One Level Up Top Level

src/lj_buf.h - luajit-2.0-src

Functions defined

Macros defined

Source code

  1. /*
  2. ** Buffer handling.
  3. ** Copyright (C) 2005-2015 Mike Pall. See Copyright Notice in luajit.h
  4. */

  6. #ifndef _LJ_BUF_H
  7. ‌#define _LJ_BUF_H

  9. #include "lj_obj.h"
  10. #include "lj_gc.h"
  11. #include "lj_str.h"

  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)))

  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);

  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. }

  36. ‌static LJ_AINLINE void lj_buf_reset(SBuf *sb)
  37. {
  38.   setmrefr(sb->p, sb->b);
  39. }

  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. }

  49. ‌static LJ_AINLINE void lj_buf_free(global_State *g, SBuf *sb)
  50. {
  51.   lj_mem_free(g, sbufB(sb), sbufsz(sb));
  52. }

  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. }

  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. }

  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);

  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. }

  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. }

  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);

  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);

  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. }

  105. #endif

One Level Up Top Level