Cocos2d-Lua-Community/cocos/scripting/lua-bindings/manual/network/LuasocketAsyn.cpp

// Created by u0u0 on 2018/10/19.

#include <string.h>
#include <thread>

// for cpp call c
#ifdef __cplusplus
extern "C" {
#endif

#include "lua.h"
#include "auxiliar.h"
#include "inet.h"
#include "tcp.h"

#ifdef __cplusplus
}
#endif

#include "cocos2d.h"
#include "scripting/lua-bindings/manual/tolua_fix.h"
#include "scripting/lua-bindings/manual/CCLuaEngine.h"

// first connect to server may block main thread, new a thread to void UI blocked
static void connectThread(p_tcp tcp, const char *address, const char *port)
{
    struct addrinfo connecthints;
    const char *err;
    memset(&connecthints, 0, sizeof(connecthints));
    connecthints.ai_socktype = SOCK_STREAM;
    /* make sure we try to connect only to the same family */
    connecthints.ai_family = tcp->family;
    timeout_markstart(&tcp->tm);
    err = inet_tryconnect(&tcp->sock, &tcp->family, address, port,
                          &tcp->tm, &connecthints);
    tcp->asynConnStat = 2;// DONE first connect
}

extern "C" int asyn_connect(lua_State *L); // for c call cpp
int asyn_connect(lua_State *L)
{
    const char *address = luaL_checkstring(L, 2);
    const char *port = luaL_checkstring(L, 3);
    p_tcp tcp = (p_tcp) auxiliar_checkgroup(L, "tcp{any}", 1);
    if (tcp->asynConnStat == 0) {
        // start thread
        tcp->asynConnStat = 1;// first connecting
        auxiliar_setclass(L, "tcp{client}", 1);
        std::thread th(connectThread, tcp, address, port);
        th.detach();//exit from main thread, auto exit
    }
    if (tcp->asynConnStat < 2) {
        lua_pushnil(L);
        lua_pushstring(L, "Operation already in progress");
        return 2;
    }
    
    // sync connect for check status
    struct addrinfo connecthints;
    const char *err;
    memset(&connecthints, 0, sizeof(connecthints));
    connecthints.ai_socktype = SOCK_STREAM;
    /* make sure we try to connect only to the same family */
    connecthints.ai_family = tcp->family;
    timeout_markstart(&tcp->tm);
    err = inet_tryconnect(&tcp->sock, &tcp->family, address, port,
                          &tcp->tm, &connecthints);
    tcp->asynConnStat = 2;// DONE first connect
    /* have to set the class even if it failed due to non-blocking connects */
    auxiliar_setclass(L, "tcp{client}", 1);
    if (err) {
        lua_pushnil(L);
        lua_pushstring(L, err);
        return 2;
    }
    lua_pushnumber(L, 1);
    return 1;
}

// new a thread to void UI blocked
static void checkIpv6Thread(lua_State *L, const char *hostname, int handler)
{
    struct addrinfo *iterator = NULL, *resolved = NULL;
    struct addrinfo hints;
    int ret = 0;
    memset(&hints, 0, sizeof(hints));
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_family = PF_UNSPEC;
    ret = getaddrinfo(hostname, NULL, &hints, &resolved);
    if (ret != 0) {
        cocos2d::Director::getInstance()->getScheduler()->performFunctionInCocosThread([=](){
            const char *err = socket_gaistrerror(ret);
            CCLOG("== socket.dns.isIpv6 getaddrinfo error:%s", err);
            cocos2d::LuaStack *stack = cocos2d::LuaEngine::getInstance()->getLuaStack();
            lua_pushstring(L, err);
            stack->executeFunctionByHandler(handler, 1);
            toluafix_remove_function_by_refid(L, handler);
        });
        return;
    }
    
    int isIpv6 = 0;
    for (iterator = resolved; iterator; iterator = iterator->ai_next) {
        char hbuf[NI_MAXHOST];
        ret = getnameinfo(iterator->ai_addr, (socklen_t) iterator->ai_addrlen,
                          hbuf, (socklen_t) sizeof(hbuf), NULL, 0, NI_NUMERICHOST);
        if (ret) {
            freeaddrinfo(resolved);
            cocos2d::Director::getInstance()->getScheduler()->performFunctionInCocosThread([=](){
                const char *err = socket_gaistrerror(ret);
                CCLOG("== socket.dns.isIpv6 getnameinfo error:%s", err);
                cocos2d::LuaStack *stack = cocos2d::LuaEngine::getInstance()->getLuaStack();
                lua_pushstring(L, err);
                stack->executeFunctionByHandler(handler, 1);
                toluafix_remove_function_by_refid(L, handler);
            });
            return;
        }
        if (iterator->ai_family == AF_INET6) {
            isIpv6 = 1;
            break;
        }
    }
    freeaddrinfo(resolved);
    cocos2d::Director::getInstance()->getScheduler()->performFunctionInCocosThread([=](){
        cocos2d::LuaStack *stack = cocos2d::LuaEngine::getInstance()->getLuaStack();
        lua_pushnil(L);
        lua_pushboolean(L, isIpv6);
        stack->executeFunctionByHandler(handler, 2);
        toluafix_remove_function_by_refid(L, handler);
    });
}

extern "C" int inet_isIpv6_asyn(lua_State *L); // for c call cpp
int inet_isIpv6_asyn(lua_State *L)
{
    const char *hostname = luaL_checkstring(L, 1);
    int handler = toluafix_ref_function(L, 2, 0);
    
    std::thread th(checkIpv6Thread, L, hostname, handler);
    th.detach();//exit from main thread, auto exit
    return 0;
}