C语言模块回调Lua函数的两种方法

技术  /  管理员 发布于 5年前   344

lua和C通过虚拟栈这种交互方式简单而又可靠，缺点就是C做栈平衡稍微会多写一点代码。 今天分享学到的C模块回调Lua函数的两种方法，都是炒冷饭，大侠勿喷。

1. C保存函数对象

C模块可以通过注册表保存Lua里面的对象，等适当时候取出再调用即可。
复制代码 代码如下:
static int lua_callback = LUA_REFNIL;

static int setnotify(lua_State *L)
{
  lua_callback = luaL_ref(L, LUA_REGISTRYINDEX);
  return 0;
}

static int testnotify(lua_State *L)
{
  lua_rawgeti(L, LUA_REGISTRYINDEX, lua_callback);
  lua_call(L, 0, 0);
}

luaL_ref把栈顶的值取出，放到指定的tabel中，然后返回一个索引（目测是数组的index)。 lua_rawgeti把之前保存的function对象取出，再由lua_call调用。
复制代码 代码如下:
function callback(  )
    print "Callback"
end

cb.setnotify(callback)
cb.testnotify()

2. C访问Lua全局环境
第二种方法更简便，C直接调用Lua中的函数，就像Lua调用C一样
复制代码 代码如下:
static int testenv(lua_State *L)
{
  lua_getglobal(L, "defcallback");
  lua_call(L, 0, 0);
}

该方法的缺点就是如果C模块独立编写，方法名就不太灵活。 用这种方法一般会在Lua端再封装一层，以隔离全局环境。

3. 完整例子
cb.c
复制代码 代码如下:
#include <stdio.h>
#include <stdlib.h>
#include "lua.h"
#include "lualib.h"
#include "lauxlib.h"

static int lua_callback = LUA_REFNIL;

static int setnotify(lua_State *L)
{
  lua_callback = luaL_ref(L, LUA_REGISTRYINDEX);
  return 0;
}

static int testnotify(lua_State *L)
{
  lua_rawgeti(L, LUA_REGISTRYINDEX, lua_callback);
  lua_call(L, 0, 0);
}

static int testenv(lua_State *L)
{
  lua_getglobal(L, "defcallback");
  lua_call(L, 0, 0);
}

static const luaL_Reg cblib[] = {
  {"setnotify", setnotify},
  {"testnotify", testnotify},
  {"testenv", testenv},
  {NULL, NULL}
};

int luaopen_cb(lua_State *L)
{
  luaL_register(L, "cb", cblib);
  return 1;
}

test.lua
复制代码 代码如下:
require("cb")

function callback(  )
  print "Callback"
end

function defcallback()
  print "Predef callback"
end

cb.setnotify(callback)
cb.testnotify()
print "Done"
cb.testenv()