Fix segfaults on streams without a codec context

CHANGELOG.rst

@@ -36,6 +36,10 @@ Features:
 - Support reusing the thread's current CUDA context via a ``current_ctx`` flag on ``CudaContext`` and ``VideoFrame.from_dlpack``, for interop with libraries like PyTorch that initialize CUDA first by :gh-user:`Yozer` (:pr:`2339`).
 - ``VideoFrame.from_dlpack`` no longer requires restating ``primary_ctx``/``current_ctx`` when passing an explicit ``cuda_context``; the flags are only validated when explicitly given by :gh-user:`WyattBlue`.
 
+Fixes:
+
+- Fix a crash when using a stream that has no ``CodecContext`` (a demuxed stream with no available decoder, such as one from a truncated file, or a stream created by ``add_mux_stream``); decoding now raises ``DecoderNotFoundError``, encoding now raises ``EncoderNotFoundError``, and ``BitStreamFilterContext`` accepts such a stream as ``out_stream`` by :gh-user:`WyattBlue`, reported by :gh-user:`justinrmiller` (:issue:`2344`).
+
 v18.0.0
 -------
 

av/audio/stream.py

@@ -1,4 +1,5 @@
 import cython
+from cython.cimports import libav as lib
 from cython.cimports.av.audio.frame import AudioFrame
 from cython.cimports.av.packet import Packet
 
@@ -31,7 +32,7 @@ def encode(self, frame: AudioFrame | None = None):
 
         .. seealso:: This is mostly a passthrough to :meth:`.CodecContext.encode`.
         """
-
+        self._assert_has_codec_context(lib.AVERROR_ENCODER_NOT_FOUND)
         packets = self.codec_context.encode(frame)
         packet: Packet
         for packet in packets:
@@ -49,5 +50,5 @@ def decode(self, packet: Packet | None = None):
 
         .. seealso:: This is a passthrough to :meth:`.CodecContext.decode`.
         """
-
+        self._assert_has_codec_context()
         return self.codec_context.decode(packet)

av/bitstream.py

@@ -61,9 +61,12 @@ def __cinit__(
                     out_stream.ptr.codecpar, self.ptr.par_out
                 )
             err_check(res)
-            lib.avcodec_parameters_to_context(
-                out_stream.codec_context.ptr, out_stream.ptr.codecpar
-            )
+            # codecpar carries everything the muxer needs; a mux-only stream
+            # (add_mux_stream) has no context to keep in sync.
+            if out_stream.codec_context is not None:
+                lib.avcodec_parameters_to_context(
+                    out_stream.codec_context.ptr, out_stream.ptr.codecpar
+                )
 
     def __dealloc__(self):
         if self.ptr:

av/stream.pxd

@@ -21,6 +21,7 @@ cdef class Stream:
 
     # Private API.
     cdef _init(self, Container, lib.AVStream*, CodecContext)
+    cdef _assert_has_codec_context(self, int err=*)
     cdef _finalize_for_output(self)
     cdef _set_id(self, value)
 

av/stream.py

@@ -128,6 +128,14 @@ def _init(
             errors=self.container.metadata_errors,
         )
 
+    @cython.cfunc
+    def _assert_has_codec_context(
+        self, err: cython.int = lib.AVERROR_DECODER_NOT_FOUND
+    ):
+        # Calling into a NULL codec_context is a segfault, not an AttributeError.
+        if self.codec_context is None:
+            err_check(err)
+
     def __repr__(self):
         name = getattr(self, "name", None)
         return (

av/subtitles/stream.py

@@ -18,6 +18,7 @@ def decode(self, packet: Packet | None = None):
 
         .. seealso:: This is a passthrough to :meth:`.CodecContext.decode`.
         """
+        self._assert_has_codec_context()
         if not packet:
             packet = Packet()
 

av/video/stream.py

@@ -40,7 +40,7 @@ def encode(self, frame: VideoFrame | None = None):
 
         .. seealso:: This is mostly a passthrough to :meth:`.CodecContext.encode`.
         """
-
+        self._assert_has_codec_context(lib.AVERROR_ENCODER_NOT_FOUND)
         packets = self.codec_context.encode(frame)
         packet: Packet
         for packet in packets:
@@ -57,6 +57,7 @@ def decode(self, packet: Packet | None = None):
 
         .. seealso:: This is a passthrough to :meth:`.CodecContext.decode`.
         """
+        self._assert_has_codec_context()
         return self.codec_context.decode(packet)
 
     @cython.cfunc

tests/test_decode.py

@@ -1,4 +1,5 @@
 import functools
+import io
 import os
 import pathlib
 from fractions import Fraction
@@ -52,6 +53,34 @@ def make_h264_test_video(path: str) -> None:
 
 
 class TestDecode(TestCase):
+    def test_decode_stream_without_codec_context(self) -> None:
+        buffer = io.BytesIO()
+        with av.open(buffer, "w", format="mp4") as output:
+            stream = output.add_mux_stream("h264", width=16, height=16)
+            packet = av.Packet(b"invalid")
+            packet.stream = stream
+            packet.pts = packet.dts = 0
+            packet.time_base = Fraction(1, 1000)
+            output.mux(packet)
+
+        # Keep the MP4 video stream while making its codec unknown to FFmpeg.
+        data = buffer.getvalue().replace(b"avc1", b"zzzz")
+        with av.open(io.BytesIO(data)) as container:
+            stream = container.streams.video[0]
+            assert stream.codec_context is None
+            with pytest.raises(av.DecoderNotFoundError):
+                list(container.decode(stream))
+
+    def test_mux_stream_without_codec_context(self) -> None:
+        with av.open(io.BytesIO(), "w", format="mp4") as output:
+            stream = output.add_mux_stream("h264", width=16, height=16)
+            assert stream.codec_context is None
+            with pytest.raises(av.EncoderNotFoundError):
+                stream.encode(None)
+
+            # A bitstream filter only needs to update codecpar for a mux stream.
+            av.BitStreamFilterContext("h264_mp4toannexb", "h264", out_stream=stream)
+
     def test_decoded_video_frame_count(self) -> None:
         container = av.open(fate_suite("h264/interlaced_crop.mp4"))
         video_stream = next(s for s in container.streams if s.type == "video")