Henri Verbeet : vkd3d-shader/sm4: Set an error state when shader_sm4_error() is called.

[Alexandre Julliard]

Module: vkd3d
Branch: master
Commit: a0c2e52b4b4c0d191677397d3030a2f0da565af9
URL:    https://source.winehq.org/git/vkd3d.git/?a=commit;h=a0c2e52b4b4c0d191677397d3030a2f0da565af9

Author: Henri Verbeet <hverbeet at codeweavers.com>
Date:   Wed Oct  6 17:11:45 2021 +0200

vkd3d-shader/sm4: Set an error state when shader_sm4_error() is called.

Signed-off-by: Henri Verbeet <hverbeet at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 libs/vkd3d-shader/dxbc.c                 | 3 +++
 libs/vkd3d-shader/glsl.c                 | 4 ++--
 libs/vkd3d-shader/trace.c                | 3 +++
 libs/vkd3d-shader/vkd3d_shader_main.c    | 5 ++++-
 libs/vkd3d-shader/vkd3d_shader_private.h | 1 +
 5 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/libs/vkd3d-shader/dxbc.c b/libs/vkd3d-shader/dxbc.c
index e9b8954..527f69b 100644
--- a/libs/vkd3d-shader/dxbc.c
+++ b/libs/vkd3d-shader/dxbc.c
@@ -183,6 +183,8 @@ static void VKD3D_PRINTF_FUNC(3, 4) shader_sm4_error(struct vkd3d_shader_sm4_par
     va_start(args, format);
     vkd3d_shader_verror(sm4->p.message_context, &sm4->p.location, error, format, args);
     va_end(args);
+
+    sm4->p.failed = true;
 }
 
 static bool shader_sm4_read_src_param(struct vkd3d_shader_sm4_parser *priv, const uint32_t **ptr,
@@ -1665,6 +1667,7 @@ void shader_sm4_reset(struct vkd3d_shader_parser *parser)
     struct vkd3d_shader_sm4_parser *sm4 = vkd3d_shader_sm4_parser(parser);
 
     parser->ptr = sm4->start;
+    parser->failed = false;
 }
 
 int vkd3d_shader_sm4_parser_create(const struct vkd3d_shader_compile_info *compile_info,
diff --git a/libs/vkd3d-shader/glsl.c b/libs/vkd3d-shader/glsl.c
index 9858098..b90a5d9 100644
--- a/libs/vkd3d-shader/glsl.c
+++ b/libs/vkd3d-shader/glsl.c
@@ -114,8 +114,8 @@ int vkd3d_glsl_generator_generate(struct vkd3d_glsl_generator *generator,
         vkd3d_glsl_handle_instruction(generator, &ins);
     }
 
-    if (generator->failed)
-        return VKD3D_ERROR;
+    if (parser->failed || generator->failed)
+        return VKD3D_ERROR_INVALID_SHADER;
 
     vkd3d_string_buffer_printf(&generator->buffer, "}\n");
 
diff --git a/libs/vkd3d-shader/trace.c b/libs/vkd3d-shader/trace.c
index 2858c7f..5710424 100644
--- a/libs/vkd3d-shader/trace.c
+++ b/libs/vkd3d-shader/trace.c
@@ -1937,6 +1937,9 @@ enum vkd3d_result vkd3d_dxbc_binary_to_text(struct vkd3d_shader_parser *parser,
         }
     }
 
+    if (parser->failed)
+        result = VKD3D_ERROR_INVALID_SHADER;
+
     if ((code = vkd3d_malloc(buffer->content_size)))
     {
         memcpy(code, buffer->buffer, buffer->content_size);
diff --git a/libs/vkd3d-shader/vkd3d_shader_main.c b/libs/vkd3d-shader/vkd3d_shader_main.c
index f02514b..696c332 100644
--- a/libs/vkd3d-shader/vkd3d_shader_main.c
+++ b/libs/vkd3d-shader/vkd3d_shader_main.c
@@ -958,7 +958,7 @@ static int scan_dxbc(const struct vkd3d_shader_compile_info *compile_info,
         }
     }
 
-    ret = VKD3D_OK;
+    ret = parser->failed ? VKD3D_ERROR_INVALID_SHADER : VKD3D_OK;
 
 done:
     vkd3d_shader_scan_context_cleanup(&context);
@@ -1084,6 +1084,9 @@ static int compile_dxbc_tpf(const struct vkd3d_shader_compile_info *compile_info
             break;
     }
 
+    if (parser->failed)
+        ret = VKD3D_ERROR_INVALID_SHADER;
+
     if (ret >= 0)
         ret = vkd3d_dxbc_compiler_generate_spirv(spirv_compiler, compile_info, out);
 
diff --git a/libs/vkd3d-shader/vkd3d_shader_private.h b/libs/vkd3d-shader/vkd3d_shader_private.h
index f937ff5..ba17d29 100644
--- a/libs/vkd3d-shader/vkd3d_shader_private.h
+++ b/libs/vkd3d-shader/vkd3d_shader_private.h
@@ -897,6 +897,7 @@ struct vkd3d_shader_parser
 {
     struct vkd3d_shader_message_context *message_context;
     struct vkd3d_shader_location location;
+    bool failed;
 
     struct vkd3d_shader_desc shader_desc;
     struct vkd3d_shader_version shader_version;