CEF/grumble
1
0
Fork 0
forked from External/grumble
Code Pull requests Activity

Make sure to always read full messages.

Browse source
This commit is contained in:
Mikkel Krautz 2011-04-11 21:14:47 +02:00
parent c3be7e6b4f
commit 91b83edf9c
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
client.go
View file

@ -13,6 +13,7 @@ import (
"goprotobuf.googlecode.com/hg/proto" "goprotobuf.googlecode.com/hg/proto"
"mumbleproto" "mumbleproto"
"cryptstate" "cryptstate"
"io"
"packetdatastream" "packetdatastream"
) )
@ -176,8 +177,10 @@ func (client *Client) RejectAuth(kind, reason string) {
// Read a protobuf message from a client // Read a protobuf message from a client
func (client *Client) readProtoMessage() (msg *Message, err os.Error) { func (client *Client) readProtoMessage() (msg *Message, err os.Error) {
var length uint32 var (
var kind uint16 length uint32
kind uint16
)
// Read the message type (16-bit big-endian unsigned integer) // Read the message type (16-bit big-endian unsigned integer)
err = binary.Read(client.reader, binary.BigEndian, &kind) err = binary.Read(client.reader, binary.BigEndian, &kind)
@ -194,7 +197,7 @@ func (client *Client) readProtoMessage() (msg *Message, err os.Error) {
} }
buf := make([]byte, length) buf := make([]byte, length)
_, err = client.reader.Read(buf) _, err = io.ReadFull(client.reader, buf)
if err != nil { if err != nil {
client.Panic("Unable to read packet content") client.Panic("Unable to read packet content")
return return