wraith-ircbot/src/dccutil.cc

/*
 * Copyright (C) 1997 Robey Pointer
 * Copyright (C) 1999 - 2002 Eggheads Development Team
 * Copyright (C) 2002 - 2014 Bryan Drewery
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

/*
 * dccutil.c -- handles:
 *   lots of little functions to send formatted text to
 *   varying types of connections
 *   '.who', '.whom', and '.dccstat' code
 *   memory management for dcc structures
 *   timeout checking for dcc connections
 *
 */


#include <sys/stat.h>
#include <sys/file.h>
#include "common.h"
#include "color.h"
#include "chanprog.h"
#include "userrec.h"
#include "dcc.h"
#include "auth.h"
#include "botnet.h"
#include "adns.h"
#include "net.h"
#include "main.h"
#include "dccutil.h"
#include "misc.h"
#include "botcmd.h"
#include <errno.h>
#include "chan.h"
#include "botmsg.h"
#include "tandem.h"
#include "core_binds.h"
#include "egg_timer.h"
#include "src/mod/server.mod/server.h"
#include <stdarg.h>
#include <bdlib/src/String.h>
#include <bdlib/src/Stream.h>

static struct portmap *root = NULL;

interval_t connect_timeout = 40;    /* How long to wait before a telnet connection times out */
int max_dcc = 200;

static int dcc_flood_thr = 3;

void
init_dcc()
{
  if (!conf.bot->hub || conf.bot->localhub)
    protect_telnet = 0;
  if (max_dcc < 1)
    max_dcc = 1;
  if (dcc)
    dcc = (struct dcc_t *) realloc(dcc, sizeof(struct dcc_t) * max_dcc);
  else
    dcc = (struct dcc_t *) calloc(1, sizeof(struct dcc_t) * max_dcc);
}

/* Replace \n with \r\n */
char *
add_cr(char *buf)
{
  static char WBUF[1024] = "";
  char *p = NULL, *q = NULL;

  for (p = buf, q = WBUF; *p; p++, q++) {
    if (*p == '\n')
      *q++ = '\r';
    *q = *p;
  }
  *q = *p;
  return WBUF;
}

static size_t
colorbuf(char *buf, size_t len, int idx, size_t bufsiz)
{
//  char *buf = *bufp;
  int cidx = coloridx(idx);
  int schar = 0;
  char buf3[1024] = "", buf2[15] = "", c = 0;
  static int8_t stdout_cflags = 0;
  int8_t *cflags;

  if (idx == -1) {
    cflags = &stdout_cflags;
  } else {
    cflags = &dcc[idx].cflags;
  }

  for (size_t i = 0; i < len; i++) {
    c = buf[i];
    buf2[0] = 0;

/*
    if (aqua) {
      if (upper) {
        upper = 0;
        c = toupper(c);
      } else {
        upper = 1;
        c = tolower(c);
      }
    }
*/
    if (cidx) {
      if (schar) {                /* These are for $X replacements */
        schar--;                  /* Unset identifier int */
        switch (c) {
          case 'b':
            if (*cflags & CFLGS_BOLD) {
              strlcpy(buf2, BOLD_END(idx), sizeof(buf2));
              *cflags &= ~CFLGS_BOLD;
            } else {
              *cflags |= CFLGS_BOLD;
              strlcpy(buf2, BOLD(idx), sizeof(buf2));
            }
            break;
          case 'u':
            if (*cflags & CFLGS_UNDERLINE) {
              strlcpy(buf2, UNDERLINE_END(idx), sizeof(buf2));
              *cflags &= ~CFLGS_UNDERLINE;
            } else {
              strlcpy(buf2, UNDERLINE(idx), sizeof(buf2));
              *cflags |= CFLGS_UNDERLINE;
            }
            break;
          case 'f':
            if (*cflags & CFLGS_FLASH) {
              strlcpy(buf2, FLASH_END(idx), sizeof(buf2));
              *cflags &= ~CFLGS_FLASH;
            } else {
              strlcpy(buf2, FLASH(idx), sizeof(buf2));
              *cflags |= CFLGS_FLASH;
            }
            break;
          default:
            /* No identifier, put the '$' back in */
            buf2[0] = '$';
            buf2[1] = c;
            buf2[2] = 0;
            break;
        }
      } else {                    /* These are character replacements */
        switch (c) {
          case '$':
            schar++;
            break;
          case ':':
            simple_snprintf(buf2, sizeof(buf2), "%s%c%s", LIGHTGREY(idx), c, COLOR_END(idx));
            break;
          case '@':
            simple_snprintf(buf2, sizeof(buf2), "%s%c%s", BOLD(idx), c, BOLD_END(idx));
            break;
          case '>':
          case ')':
          case '<':
          case '(':
            simple_snprintf(buf2, sizeof(buf2), "%s%c%s", GREEN(idx), c, COLOR_END(idx));
            break;
          default:
            buf2[0] = c;
            buf2[1] = 0;
            break;
        }
      }
    } else {
      if (schar) {
        schar--;
        switch (c) {
          case 'b':
          case 'u':
          case 'f':
            break;
          default:
            /* No identifier, put the '$' back in */
            buf2[0] = '$';
            buf2[1] = c;
            buf2[2] = 0;
        }
      } else {
        switch (c) {
          case '$':
            schar++;
            break;
          default:
            buf2[0] = c;
            buf2[1] = 0;
            break;
        }
      }
    }
    if (buf2[0])
      strlcat(buf3, buf2, sizeof(buf3));
  }
  return strlcpy(buf, buf3, bufsiz);
}

