Skip to content
LowVulnerability

CVE-2021-29512: TensorFlow heap buffer overflow in RaggedBincount through the splits argument

Identifier
CVE-2021-29512
Published
Record updated
View JSON
Affected
  • tensorflow >= 2.3.0, < 2.3.3, fixed in 2.3.3, released June 15, 2021, 32 days after disclosure
  • tensorflow >= 2.4.0, < 2.4.2, fixed in 2.4.2, released June 15, 2021, 32 days after disclosure
  • tensorflow-cpu >= 2.3.0, < 2.3.3, fixed in 2.3.3, released June 15, 2021, 32 days after disclosure
  • and 3 more
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

TensorFlow's RaggedBincount op has a heap buffer overflow when the splits argument is not a valid SparseTensor. An attacker who controls splits, set to a single element of 0, causes the while loop to read splits(1), outside the buffer bounds. The fix ships in TensorFlow 2.5.0, with cherry-picks to 2.4.2 and 2.3.3, which are also affected.

Mitigation

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