arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-10 23:40:58 +00:00
Code Issues Releases Wiki Activity

unix: use mmap to access flash file

Browse Source
This commit is contained in:
Pavol Rusnak 2017-12-08 14:18:15 +01:00
parent 44a11fa3ef
commit 7f5192701f
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 45 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
embed/unix/flash.c
View File

@ -6,8 +6,14 @@
*/ */
#include <string.h> #include <string.h>
#include <stdio.h> #include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <sys/types.h>
#include "common.h"
#include "../trezorhal/flash.h" #include "../trezorhal/flash.h"
#ifndef FLASH_FILE #ifndef FLASH_FILE
@ -16,6 +22,8 @@
#define SECTOR_COUNT 24 #define SECTOR_COUNT 24
#define FLASH_SIZE 0x200000
static const uint32_t sector_table[SECTOR_COUNT + 1] = { static const uint32_t sector_table[SECTOR_COUNT + 1] = {
[ 0] = 0x08000000, // - 0x08003FFF | 16 KiB [ 0] = 0x08000000, // - 0x08003FFF | 16 KiB
[ 1] = 0x08004000, // - 0x08007FFF | 16 KiB [ 1] = 0x08004000, // - 0x08007FFF | 16 KiB
@ -44,28 +52,49 @@ static const uint32_t sector_table[SECTOR_COUNT + 1] = {
[24] = 0x08200000, // last element - not a valid sector [24] = 0x08200000, // last element - not a valid sector
}; };
static uint8_t flash_buffer[0x200000]; static int flash_fd;
static uint8_t *flash_buffer;
static void flash_sync(void) static void flash_exit(void)
{ {
FILE *f = fopen(FLASH_FILE, "wb"); int r = munmap(flash_buffer, FLASH_SIZE);
if (f) { ensure(sectrue * (r == 0), "munmap failed");
fwrite(flash_buffer, sizeof(flash_buffer), 1, f); r = close(flash_fd);
fclose(f); ensure(sectrue * (r == 0), "close failed");
}
} }
secbool flash_init(void) secbool flash_init(void)
{ {
FILE *f = fopen(FLASH_FILE, "rb"); int r;
size_t r = 0;
if (f) { // check whether the file exists and it has the correct size
r = fread(flash_buffer, sizeof(flash_buffer), 1, f); struct stat sb;
fclose(f); r = stat(FLASH_FILE, &sb);
}
if (r != 1) { // (re)create if non existant or wrong size
memset(flash_buffer, 0xFF, sizeof(flash_buffer)); if (r != 0 || sb.st_size != FLASH_SIZE) {
int fd = open(FLASH_FILE, O_RDWR | O_CREAT | O_TRUNC, (mode_t)0600);
ensure(sectrue * (fd >= 0), "open failed");
for (int i = 0; i < FLASH_SIZE / 16; i++) {
ssize_t s = write(fd, "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF", 16);
ensure(sectrue * (s >= 0), "write failed");
}
r = close(fd);
ensure(sectrue * (r == 0), "close failed");
} }
// mmap file
int fd = open(FLASH_FILE, O_RDWR);
ensure(sectrue * (fd >= 0), "open failed");
void *map = mmap(0, FLASH_SIZE, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
ensure(sectrue * (map != MAP_FAILED), "mmap failed");
flash_fd = fd;
flash_buffer = (uint8_t *)map;
atexit(flash_exit);
return sectrue; return sectrue;
} }
@ -105,7 +134,6 @@ secbool flash_erase_sectors(const uint8_t *sectors, int len, void (*progress)(in
if (progress) { if (progress) {
progress(i + 1, len); progress(i + 1, len);
} }
flash_sync();
} }
return sectrue; return sectrue;
} }
@ -120,7 +148,6 @@ secbool flash_write_byte_rel(uint8_t sector, uint32_t offset, uint8_t data)
return secfalse; // we cannot change zeroes to ones return secfalse; // we cannot change zeroes to ones
} }
flash[0] = data; flash[0] = data;
flash_sync();
return sectrue; return sectrue;
} }
@ -137,7 +164,6 @@ secbool flash_write_word_rel(uint8_t sector, uint32_t offset, uint32_t data)
return secfalse; // we cannot change zeroes to ones return secfalse; // we cannot change zeroes to ones
} }
flash[0] = data; flash[0] = data;
flash_sync();
return sectrue; return sectrue;
} }