From 735df24617154f689cbedf1510afea35a47a40e0 Mon Sep 17 00:00:00 2001 From: Yuval Fledel Date: Wed, 25 Apr 2007 22:49:08 +0300 Subject: [PATCH] schannel: Tests for SpLsaModeInitialize, SpUserModeInitialize and GetInfo. --- .gitignore | 4 + Makefile.in | 2 + configure | 3 + configure.ac | 1 + dlls/Makefile.in | 1 + dlls/schannel/tests/Makefile.in | 13 +++ dlls/schannel/tests/main.c | 180 ++++++++++++++++++++++++++++++++ programs/winetest/Makefile.in | 3 + programs/winetest/winetest.rc | 1 + 9 files changed, 208 insertions(+) create mode 100644 dlls/schannel/tests/Makefile.in create mode 100644 dlls/schannel/tests/main.c diff --git a/.gitignore b/.gitignore index ecdb9d272b0..7bbff768275 100644 --- a/.gitignore +++ b/.gitignore @@ -455,6 +455,9 @@ dlls/rsaenh/tests/rsaenh_crosstest.exe dlls/rsaenh/tests/testlist.c dlls/rsaenh/version.res dlls/sane.ds/rsrc.res +dlls/schannel/tests/*.ok +dlls/schannel/tests/schannel_crosstest.exe +dlls/schannel/tests/testlist.c dlls/secur32/libsecur32.def dlls/secur32/tests/*.ok dlls/secur32/tests/secur32_crosstest.exe @@ -860,6 +863,7 @@ programs/winetest/riched20_test.exe programs/winetest/rpcrt4_test.exe programs/winetest/rsabase_test.exe programs/winetest/rsaenh_test.exe +programs/winetest/schannel_test.exe programs/winetest/secur32_test.exe programs/winetest/serialui_test.exe programs/winetest/setupapi_test.exe diff --git a/Makefile.in b/Makefile.in index bd311621e6b..1f8f0c9ebf2 100644 --- a/Makefile.in +++ b/Makefile.in @@ -344,6 +344,7 @@ ALL_MAKEFILES = \ dlls/rsaenh/tests/Makefile \ dlls/sane.ds/Makefile \ dlls/schannel/Makefile \ + dlls/schannel/tests/Makefile \ dlls/secur32/Makefile \ dlls/secur32/tests/Makefile \ dlls/security/Makefile \ @@ -688,6 +689,7 @@ dlls/rsaenh/Makefile: dlls/rsaenh/Makefile.in dlls/Makedll.rules dlls/rsaenh/tests/Makefile: dlls/rsaenh/tests/Makefile.in dlls/Maketest.rules dlls/sane.ds/Makefile: dlls/sane.ds/Makefile.in dlls/Makedll.rules dlls/schannel/Makefile: dlls/schannel/Makefile.in dlls/Makedll.rules +dlls/schannel/tests/Makefile: dlls/schannel/tests/Makefile.in dlls/Maketest.rules dlls/secur32/Makefile: dlls/secur32/Makefile.in dlls/Makedll.rules dlls/secur32/tests/Makefile: dlls/secur32/tests/Makefile.in dlls/Maketest.rules dlls/security/Makefile: dlls/security/Makefile.in dlls/Makedll.rules diff --git a/configure b/configure index e07ad7496d0..3d916328870 100755 --- a/configure +++ b/configure @@ -20693,6 +20693,8 @@ ac_config_files="$ac_config_files dlls/sane.ds/Makefile" ac_config_files="$ac_config_files dlls/schannel/Makefile" +ac_config_files="$ac_config_files dlls/schannel/tests/Makefile" + ac_config_files="$ac_config_files dlls/secur32/Makefile" ac_config_files="$ac_config_files dlls/secur32/tests/Makefile" @@ -21731,6 +21733,7 @@ do "dlls/rsaenh/tests/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/rsaenh/tests/Makefile" ;; "dlls/sane.ds/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/sane.ds/Makefile" ;; "dlls/schannel/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/schannel/Makefile" ;; + "dlls/schannel/tests/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/schannel/tests/Makefile" ;; "dlls/secur32/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/secur32/Makefile" ;; "dlls/secur32/tests/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/secur32/tests/Makefile" ;; "dlls/security/Makefile") CONFIG_FILES="$CONFIG_FILES dlls/security/Makefile" ;; diff --git a/configure.ac b/configure.ac index 757e9570b24..ddec5f48cd6 100644 --- a/configure.ac +++ b/configure.ac @@ -1671,6 +1671,7 @@ AC_CONFIG_FILES([dlls/rsaenh/Makefile]) AC_CONFIG_FILES([dlls/rsaenh/tests/Makefile]) AC_CONFIG_FILES([dlls/sane.ds/Makefile]) AC_CONFIG_FILES([dlls/schannel/Makefile]) +AC_CONFIG_FILES([dlls/schannel/tests/Makefile]) AC_CONFIG_FILES([dlls/secur32/Makefile]) AC_CONFIG_FILES([dlls/secur32/tests/Makefile]) AC_CONFIG_FILES([dlls/security/Makefile]) diff --git a/dlls/Makefile.in b/dlls/Makefile.in index 1a07abfbd84..f9707e7eda2 100644 --- a/dlls/Makefile.in +++ b/dlls/Makefile.in @@ -269,6 +269,7 @@ TESTSUBDIRS = \ rpcrt4/tests \ rsabase/tests \ rsaenh/tests \ + schannel/tests \ secur32/tests \ serialui/tests \ setupapi/tests \ diff --git a/dlls/schannel/tests/Makefile.in b/dlls/schannel/tests/Makefile.in new file mode 100644 index 00000000000..02621079aad --- /dev/null +++ b/dlls/schannel/tests/Makefile.in @@ -0,0 +1,13 @@ +TOPSRCDIR = @top_srcdir@ +TOPOBJDIR = ../../.. +SRCDIR = @srcdir@ +VPATH = @srcdir@ +TESTDLL = schannel.dll +IMPORTS = crypt32 advapi32 kernel32 + +CTESTS = \ + main.c + +@MAKE_TEST_RULES@ + +@DEPENDENCIES@ # everything below this line is overwritten by make depend diff --git a/dlls/schannel/tests/main.c b/dlls/schannel/tests/main.c new file mode 100644 index 00000000000..737495e267e --- /dev/null +++ b/dlls/schannel/tests/main.c @@ -0,0 +1,180 @@ +/* + * 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 +#include +#include "ntstatus.h" +#define WIN32_NO_STATUS +#include +#include +#define SECURITY_WIN32 +#include +#include +#include +#include + +#include "wine/test.h" + +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, "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: 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, "cbMaxToken: 0x%lx\n", + PackageInfo.cbMaxToken); +} + +START_TEST(main) +{ + HMODULE hMod = LoadLibraryA("schannel.dll"); + if (!hMod) { + skip("schannel.dll not found.\n"); + return; + } + + pSpLsaModeInitialize = GetProcAddress(hMod, "SpLsaModeInitialize"); + pSpUserModeInitialize = GetProcAddress(hMod, "SpUserModeInitialize"); + + if (pSpLsaModeInitialize && pSpUserModeInitialize) + { + testInitialize(); + testGetInfo(); + } + else skip( "schannel functions not found\n" ); + + FreeLibrary(hMod); +} diff --git a/programs/winetest/Makefile.in b/programs/winetest/Makefile.in index a15819cc08d..c5afe482064 100644 --- a/programs/winetest/Makefile.in +++ b/programs/winetest/Makefile.in @@ -68,6 +68,7 @@ TESTBINS = \ rpcrt4_test.exe \ rsabase_test.exe \ rsaenh_test.exe \ + schannel_test.exe \ secur32_test.exe \ serialui_test.exe \ setupapi_test.exe \ @@ -176,6 +177,8 @@ rsabase_test.exe: $(DLLDIR)/rsabase/tests/rsabase_test.exe$(DLLEXT) cp $(DLLDIR)/rsabase/tests/rsabase_test.exe$(DLLEXT) $@ && $(STRIP) $@ rsaenh_test.exe: $(DLLDIR)/rsaenh/tests/rsaenh_test.exe$(DLLEXT) cp $(DLLDIR)/rsaenh/tests/rsaenh_test.exe$(DLLEXT) $@ && $(STRIP) $@ +schannel_test.exe: $(DLLDIR)/schannel/tests/schannel_test.exe$(DLLEXT) + cp $(DLLDIR)/schannel/tests/schannel_test.exe$(DLLEXT) $@ && $(STRIP) $@ secur32_test.exe: $(DLLDIR)/secur32/tests/secur32_test.exe$(DLLEXT) cp $(DLLDIR)/secur32/tests/secur32_test.exe$(DLLEXT) $@ && $(STRIP) $@ serialui_test.exe: $(DLLDIR)/serialui/tests/serialui_test.exe$(DLLEXT) diff --git a/programs/winetest/winetest.rc b/programs/winetest/winetest.rc index 3e7d00239c8..103a03740e0 100644 --- a/programs/winetest/winetest.rc +++ b/programs/winetest/winetest.rc @@ -191,6 +191,7 @@ riched20_test.exe TESTRES "riched20_test.exe" rpcrt4_test.exe TESTRES "rpcrt4_test.exe" rsabase_test.exe TESTRES "rsabase_test.exe" rsaenh_test.exe TESTRES "rsaenh_test.exe" +schannel_test.exe TESTRES "schannel_test.exe" secur32_test.exe TESTRES "secur32_test.exe" serialui_test.exe TESTRES "serialui_test.exe" setupapi_test.exe TESTRES "setupapi_test.exe"