[crypt32] CryptUnprotectData test

Kees Cook kees at outflux.net
Wed May 18 08:04:22 CDT 2005


ChangeLog:
	Adds tests for CryptUnprotectData function.


-- 
Kees Cook                                            @outflux.net
-------------- next part --------------
--- dlls/crypt32/tests/protectdata.c.orig	2005-05-18 06:02:04.077803129 -0700
+++ dlls/crypt32/tests/protectdata.c	2005-05-18 06:03:32.475759540 -0700
@@ -113,11 +113,120 @@
     ok2(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r);
 }
 
+static void test_cryptunprotectdata(void)
+{
+    LONG r;
+    DATA_BLOB plain;
+    DATA_BLOB entropy;
+    BOOL okay;
+    WCHAR * data_desc;
+
+    entropy.pbData=(void*)key;
+    entropy.cbData=strlen(key)+1;
+
+    ok(protected, "CryptProtectData failed to run, so I can't test its output\n");
+    if (!protected) return;
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher,NULL,NULL,NULL,NULL,0,NULL);
+    ok(!okay,"Decrypting without destination\n");
+    r = GetLastError();
+    ok2(r == ERROR_INVALID_PARAMETER, "Wrong (%lu) GetLastError seen\n",r);
+
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(NULL,NULL,NULL,NULL,NULL,0,&plain);
+    ok(!okay,"Decrypting without source\n");
+    r = GetLastError();
+    ok2(r == ERROR_INVALID_PARAMETER, "Wrong (%lu) GetLastError seen\n",r);
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher_entropy,NULL,NULL,NULL,NULL,0,&plain);
+    ok(!okay,"Decrypting without needed entropy\n");
+    r = GetLastError();
+    ok2(r == ERROR_INVALID_DATA, "Wrong (%lu) GetLastError seen\n", r);
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+    data_desc=NULL;
+
+    /* without entropy */
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher,&data_desc,NULL,NULL,NULL,0,&plain);
+    ok(okay,"Decrypting without entropy\n");
+    r = GetLastError();
+    ok2(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r);
+
+    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
+    ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
+    ok(!strcmp(plain.pbData,secret),"Plain does not match secret\n");
+    ok(data_desc!=NULL,"Description not allocated\n");
+    ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
+
+    LocalFree(plain.pbData);
+    LocalFree(data_desc);
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+    data_desc=NULL;
+
+    /* with wrong entropy */
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher_entropy,&data_desc,&cipher_entropy,NULL,NULL,0,&plain);
+    ok(!okay,"Decrypting with wrong entropy\n");
+    r = GetLastError();
+    ok2(r == ERROR_INVALID_DATA, "Wrong (%lu) GetLastError seen\n",r);
+
+    /* with entropy */
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher_entropy,&data_desc,&entropy,NULL,NULL,0,&plain);
+    ok(okay,"Decrypting with entropy\n");
+    r = GetLastError();
+    ok2(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r);
+
+    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
+    ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
+    ok(!strcmp(plain.pbData,secret),"Plain does not match secret\n");
+    ok(data_desc!=NULL,"Description not allocated\n");
+    ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
+
+    LocalFree(plain.pbData);
+    LocalFree(data_desc);
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+    data_desc=NULL;
+
+    /* with entropy but no description */
+    SetLastError(0xDEADBEEF);
+    okay = CryptUnprotectData(&cipher_no_desc,&data_desc,&entropy,NULL,NULL,0,&plain);
+    ok(okay,"Decrypting with entropy and no description\n");
+    r = GetLastError();
+    ok2(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r);
+
+    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
+    ok(plain.cbData==strlen(secret2)+1,"Plain DATA_BLOB wrong length\n");
+    ok(!strcmp(plain.pbData,secret2),"Plain does not match secret\n");
+    ok(data_desc!=NULL,"Description not allocated\n");
+    ok(data_desc[0]=='\0',"Description not empty\n");
+
+    LocalFree(plain.pbData);
+
+    plain.pbData=NULL;
+    plain.cbData=0;
+}
+
 START_TEST(protectdata)
 {
     protected=FALSE;
 
     test_cryptprotectdata();
+    test_cryptunprotectdata();
 
     /* deinit globals here */
     if (cipher.pbData) LocalFree(cipher.pbData);


More information about the wine-patches mailing list