#include <errno.h>

#include <stdio.h>

#include <unistd.h>

#include "util.h"

#include "mkdir.h"

#include "strv.h"

#include "path-util.h"

#include "path-lookup.h"

#include "log.h"

#include "unit-name.h"

#include "special.h"

#include "hashmap.h"

#include "set.h"

#include "install.h"

typedef enum RunlevelType {

RUNLEVEL_UP,

RUNLEVEL_DOWN

} RunlevelType;

static const struct {

const char *path;

const char *target;

const RunlevelType type;

} rcnd_table[] = {

/* Standard SysV runlevels for start-up */

{ "rc1.d", SPECIAL_RESCUE_TARGET, RUNLEVEL_UP },

{ "rc2.d", SPECIAL_MULTI_USER_TARGET, RUNLEVEL_UP },

{ "rc3.d", SPECIAL_MULTI_USER_TARGET, RUNLEVEL_UP },

{ "rc4.d", SPECIAL_MULTI_USER_TARGET, RUNLEVEL_UP },

{ "rc5.d", SPECIAL_GRAPHICAL_TARGET, RUNLEVEL_UP },

/* Standard SysV runlevels for shutdown */

{ "rc0.d", SPECIAL_POWEROFF_TARGET, RUNLEVEL_DOWN },

{ "rc6.d", SPECIAL_REBOOT_TARGET, RUNLEVEL_DOWN }

/* Note that the order here matters, as we read the

};

const char *arg_dest = "/tmp";

typedef struct SysvStub {

char *name;

char *path;

char *description;

int sysv_start_priority;

char *pid_file;

char **before;

char **after;

char **wants;

char **wanted_by;

char **conflicts;

bool has_lsb;

bool reload;

} SysvStub;

static void free_sysvstub(SysvStub *s) {

free(s->name);

free(s->path);

free(s->description);

free(s->pid_file);

strv_free(s->before);

strv_free(s->after);

strv_free(s->wants);

strv_free(s->wanted_by);

strv_free(s->conflicts);

free(s);

}

DEFINE_TRIVIAL_CLEANUP_FUNC(SysvStub*, free_sysvstub);

static void free_sysvstub_hashmapp(Hashmap **h) {

SysvStub *stub;

while ((stub = hashmap_steal_first(*h)))

free_sysvstub(stub);

hashmap_free(*h);

}

static int add_symlink(const char *service, const char *where) {

_cleanup_free_ char *from = NULL, *to = NULL;

int r;

assert(service);

assert(where);

from = strjoin(arg_dest, "/", service, NULL);

if (!from)

return log_oom();

to = strjoin(arg_dest, "/", where, ".wants/", service, NULL);

if (!to)

return log_oom();

mkdir_parents_label(to, 0755);

r = symlink(from, to);

if (r < 0) {

if (errno == EEXIST)

return 0;

return -errno;

}

return 1;

}

static int add_alias(const char *service, const char *alias) {

_cleanup_free_ char *link = NULL;

int r;

assert(service);

assert(alias);

link = strjoin(arg_dest, "/", alias, NULL);

if (!link)

return log_oom();

r = symlink(service, link);

if (r < 0) {

if (errno == EEXIST)

return 0;

return -errno;

}

return 1;

}

static int generate_unit_file(SysvStub *s) {

char **p;

_cleanup_fclose_ FILE *f = NULL;

_cleanup_free_ char *unit = NULL,

*before = NULL, *after = NULL,

*wants = NULL, *conflicts = NULL;

int r;

before = strv_join(s->before, " ");

after = strv_join(s->after, " ");

wants = strv_join(s->wants, " ");

conflicts = strv_join(s->conflicts, " ");

unit = strjoin(arg_dest, "/", s->name, NULL);

if (!before || !after || !wants || !conflicts || !unit)

return log_oom();

/* We might already have a symlink with the same name from a Provides:,

if (is_symlink(unit) > 0) {

log_warning("Overwriting existing symlink %s with real service", unit);

(void) unlink(unit);

}

f = fopen(unit, "wxe");

if (!f)

return log_error_errno(errno, "Failed to create unit file %s: %m", unit);

fprintf(f,

"# Automatically generated by systemd-sysv-generator\n\n"

"[Unit]\n"

"Documentation=man:systemd-sysv-generator(8)\n"

"SourcePath=%s\n"

"Description=%s\n",

s->path, s->description);

if (!isempty(before))

fprintf(f, "Before=%s\n", before);

if (!isempty(after))

fprintf(f, "After=%s\n", after);

if (!isempty(wants))

fprintf(f, "Wants=%s\n", wants);

if (!isempty(conflicts))

fprintf(f, "Conflicts=%s\n", conflicts);

fprintf(f,

"\n[Service]\n"

"Type=forking\n"

"Restart=no\n"

"TimeoutSec=5min\n"

"IgnoreSIGPIPE=no\n"

"KillMode=process\n"

"GuessMainPID=no\n"

"RemainAfterExit=%s\n",

yes_no(!s->pid_file));

if (s->pid_file)

fprintf(f, "PIDFile=%s\n", s->pid_file);

fprintf(f,

"ExecStart=%s start\n"

"ExecStop=%s stop\n",

s->path, s->path);

if (s->reload)

fprintf(f, "ExecReload=%s reload\n", s->path);

STRV_FOREACH(p, s->wanted_by) {

r = add_symlink(s->name, *p);

if (r < 0)

log_error_errno(r, "Failed to create 'Wants' symlink to %s: %m", *p);

}

return 0;

}

static bool usage_contains_reload(const char *line) {

return (strcasestr(line, "{reload|") ||

strcasestr(line, "{reload}") ||

strcasestr(line, "{reload\"") ||

strcasestr(line, "|reload|") ||

strcasestr(line, "|reload}") ||

strcasestr(line, "|reload\""));

}

static char *sysv_translate_name(const char *name) {

_cleanup_free_ char *c = NULL;

char *res;

c = strdup(name);

if (!c)

return NULL;

res = endswith(c, ".sh");

if (res)

*res = 0;

if (unit_name_mangle(c, UNIT_NAME_NOGLOB, &res) < 0)

return NULL;

return res;

}

static int sysv_translate_facility(const char *name, const char *filename, char **_r) {

/* We silently ignore the $ prefix here. According to the LSB

static const char * const table[] = {

/* LSB defined facilities */

"local_fs", NULL,

"network", SPECIAL_NETWORK_ONLINE_TARGET,

"named", SPECIAL_NSS_LOOKUP_TARGET,

"portmap", SPECIAL_RPCBIND_TARGET,

"remote_fs", SPECIAL_REMOTE_FS_TARGET,

"syslog", NULL,

"time", SPECIAL_TIME_SYNC_TARGET,

};

