Changes in uspace/srv/loader/main.c [27b76ca:ffa2c8ef] in mainline

File:

• uspace/srv/loader/main.c (modified) (11 diffs)

uspace/srv/loader/main.c

@@ -52 +52 @@
 #include <ipc/services.h>
 #include <ipc/loader.h>
-#include <ns.h>
+#include <ipc/ns.h>
+#include <macros.h>
 #include <loader/pcb.h>
-#include <entry_point.h>
 #include <errno.h>
 #include <async.h>
 #include <str.h>
 #include <as.h>
-#include <elf/elf.h>
-#include <elf/elf_load.h>
-#include <vfs/vfs.h>
-
-#ifdef CONFIG_RTLD
-#include <rtld/rtld.h>
-#include <rtld/dynamic.h>
-#include <rtld/module.h>
-
-static int ldr_load_dyn_linked(elf_info_t *p_info);
-#endif
+
+#include <elf.h>
+#include <elf_load.h>
 
 #define DPRINTF(...)
@@ -90 +82 @@
 
 /** Number of preset files */
-static unsigned int filc = 0;
+static int filc = 0;
+/** Preset files vector */
+static fdi_node_t **filv = NULL;
+/** Buffer holding all preset files */
+static fdi_node_t *fil_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 = false;
-
-#ifdef CONFIG_RTLD
-/** State structure of the dynamic linker. */
-runtime_env_t dload_re;
-static module_t prog_mod;
-#endif
 
 static void ldr_get_taskid(ipc_callid_t rid, ipc_call_t *request)
@@ -236 +229 @@
 static void ldr_set_files(ipc_callid_t rid, ipc_call_t *request)
 {
-        size_t count = IPC_GET_ARG1(*request);
-
-        async_exch_t *vfs_exch = vfs_exchange_begin();
-
-        for (filc = 0; filc < count; filc++) {
-                ipc_callid_t callid;
-                int fd;
-
-                if (!async_state_change_receive(&callid, NULL, NULL, NULL)) {
-                        async_answer_0(callid, EINVAL);
-                        break;
-                }
-                async_state_change_finalize(callid, vfs_exch);
-                fd = fd_wait();
-                assert(fd == (int) filc);
-        }
-
-        vfs_exchange_end(vfs_exch);
-
+        fdi_node_t *buf;
+        size_t buf_size;
+        int rc = async_data_write_accept((void **) &buf, false, 0, 0,
+            sizeof(fdi_node_t), &buf_size);
+        
+        if (rc == EOK) {
+                int count = buf_size / sizeof(fdi_node_t);
+                
+                /*
+                 * Allocate new filv
+                 */
+                fdi_node_t **_filv = (fdi_node_t **) calloc(count + 1, sizeof(fdi_node_t *));
+                if (_filv == NULL) {
+                        free(buf);
+                        async_answer_0(rid, ENOMEM);
+                        return;
+                }
+                
+                /*
+                 * Fill the new filv with argument pointers
+                 */
+                int i;
+                for (i = 0; i < count; i++)
+                        _filv[i] = &buf[i];
+                
+                _filv[count] = NULL;
+                
+                /*
+                 * Copy temporary data to global variables
+                 */
+                if (fil_buf != NULL)
+                        free(fil_buf);
+                
+                if (filv != NULL)
+                        free(filv);
+                
+                filc = count;
+                fil_buf = buf;
+                filv = _filv;
+        }
+        
         async_answer_0(rid, EOK);
 }
@@ -268 +283 @@
         int rc;
         
-        rc = elf_load_file(pathname, 0, 0, &prog_info);
+        rc = elf_load_file(pathname, 0, &prog_info);
         if (rc != EE_OK) {
                 DPRINTF("Failed to load executable '%s'.\n", pathname);
@@ -283 +298 @@
         
         pcb.filc = filc;
+        pcb.filv = filv;
         
         if (prog_info.interp == NULL) {
                 /* Statically linked program */
+                is_dyn_linked = false;
                 async_answer_0(rid, EOK);
                 return 0;
         }
         
-        DPRINTF("Binary is dynamically linked.\n");
-#ifdef CONFIG_RTLD
-        DPRINTF(" - pcb address: %p\n", &pcb);
-        DPRINTF( "- prog dynamic: %p\n", prog_info.dynamic);
-
-        rc = ldr_load_dyn_linked(&prog_info);
-#else
-        rc = ENOTSUP;
-#endif
-        async_answer_0(rid, rc);
+        rc = elf_load_file(prog_info.interp, 0, &interp_info);
+        if (rc != EE_OK) {
+                DPRINTF("Failed to load interpreter '%s.'\n",
+                    prog_info.interp);
+                async_answer_0(rid, EINVAL);
+                return 1;
+        }
+        
+        is_dyn_linked = true;
+        async_answer_0(rid, EOK);
+        
         return 0;
 }
 
