luat_ftp_client.c 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841
  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 = {0};
  12. static void l_ftp_cb(FTP_SUCCESS_STATE_e state){
  13. if (g_s_ftp.network->ftp_cb){
  14. luat_ftp_cb_t ftp_cb = g_s_ftp.network->ftp_cb;
  15. ftp_cb(&g_s_ftp,state);
  16. }
  17. #ifndef __LUATOS__
  18. OS_DeInitBuffer(&g_s_ftp.result_buffer);
  19. #endif
  20. }
  21. uint32_t luat_ftp_release(void) {
  22. if (!g_s_ftp.network) return 0;
  23. if (g_s_ftp.network->cmd_netc){
  24. network_force_close_socket(g_s_ftp.network->cmd_netc);
  25. network_release_ctrl(g_s_ftp.network->cmd_netc);
  26. g_s_ftp.network->cmd_netc = NULL;
  27. }
  28. if (g_s_ftp.network->data_netc){
  29. network_force_close_socket(g_s_ftp.network->data_netc);
  30. network_release_ctrl(g_s_ftp.network->data_netc);
  31. g_s_ftp.network->data_netc = NULL;
  32. }
  33. luat_heap_free(g_s_ftp.network);
  34. g_s_ftp.network = NULL;
  35. return 0;
  36. }
  37. static uint32_t luat_ftp_data_send(luat_ftp_ctrl_t *ftp_ctrl, uint8_t* send_data, uint32_t send_len) {
  38. if (send_len == 0)
  39. return 0;
  40. uint32_t tx_len = 0;
  41. LLOGD("luat_ftp_data_send data:%d",send_len);
  42. network_tx(g_s_ftp.network->data_netc, send_data, send_len, 0, NULL, 0, &tx_len, 0);
  43. return tx_len;
  44. }
  45. 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) {
  46. if (send_len == 0)
  47. return 0;
  48. uint32_t tx_len = 0;
  49. LLOGD("luat_ftp_cmd_send data:%.*s",send_len,send_data);
  50. network_tx(g_s_ftp.network->cmd_netc, send_data, send_len, 0, NULL, 0, &tx_len, timeout_ms);
  51. return tx_len;
  52. }
  53. static int luat_ftp_cmd_recv(luat_ftp_ctrl_t *ftp_ctrl,uint8_t *recv_data,uint32_t *recv_len,uint32_t timeout_ms){
  54. uint8_t is_break = 0,is_timeout = 0;
  55. int ret = network_wait_rx(g_s_ftp.network->cmd_netc, timeout_ms, &is_break, &is_timeout);
  56. LLOGD("luat_ftp_cmd_recv network_wait_rx ret:%d is_break:%d is_timeout:%d",ret,is_break,is_timeout);
  57. if (ret)
  58. return -1;
  59. if (is_timeout)
  60. return 1;
  61. else if (is_break)
  62. return 2;
  63. return network_rx(g_s_ftp.network->cmd_netc, recv_data, FTP_CMD_RECV_MAX, 0, NULL, NULL, recv_len);
  64. }
  65. static int32_t luat_ftp_data_callback(void *data, void *param){
  66. OS_EVENT *event = (OS_EVENT *)data;
  67. uint8_t *rx_buffer;
  68. int ret = 0;
  69. uint32_t rx_len = 0;
  70. if (!g_s_ftp.network)
  71. {
  72. return 0;
  73. }
  74. // 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);
  75. LLOGD("luat_ftp_data_callback %d %d",event->ID - EV_NW_RESULT_BASE, event->Param1);
  76. if (event->Param1){
  77. if (EV_NW_RESULT_CONNECT == event->ID)
  78. {
  79. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0xffffffff, 0, 0, LUAT_WAIT_FOREVER);
  80. }
  81. else
  82. {
  83. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
  84. }
  85. return -1;
  86. }
  87. switch (event->ID)
  88. {
  89. case EV_NW_RESULT_TX:
  90. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_TX_DONE, 0, 0, 0, LUAT_WAIT_FOREVER);
  91. break;
  92. case EV_NW_RESULT_EVENT:
  93. rx_buffer = NULL;
  94. uint8_t tmpbuff[4];
  95. do{
  96. // 先读取长度
  97. ret = network_rx(g_s_ftp.network->data_netc, NULL, 0, 0, NULL, NULL, &rx_len);
  98. if (rx_len <= 0) {
  99. // 没数据? 那也读一次, 然后退出
  100. network_rx(g_s_ftp.network->data_netc, tmpbuff, 4, 0, NULL, NULL, &rx_len);
  101. break;
  102. }
  103. if (rx_len > 2048)
  104. rx_len = 2048;
  105. rx_buffer = luat_heap_malloc(rx_len);
  106. // 如果rx_buffer == NULL, 内存炸了
  107. if (rx_buffer == NULL) {
  108. LLOGE("out of memory when malloc ftp buff");
  109. network_close(g_s_ftp.network->data_netc, 0);
  110. return -1;
  111. }
  112. ret = network_rx(g_s_ftp.network->data_netc, rx_buffer, rx_len, 0, NULL, NULL, &rx_len);
  113. // LLOGD("luat_ftp_data_callback network_rx ret:%d rx_len:%d",ret,rx_len);
  114. if (!ret && rx_len > 0){
  115. if (g_s_ftp.fd){
  116. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_WRITE_FILE, (uint32_t)rx_buffer, rx_len, 0, LUAT_WAIT_FOREVER);
  117. rx_buffer = NULL;
  118. continue;
  119. }else{
  120. OS_BufferWrite(&g_s_ftp.result_buffer, rx_buffer, rx_len);
  121. }
  122. }
  123. luat_heap_free(rx_buffer);
  124. rx_buffer = NULL;
  125. } while (!ret && rx_len);
  126. if (rx_buffer)
  127. luat_heap_free(rx_buffer);
  128. rx_buffer = NULL;
  129. break;
  130. case EV_NW_RESULT_CLOSE:
  131. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
  132. return 0;
  133. break;
  134. case EV_NW_RESULT_CONNECT:
  135. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0, 0, 0, LUAT_WAIT_FOREVER);
  136. break;
  137. case EV_NW_RESULT_LINK:
  138. return 0;
  139. }
  140. ret = network_wait_event(g_s_ftp.network->data_netc, NULL, 0, NULL);
  141. if (ret < 0){
  142. network_close(g_s_ftp.network->data_netc, 0);
  143. return -1;
  144. }
  145. return 0;
  146. }
  147. static int32_t ftp_task_cb(void *pdata, void *param){
  148. OS_EVENT *event = pdata;
  149. if (event->ID >= FTP_EVENT_LOGIN && event->ID <= FTP_EVENT_PUSH){
  150. LLOGE("last cmd not finish, ignore %d,%u,%u,%x", event->ID - USER_EVENT_ID_START, event->Param1, event->Param2, param);
  151. return -1;
  152. }
  153. switch(event->ID){
  154. case FTP_EVENT_DATA_WRITE_FILE:
  155. if (g_s_ftp.fd){
  156. g_s_ftp.network->download_file_size += event->Param2;
  157. luat_fs_fwrite((void*)event->Param1, event->Param2, 1, g_s_ftp.fd);
  158. luat_heap_free((void*)event->Param1);
  159. }
  160. break;
  161. case FTP_EVENT_DATA_TX_DONE:
  162. g_s_ftp.network->upload_done_size = (size_t)g_s_ftp.network->data_netc->ack_size;
  163. if (g_s_ftp.network->upload_done_size >= g_s_ftp.network->local_file_size){
  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 int find_newline(void){
  345. uint32_t pos = 0;
  346. while (pos < g_s_ftp.network->cmd_recv_len){
  347. if(('\r' == g_s_ftp.network->cmd_recv_data[pos]) || ('\n' == g_s_ftp.network->cmd_recv_data[pos])){
  348. return pos;
  349. }else{
  350. pos++;
  351. }
  352. }
  353. return -1;
  354. }
  355. static int pasv_recv(void){
  356. int ret;
  357. int pos;
  358. uint8_t rx_finish = 0;
  359. 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);
  360. if (ret){
  361. LLOGE("pasv_recv %s error:%d", g_s_ftp.network->cmd_recv_data, ret);
  362. return -1;
  363. }
  364. LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ ,g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  365. g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
  366. 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)){
  367. return -1;
  368. }
  369. pos = find_newline();
  370. if (pos >= 0){
  371. memmove(g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_data + pos, g_s_ftp.network->cmd_recv_len - pos);
  372. g_s_ftp.network->cmd_recv_len -= pos;
  373. g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
  374. if (strstr((const char *)(g_s_ftp.network->cmd_recv_data), FTP_CLOSE_CONNECT)){
  375. rx_finish = 1;
  376. }
  377. }
  378. 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);
  379. while(!rx_finish || ((g_s_ftp.network->remote_file_size && (g_s_ftp.network->remote_file_size != g_s_ftp.network->download_file_size)))) //data通道未断开或者已经接收到数据了
  380. {
  381. 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/10);
  382. if (ret<0){
  383. LLOGD("%s %d rx error!%d",__FUNCTION__ ,__LINE__ , ret);
  384. return -1;
  385. } else if (!ret) {
  386. LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ , g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  387. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
  388. return -1;
  389. }
  390. else {
  391. rx_finish = 1;
  392. }
  393. }
  394. }
  395. // 数据接收完了主动关闭掉接收
  396. if (g_s_ftp.network->data_netc_online && g_s_ftp.network->data_netc){
  397. network_force_close_socket(g_s_ftp.network->data_netc);
  398. network_release_ctrl(g_s_ftp.network->data_netc);
  399. g_s_ftp.network->data_netc = NULL;
  400. }
  401. return 0;
  402. }
  403. static int string2int(char *str,size_t len){
  404. if(len==0) return 0;
  405. char flag = '+';
  406. int res = 0;
  407. size_t count = 0;
  408. if(*str=='-'){
  409. ++str;
  410. flag = '-';
  411. }
  412. while(*str>=48 && *str<=57 && count<len){
  413. res = 10*res+ *str++-48;
  414. count++;
  415. }
  416. if(flag == '-')
  417. res = -res;
  418. return res;
  419. }
  420. static void ftp_task(void *param){
  421. FTP_SUCCESS_STATE_e ftp_state = FTP_SUCCESS_NO_DATE;
  422. int ret;
  423. int count = 0;
  424. luat_rtos_task_handle task_handle = g_s_ftp.task_handle;
  425. OS_EVENT task_event;
  426. uint8_t is_timeout = 0;
  427. g_s_ftp.is_run = 1;
  428. luat_rtos_event_recv(g_s_ftp.task_handle, FTP_EVENT_LOGIN, &task_event, NULL, LUAT_WAIT_FOREVER);
  429. if (ftp_login()){
  430. LLOGE("ftp login fail");
  431. ftp_state = FTP_ERROR;
  432. l_ftp_cb(ftp_state);
  433. luat_ftp_release();
  434. g_s_ftp.task_handle = NULL;
  435. luat_rtos_task_delete(task_handle);
  436. return;
  437. }else{
  438. l_ftp_cb(ftp_state);
  439. }
  440. while (g_s_ftp.is_run) {
  441. is_timeout = 0;
  442. ret = network_wait_event(g_s_ftp.network->cmd_netc, &task_event, 3600000, &is_timeout);
  443. if (ret < 0){
  444. LLOGE("ftp network error");
  445. goto wait_event_and_out;
  446. }else if (is_timeout || !task_event.ID){
  447. if (g_s_ftp.network->cmd_netc->new_rx_flag){
  448. network_rx(g_s_ftp.network->cmd_netc, g_s_ftp.network->cmd_recv_data, 1024, 0, NULL, NULL, &ret);
  449. LLOGD("ftp rx %dbyte", ret);
  450. }
  451. continue;
  452. }
  453. ftp_state = FTP_SUCCESS_NO_DATE;
  454. switch (task_event.ID)
  455. {
  456. case FTP_EVENT_LOGIN:
  457. break;
  458. case FTP_EVENT_PULL:
  459. g_s_ftp.network->remote_file_size = 0;
  460. g_s_ftp.network->download_file_size = 0;
  461. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  462. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "SIZE %s\r\n",g_s_ftp.network->remote_name);
  463. 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);
  464. 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);
  465. if (!ret){
  466. if (!memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS, 3)){
  467. g_s_ftp.network->remote_file_size = string2int((char*)(g_s_ftp.network->cmd_recv_data+4),g_s_ftp.network->cmd_recv_len-4);
  468. }
  469. }
  470. // LLOGD("remote_file_size:%d",g_s_ftp.network->remote_file_size);
  471. if (g_s_ftp.network->data_netc){
  472. network_force_close_socket(g_s_ftp.network->data_netc);
  473. network_release_ctrl(g_s_ftp.network->data_netc);
  474. g_s_ftp.network->data_netc = NULL;
  475. }
  476. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  477. LLOGE("ftp pasv_connect fail");
  478. goto operation_failed;
  479. }
  480. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "RETR %s\r\n",g_s_ftp.network->remote_name);
  481. 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);
  482. if (pasv_recv()) goto operation_failed;
  483. if (g_s_ftp.fd){
  484. luat_fs_fclose(g_s_ftp.fd);
  485. g_s_ftp.fd = NULL;
  486. }
  487. l_ftp_cb(ftp_state);
  488. break;
  489. case FTP_EVENT_PUSH:
  490. if (g_s_ftp.network->data_netc){
  491. network_force_close_socket(g_s_ftp.network->data_netc);
  492. network_release_ctrl(g_s_ftp.network->data_netc);
  493. g_s_ftp.network->data_netc = NULL;
  494. }
  495. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  496. LLOGD("ftp pasv_connect fail");
  497. goto operation_failed;
  498. }
  499. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  500. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "STOR %s\r\n",g_s_ftp.network->remote_name);
  501. 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);
  502. 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);
  503. if (ret){
  504. goto operation_failed;
  505. }else{
  506. LLOGD("%.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
  507. 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)){
  508. LLOGD("ftp STOR wrong");
  509. goto operation_failed;
  510. }
  511. }
  512. uint8_t* buff = luat_heap_malloc(PUSH_BUFF_SIZE);
  513. int offset = 0;
  514. g_s_ftp.network->upload_done_size = 0;
  515. while (1) {
  516. memset(buff, 0, PUSH_BUFF_SIZE);
  517. int len = luat_fs_fread(buff, sizeof(uint8_t), PUSH_BUFF_SIZE, g_s_ftp.fd);
  518. if (len < 1)
  519. break;
  520. luat_ftp_data_send(&g_s_ftp, buff, len);
  521. offset += len;
  522. }
  523. luat_heap_free(buff);
  524. LLOGD("offset:%d file_size:%d",offset,g_s_ftp.network->local_file_size);
  525. 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);
  526. if (g_s_ftp.network->upload_done_size != g_s_ftp.network->local_file_size)
  527. {
  528. LLOGE("upload not finish !!! %d,%d", g_s_ftp.network->upload_done_size, g_s_ftp.network->local_file_size);
  529. }
  530. if (ret){
  531. goto operation_failed;
  532. }else{
  533. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
  534. LLOGD("ftp STOR wrong");
  535. }
  536. }
  537. while (count<3 && g_s_ftp.network->data_netc_online!=0){
  538. 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);
  539. count++;
  540. }
  541. if (g_s_ftp.fd){
  542. luat_fs_fclose(g_s_ftp.fd);
  543. g_s_ftp.fd = NULL;
  544. }
  545. l_ftp_cb(ftp_state);
  546. break;
  547. case FTP_EVENT_CLOSE:
  548. g_s_ftp.is_run = 0;
  549. break;
  550. case FTP_EVENT_COMMAND:
  551. OS_DeInitBuffer(&g_s_ftp.result_buffer);
  552. if(!memcmp(g_s_ftp.network->cmd_send_data, "LIST", 4))
  553. {
  554. if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
  555. LLOGD("ftp pasv_connect fail");
  556. goto operation_failed;
  557. }
  558. 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);
  559. if (pasv_recv()) goto operation_failed;
  560. ftp_state = FTP_SUCCESS_DATE;
  561. l_ftp_cb(ftp_state);
  562. break;
  563. }
  564. 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);
  565. 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);
  566. if (ret){
  567. goto operation_failed;
  568. }else{
  569. if (memcmp(g_s_ftp.network->cmd_send_data, "NOOP", 4)==0){
  570. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
  571. LLOGD("ftp COMMAND wrong");
  572. }
  573. }else if(memcmp(g_s_ftp.network->cmd_send_data, "TYPE", 4)==0){
  574. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
  575. LLOGD("ftp COMMAND wrong");
  576. }
  577. }else if(memcmp(g_s_ftp.network->cmd_send_data, "SIZE", 4)==0){
  578. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS, 3)){
  579. LLOGD("ftp COMMAND wrong");
  580. }
  581. }else if(memcmp(g_s_ftp.network->cmd_send_data, "SYST", 4)==0){
  582. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_SYSTEM_TYPE, 3)){
  583. LLOGD("ftp COMMAND wrong");
  584. }
  585. }else if(memcmp(g_s_ftp.network->cmd_send_data, "PWD", 3)==0){
  586. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
  587. LLOGD("ftp COMMAND wrong");
  588. }
  589. }else if(memcmp(g_s_ftp.network->cmd_send_data, "MKD", 3)==0){
  590. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
  591. LLOGD("ftp COMMAND wrong");
  592. }
  593. }else if(memcmp(g_s_ftp.network->cmd_send_data, "CWD", 3)==0){
  594. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  595. LLOGD("ftp COMMAND wrong");
  596. }
  597. }else if(memcmp(g_s_ftp.network->cmd_send_data, "CDUP", 4)==0){
  598. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  599. LLOGD("ftp COMMAND wrong");
  600. }
  601. }else if(memcmp(g_s_ftp.network->cmd_send_data, "RMD", 3)==0){
  602. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  603. LLOGD("ftp COMMAND wrong");
  604. }
  605. }else if(memcmp(g_s_ftp.network->cmd_send_data, "DELE", 4)==0){
  606. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
  607. LLOGD("ftp COMMAND wrong");
  608. }
  609. }else if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_DATA_CON_FAIL, 3)==0){
  610. LLOGD("ftp need pasv_connect");
  611. }
  612. }
  613. OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
  614. ftp_state = FTP_SUCCESS_DATE;
  615. l_ftp_cb(ftp_state);
  616. break;
  617. default:
  618. break;
  619. }
  620. continue;
  621. operation_failed:
  622. LLOGE("ftp operation failed");
  623. if (g_s_ftp.fd){
  624. luat_fs_fclose(g_s_ftp.fd);
  625. g_s_ftp.fd = NULL;
  626. }
  627. ftp_state = FTP_ERROR;
  628. l_ftp_cb(ftp_state);
  629. }
  630. ftp_state = FTP_SUCCESS_NO_DATE;
  631. luat_ftp_cmd_send(&g_s_ftp, (uint8_t*)"QUIT\r\n", strlen("QUIT\r\n"),FTP_SOCKET_TIMEOUT);
  632. 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);
  633. if (ret){
  634. ftp_state = FTP_ERROR;
  635. }else{
  636. if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONTROL, 3)){
  637. LLOGE("ftp QUIT wrong");
  638. ftp_state = FTP_ERROR;
  639. }
  640. }
  641. OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
  642. if (ftp_state == FTP_SUCCESS_NO_DATE) ftp_state = FTP_SUCCESS_DATE;
  643. l_ftp_cb(ftp_state);
  644. luat_ftp_release();
  645. g_s_ftp.task_handle = NULL;
  646. luat_rtos_task_delete(task_handle);
  647. return;
  648. wait_event_and_out:
  649. while(1)
  650. {
  651. luat_rtos_event_recv(g_s_ftp.task_handle, 0, &task_event, NULL, LUAT_WAIT_FOREVER);
  652. if (task_event.ID >= FTP_EVENT_LOGIN && task_event.ID <= FTP_EVENT_CLOSE)
  653. {
  654. luat_ftp_release();
  655. ftp_state = FTP_ERROR;
  656. l_ftp_cb(ftp_state);
  657. g_s_ftp.task_handle = NULL;
  658. luat_rtos_task_delete(task_handle);
  659. return;
  660. }
  661. }
  662. }
  663. 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){
  664. if (g_s_ftp.network){
  665. LLOGE("ftp already login, please close first");
  666. return FTP_ERROR_STATE;
  667. }
  668. g_s_ftp.network = (luat_ftp_network_t *)luat_heap_malloc(sizeof(luat_ftp_network_t));
  669. if (!g_s_ftp.network){
  670. LLOGE("out of memory when malloc g_s_ftp.network");
  671. return FTP_ERROR_NO_MEM;
  672. }
  673. memset(g_s_ftp.network, 0, sizeof(luat_ftp_network_t));
  674. if (ftp_cb){
  675. g_s_ftp.network->ftp_cb = ftp_cb;
  676. }
  677. g_s_ftp.network->adapter_index = adapter;
  678. if (g_s_ftp.network->adapter_index >= NW_ADAPTER_QTY){
  679. LLOGE("bad network adapter index %d", g_s_ftp.network->adapter_index);
  680. return FTP_ERROR_STATE;
  681. }
  682. g_s_ftp.network->cmd_netc = network_alloc_ctrl(g_s_ftp.network->adapter_index);
  683. if (!g_s_ftp.network->cmd_netc){
  684. LLOGE("cmd_netc create fail");
  685. return FTP_ERROR_NO_MEM;
  686. }
  687. g_s_ftp.network->port = port;
  688. if (strlen(ip_addr) > 0 && strlen(ip_addr) < 64)
  689. memcpy(g_s_ftp.network->addr, ip_addr, strlen(ip_addr) + 1);
  690. if (strlen(username) > 0 && strlen(username) < 64)
  691. memcpy(g_s_ftp.network->username, username, strlen(username) + 1);
  692. if (strlen(password) > 0 && strlen(password) < 64)
  693. memcpy(g_s_ftp.network->password, password, strlen(password) + 1);
  694. if (luat_ftp_tls == NULL){
  695. network_deinit_tls(g_s_ftp.network->cmd_netc);
  696. }else{
  697. if (network_init_tls(g_s_ftp.network->cmd_netc, (luat_ftp_tls->server_cert || luat_ftp_tls->client_cert)?2:0)){
  698. return FTP_ERROR_CLOSE;
  699. }
  700. if (luat_ftp_tls->server_cert){
  701. 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);
  702. }
  703. if (luat_ftp_tls->client_cert){
  704. 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,
  705. (const unsigned char *)luat_ftp_tls->client_key, strlen(luat_ftp_tls->client_key)+1,
  706. (const unsigned char *)luat_ftp_tls->client_password, strlen(luat_ftp_tls->client_password)+1);
  707. }
  708. }
  709. network_set_ip_invaild(&g_s_ftp.network->ip_addr);
  710. int result = luat_rtos_task_create(&g_s_ftp.task_handle, 2*1024, 10, "ftp", ftp_task, NULL, 16);
  711. if (result) {
  712. LLOGE("创建ftp task失败!! %d", result);
  713. return result;
  714. }
  715. network_init_ctrl(g_s_ftp.network->cmd_netc,g_s_ftp.task_handle, ftp_task_cb, NULL);
  716. network_set_base_mode(g_s_ftp.network->cmd_netc, 1, 30000, 0, 0, 0, 0);
  717. network_set_local_port(g_s_ftp.network->cmd_netc, 0);
  718. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_LOGIN, 0, 0, 0, LUAT_WAIT_FOREVER);
  719. return 0;
  720. }
  721. int luat_ftp_command(const char * command){
  722. if (!g_s_ftp.network){
  723. LLOGE("please login first");
  724. return -1;
  725. }
  726. if (memcmp(command, "NOOP", 4)==0){
  727. LLOGD("command: NOOP");
  728. }else if(memcmp(command, "SYST", 4)==0){
  729. LLOGD("command: SYST");
  730. }else if(memcmp(command, "MKD", 3)==0){
  731. LLOGD("command: MKD");
  732. }else if(memcmp(command, "CWD", 3)==0){
  733. LLOGD("command: CWD");
  734. }else if(memcmp(command, "CDUP", 4)==0){
  735. LLOGD("command: CDUP");
  736. }else if(memcmp(command, "RMD", 3)==0){
  737. LLOGD("command: RMD");
  738. }else if(memcmp(command, "PWD", 3)==0){
  739. LLOGD("command: RMD");
  740. }else if(memcmp(command, "DELE", 4)==0){
  741. LLOGD("command: DELE");
  742. }else if(memcmp(command, "TYPE", 4)==0){
  743. LLOGD("command: TYPE");
  744. }else if(memcmp(command, "SIZE", 4)==0){
  745. LLOGD("command: SIZE");
  746. }else if(memcmp(command, "LIST", 4)==0){
  747. LLOGD("command: LIST");
  748. }else{
  749. LLOGE("not support cmd:%s",command);
  750. return -1;
  751. }
  752. memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
  753. snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "%s\r\n",command);
  754. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_COMMAND, 0, 0, 0, 0);
  755. return 0;
  756. }
  757. int luat_ftp_pull(const char * local_name,const char * remote_name){
  758. if (!g_s_ftp.network){
  759. LLOGE("please login first");
  760. return -1;
  761. }
  762. luat_fs_remove(local_name);
  763. if (g_s_ftp.fd)
  764. {
  765. luat_fs_fclose(g_s_ftp.fd);
  766. g_s_ftp.fd = NULL;
  767. }
  768. g_s_ftp.fd = luat_fs_fopen(local_name, "wb+");
  769. if (g_s_ftp.fd == NULL) {
  770. LLOGE("open download file fail %s", local_name);
  771. return -1;
  772. }
  773. g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
  774. memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
  775. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PULL, 0, 0, 0, 0);
  776. return 0;
  777. }
  778. int luat_ftp_close(void){
  779. if (!g_s_ftp.network){
  780. LLOGE("please login first");
  781. return -1;
  782. }
  783. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_CLOSE, 0, 0, 0, LUAT_WAIT_FOREVER);
  784. return 0;
  785. }
  786. int luat_ftp_push(const char * local_name,const char * remote_name){
  787. if (!g_s_ftp.network){
  788. LLOGE("please login first");
  789. return -1;
  790. }
  791. if (g_s_ftp.fd)
  792. {
  793. luat_fs_fclose(g_s_ftp.fd);
  794. g_s_ftp.fd = NULL;
  795. }
  796. g_s_ftp.fd = luat_fs_fopen(local_name, "rb");
  797. if (g_s_ftp.fd == NULL) {
  798. LLOGE("open download file fail %s", local_name);
  799. return -1;
  800. }
  801. g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
  802. memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
  803. luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PUSH, 0, 0, 0, LUAT_WAIT_FOREVER);
  804. return 0;
  805. }
  806. void luat_ftp_debug(uint8_t on_off) {g_s_ftp.debug_onoff = on_off;}