X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=tools%2Fdepends.c;h=6ac31cdc5ebce5e973c1b34b9c506c3672fa7ada;hp=1422c8585f51d8b220fc04ca9144dc91d9eaacae;hb=fb4c4c3ddc24772f71a64ec02d2c9ddaeb6e9f6b;hpb=779d83085d6c6d37f6c4030130efe649a005696a diff --git a/tools/depends.c b/tools/depends.c index 1422c858..6ac31cdc 100644 --- a/tools/depends.c +++ b/tools/depends.c @@ -1,8 +1,16 @@ -#include "talloc/talloc.h" -#include "string/string.h" +#include +#include +#include +#include +#include #include "tools.h" +#include +#include +#include #include #include +#include +#include static char ** __attribute__((format(printf, 3, 4))) lines_from_cmd(const void *ctx, unsigned int *num, char *format, ...) @@ -19,7 +27,7 @@ lines_from_cmd(const void *ctx, unsigned int *num, char *format, ...) if (!p) err(1, "Executing '%s'", cmd); - buffer = grab_fd(ctx, fileno(p)); + buffer = grab_fd(ctx, fileno(p), NULL); if (!buffer) err(1, "Reading from '%s'", cmd); pclose(p); @@ -27,17 +35,133 @@ lines_from_cmd(const void *ctx, unsigned int *num, char *format, ...) return strsplit(ctx, buffer, "\n", num); } -static char **get_one_deps(const void *ctx, const char *dir, unsigned int *num) +/* Be careful about trying to compile over running programs (parallel make). + * temp_file helps here. */ +static char *compile_info(const void *ctx, const char *dir) +{ + char *info_c_file, *info, *ccandir, *compiled; + size_t len; + int fd; + + /* Copy it to a file with proper .c suffix. */ + info = grab_file(ctx, talloc_asprintf(ctx, "%s/_info", dir), &len); + if (!info) + return NULL; + + info_c_file = temp_file(ctx, ".c"); + fd = open(info_c_file, O_WRONLY|O_CREAT|O_EXCL, 0600); + if (fd < 0) + return NULL; + if (!write_all(fd, info, len)) + return NULL; + + if (close(fd) != 0) + return NULL; + + ccandir = talloc_dirname(ctx, dir); + *strrchr(ccandir, '/') = '\0'; + + compiled = temp_file(ctx, ""); + if (compile_and_link(ctx, info_c_file, ccandir, "", "", "", + compiled)) + return NULL; + return compiled; +} + +static char **get_one_deps(const void *ctx, const char *dir, + unsigned int *num, char **infofile) { char **deps, *cmd; - cmd = talloc_asprintf(ctx, "%s/_info depends", dir); + if (!*infofile) { + *infofile = compile_info(ctx, dir); + if (!*infofile) + errx(1, "Could not compile _info for '%s'", dir); + } + + cmd = talloc_asprintf(ctx, "%s depends", *infofile); deps = lines_from_cmd(cmd, num, "%s", cmd); if (!deps) err(1, "Could not run '%s'", cmd); return deps; } +/* Make copy of src, replacing "from" with "to". */ +static char *replace(const void *ctx, const char *src, + const char *from, const char *to) +{ + char *ret = talloc_strdup(ctx, ""); + unsigned int rlen, len, add; + + rlen = len = 0; + for (;;) { + const char *next = strstr(src+len, from); + if (!next) + add = strlen(src+len) + 1; + else + add = next - (src+len); + + ret = talloc_realloc(ctx, ret, char, rlen + add + strlen(to)+1); + memcpy(ret+rlen, src+len, add); + if (!next) + return ret; + len += add; + rlen += add; + strcpy(ret+rlen, to); + rlen += strlen(to); + len += strlen(from); + } +} + +/* This is a terrible hack. We scan for ccan/ strings. */ +static char **get_one_safe_deps(const void *ctx, + const char *dir, + unsigned int *num, + char **infofile) +{ + char **deps, **lines, *raw, *fname; + unsigned int i, n = 0; + + fname = talloc_asprintf(ctx, "%s/_info", dir); + raw = grab_file(fname, fname, NULL); + if (!raw) + errx(1, "Could not open %s", fname); + + /* Replace \n by actual line breaks, and split it. */ + lines = strsplit(raw, replace(raw, raw, "\\n", "\n"), "\n", &n); + + deps = talloc_array(ctx, char *, n+1); + + for (n = i = 0; lines[i]; i++) { + char *str; + unsigned int len; + + /* Ignore lines starting with # (e.g. #include) */ + if (lines[i][0] == '#') + continue; + + /* Start of line, or after ". */ + if (strstarts(lines[i], "ccan/")) + str = lines[i]; + else { + str = strstr(lines[i], "\"ccan/"); + if (!str) + continue; + str++; + } + + len = strspn(str, "/abcdefghijklmnopqrstuvxwyz12345678980_"); + if (len == 5) + continue; + deps[n++] = talloc_strndup(deps, str, len); + } + deps[n] = NULL; + talloc_free(fname); + if (num) + *num = n; + return deps; +} + static bool have_dep(char **deps, unsigned int num, const char *dep) { unsigned int i; @@ -49,20 +173,29 @@ static bool have_dep(char **deps, unsigned int num, const char *dep) } /* Gets all the dependencies, recursively. */ -char **get_deps(const void *ctx, const char *dir) +static char ** +get_all_deps(const void *ctx, const char *dir, + char **infofile, + char **(*get_one)(const void *, const char *, + unsigned int *, char **)) { char **deps; unsigned int i, num; - deps = get_one_deps(ctx, dir, &num); + deps = get_one(ctx, dir, &num, infofile); for (i = 0; i < num; i++) { char **newdeps; unsigned int j, newnum; + char *subinfo = NULL; + char *subdir; if (!strstarts(deps[i], "ccan/")) continue; - newdeps = get_one_deps(ctx, deps[i], &newnum); + subdir = talloc_asprintf(ctx, "%s/%s", + talloc_dirname(ctx, dir), + deps[i] + strlen("ccan/")); + newdeps = get_one(ctx, subdir, &newnum, &subinfo); /* Should be short, so brute-force out dups. */ for (j = 0; j < newnum; j++) { @@ -76,3 +209,51 @@ char **get_deps(const void *ctx, const char *dir) } return deps; } + +char **get_libs(const void *ctx, const char *dir, + unsigned int *num, char **infofile) +{ + char **libs, *cmd; + + if (!*infofile) { + *infofile = compile_info(ctx, dir); + if (!*infofile) + errx(1, "Could not compile _info for '%s'", dir); + } + + cmd = talloc_asprintf(ctx, "%s libs", *infofile); + libs = lines_from_cmd(cmd, num, "%s", cmd); + if (!libs) + err(1, "Could not run '%s'", cmd); + return libs; +} + +char **get_deps(const void *ctx, const char *dir, + bool recurse, char **infofile) +{ + char *temp = NULL, **ret; + if (!infofile) + infofile = &temp; + + if (!recurse) { + unsigned int num; + ret = get_one_deps(ctx, dir, &num, infofile); + } else + ret = get_all_deps(ctx, dir, infofile, get_one_deps); + + if (infofile == &temp && temp) { + unlink(temp); + talloc_free(temp); + } + return ret; +} + +char **get_safe_ccan_deps(const void *ctx, const char *dir, + bool recurse, char **infofile) +{ + if (!recurse) { + unsigned int num; + return get_one_safe_deps(ctx, dir, &num, infofile); + } + return get_all_deps(ctx, dir, infofile, get_one_safe_deps); +}