C函数返回多个参数给lua

lua通过一个运行时栈来维护参数传递及返回,使用lua_to*等函数获取lua传递到C函数的参数,使用lua_push*从C函数返回值到lua脚本。此外也可以使用lua_getglobal从C函数获取lua脚本定义的全局变量。

#include <lua.h>
    #include <lauxlib.h>

    #include <stdlib.h> /* For function exit() */
    #include <stdio.h> /* For input/output */

    void bail(lua_State *L, char *msg){
        fprintf(stderr, "
FATAL ERROR:
 %s: %s

",
            msg, lua_tostring(L, -1));
        exit(1);
    }
    int lua_func_from_c(lua_State *L)
    {
        printf("This is C
");
        int argc = lua_gettop(L);    /* number of arguments */
        const char * str = lua_tostring(L,1);    /* the first argument: string */
        int num = lua_tonumber(L,2); /* the second argument: number */
        
        printf("The first argument: %s
", str);
        printf("The second argument: %d
", num);

        lua_getglobal(L,"global_var");
        const char * global_str = lua_tostring(L,-1);
        printf("global_var is %s
", global_str);

        int the_second_ret = 2*num;
        lua_pushstring(L, "the first return");
        lua_pushnumber(L, the_second_ret);
        return 2;            /* tell lua how many variables are returned */
    }
    int main(int argc, const char *argv[])
    {
        if(argc != 2)
        {
            return 1;
        }
        lua_State *L = luaL_newstate(); 	/* Create new lua state variable */

        /* Load Lua libraries, otherwise, the lua function in *.lua will be nil */
        luaL_openlibs(L);
        
        /* register new lua function in C */
        lua_register(L, "lua_func_from_c", lua_func_from_c);

        if( luaL_loadfile(L,argv[1]) ) 	/* Only load the lua script file */
            bail(L, "luaL_loadfile() failed");

        if( lua_pcall(L,0,0,0) ) 		/* Run the loaded lua file */
            bail(L, "lua_pcall() failed");
        lua_close(L);                 	/* Close the lua state variable */    

        return 0;
    }

lua脚本(my.lua)如下所示(在lua脚本中,如果变量不用local明确声明为局部变量,则默认为全局变量):

print("Hello world")
    global_var = "this is a global string"
    first, second = lua_func_from_c("the first one", 2)
    print("the first returned", first)
    print("the second returned", second)

执行结果:

Hello world This is C The first argument: the first one The second argument: 2 global_var is this is a global string the first returned the first return the second returned 4

经验分享 程序员 微信小程序 职场和发展