netwhere.hpp

/*
 * Copyright (c) 2017, Ben Smith
 * All rights reserved.
 *
 */

#ifndef __NETWHERE_H__
#define __NETWHERE_H__

#include <tins/tins.h>
#include <boost/format.hpp>

#include "flow_collector.hpp"
#include "reader_writer_exclusion.hpp"
#include "webservice.hpp"
#include "logger.hpp"

class NetWhere {
public:
  NetWhere(const std::string interface, Tins::IPv4Range hosts_range)
    : _interface(interface), _collector(hosts_range),
      _last_stats(0), _last_prune(0) {}

  void start();

  bool handle_packet(const Tins::PDU &pdu);
  std::string handle_request(const std::string& url);

private:
  const int _print_stats_interval = 60;
  const int _prune_interval = 60*60;
  const int _prune_older_than = 60*60*2;

  FlowsCollector _collector;
  ReaderWriterExclusion _rw_exclusion;
  const std::string _interface;
  time_t _last_stats;
  time_t _last_prune;

  std::string hosts();
  std::string host_flows(const std::string& host_key);
  void print_stats() { LOG(boost::format("%1% hosts, %2% flows") % _collector.hosts().size() % _collector.flows().size()); }
  std::string get_hostname(const Host& host);
};

#endif