mirror of
https://github.com/reactos/reactos.git
synced 2024-10-03 16:05:37 +00:00
8aeb6a920a
Add testcase of kernel32!SetComputerNameExW function. ROSTESTS-227
114 lines
3.9 KiB
C
114 lines
3.9 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
|
* PURPOSE: Tests for the SetComputerNameExW API
|
|
* COPYRIGHT: Victor Martinez Calvo (victor.martinez@reactos.org)
|
|
* Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
|
|
*/
|
|
#include <apitest.h>
|
|
|
|
#define WIN32_NO_STATUS
|
|
#include <stdio.h>
|
|
#include <ndk/rtltypes.h>
|
|
|
|
#undef _WIN32_WINNT
|
|
#define _WIN32_WINNT 0x0600
|
|
#include <winreg.h>
|
|
|
|
static HKEY OpenHostNameKey(void)
|
|
{
|
|
static const WCHAR
|
|
RegHostNameKey[] = L"System\\CurrentControlSet\\Services\\Tcpip\\Parameters";
|
|
HKEY hKey = NULL;
|
|
LONG Error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, RegHostNameKey, 0, KEY_ALL_ACCESS, &hKey);
|
|
if (!Error)
|
|
return hKey;
|
|
return NULL;
|
|
}
|
|
|
|
static HKEY OpenComputerNameKey(void)
|
|
{
|
|
static const WCHAR
|
|
RegComputerNameKey[] = L"System\\CurrentControlSet\\Control\\ComputerName\\ComputerName";
|
|
HKEY hKey = NULL;
|
|
LONG Error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, RegComputerNameKey, 0, KEY_ALL_ACCESS, &hKey);
|
|
if (!Error)
|
|
return hKey;
|
|
return NULL;
|
|
}
|
|
|
|
START_TEST(SetComputerNameExW)
|
|
{
|
|
LONG Error;
|
|
BOOL ret;
|
|
HKEY hKeyHN, hKeyCN;
|
|
DWORD cbData;
|
|
WCHAR szHostNameOld[MAX_PATH], szHostNameNew[MAX_PATH];
|
|
WCHAR szComputerNameOld[MAX_PATH], szComputerNameNew[MAX_PATH];
|
|
static const WCHAR szNewName[] = L"SRVROSTEST";
|
|
|
|
/* Open keys */
|
|
hKeyHN = OpenHostNameKey();
|
|
hKeyCN = OpenComputerNameKey();
|
|
if (!hKeyHN || !hKeyCN)
|
|
{
|
|
if (hKeyHN)
|
|
RegCloseKey(hKeyHN);
|
|
if (hKeyCN)
|
|
RegCloseKey(hKeyCN);
|
|
skip("Unable to open keys (%p, %p).\n", hKeyHN, hKeyCN);
|
|
return;
|
|
}
|
|
|
|
/* Get Old Hostname */
|
|
szHostNameOld[0] = UNICODE_NULL;
|
|
cbData = sizeof(szHostNameOld);
|
|
Error = RegQueryValueExW(hKeyHN, L"Hostname", NULL, NULL, (LPBYTE)szHostNameOld, &cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
ok(szHostNameOld[0], "szHostNameOld is %S", szHostNameOld);
|
|
|
|
/* Get Old Computer Name */
|
|
szComputerNameOld[0] = UNICODE_NULL;
|
|
cbData = sizeof(szComputerNameOld);
|
|
Error = RegQueryValueExW(hKeyCN, L"ComputerName", NULL, NULL, (LPBYTE)szComputerNameOld, &cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
ok(szComputerNameOld[0], "szHostNameOld is %S", szComputerNameOld);
|
|
|
|
/* Change the value */
|
|
ret = SetComputerNameExW(ComputerNamePhysicalDnsHostname, szNewName);
|
|
ok_int(ret, TRUE);
|
|
|
|
/* Get New Hostname */
|
|
szHostNameNew[0] = UNICODE_NULL;
|
|
cbData = sizeof(szHostNameNew);
|
|
Error = RegQueryValueExW(hKeyHN, L"Hostname", NULL, NULL, (LPBYTE)szHostNameNew, &cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
ok(szHostNameNew[0], "szHostNameNew was empty.\n");
|
|
ok(lstrcmpW(szHostNameNew, szHostNameOld) == 0,
|
|
"szHostNameNew '%S' should be szHostNameOld '%S'.\n", szHostNameNew, szHostNameOld);
|
|
|
|
/* Get New Computer Name */
|
|
szComputerNameNew[0] = UNICODE_NULL;
|
|
cbData = sizeof(szComputerNameNew);
|
|
Error = RegQueryValueExW(hKeyCN, L"ComputerName", NULL, NULL, (LPBYTE)szComputerNameNew, &cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
ok(szComputerNameNew[0], "szComputerNameNew was empty.\n");
|
|
ok(lstrcmpW(szComputerNameNew, szNewName) == 0,
|
|
"szComputerNameNew '%S' should be szNewName '%S'.\n", szComputerNameNew, szNewName);
|
|
|
|
/* Restore the registry values */
|
|
cbData = (lstrlenW(szHostNameOld) + 1) * sizeof(WCHAR);
|
|
Error = RegSetValueExW(hKeyHN, L"Hostname", 0, REG_SZ, (LPBYTE)szHostNameOld, cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
|
|
cbData = (lstrlenW(szComputerNameOld) + 1) * sizeof(WCHAR);
|
|
Error = RegSetValueExW(hKeyCN, L"ComputerName", 0, REG_SZ, (LPBYTE)szComputerNameOld, cbData);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
|
|
/* Close keys */
|
|
Error = RegCloseKey(hKeyHN);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
Error = RegCloseKey(hKeyCN);
|
|
ok_long(Error, ERROR_SUCCESS);
|
|
}
|