Subversion Repositories HelenOS

Rev

Rev 4348 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 4348 Rev 4389
1
/*
1
/*
2
 * Copyright (c) 2007 Michal Konopa
2
 * Copyright (c) 2007 Michal Konopa
3
 * Copyright (c) 2007 Martin Jelen
3
 * Copyright (c) 2007 Martin Jelen
4
 * Copyright (c) 2007 Peter Majer
4
 * Copyright (c) 2007 Peter Majer
5
 * Copyright (c) 2007 Jakub Jermar
5
 * Copyright (c) 2007 Jakub Jermar
6
 * All rights reserved.
6
 * All rights reserved.
7
 *
7
 *
8
 * Redistribution and use in source and binary forms, with or without
8
 * Redistribution and use in source and binary forms, with or without
9
 * modification, are permitted provided that the following conditions
9
 * modification, are permitted provided that the following conditions
10
 * are met:
10
 * are met:
11
 *
11
 *
12
 * - Redistributions of source code must retain the above copyright
12
 * - Redistributions of source code must retain the above copyright
13
 *   notice, this list of conditions and the following disclaimer.
13
 *   notice, this list of conditions and the following disclaimer.
14
 * - Redistributions in binary form must reproduce the above copyright
14
 * - Redistributions in binary form must reproduce the above copyright
15
 *   notice, this list of conditions and the following disclaimer in the
15
 *   notice, this list of conditions and the following disclaimer in the
16
 *   documentation and/or other materials provided with the distribution.
16
 *   documentation and/or other materials provided with the distribution.
17
 * - The name of the author may not be used to endorse or promote products
17
 * - The name of the author may not be used to endorse or promote products
18
 *   derived from this software without specific prior written permission.
18
 *   derived from this software without specific prior written permission.
19
 *
19
 *
20
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
24
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
25
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
29
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
 */
30
 */
31
 
31
 
32
/** @addtogroup rd
32
/** @addtogroup rd
33
 * @{
33
 * @{
34
 */
34
 */
35
 
35
 
36
/**
36
/**
37
 * @file    rd.c
37
 * @file    rd.c
38
 * @brief   Initial RAM disk for HelenOS.
38
 * @brief   Initial RAM disk for HelenOS.
39
 */
39
 */
40
 
40
 
41
#include <ipc/ipc.h>
41
#include <ipc/ipc.h>
42
#include <ipc/services.h>
42
#include <ipc/services.h>
43
#include <ipc/ns.h>
43
#include <ipc/ns.h>
44
#include <sysinfo.h>
44
#include <sysinfo.h>
45
#include <as.h>
45
#include <as.h>
46
#include <ddi.h>
46
#include <ddi.h>
47
#include <align.h>
47
#include <align.h>
48
#include <bool.h>
48
#include <bool.h>
49
#include <errno.h>
49
#include <errno.h>
50
#include <async.h>
50
#include <async.h>
51
#include <align.h>
51
#include <align.h>
52
#include <async.h>
52
#include <async.h>
53
#include <futex.h>
53
#include <futex.h>
54
#include <stdio.h>
54
#include <stdio.h>
55
#include <ipc/devmap.h>
55
#include <ipc/devmap.h>
56
#include "rd.h"
56
#include "rd.h"
57
 
57
 
58
#define NAME "rd"
58
#define NAME "rd"
59
 
59
 
60
/** Pointer to the ramdisk's image. */
60
/** Pointer to the ramdisk's image. */
61
static void *rd_addr;
61
static void *rd_addr;
62
/** Size of the ramdisk. */
62
/** Size of the ramdisk. */
63
static size_t rd_size;
63
static size_t rd_size;
64
 
64
 
65
/**
65
/**
66
 * This futex protects the ramdisk's data.
66
 * This futex protects the ramdisk's data.
67
 * If we were to serve multiple requests (read + write or several writes)
67
 * If we were to serve multiple requests (read + write or several writes)
68
 * concurrently (i.e. from two or more threads), each read and write needs to be
68
 * concurrently (i.e. from two or more threads), each read and write needs to be
69
 * protected by this futex.
69
 * protected by this futex.
70
 */
