Update Lua to 5.3.3
parent
a684e1dd0f
commit
e2c6350978
@ -0,0 +1,45 @@
|
|||||||
|
/*
|
||||||
|
** $Id: lprefix.h,v 1.2 2014/12/29 16:54:13 roberto Exp $
|
||||||
|
** Definitions for Lua code that must come before any other header file
|
||||||
|
** See Copyright Notice in lua.h
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef lprefix_h
|
||||||
|
#define lprefix_h
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** Allows POSIX/XSI stuff
|
||||||
|
*/
|
||||||
|
#if !defined(LUA_USE_C89) /* { */
|
||||||
|
|
||||||
|
#if !defined(_XOPEN_SOURCE)
|
||||||
|
#define _XOPEN_SOURCE 600
|
||||||
|
#elif _XOPEN_SOURCE == 0
|
||||||
|
#undef _XOPEN_SOURCE /* use -D_XOPEN_SOURCE=0 to undefine it */
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*
|
||||||
|
** Allows manipulation of large files in gcc and some other compilers
|
||||||
|
*/
|
||||||
|
#if !defined(LUA_32BITS) && !defined(_FILE_OFFSET_BITS)
|
||||||
|
#define _LARGEFILE_SOURCE 1
|
||||||
|
#define _FILE_OFFSET_BITS 64
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /* } */
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** Windows stuff
|
||||||
|
*/
|
||||||
|
#if defined(_WIN32) /* { */
|
||||||
|
|
||||||
|
#if !defined(_CRT_SECURE_NO_WARNINGS)
|
||||||
|
#define _CRT_SECURE_NO_WARNINGS /* avoid warnings about ISO C functions */
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /* } */
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,173 +1,215 @@
|
|||||||
/*
|
/*
|
||||||
** $Id: ldump.c,v 2.17.1.1 2013/04/12 18:48:47 roberto Exp $
|
** $Id: ldump.c,v 2.37 2015/10/08 15:53:49 roberto Exp $
|
||||||
** save precompiled Lua chunks
|
** save precompiled Lua chunks
|
||||||
** See Copyright Notice in lua.h
|
** See Copyright Notice in lua.h
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <stddef.h>
|
|
||||||
|
|
||||||
#define ldump_c
|
#define ldump_c
|
||||||
#define LUA_CORE
|
#define LUA_CORE
|
||||||
|
|
||||||
|
#include "lprefix.h"
|
||||||
|
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
|
||||||
#include "lua.h"
|
#include "lua.h"
|
||||||
|
|
||||||
#include "lobject.h"
|
#include "lobject.h"
|
||||||
#include "lstate.h"
|
#include "lstate.h"
|
||||||
#include "lundump.h"
|
#include "lundump.h"
|
||||||
|
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
lua_State* L;
|
lua_State *L;
|
||||||
lua_Writer writer;
|
lua_Writer writer;
|
||||||
void* data;
|
void *data;
|
||||||
int strip;
|
int strip;
|
||||||
int status;
|
int status;
|
||||||
} DumpState;
|
} DumpState;
|
||||||
|
|
||||||
#define DumpMem(b,n,size,D) DumpBlock(b,(n)*(size),D)
|
|
||||||
#define DumpVar(x,D) DumpMem(&x,1,sizeof(x),D)
|
|
||||||
|
|
||||||
static void DumpBlock(const void* b, size_t size, DumpState* D)
|
/*
|
||||||
{
|
** All high-level dumps go through DumpVector; you can change it to
|
||||||
if (D->status==0)
|
** change the endianness of the result
|
||||||
{
|
*/
|
||||||
|
#define DumpVector(v,n,D) DumpBlock(v,(n)*sizeof((v)[0]),D)
|
||||||
|
|
||||||
|
#define DumpLiteral(s,D) DumpBlock(s, sizeof(s) - sizeof(char), D)
|
||||||
|
|
||||||
|
|
||||||
|
static void DumpBlock (const void *b, size_t size, DumpState *D) {
|
||||||
|
if (D->status == 0 && size > 0) {
|
||||||
lua_unlock(D->L);
|
lua_unlock(D->L);
|
||||||
D->status=(*D->writer)(D->L,b,size,D->data);
|
D->status = (*D->writer)(D->L, b, size, D->data);
|
||||||
lua_lock(D->L);
|
lua_lock(D->L);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpChar(int y, DumpState* D)
|
|
||||||
{
|
#define DumpVar(x,D) DumpVector(&x,1,D)
|
||||||
char x=(char)y;
|
|
||||||
DumpVar(x,D);
|
|
||||||
|
static void DumpByte (int y, DumpState *D) {
|
||||||
|
lu_byte x = (lu_byte)y;
|
||||||
|
DumpVar(x, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpInt(int x, DumpState* D)
|
|
||||||
{
|
static void DumpInt (int x, DumpState *D) {
|
||||||
DumpVar(x,D);
|
DumpVar(x, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpNumber(lua_Number x, DumpState* D)
|
|
||||||
{
|
static void DumpNumber (lua_Number x, DumpState *D) {
|
||||||
DumpVar(x,D);
|
DumpVar(x, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpVector(const void* b, int n, size_t size, DumpState* D)
|
|
||||||
{
|
static void DumpInteger (lua_Integer x, DumpState *D) {
|
||||||
DumpInt(n,D);
|
DumpVar(x, D);
|
||||||
DumpMem(b,n,size,D);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpString(const TString* s, DumpState* D)
|
|
||||||
{
|
static void DumpString (const TString *s, DumpState *D) {
|
||||||
if (s==NULL)
|
if (s == NULL)
|
||||||
{
|
DumpByte(0, D);
|
||||||
size_t size=0;
|
else {
|
||||||
DumpVar(size,D);
|
size_t size = tsslen(s) + 1; /* include trailing '\0' */
|
||||||
|
const char *str = getstr(s);
|
||||||
|
if (size < 0xFF)
|
||||||
|
DumpByte(cast_int(size), D);
|
||||||
|
else {
|
||||||
|
DumpByte(0xFF, D);
|
||||||
|
DumpVar(size, D);
|
||||||
}
|
}
|
||||||
else
|
DumpVector(str, size - 1, D); /* no need to save '\0' */
|
||||||
{
|
|
||||||
size_t size=s->tsv.len+1; /* include trailing '\0' */
|
|
||||||
DumpVar(size,D);
|
|
||||||
DumpBlock(getstr(s),size*sizeof(char),D);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#define DumpCode(f,D) DumpVector(f->code,f->sizecode,sizeof(Instruction),D)
|
|
||||||
|
|
||||||
static void DumpFunction(const Proto* f, DumpState* D);
|
static void DumpCode (const Proto *f, DumpState *D) {
|
||||||
|
DumpInt(f->sizecode, D);
|
||||||
|
DumpVector(f->code, f->sizecode, D);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
static void DumpConstants(const Proto* f, DumpState* D)
|
static void DumpFunction(const Proto *f, TString *psource, DumpState *D);
|
||||||
{
|
|
||||||
int i,n=f->sizek;
|
static void DumpConstants (const Proto *f, DumpState *D) {
|
||||||
DumpInt(n,D);
|
int i;
|
||||||
for (i=0; i<n; i++)
|
int n = f->sizek;
|
||||||
{
|
DumpInt(n, D);
|
||||||
const TValue* o=&f->k[i];
|
for (i = 0; i < n; i++) {
|
||||||
DumpChar(ttypenv(o),D);
|
const TValue *o = &f->k[i];
|
||||||
switch (ttypenv(o))
|
DumpByte(ttype(o), D);
|
||||||
{
|
switch (ttype(o)) {
|
||||||
case LUA_TNIL:
|
case LUA_TNIL:
|
||||||
break;
|
break;
|
||||||
case LUA_TBOOLEAN:
|
case LUA_TBOOLEAN:
|
||||||
DumpChar(bvalue(o),D);
|
DumpByte(bvalue(o), D);
|
||||||
|
break;
|
||||||
|
case LUA_TNUMFLT:
|
||||||
|
DumpNumber(fltvalue(o), D);
|
||||||
break;
|
break;
|
||||||
case LUA_TNUMBER:
|
case LUA_TNUMINT:
|
||||||
DumpNumber(nvalue(o),D);
|
DumpInteger(ivalue(o), D);
|
||||||
break;
|
break;
|
||||||
case LUA_TSTRING:
|
case LUA_TSHRSTR:
|
||||||
DumpString(rawtsvalue(o),D);
|
case LUA_TLNGSTR:
|
||||||
|
DumpString(tsvalue(o), D);
|
||||||
break;
|
break;
|
||||||
default: lua_assert(0);
|
default:
|
||||||
|
lua_assert(0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
n=f->sizep;
|
|
||||||
DumpInt(n,D);
|
|
||||||
for (i=0; i<n; i++) DumpFunction(f->p[i],D);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpUpvalues(const Proto* f, DumpState* D)
|
|
||||||
{
|
static void DumpProtos (const Proto *f, DumpState *D) {
|
||||||
int i,n=f->sizeupvalues;
|
int i;
|
||||||
DumpInt(n,D);
|
int n = f->sizep;
|
||||||
for (i=0; i<n; i++)
|
DumpInt(n, D);
|
||||||
{
|
for (i = 0; i < n; i++)
|
||||||
DumpChar(f->upvalues[i].instack,D);
|
DumpFunction(f->p[i], f->source, D);
|
||||||
DumpChar(f->upvalues[i].idx,D);
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void DumpUpvalues (const Proto *f, DumpState *D) {
|
||||||
|
int i, n = f->sizeupvalues;
|
||||||
|
DumpInt(n, D);
|
||||||
|
for (i = 0; i < n; i++) {
|
||||||
|
DumpByte(f->upvalues[i].instack, D);
|
||||||
|
DumpByte(f->upvalues[i].idx, D);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpDebug(const Proto* f, DumpState* D)
|
|
||||||
{
|
static void DumpDebug (const Proto *f, DumpState *D) {
|
||||||
int i,n;
|
int i, n;
|
||||||
DumpString((D->strip) ? NULL : f->source,D);
|
n = (D->strip) ? 0 : f->sizelineinfo;
|
||||||
n= (D->strip) ? 0 : f->sizelineinfo;
|
DumpInt(n, D);
|
||||||
DumpVector(f->lineinfo,n,sizeof(int),D);
|
DumpVector(f->lineinfo, n, D);
|
||||||
n= (D->strip) ? 0 : f->sizelocvars;
|
n = (D->strip) ? 0 : f->sizelocvars;
|
||||||
DumpInt(n,D);
|
DumpInt(n, D);
|
||||||
for (i=0; i<n; i++)
|
for (i = 0; i < n; i++) {
|
||||||
{
|
DumpString(f->locvars[i].varname, D);
|
||||||
DumpString(f->locvars[i].varname,D);
|
DumpInt(f->locvars[i].startpc, D);
|
||||||
DumpInt(f->locvars[i].startpc,D);
|
DumpInt(f->locvars[i].endpc, D);
|
||||||
DumpInt(f->locvars[i].endpc,D);
|
|
||||||
}
|
}
|
||||||
n= (D->strip) ? 0 : f->sizeupvalues;
|
n = (D->strip) ? 0 : f->sizeupvalues;
|
||||||
DumpInt(n,D);
|
DumpInt(n, D);
|
||||||
for (i=0; i<n; i++) DumpString(f->upvalues[i].name,D);
|
for (i = 0; i < n; i++)
|
||||||
|
DumpString(f->upvalues[i].name, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpFunction(const Proto* f, DumpState* D)
|
|
||||||
{
|
static void DumpFunction (const Proto *f, TString *psource, DumpState *D) {
|
||||||
DumpInt(f->linedefined,D);
|
if (D->strip || f->source == psource)
|
||||||
DumpInt(f->lastlinedefined,D);
|
DumpString(NULL, D); /* no debug info or same source as its parent */
|
||||||
DumpChar(f->numparams,D);
|
else
|
||||||
DumpChar(f->is_vararg,D);
|
DumpString(f->source, D);
|
||||||
DumpChar(f->maxstacksize,D);
|
DumpInt(f->linedefined, D);
|
||||||
DumpCode(f,D);
|
DumpInt(f->lastlinedefined, D);
|
||||||
DumpConstants(f,D);
|
DumpByte(f->numparams, D);
|
||||||
DumpUpvalues(f,D);
|
DumpByte(f->is_vararg, D);
|
||||||
DumpDebug(f,D);
|
DumpByte(f->maxstacksize, D);
|
||||||
|
DumpCode(f, D);
|
||||||
|
DumpConstants(f, D);
|
||||||
|
DumpUpvalues(f, D);
|
||||||
|
DumpProtos(f, D);
|
||||||
|
DumpDebug(f, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void DumpHeader(DumpState* D)
|
|
||||||
{
|
static void DumpHeader (DumpState *D) {
|
||||||
lu_byte h[LUAC_HEADERSIZE];
|
DumpLiteral(LUA_SIGNATURE, D);
|
||||||
luaU_header(h);
|
DumpByte(LUAC_VERSION, D);
|
||||||
DumpBlock(h,LUAC_HEADERSIZE,D);
|
DumpByte(LUAC_FORMAT, D);
|
||||||
|
DumpLiteral(LUAC_DATA, D);
|
||||||
|
DumpByte(sizeof(int), D);
|
||||||
|
DumpByte(sizeof(size_t), D);
|
||||||
|
DumpByte(sizeof(Instruction), D);
|
||||||
|
DumpByte(sizeof(lua_Integer), D);
|
||||||
|
DumpByte(sizeof(lua_Number), D);
|
||||||
|
DumpInteger(LUAC_INT, D);
|
||||||
|
DumpNumber(LUAC_NUM, D);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** dump Lua function as precompiled chunk
|
** dump Lua function as precompiled chunk
|
||||||
*/
|
*/
|
||||||
int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip)
|
int luaU_dump(lua_State *L, const Proto *f, lua_Writer w, void *data,
|
||||||
{
|
int strip) {
|
||||||
DumpState D;
|
DumpState D;
|
||||||
D.L=L;
|
D.L = L;
|
||||||
D.writer=w;
|
D.writer = w;
|
||||||
D.data=data;
|
D.data = data;
|
||||||
D.strip=strip;
|
D.strip = strip;
|
||||||
D.status=0;
|
D.status = 0;
|
||||||
DumpHeader(&D);
|
DumpHeader(&D);
|
||||||
DumpFunction(f,&D);
|
DumpByte(f->sizeupvalues, &D);
|
||||||
|
DumpFunction(f, NULL, &D);
|
||||||
return D.status;
|
return D.status;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,256 @@
|
|||||||
|
/*
|
||||||
|
** $Id: lutf8lib.c,v 1.15 2015/03/28 19:16:55 roberto Exp $
|
||||||
|
** Standard library for UTF-8 manipulation
|
||||||
|
** See Copyright Notice in lua.h
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define lutf8lib_c
|
||||||
|
#define LUA_LIB
|
||||||
|
|
||||||
|
#include "lprefix.h"
|
||||||
|
|
||||||
|
|
||||||
|
#include <assert.h>
|
||||||
|
#include <limits.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include "lua.h"
|
||||||
|
|
||||||
|
#include "lauxlib.h"
|
||||||
|
#include "lualib.h"
|
||||||
|
|
||||||
|
#define MAXUNICODE 0x10FFFF
|
||||||
|
|
||||||
|
#define iscont(p) ((*(p) & 0xC0) == 0x80)
|
||||||
|
|
||||||
|
|
||||||
|
/* from strlib */
|
||||||
|
/* translate a relative string position: negative means back from end */
|
||||||
|
static lua_Integer u_posrelat (lua_Integer pos, size_t len) {
|
||||||
|
if (pos >= 0) return pos;
|
||||||
|
else if (0u - (size_t)pos > len) return 0;
|
||||||
|
else return (lua_Integer)len + pos + 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** Decode one UTF-8 sequence, returning NULL if byte sequence is invalid.
|
||||||
|
*/
|
||||||
|
static const char *utf8_decode (const char *o, int *val) {
|
||||||
|
static const unsigned int limits[] = {0xFF, 0x7F, 0x7FF, 0xFFFF};
|
||||||
|
const unsigned char *s = (const unsigned char *)o;
|
||||||
|
unsigned int c = s[0];
|
||||||
|
unsigned int res = 0; /* final result */
|
||||||
|
if (c < 0x80) /* ascii? */
|
||||||
|
res = c;
|
||||||
|
else {
|
||||||
|
int count = 0; /* to count number of continuation bytes */
|
||||||
|
while (c & 0x40) { /* still have continuation bytes? */
|
||||||
|
int cc = s[++count]; /* read next byte */
|
||||||
|
if ((cc & 0xC0) != 0x80) /* not a continuation byte? */
|
||||||
|
return NULL; /* invalid byte sequence */
|
||||||
|
res = (res << 6) | (cc & 0x3F); /* add lower 6 bits from cont. byte */
|
||||||
|
c <<= 1; /* to test next bit */
|
||||||
|
}
|
||||||
|
res |= ((c & 0x7F) << (count * 5)); /* add first byte */
|
||||||
|
if (count > 3 || res > MAXUNICODE || res <= limits[count])
|
||||||
|
return NULL; /* invalid byte sequence */
|
||||||
|
s += count; /* skip continuation bytes read */
|
||||||
|
}
|
||||||
|
if (val) *val = res;
|
||||||
|
return (const char *)s + 1; /* +1 to include first byte */
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** utf8len(s [, i [, j]]) --> number of characters that start in the
|
||||||
|
** range [i,j], or nil + current position if 's' is not well formed in
|
||||||
|
** that interval
|
||||||
|
*/
|
||||||
|
static int utflen (lua_State *L) {
|
||||||
|
int n = 0;
|
||||||
|
size_t len;
|
||||||
|
const char *s = luaL_checklstring(L, 1, &len);
|
||||||
|
lua_Integer posi = u_posrelat(luaL_optinteger(L, 2, 1), len);
|
||||||
|
lua_Integer posj = u_posrelat(luaL_optinteger(L, 3, -1), len);
|
||||||
|
luaL_argcheck(L, 1 <= posi && --posi <= (lua_Integer)len, 2,
|
||||||
|
"initial position out of string");
|
||||||
|
luaL_argcheck(L, --posj < (lua_Integer)len, 3,
|
||||||
|
"final position out of string");
|
||||||
|
while (posi <= posj) {
|
||||||
|
const char *s1 = utf8_decode(s + posi, NULL);
|
||||||
|
if (s1 == NULL) { /* conversion error? */
|
||||||
|
lua_pushnil(L); /* return nil ... */
|
||||||
|
lua_pushinteger(L, posi + 1); /* ... and current position */
|
||||||
|
return 2;
|
||||||
|
}
|
||||||
|
posi = s1 - s;
|
||||||
|
n++;
|
||||||
|
}
|
||||||
|
lua_pushinteger(L, n);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** codepoint(s, [i, [j]]) -> returns codepoints for all characters
|
||||||
|
** that start in the range [i,j]
|
||||||
|
*/
|
||||||
|
static int codepoint (lua_State *L) {
|
||||||
|
size_t len;
|
||||||
|
const char *s = luaL_checklstring(L, 1, &len);
|
||||||
|
lua_Integer posi = u_posrelat(luaL_optinteger(L, 2, 1), len);
|
||||||
|
lua_Integer pose = u_posrelat(luaL_optinteger(L, 3, posi), len);
|
||||||
|
int n;
|
||||||
|
const char *se;
|
||||||
|
luaL_argcheck(L, posi >= 1, 2, "out of range");
|
||||||
|
luaL_argcheck(L, pose <= (lua_Integer)len, 3, "out of range");
|
||||||
|
if (posi > pose) return 0; /* empty interval; return no values */
|
||||||
|
if (pose - posi >= INT_MAX) /* (lua_Integer -> int) overflow? */
|
||||||
|
return luaL_error(L, "string slice too long");
|
||||||
|
n = (int)(pose - posi) + 1;
|
||||||
|
luaL_checkstack(L, n, "string slice too long");
|
||||||
|
n = 0;
|
||||||
|
se = s + pose;
|
||||||
|
for (s += posi - 1; s < se;) {
|
||||||
|
int code;
|
||||||
|
s = utf8_decode(s, &code);
|
||||||
|
if (s == NULL)
|
||||||
|
return luaL_error(L, "invalid UTF-8 code");
|
||||||
|
lua_pushinteger(L, code);
|
||||||
|
n++;
|
||||||
|
}
|
||||||
|
return n;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static void pushutfchar (lua_State *L, int arg) {
|
||||||
|
lua_Integer code = luaL_checkinteger(L, arg);
|
||||||
|
luaL_argcheck(L, 0 <= code && code <= MAXUNICODE, arg, "value out of range");
|
||||||
|
lua_pushfstring(L, "%U", (long)code);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** utfchar(n1, n2, ...) -> char(n1)..char(n2)...
|
||||||
|
*/
|
||||||
|
static int utfchar (lua_State *L) {
|
||||||
|
int n = lua_gettop(L); /* number of arguments */
|
||||||
|
if (n == 1) /* optimize common case of single char */
|
||||||
|
pushutfchar(L, 1);
|
||||||
|
else {
|
||||||
|
int i;
|
||||||
|
luaL_Buffer b;
|
||||||
|
luaL_buffinit(L, &b);
|
||||||
|
for (i = 1; i <= n; i++) {
|
||||||
|
pushutfchar(L, i);
|
||||||
|
luaL_addvalue(&b);
|
||||||
|
}
|
||||||
|
luaL_pushresult(&b);
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** offset(s, n, [i]) -> index where n-th character counting from
|
||||||
|
** position 'i' starts; 0 means character at 'i'.
|
||||||
|
*/
|
||||||
|
static int byteoffset (lua_State *L) {
|
||||||
|
size_t len;
|
||||||
|
const char *s = luaL_checklstring(L, 1, &len);
|
||||||
|
lua_Integer n = luaL_checkinteger(L, 2);
|
||||||
|
lua_Integer posi = (n >= 0) ? 1 : len + 1;
|
||||||
|
posi = u_posrelat(luaL_optinteger(L, 3, posi), len);
|
||||||
|
luaL_argcheck(L, 1 <= posi && --posi <= (lua_Integer)len, 3,
|
||||||
|
"position out of range");
|
||||||
|
if (n == 0) {
|
||||||
|
/* find beginning of current byte sequence */
|
||||||
|
while (posi > 0 && iscont(s + posi)) posi--;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (iscont(s + posi))
|
||||||
|
luaL_error(L, "initial position is a continuation byte");
|
||||||
|
if (n < 0) {
|
||||||
|
while (n < 0 && posi > 0) { /* move back */
|
||||||
|
do { /* find beginning of previous character */
|
||||||
|
posi--;
|
||||||
|
} while (posi > 0 && iscont(s + posi));
|
||||||
|
n++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
n--; /* do not move for 1st character */
|
||||||
|
while (n > 0 && posi < (lua_Integer)len) {
|
||||||
|
do { /* find beginning of next character */
|
||||||
|
posi++;
|
||||||
|
} while (iscont(s + posi)); /* (cannot pass final '\0') */
|
||||||
|
n--;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (n == 0) /* did it find given character? */
|
||||||
|
lua_pushinteger(L, posi + 1);
|
||||||
|
else /* no such character */
|
||||||
|
lua_pushnil(L);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int iter_aux (lua_State *L) {
|
||||||
|
size_t len;
|
||||||
|
const char *s = luaL_checklstring(L, 1, &len);
|
||||||
|
lua_Integer n = lua_tointeger(L, 2) - 1;
|
||||||
|
if (n < 0) /* first iteration? */
|
||||||
|
n = 0; /* start from here */
|
||||||
|
else if (n < (lua_Integer)len) {
|
||||||
|
n++; /* skip current byte */
|
||||||
|
while (iscont(s + n)) n++; /* and its continuations */
|
||||||
|
}
|
||||||
|
if (n >= (lua_Integer)len)
|
||||||
|
return 0; /* no more codepoints */
|
||||||
|
else {
|
||||||
|
int code;
|
||||||
|
const char *next = utf8_decode(s + n, &code);
|
||||||
|
if (next == NULL || iscont(next))
|
||||||
|
return luaL_error(L, "invalid UTF-8 code");
|
||||||
|
lua_pushinteger(L, n + 1);
|
||||||
|
lua_pushinteger(L, code);
|
||||||
|
return 2;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int iter_codes (lua_State *L) {
|
||||||
|
luaL_checkstring(L, 1);
|
||||||
|
lua_pushcfunction(L, iter_aux);
|
||||||
|
lua_pushvalue(L, 1);
|
||||||
|
lua_pushinteger(L, 0);
|
||||||
|
return 3;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* pattern to match a single UTF-8 character */
|
||||||
|
#define UTF8PATT "[\0-\x7F\xC2-\xF4][\x80-\xBF]*"
|
||||||
|
|
||||||
|
|
||||||
|
static const luaL_Reg funcs[] = {
|
||||||
|
{"offset", byteoffset},
|
||||||
|
{"codepoint", codepoint},
|
||||||
|
{"char", utfchar},
|
||||||
|
{"len", utflen},
|
||||||
|
{"codes", iter_codes},
|
||||||
|
/* placeholders */
|
||||||
|
{"charpattern", NULL},
|
||||||
|
{NULL, NULL}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
LUAMOD_API int luaopen_utf8 (lua_State *L) {
|
||||||
|
luaL_newlib(L, funcs);
|
||||||
|
lua_pushlstring(L, UTF8PATT, sizeof(UTF8PATT)/sizeof(char) - 1);
|
||||||
|
lua_setfield(L, -2, "charpattern");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue