网络编程【2】自定义消息格式实现装包与拆包

分手后的思念是犯贱 2022-10-22 04:30 57阅读 0赞

**如何获取完整的数据报文?**

一、数据编码:
    	- 定义数据格式(msg_header+content_len+content)
    	- 消息编码 encode
    	- 消息解码 decode
    二、tcp_client:
      	- 1.连接服务器
      	- 2.数据编码
    三、tcp_server:
      	- 1.创建监听连接 listener
      	- 2.等待接收握手请求 accept
      	- 3.握手成功后得套接字 socket(conn)
      	- 4.数据解码 decode(conn)

**一、数据编码(unpack/codec.go)**

// 自定义消息头
    const Msg_Header = "jstang9527"
    
    // 消息编码(消息头+消息长度+消息内容)
    func Encode(bytesBuffer io.Writer, content string) error { 
    	// 1. 向缓冲区(套接字)写入消息头
    	if err := binary.Write(bytesBuffer, binary.BigEndian, []byte(Msg_Header)); err != nil { 
    		return err
    	}
    	// 2. 向缓冲区(套接字)写入消息长度
    	if err := binary.Write(bytesBuffer, binary.BigEndian, int32(len([]byte(content)))); err != nil { 
    		return err
    	}
    	// 3. 向缓冲区(套接字)写入消息内容
    	return binary.Write(bytesBuffer, binary.BigEndian, []byte(content))
    }
    
    // 消息解码(读多少字节，缓冲区就会少多少字节)
    func Decode(bytesBuffer io.Reader) (content []byte, err error) { 
    	// 1.1 准备一个载体接收消息头, 从缓冲区读取消息头
    	headerBuf := make([]byte, len(Msg_Header))
    	if _, err := io.ReadFull(bytesBuffer, headerBuf); err != nil { 
    		return nil, err
    	}
    	// 1.2 判断消息头是否对上自定义消息头
    	if string(headerBuf) != Msg_Header { 
    		return nil, errors.New("msg_header error")
    	}
    	// 2 准备一个载体接收消息长度，从剩下的缓冲区继续读取消息长度(int32是4字节长度)
    	contentLenBuf := make([]byte, 4)
    	if _, err := io.ReadFull(bytesBuffer, contentLenBuf); err != nil { 
    		return nil, err
    	}
    	// 3 利用消息长度准备一个等长的载体(长度字节数组需要解码), 从剩余的缓冲区读取消息
    	contentLen := binary.BigEndian.Uint32(contentLenBuf)
    	content = make([]byte, contentLen)
    	_, err = io.ReadFull(bytesBuffer, content)
    	return content, err
    }

**二、TCP客户端(tcp\_client/main.go)**

func main() { 
    	conn, err := net.Dial("tcp", "127.0.0.1:9090")
    	if err != nil { 
    		log.Fatal(err)
    	}
    	defer conn.Close()
    	unpack.Encode(conn, "hello unpack")
    }

**三、TCP服务端(tcp\_server/main.go)**

func main() { 
    	listener, err := net.Listen("tcp", "127.0.0.1:9090")
    	if err != nil { 
    		log.Fatal(err)
    	}
    	defer listener.Close()
    
    	for { 
    		accept, err := listener.Accept()
    		if err != nil { 
    			log.Println(err)
    			break
    		}
    		data, err := unpack.Decode(accept)
    		if err != nil { 
    			log.Println(err)
    		}
    		fmt.Println(string(data))
    	}
    }