ReactOS  0.4.14-dev-41-g31d7680
Functions | Variables
protectdata.c File Reference
#include <stdio.h>
#include <stdarg.h>
#include <windef.h>
#include <winbase.h>
#include <winerror.h>
#include <wincrypt.h>
#include "wine/test.h"
Include dependency graph for protectdata.c:

Go to the source code of this file.

Functions

static BOOL (WINAPI *pCryptProtectData)(DATA_BLOB *
 
static void test_cryptprotectdata (void)
 
static void test_cryptunprotectdata (void)
 
static void test_simpleroundtrip (const char *plaintext)
 
 START_TEST (protectdata)
 

Variables

static LPCWSTR
 
static DATA_BLOB PVOID
 
static DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DWORD
 
static DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static LPWSTR DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static char secret [] = "I am a super secret string that no one can see!"
 
static char secret2 [] = "I am a super secret string indescribable string"
 
static char key [] = "Wibble wibble wibble"
 
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}
 
static BOOL protected = FALSE
 
static DATA_BLOB cipher
 
static DATA_BLOB cipher_entropy
 
static DATA_BLOB cipher_no_desc
 

Function Documentation

◆ BOOL()

static BOOL ( WINAPI *  pCryptProtectData)
static

◆ START_TEST()

START_TEST ( protectdata  )

Definition at line 244 of file protectdata.c.

245 {
246  HMODULE hCrypt32 = GetModuleHandleA("crypt32.dll");
247  pCryptProtectData = (void*)GetProcAddress(hCrypt32, "CryptProtectData");
248  pCryptUnprotectData = (void*)GetProcAddress(hCrypt32, "CryptUnprotectData");
249  if (!pCryptProtectData || !pCryptUnprotectData)
250  {
251  win_skip("Crypt(Un)ProtectData() is not available\n");
252  return;
253  }
254 
255  protected=FALSE;
256  test_cryptprotectdata();
257  test_cryptunprotectdata();
258  test_simpleroundtrip("");
259  test_simpleroundtrip("hello");
260 
261  /* deinit globals here */
262  if (cipher.pbData) LocalFree(cipher.pbData);
263  if (cipher_entropy.pbData) LocalFree(cipher_entropy.pbData);
264  if (cipher_no_desc.pbData) LocalFree(cipher_no_desc.pbData);
265 }
cipher
static DATA_BLOB cipher
Definition: protectdata.c:38
cipher_entropy
static DATA_BLOB cipher_entropy
Definition: protectdata.c:39
cipher_no_desc
static DATA_BLOB cipher_no_desc
Definition: protectdata.c:40
test_simpleroundtrip
static void test_simpleroundtrip(const char *plaintext)
Definition: protectdata.c:216
_CRYPTOAPI_BLOB::pbData
BYTE * pbData
Definition: wincrypt.h:103
GetModuleHandleA
HMODULE WINAPI DECLSPEC_HOTPATCH GetModuleHandleA(LPCSTR lpModuleName)
Definition: loader.c:821
LocalFree
HLOCAL NTAPI LocalFree(HLOCAL hMem)
Definition: heapmem.c:1577
test_cryptunprotectdata
static void test_cryptunprotectdata(void)
Definition: protectdata.c:109
GetProcAddress
#define GetProcAddress(x, y)
Definition: compat.h:410
win_skip
#define win_skip
Definition: test.h:141
test_cryptprotectdata
static void test_cryptprotectdata(void)
Definition: protectdata.c:42

◆ test_cryptprotectdata()

static void test_cryptprotectdata ( void  )
static

