WarpGuard: Protected-Site Control-Flow Integrity for CUDA SASS Binaries

Resumen Técnico: WARPGUARD

Planteamiento del Problema
Investigaciones recientes sobre explotación demuestran que los errores de memoria de la GPU (p. ej., desbordamientos de búfer) pueden escalar hacia la corrupción del flujo de control en el lado del dispositivo. Los atacantes pueden corromper continuaciones de retorno, punteros de función, entradas de tablas de despacho o objetivos de salto, que son consumidos posteriormente por instrucciones de la GPU para redirigir la ejecución. Aunque la Integridad del Flujo de Control (CFI) es la defensa estándar para las CPU, aplicar CFI a binarios CUDA desplegados es un desafío. Los kernels de producción suelen distribuirse como fatbinaries o cubins, donde la frontera de seguridad relevante es el código SASS (Streaming ASSembly) de NVIDIA ejecutado, no el código fuente o la representación intermedia PTX (Parallel Thread Execution).

El nivel SASS introduce complejidades ausentes en la CFI de CPU:

• Transformaciones de Reducción (Lowering): Las llamadas a nivel de fuente pueden ser integradas (inlined), los estados de retorno pueden mantenerse en registros en lugar de volcarse a memoria, y los objetivos indirectos pueden ser optimizados o eliminados.
• Ejecución SIMT: Una única instrucción estática puede ejecutarse de forma distinta a través de los carriles de un warp (warp-uniform vs. lane-local), lo que requiere comprobaciones granulares.
• Brechas de Evidencia: No todos los sitios de flujo de control en un binario poseen evidencia suficiente para derivar una política sólida. Tratar los sitios no compatibles como "permitidos" crea agujeros de seguridad, mientras que tratarlos como "protegidos" sin evidencia es carente de rigor.

Las herramientas de seguridad de GPU existentes se centran en la seguridad de memoria o la inyección de código, pero no aplican la CFI en la frontera de SASS ejecutada con una contabilidad explícita de los sitios que no pueden ser protegidos.

Metodología: WARPGUARD
WARPGUARD es el primer sistema de CFI diseñado específicamente para binarios de dispositivo CUDA que operan sobre SASS ejecutado. Aplica una política de "sitio protegido", lo que significa que solo verifica las transferencias de flujo de control donde existe evidencia binaria suficiente para derivar una política sólida.

1. Recuperación y Clasificación de Sitios:
   WARPGUARD analiza binarios SASS para identificar sitios de consumo de flujo de control (retornos, llamadas indirectas, saltos indirectos). Clasifica cada sitio en uno de seis resultados:

• Protegido: Existe evidencia suficiente; se inserta una comprobación.
• Borde fijo (Fixed-edge): El objetivo es estático (p. ej., llamada directa); no se requiere comprobación dinámica.
• No compatible (Unsupported): Falta evidencia o es ambigua; el sitio se excluye de la protección.
• Sin superficie (No-surface): No existe transferencia dinámica (p. ej., código totalmente integrado/inlined).
• Reserva/Excluido por perfil (Fallback/Profile-excluded): El análisis falló o el modo está desactivado.
  Crucialmente, WARPGUARD no permite silenciosamente los sitios no compatibles; estos se contabilizan explícitamente fuera del denominador protegido.

2. CFI de Borde Posterior (Retornos):
   Para retornos protegidos, WARPGUARD autentica el estado de la continuación antes de que la instrucción RET libere el control.

• Mecanismo: En una llamada protegida, el sistema registra la continuación de retorno esperada, el ID de la llamada, la profundidad y un token con clave (MAC) en un estado privado de confianza en el backend.
• Verificación: En el sitio de retorno, el sistema lee la continuación observada, verifica el token contra el estado de confianza y lo compara con el valor esperado.
• Seguridad: La metadata de sombra escribible no es autoritativa; solo el token con clave del host/entorno de ejecución de confianza autoriza la transferencia. Si la verificación falla, el sistema falla de forma cerrada (fails closed, lanza un trap) antes de que ocurra la transferencia inválida.

3. CFI de Borde Anterior (Llamadas Indirectas/Saltos):
   Para transferencias indirectas protegidas, WARPGUARD valida el objetivo dinámico contra un conjunto de objetivos recuperado por sitio.

• Recuperación: Los conjuntos de objetivos se derivan de la metadata del binario, las relocalizaciones de constantes y el análisis de tablas de despacho o funciones que toman direcciones.
• Verificación: Antes de que se libere una transferencia indirecta, el objetivo observado se comprueba contra el conjunto permitido para ese sitio específico.
• Granularidad: Las comprobaciones son conservadoras y locales al carril (lane-local) por defecto para manejar la divergencia SIMT, asegurando que un carril corrompido no pueda ocultarse dentro de un agregado uniforme del warp.

