corsaro_dos.c

Go to the documentation of this file.

/*
 * corsaro
 *
 * Alistair King, CAIDA, UC San Diego
 * corsaro-info@caida.org
 *
 * Copyright (C) 2012 The Regents of the University of California.
 *
 * This file is part of corsaro.
 *
 * corsaro 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 3 of the License, or
 * (at your option) any later version.
 *
 * corsaro 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 corsaro.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "config.h"
#include "corsaro_int.h"

#include <arpa/inet.h>
#include <assert.h>
#include <inttypes.h>
#include <stdio.h>
#include <string.h>
#include <time.h>

#include "libtrace.h"
#ifdef HAVE_LIBPACKETDUMP
#include "libpacketdump.h"
#endif

#include "khash.h"
#include "ksort.h"
#include "utils.h"

#include "corsaro_io.h"
#include "corsaro_file.h"
#include "corsaro_log.h"
#include "corsaro_plugin.h"

#include "corsaro_dos.h"

#define CORSARO_DOS_MAGIC 0x45444F53

#define PLUGIN_NAME "dos"

#define PLUGIN_NAME_DEPRECATED "edgar_dos"

static corsaro_plugin_t corsaro_dos_plugin = {
  PLUGIN_NAME,                                 /* name */
  CORSARO_PLUGIN_ID_DOS,                         /* id */
  CORSARO_DOS_MAGIC,                                           /* magic */
  CORSARO_PLUGIN_GENERATE_PTRS(corsaro_dos),  /* func ptrs */
  CORSARO_PLUGIN_GENERATE_TAIL,
};

#define CORSARO_DOS_INTERVAL                   300

#define CORSARO_DOS_VECTOR_TIMEOUT             CORSARO_DOS_INTERVAL

#define CORSARO_DOS_ATTACK_VECTOR_MIN_PACKETS  25

#define CORSARO_DOS_ATTACK_VECTOR_MIN_DURATION 60

#define CORSARO_DOS_ATTACK_VECTOR_MIN_PPM      30

#define CORSARO_DOS_ATTACK_VECTOR_BYTECNT      (4+4+4+4+4+8+4+8+4+8+4+4+4+4+4)

#define CORSARO_DOS_PPM_WINDOW_SIZE      60

#define CORSARO_DOS_PPM_WINDOW_PRECISION 10

#define CORSARO_DOS_PPS_BUCKET_CNT       (CORSARO_DOS_PPM_WINDOW_SIZE/  \
                                        CORSARO_DOS_PPM_WINDOW_PRECISION)

KHASH_SET_INIT_INT(32xx)


typedef struct ppm_window
{
  uint32_t window_start;
  uint64_t buckets[CORSARO_DOS_PPS_BUCKET_CNT];
  uint8_t current_bucket;
  uint64_t max_ppm;
} ppm_window_t;

typedef struct attack_vector
{
  uint8_t *initial_packet;

  uint32_t initial_packet_len;

  uint32_t attacker_ip;

  uint32_t responder_ip;

  uint32_t target_ip;

  uint64_t packet_cnt;

  uint32_t interval_packet_cnt;

  uint64_t byte_cnt;

  uint32_t interval_byte_cnt;

  ppm_window_t   ppm_window;

  struct timeval start_time;

  struct timeval latest_time;

  kh_32xx_t *attack_ip_hash;

  kh_32xx_t *attack_port_hash;

  kh_32xx_t *target_port_hash;

  uint32_t attack_ip_cnt;

} attack_vector_t;

/* need to create an attack_vector_in structure when we write the reading
   stuff */

static attack_vector_t *attack_vector_init(corsaro_t *corsaro)
{
  attack_vector_t *av = NULL;
  if((av = malloc_zero(sizeof(attack_vector_t))) == NULL)
    {
      corsaro_log(__func__, corsaro, "could not malloc memory for attack vector");
      return NULL;
    }

  av->attack_ip_hash = kh_init(32xx);
  assert(av->attack_ip_hash != NULL);

  av->attack_port_hash = kh_init(32xx);
  assert(av->attack_port_hash != NULL);

  av->target_port_hash = kh_init(32xx);
  assert(av->target_port_hash != NULL);

  return av;
}

static void attack_vector_free(attack_vector_t *av)
{
  if(av == NULL)
    {
      return;
    }

  if(av->initial_packet != NULL)
    {
      /*trace_destroy_packet(av->initial_packet);*/
      free(av->initial_packet);
    }

  if(av->attack_ip_hash != NULL)
    {
      kh_destroy(32xx, av->attack_ip_hash);
    }
  if(av->attack_port_hash != NULL)
    {
      kh_destroy(32xx, av->attack_port_hash);
    }
  if(av->target_port_hash != NULL)
    {
      kh_destroy(32xx, av->target_port_hash);
    }

  free(av);
  return;
}

static void attack_vector_reset(attack_vector_t *av)
{
  assert(av != NULL);

  av->interval_packet_cnt = 0;
  av->interval_byte_cnt = 0;
  av->attack_ip_cnt = kh_size(av->attack_ip_hash);
}

#define attack_vector_hash_equal(a, b) (                                \
                                        (a)->target_ip == (b)->target_ip \
                                                                        )

static inline khint32_t attack_vector_hash_func(attack_vector_t *av)
{
  return (khint32_t)av->target_ip*59;
}

KHASH_INIT(av, attack_vector_t*, char, 0,
           attack_vector_hash_func, attack_vector_hash_equal);

