Subversion Repositories HelenOS

Rev

Rev 3174 | Rev 3448 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

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