init

vendor/github.com/siddontang/go-mysql/packet/BUILD.bazel

@@ -0,0 +1,27 @@
+load("@io_bazel_rules_go//go:def.bzl", "go_library")
+
+go_library(
+    name = "go_default_library",
+    srcs = ["conn.go"],
+    importmap = "go-common/vendor/github.com/siddontang/go-mysql/packet",
+    importpath = "github.com/siddontang/go-mysql/packet",
+    visibility = ["//visibility:public"],
+    deps = [
+        "//vendor/github.com/juju/errors:go_default_library",
+        "//vendor/github.com/siddontang/go-mysql/mysql:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

vendor/github.com/siddontang/go-mysql/packet/conn.go

@@ -0,0 +1,161 @@
+package packet
+
+import (
+	"bufio"
+	"bytes"
+	"io"
+	"net"
+
+	"github.com/juju/errors"
+	. "github.com/siddontang/go-mysql/mysql"
+)
+
+/*
+	Conn is the base class to handle MySQL protocol.
+*/
+type Conn struct {
+	net.Conn
+	br *bufio.Reader
+
+	Sequence uint8
+}
+
+func NewConn(conn net.Conn) *Conn {
+	c := new(Conn)
+
+	c.br = bufio.NewReaderSize(conn, 4096)
+	c.Conn = conn
+
+	return c
+}
+
+func (c *Conn) ReadPacket() ([]byte, error) {
+	var buf bytes.Buffer
+
+	if err := c.ReadPacketTo(&buf); err != nil {
+		return nil, errors.Trace(err)
+	} else {
+		return buf.Bytes(), nil
+	}
+
+	// header := []byte{0, 0, 0, 0}
+
+	// if _, err := io.ReadFull(c.br, header); err != nil {
+	// 	return nil, ErrBadConn
+	// }
+
+	// length := int(uint32(header[0]) | uint32(header[1])<<8 | uint32(header[2])<<16)
+	// if length < 1 {
+	// 	return nil, fmt.Errorf("invalid payload length %d", length)
+	// }
+
+	// sequence := uint8(header[3])
+
+	// if sequence != c.Sequence {
+	// 	return nil, fmt.Errorf("invalid sequence %d != %d", sequence, c.Sequence)
+	// }
+
+	// c.Sequence++
+
+	// data := make([]byte, length)
+	// if _, err := io.ReadFull(c.br, data); err != nil {
+	// 	return nil, ErrBadConn
+	// } else {
+	// 	if length < MaxPayloadLen {
+	// 		return data, nil
+	// 	}
+
+	// 	var buf []byte
+	// 	buf, err = c.ReadPacket()
+	// 	if err != nil {
+	// 		return nil, ErrBadConn
+	// 	} else {
+	// 		return append(data, buf...), nil
+	// 	}
+	// }
+}
+
+func (c *Conn) ReadPacketTo(w io.Writer) error {
+	header := []byte{0, 0, 0, 0}
+
+	if _, err := io.ReadFull(c.br, header); err != nil {
+		return ErrBadConn
+	}
+
+	length := int(uint32(header[0]) | uint32(header[1])<<8 | uint32(header[2])<<16)
+	if length < 1 {
+		return errors.Errorf("invalid payload length %d", length)
+	}
+
+	sequence := uint8(header[3])
+
+	if sequence != c.Sequence {
+		return errors.Errorf("invalid sequence %d != %d", sequence, c.Sequence)
+	}
+
+	c.Sequence++
+	if n, err := io.CopyN(w, c.br, int64(length)); err != nil {
+		return ErrBadConn
+	} else if n != int64(length) {
+		return ErrBadConn
+	} else {
+		if length < MaxPayloadLen {
+			return nil
+		}
+		if err := c.ReadPacketTo(w); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+// data already has 4 bytes header
+// will modify data inplace
+func (c *Conn) WritePacket(data []byte) error {
+	length := len(data) - 4
+
+	for length >= MaxPayloadLen {
+		data[0] = 0xff
+		data[1] = 0xff
+		data[2] = 0xff
+
+		data[3] = c.Sequence
+
+		if n, err := c.Write(data[:4+MaxPayloadLen]); err != nil {
+			return ErrBadConn
+		} else if n != (4 + MaxPayloadLen) {
+			return ErrBadConn
+		} else {
+			c.Sequence++
+			length -= MaxPayloadLen
+			data = data[MaxPayloadLen:]
+		}
+	}
+
+	data[0] = byte(length)
+	data[1] = byte(length >> 8)
+	data[2] = byte(length >> 16)
+	data[3] = c.Sequence
+
+	if n, err := c.Write(data); err != nil {
+		return ErrBadConn
+	} else if n != len(data) {
+		return ErrBadConn
+	} else {
+		c.Sequence++
+		return nil
+	}
+}
+
+func (c *Conn) ResetSequence() {
+	c.Sequence = 0
+}
+
+func (c *Conn) Close() error {
+	c.Sequence = 0
+	if c.Conn != nil {
+		return c.Conn.Close()
+	}
+	return nil
+}