cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <arpa/inet.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <errno.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <fcntl.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <getopt.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <locale.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <net/if.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <netinet/in.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <string.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include <sys/mount.h>

7fd1b19bc9e9f5574f2877936b8ac267c7706947Harald Hoyer#include <sys/socket.h>

7fd1b19bc9e9f5574f2877936b8ac267c7706947Harald Hoyer#include <unistd.h>

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "sd-bus.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "alloc-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "bus-error.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "bus-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "cgroup-show.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "cgroup-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "copy.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "env-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "event-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "fd-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "hostname-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "import-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "log.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "logs-show.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "macro.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "mkdir.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "pager.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "parse-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "path-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "process-util.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "ptyfwd.h"

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poettering#include "signal-util.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "spawn-polkit-agent.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "strv.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "terminal-util.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "unit-name.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "util.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "verbs.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer#include "web-util.h"

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyer

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic char **arg_property = NULL;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic bool arg_all = false;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic bool arg_full = false;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic bool arg_no_pager = false;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic bool arg_legend = true;

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic const char *arg_kill_who = NULL;

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic int arg_signal = SIGTERM;

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic BusTransport arg_transport = BUS_TRANSPORT_LOCAL;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic char *arg_host = NULL;

cd34b3c6670df8a3fd49179131fe762b2dd86b01Harald Hoyerstatic bool arg_read_only = false;

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic bool arg_mkdir = false;

cbdca8525b4f36297cb9e5cb090a9648763ed1bfLennart Poetteringstatic bool arg_quiet = false;

static bool arg_ask_password = true;

static unsigned arg_lines = 10;

static OutputMode arg_output = OUTPUT_SHORT;

static bool arg_force = false;

static ImportVerify arg_verify = IMPORT_VERIFY_SIGNATURE;

static const char* arg_dkr_index_url = NULL;

static const char* arg_format = NULL;

static const char *arg_uid = NULL;

static char **arg_setenv = NULL;

static void pager_open_if_enabled(void) {

if (arg_no_pager)

return;

pager_open(false);

}

static void polkit_agent_open_if_enabled(void) {

/* Open the polkit agent as a child process if necessary */

if (!arg_ask_password)

return;

if (arg_transport != BUS_TRANSPORT_LOCAL)

return;

polkit_agent_open();

}

static OutputFlags get_output_flags(void) {

return

arg_all * OUTPUT_SHOW_ALL |

arg_full * OUTPUT_FULL_WIDTH |

(!on_tty() || pager_have()) * OUTPUT_FULL_WIDTH |

on_tty() * OUTPUT_COLOR |

!arg_quiet * OUTPUT_WARN_CUTOFF;

}

typedef struct MachineInfo {

const char *name;

const char *class;

const char *service;

} MachineInfo;

static int compare_machine_info(const void *a, const void *b) {

const MachineInfo *x = a, *y = b;

return strcmp(x->name, y->name);

}

static int list_machines(int argc, char *argv[], void *userdata) {

size_t max_name = strlen("MACHINE"), max_class = strlen("CLASS"), max_service = strlen("SERVICE");

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_free_ MachineInfo *machines = NULL;

const char *name, *class, *service, *object;

size_t n_machines = 0, n_allocated = 0, j;

sd_bus *bus = userdata;

int r;

assert(bus);

pager_open_if_enabled();

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"ListMachines",

&error,

&reply,

NULL);

if (r < 0) {

log_error("Could not get machines: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_enter_container(reply, 'a', "(ssso)");

if (r < 0)

return bus_log_parse_error(r);

while ((r = sd_bus_message_read(reply, "(ssso)", &name, &class, &service, &object)) > 0) {

size_t l;

if (name[0] == '.' && !arg_all)

continue;

if (!GREEDY_REALLOC(machines, n_allocated, n_machines + 1))

return log_oom();

machines[n_machines].name = name;

machines[n_machines].class = class;

machines[n_machines].service = service;

l = strlen(name);

if (l > max_name)

max_name = l;

l = strlen(class);

if (l > max_class)

max_class = l;

l = strlen(service);

if (l > max_service)

max_service = l;

n_machines ++;

}

if (r < 0)

return bus_log_parse_error(r);

r = sd_bus_message_exit_container(reply);

if (r < 0)

return bus_log_parse_error(r);

qsort_safe(machines, n_machines, sizeof(MachineInfo), compare_machine_info);

if (arg_legend)

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

(int) max_name, "MACHINE",

(int) max_class, "CLASS",

(int) max_service, "SERVICE");

for (j = 0; j < n_machines; j++)

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

(int) max_name, machines[j].name,

(int) max_class, machines[j].class,

(int) max_service, machines[j].service);

