/* SPDX-License-Identifier: GPL-2.0-only */ #include <stdint.h> #include <bootstate.h> #include <cbmem.h> typedef struct file { uint32_t magic; struct file *next; char *filename; char *data; int offset; int len; } FILE; #define SEEK_SET 0 /* Seek from beginning of file. */ #define DIR_SEPARATOR '/' #define IS_DIR_SEPARATOR(ch) ((ch) == DIR_SEPARATOR) #define HAS_DRIVE_SPEC(f) (0) #define COVERAGE_SIZE (32*1024) #define COVERAGE_MAGIC 0x584d4153 static FILE *current_file = NULL; static FILE *previous_file = NULL; static FILE *fopen(const char *path, const char *mode) { #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s %s with mode %s\n", __func__, path, mode); #endif if (!current_file) { current_file = cbmem_add(CBMEM_ID_COVERAGE, 32*1024); } else { previous_file = current_file; current_file = (FILE *)(ALIGN_UP(((unsigned long)previous_file->data + previous_file->len), 16)); } // TODO check if we're at the end of the CBMEM region (ENOMEM) if (current_file) { current_file->magic = COVERAGE_MAGIC; current_file->next = NULL; if (previous_file) previous_file->next = current_file; current_file->filename = (char *)¤t_file[1]; strcpy(current_file->filename, path); current_file->data = (char *)ALIGN_UP(((unsigned long)current_file->filename + strlen(path) + 1), 16); current_file->offset = 0; current_file->len = 0; } return current_file; } static int fclose(FILE *stream) { #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s %s\n", __func__, stream->filename); #endif return 0; } static int fseek(FILE *stream, long offset, int whence) { /* fseek should only be called with offset==0 and whence==SEEK_SET * to a freshly opened file. */ gcc_assert(offset == 0 && whence == SEEK_SET); #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s %s offset=%ld whence=%d\n", __func__, stream->filename, offset, whence); #endif return 0; } static long ftell(FILE *stream) { /* ftell should currently not be called */ BUG(); #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s %s\n", __func__, stream->filename); #endif return 0; } static size_t fread(void *ptr, size_t size, size_t nmemb, FILE *stream) { #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s: ptr=%p size=%zd nmemb=%zd FILE*=%p\n", __func__, ptr, size, nmemb, stream); #endif return 0; } static size_t fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream) { #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "%s: %zd * %zd bytes to file %s\n", __func__, nmemb, size, stream->filename); #endif // TODO check if file is last opened file and fail otherwise. memcpy(stream->data + stream->offset, ptr, size * nmemb); stream->len += (nmemb * size) - (stream->len - stream->offset); stream->offset += nmemb * size; return nmemb; } static void setbuf(FILE *stream, char *buf) { gcc_assert(buf == 0); } static void coverage_init(void *unused) { extern long __CTOR_LIST__; typedef void (*func_ptr)(void); func_ptr *ctor = (func_ptr *) &__CTOR_LIST__; if (ctor == NULL) return; for (; *ctor != (func_ptr) 0; ctor++) (*ctor)(); } void __gcov_flush(void); static void coverage_exit(void *unused) { #if CONFIG(DEBUG_COVERAGE) printk(BIOS_DEBUG, "Syncing coverage data.\n"); #endif __gcov_flush(); } BOOT_STATE_INIT_ENTRY(BS_PRE_DEVICE, BS_ON_ENTRY, coverage_init, NULL); BOOT_STATE_INIT_ENTRY(BS_OS_RESUME, BS_ON_ENTRY, coverage_exit, NULL); BOOT_STATE_INIT_ENTRY(BS_PAYLOAD_LOAD, BS_ON_EXIT, coverage_exit, NULL);