12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- package xugu
- import (
- "errors"
- "fmt"
- "net"
- "time"
- )
- var (
- ErrBusyBuffer = errors.New("busy buffer")
- )
- type buffer struct {
- buf []byte
- conn net.Conn
- idx int64
- length int
- timeout time.Duration
- }
- func newBuffer(nc net.Conn) buffer {
- return buffer{
- buf: make([]byte, 2048),
- conn: nc,
- }
- }
- func (b *buffer) peekChar() byte {
- ret := b.buf[b.idx]
-
- return ret
- }
- func (b *buffer) reset() {
- b.idx = 0
- b.length = 0
- b.buf = make([]byte, 2048)
- }
- func (b *buffer) readNext(need int, reverse bool) ([]byte, error) {
-
- if len(b.buf[b.idx:b.length]) < need {
- buffer := make([]byte, need+1)
- b.buf = append(b.buf[:b.length], buffer...)
- n, err := b.conn.Read(b.buf[b.length:])
- if err != nil {
-
-
-
- return nil, err
- }
- b.length += n
- for b.length-int(b.idx) < need {
- n, err := b.conn.Read(b.buf[b.length:])
- if err != nil {
-
-
-
- fmt.Println("Error reading from server:", err)
- return nil, err
- }
-
- b.length += n
- }
- }
- offset := b.idx
- b.idx += int64(need)
- if GlobalIsBig {
- reverse = false
- }
- if reverse {
-
-
- tmp := reverseBytes(b.buf[offset:b.idx])
-
- return tmp, nil
- } else {
-
-
- return b.buf[offset:b.idx], nil
- }
- }
|