mirror of
https://github.com/bjc/prosody.git
synced 2025-04-03 21:27:38 +03:00
dabdfc91be
Probably a duplicate of the comment next to Lmove, recorded by mistake Lpack can probably be removed at some point in the near future once we are confident it is not used anywhere.
73 lines
1.8 KiB
C
73 lines
1.8 KiB
C
#include <lua.h>
|
|
#include <lauxlib.h>
|
|
|
|
#ifndef LUA_MAXINTEGER
|
|
#include <stdint.h>
|
|
#define LUA_MAXINTEGER PTRDIFF_MAX
|
|
#endif
|
|
|
|
static int Lcreate_table(lua_State *L) {
|
|
lua_createtable(L, luaL_checkinteger(L, 1), luaL_checkinteger(L, 2));
|
|
return 1;
|
|
}
|
|
|
|
/* COMPAT: w/ Lua pre-5.2 */
|
|
static int Lpack(lua_State *L) {
|
|
unsigned int n_args = lua_gettop(L);
|
|
lua_createtable(L, n_args, 1);
|
|
lua_insert(L, 1);
|
|
|
|
for(int arg = n_args; arg >= 1; arg--) {
|
|
lua_rawseti(L, 1, arg);
|
|
}
|
|
|
|
lua_pushinteger(L, n_args);
|
|
lua_setfield(L, -2, "n");
|
|
return 1;
|
|
}
|
|
|
|
/* COMPAT: w/ Lua pre-5.4 */
|
|
static int Lmove (lua_State *L) {
|
|
lua_Integer f = luaL_checkinteger(L, 2);
|
|
lua_Integer e = luaL_checkinteger(L, 3);
|
|
lua_Integer t = luaL_checkinteger(L, 4);
|
|
|
|
int tt = !lua_isnoneornil(L, 5) ? 5 : 1; /* destination table */
|
|
luaL_checktype(L, 1, LUA_TTABLE);
|
|
luaL_checktype(L, tt, LUA_TTABLE);
|
|
|
|
if (e >= f) { /* otherwise, nothing to move */
|
|
lua_Integer n, i;
|
|
luaL_argcheck(L, f > 0 || e < LUA_MAXINTEGER + f, 3,
|
|
"too many elements to move");
|
|
n = e - f + 1; /* number of elements to move */
|
|
luaL_argcheck(L, t <= LUA_MAXINTEGER - n + 1, 4,
|
|
"destination wrap around");
|
|
if (t > e || t <= f || (tt != 1 && !lua_compare(L, 1, tt, LUA_OPEQ))) {
|
|
for (i = 0; i < n; i++) {
|
|
lua_rawgeti(L, 1, f + i);
|
|
lua_rawseti(L, tt, t + i);
|
|
}
|
|
} else {
|
|
for (i = n - 1; i >= 0; i--) {
|
|
lua_rawgeti(L, 1, f + i);
|
|
lua_rawseti(L, tt, t + i);
|
|
}
|
|
}
|
|
}
|
|
|
|
lua_pushvalue(L, tt); /* return destination table */
|
|
return 1;
|
|
}
|
|
|
|
int luaopen_util_table(lua_State *L) {
|
|
luaL_checkversion(L);
|
|
lua_createtable(L, 0, 2);
|
|
lua_pushcfunction(L, Lcreate_table);
|
|
lua_setfield(L, -2, "create");
|
|
lua_pushcfunction(L, Lpack);
|
|
lua_setfield(L, -2, "pack");
|
|
lua_pushcfunction(L, Lmove);
|
|
lua_setfield(L, -2, "move");
|
|
return 1;
|
|
}
|