GitHub Advisoryadvisory · PyPI
GHSA-cmgw-8vpc-rc59 Segfault on strings tensors with mistmatched dimensions, due to Go code
MEDIUM
Official page ↗
- Aliases
- CVE-2021-37692
- Ecosystem
- PyPI
- Packages
- tensorflow, tensorflow-cpu, tensorflow-gpu
- Severity
- Moderate
- CVSS vector
- CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:N/I:N/A:H
- Weaknesses
- CWE-20
- Reviewed
- 2021-08-24
Affected packages
• PyPI tensorflow: >= 2.5.0rc0, < 2.5.1 → fixed in 2.5.1
• PyPI tensorflow-cpu: >= 2.5.0rc0, < 2.5.1 → fixed in 2.5.1
• PyPI tensorflow-gpu: >= 2.5.0rc0, < 2.5.1 → fixed in 2.5.1
Description
Impact
Under certain conditions, Go code can trigger a segfault in string deallocation.
For string tensors, `C.TF_TString_Dealloc` is called during garbage collection within a finalizer function. However, tensor structure isn't checked until encoding to avoid a performance penalty. The current method for dealloc assumes that encoding succeeded, but segfaults when a string tensor is garbage collected whose encoding failed (e.g., due to mismatched dimensions).
To fix this, the call to set the finalizer function is deferred until `NewTensor` returns and, if encoding failed for a string tensor, deallocs are determined based on bytes written.
Patches
We have patched the issue in GitHub commit 8721ba96e5760c229217b594f6d2ba332beedf22 (merging #50508).
The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, which is the other affected version.
For more information
Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
Attribution
This vulnerability has been reported externally via a fixing PR.
Related
Alias · CVE (1)
- CVE-2021-37692 TensorFlow is an end-to-end open source platform for machine learning. vulnerability
Weakness · CWE (1)
- CWE-20 Improper Input Validation weakness
References
- https://github.com/tensorflow/tensorflow/security/advisories/GHSA-cmgw-8vpc-rc59 web
- https://nvd.nist.gov/vuln/detail/CVE-2021-37692 advisory
- https://github.com/tensorflow/tensorflow/pull/50508 web
- https://github.com/tensorflow/tensorflow/commit/8721ba96e5760c229217b594f6d2ba332beedf22 web
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-605.yaml web
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-803.yaml web
- https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-314.yaml web
- https://github.com/tensorflow/tensorflow package