start of proxyscan work to use iptrie ;/

[irc/quakenet/newserv.git] / proxyscan / proxyscancache.c

/*
 * proxyscancache.c:
 *  This file deals with the cache of known hosts, clean or otherwise.
 */

#include "proxyscan.h"
#include <time.h>
#include <stdio.h>
#include "../core/error.h"
#include <string.h>

time_t cleanscaninterval;
time_t dirtyscaninterval;

void cachehostinit(time_t ri) {
  cleanscaninterval=ri;
  dirtyscaninterval=ri*7;
}

cachehost *addcleanhost(time_t timestamp) {
  cachehost *chp;
  int hash;

  chp=getcachehost();
  chp->lastscan=timestamp;
  chp->proxies=NULL;
  chp->glineid=0;
  
  return chp;
}

void delcachehost(cachehost *chp) {
  cachehost **chh;
  foundproxy *fpp, *nfpp;

  for (fpp=chp->proxies;fpp;fpp=nfpp) {
    nfpp=fpp->next;
    freefoundproxy(fpp);
  }
  freecachehost(chp);
}

/*
 * Returns a cachehost * for the named IP
 */

cachehost *findcachehost(patricia_node_t *node) {
  int hash;
  cachehost *chp;

  if( (cachehost *)node->slots[ps_cache_ext] ) {
    chp = (cachehost *)node->slots[ps_cache_ext];
    if(chp->lastscan < (time(NULL)-(chp->proxies ? dirtyscaninterval : cleanscaninterval))) {
      /* Needs rescan; delete and return 1 */
      delcachehost(chp);
      patricia_deref_prefix(node->prefix);
      node->slots[ps_cache_ext] = NULL;
      return NULL;
    } else {
      /* valid: return it */
      return chp;
    }
  }

  /* Not found: return NULL */
  return NULL;
}

/* 
 * dumpcleanhosts:
 *  Dumps all cached hosts to a savefile.  Expires hosts as it goes along
 */

void dumpcachehosts(void *arg) {
  int i;
  FILE *fp;
  cachehost *chp;
  time_t now=time(NULL);
  foundproxy *fpp;
  patricia_node_t *node;

  if ((fp=fopen("cleanhosts","w"))==NULL) {
    Error("proxyscan",ERR_ERROR,"Unable to open cleanhosts file for writing!");
    return;
  }

  PATRICIA_WALK (iptree->head, node) {
    if ( node->slots[ps_cache_ext] ) {
      chp = (cachehost *) node->slots[ps_cache_ext];
      if (chp->proxies) {
        if (chp->lastscan < (now-dirtyscaninterval)) {
          patricia_deref_prefix(node->prefix);
	  delcachehost(chp);
          continue;
        }
        
        for (fpp=chp->proxies;fpp;fpp=fpp->next) 
          fprintf(fp, "%s %lu %u %i %u\n",IPtostr(node->prefix->sin),chp->lastscan,chp->glineid,fpp->type,fpp->port);
      } else {
        if (chp->lastscan < (now-cleanscaninterval)) {
          /* Needs rescan anyway, so delete it */
	  patricia_deref_prefix(node->prefix);
	  delcachehost(chp);
	  continue;
        }
        fprintf(fp,"%s %lu\n",IPtostr(node->prefix->sin),chp->lastscan);
      }
    }
  } PATRICIA_WALK_END;
  

  fclose(fp);
}

/*
 * loadcleanhosts:
 *  Loads clean hosts in from database. 
 */

void loadcachehosts() {
  FILE *fp;
  unsigned long IP,timestamp,glineid,ptype,pport;
  char buf[512];
  cachehost *chp=NULL;
  foundproxy *fpp;
  char ip[512];
  int res;
  struct irc_in_addr sin;
  unsigned char bits;
  patricia_node_t *node;

  if ((fp=fopen("cleanhosts","r"))==NULL) {
    Error("proxyscan",ERR_ERROR,"Unable to open cleanhosts file for reading!");
    return;
  }


  while (!feof(fp)) {
    fgets(buf,512,fp);
    if (feof(fp)) {
      break;
    }

    res=sscanf(buf,"%s %lu %lu %lu %lu",&ip,&timestamp,&glineid,&ptype,&pport);

    if (res<2)
      continue;

    if (0 == ipmask_parse(ip,&sin, &bits)) {
      /* invalid mask */
    } else {
      node = refnode(iptree, &sin, bits);
      if( node ) {
        chp=addcleanhost(timestamp);
        node->slots[ps_cache_ext] = chp;
      
        if (res==5) {
          chp->glineid=glineid;
          fpp=getfoundproxy();
          fpp->type=ptype;
          fpp->port=pport;
          fpp->next=chp->proxies;
          chp->proxies=fpp;
	}
      }
    }
  }
  
}

/*
 * cleancount:
 *  Returns the number of "clean" host entries present 
 */

unsigned int cleancount() {
  int i;
  unsigned int total=0;
  cachehost *chp;
  patricia_node_t *node;

  PATRICIA_WALK (iptree->head, node) {
    if ( node->slots[ps_cache_ext] ) {
      chp = (cachehost *) node->slots[ps_cache_ext];

      if (!chp->proxies)
        total++;
    }
  } PATRICIA_WALK_END;

  return total;
}

unsigned int dirtycount() {
  int i;
  unsigned int total=0;
  cachehost *chp;
  patricia_node_t *node;

  PATRICIA_WALK (iptree->head, node) {
    if ( node->slots[ps_cache_ext] ) {
      chp = (cachehost *) node->slots[ps_cache_ext];
      if (chp->proxies)
        total++;
    }
  } PATRICIA_WALK_END;

  return total;
}

/* 
 * scanall:
 *  Scans all hosts on the network for a given proxy, and updates the cache accordingly
 */

void scanall(int type, int port) {
  int i;
  cachehost *chp, *nchp;
  nick *np;
  unsigned int hostmarker;
  patricia_node_t *node;

  hostmarker=nexthostmarker();

  PATRICIA_WALK (iptree->head, node) {
    if ( node->slots[ps_cache_ext] ) {
      chp = (cachehost *) node->slots[ps_cache_ext];
      chp->marker=0;
    }
  } PATRICIA_WALK_END; 

  for (i=0;i<NICKHASHSIZE;i++) {
    for (np=nicktable[i];np;np=np->next) {
      if (np->host->marker==hostmarker)
	continue;

      np->host->marker=hostmarker;

      if (!irc_in_addr_is_ipv4(&np->p_ipaddr))
        continue;

      if ((chp=findcachehost(np->ipnode)))
	chp->marker=1;

      queuescan(np->ipnode, type, port, SCLASS_NORMAL, 0);
    }
  }
}