Loading...
--- Libc/Libc-583/gen/backtrace.c
+++ Libc/Libc-1669.0.4/gen/backtrace.c
@@ -21,58 +21,140 @@
* @APPLE_LICENSE_HEADER_END@
*/
+#include <TargetConditionals.h>
#include <mach/vm_types.h>
#include <sys/uio.h>
-
-#include <dlfcn.h>
+#include <errno.h>
+#include <mach-o/dyld_priv.h>
#include <stdint.h>
#include <stdlib.h>
+#include <uuid/uuid.h>
+#include "execinfo.h"
+#include "libc_private.h"
+
+extern void _thread_stack_pcs(vm_address_t *buffer, unsigned max,
+ unsigned *nb, unsigned skip, void *startfp);
+
+static backtrace_get_pcs_func_t backtrace_get_pcs_func = _thread_stack_pcs;
+
+void backtrace_set_pcs_func(backtrace_get_pcs_func_t func) {
+ backtrace_get_pcs_func = func ? func : _thread_stack_pcs;
+}
+
+int backtrace(void** buffer, int size) {
+ unsigned int num_frames;
+ backtrace_get_pcs_func((vm_address_t*)buffer, size, &num_frames, 1, NULL);
+ while (num_frames >= 1 && buffer[num_frames-1] == NULL) num_frames -= 1;
+ return num_frames;
+}
+
+extern unsigned int _thread_stack_async_pcs(vm_address_t *buffer, unsigned max,
+ unsigned *nb, unsigned skip, void *startfp);
+
+size_t backtrace_async(void** buffer, size_t size, uint32_t *task_id_ptr) {
+ unsigned int num_frames;
+ uint32_t task_id = _thread_stack_async_pcs((vm_address_t*)buffer, size, &num_frames, 1, NULL);
+ while (num_frames >= 1 && buffer[num_frames-1] == NULL) num_frames -= 1;
+ if (task_id_ptr) *task_id_ptr = task_id;
+ return num_frames;
+}
+
+
+int
+backtrace_from_fp(void *startfp, void **buffer, int size)
+{
+ unsigned int num_frames;
+ _thread_stack_pcs((vm_address_t*)buffer, size, &num_frames, 1, startfp);
+ while (num_frames >= 1 && buffer[num_frames-1] == NULL) num_frames -= 1;
+ return num_frames;
+}
+
+#if !TARGET_OS_DRIVERKIT
+
+#include <dlfcn.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 82
+#define _BACKTRACE_FORMAT_SIZE 83 /* %lu can take up to 20, does not include %s, includes NUL */
+#define _BACKTRACE_ADDRESS_LEN 18 /* 0x + 16 (no NUL) */
#else
#define _BACKTRACE_FORMAT "%-4d%-35s 0x%08lx %s + %lu"
-#define _BACKTRACE_FORMAT_SIZE 65
+#define _BACKTRACE_FORMAT_SIZE 65 /* %lu can take up to 10, does not include %s, includes NUL */
+#define _BACKTRACE_ADDRESS_LEN 10 /* 0x + 8 (no NUL) */
#endif
-
+#define _BACKTRACE_IMAGE_LEN 35
+#define _BACKTRACE_UUID_LEN 36
static int _backtrace_snprintf(char* buf, size_t size, int frame, const void* addr, const Dl_info* info) {
- char symbuf[19];
+ char addrbuf[_BACKTRACE_ADDRESS_LEN + 1];
+ char imagebuf[_BACKTRACE_IMAGE_LEN + 1];
+ uuid_string_t uuidbuf;
const char* image = "???";
- const char* symbol = symbuf;
+ const char* symbol = "0x0";
+ uintptr_t symbol_offset = 0;
if (info->dli_fname) {
- image = strrchr(info->dli_fname, '/') + 1;
- if (image == NULL) image = info->dli_fname;
- }
-
+ const char *tmp = strrchr(info->dli_fname, '/');
+ if (tmp == NULL) {
+ strlcpy(imagebuf, info->dli_fname, sizeof(imagebuf));
+ } else {
+ strlcpy(imagebuf, tmp + 1, sizeof(imagebuf));
+ }
+ image = imagebuf;
+ }
+
if (info->dli_sname) {
- symbol = info->dli_sname;
+ uuid_t uuid;
+ if (strcmp(info->dli_sname, "<redacted>") == 0 &&
+ _dyld_get_image_uuid(info->dli_fbase, uuid)) {
+ /*
+ * dyld returns <redacted> when the symbol name has been elided in
+ * the shared cache. To enable symbolication later, we provide the
+ * UUID and UUID-offset instead.
+ */
+ uuid_unparse(uuid, uuidbuf);
+ symbol = uuidbuf;
+ symbol_offset = (uintptr_t)addr - (uintptr_t)info->dli_fbase;
+ } else {
+ 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(addrbuf, sizeof(addrbuf), "0x%lx",
+ (uintptr_t)info->dli_saddr)) {
+ symbol = addrbuf;
+ symbol_offset = (uintptr_t)addr - (uintptr_t)info->dli_saddr;
} else {
- snprintf(symbuf, sizeof(symbuf), "0x%lx", (uintptr_t)info->dli_saddr);
- }
-
- return snprintf(buf, size,
- _BACKTRACE_FORMAT,
- frame,
- image,
- (uintptr_t)addr,
- symbol,
- (uintptr_t)addr - (uintptr_t)info->dli_saddr) + 1;
+ symbol_offset = (uintptr_t)addr;
+ }
+
+ return snprintf(buf, size, _BACKTRACE_FORMAT, frame, image,
+ (uintptr_t)addr, symbol, symbol_offset);
+}
+
+static size_t symbol_length(Dl_info *info)
+{
+ if (info->dli_sname) {
+ if (strcmp(info->dli_sname, "<redacted>") == 0) {
+ return _BACKTRACE_UUID_LEN;
+ } else {
+ return strlen(info->dli_sname);
+ }
+ } else if (info->dli_fname) {
+ const char *tmp = strrchr(info->dli_fname, '/');
+ if (tmp == NULL) {
+ return strlen(info->dli_fname);
+ } else {
+ return strlen(tmp + 1);
+ }
+ } else {
+ return _BACKTRACE_ADDRESS_LEN;
+ }
}
char** backtrace_symbols(void* const* buffer, int size) {
@@ -80,44 +162,53 @@
size_t total_bytes;
char** result;
char** ptrs;
- intptr_t strs;
+ intptr_t strs, end;
Dl_info* info = calloc(size, sizeof (Dl_info));
-
+
if (info == NULL) return NULL;
-
+
// Compute the total size for the block that is returned.
// The block will contain size number of pointers to the
// symbol descriptions.
total_bytes = sizeof(char*) * size;
-
+
// Plus each symbol description
for (i = 0 ; i < size; ++i) {
dladdr(buffer[i], &info[i]);
- total_bytes += _BACKTRACE_FORMAT_SIZE + 1;
- if (info[i].dli_sname) total_bytes += strlen(info[i].dli_sname);
- }
-
+ total_bytes += _BACKTRACE_FORMAT_SIZE;
+ total_bytes += symbol_length(&info[i]);
+ }
+
result = (char**)malloc(total_bytes);
if (result == NULL) {
- free(info);
- return NULL;
- }
-
+ goto error;
+ }
+ end = (intptr_t)result + total_bytes;
+
// Fill in the array of pointers and append the strings for
// each symbol description.
-
+
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) {
+ goto error;
+ }
+
ptrs[i] = (char*)strs;
- strs += _backtrace_snprintf((char*)strs, total_bytes, i, buffer[i], &info[i]);
- }
-
+ strs += chk + 1; // Step over the '\0'
+ }
+
free(info);
-
return result;
+
+error:
+ free(info);
+ free(result);
+ return NULL;
}
void backtrace_symbols_fd(void* const* buffer, int size, int fd) {
@@ -136,7 +227,21 @@
dladdr(buffer[i], &info);
iov[0].iov_len = _backtrace_snprintf(buf, sizeof(buf), i, buffer[i], &info);
-
+
writev(fd, iov, 2);
}
}
+
+#endif // !TARGET_OS_DRIVERKIT
+
+void
+backtrace_image_offsets(void* const* buffer, struct image_offset *imgoffs, int size)
+{
+ struct dyld_image_uuid_offset infos[size];
+ _dyld_images_for_addresses(size, (const void **)buffer, infos);
+
+ for (int i = 0; i < size; i++) {
+ uuid_copy(imgoffs[i].uuid, infos[i].uuid);
+ imgoffs[i].offset = infos[i].offsetInImage;
+ };
+}