955e4953595e9e20b2998f101efff6539d34ed56
[strongswan.git] / src / stroke / stroke.c
1 /* Stroke for charon is the counterpart to whack from pluto
2 * Copyright (C) 2007 Tobias Brunner
3 * Copyright (C) 2006 Martin Willi
4 * Hochschule fuer Technik Rapperswil
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; either version 2 of the License, or (at your
9 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
10 *
11 * This program is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * for more details.
15 */
16
17 #include <stdlib.h>
18 #include <sys/types.h>
19 #include <sys/stat.h>
20 #include <sys/socket.h>
21 #include <sys/un.h>
22 #include <sys/fcntl.h>
23 #include <unistd.h>
24 #include <dirent.h>
25 #include <errno.h>
26 #include <stdio.h>
27 #include <stddef.h>
28 #include <string.h>
29
30 #include "stroke_msg.h"
31 #include "stroke_keywords.h"
32
33 struct stroke_token {
34 char *name;
35 stroke_keyword_t kw;
36 };
37
38 static char* push_string(stroke_msg_t *msg, char *string)
39 {
40 unsigned long string_start = msg->length;
41
42 if (string == NULL || msg->length + strlen(string) >= sizeof(stroke_msg_t))
43 {
44 return NULL;
45 }
46 else
47 {
48 msg->length += strlen(string) + 1;
49 strcpy((char*)msg + string_start, string);
50 return (char*)string_start;
51 }
52 }
53
54 static int send_stroke_msg (stroke_msg_t *msg)
55 {
56 struct sockaddr_un ctl_addr;
57 int sock;
58 char buffer[512];
59 int byte_count;
60
61 ctl_addr.sun_family = AF_UNIX;
62 strcpy(ctl_addr.sun_path, STROKE_SOCKET);
63
64 msg->output_verbosity = 1; /* CONTROL */
65
66 sock = socket(AF_UNIX, SOCK_STREAM, 0);
67 if (sock < 0)
68 {
69 fprintf(stderr, "Opening unix socket %s: %s\n", STROKE_SOCKET, strerror(errno));
70 return -1;
71 }
72 if (connect(sock, (struct sockaddr *)&ctl_addr,
73 offsetof(struct sockaddr_un, sun_path) + strlen(ctl_addr.sun_path)) < 0)
74 {
75 fprintf(stderr, "Connect to socket failed: %s\n", strerror(errno));
76 close(sock);
77 return -1;
78 }
79
80 /* send message */
81 if (write(sock, msg, msg->length) != msg->length)
82 {
83 fprintf(stderr, "writing to socket failed: %s\n", strerror(errno));
84 close(sock);
85 return -1;
86 }
87
88 while ((byte_count = read(sock, buffer, sizeof(buffer)-1)) > 0)
89 {
90 buffer[byte_count] = '\0';
91 printf("%s", buffer);
92
93 /* we prompt if we receive the "Passphrase:" magic keyword */
94 if (byte_count >= 12 &&
95 strcmp(buffer + byte_count - 12, "Passphrase:\n") == 0)
96 {
97 if (fgets(buffer, sizeof(buffer), stdin))
98 {
99 if (write(sock, buffer, strlen(buffer)));
100 }
101 }
102 }
103 if (byte_count < 0)
104 {
105 fprintf(stderr, "reading from socket failed: %s\n", strerror(errno));
106 }
107
108 close(sock);
109 return 0;
110 }
111
112 static int add_connection(char *name,
113 char *my_id, char *other_id,
114 char *my_addr, char *other_addr,
115 char *my_nets, char *other_nets)
116 {
117 stroke_msg_t msg;
118
119 memset(&msg, 0, sizeof(msg));
120 msg.length = offsetof(stroke_msg_t, buffer);
121 msg.type = STR_ADD_CONN;
122
123 msg.add_conn.name = push_string(&msg, name);
124 msg.add_conn.ikev2 = 1;
125 msg.add_conn.auth_method = 2;
126 msg.add_conn.mode = 1;
127 msg.add_conn.mobike = 1;
128 msg.add_conn.dpd.action = 1;
129
130 msg.add_conn.me.id = push_string(&msg, my_id);
131 msg.add_conn.me.address = push_string(&msg, my_addr);
132 msg.add_conn.me.subnets = push_string(&msg, my_nets);
133 msg.add_conn.me.sendcert = 1;
134
135 msg.add_conn.other.id = push_string(&msg, other_id);
136 msg.add_conn.other.address = push_string(&msg, other_addr);
137 msg.add_conn.other.subnets = push_string(&msg, other_nets);
138 msg.add_conn.other.sendcert = 1;
139
140 return send_stroke_msg(&msg);
141 }
142
143 static int del_connection(char *name)
144 {
145 stroke_msg_t msg;
146
147 msg.length = offsetof(stroke_msg_t, buffer);
148 msg.type = STR_DEL_CONN;
149 msg.initiate.name = push_string(&msg, name);
150 return send_stroke_msg(&msg);
151 }
152
153 static int initiate_connection(char *name)
154 {
155 stroke_msg_t msg;
156
157 msg.length = offsetof(stroke_msg_t, buffer);
158 msg.type = STR_INITIATE;
159 msg.initiate.name = push_string(&msg, name);
160 return send_stroke_msg(&msg);
161 }
162
163 static int terminate_connection(char *name)
164 {
165 stroke_msg_t msg;
166
167 msg.type = STR_TERMINATE;
168 msg.length = offsetof(stroke_msg_t, buffer);
169 msg.initiate.name = push_string(&msg, name);
170 return send_stroke_msg(&msg);
171 }
172
173 static int terminate_connection_srcip(char *start, char *end)
174 {
175 stroke_msg_t msg;
176
177 msg.type = STR_TERMINATE_SRCIP;
178 msg.length = offsetof(stroke_msg_t, buffer);
179 msg.terminate_srcip.start = push_string(&msg, start);
180 msg.terminate_srcip.end = push_string(&msg, end);
181 return send_stroke_msg(&msg);
182 }
183
184 static int route_connection(char *name)
185 {
186 stroke_msg_t msg;
187
188 msg.type = STR_ROUTE;
189 msg.length = offsetof(stroke_msg_t, buffer);
190 msg.route.name = push_string(&msg, name);
191 return send_stroke_msg(&msg);
192 }
193
194 static int unroute_connection(char *name)
195 {
196 stroke_msg_t msg;
197
198 msg.type = STR_UNROUTE;
199 msg.length = offsetof(stroke_msg_t, buffer);
200 msg.unroute.name = push_string(&msg, name);
201 return send_stroke_msg(&msg);
202 }
203
204 static int show_status(stroke_keyword_t kw, char *connection)
205 {
206 stroke_msg_t msg;
207
208 msg.type = (kw == STROKE_STATUS)? STR_STATUS:STR_STATUS_ALL;
209 msg.length = offsetof(stroke_msg_t, buffer);
210 msg.status.name = push_string(&msg, connection);
211 return send_stroke_msg(&msg);
212 }
213
214 static int list_flags[] = {
215 LIST_PUBKEYS,
216 LIST_CERTS,
217 LIST_CACERTS,
218 LIST_OCSPCERTS,
219 LIST_AACERTS,
220 LIST_ACERTS,
221 LIST_GROUPS,
222 LIST_CAINFOS,
223 LIST_CRLS,
224 LIST_OCSP,
225 LIST_ALGS,
226 LIST_ALL
227 };
228
229 static int list(stroke_keyword_t kw, int utc)
230 {
231 stroke_msg_t msg;
232
233 msg.type = STR_LIST;
234 msg.length = offsetof(stroke_msg_t, buffer);
235 msg.list.utc = utc;
236 msg.list.flags = list_flags[kw - STROKE_LIST_FIRST];
237 return send_stroke_msg(&msg);
238 }
239
240 static int reread_flags[] = {
241 REREAD_SECRETS,
242 REREAD_CACERTS,
243 REREAD_OCSPCERTS,
244 REREAD_AACERTS,
245 REREAD_ACERTS,
246 REREAD_CRLS,
247 REREAD_ALL
248 };
249
250 static int reread(stroke_keyword_t kw)
251 {
252 stroke_msg_t msg;
253
254 msg.type = STR_REREAD;
255 msg.length = offsetof(stroke_msg_t, buffer);
256 msg.reread.flags = reread_flags[kw - STROKE_REREAD_FIRST];
257 return send_stroke_msg(&msg);
258 }
259
260 static int purge_flags[] = {
261 PURGE_OCSP,
262 PURGE_IKE,
263 };
264
265 static int purge(stroke_keyword_t kw)
266 {
267 stroke_msg_t msg;
268
269 msg.type = STR_PURGE;
270 msg.length = offsetof(stroke_msg_t, buffer);
271 msg.purge.flags = purge_flags[kw - STROKE_PURGE_FIRST];
272 return send_stroke_msg(&msg);
273 }
274
275 static int leases(stroke_keyword_t kw, char *pool, char *address)
276 {
277
278 stroke_msg_t msg;
279
280 msg.type = STR_LEASES;
281 msg.length = offsetof(stroke_msg_t, buffer);
282 msg.leases.pool = push_string(&msg, pool);
283 msg.leases.address = push_string(&msg, address);
284 return send_stroke_msg(&msg);
285 }
286
287 static int set_loglevel(char *type, u_int level)
288 {
289 stroke_msg_t msg;
290
291 msg.type = STR_LOGLEVEL;
292 msg.length = offsetof(stroke_msg_t, buffer);
293 msg.loglevel.type = push_string(&msg, type);
294 msg.loglevel.level = level;
295 return send_stroke_msg(&msg);
296 }
297
298 static void exit_error(char *error)
299 {
300 if (error)
301 {
302 fprintf(stderr, "%s\n", error);
303 }
304 exit(-1);
305 }
306
307 static void exit_usage(char *error)
308 {
309 printf("Usage:\n");
310 printf(" Add a connection:\n");
311 printf(" stroke add NAME MY_ID OTHER_ID MY_ADDR OTHER_ADDR\\\n");
312 printf(" MY_NET OTHER_NET MY_NETBITS OTHER_NETBITS\n");
313 printf(" where: ID is any IKEv2 ID \n");
314 printf(" ADDR is a IPv4 address\n");
315 printf(" NET is a IPv4 subnet in CIDR notation\n");
316 printf(" Delete a connection:\n");
317 printf(" stroke delete NAME\n");
318 printf(" where: NAME is a connection name added with \"stroke add\"\n");
319 printf(" Initiate a connection:\n");
320 printf(" stroke up NAME\n");
321 printf(" where: NAME is a connection name added with \"stroke add\"\n");
322 printf(" Terminate a connection:\n");
323 printf(" stroke down NAME\n");
324 printf(" where: NAME is a connection name added with \"stroke add\"\n");
325 printf(" Terminate a connection by remote srcip:\n");
326 printf(" stroke down-srcip START [END]\n");
327 printf(" where: START and optional END define the clients source IP\n");
328 printf(" Set loglevel for a logging type:\n");
329 printf(" stroke loglevel TYPE LEVEL\n");
330 printf(" where: TYPE is any|dmn|mgr|ike|chd|job|cfg|knl|net|enc|lib\n");
331 printf(" LEVEL is -1|0|1|2|3|4\n");
332 printf(" Show connection status:\n");
333 printf(" stroke status\n");
334 printf(" Show list of authority and attribute certificates:\n");
335 printf(" stroke listcacerts|listocspcerts|listaacerts|listacerts\n");
336 printf(" Show list of end entity certificates, ca info records and crls:\n");
337 printf(" stroke listcerts|listcainfos|listcrls|listall\n");
338 printf(" Show list of supported algorithms:\n");
339 printf(" stroke listalgs\n");
340 printf(" Reload authority and attribute certificates:\n");
341 printf(" stroke rereadcacerts|rereadocspcerts|rereadaacerts|rereadacerts\n");
342 printf(" Reload secrets and crls:\n");
343 printf(" stroke rereadsecrets|rereadcrls|rereadall\n");
344 printf(" Purge ocsp cache entries:\n");
345 printf(" stroke purgeocsp\n");
346 printf(" Purge IKE_SAs without a CHILD_SA:\n");
347 printf(" stroke purgeike\n");
348 printf(" Show leases of a pool:\n");
349 printf(" stroke leases [POOL [ADDRESS]]\n");
350 exit_error(error);
351 }
352
353 int main(int argc, char *argv[])
354 {
355 const stroke_token_t *token;
356 int res = 0;
357
358 if (argc < 2)
359 {
360 exit_usage(NULL);
361 }
362
363 token = in_word_set(argv[1], strlen(argv[1]));
364
365 if (token == NULL)
366 {
367 exit_usage("unknown keyword");
368 }
369
370 switch (token->kw)
371 {
372 case STROKE_ADD:
373 if (argc < 11)
374 {
375 exit_usage("\"add\" needs more parameters...");
376 }
377 res = add_connection(argv[2],
378 argv[3], argv[4],
379 argv[5], argv[6],
380 argv[7], argv[8]);
381 break;
382 case STROKE_DELETE:
383 case STROKE_DEL:
384 if (argc < 3)
385 {
386 exit_usage("\"delete\" needs a connection name");
387 }
388 res = del_connection(argv[2]);
389 break;
390 case STROKE_UP:
391 if (argc < 3)
392 {
393 exit_usage("\"up\" needs a connection name");
394 }
395 res = initiate_connection(argv[2]);
396 break;
397 case STROKE_DOWN:
398 if (argc < 3)
399 {
400 exit_usage("\"down\" needs a connection name");
401 }
402 res = terminate_connection(argv[2]);
403 break;
404 case STROKE_DOWN_SRCIP:
405 if (argc < 3)
406 {
407 exit_usage("\"down-srcip\" needs start and optional end address");
408 }
409 res = terminate_connection_srcip(argv[2], argc > 3 ? argv[3] : NULL);
410 break;
411 case STROKE_ROUTE:
412 if (argc < 3)
413 {
414 exit_usage("\"route\" needs a connection name");
415 }
416 res = route_connection(argv[2]);
417 break;
418 case STROKE_UNROUTE:
419 if (argc < 3)
420 {
421 exit_usage("\"unroute\" needs a connection name");
422 }
423 res = unroute_connection(argv[2]);
424 break;
425 case STROKE_LOGLEVEL:
426 if (argc < 4)
427 {
428 exit_usage("\"logtype\" needs more parameters...");
429 }
430 res = set_loglevel(argv[2], atoi(argv[3]));
431 break;
432 case STROKE_STATUS:
433 case STROKE_STATUSALL:
434 res = show_status(token->kw, argc > 2 ? argv[2] : NULL);
435 break;
436 case STROKE_LIST_PUBKEYS:
437 case STROKE_LIST_CERTS:
438 case STROKE_LIST_CACERTS:
439 case STROKE_LIST_OCSPCERTS:
440 case STROKE_LIST_AACERTS:
441 case STROKE_LIST_ACERTS:
442 case STROKE_LIST_CAINFOS:
443 case STROKE_LIST_CRLS:
444 case STROKE_LIST_OCSP:
445 case STROKE_LIST_ALGS:
446 case STROKE_LIST_ALL:
447 res = list(token->kw, argc > 2 && strcmp(argv[2], "--utc") == 0);
448 break;
449 case STROKE_REREAD_SECRETS:
450 case STROKE_REREAD_CACERTS:
451 case STROKE_REREAD_OCSPCERTS:
452 case STROKE_REREAD_AACERTS:
453 case STROKE_REREAD_ACERTS:
454 case STROKE_REREAD_CRLS:
455 case STROKE_REREAD_ALL:
456 res = reread(token->kw);
457 break;
458 case STROKE_PURGE_OCSP:
459 case STROKE_PURGE_IKE:
460 res = purge(token->kw);
461 break;
462 case STROKE_LEASES:
463 res = leases(token->kw, argc > 2 ? argv[2] : NULL,
464 argc > 3 ? argv[3] : NULL);
465 break;
466 default:
467 exit_usage(NULL);
468 }
469 return res;
470 }