indicdll.c File Reference

#include "../kbswitch.h"
#include "resource.h"
#include <wine/debug.h>

Include dependency graph for indicdll.c:

Go to the source code of this file.

Classes
struct	tagSHARED_DATA

Macros
#define	MUTEX_TIMEOUT_MS   (4 * 1000)

Typedefs
typedef struct tagSHARED_DATA	SHARED_DATA
typedef struct tagSHARED_DATA *	PSHARED_DATA

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (internat)
static BOOL	EnterProtectedSection (VOID)
static VOID	LeaveProtectedSection (VOID)
static VOID	PostMessageToMainWnd (UINT Msg, WPARAM wParam, LPARAM lParam)
static LRESULT CALLBACK	WinHookProc (INT code, WPARAM wParam, LPARAM lParam)
static LRESULT CALLBACK	ShellHookProc (INT code, WPARAM wParam, LPARAM lParam)
static BOOL	CheckVirtualKey (UINT vKey, UINT vKey0, UINT vKey1, UINT vKey2)
static LRESULT CALLBACK	KeyboardProc (INT code, WPARAM wParam, LPARAM lParam)
BOOL APIENTRY	KbSwitchSetHooks (_In_ BOOL bDoHook)
VOID APIENTRY	GetPenMenuData (PUINT pnID, PDWORD_PTR pdwItemData)
VOID APIENTRY	SetPenMenuData (_In_ UINT nID, _In_ DWORD_PTR dwItemData)
BOOL WINAPI	DllMain (IN HINSTANCE hinstDLL, IN DWORD dwReason, IN LPVOID lpvReserved)

Variables
HINSTANCE	g_hInstance = NULL
HANDLE	g_hShared = NULL
PSHARED_DATA	g_pShared = NULL
HANDLE	g_hMutex = NULL

Macro Definition Documentation

MUTEX_TIMEOUT_MS

#define MUTEX_TIMEOUT_MS   (4 * 1000)

Definition at line 31 of file indicdll.c.

Typedef Documentation

PSHARED_DATA

typedef struct tagSHARED_DATA * PSHARED_DATA

SHARED_DATA

typedef struct tagSHARED_DATA SHARED_DATA

Function Documentation

CheckVirtualKey()

static BOOL CheckVirtualKey ( UINT  vKey, UINT  vKey0, UINT  vKey1, UINT  vKey2  )

inlinestatic

Definition at line 104 of file indicdll.c.

{
    return vKey == vKey0 || vKey == vKey1 || vKey == vKey2;
}

Referenced by KeyboardProc().

DllMain()

BOOL WINAPI DllMain	(	IN HINSTANCE	hinstDLL,
		IN DWORD	dwReason,
		IN LPVOID	lpvReserved
	)

Definition at line 224 of file indicdll.c.

{
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
        {
            TRACE("DLL_PROCESS_ATTACH\n");
            g_hInstance = hinstDLL;
 
            g_hShared = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE,
                                          0, sizeof(SHARED_DATA), TEXT("InternatSHData"));
            if (!g_hShared)
            {
                ERR("!g_hShared\n");
                return FALSE;
            }
 
            BOOL bAlreadyExists = GetLastError() == ERROR_ALREADY_EXISTS;
            TRACE("bAlreadyExists: %d\n", bAlreadyExists);
 
            g_pShared = (PSHARED_DATA)MapViewOfFile(g_hShared, FILE_MAP_WRITE, 0, 0, 0);
            if (!g_pShared)
            {
                ERR("!g_pShared\n");
                return FALSE;
            }
 
            if (!bAlreadyExists)
                ZeroMemory(g_pShared, sizeof(*g_pShared));
 
            g_hMutex = CreateMutex(NULL, FALSE, TEXT("INDICDLL_PROTECTED"));
            if (!g_hMutex)
            {
                ERR("Failed to create mutex\n");
                return FALSE;
            }
            break;
        }
        case DLL_PROCESS_DETACH:
        {
            TRACE("DLL_PROCESS_DETACH\n");
            if (g_hMutex)
                CloseHandle(g_hMutex);
            UnmapViewOfFile(g_pShared);
            CloseHandle(g_hShared);
            break;
        }
    }
 
    return TRUE;
}

