WebSVN – HelenOS – Diff – /trunk/uspace/rd/rd.c


/*
 * Copyright (c) 2007 Michal Konopa
 * Copyright (c) 2007 Martin Jelen
 * Copyright (c) 2007 Peter Majer
 * Copyright (c) 2007 Jakub Jermar
 * 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 rd
 * @{
 */
 
/**
 * @file    rd.c
 * @brief   Initial RAM disk for HelenOS.
 */
 
#include <ipc/ipc.h>
#include <ipc/services.h>
#include <ipc/ns.h>
#include <sysinfo.h>
#include <as.h>
#include <ddi.h>
#include <align.h>
#include <bool.h>
#include <errno.h>
#include <async.h>
#include <align.h>
#include <async.h>
#include <futex.h>
#include "rd.h"
 
/** Pointer to the ramdisk's image. */
static void *rd_addr;
/** Size of the ramdisk. */
static size_t rd_size;
 
/**
 * This futex protects the ramdisk's data.
 * If we were to serve multiple requests (read + write or several writes)
 * concurrently (i.e. from two or more threads), each read and write needs to be
 * protected by this futex.
 */
atomic_t rd_futex = FUTEX_INITIALIZER;
 
/** Handle one connection to ramdisk.
 *
 * @param iid       Hash of the request that opened the connection.
 * @param icall     Call data of the request that opened the connection.
 */
static void rd_connection(ipc_callid_t iid, ipc_call_t *icall)
{
    ipc_callid_t callid;
    ipc_call_t call;
    int retval;
    void *fs_va = NULL;
    ipcarg_t offset;
 
    /*
     * We allocate VA for communication per connection.
     * This allows us to potentionally have more clients and work
     * concurrently.
     */
    fs_va = as_get_mappable_page(ALIGN_UP(BLOCK_SIZE, PAGE_SIZE));
    if (!fs_va) {
        /*
         * Hang up the phone if we cannot proceed any further.
         * This is the answer to the call that opened the connection.
         */
        ipc_answer_fast(iid, EHANGUP, 0, 0);
        return;
    } else {
        /*
         * Answer the first IPC_M_CONNECT_ME_TO call.
         * Return supported block size as ARG1.
         */
        ipc_answer_fast(iid, EOK, BLOCK_SIZE, 0);
    }
 
    /*
     * Now we wait for the client to send us its communication as_area.
     */
    callid = async_get_call(&call);
    if (IPC_GET_METHOD(call) == IPC_M_AS_AREA_SEND) {
        if (IPC_GET_ARG1(call) >= (ipcarg_t) BLOCK_SIZE) {
            /*
             * The client sends an as_area that can absorb the whole
             * block.
             */
            ipc_answer_fast(callid, EOK, (uintptr_t) fs_va, 0);
        } else {
            /*
             * The client offered as_area too small.
             * Close the connection.
             */
            ipc_answer_fast(callid, EHANGUP, 0, 0);
            return;    
        }
    } else {
        /*
         * The client doesn't speak the same protocol.
         * At this point we can't handle protocol variations.
         * Close the connection.
         */
        ipc_answer_fast(callid, EHANGUP, 0, 0);
        return;
    }
   
    while (1) {
        callid = async_get_call(&call);
        switch (IPC_GET_METHOD(call)) {
        case IPC_M_PHONE_HUNGUP:
            /*
             * The other side has hung up.
             * Answer the message and exit the pseudo thread.
             */
            ipc_answer_fast(callid, EOK, 0, 0);
            return;
        case RD_READ_BLOCK:
            offset = IPC_GET_ARG1(call);
            if (offset * BLOCK_SIZE > rd_size - BLOCK_SIZE) {
                /*
                 * Reading past the end of the device.
                 */
                retval = ELIMIT;
                break;
            }
            futex_down(&rd_futex);
            memcpy(fs_va, rd_addr + offset, BLOCK_SIZE);
            futex_up(&rd_futex);
            retval = EOK;
            break;
        case RD_WRITE_BLOCK:
            offset = IPC_GET_ARG1(call);
            if (offset * BLOCK_SIZE > rd_size - BLOCK_SIZE) {
                /*
                 * Writing past the end of the device.
                 */
                retval = ELIMIT;
                break;
            }
            futex_up(&rd_futex);
            memcpy(rd_addr + offset, fs_va, BLOCK_SIZE);
            futex_down(&rd_futex);
            retval = EOK;
            break;
        default:
            /*
             * The client doesn't speak the same protocol.
             * Instead of closing the connection, we just ignore
             * the call. This can be useful if the client uses a
             * newer version of the protocol.
             */
            retval = EINVAL;
            break;
        }
        ipc_answer_fast(callid, retval, 0, 0);
    }
}
 
