This repository has been archived by the owner on Jun 29, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
xiaomi.go
118 lines (101 loc) · 2.21 KB
/
xiaomi.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package xiaomi
import (
"encoding/json"
"log"
"net"
)
type (
Response struct {
Cmd string `json:"cmd"`
Model string `json:"model"`
Sid string `json:"sid"`
//ShortId int `json:"short_id,integer"`
Token string `json:"token,omitempty"`
IP string `json:"ip,omitempty"`
Port string `json:"port,omitempty"`
Data interface{} `json:"data"`
}
Request struct {
Cmd string `json:"cmd"`
Sid string `json:"sid,omitempty"`
}
Gateway struct {
Addr *net.UDPAddr
Sid string
Token string
}
object map[string]interface{}
)
var (
conn *net.UDPConn
gateways map[string]Gateway
)
const (
multicastIp = "224.0.0.50"
multicastPort = "9898"
//multicastAddr = "239.255.255.250:9898"
maxDatagramSize = 8192
)
func init() {
go serveMulticastUDP(multicastIp+":"+multicastPort, connHandler, msgHandler)
}
func GetStatus() string { //test
return ""
}
func (this *Gateway) sendMessage(msg string) {
sendMessage(this.Addr, msg)
}
func sendMessage(addr *net.UDPAddr, msg string) {
req, err := json.Marshal(Request{Cmd: msg})
if err != nil {
log.Fatal(err)
}
log.Println(string(req))
log.Println(addr)
conn.WriteMsgUDP([]byte(req), nil, addr)
}
func connHandler() {
pingAddr, err := net.ResolveUDPAddr("udp", multicastIp+":4321")
if err != nil {
log.Fatal(err)
}
sendMessage(pingAddr, "whois")
}
func msgHandler(resp *Response) {
switch resp.Cmd {
case "iam":
log.Printf("%+v", resp)
_, err := net.ResolveUDPAddr("udp", resp.IP+":"+multicastPort)
if err != nil {
log.Fatal(err)
}
case "heartbeat":
log.Printf("%+v", resp)
}
}
func serveMulticastUDP(a string, connectedHandler func(), msgHandler func(resp *Response)) {
addr, err := net.ResolveUDPAddr("udp", a)
if err != nil {
log.Fatal(err)
}
conn, err = net.ListenMulticastUDP("udp", nil, addr)
if err != nil {
log.Panic(err)
}
conn.SetReadBuffer(maxDatagramSize)
for {
b := make([]byte, maxDatagramSize)
n, _, err := conn.ReadFromUDP(b)
if err != nil {
log.Fatal("ReadFromUDP failed:", err)
}
//log.Println(n, "bytes read from", src)
//log.Println(string(b[:n]))
resp := Response{}
err = json.Unmarshal(b[:n], &resp)
if err != nil {
log.Fatal(err)
}
msgHandler(&resp)
}
}