PT-2021-18312 · Google · Tensorflow

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

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 version 2.4.2 TensorFlow version 2.3.3 TensorFlow version 2.2.3 TensorFlow version 2.1.4

Description An attacker can cause a denial of service by exploiting a CHECK-failure coming from tf.raw ops.LoadAndRemapMatrix. This is because the implementation assumes that the ckpt path is always a valid scalar. However, an attacker can send any other tensor as the first argument of LoadAndRemapMatrix. This would cause the rank CHECK in scalar<T>()() to trigger and terminate the process.

Recommendations For TensorFlow versions prior to 2.5.0, update to TensorFlow 2.5.0 or later. For TensorFlow version 2.4.2, update to a version that includes the cherrypicked commit. For TensorFlow version 2.3.3, update to a version that includes the cherrypicked commit. For TensorFlow version 2.2.3, update to a version that includes the cherrypicked commit. For TensorFlow version 2.1.4, update to a version that includes the cherrypicked commit. As a temporary workaround, consider restricting the use of the tf.raw ops.LoadAndRemapMatrix function until a patch is available. Avoid using the ckpt path parameter with invalid scalars in the affected tf.raw ops.LoadAndRemapMatrix function until the issue is resolved.

Exploit

Fix

Assertion Failure

Weakness Enumeration

CWE-617

Related Identifiers

BIT-TENSORFLOW-2021-29561

CVE-2021-29561

GHSA-GVM4-H8J3-RJRQ

PYSEC-2021-198

PYSEC-2021-489

PYSEC-2021-687

Affected Products

Tensorflow

References · 13

• 🔥 https://github.com/tensorflow/tensorflow/security/advisories/GHSA-gvm4-h8j3-rjrq · Exploit
• https://github.com/tensorflow/tensorflow/commit/77dd114513d7796e1e2b8aece214a380af26fbf4 · Patch
• https://osv.dev/vulnerability/BIT-tensorflow-2021-29561 · Vendor Advisory
• https://osv.dev/vulnerability/PYSEC-2021-198 · Vendor Advisory
• https://nvd.nist.gov/vuln/detail/CVE-2021-29561 · Security Note
• https://osv.dev/vulnerability/CVE-2021-29561 · Vendor Advisory
• https://osv.dev/vulnerability/PYSEC-2021-687 · Vendor Advisory
• https://osv.dev/vulnerability/PYSEC-2021-489 · Vendor Advisory
• https://osv.dev/vulnerability/GHSA-gvm4-h8j3-rjrq · Vendor Advisory
• https://github.com/tensorflow/tensorflow · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-687.yaml · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-489.yaml · Note
• https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-198.yaml · Note