Lua 与 C 交互之UserData(4)
lua作为脚本于要能够使用宿主语言的类型,不管是宿主基本的或者扩展的类型结构,所以Lua提供的UserData来满足扩展的需求。在Lua中使用宿主语言的类型至少要考虑到几个方面:
- 数据内存
- 生命周期
- 数据操作
下面的内容主要参考《Lua程序设计》,数据保存在Lua堆栈中,通过Metatable对数据进行操作,并通过Lua的Gc进行回收内存。
1 Full UserData
void *lua_newuserdata (lua_State *L, size_t size);
This function allocates a new block of memory with the given size, pushes onto the stack a new full userdata with the block address, and returns this address.Userdata represents C values in Lua. A full userdata represents a block of memory. It is an object (like a table): you must create it, it can have its own metatable, and you can detect when it is being collected. A full userdata is only equal to itself (under raw equality).When Lua collects a full userdata with a gc metamethod, Lua calls the metamethod and marks the userdata as finalized. When this userdata is collected again then Lua frees its corresponding memory.
函数按照指定的大小分配一块内存,将对应的userdatum放到栈内,并返回内存块的地址。Userdata可以有自己的metatable ,如果Metatable中有__Gc元方法,回收时会调用改方法。回收之后Lua释放对应的内存。
void *lua_touserdata (lua_State *L, int index);
If the value at the given acceptable index is a full userdata, returns its block address. If the value is a light userdata, returns its pointer. Otherwise, returns NULL.
操作实例
//结构定义
typedef struct NumArray {
int size;
double values[1]; /* variable part */
} NumArray;
//创建数组
static int newarray (lua_State *L) {
int n = luaL_checkint(L, 1); //检查整数
size_t nbytes = sizeof(NumArray) + (n - 1)*sizeof(double);
NumArray *a = (NumArray *)lua_newuserdata(L, nbytes);
a->size = n;
return 1; /* new userdatum is already on the stack */
}
//设置数组数值set(a, index, value)
static int setarray (lua_State *L)
{
NumArray *a = (NumArray *)lua_touserdata(L, 1);
int index = luaL_checkint(L, 2);
double value = luaL_checknumber(L, 3);
luaL_argcheck(L, a != NULL, 1, "`array' expected");
luaL_argcheck(L, 1 <= index && index <= a->size, 2, "index out of range");
a->values[index-1] = value;
return 0;
}
//获取数组元素 get(a,index)
static int getarray (lua_State *L) {
NumArray *a = (NumArray *)lua_touserdata(L, 1);
int index = luaL_checkint(L, 2);
luaL_argcheck(L, a != NULL, 1, "'array' expected");
luaL_argcheck(L, 1 <= index && index <= a->size, 2,"index out of range");
lua_pushnumber(L, a->values[index-1]);
return 1;
}
// 库定义
static const struct luaL_reg arraylib [] = {
{"new", newarray},{"set", setarray},{"get", getarray},{NULL, NULL}
};
int luaopen_array (lua_State *L) {
luaL_openlib(L, "array", arraylib, 0);
return 1;
}
导入上面的库,我们可以在Lua中进行数组的操作:
a = array.new(1000) --创建
array.set(a, 1, 10) --设置
print(array.get(a, 10)) -- 获取
2 MetaTable
为了区分不同C类型的userData,可以为userData添加不同的metatables。每次我们访问时候,我们都要检查他是否有一个正确的 metatable。 Lua 代码不能改变 userdatum 的metatable,所以他不会伪造我们的代码。
操作函数
int luaL_newmetatable (lua_State *L, const char *tname);
创建新表(将用作 metatable),将新表放到栈顶并建立表和 registry 中类型名的联系
void luaL_getmetatable (lua_State *L, const char *tname);
获取 registry 中的 tname 对应的 metatable,压入栈中
void *luaL_checkudata (lua_State *L, int index, const char *tname);
检查在栈中指定位置的对象是否为带有给定名字的 metatable 的 userdata。如果对象不
存在正确的 metatable,返回 NULL,否则,返回userdata的地址
修改案例
//首先创建名字为LuaBook.array的meta 表
int luaopen_array (lua_State *L) {
luaL_newmetatable(L, "LuaBook.array");
...
}
//创建数组时,为Userdata添加元表方法
static int newarray (lua_State *L) {
...
luaL_getmetatable(L, "LuaBook.array");
lua_setmetatable(L, -2);
...
}
//添加检测metatable名字是否正确的接口
static NumArray *checkarray (lua_State *L) {
void *ud = luaL_checkudata(L, 1, "LuaBook.array");
luaL_argcheck(L, ud != NULL, 1, "`array' expected");
return (NumArray *)ud;
}
也就是说创建相关类型的metatable,然后在创建UserData的时候设置metatable,在获取时首先检测meta是否正对
3 面向对象访问
可以通过扩展userdata中metatable的元方法来实现对C对象进行面向对象方式的访问方式。
C中定义
// Metatable增加__index , __newindex方法
int luaopen_array (lua_State *L)
{
luaL_newmetatable(L, "LuaBook.array");
lua_pushstring(L, "__index");
lua_pushvalue(L, -2); /* pushes the metatable */
lua_settable(L, -3); /* metatable.__index = metatable */
//第一次调用,当我们传递一个NULL作为库名时,luaL_openlib并没有创建任何包含函数的表;相反,他认为封装函数的表在栈内,位于临时的upvalues的下面
luaL_openlib(L, NULL, arraylib_m, 0);
//根据给定的数组名创建一个新表,并在表中注册指定的函数
luaL_openlib(L, "array", arraylib_f, 0);
return 0;
}
// 定义Userdata中的方法
static const struct luaL_reg arraylib_f [] = {
{"new", newarray}, {NULL, NULL}
};
//定义Metatable中方法
static const struct luaL_reg arraylib_m [] = {
{"set", setarray},{"get", getarray},{NULL, NULL}
};
通过为Metatable扩展元方法的方式实现对象结构在Lua中访问方便。
luaL_openlib(L, NULL, arraylib_m, 0)应该是将arraylib_m 中的方法添加到metatable中,有点不好理解,其实也可以用settable的方式将set、get方法注入到metatable中。
Lua 中定义
另外也可以在Lua中设置userdata的元表,不过需要在array库导入之后执行。
function RegsterMetatable(a)
local metaarray = getmetatable(a)
metaarray.__index = metaarray
metaarray.set = array.set
metaarray.get = array.get
metaarray.size = array.size
end
3 关于full Userdata的生命周期
上面定义的操作流程 是通过lua_newuserdata 在Lua中创建内存,并返回地址给C,C提供方法对内存进行操作。通过UserData的Metatable方式为Lua中访问提供便利。但是,C中如果保存了lua_newuserdata 返回的指针,那么Lua中释放之后,C中保存的指针就会失效. 关于UserData的生命周期问题可以参考下云风的文章
4. Light UserData
一个light userdatum是一个表示C指针的值(也就是一个void *类型的值)。由于它是一个值,我们不能创建他们,使用函数lua_pushlightuserdata将一个light userdatum入栈:
void lua_pushlightuserdata (lua_State *L, void *p);
Light userdata不是一个缓冲区,仅仅是一个指针,没有metatables。像数字一样,light userdata不需要垃圾收集器来管理她。有些人把light userdata作为一个低代价的替代实现,来代替full userdata,但是这不是light userdata的典型应用。首先,使用light userdata你必须自己管理内存,因为他们和垃圾收集器无关。第二,尽管从名字上看有轻重之分,但full userdata实现的代价也并不大,比较而言,他只是在分配给定大小的内存时候,有一点点额外的代价。
Light userdata真正的用处在于可以表示不同类型的对象。当full userdata是一个对象的时候,它等于对象自身;另一方面,light userdata表示的是一个指向对象的指针,同样的,它等于指针指向的任何类型的userdata。所以,我们在Lua中使用light userdata表示C对象。
在云风的博客中找到的一些应用场景:
[Lua 中写 C 扩展库时用到的一些技巧][http://blog.codingnow.com/2006/11/lua_c.html]
[向 lua 虚拟机传递信息][http://blog.codingnow.com/2006/01/_lua.html]
[去掉 full userdata 的 GC 元方法][http://blog.codingnow.com/2013/08/full_userdata_gc.html]
结束语
本来不想去花时间写一些书上应该有的,不过在看ulua的实现中总是对其中的各种userdata的使用完全理解明白,所以还是来认真总结一下。当然中间也做了很多其他功课,后面会对其应用做进一步的介绍。
补充:
Lua5.1支持userdata中使用__gc元方法,Lua5.2之后对普通table也支持,在设置Api交互的时候比较便利。具体可以查看:
[Lua GC 之 Finalizer][http://www.cnblogs.com/JesseFang/archive/2012/12/27/2836160.html]