Golang实现简单tcp服务器04 -- 服务器的粘包处理

服务器的粘包处理

什么是粘包

一个完成的消息可能会被TCP拆分红多个包进行发送，也有可能把多个小的包封装成一个大的数据包发送，这个就是TCP的拆包和封包问题

TCP粘包和拆包产生的缘由

1. 应用程序写入数据的字节大小大于套接字发送缓冲区的大小

2. 进行MSS大小的TCP分段。MSS是最大报文段长度的缩写。MSS是TCP报文段中的数据字段的最大长度。数据字段加上TCP首部才等于整个的TCP报文段。因此MSS并非TCP报文段的最大长度，而是：MSS=TCP报文段长度-TCP首部长度

3. 以太网的payload大于MTU进行IP分片。MTU指：一种通讯协议的某一层上面所能经过的最大数据包大小。若是IP层有一个数据包要传，并且数据的长度比链路层的MTU大，那么IP层就会进行分片，把数据包分红托干片，让每一片都不超过MTU。注意，IP分片能够发生在原始发送端主机上，也能够发生在中间路由器上。

TCP粘包和拆包的解决策略

1. 消息定长。例如100字节。
2. 在包尾部增长回车或者空格符等特殊字符进行分割，典型的如FTP协议
3. 将消息分为消息头和消息尾。
4. 其它复杂的协议，如RTMP协议等。

参考(http://blog.csdn.net/initphp/article/details/41948919)

咱们的处理方式

解决粘包问题有多种多样的方式, 咱们这里的作法是:

• 发送方在每次发送消息时将消息长度写入一个int32做为包头一并发送出去, 咱们称之为Encode
• 接受方则先读取一个int32的长度的消息长度信息, 再根据长度读取相应长的byte数据, 称之为Decode

在实验环境中的主文件夹内, 新建一个名为codec的文件夹在其之下新建一个文件codec.go, 将咱们的Encode和Decode方法写入其中, 这里给出Encode与Decode相应的代码:

codec.go

package codec

import (
	"bufio"
	"bytes"
	"encoding/binary"
)

func Encode(message string) ([]byte, error) {
	// 读取消息的长度
	var length int32 = int32(len(message))
	var pkg *bytes.Buffer = new(bytes.Buffer)
	// 写入消息头
	err := binary.Write(pkg, binary.LittleEndian, length)
	if err != nil {
		return nil, err
	}
	// 写入消息实体
	err = binary.Write(pkg, binary.LittleEndian, []byte(message))
	if err != nil {
		return nil, err
	}

	return pkg.Bytes(), nil
}

func Decode(reader *bufio.Reader) (string, error) {
	// 读取消息的长度
	lengthByte, _ := reader.Peek(4)
	lengthBuff := bytes.NewBuffer(lengthByte)
	var length int32
	err := binary.Read(lengthBuff, binary.LittleEndian, &length)
	if err != nil {
		return "", err
	}
	if int32(reader.Buffered()) < length+4 {
		return "", err
	}
	
	// 读取消息真正的内容
	pack := make([]byte, int(4+length))
	_, err = reader.Read(pack)
	if err != nil {
		return "", err
	}
	return string(pack[4:]), nil
}

这里就不帖服务器与客户端的调用代码了, 同窗们本身动手试试~

相关源码: https://git.oschina.net/victoriest/go-simple-tcp-server.git