struct corsaro_dos_state_t {
  uint32_t first_interval;
  uint16_t number_mismatched_packets;
  khash_t(av) *attack_hash;
  corsaro_file_t *outfile;
};

struct corsaro_dos_in_state_t {
  corsaro_in_record_type_t expected_type;
  int vector_total;
  int vector_cnt;
};

#define STATE(corsaro)                                          \
  (CORSARO_PLUGIN_STATE(corsaro, dos,CORSARO_PLUGIN_ID_DOS))

#define STATE_IN(corsaro)                                               \
  (CORSARO_PLUGIN_STATE(corsaro, dos_in,                        \
                        CORSARO_PLUGIN_ID_DOS))

#define PLUGIN(corsaro)                                         \
  (CORSARO_PLUGIN_PLUGIN(corsaro, CORSARO_PLUGIN_ID_DOS))

static int attack_vector_is_expired(attack_vector_t *vector,
                                    uint32_t time)
{
  if(vector->latest_time.tv_sec + CORSARO_DOS_VECTOR_TIMEOUT < time)
    {
      return 1;
    }
  return 0;
}

static void attack_vector_update_ppm(ppm_window_t *ppm_window)
{
  int i;
  uint64_t this_ppm;

  /* calculate the ppm for the current window in the buckets */
  this_ppm = 0;
  for(i = 0; i < CORSARO_DOS_PPS_BUCKET_CNT; i++)
    {
      this_ppm += ppm_window->buckets[i];
    }
  if(this_ppm > ppm_window->max_ppm)
    {
      ppm_window->max_ppm = this_ppm;
    }
}

static void attack_vector_update_ppm_window(attack_vector_t *vector,
                                           struct timeval tv)
{
  int bucket_offset;
  int i;

  ppm_window_t *ppm_window = &vector->ppm_window;

  bucket_offset = (tv.tv_sec-ppm_window->window_start)/
    CORSARO_DOS_PPM_WINDOW_PRECISION;

  /* this packet is outside of the current bucket */
  if(bucket_offset > 0)
    {
      attack_vector_update_ppm(ppm_window);

      /* zero out the first n buckets in the window */
      for(i = 0; i < bucket_offset && i < 6; i++)
        {
          ppm_window->current_bucket =
            (ppm_window->current_bucket+1) % CORSARO_DOS_PPS_BUCKET_CNT;
          ppm_window->buckets[ppm_window->current_bucket] = 0;
        }
      /* move the start of the window to the end of the zeroed buckets */
      ppm_window->window_start += bucket_offset*
        CORSARO_DOS_PPM_WINDOW_PRECISION;

    }

  /* add this packet to current bucket */
  ppm_window->buckets[ppm_window->current_bucket]++;

  return;
}

static int attack_vector_is_attack(corsaro_t *corsaro,
                                   attack_vector_t *vector,
                                   uint32_t time)
{
  struct timeval duration;
  uint64_t ppm;

  if(vector->packet_cnt < CORSARO_DOS_ATTACK_VECTOR_MIN_PACKETS)
    {
      /* not enough packets */
      return 0;
    }

  if(timeval_subtract(&duration, &vector->latest_time,
                      &vector->start_time) == 1)
    {
      corsaro_log(__func__, corsaro, "last packet seen before first packet!");
      return -1;
    }
  if(duration.tv_sec < CORSARO_DOS_ATTACK_VECTOR_MIN_DURATION)
    {
      /* not long enough */
      return 0;
    }

  attack_vector_update_ppm(&vector->ppm_window);
  ppm = vector->ppm_window.max_ppm;

  if(ppm < CORSARO_DOS_ATTACK_VECTOR_MIN_PPM)
    {
      /* not high enough velocity */
      return 0;
    }

  return 1;
}

static int ascii_dump(corsaro_t *corsaro, attack_vector_t *vector)
{
  uint32_t tmp;
  char t_ip[16];

  tmp = htonl(vector->target_ip);
  inet_ntop(AF_INET,&tmp, &t_ip[0], 16);

  corsaro_file_printf(corsaro, STATE(corsaro)->outfile,
          "%s"
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32".%06"PRIu32
          ",%"PRIu32".%06"PRIu32
          "\n",
          t_ip,
          kh_size(vector->attack_ip_hash),
          kh_size(vector->attack_ip_hash)-vector->attack_ip_cnt,
          kh_size(vector->attack_port_hash),
          kh_size(vector->target_port_hash),
          vector->packet_cnt,
          vector->interval_packet_cnt,
          vector->byte_cnt,
          vector->interval_byte_cnt,
          vector->ppm_window.max_ppm,
          (uint32_t)vector->start_time.tv_sec,
          (uint32_t)vector->start_time.tv_usec,
          (uint32_t)vector->latest_time.tv_sec,
          (uint32_t)vector->latest_time.tv_usec);
  return 0;
}

