In TensorFlow before 1.15, a heap buffer overflow in UnsortedSegmentSum can be produced when the Index template argument is int32.
Date: 2019-12
Affected: google/tensorflow
Attack vector: other
Category: vulnerability-disclosure
Severity: Low
CVE: CVE-2019-16778
CWE: CWE-122
CWE-681
CVSS: 2.6 (CVSS:3.1/AV:N/AC:H/PR:L/UI:R/S:U/C:N/I:N/A:L)

In TensorFlow before 1.15, a heap buffer overflow in UnsortedSegmentSum can be produced when the Index template argument is int32. In this case data_size and num_segments fields are truncated from int64 to int32 and can produce negative numbers, resulting in accessing out of bounds heap memory. This is unlikely to be exploitable and was detected and fixed internally in TensorFlow 1.15 and 2.0.


Mitigations:

Architectural layers (MAESTRO):