swanctl: Add a list-sas command to query active IKE_SAs

This commit is contained in:
Martin Willi
2014-05-07 15:48:13 +02:00
parent e381e69f9b
commit 86910faeca
2 changed files with 360 additions and 0 deletions
+1
View File
@@ -2,6 +2,7 @@ sbin_PROGRAMS = swanctl
swanctl_SOURCES = \
command.c command.h \
commands/list_sas.c \
swanctl.c
swanctl_LDADD = \
+359
View File
@@ -0,0 +1,359 @@
/*
* Copyright (C) 2014 Martin Willi
* Copyright (C) 2014 revosec AG
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at your
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* for more details.
*/
#define _GNU_SOURCE
#include <stdio.h>
#include <errno.h>
#include "command.h"
#include <collections/hashtable.h>
/**
* Free hashtable with contained strings
*/
static void free_hashtable(hashtable_t *hashtable)
{
enumerator_t *enumerator;
char *str;
enumerator = hashtable->create_enumerator(hashtable);
while (enumerator->enumerate(enumerator, NULL, &str))
{
free(str);
}
enumerator->destroy(enumerator);
hashtable->destroy(hashtable);
}
CALLBACK(sa_values, int,
hashtable_t *sa, vici_res_t *res, char *name, void *value, int len)
{
chunk_t chunk;
char *str;
chunk = chunk_create(value, len);
if (chunk_printable(chunk, NULL, ' '))
{
if (asprintf(&str, "%.*s", len, value) >= 0)
{
free(sa->put(sa, name, str));
}
}
return 0;
}
CALLBACK(sa_list, int,
hashtable_t *sa, vici_res_t *res, char *name, void *value, int len)
{
chunk_t chunk;
char *str;
chunk = chunk_create(value, len);
if (chunk_printable(chunk, NULL, ' '))
{
str = sa->get(sa, name);
if (asprintf(&str, "%s%s%.*s",
str ?: "", str ? " " : "", len, value) >= 0)
{
free(sa->put(sa, name, str));
}
}
return 0;
}
CALLBACK(child_sas, int,
hashtable_t *ike, vici_res_t *res, char *name)
{
hashtable_t *child;
int ret;
child = hashtable_create(hashtable_hash_str, hashtable_equals_str, 1);
ret = vici_parse_cb(res, NULL, sa_values, sa_list, child);
if (ret == 0)
{
printf(" %s: #%s, %s, %s%s, %s:",
name, child->get(child, "reqid"),
child->get(child, "state"), child->get(child, "mode"),
child->get(child, "encap") ? "-in-UDP" : "",
child->get(child, "protocol"));
if (child->get(child, "encr-alg"))
{
printf("%s", child->get(child, "encr-alg"));
if (child->get(child, "encr-keysize"))
{
printf("-%s", child->get(child, "encr-keysize"));
}
}
if (child->get(child, "integ-alg"))
{
if (child->get(child, "encr-alg"))
{
printf("/");
}
printf("%s", child->get(child, "integ-alg"));
if (child->get(child, "integ-keysize"))
{
printf("-%s", child->get(child, "integ-keysize"));
}
}
if (child->get(child, "prf-alg"))
{
printf("/%s", child->get(child, "prf-alg"));
}
if (child->get(child, "dh-group"))
{
printf("/%s", child->get(child, "dh-group"));
}
if (child->get(child, "esn"))
{
printf("/%s", child->get(child, "esn"));
}
printf("\n");
printf(" installed %s ago", child->get(child, "install-time"));
if (child->get(child, "rekey-time"))
{
printf(", rekeying in %ss", child->get(child, "rekey-time"));
}
if (child->get(child, "life-time"))
{
printf(", expires in %ss", child->get(child, "life-time"));
}
printf("\n");
printf(" in %s%s%s", child->get(child, "spi-in"),
child->get(child, "cpi-in") ? "/" : "",
child->get(child, "cpi-in") ?: "");
printf(", %6s bytes, %5s packets",
child->get(child, "bytes-in"), child->get(child, "packets-in"));
if (child->get(child, "use-in"))
{
printf(", %5ss ago", child->get(child, "use-in"));
}
printf("\n");
printf(" out %s%s%s", child->get(child, "spi-out"),
child->get(child, "cpi-out") ? "/" : "",
child->get(child, "cpi-out") ?: "");
printf(", %6s bytes, %5s packets",
child->get(child, "bytes-out"), child->get(child, "packets-out"));
if (child->get(child, "use-out"))
{
printf(", %5ss ago", child->get(child, "use-out"));
}
printf("\n");
printf(" local %s\n", child->get(child, "local-ts"));
printf(" remote %s\n", child->get(child, "remote-ts"));
}
free_hashtable(child);
return ret;
}
CALLBACK(ike_sa, int,
hashtable_t *ike, vici_res_t *res, char *name)
{
if (streq(name, "child-sas"))
{
printf("%s: #%s, %s, IKEv%s, %s:%s\n",
ike->get(ike, "name"), ike->get(ike, "uniqueid"),
ike->get(ike, "state"), ike->get(ike, "version"),
ike->get(ike, "initiator-spi"), ike->get(ike, "responder-spi"));
printf(" local '%s' @ %s\n",
ike->get(ike, "local-id"), ike->get(ike, "local-host"));
printf(" remote '%s' @ %s",
ike->get(ike, "remote-id"), ike->get(ike, "remote-host"));
if (ike->get(ike, "remote-eap-id"))
{
printf(" EAP: '%s'", ike->get(ike, "remote-eap-id"));
}
if (ike->get(ike, "remote-xauth-id"))
{
printf(" XAuth: '%s'", ike->get(ike, "remote-xauth-id"));
}
printf("\n");
if (ike->get(ike, "encr-alg"))
{
printf(" %s", ike->get(ike, "encr-alg"));
if (ike->get(ike, "encr-keysize"))
{
printf("-%s", ike->get(ike, "encr-keysize"));
}
if (ike->get(ike, "integ-alg"))
{
printf("/%s", ike->get(ike, "integ-alg"));
}
if (ike->get(ike, "integ-keysize"))
{
printf("-%s", ike->get(ike, "integ-keysize"));
}
printf("/%s", ike->get(ike, "prf-alg"));
printf("/%s", ike->get(ike, "dh-group"));
printf("\n");
}
if (ike->get(ike, "established"))
{
printf(" established %s ago", ike->get(ike, "established"));
if (ike->get(ike, "rekey-time"))
{
printf(", rekeying in %ss", ike->get(ike, "rekey-time"));
}
if (ike->get(ike, "reauth-time"))
{
printf(", reauth in %ss", ike->get(ike, "reauth-time"));
}
if (ike->get(ike, "life-time"))
{
printf(", expires in %ss", ike->get(ike, "life-time"));
}
printf("\n");
}
if (ike->get(ike, "tasks-queued"))
{
printf(" queued: %s\n", ike->get(ike, "tasks-queued"));
}
if (ike->get(ike, "tasks-active"))
{
printf(" active: %s\n", ike->get(ike, "tasks-active"));
}
if (ike->get(ike, "tasks-passive"))
{
printf(" passive: %s\n", ike->get(ike, "tasks-passive"));
}
return vici_parse_cb(res, child_sas, NULL, NULL, ike);
}
return 0;
}
CALLBACK(ike_sas, int,
void *null, vici_res_t *res, char *name)
{
hashtable_t *ike;
int ret;
ike = hashtable_create(hashtable_hash_str, hashtable_equals_str, 1);
ike->put(ike, "name", strdup(name));
ret = vici_parse_cb(res, ike_sa, sa_values, sa_list, ike);
free_hashtable(ike);
return ret;
}
CALLBACK(list_cb, void,
bool *raw, char *name, vici_res_t *res)
{
if (*raw)
{
vici_dump(res, "list-sa event", stdout);
}
else
{
if (vici_parse_cb(res, ike_sas, NULL, NULL, NULL) != 0)
{
fprintf(stderr, "parsing SA event failed: %s\n", strerror(errno));
}
}
}
static int list_sas(vici_conn_t *conn)
{
vici_req_t *req;
vici_res_t *res;
bool raw = FALSE, noblock = FALSE;
char *arg, *ike = NULL;
int ike_id = 0;
while (TRUE)
{
switch (command_getopt(&arg))
{
case 'h':
return command_usage(NULL);
case 'i':
ike = arg;
continue;
case 'I':
ike_id = atoi(arg);
continue;
case 'n':
noblock = TRUE;
continue;
case 'r':
raw = TRUE;
continue;
case EOF:
break;
default:
return command_usage("invalid --list-sas option");
}
break;
}
if (vici_register(conn, "list-sa", list_cb, &raw) != 0)
{
fprintf(stderr, "registering for SAs failed: %s\n", strerror(errno));
return errno;
}
req = vici_begin("list-sas");
if (ike)
{
vici_add_key_valuef(req, "ike", "%s", ike);
}
if (ike_id)
{
vici_add_key_valuef(req, "ike-id", "%d", ike_id);
}
if (noblock)
{
vici_add_key_valuef(req, "noblock", "yes");
}
res = vici_submit(req, conn);
if (!res)
{
fprintf(stderr, "list-sas request failed: %s\n", strerror(errno));
return errno;
}
if (raw)
{
vici_dump(res, "list-sas reply", stdout);
}
vici_free_res(res);
return 0;
}
/**
* Register the command.
*/
static void __attribute__ ((constructor))reg()
{
command_register((command_t) {
list_sas, 'l', "list-sas", "list currently active IKE_SAs",
{"[--raw]"},
{
{"help", 'h', 0, "show usage information"},
{"ike", 'i', 1, "filter IKE_SAs by name"},
{"ike-id", 'I', 1, "filter IKE_SAs by unique identifier"},
{"noblock", 'n', 0, "don't wait for IKE_SAs in use"},
{"raw", 'r', 0, "dump raw response message"},
}
});
}