Backport from xs.

This commit is contained in:
default 2022-10-25 09:32:41 +02:00
parent d2095105de
commit fa870ab358
8 changed files with 70 additions and 51 deletions

69
xs.h
View file

@ -34,6 +34,9 @@ typedef char d_char;
/* auto-destroyable strings */ /* auto-destroyable strings */
#define xs __attribute__ ((__cleanup__ (_xs_destroy))) d_char #define xs __attribute__ ((__cleanup__ (_xs_destroy))) d_char
void *xs_free(void *ptr);
void *_xs_realloc(void *ptr, size_t size, const char *file, int line);
#define xs_realloc(ptr, size) _xs_realloc(ptr, size, __FILE__, __LINE__)
int _xs_blk_size(int sz); int _xs_blk_size(int sz);
void _xs_destroy(char **var); void _xs_destroy(char **var);
#define xs_debug() raise(SIGTRAP) #define xs_debug() raise(SIGTRAP)
@ -41,7 +44,6 @@ xstype xs_type(const char *data);
int xs_size(const char *data); int xs_size(const char *data);
int xs_is_null(const char *data); int xs_is_null(const char *data);
d_char *xs_dup(const char *data); d_char *xs_dup(const char *data);
void *xs_realloc(void *ptr, size_t size);
d_char *xs_expand(d_char *data, int offset, int size); d_char *xs_expand(d_char *data, int offset, int size);
d_char *xs_collapse(d_char *data, int offset, int size); d_char *xs_collapse(d_char *data, int offset, int size);
d_char *xs_insert_m(d_char *data, int offset, const char *mem, int size); d_char *xs_insert_m(d_char *data, int offset, const char *mem, int size);
@ -80,19 +82,56 @@ d_char *xs_number_new(double f);
double xs_number_get(const char *v); double xs_number_get(const char *v);
const char *xs_number_str(const char *v); const char *xs_number_str(const char *v);
extern int _xs_debug;
#ifdef XS_IMPLEMENTATION #ifdef XS_IMPLEMENTATION
int _xs_debug = 0; void *_xs_realloc(void *ptr, size_t size, const char *file, int line)
{
d_char *ndata = realloc(ptr, size);
if (ndata == NULL) {
fprintf(stderr, "**OUT OF MEMORY**\n");
abort();
}
#ifdef XS_DEBUG
if (ndata != ptr) {
FILE *f = fopen("xs_memory.out", "a");
if (ptr != NULL)
fprintf(f, "%p b\n", ptr);
fprintf(f, "%p a %ld %s %d\n", ndata, size, file, line);
fclose(f);
}
#endif
return ndata;
}
void *xs_free(void *ptr)
{
#ifdef XS_DEBUG
if (ptr != NULL) {
FILE *f = fopen("xs_memory.out", "a");
fprintf(f, "%p b\n", ptr);
fclose(f);
}
#endif
free(ptr);
return NULL;
}
void _xs_destroy(char **var) void _xs_destroy(char **var)
{ {
/*
if (_xs_debug) if (_xs_debug)
printf("_xs_destroy %p\n", var); printf("_xs_destroy %p\n", var);
*/
free(*var); xs_free(*var);
} }
@ -222,19 +261,6 @@ d_char *xs_dup(const char *data)
} }
void *xs_realloc(void *ptr, size_t size)
{
d_char *ndata = realloc(ptr, size);
if (ndata == NULL) {
fprintf(stderr, "**OUT OF MEMORY**\n");
abort();
}
return ndata;
}
d_char *xs_expand(d_char *data, int offset, int size) d_char *xs_expand(d_char *data, int offset, int size)
/* opens a hole in data */ /* opens a hole in data */
{ {
@ -325,11 +351,10 @@ d_char *xs_fmt(const char *fmt, ...)
va_end(ap); va_end(ap);
if (n > 0) { if (n > 0) {
n = _xs_blk_size(n + 1); s = xs_realloc(NULL, _xs_blk_size(n + 1));
s = calloc(n, 1);
va_start(ap, fmt); va_start(ap, fmt);
vsnprintf(s, n, fmt, ap); vsnprintf(s, n + 1, fmt, ap);
va_end(ap); va_end(ap);
} }

View file

@ -54,7 +54,7 @@ static int _data_callback(void *buffer, size_t size,
/* open space */ /* open space */
pd->size += sz; pd->size += sz;
pd->data = realloc(pd->data, pd->size + 1); pd->data = xs_realloc(pd->data, pd->size + 1);
/* copy data */ /* copy data */
memcpy(pd->data + pd->offset, buffer, sz); memcpy(pd->data + pd->offset, buffer, sz);
@ -166,7 +166,7 @@ d_char *xs_http_request(char *method, char *url, d_char *headers,
ipd.data[ipd.size] = '\0'; ipd.data[ipd.size] = '\0';
} }
else else
free(ipd.data); xs_free(ipd.data);
return response; return response;
} }

View file

@ -50,8 +50,7 @@ d_char *xs_hex_dec(const char *hex, int *size)
int i; int i;
if (sscanf(&hex[n], "%02x", &i) == 0) { if (sscanf(&hex[n], "%02x", &i) == 0) {
/* decoding error */ /* decoding error */
free(s); return xs_free(s);
return NULL;
} }
else else
*p = i; *p = i;
@ -137,8 +136,7 @@ d_char *xs_base64_dec(const char *data, int *size)
if (ss == NULL) { if (ss == NULL) {
/* not a base64 char */ /* not a base64 char */
free(s); return xs_free(s);
return NULL;
} }
cs[n] = ss - b64_tbl; cs[n] = ss - b64_tbl;

