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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
|
// **************************************************************************
// * This file is part of the FreeFileSync project. It is distributed under *
// * GNU General Public License: http://www.gnu.org/licenses/gpl.html *
// * Copyright (C) 2008-2011 ZenJu (zhnmju123 AT gmx.de) *
// **************************************************************************
//
#include "notify_removal.h"
#include <set>
#include "last_error.h"
#include "Loki/ScopeGuard.h"
#include <algorithm>
#include <boost/bind.hpp>
#include <dbt.h>
using namespace zen;
/*
//convert bitmask into "real" drive-letter
Zstring getDriveFromMask(ULONG unitmask)
{
for (int i = 0; i < 26; ++i)
{
if (unitmask & 0x1)
return Zstring() + static_cast<DefaultChar>(DefaultChar('A') + i) + DefaultStr(":\\");
unitmask >>= 1;
}
return Zstring();
}
*/
namespace
{
bool messageProviderConstructed = false;
}
class MessageProvider //administrates a single dummy window to receive messages
{
public:
static MessageProvider& instance() //throw (FileError)
{
static MessageProvider inst;
messageProviderConstructed = true;
return inst;
}
class Listener
{
public:
virtual ~Listener() {}
virtual void onMessage(UINT message, WPARAM wParam, LPARAM lParam) = 0; //throw()!
};
void registerListener(Listener& l) { listener.insert(&l); }
void unregisterListener(Listener& l) { listener.erase(&l); } //don't unregister objects with static lifetime
HWND getWnd() const { return windowHandle; } //get handle in order to register additional notifications
private:
MessageProvider();
~MessageProvider();
MessageProvider(const MessageProvider&);
MessageProvider& operator=(const MessageProvider&);
static const wchar_t WINDOW_NAME[];
friend LRESULT CALLBACK topWndProc(HWND, UINT, WPARAM, LPARAM);
void processMessage(UINT message, WPARAM wParam, LPARAM lParam);
const HINSTANCE process;
HWND windowHandle;
std::set<Listener*> listener;
};
const wchar_t MessageProvider::WINDOW_NAME[] = L"E6AD5EB1-527B-4EEF-AC75-27883B233380"; //random name
LRESULT CALLBACK topWndProc(
HWND hwnd, //handle to window
UINT uMsg, //message identifier
WPARAM wParam, //first message parameter
LPARAM lParam) //second message parameter
{
if (messageProviderConstructed) //attention: this callback is triggered in the middle of singleton construction! It is a bad idea to to call back at this time!
try
{
MessageProvider::instance().processMessage(uMsg, wParam, lParam); //not supposed to throw
}
catch (...) {}
return ::DefWindowProc(hwnd, uMsg, wParam, lParam);
}
MessageProvider::MessageProvider() :
process(::GetModuleHandle(NULL)), //get program's module handle
windowHandle(NULL)
{
if (process == NULL)
throw zen::FileError(std::wstring(L"Could not start monitoring window notifications:") + "\n\n" + getLastErrorFormatted() + " (GetModuleHandle)");
//register the main window class
WNDCLASS wc = {};
wc.lpfnWndProc = topWndProc;
wc.hInstance = process;
wc.lpszClassName = WINDOW_NAME;
if (::RegisterClass(&wc) == 0)
throw zen::FileError(std::wstring(L"Could not start monitoring window notifications:") + "\n\n" + getLastErrorFormatted() + " (RegisterClass)");
Loki::ScopeGuard guardClass = Loki::MakeGuard(::UnregisterClass, WINDOW_NAME, process);
//create dummy-window
windowHandle = ::CreateWindow(
WINDOW_NAME, //LPCTSTR lpClassName OR ATOM in low-order word!
NULL, //LPCTSTR lpWindowName,
0, //DWORD dwStyle,
0, //int x,
0, //int y,
0, //int nWidth,
0, //int nHeight,
0, //note: we need a toplevel window to receive device arrival events, not a message-window (HWND_MESSAGE)!
NULL, //HMENU hMenu,
process, //HINSTANCE hInstance,
NULL); //LPVOID lpParam
if (windowHandle == NULL)
throw zen::FileError(std::wstring(L"Could not start monitoring window notifications:") + "\n\n" + getLastErrorFormatted() + " (CreateWindow)");
guardClass.Dismiss();
}
MessageProvider::~MessageProvider()
{
//clean-up in reverse order
::DestroyWindow(windowHandle);
::UnregisterClass(WINDOW_NAME, //LPCTSTR lpClassName OR ATOM in low-order word!
process); //HINSTANCE hInstance
}
void MessageProvider::processMessage(UINT message, WPARAM wParam, LPARAM lParam)
{
std::for_each(listener.begin(), listener.end(), boost::bind(&Listener::onMessage, _1, message, wParam, lParam));
}
//####################################################################################################
class NotifyRequestDeviceRemoval::Pimpl : private MessageProvider::Listener
{
public:
Pimpl(NotifyRequestDeviceRemoval& parent, HANDLE hDir) : //throw (FileError)
parent_(parent)
{
MessageProvider::instance().registerListener(*this); //throw (FileError)
//register handles to receive notifications
DEV_BROADCAST_HANDLE filter = {};
filter.dbch_size = sizeof(filter);
filter.dbch_devicetype = DBT_DEVTYP_HANDLE;
filter.dbch_handle = hDir;
hNotification = ::RegisterDeviceNotification(
MessageProvider::instance().getWnd(), //__in HANDLE hRecipient,
&filter, //__in LPVOID NotificationFilter,
DEVICE_NOTIFY_WINDOW_HANDLE); //__in DWORD Flags
if (hNotification == NULL)
{
const DWORD lastError = ::GetLastError();
if (lastError != ERROR_CALL_NOT_IMPLEMENTED && //fail on SAMBA share: this shouldn't be a showstopper!
lastError != ERROR_SERVICE_SPECIFIC_ERROR && //neither should be fail for "Pogoplug" mapped network drives
lastError != ERROR_INVALID_DATA) //this seems to happen for a NetDrive-mapped FTP server
throw zen::FileError(std::wstring(L"Could not register device removal notifications:") + "\n\n" + getLastErrorFormatted(lastError));
}
}
~Pimpl()
{
::UnregisterDeviceNotification(hNotification);
MessageProvider::instance().unregisterListener(*this);
}
private:
Pimpl(Pimpl&);
Pimpl& operator=(Pimpl&);
virtual void onMessage(UINT message, WPARAM wParam, LPARAM lParam) //throw()!
{
//DBT_DEVICEQUERYREMOVE example: http://msdn.microsoft.com/en-us/library/aa363427(v=VS.85).aspx
if (message == WM_DEVICECHANGE)
{
if ( wParam == DBT_DEVICEQUERYREMOVE ||
wParam == DBT_DEVICEQUERYREMOVEFAILED ||
wParam == DBT_DEVICEREMOVECOMPLETE)
{
PDEV_BROADCAST_HDR header = reinterpret_cast<PDEV_BROADCAST_HDR>(lParam);
if (header->dbch_devicetype == DBT_DEVTYP_HANDLE)
{
PDEV_BROADCAST_HANDLE body = reinterpret_cast<PDEV_BROADCAST_HANDLE>(lParam);
#ifdef __MINGW32__
const HDEVNOTIFY requestNotification = reinterpret_cast<HDEVNOTIFY>(body->dbch_hdevnotify);
#else
const HDEVNOTIFY requestNotification = body->dbch_hdevnotify;
#endif
if (requestNotification == hNotification) //is it for the notification we registered?
switch (wParam)
{
case DBT_DEVICEQUERYREMOVE:
parent_.onRequestRemoval(body->dbch_handle);
break;
case DBT_DEVICEQUERYREMOVEFAILED:
parent_.onRemovalFinished(body->dbch_handle, false);
break;
case DBT_DEVICEREMOVECOMPLETE:
parent_.onRemovalFinished(body->dbch_handle, true);
break;
}
}
}
}
}
NotifyRequestDeviceRemoval& parent_;
HDEVNOTIFY hNotification;
};
//####################################################################################################
NotifyRequestDeviceRemoval::NotifyRequestDeviceRemoval(HANDLE hDir)
{
pimpl.reset(new Pimpl(*this, hDir));
}
NotifyRequestDeviceRemoval::~NotifyRequestDeviceRemoval() {} //make sure ~auto_ptr() works with complete type
|
