X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/9701b7ab4dc4cd709ad4dcaa750fc0021f15e231..8bfaedfaf22377ff1beadebff699613bf19a1ce2:/include/stg/traffcounter.h

diff --git a/include/stg/traffcounter.h b/include/stg/traffcounter.h
index 1962ee7a..42ef0d92 100644
--- a/include/stg/traffcounter.h
+++ b/include/stg/traffcounter.h
@@ -18,14 +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 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
+}