if (arg_legend)

printf("\n%zu machines listed.\n", n_machines);

return 0;

}

typedef struct ImageInfo {

const char *name;

const char *type;

bool read_only;

usec_t crtime;

usec_t mtime;

uint64_t size;

} ImageInfo;

static int compare_image_info(const void *a, const void *b) {

const ImageInfo *x = a, *y = b;

return strcmp(x->name, y->name);

}

static int list_images(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

size_t max_name = strlen("NAME"), max_type = strlen("TYPE"), max_size = strlen("USAGE"), max_crtime = strlen("CREATED"), max_mtime = strlen("MODIFIED");

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_free_ ImageInfo *images = NULL;

size_t n_images = 0, n_allocated = 0, j;

const char *name, *type, *object;

sd_bus *bus = userdata;

uint64_t crtime, mtime, size;

int read_only, r;

assert(bus);

pager_open_if_enabled();

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"ListImages",

&error,

&reply,

"");

if (r < 0) {

log_error("Could not get images: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_enter_container(reply, SD_BUS_TYPE_ARRAY, "(ssbttto)");

if (r < 0)

return bus_log_parse_error(r);

while ((r = sd_bus_message_read(reply, "(ssbttto)", &name, &type, &read_only, &crtime, &mtime, &size, &object)) > 0) {

char buf[MAX(FORMAT_TIMESTAMP_MAX, FORMAT_BYTES_MAX)];

size_t l;

if (name[0] == '.' && !arg_all)

continue;

if (!GREEDY_REALLOC(images, n_allocated, n_images + 1))

return log_oom();

images[n_images].name = name;

images[n_images].type = type;

images[n_images].read_only = read_only;

images[n_images].crtime = crtime;

images[n_images].mtime = mtime;

images[n_images].size = size;

l = strlen(name);

if (l > max_name)

max_name = l;

l = strlen(type);

if (l > max_type)

max_type = l;

if (crtime != 0) {

l = strlen(strna(format_timestamp(buf, sizeof(buf), crtime)));

if (l > max_crtime)

max_crtime = l;

}

if (mtime != 0) {

l = strlen(strna(format_timestamp(buf, sizeof(buf), mtime)));

if (l > max_mtime)

max_mtime = l;

}

if (size != (uint64_t) -1) {

l = strlen(strna(format_bytes(buf, sizeof(buf), size)));

if (l > max_size)

max_size = l;

}

n_images++;

}

if (r < 0)

return bus_log_parse_error(r);

r = sd_bus_message_exit_container(reply);

if (r < 0)

return bus_log_parse_error(r);

qsort_safe(images, n_images, sizeof(ImageInfo), compare_image_info);

if (arg_legend)

printf("%-*s %-*s %-3s %-*s %-*s %-*s\n",

(int) max_name, "NAME",

(int) max_type, "TYPE",

"RO",

(int) max_size, "USAGE",

(int) max_crtime, "CREATED",

(int) max_mtime, "MODIFIED");

for (j = 0; j < n_images; j++) {

char crtime_buf[FORMAT_TIMESTAMP_MAX], mtime_buf[FORMAT_TIMESTAMP_MAX], size_buf[FORMAT_BYTES_MAX];

printf("%-*s %-*s %s%-3s%s %-*s %-*s %-*s\n",

(int) max_name, images[j].name,

(int) max_type, images[j].type,

images[j].read_only ? ansi_highlight_red() : "", yes_no(images[j].read_only), images[j].read_only ? ansi_normal() : "",

(int) max_size, strna(format_bytes(size_buf, sizeof(size_buf), images[j].size)),

(int) max_crtime, strna(format_timestamp(crtime_buf, sizeof(crtime_buf), images[j].crtime)),

(int) max_mtime, strna(format_timestamp(mtime_buf, sizeof(mtime_buf), images[j].mtime)));

}

if (arg_legend)

printf("\n%zu images listed.\n", n_images);

return 0;

}

static int show_unit_cgroup(sd_bus *bus, const char *unit, pid_t leader) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_free_ char *path = NULL;

const char *cgroup;

int r;

unsigned c;

assert(bus);

assert(unit);

if (arg_transport == BUS_TRANSPORT_REMOTE)

return 0;

path = unit_dbus_path_from_name(unit);

if (!path)

return log_oom();

r = sd_bus_get_property(

bus,

"org.freedesktop.systemd1",

path,

unit_dbus_interface_from_name(unit),

"ControlGroup",

&error,

&reply,

"s");

if (r < 0) {

log_error("Failed to query ControlGroup: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_read(reply, "s", &cgroup);

if (r < 0)

return bus_log_parse_error(r);

if (cg_is_empty_recursive(SYSTEMD_CGROUP_CONTROLLER, cgroup) != 0 && leader <= 0)

return 0;

c = columns();

if (c > 18)

c -= 18;

else

c = 0;

show_cgroup_and_extra(SYSTEMD_CGROUP_CONTROLLER, cgroup, "\t\t ", c, false, &leader, leader > 0, get_output_flags());

return 0;

}

static int print_addresses(sd_bus *bus, const char *name, int ifi, const char *prefix, const char *prefix2) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

int r;

assert(bus);

assert(name);

assert(prefix);

assert(prefix2);

r = sd_bus_call_method(bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"GetMachineAddresses",

NULL,

&reply,

"s", name);

if (r < 0)

return r;

r = sd_bus_message_enter_container(reply, 'a', "(iay)");

if (r < 0)

return bus_log_parse_error(r);

while ((r = sd_bus_message_enter_container(reply, 'r', "iay")) > 0) {

int family;

const void *a;

size_t sz;

char buffer[MAX(INET6_ADDRSTRLEN, INET_ADDRSTRLEN)];

r = sd_bus_message_read(reply, "i", &family);

if (r < 0)

return bus_log_parse_error(r);

r = sd_bus_message_read_array(reply, 'y', &a, &sz);

if (r < 0)

return bus_log_parse_error(r);

fputs(prefix, stdout);

fputs(inet_ntop(family, a, buffer, sizeof(buffer)), stdout);

if (family == AF_INET6 && ifi > 0)

printf("%%%i", ifi);

fputc('\n', stdout);

r = sd_bus_message_exit_container(reply);

if (r < 0)

return bus_log_parse_error(r);

if (prefix != prefix2)

prefix = prefix2;

}

if (r < 0)

return bus_log_parse_error(r);

r = sd_bus_message_exit_container(reply);

if (r < 0)

return bus_log_parse_error(r);

return 0;

}

static int print_os_release(sd_bus *bus, const char *name, const char *prefix) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

const char *k, *v, *pretty = NULL;

int r;

assert(bus);

assert(name);

assert(prefix);

r = sd_bus_call_method(bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"GetMachineOSRelease",

NULL,

&reply,

"s", name);

if (r < 0)

return r;

r = sd_bus_message_enter_container(reply, 'a', "{ss}");

if (r < 0)

return bus_log_parse_error(r);

while ((r = sd_bus_message_read(reply, "{ss}", &k, &v)) > 0) {

if (streq(k, "PRETTY_NAME"))

pretty = v;

}

if (r < 0)

return bus_log_parse_error(r);

r = sd_bus_message_exit_container(reply);

if (r < 0)

return bus_log_parse_error(r);

if (pretty)

printf("%s%s\n", prefix, pretty);

return 0;

}

