fixed sha512 dma

Author: miyazakh

src/wh_server_crypto.c

@@ -3027,14 +3027,14 @@ static int _HandleSha512Dma(whServerContext* ctx, uint16_t magic, uint16_t seq,
     whMessageCrypto_Sha512DmaResponse res;
     wc_Sha512*                        sha512 = ctx->crypto->algoCtx.sha512;
     int                               clientDevId;
+    int                               hashType = WC_HASH_TYPE_SHA512;
 
     /* Translate the request */
     ret = wh_MessageCrypto_TranslateSha512DmaRequest(
         magic, (whMessageCrypto_Sha512DmaRequest*)cryptoDataIn, &req);
     if (ret != WH_ERROR_OK) {
         return ret;
     }
-
     /* Ensure state sizes are the same */
     if (req.state.sz != sizeof(*sha512)) {
         res.dmaAddrStatus.badAddr = req.state;
@@ -3054,6 +3054,8 @@ static int _HandleSha512Dma(whServerContext* ctx, uint16_t magic, uint16_t seq,
             clientDevId = sha512->devId;
             /* overwrite the devId to that of the server for local crypto */
             sha512->devId = ctx->crypto->devId;
+            /* retrieve hash Type to handle 512, 512-224, or 512-256 */
+            hashType = sha512->hashType;
         }
     }
 
@@ -3073,8 +3075,19 @@ static int _HandleSha512Dma(whServerContext* ctx, uint16_t magic, uint16_t seq,
         if (ret == WH_ERROR_OK) {
 #ifdef DEBUG_CRYPTOCB_VERBOSE
             printf("[server]   wc_Sha512Final: outAddr=%p\n", outAddr);
+            printf("[server]   hashTpe: %d\n", hashType);
 #endif
-            ret = wc_Sha512Final(sha512, outAddr);
+             switch(hashType) {
+                case WC_HASH_TYPE_SHA512_224:
+                    ret = wc_Sha512_224Final(sha512, outAddr);
+                    break;
+                case WC_HASH_TYPE_SHA512_256:
+                    ret = wc_Sha512_256Final(sha512, outAddr);
+                    break;
+                default:
+                    ret = wc_Sha512Final(sha512, outAddr);
+                    break;
+            }
         }
 
         if (ret == WH_ERROR_OK) {