bird/nest/cli.c

144 lines
2.3 KiB
C

/*
* BIRD Internet Routing Daemon -- Command-Line Interface
*
* (c) 1999 Martin Mares <mj@ucw.cz>
*
* Can be freely distributed and used under the terms of the GNU GPL.
*/
#include "nest/bird.h"
#include "lib/string.h"
#include "nest/cli.h"
pool *cli_pool;
void
cli_printf(cli *c, int code, char *msg, ...)
{
va_list args;
byte buf[1024];
int flag = (code < 0) ? '-' : ' ';
int size;
struct cli_out *o;
va_start(args, msg);
if (code < 0)
code = -code;
bsprintf(buf, "%04d%c", code, flag);
size = bvsnprintf(buf+5, sizeof(buf)-6, msg, args);
if (size < 0)
size = bsprintf(buf, "9999%c<line overflow>", flag);
else
size += 5;
buf[size++] = '\n';
if (!(o = c->tx_write) || o->wpos + size > o->end)
{
o = mb_alloc(c->pool, sizeof(struct cli_out) + CLI_TX_BUF_SIZE);
if (c->tx_write)
c->tx_write->next = o;
else
c->tx_buf = o;
o->next = NULL;
o->wpos = o->outpos = o->buf;
o->end = o->buf + CLI_TX_BUF_SIZE;
c->tx_write = o;
}
memcpy(o->wpos, buf, size);
o->wpos += size;
}
static void
cli_free_out(cli *c)
{
struct cli_out *o, *p;
if (o = c->tx_buf)
{
c->tx_write = o;
o->wpos = o->outpos = o->buf;
while (p = o->next)
{
o->next = p->next;
mb_free(p);
}
}
}
static int
cli_flush(cli *c)
{
if (cli_write(c))
{
cli_free_out(c);
return 1;
}
return 0;
}
static int
cli_event(void *data)
{
cli *c = data;
int err;
debug("CLI EVENT\n");
if (!c->inited)
{
c->inited = 1;
cli_printf(c, 0, "Welcome!");
cli_printf(c, 0, "Here");
return cli_flush(c);
}
err = cli_get_command(c);
if (!err)
return 0;
if (err < 0)
debug("CLI CMD ERR\n");
else
debug("CLI CMD %s\n", c->rx_buf);
return 1;
}
cli *
cli_new(void *priv)
{
pool *p = rp_new(cli_pool, "CLI");
cli *c = mb_alloc(p, sizeof(cli));
c->pool = p;
c->priv = priv;
c->event = ev_new(p);
c->event->hook = cli_event;
c->event->data = c;
c->tx_buf = c->tx_pos = c->tx_write = NULL;
c->inited = 0;
cli_kick(c);
return c;
}
void
cli_kick(cli *c)
{
debug("CLI KICK\n");
ev_schedule(c->event);
}
void
cli_written(cli *c)
{
debug("CLI WRITTEN\n");
cli_free_out(c);
cli_kick(c);
}
void
cli_free(cli *c)
{
rfree(c->pool);
}
void
cli_init(void)
{
cli_pool = rp_new(&root_pool, "CLI");
}