typedef struct MachineStatusInfo {

char *name;

sd_id128_t id;

char *class;

char *service;

char *unit;

char *root_directory;

pid_t leader;

struct dual_timestamp timestamp;

int *netif;

unsigned n_netif;

} MachineStatusInfo;

static void machine_status_info_clear(MachineStatusInfo *info) {

if (info) {

free(info->name);

free(info->class);

free(info->service);

free(info->unit);

free(info->root_directory);

free(info->netif);

zero(*info);

}

}

static void print_machine_status_info(sd_bus *bus, MachineStatusInfo *i) {

char since1[FORMAT_TIMESTAMP_RELATIVE_MAX], *s1;

char since2[FORMAT_TIMESTAMP_MAX], *s2;

int ifi = -1;

assert(bus);

assert(i);

fputs(strna(i->name), stdout);

if (!sd_id128_equal(i->id, SD_ID128_NULL))

printf("(" SD_ID128_FORMAT_STR ")\n", SD_ID128_FORMAT_VAL(i->id));

else

putchar('\n');

s1 = format_timestamp_relative(since1, sizeof(since1), i->timestamp.realtime);

s2 = format_timestamp(since2, sizeof(since2), i->timestamp.realtime);

if (s1)

printf("\t Since: %s; %s\n", s2, s1);

else if (s2)

printf("\t Since: %s\n", s2);

if (i->leader > 0) {

_cleanup_free_ char *t = NULL;

printf("\t Leader: %u", (unsigned) i->leader);

get_process_comm(i->leader, &t);

if (t)

printf(" (%s)", t);

putchar('\n');

}

if (i->service) {

printf("\t Service: %s", i->service);

if (i->class)

printf("; class %s", i->class);

putchar('\n');

} else if (i->class)

printf("\t Class: %s\n", i->class);

if (i->root_directory)

printf("\t Root: %s\n", i->root_directory);

if (i->n_netif > 0) {

unsigned c;

fputs("\t Iface:", stdout);

for (c = 0; c < i->n_netif; c++) {

char name[IF_NAMESIZE+1] = "";

if (if_indextoname(i->netif[c], name)) {

fputc(' ', stdout);

fputs(name, stdout);

if (ifi < 0)

ifi = i->netif[c];

else

ifi = 0;

} else

printf(" %i", i->netif[c]);

}

fputc('\n', stdout);

}

print_addresses(bus, i->name, ifi,

"\t Address: ",

"\t ");

print_os_release(bus, i->name, "\t OS: ");

if (i->unit) {

printf("\t Unit: %s\n", i->unit);

show_unit_cgroup(bus, i->unit, i->leader);

if (arg_transport == BUS_TRANSPORT_LOCAL)

show_journal_by_unit(

stdout,

i->unit,

arg_output,

0,

i->timestamp.monotonic,

arg_lines,

0,

get_output_flags() | OUTPUT_BEGIN_NEWLINE,

SD_JOURNAL_LOCAL_ONLY,

true,

NULL);

}

}

