rpc_storage.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. #include "flipper.pb.h"
  2. #include <core/common_defines.h>
  3. #include <core/memmgr.h>
  4. #include <core/record.h>
  5. #include "pb_decode.h"
  6. #include "rpc/rpc.h"
  7. #include "rpc_i.h"
  8. #include "storage.pb.h"
  9. #include "storage/filesystem_api_defines.h"
  10. #include "storage/storage.h"
  11. #include <stdint.h>
  12. #include <lib/toolbox/md5.h>
  13. #include <lib/toolbox/path.h>
  14. #include <update_util/lfs_backup.h>
  15. #define TAG "RpcStorage"
  16. #define MAX_NAME_LENGTH 255
  17. static const size_t MAX_DATA_SIZE = 512;
  18. typedef enum {
  19. RpcStorageStateIdle = 0,
  20. RpcStorageStateWriting,
  21. } RpcStorageState;
  22. typedef struct {
  23. RpcSession* session;
  24. Storage* api;
  25. File* file;
  26. RpcStorageState state;
  27. uint32_t current_command_id;
  28. } RpcStorageSystem;
  29. static void rpc_system_storage_reset_state(
  30. RpcStorageSystem* rpc_storage,
  31. RpcSession* session,
  32. bool send_error) {
  33. furi_assert(rpc_storage);
  34. furi_assert(session);
  35. if(rpc_storage->state != RpcStorageStateIdle) {
  36. if(send_error) {
  37. rpc_send_and_release_empty(
  38. session,
  39. rpc_storage->current_command_id,
  40. PB_CommandStatus_ERROR_CONTINUOUS_COMMAND_INTERRUPTED);
  41. }
  42. if(rpc_storage->state == RpcStorageStateWriting) {
  43. storage_file_close(rpc_storage->file);
  44. storage_file_free(rpc_storage->file);
  45. furi_record_close(RECORD_STORAGE);
  46. }
  47. rpc_storage->state = RpcStorageStateIdle;
  48. }
  49. }
  50. PB_CommandStatus rpc_system_storage_get_error(FS_Error fs_error) {
  51. PB_CommandStatus pb_error;
  52. switch(fs_error) {
  53. case FSE_OK:
  54. pb_error = PB_CommandStatus_OK;
  55. break;
  56. case FSE_INVALID_NAME:
  57. pb_error = PB_CommandStatus_ERROR_STORAGE_INVALID_NAME;
  58. break;
  59. case FSE_INVALID_PARAMETER:
  60. pb_error = PB_CommandStatus_ERROR_STORAGE_INVALID_PARAMETER;
  61. break;
  62. case FSE_INTERNAL:
  63. pb_error = PB_CommandStatus_ERROR_STORAGE_INTERNAL;
  64. break;
  65. case FSE_ALREADY_OPEN:
  66. pb_error = PB_CommandStatus_ERROR_STORAGE_ALREADY_OPEN;
  67. break;
  68. case FSE_DENIED:
  69. pb_error = PB_CommandStatus_ERROR_STORAGE_DENIED;
  70. break;
  71. case FSE_EXIST:
  72. pb_error = PB_CommandStatus_ERROR_STORAGE_EXIST;
  73. break;
  74. case FSE_NOT_EXIST:
  75. pb_error = PB_CommandStatus_ERROR_STORAGE_NOT_EXIST;
  76. break;
  77. case FSE_NOT_READY:
  78. pb_error = PB_CommandStatus_ERROR_STORAGE_NOT_READY;
  79. break;
  80. case FSE_NOT_IMPLEMENTED:
  81. pb_error = PB_CommandStatus_ERROR_STORAGE_NOT_IMPLEMENTED;
  82. break;
  83. default:
  84. pb_error = PB_CommandStatus_ERROR;
  85. break;
  86. }
  87. return pb_error;
  88. }
  89. static PB_CommandStatus rpc_system_storage_get_file_error(File* file) {
  90. return rpc_system_storage_get_error(storage_file_get_error(file));
  91. }
  92. static void rpc_system_storage_info_process(const PB_Main* request, void* context) {
  93. furi_assert(request);
  94. furi_assert(context);
  95. furi_assert(request->which_content == PB_Main_storage_info_request_tag);
  96. FURI_LOG_D(TAG, "Info");
  97. RpcStorageSystem* rpc_storage = context;
  98. RpcSession* session = rpc_storage->session;
  99. furi_assert(session);
  100. rpc_system_storage_reset_state(rpc_storage, session, true);
  101. PB_Main* response = malloc(sizeof(PB_Main));
  102. response->command_id = request->command_id;
  103. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  104. FS_Error error = storage_common_fs_info(
  105. fs_api,
  106. request->content.storage_info_request.path,
  107. &response->content.storage_info_response.total_space,
  108. &response->content.storage_info_response.free_space);
  109. response->command_status = rpc_system_storage_get_error(error);
  110. if(error == FSE_OK) {
  111. response->which_content = PB_Main_storage_info_response_tag;
  112. } else {
  113. response->which_content = PB_Main_empty_tag;
  114. }
  115. rpc_send_and_release(session, response);
  116. free(response);
  117. furi_record_close(RECORD_STORAGE);
  118. }
  119. static void rpc_system_storage_stat_process(const PB_Main* request, void* context) {
  120. furi_assert(request);
  121. furi_assert(context);
  122. furi_assert(request->which_content == PB_Main_storage_stat_request_tag);
  123. FURI_LOG_D(TAG, "Stat");
  124. RpcStorageSystem* rpc_storage = context;
  125. RpcSession* session = rpc_storage->session;
  126. furi_assert(session);
  127. rpc_system_storage_reset_state(rpc_storage, session, true);
  128. PB_Main* response = malloc(sizeof(PB_Main));
  129. response->command_id = request->command_id;
  130. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  131. const char* path = request->content.storage_stat_request.path;
  132. FileInfo fileinfo;
  133. FS_Error error = storage_common_stat(fs_api, path, &fileinfo);
  134. response->command_status = rpc_system_storage_get_error(error);
  135. response->which_content = PB_Main_empty_tag;
  136. if(error == FSE_OK) {
  137. response->which_content = PB_Main_storage_stat_response_tag;
  138. response->content.storage_stat_response.has_file = true;
  139. response->content.storage_stat_response.file.type = (fileinfo.flags & FSF_DIRECTORY) ?
  140. PB_Storage_File_FileType_DIR :
  141. PB_Storage_File_FileType_FILE;
  142. response->content.storage_stat_response.file.size = fileinfo.size;
  143. }
  144. rpc_send_and_release(session, response);
  145. free(response);
  146. furi_record_close(RECORD_STORAGE);
  147. }
  148. static void rpc_system_storage_list_root(const PB_Main* request, void* context) {
  149. furi_assert(request);
  150. furi_assert(context);
  151. RpcStorageSystem* rpc_storage = context;
  152. RpcSession* session = rpc_storage->session;
  153. furi_assert(session);
  154. const char* hard_coded_dirs[] = {"any", "int", "ext"};
  155. PB_Main response = {
  156. .has_next = false,
  157. .command_id = request->command_id,
  158. .command_status = PB_CommandStatus_OK,
  159. .which_content = PB_Main_storage_list_response_tag,
  160. };
  161. furi_assert(COUNT_OF(hard_coded_dirs) < COUNT_OF(response.content.storage_list_response.file));
  162. for(uint32_t i = 0; i < COUNT_OF(hard_coded_dirs); ++i) {
  163. ++response.content.storage_list_response.file_count;
  164. response.content.storage_list_response.file[i].data = NULL;
  165. response.content.storage_list_response.file[i].size = 0;
  166. response.content.storage_list_response.file[i].type = PB_Storage_File_FileType_DIR;
  167. char* str = malloc(strlen(hard_coded_dirs[i]) + 1);
  168. strcpy(str, hard_coded_dirs[i]);
  169. response.content.storage_list_response.file[i].name = str;
  170. }
  171. rpc_send_and_release(session, &response);
  172. }
  173. static void rpc_system_storage_list_process(const PB_Main* request, void* context) {
  174. furi_assert(request);
  175. furi_assert(context);
  176. furi_assert(request->which_content == PB_Main_storage_list_request_tag);
  177. FURI_LOG_D(TAG, "List");
  178. RpcStorageSystem* rpc_storage = context;
  179. RpcSession* session = rpc_storage->session;
  180. furi_assert(session);
  181. rpc_system_storage_reset_state(rpc_storage, session, true);
  182. if(!strcmp(request->content.storage_list_request.path, "/")) {
  183. rpc_system_storage_list_root(request, context);
  184. return;
  185. }
  186. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  187. File* dir = storage_file_alloc(fs_api);
  188. PB_Main response = {
  189. .command_id = request->command_id,
  190. .has_next = false,
  191. .which_content = PB_Main_storage_list_response_tag,
  192. .command_status = PB_CommandStatus_OK,
  193. };
  194. PB_Storage_ListResponse* list = &response.content.storage_list_response;
  195. bool finish = false;
  196. int i = 0;
  197. if(!storage_dir_open(dir, request->content.storage_list_request.path)) {
  198. response.command_status = rpc_system_storage_get_file_error(dir);
  199. response.which_content = PB_Main_empty_tag;
  200. finish = true;
  201. }
  202. while(!finish) {
  203. FileInfo fileinfo;
  204. char* name = malloc(MAX_NAME_LENGTH + 1);
  205. if(storage_dir_read(dir, &fileinfo, name, MAX_NAME_LENGTH)) {
  206. if(path_contains_only_ascii(name)) {
  207. if(i == COUNT_OF(list->file)) {
  208. list->file_count = i;
  209. response.has_next = true;
  210. rpc_send_and_release(session, &response);
  211. i = 0;
  212. }
  213. list->file[i].type = (fileinfo.flags & FSF_DIRECTORY) ?
  214. PB_Storage_File_FileType_DIR :
  215. PB_Storage_File_FileType_FILE;
  216. list->file[i].size = fileinfo.size;
  217. list->file[i].data = NULL;
  218. list->file[i].name = name;
  219. ++i;
  220. } else {
  221. free(name);
  222. }
  223. } else {
  224. list->file_count = i;
  225. finish = true;
  226. free(name);
  227. }
  228. }
  229. response.has_next = false;
  230. rpc_send_and_release(session, &response);
  231. storage_dir_close(dir);
  232. storage_file_free(dir);
  233. furi_record_close(RECORD_STORAGE);
  234. }
  235. static void rpc_system_storage_read_process(const PB_Main* request, void* context) {
  236. furi_assert(request);
  237. furi_assert(context);
  238. furi_assert(request->which_content == PB_Main_storage_read_request_tag);
  239. FURI_LOG_D(TAG, "Read");
  240. RpcStorageSystem* rpc_storage = context;
  241. RpcSession* session = rpc_storage->session;
  242. furi_assert(session);
  243. rpc_system_storage_reset_state(rpc_storage, session, true);
  244. /* use same message memory to send reponse */
  245. PB_Main* response = malloc(sizeof(PB_Main));
  246. const char* path = request->content.storage_read_request.path;
  247. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  248. File* file = storage_file_alloc(fs_api);
  249. bool fs_operation_success = storage_file_open(file, path, FSAM_READ, FSOM_OPEN_EXISTING);
  250. if(fs_operation_success) {
  251. size_t size_left = storage_file_size(file);
  252. do {
  253. response->command_id = request->command_id;
  254. response->which_content = PB_Main_storage_read_response_tag;
  255. response->command_status = PB_CommandStatus_OK;
  256. size_t read_size = MIN(size_left, MAX_DATA_SIZE);
  257. if(read_size) {
  258. response->content.storage_read_response.has_file = true;
  259. response->content.storage_read_response.file.data =
  260. malloc(PB_BYTES_ARRAY_T_ALLOCSIZE(read_size));
  261. uint8_t* buffer = &response->content.storage_read_response.file.data->bytes[0];
  262. uint16_t* read_size_msg = &response->content.storage_read_response.file.data->size;
  263. *read_size_msg = storage_file_read(file, buffer, read_size);
  264. size_left -= *read_size_msg;
  265. fs_operation_success = (*read_size_msg == read_size);
  266. response->has_next = fs_operation_success && (size_left > 0);
  267. } else {
  268. response->content.storage_read_response.file.data =
  269. malloc(PB_BYTES_ARRAY_T_ALLOCSIZE(0));
  270. response->content.storage_read_response.file.data->size = 0;
  271. response->content.storage_read_response.has_file = true;
  272. response->has_next = false;
  273. fs_operation_success = true;
  274. }
  275. if(fs_operation_success) {
  276. rpc_send_and_release(session, response);
  277. }
  278. } while((size_left != 0) && fs_operation_success);
  279. }
  280. if(!fs_operation_success) {
  281. rpc_send_and_release_empty(
  282. session, request->command_id, rpc_system_storage_get_file_error(file));
  283. }
  284. free(response);
  285. storage_file_close(file);
  286. storage_file_free(file);
  287. furi_record_close(RECORD_STORAGE);
  288. }
  289. static void rpc_system_storage_write_process(const PB_Main* request, void* context) {
  290. furi_assert(request);
  291. furi_assert(context);
  292. furi_assert(request->which_content == PB_Main_storage_write_request_tag);
  293. FURI_LOG_D(TAG, "Write");
  294. RpcStorageSystem* rpc_storage = context;
  295. RpcSession* session = rpc_storage->session;
  296. furi_assert(session);
  297. bool fs_operation_success = true;
  298. if(!path_contains_only_ascii(request->content.storage_write_request.path)) {
  299. rpc_storage->current_command_id = request->command_id;
  300. rpc_send_and_release_empty(
  301. session, rpc_storage->current_command_id, PB_CommandStatus_ERROR_STORAGE_INVALID_NAME);
  302. rpc_system_storage_reset_state(rpc_storage, session, false);
  303. return;
  304. }
  305. if((request->command_id != rpc_storage->current_command_id) &&
  306. (rpc_storage->state == RpcStorageStateWriting)) {
  307. rpc_system_storage_reset_state(rpc_storage, session, true);
  308. }
  309. if(rpc_storage->state != RpcStorageStateWriting) {
  310. rpc_storage->api = furi_record_open(RECORD_STORAGE);
  311. rpc_storage->file = storage_file_alloc(rpc_storage->api);
  312. rpc_storage->current_command_id = request->command_id;
  313. rpc_storage->state = RpcStorageStateWriting;
  314. const char* path = request->content.storage_write_request.path;
  315. fs_operation_success =
  316. storage_file_open(rpc_storage->file, path, FSAM_WRITE, FSOM_CREATE_ALWAYS);
  317. }
  318. File* file = rpc_storage->file;
  319. bool send_response = false;
  320. if(fs_operation_success) {
  321. if(request->content.storage_write_request.has_file &&
  322. request->content.storage_write_request.file.data &&
  323. request->content.storage_write_request.file.data->size) {
  324. uint8_t* buffer = request->content.storage_write_request.file.data->bytes;
  325. size_t buffer_size = request->content.storage_write_request.file.data->size;
  326. uint16_t written_size = storage_file_write(file, buffer, buffer_size);
  327. fs_operation_success = (written_size == buffer_size);
  328. }
  329. send_response = !request->has_next;
  330. }
  331. PB_CommandStatus command_status = PB_CommandStatus_OK;
  332. if(!fs_operation_success) {
  333. send_response = true;
  334. command_status = rpc_system_storage_get_file_error(file);
  335. }
  336. if(send_response) {
  337. rpc_send_and_release_empty(session, rpc_storage->current_command_id, command_status);
  338. rpc_system_storage_reset_state(rpc_storage, session, false);
  339. }
  340. }
  341. static bool rpc_system_storage_is_dir_is_empty(Storage* fs_api, const char* path) {
  342. furi_assert(fs_api);
  343. furi_assert(path);
  344. FileInfo fileinfo;
  345. bool is_dir_is_empty = true;
  346. FS_Error error = storage_common_stat(fs_api, path, &fileinfo);
  347. if((error == FSE_OK) && (fileinfo.flags & FSF_DIRECTORY)) {
  348. File* dir = storage_file_alloc(fs_api);
  349. if(storage_dir_open(dir, path)) {
  350. char* name = malloc(MAX_NAME_LENGTH);
  351. while(storage_dir_read(dir, &fileinfo, name, MAX_NAME_LENGTH)) {
  352. if(path_contains_only_ascii(name)) {
  353. is_dir_is_empty = false;
  354. break;
  355. }
  356. }
  357. free(name);
  358. }
  359. storage_dir_close(dir);
  360. storage_file_free(dir);
  361. }
  362. return is_dir_is_empty;
  363. }
  364. static void rpc_system_storage_delete_process(const PB_Main* request, void* context) {
  365. furi_assert(request);
  366. furi_assert(request->which_content == PB_Main_storage_delete_request_tag);
  367. furi_assert(context);
  368. FURI_LOG_D(TAG, "Delete");
  369. RpcStorageSystem* rpc_storage = context;
  370. RpcSession* session = rpc_storage->session;
  371. furi_assert(session);
  372. PB_CommandStatus status = PB_CommandStatus_ERROR;
  373. rpc_system_storage_reset_state(rpc_storage, session, true);
  374. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  375. char* path = request->content.storage_delete_request.path;
  376. if(!path) {
  377. status = PB_CommandStatus_ERROR_INVALID_PARAMETERS;
  378. } else {
  379. FS_Error error_remove = storage_common_remove(fs_api, path);
  380. // FSE_DENIED is for empty directory, but not only for this
  381. // that's why we have to check it
  382. if((error_remove == FSE_DENIED) && !rpc_system_storage_is_dir_is_empty(fs_api, path)) {
  383. if(request->content.storage_delete_request.recursive) {
  384. bool deleted = storage_simply_remove_recursive(fs_api, path);
  385. status = deleted ? PB_CommandStatus_OK : PB_CommandStatus_ERROR;
  386. } else {
  387. status = PB_CommandStatus_ERROR_STORAGE_DIR_NOT_EMPTY;
  388. }
  389. } else if(error_remove == FSE_NOT_EXIST) {
  390. status = PB_CommandStatus_OK;
  391. } else {
  392. status = rpc_system_storage_get_error(error_remove);
  393. }
  394. }
  395. furi_record_close(RECORD_STORAGE);
  396. rpc_send_and_release_empty(session, request->command_id, status);
  397. }
  398. static void rpc_system_storage_mkdir_process(const PB_Main* request, void* context) {
  399. furi_assert(request);
  400. furi_assert(request->which_content == PB_Main_storage_mkdir_request_tag);
  401. furi_assert(context);
  402. FURI_LOG_D(TAG, "Mkdir");
  403. RpcStorageSystem* rpc_storage = context;
  404. RpcSession* session = rpc_storage->session;
  405. furi_assert(session);
  406. PB_CommandStatus status;
  407. rpc_system_storage_reset_state(rpc_storage, session, true);
  408. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  409. char* path = request->content.storage_mkdir_request.path;
  410. if(path) {
  411. if(path_contains_only_ascii(path)) {
  412. FS_Error error = storage_common_mkdir(fs_api, path);
  413. status = rpc_system_storage_get_error(error);
  414. } else {
  415. status = PB_CommandStatus_ERROR_STORAGE_INVALID_NAME;
  416. }
  417. } else {
  418. status = PB_CommandStatus_ERROR_INVALID_PARAMETERS;
  419. }
  420. furi_record_close(RECORD_STORAGE);
  421. rpc_send_and_release_empty(session, request->command_id, status);
  422. }
  423. static void rpc_system_storage_md5sum_process(const PB_Main* request, void* context) {
  424. furi_assert(request);
  425. furi_assert(request->which_content == PB_Main_storage_md5sum_request_tag);
  426. furi_assert(context);
  427. FURI_LOG_D(TAG, "Md5sum");
  428. RpcStorageSystem* rpc_storage = context;
  429. RpcSession* session = rpc_storage->session;
  430. furi_assert(session);
  431. rpc_system_storage_reset_state(rpc_storage, session, true);
  432. const char* filename = request->content.storage_md5sum_request.path;
  433. if(!filename) {
  434. rpc_send_and_release_empty(
  435. session, request->command_id, PB_CommandStatus_ERROR_INVALID_PARAMETERS);
  436. return;
  437. }
  438. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  439. File* file = storage_file_alloc(fs_api);
  440. if(storage_file_open(file, filename, FSAM_READ, FSOM_OPEN_EXISTING)) {
  441. const uint16_t size_to_read = 512;
  442. const uint8_t hash_size = 16;
  443. uint8_t* data = malloc(size_to_read);
  444. uint8_t* hash = malloc(sizeof(uint8_t) * hash_size);
  445. md5_context* md5_ctx = malloc(sizeof(md5_context));
  446. md5_starts(md5_ctx);
  447. while(true) {
  448. uint16_t read_size = storage_file_read(file, data, size_to_read);
  449. if(read_size == 0) break;
  450. md5_update(md5_ctx, data, read_size);
  451. }
  452. md5_finish(md5_ctx, hash);
  453. free(md5_ctx);
  454. PB_Main response = {
  455. .command_id = request->command_id,
  456. .command_status = PB_CommandStatus_OK,
  457. .which_content = PB_Main_storage_md5sum_response_tag,
  458. .has_next = false,
  459. };
  460. char* md5sum = response.content.storage_md5sum_response.md5sum;
  461. size_t md5sum_size = sizeof(response.content.storage_md5sum_response.md5sum);
  462. (void)md5sum_size;
  463. furi_assert(hash_size <= ((md5sum_size - 1) / 2));
  464. for(uint8_t i = 0; i < hash_size; i++) {
  465. md5sum += snprintf(md5sum, md5sum_size, "%02x", hash[i]);
  466. }
  467. free(hash);
  468. free(data);
  469. storage_file_close(file);
  470. rpc_send_and_release(session, &response);
  471. } else {
  472. rpc_send_and_release_empty(
  473. session, request->command_id, rpc_system_storage_get_file_error(file));
  474. }
  475. storage_file_free(file);
  476. furi_record_close(RECORD_STORAGE);
  477. }
  478. static void rpc_system_storage_rename_process(const PB_Main* request, void* context) {
  479. furi_assert(request);
  480. furi_assert(request->which_content == PB_Main_storage_rename_request_tag);
  481. furi_assert(context);
  482. FURI_LOG_D(TAG, "Rename");
  483. RpcStorageSystem* rpc_storage = context;
  484. RpcSession* session = rpc_storage->session;
  485. furi_assert(session);
  486. PB_CommandStatus status;
  487. rpc_system_storage_reset_state(rpc_storage, session, true);
  488. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  489. if(path_contains_only_ascii(request->content.storage_rename_request.new_path)) {
  490. FS_Error error = storage_common_rename(
  491. fs_api,
  492. request->content.storage_rename_request.old_path,
  493. request->content.storage_rename_request.new_path);
  494. status = rpc_system_storage_get_error(error);
  495. } else {
  496. status = PB_CommandStatus_ERROR_STORAGE_INVALID_NAME;
  497. }
  498. furi_record_close(RECORD_STORAGE);
  499. rpc_send_and_release_empty(session, request->command_id, status);
  500. }
  501. static void rpc_system_storage_backup_create_process(const PB_Main* request, void* context) {
  502. furi_assert(request);
  503. furi_assert(request->which_content == PB_Main_storage_backup_create_request_tag);
  504. furi_assert(context);
  505. FURI_LOG_D(TAG, "BackupCreate");
  506. RpcStorageSystem* rpc_storage = context;
  507. RpcSession* session = rpc_storage->session;
  508. furi_assert(session);
  509. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  510. bool backup_ok =
  511. lfs_backup_create(fs_api, request->content.storage_backup_create_request.archive_path);
  512. furi_record_close(RECORD_STORAGE);
  513. rpc_send_and_release_empty(
  514. session, request->command_id, backup_ok ? PB_CommandStatus_OK : PB_CommandStatus_ERROR);
  515. }
  516. static void rpc_system_storage_backup_restore_process(const PB_Main* request, void* context) {
  517. furi_assert(request);
  518. furi_assert(request->which_content == PB_Main_storage_backup_restore_request_tag);
  519. furi_assert(context);
  520. FURI_LOG_D(TAG, "BackupRestore");
  521. RpcStorageSystem* rpc_storage = context;
  522. RpcSession* session = rpc_storage->session;
  523. furi_assert(session);
  524. Storage* fs_api = furi_record_open(RECORD_STORAGE);
  525. bool backup_ok =
  526. lfs_backup_unpack(fs_api, request->content.storage_backup_restore_request.archive_path);
  527. furi_record_close(RECORD_STORAGE);
  528. rpc_send_and_release_empty(
  529. session, request->command_id, backup_ok ? PB_CommandStatus_OK : PB_CommandStatus_ERROR);
  530. }
  531. void* rpc_system_storage_alloc(RpcSession* session) {
  532. furi_assert(session);
  533. RpcStorageSystem* rpc_storage = malloc(sizeof(RpcStorageSystem));
  534. rpc_storage->api = furi_record_open(RECORD_STORAGE);
  535. rpc_storage->session = session;
  536. rpc_storage->state = RpcStorageStateIdle;
  537. RpcHandler rpc_handler = {
  538. .message_handler = NULL,
  539. .decode_submessage = NULL,
  540. .context = rpc_storage,
  541. };
  542. rpc_handler.message_handler = rpc_system_storage_info_process;
  543. rpc_add_handler(session, PB_Main_storage_info_request_tag, &rpc_handler);
  544. rpc_handler.message_handler = rpc_system_storage_stat_process;
  545. rpc_add_handler(session, PB_Main_storage_stat_request_tag, &rpc_handler);
  546. rpc_handler.message_handler = rpc_system_storage_list_process;
  547. rpc_add_handler(session, PB_Main_storage_list_request_tag, &rpc_handler);
  548. rpc_handler.message_handler = rpc_system_storage_read_process;
  549. rpc_add_handler(session, PB_Main_storage_read_request_tag, &rpc_handler);
  550. rpc_handler.message_handler = rpc_system_storage_write_process;
  551. rpc_add_handler(session, PB_Main_storage_write_request_tag, &rpc_handler);
  552. rpc_handler.message_handler = rpc_system_storage_delete_process;
  553. rpc_add_handler(session, PB_Main_storage_delete_request_tag, &rpc_handler);
  554. rpc_handler.message_handler = rpc_system_storage_mkdir_process;
  555. rpc_add_handler(session, PB_Main_storage_mkdir_request_tag, &rpc_handler);
  556. rpc_handler.message_handler = rpc_system_storage_md5sum_process;
  557. rpc_add_handler(session, PB_Main_storage_md5sum_request_tag, &rpc_handler);
  558. rpc_handler.message_handler = rpc_system_storage_rename_process;
  559. rpc_add_handler(session, PB_Main_storage_rename_request_tag, &rpc_handler);
  560. rpc_handler.message_handler = rpc_system_storage_backup_create_process;
  561. rpc_add_handler(session, PB_Main_storage_backup_create_request_tag, &rpc_handler);
  562. rpc_handler.message_handler = rpc_system_storage_backup_restore_process;
  563. rpc_add_handler(session, PB_Main_storage_backup_restore_request_tag, &rpc_handler);
  564. return rpc_storage;
  565. }
  566. void rpc_system_storage_free(void* context) {
  567. furi_assert(context);
  568. RpcStorageSystem* rpc_storage = context;
  569. RpcSession* session = rpc_storage->session;
  570. furi_assert(session);
  571. rpc_system_storage_reset_state(rpc_storage, session, false);
  572. free(rpc_storage);
  573. }