]> git.stg.codes - stg.git/commitdiff
Add rules count method to the traffcounter iface
authorMaxim Mamontov <faust.madf@gmail.com>
Wed, 31 Aug 2011 10:01:02 +0000 (13:01 +0300)
committerMaxim Mamontov <faust.madf@gmail.com>
Wed, 31 Aug 2011 10:01:02 +0000 (13:01 +0300)
include/stg/traffcounter.h
projects/stargazer/traffcounter_impl.h

index 1962ee7a2bf07e979dcbef82aa045a1340e85e7d..75c8c481c3ec3f06ec448e2c36a4927dcf4e4370 100644 (file)
@@ -26,6 +26,7 @@
 class TRAFFCOUNTER {
 public:
     virtual void Process(const RAW_PACKET & rawPacket) = 0;
 class TRAFFCOUNTER {
 public:
     virtual void Process(const RAW_PACKET & rawPacket) = 0;
+    virtual size_t RulesCount() const = 0;
 };
 
 #endif
 };
 
 #endif
index 7d01c2e4b24d9b968186e515703d8e70c518a153..225a636e19594d65e5afe6d9fc5e5ea97378ad50 100644 (file)
@@ -173,6 +173,8 @@ public:
     void        Process(const RAW_PACKET & rawPacket);
     void        SetMonitorDir(const std::string & monitorDir);
 
     void        Process(const RAW_PACKET & rawPacket);
     void        SetMonitorDir(const std::string & monitorDir);
 
+    size_t      RulesCount() const { return rules.size(); }
+
 private:
     bool        ParseAddress(const char * ta, RULE * rule) const;
     uint32_t    CalcMask(uint32_t msk) const;
 private:
     bool        ParseAddress(const char * ta, RULE * rule) const;
     uint32_t    CalcMask(uint32_t msk) const;