/* Dump a potentially super-long string of text.
 */
void dumplots(int idx, const char *prefix, const bd::String data)
{
  if (unlikely(!*data)) {
    dprintf(idx, "%s\n", prefix);
    return;
  }

  const size_t max_data_len = 120 - strlen(prefix);
  bd::Array<bd::String> lines = data.split('\n');
  size_t i = 0;

  while (i < lines.length()) {
    bd::String line(lines[i]);

    if (line.length() > max_data_len) {
      // Truncate at last space if possible
      const size_t line_max = std::min(line.length() - 1, max_data_len);
      size_t pos = line_max;

      while (pos != bd::String::npos && !strchr(",:; ", line[pos]))
        --pos;
      if (pos == bd::String::npos)
        pos = line_max;

      if (strchr(",:;", line[pos]))
        ++pos;

      if (bd::String(line(pos)).find("\n") != bd::String::npos) // Newline in remaining: dump it
        dprintf(idx, "%s%s\n", prefix, bd::String(line(pos)).c_str());
      else {
        const size_t tpos = line[pos] == ' ' ? pos + 1 : pos; // Trim out the space
        if (lines.length() - (i + 1) > 0) // Wrapped text: prepend to next line if possible
          lines[i + 1] = line(tpos) + lines[i + 1];
        else // Wrapped text (last line): Add onto list of lines
          lines << line(tpos);
      }
      line.resize(pos);
    }
    dprintf(idx, "%s%s\n", prefix, line.c_str());
    ++i;
  }
}

void
rdprintf(const char* target, int idx, const char *format, ...)
{
  char buf[1024] = "";
  size_t len = 0;
  va_list va;

  va_start(va, format);
  int vlen = egg_vsnprintf(buf, sizeof(buf), format, va);
  va_end(va);

  if (unlikely(vlen < 0)) {
    // Error parsing format..
    return;
  }

  if (size_t(vlen) > (sizeof(buf) - 1)) {
    len = sizeof(buf) - 1;
    buf[len] = 0;
  } else {
    len = size_t(vlen);
  }
  dprintf_real(idx, buf, len, sizeof(buf), target);
}

void
dprintf(int idx, const char *format, ...)
{
  char buf[1024] = "";
  size_t len = 0;
  va_list va;

  va_start(va, format);
  int vlen = egg_vsnprintf(buf, sizeof(buf), format, va);
  va_end(va);

  if (unlikely(vlen < 0)) {
    // Error parsing format..
    return;
  }

  if (size_t(vlen) > (sizeof(buf) - 1)) {
    len = sizeof(buf) - 1;
    buf[len] = 0;
  } else {
    len = size_t(vlen);
  }
  dprintf_real(idx, buf, len, sizeof(buf));
}

void
dprintf_real(int idx, char* buf, size_t len, size_t bufsiz, const char* target)
{
/* this is for color on dcc :P */

  if (unlikely(target) && strcmp(target, conf.bot->nick)) {
    char pbot[1024] = "";
    simple_snprintf(pbot, sizeof(pbot), "rd %zu %d %s", len, idx, buf);
    if (!strcmp(target, "*"))
      putallbots(pbot);
    else
      putbot(target, pbot);
  } else if (unlikely(idx < 0)) {
    tputs(-idx, buf, len);
  } else if (idx > 0x7FF0) {
    if (unlikely(idx == DP_STDOUT)) {
      len = colorbuf(buf, len, -1, bufsiz);
    }

    switch (idx) {
      case DP_DEBUG:
        sdprintf("%s", buf);
        break;
      case DP_STDOUT:
        tputs(STDOUT, buf, len);
        break;
      case DP_STDERR:
        tputs(STDERR, buf, len);
        break;
      case DP_SERVER:
      case DP_HELP:
      case DP_MODE:
      case DP_PLAY:
      case DP_MODE_NEXT:
      case DP_SERVER_NEXT:
      case DP_HELP_NEXT:
      case DP_DUMP:
      case DP_CACHE:
        if (conf.bot->hub)
          break;
        len -= remove_crlf_r(buf);

        if (unlikely((idx == DP_DUMP || (floodless && idx != DP_CACHE)))) {
         if (serv != -1) {
           if (debug_output)
             putlog(LOG_SRVOUT, "@", "[d->] %s", buf);
           write_to_server(buf, len);
         }
        } else
          queue_server(idx, buf, len);
        break;
    }
    return;
  } else {                      /* normal chat text */
    len = colorbuf(buf, len, idx, bufsiz);

    if (len > 1000) {           /* Truncate to fit */
      len = 1000;
      buf[len - 1] = '\n';
      buf[len] = 0;
    }
    if (unlikely(dcc[idx].simul >= 0 && !dcc[idx].irc)) {
      bounce_simul(idx, buf);
    } else if (unlikely(dcc[idx].irc)) {
//      size_t size = strlen(dcc[idx].simulbot) + strlen(buf) + 20;
//      char *ircbuf = (char *) calloc(1, size);

//      simple_snprintf(ircbuf, size, "PRIVMSG %s :%s", dcc[idx].simulbot, buf);
//      tputs(dcc[idx].sock, ircbuf, strlen(ircbuf));
      if (!strcmp(dcc[idx].u.chat->con_chan, "*"))
        privmsg(dcc[idx].simulbot, buf, DP_HELP);
      else
        notice(dcc[idx].simulbot, buf, DP_HELP);
//      free(ircbuf);
    } else {
      if (dcc[idx].type && ((long) (dcc[idx].type->output) == 1)) {
        char *p = add_cr(buf);

        tputs(dcc[idx].sock, p, strlen(p));
      } else if (dcc[idx].type && dcc[idx].type->output) {
        dcc[idx].type->output(idx, buf, dcc[idx].u.other);
      } else {
        tputs(dcc[idx].sock, buf, len);
      }
    }
  }
}

void
chatout(const char *format, ...)
{
  char s[1024] = "", *p = NULL;
  va_list va;

  va_start(va, format);
  egg_vsnprintf(s, sizeof(s), format, va);
  va_end(va);

  if ((p = strrchr(s, '\n')))
    *p++ = 0;

  for (int i = 0; i < dcc_total; i++)
    if (dcc[i].type && (dcc[i].type == &DCC_CHAT) && (dcc[i].simul == -1))
      if (dcc[i].u.chat->channel >= 0)
        dprintf(i, "%s\n", s);
}

/* Print to all on this channel but one.
 */
void
chanout_but(int x, int chan, const char *format, ...)
{
  char s[1024] = "", *p = NULL;
  va_list va;

  va_start(va, format);
  egg_vsnprintf(s, sizeof(s), format, va);
  va_end(va);

  if ((p = strrchr(s, '\n')))
    *p = 0;

  for (int i = 0; i < dcc_total; i++)
    if (dcc[i].type && (dcc[i].type == &DCC_CHAT) && (i != x) && (dcc[i].simul == -1))
      if (dcc[i].u.chat->channel == chan)
        dprintf(i, "%s\n", s);
}

void
dcc_chatter(int idx)
{
  struct flag_record fr = { FR_GLOBAL | FR_CHAN | FR_ANYWH, 0, 0, 0 };

  get_user_flagrec(dcc[idx].user, &fr, NULL);

  if (!glob_party(fr)) {
    dcc[idx].u.chat->channel = -1;
    dprintf(idx, "You don't have partyline chat access; commands only.\n\n");
  } 

  struct chat_info dummy;
  strlcpy(dcc[idx].u.chat->con_chan, "***", sizeof(dummy.con_chan));
  check_bind_chon(dcc[idx].nick, idx);

  dprintf(idx, "Connected to %s, running %s\n", conf.bot->nick, version);
  show_banner(idx);             /* check STAT_BANNER inside function */

  if ((dcc[idx].status & STAT_BOTS) && glob_master(fr)) {
    if ((tands + 1) > 1)
      dprintf(idx, "There are %s-%d- bots%s currently linked.\n", BOLD(idx), tands + 1, BOLD_END(idx));
    else
      dprintf(idx, "There is %s-%d- bot%s currently linked.\n", BOLD(idx), tands + 1, BOLD_END(idx));
    dprintf(idx, " \n");
  }

  if (dcc[idx].status & STAT_WHOM) {
    answer_local_whom(idx, -1);
    dprintf(idx, " \n");
  }

  if (dcc[idx].status & STAT_CHANNELS) {
    show_channels(idx, NULL);
    dprintf(idx, " \n");
  }

  show_motd(idx);

  if (dcc[idx].type == &DCC_CHAT) {
    if (!strcmp(dcc[idx].u.chat->con_chan, "***"))
      strlcpy(dcc[idx].u.chat->con_chan, "*", sizeof(dummy.con_chan));

    if (dcc[idx].u.chat->channel == -2)
      dcc[idx].u.chat->channel = 0;

    if (dcc[idx].u.chat->channel >= 0) {
      if (dcc[idx].u.chat->channel < GLOBAL_CHANS) {
        botnet_send_join_idx(idx);
      }
    }

    /* But *do* bother with sending it locally */
    if (!dcc[idx].u.chat->channel) {
      chanout_but(-1, 0, "*** %s joined the party line.\n", dcc[idx].nick);
    } else if (dcc[idx].u.chat->channel > 0) {
      chanout_but(-1, dcc[idx].u.chat->channel, "*** %s joined the channel.\n", dcc[idx].nick);
    }
  }
}

