Rev 4459 | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
4459 | decky | 1 | /* |
2 | * Copyright (c) 2009 Martin Decky |
||
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 ns |
||
30 | * @{ |
||
31 | */ |
||
32 | |||
33 | #include <ipc/ipc.h> |
||
4509 | decky | 34 | #include <adt/hash_table.h> |
4459 | decky | 35 | #include <assert.h> |
36 | #include <errno.h> |
||
37 | #include "service.h" |
||
38 | #include "ns.h" |
||
39 | |||
40 | #define SERVICE_HASH_TABLE_CHAINS 20 |
||
41 | |||
42 | /** Service hash table item. */ |
||
43 | typedef struct { |
||
44 | link_t link; |
||
45 | ipcarg_t service; /**< Number of the service. */ |
||
46 | ipcarg_t phone; /**< Phone registered with the service. */ |
||
47 | ipcarg_t in_phone_hash; /**< Incoming phone hash. */ |
||
48 | } hashed_service_t; |
||
49 | |||
50 | /** Compute hash index into service hash table. |
||
51 | * |
||
52 | * @param key Pointer keys. However, only the first key (i.e. service number) |
||
53 | * is used to compute the hash index. |
||
54 | * |
||
55 | * @return Hash index corresponding to key[0]. |
||
56 | * |
||
57 | */ |
||
58 | static hash_index_t service_hash(unsigned long *key) |
||
59 | { |
||
60 | assert(key); |
||
61 | return (*key % SERVICE_HASH_TABLE_CHAINS); |
||
62 | } |
||
63 | |||
64 | /** Compare a key with hashed item. |
||
65 | * |
||
66 | * This compare function always ignores the third key. |
||
67 | * It exists only to make it possible to remove records |
||
68 | * originating from connection with key[1] in_phone_hash |
||
69 | * value. Note that this is close to being classified |
||
70 | * as a nasty hack. |
||
71 | * |
||
72 | * @param key Array of keys. |
||
73 | * @param keys Must be lesser or equal to 3. |
||
74 | * @param item Pointer to a hash table item. |
||
75 | * |
||
76 | * @return Non-zero if the key matches the item, zero otherwise. |
||
77 | * |
||
78 | */ |
||
79 | static int service_compare(unsigned long key[], hash_count_t keys, link_t *item) |
||
80 | { |
||
81 | assert(key); |
||
82 | assert(keys <= 3); |
||
83 | assert(item); |
||
84 | |||
85 | hashed_service_t *hs = hash_table_get_instance(item, hashed_service_t, link); |
||
86 | |||
87 | if (keys == 2) |
||
88 | return (key[1] == hs->in_phone_hash); |
||
89 | else |
||
90 | return (key[0] == hs->service); |
||
91 | } |
||
92 | |||
93 | /** Perform actions after removal of item from the hash table. |
||
94 | * |
||
95 | * @param item Item that was removed from the hash table. |
||
96 | * |
||
97 | */ |
||
98 | static void service_remove(link_t *item) |
||
99 | { |
||
100 | assert(item); |
||
101 | free(hash_table_get_instance(item, hashed_service_t, link)); |
||
102 | } |
||
103 | |||
104 | /** Operations for service hash table. */ |
||
105 | static hash_table_operations_t service_hash_table_ops = { |
||
106 | .hash = service_hash, |
||
107 | .compare = service_compare, |
||
108 | .remove_callback = service_remove |
||
109 | }; |
||
110 | |||
111 | /** Service hash table structure. */ |
||
112 | static hash_table_t service_hash_table; |
||
113 | |||
114 | /** Pending connection structure. */ |
||
115 | typedef struct { |
||
116 | link_t link; |
||
117 | ipcarg_t service; /**< Number of the service. */ |
||
118 | ipc_callid_t callid; /**< Call ID waiting for the connection */ |
||
119 | ipcarg_t arg2; /**< Second argument */ |
||
120 | ipcarg_t arg3; /**< Third argument */ |
||
121 | } pending_conn_t; |
||
122 | |||
123 | static link_t pending_conn; |
||
124 | |||
125 | int service_init(void) |
||
126 | { |
||
127 | if (!hash_table_create(&service_hash_table, SERVICE_HASH_TABLE_CHAINS, |
||
128 | 3, &service_hash_table_ops)) { |
||
129 | printf(NAME ": No memory available for services\n"); |
||
130 | return ENOMEM; |
||
131 | } |
||
132 | |||
133 | list_initialize(&pending_conn); |
||
134 | |||
135 | return EOK; |
||
136 | } |
||
137 | |||
138 | /** Process pending connection requests */ |
||
139 | void process_pending_conn(void) |
||
140 | { |
||
141 | link_t *cur; |
||
142 | |||
143 | loop: |
||
144 | for (cur = pending_conn.next; cur != &pending_conn; cur = cur->next) { |
||
145 | pending_conn_t *pr = list_get_instance(cur, pending_conn_t, link); |
||
146 | |||
147 | unsigned long keys[3] = { |
||
148 | pr->service, |
||
149 | 0, |
||
150 | |||
151 | }; |
||
152 | |||
153 | link_t *link = hash_table_find(&service_hash_table, keys); |
||
154 | if (!link) |
||
155 | continue; |
||
156 | |||
157 | hashed_service_t *hs = hash_table_get_instance(link, hashed_service_t, link); |
||
158 | ipcarg_t retval = ipc_forward_fast(pr->callid, hs->phone, |
||
159 | pr->arg2, pr->arg3, 0, IPC_FF_NONE); |
||
160 | |||
161 | if (!(pr->callid & IPC_CALLID_NOTIFICATION)) |
||
162 | ipc_answer_0(pr->callid, retval); |
||
163 | |||
164 | list_remove(cur); |
||
165 | free(pr); |
||
166 | goto loop; |
||
167 | } |
||
168 | } |
||
169 | |||
170 | /** Register service. |
||
171 | * |
||
172 | * @param service Service to be registered. |
||
173 | * @param phone Phone to be used for connections to the service. |
||
174 | * @param call Pointer to call structure. |
||
175 | * |
||
176 | * @return Zero on success or a value from @ref errno.h. |
||
177 | * |
||
178 | */ |
||
179 | int register_service(ipcarg_t service, ipcarg_t phone, ipc_call_t *call) |
||
180 | { |
||
181 | unsigned long keys[3] = { |
||
182 | service, |
||
183 | call->in_phone_hash, |
||
184 | |||
185 | }; |
||
186 | |||
187 | if (hash_table_find(&service_hash_table, keys)) |
||
188 | return EEXISTS; |
||
189 | |||
190 | hashed_service_t *hs = (hashed_service_t *) malloc(sizeof(hashed_service_t)); |
||
191 | if (!hs) |
||
192 | return ENOMEM; |
||
193 | |||
194 | link_initialize(&hs->link); |
||
195 | hs->service = service; |
||
196 | hs->phone = phone; |
||
197 | hs->in_phone_hash = call->in_phone_hash; |
||
198 | hash_table_insert(&service_hash_table, keys, &hs->link); |
||
199 | |||
200 | return 0; |
||
201 | } |
||
202 | |||
203 | /** Connect client to service. |
||
204 | * |
||
205 | * @param service Service to be connected to. |
||
206 | * @param call Pointer to call structure. |
||
207 | * @param callid Call ID of the request. |
||
208 | * |
||
209 | * @return Zero on success or a value from @ref errno.h. |
||
210 | * |
||
211 | */ |
||
212 | void connect_to_service(ipcarg_t service, ipc_call_t *call, ipc_callid_t callid) |
||
213 | { |
||
214 | ipcarg_t retval; |
||
215 | unsigned long keys[3] = { |
||
216 | service, |
||
217 | 0, |
||
218 | |||
219 | }; |
||
220 | |||
221 | link_t *link = hash_table_find(&service_hash_table, keys); |
||
222 | if (!link) { |
||
223 | if (IPC_GET_ARG4(*call) & IPC_FLAG_BLOCKING) { |
||
224 | /* Blocking connection, add to pending list */ |
||
225 | pending_conn_t *pr = |
||
226 | (pending_conn_t *) malloc(sizeof(pending_conn_t)); |
||
227 | if (!pr) { |
||
228 | retval = ENOMEM; |
||
229 | goto out; |
||
230 | } |
||
231 | |||
232 | pr->service = service; |
||
233 | pr->callid = callid; |
||
234 | pr->arg2 = IPC_GET_ARG2(*call); |
||
235 | pr->arg3 = IPC_GET_ARG3(*call); |
||
236 | list_append(&pr->link, &pending_conn); |
||
237 | return; |
||
238 | } |
||
239 | retval = ENOENT; |
||
240 | goto out; |
||
241 | } |
||
242 | |||
243 | hashed_service_t *hs = hash_table_get_instance(link, hashed_service_t, link); |
||
244 | retval = ipc_forward_fast(callid, hs->phone, IPC_GET_ARG2(*call), |
||
245 | IPC_GET_ARG3(*call), 0, IPC_FF_NONE); |
||
246 | |||
247 | out: |
||
248 | if (!(callid & IPC_CALLID_NOTIFICATION)) |
||
249 | ipc_answer_0(callid, retval); |
||
250 | } |
||
251 | |||
252 | /** |
||
253 | * @} |
||
254 | */ |