First release of open-appsec source code

core/environment/context.cc

@@ -0,0 +1,101 @@
+// 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.
+
+#include "context.h"
+#include "i_environment.h"
+#include "singleton.h"
+
+using namespace std;
+
+void
+Context::activate()
+{
+    Singleton::Consume<I_Environment>::by<Context>()->registerContext(this);
+}
+
+void
+Context::deactivate()
+{
+    Singleton::Consume<I_Environment>::by<Context>()->unregisterContext(this);
+}
+
+map<string, string>
+Context::getAllStrings(const EnvKeyAttr::ParamAttr &param) const
+{
+    map<string, string> result;
+    for (auto &entry : values) {
+        if (entry.first.doesMatch(param)) {
+            auto entry_value = entry.second->getString();
+            if (entry_value.ok()) result[entry.first.first] = *entry_value;
+        }
+    }
+    return result;
+}
+
+const std::string
+Context::convertToString(MetaDataType type)
+{
+    switch (type) {
+        case MetaDataType::File: return "file";
+        case MetaDataType::SubjectIpAddr: return "subjectIp";
+        case MetaDataType::OtherIpAddr: return "otherIp";
+        case MetaDataType::Port: return "port";
+        case MetaDataType::Protocol: return "protocol";
+        case MetaDataType::Service: return "service";
+        case MetaDataType::User: return "user";
+        case MetaDataType::Domain: return "domain";
+        case MetaDataType::Url: return "url";
+        case MetaDataType::Direction: return "direction";
+        case MetaDataType::Email: return "email";
+        case MetaDataType::COUNT:
+            dbgAssert(false) << "COUNT is not a valid meta data type";
+    }
+    dbgAssert(false) << "Reached impossible case with type=" << static_cast<int>(type);
+    return "";
+}
+
+map<string, uint64_t>
+Context::getAllUints(const EnvKeyAttr::ParamAttr &param) const
+{
+    map<string, uint64_t> result;
+    for (auto &entry : values) {
+        if (entry.first.doesMatch(param)) {
+            auto entry_value = entry.second->getUint();
+            if (entry_value.ok()) result[entry.first.first] = *entry_value;
+        }
+    }
+    return result;
+}
+
+map<string, bool>
+Context::getAllBools(const EnvKeyAttr::ParamAttr &param) const
+{
+    map<string, bool> result;
+    for (auto &entry : values) {
+        if (entry.first.doesMatch(param)) {
+            auto entry_value = entry.second->getBool();
+            if (entry_value.ok()) result[entry.first.first] = *entry_value;
+        }
+    }
+    return result;
+}
+
+ScopedContext::ScopedContext()
+{
+    activate();
+}
+
+ScopedContext::~ScopedContext()
+{
+    deactivate();
+}