mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-23 02:01:55 +00:00
A simple experiment with coroutine-based CLI
This commit is contained in:
parent
a1f5e514ef
commit
fd3d15b6b5
@ -36,6 +36,8 @@ struct ssh_sock {
|
|||||||
};
|
};
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
struct coroutine;
|
||||||
|
|
||||||
typedef struct birdsock {
|
typedef struct birdsock {
|
||||||
resource r;
|
resource r;
|
||||||
pool *pool; /* Pool where incoming connections should be allocated (for SK_xxx_PASSIVE) */
|
pool *pool; /* Pool where incoming connections should be allocated (for SK_xxx_PASSIVE) */
|
||||||
@ -78,6 +80,8 @@ typedef struct birdsock {
|
|||||||
char *password; /* Password for MD5 authentication */
|
char *password; /* Password for MD5 authentication */
|
||||||
const char *err; /* Error message */
|
const char *err; /* Error message */
|
||||||
struct ssh_sock *ssh; /* Used in SK_SSH */
|
struct ssh_sock *ssh; /* Used in SK_SSH */
|
||||||
|
|
||||||
|
struct coroutine *rx_coroutine;
|
||||||
} sock;
|
} sock;
|
||||||
|
|
||||||
sock *sock_new(pool *); /* Allocate new socket */
|
sock *sock_new(pool *); /* Allocate new socket */
|
||||||
|
31
nest/cli.c
31
nest/cli.c
@ -67,6 +67,7 @@
|
|||||||
#include "nest/cli.h"
|
#include "nest/cli.h"
|
||||||
#include "conf/conf.h"
|
#include "conf/conf.h"
|
||||||
#include "lib/string.h"
|
#include "lib/string.h"
|
||||||
|
#include "sysdep/unix/unix.h" // FIXME
|
||||||
|
|
||||||
pool *cli_pool;
|
pool *cli_pool;
|
||||||
|
|
||||||
@ -251,8 +252,8 @@ cli_cmd_read_hook(byte *buf, uint max, UNUSED int fd)
|
|||||||
return max;
|
return max;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
void
|
||||||
cli_command(struct cli *c)
|
cli_command(cli *c)
|
||||||
{
|
{
|
||||||
struct config f;
|
struct config f;
|
||||||
int res;
|
int res;
|
||||||
@ -279,28 +280,15 @@ static void
|
|||||||
cli_event(void *data)
|
cli_event(void *data)
|
||||||
{
|
{
|
||||||
cli *c = data;
|
cli *c = data;
|
||||||
int err;
|
|
||||||
|
|
||||||
while (c->ring_read != c->ring_write &&
|
while (c->ring_read != c->ring_write &&
|
||||||
c->async_msg_size < CLI_MAX_ASYNC_QUEUE)
|
c->async_msg_size < CLI_MAX_ASYNC_QUEUE)
|
||||||
cli_copy_message(c);
|
cli_copy_message(c);
|
||||||
|
|
||||||
if (c->tx_pos)
|
|
||||||
;
|
|
||||||
else if (c->cont)
|
|
||||||
c->cont(c);
|
|
||||||
else
|
|
||||||
{
|
|
||||||
err = cli_get_command(c);
|
|
||||||
if (!err)
|
|
||||||
return;
|
|
||||||
if (err < 0)
|
|
||||||
cli_printf(c, 9000, "Command too long");
|
|
||||||
else
|
|
||||||
cli_command(c);
|
|
||||||
}
|
|
||||||
|
|
||||||
cli_write_trigger(c);
|
cli_write_trigger(c);
|
||||||
|
|
||||||
|
if (c->sleeping_on_yield)
|
||||||
|
coro_resume(c->coro);
|
||||||
}
|
}
|
||||||
|
|
||||||
cli *
|
cli *
|
||||||
@ -323,13 +311,6 @@ cli_new(void *priv)
|
|||||||
return c;
|
return c;
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
|
||||||
cli_kick(cli *c)
|
|
||||||
{
|
|
||||||
if (!c->cont && !c->tx_pos)
|
|
||||||
ev_schedule(c->event);
|
|
||||||
}
|
|
||||||
|
|
||||||
static list cli_log_hooks;
|
static list cli_log_hooks;
|
||||||
static int cli_log_inited;
|
static int cli_log_inited;
|
||||||
|
|
||||||
|
@ -25,6 +25,8 @@ struct cli_out {
|
|||||||
byte buf[0];
|
byte buf[0];
|
||||||
};
|
};
|
||||||
|
|
||||||
|
struct coroutine;
|
||||||
|
|
||||||
typedef struct cli {
|
typedef struct cli {
|
||||||
node n; /* Node in list of all log hooks */
|
node n; /* Node in list of all log hooks */
|
||||||
pool *pool;
|
pool *pool;
|
||||||
@ -45,6 +47,9 @@ typedef struct cli {
|
|||||||
uint log_mask; /* Mask of allowed message levels */
|
uint log_mask; /* Mask of allowed message levels */
|
||||||
uint log_threshold; /* When free < log_threshold, store only important messages */
|
uint log_threshold; /* When free < log_threshold, store only important messages */
|
||||||
uint async_msg_size; /* Total size of async messages queued in tx_buf */
|
uint async_msg_size; /* Total size of async messages queued in tx_buf */
|
||||||
|
struct coroutine *coro;
|
||||||
|
int sleeping_on_tx;
|
||||||
|
int sleeping_on_yield;
|
||||||
} cli;
|
} cli;
|
||||||
|
|
||||||
extern pool *cli_pool;
|
extern pool *cli_pool;
|
||||||
@ -66,6 +71,7 @@ void cli_free(cli *);
|
|||||||
void cli_kick(cli *);
|
void cli_kick(cli *);
|
||||||
void cli_written(cli *);
|
void cli_written(cli *);
|
||||||
void cli_echo(uint class, byte *msg);
|
void cli_echo(uint class, byte *msg);
|
||||||
|
void cli_command(cli *c);
|
||||||
|
|
||||||
static inline int cli_access_restricted(void)
|
static inline int cli_access_restricted(void)
|
||||||
{
|
{
|
||||||
|
110
sysdep/unix/io.c
110
sysdep/unix/io.c
@ -2665,3 +2665,113 @@ test_old_bird(char *path)
|
|||||||
die("I found another BIRD running.");
|
die("I found another BIRD running.");
|
||||||
close(fd);
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* EXPERIMENTAL: Support for coroutines */
|
||||||
|
|
||||||
|
#include <ucontext.h>
|
||||||
|
|
||||||
|
struct coroutine {
|
||||||
|
resource r;
|
||||||
|
ucontext_t ctx;
|
||||||
|
void *stack;
|
||||||
|
void (*entry_point)(void *arg);
|
||||||
|
void *arg;
|
||||||
|
};
|
||||||
|
|
||||||
|
static ucontext_t *main_context;
|
||||||
|
static coroutine *coro_current; // NULL for main context
|
||||||
|
|
||||||
|
static void
|
||||||
|
coro_free(resource *r)
|
||||||
|
{
|
||||||
|
coroutine *c = (coroutine *) r;
|
||||||
|
xfree(c->stack);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
coro_dump(resource *r UNUSED)
|
||||||
|
{
|
||||||
|
debug("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct resclass coro_class = {
|
||||||
|
.name = "Coroutine",
|
||||||
|
.size = sizeof(struct coroutine),
|
||||||
|
.free = coro_free,
|
||||||
|
.dump = coro_dump,
|
||||||
|
// FIXME: Implement memsize
|
||||||
|
};
|
||||||
|
|
||||||
|
static void
|
||||||
|
coro_do_start(void)
|
||||||
|
{
|
||||||
|
ASSERT(coro_current);
|
||||||
|
coro_current->entry_point(coro_current->arg);
|
||||||
|
bug("Coroutine returned unexpectedly");
|
||||||
|
}
|
||||||
|
|
||||||
|
struct coroutine *
|
||||||
|
coro_new(pool *p, void (*entry_point)(void *), void *arg)
|
||||||
|
{
|
||||||
|
if (!main_context)
|
||||||
|
{
|
||||||
|
main_context = xmalloc(sizeof(*main_context));
|
||||||
|
if (getcontext(main_context) < 0)
|
||||||
|
bug("getcontext() failed");
|
||||||
|
}
|
||||||
|
|
||||||
|
coroutine *c = ralloc(p, &coro_class);
|
||||||
|
c->entry_point = entry_point;
|
||||||
|
c->arg = arg;
|
||||||
|
if (getcontext(&c->ctx) < 0)
|
||||||
|
bug("getcontext() failed");
|
||||||
|
c->stack = xmalloc(65536);
|
||||||
|
c->ctx.uc_stack.ss_sp = c->stack;
|
||||||
|
c->ctx.uc_stack.ss_size = 65536;
|
||||||
|
|
||||||
|
makecontext(&c->ctx, coro_do_start, 0);
|
||||||
|
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Return to main context
|
||||||
|
void
|
||||||
|
coro_suspend(void)
|
||||||
|
{
|
||||||
|
ASSERT(coro_current);
|
||||||
|
ASSERT(main_context);
|
||||||
|
coroutine *c = coro_current;
|
||||||
|
coro_current = NULL;
|
||||||
|
swapcontext(&c->ctx, main_context);
|
||||||
|
ASSERT(coro_current == c);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Resume context
|
||||||
|
void
|
||||||
|
coro_resume(coroutine *c)
|
||||||
|
{
|
||||||
|
ASSERT(!coro_current);
|
||||||
|
coro_current = c;
|
||||||
|
swapcontext(main_context, &c->ctx);
|
||||||
|
ASSERT(!coro_current);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
coro_sk_rx_hook(sock *sk, uint size UNUSED)
|
||||||
|
{
|
||||||
|
ASSERT(sk->rx_coroutine);
|
||||||
|
ASSERT(!coro_current);
|
||||||
|
coro_resume(sk->rx_coroutine);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
coro_sk_read(sock *s)
|
||||||
|
{
|
||||||
|
ASSERT(coro_current);
|
||||||
|
s->rx_coroutine = coro_current;
|
||||||
|
s->rx_hook = coro_sk_rx_hook;
|
||||||
|
coro_suspend();
|
||||||
|
s->rx_hook = NULL;
|
||||||
|
return s->rpos - s->rbuf;
|
||||||
|
}
|
||||||
|
@ -374,6 +374,9 @@ cli_write(cli *c)
|
|||||||
/* Everything is written */
|
/* Everything is written */
|
||||||
s->tbuf = NULL;
|
s->tbuf = NULL;
|
||||||
cli_written(c);
|
cli_written(c);
|
||||||
|
|
||||||
|
if (c->sleeping_on_tx)
|
||||||
|
coro_resume(c->coro);
|
||||||
}
|
}
|
||||||
|
|
||||||
void
|
void
|
||||||
@ -391,42 +394,6 @@ cli_tx(sock *s)
|
|||||||
cli_write(s->data);
|
cli_write(s->data);
|
||||||
}
|
}
|
||||||
|
|
||||||
int
|
|
||||||
cli_get_command(cli *c)
|
|
||||||
{
|
|
||||||
sock *s = c->priv;
|
|
||||||
byte *t = c->rx_aux ? : s->rbuf;
|
|
||||||
byte *tend = s->rpos;
|
|
||||||
byte *d = c->rx_pos;
|
|
||||||
byte *dend = c->rx_buf + CLI_RX_BUF_SIZE - 2;
|
|
||||||
|
|
||||||
while (t < tend)
|
|
||||||
{
|
|
||||||
if (*t == '\r')
|
|
||||||
t++;
|
|
||||||
else if (*t == '\n')
|
|
||||||
{
|
|
||||||
t++;
|
|
||||||
c->rx_pos = c->rx_buf;
|
|
||||||
c->rx_aux = t;
|
|
||||||
*d = 0;
|
|
||||||
return (d < dend) ? 1 : -1;
|
|
||||||
}
|
|
||||||
else if (d < dend)
|
|
||||||
*d++ = *t++;
|
|
||||||
}
|
|
||||||
c->rx_aux = s->rpos = s->rbuf;
|
|
||||||
c->rx_pos = d;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int
|
|
||||||
cli_rx(sock *s, uint size UNUSED)
|
|
||||||
{
|
|
||||||
cli_kick(s->data);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void
|
static void
|
||||||
cli_err(sock *s, int err)
|
cli_err(sock *s, int err)
|
||||||
{
|
{
|
||||||
@ -440,6 +407,86 @@ cli_err(sock *s, int err)
|
|||||||
cli_free(s->data);
|
cli_free(s->data);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
cli_getchar(cli *c)
|
||||||
|
{
|
||||||
|
sock *s = c->priv;
|
||||||
|
|
||||||
|
if (c->rx_aux == s->rpos)
|
||||||
|
{
|
||||||
|
log(L_INFO "CLI wants to read");
|
||||||
|
c->rx_aux = s->rpos = s->rbuf;
|
||||||
|
int n = coro_sk_read(s);
|
||||||
|
log(L_INFO "CLI read %d bytes", n);
|
||||||
|
ASSERT(n);
|
||||||
|
}
|
||||||
|
return *c->rx_aux++;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
cli_yield(cli *c)
|
||||||
|
{
|
||||||
|
log(L_INFO "CLI: Yield");
|
||||||
|
c->sleeping_on_yield = 1;
|
||||||
|
ev_schedule(c->event);
|
||||||
|
coro_suspend();
|
||||||
|
c->sleeping_on_yield = 0;
|
||||||
|
log(L_INFO "CLI: Resumed after yield");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
cli_coroutine(void *_c)
|
||||||
|
{
|
||||||
|
cli *c = _c;
|
||||||
|
sock *s = c->priv;
|
||||||
|
|
||||||
|
log(L_INFO "CLI coroutine reached");
|
||||||
|
c->rx_aux = s->rbuf;
|
||||||
|
for (;;)
|
||||||
|
{
|
||||||
|
while (c->tx_pos)
|
||||||
|
{
|
||||||
|
log(L_INFO "CLI sleeps on write");
|
||||||
|
c->sleeping_on_tx = 1;
|
||||||
|
coro_suspend();
|
||||||
|
c->sleeping_on_tx = 0;
|
||||||
|
log(L_INFO "CLI wakeup on write");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (c->cont)
|
||||||
|
{
|
||||||
|
c->cont(c);
|
||||||
|
cli_write_trigger(c);
|
||||||
|
cli_yield(c);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
byte *d = c->rx_buf;
|
||||||
|
byte *dend = c->rx_buf + CLI_RX_BUF_SIZE - 2;
|
||||||
|
for (;;)
|
||||||
|
{
|
||||||
|
int ch = cli_getchar(c);
|
||||||
|
if (ch == '\r')
|
||||||
|
;
|
||||||
|
else if (ch == '\n')
|
||||||
|
break;
|
||||||
|
else if (d < dend)
|
||||||
|
*d++ = ch;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (d >= dend)
|
||||||
|
{
|
||||||
|
cli_printf(c, 9000, "Command too long");
|
||||||
|
cli_write_trigger(c);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
*d = 0;
|
||||||
|
cli_command(c);
|
||||||
|
cli_write_trigger(c);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
static int
|
static int
|
||||||
cli_connect(sock *s, uint size UNUSED)
|
cli_connect(sock *s, uint size UNUSED)
|
||||||
{
|
{
|
||||||
@ -447,7 +494,6 @@ cli_connect(sock *s, uint size UNUSED)
|
|||||||
|
|
||||||
if (config->cli_debug)
|
if (config->cli_debug)
|
||||||
log(L_INFO "CLI connect");
|
log(L_INFO "CLI connect");
|
||||||
s->rx_hook = cli_rx;
|
|
||||||
s->tx_hook = cli_tx;
|
s->tx_hook = cli_tx;
|
||||||
s->err_hook = cli_err;
|
s->err_hook = cli_err;
|
||||||
s->data = c = cli_new(s);
|
s->data = c = cli_new(s);
|
||||||
@ -456,6 +502,8 @@ cli_connect(sock *s, uint size UNUSED)
|
|||||||
c->rx_pos = c->rx_buf;
|
c->rx_pos = c->rx_buf;
|
||||||
c->rx_aux = NULL;
|
c->rx_aux = NULL;
|
||||||
rmove(s, c->pool);
|
rmove(s, c->pool);
|
||||||
|
c->coro = coro_new(c->pool, cli_coroutine, c);
|
||||||
|
coro_resume(c->coro);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -102,6 +102,13 @@ int sk_open_unix(struct birdsock *s, char *name);
|
|||||||
void *tracked_fopen(struct pool *, char *name, char *mode);
|
void *tracked_fopen(struct pool *, char *name, char *mode);
|
||||||
void test_old_bird(char *path);
|
void test_old_bird(char *path);
|
||||||
|
|
||||||
|
/* Co-routines */
|
||||||
|
typedef struct coroutine coroutine;
|
||||||
|
coroutine *coro_new(struct pool *pool, void (*entry_point)(void *arg), void *arg);
|
||||||
|
void coro_suspend(void);
|
||||||
|
void coro_resume(coroutine *c);
|
||||||
|
int coro_sk_read(struct birdsock *s);
|
||||||
|
|
||||||
/* krt.c bits */
|
/* krt.c bits */
|
||||||
|
|
||||||
void krt_io_init(void);
|
void krt_io_init(void);
|
||||||
|
Loading…
Reference in New Issue
Block a user