CVE-2021-29521
May 14, 2021
TensorFlow is an end-to-end open source platform for machine learning. Specifying a negative dense shape in `tf.raw_ops.SparseCountSparseOutput` results in a segmentation fault being thrown out from the standard library as `std::vector` invariants are broken. This is because the implementation(https://github.com/tensorflow/tensorflow/blob/8f7b60ee8c0206a2c99802e3a4d1bb55d2bc0624/tensorflow/core/kernels/count_ops.cc#L199-L213) assumes the first element of the dense shape is always positive and uses it to initialize a `BatchedMap<T>` (i.e., `std::vector<absl::flat_hash_map<int64,T>>`(https://github.com/tensorflow/tensorflow/blob/8f7b60ee8c0206a2c99802e3a4d1bb55d2bc0624/tensorflow/core/kernels/count_ops.cc#L27)) data structure. If the `shape` tensor has more than one element, `num_batches` is the first value in `shape`. Ensuring that the `dense_shape` argument is a valid tensor shape (that is, all elements are non-negative) solves this issue. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2 and TensorFlow 2.3.3.
Affected Packages
tensorflow (PYTHON):
Affected version(s) >=2.4.0 <2.4.2Fix Suggestion:
Update to version 2.4.2tensorflow-cpu (PYTHON):
Affected version(s) >=2.4.0 <2.4.2Fix Suggestion:
Update to version 2.4.2tensorflow-gpu (PYTHON):
Affected version(s) >=2.3.0 <2.3.3Fix Suggestion:
Update to version 2.3.3tensorflow (PYTHON):
Affected version(s) >=2.3.0 <2.3.3Fix Suggestion:
Update to version 2.3.3tensorflow-cpu (PYTHON):
Affected version(s) >=2.3.0 <2.3.3Fix Suggestion:
Update to version 2.3.3tensorflow-gpu (PYTHON):
Affected version(s) >=2.4.0 <2.4.2Fix Suggestion:
Update to version 2.4.2Related ResourcesĀ (6)
Do you need more information?
Contact UsCVSS v4
Base Score:
2
Attack Vector
LOCAL
Attack Complexity
HIGH
Attack Requirements
NONE
Privileges Required
LOW
User Interaction
NONE
Vulnerable System Confidentiality
NONE
Vulnerable System Integrity
NONE
Vulnerable System Availability
LOW
Subsequent System Confidentiality
NONE
Subsequent System Integrity
NONE
Subsequent System Availability
NONE
CVSS v3
Base Score:
2.5
Attack Vector
LOCAL
Attack Complexity
HIGH
Privileges Required
LOW
User Interaction
NONE
Scope
UNCHANGED
Confidentiality
NONE
Integrity
NONE
Availability
LOW
CVSS v2
Base Score:
2.1
Access Vector
LOCAL
Access Complexity
LOW
Authentication
NONE
Confidentiality Impact
NONE
Integrity Impact
NONE
Availability Impact
PARTIAL
Weakness Type (CWE)
Incorrect Calculation of Buffer Size
EPSS
Base Score:
0.01