#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <pwd.h>

#include "manager.h"

#include "unit.h"

#include "unit-name.h"

#include "unit-printf.h"

#include "install.h"

#include "specifier.h"

#include "util.h"

#include "macro.h"

#include "test-helper.h"

static void test_replacements(void) {

#define expect(pattern, repl, expected) \

{ \

_cleanup_free_ char *t = \

unit_name_replace_instance(pattern, repl); \

puts(t); \

assert_se(streq(t, expected)); \

}

expect("foo@.service", "waldo", "foo@waldo.service");

expect("foo@xyz.service", "waldo", "foo@waldo.service");

expect("xyz", "waldo", "xyz");

expect("", "waldo", "");

expect("foo.service", "waldo", "foo.service");

expect(".service", "waldo", ".service");

expect("foo@", "waldo", "foo@waldo");

expect("@bar", "waldo", "@waldo");

puts("-------------------------------------------------");

#undef expect

#define expect(path, suffix, expected) \

{ \

_cleanup_free_ char *k, *t = \

unit_name_from_path(path, suffix); \

puts(t); \

k = unit_name_to_path(t); \

puts(k); \

assert_se(streq(k, expected ? expected : path)); \

}

expect("/waldo", ".mount", NULL);

expect("/waldo/quuix", ".mount", NULL);

expect("/waldo/quuix/", ".mount", "/waldo/quuix");

expect("/", ".mount", NULL);

expect("///", ".mount", "/");

puts("-------------------------------------------------");

#undef expect

#define expect(pattern, path, suffix, expected) \

{ \

_cleanup_free_ char *t = \

unit_name_from_path_instance(pattern, path, suffix); \

puts(t); \

assert_se(streq(t, expected)); \

}

expect("waldo", "/waldo", ".mount", "waldo@waldo.mount");

expect("waldo", "/waldo////quuix////", ".mount", "waldo@waldo-quuix.mount");

expect("waldo", "/", ".mount", "waldo@-.mount");

expect("wa--ldo", "/--", ".mount", "wa--ldo@\\x2d\\x2d.mount");

puts("-------------------------------------------------");

#undef expect

#define expect(pattern) \

{ \

_cleanup_free_ char *k, *t; \

assert_se(t = unit_name_mangle(pattern, MANGLE_NOGLOB)); \

assert_se(k = unit_name_mangle(t, MANGLE_NOGLOB)); \

puts(t); \

assert_se(streq(t, k)); \

}

expect("/home");

expect("/dev/sda");

expect("üxknürz.service");

expect("foobar-meh...waldi.service");

expect("_____####----.....service");

expect("_____##@;;;,,,##----.....service");

expect("xxx@@@@/////\\\\\\\\\\yyy.service");

#undef expect

}

static int test_unit_printf(void) {

Manager *m = NULL;

Unit *u, *u2;

int r;

_cleanup_free_ char *mid, *bid, *host, *root_uid;

struct passwd *root;

assert_se(specifier_machine_id('m', NULL, NULL, &mid) >= 0 && mid);

assert_se(specifier_boot_id('b', NULL, NULL, &bid) >= 0 && bid);

assert_se((host = gethostname_malloc()));

assert_se((root = getpwnam("root")));

assert_se(asprintf(&root_uid, "%d", (int) root->pw_uid) > 0);

r = manager_new(SYSTEMD_USER, true, &m);

if (r == -EPERM || r == -EACCES || r == -EADDRINUSE) {

puts("manager_new: Permission denied. Skipping test.");

return EXIT_TEST_SKIP;

}

assert_se(r == 0);

#define expect(unit, pattern, expected) \

{ \

char *e; \

_cleanup_free_ char *t; \

assert_se(unit_full_printf(unit, pattern, &t) >= 0); \

printf("result: %s\nexpect: %s\n", t, expected); \

if ((e = endswith(expected, "*"))) \

assert_se(strncmp(t, e, e-expected)); \

else \

assert_se(streq(t, expected)); \

}

assert_se(setenv("USER", "root", 1) == 0);

assert_se(setenv("HOME", "/root", 1) == 0);

assert_se(setenv("XDG_RUNTIME_DIR", "/run/user/1/", 1) == 0);

assert_se(u = unit_new(m, sizeof(Service)));

assert_se(unit_add_name(u, "blah.service") == 0);

assert_se(unit_add_name(u, "blah.service") == 0);

/* general tests */

expect(u, "%%", "%");

expect(u, "%%s", "%s");

expect(u, "%", ""); // REALLY?

/* normal unit */

expect(u, "%n", "blah.service");

expect(u, "%N", "blah");

expect(u, "%p", "blah");

expect(u, "%P", "blah");

expect(u, "%i", "");

expect(u, "%u", root->pw_name);

expect(u, "%U", root_uid);

expect(u, "%h", root->pw_dir);

expect(u, "%m", mid);

expect(u, "%b", bid);

expect(u, "%H", host);

expect(u, "%t", "/run/user/*");

/* templated */

assert_se(u2 = unit_new(m, sizeof(Service)));

assert_se(unit_add_name(u2, "blah@foo-foo.service") == 0);

assert_se(unit_add_name(u2, "blah@foo-foo.service") == 0);

expect(u2, "%n", "blah@foo-foo.service");

expect(u2, "%N", "blah@foo-foo");

expect(u2, "%p", "blah");

expect(u2, "%P", "blah");

expect(u2, "%i", "foo-foo");

expect(u2, "%I", "foo/foo");

expect(u2, "%u", root->pw_name);

expect(u2, "%U", root_uid);

expect(u2, "%h", root->pw_dir);

expect(u2, "%m", mid);

expect(u2, "%b", bid);

expect(u2, "%H", host);

expect(u2, "%t", "/run/user/*");

manager_free(m);

return 0;

}

