utils/ -> src/utils/

1 files changed, 99 insertions, 0 deletions

diff --git a/src/utils/enum_symbols.cpp b/src/utils/enum_symbols.cpp
new file mode 100644
index 0000000..645bc13
--- /dev/null
+++ b/src/utils/enum_symbols.cpp
@@ -0,0 +1,99 @@
+// Copyright (c) 2017 Egor Tensin <Egor.Tensin@gmail.com>
+// This file is part of the "winapi-debug" project.
+// For details, see https://github.com/egor-tensin/winapi-debug.
+// Distributed under the MIT License.
+
+#include "command_line.hpp"
+
+#include <winapi/debug.hpp>
+
+#include <boost/program_options.hpp>
+
+#include <exception>
+#include <iostream>
+#include <string>
+#include <vector>
+
+namespace {
+
+class EnumSymbols : public SettingsParser {
+public:
+    explicit EnumSymbols(int argc, char** argv) : SettingsParser{argc, argv} {
+        namespace po = boost::program_options;
+
+        visible.add_options()("pdb",
+                              po::value<std::vector<std::string>>(&pdbs)->value_name("PATH"),
+                              "load a PDB file");
+        visible.add_options()(
+            "functions",
+            po::value<winapi::symbol::Tag>(&tag)->implicit_value(function_tag)->zero_tokens(),
+            "only list functions");
+        visible.add_options()(
+            "mask", po::value<std::string>(&symbol_mask)->value_name("MASK"), "symbol mask");
+    }
+
+    const char* get_short_description() const override {
+        return "[-h|--help] [--pdb PATH]... [--functions] [--mask MASK]";
+    }
+
+    std::vector<std::string> pdbs;
+
+    bool type_specified() const {
+        return tag != reserved_tag;
+    }
+
+    winapi::Symbol::Type get_type() const {
+        return static_cast<winapi::Symbol::Type>(tag);
+    }
+
+    std::string get_mask() const {
+        return symbol_mask;
+    }
+
+private:
+    static constexpr auto reserved_tag =
+        static_cast<winapi::symbol::Tag>(winapi::Symbol::Type::RESERVED);
+    static constexpr auto function_tag =
+        static_cast<winapi::symbol::Tag>(winapi::Symbol::Type::Function);
+
+    winapi::symbol::Tag tag = reserved_tag;
+    std::string symbol_mask{winapi::DbgHelp::all_symbols};
+};
+
+constexpr winapi::symbol::Tag EnumSymbols::reserved_tag;
+constexpr winapi::symbol::Tag EnumSymbols::function_tag;
+
+} // namespace
+
+int main(int argc, char* argv[]) {
+    try {
+        EnumSymbols settings{argc, argv};
+
+        try {
+            settings.parse(argc, argv);
+        } catch (const boost::program_options::error& e) {
+            settings.usage_error(e);
+            return 1;
+        }
+
+        if (settings.exit_with_usage) {
+            settings.usage();
+            return 0;
+        }
+
+        const auto dbghelp = winapi::DbgHelp::post_mortem();
+
+        for (const auto& pdb : settings.pdbs) {
+            const auto id = dbghelp.load_pdb(pdb);
+
+            dbghelp.enum_symbols(id, settings.get_mask(), [&](const winapi::SymbolInfo& symbol) {
+                if (!settings.type_specified() || settings.get_type() == symbol.get_type())
+                    std::cout << symbol.get_name() << '\n';
+            });
+        }
+    } catch (const std::exception& e) {
+        std::cerr << std::format("error: {}\n", e.what());
+        return 1;
+    }
+    return 0;
+}