User story RICPLT-2620

[ric-app/admin.git] / src / protector-plugin / NetworkProtector.h

diff --git a/src/protector-plugin/NetworkProtector.h b/src/protector-plugin/NetworkProtector.h
index e8750db..8c782e2 100644 (file)
--- a/src/protector-plugin/NetworkProtector.h
+++ b/src/protector-plugin/NetworkProtector.h
@@ -23,6 +23,7 @@
 
 #include "sliding_window.hpp"
 #include <mutex>
+#include <map>
 #include <sgnb_addition_request.hpp>  // to decode the X2AP payload
 #include <sgnb_addition_response.hpp> // to respond
 
@@ -34,39 +35,62 @@
 #define unlikely(x) (x)
 #endif
 
-extern bool report_mode_only;
+// each policy corresponds to a specific X2 subscriber  profile ID
+// and applies sliding window logic to UEs in that class (if enforce)
+class protector_policy {
+public:
+  protector_policy(bool enforce=true,  int window_size=60, int threshold=10, double block_rate=10): _enforce(enforce), _window_size(window_size), _threshold(threshold), _block_rate(block_rate){
+    _counter = 0;
+    _req = 0;
+    _rej = 0;
+    _window_ref = std::make_unique<sliding_window>(_window_size);
+  };
 
+  bool _enforce;                // do we enforce policy ?
+  int _counter;                        // count the # of attaching access
+  int _window_size;                    // time in seconds window for the # of counts
+  int _threshold;                      // count above which we start enforcing if enforce set
+  double _block_rate;          // % of rejecting rate for counter > threshold
+  std::unique_ptr<sliding_window> _window_ref;
+  unsigned long int _req; // number of requests
+  unsigned long int _rej; // number of rejects
+
+};
+
+  
 class protector
 {
 public: 
   
-  protector( bool , int , int , double , bool rep=true);
-
+  protector( bool rep=true);
+  protector(bool enforce, int window_size, int threshold, double block_rate, bool rep=true);
   bool operator()(unsigned char *, size_t , unsigned char *, size_t *);
   
-  bool configure(bool enforce, int windowSize_, int threshold_, double blockRate_);
+  bool configure(bool enforce, int windowSize_, int threshold_, double blockRate_, int id);
+  bool add_policy (bool enforce, int windowSize_, int threshold_, double blockRate_, int id);
+  bool delete_policy(int id);
+  bool query_policy(int , std::vector<double> &);
+  void get_active_policies(std::vector<int> & );
+  bool is_active(int id);
+  
   void clear();
-  bool selectiveBlock();
+  bool selectiveBlock(double);
 
-  unsigned long int get_requests(void) const;
-  unsigned long int get_rejects(void) const;
+  long int get_requests(int id) const;
+  long int get_rejects(int id) const;
   std::string get_error(void) { return error_string;};
   
 private:
-  bool m_enforce; // whether to execute logic or not 
-  int m_counter;                       // count the # of attaching access
-  int m_windowSize;                    // time in seconds window for the # of counts
-  int m_threshold;                     // count above which we start enforcing if enforce set
-  double m_blockRate;          // % of rejecting rate for counter > threshold
-  time_t m_timeWindow;      // time active window started
-  unsigned long int m_req; // number of requests
-  unsigned long int m_rej; // number of rejects
-
-  std::unique_ptr<sliding_window> m_window_ref;
+
+  std::map<int, protector_policy> policy_list;
   std::unique_ptr<std::mutex> m_access;
+
   sgnb_addition_helper sgnb_data;
   sgnb_addition_request sgnb_req;
   sgnb_addition_response sgnb_resp;
+
+  unsigned long int net_requests = 0;
+  unsigned long int net_rejects = 0;
   
   std::string error_string;
   bool report_mode_only;