CVE-2021-29552

LOW

TensorFlow < 2.1.4 - Denial of Service via UnsortedSegmentJoin num_segments Tensor

Title source: llm
STIX 2.1

Description

TensorFlow is an end-to-end open source platform for machine learning. An attacker can cause a denial of service by controlling the values of `num_segments` tensor argument for `UnsortedSegmentJoin`. This is because the implementation(https://github.com/tensorflow/tensorflow/blob/a2a607db15c7cd01d754d37e5448d72a13491bdb/tensorflow/core/kernels/unsorted_segment_join_op.cc#L92-L93) 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 and this would result in process termination. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.

References (2)

Core 2
Core References
Exploit, Patch, Third Party Advisory x_refsource_confirm
https://github.com/tensorflow/tensorflow/security/advisories/GHSA-jhq9-wm9m-cf89

Scores

CVSS v3 2.5
EPSS 0.0019
EPSS Percentile 8.7%
Attack Vector LOCAL
CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:N/I:N/A:L

Details

CWE
CWE-617
Status published
Products (4)
google/tensorflow < 2.1.4
pypi/tensorflow 0 - 2.1.4PyPI
pypi/tensorflow-cpu 0 - 2.1.4PyPI
pypi/tensorflow-gpu 0 - 2.1.4PyPI
Published May 14, 2021
Tracked Since Feb 18, 2026