Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 1 | #include "cache.h" |
Arnaldo Carvalho de Melo | e7f01d1 | 2012-03-14 12:29:29 -0300 | [diff] [blame] | 2 | #include <linux/kernel.h> |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 3 | |
| 4 | int prefixcmp(const char *str, const char *prefix) |
| 5 | { |
| 6 | for (; ; str++, prefix++) |
| 7 | if (!*prefix) |
| 8 | return 0; |
| 9 | else if (*str != *prefix) |
| 10 | return (unsigned char)*prefix - (unsigned char)*str; |
| 11 | } |
| 12 | |
| 13 | /* |
| 14 | * Used as the default ->buf value, so that people can always assume |
| 15 | * buf is non NULL and ->buf is NUL terminated even for a freshly |
| 16 | * initialized strbuf. |
| 17 | */ |
| 18 | char strbuf_slopbuf[1]; |
| 19 | |
Ingo Molnar | f37a291 | 2009-07-01 12:37:06 +0200 | [diff] [blame] | 20 | void strbuf_init(struct strbuf *sb, ssize_t hint) |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 21 | { |
| 22 | sb->alloc = sb->len = 0; |
| 23 | sb->buf = strbuf_slopbuf; |
| 24 | if (hint) |
| 25 | strbuf_grow(sb, hint); |
| 26 | } |
| 27 | |
| 28 | void strbuf_release(struct strbuf *sb) |
| 29 | { |
| 30 | if (sb->alloc) { |
Arnaldo Carvalho de Melo | 74cf249 | 2013-12-27 16:55:14 -0300 | [diff] [blame] | 31 | zfree(&sb->buf); |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 32 | strbuf_init(sb, 0); |
| 33 | } |
| 34 | } |
| 35 | |
| 36 | char *strbuf_detach(struct strbuf *sb, size_t *sz) |
| 37 | { |
| 38 | char *res = sb->alloc ? sb->buf : NULL; |
| 39 | if (sz) |
| 40 | *sz = sb->len; |
| 41 | strbuf_init(sb, 0); |
| 42 | return res; |
| 43 | } |
| 44 | |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 45 | void strbuf_grow(struct strbuf *sb, size_t extra) |
| 46 | { |
| 47 | if (sb->len + extra + 1 <= sb->len) |
| 48 | die("you want to use way too much memory"); |
| 49 | if (!sb->alloc) |
| 50 | sb->buf = NULL; |
| 51 | ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc); |
| 52 | } |
| 53 | |
Arnaldo Carvalho de Melo | 0741208 | 2016-03-23 16:56:56 -0300 | [diff] [blame] | 54 | void strbuf_addch(struct strbuf *sb, int c) |
| 55 | { |
| 56 | strbuf_grow(sb, 1); |
| 57 | sb->buf[sb->len++] = c; |
| 58 | sb->buf[sb->len] = '\0'; |
| 59 | } |
| 60 | |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 61 | void strbuf_add(struct strbuf *sb, const void *data, size_t len) |
| 62 | { |
| 63 | strbuf_grow(sb, len); |
| 64 | memcpy(sb->buf + sb->len, data, len); |
| 65 | strbuf_setlen(sb, sb->len + len); |
| 66 | } |
| 67 | |
Arnaldo Carvalho de Melo | 0741208 | 2016-03-23 16:56:56 -0300 | [diff] [blame] | 68 | static void strbuf_addv(struct strbuf *sb, const char *fmt, va_list ap) |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 69 | { |
| 70 | int len; |
Namhyung Kim | c711836 | 2015-10-25 00:49:27 +0900 | [diff] [blame] | 71 | va_list ap_saved; |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 72 | |
| 73 | if (!strbuf_avail(sb)) |
| 74 | strbuf_grow(sb, 64); |
Namhyung Kim | c711836 | 2015-10-25 00:49:27 +0900 | [diff] [blame] | 75 | |
| 76 | va_copy(ap_saved, ap); |
Namhyung Kim | f787d95 | 2012-10-23 22:44:50 +0900 | [diff] [blame] | 77 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap); |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 78 | if (len < 0) |
Namhyung Kim | f787d95 | 2012-10-23 22:44:50 +0900 | [diff] [blame] | 79 | die("your vsnprintf is broken"); |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 80 | if (len > strbuf_avail(sb)) { |
| 81 | strbuf_grow(sb, len); |
Namhyung Kim | c711836 | 2015-10-25 00:49:27 +0900 | [diff] [blame] | 82 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap_saved); |
| 83 | va_end(ap_saved); |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 84 | if (len > strbuf_avail(sb)) { |
Namhyung Kim | f787d95 | 2012-10-23 22:44:50 +0900 | [diff] [blame] | 85 | die("this should not happen, your vsnprintf is broken"); |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 86 | } |
| 87 | } |
| 88 | strbuf_setlen(sb, sb->len + len); |
| 89 | } |
| 90 | |
Namhyung Kim | c711836 | 2015-10-25 00:49:27 +0900 | [diff] [blame] | 91 | void strbuf_addf(struct strbuf *sb, const char *fmt, ...) |
| 92 | { |
| 93 | va_list ap; |
| 94 | |
| 95 | va_start(ap, fmt); |
| 96 | strbuf_addv(sb, fmt, ap); |
| 97 | va_end(ap); |
| 98 | } |
| 99 | |
Ingo Molnar | f37a291 | 2009-07-01 12:37:06 +0200 | [diff] [blame] | 100 | ssize_t strbuf_read(struct strbuf *sb, int fd, ssize_t hint) |
Ingo Molnar | 0780060 | 2009-04-20 15:00:56 +0200 | [diff] [blame] | 101 | { |
| 102 | size_t oldlen = sb->len; |
| 103 | size_t oldalloc = sb->alloc; |
| 104 | |
| 105 | strbuf_grow(sb, hint ? hint : 8192); |
| 106 | for (;;) { |
| 107 | ssize_t cnt; |
| 108 | |
| 109 | cnt = read(fd, sb->buf + sb->len, sb->alloc - sb->len - 1); |
| 110 | if (cnt < 0) { |
| 111 | if (oldalloc == 0) |
| 112 | strbuf_release(sb); |
| 113 | else |
| 114 | strbuf_setlen(sb, oldlen); |
| 115 | return -1; |
| 116 | } |
| 117 | if (!cnt) |
| 118 | break; |
| 119 | sb->len += cnt; |
| 120 | strbuf_grow(sb, 8192); |
| 121 | } |
| 122 | |
| 123 | sb->buf[sb->len] = '\0'; |
| 124 | return sb->len - oldlen; |
| 125 | } |