Segmentation fault in tensorflow-lite

Overview

Impact

If a TFLite saved model uses the same tensor as both input and output of an operator, then, depending on the operator, we can observe a segmentation fault or just memory corruption.

Patches

We have patched the issue in d58c96946b and will release patch releases for all versions between 1.15 and 2.3.

We recommend users to upgrade to TensorFlow 1.15.4, 2.0.3, 2.1.2, 2.2.1, or 2.3.1.

Workarounds

A potential workaround would be to add a custom Verifier to the model loading code to ensure that no operator reuses tensors as both inputs and outputs. Care should be taken to check all types of inputs (i.e., constant or variable tensors as well as optional tensors).

For more information

Please consult our security guide for more information regarding the security model and how to contact us with issues and questions.

Attribution

This vulnerability has been discovered from a variant analysis of GHSA-cvpc-8phh-8f45.

Affected packages

• tensorflow < 1.15.4
• tensorflow >= 2.0.0, < 2.0.3
• tensorflow >= 2.1.0, < 2.1.2
• tensorflow >= 2.2.0, < 2.2.1
• tensorflow >= 2.3.0, < 2.3.1
• tensorflow-cpu < 1.15.4
• tensorflow-cpu >= 2.0.0, < 2.0.3
• tensorflow-cpu >= 2.1.0, < 2.1.2
• tensorflow-cpu >= 2.2.0, < 2.2.1
• tensorflow-gpu < 1.15.4
• tensorflow-gpu >= 2.0.0, < 2.0.3
• tensorflow-gpu >= 2.1.0, < 2.1.2
• tensorflow-gpu >= 2.2.0, < 2.2.1
• tensorflow-cpu >= 2.3.0, < 2.3.1
• tensorflow-gpu >= 2.3.0, < 2.3.1

Remediation

Upgrade to a patched release:

• tensorflow 1.15.4
• tensorflow 2.0.3
• tensorflow 2.1.2
• tensorflow 2.2.1
• tensorflow 2.3.1
• tensorflow-cpu 1.15.4
• tensorflow-cpu 2.0.3
• tensorflow-cpu 2.1.2
• tensorflow-cpu 2.2.1
• tensorflow-gpu 1.15.4
• tensorflow-gpu 2.0.3
• tensorflow-gpu 2.1.2
• tensorflow-gpu 2.2.1
• tensorflow-cpu 2.3.1
• tensorflow-gpu 2.3.1