Loading...
gen/backtrace.c /dev/null Libc-1669.40.2
--- /dev/null
+++ Libc/Libc-1669.40.2/gen/backtrace.c
@@ -0,0 +1,247 @@
+/*
+ * Copyright (c) 2007, 2008 Apple Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+#include <TargetConditionals.h>
+#include <mach/vm_types.h>
+#include <sys/uio.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"
+
+#if __LP64__
+#define _BACKTRACE_FORMAT "%-4d%-35s 0x%016lx %s + %lu"
+#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 /* %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 addrbuf[_BACKTRACE_ADDRESS_LEN + 1];
+	char imagebuf[_BACKTRACE_IMAGE_LEN + 1];
+	uuid_string_t uuidbuf;
+	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) {
+			strlcpy(imagebuf, info->dli_fname, sizeof(imagebuf));
+		} else {
+			strlcpy(imagebuf, tmp + 1, sizeof(imagebuf));
+		}
+		image = imagebuf;
+	}
+
+	if (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 {
+		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) {
+	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;
+
+	// 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;
+		total_bytes += symbol_length(&info[i]);
+	}
+
+	result = (char**)malloc(total_bytes);
+	if (result == 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 += 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) {
+	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);
+	}
+}
+
+#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;
+	};
+}