Skip to content

Commit

Permalink
daemon: Load sd unit for callers, log it
Browse files Browse the repository at this point in the history
The high level goal is to render in a better way what caused an
update: coreos#247 (comment)

This gets us for Cockpit:
`Initiated txn DownloadUpdateRpmDiff for client(dbus:1.28 unit:session-6.scope uid:0): /org/projectatomic/rpmostree1/fedora_atomic`
which isn't as good as I'd hoped; I was thinking we'd get `cockpit.service`
but actually Cockpit does invocations as a real login for good reason.

We get a similar result from the CLI.
  • Loading branch information
cgwalters committed May 15, 2018
1 parent bb7b489 commit 83c942d
Showing 1 changed file with 73 additions and 10 deletions.
83 changes: 73 additions & 10 deletions src/daemon/rpmostreed-daemon.c
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@

#include <libglnx.h>
#include <systemd/sd-journal.h>
#include <systemd/sd-login.h>
#include <systemd/sd-daemon.h>
#include <stdio.h>

Expand All @@ -35,6 +36,9 @@
#define DAEMON_CONFIG_GROUP "Daemon"
#define EXPERIMENTAL_CONFIG_GROUP "Experimental"

struct RpmOstreeClient;
static struct RpmOstreeClient *client_new (RpmostreedDaemon *self, const char *address);

/**
* SECTION: daemon
* @title: RpmostreedDaemon
Expand Down Expand Up @@ -96,26 +100,39 @@ G_DEFINE_TYPE_WITH_CODE (RpmostreedDaemon, rpmostreed_daemon, G_TYPE_OBJECT,
struct RpmOstreeClient {
char *address;
guint name_watch_id;
/* In Rust this'd be Option<uid_t> etc. */
gboolean uid_valid;
uid_t uid;
gboolean pid_valid;
pid_t pid;
char *sd_unit;
};

static void
rpmostree_client_free (struct RpmOstreeClient *client)
{
if (!client)
return;
g_free (client->address);
g_free (client->sd_unit);
g_free (client);
}

static char *
rpmostree_client_to_string (struct RpmOstreeClient *client)
{
g_autoptr(GString) buf = g_string_new ("client ");
/* Since DBus addresses have a leading ':', let's avoid another. Yeah it's not
* symmetric, but it does read better.
*/
g_autoptr(GString) buf = g_string_new ("client(dbus");
g_string_append (buf, client->address);
if (client->sd_unit)
g_string_append_printf (buf, " unit:%s", client->sd_unit);
if (client->uid_valid)
g_string_append_printf (buf, " (uid %lu)", (unsigned long) client->uid);
g_string_append_printf (buf, " uid:%lu", (unsigned long) client->uid);
else
g_string_append (buf, " (unknown uid)");
g_string_append (buf, " uid:<unknown>");
g_string_append_c (buf, ')');
return g_string_free (g_steal_pointer (&buf), FALSE);
}

Expand Down Expand Up @@ -223,16 +240,24 @@ on_active_txn_changed (GObject *object,
g_variant_get (active_txn, "(&s&s&s)", &method, &sender, &path);
if (*method)
{
g_autofree char *client_str = rpmostreed_daemon_client_get_string (self, sender);
struct RpmOstreeClient *clientdata = g_hash_table_lookup (self->bus_clients, sender);
struct RpmOstreeClient *clientdata_owned = NULL;
/* If the caller didn't register (e.g. Cockpit doesn't today),
* then let's gather the relevant client info now.
*/
if (!clientdata)
clientdata = clientdata_owned = client_new (self, sender);
g_autofree char *client_str = rpmostree_client_to_string (clientdata);
g_autofree char *client_data_msg = NULL;
if (clientdata && clientdata->uid_valid)
if (clientdata->uid_valid)
client_data_msg = g_strdup_printf ("CLIENT_UID=%u", clientdata->uid);
/* TODO: convert other client fields to structured log */
sd_journal_send ("MESSAGE_ID=" SD_ID128_FORMAT_STR, SD_ID128_FORMAT_VAL(RPMOSTREE_MESSAGE_TRANSACTION_STARTED),
"MESSAGE=Initiated txn %s for %s: %s", method, client_str, path,
"BUS_ADDRESS=%s", sender,
client_data_msg ?: NULL,
NULL);
rpmostree_client_free (clientdata_owned);
}
}

Expand Down Expand Up @@ -616,15 +641,56 @@ rpmostreed_get_client_uid (RpmostreedDaemon *self,
return TRUE;
}

static gboolean
get_client_pid (RpmostreedDaemon *self,
const char *client,
pid_t *out_pid)
{
g_autoptr(GError) local_error = NULL;
g_autoptr(GVariant) all = g_dbus_proxy_call_sync (self->bus_proxy,
"GetConnectionUnixProcessID",
g_variant_new ("(s)", client),
G_DBUS_CALL_FLAGS_NONE,
2000, NULL, &local_error);
if (!all)
{
sd_journal_print (LOG_WARNING, "Failed to GetConnectionUnixProcessID for client %s: %s",
client, local_error->message);
return FALSE;
}

g_variant_get (all, "(u)", out_pid);
return TRUE;
}

/* Given a DBus address, load metadata for it */
static struct RpmOstreeClient *
client_new (RpmostreedDaemon *self, const char *address)
{
struct RpmOstreeClient *client = g_new0 (struct RpmOstreeClient, 1);
client->address = g_strdup (address);
if (rpmostreed_get_client_uid (self, address, &client->uid))
client->uid_valid = TRUE;
if (get_client_pid (self, address, &client->pid))
{
client->pid_valid = TRUE;
if (sd_pid_get_user_unit (client->pid, &client->sd_unit) == 0)
{}
else if (sd_pid_get_unit (client->pid, &client->sd_unit) == 0)
{}
}

return g_steal_pointer (&client);
}

void
rpmostreed_daemon_add_client (RpmostreedDaemon *self,
const char *client)
{
if (g_hash_table_lookup (self->bus_clients, client))
return;

struct RpmOstreeClient *clientdata = g_new0 (struct RpmOstreeClient, 1);
clientdata->address = g_strdup (client);
struct RpmOstreeClient *clientdata = client_new (self, client);
clientdata->name_watch_id =
g_dbus_connection_signal_subscribe (self->connection,
"org.freedesktop.DBus",
Expand All @@ -637,9 +703,6 @@ rpmostreed_daemon_add_client (RpmostreedDaemon *self,
g_object_ref (self),
g_object_unref);

if (rpmostreed_get_client_uid (self, client, &clientdata->uid))
clientdata->uid_valid = TRUE;

g_hash_table_insert (self->bus_clients, (char*)clientdata->address, clientdata);
g_autofree char *clientstr = rpmostree_client_to_string (clientdata);
sd_journal_print (LOG_INFO, "%s added; new total=%u", clientstr, g_hash_table_size (self->bus_clients));
Expand Down

0 comments on commit 83c942d

Please sign in to comment.