Momentum-Apps/evil_portal/helpers/evil_portal_storage.c

#include "evil_portal_storage.h"

static Storage *evil_portal_open_storage() {
  return furi_record_open(RECORD_STORAGE);
}

static void evil_portal_close_storage() { furi_record_close(RECORD_STORAGE); }

void evil_portal_read_index_html(void *context) {

  Evil_PortalApp *app = context;
  Storage *storage = evil_portal_open_storage();
  FileInfo fi;

  if(!storage_common_exists(storage, EVIL_PORTAL_INDEX_SAVE_PATH)) {
    FuriString* tmp = furi_string_alloc_set(EVIL_PORTAL_INDEX_DEFAULT_PATH);
    evil_portal_replace_index_html(tmp);
    furi_string_free(tmp);
  }

  if (storage_common_stat(storage, EVIL_PORTAL_INDEX_SAVE_PATH, &fi) ==
      FSE_OK) {
    File *index_html = storage_file_alloc(storage);
    if (storage_file_open(index_html, EVIL_PORTAL_INDEX_SAVE_PATH, FSAM_READ,
                          FSOM_OPEN_EXISTING)) {
      app->index_html = malloc((size_t)fi.size);
      uint8_t *buf_ptr = app->index_html;
      size_t read = 0;
      while (read < fi.size) {
        size_t to_read = fi.size - read;
        if (to_read > UINT16_MAX)
          to_read = UINT16_MAX;
        uint16_t now_read =
            storage_file_read(index_html, buf_ptr, (uint16_t)to_read);
        read += now_read;
        buf_ptr += now_read;
      }
      free(buf_ptr);
    }
    storage_file_close(index_html);
    storage_file_free(index_html);
  } else {
    char *html_error =
        "<b>Evil portal</b><br>Unable to read the html file.<br>"
        "Is the SD Card set up correctly? <br>See instructions @ "
        "github.com/bigbrodude6119/flipper-zero-evil-portal<br>"
        "Under the 'Install pre-built app on the flipper' section.";
    app->index_html = (uint8_t *)html_error;
  }

  evil_portal_close_storage();
}

void evil_portal_replace_index_html(FuriString* path) {
  Storage *storage = evil_portal_open_storage();
  FS_Error error;
  error = storage_common_remove(storage, EVIL_PORTAL_INDEX_SAVE_PATH);
  if(error != FSE_OK) {
    FURI_LOG_D("EVIL PORTAL", "Error removing file");
  } else {
    FURI_LOG_D("EVIL PORTAL", "Error removed file");
  }
  error = storage_common_copy(storage, furi_string_get_cstr(path), EVIL_PORTAL_INDEX_SAVE_PATH);
  if(error != FSE_OK) {
      FURI_LOG_D("EVIL PORTAL", "Error copying file");
  }
  evil_portal_close_storage();
}

void evil_portal_create_html_folder_if_not_exists() {
  Storage *storage = evil_portal_open_storage();
  if(storage_common_stat(storage, HTML_FOLDER, NULL) == FSE_NOT_EXIST) {
    FURI_LOG_D("Evil Portal", "Directory %s doesn't exist. Will create new.", HTML_FOLDER);
    if(!storage_simply_mkdir(storage, HTML_FOLDER)) {
      FURI_LOG_E("Evil Portal", "Error creating directory %s", HTML_FOLDER);
    }
  }
  evil_portal_close_storage();
}

void evil_portal_read_ap_name(void *context) {
  Evil_PortalApp *app = context;
  Storage *storage = evil_portal_open_storage();
  FileInfo fi;

  if (storage_common_stat(storage, EVIL_PORTAL_AP_SAVE_PATH, &fi) == FSE_OK) {
    File *ap_name = storage_file_alloc(storage);
    if (storage_file_open(ap_name, EVIL_PORTAL_AP_SAVE_PATH, FSAM_READ,
                          FSOM_OPEN_EXISTING)) {
      app->ap_name = malloc((size_t)fi.size);
      uint8_t *buf_ptr = app->ap_name;
      size_t read = 0;
      while (read < fi.size) {
        size_t to_read = fi.size - read;
        if (to_read > UINT16_MAX)
          to_read = UINT16_MAX;
        uint16_t now_read =
            storage_file_read(ap_name, buf_ptr, (uint16_t)to_read);
        read += now_read;
        buf_ptr += now_read;
      }
      free(buf_ptr);
    }
    storage_file_close(ap_name);
    storage_file_free(ap_name);
  } else {
    char *app_default = "Evil Portal";
    app->ap_name = (uint8_t *)app_default;
  }
  evil_portal_close_storage();
}

void evil_portal_write_ap_name(void *context) {
  Evil_PortalApp *app = context;
  Storage *storage = evil_portal_open_storage();

  File *ap_name = storage_file_alloc(storage);
  if (storage_file_open(ap_name, EVIL_PORTAL_AP_SAVE_PATH, FSAM_WRITE,
                          FSOM_CREATE_ALWAYS)) {
      storage_file_write(ap_name, app->text_store[0], strlen(app->text_store[0]));
  }
  storage_file_close(ap_name);
  storage_file_free(ap_name);
  evil_portal_close_storage();
}

char *sequential_file_resolve_path(Storage *storage, const char *dir,
                                   const char *prefix, const char *extension) {
  if (storage == NULL || dir == NULL || prefix == NULL || extension == NULL) {
    return NULL;
  }

  char file_path[256];
  int file_index = 0;

  do {
    if (snprintf(file_path, sizeof(file_path), "%s/%s_%d.%s", dir, prefix,
                 file_index, extension) < 0) {
      return NULL;
    }
    file_index++;
  } while (storage_file_exists(storage, file_path));

  return strdup(file_path);
}

void write_logs(FuriString *portal_logs) {
  Storage *storage = evil_portal_open_storage();

  if (!storage_file_exists(storage, EVIL_PORTAL_LOG_SAVE_PATH)) {
    storage_simply_mkdir(storage, EVIL_PORTAL_LOG_SAVE_PATH);
  }

  char *seq_file_path = sequential_file_resolve_path(
      storage, EVIL_PORTAL_LOG_SAVE_PATH, "log", "txt");

  File *file = storage_file_alloc(storage);

  if (storage_file_open(file, seq_file_path, FSAM_WRITE, FSOM_CREATE_ALWAYS)) {
    storage_file_write(file, furi_string_get_cstr(portal_logs), furi_string_utf8_length(portal_logs));
  }
  storage_file_close(file);
  storage_file_free(file);
  evil_portal_close_storage();
}