c9d8a4520d8e5a2123bc484fec574e515f3d0a68
[reactos.git] / reactos / dll / win32 / netcfgx / netcfgx.c
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Configuration of network devices
4 * FILE: dll/win32/netcfgx/netcfgx.c
5 * PURPOSE: Network devices installer
6 *
7 * PROGRAMMERS: Hervé Poussineau (hpoussin@reactos.org)
8 */
9
10
11 #include "precomp.h"
12 #include <initguid.h>
13 #include <devguid.h>
14 #define NDEBUG
15 #include <debug.h>
16
17 HINSTANCE netcfgx_hInstance;
18 const GUID CLSID_TcpipConfigNotifyObject = {0xA907657F, 0x6FDF, 0x11D0, {0x8E, 0xFB, 0x00, 0xC0, 0x4F, 0xD9, 0x12, 0xB2}};
19
20
21
22
23 static INTERFACE_TABLE InterfaceTable[] =
24 {
25 {
26 &CLSID_CNetCfg,
27 INetCfg_Constructor
28 },
29 {
30 &CLSID_TcpipConfigNotifyObject,
31 TcpipConfigNotify_Constructor
32 },
33 {
34 NULL,
35 NULL
36 }
37 };
38
39 BOOL
40 WINAPI
41 DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID fImpLoad)
42 {
43 switch (fdwReason)
44 {
45 case DLL_PROCESS_ATTACH:
46 netcfgx_hInstance = hinstDLL;
47 DisableThreadLibraryCalls(netcfgx_hInstance);
48 break;
49 default:
50 break;
51 }
52
53 return TRUE;
54 }
55
56 HRESULT
57 WINAPI
58 DllCanUnloadNow(void)
59 {
60 return S_FALSE;
61 }
62
63 STDAPI
64 DllRegisterServer(void)
65 {
66 HKEY hKey, hSubKey;
67 LPOLESTR pStr;
68 WCHAR szName[MAX_PATH] = L"CLSID\\";
69
70 if (FAILED(StringFromCLSID(&CLSID_CNetCfg, &pStr)))
71 return SELFREG_E_CLASS;
72
73 wcscpy(&szName[6], pStr);
74 CoTaskMemFree(pStr);
75
76 if (RegCreateKeyExW(HKEY_CLASSES_ROOT, szName, 0, NULL, 0, KEY_WRITE, NULL, &hKey, NULL) != ERROR_SUCCESS)
77 return SELFREG_E_CLASS;
78
79 if (RegCreateKeyExW(hKey, L"InProcServer32", 0, NULL, 0, KEY_WRITE, NULL, &hSubKey, NULL) == ERROR_SUCCESS)
80 {
81 if (!GetModuleFileNameW(netcfgx_hInstance, szName, sizeof(szName)/sizeof(WCHAR)))
82 {
83 RegCloseKey(hSubKey);
84 RegCloseKey(hKey);
85 return SELFREG_E_CLASS;
86 }
87 szName[(sizeof(szName)/sizeof(WCHAR))-1] = L'\0';
88 RegSetValueW(hSubKey, NULL, REG_SZ, szName, (wcslen(szName)+1) * sizeof(WCHAR));
89 RegSetValueExW(hSubKey, L"ThreadingModel", 0, REG_SZ, (LPBYTE)L"Both", 10);
90 RegCloseKey(hSubKey);
91 }
92
93 RegCloseKey(hKey);
94 return S_OK;
95 }
96
97 STDAPI
98 DllUnregisterServer(void)
99 {
100 //FIXME
101 // implement unregistering services
102 //
103 return S_OK;
104 }
105
106 STDAPI
107 DllGetClassObject(
108 REFCLSID rclsid,
109 REFIID riid,
110 LPVOID* ppv
111 )
112 {
113 UINT i;
114 HRESULT hres = E_OUTOFMEMORY;
115 IClassFactory * pcf = NULL;
116
117 if (!ppv)
118 return E_INVALIDARG;
119
120 *ppv = NULL;
121
122 for (i = 0; InterfaceTable[i].riid; i++)
123 {
124 if (IsEqualIID(InterfaceTable[i].riid, rclsid))
125 {
126 pcf = IClassFactory_fnConstructor(InterfaceTable[i].lpfnCI, NULL, NULL);
127 break;
128 }
129 }
130
131 if (!pcf)
132 {
133 return CLASS_E_CLASSNOTAVAILABLE;
134 }
135
136 hres = IClassFactory_QueryInterface(pcf, riid, ppv);
137 IClassFactory_Release(pcf);
138
139 return hres;
140 }
141
142
143 /* Append a REG_SZ to an existing REG_MULTI_SZ string in the registry.
144 * If the value doesn't exist, create it.
145 * Returns ERROR_SUCCESS if success. Othewise, returns an error code
146 */
147 static LONG
148 AppendStringToMultiSZ(
149 IN HKEY hKey,
150 IN PCWSTR ValueName,
151 IN PCWSTR ValueToAppend)
152 {
153 PWSTR Buffer = NULL;
154 DWORD dwRegType;
155 DWORD dwRequired, dwLength;
156 DWORD dwTmp;
157 LONG rc;
158
159 rc = RegQueryValueExW(
160 hKey,
161 ValueName,
162 NULL,
163 &dwRegType,
164 NULL,
165 &dwRequired);
166 if (rc != ERROR_FILE_NOT_FOUND)
167 {
168 if (rc != ERROR_SUCCESS)
169 goto cleanup;
170 if (dwRegType != REG_MULTI_SZ)
171 {
172 rc = ERROR_GEN_FAILURE;
173 goto cleanup;
174 }
175
176 dwTmp = dwLength = dwRequired + wcslen(ValueToAppend) * sizeof(WCHAR) + sizeof(UNICODE_NULL);
177 Buffer = HeapAlloc(GetProcessHeap(), 0, dwLength);
178 if (!Buffer)
179 {
180 rc = ERROR_NOT_ENOUGH_MEMORY;
181 goto cleanup;
182 }
183 rc = RegQueryValueExW(
184 hKey,
185 ValueName,
186 NULL,
187 NULL,
188 (BYTE*)Buffer,
189 &dwTmp);
190 if (rc != ERROR_SUCCESS)
191 goto cleanup;
192 }
193 else
194 {
195 dwRequired = sizeof(WCHAR);
196 dwLength = wcslen(ValueToAppend) * sizeof(WCHAR) + 2 * sizeof(UNICODE_NULL);
197 Buffer = HeapAlloc(GetProcessHeap(), 0, dwLength);
198 if (!Buffer)
199 {
200 rc = ERROR_NOT_ENOUGH_MEMORY;
201 goto cleanup;
202 }
203 }
204
205 /* Append the value */
206 wcscpy(&Buffer[dwRequired / sizeof(WCHAR) - 1], ValueToAppend);
207 /* Terminate the REG_MULTI_SZ string */
208 Buffer[dwLength / sizeof(WCHAR) - 1] = UNICODE_NULL;
209
210 rc = RegSetValueExW(
211 hKey,
212 ValueName,
213 0,
214 REG_MULTI_SZ,
215 (const BYTE*)Buffer,
216 dwLength);
217
218 cleanup:
219 HeapFree(GetProcessHeap(), 0, Buffer);
220 return rc;
221 }
222
223 /* Install a section of a .inf file
224 * Returns TRUE if success, FALSE if failure. Error code can
225 * be retrieved with GetLastError()
226 */
227 static BOOL
228 InstallInfSection(
229 IN HWND hWnd,
230 IN LPCWSTR InfFile,
231 IN LPCWSTR InfSection OPTIONAL,
232 IN LPCWSTR InfService OPTIONAL)
233 {
234 WCHAR Buffer[MAX_PATH];
235 HINF hInf = INVALID_HANDLE_VALUE;
236 UINT BufferSize;
237 PVOID Context = NULL;
238 BOOL ret = FALSE;
239
240 /* Get Windows directory */
241 BufferSize = MAX_PATH - 5 - wcslen(InfFile);
242 if (GetWindowsDirectoryW(Buffer, BufferSize) > BufferSize)
243 {
244 /* Function failed */
245 SetLastError(ERROR_GEN_FAILURE);
246 goto cleanup;
247 }
248 /* We have enough space to add some information in the buffer */
249 if (Buffer[wcslen(Buffer) - 1] != '\\')
250 wcscat(Buffer, L"\\");
251 wcscat(Buffer, L"Inf\\");
252 wcscat(Buffer, InfFile);
253
254 /* Install specified section */
255 hInf = SetupOpenInfFileW(Buffer, NULL, INF_STYLE_WIN4, NULL);
256 if (hInf == INVALID_HANDLE_VALUE)
257 goto cleanup;
258
259 Context = SetupInitDefaultQueueCallback(hWnd);
260 if (Context == NULL)
261 goto cleanup;
262
263 ret = TRUE;
264 if (ret && InfSection)
265 {
266 ret = SetupInstallFromInfSectionW(
267 hWnd, hInf,
268 InfSection, SPINST_ALL,
269 NULL, NULL, SP_COPY_NEWER,
270 SetupDefaultQueueCallbackW, Context,
271 NULL, NULL);
272 }
273 if (ret && InfService)
274 {
275 ret = SetupInstallServicesFromInfSectionW(
276 hInf, InfService, 0);
277 }
278
279 cleanup:
280 if (Context)
281 SetupTermDefaultQueueCallback(Context);
282 if (hInf != INVALID_HANDLE_VALUE)
283 SetupCloseInfFile(hInf);
284 return ret;
285 }
286
287 /* Add default services for network cards */
288 static DWORD
289 InstallAdditionalServices(
290 IN HWND hWnd)
291 {
292 BOOL ret;
293
294 /* Install TCP/IP protocol */
295 ret = InstallInfSection(
296 hWnd,
297 L"nettcpip.inf",
298 L"MS_TCPIP.PrimaryInstall",
299 L"MS_TCPIP.PrimaryInstall.Services");
300 if (!ret && GetLastError() != ERROR_FILE_NOT_FOUND)
301 {
302 DPRINT("InstallInfSection() failed with error 0x%lx\n", GetLastError());
303 return GetLastError();
304 }
305
306 /* You can add here more clients (SMB...) and services (DHCP server...) */
307
308 return ERROR_SUCCESS;
309 }
310
311 static DWORD
312 InstallNetDevice(
313 IN HDEVINFO DeviceInfoSet,
314 IN PSP_DEVINFO_DATA DeviceInfoData,
315 LPCWSTR UuidString,
316 DWORD Characteristics,
317 LPCWSTR BusType)
318 {
319 LPWSTR InstanceId = NULL;
320 LPWSTR DeviceName = NULL;
321 LPWSTR ExportName = NULL;
322 LONG rc;
323 HKEY hKey = NULL;
324 HKEY hNetworkKey = NULL;
325 HKEY hLinkageKey = NULL;
326 HKEY hConnectionKey = NULL;
327 DWORD dwShowIcon, dwLength;
328 SP_DEVINSTALL_PARAMS_W installParams;
329
330 /* Get Instance ID */
331 if (SetupDiGetDeviceInstanceIdW(DeviceInfoSet, DeviceInfoData, NULL, 0, &dwLength))
332 {
333 DPRINT("SetupDiGetDeviceInstanceIdW() returned TRUE. FALSE expected\n");
334 rc = ERROR_GEN_FAILURE;
335 goto cleanup;
336 }
337 InstanceId = HeapAlloc(GetProcessHeap(), 0, dwLength * sizeof(WCHAR));
338 if (!InstanceId)
339 {
340 DPRINT("HeapAlloc() failed\n");
341 rc = ERROR_NOT_ENOUGH_MEMORY;
342 goto cleanup;
343 }
344 if (!SetupDiGetDeviceInstanceIdW(DeviceInfoSet, DeviceInfoData, InstanceId, dwLength, NULL))
345 {
346 rc = GetLastError();
347 DPRINT("SetupDiGetDeviceInstanceIdW() failed with error 0x%lx\n", rc);
348 goto cleanup;
349 }
350
351 /* Create device name */
352 DeviceName = HeapAlloc(GetProcessHeap(), 0, (wcslen(L"\\Device\\") + wcslen(UuidString)) * sizeof(WCHAR) + sizeof(UNICODE_NULL));
353 if (!DeviceName)
354 {
355 DPRINT("HeapAlloc() failed\n");
356 rc = ERROR_NOT_ENOUGH_MEMORY;
357 goto cleanup;
358 }
359 wcscpy(DeviceName, L"\\Device\\");
360 wcscat(DeviceName, UuidString);
361
362 /* Create export name */
363 ExportName = HeapAlloc(GetProcessHeap(), 0, (wcslen(L"\\Device\\Tcpip_") + wcslen(UuidString)) * sizeof(WCHAR) + sizeof(UNICODE_NULL));
364 if (!ExportName)
365 {
366 DPRINT("HeapAlloc() failed\n");
367 rc = ERROR_NOT_ENOUGH_MEMORY;
368 goto cleanup;
369 }
370 wcscpy(ExportName, L"\\Device\\Tcpip_");
371 wcscat(ExportName, UuidString);
372
373 /* Write Tcpip parameters in new service Key */
374 rc = RegCreateKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Services", 0, NULL, REG_OPTION_NON_VOLATILE, 0, NULL, &hKey, NULL);
375 if (rc != ERROR_SUCCESS)
376 {
377 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
378 goto cleanup;
379 }
380 rc = RegCreateKeyExW(hKey, UuidString, 0, NULL, REG_OPTION_NON_VOLATILE, 0, NULL, &hNetworkKey, NULL);
381 if (rc != ERROR_SUCCESS)
382 {
383 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
384 goto cleanup;
385 }
386 RegCloseKey(hKey);
387 hKey = NULL;
388 rc = RegCreateKeyExW(hNetworkKey, L"Parameters\\Tcpip", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE, NULL, &hKey, NULL);
389 if (rc != ERROR_SUCCESS)
390 {
391 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
392 goto cleanup;
393 }
394 RegCloseKey(hNetworkKey);
395 hNetworkKey = NULL;
396 rc = RegSetValueExW(hKey, L"DefaultGateway", 0, REG_SZ, (const BYTE*)L"0.0.0.0", (wcslen(L"0.0.0.0") + 1) * sizeof(WCHAR));
397 if (rc != ERROR_SUCCESS)
398 {
399 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
400 goto cleanup;
401 }
402 rc = RegSetValueExW(hKey, L"IPAddress", 0, REG_SZ, (const BYTE*)L"0.0.0.0", (wcslen(L"0.0.0.0") + 1) * sizeof(WCHAR));
403 if (rc != ERROR_SUCCESS)
404 {
405 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
406 goto cleanup;
407 }
408 rc = RegSetValueExW(hKey, L"SubnetMask", 0, REG_SZ, (const BYTE*)L"0.0.0.0", (wcslen(L"0.0.0.0") + 1) * sizeof(WCHAR));
409 if (rc != ERROR_SUCCESS)
410 {
411 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
412 goto cleanup;
413 }
414 RegCloseKey(hKey);
415 hKey = NULL;
416
417 /* Write 'Linkage' key in hardware key */
418 #if _WIN32_WINNT >= 0x502
419 hKey = SetupDiOpenDevRegKey(DeviceInfoSet, DeviceInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_READ | KEY_WRITE);
420 #else
421 hKey = SetupDiOpenDevRegKey(DeviceInfoSet, DeviceInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_ALL_ACCESS);
422 #endif
423 if (hKey == INVALID_HANDLE_VALUE && GetLastError() == ERROR_FILE_NOT_FOUND)
424 hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet, DeviceInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, NULL, NULL);
425 if (hKey == INVALID_HANDLE_VALUE)
426 {
427 hKey = NULL;
428 rc = GetLastError();
429 DPRINT("SetupDiCreateDevRegKeyW() failed with error 0x%lx\n", rc);
430 goto cleanup;
431 }
432 rc = RegSetValueExW(hKey, L"NetCfgInstanceId", 0, REG_SZ, (const BYTE*)UuidString, (wcslen(UuidString) + 1) * sizeof(WCHAR));
433 if (rc != ERROR_SUCCESS)
434 {
435 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
436 goto cleanup;
437 }
438 rc = RegSetValueExW(hKey, L"Characteristics", 0, REG_DWORD, (const BYTE*)&Characteristics, sizeof(DWORD));
439 if (rc != ERROR_SUCCESS)
440 {
441 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
442 goto cleanup;
443 }
444 if (BusType)
445 rc = RegSetValueExW(hKey, L"BusType", 0, REG_SZ, (const BYTE*)BusType, (wcslen(BusType) + 1) * sizeof(WCHAR));
446 if (rc != ERROR_SUCCESS)
447 {
448 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
449 goto cleanup;
450 }
451 rc = RegCreateKeyExW(hKey, L"Linkage", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE, NULL, &hLinkageKey, NULL);
452 if (rc != ERROR_SUCCESS)
453 {
454 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
455 goto cleanup;
456 }
457 rc = RegSetValueExW(hLinkageKey, L"Export", 0, REG_SZ, (const BYTE*)DeviceName, (wcslen(DeviceName) + 1) * sizeof(WCHAR));
458 if (rc != ERROR_SUCCESS)
459 {
460 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
461 goto cleanup;
462 }
463 rc = RegSetValueExW(hLinkageKey, L"RootDevice", 0, REG_SZ, (const BYTE*)UuidString, (wcslen(UuidString) + 1) * sizeof(WCHAR));
464 if (rc != ERROR_SUCCESS)
465 {
466 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
467 goto cleanup;
468 }
469 rc = RegSetValueExW(hLinkageKey, L"UpperBind", 0, REG_SZ, (const BYTE*)L"Tcpip", (wcslen(L"Tcpip") + 1) * sizeof(WCHAR));
470 if (rc != ERROR_SUCCESS)
471 {
472 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
473 goto cleanup;
474 }
475 RegCloseKey(hKey);
476 hKey = NULL;
477
478 /* Write connection information in network subkey */
479 rc = RegCreateKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}", 0, NULL, REG_OPTION_NON_VOLATILE, 0, NULL, &hNetworkKey, NULL);
480 if (rc != ERROR_SUCCESS)
481 {
482 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
483 goto cleanup;
484 }
485 rc = RegCreateKeyExW(hNetworkKey, UuidString, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_CREATE_SUB_KEY, NULL, &hKey, NULL);
486 if (rc != ERROR_SUCCESS)
487 {
488 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
489 goto cleanup;
490 }
491 rc = RegCreateKeyExW(hKey, L"Connection", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_SET_VALUE, NULL, &hConnectionKey, NULL);
492 RegCloseKey(hKey);
493 hKey = NULL;
494 if (rc != ERROR_SUCCESS)
495 {
496 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
497 goto cleanup;
498 }
499 rc = RegSetValueExW(hConnectionKey, L"Name", 0, REG_SZ, (const BYTE*)L"Network connection", (wcslen(L"Network connection") + 1) * sizeof(WCHAR));
500 if (rc != ERROR_SUCCESS)
501 {
502 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
503 goto cleanup;
504 }
505 rc = RegSetValueExW(hConnectionKey, L"PnpInstanceId", 0, REG_SZ, (const BYTE*)InstanceId, (wcslen(InstanceId) + 1) * sizeof(WCHAR));
506 if (rc != ERROR_SUCCESS)
507 {
508 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
509 goto cleanup;
510 }
511 dwShowIcon = 1;
512 rc = RegSetValueExW(hConnectionKey, L"ShowIcon", 0, REG_DWORD, (const BYTE*)&dwShowIcon, sizeof(dwShowIcon));
513 if (rc != ERROR_SUCCESS)
514 {
515 DPRINT("RegSetValueExW() failed with error 0x%lx\n", rc);
516 goto cleanup;
517 }
518
519 /* Write linkage information in Tcpip service */
520 rc = RegCreateKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Linkage", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_QUERY_VALUE | KEY_SET_VALUE, NULL, &hKey, NULL);
521 if (rc != ERROR_SUCCESS)
522 {
523 DPRINT("RegCreateKeyExW() failed with error 0x%lx\n", rc);
524 goto cleanup;
525 }
526 rc = AppendStringToMultiSZ(hKey, L"Bind", DeviceName);
527 if (rc != ERROR_SUCCESS)
528 {
529 DPRINT("AppendStringToMultiSZ() failed with error 0x%lx\n", rc);
530 goto cleanup;
531 }
532 rc = AppendStringToMultiSZ(hKey, L"Export", ExportName);
533 if (rc != ERROR_SUCCESS)
534 {
535 DPRINT("AppendStringToMultiSZ() failed with error 0x%lx\n", rc);
536 goto cleanup;
537 }
538 rc = AppendStringToMultiSZ(hKey, L"Route", UuidString);
539 if (rc != ERROR_SUCCESS)
540 {
541 DPRINT("AppendStringToMultiSZ() failed with error 0x%lx\n", rc);
542 goto cleanup;
543 }
544
545 /* Install additionnal services */
546 rc = InstallAdditionalServices(NULL);
547 if (rc != ERROR_SUCCESS)
548 {
549 DPRINT("InstallAdditionalServices() failed with error 0x%lx\n", rc);
550 goto cleanup;
551 }
552
553 /* HACK: hpoussin, Dec 2005. TCP/IP driver is not able to manage devices
554 * which are installed after its startup. So, we have to reboot to take
555 * this new netcard into account.
556 */
557 /* Should we reboot? */
558 installParams.cbSize = sizeof(SP_DEVINSTALL_PARAMS_W);
559 if (!SetupDiGetDeviceInstallParamsW(
560 DeviceInfoSet,
561 DeviceInfoData,
562 &installParams))
563 {
564 rc = GetLastError();
565 DPRINT("SetupDiGetDeviceInstallParams() failed with error 0x%lx\n", rc);
566 goto cleanup;
567 }
568 installParams.Flags |= DI_NEEDRESTART;
569 if (!SetupDiSetDeviceInstallParamsW(
570 DeviceInfoSet,
571 DeviceInfoData,
572 &installParams))
573 {
574 rc = GetLastError();
575 DPRINT("SetupDiSetDeviceInstallParams() failed with error 0x%lx\n", rc);
576 goto cleanup;
577 }
578 rc = ERROR_SUCCESS;
579
580 cleanup:
581 HeapFree(GetProcessHeap(), 0, InstanceId);
582 HeapFree(GetProcessHeap(), 0, DeviceName);
583 HeapFree(GetProcessHeap(), 0, ExportName);
584 if (hKey != NULL)
585 RegCloseKey(hKey);
586 if (hNetworkKey != NULL)
587 RegCloseKey(hNetworkKey);
588 if (hLinkageKey != NULL)
589 RegCloseKey(hLinkageKey);
590 if (hConnectionKey != NULL)
591 RegCloseKey(hConnectionKey);
592 return rc;
593 }
594
595 static DWORD
596 InstallNetClient(VOID)
597 {
598 DPRINT1("Installation of network clients is not yet supported\n");
599 return ERROR_GEN_FAILURE;
600 }
601
602 static DWORD
603 InstallNetService(VOID)
604 {
605 DPRINT1("Installation of network services is not yet supported\n");
606 return ERROR_GEN_FAILURE;
607 }
608
609 static DWORD
610 InstallNetTransport(VOID)
611 {
612 DPRINT1("Installation of network protocols is not yet supported\n");
613 return ERROR_GEN_FAILURE;
614 }
615
616 DWORD WINAPI
617 NetClassInstaller(
618 IN DI_FUNCTION InstallFunction,
619 IN HDEVINFO DeviceInfoSet,
620 IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL)
621 {
622 SP_DRVINFO_DATA_W DriverInfoData;
623 SP_DRVINFO_DETAIL_DATA_W DriverInfoDetail;
624 WCHAR SectionName[LINE_LEN];
625 HINF hInf = INVALID_HANDLE_VALUE;
626 INFCONTEXT InfContext;
627 UINT ErrorLine;
628 INT CharacteristicsInt;
629 DWORD Characteristics;
630 LPWSTR BusType = NULL;
631 RPC_STATUS RpcStatus;
632 UUID Uuid;
633 LPWSTR UuidRpcString = NULL;
634 LPWSTR UuidString = NULL;
635 LONG rc;
636 DWORD dwLength;
637
638 if (InstallFunction != DIF_INSTALLDEVICE)
639 return ERROR_DI_DO_DEFAULT;
640
641 DPRINT("%lu %p %p\n", InstallFunction, DeviceInfoSet, DeviceInfoData);
642
643 /* Get driver info details */
644 DriverInfoData.cbSize = sizeof(SP_DRVINFO_DATA_W);
645 if (!SetupDiGetSelectedDriverW(DeviceInfoSet, DeviceInfoData, &DriverInfoData))
646 {
647 rc = GetLastError();
648 DPRINT("SetupDiGetSelectedDriverW() failed with error 0x%lx\n", rc);
649 goto cleanup;
650 }
651 DriverInfoDetail.cbSize = sizeof(SP_DRVINFO_DETAIL_DATA_W);
652 if (!SetupDiGetDriverInfoDetailW(DeviceInfoSet, DeviceInfoData, &DriverInfoData, &DriverInfoDetail, sizeof(DriverInfoDetail), NULL)
653 && GetLastError() != ERROR_INSUFFICIENT_BUFFER)
654 {
655 rc = GetLastError();
656 DPRINT("SetupDiGetDriverInfoDetailW() failed with error 0x%lx\n", rc);
657 goto cleanup;
658 }
659 hInf = SetupOpenInfFileW(DriverInfoDetail.InfFileName, NULL, INF_STYLE_WIN4, &ErrorLine);
660 if (hInf == INVALID_HANDLE_VALUE)
661 {
662 rc = GetLastError();
663 DPRINT("SetupOpenInfFileW() failed with error 0x%lx\n", rc);
664 goto cleanup;
665 }
666 if (!SetupDiGetActualSectionToInstallW(hInf, DriverInfoDetail.SectionName, SectionName, LINE_LEN, NULL, NULL))
667 {
668 rc = GetLastError();
669 DPRINT("SetupDiGetActualSectionToInstallW() failed with error 0x%lx\n", rc);
670 goto cleanup;
671 }
672
673 /* Get Characteristics and BusType (optional) from .inf file */
674 if (!SetupFindFirstLineW(hInf, SectionName, L"Characteristics", &InfContext))
675 {
676 rc = GetLastError();
677 DPRINT("Unable to find key %S in section %S of file %S (error 0x%lx)\n",
678 L"Characteristics", SectionName, DriverInfoDetail.InfFileName, rc);
679 goto cleanup;
680 }
681 if (!SetupGetIntField(&InfContext, 1, &CharacteristicsInt))
682 {
683 rc = GetLastError();
684 DPRINT("SetupGetIntField() failed with error 0x%lx\n", rc);
685 goto cleanup;
686 }
687 Characteristics = (DWORD)CharacteristicsInt;
688 if (IsEqualIID(&DeviceInfoData->ClassGuid, &GUID_DEVCLASS_NET))
689 {
690 if (SetupFindFirstLineW(hInf, SectionName, L"BusType", &InfContext))
691 {
692 if (!SetupGetStringFieldW(&InfContext, 1, NULL, 0, &dwLength))
693 {
694 rc = GetLastError();
695 DPRINT("SetupGetStringFieldW() failed with error 0x%lx\n", rc);
696 goto cleanup;
697 }
698 BusType = HeapAlloc(GetProcessHeap(), 0, dwLength * sizeof(WCHAR));
699 if (!BusType)
700 {
701 DPRINT("HeapAlloc() failed\n");
702 rc = ERROR_NOT_ENOUGH_MEMORY;
703 goto cleanup;
704 }
705 if (!SetupGetStringFieldW(&InfContext, 1, BusType, dwLength, NULL))
706 {
707 rc = GetLastError();
708 DPRINT("SetupGetStringFieldW() failed with error 0x%lx\n", rc);
709 goto cleanup;
710 }
711 }
712 }
713
714 /* Create a new UUID */
715 RpcStatus = UuidCreate(&Uuid);
716 if (RpcStatus != RPC_S_OK && RpcStatus != RPC_S_UUID_LOCAL_ONLY)
717 {
718 DPRINT("UuidCreate() failed with RPC status 0x%lx\n", RpcStatus);
719 rc = ERROR_GEN_FAILURE;
720 goto cleanup;
721 }
722 RpcStatus = UuidToStringW(&Uuid, &UuidRpcString);
723 if (RpcStatus != RPC_S_OK)
724 {
725 DPRINT("UuidToStringW() failed with RPC status 0x%lx\n", RpcStatus);
726 rc = ERROR_GEN_FAILURE;
727 goto cleanup;
728 }
729
730 /* Add curly braces around Uuid */
731 UuidString = HeapAlloc(GetProcessHeap(), 0, (2 + wcslen(UuidRpcString)) * sizeof(WCHAR) + sizeof(UNICODE_NULL));
732 if (!UuidString)
733 {
734 DPRINT("HeapAlloc() failed\n");
735 rc = ERROR_NOT_ENOUGH_MEMORY;
736 goto cleanup;
737 }
738 wcscpy(UuidString, L"{");
739 wcscat(UuidString, UuidRpcString);
740 wcscat(UuidString, L"}");
741
742 if (IsEqualIID(&DeviceInfoData->ClassGuid, &GUID_DEVCLASS_NET))
743 rc = InstallNetDevice(DeviceInfoSet, DeviceInfoData, UuidString, Characteristics, BusType);
744 else if (IsEqualIID(&DeviceInfoData->ClassGuid, &GUID_DEVCLASS_NETCLIENT))
745 rc = InstallNetClient();
746 else if (IsEqualIID(&DeviceInfoData->ClassGuid, &GUID_DEVCLASS_NETSERVICE))
747 rc = InstallNetService();
748 else if (IsEqualIID(&DeviceInfoData->ClassGuid, &GUID_DEVCLASS_NETTRANS))
749 rc = InstallNetTransport();
750 else
751 {
752 DPRINT("Invalid class guid\n");
753 rc = ERROR_GEN_FAILURE;
754 }
755
756 cleanup:
757 if (hInf != INVALID_HANDLE_VALUE)
758 SetupCloseInfFile(hInf);
759 if (UuidRpcString != NULL)
760 RpcStringFreeW(&UuidRpcString);
761 HeapFree(GetProcessHeap(), 0, BusType);
762 HeapFree(GetProcessHeap(), 0, UuidString);
763
764 if (rc == ERROR_SUCCESS)
765 rc = ERROR_DI_DO_DEFAULT;
766 DPRINT("Returning 0x%lx\n", rc);
767 return rc;
768 }