- Update to r53061
[reactos.git] / dll / win32 / netshell / lanstatusui.c
1 #include <precomp.h>
2
3 /// CLSID
4 /// HKEY_LOCAL_MACHINE\SOFTWARE\Classes\CLSID\{7007ACCF-3202-11D1-AAD2-00805FC1270E}
5 // IID B722BCCB-4E68-101B-A2BC-00AA00404770
6
7 #define WM_SHOWSTATUSDLG (WM_USER+10)
8
9 typedef struct tagNotificationItem
10 {
11 struct tagNotificationItem *pNext;
12 CLSID guidItem;
13 UINT uID;
14 HWND hwndDlg;
15 INetConnection *pNet;
16 }NOTIFICATION_ITEM;
17
18 typedef struct
19 {
20 IOleCommandTarget * lpVtbl;
21 INetConnectionManager * lpNetMan;
22 LONG ref;
23 NOTIFICATION_ITEM * pHead;
24 }ILanStatusImpl, *LPILanStatusImpl;
25
26 typedef struct
27 {
28 INetConnection *pNet;
29 HWND hwndStatusDlg; /* LanStatusDlg window */
30 HWND hwndDlg; /* status dialog window */
31 DWORD dwAdapterIndex;
32 UINT_PTR nIDEvent;
33 UINT DHCPEnabled;
34 DWORD dwInOctets;
35 DWORD dwOutOctets;
36 DWORD IpAddress;
37 DWORD SubnetMask;
38 DWORD Gateway;
39 UINT uID;
40 UINT Status;
41 }LANSTATUSUI_CONTEXT;
42
43 VOID
44 UpdateLanStatusUiDlg(
45 HWND hwndDlg,
46 MIB_IFROW * IfEntry,
47 LANSTATUSUI_CONTEXT * pContext)
48 {
49 WCHAR szFormat[MAX_PATH] = {0};
50 WCHAR szBuffer[MAX_PATH] = {0};
51 SYSTEMTIME TimeConnected;
52 DWORD DurationSeconds;
53 WCHAR Buffer[100];
54 WCHAR DayBuffer[30];
55 WCHAR LocBuffer[50];
56
57 #if 0
58 ULONGLONG Ticks;
59 #else
60 DWORD Ticks;
61 #endif
62
63 if (IfEntry->dwSpeed < 1000)
64 {
65 if (LoadStringW(netshell_hInstance, IDS_FORMAT_BIT, szFormat, sizeof(szFormat)/sizeof(WCHAR)))
66 {
67 swprintf(szBuffer, szFormat, IfEntry->dwSpeed);
68 SendDlgItemMessageW(hwndDlg, IDC_SPEED, WM_SETTEXT, 0, (LPARAM)szBuffer);
69 }
70 }
71 else if (IfEntry->dwSpeed < 1000000)
72 {
73 if (LoadStringW(netshell_hInstance, IDS_FORMAT_KBIT, szFormat, sizeof(szFormat)/sizeof(WCHAR)))
74 {
75 swprintf(szBuffer, szFormat, IfEntry->dwSpeed/1000);
76 SendDlgItemMessageW(hwndDlg, IDC_SPEED, WM_SETTEXT, 0, (LPARAM)szBuffer);
77 }
78 }
79 else if (IfEntry->dwSpeed < 1000000000)
80 {
81 if (LoadStringW(netshell_hInstance, IDS_FORMAT_MBIT, szFormat, sizeof(szFormat)/sizeof(WCHAR)))
82 {
83 swprintf(szBuffer, szFormat, IfEntry->dwSpeed/1000000);
84 SendDlgItemMessageW(hwndDlg, IDC_SPEED, WM_SETTEXT, 0, (LPARAM)szBuffer);
85 }
86 }
87 else
88 {
89 if (LoadStringW(netshell_hInstance, IDS_FORMAT_KBIT, szFormat, sizeof(szFormat)/sizeof(WCHAR)))
90 {
91 swprintf(szBuffer, szFormat, IfEntry->dwSpeed/1000000000);
92 SendDlgItemMessageW(hwndDlg, IDC_SPEED, WM_SETTEXT, 0, (LPARAM)szBuffer);
93 }
94 }
95
96 if (StrFormatByteSizeW(IfEntry->dwInOctets, szBuffer, sizeof(szFormat)/sizeof(WCHAR)))
97 {
98 SendDlgItemMessageW(hwndDlg, IDC_RECEIVED, WM_SETTEXT, 0, (LPARAM)szBuffer);
99 }
100
101 if (StrFormatByteSizeW(IfEntry->dwOutOctets, szBuffer, sizeof(szFormat)/sizeof(WCHAR)))
102 {
103 SendDlgItemMessageW(hwndDlg, IDC_SEND, WM_SETTEXT, 0, (LPARAM)szBuffer);
104 }
105
106 #if 0
107 Ticks = GetTickCount64();
108 #else
109 Ticks = GetTickCount();
110 #endif
111
112 DurationSeconds = Ticks / 1000;
113 TimeConnected.wSecond = (DurationSeconds % 60);
114 TimeConnected.wMinute = (DurationSeconds / 60) % 60;
115 TimeConnected.wHour = (DurationSeconds / (60 * 60)) % 24;
116 TimeConnected.wDay = DurationSeconds / (60 * 60 * 24);
117
118 if (!GetTimeFormatW(LOCALE_USER_DEFAULT, 0, &TimeConnected, L"HH':'mm':'ss", LocBuffer, sizeof(LocBuffer) / sizeof(LocBuffer[0])))
119 return;
120
121 if (!TimeConnected.wDay)
122 {
123 SendDlgItemMessageW(hwndDlg, IDC_DURATION, WM_SETTEXT, 0, (LPARAM)LocBuffer);
124 }
125 else
126 {
127 if (TimeConnected.wDay == 1)
128 {
129 if (!LoadStringW(netshell_hInstance, IDS_DURATION_DAY, DayBuffer, sizeof(DayBuffer) / sizeof(DayBuffer[0])))
130 DayBuffer[0] = L'\0';
131 }
132 else
133 {
134 if (!LoadStringW(netshell_hInstance, IDS_DURATION_DAYS, DayBuffer, sizeof(DayBuffer) / sizeof(DayBuffer[0])))
135 DayBuffer[0] = L'\0';
136 }
137 swprintf(Buffer, DayBuffer, TimeConnected.wDay, LocBuffer);
138 SendDlgItemMessageW(hwndDlg, IDC_DURATION, WM_SETTEXT, 0, (LPARAM)Buffer);
139 }
140
141 }
142
143 VOID
144 UpdateLanStatus(HWND hwndDlg, LANSTATUSUI_CONTEXT * pContext)
145 {
146 MIB_IFROW IfEntry;
147 HICON hIcon, hOldIcon = NULL;
148 NOTIFYICONDATAW nid;
149 NETCON_PROPERTIES * pProperties = NULL;
150
151 ZeroMemory(&IfEntry, sizeof(IfEntry));
152 IfEntry.dwIndex = pContext->dwAdapterIndex;
153 if(GetIfEntry(&IfEntry) != NO_ERROR)
154 {
155 return;
156 }
157
158 hIcon = NULL;
159 if (IfEntry.dwOperStatus == MIB_IF_OPER_STATUS_CONNECTED || IfEntry.dwOperStatus == MIB_IF_OPER_STATUS_OPERATIONAL)
160 {
161 if (pContext->dwInOctets == IfEntry.dwInOctets && pContext->dwOutOctets == IfEntry.dwOutOctets && pContext->Status != 0)
162 {
163 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_IDLE), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
164 pContext->Status = 0;
165 }
166 else if (pContext->dwInOctets != IfEntry.dwInOctets && pContext->dwOutOctets != IfEntry.dwOutOctets && pContext->Status != 1)
167 {
168 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_TRANSREC), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
169 pContext->Status = 1;
170 }
171 else if (pContext->dwInOctets != IfEntry.dwInOctets && pContext->Status != 2)
172 {
173 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_REC), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
174 pContext->Status = 2;
175 }
176 else if (pContext->dwOutOctets != IfEntry.dwOutOctets && pContext->Status != 3)
177 {
178 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_TRANS), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
179 pContext->Status = 3;
180 }
181 }
182 else if (IfEntry.dwOperStatus == MIB_IF_OPER_STATUS_UNREACHABLE || MIB_IF_OPER_STATUS_DISCONNECTED)
183 {
184 if (pContext->Status != 4)
185 {
186 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_OFF), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
187 pContext->Status = 4;
188 }
189 }
190 else if (MIB_IF_OPER_STATUS_NON_OPERATIONAL)
191 {
192 if (pContext->Status != 5)
193 {
194 hIcon = LoadImage(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_OFF), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
195 pContext->Status = 5;
196 }
197 }
198
199 if (hwndDlg && hIcon)
200 {
201 hOldIcon = (HICON)SendDlgItemMessageW(hwndDlg, IDC_NETSTAT, STM_SETICON, (WPARAM)hIcon, 0);
202 if (hOldIcon)
203 DestroyIcon(hOldIcon);
204 }
205
206 ZeroMemory(&nid, sizeof(nid));
207 nid.cbSize = sizeof(nid);
208 nid.uID = pContext->uID;
209 nid.hWnd = pContext->hwndStatusDlg;
210 nid.u.uVersion = 3;
211
212 if (INetConnection_GetProperties(pContext->pNet, &pProperties) == NOERROR)
213 {
214 if (pProperties->dwCharacter & NCCF_SHOW_ICON)
215 {
216 if (hwndDlg)
217 nid.hIcon = CopyImage(hIcon, IMAGE_ICON, 16, 16, 0);
218 else
219 nid.hIcon = hIcon;
220
221 if (nid.hIcon)
222 nid.uFlags |= NIF_ICON;
223
224 nid.uFlags |= NIF_STATE;
225 nid.dwState = 0;
226 nid.dwStateMask = NIS_HIDDEN;
227
228 if (pProperties->pszwName)
229 {
230 if (wcslen(pProperties->pszwName) * sizeof(WCHAR) < sizeof(nid.szTip))
231 {
232 nid.uFlags |= NIF_TIP;
233 wcscpy(nid.szTip, pProperties->pszwName);
234 }
235 else
236 {
237 CopyMemory(nid.szTip, pProperties->pszwName, sizeof(nid.szTip) - sizeof(WCHAR));
238 nid.szTip[(sizeof(nid.szTip)/sizeof(WCHAR))-1] = L'\0';
239 nid.uFlags |= NIF_TIP;
240 }
241 }
242 }
243 else
244 {
245 nid.uFlags |= NIF_STATE;
246 nid.dwState = NIS_HIDDEN;
247 nid.dwStateMask = NIS_HIDDEN;
248
249 }
250 NcFreeNetconProperties(pProperties);
251 }
252
253 Shell_NotifyIconW(NIM_MODIFY, &nid);
254
255 if (nid.uFlags & NIF_ICON)
256 DestroyIcon(nid.hIcon);
257
258 pContext->dwInOctets = IfEntry.dwInOctets;
259 pContext->dwOutOctets = IfEntry.dwOutOctets;
260
261 if (hwndDlg)
262 UpdateLanStatusUiDlg(hwndDlg, &IfEntry, pContext);
263 }
264
265
266 VOID
267 InitializeLANStatusUiDlg(HWND hwndDlg, LANSTATUSUI_CONTEXT * pContext)
268 {
269 WCHAR szBuffer[MAX_PATH] = {0};
270 NETCON_PROPERTIES * pProperties;
271
272 if (INetConnection_GetProperties(pContext->pNet, &pProperties) != NOERROR)
273 return;
274
275 if (pProperties->Status == NCS_DISCONNECTED)
276 LoadStringW(netshell_hInstance, IDS_STATUS_UNREACHABLE, szBuffer, MAX_PATH);
277 else if (pProperties->Status == NCS_MEDIA_DISCONNECTED)
278 LoadStringW(netshell_hInstance, IDS_STATUS_DISCONNECTED, szBuffer, MAX_PATH);
279 else if (pProperties->Status == NCS_CONNECTING)
280 LoadStringW(netshell_hInstance, IDS_STATUS_CONNECTING, szBuffer, MAX_PATH);
281 else if (pProperties->Status == NCS_CONNECTED)
282 LoadStringW(netshell_hInstance, IDS_STATUS_CONNECTED, szBuffer, MAX_PATH);
283
284 SendDlgItemMessageW(hwndDlg, IDC_STATUS, WM_SETTEXT, 0, (LPARAM)szBuffer);
285
286 pContext->dwInOctets = 0;
287 pContext->dwOutOctets = 0;
288
289 /* update adapter info */
290 pContext->Status = -1;
291 UpdateLanStatus(hwndDlg, pContext);
292 NcFreeNetconProperties(pProperties);
293 }
294
295 VOID
296 InsertColumnToListView(
297 HWND hDlgCtrl,
298 UINT ResId,
299 UINT SubItem,
300 UINT Size)
301 {
302 WCHAR szBuffer[200];
303 LVCOLUMNW lc;
304
305 if (!LoadStringW(netshell_hInstance, ResId, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
306 return;
307
308 memset(&lc, 0, sizeof(LV_COLUMN) );
309 lc.mask = LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM | LVCF_FMT;
310 lc.iSubItem = SubItem;
311 lc.fmt = LVCFMT_FIXED_WIDTH;
312 lc.cx = Size;
313 lc.cchTextMax = wcslen(szBuffer);
314 lc.pszText = szBuffer;
315
316 (void)SendMessageW(hDlgCtrl, LVM_INSERTCOLUMNW, SubItem, (LPARAM)&lc);
317 }
318
319 VOID
320 AddIPAddressToListView(
321 HWND hDlgCtrl,
322 PIP_ADDR_STRING pAddr,
323 INT Index)
324 {
325 LVITEMW li;
326 PIP_ADDR_STRING pCur;
327 WCHAR szBuffer[100];
328 UINT SubIndex;
329
330 ZeroMemory(&li, sizeof(LVITEMW));
331 li.mask = LVIF_TEXT;
332 li.iItem = Index;
333 pCur = pAddr;
334 SubIndex = 0;
335
336 do
337 {
338 if (SubIndex)
339 {
340 ZeroMemory(&li, sizeof(LVITEMW));
341 li.mask = LVIF_TEXT;
342 li.iItem = Index;
343 li.iSubItem = 0;
344 li.pszText = L"";
345 li.iItem = SendMessageW(hDlgCtrl, LVM_INSERTITEMW, 0, (LPARAM)&li);
346 }
347
348 if (MultiByteToWideChar(CP_ACP, 0, pCur->IpAddress.String, -1, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
349 {
350 li.pszText = szBuffer;
351 li.iSubItem = 1;
352 li.iItem = Index++;
353 SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
354 }
355 SubIndex++;
356 pCur = pCur->Next;
357 }while(pCur && pCur->IpAddress.String[0]);
358 }
359
360 INT
361 InsertItemToListView(
362 HWND hDlgCtrl,
363 UINT ResId)
364 {
365 LVITEMW li;
366 WCHAR szBuffer[100];
367
368 ZeroMemory(&li, sizeof(LVITEMW));
369 li.mask = LVIF_TEXT;
370 li.iItem = ListView_GetItemCount(hDlgCtrl);
371 if (LoadStringW(netshell_hInstance, ResId, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
372 {
373 li.pszText = szBuffer;
374 return (INT)SendMessageW(hDlgCtrl, LVM_INSERTITEMW, 0, (LPARAM)&li);
375 }
376 return -1;
377 }
378
379
380 INT_PTR
381 CALLBACK
382 LANStatusUiDetailsDlg(
383 HWND hwndDlg,
384 UINT uMsg,
385 WPARAM wParam,
386 LPARAM lParam
387 )
388 {
389 LANSTATUSUI_CONTEXT * pContext;
390 LVITEMW li;
391 WCHAR szBuffer[100];
392 PIP_ADAPTER_INFO pAdapterInfo, pCurAdapter;
393 PIP_PER_ADAPTER_INFO pPerAdapter;
394 DWORD dwSize;
395 HWND hDlgCtrl;
396
397 switch(uMsg)
398 {
399 case WM_INITDIALOG:
400 pContext = (LANSTATUSUI_CONTEXT*)lParam;
401
402 hDlgCtrl = GetDlgItem(hwndDlg, IDC_DETAILS);
403 InsertColumnToListView(hDlgCtrl, IDS_PROPERTY, 0, 80);
404 InsertColumnToListView(hDlgCtrl, IDS_VALUE, 1, 80);
405
406 dwSize = 0;
407 pCurAdapter = NULL;
408 pAdapterInfo = NULL;
409 if (GetAdaptersInfo(NULL, &dwSize) == ERROR_BUFFER_OVERFLOW)
410 {
411 pAdapterInfo = (PIP_ADAPTER_INFO)CoTaskMemAlloc(dwSize);
412 if (pAdapterInfo)
413 {
414 if (GetAdaptersInfo(pAdapterInfo, &dwSize) == NO_ERROR)
415 {
416 pCurAdapter = pAdapterInfo;
417 while(pCurAdapter && pCurAdapter->Index != pContext->dwAdapterIndex)
418 pCurAdapter = pCurAdapter->Next;
419
420 if(pCurAdapter->Index != pContext->dwAdapterIndex)
421 pCurAdapter = NULL;
422 }
423 }
424 }
425
426 ZeroMemory(&li, sizeof(LVITEMW));
427 li.mask = LVIF_TEXT;
428 li.iSubItem = 1;
429 li.pszText = szBuffer;
430
431 if (pCurAdapter)
432 {
433 li.iItem = InsertItemToListView(hDlgCtrl, IDS_PHYSICAL_ADDRESS);
434 if (li.iItem >= 0)
435 {
436 swprintf(szBuffer, L"%02x-%02x-%02x-%02x-%02x-%02x",pCurAdapter->Address[0], pCurAdapter->Address[1],
437 pCurAdapter->Address[2], pCurAdapter->Address[3], pCurAdapter->Address[4], pCurAdapter->Address[5]);
438 SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
439 }
440 li.iItem = InsertItemToListView(hDlgCtrl, IDS_IP_ADDRESS);
441 if (li.iItem >= 0)
442 if (MultiByteToWideChar(CP_ACP, 0, pCurAdapter->IpAddressList.IpAddress.String, -1, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
443 SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
444
445 li.iItem = InsertItemToListView(hDlgCtrl, IDS_SUBNET_MASK);
446 if (li.iItem >= 0)
447 if (MultiByteToWideChar(CP_ACP, 0, pCurAdapter->IpAddressList.IpMask.String, -1, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
448 SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
449
450 li.iItem = InsertItemToListView(hDlgCtrl, IDS_DEF_GATEWAY);
451 if (li.iItem >= 0 && pCurAdapter->GatewayList.IpAddress.String[0] != '0')
452 {
453 if (MultiByteToWideChar(CP_ACP, 0, pCurAdapter->GatewayList.IpAddress.String, -1, szBuffer, sizeof(szBuffer)/sizeof(WCHAR)))
454 SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
455 }
456 #if 0
457 li.iItem = InsertItemToListView(hDlgCtrl, IDS_LEASE_OBTAINED);
458 li.iItem = InsertItemToListView(hDlgCtrl, IDS_LEASE_EXPIRES);
459 #endif
460 }
461
462 dwSize = 0;
463 if (GetPerAdapterInfo(pContext->dwAdapterIndex, NULL, &dwSize) == ERROR_BUFFER_OVERFLOW)
464 {
465 pPerAdapter = (PIP_PER_ADAPTER_INFO)CoTaskMemAlloc(dwSize);
466 if (pPerAdapter)
467 {
468 if (GetPerAdapterInfo(pContext->dwAdapterIndex, pPerAdapter, &dwSize) == ERROR_SUCCESS)
469 {
470 li.iItem = InsertItemToListView(hDlgCtrl, IDS_DNS_SERVERS);
471 if (li.iItem >= 0)
472 AddIPAddressToListView(hDlgCtrl, &pPerAdapter->DnsServerList, li.iItem);
473 }
474 CoTaskMemFree(pPerAdapter);
475 }
476 }
477 #if 0
478 if (pCurAdapter)
479 {
480 li.iItem = InsertItemToListView(hDlgCtrl, IDS_WINS_SERVERS);
481 AddIPAddressToListView(hDlgCtrl, &pCurAdapter->PrimaryWinsServer, li.iItem);
482 AddIPAddressToListView(hDlgCtrl, &pCurAdapter->SecondaryWinsServer, li.iItem+1);
483 }
484 #endif
485 CoTaskMemFree(pAdapterInfo);
486 break;
487 case WM_COMMAND:
488 if (LOWORD(wParam) == IDC_CLOSE)
489 {
490 EndDialog(hwndDlg, FALSE);
491 break;
492 }
493 }
494 return FALSE;
495 }
496
497 INT_PTR
498 CALLBACK
499 LANStatusUiAdvancedDlg(
500 HWND hwndDlg,
501 UINT uMsg,
502 WPARAM wParam,
503 LPARAM lParam
504 )
505 {
506 WCHAR szBuffer[100] = {0};
507 PROPSHEETPAGE *page;
508 LANSTATUSUI_CONTEXT * pContext;
509 DWORD dwIpAddr;
510
511
512 switch(uMsg)
513 {
514 case WM_INITDIALOG:
515 page = (PROPSHEETPAGE*)lParam;
516 pContext = (LANSTATUSUI_CONTEXT*)page->lParam;
517 SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pContext);
518 if (pContext->DHCPEnabled)
519 LoadStringW(netshell_hInstance, IDS_ASSIGNED_DHCP, szBuffer, sizeof(szBuffer)/sizeof(WCHAR));
520 else
521 LoadStringW(netshell_hInstance, IDS_ASSIGNED_MANUAL, szBuffer, sizeof(szBuffer)/sizeof(WCHAR));
522
523 szBuffer[(sizeof(szBuffer)/sizeof(WCHAR))-1] = L'\0';
524 SendDlgItemMessageW(hwndDlg, IDC_DETAILSTYPE, WM_SETTEXT, 0, (LPARAM)szBuffer);
525
526
527 dwIpAddr = ntohl(pContext->IpAddress);
528 swprintf(szBuffer, L"%u.%u.%u.%u", FIRST_IPADDRESS(dwIpAddr), SECOND_IPADDRESS(dwIpAddr),
529 THIRD_IPADDRESS(dwIpAddr), FOURTH_IPADDRESS(dwIpAddr));
530 SendDlgItemMessageW(hwndDlg, IDC_DETAILSIP, WM_SETTEXT, 0, (LPARAM)szBuffer);
531
532 dwIpAddr = ntohl(pContext->SubnetMask);
533 swprintf(szBuffer, L"%u.%u.%u.%u", FIRST_IPADDRESS(dwIpAddr), SECOND_IPADDRESS(dwIpAddr),
534 THIRD_IPADDRESS(dwIpAddr), FOURTH_IPADDRESS(dwIpAddr));
535 SendDlgItemMessageW(hwndDlg, IDC_DETAILSSUBNET, WM_SETTEXT, 0, (LPARAM)szBuffer);
536
537 dwIpAddr = ntohl(pContext->Gateway);
538 if (dwIpAddr)
539 {
540 swprintf(szBuffer, L"%u.%u.%u.%u", FIRST_IPADDRESS(dwIpAddr), SECOND_IPADDRESS(dwIpAddr),
541 THIRD_IPADDRESS(dwIpAddr), FOURTH_IPADDRESS(dwIpAddr));
542 SendDlgItemMessageW(hwndDlg, IDC_DETAILSGATEWAY, WM_SETTEXT, 0, (LPARAM)szBuffer);
543 }
544 return TRUE;
545 case WM_COMMAND:
546 if (LOWORD(wParam) == IDC_DETAILS)
547 {
548 pContext = (LANSTATUSUI_CONTEXT*)GetWindowLongPtr(hwndDlg, DWLP_USER);
549 if (pContext)
550 {
551 DialogBoxParamW(netshell_hInstance, MAKEINTRESOURCEW(IDD_LAN_NETSTATUSDETAILS), GetParent(hwndDlg),
552 LANStatusUiDetailsDlg, (LPARAM)pContext);
553 }
554 }
555 break;
556 default:
557 break;
558 }
559 return FALSE;
560 }
561
562 BOOL
563 FindNetworkAdapter(HDEVINFO hInfo, SP_DEVINFO_DATA *pDevInfo, LPWSTR pGuid)
564 {
565 DWORD dwIndex, dwSize;
566 HKEY hSubKey;
567 WCHAR szNetCfg[50];
568 WCHAR szDetail[200] = L"SYSTEM\\CurrentControlSet\\Control\\Class\\";
569
570 dwIndex = 0;
571 do
572 {
573
574 ZeroMemory(pDevInfo, sizeof(SP_DEVINFO_DATA));
575 pDevInfo->cbSize = sizeof(SP_DEVINFO_DATA);
576
577 /* get device info */
578 if (!SetupDiEnumDeviceInfo(hInfo, dwIndex++, pDevInfo))
579 break;
580
581 /* get device software registry path */
582 if (!SetupDiGetDeviceRegistryPropertyW(hInfo, pDevInfo, SPDRP_DRIVER, NULL, (LPBYTE)&szDetail[39], sizeof(szDetail)/sizeof(WCHAR) - 40, &dwSize))
583 break;
584
585 /* open device registry key */
586 if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szDetail, 0, KEY_READ, &hSubKey) != ERROR_SUCCESS)
587 break;
588
589 /* query NetCfgInstanceId for current device */
590 dwSize = sizeof(szNetCfg);
591 if (RegQueryValueExW(hSubKey, L"NetCfgInstanceId", NULL, NULL, (LPBYTE)szNetCfg, &dwSize) != ERROR_SUCCESS)
592 {
593 RegCloseKey(hSubKey);
594 break;
595 }
596 RegCloseKey(hSubKey);
597 if (!_wcsicmp(pGuid, szNetCfg))
598 {
599 return TRUE;
600 }
601 }
602 while(TRUE);
603
604 return FALSE;
605 }
606
607
608
609 VOID
610 DisableNetworkAdapter(INetConnection * pNet, LANSTATUSUI_CONTEXT * pContext, HWND hwndDlg)
611 {
612 HKEY hKey;
613 NETCON_PROPERTIES * pProperties;
614 LPOLESTR pDisplayName;
615 WCHAR szPath[200];
616 DWORD dwSize, dwType;
617 LPWSTR pPnp;
618 HDEVINFO hInfo;
619 SP_DEVINFO_DATA DevInfo;
620 SP_PROPCHANGE_PARAMS PropChangeParams;
621 BOOL bClose = FALSE;
622 NOTIFYICONDATAW nid;
623
624 if (FAILED(INetConnection_GetProperties(pNet, &pProperties)))
625 return;
626
627
628 hInfo = SetupDiGetClassDevsW(&GUID_DEVCLASS_NET, NULL, NULL, DIGCF_PRESENT );
629 if (!hInfo)
630 {
631 NcFreeNetconProperties(pProperties);
632 return;
633 }
634
635 if (FAILED(StringFromCLSID(&pProperties->guidId, &pDisplayName)))
636 {
637 NcFreeNetconProperties(pProperties);
638 SetupDiDestroyDeviceInfoList(hInfo);
639 return;
640 }
641 NcFreeNetconProperties(pProperties);
642
643 if (FindNetworkAdapter(hInfo, &DevInfo, pDisplayName))
644 {
645 PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
646 PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE; //;
647 PropChangeParams.StateChange = DICS_DISABLE;
648 PropChangeParams.Scope = DICS_FLAG_CONFIGSPECIFIC;
649 PropChangeParams.HwProfile = 0;
650
651 if (SetupDiSetClassInstallParams(hInfo, &DevInfo, &PropChangeParams.ClassInstallHeader, sizeof(SP_PROPCHANGE_PARAMS)))
652 {
653 if (SetupDiCallClassInstaller(DIF_PROPERTYCHANGE, hInfo, &DevInfo))
654 bClose = TRUE;
655 }
656 }
657 SetupDiDestroyDeviceInfoList(hInfo);
658
659 swprintf(szPath, L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\%s\\Connection", pDisplayName);
660 CoTaskMemFree(pDisplayName);
661
662 if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szPath, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
663 return;
664
665 dwSize = 0;
666 if (RegQueryValueExW(hKey, L"PnpInstanceID", NULL, &dwType, NULL, &dwSize) != ERROR_SUCCESS || dwType != REG_SZ)
667 {
668 RegCloseKey(hKey);
669 return;
670 }
671
672 pPnp = (LPWSTR)CoTaskMemAlloc(dwSize);
673 if (!pPnp)
674 {
675 RegCloseKey(hKey);
676 return;
677 }
678
679 if (RegQueryValueExW(hKey, L"PnpInstanceID", NULL, &dwType, (LPBYTE)pPnp, &dwSize) != ERROR_SUCCESS)
680 {
681 CoTaskMemFree(pPnp);
682 RegCloseKey(hKey);
683 return;
684 }
685 RegCloseKey(hKey);
686
687 swprintf(szPath, L"System\\CurrentControlSet\\Hardware Profiles\\Current\\System\\CurrentControlSet\\Enum\\%s", pPnp);
688 CoTaskMemFree(pPnp);
689
690 if (RegCreateKeyExW(HKEY_LOCAL_MACHINE, szPath, 0, NULL, 0, KEY_WRITE, NULL, &hKey, NULL) != ERROR_SUCCESS)
691 return;
692
693 dwSize = 1; /* enable = 0, disable = 1 */
694 RegSetValueExW(hKey, L"CSConfigFlags", 0, REG_DWORD, (LPBYTE)&dwSize, sizeof(DWORD));
695 RegCloseKey(hKey);
696
697 if (!bClose)
698 return;
699
700 PropSheet_PressButton(GetParent(hwndDlg), PSBTN_CANCEL);
701 ZeroMemory(&nid, sizeof(nid));
702 nid.cbSize = sizeof(nid);
703 nid.uID = pContext->uID;
704 nid.hWnd = pContext->hwndDlg;
705 nid.uFlags = NIF_STATE;
706 nid.dwState = NIS_HIDDEN;
707 nid.dwStateMask = NIS_HIDDEN;
708
709 Shell_NotifyIconW(NIM_MODIFY, &nid);
710 }
711
712
713 INT_PTR
714 CALLBACK
715 LANStatusUiDlg(
716 HWND hwndDlg,
717 UINT uMsg,
718 WPARAM wParam,
719 LPARAM lParam
720 )
721 {
722 PROPSHEETPAGE *page;
723 LANSTATUSUI_CONTEXT * pContext;
724 LPPSHNOTIFY lppsn;
725
726 switch(uMsg)
727 {
728 case WM_INITDIALOG:
729 page = (PROPSHEETPAGE*)lParam;
730 pContext = (LANSTATUSUI_CONTEXT*)page->lParam;
731 pContext->hwndDlg = hwndDlg;
732 InitializeLANStatusUiDlg(hwndDlg, pContext);
733 SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pContext);
734 return TRUE;
735 case WM_COMMAND:
736 pContext = (LANSTATUSUI_CONTEXT*)GetWindowLongPtr(hwndDlg, DWLP_USER);
737 if (LOWORD(wParam) == IDC_STATUS_PROPERTIES)
738 {
739 if (pContext)
740 {
741 ShowNetConnectionProperties(pContext->pNet, GetParent(pContext->hwndDlg));
742 BringWindowToTop(GetParent(pContext->hwndDlg));
743 }
744 break;
745 }
746 else if (LOWORD(wParam) == IDC_ENDISABLE)
747 {
748 DisableNetworkAdapter(pContext->pNet, pContext, hwndDlg);
749 break;
750 }
751 case WM_NOTIFY:
752 lppsn = (LPPSHNOTIFY) lParam;
753 if (lppsn->hdr.code == PSN_APPLY || lppsn->hdr.code == PSN_RESET)
754 {
755 pContext = (LANSTATUSUI_CONTEXT*)GetWindowLongPtr(hwndDlg, DWLP_USER);
756 SetWindowLongPtr(hwndDlg, DWL_MSGRESULT, PSNRET_NOERROR);
757 pContext->hwndDlg = NULL;
758 return TRUE;
759 }
760 break;
761 }
762 return FALSE;
763 }
764
765 VOID
766 InitializePropertyDialog(
767 LANSTATUSUI_CONTEXT * pContext,
768 NETCON_PROPERTIES * pProperties)
769 {
770 DWORD dwSize, dwAdapterIndex, dwResult;
771 LPOLESTR pStr;
772 IP_ADAPTER_INFO * pAdapterInfo, *pCurAdapter;
773
774 if (FAILED(StringFromCLSID(&pProperties->guidId, &pStr)))
775 {
776 return;
777 }
778
779 /* get the IfTable */
780 dwSize = 0;
781 dwResult = GetAdaptersInfo(NULL, &dwSize);
782 if (dwResult!= ERROR_BUFFER_OVERFLOW)
783 {
784 CoTaskMemFree(pStr);
785 return;
786 }
787
788 pAdapterInfo = (PIP_ADAPTER_INFO)CoTaskMemAlloc(dwSize);
789 if (!pAdapterInfo)
790 {
791 CoTaskMemFree(pAdapterInfo);
792 CoTaskMemFree(pStr);
793 return;
794 }
795
796 if (GetAdaptersInfo(pAdapterInfo, &dwSize) != NO_ERROR)
797 {
798 CoTaskMemFree(pAdapterInfo);
799 CoTaskMemFree(pStr);
800 return;
801 }
802
803 if (!GetAdapterIndexFromNetCfgInstanceId(pAdapterInfo, pStr, &dwAdapterIndex))
804 {
805 CoTaskMemFree(pAdapterInfo);
806 CoTaskMemFree(pStr);
807 return;
808 }
809
810 pCurAdapter = pAdapterInfo;
811 while(pCurAdapter->Index != dwAdapterIndex)
812 pCurAdapter = pCurAdapter->Next;
813
814
815 pContext->IpAddress = inet_addr(pCurAdapter->IpAddressList.IpAddress.String);
816 pContext->SubnetMask = inet_addr(pCurAdapter->IpAddressList.IpMask.String);
817 pContext->Gateway = inet_addr(pCurAdapter->GatewayList.IpAddress.String);
818 pContext->DHCPEnabled = pCurAdapter->DhcpEnabled;
819 CoTaskMemFree(pStr);
820 CoTaskMemFree(pAdapterInfo);
821 pContext->dwAdapterIndex = dwAdapterIndex;
822 }
823
824 VOID
825 ShowStatusPropertyDialog(
826 LANSTATUSUI_CONTEXT * pContext,
827 HWND hwndDlg)
828 {
829 HPROPSHEETPAGE hppages[2];
830 PROPSHEETHEADERW pinfo;
831 NETCON_PROPERTIES * pProperties = NULL;
832
833 ZeroMemory(&pinfo, sizeof(PROPSHEETHEADERW));
834 ZeroMemory(hppages, sizeof(hppages));
835 pinfo.dwSize = sizeof(PROPSHEETHEADERW);
836 pinfo.dwFlags = PSH_NOCONTEXTHELP | PSH_PROPTITLE | PSH_NOAPPLYNOW;
837 pinfo.u3.phpage = hppages;
838 pinfo.hwndParent = hwndDlg;
839
840 if (INetConnection_GetProperties(pContext->pNet, &pProperties) == NOERROR)
841 {
842 if (pProperties->pszwName)
843 {
844 pinfo.pszCaption = pProperties->pszwName;
845 pinfo.dwFlags |= PSH_PROPTITLE;
846 }
847 InitializePropertyDialog(pContext, pProperties);
848 if (pProperties->MediaType == NCM_LAN && pProperties->Status == NCS_CONNECTED)
849 {
850 hppages[0] = InitializePropertySheetPage(MAKEINTRESOURCEW(IDD_LAN_NETSTATUS), LANStatusUiDlg, (LPARAM)pContext, NULL);
851 if (hppages[0])
852 pinfo.nPages++;
853
854 hppages[pinfo.nPages] = InitializePropertySheetPage(MAKEINTRESOURCEW(IDD_LAN_NETSTATUSADVANCED), LANStatusUiAdvancedDlg, (LPARAM)pContext, NULL);
855 if (hppages[pinfo.nPages])
856 pinfo.nPages++;
857
858 if (pinfo.nPages)
859 {
860 PropertySheetW(&pinfo);
861 }
862 }
863 else if (pProperties->Status == NCS_MEDIA_DISCONNECTED || pProperties->Status == NCS_DISCONNECTED ||
864 pProperties->Status == NCS_HARDWARE_DISABLED)
865 {
866 ShowNetConnectionProperties(pContext->pNet, pContext->hwndDlg);
867 }
868
869 NcFreeNetconProperties(pProperties);
870 }
871 }
872
873 INT_PTR
874 CALLBACK
875 LANStatusDlg(
876 HWND hwndDlg,
877 UINT uMsg,
878 WPARAM wParam,
879 LPARAM lParam
880 )
881 {
882 LANSTATUSUI_CONTEXT * pContext;
883
884 switch(uMsg)
885 {
886 case WM_INITDIALOG:
887 pContext = (LANSTATUSUI_CONTEXT *)lParam;
888 SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)lParam);
889 pContext->nIDEvent = SetTimer(hwndDlg, 0xFABC, 1000, NULL);
890 return TRUE;
891 case WM_TIMER:
892 pContext = (LANSTATUSUI_CONTEXT*)GetWindowLongPtr(hwndDlg, DWLP_USER);
893 if (wParam == (WPARAM)pContext->nIDEvent)
894 {
895 UpdateLanStatus(pContext->hwndDlg, pContext);
896 }
897 break;
898 case WM_SHOWSTATUSDLG:
899 if (LOWORD(lParam) == WM_LBUTTONDOWN)
900 {
901 pContext = (LANSTATUSUI_CONTEXT*)GetWindowLongPtr(hwndDlg, DWLP_USER);
902 if (!pContext)
903 break;
904
905 if (pContext->hwndDlg)
906 {
907 ShowWindow(GetParent(pContext->hwndDlg), SW_SHOW);
908 BringWindowToTop(GetParent(pContext->hwndDlg));
909 }
910 else
911 {
912 ShowStatusPropertyDialog(pContext, hwndDlg);
913 }
914 break;
915 }
916 break;
917 }
918 return FALSE;
919 }
920 static
921 HRESULT
922 InitializeNetTaskbarNotifications(
923 ILanStatusImpl * This)
924 {
925 NOTIFYICONDATAW nid;
926 HWND hwndDlg;
927 INetConnectionManager * INetConMan;
928 IEnumNetConnection * IEnumCon;
929 INetConnection * INetCon;
930 NETCON_PROPERTIES* pProps;
931 HRESULT hr;
932 ULONG Count;
933 ULONG Index;
934 NOTIFICATION_ITEM * pItem, *pLast = NULL;
935 LANSTATUSUI_CONTEXT * pContext;
936
937 if (This->pHead)
938 {
939 pItem = This->pHead;
940 while(pItem)
941 {
942 hr = INetConnection_GetProperties(pItem->pNet, &pProps);
943 if (SUCCEEDED(hr))
944 {
945 ZeroMemory(&nid, sizeof(nid));
946 nid.cbSize = sizeof(nid);
947 nid.uID = pItem->uID;
948 nid.hWnd = pItem->hwndDlg;
949 nid.uFlags = NIF_STATE;
950 if (pProps->dwCharacter & NCCF_SHOW_ICON)
951 nid.dwState = 0;
952 else
953 nid.dwState = NIS_HIDDEN;
954
955 nid.dwStateMask = NIS_HIDDEN;
956 Shell_NotifyIconW(NIM_MODIFY, &nid);
957 NcFreeNetconProperties(pProps);
958 }
959 pItem = pItem->pNext;
960 }
961 return S_OK;
962 }
963 /* get an instance to of IConnectionManager */
964
965 //hr = CoCreateInstance(&CLSID_ConnectionManager, NULL, CLSCTX_INPROC_SERVER, &IID_INetConnectionManager, (LPVOID*)&INetConMan);
966
967 hr = INetConnectionManager_Constructor(NULL, &IID_INetConnectionManager, (LPVOID*)&INetConMan);
968 if (FAILED(hr))
969 return hr;
970
971 hr = INetConnectionManager_EnumConnections(INetConMan, NCME_DEFAULT, &IEnumCon);
972 if (FAILED(hr))
973 {
974 INetConnectionManager_Release(INetConMan);
975 return hr;
976 }
977
978 Index = 1;
979 do
980 {
981 hr = IEnumNetConnection_Next(IEnumCon, 1, &INetCon, &Count);
982 if (hr == S_OK)
983 {
984 pItem = (NOTIFICATION_ITEM*)CoTaskMemAlloc(sizeof(NOTIFICATION_ITEM));
985 if (!pItem)
986 break;
987
988 pContext = (LANSTATUSUI_CONTEXT*)CoTaskMemAlloc(sizeof(LANSTATUSUI_CONTEXT));
989 if (!pContext)
990 {
991 CoTaskMemFree(pItem);
992 break;
993 }
994
995
996 ZeroMemory(pContext, sizeof(LANSTATUSUI_CONTEXT));
997 pContext->uID = Index;
998 pContext->pNet = INetCon;
999 pItem->uID = Index;
1000 pItem->pNext = NULL;
1001 pItem->pNet = INetCon;
1002 hwndDlg = CreateDialogParamW(netshell_hInstance, MAKEINTRESOURCEW(IDD_STATUS), NULL, LANStatusDlg, (LPARAM)pContext);
1003 if (hwndDlg)
1004 {
1005 ZeroMemory(&nid, sizeof(nid));
1006 nid.cbSize = sizeof(nid);
1007 nid.uID = Index++;
1008 nid.uFlags = NIF_MESSAGE;
1009 nid.u.uVersion = 3;
1010 nid.uCallbackMessage = WM_SHOWSTATUSDLG;
1011 nid.hWnd = hwndDlg;
1012
1013 hr = INetConnection_GetProperties(INetCon, &pProps);
1014 if (SUCCEEDED(hr))
1015 {
1016 CopyMemory(&pItem->guidItem, &pProps->guidId, sizeof(GUID));
1017 if (!(pProps->dwCharacter & NCCF_SHOW_ICON))
1018 {
1019 nid.dwState = NIS_HIDDEN;
1020 nid.dwStateMask = NIS_HIDDEN;
1021 nid.uFlags |= NIF_STATE;
1022 }
1023 if (pProps->Status == NCS_MEDIA_DISCONNECTED || pProps->Status == NCS_DISCONNECTED || pProps->Status == NCS_HARDWARE_DISABLED)
1024 nid.hIcon = LoadIcon(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_OFF));
1025 else if (pProps->Status == NCS_CONNECTED)
1026 nid.hIcon = LoadIcon(netshell_hInstance, MAKEINTRESOURCE(IDI_NET_IDLE));
1027
1028 if (nid.hIcon)
1029 nid.uFlags |= NIF_ICON;
1030
1031 wcscpy(nid.szTip, pProps->pszwName);
1032 nid.uFlags |= NIF_TIP;
1033 }
1034 pContext->hwndStatusDlg = hwndDlg;
1035 pItem->hwndDlg = hwndDlg;
1036
1037 if (Shell_NotifyIconW(NIM_ADD, &nid))
1038 {
1039 if (pLast)
1040 pLast->pNext = pItem;
1041 else
1042 This->pHead = pItem;
1043
1044 pLast = pItem;
1045 Index++;
1046 }
1047 else
1048 {
1049 CoTaskMemFree(pItem);
1050 }
1051
1052 if (nid.uFlags & NIF_ICON)
1053 DestroyIcon(nid.hIcon);
1054 }
1055 }
1056 }while(hr == S_OK);
1057
1058 This->lpNetMan = INetConMan;
1059 IEnumNetConnection_Release(IEnumCon);
1060 return S_OK;
1061 }
1062
1063 HRESULT
1064 ShowStatusDialogByCLSID(
1065 ILanStatusImpl * This,
1066 const GUID *pguidCmdGroup)
1067 {
1068 NOTIFICATION_ITEM * pItem;
1069
1070 pItem = This->pHead;
1071 while(pItem)
1072 {
1073 if (IsEqualGUID(&pItem->guidItem, pguidCmdGroup))
1074 {
1075 SendMessageW(pItem->hwndDlg, WM_SHOWSTATUSDLG, 0, WM_LBUTTONDOWN);
1076 return S_OK;
1077 }
1078 pItem = pItem->pNext;
1079 }
1080 return E_FAIL;
1081 }
1082 static
1083 HRESULT
1084 WINAPI
1085 IOleCommandTarget_fnQueryInterface(
1086 IOleCommandTarget * iface,
1087 REFIID iid,
1088 LPVOID * ppvObj)
1089 {
1090 ILanStatusImpl * This = (ILanStatusImpl*)iface;
1091 *ppvObj = NULL;
1092
1093 if (IsEqualIID (iid, &IID_IUnknown) ||
1094 IsEqualIID (iid, &IID_IOleCommandTarget))
1095 {
1096 *ppvObj = This;
1097 IUnknown_AddRef(iface);
1098 return S_OK;
1099 }
1100 MessageBoxW(NULL, L"IOleCommandTarget_fnQueryInterface", NULL, MB_OK);
1101 return E_NOINTERFACE;
1102 }
1103
1104 static
1105 ULONG
1106 WINAPI
1107 IOleCommandTarget_fnAddRef(
1108 IOleCommandTarget * iface)
1109 {
1110 ILanStatusImpl * This = (ILanStatusImpl*)iface;
1111 ULONG refCount = InterlockedIncrement(&This->ref);
1112
1113 return refCount;
1114 }
1115
1116 static
1117 ULONG
1118 WINAPI
1119 IOleCommandTarget_fnRelease(
1120 IOleCommandTarget * iface)
1121 {
1122 #if 0
1123 ILanStatusImpl * This = (ILanStatusImpl*)iface;
1124 ULONG refCount = InterlockedDecrement(&This->ref);
1125
1126 if (!refCount)
1127 {
1128 CoTaskMemFree (This);
1129 }
1130 return refCount;
1131 #else
1132 return 1;
1133 #endif
1134 }
1135
1136 static
1137 HRESULT
1138 WINAPI
1139 IOleCommandTarget_fnQueryStatus(
1140 IOleCommandTarget * iface,
1141 const GUID *pguidCmdGroup,
1142 ULONG cCmds,
1143 OLECMD *prgCmds,
1144 OLECMDTEXT *pCmdText)
1145 {
1146 MessageBoxW(NULL, pCmdText->rgwz, L"IOleCommandTarget_fnQueryStatus", MB_OK);
1147 return E_NOTIMPL;
1148 }
1149
1150 static
1151 HRESULT
1152 WINAPI
1153 IOleCommandTarget_fnExec(
1154 IOleCommandTarget * iface,
1155 const GUID *pguidCmdGroup,
1156 DWORD nCmdID,
1157 DWORD nCmdexecopt,
1158 VARIANT *pvaIn,
1159 VARIANT *pvaOut)
1160 {
1161 ILanStatusImpl * This = (ILanStatusImpl*)iface;
1162
1163 if (pguidCmdGroup)
1164 {
1165 if (IsEqualIID(pguidCmdGroup, &CGID_ShellServiceObject))
1166 {
1167 return InitializeNetTaskbarNotifications(This);
1168 }
1169 else
1170 {
1171 /* invoke status dialog */
1172 return ShowStatusDialogByCLSID(This, pguidCmdGroup);
1173 }
1174 }
1175 return S_OK;
1176 }
1177
1178
1179 static const IOleCommandTargetVtbl vt_OleCommandTarget =
1180 {
1181 IOleCommandTarget_fnQueryInterface,
1182 IOleCommandTarget_fnAddRef,
1183 IOleCommandTarget_fnRelease,
1184 IOleCommandTarget_fnQueryStatus,
1185 IOleCommandTarget_fnExec,
1186 };
1187
1188
1189 HRESULT WINAPI LanConnectStatusUI_Constructor (IUnknown * pUnkOuter, REFIID riid, LPVOID * ppv)
1190 {
1191 ILanStatusImpl * This;
1192 static volatile ILanStatusImpl *cached_This = NULL;
1193
1194 if (!ppv)
1195 return E_POINTER;
1196
1197 if (pUnkOuter)
1198 return CLASS_E_NOAGGREGATION;
1199
1200 This = (ILanStatusImpl *) CoTaskMemAlloc(sizeof (ILanStatusImpl));
1201 if (!This)
1202 return E_OUTOFMEMORY;
1203
1204 This->ref = 1;
1205 This->lpVtbl = (IOleCommandTarget*)&vt_OleCommandTarget;
1206 This->lpNetMan = NULL;
1207 This->pHead = NULL;
1208
1209 if (InterlockedCompareExchangePointer((void **)&cached_This, This, NULL) != NULL)
1210 {
1211 CoTaskMemFree(This);
1212 }
1213
1214 return IOleCommandTarget_fnQueryInterface ((IOleCommandTarget*)cached_This, riid, ppv);
1215 }
1216