First release of open-appsec source code

components/security_apps/waap/waap_clib/TuningDecisions.h

@@ -0,0 +1,89 @@
+// Copyright (C) 2022 Check Point Software Technologies Ltd. All rights reserved.
+
+// Licensed under the Apache License, Version 2.0 (the "License");
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifndef __TUNING_DECISIONS_H__
+#define __TUNING_DECISIONS_H__
+
+#include <string>
+#include <map>
+#include "i_messaging.h"
+#include "i_agent_details.h"
+#include "waap.h"
+
+enum TuningDecisionEnum
+{
+    NO_DECISION,
+    DISMISS = NO_DECISION,
+    BENIGN,
+    MALICIOUS
+};
+
+enum TuningDecisionType
+{
+    UNKNOWN,
+    SOURCE,
+    URL,
+    PARAM_NAME,
+    PARAM_VALUE
+};
+
+
+class TuningDecision
+{
+public:
+    TuningDecision(const std::string& remotePath);
+    ~TuningDecision();
+
+    TuningDecisionEnum getDecision(std::string tuningValue, TuningDecisionType tuningType);
+private:
+    void updateDecisions();
+    TuningDecisionType convertDecisionType(std::string decisionTypeStr);
+    TuningDecisionEnum convertDecision(std::string decisionStr);
+
+
+    template<typename T>
+    bool sendObject(T &obj, I_Messaging::Method method, std::string uri)
+    {
+        I_Messaging *messaging = Singleton::Consume<I_Messaging>::by<WaapComponent>();
+        I_AgentDetails *agentDetails = Singleton::Consume<I_AgentDetails>::by<WaapComponent>();
+        if (agentDetails->getOrchestrationMode() != OrchestrationMode::ONLINE) {
+            Flags <MessageConnConfig> conn_flags;
+            conn_flags.setFlag(MessageConnConfig::EXTERNAL);
+            std::string tenant_header = "X-Tenant-Id: " + agentDetails->getTenantId();
+
+            return messaging->sendObject(
+                obj,
+                method,
+                "fog-msrv-appsec-shared-files-svc",
+                80,
+                conn_flags,
+                uri,
+                tenant_header,
+                nullptr,
+                MessageTypeTag::WAAP_LEARNING);
+        }
+        return messaging->sendObject(
+            obj,
+            method,
+            uri,
+            "",
+            nullptr,
+            true,
+            MessageTypeTag::WAAP_LEARNING);
+    }
+
+    std::string m_remotePath;
+    std::map<TuningDecisionType, std::map<std::string, TuningDecisionEnum>> m_decisions;
+};
+
+#endif