Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 | // SPDX-License-Identifier: BSD-2-Clause /* * Copyright (c) 2019, Linaro Limited * Copyright (c) 2022-2023, Arm Limited */ #include <assert.h> #include <ldelf.h> #include <malloc.h> #include <printk.h> #include <string.h> #include <sys/queue.h> #include <tee_api_types.h> #include <trace.h> #include <types_ext.h> #include <util.h> #include "dl.h" #include "ftrace.h" #include "sys.h" #include "ta_elf.h" static size_t mpool_size = 4 * SMALL_PAGE_SIZE; static vaddr_t mpool_base; static void __printf(2, 0) print_to_console(void *pctx __unused, const char *fmt, va_list ap) { trace_vprintf(NULL, 0, TRACE_ERROR, true, fmt, ap); } static void __noreturn __maybe_unused dump_ta_state(struct dump_entry_arg *arg) { struct ta_elf *elf = TAILQ_FIRST(&main_elf_queue); assert(elf && elf->is_main); EMSG_RAW("Status of TA %pUl", (void *)&elf->uuid); EMSG_RAW(" arch: %s", elf->is_32bit ? "arm" : "aarch64"); ta_elf_print_mappings(NULL, print_to_console, &main_elf_queue, arg->num_maps, arg->maps, mpool_base); if (arg->is_32bit) ta_elf_stack_trace_a32(arg->arm32.regs); else ta_elf_stack_trace_a64(arg->arm64.fp, arg->arm64.sp, arg->arm64.pc); sys_return_cleanup(); } #ifdef CFG_FTRACE_SUPPORT struct print_buf_ctx { char *buf; size_t blen; size_t ret; }; static void __printf(2, 0) print_to_pbuf(void *pctx, const char *fmt, va_list ap) { struct print_buf_ctx *pbuf = pctx; char *buf = NULL; size_t blen = 0; int ret = 0; if (pbuf->buf && pbuf->blen > pbuf->ret) { buf = pbuf->buf + pbuf->ret; blen = pbuf->blen - pbuf->ret; } ret = vsnprintk(buf, blen, fmt, ap); assert(ret >= 0); pbuf->ret += ret; } static void copy_to_pbuf(void *pctx, void *b, size_t bl) { struct print_buf_ctx *pbuf = pctx; char *buf = NULL; size_t blen = 0; if (pbuf->buf && pbuf->blen > pbuf->ret) { buf = pbuf->buf + pbuf->ret; blen = pbuf->blen - pbuf->ret; memcpy(buf, b, MIN(blen, bl)); } pbuf->ret += bl; } static void __noreturn ftrace_dump(void *buf, size_t *blen) { struct print_buf_ctx pbuf = { .buf = buf, .blen = *blen }; ta_elf_print_mappings(&pbuf, print_to_pbuf, &main_elf_queue, 0, NULL, mpool_base); ftrace_copy_buf(&pbuf, copy_to_pbuf); *blen = pbuf.ret; sys_return_cleanup(); } #endif static void __noreturn dl_entry(struct dl_entry_arg *arg) { switch (arg->cmd) { case LDELF_DL_ENTRY_DLOPEN: arg->ret = dlopen_entry(arg); break; case LDELF_DL_ENTRY_DLSYM: arg->ret = dlsym_entry(arg); break; default: arg->ret = TEE_ERROR_NOT_SUPPORTED; } sys_return_cleanup(); } /* * ldelf()- Loads ELF into memory * @arg: Argument passing to/from TEE Core * * Only called from assembly */ void __noreturn ldelf(struct ldelf_arg *arg); void ldelf(struct ldelf_arg *arg) { TEE_Result res = TEE_SUCCESS; struct ta_elf *elf = NULL; DMSG("Loading TS %pUl", (void *)&arg->uuid); res = sys_map_zi(mpool_size, 0, &mpool_base, 0, 0); if (res) { EMSG("sys_map_zi(%zu): result %"PRIx32, mpool_size, res); panic(); } malloc_add_pool((void *)mpool_base, mpool_size); /* Load the main binary and get a list of dependencies, if any. */ ta_elf_load_main(&arg->uuid, &arg->is_32bit, &arg->stack_ptr, &arg->flags); /* * Load binaries, ta_elf_load() may add external libraries to the * list, so the loop will end when all the dependencies are * satisfied. */ TAILQ_FOREACH(elf, &main_elf_queue, link) ta_elf_load_dependency(elf, arg->is_32bit); TAILQ_FOREACH(elf, &main_elf_queue, link) { ta_elf_relocate(elf); ta_elf_finalize_mappings(elf); } ta_elf_finalize_load_main(&arg->entry_func, &arg->load_addr); arg->ftrace_entry = 0; #ifdef CFG_FTRACE_SUPPORT if (ftrace_init(&arg->fbuf)) arg->ftrace_entry = (vaddr_t)(void *)ftrace_dump; #endif TAILQ_FOREACH(elf, &main_elf_queue, link) DMSG("ELF (%pUl) at %#"PRIxVA, (void *)&elf->uuid, elf->load_addr); #if TRACE_LEVEL >= TRACE_ERROR arg->dump_entry = (vaddr_t)(void *)dump_ta_state; #else arg->dump_entry = 0; #endif arg->dl_entry = (vaddr_t)(void *)dl_entry; sys_return_cleanup(); } |