CVE-2021-37686: Infinite loop in TensorFlow Lite
First published: Thu Aug 12 2021(Updated: )
### Impact The strided slice implementation in TFLite has a logic bug which can allow an attacker to trigger an infinite loop. This arises from newly introduced support for [ellipsis in axis definition](https://github.com/tensorflow/tensorflow/blob/149562d49faa709ea80df1d99fc41d005b81082a/tensorflow/lite/kernels/strided_slice.cc#L103-L122): ```cc for (int i = 0; i < effective_dims;) { if ((1 << i) & op_context->params->ellipsis_mask) { // ... int ellipsis_end_idx = std::min(i + 1 + num_add_axis + op_context->input_dims - begin_count, effective_dims); // ... for (; i < ellipsis_end_idx; ++i) { // ... } continue; } // ... ++i; } ``` An attacker can craft a model such that `ellipsis_end_idx` is smaller than `i` (e.g., always negative). In this case, the inner loop does not increase `i` and the `continue` statement causes execution to skip over the preincrement at the end of the outer loop. ### Patches We have patched the issue in GitHub commit [dfa22b348b70bb89d6d6ec0ff53973bacb4f4695](https://github.com/tensorflow/tensorflow/commit/dfa22b348b70bb89d6d6ec0ff53973bacb4f4695). The fix will be included in TensorFlow 2.6.0. This is the only affected version. ### For more information Please consult [our security guide](https://github.com/tensorflow/tensorflow/blob/master/SECURITY.md) 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.
Credit: security-advisories@github.com security-advisories@github.com
| Affected Software | Affected Version | How to fix |
|---|---|---|
| Google TensorFlow | >=2.3.0<2.3.4 | |
| Google TensorFlow | >=2.4.0<2.4.3 | |
| Google TensorFlow | =2.5.0 | |
| Google TensorFlow | =2.6.0-rc0 | |
| Google TensorFlow | =2.6.0-rc1 | |
| Google TensorFlow | =2.6.0-rc2 | |
| pip/tensorflow-gpu | >=2.6.0rc0<2.6.0rc2 | 2.6.0rc2 |
| pip/tensorflow-cpu | >=2.6.0rc0<2.6.0rc2 | 2.6.0rc2 |
| pip/tensorflow | >=2.6.0rc0<2.6.0rc2 | 2.6.0rc2 |
Never miss a vulnerability like this again
Sign up to SecAlerts for real-time vulnerability data matched to your software, aggregated from hundreds of sources.
Reference Links
- https://github.com/tensorflow/tensorflow/commit/dfa22b348b70bb89d6d6ec0ff53973bacb4f4695
- https://github.com/tensorflow/tensorflow/security/advisories/GHSA-mhhc-q96p-mfm9
- https://nvd.nist.gov/vuln/detail/CVE-2021-37686
- https://github.com/tensorflow/tensorflow/releases/tag/v2.3.4
- https://github.com/tensorflow/tensorflow/releases/tag/v2.4.3
- https://github.com/tensorflow/tensorflow/releases/tag/v2.5.1
- https://github.com/tensorflow/tensorflow/releases/tag/v2.6.0
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-599.yaml
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-797.yaml
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-308.yaml
- https://github.com/advisories/GHSA-mhhc-q96p-mfm9 (Advisory)
- collector/nvd-index
- agent/type
- agent/severity
- agent/title
- agent/weakness
- agent/remedy
- agent/first-publish-date
- collector/mitre-cve
- source/MITRE
- collector/github-advisory-latest
- source/GitHub
- alias/GHSA-mhhc-q96p-mfm9
- alias/CVE-2021-37686
- agent/software-canonical-lookup
- agent/last-modified-date
- agent/references
- agent/softwarecombine
- agent/description
- agent/event
- collector/nvd-cve
- source/NVD
- agent/author
- collector/github-advisory
- agent/tags
- agent/trending
- vendor/google
- canonical/google tensorflow
- version/google tensorflow/2.3.0
- version/google tensorflow/2.4.0
- version/google tensorflow/2.5.0
- version/google tensorflow/2.6.0-rc0
- version/google tensorflow/2.6.0-rc1
- version/google tensorflow/2.6.0-rc2
- package-manager/pip