Skip to content
LowVulnerability

CVE-2021-29536: TensorFlow heap buffer overflow in QuantizedReshape via invalid thresholds

Identifier
CVE-2021-29536
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

CVE-2021-29536 is a heap buffer overflow in TensorFlow's `QuantizedReshape` operation. The implementation assumes its two arguments are always valid scalars and reads their numeric value directly, so passing an empty tensor as a quantization threshold makes `.flat<T>()` an empty buffer and accessing element 0 overflows it.

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, which are affected and still within supported range.