rsaenh: Fix the case when the CPEncrypt function is asked for the needed buffer size.

bf1ad6c3 · Karsten Elfenbein · Alexandre Julliard · 4b792c6d · bf1ad6c3
Commit bf1ad6c3 authored Jul 25, 2006 by Karsten Elfenbein Committed by Alexandre Julliard Jul 26, 2006
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

rsaenh.c dlls/rsaenh/rsaenh.c +6 -0

No files found.
--- a/dlls/rsaenh/rsaenh.c
+++ b/dlls/rsaenh/rsaenh.c
@@ -1889,6 +1889,12 @@ BOOL WINAPI RSAENH_CPEncrypt(HCRYPTPROV hProv, HCRYPTKEY hKey, HCRYPTHASH hHash,
        }

        dwEncryptedLen = (*pdwDataLen/pCryptKey->dwBlockLen+(Final?1:0))*pCryptKey->dwBlockLen;
+
+        if (pbData == NULL) {
+            *pdwDataLen = dwEncryptedLen;
+            return TRUE;
+        }
+
        for (i=*pdwDataLen; i<dwEncryptedLen && i<dwBufLen; i++) pbData[i] = dwEncryptedLen - *pdwDataLen;
        *pdwDataLen = dwEncryptedLen;