request/request_fasttrack.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include "../core/schedule.h"
   4 #include "../irc/irc.h"
   5 #include "../lib/irc_string.h"
   6 #include "request_fasttrack.h"
   7
   8 typedef struct rq_fasttrack {
   9   unsigned long userid;
  10
  11   unsigned int targets;
  12   time_t refill_time;
  13
  14   struct rq_fasttrack *next;
  15 } rq_fasttrack;
  16
  17 static rq_fasttrack *ftlist;
  18
  19 /* our fast-track extension */
  20 int rqnext;
  21
  22 static void rq_cleanup_fasttrack(void *arg);
  23 static void rqhook_account(int hook, void *arg);
  24
  25 int rq_initfasttrack(void) {
  26   rqnext = registernickext("request_fasttrack");
  27   if(rqnext < 0)
  28     return 0;
  29
  30   registerhook(HOOK_NICK_NEWNICK, &rqhook_account);
  31   registerhook(HOOK_NICK_ACCOUNT, &rqhook_account);
  32
  33   schedulerecurring(time(NULL)+1, 0, 3600, rq_cleanup_fasttrack, NULL);
  34
  35   return 1;
  36 }
  37
  38 void rq_finifasttrack(void) {
  39   rq_fasttrack *ft;
  40
  41   deregisterhook(HOOK_NICK_NEWNICK, &rqhook_account);
  42   deregisterhook(HOOK_NICK_ACCOUNT, &rqhook_account);
  43
  44   for(ft=ftlist;ft;ft=ft->next)
  45     free(ft);
  46
  47   releasenickext(rqnext);
  48 }
  49
  50 static void rqhook_account(int hook, void *arg) {
  51   nick *np = (nick *)arg;
  52   rq_fasttrack *ft;
  53
  54   /* Auth might be null for the newnick hook. */
  55   if(!np->auth)
  56     return;
  57
  58   /* Try to find an existing fasttrack record for this user. */
  59   for(ft=ftlist;ft;ft=ft->next) {
  60     if(np->auth->userid==ft->userid) {
  61       np->exts[rqnext] = ft;
  62       break;
  63     }
  64   }
  65 }
  66
  67 static void rq_cleanup_fasttrack(void *arg) {
  68
  69 }
  70
  71 static rq_fasttrack *rq_getfasttrack(nick *np) {
  72   rq_fasttrack *ft;
  73
  74   /* Use an existing fast-track record if the nick has one. */
  75   if(np->exts[rqnext])
  76     return np->exts[rqnext];
  77
  78   if(!np->auth)
  79     return NULL;
  80
  81   ft = malloc(sizeof(rq_fasttrack));
  82
  83   if(!ft)
  84     return NULL;
  85
  86   ft->userid = np->auth->userid;
  87   ft->targets = 0;
  88   ft->refill_time = 0;
  89
  90   np->exts[rqnext] = ft;
  91
  92   return ft;
  93 }
  94
  95 int rq_tryfasttrack(nick *np) {
  96   rq_fasttrack *ft = rq_getfasttrack(np);
  97
  98   /* Don't fast-track if we can't find a fast-track record. */
  99   if(!ft)
 100     return 0;
 101
 102   /* Refill targets if necessary. */
 103   if(getnettime() > ft->refill_time) {
 104     ft->targets = RQ_FASTTRACK_TARGETS;
 105     ft->refill_time = getnettime() + RQ_FASTTRACK_TIMEOUT;
 106   }
 107
 108   /* Check if we have a free target. */
 109   if(ft->targets==0)
 110     return 0;
 111
 112   ft->targets--;
 113   return 1;
 114 }
 115