patricianick/patricianick.c

   1 /* patricianick.c */
   2
   3 #include "patricianick.h"
   4 #include "../irc/irc_config.h"
   5 #include "../lib/irc_string.h"
   6 #include "../core/error.h"
   7 #include "../core/nsmalloc.h"
   8 #include "../control/control.h"
   9 #include "../core/schedule.h"
  10
  11 #include <stdio.h>
  12 #include <string.h>
  13
  14 #define ALLOCUNIT       100
  15
  16 patricianick_t *freepatricianicks;
  17 int pnode_ext;
  18 int pnick_ext;
  19
  20 int pn_cmd_nodeuserlist(void *source, int cargc, char **cargv);
  21
  22 void _init() {
  23   nick *np, *nnp;
  24   int i;
  25
  26   pnode_ext = registernodeext("patricianick");
  27   if (pnode_ext == -1) {
  28     Error("patricianick", ERR_FATAL, "Could not register a required node extension");
  29     return;
  30   }
  31
  32   pnick_ext = registernickext("patricianick");
  33   if (pnick_ext == -1) {
  34     Error("patricianick", ERR_FATAL, "Could not register a required nick extension");
  35     return;
  36   }
  37
  38   for (i=0;i<NICKHASHSIZE;i++)
  39     for (np=nicktable[i];np;np=nnp) {
  40       nnp=np->next;
  41       addnicktonode(np->ipnode, np);
  42     }
  43
  44   registerhook(HOOK_NICK_NEWNICK, &pn_hook_newuser);
  45   registerhook(HOOK_NICK_LOSTNICK, &pn_hook_lostuser);
  46
  47   registercontrolhelpcmd("nodeuserlist", NO_OPER, 1, &pn_cmd_nodeuserlist, "Usage: nodeuserlist <ipv4|ipv6|cidr4|cidr6>\nLists all users on a given IP address or CIDR range.");
  48 }
  49
  50 void _fini() {
  51   nsfreeall(POOL_PATRICIANICK);
  52
  53   if (pnode_ext != -1) {
  54     releasenodeext(pnode_ext);
  55   }
  56
  57   if (pnick_ext != -1) {
  58     releasenickext(pnick_ext);
  59   }
  60
  61   deregisterhook(HOOK_NICK_NEWNICK, &pn_hook_newuser);
  62   deregisterhook(HOOK_NICK_LOSTNICK, &pn_hook_lostuser);
  63
  64   deregistercontrolcmd("nodeuserlist", &pn_cmd_nodeuserlist);
  65 }
  66
  67 patricianick_t *getpatricianick() {
  68   int i;
  69   patricianick_t *pnp;
  70
  71   if (freepatricianicks==NULL) {
  72     freepatricianicks=(patricianick_t *)nsmalloc(POOL_PATRICIANICK, ALLOCUNIT*sizeof(patricianick_t));
  73     for(i=0;i<ALLOCUNIT-1;i++) {
  74       freepatricianicks[i].identhash[0]=(nick *)&(freepatricianicks[i+1]);
  75     }
  76     freepatricianicks[ALLOCUNIT-1].identhash[0]=NULL;
  77   }
  78
  79   pnp=freepatricianicks;
  80   freepatricianicks=(patricianick_t *)pnp->identhash[0];
  81
  82   memset(pnp, 0, sizeof(patricianick_t));
  83   return pnp;
  84 }
  85
  86 void addnicktonode(patricia_node_t *node, nick *np) {
  87   unsigned long hash;
  88
  89   patricia_ref_prefix(node->prefix);
  90
  91   if (!(node->exts[pnode_ext])) {
  92     node->exts[pnode_ext] = getpatricianick();
  93   }
  94
  95   hash = pn_getidenthash(np->ident);
  96   np->exts[pnick_ext] = ((patricianick_t *)node->exts[pnode_ext])->identhash[hash];
  97   ((patricianick_t *)node->exts[pnode_ext])->identhash[hash] = np;
  98 }
  99
 100 void deletenickfromnode(patricia_node_t *node, nick *np) {
 101   nick **tnp;
 102   int found, i;
 103
 104   found = 0;
 105
 106   for (tnp = &(((patricianick_t *)node->exts[pnode_ext])->identhash[pn_getidenthash(np->ident)]); *tnp; tnp = (nick **)(&((*tnp)->exts[pnick_ext]))) {
 107     if (*tnp == np) {
 108       *tnp = np->exts[pnick_ext];
 109       found = 1;
 110       break;
 111     }
 112   }
 113
 114   if (!found) {
 115     Error("patricianick", ERR_ERROR, "Could not remove %s!%s from %s", np->nick, np->ident, IPtostr(node->prefix->sin));
 116     return;
 117   }
 118
 119   for (i = 0; i < PATRICIANICK_HASHSIZE; i++) {
 120     if (((patricianick_t *)node->exts[pnode_ext])->identhash[i]) {
 121       return;
 122     }
 123   }
 124
 125   freepatricianick(node->exts[pnode_ext]);
 126   node->exts[pnode_ext]= NULL;
 127 }
 128
 129 void freepatricianick(patricianick_t *pnp) {
 130   pnp->identhash[0]=(nick *)freepatricianicks;
 131   freepatricianicks=pnp;
 132 }
 133
 134 void pn_hook_newuser(int hook, void *arg) {
 135   nick *np = (nick *)arg;
 136
 137   addnicktonode(np->ipnode, np);
 138 }
 139
 140 void pn_hook_lostuser(int hook, void *arg) {
 141   nick *np = (nick *)arg;
 142
 143   deletenickfromnode(np->ipnode, np);
 144 }
 145
 146 int pn_cmd_nodeuserlist(void *source, int cargc, char **cargv) {
 147   nick *np=(nick *)source;
 148   struct irc_in_addr sin;
 149   unsigned char bits;
 150   unsigned int count, i;
 151   patricia_node_t *head, *node;
 152   patricianick_t *pnp;
 153   nick *npp;
 154
 155   if (cargc < 1) {
 156     return CMD_USAGE;
 157   }
 158
 159   if (ipmask_parse(cargv[0], &sin, &bits) == 0) {
 160     controlreply(np, "Invalid mask.");
 161     return CMD_ERROR;
 162   }
 163
 164   head = refnode(iptree, &sin, bits);
 165   count = 0;
 166
 167   PATRICIA_WALK(head, node)
 168   {
 169     pnp = node->exts[pnode_ext];
 170     if (pnp) {
 171       if (count < PATRICIANICK_MAXRESULTS) {
 172         for (i = 0; i < PATRICIANICK_HASHSIZE; i++) {
 173           for (npp = pnp->identhash[i]; npp; npp=npp->exts[pnick_ext]) {
 174             controlreply(np, "%s!%s@%s%s%s (%s)", npp->nick, npp->ident, npp->host->name->content, IsAccount(npp) ? "/" : "", npp->authname, IPtostr(node->prefix->sin));
 175           }
 176         }
 177
 178         count += node->usercount;
 179
 180         if (count >= PATRICIANICK_MAXRESULTS) {
 181           controlreply(np, "Too many results, output truncated");
 182         }
 183       } else {
 184         count += node->usercount;
 185       }
 186     }
 187   }
 188   PATRICIA_WALK_END;
 189   derefnode(iptree, head);
 190
 191   controlreply(np, "Total users on %s: %d", cargv[0], count);
 192   return CMD_OK;
 193 }