Description
TensorFlow is an open source platform for machine learning. Prior to versions 2.9.0, 2.8.1, 2.7.2, and 2.6.4, the implementation of `tf.raw_ops.UnsortedSegmentJoin` does not fully validate the input arguments. This results in a `CHECK`-failure which can be used to trigger a denial of service attack. The code assumes `num_segments` is a positive scalar but there is no validation. Since this value is used to allocate the output tensor, a negative value would result in a `CHECK`-failure (assertion failure), as per TFSA-2021-198. Versions 2.9.0, 2.8.1, 2.7.2, and 2.6.4 contain a patch for this issue.
References (9)
Core 9
Core References
Patch, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/security/advisory/tfsa-2021-198.md
Release Notes, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/releases/tag/v2.6.4
Release Notes, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/releases/tag/v2.7.2
Release Notes, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/releases/tag/v2.8.1
Release Notes, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/releases/tag/v2.9.0
Exploit, Patch, Third Party Advisory x_refsource_confirm
https://github.com/tensorflow/tensorflow/security/advisories/GHSA-hx9q-2mx4-m4pg
Patch, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/commit/20cb18724b0bf6c09071a3f53434c4eec53cc147
Patch, Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/commit/84563f265f28b3c36a15335c8b005d405260e943
Third Party Advisory x_refsource_misc
https://github.com/tensorflow/tensorflow/blob/f3b9bf4c3c0597563b289c0512e98d4ce81f886e/tensorflow/core/kernels/unsorted_segment_join_op.cc#L83-L14
Scores
CVSS v3
5.5
EPSS
0.0006
EPSS Percentile
19.2%
Attack Vector
LOCAL
CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:N/I:N/A:H
CISA SSVC
Vulnrichment
Exploitation
poc
Automatable
no
Technical Impact
partial
Details
CWE
CWE-191
CWE-20
Status
published
Products (7)
google/tensorflow
2.7.0 rc0 (2 CPE variants)
google/tensorflow
2.8.0 (3 CPE variants)
google/tensorflow
2.9.0 rc0 (2 CPE variants)
google/tensorflow
1.15.0 - 2.6.4
pypi/tensorflow
0 - 2.6.4PyPI
pypi/tensorflow-cpu
0 - 2.6.4PyPI
pypi/tensorflow-gpu
0 - 2.6.4PyPI
Published
May 20, 2022
Tracked Since
Feb 18, 2026