Skip to content
MediumVulnerability

CVE-2021-37686: TensorFlow Lite strided slice infinite loop via crafted model

Identifier
CVE-2021-37686
Published
Record updated
View JSON
Affected
  • tensorflow >= 2.6.0rc0, < 2.6.0rc2
  • tensorflow-cpu >= 2.6.0rc0, < 2.6.0rc2
  • tensorflow-gpu >= 2.6.0rc0, < 2.6.0rc2
Fixed in
2.6.0rc2
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

CVE-2021-37686 affects TensorFlow Lite's strided slice implementation, which contains a logic bug introduced with new support for ellipsis in axis definition. An attacker can craft a model where `ellipsis_end_idx` is smaller than `i`, so the inner loop never increments `i` and the `continue` statement skips the outer loop's preincrement, causing an infinite loop. TensorFlow 2.6.0 is the only affected version.

Mitigation

We have patched the issue in GitHub commit dfa22b348b70bb89d6d6ec0ff53973bacb4f4695.