SearchCtx should contain 'type' - this is to make life easier when defining new searc...

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

/*
 * KILL functionality 
 */

#include "newsearch.h"

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

#include "../control/control.h"
#include "../localuser/localuser.h" /* killuser() */
#include "../lib/irc_string.h" /* IPtostr() */
#include "../lib/strlfunc.h"

/* used for *_free functions that need to warn users of certain things
   i.e. hitting too many users in a (kill) or (gline) - declared in newsearch.c */
extern nick *senderNSExtern;

void *kill_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput);
void kill_free(searchCtx *ctx, struct searchNode *thenode);
static const char *defaultreason = "You (%n) have been disconnected for violating our terms of service";

struct kill_localdata {
  unsigned int marker;
  int count;
  char reason[NSMAX_REASON_LEN];
};

struct searchNode *kill_parse(searchCtx *ctx, int argc, char **argv) {
  struct kill_localdata *localdata;
  struct searchNode *thenode;
  int len;

  if (!(localdata = (struct kill_localdata *) malloc(sizeof(struct kill_localdata)))) {
    parseError = "malloc: could not allocate memory for this search.";
    return NULL;
  }
  localdata->count = 0;
  if (ctx->type == SEARCHTYPE_CHANNEL)
    localdata->marker = nextchanmarker();
  else
    localdata->marker = nextnickmarker();

  if (argc==1) {
    char *p = argv[0];
    if(*p == '\"')
      p++;
    len = strlcpy(localdata->reason, p, sizeof(localdata->reason));
    if(len >= sizeof(localdata->reason)) {
      localdata->reason[sizeof(localdata->reason)-1] = '\0';
    } else {
      localdata->reason[len-1] = '\0';
    }
  }
  else
    strlcpy(localdata->reason, defaultreason, sizeof(localdata->reason));

  if (!(thenode=(struct searchNode *)malloc(sizeof (struct searchNode)))) {
    /* couldn't malloc() memory for thenode, so free localdata to avoid leakage */
    parseError = "malloc: could not allocate memory for this search.";
    free(localdata);
    return NULL;
  }

  thenode->returntype = RETURNTYPE_BOOL;
  thenode->localdata = localdata;
  thenode->exe = kill_exe;
  thenode->free = kill_free;

  return thenode;
}

void *kill_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) {
  struct kill_localdata *localdata;
  nick *np;
  chanindex *cip;

  localdata = thenode->localdata;

  if (ctx->type == SEARCHTYPE_CHANNEL) {
    cip = (chanindex *)theinput;
    cip->marker = localdata->marker;
    localdata->count += cip->channel->users->totalusers;
  }
  else {
    np = (nick *)theinput;
    np->marker = localdata->marker;
    localdata->count++;
  }

  return (void *)1;
}

void kill_free(searchCtx *ctx, struct searchNode *thenode) {
  struct kill_localdata *localdata;
  nick *np, *nnp;
  chanindex *cip;
  int i, j, safe=0;
  unsigned int nickmarker;
  char msgbuf[512];

  localdata = thenode->localdata;

  if (localdata->count > NSMAX_KILL_LIMIT) {
    /* need to warn the user that they have just tried to twat half the network ... */
    ctx->reply(senderNSExtern, "Warning: your pattern matches too many users (%d) - nothing done.", localdata->count);
    free(localdata);
    free(thenode);
    return;
  }

  /* For channel searches, mark up all the nicks in the relevant channels first */
  if (ctx->type == SEARCHTYPE_CHANNEL) {
    nickmarker=nextnickmarker();
    for (i=0;i<CHANNELHASHSIZE;i++) {
      for (cip=chantable[i];cip;cip=cip->next) {
        /* Skip empty and non-matching channels */
        if (!cip->channel || cip->marker != localdata->marker)
          continue;

        for (j=0;j<cip->channel->users->hashsize;j++) {
          if (cip->channel->users->content[j]==nouser)
            continue;
  
          if ((np=getnickbynumeric(cip->channel->users->content[j])))
            np->marker=nickmarker;
        }
      }
    }
  } else {
    /* For nick searches they're already marked, pick up the saved value */
    nickmarker=localdata->marker;
  }

  /* Now do the actual kills */
  for (i=0;i<NICKHASHSIZE;i++) {
    for (np=nicktable[i];np;np=nnp) {
      nnp = np->next;

      if (np->marker != nickmarker)
        continue;

      if (IsOper(np) || IsService(np) || IsXOper(np)) {
        safe++;
        continue;
      }

      nssnprintf(msgbuf, sizeof(msgbuf), localdata->reason, np);
      killuser(NULL, np, "%s", msgbuf);
    }
  }

  if (safe)
    ctx->reply(senderNSExtern, "Warning: your pattern matched privileged users (%d in total) - these have not been touched.", safe);
  /* notify opers of the action */
  ctx->wall(NL_KICKKILLS, "%s/%s killed %d %s via %s [%d untouched].", senderNSExtern->nick, senderNSExtern->authname, (localdata->count - safe), 
    (localdata->count - safe) != 1 ? "users" : "user", (ctx->type == SEARCHTYPE_CHANNEL) ? "chansearch" : "nicksearch", safe);
  free(localdata);
  free(thenode);
}