X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/57b91afcae210f9f5acd66ee5ad2908cabbf5748..c517056585ce9699f0d8b3be6423ef953e90c47b:/include/stg/traffcounter.h

diff --git a/include/stg/traffcounter.h b/include/stg/traffcounter.h
index 97d17837..42ef0d92 100644
--- a/include/stg/traffcounter.h
+++ b/include/stg/traffcounter.h
@@ -18,16 +18,20 @@
  *    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 ~TRAFFCOUNTER() {}
-    virtual void Process(const RAW_PACKET & rawPacket) = 0;
-    virtual size_t RulesCount() const = 0;
+namespace STG
+{
+
+struct RawPacket;
+
+struct TraffCounter {
+    virtual ~TraffCounter() = default;
+
+    virtual void process(const RawPacket& rawPacket) = 0;
+    virtual size_t rulesCount() const = 0;
 };
 
-#endif
+}