#include <getopt.h>

#include "strv.h"

#include "util.h"

#include "log.h"

#include "build.h"

#include "pager.h"

#include "sd-bus.h"

#include "bus-message.h"

#include "bus-internal.h"

#include "bus-util.h"

#include "bus-dump.h"

static bool arg_no_pager = false;

static bool arg_legend = true;

static char *arg_address = NULL;

static bool arg_unique = false;

static bool arg_acquired = false;

static bool arg_activatable = false;

static bool arg_show_machine = false;

static char **arg_matches = NULL;

static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;

static char *arg_host = NULL;

static bool arg_user = false;

static void pager_open_if_enabled(void) {

/* Cache result before we open the pager */

if (arg_no_pager)

return;

pager_open(false);

}

static int list_bus_names(sd_bus *bus, char **argv) {

_cleanup_strv_free_ char **acquired = NULL, **activatable = NULL;

_cleanup_free_ char **merged = NULL;

_cleanup_hashmap_free_ Hashmap *names = NULL;

char **i;

int r;

size_t max_i = 0;

unsigned n = 0;

void *v;

char *k;

Iterator iterator;

assert(bus);

r = sd_bus_list_names(bus, (arg_acquired || arg_unique) ? &acquired : NULL, arg_activatable ? &activatable : NULL);

if (r < 0) {

log_error("Failed to list names: %s", strerror(-r));

return r;

}

pager_open_if_enabled();

names = hashmap_new(string_hash_func, string_compare_func);

if (!names)

return log_oom();

STRV_FOREACH(i, acquired) {

max_i = MAX(max_i, strlen(*i));

r = hashmap_put(names, *i, INT_TO_PTR(1));

if (r < 0) {

log_error("Failed to add to hashmap: %s", strerror(-r));

return r;

}

}

STRV_FOREACH(i, activatable) {

max_i = MAX(max_i, strlen(*i));

r = hashmap_put(names, *i, INT_TO_PTR(2));

if (r < 0 && r != -EEXIST) {

log_error("Failed to add to hashmap: %s", strerror(-r));

return r;

}

}

merged = new(char*, hashmap_size(names) + 1);

HASHMAP_FOREACH_KEY(v, k, names, iterator)

merged[n++] = k;

merged[n] = NULL;

strv_sort(merged);

if (arg_legend) {

printf("%-*s %*s %-*s %-*s %-*s %-*s %-*s %-*s",

(int) max_i, "NAME", 10, "PID", 15, "PROCESS", 16, "USER", 13, "CONNECTION", 25, "UNIT", 10, "SESSION", 19, "CONNECTION-NAME");

if (arg_show_machine)

puts(" MACHINE");

else

putchar('\n');

}

STRV_FOREACH(i, merged) {

_cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;

sd_id128_t mid;

if (hashmap_get(names, *i) == INT_TO_PTR(2)) {

/* Activatable */

printf("%-*s", (int) max_i, *i);

printf(" - - - (activatable) - - ");

if (arg_show_machine)

puts(" -");

else

putchar('\n');

continue;

}

if (!arg_unique && (*i)[0] == ':')

continue;

if (!arg_acquired && (*i)[0] != ':')

continue;

printf("%-*s", (int) max_i, *i);

r = sd_bus_get_owner(bus, *i,

SD_BUS_CREDS_UID|SD_BUS_CREDS_PID|SD_BUS_CREDS_COMM|

SD_BUS_CREDS_UNIQUE_NAME|SD_BUS_CREDS_UNIT|SD_BUS_CREDS_SESSION|

SD_BUS_CREDS_CONNECTION_NAME, &creds);

if (r >= 0) {

const char *unique, *session, *unit, *cn;

pid_t pid;

uid_t uid;

r = sd_bus_creds_get_pid(creds, &pid);

if (r >= 0) {

const char *comm = NULL;

sd_bus_creds_get_comm(creds, &comm);

printf(" %10lu %-15s", (unsigned long) pid, strna(comm));

} else

fputs(" - - ", stdout);

r = sd_bus_creds_get_uid(creds, &uid);

if (r >= 0) {

_cleanup_free_ char *u = NULL;

u = uid_to_name(uid);

if (!u)

return log_oom();

if (strlen(u) > 16)

u[16] = 0;

printf(" %-16s", u);

} else

fputs(" - ", stdout);

r = sd_bus_creds_get_unique_name(creds, &unique);

if (r >= 0)

printf(" %-13s", unique);

else

fputs(" - ", stdout);

r = sd_bus_creds_get_unit(creds, &unit);

if (r >= 0) {

_cleanup_free_ char *e;

e = ellipsize(unit, 25, 100);

if (!e)

return log_oom();

printf(" %-25s", e);

} else

fputs(" - ", stdout);

r = sd_bus_creds_get_session(creds, &session);

if (r >= 0)

printf(" %-10s", session);

else

fputs(" - ", stdout);

r = sd_bus_creds_get_connection_name(creds, &cn);

if (r >= 0)

printf(" %-19s", cn);

else

fputs(" - ", stdout);

} else

printf(" - - - - - - - ");

if (arg_show_machine) {

r = sd_bus_get_owner_machine_id(bus, *i, &mid);

if (r >= 0) {

char m[SD_ID128_STRING_MAX];

printf(" %s\n", sd_id128_to_string(mid, m));

} else

puts(" -");

} else

putchar('\n');

}

return 0;

}

