Angel Pons | 118a9c7 | 2020-04-02 23:48:34 +0200 | [diff] [blame] | 1 | /* SPDX-License-Identifier: GPL-2.0-only */ |
| 2 | /* This file is part of the coreboot project. */ |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 3 | |
| 4 | #include <stdint.h> |
Aaron Durbin | 4dd87fb | 2013-04-24 16:28:52 -0500 | [diff] [blame] | 5 | #include <bootstate.h> |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 6 | #include <cbmem.h> |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 7 | |
| 8 | typedef struct file { |
| 9 | uint32_t magic; |
| 10 | struct file *next; |
| 11 | char *filename; |
| 12 | char *data; |
| 13 | int offset; |
| 14 | int len; |
| 15 | } FILE; |
| 16 | |
| 17 | #define SEEK_SET 0 /* Seek from beginning of file. */ |
| 18 | |
| 19 | #define DIR_SEPARATOR '/' |
| 20 | #define IS_DIR_SEPARATOR(ch) ((ch) == DIR_SEPARATOR) |
| 21 | #define HAS_DRIVE_SPEC(f) (0) |
| 22 | |
| 23 | #define COVERAGE_SIZE (32*1024) |
| 24 | |
Jean Lucas | 9ab9c33 | 2016-01-30 01:20:54 -0500 | [diff] [blame] | 25 | #define COVERAGE_MAGIC 0x584d4153 |
Paul Menzel | c824c26 | 2015-10-05 20:02:09 +0200 | [diff] [blame] | 26 | |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 27 | static FILE *current_file = NULL; |
| 28 | static FILE *previous_file = NULL; |
| 29 | |
| 30 | static FILE *fopen(const char *path, const char *mode) |
| 31 | { |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 32 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 33 | printk(BIOS_DEBUG, "fopen %s with mode %s\n", |
| 34 | path, mode); |
| 35 | #endif |
| 36 | if (!current_file) { |
| 37 | current_file = cbmem_add(CBMEM_ID_COVERAGE, 32*1024); |
| 38 | } else { |
| 39 | previous_file = current_file; |
Lee Leahy | 7340217 | 2017-03-10 15:23:24 -0800 | [diff] [blame] | 40 | current_file = |
| 41 | (FILE *)(ALIGN(((unsigned long)previous_file->data |
| 42 | + previous_file->len), 16)); |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 43 | } |
| 44 | |
| 45 | // TODO check if we're at the end of the CBMEM region (ENOMEM) |
| 46 | if (current_file) { |
Paul Menzel | c824c26 | 2015-10-05 20:02:09 +0200 | [diff] [blame] | 47 | current_file->magic = COVERAGE_MAGIC; |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 48 | current_file->next = NULL; |
| 49 | if (previous_file) |
| 50 | previous_file->next = current_file; |
| 51 | current_file->filename = (char *)¤t_file[1]; |
| 52 | strcpy(current_file->filename, path); |
Lee Leahy | 7340217 | 2017-03-10 15:23:24 -0800 | [diff] [blame] | 53 | current_file->data = |
| 54 | (char *)ALIGN(((unsigned long)current_file->filename |
| 55 | + strlen(path) + 1), 16); |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 56 | current_file->offset = 0; |
| 57 | current_file->len = 0; |
| 58 | } |
| 59 | |
| 60 | return current_file; |
| 61 | } |
| 62 | |
| 63 | static int fclose(FILE *stream) |
| 64 | { |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 65 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 66 | printk(BIOS_DEBUG, "fclose %s\n", stream->filename); |
| 67 | #endif |
| 68 | return 0; |
| 69 | } |
| 70 | |
| 71 | static int fseek(FILE *stream, long offset, int whence) |
| 72 | { |
| 73 | /* fseek should only be called with offset==0 and whence==SEEK_SET |
| 74 | * to a freshly opened file. */ |
Lee Leahy | 38768c3 | 2017-03-09 14:07:18 -0800 | [diff] [blame] | 75 | gcc_assert(offset == 0 && whence == SEEK_SET); |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 76 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | 84463ef | 2013-04-05 13:49:55 -0700 | [diff] [blame] | 77 | printk(BIOS_DEBUG, "fseek %s offset=%ld whence=%d\n", |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 78 | stream->filename, offset, whence); |
| 79 | #endif |
| 80 | return 0; |
| 81 | } |
| 82 | |
| 83 | static long ftell(FILE *stream) |
| 84 | { |
| 85 | /* ftell should currently not be called */ |
| 86 | gcc_assert(0); |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 87 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 88 | printk(BIOS_DEBUG, "ftell %s\n", stream->filename); |
| 89 | #endif |
| 90 | return 0; |
| 91 | } |
| 92 | |
| 93 | static size_t fread(void *ptr, size_t size, size_t nmemb, FILE *stream) |
| 94 | { |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 95 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 96 | printk(BIOS_DEBUG, "fread: ptr=%p size=%zd nmemb=%zd FILE*=%p\n", |
| 97 | ptr, size, nmemb, stream); |
| 98 | #endif |
| 99 | return 0; |
| 100 | } |
| 101 | |
| 102 | static size_t fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream) |
| 103 | { |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 104 | #if CONFIG(DEBUG_COVERAGE) |
Lee Leahy | 36984d8 | 2017-03-10 17:56:44 -0800 | [diff] [blame] | 105 | printk(BIOS_DEBUG, "fwrite: %zd * %zd bytes to file %s\n", |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 106 | nmemb, size, stream->filename); |
| 107 | #endif |
| 108 | // TODO check if file is last opened file and fail otherwise. |
| 109 | |
| 110 | memcpy(stream->data + stream->offset, ptr, size * nmemb); |
| 111 | stream->len += (nmemb * size) - (stream->len - stream->offset); |
| 112 | stream->offset += nmemb * size; |
| 113 | return nmemb; |
| 114 | } |
| 115 | |
| 116 | static void setbuf(FILE *stream, char *buf) |
| 117 | { |
| 118 | gcc_assert(buf == 0); |
| 119 | } |
| 120 | |
Aaron Durbin | 4dd87fb | 2013-04-24 16:28:52 -0500 | [diff] [blame] | 121 | static void coverage_init(void *unused) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 122 | { |
| 123 | extern long __CTOR_LIST__; |
Lee Leahy | 35af5c4 | 2017-03-09 17:35:28 -0800 | [diff] [blame] | 124 | typedef void (*func_ptr)(void); |
Lee Leahy | b2d834a | 2017-03-08 16:52:22 -0800 | [diff] [blame] | 125 | func_ptr *ctor = (func_ptr *) &__CTOR_LIST__; |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 126 | if (ctor == NULL) |
| 127 | return; |
| 128 | |
Lee Leahy | 2f919ec | 2017-03-08 17:37:06 -0800 | [diff] [blame] | 129 | for (; *ctor != (func_ptr) 0; ctor++) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 130 | (*ctor)(); |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 131 | } |
| 132 | |
| 133 | void __gcov_flush(void); |
Aaron Durbin | 4dd87fb | 2013-04-24 16:28:52 -0500 | [diff] [blame] | 134 | static void coverage_exit(void *unused) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 135 | { |
Julius Werner | cd49cce | 2019-03-05 16:53:33 -0800 | [diff] [blame] | 136 | #if CONFIG(DEBUG_COVERAGE) |
Stefan Reinauer | d37ab45 | 2012-12-18 16:23:28 -0800 | [diff] [blame] | 137 | printk(BIOS_DEBUG, "Syncing coverage data.\n"); |
| 138 | #endif |
| 139 | __gcov_flush(); |
| 140 | } |
| 141 | |
Aaron Durbin | 9ef9d85 | 2015-03-16 17:30:09 -0500 | [diff] [blame] | 142 | BOOT_STATE_INIT_ENTRY(BS_PRE_DEVICE, BS_ON_ENTRY, coverage_init, NULL); |
| 143 | BOOT_STATE_INIT_ENTRY(BS_OS_RESUME, BS_ON_ENTRY, coverage_exit, NULL); |
| 144 | BOOT_STATE_INIT_ENTRY(BS_PAYLOAD_LOAD, BS_ON_EXIT, coverage_exit, NULL); |