Skip to content
HighVulnerability

CVE-2021-29606: TensorFlow out-of-bounds read in TFLite Split_V via crafted model

Identifier
CVE-2021-29606
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 TFLite implementation of `Split_V` contains an out-of-bounds heap read. A specially crafted TFLite model can trigger it when `axis_value` falls outside the range between 0 and `NumDimensions(input)`, because `SizeOfDimension` then reads past the bounds of the tensor shape array. The fix lands in TensorFlow 2.5.0, with cherry-picks to TensorFlow 2.4.2, 2.3.3, 2.2.3 and 2.1.4, which the source identifies as affected and still in supported range.

Mitigation

Fixed 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.