shaba/systemd
1
0
Fork 0
mirror of https://github.com/systemd/systemd.git synced 2025-01-26 14:04:03 +03:00
Code

manager: make running_as configurable

Browse Source
This commit is contained in:
Lennart Poettering 2010-04-06 23:55:42 +02:00
parent f170852aea
commit a5dab5ce0f
4 changed files with 37 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
main.c
View File

@ -39,6 +39,7 @@ static enum {
} action = ACTION_RUN;
static char *default_unit = NULL;
static ManagerRunningAs running_as = _MANAGER_RUNNING_AS_INVALID;
static int set_default_unit(const char *u) {
char *c;
@ -132,13 +133,15 @@ static int parse_argv(int argc, char *argv[]) {
enum {
ARG_LOG_LEVEL = 0x100,
ARG_LOG_TARGET,
ARG_DEFAULT
ARG_DEFAULT,
ARG_RUNNING_AS
};
static const struct option options[] = {
{ "log-level", required_argument, NULL, ARG_LOG_LEVEL },
{ "log-target", required_argument, NULL, ARG_LOG_TARGET },
{ "default", required_argument, NULL, ARG_DEFAULT },
{ "running-as", required_argument, NULL, ARG_RUNNING_AS },
{ "help", no_argument, NULL, 'h' },
{ NULL, 0, NULL, 0 }
};
@ -178,6 +181,18 @@ static int parse_argv(int argc, char *argv[]) {
break;
case ARG_RUNNING_AS: {
ManagerRunningAs as;
if ((as = manager_running_as_from_string(optarg)) < 0) {
log_error("Failed to parse running as value %s", optarg);
return -EINVAL;
}
running_as = as;
break;
}
case 'h':
action = ACTION_HELP;
break;
@ -204,7 +219,8 @@ static int help(void) {
" -h --help Show this help\n"
" --default=UNIT Set default unit\n"
" --log-level=LEVEL Set log level\n"
" --log-target=TARGET Set log target (console, syslog, kmsg)\n",
" --log-target=TARGET Set log target (console, syslog, kmsg)\n"
" --running-as=AS Set running as (init, system, sesstion)\n",
__progname);
return 0;
@ -216,6 +232,13 @@ int main(int argc, char *argv[]) {
Job *job = NULL;
int r, retval = 1;
if (getpid() == 1)
running_as = MANAGER_INIT;
else if (getuid() == 0)
running_as = MANAGER_SYSTEM;
else
running_as = MANAGER_SESSION;
if (set_default_unit(SPECIAL_DEFAULT_TARGET) < 0)
goto finish;
@ -229,8 +252,9 @@ int main(int argc, char *argv[]) {
/* Close all open files */
assert_se(close_all_fds(NULL, 0) == 0);
if (parse_proc_cmdline() < 0)
goto finish;
if (running_as != MANAGER_SESSION)
if (parse_proc_cmdline() < 0)
goto finish;
log_parse_environment();
@ -260,7 +284,9 @@ int main(int argc, char *argv[]) {
log_open_syslog();
log_open_kmsg();
if ((r = manager_new(&m)) < 0) {
log_debug("systemd running in %s mode.", manager_running_as_to_string(running_as));
if ((r = manager_new(running_as, &m)) < 0) {
log_error("Failed to allocate manager object: %s", strerror(-r));
goto finish;
}

14
manager.c
View File

@ -256,15 +256,18 @@ static int manager_find_paths(Manager *m) {
return 0;
}
int manager_new(Manager **_m) {
int manager_new(ManagerRunningAs running_as, Manager **_m) {
Manager *m;
int r = -ENOMEM;
assert(_m);
assert(running_as >= 0);
assert(running_as < _MANAGER_RUNNING_AS_MAX);
if (!(m = new0(Manager, 1)))
return -ENOMEM;
m->running_as = running_as;
m->signal_watch.fd = m->mount_watch.fd = m->udev_watch.fd = m->epoll_fd = -1;
m->current_job_id = 1; /* start as id #1, so that we can leave #0 around as "null-like" value */
@ -286,15 +289,6 @@ int manager_new(Manager **_m) {
if ((m->epoll_fd = epoll_create1(EPOLL_CLOEXEC)) < 0)
goto fail;
if (getpid() == 1)
m->running_as = MANAGER_INIT;
else if (getuid() == 0)
m->running_as = MANAGER_SYSTEM;
else
m->running_as = MANAGER_SESSION;
log_debug("systemd running in %s mode.", manager_running_as_to_string(m->running_as));
if ((r = manager_find_paths(m)) < 0)
goto fail;

2
manager.h
View File

@ -166,7 +166,7 @@ struct Manager {
char *cgroup_hierarchy;
};
int manager_new(Manager **m);
int manager_new(ManagerRunningAs running_as, Manager **m);
void manager_free(Manager *m);
int manager_coldplug(Manager *m);

2
test-engine.c
View File

@ -33,7 +33,7 @@ int main(int argc, char *argv[]) {
assert_se(set_unit_path("test2") >= 0);
assert_se(manager_new(&m) >= 0);
assert_se(manager_new(MANAGER_INIT, &m) >= 0);
printf("Load1:\n");
assert_se(manager_load_unit(m, "a.service", &a) == 0);