golang 仿python pack/unpack

十二12

　　写得不完善也不完美 尤其是高低位转换那（go和c 二进制高地位相反 需要转换，还有go int转[]byte长度是4位），希望牛人看后指导一下
　　项目需要通过socket调取 客户端是go ，服务器端是python，由于需要封包解包，就参照python写的
　　python 的pack/unpack 参考  Python使用struct处理二进制(pack和unpack用法)



package includes
import (
"bytes"
"encoding/binary"
"fmt"
_ "os"
"strconv"
"strings"
)
type Protocol struct {
Format []string
}
//封包
func (p *Protocol) Pack(args ...interface{}) []byte {
la := len(args)
ls := len(p.Format)
ret := []byte{}
if ls > 0 && la > 0 && ls == la {
for i := 0; i < ls; i++ {
if p.Format == "H" {
ret = append(ret, IntToBytes2(args.(int))...)
} else if p.Format == "I" {
ret = append(ret, IntToBytes4(args.(int))...)
} else if strings.Contains(p.Format, "s") {
num, _ := strconv.Atoi(strings.TrimRight(p.Format, "s"))
ret = append(ret, []byte(fmt.Sprintf("%s%s", args.(string), strings.Repeat("\x00", num-len(args.(string)))))...)
}
}
}
return ret
}
//解包
func (p *Protocol) UnPack(msg []byte) []interface{} {
la := len(p.Format)
ret := make([]interface{}, la)
if la > 0 {
for i := 0; i < la; i++ {
if p.Format == "H" {
ret = Bytes4ToInt(msg[0:2])
msg = msg[2:len(msg)]
} else if p.Format == "I" {
ret = Bytes4ToInt(msg[0:4])
msg = msg[4:len(msg)]
} else if strings.Contains(p.Format, "s") {
num, _ := strconv.Atoi(strings.TrimRight(p.Format, "s"))
ret = string(msg[0:num])
msg = msg[num:len(msg)]
}
}
}
return ret
}
func (p *Protocol) Size() int {
size := 0
ls := len(p.Format)
if ls > 0 {
for i := 0; i < ls; i++ {
if p.Format == "H" {
size = size + 2
} else if p.Format == "I" {
size = size + 4
} else if strings.Contains(p.Format, "s") {
num, _ := strconv.Atoi(strings.TrimRight(p.Format, "s"))
size = size + num
}
}
}
return size
}
//整形转换成字节
func IntToBytes(n int) []byte {
m := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, m)
gbyte := bytesBuffer.Bytes()
return gbyte
}
//整形转换成字节4位
func IntToBytes4(n int) []byte {
m := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, m)
gbyte := bytesBuffer.Bytes()
//c++ 高低位转换
k := 4
x := len(gbyte)
nb := make([]byte, k)
for i := 0; i < k; i++ {
nb = gbyte[x-i-1]
}
return nb
}
//整形转换成字节2位
func IntToBytes2(n int) []byte {
m := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, m)
gbyte := bytesBuffer.Bytes()
//c++ 高低位转换
k := 2
x := len(gbyte)
nb := make([]byte, k)
for i := 0; i < k; i++ {
nb = gbyte[x-i-1]
}
return nb
}
//字节转换成整形
func BytesToInt(b []byte) int {
bytesBuffer := bytes.NewBuffer(b)
var x int32
binary.Read(bytesBuffer, binary.BigEndian, &x)
return int(x)
}
//4个字节转换成整形
func Bytes4ToInt(b []byte) int {
xx := make([]byte, 4)
if len(b) == 2 {
xx = []byte{b[0], b[1], 0, 0}
} else {
xx = b
}
m := len(xx)
nb := make([]byte, 4)
for i := 0; i < 4; i++ {
nb = xx[m-i-1]
}
bytesBuffer := bytes.NewBuffer(nb)
var x int32
binary.Read(bytesBuffer, binary.BigEndian, &x)
return int(x)
}
　　
　　调用



p := new(Protocol)
p.Format = []string{"H", "H", "I", "16s", "I", "I", "I"}
h_byte := p.Pack(1, 1, 1, "abc", 1, 0, len(request_buf))
conn.Write(h_byte)
　　
　　附加：



int 转 二进制 fmt.Printf("%08b\n", 97) // 01100001

二进制转int i, _ := strconv.ParseInt("1100001", 2, 32) //97
int 转 []byte IntToBytes(1000) // [0 0 3 232]  3*256+232=1000

func IntToBytes(n int) []byte {
m := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, m)
gbyte := bytesBuffer.Bytes()
return gbyte
}
[]byte 转int BytesToInt([]byte{0,0,3,232}) //1000

func BytesToInt(b []byte) int {
bytesBuffer := bytes.NewBuffer(b)
var x int32
binary.Read(bytesBuffer, binary.BigEndian, &x)
return int(x)
}
int 转 byte byte(97) //97