Assertion failure based denial of service in Tensorflow

Summary

Tensorflow is an Open Source Machine Learning Framework. The implementation of `*Bincount` operations allows malicious users to cause denial of service by passing in arguments which would trigger a `CHECK`-fail. There are several conditions that the input arguments must satisfy. Some are not caught during shape inference and others are not caught during kernel implementation. This results in `CHECK` failures later when the output tensors get allocated. 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 3, 2022, 1:43 PM UTCCVE published
2. Apr 25, 2025, 3:47 PM UTCCISA SSVC assessment

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

3. May 5, 2025, 4:31 PM UTCRecord last updated

Affected versions

• n/a n/a

  Version range not reported

• tensorflowPyPI

  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-cpuPyPI

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3

• tensorflow-gpuPyPI

  fixed in 2.5.3; From 2.6.0 · fixed in 2.6.3

References 3