-
Notifications
You must be signed in to change notification settings - Fork 0
/
daemon.h
129 lines (112 loc) · 3.14 KB
/
daemon.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#pragma once
#ifndef MOUSE_CONFIG_DAEMON_H
#define MOUSE_CONFIG_DAEMON_H
#include "mouse.h"
#include "keyboard.h"
#include "timer.h"
#include <dbt.h>
#include <usbiodef.h>
#define WND_CLASS_NAME L"MouseConfigDaemonClass"
BOOL DoRegisterDeviceInterfaceToHwnd(
IN GUID InterfaceClassGuid,
IN HWND hWnd,
OUT HDEVNOTIFY *hDeviceNotify
) {
DEV_BROADCAST_DEVICEINTERFACE NotificationFilter = {
.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE),
.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE,
.dbcc_classguid = InterfaceClassGuid
};
*hDeviceNotify = RegisterDeviceNotification(
hWnd, // events recipient
&NotificationFilter, // type of device
DEVICE_NOTIFY_WINDOW_HANDLE // type of recipient handle
);
if (NULL == *hDeviceNotify) {
printError(TEXT("RegisterDeviceNotification"));
return FALSE;
}
return TRUE;
}
INT_PTR WINAPI WinProcCallback(
HWND hWnd,
UINT message,
WPARAM wParam,
LPARAM lParam
) {
LRESULT lRet = 1;
static HDEVNOTIFY hDeviceNotify;
static HWND hEditWnd;
switch (message) {
case WM_CREATE:
//
// This is the actual registration., In this example, registration
// should happen only once, at application startup when the window
// is created.
//
// If you were using a service, you would put this in your main code
// path as part of your service initialization.
//
if (!DoRegisterDeviceInterfaceToHwnd(
GUID_DEVINTERFACE_USB_DEVICE,
hWnd,
&hDeviceNotify)) {
// Terminate on failure.
printError(TEXT("DoRegisterDeviceInterfaceToHwnd"));
ExitProcess(1);
}
break;
case WM_SETFOCUS:
break;
case WM_SIZE:
break;
case WM_DEVICECHANGE: {
// Output some messages to the window.
switch (wParam) {
case DBT_DEVICEARRIVAL:
case DBT_DEVICEREMOVECOMPLETE:
if (mouseCfg != NULL && mouseCfgCount > 0) {
StartTimer();
}
break;
case DBT_DEVNODES_CHANGED:
break;
default:
break;
}
}
break;
case WM_CLOSE:
if (!UnregisterDeviceNotification(hDeviceNotify)) {
printError(L"UnregisterDeviceNotification");
}
DestroyWindow(hWnd);
break;
case WM_DESTROY:
PostQuitMessage(0);
break;
default:
// Send all other messages on to the default windows handler.
lRet = DefWindowProc(hWnd, message, wParam, lParam);
break;
}
return lRet;
}
BOOL InitWindowClass() {
WNDCLASSEXW wndClass = {
.cbSize = sizeof(WNDCLASSEXW),
.style = CS_OWNDC,
.hInstance = (HINSTANCE) (GetModuleHandle(NULL)),
.lpfnWndProc = (WNDPROC) (WinProcCallback),
.cbClsExtra = 0,
.cbWndExtra = 0,
.lpszClassName = WND_CLASS_NAME,
.lpszMenuName = NULL
};
if (!RegisterClassEx(&wndClass)) {
printError(L"RegisterClassEx");
return FALSE;
}
return TRUE;
}
#endif //MOUSE_CONFIG_DAEMON_H