Fix panic on h264.EmitNalus #1076

AlexxIT · May 5, 2024 · 4c3de3b · 4c3de3b
1 parent 4933c14
commit 4c3de3b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/h264/payloader.go b/pkg/h264/payloader.go
@@ -67,11 +67,15 @@ func EmitNalus(nals []byte, isAVC bool, emit func([]byte)) {
 		}
 	} else {
 		for {
+			n := uint32(len(nals))
+			if n < 4 {
+				break
+			}
 			end := 4 + binary.BigEndian.Uint32(nals)
-			emit(nals[4:end])
-			if int(end) >= len(nals) {
+			if n < end {
 				break
 			}
+			emit(nals[4:end])
 			nals = nals[end:]
 		}
 	}