CVE-2021-29601: TensorFlow Integer Overflow in TFLite Concatenation
Summary
CVE-2021-29601: Integer overflow in TFLite concatentation. Affects tensorflow across 4 version ranges.
< 2.1.4
>= 2.2.0, < 2.2.3
>= 2.3.0, < 2.3.3
>= 2.4.0, < 2.4.2
Fixed in 4 releases; the minimum fixed version depends on the release line you run. How to fix ↓
Also tracked for
- tensorflow · PyPI
- tensorflow · PyPI
- tensorflow-cpu · PyPI
- tensorflow-gpu · PyPI
The same CVE-2021-29601 is tracked separately for each ecosystem and vendor. Its fix information is the same; the package and affected versions differ.
Check if this vulnerability affects your application
Upload a manifest or SBOM and we'll show whether you're affected, plus every other risky package in the same scan.
No signup required · Results in about 60 seconds
Background
TensorFlow is an end-to-end open source platform for machine learning. It provides a comprehensive ecosystem of tools, libraries, and resources for developing and deploying machine learning models. The TFLite implementation of concatenation is vulnerable to an integer overflow issue. This vulnerability arises due to the use of int to represent tensor dimensions in TFLite, while TensorFlow uses int64. As a result, when converting valid TensorFlow models to TFLite format, an attacker can craft a model with dimensions that overflow the values of int. This vulnerability affects various versions of TensorFlow, including 2.1.4, 2.2.3, 2.3.3, and 2.4.2.
Vulnerability Detail
The vulnerability in TensorFlow's TFLite implementation of concatenation allows an attacker to trigger an integer overflow by manipulating the dimensions of one of the concatenation inputs. This can lead to unexpected behavior, including crashes or potential security vulnerabilities. The issue has been identified and a fix will be included in TensorFlow 2.5.0. Additionally, the fix will be backported to TensorFlow 2.4.2, 2.3.3, 2.2.3, and 2.1.4, as these versions are still within the supported range.
How To Fix
Fixed versions by release line
The minimum fixed version depends on the release line you run.
-
< 2.1.42.1.4 -
>= 2.2.0, < 2.2.32.2.3 -
>= 2.3.0, < 2.3.32.3.3 -
>= 2.4.0, < 2.4.22.4.2
Tip — Not sure which release line you're on? Vulert can find the fix for the version you are using. Scan free →
Workaround
There is no specific workaround mentioned for this vulnerability.
Conclusion
The vulnerability in TensorFlow's TFLite implementation of concatenation can be exploited through integer overflow. It is crucial to update to the fixed version or apply the backported fixes to ensure the security and stability of TensorFlow installations. For further support on vulnerability remediation, please contact DevNack.com.
See every vulnerability recorded for tensorflow →
FAQ
TensorFlow is an open source platform for machine learning, providing tools and resources for developing and deploying machine learning models.
TFLite is a lightweight version of TensorFlow designed for mobile and embedded devices.
The vulnerability arises due to an integer overflow issue when converting TensorFlow models to TFLite format, caused by the use of int for tensor dimensions in TFLite.
The vulnerability can lead to unexpected behavior, including crashes or potential security vulnerabilities, when manipulating the dimensions of concatenation inputs.
Users should update TensorFlow to version 2.5.0 once it is released. Backported fixes will also be available for affected versions (2.4.2, 2.3.3, 2.2.3, and 2.1.4).
References
Related vulnerabilities
Recent vulnerabilities recorded for tensorflow.
