aboutsummaryrefslogtreecommitdiff
path: root/src/plugins/tests/test-port-context.c
diff options
context:
space:
mode:
authorAleksander Morgado <aleksandermj@chromium.org>2022-12-08 13:37:55 +0000
committerAleksander Morgado <aleksander@aleksander.es>2023-01-03 13:56:25 +0000
commite14b904cbd6816cb0227d519d308ae71ddaf6e07 (patch)
tree4997ab68cc606fdf4d72a571e821cec0c8df42ef /src/plugins/tests/test-port-context.c
parent072d7ac9065f444e83b390a1e2af5471ac0d48f6 (diff)
build: move plugins directory to src/plugins
We are going to allow including the plugin sources built within the ModemManager daemon binary; moving the sources within the daemon sources directory makes it easier.
Diffstat (limited to 'src/plugins/tests/test-port-context.c')
-rw-r--r--src/plugins/tests/test-port-context.c422
1 files changed, 422 insertions, 0 deletions
diff --git a/src/plugins/tests/test-port-context.c b/src/plugins/tests/test-port-context.c
new file mode 100644
index 00000000..e96cff7b
--- /dev/null
+++ b/src/plugins/tests/test-port-context.c
@@ -0,0 +1,422 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details:
+ *
+ * Copyright (C) 2013 Aleksander Morgado <aleksander@gnu.org>
+ */
+
+#include <gio/gio.h>
+#include <gio/gunixsocketaddress.h>
+#include <string.h>
+
+#include "test-port-context.h"
+
+#define BUFFER_SIZE 1024
+
+struct _TestPortContext {
+ gchar *name;
+ GThread *thread;
+ gboolean ready;
+ GCond ready_cond;
+ GMutex ready_mutex;
+ GMainLoop *loop;
+ GMainContext *context;
+ GSocket *socket;
+ GSocketService *socket_service;
+ GList *clients;
+ GHashTable *commands;
+};
+
+/*****************************************************************************/
+
+void
+test_port_context_set_command (TestPortContext *self,
+ const gchar *command,
+ const gchar *response)
+{
+ if (G_UNLIKELY (!self->commands))
+ self->commands = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_free);
+ g_hash_table_replace (self->commands, g_strdup (command), g_strcompress (response));
+}
+
+void
+test_port_context_load_commands (TestPortContext *self,
+ const gchar *file)
+{
+ GError *error = NULL;
+ gchar *contents;
+ gchar *current;
+
+ if (!g_file_get_contents (file, &contents, NULL, &error))
+ g_error ("Couldn't load commands file '%s': %s",
+ g_filename_display_name (file),
+ error->message);
+
+ current = contents;
+ while (current) {
+ gchar *next;
+
+ next = strchr (current, '\n');
+ if (next) {
+ *next = '\0';
+ next++;
+ }
+
+ g_strstrip (current);
+ if (current[0] != '\0' && current[0] != '#') {
+ gchar *response;
+
+ response = current;
+ while (*response != ' ')
+ response++;
+ g_assert (*response == ' ');
+ *response = '\0';
+ response++;
+ while (*response == ' ')
+ response++;
+ g_assert (*response != '\0');
+
+ test_port_context_set_command (self, current, response);
+ }
+ current = next;
+ }
+
+ g_free (contents);
+}
+
+static const gchar *
+process_next_command (TestPortContext *ctx,
+ GByteArray *buffer)
+{
+ gsize i = 0;
+ gchar *command;
+ const gchar *response;
+ static const gchar *error_response = "\r\nERROR\r\n";
+
+ /* Find command end */
+ while (i < buffer->len && buffer->data[i] != '\r' && buffer->data[i] != '\n')
+ i++;
+ if (i == buffer->len)
+ /* no command */
+ return NULL;
+
+ while (i < buffer->len && (buffer->data[i] == '\r' || buffer->data[i] == '\n'))
+ buffer->data[i++] = '\0';
+
+ /* Setup command and lookup response */
+ command = g_strndup ((gchar *)buffer->data, i);
+ response = g_hash_table_lookup (ctx->commands, command);
+ g_free (command);
+
+ /* Remove command from buffer */
+ g_byte_array_remove_range (buffer, 0, i);
+
+ return response ? response : error_response;
+}
+
+/*****************************************************************************/
+
+typedef struct {
+ TestPortContext *ctx;
+ GSocketConnection *connection;
+ GSource *connection_readable_source;
+ GByteArray *buffer;
+} Client;
+
+static void
+client_free (Client *client)
+{
+ g_source_destroy (client->connection_readable_source);
+ g_source_unref (client->connection_readable_source);
+ g_output_stream_close (g_io_stream_get_output_stream (G_IO_STREAM (client->connection)), NULL, NULL);
+ if (client->buffer)
+ g_byte_array_unref (client->buffer);
+ g_object_unref (client->connection);
+ g_slice_free (Client, client);
+}
+
+static void
+connection_close (Client *client)
+{
+ client->ctx->clients = g_list_remove (client->ctx->clients, client);
+ client_free (client);
+}
+
+static void
+client_parse_request (Client *client)
+{
+ const gchar *response;
+
+ do {
+ response = process_next_command (client->ctx, client->buffer);
+ if (response) {
+ GError *error = NULL;
+
+ if (!g_output_stream_write_all (g_io_stream_get_output_stream (G_IO_STREAM (client->connection)),
+ response,
+ strlen (response),
+ NULL, /* bytes_written */
+ NULL, /* cancellable */
+ &error)) {
+ g_warning ("Cannot send response to client: %s", error->message);
+ g_error_free (error);
+ }
+ }
+
+ } while (response);
+}
+
+static gboolean
+connection_readable_cb (GSocket *socket,
+ GIOCondition condition,
+ Client *client)
+{
+ guint8 buffer[BUFFER_SIZE];
+ GError *error = NULL;
+ gssize r;
+
+ if (condition & G_IO_HUP || condition & G_IO_ERR) {
+ g_debug ("client connection closed");
+ connection_close (client);
+ return FALSE;
+ }
+
+ if (!(condition & G_IO_IN || condition & G_IO_PRI))
+ return TRUE;
+
+ r = g_input_stream_read (g_io_stream_get_input_stream (G_IO_STREAM (client->connection)),
+ buffer,
+ BUFFER_SIZE,
+ NULL,
+ &error);
+
+ if (r < 0) {
+ g_warning ("Error reading from istream: %s", error ? error->message : "unknown");
+ if (error)
+ g_error_free (error);
+ /* Close the device */
+ connection_close (client);
+ return FALSE;
+ }
+
+ if (r == 0)
+ return TRUE;
+
+ /* else, r > 0 */
+ if (!G_UNLIKELY (client->buffer))
+ client->buffer = g_byte_array_sized_new (r);
+ g_byte_array_append (client->buffer, buffer, r);
+
+ /* Try to parse input messages */
+ client_parse_request (client);
+
+ return TRUE;
+}
+
+static Client *
+client_new (TestPortContext *self,
+ GSocketConnection *connection)
+{
+ Client *client;
+
+ client = g_slice_new0 (Client);
+ client->ctx = self;
+ client->connection = g_object_ref (connection);
+ client->connection_readable_source = g_socket_create_source (g_socket_connection_get_socket (client->connection),
+ G_IO_IN | G_IO_PRI | G_IO_ERR | G_IO_HUP,
+ NULL);
+ g_source_set_callback (client->connection_readable_source,
+ (GSourceFunc)connection_readable_cb,
+ client,
+ NULL);
+ g_source_attach (client->connection_readable_source, self->context);
+
+ return client;
+}
+
+/* /\*****************************************************************************\/ */
+
+static void
+incoming_cb (GSocketService *service,
+ GSocketConnection *connection,
+ GObject *unused,
+ TestPortContext *self)
+{
+ Client *client;
+
+ client = client_new (self, connection);
+ self->clients = g_list_append (self->clients, client);
+}
+
+static void
+create_socket_service (TestPortContext *self)
+{
+ GError *error = NULL;
+ GSocketService *service;
+ GSocketAddress *address;
+ GSocket *socket;
+
+ g_assert (self->socket_service == NULL);
+
+ /* Create socket */
+ socket = g_socket_new (G_SOCKET_FAMILY_UNIX,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error);
+ if (!socket)
+ g_error ("Cannot create socket: %s", error->message);
+
+ /* Bind to address */
+ address = (g_unix_socket_address_new_with_type (
+ self->name,
+ -1,
+ (g_str_has_prefix (self->name, "abstract:") ?
+ G_UNIX_SOCKET_ADDRESS_ABSTRACT :
+ G_UNIX_SOCKET_ADDRESS_PATH)));
+ if (!g_socket_bind (socket, address, TRUE, &error))
+ g_error ("Cannot bind socket: %s", error->message);
+ g_object_unref (address);
+
+ /* Listen */
+ if (!g_socket_listen (socket, &error))
+ g_error ("Cannot listen in socket: %s", error->message);
+
+ /* Create socket service */
+ service = g_socket_service_new ();
+ g_signal_connect (service, "incoming", G_CALLBACK (incoming_cb), self);
+ if (!g_socket_listener_add_socket (G_SOCKET_LISTENER (service),
+ socket,
+ NULL, /* don't pass an object, will take a reference */
+ &error))
+ g_error ("Cannot add listener to socket: %s", error->message);
+
+ /* Start it */
+ g_socket_service_start (service);
+
+ /* And store both the service and the socket.
+ * Since GLib 2.42 the socket may not be explicitly closed when the
+ * listener is diposed, so we'll do it ourselves. */
+ self->socket_service = service;
+ self->socket = socket;
+
+ /* Signal that the thread is ready */
+ g_mutex_lock (&self->ready_mutex);
+ self->ready = TRUE;
+ g_cond_signal (&self->ready_cond);
+ g_mutex_unlock (&self->ready_mutex);
+}
+
+/*****************************************************************************/
+
+static gboolean
+cancel_loop_cb (TestPortContext *self)
+{
+ g_main_loop_quit (self->loop);
+ return FALSE;
+}
+
+void
+test_port_context_stop (TestPortContext *self)
+{
+ g_assert (self->thread != NULL);
+ g_assert (self->loop != NULL);
+ g_assert (self->context != NULL);
+
+ /* Cancel main loop of the port context thread, by scheduling an idle task
+ * in the thread-owned main context */
+ g_main_context_invoke (self->context, (GSourceFunc) cancel_loop_cb, self);
+
+ g_thread_join (self->thread);
+ self->thread = NULL;
+}
+
+static gpointer
+port_context_thread_func (TestPortContext *self)
+{
+ g_assert (self->loop == NULL);
+ g_assert (self->context == NULL);
+
+ /* Define main context and loop for the thread */
+ self->context = g_main_context_new ();
+ self->loop = g_main_loop_new (self->context, FALSE);
+ g_main_context_push_thread_default (self->context);
+
+ /* Once the thread default context is setup, launch service */
+ create_socket_service (self);
+
+ g_main_loop_run (self->loop);
+
+ g_main_loop_unref (self->loop);
+ self->loop = NULL;
+ g_main_context_unref (self->context);
+ self->context = NULL;
+ return NULL;
+}
+
+void
+test_port_context_start (TestPortContext *self)
+{
+ g_assert (self->thread == NULL);
+ self->thread = g_thread_new (self->name,
+ (GThreadFunc)port_context_thread_func,
+ self);
+
+ /* Now wait until the thread has finished its initialization and is
+ * ready to serve connections */
+ g_mutex_lock (&self->ready_mutex);
+ while (!self->ready)
+ g_cond_wait (&self->ready_cond, &self->ready_mutex);
+ g_mutex_unlock (&self->ready_mutex);
+}
+
+/*****************************************************************************/
+
+void
+test_port_context_free (TestPortContext *self)
+{
+ g_assert (self->thread == NULL);
+ g_assert (self->loop == NULL);
+
+ g_cond_clear (&self->ready_cond);
+ g_mutex_clear (&self->ready_mutex);
+
+ if (self->commands)
+ g_hash_table_unref (self->commands);
+ g_list_free_full (self->clients, (GDestroyNotify)client_free);
+ if (self->socket) {
+ GError *error = NULL;
+
+ if (!g_socket_close (self->socket, &error)) {
+ g_debug ("Couldn't close socket: %s", error->message);
+ g_error_free (error);
+ }
+ g_object_unref (self->socket);
+ }
+ if (self->socket_service) {
+ if (g_socket_service_is_active (self->socket_service))
+ g_socket_service_stop (self->socket_service);
+ g_object_unref (self->socket_service);
+ }
+ g_free (self->name);
+ g_slice_free (TestPortContext, self);
+}
+
+TestPortContext *
+test_port_context_new (const gchar *name)
+{
+ TestPortContext *self;
+
+ self = g_slice_new0 (TestPortContext);
+ self->name = g_strdup (name);
+ g_cond_init (&self->ready_cond);
+ g_mutex_init (&self->ready_mutex);
+ return self;
+}