222 lines
8.3 KiB
C
222 lines
8.3 KiB
C
|
/*
|
||
|
|
* Unit test suite for crypt32.dll's OID support functions.
|
||
|
|
*
|
||
|
|
* Copyright 2005 Juan Lang
|
||
|
|
*
|
||
|
|
* 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 void test_oidFunctionSet(void)
|
||
|
|
{
|
||
|
|
HCRYPTOIDFUNCSET set1, set2;
|
||
|
|
BOOL ret;
|
||
|
|
LPWSTR buf = NULL;
|
||
|
|
DWORD size;
|
||
|
|
|
||
|
|
/* This crashes
|
||
|
|
set = CryptInitOIDFunctionSet(NULL, 0);
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* The name doesn't mean much */
|
||
|
|
set1 = CryptInitOIDFunctionSet("funky", 0);
|
||
|
|
ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08lx\n", GetLastError());
|
||
|
|
if (set1)
|
||
|
|
{
|
||
|
|
/* These crash
|
||
|
|
ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, NULL);
|
||
|
|
ret = CryptGetDefaultOIDDllList(NULL, 0, NULL, &size);
|
||
|
|
*/
|
||
|
|
size = 0;
|
||
|
|
ret = CryptGetDefaultOIDDllList(set1, 0, NULL, &size);
|
||
|
|
ok(ret, "CryptGetDefaultOIDDllList failed: %08lx\n", GetLastError());
|
||
|
|
if (ret)
|
||
|
|
{
|
||
|
|
buf = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR));
|
||
|
|
if (buf)
|
||
|
|
{
|
||
|
|
ret = CryptGetDefaultOIDDllList(set1, 0, buf, &size);
|
||
|
|
ok(ret, "CryptGetDefaultOIDDllList failed: %08lx\n",
|
||
|
|
GetLastError());
|
||
|
|
ok(!*buf, "Expected empty DLL list\n");
|
||
|
|
HeapFree(GetProcessHeap(), 0, buf);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* MSDN says flags must be 0, but it's not checked */
|
||
|
|
set1 = CryptInitOIDFunctionSet("", 1);
|
||
|
|
ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08lx\n", GetLastError());
|
||
|
|
set2 = CryptInitOIDFunctionSet("", 0);
|
||
|
|
ok(set2 != 0, "CryptInitOIDFunctionSet failed: %08lx\n", GetLastError());
|
||
|
|
/* There isn't a free function, so there must be only one set per name to
|
||
|
|
* limit leaks. (I guess the sets are freed when crypt32 is unloaded.)
|
||
|
|
*/
|
||
|
|
ok(set1 == set2, "Expected identical sets\n");
|
||
|
|
if (set1)
|
||
|
|
{
|
||
|
|
/* The empty name function set used here seems to correspond to
|
||
|
|
* DEFAULT.
|
||
|
|
*/
|
||
|
|
}
|
||
|
|
|
||
|
|
/* There's no installed function for a built-in encoding. */
|
||
|
|
set1 = CryptInitOIDFunctionSet("CryptDllEncodeObject", 0);
|
||
|
|
ok(set1 != 0, "CryptInitOIDFunctionSet failed: %08lx\n", GetLastError());
|
||
|
|
if (set1)
|
||
|
|
{
|
||
|
|
void *funcAddr;
|
||
|
|
HCRYPTOIDFUNCADDR hFuncAddr;
|
||
|
|
|
||
|
|
ret = CryptGetOIDFunctionAddress(set1, X509_ASN_ENCODING, X509_CERT, 0,
|
||
|
|
&funcAddr, &hFuncAddr);
|
||
|
|
ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
|
||
|
|
"Expected ERROR_FILE_NOT_FOUND, got %08lx\n", GetLastError());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
typedef int (*funcY)(int);
|
||
|
|
|
||
|
|
static int funky(int x)
|
||
|
|
{
|
||
|
|
return x;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void test_installOIDFunctionAddress(void)
|
||
|
|
{
|
||
|
|
BOOL ret;
|
||
|
|
CRYPT_OID_FUNC_ENTRY entry = { CRYPT_DEFAULT_OID, funky };
|
||
|
|
HCRYPTOIDFUNCSET set;
|
||
|
|
|
||
|
|
/* This crashes
|
||
|
|
ret = CryptInstallOIDFunctionAddress(NULL, 0, NULL, 0, NULL, 0);
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* Installing zero functions should work */
|
||
|
|
SetLastError(0xdeadbeef);
|
||
|
|
ret = CryptInstallOIDFunctionAddress(NULL, 0, "CryptDllEncodeObject", 0,
|
||
|
|
NULL, 0);
|
||
|
|
ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08lx\n",
|
||
|
|
GetLastError());
|
||
|
|
|
||
|
|
/* The function name doesn't much matter */
|
||
|
|
SetLastError(0xdeadbeef);
|
||
|
|
ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 0, NULL, 0);
|
||
|
|
ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08lx\n",
|
||
|
|
GetLastError());
|
||
|
|
SetLastError(0xdeadbeef);
|
||
|
|
entry.pszOID = X509_CERT;
|
||
|
|
ret = CryptInstallOIDFunctionAddress(NULL, 0, "OhSoFunky", 1, &entry, 0);
|
||
|
|
ok(ret && GetLastError() == 0xdeadbeef, "Expected success, got %08lx\n",
|
||
|
|
GetLastError());
|
||
|
|
set = CryptInitOIDFunctionSet("OhSoFunky", 0);
|
||
|
|
ok(set != 0, "CryptInitOIDFunctionSet failed: %08lx\n", GetLastError());
|
||
|
|
if (set)
|
||
|
|
{
|
||
|
|
funcY funcAddr = NULL;
|
||
|
|
HCRYPTOIDFUNCADDR hFuncAddr = NULL;
|
||
|
|
|
||
|
|
/* This crashes
|
||
|
|
ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0, NULL,
|
||
|
|
NULL);
|
||
|
|
*/
|
||
|
|
ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, 0, 0,
|
||
|
|
(void **)&funcAddr, &hFuncAddr);
|
||
|
|
ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
|
||
|
|
"Expected ERROR_FILE_NOT_FOUND, got %ld\n", GetLastError());
|
||
|
|
ret = CryptGetOIDFunctionAddress(set, X509_ASN_ENCODING, X509_CERT, 0,
|
||
|
|
(void **)&funcAddr, &hFuncAddr);
|
||
|
|
ok(!ret && GetLastError() == ERROR_FILE_NOT_FOUND,
|
||
|
|
"Expected ERROR_FILE_NOT_FOUND, got %ld\n", GetLastError());
|
||
|
|
ret = CryptGetOIDFunctionAddress(set, 0, X509_CERT, 0,
|
||
|
|
(void **)&funcAddr, &hFuncAddr);
|
||
|
|
ok(ret, "CryptGetOIDFunctionAddress failed: %ld\n", GetLastError());
|
||
|
|
if (funcAddr)
|
||
|
|
{
|
||
|
|
int y = funcAddr(0xabadc0da);
|
||
|
|
|
||
|
|
ok(y == 0xabadc0da, "Unexpected return (%d) from function\n", y);
|
||
|
|
CryptFreeOIDFunctionAddress(hFuncAddr, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void test_registerOIDFunction(void)
|
||
|
|
{
|
||
|
|
static const WCHAR bogusDll[] = { 'b','o','g','u','s','.','d','l','l',0 };
|
||
|
|
BOOL ret;
|
||
|
|
|
||
|
|
/* oddly, this succeeds under WinXP; the function name key is merely
|
||
|
|
* omitted. This may be a side effect of the registry code, I don't know.
|
||
|
|
* I don't check it because I doubt anyone would depend on it.
|
||
|
|
ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, NULL,
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
*/
|
||
|
|
/* On windows XP, GetLastError is incorrectly being set with an HRESULT,
|
||
|
|
* HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER)
|
||
|
|
*/
|
||
|
|
ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo", NULL, bogusDll,
|
||
|
|
NULL);
|
||
|
|
ok(!ret && GetLastError() == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER),
|
||
|
|
"Expected ERROR_INVALID_PARAMETER: %ld\n", GetLastError());
|
||
|
|
/* This has no effect, but "succeeds" on XP */
|
||
|
|
ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "foo",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", NULL, NULL);
|
||
|
|
ok(ret, "Expected pseudo-success, got %ld\n", GetLastError());
|
||
|
|
ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
ok(ret, "CryptRegisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10");
|
||
|
|
ok(ret, "CryptUnregisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
ret = CryptRegisterOIDFunction(X509_ASN_ENCODING, "bogus",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
ok(ret, "CryptRegisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
ret = CryptUnregisterOIDFunction(X509_ASN_ENCODING, "bogus",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10");
|
||
|
|
ok(ret, "CryptUnregisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
/* This has no effect */
|
||
|
|
ret = CryptRegisterOIDFunction(PKCS_7_ASN_ENCODING, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
ok(ret, "CryptRegisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
/* Check with bogus encoding type: */
|
||
|
|
ret = CryptRegisterOIDFunction(0, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
ok(ret, "CryptRegisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
/* This is written with value 3 verbatim. Thus, the encoding type isn't
|
||
|
|
* (for now) treated as a mask.
|
||
|
|
*/
|
||
|
|
ret = CryptRegisterOIDFunction(3, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10", bogusDll, NULL);
|
||
|
|
ok(ret, "CryptRegisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
ret = CryptUnregisterOIDFunction(3, "CryptDllEncodeObject",
|
||
|
|
"1.2.3.4.5.6.7.8.9.10");
|
||
|
|
ok(ret, "CryptUnregisterOIDFunction failed: %ld\n", GetLastError());
|
||
|
|
}
|
||
|
|
|
||
|
|
START_TEST(oid)
|
||
|
|
{
|
||
|
|
test_oidFunctionSet();
|
||
|
|
test_installOIDFunctionAddress();
|
||
|
|
test_registerOIDFunction();
|
||
|
|
}
|