static int binary_dump(corsaro_t *corsaro, attack_vector_t *vector)
{
  uint8_t av_bytes[CORSARO_DOS_ATTACK_VECTOR_BYTECNT];
  uint8_t *ptr = &av_bytes[0];

  /*
  uint8_t *pkt_buf = NULL;
  libtrace_linktype_t linktype;
  uint32_t pkt_length;
  */

  /* dump the attack vector details */

  bytes_htonl(ptr, vector->target_ip);
  ptr+=4;

  bytes_htonl(ptr, kh_size(vector->attack_ip_hash));
  ptr+=4;

  bytes_htonl(ptr, kh_size(vector->attack_ip_hash)-vector->attack_ip_cnt);
  ptr+=4;

  bytes_htonl(ptr, kh_size(vector->attack_port_hash));
  ptr+=4;

  bytes_htonl(ptr, kh_size(vector->target_port_hash));
  ptr+=4;

  bytes_htonll(ptr, vector->packet_cnt);
  ptr+=8;

  bytes_htonl(ptr, vector->interval_packet_cnt);
  ptr+=4;

  bytes_htonll(ptr, vector->byte_cnt);
  ptr+=8;

  bytes_htonl(ptr, vector->interval_byte_cnt);
  ptr+=4;

  bytes_htonll(ptr, vector->ppm_window.max_ppm);
  ptr+=8;

  bytes_htonl(ptr, vector->start_time.tv_sec);
  ptr+=4;

  bytes_htonl(ptr, vector->start_time.tv_usec);
  ptr+=4;

  bytes_htonl(ptr, vector->latest_time.tv_sec);
  ptr+=4;

  bytes_htonl(ptr, vector->latest_time.tv_usec);
  ptr+=4;

  /* dump the initial packet using trace_get_packet_buffer */
  /*
  if((pkt_buf = trace_get_packet_buffer(vector->initial_packet,
                                        &linktype, NULL)) == NULL ||
     (pkt_length = trace_get_capture_length(vector->initial_packet)) == 0)
    {
      corsaro_log(__func__, "could not get packet buffer");
      return -1;
    }
  */

  /* add the size of the packet to the byte array before we dump it */
  bytes_htonl(ptr, vector->initial_packet_len);

  if(corsaro_file_write(corsaro, STATE(corsaro)->outfile, &av_bytes[0],
                      CORSARO_DOS_ATTACK_VECTOR_BYTECNT) !=
     CORSARO_DOS_ATTACK_VECTOR_BYTECNT)
    {
      corsaro_log(__func__, corsaro, "could not dump vector byte array to file");
      return -1;
    }

  if(corsaro_file_write(corsaro, STATE(corsaro)->outfile, vector->initial_packet,
                      vector->initial_packet_len) !=
     vector->initial_packet_len)
    {
      corsaro_log(__func__, corsaro, "could not dump packet to file");
      return -1;
    }

  return 0;
}

static int read_header(corsaro_in_t *corsaro,
                       corsaro_in_record_type_t *record_type,
                       corsaro_in_record_t *record)
{
  off_t bytes_read;

  if((bytes_read =
      corsaro_io_read_bytes(corsaro, record,
                          sizeof(corsaro_dos_header_t))) !=
     sizeof(corsaro_dos_header_t))
    {
      corsaro_log_in(__func__, corsaro, "failed to read dos header from file");
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return bytes_read;
    }

  ((corsaro_dos_header_t *)record->buffer)->attack_vector_cnt =
  ntohl(((corsaro_dos_header_t *)record->buffer)->attack_vector_cnt);

  assert(bytes_read == sizeof(corsaro_dos_header_t));

  *record_type = CORSARO_IN_RECORD_TYPE_DOS_HEADER;
  STATE_IN(corsaro)->vector_total = ((corsaro_dos_header_t *)
                                  record->buffer)->attack_vector_cnt;

  STATE_IN(corsaro)->expected_type = (STATE_IN(corsaro)->vector_total == 0) ?
    CORSARO_IN_RECORD_TYPE_IO_INTERVAL_END :
    CORSARO_IN_RECORD_TYPE_DOS_ATTACK_VECTOR;

  return bytes_read;
}

static int validate_attack_vector(corsaro_dos_attack_vector_in_t *av)
{
  /* short-circuit if the packet is empty */
  if(av->initial_packet_len == 0)
    {
      return 0;
    }

  /* we need to byte swap */
  av->target_ip = ntohl(av->target_ip);
  av->attacker_ip_cnt = ntohl(av->attacker_ip_cnt);
  av->interval_attacker_ip_cnt = ntohl(av->interval_attacker_ip_cnt);
  av->attack_port_cnt = ntohl(av->attack_port_cnt);
  av->target_port_cnt = ntohl(av->target_port_cnt);
  av->packet_cnt = ntohll(av->packet_cnt);
  av->interval_packet_cnt = ntohl(av->interval_packet_cnt);
  av->byte_cnt = ntohll(av->byte_cnt);
  av->interval_byte_cnt = ntohl(av->interval_byte_cnt);
  av->max_ppm = ntohll(av->max_ppm);
  av->start_time_sec = ntohl(av->start_time_sec);
  av->start_time_usec = ntohl(av->start_time_usec);
  av->latest_time_sec = ntohl(av->latest_time_sec);
  av->latest_time_usec = ntohl(av->latest_time_usec);
  av->initial_packet_len = ntohl(av->initial_packet_len);

  /* NULL it just in case */
  av->initial_packet = NULL;
  return 1;
}

