-
#include <stdio.h>
-#include "newsearch.h"
+#include <stdarg.h>
+#include <string.h>
#include "../irc/irc_config.h"
#include "../lib/irc_string.h"
#include "../control/control.h"
#include "../lib/splitline.h"
#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 *searchCmdTree;
+searchList *globalterms = NULL;
int do_nicksearch(void *source, int cargc, char **cargv);
int do_chansearch(void *source, int cargc, char **cargv);
-struct searchNode *search_parse(int type, char *input);
+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 *);
+void printchannel_topic(searchCtx *, nick *, chanindex *);
+void printchannel_services(searchCtx *, nick *, chanindex *);
+
+UserDisplayFunc defaultuserfn = printuser;
+NickDisplayFunc defaultnickfn = printnick;
+ChanDisplayFunc defaultchanfn = printchannel;
+WhowasDisplayFunc defaultwhowasfn = printwhowas;
+
+searchCmd *reg_nicksearch, *reg_chansearch, *reg_usersearch, *reg_whowassearch;
+void displaycommandhelp(nick *, Command *);
+void displaystrerror(replyFunc, nick *, const char *);
+
+searchCmd *registersearchcommand(char *name, int level, CommandHandler cmd, void *defaultdisplayfunc) {
+ searchCmd *acmd;
+ searchList *sl;
+
+ 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;
+
+ 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 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 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;i<n;i++) {
+ /* note: we may want to only deregister a command if we've already registered it, for now, try de-registering new commands anyway */
+ if ( ((searchCmd *)cmdlist[i]->handler)->handler != cmd->handler )
+ continue;
+ acmd = ((searchCmd *)(cmdlist[i]->handler));
+
+ controlreply(np, "Usage: [flags] <criteria>\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;j<m;j++) {
+ if ( 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 );
+ }
+ }
+
+ controlreply(np, " \n");
+ controlreply(np, "Available Global Commands and Operators:\n" );
+ m=getcommandlist(acmd->searchtree,acmdlist,100);
+ for(j=0;j<m;j++) {
+ if ( acmdlist[j]->maxparams) {
+ 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;j<m;j++) {
+ if ( !acmdlist[j]->maxparams && 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 registersearchterm(char *term, parseFunc parsefunc);
-void deregistersearchterm(char *term, parseFunc parsefunc);
+void regdisp( searchCmd *cmd, const char *name, void *handler, int level, char *help) {
+ addcommandexttotree(cmd->outputtree, name, level, 0, (CommandHandler) handler, (char *)help);
+}
-void *trueval(int type);
-void *falseval(int type);
+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
nick *senderNSExtern;
void _init() {
- searchTree=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);
-
+ 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("ident",ident_parse);
- registersearchterm("host",host_parse);
- registersearchterm("channel",channel_parse);
- registersearchterm("timestamp",timestamp_parse);
- registersearchterm("country",country_parse);
- registersearchterm("ip",ip_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("namelen",namelen_parse);
- registersearchterm("name",name_parse);
- registersearchterm("topic",topic_parse);
- registersearchterm("oppct",oppct_parse);
- registersearchterm("uniquehostpct",hostpct_parse);
- registersearchterm("authedpct",authedpct_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 */
+ 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(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) <modes>)");
+ registersearchterm(reg_chansearch, "cumodes", cumodes_parse, 0, "usage: (cumodes (var x) <modes>)");
+
+ /* Notice functionality */
+ 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 */
+ regdisp(reg_nicksearch,"default",printnick, 0, "");
+ regdisp(reg_nicksearch,"channels",printnick_channels, 0, "include channels in output");
+
+ /* Channel output filters */
+ 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");
- registercontrolhelpcmd("nicksearch",NO_OPER,4,do_nicksearch, "Usage: nicksearch <criteria>\nSearches for nicknames with the given criteria.");
- registercontrolhelpcmd("chansearch",NO_OPER,4,do_chansearch, "Usage: chansearch <criteria>\nSearches for channels with the given criteria.");
+ /* Nick output filters */
+ regdisp(reg_usersearch,"default",printuser, 0, "");
+
}
void _fini() {
- destroycommandtree(searchTree);
- deregistercontrolcmd("nicksearch", do_nicksearch);
- deregistercontrolcmd("chansearch", do_chansearch);
-}
+ searchList *sl, *psl;
+ int i,n;
+ Command *cmdlist[100];
-void registersearchterm(char *term, parseFunc parsefunc) {
- addcommandtotree(searchTree, term, 0, 0, (CommandHandler) parsefunc);
-}
+ sl=globalterms;
+ while (sl) {
+ psl = sl;
+ sl=sl->next;
-void deregistersearchterm(char *term, parseFunc parsefunc) {
- deletecommandfromtree(searchTree, term, (CommandHandler) parsefunc);
-}
+ n=getcommandlist(searchCmdTree,cmdlist,100);
+ for(i=0;i<n;i++) {
+ deregistersearchterm( (searchCmd *)cmdlist[i]->handler, psl->name->content, psl->cmd);
+ }
-void printnick(nick *sender, nick *np) {
- char hostbuf[HOSTLEN+NICKLEN+USERLEN+4];
+ freesstring(psl->name);
+ if (psl->help)
+ free (psl->help);
+ free(psl);
+ }
- controlreply(sender,"%s [%s] (%s) (%s)",visiblehostmask(np,hostbuf),
- IPtostr(np->p_ipaddr), printflags(np->umodes, umodeflags), np->realname->name->content);
+ deregistersearchcommand( reg_nicksearch );
+ deregistersearchcommand( reg_chansearch );
+ deregistersearchcommand( reg_usersearch );
+ destroycommandtree( searchCmdTree );
}
-void printchannel(nick *sender, chanindex *cip) {
- /* shamelessly stolen from (now defunct) chansearch.c */
- int i;
- int op,voice,peon;
- int oper,service,hosts;
- nick *np;
- chanuserhash *cuhp;
- unsigned int marker;
-
- op=voice=peon=oper=service=hosts=0;
- marker=nexthostmarker();
-
- if (cip->channel==NULL) {
- controlreply(sender,"[ Channel currently empty ] %s",cip->name->content);
+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 {
- cuhp=cip->channel->users;
- for (i=0;i<cuhp->hashsize;i++) {
- if (cuhp->content[i]!=nouser) {
- if (cuhp->content[i]&CUMODE_OP) {
- op++;
- } else if (cuhp->content[i]&CUMODE_VOICE) {
- voice++;
- } else {
- peon++;
- }
- if ((np=getnickbynumeric(cuhp->content[i]&CU_NUMERICMASK))!=NULL) {
- if (IsOper(np)) {
- oper++;
- }
- if (IsService(np)) {
- service++;
- }
- if (np->host->marker!=marker) {
- np->host->marker=marker;
- hosts++;
- }
- }
- }
+ sl->help=NULL;
+ }
+
+
+ if ( globalterms != NULL ) {
+ sl->next = globalterms;
+ }
+ globalterms = sl;
+
+ n=getcommandlist(searchCmdTree,cmdlist,100);
+ for(i=0;i<n;i++) {
+ /* maxparams is set to 1 to indicate a global term */
+ /* access level is set to 0 for all global terms */
+ addcommandexttotree( ((searchCmd *)cmdlist[i]->handler)->searchtree,term, 0, 1, (CommandHandler) parsefunc, help);
+ }
+}
+
+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;i<n;i++) {
+ deletecommandfromtree( ((searchCmd *)cmdlist[i]->handler)->searchtree, term, (CommandHandler) parsefunc);
}
- controlreply(sender,"[ %4dU %4d@ %4d+ %4d %4d* %4dk %4dH ] %s (%s)",cuhp->totalusers,op,voice,peon,oper,
- service,hosts,cip->name->content, printflags(cip->channel->flags, cmodeflags));
+ freesstring(sl->name);
+ free(sl);
}
}
-int do_nicksearch(void *source, int cargc, char **cargv) {
- nick *sender = senderNSExtern = source, *np;
- int i;
- struct searchNode *search;
- int limit=500,matches=0;
- char *ch;
- int arg=0;
+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);
+}
- if (cargc<1)
- return CMD_USAGE;
-
+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;
+
+ va_start(ap, format);
+ vsnprintf(buf, sizeof(buf), format, ap);
+ controlwall(NO_OPER, level, "%s", buf);
+ va_end(ap);
+}
+
+int parseopts(int cargc, char **cargv, int *arg, int *limit, void **subset, void *display, CommandTree *sl, replyFunc reply, void *sender) {
+ char *ch;
+ Command *cmd;
+ struct irc_in_addr sin; unsigned char bits;
+
if (*cargv[0] == '-') {
/* options */
- arg++;
+ (*arg)++;
for (ch=cargv[0]+1;*ch;ch++) {
switch(*ch) {
case 'l':
- if (cargc<arg) {
- controlreply(sender,"Error: -l switch requires an argument");
- return CMD_USAGE;
+ if (cargc<*arg) {
+ reply(sender,"Error: -l switch requires an argument (for help, see help <searchcmd>)");
+ return CMD_ERROR;
}
- limit=strtoul(cargv[arg++],NULL,10);
+ *limit=strtoul(cargv[(*arg)++],NULL,10);
break;
+ case 'd':
+ if (cargc<*arg) {
+ reply(sender,"Error: -d switch requires an argument (for help, see help <searchcmd>)");
+ return CMD_ERROR;
+ }
+ cmd=findcommandintree(sl, cargv[*arg],1);
+ if (!cmd) {
+ reply(sender,"Error: unknown output format %s (for help, see help <searchcmd>)",cargv[*arg]);
+ return CMD_ERROR;
+ }
+ if ( !controlpermitted( cmd->level, sender) ) {
+ reply(sender,"Error: Access Denied for output format %s (for help, see help <searchcmd>)", 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 <searchcmd>)");
+ return CMD_ERROR;
+ }
+ if (ipmask_parse(cargv[*arg], &sin, &bits) == 0) {
+ reply(sender, "Error: Invalid CIDR mask supplied (for help, see help <searchcmd>)");
+ return CMD_ERROR;
+ }
+ *subset = (void *)refnode(iptree, &sin, bits);
+ (*arg)++;
+ break;
+
default:
- controlreply(sender,"Unrecognised flag -%c.",*ch);
+ reply(sender,"Unrecognised flag -%c. (for help, see help <searchcmd>)",*ch);
+ return CMD_ERROR;
}
}
}
+ return CMD_OK;
+}
+
+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->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;
+ int limit=500;
+ int arg=0;
+ NickDisplayFunc display=defaultnickfn;
+ int ret;
+ parsertree *tree;
+
+ if (cargc<1) {
+ reply( sender, "Usage: [flags] <criteria>");
+ 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;
+
if (arg>=cargc) {
- controlreply(sender,"No search terms - aborting.");
+ reply(sender,"No search terms - aborting.");
return CMD_ERROR;
}
if (arg<(cargc-1)) {
rejoinline(cargv[arg],cargc-arg);
}
-
- if (!(search = search_parse(SEARCHTYPE_NICK, cargv[arg]))) {
- controlreply(sender,"Parse error: %s",parseError);
+
+ tree = parse_string(reg_nicksearch, cargv[arg]);
+ if(!tree) {
+ displaystrerror(reply, sender, cargv[arg]);
return CMD_ERROR;
}
+
+ ast_nicksearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
+
+ parse_free(tree);
+
+ return CMD_OK;
+}
+
+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) {
+ int i, j;
+ int matches = 0;
+ unsigned int cmarker;
+ unsigned int tchans=0,uchans=0;
+ struct channel **cs;
+ 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();
+
+ /* The top-level node needs to return a BOOL */
+ search=coerceNode(ctx, search, RETURNTYPE_BOOL);
for (i=0;i<NICKHASHSIZE;i++) {
- for (np=nicktable[i];np;np=np->next) {
- if ((search->exe)(search, RETURNTYPE_BOOL, np)) {
+ 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;
+
+ /* Check channels for uniqueness */
+ cs=(channel **)np->channels->content;
+ for (j=0;j<np->channels->cursi;j++) {
+ if (cs[j]->index->marker != cmarker) {
+ cs[j]->index->marker=cmarker;
+ uchans++;
+ }
+ }
+
if (matches<limit)
- printnick(sender, np);
+ display(ctx, sender, np);
+
if (matches==limit)
- controlreply(sender, "--- More than %d matches, skipping the rest",limit);
+ ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
matches++;
}
+
+ if (ctx->target)
+ goto done;
}
}
- (search->free)(search);
+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] <criteria>");
+ 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);
- controlreply(sender,"--- End of list: %d matches", matches);
-
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 (matches<limit)
+ display(ctx, sender, ww);
+
+ if (matches==limit)
+ ctx->reply(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(void *source, int cargc, char **cargv) {
- nick *sender = senderNSExtern = source;
- chanindex *cip;
- int i;
- struct searchNode *search;
- int limit=500,matches=0;
- char *ch;
+int do_chansearch_real(replyFunc reply, wallFunc wall, void *source, int cargc, char **cargv) {
+ nick *sender = source;
+ int limit=500;
int arg=0;
+ ChanDisplayFunc display=defaultchanfn;
+ int ret;
+ parsertree *tree;
- if (cargc<1)
- return CMD_USAGE;
-
- if (*cargv[0] == '-') {
- /* options */
- arg++;
-
- for (ch=cargv[0]+1;*ch;ch++) {
- switch(*ch) {
- case 'l':
- if (cargc<arg) {
- controlreply(sender,"Error: -l switch requires an argument");
- return CMD_USAGE;
- }
- limit=strtoul(cargv[arg++],NULL,10);
- break;
-
- default:
- controlreply(sender,"Unrecognised flag -%c.",*ch);
- }
- }
+ if (cargc<1) {
+ reply( sender, "Usage: [flags] <criteria>");
+ reply( sender, "For help, see help chansearch");
+ return CMD_OK;
}
+
+ ret = parseopts(cargc, cargv, &arg, &limit, NULL, (void *)&display, reg_chansearch->outputtree, reply, sender);
+ if(ret != CMD_OK)
+ return ret;
if (arg>=cargc) {
- controlreply(sender,"No search terms - aborting.");
+ reply(sender,"No search terms - aborting.");
return CMD_ERROR;
}
if (arg<(cargc-1)) {
rejoinline(cargv[arg],cargc-arg);
}
-
- if (!(search = search_parse(SEARCHTYPE_CHANNEL, cargv[arg]))) {
- controlreply(sender,"Parse error: %s",parseError);
+
+ tree = parse_string(reg_chansearch, cargv[arg]);
+ if(!tree) {
+ displaystrerror(reply, sender, cargv[arg]);
return CMD_ERROR;
}
+
+ ast_chansearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
+
+ parse_free(tree);
+
+ return CMD_OK;
+}
+
+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) {
+ 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;i<CHANNELHASHSIZE;i++) {
- for (cip=chantable[i];cip;cip=cip->next) {
- if ((search->exe)(search, RETURNTYPE_BOOL, cip)) {
+ for (cip=ctx->target ? ctx->target : chantable[i];cip;cip=cip->next) {
+ if ((search->exe)(ctx, search, cip)) {
if (matches<limit)
- printchannel(sender, cip);
+ display(ctx, sender, cip);
if (matches==limit)
- controlreply(sender, "--- More than %d matches, skipping the rest",limit);
+ ctx->reply(sender, "--- More than %d matches, skipping the rest",limit);
matches++;
+
+ if (ctx->target)
+ goto done;
}
}
}
- (search->free)(search);
+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;
+ int limit=500;
+ int arg=0;
+ UserDisplayFunc display=defaultuserfn;
+ int ret;
+ parsertree *tree;
+
+ if (cargc<1) {
+ reply( sender, "Usage: [flags] <criteria>");
+ 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;
+
+ 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_usersearch, cargv[arg]);
+ if(!tree) {
+ displaystrerror(reply, sender, cargv[arg]);
+ return CMD_ERROR;
+ }
+
+ ast_usersearch(tree->root, reply, sender, wall, display, NULL, NULL, limit, NULL);
- controlreply(sender,"--- End of list: %d matches", matches);
+ parse_free(tree);
return CMD_OK;
}
-void *trueval(int type) {
- switch(type) {
- default:
- case RETURNTYPE_INT:
- case RETURNTYPE_BOOL:
- return (void *)1;
-
- case RETURNTYPE_STRING:
- return "1";
- }
+int do_usersearch(void *source, int cargc, char **cargv) {
+ return do_usersearch_real(controlreply, controlwallwrapper, source, cargc, cargv);
}
-void *falseval(int type) {
- switch (type) {
- default:
- case RETURNTYPE_INT:
- case RETURNTYPE_BOOL:
- return NULL;
-
- case RETURNTYPE_STRING:
- return "";
+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;i<AUTHNAMEHASHSIZE;i++) {
+ for (aup=ctx->target ? ctx->target : authnametable[i];aup;aup=aup->next) {
+ if ((search->exe)(ctx, search, aup)) {
+ if (matches<limit)
+ display(ctx, sender, aup);
+ if (matches==limit)
+ 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", matches);
}
+/* Free a coerce node */
+void free_coerce(searchCtx *ctx, struct searchNode *thenode) {
+ struct coercedata *cd=thenode->localdata;
+
+ cd->child->free(ctx, cd->child);
+ free(thenode->localdata);
+ free(thenode);
+}
-/*
- * LITERAL node type: used by the top level parse function
- */
+/* Free a coerce node with a stringbuf allocated */
+void free_coercestring(searchCtx *ctx, struct searchNode *thenode) {
+ free(((struct coercedata *)thenode->localdata)->u.stringbuf);
+ free_coerce(ctx, thenode);
+}
-struct literal_localdata {
- int intval;
- int boolval;
- sstring *stringval;
-};
+/* exe_tostr_null: return the constant string */
+void *exe_tostr_null(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ return cd->u.stringbuf;
+}
-void literal_free(struct searchNode *thenode);
-void *literal_exe(struct searchNode *thenode, int type, void *theinput);
+/* exe_val_null: return the constant value */
+void *exe_val_null(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ return (void *)cd->u.val;
+}
-/* search_parse:
- * Given an input string, return a searchNode.
- */
+/* Lots of very dull type conversion functions */
+void *exe_inttostr(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ sprintf(cd->u.stringbuf, "%lu", (unsigned long)(cd->child->exe)(ctx, cd->child, theinput));
+
+ return cd->u.stringbuf;
+}
-struct searchNode *search_parse(int type, 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,e;
- char *ch,*ch2;
- struct Command *cmd;
- struct searchNode *thenode;
- struct literal_localdata *localdata;
+void *exe_booltostr(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ if ((cd->child->exe)(ctx, cd->child, theinput)) {
+ sprintf(cd->u.stringbuf,"1");
+ } else {
+ cd->u.stringbuf[0]='\0';
+ }
+
+ return cd->u.stringbuf;
+}
- /* 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;
- }
+void *exe_strtoint(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ return (void *)strtoul((cd->child->exe)(ctx,cd->child,theinput),NULL,10);
+}
- 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)(type, j, argvector+1);
- }
+void *exe_booltoint(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ /* Don't need to do anything */
+ return (cd->child->exe)(ctx, cd->child, theinput);
+}
+
+void *exe_strtobool(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+ char *ch=(cd->child->exe)(ctx, cd->child, theinput);
+
+ if (!ch || *ch=='\0' || (*ch=='0' && ch[1]=='\0')) {
+ return (void *)0;
+ } else {
+ return (void *)1;
+ }
+}
+
+void *exe_inttobool(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ struct coercedata *cd=thenode->localdata;
+
+ if ((cd->child->exe)(ctx, cd->child, theinput)) {
+ return (void *)1;
} else {
- /* Literal */
- if (*input=='\"') {
- for (ch=input;*ch;ch++);
+ return (void *)0;
+ }
+}
+
+struct searchNode *coerceNode(searchCtx *ctx, struct searchNode *thenode, int type) {
+ struct searchNode *anode;
+ struct coercedata *cd;
+
+ /* You can't coerce a NULL */
+ if (!thenode)
+ return NULL;
+
+ /* No effort required to coerce to the same type */
+ if (type==(thenode->returntype & RETURNTYPE_TYPE))
+ return thenode;
+
+ anode=(struct searchNode *)malloc(sizeof(struct searchNode));
+ anode->localdata=cd=(struct coercedata *)malloc(sizeof(struct coercedata));
+ cd->child=thenode;
+ anode->returntype=type; /* We'll return what they want, always */
+ anode->free=free_coerce;
+
+ switch(type) {
+ case RETURNTYPE_STRING:
+ /* For a string we'll need a buffer */
+ /* A 64-bit number prints out to 20 digits, this leaves some slack */
+ cd->u.stringbuf=malloc(25);
+ anode->free=free_coercestring;
- if (*(ch-1) != '\"') {
- parseError="Quote mismatch";
- return NULL;
+ switch(thenode->returntype & RETURNTYPE_TYPE) {
+ default:
+ case RETURNTYPE_INT:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ /* Constant node: sort it out now */
+ sprintf(cd->u.stringbuf, "%lu", (unsigned long)thenode->exe(ctx, thenode, NULL));
+ anode->exe=exe_tostr_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ /* Variable data */
+ anode->exe=exe_inttostr;
+ }
+ break;
+
+ case RETURNTYPE_BOOL:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ /* Constant bool value */
+ if (thenode->exe(ctx, thenode,NULL)) {
+ /* True! */
+ sprintf(cd->u.stringbuf, "1");
+ } else {
+ cd->u.stringbuf[0] = '\0';
+ }
+ anode->exe=exe_tostr_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ /* Variable bool value */
+ anode->exe=exe_booltostr;
+ }
+ break;
}
-
- *(ch-1)='\0';
- input++;
- }
+ break;
- ch2=thestring;
- for (ch=input;*ch;ch++) {
- if (e) {
- e=0;
- *ch2++=*ch;
- } else if (*ch=='\\') {
- e=1;
- } else {
- *ch2++=*ch;
+ case RETURNTYPE_INT:
+ /* we want an int */
+ switch (thenode->returntype & RETURNTYPE_TYPE) {
+ case RETURNTYPE_STRING:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ cd->u.val=strtoul((thenode->exe)(ctx, thenode, NULL), NULL, 10);
+ anode->exe=exe_val_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ anode->exe=exe_strtoint;
+ }
+ break;
+
+ default:
+ case RETURNTYPE_BOOL:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ if ((thenode->exe)(ctx, thenode,NULL))
+ cd->u.val=1;
+ else
+ cd->u.val=0;
+
+ anode->exe=exe_val_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ anode->exe=exe_booltoint;
+ }
+ break;
+ }
+ break;
+
+ default:
+ case RETURNTYPE_BOOL:
+ /* we want a bool */
+ switch (thenode->returntype & RETURNTYPE_TYPE) {
+ case RETURNTYPE_STRING:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ char *rv=(char *)((thenode->exe)(ctx, thenode, NULL));
+ if (!rv || *rv=='\0' || (*rv=='0' && rv[1]=='\0'))
+ cd->u.val=0;
+ else
+ cd->u.val=1;
+
+ anode->exe=exe_val_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ anode->exe=exe_strtobool;
+ }
+ break;
+
+ default:
+ case RETURNTYPE_INT:
+ if (thenode->returntype & RETURNTYPE_CONST) {
+ if ((thenode->exe)(ctx, thenode,NULL))
+ cd->u.val=1;
+ else
+ cd->u.val=0;
+
+ anode->exe=exe_val_null;
+ anode->returntype |= RETURNTYPE_CONST;
+ } else {
+ anode->exe=exe_inttobool;
+ }
+ break;
}
+ break;
+ }
+
+ return anode;
+}
+
+/* Literals always return constant strings... */
+void *literal_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ if (thenode->localdata)
+ return ((sstring *)thenode->localdata)->content;
+ else
+ return "";
+}
+
+void literal_free(searchCtx *ctx, struct searchNode *thenode) {
+ freesstring(thenode->localdata);
+ free(thenode);
+}
+
+void nssnprintf(char *buf, size_t size, const char *format, nick *np) {
+ StringBuf b;
+ const char *p;
+ char *c;
+ char hostbuf[512];
+
+ if(size == 0)
+ return;
+
+ b.buf = buf;
+ b.capacity = size;
+ b.len = 0;
+
+ for(p=format;*p;p++) {
+ if(*p != '%') {
+ if(!sbaddchar(&b, *p))
+ break;
+ continue;
}
- *ch2='\0';
-
- if (!(thenode=(struct searchNode *)malloc(sizeof(struct searchNode)))) {
- parseError = "malloc: could not allocate memory for this search.";
- return NULL;
+ p++;
+ if(*p == '\0')
+ break;
+ if(*p == '%') {
+ if(!sbaddchar(&b, *p))
+ break;
+ continue;
}
- if (!(localdata=(struct literal_localdata *)malloc(sizeof (struct literal_localdata)))) {
- /* couldn't malloc() memory for localdata, so free thenode to avoid leakage */
- parseError = "malloc: could not allocate memory for this search.";
- free(thenode);
- return NULL;
- }
-
- localdata->stringval=getsstring(thestring,512);
- localdata->intval=strtol(input,NULL,10);
- if (input==NULL || *input=='\0') {
- localdata->boolval = 0;
- } else {
- localdata->boolval = 1;
+ c = NULL;
+ switch(*p) {
+ case 'n':
+ c = np->nick; break;
+ case 'i':
+ c = np->ident; break;
+ case 'h':
+ c = np->host->name->content; break;
+ case 'I':
+ 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->ipaddress));
+ c = hostbuf;
+ break;
+ default:
+ c = "(bad format specifier)";
}
+ if(c)
+ if(!sbaddstr(&b, c))
+ break;
+ }
- thenode->localdata = localdata;
- thenode->returntype = RETURNTYPE_CONST | RETURNTYPE_STRING;
- thenode->exe = literal_exe;
- thenode->free = literal_free;
+ sbterminate(&b);
- return thenode;
- }
+ /* not required */
+ /*
+ buf[size-1] = '\0';
+ */
}
-void *literal_exe(struct searchNode *thenode, int type, void *theinput) {
- struct literal_localdata *localdata;
+static char *var_tochar(searchCtx *ctx, char *arg, searchNode **variable) {
+ *variable = ctx->parser(ctx, arg);
+ if (!(*variable = coerceNode(ctx, *variable, RETURNTYPE_STRING)))
+ return NULL;
- localdata=thenode->localdata;
+ if(!((*variable)->returntype & RETURNTYPE_CONST)) {
+ parseError = "only constant variables allowed";
+ ((*variable)->free)(ctx, *variable);
+ return NULL;
+ }
- switch (type) {
- case RETURNTYPE_STRING:
- return (void *)(localdata->stringval->content);
-
- default:
- case RETURNTYPE_BOOL:
- return (void *)((long)(localdata->boolval));
+ return (char *)((*variable)->exe)(ctx, *variable, NULL);
+}
+
+void free_val_null(searchCtx *ctx, struct searchNode *thenode) {
+}
- case RETURNTYPE_INT:
- return (void *)((long)(localdata->intval));
+struct searchVariable *var_register(searchCtx *ctx, char *arg, int type) {
+ searchNode *variable;
+ struct searchVariable *us;
+ char *var;
+ int i;
+
+ if(ctx->lastvar >= MAX_VARIABLES) {
+ parseError = "Maximum number of variables reached";
+ return NULL;
}
+
+ us = &ctx->vars[ctx->lastvar];
+
+ var = var_tochar(ctx, arg, &variable);
+ if(!var)
+ return NULL;
+
+ strlcpy(us->name, var, sizeof(us->name));
+ (variable->free)(ctx, variable);
+
+ for(i=0;i<ctx->lastvar;i++) {
+ if(!strcmp(us->name, ctx->vars[i].name)) {
+ parseError = "variable name already in use";
+ return NULL;
+ }
+ }
+
+ ctx->lastvar++;
+ us->data.returntype = type;
+ us->data.localdata = &us->cdata;
+ us->data.exe = exe_val_null;
+ us->data.free = free_val_null;
+
+ us->cdata.child = NULL;
+ return us;
}
-void literal_free(struct searchNode *thenode) {
- struct literal_localdata *localdata;
+searchNode *var_get(searchCtx *ctx, char *arg) {
+ searchNode *variable, *found = NULL;
+ int i;
+ char *var = var_tochar(ctx, arg, &variable);
+ if(!var)
+ return NULL;
- localdata=thenode->localdata;
+ for(i=0;i<ctx->lastvar;i++) {
+ if(!strcmp(var, ctx->vars[i].name)) {
+ found = &ctx->vars[i].data;
+ break;
+ }
+ }
+ (variable->free)(ctx, variable);
+
+ if(!found)
+ parseError = "variable not found";
+ return found;
+}
- freesstring(localdata->stringval);
- free(localdata);
- free(thenode);
+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;i<parseStrErrorPos;i++)
+ buf[i] = ' ';
+
+ buf[i++] = '^';
+ buf[i] = '\0';
+
+ reply(np, "%s", input);
+ reply(np, "%s", buf);
+ }
+
+ reply(np, "Parse error: %s", parseStrError);
+}
+
+struct searchNode *argtoconststr(char *command, searchCtx *ctx, char *arg, char **p) {
+ struct searchNode *c;
+ static char errorbuf[512];
+
+ c = ctx->parser(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;
}