Ejemplo n.º 1
0
static int uread_chars (lua_State *L, FILE *f, size_t n) {
  size_t rlen;  /* how much to read */
  size_t nr;  /* number of chars actually read */
  luaL_Buffer b;
  luaL_wbuffinit(L, &b);
  rlen = LUAL_BUFFERSIZE / 2;  /* try to read that much each time */
  do {
    char *p = luaL_prepbuffer(&b);
    if (rlen > n) rlen = n;  /* cannot read more than asked */
    nr = fread(p, sizeof(lua_WChar), rlen, f);
    luaL_addwsize(&b, nr);
    n -= nr;  /* still have to read `n' chars */
  } while (n > 0 && nr == rlen);  /* until end of count or eof */
  luaL_pushresult(&b);  /* close buffer */
  return (n == 0 || lua_strlen(L, -1) > 0);
}
Ejemplo n.º 2
0
static int uread_line (lua_State *L, FILE *f) {
  luaL_Buffer b;
  luaL_wbuffinit(L, &b);
  for (;;) {
    size_t l;
    lua_WChar *p = (lua_WChar*)luaL_prepbuffer(&b);
    if (fgetws(p, LUAL_BUFFERSIZE / 2, f) == NULL) {  /* eof? */
      luaL_pushresult(&b);  /* close buffer */
      return (lua_strlen(L, -1) > 0);  /* check whether read something */
    }
    l = lua_WChar_len(p);
    if (p[l-1] != '\n')
      luaL_addwsize(&b, l);
    else {
      luaL_addwsize(&b, l - 1);  /* do not include `eol' */
      luaL_pushresult(&b);  /* close buffer */
      return 1;  /* read at least an `eol' */
    }
  }
}
Ejemplo n.º 3
0
static int str_gsub (lua_State *L) {
  size_t srcl;
  const lua_WChar *src = luaL_checklwstring(L, 1, &srcl);
  const lua_WChar *p = luaL_checkwstring(L, 2);
  int max_s = luaL_optint(L, 4, srcl+1);
  int anchor = (*p == '^') ? (p++, 1) : 0;
  int n = 0;
  MatchState ms;
  luaL_Buffer b;
  luaL_argcheck(L,
    lua_gettop(L) >= 3 && (lua_iswstring(L, 3) || lua_isfunction(L, 3)),
    3, "string or function expected");
  luaL_wbuffinit(L, &b);
  ms.L = L;
  ms.src_init = src;
  ms.src_end = src+srcl;
  while (n < max_s) {
    const lua_WChar *e;
    ms.level = 0;
    e = match(&ms, src, p);
    if (e) {
      n++;
      add_s(&ms, &b, src, e);
    }
    if (e && e>src) /* non empty match? */
      src = e;  /* skip it */
    else if (src < ms.src_end)
      luaL_putwchar(&b, *src++);
    else break;
    if (anchor) break;
  }
  luaL_addlwstring(&b, src, ms.src_end-src);
  luaL_pushresult(&b);
  lua_pushnumber(L, n);  /* number of substitutions */
  return 2;
}
Ejemplo n.º 4
0
static int str_lualex(lua_State *L) {
  size_t l = 0;
  const lua_WChar *str = luaL_checklwstring(L, 1, &l);
  int isWide = luaL_checkint(L, 2) != 0;
  size_t i;
  luaL_Buffer b;
  luaL_wbuffinit(L, &b);

  for (i = 0; i < l; ++i)
  {
//	int needUnicodeZero = 1;

    switch (str[i])
	{
      case '\\':
		++i;
        switch (str[i]) {
          case 'a': luaL_putwchar(&b, '\a'); break;
          case 'b': luaL_putwchar(&b, '\b'); break;
          case 'f': luaL_putwchar(&b, '\f'); break;
          case 'n': luaL_putwchar(&b, '\n'); break;
          case 'r': luaL_putwchar(&b, '\r'); break;
          case 't': luaL_putwchar(&b, '\t'); break;
          case 'v': luaL_putwchar(&b, '\v'); break;
          case 'x': {
			  int ch;
			  ++i;
			  ch = tolower(str[i]);
              if (!isdigit(ch) && !(ch >= 'a' && ch <= 'f') )
			  {
				  --i;
				  luaL_putwchar(&b, 'x');
			  }
			  else {  /* \xxx */
				  int c = 0;
				  int i = 0;
				  int numDigits = isWide ? 4 : 2;
				  do {
					  ch = towlower(str[i]);
					  if (iswdigit((wint_t)ch))
					    c = 16*c + (ch-'0');
					  else if (ch >= 'a' && ch <= 'f')
						c = 16*c + (ch-'a') + 10;
					  ++i;
					  ch = towlower(str[i]);
				  } while (++i<numDigits && iswdigit((wint_t)ch) || (ch >= 'a' && ch <= 'f'));
				  luaL_putwchar(&b, c);
//				  needUnicodeZero = 0;
			  }
			  break;
          }
          default: {
            if (!iswdigit(str[i]))
				luaL_putwchar(&b, str[i]);
            else {  /* \xxx */
              int c = 0;
              int count = 0;
              do {
                c = 10*c + (str[i]-'0');
				++i;
              } while (++count<3 && iswdigit(str[i]));
              luaL_putwchar(&b, c);
            }
          }
        }
        break;
      default:
		  luaL_putwchar(&b, str[i]);
    }
	if (isWide) //  &&  needUnicodeZero)
	  luaL_putwchar(&b, str[i]);
  }

  luaL_pushresult(&b);
  return 1;
}
Ejemplo n.º 5
0
int wstr_format_helper (luaL_Buffer* b, lua_State *L, int arg) {
  size_t sfl;
  const lua_WChar *strfrmt = luaL_checklwstring(L, arg, &sfl);
  const lua_WChar *strfrmt_end = strfrmt+sfl;
  luaL_wbuffinit(L, b);
  while (strfrmt < strfrmt_end) {
    if (*strfrmt != '%')
      luaL_putwchar(b, *strfrmt++);
    else if (*++strfrmt == '%')
      luaL_putwchar(b, *strfrmt++);  /* %% */
    else { /* format item */
      lua_WChar form[MAX_FORMAT];  /* to store the format (`%...') */
      lua_WChar buff[MAX_ITEM];  /* to store the formatted item */
      int hasprecision = 0;
      if (iswdigit(*strfrmt) && *(strfrmt+1) == '$')
        return luaL_error(L, "obsolete `format' option (d$)");
      arg++;
      strfrmt = scanformat(L, strfrmt, form, &hasprecision);
      switch (*strfrmt++) {
        case 'b':
		{
		  buff[1] = buff[2] = buff[3] = buff[4] = buff[5] = buff[6] = buff[7] = buff[8] = 0;
		  switch (*strfrmt++)
		  {
		  case 'b': {
            unsigned int num = (unsigned int)luaL_checkint(L, arg);
		    buff[0] = (unsigned char)num;
            luaL_addlwstring(b, buff, 1);
			break;
          }
		  case 'd': {
            unsigned int num = (unsigned int)luaL_checkint(L, arg);
		    *(unsigned int*)(&buff) = num;
            luaL_addlwstring(b, buff, 4);
			break;
          }
		  case 'w': {
            unsigned int num = (unsigned int)luaL_checkint(L, arg);
			*(unsigned short*)(&buff) = (unsigned short)num;
            luaL_addlwstring(b, buff, 2);
			break;
          }
		  case 'f': {
            float numF = (float)luaL_checknumber(L, arg);
			*(float*)(&buff) = numF;
            luaL_addlwstring(b, buff, 4);
			break;
          }
		  case 'F': {
            double numD = (double)luaL_checknumber(L, arg);
			*(double*)(&buff) = numD;
            luaL_addlwstring(b, buff, 8);
			break;
          }
			  
		  default:
			  break;
		  }
		  buff[0] = 0;
		  break;
		}
        case 'c':  case 'd':  case 'i':
          swprintf(buff, form, luaL_checkint(L, arg));
          break;
        case 'o':  case 'u':  case 'x':  case 'X':
          //?? How should this be for integers?
          swprintf(buff, form, (unsigned int)(luaL_checknumber(L, arg)));
          break;
        case 'e':  case 'E': case 'f':
        case 'g': case 'G':
          swprintf(buff, form, luaL_checknumber(L, arg));
          break;
        case 'q':
          luaI_addquoted(L, b, arg);
          continue;  /* skip the `addsize' at the end */
        case 'Q':
          luaI_addquotedbinary(L, b, arg);
          continue;  /* skip the `addsize' at the end */
        case 's': {
          size_t l;
          const lua_WChar *s = luaL_checklwstring(L, arg, &l);
          if (!hasprecision && l >= 100) {
            /* no precision and string is too long to be formatted;
               keep original string */
            lua_pushvalue(L, arg);
            luaL_addvalue(b);
            continue;  /* skip the `addsize' at the end */
          }
          else {
            swprintf(buff, form, s);
            break;
          }
        }
        default:  /* also treat cases `pnLlh' */
          luaL_error(L, "invalid option in `format'");
      }
      luaL_addlwstring(b, buff, lua_WChar_len(buff));
    }
  }

  return 1;
}
Ejemplo n.º 6
0
static int luaB_towstring (lua_State *L) {
  char buff[64];
  switch (lua_type(L, 1)) {
    case LUA_TNUMBER:
      lua_pushwstring(L, lua_towstring(L, 1));
      return 1;
    case LUA_TSTRING: {
	 luaL_Buffer b;
	 size_t l;
	 size_t i;
	 const char *s = lua_tostring(L, 1);
	 l = lua_strlen(L, 1);
	 if (l == 0)
	 {
	     lua_pushwstring(L, L"");
	 }
	 else
	 {
		 luaL_wbuffinit(L, &b);
		 for (i=0; i<l; i++)
			luaL_putwchar(&b, (lua_WChar)s[i]);
		 luaL_pushresult(&b);
	 }
      return 1;
    }
    case LUA_TWSTRING:
      lua_pushvalue(L, 1);
      return 1;
    case LUA_TTABLE:
      sprintf(buff, "%.40s: %p", lua_type(L, 1), lua_topointer(L, 1));
      break;
    case LUA_TFUNCTION:
      sprintf(buff, "function: %p", lua_topointer(L, 1));
      break;
    case LUA_TUSERDATA: {
      const char *t = lua_typename(L, lua_type(L, 1));
      if (strcmp(t, "userdata") == 0)
        sprintf(buff, "userdata: %p", lua_touserdata(L, 1));
      else
        sprintf(buff, "%.40s: %p", t, lua_touserdata(L, 1));
      break;
    }
    case LUA_TNIL:
      lua_pushliteral(L, "nil");
      return 1;
    case LUA_TTHREAD:
      sprintf(buff, "thread: %p", (void *)lua_tothread(L, 1));
      break;
    default:
      luaL_argerror(L, 1, "value expected");
  }

  {
	 luaL_Buffer b;
	 size_t l;
	 size_t i;
	 const lua_WChar *s = lua_towstring(L, 1);
	 l = lua_strlen(L, 1);
	 luaL_wbuffinit(L, &b);
	 for (i=0; i<l; i++)
		luaL_putwchar(&b, (lua_WChar)(s[i]));
	 luaL_pushresult(&b);
  }

/*  lua_pushstring(L, buff);*/
  return 1;
}
Ejemplo n.º 7
0
int wwstr_format_helper (luaL_Buffer* b, lua_State *L, int arg) {
    size_t sfl;
    const lua_WChar *strfrmt = luaL_checklwstring(L, arg, &sfl);
    const lua_WChar *strfrmt_end = strfrmt+sfl;
    luaL_wbuffinit(L, b);
    while (strfrmt < strfrmt_end) {
        if (*strfrmt != WL_ESC)
            luaL_addwchar(b, *strfrmt++);
        else if (*++strfrmt == WL_ESC)
            luaL_addwchar(b, *strfrmt++);  /* %% */
        else { /* format item */
            lua_WChar form[MAX_FORMAT];  /* to store the format (`%...') */
            lua_WChar buff[MAX_ITEM];  /* to store the formatted item */
            char _form[MAX_FORMAT];
            char _buff[MAX_ITEM];
            if (iswdigit(*strfrmt) && *(strfrmt+1) == '$')
                return luaL_error(L, "obsolete `format' option (d$)");
            arg++;
            strfrmt = wscanformat(L, strfrmt, form);
            switch (*strfrmt++) {
            case 'c': {
                translate_wide_to_single(form, _form);
                sprintf(_buff, _form, luaL_checkint(L, arg));
                translate_single_to_wide(_buff, buff);
                break;
            }
            case 'd':
            case 'i': {
                waddintlen(form);
                translate_wide_to_single(form, _form);
                sprintf(_buff, _form, (LUA_INTFRM_T)luaL_checknumber(L, arg));
                translate_single_to_wide(_buff, buff);
                break;
            }
            case 'o':
            case 'u':
            case 'x':
            case 'X': {
                //?? How should this be for integers?
                translate_wide_to_single(form, _form);
                sprintf(_buff, _form, (unsigned LUA_INTFRM_T)luaL_checknumber(L, arg));
                translate_single_to_wide(_buff, buff);
                break;
            }
            case 'e':
            case 'E':
            case 'f':
            case 'g':
            case 'G': {
                translate_wide_to_single(form, _form);
                sprintf(_buff, _form, luaL_checknumber(L, arg));
                translate_single_to_wide(_buff, buff);
                break;
            }
            case 'q': {
                luaI_waddquoted(L, b, arg);
                continue;  /* skip the 'addsize' at the end */
            }
            case 's': {
                size_t l;
                const lua_WChar *s = luaL_checklwstring(L, arg, &l);
                (void)s;
                if (!lua_WChar_chr(form, '.') && l >= 100) {
                    /* no precision and string is too long to be formatted;
                       keep original string */
                    lua_pushvalue(L, arg);
                    luaL_addvalue(b);
                    continue;  /* skip the `addsize' at the end */
                }
                else {
                    assert(0);
//			swprintf((wchar_t*)buff, (wchar_t*)form, s);
                    break;
                }
            }
            case 'b':
            {
                buff[1] = buff[2] = buff[3] = buff[4] = buff[5] = buff[6] = buff[7] = buff[8] = 0;
                switch (*strfrmt++)
                {
                case 'b': {
                    unsigned int num = (unsigned int)luaL_checkint(L, arg);
                    buff[0] = (unsigned char)num;
                    luaL_addlwstring(b, buff, 1);
                    break;
                }
                case 'd': {
                    unsigned int num = (unsigned int)luaL_checkint(L, arg);
                    *(unsigned int*)(&buff) = num;
                    luaL_addlwstring(b, buff, 4);
                    break;
                }
                case 'w': {
                    unsigned int num = (unsigned int)luaL_checkint(L, arg);
                    *(unsigned short*)(&buff) = (unsigned short)num;
                    luaL_addlwstring(b, buff, 2);
                    break;
                }
                case 'f': {
                    float numF = (float)luaL_checknumber(L, arg);
                    *(float*)(&buff) = numF;
                    luaL_addlwstring(b, buff, 4);
                    break;
                }
                case 'F': {
                    double numD = (double)luaL_checknumber(L, arg);
                    *(double*)(&buff) = numD;
                    luaL_addlwstring(b, buff, 8);
                    break;
                }

                default:
                    break;
                }
                buff[0] = 0;

                break;
            }
            }
            luaL_addlwstring(b, buff, lua_WChar_len(buff));
        }
    }
    return 1;
}