1 /*
2  * PROJECT:         ReactOS API tests
3  * LICENSE:         LGPLv2.1+ - See COPYING.LIB in the top level directory
4  * PURPOSE:         Test for GetComputerNameEx
5  * PROGRAMMER:      Thomas Faber <thomas.faber@reactos.org>
6  */
7 
8 #include "precomp.h"
9 
10 static
11 VOID
12 TestGetComputerNameEx(
13     _In_ COMPUTER_NAME_FORMAT NameType)
14 {
15     WCHAR Reference[128];
16     DWORD ReferenceLen;
17     WCHAR BufferW[128];
18     CHAR BufferA[128];
19     BOOL Ret;
20     DWORD Size;
21     DWORD Error;
22     ULONG i;
23 
24     Size = RTL_NUMBER_OF(Reference);
25     Ret = GetComputerNameExW(NameType, Reference, &Size);
26     ok(Ret == TRUE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
27     if (!Ret)
28     {
29         skip("[%d] Failed to get reference string\n", NameType);
30         return;
31     }
32     trace("[%d] Reference is %ls\n", NameType, Reference);
33     ReferenceLen = lstrlenW(Reference);
34     ok(ReferenceLen < RTL_NUMBER_OF(Reference),
35        "[%d] Unexpected ReferenceLen %lu\n", NameType, ReferenceLen);
36     if (NameType != ComputerNameDnsDomain && NameType != ComputerNamePhysicalDnsDomain)
37     {
38         ok(ReferenceLen != 0, "[%d] Unexpected ReferenceLen %lu\n", NameType, ReferenceLen);
39     }
40     ok(Size == ReferenceLen, "[%d] Size is %lu, expected %lu\n", NameType, Size, ReferenceLen);
41 
42     /* NULL buffer, NULL size */
43     StartSeh()
44         Ret = GetComputerNameExW(NameType, NULL, NULL);
45         Error = GetLastError();
46         ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
47         ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
48     EndSeh(STATUS_SUCCESS);
49     StartSeh()
50         Ret = GetComputerNameExA(NameType, NULL, NULL);
51         Error = GetLastError();
52         ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
53         ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
54     EndSeh(STATUS_SUCCESS);
55 
56     /* NULL buffer, nonzero size */
57     Size = 0x55555555;
58     Ret = GetComputerNameExW(NameType, NULL, &Size);
59     Error = GetLastError();
60     ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
61     ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
62     ok(Size == 0x55555555, "[%d] Got Size %lu\n", NameType, Size);
63 
64     Size = 0x55555555;
65     Ret = GetComputerNameExA(NameType, NULL, &Size);
66     Error = GetLastError();
67     ok(Ret == FALSE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
68     ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExA returned error %lu\n", NameType, Error);
69     ok(Size == 0x55555555, "[%d] Got Size %lu\n", NameType, Size);
70 
71     /* non-NULL buffer, NULL size */
72     RtlFillMemory(BufferW, sizeof(BufferW), 0x55);
73     Ret = GetComputerNameExW(NameType, BufferW, NULL);
74     Error = GetLastError();
75     ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
76     ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
77     ok(BufferW[0] == 0x5555, "[%d] BufferW[0] = 0x%x\n", NameType, BufferW[0]);
78 
79     RtlFillMemory(BufferA, sizeof(BufferA), 0x55);
80     Ret = GetComputerNameExA(NameType, BufferA, NULL);
81     Error = GetLastError();
82     ok(Ret == FALSE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
83     ok(Error == ERROR_INVALID_PARAMETER, "[%d] GetComputerNameExA returned error %lu\n", NameType, Error);
84     ok(BufferA[0] == 0x55, "[%d] BufferA[0] = 0x%x\n", NameType, BufferA[0]);
85 
86     /* NULL buffer, zero size */
87     Size = 0;
88     Ret = GetComputerNameExW(NameType, NULL, &Size);
89     Error = GetLastError();
90     ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
91     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
92     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
93 
94     Size = 0;
95     Ret = GetComputerNameExA(NameType, NULL, &Size);
96     Error = GetLastError();
97     ok(Ret == FALSE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
98     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExA returned error %lu\n", NameType, Error);
99     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
100 
101     /* non-NULL buffer, zero size */
102     RtlFillMemory(BufferW, sizeof(BufferW), 0x55);
103     Size = 0;
104     Ret = GetComputerNameExW(NameType, BufferW, &Size);
105     Error = GetLastError();
106     ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
107     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
108     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
109     ok(BufferW[0] == 0x5555, "[%d] BufferW[0] = 0x%x\n", NameType, BufferW[0]);
110 
111     RtlFillMemory(BufferA, sizeof(BufferA), 0x55);
112     Size = 0;
113     Ret = GetComputerNameExA(NameType, BufferA, &Size);
114     Error = GetLastError();
115     ok(Ret == FALSE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
116     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExA returned error %lu\n", NameType, Error);
117     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
118     ok(BufferA[0] == 0x55, "[%d] BufferA[0] = 0x%x\n", NameType, BufferA[0]);
119 
120     /* non-NULL buffer, too small size */
121     RtlFillMemory(BufferW, sizeof(BufferW), 0x55);
122     Size = ReferenceLen;
123     Ret = GetComputerNameExW(NameType, BufferW, &Size);
124     Error = GetLastError();
125     ok(Ret == FALSE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
126     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExW returned error %lu\n", NameType, Error);
127     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
128     if (NameType != ComputerNameNetBIOS && NameType != ComputerNamePhysicalNetBIOS)
129     {
130         if (ReferenceLen == 0)
131         {
132             ok(BufferW[0] == 0x5555, "[%d] BufferW[0] = 0x%x\n",
133                NameType, BufferW[0]);
134         }
135         else
136         {
137             ok(BufferW[0] == 0, "[%d] BufferW[0] = 0x%x\n",
138                NameType, BufferW[0]);
139         }
140     }
141     ok(BufferW[1] == 0x5555, "[%d] BufferW[1] = 0x%x\n", NameType, BufferW[1]);
142 
143     RtlFillMemory(BufferA, sizeof(BufferA), 0x55);
144     Size = ReferenceLen;
145     Ret = GetComputerNameExA(NameType, BufferA, &Size);
146     Error = GetLastError();
147     ok(Ret == FALSE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
148     ok(Error == ERROR_MORE_DATA, "[%d] GetComputerNameExA returned error %lu\n", NameType, Error);
149     ok(Size == ReferenceLen + 1, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
150     ok(BufferA[0] == 0x55, "[%d] BufferA[0] = 0x%x\n", NameType, BufferA[0]);
151 
152     /* non-NULL buffer, exact size */
153     RtlFillMemory(BufferW, sizeof(BufferW), 0x55);
154     Size = ReferenceLen + 1;
155     Ret = GetComputerNameExW(NameType, BufferW, &Size);
156     ok(Ret == TRUE, "[%d] GetComputerNameExW returned %d\n", NameType, Ret);
157     ok(Size == ReferenceLen, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
158     ok(BufferW[ReferenceLen] == 0, "[%d] BufferW[ReferenceLen] = 0x%x\n", NameType, BufferW[ReferenceLen]);
159     ok(BufferW[ReferenceLen + 1] == 0x5555, "[%d] BufferW[ReferenceLen + 1] = 0x%x\n", NameType, BufferW[ReferenceLen + 1]);
160     ok(!wcscmp(BufferW, Reference), "[%d] '%ls' != '%ls'\n", NameType, BufferW, Reference);
161 
162     RtlFillMemory(BufferA, sizeof(BufferA), 0x55);
163     Size = ReferenceLen + 1;
164     Ret = GetComputerNameExA(NameType, BufferA, &Size);
165     ok(Ret == TRUE, "[%d] GetComputerNameExA returned %d\n", NameType, Ret);
166     ok(Size == ReferenceLen, "[%d] Got Size %lu, expected %lu\n", NameType, Size, ReferenceLen + 1);
167     ok(BufferA[ReferenceLen] == 0, "[%d] BufferA[ReferenceLen] = 0x%x\n", NameType, BufferA[ReferenceLen]);
168     ok(BufferA[ReferenceLen + 1] == 0x55, "[%d] BufferA[ReferenceLen + 1] = 0x%x\n", NameType, BufferA[ReferenceLen + 1]);
169     for (i = 0; i < ReferenceLen; i++)
170     {
171         if (BufferA[i] != Reference[i])
172         {
173             ok(0, "[%d] BufferA[%lu] = 0x%x, expected 0x%x\n", NameType, i, BufferA[i], Reference[i]);
174         }
175     }
176 }
177 
178 static
179 LSTATUS
180 ReadRegistryValue(PCHAR ValueName, PCHAR Value)
181 {
182     INT ErrorCode;
183     HKEY ParametersKey;
184     DWORD RegType;
185     DWORD RegSize = 0;
186 
187     /* Open the database path key */
188     ErrorCode = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
189                               "System\\CurrentControlSet\\Services\\Tcpip\\Parameters",
190                               0,
191                               KEY_READ,
192                               &ParametersKey);
193     if (ErrorCode == NO_ERROR)
194     {
195         /* Read the actual path */
196         ErrorCode = RegQueryValueExA(ParametersKey,
197                                      ValueName,
198                                      NULL,
199                                      &RegType,
200                                      NULL,
201                                      &RegSize);
202         if (RegSize)
203         {
204             /* Read the actual path */
205             ErrorCode = RegQueryValueExA(ParametersKey,
206                                          ValueName,
207                                          NULL,
208                                          &RegType,
209                                          (LPBYTE)Value,
210                                          &RegSize);
211         }
212 
213         /* Close the key */
214         RegCloseKey(ParametersKey);
215     }
216     return ErrorCode;
217 }
218 
219 static
220 LSTATUS
221 ReadRegistryComputerNameValue(PCHAR ValueName, PCHAR Value)
222 {
223     INT ErrorCode;
224     HKEY ParametersKey;
225     DWORD RegType;
226     DWORD RegSize = 0;
227 
228     /* Open the database path key */
229     ErrorCode = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
230                               "System\\CurrentControlSet\\Control\\ComputerName\\ActiveComputerName",
231                               0,
232                               KEY_READ,
233                               &ParametersKey);
234     if (ErrorCode == NO_ERROR)
235     {
236         /* Read the actual path */
237         ErrorCode = RegQueryValueExA(ParametersKey,
238                                      ValueName,
239                                      NULL,
240                                      &RegType,
241                                      NULL,
242                                      &RegSize);
243         if (RegSize)
244         {
245             /* Read the actual path */
246             ErrorCode = RegQueryValueExA(ParametersKey,
247                                          ValueName,
248                                          NULL,
249                                          &RegType,
250                                          (LPBYTE)Value,
251                                          &RegSize);
252         }
253 
254         /* Close the key */
255         RegCloseKey(ParametersKey);
256     }
257     return ErrorCode;
258 }
259 
260 static
261 LSTATUS
262 WriteRegistryValue(PCHAR ValueName, PCHAR Value)
263 {
264     INT ErrorCode;
265     HKEY ParametersKey;
266 
267     /* Open the database path key */
268     ErrorCode = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
269                               "System\\CurrentControlSet\\Services\\Tcpip\\Parameters",
270                               0,
271                               KEY_WRITE,
272                               &ParametersKey);
273     if (ErrorCode == NO_ERROR)
274     {
275         /* Read the actual path */
276         ErrorCode = RegSetValueExA(ParametersKey,
277                                    ValueName,
278                                    0,
279                                    REG_SZ,
280                                    (LPBYTE)Value,
281                                    lstrlenA(Value) + 1);
282 
283         /* Close the key */
284         RegCloseKey(ParametersKey);
285     }
286     return ErrorCode;
287 }
288 
289 static
290 LSTATUS
291 DeleteRegistryValue(PCHAR ValueName)
292 {
293     INT ErrorCode;
294     HKEY ParametersKey;
295 
296     /* Open the database path key */
297     ErrorCode = RegOpenKeyExA(HKEY_LOCAL_MACHINE,
298                               "System\\CurrentControlSet\\Services\\Tcpip\\Parameters",
299                               0,
300                               KEY_WRITE,
301                               &ParametersKey);
302     if (ErrorCode == NO_ERROR)
303     {
304         /* Read the actual path */
305         ErrorCode = RegDeleteValueA(ParametersKey, ValueName);
306 
307         /* Close the key */
308         RegCloseKey(ParametersKey);
309     }
310     return ErrorCode;
311 }
312 
313 /* If this test crashes it might end up with wrong host and/or domain name in registry! */
314 static
315 VOID
316 TestReturnValues()
317 {
318     CHAR OrigNetBIOS[128];
319     CHAR OrigHostname[128];
320     CHAR OrigDomainName[128];
321     CHAR OrigDhcpHostname[128];
322     CHAR OrigDhcpDomainName[128];
323     BOOL OrigNetBIOSExists;
324     BOOL OrigHostnameExists;
325     BOOL OrigDomainNameExists;
326     BOOL OrigDhcpHostnameExists;
327     BOOL OrigDhcpDomainNameExists;
328     CHAR ComputerName[128];
329     DWORD ComputerNameSize = 0;
330     INT ErrorCode;
331 
332     memset(OrigNetBIOS, 0, sizeof(OrigNetBIOS));
333     memset(OrigHostname, 0, sizeof(OrigHostname));
334     memset(OrigDomainName, 0, sizeof(OrigDomainName));
335     memset(OrigDhcpHostname, 0, sizeof(OrigDhcpHostname));
336     memset(OrigDhcpDomainName, 0, sizeof(OrigDhcpDomainName));
337     /* read current registry values */
338     ErrorCode = ReadRegistryComputerNameValue("ComputerName", OrigNetBIOS);
339     ok(ErrorCode == ERROR_SUCCESS, "Failed to read registry key ComputerName %d\n", ErrorCode);
340     OrigNetBIOSExists = ErrorCode == STATUS_SUCCESS;
341     ErrorCode = ReadRegistryValue("Hostname", OrigHostname);
342     ok(ErrorCode == ERROR_SUCCESS, "Failed to read registry key Hostname %d\n", ErrorCode);
343     OrigHostnameExists = ErrorCode == STATUS_SUCCESS;
344     ErrorCode = ReadRegistryValue("Domain", OrigDomainName);
345     ok(ErrorCode == ERROR_SUCCESS || ErrorCode == ERROR_FILE_NOT_FOUND, "Failed to read registry key DomainName %d\n", ErrorCode);
346     OrigDomainNameExists = ErrorCode == STATUS_SUCCESS;
347     ErrorCode = ReadRegistryValue("DhcpHostname", OrigDhcpHostname);
348     ok(ErrorCode == ERROR_SUCCESS || ErrorCode == ERROR_FILE_NOT_FOUND, "Failed to read registry key DhcpHostname %d\n", ErrorCode);
349     OrigDhcpHostnameExists = ErrorCode == STATUS_SUCCESS;
350     ErrorCode = ReadRegistryValue("DhcpDomain", OrigDhcpDomainName);
351     ok(ErrorCode == ERROR_SUCCESS || ErrorCode == ERROR_FILE_NOT_FOUND, "Failed to read registry key DhcpDomainName %d\n", ErrorCode);
352     OrigDhcpDomainNameExists = ErrorCode == STATUS_SUCCESS;
353 
354     trace("Starting values:\n");
355     trace("NetBIOS: %s, exists %s\n", OrigNetBIOS, OrigNetBIOSExists ? "yes" : "no");
356     trace("Hostname: %s, exists %s\n", OrigHostname, OrigHostnameExists ? "yes" : "no");
357     trace("Domain: %s, exists %s\n", OrigDomainName, OrigDomainNameExists ? "yes" : "no");
358     trace("DhcpHostname: %s, exists %s\n", OrigDhcpHostnameExists ? OrigDhcpHostname : "", OrigDhcpHostnameExists ? "yes" : "no");
359     trace("DhcpDomain: %s, exists %s\n", OrigDhcpDomainNameExists ? OrigDhcpDomainName : "", OrigDhcpDomainNameExists ? "yes" : "no");
360 
361     /* ComputerNamePhysicalNetBIOS */
362     ComputerNameSize = 0;
363     StartSeh()
364         GetComputerNameExA(ComputerNamePhysicalNetBIOS, ComputerName, &ComputerNameSize);
365         if (ComputerNameSize)
366             ok(GetComputerNameExA(ComputerNamePhysicalNetBIOS, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalNetBIOS) failed with %ld\n", GetLastError());
367         else
368             memset(ComputerName, 0, sizeof(ComputerName));
369         ok(strcmp(ComputerName, OrigNetBIOS) == 0, "ComputerNamePhysicalNetBIOS doesn't match registry value '%s' != '%s'\n", ComputerName, OrigNetBIOS);
370     EndSeh(STATUS_SUCCESS);
371 
372     /* ComputerNamePhysicalDnsHostname */
373     ComputerNameSize = 0;
374     StartSeh()
375         GetComputerNameExA(ComputerNamePhysicalDnsHostname, ComputerName, &ComputerNameSize);
376         if (ComputerNameSize)
377             ok(GetComputerNameExA(ComputerNamePhysicalDnsHostname, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalDnsHostname) failed with %ld\n", GetLastError());
378         else
379             memset(ComputerName, 0, sizeof(ComputerName));
380         ok(strcmp(ComputerName, OrigHostname) == 0, "ComputerNamePhysicalDnsHostname doesn't match registry value '%s' != '%s'\n", ComputerName, OrigHostname);
381     EndSeh(STATUS_SUCCESS);
382 
383     /* ComputerNamePhysicalDnsDomain */
384     ComputerNameSize = 0;
385     StartSeh()
386         GetComputerNameExA(ComputerNamePhysicalDnsDomain, ComputerName, &ComputerNameSize);
387         if (ComputerNameSize)
388             ok(GetComputerNameExA(ComputerNamePhysicalDnsDomain, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalDnsDomain) failed with %ld\n", GetLastError());
389         else
390             memset(ComputerName, 0, sizeof(ComputerName));
391         ok(strcmp(ComputerName, OrigDomainName) == 0, "ComputerNamePhysicalDnsDomain doesn't match registry value '%s' != '%s'\n", ComputerName, OrigDomainName);
392     EndSeh(STATUS_SUCCESS);
393     ComputerNameSize = 0;
394 
395     /* ComputerNameNetBIOS */
396     StartSeh()
397         GetComputerNameExA(ComputerNameNetBIOS, ComputerName, &ComputerNameSize);
398         if (ComputerNameSize)
399             ok(GetComputerNameExA(ComputerNameNetBIOS, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameNetBIOS) failed with %ld\n", GetLastError());
400         else
401             memset(ComputerName, 0, sizeof(ComputerName));
402         ok(strcmp(ComputerName, OrigNetBIOS) == 0, "ComputerNameNetBIOS doesn't match registry value '%s' != '%s'\n", ComputerName, OrigNetBIOS);
403     EndSeh(STATUS_SUCCESS);
404 
405     /* ComputerNameDnsHostname */
406     ComputerNameSize = 0;
407     StartSeh()
408         GetComputerNameExA(ComputerNameDnsHostname, ComputerName, &ComputerNameSize);
409         if (ComputerNameSize)
410             ok(GetComputerNameExA(ComputerNameDnsHostname, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameDnsHostname) failed with %ld\n", GetLastError());
411         else
412             memset(ComputerName, 0, sizeof(ComputerName));
413         ok(strcmp(ComputerName, OrigHostname) == 0, "ComputerNameDnsHostname doesn't match registry value '%s' != '%s'\n", ComputerName, OrigHostname);
414     EndSeh(STATUS_SUCCESS);
415 
416     /* ComputerNameDnsDomain */
417     ComputerNameSize = 0;
418     StartSeh()
419         GetComputerNameExA(ComputerNameDnsDomain, ComputerName, &ComputerNameSize);
420         if (ComputerNameSize)
421            ok(GetComputerNameExA(ComputerNameDnsDomain, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameDnsDomain) failed with %ld\n", GetLastError());
422         else
423             memset(ComputerName, 0, sizeof(ComputerName));
424         ok(strcmp(ComputerName, OrigDomainName) == 0, "ComputerNameDnsDomain doesn't match registry value '%s' != '%s'\n", ComputerName, OrigDomainName);
425     EndSeh(STATUS_SUCCESS);
426 
427     ErrorCode = WriteRegistryValue("DhcpHostname", "testdhcproshost");
428     ok(ErrorCode == ERROR_SUCCESS, "Failed to write registry key DhcpHostname %d\n", ErrorCode);
429     ErrorCode = WriteRegistryValue("DhcpDomain", "testrosdomain");
430     ok(ErrorCode == ERROR_SUCCESS, "Failed to write registry key DhcpDomainName %d\n", ErrorCode);
431 
432     /* ComputerNamePhysicalNetBIOS */
433     ComputerNameSize = 0;
434     StartSeh()
435         GetComputerNameExA(ComputerNamePhysicalNetBIOS, ComputerName, &ComputerNameSize);
436         if (ComputerNameSize)
437             ok(GetComputerNameExA(ComputerNamePhysicalNetBIOS, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalNetBIOS) failed with %ld\n", GetLastError());
438         else
439             memset(ComputerName, 0, sizeof(ComputerName));
440         ok(strcmp(ComputerName, OrigNetBIOS) == 0, "ComputerNamePhysicalNetBIOS doesn't match registry value '%s' != '%s'\n", ComputerName, OrigNetBIOS);
441     EndSeh(STATUS_SUCCESS);
442 
443     /* ComputerNamePhysicalDnsHostname */
444     ComputerNameSize = 0;
445     StartSeh()
446         GetComputerNameExA(ComputerNamePhysicalDnsHostname, ComputerName, &ComputerNameSize);
447         if (ComputerNameSize)
448             ok(GetComputerNameExA(ComputerNamePhysicalDnsHostname, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalDnsHostname) failed with %ld\n", GetLastError());
449         else
450             memset(ComputerName, 0, sizeof(ComputerName));
451         ok(strcmp(ComputerName, OrigHostname) == 0, "ComputerNamePhysicalDnsHostname doesn't match registry value '%s' != '%s'\n", ComputerName, OrigHostname);
452     EndSeh(STATUS_SUCCESS);
453 
454     /* ComputerNamePhysicalDnsDomain */
455     ComputerNameSize = 0;
456     StartSeh()
457         GetComputerNameExA(ComputerNamePhysicalDnsDomain, ComputerName, &ComputerNameSize);
458         if (ComputerNameSize)
459             ok(GetComputerNameExA(ComputerNamePhysicalDnsDomain, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNamePhysicalDnsDomain) failed with %ld\n", GetLastError());
460         else
461             memset(ComputerName, 0, sizeof(ComputerName));
462         ok(strcmp(ComputerName, OrigDomainName) == 0, "ComputerNamePhysicalDnsDomain doesn't match registry value '%s' != '%s'\n", ComputerName, OrigDomainName);
463     EndSeh(STATUS_SUCCESS);
464     ComputerNameSize = 0;
465 
466     /* ComputerNameNetBIOS */
467     StartSeh()
468         GetComputerNameExA(ComputerNameNetBIOS, ComputerName, &ComputerNameSize);
469         if (ComputerNameSize)
470             ok(GetComputerNameExA(ComputerNameNetBIOS, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameNetBIOS) failed with %ld\n", GetLastError());
471         else
472             memset(ComputerName, 0, sizeof(ComputerName));
473         ok(strcmp(ComputerName, OrigNetBIOS) == 0, "ComputerNameNetBIOS doesn't match registry value '%s' != '%s'\n", ComputerName, OrigNetBIOS);
474     EndSeh(STATUS_SUCCESS);
475 
476     /* ComputerNameDnsHostname */
477     ComputerNameSize = 0;
478     StartSeh()
479         GetComputerNameExA(ComputerNameDnsHostname, ComputerName, &ComputerNameSize);
480         if (ComputerNameSize)
481             ok(GetComputerNameExA(ComputerNameDnsHostname, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameDnsHostname) failed with %ld\n", GetLastError());
482         else
483             memset(ComputerName, 0, sizeof(ComputerName));
484         ok(strcmp(ComputerName, OrigHostname) == 0, "ComputerNameDnsHostname doesn't match registry value '%s' != '%s'\n", ComputerName, OrigHostname);
485     EndSeh(STATUS_SUCCESS);
486 
487     /* ComputerNameDnsDomain */
488     ComputerNameSize = 0;
489     StartSeh()
490         GetComputerNameExA(ComputerNameDnsDomain, ComputerName, &ComputerNameSize);
491         if (ComputerNameSize)
492             ok(GetComputerNameExA(ComputerNameDnsDomain, ComputerName, &ComputerNameSize), "GetComputerNameExA(ComputerNameDnsDomain) failed with %ld\n", GetLastError());
493         else
494             memset(ComputerName, 0, sizeof(ComputerName));
495         ok(strcmp(ComputerName, OrigDomainName) == 0, "ComputerNameDnsDomain doesn't match registry value '%s' != '%s'\n", ComputerName, OrigDomainName);
496     EndSeh(STATUS_SUCCESS);
497 
498     /* restore registry values */
499     if (OrigDhcpHostnameExists)
500         ErrorCode = WriteRegistryValue("DhcpHostname", OrigDhcpHostname);
501     else
502         ErrorCode = DeleteRegistryValue("DhcpHostname");
503     ok(ErrorCode == ERROR_SUCCESS, "Failed to restore registry key DhcpHostname %d\n", ErrorCode);
504     if (OrigDhcpDomainNameExists)
505         ErrorCode = WriteRegistryValue("DhcpDomain", OrigDhcpDomainName);
506     else
507         ErrorCode = DeleteRegistryValue("DhcpDomain");
508     ok(ErrorCode == ERROR_SUCCESS, "Failed to restore registry key DhcpDomainName %d\n", ErrorCode);
509 }
510 
511 START_TEST(GetComputerNameEx)
512 {
513     TestGetComputerNameEx(ComputerNameNetBIOS);
514     TestGetComputerNameEx(ComputerNameDnsHostname);
515     TestGetComputerNameEx(ComputerNameDnsDomain);
516     //TestGetComputerNameEx(ComputerNameDnsFullyQualified);
517     TestGetComputerNameEx(ComputerNamePhysicalNetBIOS);
518     TestGetComputerNameEx(ComputerNamePhysicalDnsHostname);
519     TestGetComputerNameEx(ComputerNamePhysicalDnsDomain);
520     //TestGetComputerNameEx(ComputerNamePhysicalDnsFullyQualified);
521     TestReturnValues();
522 }
523