xref: /reactos/dll/win32/netapi32/wksta.c (revision 45b008dd)
1 /* Copyright 2002 Andriy Palamarchuk
2  * Copyright (c) 2003 Juan Lang
3  *
4  * netapi32 user 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 "netapi32.h"
22 
23 #include <lmwksta.h>
24 #include <lmjoin.h>
25 
26 WINE_DEFAULT_DEBUG_CHANNEL(netapi32);
27 
28 /************************************************************
29  *                NETAPI_IsLocalComputer
30  *
31  * Checks whether the server name indicates local machine.
32  */
NETAPI_IsLocalComputer(LMCSTR name)33 DECLSPEC_HIDDEN BOOL NETAPI_IsLocalComputer( LMCSTR name )
34 {
35     WCHAR buf[MAX_COMPUTERNAME_LENGTH + 1];
36     DWORD size = sizeof(buf) / sizeof(buf[0]);
37     BOOL ret;
38 
39     if (!name || !name[0]) return TRUE;
40 
41     ret = GetComputerNameW( buf,  &size );
42     if (ret && name[0] == '\\' && name[1] == '\\') name += 2;
43     return ret && !strcmpiW( name, buf );
44 }
45 
wprint_mac(WCHAR * buffer,int len,const MIB_IFROW * ifRow)46 static void wprint_mac(WCHAR* buffer, int len, const MIB_IFROW *ifRow)
47 {
48     int i;
49     unsigned char val;
50 
51     if (!buffer)
52         return;
53     if (len < 1)
54         return;
55     if (!ifRow)
56     {
57         *buffer = '\0';
58         return;
59     }
60 
61     for (i = 0; i < ifRow->dwPhysAddrLen && 2 * i < len; i++)
62     {
63         val = ifRow->bPhysAddr[i];
64         if ((val >>4) >9)
65             buffer[2*i] = (WCHAR)((val >>4) + 'A' - 10);
66         else
67             buffer[2*i] = (WCHAR)((val >>4) + '0');
68         if ((val & 0xf ) >9)
69             buffer[2*i+1] = (WCHAR)((val & 0xf) + 'A' - 10);
70         else
71             buffer[2*i+1] = (WCHAR)((val & 0xf) + '0');
72     }
73     buffer[2*i]=0;
74 }
75 
76 /* Theoretically this could be too short, except that MS defines
77  * MAX_ADAPTER_NAME as 128, and MAX_INTERFACE_NAME_LEN as 256, and both
78  * represent a count of WCHARs, so even with an extraordinarily long header
79  * this will be plenty
80  */
81 #define MAX_TRANSPORT_NAME MAX_INTERFACE_NAME_LEN
82 #define MAX_TRANSPORT_ADDR 13
83 
84 #define NBT_TRANSPORT_NAME_HEADER "\\Device\\NetBT_Tcpip_"
85 #define UNKNOWN_TRANSPORT_NAME_HEADER "\\Device\\UnknownTransport_"
86 
wprint_name(WCHAR * buffer,int len,ULONG transport,PMIB_IFROW ifRow)87 static void wprint_name(WCHAR *buffer, int len, ULONG transport,
88  PMIB_IFROW ifRow)
89 {
90     WCHAR *ptr1, *ptr2;
91     const char *name;
92 
93     if (!buffer)
94         return;
95     if (!ifRow)
96     {
97         *buffer = '\0';
98         return;
99     }
100 
101     if (!memcmp(&transport, TRANSPORT_NBT, sizeof(ULONG)))
102         name = NBT_TRANSPORT_NAME_HEADER;
103     else
104         name = UNKNOWN_TRANSPORT_NAME_HEADER;
105 
106     for (ptr1 = buffer; *name && ptr1 < buffer + len; ptr1++, name++)
107         *ptr1 = *name;
108     for (ptr2 = ifRow->wszName; *ptr2 && ptr1 < buffer + len; ptr1++, ptr2++)
109         *ptr1 = *ptr2;
110     *ptr1 = '\0';
111 }
112 
113 /***********************************************************************
114  *                NetWkstaTransportEnum  (NETAPI32.@)
115  */
116 
117 struct WkstaTransportEnumData
118 {
119     UCHAR          n_adapt;
120     UCHAR          n_read;
121     DWORD          prefmaxlen;
122     LPBYTE        *pbuf;
123     NET_API_STATUS ret;
124 };
125 
126 /**********************************************************************/
127 
WkstaEnumAdaptersCallback(UCHAR totalLANAs,UCHAR lanaIndex,ULONG transport,const NetBIOSAdapterImpl * data,void * closure)128 static BOOL WkstaEnumAdaptersCallback(UCHAR totalLANAs, UCHAR lanaIndex,
129  ULONG transport, const NetBIOSAdapterImpl *data, void *closure)
130 {
131     BOOL ret;
132     struct WkstaTransportEnumData *enumData = closure;
133 
134     if (enumData && enumData->pbuf)
135     {
136         if (lanaIndex == 0)
137         {
138             DWORD toAllocate;
139 
140             enumData->n_adapt = totalLANAs;
141             enumData->n_read = 0;
142 
143             toAllocate = totalLANAs * (sizeof(WKSTA_TRANSPORT_INFO_0)
144              + MAX_TRANSPORT_NAME * sizeof(WCHAR) +
145              MAX_TRANSPORT_ADDR * sizeof(WCHAR));
146             if (enumData->prefmaxlen != MAX_PREFERRED_LENGTH)
147                 toAllocate = enumData->prefmaxlen;
148             NetApiBufferAllocate(toAllocate, (LPVOID *)enumData->pbuf);
149         }
150         if (*(enumData->pbuf))
151         {
152             UCHAR spaceFor;
153 
154             if (enumData->prefmaxlen == MAX_PREFERRED_LENGTH)
155                 spaceFor = totalLANAs;
156             else
157                 spaceFor = enumData->prefmaxlen /
158                  (sizeof(WKSTA_TRANSPORT_INFO_0) + (MAX_TRANSPORT_NAME +
159                  MAX_TRANSPORT_ADDR) * sizeof(WCHAR));
160             if (enumData->n_read < spaceFor)
161             {
162                 PWKSTA_TRANSPORT_INFO_0 ti;
163                 LMSTR transport_name, transport_addr;
164                 MIB_IFROW ifRow;
165 
166                 ti = (PWKSTA_TRANSPORT_INFO_0)(*(enumData->pbuf) +
167                  enumData->n_read * sizeof(WKSTA_TRANSPORT_INFO_0));
168                 transport_name = (LMSTR)(*(enumData->pbuf) +
169                  totalLANAs * sizeof(WKSTA_TRANSPORT_INFO_0) +
170                  enumData->n_read * MAX_TRANSPORT_NAME * sizeof(WCHAR));
171                 transport_addr = (LMSTR)(*(enumData->pbuf) +
172                  totalLANAs * (sizeof(WKSTA_TRANSPORT_INFO_0) +
173                  MAX_TRANSPORT_NAME * sizeof(WCHAR)) +
174                  enumData->n_read * MAX_TRANSPORT_ADDR * sizeof(WCHAR));
175 
176                 ifRow.dwIndex = data->ifIndex;
177                 GetIfEntry(&ifRow);
178                 ti->wkti0_quality_of_service = 0;
179                 ti->wkti0_number_of_vcs = 0;
180                 ti->wkti0_transport_name = transport_name;
181                 wprint_name(ti->wkti0_transport_name, MAX_TRANSPORT_NAME,
182                  transport, &ifRow);
183                 ti->wkti0_transport_address = transport_addr;
184                 wprint_mac(ti->wkti0_transport_address, MAX_TRANSPORT_ADDR,
185                  &ifRow);
186                 if (!memcmp(&transport, TRANSPORT_NBT, sizeof(ULONG)))
187                     ti->wkti0_wan_ish = TRUE;
188                 else
189                     ti->wkti0_wan_ish = FALSE;
190                 TRACE("%d of %d:ti at %p\n", lanaIndex, totalLANAs, ti);
191                 TRACE("transport_name at %p %s\n",
192                  ti->wkti0_transport_name,
193                  debugstr_w(ti->wkti0_transport_name));
194                 TRACE("transport_address at %p %s\n",
195                  ti->wkti0_transport_address,
196                  debugstr_w(ti->wkti0_transport_address));
197                 enumData->n_read++;
198                 enumData->ret = NERR_Success;
199                 ret = TRUE;
200             }
201             else
202             {
203                 enumData->ret = ERROR_MORE_DATA;
204                 ret = FALSE;
205             }
206         }
207         else
208         {
209             enumData->ret = ERROR_OUTOFMEMORY;
210             ret = FALSE;
211         }
212     }
213     else
214         ret = FALSE;
215     return ret;
216 }
217 
218 /**********************************************************************/
219 
220 NET_API_STATUS WINAPI
NetWkstaTransportEnum(LMSTR ServerName,DWORD level,PBYTE * pbuf,DWORD prefmaxlen,LPDWORD read_entries,PDWORD total_entries,PDWORD hresume)221 NetWkstaTransportEnum(LMSTR ServerName, DWORD level, PBYTE* pbuf,
222       DWORD prefmaxlen, LPDWORD read_entries,
223       PDWORD total_entries, PDWORD hresume)
224 {
225     NET_API_STATUS ret;
226 
227     TRACE(":%s, 0x%08x, %p, 0x%08x, %p, %p, %p\n", debugstr_w(ServerName),
228      level, pbuf, prefmaxlen, read_entries, total_entries,hresume);
229     if (!NETAPI_IsLocalComputer(ServerName))
230     {
231         FIXME(":not implemented for non-local computers\n");
232         ret = ERROR_INVALID_LEVEL;
233     }
234     else
235     {
236         if (hresume && *hresume)
237         {
238           FIXME(":resume handle not implemented\n");
239           return ERROR_INVALID_LEVEL;
240         }
241 
242         switch (level)
243         {
244             case 0: /* transport info */
245             {
246                 ULONG allTransports;
247                 struct WkstaTransportEnumData enumData;
248 
249                 if (NetBIOSNumAdapters() == 0)
250                   return ERROR_NETWORK_UNREACHABLE;
251                 if (!read_entries)
252                   return STATUS_ACCESS_VIOLATION;
253                 if (!total_entries || !pbuf)
254                   return RPC_X_NULL_REF_POINTER;
255 
256                 enumData.prefmaxlen = prefmaxlen;
257                 enumData.pbuf = pbuf;
258                 memcpy(&allTransports, ALL_TRANSPORTS, sizeof(ULONG));
259                 NetBIOSEnumAdapters(allTransports, WkstaEnumAdaptersCallback,
260                  &enumData);
261                 *read_entries = enumData.n_read;
262                 *total_entries = enumData.n_adapt;
263                 if (hresume) *hresume= 0;
264                 ret = enumData.ret;
265                 break;
266             }
267             default:
268                 TRACE("Invalid level %d is specified\n", level);
269                 ret = ERROR_INVALID_LEVEL;
270         }
271     }
272     return ret;
273 }
274 
275 
276 /************************************************************
277  *                NetpGetComputerName  (NETAPI32.@)
278  */
NetpGetComputerName(LPWSTR * Buffer)279 NET_API_STATUS WINAPI NetpGetComputerName(LPWSTR *Buffer)
280 {
281     DWORD dwSize = MAX_COMPUTERNAME_LENGTH + 1;
282 
283     TRACE("(%p)\n", Buffer);
284     NetApiBufferAllocate(dwSize * sizeof(WCHAR), (LPVOID *) Buffer);
285     if (GetComputerNameW(*Buffer,  &dwSize))
286     {
287         return NetApiBufferReallocate(
288             *Buffer, (dwSize + 1) * sizeof(WCHAR),
289             (LPVOID *) Buffer);
290     }
291     else
292     {
293         NetApiBufferFree(*Buffer);
294         return ERROR_NOT_ENOUGH_MEMORY;
295     }
296 }
297 
I_NetNameCompare(LPVOID p1,LPWSTR wkgrp,LPWSTR comp,LPVOID p4,LPVOID p5)298 NET_API_STATUS WINAPI I_NetNameCompare(LPVOID p1, LPWSTR wkgrp, LPWSTR comp,
299  LPVOID p4, LPVOID p5)
300 {
301     FIXME("(%p %s %s %p %p): stub\n", p1, debugstr_w(wkgrp), debugstr_w(comp),
302      p4, p5);
303     return ERROR_INVALID_PARAMETER;
304 }
305 
I_NetNameValidate(LPVOID p1,LPWSTR wkgrp,LPVOID p3,LPVOID p4)306 NET_API_STATUS WINAPI I_NetNameValidate(LPVOID p1, LPWSTR wkgrp, LPVOID p3,
307  LPVOID p4)
308 {
309     FIXME("(%p %s %p %p): stub\n", p1, debugstr_w(wkgrp), p3, p4);
310     return ERROR_INVALID_PARAMETER;
311 }
312