1 files changed, 108 insertions, 0 deletions
diff --git a/src/connection.c b/src/connection.c
new file mode 100644
index 0000000..3494d08
--- /dev/null
+++ b/src/connection.c
@@ -0,0 +1,108 @@
+#include <glib.h>
+#include <gio/gio.h>
+
+#include "connection.h"
+#include "cmumble.h"
+
+static gboolean
+read_cb(GObject *pollable_stream, gpointer data)
+{
+	GPollableInputStream *input = G_POLLABLE_INPUT_STREAM(pollable_stream);
+	struct context *ctx = data;
+	gint count;
+
+	do {
+		count = recv_msg(ctx, ctx->con.callbacks);
+	} while (count && g_pollable_input_stream_is_readable(input));
+
+	return TRUE;
+}
+
+static gboolean
+do_ping(struct context *ctx)
+{
+	MumbleProto__Ping ping;
+	GTimeVal tv;
+
+	g_get_current_time(&tv);
+	mumble_proto__ping__init(&ping);
+	ping.timestamp = tv.tv_sec;
+	ping.resync = 1;
+	send_msg(ctx, &ping.base);
+
+	return TRUE;
+}
+
+static void
+setup_ping_timer(struct context *ctx)
+{
+	GSource *source;
+
+	source = g_timeout_source_new_seconds(5);
+	g_source_set_callback(source, (GSourceFunc) do_ping, ctx, NULL);
+	g_source_attach(source, NULL);
+	g_source_unref(source);
+}
+
+int
+cmumble_connection_init(struct context *ctx,
+			const char *host, int port,
+			struct mumble_callbacks *callbacks)
+{
+	struct cmumble_connection *con = &ctx->con;
+	GError *error = NULL;
+
+	con->callbacks = callbacks;
+	con->sock_client = g_socket_client_new();
+	g_socket_client_set_tls(con->sock_client, TRUE);
+	g_socket_client_set_tls_validation_flags(con->sock_client,
+						 G_TLS_CERTIFICATE_INSECURE);
+	g_socket_client_set_family(con->sock_client, G_SOCKET_FAMILY_IPV4);
+	g_socket_client_set_protocol(con->sock_client,
+				     G_SOCKET_PROTOCOL_TCP);
+	g_socket_client_set_socket_type(con->sock_client,
+					G_SOCKET_TYPE_STREAM);
+
+	con->conn =
+		g_socket_client_connect_to_host(con->sock_client,
+						host, port, NULL, &error);
+	if (error) {
+		g_printerr("connect failed: %s\n", error->message);
+		return -1;
+	}
+
+	g_object_get(G_OBJECT(con->conn),
+		     "input-stream", &con->input,
+		     "output-stream", &con->output, NULL);
+
+	if (!G_IS_POLLABLE_INPUT_STREAM(con->input) ||
+	    !g_pollable_input_stream_can_poll(con->input)) {
+		g_printerr("Error: GSocketConnection is not pollable\n");
+		return 1;
+	}
+
+	con->source = g_pollable_input_stream_create_source(con->input, NULL);
+	g_source_set_callback(con->source, (GSourceFunc) read_cb, ctx, NULL);
+	g_source_attach(con->source, NULL);
+	g_source_unref(con->source);
+
+	setup_ping_timer(ctx);
+
+	return 0;
+}
+
+int
+cmumble_connection_fini(struct context *ctx)
+{
+	g_source_remove(g_source_get_id(ctx->con.source));
+	g_source_unref(ctx->con.source);
+
+	g_object_unref(G_OBJECT(ctx->con.input));
+	g_object_unref(G_OBJECT(ctx->con.output));
+
+	g_io_stream_close(G_IO_STREAM(ctx->con.conn), NULL, NULL);
+	g_object_unref(G_OBJECT(ctx->con.conn));
+	g_object_unref(G_OBJECT(ctx->con.sock_client));
+
+	return 0;
+}