mirrors/cosmopolitan
1
0
Fork 0
mirror of https://github.com/jart/cosmopolitan.git synced 2025-08-10 20:00:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add escaping of Lua strings to redbean

Browse source
This commit is contained in:
Paul Kulchenko 2021-11-11 22:36:38 -08:00
parent 1bf1028e36
commit 290967b70e
1 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tool/net/redbean.c
View file

@ -4755,13 +4755,33 @@ static int EncodeJsonData(lua_State *L, char **buf, int level, char *numformat)
return 0; return 0;
} }
static void EscapeLuaString(char *s, size_t len, char **buf) {
static char OCT[8] = "01234567";
char escape[4] = "\\000";
appendd(buf, "\"", 1);
for (size_t i = 0; i < len; i++) {
if (s[i] == '\\' || s[i] == '\"' || s[i] == '\n' ||
s[i] == '\0' || s[i] == '\r') {
escape[1] = OCT[(s[i] >> 6) & 7];
escape[2] = OCT[(s[i] >> 3) & 7];
escape[3] = OCT[(s[i] >> 0) & 7];
appendd(buf, escape, 4);
} else {
appendd(buf, s+i, 1);
}
}
appendd(buf, "\"", 1);
}
static int EncodeLuaData(lua_State *L, char **buf, int level, char *numformat) { static int EncodeLuaData(lua_State *L, char **buf, int level, char *numformat) {
size_t idx = -1; size_t idx = -1;
size_t tbllen, buflen; size_t tbllen, buflen, slen;
char *s;
int t = lua_type(L, idx); int t = lua_type(L, idx);
if (level < 0) return luaL_argerror(L, 1, "too many nested tables"); if (level < 0) return luaL_argerror(L, 1, "too many nested tables");
if (LUA_TSTRING == t) { if (LUA_TSTRING == t) {
appendf(buf, "\"%s\"", gc(EscapeJsStringLiteral(lua_tostring(L, idx), -1, 0))); s = lua_tolstring(L, idx, &slen);
EscapeLuaString(s, slen, buf);
} else if (LUA_TNUMBER == t) { } else if (LUA_TNUMBER == t) {
// TODO: what if int64_t isn't representable as double // TODO: what if int64_t isn't representable as double
appendf(buf, numformat, lua_tonumber(L, idx)); appendf(buf, numformat, lua_tonumber(L, idx));