[gnome-software] Add support for snaps.
- From: Richard Hughes <rhughes src gnome org>
- To: commits-list gnome org
- Cc:
- Subject: [gnome-software] Add support for snaps.
- Date: Wed, 22 Jun 2016 09:25:49 +0000 (UTC)
commit 2c5392aebe3b850d937fa41ff3a5e5de2006861f
Author: Robert Ancell <robert ancell canonical com>
Date: Wed Jun 15 11:11:03 2016 +1200
Add support for snaps.
Work done by William Hua and Robert Ancell and tweaked by Richard Hughes.
https://bugzilla.gnome.org/show_bug.cgi?id=767673
src/plugins/Makefile.am | 14 +-
src/plugins/gs-plugin-snap.c | 544 ++++++++++++++++++++++++++++++++++++++++++
src/plugins/gs-snapd.c | 298 +++++++++++++++++++++++
src/plugins/gs-snapd.h | 42 ++++
4 files changed, 897 insertions(+), 1 deletions(-)
---
diff --git a/src/plugins/Makefile.am b/src/plugins/Makefile.am
index c551f61..c57b7d4 100644
--- a/src/plugins/Makefile.am
+++ b/src/plugins/Makefile.am
@@ -42,7 +42,8 @@ plugin_LTLIBRARIES = \
libgs_plugin_provenance-license.la \
libgs_plugin_fedora-tagger-usage.la \
libgs_plugin_epiphany.la \
- libgs_plugin_icons.la
+ libgs_plugin_icons.la \
+ libgs_plugin_snap.la
if HAVE_PACKAGEKIT
plugin_LTLIBRARIES += \
@@ -332,6 +333,17 @@ libgs_plugin_packagekit_proxy_la_LIBADD = $(GS_PLUGIN_LIBS)
libgs_plugin_packagekit_proxy_la_LDFLAGS = -module -avoid-version
libgs_plugin_packagekit_proxy_la_CFLAGS = $(GS_PLUGIN_CFLAGS) $(WARN_CFLAGS)
+libgs_plugin_snap_la_SOURCES = \
+ gs-plugin-snap.c \
+ gs-snapd.h \
+ gs-snapd.c
+libgs_plugin_snap_la_LIBADD = \
+ $(GS_PLUGIN_LIBS) \
+ $(SOUP_LIBS) \
+ $(JSON_GLIB_LIBS)
+libgs_plugin_snap_la_LDFLAGS = -module -avoid-version
+libgs_plugin_snap_la_CFLAGS = $(GS_PLUGIN_CFLAGS) $(WARN_CFLAGS)
+
if ENABLE_TESTS
check_PROGRAMS = \
gs-self-test
diff --git a/src/plugins/gs-plugin-snap.c b/src/plugins/gs-plugin-snap.c
new file mode 100644
index 0000000..bfcc204
--- /dev/null
+++ b/src/plugins/gs-plugin-snap.c
@@ -0,0 +1,544 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2015-2016 Canonical Ltd
+ *
+ * Licensed under the GNU General Public License Version 2
+ *
+ * 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.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <config.h>
+
+#include <json-glib/json-glib.h>
+#include <gnome-software.h>
+
+#include "gs-snapd.h"
+
+typedef gboolean (*AppFilterFunc)(const gchar *id, JsonObject *object, gpointer data);
+
+void
+gs_plugin_initialize (GsPlugin *plugin)
+{
+ if (!gs_snapd_exists ()) {
+ g_debug ("disabling '%s' as snapd not running",
+ gs_plugin_get_name (plugin));
+ gs_plugin_set_enabled (plugin, FALSE);
+ }
+
+ gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_RUN_AFTER, "desktop-categories");
+ gs_plugin_add_rule (plugin, GS_PLUGIN_RULE_RUN_AFTER, "ubuntu-reviews");
+}
+
+static JsonParser *
+parse_result (const gchar *response, const gchar *response_type, GError **error)
+{
+ g_autoptr(JsonParser) parser = NULL;
+ g_autoptr(GError) error_local = NULL;
+
+ if (response_type == NULL) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned no content type");
+ return NULL;
+ }
+ if (g_strcmp0 (response_type, "application/json") != 0) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned unexpected content type %s", response_type);
+ return NULL;
+ }
+
+ parser = json_parser_new ();
+ if (!json_parser_load_from_data (parser, response, -1, &error_local)) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to parse snapd response: %s",
+ error_local->message);
+ return NULL;
+ }
+ if (!JSON_NODE_HOLDS_OBJECT (json_parser_get_root (parser))) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd response does is not a valid JSON object");
+ return NULL;
+ }
+
+ return g_object_ref (parser);
+}
+
+static void
+refine_app (GsPlugin *plugin, GsApp *app, JsonObject *package)
+{
+ const gchar *status, *icon_url;
+ g_autoptr(GdkPixbuf) icon_pixbuf = NULL;
+ gint64 size = -1;
+
+ status = json_object_get_string_member (package, "status");
+ if (g_strcmp0 (status, "installed") == 0 || g_strcmp0 (status, "active") == 0) {
+ const gchar *update_available;
+
+ update_available = json_object_has_member (package, "update_available") ?
+ json_object_get_string_member (package, "update_available") : NULL;
+ if (update_available)
+ gs_app_set_state (app, AS_APP_STATE_UPDATABLE);
+ else
+ gs_app_set_state (app, AS_APP_STATE_INSTALLED);
+ } else if (g_strcmp0 (status, "not installed") == 0 || g_strcmp0 (status, "available") == 0) {
+ gs_app_set_state (app, AS_APP_STATE_AVAILABLE);
+ }
+ gs_app_set_name (app, GS_APP_QUALITY_HIGHEST,
+ json_object_get_string_member (package, "summary"));
+ gs_app_set_summary (app, GS_APP_QUALITY_HIGHEST,
+ json_object_get_string_member (package, "description"));
+ gs_app_set_version (app, json_object_get_string_member (package, "version"));
+ if (json_object_has_member (package, "installed-size")) {
+ size = json_object_get_int_member (package, "installed-size");
+ if (size > 0)
+ gs_app_set_size_installed (app, size);
+ }
+ if (json_object_has_member (package, "download-size")) {
+ size = json_object_get_int_member (package, "download-size");
+ if (size > 0)
+ gs_app_set_size_download (app, size);
+ }
+ gs_app_add_quirk (app, AS_APP_QUIRK_PROVENANCE);
+ icon_url = json_object_get_string_member (package, "icon");
+ if (g_str_has_prefix (icon_url, "/")) {
+ g_autofree gchar *icon_response = NULL;
+ gsize icon_response_length;
+
+ if (gs_snapd_request ("GET", icon_url, NULL, NULL, TRUE, NULL,
+ NULL, NULL, NULL,
+ &icon_response, &icon_response_length,
+ NULL)) {
+ g_autoptr(GdkPixbufLoader) loader = NULL;
+
+ loader = gdk_pixbuf_loader_new ();
+ gdk_pixbuf_loader_write (loader,
+ (guchar *) icon_response,
+ icon_response_length,
+ NULL);
+ gdk_pixbuf_loader_close (loader, NULL);
+ icon_pixbuf = g_object_ref (gdk_pixbuf_loader_get_pixbuf (loader));
+ }
+ else
+ g_printerr ("Failed to get icon\n");
+ }
+ else {
+ g_autoptr(SoupMessage) message = NULL;
+ g_autoptr(GdkPixbufLoader) loader = NULL;
+
+ message = soup_message_new (SOUP_METHOD_GET, icon_url);
+ if (message != NULL) {
+ soup_session_send_message (gs_plugin_get_soup_session (plugin), message);
+ loader = gdk_pixbuf_loader_new ();
+ gdk_pixbuf_loader_write (loader,
+ (guint8 *) message->response_body->data,
+ message->response_body->length,
+ NULL);
+ gdk_pixbuf_loader_close (loader, NULL);
+ icon_pixbuf = g_object_ref (gdk_pixbuf_loader_get_pixbuf (loader));
+ }
+ }
+
+ if (icon_pixbuf) {
+ gs_app_set_pixbuf (app, icon_pixbuf);
+ } else {
+ g_autoptr(AsIcon) icon = as_icon_new ();
+ as_icon_set_kind (icon, AS_ICON_KIND_STOCK);
+ as_icon_set_name (icon, "package-x-generic");
+ gs_app_add_icon (app, icon);
+ }
+}
+
+static gboolean
+get_apps (GsPlugin *plugin,
+ const gchar *sources,
+ gchar **search_terms,
+ GsAppList *list,
+ AppFilterFunc filter_func,
+ gpointer user_data,
+ GError **error)
+{
+ guint status_code;
+ GPtrArray *query_fields;
+ g_autoptr (GString) path = NULL;
+ g_autofree gchar *reason_phrase = NULL, *response_type = NULL, *response = NULL;
+ g_autoptr(JsonParser) parser = NULL;
+ JsonObject *root;
+ JsonArray *result;
+ GList *snaps;
+ GList *i;
+
+ /* Get all the apps */
+ query_fields = g_ptr_array_new_with_free_func (g_free);
+ if (sources != NULL)
+ g_ptr_array_add (query_fields, g_strdup_printf ("sources=%s", sources));
+ if (search_terms != NULL) {
+ g_autofree gchar *query = NULL;
+ query = g_strjoinv ("+", search_terms);
+ g_ptr_array_add (query_fields, g_strdup_printf ("q=%s", query));
+ }
+ g_ptr_array_add (query_fields, NULL);
+ path = g_string_new ("/v2/snaps");
+ if (query_fields->len > 1) {
+ g_autofree gchar *fields = NULL;
+ g_string_append (path, "?");
+ fields = g_strjoinv ("&", (gchar **) query_fields->pdata);
+ g_string_append (path, fields);
+ }
+ g_ptr_array_free (query_fields, TRUE);
+ if (!gs_snapd_request ("GET", path->str, NULL, NULL, TRUE, NULL,
+ &status_code, &reason_phrase,
+ &response_type, &response,
+ NULL, error))
+ return FALSE;
+
+ if (status_code != SOUP_STATUS_OK) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned status code %d: %s",
+ status_code, reason_phrase);
+ return FALSE;
+ }
+
+ parser = parse_result (response, response_type, error);
+ if (parser == NULL)
+ return FALSE;
+
+ root = json_node_get_object (json_parser_get_root (parser));
+ result = json_object_get_array_member (root, "result");
+ snaps = json_array_get_elements (result);
+
+ for (i = snaps; i != NULL; i = i->next) {
+ JsonObject *package = json_node_get_object (i->data);
+ g_autoptr(GsApp) app = NULL;
+ const gchar *id;
+
+ id = json_object_get_string_member (package, "name");
+
+ if (filter_func != NULL && !filter_func (id, package, user_data))
+ continue;
+
+ app = gs_app_new (id);
+ gs_app_set_management_plugin (app, "snap");
+ gs_app_set_kind (app, AS_APP_KIND_DESKTOP);
+ gs_app_add_quirk (app, AS_APP_QUIRK_NOT_REVIEWABLE);
+ gs_app_add_quirk (app, AS_APP_QUIRK_NOT_LAUNCHABLE);
+ refine_app (plugin, app, package);
+ gs_app_list_add (list, app);
+ }
+
+ g_list_free (snaps);
+
+ return TRUE;
+}
+
+static gboolean
+get_app (GsPlugin *plugin, GsApp *app, GError **error)
+{
+ guint status_code;
+ g_autofree gchar *path = NULL;
+ g_autofree gchar *reason_phrase = NULL;
+ g_autofree gchar *response = NULL;
+ g_autofree gchar *response_type = NULL;
+ g_autoptr(JsonParser) parser = NULL;
+ JsonObject *root;
+ JsonNode *result;
+ JsonArray *result_array;
+ JsonObject *result_object;
+ JsonNode *result_element;
+ guint i;
+
+ path = g_strdup_printf ("/v2/snaps/%s", gs_app_get_id (app));
+ if (!gs_snapd_request ("GET", path, NULL, NULL, TRUE, NULL,
+ &status_code, &reason_phrase,
+ &response_type, &response,
+ NULL, error))
+ return FALSE;
+
+ if (status_code == SOUP_STATUS_NOT_FOUND) {
+ g_clear_pointer (&path, g_free);
+ g_clear_pointer (&reason_phrase, g_free);
+ g_clear_pointer (&response_type, g_free);
+ g_clear_pointer (&response, g_free);
+
+ path = g_strdup_printf ("/v2/find?q=%s", gs_app_get_id (app));
+ if (!gs_snapd_request ("GET", path, NULL, NULL, TRUE, NULL,
+ &status_code, &reason_phrase,
+ &response_type, &response,
+ NULL, error))
+ return FALSE;
+ }
+
+ if (status_code != SOUP_STATUS_OK) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned status code %d: %s",
+ status_code, reason_phrase);
+ return FALSE;
+ }
+
+ parser = parse_result (response, response_type, error);
+ if (parser == NULL)
+ return FALSE;
+ root = json_node_get_object (json_parser_get_root (parser));
+ result = json_object_get_member (root, "result");
+ if (result == NULL) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned no results for %s", gs_app_get_id (app));
+ return FALSE;
+ }
+
+ if (JSON_NODE_HOLDS_ARRAY (result)) {
+ result_array = json_node_get_array (result);
+ for (i = 0; i < json_array_get_length (result_array); i++) {
+ result_element = json_array_get_element (result_array, i);
+ if (!JSON_NODE_HOLDS_OBJECT (result_element))
+ continue;
+ result_object = json_node_get_object (result_element);
+ if (g_strcmp0 (json_object_get_string_member (result_object, "name"),
+ gs_app_get_id (app)) == 0) {
+ refine_app (plugin, app, result_object);
+ break;
+ }
+ }
+ } else if (JSON_NODE_HOLDS_OBJECT (result)) {
+ result_object = json_node_get_object (result);
+ refine_app (plugin, app, result_object);
+ }
+
+ return TRUE;
+}
+
+void
+gs_plugin_destroy (GsPlugin *plugin)
+{
+}
+
+static gboolean
+is_active (const gchar *id, JsonObject *object, gpointer data)
+{
+ const gchar *status = json_object_get_string_member (object, "status");
+ return g_strcmp0 (status, "active") == 0;
+}
+
+gboolean
+gs_plugin_add_installed (GsPlugin *plugin,
+ GsAppList *list,
+ GCancellable *cancellable,
+ GError **error)
+{
+ return get_apps (plugin, "local", NULL, list, is_active, NULL, error);
+}
+
+gboolean
+gs_plugin_add_search (GsPlugin *plugin,
+ gchar **values,
+ GsAppList *list,
+ GCancellable *cancellable,
+ GError **error)
+{
+ return get_apps (plugin, NULL, values, list, NULL, values, error);
+}
+
+gboolean
+gs_plugin_refine_app (GsPlugin *plugin,
+ GsApp *app,
+ GsPluginRefineFlags flags,
+ GCancellable *cancellable,
+ GError **error)
+{
+ /* not us */
+ if (g_strcmp0 (gs_app_get_management_plugin (app), "snap") != 0)
+ return TRUE;
+
+ // Get info from snapd
+ return get_app (plugin, app, error);
+}
+
+static gboolean
+send_package_action (GsPlugin *plugin,
+ GsApp *app,
+ const gchar *id,
+ const gchar *action,
+ GError **error)
+{
+ g_autofree gchar *content = NULL, *path = NULL;
+ guint status_code;
+ g_autofree gchar *reason_phrase = NULL;
+ g_autofree gchar *response_type = NULL;
+ g_autofree gchar *response = NULL;
+ g_autofree gchar *status = NULL;
+ g_autoptr(JsonParser) parser = NULL;
+ JsonObject *root, *result, *task, *progress;
+ JsonArray *tasks;
+ GList *task_list, *l;
+ gint64 done, total, task_done, task_total;
+ const gchar *resource_path;
+ const gchar *type;
+ const gchar *change_id;
+ g_autoptr(GVariant) macaroon = NULL;
+
+ content = g_strdup_printf ("{\"action\": \"%s\"}", action);
+ path = g_strdup_printf ("/v2/snaps/%s", id);
+ if (!gs_snapd_request ("POST", path, content, NULL, TRUE,
+ &macaroon, &status_code,
+ &reason_phrase, &response_type,
+ &response, NULL, error))
+ return FALSE;
+
+ if (status_code != SOUP_STATUS_ACCEPTED) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned status code %d: %s",
+ status_code, reason_phrase);
+ return FALSE;
+ }
+
+ parser = parse_result (response, response_type, error);
+ if (parser == NULL)
+ return FALSE;
+
+ root = json_node_get_object (json_parser_get_root (parser));
+ type = json_object_get_string_member (root, "type");
+
+ if (g_strcmp0 (type, "async") == 0) {
+ change_id = json_object_get_string_member (root, "change");
+ resource_path = g_strdup_printf ("/v2/changes/%s", change_id);
+
+ while (TRUE) {
+ g_autofree gchar *status_reason_phrase = NULL;
+ g_autofree gchar *status_response_type = NULL;
+ g_autofree gchar *status_response = NULL;
+ g_autoptr(JsonParser) status_parser = NULL;
+
+ /* Wait for a little bit before polling */
+ g_usleep (100 * 1000);
+
+ if (!gs_snapd_request ("GET", resource_path, NULL, macaroon, TRUE, NULL,
+ &status_code, &status_reason_phrase,
+ &status_response_type, &status_response,
+ NULL, error)) {
+ return FALSE;
+ }
+
+ if (status_code != SOUP_STATUS_OK) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd returned status code %d: %s",
+ status_code, status_reason_phrase);
+ return FALSE;
+ }
+
+ status_parser = parse_result (status_response, status_response_type, error);
+ if (status_parser == NULL)
+ return FALSE;
+
+ root = json_node_get_object (json_parser_get_root (status_parser));
+ result = json_object_get_object_member (root, "result");
+
+ g_free (status);
+ status = g_strdup (json_object_get_string_member (result, "status"));
+
+ if (g_strcmp0 (status, "Done") == 0)
+ break;
+
+ tasks = json_object_get_array_member (result, "tasks");
+ task_list = json_array_get_elements (tasks);
+
+ done = 0;
+ total = 0;
+
+ for (l = task_list; l != NULL; l = l->next) {
+ task = json_node_get_object (l->data);
+ progress = json_object_get_object_member (task, "progress");
+ task_done = json_object_get_int_member (progress, "done");
+ task_total = json_object_get_int_member (progress, "total");
+
+ done += task_done;
+ total += task_total;
+ }
+
+ gs_app_set_progress (app, 100 * done / total);
+
+ g_list_free (task_list);
+ }
+ }
+
+ if (g_strcmp0 (status, "Done") != 0) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd operation finished with status %s", status);
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+gboolean
+gs_plugin_app_install (GsPlugin *plugin,
+ GsApp *app,
+ GCancellable *cancellable,
+ GError **error)
+{
+ gboolean ret;
+
+ /* We can only install apps we know of */
+ if (g_strcmp0 (gs_app_get_management_plugin (app), "snap") != 0)
+ return TRUE;
+
+ gs_app_set_state (app, AS_APP_STATE_INSTALLING);
+ ret = send_package_action (plugin, app, gs_app_get_id (app), "install", error);
+ if (!ret) {
+ gs_app_set_state_recover (app);
+ return FALSE;
+ }
+ gs_app_set_state (app, AS_APP_STATE_INSTALLED);
+ return TRUE;
+}
+
+gboolean
+gs_plugin_app_remove (GsPlugin *plugin,
+ GsApp *app,
+ GCancellable *cancellable,
+ GError **error)
+{
+ gboolean ret;
+
+ /* We can only remove apps we know of */
+ if (g_strcmp0 (gs_app_get_management_plugin (app), "snap") != 0)
+ return TRUE;
+
+ gs_app_set_state (app, AS_APP_STATE_REMOVING);
+ ret = send_package_action (plugin, app, gs_app_get_id (app), "remove", error);
+ if (!ret) {
+ gs_app_set_state_recover (app);
+ return FALSE;
+ }
+ gs_app_set_state (app, AS_APP_STATE_AVAILABLE);
+ return TRUE;
+}
diff --git a/src/plugins/gs-snapd.c b/src/plugins/gs-snapd.c
new file mode 100644
index 0000000..c11b1ce
--- /dev/null
+++ b/src/plugins/gs-snapd.c
@@ -0,0 +1,298 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2016 Canonical Ltd
+ *
+ * Licensed under the GNU General Public License Version 2
+ *
+ * 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.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <stdlib.h>
+#include <string.h>
+#include <gs-plugin.h>
+#include <libsoup/soup.h>
+#include <gio/gunixsocketaddress.h>
+
+#include "gs-snapd.h"
+
+// snapd API documentation is at https://github.com/snapcore/snapd/blob/master/docs/rest.md
+
+#define SNAPD_SOCKET "/run/snapd.socket"
+
+gboolean
+gs_snapd_exists (void)
+{
+ return g_file_test (SNAPD_SOCKET, G_FILE_TEST_EXISTS);
+}
+
+static GSocket *
+open_snapd_socket (GError **error)
+{
+ GSocket *socket;
+ g_autoptr(GSocketAddress) address = NULL;
+ g_autoptr(GError) error_local = NULL;
+
+ socket = g_socket_new (G_SOCKET_FAMILY_UNIX,
+ G_SOCKET_TYPE_STREAM,
+ G_SOCKET_PROTOCOL_DEFAULT,
+ &error_local);
+ if (!socket) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to open snapd socket: %s",
+ error_local->message);
+ return NULL;
+ }
+ address = g_unix_socket_address_new (SNAPD_SOCKET);
+ if (!g_socket_connect (socket, address, NULL, &error_local)) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to connect snapd socket: %s",
+ error_local->message);
+ g_object_unref (socket);
+ return NULL;
+ }
+
+ return socket;
+}
+
+static gboolean
+read_from_snapd (GSocket *socket,
+ gchar *buffer, gsize buffer_length,
+ gsize *read_offset,
+ GError **error)
+{
+ gssize n_read;
+ n_read = g_socket_receive (socket,
+ buffer + *read_offset,
+ buffer_length - *read_offset,
+ NULL,
+ error);
+ if (n_read < 0)
+ return FALSE;
+ *read_offset += n_read;
+ buffer[*read_offset] = '\0';
+
+ return TRUE;
+}
+
+gboolean
+gs_snapd_request (const gchar *method,
+ const gchar *path,
+ const gchar *content,
+ GVariant *macaroon,
+ gboolean retry_after_login,
+ GVariant **out_macaroon,
+ guint *status_code,
+ gchar **reason_phrase,
+ gchar **response_type,
+ gchar **response,
+ gsize *response_length,
+ GError **error)
+{
+ g_autoptr (GSocket) socket = NULL;
+ g_autoptr (GString) request = NULL;
+ gssize n_written;
+ gsize max_data_length = 65535, data_length = 0, header_length;
+ gchar data[max_data_length + 1], *body = NULL;
+ g_autoptr (SoupMessageHeaders) headers = NULL;
+ gsize chunk_length, n_required;
+ gchar *chunk_start = NULL;
+ const gchar *root;
+ const gchar *discharge;
+ GVariantIter *iter;
+ guint code;
+ gboolean ret;
+
+ // NOTE: Would love to use libsoup but it doesn't support unix sockets
+ // https://bugzilla.gnome.org/show_bug.cgi?id=727563
+
+ socket = open_snapd_socket (error);
+ if (socket == NULL)
+ return FALSE;
+
+ request = g_string_new ("");
+ g_string_append_printf (request, "%s %s HTTP/1.1\r\n", method, path);
+ g_string_append (request, "Host:\r\n");
+ if (macaroon != NULL) {
+ g_variant_get (macaroon, "(&sas)", &root, &iter);
+ g_string_append_printf (request, "Authorization: Macaroon root=\"%s\"", root);
+
+ while (g_variant_iter_next (iter, "&s", &discharge))
+ g_string_append_printf (request, ",discharge=\"%s\"", discharge);
+
+ g_variant_iter_free (iter);
+ g_string_append (request, "\r\n");
+ }
+ if (content)
+ g_string_append_printf (request, "Content-Length: %zi\r\n", strlen (content));
+ g_string_append (request, "\r\n");
+ if (content)
+ g_string_append (request, content);
+
+ g_debug ("begin snapd request: %s", request->str);
+
+ /* send HTTP request */
+ n_written = g_socket_send (socket, request->str, request->len, NULL, error);
+ if (n_written < 0)
+ return FALSE;
+
+ /* read HTTP headers */
+ while (data_length < max_data_length && !body) {
+ if (!read_from_snapd (socket,
+ data,
+ max_data_length,
+ &data_length,
+ error))
+ return FALSE;
+ body = strstr (data, "\r\n\r\n");
+ }
+ if (!body) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to find header separator in snapd response");
+ return FALSE;
+ }
+
+ /* body starts after header divider */
+ body += 4;
+ header_length = body - data;
+
+ /* parse headers */
+ headers = soup_message_headers_new (SOUP_MESSAGE_HEADERS_RESPONSE);
+ if (!soup_headers_parse_response (data, header_length, headers,
+ NULL, &code, reason_phrase)) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "snapd response HTTP headers not parseable");
+ return FALSE;
+ }
+
+ if (status_code != NULL)
+ *status_code = code;
+
+ if ((code == 401 || code == 403) && retry_after_login) {
+ g_socket_close (socket, NULL);
+
+ if (macaroon == NULL) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_AUTH_REQUIRED,
+ "failed to authenticate");
+ return FALSE;
+ }
+
+ ret = gs_snapd_request (method,
+ path,
+ content,
+ macaroon,
+ FALSE,
+ NULL,
+ status_code,
+ reason_phrase,
+ response_type,
+ response,
+ response_length,
+ error);
+
+ if (ret && out_macaroon != NULL) {
+ *out_macaroon = macaroon;
+ } else {
+ g_variant_unref (macaroon);
+ }
+
+ return ret;
+ }
+
+ /* work out how much data to follow */
+ if (g_strcmp0 (soup_message_headers_get_one (headers, "Transfer-Encoding"),
+ "chunked") == 0) {
+ while (data_length < max_data_length) {
+ chunk_start = strstr (body, "\r\n");
+ if (chunk_start)
+ break;
+ if (!read_from_snapd (socket,
+ data,
+ max_data_length,
+ &data_length,
+ error))
+ return FALSE;
+ }
+ if (!chunk_start) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to find chunk header in "
+ "snapd response");
+ return FALSE;
+ }
+ chunk_length = strtoul (body, NULL, 16);
+ chunk_start += 2;
+ // FIXME: support multiple chunks
+ }
+ else {
+ const gchar *value;
+ value = soup_message_headers_get_one (headers, "Content-Length");
+ if (!value) {
+ g_set_error_literal (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Unable to determine content "
+ "length of snapd response");
+ return FALSE;
+ }
+ chunk_length = strtoul (value, NULL, 10);
+ chunk_start = body;
+ }
+
+ /* check if enough space to read chunk */
+ n_required = (chunk_start - data) + chunk_length;
+ if (n_required > max_data_length) {
+ g_set_error (error,
+ GS_PLUGIN_ERROR,
+ GS_PLUGIN_ERROR_FAILED,
+ "Not enough space for snapd response, "
+ "require %zi octets, have %zi",
+ n_required, max_data_length);
+ return FALSE;
+ }
+
+ /* read chunk content */
+ while (data_length < n_required)
+ if (!read_from_snapd (socket, data,
+ n_required - data_length,
+ &data_length,
+ error))
+ return FALSE;
+
+ if (out_macaroon != NULL)
+ *out_macaroon = g_variant_ref (macaroon);
+ if (response_type)
+ *response_type = g_strdup (soup_message_headers_get_one (headers, "Content-Type"));
+ if (response) {
+ *response = g_malloc (chunk_length + 2);
+ memcpy (*response, chunk_start, chunk_length + 1);
+ (*response)[chunk_length + 1] = '\0';
+ g_debug ("snapd status %u: %s", code, *response);
+ }
+ if (response_length)
+ *response_length = chunk_length;
+
+ return TRUE;
+}
diff --git a/src/plugins/gs-snapd.h b/src/plugins/gs-snapd.h
new file mode 100644
index 0000000..58ac6fe
--- /dev/null
+++ b/src/plugins/gs-snapd.h
@@ -0,0 +1,42 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2016 Canonical Ltd
+ *
+ * Licensed under the GNU General Public License Version 2
+ *
+ * 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.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#ifndef __GS_SNAPD_H__
+#define __GS_SNAPD_H__
+
+#include <gio/gio.h>
+
+gboolean gs_snapd_exists (void);
+
+gboolean gs_snapd_request (const gchar *method,
+ const gchar *path,
+ const gchar *content,
+ GVariant *macaroon,
+ gboolean retry_after_login,
+ GVariant **out_macaroon,
+ guint *status_code,
+ gchar **reason_phrase,
+ gchar **response_type,
+ gchar **response,
+ gsize *response_length,
+ GError **error);
+
+#endif /* __GS_SNAPD_H__ */
[
Date Prev][
Date Next] [
Thread Prev][
Thread Next]
[
Thread Index]
[
Date Index]
[
Author Index]
