Rev 3914 | Rev 4075 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 3914 | Rev 3990 | ||
---|---|---|---|
1 | /* |
1 | /* |
2 | * Copyright (c) 2009 Lukas Mejdrech |
2 | * Copyright (c) 2009 Lukas Mejdrech |
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 packet |
29 | /** @addtogroup packet |
30 | * @{ |
30 | * @{ |
31 | */ |
31 | */ |
32 | 32 | ||
33 | /** @file |
33 | /** @file |
34 | * Packet map and queue implementation. |
34 | * Packet map and queue implementation. |
35 | * This file has to be compiled with both the packet server and the client. |
35 | * This file has to be compiled with both the packet server and the client. |
36 | */ |
36 | */ |
37 | 37 | ||
38 | #include <errno.h> |
38 | #include <errno.h> |
39 | #include <futex.h> |
- | |
40 | #include <malloc.h> |
39 | #include <malloc.h> |
- | 40 | #include <rwlock.h> |
|
41 | //#include <stdio.h> |
41 | //#include <stdio.h> |
42 | #include <string.h> |
42 | #include <string.h> |
43 | 43 | ||
44 | #include <sys/mman.h> |
44 | #include <sys/mman.h> |
45 | 45 | ||
46 | #include "../../err.h" |
46 | #include "../../err.h" |
47 | 47 | ||
48 | #include "../generic_field.h" |
48 | #include "../generic_field.h" |
49 | 49 | ||
50 | #include "packet_header.h" |
50 | #include "packet_header.h" |
51 | #include "packet.h" |
51 | #include "packet.h" |
52 | 52 | ||
53 | // TODO power of 2 aritmetic => div and mod speedup? |
- | |
54 | /** Packet map page size. |
53 | /** Packet map page size. |
55 | */ |
54 | */ |
56 | #define PACKET_MAP_SIZE 100 |
55 | #define PACKET_MAP_SIZE 100 |
57 | 56 | ||
58 | /** Returns the packet map page index. |
57 | /** Returns the packet map page index. |
59 | * @param packet_id The packet identifier. |
58 | * @param packet_id The packet identifier. |
60 | */ |
59 | */ |
61 | #define PACKET_MAP_PAGE( packet_id ) ((( packet_id ) - 1 ) / PACKET_MAP_SIZE ) |
60 | #define PACKET_MAP_PAGE( packet_id ) ((( packet_id ) - 1 ) / PACKET_MAP_SIZE ) |
62 | 61 | ||
63 | /** Returns the packet index in the corresponding packet map page. |
62 | /** Returns the packet index in the corresponding packet map page. |
64 | * @param packet_id The packet identifier. |
63 | * @param packet_id The packet identifier. |
65 | */ |
64 | */ |
66 | #define PACKET_MAP_INDEX( packet_id ) ((( packet_id ) - 1 ) % PACKET_MAP_SIZE ) |
65 | #define PACKET_MAP_INDEX( packet_id ) ((( packet_id ) - 1 ) % PACKET_MAP_SIZE ) |
67 | 66 | ||
68 | /** Type definition of the packet map page. |
67 | /** Type definition of the packet map page. |
69 | */ |
68 | */ |
70 | typedef packet_t packet_map_t[ PACKET_MAP_SIZE ]; |
69 | typedef packet_t packet_map_t[ PACKET_MAP_SIZE ]; |
71 | /** Type definition of the packet map page pointer. |
70 | /** Type definition of the packet map page pointer. |
72 | */ |
71 | */ |
73 | typedef packet_map_t * packet_map_ref; |
72 | typedef packet_map_t * packet_map_ref; |
74 | 73 | ||
75 | /** Packet map. |
74 | /** Packet map. |
76 | * Maps packet identifiers to the packet references. |
75 | * Maps packet identifiers to the packet references. |
77 | * @see generic_field.h |
76 | * @see generic_field.h |
78 | */ |
77 | */ |
79 | GENERIC_FIELD_DECLARE( gpm, packet_map_t ); |
78 | GENERIC_FIELD_DECLARE( gpm, packet_map_t ); |
80 | 79 | ||
81 | /** Releases the packet. |
80 | /** Releases the packet. |
82 | * @param packet The packet to be released. Input parameter. |
81 | * @param packet The packet to be released. Input parameter. |
83 | * @returns EOK on success. |
82 | * @returns EOK on success. |
84 | * @returns EINVAL if the packet is not valid. |
83 | * @returns EINVAL if the packet is not valid. |
85 | */ |
84 | */ |
86 | int packet_destroy( packet_t packet ); |
85 | int packet_destroy( packet_t packet ); |
87 | 86 | ||
88 | /** Packet map global data. |
87 | /** Packet map global data. |
89 | */ |
88 | */ |
90 | static struct{ |
89 | static struct{ |
91 | /** Safety lock. |
90 | /** Safety lock. |
92 | * Used as a mutex. |
- | |
93 | */ |
91 | */ |
94 | futex_t lock; |
92 | rwlock_t lock; |
95 | /** Packet map. |
93 | /** Packet map. |
96 | */ |
94 | */ |
97 | gpm_t packet_map; |
95 | gpm_t packet_map; |
98 | } pm_globals; |
96 | } pm_globals; |
99 | 97 | ||
100 | GENERIC_FIELD_IMPLEMENT( gpm, packet_map_t ); |
98 | GENERIC_FIELD_IMPLEMENT( gpm, packet_map_t ); |
101 | 99 | ||
102 | int packet_destroy( packet_t packet ){ |
100 | int packet_destroy( packet_t packet ){ |
103 | if( ! packet_is_valid( packet )) return EINVAL; |
101 | if( ! packet_is_valid( packet )) return EINVAL; |
104 | return munmap( packet, packet->length ); |
102 | return munmap( packet, packet->length ); |
105 | } |
103 | } |
106 | 104 | ||
107 | int pm_init( void ){ |
105 | int pm_init( void ){ |
108 | ERROR_DECLARE; |
106 | ERROR_DECLARE; |
109 | 107 | ||
110 | // start locked |
108 | rwlock_initialize( & pm_globals.lock ); |
111 | futex_initialize( & pm_globals.lock, 0 ); |
109 | rwlock_write_lock( & pm_globals.lock ); |
112 | ERROR_PROPAGATE( gpm_initialize( & pm_globals.packet_map )); |
110 | ERROR_PROPAGATE( gpm_initialize( & pm_globals.packet_map )); |
113 | // release the lock |
- | |
114 | futex_up( & pm_globals.lock ); |
111 | rwlock_write_unlock( & pm_globals.lock ); |
115 | return EOK; |
112 | return EOK; |
116 | } |
113 | } |
117 | 114 | ||
118 | packet_t pm_find( packet_id_t packet_id ){ |
115 | packet_t pm_find( packet_id_t packet_id ){ |
119 | packet_map_ref map; |
116 | packet_map_ref map; |
120 | packet_t packet; |
117 | packet_t packet; |
121 | 118 | ||
122 | if( ! packet_id ) return NULL; |
119 | if( ! packet_id ) return NULL; |
123 | futex_down( & pm_globals.lock ); |
120 | rwlock_read_lock( & pm_globals.lock ); |
124 | if( packet_id > PACKET_MAP_SIZE * gpm_count( & pm_globals.packet_map )){ |
121 | if( packet_id > PACKET_MAP_SIZE * gpm_count( & pm_globals.packet_map )){ |
125 | futex_up( & pm_globals.lock ); |
122 | rwlock_read_unlock( & pm_globals.lock ); |
126 | return NULL; |
123 | return NULL; |
127 | } |
124 | } |
128 | map = gpm_get_index( & pm_globals.packet_map, PACKET_MAP_PAGE( packet_id )); |
125 | map = gpm_get_index( & pm_globals.packet_map, PACKET_MAP_PAGE( packet_id )); |
129 | if( ! map ){ |
126 | if( ! map ){ |
130 | futex_up( & pm_globals.lock ); |
127 | rwlock_read_unlock( & pm_globals.lock ); |
131 | return NULL; |
128 | return NULL; |
132 | } |
129 | } |
133 | packet = ( * map )[ PACKET_MAP_INDEX( packet_id ) ]; |
130 | packet = ( * map )[ PACKET_MAP_INDEX( packet_id ) ]; |
134 | futex_up( & pm_globals.lock ); |
131 | rwlock_read_unlock( & pm_globals.lock ); |
135 | return packet; |
132 | return packet; |
136 | } |
133 | } |
137 | 134 | ||
138 | int pm_add( packet_t packet ){ |
135 | int pm_add( packet_t packet ){ |
139 | ERROR_DECLARE; |
136 | ERROR_DECLARE; |
140 | 137 | ||
141 | packet_map_ref map; |
138 | packet_map_ref map; |
142 | 139 | ||
143 | if( ! packet_is_valid( packet )) return EINVAL; |
140 | if( ! packet_is_valid( packet )) return EINVAL; |
144 | futex_down( & pm_globals.lock ); |
141 | rwlock_write_lock( & pm_globals.lock ); |
145 | if( PACKET_MAP_PAGE( packet->packet_id ) < gpm_count( & pm_globals.packet_map )){ |
142 | if( PACKET_MAP_PAGE( packet->packet_id ) < gpm_count( & pm_globals.packet_map )){ |
146 | map = gpm_get_index( & pm_globals.packet_map, PACKET_MAP_PAGE( packet->packet_id )); |
143 | map = gpm_get_index( & pm_globals.packet_map, PACKET_MAP_PAGE( packet->packet_id )); |
147 | }else{ |
144 | }else{ |
148 | do{ |
145 | do{ |
149 | map = ( packet_map_ref ) malloc( sizeof( packet_map_t )); |
146 | map = ( packet_map_ref ) malloc( sizeof( packet_map_t )); |
150 | if( ! map ){ |
147 | if( ! map ){ |
151 | futex_up( & pm_globals.lock ); |
148 | rwlock_write_unlock( & pm_globals.lock ); |
152 | return ENOMEM; |
149 | return ENOMEM; |
153 | } |
150 | } |
154 | memset( map, 0, sizeof( packet_map_t )); |
151 | memset( map, 0, sizeof( packet_map_t )); |
155 | if(( ERROR_CODE = gpm_add( & pm_globals.packet_map, map )) < 0 ){ |
152 | if(( ERROR_CODE = gpm_add( & pm_globals.packet_map, map )) < 0 ){ |
- | 153 | rwlock_write_unlock( & pm_globals.lock ); |
|
156 | free( map ); |
154 | free( map ); |
157 | futex_up( & pm_globals.lock ); |
- | |
158 | return ERROR_CODE; |
155 | return ERROR_CODE; |
159 | } |
156 | } |
160 | }while( PACKET_MAP_PAGE( packet->packet_id ) >= gpm_count( & pm_globals.packet_map )); |
157 | }while( PACKET_MAP_PAGE( packet->packet_id ) >= gpm_count( & pm_globals.packet_map )); |
161 | } |
158 | } |
162 | ( * map )[ PACKET_MAP_INDEX( packet->packet_id ) ] = packet; |
159 | ( * map )[ PACKET_MAP_INDEX( packet->packet_id ) ] = packet; |
163 | futex_up( & pm_globals.lock ); |
160 | rwlock_write_unlock( & pm_globals.lock ); |
164 | return EOK; |
161 | return EOK; |
165 | } |
162 | } |
166 | 163 | ||
167 | void pm_destroy( void ){ |
164 | void pm_destroy( void ){ |
168 | int count; |
165 | int count; |
169 | int index; |
166 | int index; |
170 | packet_map_ref map; |
167 | packet_map_ref map; |
171 | packet_t packet; |
168 | packet_t packet; |
172 | 169 | ||
173 | futex_down( & pm_globals.lock ); |
170 | rwlock_write_lock( & pm_globals.lock ); |
174 | count = gpm_count( & pm_globals.packet_map ); |
171 | count = gpm_count( & pm_globals.packet_map ); |
175 | while( count > 0 ){ |
172 | while( count > 0 ){ |
176 | map = gpm_get_index( & pm_globals.packet_map, count - 1 ); |
173 | map = gpm_get_index( & pm_globals.packet_map, count - 1 ); |
177 | for( index = PACKET_MAP_SIZE - 1; index >= 0; -- index ){ |
174 | for( index = PACKET_MAP_SIZE - 1; index >= 0; -- index ){ |
178 | packet = ( * map )[ index ]; |
175 | packet = ( * map )[ index ]; |
179 | if( packet_is_valid( packet )){ |
176 | if( packet_is_valid( packet )){ |
180 | munmap( packet, packet->length ); |
177 | munmap( packet, packet->length ); |
181 | } |
178 | } |
182 | } |
179 | } |
183 | } |
180 | } |
184 | gpm_destroy( & pm_globals.packet_map ); |
181 | gpm_destroy( & pm_globals.packet_map ); |
185 | // leave locked |
182 | // leave locked |
186 | } |
183 | } |
187 | 184 | ||
188 | packet_t pq_add( packet_t first, packet_t packet, int order, size_t metric ){ |
185 | packet_t pq_add( packet_t first, packet_t packet, int order, size_t metric ){ |
189 | packet_t item; |
186 | packet_t item; |
190 | 187 | ||
191 | if( ! packet_is_valid( packet )) return NULL; |
188 | if( ! packet_is_valid( packet )) return NULL; |
192 | pq_set( packet, order, metric ); |
189 | pq_set( packet, order, metric ); |
193 | if( packet_is_valid( first )){ |
190 | if( packet_is_valid( first )){ |
194 | item = first; |
191 | item = first; |
195 | do{ |
192 | do{ |
196 | if( item->order < order ){ |
193 | if( item->order < order ){ |
197 | if( item->next ){ |
194 | if( item->next ){ |
198 | item = pm_find( item->next ); |
195 | item = pm_find( item->next ); |
199 | }else{ |
196 | }else{ |
200 | item->next = packet->packet_id; |
197 | item->next = packet->packet_id; |
201 | packet->previous = item->packet_id; |
198 | packet->previous = item->packet_id; |
202 | return first; |
199 | return first; |
203 | } |
200 | } |
204 | }else{ |
201 | }else{ |
205 | packet->previous = item->previous; |
202 | packet->previous = item->previous; |
206 | packet->next = item->packet_id; |
203 | packet->next = item->packet_id; |
207 | item->previous = packet->packet_id; |
204 | item->previous = packet->packet_id; |
208 | item = pm_find( packet->previous ); |
205 | item = pm_find( packet->previous ); |
209 | if( item ) item->next = packet->packet_id; |
206 | if( item ) item->next = packet->packet_id; |
210 | return item; |
207 | return item; |
211 | } |
208 | } |
212 | }while( packet_is_valid( item )); |
209 | }while( packet_is_valid( item )); |
213 | } |
210 | } |
214 | return packet; |
211 | return packet; |
215 | } |
212 | } |
216 | 213 | ||
217 | packet_t pq_detach( packet_t packet ){ |
214 | packet_t pq_detach( packet_t packet ){ |
218 | packet_t next; |
215 | packet_t next; |
219 | packet_t previous; |
216 | packet_t previous; |
220 | 217 | ||
221 | if( ! packet_is_valid( packet )) return NULL; |
218 | if( ! packet_is_valid( packet )) return NULL; |
222 | next = pm_find( packet->next ); |
219 | next = pm_find( packet->next ); |
223 | if( next ){ |
220 | if( next ){ |
224 | next->previous = packet->previous; |
221 | next->previous = packet->previous; |
225 | previous = pm_find( next->previous ); |
222 | previous = pm_find( next->previous ); |
226 | if( previous ){ |
223 | if( previous ){ |
227 | previous->next = next->packet_id; |
224 | previous->next = next->packet_id; |
228 | } |
225 | } |
229 | } |
226 | } |
230 | packet->previous = 0; |
227 | packet->previous = 0; |
231 | packet->next = 0; |
228 | packet->next = 0; |
232 | return next; |
229 | return next; |
233 | } |
230 | } |
234 | 231 | ||
235 | int pq_set( packet_t packet, int order, size_t metric ){ |
232 | int pq_set( packet_t packet, int order, size_t metric ){ |
236 | if( ! packet_is_valid( packet )) return EINVAL; |
233 | if( ! packet_is_valid( packet )) return EINVAL; |
237 | packet->order = order; |
234 | packet->order = order; |
238 | packet->metric = metric; |
235 | packet->metric = metric; |
239 | return EOK; |
236 | return EOK; |
240 | } |
237 | } |
241 | 238 | ||
242 | void pq_destroy( packet_t first, void ( * packet_release )( packet_t packet )){ |
239 | void pq_destroy( packet_t first, void ( * packet_release )( packet_t packet )){ |
243 | packet_t actual; |
240 | packet_t actual; |
244 | packet_t next; |
241 | packet_t next; |
245 | 242 | ||
246 | actual = first; |
243 | actual = first; |
247 | while( packet_is_valid( actual )){ |
244 | while( packet_is_valid( actual )){ |
248 | next = pm_find( actual->next ); |
245 | next = pm_find( actual->next ); |
249 | actual->next = 0; |
246 | actual->next = 0; |
250 | actual->previous = 0; |
247 | actual->previous = 0; |
251 | if( packet_release ) packet_release( actual ); |
248 | if( packet_release ) packet_release( actual ); |
252 | actual = next; |
249 | actual = next; |
253 | } |
250 | } |
254 | } |
251 | } |
255 | 252 | ||
256 | /** @} |
253 | /** @} |
257 | */ |
254 | */ |
258 | 255 |