QID 980052
QID 980052: Python (pip) Security Update for tensorflow-gpu (GHSA-x7rp-74x2-mjf3)
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 `RaggedCountSparseOutput` implementation does not validate that the input arguments form a valid ragged tensor. In particular, there is no validation that the values in the `splits` tensor generate a valid partitioning of the `values` tensor. Thus, the [following code](https://github.com/tensorflow/tensorflow/blob/0e68f4d3295eb0281a517c3662f6698992b7b2cf/tensorflow/core/kernels/count_ops.cc#L248-L265
) sets up conditions to cause a heap buffer overflow:
```cc
auto per_batch_counts = BatchedMap<W>(num_batches);
int batch_idx = 0;
for (int idx = 0; idx < num_values; ++idx) {
while (idx >= splits_values(batch_idx)) {
batch_idx++;
}
const auto& value = values_values(idx);
if (value >= 0 && (maxlength_ <= 0 || value < maxlength_)) {
per_batch_counts[batch_idx - 1][value] = 1;
}
}
```
A `BatchedMap` is equivalent to a vector where each element is a hashmap. However, if the first element of `splits_values` is not 0, `batch_idx` will never be 1, hence there will be no hashmap at index 0 in `per_batch_counts`. Trying to access that in the user code results in a segmentation fault.
We recommend users to upgrade to TensorFlow 2.3.1.
- GHSA-x7rp-74x2-mjf3 -
github.com/advisories/GHSA-x7rp-74x2-mjf3
CVEs related to QID 980052
| Advisory ID | Software | Component | Link |
|---|---|---|---|
| GHSA-x7rp-74x2-mjf3 | tensorflow |
|
|
| GHSA-x7rp-74x2-mjf3 | tensorflow-cpu |
|
|
| GHSA-x7rp-74x2-mjf3 | tensorflow-gpu |
|