yuxh/gearman-go
1
0
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Check if buffer read is complete before decoding

Browse Source
This commit is contained in:
Dan Sparks 2014-09-29 18:01:50 -07:00 committed by Jian Cheung
parent f22d6d7e8d
commit bc27f6f7db
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
worker/agent.go
View File

@ -86,7 +86,8 @@ func (a *agent) work() {
if len(leftdata) > 0 { // some data left for processing if len(leftdata) > 0 { // some data left for processing
data = append(leftdata, data...) data = append(leftdata, data...)
} }
if len(data) < minPacketLength { // not enough data length := len(data) - minPacketLength
if length < 0 || length < int(binary.BigEndian.Uint32(data[8:12])) {
leftdata = data leftdata = data
continue continue
} }

2
worker/inpack.go
View File

@ -86,6 +86,8 @@ func (inpack *inPack) UpdateStatus(numerator, denominator int) {
// Decode job from byte slice // Decode job from byte slice
func decodeInPack(data []byte) (inpack *inPack, l int, err error) { func decodeInPack(data []byte) (inpack *inPack, l int, err error) {
// The next three checks should be completely unnecessary, as they are checked in
// agent.work.
if len(data) < minPacketLength { // valid package should not less 12 bytes if len(data) < minPacketLength { // valid package should not less 12 bytes
err = fmt.Errorf("Invalid data: %v", data) err = fmt.Errorf("Invalid data: %v", data)
return return