From 255541a16e1aac46082584d9c217376af95ffdfa Mon Sep 17 00:00:00 2001 From: Kees Cook <kees@outflux.net> Date: Tue, 24 May 2005 09:58:26 +0000 Subject: [PATCH] Create basic test suite for CryptProtectData/CryptUnprotectData. --- configure | 3 +- configure.ac | 1 + dlls/crypt32/Makefile.in | 2 + dlls/crypt32/tests/.cvsignore | 3 + dlls/crypt32/tests/Makefile.in | 13 ++ dlls/crypt32/tests/protectdata.c | 215 +++++++++++++++++++++++++++++++ 6 files changed, 236 insertions(+), 1 deletion(-) create mode 100644 dlls/crypt32/tests/.cvsignore create mode 100644 dlls/crypt32/tests/Makefile.in create mode 100644 dlls/crypt32/tests/protectdata.c diff --git a/configure b/configure index dec2e634703..3c254faae5b 100755 --- a/configure +++ b/configure @@ -19972,7 +19972,7 @@ MAKE_LIB_RULES=libs/Makelib.rules MAKE_PROG_RULES=programs/Makeprog.rules - ac_config_files="$ac_config_files Make.rules dlls/Makedll.rules dlls/Maketest.rules libs/Makelib.rules programs/Makeprog.rules Makefile dlls/Makefile dlls/advapi32/Makefile dlls/advapi32/tests/Makefile dlls/advpack/Makefile dlls/advpack/tests/Makefile dlls/amstream/Makefile dlls/atl/Makefile dlls/avicap32/Makefile dlls/avifil32/Makefile dlls/cabinet/Makefile dlls/capi2032/Makefile dlls/cards/Makefile dlls/cfgmgr32/Makefile dlls/comcat/Makefile dlls/comctl32/Makefile dlls/comctl32/tests/Makefile dlls/commdlg/Makefile dlls/crtdll/Makefile dlls/crypt32/Makefile dlls/cryptdll/Makefile dlls/ctl3d/Makefile dlls/d3d8/Makefile dlls/d3d9/Makefile dlls/d3dim/Makefile dlls/d3drm/Makefile dlls/d3dx8/Makefile dlls/d3dxof/Makefile dlls/dbghelp/Makefile dlls/dciman32/Makefile dlls/ddraw/Makefile dlls/ddraw/tests/Makefile dlls/devenum/Makefile dlls/dinput/Makefile dlls/dinput/tests/Makefile dlls/dinput8/Makefile dlls/dmband/Makefile dlls/dmcompos/Makefile dlls/dmime/Makefile dlls/dmloader/Makefile dlls/dmscript/Makefile dlls/dmstyle/Makefile dlls/dmsynth/Makefile dlls/dmusic/Makefile dlls/dmusic32/Makefile dlls/dplay/Makefile dlls/dplayx/Makefile dlls/dpnet/Makefile dlls/dpnhpast/Makefile dlls/dsound/Makefile dlls/dsound/tests/Makefile dlls/dswave/Makefile dlls/dxdiagn/Makefile dlls/dxerr8/Makefile dlls/dxerr9/Makefile dlls/dxguid/Makefile dlls/gdi/Makefile dlls/gdi/tests/Makefile dlls/glu32/Makefile dlls/glut32/Makefile dlls/hhctrl.ocx/Makefile dlls/iccvid/Makefile dlls/icmp/Makefile dlls/ifsmgr.vxd/Makefile dlls/imagehlp/Makefile dlls/imm32/Makefile dlls/iphlpapi/Makefile dlls/iphlpapi/tests/Makefile dlls/itss/Makefile dlls/kernel/Makefile dlls/kernel/tests/Makefile dlls/lzexpand/Makefile dlls/lzexpand/tests/Makefile dlls/mapi32/Makefile dlls/mapi32/tests/Makefile dlls/mlang/Makefile dlls/mlang/tests/Makefile dlls/mmdevldr.vxd/Makefile dlls/monodebg.vxd/Makefile dlls/mpr/Makefile dlls/msacm/Makefile dlls/msacm/imaadp32/Makefile dlls/msacm/msadp32/Makefile dlls/msacm/msg711/Makefile dlls/msacm/winemp3/Makefile dlls/msacm/tests/Makefile dlls/mscms/Makefile dlls/mscms/tests/Makefile dlls/msdmo/Makefile dlls/mshtml/Makefile dlls/msi/Makefile dlls/msi/tests/Makefile dlls/msimg32/Makefile dlls/msisys/Makefile dlls/msnet32/Makefile dlls/msrle32/Makefile dlls/msvcrt/Makefile dlls/msvcrt/tests/Makefile dlls/msvcrt20/Makefile dlls/msvcrt40/Makefile dlls/msvcrtd/Makefile dlls/msvcrtd/tests/Makefile dlls/msvidc32/Makefile dlls/msvideo/Makefile dlls/mswsock/Makefile dlls/netapi32/Makefile dlls/netapi32/tests/Makefile dlls/newdev/Makefile dlls/ntdll/Makefile dlls/ntdll/tests/Makefile dlls/odbc32/Makefile dlls/odbccp32/Makefile dlls/ole32/Makefile dlls/ole32/tests/Makefile dlls/oleacc/Makefile dlls/oleaut32/Makefile dlls/oleaut32/tests/Makefile dlls/olecli/Makefile dlls/oledlg/Makefile dlls/olepro32/Makefile dlls/olesvr/Makefile dlls/opengl32/Makefile dlls/powrprof/Makefile dlls/psapi/Makefile dlls/psapi/tests/Makefile dlls/qcap/Makefile dlls/quartz/Makefile dlls/quartz/tests/Makefile dlls/rasapi32/Makefile dlls/riched20/Makefile dlls/richedit/Makefile dlls/rpcrt4/Makefile dlls/rpcrt4/tests/Makefile dlls/rsabase/Makefile dlls/rsabase/tests/Makefile dlls/rsaenh/Makefile dlls/rsaenh/tests/Makefile dlls/secur32/Makefile dlls/sensapi/Makefile dlls/serialui/Makefile dlls/setupapi/Makefile dlls/shdocvw/Makefile dlls/shell32/Makefile dlls/shell32/tests/Makefile dlls/shfolder/Makefile dlls/shlwapi/Makefile dlls/shlwapi/tests/Makefile dlls/snmpapi/Makefile dlls/stdole32.tlb/Makefile dlls/sti/Makefile dlls/strmiids/Makefile dlls/tapi32/Makefile dlls/ttydrv/Makefile dlls/twain/Makefile dlls/unicows/Makefile dlls/url/Makefile dlls/urlmon/Makefile dlls/urlmon/tests/Makefile dlls/user/Makefile dlls/user/tests/Makefile dlls/uuid/Makefile dlls/uxtheme/Makefile dlls/vdhcp.vxd/Makefile dlls/vdmdbg/Makefile dlls/version/Makefile dlls/version/tests/Makefile dlls/vmm.vxd/Makefile dlls/vnbt.vxd/Makefile dlls/vnetbios.vxd/Makefile dlls/vtdapi.vxd/Makefile dlls/vwin32.vxd/Makefile dlls/win32s/Makefile dlls/winaspi/Makefile dlls/wined3d/Makefile dlls/winedos/Makefile dlls/wineps/Makefile dlls/wininet/Makefile dlls/wininet/tests/Makefile dlls/winmm/Makefile dlls/winmm/joystick/Makefile dlls/winmm/mcianim/Makefile dlls/winmm/mciavi/Makefile dlls/winmm/mcicda/Makefile dlls/winmm/mciseq/Makefile dlls/winmm/mciwave/Makefile dlls/winmm/midimap/Makefile dlls/winmm/tests/Makefile dlls/winmm/wavemap/Makefile dlls/winmm/winealsa/Makefile dlls/winmm/winearts/Makefile dlls/winmm/wineaudioio/Makefile dlls/winmm/winejack/Makefile dlls/winmm/winenas/Makefile dlls/winmm/wineoss/Makefile dlls/winnls/Makefile dlls/winsock/Makefile dlls/winsock/tests/Makefile dlls/winspool/Makefile dlls/winspool/tests/Makefile dlls/wintab32/Makefile dlls/wintrust/Makefile dlls/wow32/Makefile dlls/wsock32/Makefile dlls/wtsapi32/Makefile dlls/x11drv/Makefile documentation/Makefile fonts/Makefile include/Makefile libs/Makefile libs/port/Makefile libs/unicode/Makefile libs/wine/Makefile libs/wpp/Makefile loader/Makefile programs/Makefile programs/avitools/Makefile programs/clock/Makefile programs/cmdlgtst/Makefile programs/control/Makefile programs/expand/Makefile programs/msiexec/Makefile programs/notepad/Makefile programs/progman/Makefile programs/regedit/Makefile programs/regsvr32/Makefile programs/rpcss/Makefile programs/rundll32/Makefile programs/start/Makefile programs/taskmgr/Makefile programs/uninstaller/Makefile programs/view/Makefile programs/wcmd/Makefile programs/wineboot/Makefile programs/winebrowser/Makefile programs/winecfg/Makefile programs/wineconsole/Makefile programs/winedbg/Makefile programs/winefile/Makefile programs/winemenubuilder/Makefile programs/winemine/Makefile programs/winepath/Makefile programs/winetest/Makefile programs/winevdm/Makefile programs/winhelp/Makefile programs/winver/Makefile server/Makefile tools/Makefile tools/widl/Makefile tools/winapi/Makefile tools/winebuild/Makefile tools/winedump/Makefile tools/winegcc/Makefile tools/wmc/Makefile tools/wrc/Makefile" + ac_config_files="$ac_config_files Make.rules dlls/Makedll.rules dlls/Maketest.rules libs/Makelib.rules programs/Makeprog.rules Makefile dlls/Makefile dlls/advapi32/Makefile dlls/advapi32/tests/Makefile dlls/advpack/Makefile dlls/advpack/tests/Makefile dlls/amstream/Makefile dlls/atl/Makefile dlls/avicap32/Makefile dlls/avifil32/Makefile dlls/cabinet/Makefile dlls/capi2032/Makefile dlls/cards/Makefile dlls/cfgmgr32/Makefile dlls/comcat/Makefile dlls/comctl32/Makefile dlls/comctl32/tests/Makefile dlls/commdlg/Makefile dlls/crtdll/Makefile dlls/crypt32/Makefile dlls/crypt32/tests/Makefile dlls/cryptdll/Makefile dlls/ctl3d/Makefile dlls/d3d8/Makefile dlls/d3d9/Makefile dlls/d3dim/Makefile dlls/d3drm/Makefile dlls/d3dx8/Makefile dlls/d3dxof/Makefile dlls/dbghelp/Makefile dlls/dciman32/Makefile dlls/ddraw/Makefile dlls/ddraw/tests/Makefile dlls/devenum/Makefile dlls/dinput/Makefile dlls/dinput/tests/Makefile dlls/dinput8/Makefile dlls/dmband/Makefile dlls/dmcompos/Makefile dlls/dmime/Makefile dlls/dmloader/Makefile dlls/dmscript/Makefile dlls/dmstyle/Makefile dlls/dmsynth/Makefile dlls/dmusic/Makefile dlls/dmusic32/Makefile dlls/dplay/Makefile dlls/dplayx/Makefile dlls/dpnet/Makefile dlls/dpnhpast/Makefile dlls/dsound/Makefile dlls/dsound/tests/Makefile dlls/dswave/Makefile dlls/dxdiagn/Makefile dlls/dxerr8/Makefile dlls/dxerr9/Makefile dlls/dxguid/Makefile dlls/gdi/Makefile dlls/gdi/tests/Makefile dlls/glu32/Makefile dlls/glut32/Makefile dlls/hhctrl.ocx/Makefile dlls/iccvid/Makefile dlls/icmp/Makefile dlls/ifsmgr.vxd/Makefile dlls/imagehlp/Makefile dlls/imm32/Makefile dlls/iphlpapi/Makefile dlls/iphlpapi/tests/Makefile dlls/itss/Makefile dlls/kernel/Makefile dlls/kernel/tests/Makefile dlls/lzexpand/Makefile dlls/lzexpand/tests/Makefile dlls/mapi32/Makefile dlls/mapi32/tests/Makefile dlls/mlang/Makefile dlls/mlang/tests/Makefile dlls/mmdevldr.vxd/Makefile dlls/monodebg.vxd/Makefile dlls/mpr/Makefile dlls/msacm/Makefile dlls/msacm/imaadp32/Makefile dlls/msacm/msadp32/Makefile dlls/msacm/msg711/Makefile dlls/msacm/winemp3/Makefile dlls/msacm/tests/Makefile dlls/mscms/Makefile dlls/mscms/tests/Makefile dlls/msdmo/Makefile dlls/mshtml/Makefile dlls/msi/Makefile dlls/msi/tests/Makefile dlls/msimg32/Makefile dlls/msisys/Makefile dlls/msnet32/Makefile dlls/msrle32/Makefile dlls/msvcrt/Makefile dlls/msvcrt/tests/Makefile dlls/msvcrt20/Makefile dlls/msvcrt40/Makefile dlls/msvcrtd/Makefile dlls/msvcrtd/tests/Makefile dlls/msvidc32/Makefile dlls/msvideo/Makefile dlls/mswsock/Makefile dlls/netapi32/Makefile dlls/netapi32/tests/Makefile dlls/newdev/Makefile dlls/ntdll/Makefile dlls/ntdll/tests/Makefile dlls/odbc32/Makefile dlls/odbccp32/Makefile dlls/ole32/Makefile dlls/ole32/tests/Makefile dlls/oleacc/Makefile dlls/oleaut32/Makefile dlls/oleaut32/tests/Makefile dlls/olecli/Makefile dlls/oledlg/Makefile dlls/olepro32/Makefile dlls/olesvr/Makefile dlls/opengl32/Makefile dlls/powrprof/Makefile dlls/psapi/Makefile dlls/psapi/tests/Makefile dlls/qcap/Makefile dlls/quartz/Makefile dlls/quartz/tests/Makefile dlls/rasapi32/Makefile dlls/riched20/Makefile dlls/richedit/Makefile dlls/rpcrt4/Makefile dlls/rpcrt4/tests/Makefile dlls/rsabase/Makefile dlls/rsabase/tests/Makefile dlls/rsaenh/Makefile dlls/rsaenh/tests/Makefile dlls/secur32/Makefile dlls/sensapi/Makefile dlls/serialui/Makefile dlls/setupapi/Makefile dlls/shdocvw/Makefile dlls/shell32/Makefile dlls/shell32/tests/Makefile dlls/shfolder/Makefile dlls/shlwapi/Makefile dlls/shlwapi/tests/Makefile dlls/snmpapi/Makefile dlls/stdole32.tlb/Makefile dlls/sti/Makefile dlls/strmiids/Makefile dlls/tapi32/Makefile dlls/ttydrv/Makefile dlls/twain/Makefile dlls/unicows/Makefile dlls/url/Makefile dlls/urlmon/Makefile dlls/urlmon/tests/Makefile dlls/user/Makefile dlls/user/tests/Makefile dlls/uuid/Makefile dlls/uxtheme/Makefile dlls/vdhcp.vxd/Makefile dlls/vdmdbg/Makefile dlls/version/Makefile dlls/version/tests/Makefile dlls/vmm.vxd/Makefile dlls/vnbt.vxd/Makefile dlls/vnetbios.vxd/Makefile dlls/vtdapi.vxd/Makefile dlls/vwin32.vxd/Makefile dlls/win32s/Makefile dlls/winaspi/Makefile dlls/wined3d/Makefile dlls/winedos/Makefile dlls/wineps/Makefile dlls/wininet/Makefile dlls/wininet/tests/Makefile dlls/winmm/Makefile dlls/winmm/joystick/Makefile dlls/winmm/mcianim/Makefile dlls/winmm/mciavi/Makefile dlls/winmm/mcicda/Makefile dlls/winmm/mciseq/Makefile dlls/winmm/mciwave/Makefile dlls/winmm/midimap/Makefile dlls/winmm/tests/Makefile dlls/winmm/wavemap/Makefile dlls/winmm/winealsa/Makefile dlls/winmm/winearts/Makefile dlls/winmm/wineaudioio/Makefile dlls/winmm/winejack/Makefile dlls/winmm/winenas/Makefile dlls/winmm/wineoss/Makefile dlls/winnls/Makefile dlls/winsock/Makefile dlls/winsock/tests/Makefile dlls/winspool/Makefile dlls/winspool/tests/Makefile dlls/wintab32/Makefile dlls/wintrust/Makefile dlls/wow32/Makefile dlls/wsock32/Makefile dlls/wtsapi32/Makefile dlls/x11drv/Makefile documentation/Makefile fonts/Makefile include/Makefile libs/Makefile libs/port/Makefile libs/unicode/Makefile libs/wine/Makefile libs/wpp/Makefile loader/Makefile programs/Makefile programs/avitools/Makefile programs/clock/Makefile programs/cmdlgtst/Makefile programs/control/Makefile programs/expand/Makefile programs/msiexec/Makefile programs/notepad/Makefile programs/progman/Makefile programs/regedit/Makefile programs/regsvr32/Makefile programs/rpcss/Makefile programs/rundll32/Makefile programs/start/Makefile programs/taskmgr/Makefile programs/uninstaller/Makefile programs/view/Makefile programs/wcmd/Makefile programs/wineboot/Makefile programs/winebrowser/Makefile programs/winecfg/Makefile programs/wineconsole/Makefile programs/winedbg/Makefile programs/winefile/Makefile programs/winemenubuilder/Makefile programs/winemine/Makefile programs/winepath/Makefile programs/winetest/Makefile programs/winevdm/Makefile programs/winhelp/Makefile programs/winver/Makefile server/Makefile tools/Makefile tools/widl/Makefile tools/winapi/Makefile tools/winebuild/Makefile tools/winedump/Makefile tools/winegcc/Makefile tools/wmc/Makefile tools/wrc/Makefile" cat >confcache <<\_ACEOF @@ -20528,6 +20528,7 @@ do "dlls/commdlg/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/commdlg/Makefile" ;; "dlls/crtdll/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/crtdll/Makefile" ;; "dlls/crypt32/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/crypt32/Makefile" ;; + "dlls/crypt32/tests/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/crypt32/tests/Makefile" ;; "dlls/cryptdll/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/cryptdll/Makefile" ;; "dlls/ctl3d/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/ctl3d/Makefile" ;; "dlls/d3d8/Makefile" ) CONFIG_FILES="$CONFIG_FILES dlls/d3d8/Makefile" ;; diff --git a/configure.ac b/configure.ac index 383e0fde304..d09fdab4fb3 100644 --- a/configure.ac +++ b/configure.ac @@ -1548,6 +1548,7 @@ dlls/comctl32/tests/Makefile dlls/commdlg/Makefile dlls/crtdll/Makefile dlls/crypt32/Makefile +dlls/crypt32/tests/Makefile dlls/cryptdll/Makefile dlls/ctl3d/Makefile dlls/d3d8/Makefile diff --git a/dlls/crypt32/Makefile.in b/dlls/crypt32/Makefile.in index cce33396574..f1636127dee 100644 --- a/dlls/crypt32/Makefile.in +++ b/dlls/crypt32/Makefile.in @@ -13,6 +13,8 @@ C_SRCS = \ protectdata.c \ main.c +SUBDIRS = tests + @MAKE_DLL_RULES@ ### Dependencies: diff --git a/dlls/crypt32/tests/.cvsignore b/dlls/crypt32/tests/.cvsignore new file mode 100644 index 00000000000..5767c9e7997 --- /dev/null +++ b/dlls/crypt32/tests/.cvsignore @@ -0,0 +1,3 @@ +Makefile +protectdata.ok +testlist.c diff --git a/dlls/crypt32/tests/Makefile.in b/dlls/crypt32/tests/Makefile.in new file mode 100644 index 00000000000..6fee8b80248 --- /dev/null +++ b/dlls/crypt32/tests/Makefile.in @@ -0,0 +1,13 @@ +TOPSRCDIR = @top_srcdir@ +TOPOBJDIR = ../../.. +SRCDIR = @srcdir@ +VPATH = @srcdir@ +TESTDLL = crypt32.dll +IMPORTS = crypt32 + +CTESTS = \ + protectdata.c + +@MAKE_TEST_RULES@ + +### Dependencies: diff --git a/dlls/crypt32/tests/protectdata.c b/dlls/crypt32/tests/protectdata.c new file mode 100644 index 00000000000..3cc526ef558 --- /dev/null +++ b/dlls/crypt32/tests/protectdata.c @@ -0,0 +1,215 @@ +/* + * Unit test suite for crypt32.dll's CryptProtectData/CryptUnprotectData + * + * Copyright 2005 Kees Cook <kees@outflux.net> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include <stdio.h> +#include <stdarg.h> +#include <windef.h> +#include <winbase.h> +#include <winerror.h> +#include <wincrypt.h> + +#include "wine/test.h" + +static const char * secret = "I am a super secret string that no one can see!"; +static const char * secret2 = "I am a super secret string indescribable string"; +static const 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; /* if true, the unprotect tests can run */ +static DATA_BLOB cipher; +static DATA_BLOB cipher_entropy; +static DATA_BLOB cipher_no_desc; + +static void test_cryptprotectdata(void) +{ + 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,desc,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,desc,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,desc,NULL,NULL,NULL,0,&cipher); + ok(protected, "Encrypting without entropy.\n"); + r = GetLastError(); + ok(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r); + + cipher_entropy.pbData=NULL; + cipher_entropy.cbData=0; + + /* with entropy */ + SetLastError(0xDEADBEEF); + protected = CryptProtectData(&plain,desc,&entropy,NULL,NULL,0,&cipher_entropy); + ok(protected, "Encrypting with entropy.\n"); + r = GetLastError(); + ok(r == ERROR_SUCCESS, "Wrong (%lu) GetLastError seen\n",r); + + 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 and no description.\n"); + r = GetLastError(); + ok(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(); + 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"); + r = GetLastError(); + ok(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(); + 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"); + r = GetLastError(); + ok(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(); + ok(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); + if (cipher_entropy.pbData) LocalFree(cipher_entropy.pbData); + if (cipher_no_desc.pbData) LocalFree(cipher_no_desc.pbData); +} -- GitLab