CVE-2021-37690: Use after free and segfault in shape inference functions
(updated )
When running shape functions, some functions (such as MutableHashTableShape) produce extra output information in the form of a ShapeAndType struct. The shapes embedded in this struct are owned by an inference context that is cleaned up almost immediately; if the upstream code attempts to access this shape information, it can trigger a segfault.
ShapeRefiner is mitigating this for normal output shapes by cloning them (and thus putting the newly created shape under ownership of an inference context that will not die), but we were not doing the same for shapes and types. This commit fixes that by doing similar logic on output shapes and types.
References
- github.com/advisories/GHSA-3hxh-8cp2-g4hg
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-603.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-801.yaml
- github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-312.yaml
- github.com/tensorflow/tensorflow
- github.com/tensorflow/tensorflow/commit/ee119d4a498979525046fba1c3dd3f13a039fbb1
- github.com/tensorflow/tensorflow/security/advisories/GHSA-3hxh-8cp2-g4hg
- nvd.nist.gov/vuln/detail/CVE-2021-37690
Code Behaviors & Features
Detect and mitigate CVE-2021-37690 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 →