- policies contain a connections name now
[strongswan.git] / src / stroke / stroke.c
1 /* Stroke for charon is the counterpart to whack from pluto
2 * Copyright (C) 2006 Martin Willi - Hochschule fuer Technik Rapperswil
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License as published by the
6 * Free Software Foundation; either version 2 of the License, or (at your
7 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * for more details.
13 */
14
15 #include <stdlib.h>
16 #include <sys/types.h>
17 #include <sys/stat.h>
18 #include <sys/socket.h>
19 #include <sys/un.h>
20 #include <sys/fcntl.h>
21 #include <unistd.h>
22 #include <dirent.h>
23 #include <errno.h>
24 #include <stdio.h>
25 #include <linux/stddef.h>
26
27 #include "stroke.h"
28
29 #define streq(a, b) (strcmp((a), (b)) == 0) /* clearer shorthand */
30
31 static char* push_string(stroke_msg_t *msg, char *string)
32 {
33 u_int string_start = msg->length;
34
35 if (string == NULL || msg->length + strlen(string) >= sizeof(stroke_msg_t))
36 {
37 return NULL;
38 }
39 else
40 {
41 msg->length += strlen(string) + 1;
42 strcpy((char*)msg + string_start, string);
43 return (char*)string_start;
44 }
45 }
46
47 static int send_stroke_msg (stroke_msg_t *msg)
48 {
49 struct sockaddr_un ctl_addr = { AF_UNIX, STROKE_SOCKET };
50 int sock;
51 char buffer[64];
52 int byte_count;
53
54 sock = socket(AF_UNIX, SOCK_STREAM, 0);
55 if (sock < 0)
56 {
57 fprintf(stderr, "Opening unix socket %s: %s\n", STROKE_SOCKET, strerror(errno));
58 return -1;
59 }
60 if (connect(sock, (struct sockaddr *)&ctl_addr,
61 offsetof(struct sockaddr_un, sun_path) + strlen(ctl_addr.sun_path)) < 0)
62 {
63 fprintf(stderr, "Connect to socket failed: %s\n", strerror(errno));
64 close(sock);
65 return -1;
66 }
67
68 /* send message */
69 if (write(sock, msg, msg->length) != msg->length)
70 {
71 fprintf(stderr, "writing to socket failed: %s\n", strerror(errno));
72 close(sock);
73 return -1;
74 }
75
76 while ((byte_count = read(sock, buffer, sizeof(buffer)-1)) > 0)
77 {
78 buffer[byte_count] = '\0';
79 printf("%s", buffer);
80 }
81 if (byte_count < 0)
82 {
83 fprintf(stderr, "reading from socket failed: %s\n", strerror(errno));
84 }
85
86 close(sock);
87 return 0;
88 }
89
90 static int add_connection(char *name,
91 char *my_id, char *other_id,
92 char *my_addr, char *other_addr,
93 char *my_net, char *other_net,
94 u_int my_netmask, u_int other_netmask)
95 {
96 stroke_msg_t msg;
97
98 msg.length = offsetof(stroke_msg_t, buffer);
99 msg.type = STR_ADD_CONN;
100
101 msg.add_conn.name = push_string(&msg, name);
102 msg.add_conn.ikev2 = 1;
103
104 msg.add_conn.me.id = push_string(&msg, my_id);
105 msg.add_conn.me.address = push_string(&msg, my_addr);
106 msg.add_conn.me.subnet = push_string(&msg, my_net);
107 msg.add_conn.me.subnet_mask = my_netmask;
108 msg.add_conn.me.cert = NULL;
109
110 msg.add_conn.other.id = push_string(&msg, other_id);
111 msg.add_conn.other.address = push_string(&msg, other_addr);
112 msg.add_conn.other.subnet = push_string(&msg, other_net);
113 msg.add_conn.other.subnet_mask = other_netmask;
114 msg.add_conn.other.cert = NULL;
115
116 return send_stroke_msg(&msg);
117 }
118
119 static int del_connection(char *name)
120 {
121 stroke_msg_t msg;
122
123 msg.length = offsetof(stroke_msg_t, buffer);
124 msg.type = STR_DEL_CONN;
125 msg.initiate.name = push_string(&msg, name);
126 return send_stroke_msg(&msg);
127 }
128
129 static int initiate_connection(char *name)
130 {
131 stroke_msg_t msg;
132
133 msg.length = offsetof(stroke_msg_t, buffer);
134 msg.type = STR_INITIATE;
135 msg.initiate.name = push_string(&msg, name);
136 return send_stroke_msg(&msg);
137 }
138
139 static int terminate_connection(char *name)
140 {
141 stroke_msg_t msg;
142
143 msg.type = STR_TERMINATE;
144 msg.length = offsetof(stroke_msg_t, buffer);
145 msg.initiate.name = push_string(&msg, name);
146 return send_stroke_msg(&msg);
147 }
148
149 static int show_status(char *mode, char *connection)
150 {
151 stroke_msg_t msg;
152
153 if (strcmp(mode, "statusall") == 0)
154 msg.type = STR_STATUS_ALL;
155 else
156 msg.type = STR_STATUS;
157
158 msg.length = offsetof(stroke_msg_t, buffer);
159 msg.status.name = push_string(&msg, connection);
160 return send_stroke_msg(&msg);
161 }
162
163 static int list_certs(void)
164 {
165 stroke_msg_t msg;
166
167 msg.type = STR_LIST_CERTS;
168 msg.length = offsetof(stroke_msg_t, buffer);
169 return send_stroke_msg(&msg);
170 }
171
172 static int set_logtype(char *context, char *type, int enable)
173 {
174 stroke_msg_t msg;
175
176 msg.type = STR_LOGTYPE;
177 msg.length = offsetof(stroke_msg_t, buffer);
178 msg.logtype.context = push_string(&msg, context);
179 msg.logtype.type = push_string(&msg, type);
180 msg.logtype.enable = enable;
181 return send_stroke_msg(&msg);
182 }
183
184 static int set_loglevel(char *context, u_int level)
185 {
186 stroke_msg_t msg;
187
188 msg.type = STR_LOGLEVEL;
189 msg.length = offsetof(stroke_msg_t, buffer);
190 msg.loglevel.context = push_string(&msg, context);
191 msg.loglevel.level = level;
192 return send_stroke_msg(&msg);
193 }
194
195 static void exit_error(char *error)
196 {
197 if (error)
198 {
199 fprintf(stderr, "%s\n", error);
200 }
201 exit(-1);
202 }
203
204 static void exit_usage(char *error)
205 {
206 printf("Usage:\n");
207 printf(" Add a connection:\n");
208 printf(" stroke add NAME MY_ID OTHER_ID MY_ADDR OTHER_ADDR\\\n");
209 printf(" MY_NET OTHER_NET MY_NETBITS OTHER_NETBITS\n");
210 printf(" where: ID is any IKEv2 ID \n");
211 printf(" ADDR is a IPv4 address\n");
212 printf(" NET is a IPv4 address of the subnet to tunnel\n");
213 printf(" NETBITS is the size of the subnet, as the \"24\" in 192.168.0.0/24\n");
214 printf(" Delete a connection:\n");
215 printf(" stroke delete NAME\n");
216 printf(" where: NAME is a connection name added with \"stroke add\"\n");
217 printf(" Initiate a connection:\n");
218 printf(" stroke up NAME\n");
219 printf(" where: NAME is a connection name added with \"stroke add\"\n");
220 printf(" Terminate a connection:\n");
221 printf(" stroke down NAME\n");
222 printf(" where: NAME is a connection name added with \"stroke add\"\n");
223 printf(" Set logtype for a logging context:\n");
224 printf(" stroke logtype CONTEXT TYPE ENABLE\n");
225 printf(" where: CONTEXT is PARSR|GNRAT|IKESA|SAMGR|CHDSA|MESSG|TPOOL|WORKR|SCHED|\n");
226 printf(" SENDR|RECVR|SOCKT|TESTR|DAEMN|CONFG|ENCPL|PAYLD\n");
227 printf(" TYPE is CONTROL|ERROR|AUDIT|RAW|PRIVATE\n");
228 printf(" ENABLE is 0|1\n");
229 printf(" Set loglevel for a logging context:\n");
230 printf(" stroke loglevel CONTEXT LEVEL\n");
231 printf(" where: CONTEXT is PARSR|GNRAT|IKESA|SAMGR|CHDSA|MESSG|TPOOL|WORKR|SCHED|\n");
232 printf(" SENDR|RECVR|SOCKT|TESTR|DAEMN|CONFG|ENCPL|PAYLD\n");
233 printf(" LEVEL is 0|1|2|3\n");
234 printf(" Show connection status:\n");
235 printf(" stroke status\n");
236 printf(" Show list of locally loaded certificates:\n");
237 printf(" stroke listcerts\n");
238 exit_error(error);
239 }
240
241 int main(int argc, char *argv[])
242 {
243 int res = 0;
244 char *op;
245
246 if (argc < 2)
247 {
248 exit_usage(NULL);
249 }
250
251 op = argv[1];
252
253 if (streq(op, "status") || streq(op, "statusall"))
254 {
255 res = show_status(op, argc > 2 ? argv[2] : NULL);
256 }
257 else if (streq(op, "listcerts") || streq(op, "listall"))
258 {
259 res = list_certs();
260 }
261 else if (streq(op, "up"))
262 {
263 if (argc < 3)
264 {
265 exit_usage("\"up\" needs a connection name");
266 }
267 res = initiate_connection(argv[2]);
268 }
269 else if (streq(op, "down"))
270 {
271 if (argc < 3)
272 {
273 exit_usage("\"down\" needs a connection name");
274 }
275 res = terminate_connection(argv[2]);
276 }
277 else if (streq(op, "add"))
278 {
279 if (argc < 11)
280 {
281 exit_usage("\"add\" needs more parameters...");
282 }
283 res = add_connection(argv[2],
284 argv[3], argv[4],
285 argv[5], argv[6],
286 argv[7], argv[8],
287 atoi(argv[9]), atoi(argv[10]));
288 }
289 else if (streq(op, "delete"))
290 {
291 if (argc < 3)
292 {
293 exit_usage("\"delete\" needs a connection name");
294 }
295 res = del_connection(argv[2]);
296 }
297 else if (streq(op, "logtype"))
298 {
299 if (argc < 5)
300 {
301 exit_usage("\"logtype\" needs more parameters...");
302 }
303 res = set_logtype(argv[2], argv[3], atoi(argv[4]));
304 }
305 else if (streq(op, "loglevel"))
306 {
307 if (argc < 4)
308 {
309 exit_usage("\"logtype\" needs more parameters...");
310 }
311 res = set_loglevel(argv[2], atoi(argv[3]));
312 }
313 else
314 {
315 exit_usage(NULL);
316 }
317
318 return res;
319 }