]> jfr.im git - irc/quakenet/newserv.git/blob - newsearch/newsearch.c
newsearch: Implement channel operator for whowassearch.
[irc/quakenet/newserv.git] / newsearch / newsearch.c
1 #include <stdio.h>
2 #include <stdarg.h>
3 #include <string.h>
4 #include <assert.h>
5
6 #include "../irc/irc_config.h"
7 #include "../lib/irc_string.h"
8 #include "../parser/parser.h"
9 #include "../control/control.h"
10 #include "../lib/splitline.h"
11 #include "../lib/version.h"
12 #include "../lib/stringbuf.h"
13 #include "../lib/strlfunc.h"
14 #include "../lib/array.h"
15 #include "newsearch.h"
16 #include "parser.h"
17
18 MODULE_VERSION("");
19
20 CommandTree *searchCmdTree;
21 searchList *globalterms = NULL;
22
23 int do_nicksearch(void *source, int cargc, char **cargv);
24 int do_chansearch(void *source, int cargc, char **cargv);
25 int do_usersearch(void *source, int cargc, char **cargv);
26 int do_whowassearch(void *source, int cargc, char **cargv);
27
28 void printnick_channels(searchCtx *, nick *, nick *);
29 void printchannel(searchCtx *, nick *, chanindex *);
30 void printchannel_topic(searchCtx *, nick *, chanindex *);
31 void printchannel_services(searchCtx *, nick *, chanindex *);
32
33 UserDisplayFunc defaultuserfn = printuser;
34 NickDisplayFunc defaultnickfn = printnick;
35 ChanDisplayFunc defaultchanfn = printchannel;
36 WhowasDisplayFunc defaultwhowasfn = printwhowas;
37
38 searchCmd *reg_nicksearch, *reg_chansearch, *reg_usersearch, *reg_whowassearch;
39 void displaycommandhelp(nick *, Command *);
40 void displaystrerror(replyFunc, nick *, const char *);
41
42 searchCmd *registersearchcommand(char *name, int level, CommandHandler cmd, void *defaultdisplayfunc) {
43 searchCmd *acmd;
44 searchList *sl;
45
46 registercontrolhelpfunccmd(name, NO_OPER,4, cmd, &displaycommandhelp);
47 acmd=(struct searchCmd *)malloc(sizeof(struct searchCmd));
48 if (!acmd) {
49 Error("newsearch", ERR_ERROR, "malloc failed: registersearchcommand");
50 return NULL;
51 }
52 acmd->handler = cmd;
53
54 acmd->name = getsstring( name, NSMAX_COMMAND_LEN);
55 acmd->outputtree = newcommandtree();
56 acmd->searchtree = newcommandtree();
57
58 addcommandtotree(searchCmdTree, name, 0, 0, (CommandHandler)acmd);
59
60 for (sl=globalterms; sl; sl=sl->next) {
61 addcommandexttotree(acmd->searchtree, sl->name->content, 0, 1, (CommandHandler)sl->cmd, sl->help);
62 }
63
64 return acmd;
65 }
66
67 void deregistersearchcommand(searchCmd *scmd) {
68 deregistercontrolcmd(scmd->name->content, (CommandHandler)scmd->handler);
69 destroycommandtree(scmd->outputtree);
70 destroycommandtree(scmd->searchtree);
71 deletecommandfromtree(searchCmdTree, scmd->name->content, (CommandHandler) scmd);
72 freesstring(scmd->name);
73 free(scmd);
74 }
75
76 void displaycommandhelp(nick *np, Command *cmd) {
77 int i,n,j,m;
78 Command *cmdlist[100], *acmdlist[100];
79 searchCmd *acmd;
80
81 n=getcommandlist(searchCmdTree,cmdlist,100);
82 for(i=0;i<n;i++) {
83 /* note: we may want to only deregister a command if we've already registered it, for now, try de-registering new commands anyway */
84 if ( ((searchCmd *)cmdlist[i]->handler)->handler != cmd->handler )
85 continue;
86 acmd = ((searchCmd *)(cmdlist[i]->handler));
87
88 controlreply(np, "Usage: [flags] <criteria>\n");
89 controlreply(np, "Flags:\n");
90 controlreply(np, " -l int : Limit number of rows of results\n");
91 controlreply(np, " -d string : a valid output format for the results\n");
92 controlreply(np, " -s subset : ipmask subset of network to search (only for node search)\n");
93 controlreply(np, " \n");
94 controlreply(np, "Available Output Formats:\n");
95
96 m=getcommandlist(acmd->outputtree,acmdlist,100);
97 for(j=0;j<m;j++) {
98 if ( controlpermitted( acmdlist[j]->level, np) ) {
99 char *help=(char *)acmdlist[j]->ext;
100 if ( help && help[0] != '\0')
101 controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help);
102 else
103 controlreply(np, "%s\n", acmdlist[j]->command->content );
104 }
105 }
106
107 controlreply(np, " \n");
108 controlreply(np, "Available Global Commands and Operators:\n" );
109 m=getcommandlist(acmd->searchtree,acmdlist,100);
110 for(j=0;j<m;j++) {
111 if ( acmdlist[j]->maxparams) {
112 char *help=(char *)acmdlist[j]->ext;
113 if ( help && help[0] != '\0')
114 controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help );
115 else
116 controlreply(np, "%s\n", acmdlist[j]->command->content );
117 }
118 }
119
120 controlreply(np, " \n");
121 controlreply(np, "Available Commands and Operators for %s:\n", acmd->name->content);
122
123 m=getcommandlist(acmd->searchtree,acmdlist,100);
124 for(j=0;j<m;j++) {
125 if ( !acmdlist[j]->maxparams && controlpermitted( acmdlist[j]->level, np) ) {
126 char *help=(char *)acmdlist[j]->ext;
127 if ( help && help[0] != '\0')
128 controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help );
129 else
130 controlreply(np, "%s\n", acmdlist[j]->command->content );
131 }
132 }
133 }
134 }
135
136 void regdisp( searchCmd *cmd, const char *name, void *handler, int level, char *help) {
137 addcommandexttotree(cmd->outputtree, name, level, 0, (CommandHandler) handler, (char *)help);
138 }
139
140 void unregdisp( searchCmd *cmd, const char *name, void *handler ) {
141 deletecommandfromtree(cmd->outputtree, name, (CommandHandler) handler);
142 }
143
144 void *findcommandinlist( searchList *sl, char *name){
145 while (sl) {
146 if(strcmp(sl->name->content,name) == 0 ) {
147 return sl;
148 }
149 sl=sl->next;
150 }
151 return NULL;
152 }
153
154 const char *parseError;
155 /* used for *_free functions that need to warn users of certain things
156 i.e. hitting too many users in a (kill) or (gline) */
157 nick *senderNSExtern;
158
159 void _init() {
160 searchCmdTree=newcommandtree();
161
162 reg_nicksearch = (searchCmd *)registersearchcommand("nicksearch",NO_OPER,&do_nicksearch, printnick);
163 reg_chansearch = (searchCmd *)registersearchcommand("chansearch",NO_OPER,&do_chansearch, printchannel);
164 reg_usersearch = (searchCmd *)registersearchcommand("usersearch",NO_OPER,&do_usersearch, printuser);
165 reg_whowassearch = (searchCmd *)registersearchcommand("whowassearch",NO_OPER,&do_whowassearch, printwhowas);
166
167 /* Boolean operations */
168 registerglobalsearchterm("and",and_parse, "usage: (and (X) (X))" );
169 registerglobalsearchterm("not",not_parse, "usage: (not (X))");
170 registerglobalsearchterm("or",or_parse, "usage: (or (X) (X))" );
171
172 registerglobalsearchterm("eq",eq_parse, "usage: (eq (X) Y)");
173
174 registerglobalsearchterm("lt",lt_parse, "usage: (lt (X) int)");
175 registerglobalsearchterm("gt",gt_parse, "usage: (gt (X) int)");
176
177 /* String operations */
178 registerglobalsearchterm("match",match_parse, "usage: (match (X) string)");
179 registerglobalsearchterm("regex",regex_parse, "usage: (regex (X) string)");
180 registerglobalsearchterm("length",length_parse, "usage: (length string)");
181
182 /* Nickname operations */
183 registersearchterm(reg_nicksearch, "hostmask",hostmask_parse, 0, "The user's nick!user@host; \"hostmask real\" returns nick!user@host\rreal");
184 registersearchterm(reg_whowassearch, "hostmask",hostmask_parse, 0, "The user's nick!user@host; \"hostmask real\" returns nick!user@host\rreal");
185
186 registersearchterm(reg_nicksearch, "realname",realname_parse, 0, "User's current realname"); /* nick only */
187 registersearchterm(reg_whowassearch, "realname",realname_parse, 0, "User's current realname"); /* nick only */
188
189 registersearchterm(reg_nicksearch, "away",away_parse, 0, "User's current away message");
190 registersearchterm(reg_whowassearch, "away",away_parse, 0, "User's current away message");
191 registersearchterm(reg_nicksearch, "authname",authname_parse, 0, "User's current authname or false");
192 registersearchterm(reg_whowassearch, "authname",authname_parse, 0, "User's current authname or false");
193 registersearchterm(reg_nicksearch, "authts",authts_parse, 0, "User's Auth timestamp");
194 registersearchterm(reg_whowassearch, "authts",authts_parse, 0, "User's Auth timestamp");
195 registersearchterm(reg_nicksearch, "ident",ident_parse, 0, "User's current ident");
196 registersearchterm(reg_whowassearch, "ident",ident_parse, 0, "User's current ident");
197 registersearchterm(reg_nicksearch, "host",host_parse, 0, "User's host, allow \"host real\" to match real host");
198 registersearchterm(reg_whowassearch, "host",host_parse, 0, "User's host, allow \"host real\" to match real host");
199 registersearchterm(reg_nicksearch, "channel",channel_parse, 0, "Valid Channel Name to match users against");
200 registersearchterm(reg_whowassearch, "channel",channel_parse, 0, "Valid Channel Name to match users against");
201 registersearchterm(reg_nicksearch, "timestamp",timestamp_parse, 0, "User's Timestamp");
202 registersearchterm(reg_whowassearch, "timestamp",timestamp_parse, 0, "User's Timestamp");
203 registersearchterm(reg_nicksearch, "country",country_parse, 0, "2 letter country code (data source is geoip)");
204 registersearchterm(reg_nicksearch, "ip",ip_parse, 0, "User's IP - ipv4 or ipv6 format as appropriate. Note: not 6to4");
205 registersearchterm(reg_whowassearch, "ip",ip_parse, 0, "User's IP - ipv4 or ipv6 format as appropriate. Note: not 6to4");
206 registersearchterm(reg_nicksearch, "channels",channels_parse, 0, "Channel Count"); /* nick only */
207 registersearchterm(reg_nicksearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)");
208 registersearchterm(reg_whowassearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)");
209 registersearchterm(reg_whowassearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)");
210 registersearchterm(reg_nicksearch, "authid",authid_parse, 0, "User's Auth ID");
211 registersearchterm(reg_whowassearch, "authid",authid_parse, 0, "User's Auth ID");
212 registersearchterm(reg_nicksearch, "cidr",cidr_parse, 0, "CIDR matching");
213 registersearchterm(reg_whowassearch, "cidr",cidr_parse, 0, "CIDR matching");
214 registersearchterm(reg_nicksearch, "ipvsix",ipv6_parse, 0, "IPv6 user");
215 registersearchterm(reg_whowassearch, "ipvsix",ipv6_parse, 0, "IPv6 user");
216 registersearchterm(reg_nicksearch, "message",message_parse, 0, "Last message");
217
218 /* Whowas operations */
219 registersearchterm(reg_whowassearch, "quit",quit_parse, 0, "User quit");
220 registersearchterm(reg_whowassearch, "killed",killed_parse, 0, "User was killed");
221 registersearchterm(reg_whowassearch, "renamed",renamed_parse, 0, "User changed nick");
222 registersearchterm(reg_whowassearch, "age",age_parse, 0, "Whowas record age in seconds");
223 registersearchterm(reg_whowassearch, "newnick",newnick_parse, 0, "New nick (for rename whowas records)");
224 registersearchterm(reg_whowassearch, "reason",reason_parse, 0, "Quit/kill reason");
225
226 /* Channel operations */
227 registersearchterm(reg_chansearch, "exists",exists_parse, 0, "Returns if channel exists on network. Note: newserv may store data on empty channels"); /* channel only */
228 registersearchterm(reg_chansearch, "services",services_parse, 0, ""); /* channel only */
229 registersearchterm(reg_chansearch, "size",size_parse, 0, "Channel user count"); /* channel only */
230 registersearchterm(reg_chansearch, "name",name_parse, 0, "Channel Name"); /* channel only */
231 registersearchterm(reg_chansearch, "topic",topic_parse, 0, "Channel topic"); /* channel only */
232 registersearchterm(reg_chansearch, "oppct",oppct_parse, 0, "Percentage Opped"); /* channel only */
233 registersearchterm(reg_chansearch, "cumodecount",cumodecount_parse, 0, "Count of users with given channel modes"); /* channel only */
234 registersearchterm(reg_chansearch, "cumodepct",cumodepct_parse, 0, "Percentage of users with given channel modes"); /* channel only */
235 registersearchterm(reg_chansearch, "uniquehostpct",hostpct_parse, 0, "uniquehost percent"); /* channel only */
236 registersearchterm(reg_chansearch, "authedpct",authedpct_parse, 0, "Percentage of authed users"); /* channel only */
237 registersearchterm(reg_chansearch, "kick",kick_parse, 0, "KICK users channels in newsearch result. Note: evaluation order"); /* channel only */
238
239 /* Nickname / channel operations */
240 registersearchterm(reg_chansearch, "modes",modes_parse, 0, "Channel Modes");
241 registersearchterm(reg_nicksearch, "modes",modes_parse, 0, "User Modes");
242 registersearchterm(reg_whowassearch, "modes",modes_parse, 0, "User Modes");
243 registersearchterm(reg_chansearch, "nick",nick_parse, 0, "Nickname");
244 registersearchterm(reg_nicksearch, "nick",nick_parse, 0, "Nickname");
245 registersearchterm(reg_whowassearch, "nick",nick_parse, 0, "Nickname");
246
247 /* Kill / gline parameters */
248 registersearchterm(reg_chansearch,"kill",kill_parse, 0, "KILL users in newsearch result. Note: evaluation order");
249 registersearchterm(reg_chansearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order");
250 registersearchterm(reg_nicksearch,"kill",kill_parse, 0, "KILL users in newsearch result. Note: evaluation order");
251 registersearchterm(reg_nicksearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order");
252 registersearchterm(reg_whowassearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order");
253
254 /* Iteration functionality */
255 registerglobalsearchterm("any",any_parse, "usage: any (generatorfn x) (fn ... (var x) ...)");
256 registerglobalsearchterm("all",all_parse, "usage: all (generatorfn x) (fn ... (var x) ...)");
257 registerglobalsearchterm("var",var_parse, "usage: var variable");
258
259 /* Iterable functions */
260 registersearchterm(reg_nicksearch, "channeliter",channeliter_parse, 0, "Channel Iterable function - usage: (any (channeliter x) (match (var x) #twilight*))"); /* nick only */
261 registersearchterm(reg_chansearch, "nickiter",nickiter_parse, 0, "Nick Iterable function - usage: (any (nickiter x) (match (var x) nickname*))"); /* channel only */
262
263 /* Functions that work on strings?! */
264 registersearchterm(reg_nicksearch, "cumodes", cumodes_parse, 0, "usage: (cumodes (var x) <modes>)");
265 registersearchterm(reg_chansearch, "cumodes", cumodes_parse, 0, "usage: (cumodes (var x) <modes>)");
266
267 /* Notice functionality */
268 registersearchterm(reg_chansearch,"notice",notice_parse, 0, "NOTICE users in newsearch result. Note: evaluation order");
269 registersearchterm(reg_nicksearch,"notice",notice_parse, 0, "NOTICE users in newsearch result. Note: evaluation order");
270
271 /* Nick output filters */
272 regdisp(reg_nicksearch,"default",printnick, 0, "");
273 regdisp(reg_nicksearch,"channels",printnick_channels, 0, "include channels in output");
274
275 /* Channel output filters */
276 regdisp(reg_chansearch,"default",printchannel, 0, "");
277 regdisp(reg_chansearch,"topic",printchannel_topic, 0, "display channel topics");
278 regdisp(reg_chansearch,"services",printchannel_services, 0, "display services on channels");
279
280 /* Nick output filters */
281 regdisp(reg_usersearch,"default",printuser, 0, "");
282
283 }
284
285 void _fini() {
286 searchList *sl, *psl;
287 int i,n;
288 Command *cmdlist[100];
289
290 sl=globalterms;
291 while (sl) {
292 psl = sl;
293 sl=sl->next;
294
295 n=getcommandlist(searchCmdTree,cmdlist,100);
296 for(i=0;i<n;i++) {
297 deregistersearchterm( (searchCmd *)cmdlist[i]->handler, psl->name->content, psl->cmd);
298 }
299
300 freesstring(psl->name);
301 if (psl->help)
302 free (psl->help);
303 free(psl);
304 }
305
306 deregistersearchcommand( reg_nicksearch );
307 deregistersearchcommand( reg_chansearch );
308 deregistersearchcommand( reg_usersearch );
309 destroycommandtree( searchCmdTree );
310 }
311
312 void registerglobalsearchterm(char *term, parseFunc parsefunc, char *help) {
313 int i,n;
314 Command *cmdlist[100];
315 searchList *sl = malloc(sizeof(searchList));
316 if (!sl) {
317 Error("newsearch", ERR_ERROR, "malloc failed: registerglobalsearchterm");
318 return;
319 }
320
321 sl->cmd = parsefunc;
322 sl->name = getsstring(term, NSMAX_COMMAND_LEN);
323 sl->next = NULL;
324 if (help) {
325 int len=strlen(help);
326 sl->help=(char *)malloc(len+1);
327 if(!sl->help) {
328 Error("newsearch", ERR_ERROR, "malloc failed: registerglobalsearchterm");
329 return;
330 }
331 strncpy(sl->help, help, len);
332 sl->help[len] = '\0';
333 } else {
334 sl->help=NULL;
335 }
336
337
338 if ( globalterms != NULL ) {
339 sl->next = globalterms;
340 }
341 globalterms = sl;
342
343 n=getcommandlist(searchCmdTree,cmdlist,100);
344 for(i=0;i<n;i++) {
345 /* maxparams is set to 1 to indicate a global term */
346 /* access level is set to 0 for all global terms */
347 addcommandexttotree( ((searchCmd *)cmdlist[i]->handler)->searchtree,term, 0, 1, (CommandHandler) parsefunc, help);
348 }
349 }
350
351 void deregisterglobalsearchterm(char *term, parseFunc parsefunc) {
352 int i,n;
353 Command *cmdlist[100];
354 searchList *sl, *psl=NULL;
355
356 for (sl=globalterms; sl; sl=sl->next) {
357 if ( strcmp( sl->name->content, term) == 0 ) {
358 break;
359 }
360 psl = sl;
361 }
362
363 if (sl) {
364 if( psl ) {
365 psl->next = sl->next;
366 }
367
368 n=getcommandlist(searchCmdTree,cmdlist,100);
369 for(i=0;i<n;i++) {
370 deletecommandfromtree( ((searchCmd *)cmdlist[i]->handler)->searchtree, term, (CommandHandler) parsefunc);
371 }
372 freesstring(sl->name);
373 free(sl);
374 }
375 }
376
377 void registersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc, int level, char *help) {
378 /* NOTE: global terms are added to the tree elsewhere as we set maxparams to 1 to indicate global */
379 addcommandexttotree(cmd->searchtree, term, level, 0, (CommandHandler) parsefunc, help);
380 }
381
382 void deregistersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc) {
383 /* NOTE: global terms are removed from the tree within deregisterglobalsearchterm */
384 deletecommandfromtree(cmd->searchtree, term, (CommandHandler) parsefunc);
385 }
386
387 static void controlwallwrapper(int level, char *format, ...) __attribute__ ((format (printf, 2, 3)));
388 static void controlwallwrapper(int level, char *format, ...) {
389 char buf[1024];
390 va_list ap;
391
392 va_start(ap, format);
393 vsnprintf(buf, sizeof(buf), format, ap);
394 controlwall(NO_OPER, level, "%s", buf);
395 va_end(ap);
396 }
397
398 int parseopts(int cargc, char **cargv, int *arg, int *limit, void **subset, void *display, CommandTree *sl, replyFunc reply, void *sender) {
399 char *ch;
400 Command *cmd;
401 struct irc_in_addr sin; unsigned char bits;
402
403 if (*cargv[0] == '-') {
404 /* options */
405 (*arg)++;
406
407 for (ch=cargv[0]+1;*ch;ch++) {
408 switch(*ch) {
409 case 'l':
410 if (cargc<*arg) {
411 reply(sender,"Error: -l switch requires an argument (for help, see help <searchcmd>)");
412 return CMD_ERROR;
413 }
414 *limit=strtoul(cargv[(*arg)++],NULL,10);
415 break;
416
417 case 'd':
418 if (cargc<*arg) {
419 reply(sender,"Error: -d switch requires an argument (for help, see help <searchcmd>)");
420 return CMD_ERROR;
421 }
422 cmd=findcommandintree(sl, cargv[*arg],1);
423 if (!cmd) {
424 reply(sender,"Error: unknown output format %s (for help, see help <searchcmd>)",cargv[*arg]);
425 return CMD_ERROR;
426 }
427 if ( !controlpermitted( cmd->level, sender) ) {
428 reply(sender,"Error: Access Denied for output format %s (for help, see help <searchcmd>)", cargv[*arg]);
429 return CMD_ERROR;
430 }
431 *((void **)display)=(void *)cmd->handler;
432 (*arg)++;
433 break;
434
435 case 's':
436 if (subset == NULL) {
437 reply(sender,"Error: -s switch not supported for this search.");
438 return CMD_ERROR;
439 }
440 if (cargc<*arg) {
441 reply(sender,"Error: -s switch requires an argument (for help, see help <searchcmd>)");
442 return CMD_ERROR;
443 }
444 if (ipmask_parse(cargv[*arg], &sin, &bits) == 0) {
445 reply(sender, "Error: Invalid CIDR mask supplied (for help, see help <searchcmd>)");
446 return CMD_ERROR;
447 }
448 *subset = (void *)refnode(iptree, &sin, bits);
449 (*arg)++;
450 break;
451
452 default:
453 reply(sender,"Unrecognised flag -%c. (for help, see help <searchcmd>)",*ch);
454 return CMD_ERROR;
455 }
456 }
457 }
458
459 return CMD_OK;
460 }
461
462 void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, searchCmd *cmd, nick *np, void *displayfn, int limit, array *targets) {
463 memset(ctx, 0, sizeof(searchCtx));
464
465 ctx->reply = replyfn;
466 ctx->wall = wallfn;
467 ctx->parser = searchfn;
468 ctx->arg = arg;
469 ctx->searchcmd = cmd;
470 ctx->sender = np;
471 ctx->limit = limit;
472 ctx->targets = targets;
473 ctx->displayfn = displayfn;
474 }
475
476 int do_nicksearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) {
477 nick *sender = source;
478 int limit=500;
479 int arg=0;
480 NickDisplayFunc display=defaultnickfn;
481 int ret;
482 parsertree *tree;
483
484 if (cargc<1) {
485 reply( sender, "Usage: [flags] <criteria>");
486 reply( sender, "For help, see help nicksearch");
487 return CMD_OK;
488 }
489
490 ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_nicksearch->outputtree, reply, sender);
491 if(ret != CMD_OK)
492 return ret;
493
494 if (arg>=cargc) {
495 reply(sender,"No search terms - aborting.");
496 return CMD_ERROR;
497 }
498
499 if (arg<(cargc-1)) {
500 rejoinline(cargv[arg],cargc-arg);
501 }
502
503 tree = parse_string(reg_nicksearch, cargv[arg]);
504 if(!tree) {
505 displaystrerror(reply, sender, cargv[arg]);
506 return CMD_ERROR;
507 }
508
509 ast_nicksearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
510
511 parse_free(tree);
512
513 return CMD_OK;
514 }
515
516 int do_nicksearch(void *source, int cargc, char **cargv) {
517 return do_nicksearch_real(controlreply, controlwallwrapper, source, cargc, cargv);
518 }
519
520 void nicksearch_exe(struct searchNode *search, searchCtx *ctx) {
521 int i, j, k;
522 int matches = 0;
523 unsigned int cmarker;
524 unsigned int tchans=0,uchans=0;
525 struct channel **cs;
526 nick *np, *sender = ctx->sender;
527 senderNSExtern = sender;
528 NickDisplayFunc display = ctx->displayfn;
529 int limit = ctx->limit;
530
531 /* Get a marker value to mark "seen" channels for unique count */
532 cmarker=nextchanmarker();
533
534 /* The top-level node needs to return a BOOL */
535 search=coerceNode(ctx, search, RETURNTYPE_BOOL);
536
537 for (i=0;i<NICKHASHSIZE;i++) {
538 for (np=nicktable[i], k = 0;ctx->targets ? (k < ctx->targets->cursi) : (np != NULL);np=np->next, k++) {
539 if (ctx->targets) {
540 np = ((nick **)ctx->targets->content)[k];
541 if (!np)
542 continue;
543 }
544
545 if ((search->exe)(ctx, search, np)) {
546 /* Add total channels */
547 tchans += np->channels->cursi;
548
549 /* Check channels for uniqueness */
550 cs=(channel **)np->channels->content;
551 for (j=0;j<np->channels->cursi;j++) {
552 if (cs[j]->index->marker != cmarker) {
553 cs[j]->index->marker=cmarker;
554 uchans++;
555 }
556 }
557
558 if (matches<limit)
559 display(ctx, sender, np);
560
561 if (matches==limit)
562 ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
563 matches++;
564 }
565 }
566
567 if (ctx->targets)
568 break;
569 }
570
571 ctx->reply(sender,"--- End of list: %d matches; users were on %u channels (%u unique, %.1f average clones)",
572 matches, tchans, uchans, (float)tchans/uchans);
573 }
574
575 int do_whowassearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) {
576 nick *sender = source;
577 int limit=500;
578 int arg=0;
579 WhowasDisplayFunc display=defaultwhowasfn;
580 int ret;
581 parsertree *tree;
582
583 if (cargc<1) {
584 reply( sender, "Usage: [flags] <criteria>");
585 reply( sender, "For help, see help whowassearch");
586 return CMD_OK;
587 }
588
589 ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_whowassearch->outputtree, reply, sender);
590 if(ret != CMD_OK)
591 return ret;
592
593 if (arg>=cargc) {
594 reply(sender,"No search terms - aborting.");
595 return CMD_ERROR;
596 }
597
598 if (arg<(cargc-1)) {
599 rejoinline(cargv[arg],cargc-arg);
600 }
601
602 tree = parse_string(reg_whowassearch, cargv[arg]);
603 if(!tree) {
604 displaystrerror(reply, sender, cargv[arg]);
605 return CMD_ERROR;
606 }
607
608 ast_whowassearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
609
610 parse_free(tree);
611
612 return CMD_OK;
613 }
614
615 int do_whowassearch(void *source, int cargc, char **cargv) {
616 return do_whowassearch_real(controlreply, controlwallwrapper, source, cargc, cargv);
617 }
618
619 void whowassearch_exe(struct searchNode *search, searchCtx *ctx) {
620 int i, matches = 0;
621 whowas *ww;
622 nick *sender = ctx->sender;
623 senderNSExtern = sender;
624 WhowasDisplayFunc display = ctx->displayfn;
625 int limit = ctx->limit;
626
627 assert(!ctx->targets);
628
629 /* The top-level node needs to return a BOOL */
630 search=coerceNode(ctx, search, RETURNTYPE_BOOL);
631
632 for (i = whowasoffset; i < whowasoffset + WW_MAXENTRIES; i++) {
633 ww = &whowasrecs[i % WW_MAXENTRIES];
634
635 if (ww->type == WHOWAS_UNUSED)
636 continue;
637
638 /* Note: We're passing the nick to the filter function. The original
639 * whowas record is in the nick's ->next field. */
640 if ((search->exe)(ctx, search, &ww->nick)) {
641 if (matches<limit)
642 display(ctx, sender, ww);
643
644 if (matches==limit)
645 ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
646 matches++;
647 }
648 }
649
650 ctx->reply(sender,"--- End of list: %d matches", matches);
651 }
652
653 int do_chansearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) {
654 nick *sender = source;
655 int limit=500;
656 int arg=0;
657 ChanDisplayFunc display=defaultchanfn;
658 int ret;
659 parsertree *tree;
660
661 if (cargc<1) {
662 reply( sender, "Usage: [flags] <criteria>");
663 reply( sender, "For help, see help chansearch");
664 return CMD_OK;
665 }
666
667 ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_chansearch->outputtree, reply, sender);
668 if(ret != CMD_OK)
669 return ret;
670
671 if (arg>=cargc) {
672 reply(sender,"No search terms - aborting.");
673 return CMD_ERROR;
674 }
675
676 if (arg<(cargc-1)) {
677 rejoinline(cargv[arg],cargc-arg);
678 }
679
680 tree = parse_string(reg_chansearch, cargv[arg]);
681 if(!tree) {
682 displaystrerror(reply, sender, cargv[arg]);
683 return CMD_ERROR;
684 }
685
686 ast_chansearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
687
688 parse_free(tree);
689
690 return CMD_OK;
691 }
692
693 int do_chansearch(void *source, int cargc, char **cargv) {
694 return do_chansearch_real(controlreply, controlwallwrapper, source, cargc, cargv);
695 }
696
697 void chansearch_exe(struct searchNode *search, searchCtx *ctx) {
698 int i;
699 chanindex *cip;
700 int matches = 0;
701 nick *sender = ctx->sender;
702 senderNSExtern = sender;
703 ChanDisplayFunc display = ctx->displayfn;
704 int limit = ctx->limit;
705
706 assert(!ctx->targets);
707
708 search=coerceNode(ctx, search, RETURNTYPE_BOOL);
709
710 for (i=0;i<CHANNELHASHSIZE;i++) {
711 for (cip=chantable[i];cip;cip=cip->next) {
712 if ((search->exe)(ctx, search, cip)) {
713 if (matches<limit)
714 display(ctx, sender, cip);
715 if (matches==limit)
716 ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
717 matches++;
718 }
719 }
720 }
721
722 ctx->reply(sender,"--- End of list: %d matches", matches);
723 }
724
725 int do_usersearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) {
726 nick *sender = source;
727 int limit=500;
728 int arg=0;
729 UserDisplayFunc display=defaultuserfn;
730 int ret;
731 parsertree *tree;
732
733 if (cargc<1) {
734 reply( sender, "Usage: [flags] <criteria>");
735 reply( sender, "For help, see help usersearch");
736 return CMD_OK;
737 }
738
739 ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_usersearch->outputtree, reply, sender);
740 if(ret != CMD_OK)
741 return ret;
742
743 if (arg>=cargc) {
744 reply(sender,"No search terms - aborting.");
745 return CMD_ERROR;
746 }
747
748 if (arg<(cargc-1)) {
749 rejoinline(cargv[arg],cargc-arg);
750 }
751
752 tree = parse_string(reg_usersearch, cargv[arg]);
753 if(!tree) {
754 displaystrerror(reply, sender, cargv[arg]);
755 return CMD_ERROR;
756 }
757
758 ast_usersearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
759
760 parse_free(tree);
761
762 return CMD_OK;
763 }
764
765 int do_usersearch(void *source, int cargc, char **cargv) {
766 return do_usersearch_real(controlreply, controlwallwrapper, source, cargc, cargv);
767 }
768
769 void usersearch_exe(struct searchNode *search, searchCtx *ctx) {
770 int i;
771 authname *aup;
772 int matches = 0;
773 nick *sender = ctx->sender;
774 int limit = ctx->limit;
775 UserDisplayFunc display = ctx->displayfn;
776 senderNSExtern = sender;
777
778 assert(!ctx->targets);
779
780 search=coerceNode(ctx, search, RETURNTYPE_BOOL);
781
782 for (i=0;i<AUTHNAMEHASHSIZE;i++) {
783 for (aup=authnametable[i];aup;aup=aup->next) {
784 if ((search->exe)(ctx, search, aup)) {
785 if (matches<limit)
786 display(ctx, sender, aup);
787 if (matches==limit)
788 ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
789 matches++;
790 }
791 }
792 }
793
794 ctx->reply(sender,"--- End of list: %d matches", matches);
795 }
796
797 /* Free a coerce node */
798 void free_coerce(searchCtx *ctx, struct searchNode *thenode) {
799 struct coercedata *cd=thenode->localdata;
800
801 cd->child->free(ctx, cd->child);
802 free(thenode->localdata);
803 free(thenode);
804 }
805
806 /* Free a coerce node with a stringbuf allocated */
807 void free_coercestring(searchCtx *ctx, struct searchNode *thenode) {
808 free(((struct coercedata *)thenode->localdata)->u.stringbuf);
809 free_coerce(ctx, thenode);
810 }
811
812 /* exe_tostr_null: return the constant string */
813 void *exe_tostr_null(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
814 struct coercedata *cd=thenode->localdata;
815
816 return cd->u.stringbuf;
817 }
818
819 /* exe_val_null: return the constant value */
820 void *exe_val_null(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
821 struct coercedata *cd=thenode->localdata;
822
823 return (void *)cd->u.val;
824 }
825
826 /* Lots of very dull type conversion functions */
827 void *exe_inttostr(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
828 struct coercedata *cd=thenode->localdata;
829
830 sprintf(cd->u.stringbuf, "%lu", (unsigned long)(cd->child->exe)(ctx, cd->child, theinput));
831
832 return cd->u.stringbuf;
833 }
834
835 void *exe_booltostr(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
836 struct coercedata *cd=thenode->localdata;
837
838 if ((cd->child->exe)(ctx, cd->child, theinput)) {
839 sprintf(cd->u.stringbuf,"1");
840 } else {
841 cd->u.stringbuf[0]='\0';
842 }
843
844 return cd->u.stringbuf;
845 }
846
847 void *exe_strtoint(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
848 struct coercedata *cd=thenode->localdata;
849
850 return (void *)strtoul((cd->child->exe)(ctx,cd->child,theinput),NULL,10);
851 }
852
853 void *exe_booltoint(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
854 struct coercedata *cd=thenode->localdata;
855
856 /* Don't need to do anything */
857 return (cd->child->exe)(ctx, cd->child, theinput);
858 }
859
860 void *exe_strtobool(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
861 struct coercedata *cd=thenode->localdata;
862 char *ch=(cd->child->exe)(ctx, cd->child, theinput);
863
864 if (!ch || *ch=='\0' || (*ch=='0' && ch[1]=='\0')) {
865 return (void *)0;
866 } else {
867 return (void *)1;
868 }
869 }
870
871 void *exe_inttobool(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
872 struct coercedata *cd=thenode->localdata;
873
874 if ((cd->child->exe)(ctx, cd->child, theinput)) {
875 return (void *)1;
876 } else {
877 return (void *)0;
878 }
879 }
880
881 struct searchNode *coerceNode(searchCtx *ctx, struct searchNode *thenode, int type) {
882 struct searchNode *anode;
883 struct coercedata *cd;
884
885 /* You can't coerce a NULL */
886 if (!thenode)
887 return NULL;
888
889 /* No effort required to coerce to the same type */
890 if (type==(thenode->returntype & RETURNTYPE_TYPE))
891 return thenode;
892
893 anode=(struct searchNode *)malloc(sizeof(struct searchNode));
894 anode->localdata=cd=(struct coercedata *)malloc(sizeof(struct coercedata));
895 cd->child=thenode;
896 anode->returntype=type; /* We'll return what they want, always */
897 anode->free=free_coerce;
898
899 switch(type) {
900 case RETURNTYPE_STRING:
901 /* For a string we'll need a buffer */
902 /* A 64-bit number prints out to 20 digits, this leaves some slack */
903 cd->u.stringbuf=malloc(25);
904 anode->free=free_coercestring;
905
906 switch(thenode->returntype & RETURNTYPE_TYPE) {
907 default:
908 case RETURNTYPE_INT:
909 if (thenode->returntype & RETURNTYPE_CONST) {
910 /* Constant node: sort it out now */
911 sprintf(cd->u.stringbuf, "%lu", (unsigned long)thenode->exe(ctx, thenode, NULL));
912 anode->exe=exe_tostr_null;
913 anode->returntype |= RETURNTYPE_CONST;
914 } else {
915 /* Variable data */
916 anode->exe=exe_inttostr;
917 }
918 break;
919
920 case RETURNTYPE_BOOL:
921 if (thenode->returntype & RETURNTYPE_CONST) {
922 /* Constant bool value */
923 if (thenode->exe(ctx, thenode,NULL)) {
924 /* True! */
925 sprintf(cd->u.stringbuf, "1");
926 } else {
927 cd->u.stringbuf[0] = '\0';
928 }
929 anode->exe=exe_tostr_null;
930 anode->returntype |= RETURNTYPE_CONST;
931 } else {
932 /* Variable bool value */
933 anode->exe=exe_booltostr;
934 }
935 break;
936 }
937 break;
938
939 case RETURNTYPE_INT:
940 /* we want an int */
941 switch (thenode->returntype & RETURNTYPE_TYPE) {
942 case RETURNTYPE_STRING:
943 if (thenode->returntype & RETURNTYPE_CONST) {
944 cd->u.val=strtoul((thenode->exe)(ctx, thenode, NULL), NULL, 10);
945 anode->exe=exe_val_null;
946 anode->returntype |= RETURNTYPE_CONST;
947 } else {
948 anode->exe=exe_strtoint;
949 }
950 break;
951
952 default:
953 case RETURNTYPE_BOOL:
954 if (thenode->returntype & RETURNTYPE_CONST) {
955 if ((thenode->exe)(ctx, thenode,NULL))
956 cd->u.val=1;
957 else
958 cd->u.val=0;
959
960 anode->exe=exe_val_null;
961 anode->returntype |= RETURNTYPE_CONST;
962 } else {
963 anode->exe=exe_booltoint;
964 }
965 break;
966 }
967 break;
968
969 default:
970 case RETURNTYPE_BOOL:
971 /* we want a bool */
972 switch (thenode->returntype & RETURNTYPE_TYPE) {
973 case RETURNTYPE_STRING:
974 if (thenode->returntype & RETURNTYPE_CONST) {
975 char *rv=(char *)((thenode->exe)(ctx, thenode, NULL));
976 if (!rv || *rv=='\0' || (*rv=='0' && rv[1]=='\0'))
977 cd->u.val=0;
978 else
979 cd->u.val=1;
980
981 anode->exe=exe_val_null;
982 anode->returntype |= RETURNTYPE_CONST;
983 } else {
984 anode->exe=exe_strtobool;
985 }
986 break;
987
988 default:
989 case RETURNTYPE_INT:
990 if (thenode->returntype & RETURNTYPE_CONST) {
991 if ((thenode->exe)(ctx, thenode,NULL))
992 cd->u.val=1;
993 else
994 cd->u.val=0;
995
996 anode->exe=exe_val_null;
997 anode->returntype |= RETURNTYPE_CONST;
998 } else {
999 anode->exe=exe_inttobool;
1000 }
1001 break;
1002 }
1003 break;
1004 }
1005
1006 return anode;
1007 }
1008
1009 /* Literals always return constant strings... */
1010 void *literal_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
1011 if (thenode->localdata)
1012 return ((sstring *)thenode->localdata)->content;
1013 else
1014 return "";
1015 }
1016
1017 void literal_free(searchCtx *ctx, struct searchNode *thenode) {
1018 freesstring(thenode->localdata);
1019 free(thenode);
1020 }
1021
1022 void nssnprintf(char *buf, size_t size, const char *format, nick *np) {
1023 StringBuf b;
1024 const char *p;
1025 char *c;
1026 char hostbuf[512];
1027
1028 if(size == 0)
1029 return;
1030
1031 b.buf = buf;
1032 b.capacity = size;
1033 b.len = 0;
1034
1035 for(p=format;*p;p++) {
1036 if(*p != '%') {
1037 if(!sbaddchar(&b, *p))
1038 break;
1039 continue;
1040 }
1041 p++;
1042 if(*p == '\0')
1043 break;
1044 if(*p == '%') {
1045 if(!sbaddchar(&b, *p))
1046 break;
1047 continue;
1048 }
1049
1050 c = NULL;
1051 switch(*p) {
1052 case 'n':
1053 c = np->nick; break;
1054 case 'i':
1055 c = np->ident; break;
1056 case 'h':
1057 c = np->host->name->content; break;
1058 case 'I':
1059 snprintf(hostbuf, sizeof(hostbuf), "%s", IPtostr(np->ipaddress));
1060 c = hostbuf;
1061 break;
1062 case 'u':
1063 snprintf(hostbuf, sizeof(hostbuf), "%s!%s@%s", np->nick, np->ident, IPtostr(np->ipaddress));
1064 c = hostbuf;
1065 break;
1066 default:
1067 c = "(bad format specifier)";
1068 }
1069 if(c)
1070 if(!sbaddstr(&b, c))
1071 break;
1072 }
1073
1074 sbterminate(&b);
1075
1076 /* not required */
1077 /*
1078 buf[size-1] = '\0';
1079 */
1080 }
1081
1082 static char *var_tochar(searchCtx *ctx, char *arg, searchNode **variable) {
1083 *variable = ctx->parser(ctx, arg);
1084 if (!(*variable = coerceNode(ctx, *variable, RETURNTYPE_STRING)))
1085 return NULL;
1086
1087 if(!((*variable)->returntype & RETURNTYPE_CONST)) {
1088 parseError = "only constant variables allowed";
1089 ((*variable)->free)(ctx, *variable);
1090 return NULL;
1091 }
1092
1093 return (char *)((*variable)->exe)(ctx, *variable, NULL);
1094 }
1095
1096 void free_val_null(searchCtx *ctx, struct searchNode *thenode) {
1097 }
1098
1099 struct searchVariable *var_register(searchCtx *ctx, char *arg, int type) {
1100 searchNode *variable;
1101 struct searchVariable *us;
1102 char *var;
1103 int i;
1104
1105 if(ctx->lastvar >= MAX_VARIABLES) {
1106 parseError = "Maximum number of variables reached";
1107 return NULL;
1108 }
1109
1110 us = &ctx->vars[ctx->lastvar];
1111
1112 var = var_tochar(ctx, arg, &variable);
1113 if(!var)
1114 return NULL;
1115
1116 strlcpy(us->name, var, sizeof(us->name));
1117 (variable->free)(ctx, variable);
1118
1119 for(i=0;i<ctx->lastvar;i++) {
1120 if(!strcmp(us->name, ctx->vars[i].name)) {
1121 parseError = "variable name already in use";
1122 return NULL;
1123 }
1124 }
1125
1126 ctx->lastvar++;
1127 us->data.returntype = type;
1128 us->data.localdata = &us->cdata;
1129 us->data.exe = exe_val_null;
1130 us->data.free = free_val_null;
1131
1132 us->cdata.child = NULL;
1133 return us;
1134 }
1135
1136 searchNode *var_get(searchCtx *ctx, char *arg) {
1137 searchNode *variable, *found = NULL;
1138 int i;
1139 char *var = var_tochar(ctx, arg, &variable);
1140 if(!var)
1141 return NULL;
1142
1143 for(i=0;i<ctx->lastvar;i++) {
1144 if(!strcmp(var, ctx->vars[i].name)) {
1145 found = &ctx->vars[i].data;
1146 break;
1147 }
1148 }
1149 (variable->free)(ctx, variable);
1150
1151 if(!found)
1152 parseError = "variable not found";
1153 return found;
1154 }
1155
1156 void var_setstr(struct searchVariable *v, char *data) {
1157 v->cdata.u.stringbuf = data;
1158 }
1159
1160 void displaystrerror(replyFunc reply, nick *np, const char *input) {
1161 char buf[515];
1162
1163 if((parseStrErrorPos >= 0) && (parseStrErrorPos < sizeof(buf) - 3)) {
1164 int i;
1165
1166 for(i=0;i<parseStrErrorPos;i++)
1167 buf[i] = ' ';
1168
1169 buf[i++] = '^';
1170 buf[i] = '\0';
1171
1172 reply(np, "%s", input);
1173 reply(np, "%s", buf);
1174 }
1175
1176 reply(np, "Parse error: %s", parseStrError);
1177 }
1178
1179 struct searchNode *argtoconststr(char *command, searchCtx *ctx, char *arg, char **p) {
1180 struct searchNode *c;
1181 static char errorbuf[512];
1182
1183 c = ctx->parser(ctx, arg);
1184 if (!(c = coerceNode(ctx, c, RETURNTYPE_STRING))) {
1185 snprintf(errorbuf, sizeof(errorbuf), "%s: unable to coerce argument to string", command);
1186 parseError = errorbuf;
1187 return NULL;
1188 }
1189
1190 if (!(c->returntype & RETURNTYPE_CONST)) {
1191 snprintf(errorbuf, sizeof(errorbuf), "%s: constant argument required", command);
1192 parseError = errorbuf;
1193 (c->free)(ctx, c);
1194 return NULL;
1195 }
1196
1197 if(p)
1198 *p = (char *)(c->exe)(ctx, c, NULL);
1199
1200 return c;
1201 }