shell32_apitest_sub.cpp File Reference

#include "shelltest.h"
#include "shell32_apitest_sub.h"
#include <assert.h>

Include dependency graph for shell32_apitest_sub.cpp:

Go to the source code of this file.

Macros
#define	EVENTS

Typedefs
typedef enum DIRTYPE	DIRTYPE

Enumerations
enum	DIRTYPE {   DIRTYPE_DESKTOP = 0 , DIRTYPE_DESKTOP_DIR , DIRTYPE_DRIVES , DIRTYPE_PRINTERS ,   DIRTYPE_DIR1 , DIRTYPE_MAX }

Functions
LPITEMIDLIST	DoGetPidl (INT iDir)
static BOOL	OnCreate (HWND hwnd)
static BOOL	InitSHCN (HWND hwnd)
static void	UnInitSHCN (HWND hwnd)
static void	OnCommand (HWND hwnd, UINT id)
static void	OnDestroy (HWND hwnd)
static BOOL	DoSendData (LONG lEvent, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
static void	DoShellNotify (HWND hwnd, PIDLIST_ABSOLUTE pidl1, PIDLIST_ABSOLUTE pidl2, LONG lEvent)
static INT_PTR	OnShellNotify (HWND hwnd, WPARAM wParam, LPARAM lParam)
static LRESULT CALLBACK	SubWindowProc (HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
INT APIENTRY	wWinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, INT nCmdShow)

Variables
static HWND	s_hMainWnd = NULL
static HWND	s_hSubWnd = NULL
static LPITEMIDLIST	s_pidl [DIRTYPE_MAX]
static UINT	s_uRegID = 0
static INT	s_iStage = -1

Macro Definition Documentation

EVENTS

#define EVENTS

Value:

                (SHCNE_CREATE | SHCNE_DELETE | SHCNE_MKDIR | SHCNE_RMDIR | \
                SHCNE_RENAMEFOLDER | SHCNE_RENAMEITEM | SHCNE_UPDATEDIR | SHCNE_UPDATEITEM)

Definition at line 29 of file shell32_apitest_sub.cpp.

Typedef Documentation

DIRTYPE

typedef enum DIRTYPE DIRTYPE

Enumeration Type Documentation

DIRTYPE

enum DIRTYPE

Enumerator
DIRTYPE_DESKTOP
DIRTYPE_DESKTOP_DIR
DIRTYPE_DRIVES
DIRTYPE_PRINTERS
DIRTYPE_DIR1
DIRTYPE_MAX

Definition at line 14 of file shell32_apitest_sub.cpp.

{
    DIRTYPE_DESKTOP = 0,
    DIRTYPE_DESKTOP_DIR,
    DIRTYPE_DRIVES,
    DIRTYPE_PRINTERS,
    DIRTYPE_DIR1,
    DIRTYPE_MAX
} DIRTYPE;

Function Documentation

DoGetPidl()

LPITEMIDLIST DoGetPidl ( INT  iDir )

inline

Definition at line 32 of file shell32_apitest_sub.cpp.

{
    LPITEMIDLIST ret = NULL;
 
    switch (iDir)
    {
        case DIRTYPE_DESKTOP:
        {
            SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOP, &ret);
            break;
        }
        case DIRTYPE_DESKTOP_DIR:
        {
            WCHAR szPath1[MAX_PATH];
            SHGetSpecialFolderPathW(NULL, szPath1, CSIDL_DESKTOPDIRECTORY, FALSE);
            ret = ILCreateFromPathW(szPath1);
            break;
        }
        case DIRTYPE_DRIVES:
        {
            SHGetSpecialFolderLocation(NULL, CSIDL_DRIVES, &ret);
            break;
        }
        case DIRTYPE_PRINTERS:
        {
            SHGetSpecialFolderLocation(NULL, CSIDL_PRINTERS, &ret);
            break;
        }
        case DIRTYPE_DIR1:
        {
            WCHAR szPath1[MAX_PATH];
            SHGetSpecialFolderPathW(NULL, szPath1, CSIDL_PERSONAL, FALSE); // My Documents
            PathAppendW(szPath1, L"_TESTDIR_1_");
            ret = ILCreateFromPathW(szPath1);
            break;
        }
        default:
        {
            assert(0);
            break;
        }
    }
 
    return ret;
}

Referenced by OnCreate().

DoSendData()

static BOOL DoSendData ( LONG  lEvent, LPCITEMIDLIST  pidl1, LPCITEMIDLIST  pidl2  )

static

Definition at line 252 of file shell32_apitest_sub.cpp.