static int monitor(sd_bus *bus, char *argv[]) {

bool added_something = false;

char **i;

int r;

STRV_FOREACH(i, argv+1) {

_cleanup_free_ char *m = NULL;

if (!service_name_is_valid(*i)) {

log_error("Invalid service name '%s'", *i);

return -EINVAL;

}

m = strjoin("sender='", *i, "'", NULL);

if (!m)

return log_oom();

r = sd_bus_add_match(bus, m, NULL, NULL);

if (r < 0) {

log_error("Failed to add match: %s", strerror(-r));

return r;

}

added_something = true;

}

STRV_FOREACH(i, arg_matches) {

r = sd_bus_add_match(bus, *i, NULL, NULL);

if (r < 0) {

log_error("Failed to add match: %s", strerror(-r));

return r;

}

added_something = true;

}

if (!added_something) {

r = sd_bus_add_match(bus, "", NULL, NULL);

if (r < 0) {

log_error("Failed to add match: %s", strerror(-r));

return r;

}

}

for (;;) {

_cleanup_bus_message_unref_ sd_bus_message *m = NULL;

r = sd_bus_process(bus, &m);

if (r < 0) {

log_error("Failed to process bus: %s", strerror(-r));

return r;

}

if (m) {

bus_message_dump(m, stdout, true);

continue;

}

if (r > 0)

continue;

r = sd_bus_wait(bus, (uint64_t) -1);

if (r < 0) {

log_error("Failed to wait for bus: %s", strerror(-r));

return r;

}

}

}

static int status(sd_bus *bus, char *argv[]) {

_cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;

pid_t pid;

int r;

assert(bus);

if (strv_length(argv) != 2) {

log_error("Expects one argument.");

return -EINVAL;

}

r = parse_pid(argv[1], &pid);

if (r < 0)

r = sd_bus_get_owner(bus, argv[1], _SD_BUS_CREDS_ALL, &creds);

else

r = sd_bus_creds_new_from_pid(&creds, pid, _SD_BUS_CREDS_ALL);

if (r < 0) {

log_error("Failed to get credentials: %s", strerror(-r));

return r;

}

bus_creds_dump(creds, NULL);

return 0;

}

static int help(void) {

printf("%s [OPTIONS...] {COMMAND} ...\n\n"

"Introspect the bus.\n\n"

" -h --help Show this help\n"

" --version Show package version\n"

" --no-pager Do not pipe output into a pager\n"

" --no-legend Do not show the headers and footers\n"

" --system Connect to system bus\n"

" --user Connect to user bus\n"

" -H --host=[USER@]HOST Operate on remote host\n"

" -M --machine=CONTAINER Operate on local container\n"

" --address=ADDRESS Connect to bus specified by address\n"

" --show-machine Show machine ID column in list\n"

" --unique Only show unique names\n"

" --acquired Only show acquired names\n"

" --activatable Only show activatable names\n"

" --match=MATCH Only show matching messages\n\n"

"Commands:\n"

" list List bus names\n"

" monitor [SERVICE...] Show bus traffic\n"

" status NAME Show name status\n"

" help Show this help\n",

program_invocation_short_name);

return 0;

}