Definition at line 42 of file protectdata.c.

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 ||
73  broken(!protected), /* Win9x/NT4 */
74  "Encrypting without entropy.\n");
75  if (protected)
76  {
77  r = GetLastError();
78  ok(r == ERROR_SUCCESS ||
79  r == ERROR_IO_PENDING, /* win2k */
80  "Expected ERROR_SUCCESS or ERROR_IO_PENDING, got %d\n",r);
81  }
82 
83  cipher_entropy.pbData=NULL;
84  cipher_entropy.cbData=0;
85 
86  /* with entropy */
87  SetLastError(0xDEADBEEF);
88  protected = pCryptProtectData(&plain,desc,&entropy,NULL,NULL,0,&cipher_entropy);
89  ok(protected ||
90  broken(!protected), /* Win9x/NT4 */
91  "Encrypting with entropy.\n");
92 
93  cipher_no_desc.pbData=NULL;
94  cipher_no_desc.cbData=0;
95 
96  /* with entropy but no description */
97  plain.pbData=(void*)secret2;
98  plain.cbData=strlen(secret2)+1;
99  SetLastError(0xDEADBEEF);
100  protected = pCryptProtectData(&plain,NULL,&entropy,NULL,NULL,0,&cipher_no_desc);
101  if (!protected)
102  {
103  /* fails in win2k */
104  ok(GetLastError() == ERROR_INVALID_PARAMETER,
105  "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
106  }
107 }
ERROR_INVALID_PARAMETER
#define ERROR_INVALID_PARAMETER
Definition: compat.h:91
ERROR_SUCCESS
#define ERROR_SUCCESS
Definition: deptool.c:10
strlen
ACPI_SIZE strlen(const char *String)
Definition: utclib.c:269
r
GLdouble GLdouble GLdouble r
Definition: gl.h:2055
secret2
static char secret2[]
Definition: protectdata.c:34
secret
static DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static LPWSTR DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static char secret[]
Definition: protectdata.c:33
GetLastError
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1059
cipher
static DATA_BLOB cipher
Definition: protectdata.c:38
ERROR_IO_PENDING
#define ERROR_IO_PENDING
Definition: dderror.h:15
cipher_entropy
static DATA_BLOB cipher_entropy
Definition: protectdata.c:39
LONG
long LONG
Definition: pedump.c:60
desc
static const WCHAR desc[]
Definition: protectdata.c:36
NULL
smooth NULL
Definition: ftsmooth.c:416
cipher_no_desc
static DATA_BLOB cipher_no_desc
Definition: protectdata.c:40
_CRYPTOAPI_BLOB::cbData
DWORD cbData
Definition: wincrypt.h:102
_CRYPTOAPI_BLOB
Definition: wincrypt.h:101
SetLastError
#define SetLastError(x)
Definition: compat.h:409
_CRYPTOAPI_BLOB::pbData
BYTE * pbData
Definition: wincrypt.h:103
broken
#define broken(x)
Definition: _sntprintf.h:21
ok
#define ok(value,...)
Definition: atltest.h:57
key
Definition: path.c:42

Referenced by START_TEST().

◆ test_cryptunprotectdata()

static void test_cryptunprotectdata ( void  )
static

Definition at line 109 of file protectdata.c.

