WebSVN – HelenOS – Diff – /branches/network/uspace/srv/loader/main.c


/*
 * 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 <ipc/ns.h>
#include <macros.h>
#include <loader/pcb.h>
#include <errno.h>
#include <async.h>
#include <string.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;
 
/** Number of preset files */
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;
 
static void ldr_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 ldr_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 ldr_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);
}
 
/** Receive a call setting preset files of the program to execute.
 *
 * @param rid
 * @param request
 */
static void ldr_set_files(ipc_callid_t rid, ipc_call_t *request)
{
    ipc_callid_t callid;
    size_t buf_size;
    if (!ipc_data_write_receive(&callid, &buf_size)) {
        ipc_answer_0(callid, EINVAL);
        ipc_answer_0(rid, EINVAL);
        return;
    }
   
    if ((buf_size % sizeof(fdi_node_t)) != 0) {
        ipc_answer_0(callid, EINVAL);
        ipc_answer_0(rid, EINVAL);
        return;
    }
   
    if (fil_buf != NULL) {
        free(fil_buf);
        fil_buf = NULL;
    }
   
    if (filv != NULL) {
        free(filv);
        filv = NULL;
    }
   
    fil_buf = malloc(buf_size);
    if (!fil_buf) {
        ipc_answer_0(callid, ENOMEM);
        ipc_answer_0(rid, ENOMEM);
        return;
    }
   
    ipc_data_write_finalize(callid, fil_buf, buf_size);
   
    int count = buf_size / sizeof(fdi_node_t);
   
    /* Allocate filvv */
    filv = malloc((count + 1) * sizeof(fdi_node_t *));
   
    if (filv == NULL) {
        free(fil_buf);
        ipc_answer_0(rid, ENOMEM);
        return;
    }
   
    /*
     * Fill filv with argument pointers
     */
    int i;
    for (i = 0; i < count; i++)
        filv[i] = &fil_buf[i];
   
    filc = count;
    filv[count] = NULL;
   
    ipc_answer_0(rid, EOK);
}
 
/** Load the previously selected program.
 *
 * @param rid
 * @param request
 * @return 0 on success, !0 on error.
 */
static int ldr_load(ipc_callid_t rid, ipc_call_t *request)
{
    int rc;
   
    rc = elf_load_file(pathname, 0, &prog_info);
    if (rc != EE_OK) {
        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;
   
    pcb.filc = filc;
    pcb.filv = filv;
   
    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 != EE_OK) {
        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 ldr_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);
       
        ipc_answer_0(rid, EOK);
        elf_run(&interp_info, &pcb);
    } else {
        /* Statically linked program */
        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 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) {
        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:
            ldr_get_taskid(callid, &call);
            continue;
        case LOADER_SET_PATHNAME:
            ldr_set_pathname(callid, &call);
            continue;
        case LOADER_SET_ARGS:
            ldr_set_args(callid, &call);
            continue;
        case LOADER_SET_FILES:
            ldr_set_files(callid, &call);
            continue;
        case LOADER_LOAD:
            ldr_load(callid, &call);
            continue;
        case LOADER_RUN:
            ldr_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;
    task_id_t id;
    int rc;
 
    connected = false;
 
    /* Introduce this task to the NS (give it our task ID). */
    id = task_get_id();
    rc = async_req_2_0(PHONE_NS, NS_ID_INTRO, LOWER32(id), UPPER32(id));
    if (rc != EOK)
        return -1;
 
    /* Set a handler of incomming connections. */
    async_set_client_connection(ldr_connection);
   
    /* Register at naming service. */
    if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0)
        return -2;
 
    async_manager();
   
    /* Never reached */
    return 0;
}
 
/** @}
 */
 

