e0c16d35ba94468a6fcb2d5c0f10a3251fb4bd1d
[reactos.git] / rostests / winetests / netapi32 / wksta.c
1 /*
2 * Copyright 2002 Andriy Palamarchuk
3 *
4 * Conformance test of the workstation functions.
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include <stdarg.h>
22
23 #include "wine/test.h"
24 #include "windef.h"
25 #include "winbase.h"
26 #include "wingdi.h"
27 #include "winnls.h"
28 #include "winresrc.h" /* Ensure we use Unicode defns with native headers */
29 #include "nb30.h"
30 #include "lmcons.h"
31 #include "lmerr.h"
32 #include "lmwksta.h"
33 #include "lmapibuf.h"
34
35 static NET_API_STATUS (WINAPI *pNetApiBufferFree)(LPVOID)=NULL;
36 static NET_API_STATUS (WINAPI *pNetApiBufferSize)(LPVOID,LPDWORD)=NULL;
37 static NET_API_STATUS (WINAPI *pNetpGetComputerName)(LPWSTR*)=NULL;
38 static NET_API_STATUS (WINAPI *pNetWkstaUserGetInfo)(LPWSTR,DWORD,PBYTE*)=NULL;
39 static NET_API_STATUS (WINAPI *pNetWkstaTransportEnum)(LPWSTR,DWORD,LPBYTE*,
40 DWORD,LPDWORD,LPDWORD,LPDWORD)=NULL;
41
42 WCHAR user_name[UNLEN + 1];
43 WCHAR computer_name[MAX_COMPUTERNAME_LENGTH + 1];
44
45 static int init_wksta_tests(void)
46 {
47 DWORD dwSize;
48 BOOL rc;
49
50 user_name[0] = 0;
51 dwSize = sizeof(user_name)/sizeof(user_name[0]);
52 rc=GetUserNameW(user_name, &dwSize);
53 if (rc==FALSE && GetLastError()==ERROR_CALL_NOT_IMPLEMENTED) {
54 win_skip("GetUserNameW is not implemented\n");
55 return 0;
56 }
57 ok(rc, "User Name Retrieved\n");
58
59 computer_name[0] = 0;
60 dwSize = sizeof(computer_name)/sizeof(computer_name[0]);
61 ok(GetComputerNameW(computer_name, &dwSize), "Computer Name Retrieved\n");
62 return 1;
63 }
64
65 static void run_get_comp_name_tests(void)
66 {
67 LPWSTR ws = NULL;
68
69 ok(pNetpGetComputerName(&ws) == NERR_Success, "Computer name is retrieved\n");
70 ok(!lstrcmpW(computer_name, ws), "This is really computer name\n");
71 pNetApiBufferFree(ws);
72 }
73
74 static void run_wkstausergetinfo_tests(void)
75 {
76 LPWKSTA_USER_INFO_0 ui0 = NULL;
77 LPWKSTA_USER_INFO_1 ui1 = NULL;
78 LPWKSTA_USER_INFO_1101 ui1101 = NULL;
79 DWORD dwSize;
80
81 /* Level 0 */
82 ok(pNetWkstaUserGetInfo(NULL, 0, (LPBYTE *)&ui0) == NERR_Success,
83 "NetWkstaUserGetInfo is unsuccessful\n");
84
85 ok(ui0 != NULL, "ui0 is NULL\n");
86 /* This failure occurred when I ran sshd as service and didn't authenticate
87 * Since the test dereferences ui0, the rest of this test is worthless
88 */
89 if (!ui0)
90 {
91 return;
92 }
93
94 ok(!lstrcmpW(user_name, ui0->wkui0_username), "This is really user name\n");
95 pNetApiBufferSize(ui0, &dwSize);
96 ok(dwSize >= (sizeof(WKSTA_USER_INFO_0) +
97 lstrlenW(ui0->wkui0_username) * sizeof(WCHAR)),
98 "Is allocated with NetApiBufferAllocate\n");
99
100 /* Level 1 */
101 ok(pNetWkstaUserGetInfo(NULL, 1, (LPBYTE *)&ui1) == NERR_Success,
102 "NetWkstaUserGetInfo is successful\n");
103 ok(lstrcmpW(ui1->wkui1_username, ui0->wkui0_username) == 0,
104 "the same name as returned for level 0\n");
105 pNetApiBufferSize(ui1, &dwSize);
106 ok(dwSize >= (sizeof(WKSTA_USER_INFO_1) +
107 (lstrlenW(ui1->wkui1_username) +
108 lstrlenW(ui1->wkui1_logon_domain) +
109 lstrlenW(ui1->wkui1_oth_domains) +
110 lstrlenW(ui1->wkui1_logon_server)) * sizeof(WCHAR)),
111 "Is allocated with NetApiBufferAllocate\n");
112
113 /* Level 1101 */
114 ok(pNetWkstaUserGetInfo(NULL, 1101, (LPBYTE *)&ui1101) == NERR_Success,
115 "NetWkstaUserGetInfo is successful\n");
116 ok(lstrcmpW(ui1101->wkui1101_oth_domains, ui1->wkui1_oth_domains) == 0,
117 "the same oth_domains as returned for level 1\n");
118 pNetApiBufferSize(ui1101, &dwSize);
119 ok(dwSize >= (sizeof(WKSTA_USER_INFO_1101) +
120 lstrlenW(ui1101->wkui1101_oth_domains) * sizeof(WCHAR)),
121 "Is allocated with NetApiBufferAllocate\n");
122
123 pNetApiBufferFree(ui0);
124 pNetApiBufferFree(ui1);
125 pNetApiBufferFree(ui1101);
126
127 /* errors handling */
128 ok(pNetWkstaUserGetInfo(NULL, 10000, (LPBYTE *)&ui0) == ERROR_INVALID_LEVEL,
129 "Invalid level\n");
130 }
131
132 static void run_wkstatransportenum_tests(void)
133 {
134 LPBYTE bufPtr;
135 NET_API_STATUS apiReturn;
136 DWORD entriesRead, totalEntries;
137
138 /* 1st check: is param 2 (level) correct? (only if param 5 passed?) */
139 apiReturn = pNetWkstaTransportEnum(NULL, 1, NULL, MAX_PREFERRED_LENGTH,
140 NULL, &totalEntries, NULL);
141 ok(apiReturn == ERROR_INVALID_LEVEL || apiReturn == ERROR_INVALID_PARAMETER,
142 "NetWkstaTransportEnum returned %d\n", apiReturn);
143
144 /* 2nd check: is param 5 passed? (only if level passes?) */
145 apiReturn = pNetWkstaTransportEnum(NULL, 0, NULL, MAX_PREFERRED_LENGTH,
146 NULL, &totalEntries, NULL);
147
148 /* if no network adapter present, bail, the rest of the test will fail */
149 if (apiReturn == ERROR_NETWORK_UNREACHABLE)
150 return;
151
152 ok(apiReturn == STATUS_ACCESS_VIOLATION || apiReturn == ERROR_INVALID_PARAMETER,
153 "NetWkstaTransportEnum returned %d\n", apiReturn);
154
155 /* 3rd check: is param 3 passed? */
156 apiReturn = pNetWkstaTransportEnum(NULL, 0, NULL, MAX_PREFERRED_LENGTH,
157 NULL, NULL, NULL);
158 ok(apiReturn == STATUS_ACCESS_VIOLATION || apiReturn == RPC_X_NULL_REF_POINTER || apiReturn == ERROR_INVALID_PARAMETER,
159 "NetWkstaTransportEnum returned %d\n", apiReturn);
160
161 /* 4th check: is param 6 passed? */
162 apiReturn = pNetWkstaTransportEnum(NULL, 0, &bufPtr, MAX_PREFERRED_LENGTH,
163 &entriesRead, NULL, NULL);
164 ok(apiReturn == RPC_X_NULL_REF_POINTER, "null pointer\n");
165
166 /* final check: valid return, actually get data back */
167 apiReturn = pNetWkstaTransportEnum(NULL, 0, &bufPtr, MAX_PREFERRED_LENGTH,
168 &entriesRead, &totalEntries, NULL);
169 ok(apiReturn == NERR_Success || apiReturn == ERROR_NETWORK_UNREACHABLE,
170 "NetWkstaTransportEnum returned %d\n", apiReturn);
171 if (apiReturn == NERR_Success) {
172 /* WKSTA_TRANSPORT_INFO_0 *transports = (WKSTA_TRANSPORT_INFO_0 *)bufPtr; */
173
174 ok(bufPtr != NULL, "got data back\n");
175 ok(entriesRead > 0, "read at least one transport\n");
176 ok(totalEntries > 0, "at least one transport\n");
177 pNetApiBufferFree(bufPtr);
178 }
179 }
180
181 START_TEST(wksta)
182 {
183 HMODULE hnetapi32=LoadLibraryA("netapi32.dll");
184
185 pNetApiBufferFree=(void*)GetProcAddress(hnetapi32,"NetApiBufferFree");
186 pNetApiBufferSize=(void*)GetProcAddress(hnetapi32,"NetApiBufferSize");
187 pNetpGetComputerName=(void*)GetProcAddress(hnetapi32,"NetpGetComputerName");
188 pNetWkstaUserGetInfo=(void*)GetProcAddress(hnetapi32,"NetWkstaUserGetInfo");
189 pNetWkstaTransportEnum=(void*)GetProcAddress(hnetapi32,"NetWkstaTransportEnum");
190
191 /* These functions were introduced with NT. It's safe to assume that
192 * if one is not available, none are.
193 */
194 if (!pNetApiBufferFree) {
195 win_skip("Needed functions are not available\n");
196 FreeLibrary(hnetapi32);
197 return;
198 }
199
200 if (init_wksta_tests()) {
201 run_get_comp_name_tests();
202 run_wkstausergetinfo_tests();
203 run_wkstatransportenum_tests();
204 }
205
206 FreeLibrary(hnetapi32);
207 }