110 {
111  LONG r;
112  DATA_BLOB plain;
113  DATA_BLOB entropy;
114  BOOL okay;
115  WCHAR * data_desc;
116 
117  entropy.pbData=(void*)key;
118  entropy.cbData=strlen(key)+1;
119 
120  /* fails in win2k */
121  if (!protected)
122  {
123  skip("CryptProtectData failed to run\n");
124  return;
125  }
126 
127  plain.pbData=NULL;
128  plain.cbData=0;
129 
130  SetLastError(0xDEADBEEF);
131  okay = pCryptUnprotectData(&cipher,NULL,NULL,NULL,NULL,0,NULL);
132  ok(!okay,"Decrypting without destination\n");
133  r = GetLastError();
134  ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
135 
136  SetLastError(0xDEADBEEF);
137  okay = pCryptUnprotectData(NULL,NULL,NULL,NULL,NULL,0,&plain);
138  ok(!okay,"Decrypting without source\n");
139  r = GetLastError();
140  ok(r == ERROR_INVALID_PARAMETER, "Wrong (%u) GetLastError seen\n",r);
141 
142  plain.pbData=NULL;
143  plain.cbData=0;
144 
145  SetLastError(0xDEADBEEF);
146  okay = pCryptUnprotectData(&cipher_entropy,NULL,NULL,NULL,NULL,0,&plain);
147  ok(!okay,"Decrypting without needed entropy\n");
148  r = GetLastError();
149  ok(r == ERROR_INVALID_DATA, "Wrong (%u) GetLastError seen\n", r);
150 
151  plain.pbData=NULL;
152  plain.cbData=0;
153  data_desc=NULL;
154 
155  /* without entropy */
156  SetLastError(0xDEADBEEF);
157  okay = pCryptUnprotectData(&cipher,&data_desc,NULL,NULL,NULL,0,&plain);
158  ok(okay,"Decrypting without entropy\n");
159 
160  ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
161  ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
162  ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
163  ok(data_desc!=NULL,"Description not allocated\n");
164  ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
165 
166  LocalFree(plain.pbData);
167  LocalFree(data_desc);
168 
169  plain.pbData=NULL;
170  plain.cbData=0;
171  data_desc=NULL;
172 
173  /* with wrong entropy */
174  SetLastError(0xDEADBEEF);
175  okay = pCryptUnprotectData(&cipher_entropy,&data_desc,&cipher_entropy,NULL,NULL,0,&plain);
176  ok(!okay,"Decrypting with wrong entropy\n");
177  r = GetLastError();
178  ok(r == ERROR_INVALID_DATA, "Wrong (%u) GetLastError seen\n",r);
179 
180  /* with entropy */
181  SetLastError(0xDEADBEEF);
182  okay = pCryptUnprotectData(&cipher_entropy,&data_desc,&entropy,NULL,NULL,0,&plain);
183  ok(okay,"Decrypting with entropy\n");
184 
185  ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
186  ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
187  ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
188  ok(data_desc!=NULL,"Description not allocated\n");
189  ok(!lstrcmpW(data_desc,desc),"Description does not match\n");
190 
191  LocalFree(plain.pbData);
192  LocalFree(data_desc);
193 
194  plain.pbData=NULL;
195  plain.cbData=0;
196  data_desc=NULL;
197 
198  /* with entropy but no description */
199  SetLastError(0xDEADBEEF);
200  okay = pCryptUnprotectData(&cipher_no_desc,&data_desc,&entropy,NULL,NULL,0,&plain);
201  ok(okay,"Decrypting with entropy and no description\n");
202 
203  ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
204  ok(plain.cbData==strlen(secret2)+1,"Plain DATA_BLOB wrong length\n");
205  ok(!strcmp((const char*)plain.pbData,secret2),"Plain does not match secret\n");
206  ok(data_desc!=NULL,"Description not allocated\n");
207  ok(data_desc[0]=='\0',"Description not empty\n");
208 
209  LocalFree(data_desc);
210  LocalFree(plain.pbData);
211 
212  plain.pbData=NULL;
213  plain.cbData=0;
214 }
ERROR_INVALID_PARAMETER
#define ERROR_INVALID_PARAMETER
Definition: compat.h:91
strlen
ACPI_SIZE strlen(const char *String)
Definition: utclib.c:269
r
GLdouble GLdouble GLdouble r
Definition: gl.h:2055
lstrcmpW
int WINAPI lstrcmpW(LPCWSTR lpString1, LPCWSTR lpString2)
Definition: lstring.c:170
secret2
static char secret2[]
Definition: protectdata.c:34
secret
static DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static LPWSTR DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB *static char secret[]
Definition: protectdata.c:33
GetLastError
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1059
cipher
static DATA_BLOB cipher
Definition: protectdata.c:38
cipher_entropy
static DATA_BLOB cipher_entropy
Definition: protectdata.c:39
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
LONG
long LONG
Definition: pedump.c:60
desc
static const WCHAR desc[]
Definition: protectdata.c:36
NULL
smooth NULL
Definition: ftsmooth.c:416
cipher_no_desc
static DATA_BLOB cipher_no_desc
Definition: protectdata.c:40
_CRYPTOAPI_BLOB::cbData
DWORD cbData
Definition: wincrypt.h:102
_CRYPTOAPI_BLOB
Definition: wincrypt.h:101
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180
SetLastError
#define SetLastError(x)
Definition: compat.h:409
_CRYPTOAPI_BLOB::pbData
BYTE * pbData
Definition: wincrypt.h:103
ERROR_INVALID_DATA
#define ERROR_INVALID_DATA
Definition: winerror.h:116
LocalFree
HLOCAL NTAPI LocalFree(HLOCAL hMem)
Definition: heapmem.c:1577
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
strcmp
int strcmp(const char *String1, const char *String2)
Definition: utclib.c:469
key
Definition: path.c:42

Referenced by START_TEST().

◆ test_simpleroundtrip()

static void test_simpleroundtrip ( const char *  plaintext)
static

Definition at line 216 of file protectdata.c.

