fwupd/src/fu-main.c
Richard Hughes 63a407ab34 Change the DBus method for installing firmware to 'Install'
It seems a little odd to call it 'Update' when it's being used for downgrading
and reinstalling as well.

As we're making things simpler, just use a single 'install' action in fwupdmgr
rather than 'install', 'update-online', 'update-offline'. We can use the flags
and fallbacks to do the right thing in all cases, and make the typical case
(installing a local file to any matching hardware) simple.

Fixes half of https://github.com/hughsie/fwupd/pull/23
2015-07-22 10:36:00 +01:00

1522 lines
41 KiB
C

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
*
* Copyright (C) 2015 Richard Hughes <richard@hughsie.com>
*
* 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 <fwupd.h>
#include <appstream-glib.h>
#include <glib/gstdio.h>
#include <gio/gio.h>
#include <gio/gunixfdlist.h>
#include <gio/gunixinputstream.h>
#include <glib/gi18n.h>
#include <locale.h>
#include <polkit/polkit.h>
#include <stdlib.h>
#include <fcntl.h>
#include "fu-cab.h"
#include "fu-cleanup.h"
#include "fu-debug.h"
#include "fu-device.h"
#include "fu-keyring.h"
#include "fu-pending.h"
#include "fu-provider.h"
#include "fu-provider-udev.h"
#include "fu-provider-usb.h"
#include "fu-resources.h"
#ifdef HAVE_COLORHUG
#include "fu-provider-chug.h"
#endif
#ifdef HAVE_UEFI
#include "fu-provider-uefi.h"
#endif
typedef struct {
GDBusConnection *connection;
GDBusNodeInfo *introspection_daemon;
GDBusProxy *proxy_uid;
GMainLoop *loop;
GPtrArray *devices;
GPtrArray *providers;
PolkitAuthority *authority;
FwupdStatus status;
FuPending *pending;
AsStore *store;
} FuMainPrivate;
typedef struct {
FuDevice *device;
FuProvider *provider;
} FuDeviceItem;
/**
* fu_main_emit_changed:
**/
static void
fu_main_emit_changed (FuMainPrivate *priv)
{
/* not yet connected */
if (priv->connection == NULL)
return;
g_dbus_connection_emit_signal (priv->connection,
NULL,
FWUPD_DBUS_PATH,
FWUPD_DBUS_INTERFACE,
"Changed",
NULL, NULL);
}
/**
* fu_main_emit_property_changed:
**/
static void
fu_main_emit_property_changed (FuMainPrivate *priv,
const gchar *property_name,
GVariant *property_value)
{
GVariantBuilder builder;
GVariantBuilder invalidated_builder;
/* not yet connected */
if (priv->connection == NULL)
return;
/* build the dict */
g_variant_builder_init (&invalidated_builder, G_VARIANT_TYPE ("as"));
g_variant_builder_init (&builder, G_VARIANT_TYPE_ARRAY);
g_variant_builder_add (&builder,
"{sv}",
property_name,
property_value);
g_dbus_connection_emit_signal (priv->connection,
NULL,
FWUPD_DBUS_PATH,
"org.freedesktop.DBus.Properties",
"PropertiesChanged",
g_variant_new ("(sa{sv}as)",
FWUPD_DBUS_INTERFACE,
&builder,
&invalidated_builder),
NULL);
g_variant_builder_clear (&builder);
g_variant_builder_clear (&invalidated_builder);
}
/**
* fu_main_set_status:
**/
static void
fu_main_set_status (FuMainPrivate *priv, FwupdStatus status)
{
if (priv->status == status)
return;
priv->status = status;
/* emit changed */
g_debug ("Emitting PropertyChanged('Status'='%s')",
fwupd_status_to_string (status));
fu_main_emit_property_changed (priv, "Status", g_variant_new_uint32 (status));
}
/**
* fu_main_device_array_to_variant:
**/
static GVariant *
fu_main_device_array_to_variant (GPtrArray *devices, GError **error)
{
GVariantBuilder builder;
guint i;
/* no devices */
if (devices->len == 0) {
g_set_error_literal (error,
FWUPD_ERROR,
FWUPD_ERROR_NOTHING_TO_DO,
"no devices");
return NULL;
}
g_variant_builder_init (&builder, G_VARIANT_TYPE_ARRAY);
for (i = 0; i < devices->len; i++) {
GVariant *tmp;
FuDeviceItem *item;
item = g_ptr_array_index (devices, i);
tmp = fu_device_to_variant (item->device);
g_variant_builder_add_value (&builder, tmp);
}
return g_variant_new ("(a{sa{sv}})", &builder);
}
/**
* fu_main_item_free:
**/
static void
fu_main_item_free (FuDeviceItem *item)
{
g_object_unref (item->device);
g_object_unref (item->provider);
g_free (item);
}
/**
* fu_main_get_item_by_id:
**/
static FuDeviceItem *
fu_main_get_item_by_id (FuMainPrivate *priv, const gchar *id)
{
FuDeviceItem *item;
guint i;
for (i = 0; i < priv->devices->len; i++) {
item = g_ptr_array_index (priv->devices, i);
if (g_strcmp0 (fu_device_get_id (item->device), id) == 0)
return item;
}
return NULL;
}
/**
* fu_main_get_item_by_guid:
**/
static FuDeviceItem *
fu_main_get_item_by_guid (FuMainPrivate *priv, const gchar *guid)
{
FuDeviceItem *item;
guint i;
for (i = 0; i < priv->devices->len; i++) {
item = g_ptr_array_index (priv->devices, i);
if (g_strcmp0 (fu_device_get_guid (item->device), guid) == 0)
return item;
}
return NULL;
}
/**
* fu_main_get_provider_by_name:
**/
static FuProvider *
fu_main_get_provider_by_name (FuMainPrivate *priv, const gchar *name)
{
FuProvider *provider;
guint i;
for (i = 0; i < priv->providers->len; i++) {
provider = g_ptr_array_index (priv->providers, i);
if (g_strcmp0 (fu_provider_get_name (provider), name) == 0)
return provider;
}
return NULL;
}
typedef struct {
GDBusMethodInvocation *invocation;
FuCab *cab;
FuDevice *device;
FuProviderFlags flags;
gchar *id;
gint firmware_fd;
gint cab_fd;
gint vercmp;
FuMainPrivate *priv;
} FuMainAuthHelper;
/**
* fu_main_helper_free:
**/
static void
fu_main_helper_free (FuMainAuthHelper *helper)
{
/* delete temp files */
fu_cab_delete_temp_files (helper->cab, NULL);
g_object_unref (helper->cab);
/* close any open files */
if (helper->cab_fd > 0)
close (helper->cab_fd);
if (helper->firmware_fd > 0)
close (helper->firmware_fd);
/* free */
g_free (helper->id);
if (helper->device != NULL)
g_object_unref (helper->device);
g_object_unref (helper->invocation);
g_free (helper);
}
/**
* fu_main_provider_update_authenticated:
**/
static gboolean
fu_main_provider_update_authenticated (FuMainAuthHelper *helper, GError **error)
{
FuDeviceItem *item;
/* check the device still exists */
item = fu_main_get_item_by_id (helper->priv, fu_device_get_id (helper->device));
if (item == NULL) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_INVALID_FILE,
"device %s was removed",
fu_device_get_id (helper->device));
return FALSE;
}
/* run the correct provider that added this */
return fu_provider_update (item->provider,
item->device,
fu_cab_get_stream (helper->cab),
helper->firmware_fd,
helper->flags,
error);
}
/**
* fu_main_check_authorization_cb:
**/
static void
fu_main_check_authorization_cb (GObject *source, GAsyncResult *res, gpointer user_data)
{
FuMainAuthHelper *helper = (FuMainAuthHelper *) user_data;
_cleanup_error_free_ GError *error = NULL;
_cleanup_object_unref_ PolkitAuthorizationResult *auth = NULL;
/* get result */
auth = polkit_authority_check_authorization_finish (POLKIT_AUTHORITY (source),
res, &error);
if (auth == NULL) {
g_dbus_method_invocation_return_error (helper->invocation,
FWUPD_ERROR,
FWUPD_ERROR_AUTH_FAILED,
"could not check for auth: %s",
error->message);
fu_main_helper_free (helper);
return;
}
/* did not auth */
if (!polkit_authorization_result_get_is_authorized (auth)) {
g_dbus_method_invocation_return_error (helper->invocation,
FWUPD_ERROR,
FWUPD_ERROR_AUTH_FAILED,
"failed to obtain auth");
fu_main_helper_free (helper);
return;
}
/* we're good to go */
if (!fu_main_provider_update_authenticated (helper, &error)) {
g_dbus_method_invocation_return_gerror (helper->invocation, error);
fu_main_helper_free (helper);
return;
}
/* success */
g_dbus_method_invocation_return_value (helper->invocation, NULL);
fu_main_helper_free (helper);
}
/**
* fu_main_update_helper:
**/
static gboolean
fu_main_update_helper (FuMainAuthHelper *helper, GError **error)
{
const gchar *guid;
const gchar *tmp;
const gchar *version;
/* load cab file */
fu_main_set_status (helper->priv, FWUPD_STATUS_LOADING);
if (!fu_cab_load_fd (helper->cab, helper->cab_fd, NULL, error))
return FALSE;
/* are we matching *any* hardware */
guid = fu_cab_get_guid (helper->cab);
if (helper->device == NULL) {
FuDeviceItem *item;
item = fu_main_get_item_by_guid (helper->priv, guid);
if (item == NULL) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_INVALID_FILE,
"no hardware matched %s",
guid);
return FALSE;
}
helper->device = g_object_ref (item->device);
}
tmp = fu_device_get_guid (helper->device);
if (g_strcmp0 (guid, tmp) != 0) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_INVALID_FILE,
"firmware is not for this hw: required %s got %s",
tmp, guid);
return FALSE;
}
/* parse the DriverVer */
version = fu_cab_get_version (helper->cab);
fu_device_set_metadata (helper->device, FU_DEVICE_KEY_VERSION_NEW, version);
/* compare to the lowest supported version, if it exists */
tmp = fu_device_get_metadata (helper->device, FU_DEVICE_KEY_VERSION_LOWEST);
if (tmp != NULL && as_utils_vercmp (tmp, version) > 0) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_VERSION_NEWER,
"Specified firmware is older than the minimum "
"required version '%s < %s'", tmp, version);
return FALSE;
}
/* compare the versions of what we have installed */
tmp = fu_device_get_metadata (helper->device, FU_DEVICE_KEY_VERSION);
if (tmp == NULL) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_INTERNAL,
"Device %s does not yet have a current version",
fu_device_get_id (helper->device));
return FALSE;
}
helper->vercmp = as_utils_vercmp (tmp, version);
if (helper->vercmp == 0 && (helper->flags & FU_PROVIDER_UPDATE_FLAG_ALLOW_REINSTALL) == 0) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_VERSION_SAME,
"Specified firmware is already installed '%s'",
tmp);
return FALSE;
}
if (helper->vercmp > 0 && (helper->flags & FU_PROVIDER_UPDATE_FLAG_ALLOW_OLDER) == 0) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_VERSION_NEWER,
"Specified firmware is older than installed '%s < %s'",
tmp, version);
return FALSE;
}
/* now extract the firmware and set any trust flags */
fu_main_set_status (helper->priv, FWUPD_STATUS_DECOMPRESSING);
if (!fu_cab_extract (helper->cab, FU_CAB_EXTRACT_FLAG_FIRMWARE |
FU_CAB_EXTRACT_FLAG_SIGNATURE, error))
return FALSE;
if (!fu_cab_verify (helper->cab, error))
return FALSE;
/* and open it */
helper->firmware_fd = g_open (fu_cab_get_filename_firmware (helper->cab),
O_CLOEXEC, 0);
if (helper->firmware_fd < 0) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_READ,
"failed to open %s",
fu_cab_get_filename_firmware (helper->cab));
return FALSE;
}
return TRUE;
}
/**
* fu_main_dbus_get_uid:
*
* Return value: the UID, or %G_MAXUINT if it could not be obtained
**/
static guint
fu_main_dbus_get_uid (FuMainPrivate *priv, const gchar *sender)
{
guint uid;
_cleanup_error_free_ GError *error = NULL;
_cleanup_variant_unref_ GVariant *value = NULL;
if (priv->proxy_uid == NULL)
return G_MAXUINT;
value = g_dbus_proxy_call_sync (priv->proxy_uid,
"GetConnectionUnixUser",
g_variant_new ("(s)", sender),
G_DBUS_CALL_FLAGS_NONE,
-1,
NULL,
&error);
if (value == NULL) {
g_warning ("Failed to get uid for %s: %s",
sender, error->message);
return G_MAXUINT;
}
g_variant_get (value, "(u)", &uid);
return uid;
}
/**
* fu_main_get_item_by_id_fallback_pending:
**/
static FuDeviceItem *
fu_main_get_item_by_id_fallback_pending (FuMainPrivate *priv, const gchar *id, GError **error)
{
FuDevice *dev;
FuProvider *provider;
FuDeviceItem *item = NULL;
const gchar *tmp;
guint i;
_cleanup_ptrarray_unref_ GPtrArray *devices = NULL;
/* not a wildcard */
if (g_strcmp0 (id, FWUPD_DEVICE_ID_ANY) != 0) {
item = fu_main_get_item_by_id (priv, id);
if (item == NULL) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_NOT_FOUND,
"no suitable device found for %s", id);
}
return item;
}
/* allow '*' for any */
devices = fu_pending_get_devices (priv->pending, error);
if (devices == NULL)
return NULL;
for (i = 0; i < devices->len; i++) {
dev = g_ptr_array_index (devices, i);
tmp = fu_device_get_metadata (dev, FU_DEVICE_KEY_PENDING_STATE);
if (tmp == NULL)
continue;
if (g_strcmp0 (tmp, "scheduled") == 0)
continue;
/* if the device is not still connected, fake a FuDeviceItem */
item = fu_main_get_item_by_id (priv, fu_device_get_id (dev));
if (item == NULL) {
tmp = fu_device_get_metadata (dev, FU_DEVICE_KEY_PROVIDER);
provider = fu_main_get_provider_by_name (priv, tmp);
if (provider == NULL) {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_NOT_FOUND,
"no provider %s found", tmp);
}
item = g_new0 (FuDeviceItem, 1);
item->device = g_object_ref (dev);
item->provider = g_object_ref (provider);
g_ptr_array_add (priv->devices, item);
/* FIXME: just a boolean on FuDeviceItem? */
fu_device_set_metadata (dev, "FakeDevice", "TRUE");
}
break;
}
/* no device found */
if (item == NULL) {
g_set_error_literal (error,
FWUPD_ERROR,
FWUPD_ERROR_NOT_FOUND,
"no suitable devices found");
}
return item;
}
/**
* fu_main_get_action_id_for_device:
**/
static const gchar *
fu_main_get_action_id_for_device (FuMainAuthHelper *helper)
{
gboolean is_trusted;
gboolean is_downgrade;
/* only test the payload */
is_trusted = (fu_cab_get_trust_flags (helper->cab) & FWUPD_TRUST_FLAG_PAYLOAD) > 0;
is_downgrade = helper->vercmp > 0;
/* relax authentication checks for removable devices */
if ((fu_device_get_flags (helper->device) & FU_DEVICE_FLAG_INTERNAL) == 0) {
if (is_downgrade)
return "org.freedesktop.fwupd.downgrade-hotplug";
if (is_trusted)
return "org.freedesktop.fwupd.update-hotplug-trusted";
return "org.freedesktop.fwupd.update-hotplug";
}
/* internal device */
if (is_downgrade)
return "org.freedesktop.fwupd.downgrade-internal";
if (is_trusted)
return "org.freedesktop.fwupd.update-internal-trusted";
return "org.freedesktop.fwupd.update-internal";
}
/**
* _as_store_set_priority:
**/
static void
_as_store_set_priority (AsStore *store, gint priority)
{
AsApp *app;
GPtrArray *apps;
guint i;
apps = as_store_get_apps (store);
for (i = 0; i < apps->len; i++) {
app = g_ptr_array_index (apps, i);
as_app_set_priority (app, priority);
}
}
/**
* fu_main_daemon_update_metadata:
*
* Supports optionally GZipped AppStream files up to 1MiB in size.
**/
static gboolean
fu_main_daemon_update_metadata (gint fd, gint fd_sig, GError **error)
{
guint8 magic[2];
_cleanup_bytes_unref_ GBytes *bytes = NULL;
_cleanup_bytes_unref_ GBytes *bytes_raw = NULL;
_cleanup_bytes_unref_ GBytes *bytes_sig = NULL;
_cleanup_object_unref_ AsStore *store = NULL;
_cleanup_object_unref_ FuKeyring *kr = NULL;
_cleanup_object_unref_ GConverter *converter = NULL;
_cleanup_object_unref_ GFile *file = NULL;
_cleanup_object_unref_ GInputStream *stream_buf = NULL;
_cleanup_object_unref_ GInputStream *stream_fd = NULL;
_cleanup_object_unref_ GInputStream *stream = NULL;
_cleanup_object_unref_ GInputStream *stream_sig = NULL;
/* open existing file if it exists */
store = as_store_new ();
file = g_file_new_for_path ("/var/cache/app-info/xmls/fwupd.xml");
if (g_file_query_exists (file, NULL)) {
if (!as_store_from_file (store, file, NULL, NULL, error))
return FALSE;
/* ensure we don't merge existing entries */
_as_store_set_priority (store, -1);
}
/* read the entire file into memory */
stream_fd = g_unix_input_stream_new (fd, TRUE);
bytes_raw = g_input_stream_read_bytes (stream_fd, 0x100000, NULL, error);
if (bytes_raw == NULL)
return FALSE;
stream_buf = g_memory_input_stream_new ();
g_memory_input_stream_add_bytes (G_MEMORY_INPUT_STREAM (stream_buf), bytes_raw);
/* peek the file type and get data */
if (g_input_stream_read (stream_buf, magic, 2, NULL, error) == -1)
return FALSE;
if (magic[0] == 0x1f && magic[1] == 0x8b) {
g_debug ("using GZip decompressor for data");
if (!g_seekable_seek (G_SEEKABLE (stream_buf), 0, G_SEEK_SET, NULL, error))
return FALSE;
converter = G_CONVERTER (g_zlib_decompressor_new (G_ZLIB_COMPRESSOR_FORMAT_GZIP));
stream = g_converter_input_stream_new (stream_buf, converter);
bytes = g_input_stream_read_bytes (stream, 0x100000, NULL, error);
if (bytes == NULL)
return FALSE;
} else if (magic[0] == '<' && magic[1] == '?') {
g_debug ("using no decompressor for data");
bytes = g_bytes_ref (bytes_raw);
} else {
g_set_error (error,
FWUPD_ERROR,
FWUPD_ERROR_INVALID_FILE,
"file type '0x%02x,0x%02x' not supported",
magic[0], magic[1]);
return FALSE;
}
/* read signature */
stream_sig = g_unix_input_stream_new (fd_sig, TRUE);
bytes_sig = g_input_stream_read_bytes (stream_sig, 0x800, NULL, error);
if (bytes_sig == NULL)
return FALSE;
/* verify file */
kr = fu_keyring_new ();
if (!fu_keyring_add_public_keys (kr, "/etc/pki/fwupd-metadata", error))
return FALSE;
if (!fu_keyring_verify_data (kr, bytes_raw, bytes_sig, error))
return FALSE;
/* merge in the new contents */
g_debug ("Store was %i size", as_store_get_size (store));
if (!as_store_from_xml (store,
g_bytes_get_data (bytes, NULL), -1,
NULL, error))
return FALSE;
g_debug ("Store now %i size", as_store_get_size (store));
/* save the new file */
as_store_set_api_version (store, 0.9);
if (!as_store_to_file (store, file,
AS_NODE_TO_XML_FLAG_ADD_HEADER |
AS_NODE_TO_XML_FLAG_FORMAT_MULTILINE |
AS_NODE_TO_XML_FLAG_FORMAT_INDENT,
NULL, error)) {
return FALSE;
}
return TRUE;
}
/**
* fu_main_get_updates:
**/
static GPtrArray *
fu_main_get_updates (FuMainPrivate *priv, GError **error)
{
AsApp *app;
AsRelease *rel;
FuDeviceItem *item;
GPtrArray *updates;
guint i;
const gchar *tmp;
/* find any updates using the AppStream metadata */
updates = g_ptr_array_new ();
for (i = 0; i < priv->devices->len; i++) {
const gchar *version;
_cleanup_error_free_ GError *error_local = NULL;
item = g_ptr_array_index (priv->devices, i);
/* get device version */
version = fu_device_get_metadata (item->device, FU_DEVICE_KEY_VERSION);
if (version == NULL)
continue;
/* match the GUID in the XML */
app = as_store_get_app_by_id (priv->store, fu_device_get_guid (item->device));
if (app == NULL)
continue;
/* get latest release */
rel = as_app_get_release_default (app);
if (rel == NULL) {
g_debug ("%s has no firmware update metadata",
fu_device_get_id (item->device));
continue;
}
/* check if actually newer than what we have installed */
if (as_utils_vercmp (as_release_get_version (rel), version) <= 0) {
g_debug ("%s has no firmware updates",
fu_device_get_id (item->device));
continue;
}
/* set update metadata */
tmp = as_release_get_version (rel);
if (tmp != NULL) {
fu_device_set_metadata (item->device,
FU_DEVICE_KEY_UPDATE_VERSION, tmp);
}
tmp = as_release_get_checksum (rel, G_CHECKSUM_SHA1);
if (tmp != NULL) {
fu_device_set_metadata (item->device,
FU_DEVICE_KEY_UPDATE_HASH, tmp);
}
tmp = as_release_get_location_default (rel);
if (tmp != NULL) {
fu_device_set_metadata (item->device,
FU_DEVICE_KEY_UPDATE_URI, tmp);
}
tmp = as_release_get_description (rel, NULL);
if (tmp != NULL) {
_cleanup_free_ gchar *md = NULL;
md = as_markup_convert (tmp, -1,
AS_MARKUP_CONVERT_FORMAT_SIMPLE,
NULL);
if (md != NULL) {
fu_device_set_metadata (item->device,
FU_DEVICE_KEY_UPDATE_DESCRIPTION,
md);
}
}
g_ptr_array_add (updates, item);
}
return updates;
}
/**
* fu_main_daemon_method_call:
**/
static void
fu_main_daemon_method_call (GDBusConnection *connection, const gchar *sender,
const gchar *object_path, const gchar *interface_name,
const gchar *method_name, GVariant *parameters,
GDBusMethodInvocation *invocation, gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
GVariant *val;
/* return 'as' */
if (g_strcmp0 (method_name, "GetDevices") == 0) {
_cleanup_error_free_ GError *error = NULL;
g_debug ("Called %s()", method_name);
val = fu_main_device_array_to_variant (priv->devices, &error);
if (val == NULL) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
g_dbus_method_invocation_return_value (invocation, val);
fu_main_set_status (priv, FWUPD_STATUS_IDLE);
return;
}
/* return 'as' */
if (g_strcmp0 (method_name, "GetUpdates") == 0) {
_cleanup_error_free_ GError *error = NULL;
_cleanup_ptrarray_unref_ GPtrArray *updates = NULL;
g_debug ("Called %s()", method_name);
updates = fu_main_get_updates (priv, &error);
if (updates == NULL) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
val = fu_main_device_array_to_variant (updates, &error);
if (val == NULL) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
g_dbus_method_invocation_return_value (invocation, val);
fu_main_set_status (priv, FWUPD_STATUS_IDLE);
return;
}
/* return '' */
if (g_strcmp0 (method_name, "ClearResults") == 0) {
FuDeviceItem *item = NULL;
const gchar *id = NULL;
_cleanup_error_free_ GError *error = NULL;
g_variant_get (parameters, "(&s)", &id);
g_debug ("Called %s(%s)", method_name, id);
/* find device */
item = fu_main_get_item_by_id_fallback_pending (priv, id, &error);
if (item == NULL) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* call into the provider */
if (!fu_provider_clear_results (item->provider, item->device, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* success */
g_dbus_method_invocation_return_value (invocation, NULL);
return;
}
/* return 'a{sv}' */
if (g_strcmp0 (method_name, "GetResults") == 0) {
FuDeviceItem *item = NULL;
const gchar *id = NULL;
_cleanup_error_free_ GError *error = NULL;
g_variant_get (parameters, "(&s)", &id);
g_debug ("Called %s(%s)", method_name, id);
/* find device */
item = fu_main_get_item_by_id_fallback_pending (priv, id, &error);
if (item == NULL) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* call into the provider */
if (!fu_provider_get_results (item->provider, item->device, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* success */
val = fu_device_get_metadata_as_variant (item->device);
g_dbus_method_invocation_return_value (invocation, val);
return;
}
/* return '' */
if (g_strcmp0 (method_name, "UpdateMetadata") == 0) {
GDBusMessage *message;
GUnixFDList *fd_list;
gint fd_data;
gint fd_sig;
_cleanup_error_free_ GError *error = NULL;
message = g_dbus_method_invocation_get_message (invocation);
fd_list = g_dbus_message_get_unix_fd_list (message);
if (fd_list == NULL || g_unix_fd_list_get_length (fd_list) != 2) {
g_dbus_method_invocation_return_error (invocation,
FWUPD_ERROR,
FWUPD_ERROR_INTERNAL,
"invalid handle");
return;
}
fd_data = g_unix_fd_list_get (fd_list, 0, &error);
if (fd_data < 0) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
fd_sig = g_unix_fd_list_get (fd_list, 1, &error);
if (fd_sig < 0) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
if (!fu_main_daemon_update_metadata (fd_data, fd_sig, &error)) {
g_prefix_error (&error, "failed to update metadata: ");
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
g_dbus_method_invocation_return_value (invocation, NULL);
return;
}
/* return 's' */
if (g_strcmp0 (method_name, "Verify") == 0) {
FuDeviceItem *item = NULL;
const gchar *id = NULL;
_cleanup_error_free_ GError *error = NULL;
const gchar *hash = NULL;
/* check the id exists */
g_variant_get (parameters, "(&s)", &id);
g_debug ("Called %s(%s)", method_name, id);
item = fu_main_get_item_by_id (priv, id);
if (item == NULL) {
g_dbus_method_invocation_return_error (invocation,
FWUPD_ERROR,
FWUPD_ERROR_NOT_FOUND,
"no such device %s",
id);
return;
}
if (!fu_provider_verify (item->provider, item->device,
FU_PROVIDER_VERIFY_FLAG_NONE, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
hash = fu_device_get_metadata (item->device, FU_DEVICE_KEY_FIRMWARE_HASH);
g_dbus_method_invocation_return_value (invocation,
g_variant_new ("(s)", hash));
return;
}
/* return '' */
if (g_strcmp0 (method_name, "Install") == 0) {
FuDeviceItem *item = NULL;
FuMainAuthHelper *helper;
FuProviderFlags flags = FU_PROVIDER_UPDATE_FLAG_NONE;
GDBusMessage *message;
GUnixFDList *fd_list;
GVariant *prop_value;
const gchar *action_id;
const gchar *id = NULL;
gchar *prop_key;
gint32 fd_handle = 0;
gint fd;
_cleanup_error_free_ GError *error = NULL;
_cleanup_object_unref_ PolkitSubject *subject = NULL;
_cleanup_variant_iter_free_ GVariantIter *iter = NULL;
/* check the id exists */
g_variant_get (parameters, "(&sha{sv})", &id, &fd_handle, &iter);
g_debug ("Called %s(%s,%i)", method_name, id, fd_handle);
if (g_strcmp0 (id, FWUPD_DEVICE_ID_ANY) != 0) {
item = fu_main_get_item_by_id (priv, id);
if (item == NULL) {
g_dbus_method_invocation_return_error (invocation,
FWUPD_ERROR,
FWUPD_ERROR_NOT_FOUND,
"no such device %s",
id);
return;
}
}
/* get options */
while (g_variant_iter_next (iter, "{&sv}",
&prop_key, &prop_value)) {
g_debug ("got option %s", prop_key);
if (g_strcmp0 (prop_key, "offline") == 0 &&
g_variant_get_boolean (prop_value) == TRUE)
flags |= FU_PROVIDER_UPDATE_FLAG_OFFLINE;
if (g_strcmp0 (prop_key, "allow-older") == 0 &&
g_variant_get_boolean (prop_value) == TRUE)
flags |= FU_PROVIDER_UPDATE_FLAG_ALLOW_OLDER;
if (g_strcmp0 (prop_key, "allow-reinstall") == 0 &&
g_variant_get_boolean (prop_value) == TRUE)
flags |= FU_PROVIDER_UPDATE_FLAG_ALLOW_REINSTALL;
g_variant_unref (prop_value);
}
/* get the fd */
message = g_dbus_method_invocation_get_message (invocation);
fd_list = g_dbus_message_get_unix_fd_list (message);
if (fd_list == NULL || g_unix_fd_list_get_length (fd_list) != 1) {
g_dbus_method_invocation_return_error (invocation,
FWUPD_ERROR,
FWUPD_ERROR_INTERNAL,
"invalid handle");
return;
}
fd = g_unix_fd_list_get (fd_list, fd_handle, &error);
if (fd < 0) {
g_dbus_method_invocation_return_gerror (invocation,
error);
return;
}
/* process the firmware */
helper = g_new0 (FuMainAuthHelper, 1);
helper->invocation = g_object_ref (invocation);
helper->cab_fd = fd;
helper->id = g_strdup (id);
helper->flags = flags;
helper->priv = priv;
helper->cab = fu_cab_new ();
if (item != NULL)
helper->device = g_object_ref (item->device);
if (!fu_main_update_helper (helper, &error)) {
g_dbus_method_invocation_return_gerror (helper->invocation,
error);
fu_main_set_status (priv, FWUPD_STATUS_IDLE);
fu_main_helper_free (helper);
return;
}
/* is root */
if (fu_main_dbus_get_uid (priv, sender) == 0) {
if (!fu_main_provider_update_authenticated (helper, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
} else {
g_dbus_method_invocation_return_value (invocation, NULL);
}
fu_main_set_status (priv, FWUPD_STATUS_IDLE);
fu_main_helper_free (helper);
return;
}
/* authenticate */
action_id = fu_main_get_action_id_for_device (helper);
subject = polkit_system_bus_name_new (sender);
polkit_authority_check_authorization (helper->priv->authority, subject,
action_id,
NULL,
POLKIT_CHECK_AUTHORIZATION_FLAGS_ALLOW_USER_INTERACTION,
NULL,
fu_main_check_authorization_cb,
helper);
return;
}
/* return 'a{sv}' */
if (g_strcmp0 (method_name, "GetDetails") == 0) {
GDBusMessage *message;
GUnixFDList *fd_list;
GVariantBuilder builder;
FwupdTrustFlags trust_flags;
const gchar *tmp;
gint32 fd_handle = 0;
gint fd;
_cleanup_error_free_ GError *error = NULL;
_cleanup_object_unref_ FuCab *cab = NULL;
/* check the id exists */
g_variant_get (parameters, "(h)", &fd_handle);
g_debug ("Called %s(%i)", method_name, fd_handle);
/* get the fd */
message = g_dbus_method_invocation_get_message (invocation);
fd_list = g_dbus_message_get_unix_fd_list (message);
if (fd_list == NULL || g_unix_fd_list_get_length (fd_list) != 1) {
g_dbus_method_invocation_return_error (invocation,
FWUPD_ERROR,
FWUPD_ERROR_INTERNAL,
"invalid handle");
return;
}
fd = g_unix_fd_list_get (fd_list, fd_handle, &error);
if (fd < 0) {
g_dbus_method_invocation_return_gerror (invocation,
error);
return;
}
/* load file */
cab = fu_cab_new ();
if (!fu_cab_load_fd (cab, fd, NULL, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
if (!fu_cab_delete_temp_files (cab, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* we have to extract the firmware to check the signature */
if (!fu_cab_extract (cab, FU_CAB_EXTRACT_FLAG_FIRMWARE, &error)) {
g_dbus_method_invocation_return_gerror (invocation, error);
return;
}
/* create an array with all the metadata in */
g_variant_builder_init (&builder, G_VARIANT_TYPE_ARRAY);
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_VERSION,
g_variant_new_string (fu_cab_get_version (cab)));
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_GUID,
g_variant_new_string (fu_cab_get_guid (cab)));
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_SIZE,
g_variant_new_uint64 (fu_cab_get_size (cab)));
/* optional properties */
tmp = fu_cab_get_vendor (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_VENDOR,
g_variant_new_string (tmp));
}
tmp = fu_cab_get_name (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_NAME,
g_variant_new_string (tmp));
}
tmp = fu_cab_get_summary (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_SUMMARY,
g_variant_new_string (tmp));
}
tmp = fu_cab_get_description (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_DESCRIPTION,
g_variant_new_string (tmp));
}
tmp = fu_cab_get_url_homepage (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_URL_HOMEPAGE,
g_variant_new_string (tmp));
}
tmp = fu_cab_get_license (cab);
if (tmp != NULL) {
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_LICENSE,
g_variant_new_string (tmp));
}
trust_flags = fu_cab_get_trust_flags (cab);
g_variant_builder_add (&builder, "{sv}",
FU_DEVICE_KEY_TRUSTED,
g_variant_new_uint64 (trust_flags));
/* return whole array */
val = g_variant_new ("(a{sv})", &builder);
g_dbus_method_invocation_return_value (invocation, val);
return;
}
/* we suck */
g_dbus_method_invocation_return_error (invocation,
G_DBUS_ERROR,
G_DBUS_ERROR_UNKNOWN_METHOD,
"no such method %s",
method_name);
}
/**
* fu_main_daemon_get_property:
**/
static GVariant *
fu_main_daemon_get_property (GDBusConnection *connection_, const gchar *sender,
const gchar *object_path, const gchar *interface_name,
const gchar *property_name, GError **error,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
if (g_strcmp0 (property_name, "DaemonVersion") == 0)
return g_variant_new_string (VERSION);
if (g_strcmp0 (property_name, "Status") == 0)
return g_variant_new_string (fwupd_status_to_string (priv->status));
/* return an error */
g_set_error (error,
G_DBUS_ERROR,
G_DBUS_ERROR_UNKNOWN_PROPERTY,
"failed to get daemon property %s",
property_name);
return NULL;
}
/**
* fu_main_providers_coldplug:
**/
static void
fu_main_providers_coldplug (FuMainPrivate *priv)
{
FuProvider *provider;
guint i;
for (i = 0; i < priv->providers->len; i++) {
_cleanup_error_free_ GError *error = NULL;
provider = g_ptr_array_index (priv->providers, i);
if (!fu_provider_coldplug (FU_PROVIDER (provider), &error))
g_warning ("Failed to coldplug: %s", error->message);
}
}
/**
* fu_main_on_bus_acquired_cb:
**/
static void
fu_main_on_bus_acquired_cb (GDBusConnection *connection,
const gchar *name,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
guint registration_id;
_cleanup_error_free_ GError *error = NULL;
static const GDBusInterfaceVTable interface_vtable = {
fu_main_daemon_method_call,
fu_main_daemon_get_property,
NULL
};
priv->connection = g_object_ref (connection);
registration_id = g_dbus_connection_register_object (connection,
FWUPD_DBUS_PATH,
priv->introspection_daemon->interfaces[0],
&interface_vtable,
priv, /* user_data */
NULL, /* user_data_free_func */
NULL); /* GError** */
g_assert (registration_id > 0);
/* add devices */
fu_main_providers_coldplug (priv);
/* connect to D-Bus directly */
priv->proxy_uid =
g_dbus_proxy_new_sync (priv->connection,
G_DBUS_PROXY_FLAGS_DO_NOT_LOAD_PROPERTIES |
G_DBUS_PROXY_FLAGS_DO_NOT_CONNECT_SIGNALS,
NULL,
"org.freedesktop.DBus",
"/org/freedesktop/DBus",
"org.freedesktop.DBus",
NULL,
&error);
if (priv->proxy_uid == NULL) {
g_warning ("cannot connect to DBus: %s", error->message);
return;
}
}
/**
* fu_main_on_name_acquired_cb:
**/
static void
fu_main_on_name_acquired_cb (GDBusConnection *connection,
const gchar *name,
gpointer user_data)
{
g_debug ("FuMain: acquired name: %s", name);
}
/**
* fu_main_on_name_lost_cb:
**/
static void
fu_main_on_name_lost_cb (GDBusConnection *connection,
const gchar *name,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
g_debug ("FuMain: lost name: %s", name);
g_main_loop_quit (priv->loop);
}
/**
* fu_main_timed_exit_cb:
**/
static gboolean
fu_main_timed_exit_cb (gpointer user_data)
{
GMainLoop *loop = (GMainLoop *) user_data;
g_main_loop_quit (loop);
return G_SOURCE_REMOVE;
}
/**
* fu_main_load_introspection:
**/
static GDBusNodeInfo *
fu_main_load_introspection (const gchar *filename, GError **error)
{
_cleanup_bytes_unref_ GBytes *data = NULL;
_cleanup_free_ gchar *path = NULL;
/* lookup data */
path = g_build_filename ("/org/freedesktop/fwupd", filename, NULL);
data = g_resource_lookup_data (fu_get_resource (),
path,
G_RESOURCE_LOOKUP_FLAGS_NONE,
error);
if (data == NULL)
return NULL;
/* build introspection from XML */
return g_dbus_node_info_new_for_xml (g_bytes_get_data (data, NULL), error);
}
/**
* cd_main_provider_device_added_cb:
**/
static void
cd_main_provider_device_added_cb (FuProvider *provider,
FuDevice *device,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
FuDeviceItem *item;
/* remove any fake device */
item = fu_main_get_item_by_id (priv, fu_device_get_id (device));
if (item != NULL)
g_ptr_array_remove (priv->devices, item);
/* create new device */
item = g_new0 (FuDeviceItem, 1);
item->device = g_object_ref (device);
item->provider = g_object_ref (provider);
g_ptr_array_add (priv->devices, item);
fu_main_emit_changed (priv);
}
/**
* cd_main_provider_device_removed_cb:
**/
static void
cd_main_provider_device_removed_cb (FuProvider *provider,
FuDevice *device,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
FuDeviceItem *item;
item = fu_main_get_item_by_id (priv, fu_device_get_id (device));
if (item == NULL) {
g_warning ("can't remove device %s", fu_device_get_id (device));
return;
}
g_ptr_array_remove (priv->devices, item);
fu_main_emit_changed (priv);
}
/**
* cd_main_provider_status_changed_cb:
**/
static void
cd_main_provider_status_changed_cb (FuProvider *provider,
FwupdStatus status,
gpointer user_data)
{
FuMainPrivate *priv = (FuMainPrivate *) user_data;
fu_main_set_status (priv, status);
}
/**
* fu_main_add_provider:
**/
static void
fu_main_add_provider (FuMainPrivate *priv, FuProvider *provider)
{
g_signal_connect (provider, "device-added",
G_CALLBACK (cd_main_provider_device_added_cb),
priv);
g_signal_connect (provider, "device-removed",
G_CALLBACK (cd_main_provider_device_removed_cb),
priv);
g_signal_connect (provider, "status-changed",
G_CALLBACK (cd_main_provider_status_changed_cb),
priv);
g_ptr_array_add (priv->providers, provider);
}
/**
* main:
**/
int
main (int argc, char *argv[])
{
FuMainPrivate *priv = NULL;
gboolean immediate_exit = FALSE;
gboolean ret;
gboolean timed_exit = FALSE;
GOptionContext *context;
guint owner_id = 0;
guint retval = 1;
const GOptionEntry options[] = {
{ "timed-exit", '\0', 0, G_OPTION_ARG_NONE, &timed_exit,
/* TRANSLATORS: exit after we've started up, used for user profiling */
_("Exit after a small delay"), NULL },
{ "immediate-exit", '\0', 0, G_OPTION_ARG_NONE, &immediate_exit,
/* TRANSLATORS: exit straight away, used for automatic profiling */
_("Exit after the engine has loaded"), NULL },
{ NULL}
};
_cleanup_error_free_ GError *error = NULL;
setlocale (LC_ALL, "");
bindtextdomain (GETTEXT_PACKAGE, LOCALEDIR);
bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
textdomain (GETTEXT_PACKAGE);
/* TRANSLATORS: program name */
g_set_application_name (_("Firmware Update Daemon"));
context = g_option_context_new (NULL);
g_option_context_add_main_entries (context, options, NULL);
g_option_context_add_group (context, fu_debug_get_option_group ());
/* TRANSLATORS: program summary */
g_option_context_set_summary (context, _("Firmware Update D-Bus Service"));
ret = g_option_context_parse (context, &argc, &argv, &error);
if (!ret) {
g_warning ("FuMain: failed to parse command line arguments: %s",
error->message);
goto out;
}
/* create new objects */
priv = g_new0 (FuMainPrivate, 1);
priv->status = FWUPD_STATUS_IDLE;
priv->devices = g_ptr_array_new_with_free_func ((GDestroyNotify) fu_main_item_free);
priv->loop = g_main_loop_new (NULL, FALSE);
priv->pending = fu_pending_new ();
priv->store = as_store_new ();
/* load AppStream */
as_store_add_filter (priv->store, AS_ID_KIND_FIRMWARE);
if (!as_store_load (priv->store,
AS_STORE_LOAD_FLAG_APP_INFO_SYSTEM,
NULL, &error)){
g_warning ("FuMain: failed to load AppStream data: %s",
error->message);
return FALSE;
}
/* add providers */
priv->providers = g_ptr_array_new_with_free_func ((GDestroyNotify) g_object_unref);
fu_main_add_provider (priv, fu_provider_udev_new ());
fu_main_add_provider (priv, fu_provider_usb_new ());
#ifdef HAVE_COLORHUG
fu_main_add_provider (priv, fu_provider_chug_new ());
#endif
#ifdef HAVE_UEFI
fu_main_add_provider (priv, fu_provider_uefi_new ());
#endif
/* load introspection from file */
priv->introspection_daemon = fu_main_load_introspection (FWUPD_DBUS_INTERFACE ".xml",
&error);
if (priv->introspection_daemon == NULL) {
g_warning ("FuMain: failed to load daemon introspection: %s",
error->message);
goto out;
}
/* get authority */
priv->authority = polkit_authority_get_sync (NULL, &error);
if (priv->authority == NULL) {
g_warning ("FuMain: failed to load polkit authority: %s",
error->message);
goto out;
}
/* own the object */
owner_id = g_bus_own_name (G_BUS_TYPE_SYSTEM,
FWUPD_DBUS_SERVICE,
G_BUS_NAME_OWNER_FLAGS_ALLOW_REPLACEMENT |
G_BUS_NAME_OWNER_FLAGS_REPLACE,
fu_main_on_bus_acquired_cb,
fu_main_on_name_acquired_cb,
fu_main_on_name_lost_cb,
priv, NULL);
/* Only timeout and close the mainloop if we have specified it
* on the command line */
if (immediate_exit)
g_idle_add (fu_main_timed_exit_cb, priv->loop);
else if (timed_exit)
g_timeout_add_seconds (5, fu_main_timed_exit_cb, priv->loop);
/* wait */
g_info ("Daemon ready for requests");
g_main_loop_run (priv->loop);
/* success */
retval = 0;
out:
g_option_context_free (context);
if (owner_id > 0)
g_bus_unown_name (owner_id);
if (priv != NULL) {
if (priv->loop != NULL)
g_main_loop_unref (priv->loop);
if (priv->proxy_uid != NULL)
g_object_unref (priv->proxy_uid);
if (priv->connection != NULL)
g_object_unref (priv->connection);
if (priv->authority != NULL)
g_object_unref (priv->authority);
if (priv->store != NULL)
g_object_unref (priv->store);
if (priv->introspection_daemon != NULL)
g_dbus_node_info_unref (priv->introspection_daemon);
g_object_unref (priv->pending);
g_ptr_array_unref (priv->providers);
g_ptr_array_unref (priv->devices);
g_free (priv);
}
return retval;
}