int
dcc_read(bd::Stream& stream)
{
  int idx = -1;
  bool isserv = 0;
  bd::String buf, type;

  while (stream.tell() < stream.length()) {
    buf = stream.getline().chomp();

    if (buf == STR("+dcc"))
      return idx;
    
    type = newsplit(buf);
    if (type == STR("type")) {
      struct dcc_table *dcc_type = NULL;
      size_t dcc_size = 0;

//      if (buf == STR("CHAT"))
//        dcc_type = &DCC_CHAT;
      if (buf == STR("SERVER")) {
        dcc_type = &SERVER_SOCKET;
        isserv = 1;
      }
//      if (buf == STR("BOT"))
//        dcc_type = &DCC_BOT;
    
      if (dcc_type) {
        idx = new_dcc(dcc_type, dcc_size);
        if (isserv)
          servidx = idx;
      }
    }

    if (idx >= 0) {
      if (type == STR("addr"))
        dcc[idx].addr = my_atoul(buf.c_str());
      if (type == STR("sock")) {
        dcc[idx].sock = atoi(buf.c_str());
        if (isserv)
          serv = dcc[idx].sock;
      }
      if (type == STR("port"))
        dcc[idx].port = atoi(buf.c_str());
      if (type == STR("nick"))
        strlcpy(dcc[idx].nick, buf.c_str(), sizeof(dcc[idx].nick));
      if (type == STR("host"))
        strlcpy(dcc[idx].host, buf.c_str(), sizeof(dcc[idx].host));
    }
  }
  return -1;
}

void 
dcc_write(bd::Stream &stream, int idx)
{
  if (dcc[idx].sock > 0) {
    bd::String buf;

    stream << bd::String::printf(STR("-dcc\n"));
    if (dcc[idx].type)
      stream << bd::String::printf(STR("type %s\n"), dcc[idx].type->name);
//  if (user)
//  stream << bd::String::printf(STR("user %s\n"), dcc[idx].user->handle);
    if (dcc[idx].addr)
      stream << bd::String::printf(STR("addr %u\n"), dcc[idx].addr);
    if (dcc[idx].status)
      stream << bd::String::printf(STR("status %lu\n"), dcc[idx].status);
    stream << bd::String::printf(STR("sock %d\n"), dcc[idx].sock);
//  stream << bd::String::printf(STR("simul %d\n"), dcc[idx].simul);
    if (dcc[idx].port)
      stream << bd::String::printf(STR("port %d\n"), dcc[idx].port);
    if (dcc[idx].nick[0])
      stream << bd::String::printf(STR("nick %s\n"), dcc[idx].nick);
    if (dcc[idx].host[0])
      stream << bd::String::printf(STR("host %s\n"), dcc[idx].host);
    stream << bd::String::printf(STR("+dcc\n"));
  }
}

/* Mark an entry as lost and deconstruct it's contents. It will be securely
 * removed from the dcc list in the main loop.
 */
void
lostdcc(int n)
{
  sdprintf("lostdcc(%d)", n);
  /* Make sure it's a valid dcc index. */
  if (n < 0 || n >= max_dcc)
    return;

  if (n == uplink_idx)
    uplink_idx = -1;
  else if (n == dns_idx)
    dns_idx = -1;
  else if (n == servidx)
    servidx = -1;


  if (dcc[n].type && dcc[n].type->kill)
    dcc[n].type->kill(n, dcc[n].u.other);
  else if (dcc[n].u.other)
    free(dcc[n].u.other);

  dcc[n].u.other = NULL;

//  This is also done when we new_dcc(), so don't bother for now, we set sock/type to NULL, so it won't even be 
//  parsed by anything.
//  bzero(&dcc[n], sizeof(struct dcc_t));

  dcc[n].sock = -1;
  dcc[n].type = NULL;

  dccn--;

  /* last entry! make table smaller :) */
  if (n == (dcc_total - 1))
    dcc_total--;
}

/* Show list of current dcc's to a dcc-chatter
 * positive value: idx given -- negative value: sock given
 */
