Tensorflow is an Open Source Machine Learning Framework. A malicious user can cause a denial of service by altering a SavedModel
such that TensorByteSize
would trigger CHECK
failures. TensorShape
constructor throws a CHECK
-fail if shape is partial or has a number of elements that would overflow the size of an int
. The PartialTensorShape
constructor instead does not cause a CHECK
-abort if the shape is partial, which is exactly what this function needs to be able to return -1
. 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.
The product contains an assert() or similar statement that can be triggered by an attacker, which leads to an application exit or other behavior that is more severe than necessary.
Name | Vendor | Start Version | End Version |
---|---|---|---|
Tensorflow | * | 2.5.2 (including) | |
Tensorflow | 2.6.0 (including) | 2.6.2 (including) | |
Tensorflow | 2.7.0 (including) | 2.7.0 (including) |
While assertion is good for catching logic errors and reducing the chances of reaching more serious vulnerability conditions, it can still lead to a denial of service. For example, if a server handles multiple simultaneous connections, and an assert() occurs in one single connection that causes all other connections to be dropped, this is a reachable assertion that leads to a denial of service.