source/libs/luajit/LuaJIT-src/src/lj_buf.h

   1 /*
   2 ** Buffer handling.
   3 ** Copyright (C) 2005-2015 Mike Pall. See Copyright Notice in luajit.h
   4 */
   5
   6 #ifndef _LJ_BUF_H
   7 #define _LJ_BUF_H
   8
   9 #include "lj_obj.h"
  10 #include "lj_gc.h"
  11 #include "lj_str.h"
  12
  13 /* Resizable string buffers. Struct definition in lj_obj.h. */
  14 #define sbufB(sb)       (mref((sb)->b, char))
  15 #define sbufP(sb)       (mref((sb)->p, char))
  16 #define sbufE(sb)       (mref((sb)->e, char))
  17 #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)))
  23
  24 /* Buffer management */
  25 LJ_FUNC char *LJ_FASTCALL lj_buf_need2(SBuf *sb, MSize sz);
  26 LJ_FUNC char *LJ_FASTCALL lj_buf_more2(SBuf *sb, MSize sz);
  27 LJ_FUNC void LJ_FASTCALL lj_buf_shrink(lua_State *L, SBuf *sb);
  28 LJ_FUNC char * LJ_FASTCALL lj_buf_tmp(lua_State *L, MSize sz);
  29
  30 static LJ_AINLINE void lj_buf_init(lua_State *L, SBuf *sb)
  31 {
  32   setsbufL(sb, L);
  33   setmref(sb->p, NULL); setmref(sb->e, NULL); setmref(sb->b, NULL);
  34 }
  35
  36 static LJ_AINLINE void lj_buf_reset(SBuf *sb)
  37 {
  38   setmrefr(sb->p, sb->b);
  39 }
  40
  41 static LJ_AINLINE SBuf *lj_buf_tmp_(lua_State *L)
  42 {
  43   SBuf *sb = &G(L)->tmpbuf;
  44   setsbufL(sb, L);
  45   lj_buf_reset(sb);
  46   return sb;
  47 }
  48
  49 static LJ_AINLINE void lj_buf_free(global_State *g, SBuf *sb)
  50 {
  51   lj_mem_free(g, sbufB(sb), sbufsz(sb));
  52 }
  53
  54 static LJ_AINLINE char *lj_buf_need(SBuf *sb, MSize sz)
  55 {
  56   if (LJ_UNLIKELY(sz > sbufsz(sb)))
  57     return lj_buf_need2(sb, sz);
  58   return sbufB(sb);
  59 }
  60
  61 static LJ_AINLINE char *lj_buf_more(SBuf *sb, MSize sz)
  62 {
  63   if (LJ_UNLIKELY(sz > sbufleft(sb)))
  64     return lj_buf_more2(sb, sz);
  65   return sbufP(sb);
  66 }
  67
  68 /* Low-level buffer put operations */
  69 LJ_FUNC SBuf *lj_buf_putmem(SBuf *sb, const void *q, MSize len);
  70 #if LJ_HASJIT
  71 LJ_FUNC SBuf * LJ_FASTCALL lj_buf_putchar(SBuf *sb, int c);
  72 #endif
  73 LJ_FUNC SBuf * LJ_FASTCALL lj_buf_putstr(SBuf *sb, GCstr *s);
  74
  75 static LJ_AINLINE char *lj_buf_wmem(char *p, const void *q, MSize len)
  76 {
  77   return (char *)memcpy(p, q, len) + len;
  78 }
  79
  80 static LJ_AINLINE void lj_buf_putb(SBuf *sb, int c)
  81 {
  82   char *p = lj_buf_more(sb, 1);
  83   *p++ = (char)c;
  84   setsbufP(sb, p);
  85 }
  86
  87 /* High-level buffer put operations */
  88 LJ_FUNCA SBuf * LJ_FASTCALL lj_buf_putstr_reverse(SBuf *sb, GCstr *s);
  89 LJ_FUNCA SBuf * LJ_FASTCALL lj_buf_putstr_lower(SBuf *sb, GCstr *s);
  90 LJ_FUNCA SBuf * LJ_FASTCALL lj_buf_putstr_upper(SBuf *sb, GCstr *s);
  91 LJ_FUNC SBuf *lj_buf_putstr_rep(SBuf *sb, GCstr *s, int32_t rep);
  92 LJ_FUNC SBuf *lj_buf_puttab(SBuf *sb, GCtab *t, GCstr *sep,
  93                             int32_t i, int32_t e);
  94
  95 /* Miscellaneous buffer operations */
  96 LJ_FUNCA GCstr * LJ_FASTCALL lj_buf_tostr(SBuf *sb);
  97 LJ_FUNC GCstr *lj_buf_cat2str(lua_State *L, GCstr *s1, GCstr *s2);
  98 LJ_FUNC uint32_t LJ_FASTCALL lj_buf_ruleb128(const char **pp);
  99
 100 static LJ_AINLINE GCstr *lj_buf_str(lua_State *L, SBuf *sb)
 101 {
 102   return lj_str_new(L, sbufB(sb), sbuflen(sb));
 103 }
 104
 105 #endif