4. Backends de Implementación:
   WARPGUARD se implementa mediante tres backends distintos para separar las responsabilidades:

• WG-NVBit: Un amplio backend de referencia que utiliza NVBit para instrumentación dinámica y evaluación de ataques.
• WG-ST: Un carril de tiempo de trampa estática (static-trampoline timing lane) para ejecución sin callbacks en superficies específicas de sm_89, utilizado para comparar la sobrecarga sin el suelo de callbacks de NVBit.
• WG-PC: Un backend de caché de parches que aplica parches SASS revisados y respaldados por manifiestos para superficies sm_89, permitiendo la prevención sin callbacks.

Contribuciones Clave

• Formulación de CFI de Sitio Protegido: El artículo formula la CFI de binarios CUDA como una aplicación de "verificación antes de la liberación" en los sitios de consumo de SASS recuperados. Contabiliza explícitamente los bordes fijos, los sitios no compatibles y las exclusiones de perfil, separándolos de la cobertura protegida.
• Mecanismo e Implementación: WARPGUARD vincula la política al SASS recuperado, autentica el estado del borde posterior usando tokens con clave, valida los objetivos del borde anterior por sitio y mantiene la telemetría separada de la lógica de autorización.
• Evaluación con Denominadores Separados: El sistema se evalúa en 77 artefactos CUDA, clasificando 51,621 sitios de flujo de control de SASS y registrando 52.2 millones de comprobaciones dinámicas. La evaluación distingue entre la superficie recuperada, la ejecución protegida y las superficies no compatibles residuales.

Resultos

• Cobertura: WARPGUARD clasificó con éxito 51,621 sitios, incluyendo 1,343 retornos y 154 entradas de conjuntos de objetivos de borde anterior soportadas. Identificó 15 entradas de borde anterior no compatibles y 10 entradas de reserva, asegurando que estas no fueran tratadas incorrectamente como protegidas.
• Bloqueo de Ataques: En ataques representativos de corrupción de borde posterior y anterior, la ejecución nativa alcanzó el comportamiento seleccionado por el atacante, mientras que el modo de aplicación de WARPGUARD falló de forma cerrada antes de liberar la transferencia inválida. Esto se mantuvo para:
  • Secuestros de retorno y cadenas de gadgets.
  • Ataques de divergencia SIMT (retornos de carriles divergentes).
  • Corrupción de objetivos de borde anterior (tablas de despacho, punteros de función).
  • Manipulación de metadata (tokens, telemetría).
• Evidencia del Mundo Real: Casos de código público (p. ej., PPL-CUDA-SMC, PNNL SV-Sim, callbacks de cuFFT, GooFit) confirmaron que los mismos patrones de consumo de SASS existen en sistemas CUDA reales. WARPGUARD recuperó y protegió con éxito estos sitios.
• Rendimiento:
  • WG-NVBit: Como era de esperar para la instrumentación dinámica, la sobrecarga fue alta (799–927% para cargas de trabajo >50ms), dominada por el despacho de ayudantes y los costes de callback.
  • WG-ST/WG-PC: En superficies sm_89 emparejadas, los backends sin callbacks eliminaron el suelo de tiempo de NVBit. WG-PC demostró que la prevención sin callbacks es factible para manifiestos revisados, aunque requiere una validación estricta del manifiesto.

Significado y Reivindicaciones
El artículo afirma que WARPGUARD es, según conocimiento de los autores, el primer sistema de CFI evaluado para binarios de dispositivo CUDA que opera directamente sobre SASS ejecutado. Su importancia radica en:

1. Cambio de la Frontera: Mover la aplicación de la CFI desde la fuente/PTX hacia el SASS realmente ejecutado, capturando los hechos reales de flujo de control (integración, vertidos, máscaras SIMT).
2. Contabilidad Explícita: Negarse a tratar los sitios desconocidos o no compatibles como seguros, evitando así las políticas de "permitir todo" que podrían socavar la seguridad.
3. Aplicación Auditable: Proporcionar un mecanismo para separar la sobrecarga de la instrumentación dinámica de la lógica central de CFI y para distinguir entre los sitios protegidos y las superficies de ataque residuales.

Los autores señalan modestamente que WARPGUARD no proporciona seguridad de memoria, integridad de código ni protección contra el compromiso del host. Es complementario a las herramientas de seguridad de memoria, centrándose específicamente en el punto donde el estado corrupto se consume como flujo de control. El sistema depende de un host/entorno de ejecución de confianza para la gestión de claves; si el estado privado del backend se expone a la GPU, las garantías de seguridad se degradan.