658e74e8a44981f33f33199b16e5d52a725be3ed
[reactos.git] / modules / rostests / apitests / kernel32 / SetComputerNameExW.c
1 /*
2 * PROJECT: ReactOS api tests
3 * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
4 * PURPOSE: Tests for the SetComputerNameExW API
5 * COPYRIGHT: Victor Martinez Calvo (victor.martinez@reactos.org)
6 * Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
7 */
8 #include <apitest.h>
9
10 #define WIN32_NO_STATUS
11 #include <stdio.h>
12 #include <ndk/rtltypes.h>
13
14 #undef _WIN32_WINNT
15 #define _WIN32_WINNT 0x0600
16 #include <winreg.h>
17
18 static HKEY OpenHostNameKey(void)
19 {
20 static const WCHAR
21 RegHostNameKey[] = L"System\\CurrentControlSet\\Services\\Tcpip\\Parameters";
22 HKEY hKey = NULL;
23 LONG Error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, RegHostNameKey, 0, KEY_ALL_ACCESS, &hKey);
24 if (!Error)
25 return hKey;
26 return NULL;
27 }
28
29 static HKEY OpenComputerNameKey(void)
30 {
31 static const WCHAR
32 RegComputerNameKey[] = L"System\\CurrentControlSet\\Control\\ComputerName\\ComputerName";
33 HKEY hKey = NULL;
34 LONG Error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, RegComputerNameKey, 0, KEY_ALL_ACCESS, &hKey);
35 if (!Error)
36 return hKey;
37 return NULL;
38 }
39
40 START_TEST(SetComputerNameExW)
41 {
42 LONG Error;
43 BOOL ret;
44 HKEY hKeyHN, hKeyCN;
45 DWORD cbData;
46 WCHAR szHostNameOld[MAX_PATH], szHostNameNew[MAX_PATH];
47 WCHAR szComputerNameOld[MAX_PATH], szComputerNameNew[MAX_PATH];
48 static const WCHAR szNewName[] = L"SRVROSTEST";
49
50 /* Open keys */
51 hKeyHN = OpenHostNameKey();
52 hKeyCN = OpenComputerNameKey();
53 if (!hKeyHN || !hKeyCN)
54 {
55 if (hKeyHN)
56 RegCloseKey(hKeyHN);
57 if (hKeyCN)
58 RegCloseKey(hKeyCN);
59 skip("Unable to open keys (%p, %p).\n", hKeyHN, hKeyCN);
60 return;
61 }
62
63 /* Get Old Hostname */
64 szHostNameOld[0] = UNICODE_NULL;
65 cbData = sizeof(szHostNameOld);
66 Error = RegQueryValueExW(hKeyHN, L"Hostname", NULL, NULL, (LPBYTE)szHostNameOld, &cbData);
67 ok_long(Error, ERROR_SUCCESS);
68 ok(szHostNameOld[0], "szHostNameOld is %S", szHostNameOld);
69
70 /* Get Old Computer Name */
71 szComputerNameOld[0] = UNICODE_NULL;
72 cbData = sizeof(szComputerNameOld);
73 Error = RegQueryValueExW(hKeyCN, L"ComputerName", NULL, NULL, (LPBYTE)szComputerNameOld, &cbData);
74 ok_long(Error, ERROR_SUCCESS);
75 ok(szComputerNameOld[0], "szHostNameOld is %S", szComputerNameOld);
76
77 /* Change the value */
78 ret = SetComputerNameExW(ComputerNamePhysicalDnsHostname, szNewName);
79 ok_int(ret, TRUE);
80
81 /* Get New Hostname */
82 szHostNameNew[0] = UNICODE_NULL;
83 cbData = sizeof(szHostNameNew);
84 Error = RegQueryValueExW(hKeyHN, L"Hostname", NULL, NULL, (LPBYTE)szHostNameNew, &cbData);
85 ok_long(Error, ERROR_SUCCESS);
86 ok(szHostNameNew[0], "szHostNameNew was empty.\n");
87 ok(lstrcmpW(szHostNameNew, szHostNameOld) == 0,
88 "szHostNameNew '%S' should be szHostNameOld '%S'.\n", szHostNameNew, szHostNameOld);
89
90 /* Get New Computer Name */
91 szComputerNameNew[0] = UNICODE_NULL;
92 cbData = sizeof(szComputerNameNew);
93 Error = RegQueryValueExW(hKeyCN, L"ComputerName", NULL, NULL, (LPBYTE)szComputerNameNew, &cbData);
94 ok_long(Error, ERROR_SUCCESS);
95 ok(szComputerNameNew[0], "szComputerNameNew was empty.\n");
96 ok(lstrcmpW(szComputerNameNew, szNewName) == 0,
97 "szComputerNameNew '%S' should be szNewName '%S'.\n", szComputerNameNew, szNewName);
98
99 /* Restore the registry values */
100 cbData = (lstrlenW(szHostNameOld) + 1) * sizeof(WCHAR);
101 Error = RegSetValueExW(hKeyHN, L"Hostname", 0, REG_SZ, (LPBYTE)szHostNameOld, cbData);
102 ok_long(Error, ERROR_SUCCESS);
103
104 cbData = (lstrlenW(szComputerNameOld) + 1) * sizeof(WCHAR);
105 Error = RegSetValueExW(hKeyCN, L"ComputerName", 0, REG_SZ, (LPBYTE)szComputerNameOld, cbData);
106 ok_long(Error, ERROR_SUCCESS);
107
108 /* Close keys */
109 Error = RegCloseKey(hKeyHN);
110 ok_long(Error, ERROR_SUCCESS);
111 Error = RegCloseKey(hKeyCN);
112 ok_long(Error, ERROR_SUCCESS);
113 }