static int read_attack_vector(corsaro_in_t *corsaro,
                              corsaro_in_record_type_t *record_type,
                              corsaro_in_record_t *record)
{
  off_t bytes_read;

  /* the number of bytes that should be read after the first read */
  /* this is the size of the attack vector less the size of the pointer */
  off_t bsbread = sizeof(corsaro_dos_attack_vector_in_t)
    -sizeof(uint8_t*);

  corsaro_dos_attack_vector_in_t *av = NULL;

  /* read the attack vector in record, but not the pointer to the packet
     we will need to find the length before we read it in */

  if((bytes_read = corsaro_io_read_bytes(corsaro, record, bsbread)) != bsbread)
    {
      corsaro_log_in(__func__, corsaro,
                     "failed to read dos attack vector from file");
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return bytes_read;
    }

  av = (corsaro_dos_attack_vector_in_t *)record->buffer;

  if(validate_attack_vector(av) != 1)
    {
      corsaro_log_in(__func__, corsaro, "could not validate attack vector");
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return -1;
    }

  /* now read the packet into the buffer right after the attack vector */
  if((bytes_read +=
      corsaro_io_read_bytes_offset(corsaro, record,
                                   sizeof(corsaro_dos_attack_vector_in_t),
                                   av->initial_packet_len))
     != (bsbread += av->initial_packet_len))
    {
      corsaro_log_in(__func__, corsaro,
                     "failed to read initial packet from file");
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return bytes_read;
    }

  /* update the pointer */
  av->initial_packet = record->buffer+sizeof(corsaro_dos_attack_vector_in_t);

  *record_type = CORSARO_IN_RECORD_TYPE_DOS_ATTACK_VECTOR;

  if(++(STATE_IN(corsaro)->vector_cnt) == STATE_IN(corsaro)->vector_total)
    {
      STATE_IN(corsaro)->expected_type = CORSARO_IN_RECORD_TYPE_IO_INTERVAL_END;
      STATE_IN(corsaro)->vector_total = 0;
      STATE_IN(corsaro)->vector_cnt = 0;
    }

  assert(bytes_read == sizeof(corsaro_dos_attack_vector_in_t)-sizeof(uint8_t*)
         +av->initial_packet_len);

  return bytes_read;
}

static int validate_global_header(corsaro_dos_global_header_t *g)
{
  g->mismatched_pkt_cnt = ntohl(g->mismatched_pkt_cnt);
  g->attack_vector_cnt = ntohl(g->attack_vector_cnt);
  g->non_attack_vector_cnt = ntohl(g->non_attack_vector_cnt);

  return 1;
}

/* == PUBLIC PLUGIN FUNCS BELOW HERE == */

corsaro_plugin_t *corsaro_dos_alloc(corsaro_t *corsaro)
{
  return &corsaro_dos_plugin;
}

int corsaro_dos_probe_filename(const char *fname)
{
  /* look for 'corsaro_dos' in the name */
  if(corsaro_plugin_probe_filename(fname, &corsaro_dos_plugin) == 0)
    {
      if(strstr(fname, PLUGIN_NAME_DEPRECATED) != NULL)
        {
          return 1;
        }
    }
  else
    {
      return 1;
    }
  return 0;
}

int corsaro_dos_probe_magic(corsaro_in_t *corsaro, corsaro_file_in_t *file)
{
  /* unfortunately we cant detect this in corsaro 0.6 files.
     alistair was an idiot and forgot to write an magic number for the
     DOS plugin. */
  return -1;
}

int corsaro_dos_init_output(corsaro_t *corsaro)
{
  struct corsaro_dos_state_t *state;
  /* retrieve a pointer to the plugin struct with our name and id */
  corsaro_plugin_t *plugin = PLUGIN(corsaro);
  assert(plugin != NULL);

  /*
   * allocate memory for the state structure which will hold a pointer to the
   * output file and other statistics
   */
  if((state = malloc_zero(sizeof(struct corsaro_dos_state_t))) == NULL)
    {
      corsaro_log(__func__, corsaro,
                "could not malloc corsaro_dos_state_t");
      goto err;
    }
  /*
   * register the state structure with the plugin manager
   * this associates it with our plugin id so it can be retrieved later
   */
  corsaro_plugin_register_state(corsaro->plugin_manager, plugin, state);

  /* init the hash table for attack vectors */
  state->attack_hash = kh_init(av);

  return 0;

 err:
  corsaro_dos_close_output(corsaro);
  return -1;
}

int corsaro_dos_init_input(corsaro_in_t *corsaro)
{
  struct corsaro_dos_in_state_t *state;
  corsaro_plugin_t *plugin = PLUGIN(corsaro);
  assert(plugin != NULL);

  if((state = malloc_zero(sizeof(struct corsaro_dos_in_state_t)))
     == NULL)
    {
      corsaro_log_in(__func__, corsaro,
                "could not malloc corsaro_dos_state_t");
      goto err;
    }
  corsaro_plugin_register_state(corsaro->plugin_manager, plugin, state);

  /* we initially expect an corsaro interval record */
  state->expected_type = CORSARO_IN_RECORD_TYPE_IO_INTERVAL_START;

  /* don't set the vector_cnt until we actually see a header record */

  return 0;

 err:
  corsaro_dos_close_input(corsaro);
  return -1;
}

int corsaro_dos_close_input(corsaro_in_t *corsaro)
{
  struct corsaro_dos_in_state_t *state = STATE_IN(corsaro);

  if(state != NULL)
    {
      corsaro_plugin_free_state(corsaro->plugin_manager, PLUGIN(corsaro));
    }
  return 0;
}

int corsaro_dos_close_output(corsaro_t *corsaro)
{
  struct corsaro_dos_state_t *state = STATE(corsaro);

  if(state != NULL)
    {
      if(state->attack_hash != NULL)
        {
          kh_free(av, state->attack_hash, &attack_vector_free);
          kh_destroy(av, state->attack_hash);
          state->attack_hash = NULL;
        }

      if(state->outfile != NULL)
        {
          corsaro_file_close(corsaro, state->outfile);
          state->outfile = NULL;
        }
      corsaro_plugin_free_state(corsaro->plugin_manager, PLUGIN(corsaro));
    }
  return 0;
}

