codec/h264/lex.go: merge logic for start end packet writing into first if

codec/h264/lex.go

@@ -266,24 +266,20 @@ func (l *RTPLexer) handleFUA(d []byte) {
 	// If start, form new header, skip FU indicator only and set first byte to
 	// new header. Otherwise, skip over both FU indicator and FU header.
 	if start {
+		if end {
+			panic("bad fragmentation packet")
+		}
 		const FUIndicatorIdx = 0
 		newHead := (d[FUIndicatorIdx] & 0xe0) | (d[FUHeadIdx] & 0x1f)
 		d = d[1:]
 		d[0] = newHead
-	} else {
-		d = d[2:]
-	}
-
-	if start {
-		if end {
-			panic("bad fragmentation packet")
-		}
 		l.frag = true
 		l.writeWithPrefix(d)
 	} else {
 		if end {
 			l.frag = false
 		}
+		d = d[2:]
 		l.writeNoPrefix(d)
 	}
 }