SHDisconnectNetDrives.c File Reference

#include "netplwiz.h"
#include <wine/debug.h>

Include dependency graph for SHDisconnectNetDrives.c:

Go to the source code of this file.

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (netplwiz)
VOID	InitializeListView (HWND hDlg)
VOID	EnumerateConnectedDrives (HWND hDlg)
VOID	UpdateButtonStatus (WPARAM wParam, LPARAM lParam, HWND hDlg)
DWORD	DisconnectDriveExit (HWND hDlg)
static INT_PTR CALLBACK	DisconnectDlgProc (HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
HRESULT WINAPI	SHDisconnectNetDrives (PVOID Unused)

Variables
HINSTANCE	hInstance

Function Documentation

DisconnectDlgProc()

static INT_PTR CALLBACK DisconnectDlgProc ( HWND  hDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 155 of file SHDisconnectNetDrives.c.

{
    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;
}

Referenced by SHDisconnectNetDrives().

DisconnectDriveExit()

DWORD DisconnectDriveExit

(

HWND

hDlg

)

Definition at line 138 of file SHDisconnectNetDrives.c.

{
    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;
}

Referenced by DisconnectDlgProc().

EnumerateConnectedDrives()

VOID EnumerateConnectedDrives

(

HWND

hDlg

)

Definition at line 38 of file SHDisconnectNetDrives.c.

{
    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);
}

Referenced by DisconnectDlgProc().

InitializeListView()

VOID InitializeListView

(

HWND

hDlg

)

Definition at line 15 of file SHDisconnectNetDrives.c.

{
    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);
}

Referenced by DisconnectDlgProc().

SHDisconnectNetDrives()

HRESULT WINAPI SHDisconnectNetDrives

(

PVOID

Unused

)

Definition at line 193 of file SHDisconnectNetDrives.c.

{
    DialogBoxW(hInstance, MAKEINTRESOURCEW(IDD_DISCONNECTDRIVES), NULL, DisconnectDlgProc);
    return S_OK;
}

UpdateButtonStatus()

VOID UpdateButtonStatus	(	WPARAM	wParam,
		LPARAM	lParam,
		HWND	hDlg
	)

Definition at line 114 of file SHDisconnectNetDrives.c.

{
    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;
        }
    }
}

Referenced by DisconnectDlgProc().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

netplwiz

)

Variable Documentation

hInstance

HINSTANCE hInstance

Definition at line 13 of file SHDisconnectNetDrives.c.

Referenced by DisconnectDlgProc(), InitializeListView(), and SHDisconnectNetDrives().