d

broker/broker.go

@@ -0,0 +1,53 @@
+package broker
+
+import (
+	"net"
+	"time"
+
+	log "github.com/cihub/seelog"
+)
+
+type Broker struct {
+}
+
+func NewBroker() *Broker {
+	return &Broker{}
+}
+func (b *Broker) StartListening() {
+	l, e := net.Listen("tcp", "0.0.0.0:1883")
+	if e != nil {
+		log.Error("Error listening on ", e)
+		return
+	}
+	tmpDelay := 10 * ACCEPT_MIN_SLEEP
+	for {
+		conn, err := l.Accept()
+		if err != nil {
+			if ne, ok := err.(net.Error); ok && ne.Temporary() {
+				log.Error("Temporary Client Accept Error(%v), sleeping %dms",
+					ne, tmpDelay/time.Millisecond)
+				time.Sleep(tmpDelay)
+				tmpDelay *= 2
+				if tmpDelay > ACCEPT_MAX_SLEEP {
+					tmpDelay = ACCEPT_MAX_SLEEP
+				}
+			} else {
+				log.Error("Accept error: %v", err)
+			}
+			continue
+		}
+		tmpDelay = ACCEPT_MIN_SLEEP
+		go handleConnection(conn)
+	}
+}
+
+func handleConnection(conn net.Conn) {
+
+	//process connect packet
+	connMsg, err := ReadPacket(conn)
+	if err != nil {
+		log.Error("read connect packet error: ", err)
+		return
+	}
+
+}

broker/const.go

@@ -0,0 +1,31 @@
+package broker
+
+import "time"
+
+const (
+	// ACCEPT_MIN_SLEEP is the minimum acceptable sleep times on temporary errors.
+	ACCEPT_MIN_SLEEP = 100 * time.Millisecond
+	// ACCEPT_MAX_SLEEP is the maximum acceptable sleep times on temporary errors
+	ACCEPT_MAX_SLEEP = 10 * time.Second
+	// DEFAULT_ROUTE_CONNECT Route solicitation intervals.
+	DEFAULT_ROUTE_CONNECT = 5 * time.Second
+	// DEFAULT_TLS_TIMEOUT
+	DEFAULT_TLS_TIMEOUT = 5 * time.Second
+)
+
+const (
+	CONNECT = uint8(iota + 1)
+	CONNACK
+	PUBLISH
+	PUBACK
+	PUBREC
+	PUBREL
+	PUBCOMP
+	SUBSCRIBE
+	SUBACK
+	UNSUBSCRIBE
+	UNSUBACK
+	PINGREQ
+	PINGRESP
+	DISCONNECT
+)

broker/packet.go

@@ -0,0 +1,60 @@
+package broker
+
+import (
+	"errors"
+	"io"
+	"net"
+
+	log "github.com/cihub/seelog"
+)
+
+func checkError(desc string, err error) {
+	if err != nil {
+		log.Error(desc, " : ", err)
+	}
+}
+func ReadPacket(conn net.Conn) ([]byte, error) {
+	if conn == nil {
+		return nil, errors.New("conn is null")
+	}
+	// conn.SetReadDeadline(t)
+	var buf []byte
+	// read fix header
+	b := make([]byte, 1)
+	_, err := io.ReadFull(conn, b)
+	if err != nil {
+		return nil, err
+	}
+	buf = append(buf, b...)
+	// read rem msg length
+	rembuf, remlen := decodeLength(conn)
+	buf = append(buf, rembuf...)
+	// read rem msg
+	packetBytes := make([]byte, remlen)
+	_, err = io.ReadFull(conn, packetBytes)
+	if err != nil {
+		return nil, err
+	}
+	buf = append(buf, packetBytes...)
+	// log.Info("len buf: ", len(buf))
+	return buf, nil
+}
+
+func decodeLength(r io.Reader) ([]byte, int) {
+	var rLength uint32
+	var multiplier uint32
+	var buf []byte
+	b := make([]byte, 1)
+	for {
+		io.ReadFull(r, b)
+		digit := b[0]
+		buf = append(buf, b[0])
+		rLength |= uint32(digit&127) << multiplier
+		if (digit & 128) == 0 {
+			break
+		}
+		multiplier += 7
+
+	}
+	return buf, int(rLength)
+}