aboutsummaryrefslogtreecommitdiff
path: root/autistdraw.c
diff options
context:
space:
mode:
Diffstat (limited to 'autistdraw.c')
-rw-r--r--autistdraw.c309
1 files changed, 301 insertions, 8 deletions
diff --git a/autistdraw.c b/autistdraw.c
index 3938855..3e7b6c4 100644
--- a/autistdraw.c
+++ b/autistdraw.c
@@ -39,6 +39,22 @@
#define BITMAP_BLOCK_SIZE 50 ///< Step for extending bitmap size
+typedef enum network_mode network_mode_t;
+enum network_mode
+{
+ NETWORK_MODE_STANDALONE, ///< No networking taking place
+ NETWORK_MODE_SERVER, ///< We're the server
+ NETWORK_MODE_CLIENT ///< We're a client
+};
+
+typedef struct client client_t;
+struct client
+{
+ LIST_HEADER (client_t)
+
+ uv_tcp_t handle; ///< TCP connection handle
+};
+
typedef struct app_context app_context_t;
struct app_context
{
@@ -49,6 +65,16 @@ struct app_context
uv_timer_t tty_timer; ///< TTY timeout timer
uv_signal_t winch_watcher; ///< SIGWINCH watcher
+ network_mode_t mode; ///< Networking mode
+ char read_buf[8192]; ///< Global read buffer for libuv
+
+ // Client:
+ uv_tcp_t server_fd; ///< Connection to the server
+
+ // Server:
+ uv_tcp_t listen_fd; ///< Listening FD
+ client_t *clients; ///< Client connections
+
chtype palette[2 * 9]; ///< Attribute palette
uint8_t *bitmap; ///< Canvas data for drawing
@@ -79,10 +105,22 @@ app_init (app_context_t *self)
}
static void
+remove_client (app_context_t *app, client_t *client)
+{
+ // TODO: cancel any write requests?
+ // XXX: should we unref it?
+ uv_close ((uv_handle_t *) &client->handle, NULL);
+ LIST_UNLINK (app->clients, client);
+ free (client);
+}
+
+static void
app_free (app_context_t *self)
{
if (self->tk)
termo_destroy (self->tk);
+ while (self->clients)
+ remove_client (self, self->clients);
free (self->bitmap);
}
@@ -465,7 +503,7 @@ export_irc (app_context_t *app)
fclose (fp);
}
-// -----------------------------------------------------------------------------
+// --- Event handlers ----------------------------------------------------------
static bool
on_key (app_context_t *app, termo_key_t *key)
@@ -602,14 +640,158 @@ on_tty_readable (uv_poll_t *handle, int status, int events)
}
static void
-parse_program_arguments (app_context_t *app, int argc, char **argv)
+app_uv_allocator (uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf)
+{
+ // Let's just use a single "global" buffer
+ (void) suggested_size;
+
+ app_context_t *app = handle->loop->data;
+ buf->base = app->read_buf;
+ buf->len = sizeof app->read_buf;
+}
+
+static void
+on_server_data (uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf)
+{
+ (void) buf;
+
+ app_context_t *app = stream->loop->data;
+ if (nread == UV_EOF || nread < 0)
+ {
+ // TODO: cancel any write requests?
+ // XXX: should we unref it?
+ uv_close ((uv_handle_t *) &app->server_fd, NULL);
+
+ display ("Disconnected!");
+ beep (); // Beep beep! Made a boo-boo.
+
+ // Let the user save the picture at least.
+ // Also prevents us from trying to use the dead server handle.
+ app->mode = NETWORK_MODE_STANDALONE;
+ return;
+ }
+
+ // TODO: process the data
+}
+
+static void
+on_client_data (uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf)
+{
+ (void) buf;
+
+ app_context_t *app = stream->loop->data;
+ client_t *client = stream->data;
+ if (nread == UV_EOF || nread < 0)
+ {
+ remove_client (app, client);
+ return;
+ }
+
+ // TODO: process the data
+}
+
+static void
+on_new_client (uv_stream_t *server, int status)
+{
+ app_context_t *app = server->loop->data;
+ if (status)
+ return;
+
+ int err;
+ client_t *client = xcalloc (1, sizeof *client);
+ if ((err = uv_tcp_init (server->loop, &client->handle)))
+ goto free_client;
+ if ((err = uv_accept (server, (uv_stream_t *) &client->handle))
+ || (err = uv_read_start ((uv_stream_t *) &client->handle,
+ app_uv_allocator, on_client_data)))
+ // XXX: do we need to un-accept?
+ goto free_handle;
+
+ client->handle.data = client;
+ LIST_PREPEND (app->clients, client);
+ return;
+
+free_handle:
+ uv_close ((uv_handle_t *) &client->handle, NULL);
+ // XXX: should we unref it?
+free_client:
+ free (client);
+}
+
+// --- Program startup ---------------------------------------------------------
+
+typedef struct app_options app_options_t;
+struct app_options
+{
+ struct addrinfo *client_address; ///< Address to connect to
+ struct addrinfo *server_address; ///< Address to listen at
+};
+
+static void
+app_options_init (app_options_t *self)
{
- (void) app;
+ memset (self, 0, sizeof *self);
+}
+static void
+app_options_free (app_options_t *self)
+{
+ if (self->client_address) freeaddrinfo (self->client_address);
+ if (self->server_address) freeaddrinfo (self->server_address);
+}
+
+static struct addrinfo *
+parse_address (const char *address, int flags)
+{
+ char address_copy[strlen (address) + 1];
+ strcpy (address_copy, address);
+
+ char *colon = strrchr (address_copy, ':');
+ if (!colon)
+ {
+ fprintf (stderr, "error: no port number specified in `%s'\n", address);
+ return false;
+ }
+
+ char *host = address_copy, *service = colon + 1;
+
+ if (host == colon)
+ host = NULL;
+ else if (host < colon && *host == '[' && colon[-1] == ']')
+ {
+ // Remove IPv6 RFC 2732-style [] brackets from the host, if present.
+ // This also makes it possible to take the usage string literally. :))
+ host++;
+ colon[-1] = '\0';
+ }
+ else
+ *colon = '\0';
+
+ struct addrinfo *result, hints =
+ {
+ .ai_socktype = SOCK_STREAM,
+ .ai_protocol = IPPROTO_TCP,
+ .ai_flags = flags,
+ };
+ int err = getaddrinfo (host, service, &hints, &result);
+ if (err)
+ {
+ fprintf (stderr, "error: cannot resolve `%s', port `%s': %s\n",
+ host, service, gai_strerror (err));
+ return false;
+ }
+ return result;
+}
+
+static void
+parse_program_arguments (app_options_t *options, int argc, char **argv)
+{
static const struct opt opts[] =
{
{ 'h', "help", NULL, 0, "display this help and exit" },
{ 'V', "version", NULL, 0, "output version information and exit" },
+ { 's', "server", "[ADDRESS]:PORT", 0, "start a server" },
+ { 'c', "client", "[ADDRESS]:PORT", 0, "connect to a server" },
{ 0, NULL, NULL, 0, NULL }
};
@@ -627,12 +809,39 @@ parse_program_arguments (app_context_t *app, int argc, char **argv)
case 'V':
printf (PROJECT_NAME " " PROJECT_VERSION "\n");
exit (EXIT_SUCCESS);
+ case 's':
+ if (options->server_address)
+ {
+ fprintf (stderr, "%s: %s\n",
+ "error", "cannot specify multiple listening addresses");
+ exit (EXIT_FAILURE);
+ }
+ if (!(options->server_address = parse_address (optarg, AI_PASSIVE)))
+ exit (EXIT_FAILURE);
+ break;
+ case 'c':
+ if (options->client_address)
+ {
+ fprintf (stderr, "%s: %s\n",
+ "error", "cannot specify multiple addresses to connect to");
+ exit (EXIT_FAILURE);
+ }
+ if (!(options->client_address = parse_address (optarg, 0)))
+ exit (EXIT_FAILURE);
+ break;
default:
- fprintf (stderr, "%s: %s", "error", "wrong options");
+ fprintf (stderr, "%s: %s\n", "error", "wrong options");
opt_handler_usage (&oh, stderr);
exit (EXIT_FAILURE);
}
+ if (options->client_address && options->server_address)
+ {
+ fprintf (stderr, "%s: %s\n",
+ "error", "cannot be both a server and a client");
+ exit (EXIT_FAILURE);
+ }
+
argc -= optind;
argv += optind;
@@ -645,6 +854,77 @@ parse_program_arguments (app_context_t *app, int argc, char **argv)
opt_handler_free (&oh);
}
+static void
+initialize_client (app_context_t *app, struct addrinfo *address)
+{
+ app->mode = NETWORK_MODE_CLIENT;
+
+ int sock_fd, err;
+ for (; address; address = address->ai_next)
+ {
+ sock_fd = socket (address->ai_family,
+ address->ai_socktype, address->ai_protocol);
+ if (sock_fd == -1)
+ continue;
+
+ char host_buf[NI_MAXHOST], serv_buf[NI_MAXSERV];
+ err = getnameinfo (address->ai_addr, address->ai_addrlen,
+ host_buf, sizeof host_buf, serv_buf, sizeof serv_buf,
+ NI_NUMERICHOST | NI_NUMERICSERV);
+ if (err)
+ {
+ fprintf (stderr, "%s: %s: %s\n",
+ "error", "getnameinfo", gai_strerror (err));
+ fprintf (stderr, "connecting...\n");
+ }
+ else
+ {
+ char *x = format_host_port_pair (host_buf, serv_buf);
+ fprintf (stderr, "connecting to %s...\n", x);
+ free (x);
+ }
+
+ if (!connect (sock_fd, address->ai_addr, address->ai_addrlen))
+ break;
+
+ xclose (sock_fd);
+ }
+
+ if (!address)
+ {
+ fprintf (stderr, "%s: %s\n", "error", "connection failed");
+ exit (EXIT_FAILURE);
+ }
+
+ set_blocking (sock_fd, false);
+ if ((err = uv_tcp_init (uv_default_loop (), &app->server_fd))
+ || (err = uv_tcp_open (&app->server_fd, sock_fd))
+ || (err = uv_tcp_keepalive (&app->server_fd, true, 30))
+ || (err = uv_read_start ((uv_stream_t *) &app->server_fd,
+ app_uv_allocator, on_server_data)))
+ {
+ fprintf (stderr, "%s: %s: %s\n",
+ "error", "initialization failed", uv_strerror (err));
+ exit (EXIT_FAILURE);
+ }
+}
+
+static void
+initialize_server (app_context_t *app, struct addrinfo *address)
+{
+ app->mode = NETWORK_MODE_SERVER;
+
+ int err;
+ if ((err = uv_tcp_init (uv_default_loop (), &app->listen_fd))
+ || (err = uv_tcp_bind (&app->listen_fd, address->ai_addr, 0))
+ || (err = uv_listen ((uv_stream_t *) &app->listen_fd, 10, on_new_client)))
+ {
+ fprintf (stderr, "%s: %s: %s\n",
+ "error", "initialization failed", uv_strerror (err));
+ exit (EXIT_FAILURE);
+ }
+}
+
int
main (int argc, char *argv[])
{
@@ -653,12 +933,25 @@ main (int argc, char *argv[])
app_context_t app;
app_init (&app);
- parse_program_arguments (&app, argc, argv);
+
+ app_options_t options;
+ app_options_init (&options);
+ parse_program_arguments (&options, argc, argv);
+
+ if (options.client_address)
+ initialize_client (&app, options.client_address);
+ else if (options.server_address)
+ initialize_server (&app, options.server_address);
+ else
+ app.mode = NETWORK_MODE_STANDALONE;
+
+ app_options_free (&options);
termo_t *tk = termo_new (STDIN_FILENO, NULL, 0);
if (!tk)
{
- fprintf (stderr, "Cannot allocate termo instance\n");
+ fprintf (stderr, "%s: %s\n",
+ "error", "cannot allocate termo instance\n");
exit (EXIT_FAILURE);
}
@@ -669,7 +962,7 @@ main (int argc, char *argv[])
// Set up curses for our drawing needs
if (!initscr () || nonl () == ERR || curs_set (0) == ERR)
{
- fprintf (stderr, "Cannot initialize curses\n");
+ fprintf (stderr, "%s: %s\n", "error", "cannot initialize curses");
exit (EXIT_FAILURE);
}
@@ -694,8 +987,8 @@ main (int argc, char *argv[])
uv_run (loop, UV_RUN_DEFAULT);
endwin ();
- uv_loop_close (loop);
app_free (&app);
+ uv_loop_close (loop);
return 0;
}