X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fwgtpkg-unit.c;h=9f143469fc2ca3530fa1f83683f01bca3e6910b5;hb=765d08087bf422c821556400a68e1fb4868f85f3;hp=26cf324571c19db9bbacfde054bed517402a85dd;hpb=14d4c3fe7e351c12f1cc4ec6246a23ae94dca7da;p=src%2Fapp-framework-main.git

diff --git a/src/wgtpkg-unit.c b/src/wgtpkg-unit.c
index 26cf324..9f14346 100644
--- a/src/wgtpkg-unit.c
+++ b/src/wgtpkg-unit.c
@@ -1,5 +1,5 @@
 /*
- Copyright 2016, 2017 IoT.bzh
+ Copyright (C) 2016, 2017, 2018 IoT.bzh
 
  author: José Bollo <jose.bollo@iot.bzh>
 
@@ -25,12 +25,20 @@
 #include <unistd.h>
 #include <string.h>
 #include <errno.h>
- 
+#include <limits.h>
+
+#include <json-c/json.h>
+
 #include "verbose.h"
 #include "utils-file.h"
 
 #include "wgtpkg-mustach.h"
+#include "utils-json.h"
+#include "wgt-json.h"
+#include "utils-systemd.h"
+
 #include "wgtpkg-unit.h"
+#include "wgt-strings.h"
 
 #if 0
 #include <ctype.h>
@@ -46,8 +54,8 @@ static char *template;
  * Returns 1 if 'text' matches the 'pattern' or else returns 0.
  * When returning 1 and 'after' isn't NULL, the pointer to the
  * first character after the pettern in 'text' is stored in 'after'.
- * The characters '\n' and ' ' have a special mening in the search:
- *  * '\n': matches any space or tabs (including none) followed 
+ * The characters '\n' and ' ' have a special meaning in the search:
+ *  * '\n': matches any space or tabs (including none) followed
  *          either by '\n' or '\0' (end of the string)
  *  * ' ': matches any space or tabs but at least one.
  */
@@ -109,12 +117,13 @@ static size_t pack(char *text, char purge)
 	char cont;     /* flag telling whether the line continues the previous one */
 	char nextcont; /* flag telling whether the line will continues the next one */
 
-	cont = 0;
+	nextcont = 0;
 	c = *(write = read = text);
 
 	/* iteration over lines */
 	while (c) {
 		/* computes emit, nextcont, emit and start for the current line */
+		cont = nextcont;
 		emit = nextcont = 0;
 		start = NULL;
 		begin = read;
@@ -133,7 +142,7 @@ static size_t pack(char *text, char purge)
 		if (c)
 			c = *++read;
 		/* emit the line if not empty */
-		if (emit) {
+		if (emit || (cont && !nextcont)) {
 			/* removes the blanks on the left of not continuing lines */
 			if (!cont && start)
 				begin = start;
@@ -148,7 +157,6 @@ static size_t pack(char *text, char purge)
 					*write++ = *begin++;
 			}
 		}
-		cont = nextcont;
 	}
 	*write = 0;
 	return (size_t)(write - text);
@@ -183,8 +191,8 @@ static char *offset(char *text, const char *pattern, char **args)
  */
 static int process_one_unit(char *spec, struct unitdesc *desc)
 {
-	char *nsoc, *nsrv, *name;
-	int isuser, issystem, issock, isserv;
+	char *nsoc, *nsrv, *name, *wanted;
+	int isuser, issystem, issock, isserv, iswanted;
 	size_t len;
 
 	/* finds the configuration directive of the unit */
@@ -192,6 +200,7 @@ static int process_one_unit(char *spec, struct unitdesc *desc)
 	issystem = !!offset(spec, "%systemd-unit system\n", NULL);
 	issock  = !!offset(spec, "%systemd-unit socket ", &nsoc);
 	isserv  = !!offset(spec, "%systemd-unit service ", &nsrv);
+	iswanted = !!offset(spec, "%systemd-unit wanted-by ", &wanted);
 
 	/* check the unit scope */
 	if ((isuser + issystem) == 1) {
@@ -209,7 +218,7 @@ static int process_one_unit(char *spec, struct unitdesc *desc)
 			desc->type = unittype_service;
 			name = nsrv;
 		}
-		len = (size_t)(strchrnul(name, '\n') - name);
+		len = strcspn(name, " \t\n");
 		desc->name = strndup(name, len);
 		desc->name_length = desc->name ? len : 0;
 	} else {
@@ -218,6 +227,15 @@ static int process_one_unit(char *spec, struct unitdesc *desc)
 		desc->name_length = 0;
 	}
 
