|
| 1 | +/* Test case for MmioUnsanitizedMemcpy.ql |
| 2 | + * Allowlisted MMIO/DMA register macros flowing into memcpy/memmove/strncpy size parameters. |
| 3 | + */ |
| 4 | + |
| 5 | +typedef unsigned int uint32_t; |
| 6 | + |
| 7 | +void *memcpy(void *dest, const void *src, unsigned long n); |
| 8 | +void *memmove(void *dest, const void *src, unsigned long n); |
| 9 | +char *strncpy(char *dest, const char *src, unsigned long n); |
| 10 | + |
| 11 | +#define READ_REG(addr) (*(volatile uint32_t *)(addr)) |
| 12 | +#define GET_MMIO(addr) (*(volatile uint32_t *)(addr)) |
| 13 | +#define REG_READ(addr) (*(volatile uint32_t *)(addr)) |
| 14 | +#define DMA_READ(addr) (*(volatile uint32_t *)(addr)) |
| 15 | +#define MAX_DMA_LEN 64 |
| 16 | + |
| 17 | +struct VolatileField { |
| 18 | + volatile uint32_t len; |
| 19 | +}; |
| 20 | + |
| 21 | +volatile uint32_t mmio_len_reg; |
| 22 | +struct VolatileField vf; |
| 23 | + |
| 24 | +static void bad_read_reg(char *dst, char *src) { |
| 25 | + uint32_t len = READ_REG(0x40001000); // $ Source |
| 26 | + memcpy(dst, src, len); // $ Alert |
| 27 | +} |
| 28 | + |
| 29 | +static void bad_get_mmio(char *dst, char *src) { |
| 30 | + uint32_t len = GET_MMIO(0x50000000); // $ Source |
| 31 | + memmove(dst, src, len); // $ Alert |
| 32 | +} |
| 33 | + |
| 34 | +static void bad_reg_read(char *dst, char *src) { |
| 35 | + uint32_t len = REG_READ(0x51000000); // $ Source |
| 36 | + strncpy(dst, src, len); // $ Alert |
| 37 | +} |
| 38 | + |
| 39 | +static void bad_dma_read(char *dst, char *src) { |
| 40 | + uint32_t len = DMA_READ(0x60000000); // $ Source |
| 41 | + memcpy(dst, src, len); // $ Alert |
| 42 | +} |
| 43 | + |
| 44 | +static void good_bounded(char *dst, char *src) { |
| 45 | + uint32_t len = READ_REG(0x40001000); |
| 46 | + if (len <= MAX_DMA_LEN) |
| 47 | + memcpy(dst, src, len); // GOOD |
| 48 | +} |
| 49 | + |
| 50 | +static void good_early_return(char *dst, char *src) { |
| 51 | + uint32_t len = DMA_READ(0x60000000); |
| 52 | + if (len > MAX_DMA_LEN) |
| 53 | + return; |
| 54 | + memcpy(dst, src, len); // GOOD |
| 55 | +} |
| 56 | + |
| 57 | +static void good_constant_size(char *dst, char *src) { |
| 58 | + uint32_t len = READ_REG(0x40001000); |
| 59 | + memcpy(dst, src, 32); // GOOD |
| 60 | +} |
| 61 | + |
| 62 | +static void negative_volatile_global(char *dst, char *src) { |
| 63 | + uint32_t len = mmio_len_reg; |
| 64 | + memcpy(dst, src, len); // GOOD |
| 65 | +} |
| 66 | + |
| 67 | +static void negative_volatile_field(char *dst, char *src) { |
| 68 | + uint32_t len = vf.len; |
| 69 | + memcpy(dst, src, len); // GOOD |
| 70 | +} |
| 71 | + |
| 72 | +static void negative_volatile_deref(char *dst, char *src) { |
| 73 | + volatile uint32_t *reg = (volatile uint32_t *)0x40001000; |
| 74 | + uint32_t len = *reg; |
| 75 | + memcpy(dst, src, len); // GOOD |
| 76 | +} |
| 77 | + |
| 78 | +static uint32_t GET_MMIO_fn(unsigned long addr); |
| 79 | + |
| 80 | +static void negative_get_mmio_function(char *dst, char *src) { |
| 81 | + uint32_t len = GET_MMIO_fn(0x50000000); |
| 82 | + memcpy(dst, src, len); // GOOD |
| 83 | +} |
0 commit comments