#include <mach/vm_types.h>
#include <sys/uio.h>
#include <dlfcn.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "stack_logging.h"
#include "execinfo.h"
int backtrace(void** buffer, int size) {
extern void _thread_stack_pcs(vm_address_t *buffer, unsigned max, unsigned *nb, unsigned skip);
unsigned int num_frames;
_thread_stack_pcs((vm_address_t*)buffer, size, &num_frames, 1);
while (num_frames >= 1 && buffer[num_frames-1] == NULL) num_frames -= 1;
return num_frames;
}
#if __LP64__
#define _BACKTRACE_FORMAT "%-4d%-35s 0x%016lx %s + %lu"
#define _BACKTRACE_FORMAT_SIZE 83
#define _BACKTRACE_ADDRESS_LEN 18
#else
#define _BACKTRACE_FORMAT "%-4d%-35s 0x%08lx %s + %lu"
#define _BACKTRACE_FORMAT_SIZE 65
#define _BACKTRACE_ADDRESS_LEN 10
#endif
static int _backtrace_snprintf(char* buf, size_t size, int frame, const void* addr, const Dl_info* info) {
char symbuf[_BACKTRACE_ADDRESS_LEN + 1];
const char* image = "???";
const char* symbol = "0x0";
uintptr_t symbol_offset = 0;
if (info->dli_fname) {
const char *tmp = strrchr(info->dli_fname, '/');
if(tmp == NULL)
image = info->dli_fname;
else
image = tmp + 1;
}
if (info->dli_sname) {
symbol = info->dli_sname;
symbol_offset = (uintptr_t)addr - (uintptr_t)info->dli_saddr;
} else if(info->dli_fname) {
symbol = image;
symbol_offset = (uintptr_t)addr - (uintptr_t)info->dli_fbase;
} else if(0 < snprintf(symbuf, sizeof(symbuf), "0x%lx", (uintptr_t)info->dli_saddr)) {
symbol = symbuf;
symbol_offset = (uintptr_t)addr - (uintptr_t)info->dli_saddr;
} else {
symbol_offset = (uintptr_t)addr;
}
return snprintf(buf, size,
_BACKTRACE_FORMAT,
frame,
image,
(uintptr_t)addr,
symbol,
symbol_offset);
}
char** backtrace_symbols(void* const* buffer, int size) {
int i;
size_t total_bytes;
char** result;
char** ptrs;
intptr_t strs, end;
Dl_info* info = calloc(size, sizeof (Dl_info));
if (info == NULL) return NULL;
total_bytes = sizeof(char*) * size;
for (i = 0 ; i < size; ++i) {
dladdr(buffer[i], &info[i]);
total_bytes += _BACKTRACE_FORMAT_SIZE;
if (info[i].dli_sname) {
total_bytes += strlen(info[i].dli_sname);
} else if(info[i].dli_fname) {
const char *tmp = strrchr(info[i].dli_fname, '/');
if(tmp == NULL)
total_bytes += strlen(info[i].dli_fname);
else
total_bytes += strlen(tmp + 1);
} else {
total_bytes += _BACKTRACE_ADDRESS_LEN;
}
}
result = (char**)malloc(total_bytes);
if (result == NULL) {
free(info);
return NULL;
}
end = (intptr_t)result + total_bytes;
ptrs = result;
strs = ((intptr_t)result) + sizeof(char*) * size;
for (i = 0; i < size; ++i) {
int chk = _backtrace_snprintf((char*)strs, end - (intptr_t)strs, i, buffer[i], &info[i]);
if(chk < 0) {
free(info);
return NULL;
}
ptrs[i] = (char*)strs;
strs += chk + 1; }
free(info);
return result;
}
void backtrace_symbols_fd(void* const* buffer, int size, int fd) {
int i;
char buf[BUFSIZ];
Dl_info info;
struct iovec iov[2];
iov[0].iov_base = buf;
iov[1].iov_base = "\n";
iov[1].iov_len = 1;
for (i = 0; i < size; ++i) {
memset(&info, 0, sizeof(info));
dladdr(buffer[i], &info);
iov[0].iov_len = _backtrace_snprintf(buf, sizeof(buf), i, buffer[i], &info);
writev(fd, iov, 2);
}
}