mirror of https://github.com/yandex/odyssey.git
142 lines
2.6 KiB
C
142 lines
2.6 KiB
C
|
|
#include <machinarium.h>
|
|
#include <odyssey_test.h>
|
|
|
|
#include <string.h>
|
|
#include <arpa/inet.h>
|
|
|
|
static void
|
|
server(void *arg)
|
|
{
|
|
(void)arg;
|
|
machine_io_t *server = machine_io_create();
|
|
test(server != NULL);
|
|
|
|
struct sockaddr_in sa;
|
|
sa.sin_family = AF_INET;
|
|
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
|
|
sa.sin_port = htons(7778);
|
|
int rc;
|
|
rc = machine_bind(server, (struct sockaddr*)&sa);
|
|
test(rc == 0);
|
|
|
|
machine_io_t *client_a;
|
|
rc = machine_accept(server, &client_a, 16, 1, UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
machine_io_t *client_b;
|
|
rc = machine_accept(server, &client_b, 16, 1, UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
machine_io_t *client_c;
|
|
rc = machine_accept(server, &client_c, 16, 1, UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
machine_io_t *io_set_ready[3] = {NULL, NULL, NULL};
|
|
machine_io_t *io_set[3] = {client_a, client_b, client_c};
|
|
int io_count = 3;
|
|
|
|
while (io_count > 0)
|
|
{
|
|
int ready;
|
|
ready = machine_read_poll(io_set, io_set_ready, io_count, UINT32_MAX);
|
|
test(ready > 0);
|
|
|
|
int io_pos = 0;
|
|
while (io_pos < ready) {
|
|
machine_io_t *io = io_set_ready[io_pos];
|
|
|
|
char buf[1024];
|
|
rc = machine_read(io, buf, sizeof(buf), UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
/* test eof */
|
|
rc = machine_read(io, buf, sizeof(buf), UINT32_MAX);
|
|
test(rc == -1);
|
|
|
|
rc = machine_close(io);
|
|
test(rc == 0);
|
|
machine_io_free(io);
|
|
|
|
int i = 0;
|
|
int j = 0;
|
|
while (i < 3) {
|
|
if (io_set[i] == io) {
|
|
/* skip */
|
|
io_set[i] = NULL;
|
|
} else {
|
|
io_set[j] = io_set[i];
|
|
j++;
|
|
}
|
|
i++;
|
|
}
|
|
io_count--;
|
|
io_pos++;
|
|
}
|
|
}
|
|
|
|
rc = machine_close(server);
|
|
test(rc == 0);
|
|
machine_io_free(server);
|
|
}
|
|
|
|
static void
|
|
client(void *arg)
|
|
{
|
|
(void)arg;
|
|
machine_io_t *client = machine_io_create();
|
|
test(client != NULL);
|
|
|
|
struct sockaddr_in sa;
|
|
sa.sin_family = AF_INET;
|
|
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
|
|
sa.sin_port = htons(7778);
|
|
int rc;
|
|
rc = machine_connect(client, (struct sockaddr*)&sa, UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
char buf[1024];
|
|
memset(buf, 'x', sizeof(buf));
|
|
|
|
rc = machine_write(client, buf, 1024, UINT32_MAX);
|
|
test(rc == 0);
|
|
|
|
rc = machine_close(client);
|
|
test(rc == 0);
|
|
machine_io_free(client);
|
|
}
|
|
|
|
static void
|
|
test_cs(void *arg)
|
|
{
|
|
(void)arg;
|
|
int rc;
|
|
rc = machine_coroutine_create(server, NULL);
|
|
test(rc != -1);
|
|
|
|
rc = machine_coroutine_create(client, NULL);
|
|
test(rc != -1);
|
|
|
|
rc = machine_coroutine_create(client, NULL);
|
|
test(rc != -1);
|
|
|
|
rc = machine_coroutine_create(client, NULL);
|
|
test(rc != -1);
|
|
}
|
|
|
|
void
|
|
machinarium_test_read_poll1(void)
|
|
{
|
|
machinarium_init();
|
|
|
|
int id;
|
|
id = machine_create("test", test_cs, NULL);
|
|
test(id != -1);
|
|
|
|
int rc;
|
|
rc = machine_wait(id);
|
|
test(rc != -1);
|
|
|
|
machinarium_free();
|
|
}
|