Fight Optional

[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>
  */
 
-#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
+}