Signed-off-by: Lukas Sandström <lukass@xxxxxxxxxxxxxxxx> --- Junio C Hamano wrote: > Heh, after getting burned by that NUL thingy, I was waiting for somebody > to step up. Thanks. Here we go then. Two freshly baked patches. Note that this is a pretty straight buffers -> strbuf's conversion, but I think it will be a good start for further hardening of mailinfo. strbuf.c | 70 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ strbuf.h | 6 +++++ 2 files changed, 76 insertions(+), 0 deletions(-) diff --git a/strbuf.c b/strbuf.c index 4aed752..28d6776 100644 --- a/strbuf.c +++ b/strbuf.c @@ -60,6 +60,18 @@ void strbuf_grow(struct strbuf *sb, size_t extra) ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc); } +void strbuf_trim(struct strbuf *sb) +{ + char *b = sb->buf; + while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1])) + sb->len--; + while(sb->len > 0 && isspace(*b)) { + b++; + sb->len--; + } + memmove(sb->buf, b, sb->len); + sb->buf[sb->len] = '\0'; +} void strbuf_rtrim(struct strbuf *sb) { while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1])) @@ -67,6 +79,64 @@ void strbuf_rtrim(struct strbuf *sb) sb->buf[sb->len] = '\0'; } +void strbuf_ltrim(struct strbuf *sb) +{ + char *b = sb->buf; + while(sb->len > 0 && isspace(*b)) { + b++; + sb->len--; + } + memmove(sb->buf, b, sb->len); + sb->buf[sb->len] = '\0'; +} + +void strbuf_tolower(struct strbuf *sb) +{ + int i; + for (i = 0; i < sb->len; i++) + sb->buf[i] = tolower(sb->buf[i]); +} + +struct strbuf ** strbuf_split(struct strbuf *sb, int delim) +{ + int alloc = 2, pos = 0; + char *n, *p; + struct strbuf **ret; + struct strbuf *t; + + ret = xcalloc(alloc, sizeof(struct strbuf *)); + p = n = sb->buf; + while (n < sb->buf + sb->len) { + int len; + n = memchr(n, delim, sb->len - (n - sb->buf)); + if (pos + 1 >= alloc) { + alloc = alloc * 2; + ret = xrealloc(ret, sizeof(struct strbuf *) * alloc); + } + if (!n) + n = sb->buf + sb->len - 1; + len = n - p + 1; + t = xmalloc(sizeof(struct strbuf)); + strbuf_init(t, len); + strbuf_add(t, p, len); + ret[pos] = t; + ret[++pos] = NULL; + p = ++n; + } + return ret; +} + +void strbuf_list_free(struct strbuf ** sbs) +{ + struct strbuf **s = sbs; + + while(*s) { + strbuf_release(*s); + free(*s++); + } + free(sbs); +} + int strbuf_cmp(struct strbuf *a, struct strbuf *b) { int cmp; diff --git a/strbuf.h b/strbuf.h index faec229..577d14e 100644 --- a/strbuf.h +++ b/strbuf.h @@ -77,8 +77,14 @@ static inline void strbuf_setlen(struct strbuf *sb, size_t len) { #define strbuf_reset(sb) strbuf_setlen(sb, 0) /*----- content related -----*/ +extern void strbuf_trim(struct strbuf *); extern void strbuf_rtrim(struct strbuf *); +extern void strbuf_ltrim(struct strbuf *); extern int strbuf_cmp(struct strbuf *, struct strbuf *); +extern void strbuf_tolower(struct strbuf *); + +extern struct strbuf ** strbuf_split(struct strbuf*, int delim); +extern void strbuf_list_free(struct strbuf **); /*----- add data in your buffer -----*/ static inline void strbuf_addch(struct strbuf *sb, int c) { -- 1.5.4.5 -- To unsubscribe from this list: send the line "unsubscribe git" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html