Adding strbuf.
lifting a portion of the strbuf api/implementation found in archived. This will serve as a base for an generic buffer api.
This commit is contained in:
parent
378de035de
commit
5220e42038
2 changed files with 188 additions and 0 deletions
90
strbuf.c
Normal file
90
strbuf.c
Normal file
|
|
@ -0,0 +1,90 @@
|
||||||
|
/* strbuf.c
|
||||||
|
*
|
||||||
|
* Copyright (C) 2010-2011 Henrik Hautakoski <henrik@fiktivkod.org>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include "error.h"
|
||||||
|
#include "strbuf.h"
|
||||||
|
|
||||||
|
#define CHNK_SIZE 128
|
||||||
|
|
||||||
|
char strbuf_null = '\0';
|
||||||
|
|
||||||
|
void strbuf_init(strbuf_t *s) {
|
||||||
|
|
||||||
|
s->buf = &strbuf_null;
|
||||||
|
s->size = s->len = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void strbuf_expand(strbuf_t *s, size_t len) {
|
||||||
|
|
||||||
|
if (s->len + len + 1 < s->size)
|
||||||
|
return;
|
||||||
|
if (!s->size)
|
||||||
|
s->buf = NULL;
|
||||||
|
|
||||||
|
do
|
||||||
|
s->size += CHNK_SIZE;
|
||||||
|
while(s->len + len + 1 > s->size);
|
||||||
|
|
||||||
|
s->buf = realloc(s->buf, s->size);
|
||||||
|
}
|
||||||
|
|
||||||
|
char* strbuf_release(strbuf_t *s) {
|
||||||
|
|
||||||
|
char *ret;
|
||||||
|
|
||||||
|
if (!s->size)
|
||||||
|
ret = malloc(1);
|
||||||
|
else if (s->len + 1 != s->size)
|
||||||
|
ret = realloc(s->buf, s->len + 1);
|
||||||
|
else
|
||||||
|
ret = s->buf;
|
||||||
|
|
||||||
|
strbuf_init(s);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
void strbuf_free(strbuf_t *s) {
|
||||||
|
|
||||||
|
if (!s->size)
|
||||||
|
return;
|
||||||
|
|
||||||
|
free(s->buf);
|
||||||
|
strbuf_init(s);
|
||||||
|
}
|
||||||
|
|
||||||
|
void strbuf_attach(strbuf_t *s, void *buf, size_t len, size_t size) {
|
||||||
|
|
||||||
|
strbuf_free(s);
|
||||||
|
s->buf = buf;
|
||||||
|
s->len = len;
|
||||||
|
s->size = size;
|
||||||
|
strbuf_expand(s, 0);
|
||||||
|
s->buf[s->len] = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
void strbuf_append(strbuf_t *s, const void *ptr, size_t len) {
|
||||||
|
|
||||||
|
strbuf_expand(s, len);
|
||||||
|
memcpy(s->buf + s->len, ptr, len);
|
||||||
|
strbuf_setlen(s, s->len + len);
|
||||||
|
}
|
||||||
98
strbuf.h
Normal file
98
strbuf.h
Normal file
|
|
@ -0,0 +1,98 @@
|
||||||
|
/* strbuf.h
|
||||||
|
*
|
||||||
|
* Copyright (C) 2010-2011 Henrik Hautakoski <henrik@fiktivkod.org>
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
* MA 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef __STRBUF_H
|
||||||
|
#define __STRBUF_H
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
#include <stddef.h>
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
char *buf;
|
||||||
|
size_t len;
|
||||||
|
size_t size;
|
||||||
|
} strbuf_t;
|
||||||
|
|
||||||
|
extern char strbuf_null;
|
||||||
|
|
||||||
|
#define STRBUF_INIT { &strbuf_null, 0, 0 }
|
||||||
|
|
||||||
|
void strbuf_init(strbuf_t *s);
|
||||||
|
|
||||||
|
static inline size_t strbuf_avail(strbuf_t *s) {
|
||||||
|
|
||||||
|
return s->size ? s->size - (s->len + 1) : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void strbuf_expand(strbuf_t *s, size_t len);
|
||||||
|
|
||||||
|
static inline void strbuf_setlen(strbuf_t *s, size_t len) {
|
||||||
|
|
||||||
|
if (!s->size)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (len >= s->size)
|
||||||
|
len = s->size - 1;
|
||||||
|
s->len = len;
|
||||||
|
s->buf[s->len] = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void strbuf_reduce(strbuf_t *s, size_t len) {
|
||||||
|
|
||||||
|
if (len > s->len)
|
||||||
|
len = s->len;
|
||||||
|
|
||||||
|
strbuf_setlen(s, s->len - len);
|
||||||
|
}
|
||||||
|
|
||||||
|
char* strbuf_release(strbuf_t *s);
|
||||||
|
|
||||||
|
void strbuf_free(strbuf_t *s);
|
||||||
|
|
||||||
|
void strbuf_attach(strbuf_t *s, void *str, size_t len, size_t size);
|
||||||
|
|
||||||
|
void strbuf_append(strbuf_t *s, const void *ptr, size_t len);
|
||||||
|
|
||||||
|
static inline void strbuf_append_str(strbuf_t *s, const char *str) {
|
||||||
|
|
||||||
|
strbuf_append(s, str, strlen(str));
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void strbuf_append_ch(strbuf_t *s, char ch) {
|
||||||
|
|
||||||
|
strbuf_expand(s, 1);
|
||||||
|
s->buf[s->len++] = ch;
|
||||||
|
s->buf[s->len] = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void strbuf_append_repeat(strbuf_t *s, char ch, size_t len) {
|
||||||
|
|
||||||
|
strbuf_expand(s, len);
|
||||||
|
memset(s->buf + s->len, ch, len);
|
||||||
|
strbuf_setlen(s, s->len + len);
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline void strbuf_term(strbuf_t *s, char ch) {
|
||||||
|
|
||||||
|
if (s->buf[s->len-1] != ch)
|
||||||
|
strbuf_append_ch(s, ch);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif /* __STRBUF_H */
|
||||||
Reference in a new issue