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

Subversion Repositories HelenOS

(root)/branches/dynload/uspace/srv/loader/main.c – Rev 3448 → 3474