update github.com/pires/go-proxyproto to v0.5.0

This commit is contained in:
fatedier
2021-06-20 23:57:41 +08:00
parent 3f11b6a082
commit fe4e9b55f3
14 changed files with 127 additions and 43 deletions

35
test/e2e/pkg/rpc/rpc.go Normal file
View File

@@ -0,0 +1,35 @@
package rpc
import (
"bufio"
"bytes"
"encoding/binary"
"errors"
"io"
)
func WriteBytes(w io.Writer, buf []byte) (int, error) {
out := bytes.NewBuffer(nil)
binary.Write(out, binary.BigEndian, int64(len(buf)))
out.Write(buf)
return w.Write(out.Bytes())
}
func ReadBytes(r io.Reader) ([]byte, error) {
// To compatible with UDP connection, use bufio reader here to avoid lost conent.
rd := bufio.NewReader(r)
var length int64
if err := binary.Read(rd, binary.BigEndian, &length); err != nil {
return nil, err
}
buffer := make([]byte, length)
n, err := io.ReadFull(rd, buffer)
if err != nil {
return nil, err
}
if int64(n) != length {
return nil, errors.New("invalid length")
}
return buffer, nil
}