initial commit

1 files changed, 86 insertions, 0 deletions

diff --git a/utils/pdb_descr.hpp b/utils/pdb_descr.hpp
new file mode 100644
index 0000000..c57a0e9
--- /dev/null
+++ b/utils/pdb_descr.hpp
@@ -0,0 +1,86 @@
+// Copyright (c) 2017 Egor Tensin <Egor.Tensin@gmail.com>
+// This file is part of the "PDB repository" project.
+// For details, see https://github.com/egor-tensin/pdb-repo.
+// Distributed under the MIT License.
+
+#pragma once
+
+#include "pdb/all.hpp"
+
+#include <boost/program_options.hpp>
+
+#include <sstream>
+#include <string>
+#include <vector>
+
+namespace
+{
+    struct PDB
+    {
+        PDB(pdb::Address online_base, const std::string& path)
+            : online_base{online_base}
+            , path{path}
+        { }
+
+        pdb::Address online_base;
+        std::string path;
+
+        static PDB parse(std::string src)
+        {
+            static constexpr auto sep = ',';
+            const auto sep_pos = src.find(sep);
+            if (sep_pos == std::string::npos)
+                boost::throw_exception(boost::program_options::invalid_option_value{src});
+            pdb::Address online_base;
+            if (!parse_address(online_base, src.substr(0, sep_pos)))
+                boost::throw_exception(boost::program_options::invalid_option_value{src});
+            return {online_base, src.substr(sep_pos + 1)};
+        }
+
+        static bool parse_address(pdb::Address& dest, const std::string& src)
+        {
+            std::istringstream iss{src};
+            iss >> std::hex;
+            char c;
+            return iss >> dest && !iss.get(c);
+        }
+
+        static pdb::Address parse_address(const std::string& src)
+        {
+            pdb::Address dest;
+            if (!parse_address(dest, src))
+                boost::throw_exception(boost::program_options::invalid_option_value{src});
+            return dest;
+        }
+    };
+}
+
+namespace boost
+{
+    namespace program_options
+    {
+        template <typename charT>
+        void validate(
+            boost::any& dest,
+            const std::vector<std::basic_string<charT>>& src_tokens,
+            PDB*,
+            int)
+        {
+            validators::check_first_occurrence(dest);
+            const auto& src_token = validators::get_single_string(src_tokens);
+            dest = any{PDB::parse(src_token)};
+        }
+
+        template <typename charT>
+        void validate(
+            boost::any& dest,
+            const std::vector<std::basic_string<charT>>& src_tokens,
+            pdb::Address*,
+            int)
+        {
+            validators::check_first_occurrence(dest);
+            const auto& src_token = validators::get_single_string(src_tokens);
+            dest = any{PDB::parse_address(src_token)};
+        }
+    }
+}