CVE-2021-37644: `std::abort` raised from `TensorListReserve` in TensorFlow
Impact Providing a negative element to numelements list argument of tf.rawops.TensorListReserve causes the runtime to abort the process due to reallocating a std::vector to have a negative number of elements:
python import tensorflow as tf
tf.rawops.TensorListReserve( elementshape = tf.constant([1]), numelements=tf.constant([-1]), elementdtype = tf.int32)
The implementation calls std::vector.resize() with the new size controlled by input given by the user, without checking that this input is valid.
Patches We have patched the issue in GitHub commit 8a6e874437670045e6c7dc6154c7412b4a2135e2.
The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4, as these are also affected and still in supported range.
For more information Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
Attribution This vulnerability has been reported by members of the Aivul Team from Qihoo 360.
Other sources
TensorFlow is an end-to-end open source platform for machine learning. In affected versions providing a negative element to numelements list argument of tf.rawops.TensorListReserve causes the runtime to abort the process due to reallocating a std::vector to have a negative number of elements. The implementation calls std::vector.resize() with the new size controlled by input given by the user, without checking that this input is valid. We have patched the issue in GitHub commit 8a6e874437670045e6c7dc6154c7412b4a2135e2. The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4, as these are also affected and still in supported range.
Affected Software
Remediation
Recommended actions to resolve this vulnerability, in priority order.
- Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflow-gputo a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflow-cputo a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.5.1 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.4.3 - Upgrade
Upgrade
pip/tensorflowto a version that resolves this vulnerability.Fixed in 2.3.4 - Upgrade
Upgrade
tensorflowto a version that resolves this vulnerability.Patch 8a6e874437670045e6c7dc6154c7412b4a2135e2 - Upgrade
Upgrade
tensorflowto a version that resolves this vulnerability.Fixed in 2.6.0Patch 8a6e874437670045e6c7dc6154c7412b4a2135e2 - Upgrade
Upgrade
tensorflowto a version that resolves this vulnerability.Fixed in 2.5.1Patch 8a6e874437670045e6c7dc6154c7412b4a2135e2 - Upgrade
Upgrade
tensorflowto a version that resolves this vulnerability.Fixed in 2.4.3Patch 8a6e874437670045e6c7dc6154c7412b4a2135e2 - Upgrade
Upgrade
tensorflowto a version that resolves this vulnerability.Fixed in 2.3.4Patch 8a6e874437670045e6c7dc6154c7412b4a2135e2
Event History
Frequently Asked Questions
What is the severity of CVE-2021-37644?
CVE-2021-37644 has a severity rating of medium as it causes a runtime abort when a negative number is provided to `num_elements`.
How do I fix CVE-2021-37644?
To fix CVE-2021-37644, upgrade to TensorFlow version 2.3.4, 2.4.3, or 2.5.1 depending on your current version.
Which versions of TensorFlow are affected by CVE-2021-37644?
CVE-2021-37644 affects TensorFlow versions 2.3.0 to 2.3.4, 2.4.0 to 2.4.3, as well as version 2.5.0, and 2.6.0-rc0, rc1, and rc2.
What happens if I exploit CVE-2021-37644?
Exploiting CVE-2021-37644 by passing a negative `num_elements` can cause the TensorFlow process to abort, leading to service disruption.
Is CVE-2021-37644 a client-side or server-side vulnerability?
CVE-2021-37644 is considered a server-side vulnerability as it affects the runtime behavior of TensorFlow on the server.