Skip to content
LowVulnerability

CVE-2021-29552: TensorFlow denial of service through UnsortedSegmentJoin num_segments argument

Identifier
CVE-2021-29552
Published
Record updated
View JSON
Affected
  • tensorflow < 2.1.4, fixed in 2.1.4, released June 15, 2021, 31 days after disclosure
  • tensorflow >= 2.2.0, < 2.2.3, fixed in 2.2.3, released June 15, 2021, 31 days after disclosure
  • tensorflow >= 2.3.0, < 2.3.3, fixed in 2.3.3, released June 15, 2021, 31 days after disclosure
  • and 9 more
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

TensorFlow's UnsortedSegmentJoin operator assumes its num_segments tensor argument is a valid scalar. An attacker who controls that argument can pass an empty tensor, which invalidates a CHECK in .scalar<T>()() and terminates the process, causing a denial of service.

Mitigation

The fix will be included in TensorFlow 2.5.0. The commit will also be cherry-picked to TensorFlow 2.4.2, 2.3.3, 2.2.3 and 2.1.4.