void
tell_dcc(int idx)
{
  int i;
  size_t j, nicklen = 0;
  char other[160] = "", format[81] = "";

  /* calculate max nicklen */
  for (i = 0; i < dcc_total; i++) {
    if (dcc[i].type && strlen(dcc[i].nick) > (unsigned) nicklen)
      nicklen = strlen(dcc[i].nick);
  }
  if (nicklen < 9)
    nicklen = 9;

  simple_snprintf(format, sizeof format, "%%-4s %%-4s %%-4s %%-8s %%-5s %%-%zus %%-40s %%s\n", nicklen);
  dprintf(idx, format, "SOCK", "IDX", "SSL", "ADDR", "PORT", "NICK", "HOST", "TYPE");
  dprintf(idx, format, "----", "---", "---", "--------", "-----", "---------",
          "----------------------------------------", "----");

  simple_snprintf(format, sizeof format, "%%-4d %%-4d %%-4s %%08X %%5u %%-%zus %%-40s %%s\n", nicklen);

  dprintf(idx, "dccn: %d, dcc_total: %d\n", dccn, dcc_total);
  dprintf(idx, "dns_idx: %d, servidx: %d\n", dns_idx, servidx);
  for (i = 0; i < dcc_total; i++) {
    if (dcc[i].type) {
      j = strlen(dcc[i].host);
      if (j > 40)
        j -= 40;
      else
        j = 0;
      if (dcc[i].type && dcc[i].type->display)
        dcc[i].type->display(i, other, sizeof(other));
      else {
        simple_snprintf(other, sizeof(other), "?:%lX  !! ERROR !!", (long) dcc[i].type);
        break;
      }
      int snum = findanysnum(dcc[i].sock);
      dprintf(idx, format, dcc[i].sock, i, (snum != -1 && socklist[snum].ssl) ? "yes" : "no", dcc[i].addr, dcc[i].port, dcc[i].nick, dcc[i].host + j, other);
    }
  }
}

/* Mark someone on dcc chat as no longer away
 */
void
not_away(int idx)
{
  if (dcc[idx].u.chat->away == NULL) {
    dprintf(idx, "You weren't away!\n");
    return;
  }
  if (dcc[idx].u.chat->channel >= 0) {
    chanout_but(-1, dcc[idx].u.chat->channel, "*** %s is no longer away.\n", dcc[idx].nick);
    if (dcc[idx].u.chat->channel < GLOBAL_CHANS) {
      botnet_send_away(-1, conf.bot->nick, dcc[idx].sock, NULL, idx);
    }
  }
  dprintf(idx, "You're not away any more.\n");
  free(dcc[idx].u.chat->away);
  dcc[idx].u.chat->away = NULL;
}

void
set_away(int idx, char *s)
{
  if (s == NULL) {
    not_away(idx);
    return;
  }
  if (!s[0]) {
    not_away(idx);
    return;
  }
  if (dcc[idx].u.chat->away != NULL)
    free(dcc[idx].u.chat->away);
  dcc[idx].u.chat->away = strdup(s);
  if (dcc[idx].u.chat->channel >= 0) {
    chanout_but(-1, dcc[idx].u.chat->channel, "*** %s is now away: %s\n", dcc[idx].nick, s);
    if (dcc[idx].u.chat->channel < GLOBAL_CHANS) {
      botnet_send_away(-1, conf.bot->nick, dcc[idx].sock, s, idx);
    }
  }
  dprintf(idx, "You are now away. (%s)\n", s);
}


/* Make a password, 10-14 random letters and digits
 */
void
makepass(char *s)
{
  make_rand_str(s, 10 + randint(5));
}

void
flush_lines(int idx, struct chat_info *ci)
{
  int c = ci->line_count;
  struct msgq *p = ci->buffer, *o;

  while (p && c < (ci->max_line)) {
    ci->current_lines--;
    tputs(dcc[idx].sock, p->msg, p->len);
    free(p->msg);
    o = p->next;
    free(p);
    p = o;
    c++;
  }
  if (p != NULL) {
    if (dcc[idx].status & STAT_TELNET)
      tputs(dcc[idx].sock, "[More]: ", 8);
    else
      tputs(dcc[idx].sock, "[More]\n", 7);
  }
  ci->buffer = p;
  ci->line_count = 0;
}

int
new_dcc(struct dcc_table *type, int xtra_size)
{
  if (!type)
    return -1;

  if (dcc_total == max_dcc)
    return -1;

  int i = 0;

  /* Find the first gap */
  for (i = 0; i <= dcc_total; i++)
    if (!dcc[i].type)
      break;

  /* we managed to get to the end of the list! */
  if (unlikely(i == dcc_total)) {
    i = dcc_total;
    dcc_total++;
  }

  dccn++;

  /* empty out the memory for the entry */
  bzero((char *) &dcc[i], sizeof(struct dcc_t));

  dcc[i].type = type;
  if (xtra_size)
    dcc[i].u.other = (char *) calloc(1, xtra_size);
  else
    dcc[i].u.other = NULL;
  dcc[i].simul = -1;
  dcc[i].sock = -1;

  sdprintf("new_dcc (%s): %d (dccn/dcc_total: %d/%d)", type->name, i, dccn, dcc_total);
  return i;
}

/* Changes the given dcc entry to another type.
 */
void
changeover_dcc(int i, struct dcc_table *type, int xtra_size)
{
  /* Free old structure. */
  if (dcc[i].type && dcc[i].type->kill)
    dcc[i].type->kill(i, dcc[i].u.other);
  else if (dcc[i].u.other) {
    free(dcc[i].u.other);
    dcc[i].u.other = NULL;
  }
  dcc[i].type = type;
  if (xtra_size)
    dcc[i].u.other = (char *) calloc(1, xtra_size);
}

