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