EnterProtectedSection()

static BOOL EnterProtectedSection ( VOID  )

inlinestatic

Definition at line 33 of file indicdll.c.

{
    DWORD dwWaitResult = WaitForSingleObject(g_hMutex, MUTEX_TIMEOUT_MS);
    if (dwWaitResult == WAIT_OBJECT_0)
        return TRUE;
 
    if (dwWaitResult == WAIT_TIMEOUT)
        WARN("Timeout while waiting for mutex.\n");
    else
        WARN("Failed to acquire mutex. Error code: %lu\n", GetLastError());
 
    return FALSE;
}

Referenced by GetPenMenuData(), KbSwitchSetHooks(), and SetPenMenuData().

GetPenMenuData()

VOID APIENTRY GetPenMenuData	(	PUINT	pnID,
		PDWORD_PTR	pdwItemData
	)

Definition at line 191 of file indicdll.c.

{
    if (EnterProtectedSection())
    {
        *pnID = g_pShared->nHotID;
        *pdwItemData = g_pShared->dwHotMenuItemData;
        LeaveProtectedSection();
    }
    else
    {
        WARN("EnterProtectedSection failed\n");
        *pnID = 0;
        *pdwItemData = 0;
    }
}

KbSwitchSetHooks()

BOOL APIENTRY KbSwitchSetHooks

(

_In_ BOOL

bDoHook

)

Definition at line 139 of file indicdll.c.

{
    TRACE("bDoHook: %d\n", bDoHook);
 
    if (!EnterProtectedSection())
        return FALSE;
 
    if (bDoHook)
    {
        g_pShared->hWinHook = SetWindowsHookEx(WH_CBT, WinHookProc, g_hInstance, 0);
        g_pShared->hShellHook = SetWindowsHookEx(WH_SHELL, ShellHookProc, g_hInstance, 0);
        g_pShared->hKeyboardHook = SetWindowsHookEx(WH_KEYBOARD, KeyboardProc, g_hInstance, 0);
 
        if (g_pShared->hWinHook &&
            g_pShared->hShellHook &&
            g_pShared->hKeyboardHook)
        {
            // Find kbswitch window if necessary
            if (!g_pShared->hKbSwitchWnd || !IsWindow(g_pShared->hKbSwitchWnd))
            {
                g_pShared->hKbSwitchWnd = FindWindow(INDICATOR_CLASS, NULL);
                TRACE("hKbSwitchWnd: %p\n", g_pShared->hKbSwitchWnd);
            }
 
            LeaveProtectedSection();
            return TRUE;
        }
    }
 
    /* Unhook */
    if (g_pShared->hKeyboardHook)
    {
        UnhookWindowsHookEx(g_pShared->hKeyboardHook);
        g_pShared->hKeyboardHook = NULL;
    }
    if (g_pShared->hShellHook)
    {
        UnhookWindowsHookEx(g_pShared->hShellHook);
        g_pShared->hShellHook = NULL;
    }
    if (g_pShared->hWinHook)
    {
        UnhookWindowsHookEx(g_pShared->hWinHook);
        g_pShared->hWinHook = NULL;
    }
 
    LeaveProtectedSection();
    return !bDoHook;
}

KeyboardProc()

