rpc_storage.c 24 KB

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