perf: verify message length for msgTypeRegister

Signed-off-by: Jianhui Zhao <[email protected]>
zgl1104 · Oct 12, 2021 · 0e7632d · 0e7632d
1 parent 79f88f5
commit 0e7632d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/device.go b/device.go
@@ -227,6 +227,11 @@ func (dev *device) readLoop() {
 
 		switch typ {
 		case msgTypeRegister:
+			if msgLen < 2 {
+				log.Error().Msg("msgTypeRegister: invalid")
+				return
+			}
+
 			if !parseDeviceInfo(dev, b) {
 				return
 			}