1
0
mirror of https://github.com/systemd/systemd synced 2026-03-12 16:14:47 +01:00

Compare commits

..

No commits in common. "5bc11cd4e69f75fd72492cb213db5b4f0534462f" and "5b501abfc29e083dfa62c869439cd9c808634edd" have entirely different histories.

22 changed files with 619 additions and 892 deletions

View File

@ -22,7 +22,6 @@
#include "pretty-print.h" #include "pretty-print.h"
#include "string-util.h" #include "string-util.h"
#include "tpm2-util.h" #include "tpm2-util.h"
#include "varlink-util.h"
static int status_entries( static int status_entries(
const BootConfig *config, const BootConfig *config,
@ -675,21 +674,23 @@ int vl_method_list_boot_entries(sd_varlink *link, sd_json_variant *parameters, s
if (r < 0) if (r < 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.BootControl.NoSuchBootEntry"); _cleanup_(sd_json_variant_unrefp) sd_json_variant *previous = NULL;
if (r < 0)
return r;
for (size_t i = 0; i < config.n_entries; i++) { for (size_t i = 0; i < config.n_entries; i++) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (previous) {
r = sd_varlink_notifybo(link, SD_JSON_BUILD_PAIR_VARIANT("entry", previous));
if (r < 0)
return r;
r = boot_entry_to_json(&config, i, &v); previous = sd_json_variant_unref(previous);
if (r < 0) }
return r;
r = sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_VARIANT("entry", v)); r = boot_entry_to_json(&config, i, &previous);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!previous)
return sd_varlink_error(link, "io.systemd.BootControl.NoSuchBootEntry", NULL);
return sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_VARIANT("entry", previous));
} }

View File

@ -10,7 +10,6 @@
#include "uid-classification.h" #include "uid-classification.h"
#include "user-util.h" #include "user-util.h"
#include "varlink-dynamic-user.h" #include "varlink-dynamic-user.h"
#include "varlink-util.h"
typedef struct LookupParameters { typedef struct LookupParameters {
const char *user_name; const char *user_name;
@ -60,6 +59,7 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
LookupParameters p = { LookupParameters p = {
.uid = UID_INVALID, .uid = UID_INVALID,
}; };
@ -78,10 +78,6 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
if (!streq_ptr(p.service, "io.systemd.DynamicUser")) if (!streq_ptr(p.service, "io.systemd.DynamicUser"))
return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL);
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (uid_is_valid(p.uid)) if (uid_is_valid(p.uid))
r = dynamic_user_lookup_uid(m, p.uid, &found_name); r = dynamic_user_lookup_uid(m, p.uid, &found_name);
else if (p.user_name) else if (p.user_name)
@ -102,20 +98,26 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
if (!user_match_lookup_parameters(&p, d->name, uid)) if (!user_match_lookup_parameters(&p, d->name, uid))
continue; continue;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (v) {
r = build_user_json(d->name, uid, &v); r = sd_varlink_notify(link, v);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v); v = sd_json_variant_unref(v);
}
r = build_user_json(d->name, uid, &v);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!v)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, v);
} }
if (r == -ESRCH) if (r == -ESRCH)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r < 0) if (r < 0)
return r; return r;
@ -125,7 +127,6 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
if (!user_match_lookup_parameters(&p, un, uid)) if (!user_match_lookup_parameters(&p, un, uid))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_user_json(un, uid, &v); r = build_user_json(un, uid, &v);
if (r < 0) if (r < 0)
return r; return r;
@ -167,6 +168,7 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
LookupParameters p = { LookupParameters p = {
.gid = GID_INVALID, .gid = GID_INVALID,
}; };
@ -182,10 +184,6 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
if (r != 0) if (r != 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (!streq_ptr(p.service, "io.systemd.DynamicUser")) if (!streq_ptr(p.service, "io.systemd.DynamicUser"))
return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL);
@ -211,20 +209,26 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
if (!group_match_lookup_parameters(&p, d->name, (gid_t) uid)) if (!group_match_lookup_parameters(&p, d->name, (gid_t) uid))
continue; continue;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (v) {
r = build_group_json(d->name, (gid_t) uid, &v); r = sd_varlink_notify(link, v);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v); v = sd_json_variant_unref(v);
}
r = build_group_json(d->name, (gid_t) uid, &v);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!v)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, v);
} }
if (r == -ESRCH) if (r == -ESRCH)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r < 0) if (r < 0)
return r; return r;
@ -234,7 +238,6 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
if (!group_match_lookup_parameters(&p, gn, gid)) if (!group_match_lookup_parameters(&p, gn, gid))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_group_json(gn, gid, &v); r = build_group_json(gn, gid, &v);
if (r < 0) if (r < 0)
return r; return r;

View File

@ -330,10 +330,6 @@ int vl_method_enqueue_marked_jobs_manager(sd_varlink *link, sd_json_variant *par
if (r <= 0) if (r <= 0)
return r; return r;
r = varlink_set_sentinel(link, NULL);
if (r < 0)
return r;
log_info("Queuing reload/restart jobs for marked units%s", glyph(GLYPH_ELLIPSIS)); log_info("Queuing reload/restart jobs for marked units%s", glyph(GLYPH_ELLIPSIS));
Unit *u; Unit *u;
@ -377,17 +373,20 @@ int vl_method_enqueue_marked_jobs_manager(sd_varlink *link, sd_json_variant *par
const char *error_msg = bus_error.message ?: error_id ? NULL : STRERROR(r); const char *error_msg = bus_error.message ?: error_id ? NULL : STRERROR(r);
r = sd_varlink_replybo(link, r = sd_varlink_notifybo(link,
SD_JSON_BUILD_PAIR_STRING("unitID", u->id), SD_JSON_BUILD_PAIR_STRING("unitID", u->id),
JSON_BUILD_PAIR_STRING_NON_EMPTY("error", error_id), JSON_BUILD_PAIR_STRING_NON_EMPTY("error", error_id),
JSON_BUILD_PAIR_STRING_NON_EMPTY("errorMessage", error_msg)); JSON_BUILD_PAIR_STRING_NON_EMPTY("errorMessage", error_msg));
} else } else
r = sd_varlink_replybo(link, r = sd_varlink_notifybo(link,
SD_JSON_BUILD_PAIR_STRING("unitID", u->id), SD_JSON_BUILD_PAIR_STRING("unitID", u->id),
SD_JSON_BUILD_PAIR_INTEGER("jobID", job_id)); SD_JSON_BUILD_PAIR_INTEGER("jobID", job_id));
if (r < 0) if (r < 0)
return r; return r;
} }
return ret; if (ret < 0)
return ret;
return sd_varlink_reply(link, NULL);
} }

View File

@ -25,7 +25,7 @@ static int unit_active_state_build_json(MetricFamilyContext *context, void *user
context, context,
unit->id, unit->id,
unit_active_state_to_string(unit_active_state(unit)), unit_active_state_to_string(unit_active_state(unit)),
/* fields= */ NULL); /* field_pairs= */ NULL);
if (r < 0) if (r < 0)
return r; return r;
} }
@ -50,7 +50,7 @@ static int unit_load_state_build_json(MetricFamilyContext *context, void *userda
context, context,
unit->id, unit->id,
unit_load_state_to_string(unit->load_state), unit_load_state_to_string(unit->load_state),
/* fields= */ NULL); /* field_pairs= */ NULL);
if (r < 0) if (r < 0)
return r; return r;
} }
@ -66,7 +66,7 @@ static int nrestarts_build_json(MetricFamilyContext *context, void *userdata) {
LIST_FOREACH(units_by_type, unit, manager->units_by_type[UNIT_SERVICE]) { LIST_FOREACH(units_by_type, unit, manager->units_by_type[UNIT_SERVICE]) {
r = metric_build_send_unsigned( r = metric_build_send_unsigned(
context, unit->id, SERVICE(unit)->n_restarts, /* fields= */ NULL); context, unit->id, SERVICE(unit)->n_restarts, /* field_pairs= */ NULL);
if (r < 0) if (r < 0)
return r; return r;
} }
@ -81,21 +81,16 @@ static int units_by_type_total_build_json(MetricFamilyContext *context, void *us
assert(context); assert(context);
for (UnitType type = 0; type < _UNIT_TYPE_MAX; type++) { for (UnitType type = 0; type < _UNIT_TYPE_MAX; type++) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *fields = NULL;
uint64_t counter = 0; uint64_t counter = 0;
LIST_FOREACH(units_by_type, _u, manager->units_by_type[type]) LIST_FOREACH(units_by_type, _u, manager->units_by_type[type])
counter++; counter++;
r = sd_json_buildo(&fields, SD_JSON_BUILD_PAIR_STRING("type", unit_type_to_string(type)));
if (r < 0)
return r;
r = metric_build_send_unsigned( r = metric_build_send_unsigned(
context, context,
/* object= */ NULL, /* object= */ NULL,
counter, counter,
fields); STRV_MAKE("type", unit_type_to_string(type)));
if (r < 0) if (r < 0)
return r; return r;
} }
@ -122,17 +117,11 @@ static int units_by_state_total_build_json(MetricFamilyContext *context, void *u
} }
for (UnitActiveState state = 0; state < _UNIT_ACTIVE_STATE_MAX; state++) { for (UnitActiveState state = 0; state < _UNIT_ACTIVE_STATE_MAX; state++) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *fields = NULL;
r = sd_json_buildo(&fields, SD_JSON_BUILD_PAIR_STRING("state", unit_active_state_to_string(state)));
if (r < 0)
return r;
r = metric_build_send_unsigned( r = metric_build_send_unsigned(
context, context,
/* object= */ NULL, /* object= */ NULL,
counters[state], counters[state],
fields); STRV_MAKE("state", unit_active_state_to_string(state)));
if (r < 0) if (r < 0)
return r; return r;
} }
@ -146,31 +135,31 @@ const MetricFamily metric_family_table[] = {
.name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "nrestarts", .name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "nrestarts",
.description = "Per unit metric: number of restarts", .description = "Per unit metric: number of restarts",
.type = METRIC_FAMILY_TYPE_COUNTER, .type = METRIC_FAMILY_TYPE_COUNTER,
.generate = nrestarts_build_json, .generate_cb = nrestarts_build_json,
}, },
{ {
.name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "unit_active_state", .name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "unit_active_state",
.description = "Per unit metric: active state", .description = "Per unit metric: active state",
.type = METRIC_FAMILY_TYPE_STRING, .type = METRIC_FAMILY_TYPE_STRING,
.generate = unit_active_state_build_json, .generate_cb = unit_active_state_build_json,
}, },
{ {
.name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "unit_load_state", .name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "unit_load_state",
.description = "Per unit metric: load state", .description = "Per unit metric: load state",
.type = METRIC_FAMILY_TYPE_STRING, .type = METRIC_FAMILY_TYPE_STRING,
.generate = unit_load_state_build_json, .generate_cb = unit_load_state_build_json,
}, },
{ {
.name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "units_by_state_total", .name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "units_by_state_total",
.description = "Total number of units of different state", .description = "Total number of units of different state",
.type = METRIC_FAMILY_TYPE_GAUGE, .type = METRIC_FAMILY_TYPE_GAUGE,
.generate = units_by_state_total_build_json, .generate_cb = units_by_state_total_build_json,
}, },
{ {
.name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "units_by_type_total", .name = METRIC_IO_SYSTEMD_MANAGER_PREFIX "units_by_type_total",
.description = "Total number of units of different types", .description = "Total number of units of different types",
.type = METRIC_FAMILY_TYPE_GAUGE, .type = METRIC_FAMILY_TYPE_GAUGE,
.generate = units_by_type_total_build_json, .generate_cb = units_by_type_total_build_json,
}, },
{} {}
}; };

