Patch TF Inference Images (#5610)

DevakiBolleneni · DevakiBolleneni · web-flow · commit 43000050f3f7 · 2026-01-28T20:27:44.000-08:00
* Patch TF 2.19 INF SM Image

* Patch TF 2.19 INF SM Image

* Patch TF 2.19 INF SM Image

* Patch TF 2.19 INF SM Image

* Patch TF 2.19 INF SM Image

* Patch TF 2.19 INF ARM64 Image

* revert toml file

---------

Co-authored-by: DevakiBolleneni &lt;devakib@amazon.com&gt;
diff --git a/tensorflow/inference/buildspec-2-19-sm.yml b/tensorflow/inference/buildspec-2-19-sm.yml
@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
 version: &VERSION 2.19.0
 short_version: &SHORT_VERSION 2.19
 arch_type: x86
-autopatch_build: "True"
+#autopatch_build: "True"
 
 repository_info:
   inference_repository: &INFERENCE_REPOSITORY
diff --git a/tensorflow/inference/buildspec-arm64-2-19-sm.yml b/tensorflow/inference/buildspec-arm64-2-19-sm.yml
@@ -5,7 +5,7 @@ framework: &FRAMEWORK tensorflow
 version: &VERSION 2.19.0
 short_version: &SHORT_VERSION 2.19
 arch_type: arm64
-autopatch_build: "True"
+#autopatch_build: "True"
 
 repository_info:
   inference_repository: &INFERENCE_REPOSITORY
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.arm64.cpu b/tensorflow/inference/docker/2.19/py3/Dockerfile.arm64.cpu
@@ -86,7 +86,7 @@ RUN wget https://www.python.org/ftp/python/$PYTHON_VERSION/Python-$PYTHON_VERSIO
 
 RUN ${PIP} --no-cache-dir install --upgrade \
     pip \
-    setuptools
+    "setuptools>=75.0.0"
 
 # cython, falcon, gunicorn, grpc
 RUN ${PIP} install --no-cache-dir \
@@ -96,7 +96,7 @@ RUN ${PIP} install --no-cache-dir \
     gevent \
     requests \
     grpcio \
-    "protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.20.3" \
+    "protobuf==6.33.4" \
     packaging \
 # using --no-dependencies to avoid installing tensorflow binary
  && ${PIP} install --no-dependencies --no-cache-dir \
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.cpu b/tensorflow/inference/docker/2.19/py3/Dockerfile.cpu
@@ -96,7 +96,7 @@ RUN apt-get update \
 
 RUN ${PIP} --no-cache-dir install --upgrade \
     pip \
-    setuptools
+    "setuptools>=75.0.0"
 
 RUN ${PIP} install --no-cache-dir \
     "awscli<2" \
@@ -105,7 +105,7 @@ RUN ${PIP} install --no-cache-dir \
     gevent \
     requests \
     grpcio \
-    "protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.20.3" \
+    "protobuf==6.33.4" \
     packaging \
 # using --no-dependencies to avoid installing tensorflow binary
  && ${PIP} install --no-dependencies --no-cache-dir \
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.arm64.cpu.core_packages.json b/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.arm64.cpu.core_packages.json
@@ -13,12 +13,12 @@
       "version_specifier":">=1.24.3,<2.0"
    },
    "protobuf":{
-      "version_specifier":">=3.20.3,<6.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
+      "version_specifier":">=3.20.3,<7.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
    },
    "falcon":{
       "version_specifier":"==3.1.0"
    },
    "gunicorn":{
       "version_specifier":">=22.0.0"
    }
-}
+}
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.arm64.cpu.py_scan_allowlist.json b/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.arm64.cpu.py_scan_allowlist.json
@@ -0,0 +1,3 @@
+{
+  "85151": "Protocol Buffers (protobuf) vulnerability CVE-85151 affecting versions prior to 6.33.5. Using latest available version 6.33.4 as 6.33.5 is not yet released on PyPI. Will upgrade when 6.33.5 becomes available."
+}
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.cpu.core_packages.json b/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.cpu.core_packages.json
@@ -1,6 +1,6 @@
 {
     "protobuf":{
-        "version_specifier": ">=3.20.3,<5.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
+        "version_specifier": ">=3.20.3,<7.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
     },
     "cython":{
         "version_specifier": "<3.0"
@@ -21,4 +21,4 @@
     "urllib3": {
         "version_specifier": ">=2.0.7"
     }
-}
+}
diff --git a/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.cpu.py_scan_allowlist.json b/tensorflow/inference/docker/2.19/py3/Dockerfile.sagemaker.cpu.py_scan_allowlist.json
@@ -0,0 +1,3 @@
+{
+  "85151": "Protocol Buffers (protobuf) vulnerability CVE-85151 affecting versions prior to 6.33.5. Using latest available version 6.33.4 as 6.33.5 is not yet released on PyPI. Will upgrade when 6.33.5 becomes available."
+}
diff --git a/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.gpu b/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.gpu
@@ -145,7 +145,7 @@ RUN apt-get update \
 
 RUN ${PIP} --no-cache-dir install --upgrade \
     pip \