static LRESULT CALLBACK KeyboardProc ( INT  code, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 110 of file indicdll.c.

{
    if (code < 0)
        return CallNextHookEx(g_pShared->hKeyboardHook, code, wParam, lParam);
 
    if (code == HC_ACTION)
    {
        UINT vKey = (UINT)wParam;
        LONG keyFlags = HIWORD(lParam);
        if (!(keyFlags & KF_UP) && !(keyFlags & KF_REPEAT))
        {
            BOOL bShiftPressed = (GetKeyState(VK_SHIFT) < 0);
            BOOL bAltPressed = (keyFlags & KF_ALTDOWN) || (GetKeyState(VK_MENU) < 0);
            BOOL bCtrlPressed = (GetKeyState(VK_CONTROL) < 0);
            // Detect Alt+Shift and Ctrl+Shift
            if ((bAltPressed && CheckVirtualKey(vKey, VK_SHIFT, VK_LSHIFT, VK_RSHIFT)) ||
                (bShiftPressed && CheckVirtualKey(vKey, VK_MENU, VK_LMENU, VK_RMENU)) ||
                (bCtrlPressed && CheckVirtualKey(vKey, VK_SHIFT, VK_LSHIFT, VK_RSHIFT)) ||
                (bShiftPressed && CheckVirtualKey(vKey, VK_CONTROL, VK_LCONTROL, VK_RCONTROL)))
            {
                PostMessageToMainWnd(WM_LANG_CHANGED, 0, 0);
            }
        }
    }
 
    return CallNextHookEx(g_pShared->hKeyboardHook, code, wParam, lParam);
}

Referenced by KbSwitchSetHooks().

LeaveProtectedSection()

static VOID LeaveProtectedSection ( VOID  )

inlinestatic

Definition at line 47 of file indicdll.c.

{
    ReleaseMutex(g_hMutex);
}

Referenced by GetPenMenuData(), KbSwitchSetHooks(), and SetPenMenuData().

PostMessageToMainWnd()

static VOID PostMessageToMainWnd ( UINT  Msg, WPARAM  wParam, LPARAM  lParam  )

inlinestatic

Definition at line 53 of file indicdll.c.

{
    if (g_pShared->hKbSwitchWnd)
        PostMessage(g_pShared->hKbSwitchWnd, Msg, wParam, lParam);
}

Referenced by KeyboardProc(), ShellHookProc(), and WinHookProc().

SetPenMenuData()

VOID APIENTRY SetPenMenuData	(	_In_ UINT	nID,
		_In_ DWORD_PTR	dwItemData
	)

Definition at line 209 of file indicdll.c.

{
    if (EnterProtectedSection())
    {
        g_pShared->nHotID = nID;
        g_pShared->dwHotMenuItemData = dwItemData;
        LeaveProtectedSection();
    }
    else
    {
        WARN("EnterProtectedSection failed\n");
    }
}

ShellHookProc()

static LRESULT CALLBACK ShellHookProc ( INT  code, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 81 of file indicdll.c.

{
    if (code < 0)
        return CallNextHookEx(g_pShared->hShellHook, code, wParam, lParam);
 
    switch (code)
    {
        case HSHELL_WINDOWACTIVATED:
        {
            PostMessageToMainWnd(WM_WINDOW_ACTIVATE, wParam, 0);
            break;
        }
        case HSHELL_LANGUAGE:
        {
            PostMessageToMainWnd(WM_LANG_CHANGED, wParam, lParam);
            break;
        }
    }
 
    return CallNextHookEx(g_pShared->hShellHook, code, wParam, lParam);
}

Referenced by KbSwitchSetHooks().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

internat

)

WinHookProc()

static LRESULT CALLBACK WinHookProc ( INT  code, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 60 of file indicdll.c.

{
    if (code < 0)
        return CallNextHookEx(g_pShared->hWinHook, code, wParam, lParam);
 
    switch (code)
    {
        case HCBT_ACTIVATE:
        case HCBT_SETFOCUS:
        {
            HWND hwndFocus = (HWND)wParam;
            if (hwndFocus && hwndFocus != g_pShared->hKbSwitchWnd)
                PostMessageToMainWnd(WM_WINDOW_ACTIVATE, (WPARAM)hwndFocus, 0);
            break;
        }
    }
 
    return CallNextHookEx(g_pShared->hWinHook, code, wParam, lParam);
}

Referenced by KbSwitchSetHooks().

Variable Documentation

g_hInstance

HINSTANCE g_hInstance = NULL

Definition at line 26 of file indicdll.c.

Referenced by DllMain(), and KbSwitchSetHooks().

g_hMutex

HANDLE g_hMutex = NULL

Definition at line 29 of file indicdll.c.

Referenced by DllMain(), EnterProtectedSection(), and LeaveProtectedSection().

g_hShared

HANDLE g_hShared = NULL

Definition at line 27 of file indicdll.c.

Referenced by DllMain().

g_pShared

PSHARED_DATA g_pShared = NULL

Definition at line 28 of file indicdll.c.

Referenced by DllMain(), GetPenMenuData(), KbSwitchSetHooks(), KeyboardProc(), PostMessageToMainWnd(), SetPenMenuData(), ShellHookProc(), and WinHookProc().