zoukankan      html  css  js  c++  java
  • C/C++与lua实现互调

    1,在lua脚本中调用C/C++代码中的函数

    在C++中定义函数时必须以lua_State为参数, 以int为返回值才能被Lua所调用。

    /*

    typedef int (*lua_CFunction) (lua_State*L);

    C 函数的类型。

    为了正确的和 Lua 通讯,C 函数必须使用下列定义了参数以及返回值传递方法的协议: C 函数通过 Lua 中的堆栈来接受参数,参数以正序入栈(第一个参数首先入栈)。因此,当函数开始的时候, lua_gettop(L) 可以返回函数收到的参数个数。第一个参数(如果有的话)在索引 1 的地方,而最后一个参数在索引 lua_gettop(L) 处。当需要向 Lua 返回值的时候,C 函数只需要把它们以正序压到堆栈上(第一个返回值最先压入),然后返回这些返回值的个数。在这些返回值之下的,堆栈上的东西都会被 Lua 丢掉。和 Lua 函数一样,从 Lua 中调用 C 函数也可以有很多返回值。

    */

    下面这个例子中的函数将接收若干数字参数,并返回它们的平均数与和:

    #include <stdio.h>

    #ifdef __cplusplus

    extern "C" {

    #endif /* __cplusplus */

    #include <lua.h> // lua是用纯c语言写的

    #include <lualib.h>

    #include <lauxlib.h>

    #ifdef __cplusplus

    }

    #endif /* __cplusplus */

    int c_average(lua_State* L)

    {

       int n = lua_gettop(L); /* 返回栈顶元素的索引。因为索引是从1开始编号的(1表示栈底,-1表示栈顶),所以这个结果等于堆栈上的元素个数(返回0表示堆栈为空)。这里栈中元素的个数就是传入的参数个数 */

       double sum = 0;

       int i;

       for (i = 1; i <= n; i++)

        {

           if (!lua_isnumber(L, i))

           {

               lua_pushstring(L, "Incorrect argument to 'average'"); // 将错误信息压入栈中

               lua_error(L); // 抛出栈顶的错误

               /*

               int lua_error (lua_State *L);

               产生一个 Lua 错误。错误信息(实际上可以是任何类型的 Lua 值)必须被置入栈顶。这个函数会做一次长跳转,它不会再返回。

               */

           }

           sum += lua_tonumber(L, i); // lua_tonumber 将栈中指定index的值转换成数值类型的值,注意并不会从栈中弹出这个值

        }

       double avg = sum / n;

       lua_pushnumber(L, avg); // 将avg压入栈中,第1个返回值是avg

       lua_pushnumber(L, sum); // 将sum压入栈中,第2个返回值是sum

       return 2; /* return the number of results,该函数有2个返回值,即上面入栈的avg和sum */

    }

    int main(int argc, char* argv[])

    {

       lua_State* L = luaL_newstate(); // 创建一个新的独立的状态机

       

       /* register our function,告诉lua脚本其中调用的average函数(lua中的变量名)对应的是一个叫c_average的c语言函数 */

       lua_register(L, "average", c_average);

       /*

       void lua_register (lua_State *L,

       const char *name,

       lua_CFunction f);

        把 C 函数 f 设到全局变量 name 中。它通过一个宏定义:

       #define lua_register(L,n,f)

       (lua_pushcfunction(L, f), lua_setglobal(L, n))

       */

       /* run the script */

       luaL_dofile(L, "e1.lua"); // 加载脚本,脚本中的可执行语句将会得到执行

       

       /* Loads a file as a Lua chunk and then 以保护模式调用一个函数,等价于(luaL_loadfile(L,filename) || lua_pcall(L, 0, LUA_MULTRET, 0)) */

       lua_getglobal(L, "avg"); // 将全局变量avg的值入栈,等价于lua_getfield(L,LUA_GLOBALSINDEX, "name")

       printf("avg is: %d ", lua_tointeger(L, -1)); // 读栈顶的int值,注意不会从栈中弹出这个元素

       lua_pop(L, 1); // 弹出栈顶的一个元素

       

       lua_getglobal(L, "sum"); // 将全局变量sum的值入栈

       printf("sum is: %d ", lua_tointeger(L, -1));

       lua_pop(L, 1); // 弹出栈顶的一个元素

       

       /* cleanup Lua */

       lua_close(L);

       return 0;

    }

    /*

    -- e1.lua

    avg, sum = average(10, 20, 30, 40, 50)

    print("The average is ", avg)

    print("The sum is ", sum)

    */

    /*

    输出:

    The average is  30

    The sum is      150

    avg is: 30

    sum is: 150

    */

    2,在C/C++代码中调用lua脚本中的函数

    在C/C++代码中通过栈将参数传递给lua脚本中的函数,参数按照正序入栈,返回值由lua按正序压入栈中供C/C++访问。(函数调用成功后函数名和参数都出栈,返回值入栈)

    示例:

    #include <stdio.h>

    #ifdef __cplusplus

    extern "C" {

    #endif /* __cplusplus */

    #include <lua.h> // lua是用纯c语言写的

    #include <lualib.h>

    #include <lauxlib.h>

    #ifdef __cplusplus

    }

    #endif /* __cplusplus */

    lua_State* L = NULL;

    int luaAdd(int x, int y)

    {

       int sum;

       /* the function name,lua脚本中的函数名 */

       lua_getglobal(L, "add"); // lua函数名入栈

       int savedTop = lua_gettop(L); // 保存栈中的元素个数,后面要还原

       /* the first argument */

       lua_pushnumber(L, x); // 参数x入栈

       /* the second argument */

       lua_pushnumber(L, y); // 参数y入栈

       // 此时栈中有3个值,栈顶是参数y,栈底是函数名add

       /* call the function with 2 arguments, return 1 result */

       lua_call(L, 2, 1);

       /*

       void lua_call (lua_State *L, int nargs, int nresults);

        调用一个函数。

        要调用一个函数请遵循以下协议:首先,要调用的函数应该被压入堆栈;接着,把需要传递给这个函数的参数按正序压栈;这是指第一个参数首先压栈。最后调用一下 lua_call; nargs 是你压入堆栈的参数个数。当函数调用完毕后,所有的参数以及函数本身都会出栈。而函数的返回值这时则被压入堆栈。返回值的个数将被调整为 nresults 个,除非 nresults 被设置成 LUA_MULTRET。在这种情况下,所有的返回值都被压入堆栈中。 Lua 会保证返回值都放入栈空间中。函数返回值将按正序压栈(第一个返回值首先压栈),因此在调用结束后,最后一个返回值将被放在栈顶。

       lua_pcall (lua_State *L, int nargs, int nresults, int errfunc);

        以保护模式调用一个函数。

       nargs 和 nresults 的含义与 lua_call 中的相同。如果在调用过程中没有发生错误, lua_pcall 的行为和 lua_call 完全一致。但是,如果有错误发生的话, lua_pcall 会捕获它,然后把单一的值(错误信息)压入堆栈,然后返回错误码。同 lua_call 一样, lua_pcall 总是把函数本身和它的参数从栈上移除。

        如果 errfunc 是 0 ,返回在栈顶的错误信息就和原始错误信息完全一致。否则,errfunc 就被当成是错误处理函数在栈上的索引。(在当前的实现里,这个索引不能是伪索引。)在发生运行时错误时,这个函数会被调用而参数就是错误信息。错误处理函数的返回值将被 lua_pcall 作为出错信息返回在堆栈上。

        典型的用法中,错误处理函数被用来在出错信息上加上更多的调试信息,比如栈跟踪信息 (stack traceback) 。这些信息在 lua_pcall 返回后,因为栈已经展开 (unwound) ,所以收集不到了。

       lua_pcall 函数在调用成功时返回 0 ,否则返回以下(定义在 lua.h 中的)错误代码中的一个:

       LUA_ERRRUN: 运行时错误。

       LUA_ERRMEM: 内存分配错误。对于这种错,Lua 调用不了错误处理函数。

       LUA_ERRERR: 在运行错误处理函数时发生的错误。

       */

       // 因为add只有一个返回值,所以此时栈中只有1个值(如果有多个返回值,也是按正序入栈)

       /* get the result */

       sum = (int) lua_tonumber(L, -1); // -1是栈顶

       lua_pop(L, 1); // 返回值出栈

       /* 取出脚本中的变量z的值 */

       lua_getglobal(L, "z"); // 变量z的值入栈

       int z = (int) lua_tonumber(L, -1);

       printf("z = %d ", z);

       lua_pop(L, 1); // 变量z出栈

       lua_pushnumber(L, 4); // 4入栈

       lua_setglobal(L, "r");

       /*

       void lua_setglobal (lua_State *L, const char *name);

        从堆栈上弹出一个值,并将其设到全局变量 name 中。它由一个宏定义出来:

       #define lua_setglobal(L,s)  lua_setfield(L, LUA_GLOBALSINDEX, s)

       */

       lua_getglobal(L, "r"); // 变量r的值入栈

       int r = (int)lua_tonumber(L, -1);

       printf("r = %d ", r);

       lua_pop(L, 1); // 变量r出栈

       lua_settop(L, savedTop); // 函数退出之前恢复原来的栈

       return sum;

    }

    int main(int argc, char* argv[])

    {

        L= luaL_newstate(); // 创建lua运行环境

       /* load the script,加载脚本,为后面读取其中的变量做准备 */

       luaL_dofile(L, "e2.lua");

       /* call the add function */

       int sum = luaAdd(10, 15);

       /* print the result */

       printf("The sum is %d ", sum);

       /* cleanup Lua */

       lua_close(L);

       return 0;

    }

    /*

    -- e2.lua

    -- add two numbers

    function add(x, y)

       return x + y

    end

    z = 6

    */

    /*

    输出:

    z = 6

    r = 4

    The sum is 25

    */

    3,综合例子

    #include <stdio.h>

    #ifdef __cplusplus

    extern "C" {

    #endif /* __cplusplus */

    #include <lua.h> // lua是用纯c语言写的

    #include <lualib.h>

    #include <lauxlib.h>

    #ifdef __cplusplus

    }

    #endif /* __cplusplus */

    #define err_exit(num,fmt,args...) 

       do{printf("[%s:%d]"fmt" ",__FILE__,__LINE__,##args);exit(num);}while(0)

    #define err_return(num,fmt,args...) 

       do{printf("[%s:%d]"fmt" ",__FILE__,__LINE__,##args);return(num);}while(0)

    // lua中调用的c函数定义,实现加法

    int c_sum(lua_State* L)

    {

       int a = lua_tointeger(L, 1);

       int b = lua_tointeger(L, 2);

       lua_pushinteger(L, a + b);

       return 1;

    }

    int main(int argc, char* argv[])

    {

       lua_State* L = luaL_newstate(); //创建lua运行环境

       if (L == NULL)

           err_return(-1, "luaL_newstat() failed");

       int ret = 0;

       ret = luaL_loadfile(L, "e3.lua"); // Loads a file as a Luachunk. 底层调用lua_load,功能是:加载一个 Lua chunk 。如果没有错误, lua_load 把一个编译好的 chunk 作为一个 Lua 函数压入堆栈。否则,压入出错信息。

       if (ret != 0)

           err_return(-1, "luaL_loadfile failed");

       ret = lua_pcall(L, 0, 0, 0); // 以保护模式调用一个函数。 即刚刚加载的文件

       if (ret != 0)

           err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

       lua_getglobal(L, "width"); // 将全局变量width的值入栈

       lua_getglobal(L, "height");// 将全局变量height的值入栈

       printf("height: %ld, %ld ", (long) lua_tointeger(L,-1), (long) lua_tointeger(L, -2));

       

       lua_pop(L, 1); // 从栈中弹出一个值,从栈顶开始

       int a = 11;

       int b = 12;

       lua_getglobal(L, "mySum1"); // 将全局变量mySum1入栈,注意这里的mySum1是一个函数

       lua_pushinteger(L, a); // 第1个整型参数入栈,从左往右的顺序

       lua_pushinteger(L, b); // 第2个整型参数入栈,从左往右的顺序

       ret = lua_pcall(L, 2, 1, 0); // 调用函数,2个参数,1个返回值。调用成功后,函数和参数均出战,返回值入栈。如果有多个返回值,则第一个返回值先入栈

       if (ret != 0)

           err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));// 调用函数失败后栈顶存放失败原因字符串

       printf("sum: %d + %d = %ld ", a, b, (long) lua_tointeger(L,-1)); // sum只有一个返回值,调用成功后栈顶存放函数的返回值

       

       lua_pop(L, 1);

       

       const char str1[] = "hello";

       const char str2[] = "world";

       lua_getglobal(L, "myStrcat"); // 调用lua中的函数myStrcat

       lua_pushstring(L, str1);

       lua_pushstring(L, str2);

       ret = lua_pcall(L, 2, 1, 0);

       if (ret != 0)

           err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

       printf("mystrcat: %s %s = %s ", str1, str2, lua_tostring(L,-1));

       

       lua_pop(L, 1);

       

       lua_pushcfunction(L, c_sum); // 将一个 C 函数压入堆栈

       lua_setglobal(L, "mySum2");

       /*

       void lua_setglobal (lua_State *L, const char *name);

        从堆栈上弹出一个值,并将其设到全局变量 name 中。它由一个宏定义出来:

       #define lua_setglobal(L,s)  lua_setfield(L, LUA_GLOBALSINDEX, s)

       */

       // 以上两句可以用lua_register(L, "mySum2", c_sum);来代替

       /* register our function,告诉lua脚本其中调用的average函数(也是一个变量名)其实对应的是一个叫c_averaged的c代码函数 */

       // 调用lua中的mySum2函数,该函数调用c_sum函数实现加法

       lua_getglobal(L, "mySum2");

       lua_pushinteger(L, a);

       lua_pushinteger(L, b);

       ret = lua_pcall(L, 2, 1, 0);

       if (ret != 0)

           err_return(-1, "lua_pcall failed: %s", lua_tostring(L, -1));

       printf("mysum: %d + %d = %ld ", a, b, (long) lua_tointeger(L,-1));

       

       lua_pop(L, 1);

       

       lua_close(L); // 释放lua运行环境

       return 0;

    }

    /*

    -- e3.lua

    -- 变量定义

    width = 1

    height = 2

    -- lua函数定义,实现加法

    function mySum1(a, b)

       return a + b

    end

    -- lua函数定义,实现字符串相加

    function myStrcat(a, b)

       return a.. "_" .. b

    end

    -- lua函数定义,通过调用c代码中的csum函数实现加法

    function mysum(a, b)

       return csum(a, b)

    end

    */

    /*

    输出:

    height: 2, 1

    sum: 11 + 12 = 23

    mystrcat: hello world = hello_world

    mysum: 11 + 12 = 23

    */

  • 相关阅读:
    TCP 的 Keepalive 和 HTTP 的 Keep-Alive 是一个东西吗?
    图解 Redis | 差点崩溃了,还好有主从复制
    图解 Redis | 不多说了,这就是 RDB 快照
    图解 Redis | 不就是 AOF 持久化嘛
    看书的一点小建议!
    图解高性能网络架构:Reactor 和 Proactor
    面试官:什么是死锁?怎么排查死锁?怎么避免死锁?
    HTTP/1.1 有点慢,我想优化下!
    Linux 内核和 Windows 内核有什么区别?
    提高代码颜值的几个小技巧
  • 原文地址:https://www.cnblogs.com/wdd-123/p/4727709.html
Copyright © 2011-2022 走看看