ftp.c 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152
  1. /*
  2. * This file is part of the MicroPython project, http://micropython.org/
  3. *
  4. * The MIT License (MIT)
  5. *
  6. * Copyright (c) 2015 Daniel Campora
  7. *
  8. * Permission is hereby granted, free of charge, to any person obtaining a copy
  9. * of this software and associated documentation files (the "Software"), to deal
  10. * in the Software without restriction, including without limitation the rights
  11. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12. * copies of the Software, and to permit persons to whom the Software is
  13. * furnished to do so, subject to the following conditions:
  14. *
  15. * The above copyright notice and this permission notice shall be included in
  16. * all copies or substantial portions of the Software.
  17. *
  18. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24. * THE SOFTWARE.
  25. */
  26. #include <stdint.h>
  27. #include <stdio.h>
  28. #include "py/runtime.h"
  29. #include "lib/timeutils/timeutils.h"
  30. #include "lib/oofatfs/ff.h"
  31. #include "extmod/vfs.h"
  32. #include "extmod/vfs_fat.h"
  33. #include "inc/hw_types.h"
  34. #include "inc/hw_ints.h"
  35. #include "inc/hw_memmap.h"
  36. #include "rom_map.h"
  37. #include "prcm.h"
  38. #include "pybrtc.h"
  39. #include "ftp.h"
  40. #include "simplelink.h"
  41. #include "modnetwork.h"
  42. #include "modwlan.h"
  43. #include "modusocket.h"
  44. #include "debug.h"
  45. #include "serverstask.h"
  46. #include "fifo.h"
  47. #include "socketfifo.h"
  48. #include "updater.h"
  49. #include "moduos.h"
  50. /******************************************************************************
  51. DEFINE PRIVATE CONSTANTS
  52. ******************************************************************************/
  53. #define FTP_CMD_PORT 21
  54. #define FTP_ACTIVE_DATA_PORT 20
  55. #define FTP_PASIVE_DATA_PORT 2024
  56. #define FTP_BUFFER_SIZE 512
  57. #define FTP_TX_RETRIES_MAX 25
  58. #define FTP_CMD_SIZE_MAX 6
  59. #define FTP_CMD_CLIENTS_MAX 1
  60. #define FTP_DATA_CLIENTS_MAX 1
  61. #define FTP_MAX_PARAM_SIZE (MICROPY_ALLOC_PATH_MAX + 1)
  62. #define FTP_UNIX_TIME_20000101 946684800
  63. #define FTP_UNIX_TIME_20150101 1420070400
  64. #define FTP_UNIX_SECONDS_180_DAYS 15552000
  65. #define FTP_DATA_TIMEOUT_MS 5000 // 5 seconds
  66. #define FTP_SOCKETFIFO_ELEMENTS_MAX 4
  67. #define FTP_CYCLE_TIME_MS (SERVERS_CYCLE_TIME_MS * 2)
  68. /******************************************************************************
  69. DEFINE PRIVATE TYPES
  70. ******************************************************************************/
  71. typedef enum {
  72. E_FTP_RESULT_OK = 0,
  73. E_FTP_RESULT_CONTINUE,
  74. E_FTP_RESULT_FAILED
  75. } ftp_result_t;
  76. typedef enum {
  77. E_FTP_STE_DISABLED = 0,
  78. E_FTP_STE_START,
  79. E_FTP_STE_READY,
  80. E_FTP_STE_END_TRANSFER,
  81. E_FTP_STE_CONTINUE_LISTING,
  82. E_FTP_STE_CONTINUE_FILE_TX,
  83. E_FTP_STE_CONTINUE_FILE_RX
  84. } ftp_state_t;
  85. typedef enum {
  86. E_FTP_STE_SUB_DISCONNECTED = 0,
  87. E_FTP_STE_SUB_LISTEN_FOR_DATA,
  88. E_FTP_STE_SUB_DATA_CONNECTED
  89. } ftp_substate_t;
  90. typedef struct {
  91. bool uservalid : 1;
  92. bool passvalid : 1;
  93. } ftp_loggin_t;
  94. typedef enum {
  95. E_FTP_NOTHING_OPEN = 0,
  96. E_FTP_FILE_OPEN,
  97. E_FTP_DIR_OPEN
  98. } ftp_e_open_t;
  99. typedef enum {
  100. E_FTP_CLOSE_NONE = 0,
  101. E_FTP_CLOSE_DATA,
  102. E_FTP_CLOSE_CMD_AND_DATA,
  103. } ftp_e_closesocket_t;
  104. typedef struct {
  105. uint8_t *dBuffer;
  106. uint32_t ctimeout;
  107. union {
  108. FF_DIR dp;
  109. FIL fp;
  110. };
  111. int16_t lc_sd;
  112. int16_t ld_sd;
  113. int16_t c_sd;
  114. int16_t d_sd;
  115. int16_t dtimeout;
  116. uint16_t volcount;
  117. uint8_t state;
  118. uint8_t substate;
  119. uint8_t txRetries;
  120. uint8_t logginRetries;
  121. ftp_loggin_t loggin;
  122. uint8_t e_open;
  123. bool closechild;
  124. bool enabled;
  125. bool special_file;
  126. bool listroot;
  127. } ftp_data_t;
  128. typedef struct {
  129. char * cmd;
  130. } ftp_cmd_t;
  131. typedef struct {
  132. char * month;
  133. } ftp_month_t;
  134. typedef enum {
  135. E_FTP_CMD_NOT_SUPPORTED = -1,
  136. E_FTP_CMD_FEAT = 0,
  137. E_FTP_CMD_SYST,
  138. E_FTP_CMD_CDUP,
  139. E_FTP_CMD_CWD,
  140. E_FTP_CMD_PWD,
  141. E_FTP_CMD_XPWD,
  142. E_FTP_CMD_SIZE,
  143. E_FTP_CMD_MDTM,
  144. E_FTP_CMD_TYPE,
  145. E_FTP_CMD_USER,
  146. E_FTP_CMD_PASS,
  147. E_FTP_CMD_PASV,
  148. E_FTP_CMD_LIST,
  149. E_FTP_CMD_RETR,
  150. E_FTP_CMD_STOR,
  151. E_FTP_CMD_DELE,
  152. E_FTP_CMD_RMD,
  153. E_FTP_CMD_MKD,
  154. E_FTP_CMD_RNFR,
  155. E_FTP_CMD_RNTO,
  156. E_FTP_CMD_NOOP,
  157. E_FTP_CMD_QUIT,
  158. E_FTP_NUM_FTP_CMDS
  159. } ftp_cmd_index_t;
  160. /******************************************************************************
  161. DECLARE PRIVATE DATA
  162. ******************************************************************************/
  163. static ftp_data_t ftp_data;
  164. static char *ftp_path;
  165. static char *ftp_scratch_buffer;
  166. static char *ftp_cmd_buffer;
  167. static const ftp_cmd_t ftp_cmd_table[] = { { "FEAT" }, { "SYST" }, { "CDUP" }, { "CWD" },
  168. { "PWD" }, { "XPWD" }, { "SIZE" }, { "MDTM" },
  169. { "TYPE" }, { "USER" }, { "PASS" }, { "PASV" },
  170. { "LIST" }, { "RETR" }, { "STOR" }, { "DELE" },
  171. { "RMD" }, { "MKD" }, { "RNFR" }, { "RNTO" },
  172. { "NOOP" }, { "QUIT" } };
  173. static const ftp_month_t ftp_month[] = { { "Jan" }, { "Feb" }, { "Mar" }, { "Apr" },
  174. { "May" }, { "Jun" }, { "Jul" }, { "Ago" },
  175. { "Sep" }, { "Oct" }, { "Nov" }, { "Dec" } };
  176. static SocketFifoElement_t ftp_fifoelements[FTP_SOCKETFIFO_ELEMENTS_MAX];
  177. static FIFO_t ftp_socketfifo;
  178. /******************************************************************************
  179. DEFINE VFS WRAPPER FUNCTIONS
  180. ******************************************************************************/
  181. // These wrapper functions are used so that the FTP server can access the
  182. // mounted FATFS devices directly without going through the costly mp_vfs_XXX
  183. // functions. The latter may raise exceptions and we would then need to wrap
  184. // all calls in an nlr handler. The wrapper functions below assume that there
  185. // are only FATFS filesystems mounted.
  186. STATIC FATFS *lookup_path(const TCHAR **path) {
  187. mp_vfs_mount_t *fs = mp_vfs_lookup_path(*path, path);
  188. if (fs == MP_VFS_NONE || fs == MP_VFS_ROOT) {
  189. return NULL;
  190. }
  191. // here we assume that the mounted device is FATFS
  192. return &((fs_user_mount_t*)MP_OBJ_TO_PTR(fs->obj))->fatfs;
  193. }
  194. STATIC FRESULT f_open_helper(FIL *fp, const TCHAR *path, BYTE mode) {
  195. FATFS *fs = lookup_path(&path);
  196. if (fs == NULL) {
  197. return FR_NO_PATH;
  198. }
  199. return f_open(fs, fp, path, mode);
  200. }
  201. STATIC FRESULT f_opendir_helper(FF_DIR *dp, const TCHAR *path) {
  202. FATFS *fs = lookup_path(&path);
  203. if (fs == NULL) {
  204. return FR_NO_PATH;
  205. }
  206. return f_opendir(fs, dp, path);
  207. }
  208. STATIC FRESULT f_stat_helper(const TCHAR *path, FILINFO *fno) {
  209. FATFS *fs = lookup_path(&path);
  210. if (fs == NULL) {
  211. return FR_NO_PATH;
  212. }
  213. return f_stat(fs, path, fno);
  214. }
  215. STATIC FRESULT f_mkdir_helper(const TCHAR *path) {
  216. FATFS *fs = lookup_path(&path);
  217. if (fs == NULL) {
  218. return FR_NO_PATH;
  219. }
  220. return f_mkdir(fs, path);
  221. }
  222. STATIC FRESULT f_unlink_helper(const TCHAR *path) {
  223. FATFS *fs = lookup_path(&path);
  224. if (fs == NULL) {
  225. return FR_NO_PATH;
  226. }
  227. return f_unlink(fs, path);
  228. }
  229. STATIC FRESULT f_rename_helper(const TCHAR *path_old, const TCHAR *path_new) {
  230. FATFS *fs_old = lookup_path(&path_old);
  231. if (fs_old == NULL) {
  232. return FR_NO_PATH;
  233. }
  234. FATFS *fs_new = lookup_path(&path_new);
  235. if (fs_new == NULL) {
  236. return FR_NO_PATH;
  237. }
  238. if (fs_old != fs_new) {
  239. return FR_NO_PATH;
  240. }
  241. return f_rename(fs_new, path_old, path_new);
  242. }
  243. /******************************************************************************
  244. DECLARE PRIVATE FUNCTIONS
  245. ******************************************************************************/
  246. static void ftp_wait_for_enabled (void);
  247. static bool ftp_create_listening_socket (_i16 *sd, _u16 port, _u8 backlog);
  248. static ftp_result_t ftp_wait_for_connection (_i16 l_sd, _i16 *n_sd);
  249. static ftp_result_t ftp_send_non_blocking (_i16 sd, void *data, _i16 Len);
  250. static void ftp_send_reply (_u16 status, char *message);
  251. static void ftp_send_data (_u32 datasize);
  252. static void ftp_send_from_fifo (void);
  253. static ftp_result_t ftp_recv_non_blocking (_i16 sd, void *buff, _i16 Maxlen, _i32 *rxLen);
  254. static void ftp_process_cmd (void);
  255. static void ftp_close_files (void);
  256. static void ftp_close_filesystem_on_error (void);
  257. static void ftp_close_cmd_data (void);
  258. static ftp_cmd_index_t ftp_pop_command (char **str);
  259. static void ftp_pop_param (char **str, char *param);
  260. static int ftp_print_eplf_item (char *dest, uint32_t destsize, FILINFO *fno);
  261. static int ftp_print_eplf_drive (char *dest, uint32_t destsize, const char *name);
  262. static bool ftp_open_file (const char *path, int mode);
  263. static ftp_result_t ftp_read_file (char *filebuf, uint32_t desiredsize, uint32_t *actualsize);
  264. static ftp_result_t ftp_write_file (char *filebuf, uint32_t size);
  265. static ftp_result_t ftp_open_dir_for_listing (const char *path);
  266. static ftp_result_t ftp_list_dir (char *list, uint32_t maxlistsize, uint32_t *listsize);
  267. static void ftp_open_child (char *pwd, char *dir);
  268. static void ftp_close_child (char *pwd);
  269. static void ftp_return_to_previous_path (char *pwd, char *dir);
  270. /******************************************************************************
  271. DEFINE PUBLIC FUNCTIONS
  272. ******************************************************************************/
  273. void ftp_init (void) {
  274. // Allocate memory for the data buffer, and the file system structs (from the RTOS heap)
  275. ASSERT ((ftp_data.dBuffer = mem_Malloc(FTP_BUFFER_SIZE)) != NULL);
  276. ASSERT ((ftp_path = mem_Malloc(FTP_MAX_PARAM_SIZE)) != NULL);
  277. ASSERT ((ftp_scratch_buffer = mem_Malloc(FTP_MAX_PARAM_SIZE)) != NULL);
  278. ASSERT ((ftp_cmd_buffer = mem_Malloc(FTP_MAX_PARAM_SIZE + FTP_CMD_SIZE_MAX)) != NULL);
  279. SOCKETFIFO_Init (&ftp_socketfifo, (void *)ftp_fifoelements, FTP_SOCKETFIFO_ELEMENTS_MAX);
  280. ftp_data.c_sd = -1;
  281. ftp_data.d_sd = -1;
  282. ftp_data.lc_sd = -1;
  283. ftp_data.ld_sd = -1;
  284. ftp_data.e_open = E_FTP_NOTHING_OPEN;
  285. ftp_data.state = E_FTP_STE_DISABLED;
  286. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  287. ftp_data.special_file = false;
  288. ftp_data.volcount = 0;
  289. }
  290. void ftp_run (void) {
  291. switch (ftp_data.state) {
  292. case E_FTP_STE_DISABLED:
  293. ftp_wait_for_enabled();
  294. break;
  295. case E_FTP_STE_START:
  296. if (wlan_is_connected() && ftp_create_listening_socket(&ftp_data.lc_sd, FTP_CMD_PORT, FTP_CMD_CLIENTS_MAX - 1)) {
  297. ftp_data.state = E_FTP_STE_READY;
  298. }
  299. break;
  300. case E_FTP_STE_READY:
  301. if (ftp_data.c_sd < 0 && ftp_data.substate == E_FTP_STE_SUB_DISCONNECTED) {
  302. if (E_FTP_RESULT_OK == ftp_wait_for_connection(ftp_data.lc_sd, &ftp_data.c_sd)) {
  303. ftp_data.txRetries = 0;
  304. ftp_data.logginRetries = 0;
  305. ftp_data.ctimeout = 0;
  306. ftp_data.loggin.uservalid = false;
  307. ftp_data.loggin.passvalid = false;
  308. strcpy (ftp_path, "/");
  309. ftp_send_reply (220, "MicroPython FTP Server");
  310. break;
  311. }
  312. }
  313. if (SOCKETFIFO_IsEmpty()) {
  314. if (ftp_data.c_sd > 0 && ftp_data.substate != E_FTP_STE_SUB_LISTEN_FOR_DATA) {
  315. ftp_process_cmd();
  316. if (ftp_data.state != E_FTP_STE_READY) {
  317. break;
  318. }
  319. }
  320. }
  321. break;
  322. case E_FTP_STE_END_TRANSFER:
  323. break;
  324. case E_FTP_STE_CONTINUE_LISTING:
  325. // go on with listing only if the transmit buffer is empty
  326. if (SOCKETFIFO_IsEmpty()) {
  327. uint32_t listsize;
  328. ftp_list_dir((char *)ftp_data.dBuffer, FTP_BUFFER_SIZE, &listsize);
  329. if (listsize > 0) {
  330. ftp_send_data(listsize);
  331. } else {
  332. ftp_send_reply(226, NULL);
  333. ftp_data.state = E_FTP_STE_END_TRANSFER;
  334. }
  335. ftp_data.ctimeout = 0;
  336. }
  337. break;
  338. case E_FTP_STE_CONTINUE_FILE_TX:
  339. // read the next block from the file only if the previous one has been sent
  340. if (SOCKETFIFO_IsEmpty()) {
  341. uint32_t readsize;
  342. ftp_result_t result;
  343. ftp_data.ctimeout = 0;
  344. result = ftp_read_file ((char *)ftp_data.dBuffer, FTP_BUFFER_SIZE, &readsize);
  345. if (result == E_FTP_RESULT_FAILED) {
  346. ftp_send_reply(451, NULL);
  347. ftp_data.state = E_FTP_STE_END_TRANSFER;
  348. }
  349. else {
  350. if (readsize > 0) {
  351. ftp_send_data(readsize);
  352. }
  353. if (result == E_FTP_RESULT_OK) {
  354. ftp_send_reply(226, NULL);
  355. ftp_data.state = E_FTP_STE_END_TRANSFER;
  356. }
  357. }
  358. }
  359. break;
  360. case E_FTP_STE_CONTINUE_FILE_RX:
  361. if (SOCKETFIFO_IsEmpty()) {
  362. _i32 len;
  363. ftp_result_t result;
  364. if (E_FTP_RESULT_OK == (result = ftp_recv_non_blocking(ftp_data.d_sd, ftp_data.dBuffer, FTP_BUFFER_SIZE, &len))) {
  365. ftp_data.dtimeout = 0;
  366. ftp_data.ctimeout = 0;
  367. // its a software update
  368. if (ftp_data.special_file) {
  369. if (updater_write(ftp_data.dBuffer, len)) {
  370. break;
  371. }
  372. }
  373. // user file being received
  374. else if (E_FTP_RESULT_OK == ftp_write_file ((char *)ftp_data.dBuffer, len)) {
  375. break;
  376. }
  377. ftp_send_reply(451, NULL);
  378. ftp_data.state = E_FTP_STE_END_TRANSFER;
  379. }
  380. else if (result == E_FTP_RESULT_CONTINUE) {
  381. if (ftp_data.dtimeout++ > FTP_DATA_TIMEOUT_MS / FTP_CYCLE_TIME_MS) {
  382. ftp_close_files();
  383. ftp_send_reply(426, NULL);
  384. ftp_data.state = E_FTP_STE_END_TRANSFER;
  385. }
  386. }
  387. else {
  388. if (ftp_data.special_file) {
  389. ftp_data.special_file = false;
  390. updater_finnish();
  391. }
  392. ftp_close_files();
  393. ftp_send_reply(226, NULL);
  394. ftp_data.state = E_FTP_STE_END_TRANSFER;
  395. }
  396. }
  397. break;
  398. default:
  399. break;
  400. }
  401. switch (ftp_data.substate) {
  402. case E_FTP_STE_SUB_DISCONNECTED:
  403. break;
  404. case E_FTP_STE_SUB_LISTEN_FOR_DATA:
  405. if (E_FTP_RESULT_OK == ftp_wait_for_connection(ftp_data.ld_sd, &ftp_data.d_sd)) {
  406. ftp_data.dtimeout = 0;
  407. ftp_data.substate = E_FTP_STE_SUB_DATA_CONNECTED;
  408. }
  409. else if (ftp_data.dtimeout++ > FTP_DATA_TIMEOUT_MS / FTP_CYCLE_TIME_MS) {
  410. ftp_data.dtimeout = 0;
  411. // close the listening socket
  412. servers_close_socket(&ftp_data.ld_sd);
  413. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  414. }
  415. break;
  416. case E_FTP_STE_SUB_DATA_CONNECTED:
  417. if (ftp_data.state == E_FTP_STE_READY && ftp_data.dtimeout++ > FTP_DATA_TIMEOUT_MS / FTP_CYCLE_TIME_MS) {
  418. // close the listening and the data socket
  419. servers_close_socket(&ftp_data.ld_sd);
  420. servers_close_socket(&ftp_data.d_sd);
  421. ftp_close_filesystem_on_error ();
  422. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  423. }
  424. break;
  425. default:
  426. break;
  427. }
  428. // send data pending in the queue
  429. ftp_send_from_fifo();
  430. // check the state of the data sockets
  431. if (ftp_data.d_sd < 0 && (ftp_data.state > E_FTP_STE_READY)) {
  432. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  433. ftp_data.state = E_FTP_STE_READY;
  434. }
  435. }
  436. void ftp_enable (void) {
  437. ftp_data.enabled = true;
  438. }
  439. void ftp_disable (void) {
  440. ftp_reset();
  441. ftp_data.enabled = false;
  442. ftp_data.state = E_FTP_STE_DISABLED;
  443. }
  444. void ftp_reset (void) {
  445. // close all connections and start all over again
  446. servers_close_socket(&ftp_data.lc_sd);
  447. servers_close_socket(&ftp_data.ld_sd);
  448. ftp_close_cmd_data();
  449. ftp_data.state = E_FTP_STE_START;
  450. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  451. ftp_data.volcount = 0;
  452. SOCKETFIFO_Flush();
  453. }
  454. /******************************************************************************
  455. DEFINE PRIVATE FUNCTIONS
  456. ******************************************************************************/
  457. static void ftp_wait_for_enabled (void) {
  458. // Check if the telnet service has been enabled
  459. if (ftp_data.enabled) {
  460. ftp_data.state = E_FTP_STE_START;
  461. }
  462. }
  463. static bool ftp_create_listening_socket (_i16 *sd, _u16 port, _u8 backlog) {
  464. SlSockNonblocking_t nonBlockingOption;
  465. SlSockAddrIn_t sServerAddress;
  466. _i16 _sd;
  467. _i16 result;
  468. // Open a socket for ftp data listen
  469. ASSERT ((*sd = sl_Socket(SL_AF_INET, SL_SOCK_STREAM, SL_IPPROTO_IP)) > 0);
  470. _sd = *sd;
  471. if (_sd > 0) {
  472. // add the new socket to the network administration
  473. modusocket_socket_add(_sd, false);
  474. // Enable non-blocking mode
  475. nonBlockingOption.NonblockingEnabled = 1;
  476. ASSERT ((result = sl_SetSockOpt(_sd, SL_SOL_SOCKET, SL_SO_NONBLOCKING, &nonBlockingOption, sizeof(nonBlockingOption))) == SL_SOC_OK);
  477. // Bind the socket to a port number
  478. sServerAddress.sin_family = SL_AF_INET;
  479. sServerAddress.sin_addr.s_addr = SL_INADDR_ANY;
  480. sServerAddress.sin_port = sl_Htons(port);
  481. ASSERT ((result |= sl_Bind(_sd, (const SlSockAddr_t *)&sServerAddress, sizeof(sServerAddress))) == SL_SOC_OK);
  482. // Start listening
  483. ASSERT ((result |= sl_Listen (_sd, backlog)) == SL_SOC_OK);
  484. if (result == SL_SOC_OK) {
  485. return true;
  486. }
  487. servers_close_socket(sd);
  488. }
  489. return false;
  490. }
  491. static ftp_result_t ftp_wait_for_connection (_i16 l_sd, _i16 *n_sd) {
  492. SlSockAddrIn_t sClientAddress;
  493. SlSocklen_t in_addrSize;
  494. // accepts a connection from a TCP client, if there is any, otherwise returns SL_EAGAIN
  495. *n_sd = sl_Accept(l_sd, (SlSockAddr_t *)&sClientAddress, (SlSocklen_t *)&in_addrSize);
  496. _i16 _sd = *n_sd;
  497. if (_sd == SL_EAGAIN) {
  498. return E_FTP_RESULT_CONTINUE;
  499. }
  500. else if (_sd < 0) {
  501. // error
  502. ftp_reset();
  503. return E_FTP_RESULT_FAILED;
  504. }
  505. // add the new socket to the network administration
  506. modusocket_socket_add(_sd, false);
  507. // client connected, so go on
  508. return E_FTP_RESULT_OK;
  509. }
  510. static ftp_result_t ftp_send_non_blocking (_i16 sd, void *data, _i16 Len) {
  511. int16_t result = sl_Send(sd, data, Len, 0);
  512. if (result > 0) {
  513. ftp_data.txRetries = 0;
  514. return E_FTP_RESULT_OK;
  515. }
  516. else if ((FTP_TX_RETRIES_MAX >= ++ftp_data.txRetries) && (result == SL_EAGAIN)) {
  517. return E_FTP_RESULT_CONTINUE;
  518. }
  519. else {
  520. // error
  521. ftp_reset();
  522. return E_FTP_RESULT_FAILED;
  523. }
  524. }
  525. static void ftp_send_reply (_u16 status, char *message) {
  526. SocketFifoElement_t fifoelement;
  527. if (!message) {
  528. message = "";
  529. }
  530. snprintf((char *)ftp_cmd_buffer, 4, "%u", status);
  531. strcat ((char *)ftp_cmd_buffer, " ");
  532. strcat ((char *)ftp_cmd_buffer, message);
  533. strcat ((char *)ftp_cmd_buffer, "\r\n");
  534. fifoelement.sd = &ftp_data.c_sd;
  535. fifoelement.datasize = strlen((char *)ftp_cmd_buffer);
  536. fifoelement.data = mem_Malloc(fifoelement.datasize);
  537. if (status == 221) {
  538. fifoelement.closesockets = E_FTP_CLOSE_CMD_AND_DATA;
  539. }
  540. else if (status == 426 || status == 451 || status == 550) {
  541. fifoelement.closesockets = E_FTP_CLOSE_DATA;
  542. }
  543. else {
  544. fifoelement.closesockets = E_FTP_CLOSE_NONE;
  545. }
  546. fifoelement.freedata = true;
  547. if (fifoelement.data) {
  548. memcpy (fifoelement.data, ftp_cmd_buffer, fifoelement.datasize);
  549. if (!SOCKETFIFO_Push (&fifoelement)) {
  550. mem_Free(fifoelement.data);
  551. }
  552. }
  553. }
  554. static void ftp_send_data (_u32 datasize) {
  555. SocketFifoElement_t fifoelement;
  556. fifoelement.data = ftp_data.dBuffer;
  557. fifoelement.datasize = datasize;
  558. fifoelement.sd = &ftp_data.d_sd;
  559. fifoelement.closesockets = E_FTP_CLOSE_NONE;
  560. fifoelement.freedata = false;
  561. SOCKETFIFO_Push (&fifoelement);
  562. }
  563. static void ftp_send_from_fifo (void) {
  564. SocketFifoElement_t fifoelement;
  565. if (SOCKETFIFO_Peek (&fifoelement)) {
  566. _i16 _sd = *fifoelement.sd;
  567. if (_sd > 0) {
  568. if (E_FTP_RESULT_OK == ftp_send_non_blocking (_sd, fifoelement.data, fifoelement.datasize)) {
  569. SOCKETFIFO_Pop (&fifoelement);
  570. if (fifoelement.closesockets != E_FTP_CLOSE_NONE) {
  571. servers_close_socket(&ftp_data.d_sd);
  572. if (fifoelement.closesockets == E_FTP_CLOSE_CMD_AND_DATA) {
  573. servers_close_socket(&ftp_data.ld_sd);
  574. // this one is the command socket
  575. servers_close_socket(fifoelement.sd);
  576. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  577. }
  578. ftp_close_filesystem_on_error();
  579. }
  580. if (fifoelement.freedata) {
  581. mem_Free(fifoelement.data);
  582. }
  583. }
  584. }
  585. // socket closed, remove from the queue
  586. else {
  587. SOCKETFIFO_Pop (&fifoelement);
  588. if (fifoelement.freedata) {
  589. mem_Free(fifoelement.data);
  590. }
  591. }
  592. }
  593. else if (ftp_data.state == E_FTP_STE_END_TRANSFER && (ftp_data.d_sd > 0)) {
  594. // close the listening and the data sockets
  595. servers_close_socket(&ftp_data.ld_sd);
  596. servers_close_socket(&ftp_data.d_sd);
  597. if (ftp_data.special_file) {
  598. ftp_data.special_file = false;
  599. }
  600. }
  601. }
  602. static ftp_result_t ftp_recv_non_blocking (_i16 sd, void *buff, _i16 Maxlen, _i32 *rxLen) {
  603. *rxLen = sl_Recv(sd, buff, Maxlen, 0);
  604. if (*rxLen > 0) {
  605. return E_FTP_RESULT_OK;
  606. }
  607. else if (*rxLen != SL_EAGAIN) {
  608. // error
  609. return E_FTP_RESULT_FAILED;
  610. }
  611. return E_FTP_RESULT_CONTINUE;
  612. }
  613. static void ftp_get_param_and_open_child (char **bufptr) {
  614. ftp_pop_param (bufptr, ftp_scratch_buffer);
  615. ftp_open_child (ftp_path, ftp_scratch_buffer);
  616. ftp_data.closechild = true;
  617. }
  618. static void ftp_process_cmd (void) {
  619. _i32 len;
  620. char *bufptr = (char *)ftp_cmd_buffer;
  621. ftp_result_t result;
  622. FRESULT fres;
  623. FILINFO fno;
  624. ftp_data.closechild = false;
  625. // also use the reply buffer to receive new commands
  626. if (E_FTP_RESULT_OK == (result = ftp_recv_non_blocking(ftp_data.c_sd, ftp_cmd_buffer, FTP_MAX_PARAM_SIZE + FTP_CMD_SIZE_MAX, &len))) {
  627. // bufptr is moved as commands are being popped
  628. ftp_cmd_index_t cmd = ftp_pop_command(&bufptr);
  629. if (!ftp_data.loggin.passvalid && (cmd != E_FTP_CMD_USER && cmd != E_FTP_CMD_PASS && cmd != E_FTP_CMD_QUIT)) {
  630. ftp_send_reply(332, NULL);
  631. return;
  632. }
  633. switch (cmd) {
  634. case E_FTP_CMD_FEAT:
  635. ftp_send_reply(211, "no-features");
  636. break;
  637. case E_FTP_CMD_SYST:
  638. ftp_send_reply(215, "UNIX Type: L8");
  639. break;
  640. case E_FTP_CMD_CDUP:
  641. ftp_close_child(ftp_path);
  642. ftp_send_reply(250, NULL);
  643. break;
  644. case E_FTP_CMD_CWD:
  645. {
  646. fres = FR_NO_PATH;
  647. ftp_pop_param (&bufptr, ftp_scratch_buffer);
  648. ftp_open_child (ftp_path, ftp_scratch_buffer);
  649. if ((ftp_path[0] == '/' && ftp_path[1] == '\0') || ((fres = f_opendir_helper (&ftp_data.dp, ftp_path)) == FR_OK)) {
  650. if (fres == FR_OK) {
  651. f_closedir(&ftp_data.dp);
  652. }
  653. ftp_send_reply(250, NULL);
  654. }
  655. else {
  656. ftp_close_child (ftp_path);
  657. ftp_send_reply(550, NULL);
  658. }
  659. }
  660. break;
  661. case E_FTP_CMD_PWD:
  662. case E_FTP_CMD_XPWD:
  663. ftp_send_reply(257, ftp_path);
  664. break;
  665. case E_FTP_CMD_SIZE:
  666. {
  667. ftp_get_param_and_open_child (&bufptr);
  668. if (FR_OK == f_stat_helper(ftp_path, &fno)) {
  669. // send the size
  670. snprintf((char *)ftp_data.dBuffer, FTP_BUFFER_SIZE, "%u", (_u32)fno.fsize);
  671. ftp_send_reply(213, (char *)ftp_data.dBuffer);
  672. }
  673. else {
  674. ftp_send_reply(550, NULL);
  675. }
  676. }
  677. break;
  678. case E_FTP_CMD_MDTM:
  679. ftp_get_param_and_open_child (&bufptr);
  680. if (FR_OK == f_stat_helper(ftp_path, &fno)) {
  681. // send the last modified time
  682. snprintf((char *)ftp_data.dBuffer, FTP_BUFFER_SIZE, "%u%02u%02u%02u%02u%02u",
  683. 1980 + ((fno.fdate >> 9) & 0x7f), (fno.fdate >> 5) & 0x0f,
  684. fno.fdate & 0x1f, (fno.ftime >> 11) & 0x1f,
  685. (fno.ftime >> 5) & 0x3f, 2 * (fno.ftime & 0x1f));
  686. ftp_send_reply(213, (char *)ftp_data.dBuffer);
  687. }
  688. else {
  689. ftp_send_reply(550, NULL);
  690. }
  691. break;
  692. case E_FTP_CMD_TYPE:
  693. ftp_send_reply(200, NULL);
  694. break;
  695. case E_FTP_CMD_USER:
  696. ftp_pop_param (&bufptr, ftp_scratch_buffer);
  697. if (!memcmp(ftp_scratch_buffer, servers_user, MAX(strlen(ftp_scratch_buffer), strlen(servers_user)))) {
  698. ftp_data.loggin.uservalid = true && (strlen(servers_user) == strlen(ftp_scratch_buffer));
  699. }
  700. ftp_send_reply(331, NULL);
  701. break;
  702. case E_FTP_CMD_PASS:
  703. ftp_pop_param (&bufptr, ftp_scratch_buffer);
  704. if (!memcmp(ftp_scratch_buffer, servers_pass, MAX(strlen(ftp_scratch_buffer), strlen(servers_pass))) &&
  705. ftp_data.loggin.uservalid) {
  706. ftp_data.loggin.passvalid = true && (strlen(servers_pass) == strlen(ftp_scratch_buffer));
  707. if (ftp_data.loggin.passvalid) {
  708. ftp_send_reply(230, NULL);
  709. break;
  710. }
  711. }
  712. ftp_send_reply(530, NULL);
  713. break;
  714. case E_FTP_CMD_PASV:
  715. {
  716. // some servers (e.g. google chrome) send PASV several times very quickly
  717. servers_close_socket(&ftp_data.d_sd);
  718. ftp_data.substate = E_FTP_STE_SUB_DISCONNECTED;
  719. bool socketcreated = true;
  720. if (ftp_data.ld_sd < 0) {
  721. socketcreated = ftp_create_listening_socket(&ftp_data.ld_sd, FTP_PASIVE_DATA_PORT, FTP_DATA_CLIENTS_MAX - 1);
  722. }
  723. if (socketcreated) {
  724. uint32_t ip;
  725. uint8_t *pip = (uint8_t *)&ip;
  726. ftp_data.dtimeout = 0;
  727. wlan_get_ip(&ip);
  728. snprintf((char *)ftp_data.dBuffer, FTP_BUFFER_SIZE, "(%u,%u,%u,%u,%u,%u)",
  729. pip[3], pip[2], pip[1], pip[0], (FTP_PASIVE_DATA_PORT >> 8), (FTP_PASIVE_DATA_PORT & 0xFF));
  730. ftp_data.substate = E_FTP_STE_SUB_LISTEN_FOR_DATA;
  731. ftp_send_reply(227, (char *)ftp_data.dBuffer);
  732. }
  733. else {
  734. ftp_send_reply(425, NULL);
  735. }
  736. }
  737. break;
  738. case E_FTP_CMD_LIST:
  739. if (ftp_open_dir_for_listing(ftp_path) == E_FTP_RESULT_CONTINUE) {
  740. ftp_data.state = E_FTP_STE_CONTINUE_LISTING;
  741. ftp_send_reply(150, NULL);
  742. }
  743. else {
  744. ftp_send_reply(550, NULL);
  745. }
  746. break;
  747. case E_FTP_CMD_RETR:
  748. ftp_get_param_and_open_child (&bufptr);
  749. if (ftp_open_file (ftp_path, FA_READ)) {
  750. ftp_data.state = E_FTP_STE_CONTINUE_FILE_TX;
  751. ftp_send_reply(150, NULL);
  752. }
  753. else {
  754. ftp_data.state = E_FTP_STE_END_TRANSFER;
  755. ftp_send_reply(550, NULL);
  756. }
  757. break;
  758. case E_FTP_CMD_STOR:
  759. ftp_get_param_and_open_child (&bufptr);
  760. // first check if a software update is being requested
  761. if (updater_check_path (ftp_path)) {
  762. if (updater_start()) {
  763. ftp_data.special_file = true;
  764. ftp_data.state = E_FTP_STE_CONTINUE_FILE_RX;
  765. ftp_send_reply(150, NULL);
  766. }
  767. else {
  768. // to unlock the updater
  769. updater_finnish();
  770. ftp_data.state = E_FTP_STE_END_TRANSFER;
  771. ftp_send_reply(550, NULL);
  772. }
  773. }
  774. else {
  775. if (ftp_open_file (ftp_path, FA_WRITE | FA_CREATE_ALWAYS)) {
  776. ftp_data.state = E_FTP_STE_CONTINUE_FILE_RX;
  777. ftp_send_reply(150, NULL);
  778. }
  779. else {
  780. ftp_data.state = E_FTP_STE_END_TRANSFER;
  781. ftp_send_reply(550, NULL);
  782. }
  783. }
  784. break;
  785. case E_FTP_CMD_DELE:
  786. case E_FTP_CMD_RMD:
  787. ftp_get_param_and_open_child (&bufptr);
  788. if (FR_OK == f_unlink_helper(ftp_path)) {
  789. ftp_send_reply(250, NULL);
  790. }
  791. else {
  792. ftp_send_reply(550, NULL);
  793. }
  794. break;
  795. case E_FTP_CMD_MKD:
  796. ftp_get_param_and_open_child (&bufptr);
  797. if (FR_OK == f_mkdir_helper(ftp_path)) {
  798. ftp_send_reply(250, NULL);
  799. }
  800. else {
  801. ftp_send_reply(550, NULL);
  802. }
  803. break;
  804. case E_FTP_CMD_RNFR:
  805. ftp_get_param_and_open_child (&bufptr);
  806. if (FR_OK == f_stat_helper(ftp_path, &fno)) {
  807. ftp_send_reply(350, NULL);
  808. // save the current path
  809. strcpy ((char *)ftp_data.dBuffer, ftp_path);
  810. }
  811. else {
  812. ftp_send_reply(550, NULL);
  813. }
  814. break;
  815. case E_FTP_CMD_RNTO:
  816. ftp_get_param_and_open_child (&bufptr);
  817. // old path was saved in the data buffer
  818. if (FR_OK == (fres = f_rename_helper((char *)ftp_data.dBuffer, ftp_path))) {
  819. ftp_send_reply(250, NULL);
  820. }
  821. else {
  822. ftp_send_reply(550, NULL);
  823. }
  824. break;
  825. case E_FTP_CMD_NOOP:
  826. ftp_send_reply(200, NULL);
  827. break;
  828. case E_FTP_CMD_QUIT:
  829. ftp_send_reply(221, NULL);
  830. break;
  831. default:
  832. // command not implemented
  833. ftp_send_reply(502, NULL);
  834. break;
  835. }
  836. if (ftp_data.closechild) {
  837. ftp_return_to_previous_path(ftp_path, ftp_scratch_buffer);
  838. }
  839. }
  840. else if (result == E_FTP_RESULT_CONTINUE) {
  841. if (ftp_data.ctimeout++ > (servers_get_timeout() / FTP_CYCLE_TIME_MS)) {
  842. ftp_send_reply(221, NULL);
  843. }
  844. }
  845. else {
  846. ftp_close_cmd_data ();
  847. }
  848. }
  849. static void ftp_close_files (void) {
  850. if (ftp_data.e_open == E_FTP_FILE_OPEN) {
  851. f_close(&ftp_data.fp);
  852. }
  853. else if (ftp_data.e_open == E_FTP_DIR_OPEN) {
  854. f_closedir(&ftp_data.dp);
  855. }
  856. ftp_data.e_open = E_FTP_NOTHING_OPEN;
  857. }
  858. static void ftp_close_filesystem_on_error (void) {
  859. ftp_close_files();
  860. if (ftp_data.special_file) {
  861. updater_finnish ();
  862. ftp_data.special_file = false;
  863. }
  864. }
  865. static void ftp_close_cmd_data (void) {
  866. servers_close_socket(&ftp_data.c_sd);
  867. servers_close_socket(&ftp_data.d_sd);
  868. ftp_close_filesystem_on_error ();
  869. }
  870. static void stoupper (char *str) {
  871. while (str && *str != '\0') {
  872. *str = (char)unichar_toupper((int)(*str));
  873. str++;
  874. }
  875. }
  876. static ftp_cmd_index_t ftp_pop_command (char **str) {
  877. char _cmd[FTP_CMD_SIZE_MAX];
  878. ftp_pop_param (str, _cmd);
  879. stoupper (_cmd);
  880. for (ftp_cmd_index_t i = 0; i < E_FTP_NUM_FTP_CMDS; i++) {
  881. if (!strcmp (_cmd, ftp_cmd_table[i].cmd)) {
  882. // move one step further to skip the space
  883. (*str)++;
  884. return i;
  885. }
  886. }
  887. return E_FTP_CMD_NOT_SUPPORTED;
  888. }
  889. static void ftp_pop_param (char **str, char *param) {
  890. while (**str != ' ' && **str != '\r' && **str != '\n' && **str != '\0') {
  891. *param++ = **str;
  892. (*str)++;
  893. }
  894. *param = '\0';
  895. }
  896. static int ftp_print_eplf_item (char *dest, uint32_t destsize, FILINFO *fno) {
  897. char *type = (fno->fattrib & AM_DIR) ? "d" : "-";
  898. uint32_t tseconds;
  899. uint mindex = (((fno->fdate >> 5) & 0x0f) > 0) ? (((fno->fdate >> 5) & 0x0f) - 1) : 0;
  900. uint day = ((fno->fdate & 0x1f) > 0) ? (fno->fdate & 0x1f) : 1;
  901. uint fseconds = timeutils_seconds_since_2000(1980 + ((fno->fdate >> 9) & 0x7f),
  902. (fno->fdate >> 5) & 0x0f,
  903. fno->fdate & 0x1f,
  904. (fno->ftime >> 11) & 0x1f,
  905. (fno->ftime >> 5) & 0x3f,
  906. 2 * (fno->ftime & 0x1f));
  907. tseconds = pyb_rtc_get_seconds();
  908. if (FTP_UNIX_SECONDS_180_DAYS < tseconds - fseconds) {
  909. return snprintf(dest, destsize, "%srw-rw-r-- 1 root root %9u %s %2u %5u %s\r\n",
  910. type, (_u32)fno->fsize, ftp_month[mindex].month, day,
  911. 1980 + ((fno->fdate >> 9) & 0x7f), fno->fname);
  912. }
  913. else {
  914. return snprintf(dest, destsize, "%srw-rw-r-- 1 root root %9u %s %2u %02u:%02u %s\r\n",
  915. type, (_u32)fno->fsize, ftp_month[mindex].month, day,
  916. (fno->ftime >> 11) & 0x1f, (fno->ftime >> 5) & 0x3f, fno->fname);
  917. }
  918. }
  919. static int ftp_print_eplf_drive (char *dest, uint32_t destsize, const char *name) {
  920. timeutils_struct_time_t tm;
  921. uint32_t tseconds;
  922. char *type = "d";
  923. timeutils_seconds_since_2000_to_struct_time((FTP_UNIX_TIME_20150101 - FTP_UNIX_TIME_20000101), &tm);
  924. tseconds = pyb_rtc_get_seconds();
  925. if (FTP_UNIX_SECONDS_180_DAYS < tseconds - (FTP_UNIX_TIME_20150101 - FTP_UNIX_TIME_20000101)) {
  926. return snprintf(dest, destsize, "%srw-rw-r-- 1 root root %9u %s %2u %5u %s\r\n",
  927. type, 0, ftp_month[(tm.tm_mon - 1)].month, tm.tm_mday, tm.tm_year, name);
  928. }
  929. else {
  930. return snprintf(dest, destsize, "%srw-rw-r-- 1 root root %9u %s %2u %02u:%02u %s\r\n",
  931. type, 0, ftp_month[(tm.tm_mon - 1)].month, tm.tm_mday, tm.tm_hour, tm.tm_min, name);
  932. }
  933. }
  934. static bool ftp_open_file (const char *path, int mode) {
  935. FRESULT res = f_open_helper(&ftp_data.fp, path, mode);
  936. if (res != FR_OK) {
  937. return false;
  938. }
  939. ftp_data.e_open = E_FTP_FILE_OPEN;
  940. return true;
  941. }
  942. static ftp_result_t ftp_read_file (char *filebuf, uint32_t desiredsize, uint32_t *actualsize) {
  943. ftp_result_t result = E_FTP_RESULT_CONTINUE;
  944. FRESULT res = f_read(&ftp_data.fp, filebuf, desiredsize, (UINT *)actualsize);
  945. if (res != FR_OK) {
  946. ftp_close_files();
  947. result = E_FTP_RESULT_FAILED;
  948. *actualsize = 0;
  949. }
  950. else if (*actualsize < desiredsize) {
  951. ftp_close_files();
  952. result = E_FTP_RESULT_OK;
  953. }
  954. return result;
  955. }
  956. static ftp_result_t ftp_write_file (char *filebuf, uint32_t size) {
  957. ftp_result_t result = E_FTP_RESULT_FAILED;
  958. uint32_t actualsize;
  959. FRESULT res = f_write(&ftp_data.fp, filebuf, size, (UINT *)&actualsize);
  960. if ((actualsize == size) && (FR_OK == res)) {
  961. result = E_FTP_RESULT_OK;
  962. }
  963. else {
  964. ftp_close_files();
  965. }
  966. return result;
  967. }
  968. static ftp_result_t ftp_open_dir_for_listing (const char *path) {
  969. // "hack" to detect the root directory
  970. if (path[0] == '/' && path[1] == '\0') {
  971. ftp_data.listroot = true;
  972. } else {
  973. FRESULT res;
  974. res = f_opendir_helper(&ftp_data.dp, path); /* Open the directory */
  975. if (res != FR_OK) {
  976. return E_FTP_RESULT_FAILED;
  977. }
  978. ftp_data.e_open = E_FTP_DIR_OPEN;
  979. ftp_data.listroot = false;
  980. }
  981. return E_FTP_RESULT_CONTINUE;
  982. }
  983. static ftp_result_t ftp_list_dir (char *list, uint32_t maxlistsize, uint32_t *listsize) {
  984. uint next = 0;
  985. uint listcount = 0;
  986. FRESULT res;
  987. ftp_result_t result = E_FTP_RESULT_CONTINUE;
  988. FILINFO fno;
  989. #if _USE_LFN
  990. // read up to 2 directory items
  991. while (listcount < 2) {
  992. #else
  993. // read up to 4 directory items
  994. while (listcount < 4) {
  995. #endif
  996. if (ftp_data.listroot) {
  997. // root directory "hack"
  998. mp_vfs_mount_t *vfs = MP_STATE_VM(vfs_mount_table);
  999. int i = ftp_data.volcount;
  1000. while (vfs != NULL && i != 0) {
  1001. vfs = vfs->next;
  1002. i -= 1;
  1003. }
  1004. if (vfs == NULL) {
  1005. if (!next) {
  1006. // no volume found this time, we are done
  1007. ftp_data.volcount = 0;
  1008. }
  1009. break;
  1010. } else {
  1011. next += ftp_print_eplf_drive((list + next), (maxlistsize - next), vfs->str + 1);
  1012. }
  1013. ftp_data.volcount++;
  1014. } else {
  1015. // a "normal" directory
  1016. res = f_readdir(&ftp_data.dp, &fno); /* Read a directory item */
  1017. if (res != FR_OK || fno.fname[0] == 0) {
  1018. result = E_FTP_RESULT_OK;
  1019. break; /* Break on error or end of dp */
  1020. }
  1021. if (fno.fname[0] == '.' && fno.fname[1] == 0) continue; /* Ignore . entry */
  1022. if (fno.fname[0] == '.' && fno.fname[1] == '.' && fno.fname[2] == 0) continue; /* Ignore .. entry */
  1023. // add the entry to the list
  1024. next += ftp_print_eplf_item((list + next), (maxlistsize - next), &fno);
  1025. }
  1026. listcount++;
  1027. }
  1028. if (result == E_FTP_RESULT_OK) {
  1029. ftp_close_files();
  1030. }
  1031. *listsize = next;
  1032. return result;
  1033. }
  1034. static void ftp_open_child (char *pwd, char *dir) {
  1035. if (dir[0] == '/') {
  1036. strcpy (pwd, dir);
  1037. } else {
  1038. if (strlen(pwd) > 1) {
  1039. strcat (pwd, "/");
  1040. }
  1041. strcat (pwd, dir);
  1042. }
  1043. uint len = strlen(pwd);
  1044. if ((len > 1) && (pwd[len - 1] == '/')) {
  1045. pwd[len - 1] = '\0';
  1046. }
  1047. }
  1048. static void ftp_close_child (char *pwd) {
  1049. uint len = strlen(pwd);
  1050. while (len && (pwd[len] != '/')) {
  1051. len--;
  1052. }
  1053. if (len == 0) {
  1054. strcpy (pwd, "/");
  1055. } else {
  1056. pwd[len] = '\0';
  1057. }
  1058. }
  1059. static void ftp_return_to_previous_path (char *pwd, char *dir) {
  1060. uint newlen = strlen(pwd) - strlen(dir);
  1061. if ((newlen > 2) && (pwd[newlen - 1] == '/')) {
  1062. pwd[newlen - 1] = '\0';
  1063. }
  1064. else {
  1065. if (newlen == 0) {
  1066. strcpy (pwd, "/");
  1067. } else {
  1068. pwd[newlen] = '\0';
  1069. }
  1070. }
  1071. }