From b35fd8628e2bfb2d448cc0671a210217fe8c19e1 Mon Sep 17 00:00:00 2001
From: Zebediah Figura <zfigura@codeweavers.com>
Date: Thu, 4 Mar 2021 17:33:25 -0600
Subject: [PATCH] vkd3d-shader: Watch for allocation failure from yyparse().

Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
Signed-off-by: Henri Verbeet <hverbeet@codeweavers.com>
Signed-off-by: Matteo Bruni <mbruni@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 libs/vkd3d-shader/hlsl.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/libs/vkd3d-shader/hlsl.c b/libs/vkd3d-shader/hlsl.c
index 02fa4843..19bfc93a 100644
--- a/libs/vkd3d-shader/hlsl.c
+++ b/libs/vkd3d-shader/hlsl.c
@@ -1646,7 +1646,11 @@ int hlsl_compile_shader(const char *text, const struct vkd3d_shader_compile_info
     if (!hlsl_ctx_init(&ctx, message_context))
         return VKD3D_ERROR_OUT_OF_MEMORY;
 
-    hlsl_lexer_compile(&ctx, text);
+    if (hlsl_lexer_compile(&ctx, text) == 2)
+    {
+        hlsl_ctx_cleanup(&ctx);
+        return VKD3D_ERROR_OUT_OF_MEMORY;
+    }
 
     if (ctx.failed)
     {