#3703 use correct exception type for invalid data

Xpra-org · Aug 15, 2023 · cc37b9d · cc37b9d
1 parent 27ad5ad
commit cc37b9d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xpra/codecs/nvidia/nvdec/decoder.pyx b/xpra/codecs/nvidia/nvdec/decoder.pyx
@@ -37,6 +37,8 @@ ctypedef void *CUvideoparser
 ctypedef long long CUvideotimestamp
 
 
+DEF CUDA_ERROR_INVALID_IMAGE = 200
+
 ctypedef struct rect:
     short left
     short top
@@ -685,6 +687,8 @@ cdef class Decoder:
         pic.pBitstreamData = <const unsigned char*> (<uintptr_t> int(self.buffer))
         with nogil:
             r = cuvidDecodePicture(self.context, pic)
+        if r==CUDA_ERROR_INVALID_IMAGE:
+            raise ValueError("invalid image data")
         log(f"cuvidDecodePicture()={r}")
         cudacheck(r, "GPU picture decoding returned error")
         cdef CUVIDGETDECODESTATUS status