70
 */
71
atomic_t rd_futex = FUTEX_INITIALIZER;
71
atomic_t rd_futex = FUTEX_INITIALIZER;
72
 
72
 
73
/** Handle one connection to ramdisk.
73
/** Handle one connection to ramdisk.
74
 *
74
 *
75
 * @param iid       Hash of the request that opened the connection.
75
 * @param iid       Hash of the request that opened the connection.
76
 * @param icall     Call data of the request that opened the connection.
76
 * @param icall     Call data of the request that opened the connection.
77
 */
77
 */
78
static void rd_connection(ipc_callid_t iid, ipc_call_t *icall)
78
static void rd_connection(ipc_callid_t iid, ipc_call_t *icall)
79
{
79
{
80
    ipc_callid_t callid;
80
    ipc_callid_t callid;
81
    ipc_call_t call;
81
    ipc_call_t call;
82
    int retval;
82
    int retval;
83
    void *fs_va = NULL;
83
    void *fs_va = NULL;
84
    off_t offset;
84
    off_t offset;
85
    size_t block_size;
85
    size_t block_size;
86
    size_t maxblock_size;
86
    size_t maxblock_size;
87
 
87
 
88
    /*
88
    /*
89
     * Answer the first IPC_M_CONNECT_ME_TO call.
89
     * Answer the first IPC_M_CONNECT_ME_TO call.
90
     */
90
     */
91
    ipc_answer_0(iid, EOK);
91
    ipc_answer_0(iid, EOK);
92
 
92
 
93
    /*
93
    /*
94
     * Now we wait for the client to send us its communication as_area.
94
     * Now we wait for the client to send us its communication as_area.
95
     */
95
     */
96
    int flags;
96
    int flags;
97
    if (ipc_share_out_receive(&callid, &maxblock_size, &flags)) {
97
    if (ipc_share_out_receive(&callid, &maxblock_size, &flags)) {
98
        fs_va = as_get_mappable_page(maxblock_size);
98
        fs_va = as_get_mappable_page(maxblock_size);
99
        if (fs_va) {
99
        if (fs_va) {
100
            (void) ipc_share_out_finalize(callid, fs_va);
100
            (void) ipc_share_out_finalize(callid, fs_va);
101
        } else {
101
        } else {
102
            ipc_answer_0(callid, EHANGUP);
102
            ipc_answer_0(callid, EHANGUP);
103
            return;    
103
            return;    
104
        }
104
        }
105
    } else {
105
    } else {
106
        /*
106
        /*
107
         * The client doesn't speak the same protocol.
107
         * The client doesn't speak the same protocol.
108
         * At this point we can't handle protocol variations.
108
         * At this point we can't handle protocol variations.
109
         * Close the connection.
109
         * Close the connection.
110
         */
110
         */
111
        ipc_answer_0(callid, EHANGUP);
111
        ipc_answer_0(callid, EHANGUP);
112
        return;
112
        return;
113
    }
113
    }
114
   
114
   
115
    while (1) {
115
    while (1) {
116
        callid = async_get_call(&call);
116
        callid = async_get_call(&call);
117
        switch (IPC_GET_METHOD(call)) {
117
        switch (IPC_GET_METHOD(call)) {
118
        case IPC_M_PHONE_HUNGUP:
118
        case IPC_M_PHONE_HUNGUP:
119
            /*
119
            /*
120
             * The other side has hung up.
120
             * The other side has hung up.
121
             * Answer the message and exit the fibril.
121
             * Answer the message and exit the fibril.
122
             */
122
             */
123
            ipc_answer_0(callid, EOK);
123
            ipc_answer_0(callid, EOK);
124
            return;
124
            return;
125
        case RD_READ_BLOCK:
125
        case RD_READ_BLOCK:
126
            offset = IPC_GET_ARG1(call);
126
            offset = IPC_GET_ARG1(call);
127
            block_size = IPC_GET_ARG2(call);
127
            block_size = IPC_GET_ARG2(call);
128
            if (block_size > maxblock_size) {
128
            if (block_size > maxblock_size) {
129
                /*
129
                /*
130
                 * Maximum block size exceeded.
130
                 * Maximum block size exceeded.
131
                 */
131
                 */
132
                retval = ELIMIT;
132
                retval = ELIMIT;
133
                break;
133
                break;
134
            }
134
            }
135
            if (offset * block_size > rd_size - block_size) {
135
            if (offset * block_size > rd_size - block_size) {
136
                /*
136
                /*
137
                 * Reading past the end of the device.
137
                 * Reading past the end of the device.
138
                 */
138
                 */
139
                retval = ELIMIT;
139
                retval = ELIMIT;
140
                break;
140
                break;
141
            }
141
            }
142
            futex_down(&rd_futex);
142
            futex_down(&rd_futex);
143
            memcpy(fs_va, rd_addr + offset * block_size, block_size);
143
            memcpy(fs_va, rd_addr + offset * block_size, block_size);
144
            futex_up(&rd_futex);
144
            futex_up(&rd_futex);
145
            retval = EOK;
145
            retval = EOK;
146
            break;
146
            break;
147
        case RD_WRITE_BLOCK:
147
        case RD_WRITE_BLOCK:
148
            offset = IPC_GET_ARG1(call);
148
            offset = IPC_GET_ARG1(call);
149
            block_size = IPC_GET_ARG2(call);
149
            block_size = IPC_GET_ARG2(call);
150
            if (block_size > maxblock_size) {
150
            if (block_size > maxblock_size) {
151
                /*
151
                /*
152
                 * Maximum block size exceeded.
152
                 * Maximum block size exceeded.
153
                 */
153
                 */
154
                retval = ELIMIT;
154
                retval = ELIMIT;
155
                break;
155
                break;
156
            }
156
            }
157
            if (offset * block_size > rd_size - block_size) {
157
            if (offset * block_size > rd_size - block_size) {
158
                /*
158
                /*
159
                 * Writing past the end of the device.
159
                 * Writing past the end of the device.
160
                 */
160
                 */
161
                retval = ELIMIT;
161
                retval = ELIMIT;
162
                break;
162
                break;
163
            }
163
            }
164
            futex_up(&rd_futex);
164
            futex_up(&rd_futex);
165
            memcpy(rd_addr + offset * block_size, fs_va, block_size);
165
            memcpy(rd_addr + offset * block_size, fs_va, block_size);
166
            futex_down(&rd_futex);
166
            futex_down(&rd_futex);
167
            retval = EOK;
167
            retval = EOK;
168
            break;
168
            break;
169
        default:
169
        default:
170
            /*
170
            /*
171
             * The client doesn't speak the same protocol.
171
             * The client doesn't speak the same protocol.
172
             * Instead of closing the connection, we just ignore
172
             * Instead of closing the connection, we just ignore
173
             * the call. This can be useful if the client uses a
173
             * the call. This can be useful if the client uses a
174
             * newer version of the protocol.
174
             * newer version of the protocol.
175
             */
175
             */
176
            retval = EINVAL;
176
            retval = EINVAL;
177
            break;
177
            break;
178
        }
178
        }
179
        ipc_answer_0(callid, retval);
179
        ipc_answer_0(callid, retval);
180
    }
180
    }
181
}
181
}
182
 