off_t corsaro_dos_read_record(struct corsaro_in *corsaro,
                          corsaro_in_record_type_t *record_type,
                          corsaro_in_record_t *record)
{
  struct corsaro_dos_in_state_t *state = STATE_IN(corsaro);

  off_t bytes_read = -1;

  /* this code is adapted from corsaro_flowtuple.c */
  /* we have 5 different types of records that could be in this file */
  switch(state->expected_type)
    {
    case CORSARO_IN_RECORD_TYPE_IO_INTERVAL_START:
      /* ask the io subsystem to read it for us */
      bytes_read = corsaro_io_read_interval_start(corsaro, corsaro->file,
                                                  record_type, record);
      if(bytes_read == sizeof(corsaro_interval_t))
        {
          state->expected_type = CORSARO_IN_RECORD_TYPE_DOS_HEADER;
        }
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_HEADER:
      /* we'll handle this one */
      bytes_read = read_header(corsaro, record_type, record);
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_ATTACK_VECTOR:
      /* we'll handle this too */
      bytes_read = read_attack_vector(corsaro, record_type, record);
      break;

    case CORSARO_IN_RECORD_TYPE_IO_INTERVAL_END:
      /* ask the io subsystem to read it for us */
      bytes_read = corsaro_io_read_interval_end(corsaro, corsaro->file,
                                                record_type, record);
      if(bytes_read == sizeof(corsaro_interval_t))
        {
          state->expected_type = CORSARO_IN_RECORD_TYPE_IO_INTERVAL_START;
        }
      break;

    default:
      corsaro_log_in(__func__, corsaro, "invalid expected record type");
    }

  return bytes_read;
}

off_t corsaro_dos_read_global_data_record(struct corsaro_in *corsaro,
                              enum corsaro_in_record_type *record_type,
                              struct corsaro_in_record *record)
{
  off_t bytes_read;

  if((bytes_read = corsaro_io_read_bytes(corsaro, record,
                                   sizeof(corsaro_dos_global_header_t))) !=
     sizeof(corsaro_dos_global_header_t))
    {
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return bytes_read;
    }

  if(validate_global_header((corsaro_dos_global_header_t *)record->buffer)
     != 1)
    {
      corsaro_log_in(__func__, corsaro, "could not validate global header");
      *record_type = CORSARO_IN_RECORD_TYPE_NULL;
      return -1;
    }

  *record_type = CORSARO_IN_RECORD_TYPE_DOS_GLOBAL_HEADER;

  assert(bytes_read == sizeof(corsaro_dos_global_header_t));

  return bytes_read;
}

int corsaro_dos_start_interval(corsaro_t *corsaro, corsaro_interval_t *int_start)
{
  /* open the output file if it has been closed */
  if(STATE(corsaro)->outfile == NULL &&
     (STATE(corsaro)->outfile =
      corsaro_io_prepare_file(corsaro, PLUGIN(corsaro)->name,
                              int_start)) == NULL)
    {
      corsaro_log(__func__, corsaro, "could not open %s output file",
                  PLUGIN(corsaro)->name);
      return -1;
    }

  if(STATE(corsaro)->first_interval == 0)
    {
      /* -1 to simulate the end of the 'previous' interval */
      STATE(corsaro)->first_interval = int_start->time-1;
    }

  return 0;
}

