openp2p/core/udp.go

package openp2p

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(LvERROR, "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(LvERROR, "parse p2pheader error:", err)
		return nil, nil, nil, 0, err
	}
	return
}