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

Subversion Repositories HelenOS

(root)/branches/dynload/uspace/srv/loader/main.c – Rev 4347 → 4348