diff options
Diffstat (limited to 'src/lj_buf.h')
-rw-r--r-- | src/lj_buf.h | 29 |
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); | |||
30 | static LJ_AINLINE void lj_buf_init(lua_State *L, SBuf *sb) | 27 | static 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 | ||
36 | static LJ_AINLINE void lj_buf_reset(SBuf *sb) | 33 | static 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 | ||
41 | static LJ_AINLINE SBuf *lj_buf_tmp_(lua_State *L) | 38 | static 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 | ||
49 | static LJ_AINLINE void lj_buf_free(global_State *g, SBuf *sb) | 46 | static 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 | ||
54 | static LJ_AINLINE char *lj_buf_need(SBuf *sb, MSize sz) | 51 | static 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 | ||
61 | static LJ_AINLINE char *lj_buf_more(SBuf *sb, MSize sz) | 58 | static 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 | ||
78 | static LJ_AINLINE void lj_buf_putb(SBuf *sb, int c) | 75 | static 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 | ||
98 | static LJ_AINLINE GCstr *lj_buf_str(lua_State *L, SBuf *sb) | 95 | static 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. */ |