merge

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

/*
 * ANY functionality
 */

#include "newsearch.h"

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

void any_free(searchCtx *ctx, struct searchNode *thenode);
void *any_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput);

#define MAX_ITERATIONS 1000

struct any_localdata {
  searchNode *genfn;
  searchNode *lambdafn;
  int hitlimit;
};

struct searchNode *any_parse(searchCtx *ctx, int argc, char **argv) {
  searchNode *thenode;
  struct any_localdata *localdata;
  
  if(argc < 2) {
    parseError = "any: usage: any (generatorfn x) (fn ... (var x) ...)";
    return NULL;
  }

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

  localdata->hitlimit = 0;

  if(!(localdata->genfn=ctx->parser(ctx, argv[0]))) {
    free(localdata);
    return NULL;
  }

  localdata->lambdafn = ctx->parser(ctx, argv[1]);
  if(!(localdata->lambdafn = coerceNode(ctx, localdata->lambdafn, RETURNTYPE_BOOL))) {
    (localdata->genfn->free)(ctx, localdata->genfn);
    free(localdata);
    return NULL;
  }
  
  if(!(thenode=(struct searchNode *)malloc(sizeof(struct searchNode)))) {
    parseError = "malloc: could not allocate memory for this search.";
    (localdata->genfn->free)(ctx, localdata->genfn);
    (localdata->lambdafn->free)(ctx, localdata->lambdafn);
    free(localdata);
    return NULL;
  }
  
  thenode->returntype   = RETURNTYPE_BOOL;
  thenode->localdata    = localdata;
  thenode->exe          = any_exe;
  thenode->free         = any_free;

  return thenode;
}

void any_free(searchCtx *ctx, struct searchNode *thenode) {
  struct any_localdata *localdata = thenode->localdata;
  
  if(localdata->hitlimit)
    ctx->reply(senderNSExtern, "Warning: your expression was terminated for %d nicks as it hit the maximum iteration count.", localdata->hitlimit);
  
  (localdata->genfn->free)(ctx, localdata->genfn);
  (localdata->lambdafn->free)(ctx, localdata->lambdafn);
  
  free(localdata);
  free(thenode);
}

void *any_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
  struct any_localdata *localdata = thenode->localdata;
  int i;

/*  if(localdata->hitlimit)
    return (void *)0;
*/

  for(i=0;i<MAX_ITERATIONS;i++) {
    if(!(localdata->genfn->exe)(ctx, localdata->genfn, theinput))
      return (void *)0;
    
    if((localdata->lambdafn->exe)(ctx, localdata->lambdafn, theinput))
      return (void *)1;
  }
  
  localdata->hitlimit++;
  return (void *)0;
}