analyze: extend the dump command to accept patterns
The new function DumpPatterns() can be used to limit (drastically) the size of the data returned by PID1. Hence the optimization of serializing data into a file descriptor should be less relevant than having the possibility to limit the data when communicating with the service manager remotely. NB: when passing patterns, the dump command omits the version of the manager as well as the features and the timestamps.
This commit is contained in:
+26
-4
@@ -1344,7 +1344,13 @@ static int method_unsubscribe(sd_bus_message *message, void *userdata, sd_bus_er
|
||||
return sd_bus_reply_method_return(message, NULL);
|
||||
}
|
||||
|
||||
static int dump_impl(sd_bus_message *message, void *userdata, sd_bus_error *error, int (*reply)(sd_bus_message *, char *)) {
|
||||
static int dump_impl(
|
||||
sd_bus_message *message,
|
||||
void *userdata,
|
||||
sd_bus_error *error,
|
||||
char **patterns,
|
||||
int (*reply)(sd_bus_message *, char *)) {
|
||||
|
||||
_cleanup_free_ char *dump = NULL;
|
||||
Manager *m = ASSERT_PTR(userdata);
|
||||
int r;
|
||||
@@ -1357,7 +1363,7 @@ static int dump_impl(sd_bus_message *message, void *userdata, sd_bus_error *erro
|
||||
if (r < 0)
|
||||
return r;
|
||||
|
||||
r = manager_get_dump_string(m, &dump);
|
||||
r = manager_get_dump_string(m, patterns, &dump);
|
||||
if (r < 0)
|
||||
return r;
|
||||
|
||||
@@ -1369,7 +1375,7 @@ static int reply_dump(sd_bus_message *message, char *dump) {
|
||||
}
|
||||
|
||||
static int method_dump(sd_bus_message *message, void *userdata, sd_bus_error *error) {
|
||||
return dump_impl(message, userdata, error, reply_dump);
|
||||
return dump_impl(message, userdata, error, NULL, reply_dump);
|
||||
}
|
||||
|
||||
static int reply_dump_by_fd(sd_bus_message *message, char *dump) {
|
||||
@@ -1383,7 +1389,18 @@ static int reply_dump_by_fd(sd_bus_message *message, char *dump) {
|
||||
}
|
||||
|
||||
static int method_dump_by_fd(sd_bus_message *message, void *userdata, sd_bus_error *error) {
|
||||
return dump_impl(message, userdata, error, reply_dump_by_fd);
|
||||
return dump_impl(message, userdata, error, NULL, reply_dump_by_fd);
|
||||
}
|
||||
|
||||
static int method_dump_patterns(sd_bus_message *message, void *userdata, sd_bus_error *error) {
|
||||
_cleanup_strv_free_ char **patterns = NULL;
|
||||
int r;
|
||||
|
||||
r = sd_bus_message_read_strv(message, &patterns);
|
||||
if (r < 0)
|
||||
return r;
|
||||
|
||||
return dump_impl(message, userdata, error, patterns, reply_dump);
|
||||
}
|
||||
|
||||
static int method_refuse_snapshot(sd_bus_message *message, void *userdata, sd_bus_error *error) {
|
||||
@@ -3010,6 +3027,11 @@ const sd_bus_vtable bus_manager_vtable[] = {
|
||||
SD_BUS_RESULT("s", output),
|
||||
method_dump,
|
||||
SD_BUS_VTABLE_UNPRIVILEGED),
|
||||
SD_BUS_METHOD_WITH_ARGS("DumpPatterns",
|
||||
SD_BUS_ARGS("as", patterns),
|
||||
SD_BUS_RESULT("s", output),
|
||||
method_dump_patterns,
|
||||
SD_BUS_VTABLE_UNPRIVILEGED),
|
||||
SD_BUS_METHOD_WITH_ARGS("DumpByFileDescriptor",
|
||||
SD_BUS_NO_ARGS,
|
||||
SD_BUS_RESULT("h", fd),
|
||||
|
||||
+1
-1
@@ -919,7 +919,7 @@ static int ask_for_confirmation(const ExecContext *context, const char *vc, Unit
|
||||
u->id, u->description, cmdline);
|
||||
continue; /* ask again */
|
||||
case 'j':
|
||||
manager_dump_jobs(u->manager, stdout, " ");
|
||||
manager_dump_jobs(u->manager, stdout, /* patterns= */ NULL, " ");
|
||||
continue; /* ask again */
|
||||
case 'n':
|
||||
/* 'n' was removed in favor of 'f'. */
|
||||
|
||||
@@ -83,7 +83,7 @@ int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
|
||||
assert_se(g);
|
||||
|
||||
unit_dump(u, g, "");
|
||||
manager_dump(m, g, ">>>");
|
||||
manager_dump(m, g, /* patterns= */ NULL, ">>>");
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
+35
-16
@@ -7,31 +7,40 @@
|
||||
#include "manager-dump.h"
|
||||
#include "unit-serialize.h"
|
||||
|
||||
void manager_dump_jobs(Manager *s, FILE *f, const char *prefix) {
|
||||
void manager_dump_jobs(Manager *s, FILE *f, char **patterns, const char *prefix) {
|
||||
Job *j;
|
||||
|
||||
assert(s);
|
||||
assert(f);
|
||||
|
||||
HASHMAP_FOREACH(j, s->jobs)
|
||||
HASHMAP_FOREACH(j, s->jobs) {
|
||||
|
||||
if (!strv_fnmatch_or_empty(patterns, j->unit->id, FNM_NOESCAPE))
|
||||
continue;
|
||||
|
||||
job_dump(j, f, prefix);
|
||||
}
|
||||
}
|
||||
|
||||
void manager_dump_units(Manager *s, FILE *f, const char *prefix) {
|
||||
void manager_dump_units(Manager *s, FILE *f, char **patterns, const char *prefix) {
|
||||
Unit *u;
|
||||
const char *t;
|
||||
|
||||
assert(s);
|
||||
assert(f);
|
||||
|
||||
HASHMAP_FOREACH_KEY(u, t, s->units)
|
||||
if (u->id == t)
|
||||
unit_dump(u, f, prefix);
|
||||
HASHMAP_FOREACH_KEY(u, t, s->units) {
|
||||
if (u->id != t)
|
||||
continue;
|
||||
|
||||
if (!strv_fnmatch_or_empty(patterns, u->id, FNM_NOESCAPE))
|
||||
continue;
|
||||
|
||||
unit_dump(u, f, prefix);
|
||||
}
|
||||
}
|
||||
|
||||
void manager_dump(Manager *m, FILE *f, const char *prefix) {
|
||||
assert(m);
|
||||
assert(f);
|
||||
static void manager_dump_header(Manager *m, FILE *f, const char *prefix) {
|
||||
|
||||
/* NB: this is a debug interface for developers. It's not supposed to be machine readable or be
|
||||
* stable between versions. We take the liberty to restructure it entirely between versions and
|
||||
@@ -50,12 +59,22 @@ void manager_dump(Manager *m, FILE *f, const char *prefix) {
|
||||
timestamp_is_set(t->realtime) ? FORMAT_TIMESTAMP(t->realtime) :
|
||||
FORMAT_TIMESPAN(t->monotonic, 1));
|
||||
}
|
||||
|
||||
manager_dump_units(m, f, prefix);
|
||||
manager_dump_jobs(m, f, prefix);
|
||||
}
|
||||
|
||||
int manager_get_dump_string(Manager *m, char **ret) {
|
||||
void manager_dump(Manager *m, FILE *f, char **patterns, const char *prefix) {
|
||||
assert(m);
|
||||
assert(f);
|
||||
|
||||
/* If no pattern is provided, dump the full manager state including the manager version, features and
|
||||
* so on. Otherwise limit the dump to the units/jobs matching the specified patterns. */
|
||||
if (!patterns)
|
||||
manager_dump_header(m, f, prefix);
|
||||
|
||||
manager_dump_units(m, f, patterns, prefix);
|
||||
manager_dump_jobs(m, f, patterns, prefix);
|
||||
}
|
||||
|
||||
int manager_get_dump_string(Manager *m, char **patterns, char **ret) {
|
||||
_cleanup_free_ char *dump = NULL;
|
||||
_cleanup_fclose_ FILE *f = NULL;
|
||||
size_t size;
|
||||
@@ -68,7 +87,7 @@ int manager_get_dump_string(Manager *m, char **ret) {
|
||||
if (!f)
|
||||
return -errno;
|
||||
|
||||
manager_dump(m, f, NULL);
|
||||
manager_dump(m, f, patterns, NULL);
|
||||
|
||||
r = fflush_and_check(f);
|
||||
if (r < 0)
|
||||
@@ -85,8 +104,8 @@ void manager_test_summary(Manager *m) {
|
||||
assert(m);
|
||||
|
||||
printf("-> By units:\n");
|
||||
manager_dump_units(m, stdout, "\t");
|
||||
manager_dump_units(m, stdout, /* patterns= */ NULL, "\t");
|
||||
|
||||
printf("-> By jobs:\n");
|
||||
manager_dump_jobs(m, stdout, "\t");
|
||||
manager_dump_jobs(m, stdout, /* patterns= */ NULL, "\t");
|
||||
}
|
||||
|
||||
@@ -5,8 +5,8 @@
|
||||
|
||||
#include "manager.h"
|
||||
|
||||
void manager_dump_jobs(Manager *s, FILE *f, const char *prefix);
|
||||
void manager_dump_units(Manager *s, FILE *f, const char *prefix);
|
||||
void manager_dump(Manager *s, FILE *f, const char *prefix);
|
||||
int manager_get_dump_string(Manager *m, char **ret);
|
||||
void manager_dump_jobs(Manager *s, FILE *f, char **patterns, const char *prefix);
|
||||
void manager_dump_units(Manager *s, FILE *f, char **patterns, const char *prefix);
|
||||
void manager_dump(Manager *s, FILE *f, char **patterns, const char *prefix);
|
||||
int manager_get_dump_string(Manager *m, char **patterns, char **ret);
|
||||
void manager_test_summary(Manager *m);
|
||||
|
||||
+1
-1
@@ -2823,7 +2823,7 @@ static int manager_dispatch_signal_fd(sd_event_source *source, int fd, uint32_t
|
||||
case SIGUSR2: {
|
||||
_cleanup_free_ char *dump = NULL;
|
||||
|
||||
r = manager_get_dump_string(m, &dump);
|
||||
r = manager_get_dump_string(m, /* patterns= */ NULL, &dump);
|
||||
if (r < 0) {
|
||||
log_warning_errno(errno, "Failed to acquire manager dump: %m");
|
||||
break;
|
||||
|
||||
@@ -120,6 +120,10 @@
|
||||
send_interface="org.freedesktop.systemd1.Manager"
|
||||
send_member="DumpByFileDescriptor"/>
|
||||
|
||||
<allow send_destination="org.freedesktop.systemd1"
|
||||
send_interface="org.freedesktop.systemd1.Manager"
|
||||
send_member="DumpPatterns"/>
|
||||
|
||||
<allow send_destination="org.freedesktop.systemd1"
|
||||
send_interface="org.freedesktop.systemd1.Manager"
|
||||
send_member="ListUnitFiles"/>
|
||||
|
||||
Reference in New Issue
Block a user