Impact
Under certain scenarios, Grappler component of TensorFlow can trigger a null pointer dereference. There are 2 places where this can occur, for the same malicious alteration of a SavedModel
file (fixing the first one would trigger the same dereference in the second place):
First, during constant folding, the GraphDef
might not have the required nodes for the binary operation:
NodeDef* mul_left_child = node_map_->GetNode(node->input(0));
NodeDef* mul_right_child = node_map_->GetNode(node->input(1));
// One child must be constant, and the second must be Conv op.
const bool left_child_is_constant = IsReallyConstant(*mul_left_child);
const bool right_child_is_constant = IsReallyConstant(*mul_right_child);
If a node is missing, the correposning mul_*child
would be null, and the dereference in the subsequent line would be incorrect.
We have a similar issue during IsIdentityConsumingSwitch
:
NodeDef* input_node = graph.GetNode(tensor_id.node());
return IsSwitch(*input_node);
Patches
We have patched the issue in GitHub commits 0a365c029e437be0349c31f8d4c9926b69fa3fa1 and 045deec1cbdebb27d817008ad5df94d96a08b1bf.
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.
For more information
Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
References
Impact
Under certain scenarios, Grappler component of TensorFlow can trigger a null pointer dereference. There are 2 places where this can occur, for the same malicious alteration of a
SavedModel
file (fixing the first one would trigger the same dereference in the second place):First, during constant folding, the
GraphDef
might not have the required nodes for the binary operation:If a node is missing, the correposning
mul_*child
would be null, and the dereference in the subsequent line would be incorrect.We have a similar issue during
IsIdentityConsumingSwitch
:NodeDef* input_node = graph.GetNode(tensor_id.node()); return IsSwitch(*input_node);
Patches
We have patched the issue in GitHub commits 0a365c029e437be0349c31f8d4c9926b69fa3fa1 and 045deec1cbdebb27d817008ad5df94d96a08b1bf.
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.
For more information
Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.
References