static int map_netif(sd_bus *bus, const char *member, sd_bus_message *m, sd_bus_error *error, void *userdata) {

MachineStatusInfo *i = userdata;

size_t l;

const void *v;

int r;

assert_cc(sizeof(int32_t) == sizeof(int));

r = sd_bus_message_read_array(m, SD_BUS_TYPE_INT32, &v, &l);

if (r < 0)

return r;

if (r == 0)

return -EBADMSG;

i->n_netif = l / sizeof(int32_t);

i->netif = memdup(v, l);

if (!i->netif)

return -ENOMEM;

return 0;

}

static int show_machine_info(const char *verb, sd_bus *bus, const char *path, bool *new_line) {

static const struct bus_properties_map map[] = {

{ "Name", "s", NULL, offsetof(MachineStatusInfo, name) },

{ "Class", "s", NULL, offsetof(MachineStatusInfo, class) },

{ "Service", "s", NULL, offsetof(MachineStatusInfo, service) },

{ "Unit", "s", NULL, offsetof(MachineStatusInfo, unit) },

{ "RootDirectory", "s", NULL, offsetof(MachineStatusInfo, root_directory) },

{ "Leader", "u", NULL, offsetof(MachineStatusInfo, leader) },

{ "Timestamp", "t", NULL, offsetof(MachineStatusInfo, timestamp.realtime) },

{ "TimestampMonotonic", "t", NULL, offsetof(MachineStatusInfo, timestamp.monotonic) },

{ "Id", "ay", bus_map_id128, offsetof(MachineStatusInfo, id) },

{ "NetworkInterfaces", "ai", map_netif, 0 },

{}

};

_cleanup_(machine_status_info_clear) MachineStatusInfo info = {};

int r;

assert(verb);

assert(bus);

assert(path);

assert(new_line);

r = bus_map_all_properties(bus,

"org.freedesktop.machine1",

path,

map,

&info);

if (r < 0)

return log_error_errno(r, "Could not get properties: %m");

if (*new_line)

printf("\n");

*new_line = true;

print_machine_status_info(bus, &info);

return r;

}