+	if (iswanted) {
+		len = strcspn(wanted, " \t\n");
+		desc->wanted_by = strndup(wanted, len);
+		desc->wanted_by_length = len;
+	} else {
+		desc->wanted_by = NULL;
+		desc->wanted_by_length = 0;
+	}
+
 	desc->content = spec;
 	desc->content_length = pack(spec, '%');
 
@@ -232,42 +250,75 @@ static int process_one_unit(char *spec, struct unitdesc *desc)
  * with its given 'closure' and the array descripbing the units.
  * Return 0 in case of success or a negative value in case of error.
  */
-static int process_all_units(char *corpus, int (*process)(void *closure, const struct unitdesc descs[], unsigned count), void *closure)
+static int process_all_units(char *corpus, const struct unitconf *conf, int (*process)(void *closure, const struct generatedesc *desc), void *closure, struct json_object *jdesc)
 {
 	int rc, rc2;
-	unsigned n;
-	char *beg, *end;
-	struct unitdesc *descs, *d;
-
-	descs = NULL;
-	n = 0;
+	char *beg, *end, *befbeg, *aftend;
+	struct unitdesc *u;
+	struct generatedesc gdesc;
+
+	gdesc.conf = conf;
+	gdesc.desc = jdesc;
+	gdesc.units = NULL;
+	gdesc.nunits = 0;
 	rc = rc2 = 0;
 
 	/* while there is a unit in the corpus */
-	while(offset(corpus, "%begin systemd-unit\n", &beg)) {
-
-		/* get the end of the unit */
-		end = offset(beg, "%end systemd-unit\n", &corpus);
+	for(;;) {
+		befbeg = offset(corpus, "%begin ", &beg);
+		end = offset(corpus, "%end ", &aftend);
+		if (!befbeg) {
+			if (end) {
+				/* %end detected without %begin */
+				ERROR("unexpected %%end at end");
+				rc = rc ? :-EINVAL;
+			}
+			break;
+		}
 		if (!end) {
 			/* unterminated unit !! */
-			ERROR("unterminated unit description!! %s", beg);
+			ERROR("unterminated unit description!!");
 			corpus = beg;
 			rc2 = -EINVAL;
+		} else if (end < befbeg) {
+			/* sequence %end ... %begin detected !! */
+			ERROR("unexpected %%end before %%begin");
+			corpus = aftend;
+			rc2 = -EINVAL;
 		} else {
-			/* separate the unit from the corpus */
-			*end = 0;
-
-			/* allocates a descriptor for the unit */
-			d = realloc(descs, (n + 1) * sizeof *descs);
-			if (d == NULL)
-				rc2 = -ENOMEM;
-			else {
-				/* creates the unit description */
-				memset(&d[n], 0, sizeof *d);
-				descs = d;
-				rc2 = process_one_unit(beg, &descs[n]);
-				if (rc2 >= 0)
-					n++;
+			befbeg =  offset(beg, "%begin ", NULL);
+			if (befbeg && befbeg < end) {
+				/* sequence %begin ... %begin ... %end detected !! */
+				ERROR("unexpected %%begin after %%begin");
+				corpus = beg;
+				rc2 = -EINVAL;
+			} else {
+				*end = 0;
+				corpus = aftend;
+				if (matches("systemd-unit\n", beg, &beg)) {
+					if (!matches("systemd-unit\n", aftend, &corpus)) {
+						/* end doesnt match */
+						ERROR("unmatched %%begin systemd-unit (matching end mismatch)");
+						rc2 = -EINVAL;
+					} else {
+						/* allocates a descriptor for the unit */
+						u = realloc((void*)gdesc.units, ((unsigned)gdesc.nunits + 1) * sizeof *gdesc.units);
+						if (u == NULL)
+							rc2 = -ENOMEM;
+						else {
+							/* creates the unit description */
+							gdesc.units = u;
+							u = &u[gdesc.nunits];
+							memset(u, 0, sizeof *u);
+							rc2 = process_one_unit(beg, u);
+							if (rc2 >= 0)
+								gdesc.nunits++;
+						}
+					}
+				} else {
+					ERROR("unexpected %%begin name");
+					rc2 = -EINVAL;
+				}
 			}
 		}
 		/* records the error if there is an error */
@@ -279,12 +330,14 @@ static int process_all_units(char *corpus, int (*process)(void *closure, const s
 
 	/* call the function that processes the units */
 	if (rc == 0 && process)
-		rc = process(closure, descs, n);
+		rc = process(closure, &gdesc);
 
 	/* cleanup and frees */
-	while(n)
-		free((char *)(descs[--n].name));
-	free(descs);
+	while(gdesc.nunits) {
+		free((void*)(gdesc.units[--gdesc.nunits].name));
+		free((void*)(gdesc.units[gdesc.nunits].wanted_by));
+	}
+	free((void*)gdesc.units);
 
 	return rc;
 }
@@ -292,23 +345,23 @@ static int process_all_units(char *corpus, int (*process)(void *closure, const s
 /*
  * Clear the unit generator
  */
-void unit_generator_off()
+void unit_generator_close_template()
 {
 	free(template);
 	template = NULL;
 }
 
 /*
- * Initialises the unit generator with 'filename'.
+ * Initialises the unit generator with the content of the file of path 'filename'.
  * Returns 0 in case of success or a negative number in case of error.
  */
-int unit_generator_on(const char *filename)
+int unit_generator_open_template(const char *filename)
 {
 	size_t size;
 	char *tmp;
 	int rc;
 
-	unit_generator_off();
+	unit_generator_close_template();
 	rc = getfile(filename ? : FWK_UNIT_CONF, &template, NULL);
 	if (!rc) {
 		size = pack(template, ';');
@@ -319,29 +372,217 @@ int unit_generator_on(const char *filename)
 	return rc;
 }
 
+static int add_metadata(struct json_object *jdesc, const struct unitconf *conf)
+{
+	struct json_object *targets, *targ;
+	char portstr[30];
+	int port, i, n;
+
+	if (json_object_object_get_ex(jdesc, string_targets, &targets)) {
+		n = json_object_array_length(targets);
+		for (i = 0 ; i < n ; i++) {
+			targ = json_object_array_get_idx(targets, i);
+			if (!conf->port)
+				strcpy(portstr, "0");
+			else {
+				port = conf->port ? conf->port() : 0;
+				if (port < 0)
+					return port;
+				sprintf(portstr, "%d", port);
+			}
+			if (!j_add_string_m(targ, "#metatarget.http-port", portstr))
+				return -1;
+		}
+	}
+
+	return 	j_add_many_strings_m(jdesc,
+		"#metadata.install-dir", conf->installdir,
+		"#metadata.icons-dir", conf->icondir,
+		NULL) ? 0 : -1;
+}
+
 /*
- * Applies the object 'jdesc' to the current unit generator.
+ * Applies the object 'jdesc' augmented of meta data coming
+ * from 'conf' to the current unit generator.
  * The current unit generator will be set to the default one if not unit
- * was previously set using the function 'unit_generator_on'.
+ * was previously set using the function 'unit_generator_open_template'.
  * The callback function 'process' is then called with the
  * unit descriptors array and the expected closure.
  * Return what returned process in case of success or a negative
  * error code.
  */
-int unit_generator_process(struct json_object *jdesc, int (*process)(void *closure, const struct unitdesc descs[], unsigned count), void *closure)
+int unit_generator_process(struct json_object *jdesc, const struct unitconf *conf, int (*process)(void *closure, const struct generatedesc *desc), void *closure)
 {
 	int rc;
 	size_t size;
 	char *instance;
 
-	rc = template ? 0 : unit_generator_on(NULL);
-	if (!rc) {
-		instance = NULL;
-		rc = apply_mustach(template, jdesc, &instance, &size);
-		if (!rc)
-			rc = process_all_units(instance, process, closure);
-		free(instance);
+	rc = add_metadata(jdesc, conf);
+	if (rc)
+		ERROR("can't set the metadata. %m");
+	else {
+		rc = template ? 0 : unit_generator_open_template(NULL);
+		if (!rc) {
+			instance = NULL;
+			rc = apply_mustach(template, jdesc, &instance, &size);
+			if (!rc)
+				rc = process_all_units(instance, conf, process, closure, jdesc);
+			free(instance);
+		}
 	}
 	return rc;
 }
 
+/**************** SPECIALIZED PART *****************************/
+
+static int check_unit_desc(const struct unitdesc *desc, int tells)
+{
+	if (desc->scope != unitscope_unknown && desc->type != unittype_unknown && desc->name != NULL)
+		return 0;
+
+	if (tells) {
+		if (desc->scope == unitscope_unknown)
+			ERROR("unit of unknown scope");
+		if (desc->type == unittype_unknown)
+			ERROR("unit of unknown type");
+		if (desc->name == NULL)
+			ERROR("unit of unknown name");
+	}
+	errno = EINVAL;
+	return -1;
+}
+
+static int get_unit_path(char *path, size_t pathlen, const struct unitdesc *desc)
+{
+	int rc = systemd_get_unit_path(
+			path, pathlen, desc->scope == unitscope_user,
+			desc->name, desc->type == unittype_socket ? "socket" : "service");
+
+	if (rc < 0)
+		ERROR("can't get the unit path for %s", desc->name);
+
+	return rc;
+}
+
+static int get_wants_path(char *path, size_t pathlen, const struct unitdesc *desc)
+{
+	int rc = systemd_get_wants_path(
+			path, pathlen, desc->scope == unitscope_user, desc->wanted_by,
+			desc->name, desc->type == unittype_socket ? "socket" : "service");
+
+	if (rc < 0)
+		ERROR("can't get the wants path for %s and %s", desc->name, desc->wanted_by);
+
+	return rc;
+}
+
+static int get_wants_target(char *path, size_t pathlen, const struct unitdesc *desc)
+{
+	int rc = systemd_get_wants_target(
+			path, pathlen,
+			desc->name, desc->type == unittype_socket ? "socket" : "service");
+
+	if (rc < 0)
+		ERROR("can't get the wants target for %s", desc->name);
+
+	return rc;
+}
+
+static int do_uninstall_units(void *closure, const struct generatedesc *desc)
+{
+	int rc, rc2;
+	int i;
+	char path[PATH_MAX];
+	const struct unitdesc *u;
+
+	rc = 0;
+	for (i = 0 ; i < desc->nunits ; i++) {
+		u = &desc->units[i];
+		rc2 = check_unit_desc(u, 0);
+		if (rc2 == 0) {
+			rc2 = get_unit_path(path, sizeof path, u);
+			if (rc2 >= 0) {
+				rc2 = unlink(path);
+			}
+			if (rc2 < 0 && rc == 0)
+				rc = rc2;
+			if (u->wanted_by != NULL) {
+				rc2 = get_wants_path(path, sizeof path, u);
+				if (rc2 >= 0)
+					rc2 = unlink(path);
+			}
+		}
+		if (rc2 < 0 && rc == 0)
+			rc = rc2;
+	}
+	return rc;
+}
+
+static int do_install_units(void *closure, const struct generatedesc *desc)
+{
+	int rc;
+	int i;
+	char path[PATH_MAX + 1], target[PATH_MAX + 1];
+	const struct unitdesc *u;
+
+	i = 0;
+	while (i < desc->nunits) {
+		u = &desc->units[i];
+		rc = check_unit_desc(u, 1);
+		if (!rc) {
+			rc = get_unit_path(path, sizeof path, u);
+			if (rc >= 0) {
+				rc = putfile(path, u->content, u->content_length);
+				if (rc >= 0 && u->wanted_by != NULL) {
+					rc = get_wants_path(path, sizeof path, u);
+					if (rc >= 0) {
+						rc = get_wants_target(target, sizeof target, u);
+						if (rc >= 0) {
+							unlink(path); /* TODO? check? */
+							rc = symlink(target, path);
+						}
+					}
+				}
+				i++;
+			}
+		}
+		if (rc < 0)
+			goto error;
+	}
+	return 0;
+error:
+	i = errno;
+	do_uninstall_units(closure, desc);
+	errno = i;
+	return rc;
+}
+
+static int do_install_uninstall(
+		struct wgt_info *ifo,
+		const struct unitconf *conf,
+		int (*doer)(void *, const struct generatedesc *)
+)
+{
+	int rc;
+	struct json_object *jdesc;
+
+	jdesc = wgt_info_to_json(ifo);
+	if (!jdesc)
+		rc = -1;
+	else {
+		rc = unit_generator_process(jdesc, conf, doer, NULL);
+		json_object_put(jdesc);
+	}
+	return rc;
+}
+
+int unit_install(struct wgt_info *ifo, const struct unitconf *conf)
+{
+	return do_install_uninstall(ifo, conf, do_install_units);
+}
+
+int unit_uninstall(struct wgt_info *ifo, const struct unitconf *conf)
+{
+	return do_install_uninstall(ifo, conf, do_uninstall_units);
+}
+