133 lines
3.5 KiB
Go
133 lines
3.5 KiB
Go
// Copyright 2019 PingCAP, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package server
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pingcap/tidb/parser/mysql"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func BenchmarkPacketIOWrite(b *testing.B) {
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
var outBuffer bytes.Buffer
|
|
pkt := &packetIO{bufWriter: bufio.NewWriter(&outBuffer)}
|
|
_ = pkt.writePacket([]byte{0x6d, 0x44, 0x42, 0x3a, 0x35, 0x36, 0x0, 0x0, 0x0, 0xfc, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x68, 0x54, 0x49, 0x44, 0x3a, 0x31, 0x30, 0x38, 0x0, 0xfe})
|
|
}
|
|
}
|
|
|
|
func TestPacketIOWrite(t *testing.T) {
|
|
// Test write one packet
|
|
var outBuffer bytes.Buffer
|
|
pkt := &packetIO{bufWriter: bufio.NewWriter(&outBuffer)}
|
|
err := pkt.writePacket([]byte{0x00, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03})
|
|
require.NoError(t, err)
|
|
err = pkt.flush()
|
|
require.NoError(t, err)
|
|
require.Equal(t, []byte{0x03, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03}, outBuffer.Bytes())
|
|
|
|
// Test write more than one packet
|
|
outBuffer.Reset()
|
|
largeInput := make([]byte, mysql.MaxPayloadLen+4)
|
|
pkt = &packetIO{bufWriter: bufio.NewWriter(&outBuffer)}
|
|
err = pkt.writePacket(largeInput)
|
|
require.NoError(t, err)
|
|
err = pkt.flush()
|
|
require.NoError(t, err)
|
|
res := outBuffer.Bytes()
|
|
require.Equal(t, byte(0xff), res[0])
|
|
require.Equal(t, byte(0xff), res[1])
|
|
require.Equal(t, byte(0xff), res[2])
|
|
require.Equal(t, byte(0), res[3])
|
|
}
|
|
|
|
func TestPacketIORead(t *testing.T) {
|
|
var inBuffer bytes.Buffer
|
|
_, err := inBuffer.Write([]byte{0x01, 0x00, 0x00, 0x00, 0x01})
|
|
require.NoError(t, err)
|
|
// Test read one packet
|
|
brc := newBufferedReadConn(&bytesConn{inBuffer})
|
|
pkt := newPacketIO(brc)
|
|
readBytes, err := pkt.readPacket()
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint8(1), pkt.sequence)
|
|
require.Equal(t, []byte{0x01}, readBytes)
|
|
|
|
inBuffer.Reset()
|
|
buf := make([]byte, mysql.MaxPayloadLen+9)
|
|
buf[0] = 0xff
|
|
buf[1] = 0xff
|
|
buf[2] = 0xff
|
|
buf[3] = 0
|
|
buf[2+mysql.MaxPayloadLen] = 0x00
|
|
buf[3+mysql.MaxPayloadLen] = 0x00
|
|
buf[4+mysql.MaxPayloadLen] = 0x01
|
|
buf[7+mysql.MaxPayloadLen] = 0x01
|
|
buf[8+mysql.MaxPayloadLen] = 0x0a
|
|
|
|
_, err = inBuffer.Write(buf)
|
|
require.NoError(t, err)
|
|
// Test read multiple packets
|
|
brc = newBufferedReadConn(&bytesConn{inBuffer})
|
|
pkt = newPacketIO(brc)
|
|
readBytes, err = pkt.readPacket()
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint8(2), pkt.sequence)
|
|
require.Equal(t, mysql.MaxPayloadLen+1, len(readBytes))
|
|
require.Equal(t, byte(0x0a), readBytes[mysql.MaxPayloadLen])
|
|
}
|
|
|
|
type bytesConn struct {
|
|
b bytes.Buffer
|
|
}
|
|
|
|
func (c *bytesConn) Read(b []byte) (n int, err error) {
|
|
return c.b.Read(b)
|
|
}
|
|
|
|
func (c *bytesConn) Write(b []byte) (n int, err error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (c *bytesConn) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (c *bytesConn) LocalAddr() net.Addr {
|
|
return nil
|
|
}
|
|
|
|
func (c *bytesConn) RemoteAddr() net.Addr {
|
|
return nil
|
|
}
|
|
|
|
func (c *bytesConn) SetDeadline(t time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *bytesConn) SetReadDeadline(t time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *bytesConn) SetWriteDeadline(t time.Time) error {
|
|
return nil
|
|
}
|