mirror of https://github.com/yandex/odyssey.git
67 lines
1.4 KiB
C
67 lines
1.4 KiB
C
|
|
/*
|
|
* machinarium.
|
|
*
|
|
* cooperative multitasking engine.
|
|
*/
|
|
|
|
#include <machinarium_private.h>
|
|
#include <machinarium.h>
|
|
|
|
static void
|
|
mm_accept_cb(uv_stream_t *handle, int status)
|
|
{
|
|
mm_io_t *io = handle->data;
|
|
io->accept_status = status;
|
|
mm_scheduler_wakeup(io->accept_fiber);
|
|
}
|
|
|
|
static inline machine_io_t
|
|
mm_accept_client(mm_io_t *io)
|
|
{
|
|
mm_io_t *client = machine_create_io(io->machine);
|
|
if (client == NULL) {
|
|
io->accept_status = -ENOMEM;
|
|
return NULL;
|
|
}
|
|
io->accept_status =
|
|
uv_accept((uv_stream_t*)&io->handle,
|
|
(uv_stream_t*)&client->handle);
|
|
if (io->accept_status < 0) {
|
|
machine_close(client);
|
|
return NULL;
|
|
}
|
|
client->connected = 1;
|
|
uv_fileno((uv_handle_t*)&client->handle,
|
|
&client->fd);
|
|
return client;
|
|
}
|
|
|
|
MACHINE_API int
|
|
machine_accept(machine_io_t obj, int backlog, machine_io_t *client)
|
|
{
|
|
mm_io_t *io = obj;
|
|
mm_fiber_t *current = mm_scheduler_current(&io->machine->scheduler);
|
|
if (mm_fiber_is_cancelled(current))
|
|
return -ECANCELED;
|
|
if (io->accept_fiber)
|
|
return -1;
|
|
io->accept_status = 0;
|
|
io->accept_fiber = current;
|
|
int rc;
|
|
rc = uv_listen((uv_stream_t*)&io->handle, backlog, mm_accept_cb);
|
|
if (rc < 0) {
|
|
io->accept_fiber = NULL;
|
|
return rc;
|
|
}
|
|
mm_scheduler_yield(&io->machine->scheduler);
|
|
rc = io->accept_status;
|
|
io->accept_fiber = NULL;
|
|
if (rc < 0)
|
|
return rc;
|
|
*client = mm_accept_client(io);
|
|
if (*client == NULL)
|
|
rc = io->accept_status;
|
|
return rc;
|
|
}
|