TensorFlow — Vulnerability (CVE-2021-41216)
Date: 2021-11
Affected: google/tensorflow
Attack vector: other
Category: vulnerability-disclosure
Severity: Medium
CVE: CVE-2021-41216
CWE: CWE-120
CWE-787
CVSS: 5.5 (CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:N/I:N/A:H)

TensorFlow is an open source platform for machine learning. In affected versions the shape inference function for `Transpose` is vulnerable to a heap buffer overflow. This occurs whenever `perm` contains negative elements. The shape inference function does not validate that the indices in `perm` are all valid. The fix will be included in TensorFlow 2.7.0. We will also cherrypick this commit on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.


Mitigations:

Architectural layers (MAESTRO):