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/loader.h>
#include <loader/pcb.h>
#include <errno.h>
#include <async.h>
#include <as.h>
 
#include <elf.h>
#include <elf_load.h>
 
 
 
 
 
 
 
/** 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;
 
 
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, 0, &prog_info);
    if (rc < 0) {
        printf("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 < 0) {
        printf("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);
 
    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)
{
    if (is_dyn_linked == true) {
        /* Dynamically linked program */
        printf("run dynamic linker\n");
        printf("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;
 
    /* 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) {
            printf("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[])
{
    ipc_callid_t callid;
    ipc_call_t call;
    ipcarg_t phone_hash;
 
    /* The first call only communicates the incoming phone hash */
    callid = ipc_wait_for_call(&call);
 
    if (IPC_GET_METHOD(call) != LOADER_HELLO) {
        if (IPC_GET_METHOD(call) != IPC_M_PHONE_HUNGUP)
            ipc_answer_0(callid, EINVAL);
        return 1;
    }
 
    ipc_answer_0(callid, EOK);
    phone_hash = call.in_phone_hash;
 
    /*
     * Up until now async must not be used as it couldn't
     * handle incoming requests. (Which means e.g. printf()
     * cannot be used)
     */
    async_new_connection(phone_hash, 0, NULL, loader_connection);
    async_manager();
 
    /* not reached */
    return 0;
}
 
/** @}
 */
 

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

Subversion Repositories HelenOS

(root)/branches/dynload/uspace/srv/loader/main.c – Rev 3561 → 3677