182
 
183
static int driver_register(char *name)
183
static int driver_register(char *name)
184
{
184
{
185
    ipcarg_t retval;
185
    ipcarg_t retval;
186
    aid_t req;
186
    aid_t req;
187
    ipc_call_t answer;
187
    ipc_call_t answer;
188
    int phone;
188
    int phone;
189
    ipcarg_t callback_phonehash;
189
    ipcarg_t callback_phonehash;
190
 
190
 
191
    phone = ipc_connect_me_to_blocking(PHONE_NS, SERVICE_DEVMAP, DEVMAP_DRIVER, 0);
191
    phone = ipc_connect_me_to_blocking(PHONE_NS, SERVICE_DEVMAP, DEVMAP_DRIVER, 0);
192
    if (phone < 0) {
192
    if (phone < 0) {
193
        printf(NAME ": Failed to connect to device mapper\n");
193
        printf(NAME ": Failed to connect to device mapper\n");
194
        return -1;
194
        return -1;
195
    }
195
    }
196
   
196
   
197
    req = async_send_2(phone, DEVMAP_DRIVER_REGISTER, 0, 0, &answer);
197
    req = async_send_2(phone, DEVMAP_DRIVER_REGISTER, 0, 0, &answer);
198
 
198
 
199
    retval = ipc_data_write_start(phone, (char *) name, str_size(name) + 1);
199
    retval = ipc_data_write_start(phone, (char *) name, str_size(name) + 1);
200
 
200
 
201
    if (retval != EOK) {
201
    if (retval != EOK) {
202
        async_wait_for(req, NULL);
202
        async_wait_for(req, NULL);
203
        return -1;
203
        return -1;
204
    }
204
    }
205
 
205
 
206
    async_set_client_connection(rd_connection);
206
    async_set_client_connection(rd_connection);
207
 
207
 
208
    ipc_connect_to_me(phone, 0, 0, 0, &callback_phonehash);
208
    ipc_connect_to_me(phone, 0, 0, 0, &callback_phonehash);
209
    async_wait_for(req, &retval);
209
    async_wait_for(req, &retval);
210
 
210
 
211
    return phone;
211
    return phone;
212
}
212
}
213
 