{
    DWORD cbPidl1 = ILGetSize(pidl1), cbPidl2 = ILGetSize(pidl2);
    DWORD cbTotal = sizeof(lEvent) + sizeof(cbPidl1) + sizeof(cbPidl2) + cbPidl1 + cbPidl2;
    LPBYTE pbData = (LPBYTE)::LocalAlloc(LPTR, cbTotal);
    if (!pbData)
        return FALSE;
 
    LPBYTE pb = pbData;
 
    *(LONG*)pb = lEvent;
    pb += sizeof(lEvent);
 
    *(DWORD*)pb = cbPidl1;
    pb += sizeof(cbPidl1);
 
    *(DWORD*)pb = cbPidl2;
    pb += sizeof(cbPidl2);
 
    CopyMemory(pb, pidl1, cbPidl1);
    pb += cbPidl1;
 
    CopyMemory(pb, pidl2, cbPidl2);
    pb += cbPidl2;
 
    assert(INT(pb - pbData) == INT(cbTotal));
 
    COPYDATASTRUCT CopyData;
    CopyData.dwData = 0xBEEFCAFE;
    CopyData.cbData = cbTotal;
    CopyData.lpData = pbData;
    BOOL ret = (BOOL)::SendMessageW(s_hMainWnd, WM_COPYDATA, (WPARAM)s_hSubWnd, (LPARAM)&CopyData);
 
    ::LocalFree(pbData);
    return ret;
}

Referenced by DoShellNotify().

DoShellNotify()

static void DoShellNotify ( HWND  hwnd, PIDLIST_ABSOLUTE  pidl1, PIDLIST_ABSOLUTE  pidl2, LONG  lEvent  )

static

Definition at line 289 of file shell32_apitest_sub.cpp.

{
    if (s_iStage < 0)
        return;
 
    DoSendData(lEvent, pidl1, pidl2);
}

Referenced by OnShellNotify().

InitSHCN()

static BOOL InitSHCN ( HWND  hwnd )

static

Definition at line 88 of file shell32_apitest_sub.cpp.

