mirror of https://github.com/schollz/croc.git
read until 4 bytes
This commit is contained in:
parent
3c051f4283
commit
e54045ef5a
|
@ -65,18 +65,21 @@ func (c *Comm) Write(b []byte) (int, error) {
|
|||
}
|
||||
|
||||
func (c *Comm) Read() (buf []byte, numBytes int, bs []byte, err error) {
|
||||
// read until we get 5 bytes
|
||||
header := make([]byte, 4)
|
||||
n, err := c.connection.Read(header)
|
||||
if err != nil {
|
||||
// read until we get 4 bytes for the header
|
||||
var header []byte
|
||||
numBytes = 4
|
||||
for {
|
||||
tmp := make([]byte,numBytes-len(header))
|
||||
n, errRead := c.connection.Read(tmp)
|
||||
if errRead != nil {
|
||||
err = errRead
|
||||
return
|
||||
}
|
||||
if n < 4 {
|
||||
err = fmt.Errorf("not enough bytes: %d", n)
|
||||
return
|
||||
header = append(header, tmp[:n]...)
|
||||
if numBytes == len(header) {
|
||||
break
|
||||
}
|
||||
}
|
||||
// make it so it won't change
|
||||
header = append([]byte(nil), header...)
|
||||
|
||||
var numBytesUint32 uint32
|
||||
rbuf := bytes.NewReader(header)
|
||||
|
|
Loading…
Reference in New Issue