Rev 4581	Rev 4718
1	/*	1	/*
2	* Copyright (c) 2008 Jiri Svoboda	2	* Copyright (c) 2008 Jiri Svoboda
3	* All rights reserved.	3	* All rights reserved.
4	*	4	*
5	* Redistribution and use in source and binary forms, with or without	5	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions	6	* modification, are permitted provided that the following conditions
7	* are met:	7	* are met:
8	*	8	*
9	* - Redistributions of source code must retain the above copyright	9	* - Redistributions of source code must retain the above copyright
10	* notice, this list of conditions and the following disclaimer.	10	* notice, this list of conditions and the following disclaimer.
11	* - Redistributions in binary form must reproduce the above copyright	11	* - Redistributions in binary form must reproduce the above copyright
12	* notice, this list of conditions and the following disclaimer in the	12	* notice, this list of conditions and the following disclaimer in the
13	* documentation and/or other materials provided with the distribution.	13	* documentation and/or other materials provided with the distribution.
14	* - The name of the author may not be used to endorse or promote products	14	* - The name of the author may not be used to endorse or promote products
15	* derived from this software without specific prior written permission.	15	* derived from this software without specific prior written permission.
16	*	16	*
17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	*/	27	*/
28		28
29	/** @addtogroup loader	29	/** @addtogroup loader
30	* @brief Loads and runs programs from VFS.	30	* @brief Loads and runs programs from VFS.
31	* @{	31	* @{
32	*/	32	*/
33	/**	33	/**
34	* @file	34	* @file
35	* @brief Loads and runs programs from VFS.	35	* @brief Loads and runs programs from VFS.
36	*	36	*
37	* The program loader is a special init binary. Its image is used	37	* The program loader is a special init binary. Its image is used
38	* to create a new task upon a @c task_spawn syscall. The syscall	38	* to create a new task upon a @c task_spawn syscall. The syscall
39	* returns the id of a phone connected to the newly created task.	39	* returns the id of a phone connected to the newly created task.
40	*	40	*
41	* The caller uses this phone to send the pathname and various other	41	* The caller uses this phone to send the pathname and various other
42	* information to the loader. This is normally done by the C library	42	* information to the loader. This is normally done by the C library
43	* and completely hidden from applications.	43	* and completely hidden from applications.
44	*/	44	*/
45		45
46	#include <stdio.h>	46	#include <stdio.h>
47	#include <stdlib.h>	47	#include <stdlib.h>
48	#include <unistd.h>	48	#include <unistd.h>
49	#include <bool.h>	49	#include <bool.h>
50	#include <fcntl.h>	50	#include <fcntl.h>
51	#include <sys/types.h>	51	#include <sys/types.h>
52	#include <ipc/ipc.h>	52	#include <ipc/ipc.h>
53	#include <ipc/services.h>	53	#include <ipc/services.h>
54	#include <ipc/loader.h>	54	#include <ipc/loader.h>
-		55	#include <ipc/ns.h>
-		56	#include <macros.h>
55	#include <loader/pcb.h>	57	#include <loader/pcb.h>
56	#include <errno.h>	58	#include <errno.h>
57	#include <async.h>	59	#include <async.h>
58	#include <string.h>	60	#include <string.h>
59	#include <as.h>	61	#include <as.h>
60		62
61	#include <elf.h>	63	#include <elf.h>
62	#include <elf_load.h>	64	#include <elf_load.h>
63		65
64	#define DPRINTF(...)	66	#define DPRINTF(...)
65		67
66	/** Pathname of the file that will be loaded */	68	/** Pathname of the file that will be loaded */
67	static char *pathname = NULL;	69	static char *pathname = NULL;
68		70
69	/** The Program control block */	71	/** The Program control block */
70	static pcb_t pcb;	72	static pcb_t pcb;
71		73
72	/** Number of arguments */	74	/** Number of arguments */
73	static int argc = 0;	75	static int argc = 0;
74	/** Argument vector */	76	/** Argument vector */
75	static char **argv = NULL;	77	static char **argv = NULL;
76	/** Buffer holding all arguments */	78	/** Buffer holding all arguments */
77	static char *arg_buf = NULL;	79	static char *arg_buf = NULL;
78		80
79	/** Number of preset files */	81	/** Number of preset files */
80	static int filc = 0;	82	static int filc = 0;
81	/** Preset files vector */	83	/** Preset files vector */
82	static fdi_node_t **filv = NULL;	84	static fdi_node_t **filv = NULL;
83	/** Buffer holding all preset files */	85	/** Buffer holding all preset files */
84	static fdi_node_t *fil_buf = NULL;	86	static fdi_node_t *fil_buf = NULL;
85		87
86	static elf_info_t prog_info;	88	static elf_info_t prog_info;
87	static elf_info_t interp_info;	89	static elf_info_t interp_info;
88		90
89	static bool is_dyn_linked;	91	static bool is_dyn_linked;
90		92
91	/** Used to limit number of connections to one. */	93	/** Used to limit number of connections to one. */
92	static bool connected;	94	static bool connected;
93		95
94	static void ldr_get_taskid(ipc_callid_t rid, ipc_call_t *request)	96	static void ldr_get_taskid(ipc_callid_t rid, ipc_call_t *request)
95	{	97	{
96	ipc_callid_t callid;	98	ipc_callid_t callid;
97	task_id_t task_id;	99	task_id_t task_id;
98	size_t len;	100	size_t len;
99		101
100	task_id = task_get_id();	102	task_id = task_get_id();
101		103
102	if (!ipc_data_read_receive(&callid, &len)) {	104	if (!ipc_data_read_receive(&callid, &len)) {
103	ipc_answer_0(callid, EINVAL);	105	ipc_answer_0(callid, EINVAL);
104	ipc_answer_0(rid, EINVAL);	106	ipc_answer_0(rid, EINVAL);
105	return;	107	return;
106	}	108	}
107		109
108	if (len > sizeof(task_id))	110	if (len > sizeof(task_id))
109	len = sizeof(task_id);	111	len = sizeof(task_id);
110		112
111	ipc_data_read_finalize(callid, &task_id, len);	113	ipc_data_read_finalize(callid, &task_id, len);
112	ipc_answer_0(rid, EOK);	114	ipc_answer_0(rid, EOK);
113	}	115	}
114		116
115		117
116	/** Receive a call setting pathname of the program to execute.	118	/** Receive a call setting pathname of the program to execute.
117	*	119	*
118	* @param rid	120	* @param rid
119	* @param request	121	* @param request
120	*/	122	*/
121	static void ldr_set_pathname(ipc_callid_t rid, ipc_call_t *request)	123	static void ldr_set_pathname(ipc_callid_t rid, ipc_call_t *request)
122	{	124	{
123	ipc_callid_t callid;	125	ipc_callid_t callid;
124	size_t len;	126	size_t len;
125	char *name_buf;	127	char *name_buf;
126		128
127	if (!ipc_data_write_receive(&callid, &len)) {	129	if (!ipc_data_write_receive(&callid, &len)) {
128	ipc_answer_0(callid, EINVAL);	130	ipc_answer_0(callid, EINVAL);
129	ipc_answer_0(rid, EINVAL);	131	ipc_answer_0(rid, EINVAL);
130	return;	132	return;
131	}	133	}
132		134
133	name_buf = malloc(len + 1);	135	name_buf = malloc(len + 1);
134	if (!name_buf) {	136	if (!name_buf) {
135	ipc_answer_0(callid, ENOMEM);	137	ipc_answer_0(callid, ENOMEM);
136	ipc_answer_0(rid, ENOMEM);	138	ipc_answer_0(rid, ENOMEM);
137	return;	139	return;
138	}	140	}
139		141
140	ipc_data_write_finalize(callid, name_buf, len);	142	ipc_data_write_finalize(callid, name_buf, len);
141	ipc_answer_0(rid, EOK);	143	ipc_answer_0(rid, EOK);
142		144
143	if (pathname != NULL) {	145	if (pathname != NULL) {
144	free(pathname);	146	free(pathname);
145	pathname = NULL;	147	pathname = NULL;
146	}	148	}
147		149
148	name_buf[len] = '\0';	150	name_buf[len] = '\0';
149	pathname = name_buf;	151	pathname = name_buf;
150	}	152	}
151		153
152	/** Receive a call setting arguments of the program to execute.	154	/** Receive a call setting arguments of the program to execute.
153	*	155	*
154	* @param rid	156	* @param rid
155	* @param request	157	* @param request
156	*/	158	*/
157	static void ldr_set_args(ipc_callid_t rid, ipc_call_t *request)	159	static void ldr_set_args(ipc_callid_t rid, ipc_call_t *request)
158	{	160	{
159	ipc_callid_t callid;	161	ipc_callid_t callid;
160	size_t buf_size, arg_size;	162	size_t buf_size, arg_size;
161	char *p;	163	char *p;
162	int n;	164	int n;
163		165
164	if (!ipc_data_write_receive(&callid, &buf_size)) {	166	if (!ipc_data_write_receive(&callid, &buf_size)) {
165	ipc_answer_0(callid, EINVAL);	167	ipc_answer_0(callid, EINVAL);
166	ipc_answer_0(rid, EINVAL);	168	ipc_answer_0(rid, EINVAL);
167	return;	169	return;
168	}	170	}
169		171
170	if (arg_buf != NULL) {	172	if (arg_buf != NULL) {
171	free(arg_buf);	173	free(arg_buf);
172	arg_buf = NULL;	174	arg_buf = NULL;
173	}	175	}
174		176
175	if (argv != NULL) {	177	if (argv != NULL) {
176	free(argv);	178	free(argv);
177	argv = NULL;	179	argv = NULL;
178	}	180	}
179		181
180	arg_buf = malloc(buf_size + 1);	182	arg_buf = malloc(buf_size + 1);
181	if (!arg_buf) {	183	if (!arg_buf) {
182	ipc_answer_0(callid, ENOMEM);	184	ipc_answer_0(callid, ENOMEM);
183	ipc_answer_0(rid, ENOMEM);	185	ipc_answer_0(rid, ENOMEM);
184	return;	186	return;
185	}	187	}
186		188
187	ipc_data_write_finalize(callid, arg_buf, buf_size);	189	ipc_data_write_finalize(callid, arg_buf, buf_size);
188		190
189	arg_buf[buf_size] = '\0';	191	arg_buf[buf_size] = '\0';
190		192
191	/*	193	/*
192	* Count number of arguments	194	* Count number of arguments
193	*/	195	*/
194	p = arg_buf;	196	p = arg_buf;
195	n = 0;	197	n = 0;
196	while (p < arg_buf + buf_size) {	198	while (p < arg_buf + buf_size) {
197	arg_size = str_size(p);	199	arg_size = str_size(p);
198	p = p + arg_size + 1;	200	p = p + arg_size + 1;
199	++n;	201	++n;
200	}	202	}
201		203
202	/* Allocate argv */	204	/* Allocate argv */
203	argv = malloc((n + 1) * sizeof(char *));	205	argv = malloc((n + 1) * sizeof(char *));
204		206
205	if (argv == NULL) {	207	if (argv == NULL) {
206	free(arg_buf);	208	free(arg_buf);
207	ipc_answer_0(rid, ENOMEM);	209	ipc_answer_0(rid, ENOMEM);
208	return;	210	return;
209	}	211	}
210		212
211	/*	213	/*
212	* Fill argv with argument pointers	214	* Fill argv with argument pointers
213	*/	215	*/
214	p = arg_buf;	216	p = arg_buf;
215	n = 0;	217	n = 0;
216	while (p < arg_buf + buf_size) {	218	while (p < arg_buf + buf_size) {
217	argv[n] = p;	219	argv[n] = p;
218		220
219	arg_size = str_size(p);	221	arg_size = str_size(p);
220	p = p + arg_size + 1;	222	p = p + arg_size + 1;
221	++n;	223	++n;
222	}	224	}
223		225
224	argc = n;	226	argc = n;
225	argv[n] = NULL;	227	argv[n] = NULL;
226		228
227	ipc_answer_0(rid, EOK);	229	ipc_answer_0(rid, EOK);
228	}	230	}
229		231
230	/** Receive a call setting preset files of the program to execute.	232	/** Receive a call setting preset files of the program to execute.
231	*	233	*
232	* @param rid	234	* @param rid
233	* @param request	235	* @param request
234	*/	236	*/
235	static void ldr_set_files(ipc_callid_t rid, ipc_call_t *request)	237	static void ldr_set_files(ipc_callid_t rid, ipc_call_t *request)
236	{	238	{
237	ipc_callid_t callid;	239	ipc_callid_t callid;
238	size_t buf_size;	240	size_t buf_size;
239	if (!ipc_data_write_receive(&callid, &buf_size)) {	241	if (!ipc_data_write_receive(&callid, &buf_size)) {
240	ipc_answer_0(callid, EINVAL);	242	ipc_answer_0(callid, EINVAL);
241	ipc_answer_0(rid, EINVAL);	243	ipc_answer_0(rid, EINVAL);
242	return;	244	return;
243	}	245	}
244		246
245	if ((buf_size % sizeof(fdi_node_t)) != 0) {	247	if ((buf_size % sizeof(fdi_node_t)) != 0) {
246	ipc_answer_0(callid, EINVAL);	248	ipc_answer_0(callid, EINVAL);
247	ipc_answer_0(rid, EINVAL);	249	ipc_answer_0(rid, EINVAL);
248	return;	250	return;
249	}	251	}
250		252
251	if (fil_buf != NULL) {	253	if (fil_buf != NULL) {
252	free(fil_buf);	254	free(fil_buf);
253	fil_buf = NULL;	255	fil_buf = NULL;
254	}	256	}
255		257
256	if (filv != NULL) {	258	if (filv != NULL) {
257	free(filv);	259	free(filv);
258	filv = NULL;	260	filv = NULL;
259	}	261	}
260		262
261	fil_buf = malloc(buf_size);	263	fil_buf = malloc(buf_size);
262	if (!fil_buf) {	264	if (!fil_buf) {
263	ipc_answer_0(callid, ENOMEM);	265	ipc_answer_0(callid, ENOMEM);
264	ipc_answer_0(rid, ENOMEM);	266	ipc_answer_0(rid, ENOMEM);
265	return;	267	return;
266	}	268	}
267		269
268	ipc_data_write_finalize(callid, fil_buf, buf_size);	270	ipc_data_write_finalize(callid, fil_buf, buf_size);
269		271
270	int count = buf_size / sizeof(fdi_node_t);	272	int count = buf_size / sizeof(fdi_node_t);
271		273
272	/* Allocate filvv */	274	/* Allocate filvv */
273	filv = malloc((count + 1) * sizeof(fdi_node_t *));	275	filv = malloc((count + 1) * sizeof(fdi_node_t *));
274		276
275	if (filv == NULL) {	277	if (filv == NULL) {
276	free(fil_buf);	278	free(fil_buf);
277	ipc_answer_0(rid, ENOMEM);	279	ipc_answer_0(rid, ENOMEM);
278	return;	280	return;
279	}	281	}
280		282
281	/*	283	/*
282	* Fill filv with argument pointers	284	* Fill filv with argument pointers
283	*/	285	*/
284	int i;	286	int i;
285	for (i = 0; i < count; i++)	287	for (i = 0; i < count; i++)
286	filv[i] = &fil_buf[i];	288	filv[i] = &fil_buf[i];
287		289
288	filc = count;	290	filc = count;
289	filv[count] = NULL;	291	filv[count] = NULL;
290		292
291	ipc_answer_0(rid, EOK);	293	ipc_answer_0(rid, EOK);
292	}	294	}
293		295
294	/** Load the previously selected program.	296	/** Load the previously selected program.
295	*	297	*
296	* @param rid	298	* @param rid
297	* @param request	299	* @param request
298	* @return 0 on success, !0 on error.	300	* @return 0 on success, !0 on error.
299	*/	301	*/
300	static int ldr_load(ipc_callid_t rid, ipc_call_t *request)	302	static int ldr_load(ipc_callid_t rid, ipc_call_t *request)
301	{	303	{
302	int rc;	304	int rc;
303		305
304	rc = elf_load_file(pathname, 0, &prog_info);	306	rc = elf_load_file(pathname, 0, &prog_info);
305	if (rc != EE_OK) {	307	if (rc != EE_OK) {
306	DPRINTF("Failed to load executable '%s'.\n", pathname);	308	DPRINTF("Failed to load executable '%s'.\n", pathname);
307	ipc_answer_0(rid, EINVAL);	309	ipc_answer_0(rid, EINVAL);
308	return 1;	310	return 1;
309	}	311	}
310		312
311	elf_create_pcb(&prog_info, &pcb);	313	elf_create_pcb(&prog_info, &pcb);
312		314
313	pcb.argc = argc;	315	pcb.argc = argc;
314	pcb.argv = argv;	316	pcb.argv = argv;
315		317
316	pcb.filc = filc;	318	pcb.filc = filc;
317	pcb.filv = filv;	319	pcb.filv = filv;
318		320
319	if (prog_info.interp == NULL) {	321	if (prog_info.interp == NULL) {
320	/* Statically linked program */	322	/* Statically linked program */
321	is_dyn_linked = false;	323	is_dyn_linked = false;
322	ipc_answer_0(rid, EOK);	324	ipc_answer_0(rid, EOK);
323	return 0;	325	return 0;
324	}	326	}
325		327
326	rc = elf_load_file(prog_info.interp, 0, &interp_info);	328	rc = elf_load_file(prog_info.interp, 0, &interp_info);
327	if (rc != EE_OK) {	329	if (rc != EE_OK) {
328	DPRINTF("Failed to load interpreter '%s.'\n",	330	DPRINTF("Failed to load interpreter '%s.'\n",
329	prog_info.interp);	331	prog_info.interp);
330	ipc_answer_0(rid, EINVAL);	332	ipc_answer_0(rid, EINVAL);
331	return 1;	333	return 1;
332	}	334	}
333		335
334	is_dyn_linked = true;	336	is_dyn_linked = true;
335	ipc_answer_0(rid, EOK);	337	ipc_answer_0(rid, EOK);
336		338
337	return 0;	339	return 0;
338	}	340	}
339		341
340		342
341	/** Run the previously loaded program.	343	/** Run the previously loaded program.
342	*	344	*
343	* @param rid	345	* @param rid
344	* @param request	346	* @param request
345	* @return 0 on success, !0 on error.	347	* @return 0 on success, !0 on error.
346	*/	348	*/
347	static void ldr_run(ipc_callid_t rid, ipc_call_t *request)	349	static void ldr_run(ipc_callid_t rid, ipc_call_t *request)
348	{	350	{
349	const char *cp;	351	const char *cp;
350		352
351	/* Set the task name. */	353	/* Set the task name. */
352	cp = str_rchr(pathname, '/');	354	cp = str_rchr(pathname, '/');
353	cp = (cp == NULL) ? pathname : (cp + 1);	355	cp = (cp == NULL) ? pathname : (cp + 1);
354	task_set_name(cp);	356	task_set_name(cp);
355		357
356	if (is_dyn_linked == true) {	358	if (is_dyn_linked == true) {
357	/* Dynamically linked program */	359	/* Dynamically linked program */
358	DPRINTF("Run ELF interpreter.\n");	360	DPRINTF("Run ELF interpreter.\n");
359	DPRINTF("Entry point: 0x%lx\n", interp_info.entry);	361	DPRINTF("Entry point: 0x%lx\n", interp_info.entry);
360		362
361	ipc_answer_0(rid, EOK);	363	ipc_answer_0(rid, EOK);
362	elf_run(&interp_info, &pcb);	364	elf_run(&interp_info, &pcb);
363	} else {	365	} else {
364	/* Statically linked program */	366	/* Statically linked program */
365	ipc_answer_0(rid, EOK);	367	ipc_answer_0(rid, EOK);
366	elf_run(&prog_info, &pcb);	368	elf_run(&prog_info, &pcb);
367	}	369	}
368		370
369	/* Not reached */	371	/* Not reached */
370	}	372	}
371		373
372	/** Handle loader connection.	374	/** Handle loader connection.
373	*	375	*
374	* Receive and carry out commands (of which the last one should be	376	* Receive and carry out commands (of which the last one should be
375	* to execute the loaded program).	377	* to execute the loaded program).
376	*/	378	*/
377	static void ldr_connection(ipc_callid_t iid, ipc_call_t *icall)	379	static void ldr_connection(ipc_callid_t iid, ipc_call_t *icall)
378	{	380	{
379	ipc_callid_t callid;	381	ipc_callid_t callid;
380	ipc_call_t call;	382	ipc_call_t call;
381	int retval;	383	int retval;
382		384
383	/* Already have a connection? */	385	/* Already have a connection? */
384	if (connected) {	386	if (connected) {
385	ipc_answer_0(iid, ELIMIT);	387	ipc_answer_0(iid, ELIMIT);
386	return;	388	return;
387	}	389	}
388		390
389	connected = true;	391	connected = true;
390		392
391	/* Accept the connection */	393	/* Accept the connection */
392	ipc_answer_0(iid, EOK);	394	ipc_answer_0(iid, EOK);
393		395
394	/* Ignore parameters, the connection is already open */	396	/* Ignore parameters, the connection is already open */
395	(void) iid;	397	(void) iid;
396	(void) icall;	398	(void) icall;
397		399
398	while (1) {	400	while (1) {
399	callid = async_get_call(&call);	401	callid = async_get_call(&call);
400		402
401	switch (IPC_GET_METHOD(call)) {	403	switch (IPC_GET_METHOD(call)) {
402	case IPC_M_PHONE_HUNGUP:	404	case IPC_M_PHONE_HUNGUP:
403	exit(0);	405	exit(0);
404	case LOADER_GET_TASKID:	406	case LOADER_GET_TASKID:
405	ldr_get_taskid(callid, &call);	407	ldr_get_taskid(callid, &call);
406	continue;	408	continue;
407	case LOADER_SET_PATHNAME:	409	case LOADER_SET_PATHNAME:
408	ldr_set_pathname(callid, &call);	410	ldr_set_pathname(callid, &call);
409	continue;	411	continue;
410	case LOADER_SET_ARGS:	412	case LOADER_SET_ARGS:
411	ldr_set_args(callid, &call);	413	ldr_set_args(callid, &call);
412	continue;	414	continue;
413	case LOADER_SET_FILES:	415	case LOADER_SET_FILES:
414	ldr_set_files(callid, &call);	416	ldr_set_files(callid, &call);
415	continue;	417	continue;
416	case LOADER_LOAD:	418	case LOADER_LOAD:
417	ldr_load(callid, &call);	419	ldr_load(callid, &call);
418	continue;	420	continue;
419	case LOADER_RUN:	421	case LOADER_RUN:
420	ldr_run(callid, &call);	422	ldr_run(callid, &call);
421	/* Not reached */	423	/* Not reached */
422	default:	424	default:
423	retval = ENOENT;	425	retval = ENOENT;
424	break;	426	break;
425	}	427	}
426	if ((callid & IPC_CALLID_NOTIFICATION) == 0 &&	428	if ((callid & IPC_CALLID_NOTIFICATION) == 0 &&
427	IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP) {	429	IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP) {
428	DPRINTF("Responding EINVAL to method %d.\n",	430	DPRINTF("Responding EINVAL to method %d.\n",
429	IPC_GET_METHOD(call));	431	IPC_GET_METHOD(call));
430	ipc_answer_0(callid, EINVAL);	432	ipc_answer_0(callid, EINVAL);
431	}	433	}
432	}	434	}
433	}	435	}
434		436
435	/** Program loader main function.	437	/** Program loader main function.
436	*/	438	*/
437	int main(int argc, char *argv[])	439	int main(int argc, char *argv[])
438	{	440	{
439	ipcarg_t phonead;	441	ipcarg_t phonead;
-		442	task_id_t id;
-		443	int rc;
440		444
441	connected = false;	445	connected = false;
442		446
-		447	/* Introduce this task to the NS (give it our task ID). */
-		448	id = task_get_id();
-		449	rc = async_req_2_0(PHONE_NS, NS_ID_INTRO, LOWER32(id), UPPER32(id));
-		450	if (rc != EOK)
-		451	return -1;
-		452
443	/* Set a handler of incomming connections. */	453	/* Set a handler of incomming connections. */
444	async_set_client_connection(ldr_connection);	454	async_set_client_connection(ldr_connection);
445		455
446	/* Register at naming service. */	456	/* Register at naming service. */
447	if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0)	457	if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0)
448	return -1;	458	return -2;
449		459
450	async_manager();	460	async_manager();
451		461
452	/* Never reached */	462	/* Never reached */
453	return 0;	463	return 0;
454	}	464	}
455		465
456	/** @}	466	/** @}
457	*/	467	*/
458		468

Subversion Repositories HelenOS

(root)/branches/network/uspace/srv/loader/main.c – Rev 4581 → 4718