Skip to content
MediumVulnerability

CVE-2021-29601: TensorFlow Lite integer overflow in concatenation via crafted model

Identifier
CVE-2021-29601
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 concatenation contains an integer overflow in concatenation.cc. An attacker can craft a model whose concatenation input dimensions overflow the `int` type, since TFLite stores tensor dimensions as `int` while TensorFlow uses `int64`. Valid TensorFlow models can therefore trigger the overflow when converted to TFLite format.

Mitigation

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