diff --git a/main.go b/main.go index c69ba9b..7adb874 100644 --- a/main.go +++ b/main.go @@ -20,16 +20,17 @@ func main() { _, err = conn.Write(p.Bytes()) failError(err, "conn.Write") logInfoDebug("Authentication sent (%v)", p.Bytes()) + r := bufio.NewReader(conn) for { p := Packet{} - b, err := conn.ReadByte() + b, err := r.ReadByte() logErrorDebug(err, "conn.ReadByte") p.PLength = uint8(b) << 8 - b, err = conn.ReadByte() + b, err = r.ReadByte() logErrorDebug(err, "conn.ReadByte") p.PLength += uint8(b) - b, err = conn.ReadByte() + b, err = r.ReadByte() logErrorDebug(err, "conn.ReadByte") p.PType = uint8(b) if p.PLength <= 3 { @@ -39,7 +40,7 @@ func main() { d := make([]byte, p.PLength-3) for { - _, err := conn.Read(d[len(d):]) + _, err := r.Read(d[len(d):]) logErrorDebug(err, "conn.Read") if len(d) == int(p.PLength-3) { break