diff --git a/ble_handler.go b/ble_handler.go new file mode 100644 index 0000000..0ce3980 --- /dev/null +++ b/ble_handler.go @@ -0,0 +1,98 @@ +package main + +import ( + "fmt" + "log" + + "blegw/workers" + + "github.com/currantlabs/gatt" + "github.com/currantlabs/gatt/examples/option" +) + +var done = make(chan struct{}) + +type BleService struct { + workers []workers.Worker +} + +// AddWorker - add new worker +func (s BleService) AddWorker(worker workers.Worker) { + if s.workers == nil { + s.workers = []workers.Worker{} + } + s.workers = append(s.workers, worker) +} + +func (s BleService) Init() error { + d, err := gatt.NewDevice(option.DefaultClientOptions...) + if err != nil { + log.Fatalf("Failed to open device, err: %s\n", err) + return err + } + + // Register handlers. + d.Handle( + gatt.PeripheralDiscovered(s.onPeriphDiscovered), + gatt.PeripheralConnected(s.onPeriphConnected), + gatt.PeripheralDisconnected(s.onPeriphDisconnected), + ) + + d.Init(s.onStateChanged) + <-done + fmt.Println("Done") + + return nil +} + +func (s BleService) onStateChanged(d gatt.Device, state gatt.State) { + fmt.Println("State:", state) + switch state { + case gatt.StatePoweredOn: + fmt.Println("scanning...") + d.Scan([]gatt.UUID{}, false) + return + default: + d.StopScanning() + } +} + +func (s BleService) onPeriphDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) { + fmt.Printf("\nPeripheral ID:%s, NAME:(%s)\n", p.ID(), p.Name()) + fmt.Println(" Local Name =", a.LocalName) + fmt.Println(" TX Power Level =", a.TxPowerLevel) + fmt.Println(" Manufacturer Data =", a.ManufacturerData) + fmt.Println(" Service Data =", a.ServiceData) + fmt.Println(" RSSI =", rssi) + + isConnect := false + for _, worker := range s.workers { + ret := worker.OnDiscovery(p, a, rssi) + if ret && !isConnect { + isConnect = true + } + } + + if isConnect { + p.Device().Connect(p) + } +} + +func (s BleService) onPeriphConnected(p gatt.Peripheral, err error) { + fmt.Println("Connected") + defer p.Device().CancelConnection(p) + + if err := p.SetMTU(500); err != nil { + fmt.Printf("Failed to set MTU, err: %s\n", err) + } + fmt.Printf("Set mtu done.\n") + + for _, worker := range s.workers { + worker.OnConnect(p, err) + } +} + +func (s BleService) onPeriphDisconnected(p gatt.Peripheral, err error) { + fmt.Println("Disconnected") + close(done) +} diff --git a/blegw b/blegw index 1b1700e..b7a1975 100755 Binary files a/blegw and b/blegw differ diff --git a/main.go b/main.go index 65f137d..3be1c70 100644 --- a/main.go +++ b/main.go @@ -1,155 +1,12 @@ package main -import ( - "fmt" - "log" - "strings" - "time" - - "github.com/currantlabs/gatt" - "github.com/currantlabs/gatt/examples/option" -) - -var done = make(chan struct{}) - -func onStateChanged(d gatt.Device, s gatt.State) { - fmt.Println("State:", s) - switch s { - case gatt.StatePoweredOn: - fmt.Println("scanning...") - d.Scan([]gatt.UUID{}, false) - return - default: - d.StopScanning() - } -} - -func onPeriphDiscovered(p gatt.Peripheral, a *gatt.Advertisement, rssi int) { - ID := p.ID() - //xiaomi scale - if !strings.HasPrefix(ID, "88:0F:10") { - return - } - - fmt.Printf("\nPeripheral ID:%s, NAME:(%s)\n", ID, p.Name()) - - fmt.Printf("%v+", a) - fmt.Println(" Local Name =", a.LocalName) - fmt.Println(" TX Power Level =", a.TxPowerLevel) - fmt.Println(" Manufacturer Data =", a.ManufacturerData) - fmt.Println(" Service Data =", a.ServiceData) - fmt.Println(" RSSI =", rssi) - - p.Device().Connect(p) -} - -func onPeriphConnected(p gatt.Peripheral, err error) { - var WEIGHT_MEASUREMENT_SERVICE = gatt.MustParseUUID("181d") - var WEIGHT_MEASUREMENT_HISTORY_CHARACTERISTIC = gatt.MustParseUUID("00002a2f-0000-3512-2118-0009af100700") - var WEIGHT_MEASUREMENT_TIME_CHARACTERISTIC = gatt.MustParseUUID("2a2b") - var WEIGHT_MEASUREMENT_CONFIG = gatt.MustParseUUID("2902") - - fmt.Println("Connected") - defer p.Device().CancelConnection(p) - - if err := p.SetMTU(500); err != nil { - fmt.Printf("Failed to set MTU, err: %s\n", err) - } - fmt.Printf("Set mtu done.\n") - - // Discovery services - services, err := p.DiscoverServices(nil) - fmt.Printf("Services: %v+", services) - if err != nil { - fmt.Printf("Failed to discover services, err: %s\n", err) - return - } - - service := getServiceByUUID(WEIGHT_MEASUREMENT_SERVICE, services) - fmt.Printf("Service Found %s\n", service.Name()) - chars, _ := p.DiscoverCharacteristics(nil, service) - - char := getCharByUUID(WEIGHT_MEASUREMENT_TIME_CHARACTERISTIC, chars) - b, err := p.ReadCharacteristic(char) - if err != nil { - fmt.Printf("Failed to read characteristic, err: %s\n", err) - return - } - fmt.Printf(" value %x | %q\n", b, b) - - histChar := getCharByUUID(WEIGHT_MEASUREMENT_HISTORY_CHARACTERISTIC, chars) - descriptors, _ := p.DiscoverDescriptors(nil, histChar) - descriptor := getDescriptorByUUID(WEIGHT_MEASUREMENT_CONFIG, descriptors) - p.WriteDescriptor(descriptor, []byte{0x02, 0x00}) - - p.SetNotifyValue(histChar, func(c *gatt.Characteristic, b []byte, e error) { - log.Printf("Got back from %s: %#x\n", c.UUID(), b) - //weight := - if len(b) == 20 { - first := b[0:10] - parseData(first) - second := b[10:20] - parseData(second) - } else if len(b) == 10 { - parseData(b) - } - }) - p.WriteCharacteristic(histChar, []byte{0x02}, true) - - fmt.Printf("Waiting for 5 seconds to get some notifiations, if any.\n") - time.Sleep(10 * time.Second) -} - -func getDescriptorByUUID(uuid gatt.UUID, descriptors []*gatt.Descriptor) *gatt.Descriptor { - for _, descriptor := range descriptors { - if !descriptor.UUID().Equal(uuid) { - continue - } - return descriptor - } - return nil -} - -func getServiceByUUID(uuid gatt.UUID, services []*gatt.Service) *gatt.Service { - for _, service := range services { - if !service.UUID().Equal(uuid) { - continue - } - return service - } - return nil -} - -func getCharByUUID(uid gatt.UUID, chars []*gatt.Characteristic) *gatt.Characteristic { - for _, c := range chars { - if !c.UUID().Equal(uid) { - continue - } - return c - } - return nil -} - -func onPeriphDisconnected(p gatt.Peripheral, err error) { - fmt.Println("Disconnected") - close(done) -} +import "blegw/workers" func main() { - d, err := gatt.NewDevice(option.DefaultClientOptions...) - if err != nil { - log.Fatalf("Failed to open device, err: %s\n", err) - return - } + service := BleService{} - // Register handlers. - d.Handle( - gatt.PeripheralDiscovered(onPeriphDiscovered), - gatt.PeripheralConnected(onPeriphConnected), - gatt.PeripheralDisconnected(onPeriphDisconnected), - ) + scale := workers.Scale{} + service.AddWorker(scale) - d.Init(onStateChanged) - <-done - fmt.Println("Done") + service.Init() } diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..4f9ca7e --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +github.com/currantlabs/gatt diff --git a/scale.go b/scale.go deleted file mode 100644 index cec65c2..0000000 --- a/scale.go +++ /dev/null @@ -1,64 +0,0 @@ -package main - -import ( - "fmt" - "math" - "time" -) - -func parseData(weightBytes []byte) *Weight { - ctrlByte := weightBytes[0] - - isWeightRemoved := isBitSet(ctrlByte, 7) - isStabilized := isBitSet(ctrlByte, 5) - isLBSUnit := isBitSet(ctrlByte, 0) - isCattyUnit := isBitSet(ctrlByte, 4) - - // fmt.Printf("IsWeightRemoved: %v\n", isWeightRemoved) - // fmt.Printf("6 LSB Unknown: %v\n", isBitSet(ctrlByte, 6)) - // fmt.Printf("IsStabilized: %v\n", isStabilized) - // fmt.Printf("IsCattyOrKg: %v\n", isCattyUnit) - // fmt.Printf("3 LSB Unknown: %v\n", isBitSet(ctrlByte, 3)) - // fmt.Printf("2 LSB Unknown: %v\n", isBitSet(ctrlByte, 2)) - // fmt.Printf("1 LSB Unknown: %v\n", isBitSet(ctrlByte, 1)) - // fmt.Printf("IsLBS: %v\n", isLBSUnit) - - // // Only if the value is stabilized and the weight is *not* removed, the date is valid - if isStabilized && !isWeightRemoved { - wb4 := int(weightBytes[4]) - wb3 := int(weightBytes[3]) - year := (wb4<<8 | wb3) - month := int(weightBytes[5]) - day := int(weightBytes[6]) - hours := int(weightBytes[7]) - min := int(weightBytes[8]) - sec := int(weightBytes[9]) - - // Is the year plausible? Check if the year is in the range of 20 years... - now := time.Now() - if math.Abs(float64(now.Year()-year)) > 20 || year > now.Year() { - return nil - } - - wb1 := int(weightBytes[1]) - wb2 := int(weightBytes[2]) - weightVal := 0.0 - if isLBSUnit || isCattyUnit { - weightVal = float64(((wb2)<<8)|(wb1)) / 100.0 - } else { - weightVal = float64(((wb2)<<8)|(wb1)) / 200.0 - } - - t := time.Date(year, time.Month(month), day, hours, min, sec, 0, time.Local) - weight := &Weight{Weight: weightVal, Time: t} - - // fmt.Printf("Payload is: %#x\n", weightBytes) - fmt.Printf("[+] Year: %v, Month: %v, day: %v, hr: %v, min: %v, sec: %v = %v \n", year, month, day, hours, min, sec, weightVal) - return weight - } - return nil -} - -func isBitSet(value byte, bit uint) bool { - return (value & (1 << bit)) != 0 -} diff --git a/weight.go b/weight.go deleted file mode 100644 index 18a10fe..0000000 --- a/weight.go +++ /dev/null @@ -1,10 +0,0 @@ -package main - -import ( - "time" -) - -type Weight struct { - Time time.Time - Weight float64 -} diff --git a/workers/scaleworker.go b/workers/scaleworker.go new file mode 100644 index 0000000..b11bc14 --- /dev/null +++ b/workers/scaleworker.go @@ -0,0 +1,142 @@ +package workers + +import ( + "fmt" + "log" + "math" + "strings" + "time" + + "github.com/currantlabs/gatt" +) + +type Weight struct { + Time time.Time + Weight float64 +} + +type Scale struct { + Worker +} + +var WEIGHT_MEASUREMENT_SERVICE = gatt.MustParseUUID("181d") +var WEIGHT_MEASUREMENT_HISTORY_CHARACTERISTIC = gatt.MustParseUUID("00002a2f-0000-3512-2118-0009af100700") +var WEIGHT_MEASUREMENT_TIME_CHARACTERISTIC = gatt.MustParseUUID("2a2b") +var WEIGHT_MEASUREMENT_CONFIG = gatt.MustParseUUID("2902") + +func (s Scale) OnDiscovery(p gatt.Peripheral, a *gatt.Advertisement, rssi int) bool { + ID := p.ID() + if !strings.HasPrefix(ID, "88:0F:10") { + return false + } + + return true +} + +func (s Scale) OnConnect(p gatt.Peripheral, err error) error { + ID := p.ID() + //not our device + if !strings.HasPrefix(ID, "88:0F:10") { + return nil + } + + fmt.Println("Connected") + defer p.Device().CancelConnection(p) + + if err := p.SetMTU(500); err != nil { + fmt.Printf("Failed to set MTU, err: %s\n", err) + } + fmt.Printf("Set mtu done.\n") + + // Discovery services + services, err := p.DiscoverServices(nil) + fmt.Printf("Services: %v+", services) + if err != nil { + fmt.Printf("Failed to discover services, err: %s\n", err) + return nil + } + + service := getServiceByUUID(WEIGHT_MEASUREMENT_SERVICE, services) + fmt.Printf("Service Found %s\n", service.Name()) + chars, _ := p.DiscoverCharacteristics(nil, service) + + char := getCharByUUID(WEIGHT_MEASUREMENT_TIME_CHARACTERISTIC, chars) + b, err := p.ReadCharacteristic(char) + if err != nil { + fmt.Printf("Failed to read characteristic, err: %s\n", err) + return nil + } + fmt.Printf(" value %x | %q\n", b, b) + + histChar := getCharByUUID(WEIGHT_MEASUREMENT_HISTORY_CHARACTERISTIC, chars) + descriptors, _ := p.DiscoverDescriptors(nil, histChar) + descriptor := getDescriptorByUUID(WEIGHT_MEASUREMENT_CONFIG, descriptors) + p.WriteDescriptor(descriptor, []byte{0x02, 0x00}) + + p.SetNotifyValue(histChar, func(c *gatt.Characteristic, b []byte, e error) { + log.Printf("Got back from %s: %#x\n", c.UUID(), b) + //weight := + if len(b) == 20 { + first := b[0:10] + parseData(first) + second := b[10:20] + parseData(second) + } else if len(b) == 10 { + parseData(b) + } + }) + p.WriteCharacteristic(histChar, []byte{0x02}, true) + + fmt.Printf("Waiting for 5 seconds to get some notifiations, if any.\n") + time.Sleep(10 * time.Second) + + return nil +} + +func parseData(weightBytes []byte) *Weight { + ctrlByte := weightBytes[0] + + isWeightRemoved := isBitSet(ctrlByte, 7) + isStabilized := isBitSet(ctrlByte, 5) + isLBSUnit := isBitSet(ctrlByte, 0) + isCattyUnit := isBitSet(ctrlByte, 4) + + // Only if the value is stabilized and the weight is *not* removed, the date is valid + if isStabilized && !isWeightRemoved { + wb4 := int(weightBytes[4]) + wb3 := int(weightBytes[3]) + year := (wb4<<8 | wb3) + month := int(weightBytes[5]) + day := int(weightBytes[6]) + hours := int(weightBytes[7]) + min := int(weightBytes[8]) + sec := int(weightBytes[9]) + + // Is the year plausible? Check if the year is in the range of 20 years... + now := time.Now() + if math.Abs(float64(now.Year()-year)) > 20 || year > now.Year() { + return nil + } + + wb1 := int(weightBytes[1]) + wb2 := int(weightBytes[2]) + weightVal := 0.0 + if isLBSUnit || isCattyUnit { + weightVal = float64(((wb2)<<8)|(wb1)) / 100.0 + } else { + weightVal = float64(((wb2)<<8)|(wb1)) / 200.0 + } + + t := time.Date(year, time.Month(month), day, hours, min, sec, 0, time.Local) + weight := &Weight{Weight: weightVal, Time: t} + + // fmt.Printf("Payload is: %#x\n", weightBytes) + fmt.Printf("[+] Year: %v, Month: %v, day: %v, hr: %v, min: %v, sec: %v = %v \n", year, month, day, hours, min, sec, weightVal) + return weight + } + return nil +} + +func isBitSet(value byte, bit uint) bool { + return (value & (1 << bit)) != 0 +} diff --git a/workers/worker.go b/workers/worker.go new file mode 100644 index 0000000..e1885dd --- /dev/null +++ b/workers/worker.go @@ -0,0 +1,8 @@ +package workers + +import "github.com/currantlabs/gatt" + +type Worker interface { + OnDiscovery(p gatt.Peripheral, a *gatt.Advertisement, rssi int) bool + OnConnect(p gatt.Peripheral, err error) error +} diff --git a/workers/workerfuncs.go b/workers/workerfuncs.go new file mode 100644 index 0000000..d6ba72d --- /dev/null +++ b/workers/workerfuncs.go @@ -0,0 +1,33 @@ +package workers + +import "github.com/currantlabs/gatt" + +func getDescriptorByUUID(uuid gatt.UUID, descriptors []*gatt.Descriptor) *gatt.Descriptor { + for _, descriptor := range descriptors { + if !descriptor.UUID().Equal(uuid) { + continue + } + return descriptor + } + return nil +} + +func getServiceByUUID(uuid gatt.UUID, services []*gatt.Service) *gatt.Service { + for _, service := range services { + if !service.UUID().Equal(uuid) { + continue + } + return service + } + return nil +} + +func getCharByUUID(uid gatt.UUID, chars []*gatt.Characteristic) *gatt.Characteristic { + for _, c := range chars { + if !c.UUID().Equal(uid) { + continue + } + return c + } + return nil +}