int corsaro_dos_end_interval(corsaro_t *corsaro, corsaro_interval_t *int_end)
{
  int this_interval = int_end->time-STATE(corsaro)->first_interval;

  khiter_t i;
  attack_vector_t *vector;
  attack_vector_t **attack_arr = NULL;
  int attack_arr_cnt = 0;

  uint8_t gbuf[12];
  uint8_t cntbuf[4];

  if(this_interval < CORSARO_DOS_INTERVAL)
    {
      /* we haven't run for long enough to dump */
      return 0;
    }
  else
    {
      /* we either have hit exactly the right amount of time,
         or we have gone for too long, dump now and reset the counter */
      STATE(corsaro)->first_interval = int_end->time;
      /* fall through and continue to dump */
    }

  /* this is an interval we care about */

  /* malloc an array big enough to hold the entire hash even though we wont
     need it to be that big */
  if((attack_arr =
      malloc(sizeof(attack_vector_t *)*
             kh_size(STATE(corsaro)->attack_hash))) == NULL)
    {
      corsaro_log(__func__, corsaro,
                  "could not malloc array for attack vectors");
      return -1;
    }

  /* classify the flows and dump the attack ones */

  for(i = kh_begin(STATE(corsaro)->attack_hash);
      i != kh_end(STATE(corsaro)->attack_hash); ++i)
    {
      if(kh_exist(STATE(corsaro)->attack_hash, i))
        {
          vector = kh_key(STATE(corsaro)->attack_hash, i);

          if(attack_vector_is_expired(vector, int_end->time) != 0)
            {
              kh_del(av, STATE(corsaro)->attack_hash, i);
              attack_vector_free(vector);
              vector = NULL;
            }
          else if(attack_vector_is_attack(corsaro, vector, int_end->time) != 0)
            {
              /* this is an attack */
              /* add it to the attack array so we can know how many
                 before we dump it */
              attack_arr[attack_arr_cnt] = vector;
              attack_arr_cnt++;
            }
          else
            {
              attack_vector_reset(vector);
            }
        }
    }

  corsaro_io_write_interval_start(corsaro, STATE(corsaro)->outfile,
                                  &corsaro->interval_start);
  if(corsaro->global_file != NULL)
    {
      corsaro_io_write_plugin_start(corsaro, corsaro->global_file,
                                    PLUGIN(corsaro));
    }
  if(CORSARO_FILE_MODE(STATE(corsaro)->outfile) == CORSARO_FILE_MODE_ASCII)
    {
      if(corsaro->global_file != NULL)
        {
          /* global stats */
          /* dump the number of mismatched packets and vectors */
          corsaro_file_printf(corsaro, corsaro->global_file,
                              "mismatch: %"PRIu32"\n"
                              "attack_vectors: %"PRIu32"\n"
                              "non-attack_vectors: %"PRIu32"\n",
                              STATE(corsaro)->number_mismatched_packets,
                              attack_arr_cnt,
                              kh_size(STATE(corsaro)->attack_hash)
                              -attack_arr_cnt);
        }

      /* dump the number of vectors */
      corsaro_file_printf(corsaro, STATE(corsaro)->outfile, "%"PRIu32"\n",
                        attack_arr_cnt);
      /* dump the vectors */
      for(i = 0; i < attack_arr_cnt; i++)
        {
          if(ascii_dump(corsaro, attack_arr[i]) != 0)
            {
              corsaro_log(__func__, corsaro, "could not dump hash");
              return -1;
            }
          /* reset the interval stats */
          attack_vector_reset(attack_arr[i]);
        }
    }
  else if(CORSARO_FILE_MODE(STATE(corsaro)->outfile) == CORSARO_FILE_MODE_BINARY)
      {
        if(corsaro->global_file != NULL)
          {
            /* global stats */
            bytes_htonl(&gbuf[0], STATE(corsaro)->number_mismatched_packets);
            bytes_htonl(&gbuf[4], attack_arr_cnt);
            bytes_htonl(&gbuf[8],
                        kh_size(STATE(corsaro)->attack_hash)-attack_arr_cnt);
            if(corsaro_file_write(corsaro, corsaro->global_file,
                                  &gbuf[0], 12) != 12)
              {
                corsaro_log(__func__, corsaro,
                            "could not dump global stats to file");
                return -1;
              }
          }

        /* dump the number of vectors */
        bytes_htonl(&cntbuf[0], attack_arr_cnt);
        if(corsaro_file_write(corsaro, STATE(corsaro)->outfile,
                            &cntbuf[0], 4) != 4)
          {
            corsaro_log(__func__, corsaro,
                        "could not dump vector count to file");
            return -1;
          }
        /* dump the vectors */
        for(i = 0; i < attack_arr_cnt; i++)
          {
            if(binary_dump(corsaro, attack_arr[i]) != 0)
              {
                corsaro_log(__func__, corsaro, "could not dump hash");
                return -1;
              }
            attack_vector_reset(attack_arr[i]);
          }
      }
  else
    {
      corsaro_log(__func__, corsaro, "invalid mode");
      return -1;
    }
  if(corsaro->global_file != NULL)
    {
      corsaro_io_write_plugin_end(corsaro, corsaro->global_file,
                                  PLUGIN(corsaro));
    }
  corsaro_io_write_interval_end(corsaro, STATE(corsaro)->outfile, int_end);

  STATE(corsaro)->number_mismatched_packets = 0;

  free(attack_arr);

  /* if we are rotating, now is when we should do it */
  if(corsaro_is_rotate_interval(corsaro))
    {
      /* close the current file */
      if(STATE(corsaro)->outfile != NULL)
        {
          corsaro_file_close(corsaro, STATE(corsaro)->outfile);
          STATE(corsaro)->outfile = NULL;
        }
    }

  return 0;
}

