package client import ( "bytes" "encoding/binary" "github.com/juju/errors" . "github.com/siddontang/go-mysql/mysql" ) func (c *Conn) readInitialHandshake() error { data, err := c.ReadPacket() if err != nil { return errors.Trace(err) } if data[0] == ERR_HEADER { return errors.New("read initial handshake error") } if data[0] < MinProtocolVersion { return errors.Errorf("invalid protocol version %d, must >= 10", data[0]) } //skip mysql version //mysql version end with 0x00 pos := 1 + bytes.IndexByte(data[1:], 0x00) + 1 //connection id length is 4 c.connectionID = uint32(binary.LittleEndian.Uint32(data[pos : pos+4])) pos += 4 c.salt = []byte{} c.salt = append(c.salt, data[pos:pos+8]...) //skip filter pos += 8 + 1 //capability lower 2 bytes c.capability = uint32(binary.LittleEndian.Uint16(data[pos : pos+2])) pos += 2 if len(data) > pos { //skip server charset //c.charset = data[pos] pos += 1 c.status = binary.LittleEndian.Uint16(data[pos : pos+2]) pos += 2 c.capability = uint32(binary.LittleEndian.Uint16(data[pos:pos+2]))<<16 | c.capability pos += 2 //skip auth data len or [00] //skip reserved (all [00]) pos += 10 + 1 // The documentation is ambiguous about the length. // The official Python library uses the fixed length 12 // mysql-proxy also use 12 // which is not documented but seems to work. c.salt = append(c.salt, data[pos:pos+12]...) } return nil } func (c *Conn) writeAuthHandshake() error { // Adjust client capability flags based on server support capability := CLIENT_PROTOCOL_41 | CLIENT_SECURE_CONNECTION | CLIENT_LONG_PASSWORD | CLIENT_TRANSACTIONS | CLIENT_LONG_FLAG capability &= c.capability //packet length //capbility 4 //max-packet size 4 //charset 1 //reserved all[0] 23 length := 4 + 4 + 1 + 23 //username length += len(c.user) + 1 //we only support secure connection auth := CalcPassword(c.salt, []byte(c.password)) length += 1 + len(auth) if len(c.db) > 0 { capability |= CLIENT_CONNECT_WITH_DB length += len(c.db) + 1 } c.capability = capability data := make([]byte, length+4) //capability [32 bit] data[4] = byte(capability) data[5] = byte(capability >> 8) data[6] = byte(capability >> 16) data[7] = byte(capability >> 24) //MaxPacketSize [32 bit] (none) //data[8] = 0x00 //data[9] = 0x00 //data[10] = 0x00 //data[11] = 0x00 //Charset [1 byte] //use default collation id 33 here, is utf-8 data[12] = byte(DEFAULT_COLLATION_ID) //Filler [23 bytes] (all 0x00) pos := 13 + 23 //User [null terminated string] if len(c.user) > 0 { pos += copy(data[pos:], c.user) } //data[pos] = 0x00 pos++ // auth [length encoded integer] data[pos] = byte(len(auth)) pos += 1 + copy(data[pos+1:], auth) // db [null terminated string] if len(c.db) > 0 { pos += copy(data[pos:], c.db) //data[pos] = 0x00 } return c.WritePacket(data) }