Skip to content
HighVulnerability

CVE-2021-37643: TensorFlow null pointer dereference in MatrixDiagPartOp

Identifier
CVE-2021-37643
Published
Record updated
View JSON
Affected
  • tensorflow < 2.3.4, fixed in 2.3.4, released August 10, 2021, 2 days before disclosure
  • tensorflow >= 2.4.0, < 2.4.3, fixed in 2.4.3, released August 10, 2021, 2 days before disclosure
  • tensorflow = 2.5.0, fixed in 2.5.1, released August 10, 2021, 2 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.MatrixDiagPartOp crashes or misbehaves when a user omits a valid padding value. With empty input it triggers a null pointer dereference, and otherwise it ignores all values after the first. The implementation reads the first value from a tensor buffer without checking that the tensor has values to read.

Mitigation

Patched in GitHub commit 482da92095c4d48f8784b1f00dda4f81c28d2988. The fix will be included in TensorFlow 2.6.0, and the commit will also be cherry-picked to TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4.