Return to counters.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / swanctl / commands

Annotation of embedaddon/strongswan/src/swanctl/commands/counters.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2017 Tobias Brunner
                      3:  * HSR Hochschule fuer Technik Rapperswil
                      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: #define _GNU_SOURCE
                     17: #include <stdio.h>
                     18: #include <errno.h>
                     19: 
                     20: #include "command.h"
                     21: 
                     22: CALLBACK(counters_kv, int,
                     23:        void *null, vici_res_t *res, char *name, void *value, int len)
                     24: {
                     25:        if (chunk_printable(chunk_create(value, len), NULL, ' '))
                     26:        {
                     27:                printf("  %-22s: %.*s\n", name, len, value);
                     28:        }
                     29:        return 0;
                     30: }
                     31: 
                     32: CALLBACK(conns_sn, int,
                     33:        void *null, vici_res_t *res, char *name)
                     34: {
                     35:        printf("%s:\n", strlen(name) ? name : "global");
                     36:        return vici_parse_cb(res, NULL, counters_kv, NULL, NULL);
                     37: }
                     38: 
                     39: CALLBACK(counters_sn, int,
                     40:        void *null, vici_res_t *res, char *name)
                     41: {
                     42:        return vici_parse_cb(res, conns_sn, NULL, NULL, NULL);
                     43: }
                     44: 
                     45: static int counters(vici_conn_t *conn)
                     46: {
                     47:        vici_req_t *req;
                     48:        vici_res_t *res;
                     49:        command_format_options_t format = COMMAND_FORMAT_NONE;
                     50:        char *arg, *name = NULL;
                     51:        int ret = 0;
                     52:        bool all = FALSE, reset = FALSE;
                     53: 
                     54:        while (TRUE)
                     55:        {
                     56:                switch (command_getopt(&arg))
                     57:                {
                     58:                        case 'h':
                     59:                                return command_usage(NULL);
                     60:                        case 'P':
                     61:                                format |= COMMAND_FORMAT_PRETTY;
                     62:                                /* fall through to raw */
                     63:                        case 'r':
                     64:                                format |= COMMAND_FORMAT_RAW;
                     65:                                continue;
                     66:                        case 'n':
                     67:                                name = arg;
                     68:                                continue;
                     69:                        case 'a':
                     70:                                all = TRUE;
                     71:                                continue;
                     72:                        case 'R':
                     73:                                reset = TRUE;
                     74:                                continue;
                     75:                        case EOF:
                     76:                                break;
                     77:                        default:
                     78:                                return command_usage("invalid --counters option");
                     79:                }
                     80:                break;
                     81:        }
                     82:        if (reset)
                     83:        {
                     84:                req = vici_begin("reset-counters");
                     85:        }
                     86:        else
                     87:        {
                     88:                req = vici_begin("get-counters");
                     89:        }
                     90:        if (all)
                     91:        {
                     92:                vici_add_key_valuef(req, "all", "yes");
                     93:        }
                     94:        else if (name)
                     95:        {
                     96:                vici_add_key_valuef(req, "name", "%s", name);
                     97:        }
                     98: 
                     99:        res = vici_submit(req, conn);
                    100:        if (!res)
                    101:        {
                    102:                ret = errno;
                    103:                fprintf(stderr, "%s-counters request failed: %s\n",
                    104:                                reset ? "reset" : "get", strerror(errno));
                    105:                return ret;
                    106:        }
                    107:        if (format & COMMAND_FORMAT_RAW)
                    108:        {
                    109:                vici_dump(res, "counters reply", format & COMMAND_FORMAT_PRETTY,
                    110:                                  stdout);
                    111:        }
                    112:        else
                    113:        {
                    114:                if (streq(vici_find_str(res, "no", "success"), "yes"))
                    115:                {
                    116:                        if (reset)
                    117:                        {
                    118:                                printf("reset-counters completed successfully\n");
                    119:                        }
                    120:                        else if (vici_parse_cb(res, counters_sn, NULL, NULL, NULL) != 0)
                    121:                        {
                    122:                                fprintf(stderr, "parsing get-counters reply failed: %s\n",
                    123:                                                strerror(errno));
                    124:                        }
                    125:                }
                    126:                else
                    127:                {
                    128:                        fprintf(stderr, "%s-counters failed: %s\n", reset ? "reset" : "get",
                    129:                                        vici_find_str(res, "", "errmsg"));
                    130:                        ret = 1;
                    131:                }
                    132:        }
                    133:        vici_free_res(res);
                    134:        return ret;
                    135: }
                    136: 
                    137: /**
                    138:  * Register the command.
                    139:  */
                    140: static void __attribute__ ((constructor))reg()
                    141: {
                    142:        command_register((command_t) {
                    143:                counters, 'C', "counters", "list or reset IKE event counters",
                    144:                {"[--name <name>|--all] [--reset] [--raw|--pretty]"},
                    145:                {
                    146:                        {"help",                'h', 0, "show usage information"},
                    147:                        {"name",                'n', 1, "connection name, omit for global counters"},
                    148:                        {"all",                 'a', 0, "get/reset counters for all tracked connections"},
                    149:                        {"reset",               'R', 0, "reset the counters"},
                    150:                        {"raw",                 'r', 0, "dump raw response message"},
                    151:                        {"pretty",              'P', 0, "dump raw response message in pretty print"},
                    152:                }
                    153:        });
                    154: }