View file

@ -252,10 +252,8 @@ d_char *xs_httpd_request(FILE *f, d_char **payload, int *p_size)
req = xs_dict_append(req, "q_vars", q_vars); req = xs_dict_append(req, "q_vars", q_vars);
req = xs_dict_append(req, "p_vars", p_vars); req = xs_dict_append(req, "p_vars", p_vars);
if (errno) { if (errno)
free(req); req = xs_free(req);
req = NULL;
}
return req; return req;
} }

View file

@ -368,7 +368,7 @@ d_char *_xs_json_loads_array(const char **json, js_type *t)
l = xs_list_append(l, v); l = xs_list_append(l, v);
while (*t == JS_INCOMPLETE) { while (*t == JS_INCOMPLETE) {
free(_xs_json_loads_lexer(&s, &tt)); xs_free(_xs_json_loads_lexer(&s, &tt));
if (tt == JS_CBRACK) if (tt == JS_CBRACK)
*t = JS_ARRAY; *t = JS_ARRAY;
@ -392,10 +392,8 @@ d_char *_xs_json_loads_array(const char **json, js_type *t)
*t = JS_ERROR; *t = JS_ERROR;
} }
if (*t == JS_ERROR) { if (*t == JS_ERROR)
free(l); l = xs_free(l);
l = NULL;
}
*json = s; *json = s;
@ -421,7 +419,7 @@ d_char *_xs_json_loads_object(const char **json, js_type *t)
*t = JS_OBJECT; *t = JS_OBJECT;
else else
if (tt == JS_STRING) { if (tt == JS_STRING) {
free(_xs_json_loads_lexer(&s, &tt)); xs_free(_xs_json_loads_lexer(&s, &tt));
if (tt == JS_COLON) { if (tt == JS_COLON) {
xs *v1; xs *v1;
@ -433,7 +431,7 @@ d_char *_xs_json_loads_object(const char **json, js_type *t)
d = xs_dict_append(d, k1, v1); d = xs_dict_append(d, k1, v1);
while (*t == JS_INCOMPLETE) { while (*t == JS_INCOMPLETE) {
free(_xs_json_loads_lexer(&s, &tt)); xs_free(_xs_json_loads_lexer(&s, &tt));
if (tt == JS_CCURLY) if (tt == JS_CCURLY)
*t = JS_OBJECT; *t = JS_OBJECT;
@ -442,7 +440,7 @@ d_char *_xs_json_loads_object(const char **json, js_type *t)
xs *k = _xs_json_loads_lexer(&s, &tt); xs *k = _xs_json_loads_lexer(&s, &tt);
if (tt == JS_STRING) { if (tt == JS_STRING) {
free(_xs_json_loads_lexer(&s, &tt)); xs_free(_xs_json_loads_lexer(&s, &tt));
if (tt == JS_COLON) { if (tt == JS_COLON) {
xs *v; xs *v;
@ -474,10 +472,8 @@ d_char *_xs_json_loads_object(const char **json, js_type *t)
else else
*t = JS_ERROR; *t = JS_ERROR;
if (*t == JS_ERROR) { if (*t == JS_ERROR)
free(d); d = xs_free(d);
d = NULL;
}
*json = s; *json = s;
@ -491,7 +487,7 @@ d_char *xs_json_loads(const char *json)
d_char *v = NULL; d_char *v = NULL;
js_type t; js_type t;
free(_xs_json_loads_lexer(&json, &t)); xs_free(_xs_json_loads_lexer(&json, &t));
if (t == JS_OBRACK) if (t == JS_OBRACK)
v = _xs_json_loads_array(&json, &t); v = _xs_json_loads_array(&json, &t);

View file

@ -140,7 +140,7 @@ d_char *xs_rsa_sign(const char *secret, const char *mem, int size)
BIO_free(b); BIO_free(b);
RSA_free(rsa); RSA_free(rsa);
free(sig); xs_free(sig);
return signature; return signature;
} }
@ -211,7 +211,7 @@ d_char *xs_evp_sign(const char *secret, const char *mem, int size)
EVP_MD_CTX_free(mdctx); EVP_MD_CTX_free(mdctx);
EVP_PKEY_free(pkey); EVP_PKEY_free(pkey);
BIO_free(b); BIO_free(b);
free(sig); xs_free(sig);
return signature; return signature;
} }

View file

@ -22,7 +22,9 @@ xs_set *xs_set_new(int elems)
/* creates a new set with a maximum of size hashed data */ /* creates a new set with a maximum of size hashed data */
{ {
int sz = sizeof(struct _xs_set) + sizeof(int) * elems; int sz = sizeof(struct _xs_set) + sizeof(int) * elems;
xs_set *s = calloc(sz, 1); xs_set *s = xs_realloc(NULL, sz);
memset(s, '\0', sz);
/* initialize */ /* initialize */
s->elems = elems; s->elems = elems;
@ -35,8 +37,8 @@ xs_set *xs_set_new(int elems)
void xs_set_free(xs_set *s) void xs_set_free(xs_set *s)
/* frees a set */ /* frees a set */
{ {
free(s->list); xs_free(s->list);
free(s); xs_free(s);
} }

View file

@ -1 +1 @@
/* 9f90d5958755ec33c6c4946427f71de37af7500e */ /* 223ebe7a241f24ba043b32cb30d0428ac0d1dd5f */