source: mainline/uspace/srv/loader/main.c@ 8d2dd7f2

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 8d2dd7f2 was 8d2dd7f2, checked in by Jakub Jermar <jakub@…>, 8 years ago

Reduce the number of files that include <sys/types.h>
  • Property mode set to 100644
File size: 9.0 KB
Line 
1/*
2 * Copyright (c) 2008 Jiri Svoboda
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
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
15 * derived from this software without specific prior written permission.
16 *
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
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
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
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
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup loader
30 * @brief Loads and runs programs from VFS.
31 * @{
32 */
33/**
34 * @file
35 * @brief Loads and runs programs from VFS.
36 *
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
39 * returns the id of a phone connected to the newly created task.
40 *
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
43 * and completely hidden from applications.
44 */
45
46#include <stdio.h>
47#include <stdlib.h>
48#include <stdbool.h>
49#include <stddef.h>
50#include <ipc/services.h>
51#include <ipc/loader.h>
52#include <ns.h>
53#include <loader/pcb.h>
54#include <entry_point.h>
55#include <errno.h>
56#include <async.h>
57#include <str.h>
58#include <as.h>
59#include <elf/elf.h>
60#include <elf/elf_load.h>
61#include <vfs/vfs.h>
62#include <vfs/inbox.h>
63
64#define DPRINTF(...)
65
66/** File that will be loaded */
67static char *progname = NULL;
68static int program_fd = -1;
69
70/** The Program control block */
71static pcb_t pcb;
72
73/** Current working directory */
74static char *cwd = NULL;
75
76/** Number of arguments */
77static int argc = 0;
78/** Argument vector */
79static char **argv = NULL;
80/** Buffer holding all arguments */
81static char *arg_buf = NULL;
82
83/** Inbox entries. */
84static struct pcb_inbox_entry inbox[INBOX_MAX_ENTRIES];
85static int inbox_entries = 0;
86
87static elf_info_t prog_info;
88
89/** Used to limit number of connections to one. */
90static bool connected = false;
91
92static void ldr_get_taskid(ipc_callid_t rid, ipc_call_t *request)
93{
94 ipc_callid_t callid;
95 task_id_t task_id;
96 size_t len;
97
98 task_id = task_get_id();
99
100 if (!async_data_read_receive(&callid, &len)) {
101 async_answer_0(callid, EINVAL);
102 async_answer_0(rid, EINVAL);
103 return;
104 }
105
106 if (len > sizeof(task_id))
107 len = sizeof(task_id);
108
109 async_data_read_finalize(callid, &task_id, len);
110 async_answer_0(rid, EOK);
111}
112
113/** Receive a call setting the current working directory.
114 *
115 * @param rid
116 * @param request
117 */
118static void ldr_set_cwd(ipc_callid_t rid, ipc_call_t *request)
119{
120 char *buf;
121 int rc = async_data_write_accept((void **) &buf, true, 0, 0, 0, NULL);
122
123 if (rc == EOK) {
124 if (cwd != NULL)
125 free(cwd);
126
127 cwd = buf;
128 }
129
130 async_answer_0(rid, rc);
131}
132
133/** Receive a call setting the program to execute.
134 *
135 * @param rid
136 * @param request
137 */
138static void ldr_set_program(ipc_callid_t rid, ipc_call_t *request)
139{
140 ipc_callid_t writeid;
141 size_t namesize;
142 if (!async_data_write_receive(&writeid, &namesize)) {
143 async_answer_0(rid, EINVAL);
144 return;
145 }
146
147 char* name = malloc(namesize);
148 int rc = async_data_write_finalize(writeid, name, namesize);
149 if (rc != EOK) {
150 async_answer_0(rid, EINVAL);
151 return;
152 }
153
154 int file = vfs_receive_handle(true);
155 if (file < 0) {
156 async_answer_0(rid, EINVAL);
157 return;
158 }
159
160 progname = name;
161 program_fd = file;
162 async_answer_0(rid, EOK);
163}
164
165/** Receive a call setting arguments of the program to execute.
166 *
167 * @param rid
168 * @param request
169 */
170static void ldr_set_args(ipc_callid_t rid, ipc_call_t *request)
171{
172 char *buf;
173 size_t buf_size;
174 int rc = async_data_write_accept((void **) &buf, true, 0, 0, 0, &buf_size);
175
176 if (rc == EOK) {
177 /*
178 * Count number of arguments
179 */
180 char *cur = buf;
181 int count = 0;
182
183 while (cur < buf + buf_size) {
184 size_t arg_size = str_size(cur);
185 cur += arg_size + 1;
186 count++;
187 }
188
189 /*
190 * Allocate new argv
191 */
192 char **_argv = (char **) malloc((count + 1) * sizeof(char *));
193 if (_argv == NULL) {
194 free(buf);
195 async_answer_0(rid, ENOMEM);
196 return;
197 }
198
199 /*
200 * Fill the new argv with argument pointers
201 */
202 cur = buf;
203 count = 0;
204 while (cur < buf + buf_size) {
205 _argv[count] = cur;
206
207 size_t arg_size = str_size(cur);
208 cur += arg_size + 1;
209 count++;
210 }
211 _argv[count] = NULL;
212
213 /*
214 * Copy temporary data to global variables
215 */
216 if (arg_buf != NULL)
217 free(arg_buf);
218
219 if (argv != NULL)
220 free(argv);
221
222 argc = count;
223 arg_buf = buf;
224 argv = _argv;
225 }
226
227 async_answer_0(rid, rc);
228}
229
230/** Receive a call setting inbox files of the program to execute.
231 *
232 * @param rid
233 * @param request
234 */
235static void ldr_add_inbox(ipc_callid_t rid, ipc_call_t *request)
236{
237 if (inbox_entries == INBOX_MAX_ENTRIES) {
238 async_answer_0(rid, ERANGE);
239 return;
240 }
241
242 ipc_callid_t writeid;
243 size_t namesize;
244 if (!async_data_write_receive(&writeid, &namesize)) {
245 async_answer_0(rid, EINVAL);
246 return;
247 }
248
249 char* name = malloc(namesize);
250 int rc = async_data_write_finalize(writeid, name, namesize);
251 if (rc != EOK) {
252 async_answer_0(rid, EINVAL);
253 return;
254 }
255
256 int file = vfs_receive_handle(true);
257 if (file < 0) {
258 async_answer_0(rid, EINVAL);
259 return;
260 }
261
262 /*
263 * We need to set the root early for dynamically linked binaries so
264 * that the loader can use it too.
265 */
266 if (str_cmp(name, "root") == 0)
267 vfs_root_set(file);
268
269 inbox[inbox_entries].name = name;
270 inbox[inbox_entries].file = file;
271 inbox_entries++;
272 async_answer_0(rid, EOK);
273}
274
275/** Load the previously selected program.
276 *
277 * @param rid
278 * @param request
279 * @return 0 on success, !0 on error.
280 */
281static int ldr_load(ipc_callid_t rid, ipc_call_t *request)
282{
283 int rc = elf_load(program_fd, &prog_info);
284 if (rc != EE_OK) {
285 DPRINTF("Failed to load executable for '%s'.\n", progname);
286 async_answer_0(rid, EINVAL);
287 return 1;
288 }
289
290 elf_set_pcb(&prog_info, &pcb);
291
292 pcb.cwd = cwd;
293
294 pcb.argc = argc;
295 pcb.argv = argv;
296
297 pcb.inbox = inbox;
298 pcb.inbox_entries = inbox_entries;
299
300 async_answer_0(rid, rc);
301 return 0;
302}
303
304/** Run the previously loaded program.
305 *
306 * @param rid
307 * @param request
308 * @return 0 on success, !0 on error.
309 */
310static __attribute__((noreturn)) void ldr_run(ipc_callid_t rid,
311 ipc_call_t *request)
312{
313 DPRINTF("Set task name\n");
314
315 /* Set the task name. */
316 task_set_name(progname);
317
318 /* Run program */
319 DPRINTF("Reply OK\n");
320 async_answer_0(rid, EOK);
321 DPRINTF("Jump to entry point at %p\n", pcb.entry);
322 entry_point_jmp(prog_info.finfo.entry, &pcb);
323
324 /* Not reached */
325}
326
327/** Handle loader connection.
328 *
329 * Receive and carry out commands (of which the last one should be
330 * to execute the loaded program).
331 */
332static void ldr_connection(ipc_callid_t iid, ipc_call_t *icall, void *arg)
333{
334 /* Already have a connection? */
335 if (connected) {
336 async_answer_0(iid, ELIMIT);
337 return;
338 }
339
340 connected = true;
341
342 /* Accept the connection */
343 async_answer_0(iid, EOK);
344
345 /* Ignore parameters, the connection is already open */
346 (void) icall;
347
348 while (true) {
349 int retval;
350 ipc_call_t call;
351 ipc_callid_t callid = async_get_call(&call);
352
353 if (!IPC_GET_IMETHOD(call))
354 exit(0);
355
356 switch (IPC_GET_IMETHOD(call)) {
357 case LOADER_GET_TASKID:
358 ldr_get_taskid(callid, &call);
359 continue;
360 case LOADER_SET_CWD:
361 ldr_set_cwd(callid, &call);
362 continue;
363 case LOADER_SET_PROGRAM:
364 ldr_set_program(callid, &call);
365 continue;
366 case LOADER_SET_ARGS:
367 ldr_set_args(callid, &call);
368 continue;
369 case LOADER_ADD_INBOX:
370 ldr_add_inbox(callid, &call);
371 continue;
372 case LOADER_LOAD:
373 ldr_load(callid, &call);
374 continue;
375 case LOADER_RUN:
376 ldr_run(callid, &call);
377 /* Not reached */
378 default:
379 retval = EINVAL;
380 break;
381 }
382
383 async_answer_0(callid, retval);
384 }
385}
386
387/** Program loader main function.
388 */
389int main(int argc, char *argv[])
390{
391 async_set_fallback_port_handler(ldr_connection, NULL);
392
393 /* Introduce this task to the NS (give it our task ID). */
394 task_id_t id = task_get_id();
395 int rc = ns_intro(id);
396 if (rc != EOK)
397 return rc;
398
399 /* Create port */
400 port_id_t port;
401 rc = async_create_port(INTERFACE_LOADER, ldr_connection, NULL, &port);
402 if (rc != EOK)
403 return rc;
404
405 /* Register at naming service. */
406 rc = service_register(SERVICE_LOADER);
407 if (rc != EOK)
408 return rc;
409
410 async_manager();
411
412 /* Never reached */
413 return 0;
414}
415
416/** @}
417 */
Note: See TracBrowser for help on using the repository browser.