librc: add rc_import_variables api

Author: navi-desu

src/librc/librc-misc.c

@@ -56,35 +56,25 @@ rc_yesno(const char *value)
 	return false;
 }
 
-
-/**
- * Read the entire @file into the buffer and set @len to the
- * size of the buffer when finished. For C strings, this will
- * be strlen(buffer) + 1.
- * Don't forget to free the buffer afterwards!
- */
-bool
-rc_getfile(const char *file, char **buffer, size_t *len)
+static bool
+do_getfileat(int dirfd, const char *file, char **buffer, size_t *size)
 {
 	bool ret = false;
 	FILE *fp;
-	int fd;
 	struct stat st;
 	size_t done, left;
 
-	fp = fopen(file, "re");
+	fp = do_fopenat(dirfd, file, O_RDONLY);
 	if (!fp)
 		return false;
 
 	/* assume fileno() never fails */
-	fd = fileno(fp);
-
-	if (fstat(fd, &st))
+	if (fstat(fileno(fp), &st))
 		goto finished;
 
 	left = st.st_size;
-	*len = left + 1; /* NUL terminator */
-	*buffer = xrealloc(*buffer, *len);
+	*size = left + 1; /* NUL terminator */
+	*buffer = xrealloc(*buffer, *size);
 	while (left) {
 		done = fread(*buffer, sizeof(*buffer[0]), left, fp);
 		if (done == 0 && ferror(fp))
@@ -96,13 +86,25 @@ rc_getfile(const char *file, char **buffer, size_t *len)
  finished:
 	if (!ret) {
 		free(*buffer);
-		*len = 0;
+		*size = 0;
 	} else
-		(*buffer)[*len - 1] = '\0';
+		(*buffer)[*size - 1] = '\0';
 	fclose(fp);
 	return ret;
 }
 
+/**
+ * Read the entire @file into the buffer and set @len to the
+ * size of the buffer when finished. For C strings, this will
+ * be strlen(buffer) + 1.
+ * Don't forget to free the buffer afterwards!
+ */
+bool
+rc_getfile(const char *file, char **buffer, size_t *len)
+{
+	return do_getfileat(AT_FDCWD, file, buffer, len);
+}
+
 char *
 rc_proc_getent(const char *ent RC_UNUSED)
 {
@@ -472,3 +474,22 @@ bool rc_export_variable(const char *service, const char *name, const char *value
 
 	return true;
 }
+
+void rc_import_variables(const char *service) {
+	DIR *envdir = do_opendirat(rc_dirfd(RC_DIR_ENVIRONMENT), service ? service : "rc");
+	struct dirent *d;
+	if (!envdir)
+		return;
+
+	while ((d = readdir(envdir))) {
+		char *value;
+		size_t size;
+
+		if (!do_getfileat(rc_dirfd(RC_DIR_ENVIRONMENT), d->d_name, &value, &size))
+			continue;
+		setenv(d->d_name, value, true);
+		free(value);
+	}
+
+	closedir(envdir);
+}

src/librc/rc.h.in

@@ -354,6 +354,10 @@ bool rc_service_value_set(const char *, const char *, const char *);
  * @param value */
 bool rc_export_variable(const char *, const char *, const char *);
 
+/*! Imports a variable from the enviroment of a service
+ * @param service name, if null imports the global environment */
+void rc_import_variables(const char *);
+
 /*! List the services in a runlevel
  * @param runlevel to list
  * @return NULL terminated list of services */