mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-23 18:21:54 +00:00
568d9c9fae
Fix handling of async messafe in the bird light client. The async message may occure at the any moment so we need the client to liste for the message from server when it waits for user input.
430 lines
7.1 KiB
C
430 lines
7.1 KiB
C
/*
|
|
* BIRD Client
|
|
*
|
|
* (c) 1999--2004 Martin Mares <mj@ucw.cz>
|
|
* (c) 2013 Tomas Hlavacek <tomas.hlavacek@nic.cz>
|
|
*
|
|
* Can be freely distributed and used under the terms of the GNU GPL.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <termios.h>
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <sys/types.h>
|
|
#include <sys/ioctl.h>
|
|
#include <signal.h>
|
|
#include <ctype.h>
|
|
|
|
#include "nest/bird.h"
|
|
#include "lib/resource.h"
|
|
#include "lib/string.h"
|
|
#include "client/client.h"
|
|
#include "sysdep/unix/unix.h"
|
|
|
|
#define INPUT_BUF_LEN 2048
|
|
|
|
static char *opt_list = "s:vr";
|
|
static int verbose;
|
|
static char *init_cmd;
|
|
static int once;
|
|
|
|
extern char *server_path;
|
|
extern int server_fd;
|
|
|
|
extern int cstate;
|
|
extern int num_lines, skip_input, interactive;
|
|
|
|
static int term_lns=25;
|
|
static int term_cls=80;
|
|
struct termios tty_save;
|
|
|
|
void
|
|
input_start_list(void)
|
|
{
|
|
/* Empty in non-ncurses version. */
|
|
}
|
|
|
|
void
|
|
input_stop_list(void)
|
|
{
|
|
/* Empty in non-ncurses version. */
|
|
}
|
|
|
|
/*** Parsing of arguments ***/
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
fprintf(stderr, "Usage: birdc [-s <control-socket>] [-v] [-r]\n");
|
|
exit(1);
|
|
}
|
|
|
|
static void
|
|
parse_args(int argc, char **argv)
|
|
{
|
|
int c;
|
|
|
|
while ((c = getopt(argc, argv, opt_list)) >= 0)
|
|
switch (c)
|
|
{
|
|
case 's':
|
|
server_path = optarg;
|
|
break;
|
|
case 'v':
|
|
verbose++;
|
|
break;
|
|
case 'r':
|
|
init_cmd = "restrict";
|
|
break;
|
|
default:
|
|
usage();
|
|
}
|
|
|
|
/* If some arguments are not options, we take it as commands */
|
|
if (optind < argc)
|
|
{
|
|
char *tmp;
|
|
int i;
|
|
int len = 0;
|
|
|
|
if (init_cmd)
|
|
usage();
|
|
|
|
for (i = optind; i < argc; i++)
|
|
len += strlen(argv[i]) + 1;
|
|
|
|
tmp = init_cmd = malloc(len);
|
|
for (i = optind; i < argc; i++)
|
|
{
|
|
strcpy(tmp, argv[i]);
|
|
tmp += strlen(tmp);
|
|
*tmp++ = ' ';
|
|
}
|
|
tmp[-1] = 0;
|
|
|
|
once = 1;
|
|
}
|
|
}
|
|
|
|
static void
|
|
run_init_cmd(void)
|
|
{
|
|
if (init_cmd)
|
|
{
|
|
/* First transition - client received hello from BIRD
|
|
and there is waiting initial command */
|
|
submit_server_command(init_cmd);
|
|
init_cmd = NULL;
|
|
return;
|
|
}
|
|
|
|
if (!init_cmd && once && (cstate == STATE_PROMPT))
|
|
{
|
|
/* Initial command is finished and we want to exit */
|
|
cleanup();
|
|
exit(0);
|
|
}
|
|
}
|
|
|
|
/*** Input ***/
|
|
|
|
static void
|
|
got_line(char *cmd_buffer)
|
|
{
|
|
char *cmd;
|
|
|
|
if (!cmd_buffer)
|
|
{
|
|
cleanup();
|
|
exit(0);
|
|
}
|
|
if (cmd_buffer[0])
|
|
{
|
|
cmd = cmd_expand(cmd_buffer);
|
|
if (cmd)
|
|
{
|
|
if (!handle_internal_command(cmd))
|
|
submit_server_command(cmd);
|
|
|
|
free(cmd);
|
|
}
|
|
}
|
|
free(cmd_buffer);
|
|
}
|
|
|
|
void
|
|
cleanup(void)
|
|
{
|
|
/* No ncurses -> restore terminal state. */
|
|
if (interactive)
|
|
if (tcsetattr (0, TCSANOW, &tty_save) != 0)
|
|
{
|
|
perror("tcsetattr error");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
static void
|
|
print_prompt(void)
|
|
{
|
|
/* No ncurses -> no status to reveal/hide, print prompt manually. */
|
|
printf("bird> ");
|
|
fflush(stdout);
|
|
}
|
|
|
|
|
|
static int lastnb(char *str)
|
|
{
|
|
int i;
|
|
for (i=strlen(str)-1; i>0; i--)
|
|
{
|
|
if(!isblank(str[i]))
|
|
return i;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
term_read(void)
|
|
{
|
|
char *buf = malloc(INPUT_BUF_LEN);
|
|
|
|
if (fgets(buf, INPUT_BUF_LEN, stdin) == NULL) {
|
|
free(buf);
|
|
exit(0);
|
|
}
|
|
|
|
if (buf[strlen(buf)-1] != '\n')
|
|
{
|
|
printf("Input too long.\n");
|
|
free(buf);
|
|
return;
|
|
}
|
|
|
|
if (strlen(buf) <= 0)
|
|
{
|
|
free(buf);
|
|
return;
|
|
}
|
|
|
|
buf[strlen(buf)-1] = '\0';
|
|
|
|
if (!interactive)
|
|
{
|
|
print_prompt();
|
|
printf("%s\n",buf);
|
|
}
|
|
|
|
if (buf[lastnb(buf)] == '?')
|
|
{
|
|
printf("\n");
|
|
cmd_help(buf, strlen(buf));
|
|
free(buf);
|
|
return;
|
|
}
|
|
|
|
if (strlen(buf) > 0)
|
|
{
|
|
got_line(buf); /* buf is free()-ed inside */
|
|
}
|
|
else
|
|
{
|
|
free(buf); /* no command, only newline -> no-op */
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
/*** Communication with server ***/
|
|
|
|
void
|
|
more(void)
|
|
{
|
|
struct termios tty;
|
|
|
|
printf("--More--\015");
|
|
fflush(stdout);
|
|
|
|
if (tcgetattr(0, &tty) != 0)
|
|
{
|
|
perror("tcgetattr error");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
tty.c_lflag &= (~ECHO);
|
|
tty.c_lflag &= (~ICANON);
|
|
if (tcsetattr (0, TCSANOW, &tty) != 0)
|
|
{
|
|
perror("tcsetattr error");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
redo:
|
|
switch (getchar())
|
|
{
|
|
case 32:
|
|
num_lines = 2;
|
|
break;
|
|
case 13:
|
|
num_lines--;
|
|
break;
|
|
case '\n':
|
|
num_lines--;
|
|
break;
|
|
case 'q':
|
|
skip_input = 1;
|
|
break;
|
|
default:
|
|
goto redo;
|
|
}
|
|
|
|
tty.c_lflag |= ECHO;
|
|
tty.c_lflag |= ICANON;
|
|
if (tcsetattr (0, TCSANOW, &tty) != 0)
|
|
{
|
|
perror("tcsetattr error");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
printf(" \015");
|
|
fflush(stdout);
|
|
}
|
|
|
|
static void
|
|
get_term_size(void)
|
|
{
|
|
struct winsize tws;
|
|
if (ioctl(0, TIOCGWINSZ, &tws) == 0)
|
|
{
|
|
term_lns = tws.ws_row;
|
|
term_cls = tws.ws_col;
|
|
}
|
|
else
|
|
{
|
|
term_lns = 25;
|
|
term_cls = 80;
|
|
}
|
|
}
|
|
|
|
#define PRINTF(LEN, PARGS...) do { if (!skip_input) len = printf(PARGS); } while(0)
|
|
|
|
void
|
|
server_got_reply(char *x)
|
|
{
|
|
int code;
|
|
int len = 0;
|
|
|
|
if (*x == '+') /* Async reply */
|
|
PRINTF(len, ">>> %s\n", x+1);
|
|
else if (x[0] == ' ') /* Continuation */
|
|
PRINTF(len, "%s%s\n", verbose ? " " : "", x+1);
|
|
else if (strlen(x) > 4 &&
|
|
sscanf(x, "%d", &code) == 1 && code >= 0 && code < 10000 &&
|
|
(x[4] == ' ' || x[4] == '-'))
|
|
{
|
|
if (code)
|
|
PRINTF(len, "%s\n", verbose ? x : x+5);
|
|
|
|
if (x[4] == ' ')
|
|
{
|
|
cstate = STATE_PROMPT;
|
|
skip_input = 0;
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
PRINTF(len, "??? <%s>\n", x);
|
|
|
|
if (skip_input)
|
|
return;
|
|
|
|
if (interactive && (len > 0))
|
|
{
|
|
num_lines += (len + term_cls - 1) / term_cls; /* Divide and round up */
|
|
if (num_lines >= term_lns)
|
|
more();
|
|
}
|
|
}
|
|
|
|
static fd_set select_fds;
|
|
|
|
static void
|
|
select_loop(void)
|
|
{
|
|
int rv;
|
|
|
|
while (1)
|
|
{
|
|
FD_ZERO(&select_fds);
|
|
|
|
if (cstate != STATE_CMD_USER)
|
|
FD_SET(server_fd, &select_fds);
|
|
|
|
if (cstate != STATE_CMD_SERVER)
|
|
{
|
|
FD_SET(0, &select_fds);
|
|
if (interactive)
|
|
print_prompt();
|
|
}
|
|
|
|
rv = select(server_fd+1, &select_fds, NULL, NULL, NULL);
|
|
if (rv < 0)
|
|
{
|
|
if (errno == EINTR)
|
|
continue;
|
|
else
|
|
die("select: %m");
|
|
}
|
|
|
|
if (FD_ISSET(server_fd, &select_fds))
|
|
{
|
|
server_read();
|
|
run_init_cmd();
|
|
}
|
|
|
|
if (FD_ISSET(0, &select_fds))
|
|
term_read();
|
|
}
|
|
}
|
|
|
|
static void
|
|
sig_handler(int signal)
|
|
{
|
|
cleanup();
|
|
exit(0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
interactive = isatty(fileno(stdin));
|
|
if (interactive)
|
|
{
|
|
if (signal(SIGINT, sig_handler) == SIG_IGN)
|
|
signal(SIGINT, SIG_IGN);
|
|
if (signal(SIGHUP, sig_handler) == SIG_IGN)
|
|
signal(SIGHUP, SIG_IGN);
|
|
if (signal(SIGTERM, sig_handler) == SIG_IGN)
|
|
signal(SIGTERM, SIG_IGN);
|
|
|
|
get_term_size();
|
|
|
|
if (tcgetattr(0, &tty_save) != 0)
|
|
{
|
|
perror("tcgetattr error");
|
|
return(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
parse_args(argc, argv);
|
|
cmd_build_tree();
|
|
server_connect();
|
|
select_loop();
|
|
return 0;
|
|
}
|