| 0,0 → 1,382 |
|---|
| /* |
| * Copyright (c) 2008 Jiri Svoboda |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * - Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * - Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * - The name of the author may not be used to endorse or promote products |
| * derived from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
| * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
| * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
| * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
| * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| /** @addtogroup loader |
| * @brief Loads and runs programs from VFS. |
| * @{ |
| */ |
| /** |
| * @file |
| * @brief Loads and runs programs from VFS. |
| * |
| * The program loader is a special init binary. Its image is used |
| * to create a new task upon a @c task_spawn syscall. The syscall |
| * returns the id of a phone connected to the newly created task. |
| * |
| * The caller uses this phone to send the pathname and various other |
| * information to the loader. This is normally done by the C library |
| * and completely hidden from applications. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <unistd.h> |
| #include <bool.h> |
| #include <fcntl.h> |
| #include <sys/types.h> |
| #include <ipc/ipc.h> |
| #include <ipc/services.h> |
| #include <ipc/loader.h> |
| #include <loader/pcb.h> |
| #include <console.h> |
| #include <errno.h> |
| #include <async.h> |
| #include <as.h> |
| |
| #include <elf.h> |
| #include <elf_load.h> |
| |
| #define DPRINTF(...) |
| |
| /** Pathname of the file that will be loaded */ |
| static char *pathname = NULL; |
| |
| /** The Program control block */ |
| static pcb_t pcb; |
| |
| /** Number of arguments */ |
| static int argc = 0; |
| /** Argument vector */ |
| static char **argv = NULL; |
| /** Buffer holding all arguments */ |
| static char *arg_buf = NULL; |
| |
| static elf_info_t prog_info; |
| static elf_info_t interp_info; |
| |
| static bool is_dyn_linked; |
| |
| /** Used to limit number of connections to one. */ |
| static bool connected; |
| |
| static void loader_get_taskid(ipc_callid_t rid, ipc_call_t *request) |
| { |
| ipc_callid_t callid; |
| task_id_t task_id; |
| size_t len; |
| |
| task_id = task_get_id(); |
| |
| if (!ipc_data_read_receive(&callid, &len)) { |
| ipc_answer_0(callid, EINVAL); |
| ipc_answer_0(rid, EINVAL); |
| return; |
| } |
| |
| if (len > sizeof(task_id)) |
| len = sizeof(task_id); |
| |
| ipc_data_read_finalize(callid, &task_id, len); |
| ipc_answer_0(rid, EOK); |
| } |
| |
| |
| /** Receive a call setting pathname of the program to execute. |
| * |
| * @param rid |
| * @param request |
| */ |
| static void loader_set_pathname(ipc_callid_t rid, ipc_call_t *request) |
| { |
| ipc_callid_t callid; |
| size_t len; |
| char *name_buf; |
| |
| if (!ipc_data_write_receive(&callid, &len)) { |
| ipc_answer_0(callid, EINVAL); |
| ipc_answer_0(rid, EINVAL); |
| return; |
| } |
| |
| name_buf = malloc(len + 1); |
| if (!name_buf) { |
| ipc_answer_0(callid, ENOMEM); |
| ipc_answer_0(rid, ENOMEM); |
| return; |
| } |
| |
| ipc_data_write_finalize(callid, name_buf, len); |
| ipc_answer_0(rid, EOK); |
| |
| if (pathname != NULL) { |
| free(pathname); |
| pathname = NULL; |
| } |
| |
| name_buf[len] = '\0'; |
| pathname = name_buf; |
| } |
| |
| /** Receive a call setting arguments of the program to execute. |
| * |
| * @param rid |
| * @param request |
| */ |
| static void loader_set_args(ipc_callid_t rid, ipc_call_t *request) |
| { |
| ipc_callid_t callid; |
| size_t buf_size, arg_size; |
| char *p; |
| int n; |
| |
| if (!ipc_data_write_receive(&callid, &buf_size)) { |
| ipc_answer_0(callid, EINVAL); |
| ipc_answer_0(rid, EINVAL); |
| return; |
| } |
| |
| if (arg_buf != NULL) { |
| free(arg_buf); |
| arg_buf = NULL; |
| } |
| |
| if (argv != NULL) { |
| free(argv); |
| argv = NULL; |
| } |
| |
| arg_buf = malloc(buf_size + 1); |
| if (!arg_buf) { |
| ipc_answer_0(callid, ENOMEM); |
| ipc_answer_0(rid, ENOMEM); |
| return; |
| } |
| |
| ipc_data_write_finalize(callid, arg_buf, buf_size); |
| |
| arg_buf[buf_size] = '\0'; |
| |
| /* |
| * Count number of arguments |
| */ |
| p = arg_buf; |
| n = 0; |
| while (p < arg_buf + buf_size) { |
| arg_size = str_size(p); |
| p = p + arg_size + 1; |
| ++n; |
| } |
| |
| /* Allocate argv */ |
| argv = malloc((n + 1) * sizeof(char *)); |
| |
| if (argv == NULL) { |
| free(arg_buf); |
| ipc_answer_0(rid, ENOMEM); |
| return; |
| } |
| |
| /* |
| * Fill argv with argument pointers |
| */ |
| p = arg_buf; |
| n = 0; |
| while (p < arg_buf + buf_size) { |
| argv[n] = p; |
| |
| arg_size = str_size(p); |
| p = p + arg_size + 1; |
| ++n; |
| } |
| |
| argc = n; |
| argv[n] = NULL; |
| |
| ipc_answer_0(rid, EOK); |
| } |
| |
| /** Load the previously selected program. |
| * |
| * @param rid |
| * @param request |
| * @return 0 on success, !0 on error. |
| */ |
| static int loader_load(ipc_callid_t rid, ipc_call_t *request) |
| { |
| int rc; |
| |
| rc = elf_load_file(pathname, 0, &prog_info); |
| if (rc < 0) { |
| DPRINTF("Failed to load executable '%s'.\n", pathname); |
| ipc_answer_0(rid, EINVAL); |
| return 1; |
| } |
| |
| elf_create_pcb(&prog_info, &pcb); |
| |
| pcb.argc = argc; |
| pcb.argv = argv; |
| |
| if (prog_info.interp == NULL) { |
| /* Statically linked program */ |
| is_dyn_linked = false; |
| ipc_answer_0(rid, EOK); |
| return 0; |
| } |
| |
| rc = elf_load_file(prog_info.interp, 0, &interp_info); |
| if (rc < 0) { |
| DPRINTF("Failed to load interpreter '%s.'\n", |
| prog_info.interp); |
| ipc_answer_0(rid, EINVAL); |
| return 1; |
| } |
| |
| is_dyn_linked = true; |
| ipc_answer_0(rid, EOK); |
| |
| return 0; |
| } |
| |
| |
| /** Run the previously loaded program. |
| * |
| * @param rid |
| * @param request |
| * @return 0 on success, !0 on error. |
| */ |
| static void loader_run(ipc_callid_t rid, ipc_call_t *request) |
| { |
| const char *cp; |
| |
| /* Set the task name. */ |
| cp = str_rchr(pathname, '/'); |
| cp = (cp == NULL) ? pathname : (cp + 1); |
| task_set_name(cp); |
| |
| if (is_dyn_linked == true) { |
| /* Dynamically linked program */ |
| DPRINTF("Run ELF interpreter.\n"); |
| DPRINTF("Entry point: 0x%lx\n", interp_info.entry); |
| console_close(); |
| |
| ipc_answer_0(rid, EOK); |
| elf_run(&interp_info, &pcb); |
| } else { |
| /* Statically linked program */ |
| console_close(); |
| ipc_answer_0(rid, EOK); |
| elf_run(&prog_info, &pcb); |
| } |
| |
| /* Not reached */ |
| } |
| |
| /** Handle loader connection. |
| * |
| * Receive and carry out commands (of which the last one should be |
| * to execute the loaded program). |
| */ |
| static void loader_connection(ipc_callid_t iid, ipc_call_t *icall) |
| { |
| ipc_callid_t callid; |
| ipc_call_t call; |
| int retval; |
| |
| /* Already have a connection? */ |
| if (connected) { |
| ipc_answer_0(iid, ELIMIT); |
| return; |
| } |
| |
| connected = true; |
| |
| /* Accept the connection */ |
| ipc_answer_0(iid, EOK); |
| |
| /* Ignore parameters, the connection is already open */ |
| (void) iid; |
| (void) icall; |
| |
| while (1) { |
| callid = async_get_call(&call); |
| |
| switch (IPC_GET_METHOD(call)) { |
| case IPC_M_PHONE_HUNGUP: |
| exit(0); |
| case LOADER_GET_TASKID: |
| loader_get_taskid(callid, &call); |
| continue; |
| case LOADER_SET_PATHNAME: |
| loader_set_pathname(callid, &call); |
| continue; |
| case LOADER_SET_ARGS: |
| loader_set_args(callid, &call); |
| continue; |
| case LOADER_LOAD: |
| loader_load(callid, &call); |
| continue; |
| case LOADER_RUN: |
| loader_run(callid, &call); |
| /* Not reached */ |
| default: |
| retval = ENOENT; |
| break; |
| } |
| if ((callid & IPC_CALLID_NOTIFICATION) == 0 && |
| IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP) { |
| DPRINTF("Responding EINVAL to method %d.\n", |
| IPC_GET_METHOD(call)); |
| ipc_answer_0(callid, EINVAL); |
| } |
| } |
| } |
| |
| /** Program loader main function. |
| */ |
| int main(int argc, char *argv[]) |
| { |
| ipcarg_t phonead; |
| |
| connected = false; |
| |
| /* Set a handler of incomming connections. */ |
| async_set_client_connection(loader_connection); |
| |
| /* Register at naming service. */ |
| if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0) |
| return -1; |
| |
| async_manager(); |
| |
| /* Never reached */ |
| return 0; |
| } |
| |
| /** @} |
| */ |