修改: 封包大小使用常量

Author: davyxu

doc/customproc.md

@@ -47,7 +47,7 @@ https://github.com/davyxu/cellnet/blob/master/proc/tcp/transmitter.go
 
 功能 | 类型 | 备注
 ---|---|---
-包体大小 | uint16 | 只做UDP包完整性验证
+包体大小 | uint16 | 只做UDP包完整性验证,包含包体大小本身
 包体中的消息ID | uint16 | 包.消息名 的hash值(util.StringHash)
 包体中的用户消息数据 | []byte | 用户的消息大小，需要使用codec包解码
 

proc/udp/recv.go

@@ -5,12 +5,18 @@ import (
 	"github.com/davyxu/cellnet/codec"
 )
 
-const MTU = 1472
+const (
+	MTU       = 1472 // 最大传输单元
+	packetLen = 2    // 包体大小字段
+	msgIDLen  = 2    // 消息ID字段
+
+	headerSize = msgIDLen + msgIDLen // 整个UDP包头部分
+)
 
 func recvPacket(pktData []byte) (msg interface{}, err error) {
 
 	// 小于包头，使用nc指令测试时，为1
-	if len(pktData) < 2 {
+	if len(pktData) < packetLen {
 		return nil, nil
 	}
 
@@ -23,7 +29,7 @@ func recvPacket(pktData []byte) (msg interface{}, err error) {
 	}
 
 	// 读取消息ID
-	msgid := binary.LittleEndian.Uint16(pktData[2:])
+	msgid := binary.LittleEndian.Uint16(pktData[packetLen:])
 
 	msgData := pktData[headerSize:]
 

proc/udp/send.go

@@ -7,8 +7,6 @@ import (
 	"github.com/davyxu/cellnet/peer/udp"
 )
 
-const headerSize = 2 + 2
-
 func sendPacket(writer udp.DataWriter, ctx cellnet.ContextSet, msg interface{}) error {
 
 	// 将用户数据转换为字节数组和消息ID

tests/RunAll.sh

@@ -22,4 +22,4 @@ function Cleanup()
     rm -rf examplebin
 }
 
-trap CleanUp EXIT
+trap Cleanup EXIT

util/packet.go

@@ -15,14 +15,15 @@ var (
 )
 
 const (
-	packetSize = 2
+	bodySize  = 2 // 包体大小字段
+	msgIDSize = 2 // 消息ID字段
 )
 
 // 接收Length-Type-Value格式的封包流程
 func RecvLTVPacket(reader io.Reader, maxPacketSize int) (msg interface{}, err error) {
 
 	// Size为uint16，占2字节
-	var sizeBuffer = make([]byte, packetSize)
+	var sizeBuffer = make([]byte, bodySize)
 
 	// 持续读取Size直到读到为止
 	_, err = io.ReadFull(reader, sizeBuffer)
@@ -32,7 +33,7 @@ func RecvLTVPacket(reader io.Reader, maxPacketSize int) (msg interface{}, err er
 		return
 	}
 
-	if len(sizeBuffer) < packetSize {
+	if len(sizeBuffer) < bodySize {
 		return nil, ErrMinPacket
 	}
 
@@ -54,13 +55,13 @@ func RecvLTVPacket(reader io.Reader, maxPacketSize int) (msg interface{}, err er
 		return
 	}
 
-	if len(sizeBuffer) < packetSize {
+	if len(sizeBuffer) < bodySize {
 		return nil, ErrShortMsgID
 	}
 
 	msgid := binary.LittleEndian.Uint16(body)
 
-	msgData := body[2:]
+	msgData := body[msgIDSize:]
 
 	// 将字节数组和消息ID用户解出消息
 	msg, _, err = codec.DecodeMessage(int(msgid), msgData)
@@ -98,16 +99,16 @@ func SendLTVPacket(writer io.Writer, ctx cellnet.ContextSet, data interface{}) e
 		msgID = meta.ID
 	}
 
-	pkt := make([]byte, 2+2+len(msgData))
+	pkt := make([]byte, bodySize+msgIDSize+len(msgData))
 
 	// Length
-	binary.LittleEndian.PutUint16(pkt, uint16(2+len(msgData)))
+	binary.LittleEndian.PutUint16(pkt, uint16(msgIDSize+len(msgData)))
 
 	// Type
-	binary.LittleEndian.PutUint16(pkt[2:], uint16(msgID))
+	binary.LittleEndian.PutUint16(pkt[msgIDSize:], uint16(msgID))
 
 	// Value
-	copy(pkt[2+2:], msgData)
+	copy(pkt[bodySize+msgIDSize:], msgData)
 
 	// 将数据写入Socket
 	err := WriteFull(writer, pkt)