Mercurial > irccd
annotate common/system.cpp @ 282:c234b8db1816 release-2.0
Plugin auth: fix error with quakenet, closes #506
author | David Demelier <markand@malikania.fr> |
---|---|
date | Tue, 01 Nov 2016 13:10:20 +0100 |
parents | 03068f5ed79d |
children |
rev | line source |
---|---|
0 | 1 /* |
2 * system.cpp -- platform dependent functions for system inspection | |
3 * | |
4 * Copyright (c) 2013-2016 David Demelier <markand@malikania.fr> | |
5 * | |
6 * Permission to use, copy, modify, and/or distribute this software for any | |
7 * purpose with or without fee is hereby granted, provided that the above | |
8 * copyright notice and this permission notice appear in all copies. | |
9 * | |
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES | |
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF | |
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR | |
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF | |
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |
17 */ | |
18 | |
19 #include <cstdlib> | |
20 #include <ctime> | |
21 #include <stdexcept> | |
22 | |
23 #include <irccd-config.h> | |
24 | |
25 #if defined(HAVE_SETPROGNAME) | |
26 # include <cstdlib> | |
27 #endif | |
28 | |
29 #if defined(IRCCD_SYSTEM_WINDOWS) | |
30 # include <sys/types.h> | |
31 # include <sys/timeb.h> | |
32 # include <Windows.h> | |
33 # include <Shlobj.h> | |
34 #else // All non Windows | |
35 #if defined(IRCCD_SYSTEM_MAC) | |
36 # include <sys/sysctl.h> | |
37 #endif | |
38 | |
39 #if defined(IRCCD_SYSTEM_LINUX) | |
40 # include <sys/sysinfo.h> | |
41 #endif | |
42 | |
43 # include <sys/utsname.h> | |
44 # include <sys/time.h> | |
45 # include <sys/types.h> | |
46 | |
47 # include <unistd.h> | |
48 | |
49 # include <cerrno> | |
50 # include <cstring> | |
51 # include <stdexcept> | |
52 # include <ctime> | |
53 | |
54 #endif | |
55 | |
56 /* For sys::setGid */ | |
57 #if defined(HAVE_SETGID) | |
58 # include <sys/types.h> | |
59 # include <unistd.h> | |
60 # include <grp.h> | |
61 #endif | |
62 | |
63 /* For sys::setUid */ | |
64 #if defined(HAVE_SETGID) | |
65 # include <sys/types.h> | |
66 # include <unistd.h> | |
67 # include <pwd.h> | |
68 #endif | |
69 | |
70 #include "filesystem.h" | |
71 #include "logger.h" | |
72 #include "system.h" | |
73 #include "util.h" | |
74 | |
75 namespace irccd { | |
76 | |
77 namespace sys { | |
78 | |
79 namespace { | |
80 | |
81 /* | |
82 * setHelper | |
83 * ------------------------------------------------------------------ | |
84 * | |
85 * This is an helper for setting the uid or gid. It accepts both numeric and string uid and gid. | |
86 * | |
87 * If a name is specified as uid/group, the lookup function will be called and must be getpwname or | |
88 * getgrname. Then, to get the id from the returned structure (struct passwd, struct group), the getter | |
89 * function will return either pw_uid or gr_gid. | |
90 * | |
91 * Finally, when the id is resolved, the setter function (setuid, setgid) will be called. | |
92 * | |
93 * @param typeName the type of id (uid or gid) | |
94 * @param value the value (numeric or name) | |
95 * @param lookup the lookup function to resolve the name (getpwnam or getgrnam) | |
96 * @param setter the function to apply the id (setuid or setgid) | |
97 * @param getter the function to get the id from the informal structure | |
98 */ | |
99 template <typename IntType, typename LookupFunc, typename SetterFunc, typename FieldGetter> | |
100 void setHelper(const std::string &typeName, const std::string &value, LookupFunc lookup, SetterFunc setter, FieldGetter getter) | |
101 { | |
102 IntType id; | |
103 | |
104 if (util::isNumber(value)) { | |
105 id = std::stoi(value); | |
106 } else { | |
107 auto info = lookup(value.c_str()); | |
108 | |
109 if (info == nullptr) { | |
110 log::warning() << "irccd: invalid " << typeName << ": " << std::strerror(errno) << std::endl; | |
111 return; | |
112 } else { | |
113 id = getter(info); | |
114 | |
115 log::debug() << "irccd: " << typeName << " " << value << " resolved to: " << id << std::endl; | |
116 } | |
117 } | |
118 | |
119 if (setter(id) < 0) | |
120 log::warning() << "irccd: could not set " << typeName << ": " << std::strerror(errno) << std::endl; | |
121 else | |
122 log::info() << "irccd: setting " << typeName << " to " << value << std::endl; | |
123 } | |
124 | |
125 /* | |
126 * XXX: the setprogname() function keeps a pointer without copying it so when main's argv is modified, we're not using | |
127 * the same name so create our own copy. | |
128 */ | |
129 | |
130 std::string programNameCopy; | |
131 | |
132 } // !namespace | |
133 | |
134 void setProgramName(std::string name) noexcept | |
135 { | |
136 programNameCopy = std::move(name); | |
137 | |
138 #if defined(HAVE_SETPROGNAME) | |
139 setprogname(programNameCopy.c_str()); | |
140 #endif | |
141 } | |
142 | |
143 const std::string &programName() noexcept | |
144 { | |
145 return programNameCopy; | |
146 } | |
147 | |
148 std::string name() | |
149 { | |
150 #if defined(IRCCD_SYSTEM_LINUX) | |
151 return "Linux"; | |
152 #elif defined(IRCCD_SYSTEM_WINDOWS) | |
153 return "Windows"; | |
154 #elif defined(IRCCD_SYSTEM_FREEBSD) | |
155 return "FreeBSD"; | |
156 #elif defined(IRCCD_SYSTEM_OPENBSD) | |
157 return "OpenBSD"; | |
158 #elif defined(IRCCD_SYSTEM_NETBSD) | |
159 return "NetBSD"; | |
160 #elif defined(IRCCD_SYSTEM_MAC) | |
161 return "Mac"; | |
162 #else | |
163 return "Unknown"; | |
164 #endif | |
165 } | |
166 | |
167 std::string version() | |
168 { | |
169 #if defined(IRCCD_SYSTEM_WINDOWS) | |
170 auto version = GetVersion(); | |
171 auto major = (DWORD)(LOBYTE(LOWORD(version))); | |
172 auto minor = (DWORD)(HIBYTE(LOWORD(version))); | |
173 | |
174 return std::to_string(major) + "." + std::to_string(minor); | |
175 #else | |
176 struct utsname uts; | |
177 | |
178 if (uname(&uts) < 0) | |
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
179 throw std::runtime_error(std::strerror(errno)); |
0 | 180 |
181 return std::string(uts.release); | |
182 #endif | |
183 } | |
184 | |
185 uint64_t uptime() | |
186 { | |
187 #if defined(IRCCD_SYSTEM_WINDOWS) | |
188 return ::GetTickCount64() / 1000; | |
189 #elif defined(IRCCD_SYSTEM_LINUX) | |
190 struct sysinfo info; | |
191 | |
192 if (sysinfo(&info) < 0) | |
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
193 throw std::runtime_error(std::strerror(errno)); |
0 | 194 |
195 return info.uptime; | |
196 #elif defined(IRCCD_SYSTEM_MAC) | |
197 struct timeval boottime; | |
198 size_t length = sizeof (boottime); | |
199 int mib[2] = { CTL_KERN, KERN_BOOTTIME }; | |
200 | |
201 if (sysctl(mib, 2, &boottime, &length, nullptr, 0) < 0) | |
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
202 throw std::runtime_error(std::strerror(errno)); |
0 | 203 |
204 time_t bsec = boottime.tv_sec, csec = time(nullptr); | |
205 | |
206 return difftime(csec, bsec); | |
207 #else | |
208 /* BSD */ | |
209 struct timespec ts; | |
210 | |
211 if (clock_gettime(CLOCK_UPTIME, &ts) < 0) | |
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
212 throw std::runtime_error(std::strerror(errno)); |
0 | 213 |
214 return ts.tv_sec; | |
215 #endif | |
216 } | |
217 | |
218 uint64_t ticks() | |
219 { | |
220 #if defined(IRCCD_SYSTEM_WINDOWS) | |
221 _timeb tp; | |
222 | |
223 _ftime(&tp); | |
224 | |
225 return tp.time * 1000LL + tp.millitm; | |
226 #else | |
227 struct timeval tp; | |
228 | |
229 gettimeofday(&tp, NULL); | |
230 | |
231 return tp.tv_sec * 1000LL + tp.tv_usec / 1000; | |
232 #endif | |
233 } | |
234 | |
235 std::string home() | |
236 { | |
237 #if defined(IRCCD_SYSTEM_WINDOWS) | |
238 char path[MAX_PATH]; | |
239 | |
240 if (SHGetFolderPathA(nullptr, CSIDL_LOCAL_APPDATA, nullptr, 0, path) != S_OK) | |
241 return ""; | |
242 | |
243 return std::string(path); | |
244 #else | |
245 return env("HOME"); | |
246 #endif | |
247 } | |
248 | |
249 std::string env(const std::string &var) | |
250 { | |
251 auto value = std::getenv(var.c_str()); | |
252 | |
253 if (value == nullptr) | |
254 return ""; | |
255 | |
256 return value; | |
257 } | |
258 | |
259 #if defined(HAVE_SETUID) | |
260 | |
261 void setUid(const std::string &value) | |
262 { | |
263 setHelper<uid_t>("uid", value, &getpwnam, &setuid, [] (const struct passwd *pw) { | |
264 return pw->pw_uid; | |
265 }); | |
266 } | |
267 | |
268 #endif | |
269 | |
270 #if defined(HAVE_SETGID) | |
271 | |
272 void setGid(const std::string &value) | |
273 { | |
274 setHelper<gid_t>("gid", value, &getgrnam, &setgid, [] (const struct group *gr) { | |
275 return gr->gr_gid; | |
276 }); | |
277 } | |
278 | |
279 #endif | |
280 | |
281 } // !sys | |
282 | |
283 } // !irccd |