newsearch changes to support addition of trust_search/patriciasearch

[irc/quakenet/newserv.git] / newsearch / ns-country.c

/*
 * COUNTRY functionality
 */

#include "newsearch.h"

#include <stdio.h>
#include <stdlib.h>

#include "../irc/irc_config.h"
#include "../lib/irc_string.h"
#include "../geoip/geoip.h"
#include "../core/modules.h"

void *country_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput);
void country_free(searchCtx *ctx, struct searchNode *thenode);

int ext;

struct searchNode *country_parse(searchCtx *ctx, int argc, char **argv) {
  struct searchNode *thenode;
  GeoIP_LookupCode l;
  long target;

  if (argc<1) {
    parseError = "country: usage: country <country 2 character ISO code>";
    return NULL;
  }

  ext = findnickext("geoip");
  if(ext == -1) {
    parseError = "country: Geoip module not loaded.";
    return NULL;
  }
  
  l = ndlsym("geoip", "geoip_lookupcode");
  target = l(argv[0]);
  if(target == -1) {
    parseError = "country: invalid country.";
    return NULL;
  }

  if (!(thenode=(struct searchNode *)malloc(sizeof (struct searchNode)))) {
    parseError = "malloc: could not allocate memory for this search.";
    return NULL;
  }

  thenode->returntype = RETURNTYPE_BOOL;
  thenode->localdata = (void *)target;
  thenode->exe = country_exe;
  thenode->free = country_free;

  return thenode;
}

void *country_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
  nick *np = (nick *)theinput;
  long country = (long)thenode->localdata, rc = (long)np->exts[ext];

  if(country == rc)
    return (void *)1;

  return (void *)0;
}

void country_free(searchCtx *ctx, struct searchNode *thenode) {
  free(thenode);
}