| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893 |
- #include "luat_base.h"
- #include "luat_network_adapter.h"
- #include "luat_rtos.h"
- #include "luat_fs.h"
- #include "luat_mem.h"
- #include "luat_ftp.h"
- #define LUAT_LOG_TAG "ftp"
- #include "luat_log.h"
- #undef LLOGD
- #define LLOGD(format, ...) do {if (g_s_ftp.debug_onoff) {luat_log_log(LUAT_LOG_DEBUG, LUAT_LOG_TAG, format, ##__VA_ARGS__);}} while(0)
- luat_ftp_ctrl_t g_s_ftp = {0};
- static void l_ftp_cb(FTP_SUCCESS_STATE_e state){
- luat_ftp_cb_t ftp_cb;
- if (g_s_ftp.network->ftp_cb){
- ftp_cb = (luat_ftp_cb_t)g_s_ftp.network->ftp_cb;
- ftp_cb(&g_s_ftp,state);
- }
- #ifndef __LUATOS__
- OS_DeInitBuffer(&g_s_ftp.result_buffer);
- #endif
- }
- uint32_t luat_ftp_release(void) {
- if (!g_s_ftp.network) return 0;
- if (g_s_ftp.network->cmd_netc){
- network_force_close_socket(g_s_ftp.network->cmd_netc);
- network_release_ctrl(g_s_ftp.network->cmd_netc);
- g_s_ftp.network->cmd_netc = NULL;
- }
- if (g_s_ftp.network->data_netc){
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- }
- luat_heap_free(g_s_ftp.network);
- g_s_ftp.network = NULL;
- return 0;
- }
- static uint32_t luat_ftp_data_send(luat_ftp_ctrl_t *ftp_ctrl, uint8_t* send_data, uint32_t send_len) {
- if (send_len == 0)
- return 0;
- uint32_t tx_len = 0;
- LLOGD("send %p data:%d", ftp_ctrl, send_len);
- network_tx(g_s_ftp.network->data_netc, send_data, send_len, 0, NULL, 0, &tx_len, 0);
- return tx_len;
- }
- 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) {
- if (send_len == 0)
- return 0;
- uint32_t tx_len = 0;
- LLOGD("send %p cmd:%.*s", ftp_ctrl, send_len, send_data);
- network_tx(g_s_ftp.network->cmd_netc, send_data, send_len, 0, NULL, 0, &tx_len, timeout_ms);
- return tx_len;
- }
- static int luat_ftp_cmd_recv(luat_ftp_ctrl_t *ftp_ctrl,uint8_t *recv_data,uint32_t *recv_len,uint32_t timeout_ms){
- uint8_t is_break = 0,is_timeout = 0;
- int ret = network_wait_rx(g_s_ftp.network->cmd_netc, timeout_ms, &is_break, &is_timeout);
- LLOGD("cmd recv %o network_wait_rx ret:%d is_break:%d is_timeout:%d", ftp_ctrl, ret,is_break,is_timeout);
- if (ret)
- return -1;
- if (is_timeout)
- return 1;
- else if (is_break)
- return 2;
- return network_rx(g_s_ftp.network->cmd_netc, recv_data, FTP_CMD_RECV_MAX, 0, NULL, NULL, recv_len);
- }
- static int32_t luat_ftp_data_callback(void *data, void *param){
- (void)param;
- OS_EVENT *event = (OS_EVENT *)data;
- uint8_t *rx_buffer;
- int ret = 0;
- uint32_t rx_len = 0;
- if (!g_s_ftp.network)
- {
- return 0;
- }
- // 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);
- LLOGD("luat_ftp_data_callback %d %d",event->ID - EV_NW_RESULT_BASE, event->Param1);
- if (event->Param1){
- if (EV_NW_RESULT_CONNECT == event->ID)
- {
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0xffffffff, 0, 0, LUAT_WAIT_FOREVER);
- }
- else
- {
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
- }
- return -1;
- }
- switch (event->ID)
- {
- case EV_NW_RESULT_TX:
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_TX_DONE, 0, 0, 0, LUAT_WAIT_FOREVER);
- break;
- case EV_NW_RESULT_EVENT:
- rx_buffer = NULL;
- uint8_t tmpbuff[4];
- do{
- // 先读取长度
- ret = network_rx(g_s_ftp.network->data_netc, NULL, 0, 0, NULL, NULL, &rx_len);
- if (rx_len <= 0) {
- // 没数据? 那也读一次, 然后退出
- network_rx(g_s_ftp.network->data_netc, tmpbuff, 4, 0, NULL, NULL, &rx_len);
- break;
- }
- if (rx_len > 2048)
- rx_len = 2048;
- rx_buffer = luat_heap_malloc(rx_len);
- // 如果rx_buffer == NULL, 内存炸了
- if (rx_buffer == NULL) {
- LLOGE("out of memory when malloc ftp buff");
- network_close(g_s_ftp.network->data_netc, 0);
- return -1;
- }
- ret = network_rx(g_s_ftp.network->data_netc, rx_buffer, rx_len, 0, NULL, NULL, &rx_len);
- // LLOGD("luat_ftp_data_callback network_rx ret:%d rx_len:%d",ret,rx_len);
- if (!ret && rx_len > 0){
- if (g_s_ftp.fd){
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_WRITE_FILE, (uint32_t)rx_buffer, rx_len, 0, LUAT_WAIT_FOREVER);
- rx_buffer = NULL;
- continue;
- }else{
- OS_BufferWrite(&g_s_ftp.result_buffer, rx_buffer, rx_len);
- }
- }
- luat_heap_free(rx_buffer);
- rx_buffer = NULL;
- } while (!ret && rx_len);
- if (rx_buffer)
- luat_heap_free(rx_buffer);
- rx_buffer = NULL;
- break;
- case EV_NW_RESULT_CLOSE:
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CLOSED, 0, 0, 0, LUAT_WAIT_FOREVER);
- return 0;
- break;
- case EV_NW_RESULT_CONNECT:
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_DATA_CONNECT, 0, 0, 0, LUAT_WAIT_FOREVER);
- break;
- case EV_NW_RESULT_LINK:
- return 0;
- }
- ret = network_wait_event(g_s_ftp.network->data_netc, NULL, 0, NULL);
- if (ret < 0){
- network_close(g_s_ftp.network->data_netc, 0);
- return -1;
- }
- return 0;
- }
- static int32_t ftp_task_cb(void *pdata, void *param){
- OS_EVENT *event = pdata;
- if (event->ID >= FTP_EVENT_LOGIN && event->ID <= FTP_EVENT_PUSH){
- LLOGE("last cmd not finish, ignore %d,%u,%u,%x", event->ID - USER_EVENT_ID_START, event->Param1, event->Param2, param);
- return -1;
- }
- switch(event->ID){
- case FTP_EVENT_DATA_WRITE_FILE:
- if (g_s_ftp.fd){
- g_s_ftp.network->download_file_size += event->Param2;
- luat_fs_fwrite((void*)event->Param1, event->Param2, 1, g_s_ftp.fd);
- luat_heap_free((void*)event->Param1);
- }
- break;
- case FTP_EVENT_DATA_TX_DONE:
- g_s_ftp.network->upload_done_size = (size_t)g_s_ftp.network->data_netc->ack_size;
- if (g_s_ftp.network->upload_done_size >= g_s_ftp.network->local_file_size){
- LLOGD("ftp data upload done!");
- network_close(g_s_ftp.network->data_netc, 0);
- }
- break;
- case FTP_EVENT_DATA_CONNECT:
- if (g_s_ftp.network->data_netc_connecting)
- {
- g_s_ftp.network->data_netc_connecting = 0;
- g_s_ftp.network->data_netc_online = !event->Param1;
- }
- break;
- case FTP_EVENT_DATA_CLOSED:
- LLOGD("ftp data channel close");
- g_s_ftp.network->data_netc_online = 0;
- if (g_s_ftp.network->data_netc)
- {
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- }
- break;
- case FTP_EVENT_CLOSE:
- g_s_ftp.is_run = 0;
- break;
- default:
- // LLOGE("ignore %x,%x,%x", event->ID, param, EV_NW_RESULT_EVENT);
- break;
- }
- return 0;
- }
- static int luat_ftp_pasv_connect(luat_ftp_ctrl_t *ftp_ctrl,uint32_t timeout_ms){
- char h1[4]={0},h2[4]={0},h3[4]={0},h4[4]={0},p1[4]={0},p2[4]={0},data_addr[64]={0};
- uint8_t port1,port2;
- uint16_t data_port;
- luat_ftp_cmd_send(&g_s_ftp, (uint8_t*)"PASV\r\n", strlen("PASV\r\n"),FTP_SOCKET_TIMEOUT);
- // g_s_ftp.data_transfer_done = 0;
- 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);
- if (ret){
- return -1;
- }else{
- LLOGD("connect %p cmd_recv_data %.*s", ftp_ctrl, g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_ENTER_PASSIVE, 3)){
- LLOGW("ftp pasv_connect wrong %.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- return -1;
- }
- }
- char *temp = memchr(g_s_ftp.network->cmd_recv_data, '(', strlen((const char *)(g_s_ftp.network->cmd_recv_data)));
- char *temp1 = memchr(temp+1, ',', strlen(temp)-1);
- memcpy(h1, temp+1, temp1-temp-1);
- char *temp2 = memchr(temp1+1, ',', strlen(temp1)-1);
- memcpy(h2, temp1+1, temp2-temp1-1);
- char *temp3 = memchr(temp2+1, ',', strlen(temp2)-1);
- memcpy(h3, temp2+1, temp3-temp2-1);
- char *temp4 = memchr(temp3+1, ',', strlen(temp3)-1);
- memcpy(h4, temp3+1, temp4-temp3-1);
- char *temp5 = memchr(temp4+1, ',', strlen(temp4)-1);
- memcpy(p1, temp4+1, temp5-temp4-1);
- char *temp6 = memchr(temp5+1, ')', strlen(temp5)-1);
- memcpy(p2, temp5+1, temp6-temp5-1);
- snprintf_(data_addr, 64, "%s.%s.%s.%s",h1,h2,h3,h4);
- port1 = (uint8_t)atoi(p1);
- port2 = (uint8_t)atoi(p2);
- data_port = port1 * 256 + port2;
- // 由于服务器配置问题, 可能访问内网ip,空ip,0.0.0.0等不可使用的被动IP地址
- LLOGD("data_addr:%s data_port:%d",data_addr,data_port);
- if (memcmp(data_addr,"172.",4)==0 ||
- memcmp(data_addr,"192.",4)==0 ||
- memcmp(data_addr,"10.",3)==0||
- memcmp(data_addr,"127.0.0.1",9)==0||
- memcmp(data_addr,"169.254.0.0",11)==0||
- memcmp(data_addr,"169.254.0.16",12)==0||
- memcmp(data_addr,"0.",2) == 0){
- LLOGW("服务器回应不可路由的地址(%s)。使用服务器地址代替 %s", data_addr, g_s_ftp.network->addr);
- memset(data_addr,0,64);
- LLOGD("g_s_ftp.network->addr:%s",g_s_ftp.network->addr);
- memcpy(data_addr, g_s_ftp.network->addr, strlen(g_s_ftp.network->addr)+1);
- }
- LLOGD("data_addr:%s data_port:%d",data_addr,data_port);
- if (g_s_ftp.network->data_netc)
- {
- LLOGE("data_netc already create");
- return -1;
- }
- g_s_ftp.network->data_netc = network_alloc_ctrl(g_s_ftp.network->adapter_index);
- if (!g_s_ftp.network->data_netc){
- LLOGE("data_netc create fail");
- return -1;
- }
- network_init_ctrl(g_s_ftp.network->data_netc,NULL, luat_ftp_data_callback, g_s_ftp.network);
- network_set_base_mode(g_s_ftp.network->data_netc, 1, 10000, 0, 0, 0, 0);
- network_set_local_port(g_s_ftp.network->data_netc, 0);
- network_deinit_tls(g_s_ftp.network->data_netc);
- if(network_connect(g_s_ftp.network->data_netc, data_addr, strlen(data_addr), NULL, data_port, 0)<0){
- LLOGE("ftp data network connect fail");
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- return -1;
- }
- uint8_t is_timeout;
- OS_EVENT event;
- g_s_ftp.network->data_netc_connecting = 1;
- g_s_ftp.network->data_netc_online = 0;
- while(g_s_ftp.network->data_netc_connecting)
- {
- if (network_wait_event(g_s_ftp.network->cmd_netc, &event, timeout_ms, &is_timeout))
- {
- return -1;
- }
- if (is_timeout)
- {
- return -1;
- }
- if (event.ID)
- {
- ftp_task_cb(&event, NULL);
- }
- else
- {
- if (g_s_ftp.network->cmd_netc->new_rx_flag)
- {
- 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);
- LLOGD("ftp cmd rx %.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- }
- }
- }
- if (g_s_ftp.network->data_netc_online)
- {
- LLOGD("ftp pasv_connect ok");
- return 0;
- }
- return -1;
- }
- static int ftp_login(void)
- {
- int ret;
- ret = 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);
- if (ret) {
- LLOGE("ftp network_connect fail %d", ret);
- return -1;
- }
- 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);
- if (ret){
- return -1;
- }else{
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_SERVICE_NEW_OK, 3)){
- LLOGE("ftp connect error");
- return -1;
- }
- }
- LLOGD("ftp connect ok");
- memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "USER %s\r\n",g_s_ftp.network->username);
- 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);
- 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);
- if (ret){
- LLOGE("ftp username wrong");
- return -1;
- }else{
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_USERNAME_OK, 3)){
- LLOGE("ftp username wrong");
- return -1;
- }
- }
- LLOGD("ftp username ok");
- memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "PASS %s\r\n",g_s_ftp.network->password);
- 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);
- 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);
- if (ret){
- LLOGE("ftp login wrong");
- return -1;
- }else{
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_LOGIN_OK, 3)){
- LLOGE("ftp login wrong");
- return -1;
- }
- }
- LLOGD("ftp login ok");
- return 0;
- }
- static int find_newline(void){
- uint32_t pos = 0;
- while (pos < g_s_ftp.network->cmd_recv_len){
- if(('\r' == g_s_ftp.network->cmd_recv_data[pos]) || ('\n' == g_s_ftp.network->cmd_recv_data[pos])){
- return pos;
- }else{
- pos++;
- }
- }
- return -1;
- }
- static int pasv_recv(void){
- int ret;
- int pos;
- uint8_t rx_finish = 0;
- 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);
- if (ret){
- LLOGE("pasv_recv %s error:%d", g_s_ftp.network->cmd_recv_data, ret);
- return -1;
- }
- LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ ,g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
- 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)){
- return -1;
- }
- pos = find_newline();
- if (pos >= 0){
- memmove(g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_data + pos, g_s_ftp.network->cmd_recv_len - pos);
- g_s_ftp.network->cmd_recv_len -= pos;
- g_s_ftp.network->cmd_recv_data[g_s_ftp.network->cmd_recv_len] = 0;
- if (strstr((const char *)(g_s_ftp.network->cmd_recv_data), FTP_CLOSE_CONNECT)){
- rx_finish = 1;
- }
- }
- 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);
- 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通道未断开或者已经接收到数据了
- {
- 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);
- if (ret<0){
- LLOGD("%s %d rx error!%d",__FUNCTION__ ,__LINE__ , ret);
- return -1;
- } else if (!ret) {
- LLOGD("%s %d %.*s",__FUNCTION__ ,__LINE__ , g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
- return -1;
- }
- else {
- rx_finish = 1;
- }
- }
- }
- // 数据接收完了主动关闭掉接收
- if (g_s_ftp.network->data_netc_online && g_s_ftp.network->data_netc){
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- }
- return 0;
- }
- static int string2int(char *str,size_t len){
- if(len==0) return 0;
- char flag = '+';
- int res = 0;
- size_t count = 0;
- if(*str=='-'){
- ++str;
- flag = '-';
- }
- while(*str>=48 && *str<=57 && count<len){
- res = 10*res+ *str++-48;
- count++;
- }
- if(flag == '-')
- res = -res;
- return res;
- }
- static void ftp_task(void *param){
- FTP_SUCCESS_STATE_e ftp_state = FTP_SUCCESS_NO_DATE;
- int ret;
- int count = 0;
- OS_EVENT task_event;
- uint8_t is_timeout = 0;
- luat_rtos_task_sleep(10); // 起来就等10ms, 给调用者一点时间运行完毕
- g_s_ftp.is_run = 1;
- luat_rtos_event_recv(g_s_ftp.task_handle, FTP_EVENT_LOGIN, &task_event, NULL, LUAT_WAIT_FOREVER);
- if (ftp_login()){
- LLOGE("ftp login fail");
- ftp_state = FTP_ERROR;
- l_ftp_cb(ftp_state);
- luat_ftp_release();
- luat_rtos_task_delete(g_s_ftp.task_handle);
- g_s_ftp.task_handle = NULL;
- return;
- }else{
- l_ftp_cb(ftp_state);
- }
- while (g_s_ftp.is_run) {
- is_timeout = 0;
- ret = network_wait_event(g_s_ftp.network->cmd_netc, &task_event, 3600000, &is_timeout);
- if (ret < 0){
- LLOGE("ftp network error");
- goto wait_event_and_out;
- }else if (is_timeout || !task_event.ID){
- if (g_s_ftp.network->cmd_netc->new_rx_flag){
- network_rx(g_s_ftp.network->cmd_netc, g_s_ftp.network->cmd_recv_data, 1024, 0, NULL, NULL, &ret);
- LLOGD("ftp rx %dbyte", ret);
- }
- continue;
- }
- ftp_state = FTP_SUCCESS_NO_DATE;
- switch (task_event.ID)
- {
- case FTP_EVENT_LOGIN:
- break;
- case FTP_EVENT_PULL:
- g_s_ftp.network->remote_file_size = 0;
- g_s_ftp.network->download_file_size = 0;
- memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "SIZE %s\r\n",g_s_ftp.network->remote_name);
- 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);
- 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);
- if (!ret){
- if (!memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS, 3)){
- 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);
- }
- }
- // LLOGD("remote_file_size:%d",g_s_ftp.network->remote_file_size);
- if (g_s_ftp.network->data_netc){
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- }
- if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
- LLOGE("ftp pasv_connect fail");
- goto operation_failed;
- }
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "RETR %s\r\n",g_s_ftp.network->remote_name);
- 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);
- if (pasv_recv()) goto operation_failed;
- if (g_s_ftp.fd){
- luat_fs_fclose(g_s_ftp.fd);
- g_s_ftp.fd = NULL;
- }
- l_ftp_cb(ftp_state);
- break;
- case FTP_EVENT_PUSH:
- if (g_s_ftp.network->data_netc){
- network_force_close_socket(g_s_ftp.network->data_netc);
- network_release_ctrl(g_s_ftp.network->data_netc);
- g_s_ftp.network->data_netc = NULL;
- }
- if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
- LLOGD("ftp pasv_connect fail");
- goto operation_failed;
- }
- memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "STOR %s\r\n",g_s_ftp.network->remote_name);
- 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);
- 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);
- if (ret){
- goto operation_failed;
- }else{
- LLOGD("%.*s", g_s_ftp.network->cmd_recv_len, g_s_ftp.network->cmd_recv_data);
- 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)){
- LLOGD("ftp STOR wrong");
- goto operation_failed;
- }
- }
- uint8_t* buff = luat_heap_malloc(PUSH_BUFF_SIZE);
- int offset = 0;
- g_s_ftp.network->upload_done_size = 0;
- while (1) {
- memset(buff, 0, PUSH_BUFF_SIZE);
- int len = luat_fs_fread(buff, sizeof(uint8_t), PUSH_BUFF_SIZE, g_s_ftp.fd);
- if (len < 1)
- break;
- luat_ftp_data_send(&g_s_ftp, buff, len);
- offset += len;
- }
- luat_heap_free(buff);
- LLOGD("offset:%d file_size:%d",offset,g_s_ftp.network->local_file_size);
- 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);
- if (g_s_ftp.network->upload_done_size != g_s_ftp.network->local_file_size)
- {
- LLOGE("upload not finish !!! %d,%d", g_s_ftp.network->upload_done_size, g_s_ftp.network->local_file_size);
- }
- if (ret){
- goto operation_failed;
- }else{
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONNECT, 3)){
- LLOGD("ftp STOR wrong");
- }
- }
- while (count<3 && g_s_ftp.network->data_netc_online!=0){
- 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);
- count++;
- }
- if (g_s_ftp.fd){
- luat_fs_fclose(g_s_ftp.fd);
- g_s_ftp.fd = NULL;
- }
- l_ftp_cb(ftp_state);
- break;
- case FTP_EVENT_CLOSE:
- g_s_ftp.is_run = 0;
- break;
- case FTP_EVENT_COMMAND:
- OS_DeInitBuffer(&g_s_ftp.result_buffer);
- if(!memcmp(g_s_ftp.network->cmd_send_data, "LIST", 4))
- {
- if(luat_ftp_pasv_connect(&g_s_ftp,FTP_SOCKET_TIMEOUT)){
- LLOGD("ftp pasv_connect fail");
- goto operation_failed;
- }
- 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);
- if (pasv_recv()) goto operation_failed;
- ftp_state = FTP_SUCCESS_DATE;
- l_ftp_cb(ftp_state);
- break;
- }
- 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);
- 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);
- if (ret){
- goto operation_failed;
- }else{
- if (memcmp(g_s_ftp.network->cmd_send_data, "NOOP", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "TYPE", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_COMMAND_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "SIZE", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_STATUS, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "SYST", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_SYSTEM_TYPE, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "PWD", 3)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "MKD", 3)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_PATHNAME_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "CWD", 3)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "CDUP", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "RMD", 3)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if(memcmp(g_s_ftp.network->cmd_send_data, "DELE", 4)==0){
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_FILE_REQUESTED_OK, 3)){
- LLOGD("ftp COMMAND wrong");
- }
- }else if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_DATA_CON_FAIL, 3)==0){
- LLOGD("ftp need pasv_connect");
- }
- }
- OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
- ftp_state = FTP_SUCCESS_DATE;
- l_ftp_cb(ftp_state);
- break;
- default:
- break;
- }
- continue;
- operation_failed:
- LLOGE("ftp operation failed");
- if (g_s_ftp.fd){
- luat_fs_fclose(g_s_ftp.fd);
- g_s_ftp.fd = NULL;
- }
- ftp_state = FTP_ERROR;
- l_ftp_cb(ftp_state);
- }
- ftp_state = FTP_SUCCESS_NO_DATE;
- luat_ftp_cmd_send(&g_s_ftp, (uint8_t*)"QUIT\r\n", strlen("QUIT\r\n"),FTP_SOCKET_TIMEOUT);
- 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);
- if (ret){
- ftp_state = FTP_ERROR;
- }else{
- if (memcmp(g_s_ftp.network->cmd_recv_data, FTP_CLOSE_CONTROL, 3)){
- LLOGE("ftp QUIT wrong");
- ftp_state = FTP_ERROR;
- }
- }
- OS_BufferWrite(&g_s_ftp.result_buffer, g_s_ftp.network->cmd_recv_data, g_s_ftp.network->cmd_recv_len);
- if (ftp_state == FTP_SUCCESS_NO_DATE) ftp_state = FTP_SUCCESS_DATE;
- l_ftp_cb(ftp_state);
- luat_ftp_release();
- luat_rtos_task_delete(g_s_ftp.task_handle);
- g_s_ftp.task_handle = NULL;
- return;
- wait_event_and_out:
- while(1)
- {
- luat_rtos_event_recv(g_s_ftp.task_handle, 0, &task_event, NULL, LUAT_WAIT_FOREVER);
- if (task_event.ID >= FTP_EVENT_LOGIN && task_event.ID <= FTP_EVENT_CLOSE)
- {
- luat_ftp_release();
- ftp_state = FTP_ERROR;
- l_ftp_cb(ftp_state);
- luat_rtos_task_delete(g_s_ftp.task_handle);
- g_s_ftp.task_handle = NULL;
- return;
- }
- }
- }
- 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){
- int result = 0;
- if (g_s_ftp.network){
- LLOGE("ftp already login, please close first");
- return FTP_ERROR_STATE;
- }
- g_s_ftp.network = (luat_ftp_network_t *)luat_heap_malloc(sizeof(luat_ftp_network_t));
- if (!g_s_ftp.network){
- LLOGE("out of memory when malloc g_s_ftp.network");
- return FTP_ERROR_NO_MEM;
- }
- memset(g_s_ftp.network, 0, sizeof(luat_ftp_network_t));
- if (ftp_cb){
- g_s_ftp.network->ftp_cb = ftp_cb;
- }
- g_s_ftp.network->adapter_index = adapter;
- if (g_s_ftp.network->adapter_index >= NW_ADAPTER_QTY){
- LLOGE("bad network adapter index %d", g_s_ftp.network->adapter_index);
- return FTP_ERROR_STATE;
- }
- LLOGD("ftp adapter %d host %s port %d", g_s_ftp.network->adapter_index, ip_addr, port);
- g_s_ftp.network->cmd_netc = network_alloc_ctrl(g_s_ftp.network->adapter_index);
- if (!g_s_ftp.network->cmd_netc){
- LLOGE("cmd_netc create fail");
- return FTP_ERROR_NO_MEM;
- }
- g_s_ftp.network->port = port;
- if (strlen(ip_addr) > 0 && strlen(ip_addr) < 64) {
- memcpy(g_s_ftp.network->addr, ip_addr, strlen(ip_addr) + 1);
- }
- else {
- LLOGE("invalid ip address length %d", strlen(ip_addr));
- result = -20;
- goto error;
- }
- if (strlen(username) > 0 && strlen(username) < 64) {
- memcpy(g_s_ftp.network->username, username, strlen(username) + 1);
- }
- else {
- LLOGE("invalid username length %d", strlen(username));
- result = -21;
- goto error;
- }
- if (strlen(password) > 0 && strlen(password) < 64) {
- memcpy(g_s_ftp.network->password, password, strlen(password) + 1);
- }
- else {
- LLOGE("invalid password length %d", strlen(password));
- result = -22;
- goto error;
- }
- if (luat_ftp_tls == NULL){
- network_deinit_tls(g_s_ftp.network->cmd_netc);
- }
- else {
- #if 0
- if (network_init_tls(g_s_ftp.network->cmd_netc, (luat_ftp_tls->server_cert || luat_ftp_tls->client_cert)?2:0)){
- LLOGE("ftp tls init fail");
- result = -23;
- goto error;
- }
- if (luat_ftp_tls->server_cert){
- 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);
- }
- if (luat_ftp_tls->client_cert){
- 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,
- (const unsigned char *)luat_ftp_tls->client_key, strlen(luat_ftp_tls->client_key)+1,
- (const unsigned char *)luat_ftp_tls->client_password, strlen(luat_ftp_tls->client_password)+1);
- }
- #else
- LLOGE("ftp tls not support yet");
- result = -24;
- goto error;
- #endif
- }
- // task会主动等10ms
- result = luat_rtos_task_create(&g_s_ftp.task_handle, 8*1024, 10, "ftp", ftp_task, NULL, 16);
- if (result) {
- LLOGE("创建ftp task失败!! %d", result);
- goto error;
- }
-
- network_set_ip_invaild(&g_s_ftp.network->ip_addr);
- network_init_ctrl(g_s_ftp.network->cmd_netc,g_s_ftp.task_handle, ftp_task_cb, NULL);
- network_set_base_mode(g_s_ftp.network->cmd_netc, 1, 30000, 0, 0, 0, 0);
- network_set_local_port(g_s_ftp.network->cmd_netc, 0);
- // g_s_ftp.network->cmd_netc->is_debug = 1;
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_LOGIN, 0, 0, 0, LUAT_WAIT_FOREVER);
- return 0;
- error:
- // 清理资源
- if (g_s_ftp.network->cmd_netc) {
- network_force_close_socket(g_s_ftp.network->cmd_netc);
- network_release_ctrl(g_s_ftp.network->cmd_netc);
- g_s_ftp.network->cmd_netc = NULL;
- }
- return result;
- }
- int luat_ftp_command(const char * command){
- if (!g_s_ftp.network){
- LLOGE("please login first");
- return -1;
- }
- if (memcmp(command, "NOOP", 4)==0){
- LLOGD("command: NOOP");
- }else if(memcmp(command, "SYST", 4)==0){
- LLOGD("command: SYST");
- }else if(memcmp(command, "MKD", 3)==0){
- LLOGD("command: MKD");
- }else if(memcmp(command, "CWD", 3)==0){
- LLOGD("command: CWD");
- }else if(memcmp(command, "CDUP", 4)==0){
- LLOGD("command: CDUP");
- }else if(memcmp(command, "RMD", 3)==0){
- LLOGD("command: RMD");
- }else if(memcmp(command, "PWD", 3)==0){
- LLOGD("command: RMD");
- }else if(memcmp(command, "DELE", 4)==0){
- LLOGD("command: DELE");
- }else if(memcmp(command, "TYPE", 4)==0){
- LLOGD("command: TYPE");
- }else if(memcmp(command, "SIZE", 4)==0){
- LLOGD("command: SIZE");
- }else if(memcmp(command, "LIST", 4)==0){
- LLOGD("command: LIST");
- }else{
- LLOGE("not support cmd:%s",command);
- return -1;
- }
- memset(g_s_ftp.network->cmd_send_data,0,FTP_CMD_SEND_MAX);
- snprintf_((char *)(g_s_ftp.network->cmd_send_data), FTP_CMD_SEND_MAX, "%s\r\n",command);
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_COMMAND, 0, 0, 0, 0);
- return 0;
- }
- int luat_ftp_pull(const char * local_name,const char * remote_name){
- if (!g_s_ftp.network){
- LLOGE("please login first");
- return -1;
- }
- luat_fs_remove(local_name);
- if (g_s_ftp.fd)
- {
- luat_fs_fclose(g_s_ftp.fd);
- g_s_ftp.fd = NULL;
- }
- g_s_ftp.fd = luat_fs_fopen(local_name, "wb+");
- if (g_s_ftp.fd == NULL) {
- LLOGE("open download file fail %s", local_name);
- return -1;
- }
- g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
- memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PULL, 0, 0, 0, 0);
- return 0;
- }
- int luat_ftp_close(void){
- if (!g_s_ftp.network){
- LLOGE("please login first");
- return -1;
- }
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_CLOSE, 0, 0, 0, LUAT_WAIT_FOREVER);
- return 0;
- }
- int luat_ftp_push(const char * local_name,const char * remote_name){
- if (!g_s_ftp.network){
- LLOGE("please login first");
- return -1;
- }
- if (g_s_ftp.fd)
- {
- luat_fs_fclose(g_s_ftp.fd);
- g_s_ftp.fd = NULL;
- }
- g_s_ftp.fd = luat_fs_fopen(local_name, "rb");
- if (g_s_ftp.fd == NULL) {
- LLOGE("open download file fail %s", local_name);
- return -1;
- }
- g_s_ftp.network->local_file_size = luat_fs_fsize(local_name);
- memcpy(g_s_ftp.network->remote_name, remote_name, strlen(remote_name) + 1);
- luat_rtos_event_send(g_s_ftp.task_handle, FTP_EVENT_PUSH, 0, 0, 0, LUAT_WAIT_FOREVER);
- return 0;
- }
- void luat_ftp_debug(uint8_t on_off) {
- g_s_ftp.debug_onoff = on_off;
- if (NULL != g_s_ftp.network && g_s_ftp.network->cmd_netc) {
- g_s_ftp.network->cmd_netc->is_debug = on_off;
- }
- }
|