| /* |
| * tests |
| * |
| * Copyright 2006 Robert Reif |
| * |
| * 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 <windef.h> |
| #include <winbase.h> |
| #include <winnls.h> |
| #define SECURITY_WIN32 |
| #include <security.h> |
| #include <schannel.h> |
| |
| #include "wine/test.h" |
| |
| static HMODULE secdll; |
| |
| static BOOLEAN (WINAPI * pGetComputerObjectNameA)(EXTENDED_NAME_FORMAT NameFormat, LPSTR lpNameBuffer, PULONG lpnSize); |
| static BOOLEAN (WINAPI * pGetComputerObjectNameW)(EXTENDED_NAME_FORMAT NameFormat, LPWSTR lpNameBuffer, PULONG lpnSize); |
| static BOOLEAN (WINAPI * pGetUserNameExA)(EXTENDED_NAME_FORMAT NameFormat, LPSTR lpNameBuffer, PULONG lpnSize); |
| static BOOLEAN (WINAPI * pGetUserNameExW)(EXTENDED_NAME_FORMAT NameFormat, LPWSTR lpNameBuffer, PULONG lpnSize); |
| static PSecurityFunctionTableA (SEC_ENTRY * pInitSecurityInterfaceA)(void); |
| static PSecurityFunctionTableW (SEC_ENTRY * pInitSecurityInterfaceW)(void); |
| |
| static EXTENDED_NAME_FORMAT formats[] = { |
| NameUnknown, NameFullyQualifiedDN, NameSamCompatible, NameDisplay, |
| NameUniqueId, NameCanonical, NameUserPrincipal, NameCanonicalEx, |
| NameServicePrincipal, NameDnsDomain |
| }; |
| |
| static void testGetComputerObjectNameA(void) |
| { |
| char name[256]; |
| ULONG size; |
| BOOLEAN rc; |
| UINT i; |
| |
| for (i = 0; i < (sizeof(formats) / sizeof(formats[0])); i++) { |
| size = sizeof(name); |
| ZeroMemory(name, sizeof(name)); |
| rc = pGetComputerObjectNameA(formats[i], name, &size); |
| ok(rc || ((formats[i] == NameUnknown) && |
| (GetLastError() == ERROR_INVALID_PARAMETER)) || |
| (GetLastError() == ERROR_CANT_ACCESS_DOMAIN_INFO) || |
| (GetLastError() == ERROR_NO_SUCH_DOMAIN) || |
| (GetLastError() == ERROR_NO_SUCH_USER) || |
| (GetLastError() == ERROR_NONE_MAPPED) || |
| (GetLastError() == ERROR_ACCESS_DENIED), |
| "GetComputerObjectNameA(%d) failed: %d\n", |
| formats[i], GetLastError()); |
| if (rc) |
| trace("GetComputerObjectNameA() returned %s\n", name); |
| } |
| } |
| |
| static void testGetComputerObjectNameW(void) |
| { |
| WCHAR nameW[256]; |
| ULONG size; |
| BOOLEAN rc; |
| UINT i; |
| |
| for (i = 0; i < (sizeof(formats) / sizeof(formats[0])); i++) { |
| size = sizeof(nameW)/sizeof(nameW[0]); |
| ZeroMemory(nameW, sizeof(nameW)); |
| rc = pGetComputerObjectNameW(formats[i], nameW, &size); |
| ok(rc || ((formats[i] == NameUnknown) && |
| (GetLastError() == ERROR_INVALID_PARAMETER)) || |
| (GetLastError() == ERROR_CANT_ACCESS_DOMAIN_INFO) || |
| (GetLastError() == ERROR_NO_SUCH_DOMAIN) || |
| (GetLastError() == ERROR_NO_SUCH_USER) || |
| (GetLastError() == ERROR_NONE_MAPPED) || |
| (GetLastError() == ERROR_ACCESS_DENIED), |
| "GetComputerObjectNameW(%d) failed: %d\n", |
| formats[i], GetLastError()); |
| if (rc) { |
| char name[256]; |
| WideCharToMultiByte( CP_ACP, 0, nameW, -1, name, sizeof(name), NULL, NULL ); |
| trace("GetComputerObjectNameW() returned %s\n", name); |
| } |
| } |
| } |
| |
| static void testGetUserNameExA(void) |
| { |
| char name[256]; |
| ULONG size; |
| BOOLEAN rc; |
| UINT i; |
| |
| for (i = 0; i < (sizeof(formats) / sizeof(formats[0])); i++) { |
| size = sizeof(name); |
| ZeroMemory(name, sizeof(name)); |
| rc = pGetUserNameExA(formats[i], name, &size); |
| ok(rc || |
| (formats[i] == NameUnknown && |
| GetLastError() == ERROR_NO_SUCH_USER) || |
| GetLastError() == ERROR_NONE_MAPPED || |
| broken(formats[i] == NameDnsDomain && |
| GetLastError() == ERROR_INVALID_PARAMETER), |
| "GetUserNameExW(%d) failed: %d\n", |
| formats[i], GetLastError()); |
| } |
| |
| if (0) /* Crashes on Windows */ |
| pGetUserNameExA(NameSamCompatible, NULL, NULL); |
| |
| size = 0; |
| rc = pGetUserNameExA(NameSamCompatible, NULL, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(size != 0, "Expected size to be set to required size\n"); |
| |
| if (0) /* Crashes on Windows with big enough size */ |
| { |
| /* Returned size is already big enough */ |
| pGetUserNameExA(NameSamCompatible, NULL, &size); |
| } |
| |
| size = 0; |
| rc = pGetUserNameExA(NameSamCompatible, name, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(size != 0, "Expected size to be set to required size\n"); |
| size = 1; |
| name[0] = 0xff; |
| rc = pGetUserNameExA(NameSamCompatible, name, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(1 < size, "Expected size to be set to required size\n"); |
| ok(name[0] == (char) 0xff, "Expected unchanged buffer\n"); |
| } |
| |
| static void testGetUserNameExW(void) |
| { |
| WCHAR nameW[256]; |
| ULONG size; |
| BOOLEAN rc; |
| UINT i; |
| |
| for (i = 0; i < (sizeof(formats) / sizeof(formats[0])); i++) { |
| size = sizeof(nameW); |
| ZeroMemory(nameW, sizeof(nameW)); |
| rc = pGetUserNameExW(formats[i], nameW, &size); |
| ok(rc || |
| (formats[i] == NameUnknown && |
| GetLastError() == ERROR_NO_SUCH_USER) || |
| GetLastError() == ERROR_NONE_MAPPED || |
| broken(formats[i] == NameDnsDomain && |
| GetLastError() == ERROR_INVALID_PARAMETER), |
| "GetUserNameExW(%d) failed: %d\n", |
| formats[i], GetLastError()); |
| } |
| |
| if (0) /* Crashes on Windows */ |
| pGetUserNameExW(NameSamCompatible, NULL, NULL); |
| |
| size = 0; |
| rc = pGetUserNameExW(NameSamCompatible, NULL, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(size != 0, "Expected size to be set to required size\n"); |
| |
| if (0) /* Crashes on Windows with big enough size */ |
| { |
| /* Returned size is already big enough */ |
| pGetUserNameExW(NameSamCompatible, NULL, &size); |
| } |
| |
| size = 0; |
| rc = pGetUserNameExW(NameSamCompatible, nameW, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(size != 0, "Expected size to be set to required size\n"); |
| size = 1; |
| nameW[0] = 0xff; |
| rc = pGetUserNameExW(NameSamCompatible, nameW, &size); |
| ok(! rc && GetLastError() == ERROR_MORE_DATA, "Expected fail with ERROR_MORE_DATA, got %d with %u\n", rc, GetLastError()); |
| ok(1 < size, "Expected size to be set to required size\n"); |
| ok(nameW[0] == (WCHAR) 0xff, "Expected unchanged buffer\n"); |
| } |
| |
| static void test_InitSecurityInterface(void) |
| { |
| PSecurityFunctionTableA sftA; |
| PSecurityFunctionTableW sftW; |
| |
| sftA = pInitSecurityInterfaceA(); |
| ok(sftA != NULL, "pInitSecurityInterfaceA failed\n"); |
| ok(sftA->dwVersion == SECURITY_SUPPORT_PROVIDER_INTERFACE_VERSION, "wrong dwVersion %d in security function table\n", sftA->dwVersion); |
| ok(!sftA->Reserved2, |
| "Reserved2 should be NULL instead of %p in security function table\n", |
| sftA->Reserved2); |
| ok(sftA->Reserved3 == sftA->EncryptMessage, |
| "Reserved3 should be equal to EncryptMessage in the security function table\n"); |
| ok(sftA->Reserved4 == sftA->DecryptMessage, |
| "Reserved4 should be equal to DecryptMessage in the security function table\n"); |
| |
| if (!pInitSecurityInterfaceW) |
| { |
| win_skip("InitSecurityInterfaceW not exported by secur32.dll\n"); |
| return; |
| } |
| |
| sftW = pInitSecurityInterfaceW(); |
| ok(sftW != NULL, "pInitSecurityInterfaceW failed\n"); |
| ok(sftW->dwVersion == SECURITY_SUPPORT_PROVIDER_INTERFACE_VERSION, "wrong dwVersion %d in security function table\n", sftW->dwVersion); |
| ok(!sftW->Reserved2, "Reserved2 should be NULL instead of %p in security function table\n", sftW->Reserved2); |
| ok(sftW->Reserved3 == sftW->EncryptMessage, "Reserved3 should be equal to EncryptMessage in the security function table\n"); |
| ok(sftW->Reserved4 == sftW->DecryptMessage, "Reserved4 should be equal to DecryptMessage in the security function table\n"); |
| } |
| |
| START_TEST(secur32) |
| { |
| secdll = LoadLibraryA("secur32.dll"); |
| |
| if (!secdll) |
| secdll = LoadLibraryA("security.dll"); |
| |
| if (secdll) |
| { |
| pGetComputerObjectNameA = (PVOID)GetProcAddress(secdll, "GetComputerObjectNameA"); |
| pGetComputerObjectNameW = (PVOID)GetProcAddress(secdll, "GetComputerObjectNameW"); |
| pGetUserNameExA = (PVOID)GetProcAddress(secdll, "GetUserNameExA"); |
| pGetUserNameExW = (PVOID)GetProcAddress(secdll, "GetUserNameExW"); |
| pInitSecurityInterfaceA = (PVOID)GetProcAddress(secdll, "InitSecurityInterfaceA"); |
| pInitSecurityInterfaceW = (PVOID)GetProcAddress(secdll, "InitSecurityInterfaceW"); |
| |
| if (pGetComputerObjectNameA) |
| testGetComputerObjectNameA(); |
| else |
| win_skip("GetComputerObjectNameA not exported by secur32.dll\n"); |
| |
| if (pGetComputerObjectNameW) |
| testGetComputerObjectNameW(); |
| else |
| win_skip("GetComputerObjectNameW not exported by secur32.dll\n"); |
| |
| if (pGetUserNameExA) |
| testGetUserNameExA(); |
| else |
| win_skip("GetUserNameExA not exported by secur32.dll\n"); |
| |
| if (pGetUserNameExW) |
| testGetUserNameExW(); |
| else |
| win_skip("GetUserNameExW not exported by secur32.dll\n"); |
| |
| test_InitSecurityInterface(); |
| |
| FreeLibrary(secdll); |
| } |
| } |