X-Git-Url: https://jfr.im/git/irc/quakenet/newserv.git/blobdiff_plain/f33f3f52f58acdd2b9b6796713ac8f24d9881545..80e32fcd996025093c28b304d91d376b72c1496e:/newsearch/newsearch.c diff --git a/newsearch/newsearch.c b/newsearch/newsearch.c index 6726213c..0b08bdfb 100644 --- a/newsearch/newsearch.c +++ b/newsearch/newsearch.c @@ -1,7 +1,6 @@ #include #include #include -#include "newsearch.h" #include "../irc/irc_config.h" #include "../lib/irc_string.h" @@ -11,17 +10,19 @@ #include "../lib/version.h" #include "../lib/stringbuf.h" #include "../lib/strlfunc.h" +#include "../lib/array.h" +#include "newsearch.h" +#include "parser.h" MODULE_VERSION(""); -CommandTree *searchTree; -CommandTree *chanOutputTree; -CommandTree *nickOutputTree; -CommandTree *userOutputTree; +CommandTree *searchCmdTree; +searchList *globalterms = NULL; int do_nicksearch(void *source, int cargc, char **cargv); int do_chansearch(void *source, int cargc, char **cargv); int do_usersearch(void *source, int cargc, char **cargv); +int do_whowassearch(void *source, int cargc, char **cargv); void printnick_channels(searchCtx *, nick *, nick *); void printchannel(searchCtx *, nick *, chanindex *); @@ -31,140 +32,357 @@ void printchannel_services(searchCtx *, nick *, chanindex *); UserDisplayFunc defaultuserfn = printuser; NickDisplayFunc defaultnickfn = printnick; ChanDisplayFunc defaultchanfn = printchannel; +WhowasDisplayFunc defaultwhowasfn = printwhowas; -void registersearchterm(char *term, parseFunc parsefunc); -void deregistersearchterm(char *term, parseFunc parsefunc); +searchCmd *reg_nicksearch, *reg_chansearch, *reg_usersearch, *reg_whowassearch; +void displaycommandhelp(nick *, Command *); +void displaystrerror(replyFunc, nick *, const char *); -void regchandisp(const char *name, ChanDisplayFunc handler) { - addcommandtotree(chanOutputTree, name, 0, 0, (CommandHandler)handler); -} +searchCmd *registersearchcommand(char *name, int level, CommandHandler cmd, void *defaultdisplayfunc) { + searchCmd *acmd; + searchList *sl; -void unregchandisp(const char *name, ChanDisplayFunc handler) { - deletecommandfromtree(chanOutputTree, name, (CommandHandler)handler); -} + registercontrolhelpfunccmd(name, NO_OPER,4, cmd, &displaycommandhelp); + acmd=(struct searchCmd *)malloc(sizeof(struct searchCmd)); + if (!acmd) { + Error("newsearch", ERR_ERROR, "malloc failed: registersearchcommand"); + return NULL; + } + acmd->handler = cmd; -void regnickdisp(const char *name, NickDisplayFunc handler) { - addcommandtotree(nickOutputTree, name, 0, 0, (CommandHandler)handler); + acmd->name = getsstring( name, NSMAX_COMMAND_LEN); + acmd->outputtree = newcommandtree(); + acmd->searchtree = newcommandtree(); + + addcommandtotree(searchCmdTree, name, 0, 0, (CommandHandler)acmd); + + for (sl=globalterms; sl; sl=sl->next) { + addcommandexttotree(acmd->searchtree, sl->name->content, 0, 1, (CommandHandler)sl->cmd, sl->help); + } + + return acmd; } -void unregnickdisp(const char *name, NickDisplayFunc handler) { - deletecommandfromtree(nickOutputTree, name, (CommandHandler)handler); +void deregistersearchcommand(searchCmd *scmd) { + deregistercontrolcmd(scmd->name->content, (CommandHandler)scmd->handler); + destroycommandtree(scmd->outputtree); + destroycommandtree(scmd->searchtree); + deletecommandfromtree(searchCmdTree, scmd->name->content, (CommandHandler) scmd); + freesstring(scmd->name); + free(scmd); } -void reguserdisp(const char *name, UserDisplayFunc handler) { - addcommandtotree(userOutputTree, name, 0, 0, (CommandHandler)handler); +void displaycommandhelp(nick *np, Command *cmd) { + int i,n,j,m; + Command *cmdlist[100], *acmdlist[100]; + searchCmd *acmd; + + n=getcommandlist(searchCmdTree,cmdlist,100); + for(i=0;ihandler)->handler != cmd->handler ) + continue; + acmd = ((searchCmd *)(cmdlist[i]->handler)); + + controlreply(np, "Usage: [flags] \n"); + controlreply(np, "Flags:\n"); + controlreply(np, " -l int : Limit number of rows of results\n"); + controlreply(np, " -d string : a valid output format for the results\n"); + controlreply(np, " -s subset : ipmask subset of network to search (only for node search)\n"); + controlreply(np, " \n"); + controlreply(np, "Available Output Formats:\n"); + + m=getcommandlist(acmd->outputtree,acmdlist,100); + for(j=0;jlevel, np) ) { + char *help=(char *)acmdlist[j]->ext; + if ( help && help[0] != '\0') + controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help); + else + controlreply(np, "%s\n", acmdlist[j]->command->content ); + } + } + + controlreply(np, " \n"); + controlreply(np, "Available Global Commands and Operators:\n" ); + m=getcommandlist(acmd->searchtree,acmdlist,100); + for(j=0;jmaxparams) { + char *help=(char *)acmdlist[j]->ext; + if ( help && help[0] != '\0') + controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help ); + else + controlreply(np, "%s\n", acmdlist[j]->command->content ); + } + } + + controlreply(np, " \n"); + controlreply(np, "Available Commands and Operators for %s:\n", acmd->name->content); + + m=getcommandlist(acmd->searchtree,acmdlist,100); + for(j=0;jmaxparams && controlpermitted( acmdlist[j]->level, np) ) { + char *help=(char *)acmdlist[j]->ext; + if ( help && help[0] != '\0') + controlreply(np, "%-10s: %s\n", acmdlist[j]->command->content, help ); + else + controlreply(np, "%s\n", acmdlist[j]->command->content ); + } + } + } } -void unreguserdisp(const char *name, UserDisplayFunc handler) { - deletecommandfromtree(userOutputTree, name, (CommandHandler)handler); +void regdisp( searchCmd *cmd, const char *name, void *handler, int level, char *help) { + addcommandexttotree(cmd->outputtree, name, level, 0, (CommandHandler) handler, (char *)help); +} + +void unregdisp( searchCmd *cmd, const char *name, void *handler ) { + deletecommandfromtree(cmd->outputtree, name, (CommandHandler) handler); } +void *findcommandinlist( searchList *sl, char *name){ + while (sl) { + if(strcmp(sl->name->content,name) == 0 ) { + return sl; + } + sl=sl->next; + } + return NULL; +} + const char *parseError; /* used for *_free functions that need to warn users of certain things i.e. hitting too many users in a (kill) or (gline) */ nick *senderNSExtern; void _init() { - searchTree=newcommandtree(); - chanOutputTree=newcommandtree(); - nickOutputTree=newcommandtree(); - userOutputTree=newcommandtree(); + searchCmdTree=newcommandtree(); + + reg_nicksearch = (searchCmd *)registersearchcommand("nicksearch",NO_OPER,&do_nicksearch, printnick); + reg_chansearch = (searchCmd *)registersearchcommand("chansearch",NO_OPER,&do_chansearch, printchannel); + reg_usersearch = (searchCmd *)registersearchcommand("usersearch",NO_OPER,&do_usersearch, printuser); + reg_whowassearch = (searchCmd *)registersearchcommand("whowassearch",NO_OPER,&do_whowassearch, printwhowas); /* Boolean operations */ - registersearchterm("and",and_parse); - registersearchterm("not",not_parse); - registersearchterm("or",or_parse); + registerglobalsearchterm("and",and_parse, "usage: (and (X) (X))" ); + registerglobalsearchterm("not",not_parse, "usage: (not (X))"); + registerglobalsearchterm("or",or_parse, "usage: (or (X) (X))" ); - registersearchterm("eq",eq_parse); + registerglobalsearchterm("eq",eq_parse, "usage: (eq (X) Y)"); - registersearchterm("lt",lt_parse); - registersearchterm("gt",gt_parse); + registerglobalsearchterm("lt",lt_parse, "usage: (lt (X) int)"); + registerglobalsearchterm("gt",gt_parse, "usage: (gt (X) int)"); /* String operations */ - registersearchterm("match",match_parse); - registersearchterm("regex",regex_parse); - registersearchterm("length",length_parse); + registerglobalsearchterm("match",match_parse, "usage: (match (X) string)"); + registerglobalsearchterm("regex",regex_parse, "usage: (regex (X) string)"); + registerglobalsearchterm("length",length_parse, "usage: (length string)"); /* Nickname operations */ - registersearchterm("hostmask",hostmask_parse); - registersearchterm("realname",realname_parse); - registersearchterm("authname",authname_parse); - registersearchterm("authts",authts_parse); - registersearchterm("ident",ident_parse); - registersearchterm("host",host_parse); - registersearchterm("channel",channel_parse); - registersearchterm("timestamp",timestamp_parse); - registersearchterm("country",country_parse); - registersearchterm("ip",ip_parse); - registersearchterm("channels",channels_parse); - registersearchterm("server",server_parse); - registersearchterm("authid",authid_parse); + registersearchterm(reg_nicksearch, "hostmask",hostmask_parse, 0, "The user's nick!user@host; \"hostmask real\" returns nick!user@host\rreal"); + registersearchterm(reg_whowassearch, "hostmask",hostmask_parse, 0, "The user's nick!user@host; \"hostmask real\" returns nick!user@host\rreal"); + + registersearchterm(reg_nicksearch, "realname",realname_parse, 0, "User's current realname"); /* nick only */ + registersearchterm(reg_whowassearch, "realname",realname_parse, 0, "User's current realname"); /* nick only */ + + registersearchterm(reg_nicksearch, "away",away_parse, 0, "User's current away message"); + registersearchterm(reg_whowassearch, "away",away_parse, 0, "User's current away message"); + registersearchterm(reg_nicksearch, "authname",authname_parse, 0, "User's current authname or false"); + registersearchterm(reg_whowassearch, "authname",authname_parse, 0, "User's current authname or false"); + registersearchterm(reg_nicksearch, "authts",authts_parse, 0, "User's Auth timestamp"); + registersearchterm(reg_whowassearch, "authts",authts_parse, 0, "User's Auth timestamp"); + registersearchterm(reg_nicksearch, "ident",ident_parse, 0, "User's current ident"); + registersearchterm(reg_whowassearch, "ident",ident_parse, 0, "User's current ident"); + registersearchterm(reg_nicksearch, "host",host_parse, 0, "User's host, allow \"host real\" to match real host"); + registersearchterm(reg_whowassearch, "host",host_parse, 0, "User's host, allow \"host real\" to match real host"); + registersearchterm(reg_nicksearch, "channel",channel_parse, 0, "Valid Channel Name to match users against"); /* nick only */ + registersearchterm(reg_nicksearch, "timestamp",timestamp_parse, 0, "User's Timestamp"); + registersearchterm(reg_whowassearch, "timestamp",timestamp_parse, 0, "User's Timestamp"); + registersearchterm(reg_nicksearch, "country",country_parse, 0, "2 letter country code (data source is geoip)"); + registersearchterm(reg_nicksearch, "ip",ip_parse, 0, "User's IP - ipv4 or ipv6 format as appropriate. Note: not 6to4"); + registersearchterm(reg_whowassearch, "ip",ip_parse, 0, "User's IP - ipv4 or ipv6 format as appropriate. Note: not 6to4"); + registersearchterm(reg_nicksearch, "channels",channels_parse, 0, "Channel Count"); /* nick only */ + registersearchterm(reg_nicksearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)"); + registersearchterm(reg_whowassearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)"); + registersearchterm(reg_whowassearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)"); + registersearchterm(reg_nicksearch, "authid",authid_parse, 0, "User's Auth ID"); + registersearchterm(reg_whowassearch, "authid",authid_parse, 0, "User's Auth ID"); + registersearchterm(reg_nicksearch, "cidr",cidr_parse, 0, "CIDR matching"); + registersearchterm(reg_whowassearch, "cidr",cidr_parse, 0, "CIDR matching"); + registersearchterm(reg_nicksearch, "ipvsix",ipv6_parse, 0, "IPv6 user"); + registersearchterm(reg_whowassearch, "ipvsix",ipv6_parse, 0, "IPv6 user"); + registersearchterm(reg_nicksearch, "message",message_parse, 0, "Last message"); + + /* Whowas operations */ + registersearchterm(reg_whowassearch, "quit",quit_parse, 0, "User quit"); + registersearchterm(reg_whowassearch, "killed",killed_parse, 0, "User was killed"); + registersearchterm(reg_whowassearch, "renamed",renamed_parse, 0, "User changed nick"); + registersearchterm(reg_whowassearch, "age",age_parse, 0, "Whowas record age in seconds"); + registersearchterm(reg_whowassearch, "newnick",newnick_parse, 0, "New nick (for rename whowas records)"); + registersearchterm(reg_whowassearch, "reason",reason_parse, 0, "Quit/kill reason"); /* Channel operations */ - registersearchterm("exists",exists_parse); - registersearchterm("services",services_parse); - registersearchterm("size",size_parse); - registersearchterm("name",name_parse); - registersearchterm("topic",topic_parse); - registersearchterm("oppct",oppct_parse); - registersearchterm("uniquehostpct",hostpct_parse); - registersearchterm("authedpct",authedpct_parse); - registersearchterm("kick",kick_parse); + registersearchterm(reg_chansearch, "exists",exists_parse, 0, "Returns if channel exists on network. Note: newserv may store data on empty channels"); /* channel only */ + registersearchterm(reg_chansearch, "services",services_parse, 0, ""); /* channel only */ + registersearchterm(reg_chansearch, "size",size_parse, 0, "Channel user count"); /* channel only */ + registersearchterm(reg_chansearch, "name",name_parse, 0, "Channel Name"); /* channel only */ + registersearchterm(reg_chansearch, "topic",topic_parse, 0, "Channel topic"); /* channel only */ + registersearchterm(reg_chansearch, "oppct",oppct_parse, 0, "Percentage Opped"); /* channel only */ + registersearchterm(reg_chansearch, "cumodecount",cumodecount_parse, 0, "Count of users with given channel modes"); /* channel only */ + registersearchterm(reg_chansearch, "cumodepct",cumodepct_parse, 0, "Percentage of users with given channel modes"); /* channel only */ + registersearchterm(reg_chansearch, "uniquehostpct",hostpct_parse, 0, "uniquehost percent"); /* channel only */ + registersearchterm(reg_chansearch, "authedpct",authedpct_parse, 0, "Percentage of authed users"); /* channel only */ + registersearchterm(reg_chansearch, "kick",kick_parse, 0, "KICK users channels in newsearch result. Note: evaluation order"); /* channel only */ /* Nickname / channel operations */ - registersearchterm("modes",modes_parse); - registersearchterm("nick",nick_parse); + registersearchterm(reg_chansearch, "modes",modes_parse, 0, "Channel Modes"); + registersearchterm(reg_nicksearch, "modes",modes_parse, 0, "User Modes"); + registersearchterm(reg_whowassearch, "modes",modes_parse, 0, "User Modes"); + registersearchterm(reg_chansearch, "nick",nick_parse, 0, "Nickname"); + registersearchterm(reg_nicksearch, "nick",nick_parse, 0, "Nickname"); + registersearchterm(reg_whowassearch, "nick",nick_parse, 0, "Nickname"); /* Kill / gline parameters */ - registersearchterm("kill",kill_parse); - registersearchterm("gline",gline_parse); + registersearchterm(reg_chansearch,"kill",kill_parse, 0, "KILL users in newsearch result. Note: evaluation order"); + registersearchterm(reg_chansearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order"); + registersearchterm(reg_nicksearch,"kill",kill_parse, 0, "KILL users in newsearch result. Note: evaluation order"); + registersearchterm(reg_nicksearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order"); + registersearchterm(reg_whowassearch,"gline",gline_parse, 0, "GLINE users in newsearch result. Note: evaluation order"); /* Iteration functionality */ - registersearchterm("any",any_parse); - registersearchterm("all",all_parse); - registersearchterm("var",var_parse); + registerglobalsearchterm("any",any_parse, "usage: any (generatorfn x) (fn ... (var x) ...)"); + registerglobalsearchterm("all",all_parse, "usage: all (generatorfn x) (fn ... (var x) ...)"); + registerglobalsearchterm("var",var_parse, "usage: var variable"); /* Iterable functions */ - registersearchterm("channeliter",channeliter_parse); - + registersearchterm(reg_nicksearch, "channeliter",channeliter_parse, 0, "Channel Iterable function - usage: (any (channeliter x) (match (var x) #twilight*))"); /* nick only */ + registersearchterm(reg_chansearch, "nickiter",nickiter_parse, 0, "Nick Iterable function - usage: (any (nickiter x) (match (var x) nickname*))"); /* channel only */ + + /* Functions that work on strings?! */ + registersearchterm(reg_nicksearch, "cumodes", cumodes_parse, 0, "usage: (cumodes (var x) )"); + registersearchterm(reg_chansearch, "cumodes", cumodes_parse, 0, "usage: (cumodes (var x) )"); + /* Notice functionality */ - registersearchterm("notice",notice_parse); - + registersearchterm(reg_chansearch,"notice",notice_parse, 0, "NOTICE users in newsearch result. Note: evaluation order"); + registersearchterm(reg_nicksearch,"notice",notice_parse, 0, "NOTICE users in newsearch result. Note: evaluation order"); + /* Nick output filters */ - regnickdisp("default",printnick); - regnickdisp("channels",printnick_channels); + regdisp(reg_nicksearch,"default",printnick, 0, ""); + regdisp(reg_nicksearch,"channels",printnick_channels, 0, "include channels in output"); /* Channel output filters */ - regchandisp("default",printchannel); - regchandisp("topic",printchannel_topic); - regchandisp("services",printchannel_services); + regdisp(reg_chansearch,"default",printchannel, 0, ""); + regdisp(reg_chansearch,"topic",printchannel_topic, 0, "display channel topics"); + regdisp(reg_chansearch,"services",printchannel_services, 0, "display services on channels"); /* Nick output filters */ - reguserdisp("default",printuser); - - registercontrolhelpcmd("nicksearch",NO_OPER,4,do_nicksearch, "Usage: nicksearch \nSearches for nicknames with the given criteria."); - registercontrolhelpcmd("chansearch",NO_OPER,4,do_chansearch, "Usage: chansearch \nSearches for channels with the given criteria."); - registercontrolhelpcmd("usersearch",NO_OPER,4,do_usersearch, "Usage: usersearch \nSearches for users with the given criteria."); + regdisp(reg_usersearch,"default",printuser, 0, ""); + } void _fini() { - destroycommandtree(searchTree); - destroycommandtree(chanOutputTree); - destroycommandtree(nickOutputTree); - destroycommandtree(userOutputTree); - deregistercontrolcmd("nicksearch", do_nicksearch); - deregistercontrolcmd("chansearch", do_chansearch); - deregistercontrolcmd("usersearch", do_usersearch); + searchList *sl, *psl; + int i,n; + Command *cmdlist[100]; + + sl=globalterms; + while (sl) { + psl = sl; + sl=sl->next; + + n=getcommandlist(searchCmdTree,cmdlist,100); + for(i=0;ihandler, psl->name->content, psl->cmd); + } + + freesstring(psl->name); + if (psl->help) + free (psl->help); + free(psl); + } + + deregistersearchcommand( reg_nicksearch ); + deregistersearchcommand( reg_chansearch ); + deregistersearchcommand( reg_usersearch ); + destroycommandtree( searchCmdTree ); +} + +void registerglobalsearchterm(char *term, parseFunc parsefunc, char *help) { + int i,n; + Command *cmdlist[100]; + searchList *sl = malloc(sizeof(searchList)); + if (!sl) { + Error("newsearch", ERR_ERROR, "malloc failed: registerglobalsearchterm"); + return; + } + + sl->cmd = parsefunc; + sl->name = getsstring(term, NSMAX_COMMAND_LEN); + sl->next = NULL; + if (help) { + int len=strlen(help); + sl->help=(char *)malloc(len+1); + if(!sl->help) { + Error("newsearch", ERR_ERROR, "malloc failed: registerglobalsearchterm"); + return; + } + strncpy(sl->help, help, len); + sl->help[len] = '\0'; + } else { + sl->help=NULL; + } + + + if ( globalterms != NULL ) { + sl->next = globalterms; + } + globalterms = sl; + + n=getcommandlist(searchCmdTree,cmdlist,100); + for(i=0;ihandler)->searchtree,term, 0, 1, (CommandHandler) parsefunc, help); + } } -void registersearchterm(char *term, parseFunc parsefunc) { - addcommandtotree(searchTree, term, 0, 0, (CommandHandler) parsefunc); +void deregisterglobalsearchterm(char *term, parseFunc parsefunc) { + int i,n; + Command *cmdlist[100]; + searchList *sl, *psl=NULL; + + for (sl=globalterms; sl; sl=sl->next) { + if ( strcmp( sl->name->content, term) == 0 ) { + break; + } + psl = sl; + } + + if (sl) { + if( psl ) { + psl->next = sl->next; + } + + n=getcommandlist(searchCmdTree,cmdlist,100); + for(i=0;ihandler)->searchtree, term, (CommandHandler) parsefunc); + } + freesstring(sl->name); + free(sl); + } +} + +void registersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc, int level, char *help) { + /* NOTE: global terms are added to the tree elsewhere as we set maxparams to 1 to indicate global */ + addcommandexttotree(cmd->searchtree, term, level, 0, (CommandHandler) parsefunc, help); } -void deregistersearchterm(char *term, parseFunc parsefunc) { - deletecommandfromtree(searchTree, term, (CommandHandler) parsefunc); +void deregistersearchterm(searchCmd *cmd, char *term, parseFunc parsefunc) { + /* NOTE: global terms are removed from the tree within deregisterglobalsearchterm */ + deletecommandfromtree(cmd->searchtree, term, (CommandHandler) parsefunc); } +static void controlwallwrapper(int level, char *format, ...) __attribute__ ((format (printf, 2, 3))); static void controlwallwrapper(int level, char *format, ...) { char buf[1024]; va_list ap; @@ -175,10 +393,11 @@ static void controlwallwrapper(int level, char *format, ...) { va_end(ap); } -static int parseopts(int cargc, char **cargv, int *arg, int *limit, void **display, CommandTree *tree, replyFunc reply, void *sender) { +int parseopts(int cargc, char **cargv, int *arg, int *limit, void **subset, void *display, CommandTree *sl, replyFunc reply, void *sender) { char *ch; - struct Command *cmd; - + Command *cmd; + struct irc_in_addr sin; unsigned char bits; + if (*cargv[0] == '-') { /* options */ (*arg)++; @@ -187,28 +406,50 @@ static int parseopts(int cargc, char **cargv, int *arg, int *limit, void **displ switch(*ch) { case 'l': if (cargc<*arg) { - reply(sender,"Error: -l switch requires an argument"); - return CMD_USAGE; + reply(sender,"Error: -l switch requires an argument (for help, see help )"); + return CMD_ERROR; } *limit=strtoul(cargv[(*arg)++],NULL,10); break; case 'd': if (cargc<*arg) { - reply(sender,"Error: -d switch requires an argument"); - return CMD_USAGE; + reply(sender,"Error: -d switch requires an argument (for help, see help )"); + return CMD_ERROR; } - cmd=findcommandintree(tree, cargv[*arg], 1); + cmd=findcommandintree(sl, cargv[*arg],1); if (!cmd) { - reply(sender,"Error: unknown output format %s",cargv[*arg]); - return CMD_USAGE; + reply(sender,"Error: unknown output format %s (for help, see help )",cargv[*arg]); + return CMD_ERROR; + } + if ( !controlpermitted( cmd->level, sender) ) { + reply(sender,"Error: Access Denied for output format %s (for help, see help )", cargv[*arg]); + return CMD_ERROR; + } + *((void **)display)=(void *)cmd->handler; + (*arg)++; + break; + + case 's': + if (subset == NULL) { + reply(sender,"Error: -s switch not supported for this search."); + return CMD_ERROR; + } + if (cargc<*arg) { + reply(sender,"Error: -s switch requires an argument (for help, see help )"); + return CMD_ERROR; + } + if (ipmask_parse(cargv[*arg], &sin, &bits) == 0) { + reply(sender, "Error: Invalid CIDR mask supplied (for help, see help )"); + return CMD_ERROR; } - *display=(void *)cmd->handler; + *subset = (void *)refnode(iptree, &sin, bits); (*arg)++; break; default: - reply(sender,"Unrecognised flag -%c.",*ch); + reply(sender,"Unrecognised flag -%c. (for help, see help )",*ch); + return CMD_ERROR; } } } @@ -216,29 +457,35 @@ static int parseopts(int cargc, char **cargv, int *arg, int *limit, void **displ return CMD_OK; } -void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, int type) { +void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, searchCmd *cmd, nick *np, void *displayfn, int limit, void *target) { memset(ctx, 0, sizeof(searchCtx)); ctx->reply = replyfn; ctx->wall = wallfn; ctx->parser = searchfn; ctx->arg = arg; - ctx->type = type; + ctx->searchcmd = cmd; + ctx->sender = np; + ctx->limit = limit; + ctx->target = target; + ctx->displayfn = displayfn; } int do_nicksearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) { nick *sender = source; - struct searchNode *search; int limit=500; int arg=0; NickDisplayFunc display=defaultnickfn; - searchCtx ctx; int ret; + parsertree *tree; - if (cargc<1) - return CMD_USAGE; - - ret = parseopts(cargc, cargv, &arg, &limit, (void **)&display, nickOutputTree, reply, sender); + if (cargc<1) { + reply( sender, "Usage: [flags] "); + reply( sender, "For help, see help nicksearch"); + return CMD_OK; + } + + ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_nicksearch->outputtree, reply, sender); if(ret != CMD_OK) return ret; @@ -251,16 +498,15 @@ int do_nicksearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, rejoinline(cargv[arg],cargc-arg); } - newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, SEARCHTYPE_NICK); - - if (!(search = ctx.parser(&ctx, cargv[arg]))) { - reply(sender,"Parse error: %s",parseError); + tree = parse_string(reg_nicksearch, cargv[arg]); + if(!tree) { + displaystrerror(reply, sender, cargv[arg]); return CMD_ERROR; } - nicksearch_exe(search, &ctx, sender, display, limit); + ast_nicksearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL); - (search->free)(&ctx, search); + parse_free(tree); return CMD_OK; } @@ -269,15 +515,17 @@ int do_nicksearch(void *source, int cargc, char **cargv) { return do_nicksearch_real(controlreply, controlwallwrapper, source, cargc, cargv); } -void nicksearch_exe(struct searchNode *search, searchCtx *ctx, nick *sender, NickDisplayFunc display, int limit) { +void nicksearch_exe(struct searchNode *search, searchCtx *ctx) { int i, j; int matches = 0; unsigned int cmarker; unsigned int tchans=0,uchans=0; struct channel **cs; - nick *np; + nick *np, *sender = ctx->sender; senderNSExtern = sender; - + NickDisplayFunc display = ctx->displayfn; + int limit = ctx->limit; + /* Get a marker value to mark "seen" channels for unique count */ cmarker=nextchanmarker(); @@ -285,7 +533,7 @@ void nicksearch_exe(struct searchNode *search, searchCtx *ctx, nick *sender, Nic search=coerceNode(ctx, search, RETURNTYPE_BOOL); for (i=0;inext) { + for (np=ctx->target ? ctx->target : nicktable[i];np;np=np->next) { if ((search->exe)(ctx, search, np)) { /* Add total channels */ tchans += np->channels->cursi; @@ -306,26 +554,115 @@ void nicksearch_exe(struct searchNode *search, searchCtx *ctx, nick *sender, Nic ctx->reply(sender, "--- More than %d matches, skipping the rest",limit); matches++; } + + if (ctx->target) + goto done; } } +done: ctx->reply(sender,"--- End of list: %d matches; users were on %u channels (%u unique, %.1f average clones)", matches, tchans, uchans, (float)tchans/uchans); +} + +int do_whowassearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) { + nick *sender = source; + int limit=500; + int arg=0; + WhowasDisplayFunc display=defaultwhowasfn; + int ret; + parsertree *tree; + + if (cargc<1) { + reply( sender, "Usage: [flags] "); + reply( sender, "For help, see help whowassearch"); + return CMD_OK; + } + + ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_whowassearch->outputtree, reply, sender); + if(ret != CMD_OK) + return ret; + + if (arg>=cargc) { + reply(sender,"No search terms - aborting."); + return CMD_ERROR; + } + + if (arg<(cargc-1)) { + rejoinline(cargv[arg],cargc-arg); + } + + tree = parse_string(reg_whowassearch, cargv[arg]); + if(!tree) { + displaystrerror(reply, sender, cargv[arg]); + return CMD_ERROR; + } + + ast_whowassearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL); + + parse_free(tree); + + return CMD_OK; +} + +int do_whowassearch(void *source, int cargc, char **cargv) { + return do_whowassearch_real(controlreply, controlwallwrapper, source, cargc, cargv); +} + +void whowassearch_exe(struct searchNode *search, searchCtx *ctx) { + int i, matches = 0; + whowas *ww; + nick *sender = ctx->sender; + senderNSExtern = sender; + WhowasDisplayFunc display = ctx->displayfn; + int limit = ctx->limit; + + /* The top-level node needs to return a BOOL */ + search=coerceNode(ctx, search, RETURNTYPE_BOOL); + + for (i = whowasoffset; i < whowasoffset + WW_MAXENTRIES; i++) { + if (ctx->target) { + ww = ctx->target; + } else { + ww = &whowasrecs[i % WW_MAXENTRIES]; + + if (ww->type == WHOWAS_UNUSED) + continue; + } + + /* Note: We're passing the nick to the filter function. The original + * whowas record is in the nick's ->next field. */ + if ((search->exe)(ctx, search, &ww->nick)) { + if (matchesreply(sender, "--- More than %d matches, skipping the rest",limit); + matches++; + } + + if (ctx->target) + break; + } + + ctx->reply(sender,"--- End of list: %d matches", matches); } int do_chansearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) { nick *sender = source; - struct searchNode *search; int limit=500; int arg=0; ChanDisplayFunc display=defaultchanfn; - searchCtx ctx; int ret; + parsertree *tree; - if (cargc<1) - return CMD_USAGE; + if (cargc<1) { + reply( sender, "Usage: [flags] "); + reply( sender, "For help, see help chansearch"); + return CMD_OK; + } - ret = parseopts(cargc, cargv, &arg, &limit, (void **)&display, chanOutputTree, reply, sender); + ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_chansearch->outputtree, reply, sender); if(ret != CMD_OK) return ret; @@ -338,15 +675,15 @@ int do_chansearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, rejoinline(cargv[arg],cargc-arg); } - newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, SEARCHTYPE_CHANNEL); - if (!(search = ctx.parser(&ctx, cargv[arg]))) { - reply(sender,"Parse error: %s",parseError); + tree = parse_string(reg_chansearch, cargv[arg]); + if(!tree) { + displaystrerror(reply, sender, cargv[arg]); return CMD_ERROR; } - chansearch_exe(search, &ctx, sender, display, limit); + ast_chansearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL); - (search->free)(&ctx, search); + parse_free(tree); return CMD_OK; } @@ -355,42 +692,51 @@ int do_chansearch(void *source, int cargc, char **cargv) { return do_chansearch_real(controlreply, controlwallwrapper, source, cargc, cargv); } -void chansearch_exe(struct searchNode *search, searchCtx *ctx, nick *sender, ChanDisplayFunc display, int limit) { +void chansearch_exe(struct searchNode *search, searchCtx *ctx) { int i; chanindex *cip; int matches = 0; + nick *sender = ctx->sender; senderNSExtern = sender; + ChanDisplayFunc display = ctx->displayfn; + int limit = ctx->limit; search=coerceNode(ctx, search, RETURNTYPE_BOOL); for (i=0;inext) { + for (cip=ctx->target ? ctx->target : chantable[i];cip;cip=cip->next) { if ((search->exe)(ctx, search, cip)) { if (matchesreply(sender, "--- More than %d matches, skipping the rest",limit); matches++; + + if (ctx->target) + goto done; } } } +done: ctx->reply(sender,"--- End of list: %d matches", matches); } int do_usersearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) { nick *sender = source; - struct searchNode *search; int limit=500; int arg=0; UserDisplayFunc display=defaultuserfn; - searchCtx ctx; int ret; + parsertree *tree; - if (cargc<1) - return CMD_USAGE; - - ret = parseopts(cargc, cargv, &arg, &limit, (void **)&display, userOutputTree, reply, sender); + if (cargc<1) { + reply( sender, "Usage: [flags] "); + reply( sender, "For help, see help usersearch"); + return CMD_OK; + } + + ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_usersearch->outputtree, reply, sender); if(ret != CMD_OK) return ret; @@ -403,15 +749,15 @@ int do_usersearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, rejoinline(cargv[arg],cargc-arg); } - newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, SEARCHTYPE_USER); - if (!(search = ctx.parser(&ctx, cargv[arg]))) { - reply(sender,"Parse error: %s",parseError); + tree = parse_string(reg_usersearch, cargv[arg]); + if(!tree) { + displaystrerror(reply, sender, cargv[arg]); return CMD_ERROR; } - usersearch_exe(search, &ctx, sender, display, limit); + ast_usersearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL); - (search->free)(&ctx, search); + parse_free(tree); return CMD_OK; } @@ -420,16 +766,19 @@ int do_usersearch(void *source, int cargc, char **cargv) { return do_usersearch_real(controlreply, controlwallwrapper, source, cargc, cargv); } -void usersearch_exe(struct searchNode *search, searchCtx *ctx, nick *sender, UserDisplayFunc display, int limit) { +void usersearch_exe(struct searchNode *search, searchCtx *ctx) { int i; authname *aup; int matches = 0; + nick *sender = ctx->sender; + int limit = ctx->limit; + UserDisplayFunc display = ctx->displayfn; senderNSExtern = sender; - + search=coerceNode(ctx, search, RETURNTYPE_BOOL); for (i=0;inext) { + for (aup=ctx->target ? ctx->target : authnametable[i];aup;aup=aup->next) { if ((search->exe)(ctx, search, aup)) { if (matchesreply(sender, "--- More than %d matches, skipping the rest",limit); matches++; } + + if (ctx->target) + goto done; } } +done: ctx->reply(sender,"--- End of list: %d matches", matches); } @@ -668,136 +1021,6 @@ void literal_free(searchCtx *ctx, struct searchNode *thenode) { free(thenode); } -/* search_parse: - * Given an input string, return a searchNode. - */ - -struct searchNode *search_parse(searchCtx *ctx, char *input) { - /* OK, we need to split the input into chunks on spaces and brackets.. */ - char *argvector[100]; - char thestring[500]; - int i,j,q=0,e=0; - char *ch,*ch2; - struct Command *cmd; - struct searchNode *thenode; - - /* If it starts with a bracket, it's a function call.. */ - if (*input=='(') { - /* Skip past string */ - for (ch=input;*ch;ch++); - if (*(ch-1) != ')') { - parseError = "Bracket mismatch!"; - return NULL; - } - input++; - *(ch-1)='\0'; - - /* Split further args */ - i=-1; /* i = -1 BoW, 0 = inword, 1 = bracket nest depth */ - j=0; /* j = current arg */ - e=0; - q=0; - argvector[0]=""; - for (ch=input;*ch;ch++) { - if (i==-1) { - argvector[j]=ch; - if (*ch=='(') { - i=1; - } else if (*ch != ' ') { - i=0; - if (*ch=='\\') { - e=1; - } else if (*ch=='\"') { - q=1; - } - } - } else if (e==1) { - e=0; - } else if (q==1) { - if (*ch=='\"') - q=0; - } else if (i==0) { - if (*ch=='\\') { - e=1; - } else if (*ch=='\"') { - q=1; - } else if (*ch==' ') { - *ch='\0'; - j++; - if(j >= (sizeof(argvector) / sizeof(*argvector))) { - parseError = "Too many arguments"; - return NULL; - } - i=-1; - } - } else { - if (*ch=='\\') { - e=1; - } else if (*ch=='\"') { - q=1; - } else if (*ch=='(') { - i++; - } else if (*ch==')') { - i--; - } - } - } - - if (i>0) { - parseError = "Bracket mismatch!"; - return NULL; - } - - if (*(ch-1) == 0) /* if the last character was a space */ - j--; /* remove an argument */ - - if (!(cmd=findcommandintree(searchTree,argvector[0],1))) { - parseError = "Unknown command"; - return NULL; - } else { - return ((parseFunc)cmd->handler)(ctx, j, argvector+1); - } - } else { - /* Literal */ - if (*input=='\"') { - for (ch=input;*ch;ch++); - - if (*(ch-1) != '\"') { - parseError="Quote mismatch"; - return NULL; - } - - *(ch-1)='\0'; - input++; - } - - ch2=thestring; - for (ch=input;*ch;ch++) { - if (e) { - e=0; - *ch2++=*ch; - } else if (*ch=='\\') { - e=1; - } else { - *ch2++=*ch; - } - } - *ch2='\0'; - - if (!(thenode=(struct searchNode *)malloc(sizeof(struct searchNode)))) { - parseError = "malloc: could not allocate memory for this search."; - return NULL; - } - - thenode->localdata = getsstring(thestring,512); - thenode->returntype = RETURNTYPE_CONST | RETURNTYPE_STRING; - thenode->exe = literal_exe; - thenode->free = literal_free; - - return thenode; - } -} - void nssnprintf(char *buf, size_t size, const char *format, nick *np) { StringBuf b; const char *p; @@ -835,11 +1058,11 @@ void nssnprintf(char *buf, size_t size, const char *format, nick *np) { case 'h': c = np->host->name->content; break; case 'I': - snprintf(hostbuf, sizeof(hostbuf), "%s", IPtostr(np->p_ipaddr)); + snprintf(hostbuf, sizeof(hostbuf), "%s", IPtostr(np->ipaddress)); c = hostbuf; break; case 'u': - snprintf(hostbuf, sizeof(hostbuf), "%s!%s@%s", np->nick, np->ident, IPtostr(np->p_ipaddr)); + snprintf(hostbuf, sizeof(hostbuf), "%s!%s@%s", np->nick, np->ident, IPtostr(np->ipaddress)); c = hostbuf; break; default: @@ -935,3 +1158,46 @@ searchNode *var_get(searchCtx *ctx, char *arg) { void var_setstr(struct searchVariable *v, char *data) { v->cdata.u.stringbuf = data; } + +void displaystrerror(replyFunc reply, nick *np, const char *input) { + char buf[515]; + + if((parseStrErrorPos >= 0) && (parseStrErrorPos < sizeof(buf) - 3)) { + int i; + + for(i=0;iparser(ctx, arg); + if (!(c = coerceNode(ctx, c, RETURNTYPE_STRING))) { + snprintf(errorbuf, sizeof(errorbuf), "%s: unable to coerce argument to string", command); + parseError = errorbuf; + return NULL; + } + + if (!(c->returntype & RETURNTYPE_CONST)) { + snprintf(errorbuf, sizeof(errorbuf), "%s: constant argument required", command); + parseError = errorbuf; + (c->free)(ctx, c); + return NULL; + } + + if(p) + *p = (char *)(c->exe)(ctx, c, NULL); + + return c; +}