aboutsummaryrefslogtreecommitdiff
path: root/src/lj_buf.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/lj_buf.h')
-rw-r--r--src/lj_buf.h29
1 files changed, 13 insertions, 16 deletions
diff --git a/src/lj_buf.h b/src/lj_buf.h
index a720f83b..268fce5e 100644
--- a/src/lj_buf.h
+++ b/src/lj_buf.h
@@ -11,14 +11,11 @@
11#include "lj_str.h" 11#include "lj_str.h"
12 12
13/* Resizable string buffers. SBuf struct definition in lj_obj.h. */ 13/* Resizable string buffers. SBuf struct definition in lj_obj.h. */
14#define sbufB(sb) (mref((sb)->b, char)) 14#define sbufsz(sb) ((MSize)((sb)->e - (sb)->b))
15#define sbufP(sb) (mref((sb)->p, char)) 15#define sbuflen(sb) ((MSize)((sb)->w - (sb)->b))
16#define sbufE(sb) (mref((sb)->e, char)) 16#define sbufleft(sb) ((MSize)((sb)->e - (sb)->w))
17
17#define sbufL(sb) (mref((sb)->L, lua_State)) 18#define sbufL(sb) (mref((sb)->L, lua_State))
18#define sbufsz(sb) ((MSize)(sbufE((sb)) - sbufB((sb))))
19#define sbuflen(sb) ((MSize)(sbufP((sb)) - sbufB((sb))))
20#define sbufleft(sb) ((MSize)(sbufE((sb)) - sbufP((sb))))
21#define setsbufP(sb, q) (setmref((sb)->p, (q)))
22#define setsbufL(sb, l) (setmref((sb)->L, (l))) 19#define setsbufL(sb, l) (setmref((sb)->L, (l)))
23 20
24/* Buffer management */ 21/* Buffer management */
@@ -30,12 +27,12 @@ LJ_FUNC char * LJ_FASTCALL lj_buf_tmp(lua_State *L, MSize sz);
30static LJ_AINLINE void lj_buf_init(lua_State *L, SBuf *sb) 27static LJ_AINLINE void lj_buf_init(lua_State *L, SBuf *sb)
31{ 28{
32 setsbufL(sb, L); 29 setsbufL(sb, L);
33 setmref(sb->p, NULL); setmref(sb->e, NULL); setmref(sb->b, NULL); 30 sb->w = sb->e = sb->b = NULL;
34} 31}
35 32
36static LJ_AINLINE void lj_buf_reset(SBuf *sb) 33static LJ_AINLINE void lj_buf_reset(SBuf *sb)
37{ 34{
38 setmrefr(sb->p, sb->b); 35 sb->w = sb->b;
39} 36}
40 37
41static LJ_AINLINE SBuf *lj_buf_tmp_(lua_State *L) 38static LJ_AINLINE SBuf *lj_buf_tmp_(lua_State *L)
@@ -48,21 +45,21 @@ static LJ_AINLINE SBuf *lj_buf_tmp_(lua_State *L)
48 45
49static LJ_AINLINE void lj_buf_free(global_State *g, SBuf *sb) 46static LJ_AINLINE void lj_buf_free(global_State *g, SBuf *sb)
50{ 47{
51 lj_mem_free(g, sbufB(sb), sbufsz(sb)); 48 lj_mem_free(g, sb->b, sbufsz(sb));
52} 49}
53 50
54static LJ_AINLINE char *lj_buf_need(SBuf *sb, MSize sz) 51static LJ_AINLINE char *lj_buf_need(SBuf *sb, MSize sz)
55{ 52{
56 if (LJ_UNLIKELY(sz > sbufsz(sb))) 53 if (LJ_UNLIKELY(sz > sbufsz(sb)))
57 return lj_buf_need2(sb, sz); 54 return lj_buf_need2(sb, sz);
58 return sbufB(sb); 55 return sb->b;
59} 56}
60 57
61static LJ_AINLINE char *lj_buf_more(SBuf *sb, MSize sz) 58static LJ_AINLINE char *lj_buf_more(SBuf *sb, MSize sz)
62{ 59{
63 if (LJ_UNLIKELY(sz > sbufleft(sb))) 60 if (LJ_UNLIKELY(sz > sbufleft(sb)))
64 return lj_buf_more2(sb, sz); 61 return lj_buf_more2(sb, sz);
65 return sbufP(sb); 62 return sb->w;
66} 63}
67 64
68/* Low-level buffer put operations */ 65/* Low-level buffer put operations */
@@ -77,9 +74,9 @@ static LJ_AINLINE char *lj_buf_wmem(char *p, const void *q, MSize len)
77 74
78static LJ_AINLINE void lj_buf_putb(SBuf *sb, int c) 75static LJ_AINLINE void lj_buf_putb(SBuf *sb, int c)
79{ 76{
80 char *p = lj_buf_more(sb, 1); 77 char *w = lj_buf_more(sb, 1);
81 *p++ = (char)c; 78 *w++ = (char)c;
82 setsbufP(sb, p); 79 sb->w = w;
83} 80}
84 81
85/* High-level buffer put operations */ 82/* High-level buffer put operations */
@@ -97,7 +94,7 @@ LJ_FUNC uint32_t LJ_FASTCALL lj_buf_ruleb128(const char **pp);
97 94
98static LJ_AINLINE GCstr *lj_buf_str(lua_State *L, SBuf *sb) 95static LJ_AINLINE GCstr *lj_buf_str(lua_State *L, SBuf *sb)
99{ 96{
100 return lj_str_new(L, sbufB(sb), sbuflen(sb)); 97 return lj_str_new(L, sb->b, sbuflen(sb));
101} 98}
102 99
103/* Interim user-accessible string buffer. */ 100/* Interim user-accessible string buffer. */