src/mc/mc_unw_vmread.c

   1 #define _GNU_SOURCE
   2
   3 #include <sys/types.h>
   4 #include <sys/uio.h>
   5
   6 #include <libunwind.h>
   7 #include <libunwind-ptrace.h>
   8
   9 #include "mc_unw.h"
  10
  11 /** \file
  12  *  Libunwind namespace implementation using process_vm_readv.
  13  *.
  14  *  This implem
  15  */
  16
  17 /** Partial structure of libunwind-ptrace context in order to get the PID
  18  *
  19  *  The context type for libunwind-race is an opaque type. We need to get the
  20  *  PID which is the first field. This is a hack which might break if the
  21  *  libunwind-ptrace structure changes.
  22  */
  23 struct _UPT_info {
  24   pid_t pid;
  25   // Other things;
  26 };
  27
  28 /** Get the PID of a `libunwind-ptrace` context
  29  */
  30 static inline
  31 pid_t _UPT_getpid(void* arg)
  32 {
  33   struct _UPT_info* info = arg;
  34   return info->pid;
  35 }
  36
  37 /** Read from the memory, avoid using `ptrace` (libunwind method)
  38  */
  39 static int access_mem(const unw_addr_space_t as,
  40               const unw_word_t addr, unw_word_t* const  valp,
  41               const int write, void* const arg)
  42 {
  43   if (write)
  44     return - UNW_EINVAL;
  45   pid_t pid = _UPT_getpid(arg);
  46   size_t size = sizeof(unw_word_t);
  47
  48 #ifdef HAVE_PROCESS_VM_READV
  49   // process_vm_read implementation.
  50   // This is only available since Linux 3.2.
  51
  52   struct iovec local = { valp, size };
  53   struct iovec remote = { (void*) addr, size };
  54   ssize_t s = process_vm_readv(pid, &local, 1, &remote, 1, 0);
  55   if (s >= 0) {
  56     if (s != size)
  57       return - UNW_EINVAL;
  58     else
  59       return 0;
  60   }
  61   if (s < 0 && errno != ENOSYS)
  62     return - UNW_EINVAL;
  63 #endif
  64
  65   // /proc/${pid}/mem implementation.
  66   // On recent kernels, we do not need to ptrace the target process.
  67   // On older kernels, it is necessary to ptrace the target process.
  68   size_t count = size;
  69   off_t off = (off_t) addr;
  70   char* buf = (char*) valp;
  71   int fd = MC_process_vm_open(pid, O_RDONLY);
  72   if (fd < 0)
  73     return - UNW_EINVAL;
  74   while (1) {
  75     ssize_t s = pread(fd, buf, count, off);
  76     if (s == 0) {
  77       close(fd);
  78       return - UNW_EINVAL;
  79     }
  80     if (s == -1)
  81       break;
  82     count -= s;
  83     buf += s;
  84     off += s;
  85     if (count == 0) {
  86       close(fd);
  87       return 0;
  88     }
  89   }
  90   close(fd);
  91
  92   // ptrace implementation.
  93   // We need to have PTRACE_ATTACH-ed it before.
  94   return _UPT_access_mem(as, addr, valp, write, arg);
  95 }
  96
  97 unw_accessors_t mc_unw_vmread_accessors =
  98   {
  99     .find_proc_info             = &_UPT_find_proc_info,
 100     .put_unwind_info            = &_UPT_put_unwind_info,
 101     .get_dyn_info_list_addr     = &_UPT_get_dyn_info_list_addr,
 102     .access_mem                 = &access_mem,
 103     .access_reg                 = &_UPT_access_reg,
 104     .access_fpreg               = &_UPT_access_fpreg,
 105     .resume                     = &_UPT_resume,
 106     .get_proc_name              = &_UPT_get_proc_name
 107   };