lief-project-LIEF/examples/cpp/dwarf_editor.cpp

#include <LIEF/DWARF.hpp>
#include <LIEF/logging.hpp>
#include <LIEF/PE.hpp>
#include <LIEF/utils.hpp>

#include <cstdlib>
#include <iostream>

using namespace LIEF::logging;

int main(int argc, const char** argv) {
  if (!LIEF::is_extended()) {
    std::cerr << "This example requires the extended version of LIEF\n";
    return EXIT_FAILURE;
  }

  if (argc != 2) {
    std::cerr << "Usage: " << argv[0] << " <pe>\n";
    return EXIT_FAILURE;
  }

  std::unique_ptr<LIEF::PE::Binary> pe = LIEF::PE::Parser::parse(argv[1]);
  std::unique_ptr<LIEF::dwarf::Editor> editor = LIEF::dwarf::Editor::from_binary(*pe);

  std::unique_ptr<LIEF::dwarf::editor::CompilationUnit> unit = editor->create_compilation_unit();
  unit->set_producer("LIEF");

  std::unique_ptr<LIEF::dwarf::editor::Function> func = unit->create_function("hello");
  func->set_address(0x123);

  func->set_return_type(
    *unit->create_structure("my_struct_t")->pointer_to()
  );

  std::unique_ptr<LIEF::dwarf::editor::Variable> var =
    func->create_stack_variable("local_var");

  var->set_stack_offset(8);
  editor->write("/tmp/out.debug");
  return EXIT_SUCCESS;
}