path.c 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  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.perf", base, name), O_RDONLY);
  10. *
  11. * which is what it's designed for.
  12. */
  13. #include "cache.h"
  14. static char bad_path[] = "/bad-path/";
  15. /*
  16. * Two hacks:
  17. */
  18. static const char *get_perf_dir(void)
  19. {
  20. return ".";
  21. }
  22. #ifdef NO_STRLCPY
  23. size_t strlcpy(char *dest, const char *src, size_t size)
  24. {
  25. size_t ret = strlen(src);
  26. if (size) {
  27. size_t len = (ret >= size) ? size - 1 : ret;
  28. memcpy(dest, src, len);
  29. dest[len] = '\0';
  30. }
  31. return ret;
  32. }
  33. #endif
  34. static char *get_pathname(void)
  35. {
  36. static char pathname_array[4][PATH_MAX];
  37. static int idx;
  38. return pathname_array[3 & ++idx];
  39. }
  40. static char *cleanup_path(char *path)
  41. {
  42. /* Clean it up */
  43. if (!memcmp(path, "./", 2)) {
  44. path += 2;
  45. while (*path == '/')
  46. path++;
  47. }
  48. return path;
  49. }
  50. static char *perf_vsnpath(char *buf, size_t n, const char *fmt, va_list args)
  51. {
  52. const char *perf_dir = get_perf_dir();
  53. size_t len;
  54. len = strlen(perf_dir);
  55. if (n < len + 1)
  56. goto bad;
  57. memcpy(buf, perf_dir, len);
  58. if (len && !is_dir_sep(perf_dir[len-1]))
  59. buf[len++] = '/';
  60. len += vsnprintf(buf + len, n - len, fmt, args);
  61. if (len >= n)
  62. goto bad;
  63. return cleanup_path(buf);
  64. bad:
  65. strlcpy(buf, bad_path, n);
  66. return buf;
  67. }
  68. char *perf_pathdup(const char *fmt, ...)
  69. {
  70. char path[PATH_MAX];
  71. va_list args;
  72. va_start(args, fmt);
  73. (void)perf_vsnpath(path, sizeof(path), fmt, args);
  74. va_end(args);
  75. return xstrdup(path);
  76. }
  77. char *mkpath(const char *fmt, ...)
  78. {
  79. va_list args;
  80. unsigned len;
  81. char *pathname = get_pathname();
  82. va_start(args, fmt);
  83. len = vsnprintf(pathname, PATH_MAX, fmt, args);
  84. va_end(args);
  85. if (len >= PATH_MAX)
  86. return bad_path;
  87. return cleanup_path(pathname);
  88. }
  89. char *perf_path(const char *fmt, ...)
  90. {
  91. const char *perf_dir = get_perf_dir();
  92. char *pathname = get_pathname();
  93. va_list args;
  94. unsigned len;
  95. len = strlen(perf_dir);
  96. if (len > PATH_MAX-100)
  97. return bad_path;
  98. memcpy(pathname, perf_dir, len);
  99. if (len && perf_dir[len-1] != '/')
  100. pathname[len++] = '/';
  101. va_start(args, fmt);
  102. len += vsnprintf(pathname + len, PATH_MAX - len, fmt, args);
  103. va_end(args);
  104. if (len >= PATH_MAX)
  105. return bad_path;
  106. return cleanup_path(pathname);
  107. }
  108. /* strip arbitrary amount of directory separators at end of path */
  109. static inline int chomp_trailing_dir_sep(const char *path, int len)
  110. {
  111. while (len && is_dir_sep(path[len - 1]))
  112. len--;
  113. return len;
  114. }
  115. /*
  116. * If path ends with suffix (complete path components), returns the
  117. * part before suffix (sans trailing directory separators).
  118. * Otherwise returns NULL.
  119. */
  120. char *strip_path_suffix(const char *path, const char *suffix)
  121. {
  122. int path_len = strlen(path), suffix_len = strlen(suffix);
  123. while (suffix_len) {
  124. if (!path_len)
  125. return NULL;
  126. if (is_dir_sep(path[path_len - 1])) {
  127. if (!is_dir_sep(suffix[suffix_len - 1]))
  128. return NULL;
  129. path_len = chomp_trailing_dir_sep(path, path_len);
  130. suffix_len = chomp_trailing_dir_sep(suffix, suffix_len);
  131. }
  132. else if (path[--path_len] != suffix[--suffix_len])
  133. return NULL;
  134. }
  135. if (path_len && !is_dir_sep(path[path_len - 1]))
  136. return NULL;
  137. return strndup(path, chomp_trailing_dir_sep(path, path_len));
  138. }