2 * This is the first client module for newserv :)
4 * A very simple bot which should give people some ideas for how to
5 * implement stuff on this thing
8 #include "../irc/irc_config.h"
9 #include "../parser/parser.h"
10 #include "../localuser/localuser.h"
11 #include "../localuser/localuserchannel.h"
12 #include "../nick/nick.h"
13 #include "../lib/sstring.h"
14 #include "../core/config.h"
15 #include "../irc/irc.h"
16 #include "../lib/splitline.h"
17 #include "../channel/channel.h"
18 #include "../lib/flags.h"
19 #include "../core/schedule.h"
20 #include "../lib/base64.h"
21 #include "../core/modules.h"
22 #include "../lib/version.h"
23 #include "../lib/irc_string.h"
36 CommandTree
*controlcmds
;
37 ControlMsg controlreply
;
38 ControlWall controlwall
;
39 ControlPermitted controlpermitted
;
40 DestroyExt controldestroyext
;
42 void controldestroycmdext(void *ext
);
43 void handlemessages(nick
*target
, int messagetype
, void **args
);
44 int controlstatus(void *sender
, int cargc
, char **cargv
);
45 void controlconnect(void *arg
);
46 int controlwhois(void *sender
, int cargc
, char **cargv
);
47 int controlchannel(void *sender
, int cargc
, char **cargv
);
48 int relink(void *sender
, int cargc
, char **cargv
);
49 int die(void *sender
, int cargc
, char **cargv
);
50 int controlinsmod(void *sender
, int cargc
, char **cargv
);
51 int controllsmod(void *sender
, int cargc
, char **cargv
);
52 int controlrehash(void *sender
, int cargc
, char **cargv
);
53 int controlreload(void *sender
, int cargc
, char **cargv
);
54 int controlhelpcmd(void *sender
, int cargc
, char **cargv
);
55 void controlnoticeopers(flag_t permissionlevel
, flag_t noticelevel
, char *format
, ...) __attribute__ ((format (printf
, 3, 4)));
56 void controlnoticeopers(flag_t permissionlevel
, flag_t noticelevel
, char *format
, ...);
57 int controlcheckpermitted(flag_t level
, nick
*user
);
58 void handlesignal(int hooknum
, void *arg
);
61 controlcmds
=newcommandtree();
62 controlreply
=&controlmessage
;
63 controlwall
=&controlnoticeopers
;
64 controlpermitted
=&controlcheckpermitted
;
65 controldestroyext
=&controldestroycmdext
;
67 registercontrolhelpcmd("status",NO_DEVELOPER
,1,&controlstatus
,"Usage: status ?level?\nDisplays status information, increasing level gives more verbose information.");
68 registercontrolhelpcmd("whois",NO_OPERED
,1,&controlwhois
,"Usage: whois <nickname|#numeric>\nDisplays lots of information about the specified nickname or numeric.");
69 registercontrolhelpcmd("channel",NO_OPER
,1,&controlchannel
,"Usage: channel <#channel>\nDisplays channel information.");
70 registercontrolhelpcmd("relink",NO_DEVELOPER
,1,&relink
,"Usage: relink\nRelinks service to the network.");
71 registercontrolhelpcmd("die",NO_DEVELOPER
,1,&die
,"Usage: die <reason>\nTerminates the service.");
72 registercontrolhelpcmd("insmod",NO_DEVELOPER
,1,&controlinsmod
,"Usage: insmod <module>\nAdds a module to the running instance.");
73 registercontrolhelpcmd("rmmod",NO_DEVELOPER
,1,&controlrmmod
,"Usage: rmmod <module>\nRemoves a module from the running instance.");
74 registercontrolhelpcmd("lsmod",NO_OPER
,0,&controllsmod
,"Usage: lsmod\nLists currently running modules.");
75 registercontrolhelpcmd("rehash",NO_DEVELOPER
,1,&controlrehash
,"Usage: rehash\nReloads configuration file.");
76 registercontrolhelpcmd("showcommands",NO_ACCOUNT
,0,&controlshowcommands
,"Usage: showcommands\nShows all registered commands.");
77 registercontrolhelpcmd("reload",NO_DEVELOPER
,1,&controlreload
,"Usage: reload <module>\nReloads specified module.");
78 registercontrolhelpcmd("help",NO_ANYONE
,1,&controlhelpcmd
,"Usage: help <command>\nShows help for specified command.");
80 registerhook(HOOK_CORE_REHASH
, &handlesignal
);
81 registerhook(HOOK_CORE_SIGINT
, &handlesignal
);
82 scheduleoneshot(time(NULL
)+1,&controlconnect
,NULL
);
86 deleteallschedules(&controlconnect
);
88 deregisterlocaluser(mynick
,"Leaving");
91 deregistercontrolcmd("status",&controlstatus
);
92 deregistercontrolcmd("whois",&controlwhois
);
93 deregistercontrolcmd("channel",&controlchannel
);
94 deregistercontrolcmd("relink",&relink
);
95 deregistercontrolcmd("die",&die
);
96 deregistercontrolcmd("insmod",&controlinsmod
);
97 deregistercontrolcmd("rmmod",&controlrmmod
);
98 deregistercontrolcmd("lsmod",&controllsmod
);
99 deregistercontrolcmd("rehash",&controlrehash
);
100 deregistercontrolcmd("showcommands",&controlshowcommands
);
101 deregistercontrolcmd("reload",&controlreload
);
102 deregistercontrolcmd("help",&controlhelpcmd
);
104 destroycommandtree(controlcmds
);
106 deregisterhook(HOOK_CORE_REHASH
, &handlesignal
);
107 deregisterhook(HOOK_CORE_SIGINT
, &handlesignal
);
110 void registercontrolhelpcmd(const char *name
, int level
, int maxparams
, CommandHandler handler
, char *helpstr
) {
114 newcmd
= addcommandtotree(controlcmds
,name
,level
,maxparams
,handler
);
117 /* Allocate a help record */
118 help
=(cmdhelp
*)malloc(sizeof(cmdhelp
));
120 Error("control",ERR_ERROR
,"Malloc failed: registercontrolhelpcmd");
123 memset((void *)help
,0,sizeof(cmdhelp
));
125 /* this isn't an sstring atm, as sstring has a max length (512) */
126 int len
=strlen(helpstr
);
127 help
->helpstr
=(char *)malloc(len
+1);
129 strncpy(help
->helpstr
, helpstr
, len
);
130 help
->helpstr
[len
] = '\0';
132 newcmd
->ext
=(void *)help
;
133 newcmd
->destroyext
=controldestroyext
;
137 void registercontrolhelpfunccmd(const char *name
, int level
, int maxparams
, CommandHandler handler
, CommandHelp helpcmd
) {
141 /* Allocate a help record */
142 help
=(cmdhelp
*)malloc(sizeof(cmdhelp
));
144 Error("control",ERR_ERROR
,"Malloc failed: registercontrolhelpfunccmd");
147 memset((void *)help
,0,sizeof(cmdhelp
));
149 help
->helpcmd
=helpcmd
;
151 newcmd
= addcommandexttotree(controlcmds
,name
,level
,maxparams
,handler
, (void *)help
);
152 newcmd
->destroyext
=controldestroyext
;
156 int deregistercontrolcmd(const char *name
, CommandHandler handler
) {
157 return deletecommandfromtree(controlcmds
, name
, handler
);
160 void controlconnect(void *arg
) {
161 sstring
*cnick
, *myident
, *myhost
, *myrealname
, *myauthname
;
164 cnick
=getcopyconfigitem("control","nick","C",NICKLEN
);
165 myident
=getcopyconfigitem("control","ident","control",NICKLEN
);
166 myhost
=getcopyconfigitem("control","hostname",myserver
->content
,HOSTLEN
);
167 myrealname
=getcopyconfigitem("control","realname","NewServ Control Service",REALLEN
);
168 myauthname
=getcopyconfigitem("control","authname","C",ACCOUNTLEN
);
170 mynick
=registerlocaluser(cnick
->content
,myident
->content
,myhost
->content
,myrealname
->content
,myauthname
->content
,UMODE_SERVICE
|UMODE_DEAF
|UMODE_OPER
|UMODE_ACCOUNT
|UMODE_INV
,&handlemessages
);
171 triggerhook(HOOK_CONTROL_REGISTERED
, mynick
);
172 cp
=findchannel("#twilightzone");
174 localcreatechannel(mynick
,"#twilightzone");
176 localjoinchannel(mynick
,cp
);
177 localgetops(mynick
,cp
);
181 freesstring(myident
);
183 freesstring(myrealname
);
184 freesstring(myauthname
);
187 void handlestats(int hooknum
, void *arg
) {
188 controlreply(hooknick
,"%s",(char *)arg
);
191 int controlstatus(void *sender
, int cargc
, char **cargv
) {
192 unsigned long level
=999;
193 hooknick
=(nick
*)sender
;
196 level
=strtoul(cargv
[0],NULL
,10);
199 registerhook(HOOK_CORE_STATSREPLY
,&handlestats
);
201 triggerhook(HOOK_CORE_STATSREQUEST
,(void *)level
);
202 deregisterhook(HOOK_CORE_STATSREPLY
,&handlestats
);
206 int controlrehash(void *sender
, int cargc
, char **cargv
) {
207 nick
*np
=(nick
*)sender
;
209 controlreply(np
,"Rehashing the config file.");
212 triggerhook(HOOK_CORE_REHASH
,(void *)0);
216 void handlewhois(int hooknum
, void *arg
) {
217 controlreply(hooknick
,"%s",(char *)arg
);
220 int controlwhois(void *sender
, int cargc
, char **cargv
) {
229 if (cargv
[0][0]=='#') {
230 if (!(target
=getnickbynumericstr(cargv
[0]+1))) {
231 controlreply(sender
,"Sorry, couldn't find numeric %s",cargv
[0]+1);
235 if ((target
=getnickbynick(cargv
[0]))==NULL
) {
236 controlreply((nick
*)sender
,"Sorry, couldn't find that user");
241 controlreply((nick
*)sender
,"Nick : %s",target
->nick
);
242 controlreply((nick
*)sender
,"Numeric : %s",longtonumeric(target
->numeric
,5));
243 controlreply((nick
*)sender
,"User@Host : %s@%s (%d user(s) on this host)",target
->ident
,target
->host
->name
->content
,target
->host
->clonecount
);
244 if (IsSetHost(target
)) {
245 if (target
->shident
) {
246 controlreply((nick
*)sender
,"Fakehost : %s@%s",target
->shident
->content
, target
->sethost
->content
);
248 controlreply((nick
*)sender
,"Fakehost : %s",target
->sethost
->content
);
251 controlreply((nick
*)sender
,"Timestamp : %lu",target
->timestamp
);
252 controlreply((nick
*)sender
,"IP address: %s",IPtostr(target
->p_ipaddr
));
253 controlreply((nick
*)sender
,"Realname : %s (%d user(s) have this realname)",target
->realname
->name
->content
,target
->realname
->usercount
);
254 if (target
->umodes
) {
255 controlreply((nick
*)sender
,"Umode(s) : %s",printflags(target
->umodes
,umodeflags
));
257 if (IsOper(target
) && target
->opername
)
258 controlreply((nick
*)sender
,"Opered as : %s",target
->opername
->content
);
259 if (IsAccount(target
)) {
260 controlreply((nick
*)sender
,"Account : %s",target
->authname
);
261 if (target
->accountts
)
262 controlreply((nick
*)sender
,"AccountTS : %ld",target
->accountts
);
264 controlreply((nick
*)sender
,"UserID : %ld",target
->auth
->userid
);
265 if (target
->auth
->flags
)
266 controlreply((nick
*)sender
,"AccFlags : %s",printflags(target
->auth
->flags
,accountflags
));
271 controlreply((nick
*)sender
, "Away : %s",target
->away
->content
);
274 hooknick
=(nick
*)sender
;
275 registerhook(HOOK_CONTROL_WHOISREPLY
,&handlewhois
);
276 triggerhook(HOOK_CONTROL_WHOISREQUEST
,target
);
277 deregisterhook(HOOK_CONTROL_WHOISREPLY
,&handlewhois
);
279 if (target
->channels
->cursi
==0) {
280 controlreply((nick
*)sender
,"Channels : none");
281 } else if (target
->channels
->cursi
>50) {
282 controlreply((nick
*)sender
,"Channels : - (total: %d)",target
->channels
->cursi
);
285 channels
=(channel
**)target
->channels
->content
;
286 for (i
=0;i
<=target
->channels
->cursi
;i
++) {
287 if (!((i
==target
->channels
->cursi
) || ((70-strlen(buf
))<channels
[i
]->index
->name
->length
&& strlen(buf
)>0))) {
288 strcat(buf
,channels
[i
]->index
->name
->content
);
291 if (strlen(buf
)==0) {
294 controlreply((nick
*)sender
,"Channels : %s",buf
);
305 int controlinsmod(void *sender
, int cargc
, char **cargv
) {
309 switch(insmod(cargv
[0])) {
311 controlreply((nick
*)sender
,"Unable to load module %s",cargv
[0]);
315 controlreply((nick
*)sender
,"Module %s already loaded, or name not valid",cargv
[0]);
319 controlreply((nick
*)sender
,"Module %s loaded.",cargv
[0]);
323 controlreply((nick
*)sender
,"An unknown error occured.");
328 int controlrmmod(void *sender
, int cargc
, char **cargv
) {
332 switch(rmmod(cargv
[0])) {
334 controlreply((nick
*)sender
,"Module %s is not loaded.",cargv
[0]);
338 controlreply((nick
*)sender
,"Module %s unloaded.",cargv
[0]);
342 controlreply((nick
*)sender
,"An unknown error occured.");
347 int controllsmod(void *sender
, int cargc
, char **cargv
) {
351 if (cargc
< 1) { /* list all loaded modules */
352 const char *ver
, *buildid
;
353 time_t t
, t2
= time(NULL
);
354 ptr
= lsmod(i
, &ver
, &buildid
, &t
);
356 /* 9999d 24h 59m 59s fbf2a4a69ee1-tip */
357 controlreply((nick
*)sender
,"Module Loaded for Version Build id");
358 while (ptr
!= NULL
) {
359 controlreply((nick
*)sender
," %-40s %-17s %-30s %s", ptr
, longtoduration(t2
-t
, 2), ver
?ver
:"", buildid
?buildid
:"");
360 ptr
= lsmod(++i
, &ver
, &buildid
, &t
);
363 ptr
= lsmod(getindex(cargv
[0]), NULL
, NULL
, NULL
);
364 controlreply((nick
*)sender
,"Module \"%s\" %s", cargv
[0], (ptr
? "is loaded." : "is NOT loaded."));
369 int controlreload(void *sender
, int cargc
, char **cargv
) {
373 controlreply((nick
*)sender
,"Imma gonna try and reload %s",cargv
[0]);
375 safereload(cargv
[0]);
380 int relink(void *sender
, int cargc
, char **cargv
) {
382 controlreply((nick
*)sender
,"You must give a reason.");
386 irc_send("%s SQ %s 0 :%s",mynumeric
->content
,myserver
->content
,cargv
[0]);
392 int die(void *sender
, int cargc
, char **cargv
) {
393 if (cargc
<1 || (strlen(cargv
[0]) < 10)) {
394 controlreply((nick
*)sender
,"You must give a reason.");
398 controlwall(NO_OPER
,NL_OPERATIONS
,"DIE from %s: %s",((nick
*)sender
)->nick
, cargv
[0]);
400 newserv_shutdown_pending
=1;
405 int controlchannel(void *sender
, int cargc
, char **cargv
) {
411 int i
,j
, ops
=0, voice
=0;
417 if ((cp
=findchannel(cargv
[0]))==NULL
) {
418 controlreply((nick
*)sender
,"Couldn't find channel: %s",cargv
[0]);
423 sprintf(buf2
,"%d",cp
->limit
);
426 controlreply((nick
*)sender
,"Channel : %s",cp
->index
->name
->content
);
427 strftime(timebuf
, 30, "%d/%m/%y %H:%M", localtime(&(cp
->timestamp
)));
428 controlreply((nick
*)sender
,"C-time : %ld [%s]",cp
->timestamp
,timebuf
);
430 controlreply((nick
*)sender
,"Topic : %s",cp
->topic
->content
);
431 strftime(timebuf
, 30, "%d/%m/%y %H:%M", localtime(&(cp
->topictime
)));
432 controlreply((nick
*)sender
,"T-time : %ld [%s]",cp
->topictime
,timebuf
);
434 controlreply((nick
*)sender
,"Topic : (none)");
436 controlreply((nick
*)sender
,"Mode(s) : %s %s%s%s",printflags(cp
->flags
,cmodeflags
),IsLimit(cp
)?buf2
:"",
437 IsLimit(cp
)?" ":"",IsKey(cp
)?cp
->key
->content
:"");
438 controlreply((nick
*)sender
,"Users : %d (hash size %d, utilisation %.1f%%); %d unique hosts",
439 cp
->users
->totalusers
,cp
->users
->hashsize
,((float)(100*cp
->users
->totalusers
)/cp
->users
->hashsize
),
440 countuniquehosts(cp
));
444 for (j
=0;j
<=cp
->users
->hashsize
;j
++) {
445 if (i
==4 || j
==cp
->users
->hashsize
) {
447 controlreply((nick
*)sender
,"Users : %s",buf
);
451 if (j
==cp
->users
->hashsize
)
454 if (cp
->users
->content
[j
]!=nouser
) {
455 if (cp
->users
->content
[j
]&CUMODE_OP
)
457 else if (cp
->users
->content
[j
]&CUMODE_VOICE
)
459 np
=getnickbynumeric(cp
->users
->content
[j
]);
460 sprintf(&buf
[i
*18],"%c%c%-15s ",cp
->users
->content
[j
]&CUMODE_VOICE
?'+':' ',
461 cp
->users
->content
[j
]&CUMODE_OP
?'@':' ', np
?np
->nick
:"!BUG-NONICK!");
467 controlreply((nick
*)sender
, "Users : Opped: %d, Voiced: %d", ops
,voice
);
468 for (cbp
=cp
->bans
;cbp
;cbp
=cbp
->next
) {
469 controlreply((nick
*)sender
,"Ban : %s",bantostringdebug(cbp
));
474 int controlshowcommands(void *sender
, int cargc
, char **cargv
) {
475 nick
*np
=(nick
*)sender
;
476 Command
*cmdlist
[100];
479 n
=getcommandlist(controlcmds
,cmdlist
,100);
481 controlreply(np
,"The following commands are registered at present:");
484 controlreply(np
,"%s",cmdlist
[i
]->command
->content
);
487 controlreply(np
,"End of list.");
491 void handlemessages(nick
*target
, int messagetype
, void **args
) {
497 switch(messagetype
) {
500 /* If it's a message, first arg is nick and second is message */
501 sender
=(nick
*)args
[0];
503 Error("control",ERR_INFO
,"From: %s!%s@%s: %s",sender
->nick
,sender
->ident
,sender
->host
->name
->content
, (char *)args
[1]);
505 /* Split the line into params */
506 cargc
=splitline((char *)args
[1],cargv
,50,0);
513 cmd
=findcommandintree(controlcmds
,cargv
[0],1);
515 controlreply(sender
,"Unknown command.");
519 if (cmd
->level
>0 && !IsOper(sender
)) {
520 controlreply(sender
,"You need to be opered to use this command.");
524 /* If we were doing "authed user tracking" here we'd put a check in for authlevel */
526 /* Check the maxargs */
527 if (cmd
->maxparams
<(cargc
-1)) {
528 /* We need to do some rejoining */
529 rejoinline(cargv
[cmd
->maxparams
],cargc
-(cmd
->maxparams
));
530 cargc
=(cmd
->maxparams
)+1;
533 if((cmd
->handler
)((void *)sender
,cargc
-1,&(cargv
[1])) == CMD_USAGE
)
534 controlhelp(sender
, cmd
);
538 /* someone killed me? Bastards */
539 scheduleoneshot(time(NULL
)+1,&controlconnect
,NULL
);
541 triggerhook(HOOK_CONTROL_REGISTERED
, NULL
);
549 void controlmessage(nick
*target
, char *message
, ... ) {
557 va_start(va
,message
);
558 vsnprintf(buf
,512,message
,va
);
561 sendmessagetouser(mynick
,target
,"%s",buf
);
564 void controlchanmsg(channel
*cp
, char *message
, ...) {
572 va_start(va
,message
);
573 vsnprintf(buf
,512,message
,va
);
576 sendmessagetochannel(mynick
,cp
,"%s",buf
);
579 void controlnotice(nick
*target
, char *message
, ... ) {
587 va_start(va
,message
);
588 vsnprintf(buf
,512,message
,va
);
591 sendnoticetouser(mynick
,target
,"%s",buf
);
594 void controlspecialrmmod(void *arg
) {
595 struct specialsched
*a
= (struct specialsched
*)arg
;
596 sstring
*froo
= a
->modulename
;
600 rmmod(froo
->content
);
604 void controlspecialreloadmod(void *arg
) {
605 struct specialsched
*a
= (struct specialsched
*)arg
;
606 sstring
*froo
= a
->modulename
;
610 safereload(froo
->content
);
614 void controlhelp(nick
*np
, Command
*cmd
) {
617 cmdhelp
*help
=(cmdhelp
*)cmd
->ext
;
620 controlreply(np
, "Sorry, no help for this command.");
623 if ( help
->helpcmd
) {
624 (help
->helpcmd
)(np
, cmd
);
628 controlreply(np
, "Sorry, no help for this command.");
634 if(*cp
== '\0' || *cp
== '\n') {
642 controlreply(np
, "%s", sp
);
656 int controlhelpcmd(void *sender
, int cargc
, char **cargv
) {
658 nick
*np
= (nick
*)sender
;
663 cmd
=findcommandintree(controlcmds
,cargv
[0],1);
665 controlreply(np
,"Unknown command.");
669 controlhelp(np
, cmd
);
673 void controlnoticeopers(flag_t permissionlevel
, flag_t noticelevel
, char *format
, ...) {
679 va_start(va
, format
);
680 vsnprintf(broadcast
, sizeof(broadcast
), format
, va
);
683 for(i
=0;i
<NICKHASHSIZE
;i
++)
684 for(np
=nicktable
[i
];np
;np
=np
->next
)
686 controlnotice(np
, "%s", broadcast
);
689 void controlnswall(int noticelevel
, char *format
, ...) {
693 va_start(va
, format
);
694 vsnprintf(broadcast
, sizeof(broadcast
), format
, va
);
697 controlwall(NO_OPER
, noticelevel
, "%s", broadcast
);
700 int controlcheckpermitted(flag_t level
, nick
*user
) {
704 void handlesignal(int hooknum
, void *arg
) {
705 char *signal
, *action
;
707 if(hooknum
== HOOK_CORE_SIGINT
) {
709 action
= "terminating";
711 long hupped
= (long)arg
;
716 action
= "rehashing";
719 controlwall(NO_OPER
, NL_OPERATIONS
, "SIG%s received, %s...", signal
, action
);
722 void controldestroycmdext(void *ext
) {
723 if ( ((cmdhelp
*)ext
)->helpstr
)
724 free( ((cmdhelp
*)ext
)->helpstr
);
728 char *controlid(nick
*np
) {
729 static char buf
[512];
731 snprintf(buf
, sizeof(buf
), "%s!%s@%s/%s", np
->nick
, np
->ident
, np
->host
->name
->content
, np
->authname
);