View File

@ -311,16 +311,27 @@ static int unit_runtime_build_json(sd_json_variant **ret, const char *name, void
JSON_BUILD_PAIR_CALLBACK_NON_NULL("CGroup", unit_cgroup_runtime_build_json, u)); JSON_BUILD_PAIR_CALLBACK_NON_NULL("CGroup", unit_cgroup_runtime_build_json, u));
} }
static int list_unit_one(sd_varlink *link, Unit *unit) { static int list_unit_one(sd_varlink *link, Unit *unit, bool more) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
int r;
assert(link); assert(link);
assert(unit); assert(unit);
return sd_varlink_replybo(link, r = sd_json_buildo(
SD_JSON_BUILD_PAIR_CALLBACK("context", unit_context_build_json, unit), &v,
SD_JSON_BUILD_PAIR_CALLBACK("runtime", unit_runtime_build_json, unit)); SD_JSON_BUILD_PAIR_CALLBACK("context", unit_context_build_json, unit),
SD_JSON_BUILD_PAIR_CALLBACK("runtime", unit_runtime_build_json, unit));
if (r < 0)
return r;
if (more)
return sd_varlink_notify(link, v);
return sd_varlink_reply(link, v);
} }
static int list_unit_one_with_selinux_access_check(sd_varlink *link, Unit *unit) { static int list_unit_one_with_selinux_access_check(sd_varlink *link, Unit *unit, bool more) {
int r; int r;
assert(link); assert(link);
@ -332,7 +343,7 @@ static int list_unit_one_with_selinux_access_check(sd_varlink *link, Unit *unit)
* it means that SELinux enforce is on. It also does all the logging(). */ * it means that SELinux enforce is on. It also does all the logging(). */
return sd_varlink_error(link, SD_VARLINK_ERROR_PERMISSION_DENIED, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_PERMISSION_DENIED, NULL);
return list_unit_one(link, unit); return list_unit_one(link, unit, more);
} }
static int lookup_unit_by_pidref(sd_varlink *link, Manager *manager, PidRef *pidref, Unit **ret_unit) { static int lookup_unit_by_pidref(sd_varlink *link, Manager *manager, PidRef *pidref, Unit **ret_unit) {
@ -384,12 +395,7 @@ static int varlink_error_conflict_lookup_parameters(sd_varlink *v, const UnitLoo
return varlink_error_no_such_unit(v, /* name= */ NULL); return varlink_error_no_such_unit(v, /* name= */ NULL);
} }
static int lookup_unit_by_parameters( static int lookup_unit_by_parameters(sd_varlink *link, Manager *manager, UnitLookupParameters *p, Unit **ret_unit) {
sd_varlink *link,
Manager *manager,
UnitLookupParameters *p,
Unit **ret) {
/* The function can return ret_unit=NULL if no lookup parameters provided */ /* The function can return ret_unit=NULL if no lookup parameters provided */
Unit *unit = NULL; Unit *unit = NULL;
int r; int r;
@ -397,7 +403,7 @@ static int lookup_unit_by_parameters(
assert(link); assert(link);
assert(manager); assert(manager);
assert(p); assert(p);
assert(ret); assert(ret_unit);
if (p->name) { if (p->name) {
unit = manager_get_unit(manager, p->name); unit = manager_get_unit(manager, p->name);
@ -407,7 +413,6 @@ static int lookup_unit_by_parameters(
if (pidref_is_set_or_automatic(&p->pidref)) { if (pidref_is_set_or_automatic(&p->pidref)) {
Unit *pid_unit; Unit *pid_unit;
r = lookup_unit_by_pidref(link, manager, &p->pidref, &pid_unit); r = lookup_unit_by_pidref(link, manager, &p->pidref, &pid_unit);
if (r == -EINVAL) if (r == -EINVAL)
return sd_varlink_error_invalid_parameter_name(link, "pid"); return sd_varlink_error_invalid_parameter_name(link, "pid");
@ -415,7 +420,7 @@ static int lookup_unit_by_parameters(
return varlink_error_no_such_unit(link, "pid"); return varlink_error_no_such_unit(link, "pid");
if (r < 0) if (r < 0)
return r; return r;
if (unit && pid_unit != unit) if (pid_unit != unit && unit != NULL)
return varlink_error_conflict_lookup_parameters(link, p); return varlink_error_conflict_lookup_parameters(link, p);
unit = pid_unit; unit = pid_unit;
@ -428,7 +433,7 @@ static int lookup_unit_by_parameters(
Unit *cgroup_unit = manager_get_unit_by_cgroup(manager, p->cgroup); Unit *cgroup_unit = manager_get_unit_by_cgroup(manager, p->cgroup);
if (!cgroup_unit) if (!cgroup_unit)
return varlink_error_no_such_unit(link, "cgroup"); return varlink_error_no_such_unit(link, "cgroup");
if (unit && cgroup_unit != unit) if (cgroup_unit != unit && unit != NULL)
return varlink_error_conflict_lookup_parameters(link, p); return varlink_error_conflict_lookup_parameters(link, p);
unit = cgroup_unit; unit = cgroup_unit;
@ -438,14 +443,14 @@ static int lookup_unit_by_parameters(
Unit *id128_unit = hashmap_get(manager->units_by_invocation_id, &p->invocation_id); Unit *id128_unit = hashmap_get(manager->units_by_invocation_id, &p->invocation_id);
if (!id128_unit) if (!id128_unit)
return varlink_error_no_such_unit(link, "invocationID"); return varlink_error_no_such_unit(link, "invocationID");
if (unit && id128_unit != unit) if (id128_unit != unit && unit != NULL)
return varlink_error_conflict_lookup_parameters(link, p); return varlink_error_conflict_lookup_parameters(link, p);
unit = id128_unit; unit = id128_unit;
} }
*ret = unit; *ret_unit = unit;
return !!unit; return 0;
} }
int vl_method_list_units(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) { int vl_method_list_units(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
@ -461,7 +466,7 @@ int vl_method_list_units(sd_varlink *link, sd_json_variant *parameters, sd_varli
_cleanup_(unit_lookup_parameters_done) UnitLookupParameters p = { _cleanup_(unit_lookup_parameters_done) UnitLookupParameters p = {
.pidref = PIDREF_NULL, .pidref = PIDREF_NULL,
}; };
Unit *unit; Unit *unit, *previous = NULL;
const char *k; const char *k;
int r; int r;
@ -475,27 +480,30 @@ int vl_method_list_units(sd_varlink *link, sd_json_variant *parameters, sd_varli
r = lookup_unit_by_parameters(link, manager, &p, &unit); r = lookup_unit_by_parameters(link, manager, &p, &unit);
if (r < 0) if (r < 0)
return r; return r;
if (r > 0) if (unit)
return list_unit_one_with_selinux_access_check(link, unit); return list_unit_one_with_selinux_access_check(link, unit, /* more= */ false);
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL);
r = varlink_set_sentinel(link, "io.systemd.Manager.NoSuchUnit");
if (r < 0)
return r;
HASHMAP_FOREACH_KEY(unit, k, manager->units) { HASHMAP_FOREACH_KEY(unit, k, manager->units) {
/* ignore aliases */ /* ignore aliases */
if (k != unit->id) if (k != unit->id)
continue; continue;
r = list_unit_one(link, unit); if (previous) {
if (r < 0) r = list_unit_one(link, previous, /* more= */ true);
return r; if (r < 0)
return r;
}
previous = unit;
} }
return 0; if (previous)
return list_unit_one(link, previous, /* more= */ false);
return sd_varlink_error(link, "io.systemd.Manager.NoSuchUnit", NULL);
} }
int varlink_unit_queue_job_one( int varlink_unit_queue_job_one(

View File

@ -14,7 +14,6 @@
#include "user-record.h" #include "user-record.h"
#include "user-record-util.h" #include "user-record-util.h"
#include "user-util.h" #include "user-util.h"
#include "varlink-util.h"
typedef struct LookupParameters { typedef struct LookupParameters {
const char *user_name; const char *user_name;
@ -87,6 +86,7 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
LookupParameters p = { LookupParameters p = {
.uid = UID_INVALID, .uid = UID_INVALID,
}; };
@ -104,10 +104,6 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
if (!streq_ptr(p.service, m->userdb_service)) if (!streq_ptr(p.service, m->userdb_service))
return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL);
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (uid_is_valid(p.uid)) if (uid_is_valid(p.uid))
h = hashmap_get(m->homes_by_uid, UID_TO_PTR(p.uid)); h = hashmap_get(m->homes_by_uid, UID_TO_PTR(p.uid));
else if (p.user_name) { else if (p.user_name) {
@ -116,36 +112,45 @@ int vl_method_get_user_record(sd_varlink *link, sd_json_variant *parameters, sd_
return r; return r;
} else { } else {
/* If neither UID nor name was specified, then dump all homes. */ /* If neither UID nor name was specified, then dump all homes. Do so with varlink_notify()
* for all entries but the last, so that clients can stream the results, and easily process
* them piecemeal. */
HASHMAP_FOREACH(h, m->homes_by_uid) { HASHMAP_FOREACH(h, m->homes_by_uid) {
if (!home_user_match_lookup_parameters(&p, h)) if (!home_user_match_lookup_parameters(&p, h))
continue; continue;
if (v) {
/* An entry set from the previous iteration? Then send it now */
r = sd_varlink_notify(link, v);
if (r < 0)
return r;
v = sd_json_variant_unref(v);
}
trusted = client_is_trusted(link, h); trusted = client_is_trusted(link, h);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_user_json(h, trusted, &v); r = build_user_json(h, trusted, &v);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v);
if (r < 0)
return r;
} }
return 0; if (!v)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, v);
} }
if (!h) if (!h)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (!home_user_match_lookup_parameters(&p, h)) if (!home_user_match_lookup_parameters(&p, h))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
trusted = client_is_trusted(link, h); trusted = client_is_trusted(link, h);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_user_json(h, trusted, &v); r = build_user_json(h, trusted, &v);
if (r < 0) if (r < 0)
return r; return r;
@ -196,6 +201,7 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
LookupParameters p = { LookupParameters p = {
.gid = GID_INVALID, .gid = GID_INVALID,
}; };
@ -212,10 +218,6 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
if (!streq_ptr(p.service, m->userdb_service)) if (!streq_ptr(p.service, m->userdb_service))
return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL);
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (gid_is_valid(p.gid)) if (gid_is_valid(p.gid))
h = hashmap_get(m->homes_by_uid, UID_TO_PTR((uid_t) p.gid)); h = hashmap_get(m->homes_by_uid, UID_TO_PTR((uid_t) p.gid));
else if (p.group_name) { else if (p.group_name) {
@ -223,30 +225,37 @@ int vl_method_get_group_record(sd_varlink *link, sd_json_variant *parameters, sd
if (r < 0) if (r < 0)
return r; return r;
} else { } else {
HASHMAP_FOREACH(h, m->homes_by_uid) { HASHMAP_FOREACH(h, m->homes_by_uid) {
if (!home_group_match_lookup_parameters(&p, h)) if (!home_group_match_lookup_parameters(&p, h))
continue; continue;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (v) {
r = build_group_json(h, &v); r = sd_varlink_notify(link, v);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v); v = sd_json_variant_unref(v);
}
r = build_group_json(h, &v);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!v)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, v);
} }
if (!h) if (!h)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (!home_group_match_lookup_parameters(&p, h)) if (!home_group_match_lookup_parameters(&p, h))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_group_json(h, &v); r = build_group_json(h, &v);
if (r < 0) if (r < 0)
return r; return r;
@ -277,21 +286,17 @@ int vl_method_get_memberships(sd_varlink *link, sd_json_variant *parameters, sd_
if (!streq_ptr(p.service, m->userdb_service)) if (!streq_ptr(p.service, m->userdb_service))
return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.BadService", NULL);
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (p.user_name) { if (p.user_name) {
r = manager_get_home_by_name(m, p.user_name, &h); r = manager_get_home_by_name(m, p.user_name, &h);
if (r < 0) if (r < 0)
return r; return r;
if (!h) if (!h)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (p.group_name) { if (p.group_name) {
if (!strv_contains(h->record->member_of, p.group_name) && if (!strv_contains(h->record->member_of, p.group_name) &&
!user_record_matches_user_name(h->record, p.group_name)) !user_record_matches_user_name(h->record, p.group_name))
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_replybo( return sd_varlink_replybo(
link, link,
@ -300,7 +305,7 @@ int vl_method_get_memberships(sd_varlink *link, sd_json_variant *parameters, sd_
} }
STRV_FOREACH(i, h->record->member_of) { STRV_FOREACH(i, h->record->member_of) {
r = sd_varlink_replybo( r = sd_varlink_notifybo(
link, link,
SD_JSON_BUILD_PAIR_STRING("userName", h->user_name), SD_JSON_BUILD_PAIR_STRING("userName", h->user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", *i)); SD_JSON_BUILD_PAIR_STRING("groupName", *i));
@ -314,37 +319,64 @@ int vl_method_get_memberships(sd_varlink *link, sd_json_variant *parameters, sd_
SD_JSON_BUILD_PAIR_STRING("groupName", h->user_name)); SD_JSON_BUILD_PAIR_STRING("groupName", h->user_name));
} else if (p.group_name) { } else if (p.group_name) {
const char *last = NULL;
HASHMAP_FOREACH(h, m->homes_by_uid) { HASHMAP_FOREACH(h, m->homes_by_uid) {
if (!strv_contains(h->record->member_of, p.group_name) && if (!strv_contains(h->record->member_of, p.group_name) &&
!user_record_matches_user_name(h->record, p.group_name)) !user_record_matches_user_name(h->record, p.group_name))
continue; continue;
r = sd_varlink_replybo( if (last) {
link, r = sd_varlink_notifybo(
SD_JSON_BUILD_PAIR_STRING("userName", h->user_name), link,
SD_JSON_BUILD_PAIR_STRING("groupName", p.group_name)); SD_JSON_BUILD_PAIR_STRING("userName", last),
if (r < 0) SD_JSON_BUILD_PAIR_STRING("groupName", p.group_name));
return r; if (r < 0)
} return r;
} else { }
HASHMAP_FOREACH(h, m->homes_by_uid) {
r = sd_varlink_replybo(
link,
SD_JSON_BUILD_PAIR_STRING("userName", h->user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", h->user_name));
if (r < 0)
return r;
last = h->user_name;
}
if (last)
return sd_varlink_replybo(
link,
SD_JSON_BUILD_PAIR_STRING("userName", last),
SD_JSON_BUILD_PAIR_STRING("groupName", p.group_name));
} else {
const char *last = NULL;
HASHMAP_FOREACH(h, m->homes_by_uid) {
STRV_FOREACH(j, h->record->member_of) { STRV_FOREACH(j, h->record->member_of) {
r = sd_varlink_replybo( if (last) {
r = sd_varlink_notifybo(
link,
SD_JSON_BUILD_PAIR_STRING("userName", last),
SD_JSON_BUILD_PAIR_STRING("groupName", last));
if (r < 0)
return r;
last = NULL;
}
r = sd_varlink_notifybo(
link, link,
SD_JSON_BUILD_PAIR_STRING("userName", h->user_name), SD_JSON_BUILD_PAIR_STRING("userName", h->user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", *j)); SD_JSON_BUILD_PAIR_STRING("groupName", *j));
if (r < 0) if (r < 0)
return r; return r;
} }
last = h->user_name;
} }
if (last)
return sd_varlink_replybo(
link,
SD_JSON_BUILD_PAIR_STRING("userName", last),
SD_JSON_BUILD_PAIR_STRING("groupName", last));
} }
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
} }

View File

@ -1801,26 +1801,38 @@ static int vl_method_list_transfers(sd_varlink *link, sd_json_variant *parameter
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL);
r = varlink_set_sentinel(link, "io.systemd.Import.NoTransfers"); Transfer *previous = NULL, *t;
if (r < 0)
return r;
Transfer *t;
HASHMAP_FOREACH(t, m->transfers) { HASHMAP_FOREACH(t, m->transfers) {
if (p.class >= 0 && p.class != t->class) if (p.class >= 0 && p.class != t->class)
continue; continue;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (previous) {
r = make_transfer_json(t, &v); _cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
if (r < 0)
return r;
r = sd_varlink_reply(link, v); r = make_transfer_json(previous, &v);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_notify(link, v);
if (r < 0)
return r;
}
previous = t;
} }
return 0; if (previous) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = make_transfer_json(previous, &v);
if (r < 0)
return r;
return sd_varlink_reply(link, v);
}
return sd_varlink_error(link, "io.systemd.Import.NoTransfers", NULL);
} }
static JSON_DISPATCH_ENUM_DEFINE(json_dispatch_import_verify, ImportVerify, import_verify_from_string); static JSON_DISPATCH_ENUM_DEFINE(json_dispatch_import_verify, ImportVerify, import_verify_from_string);

View File

@ -119,8 +119,6 @@ typedef struct sd_varlink_field sd_varlink_field;
typedef struct sd_varlink_symbol sd_varlink_symbol; typedef struct sd_varlink_symbol sd_varlink_symbol;
typedef struct sd_varlink_interface sd_varlink_interface; typedef struct sd_varlink_interface sd_varlink_interface;
typedef int (*sd_varlink_method_t)(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata);
typedef struct sd_journal sd_journal; typedef struct sd_journal sd_journal;
typedef struct sd_resolve sd_resolve; typedef struct sd_resolve sd_resolve;

View File

@ -616,12 +616,6 @@ static void varlink_clear_current(sd_varlink *v) {
close_many(v->input_fds, v->n_input_fds); close_many(v->input_fds, v->n_input_fds);
v->input_fds = mfree(v->input_fds); v->input_fds = mfree(v->input_fds);
v->n_input_fds = 0; v->n_input_fds = 0;
v->previous = varlink_json_queue_item_free(v->previous);
if (v->sentinel != POINTER_MAX)
v->sentinel = mfree(v->sentinel);
else
v->sentinel = NULL;
} }
static void varlink_clear(sd_varlink *v) { static void varlink_clear(sd_varlink *v) {
@ -1281,154 +1275,6 @@ static int generic_method_get_interface_description(
SD_JSON_BUILD_PAIR_STRING("description", text)); SD_JSON_BUILD_PAIR_STRING("description", text));
} }
static int varlink_format_json(sd_varlink *v, sd_json_variant *m) {
_cleanup_(erase_and_freep) char *text = NULL;
int sz, r;
assert(v);
assert(m);
sz = sd_json_variant_format(m, /* flags= */ 0, &text);
if (sz < 0)
return sz;
assert(text[sz] == '\0');
if (v->output_buffer_size + sz + 1 > VARLINK_BUFFER_MAX)
return -ENOBUFS;
if (DEBUG_LOGGING) {
_cleanup_(erase_and_freep) char *censored_text = NULL;
/* Suppress sensitive fields in the debug output */
r = sd_json_variant_format(m, SD_JSON_FORMAT_CENSOR_SENSITIVE, &censored_text);
if (r < 0)
return r;
varlink_log(v, "Sending message: %s", censored_text);
}
if (v->output_buffer_size == 0) {
free_and_replace(v->output_buffer, text);
v->output_buffer_size = sz + 1;
v->output_buffer_index = 0;
} else if (v->output_buffer_index == 0) {
if (!GREEDY_REALLOC(v->output_buffer, v->output_buffer_size + sz + 1))
return -ENOMEM;
memcpy(v->output_buffer + v->output_buffer_size, text, sz + 1);
v->output_buffer_size += sz + 1;
} else {
char *n;
const size_t new_size = v->output_buffer_size + sz + 1;
n = new(char, new_size);
if (!n)
return -ENOMEM;
memcpy(mempcpy(n, v->output_buffer + v->output_buffer_index, v->output_buffer_size), text, sz + 1);
free_and_replace(v->output_buffer, n);
v->output_buffer_size = new_size;
v->output_buffer_index = 0;
}
if (sd_json_variant_is_sensitive_recursive(m))
v->output_buffer_sensitive = true; /* Propagate sensitive flag */
else
text = mfree(text); /* No point in the erase_and_free() destructor declared above */
return 0;
}
static int varlink_format_queue(sd_varlink *v) {
int r;
assert(v);
/* Takes entries out of the output queue and formats them into the output buffer. But only if this
* would not corrupt our fd message boundaries */
while (v->output_queue) {
_cleanup_free_ int *array = NULL;
assert(v->n_output_queue > 0);
VarlinkJsonQueueItem *q = v->output_queue;
if (v->n_output_fds > 0) /* unwritten fds? if we'd add more we'd corrupt the fd message boundaries, hence wait */
return 0;
if (q->n_fds > 0) {
array = newdup(int, q->fds, q->n_fds);
if (!array)
return -ENOMEM;
}
r = varlink_format_json(v, q->data);
if (r < 0)
return r;
/* Take possession of the queue element's fds */
free(v->output_fds);
v->output_fds = TAKE_PTR(array);
v->n_output_fds = q->n_fds;
q->n_fds = 0;
LIST_REMOVE(queue, v->output_queue, q);
if (!v->output_queue)
v->output_queue_tail = NULL;
v->n_output_queue--;
varlink_json_queue_item_free(q);
}
return 0;
}
static int varlink_enqueue_item(sd_varlink *v, VarlinkJsonQueueItem *q) {
assert(v);
assert(q);
if (v->n_output_queue >= VARLINK_QUEUE_MAX)
return -ENOBUFS;
LIST_INSERT_AFTER(queue, v->output_queue, v->output_queue_tail, q);
v->output_queue_tail = q;
v->n_output_queue++;
return 0;
}
static int varlink_enqueue_json(sd_varlink *v, sd_json_variant *m) {
VarlinkJsonQueueItem *q;
assert(v);
assert(m);
/* If there are no file descriptors to be queued and no queue entries yet we can shortcut things and
* append this entry directly to the output buffer */
if (v->n_pushed_fds == 0 && !v->output_queue)
return varlink_format_json(v, m);
if (v->n_output_queue >= VARLINK_QUEUE_MAX)
return -ENOBUFS;
/* Otherwise add a queue entry for this */
q = varlink_json_queue_item_new(m, v->pushed_fds, v->n_pushed_fds);
if (!q)
return -ENOMEM;
v->n_pushed_fds = 0; /* fds now belong to the queue entry */
/* We already checked the precondition ourselves so this call cannot fail. */
assert_se(varlink_enqueue_item(v, q) >= 0);
return 0;
}
static int varlink_dispatch_method(sd_varlink *v) { static int varlink_dispatch_method(sd_varlink *v) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *parameters = NULL; _cleanup_(sd_json_variant_unrefp) sd_json_variant *parameters = NULL;
sd_varlink_method_flags_t flags = 0; sd_varlink_method_flags_t flags = 0;
@ -1555,46 +1401,16 @@ static int varlink_dispatch_method(sd_varlink *v) {
if (!invalid) { if (!invalid) {
r = callback(v, parameters, flags, v->userdata); r = callback(v, parameters, flags, v->userdata);
if (VARLINK_STATE_WANTS_REPLY(v->state)) { if (r < 0 && VARLINK_STATE_WANTS_REPLY(v->state)) {
if (r < 0) { varlink_log_errno(v, r, "Callback for %s returned error: %m", method);
varlink_log_errno(v, r, "Callback for %s returned error: %m", method);
/* We got an error back from the callback. Propagate it to the client /* We got an error back from the callback. Propagate it to the client if the
* if the method call remains unanswered. */ * method call remains unanswered. */
r = sd_varlink_error_errno(v, r); r = sd_varlink_error_errno(v, r);
} else if (v->sentinel) { /* If we didn't manage to enqueue an error response, then fail the connection completely. */
if (v->previous) {
r = varlink_enqueue_item(v, v->previous);
if (r >= 0) {
TAKE_PTR(v->previous);
varlink_set_state(v, VARLINK_PROCESSED_METHOD);
}
} else {
char *sentinel = TAKE_PTR(v->sentinel);
/* Propagate the sentinel to the client if one was configured
* and no replies were enqueued by the callback. */
if (sentinel == POINTER_MAX)
r = sd_varlink_reply(v, NULL);
else
r = sd_varlink_error(v, sentinel, NULL);
if (sentinel != POINTER_MAX)
free(sentinel);
}
if (r < 0)
varlink_log_errno(v, r, "Failed to process sentinel for method '%s': %m", method);
} else {
assert(!v->previous);
r = 0;
}
/* If we didn't manage to enqueue a response, then fail the connection completely. */
if (r < 0 && VARLINK_STATE_WANTS_REPLY(v->state)) if (r < 0 && VARLINK_STATE_WANTS_REPLY(v->state))
goto fail; goto fail;
}
} else
assert(!v->previous);
} }
} else if (VARLINK_STATE_WANTS_REPLY(v->state)) { } else if (VARLINK_STATE_WANTS_REPLY(v->state)) {
r = sd_varlink_errorbo(v, SD_VARLINK_ERROR_METHOD_NOT_FOUND, SD_JSON_BUILD_PAIR_STRING("method", method)); r = sd_varlink_errorbo(v, SD_VARLINK_ERROR_METHOD_NOT_FOUND, SD_JSON_BUILD_PAIR_STRING("method", method));
@ -2082,6 +1898,141 @@ _public_ sd_varlink* sd_varlink_flush_close_unref(sd_varlink *v) {
return sd_varlink_close_unref(v); return sd_varlink_close_unref(v);
} }
static int varlink_format_json(sd_varlink *v, sd_json_variant *m) {
_cleanup_(erase_and_freep) char *text = NULL;
int sz, r;
assert(v);
assert(m);
sz = sd_json_variant_format(m, /* flags= */ 0, &text);
if (sz < 0)
return sz;
assert(text[sz] == '\0');
if (v->output_buffer_size + sz + 1 > VARLINK_BUFFER_MAX)
return -ENOBUFS;
if (DEBUG_LOGGING) {
_cleanup_(erase_and_freep) char *censored_text = NULL;
/* Suppress sensitive fields in the debug output */
r = sd_json_variant_format(m, SD_JSON_FORMAT_CENSOR_SENSITIVE, &censored_text);
if (r < 0)
return r;
varlink_log(v, "Sending message: %s", censored_text);
}
if (v->output_buffer_size == 0) {
free_and_replace(v->output_buffer, text);
v->output_buffer_size = sz + 1;
v->output_buffer_index = 0;
} else if (v->output_buffer_index == 0) {
if (!GREEDY_REALLOC(v->output_buffer, v->output_buffer_size + sz + 1))
return -ENOMEM;
memcpy(v->output_buffer + v->output_buffer_size, text, sz + 1);
v->output_buffer_size += sz + 1;
} else {
char *n;
const size_t new_size = v->output_buffer_size + sz + 1;
n = new(char, new_size);
if (!n)
return -ENOMEM;
memcpy(mempcpy(n, v->output_buffer + v->output_buffer_index, v->output_buffer_size), text, sz + 1);
free_and_replace(v->output_buffer, n);
v->output_buffer_size = new_size;
v->output_buffer_index = 0;
}
if (sd_json_variant_is_sensitive_recursive(m))
v->output_buffer_sensitive = true; /* Propagate sensitive flag */
else
text = mfree(text); /* No point in the erase_and_free() destructor declared above */
return 0;
}
static int varlink_enqueue_json(sd_varlink *v, sd_json_variant *m) {
VarlinkJsonQueueItem *q;
assert(v);
assert(m);
/* If there are no file descriptors to be queued and no queue entries yet we can shortcut things and
* append this entry directly to the output buffer */
if (v->n_pushed_fds == 0 && !v->output_queue)
return varlink_format_json(v, m);
if (v->n_output_queue >= VARLINK_QUEUE_MAX)
return -ENOBUFS;
/* Otherwise add a queue entry for this */
q = varlink_json_queue_item_new(m, v->pushed_fds, v->n_pushed_fds);
if (!q)
return -ENOMEM;
v->n_pushed_fds = 0; /* fds now belong to the queue entry */
LIST_INSERT_AFTER(queue, v->output_queue, v->output_queue_tail, q);
v->output_queue_tail = q;
v->n_output_queue++;
return 0;
}
static int varlink_format_queue(sd_varlink *v) {
int r;
assert(v);
/* Takes entries out of the output queue and formats them into the output buffer. But only if this
* would not corrupt our fd message boundaries */
while (v->output_queue) {
_cleanup_free_ int *array = NULL;
assert(v->n_output_queue > 0);
VarlinkJsonQueueItem *q = v->output_queue;
if (v->n_output_fds > 0) /* unwritten fds? if we'd add more we'd corrupt the fd message boundaries, hence wait */
return 0;
if (q->n_fds > 0) {
array = newdup(int, q->fds, q->n_fds);
if (!array)
return -ENOMEM;
}
r = varlink_format_json(v, q->data);
if (r < 0)
return r;
/* Take possession of the queue element's fds */
free(v->output_fds);
v->output_fds = TAKE_PTR(array);
v->n_output_fds = q->n_fds;
q->n_fds = 0;
LIST_REMOVE(queue, v->output_queue, q);
if (!v->output_queue)
v->output_queue_tail = NULL;
v->n_output_queue--;
varlink_json_queue_item_free(q);
}
return 0;
}
_public_ int sd_varlink_send(sd_varlink *v, const char *method, sd_json_variant *parameters) { _public_ int sd_varlink_send(sd_varlink *v, const char *method, sd_json_variant *parameters) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *m = NULL; _cleanup_(sd_json_variant_unrefp) sd_json_variant *m = NULL;
int r; int r;
@ -2550,58 +2501,33 @@ _public_ int sd_varlink_collectb(
} }
_public_ int sd_varlink_reply(sd_varlink *v, sd_json_variant *parameters) { _public_ int sd_varlink_reply(sd_varlink *v, sd_json_variant *parameters) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *m = NULL;
int r; int r;
assert_return(v, -EINVAL); assert_return(v, -EINVAL);
if (v->state == VARLINK_DISCONNECTED) if (v->state == VARLINK_DISCONNECTED)
return varlink_log_errno(v, SYNTHETIC_ERRNO(ENOTCONN), "Not connected."); return -ENOTCONN;
if (!IN_SET(v->state, if (!IN_SET(v->state,
VARLINK_PROCESSING_METHOD, VARLINK_PROCESSING_METHOD_MORE, VARLINK_PROCESSING_METHOD, VARLINK_PROCESSING_METHOD_MORE,
VARLINK_PENDING_METHOD, VARLINK_PENDING_METHOD_MORE)) VARLINK_PENDING_METHOD, VARLINK_PENDING_METHOD_MORE))
return varlink_log_errno(v, SYNTHETIC_ERRNO(EBUSY), "Connection busy."); return -EBUSY;
bool more = IN_SET(v->state, VARLINK_PROCESSING_METHOD_MORE, VARLINK_PENDING_METHOD_MORE);
/* Validate parameters BEFORE sanitization */ /* Validate parameters BEFORE sanitization */
if (v->current_method) { if (v->current_method) {
const char *bad_field = NULL; const char *bad_field = NULL;
r = varlink_idl_validate_method_reply(v->current_method, parameters, more && v->sentinel ? SD_VARLINK_REPLY_CONTINUES : 0, &bad_field); r = varlink_idl_validate_method_reply(v->current_method, parameters, /* flags= */ 0, &bad_field);
if (r == -EBADE) if (r < 0)
varlink_log_errno(v, r, "Method reply for %s() has 'continues' flag set, but IDL structure doesn't allow that, ignoring: %m",
v->current_method->name);
else if (r < 0)
/* Please adjust test/units/end.sh when updating the log message. */ /* Please adjust test/units/end.sh when updating the log message. */
varlink_log_errno(v, r, "Return parameters for method reply %s() didn't pass validation on field '%s', ignoring: %m", varlink_log_errno(v, r, "Return parameters for method reply %s() didn't pass validation on field '%s', ignoring: %m",
v->current_method->name, strna(bad_field)); v->current_method->name, strna(bad_field));
} }
_cleanup_(sd_json_variant_unrefp) sd_json_variant *m = NULL;
r = sd_json_buildo(&m, JSON_BUILD_PAIR_VARIANT_NON_EMPTY("parameters", parameters)); r = sd_json_buildo(&m, JSON_BUILD_PAIR_VARIANT_NON_EMPTY("parameters", parameters));
if (r < 0) if (r < 0)
return varlink_log_errno(v, r, "Failed to build json message: %m"); return varlink_log_errno(v, r, "Failed to build json message: %m");
if (more && v->sentinel) {
if (v->previous) {
r = sd_json_variant_set_field_boolean(&v->previous->data, "continues", true);
if (r < 0)
return r;
r = varlink_enqueue_item(v, v->previous);
if (r < 0)
return varlink_log_errno(v, r, "Failed to enqueue json message: %m");
}
v->previous = varlink_json_queue_item_new(m, v->pushed_fds, v->n_pushed_fds);
if (!v->previous)
return -ENOMEM;
v->n_pushed_fds = 0; /* fds now belong to the queue entry */
return 1;
}
r = varlink_enqueue_json(v, m); r = varlink_enqueue_json(v, m);
if (r < 0) if (r < 0)
return varlink_log_errno(v, r, "Failed to enqueue json message: %m"); return varlink_log_errno(v, r, "Failed to enqueue json message: %m");
@ -2664,21 +2590,6 @@ _public_ int sd_varlink_error(sd_varlink *v, const char *error_id, sd_json_varia
VARLINK_PENDING_METHOD, VARLINK_PENDING_METHOD_MORE)) VARLINK_PENDING_METHOD, VARLINK_PENDING_METHOD_MORE))
return varlink_log_errno(v, SYNTHETIC_ERRNO(EBUSY), "Connection busy."); return varlink_log_errno(v, SYNTHETIC_ERRNO(EBUSY), "Connection busy.");
if (v->previous) {
r = sd_json_variant_set_field_boolean(&v->previous->data, "continues", true);
if (r < 0)
return r;
/* If we have a previous reply still ready make sure we queue it before the error. We only
* ever set "previous" if we're in a streaming method so we pass more=true uncondtionally
* here as we know we're still going to queue an error afterwards. */
r = varlink_enqueue_item(v, v->previous);
if (r < 0)
return varlink_log_errno(v, r, "Failed to enqueue json message: %m");
TAKE_PTR(v->previous);
}
/* Reset the list of pushed file descriptors before sending an error reply. We do this here to /* Reset the list of pushed file descriptors before sending an error reply. We do this here to
* simplify code that puts together a complex reply message with fds, and half-way something * simplify code that puts together a complex reply message with fds, and half-way something
* fails. In that case the pushed fds need to be flushed out again. Under the assumption that it * fails. In that case the pushed fds need to be flushed out again. Under the assumption that it
@ -2810,11 +2721,6 @@ _public_ int sd_varlink_notify(sd_varlink *v, sd_json_variant *parameters) {
assert_return(v, -EINVAL); assert_return(v, -EINVAL);
if (v->sentinel)
return varlink_log_errno(v, SYNTHETIC_ERRNO(EINVAL), "Cannot use sd_varlink_notify() on method with sentinel set");
assert(!v->previous);
if (v->state == VARLINK_DISCONNECTED) if (v->state == VARLINK_DISCONNECTED)
return varlink_log_errno(v, SYNTHETIC_ERRNO(ENOTCONN), "Not connected."); return varlink_log_errno(v, SYNTHETIC_ERRNO(ENOTCONN), "Not connected.");

View File

@ -82,7 +82,7 @@ struct VarlinkJsonQueueItem {
int fds[]; int fds[];
}; };
typedef struct sd_varlink { struct sd_varlink {
unsigned n_ref; unsigned n_ref;
sd_varlink_server *server; sd_varlink_server *server;
@ -157,9 +157,6 @@ typedef struct sd_varlink {
sd_varlink_reply_flags_t current_reply_flags; sd_varlink_reply_flags_t current_reply_flags;
sd_varlink_symbol *current_method; sd_varlink_symbol *current_method;
VarlinkJsonQueueItem *previous;
char *sentinel;
int peer_pidfd; int peer_pidfd;
struct ucred ucred; struct ucred ucred;
bool ucred_acquired:1; bool ucred_acquired:1;
@ -192,7 +189,7 @@ typedef struct sd_varlink {
sd_event_source *defer_event_source; sd_event_source *defer_event_source;
PidRef exec_pidref; PidRef exec_pidref;
} sd_varlink; };
typedef struct VarlinkServerSocket VarlinkServerSocket; typedef struct VarlinkServerSocket VarlinkServerSocket;
@ -207,7 +204,7 @@ struct VarlinkServerSocket {
LIST_FIELDS(VarlinkServerSocket, sockets); LIST_FIELDS(VarlinkServerSocket, sockets);
}; };
typedef struct sd_varlink_server { struct sd_varlink_server {
unsigned n_ref; unsigned n_ref;
sd_varlink_server_flags_t flags; sd_varlink_server_flags_t flags;
@ -237,7 +234,7 @@ typedef struct sd_varlink_server {
unsigned connections_per_uid_max; unsigned connections_per_uid_max;
bool exit_on_idle; bool exit_on_idle;
} sd_varlink_server; };
#define varlink_log_errno(v, error, fmt, ...) \ #define varlink_log_errno(v, error, fmt, ...) \
log_debug_errno(error, "%s: " fmt, varlink_description(v), ##__VA_ARGS__) log_debug_errno(error, "%s: " fmt, varlink_description(v), ##__VA_ARGS__)

View File

@ -6,7 +6,6 @@
#include "pidref.h" #include "pidref.h"
#include "set.h" #include "set.h"
#include "string-util.h" #include "string-util.h"
#include "varlink-internal.h"
#include "varlink-util.h" #include "varlink-util.h"
#include "version.h" #include "version.h"
@ -205,31 +204,6 @@ int varlink_check_privileged_peer(sd_varlink *vl) {
return 0; return 0;
} }
int varlink_set_sentinel(sd_varlink *v, const char *error_id) {
_cleanup_free_ char *s = NULL;
assert(v);
/* If the caller doesn't want a reply, then don't set a sentinel. */
if (v->state == VARLINK_PROCESSING_METHOD_ONEWAY)
return 0;
/* This has to be called during a callback, and not after it has exited. */
assert(IN_SET(v->state, VARLINK_PROCESSING_METHOD, VARLINK_PROCESSING_METHOD_MORE));
if (error_id) {
s = strdup(error_id);
if (!s)
return -ENOMEM;
}
if (v->sentinel != POINTER_MAX)
free(v->sentinel);
v->sentinel = s ? TAKE_PTR(s) : POINTER_MAX;
return 0;
}
DEFINE_HASH_OPS_WITH_VALUE_DESTRUCTOR( DEFINE_HASH_OPS_WITH_VALUE_DESTRUCTOR(
varlink_hash_ops, varlink_hash_ops,
void, void,

View File

@ -28,6 +28,4 @@ int varlink_server_new(
int varlink_check_privileged_peer(sd_varlink *vl); int varlink_check_privileged_peer(sd_varlink *vl);
int varlink_set_sentinel(sd_varlink *v, const char *error_id);
extern const struct hash_ops varlink_hash_ops; extern const struct hash_ops varlink_hash_ops;

View File

@ -426,7 +426,7 @@ static int json_build_local_addresses(const struct local_address *addresses, siz
return 0; return 0;
} }
static int list_machine_one_and_maybe_read_metadata(sd_varlink *link, Machine *m, AcquireMetadata am) { static int list_machine_one_and_maybe_read_metadata(sd_varlink *link, Machine *m, bool more, AcquireMetadata am) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL, *addr_array = NULL; _cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL, *addr_array = NULL;
_cleanup_strv_free_ char **os_release = NULL; _cleanup_strv_free_ char **os_release = NULL;
uid_t shift = UID_INVALID; uid_t shift = UID_INVALID;
@ -496,6 +496,9 @@ static int list_machine_one_and_maybe_read_metadata(sd_varlink *link, Machine *m
if (r < 0) if (r < 0)
return r; return r;
if (more)
return sd_varlink_notify(link, v);
return sd_varlink_reply(link, v); return sd_varlink_reply(link, v);
} }
@ -525,6 +528,8 @@ static int vl_method_list(sd_varlink *link, sd_json_variant *parameters, sd_varl
_cleanup_(machine_lookup_parameters_done) MachineLookupParameters p = { _cleanup_(machine_lookup_parameters_done) MachineLookupParameters p = {
.pidref = PIDREF_NULL, .pidref = PIDREF_NULL,
}; };
Machine *machine;
int r; int r;
assert(link); assert(link);
@ -534,32 +539,34 @@ static int vl_method_list(sd_varlink *link, sd_json_variant *parameters, sd_varl
if (r != 0) if (r != 0)
return r; return r;
r = varlink_set_sentinel(link, VARLINK_ERROR_MACHINE_NO_SUCH_MACHINE);
if (r < 0)
return r;
if (p.name || pidref_is_set(&p.pidref) || pidref_is_automatic(&p.pidref)) { if (p.name || pidref_is_set(&p.pidref) || pidref_is_automatic(&p.pidref)) {
Machine *machine;
r = lookup_machine_by_name_or_pidref(link, m, p.name, &p.pidref, &machine); r = lookup_machine_by_name_or_pidref(link, m, p.name, &p.pidref, &machine);
if (r == -ESRCH) if (r == -ESRCH)
return 0; return sd_varlink_error(link, VARLINK_ERROR_MACHINE_NO_SUCH_MACHINE, NULL);
if (r < 0) if (r < 0)
return r; return r;
return list_machine_one_and_maybe_read_metadata(link, machine, p.acquire_metadata); return list_machine_one_and_maybe_read_metadata(link, machine, /* more= */ false, p.acquire_metadata);
} }
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL);
Machine *machine; Machine *previous = NULL, *i;
HASHMAP_FOREACH(machine, m->machines) { HASHMAP_FOREACH(i, m->machines) {
r = list_machine_one_and_maybe_read_metadata(link, machine, p.acquire_metadata); if (previous) {
if (r < 0) r = list_machine_one_and_maybe_read_metadata(link, previous, /* more= */ true, p.acquire_metadata);
return r; if (r < 0)
return r;
}
previous = i;
} }
return 0; if (previous)
return list_machine_one_and_maybe_read_metadata(link, previous, /* more= */ false, p.acquire_metadata);
return sd_varlink_error(link, VARLINK_ERROR_MACHINE_NO_SUCH_MACHINE, NULL);
} }
static int lookup_machine_and_call_method(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata, sd_varlink_method_t method) { static int lookup_machine_and_call_method(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata, sd_varlink_method_t method) {
@ -612,7 +619,7 @@ static int vl_method_open_root_directory(sd_varlink *link, sd_json_variant *para
return lookup_machine_and_call_method(link, parameters, flags, userdata, vl_method_open_root_directory_internal); return lookup_machine_and_call_method(link, parameters, flags, userdata, vl_method_open_root_directory_internal);
} }
static int list_image_one_and_maybe_read_metadata(Manager *m, sd_varlink *link, Image *image, AcquireMetadata am) { static int list_image_one_and_maybe_read_metadata(Manager *m, sd_varlink *link, Image *image, bool more, AcquireMetadata am) {
int r; int r;
assert(m); assert(m);
@ -656,6 +663,9 @@ static int list_image_one_and_maybe_read_metadata(Manager *m, sd_varlink *link,
return r; return r;
} }
if (more)
return sd_varlink_notify(link, v);
return sd_varlink_reply(link, v); return sd_varlink_reply(link, v);
} }
@ -681,10 +691,6 @@ static int vl_method_list_images(sd_varlink *link, sd_json_variant *parameters,
if (r != 0) if (r != 0)
return r; return r;
r = varlink_set_sentinel(link, VARLINK_ERROR_MACHINE_IMAGE_NO_SUCH_IMAGE);
if (r < 0)
return r;
if (p.image_name) { if (p.image_name) {
_cleanup_(image_unrefp) Image *found = NULL; _cleanup_(image_unrefp) Image *found = NULL;
@ -693,11 +699,11 @@ static int vl_method_list_images(sd_varlink *link, sd_json_variant *parameters,
r = image_find(m->runtime_scope, IMAGE_MACHINE, p.image_name, /* root= */ NULL, &found); r = image_find(m->runtime_scope, IMAGE_MACHINE, p.image_name, /* root= */ NULL, &found);
if (r == -ENOENT) if (r == -ENOENT)
return 0; return sd_varlink_error(link, VARLINK_ERROR_MACHINE_IMAGE_NO_SUCH_IMAGE, NULL);
if (r < 0) if (r < 0)
return log_debug_errno(r, "Failed to find image: %m"); return log_debug_errno(r, "Failed to find image: %m");
return list_image_one_and_maybe_read_metadata(m, link, found, p.acquire_metadata); return list_image_one_and_maybe_read_metadata(m, link, found, /* more= */ false, p.acquire_metadata);
} }
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
@ -708,14 +714,21 @@ static int vl_method_list_images(sd_varlink *link, sd_json_variant *parameters,
if (r < 0) if (r < 0)
return log_debug_errno(r, "Failed to discover images: %m"); return log_debug_errno(r, "Failed to discover images: %m");
Image *image; Image *image, *previous = NULL;
HASHMAP_FOREACH(image, images) { HASHMAP_FOREACH(image, images) {
r = list_image_one_and_maybe_read_metadata(m, link, image, p.acquire_metadata); if (previous) {
if (r < 0) r = list_image_one_and_maybe_read_metadata(m, link, previous, /* more= */ true, p.acquire_metadata);
return r; if (r < 0)
return r;
}
previous = image;
} }
return 0; if (previous)
return list_image_one_and_maybe_read_metadata(m, link, previous, /* more= */ false, p.acquire_metadata);
return sd_varlink_error(link, VARLINK_ERROR_MACHINE_IMAGE_NO_SUCH_IMAGE, NULL);
} }
static int manager_varlink_init_userdb(Manager *m) { static int manager_varlink_init_userdb(Manager *m) {

View File

@ -5417,23 +5417,24 @@ static int vl_method_read_event_log(sd_varlink *link, sd_json_variant *parameter
if (r < 0) if (r < 0)
return r; return r;
// FIXME: We can't use a NULL sentinel here because the output fields in the IDL are non-nullable. _cleanup_(sd_json_variant_unrefp) sd_json_variant *rec_cel = NULL;
r = varlink_set_sentinel(link, NULL);
if (r < 0)
return r;
FOREACH_ARRAY(rr, el->records, el->n_records) { FOREACH_ARRAY(rr, el->records, el->n_records) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *rec_cel = NULL;
r = event_log_record_to_cel(*rr, &recnum, &rec_cel);
if (r < 0)
return r;
r = sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_VARIANT("record", rec_cel)); if (rec_cel) {
r = sd_varlink_notifybo(link, SD_JSON_BUILD_PAIR_VARIANT("record", rec_cel));
if (r < 0)
return r;
rec_cel = sd_json_variant_unref(rec_cel);
}
r = event_log_record_to_cel(*rr, &recnum, &rec_cel);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; return sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_CONDITION(!!rec_cel, "record", SD_JSON_BUILD_VARIANT(rec_cel)));
} }
typedef struct MethodMakePolicyParameters { typedef struct MethodMakePolicyParameters {

View File

@ -10346,12 +10346,21 @@ static int vl_method_list_candidate_devices(
if (r < 0) if (r < 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.Repart.NoCandidateDevices"); if (n == 0)
if (r < 0) return sd_varlink_error(link, "io.systemd.Repart.NoCandidateDevices", NULL);
return r;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
FOREACH_ARRAY(d, l, n) { FOREACH_ARRAY(d, l, n) {
r = sd_varlink_replybo(link, if (v) {
r = sd_varlink_notify(link, v);
if (r < 0)
return r;
v = sd_json_variant_unref(v);
}
r = sd_json_buildo(
&v,
SD_JSON_BUILD_PAIR_STRING("node", d->node), SD_JSON_BUILD_PAIR_STRING("node", d->node),
JSON_BUILD_PAIR_STRV_NON_EMPTY("symlinks", d->symlinks), JSON_BUILD_PAIR_STRV_NON_EMPTY("symlinks", d->symlinks),
JSON_BUILD_PAIR_UNSIGNED_NOT_EQUAL("diskseq", d->diskseq, UINT64_MAX), JSON_BUILD_PAIR_UNSIGNED_NOT_EQUAL("diskseq", d->diskseq, UINT64_MAX),
@ -10363,7 +10372,8 @@ static int vl_method_list_candidate_devices(
return r; return r;
} }
return 0; assert(v);
return sd_varlink_reply(link, v);
} }
static JSON_DISPATCH_ENUM_DEFINE(json_dispatch_empty_mode, EmptyMode, empty_mode_from_string); static JSON_DISPATCH_ENUM_DEFINE(json_dispatch_empty_mode, EmptyMode, empty_mode_from_string);

View File

@ -886,9 +886,6 @@ Manager* manager_free(Manager *m) {
manager_dns_stub_stop(m); manager_dns_stub_stop(m);
manager_varlink_done(m); manager_varlink_done(m);
set_free(m->varlink_query_results_subscription);
set_free(m->varlink_dns_configuration_subscription);
manager_socket_graveyard_clear(m); manager_socket_graveyard_clear(m);
ordered_set_free(m->dns_extra_stub_listeners); ordered_set_free(m->dns_extra_stub_listeners);

View File

@ -4,9 +4,16 @@
#include "log.h" #include "log.h"
#include "metrics.h" #include "metrics.h"
#include "string-table.h" #include "string-table.h"
#include "strv.h"
#include "varlink-io.systemd.Metrics.h" #include "varlink-io.systemd.Metrics.h"
#include "varlink-util.h" #include "varlink-util.h"
static void metric_family_context_done(MetricFamilyContext *ctx) {
assert(ctx);
sd_json_variant_unref(ctx->previous);
}
int metrics_setup_varlink_server( int metrics_setup_varlink_server(
sd_varlink_server **server, /* in and out param */ sd_varlink_server **server, /* in and out param */
sd_varlink_server_flags_t flags, sd_varlink_server_flags_t flags,
@ -47,7 +54,7 @@ int metrics_setup_varlink_server(
r = sd_varlink_server_attach_event(s, event, SD_EVENT_PRIORITY_NORMAL); r = sd_varlink_server_attach_event(s, event, SD_EVENT_PRIORITY_NORMAL);
if (r < 0) if (r < 0)
return log_debug_errno(r, "Failed to attach varlink metrics server to event loop: %m"); return log_debug_errno(r, "Failed to attach varlink metrics connection to event loop: %m");
*server = TAKE_PTR(s); *server = TAKE_PTR(s);
@ -62,14 +69,25 @@ static const char * const metric_family_type_table[_METRIC_FAMILY_TYPE_MAX] = {
DEFINE_STRING_TABLE_LOOKUP_TO_STRING(metric_family_type, MetricFamilyType); DEFINE_STRING_TABLE_LOOKUP_TO_STRING(metric_family_type, MetricFamilyType);
static int metric_family_build_json(const MetricFamily *mf, sd_json_variant **ret) { static int metric_family_build_send(sd_varlink *link, const MetricFamily *mf, bool more) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
int r;
assert(link);
assert(mf); assert(mf);
return sd_json_buildo( r = sd_json_buildo(
ret, &v,
SD_JSON_BUILD_PAIR_STRING("name", mf->name), SD_JSON_BUILD_PAIR_STRING("name", mf->name),
SD_JSON_BUILD_PAIR_STRING("description", mf->description), SD_JSON_BUILD_PAIR_STRING("description", mf->description),
SD_JSON_BUILD_PAIR_STRING("type", metric_family_type_to_string(mf->type))); SD_JSON_BUILD_PAIR_STRING("type", metric_family_type_to_string(mf->type)));
if (r < 0)
return r;
if (more)
return sd_varlink_notify(link, v);
return sd_varlink_reply(link, v);
} }
int metrics_method_describe( int metrics_method_describe(
@ -92,22 +110,25 @@ int metrics_method_describe(
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL);
r = varlink_set_sentinel(link, "io.systemd.Metrics.NoSuchMetric"); const MetricFamily *previous = NULL;
if (r < 0)
return r;
for (const MetricFamily *mf = metric_family_table; mf && mf->name; mf++) { for (const MetricFamily *mf = metric_family_table; mf && mf->name; mf++) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (previous) {
r = metric_family_build_send(link, previous, /* more= */ true);
if (r < 0)
return log_debug_errno(
r, "Failed to describe metric family '%s': %m", previous->name);
}
r = metric_family_build_json(mf, &v); previous = mf;
if (r < 0)
return log_debug_errno(r, "Failed to describe metric family '%s': %m", mf->name);
r = sd_varlink_reply(link, v);
if (r < 0)
return log_debug_errno(r, "Failed to send varlink reply: %m");
} }
if (!previous)
return sd_varlink_error(link, "io.systemd.Metrics.NoSuchMetric", NULL);
r = metric_family_build_send(link, previous, /* more= */ false);
if (r < 0)
return log_debug_errno(r, "Failed to describe metric family '%s': %m", previous->name);
return 0; return 0;
} }
@ -131,47 +152,100 @@ int metrics_method_list(
if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE)) if (!FLAGS_SET(flags, SD_VARLINK_METHOD_MORE))
return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL); return sd_varlink_error(link, SD_VARLINK_ERROR_EXPECTED_MORE, NULL);
r = varlink_set_sentinel(link, "io.systemd.Metrics.NoSuchMetric"); _cleanup_(metric_family_context_done) MetricFamilyContext ctx = { .link = link };
if (r < 0)
return r;
MetricFamilyContext ctx = { .link = link };
for (const MetricFamily *mf = metric_family_table; mf && mf->name; mf++) { for (const MetricFamily *mf = metric_family_table; mf && mf->name; mf++) {
assert(mf->generate); assert(mf->generate_cb);
ctx.metric_family = mf; ctx.metric_family = mf;
r = mf->generate(&ctx, userdata); r = mf->generate_cb(&ctx, userdata);
if (r < 0) if (r < 0)
return log_debug_errno( return log_debug_errno(
r, "Failed to list metrics for metric family '%s': %m", mf->name); r, "Failed to list metrics for metric family '%s': %m", mf->name);
} }
return 0; if (!ctx.previous)
return sd_varlink_error(link, "io.systemd.Metrics.NoSuchMetric", NULL);
/* produce the last metric */
return sd_varlink_reply(link, ctx.previous);
} }
static int metric_build_send(MetricFamilyContext *context, const char *object, sd_json_variant *value, sd_json_variant *fields) { static int metric_set_fields(sd_json_variant **v, char **field_pairs) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *w = NULL;
size_t n;
int r;
assert(v);
n = strv_length(field_pairs);
if (n == 0)
return 0;
if (n % 2 != 0)
return log_debug_errno(SYNTHETIC_ERRNO(ERANGE), "Odd number of field pairs: %zu", n);
sd_json_variant **array = new0(sd_json_variant *, n);
if (!array)
return log_oom();
CLEANUP_ARRAY(array, n, sd_json_variant_unref_many);
int i = 0;
STRV_FOREACH_PAIR(key, value, field_pairs) {
r = sd_json_variant_new_string(&array[i++], *key);
if (r < 0)
return log_debug_errno(r, "Failed to create key variant: %m");
r = sd_json_variant_new_string(&array[i++], *value);
if (r < 0)
return log_debug_errno(r, "Failed to create value variant: %m");
}
r = sd_json_variant_new_object(&w, array, n);
if (r < 0)
return log_debug_errno(r, "Failed to allocate JSON object: %m");
return sd_json_variant_set_field(v, "fields", w);
}
static int metric_build_send(MetricFamilyContext *context, const char *object, sd_json_variant *value, char **field_pairs) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
int r;
assert(context); assert(context);
assert(value); assert(value);
assert(context->link); assert(context->link);
assert(context->metric_family); assert(context->metric_family);
if (fields) { r = sd_json_buildo(
assert(sd_json_variant_is_object(fields)); &v,
_unused_ const char *k;
_unused_ sd_json_variant *e;
JSON_VARIANT_OBJECT_FOREACH(k, e, fields)
assert(sd_json_variant_is_string(e));
}
return sd_varlink_replybo(context->link,
SD_JSON_BUILD_PAIR_STRING("name", context->metric_family->name), SD_JSON_BUILD_PAIR_STRING("name", context->metric_family->name),
JSON_BUILD_PAIR_STRING_NON_EMPTY("object", object), JSON_BUILD_PAIR_STRING_NON_EMPTY("object", object),
SD_JSON_BUILD_PAIR("value", SD_JSON_BUILD_VARIANT(value)), SD_JSON_BUILD_PAIR("value", SD_JSON_BUILD_VARIANT(value)));
JSON_BUILD_PAIR_VARIANT_NON_NULL("fields", fields)); /* TODO JSON_BUILD_PAIR_OBJECT_STRV_NOT_NULL */
if (r < 0)
return r;
if (field_pairs) { /* NULL => no fields object, empty strv => fields:{} */
r = metric_set_fields(&v, field_pairs);
if (r < 0)
return r;
}
if (context->previous) {
r = sd_varlink_notify(context->link, context->previous);
if (r < 0)
return r;
context->previous = sd_json_variant_unref(context->previous);
}
context->previous = TAKE_PTR(v);
return 0;
} }
int metric_build_send_string(MetricFamilyContext *context, const char *object, const char *value, sd_json_variant *fields) { int metric_build_send_string(MetricFamilyContext *context, const char *object, const char *value, char **field_pairs) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; _cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
int r; int r;
@ -181,10 +255,10 @@ int metric_build_send_string(MetricFamilyContext *context, const char *object, c
if (r < 0) if (r < 0)
return log_debug_errno(r, "Failed to allocate JSON string: %m"); return log_debug_errno(r, "Failed to allocate JSON string: %m");
return metric_build_send(context, object, v, fields); return metric_build_send(context, object, v, field_pairs);
} }
int metric_build_send_unsigned(MetricFamilyContext *context, const char *object, uint64_t value, sd_json_variant *fields) { int metric_build_send_unsigned(MetricFamilyContext *context, const char *object, uint64_t value, char **field_pairs) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; _cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
int r; int r;
@ -192,5 +266,5 @@ int metric_build_send_unsigned(MetricFamilyContext *context, const char *object,
if (r < 0) if (r < 0)
return log_debug_errno(r, "Failed to allocate JSON unsigned: %m"); return log_debug_errno(r, "Failed to allocate JSON unsigned: %m");
return metric_build_send(context, object, v, fields); return metric_build_send(context, object, v, field_pairs);
} }

View File

@ -1,7 +1,9 @@
/* SPDX-License-Identifier: LGPL-2.1-or-later */ /* SPDX-License-Identifier: LGPL-2.1-or-later */
#pragma once #pragma once
#include "shared-forward.h" #include "sd-varlink.h"
#include "macro-fundamental.h"
typedef enum MetricFamilyType { typedef enum MetricFamilyType {
METRIC_FAMILY_TYPE_COUNTER, METRIC_FAMILY_TYPE_COUNTER,
@ -16,15 +18,16 @@ typedef struct MetricFamily MetricFamily;
typedef struct MetricFamilyContext { typedef struct MetricFamilyContext {
const MetricFamily* metric_family; const MetricFamily* metric_family;
sd_varlink *link; sd_varlink *link;
sd_json_variant *previous;
} MetricFamilyContext; } MetricFamilyContext;
typedef int (*metric_family_generate_func_t) (MetricFamilyContext *mfc, void *userdata); typedef int (*metric_family_generate_cb_t) (MetricFamilyContext *mfc, void *userdata);
typedef struct MetricFamily { typedef struct MetricFamily {
const char *name; const char *name;
const char *description; const char *description;
MetricFamilyType type; MetricFamilyType type;
metric_family_generate_func_t generate; metric_family_generate_cb_t generate_cb;
} MetricFamily; } MetricFamily;
int metrics_setup_varlink_server( int metrics_setup_varlink_server(
@ -35,10 +38,9 @@ int metrics_setup_varlink_server(
sd_varlink_method_t vl_method_describe_cb, sd_varlink_method_t vl_method_describe_cb,
void *userdata); void *userdata);
DECLARE_STRING_TABLE_LOOKUP_TO_STRING(metric_family_type, MetricFamilyType); const char* metric_family_type_to_string(MetricFamilyType i) _const_;
int metrics_method_describe(const MetricFamily metric_family_table[], sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata); int metrics_method_describe(const MetricFamily metric_family_table[], sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata);
int metrics_method_list(const MetricFamily metric_family_table[], sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata); int metrics_method_list(const MetricFamily metric_family_table[], sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata);
int metric_build_send_string(MetricFamilyContext* context, const char *object, const char *value, sd_json_variant *fields); int metric_build_send_string(MetricFamilyContext* context, const char *object, const char *value, char **field_pairs);
int metric_build_send_unsigned(MetricFamilyContext* context, const char *object, uint64_t value, sd_json_variant *fields); int metric_build_send_unsigned(MetricFamilyContext* context, const char *object, uint64_t value, char **field_pairs);

View File

@ -87,7 +87,7 @@ static SD_VARLINK_DEFINE_METHOD_FULL(
ListBootEntries, ListBootEntries,
SD_VARLINK_REQUIRES_MORE, SD_VARLINK_REQUIRES_MORE,
SD_VARLINK_FIELD_COMMENT("A boot menu entry structure"), SD_VARLINK_FIELD_COMMENT("A boot menu entry structure"),
SD_VARLINK_DEFINE_OUTPUT_BY_TYPE(entry, BootEntry, 0)); SD_VARLINK_DEFINE_OUTPUT_BY_TYPE(entry, BootEntry, SD_VARLINK_NULLABLE));
static SD_VARLINK_DEFINE_METHOD( static SD_VARLINK_DEFINE_METHOD(
SetRebootToFirmware, SetRebootToFirmware,

View File

@ -2745,6 +2745,7 @@ static int vl_method_list(sd_varlink *link, sd_json_variant *parameters, sd_varl
VARLINK_DISPATCH_POLKIT_FIELD, VARLINK_DISPATCH_POLKIT_FIELD,
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
Hashmap **polkit_registry = ASSERT_PTR(userdata); Hashmap **polkit_registry = ASSERT_PTR(userdata);
int r; int r;
@ -2774,23 +2775,26 @@ static int vl_method_list(sd_varlink *link, sd_json_variant *parameters, sd_varl
if (r < 0) if (r < 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.sysext.NoImagesFound");
if (r < 0)
return r;
Image *img; Image *img;
HASHMAP_FOREACH(img, images) { HASHMAP_FOREACH(img, images) {
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (v) {
r = image_to_json(img, &v); /* Send previous item with more=true */
if (r < 0) r = sd_varlink_notify(link, v);
return r; if (r < 0)
return r;
}
r = sd_varlink_reply(link, v); v = sd_json_variant_unref(v);
r = image_to_json(img, &v);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (v) /* Send final item with more=false */
return sd_varlink_reply(link, v);
return sd_varlink_error(link, "io.systemd.sysext.NoImagesFound", NULL);
} }
static int verb_help(int argc, char **argv, void *userdata) { static int verb_help(int argc, char **argv, void *userdata) {

View File

@ -441,315 +441,4 @@ TEST(invalid_parameter) {
ASSERT_OK(sd_event_loop(e)); ASSERT_OK(sd_event_loop(e));
} }
static int method_with_error_sentinel(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
/* Set an error sentinel and return without sending a reply. The sentinel error should be sent automatically. */
ASSERT_OK(varlink_set_sentinel(link, "io.test.SentinelError"));
return 0;
}
static int reply_sentinel_error(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
ASSERT_STREQ(error_id, "io.test.SentinelError");
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
return 0;
}
TEST(sentinel_error) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, 0));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.ErrorSentinel", method_with_error_sentinel));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
ASSERT_OK(sd_varlink_bind_reply(c, reply_sentinel_error));
ASSERT_OK(sd_varlink_invoke(c, "io.test.ErrorSentinel", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
static int method_with_empty_sentinel(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
/* Set an empty sentinel and return without sending a reply. An empty reply should be sent automatically. */
ASSERT_OK(varlink_set_sentinel(link, /* error_id= */ NULL));
return 0;
}
static int reply_sentinel_empty(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
ASSERT_NULL(error_id);
ASSERT_TRUE(sd_json_variant_is_blank_object(parameters));
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
return 0;
}
TEST(sentinel_empty) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, 0));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.EmptySentinel", method_with_empty_sentinel));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
ASSERT_OK(sd_varlink_bind_reply(c, reply_sentinel_empty));
ASSERT_OK(sd_varlink_invoke(c, "io.test.EmptySentinel", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
static int method_with_sentinel_but_reply(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
/* Set a sentinel but also send a reply. The sentinel should not be used. */
ASSERT_OK(varlink_set_sentinel(link, "io.test.SentinelError"));
return sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_STRING("result", "explicit-reply"));
}
static int reply_sentinel_explicit(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
ASSERT_NULL(error_id);
ASSERT_STREQ(sd_json_variant_string(sd_json_variant_by_key(parameters, "result")), "explicit-reply");
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
return 0;
}
TEST(sentinel_with_explicit_reply) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, 0));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.SentinelButReply", method_with_sentinel_but_reply));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
ASSERT_OK(sd_varlink_bind_reply(c, reply_sentinel_explicit));
ASSERT_OK(sd_varlink_invoke(c, "io.test.SentinelButReply", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
static int method_with_oneway_sentinel(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
/* The method was called oneway, so varlink_set_sentinel() should be a no-op and the server should
* transition back to idle without sending any reply. */
ASSERT_TRUE(FLAGS_SET(flags, SD_VARLINK_METHOD_ONEWAY));
ASSERT_OK(varlink_set_sentinel(link, "io.test.SentinelError"));
return 0;
}
static int method_oneway_sentinel_pong(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
return sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_STRING("result", "pong"));
}
static int reply_oneway_sentinel_pong(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
/* If we get here, it means the oneway sentinel call didn't break the connection and the server
* properly handled a subsequent regular method call. */
ASSERT_NULL(error_id);
ASSERT_STREQ(sd_json_variant_string(sd_json_variant_by_key(parameters, "result")), "pong");
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
return 0;
}
TEST(sentinel_oneway) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, 0));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.OnewaySentinel", method_with_oneway_sentinel));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.Pong", method_oneway_sentinel_pong));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
/* Send a oneway call with a sentinel — the sentinel should be silently ignored. */
ASSERT_OK(sd_varlink_send(c, "io.test.OnewaySentinel", /* parameters= */ NULL));
/* Follow up with a regular call to verify the server is still functional. */
ASSERT_OK(sd_varlink_bind_reply(c, reply_oneway_sentinel_pong));
ASSERT_OK(sd_varlink_invoke(c, "io.test.Pong", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
static int method_with_fd_sentinel(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
_cleanup_close_ int fd1 = -EBADF, fd2 = -EBADF;
ASSERT_TRUE(FLAGS_SET(flags, SD_VARLINK_METHOD_MORE));
/* Set a sentinel so sd_varlink_reply() defers sending: each reply and its pushed fds are captured in
* the queue, and the last one is sent as the final reply when the callback returns. */
ASSERT_OK(varlink_set_sentinel(link, /* error_id= */ NULL));
/* First reply: push one fd with "alpha" content */
ASSERT_OK(fd1 = memfd_new_and_seal_string("data", "alpha"));
ASSERT_OK_EQ(sd_varlink_push_fd(link, fd1), 0);
TAKE_FD(fd1);
ASSERT_OK(sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_INTEGER("index", 0)));
/* Second reply: push one fd with "beta" content */
ASSERT_OK(fd2 = memfd_new_and_seal_string("data", "beta"));
ASSERT_OK_EQ(sd_varlink_push_fd(link, fd2), 0);
TAKE_FD(fd2);
ASSERT_OK(sd_varlink_replybo(link, SD_JSON_BUILD_PAIR_INTEGER("index", 1)));
return 0;
}
static int reply_sentinel_fd(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
int *state = ASSERT_PTR(sd_varlink_get_userdata(link));
if (*state == 0) {
/* First reply: should carry "continues" flag and fd with "alpha" */
ASSERT_NULL(error_id);
ASSERT_TRUE(FLAGS_SET(flags, SD_VARLINK_REPLY_CONTINUES));
ASSERT_EQ(sd_json_variant_integer(sd_json_variant_by_key(parameters, "index")), 0);
int fd;
ASSERT_OK(fd = sd_varlink_peek_fd(link, 0));
test_fd(fd, "alpha", STRLEN("alpha"));
(*state)++;
} else if (*state == 1) {
/* Second (final) reply: no "continues" flag, fd with "beta" */
ASSERT_NULL(error_id);
ASSERT_FALSE(FLAGS_SET(flags, SD_VARLINK_REPLY_CONTINUES));
ASSERT_EQ(sd_json_variant_integer(sd_json_variant_by_key(parameters, "index")), 1);
int fd;
ASSERT_OK(fd = sd_varlink_peek_fd(link, 0));
test_fd(fd, "beta", STRLEN("beta"));
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
} else
assert_not_reached();
return 0;
}
TEST(sentinel_with_fds) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, SD_VARLINK_SERVER_ALLOW_FD_PASSING_INPUT|SD_VARLINK_SERVER_ALLOW_FD_PASSING_OUTPUT));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.FDSentinel", method_with_fd_sentinel));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_set_allow_fd_passing_input(c, true));
ASSERT_OK(sd_varlink_set_allow_fd_passing_output(c, true));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
int state = 0;
sd_varlink_set_userdata(c, &state);
ASSERT_OK(sd_varlink_bind_reply(c, reply_sentinel_fd));
ASSERT_OK(sd_varlink_observe(c, "io.test.FDSentinel", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
static int method_with_notify_then_error(sd_varlink *link, sd_json_variant *parameters, sd_varlink_method_flags_t flags, void *userdata) {
/* Send a notify first, then return an error. The notify should be received before the error. */
ASSERT_OK(sd_varlink_notifybo(link, SD_JSON_BUILD_PAIR_STRING("status", "in-progress")));
return sd_varlink_error(link, "io.test.OperationFailed", /* parameters= */ NULL);
}
static int reply_notify_then_error(sd_varlink *link, sd_json_variant *parameters, const char *error_id, sd_varlink_reply_flags_t flags, void *userdata) {
int *state = ASSERT_PTR(sd_varlink_get_userdata(link));
if (*state == 0) {
/* First callback: should be the notify (no error, has "more" flag) */
ASSERT_NULL(error_id);
ASSERT_TRUE(FLAGS_SET(flags, SD_VARLINK_REPLY_CONTINUES));
ASSERT_STREQ(sd_json_variant_string(sd_json_variant_by_key(parameters, "status")), "in-progress");
(*state)++;
} else if (*state == 1) {
/* Second callback: should be the error */
ASSERT_STREQ(error_id, "io.test.OperationFailed");
ASSERT_FALSE(FLAGS_SET(flags, SD_VARLINK_REPLY_CONTINUES));
ASSERT_OK(sd_event_exit(sd_varlink_get_event(link), EXIT_SUCCESS));
} else
assert_not_reached();
return 0;
}
TEST(notify_then_error) {
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
ASSERT_OK(sd_event_default(&e));
_cleanup_(sd_varlink_server_unrefp) sd_varlink_server *s = NULL;
ASSERT_OK(sd_varlink_server_new(&s, 0));
ASSERT_OK(sd_varlink_server_attach_event(s, e, 0));
ASSERT_OK(sd_varlink_server_bind_method(s, "io.test.NotifyThenError", method_with_notify_then_error));
int connfd[2];
ASSERT_OK_ERRNO(socketpair(AF_UNIX, SOCK_STREAM|SOCK_NONBLOCK|SOCK_CLOEXEC, 0, connfd));
ASSERT_OK(sd_varlink_server_add_connection(s, connfd[0], /* ret= */ NULL));
_cleanup_(sd_varlink_unrefp) sd_varlink *c = NULL;
ASSERT_OK(sd_varlink_connect_fd(&c, connfd[1]));
ASSERT_OK(sd_varlink_attach_event(c, e, 0));
int state = 0;
sd_varlink_set_userdata(c, &state);
ASSERT_OK(sd_varlink_bind_reply(c, reply_notify_then_error));
ASSERT_OK(sd_varlink_observe(c, "io.test.NotifyThenError", /* parameters= */ NULL));
ASSERT_OK(sd_event_loop(e));
}
DEFINE_TEST_MAIN(LOG_DEBUG); DEFINE_TEST_MAIN(LOG_DEBUG);

View File

@ -153,6 +153,7 @@ static int vl_method_get_user_record(sd_varlink *link, sd_json_variant *paramete
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
_cleanup_(user_record_unrefp) UserRecord *hr = NULL; _cleanup_(user_record_unrefp) UserRecord *hr = NULL;
_cleanup_(lookup_parameters_done) LookupParameters p = { _cleanup_(lookup_parameters_done) LookupParameters p = {
.uid = UID_INVALID, .uid = UID_INVALID,
@ -172,16 +173,13 @@ static int vl_method_get_user_record(sd_varlink *link, sd_json_variant *paramete
* we are done'; == 0 means 'not processed, caller should process now' */ * we are done'; == 0 means 'not processed, caller should process now' */
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (uid_is_valid(p.uid)) if (uid_is_valid(p.uid))
r = userdb_by_uid(p.uid, &p.match, userdb_flags, &hr); r = userdb_by_uid(p.uid, &p.match, userdb_flags, &hr);
else if (p.name) else if (p.name)
r = userdb_by_name(p.name, &p.match, userdb_flags, &hr); r = userdb_by_name(p.name, &p.match, userdb_flags, &hr);
else { else {
_cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL; _cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *last = NULL;
r = userdb_all(&p.match, userdb_flags, &iterator); r = userdb_all(&p.match, userdb_flags, &iterator);
if (IN_SET(r, -ESRCH, -ENOLINK)) if (IN_SET(r, -ESRCH, -ENOLINK))
@ -191,7 +189,7 @@ static int vl_method_get_user_record(sd_varlink *link, sd_json_variant *paramete
* implementation detail and always return NoRecordFound in this case, since from a * implementation detail and always return NoRecordFound in this case, since from a
* client's perspective it's irrelevant if there was no entry at all or just not on * client's perspective it's irrelevant if there was no entry at all or just not on
* the service that the query was limited to. */ * the service that the query was limited to. */
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r < 0) if (r < 0)
return r; return r;
@ -204,20 +202,26 @@ static int vl_method_get_user_record(sd_varlink *link, sd_json_variant *paramete
if (r < 0) if (r < 0)
return r; return r;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (last) {
r = build_user_json(link, z, &v); r = sd_varlink_notify(link, last);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v); last = sd_json_variant_unref(last);
}
r = build_user_json(link, z, &last);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!last)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, last);
} }
if (r == -ESRCH) if (r == -ESRCH)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r == -ENOEXEC) if (r == -ENOEXEC)
return sd_varlink_error(link, "io.systemd.UserDatabase.NonMatchingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.NonMatchingRecordFound", NULL);
if (r < 0) { if (r < 0) {
@ -229,7 +233,6 @@ static int vl_method_get_user_record(sd_varlink *link, sd_json_variant *paramete
(p.name && !user_record_matches_user_name(hr, p.name))) (p.name && !user_record_matches_user_name(hr, p.name)))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_user_json(link, hr, &v); r = build_user_json(link, hr, &v);
if (r < 0) if (r < 0)
return r; return r;
@ -295,6 +298,7 @@ static int vl_method_get_group_record(sd_varlink *link, sd_json_variant *paramet
{} {}
}; };
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
_cleanup_(group_record_unrefp) GroupRecord *g = NULL; _cleanup_(group_record_unrefp) GroupRecord *g = NULL;
_cleanup_(lookup_parameters_done) LookupParameters p = { _cleanup_(lookup_parameters_done) LookupParameters p = {
.gid = GID_INVALID, .gid = GID_INVALID,
@ -313,20 +317,17 @@ static int vl_method_get_group_record(sd_varlink *link, sd_json_variant *paramet
if (r != 0) if (r != 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (gid_is_valid(p.gid)) if (gid_is_valid(p.gid))
r = groupdb_by_gid(p.gid, &p.match, userdb_flags, &g); r = groupdb_by_gid(p.gid, &p.match, userdb_flags, &g);
else if (p.name) else if (p.name)
r = groupdb_by_name(p.name, &p.match, userdb_flags, &g); r = groupdb_by_name(p.name, &p.match, userdb_flags, &g);
else { else {
_cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL; _cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *last = NULL;
r = groupdb_all(&p.match, userdb_flags, &iterator); r = groupdb_all(&p.match, userdb_flags, &iterator);
if (IN_SET(r, -ESRCH, -ENOLINK)) if (IN_SET(r, -ESRCH, -ENOLINK))
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r < 0) if (r < 0)
return r; return r;
@ -339,20 +340,26 @@ static int vl_method_get_group_record(sd_varlink *link, sd_json_variant *paramet
if (r < 0) if (r < 0)
return r; return r;
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL; if (last) {
r = build_group_json(link, z, &v); r = sd_varlink_notify(link, last);
if (r < 0) if (r < 0)
return r; return r;
r = sd_varlink_reply(link, v); last = sd_json_variant_unref(last);
}
r = build_group_json(link, z, &last);
if (r < 0) if (r < 0)
return r; return r;
} }
return 0; if (!last)
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
return sd_varlink_reply(link, last);
} }
if (r == -ESRCH) if (r == -ESRCH)
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r == -ENOEXEC) if (r == -ENOEXEC)
return sd_varlink_error(link, "io.systemd.UserDatabase.NonMatchingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.NonMatchingRecordFound", NULL);
if (r < 0) { if (r < 0) {
@ -364,7 +371,6 @@ static int vl_method_get_group_record(sd_varlink *link, sd_json_variant *paramet
(p.name && !group_record_matches_group_name(g, p.name))) (p.name && !group_record_matches_group_name(g, p.name)))
return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL); return sd_varlink_error(link, "io.systemd.UserDatabase.ConflictingRecordFound", NULL);
_cleanup_(sd_json_variant_unrefp) sd_json_variant *v = NULL;
r = build_group_json(link, g, &v); r = build_group_json(link, g, &v);
if (r < 0) if (r < 0)
return r; return r;
@ -386,6 +392,7 @@ static int vl_method_get_memberships(sd_varlink *link, sd_json_variant *paramete
{} {}
}; };
_cleanup_free_ char *last_user_name = NULL, *last_group_name = NULL;
_cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL; _cleanup_(userdb_iterator_freep) UserDBIterator *iterator = NULL;
MembershipLookupParameters p = {}; MembershipLookupParameters p = {};
UserDBFlags userdb_flags; UserDBFlags userdb_flags;
@ -401,10 +408,6 @@ static int vl_method_get_memberships(sd_varlink *link, sd_json_variant *paramete
if (r != 0) if (r != 0)
return r; return r;
r = varlink_set_sentinel(link, "io.systemd.UserDatabase.NoRecordFound");
if (r < 0)
return r;
if (p.group_name) if (p.group_name)
r = membershipdb_by_group(p.group_name, userdb_flags, &iterator); r = membershipdb_by_group(p.group_name, userdb_flags, &iterator);
else if (p.user_name) else if (p.user_name)
@ -412,7 +415,7 @@ static int vl_method_get_memberships(sd_varlink *link, sd_json_variant *paramete
else else
r = membershipdb_all(userdb_flags, &iterator); r = membershipdb_all(userdb_flags, &iterator);
if (IN_SET(r, -ESRCH, -ENOLINK)) if (IN_SET(r, -ESRCH, -ENOLINK))
return 0; return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
if (r < 0) if (r < 0)
return r; return r;
@ -429,15 +432,32 @@ static int vl_method_get_memberships(sd_varlink *link, sd_json_variant *paramete
if (p.group_name && p.user_name && !streq(group_name, p.group_name)) if (p.group_name && p.user_name && !streq(group_name, p.group_name))
continue; continue;
r = sd_varlink_replybo( if (last_user_name) {
link, assert(last_group_name);
SD_JSON_BUILD_PAIR_STRING("userName", user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", group_name)); r = sd_varlink_notifybo(
if (r < 0) link,
return r; SD_JSON_BUILD_PAIR_STRING("userName", last_user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", last_group_name));
if (r < 0)
return r;
}
free_and_replace(last_user_name, user_name);
free_and_replace(last_group_name, group_name);
} }
return 0; if (!last_user_name) {
assert(!last_group_name);
return sd_varlink_error(link, "io.systemd.UserDatabase.NoRecordFound", NULL);
}
assert(last_group_name);
return sd_varlink_replybo(
link,
SD_JSON_BUILD_PAIR_STRING("userName", last_user_name),
SD_JSON_BUILD_PAIR_STRING("groupName", last_group_name));
} }
static int process_connection(sd_varlink_server *server, int _fd) { static int process_connection(sd_varlink_server *server, int _fd) {