diff options
| -rw-r--r-- | src/librarian.c | 188 |
1 files changed, 181 insertions, 7 deletions
diff --git a/src/librarian.c b/src/librarian.c index 95f880e..98b3665 100644 --- a/src/librarian.c +++ b/src/librarian.c @@ -28,6 +28,8 @@ #include <stdlib.h> #include <errno.h> #include <dirent.h> +#include <unistd.h> +#include <fcntl.h> #include <assert.h> #define t(...) do { if (__VA_ARGS__) goto fail; } while (0) @@ -576,6 +578,148 @@ fail: /** + * Read the value of a variable in a file. + * + * @param path The pathname of the file to read. + * @param var The variable to retrieve + * @return The value of variable. `NULL` on error or if + * not found, `errno` is set to 0 if not found. + */ +static char *find_variable(const char *path, const char *var) +{ + int fd = -1, saved_errno; + size_t ptr = 0, size = 0, len; + char *buffer = NULL; + void *new; + char *p; + char *q; + char *sought = NULL; + ssize_t n; + + fd = open(path, O_RDONLY); + t (fd == -1); + + for (;;) { + if (ptr == size) { + size = size ? (size << 1) : 512; + new = realloc(buffer, size); + t (new != NULL); + buffer = new; + } + n = read(fd, buffer + ptr, size - ptr); + t (n < 0); + if (n == 0) + break; + } + + close(fd), fd = -1; + new = realloc(buffer, ptr + 3); + t (new != NULL); + buffer = new; + buffer[ptr++] = '\n'; + buffer[ptr++] = '\0'; + memmove(buffer + 1, buffer, ptr); + *buffer = '\n'; + + sought = malloc(strlen(var) + 2); + t (sought == NULL); + sought[0] = '\n'; + strcpy(sought, var); + + len = strlen(sought); + for (p = buffer; p;) { + p = strstr(p, sought); + if (p == NULL) + break; + if (!isspace(p[len])) { + p = strchr(p + 1, '\n'); + continue; + } + p += len + 1; + q = strchr(p, '\n'); + *q = '\0'; + } + + if (p == NULL) + free(buffer); + free(sought); + return p; + +fail: + saved_errno = errno; + if (fd >= 0) + close(fd); + free(buffer); + free(sought); + errno = saved_errno; + return NULL; +} + + +/** + * Get variables values stored in librarian files. + * + * @param vars Pointer to the first variable. + * @param vars_end Pointer to just after the last variable. + * @param files_start The index of the first file in `found_files` + * for which variables should be retrieved. + * @return String with all variables, `NULL` on error. + */ +static char *get_variables(const char **vars, const char **vars_end, size_t files_start) +{ + char *path; + const char **var; + char **parts = NULL; + char *part; + size_t ptr = 0; + size_t size = 0; + size_t len = 0; + void *new; + char *rc; + char *p; + int saved_errno; + + while (files_start < found_files_count) { + path = found_files[files_start++].path; + for (var = vars; var != vars_end; var++) { + part = find_variable(path, *var); + t (!part && errno); + if (!part || !*part) + continue; + if (ptr == size) { + size = size ? (size << 1) : 8; + new = realloc(parts, size * sizeof(*parts)); + t (new == NULL); + parts = new; + } + len += strlen(part) + 1; + parts[ptr++] = part; + } + } + + if (len == 0) + return strdup(""); + + p = rc = malloc(len); + t (rc == NULL); + for (size = ptr, ptr = 0; ptr < size; ptr++) { + p = stpcpy(p, parts[ptr]); + *p++ = ' '; + free(parts[ptr]); + } + p[-1] = 0; + + return rc; +fail: + saved_errno = errno; + while (ptr--) + free(parts[ptr]); + errno = saved_errno; + return NULL; +} + + +/** * @return 0: Program was successful. * 1: An error occurred. * 2: A library was not found. @@ -586,14 +730,20 @@ int main(int argc, char *argv[]) int dashed = 0, f_deps = 0, f_locate = 0, f_oldest = 0; char *arg; char **args = argv; - char **args_last = argv; - char **variables = argv; - char **variables_last = argv; + char **args_last = args; + const char **variables = (const char **)argv; + const char **variables_last = variables; struct library *libraries = NULL; struct library *libraries_last; const char *path_; char *path = NULL; int rc; + size_t start_files; + size_t start_libs, n; + char *data = NULL; + char *s; + char *end; + const char *deps_string = "deps"; /* Parse arguments. */ argv0 = argv ? (argc--, *argv++) : "pp"; @@ -644,9 +794,29 @@ int main(int argc, char *argv[]) t (path == NULL); /* Find librarian files. */ - if (find_librarian_files(libraries, (size_t)(libraries_last - libraries), path, f_oldest)) { - t (errno); - goto not_found; + start_files = found_files_count; + start_libs = 0; + while ((n = (size_t)(libraries_last - libraries) - start_libs)) { + if (find_librarian_files(libraries + start_libs, n, path, f_oldest)) { + t (errno); + goto not_found; + } + if (f_locate) + break; + if (f_deps) { + data = get_variables(&deps_string, 1 + &deps_string, start_files); + t (data == NULL); + for (end = s = data; end; s = end + 1) { + while (isspace(*s)) + s++; + end = strpbrk(s, " \t\r\n\f\v"); + if (end) + *end = '\0'; + if (parse_library(s, libraries_last++)) + goto not_found; + } + free(data), data = NULL; + } } if (f_locate) { while (found_files_count) @@ -654,7 +824,10 @@ int main(int argc, char *argv[]) goto done; } - /* TODO */ + /* Print requested data. */ + data = get_variables(variables, variables_last, 0); + t (data == NULL); + t (printf("%s\n", data) < 0); done: rc = 0; @@ -676,6 +849,7 @@ cleanup: free(found_files); free(libraries); free(path); + free(data); return rc; } |