Changes in uspace/srv/loader/main.c [153c7a29:3e6a98c5] in mainline

File:

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

uspace/srv/loader/main.c

@@ -63 +63 @@
 #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
+
 #define DPRINTF(...)
 
@@ -88 +96 @@
 /** 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)
@@ -235 +249 @@
                 }
                 async_state_change_finalize(callid, vfs_exch);
-                fd = vfs_fd_wait();
+                fd = fd_wait();
                 assert(fd == (int) filc);
         }
@@ -254 +268 @@
         int rc;
         
-        rc = elf_load(pathname, &prog_info);
+        rc = elf_load_file(pathname, 0, 0, &prog_info);
         if (rc != EE_OK) {
                 DPRINTF("Failed to load executable '%s'.\n", pathname);
@@ -261 +275 @@
         }
         
-        elf_set_pcb(&prog_info, &pcb);
+        elf_create_pcb(&prog_info, &pcb);
         
         pcb.cwd = cwd;
@@ -270 +284 @@
         pcb.filc = filc;
         
+        if (prog_info.interp == NULL) {
+                /* Statically linked program */
+                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);
         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.
@@ -295 +374 @@
         async_answer_0(rid, EOK);
         DPRINTF("Jump to entry point at %p\n", pcb.entry);
-        entry_point_jmp(prog_info.finfo.entry, &pcb);
+        entry_point_jmp(prog_info.entry, &pcb);
         
         /* Not reached */
@@ -364 +443 @@
 int main(int argc, char *argv[])
 {
-        async_set_fallback_port_handler(ldr_connection, NULL);
+        /* Set a handler of incomming connections. */
+        async_set_client_connection(ldr_connection);
         
         /* Introduce this task to the NS (give it our task ID). */
@@ -372 +452 @@
                 return rc;
         
-        /* Create port */
-        port_id_t port;
-        rc = async_create_port(INTERFACE_LOADER, ldr_connection, NULL, &port);
+        /* Register at naming service. */
+        rc = service_register(SERVICE_LOAD);
         if (rc != EOK)
                 return rc;
         
-        /* Register at naming service. */
-        rc = service_register(SERVICE_LOADER);
-        if (rc != EOK)
-                return rc;
-        
         async_manager();