196 lines
7.3 KiB
C
196 lines
7.3 KiB
C
/*
|
|
* Schannel tests
|
|
*
|
|
* Copyright 2006 Yuval Fledel
|
|
*
|
|
* 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include "ntstatus.h"
|
|
#define WIN32_NO_STATUS
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#define SECURITY_WIN32
|
|
#include <security.h>
|
|
#include <schannel.h>
|
|
#include <ntsecapi.h>
|
|
#include <ntsecpkg.h>
|
|
|
|
#include "wine/test.h"
|
|
|
|
/* Helper macros to find the size of SECPKG_FUNCTION_TABLE */
|
|
#define SECPKG_FUNCTION_TABLE_SIZE_1 FIELD_OFFSET(SECPKG_FUNCTION_TABLE, \
|
|
SetContextAttributes)
|
|
#define SECPKG_FUNCTION_TABLE_SIZE_2 FIELD_OFFSET(SECPKG_FUNCTION_TABLE, \
|
|
SetCredentialsAttributes)
|
|
#define SECPKG_FUNCTION_TABLE_SIZE_3 sizeof(SECPKG_FUNCTION_TABLE)
|
|
|
|
static NTSTATUS (NTAPI *pSpLsaModeInitialize)(ULONG, PULONG,
|
|
PSECPKG_FUNCTION_TABLE*, PULONG);
|
|
static NTSTATUS (NTAPI *pSpUserModeInitialize)(ULONG, PULONG,
|
|
PSECPKG_USER_FUNCTION_TABLE*, PULONG);
|
|
|
|
static void testInitialize(void)
|
|
{
|
|
PSECPKG_USER_FUNCTION_TABLE pUserTables, pUserTables2;
|
|
PSECPKG_FUNCTION_TABLE pTables, pTables2;
|
|
ULONG cTables = 0, cUserTables = 0, Version = 0;
|
|
NTSTATUS status;
|
|
|
|
/* Passing NULL into one of the parameters of SpLsaModeInitialize or
|
|
SpUserModeInitialize causes a crash. */
|
|
|
|
/* SpLsaModeInitialize does not care about the LSA version. */
|
|
status = pSpLsaModeInitialize(0, &Version, &pTables2, &cTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(cTables == 2, "cTables: %d\n", cTables);
|
|
ok(pTables2 != NULL,"pTables: %p\n", pTables2);
|
|
|
|
/* We can call it as many times we want. */
|
|
status = pSpLsaModeInitialize(0x10000, &Version, &pTables, &cTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(cTables == 2, "cTables: %d\n", cTables);
|
|
ok(pTables != NULL, "pTables: %p\n", pTables);
|
|
/* It will always return the same pointer. */
|
|
ok(pTables == pTables2, "pTables: %p, pTables2: %p\n", pTables, pTables2);
|
|
|
|
status = pSpLsaModeInitialize(0x23456, &Version, &pTables, &cTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(cTables == 2, "cTables: %d\n", cTables);
|
|
ok(pTables != NULL, "pTables: %p\n", pTables);
|
|
ok(pTables == pTables2, "pTables: %p, pTables2: %p\n", pTables, pTables2);
|
|
|
|
/* Bad versions to SpUserModeInitialize. Parameters unchanged */
|
|
Version = 0xdead;
|
|
cUserTables = 0xdead;
|
|
pUserTables = NULL;
|
|
status = pSpUserModeInitialize(0, &Version, &pUserTables, &cUserTables);
|
|
ok(status == STATUS_INVALID_PARAMETER, "status: 0x%x\n", status);
|
|
ok(Version == 0xdead, "Version: 0x%x\n", Version);
|
|
ok(cUserTables == 0xdead, "cTables: %d\n", cUserTables);
|
|
ok(pUserTables == NULL, "pUserTables: %p\n", pUserTables);
|
|
|
|
status = pSpUserModeInitialize(0x20000, &Version, &pUserTables,
|
|
&cUserTables);
|
|
ok(status == STATUS_INVALID_PARAMETER, "status: 0x%x\n", status);
|
|
ok(Version == 0xdead, "Version: 0x%x\n", Version);
|
|
ok(cUserTables == 0xdead, "cTables: %d\n", cUserTables);
|
|
ok(pUserTables == NULL, "pUserTables: %p\n", pUserTables);
|
|
|
|
/* Good version to SpUserModeInitialize */
|
|
status = pSpUserModeInitialize(SECPKG_INTERFACE_VERSION, &Version,
|
|
&pUserTables, &cUserTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(Version == SECPKG_INTERFACE_VERSION, "Version: 0x%x\n", Version);
|
|
ok(cUserTables == 2, "cUserTables: %d\n", cUserTables);
|
|
ok(pUserTables != NULL, "pUserTables: %p\n", pUserTables);
|
|
|
|
/* Initializing user again */
|
|
status = pSpUserModeInitialize(SECPKG_INTERFACE_VERSION, &Version,
|
|
&pUserTables2, &cTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(pUserTables == pUserTables2, "pUserTables: %p, pUserTables2: %p\n",
|
|
pUserTables, pUserTables2);
|
|
}
|
|
|
|
/* A helper function to find the dispatch table of the next package.
|
|
Needed because SECPKG_FUNCTION_TABLE's size depend on the version */
|
|
static PSECPKG_FUNCTION_TABLE getNextSecPkgTable(PSECPKG_FUNCTION_TABLE pTable,
|
|
ULONG Version)
|
|
{
|
|
size_t size;
|
|
|
|
if (Version == SECPKG_INTERFACE_VERSION)
|
|
size = SECPKG_FUNCTION_TABLE_SIZE_1;
|
|
else if (Version == SECPKG_INTERFACE_VERSION_2)
|
|
size = SECPKG_FUNCTION_TABLE_SIZE_2;
|
|
else if (Version == SECPKG_INTERFACE_VERSION_3)
|
|
size = SECPKG_FUNCTION_TABLE_SIZE_3;
|
|
else {
|
|
ok(FALSE, "Unknown package version 0x%x\n", Version);
|
|
return NULL;
|
|
}
|
|
|
|
return (PSECPKG_FUNCTION_TABLE)((PBYTE)pTable + size);
|
|
}
|
|
|
|
static void testGetInfo(void)
|
|
{
|
|
PSECPKG_FUNCTION_TABLE pTables;
|
|
SecPkgInfoW PackageInfo;
|
|
ULONG cTables, Version;
|
|
NTSTATUS status;
|
|
|
|
/* Get the dispatch table */
|
|
status = pSpLsaModeInitialize(0, &Version, &pTables, &cTables);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
|
|
/* Passing NULL into ->GetInfo causes a crash. */
|
|
|
|
/* First package: Unified */
|
|
status = pTables->GetInfo(&PackageInfo);
|
|
ok(status == STATUS_SUCCESS, "status: 0x%x\n", status);
|
|
ok(PackageInfo.fCapabilities == 0x107b3, "fCapabilities: 0x%lx\n",
|
|
PackageInfo.fCapabilities);
|
|
ok(PackageInfo.wVersion == 1, "wVersion: %d\n", PackageInfo.wVersion);
|
|
ok(PackageInfo.wRPCID == 14, "wRPCID: %d\n", PackageInfo.wRPCID);
|
|
ok(PackageInfo.cbMaxToken == 0x4000 ||
|
|
PackageInfo.cbMaxToken == 0x6000, /* Vista */
|
|
"cbMaxToken: 0x%lx\n",
|
|
PackageInfo.cbMaxToken);
|
|
|
|
/* Second package: SChannel */
|
|
pTables = getNextSecPkgTable(pTables, Version);
|
|
if (!pTables)
|
|
return;
|
|
status = pTables->GetInfo(&PackageInfo);
|
|
ok(status == STATUS_SUCCESS ||
|
|
status == SEC_E_UNSUPPORTED_FUNCTION, /* win2k3 */
|
|
"status: 0x%x\n", status);
|
|
|
|
if (status == STATUS_SUCCESS)
|
|
{
|
|
ok(PackageInfo.fCapabilities == 0x107b3, "fCapabilities: 0x%lx\n",
|
|
PackageInfo.fCapabilities);
|
|
ok(PackageInfo.wVersion == 1, "wVersion: %d\n", PackageInfo.wVersion);
|
|
ok(PackageInfo.wRPCID == 14, "wRPCID: %d\n", PackageInfo.wRPCID);
|
|
ok(PackageInfo.cbMaxToken == 0x4000, "cbMaxToken: 0x%lx\n",
|
|
PackageInfo.cbMaxToken);
|
|
}
|
|
}
|
|
|
|
START_TEST(main)
|
|
{
|
|
HMODULE hMod = LoadLibraryA("schannel.dll");
|
|
if (!hMod) {
|
|
skip("schannel.dll not found.\n");
|
|
return;
|
|
}
|
|
|
|
pSpLsaModeInitialize = (void *)GetProcAddress(hMod, "SpLsaModeInitialize");
|
|
pSpUserModeInitialize = (void *)GetProcAddress(hMod, "SpUserModeInitialize");
|
|
|
|
if (pSpLsaModeInitialize && pSpUserModeInitialize)
|
|
{
|
|
testInitialize();
|
|
testGetInfo();
|
|
}
|
|
else skip( "schannel functions not found\n" );
|
|
|
|
FreeLibrary(hMod);
|
|
}
|