{
    assert(0 <= s_iStage);
    assert(s_iStage < NUM_STAGE);
 
    SHChangeNotifyEntry entry;
    INT sources;
    LONG events;
    switch (s_iStage)
    {
        case 0:
        {
            entry.fRecursive = FALSE;
            entry.pidl = NULL;
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 1:
        {
            entry.fRecursive = TRUE;
            entry.pidl = NULL;
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 2:
        {
            entry.fRecursive = FALSE;
            entry.pidl = s_pidl[DIRTYPE_DESKTOP];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 3:
        {
            entry.fRecursive = TRUE;
            entry.pidl = s_pidl[DIRTYPE_DESKTOP];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 4:
        {
            entry.fRecursive = TRUE;
            entry.pidl = s_pidl[DIRTYPE_DESKTOP_DIR];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 5:
        {
            entry.fRecursive = FALSE;
            entry.pidl = s_pidl[DIRTYPE_DRIVES];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 6:
        {
            entry.fRecursive = TRUE;
            entry.pidl = s_pidl[DIRTYPE_DRIVES];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 7:
        {
            entry.fRecursive = TRUE;
            entry.pidl = s_pidl[DIRTYPE_PRINTERS];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 8:
        {
            entry.fRecursive = FALSE;
            entry.pidl = s_pidl[DIRTYPE_DIR1];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel;
            events = EVENTS;
            break;
        }
        case 9:
        {
            entry.fRecursive = TRUE;
            entry.pidl = s_pidl[DIRTYPE_DIR1];
            sources = SHCNRF_NewDelivery | SHCNRF_ShellLevel | SHCNRF_InterruptLevel |
                      SHCNRF_RecursiveInterrupt;
            events = EVENTS;
            break;
        }
        default:
        {
            assert(0);
            break;
        }
    }
 
    s_uRegID = SHChangeNotifyRegister(hwnd, sources, events, WM_SHELL_NOTIFY, 1, &entry);
    if (s_uRegID == 0)
        return FALSE;
 
    return TRUE;
}

Referenced by OnCommand().

OnCommand()

static void OnCommand ( HWND  hwnd, UINT  id  )

static

Definition at line 202 of file shell32_apitest_sub.cpp.

{
    switch (id)
    {
        case IDYES: // Start testing
        {
            s_hMainWnd = ::FindWindow(MAIN_CLASSNAME, MAIN_CLASSNAME);
            if (!s_hMainWnd)
            {
                ::DestroyWindow(hwnd);
                break;
            }
            s_iStage = 0;
            InitSHCN(hwnd);
            ::PostMessageW(s_hMainWnd, WM_COMMAND, IDYES, 0);
            break;
        }
        case IDRETRY: // New stage
        {
            UnInitSHCN(hwnd);
            ++s_iStage;
            InitSHCN(hwnd);
            ::PostMessageW(s_hMainWnd, WM_COMMAND, IDRETRY, 0);
            break;
        }
        case IDNO: // Quit
        {
            s_iStage = -1;
            UnInitSHCN(hwnd);
            ::DestroyWindow(hwnd);
            break;
        }
    }
}

Referenced by SubWindowProc().

OnCreate()

static BOOL OnCreate ( HWND  hwnd )

static

Definition at line 78 of file shell32_apitest_sub.cpp.

{
    s_hSubWnd = hwnd;
 
    for (INT i = 0; i < DIRTYPE_MAX; ++i)
        s_pidl[i] = DoGetPidl(i);
 
    return TRUE;
}

Referenced by SubWindowProc().

OnDestroy()

static void OnDestroy ( HWND  hwnd )

static

Definition at line 237 of file shell32_apitest_sub.cpp.

{
    UnInitSHCN(hwnd);
 
    for (auto& pidl : s_pidl)
    {
        CoTaskMemFree(pidl);
        pidl = NULL;
    }
 
    ::PostMessageW(s_hMainWnd, WM_COMMAND, IDNO, 0);
 
    PostQuitMessage(0);
}

Referenced by SubWindowProc().

OnShellNotify()

static INT_PTR OnShellNotify ( HWND  hwnd, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 297 of file shell32_apitest_sub.cpp.

{
    LONG lEvent;
    PIDLIST_ABSOLUTE *pidlAbsolute;
    HANDLE hLock = SHChangeNotification_Lock((HANDLE)wParam, (DWORD)lParam, &pidlAbsolute, &lEvent);
    if (hLock)
    {
        DoShellNotify(hwnd, pidlAbsolute[0], pidlAbsolute[1], lEvent);
        SHChangeNotification_Unlock(hLock);
    }
    else
    {
        pidlAbsolute = (PIDLIST_ABSOLUTE *)wParam;
        DoShellNotify(hwnd, pidlAbsolute[0], pidlAbsolute[1], lParam);
    }
    return TRUE;
}

Referenced by SubWindowProc().

SubWindowProc()

static LRESULT CALLBACK SubWindowProc ( HWND  hwnd, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 315 of file shell32_apitest_sub.cpp.

{
    switch (uMsg)
    {
        case WM_CREATE:
            return (OnCreate(hwnd) ? 0 : -1);
 
        case WM_COMMAND:
            OnCommand(hwnd, LOWORD(wParam));
            break;
 
        case WM_SHELL_NOTIFY:
            return OnShellNotify(hwnd, wParam, lParam);
 
        case WM_DESTROY:
            OnDestroy(hwnd);
            break;
 
        default:
            return ::DefWindowProcW(hwnd, uMsg, wParam, lParam);
    }
    return 0;
}

Referenced by wWinMain().

UnInitSHCN()

static void UnInitSHCN ( HWND  hwnd )

static

Definition at line 193 of file shell32_apitest_sub.cpp.

{
    if (s_uRegID)
    {
        SHChangeNotifyDeregister(s_uRegID);
        s_uRegID = 0;
    }
}

Referenced by OnCommand(), and OnDestroy().

wWinMain()

INT APIENTRY wWinMain	(	HINSTANCE	hInstance,
		HINSTANCE	hPrevInstance,
		LPWSTR	lpCmdLine,
		INT	nCmdShow
	)

Definition at line 340 of file shell32_apitest_sub.cpp.

{
    if (lstrcmpiW(lpCmdLine, L"") == 0 || lstrcmpiW(lpCmdLine, L"TEST") == 0)
        return 0;
 
    WNDCLASSW wc = { 0, SubWindowProc };
    wc.hInstance = hInstance;
    wc.hIcon = LoadIconW(NULL, IDI_APPLICATION);
    wc.hCursor = LoadCursorW(NULL, IDC_ARROW);
    wc.hbrBackground = (HBRUSH)(COLOR_3DFACE + 1);
    wc.lpszClassName = SUB_CLASSNAME;
    if (!RegisterClassW(&wc))
    {
        assert(0);
        return -1;
    }
 
    HWND hwnd = CreateWindowW(SUB_CLASSNAME, SUB_CLASSNAME, WS_OVERLAPPEDWINDOW,
                              CW_USEDEFAULT, CW_USEDEFAULT, 400, 100,
                              NULL, NULL, hInstance, NULL);
    if (!hwnd)
    {
        assert(0);
        return -2;
    }
 
    ShowWindow(hwnd, SW_SHOWNORMAL);
    UpdateWindow(hwnd);
 
    MSG msg;
    while (GetMessageW(&msg, NULL, 0, 0))
    {
        TranslateMessage(&msg);
        DispatchMessageW(&msg);
    }
 
    return 0;
}

Variable Documentation

s_hMainWnd

HWND s_hMainWnd = NULL

static

Definition at line 24 of file shell32_apitest_sub.cpp.

Referenced by DoSendData(), OnCommand(), and OnDestroy().

s_hSubWnd

HWND s_hSubWnd = NULL

static

Definition at line 24 of file shell32_apitest_sub.cpp.

Referenced by DoSendData(), and OnCreate().

s_iStage

INT s_iStage = -1

static

Definition at line 27 of file shell32_apitest_sub.cpp.

Referenced by DoShellNotify(), InitSHCN(), and OnCommand().

s_pidl

LPITEMIDLIST s_pidl[DIRTYPE_MAX]

static

Definition at line 25 of file shell32_apitest_sub.cpp.

Referenced by InitSHCN(), OnCreate(), and OnDestroy().

s_uRegID

UINT s_uRegID = 0

static

Definition at line 26 of file shell32_apitest_sub.cpp.

Referenced by InitSHCN(), and UnInitSHCN().