static void test_unit_instance_is_valid(void) {

assert_se(unit_instance_is_valid("fooBar"));

assert_se(unit_instance_is_valid("foo-bar"));

assert_se(unit_instance_is_valid("foo.stUff"));

assert_se(unit_instance_is_valid("fOo123.stuff"));

assert_se(unit_instance_is_valid("@f_oo123.Stuff"));

assert_se(!unit_instance_is_valid("$¢£"));

assert_se(!unit_instance_is_valid(""));

assert_se(!unit_instance_is_valid("foo bar"));

assert_se(!unit_instance_is_valid("foo/bar"));

}

static void test_unit_prefix_is_valid(void) {

assert_se(unit_prefix_is_valid("fooBar"));

assert_se(unit_prefix_is_valid("foo-bar"));

assert_se(unit_prefix_is_valid("foo.stUff"));

assert_se(unit_prefix_is_valid("fOo123.stuff"));

assert_se(unit_prefix_is_valid("foo123.Stuff"));

assert_se(!unit_prefix_is_valid("$¢£"));

assert_se(!unit_prefix_is_valid(""));

assert_se(!unit_prefix_is_valid("foo bar"));

assert_se(!unit_prefix_is_valid("foo/bar"));

assert_se(!unit_prefix_is_valid("@foo-bar"));

}

static void test_unit_name_change_suffix(void) {

char *r;

r = unit_name_change_suffix("foo.bar", ".service");

assert_se(r);

assert_se(streq(r, "foo.service"));

free(r);

r = unit_name_change_suffix("foo@stuff.bar", ".boo");

assert_se(r);

assert_se(streq(r, "foo@stuff.boo"));

free(r);

}

static void test_unit_name_build(void) {

char *r;

r = unit_name_build("foo", "bar", ".service");

assert_se(r);

assert_se(streq(r, "foo@bar.service"));

free(r);

r = unit_name_build("fo0-stUff_b", "bar", ".mount");

assert_se(r);

assert_se(streq(r, "fo0-stUff_b@bar.mount"));

free(r);

r = unit_name_build("foo", NULL, ".service");

assert_se(r);

assert_se(streq(r, "foo.service"));

free(r);

}

static void test_unit_name_is_instance(void) {

assert_se(unit_name_is_instance("a@b.service"));

assert_se(unit_name_is_instance("a-c_c01Aj@b05Dii_-oioi.service"));

assert_se(!unit_name_is_instance("a.service"));

assert_se(!unit_name_is_instance("junk"));

assert_se(!unit_name_is_instance(""));

}

static void test_build_subslice(void) {

char *a;

char *b;

assert_se(build_subslice("-.slice", "foo", &a) >= 0);

assert_se(build_subslice(a, "bar", &b) >= 0);

free(a);

assert_se(build_subslice(b, "barfoo", &a) >= 0);

free(b);

assert_se(build_subslice(a, "foobar", &b) >= 0);

free(a);

assert_se(streq(b, "foo-bar-barfoo-foobar.slice"));

free(b);

assert_se(build_subslice("foo.service", "bar", &a) < 0);

assert_se(build_subslice("foo", "bar", &a) < 0);

}

static void test_unit_name_to_instance(void) {

char *instance;

int r;

r = unit_name_to_instance("foo@bar.service", &instance);

assert_se(r >= 0);

assert_se(streq(instance, "bar"));

free(instance);

r = unit_name_to_instance("fo0-stUff_b@b.e", &instance);

assert_se(r >= 0);

assert_se(streq(instance, "b"));

free(instance);

r = unit_name_to_instance("foo.bar", &instance);

assert_se(r >= 0);

assert_se(!instance);

r = unit_name_to_instance("fooj@unk", &instance);

assert_se(r < 0);

}

static void test_unit_name_escape(void) {

_cleanup_free_ char *r;

r = unit_name_escape("ab+-c.a/bc@foo.service");

assert_se(r);

assert_se(streq(r, "ab\\x2b\\x2dc.a-bc\\x40foo.service"));

}

int main(int argc, char* argv[]) {

int rc = 0;

test_replacements();

TEST_REQ_RUNNING_SYSTEMD(rc = test_unit_printf());

test_unit_instance_is_valid();

test_unit_prefix_is_valid();

test_unit_name_change_suffix();

test_unit_name_build();

test_unit_name_is_instance();

test_build_subslice();

test_unit_name_to_instance();

test_unit_name_escape();

return rc;

}