nils/gluon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gluon-web: clean up buffer handling

Browse Source
This commit is contained in:
Matthias Schiffer 2018-02-22 17:41:38 +01:00
parent 624d969c52
commit 5a20f9794c
No known key found for this signature in database
GPG Key ID: 16EF3F64CB201D9C
5 changed files with 72 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
package/gluon-web/src/template_lmo.c
View File

@ -27,7 +27,7 @@ static inline uint16_t get_le16(const uint8_t *d) {
return (((uint16_t)d[1]) << 8) | d[0]; return (((uint16_t)d[1]) << 8) | d[0];
} }
static uint32_t sfh_hash(const void *input, int len) static uint32_t sfh_hash(const void *input, size_t len)
{ {
const uint8_t *data = input; const uint8_t *data = input;
uint32_t hash = len, tmp; uint32_t hash = len, tmp;
@ -228,7 +228,7 @@ static lmo_entry_t * lmo_find_entry(lmo_archive_t *ar, uint32_t hash)
return NULL; return NULL;
} }
int lmo_translate(const char *key, int keylen, char **out, int *outlen) int lmo_translate(const char *key, size_t keylen, char **out, size_t *outlen)
{ {
uint32_t hash; uint32_t hash;
lmo_entry_t *e; lmo_entry_t *e;

2
package/gluon-web/src/template_lmo.h
View File

@ -70,6 +70,6 @@ typedef struct lmo_catalog lmo_catalog_t;
int lmo_load_catalog(const char *lang, const char *dir); int lmo_load_catalog(const char *lang, const char *dir);
int lmo_change_catalog(const char *lang); int lmo_change_catalog(const char *lang);
int lmo_translate(const char *key, int keylen, char **out, int *outlen); int lmo_translate(const char *key, size_t keylen, char **out, size_t *outlen);
#endif #endif

8
package/gluon-web/src/template_lualib.c
View File

@ -67,13 +67,13 @@ static int template_L_parse_string(lua_State *L)
static int template_L_pcdata(lua_State *L) static int template_L_pcdata(lua_State *L)
{ {
size_t len = 0; size_t len = 0, outlen;
const char *str = luaL_checklstring(L, 1, &len); const char *str = luaL_checklstring(L, 1, &len);
char *res = pcdata(str, len); char *res = pcdata(str, len, &outlen);
if (res != NULL) if (res != NULL)
{ {
lua_pushstring(L, res); lua_pushlstring(L, res, outlen);
free(res); free(res);
return 1; return 1;
@ -92,7 +92,7 @@ static int template_L_load_catalog(lua_State *L) {
static int template_L_translate(lua_State *L) { static int template_L_translate(lua_State *L) {
size_t len; size_t len;
char *tr; char *tr;
int trlen; size_t trlen;
const char *key = luaL_checklstring(L, 1, &len); const char *key = luaL_checklstring(L, 1, &len);
switch (lmo_translate(key, len, &tr, &trlen)) switch (lmo_translate(key, len, &tr, &trlen))

121
package/gluon-web/src/template_utils.c
View File

@ -20,27 +20,24 @@
#include "template_utils.h" #include "template_utils.h"
#include "template_lmo.h" #include "template_lmo.h"
/* initialize a buffer object */ #include <stdlib.h>
struct template_buffer * buf_init(int size) #include <stdio.h>
{ #include <string.h>
struct template_buffer *buf;
if (size <= 0) /* initialize a buffer object */
struct template_buffer * buf_init(size_t size)
{
if (size < 1024)
size = 1024; size = 1024;
buf = malloc(sizeof(*buf)); struct template_buffer *buf = malloc(sizeof(*buf));
if (buf != NULL) if (buf != NULL) {
{
buf->fill = 0;
buf->size = size; buf->size = size;
buf->data = malloc(buf->size); buf->data = malloc(buf->size);
if (buf->data != NULL) if (buf->data != NULL) {
{
buf->dptr = buf->data; buf->dptr = buf->data;
buf->data[0] = 0;
return buf; return buf;
} }
@ -51,57 +48,48 @@ struct template_buffer * buf_init(int size)
} }
/* grow buffer */ /* grow buffer */
static int buf_grow(struct template_buffer *buf, int size) static bool buf_grow(struct template_buffer *buf, size_t len)
{ {
unsigned int off = (buf->dptr - buf->data); size_t off = buf->dptr - buf->data, left = buf->size - off;
char *data; if (len <= left)
return true;
if (size <= 0) size_t diff = len - left;
size = 1024; if (diff < 1024)
diff = 1024;
data = realloc(buf->data, buf->size + size); char *data = realloc(buf->data, buf->size + diff);
if (data == NULL)
return false;
if (data != NULL)
{
buf->data = data; buf->data = data;
buf->dptr = data + off; buf->dptr = data + off;
buf->size += size; buf->size += diff;
return buf->size; return true;
}
return 0;
} }
/* put one char into buffer object */ /* put one char into buffer object */
static int buf_putchar(struct template_buffer *buf, char c) static bool buf_putchar(struct template_buffer *buf, char c)
{ {
if( ((buf->fill + 1) >= buf->size) && !buf_grow(buf, 0) ) if (!buf_grow(buf, 1))
return 0; return false;
*(buf->dptr++) = c; *(buf->dptr++) = c;
*(buf->dptr) = 0;
buf->fill++; return true;
return 1;
} }
/* append data to buffer */ /* append data to buffer */
int buf_append(struct template_buffer *buf, const char *s, int len) bool buf_append(struct template_buffer *buf, const char *s, size_t len)
{ {
if ((buf->fill + len + 1) >= buf->size) if (!buf_grow(buf, len))
{ return false;
if (!buf_grow(buf, len + 1))
return 0;
}
memcpy(buf->dptr, s, len); memcpy(buf->dptr, s, len);
buf->fill += len;
buf->dptr += len; buf->dptr += len;
*(buf->dptr) = 0; return true;
return len;
} }
/* destroy buffer object and return pointer to data */ /* destroy buffer object and return pointer to data */
@ -115,7 +103,7 @@ char * buf_destroy(struct template_buffer *buf)
/* calculate the number of expected continuation chars */ /* calculate the number of expected continuation chars */
static inline int mb_num_chars(unsigned char c) static inline size_t mb_num_chars(unsigned char c)
{ {
if ((c & 0xE0) == 0xC0) if ((c & 0xE0) == 0xC0)
return 2; return 2;
@ -132,14 +120,14 @@ static inline int mb_num_chars(unsigned char c)
} }
/* test whether the given byte is a valid continuation char */ /* test whether the given byte is a valid continuation char */
static inline int mb_is_cont(unsigned char c) static inline bool mb_is_cont(unsigned char c)
{ {
return ((c >= 0x80) && (c <= 0xBF)); return ((c >= 0x80) && (c <= 0xBF));
} }
/* test whether the byte sequence at the given pointer with the given /* test whether the byte sequence at the given pointer with the given
* length is the shortest possible representation of the code point */ * length is the shortest possible representation of the code point */
static inline int mb_is_shortest(unsigned char *s, int n) static inline bool mb_is_shortest(const unsigned char *s, size_t n)
{ {
switch (n) switch (n)
{ {
@ -179,19 +167,19 @@ static inline int mb_is_shortest(unsigned char *s, int n)
((*(s+5) >> 6) == 0x02)); ((*(s+5) >> 6) == 0x02));
} }
return 1; return true;
} }
/* test whether the byte sequence at the given pointer with the given /* test whether the byte sequence at the given pointer with the given
* length is an UTF-16 surrogate */ * length is an UTF-16 surrogate */
static inline int mb_is_surrogate(unsigned char *s, int n) static inline bool mb_is_surrogate(const unsigned char *s, size_t n)
{ {
return ((n == 3) && (*s == 0xED) && (*(s+1) >= 0xA0) && (*(s+1) <= 0xBF)); return ((n == 3) && (*s == 0xED) && (*(s+1) >= 0xA0) && (*(s+1) <= 0xBF));
} }
/* test whether the byte sequence at the given pointer with the given /* test whether the byte sequence at the given pointer with the given
* length is an illegal UTF-8 code point */ * length is an illegal UTF-8 code point */
static inline int mb_is_illegal(unsigned char *s, int n) static inline bool mb_is_illegal(const unsigned char *s, size_t n)
{ {
return ((n == 3) && (*s == 0xEF) && (*(s+1) == 0xBF) && return ((n == 3) && (*s == 0xEF) && (*(s+1) == 0xBF) &&
(*(s+2) >= 0xBE) && (*(s+2) <= 0xBF)); (*(s+2) >= 0xBE) && (*(s+2) <= 0xBF));
@ -200,14 +188,13 @@ static inline int mb_is_illegal(unsigned char *s, int n)
/* scan given source string, validate UTF-8 sequence and store result /* scan given source string, validate UTF-8 sequence and store result
* in given buffer object */ * in given buffer object */
static int validate_utf8(unsigned char **s, unsigned int l, struct template_buffer *buf) static size_t validate_utf8(const unsigned char **s, size_t l, struct template_buffer *buf)
{ {
unsigned char *ptr = *s; const unsigned char *ptr = *s;
unsigned int o = 0, v, n; size_t o = 0, v, n;
/* ascii byte without null */ /* ascii byte without null */
if ((*(ptr+0) >= 0x01) && (*(ptr+0) <= 0x7F)) if ((*(ptr+0) >= 0x01) && (*(ptr+0) <= 0x7F)) {
{
if (!buf_putchar(buf, *ptr++)) if (!buf_putchar(buf, *ptr++))
return 0; return 0;
@ -215,8 +202,7 @@ static int validate_utf8(unsigned char **s, unsigned int l, struct template_buff
} }
/* multi byte sequence */ /* multi byte sequence */
else if ((n = mb_num_chars(*ptr)) > 1) else if ((n = mb_num_chars(*ptr)) > 1) {
{
/* count valid chars */ /* count valid chars */
for (v = 1; (v <= n) && ((o+v) < l) && mb_is_cont(*(ptr+v)); v++); for (v = 1; (v <= n) && ((o+v) < l) && mb_is_cont(*(ptr+v)); v++);
@ -238,7 +224,7 @@ static int validate_utf8(unsigned char **s, unsigned int l, struct template_buff
!mb_is_surrogate(ptr, n) && !mb_is_illegal(ptr, n)) !mb_is_surrogate(ptr, n) && !mb_is_illegal(ptr, n))
{ {
/* copy sequence */ /* copy sequence */
if (!buf_append(buf, (char *)ptr, n)) if (!buf_append(buf, (const char *)ptr, n))
return 0; return 0;
} }
@ -259,8 +245,7 @@ static int validate_utf8(unsigned char **s, unsigned int l, struct template_buff
} }
/* invalid byte (0x00) */ /* invalid byte (0x00) */
else else {
{
if (!buf_putchar(buf, '?')) /* or 0xEF, 0xBF, 0xBD */ if (!buf_putchar(buf, '?')) /* or 0xEF, 0xBF, 0xBD */
return 0; return 0;
@ -275,11 +260,11 @@ static int validate_utf8(unsigned char **s, unsigned int l, struct template_buff
/* Sanitize given string and strip all invalid XML bytes /* Sanitize given string and strip all invalid XML bytes
* Validate UTF-8 sequences * Validate UTF-8 sequences
* Escape XML control chars */ * Escape XML control chars */
char * pcdata(const char *s, unsigned int l) char * pcdata(const char *s, size_t l, size_t *outl)
{ {
struct template_buffer *buf = buf_init(l); struct template_buffer *buf = buf_init(l);
unsigned char *ptr = (unsigned char *)s; const unsigned char *ptr = (const unsigned char *)s;
unsigned int o, v; size_t o, v;
char esq[8]; char esq[8];
int esl; int esl;
@ -328,16 +313,17 @@ char * pcdata(const char *s, unsigned int l)
} }
} }
*outl = buf_length(buf);
return buf_destroy(buf); return buf_destroy(buf);
} }
void luastr_escape(struct template_buffer *out, const char *s, unsigned int l, int escape_xml) void luastr_escape(struct template_buffer *out, const char *s, size_t l, bool escape_xml)
{ {
int esl; int esl;
char esq[8]; char esq[8];
char *ptr; const char *ptr;
for (ptr = (char *)s; ptr < (s + l); ptr++) for (ptr = s; ptr < (s + l); ptr++)
{ {
switch (*ptr) switch (*ptr)
{ {
@ -360,8 +346,7 @@ void luastr_escape(struct template_buffer *out, const char *s, unsigned int l, i
case '&': case '&':
case '<': case '<':
case '>': case '>':
if (escape_xml) if (escape_xml) {
{
esl = snprintf(esq, sizeof(esq), "&#%i;", *ptr); esl = snprintf(esq, sizeof(esq), "&#%i;", *ptr);
buf_append(out, esq, esl); buf_append(out, esq, esl);
break; break;
@ -373,10 +358,10 @@ void luastr_escape(struct template_buffer *out, const char *s, unsigned int l, i
} }
} }
void luastr_translate(struct template_buffer *out, const char *s, unsigned int l, int escape_xml) void luastr_translate(struct template_buffer *out, const char *s, size_t l, bool escape_xml)
{ {
char *tr; char *tr;
int trlen; size_t trlen;
if (!lmo_translate(s, l, &tr, &trlen)) if (!lmo_translate(s, l, &tr, &trlen))
luastr_escape(out, tr, trlen, escape_xml); luastr_escape(out, tr, trlen, escape_xml);

22
package/gluon-web/src/template_utils.h
View File

@ -20,33 +20,31 @@
#ifndef _TEMPLATE_UTILS_H_ #ifndef _TEMPLATE_UTILS_H_
#define _TEMPLATE_UTILS_H_ #define _TEMPLATE_UTILS_H_
#include <stdlib.h> #include <stdbool.h>
#include <stdio.h> #include <stddef.h>
#include <string.h>
/* buffer object */ /* buffer object */
struct template_buffer { struct template_buffer {
char *data; char *data;
char *dptr; char *dptr;
unsigned int size; size_t size;
unsigned int fill;
}; };
struct template_buffer * buf_init(int size); struct template_buffer * buf_init(size_t size);
int buf_append(struct template_buffer *buf, const char *s, int len); bool buf_append(struct template_buffer *buf, const char *s, size_t len);
char * buf_destroy(struct template_buffer *buf); char * buf_destroy(struct template_buffer *buf);
/* read buffer length */ /* read buffer length */
static inline int buf_length(struct template_buffer *buf) static inline size_t buf_length(struct template_buffer *buf)
{ {
return buf->fill; return buf->dptr - buf->data;
} }
char * pcdata(const char *s, unsigned int l); char * pcdata(const char *s, size_t l, size_t *outl);
void luastr_escape(struct template_buffer *out, const char *s, unsigned int l, int escape_xml); void luastr_escape(struct template_buffer *out, const char *s, size_t l, bool escape_xml);
void luastr_translate(struct template_buffer *out, const char *s, unsigned int l, int escape_xml); void luastr_translate(struct template_buffer *out, const char *s, size_t l, bool escape_xml);
#endif #endif