272 lines
8.1 KiB
C
272 lines
8.1 KiB
C
/***
|
|
This file is part of systemd.
|
|
|
|
Copyright 2014 Ronny Chevalier
|
|
|
|
systemd is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation; either version 2.1 of the License, or
|
|
(at your option) any later version.
|
|
|
|
systemd is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with systemd; If not, see <http://www.gnu.org/licenses/>.
|
|
***/
|
|
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
|
|
#include "unit.h"
|
|
#include "manager.h"
|
|
#include "util.h"
|
|
#include "macro.h"
|
|
#include "strv.h"
|
|
#include "mkdir.h"
|
|
|
|
typedef void (*test_function_t)(Manager *m);
|
|
|
|
static int setup_test(Manager **m) {
|
|
char **tests_path = STRV_MAKE("exists", "existsglobFOOBAR", "changed", "modified", "unit",
|
|
"directorynotempty", "makedirectory");
|
|
char **test_path;
|
|
Manager *tmp;
|
|
int r;
|
|
|
|
assert_se(m);
|
|
|
|
r = manager_new(SYSTEMD_USER, true, &tmp);
|
|
if (IN_SET(r, -EPERM, -EACCES, -EADDRINUSE, -EHOSTDOWN, -ENOENT)) {
|
|
printf("Skipping test: manager_new: %s", strerror(-r));
|
|
return -EXIT_TEST_SKIP;
|
|
}
|
|
assert_se(r >= 0);
|
|
assert_se(manager_startup(tmp, NULL, NULL) >= 0);
|
|
|
|
STRV_FOREACH(test_path, tests_path) {
|
|
rm_rf_dangerous(strappenda("/tmp/test-path_", *test_path), false, true, false);
|
|
}
|
|
|
|
*m = tmp;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void shutdown_test(Manager *m) {
|
|
assert_se(m);
|
|
|
|
manager_free(m);
|
|
}
|
|
|
|
static void check_stop_unlink(Manager *m, Unit *unit, const char *test_path, const char *service_name) {
|
|
_cleanup_free_ char *tmp = NULL;
|
|
Unit *service_unit = NULL;
|
|
Service *service = NULL;
|
|
usec_t ts;
|
|
usec_t timeout = 2 * USEC_PER_SEC;
|
|
|
|
assert_se(m);
|
|
assert_se(unit);
|
|
assert_se(test_path);
|
|
|
|
if (!service_name) {
|
|
assert_se(tmp = strreplace(unit->id, ".path", ".service"));
|
|
service_unit = manager_get_unit(m, tmp);
|
|
} else
|
|
service_unit = manager_get_unit(m, service_name);
|
|
assert_se(service_unit);
|
|
service = SERVICE(service_unit);
|
|
|
|
ts = now(CLOCK_MONOTONIC);
|
|
/* We process events until the service related to the path has been successfully started */
|
|
while(service->result != SERVICE_SUCCESS || service->state != SERVICE_START) {
|
|
usec_t n;
|
|
int r;
|
|
|
|
r = sd_event_run(m->event, 100 * USEC_PER_MSEC);
|
|
assert_se(r >= 0);
|
|
|
|
printf("%s: state = %s; result = %s \n",
|
|
service_unit->id,
|
|
service_state_to_string(service->state),
|
|
service_result_to_string(service->result));
|
|
|
|
|
|
/* But we timeout if the service has not been started in the allocated time */
|
|
n = now(CLOCK_MONOTONIC);
|
|
if (ts + timeout < n) {
|
|
log_error("Test timeout when testing %s", unit->id);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
assert_se(UNIT_VTABLE(unit)->stop(unit) >= 0);
|
|
rm_rf_dangerous(test_path, false, true, false);
|
|
}
|
|
|
|
static void test_path_exists(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_exists";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(manager_load_unit(m, "path-exists.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_existsglob(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_existsglobFOOBAR";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
assert_se(manager_load_unit(m, "path-existsglob.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_changed(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_changed";
|
|
FILE *f;
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-changed.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
f = fopen(test_path, "w");
|
|
assert_se(f);
|
|
fclose(f);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_modified(Manager *m) {
|
|
_cleanup_fclose_ FILE *f = NULL;
|
|
const char *test_path = "/tmp/test-path_modified";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-modified.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
f = fopen(test_path, "w");
|
|
assert_se(f);
|
|
fputs("test", f);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_unit(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_unit";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(manager_load_unit(m, "path-unit.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
assert_se(touch(test_path) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, "path-mycustomunit.service");
|
|
}
|
|
|
|
static void test_path_directorynotempty(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_directorynotempty/";
|
|
Unit *unit = NULL;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-directorynotempty.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
/* MakeDirectory default to no */
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(mkdir_p(test_path, 0755) >= 0);
|
|
assert_se(touch(strappenda(test_path, "test_file")) >= 0);
|
|
|
|
check_stop_unlink(m, unit, test_path, NULL);
|
|
}
|
|
|
|
static void test_path_makedirectory_directorymode(Manager *m) {
|
|
const char *test_path = "/tmp/test-path_makedirectory/";
|
|
Unit *unit = NULL;
|
|
struct stat s;
|
|
|
|
assert_se(m);
|
|
|
|
assert_se(access(test_path, F_OK) < 0);
|
|
|
|
assert_se(manager_load_unit(m, "path-makedirectory.path", NULL, NULL, &unit) >= 0);
|
|
assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
|
|
|
|
/* Check if the directory has been created */
|
|
assert_se(access(test_path, F_OK) >= 0);
|
|
|
|
/* Check the mode we specified with DirectoryMode=0744 */
|
|
assert_se(stat(test_path, &s) >= 0);
|
|
assert_se((s.st_mode & S_IRWXU) == 0700);
|
|
assert_se((s.st_mode & S_IRWXG) == 0040);
|
|
assert_se((s.st_mode & S_IRWXO) == 0004);
|
|
|
|
assert_se(UNIT_VTABLE(unit)->stop(unit) >= 0);
|
|
rm_rf_dangerous(test_path, false, true, false);
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
test_function_t tests[] = {
|
|
test_path_exists,
|
|
test_path_existsglob,
|
|
test_path_changed,
|
|
test_path_modified,
|
|
test_path_unit,
|
|
test_path_directorynotempty,
|
|
test_path_makedirectory_directorymode,
|
|
NULL,
|
|
};
|
|
test_function_t *test = NULL;
|
|
Manager *m = NULL;
|
|
|
|
log_parse_environment();
|
|
log_open();
|
|
|
|
/* It is needed otherwise cgroup creation fails */
|
|
if (getuid() != 0)
|
|
return EXIT_TEST_SKIP;
|
|
|
|
assert_se(set_unit_path(TEST_DIR ":") >= 0);
|
|
|
|
for (test = tests; test && *test; test++) {
|
|
int r;
|
|
|
|
/* We create a clean environment for each test */
|
|
r = setup_test(&m);
|
|
if (r < 0)
|
|
return -r;
|
|
|
|
(*test)(m);
|
|
|
|
shutdown_test(m);
|
|
}
|
|
|
|
return 0;
|
|
}
|