int
detect_dcc_flood(time_t * timer, struct chat_info *chat, int idx)
{
  if (!dcc_flood_thr)
    return 0;

  time_t t = now;

  if (*timer != t) {
    *timer = t;
    chat->msgs_per_sec = 0;
  } else {
    chat->msgs_per_sec++;
    if (chat->msgs_per_sec > dcc_flood_thr) {
      /* FLOOD */
      dprintf(idx, "*** FLOOD: Goodbye.\n");
      /* Evil assumption here that flags&DCT_CHAT implies chat type */
      if ((dcc[idx].type->flags & DCT_CHAT) && chat && (chat->channel >= 0)) {
        char x[1024];

        simple_snprintf(x, sizeof x, "%s has been forcibly removed for flooding.\n", dcc[idx].nick);
        chanout_but(idx, chat->channel, "*** %s", x);
        if (chat->channel < GLOBAL_CHANS)
          botnet_send_part_idx(idx, x);
      }
      check_bind_chof(dcc[idx].nick, idx);
      if ((dcc[idx].sock != STDOUT) || backgrd) {
        killsock(dcc[idx].sock);
        lostdcc(idx);
      } else {
        dprintf(DP_STDOUT, "\n### SIMULATION RESET ###\n\n");
        dcc_chatter(idx);
      }
      return 1;                 /* <- flood */
    }
  }
  return 0;
}

/* Handle someone being booted from dcc chat.
 */
void
do_boot(int idx, const char *by, const char *reason)
{
  dprintf(idx, "-=- poof -=-\n");
  dprintf(idx, "You've been booted from the bot by %s%s%s\n", by, reason[0] ? ": " : ".", reason);
  /* If it's a partyliner (chatterer :) */
  /* Horrible assumption that DCT_CHAT using structure uses same format
   * as DCC_CHAT */
  if ((dcc[idx].type->flags & DCT_CHAT) && (dcc[idx].u.chat->channel >= 0)) {
    char x[1024] = "";

    simple_snprintf(x, sizeof x, "%s booted %s from the party line%s%s", by, dcc[idx].nick, reason[0] ? ": " : "", reason);
    chanout_but(idx, dcc[idx].u.chat->channel, "*** %s.\n", x);
    if (dcc[idx].u.chat->channel < GLOBAL_CHANS)
      botnet_send_part_idx(idx, x);
  }
  check_bind_chof(dcc[idx].nick, idx);

  if (dcc[idx].u.chat->su_nick) {
    dcc[idx].user = get_user_by_handle(userlist, dcc[idx].u.chat->su_nick);
    strlcpy(dcc[idx].nick, dcc[idx].u.chat->su_nick, sizeof(dcc[idx].nick));
    dcc[idx].type = &DCC_CHAT;
    dprintf(idx, "Returning to real nick %s!\n", dcc[idx].u.chat->su_nick);
    free(dcc[idx].u.chat->su_nick);
    dcc[idx].u.chat->su_nick = NULL;
    dcc_chatter(idx);

    if (dcc[idx].u.chat->channel < GLOBAL_CHANS && dcc[idx].u.chat->channel >= 0) {
      botnet_send_join_idx(idx);
    }
  } else if ((dcc[idx].sock != STDOUT) || backgrd) {
    killsock(dcc[idx].sock);
    lostdcc(idx);
    /* Entry must remain in the table so it can be logged by the caller */
  } else {
    dprintf(DP_STDOUT, "\n### SIMULATION RESET\n\n");
    dcc_chatter(idx);
  }
  return;
}

