PT-2021-18303 · Google · Tensorflow

Yakun Zhang + 1 · Published 2021-05-14 · Updated 2024-03-06 · CVE-2021-29552

Positive Technologies (AI Summary)

CVSS v3.1

5.5

Medium

Base vector

AV:L/AC:L/PR:L/UI:N/S:U/C:N/I:N/A:H

Name of the Vulnerable Software and Affected Versions TensorFlow versions prior to 2.5.0 TensorFlow versions 2.4.2 and earlier TensorFlow versions 2.3.3 and earlier TensorFlow versions 2.2.3 and earlier TensorFlow versions 2.1.4 and earlier

Description An attacker can cause a denial of service by controlling the values of num segments tensor argument for UnsortedSegmentJoin. This is because the implementation assumes that the num segments tensor is a valid scalar. Since the tensor is empty, the CHECK involved in .scalar<T>()() that checks that the number of elements is exactly 1 will be invalidated, resulting in process termination.

Recommendations For versions prior to 2.5.0, update to TensorFlow 2.5.0 or later. For versions 2.4.2 and earlier, update to TensorFlow 2.4.2 or later. For versions 2.3.3 and earlier, update to TensorFlow 2.3.3 or later. For versions 2.2.3 and earlier, update to TensorFlow 2.2.3 or later. For versions 2.1.4 and earlier, update to TensorFlow 2.1.4 or later. As a temporary workaround, consider restricting the use of the UnsortedSegmentJoin function with empty num segments tensors until a patch is available.

Exploit

Fix

Assertion Failure

Weakness Enumeration

CWE-617

Related Identifiers

BIT-TENSORFLOW-2021-29552

CVE-2021-29552

GHSA-JHQ9-WM9M-CF89

PYSEC-2021-189

PYSEC-2021-480

PYSEC-2021-678

Affected Products

Tensorflow

References · 13

• 🔥 https://github.com/tensorflow/tensorflow/security/advisories/GHSA-jhq9-wm9m-cf89 · Exploit
• https://github.com/tensorflow/tensorflow/commit/704866eabe03a9aeda044ec91a8d0c83fc1ebdbe · Patch
• https://osv.dev/vulnerability/CVE-2021-29552 · Vendor Advisory
• https://nvd.nist.gov/vuln/detail/CVE-2021-29552 · Security Note
• https://osv.dev/vulnerability/PYSEC-2021-678 · Vendor Advisory
• https://osv.dev/vulnerability/PYSEC-2021-480 · Vendor Advisory
• https://osv.dev/vulnerability/PYSEC-2021-189 · Vendor Advisory
• https://osv.dev/vulnerability/GHSA-jhq9-wm9m-cf89 · Vendor Advisory
• https://osv.dev/vulnerability/BIT-tensorflow-2021-29552 · Vendor Advisory
• https://github.com/tensorflow/tensorflow · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-480.yaml · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-678.yaml · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-189.yaml · Note