PT-2021-18309 · Google · Tensorflow

Yakun Zhang

+1

·

Published

2021-05-14

·

Updated

2024-03-06

·

CVE-2021-29558

CVSS v3.1

7.8

High

VectorAV:L/AC:L/PR:L/UI:N/S:U/C:H/I:H/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 heap buffer overflow in tf.raw ops.SparseSplit because the implementation accesses an array element based on a user-controlled offset. This results in overriding values on the heap. The issue has been reported by Ying Wang and Yakun Zhang of Baidu X-Team.
Recommendations For TensorFlow versions prior to 2.5.0, update to version 2.5.0 or later. For TensorFlow version 2.4.2, apply the patch from GitHub commit 8ba6fa29cd8bf9cef9b718dc31c78c73081f5b31. For TensorFlow version 2.3.3, apply the patch from GitHub commit 8ba6fa29cd8bf9cef9b718dc31c78c73081f5b31. For TensorFlow version 2.2.3, apply the patch from GitHub commit 8ba6fa29cd8bf9cef9b718dc31c78c73081f5b31. For TensorFlow version 2.1.4, apply the patch from GitHub commit 8ba6fa29cd8bf9cef9b718dc31c78c73081f5b31. As a temporary workaround, consider avoiding the use of tf.raw ops.SparseSplit until a patch is applied.

Exploit

Fix

Memory Corruption

Weakness Enumeration

Related Identifiers

BIT-TENSORFLOW-2021-29558
CVE-2021-29558
GHSA-MQH2-9WRP-VX84
PYSEC-2021-195
PYSEC-2021-486
PYSEC-2021-684

Affected Products

Tensorflow