luat_mqtt_client.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530
  1. #include "luat_base.h"
  2. #include "luat_network_adapter.h"
  3. #include "libemqtt.h"
  4. #include "luat_rtos.h"
  5. #include "luat_mem.h"
  6. #include "luat_mqtt.h"
  7. #define LUAT_LOG_TAG "mqtt"
  8. #include "luat_log.h"
  9. #ifndef LUAT_MQTT_DEBUG
  10. #define LUAT_MQTT_DEBUG 0
  11. #endif
  12. #if LUAT_MQTT_DEBUG == 0
  13. #undef LLOGD
  14. #define LLOGD(...)
  15. #endif
  16. static int luat_mqtt_msg_cb(luat_mqtt_ctrl_t *mqtt_ctrl);
  17. #ifdef __LUATOS__
  18. #include "luat_msgbus.h"
  19. int32_t luatos_mqtt_callback(lua_State *L, void* ptr);
  20. #endif
  21. int l_luat_mqtt_msg_cb(luat_mqtt_ctrl_t * ptr, int arg1, int arg2) {
  22. #ifdef __LUATOS__
  23. luat_mqtt_ctrl_t *mqtt_ctrl =(luat_mqtt_ctrl_t *)ptr;
  24. rtos_msg_t msg = {
  25. .handler = luatos_mqtt_callback,
  26. .ptr = ptr,
  27. .arg1 = arg1,
  28. .arg2 = arg2
  29. };
  30. if (mqtt_ctrl->app_cb)
  31. {
  32. luat_mqtt_cb_t mqtt_cb = mqtt_ctrl->app_cb;
  33. mqtt_cb(mqtt_ctrl, arg1);
  34. if (MQTT_MSG_PUBLISH == arg1)
  35. {
  36. return 0;
  37. }
  38. }
  39. luat_msgbus_put(&msg, 0);
  40. #else
  41. luat_mqtt_ctrl_t *mqtt_ctrl =(luat_mqtt_ctrl_t *)ptr;
  42. if (mqtt_ctrl->mqtt_cb){
  43. luat_mqtt_cb_t mqtt_cb = mqtt_ctrl->mqtt_cb;
  44. mqtt_cb(mqtt_ctrl, arg1);
  45. }
  46. #endif
  47. return 0;
  48. }
  49. LUAT_RT_RET_TYPE luat_mqtt_timer_callback(LUAT_RT_CB_PARAM){
  50. luat_mqtt_ctrl_t * mqtt_ctrl = (luat_mqtt_ctrl_t *)param;
  51. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_TIMER_PING, 0);
  52. }
  53. static LUAT_RT_RET_TYPE reconnect_timer_cb(LUAT_RT_CB_PARAM){
  54. luat_mqtt_ctrl_t * mqtt_ctrl = (luat_mqtt_ctrl_t *)param;
  55. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_RECONNECT, 0);
  56. }
  57. int luat_mqtt_reconnect(luat_mqtt_ctrl_t *mqtt_ctrl) {
  58. mqtt_ctrl->buffer_offset = 0;
  59. int ret = luat_mqtt_connect(mqtt_ctrl);
  60. if(ret){
  61. LLOGI("reconnect init socket ret=%d\n", ret);
  62. luat_mqtt_close_socket(mqtt_ctrl);
  63. }
  64. return ret;
  65. }
  66. int luat_mqtt_ping(luat_mqtt_ctrl_t *mqtt_ctrl) {
  67. mqtt_ping(&mqtt_ctrl->broker);
  68. return 0;
  69. }
  70. int luat_mqtt_init(luat_mqtt_ctrl_t *mqtt_ctrl, int adapter_index) {
  71. memset(mqtt_ctrl, 0, sizeof(luat_mqtt_ctrl_t));
  72. mqtt_ctrl->rxbuff_size = MQTT_RECV_BUF_LEN_MAX+4;
  73. mqtt_ctrl->adapter_index = adapter_index;
  74. mqtt_ctrl->netc = network_alloc_ctrl(adapter_index);
  75. if (!mqtt_ctrl->netc){
  76. LLOGW("network_alloc_ctrl fail");
  77. return -1;
  78. }
  79. network_init_ctrl(mqtt_ctrl->netc, NULL, luat_mqtt_callback, mqtt_ctrl);
  80. mqtt_ctrl->mqtt_state = MQTT_STATE_DISCONNECT;
  81. mqtt_ctrl->netc->is_debug = 0;
  82. mqtt_ctrl->keepalive = 240;
  83. network_set_base_mode(mqtt_ctrl->netc, 1, 10000, 0, 0, 0, 0);
  84. network_set_local_port(mqtt_ctrl->netc, 0);
  85. mqtt_ctrl->reconnect_timer = luat_create_rtos_timer(reconnect_timer_cb, mqtt_ctrl, NULL);
  86. mqtt_ctrl->ping_timer = luat_create_rtos_timer(luat_mqtt_timer_callback, mqtt_ctrl, NULL);
  87. return 0;
  88. }
  89. int luat_mqtt_set_rxbuff_size(luat_mqtt_ctrl_t *mqtt_ctrl, uint32_t rxbuff_size){
  90. mqtt_ctrl->rxbuff_size = rxbuff_size;
  91. return 0;
  92. }
  93. int luat_mqtt_set_keepalive(luat_mqtt_ctrl_t *mqtt_ctrl, uint32_t keepalive){
  94. mqtt_ctrl->keepalive = keepalive;
  95. return 0;
  96. }
  97. int luat_mqtt_set_auto_connect(luat_mqtt_ctrl_t *mqtt_ctrl, uint8_t auto_connect,uint32_t reconnect_time){
  98. mqtt_ctrl->reconnect = auto_connect;
  99. mqtt_ctrl->reconnect_time = reconnect_time;
  100. return 0;
  101. }
  102. int luat_mqtt_set_connopts(luat_mqtt_ctrl_t *mqtt_ctrl, luat_mqtt_connopts_t *opts) {
  103. memcpy(mqtt_ctrl->host, opts->host, strlen(opts->host) + 1);
  104. mqtt_ctrl->remote_port = opts->port;
  105. if (opts->is_tls){
  106. if (network_init_tls(mqtt_ctrl->netc, opts->verify)){
  107. LLOGE("初始化tls失败");
  108. return -1;
  109. }
  110. if (opts->server_cert){
  111. if (network_set_server_cert(mqtt_ctrl->netc, (const unsigned char *)opts->server_cert, opts->server_cert_len+1)){
  112. LLOGE("network_set_server_cert error");
  113. return -1;
  114. }
  115. }
  116. if (opts->client_cert){
  117. if (network_set_client_cert(mqtt_ctrl->netc, (const unsigned char*)opts->client_cert, opts->client_cert_len+1,
  118. (const unsigned char*)opts->client_key, opts->client_key_len+1,
  119. (const unsigned char*)opts->client_password, opts->client_password_len+1)){
  120. LLOGE("network_set_client_cert error");
  121. return -1;
  122. }
  123. }
  124. } else {
  125. network_deinit_tls(mqtt_ctrl->netc);
  126. }
  127. if (opts->is_ipv6) {
  128. network_connect_ipv6_domain(mqtt_ctrl->netc, 1);
  129. }
  130. mqtt_ctrl->broker.socket_info = mqtt_ctrl;
  131. mqtt_ctrl->broker.send = luat_mqtt_send_packet;
  132. return 0;
  133. }
  134. int luat_mqtt_set_triad(luat_mqtt_ctrl_t *mqtt_ctrl, const char* clientid, const char* username, const char* password){
  135. mqtt_init(&(mqtt_ctrl->broker), clientid);
  136. mqtt_init_auth(&(mqtt_ctrl->broker), username, password);
  137. return 0;
  138. }
  139. void luat_mqtt_close_socket(luat_mqtt_ctrl_t *mqtt_ctrl){
  140. LLOGI("mqtt closing socket netc:%p mqtt_state:%d",mqtt_ctrl->netc,mqtt_ctrl->mqtt_state);
  141. if (mqtt_ctrl->mqtt_state){
  142. mqtt_ctrl->mqtt_state = 0;
  143. mqtt_ctrl->buffer_offset = 0;
  144. if (mqtt_ctrl->netc){
  145. network_force_close_socket(mqtt_ctrl->netc);
  146. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_DISCONNECT, mqtt_ctrl->error_state==0?MQTT_ERROR_STATE_SOCKET:mqtt_ctrl->error_state);
  147. }
  148. luat_stop_rtos_timer(mqtt_ctrl->ping_timer);
  149. if (mqtt_ctrl->reconnect && mqtt_ctrl->reconnect_time > 0){
  150. luat_start_rtos_timer(mqtt_ctrl->reconnect_timer, mqtt_ctrl->reconnect_time, 0);
  151. }else{
  152. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_CLOSE, 0);
  153. }
  154. }
  155. mqtt_ctrl->buffer_offset = 0;
  156. }
  157. void luat_mqtt_release_socket(luat_mqtt_ctrl_t *mqtt_ctrl){
  158. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_RELEASE, 0);
  159. if (mqtt_ctrl->ping_timer){
  160. luat_release_rtos_timer(mqtt_ctrl->ping_timer);
  161. mqtt_ctrl->ping_timer = NULL;
  162. }
  163. if (mqtt_ctrl->reconnect_timer){
  164. luat_release_rtos_timer(mqtt_ctrl->reconnect_timer);
  165. mqtt_ctrl->reconnect_timer = NULL;
  166. }
  167. if (mqtt_ctrl->broker.will_data) {
  168. mqtt_ctrl->broker.will_len = 0;
  169. luat_heap_free(mqtt_ctrl->broker.will_data);
  170. mqtt_ctrl->broker.will_data = NULL;
  171. }
  172. if (mqtt_ctrl->netc){
  173. network_release_ctrl(mqtt_ctrl->netc);
  174. mqtt_ctrl->netc = NULL;
  175. }
  176. if (mqtt_ctrl->mqtt_packet_buffer) {
  177. luat_heap_free(mqtt_ctrl->mqtt_packet_buffer);
  178. mqtt_ctrl->mqtt_packet_buffer = NULL;
  179. }
  180. }
  181. static int mqtt_parse(luat_mqtt_ctrl_t *mqtt_ctrl) {
  182. LLOGD("mqtt_parse offset %d", mqtt_ctrl->buffer_offset);
  183. if (mqtt_ctrl->buffer_offset < 2) {
  184. LLOGD("wait more data");
  185. return 0;
  186. }
  187. // 判断数据长度, 前几个字节能判断出够不够读出mqtt的头
  188. char* buf = (char*)mqtt_ctrl->mqtt_packet_buffer;
  189. int num_bytes = 1;
  190. if ((buf[1] & 0x80) == 0x80) {
  191. num_bytes++;
  192. if (mqtt_ctrl->buffer_offset < 3) {
  193. LLOGD("wait more data for mqtt head");
  194. return 0;
  195. }
  196. if ((buf[2] & 0x80) == 0x80) {
  197. num_bytes ++;
  198. if (mqtt_ctrl->buffer_offset < 4) {
  199. LLOGD("wait more data for mqtt head");
  200. return 0;
  201. }
  202. if ((buf[3] & 0x80) == 0x80) {
  203. num_bytes ++;
  204. }
  205. }
  206. }
  207. // 判断数据总长, 这里rem_len只包含mqtt头部之外的数据
  208. uint32_t rem_len = mqtt_parse_rem_len(mqtt_ctrl->mqtt_packet_buffer);
  209. if (rem_len > mqtt_ctrl->buffer_offset - num_bytes - 1) {
  210. LLOGD("wait more data for mqtt head");
  211. return 0;
  212. }
  213. // 至此, mqtt包是完整的 解析类型, 处理之
  214. int ret = luat_mqtt_msg_cb(mqtt_ctrl);
  215. if (ret != 0){
  216. LLOGW("bad mqtt packet!! ret %d", ret);
  217. return -1;
  218. }
  219. // 处理完成后, 如果还有数据, 移动数据, 继续处理
  220. mqtt_ctrl->buffer_offset -= (1 + num_bytes + rem_len);
  221. memmove(mqtt_ctrl->mqtt_packet_buffer, mqtt_ctrl->mqtt_packet_buffer+1 + num_bytes + rem_len, mqtt_ctrl->buffer_offset);
  222. return 1;
  223. }
  224. int luat_mqtt_read_packet(luat_mqtt_ctrl_t *mqtt_ctrl){
  225. LLOGD("luat_mqtt_read_packet mqtt_ctrl->buffer_offset:%d",mqtt_ctrl->buffer_offset);
  226. uint32_t total_len = 0;
  227. uint32_t rx_len = 0;
  228. int result = network_rx(mqtt_ctrl->netc, NULL, 0, 0, NULL, NULL, &total_len);
  229. if (total_len > mqtt_ctrl->rxbuff_size - mqtt_ctrl->buffer_offset) {
  230. LLOGE("too many data wait for recv %d", total_len);
  231. luat_mqtt_close_socket(mqtt_ctrl);
  232. return -1;
  233. }
  234. if (total_len == 0 && !mqtt_ctrl->netc->tls_mode) {
  235. LLOGW("rx event but NO data wait for recv");
  236. return 0;
  237. }
  238. if (mqtt_ctrl->rxbuff_size - mqtt_ctrl->buffer_offset <= 0) {
  239. LLOGE("buff is FULL, mqtt packet too big");
  240. luat_mqtt_close_socket(mqtt_ctrl);
  241. return -1;
  242. }
  243. #define MAX_READ (1024)
  244. int recv_want = 0;
  245. while (mqtt_ctrl->rxbuff_size - mqtt_ctrl->buffer_offset > 0) {
  246. if (MAX_READ > (mqtt_ctrl->rxbuff_size - mqtt_ctrl->buffer_offset)) {
  247. recv_want = mqtt_ctrl->rxbuff_size - mqtt_ctrl->buffer_offset;
  248. }
  249. else {
  250. recv_want = MAX_READ;
  251. }
  252. // 从网络接收数据
  253. result = network_rx(mqtt_ctrl->netc, mqtt_ctrl->mqtt_packet_buffer + mqtt_ctrl->buffer_offset, recv_want, 0, NULL, NULL, &rx_len);
  254. if (rx_len == 0 || result != 0 ) {
  255. LLOGD("rx_len %d result %d", rx_len, result);
  256. break;
  257. }
  258. // 收到数据了, 传给处理函数继续处理
  259. // 数据的长度变更, 触发传递
  260. mqtt_ctrl->buffer_offset += rx_len;
  261. LLOGD("data recv %d offset %d", rx_len, mqtt_ctrl->buffer_offset);
  262. further:
  263. result = mqtt_parse(mqtt_ctrl);
  264. if (result == 0) {
  265. // OK
  266. }else if(result == 1){
  267. if (mqtt_ctrl->buffer_offset > 0)
  268. goto further;
  269. else {
  270. continue;
  271. }
  272. }
  273. else {
  274. LLOGW("mqtt_parse ret %d, closing socket",result);
  275. luat_mqtt_close_socket(mqtt_ctrl);
  276. return -1;
  277. }
  278. }
  279. return 0;
  280. }
  281. static int luat_mqtt_msg_cb(luat_mqtt_ctrl_t *mqtt_ctrl) {
  282. uint8_t msg_tp = MQTTParseMessageType(mqtt_ctrl->mqtt_packet_buffer);
  283. uint16_t msg_id = 0;
  284. uint8_t qos = 0;
  285. switch (msg_tp) {
  286. case MQTT_MSG_CONNACK: {
  287. LLOGD("MQTT_MSG_CONNACK");
  288. if(mqtt_ctrl->mqtt_packet_buffer[3] != 0x00){
  289. LLOGW("CONACK 0x%02x",mqtt_ctrl->mqtt_packet_buffer[3]);
  290. mqtt_ctrl->error_state = mqtt_ctrl->mqtt_packet_buffer[3];
  291. luat_mqtt_close_socket(mqtt_ctrl);
  292. return -1;
  293. }
  294. mqtt_ctrl->mqtt_state = MQTT_STATE_READY;
  295. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_CONNACK, 0);
  296. break;
  297. }
  298. case MQTT_MSG_PUBLISH : {
  299. LLOGD("MQTT_MSG_PUBLISH");
  300. qos = MQTTParseMessageQos(mqtt_ctrl->mqtt_packet_buffer);
  301. #ifdef __LUATOS__
  302. if (!mqtt_ctrl->app_cb)
  303. {
  304. const uint8_t* ptr;
  305. uint16_t topic_len = mqtt_parse_pub_topic_ptr(mqtt_ctrl->mqtt_packet_buffer, &ptr);
  306. uint32_t payload_len = mqtt_parse_pub_msg_ptr(mqtt_ctrl->mqtt_packet_buffer, &ptr);
  307. luat_mqtt_msg_t *mqtt_msg = (luat_mqtt_msg_t *)luat_heap_malloc(sizeof(luat_mqtt_msg_t)+topic_len+payload_len);
  308. mqtt_msg->topic_len = mqtt_parse_pub_topic(mqtt_ctrl->mqtt_packet_buffer, mqtt_msg->data);
  309. mqtt_msg->payload_len = mqtt_parse_publish_msg(mqtt_ctrl->mqtt_packet_buffer, mqtt_msg->data+topic_len);
  310. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PUBLISH, (int)mqtt_msg);
  311. }
  312. else
  313. {
  314. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PUBLISH, 0);
  315. }
  316. #else
  317. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PUBLISH, 0);
  318. #endif
  319. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  320. LLOGD("msg %d qos %d", msg_id, qos);
  321. // 还要回复puback
  322. if (qos == 1) {
  323. LLOGD("reply puback %d", msg_id);
  324. mqtt_puback(&(mqtt_ctrl->broker), msg_id);
  325. }
  326. else if (qos == 2) {
  327. LLOGD("reply pubrec %d", msg_id);
  328. mqtt_pubrec(&(mqtt_ctrl->broker), msg_id);
  329. }
  330. break;
  331. }
  332. case MQTT_MSG_PUBACK : {
  333. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  334. LLOGD("MQTT_MSG_PUBACK %d", msg_id);
  335. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PUBACK, msg_id);
  336. break;
  337. }
  338. case MQTT_MSG_PUBREC : {
  339. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  340. mqtt_pubrel(&(mqtt_ctrl->broker), msg_id);
  341. LLOGD("MQTT_MSG_PUBREC %d", msg_id);
  342. break;
  343. }
  344. case MQTT_MSG_PUBCOMP : {
  345. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  346. LLOGD("MQTT_MSG_PUBCOMP %d", msg_id);
  347. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PUBCOMP, msg_id);
  348. break;
  349. }
  350. case MQTT_MSG_PUBREL : {
  351. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  352. LLOGD("MQTT_MSG_PUBREL %d", msg_id);
  353. mqtt_pubcomp(&(mqtt_ctrl->broker), msg_id);
  354. break;
  355. }
  356. case MQTT_MSG_SUBACK : {
  357. LLOGD("MQTT_MSG_SUBACK");
  358. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  359. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_SUBACK, msg_id);
  360. break;
  361. }
  362. case MQTT_MSG_UNSUBACK : {
  363. LLOGD("MQTT_MSG_UNSUBACK");
  364. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  365. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_UNSUBACK, msg_id);
  366. break;
  367. }
  368. case MQTT_MSG_PINGRESP : {
  369. LLOGD("MQTT_MSG_PINGRESP");
  370. msg_id = mqtt_parse_msg_id(mqtt_ctrl->mqtt_packet_buffer);
  371. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_PINGRESP, msg_id);
  372. break;
  373. }
  374. case MQTT_MSG_DISCONNECT : {
  375. LLOGD("MQTT_MSG_DISCONNECT");
  376. mqtt_ctrl->error_state = MQTT_ERROR_STATE_DISCONNECT;
  377. break;
  378. }
  379. default : {
  380. LLOGD("luat_mqtt_msg_cb error msg_tp:%d",msg_tp);
  381. break;
  382. }
  383. }
  384. return 0;
  385. }
  386. int32_t luat_mqtt_callback(void *data, void *param) {
  387. OS_EVENT *event = (OS_EVENT *)data;
  388. luat_mqtt_ctrl_t *mqtt_ctrl =(luat_mqtt_ctrl_t *)param;
  389. int ret = 0;
  390. // LLOGD("LINK %08X ON_LINE %08X EVENT %08X TX_OK %08X CLOSED %d",EV_NW_RESULT_LINK & 0x0fffffff,EV_NW_RESULT_CONNECT & 0x0fffffff,EV_NW_RESULT_EVENT & 0x0fffffff,EV_NW_RESULT_TX & 0x0fffffff,EV_NW_RESULT_CLOSE & 0x0fffffff);
  391. LLOGD("network mqtt cb %8X %08X",event->ID & 0x0ffffffff, event->Param1);
  392. if (event->Param1){
  393. LLOGE("mqtt_callback param1 %d, event %d closing socket", event->Param1, event->ID - EV_NW_RESULT_BASE);
  394. luat_mqtt_close_socket(mqtt_ctrl);
  395. return -1;
  396. }
  397. if (event->ID == EV_NW_RESULT_LINK){
  398. return 0; // 这里应该直接返回, 不能往下调用network_wait_event
  399. }else if(event->ID == EV_NW_RESULT_CONNECT){
  400. mqtt_ctrl->mqtt_state = MQTT_STATE_MQTT;
  401. ret = mqtt_connect(&(mqtt_ctrl->broker));
  402. if(ret==1){
  403. luat_start_rtos_timer(mqtt_ctrl->ping_timer, mqtt_ctrl->keepalive*1000*3/4, 1);
  404. }
  405. }else if(event->ID == EV_NW_RESULT_EVENT){
  406. if (event->Param1==0){
  407. ret = luat_mqtt_read_packet(mqtt_ctrl);
  408. if (ret){
  409. return -1;
  410. }
  411. // LLOGD("luat_mqtt_read_packet ret:%d",ret);
  412. // luat_stop_rtos_timer(mqtt_ctrl->ping_timer);
  413. // luat_start_rtos_timer(mqtt_ctrl->ping_timer, mqtt_ctrl->keepalive*1000*0.75, 1);
  414. }
  415. }else if(event->ID == EV_NW_RESULT_TX){
  416. #ifdef __LUATOS__
  417. #else
  418. if (MQTT_STATE_READY == mqtt_ctrl->mqtt_state) {
  419. l_luat_mqtt_msg_cb(mqtt_ctrl, MQTT_MSG_TCP_TX_DONE, 0);
  420. }
  421. #endif
  422. // luat_stop_rtos_timer(mqtt_ctrl->ping_timer);
  423. // luat_start_rtos_timer(mqtt_ctrl->ping_timer, mqtt_ctrl->keepalive*1000*0.75, 1);
  424. }else if(event->ID == EV_NW_RESULT_CLOSE){
  425. }
  426. ret = network_wait_event(mqtt_ctrl->netc, NULL, 0, NULL);
  427. if (ret < 0){
  428. LLOGW("network_wait_event ret %d, closing socket", ret);
  429. luat_mqtt_close_socket(mqtt_ctrl);
  430. return -1;
  431. }
  432. return 0;
  433. }
  434. int luat_mqtt_send_packet(void* socket_info, const void* buf, unsigned int count){
  435. luat_mqtt_ctrl_t * mqtt_ctrl = (luat_mqtt_ctrl_t *)socket_info;
  436. uint32_t tx_len = 0;
  437. int ret = network_tx(mqtt_ctrl->netc, buf, count, 0, NULL, 0, &tx_len, 0);
  438. if (ret < 0) {
  439. LLOGW("network_tx ret %d, closing socket", ret);
  440. luat_mqtt_close_socket(mqtt_ctrl);
  441. return 0;
  442. }
  443. if (tx_len != count) {
  444. LLOGW("network_tx expect %d but %d", count, tx_len);
  445. luat_mqtt_close_socket(mqtt_ctrl);
  446. return 0;
  447. }
  448. return count;
  449. }
  450. int luat_mqtt_connect(luat_mqtt_ctrl_t *mqtt_ctrl) {
  451. int ret = 0;
  452. mqtt_ctrl->error_state=0;
  453. if (!mqtt_ctrl->mqtt_packet_buffer) {
  454. mqtt_ctrl->mqtt_packet_buffer = luat_heap_malloc(mqtt_ctrl->rxbuff_size+4);
  455. }
  456. if (mqtt_ctrl->mqtt_packet_buffer == NULL){
  457. return -1;
  458. }
  459. memset(mqtt_ctrl->mqtt_packet_buffer, 0, mqtt_ctrl->rxbuff_size+4);
  460. const char *hostname = mqtt_ctrl->host;
  461. uint16_t port = mqtt_ctrl->remote_port;
  462. uint16_t keepalive = mqtt_ctrl->keepalive;
  463. LLOGD("host %s port %d keepalive %d", hostname, port, keepalive);
  464. mqtt_set_alive(&(mqtt_ctrl->broker), keepalive);
  465. ret = network_connect(mqtt_ctrl->netc, hostname, strlen(hostname), NULL, port, 0) < 0;
  466. LLOGD("network_connect ret %d", ret);
  467. if (ret < 0) {
  468. network_close(mqtt_ctrl->netc, 0);
  469. return -1;
  470. }
  471. mqtt_ctrl->mqtt_state = MQTT_STATE_SCONNECT;
  472. return 0;
  473. }
  474. int luat_mqtt_set_will(luat_mqtt_ctrl_t *mqtt_ctrl, const char* topic,
  475. const char* payload, size_t payload_len,
  476. uint8_t qos, size_t retain) {
  477. if (mqtt_ctrl == NULL || mqtt_ctrl->netc == NULL)
  478. return -1;
  479. return mqtt_set_will(&mqtt_ctrl->broker, topic, payload, payload_len, qos, retain);
  480. }
  481. int luat_mqtt_set_cb(luat_mqtt_ctrl_t *mqtt_ctrl, luat_mqtt_cb_t mqtt_cb){
  482. if (mqtt_ctrl == NULL || mqtt_ctrl->netc == NULL)
  483. return -1;
  484. mqtt_ctrl->mqtt_cb = mqtt_cb;
  485. return 0;
  486. }
  487. LUAT_MQTT_STATE_E luat_mqtt_state_get(luat_mqtt_ctrl_t *mqtt_ctrl){
  488. return mqtt_ctrl->mqtt_state;
  489. }