56 lines
973 B
Go
56 lines
973 B
Go
|
package mysqlserver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
func ReadMysqlPacketHeader(conn io.Reader) (*MysqlPacketHeader, error) {
|
||
|
|
header := make([]byte, 4)
|
||
|
|
|
||
|
|
_, err := io.ReadFull(conn, header)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
//fmt.Printf("%+v\n", header)
|
||
|
|
|
||
|
|
return &MysqlPacketHeader{
|
||
|
|
Length: ReadUint24(header[:3]),
|
||
|
|
SequenceId: ReadByte(header[3:]),
|
||
|
|
HeaderByte: header,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadMysqlPacketByLength(conn io.Reader, dataLength int) ([]byte, error) {
|
||
|
|
data := make([]byte, dataLength)
|
||
|
|
|
||
|
|
_, err := io.ReadFull(conn, data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return data, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadMysqlPacket(conn io.Reader) (*MysqlPacket, error) {
|
||
|
|
header, err := ReadMysqlPacketHeader(conn)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
//fmt.Printf("%+v\n", header)
|
||
|
|
|
||
|
|
data := make([]byte, header.Length)
|
||
|
|
|
||
|
|
_, err = io.ReadFull(conn, data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
res := &MysqlPacket{
|
||
|
|
Payload: data,
|
||
|
|
}
|
||
|
|
res.MysqlPacketHeader = *header
|
||
|
|
|
||
|
|
return res, nil
|
||
|
|
}
|