PT-2020-14284 · Google · Tensorflow Lite
Published
2020-09-25
·
Updated
2024-03-06
·
CVE-2020-15213
CVSS v4.0
6.3
Medium
| Vector | AV:N/AC:L/AT:P/PR:N/UI:N/VC:N/VI:N/VA:L/SC:N/SI:N/SA:L |
Name of the Vulnerable Software and Affected Versions
TensorFlow Lite versions prior to 2.2.1
TensorFlow Lite versions prior to 2.3.1
Description
The issue allows attackers to trigger a denial of service by causing an out of memory allocation in the implementation of segment sum. This is possible because the code uses the last element of the tensor holding segment ids to determine the dimensionality of the output tensor, and attackers can use a very large value to trigger a large allocation.
Recommendations
For versions prior to 2.2.1, upgrade to TensorFlow 2.2.1.
For versions prior to 2.3.1, upgrade to TensorFlow 2.3.1.
As a temporary workaround, consider adding a custom
Verifier to limit the maximum value in the segment ids tensor, but this only handles the case when the segment ids are stored statically in the model. If the segment ids are generated as outputs of a tensor during inference steps, there are no possible workarounds and users are advised to upgrade to patched code.Exploit
Fix
Allocation of Resources Without Limits
Buffer Overflow
Found an issue in the description? Have something to add? Feel free to write us 👾
Related Identifiers
Affected Products
Tensorflow Lite