Giovanni Mascellani : vkd3d-shader/ir: Validate SSA registers.

[Alexandre Julliard]

Module: vkd3d
Branch: master
Commit: e91da41dea66ddc1aa21980ee9c8858a55ec5825
URL:    https://gitlab.winehq.org/wine/vkd3d/-/commit/e91da41dea66ddc1aa21980ee9c885...

Author: Giovanni Mascellani <gmascellani(a)codeweavers.com>
Date:   Thu Dec 21 15:54:13 2023 +0100

vkd3d-shader/ir: Validate SSA registers.

---

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

diff --git a/libs/vkd3d-shader/ir.c b/libs/vkd3d-shader/ir.c
index c6ac0da0..a21af2fe 100644
--- a/libs/vkd3d-shader/ir.c
+++ b/libs/vkd3d-shader/ir.c
@@ -1604,6 +1604,19 @@ static void vsir_validate_register(struct validation_context *ctx,
                         reg->idx[0].offset, temp_count);
             break;
 
+        case VKD3DSPR_SSA:
+            if (reg->idx_count != 1)
+                validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_INDEX_COUNT, "Invalid index count %u for a SSA register.",
+                        reg->idx_count);
+
+            if (reg->idx_count >= 1 && reg->idx[0].rel_addr)
+                validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_INDEX, "Non-NULL relative address for a SSA register.");
+
+            if (reg->idx_count >= 1 && reg->idx[0].offset >= ctx->parser->shader_desc.ssa_count)
+                validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_INDEX, "SSA register index %u exceeds the maximum count %u.",
+                        reg->idx[0].offset, ctx->parser->shader_desc.ssa_count);
+            break;
+
         case VKD3DSPR_NULL:
             if (reg->idx_count != 0)
                 validator_error(ctx, VKD3D_SHADER_ERROR_VSIR_INVALID_INDEX_COUNT, "Invalid index count %u for a NULL register.",