Memory leak in decoding PNG images in Tensorflow

Summary

Tensorflow is an Open Source Machine Learning Framework. When decoding PNG images TensorFlow can produce a memory leak if the image is invalid. After calling `png::CommonInitDecode(..., &decode)`, the `decode` value contains allocated buffers which can only be freed by calling `png::CommonFreeDecode(&decode)`. However, several error case in the function implementation invoke the `OP_REQUIRES` macro which immediately terminates the execution of the function, without allowing for the memory free to occur. The fix will be included in TensorFlow 2.8.0. We will also cherrypick this commit on TensorFlow 2.7.1, TensorFlow 2.6.3, and TensorFlow 2.5.3, as these are also affected and still in supported range.

Timeline

1. Feb 4, 2022, 10:32 PM UTCCVE published
2. Apr 22, 2025, 3:50 PM UTCCISA SSVC assessment

   Exploitation: poc · Automatable: no · Technical impact: partial

3. Apr 22, 2025, 6:24 PM UTCRecord last updated

Affected versions

• tensorflow tensorflow

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3; From 2.7.0 · fixed in 2.7.1

• tensorflowPyPI

  fixed in 2.5.3

• tensorflowPyPI

  From 2.6.0 · fixed in 2.6.3

• tensorflowPyPI

  From 2.7.0 · fixed in 2.7.1

• tensorflow-cpuPyPI

  fixed in 2.5.3

• tensorflow-cpuPyPI

  From 2.6.0 · fixed in 2.6.3

• tensorflow-cpuPyPI

  From 2.7.0 · fixed in 2.7.1

• tensorflow-gpuPyPI

  fixed in 2.5.3

• tensorflow-gpuPyPI

  From 2.6.0 · fixed in 2.6.3

• tensorflow-gpuPyPI

  From 2.7.0 · fixed in 2.7.1

• tensorflow-gpuPyPI

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3

• tensorflow-cpuPyPI

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3

• tensorflowPyPI

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3; From 2.7.0 · fixed in 2.7.1

References 3