Pull out independent routines from client_full.c
Pull out routines for interacting with the server and interpreting internal commands which are not dependent on libreadline and ncurses libraries. This is a preparation step for a new lightweight birdc client.
This commit is contained in:
Tomas Hlavacek
parent
5c2c4ea8b1
commit
e454916149
4 changed files with 252 additions and 207 deletions
|
|
@ -1,4 +1,4 @@
|
||||||
source=client_full.c commands.c util.c
|
source=client_full.c commands.c util.c client_common.c
|
||||||
root-rel=../
|
root-rel=../
|
||||||
dir-name=client
|
dir-name=client
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -18,3 +18,17 @@ void cmd_build_tree(void);
|
||||||
void cmd_help(char *cmd, int len);
|
void cmd_help(char *cmd, int len);
|
||||||
int cmd_complete(char *cmd, int len, char *buf, int again);
|
int cmd_complete(char *cmd, int len, char *buf, int again);
|
||||||
char *cmd_expand(char *cmd);
|
char *cmd_expand(char *cmd);
|
||||||
|
|
||||||
|
/* client_common.c */
|
||||||
|
|
||||||
|
#define STATE_PROMPT 0
|
||||||
|
#define STATE_CMD_SERVER 1
|
||||||
|
#define STATE_CMD_USER 2
|
||||||
|
|
||||||
|
#define SERVER_READ_BUF_LEN 4096
|
||||||
|
|
||||||
|
int handle_internal_command(char *cmd);
|
||||||
|
void submit_server_command(char *cmd);
|
||||||
|
void server_connect(void);
|
||||||
|
void server_read(void);
|
||||||
|
void server_send(char *cmd);
|
||||||
|
|
|
||||||
179
client/client_common.c
Normal file
179
client/client_common.c
Normal file
|
|
@ -0,0 +1,179 @@
|
||||||
|
/*
|
||||||
|
* BIRD Client
|
||||||
|
*
|
||||||
|
* (c) 1999--2004 Martin Mares <mj@ucw.cz>
|
||||||
|
* (c) 2013 Tomas Hlavacek <tmshlvck@gmail.com>
|
||||||
|
*
|
||||||
|
* 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 <errno.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <sys/un.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
|
||||||
|
#include "nest/bird.h"
|
||||||
|
#include "lib/resource.h"
|
||||||
|
#include "lib/string.h"
|
||||||
|
#include "client/client.h"
|
||||||
|
#include "sysdep/unix/unix.h"
|
||||||
|
|
||||||
|
char *server_path = PATH_CONTROL_SOCKET;
|
||||||
|
int server_fd;
|
||||||
|
byte server_read_buf[SERVER_READ_BUF_LEN];
|
||||||
|
byte *server_read_pos = server_read_buf;
|
||||||
|
|
||||||
|
int input_initialized;
|
||||||
|
int input_hidden_end;
|
||||||
|
int cstate = STATE_CMD_SERVER;
|
||||||
|
int nstate = STATE_CMD_SERVER;
|
||||||
|
|
||||||
|
int num_lines, skip_input, interactive;
|
||||||
|
|
||||||
|
|
||||||
|
/*** Input ***/
|
||||||
|
|
||||||
|
int
|
||||||
|
handle_internal_command(char *cmd)
|
||||||
|
{
|
||||||
|
if (!strncmp(cmd, "exit", 4) || !strncmp(cmd, "quit", 4))
|
||||||
|
{
|
||||||
|
cleanup();
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
if (!strncmp(cmd, "help", 4))
|
||||||
|
{
|
||||||
|
puts("Press `?' for context sensitive help.");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
submit_server_command(char *cmd)
|
||||||
|
{
|
||||||
|
server_send(cmd);
|
||||||
|
nstate = STATE_CMD_SERVER;
|
||||||
|
num_lines = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*** Communication with server ***/
|
||||||
|
|
||||||
|
void
|
||||||
|
server_connect(void)
|
||||||
|
{
|
||||||
|
struct sockaddr_un sa;
|
||||||
|
|
||||||
|
server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||||
|
if (server_fd < 0)
|
||||||
|
die("Cannot create socket: %m");
|
||||||
|
|
||||||
|
if (strlen(server_path) >= sizeof(sa.sun_path))
|
||||||
|
die("server_connect: path too long");
|
||||||
|
|
||||||
|
bzero(&sa, sizeof(sa));
|
||||||
|
sa.sun_family = AF_UNIX;
|
||||||
|
strcpy(sa.sun_path, server_path);
|
||||||
|
if (connect(server_fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) < 0)
|
||||||
|
die("Unable to connect to server control socket (%s): %m", server_path);
|
||||||
|
if (fcntl(server_fd, F_SETFL, O_NONBLOCK) < 0)
|
||||||
|
die("fcntl: %m");
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
server_read(void)
|
||||||
|
{
|
||||||
|
int c;
|
||||||
|
byte *start, *p;
|
||||||
|
|
||||||
|
redo:
|
||||||
|
c = read(server_fd, server_read_pos, server_read_buf + sizeof(server_read_buf) - server_read_pos);
|
||||||
|
if (!c)
|
||||||
|
die("Connection closed by server.");
|
||||||
|
if (c < 0)
|
||||||
|
{
|
||||||
|
if (errno == EINTR)
|
||||||
|
goto redo;
|
||||||
|
else
|
||||||
|
die("Server read error: %m");
|
||||||
|
}
|
||||||
|
|
||||||
|
start = server_read_buf;
|
||||||
|
p = server_read_pos;
|
||||||
|
server_read_pos += c;
|
||||||
|
while (p < server_read_pos)
|
||||||
|
if (*p++ == '\n')
|
||||||
|
{
|
||||||
|
p[-1] = 0;
|
||||||
|
server_got_reply(start);
|
||||||
|
start = p;
|
||||||
|
}
|
||||||
|
if (start != server_read_buf)
|
||||||
|
{
|
||||||
|
int l = server_read_pos - start;
|
||||||
|
memmove(server_read_buf, start, l);
|
||||||
|
server_read_pos = server_read_buf + l;
|
||||||
|
}
|
||||||
|
else if (server_read_pos == server_read_buf + sizeof(server_read_buf))
|
||||||
|
{
|
||||||
|
strcpy(server_read_buf, "?<too-long>");
|
||||||
|
server_read_pos = server_read_buf + 11;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
wait_for_write(int fd)
|
||||||
|
{
|
||||||
|
while (1)
|
||||||
|
{
|
||||||
|
int rv;
|
||||||
|
fd_set set;
|
||||||
|
FD_ZERO(&set);
|
||||||
|
FD_SET(fd, &set);
|
||||||
|
|
||||||
|
rv = select(fd+1, NULL, &set, NULL, NULL);
|
||||||
|
if (rv < 0)
|
||||||
|
{
|
||||||
|
if (errno == EINTR)
|
||||||
|
continue;
|
||||||
|
else
|
||||||
|
die("select: %m");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (FD_ISSET(server_fd, &set))
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
server_send(char *cmd)
|
||||||
|
{
|
||||||
|
int l = strlen(cmd);
|
||||||
|
byte *z = alloca(l + 1);
|
||||||
|
|
||||||
|
memcpy(z, cmd, l);
|
||||||
|
z[l++] = '\n';
|
||||||
|
while (l)
|
||||||
|
{
|
||||||
|
int cnt = write(server_fd, z, l);
|
||||||
|
|
||||||
|
if (cnt < 0)
|
||||||
|
{
|
||||||
|
if (errno == EAGAIN)
|
||||||
|
wait_for_write(server_fd);
|
||||||
|
else if (errno == EINTR)
|
||||||
|
continue;
|
||||||
|
else
|
||||||
|
die("Server write error: %m");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
l -= cnt;
|
||||||
|
z += cnt;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -31,21 +31,17 @@ static char *init_cmd;
|
||||||
static int once;
|
static int once;
|
||||||
static int restricted;
|
static int restricted;
|
||||||
|
|
||||||
static char *server_path = PATH_CONTROL_SOCKET;
|
extern char *server_path;
|
||||||
static int server_fd;
|
extern int server_fd;
|
||||||
static byte server_read_buf[4096];
|
extern byte server_read_buf[SERVER_READ_BUF_LEN];
|
||||||
static byte *server_read_pos = server_read_buf;
|
extern byte *server_read_pos;
|
||||||
|
|
||||||
#define STATE_PROMPT 0
|
extern int input_initialized;
|
||||||
#define STATE_CMD_SERVER 1
|
extern int input_hidden_end;
|
||||||
#define STATE_CMD_USER 2
|
extern int cstate;
|
||||||
|
extern int nstate;
|
||||||
|
|
||||||
static int input_initialized;
|
extern int num_lines, skip_input, interactive;
|
||||||
static int input_hidden_end;
|
|
||||||
static int cstate = STATE_CMD_SERVER;
|
|
||||||
static int nstate = STATE_CMD_SERVER;
|
|
||||||
|
|
||||||
static int num_lines, skip_input, interactive;
|
|
||||||
|
|
||||||
/*** Parsing of arguments ***/
|
/*** Parsing of arguments ***/
|
||||||
|
|
||||||
|
|
@ -102,37 +98,11 @@ parse_args(int argc, char **argv)
|
||||||
|
|
||||||
/*** Input ***/
|
/*** Input ***/
|
||||||
|
|
||||||
static void server_send(char *);
|
|
||||||
|
|
||||||
/* HACK: libreadline internals we need to access */
|
/* HACK: libreadline internals we need to access */
|
||||||
extern int _rl_vis_botlin;
|
extern int _rl_vis_botlin;
|
||||||
extern void _rl_move_vert(int);
|
extern void _rl_move_vert(int);
|
||||||
extern Function *rl_last_func;
|
extern Function *rl_last_func;
|
||||||
|
|
||||||
static int
|
|
||||||
handle_internal_command(char *cmd)
|
|
||||||
{
|
|
||||||
if (!strncmp(cmd, "exit", 4) || !strncmp(cmd, "quit", 4))
|
|
||||||
{
|
|
||||||
cleanup();
|
|
||||||
exit(0);
|
|
||||||
}
|
|
||||||
if (!strncmp(cmd, "help", 4))
|
|
||||||
{
|
|
||||||
puts("Press `?' for context sensitive help.");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
|
||||||
submit_server_command(char *cmd)
|
|
||||||
{
|
|
||||||
server_send(cmd);
|
|
||||||
nstate = STATE_CMD_SERVER;
|
|
||||||
num_lines = 2;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
static void
|
||||||
add_history_dedup(char *cmd)
|
add_history_dedup(char *cmd)
|
||||||
{
|
{
|
||||||
|
|
@ -142,8 +112,7 @@ add_history_dedup(char *cmd)
|
||||||
add_history(cmd);
|
add_history(cmd);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
void got_line(char *cmd_buffer)
|
||||||
got_line(char *cmd_buffer)
|
|
||||||
{
|
{
|
||||||
char *cmd;
|
char *cmd;
|
||||||
|
|
||||||
|
|
@ -156,16 +125,16 @@ got_line(char *cmd_buffer)
|
||||||
{
|
{
|
||||||
cmd = cmd_expand(cmd_buffer);
|
cmd = cmd_expand(cmd_buffer);
|
||||||
if (cmd)
|
if (cmd)
|
||||||
{
|
{
|
||||||
add_history_dedup(cmd);
|
add_history_dedup(cmd);
|
||||||
|
|
||||||
if (!handle_internal_command(cmd))
|
if (!handle_internal_command(cmd))
|
||||||
submit_server_command(cmd);
|
submit_server_command(cmd);
|
||||||
|
|
||||||
free(cmd);
|
free(cmd);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
add_history_dedup(cmd_buffer);
|
add_history_dedup(cmd_buffer);
|
||||||
}
|
}
|
||||||
free(cmd_buffer);
|
free(cmd_buffer);
|
||||||
}
|
}
|
||||||
|
|
@ -284,17 +253,6 @@ input_reveal(void)
|
||||||
rl_forced_update_display();
|
rl_forced_update_display();
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
|
||||||
cleanup(void)
|
|
||||||
{
|
|
||||||
if (input_initialized)
|
|
||||||
{
|
|
||||||
input_initialized = 0;
|
|
||||||
input_hide();
|
|
||||||
rl_callback_handler_remove();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void
|
void
|
||||||
update_state(void)
|
update_state(void)
|
||||||
{
|
{
|
||||||
|
|
@ -364,112 +322,19 @@ more(void)
|
||||||
fflush(stdout);
|
fflush(stdout);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void cleanup(void)
|
||||||
|
{
|
||||||
|
if (input_initialized)
|
||||||
|
{
|
||||||
|
input_initialized = 0;
|
||||||
|
input_hide();
|
||||||
|
rl_callback_handler_remove();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/*** Communication with server ***/
|
/*** Communication with server ***/
|
||||||
|
|
||||||
static void
|
|
||||||
server_connect(void)
|
|
||||||
{
|
|
||||||
struct sockaddr_un sa;
|
|
||||||
|
|
||||||
server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
||||||
if (server_fd < 0)
|
|
||||||
die("Cannot create socket: %m");
|
|
||||||
|
|
||||||
if (strlen(server_path) >= sizeof(sa.sun_path))
|
|
||||||
die("server_connect: path too long");
|
|
||||||
|
|
||||||
bzero(&sa, sizeof(sa));
|
|
||||||
sa.sun_family = AF_UNIX;
|
|
||||||
strcpy(sa.sun_path, server_path);
|
|
||||||
if (connect(server_fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) < 0)
|
|
||||||
die("Unable to connect to server control socket (%s): %m", server_path);
|
|
||||||
if (fcntl(server_fd, F_SETFL, O_NONBLOCK) < 0)
|
|
||||||
die("fcntl: %m");
|
|
||||||
}
|
|
||||||
|
|
||||||
#define PRINTF(LEN, PARGS...) do { if (!skip_input) len = printf(PARGS); } while(0)
|
|
||||||
|
|
||||||
static 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] == ' ')
|
|
||||||
{
|
|
||||||
nstate = STATE_PROMPT;
|
|
||||||
skip_input = 0;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
PRINTF(len, "??? <%s>\n", x);
|
|
||||||
|
|
||||||
if (skip_input)
|
|
||||||
return;
|
|
||||||
|
|
||||||
if (interactive && input_initialized && (len > 0))
|
|
||||||
{
|
|
||||||
int lns = LINES ? LINES : 25;
|
|
||||||
int cls = COLS ? COLS : 80;
|
|
||||||
num_lines += (len + cls - 1) / cls; /* Divide and round up */
|
|
||||||
if ((num_lines >= lns) && (cstate == STATE_CMD_SERVER))
|
|
||||||
more();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
|
||||||
server_read(void)
|
|
||||||
{
|
|
||||||
int c;
|
|
||||||
byte *start, *p;
|
|
||||||
|
|
||||||
redo:
|
|
||||||
c = read(server_fd, server_read_pos, server_read_buf + sizeof(server_read_buf) - server_read_pos);
|
|
||||||
if (!c)
|
|
||||||
die("Connection closed by server.");
|
|
||||||
if (c < 0)
|
|
||||||
{
|
|
||||||
if (errno == EINTR)
|
|
||||||
goto redo;
|
|
||||||
else
|
|
||||||
die("Server read error: %m");
|
|
||||||
}
|
|
||||||
|
|
||||||
start = server_read_buf;
|
|
||||||
p = server_read_pos;
|
|
||||||
server_read_pos += c;
|
|
||||||
while (p < server_read_pos)
|
|
||||||
if (*p++ == '\n')
|
|
||||||
{
|
|
||||||
p[-1] = 0;
|
|
||||||
server_got_reply(start);
|
|
||||||
start = p;
|
|
||||||
}
|
|
||||||
if (start != server_read_buf)
|
|
||||||
{
|
|
||||||
int l = server_read_pos - start;
|
|
||||||
memmove(server_read_buf, start, l);
|
|
||||||
server_read_pos = server_read_buf + l;
|
|
||||||
}
|
|
||||||
else if (server_read_pos == server_read_buf + sizeof(server_read_buf))
|
|
||||||
{
|
|
||||||
strcpy(server_read_buf, "?<too-long>");
|
|
||||||
server_read_pos = server_read_buf + 11;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static fd_set select_fds;
|
static fd_set select_fds;
|
||||||
|
|
||||||
static void
|
static void
|
||||||
|
|
@ -508,56 +373,43 @@ select_loop(void)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
#define PRINTF(LEN, PARGS...) do { if (!skip_input) len = printf(PARGS); } while(0)
|
||||||
wait_for_write(int fd)
|
|
||||||
|
void server_got_reply(char *x)
|
||||||
{
|
{
|
||||||
while (1)
|
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] == '-'))
|
||||||
{
|
{
|
||||||
int rv;
|
if (code)
|
||||||
fd_set set;
|
PRINTF(len, "%s\n", verbose ? x : x+5);
|
||||||
FD_ZERO(&set);
|
if (x[4] == ' ')
|
||||||
FD_SET(fd, &set);
|
{
|
||||||
|
nstate = STATE_PROMPT;
|
||||||
rv = select(fd+1, NULL, &set, NULL, NULL);
|
skip_input = 0;
|
||||||
if (rv < 0)
|
return;
|
||||||
{
|
}
|
||||||
if (errno == EINTR)
|
|
||||||
continue;
|
|
||||||
else
|
|
||||||
die("select: %m");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (FD_ISSET(server_fd, &set))
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
}
|
else
|
||||||
|
PRINTF(len, "??? <%s>\n", x);
|
||||||
|
|
||||||
static void
|
if (skip_input)
|
||||||
server_send(char *cmd)
|
return;
|
||||||
{
|
|
||||||
int l = strlen(cmd);
|
|
||||||
byte *z = alloca(l + 1);
|
|
||||||
|
|
||||||
memcpy(z, cmd, l);
|
if (interactive && input_initialized && (len > 0))
|
||||||
z[l++] = '\n';
|
|
||||||
while (l)
|
|
||||||
{
|
{
|
||||||
int cnt = write(server_fd, z, l);
|
int lns = LINES ? LINES : 25;
|
||||||
|
int cls = COLS ? COLS : 80;
|
||||||
if (cnt < 0)
|
num_lines += (len + cls - 1) / cls; /* Divide and round up */
|
||||||
{
|
if ((num_lines >= lns) && (cstate == STATE_CMD_SERVER))
|
||||||
if (errno == EAGAIN)
|
more();
|
||||||
wait_for_write(server_fd);
|
|
||||||
else if (errno == EINTR)
|
|
||||||
continue;
|
|
||||||
else
|
|
||||||
die("Server write error: %m");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
l -= cnt;
|
|
||||||
z += cnt;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue