Author: Isomer <isomer@coders.net>
[ircu2.10.12-pk.git] / ircd / IPcheck.c
index 48fe0445d16947e021d6badf6f02f7949e931bb6..2e9dce223c4b187d4d30482b1efbb09c99fa5200 100644 (file)
  * $Id$
  *
  */
-
-/*----------------------------------------------------------------------------
- * Platform Includes
- *--------------------------------------------------------------------------*/
-#include <assert.h>
-#include <stdio.h>
-#include <string.h>
-
-
-/*----------------------------------------------------------------------------
- * Application Includes
- *--------------------------------------------------------------------------*/
 #include "IPcheck.h"
 #include "client.h"
 #include "ircd.h"
 #include "s_user.h"
 #include "send.h"
 
+#include <assert.h>
+#include <arpa/inet.h>
+#include <stdio.h>
+#include <string.h>
 
-/*----------------------------------------------------------------------------
- * Data Structures (should be moved to IPcheck.h)
- *--------------------------------------------------------------------------*/
-typedef struct IPTargetEntry {
+#if 1
+#warning Nick collisions are horribly broken in
+#warning this version, and its known to core on
+#warning a whim.  If your even concidering
+#warning running this on something resembling a
+#warning production network, dont bother, its
+#warning not worth your time.  To those of you
+#warning who grabbed the latest CVS version to
+#warning bug test it, thanks, but I recommend
+#warning you stick to previous versions for the
+#warning time being.
+#error --- Broken code ---
+#endif
+
+struct IPTargetEntry {
   int           count;
   unsigned char targets[MAXTARGETS];
-} iptarget_entry_t;
+};
 
-typedef struct IPRegistryEntry {
+struct IPRegistryEntry {
   struct IPRegistryEntry *next;
   struct IPTargetEntry   *target;
   unsigned int             addr;
   time_t                  last_connect;
   unsigned char            connected;
   unsigned char            attempts;
-} ip_reg_entry_t;
+};
 
 
 /*
@@ -73,9 +76,10 @@ typedef struct IPRegistryEntry {
 #define IP_REGISTRY_TABLE_SIZE 0x10000
 #define MASK_16                0xffff
 
-#define IPCHECK_CLONE_LIMIT 2
-#define IPCHECK_CLONE_PERIOD 20
-#define IPCHECK_CLONE_DELAY  1
+/* We allow 6 connections in 60 seconds */
+#define IPCHECK_CLONE_LIMIT 6
+#define IPCHECK_CLONE_PERIOD 60
+#define IPCHECK_CLONE_DELAY  600
 
 
 /*----------------------------------------------------------------------------
@@ -88,8 +92,8 @@ typedef struct IPRegistryEntry {
 /*----------------------------------------------------------------------------
  * Global Data (ugly!)
  *--------------------------------------------------------------------------*/
-static ip_reg_entry_t *hashTable[IP_REGISTRY_TABLE_SIZE];
-static ip_reg_entry_t *freeList = 0;
+static struct IPRegistryEntry *hashTable[IP_REGISTRY_TABLE_SIZE];
+static struct IPRegistryEntry *freeList = 0;
 
 
 /*----------------------------------------------------------------------------
@@ -98,7 +102,8 @@ static ip_reg_entry_t *freeList = 0;
  *                    to the IP argument?  Ugly.  This should probably be a
  *                    struct in_addr.  This is asking for trouble.  --ZS)
  *--------------------------------------------------------------------------*/
