luat_ftp_client.c 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824
  1. #include "luat_base.h"
  2. #include "luat_network_adapter.h"
  3. #include "luat_rtos.h"
  4. #include "luat_fs.h"
  5. #include "luat_mem.h"
  6. #include "luat_ftp.h"
  7. #define LUAT_LOG_TAG "ftp"
  8. #include "luat_log.h"
  9. #undef LLOGD
  10. #define LLOGD(format, ...) do {if (g_s_ftp.debug_onoff) {luat_log_log(LUAT_LOG_DEBUG, LUAT_LOG_TAG, format, ##__VA_ARGS__);}} while(0)
  11. luat_ftp_ctrl_t g_s_ftp;
  12. uint32_t luat_ftp_release(void) {
  13. if (!g_s_ftp.network) return 0;
  14. if (g_s_ftp.network->cmd_netc){
  15. network_force_close_socket(g_s_ftp.network->cmd_netc);
  16. network_release_ctrl(g_s_ftp.network->cmd_netc);
  17. g_s_ftp.network->cmd_netc = NULL;
  18. }
  19. if (g_s_ftp.network->data_netc){
  20. network_force_close_socket(g_s_ftp.network->data_netc);
  21. network_release_ctrl(g_s_ftp.network->data_netc);
  22. g_s_ftp.network->data_netc = NULL;
  23. }
  24. luat_heap_free(g_s_ftp.network);
  25. g_s_ftp.network = NULL;
  26. return 0;
  27. }
  28. static uint32_t luat_ftp_data_send(luat_ftp_ctrl_t *ftp_ctrl, uint8_t* send_data, uint32_t send_len) {
  29. if (send_len == 0)
  30. return 0;
  31. uint32_t tx_len = 0;
  32. LLOGD("luat_ftp_data_send data:%d",send_len);
  33. network_tx(g_s_ftp.network->data_netc, send_data, send_len, 0, NULL, 0, &tx_len, 0);
  34. return tx_len;
  35. }
  36. static uint32_t luat_ftp_cmd_send(luat_ftp_ctrl_t *ftp_ctrl, uint8_t* send_data, uint32_t send_len,uint32_t timeout_ms) {
  37. if (send_len == 0)
  38. return 0;
  39. uint32_t tx_len = 0;
  40. LLOGD("luat_ftp_cmd_send data:%.*s",send_len,send_data);
  41. network_tx(g_s_ftp.network->cmd_netc, send_data, send_len, 0, NULL, 0, &tx_len, timeout_ms);
  42. return tx_len;
  43. }
  44. static int luat_ftp_cmd_recv(luat_ftp_ctrl_t *ftp_ctrl,uint8_t *recv_data,uint32_t *recv_len,uint32_t timeout_ms){
  45. uint8_t is_break = 0,is_timeout = 0;
  46. int ret = network_wait_rx(g_s_ftp.network->cmd_netc, timeout_ms, &is_break, &is_timeout);
  47. LLOGD("luat_ftp_cmd_recv network_wait_rx ret:%d is_break:%d is_timeout:%d",ret,is_break,is_timeout);
  48. if (ret)
  49. return -1;
  50. if (is_timeout)
  51. return 1;
  52. else if (is_break)
  53. return 2;
  54. return network_rx(g_s_ftp.network->cmd_netc, recv_data, FTP_CMD_RECV_MAX, 0, NULL, NULL, recv_len);
  55. }
  56. static int32_t luat_ftp_data_callback(void *data, void *param){
  57. OS_EVENT *event = (OS_EVENT *)data;
  58. uint8_t *rx_buffer;
  59. int ret = 0;
  60. uint32_t rx_len = 0;
  61. if (!g_s_ftp.network)
  62. {
  63. return 0;
  64. }
  65. // LLOGD("event->ID %d LINK %d ON_LINE %d EVENT %d TX_OK %d CLOSED %d",event->ID,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);
  66. LLOGD("luat_ftp_data_callback %d %d",event->ID - EV_NW_RESULT_BASE, event->Param1);
  67. if (event->Param1){
  68. if (EV_NW_RESULT_CONNECT == event->ID)
  69. {
  70. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0xffffffff, 0, 0, LUAT_WAIT_FOREVER);
  71. }
  72. else
  73. {
  74. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
  75. }
  76. return -1;
  77. }
  78. switch (event->ID)
  79. {
  80. case EV_NW_RESULT_TX:
  81. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_TX_DONE, 0, 0, 0, LUAT_WAIT_FOREVER);
  82. break;
  83. case EV_NW_RESULT_EVENT:
  84. rx_buffer = NULL;
  85. uint8_t tmpbuff[4];
  86. do
  87. {
  88. // 先读取长度
  89. ret = network_rx(g_s_ftp.network->data_netc, NULL, 0, 0, NULL, NULL, &rx_len);
  90. if (rx_len <= 0) {
  91. // 没数据? 那也读一次, 然后退出
  92. network_rx(g_s_ftp.network->data_netc, tmpbuff, 4, 0, NULL, NULL, &rx_len);
  93. break;
  94. }
  95. if (rx_len > 2048)
  96. rx_len = 2048;
  97. rx_buffer = luat_heap_malloc(rx_len);
  98. // 如果rx_buffer == NULL, 内存炸了
  99. if (rx_buffer == NULL) {
  100. LLOGE("out of memory when malloc ftp buff");
  101. network_close(g_s_ftp.network->data_netc, 0);
  102. return -1;
  103. }
  104. ret = network_rx(g_s_ftp.network->data_netc, rx_buffer, rx_len, 0, NULL, NULL, &rx_len);
  105. // LLOGD("luat_ftp_data_callback network_rx ret:%d rx_len:%d",ret,rx_len);
  106. if (!ret && rx_len > 0)
  107. {
  108. if (g_s_ftp.fd)
  109. {
  110. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_WRITE_FILE, (uint32_t)rx_buffer, rx_len, 0, LUAT_WAIT_FOREVER);
  111. rx_buffer = NULL;
  112. continue;
  113. }
  114. else
  115. {
  116. OS_BufferWrite(&g_s_ftp.result_buffer, rx_buffer, rx_len);
  117. }
  118. }
  119. luat_heap_free(rx_buffer);
  120. rx_buffer = NULL;
  121. } while (!ret && rx_len);
  122. if (rx_buffer)
  123. luat_heap_free(rx_buffer);
  124. rx_buffer = NULL;
  125. break;
  126. case EV_NW_RESULT_CLOSE:
  127. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
  128. return 0;
  129. break;
  130. case EV_NW_RESULT_CONNECT:
  131. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0, 0, 0, LUAT_WAIT_FOREVER);
  132. break;
  133. case EV_NW_RESULT_LINK:
  134. return 0;
  135. }
  136. ret = network_wait_event(g_s_ftp.network->data_netc, NULL, 0, NULL);
  137. if (ret < 0){
  138. network_close(g_s_ftp.network->data_netc, 0);
  139. return -1;
  140. }
  141. return 0;
  142. }
  143. static int32_t ftp_task_cb(void *pdata, void *param)
  144. {
  145. OS_EVENT *event = pdata;
  146. if (event->ID >= FTP_EVENT_LOGIN && event->ID <= FTP_EVENT_PUSH)
  147. {
  148. LLOGE("last cmd not finish, ignore %d,%u,%u,%x", event->ID - USER_EVENT_ID_START, event->Param1, event->Param2, param);
  149. return -1;
  150. }
  151. switch(event->ID)
  152. {
  153. case FTP_EVENT_DATA_WRITE_FILE:
  154. if (g_s_ftp.fd)
  155. {
  156. luat_fs_fwrite((void*)event->Param1, event->Param2, 1, g_s_ftp.fd);
  157. luat_heap_free((void*)event->Param1);
  158. }
  159. break;
  160. case FTP_EVENT_DATA_TX_DONE:
  161. g_s_ftp.network->upload_done_size = (size_t)g_s_ftp.network->data_netc->ack_size;
  162. if (g_s_ftp.network->upload_done_size >= g_s_ftp.network->local_file_size)
  163. {
  164. LLOGD("ftp data upload done!");
  165. network_close(g_s_ftp.network->data_netc, 0);
  166. }
  167. break;
  168. case FTP_EVENT_DATA_CONNECT:
  169. if (g_s_ftp.network->data_netc_connecting)
  170. {
  171. g_s_ftp.network->data_netc_connecting = 0;
  172. g_s_ftp.network->data_netc_online = !event->Param1;
  173. }
  174. break;
  175. case FTP_EVENT_DATA_CLOSED:
  176. LLOGD("ftp data channel close");
  177. g_s_ftp.network->data_netc_online = 0;
  178. if (g_s_ftp.network->data_netc)
  179. {
  180. network_force_close_socket(g_s_ftp.network->data_netc);
  181. network_release_ctrl(g_s_ftp.network->data_netc);
  182. g_s_ftp.network->data_netc = NULL;
  183. }
  184. break;
  185. case FTP_EVENT_CLOSE:
  186. g_s_ftp.is_run = 0;
  187. break;
  188. default:
  189. // LLOGE("ignore %x,%x,%x", event->ID, param, EV_NW_RESULT_EVENT);
  190. break;
  191. }
  192. return 0;
  193. }
  194. static int luat_ftp_pasv_connect(luat_ftp_ctrl_t *ftp_ctrl,uint32_t timeout_ms){
  195. char h1[4]={0},h2[4]={0},h3[4]={0},h4[4]={0},p1[4]={0},p2[4]={0},data_addr[64]={0};
  196. uint8_t port1,port2;
  197. uint16_t data_port;
  198. luat_ftp_cmd_send(&g_s_ftp, (uint8_t*)"PASV\r\n", strlen("PASV\r\n"),FTP_SOCKET_TIMEOUT);
  199. // g_s_ftp.data_transfer_done = 0;
  200. int ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  201. if (ret){
  202. return -1;
  203. }else{
  204. LLOGD("luat_ftp_pasv_connect cmd_recv_data %s",g_s_ftp.network->cmd_recv_data);
  205. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_ENTER_PASSIVE, 3)){
  206. LLOGD("ftp pasv_connect wrong");
  207. return -1;
  208. }
  209. }
  210. char *temp = memchr(g_s_ftp.network->cmd_recv_data, '(', strlen((const char *)(g_s_ftp.network->cmd_recv_data)));
  211. char *temp1 = memchr(temp+1, ',', strlen(temp)-1);
  212. memcpy(h1, temp+1, temp1-temp-1);
  213. char *temp2 = memchr(temp1+1, ',', strlen(temp1)-1);
  214. memcpy(h2, temp1+1, temp2-temp1-1);
  215. char *temp3 = memchr(temp2+1, ',', strlen(temp2)-1);
  216. memcpy(h3, temp2+1, temp3-temp2-1);
  217. char *temp4 = memchr(temp3+1, ',', strlen(temp3)-1);
  218. memcpy(h4, temp3+1, temp4-temp3-1);
  219. char *temp5 = memchr(temp4+1, ',', strlen(temp4)-1);
  220. memcpy(p1, temp4+1, temp5-temp4-1);
  221. char *temp6 = memchr(temp5+1, ')', strlen(temp5)-1);
  222. memcpy(p2, temp5+1, temp6-temp5-1);
  223. snprintf_(data_addr, 64, "%s.%s.%s.%s",h1,h2,h3,h4);
  224. port1 = (uint8_t)atoi(p1);
  225. port2 = (uint8_t)atoi(p2);
  226. data_port = port1 * 256 + port2;
  227. // 由于服务器配置问题, 可能访问内网ip,空ip,0.0.0.0等不可使用的被动IP地址
  228. LLOGD("data_addr:%s data_port:%d",data_addr,data_port);
  229. if (memcmp(data_addr,"172.",4)==0 ||
  230. memcmp(data_addr,"192.",4)==0 ||
  231. memcmp(data_addr,"10.",3)==0||
  232. memcmp(data_addr,"127.0.0.1",9)==0||
  233. memcmp(data_addr,"169.254.0.0",11)==0||
  234. memcmp(data_addr,"169.254.0.16",12)==0||
  235. memcmp(data_addr,"0.",2) == 0){
  236. LLOGW("服务器回应不可路由的地址(%s)。使用服务器地址代替 %s", data_addr, g_s_ftp.network->addr);
  237. memset(data_addr,0,64);
  238. LLOGD("g_s_ftp.network->addr:%s",g_s_ftp.network->addr);
  239. memcpy(data_addr, g_s_ftp.network->addr, strlen(g_s_ftp.network->addr)+1);
  240. }
  241. LLOGD("data_addr:%s data_port:%d",data_addr,data_port);
  242. if (g_s_ftp.network->data_netc)
  243. {
  244. LLOGE("data_netc already create");
  245. return -1;
  246. }
  247. g_s_ftp.network->data_netc = network_alloc_ctrl(g_s_ftp.network->adapter_index);
  248. if (!g_s_ftp.network->data_netc){
  249. LLOGE("data_netc create fail");
  250. return -1;
  251. }
  252. network_init_ctrl(g_s_ftp.network->data_netc,NULL, luat_ftp_data_callback, g_s_ftp.network);
  253. network_set_base_mode(g_s_ftp.network->data_netc, 1, 10000, 0, 0, 0, 0);
  254. network_set_local_port(g_s_ftp.network->data_netc, 0);
  255. network_deinit_tls(g_s_ftp.network->data_netc);
  256. if(network_connect(g_s_ftp.network->data_netc, data_addr, strlen(data_addr), NULL, data_port, 0)<0){
  257. LLOGE("ftp data network connect fail");
  258. network_force_close_socket(g_s_ftp.network->data_netc);
  259. network_release_ctrl(g_s_ftp.network->data_netc);
  260. g_s_ftp.network->data_netc = NULL;
  261. return -1;
  262. }
  263. uint8_t is_timeout;
  264. OS_EVENT event;
  265. g_s_ftp.network->data_netc_connecting = 1;
  266. g_s_ftp.network->data_netc_online = 0;
  267. while(g_s_ftp.network->data_netc_connecting)
  268. {
  269. if (network_wait_event(g_s_ftp.network->cmd_netc, &event, timeout_ms, &is_timeout))
  270. {
  271. return -1;
  272. }
  273. if (is_timeout)
  274. {
  275. return -1;
  276. }
  277. if (event.ID)
  278. {
  279. ftp_task_cb(&event, NULL);
  280. }
  281. else
  282. {
  283. if (g_s_ftp.network->cmd_netc->new_rx_flag)
  284. {
  285. network_rx(g_s_ftp.network->cmd_netc, g_s_ftp.network->cmd_recv_data, 1024, 0, NULL, NULL, &g_s_ftp.network->cmd_recv_len);
  286. LLOGD("ftp cmd rx %.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  287. }
  288. }
  289. }
  290. if (g_s_ftp.network->data_netc_online)
  291. {
  292. LLOGD("ftp pasv_connect ok");
  293. return 0;
  294. }
  295. return -1;
  296. }
  297. static int ftp_login(void)
  298. {
  299. int ret;
  300. if(network_connect(g_s_ftp.network->cmd_netc, g_s_ftp.network->addr, strlen(g_s_ftp.network->addr), NULL, g_s_ftp.network->port, FTP_SOCKET_TIMEOUT)){
  301. LLOGE("ftp network_connect fail");
  302. return -1;
  303. }
  304. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  305. if (ret){
  306. return -1;
  307. }else{
  308. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_SERVICE_NEW_OK, 3)){
  309. LLOGE("ftp connect error");
  310. return -1;
  311. }
  312. }
  313. LLOGD("ftp connect ok");
  314. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  315. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "USER %s\r\n",g_s_ftp.network->username);
  316. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)(g_s_ftp.network->cmd_send_data)),FTP_SOCKET_TIMEOUT);
  317. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  318. if (ret){
  319. LLOGE("ftp username wrong");
  320. return -1;
  321. }else{
  322. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_USERNAME_OK, 3)){
  323. LLOGE("ftp username wrong");
  324. return -1;
  325. }
  326. }
  327. LLOGD("ftp username ok");
  328. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  329. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "PASS %s\r\n",g_s_ftp.network->password);
  330. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)(g_s_ftp.network->cmd_send_data)),FTP_SOCKET_TIMEOUT);
  331. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  332. if (ret){
  333. LLOGE("ftp login wrong");
  334. return -1;
  335. }else{
  336. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_LOGIN_OK, 3)){
  337. LLOGE("ftp login wrong");
  338. return -1;
  339. }
  340. }
  341. LLOGD("ftp login ok");
  342. return 0;
  343. }
  344. static void l_ftp_cb(FTP_SUCCESS_STATE_e state){
  345. if (g_s_ftp.network->ftp_cb){
  346. luat_ftp_cb_t ftp_cb = g_s_ftp.network->ftp_cb;
  347. ftp_cb(&g_s_ftp,state);
  348. }
  349. #ifndef __LUATOS__
  350. OS_DeInitBuffer(&g_s_ftp.result_buffer);
  351. #endif
  352. }
  353. static int find_newline(void)
  354. {
  355. uint32_t pos = 0;
  356. while (pos < g_s_ftp.network->cmd_recv_len)
  357. {
  358. if(('\r' == g_s_ftp.network->cmd_recv_data[pos]) || ('\n' == g_s_ftp.network->cmd_recv_data[pos]))
  359. {
  360. return pos;
  361. }
  362. else
  363. {
  364. pos++;
  365. }
  366. }
  367. return -1;
  368. }
  369. static int pasv_recv(void)
  370. {
  371. int ret;
  372. int pos;
  373. uint8_t rx_finish = 0;
  374. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  375. if (ret){
  376. LLOGE("pasv_recv %s error:%d", g_s_ftp.network->cmd_recv_data, ret);
  377. return -1;
  378. }
  379. LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ ,g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  380. g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
  381. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS_OK, 3) && memcmp(g_s_ftp.network->cmd_recv_data, FTP_DATA_CON_OPEN, 3)){
  382. return -1;
  383. }
  384. pos = find_newline();
  385. if (pos >= 0)
  386. {
  387. memmove(g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_data + pos, g_s_ftp.network->cmd_recv_len - pos);
  388. g_s_ftp.network->cmd_recv_len -= pos;
  389. g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
  390. if (strstr((const char *)(g_s_ftp.network->cmd_recv_data), FTP_CLOSE_CONNECT))
  391. {
  392. rx_finish = 1;
  393. }
  394. }
  395. LLOGD("%s %d rx_finish:%d data_netc_online:%d Pos:%d ",__FUNCTION__ ,__LINE__ ,rx_finish, g_s_ftp.network->data_netc_online, g_s_ftp.result_buffer.Pos);
  396. while(!rx_finish) //data通道未断开或者已经接收到数据了
  397. {
  398. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  399. if (ret<0){
  400. LLOGD("%s %d rx error!%d",__FUNCTION__ ,__LINE__ , ret);
  401. return -1;
  402. } else if (!ret) {
  403. LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ , g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  404. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
  405. return -1;
  406. }
  407. else {
  408. rx_finish = 1;
  409. }
  410. }
  411. }
  412. if (!rx_finish) { //没接收完就断开了
  413. LLOGD("???");
  414. return -1;
  415. }
  416. //主动关闭掉接收
  417. if (g_s_ftp.network->data_netc_online && g_s_ftp.network->data_netc){
  418. network_force_close_socket(g_s_ftp.network->data_netc);
  419. network_release_ctrl(g_s_ftp.network->data_netc);
  420. g_s_ftp.network->data_netc = NULL;
  421. }
  422. return 0;
  423. }
  424. static void ftp_task(void *param){
  425. FTP_SUCCESS_STATE_e ftp_state = FTP_SUCCESS_NO_DATE;
  426. int ret;
  427. int count = 0;
  428. luat_rtos_task_handle task_handle = g_s_ftp.task_handle;
  429. OS_EVENT task_event;
  430. uint8_t is_timeout = 0;
  431. g_s_ftp.is_run = 1;
  432. luat_rtos_event_recv(g_s_ftp.task_handle, FTP_EVENT_LOGIN, &task_event, NULL, LUAT_WAIT_FOREVER);
  433. if (ftp_login()){
  434. LLOGE("ftp login fail");
  435. ftp_state = FTP_ERROR;
  436. l_ftp_cb(ftp_state);
  437. luat_ftp_release();
  438. g_s_ftp.task_handle = NULL;
  439. luat_rtos_task_delete(task_handle);
  440. return;
  441. }else{
  442. l_ftp_cb(ftp_state);
  443. }
  444. while (g_s_ftp.is_run) {
  445. is_timeout = 0;
  446. ret = network_wait_event(g_s_ftp.network->cmd_netc, &task_event, 3600000, &is_timeout);
  447. if (ret < 0){
  448. LLOGE("ftp network error");
  449. goto wait_event_and_out;
  450. }else if (is_timeout || !task_event.ID){
  451. if (g_s_ftp.network->cmd_netc->new_rx_flag){
  452. network_rx(g_s_ftp.network->cmd_netc, g_s_ftp.network->cmd_recv_data, 1024, 0, NULL, NULL, &ret);
  453. LLOGD("ftp rx %dbyte", ret);
  454. }
  455. continue;
  456. }
  457. ftp_state = FTP_SUCCESS_NO_DATE;
  458. switch (task_event.ID)
  459. {
  460. case FTP_EVENT_LOGIN:
  461. break;
  462. case FTP_EVENT_PULL:
  463. if (g_s_ftp.network->data_netc)
  464. {
  465. network_force_close_socket(g_s_ftp.network->data_netc);
  466. network_release_ctrl(g_s_ftp.network->data_netc);
  467. g_s_ftp.network->data_netc = NULL;
  468. }
  469. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  470. LLOGE("ftp pasv_connect fail");
  471. goto operation_failed;
  472. }
  473. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "RETR %s\r\n",g_s_ftp.network->remote_name);
  474. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)g_s_ftp.network->cmd_send_data),FTP_SOCKET_TIMEOUT);
  475. if (pasv_recv()) goto operation_failed;
  476. if (g_s_ftp.fd){
  477. luat_fs_fclose(g_s_ftp.fd);
  478. g_s_ftp.fd = NULL;
  479. }
  480. l_ftp_cb(ftp_state);
  481. break;
  482. case FTP_EVENT_PUSH:
  483. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  484. LLOGD("ftp pasv_connect fail");
  485. goto operation_failed;
  486. }
  487. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  488. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "STOR %s\r\n",g_s_ftp.network->remote_name);
  489. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)g_s_ftp.network->cmd_send_data),FTP_SOCKET_TIMEOUT);
  490. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  491. if (ret){
  492. goto operation_failed;
  493. }else{
  494. LLOGD("%.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  495. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS_OK, 3) && memcmp(g_s_ftp.network->cmd_recv_data, FTP_DATA_CON_OPEN, 3)){
  496. LLOGD("ftp STOR wrong");
  497. goto operation_failed;
  498. }
  499. }
  500. uint8_t* buff = luat_heap_malloc(PUSH_BUFF_SIZE);
  501. int offset = 0;
  502. g_s_ftp.network->upload_done_size = 0;
  503. while (1) {
  504. memset(buff, 0, PUSH_BUFF_SIZE);
  505. int len = luat_fs_fread(buff, sizeof(uint8_t), PUSH_BUFF_SIZE, g_s_ftp.fd);
  506. if (len < 1)
  507. break;
  508. luat_ftp_data_send(&g_s_ftp, buff, len);
  509. offset += len;
  510. }
  511. luat_heap_free(buff);
  512. LLOGD("offset:%d file_size:%d",offset,g_s_ftp.network->local_file_size);
  513. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  514. if (g_s_ftp.network->upload_done_size != g_s_ftp.network->local_file_size)
  515. {
  516. LLOGE("upload not finish !!! %d,%d", g_s_ftp.network->upload_done_size, g_s_ftp.network->local_file_size);
  517. }
  518. if (ret){
  519. goto operation_failed;
  520. }else{
  521. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
  522. LLOGD("ftp STOR wrong");
  523. }
  524. }
  525. while (count<3 && g_s_ftp.network->data_netc_online!=0){
  526. luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT/3);
  527. count++;
  528. }
  529. if (g_s_ftp.fd){
  530. luat_fs_fclose(g_s_ftp.fd);
  531. g_s_ftp.fd = NULL;
  532. }
  533. l_ftp_cb(ftp_state);
  534. break;
  535. case FTP_EVENT_CLOSE:
  536. g_s_ftp.is_run = 0;
  537. break;
  538. case FTP_EVENT_COMMAND:
  539. OS_DeInitBuffer(&g_s_ftp.result_buffer);
  540. if(!memcmp(g_s_ftp.network->cmd_send_data, "LIST", 4))
  541. {
  542. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  543. LLOGD("ftp pasv_connect fail");
  544. goto operation_failed;
  545. }
  546. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)(g_s_ftp.network->cmd_send_data)),FTP_SOCKET_TIMEOUT);
  547. if (pasv_recv()) goto operation_failed;
  548. ftp_state = FTP_SUCCESS_DATE;
  549. l_ftp_cb(ftp_state);
  550. break;
  551. }
  552. luat_ftp_cmd_send(&g_s_ftp, g_s_ftp.network->cmd_send_data, strlen((const char *)(g_s_ftp.network->cmd_send_data)),FTP_SOCKET_TIMEOUT);
  553. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  554. if (ret){
  555. goto operation_failed;
  556. }else{
  557. if (memcmp(g_s_ftp.network->cmd_send_data, "NOOP", 4)==0){
  558. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
  559. LLOGD("ftp COMMAND wrong");
  560. }
  561. }else if(memcmp(g_s_ftp.network->cmd_send_data, "TYPE", 4)==0){
  562. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
  563. LLOGD("ftp COMMAND wrong");
  564. }
  565. }else if(memcmp(g_s_ftp.network->cmd_send_data, "SYST", 4)==0){
  566. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_SYSTEM_TYPE, 3)){
  567. LLOGD("ftp COMMAND wrong");
  568. }
  569. }else if(memcmp(g_s_ftp.network->cmd_send_data, "PWD", 3)==0){
  570. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
  571. LLOGD("ftp COMMAND wrong");
  572. }
  573. }else if(memcmp(g_s_ftp.network->cmd_send_data, "MKD", 3)==0){
  574. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
  575. LLOGD("ftp COMMAND wrong");
  576. }
  577. }else if(memcmp(g_s_ftp.network->cmd_send_data, "CWD", 3)==0){
  578. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  579. LLOGD("ftp COMMAND wrong");
  580. }
  581. }else if(memcmp(g_s_ftp.network->cmd_send_data, "CDUP", 4)==0){
  582. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  583. LLOGD("ftp COMMAND wrong");
  584. }
  585. }else if(memcmp(g_s_ftp.network->cmd_send_data, "RMD", 3)==0){
  586. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  587. LLOGD("ftp COMMAND wrong");
  588. }
  589. }else if(memcmp(g_s_ftp.network->cmd_send_data, "DELE", 4)==0){
  590. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  591. LLOGD("ftp COMMAND wrong");
  592. }
  593. }else if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_DATA_CON_FAIL, 3)==0){
  594. LLOGD("ftp need pasv_connect");
  595. }
  596. }
  597. OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
  598. ftp_state = FTP_SUCCESS_DATE;
  599. l_ftp_cb(ftp_state);
  600. break;
  601. default:
  602. break;
  603. }
  604. continue;
  605. operation_failed:
  606. LLOGE("ftp operation failed");
  607. if (g_s_ftp.fd){
  608. luat_fs_fclose(g_s_ftp.fd);
  609. g_s_ftp.fd = NULL;
  610. }
  611. ftp_state = FTP_ERROR;
  612. l_ftp_cb(ftp_state);
  613. }
  614. ftp_state = FTP_SUCCESS_NO_DATE;
  615. luat_ftp_cmd_send(&g_s_ftp, (uint8_t*)"QUIT\r\n", strlen("QUIT\r\n"),FTP_SOCKET_TIMEOUT);
  616. ret = luat_ftp_cmd_recv(&g_s_ftp,g_s_ftp.network->cmd_recv_data,&g_s_ftp.network->cmd_recv_len,FTP_SOCKET_TIMEOUT);
  617. if (ret){
  618. ftp_state = FTP_ERROR;
  619. }else{
  620. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONTROL, 3)){
  621. LLOGE("ftp QUIT wrong");
  622. ftp_state = FTP_ERROR;
  623. }
  624. }
  625. OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
  626. if (ftp_state == FTP_SUCCESS_NO_DATE) ftp_state = FTP_SUCCESS_DATE;
  627. l_ftp_cb(ftp_state);
  628. luat_ftp_release();
  629. g_s_ftp.task_handle = NULL;
  630. luat_rtos_task_delete(task_handle);
  631. return;
  632. wait_event_and_out:
  633. while(1)
  634. {
  635. luat_rtos_event_recv(g_s_ftp.task_handle, 0, &task_event, NULL, LUAT_WAIT_FOREVER);
  636. if (task_event.ID >= FTP_EVENT_LOGIN && task_event.ID <= FTP_EVENT_CLOSE)
  637. {
  638. luat_ftp_release();
  639. ftp_state = FTP_ERROR;
  640. l_ftp_cb(ftp_state);
  641. g_s_ftp.task_handle = NULL;
  642. luat_rtos_task_delete(task_handle);
  643. return;
  644. }
  645. }
  646. }
  647. int luat_ftp_login(uint8_t adapter,const char * ip_addr,uint16_t port,const char * username,const char * password,luat_ftp_tls_t* luat_ftp_tls,luat_ftp_cb_t ftp_cb){
  648. if (g_s_ftp.network){
  649. LLOGE("ftp already login, please close first");
  650. return FTP_ERROR_STATE;
  651. }
  652. g_s_ftp.network = (luat_ftp_network_t *)luat_heap_malloc(sizeof(luat_ftp_network_t));
  653. if (!g_s_ftp.network){
  654. LLOGE("out of memory when malloc g_s_ftp.network");
  655. return FTP_ERROR_NO_MEM;
  656. }
  657. memset(g_s_ftp.network, 0, sizeof(luat_ftp_network_t));
  658. if (ftp_cb){
  659. g_s_ftp.network->ftp_cb = ftp_cb;
  660. }
  661. g_s_ftp.network->adapter_index = adapter;
  662. if (g_s_ftp.network->adapter_index >= NW_ADAPTER_QTY){
  663. LLOGE("bad network adapter index %d", g_s_ftp.network->adapter_index);
  664. return FTP_ERROR_STATE;
  665. }
  666. g_s_ftp.network->cmd_netc = network_alloc_ctrl(g_s_ftp.network->adapter_index);
  667. if (!g_s_ftp.network->cmd_netc){
  668. LLOGE("cmd_netc create fail");
  669. return FTP_ERROR_NO_MEM;
  670. }
  671. g_s_ftp.network->port = port;
  672. if (strlen(ip_addr) > 0 && strlen(ip_addr) < 64)
  673. memcpy(g_s_ftp.network->addr, ip_addr, strlen(ip_addr) + 1);
  674. if (strlen(username) > 0 && strlen(username) < 64)
  675. memcpy(g_s_ftp.network->username, username, strlen(username) + 1);
  676. if (strlen(password) > 0 && strlen(password) < 64)
  677. memcpy(g_s_ftp.network->password, password, strlen(password) + 1);
  678. if (luat_ftp_tls == NULL){
  679. network_deinit_tls(g_s_ftp.network->cmd_netc);
  680. }else{
  681. if (network_init_tls(g_s_ftp.network->cmd_netc, (luat_ftp_tls->server_cert || luat_ftp_tls->client_cert)?2:0)){
  682. return FTP_ERROR_CLOSE;
  683. }
  684. if (luat_ftp_tls->server_cert){
  685. network_set_server_cert(g_s_ftp.network->cmd_netc, (const unsigned char *)luat_ftp_tls->server_cert, strlen(luat_ftp_tls->server_cert)+1);
  686. }
  687. if (luat_ftp_tls->client_cert){
  688. network_set_client_cert(g_s_ftp.network->cmd_netc, (const unsigned char *)luat_ftp_tls->client_cert, strlen(luat_ftp_tls->client_cert)+1,
  689. (const unsigned char *)luat_ftp_tls->client_key, strlen(luat_ftp_tls->client_key)+1,
  690. (const unsigned char *)luat_ftp_tls->client_password, strlen(luat_ftp_tls->client_password)+1);
  691. }
  692. }
  693. network_set_ip_invaild(&g_s_ftp.network->ip_addr);
  694. int result = luat_rtos_task_create(&g_s_ftp.task_handle, 2*1024, 10, "ftp", ftp_task, NULL, 16);
  695. if (result) {
  696. LLOGE("创建ftp task失败!! %d", result);
  697. return result;
  698. }
  699. network_init_ctrl(g_s_ftp.network->cmd_netc,g_s_ftp.task_handle, ftp_task_cb, NULL);
  700. network_set_base_mode(g_s_ftp.network->cmd_netc, 1, 30000, 0, 0, 0, 0);
  701. network_set_local_port(g_s_ftp.network->cmd_netc, 0);
  702. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_LOGIN, 0, 0, 0, LUAT_WAIT_FOREVER);
  703. return 0;
  704. }
  705. int luat_ftp_command(const char * command){
  706. if (!g_s_ftp.network){
  707. LLOGE("please login first");
  708. return -1;
  709. }
  710. if (memcmp(command, "NOOP", 4)==0){
  711. LLOGD("command: NOOP");
  712. }else if(memcmp(command, "SYST", 4)==0){
  713. LLOGD("command: SYST");
  714. }else if(memcmp(command, "MKD", 3)==0){
  715. LLOGD("command: MKD");
  716. }else if(memcmp(command, "CWD", 3)==0){
  717. LLOGD("command: CWD");
  718. }else if(memcmp(command, "CDUP", 4)==0){
  719. LLOGD("command: CDUP");
  720. }else if(memcmp(command, "RMD", 3)==0){
  721. LLOGD("command: RMD");
  722. }else if(memcmp(command, "PWD", 3)==0){
  723. LLOGD("command: RMD");
  724. }else if(memcmp(command, "DELE", 4)==0){
  725. LLOGD("command: DELE");
  726. }else if(memcmp(command, "TYPE", 4)==0){
  727. LLOGD("command: TYPE");
  728. }else if(memcmp(command, "LIST", 4)==0){
  729. LLOGD("command: LIST");
  730. }else{
  731. LLOGE("not support cmd:%s",command);
  732. return -1;
  733. }
  734. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  735. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "%s\r\n",command);
  736. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_COMMAND, 0, 0, 0, 0);
  737. return 0;
  738. }
  739. int luat_ftp_pull(const char * local_name,const char * remote_name){
  740. if (!g_s_ftp.network){
  741. LLOGE("please login first");
  742. return -1;
  743. }
  744. luat_fs_remove(local_name);
  745. if (g_s_ftp.fd)
  746. {
  747. luat_fs_fclose(g_s_ftp.fd);
  748. g_s_ftp.fd = NULL;
  749. }
  750. g_s_ftp.fd = luat_fs_fopen(local_name, "wb+");
  751. if (g_s_ftp.fd == NULL) {
  752. LLOGE("open download file fail %s", local_name);
  753. return -1;
  754. }
  755. g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
  756. memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
  757. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PULL, 0, 0, 0, 0);
  758. return 0;
  759. }
  760. int luat_ftp_close(void){
  761. if (!g_s_ftp.network){
  762. LLOGE("please login first");
  763. return -1;
  764. }
  765. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_CLOSE, 0, 0, 0, LUAT_WAIT_FOREVER);
  766. return 0;
  767. }
  768. int luat_ftp_push(const char * local_name,const char * remote_name){
  769. if (!g_s_ftp.network){
  770. LLOGE("please login first");
  771. return -1;
  772. }
  773. if (g_s_ftp.fd)
  774. {
  775. luat_fs_fclose(g_s_ftp.fd);
  776. g_s_ftp.fd = NULL;
  777. }
  778. g_s_ftp.fd = luat_fs_fopen(local_name, "rb");
  779. if (g_s_ftp.fd == NULL) {
  780. LLOGE("open download file fail %s", local_name);
  781. return -1;
  782. }
  783. g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
  784. memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
  785. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PUSH, 0, 0, 0, LUAT_WAIT_FOREVER);
  786. return 0;
  787. }
  788. void luat_ftp_debug(uint8_t on_off) {g_s_ftp.debug_onoff = on_off;}