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) 2013-2015,2017 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
#include <config.h>
#include <user_registry.h>
#include <user.h>
namespace user_chk {
UserRegistry::UserRegistry() {
}
UserRegistry::~UserRegistry(){
}
void
UserRegistry::addUser(UserPtr& user) {
if (!user) {
isc_throw (UserRegistryError, "UserRegistry cannot add blank user");
}
UserPtr found_user;
if ((found_user = findUser(user->getUserId()))) {
isc_throw (UserRegistryError, "UserRegistry duplicate user: "
<< user->getUserId());
}
users_[user->getUserId()] = user;
}
const UserPtr&
UserRegistry::findUser(const UserId& id) const {
static UserPtr empty;
UserMap::const_iterator it = users_.find(id);
if (it != users_.end()) {
const UserPtr tmp = (*it).second;
return ((*it).second);
}
return empty;
}
void
UserRegistry::removeUser(const UserId& id) {
static UserPtr empty;
UserMap::iterator it = users_.find(id);
if (it != users_.end()) {
users_.erase(it);
}
}
const UserPtr&
UserRegistry::findUser(const isc::dhcp::HWAddr& hwaddr) const {
UserId id(UserId::HW_ADDRESS, hwaddr.hwaddr_);
return (findUser(id));
}
const UserPtr&
UserRegistry::findUser(const isc::dhcp::DUID& duid) const {
UserId id(UserId::DUID, duid.getDuid());
return (findUser(id));
}
void UserRegistry::refresh() {
if (!source_) {
isc_throw(UserRegistryError,
"UserRegistry: cannot refresh, no data source");
}
// If the source isn't open, open it.
if (!source_->isOpen()) {
source_->open();
}
// Make a copy in case something goes wrong midstream.
UserMap backup(users_);
// Empty the registry then read users from source until source is empty.
clearall();
try {
UserPtr user;
while ((user = source_->readNextUser())) {
addUser(user);
}
} catch (const std::exception& ex) {
// Source was compromised so restore registry from backup.
users_ = backup;
// Close the source.
source_->close();
isc_throw (UserRegistryError, "UserRegistry: refresh failed during read"
<< ex.what());
}
// Close the source.
source_->close();
}
void UserRegistry::clearall() {
users_.clear();
}
void UserRegistry::setSource(UserDataSourcePtr& source) {
if (!source) {
isc_throw (UserRegistryError,
"UserRegistry: data source cannot be set to null");
}
source_ = source;
}
const UserDataSourcePtr& UserRegistry::getSource() {
return (source_);
}
} // namespace user_chk
|
