[PATCH] rsh.c env and quoting cleanup, take 2
[git.git] / path.c
1 /*
2  * I'm tired of doing "vsnprintf()" etc just to open a
3  * file, so here's a "return static buffer with printf"
4  * interface for paths.
5  *
6  * It's obviously not thread-safe. Sue me. But it's quite
7  * useful for doing things like
8  *
9  *   f = open(mkpath("%s/%s.git", base, name), O_RDONLY);
10  *
11  * which is what it's designed for.
12  */
13 #include "cache.h"
14
15 static char pathname[PATH_MAX];
16 static char bad_path[] = "/bad-path/";
17
18 static char *cleanup_path(char *path)
19 {
20         /* Clean it up */
21         if (!memcmp(path, "./", 2)) {
22                 path += 2;
23                 while (*path == '/')
24                         path++;
25         }
26         return path;
27 }
28
29 char *mkpath(const char *fmt, ...)
30 {
31         va_list args;
32         unsigned len;
33
34         va_start(args, fmt);
35         len = vsnprintf(pathname, PATH_MAX, fmt, args);
36         va_end(args);
37         if (len >= PATH_MAX)
38                 return bad_path;
39         return cleanup_path(pathname);
40 }
41
42 char *git_path(const char *fmt, ...)
43 {
44         const char *git_dir;
45         va_list args;
46         unsigned len;
47
48         git_dir = getenv(GIT_DIR_ENVIRONMENT);
49         if (!git_dir) git_dir = DEFAULT_GIT_DIR_ENVIRONMENT;
50         len = strlen(git_dir);
51         if (len > PATH_MAX-100)
52                 return bad_path;
53         memcpy(pathname, git_dir, len);
54         if (len && git_dir[len-1] != '/')
55                 pathname[len++] = '/';
56         va_start(args, fmt);
57         len += vsnprintf(pathname + len, PATH_MAX - len, fmt, args);
58         va_end(args);
59         if (len >= PATH_MAX)
60                 return bad_path;
61         return cleanup_path(pathname);
62 }
63
64
65 /* git_mkstemp() - create tmp file honoring TMPDIR variable */
66 int git_mkstemp(char *path, size_t len, const char *template)
67 {
68         char *env, *pch = path;
69
70         if ((env = getenv("TMPDIR")) == NULL) {
71                 strcpy(pch, "/tmp/");
72                 len -= 5;
73                 pch += 5;
74         } else {
75                 size_t n = snprintf(pch, len, "%s/", env);
76
77                 len -= n;
78                 pch += n;
79         }
80
81         safe_strncpy(pch, template, len);
82
83         return mkstemp(path);
84 }
85
86
87 char *safe_strncpy(char *dest, const char *src, size_t n)
88 {
89         strncpy(dest, src, n);
90         dest[n - 1] = '\0';
91
92         return dest;
93 }