static int show_machine_properties(sd_bus *bus, const char *path, bool *new_line) {

int r;

assert(bus);

assert(path);

assert(new_line);

if (*new_line)

printf("\n");

*new_line = true;

r = bus_print_all_properties(bus, "org.freedesktop.machine1", path, arg_property, arg_all);

if (r < 0)

log_error_errno(r, "Could not get properties: %m");

return r;

}

static int show_machine(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

bool properties, new_line = false;

sd_bus *bus = userdata;

int r = 0, i;

assert(bus);

properties = !strstr(argv[0], "status");

pager_open_if_enabled();

if (properties && argc <= 1) {

/* If no argument is specified, inspect the manager

r = show_machine_properties(bus, "/org/freedesktop/machine1", &new_line);

if (r < 0)

return r;

}

for (i = 1; i < argc; i++) {

const char *path = NULL;

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"GetMachine",

&error,

&reply,

"s", argv[i]);

if (r < 0) {

log_error("Could not get path to machine: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_read(reply, "o", &path);

if (r < 0)

return bus_log_parse_error(r);

if (properties)

r = show_machine_properties(bus, path, &new_line);

else

r = show_machine_info(argv[0], bus, path, &new_line);

}

return r;

}

typedef struct ImageStatusInfo {

char *name;

char *path;

char *type;

int read_only;

usec_t crtime;

usec_t mtime;

uint64_t usage;

uint64_t limit;

uint64_t usage_exclusive;

uint64_t limit_exclusive;

} ImageStatusInfo;

static void image_status_info_clear(ImageStatusInfo *info) {

if (info) {

free(info->name);

free(info->path);

free(info->type);

zero(*info);

}

}

static void print_image_status_info(sd_bus *bus, ImageStatusInfo *i) {

char ts_relative[FORMAT_TIMESTAMP_RELATIVE_MAX], *s1;

char ts_absolute[FORMAT_TIMESTAMP_MAX], *s2;

char bs[FORMAT_BYTES_MAX], *s3;

char bs_exclusive[FORMAT_BYTES_MAX], *s4;

assert(bus);

assert(i);

if (i->name) {

fputs(i->name, stdout);

putchar('\n');

}

if (i->type)

printf("\t Type: %s\n", i->type);

if (i->path)

printf("\t Path: %s\n", i->path);

printf("\t RO: %s%s%s\n",

i->read_only ? ansi_highlight_red() : "",

i->read_only ? "read-only" : "writable",

i->read_only ? ansi_normal() : "");

s1 = format_timestamp_relative(ts_relative, sizeof(ts_relative), i->crtime);

s2 = format_timestamp(ts_absolute, sizeof(ts_absolute), i->crtime);

if (s1 && s2)

printf("\t Created: %s; %s\n", s2, s1);

else if (s2)

printf("\t Created: %s\n", s2);

s1 = format_timestamp_relative(ts_relative, sizeof(ts_relative), i->mtime);

s2 = format_timestamp(ts_absolute, sizeof(ts_absolute), i->mtime);

if (s1 && s2)

printf("\tModified: %s; %s\n", s2, s1);

else if (s2)

printf("\tModified: %s\n", s2);

s3 = format_bytes(bs, sizeof(bs), i->usage);

s4 = i->usage_exclusive != i->usage ? format_bytes(bs_exclusive, sizeof(bs_exclusive), i->usage_exclusive) : NULL;

if (s3 && s4)

printf("\t Usage: %s (exclusive: %s)\n", s3, s4);

else if (s3)

printf("\t Usage: %s\n", s3);

s3 = format_bytes(bs, sizeof(bs), i->limit);

s4 = i->limit_exclusive != i->limit ? format_bytes(bs_exclusive, sizeof(bs_exclusive), i->limit_exclusive) : NULL;

if (s3 && s4)

printf("\t Limit: %s (exclusive: %s)\n", s3, s4);

else if (s3)

printf("\t Limit: %s\n", s3);

}

static int show_image_info(sd_bus *bus, const char *path, bool *new_line) {

static const struct bus_properties_map map[] = {

{ "Name", "s", NULL, offsetof(ImageStatusInfo, name) },

{ "Path", "s", NULL, offsetof(ImageStatusInfo, path) },

{ "Type", "s", NULL, offsetof(ImageStatusInfo, type) },

{ "ReadOnly", "b", NULL, offsetof(ImageStatusInfo, read_only) },

{ "CreationTimestamp", "t", NULL, offsetof(ImageStatusInfo, crtime) },

{ "ModificationTimestamp", "t", NULL, offsetof(ImageStatusInfo, mtime) },

{ "Usage", "t", NULL, offsetof(ImageStatusInfo, usage) },

{ "Limit", "t", NULL, offsetof(ImageStatusInfo, limit) },

{ "UsageExclusive", "t", NULL, offsetof(ImageStatusInfo, usage_exclusive) },

{ "LimitExclusive", "t", NULL, offsetof(ImageStatusInfo, limit_exclusive) },

{}

};

_cleanup_(image_status_info_clear) ImageStatusInfo info = {};

int r;

assert(bus);

assert(path);

assert(new_line);

r = bus_map_all_properties(bus,

"org.freedesktop.machine1",

path,

map,

&info);

if (r < 0)

return log_error_errno(r, "Could not get properties: %m");

if (*new_line)

printf("\n");

*new_line = true;

print_image_status_info(bus, &info);

return r;

}

typedef struct PoolStatusInfo {

char *path;

uint64_t usage;

uint64_t limit;

} PoolStatusInfo;

static void pool_status_info_clear(PoolStatusInfo *info) {

if (info) {

free(info->path);

zero(*info);

info->usage = -1;

info->limit = -1;

}

}

static void print_pool_status_info(sd_bus *bus, PoolStatusInfo *i) {

char bs[FORMAT_BYTES_MAX], *s;

if (i->path)

printf("\t Path: %s\n", i->path);

s = format_bytes(bs, sizeof(bs), i->usage);

if (s)

printf("\t Usage: %s\n", s);

s = format_bytes(bs, sizeof(bs), i->limit);

if (s)

printf("\t Limit: %s\n", s);

}

static int show_pool_info(sd_bus *bus) {

static const struct bus_properties_map map[] = {

{ "PoolPath", "s", NULL, offsetof(PoolStatusInfo, path) },

{ "PoolUsage", "t", NULL, offsetof(PoolStatusInfo, usage) },

{ "PoolLimit", "t", NULL, offsetof(PoolStatusInfo, limit) },

{}

};

_cleanup_(pool_status_info_clear) PoolStatusInfo info = {

.usage = (uint64_t) -1,

.limit = (uint64_t) -1,

};

int r;

assert(bus);

r = bus_map_all_properties(bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

map,

&info);

if (r < 0)

return log_error_errno(r, "Could not get properties: %m");

print_pool_status_info(bus, &info);

return 0;

}

static int show_image_properties(sd_bus *bus, const char *path, bool *new_line) {

int r;

assert(bus);

assert(path);

assert(new_line);

if (*new_line)

printf("\n");

*new_line = true;

r = bus_print_all_properties(bus, "org.freedesktop.machine1", path, arg_property, arg_all);

if (r < 0)

log_error_errno(r, "Could not get properties: %m");

return r;

}

static int show_image(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

bool properties, new_line = false;

sd_bus *bus = userdata;

int r = 0, i;

assert(bus);

properties = !strstr(argv[0], "status");

pager_open_if_enabled();

if (argc <= 1) {

/* If no argument is specified, inspect the manager

if (properties)

r = show_image_properties(bus, "/org/freedesktop/machine1", &new_line);

else

r = show_pool_info(bus);

if (r < 0)

return r;

}

for (i = 1; i < argc; i++) {

const char *path = NULL;

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"GetImage",

&error,

&reply,

"s", argv[i]);

if (r < 0) {

log_error("Could not get path to image: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_read(reply, "o", &path);

if (r < 0)

return bus_log_parse_error(r);

if (properties)

r = show_image_properties(bus, path, &new_line);

else

r = show_image_info(bus, path, &new_line);

}

return r;

}

static int kill_machine(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

sd_bus *bus = userdata;

int r, i;

assert(bus);

polkit_agent_open_if_enabled();

if (!arg_kill_who)

arg_kill_who = "all";

for (i = 1; i < argc; i++) {

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"KillMachine",

&error,

NULL,

"ssi", argv[i], arg_kill_who, arg_signal);

if (r < 0) {

log_error("Could not kill machine: %s", bus_error_message(&error, -r));

return r;

}

}

return 0;

}

static int reboot_machine(int argc, char *argv[], void *userdata) {

arg_kill_who = "leader";

arg_signal = SIGINT; /* sysvinit + systemd */

return kill_machine(argc, argv, userdata);

}

static int poweroff_machine(int argc, char *argv[], void *userdata) {

arg_kill_who = "leader";

arg_signal = SIGRTMIN+4; /* only systemd */

return kill_machine(argc, argv, userdata);

}

static int terminate_machine(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

sd_bus *bus = userdata;

int r, i;

assert(bus);

polkit_agent_open_if_enabled();

for (i = 1; i < argc; i++) {

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"TerminateMachine",

&error,

NULL,

"s", argv[i]);

if (r < 0) {

log_error("Could not terminate machine: %s", bus_error_message(&error, -r));

return r;

}

}

return 0;

}

