Division by 0 in `DenseCountSparseOutput`

Summary

TensorFlow is an end-to-end open source platform for machine learning. An attacker can cause a denial of service via a FPE runtime error in `tf.raw_ops.DenseCountSparseOutput`. This is because the implementation(https://github.com/tensorflow/tensorflow/blob/efff014f3b2d8ef6141da30c806faf141297eca1/tensorflow/core/kernels/count_ops.cc#L123-L127) computes a divisor value from user data but does not check that the result is 0 before doing the division. Since `data` is given by the `values` argument, `num_batch_elements` is 0. The fix will be included in TensorFlow 2.5.0. We will also cherrypick this commit on TensorFlow 2.4.2, and TensorFlow 2.3.3, as these are also affected.

Timeline

1. May 14, 2021, 7:10 PM UTCCVE published
2. Aug 3, 2024, 10:11 PM UTCRecord last updated

Affected versions

• tensorflow tensorflow

  fixed in 2.3.3; From 2.4.0 · fixed in 2.4.2

• tensorflowPyPI

  From 2.3.0 · fixed in 2.3.3

• tensorflowPyPI

  From 2.4.0 · fixed in 2.4.2

• tensorflow-cpuPyPI

  From 2.3.0 · fixed in 2.3.3

• tensorflow-cpuPyPI

  From 2.4.0 · fixed in 2.4.2

• 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

• 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

References 2