#include "../lib/strlfunc.h"
#include "../lib/array.h"
#include "newsearch.h"
+#include "parser.h"
MODULE_VERSION("");
searchCmd *reg_nicksearch, *reg_chansearch, *reg_usersearch;
void displaycommandhelp(nick *, Command *);
+void displaystrerror(replyFunc, nick *, const char *);
searchCmd *registersearchcommand(char *name, int level, CommandHandler cmd, void *defaultdisplayfunc) {
searchCmd *acmd;
/* Nickname operations */
registersearchterm(reg_nicksearch, "hostmask",hostmask_parse, 0, "The user's nick!user@host; \"hostmask real\" returns nick!user@host\rreal"); /* nick only */
registersearchterm(reg_nicksearch, "realname",realname_parse, 0, "User's current realname"); /* nick only */
+ registersearchterm(reg_nicksearch, "away",away_parse, 0, "User's current away message"); /* nick only */
registersearchterm(reg_nicksearch, "authname",authname_parse, 0, "User's current authname or false"); /* nick only */
registersearchterm(reg_nicksearch, "authts",authts_parse, 0, "User's Auth timestamp"); /* nick only */
registersearchterm(reg_nicksearch, "ident",ident_parse, 0, "User's current ident"); /* nick only */
registersearchterm(reg_nicksearch, "server",server_parse, 0, "Server Name. Either (server string) or (match (server) string)"); /* nick only */
registersearchterm(reg_nicksearch, "authid",authid_parse, 0, "User's Auth ID"); /* nick only */
registersearchterm(reg_nicksearch, "cidr",cidr_parse, 0, "CIDR matching"); /* nick only */
+ registersearchterm(reg_nicksearch, "ipvsix",ipv6_parse, 0, "IPv6 user"); /* nick only */
/* Channel operations */
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, "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 */
/* 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");
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;
return CMD_OK;
}
-void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, searchCmd *cmd, nick *np) {
+void newsearch_ctxinit(searchCtx *ctx, searchParseFunc searchfn, replyFunc replyfn, wallFunc wallfn, void *arg, searchCmd *cmd, nick *np, void *displayfn, int limit) {
memset(ctx, 0, sizeof(searchCtx));
ctx->reply = replyfn;
ctx->arg = arg;
ctx->searchcmd = cmd;
ctx->sender = np;
+ ctx->limit = limit;
+ 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) {
reply( sender, "Usage: [flags] <criteria>");
rejoinline(cargv[arg],cargc-arg);
}
- newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, reg_nicksearch, sender);
-
- 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);
- (search->free)(&ctx, search);
+ parse_free(tree);
return CMD_OK;
}
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();
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) {
reply( sender, "Usage: [flags] <criteria>");
rejoinline(cargv[arg],cargc-arg);
}
- newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, reg_chansearch, sender);
- 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);
- (search->free)(&ctx, search);
+ parse_free(tree);
return CMD_OK;
}
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);
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) {
reply( sender, "Usage: [flags] <criteria>");
rejoinline(cargv[arg],cargc-arg);
}
- newsearch_ctxinit(&ctx, search_parse, reply, wall, NULL, reg_usersearch, sender);
-
- 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);
- (search->free)(&ctx, search);
+ parse_free(tree);
return CMD_OK;
}
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;i<AUTHNAMEHASHSIZE;i++) {
free(thenode);
}
-/* search_parse:
- * Given an input string, return a searchNode.
- */
-
-static int unescape(char *input, char *output, size_t buflen) {
- char *ch, *ch2;
- int e=0;
-
- if (*input=='\"') {
- for (ch=input;*ch;ch++) {
- if(ch - input >= buflen) {
- parseError="Buffer overflow";
- return 0;
- }
- }
-
- if (*(ch-1) != '\"') {
- parseError="Quote mismatch";
- return 0;
- }
-
- *(ch-1)='\0';
- input++;
- }
-
- ch2=output;
- for (ch=input;*ch;ch++) {
- if(ch - input >= buflen) {
- parseError="Buffer overflow";
- return 0;
- }
-
- if (e) {
- e=0;
- *ch2++=*ch;
- } else if (*ch=='\\') {
- e=1;
- } else {
- *ch2++=*ch;
- }
- }
- *ch2='\0';
-
- return 1;
-}
-
-struct searchNode *search_parse(searchCtx *ctx, char *cinput) {
- /* OK, we need to split the input into chunks on spaces and brackets.. */
- char *argvector[100];
- char inputb[1024];
- char *input;
- char thestring[500];
- int i,j,q=0,e=0,k;
- char *ch;
- struct Command *cmd;
- struct searchNode *thenode;
-
- strlcpy(inputb, cinput, sizeof(inputb));
- input = inputb;
-
- /* 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++) {
- /*printf("i: %d j: %d e: %d q: %d ch: '%c'\n", i, j, e, q, *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=='\\') {
- e=1;
- } else 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 */
-
- for(k=1;k<=j;k++)
- if(!unescape(argvector[k], argvector[k], sizeof(inputb)))
- return NULL;
-
- if (!(cmd=findcommandintree(ctx->searchcmd->searchtree,argvector[0],1))) {
- parseError = "Unknown command (for valid command list, see help <searchcmd>)";
- return NULL;
- } else {
- if (!controlpermitted(cmd->level, ctx->sender)) {
- parseError = "Access denied (for valid command list, see help <searchcmd>)";
- return NULL;
- }
- return ((parseFunc)cmd->handler)(ctx, j, argvector+1);
- }
- } else {
- /* Literal */
-
- /* slug: disabled now we unescape during the main parse stage */
- /*
- if(!unescape(input, thestring, sizeof(thestring)))
- return NULL;
-*/
-
- if (!(thenode=(struct searchNode *)malloc(sizeof(struct searchNode)))) {
- parseError = "malloc: could not allocate memory for this search.";
- return NULL;
- }
-
- thenode->localdata = getsstring(input,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;
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;
+}