static int copy_files(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_free_ char *abs_host_path = NULL;

char *dest, *host_path, *container_path;

sd_bus *bus = userdata;

bool copy_from;

int r;

assert(bus);

polkit_agent_open_if_enabled();

copy_from = streq(argv[0], "copy-from");

dest = argv[3] ?: argv[2];

host_path = copy_from ? dest : argv[2];

container_path = copy_from ? argv[2] : dest;

if (!path_is_absolute(host_path)) {

r = path_make_absolute_cwd(host_path, &abs_host_path);

if (r < 0)

return log_error_errno(r, "Failed to make path absolute: %m");

host_path = abs_host_path;

}

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

copy_from ? "CopyFromMachine" : "CopyToMachine",

&error,

NULL,

"sss",

argv[1],

copy_from ? container_path : host_path,

copy_from ? host_path : container_path);

if (r < 0)

return log_error_errno(r, "Failed to copy: %s", bus_error_message(&error, r));

return 0;

}

static int bind_mount(int argc, char *argv[], void *userdata) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

sd_bus *bus = userdata;

int r;

assert(bus);

polkit_agent_open_if_enabled();

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"BindMountMachine",

&error,

NULL,

"sssbb",

argv[1],

argv[2],

argv[3],

arg_read_only,

arg_mkdir);

