X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=manager.c;h=da2d4334746881728ebebdb3d77c1fb5efd510d7;hp=d6bc35a289d04034f8fe275a1400f0b5ebd20184;hb=9ea024f6b5a9626ddabeb4c4d4385481b52d211e;hpb=6091827530d6dd43479d6709fb6e9f745c11e900;ds=sidebyside diff --git a/manager.c b/manager.c index d6bc35a28..da2d43347 100644 --- a/manager.c +++ b/manager.c @@ -2,11 +2,13 @@ #include #include +#include #include "manager.h" #include "hashmap.h" #include "macro.h" #include "strv.h" +#include "log.h" Manager* manager_new(void) { Manager *m; @@ -20,10 +22,7 @@ Manager* manager_new(void) { if (!(m->jobs = hashmap_new(trivial_hash_func, trivial_compare_func))) goto fail; - if (!(m->jobs_to_add = hashmap_new(trivial_hash_func, trivial_compare_func))) - goto fail; - - if (!(m->jobs_to_remove = set_new(trivial_hash_func, trivial_compare_func))) + if (!(m->transaction_jobs = hashmap_new(trivial_hash_func, trivial_compare_func))) goto fail; return m; @@ -35,245 +34,619 @@ fail: void manager_free(Manager *m) { Name *n; + Job *j; assert(m); while ((n = hashmap_first(m->names))) name_free(n); + while ((j = hashmap_steal_first(m->transaction_jobs))) + job_free(j); + hashmap_free(m->names); hashmap_free(m->jobs); + hashmap_free(m->transaction_jobs); + + free(m); +} - /* FIXME: This is incomplete */ +static void transaction_delete_job(Manager *m, Job *j) { + assert(m); + assert(j); - hashmap_free(m->jobs_to_add); - set_free(m->jobs_to_remove); + manager_transaction_unlink_job(m, j); - free(m); + if (!j->linked) + job_free(j); } -int manager_add_job(Manager *m, JobType type, Name *name, JobMode mode, Job **_ret) { - Job *ret, *other; - void *state; - Name *dep; - int r; +static void transaction_abort(Manager *m) { + Job *j; assert(m); - assert(type < _JOB_TYPE_MAX); - assert(name); - assert(mode < _JOB_MODE_MAX); - assert(_ret); - /* Check for conflicts, first against the jobs we shall - * create */ - if ((other = hashmap_get(m->jobs_to_add, name))) { + while ((j = hashmap_first(m->transaction_jobs))) + if (j->linked) + transaction_delete_job(m, j); + else + job_free(j); - if (other->type != type) - return -EEXIST; + assert(hashmap_isempty(m->transaction_jobs)); + assert(!m->transaction_anchor); +} - } else if (name->meta.job) { +static void transaction_find_jobs_that_matter_to_anchor(Manager *m, Job *j, unsigned generation) { + JobDependency *l; - if (name->meta.job->type != type) { + assert(m); - if (mode == JOB_FAIL) - return -EEXIST; + for (l = j ? j->subject_list : m->transaction_anchor; l; l = l->subject_next) { - if ((r = set_put(m->jobs_to_remove, name->meta.job)) < 0) - return r; - } + /* This link does not matter */ + if (!l->matters) + continue; + + /* This name has already been marked */ + if (l->object->generation == generation) + continue; + + l->object->matters_to_anchor = true; + l->object->generation = generation; + + transaction_find_jobs_that_matter_to_anchor(m, l->object, generation); } +} - if (!(ret = job_new(m, type, name))) - return -ENOMEM; +static bool types_match(JobType a, JobType b, JobType c, JobType d) { + return + (a == c && b == d) || + (a == d && b == c); +} - if ((r = hashmap_put(m->jobs_to_add, name, ret)) < 0) - goto fail; +static int types_merge(JobType *a, JobType b) { + if (*a == b) + return 0; + + if (types_match(*a, b, JOB_START, JOB_VERIFY_STARTED)) + *a = JOB_START; + else if (types_match(*a, b, JOB_START, JOB_RELOAD) || + types_match(*a, b, JOB_START, JOB_RELOAD_OR_START) || + types_match(*a, b, JOB_VERIFY_STARTED, JOB_RELOAD_OR_START) || + types_match(*a, b, JOB_RELOAD, JOB_RELOAD_OR_START)) + *a = JOB_RELOAD_OR_START; + else if (types_match(*a, b, JOB_START, JOB_RESTART) || + types_match(*a, b, JOB_START, JOB_TRY_RESTART) || + types_match(*a, b, JOB_VERIFY_STARTED, JOB_RESTART) || + types_match(*a, b, JOB_RELOAD, JOB_RESTART) || + types_match(*a, b, JOB_RELOAD_OR_START, JOB_RESTART) || + types_match(*a, b, JOB_RELOAD_OR_START, JOB_TRY_RESTART) || + types_match(*a, b, JOB_RESTART, JOB_TRY_RESTART)) + *a = JOB_RESTART; + else if (types_match(*a, b, JOB_VERIFY_STARTED, JOB_RELOAD)) + *a = JOB_RELOAD; + else if (types_match(*a, b, JOB_VERIFY_STARTED, JOB_TRY_RESTART) || + types_match(*a, b, JOB_RELOAD, JOB_TRY_RESTART)) + *a = JOB_TRY_RESTART; + + return -EEXIST; +} + +static void transaction_merge_and_delete_job(Manager *m, Job *j, Job *other, JobType t) { + JobDependency *l, *last; - if (type == JOB_START || type == JOB_VERIFY_STARTED || type == JOB_RESTART_FINISH) { - SET_FOREACH(dep, ret->name->meta.requires, state) - if ((r = manager_add_job(m, type, dep, mode, NULL)) < 0) - goto fail; - SET_FOREACH(dep, ret->name->meta.soft_requires, state) - if ((r = manager_add_job(m, type, dep, JOB_FAIL, NULL)) < 0) - goto fail; - SET_FOREACH(dep, ret->name->meta.wants, state) - if ((r = manager_add_job(m, type, dep, JOB_FAIL, NULL)) < 0) - goto fail; - SET_FOREACH(dep, ret->name->meta.requisite, state) - if ((r = manager_add_job(m, JOB_VERIFY_STARTED, dep, mode, NULL)) < 0) - goto fail; - SET_FOREACH(dep, ret->name->meta.soft_requisite, state) - if ((r = manager_add_job(m, JOB_VERIFY_STARTED, dep, JOB_FAIL, NULL)) < 0) - goto fail; - SET_FOREACH(dep, ret->name->meta.conflicts, state) - if ((r = manager_add_job(m, type, dep, mode, NULL)) < 0) - goto fail; - - } else if (type == JOB_STOP || type == JOB_RESTART || type == JOB_TRY_RESTART) { - - SET_FOREACH(dep, ret->name->meta.required_by, state) - if ((r = manager_add_job(m, type, dep, mode, NULL)) < 0) - goto fail; + assert(j); + assert(other); + assert(j->name == other->name); + assert(!j->linked); + + j->type = t; + j->state = JOB_WAITING; + + j->matters_to_anchor = j->matters_to_anchor || other->matters_to_anchor; + + /* Patch us in as new owner of the JobDependency objects */ + last = NULL; + for (l = other->subject_list; l; l = l->subject_next) { + assert(l->subject == other); + l->subject = j; + last = l; } - if (_ret) - *_ret = ret; + /* Merge both lists */ + if (last) { + last->subject_next = j->subject_list; + if (j->subject_list) + j->subject_list->subject_prev = last; + j->subject_list = other->subject_list; + } - return 0; + /* Patch us in as new owner of the JobDependency objects */ + last = NULL; + for (l = other->object_list; l; l = l->object_next) { + assert(l->object == other); + l->object = j; + last = l; + } -fail: - job_free(ret); + /* Merge both lists */ + if (last) { + last->object_next = j->object_list; + if (j->object_list) + j->object_list->object_prev = last; + j->object_list = other->object_list; + } - return r; + /* Kill the other job */ + other->subject_list = NULL; + other->object_list = NULL; + transaction_delete_job(m, other); } +static int transaction_merge_jobs(Manager *m) { + Job *j; + void *state; + int r; -Job *manager_get_job(Manager *m, uint32_t id) { assert(m); - return hashmap_get(m->jobs, UINT32_TO_PTR(id)); -} + HASHMAP_FOREACH(j, m->transaction_jobs, state) { + JobType t = j->type; + Job *k; -Name *manager_get_name(Manager *m, const char *name) { - assert(m); - assert(name); + for (k = j->transaction_next; k; k = k->transaction_next) + if ((r = types_merge(&t, k->type)) < 0) + return r; - return hashmap_get(m->names, name); + while ((k = j->transaction_next)) { + if (j->linked) { + transaction_merge_and_delete_job(m, k, j, t); + j = k; + } else + transaction_merge_and_delete_job(m, j, k, t); + } + + assert(!j->transaction_next); + assert(!j->transaction_prev); + } + + return 0; } -static int detect_type(Name *name) { - char **n; +static int transaction_verify_order_one(Manager *m, Job *j, Job *from, unsigned generation) { + void *state; + Name *n; + int r; - assert(name); + assert(m); + assert(j); + + /* Did we find a cycle? */ + if (j->marker && j->generation == generation) { + Job *k; + + /* So, we already have been here. We have a + * cycle. Let's try to break it. We go backwards in our + * path and try to find a suitable job to remove. */ + + for (k = from; k; k = (k->generation == generation ? k->marker : NULL)) { + if (!k->matters_to_anchor) { + log_debug("Breaking order cycle by deleting job %s", name_id(k->name)); + transaction_delete_job(m, k); + return -EAGAIN; + } + + /* Check if this in fact was the beginning of + * the cycle */ + if (k == j) + break; + } - name->meta.type = _NAME_TYPE_INVALID; + return -ELOOP; + } - STRV_FOREACH(n, name->meta.names) { - NameType t; + j->marker = from; + j->generation = generation; - if ((t = name_type_from_string(*n)) == _NAME_TYPE_INVALID) - return -EINVAL; + /* We assume that the the dependencies are both-ways, and + * hence can ignore NAME_AFTER */ - if (name->meta.type == _NAME_TYPE_INVALID) { - name->meta.type = t; - continue; - } + SET_FOREACH(n, j->name->meta.dependencies[NAME_BEFORE], state) { + Job *o; - if (name->meta.type != t) - return -EINVAL; + if (!(o = hashmap_get(m->transaction_jobs, n))) + if (!(o = n->meta.job)) + continue; + + if ((r = transaction_verify_order_one(m, o, j, generation)) < 0) + return r; } return 0; } -static int fragment_load(Name *n) { - assert(n); +static int transaction_verify_order(Manager *m, unsigned *generation) { + bool again; + assert(m); + assert(generation); + + do { + Job *j; + int r; + void *state; + + again = false; + + HASHMAP_FOREACH(j, m->transaction_jobs, state) { - /*... */ + /* Assume merged */ + assert(!j->transaction_next); + assert(!j->transaction_prev); + + if ((r = transaction_verify_order_one(m, j, NULL, (*generation)++)) < 0) { + + /* There was a cycleq, but it was fixed, + * we need to restart our algorithm */ + if (r == -EAGAIN) { + again = true; + break; + } + + return r; + } + } + } while (again); return 0; } -static int sysv_load(Service *s) { - assert(s); +static void transaction_collect_garbage(Manager *m) { + bool again; + + assert(m); + + do { + void *state; + Job *j; + + again = false; + + HASHMAP_FOREACH(j, m->transaction_jobs, state) { + if (j->object_list) + continue; + + log_debug("Garbage collecting job %s", name_id(j->name)); + + transaction_delete_job(m, j); + again = true; + break; + } - /*... */ + } while (again); +} + +static int transaction_is_destructive(Manager *m, JobMode mode) { + void *state; + Job *j; + + assert(m); + + /* Checks whether applying this transaction means that + * existing jobs would be replaced */ + + HASHMAP_FOREACH(j, m->transaction_jobs, state) + if (j->name->meta.job && j->name->meta.job != j) + return -EEXIST; return 0; } -static int fstab_load(Name *n) { - assert(n); - assert(n->meta.type == NAME_MOUNT || n->meta.type == NAME_AUTOMOUNT); +static int transaction_apply(Manager *m, JobMode mode) { + void *state; + Job *j; + int r; + + HASHMAP_FOREACH(j, m->transaction_jobs, state) { + if (j->linked) + continue; - /*... */ + if ((r = hashmap_put(m->jobs, UINT32_TO_PTR(j->id), j)) < 0) + goto rollback; + } + + while ((j = hashmap_steal_first(m->transaction_jobs))) { + if (j->linked) + continue; + + if (j->name->meta.job) + job_free(j->name->meta.job); + + j->name->meta.job = j; + j->linked = true; + + /* We're fully installed. Now let's free data we don't + * need anymore. */ + + assert(!j->transaction_next); + assert(!j->transaction_prev); + + while (j->subject_list) + job_dependency_free(j->subject_list); + while (j->object_list) + job_dependency_free(j->object_list); + } return 0; + +rollback: + + HASHMAP_FOREACH(j, m->transaction_jobs, state) { + if (j->linked) + continue; + + hashmap_remove(m->jobs, UINT32_TO_PTR(j->id)); + } + + return r; } -static int snapshot_load(Snapshot *s) { - assert(s); - /*... */ +static int transaction_activate(Manager *m, JobMode mode) { + int r; + unsigned generation = 1; + + assert(m); + + /* This applies the changes recorded in transaction_jobs to + * the actual list of jobs, if possible. */ + + /* First step: figure out which jobs matter */ + transaction_find_jobs_that_matter_to_anchor(m, NULL, generation++); + + /* Second step: let's merge entries we can merge */ + if ((r = transaction_merge_jobs(m)) < 0) + goto rollback; + + /* Third step: verify order makes sense */ + if ((r = transaction_verify_order(m, &generation)) < 0) + goto rollback; + + /* Third step: do garbage colletion */ + transaction_collect_garbage(m); + + /* Fourth step: check whether we can actually apply this */ + if (mode == JOB_FAIL) + if ((r = transaction_is_destructive(m, mode)) < 0) + goto rollback; + + /* Fifth step: apply changes */ + if ((r = transaction_apply(m, mode)) < 0) + goto rollback; + + assert(hashmap_isempty(m->transaction_jobs)); + assert(!m->transaction_anchor); return 0; + +rollback: + transaction_abort(m); + return r; } -static int load(Name *name) { +static Job* transaction_add_one_job(Manager *m, JobType type, Name *name, bool *is_new) { + Job *j, *f; int r; + assert(m); assert(name); - if (name->meta.state != NAME_STUB) - return 0; + /* Looks for an axisting prospective job and returns that. If + * it doesn't exist it is created and added to the prospective + * jobs list. */ - if ((r = detect_type(name)) < 0) - return r; + f = hashmap_get(m->transaction_jobs, name); + + for (j = f; j; j = j->transaction_next) { + assert(j->name == name); + + if (j->type == type) { + if (is_new) + *is_new = false; + return j; + } + } + + if (name->meta.job && name->meta.job->type == type) + j = name->meta.job; + else if (!(j = job_new(m, type, name))) + return NULL; + + if ((r = hashmap_replace(m->transaction_jobs, name, j)) < 0) { + job_free(j); + return NULL; + } + + j->transaction_next = f; + + if (f) + f->transaction_prev = j; + + j->generation = 0; + j->marker = NULL; + j->matters_to_anchor = false; + + if (is_new) + *is_new = true; - if (name->meta.type == NAME_SERVICE) { + return j; +} + +void manager_transaction_unlink_job(Manager *m, Job *j) { + assert(m); + assert(j); - /* Load a .service file */ - if ((r = fragment_load(name)) == 0) - goto finish; + if (j->transaction_prev) + j->transaction_prev->transaction_next = j->transaction_next; + else if (j->transaction_next) + hashmap_replace(m->transaction_jobs, j->name, j->transaction_next); + else + hashmap_remove_value(m->transaction_jobs, j->name, j); - /* Load a classic init script */ - if (r == -ENOENT) - if ((r = sysv_load(SERVICE(name))) == 0) - goto finish; + if (j->transaction_next) + j->transaction_next->transaction_prev = j->transaction_prev; - } else if (name->meta.type == NAME_MOUNT || - name->meta.type == NAME_AUTOMOUNT) { + j->transaction_prev = j->transaction_next = NULL; - if ((r = fstab_load(name)) == 0) - goto finish; + while (j->subject_list) + job_dependency_free(j->subject_list); - } else if (name->meta.type == NAME_SNAPSHOT) { + while (j->object_list) { + Job *other = j->object_list->matters ? j->object_list->subject : NULL; - if ((r = snapshot_load(SNAPSHOT(name))) == 0) - goto finish; + job_dependency_free(j->object_list); - } else { - if ((r = fragment_load(name)) == 0) - goto finish; + if (other) { + log_debug("Deleting job %s as dependency of job %s", name_id(other->name), name_id(j->name)); + transaction_delete_job(m, other); + } } +} + +static int transaction_add_job_and_dependencies(Manager *m, JobType type, Name *name, Job *by, bool matters, bool force, Job **_ret) { + Job *ret; + void *state; + Name *dep; + int r; + bool is_new; + + assert(m); + assert(type < _JOB_TYPE_MAX); + assert(name); + + if (name->meta.state != NAME_LOADED) + return -EINVAL; - name->meta.state = NAME_FAILED; + /* First add the job. */ + if (!(ret = transaction_add_one_job(m, type, name, &is_new))) + return -ENOMEM; + + /* Then, add a link to the job. */ + if (!job_dependency_new(by, ret, matters)) + return -ENOMEM; + + if (is_new) { + /* Finally, recursively add in all dependencies. */ + if (type == JOB_START || type == JOB_RELOAD_OR_START) { + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_REQUIRES], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_START, dep, ret, true, force, NULL)) < 0) + goto fail; + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_SOFT_REQUIRES], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_START, dep, ret, !force, force, NULL)) < 0) + goto fail; + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_WANTS], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_START, dep, ret, false, force, NULL)) < 0) + goto fail; + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_REQUISITE], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_VERIFY_STARTED, dep, ret, true, force, NULL)) < 0) + goto fail; + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_SOFT_REQUISITE], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_VERIFY_STARTED, dep, ret, !force, force, NULL)) < 0) + goto fail; + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_CONFLICTS], state) + if ((r = transaction_add_job_and_dependencies(m, JOB_STOP, dep, ret, true, force, NULL)) < 0) + goto fail; + + } else if (type == JOB_STOP || type == JOB_RESTART || type == JOB_TRY_RESTART) { + + SET_FOREACH(dep, ret->name->meta.dependencies[NAME_REQUIRED_BY], state) + if ((r = transaction_add_job_and_dependencies(m, type, dep, ret, true, force, NULL)) < 0) + goto fail; + } + + /* JOB_VERIFY_STARTED, JOB_RELOAD require no dependency handling */ + } + + return 0; + +fail: return r; +} + +int manager_add_job(Manager *m, JobType type, Name *name, JobMode mode, bool force, Job **_ret) { + int r; + Job *ret; + + assert(m); + assert(type < _JOB_TYPE_MAX); + assert(name); + assert(mode < _JOB_MODE_MAX); + + if ((r = transaction_add_job_and_dependencies(m, type, name, NULL, true, force, &ret))) { + transaction_abort(m); + return r; + } + + if ((r = transaction_activate(m, mode)) < 0) + return r; + + if (_ret) + *_ret = ret; -finish: - name->meta.state = NAME_LOADED; return 0; } +Job *manager_get_job(Manager *m, uint32_t id) { + assert(m); + + return hashmap_get(m->jobs, UINT32_TO_PTR(id)); +} + +Name *manager_get_name(Manager *m, const char *name) { + assert(m); + assert(name); + + return hashmap_get(m->names, name); +} + static int dispatch_load_queue(Manager *m) { Meta *meta; assert(m); + /* Make sure we are not run recursively */ + if (m->dispatching_load_queue) + return 0; + + m->dispatching_load_queue = true; + /* Dispatches the load queue. Takes a name from the queue and * tries to load its data until the queue is empty */ while ((meta = m->load_queue)) { - load(NAME(meta)); + name_load(NAME(meta)); LIST_REMOVE(Meta, m->load_queue, meta); } + m->dispatching_load_queue = false; + return 0; } - - int manager_load_name(Manager *m, const char *name, Name **_ret) { Name *ret; NameType t; int r; + char *n; assert(m); assert(name); assert(_ret); -/* This will load the service information files, but not actually - * start any services or anything */ + if (!name_is_valid(name)) + return -EINVAL; + + /* This will load the service information files, but not actually + * start any services or anything */ if ((ret = manager_get_name(m, name))) goto finish; @@ -286,11 +659,17 @@ int manager_load_name(Manager *m, const char *name, Name **_ret) { ret->meta.type = t; - if (!(ret->meta.names = strv_new(name, NULL))) { + if (!(n = strdup(name))) { name_free(ret); return -ENOMEM; } + if (set_put(ret->meta.names, n) < 0) { + name_free(ret); + free(n); + return -ENOMEM; + } + if ((r = name_link(ret)) < 0) { name_free(ret); return r; @@ -307,3 +686,38 @@ finish: *_ret = ret; return 0; } + +void manager_dump_jobs(Manager *s, FILE *f, const char *prefix) { + void *state; + Job *j; + + assert(s); + assert(f); + + HASHMAP_FOREACH(j, s->jobs, state) + job_dump(j, f, prefix); +} + +void manager_dump_names(Manager *s, FILE *f, const char *prefix) { + void *state; + Name *n; + const char *t; + + assert(s); + assert(f); + + HASHMAP_FOREACH_KEY(n, t, s->names, state) + if (name_id(n) == t) + name_dump(n, f, prefix); +} + +void manager_clear_jobs(Manager *m) { + Job *j; + + assert(m); + + transaction_abort(m); + + while ((j = hashmap_first(m->jobs))) + job_free(j); +}