diff options
Diffstat (limited to 'gio/tests/socket-testserver.c')
-rw-r--r-- | gio/tests/socket-testserver.c | 366 |
1 files changed, 366 insertions, 0 deletions
diff --git a/gio/tests/socket-testserver.c b/gio/tests/socket-testserver.c new file mode 100644 index 000000000..61715b02d --- /dev/null +++ b/gio/tests/socket-testserver.c @@ -0,0 +1,366 @@ +#include <gio/gio.h> +#include <gio/gunixsocketaddress.h> +#include <glib.h> +#include <stdlib.h> +#include <string.h> + +GMainLoop *loop; + +int default_port = 7777; +gboolean verbose = FALSE; +gboolean dont_reuse_address = FALSE; +gboolean non_blocking = FALSE; +gboolean use_udp = 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, &default_port, + "Local port to bind to", NULL}, + {"cancel", 'c', 0, G_OPTION_ARG_INT, &cancel_timeout, + "Cancel any op after the specified amount of seconds", NULL}, + {"udp", 'u', 0, G_OPTION_ARG_NONE, &use_udp, + "Use udp instead of tcp", NULL}, + {"verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose, + "Be verbose", NULL}, + {"no-reuse", 0, 0, G_OPTION_ARG_NONE, &dont_reuse_address, + "Don't SOADDRREUSE", NULL}, + {"non-blocking", 'n', 0, G_OPTION_ARG_NONE, &non_blocking, + "Enable 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}, +#endif + {"delay", 'd', 0, G_OPTION_ARG_INT, &delay, + "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"}, + G_OPTION_ENTRY_NULL +}; + +#include "socket-common.c" + +int +main (int argc, + char *argv[]) +{ + GSocket *socket, *new_socket, *recv_socket; + GSocketAddress *src_address; + GSocketAddress *address = NULL; + GSocketType socket_type; + GSocketFamily socket_family; + GError *error = NULL; + GOptionContext *context; + GCancellable *cancellable; + char *display_addr; + GTlsCertificate *tlscert = NULL; + GIOStream *connection; + GInputStream *istream; + GOutputStream *ostream; + + context = g_option_context_new (" - Test GSocket server stuff"); + g_option_context_add_main_entries (context, cmd_entries, NULL); + if (!g_option_context_parse (context, &argc, &argv, &error)) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (unix_socket && argc != 2) + { + g_printerr ("%s: %s\n", argv[0], "Need to specify unix socket name"); + return 1; + } + + if (cancel_timeout) + { + GThread *thread; + cancellable = g_cancellable_new (); + thread = g_thread_new ("cancel", cancel_thread, cancellable); + g_thread_unref (thread); + } + else + { + 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) + socket_type = G_SOCKET_TYPE_DATAGRAM; + else + socket_type = G_SOCKET_TYPE_STREAM; + + if (unix_socket) + socket_family = G_SOCKET_FAMILY_UNIX; + else + socket_family = G_SOCKET_FAMILY_IPV4; + + socket = g_socket_new (socket_family, socket_type, 0, &error); + + if (socket == NULL) + { + g_printerr ("%s: %s\n", argv[0], error->message); + return 1; + } + + if (non_blocking) + g_socket_set_blocking (socket, FALSE); + + if (unix_socket) + { + src_address = socket_address_from_string (argv[1]); + if (src_address == NULL) + { + g_printerr ("%s: Could not parse '%s' as unix socket name\n", argv[0], argv[1]); + return 1; + } + } + else + { + src_address = g_inet_socket_address_new (g_inet_address_new_any (G_SOCKET_FAMILY_IPV4), default_port); + } + + if (!g_socket_bind (socket, src_address, !dont_reuse_address, &error)) + { + g_printerr ("Can't bind socket: %s\n", error->message); + return 1; + } + g_object_unref (src_address); + + if (!use_udp) + { + if (!g_socket_listen (socket, &error)) + { + g_printerr ("Can't listen on socket: %s\n", error->message); + return 1; + } + + address = g_socket_get_local_address (socket, &error); + if (!address) + { + g_printerr ("Error getting local address: %s\n", + error->message); + return 1; + } + display_addr = socket_address_to_string (address); + g_print ("listening on %s...\n", display_addr); + g_free (display_addr); + + ensure_socket_condition (socket, G_IO_IN, cancellable); + new_socket = g_socket_accept (socket, cancellable, &error); + if (!new_socket) + { + g_printerr ("Error accepting socket: %s\n", + error->message); + return 1; + } + + if (non_blocking) + g_socket_set_blocking (new_socket, FALSE); + if (read_timeout) + g_socket_set_timeout (new_socket, read_timeout); + + address = g_socket_get_remote_address (new_socket, &error); + if (!address) + { + g_printerr ("Error getting remote address: %s\n", + error->message); + return 1; + } + + display_addr = socket_address_to_string (address); + g_print ("got a new connection from %s\n", display_addr); + g_free(display_addr); + 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; + connection = NULL; + } + + if (tlscert) + { + GIOStream *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 = tls_conn; + } + + if (connection) + { + istream = g_io_stream_get_input_stream (connection); + ostream = g_io_stream_get_output_stream (connection); + } + else + { + g_assert (use_udp); + istream = NULL; + ostream = NULL; + } + + while (TRUE) + { + gchar buffer[4096]; + gssize size; + gsize to_send; + + if (use_udp) + { + 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); + } + + if (size < 0) + { + g_printerr ("Error receiving from socket: %s\n", + error->message); + return 1; + } + + if (size == 0) + break; + + g_print ("received %" G_GSSIZE_FORMAT " bytes of data", size); + if (use_udp) + g_print (" from %s", socket_address_to_string (address)); + g_print ("\n"); + + if (verbose) + g_print ("-------------------------\n" + "%.*s\n" + "-------------------------\n", + (int)size, buffer); + + to_send = size; + + if (delay) + { + if (verbose) + g_print ("delaying %d seconds before response\n", delay); + g_usleep (1000 * 1000 * delay); + } + + while (to_send > 0) + { + if (use_udp) + { + ensure_socket_condition (recv_socket, G_IO_OUT, cancellable); + size = g_socket_send_to (recv_socket, address, + buffer, to_send, cancellable, &error); + } + else + { + ensure_connection_condition (connection, G_IO_OUT, cancellable); + size = g_output_stream_write (ostream, + buffer, to_send, + cancellable, &error); + } + + if (size < 0) + { + if (g_error_matches (error, + G_IO_ERROR, + G_IO_ERROR_WOULD_BLOCK)) + { + g_print ("socket send would block, handling\n"); + g_error_free (error); + error = NULL; + continue; + } + else + { + g_printerr ("Error sending to socket: %s\n", + error->message); + return 1; + } + } + + g_print ("sent %" G_GSSIZE_FORMAT " bytes of data\n", size); + + if (size == 0) + { + g_printerr ("Unexpected short write\n"); + return 1; + } + + to_send -= size; + } + } + + g_print ("connection closed\n"); + + if (connection) + { + if (!g_io_stream_close (connection, NULL, &error)) + { + g_printerr ("Error closing connection stream: %s\n", + error->message); + return 1; + } + g_object_unref (connection); + } + + if (!g_socket_close (socket, &error)) + { + g_printerr ("Error closing socket: %s\n", + error->message); + return 1; + } + g_object_unref (socket); + + return 0; +} |