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

Subversion Repositories HelenOS

(root)/trunk/uspace/srv/loader/main.c @ 4014 – Rev 3896 → 3983