if (r < 0) {

log_error("Failed to bind mount: %s", bus_error_message(&error, -r));

return r;

}

return 0;

}

static int on_machine_removed(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) {

PTYForward ** forward = (PTYForward**) userdata;

int r;

assert(m);

assert(forward);

if (*forward) {

/* If the forwarder is already initialized, tell it to

r = pty_forward_set_ignore_vhangup(*forward, false);

if (r >= 0)

return 0;

log_error_errno(r, "Failed to set ignore_vhangup flag: %m");

}

/* On error, or when the forwarder is not initialized yet, quit immediately */

sd_event_exit(sd_bus_get_event(sd_bus_message_get_bus(m)), EXIT_FAILURE);

return 0;

}

static int process_forward(sd_event *event, PTYForward **forward, int master, PTYForwardFlags flags, const char *name) {

char last_char = 0;

bool machine_died;

int ret = 0, r;

assert(event);

assert(master >= 0);

assert(name);

assert_se(sigprocmask_many(SIG_BLOCK, NULL, SIGWINCH, SIGTERM, SIGINT, -1) >= 0);

if (streq(name, ".host"))

log_info("Connected to the local host. Press ^] three times within 1s to exit session.");

else

log_info("Connected to machine %s. Press ^] three times within 1s to exit session.", name);

sd_event_add_signal(event, NULL, SIGINT, NULL, NULL);

sd_event_add_signal(event, NULL, SIGTERM, NULL, NULL);

r = pty_forward_new(event, master, flags, forward);

if (r < 0)

return log_error_errno(r, "Failed to create PTY forwarder: %m");

r = sd_event_loop(event);

if (r < 0)

return log_error_errno(r, "Failed to run event loop: %m");

pty_forward_get_last_char(*forward, &last_char);

machine_died =

(flags & PTY_FORWARD_IGNORE_VHANGUP) &&

pty_forward_get_ignore_vhangup(*forward) == 0;

*forward = pty_forward_free(*forward);

if (last_char != '\n')

fputc('\n', stdout);

if (machine_died)

log_info("Machine %s terminated.", name);

else if (streq(name, ".host"))

log_info("Connection to the local host terminated.");

else

log_info("Connection to machine %s terminated.", name);

sd_event_get_exit_code(event, &ret);

return ret;

}

static int login_machine(int argc, char *argv[], void *userdata) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_(pty_forward_freep) PTYForward *forward = NULL;

_cleanup_bus_slot_unref_ sd_bus_slot *slot = NULL;

_cleanup_event_unref_ sd_event *event = NULL;

int master = -1, r;

sd_bus *bus = userdata;

const char *pty, *match, *machine;

assert(bus);

if (!strv_isempty(arg_setenv) || arg_uid) {

log_error("--setenv= and --uid= are not supported for 'login'. Use 'shell' instead.");

return -EINVAL;

}

if (arg_transport != BUS_TRANSPORT_LOCAL &&

arg_transport != BUS_TRANSPORT_MACHINE) {

log_error("Login only supported on local machines.");

return -EOPNOTSUPP;

}

polkit_agent_open_if_enabled();

r = sd_event_default(&event);

if (r < 0)

return log_error_errno(r, "Failed to get event loop: %m");

r = sd_bus_attach_event(bus, event, 0);

if (r < 0)

return log_error_errno(r, "Failed to attach bus to event loop: %m");

machine = argc < 2 || isempty(argv[1]) ? ".host" : argv[1];

match = strjoina("type='signal',"

"sender='org.freedesktop.machine1',"

"path='/org/freedesktop/machine1',",

"interface='org.freedesktop.machine1.Manager',"

"member='MachineRemoved',"

"arg0='", machine, "'");

r = sd_bus_add_match(bus, &slot, match, on_machine_removed, &forward);

if (r < 0)

return log_error_errno(r, "Failed to add machine removal match: %m");

r = sd_bus_call_method(

bus,

"org.freedesktop.machine1",

"/org/freedesktop/machine1",

"org.freedesktop.machine1.Manager",

"OpenMachineLogin",

&error,

&reply,

"s", machine);

if (r < 0) {

log_error("Failed to get login PTY: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_read(reply, "hs", &master, &pty);

if (r < 0)

return bus_log_parse_error(r);

return process_forward(event, &forward, master, PTY_FORWARD_IGNORE_VHANGUP, machine);

}

static int shell_machine(int argc, char *argv[], void *userdata) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL, *m = NULL;

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

_cleanup_(pty_forward_freep) PTYForward *forward = NULL;

_cleanup_bus_slot_unref_ sd_bus_slot *slot = NULL;

_cleanup_event_unref_ sd_event *event = NULL;

int master = -1, r;

sd_bus *bus = userdata;

const char *pty, *match, *machine, *path, *uid = NULL;

assert(bus);

if (arg_transport != BUS_TRANSPORT_LOCAL &&

arg_transport != BUS_TRANSPORT_MACHINE) {

log_error("Shell only supported on local machines.");

return -EOPNOTSUPP;

}