/** Prepare the ramdisk image for operation. */
static bool rd_init(void)
{
    int retval, flags;
 
    rd_size = sysinfo_value("rd.size");
    void *rd_ph_addr = (void *) sysinfo_value("rd.address.physical");
   
    if (rd_size == 0)
        return false;
   
    rd_addr = as_get_mappable_page(rd_size);
   
    flags = AS_AREA_READ | AS_AREA_WRITE | AS_AREA_CACHEABLE;
    retval = physmem_map(rd_ph_addr, rd_addr,
        ALIGN_UP(rd_size, PAGE_SIZE) >> PAGE_WIDTH, flags);
 
    if (retval < 0)
        return false;
    return true;
}
 
int main(int argc, char **argv)
{
    if (rd_init()) {
        ipcarg_t phonead;
       
        async_set_client_connection(rd_connection);
       
        /* Register service at nameserver */
        if (ipc_connect_to_me(PHONE_NS, SERVICE_RD, 0, &phonead) != 0)
            return -1;
       
        async_manager();
       
        /* Never reached */
        return 0;
    }
   
    return -1;
}
 
/**
 * @}
 */
 

Rev 2475	Rev 2478
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 "rd.h"	54	#include "rd.h"
55		55
56	/** Pointer to the ramdisk's image. */	56	/** Pointer to the ramdisk's image. */
57	static void *rd_addr;	57	static void *rd_addr;
-		58	/** Size of the ramdisk. */
-		59	static size_t rd_size;
58		60
59	/**	61	/**
60	* This futex protects the ramdisk's data.	62	* This futex protects the ramdisk's data.
61	* If we were to serve multiple requests (read + write or several writes)	63	* If we were to serve multiple requests (read + write or several writes)
62	* concurrently (i.e. from two or more threads), each read and write needs to be	64	* concurrently (i.e. from two or more threads), each read and write needs to be
63	* protected by this futex.	65	* protected by this futex.
64	*/	66	*/
65	atomic_t rd_futex = FUTEX_INITIALIZER;	67	atomic_t rd_futex = FUTEX_INITIALIZER;
66		68
67	/** Handle one connection to ramdisk.	69	/** Handle one connection to ramdisk.
68	*	70	*
69	* @param iid Hash of the request that opened the connection.	71	* @param iid Hash of the request that opened the connection.
70	* @param icall Call data of the request that opened the connection.	72	* @param icall Call data of the request that opened the connection.
71	*/	73	*/
72	static void rd_connection(ipc_callid_t iid, ipc_call_t *icall)	74	static void rd_connection(ipc_callid_t iid, ipc_call_t *icall)
73	{	75	{
74	ipc_callid_t callid;	76	ipc_callid_t callid;
75	ipc_call_t call;	77	ipc_call_t call;
76	int retval;	78	int retval;
77	uintptr_t fs_va = NULL;	79	void *fs_va = NULL;
78	ipcarg_t offset;	80	ipcarg_t offset;
79		81
80	/*	82	/*
81	* We allocate VA for communication per connection.	83	* We allocate VA for communication per connection.
82	* This allows us to potentionally have more clients and work	84	* This allows us to potentionally have more clients and work
83	* concurrently.	85	* concurrently.
84	*/	86	*/
85	fs_va = as_get_mappable_page(ALIGN_UP(BLOCK_SIZE, PAGE_SIZE));	87	fs_va = as_get_mappable_page(ALIGN_UP(BLOCK_SIZE, PAGE_SIZE));
86	if (!fs_va) {	88	if (!fs_va) {
87	/*	89	/*
88	* Hang up the phone if we cannot proceed any further.	90	* Hang up the phone if we cannot proceed any further.
89	* This is the answer to the call that opened the connection.	91	* This is the answer to the call that opened the connection.
90	*/	92	*/
91	ipc_answer_fast(iid, EHANGUP, 0, 0);	93	ipc_answer_fast(iid, EHANGUP, 0, 0);
92	return;	94	return;
93	} else {	95	} else {
94	/*	96	/*
95	* Answer the first IPC_M_CONNECT_ME_TO call.	97	* Answer the first IPC_M_CONNECT_ME_TO call.
96	* Return supported block size as ARG1.	98	* Return supported block size as ARG1.
97	*/	99	*/
98	ipc_answer_fast(iid, EOK, BLOCK_SIZE, 0);	100	ipc_answer_fast(iid, EOK, BLOCK_SIZE, 0);
99	}	101	}
100		102
101	/*	103	/*
102	* Now we wait for the client to send us its communication as_area.	104	* Now we wait for the client to send us its communication as_area.
103	*/	105	*/
104	callid = async_get_call(&call);	106	callid = async_get_call(&call);
105	if (IPC_GET_METHOD(call) == IPC_M_AS_AREA_SEND) {	107	if (IPC_GET_METHOD(call) == IPC_M_AS_AREA_SEND) {
106	if (IPC_GET_ARG1(call) >= (ipcarg_t) BLOCK_SIZE) {	108	if (IPC_GET_ARG1(call) >= (ipcarg_t) BLOCK_SIZE) {
107	/*	109	/*
108	* The client sends an as_area that can absorb the whole	110	* The client sends an as_area that can absorb the whole
109	* block.	111	* block.
110	*/	112	*/
111	ipc_answer_fast(callid, EOK, (uintptr_t) fs_va, 0);	113	ipc_answer_fast(callid, EOK, (uintptr_t) fs_va, 0);
112	} else {	114	} else {
113	/*	115	/*
114	* The client offered as_area too small.	116	* The client offered as_area too small.
115	* Close the connection.	117	* Close the connection.
116	*/	118	*/
117	ipc_answer_fast(callid, EHANGUP, 0, 0);	119	ipc_answer_fast(callid, EHANGUP, 0, 0);
118	return;	120	return;
119	}	121	}
120	} else {	122	} else {
121	/*	123	/*
122	* The client doesn't speak the same protocol.	124	* The client doesn't speak the same protocol.
123	* At this point we can't handle protocol variations.	125	* At this point we can't handle protocol variations.
124	* Close the connection.	126	* Close the connection.
125	*/	127	*/
126	ipc_answer_fast(callid, EHANGUP, 0, 0);	128	ipc_answer_fast(callid, EHANGUP, 0, 0);
127	return;	129	return;
128	}	130	}
129		131
130	while (1) {	132	while (1) {
131	callid = async_get_call(&call);	133	callid = async_get_call(&call);
132	switch (IPC_GET_METHOD(call)) {	134	switch (IPC_GET_METHOD(call)) {
133	case IPC_M_PHONE_HUNGUP:	135	case IPC_M_PHONE_HUNGUP:
134	/*	136	/*
135	* The other side has hung up.	137	* The other side has hung up.
136	* Answer the message and exit the pseudo thread.	138	* Answer the message and exit the pseudo thread.
137	*/	139	*/
138	ipc_answer_fast(callid, EOK, 0, 0);	140	ipc_answer_fast(callid, EOK, 0, 0);
139	return;	141	return;
140	case RD_READ_BLOCK:	142	case RD_READ_BLOCK:
141	offset = IPC_GET_ARG1(call);	143	offset = IPC_GET_ARG1(call);
-		144	if (offset * BLOCK_SIZE > rd_size - BLOCK_SIZE) {
-		145	/*
-		146	* Reading past the end of the device.
-		147	*/
-		148	retval = ELIMIT;
-		149	break;
-		150	}
142	futex_down(&rd_futex);	151	futex_down(&rd_futex);
143	memcpy((void *) fs_va, rd_addr + offset, BLOCK_SIZE);	152	memcpy(fs_va, rd_addr + offset, BLOCK_SIZE);
144	futex_up(&rd_futex);	153	futex_up(&rd_futex);
145	retval = EOK;	154	retval = EOK;
146	break;	155	break;
147	case RD_WRITE_BLOCK:	156	case RD_WRITE_BLOCK:
148	offset = IPC_GET_ARG1(call);	157	offset = IPC_GET_ARG1(call);
-		158	if (offset * BLOCK_SIZE > rd_size - BLOCK_SIZE) {
-		159	/*
-		160	* Writing past the end of the device.
-		161	*/
-		162	retval = ELIMIT;
-		163	break;
-		164	}
149	futex_up(&rd_futex);	165	futex_up(&rd_futex);
150	memcpy(rd_addr + offset, (void *) fs_va, BLOCK_SIZE);	166	memcpy(rd_addr + offset, fs_va, BLOCK_SIZE);
151	futex_down(&rd_futex);	167	futex_down(&rd_futex);
152	retval = EOK;	168	retval = EOK;
153	break;	169	break;
154	default:	170	default:
155	/*	171	/*
156	* The client doesn't speak the same protocol.	172	* The client doesn't speak the same protocol.
157	* Instead of closing the connection, we just ignore	173	* Instead of closing the connection, we just ignore
158	* the call. This can be useful if the client uses a	174	* the call. This can be useful if the client uses a
159	* newer version of the protocol.	175	* newer version of the protocol.
160	*/	176	*/
161	retval = EINVAL;	177	retval = EINVAL;
162	break;	178	break;
163	}	179	}
164	ipc_answer_fast(callid, retval, 0, 0);	180	ipc_answer_fast(callid, retval, 0, 0);
165	}	181	}
166	}	182	}
167		183
168	/** Prepare the ramdisk image for operation. */	184	/** Prepare the ramdisk image for operation. */
169	static bool rd_init(void)	185	static bool rd_init(void)
170	{	186	{
171	int retval, flags;	187	int retval, flags;
172		188
173	size_t rd_size = sysinfo_value("rd.size");	189	rd_size = sysinfo_value("rd.size");
174	void rd_ph_addr = (void ) sysinfo_value("rd.address.physical");	190	void rd_ph_addr = (void ) sysinfo_value("rd.address.physical");
175		191
176	if (rd_size == 0)	192	if (rd_size == 0)
177	return false;	193	return false;
178		194
179	rd_addr = as_get_mappable_page(rd_size);	195	rd_addr = as_get_mappable_page(rd_size);
180		196
181	flags = AS_AREA_READ \| AS_AREA_WRITE \| AS_AREA_CACHEABLE;	197	flags = AS_AREA_READ \| AS_AREA_WRITE \| AS_AREA_CACHEABLE;
182	retval = physmem_map(rd_ph_addr, rd_addr,	198	retval = physmem_map(rd_ph_addr, rd_addr,
183	ALIGN_UP(rd_size, PAGE_SIZE) >> PAGE_WIDTH, flags);	199	ALIGN_UP(rd_size, PAGE_SIZE) >> PAGE_WIDTH, flags);
184		200
185	if (retval < 0)	201	if (retval < 0)
186	return false;	202	return false;
187	return true;	203	return true;
188	}	204	}
189		205
190	int main(int argc, char **argv)	206	int main(int argc, char **argv)
191	{	207	{
192	if (rd_init()) {	208	if (rd_init()) {
193	ipcarg_t phonead;	209	ipcarg_t phonead;
194		210
195	async_set_client_connection(rd_connection);	211	async_set_client_connection(rd_connection);
196		212
197	/* Register service at nameserver */	213	/* Register service at nameserver */
198	if (ipc_connect_to_me(PHONE_NS, SERVICE_RD, 0, &phonead) != 0)	214	if (ipc_connect_to_me(PHONE_NS, SERVICE_RD, 0, &phonead) != 0)
199	return -1;	215	return -1;
200		216
201	async_manager();	217	async_manager();
202		218
203	/* Never reached */	219	/* Never reached */
204	return 0;	220	return 0;
205	}	221	}
206		222
207	return -1;	223	return -1;
208	}	224	}
209		225
210	/**	226	/**
211	* @}	227	* @}
212	*/	228	*/
213		229

Subversion Repositories HelenOS

(root)/trunk/uspace/rd/rd.c – Rev 2475 → 2478