Skip to main content
← All vulnerabilities
CVE-2021-29614PYSEC-2021-542BIT-TENSORFLOW-2021-29614GHSA-8PMX-P244-G88HPYSEC-2021-251PYSEC-2021-740

Interpreter crash from `tf.io.decode_raw`

Published
Last modified
published
Not listed in CISA KEV
Fix
Fixed in 2.1.4, 2.2.3, 2.3.3, 2.4.2
None on Exploit-DB, Metasploit or GitHub

Summary

TensorFlow is an end-to-end open source platform for machine learning. The implementation of `tf.io.decode_raw` produces incorrect results and crashes the Python interpreter when combining `fixed_length` and wider datatypes. The implementation of the padded version(https://github.com/tensorflow/tensorflow/blob/1d8903e5b167ed0432077a3db6e462daf781d1fe/tensorflow/core/kernels/decode_padded_raw_op.cc) is buggy due to a confusion about pointer arithmetic rules. First, the code computes(https://github.com/tensorflow/tensorflow/blob/1d8903e5b167ed0432077a3db6e462daf781d1fe/tensorflow/core/kernels/decode_padded_raw_op.cc#L61) the width of each output element by dividing the `fixed_length` value to the size of the type argument. The `fixed_length` argument is also used to determine the size needed for the output tensor(https://github.com/tensorflow/tensorflow/blob/1d8903e5b167ed0432077a3db6e462daf781d1fe/tensorflow/core/kernels/decode_padded_raw_op.cc#L63-L79). This is followed by reencoding code(https://github.com/tensorflow/tensorflow/blob/1d8903e5b167ed0432077a3db6e462daf781d1fe/tensorflow/core/kernels/decode_padded_raw_op.cc#L85-L94). The erroneous code is the last line above: it is moving the `out_data` pointer by `fixed_length * sizeof(T)` bytes whereas it only copied at most `fixed_length` bytes from the input. This results in parts of the input not being decoded into the output. Furthermore, because the pointer advance is far wider than desired, this quickly leads to writing to outside the bounds of the backing data. This OOB write leads to interpreter crash in the reproducer mentioned here, but more severe attacks can be mounted too, given that this gadget allows writing to periodically placed locations in memory. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, TensorFlow 2.3.3, TensorFlow 2.2.3 and TensorFlow 2.1.4, as these are also affected and still in supported range.

Timeline

  1. CVE published
  2. Record last updated

Affected versions

  • tensorflow tensorflow

    fixed in 2.1.4; From 2.2.0 · fixed in 2.2.3; From 2.3.0 · fixed in 2.3.3; From 2.4.0 · fixed in 2.4.2

  • tensorflow-cpuPyPI

    fixed in 2.1.4; From 2.2.0 · fixed in 2.2.3; From 2.3.0 · fixed in 2.3.3; From 2.4.0 · fixed in 2.4.2

  • tensorflowPyPI

    fixed in 2.1.4; From 2.2.0 · fixed in 2.2.3; From 2.3.0 · fixed in 2.3.3; From 2.4.0 · fixed in 2.4.2

  • tensorflowPyPI

    fixed in 2.1.4

  • tensorflowPyPI

    From 2.2.0 · fixed in 2.2.3

  • tensorflowPyPI

    From 2.3.0 · fixed in 2.3.3

  • tensorflowPyPI

    From 2.4.0 · fixed in 2.4.2

  • tensorflow-cpuPyPI

    fixed in 2.1.4

  • tensorflow-cpuPyPI

    From 2.2.0 · fixed in 2.2.3

  • tensorflow-cpuPyPI

    From 2.3.0 · fixed in 2.3.3

  • tensorflow-cpuPyPI

    From 2.4.0 · fixed in 2.4.2

  • tensorflow-gpuPyPI

    fixed in 2.1.4

  • tensorflow-gpuPyPI

    From 2.2.0 · fixed in 2.2.3

  • tensorflow-gpuPyPI

    From 2.3.0 · fixed in 2.3.3

  • tensorflow-gpuPyPI

    From 2.4.0 · fixed in 2.4.2

  • tensorflow-gpuPyPI

    fixed in 2.1.4; From 2.2.0 · fixed in 2.2.3; From 2.3.0 · fixed in 2.3.3; From 2.4.0 · fixed in 2.4.2

References 2

GitHub · tensorflow/tensorflow