First release of open-appsec source code

components/include/generic_rulebase/assets_config.h

@@ -0,0 +1,91 @@
+// 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 __ASSETS_CONFIG_H__
+#define __ASSETS_CONFIG_H__
+
+#include <vector>
+#include <string>
+#include <arpa/inet.h>
+
+#include "generic_rulebase_context.h"
+#include "c_common/ip_common.h"
+#include "connkey.h"
+
+class RuleAsset
+{
+public:
+    class AssetUrl
+    {
+    public:
+        void
+        load(cereal::JSONInputArchive &archive_in);
+        const std::string & getProtocol() const { return protocol; }
+        const std::string & getIp() const { return ip; }
+        const std::string & getPort() const { return port; }
+
+        uint8_t getParsedProtocol() const { return parsed_proto; }
+        const IpAddress & getParsedIp() const { return parsed_ip; }
+        uint16_t getParsedPort() const { return parsed_port; }
+        bool isAnyIp() const { return is_any_ip; }
+        bool isAnyPort() const { return is_any_port; }
+        bool isAnyProto() const { return is_any_proto; }
+
+    private:
+        static IpAddress ConvertToIpAddress(const IPAddr &addr);
+
+        std::string protocol;
+        std::string ip;
+        std::string port;
+        IpAddress parsed_ip;
+        uint16_t parsed_port;
+        uint8_t parsed_proto;
+        bool is_any_ip;
+        bool is_any_port;
+        bool is_any_proto;
+    };
+
+    void load(cereal::JSONInputArchive &archive_in);
+
+    const GenericConfigId & getId() const { return asset_id; }
+    const std::string & getName() const { return asset_name; }
+    const std::vector<AssetUrl> & getUrls() const { return asset_urls; }
+    
+private:
+    GenericConfigId asset_id;
+    std::string asset_name;
+    std::vector<AssetUrl> asset_urls;
+};
+
+class Assets
+{
+public:
+    static void preload();
+
+    void load(cereal::JSONInputArchive &archive_in)
+    {
+        try {
+            cereal::load(archive_in, assets);
+        }catch (const cereal::Exception &) {
+        }
+    }
+
+    static const Assets empty_assets_config;
+
+    const std::vector<RuleAsset> & getAssets() const { return assets; }
+
+private:
+    std::vector<RuleAsset> assets;
+};
+
+#endif //__ASSETS_CONFIG_H__