mirror of
https://github.com/openwrt/openwrt.git
synced 2024-12-22 23:12:32 +00:00
124b8c653f
Fixes compilation with Linux 4.3. Runtime tested on Ubiquiti EdgeRouter Lite with Linux 3.18, 4.1 and 4.3. Signed-off-by: Stijn Tintel <stijn@linux-ipv6.be> SVN-Revision: 47470
18159 lines
513 KiB
Diff
18159 lines
513 KiB
Diff
--- /dev/null
|
|
+++ b/extensions/LUA/byte_array.c
|
|
@@ -0,0 +1,145 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+#include "controller.h"
|
|
+
|
|
+/* Initialization helper function. This function should be used whenever
|
|
+ * a new byte array need to be initialized. Depending on the arguments it
|
|
+ * initializes the array in a different way. Have a look at the inline
|
|
+ * comments */
|
|
+lua_packet_segment * init_byte_array(lua_State *L, unsigned char * start, int length, int do_copy)
|
|
+{
|
|
+ lua_packet_segment *array;
|
|
+
|
|
+ if (length < 0)
|
|
+ luaL_error(L, "init_byte_array, requested size < 0");
|
|
+
|
|
+ if (start && do_copy) {
|
|
+ /* we have a start address where we copy from */
|
|
+ array = lua_newuserdata(L, sizeof(lua_packet_segment) + length);
|
|
+ array->start = (unsigned char *)array + sizeof(lua_packet_segment); /* aligning pointer */
|
|
+ memcpy(array->start, start, length);
|
|
+ }else if (start && !do_copy) {
|
|
+ /* just link the start pointer, in this case you have to free the memory yourself */
|
|
+ array = lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+ array->start = start;
|
|
+ }else{
|
|
+ /* create an empty array, fully managed by Lua */
|
|
+ array = lua_newuserdata(L, sizeof(lua_packet_segment) + length);
|
|
+ array->start = (unsigned char *)array + sizeof(lua_packet_segment); /* aligning pointer */
|
|
+ memset(array->start, 0, length);
|
|
+ }
|
|
+
|
|
+ array->length = length;
|
|
+ array->offset = 0;
|
|
+ array->changes = NULL;
|
|
+
|
|
+ luaL_getmetatable(L, LUA_BYTE_ARRAY);
|
|
+ lua_setmetatable(L, -2);
|
|
+
|
|
+ return array;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/* LUA_API: get one byte of the given byte array
|
|
+ * access-pattern: array[<index>] */
|
|
+static int32_t get_byte_array(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * array = checkbytearray(L, 1);
|
|
+ int32_t index = luaL_checkinteger(L, 2); /* array starts with index 0 (not 1 as usual in Lua) */
|
|
+
|
|
+ luaL_argcheck(L, 0 <= index && index < array->length, 1, "index out of range");
|
|
+ lua_pushinteger(L, (array->start + array->offset)[index]);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: set one byte of the given byte array
|
|
+ * access-pattern: array[<index>]= 0xFF */
|
|
+static int32_t set_byte_array(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * array = checkbytearray(L, 1);
|
|
+ uint8_t byte;
|
|
+ int32_t index = luaL_checkinteger(L, 2); /* array starts with index 0 (not 1 as usual in Lua) */
|
|
+ int32_t val = luaL_checkinteger(L, 3);
|
|
+ uint32_t nob = 1 << CHAR_BIT; /* we should use something like 1 << CHAR_BIT */
|
|
+
|
|
+ luaL_argcheck(L, 0 <= index && index < array->length, 1, "index out of range");
|
|
+ luaL_argcheck(L, 0 <= val && val < nob, 2, "cannot cast value to char");
|
|
+
|
|
+ byte = (uint8_t)val;
|
|
+
|
|
+ (array->start + array->offset)[index] = byte;
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+/* LUA_API: get size of the given byte array
|
|
+ * access-pattern: #array (__length meta-method) */
|
|
+static int32_t get_byte_array_size(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * array = checkbytearray(L, 1);
|
|
+
|
|
+ lua_pushnumber(L, array->length);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+/* LUA_API: converts a given byte array to a string.
|
|
+ * access-pattern: implicit through functions calling the
|
|
+ * __to_string() metamethod , e.g. print32_t */
|
|
+static int32_t byte_array_to_string(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * array = checkbytearray(L, 1);
|
|
+ uint8_t buf[(array->length * 3) + 255];
|
|
+ uint8_t hexval[16] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };
|
|
+ char res[255 + (array->length * 3)]; /* make sure the buffer is big enough*/
|
|
+ int32_t i, n;
|
|
+ uint8_t *ptr = array->start + array->offset;
|
|
+
|
|
+ for (i = 0; i < array->length; i++) {
|
|
+ buf[i * 3] = hexval[(ptr[i] >> 4) & 0xF];
|
|
+ buf[(i * 3) + 1] = hexval[ptr[i] & 0x0F];
|
|
+ buf[(i * 3) + 2] = ' '; /* seperator */
|
|
+ }
|
|
+
|
|
+ buf[array->length * 3] = '\0';
|
|
+ n = sprintf(res, "byte_array: length: %d value: %s", array->length, buf);
|
|
+
|
|
+ lua_pushlstring(L, res, n);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static const struct luaL_Reg bytearray_lib_m [] = {
|
|
+ { "__len", get_byte_array_size },
|
|
+ { "__newindex", set_byte_array },
|
|
+ { "__index", get_byte_array },
|
|
+ { "__tostring", byte_array_to_string },
|
|
+ { NULL, NULL }
|
|
+};
|
|
+
|
|
+void luaopen_bytearraylib(lua_State *L)
|
|
+{
|
|
+ luaL_newmetatable(L, LUA_BYTE_ARRAY);
|
|
+ luaL_register(L, NULL, bytearray_lib_m);
|
|
+ lua_pop(L, 1);
|
|
+}
|
|
+
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/controller.c
|
|
@@ -0,0 +1,604 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ #include <linux/mm.h>
|
|
+#endif
|
|
+#include "controller.h"
|
|
+
|
|
+/* the array 'supported_protocols' holds all pointers to the
|
|
+ * static and dynamic protocol buffers. It is filled by the
|
|
+ * call to register_protbuf */
|
|
+static struct protocol_buf * supported_protocols[MAX_NR_OF_PROTOCOLS];
|
|
+
|
|
+/* C_API: the function 'get_protocol_buf' returns the pointer
|
|
+ * to the protocol buffer of a given protocol id. */
|
|
+struct protocol_buf * get_protocol_buf(uint32_t protocol_id)
|
|
+{
|
|
+ return (struct protocol_buf *)supported_protocols[protocol_id];
|
|
+}
|
|
+
|
|
+
|
|
+/* LUA_INT: the function 'gc_packet_segment' is triggered by the
|
|
+ * garbage collector whenever a userdata annotated with one of
|
|
+ * the protocol buffer metatable should be collected. */
|
|
+static int32_t gc_packet_segment(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * seg = (lua_packet_segment *)lua_touserdata(L, 1);
|
|
+ if (seg && seg->changes) {
|
|
+ seg->changes->ref_count--;
|
|
+ if (seg->changes->ref_count <= 0) {
|
|
+ kfree(seg->changes->field_length_changes);
|
|
+ kfree(seg->changes->field_offset_changes);
|
|
+ kfree(seg->changes);
|
|
+ seg->changes = NULL;
|
|
+ }
|
|
+ }
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+/* LUA_API: the function 'set_raw' is used to set the bytes of a segment
|
|
+ * in 'raw' mode. The function is per default available in each protocol
|
|
+ * buffer until it gets overridden by a specific setter function inside
|
|
+ * a protocol buffer.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment (implicit)
|
|
+ * 2. int32_t byte_value
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index, not used in this function
|
|
+ *
|
|
+ * Return: void
|
|
+ */
|
|
+static int32_t set_raw(lua_State *L)
|
|
+{
|
|
+ int32_t i;
|
|
+ uint32_t nob;
|
|
+ uint8_t byte;
|
|
+ uint8_t *ptr;
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ int32_t val = luaL_checkinteger(L, 2);
|
|
+
|
|
+ nob = 1 << CHAR_BIT;
|
|
+
|
|
+ luaL_argcheck(L, 0 <= val && val < nob, 2, "cannot cast value to char");
|
|
+
|
|
+ byte = (uint8_t)val;
|
|
+ ptr = seg->start + seg->offset;
|
|
+
|
|
+ for (i = 0; i < seg->length; i++)
|
|
+ ptr[i] = byte;
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'get_raw' is used to get the bytes of a segment
|
|
+ * in 'raw' mode. The function is per default available in each protocol
|
|
+ * buffer until it gets overridden by a specific getter function inside
|
|
+ * a protocol buffer.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment (implicit)
|
|
+ * 2. uint32_t offset
|
|
+ * 3. uint32_t length
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index, not used in this function
|
|
+ *
|
|
+ * Return:
|
|
+ * the byte array representing the given array
|
|
+ */
|
|
+static int32_t get_raw(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ init_byte_array(L, seg->start + seg->offset, seg->length, 1);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+/* LUA_API: The function 'get_segment' is used to get a new segment in 'raw' mode.
|
|
+ * Typically this function is applied on another raw segment in order
|
|
+ * to extract a part of the segment as new segment.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment, implicit through object oriented access seg:raw(..)
|
|
+ * 2. uint32_t offset, this indicates where to start the new segment, see e.g below.
|
|
+ * 3. uint32_t length, this indicates the size of the new segment
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index, not used in this function
|
|
+ *
|
|
+ * Return:
|
|
+ * 1. A lua_packet_segment annotated with the according metatable or False in
|
|
+ * case the input data is not valid
|
|
+ *
|
|
+ * Example:
|
|
+ *
|
|
+ * +------------------------+---------------------------------------+
|
|
+ * | function call | resulting lua_packet_segment |
|
|
+ * +========================+===+===+===+===+===+===+===+===+===+===+
|
|
+ * | seg = packet:raw(0,10) | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 |
|
|
+ * +------------------------+---+---+---+---+---+---+---+---+---+---+
|
|
+ * | 1st_half = seg:raw(0,5)| 0 | 1 | 2 | 3 | 4 | |
|
|
+ * +------------------------+---+---+---+---+---+---+---+---+---+---+
|
|
+ * | 2nd_half = seg:raw(5,5)| | 5 | 6 | 7 | 8 | 9 |
|
|
+ * +------------------------+-------------------+---+---+---+---+---+
|
|
+ */
|
|
+static int32_t get_segment(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ uint32_t offset = luaL_checkinteger(L, 2);
|
|
+ uint32_t length = luaL_checkinteger(L, 3);
|
|
+ lua_packet_segment * new = (lua_packet_segment *)lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+
|
|
+ new->start = seg->start;
|
|
+ new->offset = seg->offset + offset;
|
|
+ new->changes = NULL;
|
|
+ /* we allow a seg->length == 0 , this enables processing packets where the packetsize is not fixed (0 = not fixed)*/
|
|
+ if (seg->length != 0 && length > seg->length) {
|
|
+ lua_pushboolean(L, 0);
|
|
+ return 1;
|
|
+ }
|
|
+
|
|
+ new->length = length;
|
|
+ luaL_getmetatable(L, prot_buf->name);
|
|
+ lua_setmetatable(L, -2);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'get_segment_size' is used to get the size of a segment.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment, implicit through object oriented access seg:raw(..)
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index, not used in this function
|
|
+ *
|
|
+ * Return:
|
|
+ * 1. Size as lua_Number
|
|
+ */
|
|
+static int32_t get_segment_size(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushnumber(L, seg->length);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'get_segment_offset' is used to get the real offset
|
|
+ * of a segment. This function returns the offset of the segment to the start
|
|
+ * of the buffer. This means the following
|
|
+ * seg1 = packet:raw(2,10)
|
|
+ * seg2 = seg1:raw(3,5)
|
|
+ * offset = seg2:get_offset()
|
|
+ *
|
|
+ * will give an offset of 5, since the seg1 starts at offset 2, and seg2 starts
|
|
+ * at offset (seg1:get_offset() + 3).
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment, implicit through object oriented access seg:raw(..)
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index, not used in this function
|
|
+ *
|
|
+ * Return:
|
|
+ * 1. Offset as lua_Number
|
|
+ */
|
|
+static int32_t get_segment_offset(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushnumber(L, seg->offset);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: overwrites the __tostring function of a lua_packet_segment.
|
|
+ * this will print32_t a nicely formated string, including length,
|
|
+ * offset and name of the protocol buffer.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment (implicit)
|
|
+ *
|
|
+ * Returns:
|
|
+ * 1. the representing string
|
|
+ */
|
|
+static int32_t packet_segment_tostring(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ int32_t n;
|
|
+ char buf[128];
|
|
+
|
|
+ n = sprintf(buf, "type: %s, offset: %d, length: %d", prot_buf->name, seg->offset, seg->length);
|
|
+ lua_pushlstring(L, buf, n);
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static const struct luaL_Reg seg_access_functions [] = {
|
|
+ { "set", set_raw },
|
|
+ { "get", get_raw },
|
|
+ { "raw", get_segment },
|
|
+ { "get_offset", get_segment_offset },
|
|
+ { "get_size", get_segment_size },
|
|
+ { "to_bytes", get_raw },
|
|
+ { "__tostring", packet_segment_tostring },
|
|
+ { "__gc", gc_packet_segment },
|
|
+ { NULL, NULL }
|
|
+};
|
|
+
|
|
+/* C_API: the function 'get_metatable_from_protocol_type' is a helper
|
|
+ * used in controller.c as well as it may find usage in the static
|
|
+ * protocol buffers and byte array implementation. */
|
|
+void get_metatable_from_protocol_type(lua_State *L, int32_t type)
|
|
+{
|
|
+ char * table;
|
|
+ lua_getglobal(L, SUPPORTED_PROTOCOL_TABLE);
|
|
+ lua_rawgeti(L, -1, type);
|
|
+ table = (char *)luaL_checkstring(L, -1);
|
|
+ lua_pop(L, 2); /* pop the table SUPPORTED_PROTOCOL_TABLE and the string pushed by lua_gettable */
|
|
+ luaL_getmetatable(L, table);
|
|
+ return;
|
|
+}
|
|
+
|
|
+/* C_INT: the function 'payload_contains_protocol' is used internally.
|
|
+ * Depending if static or dynamic protocol buffer it calls the right
|
|
+ * validation function. */
|
|
+static int32_t payload_contains_protocol(lua_State *L, struct protocol_buf *prot_buf, lua_packet_segment *seg, uint32_t prot_type)
|
|
+{
|
|
+ if (prot_buf->is_dynamic)
|
|
+ return has_protocol_dynamic(L, prot_buf, seg, prot_type);
|
|
+ else
|
|
+ return prot_buf->has_protocol(L, prot_buf, seg, prot_type);
|
|
+}
|
|
+
|
|
+/* C_INT: the function 'protocol_get_field_changes' is used interally.
|
|
+ * It requests the field_changes struct calling the protocol buffers
|
|
+ * 'get_field_changes' function. This funciton is called, whenever
|
|
+ * the payload field with a given protocol type is requested inside
|
|
+ * the function 'get_protocol_field' */
|
|
+static struct field_changes * protocol_get_field_changes(lua_State *L, struct protocol_buf *prot_buf, lua_packet_segment * seg)
|
|
+{
|
|
+ struct field_changes * changes = NULL;
|
|
+
|
|
+ if (prot_buf->get_field_changes) {
|
|
+ if (prot_buf->is_dynamic)
|
|
+ changes = get_field_changes_dynamic(L, prot_buf, seg);
|
|
+ else
|
|
+ changes = prot_buf->get_field_changes(L, seg);
|
|
+ /* is already 1 when set by helper 'get_allocated_field_changes,
|
|
+ * since not every prot_buf may use this function we enforce it. */
|
|
+ changes->ref_count = 1;
|
|
+ }
|
|
+ return changes;
|
|
+}
|
|
+
|
|
+/* C_INT: the function 'get_field_offset_in_bytes' wrapps the logic of
|
|
+ * calculating the new length with considering the optional field_changes. */
|
|
+static int32_t get_field_offset_in_bytes(struct protocol_field * field, lua_packet_segment * seg, int32_t field_index)
|
|
+{
|
|
+ uint32_t nr_of_bits, nr_of_bytes, field_offset;
|
|
+
|
|
+ field_offset = field->offset;
|
|
+ /* do we need to manipulate the default values stored inside the protocol buffer ?? */
|
|
+ if (seg->changes)
|
|
+ field_offset += seg->changes->field_offset_changes[field_index];
|
|
+ /* how many bits remain */
|
|
+ nr_of_bits = field_offset & (CHAR_BIT - 1);
|
|
+ /* assuming CHAR_BIT == 2 ^ 3 */
|
|
+ nr_of_bytes = (field_offset - nr_of_bits) >> 3;
|
|
+
|
|
+ return seg->offset + nr_of_bytes;
|
|
+}
|
|
+
|
|
+/* C_INT: the function 'get_field_length_in_bytes' wrapps the logic of
|
|
+ * calculating the new offset with considering the optional field_changes. */
|
|
+static int32_t get_field_length_in_bytes(struct protocol_field * field, lua_packet_segment * seg, int32_t field_index)
|
|
+{
|
|
+ uint32_t nr_of_bits, nr_of_bytes, field_length;
|
|
+
|
|
+ field_length = field->length;
|
|
+ /* if the field length is smaller than 1 byte, we take the size of one byte
|
|
+ * we treat the case where field_length == 0 in a special way ...*/
|
|
+ if (field_length < CHAR_BIT && field_length > 0)
|
|
+ field_length = CHAR_BIT;
|
|
+
|
|
+ /* do we need to manipulate the default values stored inside the protocol buffer ?? */
|
|
+ if (seg->changes)
|
|
+ field_length += seg->changes->field_length_changes[field_index];
|
|
+ /* how many bits remain */
|
|
+ nr_of_bits = field_length & (CHAR_BIT - 1);
|
|
+ /* assuming CHAR_BIT == 2 ^ 3 */
|
|
+ nr_of_bytes = (field_length - nr_of_bits) >> 3;
|
|
+ return nr_of_bytes;
|
|
+}
|
|
+
|
|
+/* C_INT: the function 'initialize_field_getter_and_setter' initializes
|
|
+ * the setter and getter function of the field, considering the optional
|
|
+ * field manipulator functions defined inside the protocol buffers. */
|
|
+static void initialize_field_getter_and_setter(lua_State *L, struct protocol_buf *prot_buf, int32_t field_index)
|
|
+{
|
|
+ /* lets check if there is a metatable on top of the stack */
|
|
+ struct protocol_field * f = (struct protocol_field *)&prot_buf->protocol_fields[field_index];
|
|
+
|
|
+ if (!lua_istable(L, -1)) luaL_error(L, "cannot initialize getter and setter for field %s->%s, "
|
|
+ "not a table on top of the stack, is '%s'", prot_buf->name, f->name, lua_typename(L, lua_type(L, -1)));
|
|
+
|
|
+ /* is there a 'getter' to initialize ? */
|
|
+ lua_pushlightuserdata(L, prot_buf); /* push upvalue 1 */
|
|
+ lua_pushinteger(L, field_index); /* push upvalue 2 */
|
|
+ if (f->get) {
|
|
+ if (prot_buf->is_dynamic)
|
|
+ lua_pushcclosure(L, field_dynamic_getter, 2);
|
|
+ else
|
|
+ lua_pushcclosure(L, f->get, 2);
|
|
+ }else
|
|
+ /* there is no specific getter defined - fall back to 'get_raw' */
|
|
+ lua_pushcclosure(L, get_raw, 2);
|
|
+ /* set the metatable field 'get' */
|
|
+ lua_setfield(L, -2, "get");
|
|
+
|
|
+ /* is there a 'setter' to initialize ? */
|
|
+ lua_pushlightuserdata(L, prot_buf); /* push upvalue 1 */
|
|
+ lua_pushinteger(L, field_index); /* push upvalue 2 */
|
|
+ if (f->set) {
|
|
+ if (prot_buf->is_dynamic)
|
|
+ lua_pushcclosure(L, field_dynamic_setter, 2);
|
|
+ else
|
|
+ lua_pushcclosure(L, f->set, 2);
|
|
+ }else
|
|
+ /* there is no specific setter defined - fall back to 'set_raw' */
|
|
+ lua_pushcclosure(L, set_raw, 2);
|
|
+ /* set the metatable field 'set' */
|
|
+ lua_setfield(L, -2, "set");
|
|
+}
|
|
+
|
|
+/* LUA_API: 'get_protocol_field' is used in Lua as a closure for each field of a protocol
|
|
+ * buffer. E.g a call to ip = packet:data(packet_ip) will go to this function,
|
|
+ * and trigger the conversion of the raw packet to a ip packet. Each call
|
|
+ * to a field function of an IP packet, like ip:daddr() uses this function
|
|
+ * to to return the right data. In each case you will end up either with a
|
|
+ * new packet segment (annotated with the proper metatable) or a boolean
|
|
+ * value (False) if something went wrong. In the case everything went fine,
|
|
+ * the newly created lua_packet_segment is annotated with the proper
|
|
+ * metatable where the fields get and set also contain the specific getter
|
|
+ * and setter functions given by the protocol buffer. E.g. the function call
|
|
+ * ip:daddr():get() or ip:daddr():set(...) will call the proper function
|
|
+ * defined inside the corresponding field definition.
|
|
+ *
|
|
+ * Parameters:
|
|
+ * 1. lua_packet_segment, implicit through object oriented access seg:raw(..)
|
|
+ * 2. type of the protocol buffer, optional, and only used if the accessed
|
|
+ * field is the payload field. If a type is provided for the access of the
|
|
+ * payload field, the function tries to convert the data pointed to by the
|
|
+ * payload field to the given type. To check if such a conversion is
|
|
+ * possible, it calls the function pointed to by the protocol buffer member
|
|
+ * has_protocol. If this function returns True, the conversion takes place.
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. struct protocol_buf*
|
|
+ * 2. int32_t field index
|
|
+ *
|
|
+ * Return:
|
|
+ * 1. A lua_packet_segment annotated with the according metatable or False in
|
|
+ * case the input data is not valid
|
|
+ */
|
|
+static int32_t get_protocol_field(lua_State *L)
|
|
+{
|
|
+ int32_t prot_type;
|
|
+ lua_packet_segment * seg, *new;
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ int32_t field_index = lua_tointeger(L, lua_upvalueindex(2));
|
|
+ struct protocol_field * field = &prot_buf->protocol_fields[field_index];
|
|
+
|
|
+ /* get the current packet segment */
|
|
+ seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ /* initialize the new packet segment */
|
|
+ new = (lua_packet_segment *)lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+ new->start = seg->start; /* the start is unchanged */
|
|
+ new->offset = get_field_offset_in_bytes(field, seg, field_index);
|
|
+ new->length = get_field_length_in_bytes(field, seg, field_index);
|
|
+
|
|
+ /* if new->length == 0 then no configuration was done, we guess the size by subtracting the
|
|
+ * new offset from the packet length. since the old length is getting initialized by the
|
|
+ * netfilter extension this assumption holds for the very last field of the protocol.
|
|
+ * this 'feature' should be used by protocol buffers containing a payload, whereas the
|
|
+ * payload field is the last field of the buffer. However, at compile-time unknown field
|
|
+ * sizes (and offsets) of fields not being placed at the end of the protocol should be
|
|
+ * initialized using the 'get_field_changes' hook system. */
|
|
+ if (new->length == 0)
|
|
+ new->length = (seg->length + seg->offset) - (new->offset);
|
|
+ /*
|
|
+ printf("%s->%s:: seg->offset %i, seg->length %i, new->offset %i, new->length %i\n",
|
|
+ prot_buf->name, field->name, seg->offset, seg->length, new->offset, new->length);
|
|
+ */
|
|
+ /* special care for packet payload requests */
|
|
+ if (prot_buf->payload_field != NULL && strcmp(prot_buf->payload_field, field->name) == 0) {
|
|
+ /* we know the payload field is requested */
|
|
+ /* the requested payload can be delivered either as a common segment or as
|
|
+ * an other packet type, such a conversion needs an extra protocol parameter
|
|
+ * ... so lets check */
|
|
+
|
|
+ if (lua_isnumber(L, 2)) {
|
|
+ /* we have an extra parameter, ... lets see if it is a valid protocol
|
|
+ * the parameter is the index of the 'supported_protocols'-array member */
|
|
+ prot_type = lua_tointeger(L, 2);
|
|
+ if (prot_type >= 0 && prot_type < PACKET_SENTINEL) {
|
|
+ /* we are sure the purpose of the request is to get the payload data,
|
|
+ * converted to the given protocol. lets check if the payload contains
|
|
+ * data of the given protocol */
|
|
+ if (payload_contains_protocol(L, prot_buf, seg, prot_type)) {
|
|
+ /* success, we can push the metatable for the given protocol */
|
|
+ get_metatable_from_protocol_type(L, prot_type);
|
|
+ if (!lua_isnil(L, -1)) /* check if the metatable was found */
|
|
+ /* perhaps the field offsets and lengths of the containing protocol
|
|
+ * are not set correctly. request the optional 'field_changes' structure
|
|
+ * holding the changes for lengths and offsets. */
|
|
+ new->changes = protocol_get_field_changes(L, get_protocol_buf(prot_type), new);
|
|
+ else{
|
|
+ /* failed, the requested protocol is not available
|
|
+ * we push false and return */
|
|
+ lua_pop(L, 1); /* pop the userdata */
|
|
+ lua_pushboolean(L, 0);
|
|
+ return 1;
|
|
+ }
|
|
+ }else{
|
|
+ /* payload does not carry the provided protocol */
|
|
+ /* we push false and return */
|
|
+ lua_pop(L, 1); /* pop the userdata */
|
|
+ lua_pushboolean(L, 0);
|
|
+ return 1;
|
|
+ }
|
|
+ }else{
|
|
+ /* unknown protocol */
|
|
+ lua_pop(L, 1); /* pop the userdata */
|
|
+ luaL_error(L, "provided protocol is unknown");
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+
|
|
+ /* if there is still the 'new' userdata on the top, we push our own metatable */
|
|
+ if (lua_isuserdata(L, -1)) {
|
|
+ luaL_getmetatable(L, prot_buf->name);
|
|
+ new->changes = seg->changes;
|
|
+ if (seg->changes)
|
|
+ new->changes->ref_count++;
|
|
+ }
|
|
+
|
|
+ /* a new packet segment is at index -2 , and the proper metatable at index -1 of the stack
|
|
+ * lets set the propper setter and getter function for the requested field */
|
|
+ initialize_field_getter_and_setter(L, prot_buf, field_index);
|
|
+
|
|
+ lua_setmetatable(L, -2);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* C_API: 'register_protbuf' is only used internally. This function takes a
|
|
+ * pointer to a fully initialized protocol buffer struct and registers it
|
|
+ * inside the Lua state. Registering means:
|
|
+ *
|
|
+ * 1. it creates a new metatable with the name of the protocol buffer.
|
|
+ * 2. it registers the default functions which are stored in the luaL_Reg
|
|
+ * array seg_access_functions.
|
|
+ * 3. it loops over the protocol fields stored at prot_buf->protocol_fields
|
|
+ * and registers a new function (using the field name) inside the
|
|
+ * metatable. Each field points to the function 'get_protocol_field'
|
|
+ * which acts as a closure taking a pointer to the protocol buffer as
|
|
+ * well as the index of the field as upvalues.
|
|
+ * 4. The protocol index, serves as numerical identifier of this protocol
|
|
+ * buffer or even of the protocol itself. This index is stored as a
|
|
+ * global value inside the Lua state as well as inside the Lua table
|
|
+ * 'supported_protocols'. Assuming the name of a procotol buffer is
|
|
+ * "packet_ip" the following statements are true:
|
|
+ *
|
|
+ * supported_protocols[protocol_index] == "packet_ip"
|
|
+ * packet_ip == protocol_index
|
|
+ *
|
|
+ * This allows you to get all registered protocols from within Lua. This
|
|
+ * is especially usefull for the dynamic protocol buffers where you have
|
|
+ * to provide your own "has_protocol"-function, which probably needs the
|
|
+ * information on which protocols it is able to contain.
|
|
+ */
|
|
+void register_protbuf(lua_State *L, struct protocol_buf * prot_buf, uint32_t protocol_index)
|
|
+{
|
|
+ int32_t field_index;
|
|
+ luaL_Reg *reg = (struct luaL_Reg *)seg_access_functions;
|
|
+ struct protocol_field * field = prot_buf->protocol_fields;
|
|
+
|
|
+ luaL_newmetatable(L, prot_buf->name);
|
|
+
|
|
+ /* metatable.__index = metatable */
|
|
+ lua_pushvalue(L, -1); /* duplicates the metatable */
|
|
+ lua_setfield(L, -2, "__index");
|
|
+
|
|
+ /* pushing default functions */
|
|
+ for (; reg->name; reg++) {
|
|
+ lua_pushlightuserdata(L, (void *)prot_buf);
|
|
+ lua_pushcclosure(L, reg->func, 1);
|
|
+ lua_setfield(L, -2, reg->name);
|
|
+ }
|
|
+
|
|
+ /* pushing functions specific to the protocol buffer */
|
|
+ for (field_index = 0; field->name; field++, field_index++) {
|
|
+ lua_pushlightuserdata(L, (void *)prot_buf); /* upvalue: prot_buf */
|
|
+ lua_pushinteger(L, field_index); /* upvalue: index of protocol field */
|
|
+ lua_pushcclosure(L, get_protocol_field, 2);
|
|
+ lua_setfield(L, -2, field->name);
|
|
+ }
|
|
+ /* pop the metatable */
|
|
+ lua_pop(L, 1);
|
|
+
|
|
+ /* registering the array-index as the protocol_id*/
|
|
+ lua_getglobal(L, "_G");
|
|
+ lua_pushinteger(L, protocol_index);
|
|
+ lua_setfield(L, -2, prot_buf->name);
|
|
+ lua_pop(L, 1); /* pop _G */
|
|
+
|
|
+ lua_getglobal(L, SUPPORTED_PROTOCOL_TABLE);
|
|
+ lua_pushstring(L, prot_buf->name);
|
|
+ lua_rawseti(L, -2, protocol_index);
|
|
+
|
|
+ lua_pop(L, 1); /* pop SUPPORTED_PROTOCOL_TABLE */
|
|
+
|
|
+ supported_protocols[protocol_index] = prot_buf;
|
|
+}
|
|
+
|
|
+void luaopen_controller(lua_State *L)
|
|
+{
|
|
+ /* registering a table inside the _G with table[protocol_index] = prot_buf->name */
|
|
+ lua_getglobal(L, "_G");
|
|
+ lua_newtable(L);
|
|
+ lua_setfield(L, -2, SUPPORTED_PROTOCOL_TABLE);
|
|
+ lua_pop(L, 1); /* pop _G */
|
|
+
|
|
+ luaopen_protbuf_raw(L);
|
|
+ luaopen_protbuf_eth(L);
|
|
+ luaopen_protbuf_ip(L);
|
|
+ luaopen_protbuf_icmp(L);
|
|
+ luaopen_protbuf_tcp(L);
|
|
+ luaopen_protbuf_tcp_options(L);
|
|
+ luaopen_protbuf_udp(L);
|
|
+ luaopen_protbuf_tftp(L);
|
|
+ luaopen_protbuf_dynamic(L);
|
|
+ /* should follow all other static buffers */
|
|
+#if defined(__KERNEL__)
|
|
+ luaopen_nflib(L);
|
|
+#endif
|
|
+
|
|
+ luaopen_bytearraylib(L);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/controller.h
|
|
@@ -0,0 +1,264 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#ifndef CONTROLLER_H_
|
|
+#define CONTROLLER_H_
|
|
+
|
|
+#include "stdlib.h" /* wrapper */
|
|
+#include "string.h" /* wrapper */
|
|
+#include "lua.h"
|
|
+#include "lualib.h"
|
|
+#include "lauxlib.h"
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+#include <linux/skbuff.h>
|
|
+#include <linux/slab.h>
|
|
+#include <linux/vmalloc.h>
|
|
+#endif
|
|
+
|
|
+
|
|
+/* to compile the stuff in userspace (for testing)*/
|
|
+#if !defined(__KERNEL__)
|
|
+#include <stdint.h>
|
|
+#define pr_debug printf;
|
|
+
|
|
+#define kmalloc(size, type) malloc(size)
|
|
+#define kfree(ptr) free(ptr)
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
+/**********************************************************************/
|
|
+/* nf Lua configuration */
|
|
+/**********************************************************************/
|
|
+#define MAX_NR_OF_PROTOCOLS 16
|
|
+#define SUPPORTED_PROTOCOL_TABLE "supported_protocols"
|
|
+
|
|
+#define MAX_NR_OF_FIELDS_IN_DYN_PROT_BUF 32
|
|
+
|
|
+
|
|
+/**********************************************************************/
|
|
+/* Static Protocol Buffer configuration */
|
|
+/**********************************************************************/
|
|
+
|
|
+/* the definitions of the stringified expression of the prot_bufs...
|
|
+ * make sure all static prot_bufs are listed and are unique */
|
|
+#define LUA_PACKET_SEG_RAW "packet_raw"
|
|
+#define LUA_PACKET_SEG_ETH "packet_eth"
|
|
+#define LUA_PACKET_SEG_ICMP "packet_icmp"
|
|
+#define LUA_PACKET_SEG_IP "packet_ip"
|
|
+#define LUA_PACKET_SEG_TCP "packet_tcp"
|
|
+#define LUA_PACKET_SEG_TCP_OPT "packet_tcp_opt"
|
|
+#define LUA_PACKET_SEG_UDP "packet_udp"
|
|
+#define LUA_PACKET_SEG_TFTP "packet_tftp"
|
|
+
|
|
+/* the enum holding all static prot_bufs... make sure it contains all
|
|
+ * static prot_bufs */
|
|
+enum PROT_BUF {
|
|
+ PACKET_RAW,
|
|
+ PACKET_ETH,
|
|
+ PACKET_IP,
|
|
+ PACKET_ICMP,
|
|
+ PACKET_TCP,
|
|
+ PACKET_TCP_OPTIONS,
|
|
+ PACKET_UDP,
|
|
+ PACKET_TFTP,
|
|
+ PACKET_DYNAMIC,
|
|
+ PACKET_SENTINEL
|
|
+};
|
|
+
|
|
+/* the luaopen-function of the prot_bufs... make sure it is called
|
|
+ * inside luaopen_controller */
|
|
+void luaopen_protbuf_raw(lua_State *L);
|
|
+void luaopen_protbuf_eth(lua_State *L);
|
|
+void luaopen_protbuf_ip(lua_State *L);
|
|
+void luaopen_protbuf_icmp(lua_State *L);
|
|
+void luaopen_protbuf_tcp(lua_State *L);
|
|
+void luaopen_protbuf_tcp_options(lua_State *L);
|
|
+void luaopen_protbuf_udp(lua_State *L);
|
|
+void luaopen_protbuf_tftp(lua_State *L);
|
|
+void luaopen_protbuf_dynamic(lua_State *L);
|
|
+
|
|
+/**********************************************************************/
|
|
+/* field changes */
|
|
+/**********************************************************************/
|
|
+struct field_changes {
|
|
+ int ref_count;
|
|
+ int *field_length_changes;
|
|
+ int *field_offset_changes;
|
|
+};
|
|
+
|
|
+/**********************************************************************/
|
|
+/* lua packet segment */
|
|
+/* ------------------ */
|
|
+/* The struct lua_packet_segment is the integral part of a Lua packet.*/
|
|
+/* At the very beginning, when a new packet arrives in `lua_tg`_ such */
|
|
+/* a struct is initialized. The field start then points to the lowest */
|
|
+/* available header inside the sk_buff structure. During packet */
|
|
+/* processing the start pointer remains the same, only the offset and */
|
|
+/* length value change. */
|
|
+/**********************************************************************/
|
|
+#define checkpacketseg(L, i, seg_type) \
|
|
+ (lua_packet_segment *)luaL_checkudata(L, i, seg_type)
|
|
+
|
|
+typedef struct lua_packet_segment {
|
|
+ unsigned int offset;
|
|
+ unsigned int length;
|
|
+ struct field_changes * changes;
|
|
+ unsigned char * start; /* need to be at the end because of the memory alignment */
|
|
+} lua_packet_segment;
|
|
+
|
|
+/**********************************************************************/
|
|
+/* protocol field */
|
|
+/* -------------- */
|
|
+/* This structure is a container for the field definitions used by the*/
|
|
+/* protocol buffer. Each protocol field is expressed using this struct*/
|
|
+/* Have a look at the protocol buffers to see how the struct gets */
|
|
+/* initialized. */
|
|
+/* */
|
|
+/* name: */
|
|
+/* This member expresses the name of the field, ending */
|
|
+/* in its own Lua function to access the field. */
|
|
+/* offset / length: */
|
|
+/* These members do specify the position inside the protocol header */
|
|
+/* in bits (not bytes!). */
|
|
+/* get / set: */
|
|
+/* The get and set functions take a function pointer pointing to the*/
|
|
+/* specific getter and setter function for this field. */
|
|
+/**********************************************************************/
|
|
+struct protocol_field {
|
|
+ const char * name;
|
|
+ uint32_t offset;
|
|
+ uint32_t length;
|
|
+ lua_CFunction get;
|
|
+ lua_CFunction set;
|
|
+};
|
|
+#define PROT_FIELD_SENTINEL { NULL, 0, 0, NULL, NULL }
|
|
+
|
|
+
|
|
+/**********************************************************************/
|
|
+/* protocol_buf */
|
|
+/**********************************************************************/
|
|
+/* This structure is a container for all the information needed for a
|
|
+ * protocol buffer. It gets initialized in each protocol buffer header
|
|
+ * file or for the dynamic protocol buffers on runtime using the
|
|
+ * 'register_dynamic_protocol_buffer' function.
|
|
+ *
|
|
+ * name:
|
|
+ * This member is used throughout the system. It is also exported
|
|
+ * to Lua as a variable name holding the index of the 'supported_protocols'
|
|
+ * array. The name is also used as the name of the generated Lua
|
|
+ * metatable, that is why inside the macro checkpacketseg_ it
|
|
+ * is always the name of a protocol buffer that is passed as the
|
|
+ * second parameter.
|
|
+ * payload_field:
|
|
+ * This member holds the string of the field responsible for payload
|
|
+ * data. The payload field of a protocol has an extra property, since
|
|
+ * it can be used to invoke another protocol buffer that is applied to
|
|
+ * the payload content.
|
|
+ * has_protocol:
|
|
+ * This member is used together with the payload_field. Since we must
|
|
+ * be sure that the payload content does really contain a protocol
|
|
+ * of type X. The function pointed to by has_protocol checks if the
|
|
+ * protocol buffer X can be applied on the payload_data.
|
|
+ * protocol_fields:
|
|
+ * This member points to the array of 'protocol_field' structures
|
|
+ * get_field_changes:
|
|
+ * This member is optional. It is used to return a pointer to an initialized
|
|
+ * field_changes struct. The function is called, whenever the payload field
|
|
+ * is requested with a given protocol type. Usually this function will
|
|
+ * initialize the field_changes struct depending on the content of the
|
|
+ * payload data. e.g.
|
|
+ * tcp = ip:data(packet_tcp)
|
|
+ * such a request will call the 'get_field_changes' function of the tcp
|
|
+ * protocol buffer. This enables, that the tcp options field have the proper
|
|
+ * length as well as the tcp data start at the right offset.
|
|
+ */
|
|
+struct protocol_buf {
|
|
+ int is_dynamic;
|
|
+ const char * name;
|
|
+ char * payload_field;
|
|
+ int (*has_protocol)(lua_State *L, struct protocol_buf *prot_buf, lua_packet_segment * seg, int type);
|
|
+ struct protocol_field * protocol_fields;
|
|
+ struct field_changes * (*get_field_changes)(lua_State *L, lua_packet_segment * seg);
|
|
+};
|
|
+
|
|
+/**********************************************************************/
|
|
+/* lua byte array library */
|
|
+/**********************************************************************/
|
|
+#define LUA_BYTE_ARRAY "byte_array"
|
|
+#define checkbytearray(L, i) \
|
|
+ (lua_packet_segment *)luaL_checkudata(L, i, LUA_BYTE_ARRAY)
|
|
+lua_packet_segment * init_byte_array(lua_State *L, unsigned char * start, int length, int do_copy);
|
|
+void luaopen_bytearraylib(lua_State *L);
|
|
+
|
|
+
|
|
+/**********************************************************************/
|
|
+/* lua netfilter environment library */
|
|
+/**********************************************************************/
|
|
+#define NETFILTER_LIB "nf"
|
|
+#if defined(__KERNEL__)
|
|
+ struct lua_env {
|
|
+ lua_State *L;
|
|
+ /* perhaps more to come here (e.g. a state per CPU) */
|
|
+ };
|
|
+ #define LUA_ENV "lua_env"
|
|
+ #define checkluaenv(L, i) \
|
|
+ (struct lua_env *)luaL_checkudata(L, i, LUA_ENV)
|
|
+
|
|
+ void luaopen_nflib(lua_State *L);
|
|
+#endif
|
|
+
|
|
+void cleanup_dynamic_prot_bufs(void); /* freeing all dynamic prot bufs */
|
|
+/**********************************************************************/
|
|
+/* lua protbuf helpers */
|
|
+/**********************************************************************/
|
|
+int get_1_bit_generic(lua_State *L);
|
|
+int set_1_bit_generic(lua_State *L);
|
|
+int get_lower_4_bit_generic(lua_State *L);
|
|
+int set_lower_4_bit_generic(lua_State *L);
|
|
+int get_upper_4_bit_generic(lua_State *L);
|
|
+int set_upper_4_bit_generic(lua_State *L);
|
|
+int get_8_bit_generic(lua_State *L);
|
|
+int set_8_bit_generic(lua_State *L);
|
|
+int get_16_bit_generic(lua_State *L);
|
|
+int set_16_bit_generic(lua_State *L);
|
|
+int get_32_bit_generic(lua_State *L);
|
|
+int set_32_bit_generic(lua_State *L);
|
|
+int set_data_generic(lua_State *L);
|
|
+int get_string_generic(lua_State *L);
|
|
+int get_byte_generic_str(lua_State *L);
|
|
+struct field_changes * get_allocated_field_changes(lua_State *L, int nr_of_fields);
|
|
+
|
|
+/* only used by the dynamic prot buf subsystem */
|
|
+#define MAX_NR_OF_DYN_PROT_BUFS 16
|
|
+int field_dynamic_setter(lua_State *L);
|
|
+int field_dynamic_getter(lua_State *L);
|
|
+int has_protocol_dynamic(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int type);
|
|
+struct field_changes * get_field_changes_dynamic(lua_State *L, struct protocol_buf *prot_buf, lua_packet_segment * seg);
|
|
+
|
|
+/**********************************************************************/
|
|
+/* lua controller API */
|
|
+/**********************************************************************/
|
|
+void luaopen_controller(lua_State *L);
|
|
+struct protocol_buf * get_protocol_buf(unsigned int protocol_id);
|
|
+void get_metatable_from_protocol_type(lua_State *L, int type);
|
|
+void register_protbuf(lua_State *L, struct protocol_buf * prot_buf, unsigned int protocol_index);
|
|
+
|
|
+
|
|
+#endif /* CONTROLLER_H_ */
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/Kbuild
|
|
@@ -0,0 +1,49 @@
|
|
+# -*- Makefile -*-
|
|
+
|
|
+# Adding debug options
|
|
+EXTRA_CFLAGS += -DDEBUG
|
|
+
|
|
+obj-m += xt_LUA.o
|
|
+
|
|
+EXTRA_CFLAGS += -I$(src)/prot_buf_new
|
|
+xt_LUA-y += xt_LUA_target.o \
|
|
+
|
|
+xt_LUA-y += nf_lua.o \
|
|
+ prot_buf_helpers.o \
|
|
+ byte_array.o \
|
|
+ controller.o \
|
|
+ prot_buf_ethernet.o \
|
|
+ prot_buf_icmp.o \
|
|
+ prot_buf_ip.o \
|
|
+ prot_buf_raw.o \
|
|
+ prot_buf_tcp.o \
|
|
+ prot_buf_udp.o \
|
|
+ prot_buf_tftp.o \
|
|
+ prot_buf_dynamic.o \
|
|
+
|
|
+
|
|
+# Adding Lua Support
|
|
+EXTRA_CFLAGS += -I$(src)/lua -I$(src)/lua/include
|
|
+xt_LUA-y += lua/lapi.o \
|
|
+ lua/lbaselib.o \
|
|
+ lua/lcode.o \
|
|
+ lua/ldebug.o \
|
|
+ lua/ldo.o \
|
|
+ lua/ldump.o \
|
|
+ lua/lfunc.o \
|
|
+ lua/lgc.o \
|
|
+ lua/llex.o \
|
|
+ lua/lmem.o \
|
|
+ lua/lobject.o \
|
|
+ lua/lopcodes.o \
|
|
+ lua/lparser.o \
|
|
+ lua/lstate.o \
|
|
+ lua/lstring.o \
|
|
+ lua/lstrlib.o \
|
|
+ lua/ltable.o \
|
|
+ lua/ltablib.o \
|
|
+ lua/ltm.o \
|
|
+ lua/lundump.o \
|
|
+ lua/lvm.o \
|
|
+ lua/lzio.o \
|
|
+ lua/lauxlib.o \
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/libxt_LUA.c
|
|
@@ -0,0 +1,191 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include <getopt.h>
|
|
+#include <stdio.h>
|
|
+#include <stdbool.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+#include <xtables.h>
|
|
+#include <linux/netfilter.h>
|
|
+#include <linux/netfilter/x_tables.h>
|
|
+#include "xt_LUA.h"
|
|
+
|
|
+enum {
|
|
+ FLAG_SCRIPT = 1 << 0,
|
|
+ FLAG_STATE = 1 << 1,
|
|
+ FLAG_FUNCTION = 1 << 2,
|
|
+};
|
|
+
|
|
+static const struct option lua_tg_opts[] = {
|
|
+ { .name = "script", .has_arg = true, .val = 's' },
|
|
+ { .name = "state", .has_arg = true, .val = 'l' },
|
|
+ { .name = "function", .has_arg = true, .val = 'f' },
|
|
+ { NULL },
|
|
+};
|
|
+
|
|
+
|
|
+static void lua_tg_help(void)
|
|
+{
|
|
+ printf(
|
|
+ "LUA target options:\n"
|
|
+ " --script SCRIPT Process packet with the Lua script given by SCRIPT\n"
|
|
+ " \n"
|
|
+ " --state ID Process packet within the Lua state given by ID.\n"
|
|
+ " Omitting --state infers the ID 0, which can be\n"
|
|
+ " refered to the 'global' state.\n"
|
|
+ " \n"
|
|
+ " --function FUNCTION Name of the function that processes the Lua packet\n"
|
|
+ "\n");
|
|
+}
|
|
+
|
|
+static void
|
|
+lua_tg_init(struct xt_entry_target *target)
|
|
+{
|
|
+ struct xt_lua_tginfo *info = (void *)target->data;
|
|
+
|
|
+ info->state_id = 0;
|
|
+ strncpy(info->function, "process_packet\0", sizeof("process_packet\0"));
|
|
+}
|
|
+
|
|
+static int
|
|
+lua_tg_parse(int32_t c, char **argv, int32_t invert, uint32_t *flags,
|
|
+ const void *entry, struct xt_entry_target **target)
|
|
+{
|
|
+ struct xt_lua_tginfo *info = (void *)(*target)->data;
|
|
+ char buf[MAX_SCRIPT_SIZE];
|
|
+ long script_size;
|
|
+ uint32_t state_id;
|
|
+ FILE *file;
|
|
+
|
|
+ switch (c) {
|
|
+ case 's':
|
|
+ if (*flags & FLAG_SCRIPT)
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Cannot specify --script more than once");
|
|
+
|
|
+ if (strlen(optarg) > sizeof(info->filename))
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Maximum script length is %zu",
|
|
+ sizeof(info->filename));
|
|
+
|
|
+ if (strchr(optarg, '\n'))
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Newlines not allowed in script name");
|
|
+ file = fopen(optarg, "rb");
|
|
+ if (file != NULL) {
|
|
+ fseek(file, 0, SEEK_END);
|
|
+ script_size = ftell(file);
|
|
+ if (script_size > MAX_SCRIPT_SIZE)
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: The size of the script is too big");
|
|
+
|
|
+ fseek(file, 0, SEEK_SET);
|
|
+ fread(buf, script_size, 1, file);
|
|
+ fclose(file);
|
|
+ } else
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Cannot open script %s", optarg);
|
|
+
|
|
+ strncpy(info->filename, optarg, sizeof(info->filename));
|
|
+ strncpy(info->buf, buf, sizeof(info->buf));
|
|
+ info->script_size = script_size;
|
|
+
|
|
+ *flags |= FLAG_SCRIPT;
|
|
+ return true;
|
|
+
|
|
+ case 'l':
|
|
+ if (*flags & FLAG_STATE)
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Cannot specify --state more than once");
|
|
+
|
|
+ if (!xtables_strtoui(optarg, NULL, &state_id, 0, 8))
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Invalid --state %s", optarg);
|
|
+
|
|
+ info->state_id = state_id;
|
|
+ *flags |= FLAG_STATE;
|
|
+ return true;
|
|
+
|
|
+ case 'f':
|
|
+ if (*flags & FLAG_FUNCTION)
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Cannot specify --function more than once");
|
|
+ if (strlen(optarg) > sizeof(info->function))
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Maximum function length is %zu",
|
|
+ sizeof(info->function));
|
|
+
|
|
+ if (strchr(optarg, '\n'))
|
|
+ xtables_error(PARAMETER_PROBLEM,
|
|
+ "LUA: Newlines not allowed in function name");
|
|
+
|
|
+ strncpy(info->function, optarg, sizeof(info->function));
|
|
+
|
|
+ *flags |= FLAG_FUNCTION;
|
|
+ return true;
|
|
+ }
|
|
+
|
|
+ return false;
|
|
+}
|
|
+
|
|
+static void
|
|
+lua_tg_check(uint32_t flags)
|
|
+{
|
|
+ if (flags == 0)
|
|
+ xtables_error(PARAMETER_PROBLEM, "LUA: --script parameter required");
|
|
+}
|
|
+
|
|
+static void
|
|
+lua_tg_print(const void *entry, const struct xt_entry_target *target,
|
|
+ int32_t numeric)
|
|
+{
|
|
+ const struct xt_lua_tginfo *info = (const void *)target->data;
|
|
+
|
|
+ printf("LUA script: %s ", info->filename);
|
|
+}
|
|
+
|
|
+static void
|
|
+lua_tg_save(const void *entry, const struct xt_entry_target *target)
|
|
+{
|
|
+ const struct xt_lua_tginfo *info = (const void *)target->data;
|
|
+
|
|
+ printf("--script %s ", info->filename);
|
|
+}
|
|
+
|
|
+static struct xtables_target lua_tg_reg = {
|
|
+ .name = "LUA",
|
|
+ .version = XTABLES_VERSION,
|
|
+ .revision = 0,
|
|
+ .family = NFPROTO_UNSPEC,
|
|
+ .size = XT_ALIGN(sizeof(struct xt_lua_tginfo)),
|
|
+ .userspacesize = XT_ALIGN(sizeof(struct xt_lua_tginfo)),
|
|
+ .help = lua_tg_help,
|
|
+ .init = lua_tg_init,
|
|
+ .parse = lua_tg_parse,
|
|
+ .final_check = lua_tg_check,
|
|
+ .print = lua_tg_print,
|
|
+ .save = lua_tg_save,
|
|
+ .extra_opts = lua_tg_opts,
|
|
+};
|
|
+
|
|
+static __attribute__((constructor)) void lua_tg_ldr(void)
|
|
+{
|
|
+ xtables_register_target(&lua_tg_reg);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/libxt_LUA.man
|
|
@@ -0,0 +1 @@
|
|
+Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/ctype.h
|
|
@@ -0,0 +1,11 @@
|
|
+#include <linux/ctype.h>
|
|
+#undef isalnum
|
|
+#define isalnum(c) (((__ismask(c)&(_U|_L|_D)) != 0) && (c > 0))
|
|
+#undef isalpha
|
|
+#define isalpha(c) (((__ismask(c)&(_U|_L)) != 0) && (c > 0))
|
|
+#undef iscntrl
|
|
+#define iscntrl(c) (((__ismask(c)&(_C)) != 0) && (c > 0))
|
|
+#undef isdigit
|
|
+#define isdigit(c) (((__ismask(c)&(_D)) != 0) && (c > 0))
|
|
+#undef isspace
|
|
+#define isspace(c) (((__ismask(c)&(_S)) != 0) && (c > 0))
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/errno.h
|
|
@@ -0,0 +1 @@
|
|
+#include <linux/errno.h>
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/locale.h
|
|
@@ -0,0 +1,5 @@
|
|
+struct lconv {
|
|
+ char * decimal_point ;
|
|
+} ;
|
|
+
|
|
+#define localeconv() NULL
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/setjmp.h
|
|
@@ -0,0 +1,26 @@
|
|
+/*
|
|
+ * arch/um/include/sysdep-i386/archsetjmp.h
|
|
+ */
|
|
+
|
|
+#ifndef _KLIBC_ARCHSETJMP_H
|
|
+#define _KLIBC_ARCHSETJMP_H
|
|
+
|
|
+struct __jmp_buf {
|
|
+ unsigned int __ebx;
|
|
+ unsigned int __esp;
|
|
+ unsigned int __ebp;
|
|
+ unsigned int __esi;
|
|
+ unsigned int __edi;
|
|
+ unsigned int __eip;
|
|
+};
|
|
+
|
|
+typedef struct __jmp_buf jmp_buf[1];
|
|
+
|
|
+#define JB_IP __eip
|
|
+#define JB_SP __esp
|
|
+
|
|
+int setjmp(jmp_buf);
|
|
+void longjmp(jmp_buf, int);
|
|
+
|
|
+#endif /* _SETJMP_H */
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/stdio.h
|
|
@@ -0,0 +1 @@
|
|
+#include <linux/kernel.h>
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/stdlib.h
|
|
@@ -0,0 +1,7 @@
|
|
+#include <linux/kernel.h>
|
|
+
|
|
+#define exit(E) return
|
|
+#define strtoul simple_strtoul
|
|
+#define strcoll strcmp
|
|
+
|
|
+#define CHAR_BIT 8
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/include/string.h
|
|
@@ -0,0 +1 @@
|
|
+#include <linux/string.h>
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lapi.c
|
|
@@ -0,0 +1,1086 @@
|
|
+/*
|
|
+** $Id: lapi.c,v 2.55.1.5 2008/07/04 18:41:18 roberto Exp $
|
|
+** Lua API
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stdarg.h>
|
|
+#include <math.h>
|
|
+#include <assert.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define lapi_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lapi.h"
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+#include "lundump.h"
|
|
+#include "lvm.h"
|
|
+
|
|
+
|
|
+
|
|
+const char lua_ident[] =
|
|
+ "$Lua: " LUA_RELEASE " " LUA_COPYRIGHT " $\n"
|
|
+ "$Authors: " LUA_AUTHORS " $\n"
|
|
+ "$URL: www.lua.org $\n";
|
|
+
|
|
+
|
|
+
|
|
+#define api_checknelems(L, n) api_check(L, (n) <= (L->top - L->base))
|
|
+
|
|
+#define api_checkvalidindex(L, i) api_check(L, (i) != luaO_nilobject)
|
|
+
|
|
+#define api_incr_top(L) {api_check(L, L->top < L->ci->top); L->top++;}
|
|
+
|
|
+
|
|
+
|
|
+static TValue *index2adr (lua_State *L, int idx) {
|
|
+ if (idx > 0) {
|
|
+ TValue *o = L->base + (idx - 1);
|
|
+ api_check(L, idx <= L->ci->top - L->base);
|
|
+ if (o >= L->top) return cast(TValue *, luaO_nilobject);
|
|
+ else return o;
|
|
+ }
|
|
+ else if (idx > LUA_REGISTRYINDEX) {
|
|
+ api_check(L, idx != 0 && -idx <= L->top - L->base);
|
|
+ return L->top + idx;
|
|
+ }
|
|
+ else switch (idx) { /* pseudo-indices */
|
|
+ case LUA_REGISTRYINDEX: return registry(L);
|
|
+ case LUA_ENVIRONINDEX: {
|
|
+ Closure *func = curr_func(L);
|
|
+ sethvalue(L, &L->env, func->c.env);
|
|
+ return &L->env;
|
|
+ }
|
|
+ case LUA_GLOBALSINDEX: return gt(L);
|
|
+ default: {
|
|
+ Closure *func = curr_func(L);
|
|
+ idx = LUA_GLOBALSINDEX - idx;
|
|
+ return (idx <= func->c.nupvalues)
|
|
+ ? &func->c.upvalue[idx-1]
|
|
+ : cast(TValue *, luaO_nilobject);
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static Table *getcurrenv (lua_State *L) {
|
|
+ if (L->ci == L->base_ci) /* no enclosing function? */
|
|
+ return hvalue(gt(L)); /* use global table as environment */
|
|
+ else {
|
|
+ Closure *func = curr_func(L);
|
|
+ return func->c.env;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaA_pushobject (lua_State *L, const TValue *o) {
|
|
+ setobj2s(L, L->top, o);
|
|
+ api_incr_top(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_checkstack (lua_State *L, int size) {
|
|
+ int res = 1;
|
|
+ lua_lock(L);
|
|
+ if (size > LUAI_MAXCSTACK || (L->top - L->base + size) > LUAI_MAXCSTACK)
|
|
+ res = 0; /* stack overflow */
|
|
+ else if (size > 0) {
|
|
+ luaD_checkstack(L, size);
|
|
+ if (L->ci->top < L->top + size)
|
|
+ L->ci->top = L->top + size;
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_xmove (lua_State *from, lua_State *to, int n) {
|
|
+ int i;
|
|
+ if (from == to) return;
|
|
+ lua_lock(to);
|
|
+ api_checknelems(from, n);
|
|
+ api_check(from, G(from) == G(to));
|
|
+ api_check(from, to->ci->top - to->top >= n);
|
|
+ from->top -= n;
|
|
+ for (i = 0; i < n; i++) {
|
|
+ setobj2s(to, to->top++, from->top + i);
|
|
+ }
|
|
+ lua_unlock(to);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_setlevel (lua_State *from, lua_State *to) {
|
|
+ to->nCcalls = from->nCcalls;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_CFunction lua_atpanic (lua_State *L, lua_CFunction panicf) {
|
|
+ lua_CFunction old;
|
|
+ lua_lock(L);
|
|
+ old = G(L)->panic;
|
|
+ G(L)->panic = panicf;
|
|
+ lua_unlock(L);
|
|
+ return old;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_State *lua_newthread (lua_State *L) {
|
|
+ lua_State *L1;
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ L1 = luaE_newthread(L);
|
|
+ setthvalue(L, L->top, L1);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+ luai_userstatethread(L, L1);
|
|
+ return L1;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** basic stack manipulation
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API int lua_gettop (lua_State *L) {
|
|
+ return cast_int(L->top - L->base);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_settop (lua_State *L, int idx) {
|
|
+ lua_lock(L);
|
|
+ if (idx >= 0) {
|
|
+ api_check(L, idx <= L->stack_last - L->base);
|
|
+ while (L->top < L->base + idx)
|
|
+ setnilvalue(L->top++);
|
|
+ L->top = L->base + idx;
|
|
+ }
|
|
+ else {
|
|
+ api_check(L, -(idx+1) <= (L->top - L->base));
|
|
+ L->top += idx+1; /* `subtract' index (index is negative) */
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_remove (lua_State *L, int idx) {
|
|
+ StkId p;
|
|
+ lua_lock(L);
|
|
+ p = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, p);
|
|
+ while (++p < L->top) setobjs2s(L, p-1, p);
|
|
+ L->top--;
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_insert (lua_State *L, int idx) {
|
|
+ StkId p;
|
|
+ StkId q;
|
|
+ lua_lock(L);
|
|
+ p = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, p);
|
|
+ for (q = L->top; q>p; q--) setobjs2s(L, q, q-1);
|
|
+ setobjs2s(L, p, L->top);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_replace (lua_State *L, int idx) {
|
|
+ StkId o;
|
|
+ lua_lock(L);
|
|
+ /* explicit test for incompatible code */
|
|
+ if (idx == LUA_ENVIRONINDEX && L->ci == L->base_ci)
|
|
+ luaG_runerror(L, "no calling environment");
|
|
+ api_checknelems(L, 1);
|
|
+ o = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, o);
|
|
+ if (idx == LUA_ENVIRONINDEX) {
|
|
+ Closure *func = curr_func(L);
|
|
+ api_check(L, ttistable(L->top - 1));
|
|
+ func->c.env = hvalue(L->top - 1);
|
|
+ luaC_barrier(L, func, L->top - 1);
|
|
+ }
|
|
+ else {
|
|
+ setobj(L, o, L->top - 1);
|
|
+ if (idx < LUA_GLOBALSINDEX) /* function upvalue? */
|
|
+ luaC_barrier(L, curr_func(L), L->top - 1);
|
|
+ }
|
|
+ L->top--;
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushvalue (lua_State *L, int idx) {
|
|
+ lua_lock(L);
|
|
+ setobj2s(L, L->top, index2adr(L, idx));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** access functions (stack -> C)
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API int lua_type (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ return (o == luaO_nilobject) ? LUA_TNONE : ttype(o);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_typename (lua_State *L, int t) {
|
|
+ UNUSED(L);
|
|
+ return (t == LUA_TNONE) ? "no value" : luaT_typenames[t];
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_iscfunction (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ return iscfunction(o);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_isnumber (lua_State *L, int idx) {
|
|
+ TValue n;
|
|
+ const TValue *o = index2adr(L, idx);
|
|
+ return tonumber(o, &n);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_isstring (lua_State *L, int idx) {
|
|
+ int t = lua_type(L, idx);
|
|
+ return (t == LUA_TSTRING || t == LUA_TNUMBER);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_isuserdata (lua_State *L, int idx) {
|
|
+ const TValue *o = index2adr(L, idx);
|
|
+ return (ttisuserdata(o) || ttislightuserdata(o));
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_rawequal (lua_State *L, int index1, int index2) {
|
|
+ StkId o1 = index2adr(L, index1);
|
|
+ StkId o2 = index2adr(L, index2);
|
|
+ return (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0
|
|
+ : luaO_rawequalObj(o1, o2);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_equal (lua_State *L, int index1, int index2) {
|
|
+ StkId o1, o2;
|
|
+ int i;
|
|
+ lua_lock(L); /* may call tag method */
|
|
+ o1 = index2adr(L, index1);
|
|
+ o2 = index2adr(L, index2);
|
|
+ i = (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0 : equalobj(L, o1, o2);
|
|
+ lua_unlock(L);
|
|
+ return i;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_lessthan (lua_State *L, int index1, int index2) {
|
|
+ StkId o1, o2;
|
|
+ int i;
|
|
+ lua_lock(L); /* may call tag method */
|
|
+ o1 = index2adr(L, index1);
|
|
+ o2 = index2adr(L, index2);
|
|
+ i = (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0
|
|
+ : luaV_lessthan(L, o1, o2);
|
|
+ lua_unlock(L);
|
|
+ return i;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+LUA_API lua_Number lua_tonumber (lua_State *L, int idx) {
|
|
+ TValue n;
|
|
+ const TValue *o = index2adr(L, idx);
|
|
+ if (tonumber(o, &n))
|
|
+ return nvalue(o);
|
|
+ else
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_Integer lua_tointeger (lua_State *L, int idx) {
|
|
+ TValue n;
|
|
+ const TValue *o = index2adr(L, idx);
|
|
+ if (tonumber(o, &n)) {
|
|
+ lua_Integer res;
|
|
+ lua_Number num = nvalue(o);
|
|
+ lua_number2integer(res, num);
|
|
+ return res;
|
|
+ }
|
|
+ else
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_toboolean (lua_State *L, int idx) {
|
|
+ const TValue *o = index2adr(L, idx);
|
|
+ return !l_isfalse(o);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_tolstring (lua_State *L, int idx, size_t *len) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ if (!ttisstring(o)) {
|
|
+ lua_lock(L); /* `luaV_tostring' may create a new string */
|
|
+ if (!luaV_tostring(L, o)) { /* conversion failed? */
|
|
+ if (len != NULL) *len = 0;
|
|
+ lua_unlock(L);
|
|
+ return NULL;
|
|
+ }
|
|
+ luaC_checkGC(L);
|
|
+ o = index2adr(L, idx); /* previous call may reallocate the stack */
|
|
+ lua_unlock(L);
|
|
+ }
|
|
+ if (len != NULL) *len = tsvalue(o)->len;
|
|
+ return svalue(o);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API size_t lua_objlen (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TSTRING: return tsvalue(o)->len;
|
|
+ case LUA_TUSERDATA: return uvalue(o)->len;
|
|
+ case LUA_TTABLE: return luaH_getn(hvalue(o));
|
|
+ case LUA_TNUMBER: {
|
|
+ size_t l;
|
|
+ lua_lock(L); /* `luaV_tostring' may create a new string */
|
|
+ l = (luaV_tostring(L, o) ? tsvalue(o)->len : 0);
|
|
+ lua_unlock(L);
|
|
+ return l;
|
|
+ }
|
|
+ default: return 0;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_CFunction lua_tocfunction (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ return (!iscfunction(o)) ? NULL : clvalue(o)->c.f;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void *lua_touserdata (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TUSERDATA: return (rawuvalue(o) + 1);
|
|
+ case LUA_TLIGHTUSERDATA: return pvalue(o);
|
|
+ default: return NULL;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_State *lua_tothread (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ return (!ttisthread(o)) ? NULL : thvalue(o);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const void *lua_topointer (lua_State *L, int idx) {
|
|
+ StkId o = index2adr(L, idx);
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TTABLE: return hvalue(o);
|
|
+ case LUA_TFUNCTION: return clvalue(o);
|
|
+ case LUA_TTHREAD: return thvalue(o);
|
|
+ case LUA_TUSERDATA:
|
|
+ case LUA_TLIGHTUSERDATA:
|
|
+ return lua_touserdata(L, idx);
|
|
+ default: return NULL;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** push functions (C -> stack)
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API void lua_pushnil (lua_State *L) {
|
|
+ lua_lock(L);
|
|
+ setnilvalue(L->top);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushnumber (lua_State *L, lua_Number n) {
|
|
+ lua_lock(L);
|
|
+ setnvalue(L->top, n);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushinteger (lua_State *L, lua_Integer n) {
|
|
+ lua_lock(L);
|
|
+ setnvalue(L->top, cast_num(n));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushlstring (lua_State *L, const char *s, size_t len) {
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ setsvalue2s(L, L->top, luaS_newlstr(L, s, len));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushstring (lua_State *L, const char *s) {
|
|
+ if (s == NULL)
|
|
+ lua_pushnil(L);
|
|
+ else
|
|
+ lua_pushlstring(L, s, strlen(s));
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_pushvfstring (lua_State *L, const char *fmt,
|
|
+ va_list argp) {
|
|
+ const char *ret;
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ ret = luaO_pushvfstring(L, fmt, argp);
|
|
+ lua_unlock(L);
|
|
+ return ret;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_pushfstring (lua_State *L, const char *fmt, ...) {
|
|
+ const char *ret;
|
|
+ va_list argp;
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ va_start(argp, fmt);
|
|
+ ret = luaO_pushvfstring(L, fmt, argp);
|
|
+ va_end(argp);
|
|
+ lua_unlock(L);
|
|
+ return ret;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushcclosure (lua_State *L, lua_CFunction fn, int n) {
|
|
+ Closure *cl;
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ api_checknelems(L, n);
|
|
+ cl = luaF_newCclosure(L, n, getcurrenv(L));
|
|
+ cl->c.f = fn;
|
|
+ L->top -= n;
|
|
+ while (n--)
|
|
+ setobj2n(L, &cl->c.upvalue[n], L->top+n);
|
|
+ setclvalue(L, L->top, cl);
|
|
+ lua_assert(iswhite(obj2gco(cl)));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushboolean (lua_State *L, int b) {
|
|
+ lua_lock(L);
|
|
+ setbvalue(L->top, (b != 0)); /* ensure that true is 1 */
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_pushlightuserdata (lua_State *L, void *p) {
|
|
+ lua_lock(L);
|
|
+ setpvalue(L->top, p);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_pushthread (lua_State *L) {
|
|
+ lua_lock(L);
|
|
+ setthvalue(L, L->top, L);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+ return (G(L)->mainthread == L);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** get functions (Lua -> stack)
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API void lua_gettable (lua_State *L, int idx) {
|
|
+ StkId t;
|
|
+ lua_lock(L);
|
|
+ t = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, t);
|
|
+ luaV_gettable(L, t, L->top - 1, L->top - 1);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_getfield (lua_State *L, int idx, const char *k) {
|
|
+ StkId t;
|
|
+ TValue key;
|
|
+ lua_lock(L);
|
|
+ t = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, t);
|
|
+ setsvalue(L, &key, luaS_new(L, k));
|
|
+ luaV_gettable(L, t, &key, L->top);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_rawget (lua_State *L, int idx) {
|
|
+ StkId t;
|
|
+ lua_lock(L);
|
|
+ t = index2adr(L, idx);
|
|
+ api_check(L, ttistable(t));
|
|
+ setobj2s(L, L->top - 1, luaH_get(hvalue(t), L->top - 1));
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_rawgeti (lua_State *L, int idx, int n) {
|
|
+ StkId o;
|
|
+ lua_lock(L);
|
|
+ o = index2adr(L, idx);
|
|
+ api_check(L, ttistable(o));
|
|
+ setobj2s(L, L->top, luaH_getnum(hvalue(o), n));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_createtable (lua_State *L, int narray, int nrec) {
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ sethvalue(L, L->top, luaH_new(L, narray, nrec));
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_getmetatable (lua_State *L, int objindex) {
|
|
+ const TValue *obj;
|
|
+ Table *mt = NULL;
|
|
+ int res;
|
|
+ lua_lock(L);
|
|
+ obj = index2adr(L, objindex);
|
|
+ switch (ttype(obj)) {
|
|
+ case LUA_TTABLE:
|
|
+ mt = hvalue(obj)->metatable;
|
|
+ break;
|
|
+ case LUA_TUSERDATA:
|
|
+ mt = uvalue(obj)->metatable;
|
|
+ break;
|
|
+ default:
|
|
+ mt = G(L)->mt[ttype(obj)];
|
|
+ break;
|
|
+ }
|
|
+ if (mt == NULL)
|
|
+ res = 0;
|
|
+ else {
|
|
+ sethvalue(L, L->top, mt);
|
|
+ api_incr_top(L);
|
|
+ res = 1;
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_getfenv (lua_State *L, int idx) {
|
|
+ StkId o;
|
|
+ lua_lock(L);
|
|
+ o = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, o);
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TFUNCTION:
|
|
+ sethvalue(L, L->top, clvalue(o)->c.env);
|
|
+ break;
|
|
+ case LUA_TUSERDATA:
|
|
+ sethvalue(L, L->top, uvalue(o)->env);
|
|
+ break;
|
|
+ case LUA_TTHREAD:
|
|
+ setobj2s(L, L->top, gt(thvalue(o)));
|
|
+ break;
|
|
+ default:
|
|
+ setnilvalue(L->top);
|
|
+ break;
|
|
+ }
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** set functions (stack -> Lua)
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API void lua_settable (lua_State *L, int idx) {
|
|
+ StkId t;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 2);
|
|
+ t = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, t);
|
|
+ luaV_settable(L, t, L->top - 2, L->top - 1);
|
|
+ L->top -= 2; /* pop index and value */
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_setfield (lua_State *L, int idx, const char *k) {
|
|
+ StkId t;
|
|
+ TValue key;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ t = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, t);
|
|
+ setsvalue(L, &key, luaS_new(L, k));
|
|
+ luaV_settable(L, t, &key, L->top - 1);
|
|
+ L->top--; /* pop value */
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_rawset (lua_State *L, int idx) {
|
|
+ StkId t;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 2);
|
|
+ t = index2adr(L, idx);
|
|
+ api_check(L, ttistable(t));
|
|
+ setobj2t(L, luaH_set(L, hvalue(t), L->top-2), L->top-1);
|
|
+ luaC_barriert(L, hvalue(t), L->top-1);
|
|
+ L->top -= 2;
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_rawseti (lua_State *L, int idx, int n) {
|
|
+ StkId o;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ o = index2adr(L, idx);
|
|
+ api_check(L, ttistable(o));
|
|
+ setobj2t(L, luaH_setnum(L, hvalue(o), n), L->top-1);
|
|
+ luaC_barriert(L, hvalue(o), L->top-1);
|
|
+ L->top--;
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_setmetatable (lua_State *L, int objindex) {
|
|
+ TValue *obj;
|
|
+ Table *mt;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ obj = index2adr(L, objindex);
|
|
+ api_checkvalidindex(L, obj);
|
|
+ if (ttisnil(L->top - 1))
|
|
+ mt = NULL;
|
|
+ else {
|
|
+ api_check(L, ttistable(L->top - 1));
|
|
+ mt = hvalue(L->top - 1);
|
|
+ }
|
|
+ switch (ttype(obj)) {
|
|
+ case LUA_TTABLE: {
|
|
+ hvalue(obj)->metatable = mt;
|
|
+ if (mt)
|
|
+ luaC_objbarriert(L, hvalue(obj), mt);
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TUSERDATA: {
|
|
+ uvalue(obj)->metatable = mt;
|
|
+ if (mt)
|
|
+ luaC_objbarrier(L, rawuvalue(obj), mt);
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ G(L)->mt[ttype(obj)] = mt;
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ L->top--;
|
|
+ lua_unlock(L);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_setfenv (lua_State *L, int idx) {
|
|
+ StkId o;
|
|
+ int res = 1;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ o = index2adr(L, idx);
|
|
+ api_checkvalidindex(L, o);
|
|
+ api_check(L, ttistable(L->top - 1));
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TFUNCTION:
|
|
+ clvalue(o)->c.env = hvalue(L->top - 1);
|
|
+ break;
|
|
+ case LUA_TUSERDATA:
|
|
+ uvalue(o)->env = hvalue(L->top - 1);
|
|
+ break;
|
|
+ case LUA_TTHREAD:
|
|
+ sethvalue(L, gt(thvalue(o)), hvalue(L->top - 1));
|
|
+ break;
|
|
+ default:
|
|
+ res = 0;
|
|
+ break;
|
|
+ }
|
|
+ if (res) luaC_objbarrier(L, gcvalue(o), hvalue(L->top - 1));
|
|
+ L->top--;
|
|
+ lua_unlock(L);
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** `load' and `call' functions (run Lua code)
|
|
+*/
|
|
+
|
|
+
|
|
+#define adjustresults(L,nres) \
|
|
+ { if (nres == LUA_MULTRET && L->top >= L->ci->top) L->ci->top = L->top; }
|
|
+
|
|
+
|
|
+#define checkresults(L,na,nr) \
|
|
+ api_check(L, (nr) == LUA_MULTRET || (L->ci->top - L->top >= (nr) - (na)))
|
|
+
|
|
+
|
|
+LUA_API void lua_call (lua_State *L, int nargs, int nresults) {
|
|
+ StkId func;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, nargs+1);
|
|
+ checkresults(L, nargs, nresults);
|
|
+ func = L->top - (nargs+1);
|
|
+ luaD_call(L, func, nresults);
|
|
+ adjustresults(L, nresults);
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Execute a protected call.
|
|
+*/
|
|
+struct CallS { /* data to `f_call' */
|
|
+ StkId func;
|
|
+ int nresults;
|
|
+};
|
|
+
|
|
+
|
|
+static void f_call (lua_State *L, void *ud) {
|
|
+ struct CallS *c = cast(struct CallS *, ud);
|
|
+ luaD_call(L, c->func, c->nresults);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+LUA_API int lua_pcall (lua_State *L, int nargs, int nresults, int errfunc) {
|
|
+ struct CallS c;
|
|
+ int status;
|
|
+ ptrdiff_t func;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, nargs+1);
|
|
+ checkresults(L, nargs, nresults);
|
|
+ if (errfunc == 0)
|
|
+ func = 0;
|
|
+ else {
|
|
+ StkId o = index2adr(L, errfunc);
|
|
+ api_checkvalidindex(L, o);
|
|
+ func = savestack(L, o);
|
|
+ }
|
|
+ c.func = L->top - (nargs+1); /* function to be called */
|
|
+ c.nresults = nresults;
|
|
+ status = luaD_pcall(L, f_call, &c, savestack(L, c.func), func);
|
|
+ adjustresults(L, nresults);
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Execute a protected C call.
|
|
+*/
|
|
+struct CCallS { /* data to `f_Ccall' */
|
|
+ lua_CFunction func;
|
|
+ void *ud;
|
|
+};
|
|
+
|
|
+
|
|
+static void f_Ccall (lua_State *L, void *ud) {
|
|
+ struct CCallS *c = cast(struct CCallS *, ud);
|
|
+ Closure *cl;
|
|
+ cl = luaF_newCclosure(L, 0, getcurrenv(L));
|
|
+ cl->c.f = c->func;
|
|
+ setclvalue(L, L->top, cl); /* push function */
|
|
+ api_incr_top(L);
|
|
+ setpvalue(L->top, c->ud); /* push only argument */
|
|
+ api_incr_top(L);
|
|
+ luaD_call(L, L->top - 2, 0);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_cpcall (lua_State *L, lua_CFunction func, void *ud) {
|
|
+ struct CCallS c;
|
|
+ int status;
|
|
+ lua_lock(L);
|
|
+ c.func = func;
|
|
+ c.ud = ud;
|
|
+ status = luaD_pcall(L, f_Ccall, &c, savestack(L, L->top), 0);
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_load (lua_State *L, lua_Reader reader, void *data,
|
|
+ const char *chunkname) {
|
|
+ ZIO z;
|
|
+ int status;
|
|
+ lua_lock(L);
|
|
+ if (!chunkname) chunkname = "?";
|
|
+ luaZ_init(L, &z, reader, data);
|
|
+ status = luaD_protectedparser(L, &z, chunkname);
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_dump (lua_State *L, lua_Writer writer, void *data) {
|
|
+ int status;
|
|
+ TValue *o;
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ o = L->top - 1;
|
|
+ if (isLfunction(o))
|
|
+ status = luaU_dump(L, clvalue(o)->l.p, writer, data, 0);
|
|
+ else
|
|
+ status = 1;
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_status (lua_State *L) {
|
|
+ return L->status;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Garbage-collection function
|
|
+*/
|
|
+
|
|
+LUA_API int lua_gc (lua_State *L, int what, int data) {
|
|
+ int res = 0;
|
|
+ global_State *g;
|
|
+ lua_lock(L);
|
|
+ g = G(L);
|
|
+ switch (what) {
|
|
+ case LUA_GCSTOP: {
|
|
+ g->GCthreshold = MAX_LUMEM;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCRESTART: {
|
|
+ g->GCthreshold = g->totalbytes;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCCOLLECT: {
|
|
+ luaC_fullgc(L);
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCCOUNT: {
|
|
+ /* GC values are expressed in Kbytes: #bytes/2^10 */
|
|
+ res = cast_int(g->totalbytes >> 10);
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCCOUNTB: {
|
|
+ res = cast_int(g->totalbytes & 0x3ff);
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCSTEP: {
|
|
+ lu_mem a = (cast(lu_mem, data) << 10);
|
|
+ if (a <= g->totalbytes)
|
|
+ g->GCthreshold = g->totalbytes - a;
|
|
+ else
|
|
+ g->GCthreshold = 0;
|
|
+ while (g->GCthreshold <= g->totalbytes) {
|
|
+ luaC_step(L);
|
|
+ if (g->gcstate == GCSpause) { /* end of cycle? */
|
|
+ res = 1; /* signal it */
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCSETPAUSE: {
|
|
+ res = g->gcpause;
|
|
+ g->gcpause = data;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_GCSETSTEPMUL: {
|
|
+ res = g->gcstepmul;
|
|
+ g->gcstepmul = data;
|
|
+ break;
|
|
+ }
|
|
+ default: res = -1; /* invalid option */
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** miscellaneous functions
|
|
+*/
|
|
+
|
|
+
|
|
+LUA_API int lua_error (lua_State *L) {
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, 1);
|
|
+ luaG_errormsg(L);
|
|
+ lua_unlock(L);
|
|
+ return 0; /* to avoid warnings */
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_next (lua_State *L, int idx) {
|
|
+ StkId t;
|
|
+ int more;
|
|
+ lua_lock(L);
|
|
+ t = index2adr(L, idx);
|
|
+ api_check(L, ttistable(t));
|
|
+ more = luaH_next(L, hvalue(t), L->top - 1);
|
|
+ if (more) {
|
|
+ api_incr_top(L);
|
|
+ }
|
|
+ else /* no more elements */
|
|
+ L->top -= 1; /* remove key */
|
|
+ lua_unlock(L);
|
|
+ return more;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_concat (lua_State *L, int n) {
|
|
+ lua_lock(L);
|
|
+ api_checknelems(L, n);
|
|
+ if (n >= 2) {
|
|
+ luaC_checkGC(L);
|
|
+ luaV_concat(L, n, cast_int(L->top - L->base) - 1);
|
|
+ L->top -= (n-1);
|
|
+ }
|
|
+ else if (n == 0) { /* push empty string */
|
|
+ setsvalue2s(L, L->top, luaS_newlstr(L, "", 0));
|
|
+ api_incr_top(L);
|
|
+ }
|
|
+ /* else n == 1; nothing to do */
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_Alloc lua_getallocf (lua_State *L, void **ud) {
|
|
+ lua_Alloc f;
|
|
+ lua_lock(L);
|
|
+ if (ud) *ud = G(L)->ud;
|
|
+ f = G(L)->frealloc;
|
|
+ lua_unlock(L);
|
|
+ return f;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_setallocf (lua_State *L, lua_Alloc f, void *ud) {
|
|
+ lua_lock(L);
|
|
+ G(L)->ud = ud;
|
|
+ G(L)->frealloc = f;
|
|
+ lua_unlock(L);
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void *lua_newuserdata (lua_State *L, size_t size) {
|
|
+ Udata *u;
|
|
+ lua_lock(L);
|
|
+ luaC_checkGC(L);
|
|
+ u = luaS_newudata(L, size, getcurrenv(L));
|
|
+ setuvalue(L, L->top, u);
|
|
+ api_incr_top(L);
|
|
+ lua_unlock(L);
|
|
+ return u + 1;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+static const char *aux_upvalue (StkId fi, int n, TValue **val) {
|
|
+ Closure *f;
|
|
+ if (!ttisfunction(fi)) return NULL;
|
|
+ f = clvalue(fi);
|
|
+ if (f->c.isC) {
|
|
+ if (!(1 <= n && n <= f->c.nupvalues)) return NULL;
|
|
+ *val = &f->c.upvalue[n-1];
|
|
+ return "";
|
|
+ }
|
|
+ else {
|
|
+ Proto *p = f->l.p;
|
|
+ if (!(1 <= n && n <= p->sizeupvalues)) return NULL;
|
|
+ *val = f->l.upvals[n-1]->v;
|
|
+ return getstr(p->upvalues[n-1]);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_getupvalue (lua_State *L, int funcindex, int n) {
|
|
+ const char *name;
|
|
+ TValue *val;
|
|
+ lua_lock(L);
|
|
+ name = aux_upvalue(index2adr(L, funcindex), n, &val);
|
|
+ if (name) {
|
|
+ setobj2s(L, L->top, val);
|
|
+ api_incr_top(L);
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+ return name;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_setupvalue (lua_State *L, int funcindex, int n) {
|
|
+ const char *name;
|
|
+ TValue *val;
|
|
+ StkId fi;
|
|
+ lua_lock(L);
|
|
+ fi = index2adr(L, funcindex);
|
|
+ api_checknelems(L, 1);
|
|
+ name = aux_upvalue(fi, n, &val);
|
|
+ if (name) {
|
|
+ L->top--;
|
|
+ setobj(L, val, L->top);
|
|
+ luaC_barrier(L, clvalue(fi), L->top);
|
|
+ }
|
|
+ lua_unlock(L);
|
|
+ return name;
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lapi.h
|
|
@@ -0,0 +1,16 @@
|
|
+/*
|
|
+** $Id: lapi.h,v 2.2.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Auxiliary functions from Lua API
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lapi_h
|
|
+#define lapi_h
|
|
+
|
|
+
|
|
+#include "lobject.h"
|
|
+
|
|
+
|
|
+LUAI_FUNC void luaA_pushobject (lua_State *L, const TValue *o);
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lauxlib.c
|
|
@@ -0,0 +1,674 @@
|
|
+/*
|
|
+** $Id: lauxlib.c,v 1.159.1.3 2008/01/21 13:20:51 roberto Exp $
|
|
+** Auxiliary functions for building Lua libraries
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stdarg.h>
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+#include <ctype.h>
|
|
+#include <errno.h>
|
|
+#include <stdio.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+#else
|
|
+#include <linux/ctype.h>
|
|
+#include <linux/errno.h>
|
|
+#include <linux/kernel.h>
|
|
+#include <linux/slab.h>
|
|
+#include <linux/string.h>
|
|
+#endif
|
|
+
|
|
+/* This file uses only the official API of Lua.
|
|
+** Any function declared here could be written as an application function.
|
|
+*/
|
|
+
|
|
+#define lauxlib_c
|
|
+#define LUA_LIB
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lauxlib.h"
|
|
+
|
|
+
|
|
+#define FREELIST_REF 0 /* free list of references */
|
|
+
|
|
+
|
|
+/* convert a stack index to positive */
|
|
+#define abs_index(L, i) ((i) > 0 || (i) <= LUA_REGISTRYINDEX ? (i) : \
|
|
+ lua_gettop(L) + (i) + 1)
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Error-report functions
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_argerror (lua_State *L, int narg, const char *extramsg) {
|
|
+ lua_Debug ar;
|
|
+ if (!lua_getstack(L, 0, &ar)) /* no stack frame? */
|
|
+ return luaL_error(L, "bad argument #%d (%s)", narg, extramsg);
|
|
+ lua_getinfo(L, "n", &ar);
|
|
+ if (strcmp(ar.namewhat, "method") == 0) {
|
|
+ narg--; /* do not count `self' */
|
|
+ if (narg == 0) /* error is in the self argument itself? */
|
|
+ return luaL_error(L, "calling " LUA_QS " on bad self (%s)",
|
|
+ ar.name, extramsg);
|
|
+ }
|
|
+ if (ar.name == NULL)
|
|
+ ar.name = "?";
|
|
+ return luaL_error(L, "bad argument #%d to " LUA_QS " (%s)",
|
|
+ narg, ar.name, extramsg);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_typerror (lua_State *L, int narg, const char *tname) {
|
|
+ const char *msg = lua_pushfstring(L, "%s expected, got %s",
|
|
+ tname, luaL_typename(L, narg));
|
|
+ return luaL_argerror(L, narg, msg);
|
|
+}
|
|
+
|
|
+
|
|
+static void tag_error (lua_State *L, int narg, int tag) {
|
|
+ luaL_typerror(L, narg, lua_typename(L, tag));
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_where (lua_State *L, int level) {
|
|
+ lua_Debug ar;
|
|
+ if (lua_getstack(L, level, &ar)) { /* check function at level */
|
|
+ lua_getinfo(L, "Sl", &ar); /* get info about it */
|
|
+ if (ar.currentline > 0) { /* is there info? */
|
|
+ lua_pushfstring(L, "%s:%d: ", ar.short_src, ar.currentline);
|
|
+ return;
|
|
+ }
|
|
+ }
|
|
+ lua_pushliteral(L, ""); /* else, no information available... */
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_error (lua_State *L, const char *fmt, ...) {
|
|
+ va_list argp;
|
|
+ va_start(argp, fmt);
|
|
+ luaL_where(L, 1);
|
|
+ lua_pushvfstring(L, fmt, argp);
|
|
+ va_end(argp);
|
|
+ lua_concat(L, 2);
|
|
+ return lua_error(L);
|
|
+}
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_checkoption (lua_State *L, int narg, const char *def,
|
|
+ const char *const lst[]) {
|
|
+ const char *name = (def) ? luaL_optstring(L, narg, def) :
|
|
+ luaL_checkstring(L, narg);
|
|
+ int i;
|
|
+ for (i=0; lst[i]; i++)
|
|
+ if (strcmp(lst[i], name) == 0)
|
|
+ return i;
|
|
+ return luaL_argerror(L, narg,
|
|
+ lua_pushfstring(L, "invalid option " LUA_QS, name));
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_newmetatable (lua_State *L, const char *tname) {
|
|
+ lua_getfield(L, LUA_REGISTRYINDEX, tname); /* get registry.name */
|
|
+ if (!lua_isnil(L, -1)) /* name already in use? */
|
|
+ return 0; /* leave previous value on top, but return 0 */
|
|
+ lua_pop(L, 1);
|
|
+ lua_newtable(L); /* create metatable */
|
|
+ lua_pushvalue(L, -1);
|
|
+ lua_setfield(L, LUA_REGISTRYINDEX, tname); /* registry.name = metatable */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void *luaL_checkudata (lua_State *L, int ud, const char *tname) {
|
|
+ void *p = lua_touserdata(L, ud);
|
|
+ if (p != NULL) { /* value is a userdata? */
|
|
+ if (lua_getmetatable(L, ud)) { /* does it have a metatable? */
|
|
+ lua_getfield(L, LUA_REGISTRYINDEX, tname); /* get correct metatable */
|
|
+ if (lua_rawequal(L, -1, -2)) { /* does it have the correct mt? */
|
|
+ lua_pop(L, 2); /* remove both metatables */
|
|
+ return p;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ luaL_typerror(L, ud, tname); /* else error */
|
|
+ return NULL; /* to avoid warnings */
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_checkstack (lua_State *L, int space, const char *mes) {
|
|
+ if (!lua_checkstack(L, space))
|
|
+ luaL_error(L, "stack overflow (%s)", mes);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_checktype (lua_State *L, int narg, int t) {
|
|
+ if (lua_type(L, narg) != t)
|
|
+ tag_error(L, narg, t);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_checkany (lua_State *L, int narg) {
|
|
+ if (lua_type(L, narg) == LUA_TNONE)
|
|
+ luaL_argerror(L, narg, "value expected");
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API const char *luaL_checklstring (lua_State *L, int narg, size_t *len) {
|
|
+ const char *s = lua_tolstring(L, narg, len);
|
|
+ if (!s) tag_error(L, narg, LUA_TSTRING);
|
|
+ return s;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API const char *luaL_optlstring (lua_State *L, int narg,
|
|
+ const char *def, size_t *len) {
|
|
+ if (lua_isnoneornil(L, narg)) {
|
|
+ if (len)
|
|
+ *len = (def ? strlen(def) : 0);
|
|
+ return def;
|
|
+ }
|
|
+ else return luaL_checklstring(L, narg, len);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API lua_Number luaL_checknumber (lua_State *L, int narg) {
|
|
+ lua_Number d = lua_tonumber(L, narg);
|
|
+ if (d == 0 && !lua_isnumber(L, narg)) /* avoid extra test when d is not 0 */
|
|
+ tag_error(L, narg, LUA_TNUMBER);
|
|
+ return d;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API lua_Number luaL_optnumber (lua_State *L, int narg, lua_Number def) {
|
|
+ return luaL_opt(L, luaL_checknumber, narg, def);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API lua_Integer luaL_checkinteger (lua_State *L, int narg) {
|
|
+ lua_Integer d = lua_tointeger(L, narg);
|
|
+ if (d == 0 && !lua_isnumber(L, narg)) /* avoid extra test when d is not 0 */
|
|
+ tag_error(L, narg, LUA_TNUMBER);
|
|
+ return d;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API lua_Integer luaL_optinteger (lua_State *L, int narg,
|
|
+ lua_Integer def) {
|
|
+ return luaL_opt(L, luaL_checkinteger, narg, def);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_getmetafield (lua_State *L, int obj, const char *event) {
|
|
+ if (!lua_getmetatable(L, obj)) /* no metatable? */
|
|
+ return 0;
|
|
+ lua_pushstring(L, event);
|
|
+ lua_rawget(L, -2);
|
|
+ if (lua_isnil(L, -1)) {
|
|
+ lua_pop(L, 2); /* remove metatable and metafield */
|
|
+ return 0;
|
|
+ }
|
|
+ else {
|
|
+ lua_remove(L, -2); /* remove only metatable */
|
|
+ return 1;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_callmeta (lua_State *L, int obj, const char *event) {
|
|
+ obj = abs_index(L, obj);
|
|
+ if (!luaL_getmetafield(L, obj, event)) /* no metafield? */
|
|
+ return 0;
|
|
+ lua_pushvalue(L, obj);
|
|
+ lua_call(L, 1, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void (luaL_register) (lua_State *L, const char *libname,
|
|
+ const luaL_Reg *l) {
|
|
+ luaI_openlib(L, libname, l, 0);
|
|
+}
|
|
+
|
|
+
|
|
+static int libsize (const luaL_Reg *l) {
|
|
+ int size = 0;
|
|
+ for (; l->name; l++) size++;
|
|
+ return size;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaI_openlib (lua_State *L, const char *libname,
|
|
+ const luaL_Reg *l, int nup) {
|
|
+ if (libname) {
|
|
+ int size = libsize(l);
|
|
+ /* check whether lib already exists */
|
|
+ luaL_findtable(L, LUA_REGISTRYINDEX, "_LOADED", 1);
|
|
+ lua_getfield(L, -1, libname); /* get _LOADED[libname] */
|
|
+ if (!lua_istable(L, -1)) { /* not found? */
|
|
+ lua_pop(L, 1); /* remove previous result */
|
|
+ /* try global variable (and create one if it does not exist) */
|
|
+ if (luaL_findtable(L, LUA_GLOBALSINDEX, libname, size) != NULL)
|
|
+ luaL_error(L, "name conflict for module " LUA_QS, libname);
|
|
+ lua_pushvalue(L, -1);
|
|
+ lua_setfield(L, -3, libname); /* _LOADED[libname] = new table */
|
|
+ }
|
|
+ lua_remove(L, -2); /* remove _LOADED table */
|
|
+ lua_insert(L, -(nup+1)); /* move library table to below upvalues */
|
|
+ }
|
|
+ for (; l->name; l++) {
|
|
+ int i;
|
|
+ for (i=0; i<nup; i++) /* copy upvalues to the top */
|
|
+ lua_pushvalue(L, -nup);
|
|
+ lua_pushcclosure(L, l->func, nup);
|
|
+ lua_setfield(L, -(nup+2), l->name);
|
|
+ }
|
|
+ lua_pop(L, nup); /* remove upvalues */
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** getn-setn: size for arrays
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+#if defined(LUA_COMPAT_GETN)
|
|
+
|
|
+static int checkint (lua_State *L, int topop) {
|
|
+ int n = (lua_type(L, -1) == LUA_TNUMBER) ? lua_tointeger(L, -1) : -1;
|
|
+ lua_pop(L, topop);
|
|
+ return n;
|
|
+}
|
|
+
|
|
+
|
|
+static void getsizes (lua_State *L) {
|
|
+ lua_getfield(L, LUA_REGISTRYINDEX, "LUA_SIZES");
|
|
+ if (lua_isnil(L, -1)) { /* no `size' table? */
|
|
+ lua_pop(L, 1); /* remove nil */
|
|
+ lua_newtable(L); /* create it */
|
|
+ lua_pushvalue(L, -1); /* `size' will be its own metatable */
|
|
+ lua_setmetatable(L, -2);
|
|
+ lua_pushliteral(L, "kv");
|
|
+ lua_setfield(L, -2, "__mode"); /* metatable(N).__mode = "kv" */
|
|
+ lua_pushvalue(L, -1);
|
|
+ lua_setfield(L, LUA_REGISTRYINDEX, "LUA_SIZES"); /* store in register */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_setn (lua_State *L, int t, int n) {
|
|
+ t = abs_index(L, t);
|
|
+ lua_pushliteral(L, "n");
|
|
+ lua_rawget(L, t);
|
|
+ if (checkint(L, 1) >= 0) { /* is there a numeric field `n'? */
|
|
+ lua_pushliteral(L, "n"); /* use it */
|
|
+ lua_pushinteger(L, n);
|
|
+ lua_rawset(L, t);
|
|
+ }
|
|
+ else { /* use `sizes' */
|
|
+ getsizes(L);
|
|
+ lua_pushvalue(L, t);
|
|
+ lua_pushinteger(L, n);
|
|
+ lua_rawset(L, -3); /* sizes[t] = n */
|
|
+ lua_pop(L, 1); /* remove `sizes' */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_getn (lua_State *L, int t) {
|
|
+ int n;
|
|
+ t = abs_index(L, t);
|
|
+ lua_pushliteral(L, "n"); /* try t.n */
|
|
+ lua_rawget(L, t);
|
|
+ if ((n = checkint(L, 1)) >= 0) return n;
|
|
+ getsizes(L); /* else try sizes[t] */
|
|
+ lua_pushvalue(L, t);
|
|
+ lua_rawget(L, -2);
|
|
+ if ((n = checkint(L, 2)) >= 0) return n;
|
|
+ return (int)lua_objlen(L, t);
|
|
+}
|
|
+
|
|
+#endif
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+
|
|
+LUALIB_API const char *luaL_gsub (lua_State *L, const char *s, const char *p,
|
|
+ const char *r) {
|
|
+ const char *wild;
|
|
+ size_t l = strlen(p);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "luaL_gsub: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ while ((wild = strstr(s, p)) != NULL) {
|
|
+ luaL_addlstring(b, s, wild - s); /* push prefix */
|
|
+ luaL_addstring(b, r); /* push replacement in place of pattern */
|
|
+ s = wild + l; /* continue after `p' */
|
|
+ }
|
|
+ luaL_addstring(b, s); /* push last suffix */
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return lua_tostring(L, -1);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API const char *luaL_findtable (lua_State *L, int idx,
|
|
+ const char *fname, int szhint) {
|
|
+ const char *e;
|
|
+ lua_pushvalue(L, idx);
|
|
+ do {
|
|
+ e = strchr(fname, '.');
|
|
+ if (e == NULL) e = fname + strlen(fname);
|
|
+ lua_pushlstring(L, fname, e - fname);
|
|
+ lua_rawget(L, -2);
|
|
+ if (lua_isnil(L, -1)) { /* no such field? */
|
|
+ lua_pop(L, 1); /* remove this nil */
|
|
+ lua_createtable(L, 0, (*e == '.' ? 1 : szhint)); /* new table for field */
|
|
+ lua_pushlstring(L, fname, e - fname);
|
|
+ lua_pushvalue(L, -2);
|
|
+ lua_settable(L, -4); /* set new table into field */
|
|
+ }
|
|
+ else if (!lua_istable(L, -1)) { /* field has a non-table value? */
|
|
+ lua_pop(L, 2); /* remove table and value */
|
|
+ return fname; /* return problematic part of the name */
|
|
+ }
|
|
+ lua_remove(L, -2); /* remove previous table */
|
|
+ fname = e + 1;
|
|
+ } while (*e == '.');
|
|
+ return NULL;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Generic Buffer manipulation
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+#define bufflen(B) ((B)->p - (B)->buffer)
|
|
+#define bufffree(B) ((size_t)(LUAL_BUFFERSIZE - bufflen(B)))
|
|
+
|
|
+#define LIMIT (LUA_MINSTACK/2)
|
|
+
|
|
+
|
|
+static int emptybuffer (luaL_Buffer *B) {
|
|
+ size_t l = bufflen(B);
|
|
+ if (l == 0) return 0; /* put nothing on stack */
|
|
+ else {
|
|
+ lua_pushlstring(B->L, B->buffer, l);
|
|
+ B->p = B->buffer;
|
|
+ B->lvl++;
|
|
+ return 1;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void adjuststack (luaL_Buffer *B) {
|
|
+ if (B->lvl > 1) {
|
|
+ lua_State *L = B->L;
|
|
+ int toget = 1; /* number of levels to concat */
|
|
+ size_t toplen = lua_strlen(L, -1);
|
|
+ do {
|
|
+ size_t l = lua_strlen(L, -(toget+1));
|
|
+ if (B->lvl - toget + 1 >= LIMIT || toplen > l) {
|
|
+ toplen += l;
|
|
+ toget++;
|
|
+ }
|
|
+ else break;
|
|
+ } while (toget < B->lvl);
|
|
+ lua_concat(L, toget);
|
|
+ B->lvl = B->lvl - toget + 1;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API char *luaL_prepbuffer (luaL_Buffer *B) {
|
|
+ if (emptybuffer(B))
|
|
+ adjuststack(B);
|
|
+ return B->buffer;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_addlstring (luaL_Buffer *B, const char *s, size_t l) {
|
|
+ while (l--)
|
|
+ luaL_addchar(B, *s++);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_addstring (luaL_Buffer *B, const char *s) {
|
|
+ luaL_addlstring(B, s, strlen(s));
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_pushresult (luaL_Buffer *B) {
|
|
+ emptybuffer(B);
|
|
+ lua_concat(B->L, B->lvl);
|
|
+ B->lvl = 1;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_addvalue (luaL_Buffer *B) {
|
|
+ lua_State *L = B->L;
|
|
+ size_t vl;
|
|
+ const char *s = lua_tolstring(L, -1, &vl);
|
|
+ if (vl <= bufffree(B)) { /* fit into buffer? */
|
|
+ memcpy(B->p, s, vl); /* put it there */
|
|
+ B->p += vl;
|
|
+ lua_pop(L, 1); /* remove from stack */
|
|
+ }
|
|
+ else {
|
|
+ if (emptybuffer(B))
|
|
+ lua_insert(L, -2); /* put buffer before new value */
|
|
+ B->lvl++; /* add new value into B stack */
|
|
+ adjuststack(B);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_buffinit (lua_State *L, luaL_Buffer *B) {
|
|
+ B->L = L;
|
|
+ B->p = B->buffer;
|
|
+ B->lvl = 0;
|
|
+}
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_ref (lua_State *L, int t) {
|
|
+ int ref;
|
|
+ t = abs_index(L, t);
|
|
+ if (lua_isnil(L, -1)) {
|
|
+ lua_pop(L, 1); /* remove from stack */
|
|
+ return LUA_REFNIL; /* `nil' has a unique fixed reference */
|
|
+ }
|
|
+ lua_rawgeti(L, t, FREELIST_REF); /* get first free element */
|
|
+ ref = (int)lua_tointeger(L, -1); /* ref = t[FREELIST_REF] */
|
|
+ lua_pop(L, 1); /* remove it from stack */
|
|
+ if (ref != 0) { /* any free element? */
|
|
+ lua_rawgeti(L, t, ref); /* remove it from list */
|
|
+ lua_rawseti(L, t, FREELIST_REF); /* (t[FREELIST_REF] = t[ref]) */
|
|
+ }
|
|
+ else { /* no free elements */
|
|
+ ref = (int)lua_objlen(L, t);
|
|
+ ref++; /* create new reference */
|
|
+ }
|
|
+ lua_rawseti(L, t, ref);
|
|
+ return ref;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API void luaL_unref (lua_State *L, int t, int ref) {
|
|
+ if (ref >= 0) {
|
|
+ t = abs_index(L, t);
|
|
+ lua_rawgeti(L, t, FREELIST_REF);
|
|
+ lua_rawseti(L, t, ref); /* t[ref] = t[FREELIST_REF] */
|
|
+ lua_pushinteger(L, ref);
|
|
+ lua_rawseti(L, t, FREELIST_REF); /* t[FREELIST_REF] = ref */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Load functions
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+typedef struct LoadF {
|
|
+ int extraline;
|
|
+ FILE *f;
|
|
+ char buff[LUAL_BUFFERSIZE];
|
|
+} LoadF;
|
|
+
|
|
+
|
|
+static const char *getF (lua_State *L, void *ud, size_t *size) {
|
|
+ LoadF *lf = (LoadF *)ud;
|
|
+ (void)L;
|
|
+ if (lf->extraline) {
|
|
+ lf->extraline = 0;
|
|
+ *size = 1;
|
|
+ return "\n";
|
|
+ }
|
|
+ if (feof(lf->f)) return NULL;
|
|
+ *size = fread(lf->buff, 1, sizeof(lf->buff), lf->f);
|
|
+ return (*size > 0) ? lf->buff : NULL;
|
|
+}
|
|
+
|
|
+
|
|
+static int errfile (lua_State *L, const char *what, int fnameindex) {
|
|
+ const char *serr = strerror(errno);
|
|
+ const char *filename = lua_tostring(L, fnameindex) + 1;
|
|
+ lua_pushfstring(L, "cannot %s %s: %s", what, filename, serr);
|
|
+ lua_remove(L, fnameindex);
|
|
+ return LUA_ERRFILE;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_loadfile (lua_State *L, const char *filename) {
|
|
+ LoadF lf;
|
|
+ int status, readstatus;
|
|
+ int c;
|
|
+ int fnameindex = lua_gettop(L) + 1; /* index of filename on the stack */
|
|
+ lf.extraline = 0;
|
|
+ if (filename == NULL) {
|
|
+ lua_pushliteral(L, "=stdin");
|
|
+ lf.f = stdin;
|
|
+ }
|
|
+ else {
|
|
+ lua_pushfstring(L, "@%s", filename);
|
|
+ lf.f = fopen(filename, "r");
|
|
+ if (lf.f == NULL) return errfile(L, "open", fnameindex);
|
|
+ }
|
|
+ c = getc(lf.f);
|
|
+ if (c == '#') { /* Unix exec. file? */
|
|
+ lf.extraline = 1;
|
|
+ while ((c = getc(lf.f)) != EOF && c != '\n') ; /* skip first line */
|
|
+ if (c == '\n') c = getc(lf.f);
|
|
+ }
|
|
+ if (c == LUA_SIGNATURE[0] && filename) { /* binary file? */
|
|
+ lf.f = freopen(filename, "rb", lf.f); /* reopen in binary mode */
|
|
+ if (lf.f == NULL) return errfile(L, "reopen", fnameindex);
|
|
+ /* skip eventual `#!...' */
|
|
+ while ((c = getc(lf.f)) != EOF && c != LUA_SIGNATURE[0]) ;
|
|
+ lf.extraline = 0;
|
|
+ }
|
|
+ ungetc(c, lf.f);
|
|
+ status = lua_load(L, getF, &lf, lua_tostring(L, -1));
|
|
+ readstatus = ferror(lf.f);
|
|
+ if (filename) fclose(lf.f); /* close file (even in case of errors) */
|
|
+ if (readstatus) {
|
|
+ lua_settop(L, fnameindex); /* ignore results from `lua_load' */
|
|
+ return errfile(L, "read", fnameindex);
|
|
+ }
|
|
+ lua_remove(L, fnameindex);
|
|
+ return status;
|
|
+}
|
|
+#endif
|
|
+
|
|
+typedef struct LoadS {
|
|
+ const char *s;
|
|
+ size_t size;
|
|
+} LoadS;
|
|
+
|
|
+
|
|
+static const char *getS (lua_State *L, void *ud, size_t *size) {
|
|
+ LoadS *ls = (LoadS *)ud;
|
|
+ (void)L;
|
|
+ if (ls->size == 0) return NULL;
|
|
+ *size = ls->size;
|
|
+ ls->size = 0;
|
|
+ return ls->s;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaL_loadbuffer (lua_State *L, const char *buff, size_t size,
|
|
+ const char *name) {
|
|
+ LoadS ls;
|
|
+ ls.s = buff;
|
|
+ ls.size = size;
|
|
+ return lua_load(L, getS, &ls, name);
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int (luaL_loadstring) (lua_State *L, const char *s) {
|
|
+ return luaL_loadbuffer(L, s, strlen(s), s);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+static void *l_alloc (void *ud, void *ptr, size_t osize, size_t nsize) {
|
|
+ (void)ud;
|
|
+ (void)osize;
|
|
+ if (nsize == 0) {
|
|
+#if !defined(__KERNEL__)
|
|
+ free(ptr);
|
|
+#else
|
|
+ kfree(ptr);
|
|
+#endif
|
|
+ return NULL;
|
|
+ }
|
|
+ else
|
|
+#if !defined(__KERNEL__)
|
|
+ return realloc(ptr, nsize);
|
|
+#else
|
|
+ return krealloc(ptr, nsize, GFP_ATOMIC);
|
|
+#endif
|
|
+}
|
|
+
|
|
+
|
|
+static int lpanic (lua_State *L) {
|
|
+ (void)L; /* to avoid warnings */
|
|
+#if !defined(__KERNEL__)
|
|
+ fprintf(stderr, "PANIC: unprotected error in call to Lua API (%s)\n",
|
|
+#else
|
|
+ printk( "PANIC: unprotected error in call to Lua API (%s)\n",
|
|
+#endif
|
|
+ lua_tostring(L, -1));
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API lua_State *luaL_newstate (void) {
|
|
+ lua_State *L = lua_newstate(l_alloc, NULL);
|
|
+ if (L) lua_atpanic(L, &lpanic);
|
|
+ return L;
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lauxlib.h
|
|
@@ -0,0 +1,184 @@
|
|
+/*
|
|
+** $Id: lauxlib.h,v 1.88.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Auxiliary functions for building Lua libraries
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lauxlib_h
|
|
+#define lauxlib_h
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+#include <linux/slab.h> /* for kmalloc and kfree when allocating luaL_Buffer */
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+#include <stdio.h>
|
|
+#endif
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+
|
|
+#if defined(LUA_COMPAT_GETN)
|
|
+LUALIB_API int (luaL_getn) (lua_State *L, int t);
|
|
+LUALIB_API void (luaL_setn) (lua_State *L, int t, int n);
|
|
+#else
|
|
+#define luaL_getn(L,i) ((int)lua_objlen(L, i))
|
|
+#define luaL_setn(L,i,j) ((void)0) /* no op! */
|
|
+#endif
|
|
+
|
|
+#if defined(LUA_COMPAT_OPENLIB)
|
|
+#define luaI_openlib luaL_openlib
|
|
+#endif
|
|
+
|
|
+
|
|
+/* extra error code for `luaL_load' */
|
|
+#define LUA_ERRFILE (LUA_ERRERR+1)
|
|
+
|
|
+
|
|
+typedef struct luaL_Reg {
|
|
+ const char *name;
|
|
+ lua_CFunction func;
|
|
+} luaL_Reg;
|
|
+
|
|
+
|
|
+
|
|
+LUALIB_API void (luaI_openlib) (lua_State *L, const char *libname,
|
|
+ const luaL_Reg *l, int nup);
|
|
+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_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_argerror) (lua_State *L, int numarg, const char *extramsg);
|
|
+LUALIB_API const char *(luaL_checklstring) (lua_State *L, int numArg,
|
|
+ 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_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_optinteger) (lua_State *L, int nArg,
|
|
+ lua_Integer def);
|
|
+
|
|
+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_checkany) (lua_State *L, int narg);
|
|
+
|
|
+LUALIB_API int (luaL_newmetatable) (lua_State *L, const char *tname);
|
|
+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 int (luaL_error) (lua_State *L, const char *fmt, ...);
|
|
+
|
|
+LUALIB_API int (luaL_checkoption) (lua_State *L, int narg, const char *def,
|
|
+ const char *const lst[]);
|
|
+
|
|
+LUALIB_API int (luaL_ref) (lua_State *L, int t);
|
|
+LUALIB_API void (luaL_unref) (lua_State *L, int t, int ref);
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+LUALIB_API int (luaL_loadfile) (lua_State *L, const char *filename);
|
|
+#endif
|
|
+
|
|
+LUALIB_API int (luaL_loadbuffer) (lua_State *L, const char *buff, size_t sz,
|
|
+ const char *name);
|
|
+LUALIB_API int (luaL_loadstring) (lua_State *L, const char *s);
|
|
+
|
|
+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_findtable) (lua_State *L, int idx,
|
|
+ const char *fname, int szhint);
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** ===============================================================
|
|
+** some useful macros
|
|
+** ===============================================================
|
|
+*/
|
|
+
|
|
+#define luaL_argcheck(L, cond,numarg,extramsg) \
|
|
+ ((void)((cond) || luaL_argerror(L, (numarg), (extramsg))))
|
|
+#define luaL_checkstring(L,n) (luaL_checklstring(L, (n), 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_optint(L,n,d) ((int)luaL_optinteger(L, (n), (d)))
|
|
+#define luaL_checklong(L,n) ((long)luaL_checkinteger(L, (n)))
|
|
+#define luaL_optlong(L,n,d) ((long)luaL_optinteger(L, (n), (d)))
|
|
+
|
|
+#define luaL_typename(L,i) lua_typename(L, lua_type(L,(i)))
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+#define luaL_dofile(L, fn) \
|
|
+ (luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
|
+#endif
|
|
+
|
|
+#define luaL_dostring(L, s) \
|
|
+ (luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
|
|
+
|
|
+#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)))
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Generic Buffer manipulation
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+
|
|
+typedef struct luaL_Buffer {
|
|
+ char *p; /* current position in buffer */
|
|
+ int lvl; /* number of strings in the stack (level) */
|
|
+ lua_State *L;
|
|
+ char buffer[LUAL_BUFFERSIZE];
|
|
+} luaL_Buffer;
|
|
+
|
|
+#define luaL_addchar(B,c) \
|
|
+ ((void)((B)->p < ((B)->buffer+LUAL_BUFFERSIZE) || luaL_prepbuffer(B)), \
|
|
+ (*(B)->p++ = (char)(c)))
|
|
+
|
|
+/* compatibility only */
|
|
+#define luaL_putchar(B,c) luaL_addchar(B,c)
|
|
+
|
|
+#define luaL_addsize(B,n) ((B)->p += (n))
|
|
+
|
|
+
|
|
+LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
|
|
+LUALIB_API char *(luaL_prepbuffer) (luaL_Buffer *B);
|
|
+LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
|
|
+LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
|
|
+LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
|
|
+LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
|
|
+
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+/* compatibility with ref system */
|
|
+
|
|
+/* pre-defined references */
|
|
+#define LUA_NOREF (-2)
|
|
+#define LUA_REFNIL (-1)
|
|
+
|
|
+#define lua_ref(L,lock) ((lock) ? luaL_ref(L, LUA_REGISTRYINDEX) : \
|
|
+ (lua_pushstring(L, "unlocked references are obsolete"), lua_error(L), 0))
|
|
+
|
|
+#define lua_unref(L,ref) luaL_unref(L, LUA_REGISTRYINDEX, (ref))
|
|
+
|
|
+#define lua_getref(L,ref) lua_rawgeti(L, LUA_REGISTRYINDEX, (ref))
|
|
+
|
|
+
|
|
+#define luaL_reg luaL_Reg
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lbaselib.c
|
|
@@ -0,0 +1,647 @@
|
|
+/*
|
|
+** $Id: lbaselib.c,v 1.191.1.6 2008/02/14 16:46:22 roberto Exp $
|
|
+** Basic library
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <linux/kernel.h>
|
|
+#include <linux/ctype.h>
|
|
+#include <linux/string.h>
|
|
+
|
|
+#define lbaselib_c
|
|
+#define LUA_LIB
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lauxlib.h"
|
|
+#include "lualib.h"
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** If your system does not support `stdout', you can just remove this function.
|
|
+** If you need, you can define your own `print' function, following this
|
|
+** model but changing `fputs' to put the strings at a proper place
|
|
+** (a console window or a log file, for instance).
|
|
+*/
|
|
+static int luaB_print (lua_State *L) {
|
|
+ int n = lua_gettop(L); /* number of arguments */
|
|
+ int i;
|
|
+ lua_getglobal(L, "tostring");
|
|
+ for (i=1; i<=n; i++) {
|
|
+ const char *s;
|
|
+ lua_pushvalue(L, -1); /* function to be called */
|
|
+ lua_pushvalue(L, i); /* value to print */
|
|
+ lua_call(L, 1, 1);
|
|
+ s = lua_tostring(L, -1); /* get result */
|
|
+ if (s == NULL)
|
|
+ return luaL_error(L, LUA_QL("tostring") " must return a string to "
|
|
+ LUA_QL("print"));
|
|
+ printk(KERN_INFO "LUA[print]: %s", s);
|
|
+ lua_pop(L, 1); /* pop result */
|
|
+ }
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_tonumber (lua_State *L) {
|
|
+ int base = luaL_optint(L, 2, 10);
|
|
+ if (base == 10) { /* standard conversion */
|
|
+ luaL_checkany(L, 1);
|
|
+ if (lua_isnumber(L, 1)) {
|
|
+ lua_pushnumber(L, lua_tonumber(L, 1));
|
|
+ return 1;
|
|
+ }
|
|
+ }
|
|
+ else {
|
|
+ const char *s1 = luaL_checkstring(L, 1);
|
|
+ char *s2;
|
|
+ unsigned long n;
|
|
+ luaL_argcheck(L, 2 <= base && base <= 36, 2, "base out of range");
|
|
+ n = simple_strtoul(s1, &s2, base);
|
|
+ if (s1 != s2) { /* at least one valid digit? */
|
|
+ while (isspace((unsigned char)(*s2))) s2++; /* skip trailing spaces */
|
|
+ if (*s2 == '\0') { /* no invalid trailing characters? */
|
|
+ lua_pushnumber(L, (lua_Number)n);
|
|
+ return 1;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ lua_pushnil(L); /* else not a number */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_error (lua_State *L) {
|
|
+ int level = luaL_optint(L, 2, 1);
|
|
+ lua_settop(L, 1);
|
|
+ if (lua_isstring(L, 1) && level > 0) { /* add extra information? */
|
|
+ luaL_where(L, level);
|
|
+ lua_pushvalue(L, 1);
|
|
+ lua_concat(L, 2);
|
|
+ }
|
|
+ return lua_error(L);
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_getmetatable (lua_State *L) {
|
|
+ luaL_checkany(L, 1);
|
|
+ if (!lua_getmetatable(L, 1)) {
|
|
+ lua_pushnil(L);
|
|
+ return 1; /* no metatable */
|
|
+ }
|
|
+ luaL_getmetafield(L, 1, "__metatable");
|
|
+ return 1; /* returns either __metatable field (if present) or metatable */
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_setmetatable (lua_State *L) {
|
|
+ int t = lua_type(L, 2);
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ luaL_argcheck(L, t == LUA_TNIL || t == LUA_TTABLE, 2,
|
|
+ "nil or table expected");
|
|
+ if (luaL_getmetafield(L, 1, "__metatable"))
|
|
+ luaL_error(L, "cannot change a protected metatable");
|
|
+ lua_settop(L, 2);
|
|
+ lua_setmetatable(L, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void getfunc (lua_State *L, int opt) {
|
|
+ if (lua_isfunction(L, 1)) lua_pushvalue(L, 1);
|
|
+ else {
|
|
+ lua_Debug ar;
|
|
+ int level = opt ? luaL_optint(L, 1, 1) : luaL_checkint(L, 1);
|
|
+ luaL_argcheck(L, level >= 0, 1, "level must be non-negative");
|
|
+ if (lua_getstack(L, level, &ar) == 0)
|
|
+ luaL_argerror(L, 1, "invalid level");
|
|
+ lua_getinfo(L, "f", &ar);
|
|
+ if (lua_isnil(L, -1))
|
|
+ luaL_error(L, "no function environment for tail call at level %d",
|
|
+ level);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_getfenv (lua_State *L) {
|
|
+ getfunc(L, 1);
|
|
+ if (lua_iscfunction(L, -1)) /* is a C function? */
|
|
+ lua_pushvalue(L, LUA_GLOBALSINDEX); /* return the thread's global env. */
|
|
+ else
|
|
+ lua_getfenv(L, -1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_setfenv (lua_State *L) {
|
|
+ luaL_checktype(L, 2, LUA_TTABLE);
|
|
+ getfunc(L, 0);
|
|
+ lua_pushvalue(L, 2);
|
|
+ if (lua_isnumber(L, 1) && lua_tonumber(L, 1) == 0) {
|
|
+ /* change environment of current thread */
|
|
+ lua_pushthread(L);
|
|
+ lua_insert(L, -2);
|
|
+ lua_setfenv(L, -2);
|
|
+ return 0;
|
|
+ }
|
|
+ else if (lua_iscfunction(L, -2) || lua_setfenv(L, -2) == 0)
|
|
+ luaL_error(L,
|
|
+ LUA_QL("setfenv") " cannot change environment of given object");
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_rawequal (lua_State *L) {
|
|
+ luaL_checkany(L, 1);
|
|
+ luaL_checkany(L, 2);
|
|
+ lua_pushboolean(L, lua_rawequal(L, 1, 2));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_rawget (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ luaL_checkany(L, 2);
|
|
+ lua_settop(L, 2);
|
|
+ lua_rawget(L, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int luaB_rawset (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ luaL_checkany(L, 2);
|
|
+ luaL_checkany(L, 3);
|
|
+ lua_settop(L, 3);
|
|
+ lua_rawset(L, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_gcinfo (lua_State *L) {
|
|
+ lua_pushinteger(L, lua_getgccount(L));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int luaB_collectgarbage (lua_State *L) {
|
|
+ static const char *const opts[] = {"stop", "restart", "collect",
|
|
+ "count", "step", "setpause", "setstepmul", NULL};
|
|
+ static const int optsnum[] = {LUA_GCSTOP, LUA_GCRESTART, LUA_GCCOLLECT,
|
|
+ LUA_GCCOUNT, LUA_GCSTEP, LUA_GCSETPAUSE, LUA_GCSETSTEPMUL};
|
|
+ int o = luaL_checkoption(L, 1, "collect", opts);
|
|
+ int ex = luaL_optint(L, 2, 0);
|
|
+ int res = lua_gc(L, optsnum[o], ex);
|
|
+ switch (optsnum[o]) {
|
|
+ case LUA_GCCOUNT: {
|
|
+ int b = lua_gc(L, LUA_GCCOUNTB, 0);
|
|
+ lua_pushnumber(L, res + ((lua_Number)b/1024));
|
|
+ return 1;
|
|
+ }
|
|
+ case LUA_GCSTEP: {
|
|
+ lua_pushboolean(L, res);
|
|
+ return 1;
|
|
+ }
|
|
+ default: {
|
|
+ lua_pushnumber(L, res);
|
|
+ return 1;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_type (lua_State *L) {
|
|
+ luaL_checkany(L, 1);
|
|
+ lua_pushstring(L, luaL_typename(L, 1));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_next (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ lua_settop(L, 2); /* create a 2nd argument if there isn't one */
|
|
+ if (lua_next(L, 1))
|
|
+ return 2;
|
|
+ else {
|
|
+ lua_pushnil(L);
|
|
+ return 1;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_pairs (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ lua_pushvalue(L, lua_upvalueindex(1)); /* return generator, */
|
|
+ lua_pushvalue(L, 1); /* state, */
|
|
+ lua_pushnil(L); /* and initial value */
|
|
+ return 3;
|
|
+}
|
|
+
|
|
+
|
|
+static int ipairsaux (lua_State *L) {
|
|
+ int i = luaL_checkint(L, 2);
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ i++; /* next value */
|
|
+ lua_pushinteger(L, i);
|
|
+ lua_rawgeti(L, 1, i);
|
|
+ return (lua_isnil(L, -1)) ? 0 : 2;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_ipairs (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ lua_pushvalue(L, lua_upvalueindex(1)); /* return generator, */
|
|
+ lua_pushvalue(L, 1); /* state, */
|
|
+ lua_pushinteger(L, 0); /* and initial value */
|
|
+ return 3;
|
|
+}
|
|
+
|
|
+
|
|
+static int load_aux (lua_State *L, int status) {
|
|
+ if (status == 0) /* OK? */
|
|
+ return 1;
|
|
+ else {
|
|
+ lua_pushnil(L);
|
|
+ lua_insert(L, -2); /* put before error message */
|
|
+ return 2; /* return nil plus error message */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_loadstring (lua_State *L) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ const char *chunkname = luaL_optstring(L, 2, s);
|
|
+ return load_aux(L, luaL_loadbuffer(L, s, l, chunkname));
|
|
+}
|
|
+
|
|
+/*
|
|
+static int luaB_loadfile (lua_State *L) {
|
|
+ const char *fname = luaL_optstring(L, 1, NULL);
|
|
+ return load_aux(L, luaL_loadfile(L, fname));
|
|
+}
|
|
+*/
|
|
+
|
|
+/*
|
|
+** Reader for generic `load' function: `lua_load' uses the
|
|
+** stack for internal stuff, so the reader cannot change the
|
|
+** stack top. Instead, it keeps its resulting string in a
|
|
+** reserved slot inside the stack.
|
|
+*/
|
|
+static const char *generic_reader (lua_State *L, void *ud, size_t *size) {
|
|
+ (void)ud; /* to avoid warnings */
|
|
+ luaL_checkstack(L, 2, "too many nested functions");
|
|
+ lua_pushvalue(L, 1); /* get function */
|
|
+ lua_call(L, 0, 1); /* call it */
|
|
+ if (lua_isnil(L, -1)) {
|
|
+ *size = 0;
|
|
+ return NULL;
|
|
+ }
|
|
+ else if (lua_isstring(L, -1)) {
|
|
+ lua_replace(L, 3); /* save string in a reserved stack slot */
|
|
+ return lua_tolstring(L, 3, size);
|
|
+ }
|
|
+ else luaL_error(L, "reader function must return a string");
|
|
+ return NULL; /* to avoid warnings */
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_load (lua_State *L) {
|
|
+ int status;
|
|
+ const char *cname = luaL_optstring(L, 2, "=(load)");
|
|
+ luaL_checktype(L, 1, LUA_TFUNCTION);
|
|
+ lua_settop(L, 3); /* function, eventual name, plus one reserved slot */
|
|
+ status = lua_load(L, generic_reader, NULL, cname);
|
|
+ return load_aux(L, status);
|
|
+}
|
|
+
|
|
+/*
|
|
+static int luaB_dofile (lua_State *L) {
|
|
+ const char *fname = luaL_optstring(L, 1, NULL);
|
|
+ int n = lua_gettop(L);
|
|
+ if (luaL_loadfile(L, fname) != 0) lua_error(L);
|
|
+ lua_call(L, 0, LUA_MULTRET);
|
|
+ return lua_gettop(L) - n;
|
|
+}
|
|
+*/
|
|
+
|
|
+static int luaB_assert (lua_State *L) {
|
|
+ luaL_checkany(L, 1);
|
|
+ if (!lua_toboolean(L, 1))
|
|
+ return luaL_error(L, "%s", luaL_optstring(L, 2, "assertion failed!"));
|
|
+ return lua_gettop(L);
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_unpack (lua_State *L) {
|
|
+ int i, e, n;
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ i = luaL_optint(L, 2, 1);
|
|
+ e = luaL_opt(L, luaL_checkint, 3, luaL_getn(L, 1));
|
|
+ if (i > e) return 0; /* empty range */
|
|
+ n = e - i + 1; /* number of elements */
|
|
+ if (n <= 0 || !lua_checkstack(L, n)) /* n <= 0 means arith. overflow */
|
|
+ return luaL_error(L, "too many results to unpack");
|
|
+ lua_rawgeti(L, 1, i); /* push arg[i] (avoiding overflow problems) */
|
|
+ while (i++ < e) /* push arg[i + 1...e] */
|
|
+ lua_rawgeti(L, 1, i);
|
|
+ return n;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_select (lua_State *L) {
|
|
+ int n = lua_gettop(L);
|
|
+ if (lua_type(L, 1) == LUA_TSTRING && *lua_tostring(L, 1) == '#') {
|
|
+ lua_pushinteger(L, n-1);
|
|
+ return 1;
|
|
+ }
|
|
+ else {
|
|
+ int i = luaL_checkint(L, 1);
|
|
+ if (i < 0) i = n + i;
|
|
+ else if (i > n) i = n;
|
|
+ luaL_argcheck(L, 1 <= i, 1, "index out of range");
|
|
+ return n - i;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_pcall (lua_State *L) {
|
|
+ int status;
|
|
+ luaL_checkany(L, 1);
|
|
+ status = lua_pcall(L, lua_gettop(L) - 1, LUA_MULTRET, 0);
|
|
+ lua_pushboolean(L, (status == 0));
|
|
+ lua_insert(L, 1);
|
|
+ return lua_gettop(L); /* return status + all results */
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_xpcall (lua_State *L) {
|
|
+ int status;
|
|
+ luaL_checkany(L, 2);
|
|
+ lua_settop(L, 2);
|
|
+ lua_insert(L, 1); /* put error function under function to be called */
|
|
+ status = lua_pcall(L, 0, LUA_MULTRET, 1);
|
|
+ lua_pushboolean(L, (status == 0));
|
|
+ lua_replace(L, 1);
|
|
+ return lua_gettop(L); /* return status + all results */
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_tostring (lua_State *L) {
|
|
+ luaL_checkany(L, 1);
|
|
+ if (luaL_callmeta(L, 1, "__tostring")) /* is there a metafield? */
|
|
+ return 1; /* use its value */
|
|
+ switch (lua_type(L, 1)) {
|
|
+ case LUA_TNUMBER:
|
|
+ lua_pushstring(L, lua_tostring(L, 1));
|
|
+ break;
|
|
+ case LUA_TSTRING:
|
|
+ lua_pushvalue(L, 1);
|
|
+ break;
|
|
+ case LUA_TBOOLEAN:
|
|
+ lua_pushstring(L, (lua_toboolean(L, 1) ? "true" : "false"));
|
|
+ break;
|
|
+ case LUA_TNIL:
|
|
+ lua_pushliteral(L, "nil");
|
|
+ break;
|
|
+ default:
|
|
+ lua_pushfstring(L, "%s: %p", luaL_typename(L, 1), lua_topointer(L, 1));
|
|
+ break;
|
|
+ }
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_newproxy (lua_State *L) {
|
|
+ lua_settop(L, 1);
|
|
+ lua_newuserdata(L, 0); /* create proxy */
|
|
+ if (lua_toboolean(L, 1) == 0)
|
|
+ return 1; /* no metatable */
|
|
+ else if (lua_isboolean(L, 1)) {
|
|
+ lua_newtable(L); /* create a new metatable `m' ... */
|
|
+ lua_pushvalue(L, -1); /* ... and mark `m' as a valid metatable */
|
|
+ lua_pushboolean(L, 1);
|
|
+ lua_rawset(L, lua_upvalueindex(1)); /* weaktable[m] = true */
|
|
+ }
|
|
+ else {
|
|
+ int validproxy = 0; /* to check if weaktable[metatable(u)] == true */
|
|
+ if (lua_getmetatable(L, 1)) {
|
|
+ lua_rawget(L, lua_upvalueindex(1));
|
|
+ validproxy = lua_toboolean(L, -1);
|
|
+ lua_pop(L, 1); /* remove value */
|
|
+ }
|
|
+ luaL_argcheck(L, validproxy, 1, "boolean or proxy expected");
|
|
+ lua_getmetatable(L, 1); /* metatable is valid; get it */
|
|
+ }
|
|
+ lua_setmetatable(L, 2);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static const luaL_Reg base_funcs[] = {
|
|
+ {"assert", luaB_assert},
|
|
+ {"collectgarbage", luaB_collectgarbage},
|
|
+// {"dofile", luaB_dofile},
|
|
+ {"error", luaB_error},
|
|
+ {"gcinfo", luaB_gcinfo},
|
|
+ {"getfenv", luaB_getfenv},
|
|
+ {"getmetatable", luaB_getmetatable},
|
|
+// {"loadfile", luaB_loadfile},
|
|
+ {"load", luaB_load},
|
|
+ {"loadstring", luaB_loadstring},
|
|
+ {"next", luaB_next},
|
|
+ {"pcall", luaB_pcall},
|
|
+ {"print", luaB_print},
|
|
+ {"rawequal", luaB_rawequal},
|
|
+ {"rawget", luaB_rawget},
|
|
+ {"rawset", luaB_rawset},
|
|
+ {"select", luaB_select},
|
|
+ {"setfenv", luaB_setfenv},
|
|
+ {"setmetatable", luaB_setmetatable},
|
|
+ {"tonumber", luaB_tonumber},
|
|
+ {"tostring", luaB_tostring},
|
|
+ {"type", luaB_type},
|
|
+ {"unpack", luaB_unpack},
|
|
+ {"xpcall", luaB_xpcall},
|
|
+ {NULL, NULL}
|
|
+};
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Coroutine library
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+#define CO_RUN 0 /* running */
|
|
+#define CO_SUS 1 /* suspended */
|
|
+#define CO_NOR 2 /* 'normal' (it resumed another coroutine) */
|
|
+#define CO_DEAD 3
|
|
+
|
|
+static const char *const statnames[] =
|
|
+ {"running", "suspended", "normal", "dead"};
|
|
+
|
|
+static int costatus (lua_State *L, lua_State *co) {
|
|
+ if (L == co) return CO_RUN;
|
|
+ switch (lua_status(co)) {
|
|
+ case LUA_YIELD:
|
|
+ return CO_SUS;
|
|
+ case 0: {
|
|
+ lua_Debug ar;
|
|
+ if (lua_getstack(co, 0, &ar) > 0) /* does it have frames? */
|
|
+ return CO_NOR; /* it is running */
|
|
+ else if (lua_gettop(co) == 0)
|
|
+ return CO_DEAD;
|
|
+ else
|
|
+ return CO_SUS; /* initial state */
|
|
+ }
|
|
+ default: /* some error occured */
|
|
+ return CO_DEAD;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_costatus (lua_State *L) {
|
|
+ lua_State *co = lua_tothread(L, 1);
|
|
+ luaL_argcheck(L, co, 1, "coroutine expected");
|
|
+ lua_pushstring(L, statnames[costatus(L, co)]);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int auxresume (lua_State *L, lua_State *co, int narg) {
|
|
+ int status = costatus(L, co);
|
|
+ if (!lua_checkstack(co, narg))
|
|
+ luaL_error(L, "too many arguments to resume");
|
|
+ if (status != CO_SUS) {
|
|
+ lua_pushfstring(L, "cannot resume %s coroutine", statnames[status]);
|
|
+ return -1; /* error flag */
|
|
+ }
|
|
+ lua_xmove(L, co, narg);
|
|
+ lua_setlevel(L, co);
|
|
+ status = lua_resume(co, narg);
|
|
+ if (status == 0 || status == LUA_YIELD) {
|
|
+ int nres = lua_gettop(co);
|
|
+ if (!lua_checkstack(L, nres + 1))
|
|
+ luaL_error(L, "too many results to resume");
|
|
+ lua_xmove(co, L, nres); /* move yielded values */
|
|
+ return nres;
|
|
+ }
|
|
+ else {
|
|
+ lua_xmove(co, L, 1); /* move error message */
|
|
+ return -1; /* error flag */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_coresume (lua_State *L) {
|
|
+ lua_State *co = lua_tothread(L, 1);
|
|
+ int r;
|
|
+ luaL_argcheck(L, co, 1, "coroutine expected");
|
|
+ r = auxresume(L, co, lua_gettop(L) - 1);
|
|
+ if (r < 0) {
|
|
+ lua_pushboolean(L, 0);
|
|
+ lua_insert(L, -2);
|
|
+ return 2; /* return false + error message */
|
|
+ }
|
|
+ else {
|
|
+ lua_pushboolean(L, 1);
|
|
+ lua_insert(L, -(r + 1));
|
|
+ return r + 1; /* return true + `resume' returns */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_auxwrap (lua_State *L) {
|
|
+ lua_State *co = lua_tothread(L, lua_upvalueindex(1));
|
|
+ int r = auxresume(L, co, lua_gettop(L));
|
|
+ if (r < 0) {
|
|
+ if (lua_isstring(L, -1)) { /* error object is a string? */
|
|
+ luaL_where(L, 1); /* add extra info */
|
|
+ lua_insert(L, -2);
|
|
+ lua_concat(L, 2);
|
|
+ }
|
|
+ lua_error(L); /* propagate error */
|
|
+ }
|
|
+ return r;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_cocreate (lua_State *L) {
|
|
+ lua_State *NL = lua_newthread(L);
|
|
+ luaL_argcheck(L, lua_isfunction(L, 1) && !lua_iscfunction(L, 1), 1,
|
|
+ "Lua function expected");
|
|
+ lua_pushvalue(L, 1); /* move function to top */
|
|
+ lua_xmove(L, NL, 1); /* move function from L to NL */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_cowrap (lua_State *L) {
|
|
+ luaB_cocreate(L);
|
|
+ lua_pushcclosure(L, luaB_auxwrap, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_yield (lua_State *L) {
|
|
+ return lua_yield(L, lua_gettop(L));
|
|
+}
|
|
+
|
|
+
|
|
+static int luaB_corunning (lua_State *L) {
|
|
+ if (lua_pushthread(L))
|
|
+ lua_pushnil(L); /* main thread is not a coroutine */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static const luaL_Reg co_funcs[] = {
|
|
+ {"create", luaB_cocreate},
|
|
+ {"resume", luaB_coresume},
|
|
+ {"running", luaB_corunning},
|
|
+ {"status", luaB_costatus},
|
|
+ {"wrap", luaB_cowrap},
|
|
+ {"yield", luaB_yield},
|
|
+ {NULL, NULL}
|
|
+};
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+static void auxopen (lua_State *L, const char *name,
|
|
+ lua_CFunction f, lua_CFunction u) {
|
|
+ lua_pushcfunction(L, u);
|
|
+ lua_pushcclosure(L, f, 1);
|
|
+ lua_setfield(L, -2, name);
|
|
+}
|
|
+
|
|
+
|
|
+static void base_open (lua_State *L) {
|
|
+ /* set global _G */
|
|
+ lua_pushvalue(L, LUA_GLOBALSINDEX);
|
|
+ lua_setglobal(L, "_G");
|
|
+ /* open lib into global table */
|
|
+ luaL_register(L, "_G", base_funcs);
|
|
+ lua_pushliteral(L, LUA_VERSION);
|
|
+ lua_setglobal(L, "_VERSION"); /* set global _VERSION */
|
|
+ /* `ipairs' and `pairs' need auxliliary functions as upvalues */
|
|
+ auxopen(L, "ipairs", luaB_ipairs, ipairsaux);
|
|
+ auxopen(L, "pairs", luaB_pairs, luaB_next);
|
|
+ /* `newproxy' needs a weaktable as upvalue */
|
|
+ lua_createtable(L, 0, 1); /* new table `w' */
|
|
+ lua_pushvalue(L, -1); /* `w' will be its own metatable */
|
|
+ lua_setmetatable(L, -2);
|
|
+ lua_pushliteral(L, "kv");
|
|
+ lua_setfield(L, -2, "__mode"); /* metatable(w).__mode = "kv" */
|
|
+ lua_pushcclosure(L, luaB_newproxy, 1);
|
|
+ lua_setglobal(L, "newproxy"); /* set global `newproxy' */
|
|
+}
|
|
+
|
|
+
|
|
+LUALIB_API int luaopen_base (lua_State *L) {
|
|
+ base_open(L);
|
|
+ luaL_register(L, LUA_COLIBNAME, co_funcs);
|
|
+ return 2;
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lcode.c
|
|
@@ -0,0 +1,838 @@
|
|
+/*
|
|
+** $Id: lcode.c,v 2.25.1.3 2007/12/28 15:32:23 roberto Exp $
|
|
+** Code generator for Lua
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stdlib.h>
|
|
+
|
|
+#define lcode_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lcode.h"
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lgc.h"
|
|
+#include "llex.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lparser.h"
|
|
+#include "ltable.h"
|
|
+
|
|
+
|
|
+#define hasjumps(e) ((e)->t != (e)->f)
|
|
+
|
|
+
|
|
+static int isnumeral(expdesc *e) {
|
|
+ return (e->k == VKNUM && e->t == NO_JUMP && e->f == NO_JUMP);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_nil (FuncState *fs, int from, int n) {
|
|
+ Instruction *previous;
|
|
+ if (fs->pc > fs->lasttarget) { /* no jumps to current position? */
|
|
+ if (fs->pc == 0) { /* function start? */
|
|
+ if (from >= fs->nactvar)
|
|
+ return; /* positions are already clean */
|
|
+ }
|
|
+ else {
|
|
+ previous = &fs->f->code[fs->pc-1];
|
|
+ if (GET_OPCODE(*previous) == OP_LOADNIL) {
|
|
+ int pfrom = GETARG_A(*previous);
|
|
+ int pto = GETARG_B(*previous);
|
|
+ if (pfrom <= from && from <= pto+1) { /* can connect both? */
|
|
+ if (from+n-1 > pto)
|
|
+ SETARG_B(*previous, from+n-1);
|
|
+ return;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ luaK_codeABC(fs, OP_LOADNIL, from, from+n-1, 0); /* else no optimization */
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_jump (FuncState *fs) {
|
|
+ int jpc = fs->jpc; /* save list of jumps to here */
|
|
+ int j;
|
|
+ fs->jpc = NO_JUMP;
|
|
+ j = luaK_codeAsBx(fs, OP_JMP, 0, NO_JUMP);
|
|
+ luaK_concat(fs, &j, jpc); /* keep them on hold */
|
|
+ return j;
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_ret (FuncState *fs, int first, int nret) {
|
|
+ luaK_codeABC(fs, OP_RETURN, first, nret+1, 0);
|
|
+}
|
|
+
|
|
+
|
|
+static int condjump (FuncState *fs, OpCode op, int A, int B, int C) {
|
|
+ luaK_codeABC(fs, op, A, B, C);
|
|
+ return luaK_jump(fs);
|
|
+}
|
|
+
|
|
+
|
|
+static void fixjump (FuncState *fs, int pc, int dest) {
|
|
+ Instruction *jmp = &fs->f->code[pc];
|
|
+ int offset = dest-(pc+1);
|
|
+ lua_assert(dest != NO_JUMP);
|
|
+ if (abs(offset) > MAXARG_sBx)
|
|
+ luaX_syntaxerror(fs->ls, "control structure too long");
|
|
+ SETARG_sBx(*jmp, offset);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** returns current `pc' and marks it as a jump target (to avoid wrong
|
|
+** optimizations with consecutive instructions not in the same basic block).
|
|
+*/
|
|
+int luaK_getlabel (FuncState *fs) {
|
|
+ fs->lasttarget = fs->pc;
|
|
+ return fs->pc;
|
|
+}
|
|
+
|
|
+
|
|
+static int getjump (FuncState *fs, int pc) {
|
|
+ int offset = GETARG_sBx(fs->f->code[pc]);
|
|
+ if (offset == NO_JUMP) /* point to itself represents end of list */
|
|
+ return NO_JUMP; /* end of list */
|
|
+ else
|
|
+ return (pc+1)+offset; /* turn offset into absolute position */
|
|
+}
|
|
+
|
|
+
|
|
+static Instruction *getjumpcontrol (FuncState *fs, int pc) {
|
|
+ Instruction *pi = &fs->f->code[pc];
|
|
+ if (pc >= 1 && testTMode(GET_OPCODE(*(pi-1))))
|
|
+ return pi-1;
|
|
+ else
|
|
+ return pi;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** check whether list has any jump that do not produce a value
|
|
+** (or produce an inverted value)
|
|
+*/
|
|
+static int need_value (FuncState *fs, int list) {
|
|
+ for (; list != NO_JUMP; list = getjump(fs, list)) {
|
|
+ Instruction i = *getjumpcontrol(fs, list);
|
|
+ if (GET_OPCODE(i) != OP_TESTSET) return 1;
|
|
+ }
|
|
+ return 0; /* not found */
|
|
+}
|
|
+
|
|
+
|
|
+static int patchtestreg (FuncState *fs, int node, int reg) {
|
|
+ Instruction *i = getjumpcontrol(fs, node);
|
|
+ if (GET_OPCODE(*i) != OP_TESTSET)
|
|
+ return 0; /* cannot patch other instructions */
|
|
+ if (reg != NO_REG && reg != GETARG_B(*i))
|
|
+ SETARG_A(*i, reg);
|
|
+ else /* no register to put value or register already has the value */
|
|
+ *i = CREATE_ABC(OP_TEST, GETARG_B(*i), 0, GETARG_C(*i));
|
|
+
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void removevalues (FuncState *fs, int list) {
|
|
+ for (; list != NO_JUMP; list = getjump(fs, list))
|
|
+ patchtestreg(fs, list, NO_REG);
|
|
+}
|
|
+
|
|
+
|
|
+static void patchlistaux (FuncState *fs, int list, int vtarget, int reg,
|
|
+ int dtarget) {
|
|
+ while (list != NO_JUMP) {
|
|
+ int next = getjump(fs, list);
|
|
+ if (patchtestreg(fs, list, reg))
|
|
+ fixjump(fs, list, vtarget);
|
|
+ else
|
|
+ fixjump(fs, list, dtarget); /* jump to default target */
|
|
+ list = next;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void dischargejpc (FuncState *fs) {
|
|
+ patchlistaux(fs, fs->jpc, fs->pc, NO_REG, fs->pc);
|
|
+ fs->jpc = NO_JUMP;
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_patchlist (FuncState *fs, int list, int target) {
|
|
+ if (target == fs->pc)
|
|
+ luaK_patchtohere(fs, list);
|
|
+ else {
|
|
+ lua_assert(target < fs->pc);
|
|
+ patchlistaux(fs, list, target, NO_REG, target);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_patchtohere (FuncState *fs, int list) {
|
|
+ luaK_getlabel(fs);
|
|
+ luaK_concat(fs, &fs->jpc, list);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_concat (FuncState *fs, int *l1, int l2) {
|
|
+ if (l2 == NO_JUMP) return;
|
|
+ else if (*l1 == NO_JUMP)
|
|
+ *l1 = l2;
|
|
+ else {
|
|
+ int list = *l1;
|
|
+ int next;
|
|
+ while ((next = getjump(fs, list)) != NO_JUMP) /* find last element */
|
|
+ list = next;
|
|
+ fixjump(fs, list, l2);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_checkstack (FuncState *fs, int n) {
|
|
+ int newstack = fs->freereg + n;
|
|
+ if (newstack > fs->f->maxstacksize) {
|
|
+ if (newstack >= MAXSTACK)
|
|
+ luaX_syntaxerror(fs->ls, "function or expression too complex");
|
|
+ fs->f->maxstacksize = cast_byte(newstack);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_reserveregs (FuncState *fs, int n) {
|
|
+ luaK_checkstack(fs, n);
|
|
+ fs->freereg += n;
|
|
+}
|
|
+
|
|
+
|
|
+static void freereg (FuncState *fs, int reg) {
|
|
+ if (!ISK(reg) && reg >= fs->nactvar) {
|
|
+ fs->freereg--;
|
|
+ lua_assert(reg == fs->freereg);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void freeexp (FuncState *fs, expdesc *e) {
|
|
+ if (e->k == VNONRELOC)
|
|
+ freereg(fs, e->u.s.info);
|
|
+}
|
|
+
|
|
+
|
|
+static int addk (FuncState *fs, TValue *k, TValue *v) {
|
|
+ lua_State *L = fs->L;
|
|
+ TValue *idx = luaH_set(L, fs->h, k);
|
|
+ Proto *f = fs->f;
|
|
+ int oldsize = f->sizek;
|
|
+ if (ttisnumber(idx)) {
|
|
+ lua_assert(luaO_rawequalObj(&fs->f->k[cast_int(nvalue(idx))], v));
|
|
+ return cast_int(nvalue(idx));
|
|
+ }
|
|
+ else { /* constant not found; create a new entry */
|
|
+ setnvalue(idx, cast_num(fs->nk));
|
|
+ luaM_growvector(L, f->k, fs->nk, f->sizek, TValue,
|
|
+ MAXARG_Bx, "constant table overflow");
|
|
+ while (oldsize < f->sizek) setnilvalue(&f->k[oldsize++]);
|
|
+ setobj(L, &f->k[fs->nk], v);
|
|
+ luaC_barrier(L, f, v);
|
|
+ return fs->nk++;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_stringK (FuncState *fs, TString *s) {
|
|
+ TValue o;
|
|
+ setsvalue(fs->L, &o, s);
|
|
+ return addk(fs, &o, &o);
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_numberK (FuncState *fs, lua_Number r) {
|
|
+ TValue o;
|
|
+ setnvalue(&o, r);
|
|
+ return addk(fs, &o, &o);
|
|
+}
|
|
+
|
|
+
|
|
+static int boolK (FuncState *fs, int b) {
|
|
+ TValue o;
|
|
+ setbvalue(&o, b);
|
|
+ return addk(fs, &o, &o);
|
|
+}
|
|
+
|
|
+
|
|
+static int nilK (FuncState *fs) {
|
|
+ TValue k, v;
|
|
+ setnilvalue(&v);
|
|
+ /* cannot use nil as key; instead use table itself to represent nil */
|
|
+ sethvalue(fs->L, &k, fs->h);
|
|
+ return addk(fs, &k, &v);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_setreturns (FuncState *fs, expdesc *e, int nresults) {
|
|
+ if (e->k == VCALL) { /* expression is an open function call? */
|
|
+ SETARG_C(getcode(fs, e), nresults+1);
|
|
+ }
|
|
+ else if (e->k == VVARARG) {
|
|
+ SETARG_B(getcode(fs, e), nresults+1);
|
|
+ SETARG_A(getcode(fs, e), fs->freereg);
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_setoneret (FuncState *fs, expdesc *e) {
|
|
+ if (e->k == VCALL) { /* expression is an open function call? */
|
|
+ e->k = VNONRELOC;
|
|
+ e->u.s.info = GETARG_A(getcode(fs, e));
|
|
+ }
|
|
+ else if (e->k == VVARARG) {
|
|
+ SETARG_B(getcode(fs, e), 2);
|
|
+ e->k = VRELOCABLE; /* can relocate its simple result */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_dischargevars (FuncState *fs, expdesc *e) {
|
|
+ switch (e->k) {
|
|
+ case VLOCAL: {
|
|
+ e->k = VNONRELOC;
|
|
+ break;
|
|
+ }
|
|
+ case VUPVAL: {
|
|
+ e->u.s.info = luaK_codeABC(fs, OP_GETUPVAL, 0, e->u.s.info, 0);
|
|
+ e->k = VRELOCABLE;
|
|
+ break;
|
|
+ }
|
|
+ case VGLOBAL: {
|
|
+ e->u.s.info = luaK_codeABx(fs, OP_GETGLOBAL, 0, e->u.s.info);
|
|
+ e->k = VRELOCABLE;
|
|
+ break;
|
|
+ }
|
|
+ case VINDEXED: {
|
|
+ freereg(fs, e->u.s.aux);
|
|
+ freereg(fs, e->u.s.info);
|
|
+ e->u.s.info = luaK_codeABC(fs, OP_GETTABLE, 0, e->u.s.info, e->u.s.aux);
|
|
+ e->k = VRELOCABLE;
|
|
+ break;
|
|
+ }
|
|
+ case VVARARG:
|
|
+ case VCALL: {
|
|
+ luaK_setoneret(fs, e);
|
|
+ break;
|
|
+ }
|
|
+ default: break; /* there is one value available (somewhere) */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int code_label (FuncState *fs, int A, int b, int jump) {
|
|
+ luaK_getlabel(fs); /* those instructions may be jump targets */
|
|
+ return luaK_codeABC(fs, OP_LOADBOOL, A, b, jump);
|
|
+}
|
|
+
|
|
+
|
|
+static void discharge2reg (FuncState *fs, expdesc *e, int reg) {
|
|
+ luaK_dischargevars(fs, e);
|
|
+ switch (e->k) {
|
|
+ case VNIL: {
|
|
+ luaK_nil(fs, reg, 1);
|
|
+ break;
|
|
+ }
|
|
+ case VFALSE: case VTRUE: {
|
|
+ luaK_codeABC(fs, OP_LOADBOOL, reg, e->k == VTRUE, 0);
|
|
+ break;
|
|
+ }
|
|
+ case VK: {
|
|
+ luaK_codeABx(fs, OP_LOADK, reg, e->u.s.info);
|
|
+ break;
|
|
+ }
|
|
+ case VKNUM: {
|
|
+ luaK_codeABx(fs, OP_LOADK, reg, luaK_numberK(fs, e->u.nval));
|
|
+ break;
|
|
+ }
|
|
+ case VRELOCABLE: {
|
|
+ Instruction *pc = &getcode(fs, e);
|
|
+ SETARG_A(*pc, reg);
|
|
+ break;
|
|
+ }
|
|
+ case VNONRELOC: {
|
|
+ if (reg != e->u.s.info)
|
|
+ luaK_codeABC(fs, OP_MOVE, reg, e->u.s.info, 0);
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ lua_assert(e->k == VVOID || e->k == VJMP);
|
|
+ return; /* nothing to do... */
|
|
+ }
|
|
+ }
|
|
+ e->u.s.info = reg;
|
|
+ e->k = VNONRELOC;
|
|
+}
|
|
+
|
|
+
|
|
+static void discharge2anyreg (FuncState *fs, expdesc *e) {
|
|
+ if (e->k != VNONRELOC) {
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ discharge2reg(fs, e, fs->freereg-1);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void exp2reg (FuncState *fs, expdesc *e, int reg) {
|
|
+ discharge2reg(fs, e, reg);
|
|
+ if (e->k == VJMP)
|
|
+ luaK_concat(fs, &e->t, e->u.s.info); /* put this jump in `t' list */
|
|
+ if (hasjumps(e)) {
|
|
+ int final; /* position after whole expression */
|
|
+ int p_f = NO_JUMP; /* position of an eventual LOAD false */
|
|
+ int p_t = NO_JUMP; /* position of an eventual LOAD true */
|
|
+ if (need_value(fs, e->t) || need_value(fs, e->f)) {
|
|
+ int fj = (e->k == VJMP) ? NO_JUMP : luaK_jump(fs);
|
|
+ p_f = code_label(fs, reg, 0, 1);
|
|
+ p_t = code_label(fs, reg, 1, 0);
|
|
+ luaK_patchtohere(fs, fj);
|
|
+ }
|
|
+ final = luaK_getlabel(fs);
|
|
+ patchlistaux(fs, e->f, final, reg, p_f);
|
|
+ patchlistaux(fs, e->t, final, reg, p_t);
|
|
+ }
|
|
+ e->f = e->t = NO_JUMP;
|
|
+ e->u.s.info = reg;
|
|
+ e->k = VNONRELOC;
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_exp2nextreg (FuncState *fs, expdesc *e) {
|
|
+ luaK_dischargevars(fs, e);
|
|
+ freeexp(fs, e);
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ exp2reg(fs, e, fs->freereg - 1);
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_exp2anyreg (FuncState *fs, expdesc *e) {
|
|
+ luaK_dischargevars(fs, e);
|
|
+ if (e->k == VNONRELOC) {
|
|
+ if (!hasjumps(e)) return e->u.s.info; /* exp is already in a register */
|
|
+ if (e->u.s.info >= fs->nactvar) { /* reg. is not a local? */
|
|
+ exp2reg(fs, e, e->u.s.info); /* put value on it */
|
|
+ return e->u.s.info;
|
|
+ }
|
|
+ }
|
|
+ luaK_exp2nextreg(fs, e); /* default */
|
|
+ return e->u.s.info;
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_exp2val (FuncState *fs, expdesc *e) {
|
|
+ if (hasjumps(e))
|
|
+ luaK_exp2anyreg(fs, e);
|
|
+ else
|
|
+ luaK_dischargevars(fs, e);
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_exp2RK (FuncState *fs, expdesc *e) {
|
|
+ luaK_exp2val(fs, e);
|
|
+ switch (e->k) {
|
|
+ case VKNUM:
|
|
+ case VTRUE:
|
|
+ case VFALSE:
|
|
+ case VNIL: {
|
|
+ if (fs->nk <= MAXINDEXRK) { /* constant fit in RK operand? */
|
|
+ e->u.s.info = (e->k == VNIL) ? nilK(fs) :
|
|
+ (e->k == VKNUM) ? luaK_numberK(fs, e->u.nval) :
|
|
+ boolK(fs, (e->k == VTRUE));
|
|
+ e->k = VK;
|
|
+ return RKASK(e->u.s.info);
|
|
+ }
|
|
+ else break;
|
|
+ }
|
|
+ case VK: {
|
|
+ if (e->u.s.info <= MAXINDEXRK) /* constant fit in argC? */
|
|
+ return RKASK(e->u.s.info);
|
|
+ else break;
|
|
+ }
|
|
+ default: break;
|
|
+ }
|
|
+ /* not a constant in the right range: put it in a register */
|
|
+ return luaK_exp2anyreg(fs, e);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_storevar (FuncState *fs, expdesc *var, expdesc *ex) {
|
|
+ switch (var->k) {
|
|
+ case VLOCAL: {
|
|
+ freeexp(fs, ex);
|
|
+ exp2reg(fs, ex, var->u.s.info);
|
|
+ return;
|
|
+ }
|
|
+ case VUPVAL: {
|
|
+ int e = luaK_exp2anyreg(fs, ex);
|
|
+ luaK_codeABC(fs, OP_SETUPVAL, e, var->u.s.info, 0);
|
|
+ break;
|
|
+ }
|
|
+ case VGLOBAL: {
|
|
+ int e = luaK_exp2anyreg(fs, ex);
|
|
+ luaK_codeABx(fs, OP_SETGLOBAL, e, var->u.s.info);
|
|
+ break;
|
|
+ }
|
|
+ case VINDEXED: {
|
|
+ int e = luaK_exp2RK(fs, ex);
|
|
+ luaK_codeABC(fs, OP_SETTABLE, var->u.s.info, var->u.s.aux, e);
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ lua_assert(0); /* invalid var kind to store */
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ freeexp(fs, ex);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_self (FuncState *fs, expdesc *e, expdesc *key) {
|
|
+ int func;
|
|
+ luaK_exp2anyreg(fs, e);
|
|
+ freeexp(fs, e);
|
|
+ func = fs->freereg;
|
|
+ luaK_reserveregs(fs, 2);
|
|
+ luaK_codeABC(fs, OP_SELF, func, e->u.s.info, luaK_exp2RK(fs, key));
|
|
+ freeexp(fs, key);
|
|
+ e->u.s.info = func;
|
|
+ e->k = VNONRELOC;
|
|
+}
|
|
+
|
|
+
|
|
+static void invertjump (FuncState *fs, expdesc *e) {
|
|
+ Instruction *pc = getjumpcontrol(fs, e->u.s.info);
|
|
+ lua_assert(testTMode(GET_OPCODE(*pc)) && GET_OPCODE(*pc) != OP_TESTSET &&
|
|
+ GET_OPCODE(*pc) != OP_TEST);
|
|
+ SETARG_A(*pc, !(GETARG_A(*pc)));
|
|
+}
|
|
+
|
|
+
|
|
+static int jumponcond (FuncState *fs, expdesc *e, int cond) {
|
|
+ if (e->k == VRELOCABLE) {
|
|
+ Instruction ie = getcode(fs, e);
|
|
+ if (GET_OPCODE(ie) == OP_NOT) {
|
|
+ fs->pc--; /* remove previous OP_NOT */
|
|
+ return condjump(fs, OP_TEST, GETARG_B(ie), 0, !cond);
|
|
+ }
|
|
+ /* else go through */
|
|
+ }
|
|
+ discharge2anyreg(fs, e);
|
|
+ freeexp(fs, e);
|
|
+ return condjump(fs, OP_TESTSET, NO_REG, e->u.s.info, cond);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_goiftrue (FuncState *fs, expdesc *e) {
|
|
+ int pc; /* pc of last jump */
|
|
+ luaK_dischargevars(fs, e);
|
|
+ switch (e->k) {
|
|
+ case VK: case VKNUM: case VTRUE: {
|
|
+ pc = NO_JUMP; /* always true; do nothing */
|
|
+ break;
|
|
+ }
|
|
+ case VFALSE: {
|
|
+ pc = luaK_jump(fs); /* always jump */
|
|
+ break;
|
|
+ }
|
|
+ case VJMP: {
|
|
+ invertjump(fs, e);
|
|
+ pc = e->u.s.info;
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ pc = jumponcond(fs, e, 0);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ luaK_concat(fs, &e->f, pc); /* insert last jump in `f' list */
|
|
+ luaK_patchtohere(fs, e->t);
|
|
+ e->t = NO_JUMP;
|
|
+}
|
|
+
|
|
+
|
|
+static void luaK_goiffalse (FuncState *fs, expdesc *e) {
|
|
+ int pc; /* pc of last jump */
|
|
+ luaK_dischargevars(fs, e);
|
|
+ switch (e->k) {
|
|
+ case VNIL: case VFALSE: {
|
|
+ pc = NO_JUMP; /* always false; do nothing */
|
|
+ break;
|
|
+ }
|
|
+ case VTRUE: {
|
|
+ pc = luaK_jump(fs); /* always jump */
|
|
+ break;
|
|
+ }
|
|
+ case VJMP: {
|
|
+ pc = e->u.s.info;
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ pc = jumponcond(fs, e, 1);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ luaK_concat(fs, &e->t, pc); /* insert last jump in `t' list */
|
|
+ luaK_patchtohere(fs, e->f);
|
|
+ e->f = NO_JUMP;
|
|
+}
|
|
+
|
|
+
|
|
+static void codenot (FuncState *fs, expdesc *e) {
|
|
+ luaK_dischargevars(fs, e);
|
|
+ switch (e->k) {
|
|
+ case VNIL: case VFALSE: {
|
|
+ e->k = VTRUE;
|
|
+ break;
|
|
+ }
|
|
+ case VK: case VKNUM: case VTRUE: {
|
|
+ e->k = VFALSE;
|
|
+ break;
|
|
+ }
|
|
+ case VJMP: {
|
|
+ invertjump(fs, e);
|
|
+ break;
|
|
+ }
|
|
+ case VRELOCABLE:
|
|
+ case VNONRELOC: {
|
|
+ discharge2anyreg(fs, e);
|
|
+ freeexp(fs, e);
|
|
+ e->u.s.info = luaK_codeABC(fs, OP_NOT, 0, e->u.s.info, 0);
|
|
+ e->k = VRELOCABLE;
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ lua_assert(0); /* cannot happen */
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ /* interchange true and false lists */
|
|
+ { int temp = e->f; e->f = e->t; e->t = temp; }
|
|
+ removevalues(fs, e->f);
|
|
+ removevalues(fs, e->t);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_indexed (FuncState *fs, expdesc *t, expdesc *k) {
|
|
+ t->u.s.aux = luaK_exp2RK(fs, k);
|
|
+ t->k = VINDEXED;
|
|
+}
|
|
+
|
|
+
|
|
+static int constfolding (OpCode op, expdesc *e1, expdesc *e2) {
|
|
+ lua_Number v1, v2, r;
|
|
+ if (!isnumeral(e1) || !isnumeral(e2)) return 0;
|
|
+ v1 = e1->u.nval;
|
|
+ v2 = e2->u.nval;
|
|
+ switch (op) {
|
|
+ case OP_ADD: r = luai_numadd(v1, v2); break;
|
|
+ case OP_SUB: r = luai_numsub(v1, v2); break;
|
|
+ case OP_MUL: r = luai_nummul(v1, v2); break;
|
|
+ case OP_DIV:
|
|
+ if (v2 == 0) return 0; /* do not attempt to divide by 0 */
|
|
+ r = luai_numdiv(v1, v2); break;
|
|
+ case OP_MOD:
|
|
+ if (v2 == 0) return 0; /* do not attempt to divide by 0 */
|
|
+ r = luai_nummod(v1, v2); break;
|
|
+ case OP_POW: r = luai_numpow(v1, v2); break;
|
|
+ case OP_UNM: r = luai_numunm(v1); break;
|
|
+ case OP_LEN: return 0; /* no constant folding for 'len' */
|
|
+ default: lua_assert(0); r = 0; break;
|
|
+ }
|
|
+ if (luai_numisnan(r)) return 0; /* do not attempt to produce NaN */
|
|
+ e1->u.nval = r;
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void codearith (FuncState *fs, OpCode op, expdesc *e1, expdesc *e2) {
|
|
+ if (constfolding(op, e1, e2))
|
|
+ return;
|
|
+ else {
|
|
+ int o2 = (op != OP_UNM && op != OP_LEN) ? luaK_exp2RK(fs, e2) : 0;
|
|
+ int o1 = luaK_exp2RK(fs, e1);
|
|
+ if (o1 > o2) {
|
|
+ freeexp(fs, e1);
|
|
+ freeexp(fs, e2);
|
|
+ }
|
|
+ else {
|
|
+ freeexp(fs, e2);
|
|
+ freeexp(fs, e1);
|
|
+ }
|
|
+ e1->u.s.info = luaK_codeABC(fs, op, 0, o1, o2);
|
|
+ e1->k = VRELOCABLE;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void codecomp (FuncState *fs, OpCode op, int cond, expdesc *e1,
|
|
+ expdesc *e2) {
|
|
+ int o1 = luaK_exp2RK(fs, e1);
|
|
+ int o2 = luaK_exp2RK(fs, e2);
|
|
+ freeexp(fs, e2);
|
|
+ freeexp(fs, e1);
|
|
+ if (cond == 0 && op != OP_EQ) {
|
|
+ int temp; /* exchange args to replace by `<' or `<=' */
|
|
+ temp = o1; o1 = o2; o2 = temp; /* o1 <==> o2 */
|
|
+ cond = 1;
|
|
+ }
|
|
+ e1->u.s.info = condjump(fs, op, cond, o1, o2);
|
|
+ e1->k = VJMP;
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_prefix (FuncState *fs, UnOpr op, expdesc *e) {
|
|
+ expdesc e2;
|
|
+ e2.t = e2.f = NO_JUMP; e2.k = VKNUM; e2.u.nval = 0;
|
|
+ switch (op) {
|
|
+ case OPR_MINUS: {
|
|
+ if (!isnumeral(e))
|
|
+ luaK_exp2anyreg(fs, e); /* cannot operate on non-numeric constants */
|
|
+ codearith(fs, OP_UNM, e, &e2);
|
|
+ break;
|
|
+ }
|
|
+ case OPR_NOT: codenot(fs, e); break;
|
|
+ case OPR_LEN: {
|
|
+ luaK_exp2anyreg(fs, e); /* cannot operate on constants */
|
|
+ codearith(fs, OP_LEN, e, &e2);
|
|
+ break;
|
|
+ }
|
|
+ default: lua_assert(0);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_infix (FuncState *fs, BinOpr op, expdesc *v) {
|
|
+ switch (op) {
|
|
+ case OPR_AND: {
|
|
+ luaK_goiftrue(fs, v);
|
|
+ break;
|
|
+ }
|
|
+ case OPR_OR: {
|
|
+ luaK_goiffalse(fs, v);
|
|
+ break;
|
|
+ }
|
|
+ case OPR_CONCAT: {
|
|
+ luaK_exp2nextreg(fs, v); /* operand must be on the `stack' */
|
|
+ break;
|
|
+ }
|
|
+ case OPR_ADD: case OPR_SUB: case OPR_MUL: case OPR_DIV:
|
|
+ case OPR_MOD: case OPR_POW: {
|
|
+ if (!isnumeral(v)) luaK_exp2RK(fs, v);
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ luaK_exp2RK(fs, v);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_posfix (FuncState *fs, BinOpr op, expdesc *e1, expdesc *e2) {
|
|
+ switch (op) {
|
|
+ case OPR_AND: {
|
|
+ lua_assert(e1->t == NO_JUMP); /* list must be closed */
|
|
+ luaK_dischargevars(fs, e2);
|
|
+ luaK_concat(fs, &e2->f, e1->f);
|
|
+ *e1 = *e2;
|
|
+ break;
|
|
+ }
|
|
+ case OPR_OR: {
|
|
+ lua_assert(e1->f == NO_JUMP); /* list must be closed */
|
|
+ luaK_dischargevars(fs, e2);
|
|
+ luaK_concat(fs, &e2->t, e1->t);
|
|
+ *e1 = *e2;
|
|
+ break;
|
|
+ }
|
|
+ case OPR_CONCAT: {
|
|
+ luaK_exp2val(fs, e2);
|
|
+ if (e2->k == VRELOCABLE && GET_OPCODE(getcode(fs, e2)) == OP_CONCAT) {
|
|
+ lua_assert(e1->u.s.info == GETARG_B(getcode(fs, e2))-1);
|
|
+ freeexp(fs, e1);
|
|
+ SETARG_B(getcode(fs, e2), e1->u.s.info);
|
|
+ e1->k = VRELOCABLE; e1->u.s.info = e2->u.s.info;
|
|
+ }
|
|
+ else {
|
|
+ luaK_exp2nextreg(fs, e2); /* operand must be on the 'stack' */
|
|
+ codearith(fs, OP_CONCAT, e1, e2);
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case OPR_ADD: codearith(fs, OP_ADD, e1, e2); break;
|
|
+ case OPR_SUB: codearith(fs, OP_SUB, e1, e2); break;
|
|
+ case OPR_MUL: codearith(fs, OP_MUL, e1, e2); break;
|
|
+ case OPR_DIV: codearith(fs, OP_DIV, e1, e2); break;
|
|
+ case OPR_MOD: codearith(fs, OP_MOD, e1, e2); break;
|
|
+ case OPR_POW: codearith(fs, OP_POW, e1, e2); break;
|
|
+ case OPR_EQ: codecomp(fs, OP_EQ, 1, e1, e2); break;
|
|
+ case OPR_NE: codecomp(fs, OP_EQ, 0, e1, e2); break;
|
|
+ case OPR_LT: codecomp(fs, OP_LT, 1, e1, e2); break;
|
|
+ case OPR_LE: codecomp(fs, OP_LE, 1, e1, e2); break;
|
|
+ case OPR_GT: codecomp(fs, OP_LT, 0, e1, e2); break;
|
|
+ case OPR_GE: codecomp(fs, OP_LE, 0, e1, e2); break;
|
|
+ default: lua_assert(0);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_fixline (FuncState *fs, int line) {
|
|
+ fs->f->lineinfo[fs->pc - 1] = line;
|
|
+}
|
|
+
|
|
+
|
|
+static int luaK_code (FuncState *fs, Instruction i, int line) {
|
|
+ Proto *f = fs->f;
|
|
+ dischargejpc(fs); /* `pc' will change */
|
|
+ /* put new instruction in code array */
|
|
+ luaM_growvector(fs->L, f->code, fs->pc, f->sizecode, Instruction,
|
|
+ MAX_INT, "code size overflow");
|
|
+ f->code[fs->pc] = i;
|
|
+ /* save corresponding line information */
|
|
+ luaM_growvector(fs->L, f->lineinfo, fs->pc, f->sizelineinfo, int,
|
|
+ MAX_INT, "code size overflow");
|
|
+ f->lineinfo[fs->pc] = line;
|
|
+ return fs->pc++;
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_codeABC (FuncState *fs, OpCode o, int a, int b, int c) {
|
|
+ lua_assert(getOpMode(o) == iABC);
|
|
+ lua_assert(getBMode(o) != OpArgN || b == 0);
|
|
+ lua_assert(getCMode(o) != OpArgN || c == 0);
|
|
+ return luaK_code(fs, CREATE_ABC(o, a, b, c), fs->ls->lastline);
|
|
+}
|
|
+
|
|
+
|
|
+int luaK_codeABx (FuncState *fs, OpCode o, int a, unsigned int bc) {
|
|
+ lua_assert(getOpMode(o) == iABx || getOpMode(o) == iAsBx);
|
|
+ lua_assert(getCMode(o) == OpArgN);
|
|
+ return luaK_code(fs, CREATE_ABx(o, a, bc), fs->ls->lastline);
|
|
+}
|
|
+
|
|
+
|
|
+void luaK_setlist (FuncState *fs, int base, int nelems, int tostore) {
|
|
+ int c = (nelems - 1)/LFIELDS_PER_FLUSH + 1;
|
|
+ int b = (tostore == LUA_MULTRET) ? 0 : tostore;
|
|
+ lua_assert(tostore != 0);
|
|
+ if (c <= MAXARG_C)
|
|
+ luaK_codeABC(fs, OP_SETLIST, base, b, c);
|
|
+ else {
|
|
+ luaK_codeABC(fs, OP_SETLIST, base, b, 0);
|
|
+ luaK_code(fs, cast(Instruction, c), fs->ls->lastline);
|
|
+ }
|
|
+ fs->freereg = base + 1; /* free registers with list values */
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lcode.h
|
|
@@ -0,0 +1,76 @@
|
|
+/*
|
|
+** $Id: lcode.h,v 1.48.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Code generator for Lua
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lcode_h
|
|
+#define lcode_h
|
|
+
|
|
+#include "llex.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lparser.h"
|
|
+
|
|
+
|
|
+/*
|
|
+** Marks the end of a patch list. It is an invalid value both as an absolute
|
|
+** address, and as a list link (would link an element to itself).
|
|
+*/
|
|
+#define NO_JUMP (-1)
|
|
+
|
|
+
|
|
+/*
|
|
+** grep "ORDER OPR" if you change these enums
|
|
+*/
|
|
+typedef enum BinOpr {
|
|
+ OPR_ADD, OPR_SUB, OPR_MUL, OPR_DIV, OPR_MOD, OPR_POW,
|
|
+ OPR_CONCAT,
|
|
+ OPR_NE, OPR_EQ,
|
|
+ OPR_LT, OPR_LE, OPR_GT, OPR_GE,
|
|
+ OPR_AND, OPR_OR,
|
|
+ OPR_NOBINOPR
|
|
+} BinOpr;
|
|
+
|
|
+
|
|
+typedef enum UnOpr { OPR_MINUS, OPR_NOT, OPR_LEN, OPR_NOUNOPR } UnOpr;
|
|
+
|
|
+
|
|
+#define getcode(fs,e) ((fs)->f->code[(e)->u.s.info])
|
|
+
|
|
+#define luaK_codeAsBx(fs,o,A,sBx) luaK_codeABx(fs,o,A,(sBx)+MAXARG_sBx)
|
|
+
|
|
+#define luaK_setmultret(fs,e) luaK_setreturns(fs, e, LUA_MULTRET)
|
|
+
|
|
+LUAI_FUNC int luaK_codeABx (FuncState *fs, OpCode o, int A, unsigned int Bx);
|
|
+LUAI_FUNC int luaK_codeABC (FuncState *fs, OpCode o, int A, int B, int C);
|
|
+LUAI_FUNC void luaK_fixline (FuncState *fs, int line);
|
|
+LUAI_FUNC void luaK_nil (FuncState *fs, int from, int n);
|
|
+LUAI_FUNC void luaK_reserveregs (FuncState *fs, int n);
|
|
+LUAI_FUNC void luaK_checkstack (FuncState *fs, int n);
|
|
+LUAI_FUNC int luaK_stringK (FuncState *fs, TString *s);
|
|
+LUAI_FUNC int luaK_numberK (FuncState *fs, lua_Number r);
|
|
+LUAI_FUNC void luaK_dischargevars (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC int luaK_exp2anyreg (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC void luaK_exp2nextreg (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC void luaK_exp2val (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC int luaK_exp2RK (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC void luaK_self (FuncState *fs, expdesc *e, expdesc *key);
|
|
+LUAI_FUNC void luaK_indexed (FuncState *fs, expdesc *t, expdesc *k);
|
|
+LUAI_FUNC void luaK_goiftrue (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC void luaK_storevar (FuncState *fs, expdesc *var, expdesc *e);
|
|
+LUAI_FUNC void luaK_setreturns (FuncState *fs, expdesc *e, int nresults);
|
|
+LUAI_FUNC void luaK_setoneret (FuncState *fs, expdesc *e);
|
|
+LUAI_FUNC int luaK_jump (FuncState *fs);
|
|
+LUAI_FUNC void luaK_ret (FuncState *fs, int first, int nret);
|
|
+LUAI_FUNC void luaK_patchlist (FuncState *fs, int list, int target);
|
|
+LUAI_FUNC void luaK_patchtohere (FuncState *fs, int list);
|
|
+LUAI_FUNC void luaK_concat (FuncState *fs, int *l1, int l2);
|
|
+LUAI_FUNC int luaK_getlabel (FuncState *fs);
|
|
+LUAI_FUNC void luaK_prefix (FuncState *fs, UnOpr op, expdesc *v);
|
|
+LUAI_FUNC void luaK_infix (FuncState *fs, BinOpr op, expdesc *v);
|
|
+LUAI_FUNC void luaK_posfix (FuncState *fs, BinOpr op, expdesc *v1, expdesc *v2);
|
|
+LUAI_FUNC void luaK_setlist (FuncState *fs, int base, int nelems, int tostore);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ldebug.c
|
|
@@ -0,0 +1,637 @@
|
|
+/*
|
|
+** $Id: ldebug.c,v 2.29.1.6 2008/05/08 16:56:26 roberto Exp $
|
|
+** Debug Interface
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <stdarg.h>
|
|
+#include <stddef.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define ldebug_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lapi.h"
|
|
+#include "lcode.h"
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+#include "lvm.h"
|
|
+
|
|
+
|
|
+
|
|
+static const char *getfuncname (lua_State *L, CallInfo *ci, const char **name);
|
|
+
|
|
+
|
|
+static int currentpc (lua_State *L, CallInfo *ci) {
|
|
+ if (!isLua(ci)) return -1; /* function is not a Lua function? */
|
|
+ if (ci == L->ci)
|
|
+ ci->savedpc = L->savedpc;
|
|
+ return pcRel(ci->savedpc, ci_func(ci)->l.p);
|
|
+}
|
|
+
|
|
+
|
|
+static int currentline (lua_State *L, CallInfo *ci) {
|
|
+ int pc = currentpc(L, ci);
|
|
+ if (pc < 0)
|
|
+ return -1; /* only active lua functions have current-line information */
|
|
+ else
|
|
+ return getline(ci_func(ci)->l.p, pc);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** this function can be called asynchronous (e.g. during a signal)
|
|
+*/
|
|
+LUA_API int lua_sethook (lua_State *L, lua_Hook func, int mask, int count) {
|
|
+ if (func == NULL || mask == 0) { /* turn off hooks? */
|
|
+ mask = 0;
|
|
+ func = NULL;
|
|
+ }
|
|
+ L->hook = func;
|
|
+ L->basehookcount = count;
|
|
+ resethookcount(L);
|
|
+ L->hookmask = cast_byte(mask);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_Hook lua_gethook (lua_State *L) {
|
|
+ return L->hook;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_gethookmask (lua_State *L) {
|
|
+ return L->hookmask;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_gethookcount (lua_State *L) {
|
|
+ return L->basehookcount;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_getstack (lua_State *L, int level, lua_Debug *ar) {
|
|
+ int status;
|
|
+ CallInfo *ci;
|
|
+ lua_lock(L);
|
|
+ for (ci = L->ci; level > 0 && ci > L->base_ci; ci--) {
|
|
+ level--;
|
|
+ if (f_isLua(ci)) /* Lua function? */
|
|
+ level -= ci->tailcalls; /* skip lost tail calls */
|
|
+ }
|
|
+ if (level == 0 && ci > L->base_ci) { /* level found? */
|
|
+ status = 1;
|
|
+ ar->i_ci = cast_int(ci - L->base_ci);
|
|
+ }
|
|
+ else if (level < 0) { /* level is of a lost tail call? */
|
|
+ status = 1;
|
|
+ ar->i_ci = 0;
|
|
+ }
|
|
+ else status = 0; /* no such level */
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+static Proto *getluaproto (CallInfo *ci) {
|
|
+ return (isLua(ci) ? ci_func(ci)->l.p : NULL);
|
|
+}
|
|
+
|
|
+
|
|
+static const char *findlocal (lua_State *L, CallInfo *ci, int n) {
|
|
+ const char *name;
|
|
+ Proto *fp = getluaproto(ci);
|
|
+ if (fp && (name = luaF_getlocalname(fp, n, currentpc(L, ci))) != NULL)
|
|
+ return name; /* is a local variable in a Lua function */
|
|
+ else {
|
|
+ StkId limit = (ci == L->ci) ? L->top : (ci+1)->func;
|
|
+ if (limit - ci->base >= n && n > 0) /* is 'n' inside 'ci' stack? */
|
|
+ return "(*temporary)";
|
|
+ else
|
|
+ return NULL;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_getlocal (lua_State *L, const lua_Debug *ar, int n) {
|
|
+ CallInfo *ci = L->base_ci + ar->i_ci;
|
|
+ const char *name = findlocal(L, ci, n);
|
|
+ lua_lock(L);
|
|
+ if (name)
|
|
+ luaA_pushobject(L, ci->base + (n - 1));
|
|
+ lua_unlock(L);
|
|
+ return name;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API const char *lua_setlocal (lua_State *L, const lua_Debug *ar, int n) {
|
|
+ CallInfo *ci = L->base_ci + ar->i_ci;
|
|
+ const char *name = findlocal(L, ci, n);
|
|
+ lua_lock(L);
|
|
+ if (name)
|
|
+ setobjs2s(L, ci->base + (n - 1), L->top - 1);
|
|
+ L->top--; /* pop value */
|
|
+ lua_unlock(L);
|
|
+ return name;
|
|
+}
|
|
+
|
|
+
|
|
+static void funcinfo (lua_Debug *ar, Closure *cl) {
|
|
+ if (cl->c.isC) {
|
|
+ ar->source = "=[C]";
|
|
+ ar->linedefined = -1;
|
|
+ ar->lastlinedefined = -1;
|
|
+ ar->what = "C";
|
|
+ }
|
|
+ else {
|
|
+ ar->source = getstr(cl->l.p->source);
|
|
+ ar->linedefined = cl->l.p->linedefined;
|
|
+ ar->lastlinedefined = cl->l.p->lastlinedefined;
|
|
+ ar->what = (ar->linedefined == 0) ? "main" : "Lua";
|
|
+ }
|
|
+ luaO_chunkid(ar->short_src, ar->source, LUA_IDSIZE);
|
|
+}
|
|
+
|
|
+
|
|
+static void info_tailcall (lua_Debug *ar) {
|
|
+ ar->name = ar->namewhat = "";
|
|
+ ar->what = "tail";
|
|
+ ar->lastlinedefined = ar->linedefined = ar->currentline = -1;
|
|
+ ar->source = "=(tail call)";
|
|
+ luaO_chunkid(ar->short_src, ar->source, LUA_IDSIZE);
|
|
+ ar->nups = 0;
|
|
+}
|
|
+
|
|
+
|
|
+static void collectvalidlines (lua_State *L, Closure *f) {
|
|
+ if (f == NULL || f->c.isC) {
|
|
+ setnilvalue(L->top);
|
|
+ }
|
|
+ else {
|
|
+ Table *t = luaH_new(L, 0, 0);
|
|
+ int *lineinfo = f->l.p->lineinfo;
|
|
+ int i;
|
|
+ for (i=0; i<f->l.p->sizelineinfo; i++)
|
|
+ setbvalue(luaH_setnum(L, t, lineinfo[i]), 1);
|
|
+ sethvalue(L, L->top, t);
|
|
+ }
|
|
+ incr_top(L);
|
|
+}
|
|
+
|
|
+
|
|
+static int auxgetinfo (lua_State *L, const char *what, lua_Debug *ar,
|
|
+ Closure *f, CallInfo *ci) {
|
|
+ int status = 1;
|
|
+ if (f == NULL) {
|
|
+ info_tailcall(ar);
|
|
+ return status;
|
|
+ }
|
|
+ for (; *what; what++) {
|
|
+ switch (*what) {
|
|
+ case 'S': {
|
|
+ funcinfo(ar, f);
|
|
+ break;
|
|
+ }
|
|
+ case 'l': {
|
|
+ ar->currentline = (ci) ? currentline(L, ci) : -1;
|
|
+ break;
|
|
+ }
|
|
+ case 'u': {
|
|
+ ar->nups = f->c.nupvalues;
|
|
+ break;
|
|
+ }
|
|
+ case 'n': {
|
|
+ ar->namewhat = (ci) ? getfuncname(L, ci, &ar->name) : NULL;
|
|
+ if (ar->namewhat == NULL) {
|
|
+ ar->namewhat = ""; /* not found */
|
|
+ ar->name = NULL;
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case 'L':
|
|
+ case 'f': /* handled by lua_getinfo */
|
|
+ break;
|
|
+ default: status = 0; /* invalid option */
|
|
+ }
|
|
+ }
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_getinfo (lua_State *L, const char *what, lua_Debug *ar) {
|
|
+ int status;
|
|
+ Closure *f = NULL;
|
|
+ CallInfo *ci = NULL;
|
|
+ lua_lock(L);
|
|
+ if (*what == '>') {
|
|
+ StkId func = L->top - 1;
|
|
+ luai_apicheck(L, ttisfunction(func));
|
|
+ what++; /* skip the '>' */
|
|
+ f = clvalue(func);
|
|
+ L->top--; /* pop function */
|
|
+ }
|
|
+ else if (ar->i_ci != 0) { /* no tail call? */
|
|
+ ci = L->base_ci + ar->i_ci;
|
|
+ lua_assert(ttisfunction(ci->func));
|
|
+ f = clvalue(ci->func);
|
|
+ }
|
|
+ status = auxgetinfo(L, what, ar, f, ci);
|
|
+ if (strchr(what, 'f')) {
|
|
+ if (f == NULL) setnilvalue(L->top);
|
|
+ else setclvalue(L, L->top, f);
|
|
+ incr_top(L);
|
|
+ }
|
|
+ if (strchr(what, 'L'))
|
|
+ collectvalidlines(L, f);
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Symbolic Execution and code checker
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+#define check(x) if (!(x)) return 0;
|
|
+
|
|
+#define checkjump(pt,pc) check(0 <= pc && pc < pt->sizecode)
|
|
+
|
|
+#define checkreg(pt,reg) check((reg) < (pt)->maxstacksize)
|
|
+
|
|
+
|
|
+
|
|
+static int precheck (const Proto *pt) {
|
|
+ check(pt->maxstacksize <= MAXSTACK);
|
|
+ check(pt->numparams+(pt->is_vararg & VARARG_HASARG) <= pt->maxstacksize);
|
|
+ check(!(pt->is_vararg & VARARG_NEEDSARG) ||
|
|
+ (pt->is_vararg & VARARG_HASARG));
|
|
+ check(pt->sizeupvalues <= pt->nups);
|
|
+ check(pt->sizelineinfo == pt->sizecode || pt->sizelineinfo == 0);
|
|
+ check(pt->sizecode > 0 && GET_OPCODE(pt->code[pt->sizecode-1]) == OP_RETURN);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+#define checkopenop(pt,pc) luaG_checkopenop((pt)->code[(pc)+1])
|
|
+
|
|
+int luaG_checkopenop (Instruction i) {
|
|
+ switch (GET_OPCODE(i)) {
|
|
+ case OP_CALL:
|
|
+ case OP_TAILCALL:
|
|
+ case OP_RETURN:
|
|
+ case OP_SETLIST: {
|
|
+ check(GETARG_B(i) == 0);
|
|
+ return 1;
|
|
+ }
|
|
+ default: return 0; /* invalid instruction after an open call */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int checkArgMode (const Proto *pt, int r, enum OpArgMask mode) {
|
|
+ switch (mode) {
|
|
+ case OpArgN: check(r == 0); break;
|
|
+ case OpArgU: break;
|
|
+ case OpArgR: checkreg(pt, r); break;
|
|
+ case OpArgK:
|
|
+ check(ISK(r) ? INDEXK(r) < pt->sizek : r < pt->maxstacksize);
|
|
+ break;
|
|
+ }
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static Instruction symbexec (const Proto *pt, int lastpc, int reg) {
|
|
+ int pc;
|
|
+ int last; /* stores position of last instruction that changed `reg' */
|
|
+ last = pt->sizecode-1; /* points to final return (a `neutral' instruction) */
|
|
+ check(precheck(pt));
|
|
+ for (pc = 0; pc < lastpc; pc++) {
|
|
+ Instruction i = pt->code[pc];
|
|
+ OpCode op = GET_OPCODE(i);
|
|
+ int a = GETARG_A(i);
|
|
+ int b = 0;
|
|
+ int c = 0;
|
|
+ check(op < NUM_OPCODES);
|
|
+ checkreg(pt, a);
|
|
+ switch (getOpMode(op)) {
|
|
+ case iABC: {
|
|
+ b = GETARG_B(i);
|
|
+ c = GETARG_C(i);
|
|
+ check(checkArgMode(pt, b, getBMode(op)));
|
|
+ check(checkArgMode(pt, c, getCMode(op)));
|
|
+ break;
|
|
+ }
|
|
+ case iABx: {
|
|
+ b = GETARG_Bx(i);
|
|
+ if (getBMode(op) == OpArgK) check(b < pt->sizek);
|
|
+ break;
|
|
+ }
|
|
+ case iAsBx: {
|
|
+ b = GETARG_sBx(i);
|
|
+ if (getBMode(op) == OpArgR) {
|
|
+ int dest = pc+1+b;
|
|
+ check(0 <= dest && dest < pt->sizecode);
|
|
+ if (dest > 0) {
|
|
+ int j;
|
|
+ /* check that it does not jump to a setlist count; this
|
|
+ is tricky, because the count from a previous setlist may
|
|
+ have the same value of an invalid setlist; so, we must
|
|
+ go all the way back to the first of them (if any) */
|
|
+ for (j = 0; j < dest; j++) {
|
|
+ Instruction d = pt->code[dest-1-j];
|
|
+ if (!(GET_OPCODE(d) == OP_SETLIST && GETARG_C(d) == 0)) break;
|
|
+ }
|
|
+ /* if 'j' is even, previous value is not a setlist (even if
|
|
+ it looks like one) */
|
|
+ check((j&1) == 0);
|
|
+ }
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ if (testAMode(op)) {
|
|
+ if (a == reg) last = pc; /* change register `a' */
|
|
+ }
|
|
+ if (testTMode(op)) {
|
|
+ check(pc+2 < pt->sizecode); /* check skip */
|
|
+ check(GET_OPCODE(pt->code[pc+1]) == OP_JMP);
|
|
+ }
|
|
+ switch (op) {
|
|
+ case OP_LOADBOOL: {
|
|
+ if (c == 1) { /* does it jump? */
|
|
+ check(pc+2 < pt->sizecode); /* check its jump */
|
|
+ check(GET_OPCODE(pt->code[pc+1]) != OP_SETLIST ||
|
|
+ GETARG_C(pt->code[pc+1]) != 0);
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case OP_LOADNIL: {
|
|
+ if (a <= reg && reg <= b)
|
|
+ last = pc; /* set registers from `a' to `b' */
|
|
+ break;
|
|
+ }
|
|
+ case OP_GETUPVAL:
|
|
+ case OP_SETUPVAL: {
|
|
+ check(b < pt->nups);
|
|
+ break;
|
|
+ }
|
|
+ case OP_GETGLOBAL:
|
|
+ case OP_SETGLOBAL: {
|
|
+ check(ttisstring(&pt->k[b]));
|
|
+ break;
|
|
+ }
|
|
+ case OP_SELF: {
|
|
+ checkreg(pt, a+1);
|
|
+ if (reg == a+1) last = pc;
|
|
+ break;
|
|
+ }
|
|
+ case OP_CONCAT: {
|
|
+ check(b < c); /* at least two operands */
|
|
+ break;
|
|
+ }
|
|
+ case OP_TFORLOOP: {
|
|
+ check(c >= 1); /* at least one result (control variable) */
|
|
+ checkreg(pt, a+2+c); /* space for results */
|
|
+ if (reg >= a+2) last = pc; /* affect all regs above its base */
|
|
+ break;
|
|
+ }
|
|
+ case OP_FORLOOP:
|
|
+ case OP_FORPREP:
|
|
+ checkreg(pt, a+3);
|
|
+ /* go through */
|
|
+ case OP_JMP: {
|
|
+ int dest = pc+1+b;
|
|
+ /* not full check and jump is forward and do not skip `lastpc'? */
|
|
+ if (reg != NO_REG && pc < dest && dest <= lastpc)
|
|
+ pc += b; /* do the jump */
|
|
+ break;
|
|
+ }
|
|
+ case OP_CALL:
|
|
+ case OP_TAILCALL: {
|
|
+ if (b != 0) {
|
|
+ checkreg(pt, a+b-1);
|
|
+ }
|
|
+ c--; /* c = num. returns */
|
|
+ if (c == LUA_MULTRET) {
|
|
+ check(checkopenop(pt, pc));
|
|
+ }
|
|
+ else if (c != 0)
|
|
+ checkreg(pt, a+c-1);
|
|
+ if (reg >= a) last = pc; /* affect all registers above base */
|
|
+ break;
|
|
+ }
|
|
+ case OP_RETURN: {
|
|
+ b--; /* b = num. returns */
|
|
+ if (b > 0) checkreg(pt, a+b-1);
|
|
+ break;
|
|
+ }
|
|
+ case OP_SETLIST: {
|
|
+ if (b > 0) checkreg(pt, a + b);
|
|
+ if (c == 0) {
|
|
+ pc++;
|
|
+ check(pc < pt->sizecode - 1);
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case OP_CLOSURE: {
|
|
+ int nup, j;
|
|
+ check(b < pt->sizep);
|
|
+ nup = pt->p[b]->nups;
|
|
+ check(pc + nup < pt->sizecode);
|
|
+ for (j = 1; j <= nup; j++) {
|
|
+ OpCode op1 = GET_OPCODE(pt->code[pc + j]);
|
|
+ check(op1 == OP_GETUPVAL || op1 == OP_MOVE);
|
|
+ }
|
|
+ if (reg != NO_REG) /* tracing? */
|
|
+ pc += nup; /* do not 'execute' these pseudo-instructions */
|
|
+ break;
|
|
+ }
|
|
+ case OP_VARARG: {
|
|
+ check((pt->is_vararg & VARARG_ISVARARG) &&
|
|
+ !(pt->is_vararg & VARARG_NEEDSARG));
|
|
+ b--;
|
|
+ if (b == LUA_MULTRET) check(checkopenop(pt, pc));
|
|
+ checkreg(pt, a+b-1);
|
|
+ break;
|
|
+ }
|
|
+ default: break;
|
|
+ }
|
|
+ }
|
|
+ return pt->code[last];
|
|
+}
|
|
+
|
|
+#undef check
|
|
+#undef checkjump
|
|
+#undef checkreg
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+int luaG_checkcode (const Proto *pt) {
|
|
+ return (symbexec(pt, pt->sizecode, NO_REG) != 0);
|
|
+}
|
|
+
|
|
+
|
|
+static const char *kname (Proto *p, int c) {
|
|
+ if (ISK(c) && ttisstring(&p->k[INDEXK(c)]))
|
|
+ return svalue(&p->k[INDEXK(c)]);
|
|
+ else
|
|
+ return "?";
|
|
+}
|
|
+
|
|
+
|
|
+static const char *getobjname (lua_State *L, CallInfo *ci, int stackpos,
|
|
+ const char **name) {
|
|
+ if (isLua(ci)) { /* a Lua function? */
|
|
+ Proto *p = ci_func(ci)->l.p;
|
|
+ int pc = currentpc(L, ci);
|
|
+ Instruction i;
|
|
+ *name = luaF_getlocalname(p, stackpos+1, pc);
|
|
+ if (*name) /* is a local? */
|
|
+ return "local";
|
|
+ i = symbexec(p, pc, stackpos); /* try symbolic execution */
|
|
+ lua_assert(pc != -1);
|
|
+ switch (GET_OPCODE(i)) {
|
|
+ case OP_GETGLOBAL: {
|
|
+ int g = GETARG_Bx(i); /* global index */
|
|
+ lua_assert(ttisstring(&p->k[g]));
|
|
+ *name = svalue(&p->k[g]);
|
|
+ return "global";
|
|
+ }
|
|
+ case OP_MOVE: {
|
|
+ int a = GETARG_A(i);
|
|
+ int b = GETARG_B(i); /* move from `b' to `a' */
|
|
+ if (b < a)
|
|
+ return getobjname(L, ci, b, name); /* get name for `b' */
|
|
+ break;
|
|
+ }
|
|
+ case OP_GETTABLE: {
|
|
+ int k = GETARG_C(i); /* key index */
|
|
+ *name = kname(p, k);
|
|
+ return "field";
|
|
+ }
|
|
+ case OP_GETUPVAL: {
|
|
+ int u = GETARG_B(i); /* upvalue index */
|
|
+ *name = p->upvalues ? getstr(p->upvalues[u]) : "?";
|
|
+ return "upvalue";
|
|
+ }
|
|
+ case OP_SELF: {
|
|
+ int k = GETARG_C(i); /* key index */
|
|
+ *name = kname(p, k);
|
|
+ return "method";
|
|
+ }
|
|
+ default: break;
|
|
+ }
|
|
+ }
|
|
+ return NULL; /* no useful name found */
|
|
+}
|
|
+
|
|
+
|
|
+static const char *getfuncname (lua_State *L, CallInfo *ci, const char **name) {
|
|
+ Instruction i;
|
|
+ if ((isLua(ci) && ci->tailcalls > 0) || !isLua(ci - 1))
|
|
+ return NULL; /* calling function is not Lua (or is unknown) */
|
|
+ ci--; /* calling function */
|
|
+ i = ci_func(ci)->l.p->code[currentpc(L, ci)];
|
|
+ if (GET_OPCODE(i) == OP_CALL || GET_OPCODE(i) == OP_TAILCALL ||
|
|
+ GET_OPCODE(i) == OP_TFORLOOP)
|
|
+ return getobjname(L, ci, GETARG_A(i), name);
|
|
+ else
|
|
+ return NULL; /* no useful name can be found */
|
|
+}
|
|
+
|
|
+
|
|
+/* only ANSI way to check whether a pointer points to an array */
|
|
+static int isinstack (CallInfo *ci, const TValue *o) {
|
|
+ StkId p;
|
|
+ for (p = ci->base; p < ci->top; p++)
|
|
+ if (o == p) return 1;
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+void luaG_typeerror (lua_State *L, const TValue *o, const char *op) {
|
|
+ const char *name = NULL;
|
|
+ const char *t = luaT_typenames[ttype(o)];
|
|
+ const char *kind = (isinstack(L->ci, o)) ?
|
|
+ getobjname(L, L->ci, cast_int(o - L->base), &name) :
|
|
+ NULL;
|
|
+ if (kind)
|
|
+ luaG_runerror(L, "attempt to %s %s " LUA_QS " (a %s value)",
|
|
+ op, kind, name, t);
|
|
+ else
|
|
+ luaG_runerror(L, "attempt to %s a %s value", op, t);
|
|
+}
|
|
+
|
|
+
|
|
+void luaG_concaterror (lua_State *L, StkId p1, StkId p2) {
|
|
+ if (ttisstring(p1) || ttisnumber(p1)) p1 = p2;
|
|
+ lua_assert(!ttisstring(p1) && !ttisnumber(p1));
|
|
+ luaG_typeerror(L, p1, "concatenate");
|
|
+}
|
|
+
|
|
+
|
|
+void luaG_aritherror (lua_State *L, const TValue *p1, const TValue *p2) {
|
|
+ TValue temp;
|
|
+ if (luaV_tonumber(p1, &temp) == NULL)
|
|
+ p2 = p1; /* first operand is wrong */
|
|
+ luaG_typeerror(L, p2, "perform arithmetic on");
|
|
+}
|
|
+
|
|
+
|
|
+int luaG_ordererror (lua_State *L, const TValue *p1, const TValue *p2) {
|
|
+ const char *t1 = luaT_typenames[ttype(p1)];
|
|
+ const char *t2 = luaT_typenames[ttype(p2)];
|
|
+ if (t1[2] == t2[2])
|
|
+ luaG_runerror(L, "attempt to compare two %s values", t1);
|
|
+ else
|
|
+ luaG_runerror(L, "attempt to compare %s with %s", t1, t2);
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static void addinfo (lua_State *L, const char *msg) {
|
|
+ CallInfo *ci = L->ci;
|
|
+ if (isLua(ci)) { /* is Lua code? */
|
|
+ char buff[LUA_IDSIZE]; /* add file:line information */
|
|
+ int line = currentline(L, ci);
|
|
+ luaO_chunkid(buff, getstr(getluaproto(ci)->source), LUA_IDSIZE);
|
|
+ luaO_pushfstring(L, "%s:%d: %s", buff, line, msg);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaG_errormsg (lua_State *L) {
|
|
+ if (L->errfunc != 0) { /* is there an error handling function? */
|
|
+ StkId errfunc = restorestack(L, L->errfunc);
|
|
+ if (!ttisfunction(errfunc)) luaD_throw(L, LUA_ERRERR);
|
|
+ setobjs2s(L, L->top, L->top - 1); /* move argument */
|
|
+ setobjs2s(L, L->top - 1, errfunc); /* push function */
|
|
+ incr_top(L);
|
|
+ luaD_call(L, L->top - 2, 1); /* call it */
|
|
+ }
|
|
+ luaD_throw(L, LUA_ERRRUN);
|
|
+}
|
|
+
|
|
+
|
|
+void luaG_runerror (lua_State *L, const char *fmt, ...) {
|
|
+ va_list argp;
|
|
+ va_start(argp, fmt);
|
|
+ addinfo(L, luaO_pushvfstring(L, fmt, argp));
|
|
+ va_end(argp);
|
|
+ luaG_errormsg(L);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ldebug.h
|
|
@@ -0,0 +1,33 @@
|
|
+/*
|
|
+** $Id: ldebug.h,v 2.3.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Auxiliary functions from Debug Interface module
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef ldebug_h
|
|
+#define ldebug_h
|
|
+
|
|
+
|
|
+#include "lstate.h"
|
|
+
|
|
+
|
|
+#define pcRel(pc, p) (cast(int, (pc) - (p)->code) - 1)
|
|
+
|
|
+#define getline(f,pc) (((f)->lineinfo) ? (f)->lineinfo[pc] : 0)
|
|
+
|
|
+#define resethookcount(L) (L->hookcount = L->basehookcount)
|
|
+
|
|
+
|
|
+LUAI_FUNC void luaG_typeerror (lua_State *L, const TValue *o,
|
|
+ const char *opname);
|
|
+LUAI_FUNC void luaG_concaterror (lua_State *L, StkId p1, StkId p2);
|
|
+LUAI_FUNC void luaG_aritherror (lua_State *L, const TValue *p1,
|
|
+ const TValue *p2);
|
|
+LUAI_FUNC int luaG_ordererror (lua_State *L, const TValue *p1,
|
|
+ const TValue *p2);
|
|
+LUAI_FUNC void luaG_runerror (lua_State *L, const char *fmt, ...);
|
|
+LUAI_FUNC void luaG_errormsg (lua_State *L);
|
|
+LUAI_FUNC int luaG_checkcode (const Proto *pt);
|
|
+LUAI_FUNC int luaG_checkopenop (Instruction i);
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ldo.c
|
|
@@ -0,0 +1,515 @@
|
|
+/*
|
|
+** $Id: ldo.c,v 2.38.1.3 2008/01/18 22:31:22 roberto Exp $
|
|
+** Stack and Call structure of Lua
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <setjmp.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define ldo_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lparser.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+#include "lundump.h"
|
|
+#include "lvm.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Error-recovery functions
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+/* chain list of long jump buffers */
|
|
+struct lua_longjmp {
|
|
+ struct lua_longjmp *previous;
|
|
+ luai_jmpbuf b;
|
|
+ volatile int status; /* error code */
|
|
+};
|
|
+
|
|
+
|
|
+void luaD_seterrorobj (lua_State *L, int errcode, StkId oldtop) {
|
|
+ switch (errcode) {
|
|
+ case LUA_ERRMEM: {
|
|
+ setsvalue2s(L, oldtop, luaS_newliteral(L, MEMERRMSG));
|
|
+ break;
|
|
+ }
|
|
+ case LUA_ERRERR: {
|
|
+ setsvalue2s(L, oldtop, luaS_newliteral(L, "error in error handling"));
|
|
+ break;
|
|
+ }
|
|
+ case LUA_ERRSYNTAX:
|
|
+ case LUA_ERRRUN: {
|
|
+ setobjs2s(L, oldtop, L->top - 1); /* error message on current top */
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ L->top = oldtop + 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void restore_stack_limit (lua_State *L) {
|
|
+ lua_assert(L->stack_last - L->stack == L->stacksize - EXTRA_STACK - 1);
|
|
+ if (L->size_ci > LUAI_MAXCALLS) { /* there was an overflow? */
|
|
+ int inuse = cast_int(L->ci - L->base_ci);
|
|
+ if (inuse + 1 < LUAI_MAXCALLS) /* can `undo' overflow? */
|
|
+ luaD_reallocCI(L, LUAI_MAXCALLS);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void resetstack (lua_State *L, int status) {
|
|
+ L->ci = L->base_ci;
|
|
+ L->base = L->ci->base;
|
|
+ luaF_close(L, L->base); /* close eventual pending closures */
|
|
+ luaD_seterrorobj(L, status, L->base);
|
|
+ L->nCcalls = L->baseCcalls;
|
|
+ L->allowhook = 1;
|
|
+ restore_stack_limit(L);
|
|
+ L->errfunc = 0;
|
|
+ L->errorJmp = NULL;
|
|
+}
|
|
+
|
|
+
|
|
+void luaD_throw (lua_State *L, int errcode) {
|
|
+ if (L->errorJmp) {
|
|
+ L->errorJmp->status = errcode;
|
|
+ LUAI_THROW(L, L->errorJmp);
|
|
+ }
|
|
+ else {
|
|
+ L->status = cast_byte(errcode);
|
|
+ if (G(L)->panic) {
|
|
+ resetstack(L, errcode);
|
|
+ lua_unlock(L);
|
|
+ G(L)->panic(L);
|
|
+ }
|
|
+ exit(EXIT_FAILURE);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+int luaD_rawrunprotected (lua_State *L, Pfunc f, void *ud) {
|
|
+ struct lua_longjmp lj;
|
|
+ lj.status = 0;
|
|
+ lj.previous = L->errorJmp; /* chain new error handler */
|
|
+ L->errorJmp = &lj;
|
|
+ LUAI_TRY(L, &lj,
|
|
+ (*f)(L, ud);
|
|
+ );
|
|
+ L->errorJmp = lj.previous; /* restore old error handler */
|
|
+ return lj.status;
|
|
+}
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+static void correctstack (lua_State *L, TValue *oldstack) {
|
|
+ CallInfo *ci;
|
|
+ GCObject *up;
|
|
+ L->top = (L->top - oldstack) + L->stack;
|
|
+ for (up = L->openupval; up != NULL; up = up->gch.next)
|
|
+ gco2uv(up)->v = (gco2uv(up)->v - oldstack) + L->stack;
|
|
+ for (ci = L->base_ci; ci <= L->ci; ci++) {
|
|
+ ci->top = (ci->top - oldstack) + L->stack;
|
|
+ ci->base = (ci->base - oldstack) + L->stack;
|
|
+ ci->func = (ci->func - oldstack) + L->stack;
|
|
+ }
|
|
+ L->base = (L->base - oldstack) + L->stack;
|
|
+}
|
|
+
|
|
+
|
|
+void luaD_reallocstack (lua_State *L, int newsize) {
|
|
+ TValue *oldstack = L->stack;
|
|
+ int realsize = newsize + 1 + EXTRA_STACK;
|
|
+ lua_assert(L->stack_last - L->stack == L->stacksize - EXTRA_STACK - 1);
|
|
+ luaM_reallocvector(L, L->stack, L->stacksize, realsize, TValue);
|
|
+ L->stacksize = realsize;
|
|
+ L->stack_last = L->stack+newsize;
|
|
+ correctstack(L, oldstack);
|
|
+}
|
|
+
|
|
+
|
|
+void luaD_reallocCI (lua_State *L, int newsize) {
|
|
+ CallInfo *oldci = L->base_ci;
|
|
+ luaM_reallocvector(L, L->base_ci, L->size_ci, newsize, CallInfo);
|
|
+ L->size_ci = newsize;
|
|
+ L->ci = (L->ci - oldci) + L->base_ci;
|
|
+ L->end_ci = L->base_ci + L->size_ci - 1;
|
|
+}
|
|
+
|
|
+
|
|
+void luaD_growstack (lua_State *L, int n) {
|
|
+ if (n <= L->stacksize) /* double size is enough? */
|
|
+ luaD_reallocstack(L, 2*L->stacksize);
|
|
+ else
|
|
+ luaD_reallocstack(L, L->stacksize + n);
|
|
+}
|
|
+
|
|
+
|
|
+static CallInfo *growCI (lua_State *L) {
|
|
+ if (L->size_ci > LUAI_MAXCALLS) /* overflow while handling overflow? */
|
|
+ luaD_throw(L, LUA_ERRERR);
|
|
+ else {
|
|
+ luaD_reallocCI(L, 2*L->size_ci);
|
|
+ if (L->size_ci > LUAI_MAXCALLS)
|
|
+ luaG_runerror(L, "stack overflow");
|
|
+ }
|
|
+ return ++L->ci;
|
|
+}
|
|
+
|
|
+
|
|
+void luaD_callhook (lua_State *L, int event, int line) {
|
|
+ lua_Hook hook = L->hook;
|
|
+ if (hook && L->allowhook) {
|
|
+ ptrdiff_t top = savestack(L, L->top);
|
|
+ ptrdiff_t ci_top = savestack(L, L->ci->top);
|
|
+ lua_Debug ar;
|
|
+ ar.event = event;
|
|
+ ar.currentline = line;
|
|
+ if (event == LUA_HOOKTAILRET)
|
|
+ ar.i_ci = 0; /* tail call; no debug information about it */
|
|
+ else
|
|
+ ar.i_ci = cast_int(L->ci - L->base_ci);
|
|
+ luaD_checkstack(L, LUA_MINSTACK); /* ensure minimum stack size */
|
|
+ L->ci->top = L->top + LUA_MINSTACK;
|
|
+ lua_assert(L->ci->top <= L->stack_last);
|
|
+ L->allowhook = 0; /* cannot call hooks inside a hook */
|
|
+ lua_unlock(L);
|
|
+ (*hook)(L, &ar);
|
|
+ lua_lock(L);
|
|
+ lua_assert(!L->allowhook);
|
|
+ L->allowhook = 1;
|
|
+ L->ci->top = restorestack(L, ci_top);
|
|
+ L->top = restorestack(L, top);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static StkId adjust_varargs (lua_State *L, Proto *p, int actual) {
|
|
+ int i;
|
|
+ int nfixargs = p->numparams;
|
|
+ Table *htab = NULL;
|
|
+ StkId base, fixed;
|
|
+ for (; actual < nfixargs; ++actual)
|
|
+ setnilvalue(L->top++);
|
|
+#if defined(LUA_COMPAT_VARARG)
|
|
+ if (p->is_vararg & VARARG_NEEDSARG) { /* compat. with old-style vararg? */
|
|
+ int nvar = actual - nfixargs; /* number of extra arguments */
|
|
+ lua_assert(p->is_vararg & VARARG_HASARG);
|
|
+ luaC_checkGC(L);
|
|
+ htab = luaH_new(L, nvar, 1); /* create `arg' table */
|
|
+ for (i=0; i<nvar; i++) /* put extra arguments into `arg' table */
|
|
+ setobj2n(L, luaH_setnum(L, htab, i+1), L->top - nvar + i);
|
|
+ /* store counter in field `n' */
|
|
+ setnvalue(luaH_setstr(L, htab, luaS_newliteral(L, "n")), cast_num(nvar));
|
|
+ }
|
|
+#endif
|
|
+ /* move fixed parameters to final position */
|
|
+ fixed = L->top - actual; /* first fixed argument */
|
|
+ base = L->top; /* final position of first argument */
|
|
+ for (i=0; i<nfixargs; i++) {
|
|
+ setobjs2s(L, L->top++, fixed+i);
|
|
+ setnilvalue(fixed+i);
|
|
+ }
|
|
+ /* add `arg' parameter */
|
|
+ if (htab) {
|
|
+ sethvalue(L, L->top++, htab);
|
|
+ lua_assert(iswhite(obj2gco(htab)));
|
|
+ }
|
|
+ return base;
|
|
+}
|
|
+
|
|
+
|
|
+static StkId tryfuncTM (lua_State *L, StkId func) {
|
|
+ const TValue *tm = luaT_gettmbyobj(L, func, TM_CALL);
|
|
+ StkId p;
|
|
+ ptrdiff_t funcr = savestack(L, func);
|
|
+ if (!ttisfunction(tm))
|
|
+ luaG_typeerror(L, func, "call");
|
|
+ /* Open a hole inside the stack at `func' */
|
|
+ for (p = L->top; p > func; p--) setobjs2s(L, p, p-1);
|
|
+ incr_top(L);
|
|
+ func = restorestack(L, funcr); /* previous call may change stack */
|
|
+ setobj2s(L, func, tm); /* tag method is the new function to be called */
|
|
+ return func;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+#define inc_ci(L) \
|
|
+ ((L->ci == L->end_ci) ? growCI(L) : \
|
|
+ (condhardstacktests(luaD_reallocCI(L, L->size_ci)), ++L->ci))
|
|
+
|
|
+
|
|
+int luaD_precall (lua_State *L, StkId func, int nresults) {
|
|
+ LClosure *cl;
|
|
+ ptrdiff_t funcr;
|
|
+ if (!ttisfunction(func)) /* `func' is not a function? */
|
|
+ func = tryfuncTM(L, func); /* check the `function' tag method */
|
|
+ funcr = savestack(L, func);
|
|
+ cl = &clvalue(func)->l;
|
|
+ L->ci->savedpc = L->savedpc;
|
|
+ if (!cl->isC) { /* Lua function? prepare its call */
|
|
+ CallInfo *ci;
|
|
+ StkId st, base;
|
|
+ Proto *p = cl->p;
|
|
+ luaD_checkstack(L, p->maxstacksize);
|
|
+ func = restorestack(L, funcr);
|
|
+ if (!p->is_vararg) { /* no varargs? */
|
|
+ base = func + 1;
|
|
+ if (L->top > base + p->numparams)
|
|
+ L->top = base + p->numparams;
|
|
+ }
|
|
+ else { /* vararg function */
|
|
+ int nargs = cast_int(L->top - func) - 1;
|
|
+ base = adjust_varargs(L, p, nargs);
|
|
+ func = restorestack(L, funcr); /* previous call may change the stack */
|
|
+ }
|
|
+ ci = inc_ci(L); /* now `enter' new function */
|
|
+ ci->func = func;
|
|
+ L->base = ci->base = base;
|
|
+ ci->top = L->base + p->maxstacksize;
|
|
+ lua_assert(ci->top <= L->stack_last);
|
|
+ L->savedpc = p->code; /* starting point */
|
|
+ ci->tailcalls = 0;
|
|
+ ci->nresults = nresults;
|
|
+ for (st = L->top; st < ci->top; st++)
|
|
+ setnilvalue(st);
|
|
+ L->top = ci->top;
|
|
+ if (L->hookmask & LUA_MASKCALL) {
|
|
+ L->savedpc++; /* hooks assume 'pc' is already incremented */
|
|
+ luaD_callhook(L, LUA_HOOKCALL, -1);
|
|
+ L->savedpc--; /* correct 'pc' */
|
|
+ }
|
|
+ return PCRLUA;
|
|
+ }
|
|
+ else { /* if is a C function, call it */
|
|
+ CallInfo *ci;
|
|
+ int n;
|
|
+ luaD_checkstack(L, LUA_MINSTACK); /* ensure minimum stack size */
|
|
+ ci = inc_ci(L); /* now `enter' new function */
|
|
+ ci->func = restorestack(L, funcr);
|
|
+ L->base = ci->base = ci->func + 1;
|
|
+ ci->top = L->top + LUA_MINSTACK;
|
|
+ lua_assert(ci->top <= L->stack_last);
|
|
+ ci->nresults = nresults;
|
|
+ if (L->hookmask & LUA_MASKCALL)
|
|
+ luaD_callhook(L, LUA_HOOKCALL, -1);
|
|
+ lua_unlock(L);
|
|
+ n = (*curr_func(L)->c.f)(L); /* do the actual call */
|
|
+ lua_lock(L);
|
|
+ if (n < 0) /* yielding? */
|
|
+ return PCRYIELD;
|
|
+ else {
|
|
+ luaD_poscall(L, L->top - n);
|
|
+ return PCRC;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static StkId callrethooks (lua_State *L, StkId firstResult) {
|
|
+ ptrdiff_t fr = savestack(L, firstResult); /* next call may change stack */
|
|
+ luaD_callhook(L, LUA_HOOKRET, -1);
|
|
+ if (f_isLua(L->ci)) { /* Lua function? */
|
|
+ while ((L->hookmask & LUA_MASKRET) && L->ci->tailcalls--) /* tail calls */
|
|
+ luaD_callhook(L, LUA_HOOKTAILRET, -1);
|
|
+ }
|
|
+ return restorestack(L, fr);
|
|
+}
|
|
+
|
|
+
|
|
+int luaD_poscall (lua_State *L, StkId firstResult) {
|
|
+ StkId res;
|
|
+ int wanted, i;
|
|
+ CallInfo *ci;
|
|
+ if (L->hookmask & LUA_MASKRET)
|
|
+ firstResult = callrethooks(L, firstResult);
|
|
+ ci = L->ci--;
|
|
+ res = ci->func; /* res == final position of 1st result */
|
|
+ wanted = ci->nresults;
|
|
+ L->base = (ci - 1)->base; /* restore base */
|
|
+ L->savedpc = (ci - 1)->savedpc; /* restore savedpc */
|
|
+ /* move results to correct place */
|
|
+ for (i = wanted; i != 0 && firstResult < L->top; i--)
|
|
+ setobjs2s(L, res++, firstResult++);
|
|
+ while (i-- > 0)
|
|
+ setnilvalue(res++);
|
|
+ L->top = res;
|
|
+ return (wanted - LUA_MULTRET); /* 0 iff wanted == LUA_MULTRET */
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Call a function (C or Lua). The function to be called is at *func.
|
|
+** The arguments are on the stack, right after the function.
|
|
+** When returns, all the results are on the stack, starting at the original
|
|
+** function position.
|
|
+*/
|
|
+void luaD_call (lua_State *L, StkId func, int nResults) {
|
|
+ if (++L->nCcalls >= LUAI_MAXCCALLS) {
|
|
+ if (L->nCcalls == LUAI_MAXCCALLS)
|
|
+ luaG_runerror(L, "C stack overflow");
|
|
+ else if (L->nCcalls >= (LUAI_MAXCCALLS + (LUAI_MAXCCALLS>>3)))
|
|
+ luaD_throw(L, LUA_ERRERR); /* error while handing stack error */
|
|
+ }
|
|
+ if (luaD_precall(L, func, nResults) == PCRLUA) /* is a Lua function? */
|
|
+ luaV_execute(L, 1); /* call it */
|
|
+ L->nCcalls--;
|
|
+ luaC_checkGC(L);
|
|
+}
|
|
+
|
|
+
|
|
+static void resume (lua_State *L, void *ud) {
|
|
+ StkId firstArg = cast(StkId, ud);
|
|
+ CallInfo *ci = L->ci;
|
|
+ if (L->status == 0) { /* start coroutine? */
|
|
+ lua_assert(ci == L->base_ci && firstArg > L->base);
|
|
+ if (luaD_precall(L, firstArg - 1, LUA_MULTRET) != PCRLUA)
|
|
+ return;
|
|
+ }
|
|
+ else { /* resuming from previous yield */
|
|
+ lua_assert(L->status == LUA_YIELD);
|
|
+ L->status = 0;
|
|
+ if (!f_isLua(ci)) { /* `common' yield? */
|
|
+ /* finish interrupted execution of `OP_CALL' */
|
|
+ lua_assert(GET_OPCODE(*((ci-1)->savedpc - 1)) == OP_CALL ||
|
|
+ GET_OPCODE(*((ci-1)->savedpc - 1)) == OP_TAILCALL);
|
|
+ if (luaD_poscall(L, firstArg)) /* complete it... */
|
|
+ L->top = L->ci->top; /* and correct top if not multiple results */
|
|
+ }
|
|
+ else /* yielded inside a hook: just continue its execution */
|
|
+ L->base = L->ci->base;
|
|
+ }
|
|
+ luaV_execute(L, cast_int(L->ci - L->base_ci));
|
|
+}
|
|
+
|
|
+
|
|
+static int resume_error (lua_State *L, const char *msg) {
|
|
+ L->top = L->ci->base;
|
|
+ setsvalue2s(L, L->top, luaS_new(L, msg));
|
|
+ incr_top(L);
|
|
+ lua_unlock(L);
|
|
+ return LUA_ERRRUN;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_resume (lua_State *L, int nargs) {
|
|
+ int status;
|
|
+ lua_lock(L);
|
|
+ if (L->status != LUA_YIELD && (L->status != 0 || L->ci != L->base_ci))
|
|
+ return resume_error(L, "cannot resume non-suspended coroutine");
|
|
+ if (L->nCcalls >= LUAI_MAXCCALLS)
|
|
+ return resume_error(L, "C stack overflow");
|
|
+ luai_userstateresume(L, nargs);
|
|
+ lua_assert(L->errfunc == 0);
|
|
+ L->baseCcalls = ++L->nCcalls;
|
|
+ status = luaD_rawrunprotected(L, resume, L->top - nargs);
|
|
+ if (status != 0) { /* error? */
|
|
+ L->status = cast_byte(status); /* mark thread as `dead' */
|
|
+ luaD_seterrorobj(L, status, L->top);
|
|
+ L->ci->top = L->top;
|
|
+ }
|
|
+ else {
|
|
+ lua_assert(L->nCcalls == L->baseCcalls);
|
|
+ status = L->status;
|
|
+ }
|
|
+ --L->nCcalls;
|
|
+ lua_unlock(L);
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API int lua_yield (lua_State *L, int nresults) {
|
|
+ luai_userstateyield(L, nresults);
|
|
+ lua_lock(L);
|
|
+ if (L->nCcalls > L->baseCcalls)
|
|
+ luaG_runerror(L, "attempt to yield across metamethod/C-call boundary");
|
|
+ L->base = L->top - nresults; /* protect stack slots below */
|
|
+ L->status = LUA_YIELD;
|
|
+ lua_unlock(L);
|
|
+ return -1;
|
|
+}
|
|
+
|
|
+
|
|
+int luaD_pcall (lua_State *L, Pfunc func, void *u,
|
|
+ ptrdiff_t old_top, ptrdiff_t ef) {
|
|
+ int status;
|
|
+ unsigned short oldnCcalls = L->nCcalls;
|
|
+ ptrdiff_t old_ci = saveci(L, L->ci);
|
|
+ lu_byte old_allowhooks = L->allowhook;
|
|
+ ptrdiff_t old_errfunc = L->errfunc;
|
|
+ L->errfunc = ef;
|
|
+ status = luaD_rawrunprotected(L, func, u);
|
|
+ if (status != 0) { /* an error occurred? */
|
|
+ StkId oldtop = restorestack(L, old_top);
|
|
+ luaF_close(L, oldtop); /* close eventual pending closures */
|
|
+ luaD_seterrorobj(L, status, oldtop);
|
|
+ L->nCcalls = oldnCcalls;
|
|
+ L->ci = restoreci(L, old_ci);
|
|
+ L->base = L->ci->base;
|
|
+ L->savedpc = L->ci->savedpc;
|
|
+ L->allowhook = old_allowhooks;
|
|
+ restore_stack_limit(L);
|
|
+ }
|
|
+ L->errfunc = old_errfunc;
|
|
+ return status;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Execute a protected parser.
|
|
+*/
|
|
+struct SParser { /* data to `f_parser' */
|
|
+ ZIO *z;
|
|
+ Mbuffer buff; /* buffer to be used by the scanner */
|
|
+ const char *name;
|
|
+};
|
|
+
|
|
+static void f_parser (lua_State *L, void *ud) {
|
|
+ int i;
|
|
+ Proto *tf;
|
|
+ Closure *cl;
|
|
+ struct SParser *p = cast(struct SParser *, ud);
|
|
+ int c = luaZ_lookahead(p->z);
|
|
+ luaC_checkGC(L);
|
|
+ tf = ((c == LUA_SIGNATURE[0]) ? luaU_undump : luaY_parser)(L, p->z,
|
|
+ &p->buff, p->name);
|
|
+ cl = luaF_newLclosure(L, tf->nups, hvalue(gt(L)));
|
|
+ cl->l.p = tf;
|
|
+ for (i = 0; i < tf->nups; i++) /* initialize eventual upvalues */
|
|
+ cl->l.upvals[i] = luaF_newupval(L);
|
|
+ setclvalue(L, L->top, cl);
|
|
+ incr_top(L);
|
|
+}
|
|
+
|
|
+
|
|
+int luaD_protectedparser (lua_State *L, ZIO *z, const char *name) {
|
|
+ struct SParser p;
|
|
+ int status;
|
|
+ p.z = z; p.name = name;
|
|
+ luaZ_initbuffer(L, &p.buff);
|
|
+ status = luaD_pcall(L, f_parser, &p, savestack(L, L->top), L->errfunc);
|
|
+ luaZ_freebuffer(L, &p.buff);
|
|
+ return status;
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ldo.h
|
|
@@ -0,0 +1,57 @@
|
|
+/*
|
|
+** $Id: ldo.h,v 2.7.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Stack and Call structure of Lua
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef ldo_h
|
|
+#define ldo_h
|
|
+
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+#define luaD_checkstack(L,n) \
|
|
+ if ((char *)L->stack_last - (char *)L->top <= (n)*(int)sizeof(TValue)) \
|
|
+ luaD_growstack(L, n); \
|
|
+ else condhardstacktests(luaD_reallocstack(L, L->stacksize - EXTRA_STACK - 1));
|
|
+
|
|
+
|
|
+#define incr_top(L) {luaD_checkstack(L,1); L->top++;}
|
|
+
|
|
+#define savestack(L,p) ((char *)(p) - (char *)L->stack)
|
|
+#define restorestack(L,n) ((TValue *)((char *)L->stack + (n)))
|
|
+
|
|
+#define saveci(L,p) ((char *)(p) - (char *)L->base_ci)
|
|
+#define restoreci(L,n) ((CallInfo *)((char *)L->base_ci + (n)))
|
|
+
|
|
+
|
|
+/* results from luaD_precall */
|
|
+#define PCRLUA 0 /* initiated a call to a Lua function */
|
|
+#define PCRC 1 /* did a call to a C function */
|
|
+#define PCRYIELD 2 /* C funtion yielded */
|
|
+
|
|
+
|
|
+/* type of protected functions, to be ran by `runprotected' */
|
|
+typedef void (*Pfunc) (lua_State *L, void *ud);
|
|
+
|
|
+LUAI_FUNC int luaD_protectedparser (lua_State *L, ZIO *z, const char *name);
|
|
+LUAI_FUNC void luaD_callhook (lua_State *L, int event, int line);
|
|
+LUAI_FUNC int luaD_precall (lua_State *L, StkId func, int nresults);
|
|
+LUAI_FUNC void luaD_call (lua_State *L, StkId func, int nResults);
|
|
+LUAI_FUNC int luaD_pcall (lua_State *L, Pfunc func, void *u,
|
|
+ ptrdiff_t oldtop, ptrdiff_t ef);
|
|
+LUAI_FUNC int luaD_poscall (lua_State *L, StkId firstResult);
|
|
+LUAI_FUNC void luaD_reallocCI (lua_State *L, int newsize);
|
|
+LUAI_FUNC void luaD_reallocstack (lua_State *L, int newsize);
|
|
+LUAI_FUNC void luaD_growstack (lua_State *L, int n);
|
|
+
|
|
+LUAI_FUNC void luaD_throw (lua_State *L, int errcode);
|
|
+LUAI_FUNC int luaD_rawrunprotected (lua_State *L, Pfunc f, void *ud);
|
|
+
|
|
+LUAI_FUNC void luaD_seterrorobj (lua_State *L, int errcode, StkId oldtop);
|
|
+
|
|
+#endif
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ldump.c
|
|
@@ -0,0 +1,164 @@
|
|
+/*
|
|
+** $Id: ldump.c,v 2.8.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** save precompiled Lua chunks
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#define ldump_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lundump.h"
|
|
+
|
|
+typedef struct {
|
|
+ lua_State* L;
|
|
+ lua_Writer writer;
|
|
+ void* data;
|
|
+ int strip;
|
|
+ int status;
|
|
+} DumpState;
|
|
+
|
|
+#define DumpMem(b,n,size,D) DumpBlock(b,(n)*(size),D)
|
|
+#define DumpVar(x,D) DumpMem(&x,1,sizeof(x),D)
|
|
+
|
|
+static void DumpBlock(const void* b, size_t size, DumpState* D)
|
|
+{
|
|
+ if (D->status==0)
|
|
+ {
|
|
+ lua_unlock(D->L);
|
|
+ D->status=(*D->writer)(D->L,b,size,D->data);
|
|
+ lua_lock(D->L);
|
|
+ }
|
|
+}
|
|
+
|
|
+static void DumpChar(int y, DumpState* D)
|
|
+{
|
|
+ char x=(char)y;
|
|
+ DumpVar(x,D);
|
|
+}
|
|
+
|
|
+static void DumpInt(int x, DumpState* D)
|
|
+{
|
|
+ DumpVar(x,D);
|
|
+}
|
|
+
|
|
+static void DumpNumber(lua_Number x, DumpState* D)
|
|
+{
|
|
+ DumpVar(x,D);
|
|
+}
|
|
+
|
|
+static void DumpVector(const void* b, int n, size_t size, DumpState* D)
|
|
+{
|
|
+ DumpInt(n,D);
|
|
+ DumpMem(b,n,size,D);
|
|
+}
|
|
+
|
|
+static void DumpString(const TString* s, DumpState* D)
|
|
+{
|
|
+ if (s==NULL || getstr(s)==NULL)
|
|
+ {
|
|
+ size_t size=0;
|
|
+ DumpVar(size,D);
|
|
+ }
|
|
+ else
|
|
+ {
|
|
+ size_t size=s->tsv.len+1; /* include trailing '\0' */
|
|
+ DumpVar(size,D);
|
|
+ DumpBlock(getstr(s),size,D);
|
|
+ }
|
|
+}
|
|
+
|
|
+#define DumpCode(f,D) DumpVector(f->code,f->sizecode,sizeof(Instruction),D)
|
|
+
|
|
+static void DumpFunction(const Proto* f, const TString* p, DumpState* D);
|
|
+
|
|
+static void DumpConstants(const Proto* f, DumpState* D)
|
|
+{
|
|
+ int i,n=f->sizek;
|
|
+ DumpInt(n,D);
|
|
+ for (i=0; i<n; i++)
|
|
+ {
|
|
+ const TValue* o=&f->k[i];
|
|
+ DumpChar(ttype(o),D);
|
|
+ switch (ttype(o))
|
|
+ {
|
|
+ case LUA_TNIL:
|
|
+ break;
|
|
+ case LUA_TBOOLEAN:
|
|
+ DumpChar(bvalue(o),D);
|
|
+ break;
|
|
+ case LUA_TNUMBER:
|
|
+ DumpNumber(nvalue(o),D);
|
|
+ break;
|
|
+ case LUA_TSTRING:
|
|
+ DumpString(rawtsvalue(o),D);
|
|
+ break;
|
|
+ default:
|
|
+ lua_assert(0); /* cannot happen */
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ n=f->sizep;
|
|
+ DumpInt(n,D);
|
|
+ for (i=0; i<n; i++) DumpFunction(f->p[i],f->source,D);
|
|
+}
|
|
+
|
|
+static void DumpDebug(const Proto* f, DumpState* D)
|
|
+{
|
|
+ int i,n;
|
|
+ n= (D->strip) ? 0 : f->sizelineinfo;
|
|
+ DumpVector(f->lineinfo,n,sizeof(int),D);
|
|
+ n= (D->strip) ? 0 : f->sizelocvars;
|
|
+ DumpInt(n,D);
|
|
+ for (i=0; i<n; i++)
|
|
+ {
|
|
+ DumpString(f->locvars[i].varname,D);
|
|
+ DumpInt(f->locvars[i].startpc,D);
|
|
+ DumpInt(f->locvars[i].endpc,D);
|
|
+ }
|
|
+ n= (D->strip) ? 0 : f->sizeupvalues;
|
|
+ DumpInt(n,D);
|
|
+ for (i=0; i<n; i++) DumpString(f->upvalues[i],D);
|
|
+}
|
|
+
|
|
+static void DumpFunction(const Proto* f, const TString* p, DumpState* D)
|
|
+{
|
|
+ DumpString((f->source==p || D->strip) ? NULL : f->source,D);
|
|
+ DumpInt(f->linedefined,D);
|
|
+ DumpInt(f->lastlinedefined,D);
|
|
+ DumpChar(f->nups,D);
|
|
+ DumpChar(f->numparams,D);
|
|
+ DumpChar(f->is_vararg,D);
|
|
+ DumpChar(f->maxstacksize,D);
|
|
+ DumpCode(f,D);
|
|
+ DumpConstants(f,D);
|
|
+ DumpDebug(f,D);
|
|
+}
|
|
+
|
|
+static void DumpHeader(DumpState* D)
|
|
+{
|
|
+ char h[LUAC_HEADERSIZE];
|
|
+ luaU_header(h);
|
|
+ DumpBlock(h,LUAC_HEADERSIZE,D);
|
|
+}
|
|
+
|
|
+/*
|
|
+** dump Lua function as precompiled chunk
|
|
+*/
|
|
+int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip)
|
|
+{
|
|
+ DumpState D;
|
|
+ D.L=L;
|
|
+ D.writer=w;
|
|
+ D.data=data;
|
|
+ D.strip=strip;
|
|
+ D.status=0;
|
|
+ DumpHeader(&D);
|
|
+ DumpFunction(f,NULL,&D);
|
|
+ return D.status;
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lfunc.c
|
|
@@ -0,0 +1,174 @@
|
|
+/*
|
|
+** $Id: lfunc.c,v 2.12.1.2 2007/12/28 14:58:43 roberto Exp $
|
|
+** Auxiliary functions to manipulate prototypes and closures
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#define lfunc_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+
|
|
+
|
|
+
|
|
+Closure *luaF_newCclosure (lua_State *L, int nelems, Table *e) {
|
|
+ Closure *c = cast(Closure *, luaM_malloc(L, sizeCclosure(nelems)));
|
|
+ luaC_link(L, obj2gco(c), LUA_TFUNCTION);
|
|
+ c->c.isC = 1;
|
|
+ c->c.env = e;
|
|
+ c->c.nupvalues = cast_byte(nelems);
|
|
+ return c;
|
|
+}
|
|
+
|
|
+
|
|
+Closure *luaF_newLclosure (lua_State *L, int nelems, Table *e) {
|
|
+ Closure *c = cast(Closure *, luaM_malloc(L, sizeLclosure(nelems)));
|
|
+ luaC_link(L, obj2gco(c), LUA_TFUNCTION);
|
|
+ c->l.isC = 0;
|
|
+ c->l.env = e;
|
|
+ c->l.nupvalues = cast_byte(nelems);
|
|
+ while (nelems--) c->l.upvals[nelems] = NULL;
|
|
+ return c;
|
|
+}
|
|
+
|
|
+
|
|
+UpVal *luaF_newupval (lua_State *L) {
|
|
+ UpVal *uv = luaM_new(L, UpVal);
|
|
+ luaC_link(L, obj2gco(uv), LUA_TUPVAL);
|
|
+ uv->v = &uv->u.value;
|
|
+ setnilvalue(uv->v);
|
|
+ return uv;
|
|
+}
|
|
+
|
|
+
|
|
+UpVal *luaF_findupval (lua_State *L, StkId level) {
|
|
+ global_State *g = G(L);
|
|
+ GCObject **pp = &L->openupval;
|
|
+ UpVal *p;
|
|
+ UpVal *uv;
|
|
+ while (*pp != NULL && (p = ngcotouv(*pp))->v >= level) {
|
|
+ lua_assert(p->v != &p->u.value);
|
|
+ if (p->v == level) { /* found a corresponding upvalue? */
|
|
+ if (isdead(g, obj2gco(p))) /* is it dead? */
|
|
+ changewhite(obj2gco(p)); /* ressurect it */
|
|
+ return p;
|
|
+ }
|
|
+ pp = &p->next;
|
|
+ }
|
|
+ uv = luaM_new(L, UpVal); /* not found: create a new one */
|
|
+ uv->tt = LUA_TUPVAL;
|
|
+ uv->marked = luaC_white(g);
|
|
+ uv->v = level; /* current value lives in the stack */
|
|
+ uv->next = *pp; /* chain it in the proper position */
|
|
+ *pp = obj2gco(uv);
|
|
+ uv->u.l.prev = &g->uvhead; /* double link it in `uvhead' list */
|
|
+ uv->u.l.next = g->uvhead.u.l.next;
|
|
+ uv->u.l.next->u.l.prev = uv;
|
|
+ g->uvhead.u.l.next = uv;
|
|
+ lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
|
+ return uv;
|
|
+}
|
|
+
|
|
+
|
|
+static void unlinkupval (UpVal *uv) {
|
|
+ lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
|
+ uv->u.l.next->u.l.prev = uv->u.l.prev; /* remove from `uvhead' list */
|
|
+ uv->u.l.prev->u.l.next = uv->u.l.next;
|
|
+}
|
|
+
|
|
+
|
|
+void luaF_freeupval (lua_State *L, UpVal *uv) {
|
|
+ if (uv->v != &uv->u.value) /* is it open? */
|
|
+ unlinkupval(uv); /* remove from open list */
|
|
+ luaM_free(L, uv); /* free upvalue */
|
|
+}
|
|
+
|
|
+
|
|
+void luaF_close (lua_State *L, StkId level) {
|
|
+ UpVal *uv;
|
|
+ global_State *g = G(L);
|
|
+ while (L->openupval != NULL && (uv = ngcotouv(L->openupval))->v >= level) {
|
|
+ GCObject *o = obj2gco(uv);
|
|
+ lua_assert(!isblack(o) && uv->v != &uv->u.value);
|
|
+ L->openupval = uv->next; /* remove from `open' list */
|
|
+ if (isdead(g, o))
|
|
+ luaF_freeupval(L, uv); /* free upvalue */
|
|
+ else {
|
|
+ unlinkupval(uv);
|
|
+ setobj(L, &uv->u.value, uv->v);
|
|
+ uv->v = &uv->u.value; /* now current value lives here */
|
|
+ luaC_linkupval(L, uv); /* link upvalue into `gcroot' list */
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+Proto *luaF_newproto (lua_State *L) {
|
|
+ Proto *f = luaM_new(L, Proto);
|
|
+ luaC_link(L, obj2gco(f), LUA_TPROTO);
|
|
+ f->k = NULL;
|
|
+ f->sizek = 0;
|
|
+ f->p = NULL;
|
|
+ f->sizep = 0;
|
|
+ f->code = NULL;
|
|
+ f->sizecode = 0;
|
|
+ f->sizelineinfo = 0;
|
|
+ f->sizeupvalues = 0;
|
|
+ f->nups = 0;
|
|
+ f->upvalues = NULL;
|
|
+ f->numparams = 0;
|
|
+ f->is_vararg = 0;
|
|
+ f->maxstacksize = 0;
|
|
+ f->lineinfo = NULL;
|
|
+ f->sizelocvars = 0;
|
|
+ f->locvars = NULL;
|
|
+ f->linedefined = 0;
|
|
+ f->lastlinedefined = 0;
|
|
+ f->source = NULL;
|
|
+ return f;
|
|
+}
|
|
+
|
|
+
|
|
+void luaF_freeproto (lua_State *L, Proto *f) {
|
|
+ luaM_freearray(L, f->code, f->sizecode, Instruction);
|
|
+ luaM_freearray(L, f->p, f->sizep, Proto *);
|
|
+ luaM_freearray(L, f->k, f->sizek, TValue);
|
|
+ luaM_freearray(L, f->lineinfo, f->sizelineinfo, int);
|
|
+ luaM_freearray(L, f->locvars, f->sizelocvars, struct LocVar);
|
|
+ luaM_freearray(L, f->upvalues, f->sizeupvalues, TString *);
|
|
+ luaM_free(L, f);
|
|
+}
|
|
+
|
|
+
|
|
+void luaF_freeclosure (lua_State *L, Closure *c) {
|
|
+ int size = (c->c.isC) ? sizeCclosure(c->c.nupvalues) :
|
|
+ sizeLclosure(c->l.nupvalues);
|
|
+ luaM_freemem(L, c, size);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Look for n-th local variable at line `line' in function `func'.
|
|
+** Returns NULL if not found.
|
|
+*/
|
|
+const char *luaF_getlocalname (const Proto *f, int local_number, int pc) {
|
|
+ int i;
|
|
+ for (i = 0; i<f->sizelocvars && f->locvars[i].startpc <= pc; i++) {
|
|
+ if (pc < f->locvars[i].endpc) { /* is variable active? */
|
|
+ local_number--;
|
|
+ if (local_number == 0)
|
|
+ return getstr(f->locvars[i].varname);
|
|
+ }
|
|
+ }
|
|
+ return NULL; /* not found */
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lfunc.h
|
|
@@ -0,0 +1,34 @@
|
|
+/*
|
|
+** $Id: lfunc.h,v 2.4.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Auxiliary functions to manipulate prototypes and closures
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lfunc_h
|
|
+#define lfunc_h
|
|
+
|
|
+
|
|
+#include "lobject.h"
|
|
+
|
|
+
|
|
+#define sizeCclosure(n) (cast(int, sizeof(CClosure)) + \
|
|
+ cast(int, sizeof(TValue)*((n)-1)))
|
|
+
|
|
+#define sizeLclosure(n) (cast(int, sizeof(LClosure)) + \
|
|
+ cast(int, sizeof(TValue *)*((n)-1)))
|
|
+
|
|
+
|
|
+LUAI_FUNC Proto *luaF_newproto (lua_State *L);
|
|
+LUAI_FUNC Closure *luaF_newCclosure (lua_State *L, int nelems, Table *e);
|
|
+LUAI_FUNC Closure *luaF_newLclosure (lua_State *L, int nelems, Table *e);
|
|
+LUAI_FUNC UpVal *luaF_newupval (lua_State *L);
|
|
+LUAI_FUNC UpVal *luaF_findupval (lua_State *L, StkId level);
|
|
+LUAI_FUNC void luaF_close (lua_State *L, StkId level);
|
|
+LUAI_FUNC void luaF_freeproto (lua_State *L, Proto *f);
|
|
+LUAI_FUNC void luaF_freeclosure (lua_State *L, Closure *c);
|
|
+LUAI_FUNC void luaF_freeupval (lua_State *L, UpVal *uv);
|
|
+LUAI_FUNC const char *luaF_getlocalname (const Proto *func, int local_number,
|
|
+ int pc);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lgc.c
|
|
@@ -0,0 +1,711 @@
|
|
+/*
|
|
+** $Id: lgc.c,v 2.38.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Garbage Collector
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define lgc_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+
|
|
+
|
|
+#define GCSTEPSIZE 1024u
|
|
+#define GCSWEEPMAX 40
|
|
+#define GCSWEEPCOST 10
|
|
+#define GCFINALIZECOST 100
|
|
+
|
|
+
|
|
+#define maskmarks cast_byte(~(bitmask(BLACKBIT)|WHITEBITS))
|
|
+
|
|
+#define makewhite(g,x) \
|
|
+ ((x)->gch.marked = cast_byte(((x)->gch.marked & maskmarks) | luaC_white(g)))
|
|
+
|
|
+#define white2gray(x) reset2bits((x)->gch.marked, WHITE0BIT, WHITE1BIT)
|
|
+#define black2gray(x) resetbit((x)->gch.marked, BLACKBIT)
|
|
+
|
|
+#define stringmark(s) reset2bits((s)->tsv.marked, WHITE0BIT, WHITE1BIT)
|
|
+
|
|
+
|
|
+#define isfinalized(u) testbit((u)->marked, FINALIZEDBIT)
|
|
+#define markfinalized(u) l_setbit((u)->marked, FINALIZEDBIT)
|
|
+
|
|
+
|
|
+#define KEYWEAK bitmask(KEYWEAKBIT)
|
|
+#define VALUEWEAK bitmask(VALUEWEAKBIT)
|
|
+
|
|
+
|
|
+
|
|
+#define markvalue(g,o) { checkconsistency(o); \
|
|
+ if (iscollectable(o) && iswhite(gcvalue(o))) reallymarkobject(g,gcvalue(o)); }
|
|
+
|
|
+#define markobject(g,t) { if (iswhite(obj2gco(t))) \
|
|
+ reallymarkobject(g, obj2gco(t)); }
|
|
+
|
|
+
|
|
+#define setthreshold(g) (g->GCthreshold = (g->estimate/100) * g->gcpause)
|
|
+
|
|
+
|
|
+static void removeentry (Node *n) {
|
|
+ lua_assert(ttisnil(gval(n)));
|
|
+ if (iscollectable(gkey(n)))
|
|
+ setttype(gkey(n), LUA_TDEADKEY); /* dead key; remove it */
|
|
+}
|
|
+
|
|
+
|
|
+static void reallymarkobject (global_State *g, GCObject *o) {
|
|
+ lua_assert(iswhite(o) && !isdead(g, o));
|
|
+ white2gray(o);
|
|
+ switch (o->gch.tt) {
|
|
+ case LUA_TSTRING: {
|
|
+ return;
|
|
+ }
|
|
+ case LUA_TUSERDATA: {
|
|
+ Table *mt = gco2u(o)->metatable;
|
|
+ gray2black(o); /* udata are never gray */
|
|
+ if (mt) markobject(g, mt);
|
|
+ markobject(g, gco2u(o)->env);
|
|
+ return;
|
|
+ }
|
|
+ case LUA_TUPVAL: {
|
|
+ UpVal *uv = gco2uv(o);
|
|
+ markvalue(g, uv->v);
|
|
+ if (uv->v == &uv->u.value) /* closed? */
|
|
+ gray2black(o); /* open upvalues are never black */
|
|
+ return;
|
|
+ }
|
|
+ case LUA_TFUNCTION: {
|
|
+ gco2cl(o)->c.gclist = g->gray;
|
|
+ g->gray = o;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TTABLE: {
|
|
+ gco2h(o)->gclist = g->gray;
|
|
+ g->gray = o;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TTHREAD: {
|
|
+ gco2th(o)->gclist = g->gray;
|
|
+ g->gray = o;
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TPROTO: {
|
|
+ gco2p(o)->gclist = g->gray;
|
|
+ g->gray = o;
|
|
+ break;
|
|
+ }
|
|
+ default: lua_assert(0);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void marktmu (global_State *g) {
|
|
+ GCObject *u = g->tmudata;
|
|
+ if (u) {
|
|
+ do {
|
|
+ u = u->gch.next;
|
|
+ makewhite(g, u); /* may be marked, if left from previous GC */
|
|
+ reallymarkobject(g, u);
|
|
+ } while (u != g->tmudata);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/* move `dead' udata that need finalization to list `tmudata' */
|
|
+size_t luaC_separateudata (lua_State *L, int all) {
|
|
+ global_State *g = G(L);
|
|
+ size_t deadmem = 0;
|
|
+ GCObject **p = &g->mainthread->next;
|
|
+ GCObject *curr;
|
|
+ while ((curr = *p) != NULL) {
|
|
+ if (!(iswhite(curr) || all) || isfinalized(gco2u(curr)))
|
|
+ p = &curr->gch.next; /* don't bother with them */
|
|
+ else if (fasttm(L, gco2u(curr)->metatable, TM_GC) == NULL) {
|
|
+ markfinalized(gco2u(curr)); /* don't need finalization */
|
|
+ p = &curr->gch.next;
|
|
+ }
|
|
+ else { /* must call its gc method */
|
|
+ deadmem += sizeudata(gco2u(curr));
|
|
+ markfinalized(gco2u(curr));
|
|
+ *p = curr->gch.next;
|
|
+ /* link `curr' at the end of `tmudata' list */
|
|
+ if (g->tmudata == NULL) /* list is empty? */
|
|
+ g->tmudata = curr->gch.next = curr; /* creates a circular list */
|
|
+ else {
|
|
+ curr->gch.next = g->tmudata->gch.next;
|
|
+ g->tmudata->gch.next = curr;
|
|
+ g->tmudata = curr;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ return deadmem;
|
|
+}
|
|
+
|
|
+
|
|
+static int traversetable (global_State *g, Table *h) {
|
|
+ int i;
|
|
+ int weakkey = 0;
|
|
+ int weakvalue = 0;
|
|
+ const TValue *mode;
|
|
+ if (h->metatable)
|
|
+ markobject(g, h->metatable);
|
|
+ mode = gfasttm(g, h->metatable, TM_MODE);
|
|
+ if (mode && ttisstring(mode)) { /* is there a weak mode? */
|
|
+ weakkey = (strchr(svalue(mode), 'k') != NULL);
|
|
+ weakvalue = (strchr(svalue(mode), 'v') != NULL);
|
|
+ if (weakkey || weakvalue) { /* is really weak? */
|
|
+ h->marked &= ~(KEYWEAK | VALUEWEAK); /* clear bits */
|
|
+ h->marked |= cast_byte((weakkey << KEYWEAKBIT) |
|
|
+ (weakvalue << VALUEWEAKBIT));
|
|
+ h->gclist = g->weak; /* must be cleared after GC, ... */
|
|
+ g->weak = obj2gco(h); /* ... so put in the appropriate list */
|
|
+ }
|
|
+ }
|
|
+ if (weakkey && weakvalue) return 1;
|
|
+ if (!weakvalue) {
|
|
+ i = h->sizearray;
|
|
+ while (i--)
|
|
+ markvalue(g, &h->array[i]);
|
|
+ }
|
|
+ i = sizenode(h);
|
|
+ while (i--) {
|
|
+ Node *n = gnode(h, i);
|
|
+ lua_assert(ttype(gkey(n)) != LUA_TDEADKEY || ttisnil(gval(n)));
|
|
+ if (ttisnil(gval(n)))
|
|
+ removeentry(n); /* remove empty entries */
|
|
+ else {
|
|
+ lua_assert(!ttisnil(gkey(n)));
|
|
+ if (!weakkey) markvalue(g, gkey(n));
|
|
+ if (!weakvalue) markvalue(g, gval(n));
|
|
+ }
|
|
+ }
|
|
+ return weakkey || weakvalue;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** All marks are conditional because a GC may happen while the
|
|
+** prototype is still being created
|
|
+*/
|
|
+static void traverseproto (global_State *g, Proto *f) {
|
|
+ int i;
|
|
+ if (f->source) stringmark(f->source);
|
|
+ for (i=0; i<f->sizek; i++) /* mark literals */
|
|
+ markvalue(g, &f->k[i]);
|
|
+ for (i=0; i<f->sizeupvalues; i++) { /* mark upvalue names */
|
|
+ if (f->upvalues[i])
|
|
+ stringmark(f->upvalues[i]);
|
|
+ }
|
|
+ for (i=0; i<f->sizep; i++) { /* mark nested protos */
|
|
+ if (f->p[i])
|
|
+ markobject(g, f->p[i]);
|
|
+ }
|
|
+ for (i=0; i<f->sizelocvars; i++) { /* mark local-variable names */
|
|
+ if (f->locvars[i].varname)
|
|
+ stringmark(f->locvars[i].varname);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+static void traverseclosure (global_State *g, Closure *cl) {
|
|
+ markobject(g, cl->c.env);
|
|
+ if (cl->c.isC) {
|
|
+ int i;
|
|
+ for (i=0; i<cl->c.nupvalues; i++) /* mark its upvalues */
|
|
+ markvalue(g, &cl->c.upvalue[i]);
|
|
+ }
|
|
+ else {
|
|
+ int i;
|
|
+ lua_assert(cl->l.nupvalues == cl->l.p->nups);
|
|
+ markobject(g, cl->l.p);
|
|
+ for (i=0; i<cl->l.nupvalues; i++) /* mark its upvalues */
|
|
+ markobject(g, cl->l.upvals[i]);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void checkstacksizes (lua_State *L, StkId max) {
|
|
+ int ci_used = cast_int(L->ci - L->base_ci); /* number of `ci' in use */
|
|
+ int s_used = cast_int(max - L->stack); /* part of stack in use */
|
|
+ if (L->size_ci > LUAI_MAXCALLS) /* handling overflow? */
|
|
+ return; /* do not touch the stacks */
|
|
+ if (4*ci_used < L->size_ci && 2*BASIC_CI_SIZE < L->size_ci)
|
|
+ luaD_reallocCI(L, L->size_ci/2); /* still big enough... */
|
|
+ condhardstacktests(luaD_reallocCI(L, ci_used + 1));
|
|
+ if (4*s_used < L->stacksize &&
|
|
+ 2*(BASIC_STACK_SIZE+EXTRA_STACK) < L->stacksize)
|
|
+ luaD_reallocstack(L, L->stacksize/2); /* still big enough... */
|
|
+ condhardstacktests(luaD_reallocstack(L, s_used));
|
|
+}
|
|
+
|
|
+
|
|
+static void traversestack (global_State *g, lua_State *l) {
|
|
+ StkId o, lim;
|
|
+ CallInfo *ci;
|
|
+ markvalue(g, gt(l));
|
|
+ lim = l->top;
|
|
+ for (ci = l->base_ci; ci <= l->ci; ci++) {
|
|
+ lua_assert(ci->top <= l->stack_last);
|
|
+ if (lim < ci->top) lim = ci->top;
|
|
+ }
|
|
+ for (o = l->stack; o < l->top; o++)
|
|
+ markvalue(g, o);
|
|
+ for (; o <= lim; o++)
|
|
+ setnilvalue(o);
|
|
+ checkstacksizes(l, lim);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** traverse one gray object, turning it to black.
|
|
+** Returns `quantity' traversed.
|
|
+*/
|
|
+static l_mem propagatemark (global_State *g) {
|
|
+ GCObject *o = g->gray;
|
|
+ lua_assert(isgray(o));
|
|
+ gray2black(o);
|
|
+ switch (o->gch.tt) {
|
|
+ case LUA_TTABLE: {
|
|
+ Table *h = gco2h(o);
|
|
+ g->gray = h->gclist;
|
|
+ if (traversetable(g, h)) /* table is weak? */
|
|
+ black2gray(o); /* keep it gray */
|
|
+ return sizeof(Table) + sizeof(TValue) * h->sizearray +
|
|
+ sizeof(Node) * sizenode(h);
|
|
+ }
|
|
+ case LUA_TFUNCTION: {
|
|
+ Closure *cl = gco2cl(o);
|
|
+ g->gray = cl->c.gclist;
|
|
+ traverseclosure(g, cl);
|
|
+ return (cl->c.isC) ? sizeCclosure(cl->c.nupvalues) :
|
|
+ sizeLclosure(cl->l.nupvalues);
|
|
+ }
|
|
+ case LUA_TTHREAD: {
|
|
+ lua_State *th = gco2th(o);
|
|
+ g->gray = th->gclist;
|
|
+ th->gclist = g->grayagain;
|
|
+ g->grayagain = o;
|
|
+ black2gray(o);
|
|
+ traversestack(g, th);
|
|
+ return sizeof(lua_State) + sizeof(TValue) * th->stacksize +
|
|
+ sizeof(CallInfo) * th->size_ci;
|
|
+ }
|
|
+ case LUA_TPROTO: {
|
|
+ Proto *p = gco2p(o);
|
|
+ g->gray = p->gclist;
|
|
+ traverseproto(g, p);
|
|
+ return sizeof(Proto) + sizeof(Instruction) * p->sizecode +
|
|
+ sizeof(Proto *) * p->sizep +
|
|
+ sizeof(TValue) * p->sizek +
|
|
+ sizeof(int) * p->sizelineinfo +
|
|
+ sizeof(LocVar) * p->sizelocvars +
|
|
+ sizeof(TString *) * p->sizeupvalues;
|
|
+ }
|
|
+ default: lua_assert(0); return 0;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static size_t propagateall (global_State *g) {
|
|
+ size_t m = 0;
|
|
+ while (g->gray) m += propagatemark(g);
|
|
+ return m;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** The next function tells whether a key or value can be cleared from
|
|
+** a weak table. Non-collectable objects are never removed from weak
|
|
+** tables. Strings behave as `values', so are never removed too. for
|
|
+** other objects: if really collected, cannot keep them; for userdata
|
|
+** being finalized, keep them in keys, but not in values
|
|
+*/
|
|
+static int iscleared (const TValue *o, int iskey) {
|
|
+ if (!iscollectable(o)) return 0;
|
|
+ if (ttisstring(o)) {
|
|
+ stringmark(rawtsvalue(o)); /* strings are `values', so are never weak */
|
|
+ return 0;
|
|
+ }
|
|
+ return iswhite(gcvalue(o)) ||
|
|
+ (ttisuserdata(o) && (!iskey && isfinalized(uvalue(o))));
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** clear collected entries from weaktables
|
|
+*/
|
|
+static void cleartable (GCObject *l) {
|
|
+ while (l) {
|
|
+ Table *h = gco2h(l);
|
|
+ int i = h->sizearray;
|
|
+ lua_assert(testbit(h->marked, VALUEWEAKBIT) ||
|
|
+ testbit(h->marked, KEYWEAKBIT));
|
|
+ if (testbit(h->marked, VALUEWEAKBIT)) {
|
|
+ while (i--) {
|
|
+ TValue *o = &h->array[i];
|
|
+ if (iscleared(o, 0)) /* value was collected? */
|
|
+ setnilvalue(o); /* remove value */
|
|
+ }
|
|
+ }
|
|
+ i = sizenode(h);
|
|
+ while (i--) {
|
|
+ Node *n = gnode(h, i);
|
|
+ if (!ttisnil(gval(n)) && /* non-empty entry? */
|
|
+ (iscleared(key2tval(n), 1) || iscleared(gval(n), 0))) {
|
|
+ setnilvalue(gval(n)); /* remove value ... */
|
|
+ removeentry(n); /* remove entry from table */
|
|
+ }
|
|
+ }
|
|
+ l = h->gclist;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void freeobj (lua_State *L, GCObject *o) {
|
|
+ switch (o->gch.tt) {
|
|
+ case LUA_TPROTO: luaF_freeproto(L, gco2p(o)); break;
|
|
+ case LUA_TFUNCTION: luaF_freeclosure(L, gco2cl(o)); break;
|
|
+ case LUA_TUPVAL: luaF_freeupval(L, gco2uv(o)); break;
|
|
+ case LUA_TTABLE: luaH_free(L, gco2h(o)); break;
|
|
+ case LUA_TTHREAD: {
|
|
+ lua_assert(gco2th(o) != L && gco2th(o) != G(L)->mainthread);
|
|
+ luaE_freethread(L, gco2th(o));
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TSTRING: {
|
|
+ G(L)->strt.nuse--;
|
|
+ luaM_freemem(L, o, sizestring(gco2ts(o)));
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TUSERDATA: {
|
|
+ luaM_freemem(L, o, sizeudata(gco2u(o)));
|
|
+ break;
|
|
+ }
|
|
+ default: lua_assert(0);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+#define sweepwholelist(L,p) sweeplist(L,p,MAX_LUMEM)
|
|
+
|
|
+
|
|
+static GCObject **sweeplist (lua_State *L, GCObject **p, lu_mem count) {
|
|
+ GCObject *curr;
|
|
+ global_State *g = G(L);
|
|
+ int deadmask = otherwhite(g);
|
|
+ while ((curr = *p) != NULL && count-- > 0) {
|
|
+ if (curr->gch.tt == LUA_TTHREAD) /* sweep open upvalues of each thread */
|
|
+ sweepwholelist(L, &gco2th(curr)->openupval);
|
|
+ if ((curr->gch.marked ^ WHITEBITS) & deadmask) { /* not dead? */
|
|
+ lua_assert(!isdead(g, curr) || testbit(curr->gch.marked, FIXEDBIT));
|
|
+ makewhite(g, curr); /* make it white (for next cycle) */
|
|
+ p = &curr->gch.next;
|
|
+ }
|
|
+ else { /* must erase `curr' */
|
|
+ lua_assert(isdead(g, curr) || deadmask == bitmask(SFIXEDBIT));
|
|
+ *p = curr->gch.next;
|
|
+ if (curr == g->rootgc) /* is the first element of the list? */
|
|
+ g->rootgc = curr->gch.next; /* adjust first */
|
|
+ freeobj(L, curr);
|
|
+ }
|
|
+ }
|
|
+ return p;
|
|
+}
|
|
+
|
|
+
|
|
+static void checkSizes (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ /* check size of string hash */
|
|
+ if (g->strt.nuse < cast(lu_int32, g->strt.size/4) &&
|
|
+ g->strt.size > MINSTRTABSIZE*2)
|
|
+ luaS_resize(L, g->strt.size/2); /* table is too big */
|
|
+ /* check size of buffer */
|
|
+ if (luaZ_sizebuffer(&g->buff) > LUA_MINBUFFER*2) { /* buffer too big? */
|
|
+ size_t newsize = luaZ_sizebuffer(&g->buff) / 2;
|
|
+ luaZ_resizebuffer(L, &g->buff, newsize);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void GCTM (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ GCObject *o = g->tmudata->gch.next; /* get first element */
|
|
+ Udata *udata = rawgco2u(o);
|
|
+ const TValue *tm;
|
|
+ /* remove udata from `tmudata' */
|
|
+ if (o == g->tmudata) /* last element? */
|
|
+ g->tmudata = NULL;
|
|
+ else
|
|
+ g->tmudata->gch.next = udata->uv.next;
|
|
+ udata->uv.next = g->mainthread->next; /* return it to `root' list */
|
|
+ g->mainthread->next = o;
|
|
+ makewhite(g, o);
|
|
+ tm = fasttm(L, udata->uv.metatable, TM_GC);
|
|
+ if (tm != NULL) {
|
|
+ lu_byte oldah = L->allowhook;
|
|
+ lu_mem oldt = g->GCthreshold;
|
|
+ L->allowhook = 0; /* stop debug hooks during GC tag method */
|
|
+ g->GCthreshold = 2*g->totalbytes; /* avoid GC steps */
|
|
+ setobj2s(L, L->top, tm);
|
|
+ setuvalue(L, L->top+1, udata);
|
|
+ L->top += 2;
|
|
+ luaD_call(L, L->top - 2, 0);
|
|
+ L->allowhook = oldah; /* restore hooks */
|
|
+ g->GCthreshold = oldt; /* restore threshold */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Call all GC tag methods
|
|
+*/
|
|
+void luaC_callGCTM (lua_State *L) {
|
|
+ while (G(L)->tmudata)
|
|
+ GCTM(L);
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_freeall (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ int i;
|
|
+ g->currentwhite = WHITEBITS | bitmask(SFIXEDBIT); /* mask to collect all elements */
|
|
+ sweepwholelist(L, &g->rootgc);
|
|
+ for (i = 0; i < g->strt.size; i++) /* free all string lists */
|
|
+ sweepwholelist(L, &g->strt.hash[i]);
|
|
+}
|
|
+
|
|
+
|
|
+static void markmt (global_State *g) {
|
|
+ int i;
|
|
+ for (i=0; i<NUM_TAGS; i++)
|
|
+ if (g->mt[i]) markobject(g, g->mt[i]);
|
|
+}
|
|
+
|
|
+
|
|
+/* mark root set */
|
|
+static void markroot (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ g->gray = NULL;
|
|
+ g->grayagain = NULL;
|
|
+ g->weak = NULL;
|
|
+ markobject(g, g->mainthread);
|
|
+ /* make global table be traversed before main stack */
|
|
+ markvalue(g, gt(g->mainthread));
|
|
+ markvalue(g, registry(L));
|
|
+ markmt(g);
|
|
+ g->gcstate = GCSpropagate;
|
|
+}
|
|
+
|
|
+
|
|
+static void remarkupvals (global_State *g) {
|
|
+ UpVal *uv;
|
|
+ for (uv = g->uvhead.u.l.next; uv != &g->uvhead; uv = uv->u.l.next) {
|
|
+ lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
|
+ if (isgray(obj2gco(uv)))
|
|
+ markvalue(g, uv->v);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void atomic (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ size_t udsize; /* total size of userdata to be finalized */
|
|
+ /* remark occasional upvalues of (maybe) dead threads */
|
|
+ remarkupvals(g);
|
|
+ /* traverse objects cautch by write barrier and by 'remarkupvals' */
|
|
+ propagateall(g);
|
|
+ /* remark weak tables */
|
|
+ g->gray = g->weak;
|
|
+ g->weak = NULL;
|
|
+ lua_assert(!iswhite(obj2gco(g->mainthread)));
|
|
+ markobject(g, L); /* mark running thread */
|
|
+ markmt(g); /* mark basic metatables (again) */
|
|
+ propagateall(g);
|
|
+ /* remark gray again */
|
|
+ g->gray = g->grayagain;
|
|
+ g->grayagain = NULL;
|
|
+ propagateall(g);
|
|
+ udsize = luaC_separateudata(L, 0); /* separate userdata to be finalized */
|
|
+ marktmu(g); /* mark `preserved' userdata */
|
|
+ udsize += propagateall(g); /* remark, to propagate `preserveness' */
|
|
+ cleartable(g->weak); /* remove collected objects from weak tables */
|
|
+ /* flip current white */
|
|
+ g->currentwhite = cast_byte(otherwhite(g));
|
|
+ g->sweepstrgc = 0;
|
|
+ g->sweepgc = &g->rootgc;
|
|
+ g->gcstate = GCSsweepstring;
|
|
+ g->estimate = g->totalbytes - udsize; /* first estimate */
|
|
+}
|
|
+
|
|
+
|
|
+static l_mem singlestep (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ /*lua_checkmemory(L);*/
|
|
+ switch (g->gcstate) {
|
|
+ case GCSpause: {
|
|
+ markroot(L); /* start a new collection */
|
|
+ return 0;
|
|
+ }
|
|
+ case GCSpropagate: {
|
|
+ if (g->gray)
|
|
+ return propagatemark(g);
|
|
+ else { /* no more `gray' objects */
|
|
+ atomic(L); /* finish mark phase */
|
|
+ return 0;
|
|
+ }
|
|
+ }
|
|
+ case GCSsweepstring: {
|
|
+ lu_mem old = g->totalbytes;
|
|
+ sweepwholelist(L, &g->strt.hash[g->sweepstrgc++]);
|
|
+ if (g->sweepstrgc >= g->strt.size) /* nothing more to sweep? */
|
|
+ g->gcstate = GCSsweep; /* end sweep-string phase */
|
|
+ lua_assert(old >= g->totalbytes);
|
|
+ g->estimate -= old - g->totalbytes;
|
|
+ return GCSWEEPCOST;
|
|
+ }
|
|
+ case GCSsweep: {
|
|
+ lu_mem old = g->totalbytes;
|
|
+ g->sweepgc = sweeplist(L, g->sweepgc, GCSWEEPMAX);
|
|
+ if (*g->sweepgc == NULL) { /* nothing more to sweep? */
|
|
+ checkSizes(L);
|
|
+ g->gcstate = GCSfinalize; /* end sweep phase */
|
|
+ }
|
|
+ lua_assert(old >= g->totalbytes);
|
|
+ g->estimate -= old - g->totalbytes;
|
|
+ return GCSWEEPMAX*GCSWEEPCOST;
|
|
+ }
|
|
+ case GCSfinalize: {
|
|
+ if (g->tmudata) {
|
|
+ GCTM(L);
|
|
+ if (g->estimate > GCFINALIZECOST)
|
|
+ g->estimate -= GCFINALIZECOST;
|
|
+ return GCFINALIZECOST;
|
|
+ }
|
|
+ else {
|
|
+ g->gcstate = GCSpause; /* end collection */
|
|
+ g->gcdept = 0;
|
|
+ return 0;
|
|
+ }
|
|
+ }
|
|
+ default: lua_assert(0); return 0;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_step (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ l_mem lim = (GCSTEPSIZE/100) * g->gcstepmul;
|
|
+ if (lim == 0)
|
|
+ lim = (MAX_LUMEM-1)/2; /* no limit */
|
|
+ g->gcdept += g->totalbytes - g->GCthreshold;
|
|
+ do {
|
|
+ lim -= singlestep(L);
|
|
+ if (g->gcstate == GCSpause)
|
|
+ break;
|
|
+ } while (lim > 0);
|
|
+ if (g->gcstate != GCSpause) {
|
|
+ if (g->gcdept < GCSTEPSIZE)
|
|
+ g->GCthreshold = g->totalbytes + GCSTEPSIZE; /* - lim/g->gcstepmul;*/
|
|
+ else {
|
|
+ g->gcdept -= GCSTEPSIZE;
|
|
+ g->GCthreshold = g->totalbytes;
|
|
+ }
|
|
+ }
|
|
+ else {
|
|
+ lua_assert(g->totalbytes >= g->estimate);
|
|
+ setthreshold(g);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_fullgc (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ if (g->gcstate <= GCSpropagate) {
|
|
+ /* reset sweep marks to sweep all elements (returning them to white) */
|
|
+ g->sweepstrgc = 0;
|
|
+ g->sweepgc = &g->rootgc;
|
|
+ /* reset other collector lists */
|
|
+ g->gray = NULL;
|
|
+ g->grayagain = NULL;
|
|
+ g->weak = NULL;
|
|
+ g->gcstate = GCSsweepstring;
|
|
+ }
|
|
+ lua_assert(g->gcstate != GCSpause && g->gcstate != GCSpropagate);
|
|
+ /* finish any pending sweep phase */
|
|
+ while (g->gcstate != GCSfinalize) {
|
|
+ lua_assert(g->gcstate == GCSsweepstring || g->gcstate == GCSsweep);
|
|
+ singlestep(L);
|
|
+ }
|
|
+ markroot(L);
|
|
+ while (g->gcstate != GCSpause) {
|
|
+ singlestep(L);
|
|
+ }
|
|
+ setthreshold(g);
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_barrierf (lua_State *L, GCObject *o, GCObject *v) {
|
|
+ global_State *g = G(L);
|
|
+ lua_assert(isblack(o) && iswhite(v) && !isdead(g, v) && !isdead(g, o));
|
|
+ lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
|
|
+ lua_assert(ttype(&o->gch) != LUA_TTABLE);
|
|
+ /* must keep invariant? */
|
|
+ if (g->gcstate == GCSpropagate)
|
|
+ reallymarkobject(g, v); /* restore invariant */
|
|
+ else /* don't mind */
|
|
+ makewhite(g, o); /* mark as white just to avoid other barriers */
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_barrierback (lua_State *L, Table *t) {
|
|
+ global_State *g = G(L);
|
|
+ GCObject *o = obj2gco(t);
|
|
+ lua_assert(isblack(o) && !isdead(g, o));
|
|
+ lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
|
|
+ black2gray(o); /* make table gray (again) */
|
|
+ t->gclist = g->grayagain;
|
|
+ g->grayagain = o;
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_link (lua_State *L, GCObject *o, lu_byte tt) {
|
|
+ global_State *g = G(L);
|
|
+ o->gch.next = g->rootgc;
|
|
+ g->rootgc = o;
|
|
+ o->gch.marked = luaC_white(g);
|
|
+ o->gch.tt = tt;
|
|
+}
|
|
+
|
|
+
|
|
+void luaC_linkupval (lua_State *L, UpVal *uv) {
|
|
+ global_State *g = G(L);
|
|
+ GCObject *o = obj2gco(uv);
|
|
+ o->gch.next = g->rootgc; /* link upvalue into `rootgc' list */
|
|
+ g->rootgc = o;
|
|
+ if (isgray(o)) {
|
|
+ if (g->gcstate == GCSpropagate) {
|
|
+ gray2black(o); /* closed upvalues need barrier */
|
|
+ luaC_barrier(L, uv, uv->v);
|
|
+ }
|
|
+ else { /* sweep phase: sweep it (turning it into white) */
|
|
+ makewhite(g, o);
|
|
+ lua_assert(g->gcstate != GCSfinalize && g->gcstate != GCSpause);
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lgc.h
|
|
@@ -0,0 +1,110 @@
|
|
+/*
|
|
+** $Id: lgc.h,v 2.15.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Garbage Collector
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lgc_h
|
|
+#define lgc_h
|
|
+
|
|
+
|
|
+#include "lobject.h"
|
|
+
|
|
+
|
|
+/*
|
|
+** Possible states of the Garbage Collector
|
|
+*/
|
|
+#define GCSpause 0
|
|
+#define GCSpropagate 1
|
|
+#define GCSsweepstring 2
|
|
+#define GCSsweep 3
|
|
+#define GCSfinalize 4
|
|
+
|
|
+
|
|
+/*
|
|
+** some userful bit tricks
|
|
+*/
|
|
+#define resetbits(x,m) ((x) &= cast(lu_byte, ~(m)))
|
|
+#define setbits(x,m) ((x) |= (m))
|
|
+#define testbits(x,m) ((x) & (m))
|
|
+#define bitmask(b) (1<<(b))
|
|
+#define bit2mask(b1,b2) (bitmask(b1) | bitmask(b2))
|
|
+#define l_setbit(x,b) setbits(x, bitmask(b))
|
|
+#define resetbit(x,b) resetbits(x, bitmask(b))
|
|
+#define testbit(x,b) testbits(x, bitmask(b))
|
|
+#define set2bits(x,b1,b2) setbits(x, (bit2mask(b1, b2)))
|
|
+#define reset2bits(x,b1,b2) resetbits(x, (bit2mask(b1, b2)))
|
|
+#define test2bits(x,b1,b2) testbits(x, (bit2mask(b1, b2)))
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Layout for bit use in `marked' field:
|
|
+** bit 0 - object is white (type 0)
|
|
+** bit 1 - object is white (type 1)
|
|
+** bit 2 - object is black
|
|
+** bit 3 - for userdata: has been finalized
|
|
+** bit 3 - for tables: has weak keys
|
|
+** bit 4 - for tables: has weak values
|
|
+** bit 5 - object is fixed (should not be collected)
|
|
+** bit 6 - object is "super" fixed (only the main thread)
|
|
+*/
|
|
+
|
|
+
|
|
+#define WHITE0BIT 0
|
|
+#define WHITE1BIT 1
|
|
+#define BLACKBIT 2
|
|
+#define FINALIZEDBIT 3
|
|
+#define KEYWEAKBIT 3
|
|
+#define VALUEWEAKBIT 4
|
|
+#define FIXEDBIT 5
|
|
+#define SFIXEDBIT 6
|
|
+#define WHITEBITS bit2mask(WHITE0BIT, WHITE1BIT)
|
|
+
|
|
+
|
|
+#define iswhite(x) test2bits((x)->gch.marked, WHITE0BIT, WHITE1BIT)
|
|
+#define isblack(x) testbit((x)->gch.marked, BLACKBIT)
|
|
+#define isgray(x) (!isblack(x) && !iswhite(x))
|
|
+
|
|
+#define otherwhite(g) (g->currentwhite ^ WHITEBITS)
|
|
+#define isdead(g,v) ((v)->gch.marked & otherwhite(g) & WHITEBITS)
|
|
+
|
|
+#define changewhite(x) ((x)->gch.marked ^= WHITEBITS)
|
|
+#define gray2black(x) l_setbit((x)->gch.marked, BLACKBIT)
|
|
+
|
|
+#define valiswhite(x) (iscollectable(x) && iswhite(gcvalue(x)))
|
|
+
|
|
+#define luaC_white(g) cast(lu_byte, (g)->currentwhite & WHITEBITS)
|
|
+
|
|
+
|
|
+#define luaC_checkGC(L) { \
|
|
+ condhardstacktests(luaD_reallocstack(L, L->stacksize - EXTRA_STACK - 1)); \
|
|
+ if (G(L)->totalbytes >= G(L)->GCthreshold) \
|
|
+ luaC_step(L); }
|
|
+
|
|
+
|
|
+#define luaC_barrier(L,p,v) { if (valiswhite(v) && isblack(obj2gco(p))) \
|
|
+ luaC_barrierf(L,obj2gco(p),gcvalue(v)); }
|
|
+
|
|
+#define luaC_barriert(L,t,v) { if (valiswhite(v) && isblack(obj2gco(t))) \
|
|
+ luaC_barrierback(L,t); }
|
|
+
|
|
+#define luaC_objbarrier(L,p,o) \
|
|
+ { if (iswhite(obj2gco(o)) && isblack(obj2gco(p))) \
|
|
+ luaC_barrierf(L,obj2gco(p),obj2gco(o)); }
|
|
+
|
|
+#define luaC_objbarriert(L,t,o) \
|
|
+ { if (iswhite(obj2gco(o)) && isblack(obj2gco(t))) luaC_barrierback(L,t); }
|
|
+
|
|
+LUAI_FUNC size_t luaC_separateudata (lua_State *L, int all);
|
|
+LUAI_FUNC void luaC_callGCTM (lua_State *L);
|
|
+LUAI_FUNC void luaC_freeall (lua_State *L);
|
|
+LUAI_FUNC void luaC_step (lua_State *L);
|
|
+LUAI_FUNC void luaC_fullgc (lua_State *L);
|
|
+LUAI_FUNC void luaC_link (lua_State *L, GCObject *o, lu_byte tt);
|
|
+LUAI_FUNC void luaC_linkupval (lua_State *L, UpVal *uv);
|
|
+LUAI_FUNC void luaC_barrierf (lua_State *L, GCObject *o, GCObject *v);
|
|
+LUAI_FUNC void luaC_barrierback (lua_State *L, Table *t);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/llex.c
|
|
@@ -0,0 +1,460 @@
|
|
+/*
|
|
+** $Id: llex.c,v 2.20.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lexical Analyzer
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <ctype.h>
|
|
+#include <locale.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define llex_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldo.h"
|
|
+#include "llex.h"
|
|
+#include "lobject.h"
|
|
+#include "lparser.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+
|
|
+#define next(ls) (ls->current = zgetc(ls->z))
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+#define currIsNewline(ls) (ls->current == '\n' || ls->current == '\r')
|
|
+
|
|
+
|
|
+/* ORDER RESERVED */
|
|
+const char *const luaX_tokens [] = {
|
|
+ "and", "break", "do", "else", "elseif",
|
|
+ "end", "false", "for", "function", "if",
|
|
+ "in", "local", "nil", "not", "or", "repeat",
|
|
+ "return", "then", "true", "until", "while",
|
|
+ "..", "...", "==", ">=", "<=", "~=",
|
|
+ "<number>", "<name>", "<string>", "<eof>",
|
|
+ NULL
|
|
+};
|
|
+
|
|
+
|
|
+#define save_and_next(ls) (save(ls, ls->current), next(ls))
|
|
+
|
|
+
|
|
+static void save (LexState *ls, int c) {
|
|
+ Mbuffer *b = ls->buff;
|
|
+ if (b->n + 1 > b->buffsize) {
|
|
+ size_t newsize;
|
|
+ if (b->buffsize >= MAX_SIZET/2)
|
|
+ luaX_lexerror(ls, "lexical element too long", 0);
|
|
+ newsize = b->buffsize * 2;
|
|
+ luaZ_resizebuffer(ls->L, b, newsize);
|
|
+ }
|
|
+ b->buffer[b->n++] = cast(char, c);
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_init (lua_State *L) {
|
|
+ int i;
|
|
+ for (i=0; i<NUM_RESERVED; i++) {
|
|
+ TString *ts = luaS_new(L, luaX_tokens[i]);
|
|
+ luaS_fix(ts); /* reserved words are never collected */
|
|
+ lua_assert(strlen(luaX_tokens[i])+1 <= TOKEN_LEN);
|
|
+ ts->tsv.reserved = cast_byte(i+1); /* reserved word */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+#define MAXSRC 80
|
|
+
|
|
+
|
|
+const char *luaX_token2str (LexState *ls, int token) {
|
|
+ if (token < FIRST_RESERVED) {
|
|
+ lua_assert(token == cast(unsigned char, token));
|
|
+ return (iscntrl(token)) ? luaO_pushfstring(ls->L, "char(%d)", token) :
|
|
+ luaO_pushfstring(ls->L, "%c", token);
|
|
+ }
|
|
+ else
|
|
+ return luaX_tokens[token-FIRST_RESERVED];
|
|
+}
|
|
+
|
|
+
|
|
+static const char *txtToken (LexState *ls, int token) {
|
|
+ switch (token) {
|
|
+ case TK_NAME:
|
|
+ case TK_STRING:
|
|
+ case TK_NUMBER:
|
|
+ save(ls, '\0');
|
|
+ return luaZ_buffer(ls->buff);
|
|
+ default:
|
|
+ return luaX_token2str(ls, token);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_lexerror (LexState *ls, const char *msg, int token) {
|
|
+ char buff[MAXSRC];
|
|
+ luaO_chunkid(buff, getstr(ls->source), MAXSRC);
|
|
+ msg = luaO_pushfstring(ls->L, "%s:%d: %s", buff, ls->linenumber, msg);
|
|
+ if (token)
|
|
+ luaO_pushfstring(ls->L, "%s near " LUA_QS, msg, txtToken(ls, token));
|
|
+ luaD_throw(ls->L, LUA_ERRSYNTAX);
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_syntaxerror (LexState *ls, const char *msg) {
|
|
+ luaX_lexerror(ls, msg, ls->t.token);
|
|
+}
|
|
+
|
|
+
|
|
+TString *luaX_newstring (LexState *ls, const char *str, size_t l) {
|
|
+ lua_State *L = ls->L;
|
|
+ TString *ts = luaS_newlstr(L, str, l);
|
|
+ TValue *o = luaH_setstr(L, ls->fs->h, ts); /* entry for `str' */
|
|
+ if (ttisnil(o))
|
|
+ setbvalue(o, 1); /* make sure `str' will not be collected */
|
|
+ return ts;
|
|
+}
|
|
+
|
|
+
|
|
+static void inclinenumber (LexState *ls) {
|
|
+ int old = ls->current;
|
|
+ lua_assert(currIsNewline(ls));
|
|
+ next(ls); /* skip `\n' or `\r' */
|
|
+ if (currIsNewline(ls) && ls->current != old)
|
|
+ next(ls); /* skip `\n\r' or `\r\n' */
|
|
+ if (++ls->linenumber >= MAX_INT)
|
|
+ luaX_syntaxerror(ls, "chunk has too many lines");
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_setinput (lua_State *L, LexState *ls, ZIO *z, TString *source) {
|
|
+ ls->decpoint = '.';
|
|
+ ls->L = L;
|
|
+ ls->lookahead.token = TK_EOS; /* no look-ahead token */
|
|
+ ls->z = z;
|
|
+ ls->fs = NULL;
|
|
+ ls->linenumber = 1;
|
|
+ ls->lastline = 1;
|
|
+ ls->source = source;
|
|
+ luaZ_resizebuffer(ls->L, ls->buff, LUA_MINBUFFER); /* initialize buffer */
|
|
+ next(ls); /* read first char */
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** =======================================================
|
|
+** LEXICAL ANALYZER
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+
|
|
+static int check_next (LexState *ls, const char *set) {
|
|
+ if (!strchr(set, ls->current))
|
|
+ return 0;
|
|
+ save_and_next(ls);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void buffreplace (LexState *ls, char from, char to) {
|
|
+ size_t n = luaZ_bufflen(ls->buff);
|
|
+ char *p = luaZ_buffer(ls->buff);
|
|
+ while (n--)
|
|
+ if (p[n] == from) p[n] = to;
|
|
+}
|
|
+
|
|
+
|
|
+static void trydecpoint (LexState *ls, SemInfo *seminfo) {
|
|
+ /* format error: try to update decimal point separator */
|
|
+ char old = ls->decpoint;
|
|
+ struct lconv *cv = localeconv();
|
|
+ ls->decpoint = (cv ? cv->decimal_point[0] : '.');
|
|
+ buffreplace(ls, old, ls->decpoint); /* try updated decimal separator */
|
|
+ if (!luaO_str2d(luaZ_buffer(ls->buff), &seminfo->r)) {
|
|
+ /* format error with correct decimal point: no more options */
|
|
+ buffreplace(ls, ls->decpoint, '.'); /* undo change (for error message) */
|
|
+ luaX_lexerror(ls, "malformed number", TK_NUMBER);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/* LUA_NUMBER */
|
|
+static void read_numeral (LexState *ls, SemInfo *seminfo) {
|
|
+ lua_assert(isdigit(ls->current));
|
|
+ do {
|
|
+ save_and_next(ls);
|
|
+ } while (isdigit(ls->current) || ls->current == '.');
|
|
+ if (check_next(ls, "Ee")) /* `E'? */
|
|
+ check_next(ls, "+-"); /* optional exponent sign */
|
|
+ while (isalnum(ls->current) || ls->current == '_')
|
|
+ save_and_next(ls);
|
|
+ save(ls, '\0');
|
|
+ buffreplace(ls, '.', ls->decpoint); /* follow locale for decimal point */
|
|
+ if (!luaO_str2d(luaZ_buffer(ls->buff), &seminfo->r)) /* format error? */
|
|
+ trydecpoint(ls, seminfo); /* try to update decimal point separator */
|
|
+}
|
|
+
|
|
+
|
|
+static int skip_sep (LexState *ls) {
|
|
+ int count = 0;
|
|
+ int s = ls->current;
|
|
+ lua_assert(s == '[' || s == ']');
|
|
+ save_and_next(ls);
|
|
+ while (ls->current == '=') {
|
|
+ save_and_next(ls);
|
|
+ count++;
|
|
+ }
|
|
+ return (ls->current == s) ? count : (-count) - 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void read_long_string (LexState *ls, SemInfo *seminfo, int sep) {
|
|
+ int cont = 0;
|
|
+ (void)(cont); /* avoid warnings when `cont' is not used */
|
|
+ save_and_next(ls); /* skip 2nd `[' */
|
|
+ if (currIsNewline(ls)) /* string starts with a newline? */
|
|
+ inclinenumber(ls); /* skip it */
|
|
+ for (;;) {
|
|
+ switch (ls->current) {
|
|
+ case EOZ:
|
|
+ luaX_lexerror(ls, (seminfo) ? "unfinished long string" :
|
|
+ "unfinished long comment", TK_EOS);
|
|
+ break; /* to avoid warnings */
|
|
+#if defined(LUA_COMPAT_LSTR)
|
|
+ case '[': {
|
|
+ if (skip_sep(ls) == sep) {
|
|
+ save_and_next(ls); /* skip 2nd `[' */
|
|
+ cont++;
|
|
+#if LUA_COMPAT_LSTR == 1
|
|
+ if (sep == 0)
|
|
+ luaX_lexerror(ls, "nesting of [[...]] is deprecated", '[');
|
|
+#endif
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+#endif
|
|
+ case ']': {
|
|
+ if (skip_sep(ls) == sep) {
|
|
+ save_and_next(ls); /* skip 2nd `]' */
|
|
+#if defined(LUA_COMPAT_LSTR) && LUA_COMPAT_LSTR == 2
|
|
+ cont--;
|
|
+ if (sep == 0 && cont >= 0) break;
|
|
+#endif
|
|
+ goto endloop;
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ case '\n':
|
|
+ case '\r': {
|
|
+ save(ls, '\n');
|
|
+ inclinenumber(ls);
|
|
+ if (!seminfo) luaZ_resetbuffer(ls->buff); /* avoid wasting space */
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ if (seminfo) save_and_next(ls);
|
|
+ else next(ls);
|
|
+ }
|
|
+ }
|
|
+ } endloop:
|
|
+ if (seminfo)
|
|
+ seminfo->ts = luaX_newstring(ls, luaZ_buffer(ls->buff) + (2 + sep),
|
|
+ luaZ_bufflen(ls->buff) - 2*(2 + sep));
|
|
+}
|
|
+
|
|
+
|
|
+static void read_string (LexState *ls, int del, SemInfo *seminfo) {
|
|
+ save_and_next(ls);
|
|
+ while (ls->current != del) {
|
|
+ switch (ls->current) {
|
|
+ case EOZ:
|
|
+ luaX_lexerror(ls, "unfinished string", TK_EOS);
|
|
+ continue; /* to avoid warnings */
|
|
+ case '\n':
|
|
+ case '\r':
|
|
+ luaX_lexerror(ls, "unfinished string", TK_STRING);
|
|
+ continue; /* to avoid warnings */
|
|
+ case '\\': {
|
|
+ int c;
|
|
+ next(ls); /* do not save the `\' */
|
|
+ switch (ls->current) {
|
|
+ case 'a': c = '\a'; break;
|
|
+ case 'b': c = '\b'; break;
|
|
+ case 'f': c = '\f'; break;
|
|
+ case 'n': c = '\n'; break;
|
|
+ case 'r': c = '\r'; break;
|
|
+ case 't': c = '\t'; break;
|
|
+ case 'v': c = '\v'; break;
|
|
+ case '\n': /* go through */
|
|
+ case '\r': save(ls, '\n'); inclinenumber(ls); continue;
|
|
+ case EOZ: continue; /* will raise an error next loop */
|
|
+ default: {
|
|
+ if (!isdigit(ls->current))
|
|
+ save_and_next(ls); /* handles \\, \", \', and \? */
|
|
+ else { /* \xxx */
|
|
+ int i = 0;
|
|
+ c = 0;
|
|
+ do {
|
|
+ c = 10*c + (ls->current-'0');
|
|
+ next(ls);
|
|
+ } while (++i<3 && isdigit(ls->current));
|
|
+ if (c > UCHAR_MAX)
|
|
+ luaX_lexerror(ls, "escape sequence too large", TK_STRING);
|
|
+ save(ls, c);
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ }
|
|
+ save(ls, c);
|
|
+ next(ls);
|
|
+ continue;
|
|
+ }
|
|
+ default:
|
|
+ save_and_next(ls);
|
|
+ }
|
|
+ }
|
|
+ save_and_next(ls); /* skip delimiter */
|
|
+ seminfo->ts = luaX_newstring(ls, luaZ_buffer(ls->buff) + 1,
|
|
+ luaZ_bufflen(ls->buff) - 2);
|
|
+}
|
|
+
|
|
+
|
|
+static int llex (LexState *ls, SemInfo *seminfo) {
|
|
+ luaZ_resetbuffer(ls->buff);
|
|
+ for (;;) {
|
|
+ switch (ls->current) {
|
|
+ case '\n':
|
|
+ case '\r': {
|
|
+ inclinenumber(ls);
|
|
+ continue;
|
|
+ }
|
|
+ case '-': {
|
|
+ next(ls);
|
|
+ if (ls->current != '-') return '-';
|
|
+ /* else is a comment */
|
|
+ next(ls);
|
|
+ if (ls->current == '[') {
|
|
+ int sep = skip_sep(ls);
|
|
+ luaZ_resetbuffer(ls->buff); /* `skip_sep' may dirty the buffer */
|
|
+ if (sep >= 0) {
|
|
+ read_long_string(ls, NULL, sep); /* long comment */
|
|
+ luaZ_resetbuffer(ls->buff);
|
|
+ continue;
|
|
+ }
|
|
+ }
|
|
+ /* else short comment */
|
|
+ while (!currIsNewline(ls) && ls->current != EOZ)
|
|
+ next(ls);
|
|
+ continue;
|
|
+ }
|
|
+ case '[': {
|
|
+ int sep = skip_sep(ls);
|
|
+ if (sep >= 0) {
|
|
+ read_long_string(ls, seminfo, sep);
|
|
+ return TK_STRING;
|
|
+ }
|
|
+ else if (sep == -1) return '[';
|
|
+ else luaX_lexerror(ls, "invalid long string delimiter", TK_STRING);
|
|
+ }
|
|
+ case '=': {
|
|
+ next(ls);
|
|
+ if (ls->current != '=') return '=';
|
|
+ else { next(ls); return TK_EQ; }
|
|
+ }
|
|
+ case '<': {
|
|
+ next(ls);
|
|
+ if (ls->current != '=') return '<';
|
|
+ else { next(ls); return TK_LE; }
|
|
+ }
|
|
+ case '>': {
|
|
+ next(ls);
|
|
+ if (ls->current != '=') return '>';
|
|
+ else { next(ls); return TK_GE; }
|
|
+ }
|
|
+ case '~': {
|
|
+ next(ls);
|
|
+ if (ls->current != '=') return '~';
|
|
+ else { next(ls); return TK_NE; }
|
|
+ }
|
|
+ case '"':
|
|
+ case '\'': {
|
|
+ read_string(ls, ls->current, seminfo);
|
|
+ return TK_STRING;
|
|
+ }
|
|
+ case '.': {
|
|
+ save_and_next(ls);
|
|
+ if (check_next(ls, ".")) {
|
|
+ if (check_next(ls, "."))
|
|
+ return TK_DOTS; /* ... */
|
|
+ else return TK_CONCAT; /* .. */
|
|
+ }
|
|
+ else if (!isdigit(ls->current)) return '.';
|
|
+ else {
|
|
+ read_numeral(ls, seminfo);
|
|
+ return TK_NUMBER;
|
|
+ }
|
|
+ }
|
|
+ case EOZ: {
|
|
+ return TK_EOS;
|
|
+ }
|
|
+ default: {
|
|
+ if (isspace(ls->current)) {
|
|
+ lua_assert(!currIsNewline(ls));
|
|
+ next(ls);
|
|
+ continue;
|
|
+ }
|
|
+ else if (isdigit(ls->current)) {
|
|
+ read_numeral(ls, seminfo);
|
|
+ return TK_NUMBER;
|
|
+ }
|
|
+ else if (isalpha(ls->current) || ls->current == '_') {
|
|
+ /* identifier or reserved word */
|
|
+ TString *ts;
|
|
+ do {
|
|
+ save_and_next(ls);
|
|
+ } while (isalnum(ls->current) || ls->current == '_');
|
|
+ ts = luaX_newstring(ls, luaZ_buffer(ls->buff),
|
|
+ luaZ_bufflen(ls->buff));
|
|
+ if (ts->tsv.reserved > 0) /* reserved word? */
|
|
+ return ts->tsv.reserved - 1 + FIRST_RESERVED;
|
|
+ else {
|
|
+ seminfo->ts = ts;
|
|
+ return TK_NAME;
|
|
+ }
|
|
+ }
|
|
+ else {
|
|
+ int c = ls->current;
|
|
+ next(ls);
|
|
+ return c; /* single-char tokens (+ - / ...) */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_next (LexState *ls) {
|
|
+ ls->lastline = ls->linenumber;
|
|
+ if (ls->lookahead.token != TK_EOS) { /* is there a look-ahead token? */
|
|
+ ls->t = ls->lookahead; /* use this one */
|
|
+ ls->lookahead.token = TK_EOS; /* and discharge it */
|
|
+ }
|
|
+ else
|
|
+ ls->t.token = llex(ls, &ls->t.seminfo); /* read next token */
|
|
+}
|
|
+
|
|
+
|
|
+void luaX_lookahead (LexState *ls) {
|
|
+ lua_assert(ls->lookahead.token == TK_EOS);
|
|
+ ls->lookahead.token = llex(ls, &ls->lookahead.seminfo);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/llex.h
|
|
@@ -0,0 +1,81 @@
|
|
+/*
|
|
+** $Id: llex.h,v 1.58.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lexical Analyzer
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef llex_h
|
|
+#define llex_h
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+#define FIRST_RESERVED 257
|
|
+
|
|
+/* maximum length of a reserved word */
|
|
+#define TOKEN_LEN (sizeof("function")/sizeof(char))
|
|
+
|
|
+
|
|
+/*
|
|
+* WARNING: if you change the order of this enumeration,
|
|
+* grep "ORDER RESERVED"
|
|
+*/
|
|
+enum RESERVED {
|
|
+ /* terminal symbols denoted by reserved words */
|
|
+ TK_AND = FIRST_RESERVED, TK_BREAK,
|
|
+ TK_DO, TK_ELSE, TK_ELSEIF, TK_END, TK_FALSE, TK_FOR, TK_FUNCTION,
|
|
+ TK_IF, TK_IN, TK_LOCAL, TK_NIL, TK_NOT, TK_OR, TK_REPEAT,
|
|
+ TK_RETURN, TK_THEN, TK_TRUE, TK_UNTIL, TK_WHILE,
|
|
+ /* other terminal symbols */
|
|
+ TK_CONCAT, TK_DOTS, TK_EQ, TK_GE, TK_LE, TK_NE, TK_NUMBER,
|
|
+ TK_NAME, TK_STRING, TK_EOS
|
|
+};
|
|
+
|
|
+/* number of reserved words */
|
|
+#define NUM_RESERVED (cast(int, TK_WHILE-FIRST_RESERVED+1))
|
|
+
|
|
+
|
|
+/* array with token `names' */
|
|
+LUAI_DATA const char *const luaX_tokens [];
|
|
+
|
|
+
|
|
+typedef union {
|
|
+ lua_Number r;
|
|
+ TString *ts;
|
|
+} SemInfo; /* semantics information */
|
|
+
|
|
+
|
|
+typedef struct Token {
|
|
+ int token;
|
|
+ SemInfo seminfo;
|
|
+} Token;
|
|
+
|
|
+
|
|
+typedef struct LexState {
|
|
+ int current; /* current character (charint) */
|
|
+ int linenumber; /* input line counter */
|
|
+ int lastline; /* line of last token `consumed' */
|
|
+ Token t; /* current token */
|
|
+ Token lookahead; /* look ahead token */
|
|
+ struct FuncState *fs; /* `FuncState' is private to the parser */
|
|
+ struct lua_State *L;
|
|
+ ZIO *z; /* input stream */
|
|
+ Mbuffer *buff; /* buffer for tokens */
|
|
+ TString *source; /* current source name */
|
|
+ char decpoint; /* locale decimal point */
|
|
+} LexState;
|
|
+
|
|
+
|
|
+LUAI_FUNC void luaX_init (lua_State *L);
|
|
+LUAI_FUNC void luaX_setinput (lua_State *L, LexState *ls, ZIO *z,
|
|
+ TString *source);
|
|
+LUAI_FUNC TString *luaX_newstring (LexState *ls, const char *str, size_t l);
|
|
+LUAI_FUNC void luaX_next (LexState *ls);
|
|
+LUAI_FUNC void luaX_lookahead (LexState *ls);
|
|
+LUAI_FUNC void luaX_lexerror (LexState *ls, const char *msg, int token);
|
|
+LUAI_FUNC void luaX_syntaxerror (LexState *ls, const char *s);
|
|
+LUAI_FUNC const char *luaX_token2str (LexState *ls, int token);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/llimits.h
|
|
@@ -0,0 +1,125 @@
|
|
+/*
|
|
+** $Id: llimits.h,v 1.69.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Limits, basic types, and some other `installation-dependent' definitions
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef llimits_h
|
|
+#define llimits_h
|
|
+
|
|
+#include <stddef.h>
|
|
+#include <limits.h>
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+typedef LUAI_UINT32 lu_int32;
|
|
+
|
|
+typedef LUAI_UMEM lu_mem;
|
|
+
|
|
+typedef LUAI_MEM l_mem;
|
|
+
|
|
+
|
|
+
|
|
+/* chars used as small naturals (so that `char' is reserved for characters) */
|
|
+typedef unsigned char lu_byte;
|
|
+
|
|
+
|
|
+#define MAX_SIZET ((size_t)(~(size_t)0)-2)
|
|
+
|
|
+#define MAX_LUMEM ((lu_mem)(~(lu_mem)0)-2)
|
|
+
|
|
+
|
|
+#define MAX_INT (LUA_INT_MAX-2) /* maximum value of an int (-2 for safety) */
|
|
+
|
|
+/*
|
|
+** conversion of pointer to integer
|
|
+** this is for hashing only; there is no problem if the integer
|
|
+** cannot hold the whole pointer value
|
|
+*/
|
|
+#define IntPoint(p) ((unsigned int)(lu_mem)(p))
|
|
+
|
|
+
|
|
+
|
|
+/* type to ensure maximum alignment */
|
|
+typedef LUAI_USER_ALIGNMENT_T L_Umaxalign;
|
|
+
|
|
+
|
|
+/* result of a `usual argument conversion' over lua_Number */
|
|
+typedef LUAI_UACNUMBER l_uacNumber;
|
|
+
|
|
+
|
|
+/* internal assertions for in-house debugging */
|
|
+#ifdef lua_assert
|
|
+
|
|
+#define check_exp(c,e) (lua_assert(c), (e))
|
|
+#define api_check(l,e) lua_assert(e)
|
|
+
|
|
+#else
|
|
+
|
|
+#define lua_assert(c) ((void)0)
|
|
+#define check_exp(c,e) (e)
|
|
+#define api_check luai_apicheck
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
+#ifndef UNUSED
|
|
+#define UNUSED(x) ((void)(x)) /* to avoid warnings */
|
|
+#endif
|
|
+
|
|
+
|
|
+#ifndef cast
|
|
+#define cast(t, exp) ((t)(exp))
|
|
+#endif
|
|
+
|
|
+#define cast_byte(i) cast(lu_byte, (i))
|
|
+#define cast_num(i) cast(lua_Number, (i))
|
|
+#define cast_int(i) cast(int, (i))
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** type for virtual-machine instructions
|
|
+** must be an unsigned with (at least) 4 bytes (see details in lopcodes.h)
|
|
+*/
|
|
+typedef lu_int32 Instruction;
|
|
+
|
|
+
|
|
+
|
|
+/* maximum stack for a Lua function */
|
|
+#define MAXSTACK 250
|
|
+
|
|
+
|
|
+
|
|
+/* minimum size for the string table (must be power of 2) */
|
|
+#ifndef MINSTRTABSIZE
|
|
+#define MINSTRTABSIZE 32
|
|
+#endif
|
|
+
|
|
+
|
|
+/* minimum size for string buffer */
|
|
+#ifndef LUA_MINBUFFER
|
|
+#define LUA_MINBUFFER 32
|
|
+#endif
|
|
+
|
|
+
|
|
+#ifndef lua_lock
|
|
+#define lua_lock(L) ((void) 0)
|
|
+#define lua_unlock(L) ((void) 0)
|
|
+#endif
|
|
+
|
|
+#ifndef luai_threadyield
|
|
+#define luai_threadyield(L) {lua_unlock(L); lua_lock(L);}
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+** macro to control inclusion of some hard tests on stack reallocation
|
|
+*/
|
|
+#ifndef HARDSTACKTESTS
|
|
+#define condhardstacktests(x) ((void)0)
|
|
+#else
|
|
+#define condhardstacktests(x) x
|
|
+#endif
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lmem.c
|
|
@@ -0,0 +1,86 @@
|
|
+/*
|
|
+** $Id: lmem.c,v 1.70.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Interface to Memory Manager
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#define lmem_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** About the realloc function:
|
|
+** void * frealloc (void *ud, void *ptr, size_t osize, size_t nsize);
|
|
+** (`osize' is the old size, `nsize' is the new size)
|
|
+**
|
|
+** Lua ensures that (ptr == NULL) iff (osize == 0).
|
|
+**
|
|
+** * frealloc(ud, NULL, 0, x) creates a new block of size `x'
|
|
+**
|
|
+** * frealloc(ud, p, x, 0) frees the block `p'
|
|
+** (in this specific case, frealloc must return NULL).
|
|
+** particularly, frealloc(ud, NULL, 0, 0) does nothing
|
|
+** (which is equivalent to free(NULL) in ANSI C)
|
|
+**
|
|
+** frealloc returns NULL if it cannot create or reallocate the area
|
|
+** (any reallocation to an equal or smaller size cannot fail!)
|
|
+*/
|
|
+
|
|
+
|
|
+
|
|
+#define MINSIZEARRAY 4
|
|
+
|
|
+
|
|
+void *luaM_growaux_ (lua_State *L, void *block, int *size, size_t size_elems,
|
|
+ int limit, const char *errormsg) {
|
|
+ void *newblock;
|
|
+ int newsize;
|
|
+ if (*size >= limit/2) { /* cannot double it? */
|
|
+ if (*size >= limit) /* cannot grow even a little? */
|
|
+ luaG_runerror(L, errormsg);
|
|
+ newsize = limit; /* still have at least one free place */
|
|
+ }
|
|
+ else {
|
|
+ newsize = (*size)*2;
|
|
+ if (newsize < MINSIZEARRAY)
|
|
+ newsize = MINSIZEARRAY; /* minimum size */
|
|
+ }
|
|
+ newblock = luaM_reallocv(L, block, *size, newsize, size_elems);
|
|
+ *size = newsize; /* update only when everything else is OK */
|
|
+ return newblock;
|
|
+}
|
|
+
|
|
+
|
|
+void *luaM_toobig (lua_State *L) {
|
|
+ luaG_runerror(L, "memory allocation error: block too big");
|
|
+ return NULL; /* to avoid warnings */
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** generic allocation routine.
|
|
+*/
|
|
+void *luaM_realloc_ (lua_State *L, void *block, size_t osize, size_t nsize) {
|
|
+ global_State *g = G(L);
|
|
+ lua_assert((osize == 0) == (block == NULL));
|
|
+ block = (*g->frealloc)(g->ud, block, osize, nsize);
|
|
+ if (block == NULL && nsize > 0)
|
|
+ luaD_throw(L, LUA_ERRMEM);
|
|
+ lua_assert((nsize == 0) == (block == NULL));
|
|
+ g->totalbytes = (g->totalbytes - osize) + nsize;
|
|
+ return block;
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lmem.h
|
|
@@ -0,0 +1,49 @@
|
|
+/*
|
|
+** $Id: lmem.h,v 1.31.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Interface to Memory Manager
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lmem_h
|
|
+#define lmem_h
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#include "llimits.h"
|
|
+#include "lua.h"
|
|
+
|
|
+#define MEMERRMSG "not enough memory"
|
|
+
|
|
+
|
|
+#define luaM_reallocv(L,b,on,n,e) \
|
|
+ ((cast(size_t, (n)+1) <= MAX_SIZET/(e)) ? /* +1 to avoid warnings */ \
|
|
+ luaM_realloc_(L, (b), (on)*(e), (n)*(e)) : \
|
|
+ luaM_toobig(L))
|
|
+
|
|
+#define luaM_freemem(L, b, s) luaM_realloc_(L, (b), (s), 0)
|
|
+#define luaM_free(L, b) luaM_realloc_(L, (b), sizeof(*(b)), 0)
|
|
+#define luaM_freearray(L, b, n, t) luaM_reallocv(L, (b), n, 0, sizeof(t))
|
|
+
|
|
+#define luaM_malloc(L,t) luaM_realloc_(L, NULL, 0, (t))
|
|
+#define luaM_new(L,t) cast(t *, luaM_malloc(L, sizeof(t)))
|
|
+#define luaM_newvector(L,n,t) \
|
|
+ cast(t *, luaM_reallocv(L, NULL, 0, n, sizeof(t)))
|
|
+
|
|
+#define luaM_growvector(L,v,nelems,size,t,limit,e) \
|
|
+ if ((nelems)+1 > (size)) \
|
|
+ ((v)=cast(t *, luaM_growaux_(L,v,&(size),sizeof(t),limit,e)))
|
|
+
|
|
+#define luaM_reallocvector(L, v,oldn,n,t) \
|
|
+ ((v)=cast(t *, luaM_reallocv(L, v, oldn, n, sizeof(t))))
|
|
+
|
|
+
|
|
+LUAI_FUNC void *luaM_realloc_ (lua_State *L, void *block, size_t oldsize,
|
|
+ size_t size);
|
|
+LUAI_FUNC void *luaM_toobig (lua_State *L);
|
|
+LUAI_FUNC void *luaM_growaux_ (lua_State *L, void *block, int *size,
|
|
+ size_t size_elem, int limit,
|
|
+ const char *errormsg);
|
|
+
|
|
+#endif
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lobject.c
|
|
@@ -0,0 +1,215 @@
|
|
+/*
|
|
+** $Id: lobject.c,v 2.22.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Some generic functions over Lua objects
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stdarg.h>
|
|
+
|
|
+#include <ctype.h>
|
|
+#include <stdio.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define lobject_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldo.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "lvm.h"
|
|
+
|
|
+
|
|
+
|
|
+const TValue luaO_nilobject_ = {{NULL}, LUA_TNIL};
|
|
+
|
|
+
|
|
+/*
|
|
+** converts an integer to a "floating point byte", represented as
|
|
+** (eeeeexxx), where the real value is (1xxx) * 2^(eeeee - 1) if
|
|
+** eeeee != 0 and (xxx) otherwise.
|
|
+*/
|
|
+int luaO_int2fb (unsigned int x) {
|
|
+ int e = 0; /* expoent */
|
|
+ while (x >= 16) {
|
|
+ x = (x+1) >> 1;
|
|
+ e++;
|
|
+ }
|
|
+ if (x < 8) return x;
|
|
+ else return ((e+1) << 3) | (cast_int(x) - 8);
|
|
+}
|
|
+
|
|
+
|
|
+/* converts back */
|
|
+int luaO_fb2int (int x) {
|
|
+ int e = (x >> 3) & 31;
|
|
+ if (e == 0) return x;
|
|
+ else return ((x & 7)+8) << (e - 1);
|
|
+}
|
|
+
|
|
+
|
|
+int luaO_log2 (unsigned int x) {
|
|
+ static const lu_byte log_2[256] = {
|
|
+ 0,1,2,2,3,3,3,3,4,4,4,4,4,4,4,4,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,
|
|
+ 6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
|
|
+ 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
|
|
+ 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
|
|
+ 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
|
|
+ 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
|
|
+ 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
|
|
+ 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8
|
|
+ };
|
|
+ int l = -1;
|
|
+ while (x >= 256) { l += 8; x >>= 8; }
|
|
+ return l + log_2[x];
|
|
+
|
|
+}
|
|
+
|
|
+
|
|
+int luaO_rawequalObj (const TValue *t1, const TValue *t2) {
|
|
+ if (ttype(t1) != ttype(t2)) return 0;
|
|
+ else switch (ttype(t1)) {
|
|
+ case LUA_TNIL:
|
|
+ return 1;
|
|
+ case LUA_TNUMBER:
|
|
+ return luai_numeq(nvalue(t1), nvalue(t2));
|
|
+ case LUA_TBOOLEAN:
|
|
+ return bvalue(t1) == bvalue(t2); /* boolean true must be 1 !! */
|
|
+ case LUA_TLIGHTUSERDATA:
|
|
+ return pvalue(t1) == pvalue(t2);
|
|
+ default:
|
|
+ lua_assert(iscollectable(t1));
|
|
+ return gcvalue(t1) == gcvalue(t2);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+int luaO_str2d (const char *s, lua_Number *result) {
|
|
+ char *endptr;
|
|
+ *result = lua_str2number(s, &endptr);
|
|
+ if (endptr == s) return 0; /* conversion failed */
|
|
+ if (*endptr == 'x' || *endptr == 'X') /* maybe an hexadecimal constant? */
|
|
+ *result = cast_num(strtoul(s, &endptr, 16));
|
|
+ if (*endptr == '\0') return 1; /* most common case */
|
|
+ while (isspace(cast(unsigned char, *endptr))) endptr++;
|
|
+ if (*endptr != '\0') return 0; /* invalid trailing characters? */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+static void pushstr (lua_State *L, const char *str) {
|
|
+ setsvalue2s(L, L->top, luaS_new(L, str));
|
|
+ incr_top(L);
|
|
+}
|
|
+
|
|
+
|
|
+/* this function handles only `%d', `%c', %f, %p, and `%s' formats */
|
|
+const char *luaO_pushvfstring (lua_State *L, const char *fmt, va_list argp) {
|
|
+ int n = 1;
|
|
+ pushstr(L, "");
|
|
+ for (;;) {
|
|
+ const char *e = strchr(fmt, '%');
|
|
+ if (e == NULL) break;
|
|
+ setsvalue2s(L, L->top, luaS_newlstr(L, fmt, e-fmt));
|
|
+ incr_top(L);
|
|
+ switch (*(e+1)) {
|
|
+ case 's': {
|
|
+ const char *s = va_arg(argp, char *);
|
|
+ if (s == NULL) s = "(null)";
|
|
+ pushstr(L, s);
|
|
+ break;
|
|
+ }
|
|
+ case 'c': {
|
|
+ char buff[2];
|
|
+ buff[0] = cast(char, va_arg(argp, int));
|
|
+ buff[1] = '\0';
|
|
+ pushstr(L, buff);
|
|
+ break;
|
|
+ }
|
|
+ case 'd': {
|
|
+ setnvalue(L->top, cast_num(va_arg(argp, int)));
|
|
+ incr_top(L);
|
|
+ break;
|
|
+ }
|
|
+ case 'f': {
|
|
+ setnvalue(L->top, cast_num(va_arg(argp, l_uacNumber)));
|
|
+ incr_top(L);
|
|
+ break;
|
|
+ }
|
|
+ case 'p': {
|
|
+ char buff[4*sizeof(void *) + 8]; /* should be enough space for a `%p' */
|
|
+ sprintf(buff, "%p", va_arg(argp, void *));
|
|
+ pushstr(L, buff);
|
|
+ break;
|
|
+ }
|
|
+ case '%': {
|
|
+ pushstr(L, "%");
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ char buff[3];
|
|
+ buff[0] = '%';
|
|
+ buff[1] = *(e+1);
|
|
+ buff[2] = '\0';
|
|
+ pushstr(L, buff);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ n += 2;
|
|
+ fmt = e+2;
|
|
+ }
|
|
+ pushstr(L, fmt);
|
|
+ luaV_concat(L, n+1, cast_int(L->top - L->base) - 1);
|
|
+ L->top -= n;
|
|
+ return svalue(L->top - 1);
|
|
+}
|
|
+
|
|
+
|
|
+const char *luaO_pushfstring (lua_State *L, const char *fmt, ...) {
|
|
+ const char *msg;
|
|
+ va_list argp;
|
|
+ va_start(argp, fmt);
|
|
+ msg = luaO_pushvfstring(L, fmt, argp);
|
|
+ va_end(argp);
|
|
+ return msg;
|
|
+}
|
|
+
|
|
+
|
|
+void luaO_chunkid (char *out, const char *source, size_t bufflen) {
|
|
+ if (*source == '=') {
|
|
+ strncpy(out, source+1, bufflen); /* remove first char */
|
|
+ out[bufflen-1] = '\0'; /* ensures null termination */
|
|
+ }
|
|
+ else { /* out = "source", or "...source" */
|
|
+ if (*source == '@') {
|
|
+ size_t l;
|
|
+ source++; /* skip the `@' */
|
|
+ bufflen -= sizeof(" '...' ");
|
|
+ l = strlen(source);
|
|
+ strcpy(out, "");
|
|
+ if (l > bufflen) {
|
|
+ source += (l-bufflen); /* get last part of file name */
|
|
+ strcat(out, "...");
|
|
+ }
|
|
+ strcat(out, source);
|
|
+ }
|
|
+ else { /* out = [string "string"] */
|
|
+ size_t len = strcspn(source, "\n\r"); /* stop at first newline */
|
|
+ bufflen -= sizeof(" [string \"...\"] ");
|
|
+ if (len > bufflen) len = bufflen;
|
|
+ strcpy(out, "[string \"");
|
|
+ if (source[len] != '\0') { /* must truncate? */
|
|
+ strncat(out, source, len);
|
|
+ strcat(out, "...");
|
|
+ }
|
|
+ else
|
|
+ strcat(out, source);
|
|
+ strcat(out, "\"]");
|
|
+ }
|
|
+ }
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lobject.h
|
|
@@ -0,0 +1,381 @@
|
|
+/*
|
|
+** $Id: lobject.h,v 2.20.1.2 2008/08/06 13:29:48 roberto Exp $
|
|
+** Type definitions for Lua objects
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lobject_h
|
|
+#define lobject_h
|
|
+
|
|
+
|
|
+#include <stdarg.h>
|
|
+
|
|
+
|
|
+#include "llimits.h"
|
|
+#include "lua.h"
|
|
+
|
|
+
|
|
+/* tags for values visible from Lua */
|
|
+#define LAST_TAG LUA_TTHREAD
|
|
+
|
|
+#define NUM_TAGS (LAST_TAG+1)
|
|
+
|
|
+
|
|
+/*
|
|
+** Extra tags for non-values
|
|
+*/
|
|
+#define LUA_TPROTO (LAST_TAG+1)
|
|
+#define LUA_TUPVAL (LAST_TAG+2)
|
|
+#define LUA_TDEADKEY (LAST_TAG+3)
|
|
+
|
|
+
|
|
+/*
|
|
+** Union of all collectable objects
|
|
+*/
|
|
+typedef union GCObject GCObject;
|
|
+
|
|
+
|
|
+/*
|
|
+** Common Header for all collectable objects (in macro form, to be
|
|
+** included in other objects)
|
|
+*/
|
|
+#define CommonHeader GCObject *next; lu_byte tt; lu_byte marked
|
|
+
|
|
+
|
|
+/*
|
|
+** Common header in struct form
|
|
+*/
|
|
+typedef struct GCheader {
|
|
+ CommonHeader;
|
|
+} GCheader;
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Union of all Lua values
|
|
+*/
|
|
+typedef union {
|
|
+ GCObject *gc;
|
|
+ void *p;
|
|
+ lua_Number n;
|
|
+ int b;
|
|
+} Value;
|
|
+
|
|
+
|
|
+/*
|
|
+** Tagged Values
|
|
+*/
|
|
+
|
|
+#define TValuefields Value value; int tt
|
|
+
|
|
+typedef struct lua_TValue {
|
|
+ TValuefields;
|
|
+} TValue;
|
|
+
|
|
+
|
|
+/* Macros to test type */
|
|
+#define ttisnil(o) (ttype(o) == LUA_TNIL)
|
|
+#define ttisnumber(o) (ttype(o) == LUA_TNUMBER)
|
|
+#define ttisstring(o) (ttype(o) == LUA_TSTRING)
|
|
+#define ttistable(o) (ttype(o) == LUA_TTABLE)
|
|
+#define ttisfunction(o) (ttype(o) == LUA_TFUNCTION)
|
|
+#define ttisboolean(o) (ttype(o) == LUA_TBOOLEAN)
|
|
+#define ttisuserdata(o) (ttype(o) == LUA_TUSERDATA)
|
|
+#define ttisthread(o) (ttype(o) == LUA_TTHREAD)
|
|
+#define ttislightuserdata(o) (ttype(o) == LUA_TLIGHTUSERDATA)
|
|
+
|
|
+/* Macros to access values */
|
|
+#define ttype(o) ((o)->tt)
|
|
+#define gcvalue(o) check_exp(iscollectable(o), (o)->value.gc)
|
|
+#define pvalue(o) check_exp(ttislightuserdata(o), (o)->value.p)
|
|
+#define nvalue(o) check_exp(ttisnumber(o), (o)->value.n)
|
|
+#define rawtsvalue(o) check_exp(ttisstring(o), &(o)->value.gc->ts)
|
|
+#define tsvalue(o) (&rawtsvalue(o)->tsv)
|
|
+#define rawuvalue(o) check_exp(ttisuserdata(o), &(o)->value.gc->u)
|
|
+#define uvalue(o) (&rawuvalue(o)->uv)
|
|
+#define clvalue(o) check_exp(ttisfunction(o), &(o)->value.gc->cl)
|
|
+#define hvalue(o) check_exp(ttistable(o), &(o)->value.gc->h)
|
|
+#define bvalue(o) check_exp(ttisboolean(o), (o)->value.b)
|
|
+#define thvalue(o) check_exp(ttisthread(o), &(o)->value.gc->th)
|
|
+
|
|
+#define l_isfalse(o) (ttisnil(o) || (ttisboolean(o) && bvalue(o) == 0))
|
|
+
|
|
+/*
|
|
+** for internal debug only
|
|
+*/
|
|
+#define checkconsistency(obj) \
|
|
+ lua_assert(!iscollectable(obj) || (ttype(obj) == (obj)->value.gc->gch.tt))
|
|
+
|
|
+#define checkliveness(g,obj) \
|
|
+ lua_assert(!iscollectable(obj) || \
|
|
+ ((ttype(obj) == (obj)->value.gc->gch.tt) && !isdead(g, (obj)->value.gc)))
|
|
+
|
|
+
|
|
+/* Macros to set values */
|
|
+#define setnilvalue(obj) ((obj)->tt=LUA_TNIL)
|
|
+
|
|
+#define setnvalue(obj,x) \
|
|
+ { TValue *i_o=(obj); i_o->value.n=(x); i_o->tt=LUA_TNUMBER; }
|
|
+
|
|
+#define setpvalue(obj,x) \
|
|
+ { TValue *i_o=(obj); i_o->value.p=(x); i_o->tt=LUA_TLIGHTUSERDATA; }
|
|
+
|
|
+#define setbvalue(obj,x) \
|
|
+ { TValue *i_o=(obj); i_o->value.b=(x); i_o->tt=LUA_TBOOLEAN; }
|
|
+
|
|
+#define setsvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TSTRING; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+#define setuvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TUSERDATA; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+#define setthvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TTHREAD; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+#define setclvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TFUNCTION; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+#define sethvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TTABLE; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+#define setptvalue(L,obj,x) \
|
|
+ { TValue *i_o=(obj); \
|
|
+ i_o->value.gc=cast(GCObject *, (x)); i_o->tt=LUA_TPROTO; \
|
|
+ checkliveness(G(L),i_o); }
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+#define setobj(L,obj1,obj2) \
|
|
+ { const TValue *o2=(obj2); TValue *o1=(obj1); \
|
|
+ o1->value = o2->value; o1->tt=o2->tt; \
|
|
+ checkliveness(G(L),o1); }
|
|
+
|
|
+
|
|
+/*
|
|
+** different types of sets, according to destination
|
|
+*/
|
|
+
|
|
+/* from stack to (same) stack */
|
|
+#define setobjs2s setobj
|
|
+/* to stack (not from same stack) */
|
|
+#define setobj2s setobj
|
|
+#define setsvalue2s setsvalue
|
|
+#define sethvalue2s sethvalue
|
|
+#define setptvalue2s setptvalue
|
|
+/* from table to same table */
|
|
+#define setobjt2t setobj
|
|
+/* to table */
|
|
+#define setobj2t setobj
|
|
+/* to new object */
|
|
+#define setobj2n setobj
|
|
+#define setsvalue2n setsvalue
|
|
+
|
|
+#define setttype(obj, tt) (ttype(obj) = (tt))
|
|
+
|
|
+
|
|
+#define iscollectable(o) (ttype(o) >= LUA_TSTRING)
|
|
+
|
|
+
|
|
+
|
|
+typedef TValue *StkId; /* index to stack elements */
|
|
+
|
|
+
|
|
+/*
|
|
+** String headers for string table
|
|
+*/
|
|
+typedef union TString {
|
|
+ L_Umaxalign dummy; /* ensures maximum alignment for strings */
|
|
+ struct {
|
|
+ CommonHeader;
|
|
+ lu_byte reserved;
|
|
+ unsigned int hash;
|
|
+ size_t len;
|
|
+ } tsv;
|
|
+} TString;
|
|
+
|
|
+
|
|
+#define getstr(ts) cast(const char *, (ts) + 1)
|
|
+#define svalue(o) getstr(rawtsvalue(o))
|
|
+
|
|
+
|
|
+
|
|
+typedef union Udata {
|
|
+ L_Umaxalign dummy; /* ensures maximum alignment for `local' udata */
|
|
+ struct {
|
|
+ CommonHeader;
|
|
+ struct Table *metatable;
|
|
+ struct Table *env;
|
|
+ size_t len;
|
|
+ } uv;
|
|
+} Udata;
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Function Prototypes
|
|
+*/
|
|
+typedef struct Proto {
|
|
+ CommonHeader;
|
|
+ TValue *k; /* constants used by the function */
|
|
+ Instruction *code;
|
|
+ struct Proto **p; /* functions defined inside the function */
|
|
+ int *lineinfo; /* map from opcodes to source lines */
|
|
+ struct LocVar *locvars; /* information about local variables */
|
|
+ TString **upvalues; /* upvalue names */
|
|
+ TString *source;
|
|
+ int sizeupvalues;
|
|
+ int sizek; /* size of `k' */
|
|
+ int sizecode;
|
|
+ int sizelineinfo;
|
|
+ int sizep; /* size of `p' */
|
|
+ int sizelocvars;
|
|
+ int linedefined;
|
|
+ int lastlinedefined;
|
|
+ GCObject *gclist;
|
|
+ lu_byte nups; /* number of upvalues */
|
|
+ lu_byte numparams;
|
|
+ lu_byte is_vararg;
|
|
+ lu_byte maxstacksize;
|
|
+} Proto;
|
|
+
|
|
+
|
|
+/* masks for new-style vararg */
|
|
+#define VARARG_HASARG 1
|
|
+#define VARARG_ISVARARG 2
|
|
+#define VARARG_NEEDSARG 4
|
|
+
|
|
+
|
|
+typedef struct LocVar {
|
|
+ TString *varname;
|
|
+ int startpc; /* first point where variable is active */
|
|
+ int endpc; /* first point where variable is dead */
|
|
+} LocVar;
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** Upvalues
|
|
+*/
|
|
+
|
|
+typedef struct UpVal {
|
|
+ CommonHeader;
|
|
+ TValue *v; /* points to stack or to its own value */
|
|
+ union {
|
|
+ TValue value; /* the value (when closed) */
|
|
+ struct { /* double linked list (when open) */
|
|
+ struct UpVal *prev;
|
|
+ struct UpVal *next;
|
|
+ } l;
|
|
+ } u;
|
|
+} UpVal;
|
|
+
|
|
+
|
|
+/*
|
|
+** Closures
|
|
+*/
|
|
+
|
|
+#define ClosureHeader \
|
|
+ CommonHeader; lu_byte isC; lu_byte nupvalues; GCObject *gclist; \
|
|
+ struct Table *env
|
|
+
|
|
+typedef struct CClosure {
|
|
+ ClosureHeader;
|
|
+ lua_CFunction f;
|
|
+ TValue upvalue[1];
|
|
+} CClosure;
|
|
+
|
|
+
|
|
+typedef struct LClosure {
|
|
+ ClosureHeader;
|
|
+ struct Proto *p;
|
|
+ UpVal *upvals[1];
|
|
+} LClosure;
|
|
+
|
|
+
|
|
+typedef union Closure {
|
|
+ CClosure c;
|
|
+ LClosure l;
|
|
+} Closure;
|
|
+
|
|
+
|
|
+#define iscfunction(o) (ttype(o) == LUA_TFUNCTION && clvalue(o)->c.isC)
|
|
+#define isLfunction(o) (ttype(o) == LUA_TFUNCTION && !clvalue(o)->c.isC)
|
|
+
|
|
+
|
|
+/*
|
|
+** Tables
|
|
+*/
|
|
+
|
|
+typedef union TKey {
|
|
+ struct {
|
|
+ TValuefields;
|
|
+ struct Node *next; /* for chaining */
|
|
+ } nk;
|
|
+ TValue tvk;
|
|
+} TKey;
|
|
+
|
|
+
|
|
+typedef struct Node {
|
|
+ TValue i_val;
|
|
+ TKey i_key;
|
|
+} Node;
|
|
+
|
|
+
|
|
+typedef struct Table {
|
|
+ CommonHeader;
|
|
+ lu_byte flags; /* 1<<p means tagmethod(p) is not present */
|
|
+ lu_byte lsizenode; /* log2 of size of `node' array */
|
|
+ struct Table *metatable;
|
|
+ TValue *array; /* array part */
|
|
+ Node *node;
|
|
+ Node *lastfree; /* any free position is before this position */
|
|
+ GCObject *gclist;
|
|
+ int sizearray; /* size of `array' array */
|
|
+} Table;
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** `module' operation for hashing (size is always a power of 2)
|
|
+*/
|
|
+#define lmod(s,size) \
|
|
+ (check_exp((size&(size-1))==0, (cast(int, (s) & ((size)-1)))))
|
|
+
|
|
+
|
|
+#define twoto(x) (1<<(x))
|
|
+#define sizenode(t) (twoto((t)->lsizenode))
|
|
+
|
|
+
|
|
+#define luaO_nilobject (&luaO_nilobject_)
|
|
+
|
|
+LUAI_DATA const TValue luaO_nilobject_;
|
|
+
|
|
+#define ceillog2(x) (luaO_log2((x)-1) + 1)
|
|
+
|
|
+LUAI_FUNC int luaO_log2 (unsigned int x);
|
|
+LUAI_FUNC int luaO_int2fb (unsigned int x);
|
|
+LUAI_FUNC int luaO_fb2int (int x);
|
|
+LUAI_FUNC int luaO_rawequalObj (const TValue *t1, const TValue *t2);
|
|
+LUAI_FUNC int luaO_str2d (const char *s, lua_Number *result);
|
|
+LUAI_FUNC const char *luaO_pushvfstring (lua_State *L, const char *fmt,
|
|
+ va_list argp);
|
|
+LUAI_FUNC const char *luaO_pushfstring (lua_State *L, const char *fmt, ...);
|
|
+LUAI_FUNC void luaO_chunkid (char *out, const char *source, size_t len);
|
|
+
|
|
+
|
|
+#endif
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lopcodes.c
|
|
@@ -0,0 +1,102 @@
|
|
+/*
|
|
+** $Id: lopcodes.c,v 1.37.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#define lopcodes_c
|
|
+#define LUA_CORE
|
|
+
|
|
+
|
|
+#include "lopcodes.h"
|
|
+
|
|
+
|
|
+/* ORDER OP */
|
|
+
|
|
+const char *const luaP_opnames[NUM_OPCODES+1] = {
|
|
+ "MOVE",
|
|
+ "LOADK",
|
|
+ "LOADBOOL",
|
|
+ "LOADNIL",
|
|
+ "GETUPVAL",
|
|
+ "GETGLOBAL",
|
|
+ "GETTABLE",
|
|
+ "SETGLOBAL",
|
|
+ "SETUPVAL",
|
|
+ "SETTABLE",
|
|
+ "NEWTABLE",
|
|
+ "SELF",
|
|
+ "ADD",
|
|
+ "SUB",
|
|
+ "MUL",
|
|
+ "DIV",
|
|
+ "MOD",
|
|
+ "POW",
|
|
+ "UNM",
|
|
+ "NOT",
|
|
+ "LEN",
|
|
+ "CONCAT",
|
|
+ "JMP",
|
|
+ "EQ",
|
|
+ "LT",
|
|
+ "LE",
|
|
+ "TEST",
|
|
+ "TESTSET",
|
|
+ "CALL",
|
|
+ "TAILCALL",
|
|
+ "RETURN",
|
|
+ "FORLOOP",
|
|
+ "FORPREP",
|
|
+ "TFORLOOP",
|
|
+ "SETLIST",
|
|
+ "CLOSE",
|
|
+ "CLOSURE",
|
|
+ "VARARG",
|
|
+ NULL
|
|
+};
|
|
+
|
|
+
|
|
+#define opmode(t,a,b,c,m) (((t)<<7) | ((a)<<6) | ((b)<<4) | ((c)<<2) | (m))
|
|
+
|
|
+const lu_byte luaP_opmodes[NUM_OPCODES] = {
|
|
+/* T A B C mode opcode */
|
|
+ opmode(0, 1, OpArgR, OpArgN, iABC) /* OP_MOVE */
|
|
+ ,opmode(0, 1, OpArgK, OpArgN, iABx) /* OP_LOADK */
|
|
+ ,opmode(0, 1, OpArgU, OpArgU, iABC) /* OP_LOADBOOL */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iABC) /* OP_LOADNIL */
|
|
+ ,opmode(0, 1, OpArgU, OpArgN, iABC) /* OP_GETUPVAL */
|
|
+ ,opmode(0, 1, OpArgK, OpArgN, iABx) /* OP_GETGLOBAL */
|
|
+ ,opmode(0, 1, OpArgR, OpArgK, iABC) /* OP_GETTABLE */
|
|
+ ,opmode(0, 0, OpArgK, OpArgN, iABx) /* OP_SETGLOBAL */
|
|
+ ,opmode(0, 0, OpArgU, OpArgN, iABC) /* OP_SETUPVAL */
|
|
+ ,opmode(0, 0, OpArgK, OpArgK, iABC) /* OP_SETTABLE */
|
|
+ ,opmode(0, 1, OpArgU, OpArgU, iABC) /* OP_NEWTABLE */
|
|
+ ,opmode(0, 1, OpArgR, OpArgK, iABC) /* OP_SELF */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_ADD */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_SUB */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_MUL */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_DIV */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_MOD */
|
|
+ ,opmode(0, 1, OpArgK, OpArgK, iABC) /* OP_POW */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iABC) /* OP_UNM */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iABC) /* OP_NOT */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iABC) /* OP_LEN */
|
|
+ ,opmode(0, 1, OpArgR, OpArgR, iABC) /* OP_CONCAT */
|
|
+ ,opmode(0, 0, OpArgR, OpArgN, iAsBx) /* OP_JMP */
|
|
+ ,opmode(1, 0, OpArgK, OpArgK, iABC) /* OP_EQ */
|
|
+ ,opmode(1, 0, OpArgK, OpArgK, iABC) /* OP_LT */
|
|
+ ,opmode(1, 0, OpArgK, OpArgK, iABC) /* OP_LE */
|
|
+ ,opmode(1, 1, OpArgR, OpArgU, iABC) /* OP_TEST */
|
|
+ ,opmode(1, 1, OpArgR, OpArgU, iABC) /* OP_TESTSET */
|
|
+ ,opmode(0, 1, OpArgU, OpArgU, iABC) /* OP_CALL */
|
|
+ ,opmode(0, 1, OpArgU, OpArgU, iABC) /* OP_TAILCALL */
|
|
+ ,opmode(0, 0, OpArgU, OpArgN, iABC) /* OP_RETURN */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iAsBx) /* OP_FORLOOP */
|
|
+ ,opmode(0, 1, OpArgR, OpArgN, iAsBx) /* OP_FORPREP */
|
|
+ ,opmode(1, 0, OpArgN, OpArgU, iABC) /* OP_TFORLOOP */
|
|
+ ,opmode(0, 0, OpArgU, OpArgU, iABC) /* OP_SETLIST */
|
|
+ ,opmode(0, 0, OpArgN, OpArgN, iABC) /* OP_CLOSE */
|
|
+ ,opmode(0, 1, OpArgU, OpArgN, iABx) /* OP_CLOSURE */
|
|
+ ,opmode(0, 1, OpArgU, OpArgN, iABC) /* OP_VARARG */
|
|
+};
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lopcodes.h
|
|
@@ -0,0 +1,268 @@
|
|
+/*
|
|
+** $Id: lopcodes.h,v 1.125.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Opcodes for Lua virtual machine
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lopcodes_h
|
|
+#define lopcodes_h
|
|
+
|
|
+#include "llimits.h"
|
|
+
|
|
+
|
|
+/*===========================================================================
|
|
+ We assume that instructions are unsigned numbers.
|
|
+ All instructions have an opcode in the first 6 bits.
|
|
+ Instructions can have the following fields:
|
|
+ `A' : 8 bits
|
|
+ `B' : 9 bits
|
|
+ `C' : 9 bits
|
|
+ `Bx' : 18 bits (`B' and `C' together)
|
|
+ `sBx' : signed Bx
|
|
+
|
|
+ A signed argument is represented in excess K; that is, the number
|
|
+ value is the unsigned value minus K. K is exactly the maximum value
|
|
+ for that argument (so that -max is represented by 0, and +max is
|
|
+ represented by 2*max), which is half the maximum for the corresponding
|
|
+ unsigned argument.
|
|
+===========================================================================*/
|
|
+
|
|
+
|
|
+enum OpMode {iABC, iABx, iAsBx}; /* basic instruction format */
|
|
+
|
|
+
|
|
+/*
|
|
+** size and position of opcode arguments.
|
|
+*/
|
|
+#define SIZE_C 9
|
|
+#define SIZE_B 9
|
|
+#define SIZE_Bx (SIZE_C + SIZE_B)
|
|
+#define SIZE_A 8
|
|
+
|
|
+#define SIZE_OP 6
|
|
+
|
|
+#define POS_OP 0
|
|
+#define POS_A (POS_OP + SIZE_OP)
|
|
+#define POS_C (POS_A + SIZE_A)
|
|
+#define POS_B (POS_C + SIZE_C)
|
|
+#define POS_Bx POS_C
|
|
+
|
|
+
|
|
+/*
|
|
+** limits for opcode arguments.
|
|
+** we use (signed) int to manipulate most arguments,
|
|
+** so they must fit in LUAI_BITSINT-1 bits (-1 for sign)
|
|
+*/
|
|
+#if SIZE_Bx < LUAI_BITSINT-1
|
|
+#define MAXARG_Bx ((1<<SIZE_Bx)-1)
|
|
+#define MAXARG_sBx (MAXARG_Bx>>1) /* `sBx' is signed */
|
|
+#else
|
|
+#define MAXARG_Bx MAX_INT
|
|
+#define MAXARG_sBx MAX_INT
|
|
+#endif
|
|
+
|
|
+
|
|
+#define MAXARG_A ((1<<SIZE_A)-1)
|
|
+#define MAXARG_B ((1<<SIZE_B)-1)
|
|
+#define MAXARG_C ((1<<SIZE_C)-1)
|
|
+
|
|
+
|
|
+/* creates a mask with `n' 1 bits at position `p' */
|
|
+#define MASK1(n,p) ((~((~(Instruction)0)<<n))<<p)
|
|
+
|
|
+/* creates a mask with `n' 0 bits at position `p' */
|
|
+#define MASK0(n,p) (~MASK1(n,p))
|
|
+
|
|
+/*
|
|
+** the following macros help to manipulate instructions
|
|
+*/
|
|
+
|
|
+#define GET_OPCODE(i) (cast(OpCode, ((i)>>POS_OP) & MASK1(SIZE_OP,0)))
|
|
+#define SET_OPCODE(i,o) ((i) = (((i)&MASK0(SIZE_OP,POS_OP)) | \
|
|
+ ((cast(Instruction, o)<<POS_OP)&MASK1(SIZE_OP,POS_OP))))
|
|
+
|
|
+#define GETARG_A(i) (cast(int, ((i)>>POS_A) & MASK1(SIZE_A,0)))
|
|
+#define SETARG_A(i,u) ((i) = (((i)&MASK0(SIZE_A,POS_A)) | \
|
|
+ ((cast(Instruction, u)<<POS_A)&MASK1(SIZE_A,POS_A))))
|
|
+
|
|
+#define GETARG_B(i) (cast(int, ((i)>>POS_B) & MASK1(SIZE_B,0)))
|
|
+#define SETARG_B(i,b) ((i) = (((i)&MASK0(SIZE_B,POS_B)) | \
|
|
+ ((cast(Instruction, b)<<POS_B)&MASK1(SIZE_B,POS_B))))
|
|
+
|
|
+#define GETARG_C(i) (cast(int, ((i)>>POS_C) & MASK1(SIZE_C,0)))
|
|
+#define SETARG_C(i,b) ((i) = (((i)&MASK0(SIZE_C,POS_C)) | \
|
|
+ ((cast(Instruction, b)<<POS_C)&MASK1(SIZE_C,POS_C))))
|
|
+
|
|
+#define GETARG_Bx(i) (cast(int, ((i)>>POS_Bx) & MASK1(SIZE_Bx,0)))
|
|
+#define SETARG_Bx(i,b) ((i) = (((i)&MASK0(SIZE_Bx,POS_Bx)) | \
|
|
+ ((cast(Instruction, b)<<POS_Bx)&MASK1(SIZE_Bx,POS_Bx))))
|
|
+
|
|
+#define GETARG_sBx(i) (GETARG_Bx(i)-MAXARG_sBx)
|
|
+#define SETARG_sBx(i,b) SETARG_Bx((i),cast(unsigned int, (b)+MAXARG_sBx))
|
|
+
|
|
+
|
|
+#define CREATE_ABC(o,a,b,c) ((cast(Instruction, o)<<POS_OP) \
|
|
+ | (cast(Instruction, a)<<POS_A) \
|
|
+ | (cast(Instruction, b)<<POS_B) \
|
|
+ | (cast(Instruction, c)<<POS_C))
|
|
+
|
|
+#define CREATE_ABx(o,a,bc) ((cast(Instruction, o)<<POS_OP) \
|
|
+ | (cast(Instruction, a)<<POS_A) \
|
|
+ | (cast(Instruction, bc)<<POS_Bx))
|
|
+
|
|
+
|
|
+/*
|
|
+** Macros to operate RK indices
|
|
+*/
|
|
+
|
|
+/* this bit 1 means constant (0 means register) */
|
|
+#define BITRK (1 << (SIZE_B - 1))
|
|
+
|
|
+/* test whether value is a constant */
|
|
+#define ISK(x) ((x) & BITRK)
|
|
+
|
|
+/* gets the index of the constant */
|
|
+#define INDEXK(r) ((int)(r) & ~BITRK)
|
|
+
|
|
+#define MAXINDEXRK (BITRK - 1)
|
|
+
|
|
+/* code a constant index as a RK value */
|
|
+#define RKASK(x) ((x) | BITRK)
|
|
+
|
|
+
|
|
+/*
|
|
+** invalid register that fits in 8 bits
|
|
+*/
|
|
+#define NO_REG MAXARG_A
|
|
+
|
|
+
|
|
+/*
|
|
+** R(x) - register
|
|
+** Kst(x) - constant (in constant table)
|
|
+** RK(x) == if ISK(x) then Kst(INDEXK(x)) else R(x)
|
|
+*/
|
|
+
|
|
+
|
|
+/*
|
|
+** grep "ORDER OP" if you change these enums
|
|
+*/
|
|
+
|
|
+typedef enum {
|
|
+/*----------------------------------------------------------------------
|
|
+name args description
|
|
+------------------------------------------------------------------------*/
|
|
+OP_MOVE,/* A B R(A) := R(B) */
|
|
+OP_LOADK,/* A Bx R(A) := Kst(Bx) */
|
|
+OP_LOADBOOL,/* A B C R(A) := (Bool)B; if (C) pc++ */
|
|
+OP_LOADNIL,/* A B R(A) := ... := R(B) := nil */
|
|
+OP_GETUPVAL,/* A B R(A) := UpValue[B] */
|
|
+
|
|
+OP_GETGLOBAL,/* A Bx R(A) := Gbl[Kst(Bx)] */
|
|
+OP_GETTABLE,/* A B C R(A) := R(B)[RK(C)] */
|
|
+
|
|
+OP_SETGLOBAL,/* A Bx Gbl[Kst(Bx)] := R(A) */
|
|
+OP_SETUPVAL,/* A B UpValue[B] := R(A) */
|
|
+OP_SETTABLE,/* A B C R(A)[RK(B)] := RK(C) */
|
|
+
|
|
+OP_NEWTABLE,/* A B C R(A) := {} (size = B,C) */
|
|
+
|
|
+OP_SELF,/* A B C R(A+1) := R(B); R(A) := R(B)[RK(C)] */
|
|
+
|
|
+OP_ADD,/* A B C R(A) := RK(B) + RK(C) */
|
|
+OP_SUB,/* A B C R(A) := RK(B) - RK(C) */
|
|
+OP_MUL,/* A B C R(A) := RK(B) * RK(C) */
|
|
+OP_DIV,/* A B C R(A) := RK(B) / RK(C) */
|
|
+OP_MOD,/* A B C R(A) := RK(B) % RK(C) */
|
|
+OP_POW,/* A B C R(A) := RK(B) ^ RK(C) */
|
|
+OP_UNM,/* A B R(A) := -R(B) */
|
|
+OP_NOT,/* A B R(A) := not R(B) */
|
|
+OP_LEN,/* A B R(A) := length of R(B) */
|
|
+
|
|
+OP_CONCAT,/* A B C R(A) := R(B).. ... ..R(C) */
|
|
+
|
|
+OP_JMP,/* sBx pc+=sBx */
|
|
+
|
|
+OP_EQ,/* A B C if ((RK(B) == RK(C)) ~= A) then pc++ */
|
|
+OP_LT,/* A B C if ((RK(B) < RK(C)) ~= A) then pc++ */
|
|
+OP_LE,/* A B C if ((RK(B) <= RK(C)) ~= A) then pc++ */
|
|
+
|
|
+OP_TEST,/* A C if not (R(A) <=> C) then pc++ */
|
|
+OP_TESTSET,/* A B C if (R(B) <=> C) then R(A) := R(B) else pc++ */
|
|
+
|
|
+OP_CALL,/* A B C R(A), ... ,R(A+C-2) := R(A)(R(A+1), ... ,R(A+B-1)) */
|
|
+OP_TAILCALL,/* A B C return R(A)(R(A+1), ... ,R(A+B-1)) */
|
|
+OP_RETURN,/* A B return R(A), ... ,R(A+B-2) (see note) */
|
|
+
|
|
+OP_FORLOOP,/* A sBx R(A)+=R(A+2);
|
|
+ if R(A) <?= R(A+1) then { pc+=sBx; R(A+3)=R(A) }*/
|
|
+OP_FORPREP,/* A sBx R(A)-=R(A+2); pc+=sBx */
|
|
+
|
|
+OP_TFORLOOP,/* A C R(A+3), ... ,R(A+2+C) := R(A)(R(A+1), R(A+2));
|
|
+ if R(A+3) ~= nil then R(A+2)=R(A+3) else pc++ */
|
|
+OP_SETLIST,/* A B C R(A)[(C-1)*FPF+i] := R(A+i), 1 <= i <= B */
|
|
+
|
|
+OP_CLOSE,/* A close all variables in the stack up to (>=) R(A)*/
|
|
+OP_CLOSURE,/* A Bx R(A) := closure(KPROTO[Bx], R(A), ... ,R(A+n)) */
|
|
+
|
|
+OP_VARARG/* A B R(A), R(A+1), ..., R(A+B-1) = vararg */
|
|
+} OpCode;
|
|
+
|
|
+
|
|
+#define NUM_OPCODES (cast(int, OP_VARARG) + 1)
|
|
+
|
|
+
|
|
+
|
|
+/*===========================================================================
|
|
+ Notes:
|
|
+ (*) In OP_CALL, if (B == 0) then B = top. C is the number of returns - 1,
|
|
+ and can be 0: OP_CALL then sets `top' to last_result+1, so
|
|
+ next open instruction (OP_CALL, OP_RETURN, OP_SETLIST) may use `top'.
|
|
+
|
|
+ (*) In OP_VARARG, if (B == 0) then use actual number of varargs and
|
|
+ set top (like in OP_CALL with C == 0).
|
|
+
|
|
+ (*) In OP_RETURN, if (B == 0) then return up to `top'
|
|
+
|
|
+ (*) In OP_SETLIST, if (B == 0) then B = `top';
|
|
+ if (C == 0) then next `instruction' is real C
|
|
+
|
|
+ (*) For comparisons, A specifies what condition the test should accept
|
|
+ (true or false).
|
|
+
|
|
+ (*) All `skips' (pc++) assume that next instruction is a jump
|
|
+===========================================================================*/
|
|
+
|
|
+
|
|
+/*
|
|
+** masks for instruction properties. The format is:
|
|
+** bits 0-1: op mode
|
|
+** bits 2-3: C arg mode
|
|
+** bits 4-5: B arg mode
|
|
+** bit 6: instruction set register A
|
|
+** bit 7: operator is a test
|
|
+*/
|
|
+
|
|
+enum OpArgMask {
|
|
+ OpArgN, /* argument is not used */
|
|
+ OpArgU, /* argument is used */
|
|
+ OpArgR, /* argument is a register or a jump offset */
|
|
+ OpArgK /* argument is a constant or register/constant */
|
|
+};
|
|
+
|
|
+LUAI_DATA const lu_byte luaP_opmodes[NUM_OPCODES];
|
|
+
|
|
+#define getOpMode(m) (cast(enum OpMode, luaP_opmodes[m] & 3))
|
|
+#define getBMode(m) (cast(enum OpArgMask, (luaP_opmodes[m] >> 4) & 3))
|
|
+#define getCMode(m) (cast(enum OpArgMask, (luaP_opmodes[m] >> 2) & 3))
|
|
+#define testAMode(m) (luaP_opmodes[m] & (1 << 6))
|
|
+#define testTMode(m) (luaP_opmodes[m] & (1 << 7))
|
|
+
|
|
+
|
|
+LUAI_DATA const char *const luaP_opnames[NUM_OPCODES+1]; /* opcode names */
|
|
+
|
|
+
|
|
+/* number of list items to accumulate before a SETLIST instruction */
|
|
+#define LFIELDS_PER_FLUSH 50
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lparser.c
|
|
@@ -0,0 +1,1339 @@
|
|
+/*
|
|
+** $Id: lparser.c,v 2.42.1.3 2007/12/28 15:32:23 roberto Exp $
|
|
+** Lua Parser
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define lparser_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lcode.h"
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "llex.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lparser.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+
|
|
+
|
|
+
|
|
+#define hasmultret(k) ((k) == VCALL || (k) == VVARARG)
|
|
+
|
|
+#define getlocvar(fs, i) ((fs)->f->locvars[(fs)->actvar[i]])
|
|
+
|
|
+#define luaY_checklimit(fs,v,l,m) if ((v)>(l)) errorlimit(fs,l,m)
|
|
+
|
|
+
|
|
+/*
|
|
+** nodes for block list (list of active blocks)
|
|
+*/
|
|
+typedef struct BlockCnt {
|
|
+ struct BlockCnt *previous; /* chain */
|
|
+ int breaklist; /* list of jumps out of this loop */
|
|
+ lu_byte nactvar; /* # active locals outside the breakable structure */
|
|
+ lu_byte upval; /* true if some variable in the block is an upvalue */
|
|
+ lu_byte isbreakable; /* true if `block' is a loop */
|
|
+} BlockCnt;
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** prototypes for recursive non-terminal functions
|
|
+*/
|
|
+static void chunk (LexState *ls);
|
|
+static void expr (LexState *ls, expdesc *v);
|
|
+
|
|
+
|
|
+static void anchor_token (LexState *ls) {
|
|
+ if (ls->t.token == TK_NAME || ls->t.token == TK_STRING) {
|
|
+ TString *ts = ls->t.seminfo.ts;
|
|
+ luaX_newstring(ls, getstr(ts), ts->tsv.len);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void error_expected (LexState *ls, int token) {
|
|
+ luaX_syntaxerror(ls,
|
|
+ luaO_pushfstring(ls->L, LUA_QS " expected", luaX_token2str(ls, token)));
|
|
+}
|
|
+
|
|
+
|
|
+static void errorlimit (FuncState *fs, int limit, const char *what) {
|
|
+ const char *msg = (fs->f->linedefined == 0) ?
|
|
+ luaO_pushfstring(fs->L, "main function has more than %d %s", limit, what) :
|
|
+ luaO_pushfstring(fs->L, "function at line %d has more than %d %s",
|
|
+ fs->f->linedefined, limit, what);
|
|
+ luaX_lexerror(fs->ls, msg, 0);
|
|
+}
|
|
+
|
|
+
|
|
+static int testnext (LexState *ls, int c) {
|
|
+ if (ls->t.token == c) {
|
|
+ luaX_next(ls);
|
|
+ return 1;
|
|
+ }
|
|
+ else return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static void check (LexState *ls, int c) {
|
|
+ if (ls->t.token != c)
|
|
+ error_expected(ls, c);
|
|
+}
|
|
+
|
|
+static void checknext (LexState *ls, int c) {
|
|
+ check(ls, c);
|
|
+ luaX_next(ls);
|
|
+}
|
|
+
|
|
+
|
|
+#define check_condition(ls,c,msg) { if (!(c)) luaX_syntaxerror(ls, msg); }
|
|
+
|
|
+
|
|
+
|
|
+static void check_match (LexState *ls, int what, int who, int where) {
|
|
+ if (!testnext(ls, what)) {
|
|
+ if (where == ls->linenumber)
|
|
+ error_expected(ls, what);
|
|
+ else {
|
|
+ luaX_syntaxerror(ls, luaO_pushfstring(ls->L,
|
|
+ LUA_QS " expected (to close " LUA_QS " at line %d)",
|
|
+ luaX_token2str(ls, what), luaX_token2str(ls, who), where));
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static TString *str_checkname (LexState *ls) {
|
|
+ TString *ts;
|
|
+ check(ls, TK_NAME);
|
|
+ ts = ls->t.seminfo.ts;
|
|
+ luaX_next(ls);
|
|
+ return ts;
|
|
+}
|
|
+
|
|
+
|
|
+static void init_exp (expdesc *e, expkind k, int i) {
|
|
+ e->f = e->t = NO_JUMP;
|
|
+ e->k = k;
|
|
+ e->u.s.info = i;
|
|
+}
|
|
+
|
|
+
|
|
+static void codestring (LexState *ls, expdesc *e, TString *s) {
|
|
+ init_exp(e, VK, luaK_stringK(ls->fs, s));
|
|
+}
|
|
+
|
|
+
|
|
+static void checkname(LexState *ls, expdesc *e) {
|
|
+ codestring(ls, e, str_checkname(ls));
|
|
+}
|
|
+
|
|
+
|
|
+static int registerlocalvar (LexState *ls, TString *varname) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ Proto *f = fs->f;
|
|
+ int oldsize = f->sizelocvars;
|
|
+ luaM_growvector(ls->L, f->locvars, fs->nlocvars, f->sizelocvars,
|
|
+ LocVar, SHRT_MAX, "too many local variables");
|
|
+ while (oldsize < f->sizelocvars) f->locvars[oldsize++].varname = NULL;
|
|
+ f->locvars[fs->nlocvars].varname = varname;
|
|
+ luaC_objbarrier(ls->L, f, varname);
|
|
+ return fs->nlocvars++;
|
|
+}
|
|
+
|
|
+
|
|
+#define new_localvarliteral(ls,v,n) \
|
|
+ new_localvar(ls, luaX_newstring(ls, "" v, (sizeof(v)/sizeof(char))-1), n)
|
|
+
|
|
+
|
|
+static void new_localvar (LexState *ls, TString *name, int n) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ luaY_checklimit(fs, fs->nactvar+n+1, LUAI_MAXVARS, "local variables");
|
|
+ fs->actvar[fs->nactvar+n] = cast(unsigned short, registerlocalvar(ls, name));
|
|
+}
|
|
+
|
|
+
|
|
+static void adjustlocalvars (LexState *ls, int nvars) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ fs->nactvar = cast_byte(fs->nactvar + nvars);
|
|
+ for (; nvars; nvars--) {
|
|
+ getlocvar(fs, fs->nactvar - nvars).startpc = fs->pc;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void removevars (LexState *ls, int tolevel) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ while (fs->nactvar > tolevel)
|
|
+ getlocvar(fs, --fs->nactvar).endpc = fs->pc;
|
|
+}
|
|
+
|
|
+
|
|
+static int indexupvalue (FuncState *fs, TString *name, expdesc *v) {
|
|
+ int i;
|
|
+ Proto *f = fs->f;
|
|
+ int oldsize = f->sizeupvalues;
|
|
+ for (i=0; i<f->nups; i++) {
|
|
+ if (fs->upvalues[i].k == v->k && fs->upvalues[i].info == v->u.s.info) {
|
|
+ lua_assert(f->upvalues[i] == name);
|
|
+ return i;
|
|
+ }
|
|
+ }
|
|
+ /* new one */
|
|
+ luaY_checklimit(fs, f->nups + 1, LUAI_MAXUPVALUES, "upvalues");
|
|
+ luaM_growvector(fs->L, f->upvalues, f->nups, f->sizeupvalues,
|
|
+ TString *, MAX_INT, "");
|
|
+ while (oldsize < f->sizeupvalues) f->upvalues[oldsize++] = NULL;
|
|
+ f->upvalues[f->nups] = name;
|
|
+ luaC_objbarrier(fs->L, f, name);
|
|
+ lua_assert(v->k == VLOCAL || v->k == VUPVAL);
|
|
+ fs->upvalues[f->nups].k = cast_byte(v->k);
|
|
+ fs->upvalues[f->nups].info = cast_byte(v->u.s.info);
|
|
+ return f->nups++;
|
|
+}
|
|
+
|
|
+
|
|
+static int searchvar (FuncState *fs, TString *n) {
|
|
+ int i;
|
|
+ for (i=fs->nactvar-1; i >= 0; i--) {
|
|
+ if (n == getlocvar(fs, i).varname)
|
|
+ return i;
|
|
+ }
|
|
+ return -1; /* not found */
|
|
+}
|
|
+
|
|
+
|
|
+static void markupval (FuncState *fs, int level) {
|
|
+ BlockCnt *bl = fs->bl;
|
|
+ while (bl && bl->nactvar > level) bl = bl->previous;
|
|
+ if (bl) bl->upval = 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int singlevaraux (FuncState *fs, TString *n, expdesc *var, int base) {
|
|
+ if (fs == NULL) { /* no more levels? */
|
|
+ init_exp(var, VGLOBAL, NO_REG); /* default is global variable */
|
|
+ return VGLOBAL;
|
|
+ }
|
|
+ else {
|
|
+ int v = searchvar(fs, n); /* look up at current level */
|
|
+ if (v >= 0) {
|
|
+ init_exp(var, VLOCAL, v);
|
|
+ if (!base)
|
|
+ markupval(fs, v); /* local will be used as an upval */
|
|
+ return VLOCAL;
|
|
+ }
|
|
+ else { /* not found at current level; try upper one */
|
|
+ if (singlevaraux(fs->prev, n, var, 0) == VGLOBAL)
|
|
+ return VGLOBAL;
|
|
+ var->u.s.info = indexupvalue(fs, n, var); /* else was LOCAL or UPVAL */
|
|
+ var->k = VUPVAL; /* upvalue in this level */
|
|
+ return VUPVAL;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void singlevar (LexState *ls, expdesc *var) {
|
|
+ TString *varname = str_checkname(ls);
|
|
+ FuncState *fs = ls->fs;
|
|
+ if (singlevaraux(fs, varname, var, 1) == VGLOBAL)
|
|
+ var->u.s.info = luaK_stringK(fs, varname); /* info points to global name */
|
|
+}
|
|
+
|
|
+
|
|
+static void adjust_assign (LexState *ls, int nvars, int nexps, expdesc *e) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ int extra = nvars - nexps;
|
|
+ if (hasmultret(e->k)) {
|
|
+ extra++; /* includes call itself */
|
|
+ if (extra < 0) extra = 0;
|
|
+ luaK_setreturns(fs, e, extra); /* last exp. provides the difference */
|
|
+ if (extra > 1) luaK_reserveregs(fs, extra-1);
|
|
+ }
|
|
+ else {
|
|
+ if (e->k != VVOID) luaK_exp2nextreg(fs, e); /* close last expression */
|
|
+ if (extra > 0) {
|
|
+ int reg = fs->freereg;
|
|
+ luaK_reserveregs(fs, extra);
|
|
+ luaK_nil(fs, reg, extra);
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void enterlevel (LexState *ls) {
|
|
+ if (++ls->L->nCcalls > LUAI_MAXCCALLS)
|
|
+ luaX_lexerror(ls, "chunk has too many syntax levels", 0);
|
|
+}
|
|
+
|
|
+
|
|
+#define leavelevel(ls) ((ls)->L->nCcalls--)
|
|
+
|
|
+
|
|
+static void enterblock (FuncState *fs, BlockCnt *bl, lu_byte isbreakable) {
|
|
+ bl->breaklist = NO_JUMP;
|
|
+ bl->isbreakable = isbreakable;
|
|
+ bl->nactvar = fs->nactvar;
|
|
+ bl->upval = 0;
|
|
+ bl->previous = fs->bl;
|
|
+ fs->bl = bl;
|
|
+ lua_assert(fs->freereg == fs->nactvar);
|
|
+}
|
|
+
|
|
+
|
|
+static void leaveblock (FuncState *fs) {
|
|
+ BlockCnt *bl = fs->bl;
|
|
+ fs->bl = bl->previous;
|
|
+ removevars(fs->ls, bl->nactvar);
|
|
+ if (bl->upval)
|
|
+ luaK_codeABC(fs, OP_CLOSE, bl->nactvar, 0, 0);
|
|
+ /* a block either controls scope or breaks (never both) */
|
|
+ lua_assert(!bl->isbreakable || !bl->upval);
|
|
+ lua_assert(bl->nactvar == fs->nactvar);
|
|
+ fs->freereg = fs->nactvar; /* free registers */
|
|
+ luaK_patchtohere(fs, bl->breaklist);
|
|
+}
|
|
+
|
|
+
|
|
+static void pushclosure (LexState *ls, FuncState *func, expdesc *v) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ Proto *f = fs->f;
|
|
+ int oldsize = f->sizep;
|
|
+ int i;
|
|
+ luaM_growvector(ls->L, f->p, fs->np, f->sizep, Proto *,
|
|
+ MAXARG_Bx, "constant table overflow");
|
|
+ while (oldsize < f->sizep) f->p[oldsize++] = NULL;
|
|
+ f->p[fs->np++] = func->f;
|
|
+ luaC_objbarrier(ls->L, f, func->f);
|
|
+ init_exp(v, VRELOCABLE, luaK_codeABx(fs, OP_CLOSURE, 0, fs->np-1));
|
|
+ for (i=0; i<func->f->nups; i++) {
|
|
+ OpCode o = (func->upvalues[i].k == VLOCAL) ? OP_MOVE : OP_GETUPVAL;
|
|
+ luaK_codeABC(fs, o, 0, func->upvalues[i].info, 0);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void open_func (LexState *ls, FuncState *fs) {
|
|
+ lua_State *L = ls->L;
|
|
+ Proto *f = luaF_newproto(L);
|
|
+ fs->f = f;
|
|
+ fs->prev = ls->fs; /* linked list of funcstates */
|
|
+ fs->ls = ls;
|
|
+ fs->L = L;
|
|
+ ls->fs = fs;
|
|
+ fs->pc = 0;
|
|
+ fs->lasttarget = -1;
|
|
+ fs->jpc = NO_JUMP;
|
|
+ fs->freereg = 0;
|
|
+ fs->nk = 0;
|
|
+ fs->np = 0;
|
|
+ fs->nlocvars = 0;
|
|
+ fs->nactvar = 0;
|
|
+ fs->bl = NULL;
|
|
+ f->source = ls->source;
|
|
+ f->maxstacksize = 2; /* registers 0/1 are always valid */
|
|
+ fs->h = luaH_new(L, 0, 0);
|
|
+ /* anchor table of constants and prototype (to avoid being collected) */
|
|
+ sethvalue2s(L, L->top, fs->h);
|
|
+ incr_top(L);
|
|
+ setptvalue2s(L, L->top, f);
|
|
+ incr_top(L);
|
|
+}
|
|
+
|
|
+
|
|
+static void close_func (LexState *ls) {
|
|
+ lua_State *L = ls->L;
|
|
+ FuncState *fs = ls->fs;
|
|
+ Proto *f = fs->f;
|
|
+ removevars(ls, 0);
|
|
+ luaK_ret(fs, 0, 0); /* final return */
|
|
+ luaM_reallocvector(L, f->code, f->sizecode, fs->pc, Instruction);
|
|
+ f->sizecode = fs->pc;
|
|
+ luaM_reallocvector(L, f->lineinfo, f->sizelineinfo, fs->pc, int);
|
|
+ f->sizelineinfo = fs->pc;
|
|
+ luaM_reallocvector(L, f->k, f->sizek, fs->nk, TValue);
|
|
+ f->sizek = fs->nk;
|
|
+ luaM_reallocvector(L, f->p, f->sizep, fs->np, Proto *);
|
|
+ f->sizep = fs->np;
|
|
+ luaM_reallocvector(L, f->locvars, f->sizelocvars, fs->nlocvars, LocVar);
|
|
+ f->sizelocvars = fs->nlocvars;
|
|
+ luaM_reallocvector(L, f->upvalues, f->sizeupvalues, f->nups, TString *);
|
|
+ f->sizeupvalues = f->nups;
|
|
+ lua_assert(luaG_checkcode(f));
|
|
+ lua_assert(fs->bl == NULL);
|
|
+ ls->fs = fs->prev;
|
|
+ L->top -= 2; /* remove table and prototype from the stack */
|
|
+ /* last token read was anchored in defunct function; must reanchor it */
|
|
+ if (fs) anchor_token(ls);
|
|
+}
|
|
+
|
|
+
|
|
+Proto *luaY_parser (lua_State *L, ZIO *z, Mbuffer *buff, const char *name) {
|
|
+ struct LexState lexstate;
|
|
+ struct FuncState funcstate;
|
|
+ lexstate.buff = buff;
|
|
+ luaX_setinput(L, &lexstate, z, luaS_new(L, name));
|
|
+ open_func(&lexstate, &funcstate);
|
|
+ funcstate.f->is_vararg = VARARG_ISVARARG; /* main func. is always vararg */
|
|
+ luaX_next(&lexstate); /* read first token */
|
|
+ chunk(&lexstate);
|
|
+ check(&lexstate, TK_EOS);
|
|
+ close_func(&lexstate);
|
|
+ lua_assert(funcstate.prev == NULL);
|
|
+ lua_assert(funcstate.f->nups == 0);
|
|
+ lua_assert(lexstate.fs == NULL);
|
|
+ return funcstate.f;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*============================================================*/
|
|
+/* GRAMMAR RULES */
|
|
+/*============================================================*/
|
|
+
|
|
+
|
|
+static void field (LexState *ls, expdesc *v) {
|
|
+ /* field -> ['.' | ':'] NAME */
|
|
+ FuncState *fs = ls->fs;
|
|
+ expdesc key;
|
|
+ luaK_exp2anyreg(fs, v);
|
|
+ luaX_next(ls); /* skip the dot or colon */
|
|
+ checkname(ls, &key);
|
|
+ luaK_indexed(fs, v, &key);
|
|
+}
|
|
+
|
|
+
|
|
+static void yindex (LexState *ls, expdesc *v) {
|
|
+ /* index -> '[' expr ']' */
|
|
+ luaX_next(ls); /* skip the '[' */
|
|
+ expr(ls, v);
|
|
+ luaK_exp2val(ls->fs, v);
|
|
+ checknext(ls, ']');
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================================
|
|
+** Rules for Constructors
|
|
+** =======================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+struct ConsControl {
|
|
+ expdesc v; /* last list item read */
|
|
+ expdesc *t; /* table descriptor */
|
|
+ int nh; /* total number of `record' elements */
|
|
+ int na; /* total number of array elements */
|
|
+ int tostore; /* number of array elements pending to be stored */
|
|
+};
|
|
+
|
|
+
|
|
+static void recfield (LexState *ls, struct ConsControl *cc) {
|
|
+ /* recfield -> (NAME | `['exp1`]') = exp1 */
|
|
+ FuncState *fs = ls->fs;
|
|
+ int reg = ls->fs->freereg;
|
|
+ expdesc key, val;
|
|
+ int rkkey;
|
|
+ if (ls->t.token == TK_NAME) {
|
|
+ luaY_checklimit(fs, cc->nh, MAX_INT, "items in a constructor");
|
|
+ checkname(ls, &key);
|
|
+ }
|
|
+ else /* ls->t.token == '[' */
|
|
+ yindex(ls, &key);
|
|
+ cc->nh++;
|
|
+ checknext(ls, '=');
|
|
+ rkkey = luaK_exp2RK(fs, &key);
|
|
+ expr(ls, &val);
|
|
+ luaK_codeABC(fs, OP_SETTABLE, cc->t->u.s.info, rkkey, luaK_exp2RK(fs, &val));
|
|
+ fs->freereg = reg; /* free registers */
|
|
+}
|
|
+
|
|
+
|
|
+static void closelistfield (FuncState *fs, struct ConsControl *cc) {
|
|
+ if (cc->v.k == VVOID) return; /* there is no list item */
|
|
+ luaK_exp2nextreg(fs, &cc->v);
|
|
+ cc->v.k = VVOID;
|
|
+ if (cc->tostore == LFIELDS_PER_FLUSH) {
|
|
+ luaK_setlist(fs, cc->t->u.s.info, cc->na, cc->tostore); /* flush */
|
|
+ cc->tostore = 0; /* no more items pending */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void lastlistfield (FuncState *fs, struct ConsControl *cc) {
|
|
+ if (cc->tostore == 0) return;
|
|
+ if (hasmultret(cc->v.k)) {
|
|
+ luaK_setmultret(fs, &cc->v);
|
|
+ luaK_setlist(fs, cc->t->u.s.info, cc->na, LUA_MULTRET);
|
|
+ cc->na--; /* do not count last expression (unknown number of elements) */
|
|
+ }
|
|
+ else {
|
|
+ if (cc->v.k != VVOID)
|
|
+ luaK_exp2nextreg(fs, &cc->v);
|
|
+ luaK_setlist(fs, cc->t->u.s.info, cc->na, cc->tostore);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void listfield (LexState *ls, struct ConsControl *cc) {
|
|
+ expr(ls, &cc->v);
|
|
+ luaY_checklimit(ls->fs, cc->na, MAX_INT, "items in a constructor");
|
|
+ cc->na++;
|
|
+ cc->tostore++;
|
|
+}
|
|
+
|
|
+
|
|
+static void constructor (LexState *ls, expdesc *t) {
|
|
+ /* constructor -> ?? */
|
|
+ FuncState *fs = ls->fs;
|
|
+ int line = ls->linenumber;
|
|
+ int pc = luaK_codeABC(fs, OP_NEWTABLE, 0, 0, 0);
|
|
+ struct ConsControl cc;
|
|
+ cc.na = cc.nh = cc.tostore = 0;
|
|
+ cc.t = t;
|
|
+ init_exp(t, VRELOCABLE, pc);
|
|
+ init_exp(&cc.v, VVOID, 0); /* no value (yet) */
|
|
+ luaK_exp2nextreg(ls->fs, t); /* fix it at stack top (for gc) */
|
|
+ checknext(ls, '{');
|
|
+ do {
|
|
+ lua_assert(cc.v.k == VVOID || cc.tostore > 0);
|
|
+ if (ls->t.token == '}') break;
|
|
+ closelistfield(fs, &cc);
|
|
+ switch(ls->t.token) {
|
|
+ case TK_NAME: { /* may be listfields or recfields */
|
|
+ luaX_lookahead(ls);
|
|
+ if (ls->lookahead.token != '=') /* expression? */
|
|
+ listfield(ls, &cc);
|
|
+ else
|
|
+ recfield(ls, &cc);
|
|
+ break;
|
|
+ }
|
|
+ case '[': { /* constructor_item -> recfield */
|
|
+ recfield(ls, &cc);
|
|
+ break;
|
|
+ }
|
|
+ default: { /* constructor_part -> listfield */
|
|
+ listfield(ls, &cc);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ } while (testnext(ls, ',') || testnext(ls, ';'));
|
|
+ check_match(ls, '}', '{', line);
|
|
+ lastlistfield(fs, &cc);
|
|
+ SETARG_B(fs->f->code[pc], luaO_int2fb(cc.na)); /* set initial array size */
|
|
+ SETARG_C(fs->f->code[pc], luaO_int2fb(cc.nh)); /* set initial table size */
|
|
+}
|
|
+
|
|
+/* }====================================================================== */
|
|
+
|
|
+
|
|
+
|
|
+static void parlist (LexState *ls) {
|
|
+ /* parlist -> [ param { `,' param } ] */
|
|
+ FuncState *fs = ls->fs;
|
|
+ Proto *f = fs->f;
|
|
+ int nparams = 0;
|
|
+ f->is_vararg = 0;
|
|
+ if (ls->t.token != ')') { /* is `parlist' not empty? */
|
|
+ do {
|
|
+ switch (ls->t.token) {
|
|
+ case TK_NAME: { /* param -> NAME */
|
|
+ new_localvar(ls, str_checkname(ls), nparams++);
|
|
+ break;
|
|
+ }
|
|
+ case TK_DOTS: { /* param -> `...' */
|
|
+ luaX_next(ls);
|
|
+#if defined(LUA_COMPAT_VARARG)
|
|
+ /* use `arg' as default name */
|
|
+ new_localvarliteral(ls, "arg", nparams++);
|
|
+ f->is_vararg = VARARG_HASARG | VARARG_NEEDSARG;
|
|
+#endif
|
|
+ f->is_vararg |= VARARG_ISVARARG;
|
|
+ break;
|
|
+ }
|
|
+ default: luaX_syntaxerror(ls, "<name> or " LUA_QL("...") " expected");
|
|
+ }
|
|
+ } while (!f->is_vararg && testnext(ls, ','));
|
|
+ }
|
|
+ adjustlocalvars(ls, nparams);
|
|
+ f->numparams = cast_byte(fs->nactvar - (f->is_vararg & VARARG_HASARG));
|
|
+ luaK_reserveregs(fs, fs->nactvar); /* reserve register for parameters */
|
|
+}
|
|
+
|
|
+
|
|
+static void body (LexState *ls, expdesc *e, int needself, int line) {
|
|
+ /* body -> `(' parlist `)' chunk END */
|
|
+ FuncState new_fs;
|
|
+ open_func(ls, &new_fs);
|
|
+ new_fs.f->linedefined = line;
|
|
+ checknext(ls, '(');
|
|
+ if (needself) {
|
|
+ new_localvarliteral(ls, "self", 0);
|
|
+ adjustlocalvars(ls, 1);
|
|
+ }
|
|
+ parlist(ls);
|
|
+ checknext(ls, ')');
|
|
+ chunk(ls);
|
|
+ new_fs.f->lastlinedefined = ls->linenumber;
|
|
+ check_match(ls, TK_END, TK_FUNCTION, line);
|
|
+ close_func(ls);
|
|
+ pushclosure(ls, &new_fs, e);
|
|
+}
|
|
+
|
|
+
|
|
+static int explist1 (LexState *ls, expdesc *v) {
|
|
+ /* explist1 -> expr { `,' expr } */
|
|
+ int n = 1; /* at least one expression */
|
|
+ expr(ls, v);
|
|
+ while (testnext(ls, ',')) {
|
|
+ luaK_exp2nextreg(ls->fs, v);
|
|
+ expr(ls, v);
|
|
+ n++;
|
|
+ }
|
|
+ return n;
|
|
+}
|
|
+
|
|
+
|
|
+static void funcargs (LexState *ls, expdesc *f) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ expdesc args;
|
|
+ int base, nparams;
|
|
+ int line = ls->linenumber;
|
|
+ switch (ls->t.token) {
|
|
+ case '(': { /* funcargs -> `(' [ explist1 ] `)' */
|
|
+ if (line != ls->lastline)
|
|
+ luaX_syntaxerror(ls,"ambiguous syntax (function call x new statement)");
|
|
+ luaX_next(ls);
|
|
+ if (ls->t.token == ')') /* arg list is empty? */
|
|
+ args.k = VVOID;
|
|
+ else {
|
|
+ explist1(ls, &args);
|
|
+ luaK_setmultret(fs, &args);
|
|
+ }
|
|
+ check_match(ls, ')', '(', line);
|
|
+ break;
|
|
+ }
|
|
+ case '{': { /* funcargs -> constructor */
|
|
+ constructor(ls, &args);
|
|
+ break;
|
|
+ }
|
|
+ case TK_STRING: { /* funcargs -> STRING */
|
|
+ codestring(ls, &args, ls->t.seminfo.ts);
|
|
+ luaX_next(ls); /* must use `seminfo' before `next' */
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ luaX_syntaxerror(ls, "function arguments expected");
|
|
+ return;
|
|
+ }
|
|
+ }
|
|
+ lua_assert(f->k == VNONRELOC);
|
|
+ base = f->u.s.info; /* base register for call */
|
|
+ if (hasmultret(args.k))
|
|
+ nparams = LUA_MULTRET; /* open call */
|
|
+ else {
|
|
+ if (args.k != VVOID)
|
|
+ luaK_exp2nextreg(fs, &args); /* close last argument */
|
|
+ nparams = fs->freereg - (base+1);
|
|
+ }
|
|
+ init_exp(f, VCALL, luaK_codeABC(fs, OP_CALL, base, nparams+1, 2));
|
|
+ luaK_fixline(fs, line);
|
|
+ fs->freereg = base+1; /* call remove function and arguments and leaves
|
|
+ (unless changed) one result */
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================================
|
|
+** Expression parsing
|
|
+** =======================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+static void prefixexp (LexState *ls, expdesc *v) {
|
|
+ /* prefixexp -> NAME | '(' expr ')' */
|
|
+ switch (ls->t.token) {
|
|
+ case '(': {
|
|
+ int line = ls->linenumber;
|
|
+ luaX_next(ls);
|
|
+ expr(ls, v);
|
|
+ check_match(ls, ')', '(', line);
|
|
+ luaK_dischargevars(ls->fs, v);
|
|
+ return;
|
|
+ }
|
|
+ case TK_NAME: {
|
|
+ singlevar(ls, v);
|
|
+ return;
|
|
+ }
|
|
+ default: {
|
|
+ luaX_syntaxerror(ls, "unexpected symbol");
|
|
+ return;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void primaryexp (LexState *ls, expdesc *v) {
|
|
+ /* primaryexp ->
|
|
+ prefixexp { `.' NAME | `[' exp `]' | `:' NAME funcargs | funcargs } */
|
|
+ FuncState *fs = ls->fs;
|
|
+ prefixexp(ls, v);
|
|
+ for (;;) {
|
|
+ switch (ls->t.token) {
|
|
+ case '.': { /* field */
|
|
+ field(ls, v);
|
|
+ break;
|
|
+ }
|
|
+ case '[': { /* `[' exp1 `]' */
|
|
+ expdesc key;
|
|
+ luaK_exp2anyreg(fs, v);
|
|
+ yindex(ls, &key);
|
|
+ luaK_indexed(fs, v, &key);
|
|
+ break;
|
|
+ }
|
|
+ case ':': { /* `:' NAME funcargs */
|
|
+ expdesc key;
|
|
+ luaX_next(ls);
|
|
+ checkname(ls, &key);
|
|
+ luaK_self(fs, v, &key);
|
|
+ funcargs(ls, v);
|
|
+ break;
|
|
+ }
|
|
+ case '(': case TK_STRING: case '{': { /* funcargs */
|
|
+ luaK_exp2nextreg(fs, v);
|
|
+ funcargs(ls, v);
|
|
+ break;
|
|
+ }
|
|
+ default: return;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void simpleexp (LexState *ls, expdesc *v) {
|
|
+ /* simpleexp -> NUMBER | STRING | NIL | true | false | ... |
|
|
+ constructor | FUNCTION body | primaryexp */
|
|
+ switch (ls->t.token) {
|
|
+ case TK_NUMBER: {
|
|
+ init_exp(v, VKNUM, 0);
|
|
+ v->u.nval = ls->t.seminfo.r;
|
|
+ break;
|
|
+ }
|
|
+ case TK_STRING: {
|
|
+ codestring(ls, v, ls->t.seminfo.ts);
|
|
+ break;
|
|
+ }
|
|
+ case TK_NIL: {
|
|
+ init_exp(v, VNIL, 0);
|
|
+ break;
|
|
+ }
|
|
+ case TK_TRUE: {
|
|
+ init_exp(v, VTRUE, 0);
|
|
+ break;
|
|
+ }
|
|
+ case TK_FALSE: {
|
|
+ init_exp(v, VFALSE, 0);
|
|
+ break;
|
|
+ }
|
|
+ case TK_DOTS: { /* vararg */
|
|
+ FuncState *fs = ls->fs;
|
|
+ check_condition(ls, fs->f->is_vararg,
|
|
+ "cannot use " LUA_QL("...") " outside a vararg function");
|
|
+ fs->f->is_vararg &= ~VARARG_NEEDSARG; /* don't need 'arg' */
|
|
+ init_exp(v, VVARARG, luaK_codeABC(fs, OP_VARARG, 0, 1, 0));
|
|
+ break;
|
|
+ }
|
|
+ case '{': { /* constructor */
|
|
+ constructor(ls, v);
|
|
+ return;
|
|
+ }
|
|
+ case TK_FUNCTION: {
|
|
+ luaX_next(ls);
|
|
+ body(ls, v, 0, ls->linenumber);
|
|
+ return;
|
|
+ }
|
|
+ default: {
|
|
+ primaryexp(ls, v);
|
|
+ return;
|
|
+ }
|
|
+ }
|
|
+ luaX_next(ls);
|
|
+}
|
|
+
|
|
+
|
|
+static UnOpr getunopr (int op) {
|
|
+ switch (op) {
|
|
+ case TK_NOT: return OPR_NOT;
|
|
+ case '-': return OPR_MINUS;
|
|
+ case '#': return OPR_LEN;
|
|
+ default: return OPR_NOUNOPR;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static BinOpr getbinopr (int op) {
|
|
+ switch (op) {
|
|
+ case '+': return OPR_ADD;
|
|
+ case '-': return OPR_SUB;
|
|
+ case '*': return OPR_MUL;
|
|
+ case '/': return OPR_DIV;
|
|
+ case '%': return OPR_MOD;
|
|
+ case '^': return OPR_POW;
|
|
+ case TK_CONCAT: return OPR_CONCAT;
|
|
+ case TK_NE: return OPR_NE;
|
|
+ case TK_EQ: return OPR_EQ;
|
|
+ case '<': return OPR_LT;
|
|
+ case TK_LE: return OPR_LE;
|
|
+ case '>': return OPR_GT;
|
|
+ case TK_GE: return OPR_GE;
|
|
+ case TK_AND: return OPR_AND;
|
|
+ case TK_OR: return OPR_OR;
|
|
+ default: return OPR_NOBINOPR;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static const struct {
|
|
+ lu_byte left; /* left priority for each binary operator */
|
|
+ lu_byte right; /* right priority */
|
|
+} priority[] = { /* ORDER OPR */
|
|
+ {6, 6}, {6, 6}, {7, 7}, {7, 7}, {7, 7}, /* `+' `-' `/' `%' */
|
|
+ {10, 9}, {5, 4}, /* power and concat (right associative) */
|
|
+ {3, 3}, {3, 3}, /* equality and inequality */
|
|
+ {3, 3}, {3, 3}, {3, 3}, {3, 3}, /* order */
|
|
+ {2, 2}, {1, 1} /* logical (and/or) */
|
|
+};
|
|
+
|
|
+#define UNARY_PRIORITY 8 /* priority for unary operators */
|
|
+
|
|
+
|
|
+/*
|
|
+** subexpr -> (simpleexp | unop subexpr) { binop subexpr }
|
|
+** where `binop' is any binary operator with a priority higher than `limit'
|
|
+*/
|
|
+static BinOpr subexpr (LexState *ls, expdesc *v, unsigned int limit) {
|
|
+ BinOpr op;
|
|
+ UnOpr uop;
|
|
+ enterlevel(ls);
|
|
+ uop = getunopr(ls->t.token);
|
|
+ if (uop != OPR_NOUNOPR) {
|
|
+ luaX_next(ls);
|
|
+ subexpr(ls, v, UNARY_PRIORITY);
|
|
+ luaK_prefix(ls->fs, uop, v);
|
|
+ }
|
|
+ else simpleexp(ls, v);
|
|
+ /* expand while operators have priorities higher than `limit' */
|
|
+ op = getbinopr(ls->t.token);
|
|
+ while (op != OPR_NOBINOPR && priority[op].left > limit) {
|
|
+ expdesc v2;
|
|
+ BinOpr nextop;
|
|
+ luaX_next(ls);
|
|
+ luaK_infix(ls->fs, op, v);
|
|
+ /* read sub-expression with higher priority */
|
|
+ nextop = subexpr(ls, &v2, priority[op].right);
|
|
+ luaK_posfix(ls->fs, op, v, &v2);
|
|
+ op = nextop;
|
|
+ }
|
|
+ leavelevel(ls);
|
|
+ return op; /* return first untreated operator */
|
|
+}
|
|
+
|
|
+
|
|
+static void expr (LexState *ls, expdesc *v) {
|
|
+ subexpr(ls, v, 0);
|
|
+}
|
|
+
|
|
+/* }==================================================================== */
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================================
|
|
+** Rules for Statements
|
|
+** =======================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+static int block_follow (int token) {
|
|
+ switch (token) {
|
|
+ case TK_ELSE: case TK_ELSEIF: case TK_END:
|
|
+ case TK_UNTIL: case TK_EOS:
|
|
+ return 1;
|
|
+ default: return 0;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void block (LexState *ls) {
|
|
+ /* block -> chunk */
|
|
+ FuncState *fs = ls->fs;
|
|
+ BlockCnt bl;
|
|
+ enterblock(fs, &bl, 0);
|
|
+ chunk(ls);
|
|
+ lua_assert(bl.breaklist == NO_JUMP);
|
|
+ leaveblock(fs);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** structure to chain all variables in the left-hand side of an
|
|
+** assignment
|
|
+*/
|
|
+struct LHS_assign {
|
|
+ struct LHS_assign *prev;
|
|
+ expdesc v; /* variable (global, local, upvalue, or indexed) */
|
|
+};
|
|
+
|
|
+
|
|
+/*
|
|
+** check whether, in an assignment to a local variable, the local variable
|
|
+** is needed in a previous assignment (to a table). If so, save original
|
|
+** local value in a safe place and use this safe copy in the previous
|
|
+** assignment.
|
|
+*/
|
|
+static void check_conflict (LexState *ls, struct LHS_assign *lh, expdesc *v) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ int extra = fs->freereg; /* eventual position to save local variable */
|
|
+ int conflict = 0;
|
|
+ for (; lh; lh = lh->prev) {
|
|
+ if (lh->v.k == VINDEXED) {
|
|
+ if (lh->v.u.s.info == v->u.s.info) { /* conflict? */
|
|
+ conflict = 1;
|
|
+ lh->v.u.s.info = extra; /* previous assignment will use safe copy */
|
|
+ }
|
|
+ if (lh->v.u.s.aux == v->u.s.info) { /* conflict? */
|
|
+ conflict = 1;
|
|
+ lh->v.u.s.aux = extra; /* previous assignment will use safe copy */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ if (conflict) {
|
|
+ luaK_codeABC(fs, OP_MOVE, fs->freereg, v->u.s.info, 0); /* make copy */
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void assignment (LexState *ls, struct LHS_assign *lh, int nvars) {
|
|
+ expdesc e;
|
|
+ check_condition(ls, VLOCAL <= lh->v.k && lh->v.k <= VINDEXED,
|
|
+ "syntax error");
|
|
+ if (testnext(ls, ',')) { /* assignment -> `,' primaryexp assignment */
|
|
+ struct LHS_assign nv;
|
|
+ nv.prev = lh;
|
|
+ primaryexp(ls, &nv.v);
|
|
+ if (nv.v.k == VLOCAL)
|
|
+ check_conflict(ls, lh, &nv.v);
|
|
+ luaY_checklimit(ls->fs, nvars, LUAI_MAXCCALLS - ls->L->nCcalls,
|
|
+ "variables in assignment");
|
|
+ assignment(ls, &nv, nvars+1);
|
|
+ }
|
|
+ else { /* assignment -> `=' explist1 */
|
|
+ int nexps;
|
|
+ checknext(ls, '=');
|
|
+ nexps = explist1(ls, &e);
|
|
+ if (nexps != nvars) {
|
|
+ adjust_assign(ls, nvars, nexps, &e);
|
|
+ if (nexps > nvars)
|
|
+ ls->fs->freereg -= nexps - nvars; /* remove extra values */
|
|
+ }
|
|
+ else {
|
|
+ luaK_setoneret(ls->fs, &e); /* close last expression */
|
|
+ luaK_storevar(ls->fs, &lh->v, &e);
|
|
+ return; /* avoid default */
|
|
+ }
|
|
+ }
|
|
+ init_exp(&e, VNONRELOC, ls->fs->freereg-1); /* default assignment */
|
|
+ luaK_storevar(ls->fs, &lh->v, &e);
|
|
+}
|
|
+
|
|
+
|
|
+static int cond (LexState *ls) {
|
|
+ /* cond -> exp */
|
|
+ expdesc v;
|
|
+ expr(ls, &v); /* read condition */
|
|
+ if (v.k == VNIL) v.k = VFALSE; /* `falses' are all equal here */
|
|
+ luaK_goiftrue(ls->fs, &v);
|
|
+ return v.f;
|
|
+}
|
|
+
|
|
+
|
|
+static void breakstat (LexState *ls) {
|
|
+ FuncState *fs = ls->fs;
|
|
+ BlockCnt *bl = fs->bl;
|
|
+ int upval = 0;
|
|
+ while (bl && !bl->isbreakable) {
|
|
+ upval |= bl->upval;
|
|
+ bl = bl->previous;
|
|
+ }
|
|
+ if (!bl)
|
|
+ luaX_syntaxerror(ls, "no loop to break");
|
|
+ if (upval)
|
|
+ luaK_codeABC(fs, OP_CLOSE, bl->nactvar, 0, 0);
|
|
+ luaK_concat(fs, &bl->breaklist, luaK_jump(fs));
|
|
+}
|
|
+
|
|
+
|
|
+static void whilestat (LexState *ls, int line) {
|
|
+ /* whilestat -> WHILE cond DO block END */
|
|
+ FuncState *fs = ls->fs;
|
|
+ int whileinit;
|
|
+ int condexit;
|
|
+ BlockCnt bl;
|
|
+ luaX_next(ls); /* skip WHILE */
|
|
+ whileinit = luaK_getlabel(fs);
|
|
+ condexit = cond(ls);
|
|
+ enterblock(fs, &bl, 1);
|
|
+ checknext(ls, TK_DO);
|
|
+ block(ls);
|
|
+ luaK_patchlist(fs, luaK_jump(fs), whileinit);
|
|
+ check_match(ls, TK_END, TK_WHILE, line);
|
|
+ leaveblock(fs);
|
|
+ luaK_patchtohere(fs, condexit); /* false conditions finish the loop */
|
|
+}
|
|
+
|
|
+
|
|
+static void repeatstat (LexState *ls, int line) {
|
|
+ /* repeatstat -> REPEAT block UNTIL cond */
|
|
+ int condexit;
|
|
+ FuncState *fs = ls->fs;
|
|
+ int repeat_init = luaK_getlabel(fs);
|
|
+ BlockCnt bl1, bl2;
|
|
+ enterblock(fs, &bl1, 1); /* loop block */
|
|
+ enterblock(fs, &bl2, 0); /* scope block */
|
|
+ luaX_next(ls); /* skip REPEAT */
|
|
+ chunk(ls);
|
|
+ check_match(ls, TK_UNTIL, TK_REPEAT, line);
|
|
+ condexit = cond(ls); /* read condition (inside scope block) */
|
|
+ if (!bl2.upval) { /* no upvalues? */
|
|
+ leaveblock(fs); /* finish scope */
|
|
+ luaK_patchlist(ls->fs, condexit, repeat_init); /* close the loop */
|
|
+ }
|
|
+ else { /* complete semantics when there are upvalues */
|
|
+ breakstat(ls); /* if condition then break */
|
|
+ luaK_patchtohere(ls->fs, condexit); /* else... */
|
|
+ leaveblock(fs); /* finish scope... */
|
|
+ luaK_patchlist(ls->fs, luaK_jump(fs), repeat_init); /* and repeat */
|
|
+ }
|
|
+ leaveblock(fs); /* finish loop */
|
|
+}
|
|
+
|
|
+
|
|
+static int exp1 (LexState *ls) {
|
|
+ expdesc e;
|
|
+ int k;
|
|
+ expr(ls, &e);
|
|
+ k = e.k;
|
|
+ luaK_exp2nextreg(ls->fs, &e);
|
|
+ return k;
|
|
+}
|
|
+
|
|
+
|
|
+static void forbody (LexState *ls, int base, int line, int nvars, int isnum) {
|
|
+ /* forbody -> DO block */
|
|
+ BlockCnt bl;
|
|
+ FuncState *fs = ls->fs;
|
|
+ int prep, endfor;
|
|
+ adjustlocalvars(ls, 3); /* control variables */
|
|
+ checknext(ls, TK_DO);
|
|
+ prep = isnum ? luaK_codeAsBx(fs, OP_FORPREP, base, NO_JUMP) : luaK_jump(fs);
|
|
+ enterblock(fs, &bl, 0); /* scope for declared variables */
|
|
+ adjustlocalvars(ls, nvars);
|
|
+ luaK_reserveregs(fs, nvars);
|
|
+ block(ls);
|
|
+ leaveblock(fs); /* end of scope for declared variables */
|
|
+ luaK_patchtohere(fs, prep);
|
|
+ endfor = (isnum) ? luaK_codeAsBx(fs, OP_FORLOOP, base, NO_JUMP) :
|
|
+ luaK_codeABC(fs, OP_TFORLOOP, base, 0, nvars);
|
|
+ luaK_fixline(fs, line); /* pretend that `OP_FOR' starts the loop */
|
|
+ luaK_patchlist(fs, (isnum ? endfor : luaK_jump(fs)), prep + 1);
|
|
+}
|
|
+
|
|
+
|
|
+static void fornum (LexState *ls, TString *varname, int line) {
|
|
+ /* fornum -> NAME = exp1,exp1[,exp1] forbody */
|
|
+ FuncState *fs = ls->fs;
|
|
+ int base = fs->freereg;
|
|
+ new_localvarliteral(ls, "(for index)", 0);
|
|
+ new_localvarliteral(ls, "(for limit)", 1);
|
|
+ new_localvarliteral(ls, "(for step)", 2);
|
|
+ new_localvar(ls, varname, 3);
|
|
+ checknext(ls, '=');
|
|
+ exp1(ls); /* initial value */
|
|
+ checknext(ls, ',');
|
|
+ exp1(ls); /* limit */
|
|
+ if (testnext(ls, ','))
|
|
+ exp1(ls); /* optional step */
|
|
+ else { /* default step = 1 */
|
|
+ luaK_codeABx(fs, OP_LOADK, fs->freereg, luaK_numberK(fs, 1));
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ }
|
|
+ forbody(ls, base, line, 1, 1);
|
|
+}
|
|
+
|
|
+
|
|
+static void forlist (LexState *ls, TString *indexname) {
|
|
+ /* forlist -> NAME {,NAME} IN explist1 forbody */
|
|
+ FuncState *fs = ls->fs;
|
|
+ expdesc e;
|
|
+ int nvars = 0;
|
|
+ int line;
|
|
+ int base = fs->freereg;
|
|
+ /* create control variables */
|
|
+ new_localvarliteral(ls, "(for generator)", nvars++);
|
|
+ new_localvarliteral(ls, "(for state)", nvars++);
|
|
+ new_localvarliteral(ls, "(for control)", nvars++);
|
|
+ /* create declared variables */
|
|
+ new_localvar(ls, indexname, nvars++);
|
|
+ while (testnext(ls, ','))
|
|
+ new_localvar(ls, str_checkname(ls), nvars++);
|
|
+ checknext(ls, TK_IN);
|
|
+ line = ls->linenumber;
|
|
+ adjust_assign(ls, 3, explist1(ls, &e), &e);
|
|
+ luaK_checkstack(fs, 3); /* extra space to call generator */
|
|
+ forbody(ls, base, line, nvars - 3, 0);
|
|
+}
|
|
+
|
|
+
|
|
+static void forstat (LexState *ls, int line) {
|
|
+ /* forstat -> FOR (fornum | forlist) END */
|
|
+ FuncState *fs = ls->fs;
|
|
+ TString *varname;
|
|
+ BlockCnt bl;
|
|
+ enterblock(fs, &bl, 1); /* scope for loop and control variables */
|
|
+ luaX_next(ls); /* skip `for' */
|
|
+ varname = str_checkname(ls); /* first variable name */
|
|
+ switch (ls->t.token) {
|
|
+ case '=': fornum(ls, varname, line); break;
|
|
+ case ',': case TK_IN: forlist(ls, varname); break;
|
|
+ default: luaX_syntaxerror(ls, LUA_QL("=") " or " LUA_QL("in") " expected");
|
|
+ }
|
|
+ check_match(ls, TK_END, TK_FOR, line);
|
|
+ leaveblock(fs); /* loop scope (`break' jumps to this point) */
|
|
+}
|
|
+
|
|
+
|
|
+static int test_then_block (LexState *ls) {
|
|
+ /* test_then_block -> [IF | ELSEIF] cond THEN block */
|
|
+ int condexit;
|
|
+ luaX_next(ls); /* skip IF or ELSEIF */
|
|
+ condexit = cond(ls);
|
|
+ checknext(ls, TK_THEN);
|
|
+ block(ls); /* `then' part */
|
|
+ return condexit;
|
|
+}
|
|
+
|
|
+
|
|
+static void ifstat (LexState *ls, int line) {
|
|
+ /* ifstat -> IF cond THEN block {ELSEIF cond THEN block} [ELSE block] END */
|
|
+ FuncState *fs = ls->fs;
|
|
+ int flist;
|
|
+ int escapelist = NO_JUMP;
|
|
+ flist = test_then_block(ls); /* IF cond THEN block */
|
|
+ while (ls->t.token == TK_ELSEIF) {
|
|
+ luaK_concat(fs, &escapelist, luaK_jump(fs));
|
|
+ luaK_patchtohere(fs, flist);
|
|
+ flist = test_then_block(ls); /* ELSEIF cond THEN block */
|
|
+ }
|
|
+ if (ls->t.token == TK_ELSE) {
|
|
+ luaK_concat(fs, &escapelist, luaK_jump(fs));
|
|
+ luaK_patchtohere(fs, flist);
|
|
+ luaX_next(ls); /* skip ELSE (after patch, for correct line info) */
|
|
+ block(ls); /* `else' part */
|
|
+ }
|
|
+ else
|
|
+ luaK_concat(fs, &escapelist, flist);
|
|
+ luaK_patchtohere(fs, escapelist);
|
|
+ check_match(ls, TK_END, TK_IF, line);
|
|
+}
|
|
+
|
|
+
|
|
+static void localfunc (LexState *ls) {
|
|
+ expdesc v, b;
|
|
+ FuncState *fs = ls->fs;
|
|
+ new_localvar(ls, str_checkname(ls), 0);
|
|
+ init_exp(&v, VLOCAL, fs->freereg);
|
|
+ luaK_reserveregs(fs, 1);
|
|
+ adjustlocalvars(ls, 1);
|
|
+ body(ls, &b, 0, ls->linenumber);
|
|
+ luaK_storevar(fs, &v, &b);
|
|
+ /* debug information will only see the variable after this point! */
|
|
+ getlocvar(fs, fs->nactvar - 1).startpc = fs->pc;
|
|
+}
|
|
+
|
|
+
|
|
+static void localstat (LexState *ls) {
|
|
+ /* stat -> LOCAL NAME {`,' NAME} [`=' explist1] */
|
|
+ int nvars = 0;
|
|
+ int nexps;
|
|
+ expdesc e;
|
|
+ do {
|
|
+ new_localvar(ls, str_checkname(ls), nvars++);
|
|
+ } while (testnext(ls, ','));
|
|
+ if (testnext(ls, '='))
|
|
+ nexps = explist1(ls, &e);
|
|
+ else {
|
|
+ e.k = VVOID;
|
|
+ nexps = 0;
|
|
+ }
|
|
+ adjust_assign(ls, nvars, nexps, &e);
|
|
+ adjustlocalvars(ls, nvars);
|
|
+}
|
|
+
|
|
+
|
|
+static int funcname (LexState *ls, expdesc *v) {
|
|
+ /* funcname -> NAME {field} [`:' NAME] */
|
|
+ int needself = 0;
|
|
+ singlevar(ls, v);
|
|
+ while (ls->t.token == '.')
|
|
+ field(ls, v);
|
|
+ if (ls->t.token == ':') {
|
|
+ needself = 1;
|
|
+ field(ls, v);
|
|
+ }
|
|
+ return needself;
|
|
+}
|
|
+
|
|
+
|
|
+static void funcstat (LexState *ls, int line) {
|
|
+ /* funcstat -> FUNCTION funcname body */
|
|
+ int needself;
|
|
+ expdesc v, b;
|
|
+ luaX_next(ls); /* skip FUNCTION */
|
|
+ needself = funcname(ls, &v);
|
|
+ body(ls, &b, needself, line);
|
|
+ luaK_storevar(ls->fs, &v, &b);
|
|
+ luaK_fixline(ls->fs, line); /* definition `happens' in the first line */
|
|
+}
|
|
+
|
|
+
|
|
+static void exprstat (LexState *ls) {
|
|
+ /* stat -> func | assignment */
|
|
+ FuncState *fs = ls->fs;
|
|
+ struct LHS_assign v;
|
|
+ primaryexp(ls, &v.v);
|
|
+ if (v.v.k == VCALL) /* stat -> func */
|
|
+ SETARG_C(getcode(fs, &v.v), 1); /* call statement uses no results */
|
|
+ else { /* stat -> assignment */
|
|
+ v.prev = NULL;
|
|
+ assignment(ls, &v, 1);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void retstat (LexState *ls) {
|
|
+ /* stat -> RETURN explist */
|
|
+ FuncState *fs = ls->fs;
|
|
+ expdesc e;
|
|
+ int first, nret; /* registers with returned values */
|
|
+ luaX_next(ls); /* skip RETURN */
|
|
+ if (block_follow(ls->t.token) || ls->t.token == ';')
|
|
+ first = nret = 0; /* return no values */
|
|
+ else {
|
|
+ nret = explist1(ls, &e); /* optional return values */
|
|
+ if (hasmultret(e.k)) {
|
|
+ luaK_setmultret(fs, &e);
|
|
+ if (e.k == VCALL && nret == 1) { /* tail call? */
|
|
+ SET_OPCODE(getcode(fs,&e), OP_TAILCALL);
|
|
+ lua_assert(GETARG_A(getcode(fs,&e)) == fs->nactvar);
|
|
+ }
|
|
+ first = fs->nactvar;
|
|
+ nret = LUA_MULTRET; /* return all values */
|
|
+ }
|
|
+ else {
|
|
+ if (nret == 1) /* only one single value? */
|
|
+ first = luaK_exp2anyreg(fs, &e);
|
|
+ else {
|
|
+ luaK_exp2nextreg(fs, &e); /* values must go to the `stack' */
|
|
+ first = fs->nactvar; /* return all `active' values */
|
|
+ lua_assert(nret == fs->freereg - first);
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ luaK_ret(fs, first, nret);
|
|
+}
|
|
+
|
|
+
|
|
+static int statement (LexState *ls) {
|
|
+ int line = ls->linenumber; /* may be needed for error messages */
|
|
+ switch (ls->t.token) {
|
|
+ case TK_IF: { /* stat -> ifstat */
|
|
+ ifstat(ls, line);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_WHILE: { /* stat -> whilestat */
|
|
+ whilestat(ls, line);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_DO: { /* stat -> DO block END */
|
|
+ luaX_next(ls); /* skip DO */
|
|
+ block(ls);
|
|
+ check_match(ls, TK_END, TK_DO, line);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_FOR: { /* stat -> forstat */
|
|
+ forstat(ls, line);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_REPEAT: { /* stat -> repeatstat */
|
|
+ repeatstat(ls, line);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_FUNCTION: {
|
|
+ funcstat(ls, line); /* stat -> funcstat */
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_LOCAL: { /* stat -> localstat */
|
|
+ luaX_next(ls); /* skip LOCAL */
|
|
+ if (testnext(ls, TK_FUNCTION)) /* local function? */
|
|
+ localfunc(ls);
|
|
+ else
|
|
+ localstat(ls);
|
|
+ return 0;
|
|
+ }
|
|
+ case TK_RETURN: { /* stat -> retstat */
|
|
+ retstat(ls);
|
|
+ return 1; /* must be last statement */
|
|
+ }
|
|
+ case TK_BREAK: { /* stat -> breakstat */
|
|
+ luaX_next(ls); /* skip BREAK */
|
|
+ breakstat(ls);
|
|
+ return 1; /* must be last statement */
|
|
+ }
|
|
+ default: {
|
|
+ exprstat(ls);
|
|
+ return 0; /* to avoid warnings */
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void chunk (LexState *ls) {
|
|
+ /* chunk -> { stat [`;'] } */
|
|
+ int islast = 0;
|
|
+ enterlevel(ls);
|
|
+ while (!islast && !block_follow(ls->t.token)) {
|
|
+ islast = statement(ls);
|
|
+ testnext(ls, ';');
|
|
+ lua_assert(ls->fs->f->maxstacksize >= ls->fs->freereg &&
|
|
+ ls->fs->freereg >= ls->fs->nactvar);
|
|
+ ls->fs->freereg = ls->fs->nactvar; /* free registers */
|
|
+ }
|
|
+ leavelevel(ls);
|
|
+}
|
|
+
|
|
+/* }====================================================================== */
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lparser.h
|
|
@@ -0,0 +1,82 @@
|
|
+/*
|
|
+** $Id: lparser.h,v 1.57.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lua Parser
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lparser_h
|
|
+#define lparser_h
|
|
+
|
|
+#include "llimits.h"
|
|
+#include "lobject.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+/*
|
|
+** Expression descriptor
|
|
+*/
|
|
+
|
|
+typedef enum {
|
|
+ VVOID, /* no value */
|
|
+ VNIL,
|
|
+ VTRUE,
|
|
+ VFALSE,
|
|
+ VK, /* info = index of constant in `k' */
|
|
+ VKNUM, /* nval = numerical value */
|
|
+ VLOCAL, /* info = local register */
|
|
+ VUPVAL, /* info = index of upvalue in `upvalues' */
|
|
+ VGLOBAL, /* info = index of table; aux = index of global name in `k' */
|
|
+ VINDEXED, /* info = table register; aux = index register (or `k') */
|
|
+ VJMP, /* info = instruction pc */
|
|
+ VRELOCABLE, /* info = instruction pc */
|
|
+ VNONRELOC, /* info = result register */
|
|
+ VCALL, /* info = instruction pc */
|
|
+ VVARARG /* info = instruction pc */
|
|
+} expkind;
|
|
+
|
|
+typedef struct expdesc {
|
|
+ expkind k;
|
|
+ union {
|
|
+ struct { int info, aux; } s;
|
|
+ lua_Number nval;
|
|
+ } u;
|
|
+ int t; /* patch list of `exit when true' */
|
|
+ int f; /* patch list of `exit when false' */
|
|
+} expdesc;
|
|
+
|
|
+
|
|
+typedef struct upvaldesc {
|
|
+ lu_byte k;
|
|
+ lu_byte info;
|
|
+} upvaldesc;
|
|
+
|
|
+
|
|
+struct BlockCnt; /* defined in lparser.c */
|
|
+
|
|
+
|
|
+/* state needed to generate code for a given function */
|
|
+typedef struct FuncState {
|
|
+ Proto *f; /* current function header */
|
|
+ Table *h; /* table to find (and reuse) elements in `k' */
|
|
+ struct FuncState *prev; /* enclosing function */
|
|
+ struct LexState *ls; /* lexical state */
|
|
+ struct lua_State *L; /* copy of the Lua state */
|
|
+ struct BlockCnt *bl; /* chain of current blocks */
|
|
+ int pc; /* next position to code (equivalent to `ncode') */
|
|
+ int lasttarget; /* `pc' of last `jump target' */
|
|
+ int jpc; /* list of pending jumps to `pc' */
|
|
+ int freereg; /* first free register */
|
|
+ int nk; /* number of elements in `k' */
|
|
+ int np; /* number of elements in `p' */
|
|
+ short nlocvars; /* number of elements in `locvars' */
|
|
+ lu_byte nactvar; /* number of active local variables */
|
|
+ upvaldesc upvalues[LUAI_MAXUPVALUES]; /* upvalues */
|
|
+ unsigned short actvar[LUAI_MAXVARS]; /* declared-variable stack */
|
|
+} FuncState;
|
|
+
|
|
+
|
|
+LUAI_FUNC Proto *luaY_parser (lua_State *L, ZIO *z, Mbuffer *buff,
|
|
+ const char *name);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lstate.c
|
|
@@ -0,0 +1,214 @@
|
|
+/*
|
|
+** $Id: lstate.c,v 2.36.1.2 2008/01/03 15:20:39 roberto Exp $
|
|
+** Global State
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#define lstate_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "llex.h"
|
|
+#include "lmem.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+
|
|
+
|
|
+#define state_size(x) (sizeof(x) + LUAI_EXTRASPACE)
|
|
+#define fromstate(l) (cast(lu_byte *, (l)) - LUAI_EXTRASPACE)
|
|
+#define tostate(l) (cast(lua_State *, cast(lu_byte *, l) + LUAI_EXTRASPACE))
|
|
+
|
|
+
|
|
+/*
|
|
+** Main thread combines a thread state and the global state
|
|
+*/
|
|
+typedef struct LG {
|
|
+ lua_State l;
|
|
+ global_State g;
|
|
+} LG;
|
|
+
|
|
+
|
|
+
|
|
+static void stack_init (lua_State *L1, lua_State *L) {
|
|
+ /* initialize CallInfo array */
|
|
+ L1->base_ci = luaM_newvector(L, BASIC_CI_SIZE, CallInfo);
|
|
+ L1->ci = L1->base_ci;
|
|
+ L1->size_ci = BASIC_CI_SIZE;
|
|
+ L1->end_ci = L1->base_ci + L1->size_ci - 1;
|
|
+ /* initialize stack array */
|
|
+ L1->stack = luaM_newvector(L, BASIC_STACK_SIZE + EXTRA_STACK, TValue);
|
|
+ L1->stacksize = BASIC_STACK_SIZE + EXTRA_STACK;
|
|
+ L1->top = L1->stack;
|
|
+ L1->stack_last = L1->stack+(L1->stacksize - EXTRA_STACK)-1;
|
|
+ /* initialize first ci */
|
|
+ L1->ci->func = L1->top;
|
|
+ setnilvalue(L1->top++); /* `function' entry for this `ci' */
|
|
+ L1->base = L1->ci->base = L1->top;
|
|
+ L1->ci->top = L1->top + LUA_MINSTACK;
|
|
+}
|
|
+
|
|
+
|
|
+static void freestack (lua_State *L, lua_State *L1) {
|
|
+ luaM_freearray(L, L1->base_ci, L1->size_ci, CallInfo);
|
|
+ luaM_freearray(L, L1->stack, L1->stacksize, TValue);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** open parts that may cause memory-allocation errors
|
|
+*/
|
|
+static void f_luaopen (lua_State *L, void *ud) {
|
|
+ global_State *g = G(L);
|
|
+ UNUSED(ud);
|
|
+ stack_init(L, L); /* init stack */
|
|
+ sethvalue(L, gt(L), luaH_new(L, 0, 2)); /* table of globals */
|
|
+ sethvalue(L, registry(L), luaH_new(L, 0, 2)); /* registry */
|
|
+ luaS_resize(L, MINSTRTABSIZE); /* initial size of string table */
|
|
+ luaT_init(L);
|
|
+ luaX_init(L);
|
|
+ luaS_fix(luaS_newliteral(L, MEMERRMSG));
|
|
+ g->GCthreshold = 4*g->totalbytes;
|
|
+}
|
|
+
|
|
+
|
|
+static void preinit_state (lua_State *L, global_State *g) {
|
|
+ G(L) = g;
|
|
+ L->stack = NULL;
|
|
+ L->stacksize = 0;
|
|
+ L->errorJmp = NULL;
|
|
+ L->hook = NULL;
|
|
+ L->hookmask = 0;
|
|
+ L->basehookcount = 0;
|
|
+ L->allowhook = 1;
|
|
+ resethookcount(L);
|
|
+ L->openupval = NULL;
|
|
+ L->size_ci = 0;
|
|
+ L->nCcalls = L->baseCcalls = 0;
|
|
+ L->status = 0;
|
|
+ L->base_ci = L->ci = NULL;
|
|
+ L->savedpc = NULL;
|
|
+ L->errfunc = 0;
|
|
+ setnilvalue(gt(L));
|
|
+}
|
|
+
|
|
+
|
|
+static void close_state (lua_State *L) {
|
|
+ global_State *g = G(L);
|
|
+ luaF_close(L, L->stack); /* close all upvalues for this thread */
|
|
+ luaC_freeall(L); /* collect all objects */
|
|
+ lua_assert(g->rootgc == obj2gco(L));
|
|
+ lua_assert(g->strt.nuse == 0);
|
|
+ luaM_freearray(L, G(L)->strt.hash, G(L)->strt.size, TString *);
|
|
+ luaZ_freebuffer(L, &g->buff);
|
|
+ freestack(L, L);
|
|
+ lua_assert(g->totalbytes == sizeof(LG));
|
|
+ (*g->frealloc)(g->ud, fromstate(L), state_size(LG), 0);
|
|
+}
|
|
+
|
|
+
|
|
+lua_State *luaE_newthread (lua_State *L) {
|
|
+ lua_State *L1 = tostate(luaM_malloc(L, state_size(lua_State)));
|
|
+ luaC_link(L, obj2gco(L1), LUA_TTHREAD);
|
|
+ preinit_state(L1, G(L));
|
|
+ stack_init(L1, L); /* init stack */
|
|
+ setobj2n(L, gt(L1), gt(L)); /* share table of globals */
|
|
+ L1->hookmask = L->hookmask;
|
|
+ L1->basehookcount = L->basehookcount;
|
|
+ L1->hook = L->hook;
|
|
+ resethookcount(L1);
|
|
+ lua_assert(iswhite(obj2gco(L1)));
|
|
+ return L1;
|
|
+}
|
|
+
|
|
+
|
|
+void luaE_freethread (lua_State *L, lua_State *L1) {
|
|
+ luaF_close(L1, L1->stack); /* close all upvalues for this thread */
|
|
+ lua_assert(L1->openupval == NULL);
|
|
+ luai_userstatefree(L1);
|
|
+ freestack(L, L1);
|
|
+ luaM_freemem(L, fromstate(L1), state_size(lua_State));
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API lua_State *lua_newstate (lua_Alloc f, void *ud) {
|
|
+ int i;
|
|
+ lua_State *L;
|
|
+ global_State *g;
|
|
+ void *l = (*f)(ud, NULL, 0, state_size(LG));
|
|
+ if (l == NULL) return NULL;
|
|
+ L = tostate(l);
|
|
+ g = &((LG *)L)->g;
|
|
+ L->next = NULL;
|
|
+ L->tt = LUA_TTHREAD;
|
|
+ g->currentwhite = bit2mask(WHITE0BIT, FIXEDBIT);
|
|
+ L->marked = luaC_white(g);
|
|
+ set2bits(L->marked, FIXEDBIT, SFIXEDBIT);
|
|
+ preinit_state(L, g);
|
|
+ g->frealloc = f;
|
|
+ g->ud = ud;
|
|
+ g->mainthread = L;
|
|
+ g->uvhead.u.l.prev = &g->uvhead;
|
|
+ g->uvhead.u.l.next = &g->uvhead;
|
|
+ g->GCthreshold = 0; /* mark it as unfinished state */
|
|
+ g->strt.size = 0;
|
|
+ g->strt.nuse = 0;
|
|
+ g->strt.hash = NULL;
|
|
+ setnilvalue(registry(L));
|
|
+ luaZ_initbuffer(L, &g->buff);
|
|
+ g->panic = NULL;
|
|
+ g->gcstate = GCSpause;
|
|
+ g->rootgc = obj2gco(L);
|
|
+ g->sweepstrgc = 0;
|
|
+ g->sweepgc = &g->rootgc;
|
|
+ g->gray = NULL;
|
|
+ g->grayagain = NULL;
|
|
+ g->weak = NULL;
|
|
+ g->tmudata = NULL;
|
|
+ g->totalbytes = sizeof(LG);
|
|
+ g->gcpause = LUAI_GCPAUSE;
|
|
+ g->gcstepmul = LUAI_GCMUL;
|
|
+ g->gcdept = 0;
|
|
+ for (i=0; i<NUM_TAGS; i++) g->mt[i] = NULL;
|
|
+ if (luaD_rawrunprotected(L, f_luaopen, NULL) != 0) {
|
|
+ /* memory allocation error: free partial state */
|
|
+ close_state(L);
|
|
+ L = NULL;
|
|
+ }
|
|
+ else
|
|
+ luai_userstateopen(L);
|
|
+ return L;
|
|
+}
|
|
+
|
|
+
|
|
+static void callallgcTM (lua_State *L, void *ud) {
|
|
+ UNUSED(ud);
|
|
+ luaC_callGCTM(L); /* call GC metamethods for all udata */
|
|
+}
|
|
+
|
|
+
|
|
+LUA_API void lua_close (lua_State *L) {
|
|
+ L = G(L)->mainthread; /* only the main thread can be closed */
|
|
+ lua_lock(L);
|
|
+ luaF_close(L, L->stack); /* close all upvalues for this thread */
|
|
+ luaC_separateudata(L, 1); /* separate udata that have GC metamethods */
|
|
+ L->errfunc = 0; /* no error function during GC metamethods */
|
|
+ do { /* repeat until no more errors */
|
|
+ L->ci = L->base_ci;
|
|
+ L->base = L->top = L->ci->base;
|
|
+ L->nCcalls = L->baseCcalls = 0;
|
|
+ } while (luaD_rawrunprotected(L, callallgcTM, NULL) != 0);
|
|
+ lua_assert(G(L)->tmudata == NULL);
|
|
+ luai_userstateclose(L);
|
|
+ close_state(L);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lstate.h
|
|
@@ -0,0 +1,169 @@
|
|
+/*
|
|
+** $Id: lstate.h,v 2.24.1.2 2008/01/03 15:20:39 roberto Exp $
|
|
+** Global State
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lstate_h
|
|
+#define lstate_h
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "ltm.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+
|
|
+struct lua_longjmp; /* defined in ldo.c */
|
|
+
|
|
+
|
|
+/* table of globals */
|
|
+#define gt(L) (&L->l_gt)
|
|
+
|
|
+/* registry */
|
|
+#define registry(L) (&G(L)->l_registry)
|
|
+
|
|
+
|
|
+/* extra stack space to handle TM calls and some other extras */
|
|
+#define EXTRA_STACK 5
|
|
+
|
|
+
|
|
+#define BASIC_CI_SIZE 8
|
|
+
|
|
+#define BASIC_STACK_SIZE (2*LUA_MINSTACK)
|
|
+
|
|
+
|
|
+
|
|
+typedef struct stringtable {
|
|
+ GCObject **hash;
|
|
+ lu_int32 nuse; /* number of elements */
|
|
+ int size;
|
|
+} stringtable;
|
|
+
|
|
+
|
|
+/*
|
|
+** informations about a call
|
|
+*/
|
|
+typedef struct CallInfo {
|
|
+ StkId base; /* base for this function */
|
|
+ StkId func; /* function index in the stack */
|
|
+ StkId top; /* top for this function */
|
|
+ const Instruction *savedpc;
|
|
+ int nresults; /* expected number of results from this function */
|
|
+ int tailcalls; /* number of tail calls lost under this entry */
|
|
+} CallInfo;
|
|
+
|
|
+
|
|
+
|
|
+#define curr_func(L) (clvalue(L->ci->func))
|
|
+#define ci_func(ci) (clvalue((ci)->func))
|
|
+#define f_isLua(ci) (!ci_func(ci)->c.isC)
|
|
+#define isLua(ci) (ttisfunction((ci)->func) && f_isLua(ci))
|
|
+
|
|
+
|
|
+/*
|
|
+** `global state', shared by all threads of this state
|
|
+*/
|
|
+typedef struct global_State {
|
|
+ stringtable strt; /* hash table for strings */
|
|
+ lua_Alloc frealloc; /* function to reallocate memory */
|
|
+ void *ud; /* auxiliary data to `frealloc' */
|
|
+ lu_byte currentwhite;
|
|
+ lu_byte gcstate; /* state of garbage collector */
|
|
+ int sweepstrgc; /* position of sweep in `strt' */
|
|
+ GCObject *rootgc; /* list of all collectable objects */
|
|
+ GCObject **sweepgc; /* position of sweep in `rootgc' */
|
|
+ GCObject *gray; /* list of gray objects */
|
|
+ GCObject *grayagain; /* list of objects to be traversed atomically */
|
|
+ GCObject *weak; /* list of weak tables (to be cleared) */
|
|
+ GCObject *tmudata; /* last element of list of userdata to be GC */
|
|
+ Mbuffer buff; /* temporary buffer for string concatentation */
|
|
+ lu_mem GCthreshold;
|
|
+ lu_mem totalbytes; /* number of bytes currently allocated */
|
|
+ lu_mem estimate; /* an estimate of number of bytes actually in use */
|
|
+ lu_mem gcdept; /* how much GC is `behind schedule' */
|
|
+ int gcpause; /* size of pause between successive GCs */
|
|
+ int gcstepmul; /* GC `granularity' */
|
|
+ lua_CFunction panic; /* to be called in unprotected errors */
|
|
+ TValue l_registry;
|
|
+ struct lua_State *mainthread;
|
|
+ UpVal uvhead; /* head of double-linked list of all open upvalues */
|
|
+ struct Table *mt[NUM_TAGS]; /* metatables for basic types */
|
|
+ TString *tmname[TM_N]; /* array with tag-method names */
|
|
+} global_State;
|
|
+
|
|
+
|
|
+/*
|
|
+** `per thread' state
|
|
+*/
|
|
+struct lua_State {
|
|
+ CommonHeader;
|
|
+ lu_byte status;
|
|
+ StkId top; /* first free slot in the stack */
|
|
+ StkId base; /* base of current function */
|
|
+ global_State *l_G;
|
|
+ CallInfo *ci; /* call info for current function */
|
|
+ const Instruction *savedpc; /* `savedpc' of current function */
|
|
+ StkId stack_last; /* last free slot in the stack */
|
|
+ StkId stack; /* stack base */
|
|
+ CallInfo *end_ci; /* points after end of ci array*/
|
|
+ CallInfo *base_ci; /* array of CallInfo's */
|
|
+ int stacksize;
|
|
+ int size_ci; /* size of array `base_ci' */
|
|
+ unsigned short nCcalls; /* number of nested C calls */
|
|
+ unsigned short baseCcalls; /* nested C calls when resuming coroutine */
|
|
+ lu_byte hookmask;
|
|
+ lu_byte allowhook;
|
|
+ int basehookcount;
|
|
+ int hookcount;
|
|
+ lua_Hook hook;
|
|
+ TValue l_gt; /* table of globals */
|
|
+ TValue env; /* temporary place for environments */
|
|
+ GCObject *openupval; /* list of open upvalues in this stack */
|
|
+ GCObject *gclist;
|
|
+ struct lua_longjmp *errorJmp; /* current error recover point */
|
|
+ ptrdiff_t errfunc; /* current error handling function (stack index) */
|
|
+};
|
|
+
|
|
+
|
|
+#define G(L) (L->l_G)
|
|
+
|
|
+
|
|
+/*
|
|
+** Union of all collectable objects
|
|
+*/
|
|
+union GCObject {
|
|
+ GCheader gch;
|
|
+ union TString ts;
|
|
+ union Udata u;
|
|
+ union Closure cl;
|
|
+ struct Table h;
|
|
+ struct Proto p;
|
|
+ struct UpVal uv;
|
|
+ struct lua_State th; /* thread */
|
|
+};
|
|
+
|
|
+
|
|
+/* macros to convert a GCObject into a specific value */
|
|
+#define rawgco2ts(o) check_exp((o)->gch.tt == LUA_TSTRING, &((o)->ts))
|
|
+#define gco2ts(o) (&rawgco2ts(o)->tsv)
|
|
+#define rawgco2u(o) check_exp((o)->gch.tt == LUA_TUSERDATA, &((o)->u))
|
|
+#define gco2u(o) (&rawgco2u(o)->uv)
|
|
+#define gco2cl(o) check_exp((o)->gch.tt == LUA_TFUNCTION, &((o)->cl))
|
|
+#define gco2h(o) check_exp((o)->gch.tt == LUA_TTABLE, &((o)->h))
|
|
+#define gco2p(o) check_exp((o)->gch.tt == LUA_TPROTO, &((o)->p))
|
|
+#define gco2uv(o) check_exp((o)->gch.tt == LUA_TUPVAL, &((o)->uv))
|
|
+#define ngcotouv(o) \
|
|
+ check_exp((o) == NULL || (o)->gch.tt == LUA_TUPVAL, &((o)->uv))
|
|
+#define gco2th(o) check_exp((o)->gch.tt == LUA_TTHREAD, &((o)->th))
|
|
+
|
|
+/* macro to convert any Lua object into a GCObject */
|
|
+#define obj2gco(v) (cast(GCObject *, (v)))
|
|
+
|
|
+
|
|
+LUAI_FUNC lua_State *luaE_newthread (lua_State *L);
|
|
+LUAI_FUNC void luaE_freethread (lua_State *L, lua_State *L1);
|
|
+
|
|
+#endif
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lstring.c
|
|
@@ -0,0 +1,110 @@
|
|
+/*
|
|
+** $Id: lstring.c,v 2.8.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** String table (keeps all strings handled by Lua)
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define lstring_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+
|
|
+
|
|
+
|
|
+void luaS_resize (lua_State *L, int newsize) {
|
|
+ GCObject **newhash;
|
|
+ stringtable *tb;
|
|
+ int i;
|
|
+ if (G(L)->gcstate == GCSsweepstring)
|
|
+ return; /* cannot resize during GC traverse */
|
|
+ newhash = luaM_newvector(L, newsize, GCObject *);
|
|
+ tb = &G(L)->strt;
|
|
+ for (i=0; i<newsize; i++) newhash[i] = NULL;
|
|
+ /* rehash */
|
|
+ for (i=0; i<tb->size; i++) {
|
|
+ GCObject *p = tb->hash[i];
|
|
+ while (p) { /* for each node in the list */
|
|
+ GCObject *next = p->gch.next; /* save next */
|
|
+ unsigned int h = gco2ts(p)->hash;
|
|
+ int h1 = lmod(h, newsize); /* new position */
|
|
+ lua_assert(cast_int(h%newsize) == lmod(h, newsize));
|
|
+ p->gch.next = newhash[h1]; /* chain it */
|
|
+ newhash[h1] = p;
|
|
+ p = next;
|
|
+ }
|
|
+ }
|
|
+ luaM_freearray(L, tb->hash, tb->size, TString *);
|
|
+ tb->size = newsize;
|
|
+ tb->hash = newhash;
|
|
+}
|
|
+
|
|
+
|
|
+static TString *newlstr (lua_State *L, const char *str, size_t l,
|
|
+ unsigned int h) {
|
|
+ TString *ts;
|
|
+ stringtable *tb;
|
|
+ if (l+1 > (MAX_SIZET - sizeof(TString))/sizeof(char))
|
|
+ luaM_toobig(L);
|
|
+ ts = cast(TString *, luaM_malloc(L, (l+1)*sizeof(char)+sizeof(TString)));
|
|
+ ts->tsv.len = l;
|
|
+ ts->tsv.hash = h;
|
|
+ ts->tsv.marked = luaC_white(G(L));
|
|
+ ts->tsv.tt = LUA_TSTRING;
|
|
+ ts->tsv.reserved = 0;
|
|
+ memcpy(ts+1, str, l*sizeof(char));
|
|
+ ((char *)(ts+1))[l] = '\0'; /* ending 0 */
|
|
+ tb = &G(L)->strt;
|
|
+ h = lmod(h, tb->size);
|
|
+ ts->tsv.next = tb->hash[h]; /* chain new entry */
|
|
+ tb->hash[h] = obj2gco(ts);
|
|
+ tb->nuse++;
|
|
+ if (tb->nuse > cast(lu_int32, tb->size) && tb->size <= MAX_INT/2)
|
|
+ luaS_resize(L, tb->size*2); /* too crowded */
|
|
+ return ts;
|
|
+}
|
|
+
|
|
+
|
|
+TString *luaS_newlstr (lua_State *L, const char *str, size_t l) {
|
|
+ GCObject *o;
|
|
+ unsigned int h = cast(unsigned int, l); /* seed */
|
|
+ size_t step = (l>>5)+1; /* if string is too long, don't hash all its chars */
|
|
+ size_t l1;
|
|
+ for (l1=l; l1>=step; l1-=step) /* compute hash */
|
|
+ h = h ^ ((h<<5)+(h>>2)+cast(unsigned char, str[l1-1]));
|
|
+ for (o = G(L)->strt.hash[lmod(h, G(L)->strt.size)];
|
|
+ o != NULL;
|
|
+ o = o->gch.next) {
|
|
+ TString *ts = rawgco2ts(o);
|
|
+ if (ts->tsv.len == l && (memcmp(str, getstr(ts), l) == 0)) {
|
|
+ /* string may be dead */
|
|
+ if (isdead(G(L), o)) changewhite(o);
|
|
+ return ts;
|
|
+ }
|
|
+ }
|
|
+ return newlstr(L, str, l, h); /* not found */
|
|
+}
|
|
+
|
|
+
|
|
+Udata *luaS_newudata (lua_State *L, size_t s, Table *e) {
|
|
+ Udata *u;
|
|
+ if (s > MAX_SIZET - sizeof(Udata))
|
|
+ luaM_toobig(L);
|
|
+ u = cast(Udata *, luaM_malloc(L, s + sizeof(Udata)));
|
|
+ u->uv.marked = luaC_white(G(L)); /* is not finalized */
|
|
+ u->uv.tt = LUA_TUSERDATA;
|
|
+ u->uv.len = s;
|
|
+ u->uv.metatable = NULL;
|
|
+ u->uv.env = e;
|
|
+ /* chain it on udata list (after main thread) */
|
|
+ u->uv.next = G(L)->mainthread->next;
|
|
+ G(L)->mainthread->next = obj2gco(u);
|
|
+ return u;
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lstring.h
|
|
@@ -0,0 +1,31 @@
|
|
+/*
|
|
+** $Id: lstring.h,v 1.43.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** String table (keep all strings handled by Lua)
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lstring_h
|
|
+#define lstring_h
|
|
+
|
|
+
|
|
+#include "lgc.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+
|
|
+
|
|
+#define sizestring(s) (sizeof(union TString)+((s)->len+1)*sizeof(char))
|
|
+
|
|
+#define sizeudata(u) (sizeof(union Udata)+(u)->len)
|
|
+
|
|
+#define luaS_new(L, s) (luaS_newlstr(L, s, strlen(s)))
|
|
+#define luaS_newliteral(L, s) (luaS_newlstr(L, "" s, \
|
|
+ (sizeof(s)/sizeof(char))-1))
|
|
+
|
|
+#define luaS_fix(s) l_setbit((s)->tsv.marked, FIXEDBIT)
|
|
+
|
|
+LUAI_FUNC void luaS_resize (lua_State *L, int newsize);
|
|
+LUAI_FUNC Udata *luaS_newudata (lua_State *L, size_t s, Table *e);
|
|
+LUAI_FUNC TString *luaS_newlstr (lua_State *L, const char *str, size_t l);
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lstrlib.c
|
|
@@ -0,0 +1,883 @@
|
|
+/*
|
|
+** $Id: lstrlib.c,v 1.132.1.4 2008/07/11 17:27:21 roberto Exp $
|
|
+** Standard library for string operations and pattern-matching
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <ctype.h>
|
|
+#include <stddef.h>
|
|
+#include <stdio.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define lstrlib_c
|
|
+#define LUA_LIB
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lauxlib.h"
|
|
+#include "lualib.h"
|
|
+
|
|
+
|
|
+/* macro to `unsign' a character */
|
|
+#define uchar(c) ((unsigned char)(c))
|
|
+
|
|
+
|
|
+
|
|
+static int str_len (lua_State *L) {
|
|
+ size_t l;
|
|
+ luaL_checklstring(L, 1, &l);
|
|
+ lua_pushinteger(L, l);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static ptrdiff_t posrelat (ptrdiff_t pos, size_t len) {
|
|
+ /* relative string position: negative means back from end */
|
|
+ if (pos < 0) pos += (ptrdiff_t)len + 1;
|
|
+ return (pos >= 0) ? pos : 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_sub (lua_State *L) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ ptrdiff_t start = posrelat(luaL_checkinteger(L, 2), l);
|
|
+ ptrdiff_t end = posrelat(luaL_optinteger(L, 3, -1), l);
|
|
+ if (start < 1) start = 1;
|
|
+ if (end > (ptrdiff_t)l) end = (ptrdiff_t)l;
|
|
+ if (start <= end)
|
|
+ lua_pushlstring(L, s+start-1, end-start+1);
|
|
+ else lua_pushliteral(L, "");
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_reverse (lua_State *L) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_reverse: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ while (l--) luaL_addchar(b, s[l]);
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_lower (lua_State *L) {
|
|
+ size_t l;
|
|
+ size_t i;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_lower: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ for (i=0; i<l; i++)
|
|
+ luaL_addchar(b, tolower(uchar(s[i])));
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_upper (lua_State *L) {
|
|
+ size_t l;
|
|
+ size_t i;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_upper: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ for (i=0; i<l; i++)
|
|
+ luaL_addchar(b, toupper(uchar(s[i])));
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int str_rep (lua_State *L) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ int n = luaL_checkint(L, 2);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_rep: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ while (n-- > 0)
|
|
+ luaL_addlstring(b, s, l);
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_byte (lua_State *L) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, 1, &l);
|
|
+ ptrdiff_t posi = posrelat(luaL_optinteger(L, 2, 1), l);
|
|
+ ptrdiff_t pose = posrelat(luaL_optinteger(L, 3, posi), l);
|
|
+ int n, i;
|
|
+ if (posi <= 0) posi = 1;
|
|
+ if ((size_t)pose > l) pose = l;
|
|
+ if (posi > pose) return 0; /* empty interval; return no values */
|
|
+ n = (int)(pose - posi + 1);
|
|
+ if (posi + n <= pose) /* overflow? */
|
|
+ luaL_error(L, "string slice too long");
|
|
+ luaL_checkstack(L, n, "string slice too long");
|
|
+ for (i=0; i<n; i++)
|
|
+ lua_pushinteger(L, uchar(s[posi+i-1]));
|
|
+ return n;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_char (lua_State *L) {
|
|
+ int n = lua_gettop(L); /* number of arguments */
|
|
+ int i;
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_char: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ for (i=1; i<=n; i++) {
|
|
+ int c = luaL_checkint(L, i);
|
|
+ luaL_argcheck(L, uchar(c) == c, i, "invalid value");
|
|
+ luaL_addchar(b, uchar(c));
|
|
+ }
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int writer (lua_State *L, const void* b, size_t size, void* B) {
|
|
+ (void)L;
|
|
+ luaL_addlstring((luaL_Buffer*) B, (const char *)b, size);
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_dump (lua_State *L) {
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_dump: cannot allocate memory");
|
|
+ luaL_checktype(L, 1, LUA_TFUNCTION);
|
|
+ lua_settop(L, 1);
|
|
+ luaL_buffinit(L,b);
|
|
+ if (lua_dump(L, writer, b) != 0){
|
|
+ kfree(b);
|
|
+ luaL_error(L, "unable to dump given function");
|
|
+ }
|
|
+
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** PATTERN MATCHING
|
|
+** =======================================================
|
|
+*/
|
|
+
|
|
+
|
|
+#define CAP_UNFINISHED (-1)
|
|
+#define CAP_POSITION (-2)
|
|
+
|
|
+typedef struct MatchState {
|
|
+ const char *src_init; /* init of source string */
|
|
+ const char *src_end; /* end (`\0') of source string */
|
|
+ lua_State *L;
|
|
+ int level; /* total number of captures (finished or unfinished) */
|
|
+ struct {
|
|
+ const char *init;
|
|
+ ptrdiff_t len;
|
|
+ } capture[LUA_MAXCAPTURES];
|
|
+} MatchState;
|
|
+
|
|
+
|
|
+#define L_ESC '%'
|
|
+#define SPECIALS "^$*+?.([%-"
|
|
+
|
|
+
|
|
+static int check_capture (MatchState *ms, int l) {
|
|
+ l -= '1';
|
|
+ if (l < 0 || l >= ms->level || ms->capture[l].len == CAP_UNFINISHED)
|
|
+ return luaL_error(ms->L, "invalid capture index");
|
|
+ return l;
|
|
+}
|
|
+
|
|
+
|
|
+static int capture_to_close (MatchState *ms) {
|
|
+ int level = ms->level;
|
|
+ for (level--; level>=0; level--)
|
|
+ if (ms->capture[level].len == CAP_UNFINISHED) return level;
|
|
+ return luaL_error(ms->L, "invalid pattern capture");
|
|
+}
|
|
+
|
|
+
|
|
+static const char *classend (MatchState *ms, const char *p) {
|
|
+ switch (*p++) {
|
|
+ case L_ESC: {
|
|
+ if (*p == '\0')
|
|
+ luaL_error(ms->L, "malformed pattern (ends with " LUA_QL("%%") ")");
|
|
+ return p+1;
|
|
+ }
|
|
+ case '[': {
|
|
+ if (*p == '^') p++;
|
|
+ do { /* look for a `]' */
|
|
+ if (*p == '\0')
|
|
+ luaL_error(ms->L, "malformed pattern (missing " LUA_QL("]") ")");
|
|
+ if (*(p++) == L_ESC && *p != '\0')
|
|
+ p++; /* skip escapes (e.g. `%]') */
|
|
+ } while (*p != ']');
|
|
+ return p+1;
|
|
+ }
|
|
+ default: {
|
|
+ return p;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int match_class (int c, int cl) {
|
|
+ int res;
|
|
+ switch (tolower(cl)) {
|
|
+ case 'a' : res = isalpha(c); break;
|
|
+ case 'c' : res = iscntrl(c); break;
|
|
+ case 'd' : res = isdigit(c); break;
|
|
+ case 'l' : res = islower(c); break;
|
|
+ case 'p' : res = ispunct(c); break;
|
|
+ case 's' : res = isspace(c); break;
|
|
+ case 'u' : res = isupper(c); break;
|
|
+ case 'w' : res = isalnum(c); break;
|
|
+ case 'x' : res = isxdigit(c); break;
|
|
+ case 'z' : res = (c == 0); break;
|
|
+ default: return (cl == c);
|
|
+ }
|
|
+ return (islower(cl) ? res : !res);
|
|
+}
|
|
+
|
|
+
|
|
+static int matchbracketclass (int c, const char *p, const char *ec) {
|
|
+ int sig = 1;
|
|
+ if (*(p+1) == '^') {
|
|
+ sig = 0;
|
|
+ p++; /* skip the `^' */
|
|
+ }
|
|
+ while (++p < ec) {
|
|
+ if (*p == L_ESC) {
|
|
+ p++;
|
|
+ if (match_class(c, uchar(*p)))
|
|
+ return sig;
|
|
+ }
|
|
+ else if ((*(p+1) == '-') && (p+2 < ec)) {
|
|
+ p+=2;
|
|
+ if (uchar(*(p-2)) <= c && c <= uchar(*p))
|
|
+ return sig;
|
|
+ }
|
|
+ else if (uchar(*p) == c) return sig;
|
|
+ }
|
|
+ return !sig;
|
|
+}
|
|
+
|
|
+
|
|
+static int singlematch (int c, const char *p, const char *ep) {
|
|
+ switch (*p) {
|
|
+ case '.': return 1; /* matches any char */
|
|
+ case L_ESC: return match_class(c, uchar(*(p+1)));
|
|
+ case '[': return matchbracketclass(c, p, ep-1);
|
|
+ default: return (uchar(*p) == c);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static const char *match (MatchState *ms, const char *s, const char *p);
|
|
+
|
|
+
|
|
+static const char *matchbalance (MatchState *ms, const char *s,
|
|
+ const char *p) {
|
|
+ if (*p == 0 || *(p+1) == 0)
|
|
+ luaL_error(ms->L, "unbalanced pattern");
|
|
+ if (*s != *p) return NULL;
|
|
+ else {
|
|
+ int b = *p;
|
|
+ int e = *(p+1);
|
|
+ int cont = 1;
|
|
+ while (++s < ms->src_end) {
|
|
+ if (*s == e) {
|
|
+ if (--cont == 0) return s+1;
|
|
+ }
|
|
+ else if (*s == b) cont++;
|
|
+ }
|
|
+ }
|
|
+ return NULL; /* string ends out of balance */
|
|
+}
|
|
+
|
|
+
|
|
+static const char *max_expand (MatchState *ms, const char *s,
|
|
+ const char *p, const char *ep) {
|
|
+ ptrdiff_t i = 0; /* counts maximum expand for item */
|
|
+ while ((s+i)<ms->src_end && singlematch(uchar(*(s+i)), p, ep))
|
|
+ i++;
|
|
+ /* keeps trying to match with the maximum repetitions */
|
|
+ while (i>=0) {
|
|
+ const char *res = match(ms, (s+i), ep+1);
|
|
+ if (res) return res;
|
|
+ i--; /* else didn't match; reduce 1 repetition to try again */
|
|
+ }
|
|
+ return NULL;
|
|
+}
|
|
+
|
|
+
|
|
+static const char *min_expand (MatchState *ms, const char *s,
|
|
+ const char *p, const char *ep) {
|
|
+ for (;;) {
|
|
+ const char *res = match(ms, s, ep+1);
|
|
+ if (res != NULL)
|
|
+ return res;
|
|
+ else if (s<ms->src_end && singlematch(uchar(*s), p, ep))
|
|
+ s++; /* try with one more repetition */
|
|
+ else return NULL;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static const char *start_capture (MatchState *ms, const char *s,
|
|
+ const char *p, int what) {
|
|
+ const char *res;
|
|
+ int level = ms->level;
|
|
+ if (level >= LUA_MAXCAPTURES) luaL_error(ms->L, "too many captures");
|
|
+ ms->capture[level].init = s;
|
|
+ ms->capture[level].len = what;
|
|
+ ms->level = level+1;
|
|
+ if ((res=match(ms, s, p)) == NULL) /* match failed? */
|
|
+ ms->level--; /* undo capture */
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+static const char *end_capture (MatchState *ms, const char *s,
|
|
+ const char *p) {
|
|
+ int l = capture_to_close(ms);
|
|
+ const char *res;
|
|
+ ms->capture[l].len = s - ms->capture[l].init; /* close capture */
|
|
+ if ((res = match(ms, s, p)) == NULL) /* match failed? */
|
|
+ ms->capture[l].len = CAP_UNFINISHED; /* undo capture */
|
|
+ return res;
|
|
+}
|
|
+
|
|
+
|
|
+static const char *match_capture (MatchState *ms, const char *s, int l) {
|
|
+ size_t len;
|
|
+ l = check_capture(ms, l);
|
|
+ len = ms->capture[l].len;
|
|
+ if ((size_t)(ms->src_end-s) >= len &&
|
|
+ memcmp(ms->capture[l].init, s, len) == 0)
|
|
+ return s+len;
|
|
+ else return NULL;
|
|
+}
|
|
+
|
|
+
|
|
+static const char *match (MatchState *ms, const char *s, const char *p) {
|
|
+ init: /* using goto's to optimize tail recursion */
|
|
+ switch (*p) {
|
|
+ case '(': { /* start capture */
|
|
+ if (*(p+1) == ')') /* position capture? */
|
|
+ return start_capture(ms, s, p+2, CAP_POSITION);
|
|
+ else
|
|
+ return start_capture(ms, s, p+1, CAP_UNFINISHED);
|
|
+ }
|
|
+ case ')': { /* end capture */
|
|
+ return end_capture(ms, s, p+1);
|
|
+ }
|
|
+ case L_ESC: {
|
|
+ switch (*(p+1)) {
|
|
+ case 'b': { /* balanced string? */
|
|
+ s = matchbalance(ms, s, p+2);
|
|
+ if (s == NULL) return NULL;
|
|
+ p+=4; goto init; /* else return match(ms, s, p+4); */
|
|
+ }
|
|
+ case 'f': { /* frontier? */
|
|
+ const char *ep; char previous;
|
|
+ p += 2;
|
|
+ if (*p != '[')
|
|
+ luaL_error(ms->L, "missing " LUA_QL("[") " after "
|
|
+ LUA_QL("%%f") " in pattern");
|
|
+ ep = classend(ms, p); /* points to what is next */
|
|
+ previous = (s == ms->src_init) ? '\0' : *(s-1);
|
|
+ if (matchbracketclass(uchar(previous), p, ep-1) ||
|
|
+ !matchbracketclass(uchar(*s), p, ep-1)) return NULL;
|
|
+ p=ep; goto init; /* else return match(ms, s, ep); */
|
|
+ }
|
|
+ default: {
|
|
+ if (isdigit(uchar(*(p+1)))) { /* capture results (%0-%9)? */
|
|
+ s = match_capture(ms, s, uchar(*(p+1)));
|
|
+ if (s == NULL) return NULL;
|
|
+ p+=2; goto init; /* else return match(ms, s, p+2) */
|
|
+ }
|
|
+ goto dflt; /* case default */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ case '\0': { /* end of pattern */
|
|
+ return s; /* match succeeded */
|
|
+ }
|
|
+ case '$': {
|
|
+ if (*(p+1) == '\0') /* is the `$' the last char in pattern? */
|
|
+ return (s == ms->src_end) ? s : NULL; /* check end of string */
|
|
+ else goto dflt;
|
|
+ }
|
|
+ default: dflt: { /* it is a pattern item */
|
|
+ const char *ep = classend(ms, p); /* points to what is next */
|
|
+ int m = s<ms->src_end && singlematch(uchar(*s), p, ep);
|
|
+ switch (*ep) {
|
|
+ case '?': { /* optional */
|
|
+ const char *res;
|
|
+ if (m && ((res=match(ms, s+1, ep+1)) != NULL))
|
|
+ return res;
|
|
+ p=ep+1; goto init; /* else return match(ms, s, ep+1); */
|
|
+ }
|
|
+ case '*': { /* 0 or more repetitions */
|
|
+ return max_expand(ms, s, p, ep);
|
|
+ }
|
|
+ case '+': { /* 1 or more repetitions */
|
|
+ return (m ? max_expand(ms, s+1, p, ep) : NULL);
|
|
+ }
|
|
+ case '-': { /* 0 or more repetitions (minimum) */
|
|
+ return min_expand(ms, s, p, ep);
|
|
+ }
|
|
+ default: {
|
|
+ if (!m) return NULL;
|
|
+ s++; p=ep; goto init; /* else return match(ms, s+1, ep); */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+static const char *lmemfind (const char *s1, size_t l1,
|
|
+ const char *s2, size_t l2) {
|
|
+ if (l2 == 0) return s1; /* empty strings are everywhere */
|
|
+ else if (l2 > l1) return NULL; /* avoids a negative `l1' */
|
|
+ else {
|
|
+ const char *init; /* to search for a `*s2' inside `s1' */
|
|
+ l2--; /* 1st char will be checked by `memchr' */
|
|
+ l1 = l1-l2; /* `s2' cannot be found after that */
|
|
+ while (l1 > 0 && (init = (const char *)memchr(s1, *s2, l1)) != NULL) {
|
|
+ init++; /* 1st char is already checked */
|
|
+ if (memcmp(init, s2+1, l2) == 0)
|
|
+ return init-1;
|
|
+ else { /* correct `l1' and `s1' to try again */
|
|
+ l1 -= init-s1;
|
|
+ s1 = init;
|
|
+ }
|
|
+ }
|
|
+ return NULL; /* not found */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void push_onecapture (MatchState *ms, int i, const char *s,
|
|
+ const char *e) {
|
|
+ if (i >= ms->level) {
|
|
+ if (i == 0) /* ms->level == 0, too */
|
|
+ lua_pushlstring(ms->L, s, e - s); /* add whole match */
|
|
+ else
|
|
+ luaL_error(ms->L, "invalid capture index");
|
|
+ }
|
|
+ else {
|
|
+ ptrdiff_t l = ms->capture[i].len;
|
|
+ if (l == CAP_UNFINISHED) luaL_error(ms->L, "unfinished capture");
|
|
+ if (l == CAP_POSITION)
|
|
+ lua_pushinteger(ms->L, ms->capture[i].init - ms->src_init + 1);
|
|
+ else
|
|
+ lua_pushlstring(ms->L, ms->capture[i].init, l);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int push_captures (MatchState *ms, const char *s, const char *e) {
|
|
+ int i;
|
|
+ int nlevels = (ms->level == 0 && s) ? 1 : ms->level;
|
|
+ luaL_checkstack(ms->L, nlevels, "too many captures");
|
|
+ for (i = 0; i < nlevels; i++)
|
|
+ push_onecapture(ms, i, s, e);
|
|
+ return nlevels; /* number of strings pushed */
|
|
+}
|
|
+
|
|
+
|
|
+static int str_find_aux (lua_State *L, int find) {
|
|
+ size_t l1, l2;
|
|
+ const char *s = luaL_checklstring(L, 1, &l1);
|
|
+ const char *p = luaL_checklstring(L, 2, &l2);
|
|
+ ptrdiff_t init = posrelat(luaL_optinteger(L, 3, 1), l1) - 1;
|
|
+ if (init < 0) init = 0;
|
|
+ else if ((size_t)(init) > l1) init = (ptrdiff_t)l1;
|
|
+ if (find && (lua_toboolean(L, 4) || /* explicit request? */
|
|
+ strpbrk(p, SPECIALS) == NULL)) { /* or no special characters? */
|
|
+ /* do a plain search */
|
|
+ const char *s2 = lmemfind(s+init, l1-init, p, l2);
|
|
+ if (s2) {
|
|
+ lua_pushinteger(L, s2-s+1);
|
|
+ lua_pushinteger(L, s2-s+l2);
|
|
+ return 2;
|
|
+ }
|
|
+ }
|
|
+ else {
|
|
+ MatchState ms;
|
|
+ int anchor = (*p == '^') ? (p++, 1) : 0;
|
|
+ const char *s1=s+init;
|
|
+ ms.L = L;
|
|
+ ms.src_init = s;
|
|
+ ms.src_end = s+l1;
|
|
+ do {
|
|
+ const char *res;
|
|
+ ms.level = 0;
|
|
+ if ((res=match(&ms, s1, p)) != NULL) {
|
|
+ if (find) {
|
|
+ lua_pushinteger(L, s1-s+1); /* start */
|
|
+ lua_pushinteger(L, res-s); /* end */
|
|
+ return push_captures(&ms, NULL, 0) + 2;
|
|
+ }
|
|
+ else
|
|
+ return push_captures(&ms, s1, res);
|
|
+ }
|
|
+ } while (s1++ < ms.src_end && !anchor);
|
|
+ }
|
|
+ lua_pushnil(L); /* not found */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int str_find (lua_State *L) {
|
|
+ return str_find_aux(L, 1);
|
|
+}
|
|
+
|
|
+
|
|
+static int str_match (lua_State *L) {
|
|
+ return str_find_aux(L, 0);
|
|
+}
|
|
+
|
|
+
|
|
+static int gmatch_aux (lua_State *L) {
|
|
+ MatchState ms;
|
|
+ size_t ls;
|
|
+ const char *s = lua_tolstring(L, lua_upvalueindex(1), &ls);
|
|
+ const char *p = lua_tostring(L, lua_upvalueindex(2));
|
|
+ const char *src;
|
|
+ ms.L = L;
|
|
+ ms.src_init = s;
|
|
+ ms.src_end = s+ls;
|
|
+ for (src = s + (size_t)lua_tointeger(L, lua_upvalueindex(3));
|
|
+ src <= ms.src_end;
|
|
+ src++) {
|
|
+ const char *e;
|
|
+ ms.level = 0;
|
|
+ if ((e = match(&ms, src, p)) != NULL) {
|
|
+ lua_Integer newstart = e-s;
|
|
+ if (e == src) newstart++; /* empty match? go at least one position */
|
|
+ lua_pushinteger(L, newstart);
|
|
+ lua_replace(L, lua_upvalueindex(3));
|
|
+ return push_captures(&ms, src, e);
|
|
+ }
|
|
+ }
|
|
+ return 0; /* not found */
|
|
+}
|
|
+
|
|
+
|
|
+static int gmatch (lua_State *L) {
|
|
+ luaL_checkstring(L, 1);
|
|
+ luaL_checkstring(L, 2);
|
|
+ lua_settop(L, 2);
|
|
+ lua_pushinteger(L, 0);
|
|
+ lua_pushcclosure(L, gmatch_aux, 3);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int gfind_nodef (lua_State *L) {
|
|
+ return luaL_error(L, LUA_QL("string.gfind") " was renamed to "
|
|
+ LUA_QL("string.gmatch"));
|
|
+}
|
|
+
|
|
+
|
|
+static void add_s (MatchState *ms, luaL_Buffer *b, const char *s,
|
|
+ const char *e) {
|
|
+ size_t l, i;
|
|
+ const char *news = lua_tolstring(ms->L, 3, &l);
|
|
+ for (i = 0; i < l; i++) {
|
|
+ if (news[i] != L_ESC)
|
|
+ luaL_addchar(b, news[i]);
|
|
+ else {
|
|
+ i++; /* skip ESC */
|
|
+ if (!isdigit(uchar(news[i])))
|
|
+ luaL_addchar(b, news[i]);
|
|
+ else if (news[i] == '0')
|
|
+ luaL_addlstring(b, s, e - s);
|
|
+ else {
|
|
+ push_onecapture(ms, news[i] - '1', s, e);
|
|
+ luaL_addvalue(b); /* add capture to accumulated result */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void add_value (MatchState *ms, luaL_Buffer *b, const char *s,
|
|
+ const char *e) {
|
|
+ lua_State *L = ms->L;
|
|
+ switch (lua_type(L, 3)) {
|
|
+ case LUA_TNUMBER:
|
|
+ case LUA_TSTRING: {
|
|
+ add_s(ms, b, s, e);
|
|
+ return;
|
|
+ }
|
|
+ case LUA_TFUNCTION: {
|
|
+ int n;
|
|
+ lua_pushvalue(L, 3);
|
|
+ n = push_captures(ms, s, e);
|
|
+ lua_call(L, n, 1);
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TTABLE: {
|
|
+ push_onecapture(ms, 0, s, e);
|
|
+ lua_gettable(L, 3);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ if (!lua_toboolean(L, -1)) { /* nil or false? */
|
|
+ lua_pop(L, 1);
|
|
+ lua_pushlstring(L, s, e - s); /* keep original text */
|
|
+ }
|
|
+ else if (!lua_isstring(L, -1))
|
|
+ luaL_error(L, "invalid replacement value (a %s)", luaL_typename(L, -1));
|
|
+ luaL_addvalue(b); /* add result to accumulator */
|
|
+}
|
|
+
|
|
+
|
|
+static int str_gsub (lua_State *L) {
|
|
+ size_t srcl;
|
|
+ const char *src = luaL_checklstring(L, 1, &srcl);
|
|
+ const char *p = luaL_checkstring(L, 2);
|
|
+ int tr = lua_type(L, 3);
|
|
+ int max_s = luaL_optint(L, 4, srcl+1);
|
|
+ int anchor = (*p == '^') ? (p++, 1) : 0;
|
|
+ int n = 0;
|
|
+ MatchState ms;
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_gsub: cannot allocate memory");
|
|
+ luaL_argcheck(L, tr == LUA_TNUMBER || tr == LUA_TSTRING ||
|
|
+ tr == LUA_TFUNCTION || tr == LUA_TTABLE, 3,
|
|
+ "string/function/table expected");
|
|
+ luaL_buffinit(L, b);
|
|
+ ms.L = L;
|
|
+ ms.src_init = src;
|
|
+ ms.src_end = src+srcl;
|
|
+ while (n < max_s) {
|
|
+ const char *e;
|
|
+ ms.level = 0;
|
|
+ e = match(&ms, src, p);
|
|
+ if (e) {
|
|
+ n++;
|
|
+ add_value(&ms, b, src, e);
|
|
+ }
|
|
+ if (e && e>src) /* non empty match? */
|
|
+ src = e; /* skip it */
|
|
+ else if (src < ms.src_end)
|
|
+ luaL_addchar(b, *src++);
|
|
+ else break;
|
|
+ if (anchor) break;
|
|
+ }
|
|
+ luaL_addlstring(b, src, ms.src_end-src);
|
|
+ luaL_pushresult(b);
|
|
+ lua_pushinteger(L, n); /* number of substitutions */
|
|
+ kfree(b);
|
|
+ return 2;
|
|
+}
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+/* maximum size of each formatted item (> len(format('%99.99f', -1e308))) */
|
|
+#define MAX_ITEM 512
|
|
+/* valid flags in a format specification */
|
|
+#define FLAGS "-+ #0"
|
|
+/*
|
|
+** maximum size of each format specification (such as '%-099.99d')
|
|
+** (+10 accounts for %99.99x plus margin of error)
|
|
+*/
|
|
+#define MAX_FORMAT (sizeof(FLAGS) + sizeof(LUA_INTFRMLEN) + 10)
|
|
+
|
|
+
|
|
+static void addquoted (lua_State *L, luaL_Buffer *b, int arg) {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, arg, &l);
|
|
+ luaL_addchar(b, '"');
|
|
+ while (l--) {
|
|
+ switch (*s) {
|
|
+ case '"': case '\\': case '\n': {
|
|
+ luaL_addchar(b, '\\');
|
|
+ luaL_addchar(b, *s);
|
|
+ break;
|
|
+ }
|
|
+ case '\r': {
|
|
+ luaL_addlstring(b, "\\r", 2);
|
|
+ break;
|
|
+ }
|
|
+ case '\0': {
|
|
+ luaL_addlstring(b, "\\000", 4);
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ luaL_addchar(b, *s);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ s++;
|
|
+ }
|
|
+ luaL_addchar(b, '"');
|
|
+}
|
|
+
|
|
+static const char *scanformat (lua_State *L, const char *strfrmt, char *form) {
|
|
+ const char *p = strfrmt;
|
|
+ while (*p != '\0' && strchr(FLAGS, *p) != NULL) p++; /* skip flags */
|
|
+ if ((size_t)(p - strfrmt) >= sizeof(FLAGS))
|
|
+ luaL_error(L, "invalid format (repeated flags)");
|
|
+ if (isdigit(uchar(*p))) p++; /* skip width */
|
|
+ if (isdigit(uchar(*p))) p++; /* (2 digits at most) */
|
|
+ if (*p == '.') {
|
|
+ p++;
|
|
+ if (isdigit(uchar(*p))) p++; /* skip precision */
|
|
+ if (isdigit(uchar(*p))) p++; /* (2 digits at most) */
|
|
+ }
|
|
+ if (isdigit(uchar(*p)))
|
|
+ luaL_error(L, "invalid format (width or precision too long)");
|
|
+ *(form++) = '%';
|
|
+ strncpy(form, strfrmt, p - strfrmt + 1);
|
|
+ form += p - strfrmt + 1;
|
|
+ *form = '\0';
|
|
+ return p;
|
|
+}
|
|
+
|
|
+
|
|
+static void addintlen (char *form) {
|
|
+ size_t l = strlen(form);
|
|
+ char spec = form[l - 1];
|
|
+ strcpy(form + l - 1, LUA_INTFRMLEN);
|
|
+ form[l + sizeof(LUA_INTFRMLEN) - 2] = spec;
|
|
+ form[l + sizeof(LUA_INTFRMLEN) - 1] = '\0';
|
|
+}
|
|
+
|
|
+
|
|
+static int str_format (lua_State *L) {
|
|
+ int arg = 1;
|
|
+ size_t sfl;
|
|
+ const char *strfrmt = luaL_checklstring(L, arg, &sfl);
|
|
+ const char *strfrmt_end = strfrmt+sfl;
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "str_format: cannot allocate memory");
|
|
+ luaL_buffinit(L, b);
|
|
+ while (strfrmt < strfrmt_end) {
|
|
+ if (*strfrmt != L_ESC)
|
|
+ luaL_addchar(b, *strfrmt++);
|
|
+ else if (*++strfrmt == L_ESC)
|
|
+ luaL_addchar(b, *strfrmt++); /* %% */
|
|
+ else { /* format item */
|
|
+ char form[MAX_FORMAT]; /* to store the format (`%...') */
|
|
+ char buff[MAX_ITEM]; /* to store the formatted item */
|
|
+ arg++;
|
|
+ strfrmt = scanformat(L, strfrmt, form);
|
|
+ switch (*strfrmt++) {
|
|
+ case 'c': {
|
|
+ sprintf(buff, form, (int)luaL_checknumber(L, arg));
|
|
+ break;
|
|
+ }
|
|
+ case 'd': case 'i': {
|
|
+ addintlen(form);
|
|
+ sprintf(buff, form, (LUA_INTFRM_T)luaL_checknumber(L, arg));
|
|
+ break;
|
|
+ }
|
|
+ case 'o': case 'u': case 'x': case 'X': {
|
|
+ addintlen(form);
|
|
+ sprintf(buff, form, (unsigned LUA_INTFRM_T)luaL_checknumber(L, arg));
|
|
+ break;
|
|
+ }
|
|
+ case 'q': {
|
|
+ addquoted(L, b, arg);
|
|
+ continue; /* skip the 'addsize' at the end */
|
|
+ }
|
|
+ case 's': {
|
|
+ size_t l;
|
|
+ const char *s = luaL_checklstring(L, arg, &l);
|
|
+ if (!strchr(form, '.') && l >= 100) {
|
|
+ /* no precision and string is too long to be formatted;
|
|
+ keep original string */
|
|
+ lua_pushvalue(L, arg);
|
|
+ luaL_addvalue(b);
|
|
+ continue; /* skip the `addsize' at the end */
|
|
+ }
|
|
+ else {
|
|
+ sprintf(buff, form, s);
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ default: { /* also treat cases `pnLlh' */
|
|
+ kfree(b);
|
|
+ return luaL_error(L, "invalid option " LUA_QL("%%%c") " to "
|
|
+ LUA_QL("format"), *(strfrmt - 1));
|
|
+ }
|
|
+ }
|
|
+ luaL_addlstring(b, buff, strlen(buff));
|
|
+ }
|
|
+ }
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static const luaL_Reg strlib[] = {
|
|
+ {"byte", str_byte},
|
|
+ {"char", str_char},
|
|
+ {"dump", str_dump},
|
|
+ {"find", str_find},
|
|
+ {"format", str_format},
|
|
+ {"gfind", gfind_nodef},
|
|
+ {"gmatch", gmatch},
|
|
+ {"gsub", str_gsub},
|
|
+ {"len", str_len},
|
|
+ {"lower", str_lower},
|
|
+ {"match", str_match},
|
|
+ {"rep", str_rep},
|
|
+ {"reverse", str_reverse},
|
|
+ {"sub", str_sub},
|
|
+ {"upper", str_upper},
|
|
+ {NULL, NULL}
|
|
+};
|
|
+
|
|
+
|
|
+static void createmetatable (lua_State *L) {
|
|
+ lua_createtable(L, 0, 1); /* create metatable for strings */
|
|
+ lua_pushliteral(L, ""); /* dummy string */
|
|
+ lua_pushvalue(L, -2);
|
|
+ lua_setmetatable(L, -2); /* set string metatable */
|
|
+ lua_pop(L, 1); /* pop dummy string */
|
|
+ lua_pushvalue(L, -2); /* string library... */
|
|
+ lua_setfield(L, -2, "__index"); /* ...is the __index metamethod */
|
|
+ lua_pop(L, 1); /* pop metatable */
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Open string library
|
|
+*/
|
|
+LUALIB_API int luaopen_string (lua_State *L) {
|
|
+ luaL_register(L, LUA_STRLIBNAME, strlib);
|
|
+#if defined(LUA_COMPAT_GFIND)
|
|
+ lua_getfield(L, -1, "gmatch");
|
|
+ lua_setfield(L, -2, "gfind");
|
|
+#endif
|
|
+ createmetatable(L);
|
|
+ return 1;
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ltable.c
|
|
@@ -0,0 +1,588 @@
|
|
+/*
|
|
+** $Id: ltable.c,v 2.32.1.2 2007/12/28 15:32:23 roberto Exp $
|
|
+** Lua tables (hash)
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+/*
|
|
+** Implementation of tables (aka arrays, objects, or hash tables).
|
|
+** Tables keep its elements in two parts: an array part and a hash part.
|
|
+** Non-negative integer keys are all candidates to be kept in the array
|
|
+** part. The actual size of the array is the largest `n' such that at
|
|
+** least half the slots between 0 and n are in use.
|
|
+** Hash uses a mix of chained scatter table with Brent's variation.
|
|
+** A main invariant of these tables is that, if an element is not
|
|
+** in its main position (i.e. the `original' position that its hash gives
|
|
+** to it), then the colliding element is in its own main position.
|
|
+** Hence even when the load factor reaches 100%, performance remains good.
|
|
+*/
|
|
+
|
|
+#include <math.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define ltable_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lgc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "ltable.h"
|
|
+
|
|
+
|
|
+/*
|
|
+** max size of array part is 2^MAXBITS
|
|
+*/
|
|
+#if LUAI_BITSINT > 26
|
|
+#define MAXBITS 26
|
|
+#else
|
|
+#define MAXBITS (LUAI_BITSINT-2)
|
|
+#endif
|
|
+
|
|
+#define MAXASIZE (1 << MAXBITS)
|
|
+
|
|
+
|
|
+#define hashpow2(t,n) (gnode(t, lmod((n), sizenode(t))))
|
|
+
|
|
+#define hashstr(t,str) hashpow2(t, (str)->tsv.hash)
|
|
+#define hashboolean(t,p) hashpow2(t, p)
|
|
+
|
|
+
|
|
+/*
|
|
+** for some types, it is better to avoid modulus by power of 2, as
|
|
+** they tend to have many 2 factors.
|
|
+*/
|
|
+#define hashmod(t,n) (gnode(t, ((n) % ((sizenode(t)-1)|1))))
|
|
+
|
|
+
|
|
+#define hashpointer(t,p) hashmod(t, IntPoint(p))
|
|
+
|
|
+
|
|
+/*
|
|
+** number of ints inside a lua_Number
|
|
+*/
|
|
+#define numints cast_int(sizeof(lua_Number)/sizeof(int))
|
|
+
|
|
+
|
|
+
|
|
+#define dummynode (&dummynode_)
|
|
+
|
|
+static const Node dummynode_ = {
|
|
+ {{NULL}, LUA_TNIL}, /* value */
|
|
+ {{{NULL}, LUA_TNIL, NULL}} /* key */
|
|
+};
|
|
+
|
|
+
|
|
+/*
|
|
+** hash for lua_Numbers
|
|
+*/
|
|
+static Node *hashnum (const Table *t, lua_Number n) {
|
|
+ unsigned int a[numints];
|
|
+ int i;
|
|
+ if (luai_numeq(n, 0)) /* avoid problems with -0 */
|
|
+ return gnode(t, 0);
|
|
+ memcpy(a, &n, sizeof(a));
|
|
+ for (i = 1; i < numints; i++) a[0] += a[i];
|
|
+ return hashmod(t, a[0]);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** returns the `main' position of an element in a table (that is, the index
|
|
+** of its hash value)
|
|
+*/
|
|
+static Node *mainposition (const Table *t, const TValue *key) {
|
|
+ switch (ttype(key)) {
|
|
+ case LUA_TNUMBER:
|
|
+ return hashnum(t, nvalue(key));
|
|
+ case LUA_TSTRING:
|
|
+ return hashstr(t, rawtsvalue(key));
|
|
+ case LUA_TBOOLEAN:
|
|
+ return hashboolean(t, bvalue(key));
|
|
+ case LUA_TLIGHTUSERDATA:
|
|
+ return hashpointer(t, pvalue(key));
|
|
+ default:
|
|
+ return hashpointer(t, gcvalue(key));
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** returns the index for `key' if `key' is an appropriate key to live in
|
|
+** the array part of the table, -1 otherwise.
|
|
+*/
|
|
+static int arrayindex (const TValue *key) {
|
|
+ if (ttisnumber(key)) {
|
|
+ lua_Number n = nvalue(key);
|
|
+ int k;
|
|
+ lua_number2int(k, n);
|
|
+ if (luai_numeq(cast_num(k), n))
|
|
+ return k;
|
|
+ }
|
|
+ return -1; /* `key' did not match some condition */
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** returns the index of a `key' for table traversals. First goes all
|
|
+** elements in the array part, then elements in the hash part. The
|
|
+** beginning of a traversal is signalled by -1.
|
|
+*/
|
|
+static int findindex (lua_State *L, Table *t, StkId key) {
|
|
+ int i;
|
|
+ if (ttisnil(key)) return -1; /* first iteration */
|
|
+ i = arrayindex(key);
|
|
+ if (0 < i && i <= t->sizearray) /* is `key' inside array part? */
|
|
+ return i-1; /* yes; that's the index (corrected to C) */
|
|
+ else {
|
|
+ Node *n = mainposition(t, key);
|
|
+ do { /* check whether `key' is somewhere in the chain */
|
|
+ /* key may be dead already, but it is ok to use it in `next' */
|
|
+ if (luaO_rawequalObj(key2tval(n), key) ||
|
|
+ (ttype(gkey(n)) == LUA_TDEADKEY && iscollectable(key) &&
|
|
+ gcvalue(gkey(n)) == gcvalue(key))) {
|
|
+ i = cast_int(n - gnode(t, 0)); /* key index in hash table */
|
|
+ /* hash elements are numbered after array ones */
|
|
+ return i + t->sizearray;
|
|
+ }
|
|
+ else n = gnext(n);
|
|
+ } while (n);
|
|
+ luaG_runerror(L, "invalid key to " LUA_QL("next")); /* key not found */
|
|
+ return 0; /* to avoid warnings */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+int luaH_next (lua_State *L, Table *t, StkId key) {
|
|
+ int i = findindex(L, t, key); /* find original element */
|
|
+ for (i++; i < t->sizearray; i++) { /* try first array part */
|
|
+ if (!ttisnil(&t->array[i])) { /* a non-nil value? */
|
|
+ setnvalue(key, cast_num(i+1));
|
|
+ setobj2s(L, key+1, &t->array[i]);
|
|
+ return 1;
|
|
+ }
|
|
+ }
|
|
+ for (i -= t->sizearray; i < sizenode(t); i++) { /* then hash part */
|
|
+ if (!ttisnil(gval(gnode(t, i)))) { /* a non-nil value? */
|
|
+ setobj2s(L, key, key2tval(gnode(t, i)));
|
|
+ setobj2s(L, key+1, gval(gnode(t, i)));
|
|
+ return 1;
|
|
+ }
|
|
+ }
|
|
+ return 0; /* no more elements */
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** {=============================================================
|
|
+** Rehash
|
|
+** ==============================================================
|
|
+*/
|
|
+
|
|
+
|
|
+static int computesizes (int nums[], int *narray) {
|
|
+ int i;
|
|
+ int twotoi; /* 2^i */
|
|
+ int a = 0; /* number of elements smaller than 2^i */
|
|
+ int na = 0; /* number of elements to go to array part */
|
|
+ int n = 0; /* optimal size for array part */
|
|
+ for (i = 0, twotoi = 1; twotoi/2 < *narray; i++, twotoi *= 2) {
|
|
+ if (nums[i] > 0) {
|
|
+ a += nums[i];
|
|
+ if (a > twotoi/2) { /* more than half elements present? */
|
|
+ n = twotoi; /* optimal size (till now) */
|
|
+ na = a; /* all elements smaller than n will go to array part */
|
|
+ }
|
|
+ }
|
|
+ if (a == *narray) break; /* all elements already counted */
|
|
+ }
|
|
+ *narray = n;
|
|
+ lua_assert(*narray/2 <= na && na <= *narray);
|
|
+ return na;
|
|
+}
|
|
+
|
|
+
|
|
+static int countint (const TValue *key, int *nums) {
|
|
+ int k = arrayindex(key);
|
|
+ if (0 < k && k <= MAXASIZE) { /* is `key' an appropriate array index? */
|
|
+ nums[ceillog2(k)]++; /* count as such */
|
|
+ return 1;
|
|
+ }
|
|
+ else
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int numusearray (const Table *t, int *nums) {
|
|
+ int lg;
|
|
+ int ttlg; /* 2^lg */
|
|
+ int ause = 0; /* summation of `nums' */
|
|
+ int i = 1; /* count to traverse all array keys */
|
|
+ for (lg=0, ttlg=1; lg<=MAXBITS; lg++, ttlg*=2) { /* for each slice */
|
|
+ int lc = 0; /* counter */
|
|
+ int lim = ttlg;
|
|
+ if (lim > t->sizearray) {
|
|
+ lim = t->sizearray; /* adjust upper limit */
|
|
+ if (i > lim)
|
|
+ break; /* no more elements to count */
|
|
+ }
|
|
+ /* count elements in range (2^(lg-1), 2^lg] */
|
|
+ for (; i <= lim; i++) {
|
|
+ if (!ttisnil(&t->array[i-1]))
|
|
+ lc++;
|
|
+ }
|
|
+ nums[lg] += lc;
|
|
+ ause += lc;
|
|
+ }
|
|
+ return ause;
|
|
+}
|
|
+
|
|
+
|
|
+static int numusehash (const Table *t, int *nums, int *pnasize) {
|
|
+ int totaluse = 0; /* total number of elements */
|
|
+ int ause = 0; /* summation of `nums' */
|
|
+ int i = sizenode(t);
|
|
+ while (i--) {
|
|
+ Node *n = &t->node[i];
|
|
+ if (!ttisnil(gval(n))) {
|
|
+ ause += countint(key2tval(n), nums);
|
|
+ totaluse++;
|
|
+ }
|
|
+ }
|
|
+ *pnasize += ause;
|
|
+ return totaluse;
|
|
+}
|
|
+
|
|
+
|
|
+static void setarrayvector (lua_State *L, Table *t, int size) {
|
|
+ int i;
|
|
+ luaM_reallocvector(L, t->array, t->sizearray, size, TValue);
|
|
+ for (i=t->sizearray; i<size; i++)
|
|
+ setnilvalue(&t->array[i]);
|
|
+ t->sizearray = size;
|
|
+}
|
|
+
|
|
+
|
|
+static void setnodevector (lua_State *L, Table *t, int size) {
|
|
+ int lsize;
|
|
+ if (size == 0) { /* no elements to hash part? */
|
|
+ t->node = cast(Node *, dummynode); /* use common `dummynode' */
|
|
+ lsize = 0;
|
|
+ }
|
|
+ else {
|
|
+ int i;
|
|
+ lsize = ceillog2(size);
|
|
+ if (lsize > MAXBITS)
|
|
+ luaG_runerror(L, "table overflow");
|
|
+ size = twoto(lsize);
|
|
+ t->node = luaM_newvector(L, size, Node);
|
|
+ for (i=0; i<size; i++) {
|
|
+ Node *n = gnode(t, i);
|
|
+ gnext(n) = NULL;
|
|
+ setnilvalue(gkey(n));
|
|
+ setnilvalue(gval(n));
|
|
+ }
|
|
+ }
|
|
+ t->lsizenode = cast_byte(lsize);
|
|
+ t->lastfree = gnode(t, size); /* all positions are free */
|
|
+}
|
|
+
|
|
+
|
|
+static void resize (lua_State *L, Table *t, int nasize, int nhsize) {
|
|
+ int i;
|
|
+ int oldasize = t->sizearray;
|
|
+ int oldhsize = t->lsizenode;
|
|
+ Node *nold = t->node; /* save old hash ... */
|
|
+ if (nasize > oldasize) /* array part must grow? */
|
|
+ setarrayvector(L, t, nasize);
|
|
+ /* create new hash part with appropriate size */
|
|
+ setnodevector(L, t, nhsize);
|
|
+ if (nasize < oldasize) { /* array part must shrink? */
|
|
+ t->sizearray = nasize;
|
|
+ /* re-insert elements from vanishing slice */
|
|
+ for (i=nasize; i<oldasize; i++) {
|
|
+ if (!ttisnil(&t->array[i]))
|
|
+ setobjt2t(L, luaH_setnum(L, t, i+1), &t->array[i]);
|
|
+ }
|
|
+ /* shrink array */
|
|
+ luaM_reallocvector(L, t->array, oldasize, nasize, TValue);
|
|
+ }
|
|
+ /* re-insert elements from hash part */
|
|
+ for (i = twoto(oldhsize) - 1; i >= 0; i--) {
|
|
+ Node *old = nold+i;
|
|
+ if (!ttisnil(gval(old)))
|
|
+ setobjt2t(L, luaH_set(L, t, key2tval(old)), gval(old));
|
|
+ }
|
|
+ if (nold != dummynode)
|
|
+ luaM_freearray(L, nold, twoto(oldhsize), Node); /* free old array */
|
|
+}
|
|
+
|
|
+
|
|
+void luaH_resizearray (lua_State *L, Table *t, int nasize) {
|
|
+ int nsize = (t->node == dummynode) ? 0 : sizenode(t);
|
|
+ resize(L, t, nasize, nsize);
|
|
+}
|
|
+
|
|
+
|
|
+static void rehash (lua_State *L, Table *t, const TValue *ek) {
|
|
+ int nasize, na;
|
|
+ int nums[MAXBITS+1]; /* nums[i] = number of keys between 2^(i-1) and 2^i */
|
|
+ int i;
|
|
+ int totaluse;
|
|
+ for (i=0; i<=MAXBITS; i++) nums[i] = 0; /* reset counts */
|
|
+ nasize = numusearray(t, nums); /* count keys in array part */
|
|
+ totaluse = nasize; /* all those keys are integer keys */
|
|
+ totaluse += numusehash(t, nums, &nasize); /* count keys in hash part */
|
|
+ /* count extra key */
|
|
+ nasize += countint(ek, nums);
|
|
+ totaluse++;
|
|
+ /* compute new size for array part */
|
|
+ na = computesizes(nums, &nasize);
|
|
+ /* resize the table to new computed sizes */
|
|
+ resize(L, t, nasize, totaluse - na);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** }=============================================================
|
|
+*/
|
|
+
|
|
+
|
|
+Table *luaH_new (lua_State *L, int narray, int nhash) {
|
|
+ Table *t = luaM_new(L, Table);
|
|
+ luaC_link(L, obj2gco(t), LUA_TTABLE);
|
|
+ t->metatable = NULL;
|
|
+ t->flags = cast_byte(~0);
|
|
+ /* temporary values (kept only if some malloc fails) */
|
|
+ t->array = NULL;
|
|
+ t->sizearray = 0;
|
|
+ t->lsizenode = 0;
|
|
+ t->node = cast(Node *, dummynode);
|
|
+ setarrayvector(L, t, narray);
|
|
+ setnodevector(L, t, nhash);
|
|
+ return t;
|
|
+}
|
|
+
|
|
+
|
|
+void luaH_free (lua_State *L, Table *t) {
|
|
+ if (t->node != dummynode)
|
|
+ luaM_freearray(L, t->node, sizenode(t), Node);
|
|
+ luaM_freearray(L, t->array, t->sizearray, TValue);
|
|
+ luaM_free(L, t);
|
|
+}
|
|
+
|
|
+
|
|
+static Node *getfreepos (Table *t) {
|
|
+ while (t->lastfree-- > t->node) {
|
|
+ if (ttisnil(gkey(t->lastfree)))
|
|
+ return t->lastfree;
|
|
+ }
|
|
+ return NULL; /* could not find a free place */
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** inserts a new key into a hash table; first, check whether key's main
|
|
+** position is free. If not, check whether colliding node is in its main
|
|
+** position or not: if it is not, move colliding node to an empty place and
|
|
+** put new key in its main position; otherwise (colliding node is in its main
|
|
+** position), new key goes to an empty position.
|
|
+*/
|
|
+static TValue *newkey (lua_State *L, Table *t, const TValue *key) {
|
|
+ Node *mp = mainposition(t, key);
|
|
+ if (!ttisnil(gval(mp)) || mp == dummynode) {
|
|
+ Node *othern;
|
|
+ Node *n = getfreepos(t); /* get a free place */
|
|
+ if (n == NULL) { /* cannot find a free place? */
|
|
+ rehash(L, t, key); /* grow table */
|
|
+ return luaH_set(L, t, key); /* re-insert key into grown table */
|
|
+ }
|
|
+ lua_assert(n != dummynode);
|
|
+ othern = mainposition(t, key2tval(mp));
|
|
+ if (othern != mp) { /* is colliding node out of its main position? */
|
|
+ /* yes; move colliding node into free position */
|
|
+ while (gnext(othern) != mp) othern = gnext(othern); /* find previous */
|
|
+ gnext(othern) = n; /* redo the chain with `n' in place of `mp' */
|
|
+ *n = *mp; /* copy colliding node into free pos. (mp->next also goes) */
|
|
+ gnext(mp) = NULL; /* now `mp' is free */
|
|
+ setnilvalue(gval(mp));
|
|
+ }
|
|
+ else { /* colliding node is in its own main position */
|
|
+ /* new node will go into free position */
|
|
+ gnext(n) = gnext(mp); /* chain new position */
|
|
+ gnext(mp) = n;
|
|
+ mp = n;
|
|
+ }
|
|
+ }
|
|
+ gkey(mp)->value = key->value; gkey(mp)->tt = key->tt;
|
|
+ luaC_barriert(L, t, key);
|
|
+ lua_assert(ttisnil(gval(mp)));
|
|
+ return gval(mp);
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** search function for integers
|
|
+*/
|
|
+const TValue *luaH_getnum (Table *t, int key) {
|
|
+ /* (1 <= key && key <= t->sizearray) */
|
|
+ if (cast(unsigned int, key-1) < cast(unsigned int, t->sizearray))
|
|
+ return &t->array[key-1];
|
|
+ else {
|
|
+ lua_Number nk = cast_num(key);
|
|
+ Node *n = hashnum(t, nk);
|
|
+ do { /* check whether `key' is somewhere in the chain */
|
|
+ if (ttisnumber(gkey(n)) && luai_numeq(nvalue(gkey(n)), nk))
|
|
+ return gval(n); /* that's it */
|
|
+ else n = gnext(n);
|
|
+ } while (n);
|
|
+ return luaO_nilobject;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** search function for strings
|
|
+*/
|
|
+const TValue *luaH_getstr (Table *t, TString *key) {
|
|
+ Node *n = hashstr(t, key);
|
|
+ do { /* check whether `key' is somewhere in the chain */
|
|
+ if (ttisstring(gkey(n)) && rawtsvalue(gkey(n)) == key)
|
|
+ return gval(n); /* that's it */
|
|
+ else n = gnext(n);
|
|
+ } while (n);
|
|
+ return luaO_nilobject;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** main search function
|
|
+*/
|
|
+const TValue *luaH_get (Table *t, const TValue *key) {
|
|
+ switch (ttype(key)) {
|
|
+ case LUA_TNIL: return luaO_nilobject;
|
|
+ case LUA_TSTRING: return luaH_getstr(t, rawtsvalue(key));
|
|
+ case LUA_TNUMBER: {
|
|
+ int k;
|
|
+ lua_Number n = nvalue(key);
|
|
+ lua_number2int(k, n);
|
|
+ if (luai_numeq(cast_num(k), nvalue(key))) /* index is int? */
|
|
+ return luaH_getnum(t, k); /* use specialized version */
|
|
+ /* else go through */
|
|
+ }
|
|
+ default: {
|
|
+ Node *n = mainposition(t, key);
|
|
+ do { /* check whether `key' is somewhere in the chain */
|
|
+ if (luaO_rawequalObj(key2tval(n), key))
|
|
+ return gval(n); /* that's it */
|
|
+ else n = gnext(n);
|
|
+ } while (n);
|
|
+ return luaO_nilobject;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+TValue *luaH_set (lua_State *L, Table *t, const TValue *key) {
|
|
+ const TValue *p = luaH_get(t, key);
|
|
+ t->flags = 0;
|
|
+ if (p != luaO_nilobject)
|
|
+ return cast(TValue *, p);
|
|
+ else {
|
|
+ if (ttisnil(key)) luaG_runerror(L, "table index is nil");
|
|
+ else if (ttisnumber(key) && luai_numisnan(nvalue(key)))
|
|
+ luaG_runerror(L, "table index is NaN");
|
|
+ return newkey(L, t, key);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+TValue *luaH_setnum (lua_State *L, Table *t, int key) {
|
|
+ const TValue *p = luaH_getnum(t, key);
|
|
+ if (p != luaO_nilobject)
|
|
+ return cast(TValue *, p);
|
|
+ else {
|
|
+ TValue k;
|
|
+ setnvalue(&k, cast_num(key));
|
|
+ return newkey(L, t, &k);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+TValue *luaH_setstr (lua_State *L, Table *t, TString *key) {
|
|
+ const TValue *p = luaH_getstr(t, key);
|
|
+ if (p != luaO_nilobject)
|
|
+ return cast(TValue *, p);
|
|
+ else {
|
|
+ TValue k;
|
|
+ setsvalue(L, &k, key);
|
|
+ return newkey(L, t, &k);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static int unbound_search (Table *t, unsigned int j) {
|
|
+ unsigned int i = j; /* i is zero or a present index */
|
|
+ j++;
|
|
+ /* find `i' and `j' such that i is present and j is not */
|
|
+ while (!ttisnil(luaH_getnum(t, j))) {
|
|
+ i = j;
|
|
+ j *= 2;
|
|
+ if (j > cast(unsigned int, MAX_INT)) { /* overflow? */
|
|
+ /* table was built with bad purposes: resort to linear search */
|
|
+ i = 1;
|
|
+ while (!ttisnil(luaH_getnum(t, i))) i++;
|
|
+ return i - 1;
|
|
+ }
|
|
+ }
|
|
+ /* now do a binary search between them */
|
|
+ while (j - i > 1) {
|
|
+ unsigned int m = (i+j)/2;
|
|
+ if (ttisnil(luaH_getnum(t, m))) j = m;
|
|
+ else i = m;
|
|
+ }
|
|
+ return i;
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** Try to find a boundary in table `t'. A `boundary' is an integer index
|
|
+** such that t[i] is non-nil and t[i+1] is nil (and 0 if t[1] is nil).
|
|
+*/
|
|
+int luaH_getn (Table *t) {
|
|
+ unsigned int j = t->sizearray;
|
|
+ if (j > 0 && ttisnil(&t->array[j - 1])) {
|
|
+ /* there is a boundary in the array part: (binary) search for it */
|
|
+ unsigned int i = 0;
|
|
+ while (j - i > 1) {
|
|
+ unsigned int m = (i+j)/2;
|
|
+ if (ttisnil(&t->array[m - 1])) j = m;
|
|
+ else i = m;
|
|
+ }
|
|
+ return i;
|
|
+ }
|
|
+ /* else must find a boundary in hash part */
|
|
+ else if (t->node == dummynode) /* hash part is empty? */
|
|
+ return j; /* that is easy... */
|
|
+ else return unbound_search(t, j);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+#if defined(LUA_DEBUG)
|
|
+
|
|
+Node *luaH_mainposition (const Table *t, const TValue *key) {
|
|
+ return mainposition(t, key);
|
|
+}
|
|
+
|
|
+int luaH_isdummy (Node *n) { return n == dummynode; }
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ltable.h
|
|
@@ -0,0 +1,40 @@
|
|
+/*
|
|
+** $Id: ltable.h,v 2.10.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lua tables (hash)
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef ltable_h
|
|
+#define ltable_h
|
|
+
|
|
+#include "lobject.h"
|
|
+
|
|
+
|
|
+#define gnode(t,i) (&(t)->node[i])
|
|
+#define gkey(n) (&(n)->i_key.nk)
|
|
+#define gval(n) (&(n)->i_val)
|
|
+#define gnext(n) ((n)->i_key.nk.next)
|
|
+
|
|
+#define key2tval(n) (&(n)->i_key.tvk)
|
|
+
|
|
+
|
|
+LUAI_FUNC const TValue *luaH_getnum (Table *t, int key);
|
|
+LUAI_FUNC TValue *luaH_setnum (lua_State *L, Table *t, int key);
|
|
+LUAI_FUNC const TValue *luaH_getstr (Table *t, TString *key);
|
|
+LUAI_FUNC TValue *luaH_setstr (lua_State *L, Table *t, TString *key);
|
|
+LUAI_FUNC const TValue *luaH_get (Table *t, const TValue *key);
|
|
+LUAI_FUNC TValue *luaH_set (lua_State *L, Table *t, const TValue *key);
|
|
+LUAI_FUNC Table *luaH_new (lua_State *L, int narray, int lnhash);
|
|
+LUAI_FUNC void luaH_resizearray (lua_State *L, Table *t, int nasize);
|
|
+LUAI_FUNC void luaH_free (lua_State *L, Table *t);
|
|
+LUAI_FUNC int luaH_next (lua_State *L, Table *t, StkId key);
|
|
+LUAI_FUNC int luaH_getn (Table *t);
|
|
+
|
|
+
|
|
+#if defined(LUA_DEBUG)
|
|
+LUAI_FUNC Node *luaH_mainposition (const Table *t, const TValue *key);
|
|
+LUAI_FUNC int luaH_isdummy (Node *n);
|
|
+#endif
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ltablib.c
|
|
@@ -0,0 +1,288 @@
|
|
+/*
|
|
+** $Id: ltablib.c,v 1.38.1.3 2008/02/14 16:46:58 roberto Exp $
|
|
+** Library for Table Manipulation
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#define ltablib_c
|
|
+#define LUA_LIB
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lauxlib.h"
|
|
+#include "lualib.h"
|
|
+
|
|
+
|
|
+#define aux_getn(L,n) (luaL_checktype(L, n, LUA_TTABLE), luaL_getn(L, n))
|
|
+
|
|
+
|
|
+static int foreachi (lua_State *L) {
|
|
+ int i;
|
|
+ int n = aux_getn(L, 1);
|
|
+ luaL_checktype(L, 2, LUA_TFUNCTION);
|
|
+ for (i=1; i <= n; i++) {
|
|
+ lua_pushvalue(L, 2); /* function */
|
|
+ lua_pushinteger(L, i); /* 1st argument */
|
|
+ lua_rawgeti(L, 1, i); /* 2nd argument */
|
|
+ lua_call(L, 2, 1);
|
|
+ if (!lua_isnil(L, -1))
|
|
+ return 1;
|
|
+ lua_pop(L, 1); /* remove nil result */
|
|
+ }
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int foreach (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ luaL_checktype(L, 2, LUA_TFUNCTION);
|
|
+ lua_pushnil(L); /* first key */
|
|
+ while (lua_next(L, 1)) {
|
|
+ lua_pushvalue(L, 2); /* function */
|
|
+ lua_pushvalue(L, -3); /* key */
|
|
+ lua_pushvalue(L, -3); /* value */
|
|
+ lua_call(L, 2, 1);
|
|
+ if (!lua_isnil(L, -1))
|
|
+ return 1;
|
|
+ lua_pop(L, 2); /* remove value and result */
|
|
+ }
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int maxn (lua_State *L) {
|
|
+ lua_Number max = 0;
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ lua_pushnil(L); /* first key */
|
|
+ while (lua_next(L, 1)) {
|
|
+ lua_pop(L, 1); /* remove value */
|
|
+ if (lua_type(L, -1) == LUA_TNUMBER) {
|
|
+ lua_Number v = lua_tonumber(L, -1);
|
|
+ if (v > max) max = v;
|
|
+ }
|
|
+ }
|
|
+ lua_pushnumber(L, max);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int getn (lua_State *L) {
|
|
+ lua_pushinteger(L, aux_getn(L, 1));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int setn (lua_State *L) {
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+#ifndef luaL_setn
|
|
+ luaL_setn(L, 1, luaL_checkint(L, 2));
|
|
+#else
|
|
+ luaL_error(L, LUA_QL("setn") " is obsolete");
|
|
+#endif
|
|
+ lua_pushvalue(L, 1);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static int tinsert (lua_State *L) {
|
|
+ int e = aux_getn(L, 1) + 1; /* first empty element */
|
|
+ int pos; /* where to insert new element */
|
|
+ switch (lua_gettop(L)) {
|
|
+ case 2: { /* called with only 2 arguments */
|
|
+ pos = e; /* insert new element at the end */
|
|
+ break;
|
|
+ }
|
|
+ case 3: {
|
|
+ int i;
|
|
+ pos = luaL_checkint(L, 2); /* 2nd argument is the position */
|
|
+ if (pos > e) e = pos; /* `grow' array if necessary */
|
|
+ for (i = e; i > pos; i--) { /* move up elements */
|
|
+ lua_rawgeti(L, 1, i-1);
|
|
+ lua_rawseti(L, 1, i); /* t[i] = t[i-1] */
|
|
+ }
|
|
+ break;
|
|
+ }
|
|
+ default: {
|
|
+ return luaL_error(L, "wrong number of arguments to " LUA_QL("insert"));
|
|
+ }
|
|
+ }
|
|
+ luaL_setn(L, 1, e); /* new size */
|
|
+ lua_rawseti(L, 1, pos); /* t[pos] = v */
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static int tremove (lua_State *L) {
|
|
+ int e = aux_getn(L, 1);
|
|
+ int pos = luaL_optint(L, 2, e);
|
|
+ if (!(1 <= pos && pos <= e)) /* position is outside bounds? */
|
|
+ return 0; /* nothing to remove */
|
|
+ luaL_setn(L, 1, e - 1); /* t.n = n-1 */
|
|
+ lua_rawgeti(L, 1, pos); /* result = t[pos] */
|
|
+ for ( ;pos<e; pos++) {
|
|
+ lua_rawgeti(L, 1, pos+1);
|
|
+ lua_rawseti(L, 1, pos); /* t[pos] = t[pos+1] */
|
|
+ }
|
|
+ lua_pushnil(L);
|
|
+ lua_rawseti(L, 1, e); /* t[e] = nil */
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static void addfield (lua_State *L, luaL_Buffer *b, int i) {
|
|
+ lua_rawgeti(L, 1, i);
|
|
+ if (!lua_isstring(L, -1))
|
|
+ luaL_error(L, "invalid value (%s) at index %d in table for "
|
|
+ LUA_QL("concat"), luaL_typename(L, -1), i);
|
|
+ luaL_addvalue(b);
|
|
+}
|
|
+
|
|
+
|
|
+static int tconcat (lua_State *L) {
|
|
+ size_t lsep;
|
|
+ int i, last;
|
|
+ const char *sep = luaL_optlstring(L, 2, "", &lsep);
|
|
+ luaL_Buffer *b = (luaL_Buffer *)kmalloc(sizeof(luaL_Buffer) + BUFSIZ, GFP_ATOMIC);
|
|
+ if(!b) luaL_error(L, "tconcat: cannot allocate memory");
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+ i = luaL_optint(L, 3, 1);
|
|
+ last = luaL_opt(L, luaL_checkint, 4, luaL_getn(L, 1));
|
|
+ luaL_buffinit(L, b);
|
|
+ for (; i < last; i++) {
|
|
+ addfield(L, b, i);
|
|
+ luaL_addlstring(b, sep, lsep);
|
|
+ }
|
|
+ if (i == last) /* add last value (if interval was not empty) */
|
|
+ addfield(L, b, i);
|
|
+ luaL_pushresult(b);
|
|
+ kfree(b);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================
|
|
+** Quicksort
|
|
+** (based on `Algorithms in MODULA-3', Robert Sedgewick;
|
|
+** Addison-Wesley, 1993.)
|
|
+*/
|
|
+
|
|
+
|
|
+static void set2 (lua_State *L, int i, int j) {
|
|
+ lua_rawseti(L, 1, i);
|
|
+ lua_rawseti(L, 1, j);
|
|
+}
|
|
+
|
|
+static int sort_comp (lua_State *L, int a, int b) {
|
|
+ if (!lua_isnil(L, 2)) { /* function? */
|
|
+ int res;
|
|
+ lua_pushvalue(L, 2);
|
|
+ lua_pushvalue(L, a-1); /* -1 to compensate function */
|
|
+ lua_pushvalue(L, b-2); /* -2 to compensate function and `a' */
|
|
+ lua_call(L, 2, 1);
|
|
+ res = lua_toboolean(L, -1);
|
|
+ lua_pop(L, 1);
|
|
+ return res;
|
|
+ }
|
|
+ else /* a < b? */
|
|
+ return lua_lessthan(L, a, b);
|
|
+}
|
|
+
|
|
+static void auxsort (lua_State *L, int l, int u) {
|
|
+ while (l < u) { /* for tail recursion */
|
|
+ int i, j;
|
|
+ /* sort elements a[l], a[(l+u)/2] and a[u] */
|
|
+ lua_rawgeti(L, 1, l);
|
|
+ lua_rawgeti(L, 1, u);
|
|
+ if (sort_comp(L, -1, -2)) /* a[u] < a[l]? */
|
|
+ set2(L, l, u); /* swap a[l] - a[u] */
|
|
+ else
|
|
+ lua_pop(L, 2);
|
|
+ if (u-l == 1) break; /* only 2 elements */
|
|
+ i = (l+u)/2;
|
|
+ lua_rawgeti(L, 1, i);
|
|
+ lua_rawgeti(L, 1, l);
|
|
+ if (sort_comp(L, -2, -1)) /* a[i]<a[l]? */
|
|
+ set2(L, i, l);
|
|
+ else {
|
|
+ lua_pop(L, 1); /* remove a[l] */
|
|
+ lua_rawgeti(L, 1, u);
|
|
+ if (sort_comp(L, -1, -2)) /* a[u]<a[i]? */
|
|
+ set2(L, i, u);
|
|
+ else
|
|
+ lua_pop(L, 2);
|
|
+ }
|
|
+ if (u-l == 2) break; /* only 3 elements */
|
|
+ lua_rawgeti(L, 1, i); /* Pivot */
|
|
+ lua_pushvalue(L, -1);
|
|
+ lua_rawgeti(L, 1, u-1);
|
|
+ set2(L, i, u-1);
|
|
+ /* a[l] <= P == a[u-1] <= a[u], only need to sort from l+1 to u-2 */
|
|
+ i = l; j = u-1;
|
|
+ for (;;) { /* invariant: a[l..i] <= P <= a[j..u] */
|
|
+ /* repeat ++i until a[i] >= P */
|
|
+ while (lua_rawgeti(L, 1, ++i), sort_comp(L, -1, -2)) {
|
|
+ if (i>u) luaL_error(L, "invalid order function for sorting");
|
|
+ lua_pop(L, 1); /* remove a[i] */
|
|
+ }
|
|
+ /* repeat --j until a[j] <= P */
|
|
+ while (lua_rawgeti(L, 1, --j), sort_comp(L, -3, -1)) {
|
|
+ if (j<l) luaL_error(L, "invalid order function for sorting");
|
|
+ lua_pop(L, 1); /* remove a[j] */
|
|
+ }
|
|
+ if (j<i) {
|
|
+ lua_pop(L, 3); /* pop pivot, a[i], a[j] */
|
|
+ break;
|
|
+ }
|
|
+ set2(L, i, j);
|
|
+ }
|
|
+ lua_rawgeti(L, 1, u-1);
|
|
+ lua_rawgeti(L, 1, i);
|
|
+ set2(L, u-1, i); /* swap pivot (a[u-1]) with a[i] */
|
|
+ /* a[l..i-1] <= a[i] == P <= a[i+1..u] */
|
|
+ /* adjust so that smaller half is in [j..i] and larger one in [l..u] */
|
|
+ if (i-l < u-i) {
|
|
+ j=l; i=i-1; l=i+2;
|
|
+ }
|
|
+ else {
|
|
+ j=i+1; i=u; u=j-2;
|
|
+ }
|
|
+ auxsort(L, j, i); /* call recursively the smaller one */
|
|
+ } /* repeat the routine for the larger one */
|
|
+}
|
|
+
|
|
+static int sort (lua_State *L) {
|
|
+ int n = aux_getn(L, 1);
|
|
+ luaL_checkstack(L, 40, ""); /* assume array is smaller than 2^40 */
|
|
+ if (!lua_isnoneornil(L, 2)) /* is there a 2nd argument? */
|
|
+ luaL_checktype(L, 2, LUA_TFUNCTION);
|
|
+ lua_settop(L, 2); /* make sure there is two arguments */
|
|
+ auxsort(L, 1, n);
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+/* }====================================================== */
|
|
+
|
|
+
|
|
+static const luaL_Reg tab_funcs[] = {
|
|
+ {"concat", tconcat},
|
|
+ {"foreach", foreach},
|
|
+ {"foreachi", foreachi},
|
|
+ {"getn", getn},
|
|
+ {"maxn", maxn},
|
|
+ {"insert", tinsert},
|
|
+ {"remove", tremove},
|
|
+ {"setn", setn},
|
|
+ {"sort", sort},
|
|
+ {NULL, NULL}
|
|
+};
|
|
+
|
|
+
|
|
+LUALIB_API int luaopen_table (lua_State *L) {
|
|
+ luaL_register(L, LUA_TABLIBNAME, tab_funcs);
|
|
+ return 1;
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ltm.c
|
|
@@ -0,0 +1,74 @@
|
|
+/*
|
|
+** $Id: ltm.c,v 2.8.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Tag methods
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define ltm_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+
|
|
+
|
|
+
|
|
+const char *const luaT_typenames[] = {
|
|
+ "nil", "boolean", "userdata", "number",
|
|
+ "string", "table", "function", "userdata", "thread",
|
|
+ "proto", "upval"
|
|
+};
|
|
+
|
|
+
|
|
+void luaT_init (lua_State *L) {
|
|
+ static const char *const luaT_eventname[] = { /* ORDER TM */
|
|
+ "__index", "__newindex",
|
|
+ "__gc", "__mode", "__eq",
|
|
+ "__add", "__sub", "__mul", "__div", "__mod",
|
|
+ "__pow", "__unm", "__len", "__lt", "__le",
|
|
+ "__concat", "__call"
|
|
+ };
|
|
+ int i;
|
|
+ for (i=0; i<TM_N; i++) {
|
|
+ G(L)->tmname[i] = luaS_new(L, luaT_eventname[i]);
|
|
+ luaS_fix(G(L)->tmname[i]); /* never collect these names */
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+/*
|
|
+** function to be used with macro "fasttm": optimized for absence of
|
|
+** tag methods
|
|
+*/
|
|
+const TValue *luaT_gettm (Table *events, TMS event, TString *ename) {
|
|
+ const TValue *tm = luaH_getstr(events, ename);
|
|
+ lua_assert(event <= TM_EQ);
|
|
+ if (ttisnil(tm)) { /* no tag method? */
|
|
+ events->flags |= cast_byte(1u<<event); /* cache this fact */
|
|
+ return NULL;
|
|
+ }
|
|
+ else return tm;
|
|
+}
|
|
+
|
|
+
|
|
+const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o, TMS event) {
|
|
+ Table *mt;
|
|
+ switch (ttype(o)) {
|
|
+ case LUA_TTABLE:
|
|
+ mt = hvalue(o)->metatable;
|
|
+ break;
|
|
+ case LUA_TUSERDATA:
|
|
+ mt = uvalue(o)->metatable;
|
|
+ break;
|
|
+ default:
|
|
+ mt = G(L)->mt[ttype(o)];
|
|
+ }
|
|
+ return (mt ? luaH_getstr(mt, G(L)->tmname[event]) : luaO_nilobject);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/ltm.h
|
|
@@ -0,0 +1,54 @@
|
|
+/*
|
|
+** $Id: ltm.h,v 2.6.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Tag methods
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef ltm_h
|
|
+#define ltm_h
|
|
+
|
|
+
|
|
+#include "lobject.h"
|
|
+
|
|
+
|
|
+/*
|
|
+* WARNING: if you change the order of this enumeration,
|
|
+* grep "ORDER TM"
|
|
+*/
|
|
+typedef enum {
|
|
+ TM_INDEX,
|
|
+ TM_NEWINDEX,
|
|
+ TM_GC,
|
|
+ TM_MODE,
|
|
+ TM_EQ, /* last tag method with `fast' access */
|
|
+ TM_ADD,
|
|
+ TM_SUB,
|
|
+ TM_MUL,
|
|
+ TM_DIV,
|
|
+ TM_MOD,
|
|
+ TM_POW,
|
|
+ TM_UNM,
|
|
+ TM_LEN,
|
|
+ TM_LT,
|
|
+ TM_LE,
|
|
+ TM_CONCAT,
|
|
+ TM_CALL,
|
|
+ TM_N /* number of elements in the enum */
|
|
+} TMS;
|
|
+
|
|
+
|
|
+
|
|
+#define gfasttm(g,et,e) ((et) == NULL ? NULL : \
|
|
+ ((et)->flags & (1u<<(e))) ? NULL : luaT_gettm(et, e, (g)->tmname[e]))
|
|
+
|
|
+#define fasttm(l,et,e) gfasttm(G(l), et, e)
|
|
+
|
|
+LUAI_DATA const char *const luaT_typenames[];
|
|
+
|
|
+
|
|
+LUAI_FUNC const TValue *luaT_gettm (Table *events, TMS event, TString *ename);
|
|
+LUAI_FUNC const TValue *luaT_gettmbyobj (lua_State *L, const TValue *o,
|
|
+ TMS event);
|
|
+LUAI_FUNC void luaT_init (lua_State *L);
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/luaconf.h
|
|
@@ -0,0 +1,797 @@
|
|
+/*
|
|
+** $Id: luaconf.h,v 1.82.1.7 2008/02/11 16:25:08 roberto Exp $
|
|
+** Configuration file for Lua
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lconfig_h
|
|
+#define lconfig_h
|
|
+
|
|
+#include <stddef.h>
|
|
+
|
|
+#if !defined(__KERNEL__)
|
|
+#include <limits.h>
|
|
+#else
|
|
+#define UCHAR_MAX 255
|
|
+#define SHRT_MAX 32767
|
|
+#define BUFSIZ 8192
|
|
+#define NO_FPU
|
|
+#endif
|
|
+
|
|
+/*
|
|
+** ==================================================================
|
|
+** Search for "@@" to find all configurable definitions.
|
|
+** ===================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_ANSI controls the use of non-ansi features.
|
|
+** CHANGE it (define it) if you want Lua to avoid the use of any
|
|
+** non-ansi feature or library.
|
|
+*/
|
|
+#if defined(__STRICT_ANSI__)
|
|
+#define LUA_ANSI
|
|
+#endif
|
|
+
|
|
+
|
|
+#if !defined(LUA_ANSI) && defined(_WIN32)
|
|
+#define LUA_WIN
|
|
+#endif
|
|
+
|
|
+#if defined(LUA_USE_LINUX)
|
|
+#define LUA_USE_POSIX
|
|
+#define LUA_USE_DLOPEN /* needs an extra library: -ldl */
|
|
+#define LUA_USE_READLINE /* needs some extra libraries */
|
|
+#endif
|
|
+
|
|
+#if defined(LUA_USE_MACOSX)
|
|
+#define LUA_USE_POSIX
|
|
+#define LUA_DL_DYLD /* does not need extra library */
|
|
+#endif
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_USE_POSIX includes all functionallity listed as X/Open System
|
|
+@* Interfaces Extension (XSI).
|
|
+** CHANGE it (define it) if your system is XSI compatible.
|
|
+*/
|
|
+#if defined(LUA_USE_POSIX)
|
|
+#define LUA_USE_MKSTEMP
|
|
+#define LUA_USE_ISATTY
|
|
+#define LUA_USE_POPEN
|
|
+#define LUA_USE_ULONGJMP
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_PATH and LUA_CPATH are the names of the environment variables that
|
|
+@* Lua check to set its paths.
|
|
+@@ LUA_INIT is the name of the environment variable that Lua
|
|
+@* checks for initialization code.
|
|
+** CHANGE them if you want different names.
|
|
+*/
|
|
+#define LUA_PATH "LUA_PATH"
|
|
+#define LUA_CPATH "LUA_CPATH"
|
|
+#define LUA_INIT "LUA_INIT"
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_PATH_DEFAULT is the default path that Lua uses to look for
|
|
+@* Lua libraries.
|
|
+@@ LUA_CPATH_DEFAULT is the default path that Lua uses to look for
|
|
+@* C libraries.
|
|
+** CHANGE them if your machine has a non-conventional directory
|
|
+** hierarchy or if you want to install your libraries in
|
|
+** non-conventional directories.
|
|
+*/
|
|
+#if defined(_WIN32)
|
|
+/*
|
|
+** In Windows, any exclamation mark ('!') in the path is replaced by the
|
|
+** path of the directory of the executable file of the current process.
|
|
+*/
|
|
+#define LUA_LDIR "!\\lua\\"
|
|
+#define LUA_CDIR "!\\"
|
|
+#define LUA_PATH_DEFAULT \
|
|
+ ".\\?.lua;" LUA_LDIR"?.lua;" LUA_LDIR"?\\init.lua;" \
|
|
+ LUA_CDIR"?.lua;" LUA_CDIR"?\\init.lua"
|
|
+#define LUA_CPATH_DEFAULT \
|
|
+ ".\\?.dll;" LUA_CDIR"?.dll;" LUA_CDIR"loadall.dll"
|
|
+
|
|
+#else
|
|
+#define LUA_ROOT "/usr/local/"
|
|
+#define LUA_LDIR LUA_ROOT "share/lua/5.1/"
|
|
+#define LUA_CDIR LUA_ROOT "lib/lua/5.1/"
|
|
+#define LUA_PATH_DEFAULT \
|
|
+ "./?.lua;" LUA_LDIR"?.lua;" LUA_LDIR"?/init.lua;" \
|
|
+ LUA_CDIR"?.lua;" LUA_CDIR"?/init.lua"
|
|
+#define LUA_CPATH_DEFAULT \
|
|
+ "./?.so;" LUA_CDIR"?.so;" LUA_CDIR"loadall.so"
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_DIRSEP is the directory separator (for submodules).
|
|
+** CHANGE it if your machine does not use "/" as the directory separator
|
|
+** and is not Windows. (On Windows Lua automatically uses "\".)
|
|
+*/
|
|
+#if defined(_WIN32)
|
|
+#define LUA_DIRSEP "\\"
|
|
+#else
|
|
+#define LUA_DIRSEP "/"
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_PATHSEP is the character that separates templates in a path.
|
|
+@@ LUA_PATH_MARK is the string that marks the substitution points in a
|
|
+@* template.
|
|
+@@ LUA_EXECDIR in a Windows path is replaced by the executable's
|
|
+@* directory.
|
|
+@@ LUA_IGMARK is a mark to ignore all before it when bulding the
|
|
+@* luaopen_ function name.
|
|
+** CHANGE them if for some reason your system cannot use those
|
|
+** characters. (E.g., if one of those characters is a common character
|
|
+** in file/directory names.) Probably you do not need to change them.
|
|
+*/
|
|
+#define LUA_PATHSEP ";"
|
|
+#define LUA_PATH_MARK "?"
|
|
+#define LUA_EXECDIR "!"
|
|
+#define LUA_IGMARK "-"
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_INTEGER is the integral type used by lua_pushinteger/lua_tointeger.
|
|
+** CHANGE that if ptrdiff_t is not adequate on your machine. (On most
|
|
+** machines, ptrdiff_t gives a good choice between int or long.)
|
|
+*/
|
|
+#define LUA_INTEGER ptrdiff_t
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_API is a mark for all core API functions.
|
|
+@@ LUALIB_API is a mark for all standard library functions.
|
|
+** CHANGE them if you need to define those functions in some special way.
|
|
+** For instance, if you want to create one Windows DLL with the core and
|
|
+** the libraries, you may want to use the following definition (define
|
|
+** LUA_BUILD_AS_DLL to get it).
|
|
+*/
|
|
+#if defined(LUA_BUILD_AS_DLL)
|
|
+
|
|
+#if defined(LUA_CORE) || defined(LUA_LIB)
|
|
+#define LUA_API __declspec(dllexport)
|
|
+#else
|
|
+#define LUA_API __declspec(dllimport)
|
|
+#endif
|
|
+
|
|
+#else
|
|
+
|
|
+#define LUA_API extern
|
|
+
|
|
+#endif
|
|
+
|
|
+/* more often than not the libs go together with the core */
|
|
+#define LUALIB_API LUA_API
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_FUNC is a mark for all extern functions that are not to be
|
|
+@* exported to outside modules.
|
|
+@@ LUAI_DATA is a mark for all extern (const) variables that are not to
|
|
+@* be exported to outside modules.
|
|
+** CHANGE them if you need to mark them in some special way. Elf/gcc
|
|
+** (versions 3.2 and later) mark them as "hidden" to optimize access
|
|
+** when Lua is compiled as a shared library.
|
|
+*/
|
|
+#if defined(luaall_c)
|
|
+#define LUAI_FUNC static
|
|
+#define LUAI_DATA /* empty */
|
|
+
|
|
+#elif defined(__GNUC__) && ((__GNUC__*100 + __GNUC_MINOR__) >= 302) && \
|
|
+ defined(__ELF__)
|
|
+#define LUAI_FUNC __attribute__((visibility("hidden"))) extern
|
|
+#define LUAI_DATA LUAI_FUNC
|
|
+
|
|
+#else
|
|
+#define LUAI_FUNC extern
|
|
+#define LUAI_DATA extern
|
|
+#endif
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_QL describes how error messages quote program elements.
|
|
+** CHANGE it if you want a different appearance.
|
|
+*/
|
|
+#define LUA_QL(x) "'" x "'"
|
|
+#define LUA_QS LUA_QL("%s")
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_IDSIZE gives the maximum size for the description of the source
|
|
+@* of a function in debug information.
|
|
+** CHANGE it if you want a different size.
|
|
+*/
|
|
+#define LUA_IDSIZE 60
|
|
+
|
|
+
|
|
+/*
|
|
+** {==================================================================
|
|
+** Stand-alone configuration
|
|
+** ===================================================================
|
|
+*/
|
|
+
|
|
+#if defined(lua_c) || defined(luaall_c)
|
|
+
|
|
+/*
|
|
+@@ lua_stdin_is_tty detects whether the standard input is a 'tty' (that
|
|
+@* is, whether we're running lua interactively).
|
|
+** CHANGE it if you have a better definition for non-POSIX/non-Windows
|
|
+** systems.
|
|
+*/
|
|
+#if defined(LUA_USE_ISATTY)
|
|
+#include <unistd.h>
|
|
+#define lua_stdin_is_tty() isatty(0)
|
|
+#elif defined(LUA_WIN)
|
|
+#include <io.h>
|
|
+#include <stdio.h>
|
|
+#define lua_stdin_is_tty() _isatty(_fileno(stdin))
|
|
+#else
|
|
+#define lua_stdin_is_tty() 1 /* assume stdin is a tty */
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_PROMPT is the default prompt used by stand-alone Lua.
|
|
+@@ LUA_PROMPT2 is the default continuation prompt used by stand-alone Lua.
|
|
+** CHANGE them if you want different prompts. (You can also change the
|
|
+** prompts dynamically, assigning to globals _PROMPT/_PROMPT2.)
|
|
+*/
|
|
+#define LUA_PROMPT "> "
|
|
+#define LUA_PROMPT2 ">> "
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_PROGNAME is the default name for the stand-alone Lua program.
|
|
+** CHANGE it if your stand-alone interpreter has a different name and
|
|
+** your system is not able to detect that name automatically.
|
|
+*/
|
|
+#define LUA_PROGNAME "lua"
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_MAXINPUT is the maximum length for an input line in the
|
|
+@* stand-alone interpreter.
|
|
+** CHANGE it if you need longer lines.
|
|
+*/
|
|
+#define LUA_MAXINPUT 512
|
|
+
|
|
+
|
|
+/*
|
|
+@@ lua_readline defines how to show a prompt and then read a line from
|
|
+@* the standard input.
|
|
+@@ lua_saveline defines how to "save" a read line in a "history".
|
|
+@@ lua_freeline defines how to free a line read by lua_readline.
|
|
+** CHANGE them if you want to improve this functionality (e.g., by using
|
|
+** GNU readline and history facilities).
|
|
+*/
|
|
+#if defined(LUA_USE_READLINE)
|
|
+#include <stdio.h>
|
|
+#include <readline/readline.h>
|
|
+#include <readline/history.h>
|
|
+#define lua_readline(L,b,p) ((void)L, ((b)=readline(p)) != NULL)
|
|
+#define lua_saveline(L,idx) \
|
|
+ if (lua_strlen(L,idx) > 0) /* non-empty line? */ \
|
|
+ add_history(lua_tostring(L, idx)); /* add it to history */
|
|
+#define lua_freeline(L,b) ((void)L, free(b))
|
|
+#else
|
|
+#define lua_readline(L,b,p) \
|
|
+ ((void)L, fputs(p, stdout), fflush(stdout), /* show prompt */ \
|
|
+ fgets(b, LUA_MAXINPUT, stdin) != NULL) /* get line */
|
|
+#define lua_saveline(L,idx) { (void)L; (void)idx; }
|
|
+#define lua_freeline(L,b) { (void)L; (void)b; }
|
|
+#endif
|
|
+
|
|
+#endif
|
|
+
|
|
+/* }================================================================== */
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_GCPAUSE defines the default pause between garbage-collector cycles
|
|
+@* as a percentage.
|
|
+** CHANGE it if you want the GC to run faster or slower (higher values
|
|
+** mean larger pauses which mean slower collection.) You can also change
|
|
+** this value dynamically.
|
|
+*/
|
|
+#define LUAI_GCPAUSE 200 /* 200% (wait memory to double before next GC) */
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_GCMUL defines the default speed of garbage collection relative to
|
|
+@* memory allocation as a percentage.
|
|
+** CHANGE it if you want to change the granularity of the garbage
|
|
+** collection. (Higher values mean coarser collections. 0 represents
|
|
+** infinity, where each step performs a full collection.) You can also
|
|
+** change this value dynamically.
|
|
+*/
|
|
+#define LUAI_GCMUL 200 /* GC runs 'twice the speed' of memory allocation */
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_GETN controls compatibility with old getn behavior.
|
|
+** CHANGE it (define it) if you want exact compatibility with the
|
|
+** behavior of setn/getn in Lua 5.0.
|
|
+*/
|
|
+#undef LUA_COMPAT_GETN
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_LOADLIB controls compatibility about global loadlib.
|
|
+** CHANGE it to undefined as soon as you do not need a global 'loadlib'
|
|
+** function (the function is still available as 'package.loadlib').
|
|
+*/
|
|
+#undef LUA_COMPAT_LOADLIB
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_VARARG controls compatibility with old vararg feature.
|
|
+** CHANGE it to undefined as soon as your programs use only '...' to
|
|
+** access vararg parameters (instead of the old 'arg' table).
|
|
+*/
|
|
+#define LUA_COMPAT_VARARG
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_MOD controls compatibility with old math.mod function.
|
|
+** CHANGE it to undefined as soon as your programs use 'math.fmod' or
|
|
+** the new '%' operator instead of 'math.mod'.
|
|
+*/
|
|
+#define LUA_COMPAT_MOD
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_LSTR controls compatibility with old long string nesting
|
|
+@* facility.
|
|
+** CHANGE it to 2 if you want the old behaviour, or undefine it to turn
|
|
+** off the advisory error when nesting [[...]].
|
|
+*/
|
|
+#define LUA_COMPAT_LSTR 1
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_GFIND controls compatibility with old 'string.gfind' name.
|
|
+** CHANGE it to undefined as soon as you rename 'string.gfind' to
|
|
+** 'string.gmatch'.
|
|
+*/
|
|
+#define LUA_COMPAT_GFIND
|
|
+
|
|
+/*
|
|
+@@ LUA_COMPAT_OPENLIB controls compatibility with old 'luaL_openlib'
|
|
+@* behavior.
|
|
+** CHANGE it to undefined as soon as you replace to 'luaL_register'
|
|
+** your uses of 'luaL_openlib'
|
|
+*/
|
|
+#define LUA_COMPAT_OPENLIB
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+@@ luai_apicheck is the assert macro used by the Lua-C API.
|
|
+** CHANGE luai_apicheck if you want Lua to perform some checks in the
|
|
+** parameters it gets from API calls. This may slow down the interpreter
|
|
+** a bit, but may be quite useful when debugging C code that interfaces
|
|
+** with Lua. A useful redefinition is to use assert.h.
|
|
+*/
|
|
+#if defined(LUA_USE_APICHECK)
|
|
+#include <assert.h>
|
|
+#define luai_apicheck(L,o) { (void)L; assert(o); }
|
|
+#else
|
|
+#define luai_apicheck(L,o) { (void)L; }
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_BITSINT defines the number of bits in an int.
|
|
+** CHANGE here if Lua cannot automatically detect the number of bits of
|
|
+** your machine. Probably you do not need to change this.
|
|
+*/
|
|
+/* avoid overflows in comparison */
|
|
+#if !defined(__KERNEL__)
|
|
+#include <limits.h>
|
|
+#define LUA_INT_MAX INT_MAX
|
|
+#else
|
|
+#define LUA_INT_MAX (~0U>>1)
|
|
+#endif
|
|
+
|
|
+#if LUA_INT_MAX-20 < 32760
|
|
+#define LUAI_BITSINT 16
|
|
+#elif LUA_INT_MAX > 2147483640L
|
|
+/* int has at least 32 bits */
|
|
+#define LUAI_BITSINT 32
|
|
+#else
|
|
+#error "you must define LUA_BITSINT with number of bits in an integer"
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_UINT32 is an unsigned integer with at least 32 bits.
|
|
+@@ LUAI_INT32 is an signed integer with at least 32 bits.
|
|
+@@ LUAI_UMEM is an unsigned integer big enough to count the total
|
|
+@* memory used by Lua.
|
|
+@@ LUAI_MEM is a signed integer big enough to count the total memory
|
|
+@* used by Lua.
|
|
+** CHANGE here if for some weird reason the default definitions are not
|
|
+** good enough for your machine. (The definitions in the 'else'
|
|
+** part always works, but may waste space on machines with 64-bit
|
|
+** longs.) Probably you do not need to change this.
|
|
+*/
|
|
+#if LUAI_BITSINT >= 32
|
|
+#define LUAI_UINT32 unsigned int
|
|
+#define LUAI_INT32 int
|
|
+#define LUAI_MAXINT32 INT_MAX
|
|
+#define LUAI_UMEM size_t
|
|
+#define LUAI_MEM ptrdiff_t
|
|
+#else
|
|
+/* 16-bit ints */
|
|
+#define LUAI_UINT32 unsigned long
|
|
+#define LUAI_INT32 long
|
|
+#define LUAI_MAXINT32 LONG_MAX
|
|
+#define LUAI_UMEM unsigned long
|
|
+#define LUAI_MEM long
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_MAXCALLS limits the number of nested calls.
|
|
+** CHANGE it if you need really deep recursive calls. This limit is
|
|
+** arbitrary; its only purpose is to stop infinite recursion before
|
|
+** exhausting memory.
|
|
+*/
|
|
+#define LUAI_MAXCALLS 20000
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_MAXCSTACK limits the number of Lua stack slots that a C function
|
|
+@* can use.
|
|
+** CHANGE it if you need lots of (Lua) stack space for your C
|
|
+** functions. This limit is arbitrary; its only purpose is to stop C
|
|
+** functions to consume unlimited stack space. (must be smaller than
|
|
+** -LUA_REGISTRYINDEX)
|
|
+*/
|
|
+#define LUAI_MAXCSTACK 8000
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {==================================================================
|
|
+** CHANGE (to smaller values) the following definitions if your system
|
|
+** has a small C stack. (Or you may want to change them to larger
|
|
+** values if your system has a large C stack and these limits are
|
|
+** too rigid for you.) Some of these constants control the size of
|
|
+** stack-allocated arrays used by the compiler or the interpreter, while
|
|
+** others limit the maximum number of recursive calls that the compiler
|
|
+** or the interpreter can perform. Values too large may cause a C stack
|
|
+** overflow for some forms of deep constructs.
|
|
+** ===================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_MAXCCALLS is the maximum depth for nested C calls (short) and
|
|
+@* syntactical nested non-terminals in a program.
|
|
+*/
|
|
+#define LUAI_MAXCCALLS 200
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_MAXVARS is the maximum number of local variables per function
|
|
+@* (must be smaller than 250).
|
|
+*/
|
|
+#define LUAI_MAXVARS 200
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_MAXUPVALUES is the maximum number of upvalues per function
|
|
+@* (must be smaller than 250).
|
|
+*/
|
|
+#define LUAI_MAXUPVALUES 60
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAL_BUFFERSIZE is the buffer size used by the lauxlib buffer system.
|
|
+*/
|
|
+#define LUAL_BUFFERSIZE BUFSIZ
|
|
+
|
|
+/* }================================================================== */
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** {==================================================================
|
|
+@@ LUA_NUMBER is the type of numbers in Lua.
|
|
+** CHANGE the following definitions only if you want to build Lua
|
|
+** with a number type different from double. You may also need to
|
|
+** change lua_number2int & lua_number2integer.
|
|
+** ===================================================================
|
|
+*/
|
|
+#if !defined(NO_FPU)
|
|
+#define LUA_NUMBER_DOUBLE
|
|
+#define LUA_NUMBER double
|
|
+#else
|
|
+#define LUA_NUMBER long
|
|
+#endif
|
|
+
|
|
+/*
|
|
+@@ LUAI_UACNUMBER is the result of an 'usual argument conversion'
|
|
+@* over a number.
|
|
+*/
|
|
+#define LUAI_UACNUMBER LUA_NUMBER
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_NUMBER_SCAN is the format for reading numbers.
|
|
+@@ LUA_NUMBER_FMT is the format for writing numbers.
|
|
+@@ lua_number2str converts a number to a string.
|
|
+@@ LUAI_MAXNUMBER2STR is maximum size of previous conversion.
|
|
+@@ lua_str2number converts a string to a number.
|
|
+*/
|
|
+#if !defined(NO_FPU)
|
|
+#define LUA_NUMBER_SCAN "%lf"
|
|
+#define LUA_NUMBER_FMT "%.14g"
|
|
+#define lua_str2number(s,p) strtod((s), (p))
|
|
+#else
|
|
+#define LUA_NUMBER_SCAN "%ld"
|
|
+#define LUA_NUMBER_FMT "%ld"
|
|
+#if !defined(__KERNEL__)
|
|
+#define lua_str2number(s,p) strtol((s), (p), 10)
|
|
+#else
|
|
+#define lua_str2number(s,p) simple_strtol((s), (p), 10)
|
|
+#endif
|
|
+#endif
|
|
+
|
|
+#define LUAI_MAXNUMBER2STR 32 /* 16 digits, sign, point, and \0 */
|
|
+#define lua_number2str(s,n) sprintf((s), LUA_NUMBER_FMT, (n))
|
|
+
|
|
+/*
|
|
+@@ The luai_num* macros define the primitive operations over numbers.
|
|
+*/
|
|
+#if defined(LUA_CORE)
|
|
+#define luai_numadd(a,b) ((a)+(b))
|
|
+#define luai_numsub(a,b) ((a)-(b))
|
|
+#define luai_nummul(a,b) ((a)*(b))
|
|
+#define luai_numdiv(a,b) ((a)/(b))
|
|
+#define luai_numunm(a) (-(a))
|
|
+#define luai_numeq(a,b) ((a)==(b))
|
|
+#define luai_numlt(a,b) ((a)<(b))
|
|
+#define luai_numle(a,b) ((a)<=(b))
|
|
+#define luai_numisnan(a) (!luai_numeq((a), (a)))
|
|
+#if !defined(NO_FPU)
|
|
+#include <math.h>
|
|
+#define luai_nummod(a,b) ((a) - floor((a)/(b))*(b))
|
|
+#define luai_numpow(a,b) (pow(a,b))
|
|
+#else
|
|
+#define luai_nummod(a,b) ((a)%(b))
|
|
+#define luai_numpow(a,b) luai_nummul(a,b)
|
|
+#endif
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ lua_number2int is a macro to convert lua_Number to int.
|
|
+@@ lua_number2integer is a macro to convert lua_Number to lua_Integer.
|
|
+** CHANGE them if you know a faster way to convert a lua_Number to
|
|
+** int (with any rounding method and without throwing errors) in your
|
|
+** system. In Pentium machines, a naive typecast from double to int
|
|
+** in C is extremely slow, so any alternative is worth trying.
|
|
+*/
|
|
+
|
|
+/* On a Pentium, resort to a trick */
|
|
+#if defined(LUA_NUMBER_DOUBLE) && !defined(LUA_ANSI) && !defined(__SSE2__) && \
|
|
+ (defined(__i386) || defined (_M_IX86) || defined(__i386__))
|
|
+
|
|
+/* On a Microsoft compiler, use assembler */
|
|
+#if defined(_MSC_VER)
|
|
+
|
|
+#define lua_number2int(i,d) __asm fld d __asm fistp i
|
|
+#define lua_number2integer(i,n) lua_number2int(i, n)
|
|
+
|
|
+/* the next trick should work on any Pentium, but sometimes clashes
|
|
+ with a DirectX idiosyncrasy */
|
|
+#else
|
|
+
|
|
+union luai_Cast { double l_d; long l_l; };
|
|
+#define lua_number2int(i,d) \
|
|
+ { volatile union luai_Cast u; u.l_d = (d) + 6755399441055744.0; (i) = u.l_l; }
|
|
+#define lua_number2integer(i,n) lua_number2int(i, n)
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
+/* this option always works, but may be slow */
|
|
+#else
|
|
+#define lua_number2int(i,d) ((i)=(int)(d))
|
|
+#define lua_number2integer(i,d) ((i)=(lua_Integer)(d))
|
|
+
|
|
+#endif
|
|
+
|
|
+/* }================================================================== */
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_USER_ALIGNMENT_T is a type that requires maximum alignment.
|
|
+** CHANGE it if your system requires alignments larger than double. (For
|
|
+** instance, if your system supports long doubles and they must be
|
|
+** aligned in 16-byte boundaries, then you should add long double in the
|
|
+** union.) Probably you do not need to change this.
|
|
+*/
|
|
+#define LUAI_USER_ALIGNMENT_T union { double u; void *s; long l; }
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_THROW/LUAI_TRY define how Lua does exception handling.
|
|
+** CHANGE them if you prefer to use longjmp/setjmp even with C++
|
|
+** or if want/don't to use _longjmp/_setjmp instead of regular
|
|
+** longjmp/setjmp. By default, Lua handles errors with exceptions when
|
|
+** compiling as C++ code, with _longjmp/_setjmp when asked to use them,
|
|
+** and with longjmp/setjmp otherwise.
|
|
+*/
|
|
+#if defined(__KERNEL__)
|
|
+#undef LUA_USE_ULONGJMP
|
|
+#endif
|
|
+
|
|
+#if defined(__cplusplus)
|
|
+/* C++ exceptions */
|
|
+#define LUAI_THROW(L,c) throw(c)
|
|
+#define LUAI_TRY(L,c,a) try { a } catch(...) \
|
|
+ { if ((c)->status == 0) (c)->status = -1; }
|
|
+#define luai_jmpbuf int /* dummy variable */
|
|
+
|
|
+#elif defined(LUA_USE_ULONGJMP)
|
|
+/* in Unix, try _longjmp/_setjmp (more efficient) */
|
|
+#define LUAI_THROW(L,c) _longjmp((c)->b, 1)
|
|
+#define LUAI_TRY(L,c,a) if (_setjmp((c)->b) == 0) { a }
|
|
+#define luai_jmpbuf jmp_buf
|
|
+
|
|
+#else
|
|
+/* default handling with long jumps */
|
|
+#define LUAI_THROW(L,c) longjmp((c)->b, 1)
|
|
+#define LUAI_TRY(L,c,a) if (setjmp((c)->b) == 0) { a }
|
|
+#define luai_jmpbuf jmp_buf
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_MAXCAPTURES is the maximum number of captures that a pattern
|
|
+@* can do during pattern-matching.
|
|
+** CHANGE it if you need more captures. This limit is arbitrary.
|
|
+*/
|
|
+#define LUA_MAXCAPTURES 32
|
|
+
|
|
+
|
|
+/*
|
|
+@@ lua_tmpnam is the function that the OS library uses to create a
|
|
+@* temporary name.
|
|
+@@ LUA_TMPNAMBUFSIZE is the maximum size of a name created by lua_tmpnam.
|
|
+** CHANGE them if you have an alternative to tmpnam (which is considered
|
|
+** insecure) or if you want the original tmpnam anyway. By default, Lua
|
|
+** uses tmpnam except when POSIX is available, where it uses mkstemp.
|
|
+*/
|
|
+#if defined(loslib_c) || defined(luaall_c)
|
|
+
|
|
+#if defined(LUA_USE_MKSTEMP)
|
|
+#include <unistd.h>
|
|
+#define LUA_TMPNAMBUFSIZE 32
|
|
+#define lua_tmpnam(b,e) { \
|
|
+ strcpy(b, "/tmp/lua_XXXXXX"); \
|
|
+ e = mkstemp(b); \
|
|
+ if (e != -1) close(e); \
|
|
+ e = (e == -1); }
|
|
+
|
|
+#else
|
|
+#define LUA_TMPNAMBUFSIZE L_tmpnam
|
|
+#define lua_tmpnam(b,e) { e = (tmpnam(b) == NULL); }
|
|
+#endif
|
|
+
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ lua_popen spawns a new process connected to the current one through
|
|
+@* the file streams.
|
|
+** CHANGE it if you have a way to implement it in your system.
|
|
+*/
|
|
+#if defined(LUA_USE_POPEN)
|
|
+
|
|
+#define lua_popen(L,c,m) ((void)L, fflush(NULL), popen(c,m))
|
|
+#define lua_pclose(L,file) ((void)L, (pclose(file) != -1))
|
|
+
|
|
+#elif defined(LUA_WIN)
|
|
+
|
|
+#define lua_popen(L,c,m) ((void)L, _popen(c,m))
|
|
+#define lua_pclose(L,file) ((void)L, (_pclose(file) != -1))
|
|
+
|
|
+#else
|
|
+
|
|
+#define lua_popen(L,c,m) ((void)((void)c, m), \
|
|
+ luaL_error(L, LUA_QL("popen") " not supported"), (FILE*)0)
|
|
+#define lua_pclose(L,file) ((void)((void)L, file), 0)
|
|
+
|
|
+#endif
|
|
+
|
|
+/*
|
|
+@@ LUA_DL_* define which dynamic-library system Lua should use.
|
|
+** CHANGE here if Lua has problems choosing the appropriate
|
|
+** dynamic-library system for your platform (either Windows' DLL, Mac's
|
|
+** dyld, or Unix's dlopen). If your system is some kind of Unix, there
|
|
+** is a good chance that it has dlopen, so LUA_DL_DLOPEN will work for
|
|
+** it. To use dlopen you also need to adapt the src/Makefile (probably
|
|
+** adding -ldl to the linker options), so Lua does not select it
|
|
+** automatically. (When you change the makefile to add -ldl, you must
|
|
+** also add -DLUA_USE_DLOPEN.)
|
|
+** If you do not want any kind of dynamic library, undefine all these
|
|
+** options.
|
|
+** By default, _WIN32 gets LUA_DL_DLL and MAC OS X gets LUA_DL_DYLD.
|
|
+*/
|
|
+#if defined(LUA_USE_DLOPEN)
|
|
+#define LUA_DL_DLOPEN
|
|
+#endif
|
|
+
|
|
+#if defined(LUA_WIN)
|
|
+#define LUA_DL_DLL
|
|
+#endif
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUAI_EXTRASPACE allows you to add user-specific data in a lua_State
|
|
+@* (the data goes just *before* the lua_State pointer).
|
|
+** CHANGE (define) this if you really need that. This value must be
|
|
+** a multiple of the maximum alignment required for your machine.
|
|
+*/
|
|
+#define LUAI_EXTRASPACE 0
|
|
+
|
|
+
|
|
+/*
|
|
+@@ luai_userstate* allow user-specific actions on threads.
|
|
+** CHANGE them if you defined LUAI_EXTRASPACE and need to do something
|
|
+** extra when a thread is created/deleted/resumed/yielded.
|
|
+*/
|
|
+#define luai_userstateopen(L) ((void)L)
|
|
+#define luai_userstateclose(L) ((void)L)
|
|
+#define luai_userstatethread(L,L1) ((void)L)
|
|
+#define luai_userstatefree(L) ((void)L)
|
|
+#define luai_userstateresume(L,n) ((void)L)
|
|
+#define luai_userstateyield(L,n) ((void)L)
|
|
+
|
|
+
|
|
+/*
|
|
+@@ LUA_INTFRMLEN is the length modifier for integer conversions
|
|
+@* in 'string.format'.
|
|
+@@ LUA_INTFRM_T is the integer type correspoding to the previous length
|
|
+@* modifier.
|
|
+** CHANGE them if your system supports long long or does not support long.
|
|
+*/
|
|
+
|
|
+#if defined(LUA_USELONGLONG)
|
|
+
|
|
+#define LUA_INTFRMLEN "ll"
|
|
+#define LUA_INTFRM_T long long
|
|
+
|
|
+#else
|
|
+
|
|
+#define LUA_INTFRMLEN "l"
|
|
+#define LUA_INTFRM_T long
|
|
+
|
|
+#endif
|
|
+
|
|
+/* =================================================================== */
|
|
+
|
|
+/*
|
|
+** Local configuration. You can use this space to add your redefinitions
|
|
+** without modifying the main part of the file.
|
|
+*/
|
|
+
|
|
+
|
|
+
|
|
+#endif
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lua.h
|
|
@@ -0,0 +1,387 @@
|
|
+/*
|
|
+** $Id: lua.h,v 1.218.1.5 2008/08/06 13:30:12 roberto Exp $
|
|
+** Lua - An Extensible Extension Language
|
|
+** Lua.org, PUC-Rio, Brazil (http://www.lua.org)
|
|
+** See Copyright Notice at the end of this file
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lua_h
|
|
+#define lua_h
|
|
+
|
|
+#include <stdarg.h>
|
|
+#include <stddef.h>
|
|
+
|
|
+#include "luaconf.h"
|
|
+
|
|
+
|
|
+#define LUA_VERSION "Lua 5.1"
|
|
+#define LUA_RELEASE "Lua 5.1.4"
|
|
+#define LUA_VERSION_NUM 501
|
|
+#define LUA_COPYRIGHT "Copyright (C) 1994-2008 Lua.org, PUC-Rio"
|
|
+#define LUA_AUTHORS "R. Ierusalimschy, L. H. de Figueiredo & W. Celes"
|
|
+
|
|
+
|
|
+/* mark for precompiled code (`<esc>Lua') */
|
|
+#define LUA_SIGNATURE "\033Lua"
|
|
+
|
|
+/* option for multiple returns in `lua_pcall' and `lua_call' */
|
|
+#define LUA_MULTRET (-1)
|
|
+
|
|
+
|
|
+/*
|
|
+** pseudo-indices
|
|
+*/
|
|
+#define LUA_REGISTRYINDEX (-10000)
|
|
+#define LUA_ENVIRONINDEX (-10001)
|
|
+#define LUA_GLOBALSINDEX (-10002)
|
|
+#define lua_upvalueindex(i) (LUA_GLOBALSINDEX-(i))
|
|
+
|
|
+
|
|
+/* thread status; 0 is OK */
|
|
+#define LUA_YIELD 1
|
|
+#define LUA_ERRRUN 2
|
|
+#define LUA_ERRSYNTAX 3
|
|
+#define LUA_ERRMEM 4
|
|
+#define LUA_ERRERR 5
|
|
+
|
|
+
|
|
+typedef struct lua_State lua_State;
|
|
+
|
|
+typedef int (*lua_CFunction) (lua_State *L);
|
|
+
|
|
+
|
|
+/*
|
|
+** functions that read/write blocks when loading/dumping Lua chunks
|
|
+*/
|
|
+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);
|
|
+
|
|
+
|
|
+/*
|
|
+** prototype for memory-allocation functions
|
|
+*/
|
|
+typedef void * (*lua_Alloc) (void *ud, void *ptr, size_t osize, size_t nsize);
|
|
+
|
|
+
|
|
+/*
|
|
+** basic types
|
|
+*/
|
|
+#define LUA_TNONE (-1)
|
|
+
|
|
+#define LUA_TNIL 0
|
|
+#define LUA_TBOOLEAN 1
|
|
+#define LUA_TLIGHTUSERDATA 2
|
|
+#define LUA_TNUMBER 3
|
|
+#define LUA_TSTRING 4
|
|
+#define LUA_TTABLE 5
|
|
+#define LUA_TFUNCTION 6
|
|
+#define LUA_TUSERDATA 7
|
|
+#define LUA_TTHREAD 8
|
|
+
|
|
+
|
|
+
|
|
+/* minimum Lua stack available to a C function */
|
|
+#define LUA_MINSTACK 20
|
|
+
|
|
+
|
|
+/*
|
|
+** generic extra include file
|
|
+*/
|
|
+#if defined(LUA_USER_H)
|
|
+#include LUA_USER_H
|
|
+#endif
|
|
+
|
|
+
|
|
+/* type of numbers in Lua */
|
|
+typedef LUA_NUMBER lua_Number;
|
|
+
|
|
+
|
|
+/* type for integer functions */
|
|
+typedef LUA_INTEGER lua_Integer;
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** state manipulation
|
|
+*/
|
|
+LUA_API lua_State *(lua_newstate) (lua_Alloc f, void *ud);
|
|
+LUA_API void (lua_close) (lua_State *L);
|
|
+LUA_API lua_State *(lua_newthread) (lua_State *L);
|
|
+
|
|
+LUA_API lua_CFunction (lua_atpanic) (lua_State *L, lua_CFunction panicf);
|
|
+
|
|
+
|
|
+/*
|
|
+** basic stack manipulation
|
|
+*/
|
|
+LUA_API int (lua_gettop) (lua_State *L);
|
|
+LUA_API void (lua_settop) (lua_State *L, int idx);
|
|
+LUA_API void (lua_pushvalue) (lua_State *L, int idx);
|
|
+LUA_API void (lua_remove) (lua_State *L, int idx);
|
|
+LUA_API void (lua_insert) (lua_State *L, int idx);
|
|
+LUA_API void (lua_replace) (lua_State *L, int idx);
|
|
+LUA_API int (lua_checkstack) (lua_State *L, int sz);
|
|
+
|
|
+LUA_API void (lua_xmove) (lua_State *from, lua_State *to, int n);
|
|
+
|
|
+
|
|
+/*
|
|
+** access functions (stack -> C)
|
|
+*/
|
|
+
|
|
+LUA_API int (lua_isnumber) (lua_State *L, int idx);
|
|
+LUA_API int (lua_isstring) (lua_State *L, int idx);
|
|
+LUA_API int (lua_iscfunction) (lua_State *L, int idx);
|
|
+LUA_API int (lua_isuserdata) (lua_State *L, int idx);
|
|
+LUA_API int (lua_type) (lua_State *L, int idx);
|
|
+LUA_API const char *(lua_typename) (lua_State *L, int tp);
|
|
+
|
|
+LUA_API int (lua_equal) (lua_State *L, int idx1, int idx2);
|
|
+LUA_API int (lua_rawequal) (lua_State *L, int idx1, int idx2);
|
|
+LUA_API int (lua_lessthan) (lua_State *L, int idx1, int idx2);
|
|
+
|
|
+LUA_API lua_Number (lua_tonumber) (lua_State *L, int idx);
|
|
+LUA_API lua_Integer (lua_tointeger) (lua_State *L, int idx);
|
|
+LUA_API int (lua_toboolean) (lua_State *L, int idx);
|
|
+LUA_API const char *(lua_tolstring) (lua_State *L, int idx, size_t *len);
|
|
+LUA_API size_t (lua_objlen) (lua_State *L, int idx);
|
|
+LUA_API lua_CFunction (lua_tocfunction) (lua_State *L, int idx);
|
|
+LUA_API void *(lua_touserdata) (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);
|
|
+
|
|
+
|
|
+/*
|
|
+** push functions (C -> stack)
|
|
+*/
|
|
+LUA_API void (lua_pushnil) (lua_State *L);
|
|
+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_pushlstring) (lua_State *L, const char *s, size_t l);
|
|
+LUA_API void (lua_pushstring) (lua_State *L, const char *s);
|
|
+LUA_API const char *(lua_pushvfstring) (lua_State *L, const char *fmt,
|
|
+ va_list argp);
|
|
+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_pushboolean) (lua_State *L, int b);
|
|
+LUA_API void (lua_pushlightuserdata) (lua_State *L, void *p);
|
|
+LUA_API int (lua_pushthread) (lua_State *L);
|
|
+
|
|
+
|
|
+/*
|
|
+** get functions (Lua -> stack)
|
|
+*/
|
|
+LUA_API void (lua_gettable) (lua_State *L, int idx);
|
|
+LUA_API void (lua_getfield) (lua_State *L, int idx, const char *k);
|
|
+LUA_API void (lua_rawget) (lua_State *L, int idx);
|
|
+LUA_API void (lua_rawgeti) (lua_State *L, int idx, int n);
|
|
+LUA_API void (lua_createtable) (lua_State *L, int narr, int nrec);
|
|
+LUA_API void *(lua_newuserdata) (lua_State *L, size_t sz);
|
|
+LUA_API int (lua_getmetatable) (lua_State *L, int objindex);
|
|
+LUA_API void (lua_getfenv) (lua_State *L, int idx);
|
|
+
|
|
+
|
|
+/*
|
|
+** set functions (stack -> Lua)
|
|
+*/
|
|
+LUA_API void (lua_settable) (lua_State *L, int idx);
|
|
+LUA_API void (lua_setfield) (lua_State *L, int idx, const char *k);
|
|
+LUA_API void (lua_rawset) (lua_State *L, int idx);
|
|
+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_setfenv) (lua_State *L, int idx);
|
|
+
|
|
+
|
|
+/*
|
|
+** `load' and `call' functions (load and run Lua code)
|
|
+*/
|
|
+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_cpcall) (lua_State *L, lua_CFunction func, void *ud);
|
|
+LUA_API int (lua_load) (lua_State *L, lua_Reader reader, void *dt,
|
|
+ const char *chunkname);
|
|
+
|
|
+LUA_API int (lua_dump) (lua_State *L, lua_Writer writer, void *data);
|
|
+
|
|
+
|
|
+/*
|
|
+** coroutine functions
|
|
+*/
|
|
+LUA_API int (lua_yield) (lua_State *L, int nresults);
|
|
+LUA_API int (lua_resume) (lua_State *L, int narg);
|
|
+LUA_API int (lua_status) (lua_State *L);
|
|
+
|
|
+/*
|
|
+** garbage-collection function and options
|
|
+*/
|
|
+
|
|
+#define LUA_GCSTOP 0
|
|
+#define LUA_GCRESTART 1
|
|
+#define LUA_GCCOLLECT 2
|
|
+#define LUA_GCCOUNT 3
|
|
+#define LUA_GCCOUNTB 4
|
|
+#define LUA_GCSTEP 5
|
|
+#define LUA_GCSETPAUSE 6
|
|
+#define LUA_GCSETSTEPMUL 7
|
|
+
|
|
+LUA_API int (lua_gc) (lua_State *L, int what, int data);
|
|
+
|
|
+
|
|
+/*
|
|
+** miscellaneous functions
|
|
+*/
|
|
+
|
|
+LUA_API int (lua_error) (lua_State *L);
|
|
+
|
|
+LUA_API int (lua_next) (lua_State *L, int idx);
|
|
+
|
|
+LUA_API void (lua_concat) (lua_State *L, int n);
|
|
+
|
|
+LUA_API lua_Alloc (lua_getallocf) (lua_State *L, void **ud);
|
|
+LUA_API void lua_setallocf (lua_State *L, lua_Alloc f, void *ud);
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** ===============================================================
|
|
+** some useful macros
|
|
+** ===============================================================
|
|
+*/
|
|
+
|
|
+#define lua_pop(L,n) lua_settop(L, -(n)-1)
|
|
+
|
|
+#define lua_newtable(L) lua_createtable(L, 0, 0)
|
|
+
|
|
+#define lua_register(L,n,f) (lua_pushcfunction(L, (f)), lua_setglobal(L, (n)))
|
|
+
|
|
+#define lua_pushcfunction(L,f) lua_pushcclosure(L, (f), 0)
|
|
+
|
|
+#define lua_strlen(L,i) lua_objlen(L, (i))
|
|
+
|
|
+#define lua_isfunction(L,n) (lua_type(L, (n)) == LUA_TFUNCTION)
|
|
+#define lua_istable(L,n) (lua_type(L, (n)) == LUA_TTABLE)
|
|
+#define lua_islightuserdata(L,n) (lua_type(L, (n)) == LUA_TLIGHTUSERDATA)
|
|
+#define lua_isnil(L,n) (lua_type(L, (n)) == LUA_TNIL)
|
|
+#define lua_isboolean(L,n) (lua_type(L, (n)) == LUA_TBOOLEAN)
|
|
+#define lua_isthread(L,n) (lua_type(L, (n)) == LUA_TTHREAD)
|
|
+#define lua_isnone(L,n) (lua_type(L, (n)) == LUA_TNONE)
|
|
+#define lua_isnoneornil(L, n) (lua_type(L, (n)) <= 0)
|
|
+
|
|
+#define lua_pushliteral(L, s) \
|
|
+ lua_pushlstring(L, "" s, (sizeof(s)/sizeof(char))-1)
|
|
+
|
|
+#define lua_setglobal(L,s) lua_setfield(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)
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** compatibility macros and functions
|
|
+*/
|
|
+
|
|
+#define lua_open() luaL_newstate()
|
|
+
|
|
+#define lua_getregistry(L) lua_pushvalue(L, LUA_REGISTRYINDEX)
|
|
+
|
|
+#define lua_getgccount(L) lua_gc(L, LUA_GCCOUNT, 0)
|
|
+
|
|
+#define lua_Chunkreader lua_Reader
|
|
+#define lua_Chunkwriter lua_Writer
|
|
+
|
|
+
|
|
+/* hack */
|
|
+LUA_API void lua_setlevel (lua_State *from, lua_State *to);
|
|
+
|
|
+
|
|
+/*
|
|
+** {======================================================================
|
|
+** Debug API
|
|
+** =======================================================================
|
|
+*/
|
|
+
|
|
+
|
|
+/*
|
|
+** Event codes
|
|
+*/
|
|
+#define LUA_HOOKCALL 0
|
|
+#define LUA_HOOKRET 1
|
|
+#define LUA_HOOKLINE 2
|
|
+#define LUA_HOOKCOUNT 3
|
|
+#define LUA_HOOKTAILRET 4
|
|
+
|
|
+
|
|
+/*
|
|
+** Event masks
|
|
+*/
|
|
+#define LUA_MASKCALL (1 << LUA_HOOKCALL)
|
|
+#define LUA_MASKRET (1 << LUA_HOOKRET)
|
|
+#define LUA_MASKLINE (1 << LUA_HOOKLINE)
|
|
+#define LUA_MASKCOUNT (1 << LUA_HOOKCOUNT)
|
|
+
|
|
+typedef struct lua_Debug lua_Debug; /* activation record */
|
|
+
|
|
+
|
|
+/* Functions to be called by the debuger in specific events */
|
|
+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_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_setlocal (lua_State *L, const lua_Debug *ar, int n);
|
|
+LUA_API const char *lua_getupvalue (lua_State *L, int funcindex, int n);
|
|
+LUA_API const char *lua_setupvalue (lua_State *L, int funcindex, int n);
|
|
+
|
|
+LUA_API int lua_sethook (lua_State *L, lua_Hook func, int mask, int count);
|
|
+LUA_API lua_Hook lua_gethook (lua_State *L);
|
|
+LUA_API int lua_gethookmask (lua_State *L);
|
|
+LUA_API int lua_gethookcount (lua_State *L);
|
|
+
|
|
+
|
|
+struct lua_Debug {
|
|
+ int event;
|
|
+ const char *name; /* (n) */
|
|
+ const char *namewhat; /* (n) `global', `local', `field', `method' */
|
|
+ const char *what; /* (S) `Lua', `C', `main', `tail' */
|
|
+ const char *source; /* (S) */
|
|
+ int currentline; /* (l) */
|
|
+ int nups; /* (u) number of upvalues */
|
|
+ int linedefined; /* (S) */
|
|
+ int lastlinedefined; /* (S) */
|
|
+ char short_src[LUA_IDSIZE]; /* (S) */
|
|
+ /* private part */
|
|
+ int i_ci; /* active function */
|
|
+};
|
|
+
|
|
+/* }====================================================================== */
|
|
+
|
|
+
|
|
+/******************************************************************************
|
|
+* Copyright (C) 1994-2008 Lua.org, PUC-Rio. All rights reserved.
|
|
+*
|
|
+* Permission is hereby granted, free of charge, to any person obtaining
|
|
+* a copy of this software and associated documentation files (the
|
|
+* "Software"), to deal in the Software without restriction, including
|
|
+* without limitation the rights to use, copy, modify, merge, publish,
|
|
+* distribute, sublicense, and/or sell copies of the Software, and to
|
|
+* permit persons to whom the Software is furnished to do so, subject to
|
|
+* the following conditions:
|
|
+*
|
|
+* The above copyright notice and this permission notice shall be
|
|
+* included in all copies or substantial portions of the Software.
|
|
+*
|
|
+* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
+* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
+* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
+* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
+* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
+* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
+* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
+******************************************************************************/
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lualib.h
|
|
@@ -0,0 +1,55 @@
|
|
+/*
|
|
+** $Id: lualib.h,v 1.36.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lua standard libraries
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lualib_h
|
|
+#define lualib_h
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+
|
|
+/* Key to file-handle type */
|
|
+#define LUA_FILEHANDLE "FILE*"
|
|
+
|
|
+
|
|
+#define LUA_COLIBNAME "coroutine"
|
|
+LUALIB_API int (luaopen_base) (lua_State *L);
|
|
+
|
|
+#define LUA_TABLIBNAME "table"
|
|
+LUALIB_API int (luaopen_table) (lua_State *L);
|
|
+/*
|
|
+#define LUA_IOLIBNAME "io"
|
|
+LUALIB_API int (luaopen_io) (lua_State *L);
|
|
+
|
|
+#define LUA_OSLIBNAME "os"
|
|
+LUALIB_API int (luaopen_os) (lua_State *L);
|
|
+*/
|
|
+
|
|
+#define LUA_STRLIBNAME "string"
|
|
+LUALIB_API int (luaopen_string) (lua_State *L);
|
|
+
|
|
+/*
|
|
+#define LUA_MATHLIBNAME "math"
|
|
+LUALIB_API int (luaopen_math) (lua_State *L);
|
|
+
|
|
+#define LUA_DBLIBNAME "debug"
|
|
+LUALIB_API int (luaopen_debug) (lua_State *L);
|
|
+
|
|
+#define LUA_LOADLIBNAME "package"
|
|
+LUALIB_API int (luaopen_package) (lua_State *L);
|
|
+*/
|
|
+
|
|
+/* open all previous libraries */
|
|
+LUALIB_API void (luaL_openlibs) (lua_State *L);
|
|
+
|
|
+
|
|
+
|
|
+#ifndef lua_assert
|
|
+#define lua_assert(x) ((void)0)
|
|
+#endif
|
|
+
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lundump.c
|
|
@@ -0,0 +1,227 @@
|
|
+/*
|
|
+** $Id: lundump.c,v 2.7.1.4 2008/04/04 19:51:41 roberto Exp $
|
|
+** load precompiled Lua chunks
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define lundump_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lmem.h"
|
|
+#include "lobject.h"
|
|
+#include "lstring.h"
|
|
+#include "lundump.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+typedef struct {
|
|
+ lua_State* L;
|
|
+ ZIO* Z;
|
|
+ Mbuffer* b;
|
|
+ const char* name;
|
|
+} LoadState;
|
|
+
|
|
+#ifdef LUAC_TRUST_BINARIES
|
|
+#define IF(c,s)
|
|
+#define error(S,s)
|
|
+#else
|
|
+#define IF(c,s) if (c) error(S,s)
|
|
+
|
|
+static void error(LoadState* S, const char* why)
|
|
+{
|
|
+ luaO_pushfstring(S->L,"%s: %s in precompiled chunk",S->name,why);
|
|
+ luaD_throw(S->L,LUA_ERRSYNTAX);
|
|
+}
|
|
+#endif
|
|
+
|
|
+#define LoadMem(S,b,n,size) LoadBlock(S,b,(n)*(size))
|
|
+#define LoadByte(S) (lu_byte)LoadChar(S)
|
|
+#define LoadVar(S,x) LoadMem(S,&x,1,sizeof(x))
|
|
+#define LoadVector(S,b,n,size) LoadMem(S,b,n,size)
|
|
+
|
|
+static void LoadBlock(LoadState* S, void* b, size_t size)
|
|
+{
|
|
+ size_t r=luaZ_read(S->Z,b,size);
|
|
+ IF (r!=0, "unexpected end");
|
|
+}
|
|
+
|
|
+static int LoadChar(LoadState* S)
|
|
+{
|
|
+ char x;
|
|
+ LoadVar(S,x);
|
|
+ return x;
|
|
+}
|
|
+
|
|
+static int LoadInt(LoadState* S)
|
|
+{
|
|
+ int x;
|
|
+ LoadVar(S,x);
|
|
+ IF (x<0, "bad integer");
|
|
+ return x;
|
|
+}
|
|
+
|
|
+static lua_Number LoadNumber(LoadState* S)
|
|
+{
|
|
+ lua_Number x;
|
|
+ LoadVar(S,x);
|
|
+ return x;
|
|
+}
|
|
+
|
|
+static TString* LoadString(LoadState* S)
|
|
+{
|
|
+ size_t size;
|
|
+ LoadVar(S,size);
|
|
+ if (size==0)
|
|
+ return NULL;
|
|
+ else
|
|
+ {
|
|
+ char* s=luaZ_openspace(S->L,S->b,size);
|
|
+ LoadBlock(S,s,size);
|
|
+ return luaS_newlstr(S->L,s,size-1); /* remove trailing '\0' */
|
|
+ }
|
|
+}
|
|
+
|
|
+static void LoadCode(LoadState* S, Proto* f)
|
|
+{
|
|
+ int n=LoadInt(S);
|
|
+ f->code=luaM_newvector(S->L,n,Instruction);
|
|
+ f->sizecode=n;
|
|
+ LoadVector(S,f->code,n,sizeof(Instruction));
|
|
+}
|
|
+
|
|
+static Proto* LoadFunction(LoadState* S, TString* p);
|
|
+
|
|
+static void LoadConstants(LoadState* S, Proto* f)
|
|
+{
|
|
+ int i,n;
|
|
+ n=LoadInt(S);
|
|
+ f->k=luaM_newvector(S->L,n,TValue);
|
|
+ f->sizek=n;
|
|
+ for (i=0; i<n; i++) setnilvalue(&f->k[i]);
|
|
+ for (i=0; i<n; i++)
|
|
+ {
|
|
+ TValue* o=&f->k[i];
|
|
+ int t=LoadChar(S);
|
|
+ switch (t)
|
|
+ {
|
|
+ case LUA_TNIL:
|
|
+ setnilvalue(o);
|
|
+ break;
|
|
+ case LUA_TBOOLEAN:
|
|
+ setbvalue(o,LoadChar(S)!=0);
|
|
+ break;
|
|
+ case LUA_TNUMBER:
|
|
+ setnvalue(o,LoadNumber(S));
|
|
+ break;
|
|
+ case LUA_TSTRING:
|
|
+ setsvalue2n(S->L,o,LoadString(S));
|
|
+ break;
|
|
+ default:
|
|
+ error(S,"bad constant");
|
|
+ break;
|
|
+ }
|
|
+ }
|
|
+ n=LoadInt(S);
|
|
+ f->p=luaM_newvector(S->L,n,Proto*);
|
|
+ f->sizep=n;
|
|
+ for (i=0; i<n; i++) f->p[i]=NULL;
|
|
+ for (i=0; i<n; i++) f->p[i]=LoadFunction(S,f->source);
|
|
+}
|
|
+
|
|
+static void LoadDebug(LoadState* S, Proto* f)
|
|
+{
|
|
+ int i,n;
|
|
+ n=LoadInt(S);
|
|
+ f->lineinfo=luaM_newvector(S->L,n,int);
|
|
+ f->sizelineinfo=n;
|
|
+ LoadVector(S,f->lineinfo,n,sizeof(int));
|
|
+ n=LoadInt(S);
|
|
+ f->locvars=luaM_newvector(S->L,n,LocVar);
|
|
+ f->sizelocvars=n;
|
|
+ for (i=0; i<n; i++) f->locvars[i].varname=NULL;
|
|
+ for (i=0; i<n; i++)
|
|
+ {
|
|
+ f->locvars[i].varname=LoadString(S);
|
|
+ f->locvars[i].startpc=LoadInt(S);
|
|
+ f->locvars[i].endpc=LoadInt(S);
|
|
+ }
|
|
+ n=LoadInt(S);
|
|
+ f->upvalues=luaM_newvector(S->L,n,TString*);
|
|
+ f->sizeupvalues=n;
|
|
+ for (i=0; i<n; i++) f->upvalues[i]=NULL;
|
|
+ for (i=0; i<n; i++) f->upvalues[i]=LoadString(S);
|
|
+}
|
|
+
|
|
+static Proto* LoadFunction(LoadState* S, TString* p)
|
|
+{
|
|
+ Proto* f;
|
|
+ if (++S->L->nCcalls > LUAI_MAXCCALLS) error(S,"code too deep");
|
|
+ f=luaF_newproto(S->L);
|
|
+ setptvalue2s(S->L,S->L->top,f); incr_top(S->L);
|
|
+ f->source=LoadString(S); if (f->source==NULL) f->source=p;
|
|
+ f->linedefined=LoadInt(S);
|
|
+ f->lastlinedefined=LoadInt(S);
|
|
+ f->nups=LoadByte(S);
|
|
+ f->numparams=LoadByte(S);
|
|
+ f->is_vararg=LoadByte(S);
|
|
+ f->maxstacksize=LoadByte(S);
|
|
+ LoadCode(S,f);
|
|
+ LoadConstants(S,f);
|
|
+ LoadDebug(S,f);
|
|
+ IF (!luaG_checkcode(f), "bad code");
|
|
+ S->L->top--;
|
|
+ S->L->nCcalls--;
|
|
+ return f;
|
|
+}
|
|
+
|
|
+static void LoadHeader(LoadState* S)
|
|
+{
|
|
+ char h[LUAC_HEADERSIZE];
|
|
+ char s[LUAC_HEADERSIZE];
|
|
+ luaU_header(h);
|
|
+ LoadBlock(S,s,LUAC_HEADERSIZE);
|
|
+ IF (memcmp(h,s,LUAC_HEADERSIZE)!=0, "bad header");
|
|
+}
|
|
+
|
|
+/*
|
|
+** load precompiled chunk
|
|
+*/
|
|
+Proto* luaU_undump (lua_State* L, ZIO* Z, Mbuffer* buff, const char* name)
|
|
+{
|
|
+ LoadState S;
|
|
+ if (*name=='@' || *name=='=')
|
|
+ S.name=name+1;
|
|
+ else if (*name==LUA_SIGNATURE[0])
|
|
+ S.name="binary string";
|
|
+ else
|
|
+ S.name=name;
|
|
+ S.L=L;
|
|
+ S.Z=Z;
|
|
+ S.b=buff;
|
|
+ LoadHeader(&S);
|
|
+ return LoadFunction(&S,luaS_newliteral(L,"=?"));
|
|
+}
|
|
+
|
|
+/*
|
|
+* make header
|
|
+*/
|
|
+void luaU_header (char* h)
|
|
+{
|
|
+ int x=1;
|
|
+ memcpy(h,LUA_SIGNATURE,sizeof(LUA_SIGNATURE)-1);
|
|
+ h+=sizeof(LUA_SIGNATURE)-1;
|
|
+ *h++=(char)LUAC_VERSION;
|
|
+ *h++=(char)LUAC_FORMAT;
|
|
+ *h++=(char)*(char*)&x; /* endianness */
|
|
+ *h++=(char)sizeof(int);
|
|
+ *h++=(char)sizeof(size_t);
|
|
+ *h++=(char)sizeof(Instruction);
|
|
+ *h++=(char)sizeof(lua_Number);
|
|
+ *h++=(char)(((lua_Number)0.5)==0); /* is lua_Number integral? */
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lundump.h
|
|
@@ -0,0 +1,36 @@
|
|
+/*
|
|
+** $Id: lundump.h,v 1.37.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** load precompiled Lua chunks
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lundump_h
|
|
+#define lundump_h
|
|
+
|
|
+#include "lobject.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+/* load one chunk; from lundump.c */
|
|
+LUAI_FUNC Proto* luaU_undump (lua_State* L, ZIO* Z, Mbuffer* buff, const char* name);
|
|
+
|
|
+/* make header; from lundump.c */
|
|
+LUAI_FUNC void luaU_header (char* h);
|
|
+
|
|
+/* dump one chunk; from ldump.c */
|
|
+LUAI_FUNC int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip);
|
|
+
|
|
+#ifdef luac_c
|
|
+/* print one chunk; from print.c */
|
|
+LUAI_FUNC void luaU_print (const Proto* f, int full);
|
|
+#endif
|
|
+
|
|
+/* for header of binary files -- this is Lua 5.1 */
|
|
+#define LUAC_VERSION 0x51
|
|
+
|
|
+/* for header of binary files -- this is the official format */
|
|
+#define LUAC_FORMAT 0
|
|
+
|
|
+/* size of header of binary files */
|
|
+#define LUAC_HEADERSIZE 12
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lvm.c
|
|
@@ -0,0 +1,762 @@
|
|
+/*
|
|
+** $Id: lvm.c,v 2.63.1.3 2007/12/28 15:32:23 roberto Exp $
|
|
+** Lua virtual machine
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <stdio.h>
|
|
+#include <stdlib.h>
|
|
+#include <string.h>
|
|
+
|
|
+#define lvm_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "ldebug.h"
|
|
+#include "ldo.h"
|
|
+#include "lfunc.h"
|
|
+#include "lgc.h"
|
|
+#include "lobject.h"
|
|
+#include "lopcodes.h"
|
|
+#include "lstate.h"
|
|
+#include "lstring.h"
|
|
+#include "ltable.h"
|
|
+#include "ltm.h"
|
|
+#include "lvm.h"
|
|
+
|
|
+
|
|
+
|
|
+/* limit for table tag-method chains (to avoid loops) */
|
|
+#define MAXTAGLOOP 100
|
|
+
|
|
+
|
|
+const TValue *luaV_tonumber (const TValue *obj, TValue *n) {
|
|
+ lua_Number num;
|
|
+ if (ttisnumber(obj)) return obj;
|
|
+ if (ttisstring(obj) && luaO_str2d(svalue(obj), &num)) {
|
|
+ setnvalue(n, num);
|
|
+ return n;
|
|
+ }
|
|
+ else
|
|
+ return NULL;
|
|
+}
|
|
+
|
|
+
|
|
+int luaV_tostring (lua_State *L, StkId obj) {
|
|
+ if (!ttisnumber(obj))
|
|
+ return 0;
|
|
+ else {
|
|
+ char s[LUAI_MAXNUMBER2STR];
|
|
+ lua_Number n = nvalue(obj);
|
|
+ lua_number2str(s, n);
|
|
+ setsvalue2s(L, obj, luaS_new(L, s));
|
|
+ return 1;
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void traceexec (lua_State *L, const Instruction *pc) {
|
|
+ lu_byte mask = L->hookmask;
|
|
+ const Instruction *oldpc = L->savedpc;
|
|
+ L->savedpc = pc;
|
|
+ if ((mask & LUA_MASKCOUNT) && L->hookcount == 0) {
|
|
+ resethookcount(L);
|
|
+ luaD_callhook(L, LUA_HOOKCOUNT, -1);
|
|
+ }
|
|
+ if (mask & LUA_MASKLINE) {
|
|
+ Proto *p = ci_func(L->ci)->l.p;
|
|
+ int npc = pcRel(pc, p);
|
|
+ int newline = getline(p, npc);
|
|
+ /* call linehook when enter a new function, when jump back (loop),
|
|
+ or when enter a new line */
|
|
+ if (npc == 0 || pc <= oldpc || newline != getline(p, pcRel(oldpc, p)))
|
|
+ luaD_callhook(L, LUA_HOOKLINE, newline);
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+static void callTMres (lua_State *L, StkId res, const TValue *f,
|
|
+ const TValue *p1, const TValue *p2) {
|
|
+ ptrdiff_t result = savestack(L, res);
|
|
+ setobj2s(L, L->top, f); /* push function */
|
|
+ setobj2s(L, L->top+1, p1); /* 1st argument */
|
|
+ setobj2s(L, L->top+2, p2); /* 2nd argument */
|
|
+ luaD_checkstack(L, 3);
|
|
+ L->top += 3;
|
|
+ luaD_call(L, L->top - 3, 1);
|
|
+ res = restorestack(L, result);
|
|
+ L->top--;
|
|
+ setobjs2s(L, res, L->top);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+static void callTM (lua_State *L, const TValue *f, const TValue *p1,
|
|
+ const TValue *p2, const TValue *p3) {
|
|
+ setobj2s(L, L->top, f); /* push function */
|
|
+ setobj2s(L, L->top+1, p1); /* 1st argument */
|
|
+ setobj2s(L, L->top+2, p2); /* 2nd argument */
|
|
+ setobj2s(L, L->top+3, p3); /* 3th argument */
|
|
+ luaD_checkstack(L, 4);
|
|
+ L->top += 4;
|
|
+ luaD_call(L, L->top - 4, 0);
|
|
+}
|
|
+
|
|
+
|
|
+void luaV_gettable (lua_State *L, const TValue *t, TValue *key, StkId val) {
|
|
+ int loop;
|
|
+ for (loop = 0; loop < MAXTAGLOOP; loop++) {
|
|
+ const TValue *tm;
|
|
+ if (ttistable(t)) { /* `t' is a table? */
|
|
+ Table *h = hvalue(t);
|
|
+ const TValue *res = luaH_get(h, key); /* do a primitive get */
|
|
+ if (!ttisnil(res) || /* result is no nil? */
|
|
+ (tm = fasttm(L, h->metatable, TM_INDEX)) == NULL) { /* or no TM? */
|
|
+ setobj2s(L, val, res);
|
|
+ return;
|
|
+ }
|
|
+ /* else will try the tag method */
|
|
+ }
|
|
+ else if (ttisnil(tm = luaT_gettmbyobj(L, t, TM_INDEX)))
|
|
+ luaG_typeerror(L, t, "index");
|
|
+ if (ttisfunction(tm)) {
|
|
+ callTMres(L, val, tm, t, key);
|
|
+ return;
|
|
+ }
|
|
+ t = tm; /* else repeat with `tm' */
|
|
+ }
|
|
+ luaG_runerror(L, "loop in gettable");
|
|
+}
|
|
+
|
|
+
|
|
+void luaV_settable (lua_State *L, const TValue *t, TValue *key, StkId val) {
|
|
+ int loop;
|
|
+ for (loop = 0; loop < MAXTAGLOOP; loop++) {
|
|
+ const TValue *tm;
|
|
+ if (ttistable(t)) { /* `t' is a table? */
|
|
+ Table *h = hvalue(t);
|
|
+ TValue *oldval = luaH_set(L, h, key); /* do a primitive set */
|
|
+ if (!ttisnil(oldval) || /* result is no nil? */
|
|
+ (tm = fasttm(L, h->metatable, TM_NEWINDEX)) == NULL) { /* or no TM? */
|
|
+ setobj2t(L, oldval, val);
|
|
+ luaC_barriert(L, h, val);
|
|
+ return;
|
|
+ }
|
|
+ /* else will try the tag method */
|
|
+ }
|
|
+ else if (ttisnil(tm = luaT_gettmbyobj(L, t, TM_NEWINDEX)))
|
|
+ luaG_typeerror(L, t, "index");
|
|
+ if (ttisfunction(tm)) {
|
|
+ callTM(L, tm, t, key, val);
|
|
+ return;
|
|
+ }
|
|
+ t = tm; /* else repeat with `tm' */
|
|
+ }
|
|
+ luaG_runerror(L, "loop in settable");
|
|
+}
|
|
+
|
|
+
|
|
+static int call_binTM (lua_State *L, const TValue *p1, const TValue *p2,
|
|
+ StkId res, TMS event) {
|
|
+ const TValue *tm = luaT_gettmbyobj(L, p1, event); /* try first operand */
|
|
+ if (ttisnil(tm))
|
|
+ tm = luaT_gettmbyobj(L, p2, event); /* try second operand */
|
|
+ if (ttisnil(tm)) return 0;
|
|
+ callTMres(L, res, tm, p1, p2);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+static const TValue *get_compTM (lua_State *L, Table *mt1, Table *mt2,
|
|
+ TMS event) {
|
|
+ const TValue *tm1 = fasttm(L, mt1, event);
|
|
+ const TValue *tm2;
|
|
+ if (tm1 == NULL) return NULL; /* no metamethod */
|
|
+ if (mt1 == mt2) return tm1; /* same metatables => same metamethods */
|
|
+ tm2 = fasttm(L, mt2, event);
|
|
+ if (tm2 == NULL) return NULL; /* no metamethod */
|
|
+ if (luaO_rawequalObj(tm1, tm2)) /* same metamethods? */
|
|
+ return tm1;
|
|
+ return NULL;
|
|
+}
|
|
+
|
|
+
|
|
+static int call_orderTM (lua_State *L, const TValue *p1, const TValue *p2,
|
|
+ TMS event) {
|
|
+ const TValue *tm1 = luaT_gettmbyobj(L, p1, event);
|
|
+ const TValue *tm2;
|
|
+ if (ttisnil(tm1)) return -1; /* no metamethod? */
|
|
+ tm2 = luaT_gettmbyobj(L, p2, event);
|
|
+ if (!luaO_rawequalObj(tm1, tm2)) /* different metamethods? */
|
|
+ return -1;
|
|
+ callTMres(L, L->top, tm1, p1, p2);
|
|
+ return !l_isfalse(L->top);
|
|
+}
|
|
+
|
|
+
|
|
+static int l_strcmp (const TString *ls, const TString *rs) {
|
|
+ const char *l = getstr(ls);
|
|
+ size_t ll = ls->tsv.len;
|
|
+ const char *r = getstr(rs);
|
|
+ size_t lr = rs->tsv.len;
|
|
+ for (;;) {
|
|
+ int temp = strcoll(l, r);
|
|
+ if (temp != 0) return temp;
|
|
+ else { /* strings are equal up to a `\0' */
|
|
+ size_t len = strlen(l); /* index of first `\0' in both strings */
|
|
+ if (len == lr) /* r is finished? */
|
|
+ return (len == ll) ? 0 : 1;
|
|
+ else if (len == ll) /* l is finished? */
|
|
+ return -1; /* l is smaller than r (because r is not finished) */
|
|
+ /* both strings longer than `len'; go on comparing (after the `\0') */
|
|
+ len++;
|
|
+ l += len; ll -= len; r += len; lr -= len;
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
+
|
|
+int luaV_lessthan (lua_State *L, const TValue *l, const TValue *r) {
|
|
+ int res;
|
|
+ if (ttype(l) != ttype(r))
|
|
+ return luaG_ordererror(L, l, r);
|
|
+ else if (ttisnumber(l))
|
|
+ return luai_numlt(nvalue(l), nvalue(r));
|
|
+ else if (ttisstring(l))
|
|
+ return l_strcmp(rawtsvalue(l), rawtsvalue(r)) < 0;
|
|
+ else if ((res = call_orderTM(L, l, r, TM_LT)) != -1)
|
|
+ return res;
|
|
+ return luaG_ordererror(L, l, r);
|
|
+}
|
|
+
|
|
+
|
|
+static int lessequal (lua_State *L, const TValue *l, const TValue *r) {
|
|
+ int res;
|
|
+ if (ttype(l) != ttype(r))
|
|
+ return luaG_ordererror(L, l, r);
|
|
+ else if (ttisnumber(l))
|
|
+ return luai_numle(nvalue(l), nvalue(r));
|
|
+ else if (ttisstring(l))
|
|
+ return l_strcmp(rawtsvalue(l), rawtsvalue(r)) <= 0;
|
|
+ else if ((res = call_orderTM(L, l, r, TM_LE)) != -1) /* first try `le' */
|
|
+ return res;
|
|
+ else if ((res = call_orderTM(L, r, l, TM_LT)) != -1) /* else try `lt' */
|
|
+ return !res;
|
|
+ return luaG_ordererror(L, l, r);
|
|
+}
|
|
+
|
|
+
|
|
+int luaV_equalval (lua_State *L, const TValue *t1, const TValue *t2) {
|
|
+ const TValue *tm;
|
|
+ lua_assert(ttype(t1) == ttype(t2));
|
|
+ switch (ttype(t1)) {
|
|
+ case LUA_TNIL: return 1;
|
|
+ case LUA_TNUMBER: return luai_numeq(nvalue(t1), nvalue(t2));
|
|
+ case LUA_TBOOLEAN: return bvalue(t1) == bvalue(t2); /* true must be 1 !! */
|
|
+ case LUA_TLIGHTUSERDATA: return pvalue(t1) == pvalue(t2);
|
|
+ case LUA_TUSERDATA: {
|
|
+ if (uvalue(t1) == uvalue(t2)) return 1;
|
|
+ tm = get_compTM(L, uvalue(t1)->metatable, uvalue(t2)->metatable,
|
|
+ TM_EQ);
|
|
+ break; /* will try TM */
|
|
+ }
|
|
+ case LUA_TTABLE: {
|
|
+ if (hvalue(t1) == hvalue(t2)) return 1;
|
|
+ tm = get_compTM(L, hvalue(t1)->metatable, hvalue(t2)->metatable, TM_EQ);
|
|
+ break; /* will try TM */
|
|
+ }
|
|
+ default: return gcvalue(t1) == gcvalue(t2);
|
|
+ }
|
|
+ if (tm == NULL) return 0; /* no TM? */
|
|
+ callTMres(L, L->top, tm, t1, t2); /* call TM */
|
|
+ return !l_isfalse(L->top);
|
|
+}
|
|
+
|
|
+
|
|
+void luaV_concat (lua_State *L, int total, int last) {
|
|
+ do {
|
|
+ StkId top = L->base + last + 1;
|
|
+ int n = 2; /* number of elements handled in this pass (at least 2) */
|
|
+ if (!(ttisstring(top-2) || ttisnumber(top-2)) || !tostring(L, top-1)) {
|
|
+ if (!call_binTM(L, top-2, top-1, top-2, TM_CONCAT))
|
|
+ luaG_concaterror(L, top-2, top-1);
|
|
+ } else if (tsvalue(top-1)->len == 0) /* second op is empty? */
|
|
+ (void)tostring(L, top - 2); /* result is first op (as string) */
|
|
+ else {
|
|
+ /* at least two string values; get as many as possible */
|
|
+ size_t tl = tsvalue(top-1)->len;
|
|
+ char *buffer;
|
|
+ int i;
|
|
+ /* collect total length */
|
|
+ for (n = 1; n < total && tostring(L, top-n-1); n++) {
|
|
+ size_t l = tsvalue(top-n-1)->len;
|
|
+ if (l >= MAX_SIZET - tl) luaG_runerror(L, "string length overflow");
|
|
+ tl += l;
|
|
+ }
|
|
+ buffer = luaZ_openspace(L, &G(L)->buff, tl);
|
|
+ tl = 0;
|
|
+ for (i=n; i>0; i--) { /* concat all strings */
|
|
+ size_t l = tsvalue(top-i)->len;
|
|
+ memcpy(buffer+tl, svalue(top-i), l);
|
|
+ tl += l;
|
|
+ }
|
|
+ setsvalue2s(L, top-n, luaS_newlstr(L, buffer, tl));
|
|
+ }
|
|
+ total -= n-1; /* got `n' strings to create 1 new */
|
|
+ last -= n-1;
|
|
+ } while (total > 1); /* repeat until only 1 result left */
|
|
+}
|
|
+
|
|
+
|
|
+static void Arith (lua_State *L, StkId ra, const TValue *rb,
|
|
+ const TValue *rc, TMS op) {
|
|
+ TValue tempb, tempc;
|
|
+ const TValue *b, *c;
|
|
+ if ((b = luaV_tonumber(rb, &tempb)) != NULL &&
|
|
+ (c = luaV_tonumber(rc, &tempc)) != NULL) {
|
|
+ lua_Number nb = nvalue(b), nc = nvalue(c);
|
|
+ switch (op) {
|
|
+ case TM_ADD: setnvalue(ra, luai_numadd(nb, nc)); break;
|
|
+ case TM_SUB: setnvalue(ra, luai_numsub(nb, nc)); break;
|
|
+ case TM_MUL: setnvalue(ra, luai_nummul(nb, nc)); break;
|
|
+ case TM_DIV: setnvalue(ra, luai_numdiv(nb, nc)); break;
|
|
+ case TM_MOD: setnvalue(ra, luai_nummod(nb, nc)); break;
|
|
+ case TM_POW: setnvalue(ra, luai_numpow(nb, nc)); break;
|
|
+ case TM_UNM: setnvalue(ra, luai_numunm(nb)); break;
|
|
+ default: lua_assert(0); break;
|
|
+ }
|
|
+ }
|
|
+ else if (!call_binTM(L, rb, rc, ra, op))
|
|
+ luaG_aritherror(L, rb, rc);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+/*
|
|
+** some macros for common tasks in `luaV_execute'
|
|
+*/
|
|
+
|
|
+#define runtime_check(L, c) { if (!(c)) break; }
|
|
+
|
|
+#define RA(i) (base+GETARG_A(i))
|
|
+/* to be used after possible stack reallocation */
|
|
+#define RB(i) check_exp(getBMode(GET_OPCODE(i)) == OpArgR, base+GETARG_B(i))
|
|
+#define RC(i) check_exp(getCMode(GET_OPCODE(i)) == OpArgR, base+GETARG_C(i))
|
|
+#define RKB(i) check_exp(getBMode(GET_OPCODE(i)) == OpArgK, \
|
|
+ ISK(GETARG_B(i)) ? k+INDEXK(GETARG_B(i)) : base+GETARG_B(i))
|
|
+#define RKC(i) check_exp(getCMode(GET_OPCODE(i)) == OpArgK, \
|
|
+ ISK(GETARG_C(i)) ? k+INDEXK(GETARG_C(i)) : base+GETARG_C(i))
|
|
+#define KBx(i) check_exp(getBMode(GET_OPCODE(i)) == OpArgK, k+GETARG_Bx(i))
|
|
+
|
|
+
|
|
+#define dojump(L,pc,i) {(pc) += (i); luai_threadyield(L);}
|
|
+
|
|
+
|
|
+#define Protect(x) { L->savedpc = pc; {x;}; base = L->base; }
|
|
+
|
|
+
|
|
+#define arith_op(op,tm) { \
|
|
+ TValue *rb = RKB(i); \
|
|
+ TValue *rc = RKC(i); \
|
|
+ if (ttisnumber(rb) && ttisnumber(rc)) { \
|
|
+ lua_Number nb = nvalue(rb), nc = nvalue(rc); \
|
|
+ setnvalue(ra, op(nb, nc)); \
|
|
+ } \
|
|
+ else \
|
|
+ Protect(Arith(L, ra, rb, rc, tm)); \
|
|
+ }
|
|
+
|
|
+
|
|
+
|
|
+void luaV_execute (lua_State *L, int nexeccalls) {
|
|
+ LClosure *cl;
|
|
+ StkId base;
|
|
+ TValue *k;
|
|
+ const Instruction *pc;
|
|
+ reentry: /* entry point */
|
|
+ lua_assert(isLua(L->ci));
|
|
+ pc = L->savedpc;
|
|
+ cl = &clvalue(L->ci->func)->l;
|
|
+ base = L->base;
|
|
+ k = cl->p->k;
|
|
+ /* main loop of interpreter */
|
|
+ for (;;) {
|
|
+ const Instruction i = *pc++;
|
|
+ StkId ra;
|
|
+ if ((L->hookmask & (LUA_MASKLINE | LUA_MASKCOUNT)) &&
|
|
+ (--L->hookcount == 0 || L->hookmask & LUA_MASKLINE)) {
|
|
+ traceexec(L, pc);
|
|
+ if (L->status == LUA_YIELD) { /* did hook yield? */
|
|
+ L->savedpc = pc - 1;
|
|
+ return;
|
|
+ }
|
|
+ base = L->base;
|
|
+ }
|
|
+ /* warning!! several calls may realloc the stack and invalidate `ra' */
|
|
+ ra = RA(i);
|
|
+ lua_assert(base == L->base && L->base == L->ci->base);
|
|
+ lua_assert(base <= L->top && L->top <= L->stack + L->stacksize);
|
|
+ lua_assert(L->top == L->ci->top || luaG_checkopenop(i));
|
|
+ switch (GET_OPCODE(i)) {
|
|
+ case OP_MOVE: {
|
|
+ setobjs2s(L, ra, RB(i));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LOADK: {
|
|
+ setobj2s(L, ra, KBx(i));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LOADBOOL: {
|
|
+ setbvalue(ra, GETARG_B(i));
|
|
+ if (GETARG_C(i)) pc++; /* skip next instruction (if C) */
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LOADNIL: {
|
|
+ TValue *rb = RB(i);
|
|
+ do {
|
|
+ setnilvalue(rb--);
|
|
+ } while (rb >= ra);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_GETUPVAL: {
|
|
+ int b = GETARG_B(i);
|
|
+ setobj2s(L, ra, cl->upvals[b]->v);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_GETGLOBAL: {
|
|
+ TValue g;
|
|
+ TValue *rb = KBx(i);
|
|
+ sethvalue(L, &g, cl->env);
|
|
+ lua_assert(ttisstring(rb));
|
|
+ Protect(luaV_gettable(L, &g, rb, ra));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_GETTABLE: {
|
|
+ Protect(luaV_gettable(L, RB(i), RKC(i), ra));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SETGLOBAL: {
|
|
+ TValue g;
|
|
+ sethvalue(L, &g, cl->env);
|
|
+ lua_assert(ttisstring(KBx(i)));
|
|
+ Protect(luaV_settable(L, &g, KBx(i), ra));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SETUPVAL: {
|
|
+ UpVal *uv = cl->upvals[GETARG_B(i)];
|
|
+ setobj(L, uv->v, ra);
|
|
+ luaC_barrier(L, uv, ra);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SETTABLE: {
|
|
+ Protect(luaV_settable(L, ra, RKB(i), RKC(i)));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_NEWTABLE: {
|
|
+ int b = GETARG_B(i);
|
|
+ int c = GETARG_C(i);
|
|
+ sethvalue(L, ra, luaH_new(L, luaO_fb2int(b), luaO_fb2int(c)));
|
|
+ Protect(luaC_checkGC(L));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SELF: {
|
|
+ StkId rb = RB(i);
|
|
+ setobjs2s(L, ra+1, rb);
|
|
+ Protect(luaV_gettable(L, rb, RKC(i), ra));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_ADD: {
|
|
+ arith_op(luai_numadd, TM_ADD);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SUB: {
|
|
+ arith_op(luai_numsub, TM_SUB);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_MUL: {
|
|
+ arith_op(luai_nummul, TM_MUL);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_DIV: {
|
|
+ arith_op(luai_numdiv, TM_DIV);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_MOD: {
|
|
+ arith_op(luai_nummod, TM_MOD);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_POW: {
|
|
+ arith_op(luai_numpow, TM_POW);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_UNM: {
|
|
+ TValue *rb = RB(i);
|
|
+ if (ttisnumber(rb)) {
|
|
+ lua_Number nb = nvalue(rb);
|
|
+ setnvalue(ra, luai_numunm(nb));
|
|
+ }
|
|
+ else {
|
|
+ Protect(Arith(L, ra, rb, rb, TM_UNM));
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ case OP_NOT: {
|
|
+ int res = l_isfalse(RB(i)); /* next assignment may change this value */
|
|
+ setbvalue(ra, res);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LEN: {
|
|
+ const TValue *rb = RB(i);
|
|
+ switch (ttype(rb)) {
|
|
+ case LUA_TTABLE: {
|
|
+ setnvalue(ra, cast_num(luaH_getn(hvalue(rb))));
|
|
+ break;
|
|
+ }
|
|
+ case LUA_TSTRING: {
|
|
+ setnvalue(ra, cast_num(tsvalue(rb)->len));
|
|
+ break;
|
|
+ }
|
|
+ default: { /* try metamethod */
|
|
+ Protect(
|
|
+ if (!call_binTM(L, rb, luaO_nilobject, ra, TM_LEN))
|
|
+ luaG_typeerror(L, rb, "get length of");
|
|
+ )
|
|
+ }
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ case OP_CONCAT: {
|
|
+ int b = GETARG_B(i);
|
|
+ int c = GETARG_C(i);
|
|
+ Protect(luaV_concat(L, c-b+1, c); luaC_checkGC(L));
|
|
+ setobjs2s(L, RA(i), base+b);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_JMP: {
|
|
+ dojump(L, pc, GETARG_sBx(i));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_EQ: {
|
|
+ TValue *rb = RKB(i);
|
|
+ TValue *rc = RKC(i);
|
|
+ Protect(
|
|
+ if (equalobj(L, rb, rc) == GETARG_A(i))
|
|
+ dojump(L, pc, GETARG_sBx(*pc));
|
|
+ )
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LT: {
|
|
+ Protect(
|
|
+ if (luaV_lessthan(L, RKB(i), RKC(i)) == GETARG_A(i))
|
|
+ dojump(L, pc, GETARG_sBx(*pc));
|
|
+ )
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_LE: {
|
|
+ Protect(
|
|
+ if (lessequal(L, RKB(i), RKC(i)) == GETARG_A(i))
|
|
+ dojump(L, pc, GETARG_sBx(*pc));
|
|
+ )
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_TEST: {
|
|
+ if (l_isfalse(ra) != GETARG_C(i))
|
|
+ dojump(L, pc, GETARG_sBx(*pc));
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_TESTSET: {
|
|
+ TValue *rb = RB(i);
|
|
+ if (l_isfalse(rb) != GETARG_C(i)) {
|
|
+ setobjs2s(L, ra, rb);
|
|
+ dojump(L, pc, GETARG_sBx(*pc));
|
|
+ }
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_CALL: {
|
|
+ int b = GETARG_B(i);
|
|
+ int nresults = GETARG_C(i) - 1;
|
|
+ if (b != 0) L->top = ra+b; /* else previous instruction set top */
|
|
+ L->savedpc = pc;
|
|
+ switch (luaD_precall(L, ra, nresults)) {
|
|
+ case PCRLUA: {
|
|
+ nexeccalls++;
|
|
+ goto reentry; /* restart luaV_execute over new Lua function */
|
|
+ }
|
|
+ case PCRC: {
|
|
+ /* it was a C function (`precall' called it); adjust results */
|
|
+ if (nresults >= 0) L->top = L->ci->top;
|
|
+ base = L->base;
|
|
+ continue;
|
|
+ }
|
|
+ default: {
|
|
+ return; /* yield */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ case OP_TAILCALL: {
|
|
+ int b = GETARG_B(i);
|
|
+ if (b != 0) L->top = ra+b; /* else previous instruction set top */
|
|
+ L->savedpc = pc;
|
|
+ lua_assert(GETARG_C(i) - 1 == LUA_MULTRET);
|
|
+ switch (luaD_precall(L, ra, LUA_MULTRET)) {
|
|
+ case PCRLUA: {
|
|
+ /* tail call: put new frame in place of previous one */
|
|
+ CallInfo *ci = L->ci - 1; /* previous frame */
|
|
+ int aux;
|
|
+ StkId func = ci->func;
|
|
+ StkId pfunc = (ci+1)->func; /* previous function index */
|
|
+ if (L->openupval) luaF_close(L, ci->base);
|
|
+ L->base = ci->base = ci->func + ((ci+1)->base - pfunc);
|
|
+ for (aux = 0; pfunc+aux < L->top; aux++) /* move frame down */
|
|
+ setobjs2s(L, func+aux, pfunc+aux);
|
|
+ ci->top = L->top = func+aux; /* correct top */
|
|
+ lua_assert(L->top == L->base + clvalue(func)->l.p->maxstacksize);
|
|
+ ci->savedpc = L->savedpc;
|
|
+ ci->tailcalls++; /* one more call lost */
|
|
+ L->ci--; /* remove new frame */
|
|
+ goto reentry;
|
|
+ }
|
|
+ case PCRC: { /* it was a C function (`precall' called it) */
|
|
+ base = L->base;
|
|
+ continue;
|
|
+ }
|
|
+ default: {
|
|
+ return; /* yield */
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+ case OP_RETURN: {
|
|
+ int b = GETARG_B(i);
|
|
+ if (b != 0) L->top = ra+b-1;
|
|
+ if (L->openupval) luaF_close(L, base);
|
|
+ L->savedpc = pc;
|
|
+ b = luaD_poscall(L, ra);
|
|
+ if (--nexeccalls == 0) /* was previous function running `here'? */
|
|
+ return; /* no: return */
|
|
+ else { /* yes: continue its execution */
|
|
+ if (b) L->top = L->ci->top;
|
|
+ lua_assert(isLua(L->ci));
|
|
+ lua_assert(GET_OPCODE(*((L->ci)->savedpc - 1)) == OP_CALL);
|
|
+ goto reentry;
|
|
+ }
|
|
+ }
|
|
+ case OP_FORLOOP: {
|
|
+ lua_Number step = nvalue(ra+2);
|
|
+ lua_Number idx = luai_numadd(nvalue(ra), step); /* increment index */
|
|
+ lua_Number limit = nvalue(ra+1);
|
|
+ if (luai_numlt(0, step) ? luai_numle(idx, limit)
|
|
+ : luai_numle(limit, idx)) {
|
|
+ dojump(L, pc, GETARG_sBx(i)); /* jump back */
|
|
+ setnvalue(ra, idx); /* update internal index... */
|
|
+ setnvalue(ra+3, idx); /* ...and external index */
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ case OP_FORPREP: {
|
|
+ const TValue *init = ra;
|
|
+ const TValue *plimit = ra+1;
|
|
+ const TValue *pstep = ra+2;
|
|
+ L->savedpc = pc; /* next steps may throw errors */
|
|
+ if (!tonumber(init, ra))
|
|
+ luaG_runerror(L, LUA_QL("for") " initial value must be a number");
|
|
+ else if (!tonumber(plimit, ra+1))
|
|
+ luaG_runerror(L, LUA_QL("for") " limit must be a number");
|
|
+ else if (!tonumber(pstep, ra+2))
|
|
+ luaG_runerror(L, LUA_QL("for") " step must be a number");
|
|
+ setnvalue(ra, luai_numsub(nvalue(ra), nvalue(pstep)));
|
|
+ dojump(L, pc, GETARG_sBx(i));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_TFORLOOP: {
|
|
+ StkId cb = ra + 3; /* call base */
|
|
+ setobjs2s(L, cb+2, ra+2);
|
|
+ setobjs2s(L, cb+1, ra+1);
|
|
+ setobjs2s(L, cb, ra);
|
|
+ L->top = cb+3; /* func. + 2 args (state and index) */
|
|
+ Protect(luaD_call(L, cb, GETARG_C(i)));
|
|
+ L->top = L->ci->top;
|
|
+ cb = RA(i) + 3; /* previous call may change the stack */
|
|
+ if (!ttisnil(cb)) { /* continue loop? */
|
|
+ setobjs2s(L, cb-1, cb); /* save control variable */
|
|
+ dojump(L, pc, GETARG_sBx(*pc)); /* jump back */
|
|
+ }
|
|
+ pc++;
|
|
+ continue;
|
|
+ }
|
|
+ case OP_SETLIST: {
|
|
+ int n = GETARG_B(i);
|
|
+ int c = GETARG_C(i);
|
|
+ int last;
|
|
+ Table *h;
|
|
+ if (n == 0) {
|
|
+ n = cast_int(L->top - ra) - 1;
|
|
+ L->top = L->ci->top;
|
|
+ }
|
|
+ if (c == 0) c = cast_int(*pc++);
|
|
+ runtime_check(L, ttistable(ra));
|
|
+ h = hvalue(ra);
|
|
+ last = ((c-1)*LFIELDS_PER_FLUSH) + n;
|
|
+ if (last > h->sizearray) /* needs more space? */
|
|
+ luaH_resizearray(L, h, last); /* pre-alloc it at once */
|
|
+ for (; n > 0; n--) {
|
|
+ TValue *val = ra+n;
|
|
+ setobj2t(L, luaH_setnum(L, h, last--), val);
|
|
+ luaC_barriert(L, h, val);
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ case OP_CLOSE: {
|
|
+ luaF_close(L, ra);
|
|
+ continue;
|
|
+ }
|
|
+ case OP_CLOSURE: {
|
|
+ Proto *p;
|
|
+ Closure *ncl;
|
|
+ int nup, j;
|
|
+ p = cl->p->p[GETARG_Bx(i)];
|
|
+ nup = p->nups;
|
|
+ ncl = luaF_newLclosure(L, nup, cl->env);
|
|
+ ncl->l.p = p;
|
|
+ for (j=0; j<nup; j++, pc++) {
|
|
+ if (GET_OPCODE(*pc) == OP_GETUPVAL)
|
|
+ ncl->l.upvals[j] = cl->upvals[GETARG_B(*pc)];
|
|
+ else {
|
|
+ lua_assert(GET_OPCODE(*pc) == OP_MOVE);
|
|
+ ncl->l.upvals[j] = luaF_findupval(L, base + GETARG_B(*pc));
|
|
+ }
|
|
+ }
|
|
+ setclvalue(L, ra, ncl);
|
|
+ Protect(luaC_checkGC(L));
|
|
+ continue;
|
|
+ }
|
|
+ case OP_VARARG: {
|
|
+ int b = GETARG_B(i) - 1;
|
|
+ int j;
|
|
+ CallInfo *ci = L->ci;
|
|
+ int n = cast_int(ci->base - ci->func) - cl->p->numparams - 1;
|
|
+ if (b == LUA_MULTRET) {
|
|
+ Protect(luaD_checkstack(L, n));
|
|
+ ra = RA(i); /* previous call may change the stack */
|
|
+ b = n;
|
|
+ L->top = ra + n;
|
|
+ }
|
|
+ for (j = 0; j < b; j++) {
|
|
+ if (j < n) {
|
|
+ setobjs2s(L, ra + j, ci->base - n + j);
|
|
+ }
|
|
+ else {
|
|
+ setnilvalue(ra + j);
|
|
+ }
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ }
|
|
+ }
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lvm.h
|
|
@@ -0,0 +1,36 @@
|
|
+/*
|
|
+** $Id: lvm.h,v 2.5.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Lua virtual machine
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#ifndef lvm_h
|
|
+#define lvm_h
|
|
+
|
|
+
|
|
+#include "ldo.h"
|
|
+#include "lobject.h"
|
|
+#include "ltm.h"
|
|
+
|
|
+
|
|
+#define tostring(L,o) ((ttype(o) == LUA_TSTRING) || (luaV_tostring(L, o)))
|
|
+
|
|
+#define tonumber(o,n) (ttype(o) == LUA_TNUMBER || \
|
|
+ (((o) = luaV_tonumber(o,n)) != NULL))
|
|
+
|
|
+#define equalobj(L,o1,o2) \
|
|
+ (ttype(o1) == ttype(o2) && luaV_equalval(L, o1, o2))
|
|
+
|
|
+
|
|
+LUAI_FUNC int luaV_lessthan (lua_State *L, const TValue *l, const TValue *r);
|
|
+LUAI_FUNC int luaV_equalval (lua_State *L, const TValue *t1, const TValue *t2);
|
|
+LUAI_FUNC const TValue *luaV_tonumber (const TValue *obj, TValue *n);
|
|
+LUAI_FUNC int luaV_tostring (lua_State *L, StkId obj);
|
|
+LUAI_FUNC void luaV_gettable (lua_State *L, const TValue *t, TValue *key,
|
|
+ StkId val);
|
|
+LUAI_FUNC void luaV_settable (lua_State *L, const TValue *t, TValue *key,
|
|
+ StkId val);
|
|
+LUAI_FUNC void luaV_execute (lua_State *L, int nexeccalls);
|
|
+LUAI_FUNC void luaV_concat (lua_State *L, int total, int last);
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lzio.c
|
|
@@ -0,0 +1,81 @@
|
|
+/*
|
|
+** $Id: lzio.c,v 1.31.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** a generic input stream interface
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+#include <string.h>
|
|
+
|
|
+#define lzio_c
|
|
+#define LUA_CORE
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "llimits.h"
|
|
+#include "lmem.h"
|
|
+#include "lstate.h"
|
|
+#include "lzio.h"
|
|
+
|
|
+
|
|
+int luaZ_fill (ZIO *z) {
|
|
+ size_t size;
|
|
+ lua_State *L = z->L;
|
|
+ const char *buff;
|
|
+ lua_unlock(L);
|
|
+ buff = z->reader(L, z->data, &size);
|
|
+ lua_lock(L);
|
|
+ if (buff == NULL || size == 0) return EOZ;
|
|
+ z->n = size - 1;
|
|
+ z->p = buff;
|
|
+ return char2int(*(z->p++));
|
|
+}
|
|
+
|
|
+
|
|
+int luaZ_lookahead (ZIO *z) {
|
|
+ if (z->n == 0) {
|
|
+ if (luaZ_fill(z) == EOZ)
|
|
+ return EOZ;
|
|
+ else {
|
|
+ z->n++; /* luaZ_fill removed first byte; put back it */
|
|
+ z->p--;
|
|
+ }
|
|
+ }
|
|
+ return char2int(*z->p);
|
|
+}
|
|
+
|
|
+
|
|
+void luaZ_init (lua_State *L, ZIO *z, lua_Reader reader, void *data) {
|
|
+ z->L = L;
|
|
+ z->reader = reader;
|
|
+ z->data = data;
|
|
+ z->n = 0;
|
|
+ z->p = NULL;
|
|
+}
|
|
+
|
|
+
|
|
+/* --------------------------------------------------------------- read --- */
|
|
+size_t luaZ_read (ZIO *z, void *b, size_t n) {
|
|
+ while (n) {
|
|
+ size_t m;
|
|
+ if (luaZ_lookahead(z) == EOZ)
|
|
+ return n; /* return number of missing bytes */
|
|
+ m = (n <= z->n) ? n : z->n; /* min. between n and z->n */
|
|
+ memcpy(b, z->p, m);
|
|
+ z->n -= m;
|
|
+ z->p += m;
|
|
+ b = (char *)b + m;
|
|
+ n -= m;
|
|
+ }
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+/* ------------------------------------------------------------------------ */
|
|
+char *luaZ_openspace (lua_State *L, Mbuffer *buff, size_t n) {
|
|
+ if (n > buff->buffsize) {
|
|
+ if (n < LUA_MINBUFFER) n = LUA_MINBUFFER;
|
|
+ luaZ_resizebuffer(L, buff, n);
|
|
+ }
|
|
+ return buff->buffer;
|
|
+}
|
|
+
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/lua/lzio.h
|
|
@@ -0,0 +1,67 @@
|
|
+/*
|
|
+** $Id: lzio.h,v 1.21.1.1 2007/12/27 13:02:25 roberto Exp $
|
|
+** Buffered streams
|
|
+** See Copyright Notice in lua.h
|
|
+*/
|
|
+
|
|
+
|
|
+#ifndef lzio_h
|
|
+#define lzio_h
|
|
+
|
|
+#include "lua.h"
|
|
+
|
|
+#include "lmem.h"
|
|
+
|
|
+
|
|
+#define EOZ (-1) /* end of stream */
|
|
+
|
|
+typedef struct Zio ZIO;
|
|
+
|
|
+#define char2int(c) cast(int, cast(unsigned char, (c)))
|
|
+
|
|
+#define zgetc(z) (((z)->n--)>0 ? char2int(*(z)->p++) : luaZ_fill(z))
|
|
+
|
|
+typedef struct Mbuffer {
|
|
+ char *buffer;
|
|
+ size_t n;
|
|
+ size_t buffsize;
|
|
+} Mbuffer;
|
|
+
|
|
+#define luaZ_initbuffer(L, buff) ((buff)->buffer = NULL, (buff)->buffsize = 0)
|
|
+
|
|
+#define luaZ_buffer(buff) ((buff)->buffer)
|
|
+#define luaZ_sizebuffer(buff) ((buff)->buffsize)
|
|
+#define luaZ_bufflen(buff) ((buff)->n)
|
|
+
|
|
+#define luaZ_resetbuffer(buff) ((buff)->n = 0)
|
|
+
|
|
+
|
|
+#define luaZ_resizebuffer(L, buff, size) \
|
|
+ (luaM_reallocvector(L, (buff)->buffer, (buff)->buffsize, size, char), \
|
|
+ (buff)->buffsize = size)
|
|
+
|
|
+#define luaZ_freebuffer(L, buff) luaZ_resizebuffer(L, buff, 0)
|
|
+
|
|
+
|
|
+LUAI_FUNC char *luaZ_openspace (lua_State *L, Mbuffer *buff, size_t n);
|
|
+LUAI_FUNC void luaZ_init (lua_State *L, ZIO *z, lua_Reader reader,
|
|
+ void *data);
|
|
+LUAI_FUNC size_t luaZ_read (ZIO* z, void* b, size_t n); /* read next n bytes */
|
|
+LUAI_FUNC int luaZ_lookahead (ZIO *z);
|
|
+
|
|
+
|
|
+
|
|
+/* --------- Private Part ------------------ */
|
|
+
|
|
+struct Zio {
|
|
+ size_t n; /* bytes still unread */
|
|
+ const char *p; /* current position in buffer */
|
|
+ lua_Reader reader;
|
|
+ void* data; /* additional data */
|
|
+ lua_State *L; /* Lua state (for reader) */
|
|
+};
|
|
+
|
|
+
|
|
+LUAI_FUNC int luaZ_fill (ZIO *z);
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/Makefile
|
|
@@ -0,0 +1,389 @@
|
|
+# Makefile.in generated by automake 1.11.1 from Makefile.am.
|
|
+# extensions/LUA/Makefile. Generated from Makefile.in by configure.
|
|
+
|
|
+# Copyright (C) 1994, 1995, 1996, 1997, 1998, 1999, 2000, 2001, 2002,
|
|
+# 2003, 2004, 2005, 2006, 2007, 2008, 2009 Free Software Foundation,
|
|
+# Inc.
|
|
+# This Makefile.in is free software; the Free Software Foundation
|
|
+# gives unlimited permission to copy and/or distribute it,
|
|
+# with or without modifications, as long as this notice is preserved.
|
|
+
|
|
+# This program is distributed in the hope that it will be useful,
|
|
+# but WITHOUT ANY WARRANTY, to the extent permitted by law; without
|
|
+# even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
+# PARTICULAR PURPOSE.
|
|
+
|
|
+
|
|
+
|
|
+# -*- Makefile -*-
|
|
+# AUTOMAKE
|
|
+
|
|
+pkgdatadir = $(datadir)/xtables-addons
|
|
+pkgincludedir = $(includedir)/xtables-addons
|
|
+pkglibdir = $(libdir)/xtables-addons
|
|
+pkglibexecdir = $(libexecdir)/xtables-addons
|
|
+am__cd = CDPATH="$${ZSH_VERSION+.}$(PATH_SEPARATOR)" && cd
|
|
+install_sh_DATA = $(install_sh) -c -m 644
|
|
+install_sh_PROGRAM = $(install_sh) -c
|
|
+install_sh_SCRIPT = $(install_sh) -c
|
|
+INSTALL_HEADER = $(INSTALL_DATA)
|
|
+transform = $(program_transform_name)
|
|
+NORMAL_INSTALL = :
|
|
+PRE_INSTALL = :
|
|
+POST_INSTALL = :
|
|
+NORMAL_UNINSTALL = :
|
|
+PRE_UNINSTALL = :
|
|
+POST_UNINSTALL = :
|
|
+build_triplet = i686-pc-linux-gnu
|
|
+host_triplet = i686-pc-linux-gnu
|
|
+DIST_COMMON = $(srcdir)/../../Makefile.extra $(srcdir)/Makefile.am \
|
|
+ $(srcdir)/Makefile.in
|
|
+subdir = extensions/LUA
|
|
+ACLOCAL_M4 = $(top_srcdir)/aclocal.m4
|
|
+am__aclocal_m4_deps = $(top_srcdir)/m4/libtool.m4 \
|
|
+ $(top_srcdir)/m4/ltoptions.m4 $(top_srcdir)/m4/ltsugar.m4 \
|
|
+ $(top_srcdir)/m4/ltversion.m4 $(top_srcdir)/m4/lt~obsolete.m4 \
|
|
+ $(top_srcdir)/configure.ac
|
|
+am__configure_deps = $(am__aclocal_m4_deps) $(CONFIGURE_DEPENDENCIES) \
|
|
+ $(ACLOCAL_M4)
|
|
+mkinstalldirs = $(install_sh) -d
|
|
+CONFIG_HEADER = $(top_builddir)/config.h
|
|
+CONFIG_CLEAN_FILES =
|
|
+CONFIG_CLEAN_VPATH_FILES =
|
|
+SOURCES =
|
|
+DIST_SOURCES =
|
|
+DISTFILES = $(DIST_COMMON) $(DIST_SOURCES) $(TEXINFOS) $(EXTRA_DIST)
|
|
+ACLOCAL = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run aclocal-1.11
|
|
+AMTAR = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run tar
|
|
+AR = ar
|
|
+AUTOCONF = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run autoconf
|
|
+AUTOHEADER = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run autoheader
|
|
+AUTOMAKE = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run automake-1.11
|
|
+AWK = mawk
|
|
+CC = gcc
|
|
+CCDEPMODE = depmode=gcc3
|
|
+CFLAGS = -g -O2
|
|
+CPP = gcc -E
|
|
+CPPFLAGS =
|
|
+CYGPATH_W = echo
|
|
+DEFS = -DHAVE_CONFIG_H
|
|
+DEPDIR = .deps
|
|
+DSYMUTIL =
|
|
+DUMPBIN =
|
|
+ECHO_C =
|
|
+ECHO_N = -n
|
|
+ECHO_T =
|
|
+EGREP = /bin/grep -E
|
|
+EXEEXT =
|
|
+FGREP = /bin/grep -F
|
|
+GREP = /bin/grep
|
|
+INSTALL = /usr/bin/install -c
|
|
+INSTALL_DATA = ${INSTALL} -m 644
|
|
+INSTALL_PROGRAM = ${INSTALL}
|
|
+INSTALL_SCRIPT = ${INSTALL}
|
|
+INSTALL_STRIP_PROGRAM = $(install_sh) -c -s
|
|
+LD = /usr/bin/ld
|
|
+LDFLAGS =
|
|
+LIBOBJS =
|
|
+LIBS =
|
|
+LIBTOOL = $(SHELL) $(top_builddir)/libtool
|
|
+LIPO =
|
|
+LN_S = ln -s
|
|
+LTLIBOBJS =
|
|
+MAKEINFO = ${SHELL} /home/andre/Dropbox/xtables-addons/missing --run makeinfo
|
|
+MKDIR_P = /bin/mkdir -p
|
|
+NM = /usr/bin/nm -B
|
|
+NMEDIT =
|
|
+OBJDUMP = objdump
|
|
+OBJEXT = o
|
|
+OTOOL =
|
|
+OTOOL64 =
|
|
+PACKAGE = xtables-addons
|
|
+PACKAGE_BUGREPORT =
|
|
+PACKAGE_NAME = xtables-addons
|
|
+PACKAGE_STRING = xtables-addons 1.21
|
|
+PACKAGE_TARNAME = xtables-addons
|
|
+PACKAGE_URL =
|
|
+PACKAGE_VERSION = 1.21
|
|
+PATH_SEPARATOR = :
|
|
+PKG_CONFIG = /usr/bin/pkg-config
|
|
+RANLIB = ranlib
|
|
+SED = /bin/sed
|
|
+SET_MAKE =
|
|
+SHELL = /bin/bash
|
|
+STRIP = strip
|
|
+VERSION = 1.21
|
|
+abs_builddir = /home/andre/Dropbox/xtables-addons/extensions/LUA
|
|
+abs_srcdir = /home/andre/Dropbox/xtables-addons/extensions/LUA
|
|
+abs_top_builddir = /home/andre/Dropbox/xtables-addons
|
|
+abs_top_srcdir = /home/andre/Dropbox/xtables-addons
|
|
+ac_ct_CC = gcc
|
|
+ac_ct_DUMPBIN =
|
|
+am__include = include
|
|
+am__leading_dot = .
|
|
+am__quote =
|
|
+am__tar = ${AMTAR} chof - "$$tardir"
|
|
+am__untar = ${AMTAR} xf -
|
|
+bindir = ${exec_prefix}/bin
|
|
+build = i686-pc-linux-gnu
|
|
+build_alias =
|
|
+build_cpu = i686
|
|
+build_os = linux-gnu
|
|
+build_vendor = pc
|
|
+builddir = .
|
|
+datadir = ${datarootdir}
|
|
+datarootdir = ${prefix}/share
|
|
+docdir = ${datarootdir}/doc/${PACKAGE_TARNAME}
|
|
+dvidir = ${docdir}
|
|
+exec_prefix = ${prefix}
|
|
+host = i686-pc-linux-gnu
|
|
+host_alias =
|
|
+host_cpu = i686
|
|
+host_os = linux-gnu
|
|
+host_vendor = pc
|
|
+htmldir = ${docdir}
|
|
+includedir = ${prefix}/include
|
|
+infodir = ${datarootdir}/info
|
|
+install_sh = ${SHELL} /home/andre/Dropbox/xtables-addons/install-sh
|
|
+kbuilddir = /lib/modules/2.6.33-020633-generic/build
|
|
+kinclude_CFLAGS = -I /lib/modules/2.6.33-020633-generic/build/include
|
|
+ksourcedir =
|
|
+libdir = ${exec_prefix}/lib
|
|
+libexecdir = ${exec_prefix}/libexec
|
|
+libxtables_CFLAGS =
|
|
+libxtables_LIBS = -L/lib -lxtables
|
|
+localedir = ${datarootdir}/locale
|
|
+localstatedir = ${prefix}/var
|
|
+lt_ECHO = echo
|
|
+mandir = ${datarootdir}/man
|
|
+mkdir_p = /bin/mkdir -p
|
|
+oldincludedir = /usr/include
|
|
+pdfdir = ${docdir}
|
|
+prefix = /usr/local
|
|
+program_transform_name = s,x,x,
|
|
+psdir = ${docdir}
|
|
+regular_CFLAGS = -D_LARGEFILE_SOURCE=1 -D_LARGE_FILES -D_FILE_OFFSET_BITS=64 -D_REENTRANT -Wall -Waggregate-return -Wmissing-declarations -Wmissing-prototypes -Wredundant-decls -Wshadow -Wstrict-prototypes -Winline -pipe -DXTABLES_LIBDIR=\"${xtlibdir}\"
|
|
+sbindir = ${exec_prefix}/sbin
|
|
+sharedstatedir = ${prefix}/com
|
|
+srcdir = .
|
|
+sysconfdir = ${prefix}/etc
|
|
+target_alias =
|
|
+top_build_prefix = ../../
|
|
+top_builddir = ../..
|
|
+top_srcdir = ../..
|
|
+xtlibdir = ${libexecdir}/xtables
|
|
+XA_SRCDIR = ${srcdir}
|
|
+XA_TOPSRCDIR = ${top_srcdir}
|
|
+XA_ABSTOPSRCDIR = ${abs_top_srcdir}
|
|
+_mcall = -f ${top_builddir}/Makefile.iptrules
|
|
+all: all-am
|
|
+
|
|
+.SUFFIXES:
|
|
+$(srcdir)/Makefile.in: $(srcdir)/Makefile.am $(srcdir)/../../Makefile.extra $(am__configure_deps)
|
|
+ @for dep in $?; do \
|
|
+ case '$(am__configure_deps)' in \
|
|
+ *$$dep*) \
|
|
+ ( cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh ) \
|
|
+ && { if test -f $@; then exit 0; else break; fi; }; \
|
|
+ exit 1;; \
|
|
+ esac; \
|
|
+ done; \
|
|
+ echo ' cd $(top_srcdir) && $(AUTOMAKE) --foreign extensions/LUA/Makefile'; \
|
|
+ $(am__cd) $(top_srcdir) && \
|
|
+ $(AUTOMAKE) --foreign extensions/LUA/Makefile
|
|
+.PRECIOUS: Makefile
|
|
+Makefile: $(srcdir)/Makefile.in $(top_builddir)/config.status
|
|
+ @case '$?' in \
|
|
+ *config.status*) \
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh;; \
|
|
+ *) \
|
|
+ echo ' cd $(top_builddir) && $(SHELL) ./config.status $(subdir)/$@ $(am__depfiles_maybe)'; \
|
|
+ cd $(top_builddir) && $(SHELL) ./config.status $(subdir)/$@ $(am__depfiles_maybe);; \
|
|
+ esac;
|
|
+
|
|
+$(top_builddir)/config.status: $(top_srcdir)/configure $(CONFIG_STATUS_DEPENDENCIES)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+
|
|
+$(top_srcdir)/configure: $(am__configure_deps)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+$(ACLOCAL_M4): $(am__aclocal_m4_deps)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+$(am__aclocal_m4_deps):
|
|
+
|
|
+mostlyclean-libtool:
|
|
+ -rm -f *.lo
|
|
+
|
|
+clean-libtool:
|
|
+ -rm -rf .libs _libs
|
|
+tags: TAGS
|
|
+TAGS:
|
|
+
|
|
+ctags: CTAGS
|
|
+CTAGS:
|
|
+
|
|
+
|
|
+distdir: $(DISTFILES)
|
|
+ @srcdirstrip=`echo "$(srcdir)" | sed 's/[].[^$$\\*]/\\\\&/g'`; \
|
|
+ topsrcdirstrip=`echo "$(top_srcdir)" | sed 's/[].[^$$\\*]/\\\\&/g'`; \
|
|
+ list='$(DISTFILES)'; \
|
|
+ dist_files=`for file in $$list; do echo $$file; done | \
|
|
+ sed -e "s|^$$srcdirstrip/||;t" \
|
|
+ -e "s|^$$topsrcdirstrip/|$(top_builddir)/|;t"`; \
|
|
+ case $$dist_files in \
|
|
+ */*) $(MKDIR_P) `echo "$$dist_files" | \
|
|
+ sed '/\//!d;s|^|$(distdir)/|;s,/[^/]*$$,,' | \
|
|
+ sort -u` ;; \
|
|
+ esac; \
|
|
+ for file in $$dist_files; do \
|
|
+ if test -f $$file || test -d $$file; then d=.; else d=$(srcdir); fi; \
|
|
+ if test -d $$d/$$file; then \
|
|
+ dir=`echo "/$$file" | sed -e 's,/[^/]*$$,,'`; \
|
|
+ if test -d "$(distdir)/$$file"; then \
|
|
+ find "$(distdir)/$$file" -type d ! -perm -700 -exec chmod u+rwx {} \;; \
|
|
+ fi; \
|
|
+ if test -d $(srcdir)/$$file && test $$d != $(srcdir); then \
|
|
+ cp -fpR $(srcdir)/$$file "$(distdir)$$dir" || exit 1; \
|
|
+ find "$(distdir)/$$file" -type d ! -perm -700 -exec chmod u+rwx {} \;; \
|
|
+ fi; \
|
|
+ cp -fpR $$d/$$file "$(distdir)$$dir" || exit 1; \
|
|
+ else \
|
|
+ test -f "$(distdir)/$$file" \
|
|
+ || cp -p $$d/$$file "$(distdir)/$$file" \
|
|
+ || exit 1; \
|
|
+ fi; \
|
|
+ done
|
|
+check-am: all-am
|
|
+check: check-am
|
|
+all-am: Makefile all-local
|
|
+installdirs:
|
|
+install: install-am
|
|
+install-exec: install-exec-am
|
|
+install-data: install-data-am
|
|
+uninstall: uninstall-am
|
|
+
|
|
+install-am: all-am
|
|
+ @$(MAKE) $(AM_MAKEFLAGS) install-exec-am install-data-am
|
|
+
|
|
+installcheck: installcheck-am
|
|
+install-strip:
|
|
+ $(MAKE) $(AM_MAKEFLAGS) INSTALL_PROGRAM="$(INSTALL_STRIP_PROGRAM)" \
|
|
+ install_sh_PROGRAM="$(INSTALL_STRIP_PROGRAM)" INSTALL_STRIP_FLAG=-s \
|
|
+ `test -z '$(STRIP)' || \
|
|
+ echo "INSTALL_PROGRAM_ENV=STRIPPROG='$(STRIP)'"` install
|
|
+mostlyclean-generic:
|
|
+
|
|
+clean-generic:
|
|
+
|
|
+distclean-generic:
|
|
+ -test -z "$(CONFIG_CLEAN_FILES)" || rm -f $(CONFIG_CLEAN_FILES)
|
|
+ -test . = "$(srcdir)" || test -z "$(CONFIG_CLEAN_VPATH_FILES)" || rm -f $(CONFIG_CLEAN_VPATH_FILES)
|
|
+
|
|
+maintainer-clean-generic:
|
|
+ @echo "This command is intended for maintainers to use"
|
|
+ @echo "it deletes files that may require special tools to rebuild."
|
|
+clean: clean-am
|
|
+
|
|
+clean-am: clean-generic clean-libtool clean-local mostlyclean-am
|
|
+
|
|
+distclean: distclean-am
|
|
+ -rm -f Makefile
|
|
+distclean-am: clean-am distclean-generic
|
|
+
|
|
+dvi: dvi-am
|
|
+
|
|
+dvi-am:
|
|
+
|
|
+html: html-am
|
|
+
|
|
+html-am:
|
|
+
|
|
+info: info-am
|
|
+
|
|
+info-am:
|
|
+
|
|
+install-data-am:
|
|
+
|
|
+install-dvi: install-dvi-am
|
|
+
|
|
+install-dvi-am:
|
|
+
|
|
+install-exec-am: install-exec-local
|
|
+
|
|
+install-html: install-html-am
|
|
+
|
|
+install-html-am:
|
|
+
|
|
+install-info: install-info-am
|
|
+
|
|
+install-info-am:
|
|
+
|
|
+install-man:
|
|
+
|
|
+install-pdf: install-pdf-am
|
|
+
|
|
+install-pdf-am:
|
|
+
|
|
+install-ps: install-ps-am
|
|
+
|
|
+install-ps-am:
|
|
+
|
|
+installcheck-am:
|
|
+
|
|
+maintainer-clean: maintainer-clean-am
|
|
+ -rm -f Makefile
|
|
+maintainer-clean-am: distclean-am maintainer-clean-generic
|
|
+
|
|
+mostlyclean: mostlyclean-am
|
|
+
|
|
+mostlyclean-am: mostlyclean-generic mostlyclean-libtool
|
|
+
|
|
+pdf: pdf-am
|
|
+
|
|
+pdf-am:
|
|
+
|
|
+ps: ps-am
|
|
+
|
|
+ps-am:
|
|
+
|
|
+uninstall-am:
|
|
+
|
|
+.MAKE: install-am install-strip
|
|
+
|
|
+.PHONY: all all-am all-local check check-am clean clean-generic \
|
|
+ clean-libtool clean-local distclean distclean-generic \
|
|
+ distclean-libtool distdir dvi dvi-am html html-am info info-am \
|
|
+ install install-am install-data install-data-am install-dvi \
|
|
+ install-dvi-am install-exec install-exec-am install-exec-local \
|
|
+ install-html install-html-am install-info install-info-am \
|
|
+ install-man install-pdf install-pdf-am install-ps \
|
|
+ install-ps-am install-strip installcheck installcheck-am \
|
|
+ installdirs maintainer-clean maintainer-clean-generic \
|
|
+ mostlyclean mostlyclean-generic mostlyclean-libtool pdf pdf-am \
|
|
+ ps ps-am uninstall uninstall-am
|
|
+
|
|
+export XA_SRCDIR
|
|
+export XA_TOPSRCDIR
|
|
+export XA_ABSTOPSRCDIR
|
|
+
|
|
+all-local: user-all-local
|
|
+
|
|
+install-exec-local: user-install-local
|
|
+
|
|
+clean-local: user-clean-local
|
|
+
|
|
+user-all-local:
|
|
+ ${MAKE} ${_mcall} all;
|
|
+
|
|
+# Have no user-install-data-local ATM
|
|
+user-install-local: user-install-exec-local
|
|
+
|
|
+user-install-exec-local:
|
|
+ ${MAKE} ${_mcall} install;
|
|
+
|
|
+user-clean-local:
|
|
+ ${MAKE} ${_mcall} clean;
|
|
+
|
|
+# Tell versions [3.59,3.63) of GNU make to not export all variables.
|
|
+# Otherwise a system limit (for SysV at least) may be exceeded.
|
|
+.NOEXPORT:
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/Makefile.am
|
|
@@ -0,0 +1 @@
|
|
+include ../../Makefile.extra
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/Makefile.in
|
|
@@ -0,0 +1,389 @@
|
|
+# Makefile.in generated by automake 1.11.1 from Makefile.am.
|
|
+# @configure_input@
|
|
+
|
|
+# Copyright (C) 1994, 1995, 1996, 1997, 1998, 1999, 2000, 2001, 2002,
|
|
+# 2003, 2004, 2005, 2006, 2007, 2008, 2009 Free Software Foundation,
|
|
+# Inc.
|
|
+# This Makefile.in is free software; the Free Software Foundation
|
|
+# gives unlimited permission to copy and/or distribute it,
|
|
+# with or without modifications, as long as this notice is preserved.
|
|
+
|
|
+# This program is distributed in the hope that it will be useful,
|
|
+# but WITHOUT ANY WARRANTY, to the extent permitted by law; without
|
|
+# even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
+# PARTICULAR PURPOSE.
|
|
+
|
|
+@SET_MAKE@
|
|
+
|
|
+# -*- Makefile -*-
|
|
+# AUTOMAKE
|
|
+VPATH = @srcdir@
|
|
+pkgdatadir = $(datadir)/@PACKAGE@
|
|
+pkgincludedir = $(includedir)/@PACKAGE@
|
|
+pkglibdir = $(libdir)/@PACKAGE@
|
|
+pkglibexecdir = $(libexecdir)/@PACKAGE@
|
|
+am__cd = CDPATH="$${ZSH_VERSION+.}$(PATH_SEPARATOR)" && cd
|
|
+install_sh_DATA = $(install_sh) -c -m 644
|
|
+install_sh_PROGRAM = $(install_sh) -c
|
|
+install_sh_SCRIPT = $(install_sh) -c
|
|
+INSTALL_HEADER = $(INSTALL_DATA)
|
|
+transform = $(program_transform_name)
|
|
+NORMAL_INSTALL = :
|
|
+PRE_INSTALL = :
|
|
+POST_INSTALL = :
|
|
+NORMAL_UNINSTALL = :
|
|
+PRE_UNINSTALL = :
|
|
+POST_UNINSTALL = :
|
|
+build_triplet = @build@
|
|
+host_triplet = @host@
|
|
+DIST_COMMON = $(srcdir)/../../Makefile.extra $(srcdir)/Makefile.am \
|
|
+ $(srcdir)/Makefile.in
|
|
+subdir = extensions/LUA
|
|
+ACLOCAL_M4 = $(top_srcdir)/aclocal.m4
|
|
+am__aclocal_m4_deps = $(top_srcdir)/m4/libtool.m4 \
|
|
+ $(top_srcdir)/m4/ltoptions.m4 $(top_srcdir)/m4/ltsugar.m4 \
|
|
+ $(top_srcdir)/m4/ltversion.m4 $(top_srcdir)/m4/lt~obsolete.m4 \
|
|
+ $(top_srcdir)/configure.ac
|
|
+am__configure_deps = $(am__aclocal_m4_deps) $(CONFIGURE_DEPENDENCIES) \
|
|
+ $(ACLOCAL_M4)
|
|
+mkinstalldirs = $(install_sh) -d
|
|
+CONFIG_HEADER = $(top_builddir)/config.h
|
|
+CONFIG_CLEAN_FILES =
|
|
+CONFIG_CLEAN_VPATH_FILES =
|
|
+SOURCES =
|
|
+DIST_SOURCES =
|
|
+DISTFILES = $(DIST_COMMON) $(DIST_SOURCES) $(TEXINFOS) $(EXTRA_DIST)
|
|
+ACLOCAL = @ACLOCAL@
|
|
+AMTAR = @AMTAR@
|
|
+AR = @AR@
|
|
+AUTOCONF = @AUTOCONF@
|
|
+AUTOHEADER = @AUTOHEADER@
|
|
+AUTOMAKE = @AUTOMAKE@
|
|
+AWK = @AWK@
|
|
+CC = @CC@
|
|
+CCDEPMODE = @CCDEPMODE@
|
|
+CFLAGS = @CFLAGS@
|
|
+CPP = @CPP@
|
|
+CPPFLAGS = @CPPFLAGS@
|
|
+CYGPATH_W = @CYGPATH_W@
|
|
+DEFS = @DEFS@
|
|
+DEPDIR = @DEPDIR@
|
|
+DSYMUTIL = @DSYMUTIL@
|
|
+DUMPBIN = @DUMPBIN@
|
|
+ECHO_C = @ECHO_C@
|
|
+ECHO_N = @ECHO_N@
|
|
+ECHO_T = @ECHO_T@
|
|
+EGREP = @EGREP@
|
|
+EXEEXT = @EXEEXT@
|
|
+FGREP = @FGREP@
|
|
+GREP = @GREP@
|
|
+INSTALL = @INSTALL@
|
|
+INSTALL_DATA = @INSTALL_DATA@
|
|
+INSTALL_PROGRAM = @INSTALL_PROGRAM@
|
|
+INSTALL_SCRIPT = @INSTALL_SCRIPT@
|
|
+INSTALL_STRIP_PROGRAM = @INSTALL_STRIP_PROGRAM@
|
|
+LD = @LD@
|
|
+LDFLAGS = @LDFLAGS@
|
|
+LIBOBJS = @LIBOBJS@
|
|
+LIBS = @LIBS@
|
|
+LIBTOOL = @LIBTOOL@
|
|
+LIPO = @LIPO@
|
|
+LN_S = @LN_S@
|
|
+LTLIBOBJS = @LTLIBOBJS@
|
|
+MAKEINFO = @MAKEINFO@
|
|
+MKDIR_P = @MKDIR_P@
|
|
+NM = @NM@
|
|
+NMEDIT = @NMEDIT@
|
|
+OBJDUMP = @OBJDUMP@
|
|
+OBJEXT = @OBJEXT@
|
|
+OTOOL = @OTOOL@
|
|
+OTOOL64 = @OTOOL64@
|
|
+PACKAGE = @PACKAGE@
|
|
+PACKAGE_BUGREPORT = @PACKAGE_BUGREPORT@
|
|
+PACKAGE_NAME = @PACKAGE_NAME@
|
|
+PACKAGE_STRING = @PACKAGE_STRING@
|
|
+PACKAGE_TARNAME = @PACKAGE_TARNAME@
|
|
+PACKAGE_URL = @PACKAGE_URL@
|
|
+PACKAGE_VERSION = @PACKAGE_VERSION@
|
|
+PATH_SEPARATOR = @PATH_SEPARATOR@
|
|
+PKG_CONFIG = @PKG_CONFIG@
|
|
+RANLIB = @RANLIB@
|
|
+SED = @SED@
|
|
+SET_MAKE = @SET_MAKE@
|
|
+SHELL = @SHELL@
|
|
+STRIP = @STRIP@
|
|
+VERSION = @VERSION@
|
|
+abs_builddir = @abs_builddir@
|
|
+abs_srcdir = @abs_srcdir@
|
|
+abs_top_builddir = @abs_top_builddir@
|
|
+abs_top_srcdir = @abs_top_srcdir@
|
|
+ac_ct_CC = @ac_ct_CC@
|
|
+ac_ct_DUMPBIN = @ac_ct_DUMPBIN@
|
|
+am__include = @am__include@
|
|
+am__leading_dot = @am__leading_dot@
|
|
+am__quote = @am__quote@
|
|
+am__tar = @am__tar@
|
|
+am__untar = @am__untar@
|
|
+bindir = @bindir@
|
|
+build = @build@
|
|
+build_alias = @build_alias@
|
|
+build_cpu = @build_cpu@
|
|
+build_os = @build_os@
|
|
+build_vendor = @build_vendor@
|
|
+builddir = @builddir@
|
|
+datadir = @datadir@
|
|
+datarootdir = @datarootdir@
|
|
+docdir = @docdir@
|
|
+dvidir = @dvidir@
|
|
+exec_prefix = @exec_prefix@
|
|
+host = @host@
|
|
+host_alias = @host_alias@
|
|
+host_cpu = @host_cpu@
|
|
+host_os = @host_os@
|
|
+host_vendor = @host_vendor@
|
|
+htmldir = @htmldir@
|
|
+includedir = @includedir@
|
|
+infodir = @infodir@
|
|
+install_sh = @install_sh@
|
|
+kbuilddir = @kbuilddir@
|
|
+kinclude_CFLAGS = @kinclude_CFLAGS@
|
|
+ksourcedir = @ksourcedir@
|
|
+libdir = @libdir@
|
|
+libexecdir = @libexecdir@
|
|
+libxtables_CFLAGS = @libxtables_CFLAGS@
|
|
+libxtables_LIBS = @libxtables_LIBS@
|
|
+localedir = @localedir@
|
|
+localstatedir = @localstatedir@
|
|
+lt_ECHO = @lt_ECHO@
|
|
+mandir = @mandir@
|
|
+mkdir_p = @mkdir_p@
|
|
+oldincludedir = @oldincludedir@
|
|
+pdfdir = @pdfdir@
|
|
+prefix = @prefix@
|
|
+program_transform_name = @program_transform_name@
|
|
+psdir = @psdir@
|
|
+regular_CFLAGS = @regular_CFLAGS@
|
|
+sbindir = @sbindir@
|
|
+sharedstatedir = @sharedstatedir@
|
|
+srcdir = @srcdir@
|
|
+sysconfdir = @sysconfdir@
|
|
+target_alias = @target_alias@
|
|
+top_build_prefix = @top_build_prefix@
|
|
+top_builddir = @top_builddir@
|
|
+top_srcdir = @top_srcdir@
|
|
+xtlibdir = @xtlibdir@
|
|
+XA_SRCDIR = ${srcdir}
|
|
+XA_TOPSRCDIR = ${top_srcdir}
|
|
+XA_ABSTOPSRCDIR = ${abs_top_srcdir}
|
|
+_mcall = -f ${top_builddir}/Makefile.iptrules
|
|
+all: all-am
|
|
+
|
|
+.SUFFIXES:
|
|
+$(srcdir)/Makefile.in: $(srcdir)/Makefile.am $(srcdir)/../../Makefile.extra $(am__configure_deps)
|
|
+ @for dep in $?; do \
|
|
+ case '$(am__configure_deps)' in \
|
|
+ *$$dep*) \
|
|
+ ( cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh ) \
|
|
+ && { if test -f $@; then exit 0; else break; fi; }; \
|
|
+ exit 1;; \
|
|
+ esac; \
|
|
+ done; \
|
|
+ echo ' cd $(top_srcdir) && $(AUTOMAKE) --foreign extensions/LUA/Makefile'; \
|
|
+ $(am__cd) $(top_srcdir) && \
|
|
+ $(AUTOMAKE) --foreign extensions/LUA/Makefile
|
|
+.PRECIOUS: Makefile
|
|
+Makefile: $(srcdir)/Makefile.in $(top_builddir)/config.status
|
|
+ @case '$?' in \
|
|
+ *config.status*) \
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh;; \
|
|
+ *) \
|
|
+ echo ' cd $(top_builddir) && $(SHELL) ./config.status $(subdir)/$@ $(am__depfiles_maybe)'; \
|
|
+ cd $(top_builddir) && $(SHELL) ./config.status $(subdir)/$@ $(am__depfiles_maybe);; \
|
|
+ esac;
|
|
+
|
|
+$(top_builddir)/config.status: $(top_srcdir)/configure $(CONFIG_STATUS_DEPENDENCIES)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+
|
|
+$(top_srcdir)/configure: $(am__configure_deps)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+$(ACLOCAL_M4): $(am__aclocal_m4_deps)
|
|
+ cd $(top_builddir) && $(MAKE) $(AM_MAKEFLAGS) am--refresh
|
|
+$(am__aclocal_m4_deps):
|
|
+
|
|
+mostlyclean-libtool:
|
|
+ -rm -f *.lo
|
|
+
|
|
+clean-libtool:
|
|
+ -rm -rf .libs _libs
|
|
+tags: TAGS
|
|
+TAGS:
|
|
+
|
|
+ctags: CTAGS
|
|
+CTAGS:
|
|
+
|
|
+
|
|
+distdir: $(DISTFILES)
|
|
+ @srcdirstrip=`echo "$(srcdir)" | sed 's/[].[^$$\\*]/\\\\&/g'`; \
|
|
+ topsrcdirstrip=`echo "$(top_srcdir)" | sed 's/[].[^$$\\*]/\\\\&/g'`; \
|
|
+ list='$(DISTFILES)'; \
|
|
+ dist_files=`for file in $$list; do echo $$file; done | \
|
|
+ sed -e "s|^$$srcdirstrip/||;t" \
|
|
+ -e "s|^$$topsrcdirstrip/|$(top_builddir)/|;t"`; \
|
|
+ case $$dist_files in \
|
|
+ */*) $(MKDIR_P) `echo "$$dist_files" | \
|
|
+ sed '/\//!d;s|^|$(distdir)/|;s,/[^/]*$$,,' | \
|
|
+ sort -u` ;; \
|
|
+ esac; \
|
|
+ for file in $$dist_files; do \
|
|
+ if test -f $$file || test -d $$file; then d=.; else d=$(srcdir); fi; \
|
|
+ if test -d $$d/$$file; then \
|
|
+ dir=`echo "/$$file" | sed -e 's,/[^/]*$$,,'`; \
|
|
+ if test -d "$(distdir)/$$file"; then \
|
|
+ find "$(distdir)/$$file" -type d ! -perm -700 -exec chmod u+rwx {} \;; \
|
|
+ fi; \
|
|
+ if test -d $(srcdir)/$$file && test $$d != $(srcdir); then \
|
|
+ cp -fpR $(srcdir)/$$file "$(distdir)$$dir" || exit 1; \
|
|
+ find "$(distdir)/$$file" -type d ! -perm -700 -exec chmod u+rwx {} \;; \
|
|
+ fi; \
|
|
+ cp -fpR $$d/$$file "$(distdir)$$dir" || exit 1; \
|
|
+ else \
|
|
+ test -f "$(distdir)/$$file" \
|
|
+ || cp -p $$d/$$file "$(distdir)/$$file" \
|
|
+ || exit 1; \
|
|
+ fi; \
|
|
+ done
|
|
+check-am: all-am
|
|
+check: check-am
|
|
+all-am: Makefile all-local
|
|
+installdirs:
|
|
+install: install-am
|
|
+install-exec: install-exec-am
|
|
+install-data: install-data-am
|
|
+uninstall: uninstall-am
|
|
+
|
|
+install-am: all-am
|
|
+ @$(MAKE) $(AM_MAKEFLAGS) install-exec-am install-data-am
|
|
+
|
|
+installcheck: installcheck-am
|
|
+install-strip:
|
|
+ $(MAKE) $(AM_MAKEFLAGS) INSTALL_PROGRAM="$(INSTALL_STRIP_PROGRAM)" \
|
|
+ install_sh_PROGRAM="$(INSTALL_STRIP_PROGRAM)" INSTALL_STRIP_FLAG=-s \
|
|
+ `test -z '$(STRIP)' || \
|
|
+ echo "INSTALL_PROGRAM_ENV=STRIPPROG='$(STRIP)'"` install
|
|
+mostlyclean-generic:
|
|
+
|
|
+clean-generic:
|
|
+
|
|
+distclean-generic:
|
|
+ -test -z "$(CONFIG_CLEAN_FILES)" || rm -f $(CONFIG_CLEAN_FILES)
|
|
+ -test . = "$(srcdir)" || test -z "$(CONFIG_CLEAN_VPATH_FILES)" || rm -f $(CONFIG_CLEAN_VPATH_FILES)
|
|
+
|
|
+maintainer-clean-generic:
|
|
+ @echo "This command is intended for maintainers to use"
|
|
+ @echo "it deletes files that may require special tools to rebuild."
|
|
+clean: clean-am
|
|
+
|
|
+clean-am: clean-generic clean-libtool clean-local mostlyclean-am
|
|
+
|
|
+distclean: distclean-am
|
|
+ -rm -f Makefile
|
|
+distclean-am: clean-am distclean-generic
|
|
+
|
|
+dvi: dvi-am
|
|
+
|
|
+dvi-am:
|
|
+
|
|
+html: html-am
|
|
+
|
|
+html-am:
|
|
+
|
|
+info: info-am
|
|
+
|
|
+info-am:
|
|
+
|
|
+install-data-am:
|
|
+
|
|
+install-dvi: install-dvi-am
|
|
+
|
|
+install-dvi-am:
|
|
+
|
|
+install-exec-am: install-exec-local
|
|
+
|
|
+install-html: install-html-am
|
|
+
|
|
+install-html-am:
|
|
+
|
|
+install-info: install-info-am
|
|
+
|
|
+install-info-am:
|
|
+
|
|
+install-man:
|
|
+
|
|
+install-pdf: install-pdf-am
|
|
+
|
|
+install-pdf-am:
|
|
+
|
|
+install-ps: install-ps-am
|
|
+
|
|
+install-ps-am:
|
|
+
|
|
+installcheck-am:
|
|
+
|
|
+maintainer-clean: maintainer-clean-am
|
|
+ -rm -f Makefile
|
|
+maintainer-clean-am: distclean-am maintainer-clean-generic
|
|
+
|
|
+mostlyclean: mostlyclean-am
|
|
+
|
|
+mostlyclean-am: mostlyclean-generic mostlyclean-libtool
|
|
+
|
|
+pdf: pdf-am
|
|
+
|
|
+pdf-am:
|
|
+
|
|
+ps: ps-am
|
|
+
|
|
+ps-am:
|
|
+
|
|
+uninstall-am:
|
|
+
|
|
+.MAKE: install-am install-strip
|
|
+
|
|
+.PHONY: all all-am all-local check check-am clean clean-generic \
|
|
+ clean-libtool clean-local distclean distclean-generic \
|
|
+ distclean-libtool distdir dvi dvi-am html html-am info info-am \
|
|
+ install install-am install-data install-data-am install-dvi \
|
|
+ install-dvi-am install-exec install-exec-am install-exec-local \
|
|
+ install-html install-html-am install-info install-info-am \
|
|
+ install-man install-pdf install-pdf-am install-ps \
|
|
+ install-ps-am install-strip installcheck installcheck-am \
|
|
+ installdirs maintainer-clean maintainer-clean-generic \
|
|
+ mostlyclean mostlyclean-generic mostlyclean-libtool pdf pdf-am \
|
|
+ ps ps-am uninstall uninstall-am
|
|
+
|
|
+export XA_SRCDIR
|
|
+export XA_TOPSRCDIR
|
|
+export XA_ABSTOPSRCDIR
|
|
+
|
|
+all-local: user-all-local
|
|
+
|
|
+install-exec-local: user-install-local
|
|
+
|
|
+clean-local: user-clean-local
|
|
+
|
|
+user-all-local:
|
|
+ ${MAKE} ${_mcall} all;
|
|
+
|
|
+# Have no user-install-data-local ATM
|
|
+user-install-local: user-install-exec-local
|
|
+
|
|
+user-install-exec-local:
|
|
+ ${MAKE} ${_mcall} install;
|
|
+
|
|
+user-clean-local:
|
|
+ ${MAKE} ${_mcall} clean;
|
|
+
|
|
+# Tell versions [3.59,3.63) of GNU make to not export all variables.
|
|
+# Otherwise a system limit (for SysV at least) may be exceeded.
|
|
+.NOEXPORT:
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/Mbuild
|
|
@@ -0,0 +1,3 @@
|
|
+# -*- Makefile -*-
|
|
+
|
|
+obj-${build_LUA} += libxt_LUA.so
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/nf_lua.c
|
|
@@ -0,0 +1,64 @@
|
|
+#if defined(__KERNEL__)
|
|
+
|
|
+#include <linux/kernel.h>
|
|
+#include <linux/slab.h>
|
|
+#include <linux/string.h>
|
|
+#include <linux/timer.h>
|
|
+#include <linux/random.h>
|
|
+#include <linux/netfilter/x_tables.h>
|
|
+
|
|
+#endif
|
|
+
|
|
+#include "lua.h"
|
|
+#include "lobject.h" /*sizeof(udata) */
|
|
+#include "lauxlib.h"
|
|
+#include "controller.h"
|
|
+
|
|
+#if defined(__KERNEL__) /* reachs until luaopen_nflib */
|
|
+
|
|
+
|
|
+static int32_t nf_get_random(lua_State *L)
|
|
+{
|
|
+ uint32_t rand = 0;
|
|
+
|
|
+ get_random_bytes(&rand, sizeof(uint32_t ));
|
|
+ lua_pushnumber(L, rand);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int32_t nf_get_time(lua_State *L)
|
|
+{
|
|
+ lua_pushnumber(L, jiffies_to_msecs(jiffies_64));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static const struct luaL_Reg nf_lua_lib_f [] = {
|
|
+ { "get_random", nf_get_random },
|
|
+ { "get_time", nf_get_time },
|
|
+ { NULL, NULL }
|
|
+};
|
|
+
|
|
+void luaopen_nflib(lua_State *L)
|
|
+{
|
|
+ int32_t top;
|
|
+
|
|
+ luaL_register(L, NETFILTER_LIB, nf_lua_lib_f);
|
|
+ lua_pop(L, 1);
|
|
+
|
|
+ /* registering verdicts inside the _G */
|
|
+ lua_getglobal(L, "_G");
|
|
+ top = lua_gettop(L);
|
|
+
|
|
+ lua_pushinteger(L, XT_CONTINUE);
|
|
+ lua_setfield(L, top, "XT_CONTINUE"); /* continiue with next rule */
|
|
+
|
|
+ lua_pushinteger(L, NF_DROP);
|
|
+ lua_setfield(L, top, "NF_DROP"); /* stop traversal in the current table hook and drop packet */
|
|
+
|
|
+ lua_pushinteger(L, NF_ACCEPT);
|
|
+ lua_setfield(L, top, "NF_ACCEPT"); /* stop traversal in the current table hook and accept packet */
|
|
+
|
|
+ lua_pop(L, 1); /* pop _G */
|
|
+}
|
|
+
|
|
+#endif
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_dynamic.c
|
|
@@ -0,0 +1,486 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+
|
|
+
|
|
+struct protocol_buf * dyn_prot_buf_array[MAX_NR_OF_DYN_PROT_BUFS] = { NULL };
|
|
+
|
|
+
|
|
+/* LUA_API: the function 'field_dynamic_setter' acts as a wrapper around
|
|
+ * a given Lua field setter function of a dynamic protocol buffer. The
|
|
+ * string containing the lua function name was piggybacked in the 'set'
|
|
+ * member of the protocol_field. We call this function passing the actual
|
|
+ * segment as byte array and the set value.
|
|
+ *
|
|
+ * Paramters:
|
|
+ * 1. lua_packet_segment (implicit)
|
|
+ * 2. some lua value
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. pointer to the protocol buffer
|
|
+ * 2. field index
|
|
+ *
|
|
+ * Returns:
|
|
+ * 1. true or false if the 'set' was successful
|
|
+ */
|
|
+int32_t field_dynamic_setter(lua_State *L)
|
|
+{
|
|
+ size_t nbytes;
|
|
+ lua_packet_segment * array;
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ int32_t field_index = lua_tointeger(L, lua_upvalueindex(2));
|
|
+
|
|
+ /* the function name is piggybacked as a string */
|
|
+ lua_getglobal(L, (char *)prot_buf->protocol_fields[field_index].set);
|
|
+ if (!lua_isfunction(L, -1)) {
|
|
+ lua_pushboolean(L, 0);
|
|
+ return 1;
|
|
+ }
|
|
+
|
|
+ nbytes = sizeof(lua_packet_segment) + seg->length * sizeof(uint8_t);
|
|
+ array = (lua_packet_segment *)lua_newuserdata(L, nbytes);
|
|
+ array->length = seg->length;
|
|
+ array->start = seg->start + seg->offset;
|
|
+ array->changes = NULL;
|
|
+
|
|
+ luaL_getmetatable(L, LUA_BYTE_ARRAY);
|
|
+ lua_setmetatable(L, -2);
|
|
+ lua_pushvalue(L, 2); /* push value to set */
|
|
+ if (lua_pcall(L, 2, 1, 0) != 0) {
|
|
+ pr_debug("Error: %s \n", lua_tostring(L, -1));
|
|
+ lua_pop(L, 1);
|
|
+ lua_pushboolean(L, 0);
|
|
+ }
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'field_dynamic_getter' acts as a wrapper around
|
|
+ * a given Lua field getter function of a dynamic protocol buffer. The
|
|
+ * string containing the lua function name was piggybacked in the 'get'
|
|
+ * member of the protocol_field. We call this function passing the actual
|
|
+ * segment as byte array.
|
|
+ *
|
|
+ * Paramters:
|
|
+ * 1. lua_packet_segment (implicit)
|
|
+ *
|
|
+ * Upvalues:
|
|
+ * 1. pointer to the protocol buffer
|
|
+ * 2. field index
|
|
+ *
|
|
+ * Returns:
|
|
+ * 1. true or false if the 'get' was successful
|
|
+ */
|
|
+int32_t field_dynamic_getter(lua_State *L)
|
|
+{
|
|
+ size_t nbytes;
|
|
+ lua_packet_segment * array;
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ int32_t field_index = lua_tointeger(L, lua_upvalueindex(2));
|
|
+
|
|
+ /* the function name is piggybacked as a string */
|
|
+ lua_getglobal(L, (char *)prot_buf->protocol_fields[field_index].get);
|
|
+ if (!lua_isfunction(L, -1)) {
|
|
+ lua_pushboolean(L, 0);
|
|
+ return 1;
|
|
+ }
|
|
+
|
|
+ nbytes = sizeof(lua_packet_segment) + seg->length * sizeof(uint8_t);
|
|
+ array = (lua_packet_segment *)lua_newuserdata(L, nbytes);
|
|
+ array->length = seg->length;
|
|
+ array->start = seg->start + seg->offset;
|
|
+ array->changes = NULL;
|
|
+
|
|
+ luaL_getmetatable(L, LUA_BYTE_ARRAY);
|
|
+ lua_setmetatable(L, -2);
|
|
+ if (lua_pcall(L, 1, 1, 0) != 0) {
|
|
+ pr_debug("Error: %s \n", luaL_checkstring(L, -1));
|
|
+ lua_pop(L, 1);
|
|
+ lua_pushboolean(L, 0);
|
|
+ }
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'has_protocol_dynamic' acts as a wrapper around
|
|
+ * a given lua has_protocol function of a dynamic protocol buffer. The
|
|
+ * string containing the lua function name was piggybacked in the 'has_protocol'
|
|
+ * member of the protocol_buffer. We call this function passing the actual
|
|
+ * segment.
|
|
+ *
|
|
+ * Paramters:
|
|
+ * 1. lua_packet_segment
|
|
+ * 2. protocol type
|
|
+ *
|
|
+ * Returns:
|
|
+ * 1. true or false if the payload field contains the given protocol
|
|
+ */
|
|
+int32_t has_protocol_dynamic(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t type)
|
|
+{
|
|
+ lua_packet_segment *seg_new;
|
|
+ int32_t res = 0;
|
|
+
|
|
+ /* the function name is piggybacked as a string */
|
|
+ lua_getglobal(L, (char *)prot_buf->has_protocol);
|
|
+ seg_new = (lua_packet_segment *)lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+ seg_new->start = seg->start;
|
|
+ seg_new->offset = seg->offset;
|
|
+ seg_new->length = seg->length;
|
|
+ seg_new->changes = NULL;
|
|
+ luaL_getmetatable(L, prot_buf->name);
|
|
+ lua_setmetatable(L, -2);
|
|
+ lua_pushinteger(L, type); /* push the protocol type */
|
|
+ if (lua_pcall(L, 2, 1, 0) != 0) {
|
|
+ pr_debug("Error: %s \n", luaL_checkstring(L, -1));
|
|
+ lua_pop(L, 1);
|
|
+ return 0;
|
|
+ }
|
|
+ res = lua_toboolean(L, -1);
|
|
+ lua_pop(L, 1);
|
|
+
|
|
+ return res;
|
|
+}
|
|
+
|
|
+/* LUA_API: the function 'get_field_changes_dynamic' acts as a wrapper around
|
|
+ * a given lua get_field_changes function of a dynamic protocol buffer. The
|
|
+ * string containing the lua function name was piggybacked in the 'get_field_changes'
|
|
+ * member of the protocol_buffer. We call this function passing the actual
|
|
+ * segment. The lua function must return two lua table containing the offset
|
|
+ * and length changes (in bits).
|
|
+ *
|
|
+ * Paramters:
|
|
+ * 1. lua_packet_segment
|
|
+ *
|
|
+ * Returns:
|
|
+ * 1. new allocated field_changes struct
|
|
+ */
|
|
+struct field_changes * get_field_changes_dynamic(lua_State *L, struct protocol_buf *prot_buf, lua_packet_segment * seg)
|
|
+{
|
|
+ lua_packet_segment *seg_new;
|
|
+ struct field_changes * changes;
|
|
+ int32_t nr_of_changes, i;
|
|
+
|
|
+ lua_getglobal(L, (char *)prot_buf->get_field_changes);
|
|
+
|
|
+ seg_new = (lua_packet_segment *)lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+ seg_new->start = seg->start;
|
|
+ seg_new->offset = seg->offset;
|
|
+ seg_new->length = seg->length;
|
|
+ seg_new->changes = NULL;
|
|
+ luaL_getmetatable(L, prot_buf->name);
|
|
+ lua_setmetatable(L, -2);
|
|
+
|
|
+ if (lua_pcall(L, 1, 2, 0) != 0)
|
|
+ luaL_error(L, "inside get_field_changes_dynamic. %s\n", lua_tostring(L, -1));
|
|
+
|
|
+ /* the function call must return a table containing length changes */
|
|
+ luaL_checktype(L, -1, LUA_TTABLE);
|
|
+ /* the function call must return a table containing offset changes */
|
|
+ luaL_checktype(L, -2, LUA_TTABLE);
|
|
+ /* both tables have to be of same size */
|
|
+ if (lua_objlen(L, -1) != lua_objlen(L, -2))
|
|
+ luaL_error(L, "the provided tables are not of equal size");
|
|
+
|
|
+ nr_of_changes = lua_objlen(L, -1);
|
|
+ changes = get_allocated_field_changes(L, nr_of_changes);
|
|
+
|
|
+ /* loop over the tables */
|
|
+ for (i = 1; i < nr_of_changes; i++) {
|
|
+ lua_rawgeti(L, -1, i); /* push length value of field at index i */
|
|
+ changes->field_length_changes[i - 1] = luaL_checkinteger(L, -1);
|
|
+ lua_pop(L, 1); /* pop offset value */
|
|
+
|
|
+ lua_rawgeti(L, -2, i); /* push offset value of field at index i */
|
|
+ changes->field_offset_changes[i - 1] = luaL_checkinteger(L, -1);
|
|
+ lua_pop(L, 1); /* pop length value */
|
|
+ }
|
|
+
|
|
+ /* pop both tables */
|
|
+ lua_pop(L, 2);
|
|
+
|
|
+ return changes;
|
|
+}
|
|
+
|
|
+/* C_INT: 'get_free_protocol_index' is only used internally. This function
|
|
+ * gets a free slot inside the array holding all the protocol buffers.
|
|
+ * There are several ways to get to this information. In this case I take
|
|
+ * the way over the reflected array SUPPORTED_PROTOCOL_TABLE inside the
|
|
+ * Lua state. Since this function is called at laodtime, we do not have
|
|
+ * to care about performance.
|
|
+ */
|
|
+static int32_t get_free_protocol_index(lua_State *L)
|
|
+{
|
|
+ int32_t protocol_index;
|
|
+
|
|
+ lua_getglobal(L, SUPPORTED_PROTOCOL_TABLE);
|
|
+ protocol_index = lua_objlen(L, -1) + 1;
|
|
+ lua_pop(L, 1);
|
|
+ return protocol_index;
|
|
+}
|
|
+
|
|
+/* C_API: 'free_dynamic_prot_buf' frees the allocated memory of a given
|
|
+ * dynamic protocol buffer. this function is normally called inside a
|
|
+ * cleanup routine. Be aware, before running this function you must be
|
|
+ * sure that no references to the dynamic protocol buffers were available.
|
|
+ * It's recomended to close the Lua state before calling the function. */
|
|
+void free_dynamic_prot_buf(struct protocol_buf * prot_buf)
|
|
+{
|
|
+ struct protocol_field * field = prot_buf->protocol_fields;
|
|
+
|
|
+ for (; field->name != NULL; field++) {
|
|
+ if (field->get) kfree(field->get);
|
|
+ if (field->set) kfree(field->set);
|
|
+ if (field->name) kfree((char *)field->name);
|
|
+ }
|
|
+
|
|
+ if (prot_buf->payload_field) kfree(prot_buf->payload_field);
|
|
+ if (prot_buf->has_protocol) kfree(prot_buf->has_protocol);
|
|
+
|
|
+ if (prot_buf->get_field_changes) kfree(prot_buf->get_field_changes);
|
|
+ kfree((char *)prot_buf->name);
|
|
+ kfree(prot_buf);
|
|
+ return;
|
|
+}
|
|
+
|
|
+void cleanup_dynamic_prot_bufs(void)
|
|
+{
|
|
+ int32_t i;
|
|
+
|
|
+ for (i = 0; i < MAX_NR_OF_DYN_PROT_BUFS; i++) {
|
|
+ if (dyn_prot_buf_array[i]) {
|
|
+ free_dynamic_prot_buf(dyn_prot_buf_array[i]);
|
|
+ dyn_prot_buf_array[i] = NULL;
|
|
+ }
|
|
+ }
|
|
+ return;
|
|
+}
|
|
+
|
|
+
|
|
+/* C_INT: 'free_protocol_fields' is used internally as a helper function for
|
|
+ * 'register_dynamic_protbuf'. It is used when durin registration an error
|
|
+ * occurs and the afore allocated fields needed to be freed. */
|
|
+static inline void free_protocol_fields(struct protocol_field * prot_fields, int32_t i)
|
|
+{
|
|
+ struct protocol_field * f;
|
|
+
|
|
+ while (i >= 0) {
|
|
+ f = &prot_fields[i];
|
|
+ if (f->name) kfree((void *)f->name);
|
|
+ if (f->get) kfree((void *)f->get);
|
|
+ if (f->set) kfree((void *)f->set);
|
|
+ kfree((void *)f);
|
|
+ i--;
|
|
+ }
|
|
+}
|
|
+
|
|
+/* LUA_API: 'register_dynamic_protbuf' is called from within the Lua script.
|
|
+ * it takes a Lua table representing the dynamic protocol buffer as parameter.
|
|
+ * e.g.:
|
|
+ * eth_prot_buf = {
|
|
+ * name = "packet_eth_dyn",
|
|
+ * payload_field = "data",
|
|
+ * protocol_fields = {
|
|
+ * {"dmac", 0, 48, nil, nil },
|
|
+ * {"smac", 48, 48, nil, nil },
|
|
+ * {"type", 96, 16, nil, nil },
|
|
+ * {"data", 112, 0, nil, nil },
|
|
+ * },
|
|
+ * has_protocol = "eth_dyn_has_protocol",
|
|
+ * get_field_changes = "eth_dyn_get_field_changes"
|
|
+ * }
|
|
+ * register_dynamic_protbuf(eth_prot_buf)
|
|
+ *
|
|
+ * the table gets parsed and a new protocol_buf struct is allocated and
|
|
+ * initialized using 'register_protbuf', which is also used for the static
|
|
+ * protocol buffers. This enables an identical behavior like the static
|
|
+ * protocol buffers. The dynamic protocol buffers are not garbage collected,
|
|
+ * use 'free_dynamic_protbuf' to free them after closing the Lua state.
|
|
+ */
|
|
+static int32_t register_dynamic_protbuf(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf *prot_buf;
|
|
+ struct protocol_field *field, sentinel = PROT_FIELD_SENTINEL;
|
|
+ int32_t nr_of_fields, i;
|
|
+
|
|
+ prot_buf = (struct protocol_buf *)kmalloc(sizeof(struct protocol_buf), GFP_KERNEL);
|
|
+ prot_buf->is_dynamic = 1;
|
|
+
|
|
+ /* check if parameter is a table */
|
|
+ luaL_checktype(L, 1, LUA_TTABLE);
|
|
+
|
|
+ /* initialize prot_buf.name */
|
|
+ lua_getfield(L, 1, "name");
|
|
+ prot_buf->name = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char *)prot_buf->name, luaL_checkstring(L, -1));
|
|
+ lua_pop(L, 1); /* pop res from lua_getfield */
|
|
+
|
|
+ /* check if protocol buffer is already registered */
|
|
+ lua_getglobal(L, prot_buf->name);
|
|
+ if (!lua_isnil(L, -1)) {
|
|
+ lua_pop(L, 1); /* pop res from lua_getglobal */
|
|
+ pr_debug("protocol_buf '%s' already registered.\n", prot_buf->name);
|
|
+ goto free_prot_buf;
|
|
+ }
|
|
+ lua_pop(L, 1); /* pop res from lua_getglobal */
|
|
+
|
|
+ /* initialize payload field */
|
|
+ lua_getfield(L, 1, "payload_field");
|
|
+ if (lua_isstring(L, -1)) {
|
|
+ prot_buf->payload_field = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy(prot_buf->payload_field, lua_tostring(L, -1));
|
|
+ }else
|
|
+ prot_buf->payload_field = NULL;
|
|
+ lua_pop(L, 1); /* pop res from lua_getfield */
|
|
+
|
|
+ /* initialize protocol_fields field*/
|
|
+ lua_getfield(L, 1, "protocol_fields");
|
|
+ if (!lua_istable(L, -1)) {
|
|
+ pr_debug("invalid protocol_fields table.\n");
|
|
+ goto err2;
|
|
+
|
|
+ }
|
|
+
|
|
+ nr_of_fields = lua_objlen(L, -1);
|
|
+ prot_buf->protocol_fields = (struct protocol_field *)kmalloc((nr_of_fields + 1) * sizeof(struct protocol_field), GFP_KERNEL);
|
|
+
|
|
+ for (i = 1; i <= nr_of_fields; i++) {
|
|
+ field = &prot_buf->protocol_fields[i - 1];
|
|
+ /* initialize protocol field */
|
|
+ lua_rawgeti(L, -1, i); /* push field-table */
|
|
+ if (!lua_istable(L, -1)) {
|
|
+ free_protocol_fields(prot_buf->protocol_fields, i);
|
|
+ pr_debug("invalid protocol_field at %i.\n", i);
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
+ /* initialize protocol field name */
|
|
+ lua_rawgeti(L, -1, 1);
|
|
+ if (!lua_isstring(L, -1)) {
|
|
+ free_protocol_fields(prot_buf->protocol_fields, i);
|
|
+ pr_debug("invalid protocol_field name at %i.\n", i);
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
+ field->name = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char*)field->name, lua_tostring(L, -1));
|
|
+ lua_pop(L, 1); /* pop field name */
|
|
+
|
|
+ /* initialize protocol field offset */
|
|
+ lua_rawgeti(L, -1, 2);
|
|
+ if (!lua_isnumber(L, -1)) {
|
|
+ free_protocol_fields(prot_buf->protocol_fields, i);
|
|
+ pr_debug("invalid protocol_field offset at %i.\n", i);
|
|
+ goto err;
|
|
+ }
|
|
+ field->offset = lua_tointeger(L, -1);
|
|
+ lua_pop(L, 1); /* pop field offset */
|
|
+
|
|
+ /* initialize protocol field length */
|
|
+ lua_rawgeti(L, -1, 3);
|
|
+ if (!lua_isnumber(L, -1)) {
|
|
+ free_protocol_fields(prot_buf->protocol_fields, i);
|
|
+ pr_debug("invalid protocol_field length at %i.\n", i);
|
|
+ goto err;
|
|
+ }
|
|
+ field->length = lua_tointeger(L, -1);
|
|
+ lua_pop(L, 1); /* pop field length */
|
|
+
|
|
+ /* initialize protocol field getter */
|
|
+ lua_rawgeti(L, -1, 4);
|
|
+ if (lua_isstring(L, -1)) {
|
|
+ field->get = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char *)field->get, lua_tostring(L, -1)); /* the get-wrapper knows about the piggybacked string */
|
|
+ }else
|
|
+ field->get = NULL;
|
|
+ lua_pop(L, 1); /* pop field getter */
|
|
+
|
|
+ /* initialize protocol field setter */
|
|
+ lua_rawgeti(L, -1, 5);
|
|
+ if (lua_isstring(L, -1)) {
|
|
+ field->set = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char *)field->set, lua_tostring(L, -1)); /* the set-wrapper knows about the piggybacked string */
|
|
+ }else
|
|
+ field->set = NULL;
|
|
+ lua_pop(L, 1); /* pop field setter */
|
|
+
|
|
+ /* field initialization completed */
|
|
+ lua_pop(L, 1); /* pop field-table */
|
|
+ }
|
|
+
|
|
+ /* put sentinel at the end of protocol_fields */
|
|
+ memcpy(&prot_buf->protocol_fields[nr_of_fields], &sentinel, sizeof(sentinel));
|
|
+ lua_pop(L, 1); /* pop protocol-fields-table */
|
|
+
|
|
+ /* initialize has_protocol field */
|
|
+ lua_getfield(L, 1, "has_protocol");
|
|
+ if (lua_isstring(L, -1)) {
|
|
+ prot_buf->has_protocol = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char *)prot_buf->has_protocol, lua_tostring(L, -1)); /* the has_protocol-wrapper knows about the piggybacked string */
|
|
+ }else
|
|
+ prot_buf->has_protocol = NULL;
|
|
+ lua_pop(L, 1); /* pop has_protocol */
|
|
+
|
|
+ /* initialize get_field_changes field */
|
|
+ lua_getfield(L, 1, "get_field_changes");
|
|
+ if (lua_isstring(L, -1)) {
|
|
+ prot_buf->get_field_changes = kmalloc(lua_objlen(L, -1), GFP_KERNEL);
|
|
+ strcpy((char *)prot_buf->get_field_changes, lua_tostring(L, -1)); /* the get_field_changes-wrapper knows about the piggybacked string */
|
|
+ }else
|
|
+ prot_buf->get_field_changes = NULL;
|
|
+ lua_pop(L, 1); /* pop get_field_changes */
|
|
+
|
|
+ /* Storing the pointer to the DYNAMIC protbuf within dyn_prot_buf_array, in order to free it at cleanup */
|
|
+ for (i = 0; i < MAX_NR_OF_DYN_PROT_BUFS; i++) {
|
|
+ if (!dyn_prot_buf_array[i]) {
|
|
+ dyn_prot_buf_array[i] = prot_buf;
|
|
+ break;
|
|
+ }else
|
|
+ goto err;
|
|
+ }
|
|
+
|
|
+ /* call the "common" register_protbuf */
|
|
+ register_protbuf(L, prot_buf, get_free_protocol_index(L)); /* register prot_buf as it is done with the static ones */
|
|
+
|
|
+ return 0;
|
|
+
|
|
+err:
|
|
+ kfree(prot_buf->protocol_fields);
|
|
+err2:
|
|
+ if (prot_buf->payload_field) kfree(prot_buf->payload_field);
|
|
+free_prot_buf:
|
|
+ kfree((void *)prot_buf->name);
|
|
+ kfree(prot_buf);
|
|
+
|
|
+ luaL_error(L, "one or more error happend while registering a dynamic protocol buffer, please consult the debug log");
|
|
+
|
|
+ return 0;
|
|
+
|
|
+}
|
|
+
|
|
+void luaopen_protbuf_dynamic(lua_State *L)
|
|
+{
|
|
+ lua_getglobal(L, "_G");
|
|
+ lua_pushcclosure(L, register_dynamic_protbuf, 0);
|
|
+ lua_setfield(L, -2, "register_dynamic_protbuf");
|
|
+ lua_pop(L, 1); /* pop _G */
|
|
+ return;
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_ethernet.c
|
|
@@ -0,0 +1,60 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+
|
|
+static int32_t eth_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ uint8_t *embedded_protocol = seg->start + seg->offset + 12 /*bytes*/;
|
|
+ unsigned short res = (unsigned short)((embedded_protocol[1] << CHAR_BIT) | (embedded_protocol[0] << CHAR_BIT));
|
|
+
|
|
+ switch (res) {
|
|
+ case 0x0800: /* 1: Internet Protocol (IP) */
|
|
+ if (protocol_type == PACKET_IP) return 1;
|
|
+ break;
|
|
+ default:
|
|
+ return 0;
|
|
+ }
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+static const struct protocol_field eth_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "dmac", 0, 48, NULL, NULL },
|
|
+ { "smac", 48, 48, NULL, NULL },
|
|
+ { "type", 96, 16, NULL, NULL },
|
|
+ { "data", 112, 0, NULL, NULL },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+static const struct protocol_buf eth_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_ETH,
|
|
+ .payload_field = "data",
|
|
+ .protocol_fields = (struct protocol_field *)ð_protocol_fields,
|
|
+ .has_protocol = ð_has_protocol,
|
|
+ .get_field_changes = NULL,
|
|
+};
|
|
+
|
|
+
|
|
+void luaopen_protbuf_eth(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)ð_protocol_buf, PACKET_ETH);
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_helpers.c
|
|
@@ -0,0 +1,216 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+#include <linux/netfilter_ipv4.h>
|
|
+#include <linux/slab.h> /* kmalloc */
|
|
+#endif
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+int32_t get_header_size(struct protocol_buf * prot_buf)
|
|
+{
|
|
+ int32_t bit_counter = 0;
|
|
+ struct protocol_field * field = prot_buf->protocol_fields;
|
|
+
|
|
+ for (; field->name; field++)
|
|
+ bit_counter += field->length;
|
|
+
|
|
+ return bit_counter >> 3;
|
|
+}
|
|
+
|
|
+
|
|
+int32_t set_32_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ *(uint32_t *)(seg->start + seg->offset) = (uint32_t )htonl(luaL_checkinteger(L, 2));
|
|
+ return 0;
|
|
+}
|
|
+int32_t get_32_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushinteger(L, ntohl(*((uint32_t *)(seg->start + seg->offset))));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t set_16_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ *(uint16_t *)(seg->start + seg->offset) = (uint16_t)htons(luaL_checkinteger(L, 2));
|
|
+ return 0;
|
|
+}
|
|
+int32_t get_16_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushinteger(L, ntohs(*((uint16_t *)(seg->start + seg->offset))));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t set_lower_4_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ uint8_t b = (uint8_t)luaL_checkinteger(L, 2) << 4;
|
|
+ uint8_t * pos = (uint8_t *)(seg->start + seg->offset);
|
|
+
|
|
+ *pos &= 0x0F; /* reset lower 4 bits*/
|
|
+ *pos |= b;
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+int32_t get_lower_4_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushinteger(L, (*(uint8_t *)(seg->start + seg->offset)) >> 4);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t set_upper_4_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ uint8_t b = (uint8_t)luaL_checkinteger(L, 2) << 4;
|
|
+ uint8_t * pos = (uint8_t *)(seg->start + seg->offset);
|
|
+
|
|
+ *pos &= 0xF0; /* reset upper 4 bits*/
|
|
+ *pos |= (b >> 4);
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+int32_t get_upper_4_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushinteger(L, (*(uint8_t *)(seg->start + seg->offset)) & 0x0F);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+
|
|
+int32_t set_8_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ *(uint8_t *)(seg->start + seg->offset) = (uint8_t)luaL_checkinteger(L, 2);
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+int32_t get_8_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ lua_pushinteger(L, *(uint8_t *)(seg->start + seg->offset));
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t set_1_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ unsigned long l = 0;
|
|
+
|
|
+ memcpy(&l, (seg->start + seg->offset), seg->length);
|
|
+ l |= (1 << ((CHAR_BIT * seg->length) - luaL_checkinteger(L, 2)));
|
|
+ memcpy((seg->start + seg->offset), &l, seg->length);
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+int32_t get_1_bit_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ unsigned long l = 0;
|
|
+ uint32_t bit = 0;
|
|
+
|
|
+ memcpy(&l, (seg->start + seg->offset), seg->length);
|
|
+ bit = l & (1 << ((CHAR_BIT * seg->length) - luaL_checkinteger(L, 2)));
|
|
+
|
|
+ lua_pushboolean(L, bit);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t get_string_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+ /* Warning we cast from uchar to char */
|
|
+ lua_pushlstring(L, (char *)seg->start + seg->offset, seg->length);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+int32_t set_data_generic(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+ lua_packet_segment * data = checkbytearray(L, 2);
|
|
+
|
|
+ pr_debug("seg->length %u, data->length %u\n", seg->length, data->length);
|
|
+
|
|
+ if (seg->length >= data->length)
|
|
+ memcpy((seg->start + seg->offset), data->start, data->length);
|
|
+ else
|
|
+ luaL_error(L, "provided byte array too big for given packet segment");
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+struct field_changes * get_allocated_field_changes(lua_State *L, int32_t nr_of_fields)
|
|
+{
|
|
+ struct field_changes * changes;
|
|
+
|
|
+ changes = kmalloc(sizeof(struct field_changes), GFP_ATOMIC);
|
|
+
|
|
+ if (!changes)
|
|
+ goto failure;
|
|
+
|
|
+ changes->field_length_changes = kmalloc(nr_of_fields * sizeof(int), GFP_ATOMIC);
|
|
+ if (!changes->field_length_changes)
|
|
+ goto free1;
|
|
+
|
|
+ changes->field_offset_changes = kmalloc(nr_of_fields * sizeof(int), GFP_ATOMIC);
|
|
+ if (!changes->field_offset_changes)
|
|
+ goto free2;
|
|
+
|
|
+ memset(changes->field_length_changes, 0, nr_of_fields * sizeof(int));
|
|
+ memset(changes->field_offset_changes, 0, nr_of_fields * sizeof(int));
|
|
+
|
|
+ changes->ref_count = 1;
|
|
+
|
|
+ return changes;
|
|
+
|
|
+free2: kfree(changes->field_length_changes);
|
|
+free1: kfree(changes);
|
|
+failure:
|
|
+ if (!changes) luaL_error(L, "couldnt allocate memory inside 'get_allocated_field_changes'");
|
|
+ return NULL; /* only to omit warnings */
|
|
+}
|
|
\ No newline at end of file
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_icmp.c
|
|
@@ -0,0 +1,49 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+static int32_t icmp_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+static const struct protocol_field icmp_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "type", 0, 8, NULL, NULL },
|
|
+ { "code", 8, 8, NULL, NULL },
|
|
+ { "checksum", 16, 16, NULL, NULL },
|
|
+ { "id", 32, 16, NULL, NULL },
|
|
+ { "sequence", 48, 16, NULL, NULL },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+static const struct protocol_buf icmp_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_ICMP,
|
|
+ .payload_field = NULL,
|
|
+ .protocol_fields = (struct protocol_field *)&icmp_protocol_fields,
|
|
+ .has_protocol = &icmp_has_protocol,
|
|
+ .get_field_changes = NULL,
|
|
+};
|
|
+
|
|
+void luaopen_protbuf_icmp(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&icmp_protocol_buf, PACKET_ICMP);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_ip.c
|
|
@@ -0,0 +1,209 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ #include <net/checksum.h>
|
|
+ #include <net/tcp.h>
|
|
+#endif
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+
|
|
+#define IP_FMT "%u.%u.%u.%u"
|
|
+#define IP_ACC(buf) buf[0], buf[1], buf[2], buf[3]
|
|
+
|
|
+
|
|
+static int32_t ip_version_set(lua_State *L)
|
|
+{
|
|
+ uint8_t version_checked;
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *version_seg = seg->start + seg->offset;
|
|
+ int32_t version = luaL_checkinteger(L, 2);
|
|
+
|
|
+ luaL_argcheck(L, version >= 0 && version <= 15, 1, "version number invalid");
|
|
+
|
|
+ version_checked = (uint8_t)version;
|
|
+
|
|
+ version_seg[0] &= (uint8_t)0x0F; /* reset version bits */
|
|
+ version_seg[0] |= version_checked << 4;
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+static int32_t ip_version_get(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *version_seg = seg->start + seg->offset;
|
|
+ uint8_t v = version_seg[0] & 0xF0;
|
|
+
|
|
+ v >>= 4;
|
|
+
|
|
+ lua_pushinteger(L, v);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int32_t ip_ihl_set(lua_State *L)
|
|
+{
|
|
+ uint8_t ihl_checked;
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *ihl_seg = seg->start + seg->offset;
|
|
+ int32_t ihl = luaL_checkinteger(L, 2);
|
|
+
|
|
+ luaL_argcheck(L, ihl >= 5 && ihl <= 15, 1, "ip header length invalid"); // RFC 791 5x32 = 160 bits
|
|
+
|
|
+ ihl_checked = (uint8_t)ihl;
|
|
+
|
|
+ ihl_seg[0] &= (uint8_t)0xF0; /* reset ihl bits */
|
|
+ ihl_seg[0] |= ihl_checked;
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+static int32_t ip_ihl_get(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *ihl_seg = seg->start + seg->offset;
|
|
+ uint8_t v = ihl_seg[0] & 0x0F;
|
|
+
|
|
+ lua_pushinteger(L, v);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int32_t ip_addr_set(lua_State *L)
|
|
+{
|
|
+ int32_t field_id = lua_tointeger(L, lua_upvalueindex(2));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *addr_seg = seg->start + seg->offset;
|
|
+ uint32_t old_addr;
|
|
+ char *ip = (char *)luaL_checkstring(L, 2);
|
|
+ uint32_t a, b, c, d;
|
|
+ struct sk_buff * skb = (struct sk_buff *)lua_touserdata(L, 3);
|
|
+
|
|
+ /* for tcp / udp checksumming*/
|
|
+ uint32_t prot_offset;
|
|
+ uint8_t *check, *protocol_seg;
|
|
+
|
|
+ /* end */
|
|
+
|
|
+ sscanf(ip, IP_FMT, &a, &b, &c, &d);
|
|
+
|
|
+ luaL_argcheck(L, a < 256 && b < 256 && c < 256 && d < 256, 1, "invalid ip addr");
|
|
+
|
|
+ old_addr = *((uint32_t *)addr_seg);
|
|
+ addr_seg[0] = (uint8_t)a;
|
|
+ addr_seg[1] = (uint8_t)b;
|
|
+ addr_seg[2] = (uint8_t)c;
|
|
+ addr_seg[3] = (uint8_t)d;
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ if (old_addr != *(uint32_t *)addr_seg) {
|
|
+ int32_t offset = (field_id == 10) ? -2 : -6; /* offset from saddr or daddr */
|
|
+
|
|
+ csum_replace4((uint16_t *)(addr_seg + offset), old_addr, *(uint32_t *)addr_seg);
|
|
+
|
|
+ prot_offset = (field_id == 10) ? -3 : -7; /* offset from saddr or daddr */
|
|
+ protocol_seg = seg->start + seg->offset + prot_offset;
|
|
+
|
|
+ if (skb && (protocol_seg[0] == 0x06 || protocol_seg[0] == 0x11)) { /* is payload TCP or UDP ? */
|
|
+
|
|
+ check = seg->start + seg->offset; /* tmp res */
|
|
+ check += (field_id == 10) ? 8 : 16; /* the start of the payload, depending saddr or daddr */
|
|
+ check += (protocol_seg[0] == 0x06) ? 16 : 6; /* the start of the checksum, depending on TCP or UDP */
|
|
+
|
|
+ inet_proto_csum_replace4((__sum16 *)check, skb, old_addr, *(uint32_t *)addr_seg, 1);
|
|
+
|
|
+ lua_pop(L, 1);
|
|
+ }
|
|
+ }
|
|
+#endif
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+static int32_t ip_addr_get(lua_State *L)
|
|
+{
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, LUA_PACKET_SEG_IP);
|
|
+ uint8_t *addr_seg = seg->start + seg->offset;
|
|
+
|
|
+ char buf[16]; /*max: 255.255.255.255\0 --> 16 chars */
|
|
+
|
|
+ sprintf(buf, IP_FMT, IP_ACC(addr_seg));
|
|
+ lua_pushstring(L, buf);
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int32_t ip_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ uint8_t * embedded_protocol = seg->start + seg->offset + 9 /*bytes*/;
|
|
+
|
|
+ switch (embedded_protocol[0]) {
|
|
+ case 0x01: /* 1: Internet Control Message Protocol (ICMP) */
|
|
+ if (protocol_type == PACKET_ICMP) return 1;
|
|
+ break;
|
|
+ case 0x02: /* 2: Internet Group Management Protocol (IGMP) */
|
|
+ break;
|
|
+ case 0x06: /* 6: Transmission Control Protocol (TCP) */
|
|
+ if (protocol_type == PACKET_TCP) return 1;
|
|
+ break;
|
|
+ case 0x11: /* 17: User Datagram Protocol (UDP) */
|
|
+ if (protocol_type == PACKET_UDP) return 1;
|
|
+ break;
|
|
+ case 0x59: /* 89: Open Shortest Path First (OSPF) */
|
|
+ break;
|
|
+ case 0x84: /* 132: Stream Control Transmission Protocol (SCTP) */
|
|
+ break;
|
|
+ default:
|
|
+ break;
|
|
+ }
|
|
+
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+static const struct protocol_field ip_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "version", 0, 4, ip_version_get, ip_version_set },
|
|
+ { "ihl", 4, 4, ip_ihl_get, ip_ihl_set },
|
|
+ { "tos", 8, 8, get_8_bit_generic, set_8_bit_generic },
|
|
+ { "tot_len", 16, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "id", 32, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "flags", 48, 3, get_1_bit_generic, set_1_bit_generic },
|
|
+ { "frag_off", 51, 13, NULL, NULL },
|
|
+ { "ttl", 64, 8, get_8_bit_generic, set_8_bit_generic },
|
|
+ { "protocol", 72, 8, get_8_bit_generic, set_8_bit_generic },
|
|
+ { "check", 80, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "saddr", 96, 32, ip_addr_get, ip_addr_set },
|
|
+ { "daddr", 128, 32, ip_addr_get, ip_addr_set },
|
|
+ { "data", 160, 0, NULL, set_data_generic },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+static const struct protocol_buf ip_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_IP,
|
|
+ .payload_field = "data",
|
|
+ .protocol_fields = (struct protocol_field *)&ip_protocol_fields,
|
|
+ .has_protocol = &ip_has_protocol,
|
|
+ .get_field_changes = NULL,
|
|
+};
|
|
+
|
|
+void luaopen_protbuf_ip(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&ip_protocol_buf, PACKET_IP);
|
|
+}
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_raw.c
|
|
@@ -0,0 +1,43 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include "controller.h"
|
|
+static int32_t raw_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static const struct protocol_field raw_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "data", 0, 0, NULL, NULL },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+static const struct protocol_buf raw_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_RAW,
|
|
+ .payload_field = "data",
|
|
+ .protocol_fields = (struct protocol_field *)&raw_protocol_fields,
|
|
+ .has_protocol = &raw_has_protocol,
|
|
+ .get_field_changes = NULL,
|
|
+};
|
|
+
|
|
+void luaopen_protbuf_raw(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&raw_protocol_buf, PACKET_RAW);
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_tcp.c
|
|
@@ -0,0 +1,188 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ #include <net/checksum.h>
|
|
+ #include <net/tcp.h>
|
|
+#endif
|
|
+#include "controller.h"
|
|
+
|
|
+
|
|
+static int32_t tcp_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static int32_t tcp_set_checksum(lua_State *L)
|
|
+{
|
|
+ struct protocol_buf * prot_buf = (struct protocol_buf *)lua_topointer(L, lua_upvalueindex(1));
|
|
+ lua_packet_segment * seg = checkpacketseg(L, 1, prot_buf->name);
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ uint8_t * check_seg = seg->start + seg->offset;
|
|
+ uint8_t * tcp_hdr = check_seg - 16;
|
|
+ uint8_t * saddr = tcp_hdr - 8;
|
|
+ uint8_t * daddr = saddr + 4;
|
|
+ uint32_t len = 20 + (seg->changes->field_length_changes[11] / 8) + (seg->changes->field_length_changes[10] / 8);
|
|
+ unsigned short checksum = tcp_v4_check(len, *(uint32_t *)saddr, *(uint32_t *)daddr,
|
|
+ csum_partial(tcp_hdr, len, 0));
|
|
+
|
|
+ memcpy(check_seg, &checksum, sizeof(unsigned short));
|
|
+#endif
|
|
+ return 0;
|
|
+}
|
|
+
|
|
+
|
|
+static const struct protocol_field tcp_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "sport", 0, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "dport", 16, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "seq", 32, 32, get_32_bit_generic, set_32_bit_generic },
|
|
+ { "ack", 64, 32, get_32_bit_generic, set_32_bit_generic },
|
|
+ { "data_off", 96, 4, get_lower_4_bit_generic, set_lower_4_bit_generic },
|
|
+ { "reserved", 100, 4, get_upper_4_bit_generic, set_upper_4_bit_generic },
|
|
+ { "flags", 104, 8, get_1_bit_generic, set_1_bit_generic },
|
|
+ { "window_size", 112, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "check", 128, 16, get_16_bit_generic, tcp_set_checksum },
|
|
+ { "urgent", 144, 16, NULL, NULL },
|
|
+ { "options", 160, 0, NULL, set_data_generic },
|
|
+ { "data", 160, 0, NULL, set_data_generic }, /* begin of data depends on options */
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+
|
|
+static const struct protocol_field tcp_options_and_data[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "MSS", 0, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "WS", 0, 8, get_8_bit_generic, set_8_bit_generic },
|
|
+ { "SACK", 0, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "TSVAL", 0, 32, get_32_bit_generic, set_32_bit_generic },
|
|
+ { "TSER", 0, 32, get_32_bit_generic, set_32_bit_generic },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+
|
|
+static struct field_changes * tcp_get_field_changes(lua_State *L, lua_packet_segment * seg);
|
|
+
|
|
+static const struct protocol_buf tcp_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_TCP,
|
|
+ .payload_field = "data",
|
|
+ .protocol_fields = (struct protocol_field *)&tcp_protocol_fields,
|
|
+ .has_protocol = &tcp_has_protocol,
|
|
+ .get_field_changes = &tcp_get_field_changes,
|
|
+};
|
|
+
|
|
+
|
|
+static struct field_changes * tcp_options_get_field_changes(lua_State *L, lua_packet_segment * seg);
|
|
+
|
|
+static const struct protocol_buf tcp_options_and_data_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_TCP_OPT,
|
|
+ .payload_field = NULL,
|
|
+ .protocol_fields = (struct protocol_field *)&tcp_options_and_data,
|
|
+ .has_protocol = NULL,
|
|
+ .get_field_changes = &tcp_options_get_field_changes,
|
|
+};
|
|
+
|
|
+struct field_changes * tcp_get_field_changes(lua_State *L, lua_packet_segment * seg)
|
|
+{
|
|
+ /* depending on the value stored inside the 'data_off'-field, the length of
|
|
+ * the 'options' field has to be changed, as well as the length and offset
|
|
+ * of the 'data' field */
|
|
+ uint8_t *tcp_hdr = seg->start + seg->offset;
|
|
+
|
|
+ /* get the pointer to the 'data_off' field */
|
|
+ uint8_t * data_off_field = tcp_hdr + 12; /* 12 bytes offset */
|
|
+ /* extract the stored header length in bits */
|
|
+ uint32_t tcp_hdr_len = ((*(uint8_t *)data_off_field) >> 4) * 32;
|
|
+
|
|
+ /* get an allocated 'field_changes' structure */
|
|
+ struct field_changes * changes = get_allocated_field_changes(L, 12);
|
|
+
|
|
+ /* depending on the tcp header length, change the length of the options*/
|
|
+ changes->field_length_changes[10] = tcp_hdr_len - 160;
|
|
+ /* depending on the options length, change the offset of the data */
|
|
+ changes->field_offset_changes[11] = changes->field_length_changes[10];
|
|
+ changes->field_length_changes[11] = (seg->length * 8) - tcp_hdr_len;
|
|
+
|
|
+ return changes;
|
|
+
|
|
+}
|
|
+
|
|
+struct field_changes * tcp_options_get_field_changes(lua_State *L, lua_packet_segment * seg)
|
|
+{
|
|
+ /* depending on the value stored inside the 'data_off'-field, the length of
|
|
+ * the 'options' field has to be changed, as well as the length and offset
|
|
+ * of the 'data' field */
|
|
+ uint8_t *tcp_opt_hdr = seg->start + seg->offset;
|
|
+
|
|
+ /* get an allocated 'field_changes' structure */
|
|
+ struct field_changes * changes = get_allocated_field_changes(L, 5);
|
|
+
|
|
+ int32_t MSS = 0, WS = 0, SACK = 0, TS = 0, i;
|
|
+
|
|
+ uint8_t b1, b2;
|
|
+
|
|
+ for (i = 0; i < seg->length; i++) {
|
|
+ b1 = tcp_opt_hdr[i];
|
|
+ b2 = tcp_opt_hdr[i + 1];
|
|
+
|
|
+ if (b1 == 0x00)
|
|
+ break;
|
|
+
|
|
+ /* test for MSS */
|
|
+ if (!MSS && (b1 == 0x02 && b2 == 0x04)) {
|
|
+ changes->field_offset_changes[0] = (i + 2) * CHAR_BIT;
|
|
+ MSS = 1;
|
|
+ }
|
|
+
|
|
+ /* test for WS --- yet buggy somehow */
|
|
+ if (!WS && (b1 == 0x03 && b2 == 0x03)) {
|
|
+ changes->field_offset_changes[1] = (i + 2) * CHAR_BIT;
|
|
+ WS = 1;
|
|
+ }
|
|
+
|
|
+ /* test for SACK*/
|
|
+ if (!SACK && (b1 == 0x04 && b2 == 0x02)) {
|
|
+ changes->field_offset_changes[2] = i * CHAR_BIT; /* has no value */
|
|
+ SACK = 1;
|
|
+ }
|
|
+
|
|
+ /* test for TS */
|
|
+ if (!TS && (b1 == 0x08 && b2 == 0x0A)) {
|
|
+ changes->field_offset_changes[3] = (i + 2) * CHAR_BIT;
|
|
+ changes->field_offset_changes[4] = (i + 2 + 4) * CHAR_BIT;
|
|
+ TS = 1;
|
|
+ }
|
|
+ }
|
|
+
|
|
+ return changes;
|
|
+
|
|
+}
|
|
+
|
|
+void luaopen_protbuf_tcp(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&tcp_protocol_buf, PACKET_TCP);
|
|
+}
|
|
+void luaopen_protbuf_tcp_options(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&tcp_options_and_data_buf, PACKET_TCP_OPTIONS);
|
|
+}
|
|
+
|
|
+
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_tftp.c
|
|
@@ -0,0 +1,87 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+#include "controller.h"
|
|
+
|
|
+static const struct protocol_field tftp_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "opcode", 0, 16, get_16_bit_generic, NULL},
|
|
+ { "filename", 0, 0, get_string_generic, NULL},
|
|
+ { "mode", 0, 0, get_string_generic, NULL},
|
|
+ { "block_nr", 0, 16, get_16_bit_generic, NULL},
|
|
+ { "data", 0, 0, NULL, NULL},
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+struct field_changes * tftp_get_field_changes(lua_State *L, lua_packet_segment * seg)
|
|
+{
|
|
+ /* depending on the value stored inside the 'opcode'-field we have to change
|
|
+ * offsets and lengths */
|
|
+ uint8_t *tftp_hdr = seg->start + seg->offset;
|
|
+ short opcode = ntohs(*((uint16_t *)tftp_hdr));
|
|
+ /* get an allocated 'field_changes' structure */
|
|
+ struct field_changes * changes = get_allocated_field_changes(L, 5);
|
|
+ switch (opcode) {
|
|
+ case 1: /* Read Request (RRQ) */
|
|
+ /* setting offset and length of field 'filename' */
|
|
+ changes->field_offset_changes[1] = sizeof(unsigned short) << 3;
|
|
+ changes->field_length_changes[1] = strlen((char *)tftp_hdr + sizeof(unsigned short)) << 3;
|
|
+ /* setting offset and length of field 'mode' */
|
|
+ changes->field_offset_changes[2] = changes->field_offset_changes[1] + changes->field_length_changes[1];
|
|
+ changes->field_length_changes[2] = strlen((char *)tftp_hdr + (changes->field_offset_changes[2] >> 3));
|
|
+ break;
|
|
+ case 2: /* Write Request (WRQ) */
|
|
+ /* setting offset and length of field 'filename' */
|
|
+ changes->field_offset_changes[1] = sizeof(unsigned short) << 3;
|
|
+ changes->field_length_changes[1] = strlen((char *)tftp_hdr + sizeof(unsigned short)) << 3;
|
|
+ /* setting offset and length of field 'mode' */
|
|
+ changes->field_offset_changes[2] = changes->field_offset_changes[1] + changes->field_length_changes[1];
|
|
+ changes->field_length_changes[2] = strlen((char *)tftp_hdr + (changes->field_offset_changes[2] >> 3));
|
|
+ break;
|
|
+ case 3: /* Data (DATA) */
|
|
+ /* setting offset of field 'block_nr' */
|
|
+ changes->field_offset_changes[3] = sizeof(unsigned short) << 3;
|
|
+ /* setting offset of field 'data' */
|
|
+ changes->field_offset_changes[4] = changes->field_offset_changes[3] + (sizeof(unsigned short) << 3);
|
|
+ break;
|
|
+ case 4: /* Acknowledgment (ACK) */
|
|
+ /* setting offset of field 'block_nr' */
|
|
+ changes->field_offset_changes[3] = sizeof(unsigned short) << 3;
|
|
+ break;
|
|
+ case 5: /* Error (ERROR) */
|
|
+ /* we don't care ... yet */
|
|
+ break;
|
|
+ default:
|
|
+ break;
|
|
+ }
|
|
+
|
|
+ return changes;
|
|
+}
|
|
+
|
|
+static const struct protocol_buf tftp_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_TFTP,
|
|
+ .payload_field = NULL,
|
|
+ .protocol_fields = (struct protocol_field *)&tftp_protocol_fields,
|
|
+ .has_protocol = NULL, /* we don't need it, since we don't provide a payload field */
|
|
+ .get_field_changes = tftp_get_field_changes,
|
|
+};
|
|
+
|
|
+void luaopen_protbuf_tftp(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&tftp_protocol_buf, PACKET_TFTP);
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/prot_buf_udp.c
|
|
@@ -0,0 +1,53 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#if defined(__KERNEL__)
|
|
+ #include <net/checksum.h>
|
|
+#endif
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+
|
|
+static int32_t udp_has_protocol(lua_State *L, struct protocol_buf * prot_buf, lua_packet_segment * seg, int32_t protocol_type)
|
|
+{
|
|
+ return 1;
|
|
+}
|
|
+
|
|
+static const struct protocol_field udp_protocol_fields[] = {
|
|
+ /* field name offset length getter setter */
|
|
+ { "sport", 0, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "dport", 16, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "length", 32, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "check", 48, 16, get_16_bit_generic, set_16_bit_generic },
|
|
+ { "data", 64, 0, NULL, NULL },
|
|
+ PROT_FIELD_SENTINEL,
|
|
+};
|
|
+
|
|
+static const struct protocol_buf udp_protocol_buf = {
|
|
+ .is_dynamic = 0,
|
|
+ .name = LUA_PACKET_SEG_UDP,
|
|
+ .payload_field = "data",
|
|
+ .protocol_fields = (struct protocol_field *)&udp_protocol_fields,
|
|
+ .has_protocol = &udp_has_protocol,
|
|
+ .get_field_changes = NULL,
|
|
+};
|
|
+
|
|
+void luaopen_protbuf_udp(lua_State *L)
|
|
+{
|
|
+ register_protbuf(L, (struct protocol_buf *)&udp_protocol_buf, PACKET_UDP);
|
|
+}
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/xt_LUA.h
|
|
@@ -0,0 +1,36 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre.graf@stud.unibas.ch>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#ifndef XT_LUA_H_
|
|
+#define XT_LUA_H_
|
|
+
|
|
+#define MAX_FILENAME_SIZE 256
|
|
+#define MAX_FUNCTION_SIZE 256
|
|
+#define MAX_SCRIPT_SIZE 32768
|
|
+#define LUA_STATE_ARRAY_SIZE 128
|
|
+
|
|
+/* the targetsize is stored in a u16, so max size of the xt_lua_tginfo cannot exceed 64K*/
|
|
+struct xt_lua_tginfo {
|
|
+ char buf[MAX_SCRIPT_SIZE];
|
|
+ char filename[MAX_FILENAME_SIZE];
|
|
+ char function[MAX_FUNCTION_SIZE];
|
|
+ __u64 script_size;
|
|
+ __u32 state_id;
|
|
+};
|
|
+
|
|
+#endif /* XT_LUA_H_ */
|
|
--- /dev/null
|
|
+++ b/extensions/LUA/xt_LUA_target.c
|
|
@@ -0,0 +1,286 @@
|
|
+/*
|
|
+ * Copyright (C) 2010 University of Basel <http://cn.cs.unibas.ch/>
|
|
+ * by Andre Graf <andre@dergraf.org>
|
|
+ *
|
|
+ * This program is free software; you can redistribute it and/or modify
|
|
+ * it under the terms of the GNU General Public License as published by
|
|
+ * the Free Software Foundation; either version 2 of the License, or
|
|
+ * (at your option) any later version.
|
|
+ *
|
|
+ * This program is distributed in the hope that it will be useful,
|
|
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
+ * GNU General Public License for more details.
|
|
+ *
|
|
+ * You should have received a copy of the GNU General Public License
|
|
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
+ */
|
|
+
|
|
+#include <linux/kernel.h>
|
|
+#include <linux/slab.h>
|
|
+#include <linux/module.h>
|
|
+#include <asm/uaccess.h>
|
|
+#include <net/ip.h>
|
|
+#include <linux/netfilter/x_tables.h>
|
|
+#include "xt_LUA.h"
|
|
+
|
|
+#include "controller.h"
|
|
+
|
|
+/*::*
|
|
+ * lua_envs
|
|
+ * ----------
|
|
+ * This array holds a defined number of `lua_envs`_ structures.
|
|
+ * The used array index is also used as the Lua state identifier.
|
|
+ * The size of the array is defined in `LUA_STATE_ARRAY_SIZE`_.
|
|
+ */
|
|
+struct lua_env * lua_envs[LUA_STATE_ARRAY_SIZE];
|
|
+
|
|
+/*::*
|
|
+ * lua_state_refs
|
|
+ * --------------
|
|
+ * This array holds the reference counts of the several `lua_nf_state`_s
|
|
+ * which are stored inside the array `lua_states`_.
|
|
+ */
|
|
+uint32_t lua_state_refs[LUA_STATE_ARRAY_SIZE] = { 0 };
|
|
+
|
|
+/*::*
|
|
+ * lua_tg
|
|
+ * ------
|
|
+ * This function is called whenever a packet matches all matching conditions
|
|
+ * inside a rule. It is the target. It extracts the state identifier comming
|
|
+ * inside the *xt_target_param* structure and uses it to access the proper
|
|
+ * Lua state inside the `lua_states`_ array.
|
|
+ *
|
|
+ * It then constructs a new Lua userdata of type *lua_packet_segment* and
|
|
+ * initializes it with the lowest network header available. This userdata
|
|
+ * is annotated with the Lua metatable `LUA_PACKET_SEG_RAW`_ which converts
|
|
+ * the userdata to a raw lua packet having all raw functions available.
|
|
+ * This raw packet is the single parameter to the Lua function *process_packet*
|
|
+ * which must be defined inside the Lua script provided by the user. So far
|
|
+ * hardcoded, may be later configured by Lua - subject to change.
|
|
+ *
|
|
+ * The process_packet function must return an integer value, the verdict. For
|
|
+ * convenience reasons xt_LUA exports the verdicts NF_ACCEPT, NF_DROP and
|
|
+ * XT_CONTINUE inside the *register_lua_packet_lib* function.
|
|
+ */
|
|
+
|
|
+spinlock_t lock = SPIN_LOCK_UNLOCKED;
|
|
+
|
|
+static uint32_t
|
|
+lua_tg(struct sk_buff *pskb, const struct xt_target_param *par)
|
|
+{
|
|
+ uint32_t verdict;
|
|
+ lua_packet_segment *p;
|
|
+ const struct xt_lua_tginfo *info = par->targinfo;
|
|
+ lua_State * L;
|
|
+
|
|
+ /* START critical section on SMP, PacketScript is on the sequential trail at the moment TODO*/
|
|
+ spin_lock_irq(&lock);
|
|
+
|
|
+ L = lua_envs[info->state_id]->L;
|
|
+
|
|
+ if (!skb_make_writable(pskb, pskb->len))
|
|
+ return NF_DROP;
|
|
+
|
|
+ /* call the function provided by --function parameter or the default 'process_packet' defined in Lua */
|
|
+ lua_getglobal(L, info->function);
|
|
+
|
|
+ /* push the lua_packet_segment as a parameter */
|
|
+ p = (lua_packet_segment *)lua_newuserdata(L, sizeof(lua_packet_segment));
|
|
+ if (pskb->mac_header)
|
|
+ p->start = pskb->mac_header;
|
|
+ else if (pskb->network_header)
|
|
+ p->start = pskb->network_header;
|
|
+ else if (pskb->transport_header)
|
|
+ p->start = pskb->transport_header;
|
|
+ p->offset = 0;
|
|
+ p->length = (unsigned long)pskb->tail - (unsigned long)p->start;
|
|
+ p->changes = NULL;
|
|
+
|
|
+ /* marking userdata 'lua_packet_seg' with the corresponding metatable */
|
|
+ luaL_getmetatable(L, LUA_PACKET_SEG_RAW);
|
|
+ lua_setmetatable(L, -2);
|
|
+
|
|
+ /* push a reference to the skb as a parameter, needed at the moment for calculating TCP checksum, but I am not happy with it*/
|
|
+ lua_pushlightuserdata(L, (void *)skb_get(pskb));
|
|
+
|
|
+ /* do the function call (2 argument, 1 result) */
|
|
+ if (lua_pcall(L, 2, 1, 0) != 0) {
|
|
+ printk(KERN_ERR "LUA [%d]: pcall '%s' failed: %s\n", info->state_id, info->function, lua_tostring(L, -1));
|
|
+ lua_pop(L, 1);
|
|
+ return NF_DROP;
|
|
+ }
|
|
+
|
|
+ if (!lua_isnumber(L, -1)) {
|
|
+ printk(KERN_ERR "LUA [%d]: function '%s' must return a verdict\n", info->state_id, info->function);
|
|
+ lua_pop(L, 1);
|
|
+ return NF_DROP;
|
|
+ }
|
|
+
|
|
+ verdict = lua_tonumber(L, -1);
|
|
+ lua_pop(L, 1);
|
|
+
|
|
+ kfree_skb(pskb);
|
|
+
|
|
+ /* END critical section on SMP */
|
|
+ spin_unlock_irq(&lock);
|
|
+
|
|
+
|
|
+ return verdict;
|
|
+
|
|
+}
|
|
+/* Helper for checkentry */
|
|
+static bool load_script_into_state(uint32_t state_id, unsigned long script_size, char *script_buf)
|
|
+{
|
|
+ char *buf = kmalloc(script_size, GFP_KERNEL);
|
|
+ int32_t ret;
|
|
+ struct lua_env * env = kmalloc(sizeof(struct lua_env), GFP_KERNEL);
|
|
+
|
|
+ if (!script_size > 0) {
|
|
+ pr_debug("LUA [%d]: script_size %lu < 0\n", state_id, script_size);
|
|
+ return false;
|
|
+ }
|
|
+
|
|
+ env->L = lua_open();
|
|
+ luaopen_base(env->L);
|
|
+ luaopen_controller(env->L);
|
|
+
|
|
+ lua_getglobal(env->L, "_G");
|
|
+ lua_pushinteger(env->L, state_id);
|
|
+ lua_setfield(env->L, -2, "STATE_ID");
|
|
+ lua_pop(env->L, 1); /* pop _G */
|
|
+
|
|
+ strncpy(buf, script_buf, script_size);
|
|
+ ret = luaL_loadbuffer(env->L, buf, script_size, "PacketScript, loadbuffer") ||
|
|
+ lua_pcall(env->L, 0, 1, 0);
|
|
+
|
|
+ if (ret != 0) {
|
|
+ printk(KERN_ERR "LUA [%d]: failure loading script, error %s \n", state_id, lua_tostring(env->L, -1));
|
|
+ lua_pop(env->L, 1);
|
|
+ kfree(buf);
|
|
+ kfree(env);
|
|
+ return false;
|
|
+ }
|
|
+
|
|
+ lua_envs[state_id] = env;
|
|
+
|
|
+ kfree(buf);
|
|
+
|
|
+ return true;
|
|
+}
|
|
+/*::*
|
|
+ * lua_tg_checkentry
|
|
+ * -----------------
|
|
+ * This function is used as a kernel-side sanity check of the data comming
|
|
+ * from the iptables userspace program. Since this is the function which is
|
|
+ * called everytime a new rule (with -j xt_LUA) is injected, this function
|
|
+ * is used to do the bookkeeping work, such as counting the reference of
|
|
+ * several Lua states and the initialization of new states if needed. As an
|
|
+ * extra initialization step it loads the provided Lua script into the Lua
|
|
+ * state.
|
|
+ *
|
|
+ * Lua state initialization
|
|
+ * ~~~~~~~~~~~~~~~~~~~~~~~~
|
|
+ * 1. If a new rule is inserted and there is no existing state for the given
|
|
+ * state identifier (default state identifier is 0) a new Lua state is
|
|
+ * initialized using *lua_open*.
|
|
+ * 2. The Lua base library is registered inside the newly initialized state.
|
|
+ * Have a look at *lua/lbaselib.c* to see what functions of the Lua base
|
|
+ * library are available inside Lua.
|
|
+ * 3. The Lua packet library is registered inside the Lua state using the
|
|
+ * function *register_lua_packet_lib*. So far this function only registers
|
|
+ * the Netfilter verdicts NF_ACCEPT, NF_DROP and XT_CONTINUE inside the
|
|
+ * global environment of the given Lua state.
|
|
+ * 4. All the protocol Buffers, and the functions for accessing the bytes are
|
|
+ * registered using *register_protocols*.
|
|
+ *
|
|
+ * Lua state reference counting
|
|
+ * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
+ * Bookkeeping of the Lua states inside the *lua_state_refs* array. The
|
|
+ * state identifier is mapped to the array index, which holds an integer
|
|
+ * counting the several initialized states.
|
|
+ *
|
|
+ * Loading the Lua script
|
|
+ * ~~~~~~~~~~~~~~~~~~~~~~
|
|
+ * Copying the buffer which was initialized by the userspace program to a
|
|
+ * buffer with the proper size. The script is then loaded by the function
|
|
+ * xt_LUA_loadcode, which wrapps the *luaL_loadbuffer* function and does
|
|
+ * some workqueue initialization. So far this is done each time this function
|
|
+ * is called, subject to change.
|
|
+ */
|
|
+static bool
|
|
+lua_tg_checkentry(const struct xt_tgchk_param *par)
|
|
+{
|
|
+ const struct xt_lua_tginfo *info = par->targinfo;
|
|
+
|
|
+ if (load_script_into_state(info->state_id, info->script_size, (char *)info->buf)) {
|
|
+ lua_state_refs[info->state_id]++;
|
|
+ return true;
|
|
+ }
|
|
+ return false;
|
|
+}
|
|
+
|
|
+/*::*
|
|
+ * lua_tg_destroy
|
|
+ * --------------
|
|
+ * This function is the counterpart of the `lua_tg_checkentry`_ function. It is
|
|
+ * responsible to free all the resources alocated inside the checkentry process.
|
|
+ * To be more specific it frees the Lua state using *lua_close* and kfree on all
|
|
+ * the dynamically allocated pointers to the registered dynamic protocol buffers.
|
|
+ *
|
|
+ * Additionally the function cares about decrementing the reference counters
|
|
+ * inside the array `lua_states`_.
|
|
+ */
|
|
+static void
|
|
+lua_tg_destroy(const struct xt_tgdtor_param *par)
|
|
+{
|
|
+ const struct xt_lua_tginfo *info = par->targinfo;
|
|
+ struct lua_env * env = lua_envs[info->state_id];
|
|
+
|
|
+ if (lua_state_refs[info->state_id] == 1) {
|
|
+ lua_close(env->L);
|
|
+ cleanup_dynamic_prot_bufs(); /* clean memory allocated by protocols defined in Lua */
|
|
+ kfree(env);
|
|
+ pr_debug("LUA [%d]: Rule removed, close Lua state\n", info->state_id);
|
|
+ } else
|
|
+ pr_debug("LUA [%d]: Rule removed, Lua state stays open, referenced %d time(s)\n",
|
|
+ info->state_id, lua_state_refs[info->state_id] - 1);
|
|
+
|
|
+ lua_state_refs[info->state_id]--;
|
|
+}
|
|
+
|
|
+static struct xt_target lua_tg_reg __read_mostly = {
|
|
+ .name = "LUA",
|
|
+ .revision = 0,
|
|
+ .family = NFPROTO_UNSPEC,
|
|
+ .targetsize = XT_ALIGN(sizeof(struct xt_lua_tginfo)),
|
|
+ .target = lua_tg,
|
|
+ .checkentry = lua_tg_checkentry,
|
|
+ .destroy = lua_tg_destroy,
|
|
+ .me = THIS_MODULE,
|
|
+};
|
|
+
|
|
+
|
|
+static int32_t lua_tg_init(void)
|
|
+{
|
|
+ return xt_register_target(&lua_tg_reg);
|
|
+}
|
|
+
|
|
+static void lua_tg_exit(void)
|
|
+{
|
|
+ xt_unregister_target(&lua_tg_reg);
|
|
+}
|
|
+
|
|
+module_init(lua_tg_init);
|
|
+module_exit(lua_tg_exit);
|
|
+
|
|
+MODULE_AUTHOR("Andre Graf <andre@dergraf.org>");
|
|
+MODULE_DESCRIPTION("Xtables: Processing of matched packets using the Lua scripting environment");
|
|
+MODULE_ALIAS("ipt_LUA");
|
|
+MODULE_ALIAS("ipt6t_LUA");
|
|
+MODULE_ALIAS("arpt_LUA");
|
|
+MODULE_ALIAS("ebt_LUA");
|
|
+MODULE_LICENSE("GPL");
|
|
+
|
|
+
|
|
+
|
|
--- a/extensions/Kbuild
|
|
+++ b/extensions/Kbuild
|
|
@@ -27,6 +27,7 @@ obj-${build_pknock} += pknock/
|
|
obj-${build_psd} += xt_psd.o
|
|
obj-${build_quota2} += xt_quota2.o
|
|
obj-${build_rtsp} += rtsp/
|
|
+obj-${build_LUA} += LUA/
|
|
|
|
-include ${M}/*.Kbuild
|
|
-include ${M}/Kbuild.*
|
|
--- a/extensions/Mbuild
|
|
+++ b/extensions/Mbuild
|
|
@@ -22,3 +22,4 @@ obj-${build_pknock} += pknock/
|
|
obj-${build_psd} += libxt_psd.so
|
|
obj-${build_quota2} += libxt_quota2.so
|
|
obj-${build_gradm} += libxt_gradm.so
|
|
+obj-${build_LUA} += LUA/
|
|
--- a/mconfig
|
|
+++ b/mconfig
|
|
@@ -23,3 +23,4 @@ build_pknock=m
|
|
build_psd=m
|
|
build_quota2=m
|
|
build_rtsp=m
|
|
+build_LUA=m
|