char *filename_no_sh, *e, *r;

const char *n;

unsigned i;

assert(name);

assert(_r);

n = *name == '$' ? name + 1 : name;

for (i = 0; i < ELEMENTSOF(table); i += 2) {

if (!streq(table[i], n))

continue;

if (!table[i+1])

return 0;

r = strdup(table[i+1]);

if (!r)

return log_oom();

goto finish;

}

/* strip ".sh" suffix from file name for comparison */

filename_no_sh = strdupa(filename);

e = endswith(filename_no_sh, ".sh");

if (e) {

*e = '\0';

filename = filename_no_sh;

}

/* If we don't know this name, fallback heuristics to figure

if (*name == '$') {

int k;

/* Facilities starting with $ are most likely targets */

k = unit_name_build(n, NULL, ".target", &r);

if (k < 0)

return k;

} else if (streq_ptr(n, filename))

/* Names equaling the file name of the services are redundant */

return 0;

else

/* Everything else we assume to be normal service names */

r = sysv_translate_name(n);

if (!r)

return -ENOMEM;

finish:

*_r = r;

return 1;

}

static int handle_provides(SysvStub *s, unsigned line, const char *full_text, const char *text) {

const char *word, *state_;

size_t z;

int r;

FOREACH_WORD_QUOTED(word, z, text, state_) {

_cleanup_free_ char *n = NULL, *m = NULL;

UnitType t;

n = strndup(word, z);

if (!n)

return log_oom();

r = sysv_translate_facility(n, basename(s->path), &m);

if (r < 0)

return r;

if (r == 0)

continue;

t = unit_name_to_type(m);

if (t == UNIT_SERVICE) {

log_debug("Adding Provides: alias '%s' for '%s'", m, s->name);

r = add_alias(s->name, m);

if (r < 0)

log_warning_errno(r, "[%s:%u] Failed to add LSB Provides name %s, ignoring: %m", s->path, line, m);

} else if (t == UNIT_TARGET) {

/* NB: SysV targets which are provided by a

r = strv_extend(&s->before, m);

if (r < 0)

return log_oom();

r = strv_extend(&s->wants, m);

if (r < 0)

return log_oom();

if (streq(m, SPECIAL_NETWORK_ONLINE_TARGET)) {

r = strv_extend(&s->before, SPECIAL_NETWORK_TARGET);

if (r < 0)

return log_oom();

}

} else if (t == _UNIT_TYPE_INVALID)

log_warning("Unit name '%s' is invalid", m);

else

log_warning("Unknown unit type for unit '%s'", m);

}

if (!isempty(state_))

log_error("[%s:%u] Trailing garbage in Provides, ignoring.", s->path, line);

return 0;

}

static int handle_dependencies(SysvStub *s, unsigned line, const char *full_text, const char *text) {

const char *word, *state_;

size_t z;

int r;

FOREACH_WORD_QUOTED(word, z, text, state_) {

_cleanup_free_ char *n = NULL, *m = NULL;

bool is_before;

n = strndup(word, z);

if (!n)

return log_oom();

r = sysv_translate_facility(n, basename(s->path), &m);

if (r < 0) {

log_warning_errno(r, "[%s:%u] Failed to translate LSB dependency %s, ignoring: %m", s->path, line, n);

continue;

}

if (r == 0)

continue;

is_before = startswith_no_case(full_text, "X-Start-Before:");

if (streq(m, SPECIAL_NETWORK_ONLINE_TARGET) && !is_before) {

/* the network-online target is special, as it needs to be actively pulled in */

r = strv_extend(&s->after, m);

if (r < 0)

return log_oom();

r = strv_extend(&s->wants, m);

} else

r = strv_extend(is_before ? &s->before : &s->after, m);

if (r < 0)

return log_oom();

}

if (!isempty(state_))

log_warning("[%s:%u] Trailing garbage in %*s, ignoring.", s->path, line, (int)(strchr(full_text, ':') - full_text), full_text);

return 0;

}

