CVE-2021-37650 - CVE House
Back to Database
Status published High CVE-2021-37650

Segfault and heap buffer overflow in `{Experimental,}DatasetToTFRecord` in TensorFlow

Vulnerability Description

TensorFlow is an end-to-end open source platform for machine learning. In affected versions the implementation for `tf.raw_ops.ExperimentalDatasetToTFRecord` and `tf.raw_ops.DatasetToTFRecord` can trigger heap buffer overflow and segmentation fault. The [implementation](https://github.com/tensorflow/tensorflow/blob/f24faa153ad31a4b51578f8181d3aaab77a1ddeb/tensorflow/core/kernels/data/experimental/to_tf_record_op.cc#L93-L102) assumes that all records in the dataset are of string type. However, there is no check for that, and the example given above uses numeric types. We have patched the issue in GitHub commit e0b6e58c328059829c3eb968136f17aa72b6c876. The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4, as these are also affected and still in supported range.

Impact Analysis

Refer to official advisory for detailed impact metrics.

Remediation

Ensure systems are updated to the latest vendor-supplied patch levels.

THREAT MONITOR

Am I Vulnerable?

Launch our assessment wizard to check if your infrastructure is exposed to • CVE-2021-37650

Credits & Attribution

No credits recorded in the NVD database.

References

More from tensorflow

View All →
CVE-2024-3660
Arbitrary code injection vulnerability in Keras framework < 2.13
Unknown 0
CVE-2023-33976
TensorFlow segfault in array_ops.upper_bound
High 7.5
CVE-2023-27579
TensorFlow has Floating Point Exception in TFLite in conv kernel
High 7.5
CVE-2023-25801
TensorFlow has double free in Fractional(Max/Avg)Pool
High 8
CVE-2023-25676
TensorFlow has null dereference on ParallelConcat with XLA
High 7.5

Affected Vendor

Affected Software

tensorflow
Vulnerable Versions:
>= 2.5.0, < 2.5.1, >= 2.4.0, < 2.4.3, < 2.3.4

Timeline

Official Publish: August 12th, 2021
Last Modified: August 4th, 2024
Added to House: July 21st, 2026

CVSS Vectors

V3: CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:H/I:H/A:H

Weaknesses (CWE)

CWE-120

MITRE ATT&CK TTPs

T1203 Exploitation for Client Execution
Execution
T1068 Exploitation for Privilege Escalation
Privilege Escalation
T1055 Process Injection
Defense Evasion
T1105 Ingress Tool Transfer
Command and Control