From 02d49fa6762cebf94c9806c95d5e41470c0a1dec Mon Sep 17 00:00:00 2001
From: Brett Nicholas <7547222+bigbrett@users.noreply.github.com>
Date: Fri, 17 Oct 2025 09:23:58 -0600
Subject: [PATCH] add missing post-write DMA callbacks for AES GCM

---
 src/wh_server_crypto.c | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/src/wh_server_crypto.c b/src/wh_server_crypto.c
index 2b905f53c..039c89e15 100644
--- a/src/wh_server_crypto.c
+++ b/src/wh_server_crypto.c
@@ -2112,6 +2112,33 @@ static int _HandleAesGcmDma(whServerContext* ctx, uint16_t magic, uint16_t seq,
         }
     }
 
+    /* Post-write DMA address processing for output/authTag (on success) */
+    if (ret == WH_ERROR_OK) {
+        if (req.output.sz > 0) {
+            int rc2 = wh_Server_DmaProcessClientAddress(
+                ctx, req.output.addr, &outAddr, req.output.sz,
+                WH_DMA_OPER_CLIENT_WRITE_POST, (whServerDmaFlags){0});
+            if (rc2 != WH_ERROR_OK) {
+                if (rc2 == WH_ERROR_ACCESS) {
+                    res.dmaAddrStatus.badAddr = req.output;
+                }
+                ret = rc2;
+            }
+        }
+        /* During encryption, the auth tag is written to client memory */
+        if (ret == WH_ERROR_OK && req.enc && req.authTag.sz > 0) {
+            int rc2 = wh_Server_DmaProcessClientAddress(
+                ctx, req.authTag.addr, &authTagAddr, req.authTag.sz,
+                WH_DMA_OPER_CLIENT_WRITE_POST, (whServerDmaFlags){0});
+            if (rc2 != WH_ERROR_OK) {
+                if (rc2 == WH_ERROR_ACCESS) {
+                    res.dmaAddrStatus.badAddr = req.authTag;
+                }
+                ret = rc2;
+            }
+        }
+    }
+
     wc_AesFree(aes);
     res.outSz = outSz;