dlls/crypt32/tests/protectdata.c

   1 /*
   2  * Unit test suite for crypt32.dll's CryptProtectData/CryptUnprotectData
   3  *
   4  * Copyright 2005 Kees Cook <kees@outflux.net>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdio.h>
  22 #include <stdarg.h>
  23 #include <windef.h>
  24 #include <winbase.h>
  25 #include <winerror.h>
  26 #include <wincrypt.h>
  27
  28 #include "wine/test.h"
  29
  30 static BOOL (WINAPI *pCryptProtectData)(DATA_BLOB*,LPCWSTR,DATA_BLOB*,PVOID,CRYPTPROTECT_PROMPTSTRUCT*,DWORD,DATA_BLOB*);
  31 static BOOL (WINAPI *pCryptUnprotectData)(DATA_BLOB*,LPWSTR*,DATA_BLOB*,PVOID,CRYPTPROTECT_PROMPTSTRUCT*,DWORD,DATA_BLOB*);
  32
  33 static char  secret[]     = "I am a super secret string that no one can see!";
  34 static char  secret2[]    = "I am a super secret string indescribable string";
  35 static char  key[]        = "Wibble wibble wibble";
  36 static const WCHAR desc[] = {'U','l','t','r','a',' ','s','e','c','r','e','t',' ','t','e','s','t',' ','m','e','s','s','a','g','e',0};
  37 static BOOL protected = FALSE; /* if true, the unprotect tests can run */
  38 static DATA_BLOB cipher;
  39 static DATA_BLOB cipher_entropy;
  40 static DATA_BLOB cipher_no_desc;
  41
  42 static void test_cryptprotectdata(void)
  43 {
  44     LONG r;
  45     DATA_BLOB plain;
  46     DATA_BLOB entropy;
  47
  48     plain.pbData=(void*)secret;
  49     plain.cbData=strlen(secret)+1;
  50
  51     entropy.pbData=(void*)key;
  52     entropy.cbData=strlen(key)+1;
  53
  54     SetLastError(0xDEADBEEF);
  55     protected = pCryptProtectData(NULL,desc,NULL,NULL,NULL,0,&cipher);
  56     ok(!protected, "Encrypting without plain data source.\n");
  57     r = GetLastError();
  58     ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
  59
  60     SetLastError(0xDEADBEEF);
  61     protected = pCryptProtectData(&plain,desc,NULL,NULL,NULL,0,NULL);
  62     ok(!protected, "Encrypting without cipher destination.\n");
  63     r = GetLastError();
  64     ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
  65
  66     cipher.pbData=NULL;
  67     cipher.cbData=0;
  68
  69     /* without entropy */
  70     SetLastError(0xDEADBEEF);
  71     protected = pCryptProtectData(&plain,desc,NULL,NULL,NULL,0,&cipher);
  72     ok(protected, "Encrypting without entropy.\n");
  73     r = GetLastError();
  74     ok(r == ERROR_SUCCESS ||
  75        r == ERROR_IO_PENDING, /* win2k */
  76        "Expected ERROR_SUCCESS or ERROR_IO_PENDING, got %d\n",r);
  77
  78     cipher_entropy.pbData=NULL;
  79     cipher_entropy.cbData=0;
  80
  81     /* with entropy */
  82     SetLastError(0xDEADBEEF);
  83     protected = pCryptProtectData(&plain,desc,&entropy,NULL,NULL,0,&cipher_entropy);
  84     ok(protected, "Encrypting with entropy.\n");
  85
  86     cipher_no_desc.pbData=NULL;
  87     cipher_no_desc.cbData=0;
  88
  89     /* with entropy but no description */
  90     plain.pbData=(void*)secret2;
  91     plain.cbData=strlen(secret2)+1;
  92     SetLastError(0xDEADBEEF);
  93     protected = pCryptProtectData(&plain,NULL,&entropy,NULL,NULL,0,&cipher_no_desc);
  94     if (!protected)
  95     {
  96         /* fails in win2k */
  97         ok(GetLastError() == ERROR_INVALID_PARAMETER,
  98            "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
  99     }
 100 }
 101
 102 static void test_cryptunprotectdata(void)
 103 {
 104     LONG r;
 105     DATA_BLOB plain;
 106     DATA_BLOB entropy;
 107     BOOL okay;
 108     WCHAR * data_desc;
 109
 110     entropy.pbData=(void*)key;
 111     entropy.cbData=strlen(key)+1;
 112
 113     /* fails in win2k */
 114     if (!protected)
 115     {
 116         skip("CryptProtectData failed to run\n");
 117         return;
 118     }
 119
 120     plain.pbData=NULL;
 121     plain.cbData=0;
 122
 123     SetLastError(0xDEADBEEF);
 124     okay = pCryptUnprotectData(&cipher,NULL,NULL,NULL,NULL,0,NULL);
 125     ok(!okay,"Decrypting without destination\n");
 126     r = GetLastError();
 127     ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
 128
 129     SetLastError(0xDEADBEEF);
 130     okay = pCryptUnprotectData(NULL,NULL,NULL,NULL,NULL,0,&plain);
 131     ok(!okay,"Decrypting without source\n");
 132     r = GetLastError();
 133     ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
 134
 135     plain.pbData=NULL;
 136     plain.cbData=0;
 137
 138     SetLastError(0xDEADBEEF);
 139     okay = pCryptUnprotectData(&cipher_entropy,NULL,NULL,NULL,NULL,0,&plain);
 140     ok(!okay,"Decrypting without needed entropy\n");
 141     r = GetLastError();
 142     ok(r == ERROR_INVALID_DATA, "Wrong (%u) GetLastError seen\n", r);
 143
 144     plain.pbData=NULL;
 145     plain.cbData=0;
 146     data_desc=NULL;
 147
 148     /* without entropy */
 149     SetLastError(0xDEADBEEF);
 150     okay = pCryptUnprotectData(&cipher,&data_desc,NULL,NULL,NULL,0,&plain);
 151     ok(okay,"Decrypting without entropy\n");
 152
 153     ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
 154     ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
 155     ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
 156     ok(data_desc!=NULL,"Description not allocated\n");
 157     ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
 158
 159     LocalFree(plain.pbData);
 160     LocalFree(data_desc);
 161
 162     plain.pbData=NULL;
 163     plain.cbData=0;
 164     data_desc=NULL;
 165
 166     /* with wrong entropy */
 167     SetLastError(0xDEADBEEF);
 168     okay = pCryptUnprotectData(&cipher_entropy,&data_desc,&cipher_entropy,NULL,NULL,0,&plain);
 169     ok(!okay,"Decrypting with wrong entropy\n");
 170     r = GetLastError();
 171     ok(r == ERROR_INVALID_DATA, "Wrong (%u) GetLastError seen\n",r);
 172
 173     /* with entropy */
 174     SetLastError(0xDEADBEEF);
 175     okay = pCryptUnprotectData(&cipher_entropy,&data_desc,&entropy,NULL,NULL,0,&plain);
 176     ok(okay,"Decrypting with entropy\n");
 177
 178     ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
 179     ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
 180     ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
 181     ok(data_desc!=NULL,"Description not allocated\n");
 182     ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
 183
 184     LocalFree(plain.pbData);
 185     LocalFree(data_desc);
 186
 187     plain.pbData=NULL;
 188     plain.cbData=0;
 189     data_desc=NULL;
 190
 191     /* with entropy but no description */
 192     SetLastError(0xDEADBEEF);
 193     okay = pCryptUnprotectData(&cipher_no_desc,&data_desc,&entropy,NULL,NULL,0,&plain);
 194     ok(okay,"Decrypting with entropy and no description\n");
 195
 196     ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
 197     ok(plain.cbData==strlen(secret2)+1,"Plain DATA_BLOB wrong length\n");
 198     ok(!strcmp((const char*)plain.pbData,secret2),"Plain does not match secret\n");
 199     ok(data_desc!=NULL,"Description not allocated\n");
 200     ok(data_desc[0]=='\0',"Description not empty\n");
 201
 202     LocalFree(data_desc);
 203     LocalFree(plain.pbData);
 204
 205     plain.pbData=NULL;
 206     plain.cbData=0;
 207 }
 208
 209 START_TEST(protectdata)
 210 {
 211     HMODULE hCrypt32 = GetModuleHandleA("crypt32.dll");
 212     hCrypt32 = GetModuleHandleA("crypt32.dll");
 213     pCryptProtectData = (void*)GetProcAddress(hCrypt32, "CryptProtectData");
 214     pCryptUnprotectData = (void*)GetProcAddress(hCrypt32, "CryptUnprotectData");
 215     if (!pCryptProtectData || !pCryptUnprotectData)
 216     {
 217         skip("Crypt(Un)ProtectData() is not available\n");
 218         return;
 219     }
 220
 221     protected=FALSE;
 222     test_cryptprotectdata();
 223     test_cryptunprotectdata();
 224
 225     /* deinit globals here */
 226     if (cipher.pbData) LocalFree(cipher.pbData);
 227     if (cipher_entropy.pbData) LocalFree(cipher_entropy.pbData);
 228     if (cipher_no_desc.pbData) LocalFree(cipher_no_desc.pbData);
 229 }