🛡️ CVE-2021-37664 — tensorflow

🟠 CVSS 8.0 — High ✅ No Known Exploit CWE-125 OSV
8.0
CVSS Score
0 Low4 Medium7 High9 Critical10

Description

Heap OOB in boosted trees

Impact

An attacker can read from outside of bounds of heap allocated data by sending specially crafted illegal arguments to BoostedTreesSparseCalculateBestFeatureSplit:

```python

import tensorflow as tf

tf.raw_ops.BoostedTreesSparseCalculateBestFeatureSplit(

node_id_range=[0,10],

stats_summary_indices=[[1, 2, 3, 0x1000000]],

stats_summary_values=[1.0],

stats_summary_shape=[1,1,1,1],

l1=l2=[1.0],

tree_complexity=[0.5],

min_node_weight=[1.0],

logits_dimension=3,

split_type='inequality')

```

The [implementation](https://github.com/tensorflow/tensorflow/blob/84d053187cb80d975ef2b9684d4b61981bca0c41/tensorflow/core/kernels/boosted_trees/stats_ops.cc) needs to validate that each value in stats_summary_indices is in range.

Patches

We have patched the issue in GitHub commit [e84c975313e8e8e38bb2ea118196369c45c51378](https://github.com/tensorflow/tensorflow/commit/e84c975313e8e8e38bb2ea118196369c45c51378).

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.

For more information

Please consult [our security guide](https://github.com/tensorflow/tensorflow/blob/master/SECURITY.md) for more information regarding the security model and how to contact us with issues and questions.

Attribution

This vulnerability has been reported by members of the Aivul Team from Qihoo 360.

How this vulnerability can be exploited

This issue can be reached with local access to the system, attack complexity is low, an attacker needs low-level privileges on the target. No user interaction is required. Rated impact: confidentiality high, integrity low, availability high.

Weakness class

CVE-2021-37664 is classified as CWE-125: Out-of-bounds Read. The code reads past the limits of a buffer, exposing adjacent memory contents or crashing the process.

Affected software

CVE-2021-37664 is recorded against 3 packages.

  • tensorflow
  • tensorflow-cpu (from 2.5.0 up to 2.5.1)
  • tensorflow-gpu (from 2.5.0 up to 2.5.1)

Timeline and source

Published on 6 March 2024 and last revised on 8 July 2026. No public exploit is currently recorded for this entry. A vendor advisory or fix has been published. Record sourced from OSV.

References

github.com (Web)
github.com (Web)
nvd.nist.gov (Web)

Details

Severity HIGH
CVSS Score 8.0
CVSS Vector CVSS:4.0/AV:L/AC:L/AT:N/PR:L/UI:N/VC:H/VI:L/VA:H/SC:N/SI:N/SA:N
CWE CWE-125
Public Exploit ✅ No
Source OSV
Published 2024-03-06
Updated 2026-08-12
Modified 2026-07-08

Affected Packages

Software From version Fixed in
tensorflow — —
tensorflow-cpu 2.5.0 2.5.1
tensorflow-gpu 2.5.0 2.5.1

Similar Threats

Site Security Check

Is tensorflow part of your stack?

CVE-2021-37664 is rated CVSS 8.0 High. BotEraser scans your installation against known CVE records and tells you whether this vulnerability applies to the versions you actually run.

Scan My Site Free →

No credit card required  ·  Results in minutes

ⓘ Data Notice: The information presented above has been compiled from publicly available internet sources. Boteraser aggregates this data solely for informational purposes and does not independently classify, evaluate, or endorse any findings about the vulnerabilities listed. The accuracy and completeness of this information is the sole responsibility of the original publishers. Boteraser and its operators accept no liability for any decisions made based on this data.