QID 981519
QID 981519: Python (pip) Security Update for tensorflow-gpu (GHSA-f8h4-7rgh-q2gm)
Security update has been released for tensorflow-gpu,tensorflow,tensorflow-cpu to fix the vulnerability.
Note: The preceding description block is extracted directly from the security advisory. Using automation, we have attempted to clean and format it as much as possible without introducing additional issues.
The implementation for `tf.raw_ops.ExperimentalDatasetToTFRecord` and `tf.raw_ops.DatasetToTFRecord` can trigger heap buffer overflow and segmentation fault:
```python
import tensorflow as tf
dataset = tf.data.Dataset.range(3)
dataset = tf.data.experimental.to_variant(dataset)
tf.raw_ops.ExperimentalDatasetToTFRecord(
input_dataset=dataset,
filename='/tmp/output',
compression_type=')
```
The [implementation](https://github.com/tensorflow/tensorflow/blob/f24faa153ad31a4b51578f8181d3aaab77a1ddeb/tensorflow/core/kernels/data/experimental/to_tf_record_op.cc#L93-L102) assumes that all records in the dataset are of string type. However, there is no check for that, and the example given above uses numeric types.
The fix will be included in TensorFlow 2.6.0. We will also cherrypick this commit on TensorFlow 2.5.1, TensorFlow 2.4.3, and TensorFlow 2.3.4, as these are also affected and still in supported range.
- GHSA-f8h4-7rgh-q2gm -
github.com/advisories/GHSA-f8h4-7rgh-q2gm
CVEs related to QID 981519
| Advisory ID | Software | Component | Link |
|---|---|---|---|
| GHSA-f8h4-7rgh-q2gm | tensorflow |
|
|
| GHSA-f8h4-7rgh-q2gm | tensorflow-cpu |
|
|
| GHSA-f8h4-7rgh-q2gm | tensorflow-gpu |
|