213
 
214
static int device_register(int driver_phone, char *name, int *handle)
214
static int device_register(int driver_phone, char *name, int *handle)
215
{
215
{
216
    ipcarg_t retval;
216
    ipcarg_t retval;
217
    aid_t req;
217
    aid_t req;
218
    ipc_call_t answer;
218
    ipc_call_t answer;
219
 
219
 
220
    req = async_send_2(driver_phone, DEVMAP_DEVICE_REGISTER, 0, 0, &answer);
220
    req = async_send_2(driver_phone, DEVMAP_DEVICE_REGISTER, 0, 0, &answer);
221
 
221
 
222
    retval = ipc_data_write_start(driver_phone, (char *) name,
222
    retval = ipc_data_write_start(driver_phone, (char *) name,
223
        str_size(name) + 1);
223
        str_size(name) + 1);
224
 
224
 
225
    if (retval != EOK) {
225
    if (retval != EOK) {
226
        async_wait_for(req, NULL);
226
        async_wait_for(req, NULL);
227
        return retval;
227
        return retval;
228
    }
228
    }
229
 
229
 
230
    async_wait_for(req, &retval);
230
    async_wait_for(req, &retval);
231
 
231
 
232
    if (handle != NULL)
232
    if (handle != NULL)
233
        *handle = -1;
233
        *handle = -1;
234
   
234
   
235
    if (EOK == retval) {
235
    if (EOK == retval) {
236
        if (NULL != handle)
236
        if (NULL != handle)
237
            *handle = (int) IPC_GET_ARG1(answer);
237
            *handle = (int) IPC_GET_ARG1(answer);
238
    }
238
    }
239
   
239
   
240
    return retval;
240
    return retval;
241
}
241
}
242
 
242
 
