From 61804dad1c733156d658c7f41217894382fe61f5 Mon Sep 17 00:00:00 2001
From: Aryan Salmanpour <aryan.salmanpour@amd.com>
Date: Mon, 12 Jan 2026 14:51:46 -0800
Subject: [PATCH] Add a check to detect JPEG images that contain an unsupported
 number of components for VCN (#217)

[ROCm/rocjpeg commit: 772c0ee3f5d501cad78e935084dc3d3c3534cbf7]
---
 projects/rocjpeg/src/rocjpeg_parser.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/rocjpeg/src/rocjpeg_parser.cpp b/projects/rocjpeg/src/rocjpeg_parser.cpp
index 3280edcd0a..e79a589a24 100644
--- a/projects/rocjpeg/src/rocjpeg_parser.cpp
+++ b/projects/rocjpeg/src/rocjpeg_parser.cpp
@@ -330,7 +330,7 @@ bool RocJpegStreamParser::ParseSOS() {
 
     uint32_t num_components = stream_[2];
 
-    if (num_components > NUM_COMPONENTS - 1) {
+    if (num_components > NUM_COMPONENTS - 1 || num_components != jpeg_stream_parameters_.picture_parameter_buffer.num_components) {
         ERR("invalid number of component!")
         return false;
     }