rtmp: support empty H264 access units fro DJI (#2375)

bluenviron · Sep 24, 2023 · 4e60e5c · 4e60e5c
1 parent f1e28d0
commit 4e60e5c
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/internal/rtmp/reader.go b/internal/rtmp/reader.go
@@ -480,23 +480,29 @@ func (r *Reader) OnDataH265(cb OnDataH26xFunc) {
 				return fmt.Errorf("unable to decode AVCC: %v", err)
 			}
 
-			cb(msg.DTS+msg.PTSDelta, au)
+			if au != nil {
+				cb(msg.DTS+msg.PTSDelta, au)
+			}
 
 		case *message.ExtendedFramesX:
 			au, err := h264.AVCCUnmarshal(msg.Payload)
 			if err != nil {
 				return fmt.Errorf("unable to decode AVCC: %v", err)
 			}
 
-			cb(msg.DTS, au)
+			if au != nil {
+				cb(msg.DTS, au)
+			}
 
 		case *message.ExtendedCodedFrames:
 			au, err := h264.AVCCUnmarshal(msg.Payload)
 			if err != nil {
 				return fmt.Errorf("unable to decode AVCC: %v", err)
 			}
 
-			cb(msg.DTS+msg.PTSDelta, au)
+			if au != nil {
+				cb(msg.DTS+msg.PTSDelta, au)
+			}
 		}
 
 		return nil
@@ -528,7 +534,9 @@ func (r *Reader) OnDataH264(cb OnDataH26xFunc) {
 					return fmt.Errorf("unable to decode AVCC: %v", err)
 				}
 
-				cb(msg.DTS+msg.PTSDelta, au)
+				if au != nil {
+					cb(msg.DTS+msg.PTSDelta, au)
+				}
 			}
 		}