Skip to content
HighVulnerability

CVE-2021-37655: TensorFlow out-of-bounds heap read through ResourceScatterUpdate

Identifier
CVE-2021-37655
Published
Record updated
View JSON
Affected
  • tensorflow < 2.3.4, fixed in 2.3.4, released August 10, 2021, 3 days before disclosure
  • tensorflow >= 2.4.0, < 2.4.3, fixed in 2.4.3, released August 10, 2021, 3 days before disclosure
  • tensorflow = 2.5.0, fixed in 2.5.1, released August 10, 2021, 3 days before disclosure
  • and 6 more
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.2%

Summary

TensorFlow's `tf.raw_ops.ResourceScatterUpdate` in affected versions can be triggered with invalid arguments to read outside the bounds of heap-allocated data. The cause is incomplete validation of the `indices` and `updates` shapes: the code checks only that their element counts are divisible, not that the shape of `indices` is a prefix of the shape of `updates`.

Mitigation

Fixed in GitHub commit 01cff3f986259d661103412a20745928c727326f. The fix will be included in TensorFlow 2.6.0, and the commit will be cherry-picked to TensorFlow 2.5.1, 2.4.3 and 2.3.4.