int
listen_all(in_port_t lport, bool off, bool should_v6)
{
  int idx = -1;
  in_port_t port, realport;

#ifdef USE_IPV6
  int i6 = -1;
#endif /* USE_IPV6 */
  int i = -1;
  struct portmap *pmap = NULL, *pold = NULL;

  port = realport = lport;
  // If using a random port, lookup the port mapping
  if (lport == 0) {
    for (pmap = root; pmap; pold = pmap, pmap = pmap->next)
      if (pmap->realport == port) {
        port = pmap->mappedto;
        break;
      }
  }

  // Look for an existing open port and close if requested
  for (int ii = 0; ii < dcc_total; ii++) {
    if (dcc[ii].type && (dcc[ii].type == &DCC_TELNET) && (dcc[ii].port == port) &&
           (!strcmp(dcc[ii].nick, "(telnet)") || !strcmp(dcc[ii].nick, "(telnet6)"))) {
      idx = ii;

      if (off) {
        if (lport == 0 && pmap) {
          if (pold) pold->next = pmap->next;
          else root = pmap->next;
          free(pmap);
        }
#ifdef USE_IPV6
        if (sockprotocol(dcc[idx].sock) == AF_INET6)
          putlog(LOG_DEBUG, "*", "Closing IPv6 listening port %d", dcc[idx].port);
        else
#endif /* USE_IPV6 */
          putlog(LOG_DEBUG, "*", "Closing IPv4 listening port %d", dcc[idx].port);
        killsock(dcc[idx].sock);
        lostdcc(idx);
        return idx;
      }
    }
  }
  if (idx < 0) {
    if (off) {
      putlog(LOG_ERRORS, "*", "No such listening port open - %d", lport);
      return idx;
    }
    /* make new one */
    if (dcc_total >= max_dcc) {
      putlog(LOG_ERRORS, "*", "Can't open listening port - no more DCC Slots");
    } else {
      if (should_v6 && (conf.bot->net.ip6 || conf.bot->net.host6)) {
#ifdef USE_IPV6
        i6 = open_listen_by_af(&port, AF_INET6);
        if (i6 < 0) {
          putlog(LOG_ERRORS, "*", "Can't open IPv6 listening port %d - %s", port, i6 == -1 ? "it's taken." : "couldn't assign ip.");
        } else {
          /* now setup ipv4/ipv6 listening port */
          idx = new_dcc(&DCC_TELNET, 0);
          dcc[idx].addr = 0L;
          strlcpy(dcc[idx].host6, myipstr(AF_INET6), sizeof(dcc[idx].host6));
          dcc[idx].port = port;
          dcc[idx].sock = i6;
          dcc[idx].timeval = now;
          strlcpy(dcc[idx].nick, "(telnet6)", sizeof(dcc[idx].nick));
          strlcpy(dcc[idx].host, "*", sizeof(dcc[idx].host));
          putlog(LOG_DEBUG, "*", "Listening on IPv6 at telnet port %d", port);
        }
#endif
      }

      /* now setup ipv4 listening port */
#ifdef USE_IPV6
      i = open_listen_by_af(&port, AF_INET);
#else
      i = open_listen(&port);
#endif /* USE_IPV6 */

      if (i < 0) {
        putlog(LOG_ERRORS, "*", "Can't open IPv4 listening port %d - %s", port, i == -1 ? "it's taken." : "couldn't assign ip.");
      } else {
        idx = new_dcc(&DCC_TELNET, 0);
        dcc[idx].addr = iptolong(getmyip());
        dcc[idx].port = port;
        dcc[idx].sock = i;
        dcc[idx].timeval = now;
        strlcpy(dcc[idx].nick, "(telnet)", sizeof(dcc[idx].nick));
        strlcpy(dcc[idx].host, "*", sizeof(dcc[idx].host));
        putlog(LOG_DEBUG, "*", "Listening on IPv4 at telnet port %d", port);
      }

      // If was asked for a random listen, save it in a mapping
      if (lport == 0) {
#ifdef USE_IPV6
        if (i > 0 || (should_v6 && i6 > 0)) {
#else
          if (i > 0) {
#endif /* USE_IPV6 */
            if (!pmap) {
              pmap = (struct portmap *) calloc(1, sizeof(struct portmap));
              pmap->next = root;
              root = pmap;
            }
            pmap->realport = realport;
            pmap->mappedto = port;
          }
        }
    }
  }
  /* if one of the protocols failed, the one which worked will be returned
   * if both were successful, it wont matter which idx is returned, because the 
   * code reading listen_all will only be reading dcc[idx].port, which would be
   * open on both protocols.
   * -bryan (10/29/03)
   */
  return idx;
}

// @param identd 2 = use oidentd
void
identd_open(const char *sourceIp, const char *destIp, int identd)
{
  // Only open identd socket if running as root
  if (!conf.uid)
  {
    int idx;
    int i = -1;
    in_port_t port = 113;

    for (idx = 0; idx < dcc_total; idx++)
      if (dcc[idx].type == &DCC_IDENTD_CONNECT)
        return;                   /* it's already open :) */

    identd_hack = 1;
#ifdef USE_IPV6
    i = open_listen_by_af(&port, AF_INET6);
#else
    i = open_listen(&port);
#endif /* USE_IPV6 */
    identd_hack = 0;
    if (i >= 0) {
      idx = new_dcc(&DCC_IDENTD_CONNECT, 0);
      if (idx >= 0) {
        egg_timeval_t howlong;

        dcc[idx].addr = iptolong(getmyip());
        dcc[idx].port = port;
        dcc[idx].sock = i;
        dcc[idx].timeval = now;
        strlcpy(dcc[idx].nick, "(identd)", sizeof(dcc[idx].nick));
        strlcpy(dcc[idx].host, "*", sizeof(dcc[idx].host));
        putlog(LOG_DEBUG, "*", "Identd daemon started.");
        howlong.sec = 15;
        howlong.usec = 0;
        timer_create(&howlong, "identd_close()", (Function) identd_close);
      } else
        killsock(i);
    }
  }

  /* Only makes sense if we're spoofing by nick */
  if (identd == 2 && conf.homedir && oidentd && ident_botnick) {
    char oidentd_conf[1024] = "";

    simple_snprintf(oidentd_conf, sizeof(oidentd_conf), "%s/.oidentd.conf", conf.homedir);

    sdprintf("Attempting to spoof ident with oidentd (%s)", oidentd_conf);
    FILE *f = NULL;

    /* Wait for any locks to finish up */
    while ((f = fopen(oidentd_conf, "a+")) == NULL)
      ;

    if (f) {
#ifdef LOCK_EX
      flock(fileno(f), LOCK_EX);
#endif
      fseek(f, 0, SEEK_SET);

      char inbuf[100] = "", outbuf[2048] = "";

      /* Clear any of my matching ips and username */

      while (fgets(inbuf, sizeof(inbuf), f) != NULL) {
        if(inbuf[0] == '\n') continue;
        char *line = strdup(inbuf), *p = line;
        newsplit(&line); /* to */
        newsplit(&line); /* ip */
        if (!strcmp(newsplit(&line), "from") && !strcmp(newsplit(&line), sourceIp)) {
          free(p);
          continue;
        }


        char reply[100] = "";
        simple_snprintf(reply, sizeof(reply), "reply \"%s\"", origbotname);

        if (strstr(line, reply)) {
          free(p);
          continue;
        }

        strlcat(outbuf, inbuf, sizeof(outbuf));
        free(p);
      }

      if (ftruncate(fileno(f), 0))
        goto failure;

      if (strlen(outbuf))
        if (!fwrite(outbuf, 1, strlen(outbuf), f))
          goto failure;

      //And make a record in the oidentd.conf to spoof the ident request from this port->dest-prot
      fprintf(f, "to %s from %s { reply \"%s\" }\n", destIp, sourceIp, origbotname);
      
failure:
      fflush(f);
#ifdef LOCK_EX
      flock(fileno(f), LOCK_UN);
#endif
      fclose(f);
    }
  }
}

void
identd_close()
{
  for (int idx = 0; idx < dcc_total; idx++) {
    if (dcc[idx].type == &DCC_IDENTD_CONNECT) {
      killsock(dcc[idx].sock);
      lostdcc(idx);
      putlog(LOG_DEBUG, "*", "Identd daemon stopped.");
      break;
    }
  }
}

bool
valid_idx(int idx)
{
  if ((idx == -1) || (idx >= dcc_total) || (!dcc[idx].type))
    return 0;
  return 1;
}

int check_cmd_pass(const char *cmd, char *pass)
{
  struct cmd_pass *cp = NULL;

  for (cp = cmdpass; cp; cp = cp->next)
    if (!strcasecmp(cmd, cp->name)) {
      char *epass = NULL;

      /* Does the old pass need to be converted? */
      if (strlen(cp->pass) < SHA1_SALTED_LEN) {
        char out[MAXPASSLEN + 1] = "", *tmp = encrypt_string(pass, pass);
        strlcpy(out, "+", 2);
        strlcat(out, tmp, MAXPASSLEN + 1);
        out[MAXPASSLEN] = 0;
        free(tmp);

        /* No match */
        if (strcmp(out, cp->pass))
          return 0;

        /* Successful match on the old version, convert it and save it */
        char ctmp[256] = "";
        epass = salted_sha1(pass);

        simple_snprintf(ctmp, sizeof(ctmp), "%s %s", cmd, epass);
        free(epass);
        set_cmd_pass(ctmp, 1);
        return 1;
      }

      if (!salted_sha1cmp(cp->pass, pass))
        return 1;
      return 0;
    }
  return 0;
}

int has_cmd_pass(const char *cmd)
{
  struct cmd_pass *cp = NULL;

  for (cp = cmdpass; cp; cp = cp->next)
    if (!strcasecmp(cmd, cp->name))
      return 1;
  return 0;
}

void set_cmd_pass(char *ln, int shareit)
{
  struct cmd_pass *cp = NULL;
  char *cmd = NULL;

  cmd = newsplit(&ln);
  for (cp = cmdpass; cp; cp = cp->next)
    if (!strcmp(cmd, cp->name))
      break;
  if (cp)
    if (ln[0]) {
      /* change */
      strlcpy(cp->pass, ln, sizeof(cp->pass));
      if (shareit)
        botnet_send_cmdpass(-1, cp->name, cp->pass);
    } else {
      if (cp == cmdpass)
        cmdpass = cp->next;
      else {
        struct cmd_pass *cp2;

        cp2 = cmdpass;
        while (cp2->next != cp)
          cp2 = cp2->next;
        cp2->next = cp->next;
      }
      if (shareit)
        botnet_send_cmdpass(-1, cp->name, "");
      free(cp->name);
      free(cp);
  } else if (ln[0]) {
    /* create */
    cp = (struct cmd_pass *) calloc(1, sizeof(struct cmd_pass));
    cp->next = cmdpass;
    cmdpass = cp;
    cp->name = strdup(cmd);
    strlcpy(cp->pass, ln, sizeof(cp->pass));
    if (shareit)
      botnet_send_cmdpass(-1, cp->name, cp->pass);
  }
}

void cmdpass_free(struct cmd_pass *x) 
{
  struct cmd_pass *cp = NULL, *cp_n = NULL;

  for (cp = x; cp; cp = cp_n) {
    cp_n = cp->next;
    list_delete((struct list_type **) &x, (struct list_type *) cp);
    free(cp->name);
    free(cp);
  }
}
/* vim: set sts=2 sw=2 ts=8 et: */