#include "../irc/irc_config.h"
#include "../lib/irc_string.h"
-void *hostmask_exe(struct searchNode *thenode, int type, void *theinput);
-void *hostmask_exe_real(struct searchNode *thenode, int type, void *theinput);
-void hostmask_free(struct searchNode *thenode);
+void *hostmask_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput);
+void *hostmask_exe_rn(searchCtx *ctx, struct searchNode *thenode, void *theinput);
+void *hostmask_exe_rh(searchCtx *ctx, struct searchNode *thenode, void *theinput);
+void *hostmask_exe_rn_rh(searchCtx *ctx, struct searchNode *thenode, void *theinput);
+void hostmask_free(searchCtx *ctx, struct searchNode *thenode);
-struct searchNode *hostmask_parse(int type, int argc, char **argv) {
+struct searchNode *hostmask_parse(searchCtx *ctx, int argc, char **argv) {
struct searchNode *thenode;
+ int realhost = 0, realname = 0, i, error = 0;
+
+ for(i=0;i<argc;i++) {
+ struct searchNode *convsn;
+ char *p;
+
+ if (!(convsn=argtoconststr("hostmask", ctx, argv[i], &p)))
+ return NULL;
+
+ if(!ircd_strcmp(p, "realhost")) {
+ realhost = 1;
+ } else if(!ircd_strcmp(p, "realname")) {
+ realname = 1;
+ } else {
+ error = 1;
+ }
+ convsn->free(ctx, convsn);
+
+ if(error) {
+ parseError = "bad argument: use realhost and/or realname";
+ return NULL;
+ }
+ }
- if (type != SEARCHTYPE_NICK) {
- parseError = "hostmask: this function is only valid for nick searches.";
+ if (!(thenode=(struct searchNode *)malloc(sizeof (struct searchNode)))) {
+ parseError = "malloc: could not allocate memory for this search.";
return NULL;
}
- thenode=(struct searchNode *)malloc(sizeof (struct searchNode));
-
thenode->returntype = RETURNTYPE_STRING;
- thenode->localdata = (void *)malloc(HOSTLEN+USERLEN+NICKLEN+3);
- thenode->exe = hostmask_exe;
+ if (!(thenode->localdata = (void *)malloc(HOSTLEN+USERLEN+NICKLEN+REALLEN+10))) {
+ /* couldn't malloc() memory for thenode->localdata, so free thenode to avoid leakage */
+ parseError = "malloc: could not allocate memory for this search.";
+ free(thenode);
+ return NULL;
+ }
thenode->free = hostmask_free;
- /* Allow "hostmask real" to match realhost */
-
- if (argc>0 && !ircd_strcmp(argv[0],"real")) {
- thenode->exe = hostmask_exe_real;
+ if(realname) {
+ if(realhost) {
+ thenode->exe = hostmask_exe_rn_rh;
+ } else {
+ thenode->exe = hostmask_exe_rn;
+ }
+ } else {
+ if(realhost) {
+ thenode->exe = hostmask_exe_rh;
+ } else {
+ thenode->exe = hostmask_exe;
+ }
}
return thenode;
}
-void *hostmask_exe(struct searchNode *thenode, int type, void *theinput) {
+void *hostmask_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
nick *np = (nick *)theinput;
char *buf = thenode->localdata;
- if (type != RETURNTYPE_STRING) {
- return (void *)1;
- }
-
return visiblehostmask(np, buf);
}
-void *hostmask_exe_real(struct searchNode *thenode, int type, void *theinput) {
+void *hostmask_exe_rh(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
nick *np = (nick *)theinput;
char *buf = thenode->localdata;
- if (type != RETURNTYPE_STRING) {
- return (void *)1;
- }
-
sprintf(buf,"%s!%s@%s",np->nick,np->ident,np->host->name->content);
return buf;
}
-void hostmask_free(struct searchNode *thenode) {
+void *hostmask_exe_rn_rh(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ nick *np = (nick *)theinput;
+ char *buf = thenode->localdata;
+
+ sprintf(buf,"%s!%s@%s\r%s",np->nick,np->ident,np->host->name->content,np->realname->name->content);
+
+ return buf;
+}
+
+void *hostmask_exe_rn(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
+ nick *np = (nick *)theinput;
+ char *buf = thenode->localdata;
+
+ sprintf(buf,"%s\r%s",visiblehostmask(np, buf),np->realname->name->content);
+
+ return buf;
+}
+
+void hostmask_free(searchCtx *ctx, struct searchNode *thenode) {
free(thenode->localdata);
free(thenode);
}