1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-16 03:18:09 +00:00
trezor-firmware/embed/unix/profile.c
2018-10-11 16:51:04 +02:00

129 lines
3.2 KiB
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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/errno.h>
#include <libgen.h>
#include <unistd.h>
#include "profile.h"
#include "common.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;
}