Instituto Nacional de ciberseguridad. Sección Incibe
Instituto Nacional de Ciberseguridad. Sección INCIBE-CERT

Vulnerabilidad en la implementación de las operaciones "tf.math.segment_*" en TensorFlow (CVE-2021-41195)

Gravedad CVSS v3.1:
MEDIA
Tipo:
CWE-190 Desbordamiento o ajuste de enteros
Fecha de publicación:
05/11/2021
Última modificación:
09/11/2021

Descripción

TensorFlow es una plataforma de código abierto para el aprendizaje automático. En las versiones afectadas, la implementación de las operaciones "tf.math.segment_*" resulta en un aborto relacionado con el fallo "CHECK" (y a una denegación de servicio) si un id de segmento en "segment_ids" es grande. Esto es similar a CVE-2021-29584 (y otras vulnerabilidades similares reportadas en TensorFlow, localizadas en APIs específicas): la implementación (tanto en CPU como en GPU) computa la forma de salida usando "AddDim". Sin embargo, si el número de elementos en el tensor sobrepasa un valor "int64_t", "AddDim" resulta en un fallo "CHECK" que provoca un "std::abort". En su lugar, el código debería usar "AddDimWithStatus". La corrección será incluida en TensorFlow versión 2.7.0. También será incluida este commit en TensorFlow versión 2.6.1, TensorFlow versión 2.5.2, y TensorFlow versión 2.4.4, ya que estos también están afectados y todavía están en el rango admitido

Productos y versiones vulnerables

CPE Desde Hasta
cpe:2.3:a:google:tensorflow:*:*:*:*:*:*:*:* 2.4.4 (excluyendo)
cpe:2.3:a:google:tensorflow:*:*:*:*:*:*:*:* 2.5.0 (incluyendo) 2.5.2 (excluyendo)
cpe:2.3:a:google:tensorflow:*:*:*:*:*:*:*:* 2.6.0 (incluyendo) 2.6.1 (excluyendo)