WebSVN – HelenOS – Diff – /branches/dd/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 <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 != 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;
   
    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 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;
}
 
/** @}
 */
 

Rev 4419	Rev 4420
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 <loader/pcb.h>	55	#include <loader/pcb.h>
56	#include <console.h>	56	#include <console.h>
57	#include <errno.h>	57	#include <errno.h>
58	#include <async.h>	58	#include <async.h>
59	#include <as.h>	59	#include <as.h>
60		60
61	#include <elf.h>	61	#include <elf.h>
62	#include <elf_load.h>	62	#include <elf_load.h>
63		63
64	#define DPRINTF(...)	64	#define DPRINTF(...)
65		65
66	/** Pathname of the file that will be loaded */	66	/** Pathname of the file that will be loaded */
67	static char *pathname = NULL;	67	static char *pathname = NULL;
68		68
69	/** The Program control block */	69	/** The Program control block */
70	static pcb_t pcb;	70	static pcb_t pcb;
71		71
72	/** Number of arguments */	72	/** Number of arguments */
73	static int argc = 0;	73	static int argc = 0;
74	/** Argument vector */	74	/** Argument vector */
75	static char **argv = NULL;	75	static char **argv = NULL;
76	/** Buffer holding all arguments */	76	/** Buffer holding all arguments */
77	static char *arg_buf = NULL;	77	static char *arg_buf = NULL;
78		78
79	static elf_info_t prog_info;	79	static elf_info_t prog_info;
80	static elf_info_t interp_info;	80	static elf_info_t interp_info;
81		81
82	static bool is_dyn_linked;	82	static bool is_dyn_linked;
83		83
84	/** Used to limit number of connections to one. */	84	/** Used to limit number of connections to one. */
85	static bool connected;	85	static bool connected;
86		86
87	static void loader_get_taskid(ipc_callid_t rid, ipc_call_t *request)	87	static void loader_get_taskid(ipc_callid_t rid, ipc_call_t *request)
88	{	88	{
89	ipc_callid_t callid;	89	ipc_callid_t callid;
90	task_id_t task_id;	90	task_id_t task_id;
91	size_t len;	91	size_t len;
92		92
93	task_id = task_get_id();	93	task_id = task_get_id();
94		94
95	if (!ipc_data_read_receive(&callid, &len)) {	95	if (!ipc_data_read_receive(&callid, &len)) {
96	ipc_answer_0(callid, EINVAL);	96	ipc_answer_0(callid, EINVAL);
97	ipc_answer_0(rid, EINVAL);	97	ipc_answer_0(rid, EINVAL);
98	return;	98	return;
99	}	99	}
100		100
101	if (len > sizeof(task_id))	101	if (len > sizeof(task_id))
102	len = sizeof(task_id);	102	len = sizeof(task_id);
103		103
104	ipc_data_read_finalize(callid, &task_id, len);	104	ipc_data_read_finalize(callid, &task_id, len);
105	ipc_answer_0(rid, EOK);	105	ipc_answer_0(rid, EOK);
106	}	106	}
107		107
108		108
109	/** Receive a call setting pathname of the program to execute.	109	/** Receive a call setting pathname of the program to execute.
110	*	110	*
111	* @param rid	111	* @param rid
112	* @param request	112	* @param request
113	*/	113	*/
114	static void loader_set_pathname(ipc_callid_t rid, ipc_call_t *request)	114	static void loader_set_pathname(ipc_callid_t rid, ipc_call_t *request)
115	{	115	{
116	ipc_callid_t callid;	116	ipc_callid_t callid;
117	size_t len;	117	size_t len;
118	char *name_buf;	118	char *name_buf;
119		119
120	if (!ipc_data_write_receive(&callid, &len)) {	120	if (!ipc_data_write_receive(&callid, &len)) {
121	ipc_answer_0(callid, EINVAL);	121	ipc_answer_0(callid, EINVAL);
122	ipc_answer_0(rid, EINVAL);	122	ipc_answer_0(rid, EINVAL);
123	return;	123	return;
124	}	124	}
125		125
126	name_buf = malloc(len + 1);	126	name_buf = malloc(len + 1);
127	if (!name_buf) {	127	if (!name_buf) {
128	ipc_answer_0(callid, ENOMEM);	128	ipc_answer_0(callid, ENOMEM);
129	ipc_answer_0(rid, ENOMEM);	129	ipc_answer_0(rid, ENOMEM);
130	return;	130	return;
131	}	131	}
132		132
133	ipc_data_write_finalize(callid, name_buf, len);	133	ipc_data_write_finalize(callid, name_buf, len);
134	ipc_answer_0(rid, EOK);	134	ipc_answer_0(rid, EOK);
135		135
136	if (pathname != NULL) {	136	if (pathname != NULL) {
137	free(pathname);	137	free(pathname);
138	pathname = NULL;	138	pathname = NULL;
139	}	139	}
140		140
141	name_buf[len] = '\0';	141	name_buf[len] = '\0';
142	pathname = name_buf;	142	pathname = name_buf;
143	}	143	}
144		144
145	/** Receive a call setting arguments of the program to execute.	145	/** Receive a call setting arguments of the program to execute.
146	*	146	*
147	* @param rid	147	* @param rid
148	* @param request	148	* @param request
149	*/	149	*/
150	static void loader_set_args(ipc_callid_t rid, ipc_call_t *request)	150	static void loader_set_args(ipc_callid_t rid, ipc_call_t *request)
151	{	151	{
152	ipc_callid_t callid;	152	ipc_callid_t callid;
153	size_t buf_size, arg_size;	153	size_t buf_size, arg_size;
154	char *p;	154	char *p;
155	int n;	155	int n;
156		156
157	if (!ipc_data_write_receive(&callid, &buf_size)) {	157	if (!ipc_data_write_receive(&callid, &buf_size)) {
158	ipc_answer_0(callid, EINVAL);	158	ipc_answer_0(callid, EINVAL);
159	ipc_answer_0(rid, EINVAL);	159	ipc_answer_0(rid, EINVAL);
160	return;	160	return;
161	}	161	}
162		162
163	if (arg_buf != NULL) {	163	if (arg_buf != NULL) {
164	free(arg_buf);	164	free(arg_buf);
165	arg_buf = NULL;	165	arg_buf = NULL;
166	}	166	}
167		167
168	if (argv != NULL) {	168	if (argv != NULL) {
169	free(argv);	169	free(argv);
170	argv = NULL;	170	argv = NULL;
171	}	171	}
172		172
173	arg_buf = malloc(buf_size + 1);	173	arg_buf = malloc(buf_size + 1);
174	if (!arg_buf) {	174	if (!arg_buf) {
175	ipc_answer_0(callid, ENOMEM);	175	ipc_answer_0(callid, ENOMEM);
176	ipc_answer_0(rid, ENOMEM);	176	ipc_answer_0(rid, ENOMEM);
177	return;	177	return;
178	}	178	}
179		179
180	ipc_data_write_finalize(callid, arg_buf, buf_size);	180	ipc_data_write_finalize(callid, arg_buf, buf_size);
181		181
182	arg_buf[buf_size] = '\0';	182	arg_buf[buf_size] = '\0';
183		183
184	/*	184	/*
185	* Count number of arguments	185	* Count number of arguments
186	*/	186	*/
187	p = arg_buf;	187	p = arg_buf;
188	n = 0;	188	n = 0;
189	while (p < arg_buf + buf_size) {	189	while (p < arg_buf + buf_size) {
190	arg_size = str_size(p);	190	arg_size = str_size(p);
191	p = p + arg_size + 1;	191	p = p + arg_size + 1;
192	++n;	192	++n;
193	}	193	}
194		194
195	/* Allocate argv */	195	/* Allocate argv */
196	argv = malloc((n + 1) * sizeof(char *));	196	argv = malloc((n + 1) * sizeof(char *));
197		197
198	if (argv == NULL) {	198	if (argv == NULL) {
199	free(arg_buf);	199	free(arg_buf);
200	ipc_answer_0(rid, ENOMEM);	200	ipc_answer_0(rid, ENOMEM);
201	return;	201	return;
202	}	202	}
203		203
204	/*	204	/*
205	* Fill argv with argument pointers	205	* Fill argv with argument pointers
206	*/	206	*/
207	p = arg_buf;	207	p = arg_buf;
208	n = 0;	208	n = 0;
209	while (p < arg_buf + buf_size) {	209	while (p < arg_buf + buf_size) {
210	argv[n] = p;	210	argv[n] = p;
211		211
212	arg_size = str_size(p);	212	arg_size = str_size(p);
213	p = p + arg_size + 1;	213	p = p + arg_size + 1;
214	++n;	214	++n;
215	}	215	}
216		216
217	argc = n;	217	argc = n;
218	argv[n] = NULL;	218	argv[n] = NULL;
219		219
220	ipc_answer_0(rid, EOK);	220	ipc_answer_0(rid, EOK);
221	}	221	}
222		222
223	/** Load the previously selected program.	223	/** Load the previously selected program.
224	*	224	*
225	* @param rid	225	* @param rid
226	* @param request	226	* @param request
227	* @return 0 on success, !0 on error.	227	* @return 0 on success, !0 on error.
228	*/	228	*/
229	static int loader_load(ipc_callid_t rid, ipc_call_t *request)	229	static int loader_load(ipc_callid_t rid, ipc_call_t *request)
230	{	230	{
231	int rc;	231	int rc;
232		232
233	rc = elf_load_file(pathname, 0, &prog_info);	233	rc = elf_load_file(pathname, 0, &prog_info);
234	if (rc < 0) {	234	if (rc != EE_OK) {
235	DPRINTF("Failed to load executable '%s'.\n", pathname);	235	DPRINTF("Failed to load executable '%s'.\n", pathname);
236	ipc_answer_0(rid, EINVAL);	236	ipc_answer_0(rid, EINVAL);
237	return 1;	237	return 1;
238	}	238	}
239		239
240	elf_create_pcb(&prog_info, &pcb);	240	elf_create_pcb(&prog_info, &pcb);
241		241
242	pcb.argc = argc;	242	pcb.argc = argc;
243	pcb.argv = argv;	243	pcb.argv = argv;
244		244
245	if (prog_info.interp == NULL) {	245	if (prog_info.interp == NULL) {
246	/* Statically linked program */	246	/* Statically linked program */
247	is_dyn_linked = false;	247	is_dyn_linked = false;
248	ipc_answer_0(rid, EOK);	248	ipc_answer_0(rid, EOK);
249	return 0;	249	return 0;
250	}	250	}
251		251
252	rc = elf_load_file(prog_info.interp, 0, &interp_info);	252	rc = elf_load_file(prog_info.interp, 0, &interp_info);
253	if (rc < 0) {	253	if (rc != EE_OK) {
254	DPRINTF("Failed to load interpreter '%s.'\n",	254	DPRINTF("Failed to load interpreter '%s.'\n",
255	prog_info.interp);	255	prog_info.interp);
256	ipc_answer_0(rid, EINVAL);	256	ipc_answer_0(rid, EINVAL);
257	return 1;	257	return 1;
258	}	258	}
259		259
260	is_dyn_linked = true;	260	is_dyn_linked = true;
261	ipc_answer_0(rid, EOK);	261	ipc_answer_0(rid, EOK);
262		262
263	return 0;	263	return 0;
264	}	264	}
265		265
266		266
267	/** Run the previously loaded program.	267	/** Run the previously loaded program.
268	*	268	*
269	* @param rid	269	* @param rid
270	* @param request	270	* @param request
271	* @return 0 on success, !0 on error.	271	* @return 0 on success, !0 on error.
272	*/	272	*/
273	static void loader_run(ipc_callid_t rid, ipc_call_t *request)	273	static void loader_run(ipc_callid_t rid, ipc_call_t *request)
274	{	274	{
275	const char *cp;	275	const char *cp;
276		276
277	/* Set the task name. */	277	/* Set the task name. */
278	cp = str_rchr(pathname, '/');	278	cp = str_rchr(pathname, '/');
279	cp = (cp == NULL) ? pathname : (cp + 1);	279	cp = (cp == NULL) ? pathname : (cp + 1);
280	task_set_name(cp);	280	task_set_name(cp);
281		281
282	if (is_dyn_linked == true) {	282	if (is_dyn_linked == true) {
283	/* Dynamically linked program */	283	/* Dynamically linked program */
284	DPRINTF("Run ELF interpreter.\n");	284	DPRINTF("Run ELF interpreter.\n");
285	DPRINTF("Entry point: 0x%lx\n", interp_info.entry);	285	DPRINTF("Entry point: 0x%lx\n", interp_info.entry);
286	console_close();	286	console_close();
287		287
288	ipc_answer_0(rid, EOK);	288	ipc_answer_0(rid, EOK);
289	elf_run(&interp_info, &pcb);	289	elf_run(&interp_info, &pcb);
290	} else {	290	} else {
291	/* Statically linked program */	291	/* Statically linked program */
292	console_close();	292	console_close();
293	ipc_answer_0(rid, EOK);	293	ipc_answer_0(rid, EOK);
294	elf_run(&prog_info, &pcb);	294	elf_run(&prog_info, &pcb);
295	}	295	}
296		296
297	/* Not reached */	297	/* Not reached */
298	}	298	}
299		299
300	/** Handle loader connection.	300	/** Handle loader connection.
301	*	301	*
302	* Receive and carry out commands (of which the last one should be	302	* Receive and carry out commands (of which the last one should be
303	* to execute the loaded program).	303	* to execute the loaded program).
304	*/	304	*/
305	static void loader_connection(ipc_callid_t iid, ipc_call_t *icall)	305	static void loader_connection(ipc_callid_t iid, ipc_call_t *icall)
306	{	306	{
307	ipc_callid_t callid;	307	ipc_callid_t callid;
308	ipc_call_t call;	308	ipc_call_t call;
309	int retval;	309	int retval;
310		310
311	/* Already have a connection? */	311	/* Already have a connection? */
312	if (connected) {	312	if (connected) {
313	ipc_answer_0(iid, ELIMIT);	313	ipc_answer_0(iid, ELIMIT);
314	return;	314	return;
315	}	315	}
316		316
317	connected = true;	317	connected = true;
318		318
319	/* Accept the connection */	319	/* Accept the connection */
320	ipc_answer_0(iid, EOK);	320	ipc_answer_0(iid, EOK);
321		321
322	/* Ignore parameters, the connection is already open */	322	/* Ignore parameters, the connection is already open */
323	(void) iid;	323	(void) iid;
324	(void) icall;	324	(void) icall;
325		325
326	while (1) {	326	while (1) {
327	callid = async_get_call(&call);	327	callid = async_get_call(&call);
328		328
329	switch (IPC_GET_METHOD(call)) {	329	switch (IPC_GET_METHOD(call)) {
330	case IPC_M_PHONE_HUNGUP:	330	case IPC_M_PHONE_HUNGUP:
331	exit(0);	331	exit(0);
332	case LOADER_GET_TASKID:	332	case LOADER_GET_TASKID:
333	loader_get_taskid(callid, &call);	333	loader_get_taskid(callid, &call);
334	continue;	334	continue;
335	case LOADER_SET_PATHNAME:	335	case LOADER_SET_PATHNAME:
336	loader_set_pathname(callid, &call);	336	loader_set_pathname(callid, &call);
337	continue;	337	continue;
338	case LOADER_SET_ARGS:	338	case LOADER_SET_ARGS:
339	loader_set_args(callid, &call);	339	loader_set_args(callid, &call);
340	continue;	340	continue;
341	case LOADER_LOAD:	341	case LOADER_LOAD:
342	loader_load(callid, &call);	342	loader_load(callid, &call);
343	continue;	343	continue;
344	case LOADER_RUN:	344	case LOADER_RUN:
345	loader_run(callid, &call);	345	loader_run(callid, &call);
346	/* Not reached */	346	/* Not reached */
347	default:	347	default:
348	retval = ENOENT;	348	retval = ENOENT;
349	break;	349	break;
350	}	350	}
351	if ((callid & IPC_CALLID_NOTIFICATION) == 0 &&	351	if ((callid & IPC_CALLID_NOTIFICATION) == 0 &&
352	IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP) {	352	IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP) {
353	DPRINTF("Responding EINVAL to method %d.\n",	353	DPRINTF("Responding EINVAL to method %d.\n",
354	IPC_GET_METHOD(call));	354	IPC_GET_METHOD(call));
355	ipc_answer_0(callid, EINVAL);	355	ipc_answer_0(callid, EINVAL);
356	}	356	}
357	}	357	}
358	}	358	}
359		359
360	/** Program loader main function.	360	/** Program loader main function.
361	*/	361	*/
362	int main(int argc, char *argv[])	362	int main(int argc, char *argv[])
363	{	363	{
364	ipcarg_t phonead;	364	ipcarg_t phonead;
365		365
366	connected = false;	366	connected = false;
367		367
368	/* Set a handler of incomming connections. */	368	/* Set a handler of incomming connections. */
369	async_set_client_connection(loader_connection);	369	async_set_client_connection(loader_connection);
370		370
371	/* Register at naming service. */	371	/* Register at naming service. */
372	if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0)	372	if (ipc_connect_to_me(PHONE_NS, SERVICE_LOAD, 0, 0, &phonead) != 0)
373	return -1;	373	return -1;
374		374
375	async_manager();	375	async_manager();
376		376
377	/* Never reached */	377	/* Never reached */
378	return 0;	378	return 0;
379	}	379	}
380		380
381	/** @}	381	/** @}
382	*/	382	*/
383		383

Subversion Repositories HelenOS

(root)/branches/dd/uspace/srv/loader/main.c – Rev 4419 → 4420