]> git.proxmox.com Git - mirror_frr.git/blame - vtysh/vtysh_main.c
Zebra: Fix multiple RNH deletes
[mirror_frr.git] / vtysh / vtysh_main.c
CommitLineData
718e3744 1/* Virtual terminal interface shell.
2 * Copyright (C) 2000 Kunihiro Ishiguro
3 *
4 * This file is part of GNU Zebra.
5 *
6 * GNU Zebra 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, or (at your option) any
9 * later version.
10 *
11 * GNU Zebra is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with GNU Zebra; see the file COPYING. If not, write to the Free
18 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
19 * 02111-1307, USA.
20 */
21
22#include <zebra.h>
23
24#include <sys/un.h>
25#include <setjmp.h>
26#include <sys/wait.h>
27#include <pwd.h>
28
29#include <readline/readline.h>
30#include <readline/history.h>
31
5e4fa164 32#include <lib/version.h>
718e3744 33#include "getopt.h"
34#include "command.h"
f366ad31 35#include "memory.h"
718e3744 36
37#include "vtysh/vtysh.h"
38#include "vtysh/vtysh_user.h"
b094d260 39
718e3744 40/* VTY shell program name. */
41char *progname;
42
67e29abc 43/* Configuration file name and directory. */
718e3744 44char config_default[] = SYSCONFDIR VTYSH_DEFAULT_CONFIG;
3f4ab7f9 45char history_file[MAXPATHLEN];
718e3744 46
718e3744 47/* Flag for indicate executing child command. */
48int execute_flag = 0;
49
50/* For sigsetjmp() & siglongjmp(). */
51static sigjmp_buf jmpbuf;
52
53/* Flag for avoid recursive siglongjmp() call. */
54static int jmpflag = 0;
55
56/* A static variable for holding the line. */
57static char *line_read;
58
59/* Master of threads. */
60struct thread_master *master;
b094d260 61
57fb9748
SH
62/* Command logging */
63FILE *logfile;
64
718e3744 65/* SIGTSTP handler. This function care user's ^Z input. */
66void
67sigtstp (int sig)
68{
69 /* Execute "end" command. */
70 vtysh_execute ("end");
71
72 /* Initialize readline. */
73 rl_initialize ();
74 printf ("\n");
75
76 /* Check jmpflag for duplicate siglongjmp(). */
77 if (! jmpflag)
78 return;
79
80 jmpflag = 0;
81
82 /* Back to main command loop. */
83 siglongjmp (jmpbuf, 1);
84}
85
86/* SIGINT handler. This function care user's ^Z input. */
87void
88sigint (int sig)
89{
90 /* Check this process is not child process. */
91 if (! execute_flag)
92 {
93 rl_initialize ();
94 printf ("\n");
95 rl_forced_update_display ();
96 }
97}
98
e42f5a37 99/* Signale wrapper for vtysh. We don't use sigevent because
100 * vtysh doesn't use threads. TODO */
718e3744 101RETSIGTYPE *
e42f5a37 102vtysh_signal_set (int signo, void (*func)(int))
718e3744 103{
104 int ret;
105 struct sigaction sig;
106 struct sigaction osig;
107
108 sig.sa_handler = func;
109 sigemptyset (&sig.sa_mask);
110 sig.sa_flags = 0;
111#ifdef SA_RESTART
112 sig.sa_flags |= SA_RESTART;
113#endif /* SA_RESTART */
114
115 ret = sigaction (signo, &sig, &osig);
116
117 if (ret < 0)
118 return (SIG_ERR);
119 else
120 return (osig.sa_handler);
121}
122
123/* Initialization of signal handles. */
124void
e42f5a37 125vtysh_signal_init ()
718e3744 126{
e42f5a37 127 vtysh_signal_set (SIGINT, sigint);
128 vtysh_signal_set (SIGTSTP, sigtstp);
129 vtysh_signal_set (SIGPIPE, SIG_IGN);
718e3744 130}
b094d260 131
718e3744 132/* Help information display. */
133static void
134usage (int status)
135{
136 if (status != 0)
137 fprintf (stderr, "Try `%s --help' for more information.\n", progname);
138 else
95e735b5 139 printf ("Usage : %s [OPTION...]\n\n" \
f366ad31 140 "Integrated shell for Quagga routing software suite. \n\n" \
95e735b5 141 "-b, --boot Execute boot startup configuration\n" \
f366ad31
AS
142 "-c, --command Execute argument as command\n" \
143 "-d, --daemon Connect only to the specified daemon\n" \
144 "-E, --echo Echo prompt and command in -c mode\n" \
876b8be0 145 "-C, --dryrun Check configuration for validity and exit\n" \
95e735b5 146 "-h, --help Display this help and exit\n\n" \
f366ad31
AS
147 "Note that multiple commands may be executed from the command\n" \
148 "line by passing multiple -c args, or by embedding linefeed\n" \
149 "characters in one or more of the commands.\n\n" \
95e735b5 150 "Report bugs to %s\n", progname, ZEBRA_BUG_ADDRESS);
151
718e3744 152 exit (status);
153}
154
155/* VTY shell options, we use GNU getopt library. */
156struct option longopts[] =
157{
b094d260 158 { "boot", no_argument, NULL, 'b'},
4991f6ca 159 /* For compatibility with older zebra/quagga versions */
718e3744 160 { "eval", required_argument, NULL, 'e'},
4991f6ca 161 { "command", required_argument, NULL, 'c'},
f366ad31
AS
162 { "daemon", required_argument, NULL, 'd'},
163 { "echo", no_argument, NULL, 'E'},
876b8be0 164 { "dryrun", no_argument, NULL, 'C'},
718e3744 165 { "help", no_argument, NULL, 'h'},
914131f8 166 { "noerror", no_argument, NULL, 'n'},
718e3744 167 { 0 }
168};
b094d260 169
718e3744 170/* Read a string, and return a pointer to it. Returns NULL on EOF. */
171char *
172vtysh_rl_gets ()
173{
4991f6ca 174 HIST_ENTRY *last;
718e3744 175 /* If the buffer has already been allocated, return the memory
95e735b5 176 * to the free pool. */
718e3744 177 if (line_read)
178 {
179 free (line_read);
180 line_read = NULL;
181 }
182
183 /* Get a line from the user. Change prompt according to node. XXX. */
184 line_read = readline (vtysh_prompt ());
185
4991f6ca 186 /* If the line has any text in it, save it on the history. But only if
95e735b5 187 * last command in history isn't the same one. */
718e3744 188 if (line_read && *line_read)
4991f6ca 189 {
190 using_history();
191 last = previous_history();
3f4ab7f9 192 if (!last || strcmp (last->line, line_read) != 0) {
4991f6ca 193 add_history (line_read);
3f4ab7f9
TP
194 append_history(1,history_file);
195 }
4991f6ca 196 }
718e3744 197
198 return (line_read);
199}
b094d260 200
57fb9748
SH
201static void log_it(const char *line)
202{
203 time_t t = time(NULL);
204 struct tm *tmp = localtime(&t);
205 char *user = getenv("USER") ? : "boot";
206 char tod[64];
207
208 strftime(tod, sizeof tod, "%Y%m%d-%H:%M.%S", tmp);
209
210 fprintf(logfile, "%s:%s %s\n", tod, user, line);
211}
212
718e3744 213/* VTY shell main routine. */
214int
215main (int argc, char **argv, char **env)
216{
217 char *p;
218 int opt;
876b8be0 219 int dryrun = 0;
718e3744 220 int boot_flag = 0;
f366ad31
AS
221 const char *daemon_name = NULL;
222 struct cmd_rec {
223 const char *line;
224 struct cmd_rec *next;
225 } *cmd = NULL;
226 struct cmd_rec *tail = NULL;
227 int echo_command = 0;
914131f8 228 int no_error = 0;
718e3744 229
230 /* Preserve name of myself. */
231 progname = ((p = strrchr (argv[0], '/')) ? ++p : argv[0]);
232
57fb9748
SH
233 /* if logging open now */
234 if ((p = getenv("VTYSH_LOG")) != NULL)
235 logfile = fopen(p, "a");
236
718e3744 237 /* Option handling. */
238 while (1)
239 {
914131f8 240 opt = getopt_long (argc, argv, "be:c:d:nEhC", longopts, 0);
718e3744 241
242 if (opt == EOF)
243 break;
244
245 switch (opt)
246 {
247 case 0:
248 break;
249 case 'b':
250 boot_flag = 1;
251 break;
252 case 'e':
4991f6ca 253 case 'c':
f366ad31
AS
254 {
255 struct cmd_rec *cr;
256 cr = XMALLOC(0, sizeof(*cr));
257 cr->line = optarg;
258 cr->next = NULL;
259 if (tail)
260 tail->next = cr;
261 else
262 cmd = cr;
263 tail = cr;
264 }
265 break;
266 case 'd':
267 daemon_name = optarg;
268 break;
914131f8
SH
269 case 'n':
270 no_error = 1;
271 break;
f366ad31
AS
272 case 'E':
273 echo_command = 1;
718e3744 274 break;
876b8be0
PJ
275 case 'C':
276 dryrun = 1;
277 break;
718e3744 278 case 'h':
279 usage (0);
280 break;
718e3744 281 default:
282 usage (1);
283 break;
284 }
285 }
286
287 /* Initialize user input buffer. */
288 line_read = NULL;
0fbd62a1 289 setlinebuf(stdout);
718e3744 290
291 /* Signal and others. */
e42f5a37 292 vtysh_signal_init ();
718e3744 293
294 /* Make vty structure and register commands. */
295 vtysh_init_vty ();
296 vtysh_init_cmd ();
297 vtysh_user_init ();
298 vtysh_config_init ();
299
300 vty_init_vtysh ();
301
f366ad31 302 /* Read vtysh configuration file before connecting to daemons. */
e7168df4 303 vtysh_read_config (config_default);
718e3744 304
876b8be0
PJ
305 /* Start execution only if not in dry-run mode */
306 if(dryrun)
307 return(0);
308
914131f8
SH
309 /* Ignore error messages */
310 if (no_error)
311 freopen("/dev/null", "w", stdout);
312
f366ad31
AS
313 /* Make sure we pass authentication before proceeding. */
314 vtysh_auth ();
315
316 /* Do not connect until we have passed authentication. */
317 if (vtysh_connect_all (daemon_name) <= 0)
318 {
319 fprintf(stderr, "Exiting: failed to connect to any daemons.\n");
320 exit(1);
321 }
322
95e735b5 323 /* If eval mode. */
f366ad31 324 if (cmd)
718e3744 325 {
f366ad31
AS
326 /* Enter into enable node. */
327 vtysh_execute ("enable");
328
329 while (cmd != NULL)
330 {
57fb9748 331 int ret;
f366ad31
AS
332 char *eol;
333
334 while ((eol = strchr(cmd->line, '\n')) != NULL)
335 {
336 *eol = '\0';
57fb9748 337
f366ad31 338 if (echo_command)
57fb9748
SH
339 printf("%s%s\n", vtysh_prompt(), cmd->line);
340
341 if (logfile)
342 log_it(cmd->line);
343
344 ret = vtysh_execute_no_pager(cmd->line);
914131f8
SH
345 if (!no_error &&
346 ! (ret == CMD_SUCCESS ||
347 ret == CMD_SUCCESS_DAEMON ||
348 ret == CMD_WARNING))
57fb9748
SH
349 exit(1);
350
f366ad31
AS
351 cmd->line = eol+1;
352 }
57fb9748 353
f366ad31
AS
354 if (echo_command)
355 printf("%s%s\n", vtysh_prompt(), cmd->line);
57fb9748
SH
356
357 if (logfile)
358 log_it(cmd->line);
359
360 ret = vtysh_execute_no_pager(cmd->line);
914131f8
SH
361 if (!no_error &&
362 ! (ret == CMD_SUCCESS ||
363 ret == CMD_SUCCESS_DAEMON ||
364 ret == CMD_WARNING))
57fb9748 365 exit(1);
f366ad31
AS
366
367 {
368 struct cmd_rec *cr;
369 cr = cmd;
370 cmd = cmd->next;
371 XFREE(0, cr);
372 }
373 }
718e3744 374 exit (0);
375 }
376
377 /* Boot startup configuration file. */
378 if (boot_flag)
379 {
e7168df4 380 if (vtysh_read_config (integrate_default))
381 {
382 fprintf (stderr, "Can't open configuration file [%s]\n",
383 integrate_default);
384 exit (1);
385 }
386 else
387 exit (0);
718e3744 388 }
389
390 vtysh_pager_init ();
391
392 vtysh_readline_init ();
393
394 vty_hello (vty);
395
e7168df4 396 /* Enter into enable node. */
397 vtysh_execute ("enable");
398
718e3744 399 /* Preparation for longjmp() in sigtstp(). */
400 sigsetjmp (jmpbuf, 1);
401 jmpflag = 1;
402
3f4ab7f9
TP
403 snprintf(history_file, sizeof(history_file), "%s/.history_quagga", getenv("HOME"));
404 read_history(history_file);
718e3744 405 /* Main command loop. */
406 while (vtysh_rl_gets ())
407 vtysh_execute (line_read);
408
3f4ab7f9 409 history_truncate_file(history_file,1000);
718e3744 410 printf ("\n");
411
412 /* Rest in peace. */
413 exit (0);
414}