chanserv/chanserv.c

   1 /*
   2  * chanserv.c:
   3  *  The main chanserv core file.
   4  */
   5
   6 #include "chanserv.h"
   7 #include "authlib.h"
   8 #include "../core/hooks.h"
   9 #include "../core/schedule.h"
  10 #include "../lib/version.h"
  11
  12 MODULE_VERSION(QVERSION);
  13
  14 int chanservnext;
  15
  16 int chanserv_init_status;
  17
  18 sstring **chantypes;
  19 sstring *cs_quitreason;
  20
  21 void chanservfreestuff();
  22 void chanservfinishinit(int hooknum, void *arg);
  23 static void cs_hourlyfunc(void *arg);
  24
  25 DBModuleIdentifier q9dbid;
  26
  27 void chanservdumpstuff(void *arg) {
  28   dumplastjoindata("lastjoin.dump");
  29 }
  30
  31 void _init() {
  32   /* Register the nick extension - the others are registered in the db module */
  33   chanservnext=registernickext("nickserv");
  34
  35   chanservcryptoinit();
  36   csa_initregex();
  37
  38   q9dbid = dbgetid();
  39
  40   if (chanservext!=-1 && chanservnext!=-1 && chanservaext!=-1) {
  41     /* Set up the chantypes */
  42     chantypes=(sstring **)malloc(CHANTYPES*sizeof(sstring *));
  43     chantypes[0]=getsstring("(unspecified)",20);
  44     chantypes[1]=getsstring("clan",20);
  45     chantypes[2]=getsstring("league",20);
  46     chantypes[3]=getsstring("private",20);
  47     chantypes[4]=getsstring("special",20);
  48     chantypes[5]=getsstring("gamesite",20);
  49     chantypes[6]=getsstring("game",20);
  50     chantypes[7]=getsstring("upgrade",20);
  51
  52     /* And the log system */
  53     cs_initlog();
  54
  55     /* Set up the command handler, and built in commands */
  56     chanservcommandinit();
  57     chanservaddcommand("showcommands", 0, 1, cs_doshowcommands, "Lists available commands.","Usage: SHOWCOMMANDS [<mask>]\nPrints a list of commands currently available to you, where:\nmask - Mask of commands to list (* or ? are wildcards).  If no mask is specified,\n       all available commands are displayed.");
  58     chanservaddcommand("quit", QCMD_DEV, 1, cs_doquit, "Makes the bot QUIT and \"reconnect\".","");
  59     chanservaddcommand("setquitreason", QCMD_DEV, 1, cs_dosetquitreason, "Sets the reason to be sent when quitting due to an unload.","");
  60     chanservaddcommand("rename", QCMD_DEV, 1, cs_dorename, "Changes the bot's name.","");
  61     chanservaddcommand("rehash", QCMD_DEV, 0, cs_dorehash, "Reloads all text from database.","");
  62     /* Make "HELP" take 2 arguments so things like "HELP chanflags #channel" work.  Any junk after the command will go into arg 2 and be ignored. */
  63     chanservaddcommand("help", 0, 2, cs_dohelp, "Displays help on a specific command.","Usage: HELP <command>\nShows help for a command, where:\ncommand - the command to show help for.\nFor a list of available commands, see SHOWCOMMANDS.\n");
  64     chanservaddcommand("version", 0, 1, cs_doversion, "Show Version.","Usage: VERSION\nShows the version number of the running bot.");
  65
  66     chanservaddctcpcommand("ping",cs_doctcpping);
  67     chanservaddctcpcommand("version",cs_doctcpversion);
  68     chanservaddctcpcommand("gender",cs_doctcpgender);
  69
  70     registerhook(HOOK_CHANSERV_DBLOADED, chanservfinishinit);
  71
  72     /* Now that the database is in a separate module it might be loaded already. */
  73     if (chanservdb_ready)
  74       chanservfinishinit(HOOK_CHANSERV_DBLOADED, NULL);
  75   }
  76 }
  77
  78 void chanservfinishinit(int hooknum, void *arg) {
  79   Error("chanserv",ERR_INFO,"Database loaded, finishing initialisation.");
  80
  81   deregisterhook(HOOK_CHANSERV_DBLOADED, chanservfinishinit);
  82
  83   readlastjoindata("lastjoin.dump");
  84
  85   /* Schedule the dumps */
  86   schedulerecurring(time(NULL)+DUMPINTERVAL,0,DUMPINTERVAL,chanservdumpstuff,NULL);
  87   schedulerecurring(time(NULL)+5,0,5,csdb_flushchannelcounters,NULL);
  88
  89   chanserv_init_status = CS_INIT_NOUSER;
  90
  91   /* Register the user */
  92   scheduleoneshot(time(NULL)+1,&chanservreguser,NULL);
  93 }
  94
  95 void chanserv_finalinit() {
  96   int i;
  97   nick *np, *nnp;
  98
  99   /* Scan for users */
 100   for (i=0;i<NICKHASHSIZE;i++)
 101     for (np=nicktable[i];np;np=nnp) {
 102       nnp=np->next;
 103       cs_checknick(np);
 104     }
 105
 106   /* Register core hooks */
 107   registerhook(HOOK_NICK_NEWNICK, cs_handlenick);
 108   registerhook(HOOK_NICK_ACCOUNT, cs_handlenick);
 109   registerhook(HOOK_NICK_LOSTNICK, cs_handlelostnick);
 110   registerhook(HOOK_NICK_SETHOST, cs_handlesethost);
 111   registerhook(HOOK_CHANNEL_NEWCHANNEL, cs_handlenewchannel);
 112   registerhook(HOOK_CHANNEL_LOSTCHANNEL, cs_handlelostchannel);
 113   registerhook(HOOK_CHANNEL_JOIN, cs_handlejoin);
 114   registerhook(HOOK_CHANNEL_CREATE, cs_handlejoin);
 115   registerhook(HOOK_CHANNEL_MODECHANGE, cs_handlemodechange);
 116   registerhook(HOOK_CHANNEL_BURST, cs_handleburst);
 117   registerhook(HOOK_CHANNEL_TOPIC, cs_handletopicchange);
 118   registerhook(HOOK_CHANNEL_LOSTNICK, cs_handlechanlostuser);
 119
 120   chanserv_init_status = CS_INIT_READY;
 121   triggerhook(HOOK_CHANSERV_RUNNING, NULL);
 122
 123   /* run every 60 minutes, first one 5 minutes after start */
 124   schedulerecurring(time(NULL)+60*5,0,3600,cs_hourlyfunc,NULL);
 125
 126   Error("chanserv",ERR_INFO,"Ready to roll.");
 127 }
 128
 129 void _fini() {
 130   csdb_flushchannelcounters(NULL);
 131
 132   dbfreeid(q9dbid);
 133
 134   deleteallschedules(cs_hourlyfunc);
 135   deleteallschedules(cs_timerfunc);
 136   deleteallschedules(chanservreguser);
 137   deleteallschedules(chanservdumpstuff);
 138   deleteallschedules(chanservdgline);
 139   deleteallschedules(csdb_flushchannelcounters);
 140
 141   if (chanservext>-1 && chanservnext>-1 && chanservaext>-1) {
 142     int i;
 143     for (i=0;i<CHANTYPES;i++)
 144       freesstring(chantypes[i]);
 145
 146     free(chantypes);
 147   }
 148
 149   /* Free everything */
 150   if (chanservnext!=-1) {
 151     releasenickext(chanservnext);
 152   }
 153
 154   if (chanservnick)
 155     deregisterlocaluser(chanservnick, cs_quitreason?cs_quitreason->content:"Leaving");
 156
 157   freesstring(cs_quitreason);
 158
 159   deregisterhook(HOOK_NICK_NEWNICK, cs_handlenick);
 160   deregisterhook(HOOK_NICK_ACCOUNT, cs_handlenick);
 161   deregisterhook(HOOK_NICK_LOSTNICK, cs_handlelostnick);
 162   deregisterhook(HOOK_NICK_SETHOST, cs_handlesethost);
 163   deregisterhook(HOOK_CHANNEL_NEWCHANNEL, cs_handlenewchannel);
 164   deregisterhook(HOOK_CHANNEL_LOSTCHANNEL, cs_handlelostchannel);
 165   deregisterhook(HOOK_CHANNEL_JOIN, cs_handlejoin);
 166   deregisterhook(HOOK_CHANNEL_CREATE, cs_handlejoin);
 167   deregisterhook(HOOK_CHANNEL_MODECHANGE, cs_handlemodechange);
 168   deregisterhook(HOOK_CHANNEL_BURST, cs_handleburst);
 169   deregisterhook(HOOK_CHANNEL_TOPIC, cs_handletopicchange);
 170   deregisterhook(HOOK_CHANNEL_LOSTNICK, cs_handlechanlostuser);
 171   /*  deregisterhook(HOOK_CHANNEL_OPPED, cs_handleopchange);
 172   deregisterhook(HOOK_CHANNEL_DEOPPED, cs_handleopchange);
 173   deregisterhook(HOOK_CHANNEL_DEVOICED, cs_handleopchange);
 174   deregisterhook(HOOK_CHANNEL_BANSET, cs_handlenewban); */
 175   deregisterhook(HOOK_CHANSERV_DBLOADED, chanservfinishinit); /* for safety */
 176
 177   chanservremovecommand("showcommands", cs_doshowcommands);
 178   chanservremovecommand("quit", cs_doquit);
 179   chanservremovecommand("setquitreason", cs_dosetquitreason);
 180   chanservremovecommand("rename", cs_dorename);
 181   chanservremovecommand("rehash", cs_dorehash);
 182   chanservremovecommand("help", cs_dohelp);
 183   chanservremovecommand("version", cs_doversion);
 184   chanservremovectcpcommand("ping",cs_doctcpping);
 185   chanservremovectcpcommand("version",cs_doctcpversion);
 186   chanservremovectcpcommand("gender",cs_doctcpgender);
 187   chanservcommandclose();
 188
 189   csa_freeregex();
 190   chanservcryptofree();
 191
 192   cs_closelog();
 193 }
 194
 195 static void cs_hourlyfunc(void *arg) {
 196   int i, total = 0, touched = 0, toorecent = 0;
 197   chanindex *cip, *ncip;
 198   regchan *rcp;
 199   time_t t = time(NULL);
 200
 201   for (i=0;i<CHANNELHASHSIZE;i++) {
 202     for (cip=chantable[i];cip;cip=ncip) {
 203       ncip=cip->next;
 204       if (!(rcp=cip->exts[chanservext]))
 205         continue;
 206
 207       total++;
 208
 209       if(cip->channel && cs_ischannelactive(cip->channel, rcp)) {
 210         rcp->lastactive = t;
 211         if (rcp->lastcountersync < (t - COUNTERSYNCINTERVAL)) {
 212           csdb_updatechannelcounters(rcp);
 213           rcp->lastcountersync=t;
 214           touched++;
 215         } else {
 216           toorecent++;
 217         }
 218       }
 219
 220     }
 221   }
 222   cs_log(NULL,"hourly update active completed, %d seen, %d touched, %d too recent to update.",total,touched,toorecent);
 223 }