X-Git-Url: https://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=tools%2Ftools.c;h=eacfbf9c8d353ac89bb7e782b19d379e0aefd1a1;hp=5db43df08b20ed8b45139055ee42c58d463322cc;hb=64eca8b85ef0bd13d72381b6c70110e4c342339b;hpb=92ee7edb3a85d458fe7a456a9e9b4a6bb9351059 diff --git a/tools/tools.c b/tools/tools.c index 5db43df0..eacfbf9c 100644 --- a/tools/tools.c +++ b/tools/tools.c @@ -63,10 +63,8 @@ static void killme(int sig) kill(-getpid(), SIGKILL); } -static char *run_with_timeout(const void *ctx, - const char *cmd, - bool *ok, - unsigned *timeout_ms) +char *run_with_timeout(const void *ctx, const char *cmd, + bool *ok, unsigned *timeout_ms) { pid_t pid; int p[2]; @@ -99,7 +97,6 @@ static char *run_with_timeout(const void *ctx, || open("/dev/null", O_RDONLY) != STDIN_FILENO) exit(128); - setpgid(0, 0); signal(SIGALRM, killme); itim.it_interval.tv_sec = itim.it_interval.tv_usec = 0; itim.it_value.tv_sec = *timeout_ms / 1000; @@ -175,7 +172,7 @@ static int unlink_all(char *dir) return 0; } -char *temp_file(const void *ctx, const char *extension) +char *temp_dir(const void *ctx) { /* For first call, create dir. */ while (!tmpdir) { @@ -195,8 +192,25 @@ char *temp_file(const void *ctx, const char *extension) } talloc_set_destructor(tmpdir, unlink_all); } + return tmpdir; +} + +char *temp_file(const void *ctx, const char *extension) +{ + return talloc_asprintf(ctx, "%s/%u%s", + temp_dir(ctx), count++, extension); +} + +char *maybe_temp_file(const void *ctx, const char *extension, bool keep, + const char *srcname) +{ + size_t baselen; + + if (!keep) + return temp_file(ctx, extension); - return talloc_asprintf(ctx, "%s/%u%s", tmpdir, count++, extension); + baselen = strrchr(srcname, '.') - srcname; + return talloc_asprintf(ctx, "%.*s%s", baselen, srcname, extension); } bool move_file(const char *oldname, const char *newname)