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