217 {
218  DATA_BLOB input;
219  DATA_BLOB encrypted;
220  DATA_BLOB output;
221  int res;
222  WCHAR emptyW[1];
223 
224  emptyW[0] = 0;
225  input.pbData = (unsigned char *)plaintext;
226  input.cbData = strlen(plaintext);
227  res = pCryptProtectData(&input, emptyW, NULL, NULL, NULL, 0, &encrypted);
228  ok(res != 0 || broken(!res), "can't protect\n");
229  if (!res)
230  {
231  /* Fails on Win9x, NT4 */
232  win_skip("CryptProtectData failed\n");
233  return;
234  }
235 
236  res = pCryptUnprotectData(&encrypted, NULL, NULL, NULL, NULL, 0, &output);
237  ok(res != 0, "can't unprotect; last error %u\n", GetLastError());
238  ok(output.cbData == strlen(plaintext), "output wrong length %d for input '%s', wanted %d\n", output.cbData, plaintext, lstrlenA(plaintext));
239  ok(!memcmp(plaintext, (char *)output.pbData, output.cbData), "output wrong contents for input '%s'\n", plaintext);
240  LocalFree(output.pbData);
241  LocalFree(encrypted.pbData);
242 }
output
static UCHAR ULONG UCHAR ULONG UCHAR * output
Definition: bcrypt.c:29
memcmp
int memcmp(void *Buffer1, void *Buffer2, ACPI_SIZE Count)
Definition: utclib.c:112
strlen
ACPI_SIZE strlen(const char *String)
Definition: utclib.c:269
GetLastError
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1059
NULL
smooth NULL
Definition: ftsmooth.c:416
_CRYPTOAPI_BLOB
Definition: wincrypt.h:101
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180
_CRYPTOAPI_BLOB::pbData
BYTE * pbData
Definition: wincrypt.h:103
broken
#define broken(x)
Definition: _sntprintf.h:21
lstrlenA
int WINAPI lstrlenA(LPCSTR lpString)
Definition: lstring.c:145
input
GLenum GLenum GLenum input
Definition: glext.h:9031
emptyW
static const WCHAR emptyW[]
Definition: navigate.c:36
LocalFree
HLOCAL NTAPI LocalFree(HLOCAL hMem)
Definition: heapmem.c:1577
ok
#define ok(value,...)
Definition: atltest.h:57
res
GLuint res
Definition: glext.h:9613
win_skip
#define win_skip
Definition: test.h:141

Referenced by START_TEST().

Variable Documentation

◆ cipher

DATA_BLOB cipher
static

Definition at line 38 of file protectdata.c.

Referenced by rc2_ecb_decrypt(), rc2_ecb_encrypt(), START_TEST(), test_cryptprotectdata(), and test_cryptunprotectdata().

◆ cipher_entropy

DATA_BLOB cipher_entropy
static

Definition at line 39 of file protectdata.c.

Referenced by START_TEST(), test_cryptprotectdata(), and test_cryptunprotectdata().

◆ cipher_no_desc

DATA_BLOB cipher_no_desc
static

Definition at line 40 of file protectdata.c.

Referenced by START_TEST(), test_cryptprotectdata(), and test_cryptunprotectdata().

◆ desc

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}
static

Definition at line 36 of file protectdata.c.

Referenced by test_cryptprotectdata(), and test_cryptunprotectdata().

◆ DWORD

DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB* static LPWSTR DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DWORD

Definition at line 30 of file protectdata.c.

Referenced by convert_hash_to_blob(), serialize(), serialize_dword(), and unserialize_dword().

◆ key

char key[] = "Wibble wibble wibble"
static

Definition at line 35 of file protectdata.c.

◆ LPCWSTR

LPCWSTR

Definition at line 30 of file protectdata.c.

◆ protected

BOOL protected = FALSE
static

Definition at line 37 of file protectdata.c.

◆ PVOID

DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB* static LPWSTR DATA_BLOB PVOID

Definition at line 30 of file protectdata.c.

◆ secret

DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB* static LPWSTR DATA_BLOB CRYPTPROTECT_PROMPTSTRUCT DATA_BLOB* static char secret[] = "I am a super secret string that no one can see!"
static

Definition at line 33 of file protectdata.c.

Referenced by test_cryptprotectdata(), and test_cryptunprotectdata().

◆ secret2

char secret2[] = "I am a super secret string indescribable string"
static

Definition at line 34 of file protectdata.c.

Referenced by test_cryptprotectdata(), and test_cryptunprotectdata().