implement kallsyms

include/lo2s/mmap.hpp

@@ -32,6 +32,7 @@
 #include <deque>
 #include <map>
 #include <mutex>
+#include <regex>
 #include <sstream>
 #include <stdexcept>
 
@@ -140,6 +141,88 @@ class BfdRadareBinary : public Binary
 #endif // HAVE_RADARE
 };
 
+class Kallsyms : public Binary
+{
+public:
+    Kallsyms() : Binary("kernel")
+    {
+        std::map<Address, std::string> entries;
+        std::ifstream ksyms_file("/proc/kallsyms");
+
+        std::regex ksym_regex("([0-9a-f]+) (?:t|T) ([^[:space:]]+).+");
+        std::smatch ksym_match;
+
+        std::string line;
+
+        while (getline(ksyms_file, line))
+        {
+            if (std::regex_match(line, ksym_match, ksym_regex))
+            {
+                std::string sym_str = ksym_match[2];
+                uint64_t sym_addr = stoull(ksym_match[1], nullptr, 16);
+                entries.emplace(std::piecewise_construct, std::forward_as_tuple(sym_addr),
+                                std::forward_as_tuple(sym_str));
+            }
+        }
+
+        std::string sym_str = "";
+        Address prev = 0;
+
+        for (auto& entry : entries)
+        {
+            if (prev != 0 && prev != entry.first)
+            {
+                kallsyms_.emplace_back(
+                    std::pair<Range, std::string>(Range(prev, entry.first), sym_str));
+            }
+            else
+            {
+                start_ = entry.first.value();
+            }
+            sym_str = entry.second;
+            prev = entry.first;
+        }
+
+        if (sym_str != "")
+        {
+            kallsyms_.emplace_back(
+                std::pair<Range, std::string>(Range(prev, Address((uint64_t)-1)), sym_str));
+        }
+    }
+
+    static Kallsyms& cache()
+    {
+        static Kallsyms k;
+        return k;
+    }
+
+    uint64_t start()
+    {
+        return start_;
+    }
+
+    virtual std::string lookup_instruction(Address) override
+    {
+        throw std::domain_error("Unknown instruction.");
+    }
+
+    virtual LineInfo lookup_line_info(Address addr) override
+    {
+        for (auto& entry : kallsyms_)
+        {
+            if (addr + start_ >= entry.first.start && addr + start_ < entry.first.end)
+            {
+                return LineInfo::for_function("kernel", entry.second.c_str(), 1, "");
+            }
+        }
+        return LineInfo::for_unknown_function();
+    }
+
+private:
+    std::vector<std::pair<Range, std::string>> kallsyms_;
+    uint64_t start_;
+};
+
 struct RecordMmapType
 {
     // BAD things happen if you try this

src/mmap.cpp

@@ -40,6 +40,10 @@ MemoryMap::MemoryMap()
 
 MemoryMap::MemoryMap(Process process, bool read_initial)
 {
+    Binary* kall = &Kallsyms::cache();
+    auto ret = map_.emplace(
+        std::piecewise_construct, std::forward_as_tuple(Kallsyms::cache().start(), (uint64_t)-1),
+        std::forward_as_tuple(Kallsyms::cache().start(), (uint64_t)-1, 0, *kall));
     if (!read_initial)
     {
         return;