[PATCH v5 7/9] vkd3d-shader: Validate register types.

[Giovanni Mascellani]

From: Giovanni Mascellani

---
 libs/vkd3d-shader/ir.c                   | 34 ++++++++++++++++++++++++
 libs/vkd3d-shader/vkd3d_shader_private.h |  1 +
 2 files changed, 35 insertions(+)

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index ce725672..2015ab25 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -1334,12 +1334,46 @@ static void validator_error(struct validation_context *ctx, enum vkd3d_shader_er
     vkd3d_string_buffer_cleanup(&buf);
 }

+static void vkd3d_shader_validate_register(struct validation_context *ctx,
+        const struct vkd3d_shader_register *reg)
+{
+    switch (reg->type)
+    {
+        case VKD3DSPR_COUNT:
+        case VKD3DSPR_INVALID:
+            validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_REGISTER_TYPE, "Invalid register type.\n");
+            break;
+
+        default:
+            break;
+    }
+}
+
+static void vkd3d_shader_validate_dst_param(struct validation_context *ctx,
+        const struct vkd3d_shader_dst_param *dst)
+{
+    vkd3d_shader_validate_register(ctx, &dst->reg);
+}
+
+static void vkd3d_shader_validate_src_param(struct validation_context *ctx,
+        const struct vkd3d_shader_src_param *src)
+{
+    vkd3d_shader_validate_register(ctx, &src->reg);
+}
+
 static void vkd3d_shader_validate_instruction(struct validation_context *ctx)
 {
     const struct vkd3d_shader_instruction *instruction = &ctx->parser->instructions.elements[ctx->instruction_idx];
+    size_t i;

ctx->parser->location = instruction->location;

+    for (i = 0; i < instruction->dst_count; ++i)
+        vkd3d_shader_validate_dst_param(ctx, &instruction->dst[i]);
+
+    for (i = 0; i < instruction->src_count; ++i)
+        vkd3d_shader_validate_src_param(ctx, &instruction->src[i]);
+
     switch (instruction->handler_idx)
     {
         case VKD3DSIH_INVALID:
diff --git a/libs/vkd3d-shader/vkd3d_shader_private.h b/libs/vkd3d-shader/vkd3d_shader_private.h
index 3525f6f4..cf106693 100644
--- a/libs/vkd3d-shader/vkd3d_shader_private.h
+++ b/libs/vkd3d-shader/vkd3d_shader_private.h
@@ -184,6 +184,7 @@ enum vkd3d_shader_error

VKD3D_SHADER_ERROR_VSIR_NOT_IMPLEMENTED             = 9000,
     VKD3D_SHADER_ERROR_VSIR_INVALID_HANDLER             = 9001,
+    VKD3D_SHADER_ERROR_VSIR_INVALID_REGISTER_TYPE       = 9002,
 };

enum vkd3d_shader_opcode