Giovanni Mascellani : vkd3d-shader/ir: Check that REP blocks are correctly nested.

[Alexandre Julliard]

Module: vkd3d
Branch: master
Commit: 93632fb40710fb852e9659db9e2155a82a2475d9
URL:    https://gitlab.winehq.org/wine/vkd3d/-/commit/93632fb40710fb852e9659db9e2155...

Author: Giovanni Mascellani <gmascellani(a)codeweavers.com>
Date:   Mon Nov  6 13:56:38 2023 +0100

vkd3d-shader/ir: Check that REP blocks are correctly nested.

---

 libs/vkd3d-shader/ir.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index 71a23fa0..bc48df8c 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -1684,6 +1684,23 @@ static void vsir_validate_instruction(struct validation_context *ctx)
                 --ctx->depth;
             break;
 
+        case VKD3DSIH_REP:
+            vsir_validate_dst_count(ctx, instruction, 0);
+            vsir_validate_src_count(ctx, instruction, 1);
+            if (!vkd3d_array_reserve((void **)&ctx->blocks, &ctx->blocks_capacity, ctx->depth + 1, sizeof(*ctx->blocks)))
+                return;
+            ctx->blocks[ctx->depth++] = instruction->handler_idx;
+            break;
+
+        case VKD3DSIH_ENDREP:
+            vsir_validate_dst_count(ctx, instruction, 0);
+            vsir_validate_src_count(ctx, instruction, 0);
+            if (ctx->depth == 0 || ctx->blocks[ctx->depth - 1] != VKD3DSIH_REP)
+                validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_INSTRUCTION_NESTING, "ENDREP instruction doesn't terminate REP block.");
+            else
+                --ctx->depth;
+            break;
+
         default:
             break;
     }