control/control.c

   1 /*
   2  * This is the first client module for newserv :)
   3  *
   4  * A very simple bot which should give people some ideas for how to
   5  * implement stuff on this thing
   6  */
   7
   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 "control.h"
  24
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <stdarg.h>
  28
  29 MODULE_VERSION("");
  30
  31 nick *hooknick;
  32
  33 nick *mynick;
  34
  35 CommandTree *controlcmds;
  36 ControlMsg controlreply;
  37 ControlWall controlwall;
  38
  39 void handlemessages(nick *target, int messagetype, void **args);
  40 int controlstatus(void *sender, int cargc, char **cargv);
  41 void controlconnect(void *arg);
  42 int controlwhois(void *sender, int cargc, char **cargv);
  43 int controlchannel(void *sender, int cargc, char **cargv);
  44 int relink(void *sender, int cargc, char **cargv);
  45 int die(void *sender, int cargc, char **cargv);
  46 int controlinsmod(void *sender, int cargc, char **cargv);
  47 int controllsmod(void *sender, int cargc, char **cargv);
  48 int controlrehash(void *sender, int cargc, char **cargv);
  49 int controlreload(void *sender, int cargc, char **cargv);
  50 int controlhelpcmd(void *sender, int cargc, char **cargv);
  51 void controlnoticeopers(flag_t permissionlevel, flag_t noticelevel, char *format, ...);
  52
  53 void _init() {
  54   controlcmds=newcommandtree();
  55   controlreply=&controlmessage;
  56   controlwall=&controlnoticeopers;
  57
  58   registercontrolhelpcmd("status",NO_DEVELOPER,1,&controlstatus,"Usage: status ?level?\nDisplays status information, increasing level gives more verbose information.");
  59   registercontrolhelpcmd("whois",NO_OPERED,1,&controlwhois,"Usage: whois <nickname|#numeric>\nDisplays lots of information about the specified nickname or numeric.");
  60   registercontrolhelpcmd("channel",NO_OPER,1,&controlchannel,"Usage: channel <#channel>\nDisplays channel information.");
  61   registercontrolhelpcmd("relink",NO_DEVELOPER,1,&relink,"Usage: relink\nRelinks service to the network.");
  62   registercontrolhelpcmd("die",NO_DEVELOPER,1,&die,"Usage: die <reason>\nTerminates the service.");
  63   registercontrolhelpcmd("insmod",NO_DEVELOPER,1,&controlinsmod,"Usage: insmod <module>\nAdds a module to the running instance.");
  64   registercontrolhelpcmd("rmmod",NO_DEVELOPER,1,&controlrmmod,"Usage: rmmod <module>\nRemoves a module from the running instance.");
  65   registercontrolhelpcmd("lsmod",NO_DEVELOPER,0,&controllsmod,"Usage: lsmod\nLists currently running modules.");
  66   registercontrolhelpcmd("rehash",NO_DEVELOPER,1,&controlrehash,"Usage: rehash\nReloads configuration file.");
  67   registercontrolhelpcmd("showcommands",NO_ACCOUNT,0,&controlshowcommands,"Usage: showcommands\nShows all registered commands.");
  68   registercontrolhelpcmd("reload",NO_DEVELOPER,1,&controlreload,"Usage: reload <module>\nReloads specified module.");
  69   registercontrolhelpcmd("help",NO_ANYONE,1,&controlhelpcmd,"Usage: help <command>\nShows help for specified command.");
  70
  71   scheduleoneshot(time(NULL)+1,&controlconnect,NULL);
  72 }
  73
  74 void _fini() {
  75   deleteallschedules(&controlconnect);
  76   if (mynick) {
  77     deregisterlocaluser(mynick,"Leaving");
  78   }
  79
  80   deregistercontrolcmd("status",&controlstatus);
  81   deregistercontrolcmd("whois",&controlwhois);
  82   deregistercontrolcmd("channel",&controlchannel);
  83   deregistercontrolcmd("relink",&relink);
  84   deregistercontrolcmd("die",&die);
  85   deregistercontrolcmd("insmod",&controlinsmod);
  86   deregistercontrolcmd("rmmod",&controlrmmod);
  87   deregistercontrolcmd("lsmod",&controllsmod);
  88   deregistercontrolcmd("rehash",&controlrehash);
  89   deregistercontrolcmd("showcommands",&controlshowcommands);
  90   deregistercontrolcmd("reload",&controlreload);
  91   deregistercontrolcmd("help",&controlhelpcmd);
  92
  93   destroycommandtree(controlcmds);
  94 }
  95
  96 void registercontrolhelpcmd(const char *name, int level, int maxparams, CommandHandler handler, char *help) {
  97   addcommandhelptotree(controlcmds,name,level,maxparams,handler,help);
  98 }
  99
 100 int deregistercontrolcmd(const char *name, CommandHandler handler) {
 101   return deletecommandfromtree(controlcmds, name, handler);
 102 }
 103
 104 void controlconnect(void *arg) {
 105   sstring *cnick, *myident, *myhost, *myrealname, *myauthname;
 106   channel *cp;
 107
 108   cnick=getcopyconfigitem("control","nick","C",NICKLEN);
 109   myident=getcopyconfigitem("control","ident","control",NICKLEN);
 110   myhost=getcopyconfigitem("control","hostname",myserver->content,HOSTLEN);
 111   myrealname=getcopyconfigitem("control","realname","NewServ Control Service",REALLEN);
 112   myauthname=getcopyconfigitem("control","authname","C",ACCOUNTLEN);
 113
 114   mynick=registerlocaluser(cnick->content,myident->content,myhost->content,myrealname->content,myauthname->content,UMODE_SERVICE|UMODE_DEAF|UMODE_OPER|UMODE_ACCOUNT|UMODE_INV,&handlemessages);
 115   triggerhook(HOOK_CONTROL_REGISTERED, mynick);
 116   cp=findchannel("#twilightzone");
 117   if (!cp) {
 118     localcreatechannel(mynick,"#twilightzone");
 119   } else {
 120     localjoinchannel(mynick,cp);
 121     localgetops(mynick,cp);
 122   }
 123
 124   freesstring(cnick);
 125   freesstring(myident);
 126   freesstring(myhost);
 127   freesstring(myrealname);
 128   freesstring(myauthname);
 129 }
 130
 131 void handlestats(int hooknum, void *arg) {
 132   controlreply(hooknick,"%s",(char *)arg);
 133 }
 134
 135 int controlstatus(void *sender, int cargc, char **cargv) {
 136   unsigned long level=999;
 137   hooknick=(nick *)sender;
 138
 139   if (cargc>0) {
 140     level=strtoul(cargv[0],NULL,10);
 141   }
 142
 143   registerhook(HOOK_CORE_STATSREPLY,&handlestats);
 144
 145   triggerhook(HOOK_CORE_STATSREQUEST,(void *)level);
 146   deregisterhook(HOOK_CORE_STATSREPLY,&handlestats);
 147   return CMD_OK;
 148 }
 149
 150 int controlrehash(void *sender, int cargc, char **cargv) {
 151   nick *np=(nick *)sender;
 152
 153   controlreply(np,"Rehashing the config file.");
 154
 155   rehashconfig();
 156   triggerhook(HOOK_CORE_REHASH,(void *)0);
 157   return CMD_OK;
 158 }
 159
 160 void handlewhois(int hooknum, void *arg) {
 161   controlreply(hooknick,"%s",(char *)arg);
 162 }
 163
 164 int controlwhois(void *sender, int cargc, char **cargv) {
 165   nick *target;
 166   channel **channels;
 167   char buf[BUFSIZE];
 168   int i;
 169
 170   if (cargc<1)
 171     return CMD_USAGE;
 172
 173   if (cargv[0][0]=='#') {
 174     if (!(target=getnickbynumericstr(cargv[0]+1))) {
 175       controlreply(sender,"Sorry, couldn't find numeric %s",cargv[0]+1);
 176       return CMD_ERROR;
 177     }
 178   } else {
 179     if ((target=getnickbynick(cargv[0]))==NULL) {
 180       controlreply((nick *)sender,"Sorry, couldn't find that user");
 181       return CMD_ERROR;
 182     }
 183   }
 184
 185   controlreply((nick *)sender,"Nick      : %s",target->nick);
 186   controlreply((nick *)sender,"Numeric   : %s",longtonumeric(target->numeric,5));
 187   controlreply((nick *)sender,"User@Host : %s@%s (%d user(s) on this host)",target->ident,target->host->name->content,target->host->clonecount);
 188   if (IsSetHost(target)) {
 189     if (target->shident) {
 190       controlreply((nick *)sender,"Fakehost  : %s@%s",target->shident->content, target->sethost->content);
 191     } else {
 192       controlreply((nick *)sender,"Fakehost  : %s",target->sethost->content);
 193     }
 194   }
 195   controlreply((nick *)sender,"Timestamp : %lu",target->timestamp);
 196   controlreply((nick *)sender,"IP address: %s",IPtostr(target->p_ipaddr));
 197   controlreply((nick *)sender,"Realname  : %s (%d user(s) have this realname)",target->realname->name->content,target->realname->usercount);
 198   if (target->umodes) {
 199     controlreply((nick *)sender,"Umode(s)  : %s",printflags(target->umodes,umodeflags));
 200   }
 201   if (IsOper(target) && target->opername)
 202     controlreply((nick *)sender,"Opered as : %s",target->opername->content);
 203   if (IsAccount(target)) {
 204     controlreply((nick *)sender,"Account   : %s",target->authname);
 205     if (target->accountts)
 206       controlreply((nick *)sender,"AccountTS : %ld",target->accountts);
 207     if (target->auth)  {
 208       controlreply((nick *)sender,"UserID    : %ld",target->auth->userid);
 209       if (target->auth->flags)
 210         controlreply((nick *)sender,"AccFlags  : %s",printflags(target->auth->flags,accountflags));
 211     }
 212   }
 213
 214   hooknick=(nick *)sender;
 215   registerhook(HOOK_CONTROL_WHOISREPLY,&handlewhois);
 216   triggerhook(HOOK_CONTROL_WHOISREQUEST,target);
 217   deregisterhook(HOOK_CONTROL_WHOISREPLY,&handlewhois);
 218
 219   if (target->channels->cursi==0) {
 220     controlreply((nick *)sender,"Channels  : none");
 221   } else if (target->channels->cursi>50) {
 222     controlreply((nick *)sender,"Channels  : - (total: %d)",target->channels->cursi);
 223   } else {
 224     buf[0]='\0';
 225     channels=(channel **)target->channels->content;
 226     for (i=0;i<=target->channels->cursi;i++) {
 227       if (!((i==target->channels->cursi) || ((70-strlen(buf))<channels[i]->index->name->length && strlen(buf)>0))) {
 228         strcat(buf,channels[i]->index->name->content);
 229         strcat(buf," ");
 230       } else {
 231         if (strlen(buf)==0) {
 232           break;
 233         } else {
 234           controlreply((nick *)sender,"Channels  : %s",buf);
 235           buf[0]='\0';
 236           i--;
 237         }
 238       }
 239     }
 240   }
 241
 242   return CMD_OK;
 243 }
 244
 245 int controlinsmod(void *sender, int cargc, char **cargv) {
 246   if (cargc<1)
 247     return CMD_USAGE;
 248
 249   switch(insmod(cargv[0])) {
 250     case -1:
 251       controlreply((nick *)sender,"Unable to load module %s",cargv[0]);
 252       return CMD_ERROR;
 253
 254     case 1:
 255       controlreply((nick *)sender,"Module %s already loaded, or name not valid",cargv[0]);
 256       return CMD_ERROR;
 257
 258     case 0:
 259       controlreply((nick *)sender,"Module %s loaded.",cargv[0]);
 260       return CMD_OK;
 261
 262     default:
 263       controlreply((nick *)sender,"An unknown error occured.");
 264       return CMD_ERROR;
 265   }
 266 }
 267
 268 int controlrmmod(void *sender, int cargc, char **cargv) {
 269   if (cargc<1)
 270     return CMD_USAGE;
 271
 272   switch(rmmod(cargv[0])) {
 273     case 1:
 274       controlreply((nick *)sender,"Module %s is not loaded.",cargv[0]);
 275       return CMD_ERROR;
 276
 277     case 0:
 278       controlreply((nick *)sender,"Module %s unloaded.",cargv[0]);
 279       return CMD_OK;
 280
 281     default:
 282       controlreply((nick *)sender,"An unknown error occured.");
 283       return CMD_ERROR;
 284   }
 285 }
 286
 287 int controllsmod(void *sender, int cargc, char **cargv) {
 288   int i=0;
 289   char *ptr;
 290
 291   if (cargc < 1) { /* list all loaded modules */
 292     ptr = lsmod(i);
 293     controlreply((nick *)sender,"Module");
 294     while (ptr != NULL) {
 295       const char *ver = lsmodver(i);
 296       controlreply((nick *)sender,"%s%s%s%s", ptr, ver?" (":"", ver?ver:"", ver?")":"");
 297       ptr = lsmod(++i);
 298     }
 299   } else {
 300     ptr = lsmod(getindex(cargv[0]));
 301     controlreply((nick *)sender,"Module \"%s\" %s", cargv[0], (ptr ? "is loaded." : "is NOT loaded."));
 302   }
 303   return CMD_OK;
 304 }
 305
 306 int controlreload(void *sender, int cargc, char **cargv) {
 307   if (cargc<1)
 308     return CMD_USAGE;
 309
 310   controlreply((nick *)sender,"Imma gonna try and reload %s",cargv[0]);
 311
 312   safereload(cargv[0]);
 313
 314   return CMD_OK;
 315 }
 316
 317 int relink(void *sender, int cargc, char **cargv) {
 318   if (cargc<1) {
 319     controlreply((nick *)sender,"You must give a reason.");
 320     return CMD_USAGE;
 321   }
 322
 323   irc_send("%s SQ %s 0 :%s",mynumeric->content,myserver->content,cargv[0]);
 324   irc_disconnected();
 325
 326   return CMD_OK;
 327 }
 328
 329 int die(void *sender, int cargc, char **cargv) {
 330   if (cargc<1) {
 331     controlreply((nick *)sender,"You must give a reason.");
 332     return CMD_USAGE;
 333   }
 334
 335   controlwall(NO_OPER,NL_OPERATIONS,"DIE from %s: %s",((nick *)sender)->nick, cargv[0]);
 336
 337   newserv_shutdown_pending=1;
 338
 339   return CMD_OK;
 340 }
 341
 342 int controlchannel(void *sender, int cargc, char **cargv) {
 343   channel *cp;
 344   nick *np;
 345   chanban *cbp;
 346   char buf[BUFSIZE];
 347   char buf2[12];
 348   int i,j, ops=0, voice=0;
 349   char timebuf[30];
 350
 351   if (cargc<1)
 352     return CMD_USAGE;
 353
 354   if ((cp=findchannel(cargv[0]))==NULL) {
 355     controlreply((nick *)sender,"Couldn't find channel: %s",cargv[0]);
 356     return CMD_ERROR;
 357   }
 358
 359   if (IsLimit(cp)) {
 360     sprintf(buf2,"%d",cp->limit);
 361   }
 362
 363   controlreply((nick *)sender,"Channel : %s",cp->index->name->content);
 364   strftime(timebuf, 30, "%d/%m/%y %H:%M", localtime(&(cp->timestamp)));
 365   controlreply((nick *)sender,"C-time  : %ld [%s]",cp->timestamp,timebuf);
 366   if (cp->topic) {
 367     controlreply((nick *)sender,"Topic   : %s",cp->topic->content);
 368     strftime(timebuf, 30, "%d/%m/%y %H:%M", localtime(&(cp->topictime)));
 369     controlreply((nick *)sender,"T-time  : %ld [%s]",cp->topictime,timebuf);
 370   } else {
 371     controlreply((nick *)sender,"Topic   : (none)");
 372   }
 373   controlreply((nick *)sender,"Mode(s) : %s %s%s%s",printflags(cp->flags,cmodeflags),IsLimit(cp)?buf2:"",
 374     IsLimit(cp)?" ":"",IsKey(cp)?cp->key->content:"");
 375   controlreply((nick *)sender,"Users   : %d (hash size %d, utilisation %.1f%%); %d unique hosts",
 376     cp->users->totalusers,cp->users->hashsize,((float)(100*cp->users->totalusers)/cp->users->hashsize),
 377     countuniquehosts(cp));
 378   i=0;
 379   memset(buf,' ',90);
 380   buf[72]='\0';
 381   for (j=0;j<=cp->users->hashsize;j++) {
 382     if (i==4 || j==cp->users->hashsize) {
 383       if(i>0) {
 384         controlreply((nick *)sender,"Users   : %s",buf);
 385       }
 386       i=0;
 387       memset(buf,' ',72);
 388       if (j==cp->users->hashsize)
 389         break;
 390     }
 391     if (cp->users->content[j]!=nouser) {
 392       if (cp->users->content[j]&CUMODE_OP)
 393         ops++;
 394       else if (cp->users->content[j]&CUMODE_VOICE)
 395         voice++;
 396       np=getnickbynumeric(cp->users->content[j]);
 397       sprintf(&buf[i*18],"%c%c%-15s ",cp->users->content[j]&CUMODE_VOICE?'+':' ',
 398         cp->users->content[j]&CUMODE_OP?'@':' ', np?np->nick:"!BUG-NONICK!");
 399       i++;
 400       if (i<4)
 401         buf[i*18]=' ';
 402     }
 403   }
 404   controlreply((nick *)sender, "Users   : Opped: %d, Voiced: %d", ops,voice);
 405   for (cbp=cp->bans;cbp;cbp=cbp->next) {
 406     controlreply((nick *)sender,"Ban     : %s",bantostringdebug(cbp));
 407   }
 408   return CMD_OK;
 409 }
 410
 411 int controlshowcommands(void *sender, int cargc, char **cargv) {
 412   nick *np=(nick *)sender;
 413   Command *cmdlist[100];
 414   int i,n;
 415
 416   n=getcommandlist(controlcmds,cmdlist,100);
 417
 418   controlreply(np,"The following commands are registered at present:");
 419
 420   for(i=0;i<n;i++) {
 421     controlreply(np,"%s",cmdlist[i]->command->content);
 422   }
 423
 424   controlreply(np,"End of list.");
 425   return CMD_OK;
 426 }
 427
 428 void handlemessages(nick *target, int messagetype, void **args) {
 429   Command *cmd;
 430   char *cargv[50];
 431   int cargc;
 432   nick *sender;
 433
 434   switch(messagetype) {
 435     case LU_PRIVMSG:
 436     case LU_SECUREMSG:
 437       /* If it's a message, first arg is nick and second is message */
 438       sender=(nick *)args[0];
 439
 440       Error("control",ERR_INFO,"From: %s!%s@%s: %s",sender->nick,sender->ident,sender->host->name->content, (char *)args[1]);
 441
 442       /* Split the line into params */
 443       cargc=splitline((char *)args[1],cargv,50,0);
 444
 445       if (!cargc) {
 446         /* Blank line */
 447         return;
 448       }
 449
 450       cmd=findcommandintree(controlcmds,cargv[0],1);
 451       if (cmd==NULL) {
 452         controlreply(sender,"Unknown command.");
 453         return;
 454       }
 455
 456       if (cmd->level>0 && !IsOper(sender)) {
 457         controlreply(sender,"You need to be opered to use this command.");
 458         return;
 459       }
 460
 461       /* If we were doing "authed user tracking" here we'd put a check in for authlevel */
 462
 463       /* Check the maxargs */
 464       if (cmd->maxparams<(cargc-1)) {
 465         /* We need to do some rejoining */
 466         rejoinline(cargv[cmd->maxparams],cargc-(cmd->maxparams));
 467         cargc=(cmd->maxparams)+1;
 468       }
 469
 470       if((cmd->handler)((void *)sender,cargc-1,&(cargv[1])) == CMD_USAGE)
 471         controlhelp(sender, cmd);
 472       break;
 473
 474     case LU_KILLED:
 475       /* someone killed me?  Bastards */
 476       scheduleoneshot(time(NULL)+1,&controlconnect,NULL);
 477       mynick=NULL;
 478       break;
 479
 480     default:
 481       break;
 482   }
 483 }
 484
 485 void controlmessage(nick *target, char *message, ... ) {
 486   char buf[512];
 487   va_list va;
 488
 489   if (mynick==NULL) {
 490     return;
 491   }
 492
 493   va_start(va,message);
 494   vsnprintf(buf,512,message,va);
 495   va_end(va);
 496
 497   sendmessagetouser(mynick,target,"%s",buf);
 498 }
 499
 500 void controlchanmsg(channel *cp, char *message, ...) {
 501   char buf[512];
 502   va_list va;
 503
 504   if (mynick==NULL) {
 505     return;
 506   }
 507
 508   va_start(va,message);
 509   vsnprintf(buf,512,message,va);
 510   va_end(va);
 511
 512   sendmessagetochannel(mynick,cp,"%s",buf);
 513 }
 514
 515 void controlnotice(nick *target, char *message, ... ) {
 516   char buf[512];
 517   va_list va;
 518
 519   if (mynick==NULL) {
 520     return;
 521   }
 522
 523   va_start(va,message);
 524   vsnprintf(buf,512,message,va);
 525   va_end(va);
 526
 527   sendnoticetouser(mynick,target,"%s",buf);
 528 }
 529
 530 void controlspecialrmmod(void *arg) {
 531   struct specialsched *a = (struct specialsched *)arg;
 532
 533   a->schedule = NULL;
 534   sstring *froo = a->modulename;
 535
 536   rmmod(froo->content);
 537   freesstring(froo);
 538 }
 539
 540 void controlspecialreloadmod(void *arg) {
 541   struct specialsched *a = (struct specialsched *)arg;
 542
 543   a->schedule = NULL;
 544   sstring *froo = a->modulename;
 545
 546   safereload(froo->content);
 547   freesstring(froo);
 548 }
 549
 550 void controlhelp(nick *np, Command *cmd) {
 551   char *cp = cmd->help, *sp = cp;
 552   if(!cp || !*cp) {
 553     controlreply(np, "Sorry, no help for this command.");
 554   } else {
 555     int finished = 0;
 556     for(;;cp++) {
 557       if(*cp == '\0' || *cp == '\n') {
 558         if(*cp == '\0') {
 559           finished = 1;
 560         } else {
 561           *cp = '\0';
 562         }
 563
 564         if(sp != cp)
 565           controlreply(np, "%s", sp);
 566
 567         if(finished)
 568           break;
 569
 570         *cp = '\n';
 571
 572         sp = cp + 1;
 573       }
 574     }
 575   }
 576 }
 577
 578 int controlhelpcmd(void *sender, int cargc, char **cargv) {
 579   Command *cmd;
 580   nick *np = (nick *)sender;
 581
 582   if (cargc<1)
 583     return CMD_USAGE;
 584
 585   cmd=findcommandintree(controlcmds,cargv[0],1);
 586   if (cmd==NULL) {
 587     controlreply(np,"Unknown command.");
 588     return CMD_ERROR;
 589   }
 590
 591   controlhelp(np, cmd);
 592   return CMD_OK;
 593 }
 594
 595 void controlnoticeopers(flag_t permissionlevel, flag_t noticelevel, char *format, ...) {
 596   int i;
 597   nick *np;
 598   char broadcast[512];
 599   va_list va;
 600
 601   va_start(va, format);
 602   vsnprintf(broadcast, sizeof(broadcast), format, va);
 603   va_end(va);
 604
 605   for(i=0;i<NICKHASHSIZE;i++)
 606     for(np=nicktable[i];np;np=np->next)
 607       if (IsOper(np))
 608         controlnotice(np, "%s", broadcast);
 609 }
 610
 611 void controlnswall(int noticelevel, char *format, ...) {
 612   char broadcast[512];
 613   va_list va;
 614
 615   va_start(va, format);
 616   vsnprintf(broadcast, sizeof(broadcast), format, va);
 617   va_end(va);
 618
 619   controlwall(NO_OPER, noticelevel, "%s", broadcast);
 620 }
 621