2 * Copyright (c) 2012-2018, JANET(UK)
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of JANET(UK) nor the names of its contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
31 * OF THE POSSIBILITY OF SUCH DAMAGE.
41 #include <mon_internal.h>
44 #include <trust_router/tr_dh.h>
47 /* command-line option setup */
49 /* argp global parameters */
50 const char *argp_program_bug_address=PACKAGE_BUGREPORT; /* bug reporting address */
53 static const char doc[]=PACKAGE_NAME " - TR Monitoring Client";
54 static const char arg_doc[]="<message> <server> [<port>]"; /* string describing arguments, if any */
56 /* define the options here. Fields are:
57 * { long-name, short-name, variable name, options, help description } */
58 static const struct argp_option cmdline_options[] = {
59 { "repeat", 'r', "N", OPTION_ARG_OPTIONAL, "Repeat message until terminated, or N times." },
63 /* structure for communicating with option parser */
67 unsigned int port; /* optional */
68 int repeat; /* how many times to repeat, or -1 for infinite */
71 /* parser for individual options - fills in a struct cmdline_args */
72 static error_t parse_option(int key, char *arg, struct argp_state *state)
76 /* get a shorthand to the command line argument structure, part of state */
77 struct cmdline_args *arguments=state->input;
84 tmp_l = strtol(arg, NULL, 10);
85 if ((errno == 0) && (tmp_l > 0) && (tmp_l < INT_MAX))
86 arguments->repeat = (int) tmp_l;
91 case ARGP_KEY_ARG: /* handle argument (not option) */
92 switch (state->arg_num) {
98 arguments->server=arg;
102 tmp_l = strtol(arg, NULL, 10);
103 if (errno || (tmp_l < 0) || (tmp_l > 65535)) /* max valid port */
106 arguments->port=(unsigned int) tmp_l;
110 /* too many arguments */
115 case ARGP_KEY_END: /* no more arguments */
116 if (state->arg_num < 2) {
117 /* not enough arguments encountered */
123 return ARGP_ERR_UNKNOWN;
126 return 0; /* success */
130 /* assemble the argp parser */
131 static struct argp argp = {cmdline_options, parse_option, arg_doc, doc};
133 int main(int argc, char *argv[])
135 TALLOC_CTX *main_ctx=talloc_new(NULL);
136 MONC_INSTANCE *monc = NULL;
138 MON_RESP *resp = NULL;
140 struct cmdline_args opts;
141 int retval=1; /* exit with an error status unless this gets set to zero */
143 /* parse the command line*/
150 argp_parse(&argp, argc, argv, 0, 0, &opts);
152 /* Use standalone logging */
155 /* set logging levels */
156 talloc_set_log_stderr();
157 tr_log_threshold(LOG_CRIT);
158 tr_console_threshold(LOG_DEBUG);
160 printf("TR Monitor:\nServer = %s, port = %i\n", opts.server, opts.port);
162 /* Create a MON client instance & the client DH */
163 monc = monc_new(main_ctx);
165 printf("Error allocating client instance.\n");
170 /* fill in the DH parameters */
171 monc_set_dh(monc, tr_create_dh_params(NULL, 0));
172 if (monc_get_dh(monc) == NULL) {
173 printf("Error creating client DH params.\n");
177 /* Set-up MON connection */
178 if (0 != monc_open_connection(monc, opts.server, opts.port)) {
180 printf("Error opening connection to %s:%d.\n", opts.server, opts.port);
184 req = mon_req_new(main_ctx, MON_CMD_SHOW);
186 /* Send a MON request and get the response */
187 resp = monc_send_request(main_ctx, monc, req);
191 printf("Error executing monitoring request.\n");
198 /* Clean-up the MON client instance, and exit */
200 talloc_free(main_ctx);