static int load_sysv(SysvStub *s) {

_cleanup_fclose_ FILE *f;

unsigned line = 0;

int r;

enum {

NORMAL,

DESCRIPTION,

LSB,

LSB_DESCRIPTION,

USAGE_CONTINUATION

} state = NORMAL;

_cleanup_free_ char *short_description = NULL, *long_description = NULL, *chkconfig_description = NULL;

char *description;

bool supports_reload = false;

assert(s);

f = fopen(s->path, "re");

if (!f)

return errno == ENOENT ? 0 : -errno;

log_debug("Loading SysV script %s", s->path);

while (!feof(f)) {

char l[LINE_MAX], *t;

if (!fgets(l, sizeof(l), f)) {

if (feof(f))

break;

return log_error_errno(errno, "Failed to read configuration file '%s': %m", s->path);

}

line++;

t = strstrip(l);

if (*t != '#') {

/* Try to figure out whether this init script supports

if ( state == USAGE_CONTINUATION ||

(state == NORMAL && strcasestr(t, "usage"))) {

if (usage_contains_reload(t)) {

supports_reload = true;

state = NORMAL;

} else if (t[strlen(t)-1] == '\\')

state = USAGE_CONTINUATION;

else

state = NORMAL;

}

continue;

}

if (state == NORMAL && streq(t, "### BEGIN INIT INFO")) {

state = LSB;

s->has_lsb = true;

continue;

}

if ((state == LSB_DESCRIPTION || state == LSB) && streq(t, "### END INIT INFO")) {

state = NORMAL;

continue;

}

t++;

t += strspn(t, WHITESPACE);

if (state == NORMAL) {

/* Try to parse Red Hat style description */

if (startswith_no_case(t, "description:")) {

size_t k = strlen(t);

char *d;

const char *j;

if (t[k-1] == '\\') {

state = DESCRIPTION;

t[k-1] = 0;

}

j = strstrip(t+12);

if (j && *j) {

d = strdup(j);

if (!d)

return -ENOMEM;

} else

d = NULL;

free(chkconfig_description);

chkconfig_description = d;

} else if (startswith_no_case(t, "pidfile:")) {

char *fn;

state = NORMAL;

fn = strstrip(t+8);

if (!path_is_absolute(fn)) {

log_error("[%s:%u] PID file not absolute. Ignoring.", s->path, line);

continue;

}

fn = strdup(fn);

if (!fn)

return -ENOMEM;

free(s->pid_file);

s->pid_file = fn;

}

} else if (state == DESCRIPTION) {

/* Try to parse Red Hat style description

size_t k = strlen(t);

char *j;

if (t[k-1] == '\\')

t[k-1] = 0;

else

state = NORMAL;

j = strstrip(t);

if (j && *j) {

char *d = NULL;

if (chkconfig_description)

d = strjoin(chkconfig_description, " ", j, NULL);

else

d = strdup(j);

if (!d)

return -ENOMEM;

free(chkconfig_description);

chkconfig_description = d;

}

} else if (state == LSB || state == LSB_DESCRIPTION) {

if (startswith_no_case(t, "Provides:")) {

state = LSB;

r = handle_provides(s, line, t, t + 9);

if (r < 0)

return r;

} else if (startswith_no_case(t, "Required-Start:") ||

startswith_no_case(t, "Should-Start:") ||

startswith_no_case(t, "X-Start-Before:") ||

startswith_no_case(t, "X-Start-After:")) {

state = LSB;

r = handle_dependencies(s, line, t, strchr(t, ':') + 1);

if (r < 0)

return r;

} else if (startswith_no_case(t, "Description:")) {

char *d, *j;

state = LSB_DESCRIPTION;

j = strstrip(t+12);

if (j && *j) {

d = strdup(j);

if (!d)

return -ENOMEM;

} else

d = NULL;

free(long_description);

long_description = d;

} else if (startswith_no_case(t, "Short-Description:")) {

char *d, *j;

state = LSB;

j = strstrip(t+18);

if (j && *j) {

d = strdup(j);

if (!d)

return -ENOMEM;

} else

d = NULL;

free(short_description);

short_description = d;

} else if (state == LSB_DESCRIPTION) {

if (startswith(l, "#\t") || startswith(l, "# ")) {

char *j;

j = strstrip(t);

if (j && *j) {

char *d = NULL;

if (long_description)

d = strjoin(long_description, " ", t, NULL);

else

d = strdup(j);

if (!d)

return -ENOMEM;

free(long_description);

long_description = d;

}

} else

state = LSB;

}

}

}

s->reload = supports_reload;

/* We use the long description only if

if (short_description)

description = short_description;

else if (chkconfig_description)

description = chkconfig_description;

else if (long_description)

description = long_description;

else

description = NULL;

if (description) {

char *d;

d = strappend(s->has_lsb ? "LSB: " : "SYSV: ", description);

if (!d)

return -ENOMEM;

s->description = d;

}

return 0;

}

static int fix_order(SysvStub *s, Hashmap *all_services) {

SysvStub *other;

Iterator j;

int r;

assert(s);

if (s->sysv_start_priority < 0)

return 0;

HASHMAP_FOREACH(other, all_services, j) {

if (s == other)

continue;

if (other->sysv_start_priority < 0)

continue;

/* If both units have modern headers we don't care

if (s->has_lsb && other->has_lsb)

continue;

if (other->sysv_start_priority < s->sysv_start_priority) {

r = strv_extend(&s->after, other->name);

if (r < 0)

return log_oom();

}

else if (other->sysv_start_priority > s->sysv_start_priority) {

r = strv_extend(&s->before, other->name);

if (r < 0)

return log_oom();

}

else

continue;

/* FIXME: Maybe we should compare the name here lexicographically? */

}

return 0;

}

static int enumerate_sysv(const LookupPaths *lp, Hashmap *all_services) {

char **path;

STRV_FOREACH(path, lp->sysvinit_path) {

_cleanup_closedir_ DIR *d = NULL;

struct dirent *de;

d = opendir(*path);

if (!d) {

if (errno != ENOENT)

log_warning_errno(errno, "opendir(%s) failed: %m", *path);

continue;

}

while ((de = readdir(d))) {

_cleanup_free_ char *fpath = NULL, *name = NULL;

_cleanup_(free_sysvstubp) SysvStub *service = NULL;

struct stat st;

int r;

if (hidden_file(de->d_name))

continue;

if (fstatat(dirfd(d), de->d_name, &st, 0) < 0) {

log_warning_errno(errno, "stat() failed on %s/%s: %m", *path, de->d_name);

continue;

}

if (!(st.st_mode & S_IXUSR))

continue;

if (!S_ISREG(st.st_mode))

continue;

name = sysv_translate_name(de->d_name);

if (!name)

return log_oom();

if (hashmap_contains(all_services, name))

continue;

fpath = strjoin(*path, "/", de->d_name, NULL);

if (!fpath)

return log_oom();

if (unit_file_lookup_state(UNIT_FILE_SYSTEM, NULL, lp, name) >= 0) {

log_debug("Native unit for %s already exists, skipping", name);

continue;

}

service = new0(SysvStub, 1);

if (!service)

return log_oom();

service->sysv_start_priority = -1;

service->name = name;

service->path = fpath;

r = hashmap_put(all_services, service->name, service);

if (r < 0)

return log_oom();

name = fpath = NULL;

service = NULL;

}

}

return 0;

}

static int set_dependencies_from_rcnd(const LookupPaths *lp, Hashmap *all_services) {

char **p;

unsigned i;

_cleanup_closedir_ DIR *d = NULL;

_cleanup_free_ char *path = NULL, *fpath = NULL;

SysvStub *service;

Iterator j;

Set *runlevel_services[ELEMENTSOF(rcnd_table)] = {};

_cleanup_set_free_ Set *shutdown_services = NULL;

int r = 0;

STRV_FOREACH(p, lp->sysvrcnd_path)

for (i = 0; i < ELEMENTSOF(rcnd_table); i ++) {

struct dirent *de;

free(path);

path = strjoin(*p, "/", rcnd_table[i].path, NULL);

if (!path)

return -ENOMEM;

if (d)

closedir(d);

d = opendir(path);

if (!d) {

if (errno != ENOENT)

log_warning_errno(errno, "opendir(%s) failed: %m", path);

continue;

}

while ((de = readdir(d))) {

_cleanup_free_ char *name = NULL;

int a, b;

if (hidden_file(de->d_name))

continue;

if (de->d_name[0] != 'S' && de->d_name[0] != 'K')

continue;

if (strlen(de->d_name) < 4)

continue;

a = undecchar(de->d_name[1]);

b = undecchar(de->d_name[2]);

if (a < 0 || b < 0)

continue;

free(fpath);

fpath = strjoin(*p, "/", de->d_name, NULL);

if (!fpath) {

r = -ENOMEM;

goto finish;

}

name = sysv_translate_name(de->d_name + 3);

if (!name) {

r = log_oom();

goto finish;

}

service = hashmap_get(all_services, name);

if (!service){

log_debug("Ignoring %s symlink in %s, not generating %s.",

de->d_name, rcnd_table[i].path, name);

continue;

}

if (de->d_name[0] == 'S') {

if (rcnd_table[i].type == RUNLEVEL_UP) {

service->sysv_start_priority =

MAX(a*10 + b, service->sysv_start_priority);

}

r = set_ensure_allocated(&runlevel_services[i], NULL);

if (r < 0)

goto finish;

r = set_put(runlevel_services[i], service);

if (r < 0)

goto finish;

} else if (de->d_name[0] == 'K' &&

(rcnd_table[i].type == RUNLEVEL_DOWN)) {

r = set_ensure_allocated(&shutdown_services, NULL);

if (r < 0)

goto finish;

r = set_put(shutdown_services, service);

if (r < 0)

goto finish;

}

}

}

for (i = 0; i < ELEMENTSOF(rcnd_table); i ++)

SET_FOREACH(service, runlevel_services[i], j) {

r = strv_extend(&service->before, rcnd_table[i].target);

if (r < 0)

return log_oom();

r = strv_extend(&service->wanted_by, rcnd_table[i].target);

if (r < 0)

return log_oom();

}

SET_FOREACH(service, shutdown_services, j) {

r = strv_extend(&service->before, SPECIAL_SHUTDOWN_TARGET);

if (r < 0)

return log_oom();

r = strv_extend(&service->conflicts, SPECIAL_SHUTDOWN_TARGET);

if (r < 0)

return log_oom();

}

r = 0;

finish:

for (i = 0; i < ELEMENTSOF(rcnd_table); i++)

set_free(runlevel_services[i]);

return r;

}

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

int r, q;

_cleanup_lookup_paths_free_ LookupPaths lp = {};

_cleanup_(free_sysvstub_hashmapp) Hashmap *all_services = NULL;

SysvStub *service;

Iterator j;

if (argc > 1 && argc != 4) {

log_error("This program takes three or no arguments.");

return EXIT_FAILURE;

}

if (argc > 1)

arg_dest = argv[3];

log_set_target(LOG_TARGET_SAFE);

log_parse_environment();

log_open();

umask(0022);

r = lookup_paths_init(&lp, MANAGER_SYSTEM, true, NULL, NULL, NULL, NULL);

if (r < 0) {

log_error("Failed to find lookup paths.");

return EXIT_FAILURE;

}

all_services = hashmap_new(&string_hash_ops);

if (!all_services) {

log_oom();

return EXIT_FAILURE;

}

r = enumerate_sysv(&lp, all_services);

if (r < 0) {

log_error("Failed to generate units for all init scripts.");

return EXIT_FAILURE;

}

r = set_dependencies_from_rcnd(&lp, all_services);

if (r < 0) {

log_error("Failed to read runlevels from rcnd links.");

return EXIT_FAILURE;

}

HASHMAP_FOREACH(service, all_services, j) {

q = load_sysv(service);

if (q < 0)

continue;

}

HASHMAP_FOREACH(service, all_services, j) {

q = fix_order(service, all_services);

if (q < 0)

continue;

q = generate_unit_file(service);

if (q < 0)

continue;

}

return EXIT_SUCCESS;

}