int corsaro_dos_process_packet(corsaro_t *corsaro,
                             corsaro_packet_t *packet)
{
  libtrace_packet_t *ltpacket = LT_PKT(packet);
  void *temp = NULL;
  uint8_t proto;
  uint32_t remaining;

  libtrace_ip_t *ip_hdr = NULL;
  libtrace_icmp_t *icmp_hdr = NULL;
  libtrace_ip_t *inner_ip_hdr = NULL;

  /* borrowed from libtrace's protocols.h (used by trace_get_*_port) */
  struct ports_t {
    uint16_t src;           
    uint16_t dst;           
  };

  uint16_t attacker_port = 0;
  uint16_t target_port = 0;

  attack_vector_t findme;

  int khret;
  khiter_t khiter;
  attack_vector_t *vector = NULL;
  uint8_t *pkt_buf = NULL;
  libtrace_linktype_t linktype;

  struct timeval tv;

  if((packet->state.flags & CORSARO_PACKET_STATE_FLAG_BACKSCATTER) == 0)
    {
      /* not a backscatter packet */
      return 0;
    }

  /* backscatter packet, lets find the flow */
  /* check for ipv4 */
  /* 10/19/12 ak replaced much more verbose code to get header with this */
  if((ip_hdr = trace_get_ip(ltpacket)) == NULL)
    {
      /* non-ipv4 packet */
      return 0;
    }

  /* get the transport header */
  if((temp = trace_get_transport(ltpacket, &proto, &remaining)) == NULL)
    {
      /* not enough payload */
      return 0;
    }

  findme.target_ip = 0;

  if(ip_hdr->ip_p == TRACE_IPPROTO_ICMP && remaining >= 2)
    {
      icmp_hdr = (libtrace_icmp_t *)temp;

      if((icmp_hdr->type == 3  ||
          icmp_hdr->type == 4  ||
          icmp_hdr->type == 5  ||
          icmp_hdr->type == 11 ||
          icmp_hdr->type == 12) &&
         ((temp = trace_get_payload_from_icmp(icmp_hdr, &remaining)) != NULL
         && remaining >= 20 && (inner_ip_hdr = (libtrace_ip_t *)temp) &&
          inner_ip_hdr->ip_v == 4))
        {
          /* icmp error message */
          if(inner_ip_hdr->ip_src.s_addr != ip_hdr->ip_dst.s_addr)
            {
              STATE(corsaro)->number_mismatched_packets++;
            }

          findme.target_ip = ntohl(inner_ip_hdr->ip_dst.s_addr);

          /* just extract the first four bytes of payload as ports */
          if((temp = trace_get_payload_from_ip(inner_ip_hdr, NULL,
                                               &remaining)) != NULL
             && remaining >= 4)
            {
              attacker_port = ntohs(((struct ports_t *)temp)->src);
              target_port = ntohs(((struct ports_t *)temp)->dst);
            }
        }
      else
        {
          findme.target_ip =  ntohl(ip_hdr->ip_src.s_addr);
          attacker_port = ntohs(icmp_hdr->code);
          target_port = ntohs(icmp_hdr->type);
        }
    }
  else if((ip_hdr->ip_p == TRACE_IPPROTO_TCP ||
          ip_hdr->ip_p == TRACE_IPPROTO_UDP) &&
          remaining >= 4)
    {
      findme.target_ip = ntohl(ip_hdr->ip_src.s_addr);
      attacker_port = trace_get_destination_port(ltpacket);
      target_port = trace_get_source_port(ltpacket);
    }

  if(findme.target_ip == 0)
    {
      /* the packet is none of ICMP, TCP or UDP */
      return 0;
    }

  tv = trace_get_timeval(ltpacket);

  /* is this vector in the hash? */
  assert(STATE(corsaro)->attack_hash != NULL);
  if((khiter = kh_get(av, STATE(corsaro)->attack_hash, &findme))
     != kh_end(STATE(corsaro)->attack_hash))
    {
      /* the vector is in the hash */
      vector = kh_key(STATE(corsaro)->attack_hash, khiter);

      if(attack_vector_is_expired(vector, tv.tv_sec) != 0)
        {
          kh_del(av, STATE(corsaro)->attack_hash, khiter);
          attack_vector_free(vector);
          vector = NULL;
        }
    }

  if(vector == NULL)
    {
      /* create a new vector and fill it */
      if((vector = attack_vector_init(corsaro)) == NULL)
        {
          corsaro_log(__func__, corsaro, "failed to create new attack vector");
          return -1;
        }

      /* i think this may be buggy. do it the safe way for now
      vector->initial_packet = corsaro_mincopy_packet(packet);
      */
      vector->initial_packet_len = trace_get_capture_length(ltpacket);

      if((vector->initial_packet = malloc(vector->initial_packet_len)) == NULL)
        {
          corsaro_log(__func__, corsaro, "could not malloc initial packet");
          return -1;
        }

      if((pkt_buf = trace_get_packet_buffer(ltpacket,
                                            &linktype, NULL)) == NULL)
        {
          corsaro_log(__func__, corsaro, "could not get packet buffer");
          return -1;
        }

      memcpy(vector->initial_packet, pkt_buf, vector->initial_packet_len);

      vector->attacker_ip = ntohl(ip_hdr->ip_dst.s_addr);
      vector->responder_ip = ntohl(ip_hdr->ip_src.s_addr);
      vector->target_ip = findme.target_ip;

      vector->start_time = tv;

      vector->ppm_window.window_start = tv.tv_sec;

      /* add to the hash */
      khiter = kh_put(av, STATE(corsaro)->attack_hash, vector, &khret);
    }

  assert(vector != NULL);

  vector->packet_cnt++;
  vector->interval_packet_cnt++;
  vector->byte_cnt += ntohs(ip_hdr->ip_len);
  vector->interval_byte_cnt += ntohs(ip_hdr->ip_len);

  vector->latest_time = tv;
  /* update the pps window */
  attack_vector_update_ppm_window(vector, tv);

  /* add the attacker ip to the hash */
  kh_put(32xx, vector->attack_ip_hash, ntohl(ip_hdr->ip_dst.s_addr), &khret);

  /* add the ports to the hashes */
  kh_put(32xx, vector->attack_port_hash, attacker_port, &khret);
  kh_put(32xx, vector->target_port_hash, target_port, &khret);

  return 0;
}

/* ==== External Output Convenience Functions ==== */

void corsaro_dos_attack_vector_get_packet(
                            corsaro_dos_attack_vector_in_t *attack_vector,
                            libtrace_packet_t *packet)
{
  assert(packet != NULL);

  trace_construct_packet(packet, TRACE_TYPE_ETH,
                         attack_vector->initial_packet,
                         attack_vector->initial_packet_len);
}

off_t corsaro_dos_global_header_fprint(corsaro_t *corsaro,
                                corsaro_file_t *file,
                                corsaro_dos_global_header_t *header)
{
  assert(corsaro != NULL);
  assert(file != NULL);
  assert(header != NULL);

  return corsaro_file_printf(corsaro, file,
                             "mismatch: %"PRIu32"\n"
                             "attack_vectors: %"PRIu32"\n"
                             "non-attack_vectors: %"PRIu32"\n",
                             header->mismatched_pkt_cnt,
                             header->attack_vector_cnt,
                             header->non_attack_vector_cnt);

}

