X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/e5499c61083684b28bcbc6950aae66cbf0938703..e9ae1f101b5418c0ba2e6c9d86b23c12f0140982:/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
+}