/*
 * machinarium.
 *
 * Cooperative multitasking engine.
*/

#include <machinarium.h>
#include <machinarium_private.h>
#include <string.h>
#include <assert.h>

static void
server(void *arg)
{
	printf("server: started\n");

	machine_t machine = arg;
	machine_io_t server = machine_create_io(machine);

	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);
	if (rc < 0) {
		printf("server: bind failed\n");
		machine_close(server);
		return;
	}

	printf("server: waiting for connections (127.0.0.1:7778)\n");
	machine_io_t client;
	rc = machine_accept(server, &client, 16, 0);
	if (rc < 0) {
		printf("accept error: %s\n", machine_error(server));
		machine_close(server);
		machine_free_io(server);
		return;
	}
	char msg[] = "hello world";
	rc = machine_write(client, msg, sizeof(msg), 0);
	if (rc < 0) {
		printf("server: write error: %s\n", machine_error(client));
		machine_close(server);
		machine_free_io(server);
		return;
	}

	machine_close(client);
	machine_free_io(client);
	machine_close(server);
	machine_free_io(server);

	printf("server: done\n");
}

static void
client(void *arg)
{
	printf("client: started\n");

	machine_t machine = arg;
	machine_io_t client = machine_create_io(machine);

	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, 0);
	if (rc < 0) {
		printf("client: connect failed: %s\n", machine_error(client));
		machine_close(client);
		machine_free_io(client);
		return;
	}

	printf("client: connected\n");

	char buf[16];
	rc = machine_read(client, buf, 12, 0);
	if (rc < 0) {
		printf("client: read failed: %s\n", machine_error(client));
		machine_close(client);
		machine_free_io(client);
		return;
	}

	assert(memcmp(buf, "hello world", 12) == 0);

	rc = machine_read(client, buf, 1, 0);
	/* eof */
	assert(rc == -1);

	machine_close(client);
	machine_free_io(client);

	printf("client: done\n");
	machine_stop(machine);
}

int
main(int argc, char *argv[])
{
	machine_t machine = machine_create();
	machine_create_fiber(machine, server, machine);
	machine_create_fiber(machine, client, machine);
	machine_start(machine);
	machine_free(machine);
	return 0;
}