123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785 |
- // Module for network
- #include "module.h"
- #include "lauxlib.h"
- #include "platform.h"
- #include "lmem.h"
- #include "c_string.h"
- #include "c_stdlib.h"
- #include "c_types.h"
- #include "mem.h"
- #include "lwip/ip_addr.h"
- #include "espconn.h"
- #include "lwip/dns.h"
- #define TCP ESPCONN_TCP
- #define UDP ESPCONN_UDP
- static ip_addr_t host_ip; // for dns
- #ifdef HAVE_SSL_SERVER_CRT
- #include HAVE_SSL_SERVER_CRT
- #else
- __attribute__((section(".servercert.flash"))) unsigned char net_server_cert_area[INTERNAL_FLASH_SECTOR_SIZE];
- #endif
- __attribute__((section(".clientcert.flash"))) unsigned char net_client_cert_area[INTERNAL_FLASH_SECTOR_SIZE];
- #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(nud->self_ref == LUA_NOREF){
- NODE_DBG("self_ref null.\n");
- return;
- }
- /* original
- if(ipaddr == NULL)
- {
- NODE_ERR( "DNS Fail!\n" );
- goto end;
- }
- // ipaddr->addr is a uint32_t ip
- char ip_str[20];
- c_memset(ip_str, 0, sizeof(ip_str));
- if(ipaddr->addr != 0)
- {
- c_sprintf(ip_str, IPSTR, IP2STR(&(ipaddr->addr)));
- }
- 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
- */
- // "enhanced"
- 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
- if(ipaddr == NULL)
- {
- NODE_DBG( "DNS Fail!\n" );
- lua_pushnil(gL);
- }else{
- // ipaddr->addr is a uint32_t ip
- char ip_str[20];
- c_memset(ip_str, 0, sizeof(ip_str));
- if(ipaddr->addr != 0)
- {
- c_sprintf(ip_str, IPSTR, IP2STR(&(ipaddr->addr)));
- }
- lua_pushstring(gL, ip_str); // the ip para
- }
- // "enhanced" end
- lua_call(gL, 2, 0);
- end:
- 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_set_size(ESPCONN_CLIENT, 5120); /* set SSL buffer size */
- 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 int 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;
- }
- lnet_userdata *nud = (lnet_userdata *)pesp_conn->reverse;
- if(nud == NULL)
- return;
- if(gL == NULL)
- return;
- if(ipaddr == NULL)
- {
- dns_reconn_count++;
- if( dns_reconn_count >= 5 ){
- NODE_ERR( "DNS Fail!\n" );
- 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);
- 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;
- }
- if(nud->pesp_conn == NULL){
- NODE_DBG("nud->pesp_conn is NULL.\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);
- }
- // SDK 1.4.0 changed behaviour, for UDP server need to look up remote ip/port
- if (isserver && pesp_conn->type == ESPCONN_UDP)
- {
- remot_info *pr = 0;
- if (espconn_get_connection_info (pesp_conn, &pr, 0) != ESPCONN_OK)
- return luaL_error (L, "remote ip/port unavailable");
- pesp_conn->proto.udp->remote_port = pr->remote_port;
- os_memmove (pesp_conn->proto.udp->remote_ip, pr->remote_ip, 4);
- // The remot_info apparently should *not* be os_free()d, fyi
- }
- #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;
- }
- if(nud->pesp_conn == NULL){
- NODE_DBG("nud->pesp_conn is NULL.\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;
- if(ESPCONN_OK == espconn_gethostbyname(pesp_conn, domain, &host_ip, net_dns_found))
- net_dns_found(domain, &host_ip, pesp_conn); // ip is returned in host_ip.
- return 0;
- }
- // Lua: net.dns.resolve( domain, function(ip) )
- static int net_dns_static( lua_State* L )
- {
- const char *mt = "net.socket";
- if (!lua_isstring( L, 1 ))
- return luaL_error( L, "wrong parameter type (domain)" );
-
- int rfunc = LUA_NOREF; //save reference to func
- if (lua_type(L, 2) == LUA_TFUNCTION || lua_type(L, 2) == LUA_TLIGHTFUNCTION){
- rfunc = luaL_ref(L, LUA_REGISTRYINDEX);
- }
- int rdom = luaL_ref(L, LUA_REGISTRYINDEX); //save reference to domain
- lua_settop(L,0); //empty stack
- lua_getfield(L, LUA_GLOBALSINDEX, "net");
- lua_getfield(L, -1, "createConnection");
- lua_remove(L, -2); //remove "net" from stack
- lua_pushinteger(L, UDP); // we are going to create a new dummy UDP socket
- lua_call(L,1,1);// after this the stack should have a socket
- lua_rawgeti(gL, LUA_REGISTRYINDEX, rdom); // load domain back to the stack
- lua_rawgeti(gL, LUA_REGISTRYINDEX, rfunc); // load the callback function back to the stack
- luaL_unref(L, LUA_REGISTRYINDEX, rdom); //free reference
- luaL_unref(L, LUA_REGISTRYINDEX, rfunc); //free reference
- 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;
- 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;
- if(ESPCONN_OK == espconn_gethostbyname(pesp_conn, domain, &host_ip, net_dns_found))
- net_dns_found(domain, &host_ip, pesp_conn); // ip is returned in host_ip.
- 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);
- }
- static int net_socket_hold( lua_State* L )
- {
- const char *mt = "net.socket";
- 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;
- espconn_recv_hold(pesp_conn);
- return 0;
- }
- static int net_socket_unhold( lua_State* L )
- {
- const char *mt = "net.socket";
- 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;
- espconn_recv_unhold(pesp_conn);
- return 0;
- }
- // Lua: ip,port = sk:getpeer()
- static int net_socket_getpeer( lua_State* L )
- {
- lnet_userdata *nud;
- const char *mt = "net.socket";
- nud = (lnet_userdata *)luaL_checkudata(L, 1, mt);
- luaL_argcheck(L, nud, 1, "Server/Socket expected");
- if(nud!=NULL && nud->pesp_conn!=NULL ){
- char temp[20] = {0};
- c_sprintf(temp, IPSTR, IP2STR( &(nud->pesp_conn->proto.tcp->remote_ip) ) );
- if ( nud->pesp_conn->proto.tcp->remote_port != 0 ) {
- lua_pushstring( L, temp );
- lua_pushinteger( L, nud->pesp_conn->proto.tcp->remote_port );
- } else {
- lua_pushnil( L );
- lua_pushnil( L );
- }
- } else {
- lua_pushnil( L );
- lua_pushnil( L );
- }
- return 2;
- }
- // Lua: socket:dns( string, function(ip) )
- static int net_socket_dns( lua_State* L )
- {
- const char *mt = "net.socket";
- return net_dns(L, mt);
- }
- static int net_multicastJoinLeave( lua_State *L, int join)
- {
- size_t il;
- ip_addr_t multicast_addr;
- ip_addr_t if_addr;
- const char *multicast_ip;
- const char *if_ip;
- NODE_DBG("net_multicastJoin is called.\n");
- if(! lua_isstring(L,1) ) return luaL_error( L, "wrong arg type" );
- if_ip = luaL_checklstring( L, 1, &il );
- if (if_ip != NULL)
- if ( if_ip[0] == '\0' || stricmp(if_ip,"any") == 0)
- {
- if_ip = "0.0.0.0";
- il = 7;
- }
- if (if_ip == NULL || il > 15 || il < 7) return luaL_error( L, "invalid if ip" );
- if_addr.addr = ipaddr_addr(if_ip);
- if(! lua_isstring(L,2) ) return luaL_error( L, "wrong arg type" );
- multicast_ip = luaL_checklstring( L, 2, &il );
- if (multicast_ip == NULL || il > 15 || il < 7) return luaL_error( L, "invalid multicast ip" );
- multicast_addr.addr = ipaddr_addr(multicast_ip);
- if (join)
- {
- espconn_igmp_join(&if_addr, &multicast_addr);
- }
- else
- {
- espconn_igmp_leave(&if_addr, &multicast_addr);
- }
- return 0;
- }
- // Lua: net.multicastJoin(ifip, multicastip)
- // if ifip "" or "any" all interfaces are affected
- static int net_multicastJoin( lua_State* L )
- {
- return net_multicastJoinLeave(L,1);
- }
- // Lua: net.multicastLeave(ifip, multicastip)
- // if ifip "" or "any" all interfaces are affected
- static int net_multicastLeave( lua_State* L )
- {
- return net_multicastJoinLeave(L,0);
- }
- // Returns NULL on success, error message otherwise
- static const char *append_pem_blob(const char *pem, const char *type, uint8_t **buffer_p, uint8_t *buffer_limit, const char *name) {
- char unb64[256];
- memset(unb64, 0xff, sizeof(unb64));
- int i;
- for (i = 0; i < 64; i++) {
- unb64["ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"[i]] = i;
- }
- if (!pem) {
- return "No PEM blob";
- }
- // Scan for -----BEGIN CERT
- pem = strstr(pem, "-----BEGIN ");
- if (!pem) {
- return "No PEM header";
- }
- if (strncmp(pem + 11, type, strlen(type))) {
- return "Wrong PEM type";
- }
- pem = strchr(pem, '\n');
- if (!pem) {
- return "Incorrect PEM format";
- }
- //
- // Base64 encoded data starts here
- // Get all the base64 data into a single buffer....
- // We will use the back end of the buffer....
- //
- uint8_t *buffer = *buffer_p;
- uint8_t *dest = buffer + 32 + 2; // Leave space for name and length
- int bitcount = 0;
- int accumulator = 0;
- for (; *pem && dest < buffer_limit; pem++) {
- int val = unb64[*(uint8_t*) pem];
- if (val & 0xC0) {
- // not a base64 character
- if (isspace(*(uint8_t*) pem)) {
- continue;
- }
- if (*pem == '=') {
- // just ignore -- at the end
- bitcount = 0;
- continue;
- }
- if (*pem == '-') {
- break;
- }
- return "Invalid character in PEM";
- } else {
- bitcount += 6;
- accumulator = (accumulator << 6) + val;
- if (bitcount >= 8) {
- bitcount -= 8;
- *dest++ = accumulator >> bitcount;
- }
- }
- }
- if (dest >= buffer_limit || strncmp(pem, "-----END ", 9) || strncmp(pem + 9, type, strlen(type)) || bitcount) {
- return "Invalid PEM format data";
- }
- size_t len = dest - (buffer + 32 + 2);
- memset(buffer, 0, 32);
- strcpy(buffer, name);
- buffer[32] = len & 0xff;
- buffer[33] = (len >> 8) & 0xff;
- *buffer_p = dest;
- return NULL;
- }
- static const char *fill_page_with_pem(lua_State *L, const unsigned char *flash_memory, int flash_offset, const char **types, const char **names)
- {
- uint8_t *buffer = luaM_malloc(L, INTERNAL_FLASH_SECTOR_SIZE);
- uint8_t *buffer_base = buffer;
- uint8_t *buffer_limit = buffer + INTERNAL_FLASH_SECTOR_SIZE;
- int argno;
- for (argno = 1; argno <= lua_gettop(L) && types[argno - 1]; argno++) {
- const char *pem = lua_tostring(L, argno);
- const char *error = append_pem_blob(pem, types[argno - 1], &buffer, buffer_limit, names[argno - 1]);
- if (error) {
- luaM_free(L, buffer_base);
- return error;
- }
- }
- memset(buffer, 0xff, buffer_limit - buffer);
- // Lets see if it matches what is already there....
- if (c_memcmp(buffer_base, flash_memory, INTERNAL_FLASH_SECTOR_SIZE) != 0) {
- // Starts being dangerous
- if (platform_flash_erase_sector(flash_offset / INTERNAL_FLASH_SECTOR_SIZE) != PLATFORM_OK) {
- luaM_free(L, buffer_base);
- return "Failed to erase sector";
- }
- if (platform_s_flash_write(buffer_base, flash_offset, INTERNAL_FLASH_SECTOR_SIZE) != INTERNAL_FLASH_SECTOR_SIZE) {
- luaM_free(L, buffer_base);
- return "Failed to write sector";
- }
- // ends being dangerous
- }
- luaM_free(L, buffer_base);
- return NULL;
- }
- // Lua: net.cert.auth(true / false | PEM data [, PEM data] )
- static int net_cert_auth(lua_State *L)
- {
- int enable;
- uint32_t flash_offset = platform_flash_mapped2phys((uint32_t) &net_client_cert_area[0]);
- if ((flash_offset & 0xfff) || flash_offset > 0xff000 || INTERNAL_FLASH_SECTOR_SIZE != 0x1000) {
- // THis should never happen
- return luaL_error( L, "bad offset" );
- }
- if (lua_type(L, 1) == LUA_TSTRING) {
- const char *types[3] = { "CERTIFICATE", "RSA PRIVATE KEY", NULL };
- const char *names[2] = { "certificate", "private_key" };
- const char *error = fill_page_with_pem(L, &net_client_cert_area[0], flash_offset, types, names);
- if (error) {
- return luaL_error(L, error);
- }
- enable = 1;
- } else {
- enable = lua_toboolean(L, 1);
- }
- bool rc;
- if (enable) {
- // See if there is a cert there
- if (net_client_cert_area[0] == 0x00 || net_client_cert_area[0] == 0xff) {
- return luaL_error( L, "no certificates found" );
- }
- rc = espconn_secure_cert_req_enable(1, flash_offset / INTERNAL_FLASH_SECTOR_SIZE);
- } else {
- rc = espconn_secure_cert_req_disable(1);
- }
- lua_pushboolean(L, rc);
- return 1;
- }
- // Lua: net.cert.verify(true / false | PEM data [, PEM data] )
- static int net_cert_verify(lua_State *L)
- {
- int enable;
- uint32_t flash_offset = platform_flash_mapped2phys((uint32_t) &net_server_cert_area[0]);
- if ((flash_offset & 0xfff) || flash_offset > 0xff000 || INTERNAL_FLASH_SECTOR_SIZE != 0x1000) {
- // THis should never happen
- return luaL_error( L, "bad offset" );
- }
- if (lua_type(L, 1) == LUA_TSTRING) {
- const char *types[2] = { "CERTIFICATE", NULL };
- const char *names[1] = { "certificate" };
- const char *error = fill_page_with_pem(L, &net_server_cert_area[0], flash_offset, types, names);
- if (error) {
- return luaL_error(L, error);
- }
- enable = 1;
- } else {
- enable = lua_toboolean(L, 1);
- }
- bool rc;
- if (enable) {
- // See if there is a cert there
- if (net_server_cert_area[0] == 0x00 || net_server_cert_area[0] == 0xff) {
- return luaL_error( L, "no certificates found" );
- }
- rc = espconn_secure_ca_enable(1, flash_offset / INTERNAL_FLASH_SECTOR_SIZE);
- } else {
- rc = espconn_secure_ca_disable(1);
- }
- lua_pushboolean(L, rc);
- return 1;
- }
- // Lua: s = net.dns.setdnsserver(ip_addr, [index])
- static int net_setdnsserver( lua_State* L )
- {
- size_t l;
- u32_t ip32;
- const char *server = luaL_checklstring( L, 1, &l );
- if (l>16 || server == NULL || (ip32 = ipaddr_addr(server)) == IPADDR_NONE || ip32 == IPADDR_ANY)
- return luaL_error( L, "invalid dns server ip" );
- int numdns = luaL_optint(L, 2, 0);
- if (numdns >= DNS_MAX_SERVERS)
- return luaL_error( L, "server index out of range [0-%d]", DNS_MAX_SERVERS - 1);
- ip_addr_t ipaddr;
- ip4_addr_set_u32(&ipaddr, ip32);
- dns_setserver(numdns,&ipaddr);
- return 0;
- }
- // Lua: s = net.dns.getdnsserver([index])
- static int net_getdnsserver( lua_State* L )
- {
- int numdns = luaL_optint(L, 1, 0);
- if (numdns >= DNS_MAX_SERVERS)
- return luaL_error( L, "server index out of range [0-%d]", DNS_MAX_SERVERS - 1);
- // ip_addr_t ipaddr;
- // dns_getserver(numdns,&ipaddr);
- // Bug fix by @md5crypt https://github.com/nodemcu/nodemcu-firmware/pull/500
- ip_addr_t ipaddr = dns_getserver(numdns);
- if ( ip_addr_isany(&ipaddr) ) {
- lua_pushnil( L );
- } else {
- char temp[20] = {0};
- c_sprintf(temp, IPSTR, IP2STR( &ipaddr.addr ) );
- lua_pushstring( L, temp );
- }
- return 1;
- }
- #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
- 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 ) },
- { LSTRKEY( "__index" ), LROVAL( net_server_map ) },
- { 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( "hold" ), LFUNCVAL( net_socket_hold ) },
- { LSTRKEY( "unhold" ), LFUNCVAL( net_socket_unhold ) },
- { LSTRKEY( "dns" ), LFUNCVAL( net_socket_dns ) },
- { LSTRKEY( "getpeer" ), LFUNCVAL( net_socket_getpeer ) },
- //{ LSTRKEY( "delete" ), LFUNCVAL( net_socket_delete ) },
- { LSTRKEY( "__gc" ), LFUNCVAL( net_socket_delete ) },
- { LSTRKEY( "__index" ), LROVAL( net_socket_map ) },
- { 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
- static const LUA_REG_TYPE net_cert_map[] = {
- { LSTRKEY( "verify" ), LFUNCVAL( net_cert_verify ) },
- #ifdef CLIENT_SSL_CERT_AUTH_ENABLE
- { LSTRKEY( "auth" ), LFUNCVAL( net_cert_auth ) },
- #endif
- { LNILKEY, LNILVAL }
- };
- static const LUA_REG_TYPE net_dns_map[] = {
- { LSTRKEY( "setdnsserver" ), LFUNCVAL( net_setdnsserver ) },
- { LSTRKEY( "getdnsserver" ), LFUNCVAL( net_getdnsserver ) },
- { LSTRKEY( "resolve" ), LFUNCVAL( net_dns_static ) },
- { LNILKEY, LNILVAL }
- };
- static const LUA_REG_TYPE net_map[] = {
- { LSTRKEY( "createServer" ), LFUNCVAL( net_createServer ) },
- { LSTRKEY( "createConnection" ), LFUNCVAL( net_createConnection ) },
- { LSTRKEY( "multicastJoin"), LFUNCVAL( net_multicastJoin ) },
- { LSTRKEY( "multicastLeave"), LFUNCVAL( net_multicastLeave ) },
- { LSTRKEY( "dns" ), LROVAL( net_dns_map ) },
- #ifdef CLIENT_SSL_ENABLE
- { LSTRKEY( "cert" ), LROVAL(net_cert_map) },
- #endif
- { LSTRKEY( "TCP" ), LNUMVAL( TCP ) },
- { LSTRKEY( "UDP" ), LNUMVAL( UDP ) },
- { LSTRKEY( "__metatable" ), LROVAL( net_map ) },
- { LNILKEY, LNILVAL }
- };
- int luaopen_net( lua_State *L ) {
- int i;
- for(i=0;i<MAX_SOCKET;i++)
- {
- socket[i] = LUA_NOREF;
- }
- 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;
- }
- NODEMCU_MODULE(NET, "net", net_map, luaopen_net);
|