]>
Commit | Line | Data |
---|---|---|
b7eccf5f | 1 | #include "../core/schedule.h" |
475b3d9d | 2 | #include "../irc/irc.h" |
b7eccf5f P |
3 | #include "../lib/irc_string.h" |
4 | #include "../localuser/localuserchannel.h" | |
5 | #include "../control/control.h" | |
074a4529 | 6 | #include "../usercount/usercount.h" |
240bbd21 | 7 | #include "../lib/version.h" |
81f6c579 CP |
8 | #include "../serverlist/serverlist.h" |
9 | #include "../core/config.h" | |
10 | #include "../lib/strlfunc.h" | |
240bbd21 CP |
11 | |
12 | MODULE_VERSION("") | |
b7eccf5f P |
13 | |
14 | #include <stdlib.h> | |
15 | #include <string.h> | |
16 | #include <stdarg.h> | |
41b5727a | 17 | #include <stdio.h> |
475b3d9d | 18 | #include <sys/time.h> |
81f6c579 | 19 | #include <pcre.h> |
b7eccf5f P |
20 | |
21 | int cmd_serverlist(void *sender, int cargc, char **cargv); | |
22 | void serverlist_hook_newserver(int hook, void *arg); | |
23 | void serverlist_hook_lostserver(int hook, void *arg); | |
24 | int serverlist_versionreply(void *source, int cargc, char **cargv); | |
475b3d9d GB |
25 | void serverlist_pingservers(void *arg); |
26 | int serverlist_rpong(void *source, int cargc, char **cargv); | |
b7eccf5f | 27 | |
81f6c579 CP |
28 | const flag servertypeflags[] = { |
29 | { 'c', SERVERTYPEFLAG_CLIENT_SERVER }, | |
30 | { 'h', SERVERTYPEFLAG_HUB }, | |
31 | { 's', SERVERTYPEFLAG_SERVICE }, | |
32 | { 'Q', SERVERTYPEFLAG_CHANSERV }, | |
33 | { 'S', SERVERTYPEFLAG_SPAMSCAN }, | |
34 | { 'X', SERVERTYPEFLAG_CRITICAL_SERVICE }, | |
35 | { '\0', 0 } | |
36 | }; | |
37 | ||
b7eccf5f P |
38 | struct { |
39 | int used; | |
475b3d9d | 40 | int lag; |
b7eccf5f P |
41 | sstring *version1; |
42 | sstring *version2; | |
81f6c579 | 43 | flag_t type; |
b7eccf5f P |
44 | } serverinfo[MAXSERVERS]; |
45 | ||
46 | void serverlist_doversion(void); | |
47 | ||
81f6c579 CP |
48 | static sstring *s_server, *q_server; |
49 | static pcre *service_re, *hub_re, *not_client_re; | |
50 | ||
51 | static pcre *compilefree(sstring *re) { | |
52 | const char *err; | |
53 | int erroffset; | |
54 | pcre *r; | |
55 | ||
56 | r = pcre_compile(re->content, 0, &err, &erroffset, NULL); | |
57 | if(r == NULL) { | |
58 | Error("serverlist", ERR_WARNING, "Unable to compile RE %s (offset: %d, reason: %s)", re->content, erroffset, err); | |
59 | freesstring(re); | |
60 | return NULL; | |
61 | } | |
62 | ||
63 | freesstring(re); | |
64 | return r; | |
65 | } | |
b7eccf5f P |
66 | |
67 | void _init(void) { | |
68 | registercontrolhelpcmd("serverlist",NO_OPER,1,&cmd_serverlist,"Usage: serverlist [pattern]\nShows all currently connected servers"); | |
69 | /* hooks for serverlist */ | |
70 | registerhook(HOOK_SERVER_NEWSERVER, &serverlist_hook_newserver); | |
71 | registerhook(HOOK_SERVER_LOSTSERVER, &serverlist_hook_lostserver); | |
72 | int i; | |
73 | ||
81f6c579 CP |
74 | q_server = getcopyconfigitem("serverlist", "q_server", "CServe.quakenet.org", HOSTLEN); |
75 | s_server = getcopyconfigitem("serverlist", "s_server", "services2.uk.quakenet.org", HOSTLEN); | |
76 | ||
77 | service_re = compilefree(getcopyconfigitem("serverlist", "service_re", "^services\\d*\\..*$", 256)); | |
78 | hub_re = compilefree(getcopyconfigitem("serverlist", "hub_re", "^hub\\d*\\..*$", 256)); | |
b8bca1ed | 79 | not_client_re = compilefree(getcopyconfigitem("serverlist", "not_client_re", "^(testserv\\d*\\.).*$", 256)); |
81f6c579 | 80 | |
b7eccf5f P |
81 | for (i = 0; i < MAXSERVERS; i++) { |
82 | if (serverlist[i].linkstate == LS_LINKED) | |
83 | serverinfo[i].used = 1; | |
84 | else | |
85 | serverinfo[i].used = 0; | |
86 | ||
475b3d9d | 87 | serverinfo[i].lag = -1; |
b7eccf5f P |
88 | serverinfo[i].version1 = NULL; |
89 | serverinfo[i].version2 = NULL; | |
81f6c579 | 90 | serverinfo[i].type = getservertype(&serverlist[i]); |
b7eccf5f P |
91 | } |
92 | registernumerichandler(351, &serverlist_versionreply, 2); | |
475b3d9d | 93 | registerserverhandler("RO", &serverlist_rpong, 4); |
b7eccf5f | 94 | |
b2615f8b | 95 | schedulerecurring(time(NULL)+1, 0, 60, &serverlist_pingservers, NULL); |
b7eccf5f P |
96 | } |
97 | ||
98 | void _fini(void) { | |
99 | int i; | |
100 | for (i = 0; i < MAXSERVERS; i++) { | |
101 | if (serverinfo[i].used) { | |
102 | freesstring(serverinfo[i].version1); | |
103 | freesstring(serverinfo[i].version2); | |
104 | } | |
105 | } | |
106 | deregisternumerichandler(351, &serverlist_versionreply); | |
475b3d9d | 107 | deregisterserverhandler("RO", &serverlist_rpong); |
b7eccf5f P |
108 | |
109 | deregisterhook(HOOK_SERVER_NEWSERVER, &serverlist_hook_newserver); | |
110 | deregisterhook(HOOK_SERVER_LOSTSERVER, &serverlist_hook_lostserver); | |
111 | ||
112 | deregistercontrolcmd("serverlist",cmd_serverlist); | |
475b3d9d GB |
113 | |
114 | deleteschedule(NULL, &serverlist_pingservers, NULL); | |
ea3692b9 | 115 | |
81f6c579 CP |
116 | freesstring(q_server); |
117 | freesstring(s_server); | |
118 | pcre_free(service_re); | |
119 | pcre_free(hub_re); | |
120 | pcre_free(not_client_re); | |
b7eccf5f P |
121 | } |
122 | ||
123 | int cmd_serverlist(void *sender, int cargc, char **cargv) { | |
124 | nick *np = (nick*)sender; | |
125 | int a, i, ucount, acount, scount; | |
41b5727a | 126 | char lagstr[20]; |
81f6c579 | 127 | char buf[512]; |
b7eccf5f | 128 | |
6b129155 | 129 | controlreply(np, "%-7s %-35s %5s/%5s/%-7s %-7s %-7s %-20s %-8s %-20s", "Numeric", "Hostname", "ECl", "Cl", "MaxCl", "Flags", "Type", "Connected for", "Lag", "Version"); |
b7eccf5f P |
130 | |
131 | scount = acount = 0; | |
132 | ||
133 | for (i = 0; i < MAXSERVERS; i++) { | |
134 | if (serverlist[i].linkstate == LS_LINKED && (cargc < 1 || match2strings(cargv[0], serverlist[i].name->content))) { | |
135 | ucount = 0; | |
136 | ||
19f8fda5 | 137 | for (a = 0; a <= serverlist[i].maxusernum; a++) |
b7eccf5f P |
138 | if (servernicks[i][a] != NULL) |
139 | ucount++; | |
140 | ||
141 | acount += ucount; | |
142 | scount++; | |
143 | ||
41b5727a GB |
144 | if (serverinfo[i].lag == -1) |
145 | strcpy(lagstr, "-"); | |
146 | else | |
147 | snprintf(lagstr, sizeof(lagstr), "%d", serverinfo[i].lag); | |
148 | ||
81f6c579 CP |
149 | strlcpy(buf, printflags(serverinfo[i].type, servertypeflags), sizeof(buf)); |
150 | ||
6b129155 | 151 | controlreply(np, "%-7d %-35s %5d/%5d/%-7d %-7s %-7s %-20s %-8s %-20s - %s", i, serverlist[i].name->content, |
4f2ceb49 | 152 | servercount[i], ucount, serverlist[i].maxusernum, |
81f6c579 | 153 | printflags(serverlist[i].flags, smodeflags), buf, |
618cddb2 | 154 | longtoduration(getnettime() - serverlist[i].ts, 0), |
41b5727a | 155 | lagstr, |
b7eccf5f P |
156 | serverinfo[i].version1 ? serverinfo[i].version1->content : "Unknown", |
157 | serverinfo[i].version2 ? serverinfo[i].version2->content : "Unknown"); | |
158 | } | |
159 | } | |
160 | ||
161 | controlreply(np, "--- End of list. %d users and %d servers on the net.", acount, scount); | |
162 | ||
163 | /* update version info for next time */ | |
164 | serverlist_doversion(); | |
165 | ||
166 | return CMD_OK; | |
167 | } | |
168 | ||
169 | int serverlist_versionreply(void *source, int cargc, char **cargv) { | |
170 | int num; | |
171 | ||
172 | if (cargc < 6) | |
173 | return CMD_OK; | |
174 | ||
175 | num = numerictolong(cargv[0], 2); | |
176 | ||
177 | if (serverinfo[num].used) { | |
178 | freesstring(serverinfo[num].version1); | |
179 | freesstring(serverinfo[num].version2); | |
180 | ||
181 | serverinfo[num].version1 = getsstring(cargv[3], strlen(cargv[3])); | |
182 | serverinfo[num].version2 = getsstring(cargv[5], strlen(cargv[5])); | |
183 | } | |
184 | return CMD_OK; | |
185 | } | |
186 | ||
187 | void serverlist_doversion(void) { | |
188 | int i; | |
189 | char *num1, *numeric; | |
190 | ||
191 | if (mynick == NULL) | |
192 | return; | |
193 | ||
194 | for (i = 0; i < MAXSERVERS; i++) { | |
195 | if (serverlist[i].linkstate == LS_LINKED && serverinfo[i].version1 == NULL) { | |
196 | numeric = longtonumeric(mynick->numeric,5); | |
197 | num1 = (char*)malloc(strlen(numeric) + 1); /* bleh.. longtonumeric() is using static variables */ | |
198 | strcpy(num1, numeric); | |
199 | ||
200 | irc_send("%s V :%s", num1, longtonumeric(i,2)); | |
201 | free(num1); | |
202 | } | |
203 | } | |
204 | } | |
205 | ||
206 | void serverlist_hook_newserver(int hook, void *arg) { | |
207 | char *num1, *numeric; | |
074a4529 | 208 | long num = (long)arg; |
b7eccf5f | 209 | |
b7eccf5f | 210 | serverinfo[num].used = 1; |
b7eccf5f P |
211 | serverinfo[num].version1 = NULL; |
212 | serverinfo[num].version2 = NULL; | |
81f6c579 CP |
213 | serverinfo[num].type = getservertype(&serverlist[num]); |
214 | ||
215 | if (mynick == NULL) /* bleh this is buggy... doesn't do it when mynick appears */ | |
216 | return; | |
b7eccf5f P |
217 | |
218 | numeric = longtonumeric(mynick->numeric,5); | |
219 | num1 = (char*)malloc(strlen(numeric) + 1); /* bleh.. longtonumeric() is using static variables */ | |
220 | strcpy(num1, numeric); | |
221 | ||
222 | irc_send("%s V :%s", num1, longtonumeric(num,2)); | |
223 | ||
224 | free(num1); | |
225 | } | |
226 | ||
227 | void serverlist_hook_lostserver(int hook, void *arg) { | |
074a4529 | 228 | long num = (long)arg; |
b7eccf5f P |
229 | |
230 | serverinfo[num].used = 0; | |
231 | freesstring(serverinfo[num].version1); | |
232 | freesstring(serverinfo[num].version2); | |
233 | } | |
234 | ||
81f6c579 CP |
235 | flag_t getservertype(server *server) { |
236 | flag_t result = 0; | |
237 | char *server_name; | |
238 | int server_len; | |
239 | ||
240 | if(server == NULL || server->name == NULL) | |
241 | return 0; | |
242 | ||
243 | server_name = server->name->content; | |
244 | server_len = server->name->length; | |
245 | ||
246 | if(server->flags & SMODE_SERVICE) | |
247 | result|=SERVERTYPEFLAG_SERVICE; | |
248 | ||
249 | if(!strcmp(server_name, q_server->content)) { | |
250 | result|=SERVERTYPEFLAG_CHANSERV; | |
251 | result|=SERVERTYPEFLAG_CRITICAL_SERVICE; | |
252 | } else if(!strcmp(server_name, s_server->content)) { | |
253 | result|=SERVERTYPEFLAG_SPAMSCAN; | |
254 | result|=SERVERTYPEFLAG_CRITICAL_SERVICE; | |
255 | } else { | |
256 | if(service_re && (pcre_exec(service_re, NULL, server_name, server_len, 0, 0, NULL, 0) >= 0)) { | |
257 | /* matches service re */ | |
258 | if((server->flags & SMODE_SERVICE) == 0) { | |
259 | /* is not a service */ | |
260 | Error("serverlist", ERR_WARNING, "Non-service server (%s) matched service RE.", server_name); | |
261 | } | |
262 | } else { | |
263 | /* does not match service re */ | |
264 | if((server->flags & SMODE_SERVICE) != 0) { | |
265 | result|=SERVERTYPEFLAG_SERVICE; | |
266 | Error("serverlist", ERR_WARNING, "Service server (%s) that does not match service RE.", server_name); | |
267 | } | |
268 | } | |
269 | } | |
270 | ||
271 | if(hub_re && (pcre_exec(hub_re, NULL, server_name, server_len, 0, 0, NULL, 0) >= 0)) { | |
272 | if((server->flags & SMODE_HUB) != 0) { | |
273 | result|=SERVERTYPEFLAG_HUB; | |
274 | } else { | |
275 | Error("serverlist", ERR_WARNING, "Server matched hub re but isn't a hub (%s).", server_name); | |
276 | } | |
277 | } | |
278 | ||
81f6c579 CP |
279 | if(not_client_re && (pcre_exec(not_client_re, NULL, server_name, server_len, 0, 0, NULL, 0) >= 0)) { |
280 | /* noop */ | |
281 | } else if(result == 0) { | |
282 | result|=SERVERTYPEFLAG_CLIENT_SERVER; | |
283 | } | |
284 | ||
285 | return result; | |
286 | } | |
475b3d9d GB |
287 | |
288 | void serverlist_pingservers(void *arg) { | |
289 | int i; | |
290 | server *from, *to; | |
291 | char fnum[10], tnum[10]; | |
292 | struct timeval tv; | |
293 | ||
294 | if (!mynick) | |
295 | return; | |
296 | ||
297 | for(i=0;i<MAXSERVERS;i++) { | |
298 | to = &serverlist[i]; | |
299 | ||
300 | if (to->parent == -1) | |
301 | continue; | |
302 | ||
303 | from = &serverlist[to->parent]; | |
304 | ||
305 | if (to->linkstate != LS_LINKED || from->linkstate != LS_LINKED) | |
306 | continue; | |
307 | ||
308 | strcpy(fnum, longtonumeric(to->parent,2)); | |
309 | strcpy(tnum, longtonumeric(i, 2)); | |
310 | ||
311 | if (from->parent == -1) { /* Are we the source? */ | |
312 | (void) gettimeofday(&tv, NULL); | |
313 | irc_send("%s RI %s %s %jd %jd :RP", mynumeric->content, tnum, longtonumeric(mynick->numeric,5), (intmax_t)tv.tv_sec, (intmax_t)tv.tv_usec); | |
314 | } else { | |
315 | strcpy(fnum, longtonumeric(to->parent,2)); | |
316 | strcpy(tnum, longtonumeric(i, 2)); | |
317 | irc_send("%s RI %s %s :RP", longtonumeric(mynick->numeric,5), to->name->content, fnum); | |
318 | } | |
319 | } | |
320 | } | |
321 | ||
322 | int serverlist_rpong(void *source, int cargc, char **cargv) { | |
323 | int to, lag; | |
324 | struct timeval tv; | |
325 | ||
326 | if (cargc < 4) | |
327 | return CMD_ERROR; | |
328 | ||
329 | if (cargc == 5) { /* From target to source server */ | |
330 | if (strcmp(cargv[4], "RP") != 0) | |
331 | return CMD_OK; | |
332 | ||
333 | to = numerictolong(source, 2); | |
334 | ||
335 | (void) gettimeofday(&tv, NULL); | |
336 | lag = (tv.tv_sec - strtoull(cargv[2], NULL, 10)) * 1000 + (tv.tv_usec - strtoull(cargv[3], NULL, 10)) / 1000; | |
337 | } else { /* From source server to client */ | |
338 | if (strcmp(cargv[3], "RP") != 0) | |
339 | return CMD_OK; | |
340 | ||
341 | to = findserver(cargv[1]); | |
342 | ||
343 | if (to == -1) | |
344 | return CMD_ERROR; | |
345 | ||
346 | lag = atoi(cargv[2]); | |
347 | } | |
348 | ||
349 | serverinfo[to].lag = lag; | |
350 | ||
351 | return CMD_OK; | |
352 | } |