luat_lib_socket.c 43 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391
  1. /*
  2. @module socket
  3. @summary 网络接口
  4. @version 1.0
  5. @date 2022.11.13
  6. @demo socket
  7. @tag LUAT_USE_NETWORK
  8. */
  9. #include "luat_base.h"
  10. #include "luat_mem.h"
  11. // #ifdef LUAT_USE_NETWORK
  12. #include "luat_network_adapter.h"
  13. #include "luat_rtos.h"
  14. #include "luat_zbuff.h"
  15. #define LUAT_LOG_TAG "socket"
  16. #include "luat_log.h"
  17. //static const char NW_TYPE[] = "NWA*";
  18. #define LUAT_NW_CTRL_TYPE "NWCTRL*"
  19. typedef struct
  20. {
  21. network_ctrl_t *netc;
  22. int cb_ref; //回调函数
  23. char *task_name;
  24. uint8_t adapter_index;
  25. }luat_socket_ctrl_t;
  26. #define L_CTRL_CHECK do {if (!l_ctrl || !l_ctrl->netc){return 0;}}while(0)
  27. network_adapter_info* network_adapter_fetch(int id, void** userdata);
  28. /*
  29. 获取本地ip
  30. @api socket.localIP(adapter)
  31. @int 适配器序号, 只能是socket.ETH0(外置以太网),socket.LWIP_ETH(内置以太网),socket.LWIP_STA(内置WIFI的STA),socket.LWIP_AP(内置WIFI的AP),socket.LWIP_GP(内置蜂窝网络的GPRS),socket.USB(外置USB网卡),如果不填,优先选择soc平台自带能上外网的适配器,若仍然没有,选择最后一个注册的适配器
  32. @return string 通常是内网ip, 也可能是外网ip, 取决于运营商的分配
  33. @return string 网络掩码
  34. @return string 网关IP
  35. @usage
  36. sys.taskInit(function()
  37. while 1 do
  38. sys.wait(3000)
  39. log.info("socket", "ip", socket.localIP())
  40. -- 输出示例
  41. -- 62.39.244.10 255.255.255.255 0.0.0.0
  42. end
  43. end)
  44. */
  45. static int l_socket_local_ip(lua_State *L)
  46. {
  47. luat_ip_addr_t local_ip, net_mask, gate_way, ipv6;
  48. int adapter_index = luaL_optinteger(L, 1, network_register_get_default());
  49. if (adapter_index < 0 || adapter_index >= NW_ADAPTER_QTY)
  50. {
  51. return 0;
  52. }
  53. #ifdef LUAT_USE_LWIP
  54. network_set_ip_invaild(&ipv6);
  55. int ret = network_get_full_local_ip_info(NULL, adapter_index, &local_ip, &net_mask, &gate_way, &ipv6);
  56. #else
  57. void* userdata = NULL;
  58. network_adapter_info* info = network_adapter_fetch(adapter_index, &userdata);
  59. if (info == NULL)
  60. return 0;
  61. int ret = info->get_local_ip_info(&local_ip, &net_mask, &gate_way, userdata);
  62. #endif
  63. if (ret == 0) {
  64. #ifdef LUAT_USE_LWIP
  65. lua_pushfstring(L, "%s", ipaddr_ntoa(&local_ip));
  66. lua_pushfstring(L, "%s", ipaddr_ntoa(&net_mask));
  67. lua_pushfstring(L, "%s", ipaddr_ntoa(&gate_way));
  68. #if LWIP_IPV6
  69. if (IPADDR_TYPE_V6 == ipv6.type)
  70. {
  71. char *ipv6_string = ip6addr_ntoa(&ipv6.u_addr.ip6);
  72. lua_pushfstring(L, "%s", ipv6_string);
  73. }
  74. else
  75. #endif
  76. {
  77. lua_pushnil(L);
  78. }
  79. return 4;
  80. #else
  81. lua_pushfstring(L, "%d.%d.%d.%d", (local_ip.ipv4 >> 0) & 0xFF, (local_ip.ipv4 >> 8) & 0xFF, (local_ip.ipv4 >> 16) & 0xFF, (local_ip.ipv4 >> 24) & 0xFF);
  82. lua_pushfstring(L, "%d.%d.%d.%d", (net_mask.ipv4 >> 0) & 0xFF, (net_mask.ipv4 >> 8) & 0xFF, (net_mask.ipv4 >> 16) & 0xFF, (net_mask.ipv4 >> 24) & 0xFF);
  83. lua_pushfstring(L, "%d.%d.%d.%d", (gate_way.ipv4 >> 0) & 0xFF, (gate_way.ipv4 >> 8) & 0xFF, (gate_way.ipv4 >> 16) & 0xFF, (gate_way.ipv4 >> 24) & 0xFF);
  84. return 3;
  85. #endif
  86. }
  87. return 0;
  88. }
  89. static int32_t l_socket_callback(lua_State *L, void* ptr)
  90. {
  91. rtos_msg_t* msg = (rtos_msg_t*)lua_topointer(L, -1);
  92. luat_socket_ctrl_t *l_ctrl =(luat_socket_ctrl_t *)msg->ptr;
  93. if (l_ctrl->netc)
  94. {
  95. if (l_ctrl->cb_ref)
  96. {
  97. lua_geti(L, LUA_REGISTRYINDEX, l_ctrl->cb_ref);
  98. if (lua_isfunction(L, -1)) {
  99. lua_pushlightuserdata(L, l_ctrl);
  100. lua_pushinteger(L, msg->arg1);
  101. lua_pushinteger(L, msg->arg2);
  102. lua_call(L, 3, 0);
  103. }
  104. }
  105. else if (l_ctrl->task_name)
  106. {
  107. lua_getglobal(L, "sys_send");
  108. if (lua_isfunction(L, -1)) {
  109. lua_pushstring(L, l_ctrl->task_name);
  110. lua_pushinteger(L, msg->arg1);
  111. lua_pushinteger(L, msg->arg2);
  112. lua_call(L, 3, 0);
  113. }
  114. }
  115. else
  116. {
  117. lua_getglobal(L, "sys_pub");
  118. if (lua_isfunction(L, -1)) {
  119. lua_pushstring(L, LUAT_NW_CTRL_TYPE);
  120. lua_pushinteger(L, l_ctrl->netc->adapter_index);
  121. lua_pushinteger(L, l_ctrl->netc->socket_id);
  122. lua_pushinteger(L, msg->arg1);
  123. lua_pushinteger(L, msg->arg2);
  124. lua_call(L, 5, 0);
  125. }
  126. }
  127. }
  128. lua_pushinteger(L, 0);
  129. return 1;
  130. }
  131. static int32_t luat_lib_socket_callback(void *data, void *param)
  132. {
  133. OS_EVENT *event = (OS_EVENT *)data;
  134. rtos_msg_t msg;
  135. msg.handler = l_socket_callback;
  136. msg.ptr = param;
  137. msg.arg1 = event->ID & 0x0fffffff;
  138. msg.arg2 = event->Param1;
  139. luat_msgbus_put(&msg, 0);
  140. return 0;
  141. }
  142. static luat_socket_ctrl_t * l_get_ctrl(lua_State *L, int index)
  143. {
  144. if (luaL_testudata(L, 1, LUAT_NW_CTRL_TYPE))
  145. {
  146. return ((luat_socket_ctrl_t *)luaL_checkudata(L, 1, LUAT_NW_CTRL_TYPE));
  147. }
  148. else
  149. {
  150. return ((luat_socket_ctrl_t *)lua_touserdata(L, 1));
  151. }
  152. }
  153. // __gc
  154. static int l_socket_gc(lua_State *L)
  155. {
  156. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  157. if (!l_ctrl){return 0;}
  158. if (l_ctrl->netc)
  159. {
  160. network_force_close_socket(l_ctrl->netc);
  161. network_release_ctrl(l_ctrl->netc);
  162. l_ctrl->netc = NULL;
  163. }
  164. if (l_ctrl->cb_ref)
  165. {
  166. luaL_unref(L, LUA_REGISTRYINDEX, l_ctrl->cb_ref);
  167. l_ctrl->cb_ref = 0;
  168. }
  169. if (l_ctrl->task_name)
  170. {
  171. luat_heap_free(l_ctrl->task_name);
  172. l_ctrl->task_name = 0;
  173. }
  174. return 0;
  175. }
  176. /*
  177. 在指定网卡上申请一个socket_ctrl
  178. @api socket.create(adapter, cb)
  179. @int 适配器序号,默认是平台自带的能上外网的适配器,通过socket.dft()可以获取和修改
  180. @string/function string为消息通知的taskName,function则为回调函数
  181. @return userdata 成功返回network_ctrl,失败返回nil
  182. @usage
  183. --以太网网卡上申请一个network_ctrl,通过socket_cb_fun回调相关消息
  184. local netc = socket.create(socket.LWIP_GP, socket_cb_fun)
  185. --以太网网卡上申请一个network_ctrl,通过sendMsg方式通知taskName为"IOT_TASK"回调相关消息
  186. local netc = socket.create(socket.LWIP_GP, "IOT_TASK")
  187. -- 在默认网络适配器上创建一个network_ctrl
  188. local netc = socket.create(nil, "MySocket")
  189. --[[
  190. 当通过回调函数回调消息时,输入给function一共3个参数:
  191. param1为申请的network_ctrl
  192. param2为具体的消息,只能是socket.LINK, socket.ON_LINE, socket.TX_OK, socket.EVENT, socket.CLOSED等等
  193. param3为消息对应的参数,目前只有0和-1,0表示成功或者可能有新数据(具体消息为socket.EVENT),-1表示失败或者有异常,需要断开重连
  194. ]]
  195. --[[
  196. Air8000内核固件运行起来之后,在脚本调用socket.dft(id)之前,默认适配器编号为socket.LWIP_GP,最后一个注册的适配器编号为socket.LWIP_AP;
  197. Air780xx内核固件运行起来之后,在脚本调用socket.dft(id)之前,默认适配器编号为socket.LWIP_GP,最后一个注册的适配器编号为socket.LWIP_GP;
  198. Air8101内核固件运行起来之后,在脚本调用socket.dft(id)之前,默认适配器编号为socket.LWIP_STA,最后一个注册的适配器编号为socket.LWIP_STA;
  199. ]]
  200. */
  201. static int l_socket_create(lua_State *L)
  202. {
  203. int adapter_index = luaL_optinteger(L, 1, network_register_get_default());
  204. if (adapter_index < 0 || adapter_index >= NW_ADAPTER_QTY)
  205. {
  206. lua_pushnil(L);
  207. return 1;
  208. }
  209. luat_socket_ctrl_t *l_ctrl = (luat_socket_ctrl_t *)lua_newuserdata(L, sizeof(luat_socket_ctrl_t));
  210. if (!l_ctrl)
  211. {
  212. lua_pushnil(L);
  213. return 1;
  214. }
  215. l_ctrl->adapter_index = adapter_index;
  216. l_ctrl->netc = network_alloc_ctrl(adapter_index);
  217. if (!l_ctrl->netc)
  218. {
  219. LLOGD("create fail");
  220. lua_pushnil(L);
  221. return 1;
  222. }
  223. network_init_ctrl(l_ctrl->netc, NULL, luat_lib_socket_callback, l_ctrl);
  224. if (lua_isfunction(L, 2))
  225. {
  226. lua_pushvalue(L, 2);
  227. l_ctrl->cb_ref = luaL_ref(L, LUA_REGISTRYINDEX);
  228. l_ctrl->task_name = NULL;
  229. }
  230. else if (lua_isstring(L, 2))
  231. {
  232. l_ctrl->cb_ref = 0;
  233. size_t len;
  234. const char *buf;
  235. buf = lua_tolstring(L, 2, &len);//取出字符串数据
  236. l_ctrl->task_name = luat_heap_malloc(len + 1);
  237. memset(l_ctrl->task_name, 0, len + 1);
  238. memcpy(l_ctrl->task_name, buf, len);
  239. }
  240. luaL_setmetatable(L, LUAT_NW_CTRL_TYPE);
  241. return 1;
  242. }
  243. /*
  244. 配置是否打开debug信息
  245. @api socket.debug(ctrl, onoff)
  246. @user_data socket.create得到的ctrl
  247. @boolean true 打开debug开关
  248. @return nil 无返回值
  249. @usage
  250. -- 打开调试信息,默认是关闭状态
  251. socket.debug(ctrl, true)
  252. */
  253. static int l_socket_set_debug(lua_State *L)
  254. {
  255. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  256. L_CTRL_CHECK;
  257. if (lua_isboolean(L, 2))
  258. {
  259. l_ctrl->netc->is_debug = lua_toboolean(L, 2);
  260. }
  261. return 0;
  262. }
  263. /*
  264. 配置network一些信息,
  265. @api socket.config(ctrl, local_port, is_udp, is_tls, keep_idle, keep_interval, keep_cnt, server_cert, client_cert, client_key, client_password)
  266. @user_data socket.create得到的ctrl
  267. @int 本地端口号,小端格式,如果不写,则自动分配一个,如果用户填了端口号则需要小于60000, 默认不写
  268. @boolean 是否是UDP,默认false
  269. @boolean 是否是加密传输,默认false
  270. @int tcp keep live模式下的idle时间(秒),如果留空则表示不启用,如果是不支持标准posix接口的网卡(比如W5500),则为心跳间隔
  271. @int tcp keep live模式下的探测间隔时间(秒)
  272. @int tcp keep live模式下的探测次数
  273. @string TCP模式下的服务器ca证书数据,UDP模式下的PSK,不需要加密传输写nil,后续参数也全部nil
  274. @string TCP模式下的客户端证书数据,UDP模式下的PSK-ID,TCP模式下如果不需要验证客户端证书时,忽略,一般不需要验证客户端证书
  275. @string TCP模式下的客户端私钥加密数据
  276. @string TCP模式下的客户端私钥口令数据
  277. @return boolean 成功返回true,失败返回false
  278. @usage
  279. --最普通的TCP传输
  280. socket.config(ctrl)
  281. --最普通的加密TCP传输,证书都不用验证的那种
  282. socket.config(ctrl, nil, nil ,true)
  283. */
  284. static int l_socket_config(lua_State *L)
  285. {
  286. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  287. L_CTRL_CHECK;
  288. uint8_t is_udp = 0;
  289. uint8_t is_tls = 0;
  290. int param_pos = 1;
  291. uint32_t keep_idle, keep_interval, keep_cnt;
  292. const char *server_cert = NULL;
  293. const char *client_cert = NULL;
  294. const char *client_key = NULL;
  295. const char *client_password = NULL;
  296. size_t server_cert_len, client_cert_len, client_key_len, client_password_len;
  297. uint16_t local_port = luaL_optinteger(L, ++param_pos, 0);
  298. if (lua_isboolean(L, ++param_pos))
  299. {
  300. is_udp = lua_toboolean(L, param_pos);
  301. }
  302. if (lua_isboolean(L, ++param_pos))
  303. {
  304. is_tls = lua_toboolean(L, param_pos);
  305. }
  306. #ifndef LUAT_USE_TLS
  307. if (is_tls){
  308. LLOGE("NOT SUPPORT TLS");
  309. lua_pushboolean(L, 0);
  310. return 1;
  311. }
  312. #endif
  313. keep_idle = luaL_optinteger(L, ++param_pos, 0);
  314. keep_interval = luaL_optinteger(L, ++param_pos, 0);
  315. keep_cnt = luaL_optinteger(L, ++param_pos, 0);
  316. if (lua_isstring(L, ++param_pos))
  317. {
  318. server_cert_len = 0;
  319. server_cert = luaL_checklstring(L, param_pos, &server_cert_len);
  320. }
  321. if (lua_isstring(L, ++param_pos))
  322. {
  323. client_cert_len = 0;
  324. client_cert = luaL_checklstring(L, param_pos, &client_cert_len);
  325. }
  326. if (lua_isstring(L, ++param_pos))
  327. {
  328. client_key_len = 0;
  329. client_key = luaL_checklstring(L, param_pos, &client_key_len);
  330. }
  331. if (lua_isstring(L, ++param_pos))
  332. {
  333. client_password_len = 0;
  334. client_password = luaL_checklstring(L, param_pos, &client_password_len);
  335. }
  336. network_set_base_mode(l_ctrl->netc, !is_udp, 10000, keep_idle, keep_idle, keep_interval, keep_cnt);
  337. network_set_local_port(l_ctrl->netc, local_port);
  338. if (is_tls)
  339. {
  340. network_init_tls(l_ctrl->netc, (server_cert || client_cert)?2:0);
  341. if (is_udp)
  342. {
  343. network_set_psk_info(l_ctrl->netc, (const unsigned char *)server_cert, server_cert_len, (const unsigned char *)client_key, client_key_len);
  344. }
  345. else
  346. {
  347. if (server_cert)
  348. {
  349. network_set_server_cert(l_ctrl->netc, (const unsigned char *)server_cert, server_cert_len + 1);
  350. }
  351. if (client_cert)
  352. {
  353. network_set_client_cert(l_ctrl->netc, (const unsigned char *)client_cert, client_cert_len + 1,
  354. (const unsigned char *)client_key, client_key_len + 1,
  355. (const unsigned char *)client_password, client_password_len + 1);
  356. }
  357. }
  358. }
  359. else
  360. {
  361. network_deinit_tls(l_ctrl->netc);
  362. }
  363. lua_pushboolean(L, 1);
  364. return 1;
  365. }
  366. /*
  367. 等待网卡linkup
  368. @api socket.linkup(ctrl)
  369. @user_data socket.create得到的ctrl
  370. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了
  371. @return boolean true已经linkup,false没有linkup,之后需要接收socket.LINK消息
  372. @usage
  373. -- 判断一下是否已经联网
  374. local succ, result = socket.linkup(ctrl)
  375. */
  376. static int l_socket_linkup(lua_State *L)
  377. {
  378. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  379. L_CTRL_CHECK;
  380. int result = network_wait_link_up(l_ctrl->netc, 0);
  381. lua_pushboolean(L, (result < 0)?0:1);
  382. lua_pushboolean(L, result == 0);
  383. return 2;
  384. }
  385. /*
  386. 作为客户端连接服务器
  387. @api socket.connect(ctrl, ip, remote_port, need_ipv6_dns)
  388. @user_data socket.create得到的ctrl
  389. @string or int ip或者域名,如果是IPV4,可以是大端格式的int值
  390. @int 服务器端口号,小端格式
  391. @boolean 域名解析是否要IPV6,true要,false不要,默认false不要,只有支持IPV6的协议栈才有效果
  392. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果有异常,后续要close
  393. @return boolean true已经connect,false没有connect,之后需要接收socket.ON_LINE消息
  394. @usage
  395. local succ, result = socket.connect(ctrl, "netlab.luatos.com", 40123)
  396. --[[
  397. 常见的连接失败的code值, 会在日志中显示
  398. -1 底层内存不足
  399. -3 超时
  400. -8 端口已经被占用
  401. -11 链接未建立
  402. -13 模块主动断开连接
  403. -14 服务器主动断开连接
  404. ]]
  405. */
  406. static int l_socket_connect(lua_State *L)
  407. {
  408. #ifdef LUAT_USE_LWIP
  409. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  410. L_CTRL_CHECK;
  411. luat_ip_addr_t ip_addr;
  412. const char *ip = NULL;
  413. size_t ip_len;
  414. network_set_ip_invaild(&ip_addr);
  415. if (lua_isinteger(L, 2))
  416. {
  417. network_set_ip_ipv4(&ip_addr, lua_tointeger(L, 2));
  418. ip = NULL;
  419. ip_len = 0;
  420. }
  421. else
  422. {
  423. ip_len = 0;
  424. ip = luaL_checklstring(L, 2, &ip_len);
  425. }
  426. uint16_t remote_port = luaL_checkinteger(L, 3);
  427. LLOGD("connect to %s,%d", ip, remote_port);
  428. if (!network_ip_is_vaild_ipv4(&ip_addr))
  429. {
  430. if (LUA_TBOOLEAN == lua_type(L, 4))
  431. {
  432. network_connect_ipv6_domain(l_ctrl->netc, lua_toboolean(L, 4));
  433. }
  434. else
  435. {
  436. network_connect_ipv6_domain(l_ctrl->netc, 0);
  437. }
  438. }
  439. int result = network_connect(l_ctrl->netc, ip, ip_len, (!network_ip_is_vaild_ipv4(&ip_addr))?NULL:&ip_addr, remote_port, 0);
  440. lua_pushboolean(L, (result < 0)?0:1);
  441. lua_pushboolean(L, result == 0);
  442. return 2;
  443. #else
  444. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  445. L_CTRL_CHECK;
  446. luat_ip_addr_t ip_addr;
  447. const char *ip = NULL;
  448. size_t ip_len;
  449. ip_addr.is_ipv6 = 0xff;
  450. if (lua_isinteger(L, 2))
  451. {
  452. ip_addr.is_ipv6 = 0;
  453. ip_addr.ipv4 = lua_tointeger(L, 2);
  454. ip = NULL;
  455. ip_len = 0;
  456. }
  457. else
  458. {
  459. ip_len = 0;
  460. ip = luaL_checklstring(L, 2, &ip_len);
  461. }
  462. uint16_t remote_port = luaL_checkinteger(L, 3);
  463. LLOGD("connect to %s,%d", ip, remote_port);
  464. int result = network_connect(l_ctrl->netc, ip, ip_len, ip_addr.is_ipv6?NULL:&ip_addr, remote_port, 0);
  465. lua_pushboolean(L, (result < 0)?0:1);
  466. lua_pushboolean(L, result == 0);
  467. return 2;
  468. #endif
  469. }
  470. /*
  471. 作为客户端断开连接
  472. @api socket.discon(ctrl)
  473. @user_data socket.create得到的ctrl
  474. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了
  475. @return boolean true已经断开,false没有断开,之后需要接收socket.CLOSED消息
  476. @usage
  477. local succ, result = socket.discon(ctrl)
  478. */
  479. static int l_socket_disconnect(lua_State *L)
  480. {
  481. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  482. L_CTRL_CHECK;
  483. int result = network_close(l_ctrl->netc, 0);
  484. lua_pushboolean(L, (result < 0)?0:1);
  485. lua_pushboolean(L, result == 0);
  486. return 2;
  487. }
  488. /*
  489. 强制关闭socket
  490. @api socket.close(ctrl)
  491. @user_data socket.create得到的ctrl
  492. @return nil 无返回值
  493. */
  494. static int l_socket_close(lua_State *L)
  495. {
  496. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  497. L_CTRL_CHECK;
  498. network_force_close_socket(l_ctrl->netc);
  499. return 0;
  500. }
  501. /*
  502. 发送数据给对端,UDP单次发送不要超过1460字节,否则很容易失败
  503. @api socket.tx(ctrl, data, ip, port, flag)
  504. @user_data socket.create得到的ctrl
  505. @string or user_data zbuff 要发送的数据
  506. @string or int 对端IP,如果是TCP应用则忽略,如果是UDP,如果留空则用connect时候的参数,如果是IPV4,可以是大端格式的int值
  507. @int 对端端口号,小端格式,如果是TCP应用则忽略,如果是UDP,如果留空则用connect时候的参数
  508. @int 发送参数,目前预留,不起作用
  509. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果false,后续要close
  510. @return boolean true缓冲区满了,false没有满,如果true,则需要等待一段时间或者等到socket.TX_OK消息后再尝试发送,同时忽略下一个返回值
  511. @return boolean true已经收到应答,false没有收到应答,之后需要接收socket.TX_OK消息, 也可以忽略继续发送,直到full==true
  512. @usage
  513. local succ, full, result = socket.tx(ctrl, "123456", "xxx.xxx.xxx.xxx", xxxx)
  514. */
  515. static int l_socket_tx(lua_State *L)
  516. {
  517. #ifdef LUAT_USE_LWIP
  518. char ip_buf[68] = {0};
  519. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  520. L_CTRL_CHECK;
  521. luat_ip_addr_t ip_addr = {0};
  522. luat_zbuff_t *buff = NULL;
  523. const char *ip = NULL;
  524. const char *data = NULL;
  525. size_t ip_len = 0, data_len = 0;
  526. network_set_ip_invaild(&ip_addr);
  527. if (lua_isstring(L, 2))
  528. {
  529. data_len = 0;
  530. data = luaL_checklstring(L, 2, &data_len);
  531. }
  532. else
  533. {
  534. buff = ((luat_zbuff_t *)luaL_checkudata(L, 2, LUAT_ZBUFF_TYPE));
  535. data = (const char*)buff->addr;
  536. data_len = buff->used;
  537. }
  538. if (lua_isinteger(L, 3))
  539. {
  540. network_set_ip_ipv4(&ip_addr, lua_tointeger(L, 3));
  541. }
  542. else if (lua_isstring(L, 3))
  543. {
  544. ip_len = 0;
  545. ip = luaL_checklstring(L, 3, &ip_len);
  546. memcpy(ip_buf, ip, ip_len);
  547. ip_buf[ip_len] = 0;
  548. ipaddr_aton(ip_buf, &ip_addr);
  549. }
  550. uint32_t tx_len;
  551. int result = network_tx(l_ctrl->netc, (const uint8_t *)data, data_len, luaL_optinteger(L, 5, 0), network_ip_is_vaild(&ip_addr)?&ip_addr:NULL, luaL_optinteger(L, 4, 0), &tx_len, 0);
  552. #else
  553. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  554. L_CTRL_CHECK;
  555. luat_ip_addr_t ip_addr = {0};
  556. luat_zbuff_t *buff = NULL;
  557. const char *ip = NULL;
  558. const char *data = NULL;
  559. size_t ip_len = 0, data_len = 0;
  560. ip_addr.is_ipv6 = 0xff;
  561. if (lua_isstring(L, 2))
  562. {
  563. data_len = 0;
  564. data = luaL_checklstring(L, 2, &data_len);
  565. }
  566. else
  567. {
  568. buff = ((luat_zbuff_t *)luaL_checkudata(L, 2, LUAT_ZBUFF_TYPE));
  569. data = (const char *)buff->addr;
  570. data_len = buff->used;
  571. }
  572. if (lua_isinteger(L, 3))
  573. {
  574. ip_addr.is_ipv6 = 0;
  575. ip_addr.ipv4 = lua_tointeger(L, 3);
  576. }
  577. else if (lua_isstring(L, 3))
  578. {
  579. ip_len = 0;
  580. ip = luaL_checklstring(L, 3, &ip_len);
  581. if (network_string_is_ipv4(ip, ip_len))
  582. {
  583. ip_addr.is_ipv6 = 0;
  584. ip_addr.ipv4 = network_string_to_ipv4(ip, ip_len);
  585. }
  586. else
  587. {
  588. char *name = luat_heap_malloc(ip_len + 1);
  589. memcpy(name, ip, ip_len);
  590. name[ip_len] = 0;
  591. network_string_to_ipv6(name, &ip_addr);
  592. free(name);
  593. }
  594. }
  595. uint32_t tx_len;
  596. int result = network_tx(l_ctrl->netc, (const uint8_t *)data, data_len, luaL_optinteger(L, 5, 0), (ip_addr.is_ipv6 != 0xff)?&ip_addr:NULL, luaL_optinteger(L, 4, 0), &tx_len, 0);
  597. #endif
  598. lua_pushboolean(L, (result < 0)?0:1);
  599. lua_pushboolean(L, tx_len != data_len);
  600. lua_pushboolean(L, result == 0);
  601. return 3;
  602. }
  603. /*
  604. 接收对端发出的数据,注意数据已经缓存在底层,使用本函数只是提取出来,UDP模式下一次只会取出一个数据包
  605. @api socket.rx(ctrl, buff, flag, limit)
  606. @user_data socket.create得到的ctrl
  607. @user_data zbuff 存放接收的数据,如果缓冲区不够大会自动扩容
  608. @int 接收参数,目前预留,不起作用
  609. @int 接收数据长度限制,如果指定了,则只取前N个字节. 2024.1.5 新增
  610. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果false,后续要close
  611. @return int 本次接收到数据长度
  612. @return string 对端IP,只有UDP模式下才有意义,TCP模式返回nil,注意返回的格式,如果是IPV4,1byte 0x00 + 4byte地址 如果是IPV6,1byte 0x01 + 16byte地址
  613. @return int 对端port,只有UDP模式下才有意义,TCP模式返回0
  614. @usage
  615. -- 从socket中读取数据, ctrl是socket.create返回的, 请查阅demo/socket
  616. local buff = zbuff.create(2048)
  617. local succ, data_len, remote_ip, remote_port = socket.rx(ctrl, buff)
  618. -- 限制读取长度, 2024.1.5 新增
  619. -- 注意
  620. -- 如果是UDP数据, 如果limit小于UDP数据包长度, 只会取前limit个字节, 剩余数据会丢弃
  621. -- 如果是TCP数据, 如果有剩余数据, 不会丢弃, 可继续读取.
  622. -- 有新的数据到来才会有新的EVENT数据, 未读取完成的数据不会触发新EVENT事件
  623. local succ, data_len, remote_ip, remote_port = socket.rx(ctrl, buff, 1500)
  624. -- 读取缓冲区大小, 2024.1.5 新增, 注意,老版本固件不传buff参数会报错的
  625. -- 对于TCP数据, 这里返回的是待读取的数据的总长度
  626. -- 对于UDP数据, 这里返回的是单个UDP数据包的长度
  627. local succ, data_len = socket.rx(ctrl)
  628. if succ then
  629. log.info("待收取数据长度", data_len)
  630. end
  631. */
  632. static int l_socket_rx(lua_State *L)
  633. {
  634. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  635. L_CTRL_CHECK;
  636. luat_zbuff_t *buff = NULL;
  637. luat_ip_addr_t ip_addr = {0};
  638. uint8_t ip[17] = {0};
  639. uint16_t port = 0;
  640. // uint8_t new_flag = 0;
  641. uint32_t rx_len = 0;
  642. uint32_t total_len = 0;
  643. int result = network_rx(l_ctrl->netc, NULL, 0, 0, NULL, NULL, &total_len);
  644. if (result < 0)
  645. {
  646. lua_pushboolean(L, 0);
  647. lua_pushinteger(L, 0);
  648. return 2;
  649. }
  650. if (!total_len)
  651. {
  652. lua_pushboolean(L, 1);
  653. lua_pushinteger(L, 0);
  654. return 2;
  655. }
  656. // 是否传入的zbuff呢? 没有的话就返回值长度
  657. if (lua_isuserdata(L, 2)) {
  658. buff = ((luat_zbuff_t *)luaL_checkudata(L, 2, LUAT_ZBUFF_TYPE));
  659. }
  660. else {
  661. // 仅返回长度
  662. lua_pushboolean(L, 1);
  663. lua_pushinteger(L, total_len);
  664. return 2;
  665. }
  666. // 是否限制接收数据长度
  667. if (lua_gettop(L) >= 4 && lua_isinteger(L, 4)) {
  668. uint32_t limit = lua_tointeger(L, 4);
  669. if (total_len > limit)
  670. {
  671. LLOGD("待数据数据长度 %d 限制读取长度 %d", total_len, limit);
  672. total_len = limit;
  673. }
  674. }
  675. if (1)
  676. {
  677. if ((buff->len - buff->used) < total_len)
  678. {
  679. result = __zbuff_resize(buff, total_len + buff->used);
  680. if (result < 0) {
  681. if (buff->len > buff->used) {
  682. LLOGW("zbuff自动扩容失败, 减少接收到数据长度 %d -> %d", total_len, buff->len - buff->used);
  683. total_len = buff->len - buff->used;
  684. }
  685. else {
  686. LLOGE("zbuff自动扩容失败, 且zbuff没有剩余空间,无法读取剩余数据");
  687. lua_pushboolean(L, 0);
  688. lua_pushinteger(L, 0);
  689. return 2;
  690. }
  691. }
  692. }
  693. result = network_rx(l_ctrl->netc, buff->addr + buff->used, total_len, 0, &ip_addr, &port, &rx_len);
  694. if (result < 0)
  695. {
  696. lua_pushboolean(L, 0);
  697. lua_pushinteger(L, 0);
  698. return 2;
  699. }
  700. else if (!rx_len)
  701. {
  702. lua_pushboolean(L, 1);
  703. lua_pushinteger(L, 0);
  704. return 2;
  705. }
  706. else
  707. {
  708. buff->used += rx_len;
  709. lua_pushboolean(L, 1);
  710. lua_pushinteger(L, rx_len);
  711. if (l_ctrl->netc->is_tcp)
  712. {
  713. lua_pushnil(L);
  714. lua_pushnil(L);
  715. }
  716. else
  717. {
  718. #ifdef LUAT_USE_LWIP
  719. #if LWIP_IPV6
  720. if (IPADDR_TYPE_V4 == ip_addr.type)
  721. {
  722. ip[0] = 0;
  723. memcpy(ip + 1, &ip_addr.u_addr.ip4.addr, 4);
  724. lua_pushlstring(L, (const char*)ip, 5);
  725. }
  726. else
  727. {
  728. ip[0] = 1;
  729. memcpy(ip + 1, ip_addr.u_addr.ip6.addr, 16);
  730. lua_pushlstring(L, (const char*)ip, 17);
  731. }
  732. #else
  733. ip[0] = 0;
  734. memcpy(ip + 1, &ip_addr.addr, 4);
  735. lua_pushlstring(L, (const char*)ip, 5);
  736. #endif
  737. #else
  738. if (!ip_addr.is_ipv6)
  739. {
  740. ip[0] = 0;
  741. memcpy(ip + 1, &ip_addr.ipv4, 4);
  742. lua_pushlstring(L, (const char*)ip, 5);
  743. }
  744. else
  745. {
  746. ip[0] = 1;
  747. memcpy(ip + 1, &ip_addr.ipv6_u8_addr, 16);
  748. lua_pushlstring(L, (const char*)ip, 17);
  749. }
  750. #endif
  751. lua_pushinteger(L, port);
  752. }
  753. }
  754. }
  755. return 4;
  756. }
  757. /*
  758. 读取数据(非zbuff版本)
  759. @api socket.read(netc, len)
  760. @userdata socket.create得到的ctrl
  761. @int 限制读取数据长度,可选,不传就是读出全部
  762. @return boolean 读取成功与否
  763. @return string 读取的数据,仅当读取成功时有效
  764. @return string 对方IP地址,仅当读取成功且UDP通信时有效
  765. @return int 对方端口,仅当读取成功且UDP通信时有效
  766. @usage
  767. -- 本函数于2024.4.8添加, 用于简易读取不大的数据
  768. -- 请优先使用socket.rx函数, 本函数主要用于固件不含zbuff库时的变通调用
  769. local ok, data = socket.read(netc, 1500)
  770. if ok and #data > 0 then
  771. log.info("读取到的数据", data)
  772. end
  773. */
  774. static int l_socket_read(lua_State *L) {
  775. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  776. L_CTRL_CHECK;
  777. luat_ip_addr_t ip_addr = {0};
  778. uint8_t ip[17] = {0};
  779. uint16_t port = 0;
  780. // uint8_t new_flag = 0;
  781. uint32_t rx_len = 0;
  782. uint32_t total_len = 0;
  783. int result = network_rx(l_ctrl->netc, NULL, 0, 0, NULL, NULL, &total_len);
  784. if (result < 0)
  785. {
  786. lua_pushboolean(L, 0);
  787. return 1;
  788. }
  789. if (!total_len)
  790. {
  791. lua_pushboolean(L, 1);
  792. lua_pushstring(L, "");
  793. return 2;
  794. }
  795. // 是否限制接收数据长度
  796. if (lua_gettop(L) >= 2 && lua_isinteger(L, 2)) {
  797. uint32_t limit = lua_tointeger(L, 2);
  798. if (limit > 0 && total_len > limit)
  799. {
  800. LLOGD("待数据数据长度 %d 限制读取长度 %d", total_len, limit);
  801. total_len = limit;
  802. }
  803. }
  804. luaL_Buffer bf = {0};
  805. char* buff = luaL_buffinitsize(L, &bf, total_len);
  806. result = network_rx(l_ctrl->netc, (uint8_t*)buff, total_len, 0, &ip_addr, &port, &rx_len);
  807. if (result < 0)
  808. {
  809. lua_pushboolean(L, 0);
  810. return 1;
  811. }
  812. else if (!rx_len)
  813. {
  814. lua_pushboolean(L, 1);
  815. lua_pushstring(L, "");
  816. return 2;
  817. }
  818. else
  819. {
  820. lua_pushboolean(L, 1);
  821. luaL_pushresultsize(&bf, rx_len);
  822. if (l_ctrl->netc->is_tcp)
  823. {
  824. return 2;
  825. }
  826. else
  827. {
  828. #ifdef LUAT_USE_LWIP
  829. #if LWIP_IPV6
  830. if (IPADDR_TYPE_V4 == ip_addr.type)
  831. {
  832. ip[0] = 0;
  833. memcpy(ip + 1, &ip_addr.u_addr.ip4.addr, 4);
  834. lua_pushlstring(L, (const char*)ip, 5);
  835. }
  836. else
  837. {
  838. ip[0] = 1;
  839. memcpy(ip + 1, ip_addr.u_addr.ip6.addr, 16);
  840. lua_pushlstring(L, (const char*)ip, 17);
  841. }
  842. #else
  843. ip[0] = 0;
  844. memcpy(ip + 1, &ip_addr.addr, 4);
  845. lua_pushlstring(L, (const char*)ip, 5);
  846. #endif
  847. #else
  848. if (!ip_addr.is_ipv6)
  849. {
  850. ip[0] = 0;
  851. memcpy(ip + 1, &ip_addr.ipv4, 4);
  852. lua_pushlstring(L, (const char*)ip, 5);
  853. }
  854. else
  855. {
  856. ip[0] = 1;
  857. memcpy(ip + 1, &ip_addr.ipv6_u8_addr, 16);
  858. lua_pushlstring(L, (const char*)ip, 17);
  859. }
  860. #endif
  861. lua_pushinteger(L, port);
  862. return 4;
  863. }
  864. }
  865. }
  866. /*
  867. 等待新的socket消息,在连接成功和发送数据成功后,使用一次将network状态转换到接收新数据
  868. @api socket.wait(ctrl)
  869. @user_data socket.create得到的ctrl
  870. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果false,后续要close
  871. @return boolean true有新的数据需要接收,false没有数据,之后需要接收socket.EVENT消息
  872. @usage
  873. local succ, result = socket.wait(ctrl)
  874. */
  875. static int l_socket_wait(lua_State *L)
  876. {
  877. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  878. L_CTRL_CHECK;
  879. int result = network_wait_event(l_ctrl->netc, NULL, 0, NULL);
  880. lua_pushboolean(L, (result < 0)?0:1);
  881. lua_pushboolean(L, result == 0);
  882. return 2;
  883. }
  884. /*
  885. 作为服务端开始监听
  886. @api socket.listen(ctrl)
  887. @user_data socket.create得到的ctrl
  888. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果false,后续要close
  889. @return boolean true已经connect,false没有connect,之后需要接收socket.ON_LINE消息
  890. @usage
  891. local succ, result = socket.listen(ctrl)
  892. */
  893. static int l_socket_listen(lua_State *L)
  894. {
  895. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  896. L_CTRL_CHECK;
  897. int result = network_listen(l_ctrl->netc, 0);
  898. lua_pushboolean(L, (result < 0)?0:1);
  899. lua_pushboolean(L, result == 0);
  900. return 2;
  901. }
  902. /*
  903. 作为服务端接收到一个新的客户端,注意,如果是类似W5500的硬件协议栈不支持1对多,则不需要第二个参数
  904. @api socket.accept(ctrl)
  905. @user_data socket.create得到的ctrl,这里是服务器端
  906. @string or function or nil string为消息通知的taskName,function则为回调函数,和socket.create参数一致
  907. @return boolean true没有异常发生,false失败了,如果false则不需要看下一个返回值了,如果false,后续要close
  908. @return user_data or nil 如果支持1对多,则会返回新的ctrl,自动create,如果不支持则返回nil
  909. @usage
  910. local succ, new_netc = socket.accept(ctrl, cb)
  911. */
  912. static int l_socket_accept(lua_State *L)
  913. {
  914. luat_socket_ctrl_t *old_ctrl = l_get_ctrl(L, 1);
  915. if (!old_ctrl) return 0;
  916. if (network_accept_enable(old_ctrl->netc))
  917. {
  918. luat_socket_ctrl_t *new_ctrl = (luat_socket_ctrl_t *)lua_newuserdata(L, sizeof(luat_socket_ctrl_t));
  919. if (!new_ctrl)
  920. {
  921. lua_pushboolean(L, 0);
  922. lua_pushnil(L);
  923. return 2;
  924. }
  925. new_ctrl->adapter_index = old_ctrl->adapter_index;
  926. new_ctrl->netc = network_alloc_ctrl(old_ctrl->adapter_index);
  927. if (!new_ctrl->netc)
  928. {
  929. LLOGD("create fail");
  930. lua_pushboolean(L, 0);
  931. lua_pushnil(L);
  932. return 2;
  933. }
  934. network_init_ctrl(new_ctrl->netc, NULL, luat_lib_socket_callback, new_ctrl);
  935. if (lua_isfunction(L, 2))
  936. {
  937. lua_pushvalue(L, 2);
  938. new_ctrl->cb_ref = luaL_ref(L, LUA_REGISTRYINDEX);
  939. new_ctrl->task_name = NULL;
  940. }
  941. else if (lua_isstring(L, 2))
  942. {
  943. new_ctrl->cb_ref = 0;
  944. size_t len;
  945. const char *buf;
  946. buf = lua_tolstring(L, 2, &len);//取出字符串数据
  947. new_ctrl->task_name = luat_heap_malloc(len + 1);
  948. memset(new_ctrl->task_name, 0, len + 1);
  949. memcpy(new_ctrl->task_name, buf, len);
  950. }
  951. if (network_socket_accept(old_ctrl, new_ctrl))
  952. {
  953. lua_pushboolean(L, 0);
  954. lua_pushnil(L);
  955. return 2;
  956. }
  957. else
  958. {
  959. lua_pushboolean(L, 1);
  960. luaL_setmetatable(L, LUAT_NW_CTRL_TYPE);
  961. return 2;
  962. }
  963. }
  964. else
  965. {
  966. lua_pushboolean(L, !network_socket_accept(old_ctrl->netc, NULL));
  967. lua_pushnil(L);
  968. return 2;
  969. }
  970. }
  971. /*
  972. 获取socket当前状态
  973. @api socket.state(ctrl)
  974. @user_data socket.create得到的ctrl
  975. @return int or nil,输入参数正确的情况下,返回状态的数值,否则返回nil
  976. @return string or nil,输入参数正确的情况下,返回状态的中文描述,否则返回nil
  977. @usage
  978. local state, str = socket.state(ctrl)
  979. log.info("state", state, str)
  980. state 0 "硬件离线",
  981. 1 "离线",
  982. 2 "等待DNS",
  983. 3 "正在连接",
  984. 4 "正在TLS握手",
  985. 5 "在线",
  986. 6 "在监听",
  987. 7 "正在离线",
  988. 8 "未知"
  989. */
  990. static int l_socket_state(lua_State *L)
  991. {
  992. luat_socket_ctrl_t *l_ctrl = l_get_ctrl(L, 1);
  993. L_CTRL_CHECK;
  994. lua_pushinteger(L, l_ctrl->netc->state);
  995. lua_pushstring(L, network_ctrl_state_string(l_ctrl->netc->state));
  996. return 2;
  997. }
  998. /*
  999. 主动释放掉network_ctrl
  1000. @api socket.release(ctrl)
  1001. @user_data socket.create得到的ctrl
  1002. @usage
  1003. -- 释放后就不能再使用了
  1004. socket.release(ctrl)
  1005. */
  1006. static int l_socket_release(lua_State *L)
  1007. {
  1008. return l_socket_gc(L);
  1009. }
  1010. /*
  1011. 设置DNS服务器
  1012. @api socket.setDNS(adapter_index, dns_index, ip)
  1013. @int 适配器序号, 只能是socket.ETH0,socket.STA,socket.AP,如果不填,会选择最后一个注册的适配器
  1014. @int dns服务器序号,从1开始
  1015. @string or int dns,如果是IPV4,可以是大端格式的int值
  1016. @return boolean 成功返回true,失败返回false
  1017. @usage
  1018. -- 设置默认网络适配器的DNS配置
  1019. socket.setDNS(nil, 1, "114.114.114.114")
  1020. -- 设置制定网络适配器的DNS配置
  1021. socket.setDNS(socket.ETH0, 1, "114.114.114.114")
  1022. */
  1023. static int l_socket_set_dns(lua_State *L)
  1024. {
  1025. #ifdef LUAT_USE_LWIP
  1026. char ip_buf[68];
  1027. int adapter_index = luaL_optinteger(L, 1, network_register_get_default());
  1028. if (adapter_index < 0 || adapter_index >= NW_ADAPTER_QTY)
  1029. {
  1030. lua_pushboolean(L, 0);
  1031. return 1;
  1032. }
  1033. int dns_index = luaL_optinteger(L, 2, 1);
  1034. luat_ip_addr_t ip_addr;
  1035. const char *ip;
  1036. size_t ip_len;
  1037. network_set_ip_invaild(&ip_addr);
  1038. if (lua_isinteger(L, 3))
  1039. {
  1040. network_set_ip_ipv4(&ip_addr, lua_tointeger(L, 3));
  1041. ip = NULL;
  1042. ip_len = 0;
  1043. }
  1044. else
  1045. {
  1046. ip_len = 0;
  1047. ip = luaL_checklstring(L, 3, &ip_len);
  1048. memcpy(ip_buf, ip, ip_len);
  1049. ip_buf[ip_len] = 0;
  1050. ipaddr_aton(ip_buf, &ip_addr);
  1051. }
  1052. #else
  1053. int adapter_index = luaL_optinteger(L, 1, network_register_get_default());
  1054. if (adapter_index < 0 || adapter_index >= NW_ADAPTER_QTY)
  1055. {
  1056. lua_pushboolean(L, 0);
  1057. return 1;
  1058. }
  1059. int dns_index = luaL_optinteger(L, 2, 1);
  1060. luat_ip_addr_t ip_addr;
  1061. const char *ip;
  1062. size_t ip_len;
  1063. ip_addr.is_ipv6 = 0xff;
  1064. if (lua_isinteger(L, 3))
  1065. {
  1066. ip_addr.is_ipv6 = 0;
  1067. ip_addr.ipv4 = lua_tointeger(L, 3);
  1068. ip = NULL;
  1069. ip_len = 0;
  1070. }
  1071. else
  1072. {
  1073. ip_len = 0;
  1074. ip = luaL_checklstring(L, 3, &ip_len);
  1075. ip_addr.is_ipv6 = !network_string_is_ipv4(ip, ip_len);
  1076. if (ip_addr.is_ipv6)
  1077. {
  1078. char *temp = luat_heap_malloc(ip_len + 1);
  1079. memcpy(temp, ip, ip_len);
  1080. temp[ip_len] = 0;
  1081. network_string_to_ipv6(temp, &ip_addr);
  1082. luat_heap_free(temp);
  1083. }
  1084. else
  1085. {
  1086. ip_addr.ipv4 = network_string_to_ipv4(ip, ip_len);
  1087. }
  1088. }
  1089. #endif
  1090. void* userdata = NULL;
  1091. network_adapter_info* info = network_adapter_fetch(adapter_index, &userdata);
  1092. if (info == NULL) {
  1093. LLOGW("adapter_index is invaild %d", adapter_index);
  1094. lua_pushboolean(L, 0);
  1095. }
  1096. else {
  1097. network_set_dns_server(adapter_index, dns_index - 1, &ip_addr);
  1098. lua_pushboolean(L, 1);
  1099. }
  1100. return 1;
  1101. }
  1102. /*
  1103. 设置SSL的log登记
  1104. @api socket.sslLog(log_level)
  1105. @int mbedtls log等级
  1106. @return nil 无返回值
  1107. @usage
  1108. --[[
  1109. SSL/TLS log级别说明
  1110. 0不打印
  1111. 1只打印错误和警
  1112. 2大部分info
  1113. 3及3以上详细的debug
  1114. 过多的信息可能会造成内存碎片化
  1115. ]]
  1116. -- 打印大部分info日志
  1117. socket.sslLog(2)
  1118. */
  1119. static int l_socket_set_ssl_log(lua_State *L)
  1120. {
  1121. #if defined(MBEDTLS_DEBUG_C)
  1122. mbedtls_debug_set_threshold(luaL_optinteger(L, 1, 1));
  1123. #endif
  1124. return 0;
  1125. }
  1126. #ifdef LUAT_USE_SNTP
  1127. #include "luat_sntp.h"
  1128. #endif
  1129. /*
  1130. 查看网卡适配器的联网状态
  1131. @api socket.adapter(index)
  1132. @int 需要查看的适配器序号,可以留空会查看全部网卡,直到遇到IP READY的,如果指定网卡,只能是socket.ETH0(外置以太网),socket.LWIP_ETH(内置以太网),socket.LWIP_STA(内置WIFI的STA),socket.LWIP_AP(内置WIFI的AP),socket.LWIP_GP(内置蜂窝网络的GPRS),socket.USB(外置USB网卡)
  1133. @return boolean 被查看的适配器是否IP READY,true表示已经准备好可以联网了,false暂时不可以联网
  1134. @return int 最后一个被查看的适配器序号
  1135. @usage
  1136. -- 查看全部网卡,直到找到一个是IP READY的
  1137. local isReady,index = socket.adapter() --如果isReady为true,则index为IP READY的网卡适配器序号
  1138. --查看外置以太网(比如W5500)是否IP READY
  1139. local isReady,default = socket.adapter(socket.ETH0)
  1140. */
  1141. static int l_socket_adapter(lua_State *L)
  1142. {
  1143. int adapter_index = luaL_optinteger(L, 1, -1);
  1144. if (adapter_index > NW_ADAPTER_INDEX_LWIP_NONE && adapter_index < NW_ADAPTER_QTY)
  1145. {
  1146. lua_pushboolean(L, network_check_ready(NULL, adapter_index));
  1147. lua_pushinteger(L, adapter_index);
  1148. }
  1149. else
  1150. {
  1151. for(int i = NW_ADAPTER_INDEX_LWIP_GPRS; i < NW_ADAPTER_QTY; i++)
  1152. {
  1153. if (network_check_ready(NULL, i))
  1154. {
  1155. lua_pushboolean(L, 1);
  1156. lua_pushinteger(L, i);
  1157. return 2;
  1158. }
  1159. }
  1160. lua_pushboolean(L, 0);
  1161. lua_pushinteger(L, NW_ADAPTER_QTY - 1);
  1162. }
  1163. return 2;
  1164. }
  1165. /*
  1166. 获取对端ip
  1167. @api socket.remoteIP(ctrl)
  1168. @user_data socket.create得到的ctrl
  1169. @return string IP1,如果为nil,则表示没有获取到IP地址
  1170. @return string IP2,如果为nil,则表示没有IP2
  1171. @return string IP3,如果为nil,则表示没有IP3
  1172. @return string IP4,如果为nil,则表示没有IP4
  1173. @usage
  1174. -- 注意: ,必须在接收到socket.ON_LINE消息之后才可能获取到,最多返回4个IP。
  1175. -- socket.connect里如果remote_port设置成0,则当DNS完成时就返回socket.ON_LINE消息
  1176. local ip1,ip2,ip3,ip4 = socket.remoteIP(ctrl)
  1177. */
  1178. static int l_socket_remote_ip(lua_State *L)
  1179. {
  1180. luat_socket_ctrl_t *ctrl = l_get_ctrl(L, 1);
  1181. uint8_t i;
  1182. uint8_t total;
  1183. if (!ctrl)
  1184. {
  1185. goto NO_REMOTE_IP;
  1186. }
  1187. if (!ctrl->netc->dns_ip_nums || !ctrl->netc->dns_ip)
  1188. {
  1189. goto NO_REMOTE_IP;
  1190. }
  1191. total = (ctrl->netc->dns_ip_nums > 4)?4:ctrl->netc->dns_ip_nums;
  1192. for(i = 0; i < total; i++)
  1193. {
  1194. #ifdef LUAT_USE_LWIP
  1195. lua_pushfstring(L, "%s", ipaddr_ntoa(&ctrl->netc->dns_ip[i].ip));
  1196. #else
  1197. PV_Union uPV;
  1198. uPV.u32 = &ctrl->netc->dns_ip[i].ip.ipv4;
  1199. lua_pushfstring(L, "%d.%d.%d.%d", uPV.u8[0], uPV.u8[1], uPV.u8[2], uPV.u8[3]);
  1200. #endif
  1201. }
  1202. if (total < 4)
  1203. {
  1204. for(i = total; i < 4; i++)
  1205. {
  1206. lua_pushnil(L);
  1207. }
  1208. }
  1209. return 4;
  1210. NO_REMOTE_IP:
  1211. lua_pushnil(L);
  1212. lua_pushnil(L);
  1213. lua_pushnil(L);
  1214. lua_pushnil(L);
  1215. return 4;
  1216. }
  1217. /*
  1218. 设置默认网络适配器编号
  1219. @api socket.dft(id)
  1220. @int 默认适配器编号,若不传,则打包获取
  1221. @return int 默认适配器编号
  1222. @return int 最后一个注册的适配器编号(2025.7.25新增)
  1223. @usage
  1224. -- 本函数于 2025.1.6 新增
  1225. -- 获取当前默认适配器编号
  1226. local id = socket.dft()
  1227. -- 设置默认适配器编号
  1228. socket.dft(socket.LWIP_ETH)
  1229. -- 获取当前默认适配器编号, 及最后一个注册的适配器编号
  1230. local id, last_id = socket.dft()
  1231. log.info("当前默认适配器编号", id, "最后一个注册的适配器编号", last_id)
  1232. -- 1. 当前的默认网卡, 可以获取, 可以设置, 就是socket.dft(id)的第一个参数, 也是第一个返回值
  1233. -- 2. 最后注册的网卡, 可以获取, 不支持设置, 就是socket.dft()的第二个返回值
  1234. */
  1235. static int l_socket_default(lua_State *L) {
  1236. uint8_t id = 0;
  1237. if (lua_type(L, 1) == LUA_TNUMBER) {
  1238. id = (uint8_t)luaL_checkinteger(L, 1);
  1239. if (id < NW_ADAPTER_QTY) {
  1240. network_register_set_default(id);
  1241. }
  1242. }
  1243. lua_pushinteger(L, network_register_get_default());
  1244. lua_pushinteger(L, network_get_last_register_adapter_real());
  1245. return 2;
  1246. }
  1247. #include "rotable2.h"
  1248. static const rotable_Reg_t reg_socket_adapter[] =
  1249. {
  1250. {"create", ROREG_FUNC(l_socket_create)},
  1251. {"debug", ROREG_FUNC(l_socket_set_debug)},
  1252. {"config", ROREG_FUNC(l_socket_config)},
  1253. {"linkup", ROREG_FUNC(l_socket_linkup)},
  1254. {"connect", ROREG_FUNC(l_socket_connect)},
  1255. {"listen", ROREG_FUNC(l_socket_listen)},
  1256. {"accept", ROREG_FUNC(l_socket_accept)},
  1257. {"discon", ROREG_FUNC(l_socket_disconnect)},
  1258. {"close", ROREG_FUNC(l_socket_close)},
  1259. {"tx", ROREG_FUNC(l_socket_tx)},
  1260. {"rx", ROREG_FUNC(l_socket_rx)},
  1261. {"read", ROREG_FUNC(l_socket_read)},
  1262. {"wait", ROREG_FUNC(l_socket_wait)},
  1263. {"state", ROREG_FUNC(l_socket_state)},
  1264. {"release", ROREG_FUNC(l_socket_release)},
  1265. { "setDNS", ROREG_FUNC(l_socket_set_dns)},
  1266. { "sslLog", ROREG_FUNC(l_socket_set_ssl_log)},
  1267. {"localIP", ROREG_FUNC(l_socket_local_ip)},
  1268. {"remoteIP", ROREG_FUNC(l_socket_remote_ip)},
  1269. {"adapter", ROREG_FUNC(l_socket_adapter)},
  1270. {"dft", ROREG_FUNC(l_socket_default)},
  1271. #ifdef LUAT_USE_SNTP
  1272. {"sntp", ROREG_FUNC(l_sntp_get)},
  1273. {"ntptm", ROREG_FUNC(l_sntp_tm)},
  1274. {"sntp_port", ROREG_FUNC(l_sntp_port)},
  1275. #endif
  1276. //@const ETH0 number 带硬件协议栈的ETH0,值为5
  1277. { "ETH0", ROREG_INT(NW_ADAPTER_INDEX_ETH0)},
  1278. //@const LWIP_ETH number 使用LWIP协议栈的以太网卡,值为4
  1279. { "LWIP_ETH", ROREG_INT(NW_ADAPTER_INDEX_LWIP_ETH)},
  1280. //@const LWIP_STA number 使用LWIP协议栈的WIFI STA,值为2
  1281. { "LWIP_STA", ROREG_INT(NW_ADAPTER_INDEX_LWIP_WIFI_STA)},
  1282. //@const LWIP_AP number 使用LWIP协议栈的WIFI AP,值为3
  1283. { "LWIP_AP", ROREG_INT(NW_ADAPTER_INDEX_LWIP_WIFI_AP)},
  1284. //@const LWIP_GP number 使用LWIP协议栈的移动蜂窝模块,值为1
  1285. { "LWIP_GP", ROREG_INT(NW_ADAPTER_INDEX_LWIP_GPRS)},
  1286. //@const USB number 使用LWIP协议栈的USB网卡,值为6
  1287. { "USB", ROREG_INT(NW_ADAPTER_INDEX_USB)},
  1288. //@const LINK number LINK事件
  1289. { "LINK", ROREG_INT(EV_NW_RESULT_LINK & 0x0fffffff)},
  1290. //@const ON_LINE number ON_LINE事件
  1291. { "ON_LINE", ROREG_INT(EV_NW_RESULT_CONNECT & 0x0fffffff)},
  1292. //@const EVENT number EVENT事件
  1293. { "EVENT", ROREG_INT(EV_NW_RESULT_EVENT & 0x0fffffff)},
  1294. //@const TX_OK number TX_OK事件
  1295. { "TX_OK", ROREG_INT(EV_NW_RESULT_TX & 0x0fffffff)},
  1296. //@const CLOSED number CLOSED事件
  1297. { "CLOSED", ROREG_INT(EV_NW_RESULT_CLOSE & 0x0fffffff)},
  1298. //@const LWIP_SPI number 使用LWIP协议栈的SPI网络
  1299. { "LWIP_SPI", ROREG_INT(NW_ADAPTER_INDEX_LWIP_SPI)},
  1300. //@const LWIP_USER0 number 使用LWIP协议栈的自定义网卡0, 2025.1.12新增
  1301. { "LWIP_USER0", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER0)},
  1302. //@const LWIP_USER1 number 使用LWIP协议栈的自定义网卡1, 2025.1.12新增
  1303. { "LWIP_USER1", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER1)},
  1304. //@const LWIP_USER2 number 使用LWIP协议栈的自定义网卡2, 2025.1.12新增
  1305. { "LWIP_USER2", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER2)},
  1306. //@const LWIP_USER3 number 使用LWIP协议栈的自定义网卡3, 2025.1.12新增
  1307. { "LWIP_USER3", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER3)},
  1308. //@const LWIP_USER4 number 使用LWIP协议栈的自定义网卡4, 2025.1.12新增
  1309. { "LWIP_USER4", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER4)},
  1310. //@const LWIP_USER5 number 使用LWIP协议栈的自定义网卡5, 2025.1.12新增
  1311. { "LWIP_USER5", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER5)},
  1312. //@const LWIP_USER6 number 使用LWIP协议栈的自定义网卡6, 2025.1.12新增
  1313. { "LWIP_USER6", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER6)},
  1314. //@const LWIP_USER7 number 使用LWIP协议栈的自定义网卡7, 2025.1.12新增
  1315. { "LWIP_USER7", ROREG_INT(NW_ADAPTER_INDEX_LWIP_USER7)},
  1316. //@const LWIP_GP_GW number 4G代理网关
  1317. { "LWIP_GP_GW", ROREG_INT(NW_ADAPTER_INDEX_LWIP_GP_GW)},
  1318. { NULL, ROREG_INT(0)}
  1319. };
  1320. LUAMOD_API int luaopen_socket_adapter( lua_State *L ) {
  1321. luat_newlib2(L, reg_socket_adapter);
  1322. luaL_newmetatable(L, LUAT_NW_CTRL_TYPE); /* create metatable for file handles */
  1323. lua_pushcfunction(L, l_socket_gc);
  1324. lua_setfield(L, -2, "__gc");
  1325. lua_pop(L, 1); /* pop new metatable */
  1326. return 1;
  1327. }
  1328. // #endif