forked from redo/BlockLua
Compare commits
41 Commits
74745108b6
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
| b328f0b21a | |||
| 71b73c816b | |||
| c5dc8b15f9 | |||
|
|
8399b11322 | ||
| d494f02fe3 | |||
| a7db0d8e81 | |||
| 33f5ec9bbe | |||
| f6bf18efaa | |||
| 4f42801da6 | |||
| 15f67e0eef | |||
| 5885dcbed3 | |||
| d9a416f5d5 | |||
|
|
2191e004ad | ||
|
|
ae34bb8b7a | ||
|
|
b71bfdb73e | ||
|
|
7232ede09d | ||
| ed5c254480 | |||
| 5718ba8e6b | |||
| a4f78b7425 | |||
| e47f6d4651 | |||
| 93a47d54be | |||
|
|
76c758a47b | ||
| 87e199ea5c | |||
| 66ed695010 | |||
| e309e2427b | |||
|
|
5f98dc017b | ||
|
|
edf31c178c | ||
|
|
7da249cca1 | ||
| ee784869f1 | |||
| 0815a6d229 | |||
| cbd0c29495 | |||
| 01f216f31e | |||
| f447c039c7 | |||
| 30fa81f513 | |||
| 1144504ac3 | |||
| ce3f7c8b3f | |||
| 4d74df209e | |||
|
|
1a4c7bfefc | ||
|
|
eaafb42317 | ||
|
|
9c349a9352 | ||
|
|
34345a7eeb |
42
.clang-format
Normal file
42
.clang-format
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
BasedOnStyle: LLVM
|
||||||
|
|
||||||
|
# Include sorting and grouping
|
||||||
|
SortIncludes: CaseSensitive
|
||||||
|
IncludeBlocks: Preserve
|
||||||
|
IncludeCategories:
|
||||||
|
- Regex: '<Windows\.h>'
|
||||||
|
Priority: 1
|
||||||
|
- Regex: '<Psapi\.h>'
|
||||||
|
Priority: 2
|
||||||
|
- Regex: '"BlHooks\\.hpp"'
|
||||||
|
Priority: 3
|
||||||
|
- Regex: '"BlFuncs\\.hpp"'
|
||||||
|
Priority: 4
|
||||||
|
- Regex: ".*"
|
||||||
|
Priority: 5
|
||||||
|
|
||||||
|
# Indentation settings
|
||||||
|
IndentWidth: 2
|
||||||
|
UseTab: Never
|
||||||
|
IndentAccessModifiers: false
|
||||||
|
AccessModifierOffset: 0
|
||||||
|
NamespaceIndentation: All
|
||||||
|
|
||||||
|
# Line breaking and wrapping
|
||||||
|
ColumnLimit: 120
|
||||||
|
BreakBeforeTernaryOperators: false
|
||||||
|
BreakBeforeBinaryOperators: None
|
||||||
|
AlwaysBreakAfterReturnType: None
|
||||||
|
AlwaysBreakAfterDefinitionReturnType: None
|
||||||
|
|
||||||
|
# Alignment
|
||||||
|
AlignOperands: DontAlign
|
||||||
|
AlignAfterOpenBracket: AlwaysBreak
|
||||||
|
PointerAlignment: Left
|
||||||
|
|
||||||
|
# Parameter formatting
|
||||||
|
BinPackParameters: false
|
||||||
|
AllowAllParametersOfDeclarationOnNextLine: false
|
||||||
|
|
||||||
|
# Code organization
|
||||||
|
SeparateDefinitionBlocks: Always
|
||||||
9
.editorconfig
Normal file
9
.editorconfig
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
root = true
|
||||||
|
|
||||||
|
[*]
|
||||||
|
indent_style = space
|
||||||
|
indent_size = 2
|
||||||
|
end_of_line = lf
|
||||||
|
charset = utf-8
|
||||||
|
trim_trailing_whitespace = true
|
||||||
|
insert_final_newline = true
|
||||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -1,2 +1,2 @@
|
|||||||
.*
|
build/
|
||||||
!.gitignore
|
.cache/
|
||||||
|
|||||||
11
.vscode/settings.json
vendored
Normal file
11
.vscode/settings.json
vendored
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
{
|
||||||
|
"Lua.diagnostics.globals": [
|
||||||
|
"_bllua_ts",
|
||||||
|
"_bllua_requiresecure",
|
||||||
|
"_bllua_on_unload"
|
||||||
|
],
|
||||||
|
"Lua.runtime.version": "Lua 5.1",
|
||||||
|
"Lua.diagnostics.disable": ["lowercase-global", "undefined-global"],
|
||||||
|
"C_Cpp.default.compileCommands": "${workspaceFolder}/build/compile_commands.json",
|
||||||
|
"C_Cpp.default.compilerPath": "C:/msys64/mingw32/bin/g++.exe"
|
||||||
|
}
|
||||||
Binary file not shown.
BIN
BlockLua.dll
BIN
BlockLua.dll
Binary file not shown.
56
CMakeLists.txt
Normal file
56
CMakeLists.txt
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
cmake_minimum_required(VERSION 3.10)
|
||||||
|
|
||||||
|
project(BlockLua CXX)
|
||||||
|
|
||||||
|
# Export compile_commands.json for VSCode IntelliSense
|
||||||
|
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
|
||||||
|
|
||||||
|
# Output directories to mirror build.bat's build folder
|
||||||
|
set(OUTPUT_DIR ${CMAKE_SOURCE_DIR}/build)
|
||||||
|
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${OUTPUT_DIR})
|
||||||
|
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${OUTPUT_DIR})
|
||||||
|
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${OUTPUT_DIR})
|
||||||
|
|
||||||
|
# Global build options to mirror build.bat
|
||||||
|
add_compile_options(
|
||||||
|
-Wall
|
||||||
|
-Werror
|
||||||
|
-m32
|
||||||
|
-static-libgcc
|
||||||
|
-static-libstdc++
|
||||||
|
)
|
||||||
|
|
||||||
|
# Include paths
|
||||||
|
include_directories(
|
||||||
|
${CMAKE_SOURCE_DIR}/src
|
||||||
|
${CMAKE_SOURCE_DIR}/inc/tsfuncs
|
||||||
|
${CMAKE_SOURCE_DIR}/inc/lua
|
||||||
|
)
|
||||||
|
|
||||||
|
# Link directories (for -L.) and libraries from build.bat
|
||||||
|
link_directories(
|
||||||
|
${CMAKE_SOURCE_DIR}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Safe DLL
|
||||||
|
add_library(BlockLua SHARED src/bllua4.cpp)
|
||||||
|
# Ensure output name matches build.bat
|
||||||
|
set_target_properties(BlockLua PROPERTIES OUTPUT_NAME "BlockLua")
|
||||||
|
# Linker flags and libraries
|
||||||
|
if(MSVC)
|
||||||
|
# Not expected with mingw, but keep placeholder
|
||||||
|
else()
|
||||||
|
target_link_libraries(BlockLua PRIVATE psapi lua5.1)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
# Unsafe DLL (with BLLUA_UNSAFE definition)
|
||||||
|
add_library(BlockLuaUnsafe SHARED src/bllua4.cpp)
|
||||||
|
set_target_properties(BlockLuaUnsafe PROPERTIES OUTPUT_NAME "BlockLua-Unsafe")
|
||||||
|
|
||||||
|
target_compile_definitions(BlockLuaUnsafe PRIVATE BLLUA_UNSAFE)
|
||||||
|
|
||||||
|
if(MSVC)
|
||||||
|
# Not expected with mingw, but keep placeholder
|
||||||
|
else()
|
||||||
|
target_link_libraries(BlockLuaUnsafe PRIVATE psapi lua5.1)
|
||||||
|
endif()
|
||||||
27
build.bat
Normal file
27
build.bat
Normal file
@@ -0,0 +1,27 @@
|
|||||||
|
@echo off
|
||||||
|
cd /d %~dp0
|
||||||
|
|
||||||
|
REM Ensure MinGW32 toolchain is first in PATH
|
||||||
|
set "PATH=C:\msys64\mingw32\bin;%PATH%"
|
||||||
|
|
||||||
|
REM Configure CMake (generate into build/)
|
||||||
|
cmake -S . -B build -G "MinGW Makefiles"
|
||||||
|
if errorlevel 1 goto :error
|
||||||
|
|
||||||
|
REM Build (Release by default)
|
||||||
|
cmake --build build --config Release -j
|
||||||
|
if errorlevel 1 goto :error
|
||||||
|
|
||||||
|
echo.
|
||||||
|
echo Build completed.
|
||||||
|
echo Outputs in .\build :
|
||||||
|
echo - BlockLua.dll
|
||||||
|
|
||||||
|
echo - BlockLua-Unsafe.dll
|
||||||
|
|
||||||
|
exit /b 0
|
||||||
|
|
||||||
|
:error
|
||||||
|
echo.
|
||||||
|
echo Build failed. See errors above.
|
||||||
|
exit /b 1
|
||||||
13
compile.bat
13
compile.bat
@@ -1,13 +0,0 @@
|
|||||||
@echo off
|
|
||||||
cd /d %~dp0
|
|
||||||
|
|
||||||
set buildargs=-Wall -Werror -m32 -shared -Isrc -Iinc/tsfuncs -Iinc/lua -lpsapi -L. -llua5.1 -static-libgcc -static-libstdc++
|
|
||||||
|
|
||||||
echo on
|
|
||||||
g++ src/bllua4.cpp %buildargs% -o BlockLua.dll && g++ -DBLLUA_UNSAFE src/bllua4.cpp %buildargs% -o BlockLua-Unsafe.dll
|
|
||||||
@echo off
|
|
||||||
|
|
||||||
rem objdump -d BlockLua.dll > BlockLua.dll.dump.txt
|
|
||||||
rem objdump -d BlockLua-Unsafe.dll > BlockLua-Unsafe.dll.dump.txt
|
|
||||||
|
|
||||||
pause
|
|
||||||
52
compiling.md
Normal file
52
compiling.md
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
## Compiling on Windows with MSYS2 (32-bit)
|
||||||
|
|
||||||
|
Follow these steps to build `BlockLua.dll` for 32-bit Windows using MSYS2's MinGW-w64 i686 toolchain.
|
||||||
|
|
||||||
|
### 1) Install MSYS2
|
||||||
|
|
||||||
|
- Download and install MSYS2 from `https://www.msys2.org/`.
|
||||||
|
- After installing, open the "MSYS2 MSYS" terminal once and update the package database if prompted.
|
||||||
|
|
||||||
|
### 2) Install required packages (i686 / 32-bit)
|
||||||
|
|
||||||
|
Run this in the "MSYS2 MSYS" or "MSYS2 MinGW 32-bit" terminal:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pacman -Sy --needed mingw-w64-i686-toolchain mingw-w64-i686-binutils mingw-w64-i686-lua51
|
||||||
|
```
|
||||||
|
|
||||||
|
What these packages are for:
|
||||||
|
|
||||||
|
- mingw-w64-i686-toolchain: 32-bit C/C++ compiler suite (g++, libstdc++, runtime libs) to build Windows binaries.
|
||||||
|
- mingw-w64-i686-binutils: Linker and binary utilities (ld, as, objdump) used by the compiler and for optional inspection.
|
||||||
|
- mingw-w64-i686-lua51: Lua 5.1 development files for 32-bit (import library). We use its import library to link; at runtime you can use the provided `lua5.1.dll`.
|
||||||
|
|
||||||
|
### 3) Build (PowerShell, recommended)
|
||||||
|
|
||||||
|
- Open PowerShell in the repo root.
|
||||||
|
- Run the script:
|
||||||
|
|
||||||
|
```powershell
|
||||||
|
build.bat
|
||||||
|
```
|
||||||
|
|
||||||
|
What the script does:
|
||||||
|
|
||||||
|
- Temporarily prepends `C:\\msys64\\mingw32\\bin` to PATH so the 32-bit toolchain is used.
|
||||||
|
- Compiles the project with `-m32` and links against `lua5.1`.
|
||||||
|
- Produces `build\BlockLua.dll` and `build\BlockLua-Unsafe.dll`.
|
||||||
|
|
||||||
|
### 4) Optional: Verify 32-bit output
|
||||||
|
|
||||||
|
If you installed binutils, you can check the architecture:
|
||||||
|
|
||||||
|
```powershell
|
||||||
|
objdump -f build\BlockLua.dll | Select-String i386
|
||||||
|
```
|
||||||
|
|
||||||
|
You should see `architecture: i386` in the output.
|
||||||
|
|
||||||
|
### Notes
|
||||||
|
|
||||||
|
- Ensure you installed the i686 (32-bit) variants of the packages; x86_64 packages won't work for a 32-bit build.
|
||||||
|
- If the linker cannot find `-llua5.1`, confirm `mingw-w64-i686-lua51` is installed and you are using the `mingw32` toolchain (not `x86_64`).
|
||||||
@@ -4,17 +4,14 @@
|
|||||||
** See Copyright Notice in lua.h
|
** See Copyright Notice in lua.h
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
#ifndef lauxlib_h
|
#ifndef lauxlib_h
|
||||||
#define lauxlib_h
|
#define lauxlib_h
|
||||||
|
|
||||||
|
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
#include "lua.h"
|
#include "lua.h"
|
||||||
|
|
||||||
|
|
||||||
/* extra error code for `luaL_load' */
|
/* extra error code for `luaL_load' */
|
||||||
#define LUA_ERRFILE (LUA_ERRERR + 1)
|
#define LUA_ERRFILE (LUA_ERRERR + 1)
|
||||||
|
|
||||||
@@ -23,24 +20,19 @@ typedef struct luaL_Reg {
|
|||||||
lua_CFunction func;
|
lua_CFunction func;
|
||||||
} luaL_Reg;
|
} luaL_Reg;
|
||||||
|
|
||||||
LUALIB_API void (luaL_openlib) (lua_State *L, const char *libname,
|
LUALIB_API void(luaL_openlib)(lua_State* L, const char* libname, const luaL_Reg* l, int nup);
|
||||||
const luaL_Reg *l, int nup);
|
LUALIB_API void(luaL_register)(lua_State* L, const char* libname, const luaL_Reg* l);
|
||||||
LUALIB_API void (luaL_register) (lua_State *L, const char *libname,
|
|
||||||
const luaL_Reg *l);
|
|
||||||
LUALIB_API int(luaL_getmetafield)(lua_State* L, int obj, const char* e);
|
LUALIB_API int(luaL_getmetafield)(lua_State* L, int obj, const char* e);
|
||||||
LUALIB_API int(luaL_callmeta)(lua_State* L, int obj, const char* e);
|
LUALIB_API int(luaL_callmeta)(lua_State* L, int obj, const char* e);
|
||||||
LUALIB_API int(luaL_typerror)(lua_State* L, int narg, const char* tname);
|
LUALIB_API int(luaL_typerror)(lua_State* L, int narg, const char* tname);
|
||||||
LUALIB_API int(luaL_argerror)(lua_State* L, int numarg, const char* extramsg);
|
LUALIB_API int(luaL_argerror)(lua_State* L, int numarg, const char* extramsg);
|
||||||
LUALIB_API const char *(luaL_checklstring) (lua_State *L, int numArg,
|
LUALIB_API const char*(luaL_checklstring)(lua_State * L, int numArg, size_t* l);
|
||||||
size_t *l);
|
LUALIB_API const char*(luaL_optlstring)(lua_State * L, int numArg, const char* def, size_t* l);
|
||||||
LUALIB_API const char *(luaL_optlstring) (lua_State *L, int numArg,
|
|
||||||
const char *def, size_t *l);
|
|
||||||
LUALIB_API lua_Number(luaL_checknumber)(lua_State* L, int numArg);
|
LUALIB_API lua_Number(luaL_checknumber)(lua_State* L, int numArg);
|
||||||
LUALIB_API lua_Number(luaL_optnumber)(lua_State* L, int nArg, lua_Number def);
|
LUALIB_API lua_Number(luaL_optnumber)(lua_State* L, int nArg, lua_Number def);
|
||||||
|
|
||||||
LUALIB_API lua_Integer(luaL_checkinteger)(lua_State* L, int numArg);
|
LUALIB_API lua_Integer(luaL_checkinteger)(lua_State* L, int numArg);
|
||||||
LUALIB_API lua_Integer (luaL_optinteger) (lua_State *L, int nArg,
|
LUALIB_API lua_Integer(luaL_optinteger)(lua_State* L, int nArg, lua_Integer def);
|
||||||
lua_Integer def);
|
|
||||||
|
|
||||||
LUALIB_API void(luaL_checkstack)(lua_State* L, int sz, const char* msg);
|
LUALIB_API void(luaL_checkstack)(lua_State* L, int sz, const char* msg);
|
||||||
LUALIB_API void(luaL_checktype)(lua_State* L, int narg, int t);
|
LUALIB_API void(luaL_checktype)(lua_State* L, int narg, int t);
|
||||||
@@ -52,8 +44,7 @@ LUALIB_API void *(luaL_checkudata) (lua_State *L, int ud, const char *tname);
|
|||||||
LUALIB_API void(luaL_where)(lua_State* L, int lvl);
|
LUALIB_API void(luaL_where)(lua_State* L, int lvl);
|
||||||
LUALIB_API int(luaL_error)(lua_State* L, const char* fmt, ...);
|
LUALIB_API int(luaL_error)(lua_State* L, const char* fmt, ...);
|
||||||
|
|
||||||
LUALIB_API int (luaL_checkoption) (lua_State *L, int narg, const char *def,
|
LUALIB_API int(luaL_checkoption)(lua_State* L, int narg, const char* def, const char* const lst[]);
|
||||||
const char *const lst[]);
|
|
||||||
|
|
||||||
/* pre-defined references */
|
/* pre-defined references */
|
||||||
#define LUA_NOREF (-2)
|
#define LUA_NOREF (-2)
|
||||||
@@ -63,43 +54,33 @@ LUALIB_API int (luaL_ref) (lua_State *L, int t);
|
|||||||
LUALIB_API void(luaL_unref)(lua_State* L, int t, int ref);
|
LUALIB_API void(luaL_unref)(lua_State* L, int t, int ref);
|
||||||
|
|
||||||
LUALIB_API int(luaL_loadfile)(lua_State* L, const char* filename);
|
LUALIB_API int(luaL_loadfile)(lua_State* L, const char* filename);
|
||||||
LUALIB_API int (luaL_loadbuffer) (lua_State *L, const char *buff, size_t sz,
|
LUALIB_API int(luaL_loadbuffer)(lua_State* L, const char* buff, size_t sz, const char* name);
|
||||||
const char *name);
|
|
||||||
LUALIB_API int(luaL_loadstring)(lua_State* L, const char* s);
|
LUALIB_API int(luaL_loadstring)(lua_State* L, const char* s);
|
||||||
|
|
||||||
LUALIB_API lua_State*(luaL_newstate)(void);
|
LUALIB_API lua_State*(luaL_newstate)(void);
|
||||||
|
|
||||||
|
LUALIB_API const char*(luaL_gsub)(lua_State * L, const char* s, const char* p, const char* r);
|
||||||
|
|
||||||
LUALIB_API const char *(luaL_gsub) (lua_State *L, const char *s, const char *p,
|
LUALIB_API const char*(luaL_findtable)(lua_State * L, int idx, const char* fname, int szhint);
|
||||||
const char *r);
|
|
||||||
|
|
||||||
LUALIB_API const char *(luaL_findtable) (lua_State *L, int idx,
|
|
||||||
const char *fname, int szhint);
|
|
||||||
|
|
||||||
/* From Lua 5.2. */
|
/* From Lua 5.2. */
|
||||||
LUALIB_API int luaL_fileresult(lua_State* L, int stat, const char* fname);
|
LUALIB_API int luaL_fileresult(lua_State* L, int stat, const char* fname);
|
||||||
LUALIB_API int luaL_execresult(lua_State* L, int stat);
|
LUALIB_API int luaL_execresult(lua_State* L, int stat);
|
||||||
LUALIB_API int (luaL_loadfilex) (lua_State *L, const char *filename,
|
LUALIB_API int(luaL_loadfilex)(lua_State* L, const char* filename, const char* mode);
|
||||||
const char *mode);
|
LUALIB_API int(luaL_loadbufferx)(lua_State* L, const char* buff, size_t sz, const char* name, const char* mode);
|
||||||
LUALIB_API int (luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz,
|
LUALIB_API void luaL_traceback(lua_State* L, lua_State* L1, const char* msg, int level);
|
||||||
const char *name, const char *mode);
|
|
||||||
LUALIB_API void luaL_traceback (lua_State *L, lua_State *L1, const char *msg,
|
|
||||||
int level);
|
|
||||||
LUALIB_API void(luaL_setfuncs)(lua_State* L, const luaL_Reg* l, int nup);
|
LUALIB_API void(luaL_setfuncs)(lua_State* L, const luaL_Reg* l, int nup);
|
||||||
LUALIB_API void (luaL_pushmodule) (lua_State *L, const char *modname,
|
LUALIB_API void(luaL_pushmodule)(lua_State* L, const char* modname, int sizehint);
|
||||||
int sizehint);
|
|
||||||
LUALIB_API void*(luaL_testudata)(lua_State * L, int ud, const char* tname);
|
LUALIB_API void*(luaL_testudata)(lua_State * L, int ud, const char* tname);
|
||||||
LUALIB_API void(luaL_setmetatable)(lua_State* L, const char* tname);
|
LUALIB_API void(luaL_setmetatable)(lua_State* L, const char* tname);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** ===============================================================
|
** ===============================================================
|
||||||
** some useful macros
|
** some useful macros
|
||||||
** ===============================================================
|
** ===============================================================
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#define luaL_argcheck(L, cond,numarg,extramsg) \
|
#define luaL_argcheck(L, cond, numarg, extramsg) ((void)((cond) || luaL_argerror(L, (numarg), (extramsg))))
|
||||||
((void)((cond) || luaL_argerror(L, (numarg), (extramsg))))
|
|
||||||
#define luaL_checkstring(L, n) (luaL_checklstring(L, (n), NULL))
|
#define luaL_checkstring(L, n) (luaL_checklstring(L, (n), NULL))
|
||||||
#define luaL_optstring(L, n, d) (luaL_optlstring(L, (n), (d), NULL))
|
#define luaL_optstring(L, n, d) (luaL_optlstring(L, (n), (d), NULL))
|
||||||
#define luaL_checkint(L, n) ((int)luaL_checkinteger(L, (n)))
|
#define luaL_checkint(L, n) ((int)luaL_checkinteger(L, (n)))
|
||||||
@@ -109,19 +90,16 @@ LUALIB_API void (luaL_setmetatable) (lua_State *L, const char *tname);
|
|||||||
|
|
||||||
#define luaL_typename(L, i) lua_typename(L, lua_type(L, (i)))
|
#define luaL_typename(L, i) lua_typename(L, lua_type(L, (i)))
|
||||||
|
|
||||||
#define luaL_dofile(L, fn) \
|
#define luaL_dofile(L, fn) (luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
||||||
(luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
|
||||||
|
|
||||||
#define luaL_dostring(L, s) \
|
#define luaL_dostring(L, s) (luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
||||||
(luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
|
||||||
|
|
||||||
#define luaL_getmetatable(L, n) (lua_getfield(L, LUA_REGISTRYINDEX, (n)))
|
#define luaL_getmetatable(L, n) (lua_getfield(L, LUA_REGISTRYINDEX, (n)))
|
||||||
|
|
||||||
#define luaL_opt(L, f, n, d) (lua_isnoneornil(L, (n)) ? (d) : f(L, (n)))
|
#define luaL_opt(L, f, n, d) (lua_isnoneornil(L, (n)) ? (d) : f(L, (n)))
|
||||||
|
|
||||||
/* From Lua 5.2. */
|
/* From Lua 5.2. */
|
||||||
#define luaL_newlibtable(L, l) \
|
#define luaL_newlibtable(L, l) lua_createtable(L, 0, sizeof(l) / sizeof((l)[0]) - 1)
|
||||||
lua_createtable(L, 0, sizeof(l)/sizeof((l)[0]) - 1)
|
|
||||||
#define luaL_newlib(L, l) (luaL_newlibtable(L, l), luaL_setfuncs(L, l, 0))
|
#define luaL_newlib(L, l) (luaL_newlibtable(L, l), luaL_setfuncs(L, l, 0))
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -130,8 +108,6 @@ LUALIB_API void (luaL_setmetatable) (lua_State *L, const char *tname);
|
|||||||
** =======================================================
|
** =======================================================
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
typedef struct luaL_Buffer {
|
typedef struct luaL_Buffer {
|
||||||
char* p; /* current position in buffer */
|
char* p; /* current position in buffer */
|
||||||
int lvl; /* number of strings in the stack (level) */
|
int lvl; /* number of strings in the stack (level) */
|
||||||
@@ -140,8 +116,7 @@ typedef struct luaL_Buffer {
|
|||||||
} luaL_Buffer;
|
} luaL_Buffer;
|
||||||
|
|
||||||
#define luaL_addchar(B, c) \
|
#define luaL_addchar(B, c) \
|
||||||
((void)((B)->p < ((B)->buffer+LUAL_BUFFERSIZE) || luaL_prepbuffer(B)), \
|
((void)((B)->p < ((B)->buffer + LUAL_BUFFERSIZE) || luaL_prepbuffer(B)), (*(B)->p++ = (char)(c)))
|
||||||
(*(B)->p++ = (char)(c)))
|
|
||||||
|
|
||||||
/* compatibility only */
|
/* compatibility only */
|
||||||
#define luaL_putchar(B, c) luaL_addchar(B, c)
|
#define luaL_putchar(B, c) luaL_addchar(B, c)
|
||||||
@@ -155,7 +130,6 @@ LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
|
|||||||
LUALIB_API void(luaL_addvalue)(luaL_Buffer* B);
|
LUALIB_API void(luaL_addvalue)(luaL_Buffer* B);
|
||||||
LUALIB_API void(luaL_pushresult)(luaL_Buffer* B);
|
LUALIB_API void(luaL_pushresult)(luaL_Buffer* B);
|
||||||
|
|
||||||
|
|
||||||
/* }====================================================== */
|
/* }====================================================== */
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -5,31 +5,26 @@
|
|||||||
** See Copyright Notice at the end of this file
|
** See Copyright Notice at the end of this file
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
#ifndef lua_h
|
#ifndef lua_h
|
||||||
#define lua_h
|
#define lua_h
|
||||||
|
|
||||||
#include <stdarg.h>
|
#include <stdarg.h>
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
|
|
||||||
|
|
||||||
#include "luaconf.h"
|
#include "luaconf.h"
|
||||||
|
|
||||||
|
|
||||||
#define LUA_VERSION "Lua 5.1"
|
#define LUA_VERSION "Lua 5.1"
|
||||||
#define LUA_RELEASE "Lua 5.1.4"
|
#define LUA_RELEASE "Lua 5.1.4"
|
||||||
#define LUA_VERSION_NUM 501
|
#define LUA_VERSION_NUM 501
|
||||||
#define LUA_COPYRIGHT "Copyright (C) 1994-2008 Lua.org, PUC-Rio"
|
#define LUA_COPYRIGHT "Copyright (C) 1994-2008 Lua.org, PUC-Rio"
|
||||||
#define LUA_AUTHORS "R. Ierusalimschy, L. H. de Figueiredo & W. Celes"
|
#define LUA_AUTHORS "R. Ierusalimschy, L. H. de Figueiredo & W. Celes"
|
||||||
|
|
||||||
|
|
||||||
/* mark for precompiled code (`<esc>Lua') */
|
/* mark for precompiled code (`<esc>Lua') */
|
||||||
#define LUA_SIGNATURE "\033Lua"
|
#define LUA_SIGNATURE "\033Lua"
|
||||||
|
|
||||||
/* option for multiple returns in `lua_pcall' and `lua_call' */
|
/* option for multiple returns in `lua_pcall' and `lua_call' */
|
||||||
#define LUA_MULTRET (-1)
|
#define LUA_MULTRET (-1)
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** pseudo-indices
|
** pseudo-indices
|
||||||
*/
|
*/
|
||||||
@@ -38,7 +33,6 @@
|
|||||||
#define LUA_GLOBALSINDEX (-10002)
|
#define LUA_GLOBALSINDEX (-10002)
|
||||||
#define lua_upvalueindex(i) (LUA_GLOBALSINDEX - (i))
|
#define lua_upvalueindex(i) (LUA_GLOBALSINDEX - (i))
|
||||||
|
|
||||||
|
|
||||||
/* thread status */
|
/* thread status */
|
||||||
#define LUA_OK 0
|
#define LUA_OK 0
|
||||||
#define LUA_YIELD 1
|
#define LUA_YIELD 1
|
||||||
@@ -47,12 +41,10 @@
|
|||||||
#define LUA_ERRMEM 4
|
#define LUA_ERRMEM 4
|
||||||
#define LUA_ERRERR 5
|
#define LUA_ERRERR 5
|
||||||
|
|
||||||
|
|
||||||
typedef struct lua_State lua_State;
|
typedef struct lua_State lua_State;
|
||||||
|
|
||||||
typedef int (*lua_CFunction)(lua_State* L);
|
typedef int (*lua_CFunction)(lua_State* L);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** functions that read/write blocks when loading/dumping Lua chunks
|
** functions that read/write blocks when loading/dumping Lua chunks
|
||||||
*/
|
*/
|
||||||
@@ -60,13 +52,11 @@ typedef const char * (*lua_Reader) (lua_State *L, void *ud, size_t *sz);
|
|||||||
|
|
||||||
typedef int (*lua_Writer)(lua_State* L, const void* p, size_t sz, void* ud);
|
typedef int (*lua_Writer)(lua_State* L, const void* p, size_t sz, void* ud);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** prototype for memory-allocation functions
|
** prototype for memory-allocation functions
|
||||||
*/
|
*/
|
||||||
typedef void* (*lua_Alloc)(void* ud, void* ptr, size_t osize, size_t nsize);
|
typedef void* (*lua_Alloc)(void* ud, void* ptr, size_t osize, size_t nsize);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** basic types
|
** basic types
|
||||||
*/
|
*/
|
||||||
@@ -82,12 +72,9 @@ typedef void * (*lua_Alloc) (void *ud, void *ptr, size_t osize, size_t nsize);
|
|||||||
#define LUA_TUSERDATA 7
|
#define LUA_TUSERDATA 7
|
||||||
#define LUA_TTHREAD 8
|
#define LUA_TTHREAD 8
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/* minimum Lua stack available to a C function */
|
/* minimum Lua stack available to a C function */
|
||||||
#define LUA_MINSTACK 20
|
#define LUA_MINSTACK 20
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** generic extra include file
|
** generic extra include file
|
||||||
*/
|
*/
|
||||||
@@ -95,16 +82,12 @@ typedef void * (*lua_Alloc) (void *ud, void *ptr, size_t osize, size_t nsize);
|
|||||||
#include LUA_USER_H
|
#include LUA_USER_H
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
/* type of numbers in Lua */
|
/* type of numbers in Lua */
|
||||||
typedef LUA_NUMBER lua_Number;
|
typedef LUA_NUMBER lua_Number;
|
||||||
|
|
||||||
|
|
||||||
/* type for integer functions */
|
/* type for integer functions */
|
||||||
typedef LUA_INTEGER lua_Integer;
|
typedef LUA_INTEGER lua_Integer;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** state manipulation
|
** state manipulation
|
||||||
*/
|
*/
|
||||||
@@ -114,7 +97,6 @@ LUA_API lua_State *(lua_newthread) (lua_State *L);
|
|||||||
|
|
||||||
LUA_API lua_CFunction(lua_atpanic)(lua_State* L, lua_CFunction panicf);
|
LUA_API lua_CFunction(lua_atpanic)(lua_State* L, lua_CFunction panicf);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** basic stack manipulation
|
** basic stack manipulation
|
||||||
*/
|
*/
|
||||||
@@ -128,7 +110,6 @@ LUA_API int (lua_checkstack) (lua_State *L, int sz);
|
|||||||
|
|
||||||
LUA_API void(lua_xmove)(lua_State* from, lua_State* to, int n);
|
LUA_API void(lua_xmove)(lua_State* from, lua_State* to, int n);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** access functions (stack -> C)
|
** access functions (stack -> C)
|
||||||
*/
|
*/
|
||||||
@@ -154,7 +135,6 @@ LUA_API void *(lua_touserdata) (lua_State *L, int idx);
|
|||||||
LUA_API lua_State*(lua_tothread)(lua_State * L, int idx);
|
LUA_API lua_State*(lua_tothread)(lua_State * L, int idx);
|
||||||
LUA_API const void*(lua_topointer)(lua_State * L, int idx);
|
LUA_API const void*(lua_topointer)(lua_State * L, int idx);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** push functions (C -> stack)
|
** push functions (C -> stack)
|
||||||
*/
|
*/
|
||||||
@@ -163,15 +143,13 @@ LUA_API void (lua_pushnumber) (lua_State *L, lua_Number n);
|
|||||||
LUA_API void(lua_pushinteger)(lua_State* L, lua_Integer n);
|
LUA_API void(lua_pushinteger)(lua_State* L, lua_Integer n);
|
||||||
LUA_API void(lua_pushlstring)(lua_State* L, const char* s, size_t l);
|
LUA_API void(lua_pushlstring)(lua_State* L, const char* s, size_t l);
|
||||||
LUA_API void(lua_pushstring)(lua_State* L, const char* s);
|
LUA_API void(lua_pushstring)(lua_State* L, const char* s);
|
||||||
LUA_API const char *(lua_pushvfstring) (lua_State *L, const char *fmt,
|
LUA_API const char*(lua_pushvfstring)(lua_State * L, const char* fmt, va_list argp);
|
||||||
va_list argp);
|
|
||||||
LUA_API const char*(lua_pushfstring)(lua_State * L, const char* fmt, ...);
|
LUA_API const char*(lua_pushfstring)(lua_State * L, const char* fmt, ...);
|
||||||
LUA_API void(lua_pushcclosure)(lua_State* L, lua_CFunction fn, int n);
|
LUA_API void(lua_pushcclosure)(lua_State* L, lua_CFunction fn, int n);
|
||||||
LUA_API void(lua_pushboolean)(lua_State* L, int b);
|
LUA_API void(lua_pushboolean)(lua_State* L, int b);
|
||||||
LUA_API void(lua_pushlightuserdata)(lua_State* L, void* p);
|
LUA_API void(lua_pushlightuserdata)(lua_State* L, void* p);
|
||||||
LUA_API int(lua_pushthread)(lua_State* L);
|
LUA_API int(lua_pushthread)(lua_State* L);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** get functions (Lua -> stack)
|
** get functions (Lua -> stack)
|
||||||
*/
|
*/
|
||||||
@@ -184,7 +162,6 @@ LUA_API void *(lua_newuserdata) (lua_State *L, size_t sz);
|
|||||||
LUA_API int(lua_getmetatable)(lua_State* L, int objindex);
|
LUA_API int(lua_getmetatable)(lua_State* L, int objindex);
|
||||||
LUA_API void(lua_getfenv)(lua_State* L, int idx);
|
LUA_API void(lua_getfenv)(lua_State* L, int idx);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** set functions (stack -> Lua)
|
** set functions (stack -> Lua)
|
||||||
*/
|
*/
|
||||||
@@ -195,19 +172,16 @@ LUA_API void (lua_rawseti) (lua_State *L, int idx, int n);
|
|||||||
LUA_API int(lua_setmetatable)(lua_State* L, int objindex);
|
LUA_API int(lua_setmetatable)(lua_State* L, int objindex);
|
||||||
LUA_API int(lua_setfenv)(lua_State* L, int idx);
|
LUA_API int(lua_setfenv)(lua_State* L, int idx);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** `load' and `call' functions (load and run Lua code)
|
** `load' and `call' functions (load and run Lua code)
|
||||||
*/
|
*/
|
||||||
LUA_API void(lua_call)(lua_State* L, int nargs, int nresults);
|
LUA_API void(lua_call)(lua_State* L, int nargs, int nresults);
|
||||||
LUA_API int(lua_pcall)(lua_State* L, int nargs, int nresults, int errfunc);
|
LUA_API int(lua_pcall)(lua_State* L, int nargs, int nresults, int errfunc);
|
||||||
LUA_API int(lua_cpcall)(lua_State* L, lua_CFunction func, void* ud);
|
LUA_API int(lua_cpcall)(lua_State* L, lua_CFunction func, void* ud);
|
||||||
LUA_API int (lua_load) (lua_State *L, lua_Reader reader, void *dt,
|
LUA_API int(lua_load)(lua_State* L, lua_Reader reader, void* dt, const char* chunkname);
|
||||||
const char *chunkname);
|
|
||||||
|
|
||||||
LUA_API int(lua_dump)(lua_State* L, lua_Writer writer, void* data);
|
LUA_API int(lua_dump)(lua_State* L, lua_Writer writer, void* data);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** coroutine functions
|
** coroutine functions
|
||||||
*/
|
*/
|
||||||
@@ -231,7 +205,6 @@ LUA_API int (lua_status) (lua_State *L);
|
|||||||
|
|
||||||
LUA_API int(lua_gc)(lua_State* L, int what, int data);
|
LUA_API int(lua_gc)(lua_State* L, int what, int data);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** miscellaneous functions
|
** miscellaneous functions
|
||||||
*/
|
*/
|
||||||
@@ -245,8 +218,6 @@ LUA_API void (lua_concat) (lua_State *L, int n);
|
|||||||
LUA_API lua_Alloc(lua_getallocf)(lua_State* L, void** ud);
|
LUA_API lua_Alloc(lua_getallocf)(lua_State* L, void** ud);
|
||||||
LUA_API void lua_setallocf(lua_State* L, lua_Alloc f, void* ud);
|
LUA_API void lua_setallocf(lua_State* L, lua_Alloc f, void* ud);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** ===============================================================
|
** ===============================================================
|
||||||
** some useful macros
|
** some useful macros
|
||||||
@@ -272,16 +243,13 @@ LUA_API void lua_setallocf (lua_State *L, lua_Alloc f, void *ud);
|
|||||||
#define lua_isnone(L, n) (lua_type(L, (n)) == LUA_TNONE)
|
#define lua_isnone(L, n) (lua_type(L, (n)) == LUA_TNONE)
|
||||||
#define lua_isnoneornil(L, n) (lua_type(L, (n)) <= 0)
|
#define lua_isnoneornil(L, n) (lua_type(L, (n)) <= 0)
|
||||||
|
|
||||||
#define lua_pushliteral(L, s) \
|
#define lua_pushliteral(L, s) lua_pushlstring(L, "" s, (sizeof(s) / sizeof(char)) - 1)
|
||||||
lua_pushlstring(L, "" s, (sizeof(s)/sizeof(char))-1)
|
|
||||||
|
|
||||||
#define lua_setglobal(L, s) lua_setfield(L, LUA_GLOBALSINDEX, (s))
|
#define lua_setglobal(L, s) lua_setfield(L, LUA_GLOBALSINDEX, (s))
|
||||||
#define lua_getglobal(L, s) lua_getfield(L, LUA_GLOBALSINDEX, (s))
|
#define lua_getglobal(L, s) lua_getfield(L, LUA_GLOBALSINDEX, (s))
|
||||||
|
|
||||||
#define lua_tostring(L, i) lua_tolstring(L, (i), NULL)
|
#define lua_tostring(L, i) lua_tolstring(L, (i), NULL)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** compatibility macros and functions
|
** compatibility macros and functions
|
||||||
*/
|
*/
|
||||||
@@ -295,18 +263,15 @@ LUA_API void lua_setallocf (lua_State *L, lua_Alloc f, void *ud);
|
|||||||
#define lua_Chunkreader lua_Reader
|
#define lua_Chunkreader lua_Reader
|
||||||
#define lua_Chunkwriter lua_Writer
|
#define lua_Chunkwriter lua_Writer
|
||||||
|
|
||||||
|
|
||||||
/* hack */
|
/* hack */
|
||||||
LUA_API void lua_setlevel(lua_State* from, lua_State* to);
|
LUA_API void lua_setlevel(lua_State* from, lua_State* to);
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** {======================================================================
|
** {======================================================================
|
||||||
** Debug API
|
** Debug API
|
||||||
** =======================================================================
|
** =======================================================================
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Event codes
|
** Event codes
|
||||||
*/
|
*/
|
||||||
@@ -316,7 +281,6 @@ LUA_API void lua_setlevel (lua_State *from, lua_State *to);
|
|||||||
#define LUA_HOOKCOUNT 3
|
#define LUA_HOOKCOUNT 3
|
||||||
#define LUA_HOOKTAILRET 4
|
#define LUA_HOOKTAILRET 4
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Event masks
|
** Event masks
|
||||||
*/
|
*/
|
||||||
@@ -327,11 +291,9 @@ LUA_API void lua_setlevel (lua_State *from, lua_State *to);
|
|||||||
|
|
||||||
typedef struct lua_Debug lua_Debug; /* activation record */
|
typedef struct lua_Debug lua_Debug; /* activation record */
|
||||||
|
|
||||||
|
|
||||||
/* Functions to be called by the debuger in specific events */
|
/* Functions to be called by the debuger in specific events */
|
||||||
typedef void (*lua_Hook)(lua_State* L, lua_Debug* ar);
|
typedef void (*lua_Hook)(lua_State* L, lua_Debug* ar);
|
||||||
|
|
||||||
|
|
||||||
LUA_API int lua_getstack(lua_State* L, int level, lua_Debug* ar);
|
LUA_API int lua_getstack(lua_State* L, int level, lua_Debug* ar);
|
||||||
LUA_API int lua_getinfo(lua_State* L, const char* what, lua_Debug* ar);
|
LUA_API int lua_getinfo(lua_State* L, const char* what, lua_Debug* ar);
|
||||||
LUA_API const char* lua_getlocal(lua_State* L, const lua_Debug* ar, int n);
|
LUA_API const char* lua_getlocal(lua_State* L, const lua_Debug* ar, int n);
|
||||||
@@ -346,8 +308,7 @@ LUA_API int lua_gethookcount (lua_State *L);
|
|||||||
/* From Lua 5.2. */
|
/* From Lua 5.2. */
|
||||||
LUA_API void* lua_upvalueid(lua_State* L, int idx, int n);
|
LUA_API void* lua_upvalueid(lua_State* L, int idx, int n);
|
||||||
LUA_API void lua_upvaluejoin(lua_State* L, int idx1, int n1, int idx2, int n2);
|
LUA_API void lua_upvaluejoin(lua_State* L, int idx1, int n1, int idx2, int n2);
|
||||||
LUA_API int lua_loadx (lua_State *L, lua_Reader reader, void *dt,
|
LUA_API int lua_loadx(lua_State* L, lua_Reader reader, void* dt, const char* chunkname, const char* mode);
|
||||||
const char *chunkname, const char *mode);
|
|
||||||
LUA_API const lua_Number* lua_version(lua_State* L);
|
LUA_API const lua_Number* lua_version(lua_State* L);
|
||||||
LUA_API void lua_copy(lua_State* L, int fromidx, int toidx);
|
LUA_API void lua_copy(lua_State* L, int fromidx, int toidx);
|
||||||
LUA_API lua_Number lua_tonumberx(lua_State* L, int idx, int* isnum);
|
LUA_API lua_Number lua_tonumberx(lua_State* L, int idx, int* isnum);
|
||||||
@@ -356,7 +317,6 @@ LUA_API lua_Integer lua_tointegerx (lua_State *L, int idx, int *isnum);
|
|||||||
/* From Lua 5.3. */
|
/* From Lua 5.3. */
|
||||||
LUA_API int lua_isyieldable(lua_State* L);
|
LUA_API int lua_isyieldable(lua_State* L);
|
||||||
|
|
||||||
|
|
||||||
struct lua_Debug {
|
struct lua_Debug {
|
||||||
int event;
|
int event;
|
||||||
const char* name; /* (n) */
|
const char* name; /* (n) */
|
||||||
@@ -374,7 +334,6 @@ struct lua_Debug {
|
|||||||
|
|
||||||
/* }====================================================================== */
|
/* }====================================================================== */
|
||||||
|
|
||||||
|
|
||||||
/******************************************************************************
|
/******************************************************************************
|
||||||
* Copyright (C) 1994-2008 Lua.org, PUC-Rio. All rights reserved.
|
* Copyright (C) 1994-2008 Lua.org, PUC-Rio. All rights reserved.
|
||||||
*
|
*
|
||||||
@@ -398,5 +357,4 @@ struct lua_Debug {
|
|||||||
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
******************************************************************************/
|
******************************************************************************/
|
||||||
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -1,9 +1,8 @@
|
|||||||
// C++ wrapper for LuaJIT header files.
|
// C++ wrapper for LuaJIT header files.
|
||||||
|
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#include "lua.h"
|
|
||||||
#include "lauxlib.h"
|
#include "lauxlib.h"
|
||||||
#include "lualib.h"
|
#include "lua.h"
|
||||||
#include "luajit.h"
|
#include "luajit.h"
|
||||||
|
#include "lualib.h"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -9,6 +9,7 @@
|
|||||||
#ifndef WINVER
|
#ifndef WINVER
|
||||||
#define WINVER 0x0501
|
#define WINVER 0x0501
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <limits.h>
|
#include <limits.h>
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
|
|
||||||
@@ -20,10 +21,8 @@
|
|||||||
*/
|
*/
|
||||||
#define LUA_LDIR "!\\lua\\"
|
#define LUA_LDIR "!\\lua\\"
|
||||||
#define LUA_CDIR "!\\"
|
#define LUA_CDIR "!\\"
|
||||||
#define LUA_PATH_DEFAULT \
|
#define LUA_PATH_DEFAULT ".\\?.lua;" LUA_LDIR "?.lua;" LUA_LDIR "?\\init.lua;"
|
||||||
".\\?.lua;" LUA_LDIR"?.lua;" LUA_LDIR"?\\init.lua;"
|
#define LUA_CPATH_DEFAULT ".\\?.dll;" LUA_CDIR "?.dll;" LUA_CDIR "loadall.dll"
|
||||||
#define LUA_CPATH_DEFAULT \
|
|
||||||
".\\?.dll;" LUA_CDIR"?.dll;" LUA_CDIR"loadall.dll"
|
|
||||||
#else
|
#else
|
||||||
/*
|
/*
|
||||||
** Note to distribution maintainers: do NOT patch the following lines!
|
** Note to distribution maintainers: do NOT patch the following lines!
|
||||||
@@ -77,9 +76,7 @@
|
|||||||
#define LUA_PATH_MARK "?"
|
#define LUA_PATH_MARK "?"
|
||||||
#define LUA_EXECDIR "!"
|
#define LUA_EXECDIR "!"
|
||||||
#define LUA_IGMARK "-"
|
#define LUA_IGMARK "-"
|
||||||
#define LUA_PATH_CONFIG \
|
#define LUA_PATH_CONFIG LUA_DIRSEP "\n" LUA_PATHSEP "\n" LUA_PATH_MARK "\n" LUA_EXECDIR "\n" LUA_IGMARK "\n"
|
||||||
LUA_DIRSEP "\n" LUA_PATHSEP "\n" LUA_PATH_MARK "\n" \
|
|
||||||
LUA_EXECDIR "\n" LUA_IGMARK "\n"
|
|
||||||
|
|
||||||
/* Quoting in error messages. */
|
/* Quoting in error messages. */
|
||||||
#define LUA_QL(x) "'" x "'"
|
#define LUA_QL(x) "'" x "'"
|
||||||
@@ -144,9 +141,16 @@
|
|||||||
#define lua_assert(x) assert(x)
|
#define lua_assert(x) assert(x)
|
||||||
#endif
|
#endif
|
||||||
#ifdef LUA_USE_APICHECK
|
#ifdef LUA_USE_APICHECK
|
||||||
#define luai_apicheck(L, o) { (void)L; assert(o); }
|
#define luai_apicheck(L, o) \
|
||||||
|
{ \
|
||||||
|
(void)L; \
|
||||||
|
assert(o); \
|
||||||
|
}
|
||||||
#else
|
#else
|
||||||
#define luai_apicheck(L, o) { (void)L; }
|
#define luai_apicheck(L, o) \
|
||||||
|
{ \
|
||||||
|
(void)L; \
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -65,13 +65,10 @@ enum {
|
|||||||
LUA_API int luaJIT_setmode(lua_State* L, int idx, int mode);
|
LUA_API int luaJIT_setmode(lua_State* L, int idx, int mode);
|
||||||
|
|
||||||
/* Low-overhead profiling API. */
|
/* Low-overhead profiling API. */
|
||||||
typedef void (*luaJIT_profile_callback)(void *data, lua_State *L,
|
typedef void (*luaJIT_profile_callback)(void* data, lua_State* L, int samples, int vmstate);
|
||||||
int samples, int vmstate);
|
LUA_API void luaJIT_profile_start(lua_State* L, const char* mode, luaJIT_profile_callback cb, void* data);
|
||||||
LUA_API void luaJIT_profile_start(lua_State *L, const char *mode,
|
|
||||||
luaJIT_profile_callback cb, void *data);
|
|
||||||
LUA_API void luaJIT_profile_stop(lua_State* L);
|
LUA_API void luaJIT_profile_stop(lua_State* L);
|
||||||
LUA_API const char *luaJIT_profile_dumpstack(lua_State *L, const char *fmt,
|
LUA_API const char* luaJIT_profile_dumpstack(lua_State* L, const char* fmt, int depth, size_t* len);
|
||||||
int depth, size_t *len);
|
|
||||||
|
|
||||||
/* Enforce (dynamic) linker error for version mismatches. Call from main. */
|
/* Enforce (dynamic) linker error for version mismatches. Call from main. */
|
||||||
LUA_API void LUAJIT_VERSION_SYM(void);
|
LUA_API void LUAJIT_VERSION_SYM(void);
|
||||||
|
|||||||
@@ -2,15 +2,18 @@
|
|||||||
//////////////////////////////////////////////////
|
//////////////////////////////////////////////////
|
||||||
// BlFuncs Version 1.0
|
// BlFuncs Version 1.0
|
||||||
|
|
||||||
|
|
||||||
// Includes
|
// Includes
|
||||||
|
#ifndef WIN32_LEAN_AND_MEAN
|
||||||
#include "BlHooks.hpp"
|
#define WIN32_LEAN_AND_MEAN
|
||||||
|
#endif
|
||||||
#include "BlFuncs.hpp"
|
#include "BlFuncs.hpp"
|
||||||
|
#include <Windows.h>
|
||||||
|
#include "BlHooks.hpp"
|
||||||
|
|
||||||
|
#include <cstdarg>
|
||||||
|
#include <cstring>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
|
|
||||||
// Scanned structures
|
// Scanned structures
|
||||||
|
|
||||||
ADDR tsf_mCacheSequence;
|
ADDR tsf_mCacheSequence;
|
||||||
@@ -33,7 +36,6 @@ BlFunctionDefIntern(tsf_BlSimObject__getDataField );
|
|||||||
BlFunctionDefIntern(tsf_BlSimObject__setDataField);
|
BlFunctionDefIntern(tsf_BlSimObject__setDataField);
|
||||||
BlFunctionDefIntern(tsf_BlCon__getReturnBuffer);
|
BlFunctionDefIntern(tsf_BlCon__getReturnBuffer);
|
||||||
|
|
||||||
|
|
||||||
// C->TS Args
|
// C->TS Args
|
||||||
|
|
||||||
char* tsf_GetIntArg(signed int value) {
|
char* tsf_GetIntArg(signed int value) {
|
||||||
@@ -41,21 +43,21 @@ char* tsf_GetIntArg(signed int value) {
|
|||||||
snprintf(ret, 16, "%d", value);
|
snprintf(ret, 16, "%d", value);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* tsf_GetFloatArg(float value) {
|
char* tsf_GetFloatArg(float value) {
|
||||||
char* ret = tsf_BlStringStack__getArgBuffer(32);
|
char* ret = tsf_BlStringStack__getArgBuffer(32);
|
||||||
snprintf(ret, 32, "%g", value);
|
snprintf(ret, 32, "%g", value);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* tsf_GetStringArg(char* value) {
|
char* tsf_GetStringArg(char* value) {
|
||||||
int len = strlen(value) + 1;
|
int len = strlen(value) + 1;
|
||||||
char* ret = tsf_BlStringStack__getArgBuffer(len);
|
char* ret = tsf_BlStringStack__getArgBuffer(len);
|
||||||
memcpy(ret, value, len);
|
memcpy(ret, value, len);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
char* tsf_GetThisArg(ADDR obj) {
|
|
||||||
return tsf_GetIntArg(*(signed int *)(obj + 32));
|
|
||||||
}
|
|
||||||
|
|
||||||
|
char* tsf_GetThisArg(ADDR obj) { return tsf_GetIntArg(*(signed int*)(obj + 32)); }
|
||||||
|
|
||||||
// Eval
|
// Eval
|
||||||
|
|
||||||
@@ -74,24 +76,23 @@ const char* tsf_Evalf(const char *fmt, ...) {
|
|||||||
return tsf_Eval((const char*)code);
|
return tsf_Eval((const char*)code);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Objects
|
// Objects
|
||||||
|
|
||||||
ADDR tsf_FindObject(unsigned int id) {
|
ADDR tsf_FindObject(unsigned int id) {
|
||||||
ADDR obj = *(ADDR*)(*(ADDR*)(tsf_gIdDictionary) + 4 * (id & 0xFFF));
|
ADDR obj = *(ADDR*)(*(ADDR*)(tsf_gIdDictionary) + 4 * (id & 0xFFF));
|
||||||
if(!obj) return 0;
|
if (!obj)
|
||||||
|
return 0;
|
||||||
|
|
||||||
while (obj && *(unsigned int*)(obj + 32) != id) {
|
while (obj && *(unsigned int*)(obj + 32) != id) {
|
||||||
obj = *(ADDR*)(obj + 16);
|
obj = *(ADDR*)(obj + 16);
|
||||||
if(!obj) return 0;
|
if (!obj)
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
return obj;
|
return obj;
|
||||||
}
|
}
|
||||||
|
|
||||||
ADDR tsf_FindObject(const char* name) {
|
ADDR tsf_FindObject(const char* name) { return (ADDR)tsf_BlSim__findObject_name(name); }
|
||||||
return (ADDR)tsf_BlSim__findObject_name(name);
|
|
||||||
}
|
|
||||||
|
|
||||||
ADDR tsf_LookupNamespace(const char* ns, const char* package) {
|
ADDR tsf_LookupNamespace(const char* ns, const char* package) {
|
||||||
const char* ste_package;
|
const char* ste_package;
|
||||||
@@ -108,10 +109,8 @@ ADDR tsf_LookupNamespace(const char* ns, const char* package) {
|
|||||||
return tsf_BlNamespace__find(nullptr, ste_package);
|
return tsf_BlNamespace__find(nullptr, ste_package);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ADDR tsf_LookupNamespace(const char* ns) {
|
|
||||||
return tsf_LookupNamespace(ns, nullptr);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
ADDR tsf_LookupNamespace(const char* ns) { return tsf_LookupNamespace(ns, nullptr); }
|
||||||
|
|
||||||
// Object Fields
|
// Object Fields
|
||||||
|
|
||||||
@@ -137,34 +136,32 @@ void tsf_SetDataField(ADDR simObject, const char* slotName, const char* array, c
|
|||||||
tsf_BlSimObject__setDataField(simObject, ste_slotName, array, value);
|
tsf_BlSimObject__setDataField(simObject, ste_slotName, array, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// TS Global Variables
|
// TS Global Variables
|
||||||
|
|
||||||
const char *tsf_GetVar(const char* name) {
|
const char* tsf_GetVar(const char* name) { return tsf_BlCon__getVariable(name); }
|
||||||
return tsf_BlCon__getVariable(name);
|
|
||||||
}
|
|
||||||
|
|
||||||
void tsf_AddVarInternal(const char* name, signed int varType, void* data) {
|
void tsf_AddVarInternal(const char* name, signed int varType, void* data) {
|
||||||
tsf_BlDictionary__addVariable((ADDR*)tsf_gEvalState_globalVars, name, varType, data);
|
tsf_BlDictionary__addVariable((ADDR*)tsf_gEvalState_globalVars, name, varType, data);
|
||||||
}
|
}
|
||||||
|
|
||||||
void tsf_AddVar(const char* name, const char** data) {
|
void tsf_AddVar(const char* name, const char** data) { tsf_AddVarInternal(name, 10, data); }
|
||||||
tsf_AddVarInternal(name, 10, data);
|
|
||||||
}
|
|
||||||
void tsf_AddVar(const char* name, signed int* data) {
|
|
||||||
tsf_AddVarInternal(name, 4, data);
|
|
||||||
}
|
|
||||||
void tsf_AddVar(const char* name, float* data) {
|
|
||||||
tsf_AddVarInternal(name, 8, data);
|
|
||||||
}
|
|
||||||
void tsf_AddVar(const char* name, bool* data) {
|
|
||||||
tsf_AddVarInternal(name, 6, data);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
void tsf_AddVar(const char* name, signed int* data) { tsf_AddVarInternal(name, 4, data); }
|
||||||
|
|
||||||
|
void tsf_AddVar(const char* name, float* data) { tsf_AddVarInternal(name, 8, data); }
|
||||||
|
|
||||||
|
void tsf_AddVar(const char* name, bool* data) { tsf_AddVarInternal(name, 6, data); }
|
||||||
|
|
||||||
// TS->C Functions
|
// TS->C Functions
|
||||||
|
|
||||||
ADDR tsf_AddConsoleFuncInternal(const char* pname, const char* cname, const char* fname, signed int cbtype, const char* usage, signed int mina, signed int maxa) {
|
ADDR tsf_AddConsoleFuncInternal(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
signed int cbtype,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
const char* ste_fname = tsf_BlStringTable__insert(fname, 0);
|
const char* ste_fname = tsf_BlStringTable__insert(fname, 0);
|
||||||
ADDR ns = tsf_LookupNamespace(cname, pname);
|
ADDR ns = tsf_LookupNamespace(cname, pname);
|
||||||
ADDR ent = tsf_BlNamespace__createLocalEntry(ns, ste_fname);
|
ADDR ent = tsf_BlNamespace__createLocalEntry(ns, ste_fname);
|
||||||
@@ -180,50 +177,121 @@ ADDR tsf_AddConsoleFuncInternal(const char* pname, const char* cname, const char
|
|||||||
return ent;
|
return ent;
|
||||||
}
|
}
|
||||||
|
|
||||||
void tsf_AddConsoleFunc(const char* pname, const char* cname, const char* fname, tsf_StringCallback sc, const char* usage, signed int mina, signed int maxa) {
|
void tsf_AddConsoleFunc(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
tsf_StringCallback sc,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 1, usage, mina, maxa);
|
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 1, usage, mina, maxa);
|
||||||
*(tsf_StringCallback*)(ent + 40) = sc;
|
*(tsf_StringCallback*)(ent + 40) = sc;
|
||||||
}
|
}
|
||||||
void tsf_AddConsoleFunc(const char* pname, const char* cname, const char* fname, tsf_IntCallback ic, const char* usage, signed int mina, signed int maxa) {
|
|
||||||
|
void tsf_AddConsoleFunc(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
tsf_IntCallback ic,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 2, usage, mina, maxa);
|
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 2, usage, mina, maxa);
|
||||||
*(tsf_IntCallback*)(ent + 40) = ic;
|
*(tsf_IntCallback*)(ent + 40) = ic;
|
||||||
}
|
}
|
||||||
void tsf_AddConsoleFunc(const char* pname, const char* cname, const char* fname, tsf_FloatCallback fc, const char* usage, signed int mina, signed int maxa) {
|
|
||||||
|
void tsf_AddConsoleFunc(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
tsf_FloatCallback fc,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 3, usage, mina, maxa);
|
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 3, usage, mina, maxa);
|
||||||
*(tsf_FloatCallback*)(ent + 40) = fc;
|
*(tsf_FloatCallback*)(ent + 40) = fc;
|
||||||
}
|
}
|
||||||
void tsf_AddConsoleFunc(const char* pname, const char* cname, const char* fname, tsf_VoidCallback vc, const char* usage, signed int mina, signed int maxa) {
|
|
||||||
|
void tsf_AddConsoleFunc(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
tsf_VoidCallback vc,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 4, usage, mina, maxa);
|
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 4, usage, mina, maxa);
|
||||||
*(tsf_VoidCallback*)(ent + 40) = vc;
|
*(tsf_VoidCallback*)(ent + 40) = vc;
|
||||||
}
|
}
|
||||||
void tsf_AddConsoleFunc(const char* pname, const char* cname, const char* fname, tsf_BoolCallback bc, const char* usage, signed int mina, signed int maxa) {
|
|
||||||
|
void tsf_AddConsoleFunc(
|
||||||
|
const char* pname,
|
||||||
|
const char* cname,
|
||||||
|
const char* fname,
|
||||||
|
tsf_BoolCallback bc,
|
||||||
|
const char* usage,
|
||||||
|
signed int mina,
|
||||||
|
signed int maxa) {
|
||||||
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 5, usage, mina, maxa);
|
ADDR ent = tsf_AddConsoleFuncInternal(pname, cname, fname, 5, usage, mina, maxa);
|
||||||
*(tsf_BoolCallback*)(ent + 40) = bc;
|
*(tsf_BoolCallback*)(ent + 40) = bc;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Initialization
|
// Initialization
|
||||||
|
|
||||||
bool tsf_InitInternal() {
|
bool tsf_InitInternal() {
|
||||||
BlScanFunctionText(tsf_BlStringTable__insert, "83 EC 0C 80 3D ? ? ? ? ?");
|
BlScanFunctionText(tsf_BlStringTable__insert, "83 EC 0C 80 3D ? ? ? ? ?");
|
||||||
BlScanFunctionText(tsf_BlNamespace__find , "55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 83 EC 0C 53 56 57 A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B DA 8B D1" );
|
BlScanFunctionText(
|
||||||
BlScanFunctionText(tsf_BlNamespace__createLocalEntry, "55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 83 EC 08 53 56 57 A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 89 4D F0" );
|
tsf_BlNamespace__find,
|
||||||
BlScanFunctionText(tsf_BlDataChunker__freeBlocks , "55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 51 53 56 57 A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B D9 8B 33" );
|
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 83 EC 0C 53 56 57 A1 ? ? ? ? "
|
||||||
BlScanFunctionText(tsf_BlCon__evaluate , "55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 56 57 A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B 75 10" );
|
"33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B DA 8B D1");
|
||||||
BlScanFunctionText(tsf_BlCon__executef , "81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 55 56 8B B4 24 ? ? ? ? 33 C9" );
|
BlScanFunctionText(
|
||||||
BlScanFunctionText(tsf_BlCon__executefSimObj , "81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 56 8B B4 24 ? ? ? ? 33 C9" );
|
tsf_BlNamespace__createLocalEntry,
|
||||||
|
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 83 EC 08 53 56 57 A1 ? ? ? ? "
|
||||||
|
"33 C5 50 8D 45 F4 64 A3 ? ? ? ? 89 4D F0");
|
||||||
|
BlScanFunctionText(
|
||||||
|
tsf_BlDataChunker__freeBlocks,
|
||||||
|
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 51 53 56 57 "
|
||||||
|
"A1 ? ? ? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B D9 8B 33");
|
||||||
|
BlScanFunctionText(
|
||||||
|
tsf_BlCon__evaluate,
|
||||||
|
"55 8B EC 6A FF 68 ? ? ? ? 64 A1 ? ? ? ? 50 56 57 A1 ? ? "
|
||||||
|
"? ? 33 C5 50 8D 45 F4 64 A3 ? ? ? ? 8B 75 10");
|
||||||
|
BlScanFunctionText(
|
||||||
|
tsf_BlCon__executef,
|
||||||
|
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 55 56 "
|
||||||
|
"8B B4 24 ? ? ? ? 33 C9");
|
||||||
|
BlScanFunctionText(
|
||||||
|
tsf_BlCon__executefSimObj,
|
||||||
|
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 53 56 8B "
|
||||||
|
"B4 24 ? ? ? ? 33 C9");
|
||||||
BlScanFunctionText(tsf_BlCon__getVariable, "53 56 8B F1 57 85 F6 0F 84 ? ? ? ?");
|
BlScanFunctionText(tsf_BlCon__getVariable, "53 56 8B F1 57 85 F6 0F 84 ? ? ? ?");
|
||||||
BlScanFunctionText(tsf_BlDictionary__addVariable, "8B 44 24 04 56 57 8B F9");
|
BlScanFunctionText(tsf_BlDictionary__addVariable, "8B 44 24 04 56 57 8B F9");
|
||||||
BlScanFunctionText(tsf_BlSim__findObject_name, "57 8B F9 8A 17");
|
BlScanFunctionText(tsf_BlSim__findObject_name, "57 8B F9 8A 17");
|
||||||
BlScanFunctionText(tsf_BlStringStack__getArgBuffer , "55 8B EC 83 E4 F8 8B 0D ? ? ? ? A1 ? ? ? ? 56 57 8B 7D 08 8D 14 01 03 D7 3B 15 ? ? ? ? 72 2C 8B 0D" );
|
BlScanFunctionText(
|
||||||
|
tsf_BlStringStack__getArgBuffer,
|
||||||
|
"55 8B EC 83 E4 F8 8B 0D ? ? ? ? A1 ? ? ? ? 56 57 8B 7D "
|
||||||
|
"08 8D 14 01 03 D7 3B 15 ? ? ? ? 72 2C 8B 0D");
|
||||||
BlScanFunctionText(tsf_BlSimObject__getDataField, "51 53 8B D9 55 56 8B 74 24 14");
|
BlScanFunctionText(tsf_BlSimObject__getDataField, "51 53 8B D9 55 56 8B 74 24 14");
|
||||||
BlScanFunctionText(tsf_BlSimObject__setDataField , "81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 8B 84 24 ? ? ? ? 53 8B D9 89 44 24 04" );
|
BlScanFunctionText(
|
||||||
|
tsf_BlSimObject__setDataField,
|
||||||
|
"81 EC ? ? ? ? A1 ? ? ? ? 33 C4 89 84 24 ? ? ? ? 8B 84 24 "
|
||||||
|
"? ? ? ? 53 8B D9 89 44 24 04");
|
||||||
BlScanFunctionText(tsf_BlCon__getReturnBuffer, "81 F9 ? ? ? ? 76 2B");
|
BlScanFunctionText(tsf_BlCon__getReturnBuffer, "81 F9 ? ? ? ? 76 2B");
|
||||||
|
|
||||||
ADDR BlScanText (tsf_mCacheSequenceLoc , "FF 05 ? ? ? ? B9 ? ? ? ? 8B F8 E8 ? ? ? ? 8B 44 24 1C 89 47 18 8B 44 24 14" );
|
ADDR BlScanText(
|
||||||
ADDR BlScanText (tsf_mCacheAllocatorLoc , "89 35 ? ? ? ? C7 06 ? ? ? ? A1 ? ? ? ? 68 ? ? ? ? C7 40 ? ? ? ? ? E8 ? ? ? ? 83 C4 04 8B 4D F4 64 89 0D ? ? ? ? 59 5E 8B E5 5D C3");
|
tsf_mCacheSequenceLoc,
|
||||||
|
"FF 05 ? ? ? ? B9 ? ? ? ? 8B F8 E8 ? ? ? ? 8B 44 24 1C 89 47 "
|
||||||
|
"18 8B 44 24 14");
|
||||||
|
ADDR BlScanText(
|
||||||
|
tsf_mCacheAllocatorLoc,
|
||||||
|
"89 35 ? ? ? ? C7 06 ? ? ? ? A1 ? ? ? ? 68 ? ? ? ? C7 40 ? ? ? ? ? E8 ? "
|
||||||
|
"? ? ? 83 C4 04 8B 4D F4 64 89 0D ? ? ? ? 59 5E 8B E5 5D C3");
|
||||||
ADDR BlScanText(tsf_gIdDictionaryLoc, "89 15 ? ? ? ? E8 ? ? ? ? 8B F0 89 75 F0");
|
ADDR BlScanText(tsf_gIdDictionaryLoc, "89 15 ? ? ? ? E8 ? ? ? ? 8B F0 89 75 F0");
|
||||||
ADDR BlScanText (tsf_gEvalState_globalVarsLoc , "B9 ? ? ? ? E8 ? ? ? ? 68 ? ? ? ? 6A 0A 68 ? ? ? ? B9 ? ? ? ? E8 ? ? ? ? E8 ? ? ? ?" );
|
ADDR BlScanText(
|
||||||
|
tsf_gEvalState_globalVarsLoc,
|
||||||
|
"B9 ? ? ? ? E8 ? ? ? ? 68 ? ? ? ? 6A 0A 68 ? ? ? ? B9 ? ? ? "
|
||||||
|
"? E8 ? ? ? ? E8 ? ? ? ?");
|
||||||
|
|
||||||
tsf_mCacheSequence = *(ADDR*)(tsf_mCacheSequenceLoc + 2);
|
tsf_mCacheSequence = *(ADDR*)(tsf_mCacheSequenceLoc + 2);
|
||||||
tsf_mCacheAllocator = *(ADDR*)(tsf_mCacheAllocatorLoc + 2);
|
tsf_mCacheAllocator = *(ADDR*)(tsf_mCacheAllocatorLoc + 2);
|
||||||
@@ -233,6 +301,4 @@ bool tsf_InitInternal() {
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool tsf_DeinitInternal() {
|
bool tsf_DeinitInternal() { return true; }
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -5,10 +5,10 @@
|
|||||||
#ifndef _H_BLFUNCS
|
#ifndef _H_BLFUNCS
|
||||||
#define _H_BLFUNCS
|
#define _H_BLFUNCS
|
||||||
|
|
||||||
// Require BlHooks to be included before this header
|
// Ensure BlHooks is available (include it if not already included)
|
||||||
#ifndef _H_BLHOOKS
|
#ifndef _H_BLHOOKS
|
||||||
#error "BlFuncs.hpp: You must include BlHooks.hpp first"
|
#include "BlHooks.hpp"
|
||||||
#else
|
#endif
|
||||||
|
|
||||||
typedef const char* (*tsf_StringCallback)(ADDR, signed int, const char*[]);
|
typedef const char* (*tsf_StringCallback)(ADDR, signed int, const char*[]);
|
||||||
typedef signed int (*tsf_IntCallback)(ADDR, signed int, const char*[]);
|
typedef signed int (*tsf_IntCallback)(ADDR, signed int, const char*[]);
|
||||||
@@ -71,7 +71,6 @@ BlFunctionDefExtern(const char *, __thiscall, tsf_BlSimObject__getDataField, ADD
|
|||||||
BlFunctionDefExtern(void, __thiscall, tsf_BlSimObject__setDataField, ADDR, const char*, const char*, const char*);
|
BlFunctionDefExtern(void, __thiscall, tsf_BlSimObject__setDataField, ADDR, const char*, const char*, const char*);
|
||||||
BlFunctionDefExtern(char*, __fastcall, tsf_BlCon__getReturnBuffer, unsigned int);
|
BlFunctionDefExtern(char*, __fastcall, tsf_BlCon__getReturnBuffer, unsigned int);
|
||||||
|
|
||||||
|
|
||||||
// Function short names
|
// Function short names
|
||||||
|
|
||||||
#define BlEval tsf_Eval
|
#define BlEval tsf_Eval
|
||||||
@@ -94,16 +93,21 @@ BlFunctionDefExtern(char *, __fastcall, tsf_BlCon__getReturnBuffer, unsigned int
|
|||||||
|
|
||||||
#define BlAddFunction tsf_AddConsoleFunc
|
#define BlAddFunction tsf_AddConsoleFunc
|
||||||
|
|
||||||
#define __22ND_ARGUMENT(a1, a2, a3, a4, a5, a6, a7, a8, a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20, a21, a22, ...) a22
|
#define __22ND_ARGUMENT( \
|
||||||
#define __NUM_LIST(...) __22ND_ARGUMENT(dummy, ##__VA_ARGS__, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
|
a1, a2, a3, a4, a5, a6, a7, a8, a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20, a21, a22, ...) \
|
||||||
#define BlCall(...) \
|
a22
|
||||||
tsf_BlCon__executef(__NUM_LIST(__VA_ARGS__), __VA_ARGS__)
|
#define __NUM_LIST(...) \
|
||||||
#define BlCallObj(obj, ...) \
|
__22ND_ARGUMENT(dummy, ##__VA_ARGS__, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
|
||||||
tsf_BlCon__executefSimObj((ADDR*)obj, __NUM_LIST(__VA_ARGS__), __VA_ARGS__)
|
#define BlCall(...) tsf_BlCon__executef(__NUM_LIST(__VA_ARGS__), __VA_ARGS__)
|
||||||
|
#define BlCallObj(obj, ...) tsf_BlCon__executefSimObj((ADDR*)obj, __NUM_LIST(__VA_ARGS__), __VA_ARGS__)
|
||||||
#define BlFuncsInit() if(!tsf_InitInternal()) { return false; }
|
|
||||||
#define BlFuncsDeinit() if(!tsf_DeinitInternal()) { return false; }
|
|
||||||
|
|
||||||
|
#define BlFuncsInit() \
|
||||||
|
if (!tsf_InitInternal()) { \
|
||||||
|
return false; \
|
||||||
|
}
|
||||||
|
#define BlFuncsDeinit() \
|
||||||
|
if (!tsf_DeinitInternal()) { \
|
||||||
|
return false; \
|
||||||
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
#endif
|
|
||||||
|
|||||||
@@ -2,21 +2,22 @@
|
|||||||
//////////////////////////////////////////////////
|
//////////////////////////////////////////////////
|
||||||
// RedoBlHooks Version 3.0
|
// RedoBlHooks Version 3.0
|
||||||
|
|
||||||
|
|
||||||
// Includes
|
// Includes
|
||||||
|
|
||||||
|
#ifndef WIN32_LEAN_AND_MEAN
|
||||||
|
#define WIN32_LEAN_AND_MEAN
|
||||||
|
#include <cstdlib>
|
||||||
|
#endif
|
||||||
#include <Windows.h>
|
#include <Windows.h>
|
||||||
#include <Psapi.h>
|
#include <Psapi.h>
|
||||||
// #include <map>
|
// #include <map>
|
||||||
|
|
||||||
#include "BlHooks.hpp"
|
#include "BlHooks.hpp"
|
||||||
|
|
||||||
|
|
||||||
// Scanned structures
|
// Scanned structures
|
||||||
|
|
||||||
BlFunctionDefIntern(tsh_BlPrintf);
|
BlFunctionDefIntern(tsh_BlPrintf);
|
||||||
|
|
||||||
|
|
||||||
// Sig Scanning
|
// Sig Scanning
|
||||||
|
|
||||||
ADDR ImageBase;
|
ADDR ImageBase;
|
||||||
@@ -87,7 +88,6 @@ void tsh_i_PatternTextToCode(char* text, char** opatt, char** omask) {
|
|||||||
*omask = mask;
|
*omask = mask;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Public functions for sig scanning
|
// Public functions for sig scanning
|
||||||
|
|
||||||
// Scan using code-style pattern
|
// Scan using code-style pattern
|
||||||
@@ -109,7 +109,6 @@ ADDR tsh_ScanText(char* text) {
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Call Patching and Hooking
|
// Call Patching and Hooking
|
||||||
|
|
||||||
// Remove protection from address
|
// Remove protection from address
|
||||||
@@ -127,21 +126,13 @@ void tsh_PatchBytes(ADDR length, ADDR location, BYTE* repl) {
|
|||||||
memcpy((void*)location, (void*)repl, (size_t)length);
|
memcpy((void*)location, (void*)repl, (size_t)length);
|
||||||
}
|
}
|
||||||
|
|
||||||
void tsh_PatchByte(ADDR location, BYTE value) {
|
void tsh_PatchByte(ADDR location, BYTE value) { tsh_PatchBytes(location, 1, &value); }
|
||||||
tsh_PatchBytes(location, 1, &value);
|
|
||||||
}
|
|
||||||
|
|
||||||
void tsh_ReplaceInt(ADDR addr, int rval) {
|
void tsh_ReplaceInt(ADDR addr, int rval) { tsh_PatchBytes(4, addr, (BYTE*)(&rval)); }
|
||||||
tsh_PatchBytes(4, addr, (BYTE*)(&rval));
|
|
||||||
}
|
|
||||||
|
|
||||||
int tsh_i_CallOffset(ADDR instr, ADDR func) {
|
int tsh_i_CallOffset(ADDR instr, ADDR func) { return func - (instr + 4); }
|
||||||
return func - (instr+4);
|
|
||||||
}
|
|
||||||
|
|
||||||
void tsh_i_ReplaceCall(ADDR instr, ADDR target) {
|
void tsh_i_ReplaceCall(ADDR instr, ADDR target) { tsh_ReplaceInt(instr, tsh_i_CallOffset(instr, target)); }
|
||||||
tsh_ReplaceInt(instr, tsh_i_CallOffset(instr, target));
|
|
||||||
}
|
|
||||||
|
|
||||||
void tsh_i_PatchCopy(ADDR dest, ADDR src, unsigned int len) {
|
void tsh_i_PatchCopy(ADDR dest, ADDR src, unsigned int len) {
|
||||||
for (unsigned int i = 0; i < len; i++) {
|
for (unsigned int i = 0; i < len; i++) {
|
||||||
@@ -165,7 +156,8 @@ int tsh_PatchAllMatchesCode(ADDR len, char* patt, char* mask, char* replace, boo
|
|||||||
int numpatched = 0;
|
int numpatched = 0;
|
||||||
for (ADDR i = ImageBase; i < ImageBase + ImageSize - len; i++) {
|
for (ADDR i = ImageBase; i < ImageBase + ImageSize - len; i++) {
|
||||||
if (tsh_i_CompareData((BYTE*)i, (BYTE*)patt, mask)) {
|
if (tsh_i_CompareData((BYTE*)i, (BYTE*)patt, mask)) {
|
||||||
if(debugprint) BlPrintf("RedoBlHooks: Patching call at %08x", i);
|
if (debugprint)
|
||||||
|
BlPrintf("RedoBlHooks: Patching call at %08x", i);
|
||||||
|
|
||||||
numpatched++;
|
numpatched++;
|
||||||
tsh_DeprotectAddress(i, len);
|
tsh_DeprotectAddress(i, len);
|
||||||
@@ -190,7 +182,6 @@ int tsh_PatchAllMatchesHex(ADDR len, char* text, char* replace, bool debugprint)
|
|||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Initialization
|
// Initialization
|
||||||
|
|
||||||
bool tsh_InitInternal() {
|
bool tsh_InitInternal() {
|
||||||
@@ -201,7 +192,4 @@ bool tsh_InitInternal(){
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool tsh_DeinitInternal() {
|
bool tsh_DeinitInternal() { return true; }
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
@@ -5,13 +5,11 @@
|
|||||||
#ifndef _H_BLHOOKS
|
#ifndef _H_BLHOOKS
|
||||||
#define _H_BLHOOKS
|
#define _H_BLHOOKS
|
||||||
|
|
||||||
|
|
||||||
// Typedefs
|
// Typedefs
|
||||||
|
|
||||||
typedef unsigned char BYTE;
|
typedef unsigned char BYTE;
|
||||||
typedef unsigned int ADDR;
|
typedef unsigned int ADDR;
|
||||||
|
|
||||||
|
|
||||||
// Prototypes
|
// Prototypes
|
||||||
|
|
||||||
bool tsh_InitInternal();
|
bool tsh_InitInternal();
|
||||||
@@ -25,7 +23,6 @@ void tsh_PatchByte(ADDR, BYTE);
|
|||||||
void tsh_PatchBytes(unsigned int, ADDR, BYTE*);
|
void tsh_PatchBytes(unsigned int, ADDR, BYTE*);
|
||||||
void tsh_PatchInt(ADDR, int);
|
void tsh_PatchInt(ADDR, int);
|
||||||
|
|
||||||
|
|
||||||
// Debug print settings
|
// Debug print settings
|
||||||
|
|
||||||
#ifndef TSH_NO_DEBUG_PRINT
|
#ifndef TSH_NO_DEBUG_PRINT
|
||||||
@@ -34,29 +31,31 @@ void tsh_PatchInt(ADDR, int);
|
|||||||
#define tsh_DEBUGPRINT true
|
#define tsh_DEBUGPRINT true
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
// Function short names
|
// Function short names
|
||||||
|
|
||||||
// Use in code when the def is not shared in a header
|
// Use in code when the def is not shared in a header
|
||||||
#define BlFunctionDef(returnType, convention, name, ...) \
|
#define BlFunctionDef(returnType, convention, name, ...) \
|
||||||
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
|
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
|
||||||
tsh_##name##FnT name;
|
tsh_##name##FnT name;
|
||||||
// Use in header for shared function defs when a BlFunctionDefIntern exists in code
|
// Use in header for shared function defs when a BlFunctionDefIntern exists in
|
||||||
|
// code
|
||||||
#define BlFunctionDefExtern(returnType, convention, name, ...) \
|
#define BlFunctionDefExtern(returnType, convention, name, ...) \
|
||||||
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
|
typedef returnType(convention* tsh_##name##FnT)(__VA_ARGS__); \
|
||||||
extern tsh_##name##FnT name;
|
extern tsh_##name##FnT name;
|
||||||
// Use in code for shared function defs when a BlFunctionDefExtern exists in header
|
// Use in code for shared function defs when a BlFunctionDefExtern exists in
|
||||||
#define BlFunctionDefIntern(name) \
|
// header
|
||||||
tsh_##name##FnT name;
|
#define BlFunctionDefIntern(name) tsh_##name##FnT name;
|
||||||
|
|
||||||
// Scan for and assign the pattern to the variable, or err and return if not found
|
// Scan for and assign the pattern to the variable, or err and return if not
|
||||||
|
// found
|
||||||
#define BlScanFunctionCode(target, patt, mask) \
|
#define BlScanFunctionCode(target, patt, mask) \
|
||||||
target = (tsh_##target##FnT)tsh_ScanCode((char*)patt, (char*)mask); \
|
target = (tsh_##target##FnT)tsh_ScanCode((char*)patt, (char*)mask); \
|
||||||
if (!target) { \
|
if (!target) { \
|
||||||
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
|
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
|
||||||
return false; \
|
return false; \
|
||||||
} else { \
|
} else { \
|
||||||
if(tsh_DEBUGPRINT) BlPrintf("RedoBlHooks | Found function "#target" at %08x", (int)target); \
|
if (tsh_DEBUGPRINT) \
|
||||||
|
BlPrintf("RedoBlHooks | Found function " #target " at %08x", (int)target); \
|
||||||
}
|
}
|
||||||
#define BlScanFunctionText(target, text) \
|
#define BlScanFunctionText(target, text) \
|
||||||
target = (tsh_##target##FnT)tsh_ScanText((char*)text); \
|
target = (tsh_##target##FnT)tsh_ScanText((char*)text); \
|
||||||
@@ -64,7 +63,8 @@ void tsh_PatchInt(ADDR, int);
|
|||||||
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
|
BlPrintf("RedoBlHooks | Cannot find function " #target "!"); \
|
||||||
return false; \
|
return false; \
|
||||||
} else { \
|
} else { \
|
||||||
if(tsh_DEBUGPRINT) BlPrintf("RedoBlHooks | Found function "#target" at %08x", (int)target); \
|
if (tsh_DEBUGPRINT) \
|
||||||
|
BlPrintf("RedoBlHooks | Found function " #target " at %08x", (int)target); \
|
||||||
}
|
}
|
||||||
#define BlScanCode(target, patt, mask) \
|
#define BlScanCode(target, patt, mask) \
|
||||||
target = tsh_ScanCode((char*)patt, (char*)mask); \
|
target = tsh_ScanCode((char*)patt, (char*)mask); \
|
||||||
@@ -72,7 +72,8 @@ void tsh_PatchInt(ADDR, int);
|
|||||||
BlPrintf("RedoBlHooks | Cannot find pattern " #target "!"); \
|
BlPrintf("RedoBlHooks | Cannot find pattern " #target "!"); \
|
||||||
return false; \
|
return false; \
|
||||||
} else { \
|
} else { \
|
||||||
if(tsh_DEBUGPRINT) BlPrintf("RedoBlHooks | Found "#target" at %08x", (int)target); \
|
if (tsh_DEBUGPRINT) \
|
||||||
|
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
|
||||||
}
|
}
|
||||||
#define BlScanText(target, text) \
|
#define BlScanText(target, text) \
|
||||||
target = tsh_ScanText((char*)text); \
|
target = tsh_ScanText((char*)text); \
|
||||||
@@ -80,7 +81,8 @@ void tsh_PatchInt(ADDR, int);
|
|||||||
BlPrintf("RedoBlHooks | Cannot find " #target "!"); \
|
BlPrintf("RedoBlHooks | Cannot find " #target "!"); \
|
||||||
return false; \
|
return false; \
|
||||||
} else { \
|
} else { \
|
||||||
if(tsh_DEBUGPRINT) BlPrintf("RedoBlHooks | Found "#target" at %08x", (int)target); \
|
if (tsh_DEBUGPRINT) \
|
||||||
|
BlPrintf("RedoBlHooks | Found " #target " at %08x", (int)target); \
|
||||||
}
|
}
|
||||||
|
|
||||||
// Use in code to define the data and functions for hooking a function
|
// Use in code to define the data and functions for hooking a function
|
||||||
@@ -100,23 +102,30 @@ void tsh_PatchInt(ADDR, int);
|
|||||||
tsh_PatchAllMatchesCode((ADDR)len, (char*)text, (char*)repl, tsh_DEBUGPRINT);
|
tsh_PatchAllMatchesCode((ADDR)len, (char*)text, (char*)repl, tsh_DEBUGPRINT);
|
||||||
|
|
||||||
// Deprotect and replace one byte
|
// Deprotect and replace one byte
|
||||||
#define BlPatchByte(addr, byte) \
|
#define BlPatchByte(addr, byte) tsh_PatchByte((ADDR)addr, (BYTE)byte);
|
||||||
tsh_PatchByte((ADDR)addr, (BYTE)byte);
|
|
||||||
// Deprotect and replace a byte string
|
// Deprotect and replace a byte string
|
||||||
#define BlPatchBytes(len, addr, repl) \
|
#define BlPatchBytes(len, addr, repl) tsh_PatchBytes((ADDR)len, (ADDR)addr, (BYTE*)repl);
|
||||||
tsh_PatchBytes((ADDR)len, (ADDR)addr, (BYTE*)repl);
|
|
||||||
|
|
||||||
// BlPrintf(char* format, ...)
|
// BlPrintf(char* format, ...)
|
||||||
#define BlPrintf(...) if(tsh_BlPrintf) { tsh_BlPrintf(__VA_ARGS__); }
|
#define BlPrintf(...) \
|
||||||
|
if (tsh_BlPrintf) { \
|
||||||
|
tsh_BlPrintf(__VA_ARGS__); \
|
||||||
|
}
|
||||||
// BlHooksInit() -> bool: success
|
// BlHooksInit() -> bool: success
|
||||||
#define BlHooksInit() if(!tsh_InitInternal()) { BlPrintf("BlHooksInit failed"); return false; }
|
#define BlHooksInit() \
|
||||||
|
if (!tsh_InitInternal()) { \
|
||||||
|
BlPrintf("BlHooksInit failed"); \
|
||||||
|
return false; \
|
||||||
|
}
|
||||||
// BlHooksDeinit() -> bool: success
|
// BlHooksDeinit() -> bool: success
|
||||||
#define BlHooksDeinit() if(!tsh_DeinitInternal()) { BlPrintf("BlHooksDeinit failed"); return false; }
|
#define BlHooksDeinit() \
|
||||||
|
if (!tsh_DeinitInternal()) { \
|
||||||
|
BlPrintf("BlHooksDeinit failed"); \
|
||||||
|
return false; \
|
||||||
|
}
|
||||||
|
|
||||||
// Scanned structures
|
// Scanned structures
|
||||||
|
|
||||||
BlFunctionDefExtern(void, , tsh_BlPrintf, const char*, ...);
|
BlFunctionDefExtern(void, , tsh_BlPrintf, const char*, ...);
|
||||||
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
BIN
lua5.1.dll
BIN
lua5.1.dll
Binary file not shown.
BIN
lualib/https.dll
BIN
lualib/https.dll
Binary file not shown.
292
lualib/ltn12.lua
292
lualib/ltn12.lua
@@ -1,292 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- LTN12 - Filters, sources, sinks and pumps.
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: ltn12.lua,v 1.31 2006/04/03 04:45:42 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local string = require("string")
|
|
||||||
local table = require("table")
|
|
||||||
local base = _G
|
|
||||||
module("ltn12")
|
|
||||||
|
|
||||||
filter = {}
|
|
||||||
source = {}
|
|
||||||
sink = {}
|
|
||||||
pump = {}
|
|
||||||
|
|
||||||
-- 2048 seems to be better in windows...
|
|
||||||
BLOCKSIZE = 2048
|
|
||||||
_VERSION = "LTN12 1.0.1"
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Filter stuff
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- returns a high level filter that cycles a low-level filter
|
|
||||||
function filter.cycle(low, ctx, extra)
|
|
||||||
base.assert(low)
|
|
||||||
return function(chunk)
|
|
||||||
local ret
|
|
||||||
ret, ctx = low(ctx, chunk, extra)
|
|
||||||
return ret
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- chains a bunch of filters together
|
|
||||||
-- (thanks to Wim Couwenberg)
|
|
||||||
function filter.chain(...)
|
|
||||||
local n = table.getn(arg)
|
|
||||||
local top, index = 1, 1
|
|
||||||
local retry = ""
|
|
||||||
return function(chunk)
|
|
||||||
retry = chunk and retry
|
|
||||||
while true do
|
|
||||||
if index == top then
|
|
||||||
chunk = arg[index](chunk)
|
|
||||||
if chunk == "" or top == n then return chunk
|
|
||||||
elseif chunk then index = index + 1
|
|
||||||
else
|
|
||||||
top = top+1
|
|
||||||
index = top
|
|
||||||
end
|
|
||||||
else
|
|
||||||
chunk = arg[index](chunk or "")
|
|
||||||
if chunk == "" then
|
|
||||||
index = index - 1
|
|
||||||
chunk = retry
|
|
||||||
elseif chunk then
|
|
||||||
if index == n then return chunk
|
|
||||||
else index = index + 1 end
|
|
||||||
else base.error("filter returned inappropriate nil") end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Source stuff
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- create an empty source
|
|
||||||
local function empty()
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
|
|
||||||
function source.empty()
|
|
||||||
return empty
|
|
||||||
end
|
|
||||||
|
|
||||||
-- returns a source that just outputs an error
|
|
||||||
function source.error(err)
|
|
||||||
return function()
|
|
||||||
return nil, err
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates a file source
|
|
||||||
function source.file(handle, io_err)
|
|
||||||
if handle then
|
|
||||||
return function()
|
|
||||||
local chunk = handle:read(BLOCKSIZE)
|
|
||||||
if not chunk then handle:close() end
|
|
||||||
return chunk
|
|
||||||
end
|
|
||||||
else return source.error(io_err or "unable to open file") end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- turns a fancy source into a simple source
|
|
||||||
function source.simplify(src)
|
|
||||||
base.assert(src)
|
|
||||||
return function()
|
|
||||||
local chunk, err_or_new = src()
|
|
||||||
src = err_or_new or src
|
|
||||||
if not chunk then return nil, err_or_new
|
|
||||||
else return chunk end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates string source
|
|
||||||
function source.string(s)
|
|
||||||
if s then
|
|
||||||
local i = 1
|
|
||||||
return function()
|
|
||||||
local chunk = string.sub(s, i, i+BLOCKSIZE-1)
|
|
||||||
i = i + BLOCKSIZE
|
|
||||||
if chunk ~= "" then return chunk
|
|
||||||
else return nil end
|
|
||||||
end
|
|
||||||
else return source.empty() end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates rewindable source
|
|
||||||
function source.rewind(src)
|
|
||||||
base.assert(src)
|
|
||||||
local t = {}
|
|
||||||
return function(chunk)
|
|
||||||
if not chunk then
|
|
||||||
chunk = table.remove(t)
|
|
||||||
if not chunk then return src()
|
|
||||||
else return chunk end
|
|
||||||
else
|
|
||||||
table.insert(t, chunk)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function source.chain(src, f)
|
|
||||||
base.assert(src and f)
|
|
||||||
local last_in, last_out = "", ""
|
|
||||||
local state = "feeding"
|
|
||||||
local err
|
|
||||||
return function()
|
|
||||||
if not last_out then
|
|
||||||
base.error('source is empty!', 2)
|
|
||||||
end
|
|
||||||
while true do
|
|
||||||
if state == "feeding" then
|
|
||||||
last_in, err = src()
|
|
||||||
if err then return nil, err end
|
|
||||||
last_out = f(last_in)
|
|
||||||
if not last_out then
|
|
||||||
if last_in then
|
|
||||||
base.error('filter returned inappropriate nil')
|
|
||||||
else
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
elseif last_out ~= "" then
|
|
||||||
state = "eating"
|
|
||||||
if last_in then last_in = "" end
|
|
||||||
return last_out
|
|
||||||
end
|
|
||||||
else
|
|
||||||
last_out = f(last_in)
|
|
||||||
if last_out == "" then
|
|
||||||
if last_in == "" then
|
|
||||||
state = "feeding"
|
|
||||||
else
|
|
||||||
base.error('filter returned ""')
|
|
||||||
end
|
|
||||||
elseif not last_out then
|
|
||||||
if last_in then
|
|
||||||
base.error('filter returned inappropriate nil')
|
|
||||||
else
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
else
|
|
||||||
return last_out
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates a source that produces contents of several sources, one after the
|
|
||||||
-- other, as if they were concatenated
|
|
||||||
-- (thanks to Wim Couwenberg)
|
|
||||||
function source.cat(...)
|
|
||||||
local src = table.remove(arg, 1)
|
|
||||||
return function()
|
|
||||||
while src do
|
|
||||||
local chunk, err = src()
|
|
||||||
if chunk then return chunk end
|
|
||||||
if err then return nil, err end
|
|
||||||
src = table.remove(arg, 1)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Sink stuff
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- creates a sink that stores into a table
|
|
||||||
function sink.table(t)
|
|
||||||
t = t or {}
|
|
||||||
local f = function(chunk, err)
|
|
||||||
if chunk then table.insert(t, chunk) end
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
return f, t
|
|
||||||
end
|
|
||||||
|
|
||||||
-- turns a fancy sink into a simple sink
|
|
||||||
function sink.simplify(snk)
|
|
||||||
base.assert(snk)
|
|
||||||
return function(chunk, err)
|
|
||||||
local ret, err_or_new = snk(chunk, err)
|
|
||||||
if not ret then return nil, err_or_new end
|
|
||||||
snk = err_or_new or snk
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates a file sink
|
|
||||||
function sink.file(handle, io_err)
|
|
||||||
if handle then
|
|
||||||
return function(chunk, err)
|
|
||||||
if not chunk then
|
|
||||||
handle:close()
|
|
||||||
return 1
|
|
||||||
else return handle:write(chunk) end
|
|
||||||
end
|
|
||||||
else return sink.error(io_err or "unable to open file") end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates a sink that discards data
|
|
||||||
local function null()
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function sink.null()
|
|
||||||
return null
|
|
||||||
end
|
|
||||||
|
|
||||||
-- creates a sink that just returns an error
|
|
||||||
function sink.error(err)
|
|
||||||
return function()
|
|
||||||
return nil, err
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- chains a sink with a filter
|
|
||||||
function sink.chain(f, snk)
|
|
||||||
base.assert(f and snk)
|
|
||||||
return function(chunk, err)
|
|
||||||
if chunk ~= "" then
|
|
||||||
local filtered = f(chunk)
|
|
||||||
local done = chunk and ""
|
|
||||||
while true do
|
|
||||||
local ret, snkerr = snk(filtered, err)
|
|
||||||
if not ret then return nil, snkerr end
|
|
||||||
if filtered == done then return 1 end
|
|
||||||
filtered = f(done)
|
|
||||||
end
|
|
||||||
else return 1 end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Pump stuff
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- pumps one chunk from the source to the sink
|
|
||||||
function pump.step(src, snk)
|
|
||||||
local chunk, src_err = src()
|
|
||||||
local ret, snk_err = snk(chunk, src_err)
|
|
||||||
if chunk and ret then return 1
|
|
||||||
else return nil, src_err or snk_err end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- pumps all data from a source to a sink, using a step function
|
|
||||||
function pump.all(src, snk, step)
|
|
||||||
base.assert(src and snk)
|
|
||||||
step = step or pump.step
|
|
||||||
while true do
|
|
||||||
local ret, err = step(src, snk)
|
|
||||||
if not ret then
|
|
||||||
if err then return nil, err
|
|
||||||
else return 1 end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
@@ -1,87 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- MIME support for the Lua language.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- Conforming to RFCs 2045-2049
|
|
||||||
-- RCS ID: $Id: mime.lua,v 1.29 2007/06/11 23:44:54 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local base = _G
|
|
||||||
local ltn12 = require("ltn12")
|
|
||||||
local mime = require("mime.core")
|
|
||||||
--local io = require("io")
|
|
||||||
local string = require("string")
|
|
||||||
module("mime")
|
|
||||||
|
|
||||||
-- encode, decode and wrap algorithm tables
|
|
||||||
encodet = {}
|
|
||||||
decodet = {}
|
|
||||||
wrapt = {}
|
|
||||||
|
|
||||||
-- creates a function that chooses a filter by name from a given table
|
|
||||||
local function choose(table)
|
|
||||||
return function(name, opt1, opt2)
|
|
||||||
if base.type(name) ~= "string" then
|
|
||||||
name, opt1, opt2 = "default", name, opt1
|
|
||||||
end
|
|
||||||
local f = table[name or "nil"]
|
|
||||||
if not f then
|
|
||||||
base.error("unknown key (" .. base.tostring(name) .. ")", 3)
|
|
||||||
else return f(opt1, opt2) end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- define the encoding filters
|
|
||||||
encodet['base64'] = function()
|
|
||||||
return ltn12.filter.cycle(b64, "")
|
|
||||||
end
|
|
||||||
|
|
||||||
encodet['quoted-printable'] = function(mode)
|
|
||||||
return ltn12.filter.cycle(qp, "",
|
|
||||||
(mode == "binary") and "=0D=0A" or "\r\n")
|
|
||||||
end
|
|
||||||
|
|
||||||
-- define the decoding filters
|
|
||||||
decodet['base64'] = function()
|
|
||||||
return ltn12.filter.cycle(unb64, "")
|
|
||||||
end
|
|
||||||
|
|
||||||
decodet['quoted-printable'] = function()
|
|
||||||
return ltn12.filter.cycle(unqp, "")
|
|
||||||
end
|
|
||||||
|
|
||||||
local function format(chunk)
|
|
||||||
if chunk then
|
|
||||||
if chunk == "" then return "''"
|
|
||||||
else return string.len(chunk) end
|
|
||||||
else return "nil" end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- define the line-wrap filters
|
|
||||||
wrapt['text'] = function(length)
|
|
||||||
length = length or 76
|
|
||||||
return ltn12.filter.cycle(wrp, length, length)
|
|
||||||
end
|
|
||||||
wrapt['base64'] = wrapt['text']
|
|
||||||
wrapt['default'] = wrapt['text']
|
|
||||||
|
|
||||||
wrapt['quoted-printable'] = function()
|
|
||||||
return ltn12.filter.cycle(qpwrp, 76, 76)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- function that choose the encoding, decoding or wrap algorithm
|
|
||||||
encode = choose(encodet)
|
|
||||||
decode = choose(decodet)
|
|
||||||
wrap = choose(wrapt)
|
|
||||||
|
|
||||||
-- define the end-of-line normalization filter
|
|
||||||
function normalize(marker)
|
|
||||||
return ltn12.filter.cycle(eol, 0, marker)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- high level stuffing filter
|
|
||||||
function stuff()
|
|
||||||
return ltn12.filter.cycle(dot, 2)
|
|
||||||
end
|
|
||||||
Binary file not shown.
@@ -1,133 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- LuaSocket helper module
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: socket.lua,v 1.22 2005/11/22 08:33:29 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local base = _G
|
|
||||||
local string = require("string")
|
|
||||||
local math = require("math")
|
|
||||||
local socket = require("socket.core")
|
|
||||||
module("socket")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Exported auxiliar functions
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function connect(address, port, laddress, lport)
|
|
||||||
local sock, err = socket.tcp()
|
|
||||||
if not sock then return nil, err end
|
|
||||||
if laddress then
|
|
||||||
local res, err = sock:bind(laddress, lport, -1)
|
|
||||||
if not res then return nil, err end
|
|
||||||
end
|
|
||||||
local res, err = sock:connect(address, port)
|
|
||||||
if not res then return nil, err end
|
|
||||||
return sock
|
|
||||||
end
|
|
||||||
|
|
||||||
function bind(host, port, backlog)
|
|
||||||
local sock, err = socket.tcp()
|
|
||||||
if not sock then return nil, err end
|
|
||||||
sock:setoption("reuseaddr", true)
|
|
||||||
local res, err = sock:bind(host, port)
|
|
||||||
if not res then return nil, err end
|
|
||||||
res, err = sock:listen(backlog)
|
|
||||||
if not res then return nil, err end
|
|
||||||
return sock
|
|
||||||
end
|
|
||||||
|
|
||||||
try = newtry()
|
|
||||||
|
|
||||||
function choose(table)
|
|
||||||
return function(name, opt1, opt2)
|
|
||||||
if base.type(name) ~= "string" then
|
|
||||||
name, opt1, opt2 = "default", name, opt1
|
|
||||||
end
|
|
||||||
local f = table[name or "nil"]
|
|
||||||
if not f then base.error("unknown key (".. base.tostring(name) ..")", 3)
|
|
||||||
else return f(opt1, opt2) end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Socket sources and sinks, conforming to LTN12
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- create namespaces inside LuaSocket namespace
|
|
||||||
sourcet = {}
|
|
||||||
sinkt = {}
|
|
||||||
|
|
||||||
BLOCKSIZE = 2048
|
|
||||||
|
|
||||||
sinkt["close-when-done"] = function(sock)
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function(self, chunk, err)
|
|
||||||
if not chunk then
|
|
||||||
sock:close()
|
|
||||||
return 1
|
|
||||||
else return sock:send(chunk) end
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
sinkt["keep-open"] = function(sock)
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function(self, chunk, err)
|
|
||||||
if chunk then return sock:send(chunk)
|
|
||||||
else return 1 end
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
sinkt["default"] = sinkt["keep-open"]
|
|
||||||
|
|
||||||
sink = choose(sinkt)
|
|
||||||
|
|
||||||
sourcet["by-length"] = function(sock, length)
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function()
|
|
||||||
if length <= 0 then return nil end
|
|
||||||
local size = math.min(socket.BLOCKSIZE, length)
|
|
||||||
local chunk, err = sock:receive(size)
|
|
||||||
if err then return nil, err end
|
|
||||||
length = length - string.len(chunk)
|
|
||||||
return chunk
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
sourcet["until-closed"] = function(sock)
|
|
||||||
local done
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function()
|
|
||||||
if done then return nil end
|
|
||||||
local chunk, err, partial = sock:receive(socket.BLOCKSIZE)
|
|
||||||
if not err then return chunk
|
|
||||||
elseif err == "closed" then
|
|
||||||
sock:close()
|
|
||||||
done = 1
|
|
||||||
return partial
|
|
||||||
else return nil, err end
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
sourcet["default"] = sourcet["until-closed"]
|
|
||||||
|
|
||||||
source = choose(sourcet)
|
|
||||||
|
|
||||||
Binary file not shown.
@@ -1,281 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- FTP support for the Lua language
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: ftp.lua,v 1.45 2007/07/11 19:25:47 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local base = _G
|
|
||||||
local table = require("table")
|
|
||||||
local string = require("string")
|
|
||||||
local math = require("math")
|
|
||||||
local socket = require("socket")
|
|
||||||
local url = require("socket.url")
|
|
||||||
local tp = require("socket.tp")
|
|
||||||
local ltn12 = require("ltn12")
|
|
||||||
module("socket.ftp")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Program constants
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- timeout in seconds before the program gives up on a connection
|
|
||||||
TIMEOUT = 60
|
|
||||||
-- default port for ftp service
|
|
||||||
PORT = 21
|
|
||||||
-- this is the default anonymous password. used when no password is
|
|
||||||
-- provided in url. should be changed to your e-mail.
|
|
||||||
USER = "ftp"
|
|
||||||
PASSWORD = "anonymous@anonymous.org"
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Low level FTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local metat = { __index = {} }
|
|
||||||
|
|
||||||
function open(server, port, create)
|
|
||||||
local tp = socket.try(tp.connect(server, port or PORT, TIMEOUT, create))
|
|
||||||
local f = base.setmetatable({ tp = tp }, metat)
|
|
||||||
-- make sure everything gets closed in an exception
|
|
||||||
f.try = socket.newtry(function() f:close() end)
|
|
||||||
return f
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:portconnect()
|
|
||||||
self.try(self.server:settimeout(TIMEOUT))
|
|
||||||
self.data = self.try(self.server:accept())
|
|
||||||
self.try(self.data:settimeout(TIMEOUT))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:pasvconnect()
|
|
||||||
self.data = self.try(socket.tcp())
|
|
||||||
self.try(self.data:settimeout(TIMEOUT))
|
|
||||||
self.try(self.data:connect(self.pasvt.ip, self.pasvt.port))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:login(user, password)
|
|
||||||
self.try(self.tp:command("user", user or USER))
|
|
||||||
local code, reply = self.try(self.tp:check{"2..", 331})
|
|
||||||
if code == 331 then
|
|
||||||
self.try(self.tp:command("pass", password or PASSWORD))
|
|
||||||
self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:pasv()
|
|
||||||
self.try(self.tp:command("pasv"))
|
|
||||||
local code, reply = self.try(self.tp:check("2.."))
|
|
||||||
local pattern = "(%d+)%D(%d+)%D(%d+)%D(%d+)%D(%d+)%D(%d+)"
|
|
||||||
local a, b, c, d, p1, p2 = socket.skip(2, string.find(reply, pattern))
|
|
||||||
self.try(a and b and c and d and p1 and p2, reply)
|
|
||||||
self.pasvt = {
|
|
||||||
ip = string.format("%d.%d.%d.%d", a, b, c, d),
|
|
||||||
port = p1*256 + p2
|
|
||||||
}
|
|
||||||
if self.server then
|
|
||||||
self.server:close()
|
|
||||||
self.server = nil
|
|
||||||
end
|
|
||||||
return self.pasvt.ip, self.pasvt.port
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:port(ip, port)
|
|
||||||
self.pasvt = nil
|
|
||||||
if not ip then
|
|
||||||
ip, port = self.try(self.tp:getcontrol():getsockname())
|
|
||||||
self.server = self.try(socket.bind(ip, 0))
|
|
||||||
ip, port = self.try(self.server:getsockname())
|
|
||||||
self.try(self.server:settimeout(TIMEOUT))
|
|
||||||
end
|
|
||||||
local pl = math.mod(port, 256)
|
|
||||||
local ph = (port - pl)/256
|
|
||||||
local arg = string.gsub(string.format("%s,%d,%d", ip, ph, pl), "%.", ",")
|
|
||||||
self.try(self.tp:command("port", arg))
|
|
||||||
self.try(self.tp:check("2.."))
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:send(sendt)
|
|
||||||
self.try(self.pasvt or self.server, "need port or pasv first")
|
|
||||||
-- if there is a pasvt table, we already sent a PASV command
|
|
||||||
-- we just get the data connection into self.data
|
|
||||||
if self.pasvt then self:pasvconnect() end
|
|
||||||
-- get the transfer argument and command
|
|
||||||
local argument = sendt.argument or
|
|
||||||
url.unescape(string.gsub(sendt.path or "", "^[/\\]", ""))
|
|
||||||
if argument == "" then argument = nil end
|
|
||||||
local command = sendt.command or "stor"
|
|
||||||
-- send the transfer command and check the reply
|
|
||||||
self.try(self.tp:command(command, argument))
|
|
||||||
local code, reply = self.try(self.tp:check{"2..", "1.."})
|
|
||||||
-- if there is not a a pasvt table, then there is a server
|
|
||||||
-- and we already sent a PORT command
|
|
||||||
if not self.pasvt then self:portconnect() end
|
|
||||||
-- get the sink, source and step for the transfer
|
|
||||||
local step = sendt.step or ltn12.pump.step
|
|
||||||
local readt = {self.tp.c}
|
|
||||||
local checkstep = function(src, snk)
|
|
||||||
-- check status in control connection while downloading
|
|
||||||
local readyt = socket.select(readt, nil, 0)
|
|
||||||
if readyt[tp] then code = self.try(self.tp:check("2..")) end
|
|
||||||
return step(src, snk)
|
|
||||||
end
|
|
||||||
local sink = socket.sink("close-when-done", self.data)
|
|
||||||
-- transfer all data and check error
|
|
||||||
self.try(ltn12.pump.all(sendt.source, sink, checkstep))
|
|
||||||
if string.find(code, "1..") then self.try(self.tp:check("2..")) end
|
|
||||||
-- done with data connection
|
|
||||||
self.data:close()
|
|
||||||
-- find out how many bytes were sent
|
|
||||||
local sent = socket.skip(1, self.data:getstats())
|
|
||||||
self.data = nil
|
|
||||||
return sent
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receive(recvt)
|
|
||||||
self.try(self.pasvt or self.server, "need port or pasv first")
|
|
||||||
if self.pasvt then self:pasvconnect() end
|
|
||||||
local argument = recvt.argument or
|
|
||||||
url.unescape(string.gsub(recvt.path or "", "^[/\\]", ""))
|
|
||||||
if argument == "" then argument = nil end
|
|
||||||
local command = recvt.command or "retr"
|
|
||||||
self.try(self.tp:command(command, argument))
|
|
||||||
local code = self.try(self.tp:check{"1..", "2.."})
|
|
||||||
if not self.pasvt then self:portconnect() end
|
|
||||||
local source = socket.source("until-closed", self.data)
|
|
||||||
local step = recvt.step or ltn12.pump.step
|
|
||||||
self.try(ltn12.pump.all(source, recvt.sink, step))
|
|
||||||
if string.find(code, "1..") then self.try(self.tp:check("2..")) end
|
|
||||||
self.data:close()
|
|
||||||
self.data = nil
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:cwd(dir)
|
|
||||||
self.try(self.tp:command("cwd", dir))
|
|
||||||
self.try(self.tp:check(250))
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:type(type)
|
|
||||||
self.try(self.tp:command("type", type))
|
|
||||||
self.try(self.tp:check(200))
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:greet()
|
|
||||||
local code = self.try(self.tp:check{"1..", "2.."})
|
|
||||||
if string.find(code, "1..") then self.try(self.tp:check("2..")) end
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:quit()
|
|
||||||
self.try(self.tp:command("quit"))
|
|
||||||
self.try(self.tp:check("2.."))
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:close()
|
|
||||||
if self.data then self.data:close() end
|
|
||||||
if self.server then self.server:close() end
|
|
||||||
return self.tp:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- High level FTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local function override(t)
|
|
||||||
if t.url then
|
|
||||||
local u = url.parse(t.url)
|
|
||||||
for i,v in base.pairs(t) do
|
|
||||||
u[i] = v
|
|
||||||
end
|
|
||||||
return u
|
|
||||||
else return t end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function tput(putt)
|
|
||||||
putt = override(putt)
|
|
||||||
socket.try(putt.host, "missing hostname")
|
|
||||||
local f = open(putt.host, putt.port, putt.create)
|
|
||||||
f:greet()
|
|
||||||
f:login(putt.user, putt.password)
|
|
||||||
if putt.type then f:type(putt.type) end
|
|
||||||
f:pasv()
|
|
||||||
local sent = f:send(putt)
|
|
||||||
f:quit()
|
|
||||||
f:close()
|
|
||||||
return sent
|
|
||||||
end
|
|
||||||
|
|
||||||
local default = {
|
|
||||||
path = "/",
|
|
||||||
scheme = "ftp"
|
|
||||||
}
|
|
||||||
|
|
||||||
local function parse(u)
|
|
||||||
local t = socket.try(url.parse(u, default))
|
|
||||||
socket.try(t.scheme == "ftp", "wrong scheme '" .. t.scheme .. "'")
|
|
||||||
socket.try(t.host, "missing hostname")
|
|
||||||
local pat = "^type=(.)$"
|
|
||||||
if t.params then
|
|
||||||
t.type = socket.skip(2, string.find(t.params, pat))
|
|
||||||
socket.try(t.type == "a" or t.type == "i",
|
|
||||||
"invalid type '" .. t.type .. "'")
|
|
||||||
end
|
|
||||||
return t
|
|
||||||
end
|
|
||||||
|
|
||||||
local function sput(u, body)
|
|
||||||
local putt = parse(u)
|
|
||||||
putt.source = ltn12.source.string(body)
|
|
||||||
return tput(putt)
|
|
||||||
end
|
|
||||||
|
|
||||||
put = socket.protect(function(putt, body)
|
|
||||||
if base.type(putt) == "string" then return sput(putt, body)
|
|
||||||
else return tput(putt) end
|
|
||||||
end)
|
|
||||||
|
|
||||||
local function tget(gett)
|
|
||||||
gett = override(gett)
|
|
||||||
socket.try(gett.host, "missing hostname")
|
|
||||||
local f = open(gett.host, gett.port, gett.create)
|
|
||||||
f:greet()
|
|
||||||
f:login(gett.user, gett.password)
|
|
||||||
if gett.type then f:type(gett.type) end
|
|
||||||
f:pasv()
|
|
||||||
f:receive(gett)
|
|
||||||
f:quit()
|
|
||||||
return f:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
local function sget(u)
|
|
||||||
local gett = parse(u)
|
|
||||||
local t = {}
|
|
||||||
gett.sink = ltn12.sink.table(t)
|
|
||||||
tget(gett)
|
|
||||||
return table.concat(t)
|
|
||||||
end
|
|
||||||
|
|
||||||
command = socket.protect(function(cmdt)
|
|
||||||
cmdt = override(cmdt)
|
|
||||||
socket.try(cmdt.host, "missing hostname")
|
|
||||||
socket.try(cmdt.command, "missing command")
|
|
||||||
local f = open(cmdt.host, cmdt.port, cmdt.create)
|
|
||||||
f:greet()
|
|
||||||
f:login(cmdt.user, cmdt.password)
|
|
||||||
f.try(f.tp:command(cmdt.command, cmdt.argument))
|
|
||||||
if cmdt.check then f.try(f.tp:check(cmdt.check)) end
|
|
||||||
f:quit()
|
|
||||||
return f:close()
|
|
||||||
end)
|
|
||||||
|
|
||||||
get = socket.protect(function(gett)
|
|
||||||
if base.type(gett) == "string" then return sget(gett)
|
|
||||||
else return tget(gett) end
|
|
||||||
end)
|
|
||||||
|
|
||||||
@@ -1,350 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- HTTP/1.1 client support for the Lua language.
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: http.lua,v 1.70 2007/03/12 04:08:40 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-------------------------------------------------------------------------------
|
|
||||||
local socket = require("socket")
|
|
||||||
local url = require("socket.url")
|
|
||||||
local ltn12 = require("ltn12")
|
|
||||||
local mime = require("mime")
|
|
||||||
local string = require("string")
|
|
||||||
local base = _G
|
|
||||||
local table = require("table")
|
|
||||||
module("socket.http")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Program constants
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- connection timeout in seconds
|
|
||||||
TIMEOUT = 60
|
|
||||||
-- default port for document retrieval
|
|
||||||
PORT = 80
|
|
||||||
-- user agent field sent in request
|
|
||||||
USERAGENT = socket._VERSION
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Reads MIME headers from a connection, unfolding where needed
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local function receiveheaders(sock, headers)
|
|
||||||
local line, name, value, err
|
|
||||||
headers = headers or {}
|
|
||||||
-- get first line
|
|
||||||
line, err = sock:receive()
|
|
||||||
if err then return nil, err end
|
|
||||||
-- headers go until a blank line is found
|
|
||||||
while line ~= "" do
|
|
||||||
-- get field-name and value
|
|
||||||
name, value = socket.skip(2, string.find(line, "^(.-):%s*(.*)"))
|
|
||||||
if not (name and value) then return nil, "malformed reponse headers" end
|
|
||||||
name = string.lower(name)
|
|
||||||
-- get next line (value might be folded)
|
|
||||||
line, err = sock:receive()
|
|
||||||
if err then return nil, err end
|
|
||||||
-- unfold any folded values
|
|
||||||
while string.find(line, "^%s") do
|
|
||||||
value = value .. line
|
|
||||||
line = sock:receive()
|
|
||||||
if err then return nil, err end
|
|
||||||
end
|
|
||||||
-- save pair in table
|
|
||||||
if headers[name] then headers[name] = headers[name] .. ", " .. value
|
|
||||||
else headers[name] = value end
|
|
||||||
end
|
|
||||||
return headers
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Extra sources and sinks
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
socket.sourcet["http-chunked"] = function(sock, headers)
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function()
|
|
||||||
-- get chunk size, skip extention
|
|
||||||
local line, err = sock:receive()
|
|
||||||
if err then return nil, err end
|
|
||||||
local size = base.tonumber(string.gsub(line, ";.*", ""), 16)
|
|
||||||
if not size then return nil, "invalid chunk size" end
|
|
||||||
-- was it the last chunk?
|
|
||||||
if size > 0 then
|
|
||||||
-- if not, get chunk and skip terminating CRLF
|
|
||||||
local chunk, err, part = sock:receive(size)
|
|
||||||
if chunk then sock:receive() end
|
|
||||||
return chunk, err
|
|
||||||
else
|
|
||||||
-- if it was, read trailers into headers table
|
|
||||||
headers, err = receiveheaders(sock, headers)
|
|
||||||
if not headers then return nil, err end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
socket.sinkt["http-chunked"] = function(sock)
|
|
||||||
return base.setmetatable({
|
|
||||||
getfd = function() return sock:getfd() end,
|
|
||||||
dirty = function() return sock:dirty() end
|
|
||||||
}, {
|
|
||||||
__call = function(self, chunk, err)
|
|
||||||
if not chunk then return sock:send("0\r\n\r\n") end
|
|
||||||
local size = string.format("%X\r\n", string.len(chunk))
|
|
||||||
return sock:send(size .. chunk .. "\r\n")
|
|
||||||
end
|
|
||||||
})
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Low level HTTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local metat = { __index = {} }
|
|
||||||
|
|
||||||
function open(host, port, create)
|
|
||||||
-- create socket with user connect function, or with default
|
|
||||||
local c = socket.try((create or socket.tcp)())
|
|
||||||
local h = base.setmetatable({ c = c }, metat)
|
|
||||||
-- create finalized try
|
|
||||||
h.try = socket.newtry(function() h:close() end)
|
|
||||||
-- set timeout before connecting
|
|
||||||
h.try(c:settimeout(TIMEOUT))
|
|
||||||
h.try(c:connect(host, port or PORT))
|
|
||||||
-- here everything worked
|
|
||||||
return h
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:sendrequestline(method, uri)
|
|
||||||
local reqline = string.format("%s %s HTTP/1.1\r\n", method or "GET", uri)
|
|
||||||
return self.try(self.c:send(reqline))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:sendheaders(headers)
|
|
||||||
local h = "\r\n"
|
|
||||||
for i, v in base.pairs(headers) do
|
|
||||||
h = i .. ": " .. v .. "\r\n" .. h
|
|
||||||
end
|
|
||||||
self.try(self.c:send(h))
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:sendbody(headers, source, step)
|
|
||||||
source = source or ltn12.source.empty()
|
|
||||||
step = step or ltn12.pump.step
|
|
||||||
-- if we don't know the size in advance, send chunked and hope for the best
|
|
||||||
local mode = "http-chunked"
|
|
||||||
if headers["content-length"] then mode = "keep-open" end
|
|
||||||
return self.try(ltn12.pump.all(source, socket.sink(mode, self.c), step))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receivestatusline()
|
|
||||||
local status = self.try(self.c:receive(5))
|
|
||||||
-- identify HTTP/0.9 responses, which do not contain a status line
|
|
||||||
-- this is just a heuristic, but is what the RFC recommends
|
|
||||||
if status ~= "HTTP/" then return nil, status end
|
|
||||||
-- otherwise proceed reading a status line
|
|
||||||
status = self.try(self.c:receive("*l", status))
|
|
||||||
local code = socket.skip(2, string.find(status, "HTTP/%d*%.%d* (%d%d%d)"))
|
|
||||||
return self.try(base.tonumber(code), status)
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receiveheaders()
|
|
||||||
return self.try(receiveheaders(self.c))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receivebody(headers, sink, step)
|
|
||||||
sink = sink or ltn12.sink.null()
|
|
||||||
step = step or ltn12.pump.step
|
|
||||||
local length = base.tonumber(headers["content-length"])
|
|
||||||
local t = headers["transfer-encoding"] -- shortcut
|
|
||||||
local mode = "default" -- connection close
|
|
||||||
if t and t ~= "identity" then mode = "http-chunked"
|
|
||||||
elseif base.tonumber(headers["content-length"]) then mode = "by-length" end
|
|
||||||
return self.try(ltn12.pump.all(socket.source(mode, self.c, length),
|
|
||||||
sink, step))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receive09body(status, sink, step)
|
|
||||||
local source = ltn12.source.rewind(socket.source("until-closed", self.c))
|
|
||||||
source(status)
|
|
||||||
return self.try(ltn12.pump.all(source, sink, step))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:close()
|
|
||||||
return self.c:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- High level HTTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local function adjusturi(reqt)
|
|
||||||
local u = reqt
|
|
||||||
-- if there is a proxy, we need the full url. otherwise, just a part.
|
|
||||||
if not reqt.proxy and not PROXY then
|
|
||||||
u = {
|
|
||||||
path = socket.try(reqt.path, "invalid path 'nil'"),
|
|
||||||
params = reqt.params,
|
|
||||||
query = reqt.query,
|
|
||||||
fragment = reqt.fragment
|
|
||||||
}
|
|
||||||
end
|
|
||||||
return url.build(u)
|
|
||||||
end
|
|
||||||
|
|
||||||
local function adjustproxy(reqt)
|
|
||||||
local proxy = reqt.proxy or PROXY
|
|
||||||
if proxy then
|
|
||||||
proxy = url.parse(proxy)
|
|
||||||
return proxy.host, proxy.port or 3128
|
|
||||||
else
|
|
||||||
return reqt.host, reqt.port
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function adjustheaders(reqt)
|
|
||||||
-- default headers
|
|
||||||
local lower = {
|
|
||||||
["user-agent"] = USERAGENT,
|
|
||||||
["host"] = reqt.host,
|
|
||||||
["connection"] = "close, TE",
|
|
||||||
["te"] = "trailers"
|
|
||||||
}
|
|
||||||
-- if we have authentication information, pass it along
|
|
||||||
if reqt.user and reqt.password then
|
|
||||||
lower["authorization"] =
|
|
||||||
"Basic " .. (mime.b64(reqt.user .. ":" .. reqt.password))
|
|
||||||
end
|
|
||||||
-- override with user headers
|
|
||||||
for i,v in base.pairs(reqt.headers or lower) do
|
|
||||||
lower[string.lower(i)] = v
|
|
||||||
end
|
|
||||||
return lower
|
|
||||||
end
|
|
||||||
|
|
||||||
-- default url parts
|
|
||||||
local default = {
|
|
||||||
host = "",
|
|
||||||
port = PORT,
|
|
||||||
path ="/",
|
|
||||||
scheme = "http"
|
|
||||||
}
|
|
||||||
|
|
||||||
local function adjustrequest(reqt)
|
|
||||||
-- parse url if provided
|
|
||||||
local nreqt = reqt.url and url.parse(reqt.url, default) or {}
|
|
||||||
-- explicit components override url
|
|
||||||
for i,v in base.pairs(reqt) do nreqt[i] = v end
|
|
||||||
if nreqt.port == "" then nreqt.port = 80 end
|
|
||||||
socket.try(nreqt.host and nreqt.host ~= "",
|
|
||||||
"invalid host '" .. base.tostring(nreqt.host) .. "'")
|
|
||||||
-- compute uri if user hasn't overriden
|
|
||||||
nreqt.uri = reqt.uri or adjusturi(nreqt)
|
|
||||||
-- ajust host and port if there is a proxy
|
|
||||||
nreqt.host, nreqt.port = adjustproxy(nreqt)
|
|
||||||
-- adjust headers in request
|
|
||||||
nreqt.headers = adjustheaders(nreqt)
|
|
||||||
return nreqt
|
|
||||||
end
|
|
||||||
|
|
||||||
local function shouldredirect(reqt, code, headers)
|
|
||||||
return headers.location and
|
|
||||||
string.gsub(headers.location, "%s", "") ~= "" and
|
|
||||||
(reqt.redirect ~= false) and
|
|
||||||
(code == 301 or code == 302) and
|
|
||||||
(not reqt.method or reqt.method == "GET" or reqt.method == "HEAD")
|
|
||||||
and (not reqt.nredirects or reqt.nredirects < 5)
|
|
||||||
end
|
|
||||||
|
|
||||||
local function shouldreceivebody(reqt, code)
|
|
||||||
if reqt.method == "HEAD" then return nil end
|
|
||||||
if code == 204 or code == 304 then return nil end
|
|
||||||
if code >= 100 and code < 200 then return nil end
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
-- forward declarations
|
|
||||||
local trequest, tredirect
|
|
||||||
|
|
||||||
function tredirect(reqt, location)
|
|
||||||
local result, code, headers, status = trequest {
|
|
||||||
-- the RFC says the redirect URL has to be absolute, but some
|
|
||||||
-- servers do not respect that
|
|
||||||
url = url.absolute(reqt.url, location),
|
|
||||||
source = reqt.source,
|
|
||||||
sink = reqt.sink,
|
|
||||||
headers = reqt.headers,
|
|
||||||
proxy = reqt.proxy,
|
|
||||||
nredirects = (reqt.nredirects or 0) + 1,
|
|
||||||
create = reqt.create
|
|
||||||
}
|
|
||||||
-- pass location header back as a hint we redirected
|
|
||||||
headers = headers or {}
|
|
||||||
headers.location = headers.location or location
|
|
||||||
return result, code, headers, status
|
|
||||||
end
|
|
||||||
|
|
||||||
function trequest(reqt)
|
|
||||||
-- we loop until we get what we want, or
|
|
||||||
-- until we are sure there is no way to get it
|
|
||||||
local nreqt = adjustrequest(reqt)
|
|
||||||
local h = open(nreqt.host, nreqt.port, nreqt.create)
|
|
||||||
-- send request line and headers
|
|
||||||
h:sendrequestline(nreqt.method, nreqt.uri)
|
|
||||||
h:sendheaders(nreqt.headers)
|
|
||||||
-- if there is a body, send it
|
|
||||||
if nreqt.source then
|
|
||||||
h:sendbody(nreqt.headers, nreqt.source, nreqt.step)
|
|
||||||
end
|
|
||||||
local code, status = h:receivestatusline()
|
|
||||||
-- if it is an HTTP/0.9 server, simply get the body and we are done
|
|
||||||
if not code then
|
|
||||||
h:receive09body(status, nreqt.sink, nreqt.step)
|
|
||||||
return 1, 200
|
|
||||||
end
|
|
||||||
local headers
|
|
||||||
-- ignore any 100-continue messages
|
|
||||||
while code == 100 do
|
|
||||||
headers = h:receiveheaders()
|
|
||||||
code, status = h:receivestatusline()
|
|
||||||
end
|
|
||||||
headers = h:receiveheaders()
|
|
||||||
-- at this point we should have a honest reply from the server
|
|
||||||
-- we can't redirect if we already used the source, so we report the error
|
|
||||||
if shouldredirect(nreqt, code, headers) and not nreqt.source then
|
|
||||||
h:close()
|
|
||||||
return tredirect(reqt, headers.location)
|
|
||||||
end
|
|
||||||
-- here we are finally done
|
|
||||||
if shouldreceivebody(nreqt, code) then
|
|
||||||
h:receivebody(headers, nreqt.sink, nreqt.step)
|
|
||||||
end
|
|
||||||
h:close()
|
|
||||||
return 1, code, headers, status
|
|
||||||
end
|
|
||||||
|
|
||||||
local function srequest(u, b)
|
|
||||||
local t = {}
|
|
||||||
local reqt = {
|
|
||||||
url = u,
|
|
||||||
sink = ltn12.sink.table(t)
|
|
||||||
}
|
|
||||||
if b then
|
|
||||||
reqt.source = ltn12.source.string(b)
|
|
||||||
reqt.headers = {
|
|
||||||
["content-length"] = string.len(b),
|
|
||||||
["content-type"] = "application/x-www-form-urlencoded"
|
|
||||||
}
|
|
||||||
reqt.method = "POST"
|
|
||||||
end
|
|
||||||
local code, headers, status = socket.skip(1, trequest(reqt))
|
|
||||||
return table.concat(t), code, headers, status
|
|
||||||
end
|
|
||||||
|
|
||||||
request = socket.protect(function(reqt, body)
|
|
||||||
if base.type(reqt) == "string" then return srequest(reqt, body)
|
|
||||||
else return trequest(reqt) end
|
|
||||||
end)
|
|
||||||
@@ -1,251 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- SMTP client support for the Lua language.
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: smtp.lua,v 1.46 2007/03/12 04:08:40 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local base = _G
|
|
||||||
local coroutine = require("coroutine")
|
|
||||||
local string = require("string")
|
|
||||||
local math = require("math")
|
|
||||||
local os = require("os")
|
|
||||||
local socket = require("socket")
|
|
||||||
local tp = require("socket.tp")
|
|
||||||
local ltn12 = require("ltn12")
|
|
||||||
local mime = require("mime")
|
|
||||||
module("socket.smtp")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Program constants
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- timeout for connection
|
|
||||||
TIMEOUT = 60
|
|
||||||
-- default server used to send e-mails
|
|
||||||
SERVER = "localhost"
|
|
||||||
-- default port
|
|
||||||
PORT = 25
|
|
||||||
-- domain used in HELO command and default sendmail
|
|
||||||
-- If we are under a CGI, try to get from environment
|
|
||||||
DOMAIN = os.getenv("SERVER_NAME") or "localhost"
|
|
||||||
-- default time zone (means we don't know)
|
|
||||||
ZONE = "-0000"
|
|
||||||
|
|
||||||
---------------------------------------------------------------------------
|
|
||||||
-- Low level SMTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local metat = { __index = {} }
|
|
||||||
|
|
||||||
function metat.__index:greet(domain)
|
|
||||||
self.try(self.tp:check("2.."))
|
|
||||||
self.try(self.tp:command("EHLO", domain or DOMAIN))
|
|
||||||
return socket.skip(1, self.try(self.tp:check("2..")))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:mail(from)
|
|
||||||
self.try(self.tp:command("MAIL", "FROM:" .. from))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:rcpt(to)
|
|
||||||
self.try(self.tp:command("RCPT", "TO:" .. to))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:data(src, step)
|
|
||||||
self.try(self.tp:command("DATA"))
|
|
||||||
self.try(self.tp:check("3.."))
|
|
||||||
self.try(self.tp:source(src, step))
|
|
||||||
self.try(self.tp:send("\r\n.\r\n"))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:quit()
|
|
||||||
self.try(self.tp:command("QUIT"))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:close()
|
|
||||||
return self.tp:close()
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:login(user, password)
|
|
||||||
self.try(self.tp:command("AUTH", "LOGIN"))
|
|
||||||
self.try(self.tp:check("3.."))
|
|
||||||
self.try(self.tp:command(mime.b64(user)))
|
|
||||||
self.try(self.tp:check("3.."))
|
|
||||||
self.try(self.tp:command(mime.b64(password)))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:plain(user, password)
|
|
||||||
local auth = "PLAIN " .. mime.b64("\0" .. user .. "\0" .. password)
|
|
||||||
self.try(self.tp:command("AUTH", auth))
|
|
||||||
return self.try(self.tp:check("2.."))
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:auth(user, password, ext)
|
|
||||||
if not user or not password then return 1 end
|
|
||||||
if string.find(ext, "AUTH[^\n]+LOGIN") then
|
|
||||||
return self:login(user, password)
|
|
||||||
elseif string.find(ext, "AUTH[^\n]+PLAIN") then
|
|
||||||
return self:plain(user, password)
|
|
||||||
else
|
|
||||||
self.try(nil, "authentication not supported")
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- send message or throw an exception
|
|
||||||
function metat.__index:send(mailt)
|
|
||||||
self:mail(mailt.from)
|
|
||||||
if base.type(mailt.rcpt) == "table" then
|
|
||||||
for i,v in base.ipairs(mailt.rcpt) do
|
|
||||||
self:rcpt(v)
|
|
||||||
end
|
|
||||||
else
|
|
||||||
self:rcpt(mailt.rcpt)
|
|
||||||
end
|
|
||||||
self:data(ltn12.source.chain(mailt.source, mime.stuff()), mailt.step)
|
|
||||||
end
|
|
||||||
|
|
||||||
function open(server, port, create)
|
|
||||||
local tp = socket.try(tp.connect(server or SERVER, port or PORT,
|
|
||||||
TIMEOUT, create))
|
|
||||||
local s = base.setmetatable({tp = tp}, metat)
|
|
||||||
-- make sure tp is closed if we get an exception
|
|
||||||
s.try = socket.newtry(function()
|
|
||||||
s:close()
|
|
||||||
end)
|
|
||||||
return s
|
|
||||||
end
|
|
||||||
|
|
||||||
-- convert headers to lowercase
|
|
||||||
local function lower_headers(headers)
|
|
||||||
local lower = {}
|
|
||||||
for i,v in base.pairs(headers or lower) do
|
|
||||||
lower[string.lower(i)] = v
|
|
||||||
end
|
|
||||||
return lower
|
|
||||||
end
|
|
||||||
|
|
||||||
---------------------------------------------------------------------------
|
|
||||||
-- Multipart message source
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- returns a hopefully unique mime boundary
|
|
||||||
local seqno = 0
|
|
||||||
local function newboundary()
|
|
||||||
seqno = seqno + 1
|
|
||||||
return string.format('%s%05d==%05u', os.date('%d%m%Y%H%M%S'),
|
|
||||||
math.random(0, 99999), seqno)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- send_message forward declaration
|
|
||||||
local send_message
|
|
||||||
|
|
||||||
-- yield the headers all at once, it's faster
|
|
||||||
local function send_headers(headers)
|
|
||||||
local h = "\r\n"
|
|
||||||
for i,v in base.pairs(headers) do
|
|
||||||
h = i .. ': ' .. v .. "\r\n" .. h
|
|
||||||
end
|
|
||||||
coroutine.yield(h)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- yield multipart message body from a multipart message table
|
|
||||||
local function send_multipart(mesgt)
|
|
||||||
-- make sure we have our boundary and send headers
|
|
||||||
local bd = newboundary()
|
|
||||||
local headers = lower_headers(mesgt.headers or {})
|
|
||||||
headers['content-type'] = headers['content-type'] or 'multipart/mixed'
|
|
||||||
headers['content-type'] = headers['content-type'] ..
|
|
||||||
'; boundary="' .. bd .. '"'
|
|
||||||
send_headers(headers)
|
|
||||||
-- send preamble
|
|
||||||
if mesgt.body.preamble then
|
|
||||||
coroutine.yield(mesgt.body.preamble)
|
|
||||||
coroutine.yield("\r\n")
|
|
||||||
end
|
|
||||||
-- send each part separated by a boundary
|
|
||||||
for i, m in base.ipairs(mesgt.body) do
|
|
||||||
coroutine.yield("\r\n--" .. bd .. "\r\n")
|
|
||||||
send_message(m)
|
|
||||||
end
|
|
||||||
-- send last boundary
|
|
||||||
coroutine.yield("\r\n--" .. bd .. "--\r\n\r\n")
|
|
||||||
-- send epilogue
|
|
||||||
if mesgt.body.epilogue then
|
|
||||||
coroutine.yield(mesgt.body.epilogue)
|
|
||||||
coroutine.yield("\r\n")
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- yield message body from a source
|
|
||||||
local function send_source(mesgt)
|
|
||||||
-- make sure we have a content-type
|
|
||||||
local headers = lower_headers(mesgt.headers or {})
|
|
||||||
headers['content-type'] = headers['content-type'] or
|
|
||||||
'text/plain; charset="iso-8859-1"'
|
|
||||||
send_headers(headers)
|
|
||||||
-- send body from source
|
|
||||||
while true do
|
|
||||||
local chunk, err = mesgt.body()
|
|
||||||
if err then coroutine.yield(nil, err)
|
|
||||||
elseif chunk then coroutine.yield(chunk)
|
|
||||||
else break end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- yield message body from a string
|
|
||||||
local function send_string(mesgt)
|
|
||||||
-- make sure we have a content-type
|
|
||||||
local headers = lower_headers(mesgt.headers or {})
|
|
||||||
headers['content-type'] = headers['content-type'] or
|
|
||||||
'text/plain; charset="iso-8859-1"'
|
|
||||||
send_headers(headers)
|
|
||||||
-- send body from string
|
|
||||||
coroutine.yield(mesgt.body)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- message source
|
|
||||||
function send_message(mesgt)
|
|
||||||
if base.type(mesgt.body) == "table" then send_multipart(mesgt)
|
|
||||||
elseif base.type(mesgt.body) == "function" then send_source(mesgt)
|
|
||||||
else send_string(mesgt) end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- set defaul headers
|
|
||||||
local function adjust_headers(mesgt)
|
|
||||||
local lower = lower_headers(mesgt.headers)
|
|
||||||
lower["date"] = lower["date"] or
|
|
||||||
os.date("!%a, %d %b %Y %H:%M:%S ") .. (mesgt.zone or ZONE)
|
|
||||||
lower["x-mailer"] = lower["x-mailer"] or socket._VERSION
|
|
||||||
-- this can't be overriden
|
|
||||||
lower["mime-version"] = "1.0"
|
|
||||||
return lower
|
|
||||||
end
|
|
||||||
|
|
||||||
function message(mesgt)
|
|
||||||
mesgt.headers = adjust_headers(mesgt)
|
|
||||||
-- create and return message source
|
|
||||||
local co = coroutine.create(function() send_message(mesgt) end)
|
|
||||||
return function()
|
|
||||||
local ret, a, b = coroutine.resume(co)
|
|
||||||
if ret then return a, b
|
|
||||||
else return nil, a end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
---------------------------------------------------------------------------
|
|
||||||
-- High level SMTP API
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
send = socket.protect(function(mailt)
|
|
||||||
local s = open(mailt.server, mailt.port, mailt.create)
|
|
||||||
local ext = s:greet(mailt.domain)
|
|
||||||
s:auth(mailt.user, mailt.password, ext)
|
|
||||||
s:send(mailt)
|
|
||||||
s:quit()
|
|
||||||
return s:close()
|
|
||||||
end)
|
|
||||||
@@ -1,123 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Unified SMTP/FTP subsystem
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: tp.lua,v 1.22 2006/03/14 09:04:15 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module and import dependencies
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local base = _G
|
|
||||||
local string = require("string")
|
|
||||||
local socket = require("socket")
|
|
||||||
local ltn12 = require("ltn12")
|
|
||||||
module("socket.tp")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Program constants
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
TIMEOUT = 60
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Implementation
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- gets server reply (works for SMTP and FTP)
|
|
||||||
local function get_reply(c)
|
|
||||||
local code, current, sep
|
|
||||||
local line, err = c:receive()
|
|
||||||
local reply = line
|
|
||||||
if err then return nil, err end
|
|
||||||
code, sep = socket.skip(2, string.find(line, "^(%d%d%d)(.?)"))
|
|
||||||
if not code then return nil, "invalid server reply" end
|
|
||||||
if sep == "-" then -- reply is multiline
|
|
||||||
repeat
|
|
||||||
line, err = c:receive()
|
|
||||||
if err then return nil, err end
|
|
||||||
current, sep = socket.skip(2, string.find(line, "^(%d%d%d)(.?)"))
|
|
||||||
reply = reply .. "\n" .. line
|
|
||||||
-- reply ends with same code
|
|
||||||
until code == current and sep == " "
|
|
||||||
end
|
|
||||||
return code, reply
|
|
||||||
end
|
|
||||||
|
|
||||||
-- metatable for sock object
|
|
||||||
local metat = { __index = {} }
|
|
||||||
|
|
||||||
function metat.__index:check(ok)
|
|
||||||
local code, reply = get_reply(self.c)
|
|
||||||
if not code then return nil, reply end
|
|
||||||
if base.type(ok) ~= "function" then
|
|
||||||
if base.type(ok) == "table" then
|
|
||||||
for i, v in base.ipairs(ok) do
|
|
||||||
if string.find(code, v) then
|
|
||||||
return base.tonumber(code), reply
|
|
||||||
end
|
|
||||||
end
|
|
||||||
return nil, reply
|
|
||||||
else
|
|
||||||
if string.find(code, ok) then return base.tonumber(code), reply
|
|
||||||
else return nil, reply end
|
|
||||||
end
|
|
||||||
else return ok(base.tonumber(code), reply) end
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:command(cmd, arg)
|
|
||||||
if arg then
|
|
||||||
return self.c:send(cmd .. " " .. arg.. "\r\n")
|
|
||||||
else
|
|
||||||
return self.c:send(cmd .. "\r\n")
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:sink(snk, pat)
|
|
||||||
local chunk, err = c:receive(pat)
|
|
||||||
return snk(chunk, err)
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:send(data)
|
|
||||||
return self.c:send(data)
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:receive(pat)
|
|
||||||
return self.c:receive(pat)
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:getfd()
|
|
||||||
return self.c:getfd()
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:dirty()
|
|
||||||
return self.c:dirty()
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:getcontrol()
|
|
||||||
return self.c
|
|
||||||
end
|
|
||||||
|
|
||||||
function metat.__index:source(source, step)
|
|
||||||
local sink = socket.sink("keep-open", self.c)
|
|
||||||
local ret, err = ltn12.pump.all(source, sink, step or ltn12.pump.step)
|
|
||||||
return ret, err
|
|
||||||
end
|
|
||||||
|
|
||||||
-- closes the underlying c
|
|
||||||
function metat.__index:close()
|
|
||||||
self.c:close()
|
|
||||||
return 1
|
|
||||||
end
|
|
||||||
|
|
||||||
-- connect with server and return c object
|
|
||||||
function connect(host, port, timeout, create)
|
|
||||||
local c, e = (create or socket.tcp)()
|
|
||||||
if not c then return nil, e end
|
|
||||||
c:settimeout(timeout or TIMEOUT)
|
|
||||||
local r, e = c:connect(host, port)
|
|
||||||
if not r then
|
|
||||||
c:close()
|
|
||||||
return nil, e
|
|
||||||
end
|
|
||||||
return base.setmetatable({c = c}, metat)
|
|
||||||
end
|
|
||||||
|
|
||||||
@@ -1,297 +0,0 @@
|
|||||||
-----------------------------------------------------------------------------
|
|
||||||
-- URI parsing, composition and relative URL resolution
|
|
||||||
-- LuaSocket toolkit.
|
|
||||||
-- Author: Diego Nehab
|
|
||||||
-- RCS ID: $Id: url.lua,v 1.38 2006/04/03 04:45:42 diego Exp $
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Declare module
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local string = require("string")
|
|
||||||
local base = _G
|
|
||||||
local table = require("table")
|
|
||||||
module("socket.url")
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Module version
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
_VERSION = "URL 1.0.1"
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Encodes a string into its escaped hexadecimal representation
|
|
||||||
-- Input
|
|
||||||
-- s: binary string to be encoded
|
|
||||||
-- Returns
|
|
||||||
-- escaped representation of string binary
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function escape(s)
|
|
||||||
return string.gsub(s, "([^A-Za-z0-9_])", function(c)
|
|
||||||
return string.format("%%%02x", string.byte(c))
|
|
||||||
end)
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Protects a path segment, to prevent it from interfering with the
|
|
||||||
-- url parsing.
|
|
||||||
-- Input
|
|
||||||
-- s: binary string to be encoded
|
|
||||||
-- Returns
|
|
||||||
-- escaped representation of string binary
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local function make_set(t)
|
|
||||||
local s = {}
|
|
||||||
for i,v in base.ipairs(t) do
|
|
||||||
s[t[i]] = 1
|
|
||||||
end
|
|
||||||
return s
|
|
||||||
end
|
|
||||||
|
|
||||||
-- these are allowed withing a path segment, along with alphanum
|
|
||||||
-- other characters must be escaped
|
|
||||||
local segment_set = make_set {
|
|
||||||
"-", "_", ".", "!", "~", "*", "'", "(",
|
|
||||||
")", ":", "@", "&", "=", "+", "$", ",",
|
|
||||||
}
|
|
||||||
|
|
||||||
local function protect_segment(s)
|
|
||||||
return string.gsub(s, "([^A-Za-z0-9_])", function (c)
|
|
||||||
if segment_set[c] then return c
|
|
||||||
else return string.format("%%%02x", string.byte(c)) end
|
|
||||||
end)
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Encodes a string into its escaped hexadecimal representation
|
|
||||||
-- Input
|
|
||||||
-- s: binary string to be encoded
|
|
||||||
-- Returns
|
|
||||||
-- escaped representation of string binary
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function unescape(s)
|
|
||||||
return string.gsub(s, "%%(%x%x)", function(hex)
|
|
||||||
return string.char(base.tonumber(hex, 16))
|
|
||||||
end)
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Builds a path from a base path and a relative path
|
|
||||||
-- Input
|
|
||||||
-- base_path
|
|
||||||
-- relative_path
|
|
||||||
-- Returns
|
|
||||||
-- corresponding absolute path
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
local function absolute_path(base_path, relative_path)
|
|
||||||
if string.sub(relative_path, 1, 1) == "/" then return relative_path end
|
|
||||||
local path = string.gsub(base_path, "[^/]*$", "")
|
|
||||||
path = path .. relative_path
|
|
||||||
path = string.gsub(path, "([^/]*%./)", function (s)
|
|
||||||
if s ~= "./" then return s else return "" end
|
|
||||||
end)
|
|
||||||
path = string.gsub(path, "/%.$", "/")
|
|
||||||
local reduced
|
|
||||||
while reduced ~= path do
|
|
||||||
reduced = path
|
|
||||||
path = string.gsub(reduced, "([^/]*/%.%./)", function (s)
|
|
||||||
if s ~= "../../" then return "" else return s end
|
|
||||||
end)
|
|
||||||
end
|
|
||||||
path = string.gsub(reduced, "([^/]*/%.%.)$", function (s)
|
|
||||||
if s ~= "../.." then return "" else return s end
|
|
||||||
end)
|
|
||||||
return path
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Parses a url and returns a table with all its parts according to RFC 2396
|
|
||||||
-- The following grammar describes the names given to the URL parts
|
|
||||||
-- <url> ::= <scheme>://<authority>/<path>;<params>?<query>#<fragment>
|
|
||||||
-- <authority> ::= <userinfo>@<host>:<port>
|
|
||||||
-- <userinfo> ::= <user>[:<password>]
|
|
||||||
-- <path> :: = {<segment>/}<segment>
|
|
||||||
-- Input
|
|
||||||
-- url: uniform resource locator of request
|
|
||||||
-- default: table with default values for each field
|
|
||||||
-- Returns
|
|
||||||
-- table with the following fields, where RFC naming conventions have
|
|
||||||
-- been preserved:
|
|
||||||
-- scheme, authority, userinfo, user, password, host, port,
|
|
||||||
-- path, params, query, fragment
|
|
||||||
-- Obs:
|
|
||||||
-- the leading '/' in {/<path>} is considered part of <path>
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function parse(url, default)
|
|
||||||
-- initialize default parameters
|
|
||||||
local parsed = {}
|
|
||||||
for i,v in base.pairs(default or parsed) do parsed[i] = v end
|
|
||||||
-- empty url is parsed to nil
|
|
||||||
if not url or url == "" then return nil, "invalid url" end
|
|
||||||
-- remove whitespace
|
|
||||||
-- url = string.gsub(url, "%s", "")
|
|
||||||
-- get fragment
|
|
||||||
url = string.gsub(url, "#(.*)$", function(f)
|
|
||||||
parsed.fragment = f
|
|
||||||
return ""
|
|
||||||
end)
|
|
||||||
-- get scheme
|
|
||||||
url = string.gsub(url, "^([%w][%w%+%-%.]*)%:",
|
|
||||||
function(s) parsed.scheme = s; return "" end)
|
|
||||||
-- get authority
|
|
||||||
url = string.gsub(url, "^//([^/]*)", function(n)
|
|
||||||
parsed.authority = n
|
|
||||||
return ""
|
|
||||||
end)
|
|
||||||
-- get query stringing
|
|
||||||
url = string.gsub(url, "%?(.*)", function(q)
|
|
||||||
parsed.query = q
|
|
||||||
return ""
|
|
||||||
end)
|
|
||||||
-- get params
|
|
||||||
url = string.gsub(url, "%;(.*)", function(p)
|
|
||||||
parsed.params = p
|
|
||||||
return ""
|
|
||||||
end)
|
|
||||||
-- path is whatever was left
|
|
||||||
if url ~= "" then parsed.path = url end
|
|
||||||
local authority = parsed.authority
|
|
||||||
if not authority then return parsed end
|
|
||||||
authority = string.gsub(authority,"^([^@]*)@",
|
|
||||||
function(u) parsed.userinfo = u; return "" end)
|
|
||||||
authority = string.gsub(authority, ":([^:]*)$",
|
|
||||||
function(p) parsed.port = p; return "" end)
|
|
||||||
if authority ~= "" then parsed.host = authority end
|
|
||||||
local userinfo = parsed.userinfo
|
|
||||||
if not userinfo then return parsed end
|
|
||||||
userinfo = string.gsub(userinfo, ":([^:]*)$",
|
|
||||||
function(p) parsed.password = p; return "" end)
|
|
||||||
parsed.user = userinfo
|
|
||||||
return parsed
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Rebuilds a parsed URL from its components.
|
|
||||||
-- Components are protected if any reserved or unallowed characters are found
|
|
||||||
-- Input
|
|
||||||
-- parsed: parsed URL, as returned by parse
|
|
||||||
-- Returns
|
|
||||||
-- a stringing with the corresponding URL
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function build(parsed)
|
|
||||||
local ppath = parse_path(parsed.path or "")
|
|
||||||
local url = build_path(ppath)
|
|
||||||
if parsed.params then url = url .. ";" .. parsed.params end
|
|
||||||
if parsed.query then url = url .. "?" .. parsed.query end
|
|
||||||
local authority = parsed.authority
|
|
||||||
if parsed.host then
|
|
||||||
authority = parsed.host
|
|
||||||
if parsed.port then authority = authority .. ":" .. parsed.port end
|
|
||||||
local userinfo = parsed.userinfo
|
|
||||||
if parsed.user then
|
|
||||||
userinfo = parsed.user
|
|
||||||
if parsed.password then
|
|
||||||
userinfo = userinfo .. ":" .. parsed.password
|
|
||||||
end
|
|
||||||
end
|
|
||||||
if userinfo then authority = userinfo .. "@" .. authority end
|
|
||||||
end
|
|
||||||
if authority then url = "//" .. authority .. url end
|
|
||||||
if parsed.scheme then url = parsed.scheme .. ":" .. url end
|
|
||||||
if parsed.fragment then url = url .. "#" .. parsed.fragment end
|
|
||||||
-- url = string.gsub(url, "%s", "")
|
|
||||||
return url
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Builds a absolute URL from a base and a relative URL according to RFC 2396
|
|
||||||
-- Input
|
|
||||||
-- base_url
|
|
||||||
-- relative_url
|
|
||||||
-- Returns
|
|
||||||
-- corresponding absolute url
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function absolute(base_url, relative_url)
|
|
||||||
if base.type(base_url) == "table" then
|
|
||||||
base_parsed = base_url
|
|
||||||
base_url = build(base_parsed)
|
|
||||||
else
|
|
||||||
base_parsed = parse(base_url)
|
|
||||||
end
|
|
||||||
local relative_parsed = parse(relative_url)
|
|
||||||
if not base_parsed then return relative_url
|
|
||||||
elseif not relative_parsed then return base_url
|
|
||||||
elseif relative_parsed.scheme then return relative_url
|
|
||||||
else
|
|
||||||
relative_parsed.scheme = base_parsed.scheme
|
|
||||||
if not relative_parsed.authority then
|
|
||||||
relative_parsed.authority = base_parsed.authority
|
|
||||||
if not relative_parsed.path then
|
|
||||||
relative_parsed.path = base_parsed.path
|
|
||||||
if not relative_parsed.params then
|
|
||||||
relative_parsed.params = base_parsed.params
|
|
||||||
if not relative_parsed.query then
|
|
||||||
relative_parsed.query = base_parsed.query
|
|
||||||
end
|
|
||||||
end
|
|
||||||
else
|
|
||||||
relative_parsed.path = absolute_path(base_parsed.path or "",
|
|
||||||
relative_parsed.path)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
return build(relative_parsed)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Breaks a path into its segments, unescaping the segments
|
|
||||||
-- Input
|
|
||||||
-- path
|
|
||||||
-- Returns
|
|
||||||
-- segment: a table with one entry per segment
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function parse_path(path)
|
|
||||||
local parsed = {}
|
|
||||||
path = path or ""
|
|
||||||
--path = string.gsub(path, "%s", "")
|
|
||||||
string.gsub(path, "([^/]+)", function (s) table.insert(parsed, s) end)
|
|
||||||
for i = 1, table.getn(parsed) do
|
|
||||||
parsed[i] = unescape(parsed[i])
|
|
||||||
end
|
|
||||||
if string.sub(path, 1, 1) == "/" then parsed.is_absolute = 1 end
|
|
||||||
if string.sub(path, -1, -1) == "/" then parsed.is_directory = 1 end
|
|
||||||
return parsed
|
|
||||||
end
|
|
||||||
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
-- Builds a path component from its segments, escaping protected characters.
|
|
||||||
-- Input
|
|
||||||
-- parsed: path segments
|
|
||||||
-- unsafe: if true, segments are not protected before path is built
|
|
||||||
-- Returns
|
|
||||||
-- path: corresponding path stringing
|
|
||||||
-----------------------------------------------------------------------------
|
|
||||||
function build_path(parsed, unsafe)
|
|
||||||
local path = ""
|
|
||||||
local n = table.getn(parsed)
|
|
||||||
if unsafe then
|
|
||||||
for i = 1, n-1 do
|
|
||||||
path = path .. parsed[i]
|
|
||||||
path = path .. "/"
|
|
||||||
end
|
|
||||||
if n > 0 then
|
|
||||||
path = path .. parsed[n]
|
|
||||||
if parsed.is_directory then path = path .. "/" end
|
|
||||||
end
|
|
||||||
else
|
|
||||||
for i = 1, n-1 do
|
|
||||||
path = path .. protect_segment(parsed[i])
|
|
||||||
path = path .. "/"
|
|
||||||
end
|
|
||||||
if n > 0 then
|
|
||||||
path = path .. protect_segment(parsed[n])
|
|
||||||
if parsed.is_directory then path = path .. "/" end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
if parsed.is_absolute then path = "/" .. path end
|
|
||||||
return path
|
|
||||||
end
|
|
||||||
232
readme.md
232
readme.md
@@ -1,33 +1,35 @@
|
|||||||
|
|
||||||
# BlockLua
|
# BlockLua
|
||||||
|
|
||||||
Lua scripting for Blockland
|
Lua scripting for Blockland
|
||||||
|
|
||||||
## How to Install
|
## How to Install
|
||||||
|
|
||||||
- Install RedBlocklandLoader
|
- Install RedBlocklandLoader
|
||||||
- Copy `lua5.1.dll` into your Blockland install folder, next to `Blockland.exe`
|
- Copy `lua5.1.dll` into your Blockland install folder, next to `Blockland.exe`
|
||||||
- Copy `BlockLua.dll` into the `modules` folder within the Blockland folder
|
- Copy `BlockLua.dll` into the `modules` folder within the Blockland folder
|
||||||
- Optional: Copy the `lualib` folder into `modules`
|
|
||||||
|
|
||||||
|
|
||||||
## Quick Reference
|
## Quick Reference
|
||||||
|
|
||||||
### From TorqueScript
|
### From TorqueScript
|
||||||
`'print('hello world')` - Execute Lua code in the console by prepending a `'` (single quote)
|
|
||||||
`luaeval("code");` - Eval Lua code
|
`'print('hello world')` - Execute Lua in the console by prepending a `'` (single quote)
|
||||||
`luacall("funcName", %args);` - Call a Lua global function
|
`luaeval("code");` - Execute Lua code
|
||||||
`luaexec("fileName");` - Execute a Lua file. Path rules are the same as executing .cs files.
|
`luacall("funcName", %args...);` - Call a Lua function (supports indexing tables and object methods)
|
||||||
`luaget("varName");` - Read a Lua global variable
|
`luaexec("fileName");` - Execute a Lua file. Path rules are the same as when executing .cs files, relative paths are allowed.
|
||||||
`luaset("varName");` - Write a Lua global variable
|
`luaget("varName");` - Read a Lua global variable (supports indexing tables)
|
||||||
|
`luaset("varName", %value);` - Write a Lua global variable (supports indexing tables)
|
||||||
|
|
||||||
### From Lua
|
### From Lua
|
||||||
|
|
||||||
`bl.eval('code')` - Eval TorqueScript code
|
`bl.eval('code')` - Eval TorqueScript code
|
||||||
`bl.funcName(args)` - Call a TorqueScript function
|
`bl.funcName(args)` - Call a TorqueScript function
|
||||||
`bl.varName` - Read a TorqueScript global variable
|
`bl.varName` - Read a TorqueScript global variable
|
||||||
`bl['varName']` - Read a TorqueScript global variable (i.e. with special characters in the name, or from an array)
|
`bl['varName']` - Read a TorqueScript global variable (i.e. with special characters in the name, or from an array)
|
||||||
`bl.set('varName', value)` - Write a TorqueScript global variable
|
`bl.set('varName', value)` - Write a TorqueScript global variable
|
||||||
|
`bl['namespaceName::funcName'](args)` - Call a namespaced TorqueScript function
|
||||||
|
|
||||||
### Accessing Torque Objects from Lua
|
### Accessing Torque Objects from Lua
|
||||||
|
|
||||||
`bl.objectName` - Access a Torque object by name
|
`bl.objectName` - Access a Torque object by name
|
||||||
`bl[objectID]` - Access a Torque object by ID (or name)
|
`bl[objectID]` - Access a Torque object by ID (or name)
|
||||||
`object.fieldOrKey` - Read a field or Lua key from a Torque object
|
`object.fieldOrKey` - Read a field or Lua key from a Torque object
|
||||||
@@ -35,31 +37,69 @@ Lua scripting for Blockland
|
|||||||
`object.key = value` - Associate Lua data with a Torque object
|
`object.key = value` - Associate Lua data with a Torque object
|
||||||
`object:method(args)` - Call a Torque object method
|
`object:method(args)` - Call a Torque object method
|
||||||
`object[index]` - Access a member of a Torque set or group
|
`object[index]` - Access a member of a Torque set or group
|
||||||
`for childIndex, child in object:members() do` - Iterate objects within of a Torque set or group. Indices start at 0 like in Torque.
|
`for child in object:members() do` - Iterate objects within of a Torque set or group. Indices start at 0 like in Torque.
|
||||||
|
`bl.isObject(object, objectID, or 'objectName')` - Check if an object exists
|
||||||
|
`object:exists()` - Check if an object exists
|
||||||
|
|
||||||
|
### Timing/Schedules
|
||||||
|
|
||||||
### Advanced
|
|
||||||
`sched = bl.schedule(timeMs, function, args...)` - Schedule a Lua function to be called later, similar to schedule in Torque
|
`sched = bl.schedule(timeMs, function, args...)` - Schedule a Lua function to be called later, similar to schedule in Torque
|
||||||
`sched:cancel()` - Cancel a previously scheduled timer
|
`sched:cancel()` - Cancel a previously scheduled timer
|
||||||
`hitObject, hisPos, hitNormal = bl.raycast(vector{startPosX,y,z}, vector{endPosX,y,z}, 'objtype'/{'objtypes',...}, ignoreObjects...?)` - Cast a ray in the world over objects of the specified type(s) (possibly excluding some objects), and return the object hit, the position of the hit, and the normal vector to the surface hit. See the Types section for a list of valid object types.
|
|
||||||
|
### Raycasts and Searches
|
||||||
|
|
||||||
|
`hitObject, hitPos, hitNormal = bl.raycast(vector{startPosX,y,z}, vector{endPosX,y,z}, 'objtype'/{'objtypes',...}, ignoreObjects...?)` - Cast a ray in the world over objects of the specified type(s) (possibly excluding some objects), and return the object hit, the position of the hit, and the normal vector to the surface hit. See the Types section for a list of valid object types.
|
||||||
`for object in bl.boxSearch(vector{centerX,y,z}, vector{sizeX,y,z}, 'objtype'/{'objtypes',...}) do` - Find all objects in the world of the specified type(s) whose bounding box overlaps with the specified box. See the Types section for a list of valid object types.
|
`for object in bl.boxSearch(vector{centerX,y,z}, vector{sizeX,y,z}, 'objtype'/{'objtypes',...}) do` - Find all objects in the world of the specified type(s) whose bounding box overlaps with the specified box. See the Types section for a list of valid object types.
|
||||||
`for object in bl.radiusSearch(vector{centerX,y,z}, radius, 'objtype'/{'objtypes',...}) do` - Find all objects of the specified type(s) whose bounding box overlaps with the specified sphere. See the Types section for a list of valid object types.
|
`for object in bl.radiusSearch(vector{centerX,y,z}, radius, 'objtype'/{'objtypes',...}) do` - Find all objects of the specified type(s) whose bounding box overlaps with the specified sphere. See the Types section for a list of valid object types.
|
||||||
`bl.serverCmd('commandName', function(client, args...) code() end)` - Register a /-command on the server
|
|
||||||
`bl.clientCmd('commandName', function(args...) code() end)` - Register a client command on the client
|
### List of Object Classes (for raycasts and searches)
|
||||||
|
|
||||||
|
`'all'` - Any object
|
||||||
|
`'player'` - Players or bots
|
||||||
|
`'item'` - Items
|
||||||
|
`'vehicle'` - Vehicles
|
||||||
|
`'projectile'` - Projectiles
|
||||||
|
`'brick'` - Bricks with raycasting enabled
|
||||||
|
`'brickalways'` - All bricks including those with raycasting disabled
|
||||||
|
Other types: `'static'`, `'environment'`, `'terrain'`, `'water'`, `'trigger'`, `'marker'`, `'gamebase'`, `'shapebase'`, `'camera'`, `'staticshape'`, `'vehicleblocker'`, `'explosion'`, `'corpse'`, `'debris'`, `'physicalzone'`, `'staticts'`, `'staticrendered'`, `'damagableitem'`
|
||||||
|
|
||||||
|
### Server-Client Communication
|
||||||
|
|
||||||
|
`bl.addServerCmd('commandName', function(client, args...) ... end)` - Register a /command on the server
|
||||||
|
`bl.addClientCmd('commandName', function(args...) ... end)` - Register a client command on the client
|
||||||
|
`bl.commandToServer('commandName', args...)` - As a client, execute a server command
|
||||||
|
`bl.commandToClient(client, 'commandName', args...)` - As the server, execute a client command on a specific client
|
||||||
|
`bl.commandToAll('commandName', args...)` - As the server, execute a client command on all clients
|
||||||
|
|
||||||
### Packages/Hooks
|
### Packages/Hooks
|
||||||
`bl.hook('packageName', 'functionName', 'before'/'after'/'override', function(args...) code() end)` - Hook a Torque function with a Lua function
|
|
||||||
`bl.unhook('packageName', 'functionName', 'before'/'after'/'override')` - Remove a previously defined hook
|
|
||||||
|
|
||||||
### Classes and Types
|
`bl.hook('packageName', 'functionName', 'before'/'after', function(args) ... end)` - Hook a Torque function with a Lua function.
|
||||||
`bl.bool(thing)` - Convert a Torque boolean (0 or 1) into a Lua boolean. Done automatically for all built-in functions that return bools.
|
`args` is an array containing the arguments provided to the function. If the hook is `before`, these can be modified before being passed to the parent function.
|
||||||
`bl.object(thing)` - Convert a Torque object reference (object ID or name) into a Lua object. Done automatically for all built-in functions that return objects.
|
If `args._return` is set to anything other than nil by a `before` hook, the parent function will not be called, and the function will simply return that value. Also in this case, any `after` hook will not be executed.
|
||||||
`bl.type('varName', 'type')` - Register the type of a Torque global variable, for conversion when accessing from Lua. Valid types are 'bool', 'object', and nil - all other conversion is automatic.
|
In an `after` hook, `args._return` is set to the value returned by the parent function, and can be modified.
|
||||||
`bl.type('funcName', 'type')` - Register the return type of a Torque function, for conversion when calling from Lua. Valid types are 'bool', 'object', and nil - all other conversion is automatic. Already done for all built-in functions that return objects.
|
`bl.unhook('packageName', 'functionName', 'before'/'after')` - Remove a previously defined hook
|
||||||
`bl.type('className::funcName', 'type')` - Register the return type of a Torque object method.
|
`bl.unhook('packageName', 'functionName')` - Remove any previously defined hooks on the function within the package
|
||||||
`bl.class('className')` - Register a Torque class to be used from Lua (Already done for all built-in classes)
|
`bl.unhook('packageName')` - Remove any previously defined hooks within the package
|
||||||
`bl.class('className', 'parentClassName')` - Same as above, with inheritance
|
|
||||||
|
### Modules and Dependencies
|
||||||
|
|
||||||
|
`dofile('Add-Ons/Path/file.lua')` - Execute a Lua file. Relative paths (`./file.lua`) are allowed. `..` is not allowed.
|
||||||
|
|
||||||
|
`require('modulePath.moduleName')` - Load a Lua file or external library.
|
||||||
|
`require` replaces `.` with `/` in the path, and then searches for files in the following order:
|
||||||
|
|
||||||
|
- `./modulePath/moduleName.lua`
|
||||||
|
- `./modulePath/moduleName/init.lua`
|
||||||
|
- `modulePath/moduleName.lua` (Relative to game directory)
|
||||||
|
- `modulePath/moduleName/init.lua` (Relative to game directory)
|
||||||
|
- `modules/lualib/modulePath/moduleName.lua`
|
||||||
|
- `modules/lualib/modulePath/moduleName/init.lua`
|
||||||
|
- `modules/lualib/modulePath/moduleName.dll` - C libraries for Lua can be loaded
|
||||||
|
|
||||||
|
Like in standard Lua, modules loaded using `require` are only executed the first time `require` is called with that path (from anywhere). Subsequent calls simply return the result from the initial execution. To allow hot reloading, use `dofile`.
|
||||||
|
|
||||||
### File I/O
|
### File I/O
|
||||||
|
|
||||||
Lua's builtin file I/O is emulated, and is confined to the same directories as TorqueScript file I/O.
|
Lua's builtin file I/O is emulated, and is confined to the same directories as TorqueScript file I/O.
|
||||||
Relative paths (`./`) are allowed. `..` is not allowed.
|
Relative paths (`./`) are allowed. `..` is not allowed.
|
||||||
`file = io.open('./file.txt', 'r'/'w'/'a'/'rb'/'wb'/'ab')` - Open a file
|
`file = io.open('./file.txt', 'r'/'w'/'a'/'rb'/'wb'/'ab')` - Open a file
|
||||||
@@ -69,54 +109,132 @@ Relative paths (`./`) are allowed. `..` is not allowed.
|
|||||||
Reading files from ZIPs is supported, with caveats. Null characters are not allowed, and \r\n becomes \n. Generally, text formats work, and binary formats don't.
|
Reading files from ZIPs is supported, with caveats. Null characters are not allowed, and \r\n becomes \n. Generally, text formats work, and binary formats don't.
|
||||||
When reading from outside ZIPs, binary files are fully supported.
|
When reading from outside ZIPs, binary files are fully supported.
|
||||||
|
|
||||||
### Modules and Dependencies
|
### Object Creation
|
||||||
`dofile('Add-Ons/Path/file.lua')` - Execute a Lua file. Relative paths (`./file.lua`) are allowed. `..` is not allowed.
|
|
||||||
|
|
||||||
`require('modulePath.moduleName')` - Load a Lua file or or external library.
|
`bl.new('className')` - Create a new Torque object
|
||||||
Require replaces `.` with `/` in the path, and then searches for files in the following order:
|
`bl.new('className', {fieldName = value, ...})` - Create a new Torque object with the given fields
|
||||||
- `./modulePath/moduleName.lua`
|
`bl.new('className objectName', fields?)` - Create a new named Torque object
|
||||||
- `./modulePath/moduleName/init.lua`
|
`bl.new('className objectName:parentName', fields?)` - Create a new named Torque object with inheritance
|
||||||
- `modulePath/moduleName.lua` (Relative to game directory)
|
`bl.datablock('datablockClassName datablockName', fields?)` - Create a new datablock
|
||||||
- `modulePath/moduleName/init.lua` (Relative to game directory)
|
`bl.datablock('datablockClassName datablockName:parentDatablockName', fields?)` - Create a new datablock with inheritance
|
||||||
- `modules/lualib/modulePath/moduleName.lua`
|
|
||||||
- `modules/lualib/modulePath/moduleName/init.lua`
|
|
||||||
- `modules/lualib/modulePath/moduleName.dll`
|
|
||||||
|
|
||||||
Like in standard Lua, modules loaded using `require` are only executed the first time `require` is called with that path. Subsequent calls simply return the result from the initial execution. To allow hot reloading, use `dofile`.
|
### Classes and Types
|
||||||
|
|
||||||
|
`bl.type('varName', 'type')` - Register the type of a Torque global variable, for conversion when accessing from Lua. Valid types are 'boolean', 'object', 'string' (prevents automatic conversion), and nil (default, applies automatic conversion).
|
||||||
|
`bl.type('funcName', 'type')` - Register the return type of a Torque function, for conversion when calling from Lua. Valid types are 'bool', 'object', and nil - all other conversion is automatic. Already done for all default functions.
|
||||||
|
`bl.type('className::funcName', 'type')` - Register the return type of a Torque object method.
|
||||||
|
`bl.class('className')` - Register an existing Torque class to be used from Lua. Already done for all built-in classes.
|
||||||
|
`bl.class('className', 'parentClassName')` - Same as above, with inheritance
|
||||||
|
`bl.boolean(arg)` - Manually convert a Torque boolean (0 or 1) into a Lua boolean.
|
||||||
|
`bl.object(arg)` - Manually convert a Torque object reference (object ID or name) into a Lua object.
|
||||||
|
`bl.string(arg)` - Manually convert any automatically-converted Torque value back into a string. This is not as reliable as using `bl.type` to specify the type as a string beforehand.
|
||||||
|
|
||||||
|
### Vector
|
||||||
|
|
||||||
|
`vec = vector{x,y,z}` - Create a vector. Can have any number of elements
|
||||||
|
`vec1 + vec2` - Add
|
||||||
|
`vec1 - vec2` - Subtract
|
||||||
|
`vec * number` - Scale (x\*n, y\*n, z\*n)
|
||||||
|
`vec / number` - Scale (x/n, y/n, z/n)
|
||||||
|
`vec ^ number` - Exponentiate (x^n, y^n, z^n)
|
||||||
|
`vec1 * vec2` - Multiply elements piecewise (x1\*x2, y1\*y2, z1\*z2)
|
||||||
|
`vec1 / vec2` - Divide elements piecewise (x1/x2, y1/y2, z1/z2)
|
||||||
|
`vec1 ^ vec2` - Exponentiate elements piecewise (x1^x2, y1^y2, z1^z2)
|
||||||
|
`-vec` - Negate/invert
|
||||||
|
`vec1 == vec2` - Compare by value
|
||||||
|
`vec:length()` - Length
|
||||||
|
`vec:normalize()` - Preserve direction but scale so magnitude is 1
|
||||||
|
`vec:floor()` - Floor each element
|
||||||
|
`vec:ceil()` - Ceil each element
|
||||||
|
`vec:abs()` - Absolute value each element
|
||||||
|
`vec1:dot(vec2)` - Dot product
|
||||||
|
`vec1:cross(vec2)` - Cross product (Only defined for two vectors of length 3)
|
||||||
|
`vec:rotateZ(radians)` - Rotate counterclockwise about the Z (vertical) axis
|
||||||
|
`vec:rotateByAngleId(0/1/2/3)` - Rotate counterclockwise about the Z (vertical) axis in increments of 90 degrees
|
||||||
|
`vec:tsString()` - Convert to string usable by Torque. Done automatically when a vector is passed to Torque.
|
||||||
|
`vec1:distance(vec2)` - Distance between two points
|
||||||
|
`vec2 = vec:copy()` - Clone a vector so its elements can be modified without affecting the original. Usually not needed - the builtin vector functions never modify vectors in-place.
|
||||||
|
|
||||||
|
### Matrix
|
||||||
|
|
||||||
|
WIP
|
||||||
|
|
||||||
|
### Extended Standard Lua Library
|
||||||
|
|
||||||
|
`str[index]`
|
||||||
|
`str[{start,stop}]`
|
||||||
|
`string.split(str, separator='' (splits into chars), noregex=false)`
|
||||||
|
`string.bytes(str)`
|
||||||
|
`string.trim(str, charsToTrim=' \t\r\n')`
|
||||||
|
`table.empty`
|
||||||
|
`table.map(func, ...)`
|
||||||
|
`table.mapk(func, ...)`
|
||||||
|
`table.map_list(func, ...)`
|
||||||
|
`table.mapi_list(func, ...)`
|
||||||
|
`table.swap(tbl)`
|
||||||
|
`table.reverse(list)`
|
||||||
|
`table.islist(list)`
|
||||||
|
`table.append(list, ...)`
|
||||||
|
`table.join(...)`
|
||||||
|
`table.contains(tbl, val)`
|
||||||
|
`table.contains_list(list, val)`
|
||||||
|
`table.copy(tbl)`
|
||||||
|
`table.copy_list(list)`
|
||||||
|
`table.sortcopy(tbl, sortFunction?)`
|
||||||
|
`table.removevalue(tbl, val)`
|
||||||
|
`table.removevalue_list(tbl, val)`
|
||||||
|
`table.tostring(tbl)`
|
||||||
|
`for char in string.chars(str) do`
|
||||||
|
`string.escape(str, escapes={['\n']='\\n', etc. (C standard)})`
|
||||||
|
`string.unescape(str, escapeChar='\\', unescapes={['\\n']='\n', etc.})`
|
||||||
|
`io.readall(filename)`
|
||||||
|
`io.writeall(filename, str)`
|
||||||
|
`math.round(num)`
|
||||||
|
`math.mod(divisor, modulus)`
|
||||||
|
`math.clamp(num, min, max)`
|
||||||
|
|
||||||
## Type Conversion
|
## Type Conversion
|
||||||
|
|
||||||
When a TorqueScript function is called from Lua or vice-versa, the arguments and return value must be converted between the two languages' type systems.
|
When a TorqueScript function is called from Lua or vice-versa, the arguments and return value must be converted between the two languages' type systems.
|
||||||
TorqueScript stores no type information; all values in TorqueScript are strings. So it's necessary to make some inferences when converting values between the two languages.
|
TorqueScript stores no type information; all values in TorqueScript are strings. So it's necessary to make some inferences when converting values between the two languages.
|
||||||
### From TorqueScript to Lua
|
|
||||||
- Any numeric value becomes a Lua `number`, except as specified with `ts.type`, which may convert a value into a `boolean` or a Torque object container.
|
|
||||||
- The empty string "" becomes `nil`
|
|
||||||
- A string containing three numbers separated by spaces becomes a `vector`
|
|
||||||
- A string containing six numbers separated by spaces becomes a table of two vectors
|
|
||||||
- Any other string is passed directly as a `string`
|
|
||||||
### From Lua to TorqueScript
|
### From Lua to TorqueScript
|
||||||
|
|
||||||
- `nil` becomes the empty string ""
|
- `nil` becomes the empty string ""
|
||||||
- `true` and `false` become "1" and "0" respectively
|
- `true` and `false` become "1" and "0" respectively
|
||||||
- Torque containers become their object ID
|
- A Torque object container becomes its object ID
|
||||||
- A `vector` becomes a string containing three numbers separated by spaces
|
- A `vector` becomes a string containing three numbers separated by spaces
|
||||||
- A table of two vectors becomes a string containing six numbers separated by spaces
|
- A table of two `vector`s becomes a string containing six numbers separated by spaces
|
||||||
|
- (WIP) A `matrix` is converted into an axis-angle (a "transform"), a string containing seven numbers separated by spaces
|
||||||
- Any `string` is passed directly as a string
|
- Any `string` is passed directly as a string
|
||||||
- Tables cannot be passed and will throw an error
|
- Tables cannot be passed and will throw an error
|
||||||
|
|
||||||
|
### From TorqueScript to Lua
|
||||||
|
|
||||||
|
- The empty string "" becomes `nil`
|
||||||
|
- Any numeric value becomes a Lua `number`, except as specified with `bl.type`, which may convert a value into a `boolean` or a Torque object container.
|
||||||
|
- A string containing two or three numbers separated by single spaces becomes a `vector`
|
||||||
|
- A string containing six numbers separated by single spaces becomes a table of two vectors, usually defining the corners a bounding box
|
||||||
|
- (WIP) A string containing seven numbers separated by single spaces is treated as an axis-angle (a "transform"), and is converted into a `matrix` representing the translation and rotation
|
||||||
|
- Any other string is passed directly as a `string`
|
||||||
|
|
||||||
|
For scenarios where the automatic TorqueScript->Lua conversion rules are insufficient or incorrect, use `bl.type`.
|
||||||
|
To convert things by hand, use `bl.object`, `bl.boolean`, or `bl.string`.
|
||||||
|
|
||||||
## I/O and Safety
|
## I/O and Safety
|
||||||
All Lua code is sandboxed, and file access is confied to the default directories in the same way TorqueScript is.
|
|
||||||
|
All Lua code is sandboxed, and file access is confined to the default directories in the same way TorqueScript is.
|
||||||
BlockLua also has access to any C libraries installed in the `modules/lualib` folder, so be careful throwing things in there.
|
BlockLua also has access to any C libraries installed in the `modules/lualib` folder, so be careful throwing things in there.
|
||||||
|
|
||||||
### Unsafe Mode
|
### Unsafe Mode
|
||||||
BlockLua-Unsafe.dll can be used in place of BlockLua.dll, to remove the sandboxing of Lua code. This allows Lua code to access any file and use any library, including ffi.
|
|
||||||
Please do not publish add-ons that require unsafe mode.
|
|
||||||
|
|
||||||
## Other Reference
|
BlockLua can be built in Unsafe Mode by specifying the `-DBLLUA_UNSAFE` compiler flag. This removes the sandboxing of Lua code, allowing it to access any file and use any library, including ffi.
|
||||||
|
A more limited option is `-DBLLUA_ALLOWFFI`, which allows the use of the `ffi` library. This can still be exploited to grant all the same access as full unsafe mode.
|
||||||
|
Please do not publish add-ons that require either of these.
|
||||||
|
|
||||||
### List of Object Types
|
## Compiling
|
||||||
'all' - Any object
|
|
||||||
'brick' - Bricks with Ray-Casting enabled
|
With any _32-bit_ variant of GCC installed (such as MinGW or MSYS2), run the following command in the repo directory:
|
||||||
'brickalways' - All bricks including those with Ray-Casting disabled
|
`g++ src/bllua4.cpp -o BlockLua.dll -m32 -shared -static-libgcc -Isrc -Iinc/tsfuncs -Iinc/lua -lpsapi -L. -llua5.1`
|
||||||
'player' - Players or bots
|
|
||||||
'item' - Items
|
LuaJIT (lua5.1.dll) can be obtained from https://luajit.org/
|
||||||
'vehicle' - Vehicles
|
|
||||||
'projectile' - Projectiles
|
|
||||||
Other types: 'static', 'environment', 'terrain', 'water', 'trigger', 'marker', 'gamebase', 'shapebase', 'camera', 'staticshape', 'vehicleblocker', 'explosion', 'corpse', 'debris', 'physicalzone', 'staticts', 'staticrendered', 'damagableitem'
|
|
||||||
|
|||||||
@@ -1,13 +1,16 @@
|
|||||||
// BlockLua (bllua4): Simple Lua interface for TorqueScript
|
// BlockLua (bllua4): Advanced Lua interface for TorqueScript
|
||||||
|
|
||||||
// Includes
|
// Includes
|
||||||
|
|
||||||
|
#ifndef WIN32_LEAN_AND_MEAN
|
||||||
|
#define WIN32_LEAN_AND_MEAN
|
||||||
|
#endif
|
||||||
#include <Windows.h>
|
#include <Windows.h>
|
||||||
#include <Psapi.h>
|
#include <Psapi.h>
|
||||||
|
|
||||||
#include "lua.hpp"
|
|
||||||
#include "BlHooks.cpp"
|
|
||||||
#include "BlFuncs.cpp"
|
#include "BlFuncs.cpp"
|
||||||
|
#include "BlHooks.cpp"
|
||||||
|
#include "lua.hpp"
|
||||||
|
|
||||||
#include "luainterp.cpp"
|
#include "luainterp.cpp"
|
||||||
#include "lualibts.cpp"
|
#include "lualibts.cpp"
|
||||||
@@ -28,8 +31,9 @@ INCLUDE_BIN(bll_fileLuaEnv , "lua-env.lua");
|
|||||||
INCLUDE_BIN(bll_fileTsEnv, "ts-env.cs");
|
INCLUDE_BIN(bll_fileTsEnv, "ts-env.cs");
|
||||||
INCLUDE_BIN(bll_fileLuaStd, "util/std.lua");
|
INCLUDE_BIN(bll_fileLuaStd, "util/std.lua");
|
||||||
INCLUDE_BIN(bll_fileLuaVector, "util/vector.lua");
|
INCLUDE_BIN(bll_fileLuaVector, "util/vector.lua");
|
||||||
INCLUDE_BIN(bll_fileLuaLibts , "util/libts.lua");
|
INCLUDE_BIN(bll_fileLuaMatrix, "util/matrix.lua");
|
||||||
INCLUDE_BIN(bll_fileTsLibtsSupport, "util/libts-support.cs");
|
INCLUDE_BIN(bll_fileLuaLibts, "util/libts-lua.lua");
|
||||||
|
INCLUDE_BIN(bll_fileTsLibts, "util/libts-ts.cs");
|
||||||
INCLUDE_BIN(bll_fileLuaLibbl, "util/libbl.lua");
|
INCLUDE_BIN(bll_fileLuaLibbl, "util/libbl.lua");
|
||||||
INCLUDE_BIN(bll_fileLuaLibblTypes, "util/libbl-types.lua");
|
INCLUDE_BIN(bll_fileLuaLibblTypes, "util/libbl-types.lua");
|
||||||
INCLUDE_BIN(bll_fileTsLibblSupport, "util/libbl-support.cs");
|
INCLUDE_BIN(bll_fileTsLibblSupport, "util/libbl-support.cs");
|
||||||
@@ -56,21 +60,27 @@ bool init() {
|
|||||||
|
|
||||||
// Set up Lua environment
|
// Set up Lua environment
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaEnv);
|
BLL_LOAD_LUA(gL, bll_fileLuaEnv);
|
||||||
|
#ifdef BLLUA_ALLOWFFI
|
||||||
|
lua_pushboolean(gL, true);
|
||||||
|
lua_setglobal(gL, "_bllua_allowffi");
|
||||||
|
#endif
|
||||||
#ifndef BLLUA_UNSAFE
|
#ifndef BLLUA_UNSAFE
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaEnvSafe);
|
BLL_LOAD_LUA(gL, bll_fileLuaEnvSafe);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Expose Lua API to TS
|
// Load utilities in Lua
|
||||||
BlAddFunction(NULL, NULL, "_bllua_luacall", bll_ts_luacall, "LuaCall(name, ...) - Call Lua function and return result", 2, 20);
|
|
||||||
BlEval(bll_fileTsEnv);
|
|
||||||
|
|
||||||
// Load utilities
|
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaStd);
|
BLL_LOAD_LUA(gL, bll_fileLuaStd);
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaVector);
|
BLL_LOAD_LUA(gL, bll_fileLuaVector);
|
||||||
|
BLL_LOAD_LUA(gL, bll_fileLuaMatrix);
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaLibts);
|
BLL_LOAD_LUA(gL, bll_fileLuaLibts);
|
||||||
BlEval(bll_fileTsLibtsSupport);
|
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaLibbl);
|
BLL_LOAD_LUA(gL, bll_fileLuaLibbl);
|
||||||
BLL_LOAD_LUA(gL, bll_fileLuaLibblTypes);
|
BLL_LOAD_LUA(gL, bll_fileLuaLibblTypes);
|
||||||
|
|
||||||
|
// Expose Lua API to TS
|
||||||
|
BlAddFunction(
|
||||||
|
NULL, NULL, "_bllua_luacall", bll_ts_luacall, "LuaCall(name, ...) - Call Lua function and return result", 2, 20);
|
||||||
|
BlEval(bll_fileTsEnv);
|
||||||
|
BlEval(bll_fileTsLibts);
|
||||||
BlEval(bll_fileTsLibblSupport);
|
BlEval(bll_fileTsLibblSupport);
|
||||||
BlEval(bll_fileLoadaddons);
|
BlEval(bll_fileLoadaddons);
|
||||||
|
|
||||||
@@ -83,8 +93,7 @@ bool init() {
|
|||||||
bool deinit() {
|
bool deinit() {
|
||||||
BlPrintf("BlockLua: Unloading");
|
BlPrintf("BlockLua: Unloading");
|
||||||
|
|
||||||
BlEval("deactivatePackage(_bllua_main);");
|
BlEval("$_bllua_active=0;deactivatePackage(_bllua_main);");
|
||||||
BlEval("$_bllua_active = 0;");
|
|
||||||
bll_LuaEval(gL, "for _,f in pairs(_bllua_on_unload) do f() end");
|
bll_LuaEval(gL, "for _,f in pairs(_bllua_on_unload) do f() end");
|
||||||
|
|
||||||
lua_close(gL);
|
lua_close(gL);
|
||||||
@@ -98,8 +107,11 @@ bool deinit() {
|
|||||||
|
|
||||||
bool __stdcall DllMain(HINSTANCE hinstance, DWORD reason, void* reserved) {
|
bool __stdcall DllMain(HINSTANCE hinstance, DWORD reason, void* reserved) {
|
||||||
switch (reason) {
|
switch (reason) {
|
||||||
case DLL_PROCESS_ATTACH: return init();
|
case DLL_PROCESS_ATTACH:
|
||||||
case DLL_PROCESS_DETACH: return deinit();
|
return init();
|
||||||
default : return true;
|
case DLL_PROCESS_DETACH:
|
||||||
|
return deinit();
|
||||||
|
default:
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,7 +4,9 @@
|
|||||||
|
|
||||||
|
|
||||||
-- Utility: Convert a list of strings into a map of string->true
|
-- Utility: Convert a list of strings into a map of string->true
|
||||||
local function tmap(t) local u = {}; for _, n in ipairs(t) do u[n] = true end; return u; end
|
local function tmap(t)
|
||||||
|
local u = {}; for _, n in ipairs(t) do u[n] = true end; return u;
|
||||||
|
end
|
||||||
|
|
||||||
-- Save banned global variables for wrapping with safe functions
|
-- Save banned global variables for wrapping with safe functions
|
||||||
local old_io = io
|
local old_io = io
|
||||||
@@ -12,6 +14,7 @@ local old_require = require
|
|||||||
local old_os = os
|
local old_os = os
|
||||||
local old_debug = debug
|
local old_debug = debug
|
||||||
local old_package = package
|
local old_package = package
|
||||||
|
local old_allowffi = _bllua_allowffi
|
||||||
|
|
||||||
-- Remove all global variables except a whitelist
|
-- Remove all global variables except a whitelist
|
||||||
local ok_names = tmap {
|
local ok_names = tmap {
|
||||||
@@ -37,13 +40,10 @@ end
|
|||||||
|
|
||||||
-- Sanitize file paths to point only to allowed files within the game directory
|
-- Sanitize file paths to point only to allowed files within the game directory
|
||||||
-- List of allowed directories for reading/writing
|
-- List of allowed directories for reading/writing
|
||||||
|
-- modules/lualib is also allowed as read-only
|
||||||
local allowed_dirs = tmap {
|
local allowed_dirs = tmap {
|
||||||
'add-ons', 'base', 'config', 'saves', 'screenshots', 'shaders'
|
'add-ons', 'base', 'config', 'saves', 'screenshots', 'shaders'
|
||||||
}
|
}
|
||||||
-- List of allowed directories for reading only
|
|
||||||
local allowed_dirs_readonly = tmap {
|
|
||||||
'lualib'
|
|
||||||
}
|
|
||||||
-- List of disallowed file extensions - basically executable file extensions
|
-- List of disallowed file extensions - basically executable file extensions
|
||||||
-- Note that even without this protection, exploiting would still require somehow
|
-- Note that even without this protection, exploiting would still require somehow
|
||||||
-- getting a file within the allowed directories to autorun,
|
-- getting a file within the allowed directories to autorun,
|
||||||
@@ -62,6 +62,9 @@ local disallowed_exts = tmap {
|
|||||||
-- Return: clean file path if allowed (or nil if disallowed),
|
-- Return: clean file path if allowed (or nil if disallowed),
|
||||||
-- error string (or nil if allowed)
|
-- error string (or nil if allowed)
|
||||||
local function safe_path(fn, readonly)
|
local function safe_path(fn, readonly)
|
||||||
|
if type(fn) ~= 'string' then
|
||||||
|
return nil, 'Filename must be a string'
|
||||||
|
end
|
||||||
fn = fn:gsub('\\', '/')
|
fn = fn:gsub('\\', '/')
|
||||||
fn = fn:gsub('^ +', '')
|
fn = fn:gsub('^ +', '')
|
||||||
fn = fn:gsub(' +$', '')
|
fn = fn:gsub(' +$', '')
|
||||||
@@ -79,14 +82,15 @@ local function safe_path(fn, readonly)
|
|||||||
end
|
end
|
||||||
-- allow only whitelisted dirs
|
-- allow only whitelisted dirs
|
||||||
local dir = fn:match('^([^/]+)/')
|
local dir = fn:match('^([^/]+)/')
|
||||||
if (not dir) or (
|
if not (dir and (
|
||||||
(not allowed_dirs[dir:lower()]) and
|
allowed_dirs[dir:lower()] or
|
||||||
((not readonly) or (not allowed_dirs_readonly[dir:lower()])) ) then
|
(readonly and fn:find('^modules/lualib/'))))
|
||||||
return nil, 'filename is in disallowed directory '..(dir or 'nil')
|
then
|
||||||
|
return nil, 'File is in disallowed directory ' .. (dir or 'nil')
|
||||||
end
|
end
|
||||||
-- disallow blacklisted extensions or no extension
|
-- disallow blacklisted extensions
|
||||||
local ext = fn:match('%.([^/%.]+)$')
|
local ext = fn:match('%.([^/%.]+)$')
|
||||||
if (not ext) or (disallowed_exts[ext:lower()]) then
|
if ext and disallowed_exts[ext:lower()] then
|
||||||
return nil, 'Filename \'' .. fn .. '\' has disallowed extension \'' ..
|
return nil, 'Filename \'' .. fn .. '\' has disallowed extension \'' ..
|
||||||
(ext or '') .. '\''
|
(ext or '') .. '\''
|
||||||
end
|
end
|
||||||
@@ -104,6 +108,7 @@ function _bllua_io_open(fn, md)
|
|||||||
return nil, err
|
return nil, err
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Allow io.type (works on file handles returned by io.open)
|
-- Allow io.type (works on file handles returned by io.open)
|
||||||
function _bllua_io_type(f)
|
function _bllua_io_type(f)
|
||||||
return old_io.type(f)
|
return old_io.type(f)
|
||||||
@@ -117,6 +122,7 @@ local disallowed_packages = tmap {
|
|||||||
'ffi', 'debug', 'package', 'io', 'os',
|
'ffi', 'debug', 'package', 'io', 'os',
|
||||||
'_bllua_ts',
|
'_bllua_ts',
|
||||||
}
|
}
|
||||||
|
if old_allowffi then disallowed_packages['ffi'] = nil end
|
||||||
function _bllua_requiresecure(name)
|
function _bllua_requiresecure(name)
|
||||||
if name:find('[^a-zA-Z0-9_%-%.]') or name:find('%.%.') or
|
if name:find('[^a-zA-Z0-9_%-%.]') or name:find('%.%.') or
|
||||||
name:find('^%.') or name:find('%.$') then
|
name:find('^%.') or name:find('%.$') then
|
||||||
@@ -128,6 +134,7 @@ function _bllua_requiresecure(name)
|
|||||||
return old_require(name)
|
return old_require(name)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
package = {
|
package = {
|
||||||
seeall = old_package.seeall,
|
seeall = old_package.seeall,
|
||||||
}
|
}
|
||||||
@@ -139,4 +146,5 @@ debug = {
|
|||||||
getfilename = old_debug.getfilename, -- defined in lua.env.lua
|
getfilename = old_debug.getfilename, -- defined in lua.env.lua
|
||||||
}
|
}
|
||||||
|
|
||||||
_bllua_ts.echo(' Executed bllua-env-safe.lua')
|
print = _bllua_ts.echo
|
||||||
|
print(' Executed bllua-env-safe.lua')
|
||||||
|
|||||||
@@ -18,6 +18,8 @@ end
|
|||||||
|
|
||||||
-- Called when pcall fails on a ts->lua call, used to print detailed error info
|
-- Called when pcall fails on a ts->lua call, used to print detailed error info
|
||||||
function _bllua_on_error(err)
|
function _bllua_on_error(err)
|
||||||
|
-- Convert error to string if it's not already
|
||||||
|
err = tostring(err)
|
||||||
err = err:match(': (.+)$') or err
|
err = err:match(': (.+)$') or err
|
||||||
local tracelines = { err }
|
local tracelines = { err }
|
||||||
local level = 2
|
local level = 2
|
||||||
@@ -25,7 +27,7 @@ function _bllua_on_error(err)
|
|||||||
local info = debug.getinfo(level)
|
local info = debug.getinfo(level)
|
||||||
if not info then break end
|
if not info then break end
|
||||||
local filename = debug.getfilename(level) or info.short_src
|
local filename = debug.getfilename(level) or info.short_src
|
||||||
local funcname = info.name
|
local funcname = info.name or '<unknown>'
|
||||||
if funcname == 'dofile' then break end
|
if funcname == 'dofile' then break end
|
||||||
table.insert(tracelines, string.format('%s:%s in function \'%s\'',
|
table.insert(tracelines, string.format('%s:%s in function \'%s\'',
|
||||||
filename,
|
filename,
|
||||||
@@ -37,4 +39,9 @@ function _bllua_on_error(err)
|
|||||||
return table.concat(tracelines, '\n')
|
return table.concat(tracelines, '\n')
|
||||||
end
|
end
|
||||||
|
|
||||||
_bllua_ts.echo(' Executed bllua-env.lua')
|
-- overridden in lua-env-safe.lua (executed if not BLLUA_UNSAFE)
|
||||||
|
_bllua_io_open = io.open
|
||||||
|
_bllua_requiresecure = require
|
||||||
|
|
||||||
|
print = _bllua_ts.echo
|
||||||
|
print(' Executed bllua-env.lua')
|
||||||
|
|||||||
@@ -1,5 +1,11 @@
|
|||||||
|
|
||||||
// Handle errors with a Lua function, defined in lua-env.lua
|
// Handle errors with a Lua function, defined in lua-env.lua
|
||||||
|
#include "luainterp.hpp"
|
||||||
|
#include "BlHooks.hpp"
|
||||||
|
#include "lauxlib.h"
|
||||||
|
#include "lua.h"
|
||||||
|
#include <cstring>
|
||||||
|
|
||||||
int bll_error_handler(lua_State* L) {
|
int bll_error_handler(lua_State* L) {
|
||||||
lua_getfield(L, LUA_GLOBALSINDEX, "_bllua_on_error");
|
lua_getfield(L, LUA_GLOBALSINDEX, "_bllua_on_error");
|
||||||
if (!lua_isfunction(L, -1)) {
|
if (!lua_isfunction(L, -1)) {
|
||||||
@@ -15,6 +21,7 @@ int bll_error_handler(lua_State *L) {
|
|||||||
lua_pcall(L, 1, 1, 0);
|
lua_pcall(L, 1, 1, 0);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int bll_pcall(lua_State* L, int nargs, int nret) {
|
int bll_pcall(lua_State* L, int nargs, int nret) {
|
||||||
// calculate stack position for message handler
|
// calculate stack position for message handler
|
||||||
int hpos = lua_gettop(L) - nargs;
|
int hpos = lua_gettop(L) - nargs;
|
||||||
@@ -48,21 +55,27 @@ bool bll_LuaEval(lua_State* L, const char* str) {
|
|||||||
|
|
||||||
// Convert a Lua stack entry into a string for providing to TS
|
// Convert a Lua stack entry into a string for providing to TS
|
||||||
// Use static buffer to avoid excessive malloc
|
// Use static buffer to avoid excessive malloc
|
||||||
#define BLL_ARG_COUNT 20
|
|
||||||
#define BLL_ARG_MAX 8192
|
|
||||||
char bll_arg_buffer[BLL_ARG_COUNT][BLL_ARG_MAX];
|
char bll_arg_buffer[BLL_ARG_COUNT][BLL_ARG_MAX];
|
||||||
|
|
||||||
bool bll_toarg(lua_State* L, char* buf, int i, bool err) {
|
bool bll_toarg(lua_State* L, char* buf, int i, bool err) {
|
||||||
if (lua_isstring(L, i)) {
|
if (lua_isstring(L, i)) {
|
||||||
const char* str = lua_tostring(L, i);
|
const char* str = lua_tostring(L, i);
|
||||||
if (strlen(str) >= BLL_ARG_MAX) {
|
if (strlen(str) >= BLL_ARG_MAX) {
|
||||||
if(err) luaL_error(L, "argument to TS is too long - max length is 8192");
|
if (err)
|
||||||
|
luaL_error(L, "argument to TS is too long - max length is 8192");
|
||||||
return true;
|
return true;
|
||||||
} else {
|
} else {
|
||||||
strcpy(buf, str);
|
#ifdef _MSC_VER
|
||||||
|
strncpy_s(buf, BLL_ARG_MAX, str, _TRUNCATE);
|
||||||
|
#else
|
||||||
|
strncpy(buf, str, BLL_ARG_MAX - 1);
|
||||||
|
buf[BLL_ARG_MAX - 1] = '\0';
|
||||||
|
#endif
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if(err) luaL_error(L, "argument to TS must be a string");
|
if (err)
|
||||||
|
luaL_error(L, "argument to TS must be a string");
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
17
src/luainterp.hpp
Normal file
17
src/luainterp.hpp
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
// Shared declarations for Lua <-> TS argument handling
|
||||||
|
#ifndef _H_LUAINTERP_SHARED
|
||||||
|
#define _H_LUAINTERP_SHARED
|
||||||
|
|
||||||
|
#include "lua.h"
|
||||||
|
|
||||||
|
#define BLL_ARG_COUNT 20
|
||||||
|
#define BLL_ARG_MAX 8192
|
||||||
|
|
||||||
|
extern char bll_arg_buffer[BLL_ARG_COUNT][BLL_ARG_MAX];
|
||||||
|
bool bll_toarg(lua_State* L, char* buf, int i, bool err);
|
||||||
|
int bll_pcall(lua_State* L, int nargs, int nret);
|
||||||
|
void bll_printError(lua_State* L, const char* operation, const char* item);
|
||||||
|
|
||||||
|
extern lua_State* gL;
|
||||||
|
|
||||||
|
#endif
|
||||||
256
src/lualibts.cpp
256
src/lualibts.cpp
@@ -3,14 +3,24 @@
|
|||||||
// TS -> Lua API
|
// TS -> Lua API
|
||||||
|
|
||||||
// Call a TS function from Lua, push the result to Lua stack
|
// Call a TS function from Lua, push the result to Lua stack
|
||||||
|
#include "BlFuncs.hpp"
|
||||||
|
#include "BlHooks.hpp"
|
||||||
|
#include "lauxlib.h"
|
||||||
|
#include "lua.h"
|
||||||
|
#include "luainterp.hpp"
|
||||||
|
|
||||||
int bll_TsCall(lua_State* L, const char* oname, const char* fname, int argc, int ofs) {
|
int bll_TsCall(lua_State* L, const char* oname, const char* fname, int argc, int ofs) {
|
||||||
ADDR obj = (ADDR)NULL;
|
ADDR obj = (ADDR)NULL;
|
||||||
if (oname) {
|
if (oname) {
|
||||||
obj = BlObject(oname);
|
obj = BlObject(oname);
|
||||||
if(!obj) { return luaL_error(L, "Lua->TS call: Object not found"); }
|
if (!obj) {
|
||||||
|
return luaL_error(L, "Lua->TS call: Object not found");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if(argc > BLL_ARG_COUNT) { return luaL_error(L, "Lua->TS call: Too many arguments (Max is 19)"); }
|
if (argc > BLL_ARG_COUNT) {
|
||||||
|
return luaL_error(L, "Lua->TS call: Too many arguments (Max is 19)");
|
||||||
|
}
|
||||||
|
|
||||||
char* argv[BLL_ARG_COUNT];
|
char* argv[BLL_ARG_COUNT];
|
||||||
for (int i = 0; i < argc; i++) {
|
for (int i = 0; i < argc; i++) {
|
||||||
@@ -23,52 +33,174 @@ int bll_TsCall(lua_State* L, const char* oname, const char* fname, int argc, int
|
|||||||
const char* res;
|
const char* res;
|
||||||
if (obj) {
|
if (obj) {
|
||||||
switch (argc) {
|
switch (argc) {
|
||||||
case 0: res = BlCallObj(obj, fname); break; // no idea why this happens sometimes, it shouldnt be possible
|
case 0:
|
||||||
case 1: res = BlCallObj(obj, fname); break;
|
res = BlCallObj(obj, fname);
|
||||||
case 2: res = BlCallObj(obj, fname, argv[0]); break;
|
break; // no idea why this happens sometimes, it shouldnt be possible
|
||||||
case 3: res = BlCallObj(obj, fname, argv[0], argv[1]); break;
|
case 1:
|
||||||
case 4: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2]); break;
|
res = BlCallObj(obj, fname);
|
||||||
case 5: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3]); break;
|
break;
|
||||||
case 6: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4]); break;
|
case 2:
|
||||||
case 7: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]); break;
|
res = BlCallObj(obj, fname, argv[0]);
|
||||||
case 8: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6]); break;
|
break;
|
||||||
case 9: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7]); break;
|
case 3:
|
||||||
case 10: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8]); break;
|
res = BlCallObj(obj, fname, argv[0], argv[1]);
|
||||||
case 11: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9]); break;
|
break;
|
||||||
case 12: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10]); break;
|
case 4:
|
||||||
case 13: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11]); break;
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2]);
|
||||||
case 14: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12]); break;
|
break;
|
||||||
case 15: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13]); break;
|
case 5:
|
||||||
case 16: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14]); break;
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3]);
|
||||||
case 17: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15]); break;
|
break;
|
||||||
case 18: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16]); break;
|
case 6:
|
||||||
case 19: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17]); break;
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4]);
|
||||||
case 20: res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18]); break;
|
break;
|
||||||
default: res = ""; luaL_error(L, "Lua->TS object call: Too many arguments (Max is 19)");
|
case 7:
|
||||||
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]);
|
||||||
|
break;
|
||||||
|
case 8:
|
||||||
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6]);
|
||||||
|
break;
|
||||||
|
case 9:
|
||||||
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7]);
|
||||||
|
break;
|
||||||
|
case 10:
|
||||||
|
res = BlCallObj(obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8]);
|
||||||
|
break;
|
||||||
|
case 11:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9]);
|
||||||
|
break;
|
||||||
|
case 12:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10]);
|
||||||
|
break;
|
||||||
|
case 13:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11]);
|
||||||
|
break;
|
||||||
|
case 14:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12]);
|
||||||
|
break;
|
||||||
|
case 15:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13]);
|
||||||
|
break;
|
||||||
|
case 16:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13], argv[14]);
|
||||||
|
break;
|
||||||
|
case 17:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13], argv[14], argv[15]);
|
||||||
|
break;
|
||||||
|
case 18:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16]);
|
||||||
|
break;
|
||||||
|
case 19:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17]);
|
||||||
|
break;
|
||||||
|
case 20:
|
||||||
|
res = BlCallObj(
|
||||||
|
obj, fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9],
|
||||||
|
argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18]);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
res = "";
|
||||||
|
luaL_error(L, "Lua->TS object call: Too many arguments (Max is 19)");
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
switch (argc) {
|
switch (argc) {
|
||||||
case 0: res = BlCall(fname); break;
|
case 0:
|
||||||
case 1: res = BlCall(fname, argv[0]); break;
|
res = BlCall(fname);
|
||||||
case 2: res = BlCall(fname, argv[0], argv[1]); break;
|
break;
|
||||||
case 3: res = BlCall(fname, argv[0], argv[1], argv[2]); break;
|
case 1:
|
||||||
case 4: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3]); break;
|
res = BlCall(fname, argv[0]);
|
||||||
case 5: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4]); break;
|
break;
|
||||||
case 6: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]); break;
|
case 2:
|
||||||
case 7: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6]); break;
|
res = BlCall(fname, argv[0], argv[1]);
|
||||||
case 8: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7]); break;
|
break;
|
||||||
case 9: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8]); break;
|
case 3:
|
||||||
case 10: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9]); break;
|
res = BlCall(fname, argv[0], argv[1], argv[2]);
|
||||||
case 11: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10]); break;
|
break;
|
||||||
case 12: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11]); break;
|
case 4:
|
||||||
case 13: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12]); break;
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3]);
|
||||||
case 14: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13]); break;
|
break;
|
||||||
case 15: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14]); break;
|
case 5:
|
||||||
case 16: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15]); break;
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4]);
|
||||||
case 17: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16]); break;
|
break;
|
||||||
case 18: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17]); break;
|
case 6:
|
||||||
case 19: res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10], argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18]); break;
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5]);
|
||||||
default: res = ""; luaL_error(L, "Lua->TS call: Too many arguments (Max is 19)");
|
break;
|
||||||
|
case 7:
|
||||||
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6]);
|
||||||
|
break;
|
||||||
|
case 8:
|
||||||
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7]);
|
||||||
|
break;
|
||||||
|
case 9:
|
||||||
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8]);
|
||||||
|
break;
|
||||||
|
case 10:
|
||||||
|
res = BlCall(fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9]);
|
||||||
|
break;
|
||||||
|
case 11:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10]);
|
||||||
|
break;
|
||||||
|
case 12:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11]);
|
||||||
|
break;
|
||||||
|
case 13:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12]);
|
||||||
|
break;
|
||||||
|
case 14:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13]);
|
||||||
|
break;
|
||||||
|
case 15:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13], argv[14]);
|
||||||
|
break;
|
||||||
|
case 16:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13], argv[14], argv[15]);
|
||||||
|
break;
|
||||||
|
case 17:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13], argv[14], argv[15], argv[16]);
|
||||||
|
break;
|
||||||
|
case 18:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17]);
|
||||||
|
break;
|
||||||
|
case 19:
|
||||||
|
res = BlCall(
|
||||||
|
fname, argv[0], argv[1], argv[2], argv[3], argv[4], argv[5], argv[6], argv[7], argv[8], argv[9], argv[10],
|
||||||
|
argv[11], argv[12], argv[13], argv[14], argv[15], argv[16], argv[17], argv[18]);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
res = "";
|
||||||
|
luaL_error(L, "Lua->TS call: Too many arguments (Max is 19)");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -76,19 +208,23 @@ int bll_TsCall(lua_State* L, const char* oname, const char* fname, int argc, int
|
|||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lua lib function: ts.call
|
// Lua lib function: ts.call
|
||||||
int bll_lua_tscall(lua_State* L) {
|
int bll_lua_tscall(lua_State* L) {
|
||||||
int argc = lua_gettop(L) - 1; // number of arguments after function name
|
int argc = lua_gettop(L) - 1; // number of arguments after function name
|
||||||
if(argc < 0) return luaL_error(L, "_bllua_ts.call: Must provide a function name");
|
if (argc < 0)
|
||||||
|
return luaL_error(L, "_bllua_ts.call: Must provide a function name");
|
||||||
|
|
||||||
const char* fname = luaL_checkstring(L, 1);
|
const char* fname = luaL_checkstring(L, 1);
|
||||||
|
|
||||||
return bll_TsCall(L, NULL, fname, argc, 1);
|
return bll_TsCall(L, NULL, fname, argc, 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lua lib function: ts.callobj
|
// Lua lib function: ts.callobj
|
||||||
int bll_lua_tscallobj(lua_State* L) {
|
int bll_lua_tscallobj(lua_State* L) {
|
||||||
int argc = lua_gettop(L) - 2; // number of arguments after function name and object?
|
int argc = lua_gettop(L) - 2; // number of arguments after function name and object?
|
||||||
if(argc < 0) return luaL_error(L, "_bllua_ts.callobj: Must provide an object and function name");
|
if (argc < 0)
|
||||||
|
return luaL_error(L, "_bllua_ts.callobj: Must provide an object and function name");
|
||||||
|
|
||||||
const char* oname = luaL_checkstring(L, 1);
|
const char* oname = luaL_checkstring(L, 1);
|
||||||
const char* fname = luaL_checkstring(L, 2);
|
const char* fname = luaL_checkstring(L, 2);
|
||||||
@@ -110,19 +246,24 @@ int bll_lua_tsgetfield(lua_State* L) {
|
|||||||
const char* oname = luaL_checkstring(L, 1);
|
const char* oname = luaL_checkstring(L, 1);
|
||||||
const char* vname = luaL_checkstring(L, 2);
|
const char* vname = luaL_checkstring(L, 2);
|
||||||
ADDR obj = BlObject(oname);
|
ADDR obj = BlObject(oname);
|
||||||
if(!obj) { return luaL_error(L, "_bllua_ts.getfield: Object not found"); }
|
if (!obj) {
|
||||||
|
return luaL_error(L, "_bllua_ts.getfield: Object not found");
|
||||||
|
}
|
||||||
|
|
||||||
const char* val = BlGetField(obj, vname, NULL);
|
const char* val = BlGetField(obj, vname, NULL);
|
||||||
lua_pushstring(L, val);
|
lua_pushstring(L, val);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lua lib function: ts.setfield
|
// Lua lib function: ts.setfield
|
||||||
int bll_lua_tssetfield(lua_State* L) {
|
int bll_lua_tssetfield(lua_State* L) {
|
||||||
const char* oname = luaL_checkstring(L, 1);
|
const char* oname = luaL_checkstring(L, 1);
|
||||||
const char* vname = luaL_checkstring(L, 2);
|
const char* vname = luaL_checkstring(L, 2);
|
||||||
const char* val = luaL_checkstring(L, 3);
|
const char* val = luaL_checkstring(L, 3);
|
||||||
ADDR obj = BlObject(oname);
|
ADDR obj = BlObject(oname);
|
||||||
if(!obj) { return luaL_error(L, "_bllua_ts.setfield: Object not found"); }
|
if (!obj) {
|
||||||
|
return luaL_error(L, "_bllua_ts.setfield: Object not found");
|
||||||
|
}
|
||||||
|
|
||||||
BlSetField(obj, vname, NULL, val);
|
BlSetField(obj, vname, NULL, val);
|
||||||
return 0;
|
return 0;
|
||||||
@@ -145,15 +286,10 @@ int bll_lua_tsecho(lua_State* L) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const luaL_Reg bll_lua_reg[] = {
|
const luaL_Reg bll_lua_reg[] = {
|
||||||
{"call" , bll_lua_tscall },
|
{"call", bll_lua_tscall}, {"callobj", bll_lua_tscallobj},
|
||||||
{"callobj" , bll_lua_tscallobj },
|
{"getvar", bll_lua_tsgetvar}, {"getfield", bll_lua_tsgetfield},
|
||||||
{"getvar" , bll_lua_tsgetvar },
|
{"setfield", bll_lua_tssetfield}, {"eval", bll_lua_tseval},
|
||||||
{"getfield", bll_lua_tsgetfield},
|
{"echo", bll_lua_tsecho}, {NULL, NULL},
|
||||||
{"setfield", bll_lua_tssetfield},
|
|
||||||
{"eval" , bll_lua_tseval },
|
|
||||||
{"echo" , bll_lua_tsecho },
|
|
||||||
{NULL, NULL},
|
|
||||||
};
|
};
|
||||||
void llibbl_init(lua_State* L) {
|
|
||||||
luaL_register(L, "_bllua_ts", bll_lua_reg);
|
void llibbl_init(lua_State* L) { luaL_register(L, "_bllua_ts", bll_lua_reg); }
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,26 +0,0 @@
|
|||||||
// Built-in functions
|
|
||||||
// Eval'd after BLLua4 has loaded the Lua environment and API
|
|
||||||
|
|
||||||
// Public Lua library for TS
|
|
||||||
function luacall(%func, %a,%b,%c,%d,%e,%f,%g,%h,%i,%j,%k,%l,%m,%n,%o,%p) {
|
|
||||||
if($_bllua_active)
|
|
||||||
return _bllua_luacall(%func, %a,%b,%c,%d,%e,%f,%g,%h,%i,%j,%k,%l,%m,%n,%o,%p);
|
|
||||||
}
|
|
||||||
function luaexec(%fn) {
|
|
||||||
if($_bllua_active)
|
|
||||||
return _bllua_luacall("_bllua_exec", %fn);
|
|
||||||
}
|
|
||||||
function luaeval(%code) {
|
|
||||||
if($_bllua_active)
|
|
||||||
return _bllua_luacall("_bllua_eval", %code);
|
|
||||||
}
|
|
||||||
function luaget(%name) {
|
|
||||||
if($_bllua_active)
|
|
||||||
return _bllua_luacall("_bllua_getvar", %name);
|
|
||||||
}
|
|
||||||
function luaset(%name, %val) {
|
|
||||||
if($_bllua_active)
|
|
||||||
_bllua_luacall("_bllua_setvar", %name, %val);
|
|
||||||
}
|
|
||||||
|
|
||||||
echo(" Executed bllua-env.cs");
|
|
||||||
|
|||||||
@@ -1,5 +1,10 @@
|
|||||||
|
|
||||||
// Call a Lua function from TS, return true if success - result will be on Lua stack
|
// Call a Lua function from TS, return true if success - result will be on Lua
|
||||||
|
// stack
|
||||||
|
#include "BlFuncs.hpp"
|
||||||
|
#include "BlHooks.hpp"
|
||||||
|
#include "luainterp.hpp"
|
||||||
|
|
||||||
bool bll_LuaCall(const char* fname, int argc, const char* argv[]) {
|
bool bll_LuaCall(const char* fname, int argc, const char* argv[]) {
|
||||||
lua_getglobal(gL, fname);
|
lua_getglobal(gL, fname);
|
||||||
for (int i = 0; i < argc; i++) {
|
for (int i = 0; i < argc; i++) {
|
||||||
@@ -14,9 +19,12 @@ bool bll_LuaCall(const char* fname, int argc, const char* argv[]) {
|
|||||||
|
|
||||||
// TS lib function: luacall
|
// TS lib function: luacall
|
||||||
const char* bll_ts_luacall(ADDR obj, int argc, const char* argv[]) {
|
const char* bll_ts_luacall(ADDR obj, int argc, const char* argv[]) {
|
||||||
if(argc<2) return "";
|
if (argc < 2)
|
||||||
|
return "";
|
||||||
|
|
||||||
if(!bll_LuaCall(argv[1], argc-2, &argv[2])) { return ""; }
|
if (!bll_LuaCall(argv[1], argc - 2, &argv[2])) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
char* retbuf = BlReturnBuffer(BLL_ARG_MAX);
|
char* retbuf = BlReturnBuffer(BLL_ARG_MAX);
|
||||||
bll_toarg(gL, retbuf, -1, false); // provide returned value to ts
|
bll_toarg(gL, retbuf, -1, false); // provide returned value to ts
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
|
|
||||||
// Private - Utilities used by libbl from the Lua side
|
// Private - Utilities used by libbl from the Lua side
|
||||||
|
|
||||||
package _bllua_smartEval {
|
package smartEval {
|
||||||
// Allow prepending ' to console commands to eval in lua instead of TS
|
// Allow prepending ' to console commands to eval in lua instead of TS
|
||||||
// Also wraps TS lines with echo(...); if they don't end in ; or }
|
// Also wraps TS lines with echo(...); if they don't end in ; or }
|
||||||
function ConsoleEntry::eval() {
|
function ConsoleEntry::eval() {
|
||||||
@@ -10,7 +10,7 @@ package _bllua_smartEval {
|
|||||||
if($_bllua_active) {
|
if($_bllua_active) {
|
||||||
%text = getSubStr(%text, 1, strLen(%text));
|
%text = getSubStr(%text, 1, strLen(%text));
|
||||||
echo("Lua ==> " @ %text);
|
echo("Lua ==> " @ %text);
|
||||||
luacall("_bllua_smarteval", %text);
|
_bllua_luacall("_bllua_smarteval", %text);
|
||||||
} else {
|
} else {
|
||||||
echo("Lua: not loaded");
|
echo("Lua: not loaded");
|
||||||
}
|
}
|
||||||
@@ -28,7 +28,7 @@ package _bllua_smartEval {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
activatePackage(_bllua_smartEval);
|
activatePackage(smartEval);
|
||||||
|
|
||||||
package _bllua_objectDeletionHook {
|
package _bllua_objectDeletionHook {
|
||||||
// Hook object deletion to clean up its lua data
|
// Hook object deletion to clean up its lua data
|
||||||
@@ -36,9 +36,32 @@ package _bllua_objectDeletionHook {
|
|||||||
// note: no parent function exists by default,
|
// note: no parent function exists by default,
|
||||||
// and this is loaded before any addons
|
// and this is loaded before any addons
|
||||||
//parent::onRemove(%obj);
|
//parent::onRemove(%obj);
|
||||||
if($_bllua_active) luacall("_bllua_objectDeleted", %obj);
|
// assuming obj is an ID and never a name
|
||||||
|
if($_bllua_active) _bllua_luacall("_bllua_objectDeleted", %obj);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
activatePackage(_bllua_objectDeletionHook);
|
activatePackage(_bllua_objectDeletionHook);
|
||||||
|
|
||||||
|
// Public Lua library for TS
|
||||||
|
function luacall(%func, %a,%b,%c,%d,%e,%f,%g,%h,%i,%j,%k,%l,%m,%n,%o,%p) {
|
||||||
|
if($_bllua_active)
|
||||||
|
return _bllua_luacall("_bllua_call", %func, %a,%b,%c,%d,%e,%f,%g,%h,%i,%j,%k,%l,%m,%n,%o,%p);
|
||||||
|
}
|
||||||
|
function luaexec(%fn) {
|
||||||
|
if($_bllua_active)
|
||||||
|
return _bllua_luacall("_bllua_exec", %fn);
|
||||||
|
}
|
||||||
|
function luaeval(%code) {
|
||||||
|
if($_bllua_active)
|
||||||
|
return _bllua_luacall("_bllua_eval", %code);
|
||||||
|
}
|
||||||
|
function luaget(%name) {
|
||||||
|
if($_bllua_active)
|
||||||
|
return _bllua_luacall("_bllua_getvar", %name);
|
||||||
|
}
|
||||||
|
function luaset(%name, %val) {
|
||||||
|
if($_bllua_active)
|
||||||
|
_bllua_luacall("_bllua_setvar", %name, %val);
|
||||||
|
}
|
||||||
|
|
||||||
echo(" Executed libbl-support.cs");
|
echo(" Executed libbl-support.cs");
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
231
src/util/libts-lua.lua
Normal file
231
src/util/libts-lua.lua
Normal file
@@ -0,0 +1,231 @@
|
|||||||
|
-- This Lua code provides some built-in utilities for writing Lua add-ons
|
||||||
|
-- It is eval'd automatically once BLLua3 has loaded the TS API and environment
|
||||||
|
-- It only has access to the sandboxed lua environment, just like user code.
|
||||||
|
|
||||||
|
ts = _bllua_ts
|
||||||
|
|
||||||
|
-- Provide limited OS functions
|
||||||
|
os = os or {}
|
||||||
|
---@diagnostic disable-next-line: duplicate-set-field
|
||||||
|
function os.time() return math.floor(tonumber(_bllua_ts.call('getSimTime')) / 1000) end
|
||||||
|
|
||||||
|
---@diagnostic disable-next-line: duplicate-set-field
|
||||||
|
function os.clock() return tonumber(_bllua_ts.call('getSimTime')) / 1000 end
|
||||||
|
|
||||||
|
-- Virtual file class, emulating a file object as returned by io.open
|
||||||
|
-- Used to wrap io.open to allow reading from zips (using TS)
|
||||||
|
-- Not perfect because TS file I/O sucks
|
||||||
|
-- Can't read nulls, can't distinguish between CRLF and LF.
|
||||||
|
-- Todo someday: actually read the zip in lua?
|
||||||
|
local file_meta = {
|
||||||
|
read = function(file, mode)
|
||||||
|
file:_init()
|
||||||
|
if not file or type(file) ~= 'table' or not file._is_file then error('File:read: Not a file', 2) end
|
||||||
|
if file._is_open ~= true then error('File:read: File is closed', 2) end
|
||||||
|
if mode == '*n' then
|
||||||
|
local ws, n = file.data:match('^([ \t\r\n]*)([0-9%.%-e]+)', file.pos)
|
||||||
|
if n then
|
||||||
|
file.pos = file.pos + #ws + #n
|
||||||
|
return n
|
||||||
|
else
|
||||||
|
return nil
|
||||||
|
end
|
||||||
|
elseif mode == '*a' then
|
||||||
|
local d = file.data:sub(file.pos, #file.data)
|
||||||
|
file.pos = #file.data + 1
|
||||||
|
return d
|
||||||
|
elseif mode == '*l' then
|
||||||
|
local l, ws = file.data:match('^([^\r\n]*)(\r?\n)', file.pos)
|
||||||
|
if not l then
|
||||||
|
l = file.data:match('^([^\r\n]*)$', file.pos); ws = '';
|
||||||
|
if l == '' then return nil end
|
||||||
|
end
|
||||||
|
if l then
|
||||||
|
file.pos = file.pos + #l + #ws
|
||||||
|
return l
|
||||||
|
else
|
||||||
|
return nil
|
||||||
|
end
|
||||||
|
elseif type(mode) == 'number' then
|
||||||
|
local d = file.data:sub(file.pos, file.pos + mode)
|
||||||
|
file.pos = file.pos + #d
|
||||||
|
return d
|
||||||
|
else
|
||||||
|
error('File:read: Invalid mode \'' .. mode .. '\'', 2)
|
||||||
|
end
|
||||||
|
end,
|
||||||
|
lines = function(file)
|
||||||
|
file:_init()
|
||||||
|
return function()
|
||||||
|
return file:read('*l')
|
||||||
|
end
|
||||||
|
end,
|
||||||
|
close = function(file)
|
||||||
|
if not file._is_open then error('File:close: File is not open', 2) end
|
||||||
|
file._is_open = false
|
||||||
|
end,
|
||||||
|
__index = function(f, k) return rawget(f, k) or getmetatable(f)[k] end,
|
||||||
|
_init = function(f)
|
||||||
|
if not f.data then
|
||||||
|
f.data = _bllua_ts.call('_bllua_ReadEntireFile', f.filename)
|
||||||
|
end
|
||||||
|
end,
|
||||||
|
}
|
||||||
|
local function new_file_obj(fn)
|
||||||
|
local file = {
|
||||||
|
_is_file = true,
|
||||||
|
_is_open = true,
|
||||||
|
pos = 1,
|
||||||
|
__index = file_meta.__index,
|
||||||
|
filename = fn,
|
||||||
|
data = nil,
|
||||||
|
}
|
||||||
|
setmetatable(file, file_meta)
|
||||||
|
return file
|
||||||
|
end
|
||||||
|
|
||||||
|
local function tflip(t)
|
||||||
|
local u = {}; for _, n in ipairs(t) do u[n] = true end; return u;
|
||||||
|
end
|
||||||
|
local allowed_zip_dirs = tflip {
|
||||||
|
'add-ons', 'base', 'config', 'saves', 'screenshots', 'shaders'
|
||||||
|
}
|
||||||
|
local function io_open_absolute(fn, mode)
|
||||||
|
-- if file exists, use original mode
|
||||||
|
local res, err = _bllua_io_open(fn, mode)
|
||||||
|
if res then
|
||||||
|
return res
|
||||||
|
elseif err and not err:find('No such file or directory$') then
|
||||||
|
return nil, err
|
||||||
|
end
|
||||||
|
|
||||||
|
-- otherwise, if TS sees file but Lua doesn't, it must be in a zip, so use TS reader
|
||||||
|
local dir = fn:match('^[^/]+')
|
||||||
|
if not allowed_zip_dirs[dir:lower()] then return nil, 'Zip is not in one of the allowed directories' end
|
||||||
|
local exist = _bllua_ts.call('isFile', fn) == '1'
|
||||||
|
if not exist then return nil, err end
|
||||||
|
|
||||||
|
if mode ~= nil and mode ~= 'r' and mode ~= 'rb' then
|
||||||
|
return nil, 'Files in zips can only be opened in read mode'
|
||||||
|
end
|
||||||
|
|
||||||
|
-- return a temp lua file object with the data
|
||||||
|
local fi = new_file_obj(fn)
|
||||||
|
return fi
|
||||||
|
end
|
||||||
|
|
||||||
|
io = io or {}
|
||||||
|
---@diagnostic disable-next-line: duplicate-set-field
|
||||||
|
function io.open(fn, mode, errn)
|
||||||
|
errn = errn or 1
|
||||||
|
|
||||||
|
-- try to open the file with relative path, otherwise use absolute path
|
||||||
|
local curfn = debug.getfilename(errn + 1) or _bllua_ts.getvar('Con::File')
|
||||||
|
if curfn == '' then curfn = nil end
|
||||||
|
if fn:find('^%.') then
|
||||||
|
local relfn = curfn and fn:find('^%./') and
|
||||||
|
curfn:gsub('[^/]+$', '') .. fn:gsub('^%./', '')
|
||||||
|
if relfn then
|
||||||
|
local fi, err = io_open_absolute(relfn, mode)
|
||||||
|
return fi, err, relfn
|
||||||
|
else
|
||||||
|
return nil, 'Invalid path', fn
|
||||||
|
end
|
||||||
|
else
|
||||||
|
local fi, err = io_open_absolute(fn, mode)
|
||||||
|
return fi, err, fn
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
---@diagnostic disable-next-line: duplicate-set-field
|
||||||
|
function io.lines(fn)
|
||||||
|
local fi, err, fn2 = io.open(fn, nil, 2)
|
||||||
|
if not fi then error('Error opening file \'' .. fn2 .. '\': ' .. err, 2) end
|
||||||
|
return fi:lines()
|
||||||
|
end
|
||||||
|
|
||||||
|
---@diagnostic disable-next-line: duplicate-set-field
|
||||||
|
function io.type(f)
|
||||||
|
---@diagnostic disable-next-line: undefined-field
|
||||||
|
if type(f) == 'table' and f._is_file then
|
||||||
|
---@diagnostic disable-next-line: undefined-field
|
||||||
|
return f._is_open and 'file' or 'closed file'
|
||||||
|
else
|
||||||
|
return _bllua_io_type(f)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- provide dofile
|
||||||
|
function dofile(fn, errn)
|
||||||
|
errn = errn or 1
|
||||||
|
|
||||||
|
local fi, err, fn2 = io.open(fn, 'r', errn + 1)
|
||||||
|
if not fi then error('Error executing file \'' .. fn2 .. '\': ' .. err, errn + 1) end
|
||||||
|
|
||||||
|
print('Executing ' .. fn2)
|
||||||
|
local text = fi:read('*a')
|
||||||
|
fi:close()
|
||||||
|
return assert(loadstring('--[[' .. fn2 .. ']]' .. text))()
|
||||||
|
end
|
||||||
|
|
||||||
|
-- provide require (just a wrapper for dofile)
|
||||||
|
-- searches for ?.lua and ?/init.lua in the following directories:
|
||||||
|
-- location of current file
|
||||||
|
-- blockland directory
|
||||||
|
-- current add-on
|
||||||
|
local function file_exists(fn, errn)
|
||||||
|
local fi, err, fn2 = io.open(fn, 'r', errn + 1)
|
||||||
|
if fi then
|
||||||
|
fi:close()
|
||||||
|
return fn2
|
||||||
|
else
|
||||||
|
return nil
|
||||||
|
end
|
||||||
|
end
|
||||||
|
local require_memo = {}
|
||||||
|
function require(mod)
|
||||||
|
if require_memo[mod] then return unpack(require_memo[mod]) end
|
||||||
|
local fp = mod:gsub('%.', '/')
|
||||||
|
local fns = {
|
||||||
|
'./' .. fp .. '.lua', -- local file
|
||||||
|
'./' .. fp .. '/init.lua', -- local library
|
||||||
|
fp .. '.lua', -- global file
|
||||||
|
fp .. '/init.lua', -- global library
|
||||||
|
}
|
||||||
|
if fp:lower():find('^add-ons/') then
|
||||||
|
local addonpath = fp:lower():match('^add-ons/[^/]+') .. '/'
|
||||||
|
table.insert(fns, addonpath .. fp .. '.lua') -- add-on file
|
||||||
|
table.insert(fns, addonpath .. fp .. '/init.lua') -- add-on library
|
||||||
|
end
|
||||||
|
for _, fn in ipairs(fns) do
|
||||||
|
local fne = file_exists(fn, 2)
|
||||||
|
if fne then
|
||||||
|
local res = { dofile(fne, 2) }
|
||||||
|
require_memo[mod] = res
|
||||||
|
return unpack(res)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return _bllua_requiresecure(mod)
|
||||||
|
end
|
||||||
|
|
||||||
|
local function isValidCode(code)
|
||||||
|
local f, e = loadstring(code)
|
||||||
|
return f ~= nil
|
||||||
|
end
|
||||||
|
function _bllua_smarteval(code)
|
||||||
|
if (not code:find('^print%(')) and isValidCode('print(' .. code .. ')') then
|
||||||
|
code = 'print(' .. code .. ')'
|
||||||
|
end
|
||||||
|
local f, e = loadstring(code)
|
||||||
|
if f then
|
||||||
|
return f()
|
||||||
|
else
|
||||||
|
print(e)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
function ts.setvar(name, val)
|
||||||
|
_bllua_ts.call('_bllua_set_var', name, val)
|
||||||
|
end
|
||||||
|
|
||||||
|
_bllua_ts.call('echo', ' Executed libts-lua.lua')
|
||||||
@@ -49,4 +49,4 @@ function _bllua_set_var(%name, %val) {
|
|||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
|
||||||
echo(" Executed libts-support.cs");
|
echo(" Executed libts-ts.cs");
|
||||||
@@ -1,216 +0,0 @@
|
|||||||
|
|
||||||
-- This Lua code provides some built-in utilities for writing Lua add-ons
|
|
||||||
-- It is eval'd automatically once BLLua3 has loaded the TS API and environment
|
|
||||||
-- It only has access to the sandboxed lua environment, just like user code.
|
|
||||||
|
|
||||||
ts = _bllua_ts
|
|
||||||
|
|
||||||
-- Provide limited OS functions
|
|
||||||
os = os or {}
|
|
||||||
function os.time() return math.floor(tonumber(_bllua_ts.call('getSimTime'))/1000) end
|
|
||||||
function os.clock() return tonumber(_bllua_ts.call('getSimTime'))/1000 end
|
|
||||||
|
|
||||||
-- Virtual file class, emulating a file object as returned by io.open
|
|
||||||
-- Used to wrap io.open to allow reading from zips (using TS)
|
|
||||||
-- Not perfect because TS file I/O sucks
|
|
||||||
-- Can't read nulls, can't distinguish between CRLF and LF.
|
|
||||||
-- Todo someday: actually read the zip in lua?
|
|
||||||
local file_meta = {
|
|
||||||
read = function(file, mode)
|
|
||||||
file:_init()
|
|
||||||
if not file or type(file)~='table' or not file._is_file then error('File:read: Not a file', 2) end
|
|
||||||
if file._is_open ~= true then error('File:read: File is closed', 2) end
|
|
||||||
if mode=='*n' then
|
|
||||||
local ws, n = file.data:match('^([ \t\r\n]*)([0-9%.%-e]+)', file.pos)
|
|
||||||
if n then
|
|
||||||
file.pos = file.pos + #ws + #n
|
|
||||||
return n
|
|
||||||
else
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
elseif mode=='*a' then
|
|
||||||
local d = file.data:sub(file.pos, #file.data)
|
|
||||||
file.pos = #file.data + 1
|
|
||||||
return d
|
|
||||||
elseif mode=='*l' then
|
|
||||||
local l, ws = file.data:match('^([^\r\n]*)(\r?\n)', file.pos)
|
|
||||||
if not l then
|
|
||||||
l = file.data:match('^([^\r\n]*)$', file.pos); ws = '';
|
|
||||||
if l=='' then return nil end
|
|
||||||
end
|
|
||||||
if l then
|
|
||||||
file.pos = file.pos + #l + #ws
|
|
||||||
return l
|
|
||||||
else
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
elseif type(mode)=='number' then
|
|
||||||
local d = file.data:sub(file.pos, file.pos+mode)
|
|
||||||
file.pos = file.pos + #d
|
|
||||||
return d
|
|
||||||
else
|
|
||||||
error('File:read: Invalid mode \''..mode..'\'', 2)
|
|
||||||
end
|
|
||||||
end,
|
|
||||||
lines = function(file)
|
|
||||||
file:_init()
|
|
||||||
return function()
|
|
||||||
return file:read('*l')
|
|
||||||
end
|
|
||||||
end,
|
|
||||||
close = function(file)
|
|
||||||
if not file._is_open then error('File:close: File is not open', 2) end
|
|
||||||
file._is_open = false
|
|
||||||
end,
|
|
||||||
__index = function(f, k) return rawget(f, k) or getmetatable(f)[k] end,
|
|
||||||
_init = function(f)
|
|
||||||
if not f.data then
|
|
||||||
f.data = _bllua_ts.call('_bllua_ReadEntireFile', f.filename)
|
|
||||||
end
|
|
||||||
end,
|
|
||||||
}
|
|
||||||
local function new_file_obj(fn)
|
|
||||||
local file = {
|
|
||||||
_is_file = true,
|
|
||||||
_is_open = true,
|
|
||||||
pos = 1,
|
|
||||||
__index = file_meta.__index,
|
|
||||||
filename = fn,
|
|
||||||
data = nil,
|
|
||||||
}
|
|
||||||
setmetatable(file, file_meta)
|
|
||||||
return file
|
|
||||||
end
|
|
||||||
|
|
||||||
local function tflip(t) local u = {}; for _, n in ipairs(t) do u[n] = true end; return u; end
|
|
||||||
local allowed_zip_dirs = tflip{
|
|
||||||
'add-ons', 'base', 'config', 'saves', 'screenshots', 'shaders'
|
|
||||||
}
|
|
||||||
local function io_open_absolute(fn, mode)
|
|
||||||
-- if file exists, use original mode
|
|
||||||
local res, err = _bllua_io_open(fn, mode)
|
|
||||||
if res then return res end
|
|
||||||
|
|
||||||
-- otherwise, if TS sees file but Lua doesn't, it must be in a zip, so use TS reader
|
|
||||||
local dir = fn:match('^[^/]+')
|
|
||||||
if not allowed_zip_dirs[dir:lower()] then return nil, 'File is not in one of the allowed directories' end
|
|
||||||
local exist = _bllua_ts.call('isFile', fn) == '1'
|
|
||||||
if not exist then return nil, err end
|
|
||||||
|
|
||||||
if mode~=nil and mode~='r' and mode~='rb' then
|
|
||||||
return nil, 'Files in zips can only be opened in read mode' end
|
|
||||||
|
|
||||||
-- return a temp lua file object with the data
|
|
||||||
local fi = new_file_obj(fn)
|
|
||||||
return fi
|
|
||||||
end
|
|
||||||
|
|
||||||
io = io or {}
|
|
||||||
function io.open(fn, mode, errn)
|
|
||||||
errn = errn or 1
|
|
||||||
|
|
||||||
-- try to open the file with relative path, otherwise use absolute path
|
|
||||||
local curfn = debug.getfilename(errn + 1) or _bllua_ts.getvar('Con::File')
|
|
||||||
if curfn == '' then curfn = nil end
|
|
||||||
if fn:find('^%.') then
|
|
||||||
local relfn = curfn and fn:find('^%./') and
|
|
||||||
curfn:gsub('[^/]+$', '')..fn:gsub('^%./', '')
|
|
||||||
if relfn then
|
|
||||||
local fi, err = io_open_absolute(relfn, mode, errn+1)
|
|
||||||
return fi, err, relfn
|
|
||||||
else
|
|
||||||
return nil, 'Invalid path', fn
|
|
||||||
end
|
|
||||||
else
|
|
||||||
local fi, err = io_open_absolute(fn, mode, errn+1)
|
|
||||||
return fi, err, fn
|
|
||||||
end
|
|
||||||
end
|
|
||||||
function io.lines(fn)
|
|
||||||
local fi, err, fn2 = io.open(fn, nil, 2)
|
|
||||||
if not fi then error('Error opening file \''..fn2..'\': '..err, 2) end
|
|
||||||
return fi:lines()
|
|
||||||
end
|
|
||||||
function io.type(f)
|
|
||||||
if type(f)=='table' and f._is_file then
|
|
||||||
return f._is_open and 'file' or 'closed file'
|
|
||||||
else
|
|
||||||
return _bllua_io_type(f)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- provide dofile
|
|
||||||
function dofile(fn, errn)
|
|
||||||
errn = errn or 1
|
|
||||||
|
|
||||||
local fi, err, fn2 = io.open(fn, 'r', errn+1)
|
|
||||||
if not fi then error('Error executing file \''..fn2..'\': '..err, errn+1) end
|
|
||||||
|
|
||||||
print('Executing '..fn2)
|
|
||||||
local text = fi:read('*a')
|
|
||||||
fi:close()
|
|
||||||
return assert(loadstring('--[['..fn2..']]'..text))()
|
|
||||||
end
|
|
||||||
|
|
||||||
-- provide require (just a wrapper for dofile)
|
|
||||||
-- searches for ?.lua and ?/init.lua in the following directories:
|
|
||||||
-- location of current file
|
|
||||||
-- blockland directory
|
|
||||||
-- current add-on
|
|
||||||
local function file_exists(fn, errn)
|
|
||||||
local fi, err, fn2 = io.open(fn, 'r', errn+1)
|
|
||||||
if fi then
|
|
||||||
fi:close()
|
|
||||||
return fn2
|
|
||||||
else
|
|
||||||
return nil
|
|
||||||
end
|
|
||||||
end
|
|
||||||
local require_memo = {}
|
|
||||||
function require(mod)
|
|
||||||
if require_memo[mod] then return unpack(require_memo[mod]) end
|
|
||||||
local fp = mod:gsub('%.', '/')
|
|
||||||
local fns = {
|
|
||||||
'./'..fp..'.lua', -- local file
|
|
||||||
'./'..fp..'/init.lua', -- local library
|
|
||||||
fp..'.lua', -- global file
|
|
||||||
fp..'/init.lua', -- global library
|
|
||||||
}
|
|
||||||
if fp:lower():find('^add-ons/') then
|
|
||||||
local addonpath = fp:lower():match('^add-ons/[^/]+')..'/'
|
|
||||||
table.insert(fns, addonpath..fp..'.lua') -- add-on file
|
|
||||||
table.insert(fns, addonpath..fp..'/init.lua') -- add-on library
|
|
||||||
end
|
|
||||||
for _,fn in ipairs(fns) do
|
|
||||||
local fne = file_exists(fn, 2)
|
|
||||||
if fne then
|
|
||||||
local res = {dofile(fne, 2)}
|
|
||||||
require_memo[mod] = res
|
|
||||||
return unpack(res)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
return _bllua_requiresecure(mod)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- Exposure to TS
|
|
||||||
function _bllua_getvar(name) return _G[name] end
|
|
||||||
function _bllua_setvar(name, val) _G[name] = val end
|
|
||||||
function _bllua_eval(code) return loadstring(code)() end
|
|
||||||
function _bllua_exec(fn) return dofile(fn, 2) end
|
|
||||||
|
|
||||||
local function isValidCode(code)
|
|
||||||
local f,e = loadstring(code)
|
|
||||||
return f~=nil
|
|
||||||
end
|
|
||||||
function _bllua_smarteval(code)
|
|
||||||
if (not code:find('^print%(')) and isValidCode('print('..code..')') then
|
|
||||||
code = 'print('..code..')' end
|
|
||||||
local f,e = loadstring(code)
|
|
||||||
if f then
|
|
||||||
return f()
|
|
||||||
else
|
|
||||||
print(e)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
_bllua_ts.call('echo', ' Executed libts.lua')
|
|
||||||
6
src/util/matrix.lua
Normal file
6
src/util/matrix.lua
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
-- todo
|
||||||
|
-- Matrix class with math operators
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
print(' Executed matrix.lua')
|
||||||
@@ -1,24 +1,35 @@
|
|||||||
|
|
||||||
-- Basic functionality that should be standard in Lua
|
-- Basic functionality that should be standard in Lua
|
||||||
|
|
||||||
|
|
||||||
-- Table / List
|
-- Table / List
|
||||||
-- Whether a table contains no keys
|
-- Whether a table contains no keys
|
||||||
function table.empty(t)
|
function table.empty(t)
|
||||||
for _,_ in pairs(t) do return false end
|
return next(t) == nil
|
||||||
return true
|
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Apply a function to each key in a table
|
-- Apply a function to each key in a table
|
||||||
function table.map(f, ...)
|
function table.map(f, ...)
|
||||||
local ts = { ... }
|
local ts = { ... }
|
||||||
local u = {}
|
local u = {}
|
||||||
for k, _ in pairs(ts[1]) do
|
for k, _ in pairs(ts[1]) do
|
||||||
local args = {}
|
local args = {}
|
||||||
for j=1,#ts do args[j] = ts[j][i] end
|
for j = 1, #ts do args[j] = ts[j][k] end
|
||||||
u[i] = f(unpack(args))
|
u[k] = f(unpack(args))
|
||||||
end
|
end
|
||||||
return u
|
return u
|
||||||
end
|
end
|
||||||
|
|
||||||
|
function table.mapk(f, ...)
|
||||||
|
local ts = { ... }
|
||||||
|
local u = {}
|
||||||
|
for k, _ in pairs(ts[1]) do
|
||||||
|
local args = {}
|
||||||
|
for j = 1, #ts do args[j] = ts[j][k] end
|
||||||
|
u[k] = f(k, unpack(args))
|
||||||
|
end
|
||||||
|
return u
|
||||||
|
end
|
||||||
|
|
||||||
function table.map_list(f, ...)
|
function table.map_list(f, ...)
|
||||||
local ts = { ... }
|
local ts = { ... }
|
||||||
local u = {}
|
local u = {}
|
||||||
@@ -29,30 +40,32 @@ function table.map_list(f, ...)
|
|||||||
end
|
end
|
||||||
return u
|
return u
|
||||||
end
|
end
|
||||||
|
|
||||||
|
function table.mapi_list(f, ...)
|
||||||
|
local ts = { ... }
|
||||||
|
local u = {}
|
||||||
|
for i = 1, #ts[1] do
|
||||||
|
local args = {}
|
||||||
|
for j = 1, #ts do args[j] = ts[j][i] end
|
||||||
|
u[i] = f(i, unpack(args))
|
||||||
|
end
|
||||||
|
return u
|
||||||
|
end
|
||||||
|
|
||||||
-- Swap keys/values
|
-- Swap keys/values
|
||||||
function table.swap(t)
|
function table.swap(t)
|
||||||
local u = {}
|
local u = {}
|
||||||
for k, v in pairs(t) do u[v] = k end
|
for k, v in pairs(t) do u[v] = k end
|
||||||
return u
|
return u
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Reverse a list
|
-- Reverse a list
|
||||||
function table.reverse(l)
|
function table.reverse(l)
|
||||||
local m = {}
|
local m = {}
|
||||||
for i = 1, #l do m[#l - i + 1] = l[i] end
|
for i = 1, #l do m[#l - i + 1] = l[i] end
|
||||||
return m
|
return m
|
||||||
end
|
end
|
||||||
-- Convert i->v to v->true
|
|
||||||
function table.values(l)
|
|
||||||
local u = {}
|
|
||||||
for _,v in ipairs(l) do u[v] = true end
|
|
||||||
return u
|
|
||||||
end
|
|
||||||
-- Make a list of keys
|
|
||||||
function table.keys(t)
|
|
||||||
local u = {}
|
|
||||||
for k,_ in pairs(t) do table.insert(u, k) end
|
|
||||||
return u
|
|
||||||
end
|
|
||||||
-- Whether a table is a list/array (has only monotonic integer keys)
|
-- Whether a table is a list/array (has only monotonic integer keys)
|
||||||
function table.islist(t)
|
function table.islist(t)
|
||||||
local n = 0
|
local n = 0
|
||||||
@@ -62,6 +75,7 @@ function table.islist(t)
|
|||||||
end
|
end
|
||||||
return n == #t
|
return n == #t
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Append contents of other tables to first table
|
-- Append contents of other tables to first table
|
||||||
function table.append(t, ...)
|
function table.append(t, ...)
|
||||||
local a = { ... }
|
local a = { ... }
|
||||||
@@ -70,6 +84,7 @@ function table.append(t, ...)
|
|||||||
end
|
end
|
||||||
return t
|
return t
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Create a new table containing all keys from any number of tables
|
-- Create a new table containing all keys from any number of tables
|
||||||
-- latter tables in the arg list override prior ones
|
-- latter tables in the arg list override prior ones
|
||||||
-- overlaps, NOT appends, integer keys
|
-- overlaps, NOT appends, integer keys
|
||||||
@@ -81,6 +96,7 @@ function table.join(...)
|
|||||||
end
|
end
|
||||||
return w
|
return w
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Whether a table contains a certain value in any key
|
-- Whether a table contains a certain value in any key
|
||||||
function table.contains(t, s)
|
function table.contains(t, s)
|
||||||
for _, v in pairs(t) do
|
for _, v in pairs(t) do
|
||||||
@@ -88,29 +104,34 @@ function table.contains(t,s)
|
|||||||
end
|
end
|
||||||
return false
|
return false
|
||||||
end
|
end
|
||||||
|
|
||||||
function table.contains_list(t, s)
|
function table.contains_list(t, s)
|
||||||
for _, v in ipairs(t) do
|
for _, v in ipairs(t) do
|
||||||
if v == s then return true end
|
if v == s then return true end
|
||||||
end
|
end
|
||||||
return false
|
return false
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Copy a table to another table
|
-- Copy a table to another table
|
||||||
function table.copy(t)
|
function table.copy(t)
|
||||||
local u = {}
|
local u = {}
|
||||||
for k, v in pairs(t) do u[k] = v end
|
for k, v in pairs(t) do u[k] = v end
|
||||||
return u
|
return u
|
||||||
end
|
end
|
||||||
|
|
||||||
function table.copy_list(l)
|
function table.copy_list(l)
|
||||||
local m = {}
|
local m = {}
|
||||||
for i, v in ipairs(l) do m[i] = v end
|
for i, v in ipairs(l) do m[i] = v end
|
||||||
return m
|
return m
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Sort a table in a new copy
|
-- Sort a table in a new copy
|
||||||
function table.sortcopy(t, f)
|
function table.sortcopy(t, f)
|
||||||
local u = table.copy_list(t)
|
local u = table.copy_list(t)
|
||||||
table.sort(u, f)
|
table.sort(u, f)
|
||||||
return u
|
return u
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Remove a value from a table
|
-- Remove a value from a table
|
||||||
function table.removevalue(t, r)
|
function table.removevalue(t, r)
|
||||||
local rem = {}
|
local rem = {}
|
||||||
@@ -119,6 +140,7 @@ function table.removevalue(t, r)
|
|||||||
end
|
end
|
||||||
for _, k in ipairs(rem) do t[k] = nil end
|
for _, k in ipairs(rem) do t[k] = nil end
|
||||||
end
|
end
|
||||||
|
|
||||||
function table.removevalue_list(t, r)
|
function table.removevalue_list(t, r)
|
||||||
for i = #t, 1, -1 do
|
for i = #t, 1, -1 do
|
||||||
if t[i] == r then
|
if t[i] == r then
|
||||||
@@ -126,6 +148,7 @@ function table.removevalue_list(t, r)
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Export tables into formatted executable strings
|
-- Export tables into formatted executable strings
|
||||||
local function tabs(tabLevel)
|
local function tabs(tabLevel)
|
||||||
return (' '):rep(tabLevel)
|
return (' '):rep(tabLevel)
|
||||||
@@ -192,14 +215,13 @@ valueToString = function(v, tabLevel, seen)
|
|||||||
return tostring(v)
|
return tostring(v)
|
||||||
else
|
else
|
||||||
--error('table.tostring: table contains a '..t..' value, cannot serialize')
|
--error('table.tostring: table contains a '..t..' value, cannot serialize')
|
||||||
return 'nil --[[ cannot serialize '..t..': '..tostring(v)..' ]]'
|
return 'nil --[[ ' .. tostring(v) .. ' ]]'
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
function table.tostring(t)
|
function table.tostring(t)
|
||||||
return tableToString(t, 0, {})
|
return tableToString(t, 0, {})
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
-- String
|
-- String
|
||||||
|
|
||||||
-- Split string into table by separator
|
-- Split string into table by separator
|
||||||
@@ -207,7 +229,8 @@ end
|
|||||||
-- if regex is not true, sep is treated as a regex pattern
|
-- if regex is not true, sep is treated as a regex pattern
|
||||||
function string.split(str, sep, noregex)
|
function string.split(str, sep, noregex)
|
||||||
if type(str) ~= 'string' then
|
if type(str) ~= 'string' then
|
||||||
error('string.split: argument #1: expected string, got '..type(str), 2) end
|
error('string.split: argument #1: expected string, got ' .. type(str), 2)
|
||||||
|
end
|
||||||
if sep == nil or sep == '' then
|
if sep == nil or sep == '' then
|
||||||
local t = {}
|
local t = {}
|
||||||
local ns = #str
|
local ns = #str
|
||||||
@@ -233,6 +256,7 @@ function string.split(str, sep, noregex)
|
|||||||
'string.split: argument #2: expected string or nil, got ' .. type(sep), 2)
|
'string.split: argument #2: expected string or nil, got ' .. type(sep), 2)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Split string to a list of char bytes
|
-- Split string to a list of char bytes
|
||||||
function string.bytes(s)
|
function string.bytes(s)
|
||||||
local b = {}
|
local b = {}
|
||||||
@@ -242,11 +266,13 @@ function string.bytes(s)
|
|||||||
end
|
end
|
||||||
return b
|
return b
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Trim leading and trailing whitespace
|
-- Trim leading and trailing whitespace
|
||||||
function string.trim(s, ws)
|
function string.trim(s, ws)
|
||||||
ws = ws or '[ \t\r\t]'
|
ws = ws or ' \t\r\n'
|
||||||
return s:gsub('^'..ws..'+', ''):gsub(ws..'+$', '')..''
|
return s:gsub('^[' .. ws .. ']+', ''):gsub('[' .. ws .. ']+$', '') .. ''
|
||||||
end
|
end
|
||||||
|
|
||||||
-- String slicing and searching using [] operator
|
-- String slicing and searching using [] operator
|
||||||
local str_meta = getmetatable('')
|
local str_meta = getmetatable('')
|
||||||
local str_meta_index_old = str_meta.__index
|
local str_meta_index_old = str_meta.__index
|
||||||
@@ -262,15 +288,20 @@ function str_meta.__index(s,k)
|
|||||||
return string.sub(s, a, b)
|
return string.sub(s, a, b)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
-- String iterator
|
-- String iterator
|
||||||
function string.chars(s)
|
function string.chars(s)
|
||||||
local i = 0
|
local i = 0
|
||||||
return function()
|
return function()
|
||||||
i = i + 1
|
i = i + 1
|
||||||
if i<=#s then return s:sub(i,i)
|
if i <= #s then
|
||||||
else return nil end
|
return s:sub(i, i)
|
||||||
|
else
|
||||||
|
return nil
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
end
|
||||||
|
|
||||||
-- Escape sequences
|
-- Escape sequences
|
||||||
local defaultEscapes = {
|
local defaultEscapes = {
|
||||||
['\\'] = '\\\\',
|
['\\'] = '\\\\',
|
||||||
@@ -290,6 +321,7 @@ function string.escape(s, escapes)
|
|||||||
end
|
end
|
||||||
return table.concat(t)
|
return table.concat(t)
|
||||||
end
|
end
|
||||||
|
|
||||||
local defaultEscapeChar = '\\'
|
local defaultEscapeChar = '\\'
|
||||||
local defaultUnescapes = {
|
local defaultUnescapes = {
|
||||||
['\\'] = '\\',
|
['\\'] = '\\',
|
||||||
@@ -320,20 +352,20 @@ function string.unescape(s, escapeChar, unescapes)
|
|||||||
return table.concat(t)
|
return table.concat(t)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
-- IO
|
-- IO
|
||||||
|
|
||||||
io = io or {}
|
io = io or {}
|
||||||
-- Read entire file at once, return nil,err if access failed
|
-- Read entire file at once, return nil,err if access failed
|
||||||
function io.readfile(filename)
|
function io.readall(filename)
|
||||||
local fi, err = io.open(filename, 'rb')
|
local fi, err = io.open(filename, 'rb')
|
||||||
if not fi then return nil, err end
|
if not fi then return nil, err end
|
||||||
local s = fi:read("*a")
|
local s = fi:read("*a")
|
||||||
fi:close()
|
fi:close()
|
||||||
return s
|
return s
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Write data to file all at once, return true if success / false,err if failure
|
-- Write data to file all at once, return true if success / false,err if failure
|
||||||
function io.writefile(filename, data)
|
function io.writeall(filename, data)
|
||||||
local fi, err = io.open(filename, 'wb')
|
local fi, err = io.open(filename, 'wb')
|
||||||
if not fi then return false, err end
|
if not fi then return false, err end
|
||||||
fi:write(data)
|
fi:write(data)
|
||||||
@@ -341,20 +373,26 @@ function io.writefile(filename, data)
|
|||||||
return true, nil
|
return true, nil
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
-- Math
|
-- Math
|
||||||
|
|
||||||
-- Round
|
-- Round
|
||||||
function math.round(x)
|
function math.round(x)
|
||||||
return math.floor(x + 0.5)
|
return math.floor(x + 0.5)
|
||||||
end
|
end
|
||||||
|
|
||||||
-- Mod that accounts for floating point inaccuracy
|
-- Mod that accounts for floating point inaccuracy
|
||||||
function math.mod(a, b)
|
function math.mod(a, b)
|
||||||
local m = a % b
|
local m = a % b
|
||||||
if m==0 or math.abs(m)<1e-15 or math.abs(m-b)<1e-15 then return 0
|
if m == 0 or math.abs(m) < 1e-15 or math.abs(m - b) < 1e-15 then
|
||||||
else return m end
|
return 0
|
||||||
|
else
|
||||||
|
return m
|
||||||
end
|
end
|
||||||
|
end
|
||||||
|
|
||||||
-- Clamp value between min and max
|
-- Clamp value between min and max
|
||||||
function math.clamp(v, n, x)
|
function math.clamp(v, n, x)
|
||||||
return math.min(x, math.max(v, n))
|
return math.min(x, math.max(v, n))
|
||||||
end
|
end
|
||||||
|
|
||||||
|
print(' Executed std.lua')
|
||||||
|
|||||||
@@ -1,18 +1,19 @@
|
|||||||
|
|
||||||
-- Vector math class with operators
|
-- Vector math class with operators
|
||||||
local vector_meta
|
local vector_meta
|
||||||
local vector_new
|
local vector_new
|
||||||
local function vector_check(v, n, name, argn)
|
local function vector_check(v, n, name, argn)
|
||||||
if not v.__is_vector then
|
if not v.__is_vector then
|
||||||
error('vector ' .. name .. ': argument #' .. (argn or 1)
|
error('vector ' .. name .. ': argument #' .. (argn or 1)
|
||||||
..': expected vector, got '..type(v), n+1) end
|
.. ': expected vector, got ' .. type(v), n + 1)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
local function vector_checksamelen(v1, v2, name)
|
local function vector_checksamelen(v1, v2, name)
|
||||||
vector_check(v1, 3, name, 1)
|
vector_check(v1, 3, name, 1)
|
||||||
vector_check(v2, 3, name, 2)
|
vector_check(v2, 3, name, 2)
|
||||||
if #v1 ~= #v2 then
|
if #v1 ~= #v2 then
|
||||||
error('vector ' .. name .. ': vector lengths do not match (lengths are '
|
error('vector ' .. name .. ': vector lengths do not match (lengths are '
|
||||||
..#v1..' and '..#v2..')', 3) end
|
.. #v1 .. ' and ' .. #v2 .. ')', 3)
|
||||||
|
end
|
||||||
return #v1
|
return #v1
|
||||||
end
|
end
|
||||||
local function vector_checklen(v1, v2, name, len)
|
local function vector_checklen(v1, v2, name, len)
|
||||||
@@ -20,7 +21,8 @@ local function vector_checklen(v1, v2, name, len)
|
|||||||
vector_check(v2, 3, name, 2)
|
vector_check(v2, 3, name, 2)
|
||||||
if #v1 ~= len or #v2 ~= len then
|
if #v1 ~= len or #v2 ~= len then
|
||||||
error('vector ' .. name .. ': vector lengths are not ' .. len .. ' (lengths are '
|
error('vector ' .. name .. ': vector lengths are not ' .. len .. ' (lengths are '
|
||||||
..#v1..' and '..#v2..')', 3) end
|
.. #v1 .. ' and ' .. #v2 .. ')', 3)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
local function vector_opnnn(name, op)
|
local function vector_opnnn(name, op)
|
||||||
return function(v1, v2)
|
return function(v1, v2)
|
||||||
@@ -69,15 +71,21 @@ local vector_indices = {x = 1, y = 2, z = 3, w = 4, r = 1, g = 2, b = 3, a = 4}
|
|||||||
local vector_meta = {
|
local vector_meta = {
|
||||||
__is_vector = true,
|
__is_vector = true,
|
||||||
__index = function(t, k)
|
__index = function(t, k)
|
||||||
if tonumber(k) then return rawget(t, k)
|
if tonumber(k) then
|
||||||
elseif vector_indices[k] then return rawget(t, vector_indices[k])
|
return rawget(t, k)
|
||||||
else return getmetatable(t)[k]
|
elseif vector_indices[k] then
|
||||||
|
return rawget(t, vector_indices[k])
|
||||||
|
else
|
||||||
|
return getmetatable(t)[k]
|
||||||
end
|
end
|
||||||
end,
|
end,
|
||||||
__newindex = function(t, k, v)
|
__newindex = function(t, k, v)
|
||||||
if tonumber(k) then rawset(t, k, v)
|
if tonumber(k) then
|
||||||
elseif vector_indices[k] then rawset(t, vector_indices[k], v)
|
rawset(t, k, v)
|
||||||
else return
|
elseif vector_indices[k] then
|
||||||
|
rawset(t, vector_indices[k], v)
|
||||||
|
else
|
||||||
|
return
|
||||||
end
|
end
|
||||||
end,
|
end,
|
||||||
__add = vector_opnnn('add', function(x1, x2) return x1 + x2 end),
|
__add = vector_opnnn('add', function(x1, x2) return x1 + x2 end),
|
||||||
@@ -115,8 +123,11 @@ local vector_meta = {
|
|||||||
local len = #v1
|
local len = #v1
|
||||||
local v3 = {}
|
local v3 = {}
|
||||||
for i = 1, len do
|
for i = 1, len do
|
||||||
if length==0 then v3[i] = 0
|
if length == 0 then
|
||||||
else v3[i] = v1[i]/length end
|
v3[i] = 0
|
||||||
|
else
|
||||||
|
v3[i] = v1[i] / length
|
||||||
|
end
|
||||||
end
|
end
|
||||||
return vector_new(v3)
|
return vector_new(v3)
|
||||||
end,
|
end,
|
||||||
@@ -127,7 +138,7 @@ local vector_meta = {
|
|||||||
for i = 1, len do
|
for i = 1, len do
|
||||||
table.insert(st, tostring(v1[i]))
|
table.insert(st, tostring(v1[i]))
|
||||||
end
|
end
|
||||||
return '{ '..table.concat(st, ', ')..' }'
|
return 'vector{ ' .. table.concat(st, ', ') .. ' }'
|
||||||
end,
|
end,
|
||||||
unpack = function(v1) return unpack(v1) end,
|
unpack = function(v1) return unpack(v1) end,
|
||||||
floor = vector_opn0n('floor', function(x1) return math.floor(x1) end),
|
floor = vector_opn0n('floor', function(x1) return math.floor(x1) end),
|
||||||
@@ -152,20 +163,28 @@ local vector_meta = {
|
|||||||
rotateByAngleId = function(v1, r)
|
rotateByAngleId = function(v1, r)
|
||||||
--vector_check(v1, 2, 'rotate')
|
--vector_check(v1, 2, 'rotate')
|
||||||
if type(r) ~= 'number' or r % 1 ~= 0 then
|
if type(r) ~= 'number' or r % 1 ~= 0 then
|
||||||
error('vector rotateByAngleId: invalid rotation '..tostring(r), 2) end
|
error('vector rotateByAngleId: invalid rotation ' .. tostring(r), 2)
|
||||||
|
end
|
||||||
r = r % 4
|
r = r % 4
|
||||||
local v2
|
local v2
|
||||||
if r==0 then v2 = vector_new{ v1[1], v1[2], v1[3] }
|
if r == 0 then
|
||||||
elseif r==1 then v2 = vector_new{ v1[2], -v1[1], v1[3] }
|
v2 = vector_new { v1[1], v1[2], v1[3] }
|
||||||
elseif r==2 then v2 = vector_new{ -v1[1], -v1[2], v1[3] }
|
elseif r == 1 then
|
||||||
elseif r==3 then v2 = vector_new{ -v1[2], v1[1], v1[3] }
|
v2 = vector_new { v1[2], -v1[1], v1[3] }
|
||||||
else error('vector rotateByAngleId: invalid rotation '..r, 2) end
|
elseif r == 2 then
|
||||||
|
v2 = vector_new { -v1[1], -v1[2], v1[3] }
|
||||||
|
elseif r == 3 then
|
||||||
|
v2 = vector_new { -v1[2], v1[1], v1[3] }
|
||||||
|
else
|
||||||
|
error('vector rotateByAngleId: invalid rotation ' .. r, 2)
|
||||||
|
end
|
||||||
return v2
|
return v2
|
||||||
end,
|
end,
|
||||||
rotate2d = function(v, r)
|
rotateZ = function(v, r)
|
||||||
--vector_check(v, 2, 'rotate2d')
|
--vector_check(v, 2, 'rotate2d')
|
||||||
if type(r) ~= 'number' then
|
if type(r) ~= 'number' then
|
||||||
error('vector rotate2d: invalid rotation '..tostring(r), 2) end
|
error('vector rotateZ: invalid rotation ' .. tostring(r), 2)
|
||||||
|
end
|
||||||
local len = math.sqrt(v[1] ^ 2 + v[2] ^ 2)
|
local len = math.sqrt(v[1] ^ 2 + v[2] ^ 2)
|
||||||
local ang = math.atan2(v[2], v[1]) + r
|
local ang = math.atan2(v[2], v[1]) + r
|
||||||
local v2 = vector_new { math.cos(ang) * len, math.sin(ang) * len }
|
local v2 = vector_new { math.cos(ang) * len, math.sin(ang) * len }
|
||||||
@@ -216,4 +235,7 @@ vector_new = function(vi)
|
|||||||
end
|
end
|
||||||
|
|
||||||
vector = vector_new
|
vector = vector_new
|
||||||
|
|
||||||
|
print(' Executed vector.lua')
|
||||||
|
|
||||||
return vector_new
|
return vector_new
|
||||||
|
|||||||
Reference in New Issue
Block a user