CVE-2021-29599: Division by zero in TFLite's implementation of Split
(updated )
The implementation of the Split TFLite operator is vulnerable to a division by zero error:
TF_LITE_ENSURE_MSG(context, input_size % num_splits == 0, "Not an even split");
const int slice_size = input_size / num_splits;
An attacker can craft a model such that num_splits would be 0.
References
- github.com/advisories/GHSA-97wf-p777-86jq
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-527.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-725.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-236.yaml
- github.com/tensorflow/tensorflow
- github.com/tensorflow/tensorflow/blob/e2752089ef7ce9bcf3db0ec618ebd23ea119d0c7/tensorflow/lite/kernels/split.cc
- github.com/tensorflow/tensorflow/commit/b22786e7e9b7bdb6a56936ff29cc7e9968d7bc1d
- github.com/tensorflow/tensorflow/security/advisories/GHSA-97wf-p777-86jq
- nvd.nist.gov/vuln/detail/CVE-2021-29599
Code Behaviors & Features
Detect and mitigate CVE-2021-29599 with GitLab Dependency Scanning
Secure your software supply chain by verifying that all open source dependencies used in your projects contain no disclosed vulnerabilities. Learn more about Dependency Scanning →