xpc: support termination of IKE_SAs using XPC RPC on connection channel
[strongswan.git] / src / frontends / osx / charon-xpc / xpc_channels.c
1 /*
2 * Copyright (C) 2013 Martin Willi
3 * Copyright (C) 2013 revosec AG
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13 * for more details.
14 */
15
16 #include "xpc_channels.h"
17 #include "xpc_logger.h"
18
19 #include <credentials/sets/callback_cred.h>
20 #include <collections/hashtable.h>
21 #include <threading/rwlock.h>
22 #include <daemon.h>
23
24 typedef struct private_xpc_channels_t private_xpc_channels_t;
25
26 /**
27 * Private data of an xpc_channels_t object.
28 */
29 struct private_xpc_channels_t {
30
31 /**
32 * Public xpc_channels_t interface.
33 */
34 xpc_channels_t public;
35
36 /**
37 * Registered channels, IKE_SA unique ID => entry_t
38 */
39 hashtable_t *channels;
40
41 /**
42 * Lock for channels list
43 */
44 rwlock_t *lock;
45
46 /**
47 * Callback credential set for passwords
48 */
49 callback_cred_t *creds;
50 };
51
52 /**
53 * Channel entry
54 */
55 typedef struct {
56 /* XPC channel to App */
57 xpc_connection_t conn;
58 /* associated IKE_SA unique identifier */
59 uintptr_t sa;
60 /* did we already ask for a password? */
61 bool passworded;
62 /* channel specific logger */
63 xpc_logger_t *logger;
64 } entry_t;
65
66 /**
67 * Clean up an entry, cancelling connection
68 */
69 static void destroy_entry(entry_t *entry)
70 {
71 charon->bus->remove_logger(charon->bus, &entry->logger->logger);
72 entry->logger->destroy(entry->logger);
73 xpc_connection_suspend(entry->conn);
74 xpc_release(entry->conn);
75 free(entry);
76 }
77
78 /**
79 * Find an IKE_SA unique identifier by a given XPC channel
80 */
81 static u_int32_t find_ike_sa_by_conn(private_xpc_channels_t *this,
82 xpc_connection_t conn)
83 {
84 enumerator_t *enumerator;
85 entry_t *entry;
86 u_int32_t ike_sa = 0;
87
88 this->lock->read_lock(this->lock);
89 enumerator = this->channels->create_enumerator(this->channels);
90 while (enumerator->enumerate(enumerator, NULL, &entry))
91 {
92 if (entry->conn == conn)
93 {
94 ike_sa = entry->sa;
95 break;
96 }
97 }
98 enumerator->destroy(enumerator);
99 this->lock->unlock(this->lock);
100
101 return ike_sa;
102 }
103
104 /**
105 * Remove an entry for a given XPC connection
106 */
107 static void remove_conn(private_xpc_channels_t *this, xpc_connection_t conn)
108 {
109 uintptr_t ike_sa;
110 entry_t *entry;
111
112 ike_sa = find_ike_sa_by_conn(this, conn);
113 if (ike_sa)
114 {
115 this->lock->write_lock(this->lock);
116 entry = this->channels->remove(this->channels, (void*)ike_sa);
117 this->lock->unlock(this->lock);
118
119 if (entry)
120 {
121 destroy_entry(entry);
122 }
123 }
124 }
125
126 /**
127 * Trigger termination of a connection
128 */
129 static void stop_connection(private_xpc_channels_t *this, u_int32_t ike_sa,
130 xpc_object_t request, xpc_object_t reply)
131 {
132 status_t status;
133
134 status = charon->controller->terminate_ike(charon->controller, ike_sa,
135 NULL, NULL, 0);
136 xpc_dictionary_set_bool(reply, "success", status != NOT_FOUND);
137 }
138
139 /**
140 * XPC RPC command dispatch table
141 */
142 static struct {
143 char *name;
144 void (*handler)(private_xpc_channels_t *this, u_int32_t ike_sa,
145 xpc_object_t request, xpc_object_t reply);
146 } commands[] = {
147 { "stop_connection", stop_connection },
148 };
149
150 /**
151 * Handle a request message from App
152 */
153 static void handle(private_xpc_channels_t *this, xpc_connection_t conn,
154 xpc_object_t request)
155 {
156 xpc_object_t reply;
157 const char *type, *rpc;
158 bool found = FALSE;
159 u_int32_t ike_sa;
160 int i;
161
162 type = xpc_dictionary_get_string(request, "type");
163 if (type)
164 {
165 if (streq(type, "rpc"))
166 {
167 reply = xpc_dictionary_create_reply(request);
168 rpc = xpc_dictionary_get_string(request, "rpc");
169 ike_sa = find_ike_sa_by_conn(this, conn);
170 if (reply && rpc && ike_sa)
171 {
172 for (i = 0; i < countof(commands); i++)
173 {
174 if (streq(commands[i].name, rpc))
175 {
176 found = TRUE;
177 commands[i].handler(this, ike_sa, request, reply);
178 break;
179 }
180 }
181 }
182 if (!found)
183 {
184 DBG1(DBG_CFG, "received invalid XPC rpc command: %s", rpc);
185 }
186 if (reply)
187 {
188 xpc_connection_send_message(conn, reply);
189 xpc_release(reply);
190 }
191 }
192 else
193 {
194 DBG1(DBG_CFG, "received unknown XPC message type: %s", type);
195 }
196 }
197 else
198 {
199 DBG1(DBG_CFG, "received XPC message without a type");
200 }
201 }
202
203 METHOD(xpc_channels_t, add, void,
204 private_xpc_channels_t *this, xpc_connection_t conn, u_int32_t ike_sa)
205 {
206 entry_t *entry;
207
208 INIT(entry,
209 .conn = conn,
210 .sa = ike_sa,
211 .logger = xpc_logger_create(conn),
212 );
213
214 xpc_connection_set_event_handler(entry->conn, ^(xpc_object_t event)
215 {
216 if (event == XPC_ERROR_CONNECTION_INVALID ||
217 event == XPC_ERROR_CONNECTION_INTERRUPTED)
218 {
219 remove_conn(this, conn);
220 }
221 else if (xpc_get_type(event) == XPC_TYPE_DICTIONARY)
222 {
223 handle(this, conn, event);
224 }
225 });
226
227 entry->logger->set_ike_sa(entry->logger, entry->sa);
228 charon->bus->add_logger(charon->bus, &entry->logger->logger);
229
230 this->lock->write_lock(this->lock);
231 this->channels->put(this->channels, (void*)entry->sa, entry);
232 this->lock->unlock(this->lock);
233
234 xpc_connection_resume(conn);
235 }
236
237 METHOD(listener_t, ike_rekey, bool,
238 private_xpc_channels_t *this, ike_sa_t *old, ike_sa_t *new)
239 {
240 entry_t *entry;
241 uintptr_t sa;
242
243 sa = old->get_unique_id(old);
244 this->lock->write_lock(this->lock);
245 entry = this->channels->remove(this->channels, (void*)sa);
246 if (entry)
247 {
248 entry->sa = new->get_unique_id(new);
249 entry->logger->set_ike_sa(entry->logger, entry->sa);
250 this->channels->put(this->channels, (void*)entry->sa, entry);
251 }
252 this->lock->unlock(this->lock);
253
254 return TRUE;
255 }
256
257 METHOD(listener_t, ike_updown, bool,
258 private_xpc_channels_t *this, ike_sa_t *ike_sa, bool up)
259 {
260 xpc_object_t msg;
261 entry_t *entry;
262 uintptr_t sa;
263
264 sa = ike_sa->get_unique_id(ike_sa);
265 if (up)
266 {
267 this->lock->read_lock(this->lock);
268 entry = this->channels->get(this->channels, (void*)sa);
269 if (entry)
270 {
271 msg = xpc_dictionary_create(NULL, NULL, 0);
272 xpc_dictionary_set_string(msg, "type", "event");
273 xpc_dictionary_set_string(msg, "event", "up");
274 xpc_connection_send_message(entry->conn, msg);
275 xpc_release(msg);
276 }
277 this->lock->unlock(this->lock);
278 }
279 else
280 {
281 this->lock->write_lock(this->lock);
282 entry = this->channels->remove(this->channels, (void*)sa);
283 this->lock->unlock(this->lock);
284 if (entry)
285 {
286 msg = xpc_dictionary_create(NULL, NULL, 0);
287 xpc_dictionary_set_string(msg, "type", "event");
288 xpc_dictionary_set_string(msg, "event", "down");
289 xpc_connection_send_message(entry->conn, msg);
290 xpc_release(msg);
291 xpc_connection_send_barrier(entry->conn, ^() {
292 destroy_entry(entry);
293 });
294 }
295 }
296 return TRUE;
297 }
298
299 /**
300 * Query password from App using XPC channel
301 */
302 static shared_key_t *query_password(xpc_connection_t conn, identification_t *id)
303 {
304 char buf[128], *password;
305 xpc_object_t request, response;
306 shared_key_t *shared = NULL;
307
308 request = xpc_dictionary_create(NULL, NULL, 0);
309 xpc_dictionary_set_string(request, "type", "rpc");
310 xpc_dictionary_set_string(request, "rpc", "get_password");
311 snprintf(buf, sizeof(buf), "%Y", id);
312 xpc_dictionary_set_string(request, "username", buf);
313
314 response = xpc_connection_send_message_with_reply_sync(conn, request);
315 xpc_release(request);
316 if (xpc_get_type(response) == XPC_TYPE_DICTIONARY)
317 {
318 password = (char*)xpc_dictionary_get_string(response, "password");
319 if (password)
320 {
321 shared = shared_key_create(SHARED_EAP,
322 chunk_clone(chunk_from_str(password)));
323 }
324 }
325 xpc_release(response);
326 return shared;
327 }
328
329 /**
330 * Password query callback
331 */
332 static shared_key_t* password_cb(private_xpc_channels_t *this,
333 shared_key_type_t type,
334 identification_t *me, identification_t *other,
335 id_match_t *match_me, id_match_t *match_other)
336 {
337 shared_key_t *shared = NULL;
338 ike_sa_t *ike_sa;
339 entry_t *entry;
340 u_int32_t sa;
341
342 switch (type)
343 {
344 case SHARED_EAP:
345 break;
346 default:
347 return NULL;
348 }
349 ike_sa = charon->bus->get_sa(charon->bus);
350 if (ike_sa)
351 {
352 sa = ike_sa->get_unique_id(ike_sa);
353 this->lock->read_lock(this->lock);
354 entry = this->channels->get(this->channels, (void*)sa);
355 if (entry && !entry->passworded)
356 {
357 entry->passworded = TRUE;
358
359 shared = query_password(entry->conn, me);
360 if (shared)
361 {
362 if (match_me)
363 {
364 *match_me = ID_MATCH_PERFECT;
365 }
366 if (match_other)
367 {
368 *match_other = ID_MATCH_PERFECT;
369 }
370 }
371 }
372 this->lock->unlock(this->lock);
373 }
374 return shared;
375 }
376
377 METHOD(xpc_channels_t, destroy, void,
378 private_xpc_channels_t *this)
379 {
380 lib->credmgr->remove_set(lib->credmgr, &this->creds->set);
381 this->creds->destroy(this->creds);
382 this->channels->destroy(this->channels);
383 this->lock->destroy(this->lock);
384 free(this);
385 }
386
387 /**
388 * See header
389 */
390 xpc_channels_t *xpc_channels_create()
391 {
392 private_xpc_channels_t *this;
393
394 INIT(this,
395 .public = {
396 .listener = {
397 .ike_updown = _ike_updown,
398 .ike_rekey = _ike_rekey,
399 },
400 .add = _add,
401 .destroy = _destroy,
402 },
403 .channels = hashtable_create(hashtable_hash_ptr,
404 hashtable_equals_ptr, 4),
405 .lock = rwlock_create(RWLOCK_TYPE_DEFAULT),
406 );
407
408 this->creds = callback_cred_create_shared(
409 (callback_cred_shared_cb_t)password_cb, this);
410 lib->credmgr->add_set(lib->credmgr, &this->creds->set);
411
412 return &this->public;
413 }