feat(utility): add location utility

Author: liblaf

CMakeLists.txt

@@ -22,6 +22,7 @@ target_compile_definitions(call-hook++ PUBLIC RUNTIME_HOME="${CMAKE_SOURCE_DIR}/
 target_compile_definitions(call-hook++ PUBLIC USE_CXX)
 
 list(APPEND STRUCT_PARSER_SRCS ${BASE_SRCS} src/pass/pass_struct_parser.cc src/wrapper/wrapper_struct_parser.cc src/utility/struct_parser.cc)
+list(APPEND STRUCT_PARSER_SRCS src/utility/location.cc)
 add_executable(struct-parser   ${STRUCT_PARSER_SRCS} src/main/struct_parser.cc)
 add_executable(struct-parser++ ${STRUCT_PARSER_SRCS} src/main/struct_parser.cc)
 target_compile_definitions(struct-parser++ PUBLIC USE_CXX)

src/pass/pass_struct_parser.cc

@@ -5,6 +5,7 @@
 #include "common/logging.h"
 #include "llvm/IR/Instructions.h"
 #include "pass/pass_base.h"
+#include "utility/location.h"
 #include "utility/struct_parser.h"
 
 using namespace llvm;
@@ -32,6 +33,7 @@ bool PassStructParser::run_on_load_inst(LoadInst& inst) {
   bool modified = false;
   auto name = this->struct_parser().get_field_name_from_value(inst);
   if (name.empty()) return modified;
+  logger().info("at {}", get_inst_loc(inst));
   logger().info("{}: {}", name, inst);
   return modified;
 }
@@ -40,6 +42,7 @@ bool PassStructParser::run_on_get_element_ptr_inst(GetElementPtrInst& inst) {
   bool modified = false;
   auto name = this->struct_parser().get_field_name_from_address(inst);
   if (name.empty()) return modified;
+  logger().info("at {}", get_inst_loc(inst));
   logger().info("{}: {}", name, inst);
   return modified;
 }

src/utility/location.cc

@@ -0,0 +1,32 @@
+#include "location.h"
+
+#include <sstream>
+#include <string>
+
+#include "llvm/IR/DebugInfoMetadata.h"
+#include "llvm/IR/DebugLoc.h"
+#include "llvm/IR/Function.h"
+#include "llvm/IR/Instruction.h"
+#include "llvm/IR/Module.h"
+#include "llvm/Support/raw_ostream.h"
+
+using namespace llvm;
+
+std::string get_inst_loc(Instruction& inst) {
+  std::string str;
+  auto oss = raw_string_ostream(str);
+  auto debug_loc = inst.getDebugLoc();
+  if (debug_loc) {
+    int line = debug_loc.getLine();
+    int col = debug_loc.getCol();
+    auto dir = debug_loc->getDirectory();
+    auto file = debug_loc->getFilename();
+    oss << dir.str() << "/" << file.str() << ":" << line << ":" << col;
+  } else {
+    auto mod = inst.getModule()->getName();
+    auto func = inst.getFunction()->getName();
+    auto opcode = inst.getOpcodeName();
+    oss << mod.str() << ":" << func.str() << ":" << opcode;
+  }
+  return oss.str();
+}

src/utility/location.h

@@ -0,0 +1,10 @@
+#ifndef LOCATION_H_
+#define LOCATION_H_
+
+#include <string>
+
+#include "llvm/IR/Instruction.h"
+
+std::string get_inst_loc(llvm::Instruction& inst);
+
+#endif  // LOCATION_H_