init

udp.go

@@ -0,0 +1,44 @@
+package main
+
+import (
+	"bytes"
+	"encoding/binary"
+	"net"
+	"time"
+)
+
+func UDPWrite(conn *net.UDPConn, dst net.Addr, mainType uint16, subType uint16, packet interface{}) (len int, err error) {
+	msg, err := newMessage(mainType, subType, packet)
+	if err != nil {
+		return 0, err
+	}
+	if dst == nil {
+		return conn.Write(msg)
+	}
+	return conn.WriteTo(msg, dst)
+}
+
+func UDPRead(conn *net.UDPConn, timeout int) (ra net.Addr, head *openP2PHeader, result []byte, len int, err error) {
+	if timeout > 0 {
+		deadline := time.Now().Add(time.Millisecond * time.Duration(timeout))
+		err = conn.SetReadDeadline(deadline)
+		if err != nil {
+			gLog.Println(LevelERROR, "SetReadDeadline error")
+			return nil, nil, nil, 0, err
+		}
+	}
+
+	result = make([]byte, 1024)
+	len, ra, err = conn.ReadFrom(result)
+	if err != nil {
+		// gLog.Println(LevelDEBUG, "ReadFrom error")
+		return nil, nil, nil, 0, err
+	}
+	head = &openP2PHeader{}
+	err = binary.Read(bytes.NewReader(result[:openP2PHeaderSize]), binary.LittleEndian, head)
+	if err != nil {
+		gLog.Println(LevelERROR, "parse p2pheader error:", err)
+		return nil, nil, nil, 0, err
+	}
+	return
+}