-#ifdef CONFIG_RTLD
-
-static int ldr_load_dyn_linked(elf_info_t *p_info)
-{
-        runtime_env = &dload_re;
-
-        DPRINTF("Load dynamically linked program.\n");
-
-        /*
-         * First we need to process dynamic sections of the executable
-         * program and insert it into the module graph.
-         */
-
-        DPRINTF("Parse program .dynamic section at %p\n", p_info->dynamic);
-        dynamic_parse(p_info->dynamic, 0, &prog_mod.dyn);
-        prog_mod.bias = 0;
-        prog_mod.dyn.soname = "[program]";
-
-        /* Initialize list of loaded modules */
-        list_initialize(&runtime_env->modules);
-        list_append(&prog_mod.modules_link, &runtime_env->modules);
-
-        /* Pointer to program module. Used as root of the module graph. */
-        runtime_env->program = &prog_mod;
-
-        /* Work around non-existent memory space allocation. */
-        runtime_env->next_bias = 0x1000000;
-
-        /*
-         * Now we can continue with loading all other modules.
-         */
-
-        DPRINTF("Load all program dependencies\n");
-        module_load_deps(&prog_mod);
-
-        /*
-         * Now relocate/link all modules together.
-         */
-
-        /* Process relocations in all modules */
-        DPRINTF("Relocate all modules\n");
-        modules_process_relocs(&prog_mod);
-
-        /* Pass runtime evironment pointer through PCB. */
-        pcb.rtld_runtime = (void *) runtime_env;
-
-        return 0;
-}
-#endif
 
 /** Run the previously loaded program.
@@ -363 +332 @@
         const char *cp;
         
-        DPRINTF("Set task name\n");
-
         /* Set the task name. */
         cp = str_rchr(pathname, '/');
@@ -370 +337 @@
         task_set_name(cp);
         
-        /* Run program */
-        DPRINTF("Reply OK\n");
-        async_answer_0(rid, EOK);
-        DPRINTF("Jump to entry point at %p\n", pcb.entry);
-        entry_point_jmp(prog_info.entry, &pcb);
+        if (is_dyn_linked == true) {
+                /* Dynamically linked program */
+                DPRINTF("Run ELF interpreter.\n");
+                DPRINTF("Entry point: %p\n", interp_info.entry);
+                
+                async_answer_0(rid, EOK);
+                elf_run(&interp_info, &pcb);
+        } else {
+                /* Statically linked program */
+                async_answer_0(rid, EOK);
+                elf_run(&prog_info, &pcb);
+        }
         
         /* Not reached */
@@ -384 +358 @@
  * to execute the loaded program).
  */
-static void ldr_connection(ipc_callid_t iid, ipc_call_t *icall, void *arg)
-{
+static void ldr_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) {
@@ -398 +376 @@
         
         /* Ignore parameters, the connection is already open */
+        (void) iid;
         (void) icall;
         
-        while (true) {
-                int retval;
-                ipc_call_t call;
-                ipc_callid_t callid = async_get_call(&call);
-                
-                if (!IPC_GET_IMETHOD(call))
+        while (1) {
+                callid = async_get_call(&call);
+                
+                switch (IPC_GET_IMETHOD(call)) {
+                case IPC_M_PHONE_HUNGUP:
                         exit(0);
-                
-                switch (IPC_GET_IMETHOD(call)) {
                 case LOADER_GET_TASKID:
                         ldr_get_taskid(callid, &call);
@@ -431 +407 @@
                         /* Not reached */
                 default:
-                        retval = EINVAL;
+                        retval = ENOENT;
                         break;
                 }
-                
-                async_answer_0(callid, retval);
+                if (IPC_GET_IMETHOD(call) != IPC_M_PHONE_HUNGUP) {
+                        DPRINTF("Responding EINVAL to method %d.\n",
+                            IPC_GET_IMETHOD(call));
+                        async_answer_0(callid, EINVAL);
+                }
         }
 }
@@ -448 +427 @@
         /* Introduce this task to the NS (give it our task ID). */
         task_id_t id = task_get_id();
-        int rc = ns_intro(id);
+        int rc = async_req_2_0(PHONE_NS, NS_ID_INTRO, LOWER32(id), UPPER32(id));
         if (rc != EOK)
                 return -1;