#ifndef _L4_STATS_H_

#define _L4_STATS_H_

class L4Stats {

 private:

  TrafficStats tcp_sent, tcp_rcvd;

  TrafficStats udp_sent, udp_rcvd;

  TrafficStats icmp_sent, icmp_rcvd;

  TrafficStats other_ip_sent, other_ip_rcvd;

 public:

  void luaStats(lua_State* vm);

  void incStats(time_t when, u_int8_t l4_proto, u_int64_t rcvd_packets,

                u_int64_t rcvd_bytes, u_int64_t sent_packets,

                u_int64_t sent_bytes);

  inline TrafficStats* getTCPSent() { return (&tcp_sent); }

  inline TrafficStats* getTCPRcvd() { return (&tcp_rcvd); }

  inline TrafficStats* getUDPSent() { return (&udp_sent); }

  inline TrafficStats* getUDPRcvd() { return (&udp_rcvd); }

  inline TrafficStats* getICMPSent() { return (&icmp_sent); }

  inline TrafficStats* getICMPRcvd() { return (&icmp_rcvd); }

  inline TrafficStats* getOtherIPSent() { return (&other_ip_sent); }

  inline TrafficStats* getOtherIPRcvd() { return (&other_ip_rcvd); }

};

#endif