void corsaro_dos_global_header_print(corsaro_dos_global_header_t *header)
{
  assert(header != NULL);

  fprintf(stdout, "mismatch: %"PRIu32"\n"
          "attack_vectors: %"PRIu32"\n"
          "non-attack_vectors: %"PRIu32"\n",
          header->mismatched_pkt_cnt,
          header->attack_vector_cnt,
          header->non_attack_vector_cnt);
}

off_t corsaro_dos_attack_vector_fprint(corsaro_t *corsaro,
                                       corsaro_file_t *file,
                                       corsaro_dos_attack_vector_in_t *av)
{
  uint32_t tmp;
  char t_ip[16];

  assert(corsaro != NULL);
  assert(file != NULL);
  assert(av != NULL);

  tmp = htonl(av->target_ip);
  inet_ntop(AF_INET,&tmp, &t_ip[0], 16);

  return corsaro_file_printf(corsaro, file,
                             "%s"
                             ",%"PRIu32
                             ",%"PRIu32
                             ",%"PRIu32
                             ",%"PRIu32
                             ",%"PRIu64
                             ",%"PRIu32
                             ",%"PRIu64
                             ",%"PRIu32
                             ",%"PRIu64
                             ",%"PRIu32".%06"PRIu32
                             ",%"PRIu32".%06"PRIu32
                             "\n",
                             t_ip,
                             av->attacker_ip_cnt,
                             av->interval_attacker_ip_cnt,
                             av->attack_port_cnt,
                             av->target_port_cnt,
                             av->packet_cnt,
                             av->interval_packet_cnt,
                             av->byte_cnt,
                             av->interval_byte_cnt,
                             av->max_ppm,
                             av->start_time_sec,
                             av->start_time_usec,
                             av->latest_time_sec,
                             av->latest_time_usec);

}

void corsaro_dos_attack_vector_print(corsaro_dos_attack_vector_in_t *av)
{
  uint32_t tmp;
  char t_ip[16];
  libtrace_packet_t *packet;

  assert(av != NULL);

  tmp = htonl(av->target_ip);
  inet_ntop(AF_INET,&tmp, &t_ip[0], 16);

  fprintf(stdout,
          "%s"
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32
          ",%"PRIu64
          ",%"PRIu32".%06"PRIu32
          ",%"PRIu32".%06"PRIu32
          "\n",
          t_ip,
          av->attacker_ip_cnt,
          av->interval_attacker_ip_cnt,
          av->attack_port_cnt,
          av->target_port_cnt,
          av->packet_cnt,
          av->interval_packet_cnt,
          av->byte_cnt,
          av->interval_byte_cnt,
          av->max_ppm,
          av->start_time_sec,
          av->start_time_usec,
          av->latest_time_sec,
          av->latest_time_usec);

  /* this may get slow if you are dumping *lots* of dos records */
  if ((packet = trace_create_packet()) == NULL) {
    corsaro_log_file(__func__, NULL, "error creating libtrace packet");
    return;
  }

  corsaro_dos_attack_vector_get_packet(av, packet);

#if HAVE_LIBPACKETDUMP
  fprintf(stdout, "START PACKET\n");
  trace_dump_packet(packet);
  fprintf(stdout, "\nEND PACKET\n");
#else
  fprintf(stdout, "corsaro not built with libpacketdump support\n"
          "not dumping initial packet\n");
#endif
}

off_t corsaro_dos_header_fprint(corsaro_t *corsaro,
                                corsaro_file_t *file,
                                corsaro_dos_header_t *header)
{
  assert(corsaro != NULL);
  assert(file != NULL);
  assert(header != NULL);

  return corsaro_file_printf(corsaro, file,
                             "%"PRIu32"\n",
                             header->attack_vector_cnt);

}

void corsaro_dos_header_print(corsaro_dos_header_t *header)
{
  assert(header != NULL);

  fprintf(stdout, "%"PRIu32"\n", header->attack_vector_cnt);

}

off_t corsaro_dos_record_fprint(corsaro_t *corsaro,
                                corsaro_file_t *file,
                                corsaro_in_record_type_t record_type,
                                corsaro_in_record_t *record)
{
  switch(record_type)
    {
    case CORSARO_IN_RECORD_TYPE_DOS_GLOBAL_HEADER:
      return corsaro_dos_global_header_fprint(corsaro, file,
                            (corsaro_dos_global_header_t *)record->buffer);
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_HEADER:
      return corsaro_dos_header_fprint(corsaro, file,
                            (corsaro_dos_header_t *)record->buffer);
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_ATTACK_VECTOR:
      return corsaro_dos_attack_vector_fprint(corsaro, file,
                            (corsaro_dos_attack_vector_in_t *)record->buffer);
      break;

    default:
      corsaro_log(__func__, corsaro, "record_type %d not a dos record",
                record_type);
      return -1;
      break;
    }

  return -1;
}

int corsaro_dos_record_print(corsaro_in_record_type_t record_type,
                             corsaro_in_record_t *record)
{
  switch(record_type)
    {
    case CORSARO_IN_RECORD_TYPE_DOS_GLOBAL_HEADER:
      corsaro_dos_global_header_print(
                        (corsaro_dos_global_header_t *)record->buffer);
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_HEADER:
      corsaro_dos_header_print(
                        (corsaro_dos_header_t *)record->buffer);
      break;

    case CORSARO_IN_RECORD_TYPE_DOS_ATTACK_VECTOR:
      corsaro_dos_attack_vector_print(
                        (corsaro_dos_attack_vector_in_t *)record->buffer);
      break;

    default:
      corsaro_log_file(__func__, NULL,
                       "record_type %d not a dos record",
                       record_type);
      return -1;
      break;
    }

  return 0;
}