static int parse_argv(int argc, char *argv[]) {

enum {

ARG_VERSION = 0x100,

ARG_NO_PAGER,

ARG_NO_LEGEND,

ARG_SYSTEM,

ARG_USER,

ARG_ADDRESS,

ARG_MATCH,

ARG_SHOW_MACHINE,

ARG_UNIQUE,

ARG_ACQUIRED,

ARG_ACTIVATABLE

};

static const struct option options[] = {

{ "help", no_argument, NULL, 'h' },

{ "version", no_argument, NULL, ARG_VERSION },

{ "no-pager", no_argument, NULL, ARG_NO_PAGER },

{ "no-legend", no_argument, NULL, ARG_NO_LEGEND },

{ "system", no_argument, NULL, ARG_SYSTEM },

{ "user", no_argument, NULL, ARG_USER },

{ "address", required_argument, NULL, ARG_ADDRESS },

{ "show-machine", no_argument, NULL, ARG_SHOW_MACHINE },

{ "unique", no_argument, NULL, ARG_UNIQUE },

{ "acquired", no_argument, NULL, ARG_ACQUIRED },

{ "activatable", no_argument, NULL, ARG_ACTIVATABLE },

{ "match", required_argument, NULL, ARG_MATCH },

{ "host", required_argument, NULL, 'H' },

{ "machine", required_argument, NULL, 'M' },

{},

};

int c;

assert(argc >= 0);

assert(argv);

while ((c = getopt_long(argc, argv, "hH:M:", options, NULL)) >= 0) {

switch (c) {

case 'h':

return help();

case ARG_VERSION:

puts(PACKAGE_STRING);

puts(SYSTEMD_FEATURES);

return 0;

case ARG_NO_PAGER:

arg_no_pager = true;

break;

case ARG_NO_LEGEND:

arg_legend = false;

break;

case ARG_USER:

arg_user = true;

break;

case ARG_SYSTEM:

arg_user = false;

break;

case ARG_ADDRESS:

arg_address = optarg;

break;

case ARG_SHOW_MACHINE:

arg_show_machine = true;

break;

case ARG_UNIQUE:

arg_unique = true;

break;

case ARG_ACQUIRED:

arg_acquired = true;

break;

case ARG_ACTIVATABLE:

arg_activatable = true;

break;

case ARG_MATCH:

if (strv_extend(&arg_matches, optarg) < 0)

return log_oom();

break;

case 'H':

arg_transport = BUS_TRANSPORT_REMOTE;

arg_host = optarg;

break;

case 'M':

arg_transport = BUS_TRANSPORT_CONTAINER;

arg_host = optarg;

break;

case '?':

return -EINVAL;

default:

assert_not_reached("Unhandled option");

}

}

if (!arg_unique && !arg_acquired && !arg_activatable)

arg_unique = arg_acquired = arg_activatable = true;

return 1;

}

static int busctl_main(sd_bus *bus, int argc, char *argv[]) {

assert(bus);

if (optind >= argc ||

streq(argv[optind], "list"))

return list_bus_names(bus, argv + optind);

if (streq(argv[optind], "monitor"))

return monitor(bus, argv + optind);

if (streq(argv[optind], "status"))

return status(bus, argv + optind);

if (streq(argv[optind], "help"))

return help();

log_error("Unknown command '%s'", argv[optind]);

return -EINVAL;

}

int main(int argc, char *argv[]) {

_cleanup_bus_unref_ sd_bus *bus = NULL;

int r;

log_parse_environment();

log_open();

r = parse_argv(argc, argv);

if (r <= 0)

goto finish;

r = sd_bus_new(&bus);

if (r < 0) {

log_error("Failed to allocate bus: %s", strerror(-r));

goto finish;

}

if (streq_ptr(argv[optind], "monitor")) {

r = sd_bus_set_monitor(bus, true);

if (r < 0) {

log_error("Failed to set monitor mode: %s", strerror(-r));

goto finish;

}

}

if (arg_address)

r = sd_bus_set_address(bus, arg_address);

else {

switch (arg_transport) {

case BUS_TRANSPORT_LOCAL:

if (arg_user)

r = bus_set_address_user(bus);

else

r = bus_set_address_system(bus);

break;

case BUS_TRANSPORT_REMOTE:

r = bus_set_address_system_remote(bus, arg_host);

break;

case BUS_TRANSPORT_CONTAINER:

r = bus_set_address_system_container(bus, arg_host);

break;

default:

assert_not_reached("Hmm, unknown transport type.");

}

}

if (r < 0) {

log_error("Failed to set address: %s", strerror(-r));

goto finish;

}

r = sd_bus_set_bus_client(bus, true);

if (r < 0) {

log_error("Failed to set bus client: %s", strerror(-r));

goto finish;

}

r = sd_bus_start(bus);

if (r < 0) {

log_error("Failed to connect to bus: %s", strerror(-r));

goto finish;

}

r = busctl_main(bus, argc, argv);

finish:

pager_close();

strv_free(arg_matches);

return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;

}