mirror of
https://github.com/reactos/reactos.git
synced 2024-08-20 02:13:29 +00:00
198 lines
5.4 KiB
C
198 lines
5.4 KiB
C
/*
|
|
* PROJECT: ReactOS Shell
|
|
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
|
|
* PURPOSE: Implements the Disconnct Network Drive dialog
|
|
* COPYRIGHT: Copyright 2018 Jared Smudde (computerwhiz02@hotmail.com)
|
|
*/
|
|
|
|
#include "netplwiz.h"
|
|
#include <wine/debug.h>
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(netplwiz);
|
|
|
|
HINSTANCE hInstance;
|
|
|
|
VOID InitializeListView(HWND hDlg)
|
|
{
|
|
HWND hListView = GetDlgItem(hDlg, IDC_CONNECTEDDRIVELIST);
|
|
LV_COLUMN column;
|
|
WCHAR szLetter[100], szPath[100];
|
|
|
|
if (hListView == NULL)
|
|
return;
|
|
|
|
LoadStringW(hInstance, IDS_DRIVE_LETTER, szLetter, _countof(szLetter));
|
|
LoadStringW(hInstance, IDS_NETWORK_PATH, szPath, _countof(szPath));
|
|
|
|
column.mask = LVCF_WIDTH | LVCF_TEXT;
|
|
column.pszText = szLetter;
|
|
column.cx = 75;
|
|
ListView_InsertColumn(hListView, 0, &column);
|
|
|
|
column.mask = LVCF_WIDTH | LVCF_TEXT;
|
|
column.pszText = szPath;
|
|
column.cx = 150;
|
|
ListView_InsertColumn(hListView, 1, &column);
|
|
}
|
|
|
|
VOID EnumerateConnectedDrives(HWND hDlg)
|
|
{
|
|
LV_ITEM item;
|
|
HIMAGELIST hImgList;
|
|
HICON hIconDrive = NULL;
|
|
HMODULE hShell32;
|
|
HWND hListView = GetDlgItem(hDlg, IDC_CONNECTEDDRIVELIST);
|
|
|
|
DWORD dRet;
|
|
HANDLE hEnum;
|
|
LPNETRESOURCE lpRes;
|
|
DWORD dwSize = 0x1000;
|
|
DWORD dwCount = -1;
|
|
LPNETRESOURCE lpCur;
|
|
|
|
/* List View Icons */
|
|
hShell32 = GetModuleHandleW(L"shell32.dll");
|
|
if (hShell32 == NULL)
|
|
return;
|
|
|
|
hIconDrive = LoadImageW(hShell32, MAKEINTRESOURCEW(10), IMAGE_ICON,
|
|
GetSystemMetrics(SM_CXSMICON),
|
|
GetSystemMetrics(SM_CYSMICON), LR_DEFAULTCOLOR);
|
|
if (hIconDrive == NULL)
|
|
return;
|
|
|
|
hImgList = ImageList_Create(16, 16, ILC_COLOR32 | ILC_MASK, 1, 1);
|
|
if (hImgList == NULL)
|
|
{
|
|
DestroyIcon(hIconDrive);
|
|
return;
|
|
}
|
|
|
|
ImageList_AddIcon(hImgList, hIconDrive);
|
|
DestroyIcon(hIconDrive);
|
|
ListView_SetImageList(hListView, hImgList, LVSIL_SMALL);
|
|
|
|
dRet = WNetOpenEnum(RESOURCE_CONNECTED, RESOURCETYPE_DISK, 0, NULL, &hEnum);
|
|
if (dRet != WN_SUCCESS)
|
|
{
|
|
return;
|
|
}
|
|
|
|
lpRes = HeapAlloc(GetProcessHeap(), 0, dwSize);
|
|
if (!lpRes)
|
|
{
|
|
WNetCloseEnum(hEnum);
|
|
return;
|
|
}
|
|
|
|
do
|
|
{
|
|
ZeroMemory(lpRes, dwSize);
|
|
dRet = WNetEnumResource(hEnum, &dwCount, lpRes, &dwSize);
|
|
if (dRet == WN_SUCCESS || dRet == WN_MORE_DATA)
|
|
{
|
|
lpCur = lpRes;
|
|
for (; dwCount; dwCount--)
|
|
{
|
|
ZeroMemory(&item, sizeof(item));
|
|
item.mask = LVIF_TEXT | LVIF_IMAGE;
|
|
item.iImage = 0;
|
|
item.pszText = lpCur->lpLocalName;
|
|
item.lParam = 0;
|
|
item.iItem = ListView_InsertItem(hListView, &item);
|
|
ListView_SetItemText(hListView, item.iItem, 1, lpCur->lpRemoteName);
|
|
lpCur++;
|
|
}
|
|
}
|
|
} while (dRet != WN_NO_MORE_ENTRIES);
|
|
|
|
HeapFree(GetProcessHeap(), 0, lpRes);
|
|
WNetCloseEnum(hEnum);
|
|
}
|
|
|
|
|
|
VOID UpdateButtonStatus(WPARAM wParam, LPARAM lParam, HWND hDlg)
|
|
{
|
|
LPNMHDR pnmh = (LPNMHDR)lParam;
|
|
HWND hListView = GetDlgItem(hDlg, IDC_CONNECTEDDRIVELIST);
|
|
HWND hOkButton = GetDlgItem(hDlg, ID_OK);
|
|
|
|
if (pnmh->hwndFrom == hListView)
|
|
{
|
|
switch (pnmh->code)
|
|
{
|
|
case LVN_ITEMCHANGED:
|
|
if (ListView_GetSelectedCount(hListView))
|
|
{
|
|
EnableWindow(hOkButton, TRUE);
|
|
}
|
|
else
|
|
{
|
|
EnableWindow(hOkButton, FALSE);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
DWORD DisconnectDriveExit(HWND hDlg)
|
|
{
|
|
TCHAR driveLetter[10];
|
|
HRESULT hr;
|
|
HWND hListView = GetDlgItem(hDlg, IDC_CONNECTEDDRIVELIST);
|
|
INT nItem = ListView_GetNextItem(hListView, -1, LVNI_SELECTED);
|
|
|
|
if (nItem == -1)
|
|
return 0;
|
|
|
|
ListView_GetItemText(hListView, nItem, 0, driveLetter, _countof(driveLetter));
|
|
hr = WNetCancelConnection2(driveLetter, CONNECT_UPDATE_PROFILE, FALSE);
|
|
|
|
EndDialog(hDlg, ID_OK);
|
|
return hr;
|
|
}
|
|
|
|
static INT_PTR CALLBACK DisconnectDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
HICON hIcon, hIconSm;
|
|
switch (uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
hIcon = (HICON)LoadImageW(hInstance, MAKEINTRESOURCEW(IDI_DISCONNECT_NET_DRIVES), IMAGE_ICON, 0, 0, LR_SHARED | LR_DEFAULTSIZE);
|
|
hIconSm = (HICON)CopyImage(hIcon, IMAGE_ICON, GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), LR_COPYFROMRESOURCE);
|
|
SendMessageW(hDlg, WM_SETICON, ICON_BIG, (LPARAM)hIcon);
|
|
SendMessageW(hDlg, WM_SETICON, ICON_SMALL, (LPARAM)hIconSm);
|
|
EnableWindow(GetDlgItem(hDlg, ID_OK), FALSE);
|
|
InitializeListView(hDlg);
|
|
EnumerateConnectedDrives(hDlg);
|
|
return TRUE;
|
|
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam))
|
|
{
|
|
case ID_OK:
|
|
DisconnectDriveExit(hDlg);
|
|
break;
|
|
case IDCANCEL:
|
|
EndDialog(hDlg, IDCANCEL);
|
|
break;
|
|
}
|
|
break;
|
|
|
|
case WM_NOTIFY:
|
|
UpdateButtonStatus(wParam, lParam, hDlg);
|
|
break;
|
|
|
|
default:
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
HRESULT WINAPI SHDisconnectNetDrives(PVOID Unused)
|
|
{
|
|
DialogBoxW(hInstance, MAKEINTRESOURCEW(IDD_DISCONNECTDRIVES), NULL, DisconnectDlgProc);
|
|
return S_OK;
|
|
}
|