Description
Tensorflow is an Open Source Machine Learning Framework. An attacker can craft a TFLite model that would cause an integer overflow in embedding lookup operations. Both `embedding_size` and `lookup_size` are products of values provided by the user. Hence, a malicious user could trigger overflows in the multiplication. In certain scenarios, this can then result in heap OOB read/write. Users are advised to upgrade to a patched version.
Affected products
| Vendor | Product | Affected | Fixed |
|---|---|---|---|
| tensorflow | tensorflow | >= 2.7.0, < 2.7.1, >= 2.6.0, < 2.6.3, < 2.5.3 | — |
Affected packages
| Ecosystem | Package | Affected | Fixed | Record |
|---|---|---|---|---|
| PyPI | tensorflow | < 2.5.3 | 2.5.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow | >= 2.6.0, < 2.6.3 | 2.6.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow | >= 2.7.0, < 2.7.1 | 2.7.1 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow | < 2.5.3 · >= 2.6.0, < 2.6.3 · >= 2.7.0, < 2.7.1 | 2.5.3, 2.6.3, 2.7.1 | PYSEC-2026-3146 |
| PyPI | tensorflow-cpu | < 2.5.3 | 2.5.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-cpu | >= 2.6.0, < 2.6.3 | 2.6.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-cpu | >= 2.7.0, < 2.7.1 | 2.7.1 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-cpu | < 2.5.3 · >= 2.6.0, < 2.6.3 | 2.5.3, 2.6.3 | PYSEC-2022-68 |
| PyPI | tensorflow-gpu | < 2.5.3 | 2.5.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-gpu | >= 2.6.0, < 2.6.3 | 2.6.3 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-gpu | >= 2.7.0, < 2.7.1 | 2.7.1 | GHSA-98p5-x8x4-c9m5 |
| PyPI | tensorflow-gpu | < 2.5.3 · >= 2.6.0, < 2.6.3 | 2.5.3, 2.6.3 | PYSEC-2022-123 |
References
- https://github.com/tensorflow/tensorflow/security/advisories/GHSA-98p5-x8x4-c9m5
- https://github.com/tensorflow/tensorflow/commit/1de49725a5fc4e48f1a3b902ec3599ee99283043
- https://github.com/tensorflow/tensorflow/commit/a4e401da71458d253b05e41f28637b65baf64be4
- https://github.com/tensorflow/tensorflow/commit/f19be71717c497723ba0cea0379e84f061a75e01
- https://github.com/tensorflow/tensorflow/blob/ca6f96b62ad84207fbec580404eaa7dd7403a550/tensorflow/lite/kernels/embedding_lookup_sparse.cc#L179-L189