243
/** Prepare the ramdisk image for operation. */
243
/** Prepare the ramdisk image for operation. */
244
static bool rd_init(void)
244
static bool rd_init(void)
245
{
245
{
246
    rd_size = sysinfo_value("rd.size");
246
    rd_size = sysinfo_value("rd.size");
247
    void *rd_ph_addr = (void *) sysinfo_value("rd.address.physical");
247
    void *rd_ph_addr = (void *) sysinfo_value("rd.address.physical");
248
   
248
   
249
    if (rd_size == 0) {
249
    if (rd_size == 0) {
250
        printf(NAME ": No RAM disk found\n");
250
        printf(NAME ": No RAM disk found\n");
251
        return false;
251
        return false;
252
    }
252
    }
253
   
253
   
254
    rd_addr = as_get_mappable_page(rd_size);
254
    rd_addr = as_get_mappable_page(rd_size);
255
   
255
   
256
    int flags = AS_AREA_READ | AS_AREA_WRITE | AS_AREA_CACHEABLE;
256
    int flags = AS_AREA_READ | AS_AREA_WRITE | AS_AREA_CACHEABLE;
257
    int retval = physmem_map(rd_ph_addr, rd_addr,
257
    int retval = physmem_map(rd_ph_addr, rd_addr,
258
        ALIGN_UP(rd_size, PAGE_SIZE) >> PAGE_WIDTH, flags);
258
        ALIGN_UP(rd_size, PAGE_SIZE) >> PAGE_WIDTH, flags);
259
   
259
   
260
    if (retval < 0) {
260
    if (retval < 0) {
261
        printf(NAME ": Error mapping RAM disk\n");
261
        printf(NAME ": Error mapping RAM disk\n");
262
        return false;
262
        return false;
263
    }
263
    }
264
   
264
   
265
    printf(NAME ": Found RAM disk at %p, %d bytes\n", rd_ph_addr, rd_size);
265
    printf(NAME ": Found RAM disk at %p, %d bytes\n", rd_ph_addr, rd_size);
266
   
266
   
267
    int driver_phone = driver_register(NAME);
267
    int driver_phone = driver_register(NAME);
268
    if (driver_phone < 0) {
268
    if (driver_phone < 0) {
269
        printf(NAME ": Unable to register driver\n");
269
        printf(NAME ": Unable to register driver\n");
270
        return false;
270
        return false;
271
    }
271
    }
272
   
272
   
273
    int dev_handle;
273
    int dev_handle;
274
    if (EOK != device_register(driver_phone, "initrd", &dev_handle)) {
274
    if (EOK != device_register(driver_phone, "initrd", &dev_handle)) {
275
        ipc_hangup(driver_phone);
275
        ipc_hangup(driver_phone);
276
        printf(NAME ": Unable to register device\n");
276
        printf(NAME ": Unable to register device\n");
277
        return false;
277
        return false;
278
    }
278
    }
-
 
279
 
-
 
280
    /*
-
 
281
     * Create the second device.
-
 
282
     * We need at least two devices for the sake of testing of non-root
-
 
283
     * mounts. Of course it would be better to allow the second device
-
 
284
     * be created dynamically...
-
 
285
     */
-
 
286
    if (EOK != device_register(driver_phone, "spared", &dev_handle)) {
-
 
287
        ipc_hangup(driver_phone);
-
 
288
        printf(NAME ": Unable to register device\n");
-
 
289
        return false;
-
 
290
    }
279
   
291
   
280
    return true;
292
    return true;
281
}
293
}
282
 
294
 
283
int main(int argc, char **argv)
295
int main(int argc, char **argv)
284
{
296
{
285
    printf(NAME ": HelenOS RAM disk server\n");
297
    printf(NAME ": HelenOS RAM disk server\n");
286
   
298
   
287
    if (!rd_init())
299
    if (!rd_init())
288
        return -1;
300
        return -1;
289
   
301
   
290
    printf(NAME ": Accepting connections\n");
302
    printf(NAME ": Accepting connections\n");
291
    async_manager();
303
    async_manager();
292
 
304
 
293
    /* Never reached */
305
    /* Never reached */
294
    return 0;
306
    return 0;
295
}
307
}
296
 
308
 
297
/**
309
/**
298
 * @}
310
 * @}
299
 */
311
 */
300
 
312