123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- package xugu
- import "encoding/binary"
- var BUFF_SIZE = 8 * 1024
- func EncryptBuff(pconn *xugusqlConn, data []byte) {
-
- }
- func sockSendCommand0(conn *xugusqlConn, sql string) (int, error) {
- cmdLen := len(sql)
-
- ret, err := RhSend(conn, []byte("?"), cmdLen)
- if err != nil {
- return 0, err
- }
-
- lengthInt32 := uint32(conn.sendBuff.Len())
- ret, err = rhSendInt32(conn, lengthInt32)
- if err != nil {
- return 0, err
- }
-
- ret, err = RhSend(conn, []byte(sql), cmdLen+1)
- if err != nil {
- return 0, err
- }
-
- ret, err = rhSendInt32(conn, 0)
- if err != nil {
- return 0, err
- }
- return ret, nil
- }
- func RhSend(pconn *xugusqlConn, buff []byte, length int) (int, error) {
- pconn.mu.Lock()
- defer pconn.mu.Unlock()
-
- if length == 0 {
- return 0, nil
- }
-
- if pconn.turingDataR != nil && pconn.useSSL {
- if length > 2*1024 {
- EncryptBuff(pconn, buff)
- if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
- pconn.sendBuff.Reset()
- return 0, err
- }
- pconn.sendBuff.Reset()
- _, err := pconn.conn.Write(buff)
- return len(buff), err
- } else if pconn.sendBuff.Len()+length >= BUFF_SIZE {
- if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
- pconn.sendBuff.Reset()
- return 0, err
- }
- pconn.sendBuff.Reset()
- _, err := pconn.sendBuff.Write(buff)
- if err != nil {
- return 0, err
- }
- EncryptBuff(pconn, pconn.sendBuff.Bytes())
- return len(buff), nil
- } else {
- pconn.sendBuff.Write(buff)
- EncryptBuff(pconn, pconn.sendBuff.Bytes()[pconn.sendBuff.Len()-length:])
- return length, nil
- }
- } else {
-
- if length > 2048 {
- if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
-
- pconn.sendBuff.Reset()
- return 0, err
- }
-
- pconn.sendBuff.Reset()
-
- _, err := pconn.conn.Write(buff)
- return length, err
- } else if pconn.sendBuff.Len()+length >= BUFF_SIZE {
-
-
- if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
- pconn.sendBuff.Reset()
- return 0, err
- }
- pconn.sendBuff.Reset()
- pconn.sendBuff.Write(buff)
- return length, nil
- } else {
-
- pconn.sendBuff.Write(buff)
- return length, nil
- }
- }
- }
- func rhSendInt32(pconn *xugusqlConn, i uint32) (int, error) {
-
- var networkBytes [4]byte
-
- binary.BigEndian.PutUint32(networkBytes[:], i)
- return RhSend(pconn, networkBytes[:], 4)
- }
- func RhFlush(pconn *xugusqlConn) (int, error) {
- if pconn.sendBuff.Len() > 0 {
- if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
- pconn.sendBuff.Reset()
- return 0, err
- }
- pconn.sendBuff.Reset()
- }
- return 0, nil
- }
|