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 void	test_cryptprotectdata (void)
static void	test_cryptunprotectdata (void)
static void	test_simpleroundtrip (const char *plaintext)
	START_TEST (protectdata)

Variables
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 BOOL	protected = FALSE
static DATA_BLOB	cipher
static DATA_BLOB	cipher_entropy
static DATA_BLOB	cipher_no_desc

Function Documentation

START_TEST()

START_TEST

(

protectdata

)

Definition at line 216 of file protectdata.c.

{
    protected = FALSE;
    test_cryptprotectdata();
    test_cryptunprotectdata();
    test_simpleroundtrip("");
    test_simpleroundtrip("hello");
 
    /* deinit globals here */
    if (cipher.pbData) LocalFree(cipher.pbData);
    if (cipher_entropy.pbData) LocalFree(cipher_entropy.pbData);
    if (cipher_no_desc.pbData) LocalFree(cipher_no_desc.pbData);
}

test_cryptprotectdata()

static void test_cryptprotectdata ( void  )

static

Definition at line 38 of file protectdata.c.

{
    LONG r;
    DATA_BLOB plain;
    DATA_BLOB entropy;
 
    plain.pbData=(void*)secret;
    plain.cbData=strlen(secret)+1;
 
    entropy.pbData=(void*)key;
    entropy.cbData=strlen(key)+1;
 
    SetLastError(0xDEADBEEF);
    protected = CryptProtectData(NULL, L"Ultra secret test message", NULL, NULL, NULL, 0, &cipher);
    ok(!protected, "Encrypting without plain data source.\n");
    r = GetLastError();
    ok(r == ERROR_INVALID_PARAMETER, "Wrong (%lu) GetLastError seen\n",r);
 
    SetLastError(0xDEADBEEF);
    protected = CryptProtectData(&plain, L"Ultra secret test message", NULL, NULL, NULL, 0, NULL);
    ok(!protected, "Encrypting without cipher destination.\n");
    r = GetLastError();
    ok(r == ERROR_INVALID_PARAMETER, "Wrong (%lu) GetLastError seen\n",r);
 
    cipher.pbData=NULL;
    cipher.cbData=0;
 
    /* without entropy */
    SetLastError(0xDEADBEEF);
    protected = CryptProtectData(&plain, L"Ultra secret test message", NULL, NULL, NULL, 0, &cipher);
    ok(protected, "Encrypting without entropy.\n");
    if (protected)
    {
        r = GetLastError();
        ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %ld\n",r);
    }
 
    cipher_entropy.pbData=NULL;
    cipher_entropy.cbData=0;
 
    /* with entropy */
    SetLastError(0xDEADBEEF);
    protected = CryptProtectData(&plain, L"Ultra secret test message", &entropy, NULL, NULL, 0, &cipher_entropy);
    ok(protected, "Encrypting with entropy.\n");
 
    cipher_no_desc.pbData=NULL;
    cipher_no_desc.cbData=0;
 
    /* with entropy but no description */
    plain.pbData=(void*)secret2;
    plain.cbData=strlen(secret2)+1;
    SetLastError(0xDEADBEEF);
    protected = CryptProtectData(&plain,NULL,&entropy,NULL,NULL,0,&cipher_no_desc);
    ok(protected, "Encrypting with entropy.\n");
}

Referenced by START_TEST().

test_cryptunprotectdata()

static void test_cryptunprotectdata ( void  )

static

Definition at line 94 of file protectdata.c.

{
    LONG r;
    DATA_BLOB plain;
    DATA_BLOB entropy;
    BOOL okay;
    WCHAR * data_desc;
 
    entropy.pbData=(void*)key;
    entropy.cbData=strlen(key)+1;
 
    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();
    ok(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();
    ok(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();
    ok(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");
 
    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
    ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
    ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
    ok(data_desc!=NULL,"Description not allocated\n");
    ok(!lstrcmpW(data_desc, L"Ultra secret test message"),"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();
    ok(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");
 
    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
    ok(plain.cbData==strlen(secret)+1,"Plain DATA_BLOB wrong length\n");
    ok(!strcmp((const char*)plain.pbData,secret),"Plain does not match secret\n");
    ok(data_desc!=NULL,"Description not allocated\n");
    ok(!lstrcmpW(data_desc, L"Ultra secret test message"),"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");
 
    ok(plain.pbData!=NULL,"Plain DATA_BLOB missing data\n");
    ok(plain.cbData==strlen(secret2)+1,"Plain DATA_BLOB wrong length\n");
    ok(!strcmp((const char*)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(data_desc);
    LocalFree(plain.pbData);
 
    plain.pbData=NULL;
    plain.cbData=0;
}

Referenced by START_TEST().

test_simpleroundtrip()

static void test_simpleroundtrip ( const char *  plaintext )

static

Definition at line 194 of file protectdata.c.

{
    DATA_BLOB input;
    DATA_BLOB encrypted;
    DATA_BLOB output;
    int res;
    WCHAR emptyW[1];
 
    emptyW[0] = 0;
    input.pbData = (unsigned char *)plaintext;
    input.cbData = strlen(plaintext);
    res = CryptProtectData(&input, emptyW, NULL, NULL, NULL, 0, &encrypted);
    ok(res != 0, "can't protect\n");
 
    res = CryptUnprotectData(&encrypted, NULL, NULL, NULL, NULL, 0, &output);
    ok(res != 0, "can't unprotect; last error %lu\n", GetLastError());
    ok(output.cbData == strlen(plaintext), "output wrong length %ld for input '%s', wanted %d\n", output.cbData, plaintext, lstrlenA(plaintext));
    ok(!memcmp(plaintext, (char *)output.pbData, output.cbData), "output wrong contents for input '%s'\n", plaintext);
    LocalFree(output.pbData);
    LocalFree(encrypted.pbData);
}

Referenced by START_TEST().

Variable Documentation

cipher

DATA_BLOB cipher

static

Definition at line 34 of file protectdata.c.

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

cipher_entropy

DATA_BLOB cipher_entropy

static

Definition at line 35 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 36 of file protectdata.c.

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

key

char key[] = "Wibble wibble wibble"

static

Definition at line 32 of file protectdata.c.

protected

BOOL protected = FALSE

static

Definition at line 33 of file protectdata.c.

secret

char secret[] = "I am a super secret string that no one can see!"

static

Definition at line 30 of file protectdata.c.

secret2

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

static

Definition at line 31 of file protectdata.c.

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