12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331 |
- // Module for network
- //#include "lua.h"
- #include "lualib.h"
- #include "lauxlib.h"
- #include "platform.h"
- #include "auxmods.h"
- #include "lrotable.h"
- #include "c_string.h"
- #include "c_stdlib.h"
- #include "c_types.h"
- #include "mem.h"
- #include "espconn.h"
- #ifdef CLIENT_SSL_ENABLE
- unsigned char *default_certificate;
- unsigned int default_certificate_len = 0;
- unsigned char *default_private_key;
- unsigned int default_private_key_len = 0;
- #endif
- #define TCP ESPCONN_TCP
- #define UDP ESPCONN_UDP
- static ip_addr_t host_ip; // for dns
- #if 0
- static int expose_array(lua_State* L, char *array, unsigned short len);
- #endif
- #define MAX_SOCKET 5
- static int socket_num = 0;
- static int socket[MAX_SOCKET];
- static lua_State *gL = NULL;
- static int tcpserver_cb_connect_ref = LUA_NOREF; // for tcp server connected callback
- static uint16_t tcp_server_timeover = 30;
- static struct espconn *pTcpServer = NULL;
- static struct espconn *pUdpServer = NULL;
- typedef struct lnet_userdata
- {
- struct espconn *pesp_conn;
- int self_ref;
- int cb_connect_ref;
- int cb_reconnect_ref;
- int cb_disconnect_ref;
- int cb_receive_ref;
- int cb_send_ref;
- int cb_dns_found_ref;
- #ifdef CLIENT_SSL_ENABLE
- uint8_t secure;
- #endif
- }lnet_userdata;
- static void net_server_disconnected(void *arg) // for tcp server only
- {
- NODE_DBG("net_server_disconnected is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if(gL == NULL)
- return;
- #if 0
- char temp[20] = {0};
- c_sprintf(temp, IPSTR, IP2STR( &(pesp_conn->proto.tcp->remote_ip) ) );
- NODE_DBG("remote ");
- NODE_DBG(temp);
- NODE_DBG(":");
- NODE_DBG("%d",pesp_conn->proto.tcp->remote_port);
- NODE_DBG(" disconnected.\n");
- #endif
- if(nud->cb_disconnect_ref != LUA_NOREF && nud->self_ref != LUA_NOREF)
- {
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_disconnect_ref);
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(client) to callback func in lua
- lua_call(gL, 1, 0);
- }
- int i;
- lua_gc(gL, LUA_GCSTOP, 0);
- for(i=0;i<MAX_SOCKET;i++){
- if( (LUA_NOREF!=socket[i]) && (socket[i] == nud->self_ref) ){
- // found the saved client
- nud->pesp_conn->reverse = NULL;
- nud->pesp_conn = NULL; // the espconn is made by low level sdk, do not need to free, delete() will not free it.
- nud->self_ref = LUA_NOREF; // unref this, and the net.socket userdata will delete it self
- luaL_unref(gL, LUA_REGISTRYINDEX, socket[i]);
- socket[i] = LUA_NOREF;
- socket_num--;
- break;
- }
- }
- lua_gc(gL, LUA_GCRESTART, 0);
- }
- static void net_socket_disconnected(void *arg) // tcp only
- {
- NODE_DBG("net_socket_disconnected is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if(nud->cb_disconnect_ref != LUA_NOREF && nud->self_ref != LUA_NOREF)
- {
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_disconnect_ref);
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(client) to callback func in lua
- lua_call(gL, 1, 0);
- }
- if(pesp_conn->proto.tcp)
- c_free(pesp_conn->proto.tcp);
- pesp_conn->proto.tcp = NULL;
- if(nud->pesp_conn)
- c_free(nud->pesp_conn);
- nud->pesp_conn = NULL; // espconn is already disconnected
- lua_gc(gL, LUA_GCSTOP, 0);
- if(nud->self_ref != LUA_NOREF){
- luaL_unref(gL, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = LUA_NOREF; // unref this, and the net.socket userdata will delete it self
- }
- lua_gc(gL, LUA_GCRESTART, 0);
- }
- static void net_server_reconnected(void *arg, sint8_t err)
- {
- NODE_DBG("net_server_reconnected is called.\n");
- net_server_disconnected(arg);
- }
- static void net_socket_reconnected(void *arg, sint8_t err)
- {
- NODE_DBG("net_socket_reconnected is called.\n");
- net_socket_disconnected(arg);
- }
- static void net_socket_received(void *arg, char *pdata, unsigned short len)
- {
- NODE_DBG("net_socket_received is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if(nud->cb_receive_ref == LUA_NOREF)
- return;
- if(nud->self_ref == LUA_NOREF)
- return;
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_receive_ref);
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(server) to callback func in lua
- // expose_array(gL, pdata, len);
- // *(pdata+len) = 0;
- // NODE_DBG(pdata);
- // NODE_DBG("\n");
- lua_pushlstring(gL, pdata, len);
- // lua_pushinteger(gL, len);
- lua_call(gL, 2, 0);
- }
- static void net_socket_sent(void *arg)
- {
- // NODE_DBG("net_socket_sent is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if(nud->cb_send_ref == LUA_NOREF)
- return;
- if(nud->self_ref == LUA_NOREF)
- return;
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_send_ref);
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(server) to callback func in lua
- lua_call(gL, 1, 0);
- }
- static void net_dns_found(const char *name, ip_addr_t *ipaddr, void *arg)
- {
- NODE_DBG("net_dns_found is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL){
- NODE_DBG("pesp_conn null.\n");
- return;
- }
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL){
- NODE_DBG("nud null.\n");
- return;
- }
- if(nud->cb_dns_found_ref == LUA_NOREF){
- NODE_DBG("cb_dns_found_ref null.\n");
- return;
- }
- if(ipaddr == NULL)
- {
- NODE_ERR( "DNS Fail!\n" );
- return;
- }
- // ipaddr->addr is a uint32_t ip
- char ip_str[20];
- c_memset(ip_str, 0, sizeof(ip_str));
- if(host_ip.addr == 0 && ipaddr->addr != 0)
- {
- c_sprintf(ip_str, IPSTR, IP2STR(&(ipaddr->addr)));
- }
- if(nud->self_ref == LUA_NOREF){
- NODE_DBG("self_ref null.\n");
- return;
- }
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_dns_found_ref); // the callback function
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(conn) to callback func in lua
- lua_pushstring(gL, ip_str); // the ip para
- lua_call(gL, 2, 0);
- if((pesp_conn->type == ESPCONN_TCP && pesp_conn->proto.tcp->remote_port == 0)
- || (pesp_conn->type == ESPCONN_UDP && pesp_conn->proto.udp->remote_port == 0) ){
- lua_gc(gL, LUA_GCSTOP, 0);
- if(nud->self_ref != LUA_NOREF){
- luaL_unref(gL, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = LUA_NOREF; // unref this, and the net.socket userdata will delete it self
- }
- lua_gc(gL, LUA_GCRESTART, 0);
- }
- }
- static void net_server_connected(void *arg) // for tcp only
- {
- NODE_DBG("net_server_connected is called.\n");
- struct espconn *pesp_conn = arg;
- int i = 0;
- lnet_userdata *skt = NULL;
- if(pesp_conn == NULL)
- return;
- #if 0
- char temp[20] = {0};
- c_sprintf(temp, IPSTR, IP2STR( &(pesp_conn->proto.tcp->remote_ip) ) );
- NODE_DBG("remote ");
- NODE_DBG(temp);
- NODE_DBG(":");
- NODE_DBG("%d",pesp_conn->proto.tcp->remote_port);
- NODE_DBG(" connected.\n");
- #endif
- for(i=0;i<MAX_SOCKET;i++){
- if(socket[i] == LUA_NOREF) // found empty slot
- {
- break;
- }
- }
- if(i>=MAX_SOCKET) // can't create more socket
- {
- NODE_ERR("MAX_SOCKET\n");
- pesp_conn->reverse = NULL; // not accept this conn
- if(pesp_conn->proto.tcp->remote_port || pesp_conn->proto.tcp->local_port)
- espconn_disconnect(pesp_conn);
- return;
- }
- if(tcpserver_cb_connect_ref == LUA_NOREF)
- return;
- if(!gL)
- return;
- lua_rawgeti(gL, LUA_REGISTRYINDEX, tcpserver_cb_connect_ref); // get function
- // create a new client object
- skt = (lnet_userdata *)lua_newuserdata(gL, sizeof(lnet_userdata));
- if(!skt){
- NODE_ERR("can't newudata\n");
- lua_pop(gL, 1);
- return;
- }
- // set its metatable
- luaL_getmetatable(gL, "net.socket");
- lua_setmetatable(gL, -2);
- // pre-initialize it, in case of errors
- skt->self_ref = LUA_NOREF;
- lua_pushvalue(gL, -1); // copy the top of stack
- skt->self_ref = luaL_ref(gL, LUA_REGISTRYINDEX); // ref to it self, for module api to find the userdata
- socket[i] = skt->self_ref; // save to socket array
- socket_num++;
- skt->cb_connect_ref = LUA_NOREF; // this socket already connected
- skt->cb_reconnect_ref = LUA_NOREF;
- skt->cb_disconnect_ref = LUA_NOREF;
- skt->cb_receive_ref = LUA_NOREF;
- skt->cb_send_ref = LUA_NOREF;
- skt->cb_dns_found_ref = LUA_NOREF;
- #ifdef CLIENT_SSL_ENABLE
- skt->secure = 0; // as a server SSL is not supported.
- #endif
- skt->pesp_conn = pesp_conn; // point to the espconn made by low level sdk
- pesp_conn->reverse = skt; // let espcon carray the info of this userdata(net.socket)
- espconn_regist_recvcb(pesp_conn, net_socket_received);
- espconn_regist_sentcb(pesp_conn, net_socket_sent);
- espconn_regist_disconcb(pesp_conn, net_server_disconnected);
- espconn_regist_reconcb(pesp_conn, net_server_reconnected);
- // now socket[i] has the client ref, and stack top has the userdata
- lua_call(gL, 1, 0); // function(conn)
- }
- static void net_socket_connected(void *arg)
- {
- NODE_DBG("net_socket_connected is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- // can receive and send data, even if there is no connected callback in lua.
- espconn_regist_recvcb(pesp_conn, net_socket_received);
- espconn_regist_sentcb(pesp_conn, net_socket_sent);
- espconn_regist_disconcb(pesp_conn, net_socket_disconnected);
- if(nud->cb_connect_ref == LUA_NOREF)
- return;
- if(nud->self_ref == LUA_NOREF)
- return;
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->cb_connect_ref);
- lua_rawgeti(gL, LUA_REGISTRYINDEX, nud->self_ref); // pass the userdata(client) to callback func in lua
- lua_call(gL, 1, 0);
- }
- // Lua: s = net.create(type, secure/timeout, function(conn))
- static int net_create( lua_State* L, const char* mt )
- {
- NODE_DBG("net_create is called.\n");
- struct espconn *pesp_conn = NULL;
- lnet_userdata *nud, *temp = NULL;
- unsigned type;
- #ifdef CLIENT_SSL_ENABLE
- unsigned secure = 0;
- #endif
- uint8_t stack = 1;
- bool isserver = false;
-
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_create.\n");
- return 0;
- }
- type = luaL_checkinteger( L, stack );
- if ( type != ESPCONN_TCP && type != ESPCONN_UDP )
- return luaL_error( L, "wrong arg type" );
- stack++;
- #ifdef CLIENT_SSL_ENABLE
- if(!isserver){
- if ( lua_isnumber(L, stack) )
- {
- secure = lua_tointeger(L, stack);
- stack++;
- if ( secure != 0 && secure != 1 ){
- return luaL_error( L, "wrong arg type" );
- }
- } else {
- secure = 0; // default to 0
- }
- }
- #endif
- if(isserver && type == ESPCONN_TCP){
- if ( lua_isnumber(L, stack) )
- {
- unsigned to = lua_tointeger(L, stack);
- stack++;
- if ( to < 1 || to > 28800 ){
- return luaL_error( L, "wrong arg type" );
- }
- tcp_server_timeover = (uint16_t)to;
- } else {
- tcp_server_timeover = 30; // default to 30
- }
- }
- // create a object
- nud = (lnet_userdata *)lua_newuserdata(L, sizeof(lnet_userdata));
- // pre-initialize it, in case of errors
- nud->self_ref = LUA_NOREF;
- nud->cb_connect_ref = LUA_NOREF;
- nud->cb_reconnect_ref = LUA_NOREF;
- nud->cb_disconnect_ref = LUA_NOREF;
- nud->cb_receive_ref = LUA_NOREF;
- nud->cb_send_ref = LUA_NOREF;
- nud->cb_dns_found_ref = LUA_NOREF;
- nud->pesp_conn = NULL;
- #ifdef CLIENT_SSL_ENABLE
- nud->secure = secure;
- #endif
- // set its metatable
- luaL_getmetatable(L, mt);
- lua_setmetatable(L, -2);
- // create the espconn struct
- if(isserver && type==ESPCONN_TCP && pTcpServer){
- if(tcpserver_cb_connect_ref != LUA_NOREF){ // self_ref should be unref in close()
- lua_pop(L,1);
- return luaL_error(L, "only one tcp server allowed");
- }
- pesp_conn = nud->pesp_conn = pTcpServer;
- } else if(isserver && type==ESPCONN_UDP && pUdpServer){
- temp = (lnet_userdata *)pUdpServer->reverse;
- if(temp && temp->self_ref != LUA_NOREF){
- lua_pop(L,1);
- return luaL_error(L, "only one udp server allowed");
- }
- pesp_conn = nud->pesp_conn = pUdpServer;
- } else {
- pesp_conn = nud->pesp_conn = (struct espconn *)c_zalloc(sizeof(struct espconn));
- if(!pesp_conn)
- return luaL_error(L, "not enough memory");
- pesp_conn->proto.tcp = NULL;
- pesp_conn->proto.udp = NULL;
- pesp_conn->reverse = NULL;
- if( type==ESPCONN_TCP )
- {
- pesp_conn->proto.tcp = (esp_tcp *)c_zalloc(sizeof(esp_tcp));
- if(!pesp_conn->proto.tcp){
- c_free(pesp_conn);
- pesp_conn = nud->pesp_conn = NULL;
- return luaL_error(L, "not enough memory");
- }
- NODE_DBG("TCP server/socket is set.\n");
- }
- else if( type==ESPCONN_UDP )
- {
- pesp_conn->proto.udp = (esp_udp *)c_zalloc(sizeof(esp_udp));
- if(!pesp_conn->proto.udp){
- c_free(pesp_conn);
- pesp_conn = nud->pesp_conn = NULL;
- return luaL_error(L, "not enough memory");
- }
- NODE_DBG("UDP server/socket is set.\n");
- }
- }
- pesp_conn->type = type;
- pesp_conn->state = ESPCONN_NONE;
- // reverse is for the callback function
- pesp_conn->reverse = nud;
- if(isserver && type==ESPCONN_TCP && pTcpServer==NULL){
- pTcpServer = pesp_conn;
- } else if(isserver && type==ESPCONN_UDP && pUdpServer==NULL){
- pUdpServer = pesp_conn;
- }
- gL = L; // global L for net module.
- // if call back function is specified, call it with para userdata
- // luaL_checkanyfunction(L, 2);
- if (lua_type(L, stack) == LUA_TFUNCTION || lua_type(L, stack) == LUA_TLIGHTFUNCTION){
- lua_pushvalue(L, stack); // copy argument (func) to the top of stack
- lua_pushvalue(L, -2); // copy the self_ref(userdata) to the top
- lua_call(L, 1, 0);
- }
- return 1;
- }
- // static int net_close( lua_State* L, const char* mt );
- // Lua: net.delete( socket/server )
- // call close() first
- // server: disconnect server, unref everything
- // socket: unref everything
- static int net_delete( lua_State* L, const char* mt )
- {
- NODE_DBG("net_delete is called.\n");
- bool isserver = false;
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_delete.\n");
- return 0;
- }
- // net_close( L, mt ); // close it first
- lnet_userdata *nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud==NULL){
- NODE_DBG("userdata is nil.\n");
- return 0;
- }
- if(nud->pesp_conn){ // for client connected to tcp server, this should set NULL in disconnect cb
- nud->pesp_conn->reverse = NULL;
- if(!isserver) // socket is freed here
- {
- if(nud->pesp_conn->type == ESPCONN_UDP){
- if(nud->pesp_conn->proto.udp)
- c_free(nud->pesp_conn->proto.udp);
- nud->pesp_conn->proto.udp = NULL;
- } else if (nud->pesp_conn->type == ESPCONN_TCP) {
- if(nud->pesp_conn->proto.tcp)
- c_free(nud->pesp_conn->proto.tcp);
- nud->pesp_conn->proto.tcp = NULL;
- }
- c_free(nud->pesp_conn);
- }
- nud->pesp_conn = NULL; // for socket, it will free this when disconnected
- }
- // free (unref) callback ref
- if(LUA_NOREF!=nud->cb_connect_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_connect_ref);
- nud->cb_connect_ref = LUA_NOREF;
- }
- if(LUA_NOREF!=nud->cb_reconnect_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_reconnect_ref);
- nud->cb_reconnect_ref = LUA_NOREF;
- }
- if(LUA_NOREF!=nud->cb_disconnect_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_disconnect_ref);
- nud->cb_disconnect_ref = LUA_NOREF;
- }
- if(LUA_NOREF!=nud->cb_receive_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_receive_ref);
- nud->cb_receive_ref = LUA_NOREF;
- }
- if(LUA_NOREF!=nud->cb_send_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_send_ref);
- nud->cb_send_ref = LUA_NOREF;
- }
- if(LUA_NOREF!=nud->cb_dns_found_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_dns_found_ref);
- nud->cb_dns_found_ref = LUA_NOREF;
- }
- lua_gc(gL, LUA_GCSTOP, 0);
- if(LUA_NOREF!=nud->self_ref){
- luaL_unref(L, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = LUA_NOREF;
- }
- lua_gc(gL, LUA_GCRESTART, 0);
- return 0;
- }
- static void socket_connect(struct espconn *pesp_conn)
- {
- if(pesp_conn == NULL)
- return;
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if( pesp_conn->type == ESPCONN_TCP )
- {
- #ifdef CLIENT_SSL_ENABLE
- if(nud->secure){
- espconn_secure_connect(pesp_conn);
- }
- else
- #endif
- {
- espconn_connect(pesp_conn);
- }
- }
- else if (pesp_conn->type == ESPCONN_UDP)
- {
- espconn_create(pesp_conn);
- }
- NODE_DBG("socket_connect is called.\n");
- }
- static void socket_dns_found(const char *name, ip_addr_t *ipaddr, void *arg);
- static dns_reconn_count = 0;
- static void socket_dns_found(const char *name, ip_addr_t *ipaddr, void *arg)
- {
- NODE_DBG("socket_dns_found is called.\n");
- struct espconn *pesp_conn = arg;
- if(pesp_conn == NULL){
- NODE_DBG("pesp_conn null.\n");
- return;
- }
- if(ipaddr == NULL)
- {
- dns_reconn_count++;
- if( dns_reconn_count >= 5 ){
- NODE_ERR( "DNS Fail!\n" );
- return;
- }
- NODE_ERR( "DNS retry %d!\n", dns_reconn_count );
- host_ip.addr = 0;
- espconn_gethostbyname(pesp_conn, name, &host_ip, socket_dns_found);
- return;
- }
- // ipaddr->addr is a uint32_t ip
- if(ipaddr->addr != 0)
- {
- dns_reconn_count = 0;
- if( pesp_conn->type == ESPCONN_TCP )
- {
- c_memcpy(pesp_conn->proto.tcp->remote_ip, &(ipaddr->addr), 4);
- NODE_DBG("TCP ip is set: ");
- NODE_DBG(IPSTR, IP2STR(&(ipaddr->addr)));
- NODE_DBG("\n");
- }
- else if (pesp_conn->type == ESPCONN_UDP)
- {
- c_memcpy(pesp_conn->proto.udp->remote_ip, &(ipaddr->addr), 4);
- NODE_DBG("UDP ip is set: ");
- NODE_DBG(IPSTR, IP2STR(&(ipaddr->addr)));
- NODE_DBG("\n");
- }
- socket_connect(pesp_conn);
- }
- }
- // Lua: server:listen( port, ip, function(con) )
- // Lua: socket:connect( port, ip, function(con) )
- static int net_start( lua_State* L, const char* mt )
- {
- NODE_DBG("net_start is called.\n");
- struct espconn *pesp_conn = NULL;
- lnet_userdata *nud;
- unsigned port;
- size_t il;
- bool isserver = false;
- ip_addr_t ipaddr;
- const char *domain;
- uint8_t stack = 1;
-
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_start.\n");
- return 0;
- }
- nud = (lnet_userdata *)luaL_checkudata(L, stack, mt);
- luaL_argcheck(L, nud, stack, "Server/Socket expected");
- stack++;
- if(nud==NULL){
- NODE_DBG("userdata is nil.\n");
- return 0;
- }
- pesp_conn = nud->pesp_conn;
- port = luaL_checkinteger( L, stack );
- stack++;
- if( pesp_conn->type == ESPCONN_TCP )
- {
- if(isserver)
- pesp_conn->proto.tcp->local_port = port;
- else{
- pesp_conn->proto.tcp->remote_port = port;
- pesp_conn->proto.tcp->local_port = espconn_port();
- }
- NODE_DBG("TCP port is set: %d.\n", port);
- }
- else if (pesp_conn->type == ESPCONN_UDP)
- {
- if(isserver)
- pesp_conn->proto.udp->local_port = port;
- else{
- pesp_conn->proto.udp->remote_port = port;
- pesp_conn->proto.udp->local_port = espconn_port();
- }
- NODE_DBG("UDP port is set: %d.\n", port);
- }
- if( lua_isstring(L,stack) ) // deal with the domain string
- {
- domain = luaL_checklstring( L, stack, &il );
- stack++;
- if (domain == NULL)
- {
- if(isserver)
- domain = "0.0.0.0";
- else
- domain = "127.0.0.1";
- }
- ipaddr.addr = ipaddr_addr(domain);
- if( pesp_conn->type == ESPCONN_TCP )
- {
- if(isserver)
- c_memcpy(pesp_conn->proto.tcp->local_ip, &ipaddr.addr, 4);
- else
- c_memcpy(pesp_conn->proto.tcp->remote_ip, &ipaddr.addr, 4);
- NODE_DBG("TCP ip is set: ");
- NODE_DBG(IPSTR, IP2STR(&ipaddr.addr));
- NODE_DBG("\n");
- }
- else if (pesp_conn->type == ESPCONN_UDP)
- {
- if(isserver)
- c_memcpy(pesp_conn->proto.udp->local_ip, &ipaddr.addr, 4);
- else
- c_memcpy(pesp_conn->proto.udp->remote_ip, &ipaddr.addr, 4);
- NODE_DBG("UDP ip is set: ");
- NODE_DBG(IPSTR, IP2STR(&ipaddr.addr));
- NODE_DBG("\n");
- }
- }
- // call back function when a connection is obtained, tcp only
- if ( pesp_conn->type == ESPCONN_TCP ) {
- if (lua_type(L, stack) == LUA_TFUNCTION || lua_type(L, stack) == LUA_TLIGHTFUNCTION){
- lua_pushvalue(L, stack); // copy argument (func) to the top of stack
- if(isserver) // for tcp server connected callback
- {
- if(tcpserver_cb_connect_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, tcpserver_cb_connect_ref);
- tcpserver_cb_connect_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- else
- {
- if(nud->cb_connect_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_connect_ref);
- nud->cb_connect_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- }
- }
- if(!isserver || pesp_conn->type == ESPCONN_UDP){ // self_ref is only needed by socket userdata, or udp server
- lua_pushvalue(L, 1); // copy to the top of stack
- if(nud->self_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- if( pesp_conn->type == ESPCONN_TCP )
- {
- if(isserver){ // no secure server support for now
- espconn_regist_connectcb(pesp_conn, net_server_connected);
- // tcp server, SSL is not supported
- #ifdef CLIENT_SSL_ENABLE
- // if(nud->secure)
- // espconn_secure_accept(pesp_conn);
- // else
- #endif
- espconn_accept(pesp_conn); // if it's a server, no need to dns.
- espconn_regist_time(pesp_conn, tcp_server_timeover, 0);
- }
- else{
- espconn_regist_connectcb(pesp_conn, net_socket_connected);
- espconn_regist_reconcb(pesp_conn, net_socket_reconnected);
- #ifdef CLIENT_SSL_ENABLE
- if(nud->secure){
- if(pesp_conn->proto.tcp->remote_port || pesp_conn->proto.tcp->local_port)
- espconn_secure_disconnect(pesp_conn);
- // espconn_secure_connect(pesp_conn);
- }
- else
- #endif
- {
- if(pesp_conn->proto.tcp->remote_port || pesp_conn->proto.tcp->local_port)
- espconn_disconnect(pesp_conn);
- // espconn_connect(pesp_conn);
- }
- }
- }
- else if (pesp_conn->type == ESPCONN_UDP)
- {
- espconn_regist_recvcb(pesp_conn, net_socket_received);
- espconn_regist_sentcb(pesp_conn, net_socket_sent);
- if(pesp_conn->proto.tcp->remote_port || pesp_conn->proto.tcp->local_port)
- espconn_delete(pesp_conn);
- if(isserver)
- espconn_create(pesp_conn); // if it's a server, no need to dns.
- }
- if(!isserver){
- if((ipaddr.addr == IPADDR_NONE) && (c_memcmp(domain,"255.255.255.255",16) != 0))
- {
- host_ip.addr = 0;
- dns_reconn_count = 0;
- if(ESPCONN_OK == espconn_gethostbyname(pesp_conn, domain, &host_ip, socket_dns_found)){
- socket_dns_found(domain, &host_ip, pesp_conn); // ip is returned in host_ip.
- }
- }
- else
- {
- socket_connect(pesp_conn);
- }
- }
- return 0;
- }
- // Lua: server/socket:close()
- // server disconnect everything, unref everything
- // client disconnect and unref itself
- static int net_close( lua_State* L, const char* mt )
- {
- NODE_DBG("net_close is called.\n");
- bool isserver = false;
- int i = 0;
- lnet_userdata *nud = NULL, *skt = NULL;
- nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud == NULL)
- return 0;
- if(nud->pesp_conn == NULL)
- return 0;
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_close.\n");
- return 0;
- }
- if(isserver && nud->pesp_conn->type == ESPCONN_TCP && tcpserver_cb_connect_ref != LUA_NOREF){
- luaL_unref(L, LUA_REGISTRYINDEX, tcpserver_cb_connect_ref);
- tcpserver_cb_connect_ref = LUA_NOREF;
- }
- int n = lua_gettop(L);
- skt = nud;
- do{
- if(isserver && skt == NULL){
- if(socket[i] != LUA_NOREF){ // there is client socket exists
- lua_rawgeti(L, LUA_REGISTRYINDEX, socket[i]); // get the referenced user_data to stack top
- #if 0
- socket[i] = LUA_NOREF;
- socket_num--;
- #endif // do this in net_server_disconnected
- i++;
- if(lua_isuserdata(L,-1)){
- skt = lua_touserdata(L,-1);
- } else {
- lua_pop(L, 1);
- continue;
- }
- }else{
- // skip LUA_NOREF
- i++;
- continue;
- }
- }
- if(skt==NULL){
- NODE_DBG("userdata is nil.\n");
- continue;
- }
- if(skt->pesp_conn) // disconnect the connection
- {
- if(skt->pesp_conn->type == ESPCONN_TCP)
- {
- #ifdef CLIENT_SSL_ENABLE
- if(skt->secure){
- if(skt->pesp_conn->proto.tcp->remote_port || skt->pesp_conn->proto.tcp->local_port)
- espconn_secure_disconnect(skt->pesp_conn);
- }
- else
- #endif
- {
- if(skt->pesp_conn->proto.tcp->remote_port || skt->pesp_conn->proto.tcp->local_port)
- espconn_disconnect(skt->pesp_conn);
- }
- }else if(skt->pesp_conn->type == ESPCONN_UDP)
- {
- if(skt->pesp_conn->proto.tcp->remote_port || skt->pesp_conn->proto.tcp->local_port)
- espconn_delete(skt->pesp_conn);
- // a udp server/socket unref it self here. not in disconnect.
- if(LUA_NOREF!=skt->self_ref){ // for a udp self_ref is NOREF
- luaL_unref(L, LUA_REGISTRYINDEX, skt->self_ref);
- skt->self_ref = LUA_NOREF; // for a socket, now only var in lua is ref to the userdata
- }
- }
- }
- #if 0
- // unref the self_ref
- if(LUA_NOREF!=skt->self_ref){ // for a server self_ref is NOREF
- luaL_unref(L, LUA_REGISTRYINDEX, skt->self_ref);
- skt->self_ref = LUA_NOREF; // for a socket, now only var in lua is ref to the userdata
- }
- #endif
- lua_settop(L, n); // reset the stack top
- skt = NULL;
- } while( isserver && i<MAX_SOCKET);
- #if 0
- // unref the self_ref, for both socket and server
- if(LUA_NOREF!=nud->self_ref){ // for a server self_ref is NOREF
- luaL_unref(L, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = LUA_NOREF; // now only var in lua is ref to the userdata
- }
- #endif
- return 0;
- }
- // Lua: socket/udpserver:on( "method", function(s) )
- static int net_on( lua_State* L, const char* mt )
- {
- NODE_DBG("net_on is called.\n");
- bool isserver = false;
- lnet_userdata *nud;
- size_t sl;
-
- nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud==NULL){
- NODE_DBG("userdata is nil.\n");
- return 0;
- }
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_on.\n");
- return 0;
- }
- const char *method = luaL_checklstring( L, 2, &sl );
- if (method == NULL)
- return luaL_error( L, "wrong arg type" );
- luaL_checkanyfunction(L, 3);
- lua_pushvalue(L, 3); // copy argument (func) to the top of stack
- if(!isserver && nud->pesp_conn->type == ESPCONN_TCP && sl == 10 && c_strcmp(method, "connection") == 0){
- if(nud->cb_connect_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_connect_ref);
- nud->cb_connect_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else if(!isserver && nud->pesp_conn->type == ESPCONN_TCP && sl == 12 && c_strcmp(method, "reconnection") == 0){
- if(nud->cb_reconnect_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_reconnect_ref);
- nud->cb_reconnect_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else if(!isserver && nud->pesp_conn->type == ESPCONN_TCP && sl == 13 && c_strcmp(method, "disconnection") == 0){
- if(nud->cb_disconnect_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_disconnect_ref);
- nud->cb_disconnect_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else if((!isserver || nud->pesp_conn->type == ESPCONN_UDP) && sl == 7 && c_strcmp(method, "receive") == 0){
- if(nud->cb_receive_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_receive_ref);
- nud->cb_receive_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else if((!isserver || nud->pesp_conn->type == ESPCONN_UDP) && sl == 4 && c_strcmp(method, "sent") == 0){
- if(nud->cb_send_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_send_ref);
- nud->cb_send_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else if(!isserver && nud->pesp_conn->type == ESPCONN_TCP && sl == 3 && c_strcmp(method, "dns") == 0){
- if(nud->cb_dns_found_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_dns_found_ref);
- nud->cb_dns_found_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }else{
- lua_pop(L, 1);
- return luaL_error( L, "method not supported" );
- }
- return 0;
- }
- // Lua: server/socket:send( string, function(sent) )
- static int net_send( lua_State* L, const char* mt )
- {
- // NODE_DBG("net_send is called.\n");
- bool isserver = false;
- struct espconn *pesp_conn = NULL;
- lnet_userdata *nud;
- size_t l;
-
- nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud==NULL){
- NODE_DBG("userdata is nil.\n");
- return 0;
- }
- if(nud->pesp_conn == NULL){
- NODE_DBG("nud->pesp_conn is NULL.\n");
- return 0;
- }
- pesp_conn = nud->pesp_conn;
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_send.\n");
- return 0;
- }
- if(isserver && nud->pesp_conn->type == ESPCONN_TCP){
- return luaL_error( L, "tcp server send not supported" );
- }
- #if 0
- char temp[20] = {0};
- c_sprintf(temp, IPSTR, IP2STR( &(pesp_conn->proto.tcp->remote_ip) ) );
- NODE_DBG("remote ");
- NODE_DBG(temp);
- NODE_DBG(":");
- NODE_DBG("%d",pesp_conn->proto.tcp->remote_port);
- NODE_DBG(" sending data.\n");
- #endif
- const char *payload = luaL_checklstring( L, 2, &l );
- if (l>1460 || payload == NULL)
- return luaL_error( L, "need <1460 payload" );
- if (lua_type(L, 3) == LUA_TFUNCTION || lua_type(L, 3) == LUA_TLIGHTFUNCTION){
- lua_pushvalue(L, 3); // copy argument (func) to the top of stack
- if(nud->cb_send_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_send_ref);
- nud->cb_send_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- #ifdef CLIENT_SSL_ENABLE
- if(nud->secure)
- espconn_secure_sent(pesp_conn, (unsigned char *)payload, l);
- else
- #endif
- espconn_sent(pesp_conn, (unsigned char *)payload, l);
- return 0;
- }
- // Lua: socket:dns( string, function(socket, ip) )
- static int net_dns( lua_State* L, const char* mt )
- {
- NODE_DBG("net_dns is called.\n");
- bool isserver = false;
- struct espconn *pesp_conn = NULL;
- lnet_userdata *nud;
- size_t l;
-
- nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud==NULL){
- NODE_DBG("userdata is nil.\n");
- return 0;
- }
- if (mt!=NULL && c_strcmp(mt, "net.server")==0)
- isserver = true;
- else if (mt!=NULL && c_strcmp(mt, "net.socket")==0)
- isserver = false;
- else
- {
- NODE_DBG("wrong metatable for net_send.\n");
- return 0;
- }
- pesp_conn = nud->pesp_conn;
- if(!isserver || pesp_conn->type == ESPCONN_UDP){ // self_ref is only needed by socket userdata, or udp server
- lua_pushvalue(L, 1); // copy to the top of stack
- if(nud->self_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->self_ref);
- nud->self_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- const char *domain = luaL_checklstring( L, 2, &l );
- if (l>128 || domain == NULL)
- return luaL_error( L, "need <128 domain" );
- if (lua_type(L, 3) == LUA_TFUNCTION || lua_type(L, 3) == LUA_TLIGHTFUNCTION){
- lua_pushvalue(L, 3); // copy argument (func) to the top of stack
- if(nud->cb_dns_found_ref != LUA_NOREF)
- luaL_unref(L, LUA_REGISTRYINDEX, nud->cb_dns_found_ref);
- nud->cb_dns_found_ref = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- host_ip.addr = 0;
- espconn_gethostbyname(pesp_conn, domain, &host_ip, net_dns_found);
- return 0;
- }
- // Lua: s = net.createServer(type, function(server))
- static int net_createServer( lua_State* L )
- {
- const char *mt = "net.server";
- return net_create(L, mt);
- }
- // Lua: server:delete()
- static int net_server_delete( lua_State* L )
- {
- const char *mt = "net.server";
- return net_delete(L, mt);
- }
- // Lua: server:listen( port, ip )
- static int net_server_listen( lua_State* L )
- {
- const char *mt = "net.server";
- return net_start(L, mt);
- }
- // Lua: server:close()
- static int net_server_close( lua_State* L )
- {
- const char *mt = "net.server";
- return net_close(L, mt);
- }
- // Lua: udpserver:on( "method", function(udpserver) )
- static int net_udpserver_on( lua_State* L )
- {
- const char *mt = "net.server";
- return net_on(L, mt);
- }
- // Lua: udpserver:send(string, function() )
- static int net_udpserver_send( lua_State* L )
- {
- const char *mt = "net.server";
- return net_send(L, mt);;
- }
- // Lua: s = net.createConnection(type, function(conn))
- static int net_createConnection( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_create(L, mt);
- }
- // Lua: socket:delete()
- static int net_socket_delete( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_delete(L, mt);
- }
- // Lua: socket:connect( port, ip )
- static int net_socket_connect( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_start(L, mt);
- }
- // Lua: socket:close()
- static int net_socket_close( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_close(L, mt);
- }
- // Lua: socket:on( "method", function(socket) )
- static int net_socket_on( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_on(L, mt);
- }
- // Lua: socket:send( string, function() )
- static int net_socket_send( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_send(L, mt);
- }
- // Lua: socket:dns( string, function(ip) )
- static int net_socket_dns( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_dns(L, mt);
- }
- #if 0
- static int net_array_index( lua_State* L )
- {
- char** parray = luaL_checkudata(L, 1, "net.array");
- int index = luaL_checkint(L, 2);
- lua_pushnumber(L, (*parray)[index-1]);
- return 1;
- }
- static int net_array_newindex( lua_State* L )
- {
- char** parray = luaL_checkudata(L, 1, "net.array");
- int index = luaL_checkint(L, 2);
- int value = luaL_checkint(L, 3);
- (*parray)[index-1] = value;
- return 0;
- }
- // expose an array to lua, by storing it in a userdata with the array metatable
- static int expose_array(lua_State* L, char *array, unsigned short len) {
- char** parray = lua_newuserdata(L, len);
- *parray = array;
- luaL_getmetatable(L, "net.array");
- lua_setmetatable(L, -2);
- return 1;
- }
- #endif
- // Module function map
- #define MIN_OPT_LEVEL 2
- #include "lrodefs.h"
- static const LUA_REG_TYPE net_server_map[] =
- {
- { LSTRKEY( "listen" ), LFUNCVAL ( net_server_listen ) },
- { LSTRKEY( "close" ), LFUNCVAL ( net_server_close ) },
- { LSTRKEY( "on" ), LFUNCVAL ( net_udpserver_on ) },
- { LSTRKEY( "send" ), LFUNCVAL ( net_udpserver_send ) },
- // { LSTRKEY( "delete" ), LFUNCVAL ( net_server_delete ) },
- { LSTRKEY( "__gc" ), LFUNCVAL ( net_server_delete ) },
- #if LUA_OPTIMIZE_MEMORY > 0
- { LSTRKEY( "__index" ), LROVAL ( net_server_map ) },
- #endif
- { LNILKEY, LNILVAL }
- };
- static const LUA_REG_TYPE net_socket_map[] =
- {
- { LSTRKEY( "connect" ), LFUNCVAL( net_socket_connect ) },
- { LSTRKEY( "close" ), LFUNCVAL ( net_socket_close ) },
- { LSTRKEY( "on" ), LFUNCVAL ( net_socket_on ) },
- { LSTRKEY( "send" ), LFUNCVAL ( net_socket_send ) },
- { LSTRKEY( "dns" ), LFUNCVAL ( net_socket_dns ) },
- // { LSTRKEY( "delete" ), LFUNCVAL ( net_socket_delete ) },
- { LSTRKEY( "__gc" ), LFUNCVAL ( net_socket_delete ) },
- #if LUA_OPTIMIZE_MEMORY > 0
- { LSTRKEY( "__index" ), LROVAL ( net_socket_map ) },
- #endif
- { LNILKEY, LNILVAL }
- };
- #if 0
- static const LUA_REG_TYPE net_array_map[] =
- {
- { LSTRKEY( "__index" ), LFUNCVAL( net_array_index ) },
- { LSTRKEY( "__newindex" ), LFUNCVAL( net_array_newindex ) },
- { LNILKEY, LNILVAL }
- };
- #endif
- const LUA_REG_TYPE net_map[] =
- {
- { LSTRKEY( "createServer" ), LFUNCVAL ( net_createServer ) },
- { LSTRKEY( "createConnection" ), LFUNCVAL ( net_createConnection ) },
- #if LUA_OPTIMIZE_MEMORY > 0
- { LSTRKEY( "TCP" ), LNUMVAL( TCP ) },
- { LSTRKEY( "UDP" ), LNUMVAL( UDP ) },
- { LSTRKEY( "__metatable" ), LROVAL( net_map ) },
- #endif
- { LNILKEY, LNILVAL }
- };
- LUALIB_API int luaopen_net( lua_State *L )
- {
- int i;
- for(i=0;i<MAX_SOCKET;i++)
- {
- socket[i] = LUA_NOREF;
- }
- #if LUA_OPTIMIZE_MEMORY > 0
- luaL_rometatable(L, "net.server", (void *)net_server_map); // create metatable for net.server
- luaL_rometatable(L, "net.socket", (void *)net_socket_map); // create metatable for net.socket
- #if 0
- luaL_rometatable(L, "net.array", (void *)net_array_map); // create metatable for net.array
- #endif
- return 0;
- #else // #if LUA_OPTIMIZE_MEMORY > 0
- int n;
- luaL_register( L, AUXLIB_NET, net_map );
- // Set it as its own metatable
- lua_pushvalue( L, -1 );
- lua_setmetatable( L, -2 );
- // Module constants
- MOD_REG_NUMBER( L, "TCP", TCP );
- MOD_REG_NUMBER( L, "UDP", UDP );
-
- n = lua_gettop(L);
- // create metatable
- luaL_newmetatable(L, "net.server");
- // metatable.__index = metatable
- lua_pushliteral(L, "__index");
- lua_pushvalue(L,-2);
- lua_rawset(L,-3);
- // Setup the methods inside metatable
- luaL_register( L, NULL, net_server_map );
- lua_settop(L, n);
- // create metatable
- luaL_newmetatable(L, "net.socket");
- // metatable.__index = metatable
- lua_pushliteral(L, "__index");
- lua_pushvalue(L,-2);
- lua_rawset(L,-3);
- // Setup the methods inside metatable
- luaL_register( L, NULL, net_socket_map );
- #if 0
- lua_settop(L, n);
- // create metatable
- luaL_newmetatable(L, "net.array");
- // Setup the methods inside metatable
- luaL_register( L, NULL, net_array_map );
- #endif
- return 1;
- #endif // #if LUA_OPTIMIZE_MEMORY > 0
- }
|