Christopher Li | 812666c | 2005-04-26 12:00:58 -0700 | [diff] [blame] | 1 | #include "cache.h" |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 2 | |
Junio C Hamano | 698a68b | 2008-01-03 01:23:12 -0800 | [diff] [blame] | 3 | int prefixcmp(const char *str, const char *prefix) |
| 4 | { |
| 5 | for (; ; str++, prefix++) |
| 6 | if (!*prefix) |
| 7 | return 0; |
| 8 | else if (*str != *prefix) |
| 9 | return (unsigned char)*prefix - (unsigned char)*str; |
| 10 | } |
| 11 | |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 12 | /* |
| 13 | * Used as the default ->buf value, so that people can always assume |
| 14 | * buf is non NULL and ->buf is NUL terminated even for a freshly |
| 15 | * initialized strbuf. |
| 16 | */ |
| 17 | char strbuf_slopbuf[1]; |
| 18 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 19 | void strbuf_init(struct strbuf *sb, size_t hint) |
| 20 | { |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 21 | sb->alloc = sb->len = 0; |
| 22 | sb->buf = strbuf_slopbuf; |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 23 | if (hint) |
| 24 | strbuf_grow(sb, hint); |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 25 | } |
| 26 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 27 | void strbuf_release(struct strbuf *sb) |
| 28 | { |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 29 | if (sb->alloc) { |
| 30 | free(sb->buf); |
| 31 | strbuf_init(sb, 0); |
| 32 | } |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 33 | } |
| 34 | |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 35 | char *strbuf_detach(struct strbuf *sb, size_t *sz) |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 36 | { |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 37 | char *res = sb->alloc ? sb->buf : NULL; |
| 38 | if (sz) |
| 39 | *sz = sb->len; |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 40 | strbuf_init(sb, 0); |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 41 | return res; |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 42 | } |
| 43 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 44 | void strbuf_attach(struct strbuf *sb, void *buf, size_t len, size_t alloc) |
| 45 | { |
| 46 | strbuf_release(sb); |
| 47 | sb->buf = buf; |
| 48 | sb->len = len; |
| 49 | sb->alloc = alloc; |
| 50 | strbuf_grow(sb, 0); |
| 51 | sb->buf[sb->len] = '\0'; |
| 52 | } |
| 53 | |
| 54 | void strbuf_grow(struct strbuf *sb, size_t extra) |
| 55 | { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 56 | if (sb->len + extra + 1 <= sb->len) |
| 57 | die("you want to use way too much memory"); |
Pierre Habouzit | b315c5c | 2007-09-27 12:58:23 +0200 | [diff] [blame] | 58 | if (!sb->alloc) |
| 59 | sb->buf = NULL; |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 60 | ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc); |
| 61 | } |
| 62 | |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 63 | void strbuf_rtrim(struct strbuf *sb) |
| 64 | { |
| 65 | while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1])) |
| 66 | sb->len--; |
| 67 | sb->buf[sb->len] = '\0'; |
| 68 | } |
| 69 | |
Pierre Habouzit | 45f66f6 | 2007-09-24 11:25:03 +0200 | [diff] [blame] | 70 | int strbuf_cmp(struct strbuf *a, struct strbuf *b) |
| 71 | { |
| 72 | int cmp; |
| 73 | if (a->len < b->len) { |
| 74 | cmp = memcmp(a->buf, b->buf, a->len); |
| 75 | return cmp ? cmp : -1; |
| 76 | } else { |
| 77 | cmp = memcmp(a->buf, b->buf, b->len); |
| 78 | return cmp ? cmp : a->len != b->len; |
| 79 | } |
| 80 | } |
| 81 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 82 | void strbuf_splice(struct strbuf *sb, size_t pos, size_t len, |
| 83 | const void *data, size_t dlen) |
| 84 | { |
| 85 | if (pos + len < pos) |
| 86 | die("you want to use way too much memory"); |
| 87 | if (pos > sb->len) |
| 88 | die("`pos' is too far after the end of the buffer"); |
| 89 | if (pos + len > sb->len) |
| 90 | die("`pos + len' is too far after the end of the buffer"); |
| 91 | |
| 92 | if (dlen >= len) |
| 93 | strbuf_grow(sb, dlen - len); |
| 94 | memmove(sb->buf + pos + dlen, |
| 95 | sb->buf + pos + len, |
| 96 | sb->len - pos - len); |
| 97 | memcpy(sb->buf + pos, data, dlen); |
| 98 | strbuf_setlen(sb, sb->len + dlen - len); |
| 99 | } |
| 100 | |
Pierre Habouzit | c76689d | 2007-09-20 00:42:12 +0200 | [diff] [blame] | 101 | void strbuf_insert(struct strbuf *sb, size_t pos, const void *data, size_t len) |
| 102 | { |
| 103 | strbuf_splice(sb, pos, 0, data, len); |
| 104 | } |
| 105 | |
| 106 | void strbuf_remove(struct strbuf *sb, size_t pos, size_t len) |
| 107 | { |
| 108 | strbuf_splice(sb, pos, len, NULL, 0); |
| 109 | } |
| 110 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 111 | void strbuf_add(struct strbuf *sb, const void *data, size_t len) |
| 112 | { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 113 | strbuf_grow(sb, len); |
| 114 | memcpy(sb->buf + sb->len, data, len); |
| 115 | strbuf_setlen(sb, sb->len + len); |
| 116 | } |
| 117 | |
René Scharfe | 91db267 | 2007-11-10 12:16:05 +0100 | [diff] [blame] | 118 | void strbuf_adddup(struct strbuf *sb, size_t pos, size_t len) |
| 119 | { |
| 120 | strbuf_grow(sb, len); |
| 121 | memcpy(sb->buf + sb->len, sb->buf + pos, len); |
| 122 | strbuf_setlen(sb, sb->len + len); |
| 123 | } |
| 124 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 125 | void strbuf_addf(struct strbuf *sb, const char *fmt, ...) |
| 126 | { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 127 | int len; |
| 128 | va_list ap; |
| 129 | |
Shawn O. Pearce | f141bd8 | 2007-11-13 03:22:44 -0500 | [diff] [blame] | 130 | if (!strbuf_avail(sb)) |
| 131 | strbuf_grow(sb, 64); |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 132 | va_start(ap, fmt); |
| 133 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap); |
| 134 | va_end(ap); |
Shawn O. Pearce | f141bd8 | 2007-11-13 03:22:44 -0500 | [diff] [blame] | 135 | if (len < 0) |
| 136 | die("your vsnprintf is broken"); |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 137 | if (len > strbuf_avail(sb)) { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 138 | strbuf_grow(sb, len); |
| 139 | va_start(ap, fmt); |
| 140 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap); |
| 141 | va_end(ap); |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 142 | if (len > strbuf_avail(sb)) { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 143 | die("this should not happen, your snprintf is broken"); |
| 144 | } |
| 145 | } |
| 146 | strbuf_setlen(sb, sb->len + len); |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 147 | } |
| 148 | |
Marco Costalba | c3a670d | 2008-02-09 15:40:19 +0100 | [diff] [blame] | 149 | void strbuf_expand(struct strbuf *sb, const char *format, expand_fn_t fn, |
| 150 | void *context) |
René Scharfe | cde75e5 | 2007-11-09 01:49:42 +0100 | [diff] [blame] | 151 | { |
| 152 | for (;;) { |
Marco Costalba | c3a670d | 2008-02-09 15:40:19 +0100 | [diff] [blame] | 153 | const char *percent; |
| 154 | size_t consumed; |
René Scharfe | cde75e5 | 2007-11-09 01:49:42 +0100 | [diff] [blame] | 155 | |
| 156 | percent = strchrnul(format, '%'); |
| 157 | strbuf_add(sb, format, percent - format); |
| 158 | if (!*percent) |
| 159 | break; |
| 160 | format = percent + 1; |
| 161 | |
Marco Costalba | c3a670d | 2008-02-09 15:40:19 +0100 | [diff] [blame] | 162 | consumed = fn(sb, format, context); |
| 163 | if (consumed) |
| 164 | format += consumed; |
| 165 | else |
René Scharfe | cde75e5 | 2007-11-09 01:49:42 +0100 | [diff] [blame] | 166 | strbuf_addch(sb, '%'); |
| 167 | } |
| 168 | } |
| 169 | |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 170 | size_t strbuf_fread(struct strbuf *sb, size_t size, FILE *f) |
| 171 | { |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 172 | size_t res; |
| 173 | |
| 174 | strbuf_grow(sb, size); |
| 175 | res = fread(sb->buf + sb->len, 1, size, f); |
| 176 | if (res > 0) { |
| 177 | strbuf_setlen(sb, sb->len + res); |
| 178 | } |
| 179 | return res; |
| 180 | } |
| 181 | |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 182 | ssize_t strbuf_read(struct strbuf *sb, int fd, size_t hint) |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 183 | { |
| 184 | size_t oldlen = sb->len; |
| 185 | |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 186 | strbuf_grow(sb, hint ? hint : 8192); |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 187 | for (;;) { |
| 188 | ssize_t cnt; |
| 189 | |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 190 | cnt = xread(fd, sb->buf + sb->len, sb->alloc - sb->len - 1); |
| 191 | if (cnt < 0) { |
| 192 | strbuf_setlen(sb, oldlen); |
| 193 | return -1; |
| 194 | } |
| 195 | if (!cnt) |
| 196 | break; |
| 197 | sb->len += cnt; |
Pierre Habouzit | f1696ee | 2007-09-10 12:35:04 +0200 | [diff] [blame] | 198 | strbuf_grow(sb, 8192); |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 199 | } |
| 200 | |
| 201 | sb->buf[sb->len] = '\0'; |
| 202 | return sb->len - oldlen; |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 203 | } |
| 204 | |
Pierre Habouzit | e6c019d | 2007-09-17 11:19:04 +0200 | [diff] [blame] | 205 | int strbuf_getline(struct strbuf *sb, FILE *fp, int term) |
Pierre Habouzit | 917c9a7 | 2007-09-15 15:56:50 +0200 | [diff] [blame] | 206 | { |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 207 | int ch; |
Pierre Habouzit | e6c019d | 2007-09-17 11:19:04 +0200 | [diff] [blame] | 208 | |
| 209 | strbuf_grow(sb, 0); |
| 210 | if (feof(fp)) |
| 211 | return EOF; |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 212 | |
| 213 | strbuf_reset(sb); |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 214 | while ((ch = fgetc(fp)) != EOF) { |
| 215 | if (ch == term) |
| 216 | break; |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 217 | strbuf_grow(sb, 1); |
| 218 | sb->buf[sb->len++] = ch; |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 219 | } |
Pierre Habouzit | e6c019d | 2007-09-17 11:19:04 +0200 | [diff] [blame] | 220 | if (ch == EOF && sb->len == 0) |
| 221 | return EOF; |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 222 | |
Pierre Habouzit | b449f4c | 2007-09-06 13:20:05 +0200 | [diff] [blame] | 223 | sb->buf[sb->len] = '\0'; |
Pierre Habouzit | e6c019d | 2007-09-17 11:19:04 +0200 | [diff] [blame] | 224 | return 0; |
Junio C Hamano | d1df574 | 2005-04-25 18:26:45 -0700 | [diff] [blame] | 225 | } |
Kristian Høgsberg | a9390b9 | 2007-09-17 20:06:46 -0400 | [diff] [blame] | 226 | |
Pierre Habouzit | 387e7e1 | 2007-09-27 15:25:55 +0200 | [diff] [blame] | 227 | int strbuf_read_file(struct strbuf *sb, const char *path, size_t hint) |
Kristian Høgsberg | a9390b9 | 2007-09-17 20:06:46 -0400 | [diff] [blame] | 228 | { |
| 229 | int fd, len; |
| 230 | |
| 231 | fd = open(path, O_RDONLY); |
| 232 | if (fd < 0) |
| 233 | return -1; |
Pierre Habouzit | 387e7e1 | 2007-09-27 15:25:55 +0200 | [diff] [blame] | 234 | len = strbuf_read(sb, fd, hint); |
Kristian Høgsberg | a9390b9 | 2007-09-17 20:06:46 -0400 | [diff] [blame] | 235 | close(fd); |
| 236 | if (len < 0) |
| 237 | return -1; |
| 238 | |
| 239 | return len; |
| 240 | } |