-static unsigned int ip_registry_hash(unsigned int ip) {
+static unsigned int ip_registry_hash(unsigned int ip)
+{
   return ((ip >> 16) ^ ip) & (IP_REGISTRY_TABLE_SIZE - 1);
 }
 
@@ -106,22 +111,24 @@ static unsigned int ip_registry_hash(unsigned int ip) {
 /*----------------------------------------------------------------------------
  * ip_registry_find:  Find a given IP registry entry and return it.
  *--------------------------------------------------------------------------*/
-static ip_reg_entry_t *ip_registry_find(unsigned int ip) {
-  ip_reg_entry_t *entry;
+static struct IPRegistryEntry *ip_registry_find(unsigned int ip) 
+{
+  struct IPRegistryEntry *entry = NULL;
 
   for (entry = hashTable[ip_registry_hash(ip)]; entry; entry = entry->next) {
     if (entry->addr == ip)
-      break;
+      return entry;
   }
 
-  return entry;
+  return NULL;
 }
 
 
 /*----------------------------------------------------------------------------
  * ip_registry_add:  Add an entry to the IP registry
  *--------------------------------------------------------------------------*/
-static void ip_registry_add(ip_reg_entry_t *entry) {
+static void ip_registry_add(struct IPRegistryEntry *entry) 
+{
   unsigned int bucket = ip_registry_hash(entry->addr);
 
   entry->next = hashTable[bucket];
@@ -132,13 +139,14 @@ static void ip_registry_add(ip_reg_entry_t *entry) {
 /*----------------------------------------------------------------------------
  * ip_registry_remove:  Remove an entry from the IP registry
  *--------------------------------------------------------------------------*/
-static void ip_registry_remove(ip_reg_entry_t *entry) {
+static void ip_registry_remove(struct IPRegistryEntry* entry) 
+{
   unsigned int bucket = ip_registry_hash(entry->addr);
 
   if (hashTable[bucket] == entry)
     hashTable[bucket] = entry->next;
   else {
-    ip_reg_entry_t *prev;
+    struct IPRegistryEntry *prev;
 
     for (prev = hashTable[bucket]; prev; prev = prev->next) {
       if (prev->next == entry) {
@@ -154,17 +162,18 @@ static void ip_registry_remove(ip_reg_entry_t *entry) {
  * ip_registry_new_entry():  Creates and initializes an IP Registry entry.
  *                           NOW ALSO ADDS IT TO THE LIST! --ZS
  *--------------------------------------------------------------------------*/
-static ip_reg_entry_t *ip_registry_new_entry(unsigned int addr, int attempt) {
-  ip_reg_entry_t *entry = freeList;
+static struct IPRegistryEntry *ip_registry_new_entry(unsigned int addr, int attempt)
+{
+  struct IPRegistryEntry* entry = freeList;
 
   if (entry)
     freeList = entry->next;
   else
-    entry = (ip_reg_entry_t *)MyMalloc(sizeof(ip_reg_entry_t));
+    entry = (struct IPRegistryEntry *)MyMalloc(sizeof(struct IPRegistryEntry));
 
   assert(0 != entry);
 
-  memset(entry, 0, sizeof(ip_reg_entry_t));
+  memset(entry, 0, sizeof(struct IPRegistryEntry));
   entry->last_connect = NOW;     /* Seconds since last connect attempt */
   entry->connected    = 1;       /* connected clients for this IP */
   entry->attempts     = attempt; /* Number attempts for this IP        */
@@ -182,7 +191,8 @@ static ip_reg_entry_t *ip_registry_new_entry(unsigned int addr, int attempt) {
  *                            the freelist every once in a while!  This is
  *                            potentially a way to DoS the server...  -ZS)
  *--------------------------------------------------------------------------*/
-static void ip_registry_delete_entry(ip_reg_entry_t *entry) {
+static void ip_registry_delete_entry(struct IPRegistryEntry *entry)
+{
   if (entry->target)
     MyFree(entry->target);
 
@@ -194,7 +204,8 @@ static void ip_registry_delete_entry(ip_reg_entry_t *entry) {
 /*----------------------------------------------------------------------------
  * ip_registry_update_free_targets:  
  *--------------------------------------------------------------------------*/
-static unsigned int ip_registry_update_free_targets(ip_reg_entry_t  *entry) {
+static unsigned int ip_registry_update_free_targets(struct IPRegistryEntry  *entry)
+{
   unsigned int free_targets = STARTTARGETS;
 
   if (entry->target) {
@@ -216,18 +227,21 @@ static unsigned int ip_registry_update_free_targets(ip_reg_entry_t  *entry) {
  *                            entry isn't expired, then also check the target
  *                            list to see if it needs to be expired.
  *--------------------------------------------------------------------------*/
-static void ip_registry_expire_entry(ip_reg_entry_t *entry) {
+static void ip_registry_expire_entry(struct IPRegistryEntry *entry)
+{
   /*
    * Don't touch this number, it has statistical significance
    * XXX - blah blah blah
    * ZS - Just -what- statistical significance does it -have-?
+   * Iso - Noone knows, we've just been told not to touch it.
    */
+  if (CONNECTED_SINCE(entry) > 120 && 0 != entry->target) {
+    MyFree(entry->target);
+    entry->target = 0;
+  }
   if (CONNECTED_SINCE(entry) > 600) {
     ip_registry_remove(entry);
     ip_registry_delete_entry(entry);
-  } else if (CONNECTED_SINCE(entry) > 120 && 0 != entry->target) {
-    MyFree(entry->target);
-    entry->target = 0;
   }
 }
 
@@ -235,11 +249,17 @@ static void ip_registry_expire_entry(ip_reg_entry_t *entry) {
 /*----------------------------------------------------------------------------
  * ip_registry_expire:  Expire all of the needed entries in the hash table
  *--------------------------------------------------------------------------*/
-static void ip_registry_expire(void) {
-  ip_reg_entry_t *entry;
-  ip_reg_entry_t *entry_next;
+void ip_registry_expire(void)
+{
+  struct IPRegistryEntry *entry;
+  struct IPRegistryEntry *entry_next;
+  static time_t   next_expire = 0;
   int i;
 
+  /* Only do this if we're ready to */
+  if (next_expire >= CurrentTime)
+    return;
+
   for (i = 0; i < IP_REGISTRY_TABLE_SIZE; ++i) {
     for (entry = hashTable[i]; entry; entry = entry_next) {
       entry_next = entry->next;
@@ -247,11 +267,12 @@ static void ip_registry_expire(void) {
         ip_registry_expire_entry(entry);
     }
   }
+
+  next_expire = CurrentTime + 60;
 }
 
 
 /*----------------------------------------------------------------------------
- * IPcheck_local_connect
  *
  * Event:
  *   A new connection was accept()-ed with IP number `cptr->ip.s_addr'.
@@ -279,100 +300,110 @@ static void ip_registry_expire(void) {
  *--------------------------------------------------------------------------*/
 int ip_registry_check_local(unsigned int addr, time_t *next_target_out)
 {
-  ip_reg_entry_t *entry        = ip_registry_find(addr);
+  struct IPRegistryEntry *entry        = ip_registry_find(addr);
   unsigned int free_targets = STARTTARGETS;
  
+  assert(0 != next_target_out);
+
+  /* If they've never connected before, let them on */
   if (0 == entry) {
+    Debug((DEBUG_DEBUG,"IPcheck: Local user allowed - unseen"));
     entry = ip_registry_new_entry(addr, 1);
     return 1;
   }
+  
+  /* Keep track of how many people have connected */
+  entry->connected++;
 
-  /* Do not allow more than 255 connects from a single IP, EVER. */
-  if (0 == ++entry->connected)
+  /* Do not allow more than 250 connects from a single IP, EVER. */
+  if (250 <= entry->connected) {
+    Debug((DEBUG_DEBUG,"IPcheck: Local user disallowed - Too many connections"));
+    entry->connected--;
     return 0;
+  }
 
-  /* If our threshhold has elapsed, reset the counter so we don't throttle */
-  if (CONNECTED_SINCE(entry) > IPCHECK_CLONE_PERIOD)
+  /* If our threshhold has elapsed, reset the counter so we don't throttle,
+   * IPCHECK_CLONE_LIMIT connections every IPCHECK_CLONE_PERIOD
+   */
+  if (CONNECTED_SINCE(entry) > IPCHECK_CLONE_PERIOD) {
     entry->attempts = 0;
-  else if (0 == ++entry->attempts)
+    entry->last_connect = NOW;
+  }
+
+  /* Count the number of recent attempts */ 
+  entry->attempts++;
+  
+  if (250 <= entry->attempts)
     --entry->attempts;  /* Disallow overflow */
 
-  entry->last_connect = NOW;
+
   free_targets = ip_registry_update_free_targets(entry);
 
-  if (entry->attempts < IPCHECK_CLONE_LIMIT && next_target_out)
+  /* Have they connected less than IPCHECK_CLONE_LIMIT times && next_target_out */
+  if (entry->attempts < IPCHECK_CLONE_LIMIT && next_target_out) {
       *next_target_out = CurrentTime - (TARGET_DELAY * free_targets - 1);
-  else if ((CurrentTime - me.since) > IPCHECK_CLONE_DELAY) {
-#ifdef NOTHROTTLE 
+      entry->last_connect = NOW;
+      Debug((DEBUG_DEBUG,"IPcheck: Local user allowed"));
+      return 1;
+  }
+  
+  /* If the server is younger than IPCHECK_CLONE_DELAY then the person
+   * is allowed on.
+   */
+  if ((CurrentTime - me.since) < IPCHECK_CLONE_DELAY) {
+    Debug((DEBUG_DEBUG,"IPcheck: Local user allowed during server startup"));
     return 1;
-#else
-    --entry->connected;
-    return 0;
-#endif        
   }
-
-  return 1;
+  
+  /* Otherwise they're throttled */
+  entry->connected--;
+  Debug((DEBUG_DEBUG,"IPcheck: Throttling local user"));
+  return 0;
 }
 
 
 /*----------------------------------------------------------------------------
- * IPcheck_remote_connect
- *
- * Event:
- *   A remote client connected to Undernet, with IP number `cptr->ip.s_addr'
- *   and hostname `hostname'.
+ * ip_registry_remote_connect
  *
- * Action:
- *   Update the IPcheck registry.
- *   Return 0 on failure, 1 on success.
+ * Does anything that needs to be done once we actually have a client
+ * structure to play with on a remote connection.
+ * returns:
+ *  1 - allowed to connect
+ *  0 - disallowed.
  *--------------------------------------------------------------------------*/
-int ip_registry_check_remote(struct Client* cptr, int is_burst) {
-  ip_reg_entry_t *entry = ip_registry_find(cptr->ip.s_addr);
+int ip_registry_remote_connect(struct Client *cptr)
+{
+  struct IPRegistryEntry *entry        = ip_registry_find(cptr->ip.s_addr);
+  assert(0 != cptr);
 
+  /* If they've never connected before, let them on */
+  if (0 == entry) {
+    entry = ip_registry_new_entry(cptr->ip.s_addr, 1);
+    SetIPChecked(cptr);
+    Debug((DEBUG_DEBUG,"IPcheck: First remote connection.  connected=%i",entry->connected));
+    return 1;
+  }
+  
+  /* Keep track of how many people have connected */
+  entry->connected++;
   SetIPChecked(cptr);
 
-  if (0 == entry)
-    entry = ip_registry_new_entry(cptr->ip.s_addr, (is_burst ? 0 : 1));
-  else {
-    /* NEVER more than 255 connections. */
-    if (0 == ++entry->connected)
-      return 0;
-
-    /* Make sure we don't bounce if our threshhold has expired */
-    if (CONNECTED_SINCE(entry) > IPCHECK_CLONE_PERIOD)
-      entry->attempts = 0;
-
-    /* If we're not part of a burst, go ahead and process the rest */
-    if (!is_burst) {
-      if (0 == ++entry->attempts)
-        --entry->attempts;  /* Overflows are bad, mmmkay? */
-      ip_registry_update_free_targets(entry);
-      entry->last_connect = NOW;
-    }
+  /* Do not allow more than 250 connections from one IP.
+   * This can happen by having 128 clients on one server, and 128 on another
+   * and then the servers joining after a netsplit
+   */ 
+  if (250 <= entry->connected) {
+    sendto_ops("IPcheck Ghost! [%s]",inet_ntoa(cptr->ip));
+    Debug((DEBUG_DEBUG,"IPcheck: Too many connected from IP: %i",entry->connected));
+    return 0;
   }
-
+  
+  Debug((DEBUG_DEBUG,"IPcheck: %i people connected",entry->connected));
+  
+  /* They are allowed to connect */
   return 1;
 }
 
-/*----------------------------------------------------------------------------
- * IPcheck_connect_fail
- *
- * Event:
- *   This local client failed to connect due to legal reasons.
- *
- * Action:
- *   Neutralize the effect of calling IPcheck_local_connect, in such
- *   a way that the client won't be penalized when trying to reconnect
- *   again.
- *--------------------------------------------------------------------------*/
-void ip_registry_connect_fail(unsigned int addr) {
-  ip_reg_entry_t *entry = ip_registry_find(addr);
-
-  if (entry)
-    --entry->attempts;
-}
-
-
 /*----------------------------------------------------------------------------
  * IPcheck_connect_succeeded
  *
@@ -381,25 +412,31 @@ void ip_registry_connect_fail(unsigned int addr) {
  *
  * Finish IPcheck registration of a successfully, locally connected client.
  *--------------------------------------------------------------------------*/
-void ip_registry_connect_succeeded(struct Client *cptr) {
-  const char     *tr           = "";
+void ip_registry_connect_succeeded(struct Client *cptr)
+{
   unsigned int free_targets     = STARTTARGETS;
-  ip_reg_entry_t *entry        = ip_registry_find(cptr->ip.s_addr);
+  struct IPRegistryEntry *entry;
 
-  if (!entry) {
-    Debug((DEBUG_ERROR, "Missing registry entry for: %s", cptr->sock_ip));
-    return;
-  }
+  assert(cptr);
+
+  entry = ip_registry_find(cptr->ip.s_addr);
+
+
+  assert(entry);
 
   if (entry->target) {
     memcpy(cptr->targets, entry->target->targets, MAXTARGETS);
     free_targets = entry->target->count;
-    tr = " tr";
   }
 
-  sendcmdto_one(&me, CMD_NOTICE, cptr, "%C :on %u ca %u(%u) ft %u(%u)%s",
+  sendcmdto_one(&me, CMD_NOTICE, cptr, "%C :connected %u attempts %u/%u free targets %u/%u%s"
+               " IPcheck: %s",
                cptr, entry->connected, entry->attempts, IPCHECK_CLONE_LIMIT,
-               free_targets, STARTTARGETS, tr);
+               free_targets, STARTTARGETS, 
+               ((entry->target) ? " [Inherited Targets]" : ""), 
+               ((CurrentTime - me.since) < IPCHECK_CLONE_DELAY) ? "Disabled" : "Enabled");
+               
+  SetIPChecked(cptr);
 }
 
 
@@ -407,202 +444,141 @@ void ip_registry_connect_succeeded(struct Client *cptr) {
  * IPcheck_disconnect
  *
  * Event:
- *   A local client disconnected or a remote client left Undernet.
+ *   A local client disconnected.
  *
  * Action:
  *   Update the IPcheck registry.
  *   Remove all expired IPregistry structures from the hash bucket
  *     that belongs to this clients IP number.
  *--------------------------------------------------------------------------*/
-void ip_registry_disconnect(struct Client *cptr) {
-  ip_reg_entry_t *entry = ip_registry_find(cptr->ip.s_addr);
+void ip_registry_local_disconnect(struct Client *cptr)
+{
+  struct IPRegistryEntry *entry;
+  unsigned int free_targets;
 
-  /* Entry is probably a server if this happens. */
-  if (0 == entry)
-    return;
+  assert(0 != cptr);
+
+  entry = ip_registry_find(cptr->ip.s_addr);
+
+  /* Servers might not be in IPcheck because we connected to them, not visa
+   * versa.
+   * We can't use IsServer() here, because it might be in the 'unregistered'
+   * state.
+   */
+  if (0 != cptr->serv && !entry) {
+        Debug((DEBUG_DEBUG,"IPcheck: Server ignored"));
+       return;
+  }
+  Debug((DEBUG_DEBUG,"IPcheck: Local Disconnect"));
+       
+  assert(IsIPChecked(cptr));
+  
+  assert(entry);
 
+  assert(entry->connected > 0);
+  
+  if (entry->connected > 0) {
+    entry->connected--;
+  }
 
   /*
    * If this was the last one, set `last_connect' to disconnect time
    * (used for expiration)   Note that we reset attempts here as well if our
    * threshhold hasn't been crossed.
    */
-  if (0 == --entry->connected) {
-    if (CONNECTED_SINCE(entry) > IPCHECK_CLONE_LIMIT * IPCHECK_CLONE_PERIOD)
-      entry->attempts = 0;
+  if (0 == entry->connected) {
     ip_registry_update_free_targets(entry);
     entry->last_connect = NOW;
   }
+  
+  assert(MyConnect(cptr));
 
-
-  if (MyConnect(cptr)) {
-    unsigned int free_targets;
-
-    if (0 == entry->target) {
-      entry->target = (iptarget_entry_t *)MyMalloc(sizeof(iptarget_entry_t));
-      assert(0 != entry->target);
-      entry->target->count = STARTTARGETS;
-    }
-    memcpy(entry->target->targets, cptr->targets, MAXTARGETS);
-
-    /*
-     * This calculation can be pretty unfair towards large multi-user hosts,
-     * but there is "nothing" we can do without also allowing spam bots to
-     * send more messages or by drastically increasing the ammount of memory
-     * used in the IPregistry.
-     *
-     * The problem is that when a client disconnects, leaving no free targets,
-     * then the next client from that IP number has to pay for it (getting no
-     * free targets).  But ALSO the next client, and the next client, and the
-     * next client etc - until another client disconnects that DOES leave free
-     * targets.  The reason for this is that if there are 10 SPAM bots, and
-     * they all disconnect at once, then they ALL should get no free targets
-     * when reconnecting.  We'd need to store an entry per client (instead of
-     * per IP number) to avoid this.  
-         */
-    if (cptr->nexttarget < CurrentTime)
-      free_targets = (CurrentTime - cptr->nexttarget) / TARGET_DELAY + 1;
-    else
-      free_targets = 0;
-
-    /* Add bonus, this is pretty fuzzy, but it will help in some cases. */
-    if ((CurrentTime - cptr->firsttime) > 600)
-      free_targets += (CurrentTime - cptr->firsttime - 600) / TARGET_DELAY;
-
-    /* Finally, store smallest value for Judgement Day */
-    if (free_targets < entry->target->count)
-      entry->target->count = free_targets;
+  if (0 == entry->target) {
+    entry->target = (struct IPTargetEntry *)MyMalloc(sizeof(struct IPTargetEntry));
+    assert(0 != entry->target);
+    entry->target->count = STARTTARGETS;
   }
-}
+  memcpy(entry->target->targets, cptr->targets, MAXTARGETS);
 
-/*----------------------------------------------------------------------------
- * IPcheck_nr
- *
- * Returns number of clients with the same IP number
- *--------------------------------------------------------------------------*/
-int ip_registry_count(unsigned int addr) {
-  ip_reg_entry_t *entry = ip_registry_find(addr);
-  return (entry) ? entry->connected : 0;
-}
+  /*
+   * This calculation can be pretty unfair towards large multi-user hosts,
+   * but there is "nothing" we can do without also allowing spam bots to
+   * send more messages or by drastically increasing the ammount of memory
+   * used in the IPregistry.
+   *
+   * The problem is that when a client disconnects, leaving no free targets,
+   * then the next client from that IP number has to pay for it (getting no
+   * free targets).  But ALSO the next client, and the next client, and the
+   * next client etc - until another client disconnects that DOES leave free
+   * targets.  The reason for this is that if there are 10 SPAM bots, and
+   * they all disconnect at once, then they ALL should get no free targets
+   * when reconnecting.  We'd need to store an entry per client (instead of
+   * per IP number) to avoid this.  
+   */
+  if (cptr->nexttarget < CurrentTime)
+    free_targets = (CurrentTime - cptr->nexttarget) / TARGET_DELAY + 1;
+  else
+    free_targets = 0;
 
+  /* Add bonus, if you've been connected for more than 10 minutes you
+   * get a free target every TARGET_DELAY seconds.
+   * this is pretty fuzzy, but it will help in some cases. 
+   */
+  if ((CurrentTime - cptr->firsttime) > 600)
+    free_targets += (CurrentTime - cptr->firsttime - 600) / TARGET_DELAY;
 
-/*----------------------------------------------------------------------------
- * IPcheck_local_connect
- *
- * Event:
- *   A new connection was accept()-ed with IP number `cptr->ip.s_addr'.
- *
- * Action:
- *   Update the IPcheck registry.
- *   Return:
- *     1 : You're allowed to connect.
- *     0 : You're not allowed to connect.
- *
- * Throttling:
- *
- * A connection should be rejected when a connection from the same IP number was
- * received IPCHECK_CLONE_LIMIT times before this connect attempt, with
- * reconnect intervals of IPCHECK_CLONE_PERIOD seconds or less.
- *
- * Free target inheritance:
- *
- * When the client is accepted, then the number of Free Targets
- * of the cptr is set to the value stored in the found IPregistry
- * structure, or left at STARTTARGETS.  This can be done by changing
- * cptr->nexttarget to be `now - (TARGET_DELAY * (FREE_TARGETS - 1))',
- * where FREE_TARGETS may range from 0 till STARTTARGETS.
- *--------------------------------------------------------------------------*/
-int IPcheck_local_connect(struct in_addr a, time_t* next_target_out) {
-  assert(0 != next_target_out);
-  return ip_registry_check_local(a.s_addr, next_target_out);
+  /* Finally, store smallest value for Judgement Day */
+  if (free_targets < entry->target->count)
+    entry->target->count = free_targets;
+  
 }
 
-
 /*----------------------------------------------------------------------------
- * IPcheck_remote_connect
+ * ip_registry_remote_disconnect
  *
  * Event:
- *   A remote client connected to Undernet, with IP number `cptr->ip.s_addr'
- *   and hostname `hostname'.
+ *   A remote client disconnected.
  *
  * Action:
  *   Update the IPcheck registry.
- *   Return 0 on failure, 1 on success.
- *--------------------------------------------------------------------------*/
-int IPcheck_remote_connect(struct Client *cptr, int is_burst) {
-  assert(0 != cptr);
-  return ip_registry_check_remote(cptr, is_burst);
-}
-
-
-/*----------------------------------------------------------------------------
- * IPcheck_connect_fail
- *
- * Event:
- *   This local client failed to connect due to legal reasons.
- *
- * Action:
- *   Neutralize the effect of calling IPcheck_local_connect, in such
- *   a way that the client won't be penalized when trying to reconnect
- *   again.
+ *   Remove all expired IPregistry structures from the hash bucket
+ *     that belongs to this clients IP number.
  *--------------------------------------------------------------------------*/
-void IPcheck_connect_fail(struct in_addr a) {
-  ip_registry_connect_fail(a.s_addr);
-}
-
+void ip_registry_remote_disconnect(struct Client *cptr)
+{
+  struct IPRegistryEntry *entry;
 
-/*----------------------------------------------------------------------------
- * IPcheck_connect_succeeded
- *
- * Event:
- *   A client succeeded to finish the registration.
- *
- * Finish IPcheck registration of a successfully, locally connected client.
- *--------------------------------------------------------------------------*/
-void IPcheck_connect_succeeded(struct Client *cptr) {
   assert(0 != cptr);
-  ip_registry_connect_succeeded(cptr);
-}
 
+  entry = ip_registry_find(cptr->ip.s_addr);
+  
+  assert(entry);
+  
+  assert(entry->connected > 0);
+  Debug((DEBUG_DEBUG,"IPcheck: Remote Disconnect"));
 
-/*----------------------------------------------------------------------------
- * IPcheck_disconnect
- *
- * Event:
- *   A local client disconnected or a remote client left Undernet.
- *
- * Action:
- *   Update the IPcheck registry.
- *   Remove all expired IPregistry structures from the hash bucket
- *     that belongs to this clients IP number.
- *--------------------------------------------------------------------------*/
-void IPcheck_disconnect(struct Client *cptr) {
-  assert(0 != cptr);
-  ip_registry_disconnect(cptr);
-}
+  if (entry->connected > 0) {
+    entry->connected--;
+  }
 
+  /*
+   * If this was the last one, set `last_connect' to disconnect time
+   * (used for expiration)   Note that we reset attempts here as well if our
+   * threshhold hasn't been crossed.
+   */
+  if (0 == entry->connected) {
+    ip_registry_update_free_targets(entry);
+    entry->last_connect=NOW;
+  }
+}
 
 /*----------------------------------------------------------------------------
  * IPcheck_nr
  *
  * Returns number of clients with the same IP number
  *--------------------------------------------------------------------------*/
-unsigned short IPcheck_nr(struct Client *cptr) {
-  assert(0 != cptr);
-  return ip_registry_count(cptr->ip.s_addr);
-}
-
-
-/*----------------------------------------------------------------------------
- * IPcheck_expire
- *
- * Expire old entries
- *--------------------------------------------------------------------------*/
-void IPcheck_expire() {
-  static time_t next_expire = 0;
-
-  if (next_expire < CurrentTime) {
-    ip_registry_expire();
-    next_expire = CurrentTime + 60;
-  }
+int ip_registry_count(unsigned int addr)
+{
+  struct IPRegistryEntry *entry = ip_registry_find(addr);
+  return (entry) ? entry->connected : 0;
 }