trezor-firmware/core/embed/unix/profile.c

/*
 * This file is part of the TREZOR project, https://trezor.io/
 *
 * Copyright (c) SatoshiLabs
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#define _GNU_SOURCE
#include <libgen.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/errno.h>
#include <sys/stat.h>
#include <unistd.h>

#include "common.h"
#include "profile.h"

static int mkpath(const char *path, mode_t mode) {
  if (!path) {
    errno = EINVAL;
    return 1;
  }

  struct stat sb;
  if (!stat(strdup(path), &sb)) {
    return 0;
  }

  // dirname has multiple incompatible implementations.
  // Some modify input argument some share output buffer.
  char *pathdup = strdup(path);
  char *subpath = strdup(dirname(pathdup));
  mkpath(subpath, mode);
  free(pathdup);
  free(subpath);
  return mkdir(path, mode);
}

void profile_init(void) {
  const char *dir = profile_dir();
  if (mkpath(dir, 0755)) {
    perror(dir);
    printf("!!! Unable to initialize profile directory `%s`. Quitting\n", dir);
    exit(1);
  }
  printf("Profile directory: %s\n", dir);
}

const char *profile_dir(void) {
  static const char *_profile_dir;

  if (_profile_dir) {
    return _profile_dir;
  }

  char *trezor_profile = getenv("TREZOR_PROFILE");
  if (!trezor_profile || strlen(trezor_profile) < 1) {
    trezor_profile = PROFILE_DEFAULT;
  }

  char *path;
  if (trezor_profile[0] == '/') {
    // TREZOR_PROFILE is a full path to profile directory
    path = strdup(trezor_profile);
  } else {
    // TREZOR_PROFILE is just a profile name and will be put in ~/.trezoremu/
    int print_length = asprintf(&path, "%s/" PROFILE_HOMEDOT "/%s",
                                getenv("HOME"), trezor_profile);
    if (print_length == -1) {
      path = NULL;
    }
  }

  if (!path) {  // last resort fallback
    path = PROFILE_DEFAULT;
  }

  _profile_dir = path;

  return _profile_dir;
}

const char *profile_flash_path(void) {
  static char *_flash_path;
  if (_flash_path) {
    return _flash_path;
  }

  if (asprintf(&_flash_path, "%s/trezor.flash", profile_dir()) < 0) {
    _flash_path = NULL;
  }

  if (!_flash_path) {  // last resort fallback
    _flash_path = PROFILE_DEFAULT "/trezor.flash";
  }

  return _flash_path;
}

const char *profile_sdcard_path(void) {
  static char *_sdcard_path;
  if (_sdcard_path) {
    return _sdcard_path;
  }

  if (asprintf(&_sdcard_path, "%s/trezor.sdcard", profile_dir()) < 0) {
    _sdcard_path = NULL;
  }

  if (!_sdcard_path) {  // last resort fallback
    _sdcard_path = PROFILE_DEFAULT "/trezor.sdcard";
  }

  return _sdcard_path;
}