Move dropin listing to shared

No functional change. This is in preparation for using this in
systemctl in the future.
This commit is contained in:
Zbigniew Jędrzejewski-Szmek 2014-12-14 23:12:40 -05:00
parent 6a057a9985
commit 1a7f1b385c
5 changed files with 190 additions and 126 deletions

View file

@ -31,137 +31,27 @@
#include "load-fragment.h"
#include "conf-files.h"
static int iterate_dir(
Unit *u,
const char *path,
static int add_dependency_consumer(
UnitDependency dependency,
char ***strv) {
_cleanup_closedir_ DIR *d = NULL;
int r;
assert(u);
assert(path);
/* The config directories are special, since the order of the
* drop-ins matters */
if (dependency < 0) {
r = strv_extend(strv, path);
if (r < 0)
return log_oom();
return 0;
}
d = opendir(path);
if (!d) {
if (errno == ENOENT)
return 0;
log_error_errno(errno, "Failed to open directory %s: %m", path);
return -errno;
}
for (;;) {
struct dirent *de;
_cleanup_free_ char *f = NULL;
errno = 0;
de = readdir(d);
if (!de && errno != 0)
return log_error_errno(errno, "Failed to read directory %s: %m", path);
if (!de)
break;
if (ignore_file(de->d_name))
continue;
f = strjoin(path, "/", de->d_name, NULL);
if (!f)
return log_oom();
r = unit_add_dependency_by_name(u, dependency, de->d_name, f, true);
if (r < 0)
log_error_errno(r, "Cannot add dependency %s to %s, ignoring: %m", de->d_name, u->id);
}
return 0;
}
static int process_dir(
Unit *u,
const char *unit_path,
const char *name,
const char *suffix,
UnitDependency dependency,
char ***strv) {
_cleanup_free_ char *path = NULL;
assert(u);
assert(unit_path);
assert(name);
assert(suffix);
path = strjoin(unit_path, "/", name, suffix, NULL);
if (!path)
return log_oom();
if (!u->manager->unit_path_cache || set_get(u->manager->unit_path_cache, path))
iterate_dir(u, path, dependency, strv);
if (u->instance) {
_cleanup_free_ char *template = NULL, *p = NULL;
/* Also try the template dir */
template = unit_name_template(name);
if (!template)
return log_oom();
p = strjoin(unit_path, "/", template, suffix, NULL);
if (!p)
return log_oom();
if (!u->manager->unit_path_cache || set_get(u->manager->unit_path_cache, p))
iterate_dir(u, p, dependency, strv);
}
return 0;
}
char **unit_find_dropin_paths(Unit *u) {
_cleanup_strv_free_ char **strv = NULL;
char **configs = NULL;
Iterator i;
char *t;
const char *entry,
const char* filepath,
void *arg) {
Unit *u = arg;
int r;
assert(u);
SET_FOREACH(t, u->names, i) {
char **p;
r = unit_add_dependency_by_name(u, dependency, entry, filepath, true);
if (r < 0)
log_error_errno(r, "Cannot add dependency %s to %s, ignoring: %m", entry, u->id);
STRV_FOREACH(p, u->manager->lookup_paths.unit_path)
process_dir(u, *p, t, ".d", _UNIT_DEPENDENCY_INVALID, &strv);
}
if (strv_isempty(strv))
return NULL;
r = conf_files_list_strv(&configs, ".conf", NULL, (const char**) strv);
if (r < 0) {
log_error_errno(r, "Failed to get list of configuration files: %m");
strv_free(configs);
return NULL;
}
return configs;
return 0;
}
int unit_load_dropin(Unit *u) {
Iterator i;
char *t, **f;
int r;
assert(u);
@ -171,13 +61,15 @@ int unit_load_dropin(Unit *u) {
char **p;
STRV_FOREACH(p, u->manager->lookup_paths.unit_path) {
process_dir(u, *p, t, ".wants", UNIT_WANTS, NULL);
process_dir(u, *p, t, ".requires", UNIT_REQUIRES, NULL);
unit_file_process_dir(u->manager->unit_path_cache, *p, t, ".wants", UNIT_WANTS,
add_dependency_consumer, u, NULL);
unit_file_process_dir(u->manager->unit_path_cache, *p, t, ".requires", UNIT_REQUIRES,
add_dependency_consumer, u, NULL);
}
}
u->dropin_paths = unit_find_dropin_paths(u);
if (!u->dropin_paths)
r = unit_find_dropin_paths(u, &u->dropin_paths);
if (r <= 0)
return 0;
STRV_FOREACH(f, u->dropin_paths) {

View file

@ -22,8 +22,15 @@
***/
#include "unit.h"
#include "dropin.h"
/* Read service data supplementary drop-in directories */
char **unit_find_dropin_paths(Unit *u);
static inline int unit_find_dropin_paths(Unit *u, char ***paths) {
return unit_file_find_dropin_paths(u->manager->lookup_paths.unit_path,
u->manager->unit_path_cache,
u->names,
paths);
}
int unit_load_dropin(Unit *u);

View file

@ -2894,7 +2894,7 @@ bool unit_need_daemon_reload(Unit *u) {
return true;
}
t = unit_find_dropin_paths(u);
(void) unit_find_dropin_paths(u, &t);
loaded_cnt = strv_length(t);
current_cnt = strv_length(u->dropin_paths);

View file

@ -21,8 +21,10 @@
#include "dropin.h"
#include "util.h"
#include "strv.h"
#include "mkdir.h"
#include "fileio-label.h"
#include "conf-files.h"
int drop_in_file(const char *dir, const char *unit, unsigned level,
const char *name, char **_p, char **_q) {
@ -100,3 +102,137 @@ int write_drop_in_format(const char *dir, const char *unit, unsigned level,
return write_drop_in(dir, unit, level, name, p);
}
static int iterate_dir(
const char *path,
UnitDependency dependency,
dependency_consumer_t consumer,
void *arg,
char ***strv) {
_cleanup_closedir_ DIR *d = NULL;
int r;
assert(path);
/* The config directories are special, since the order of the
* drop-ins matters */
if (dependency < 0) {
r = strv_extend(strv, path);
if (r < 0)
return log_oom();
return 0;
}
assert(consumer);
d = opendir(path);
if (!d) {
if (errno == ENOENT)
return 0;
log_error_errno(errno, "Failed to open directory %s: %m", path);
return -errno;
}
for (;;) {
struct dirent *de;
_cleanup_free_ char *f = NULL;
errno = 0;
de = readdir(d);
if (!de && errno != 0)
return log_error_errno(errno, "Failed to read directory %s: %m", path);
if (!de)
break;
if (ignore_file(de->d_name))
continue;
f = strjoin(path, "/", de->d_name, NULL);
if (!f)
return log_oom();
r = consumer(dependency, de->d_name, f, arg);
if (r < 0)
return r;
}
return 0;
}
int unit_file_process_dir(
Set * unit_path_cache,
const char *unit_path,
const char *name,
const char *suffix,
UnitDependency dependency,
dependency_consumer_t consumer,
void *arg,
char ***strv) {
_cleanup_free_ char *path = NULL;
assert(unit_path);
assert(name);
assert(suffix);
path = strjoin(unit_path, "/", name, suffix, NULL);
if (!path)
return log_oom();
if (!unit_path_cache || set_get(unit_path_cache, path))
iterate_dir(path, dependency, consumer, arg, strv);
if (unit_name_is_instance(name)) {
_cleanup_free_ char *template = NULL, *p = NULL;
/* Also try the template dir */
template = unit_name_template(name);
if (!template)
return log_oom();
p = strjoin(unit_path, "/", template, suffix, NULL);
if (!p)
return log_oom();
if (!unit_path_cache || set_get(unit_path_cache, p))
iterate_dir(p, dependency, consumer, arg, strv);
}
return 0;
}
int unit_file_find_dropin_paths(
char **lookup_path,
Set *unit_path_cache,
Set *names,
char ***paths) {
_cleanup_strv_free_ char **strv = NULL, **ans = NULL;
Iterator i;
char *t;
int r;
assert(paths);
SET_FOREACH(t, names, i) {
char **p;
STRV_FOREACH(p, lookup_path)
unit_file_process_dir(unit_path_cache, *p, t, ".d", _UNIT_DEPENDENCY_INVALID, NULL, NULL, &strv);
}
if (strv_isempty(strv))
return 0;
r = conf_files_list_strv(&ans, ".conf", NULL, (const char**) strv);
if (r < 0)
return log_warning_errno(r, "Failed to get list of configuration files: %m");
*paths = ans;
ans = NULL;
return 1;
}

View file

@ -22,6 +22,8 @@
***/
#include "macro.h"
#include "set.h"
#include "unit-name.h"
int drop_in_file(const char *dir, const char *unit, unsigned level,
const char *name, char **_p, char **_q);
@ -31,3 +33,30 @@ int write_drop_in(const char *dir, const char *unit, unsigned level,
int write_drop_in_format(const char *dir, const char *unit, unsigned level,
const char *name, const char *format, ...) _printf_(5, 6);
/**
* This callback will be called for each directory entry @entry,
* with @filepath being the full path to the entry.
*
* If return value is negative, loop will be aborted.
*/
typedef int (*dependency_consumer_t)(UnitDependency dependency,
const char *entry,
const char* filepath,
void *arg);
int unit_file_process_dir(
Set * unit_path_cache,
const char *unit_path,
const char *name,
const char *suffix,
UnitDependency dependency,
dependency_consumer_t consumer,
void *arg,
char ***strv);
int unit_file_find_dropin_paths(
char **lookup_path,
Set *unit_path_cache,
Set *names,
char ***paths);