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
|
/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
*
* This is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this software; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
* USA.
*/
// -=- RegConfig.cxx
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <malloc.h>
#include <rfb_win32/RegConfig.h>
#include <rfb/LogWriter.h>
//#include <rdr/HexOutStream.h>
using namespace rfb;
using namespace rfb::win32;
static LogWriter vlog("RegConfig");
RegConfig::RegConfig(EventManager* em)
: eventMgr(em), event(CreateEvent(nullptr, TRUE, FALSE, nullptr)),
callback(nullptr)
{
if (em->addEvent(event, this))
eventMgr = em;
}
RegConfig::~RegConfig() {
if (eventMgr)
eventMgr->removeEvent(event);
}
bool RegConfig::setKey(const HKEY rootkey, const char* keyname) {
try {
key.createKey(rootkey, keyname);
processEvent(event);
return true;
} catch (rdr::Exception& e) {
vlog.debug("%s", e.str());
return false;
}
}
void RegConfig::loadRegistryConfig(RegKey& key) {
DWORD i = 0;
try {
while (1) {
const char *name = key.getValueName(i++);
if (!name) break;
std::string value = key.getRepresentation(name);
if (!Configuration::setParam(name, value.c_str()))
vlog.info("unable to process %s", name);
}
} catch (rdr::SystemException& e) {
if (e.err != 6)
vlog.error("%s", e.str());
}
}
void RegConfig::processEvent(HANDLE /*event*/) {
vlog.info("registry changed");
// Reinstate the registry change notifications
ResetEvent(event);
key.awaitChange(true, REG_NOTIFY_CHANGE_NAME | REG_NOTIFY_CHANGE_LAST_SET, event);
// Load settings
loadRegistryConfig(key);
// Notify the callback, if supplied
if (callback)
callback->regConfigChanged();
}
RegConfigThread::RegConfigThread() : config(&eventMgr), thread_id(-1) {
}
RegConfigThread::~RegConfigThread() {
PostThreadMessage(thread_id, WM_QUIT, 0, 0);
wait();
}
bool RegConfigThread::start(const HKEY rootKey, const char* keyname) {
if (config.setKey(rootKey, keyname)) {
Thread::start();
while (thread_id == (DWORD)-1)
Sleep(0);
return true;
}
return false;
}
void RegConfigThread::worker() {
BOOL result = 0;
MSG msg;
thread_id = GetCurrentThreadId();
while ((result = eventMgr.getMessage(&msg, nullptr, 0, 0)) > 0) {}
if (result < 0)
throw rdr::SystemException("RegConfigThread failed", GetLastError());
}
|