Pretty printing.

[stg.git] / include / stg / traffcounter.h

diff --git a/include/stg/traffcounter.h b/include/stg/traffcounter.h
index 1962ee7a2bf07e979dcbef82aa045a1340e85e7d..42ef0d9235b55ea6cffc29ccf218b1c0cf6cb9d5 100644 (file)
--- a/include/stg/traffcounter.h
+++ b/include/stg/traffcounter.h
@@ -18,14 +18,20 @@
  *    Author : maxim Mamontov <faust@stargazer.dp.ua>
  */
 
  *    Author : maxim Mamontov <faust@stargazer.dp.ua>
  */
 

-#ifndef TRAFFCOUNTER_H
-#define TRAFFCOUNTER_H
+#pragma once

 
 

-#include "raw_ip_packet.h"
+#include <cstddef> // size_t

 
 

-class TRAFFCOUNTER {
-public:
-    virtual void Process(const RAW_PACKET & rawPacket) = 0;
+namespace STG
+{
+
+struct RawPacket;
+
+struct TraffCounter {
+    virtual ~TraffCounter() = default;
+
+    virtual void process(const RawPacket& rawPacket) = 0;
+    virtual size_t rulesCount() const = 0;

 };
 
 };
 

-#endif
+}