luat_lib_socket.c 46 KB

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