-    setuptools
+    "setuptools>=75.0.0"
 
 # Upgrade libsasl2-2 for fixing cyrus-sasl2 related CVE
 RUN apt-get install -y --only-upgrade libsasl2-2
@@ -182,7 +182,7 @@ RUN ${PIP} install -U --no-cache-dir \
     gevent \
     requests \
     grpcio \
-    "protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<5.0.0dev,>=3.20.3" \
+    "protobuf==6.33.4" \
     packaging \
 # using --no-dependencies to avoid installing tensorflow binary
  && ${PIP} install --no-dependencies --no-cache-dir \
diff --git a/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.core_packages.json b/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.core_packages.json
@@ -1,6 +1,6 @@
 {
     "protobuf":{
-        "version_specifier": ">=3.20.3,<5.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
+        "version_specifier": ">=3.20.3,<7.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"
     },
     "cython":{
         "version_specifier": "<3.0"
@@ -21,4 +21,4 @@
     "urllib3": {
         "version_specifier": ">=2.0.7"
     }
-}
+}
diff --git a/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.os_scan_allowlist.json b/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.os_scan_allowlist.json
@@ -474,6 +474,31 @@
       "status": "ACTIVE",
       "title": "CVE-2022-2068 - openssl",
       "reason_to_ignore": "N/A"
+    },
+    {
+      "description": "Issue summary: An application trying to decrypt CMS messages encrypted using password based encryption can trigger an out-of-bounds read and write. Impact summary: This out-of-bounds read may trigger a crash which leads to Denial of Service for an application. The out-of-bounds write can cause a memory corruption which can have various consequences including a Denial of Service or Execution of attacker-supplied code. Although the consequences of a successful exploit of this vulnerability could be severe, the probability that the attacker would be able to perform it is low. Besides, password based (PWRI) encryption support in CMS messages is very rarely used. For that reason the issue was assessed as Moderate severity according to our Security Policy. The FIPS modules in 3.5, 3.4, 3.3, 3.2, 3.1 and 3.0 are not affected by this issue, as the CMS implementation is outside the OpenSSL FIPS module boundary.",
+      "vulnerability_id": "CVE-2025-9230",
+      "name": "CVE-2025-9230",
+      "package_name": "openssl",
+      "package_details": {
+        "file_path": null,
+        "name": "openssl",
+        "package_manager": "OS",
+        "version": "1.1.1",
+        "release": "1ubuntu2.1~18.04.23"
+      },
+      "remediation": { "recommendation": { "text": "None Provided" } },
+      "cvss_v3_score": 7.5,
+      "cvss_v30_score": 0.0,
+      "cvss_v31_score": 7.5,
+      "cvss_v2_score": 0.0,
+      "cvss_v3_severity": "HIGH",
+      "source_url": "https://people.canonical.com/~ubuntu-security/cve/2025/CVE-2025-9230.html",
+      "source": "UBUNTU_CVE",
+      "severity": "HIGH",
+      "status": "ACTIVE",
+      "title": "CVE-2025-9230 - openssl",
+      "reason_to_ignore": "N/A"
     }
   ],
   "dpkg": [
diff --git a/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.py_scan_allowlist.json b/tensorflow/inference/docker/2.19/py3/cu122/Dockerfile.sagemaker.gpu.py_scan_allowlist.json
@@ -0,0 +1,3 @@
+{
+  "85151": "Protocol Buffers (protobuf) vulnerability CVE-85151 affecting versions prior to 6.33.5. Using latest available version 6.33.4 as 6.33.5 is not yet released on PyPI. Will upgrade when 6.33.5 becomes available."
+}

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+{`
	`2`	`+ "85151": "Protocol Buffers (protobuf) vulnerability CVE-85151 affecting versions prior to 6.33.5. Using latest available version 6.33.4 as 6.33.5 is not yet released on PyPI. Will upgrade when 6.33.5 becomes available."`
	`3`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"protobuf":{`
`3`		`- "version_specifier": ">=3.20.3,<5.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"`
	`3`	`+ "version_specifier": ">=3.20.3,<7.0.0dev,!=4.21.5,!=4.21.4,!=4.21.3,!=4.21.2,!=4.21.1,!=4.21.0"`
`4`	`4`	`},`
`5`	`5`	`"cython":{`
`6`	`6`	`"version_specifier": "<3.0"`
`@@ -21,4 +21,4 @@`
`21`	`21`	`"urllib3": {`
`22`	`22`	`"version_specifier": ">=2.0.7"`
`23`	`23`	`}`
`24`		`-}`
	`24`	`+}`