mirror of
https://gitlab.gnome.org/GNOME/glib.git
synced 2025-08-30 21:04:13 +02:00
Add initial TLS (SSL) support to gio
This adds an extension point for TLS connections to gio, with a gnutls-based implementation in glib-networking. Full TLS support is still a work in progress; the current API is missing some features, and parts of it may still be changed before 2.28. https://bugzilla.gnome.org/show_bug.cgi?id=588189
This commit is contained in:
@@ -4,8 +4,6 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "socket-common.c"
|
||||
|
||||
GMainLoop *loop;
|
||||
|
||||
int port = 7777;
|
||||
@@ -13,11 +11,11 @@ gboolean verbose = FALSE;
|
||||
gboolean dont_reuse_address = FALSE;
|
||||
gboolean non_blocking = FALSE;
|
||||
gboolean use_udp = FALSE;
|
||||
gboolean use_source = FALSE;
|
||||
int cancel_timeout = 0;
|
||||
int read_timeout = 0;
|
||||
int delay = 0;
|
||||
gboolean unix_socket = FALSE;
|
||||
const char *tls_cert_file = NULL;
|
||||
|
||||
static GOptionEntry cmd_entries[] = {
|
||||
{"port", 'p', 0, G_OPTION_ARG_INT, &port,
|
||||
@@ -32,8 +30,6 @@ static GOptionEntry cmd_entries[] = {
|
||||
"Don't SOADDRREUSE", NULL},
|
||||
{"non-blocking", 'n', 0, G_OPTION_ARG_NONE, &non_blocking,
|
||||
"Enable non-blocking i/o", NULL},
|
||||
{"use-source", 's', 0, G_OPTION_ARG_NONE, &use_source,
|
||||
"Use GSource to wait for non-blocking i/o", NULL},
|
||||
#ifdef G_OS_UNIX
|
||||
{"unix", 'U', 0, G_OPTION_ARG_NONE, &unix_socket,
|
||||
"Use a unix socket instead of IP", NULL},
|
||||
@@ -42,63 +38,12 @@ static GOptionEntry cmd_entries[] = {
|
||||
"Delay responses by the specified number of seconds", NULL},
|
||||
{"timeout", 't', 0, G_OPTION_ARG_INT, &read_timeout,
|
||||
"Time out reads after the specified number of seconds", NULL},
|
||||
{"tls", 'T', 0, G_OPTION_ARG_STRING, &tls_cert_file,
|
||||
"Use TLS (SSL) with indicated server certificate", "CERTFILE"},
|
||||
{NULL}
|
||||
};
|
||||
|
||||
static gboolean
|
||||
source_ready (gpointer data,
|
||||
GIOCondition condition)
|
||||
{
|
||||
g_main_loop_quit (loop);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
static void
|
||||
ensure_condition (GSocket *socket,
|
||||
const char *where,
|
||||
GCancellable *cancellable,
|
||||
GIOCondition condition)
|
||||
{
|
||||
GError *error = NULL;
|
||||
GSource *source;
|
||||
|
||||
if (!non_blocking)
|
||||
return;
|
||||
|
||||
if (use_source)
|
||||
{
|
||||
source = g_socket_create_source (socket,
|
||||
condition,
|
||||
cancellable);
|
||||
g_source_set_callback (source,
|
||||
(GSourceFunc) source_ready,
|
||||
NULL, NULL);
|
||||
g_source_attach (source, NULL);
|
||||
g_source_unref (source);
|
||||
g_main_loop_run (loop);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (!g_socket_condition_wait (socket, condition, cancellable, &error))
|
||||
{
|
||||
g_printerr ("condition wait error for %s: %s\n",
|
||||
where,
|
||||
error->message);
|
||||
exit (1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static gpointer
|
||||
cancel_thread (gpointer data)
|
||||
{
|
||||
GCancellable *cancellable = data;
|
||||
|
||||
g_usleep (1000*1000*cancel_timeout);
|
||||
g_print ("Cancelling\n");
|
||||
g_cancellable_cancel (cancellable);
|
||||
return NULL;
|
||||
}
|
||||
#include "socket-common.c"
|
||||
|
||||
int
|
||||
main (int argc,
|
||||
@@ -113,6 +58,10 @@ main (int argc,
|
||||
GOptionContext *context;
|
||||
GCancellable *cancellable;
|
||||
char *display_addr;
|
||||
GTlsCertificate *tlscert = NULL;
|
||||
GIOStream *connection;
|
||||
GInputStream *istream;
|
||||
GOutputStream *ostream;
|
||||
|
||||
g_thread_init (NULL);
|
||||
|
||||
@@ -142,6 +91,23 @@ main (int argc,
|
||||
cancellable = NULL;
|
||||
}
|
||||
|
||||
if (tls_cert_file)
|
||||
{
|
||||
if (use_udp)
|
||||
{
|
||||
g_printerr ("DTLS (TLS over UDP) is not supported");
|
||||
return 1;
|
||||
}
|
||||
|
||||
tlscert = g_tls_certificate_new_from_file (tls_cert_file, &error);
|
||||
if (!tlscert)
|
||||
{
|
||||
g_printerr ("Could not read server certificate '%s': %s\n",
|
||||
tls_cert_file, error->message);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
loop = g_main_loop_new (NULL, FALSE);
|
||||
|
||||
if (use_udp)
|
||||
@@ -205,7 +171,7 @@ main (int argc,
|
||||
g_print ("listening on %s...\n", display_addr);
|
||||
g_free (display_addr);
|
||||
|
||||
ensure_condition (socket, "accept", cancellable, G_IO_IN);
|
||||
ensure_socket_condition (socket, G_IO_IN, cancellable);
|
||||
new_socket = g_socket_accept (socket, cancellable, &error);
|
||||
if (!new_socket)
|
||||
{
|
||||
@@ -233,13 +199,45 @@ main (int argc,
|
||||
g_object_unref (address);
|
||||
|
||||
recv_socket = new_socket;
|
||||
|
||||
connection = G_IO_STREAM (g_socket_connection_factory_create_connection (recv_socket));
|
||||
g_object_unref (new_socket);
|
||||
}
|
||||
else
|
||||
{
|
||||
recv_socket = socket;
|
||||
new_socket = NULL;
|
||||
connection = NULL;
|
||||
}
|
||||
|
||||
if (tlscert)
|
||||
{
|
||||
GTlsServerConnection *tls_conn;
|
||||
|
||||
tls_conn = g_tls_server_connection_new (connection, tlscert, &error);
|
||||
if (!tls_conn)
|
||||
{
|
||||
g_printerr ("Could not create TLS connection: %s\n",
|
||||
error->message);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!g_tls_connection_handshake (G_TLS_CONNECTION (tls_conn),
|
||||
cancellable, &error))
|
||||
{
|
||||
g_printerr ("Error during TLS handshake: %s\n",
|
||||
error->message);
|
||||
return 1;
|
||||
}
|
||||
|
||||
g_object_unref (connection);
|
||||
connection = G_IO_STREAM (tls_conn);
|
||||
}
|
||||
|
||||
if (connection)
|
||||
{
|
||||
istream = g_io_stream_get_input_stream (connection);
|
||||
ostream = g_io_stream_get_output_stream (connection);
|
||||
}
|
||||
|
||||
while (TRUE)
|
||||
{
|
||||
@@ -247,14 +245,20 @@ main (int argc,
|
||||
gssize size;
|
||||
gsize to_send;
|
||||
|
||||
ensure_condition (recv_socket, "receive", cancellable, G_IO_IN);
|
||||
if (use_udp)
|
||||
size = g_socket_receive_from (recv_socket, &address,
|
||||
{
|
||||
ensure_socket_condition (recv_socket, G_IO_IN, cancellable);
|
||||
size = g_socket_receive_from (recv_socket, &address,
|
||||
buffer, sizeof buffer,
|
||||
cancellable, &error);
|
||||
}
|
||||
else
|
||||
{
|
||||
ensure_connection_condition (connection, G_IO_IN, cancellable);
|
||||
size = g_input_stream_read (istream,
|
||||
buffer, sizeof buffer,
|
||||
cancellable, &error);
|
||||
else
|
||||
size = g_socket_receive (recv_socket, buffer, sizeof buffer,
|
||||
cancellable, &error);
|
||||
}
|
||||
|
||||
if (size < 0)
|
||||
{
|
||||
@@ -288,13 +292,19 @@ main (int argc,
|
||||
|
||||
while (to_send > 0)
|
||||
{
|
||||
ensure_condition (recv_socket, "send", cancellable, G_IO_OUT);
|
||||
if (use_udp)
|
||||
size = g_socket_send_to (recv_socket, address,
|
||||
buffer, to_send, cancellable, &error);
|
||||
{
|
||||
ensure_socket_condition (recv_socket, G_IO_OUT, cancellable);
|
||||
size = g_socket_send_to (recv_socket, address,
|
||||
buffer, to_send, cancellable, &error);
|
||||
}
|
||||
else
|
||||
size = g_socket_send (recv_socket, buffer, to_send,
|
||||
cancellable, &error);
|
||||
{
|
||||
ensure_connection_condition (connection, G_IO_OUT, cancellable);
|
||||
size = g_output_stream_write (ostream,
|
||||
buffer, to_send,
|
||||
cancellable, &error);
|
||||
}
|
||||
|
||||
if (size < 0)
|
||||
{
|
||||
@@ -329,16 +339,15 @@ main (int argc,
|
||||
|
||||
g_print ("connection closed\n");
|
||||
|
||||
if (new_socket)
|
||||
if (connection)
|
||||
{
|
||||
if (!g_socket_close (new_socket, &error))
|
||||
if (!g_io_stream_close (connection, NULL, &error))
|
||||
{
|
||||
g_printerr ("Error closing connection socket: %s\n",
|
||||
g_printerr ("Error closing connection stream: %s\n",
|
||||
error->message);
|
||||
return 1;
|
||||
}
|
||||
|
||||
g_object_unref (G_OBJECT (new_socket));
|
||||
g_object_unref (connection);
|
||||
}
|
||||
|
||||
if (!g_socket_close (socket, &error))
|
||||
@@ -347,8 +356,7 @@ main (int argc,
|
||||
error->message);
|
||||
return 1;
|
||||
}
|
||||
|
||||
g_object_unref (G_OBJECT (socket));
|
||||
g_object_unref (socket);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
Reference in New Issue
Block a user