Home | History | Annotate | Line # | Download | only in kern
subr_asan.c revision 1.26.2.1
      1 /*	$NetBSD: subr_asan.c,v 1.26.2.1 2021/01/03 16:35:04 thorpej Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2018-2020 Maxime Villard, m00nbsd.net
      5  * All rights reserved.
      6  *
      7  * This code is part of the KASAN subsystem of the NetBSD kernel.
      8  *
      9  * Redistribution and use in source and binary forms, with or without
     10  * modification, are permitted provided that the following conditions
     11  * are met:
     12  * 1. Redistributions of source code must retain the above copyright
     13  *    notice, this list of conditions and the following disclaimer.
     14  * 2. Redistributions in binary form must reproduce the above copyright
     15  *    notice, this list of conditions and the following disclaimer in the
     16  *    documentation and/or other materials provided with the distribution.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     23  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     26  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     28  * SUCH DAMAGE.
     29  */
     30 
     31 #include <sys/cdefs.h>
     32 __KERNEL_RCSID(0, "$NetBSD: subr_asan.c,v 1.26.2.1 2021/01/03 16:35:04 thorpej Exp $");
     33 
     34 #include <sys/param.h>
     35 #include <sys/device.h>
     36 #include <sys/kernel.h>
     37 #include <sys/param.h>
     38 #include <sys/conf.h>
     39 #include <sys/systm.h>
     40 #include <sys/types.h>
     41 #include <sys/asan.h>
     42 
     43 #include <uvm/uvm_extern.h>
     44 
     45 #ifdef DDB
     46 #include <machine/db_machdep.h>
     47 #include <ddb/db_extern.h>
     48 #endif
     49 
     50 #ifdef KASAN_PANIC
     51 #define REPORT panic
     52 #else
     53 #define REPORT printf
     54 #endif
     55 
     56 /* ASAN constants. Part of the compiler ABI. */
     57 #define KASAN_SHADOW_SCALE_SIZE		(1UL << KASAN_SHADOW_SCALE_SHIFT)
     58 #define KASAN_SHADOW_MASK		(KASAN_SHADOW_SCALE_SIZE - 1)
     59 #define KASAN_ALLOCA_SCALE_SIZE		32
     60 
     61 /* The MD code. */
     62 #include <machine/asan.h>
     63 
     64 /* ASAN ABI version. */
     65 #if defined(__clang__) && (__clang_major__ - 0 >= 6)
     66 #define ASAN_ABI_VERSION	8
     67 #elif __GNUC_PREREQ__(7, 1) && !defined(__clang__)
     68 #define ASAN_ABI_VERSION	8
     69 #elif __GNUC_PREREQ__(6, 1) && !defined(__clang__)
     70 #define ASAN_ABI_VERSION	6
     71 #else
     72 #error "Unsupported compiler version"
     73 #endif
     74 
     75 #define __RET_ADDR	(unsigned long)__builtin_return_address(0)
     76 
     77 /* Global variable descriptor. Part of the compiler ABI.  */
     78 struct __asan_global_source_location {
     79 	const char *filename;
     80 	int line_no;
     81 	int column_no;
     82 };
     83 struct __asan_global {
     84 	const void *beg;		/* address of the global variable */
     85 	size_t size;			/* size of the global variable */
     86 	size_t size_with_redzone;	/* size with the redzone */
     87 	const void *name;		/* name of the variable */
     88 	const void *module_name;	/* name of the module where the var is declared */
     89 	unsigned long has_dynamic_init;	/* the var has dyn initializer (c++) */
     90 	struct __asan_global_source_location *location;
     91 #if ASAN_ABI_VERSION >= 7
     92 	uintptr_t odr_indicator;	/* the address of the ODR indicator symbol */
     93 #endif
     94 };
     95 
     96 static bool kasan_enabled __read_mostly = false;
     97 
     98 /* -------------------------------------------------------------------------- */
     99 
    100 void
    101 kasan_shadow_map(void *addr, size_t size)
    102 {
    103 	size_t sz, npages, i;
    104 	vaddr_t sva, eva;
    105 
    106 	KASSERT((vaddr_t)addr % KASAN_SHADOW_SCALE_SIZE == 0);
    107 
    108 	sz = roundup(size, KASAN_SHADOW_SCALE_SIZE) / KASAN_SHADOW_SCALE_SIZE;
    109 
    110 	sva = (vaddr_t)kasan_md_addr_to_shad(addr);
    111 	eva = (vaddr_t)kasan_md_addr_to_shad(addr) + sz;
    112 
    113 	sva = rounddown(sva, PAGE_SIZE);
    114 	eva = roundup(eva, PAGE_SIZE);
    115 
    116 	npages = (eva - sva) / PAGE_SIZE;
    117 
    118 	KASSERT(sva >= KASAN_MD_SHADOW_START && eva < KASAN_MD_SHADOW_END);
    119 
    120 	for (i = 0; i < npages; i++) {
    121 		kasan_md_shadow_map_page(sva + i * PAGE_SIZE);
    122 	}
    123 }
    124 
    125 static void
    126 kasan_ctors(void)
    127 {
    128 	extern Elf_Addr __CTOR_LIST__, __CTOR_END__;
    129 	size_t nentries, i;
    130 	Elf_Addr *ptr;
    131 
    132 	nentries = ((size_t)&__CTOR_END__ - (size_t)&__CTOR_LIST__) /
    133 	    sizeof(uintptr_t);
    134 
    135 	ptr = &__CTOR_LIST__;
    136 	for (i = 0; i < nentries; i++) {
    137 		void (*func)(void);
    138 
    139 		func = (void *)(*ptr);
    140 		(*func)();
    141 
    142 		ptr++;
    143 	}
    144 }
    145 
    146 void
    147 kasan_early_init(void *stack)
    148 {
    149 	kasan_md_early_init(stack);
    150 }
    151 
    152 void
    153 kasan_init(void)
    154 {
    155 	/* MD initialization. */
    156 	kasan_md_init();
    157 
    158 	/* Now officially enabled. */
    159 	kasan_enabled = true;
    160 
    161 	/* Call the ASAN constructors. */
    162 	kasan_ctors();
    163 }
    164 
    165 static inline const char *
    166 kasan_code_name(uint8_t code)
    167 {
    168 	switch (code) {
    169 	case KASAN_GENERIC_REDZONE:
    170 		return "GenericRedZone";
    171 	case KASAN_MALLOC_REDZONE:
    172 		return "MallocRedZone";
    173 	case KASAN_KMEM_REDZONE:
    174 		return "KmemRedZone";
    175 	case KASAN_POOL_REDZONE:
    176 		return "PoolRedZone";
    177 	case KASAN_POOL_FREED:
    178 		return "PoolUseAfterFree";
    179 	case 1 ... 7:
    180 		return "RedZonePartial";
    181 	case KASAN_STACK_LEFT:
    182 		return "StackLeft";
    183 	case KASAN_STACK_MID:
    184 		return "StackMiddle";
    185 	case KASAN_STACK_RIGHT:
    186 		return "StackRight";
    187 	case KASAN_USE_AFTER_RET:
    188 		return "UseAfterRet";
    189 	case KASAN_USE_AFTER_SCOPE:
    190 		return "UseAfterScope";
    191 	default:
    192 		return "Unknown";
    193 	}
    194 }
    195 
    196 static void
    197 kasan_report(unsigned long addr, size_t size, bool write, unsigned long pc,
    198     uint8_t code)
    199 {
    200 	REPORT("ASan: Unauthorized Access In %p: Addr %p [%zu byte%s, %s,"
    201 	    " %s]\n",
    202 	    (void *)pc, (void *)addr, size, (size > 1 ? "s" : ""),
    203 	    (write ? "write" : "read"), kasan_code_name(code));
    204 	kasan_md_unwind();
    205 }
    206 
    207 static __always_inline void
    208 kasan_shadow_1byte_markvalid(unsigned long addr)
    209 {
    210 	int8_t *byte = kasan_md_addr_to_shad((void *)addr);
    211 	int8_t last = (addr & KASAN_SHADOW_MASK) + 1;
    212 
    213 	*byte = last;
    214 }
    215 
    216 static __always_inline void
    217 kasan_shadow_Nbyte_markvalid(const void *addr, size_t size)
    218 {
    219 	size_t i;
    220 
    221 	for (i = 0; i < size; i++) {
    222 		kasan_shadow_1byte_markvalid((unsigned long)addr+i);
    223 	}
    224 }
    225 
    226 static __always_inline void
    227 kasan_shadow_Nbyte_fill(const void *addr, size_t size, uint8_t code)
    228 {
    229 	void *shad;
    230 
    231 	if (__predict_false(size == 0))
    232 		return;
    233 	if (__predict_false(kasan_md_unsupported((vaddr_t)addr)))
    234 		return;
    235 
    236 	KASSERT((vaddr_t)addr % KASAN_SHADOW_SCALE_SIZE == 0);
    237 	KASSERT(size % KASAN_SHADOW_SCALE_SIZE == 0);
    238 
    239 	shad = (void *)kasan_md_addr_to_shad(addr);
    240 	size = size >> KASAN_SHADOW_SCALE_SHIFT;
    241 
    242 	__builtin_memset(shad, code, size);
    243 }
    244 
    245 void
    246 kasan_add_redzone(size_t *size)
    247 {
    248 	*size = roundup(*size, KASAN_SHADOW_SCALE_SIZE);
    249 	*size += KASAN_SHADOW_SCALE_SIZE;
    250 }
    251 
    252 void
    253 kasan_softint(struct lwp *l)
    254 {
    255 	const void *stk = (const void *)uvm_lwp_getuarea(l);
    256 
    257 	kasan_shadow_Nbyte_fill(stk, USPACE, 0);
    258 }
    259 
    260 /*
    261  * In an area of size 'sz_with_redz', mark the 'size' first bytes as valid,
    262  * and the rest as invalid. There are generally two use cases:
    263  *
    264  *  o kasan_mark(addr, origsize, size, code), with origsize < size. This marks
    265  *    the redzone at the end of the buffer as invalid.
    266  *
    267  *  o kasan_mark(addr, size, size, 0). This marks the entire buffer as valid.
    268  */
    269 void
    270 kasan_mark(const void *addr, size_t size, size_t sz_with_redz, uint8_t code)
    271 {
    272 	size_t i, n, redz;
    273 	int8_t *shad;
    274 
    275 	KASSERT((vaddr_t)addr % KASAN_SHADOW_SCALE_SIZE == 0);
    276 	redz = sz_with_redz - roundup(size, KASAN_SHADOW_SCALE_SIZE);
    277 	KASSERT(redz % KASAN_SHADOW_SCALE_SIZE == 0);
    278 	shad = kasan_md_addr_to_shad(addr);
    279 
    280 	/* Chunks of 8 bytes, valid. */
    281 	n = size / KASAN_SHADOW_SCALE_SIZE;
    282 	for (i = 0; i < n; i++) {
    283 		*shad++ = 0;
    284 	}
    285 
    286 	/* Possibly one chunk, mid. */
    287 	if ((size & KASAN_SHADOW_MASK) != 0) {
    288 		*shad++ = (size & KASAN_SHADOW_MASK);
    289 	}
    290 
    291 	/* Chunks of 8 bytes, invalid. */
    292 	n = redz / KASAN_SHADOW_SCALE_SIZE;
    293 	for (i = 0; i < n; i++) {
    294 		*shad++ = code;
    295 	}
    296 }
    297 
    298 /* -------------------------------------------------------------------------- */
    299 
    300 #define ADDR_CROSSES_SCALE_BOUNDARY(addr, size) 		\
    301 	(addr >> KASAN_SHADOW_SCALE_SHIFT) !=			\
    302 	    ((addr + size - 1) >> KASAN_SHADOW_SCALE_SHIFT)
    303 
    304 static __always_inline bool
    305 kasan_shadow_1byte_isvalid(unsigned long addr, uint8_t *code)
    306 {
    307 	int8_t *byte = kasan_md_addr_to_shad((void *)addr);
    308 	int8_t last = (addr & KASAN_SHADOW_MASK) + 1;
    309 
    310 	if (__predict_true(*byte == 0 || last <= *byte)) {
    311 		return true;
    312 	}
    313 	*code = *byte;
    314 	return false;
    315 }
    316 
    317 static __always_inline bool
    318 kasan_shadow_2byte_isvalid(unsigned long addr, uint8_t *code)
    319 {
    320 	int8_t *byte, last;
    321 
    322 	if (ADDR_CROSSES_SCALE_BOUNDARY(addr, 2)) {
    323 		return (kasan_shadow_1byte_isvalid(addr, code) &&
    324 		    kasan_shadow_1byte_isvalid(addr+1, code));
    325 	}
    326 
    327 	byte = kasan_md_addr_to_shad((void *)addr);
    328 	last = ((addr + 1) & KASAN_SHADOW_MASK) + 1;
    329 
    330 	if (__predict_true(*byte == 0 || last <= *byte)) {
    331 		return true;
    332 	}
    333 	*code = *byte;
    334 	return false;
    335 }
    336 
    337 static __always_inline bool
    338 kasan_shadow_4byte_isvalid(unsigned long addr, uint8_t *code)
    339 {
    340 	int8_t *byte, last;
    341 
    342 	if (ADDR_CROSSES_SCALE_BOUNDARY(addr, 4)) {
    343 		return (kasan_shadow_2byte_isvalid(addr, code) &&
    344 		    kasan_shadow_2byte_isvalid(addr+2, code));
    345 	}
    346 
    347 	byte = kasan_md_addr_to_shad((void *)addr);
    348 	last = ((addr + 3) & KASAN_SHADOW_MASK) + 1;
    349 
    350 	if (__predict_true(*byte == 0 || last <= *byte)) {
    351 		return true;
    352 	}
    353 	*code = *byte;
    354 	return false;
    355 }
    356 
    357 static __always_inline bool
    358 kasan_shadow_8byte_isvalid(unsigned long addr, uint8_t *code)
    359 {
    360 	int8_t *byte, last;
    361 
    362 	if (ADDR_CROSSES_SCALE_BOUNDARY(addr, 8)) {
    363 		return (kasan_shadow_4byte_isvalid(addr, code) &&
    364 		    kasan_shadow_4byte_isvalid(addr+4, code));
    365 	}
    366 
    367 	byte = kasan_md_addr_to_shad((void *)addr);
    368 	last = ((addr + 7) & KASAN_SHADOW_MASK) + 1;
    369 
    370 	if (__predict_true(*byte == 0 || last <= *byte)) {
    371 		return true;
    372 	}
    373 	*code = *byte;
    374 	return false;
    375 }
    376 
    377 static __always_inline bool
    378 kasan_shadow_Nbyte_isvalid(unsigned long addr, size_t size, uint8_t *code)
    379 {
    380 	size_t i;
    381 
    382 	for (i = 0; i < size; i++) {
    383 		if (!kasan_shadow_1byte_isvalid(addr+i, code))
    384 			return false;
    385 	}
    386 
    387 	return true;
    388 }
    389 
    390 static __always_inline void
    391 kasan_shadow_check(unsigned long addr, size_t size, bool write,
    392     unsigned long retaddr)
    393 {
    394 	uint8_t code;
    395 	bool valid;
    396 
    397 	if (__predict_false(!kasan_enabled))
    398 		return;
    399 #ifdef DDB
    400 	if (__predict_false(db_recover != NULL))
    401 		return;
    402 #endif
    403 	if (__predict_false(size == 0))
    404 		return;
    405 	if (__predict_false(kasan_md_unsupported(addr)))
    406 		return;
    407 
    408 	if (__builtin_constant_p(size)) {
    409 		switch (size) {
    410 		case 1:
    411 			valid = kasan_shadow_1byte_isvalid(addr, &code);
    412 			break;
    413 		case 2:
    414 			valid = kasan_shadow_2byte_isvalid(addr, &code);
    415 			break;
    416 		case 4:
    417 			valid = kasan_shadow_4byte_isvalid(addr, &code);
    418 			break;
    419 		case 8:
    420 			valid = kasan_shadow_8byte_isvalid(addr, &code);
    421 			break;
    422 		default:
    423 			valid = kasan_shadow_Nbyte_isvalid(addr, size, &code);
    424 			break;
    425 		}
    426 	} else {
    427 		valid = kasan_shadow_Nbyte_isvalid(addr, size, &code);
    428 	}
    429 
    430 	if (__predict_false(!valid)) {
    431 		kasan_report(addr, size, write, retaddr, code);
    432 	}
    433 }
    434 
    435 /* -------------------------------------------------------------------------- */
    436 
    437 void *
    438 kasan_memcpy(void *dst, const void *src, size_t len)
    439 {
    440 	kasan_shadow_check((unsigned long)src, len, false, __RET_ADDR);
    441 	kasan_shadow_check((unsigned long)dst, len, true, __RET_ADDR);
    442 	return __builtin_memcpy(dst, src, len);
    443 }
    444 
    445 int
    446 kasan_memcmp(const void *b1, const void *b2, size_t len)
    447 {
    448 	kasan_shadow_check((unsigned long)b1, len, false, __RET_ADDR);
    449 	kasan_shadow_check((unsigned long)b2, len, false, __RET_ADDR);
    450 	return __builtin_memcmp(b1, b2, len);
    451 }
    452 
    453 void *
    454 kasan_memset(void *b, int c, size_t len)
    455 {
    456 	kasan_shadow_check((unsigned long)b, len, true, __RET_ADDR);
    457 	return __builtin_memset(b, c, len);
    458 }
    459 
    460 void *
    461 kasan_memmove(void *dst, const void *src, size_t len)
    462 {
    463 	kasan_shadow_check((unsigned long)src, len, false, __RET_ADDR);
    464 	kasan_shadow_check((unsigned long)dst, len, true, __RET_ADDR);
    465 	return __builtin_memmove(dst, src, len);
    466 }
    467 
    468 char *
    469 kasan_strcpy(char *dst, const char *src)
    470 {
    471 	char *save = dst;
    472 
    473 	while (1) {
    474 		kasan_shadow_check((unsigned long)src, 1, false, __RET_ADDR);
    475 		kasan_shadow_check((unsigned long)dst, 1, true, __RET_ADDR);
    476 		*dst = *src;
    477 		if (*src == '\0')
    478 			break;
    479 		src++, dst++;
    480 	}
    481 
    482 	return save;
    483 }
    484 
    485 int
    486 kasan_strcmp(const char *s1, const char *s2)
    487 {
    488 	while (1) {
    489 		kasan_shadow_check((unsigned long)s1, 1, false, __RET_ADDR);
    490 		kasan_shadow_check((unsigned long)s2, 1, false, __RET_ADDR);
    491 		if (*s1 != *s2)
    492 			break;
    493 		if (*s1 == '\0')
    494 			return 0;
    495 		s1++, s2++;
    496 	}
    497 
    498 	return (*(const unsigned char *)s1 - *(const unsigned char *)s2);
    499 }
    500 
    501 size_t
    502 kasan_strlen(const char *str)
    503 {
    504 	const char *s;
    505 
    506 	s = str;
    507 	while (1) {
    508 		kasan_shadow_check((unsigned long)s, 1, false, __RET_ADDR);
    509 		if (*s == '\0')
    510 			break;
    511 		s++;
    512 	}
    513 
    514 	return (s - str);
    515 }
    516 
    517 char *
    518 kasan_strcat(char *dst, const char *src)
    519 {
    520 	size_t ldst, lsrc;
    521 
    522 	ldst = __builtin_strlen(dst);
    523 	lsrc = __builtin_strlen(src);
    524 	kasan_shadow_check((unsigned long)dst, ldst + lsrc + 1, true,
    525 	    __RET_ADDR);
    526 	kasan_shadow_check((unsigned long)src, lsrc + 1, false,
    527 	    __RET_ADDR);
    528 
    529 	return __builtin_strcat(dst, src);
    530 }
    531 
    532 char *
    533 kasan_strchr(const char *s, int c)
    534 {
    535 	kasan_shadow_check((unsigned long)s, __builtin_strlen(s) + 1, false,
    536 	    __RET_ADDR);
    537 	return __builtin_strchr(s, c);
    538 }
    539 
    540 char *
    541 kasan_strrchr(const char *s, int c)
    542 {
    543 	kasan_shadow_check((unsigned long)s, __builtin_strlen(s) + 1, false,
    544 	    __RET_ADDR);
    545 	return __builtin_strrchr(s, c);
    546 }
    547 
    548 #undef kcopy
    549 #undef copyinstr
    550 #undef copyoutstr
    551 #undef copyin
    552 
    553 int	kasan_kcopy(const void *, void *, size_t);
    554 int	kasan_copyinstr(const void *, void *, size_t, size_t *);
    555 int	kasan_copyoutstr(const void *, void *, size_t, size_t *);
    556 int	kasan_copyin(const void *, void *, size_t);
    557 int	kcopy(const void *, void *, size_t);
    558 int	copyinstr(const void *, void *, size_t, size_t *);
    559 int	copyoutstr(const void *, void *, size_t, size_t *);
    560 int	copyin(const void *, void *, size_t);
    561 
    562 int
    563 kasan_kcopy(const void *src, void *dst, size_t len)
    564 {
    565 	kasan_shadow_check((unsigned long)src, len, false, __RET_ADDR);
    566 	kasan_shadow_check((unsigned long)dst, len, true, __RET_ADDR);
    567 	return kcopy(src, dst, len);
    568 }
    569 
    570 int
    571 kasan_copyin(const void *uaddr, void *kaddr, size_t len)
    572 {
    573 	kasan_shadow_check((unsigned long)kaddr, len, true, __RET_ADDR);
    574 	return copyin(uaddr, kaddr, len);
    575 }
    576 
    577 int
    578 kasan_copyinstr(const void *uaddr, void *kaddr, size_t len, size_t *done)
    579 {
    580 	kasan_shadow_check((unsigned long)kaddr, len, true, __RET_ADDR);
    581 	return copyinstr(uaddr, kaddr, len, done);
    582 }
    583 
    584 int
    585 kasan_copyoutstr(const void *kaddr, void *uaddr, size_t len, size_t *done)
    586 {
    587 	kasan_shadow_check((unsigned long)kaddr, len, false, __RET_ADDR);
    588 	return copyoutstr(kaddr, uaddr, len, done);
    589 }
    590 
    591 /* -------------------------------------------------------------------------- */
    592 
    593 #undef _ucas_32
    594 #undef _ucas_32_mp
    595 #undef _ucas_64
    596 #undef _ucas_64_mp
    597 #undef _ufetch_8
    598 #undef _ufetch_16
    599 #undef _ufetch_32
    600 #undef _ufetch_64
    601 
    602 int _ucas_32(volatile uint32_t *, uint32_t, uint32_t, uint32_t *);
    603 int kasan__ucas_32(volatile uint32_t *, uint32_t, uint32_t, uint32_t *);
    604 int
    605 kasan__ucas_32(volatile uint32_t *uaddr, uint32_t old, uint32_t new,
    606     uint32_t *ret)
    607 {
    608 	kasan_shadow_check((unsigned long)ret, sizeof(*ret), true,
    609 	    __RET_ADDR);
    610 	return _ucas_32(uaddr, old, new, ret);
    611 }
    612 
    613 #ifdef __HAVE_UCAS_MP
    614 int _ucas_32_mp(volatile uint32_t *, uint32_t, uint32_t, uint32_t *);
    615 int kasan__ucas_32_mp(volatile uint32_t *, uint32_t, uint32_t, uint32_t *);
    616 int
    617 kasan__ucas_32_mp(volatile uint32_t *uaddr, uint32_t old, uint32_t new,
    618     uint32_t *ret)
    619 {
    620 	kasan_shadow_check((unsigned long)ret, sizeof(*ret), true,
    621 	    __RET_ADDR);
    622 	return _ucas_32_mp(uaddr, old, new, ret);
    623 }
    624 #endif
    625 
    626 #ifdef _LP64
    627 int _ucas_64(volatile uint64_t *, uint64_t, uint64_t, uint64_t *);
    628 int kasan__ucas_64(volatile uint64_t *, uint64_t, uint64_t, uint64_t *);
    629 int
    630 kasan__ucas_64(volatile uint64_t *uaddr, uint64_t old, uint64_t new,
    631     uint64_t *ret)
    632 {
    633 	kasan_shadow_check((unsigned long)ret, sizeof(*ret), true,
    634 	    __RET_ADDR);
    635 	return _ucas_64(uaddr, old, new, ret);
    636 }
    637 
    638 #ifdef __HAVE_UCAS_MP
    639 int _ucas_64_mp(volatile uint64_t *, uint64_t, uint64_t, uint64_t *);
    640 int kasan__ucas_64_mp(volatile uint64_t *, uint64_t, uint64_t, uint64_t *);
    641 int
    642 kasan__ucas_64_mp(volatile uint64_t *uaddr, uint64_t old, uint64_t new,
    643     uint64_t *ret)
    644 {
    645 	kasan_shadow_check((unsigned long)ret, sizeof(*ret), true,
    646 	    __RET_ADDR);
    647 	return _ucas_64_mp(uaddr, old, new, ret);
    648 }
    649 #endif
    650 #endif
    651 
    652 int _ufetch_8(const uint8_t *, uint8_t *);
    653 int kasan__ufetch_8(const uint8_t *, uint8_t *);
    654 int
    655 kasan__ufetch_8(const uint8_t *uaddr, uint8_t *valp)
    656 {
    657 	kasan_shadow_check((unsigned long)valp, sizeof(*valp), true,
    658 	    __RET_ADDR);
    659 	return _ufetch_8(uaddr, valp);
    660 }
    661 
    662 int _ufetch_16(const uint16_t *, uint16_t *);
    663 int kasan__ufetch_16(const uint16_t *, uint16_t *);
    664 int
    665 kasan__ufetch_16(const uint16_t *uaddr, uint16_t *valp)
    666 {
    667 	kasan_shadow_check((unsigned long)valp, sizeof(*valp), true,
    668 	    __RET_ADDR);
    669 	return _ufetch_16(uaddr, valp);
    670 }
    671 
    672 int _ufetch_32(const uint32_t *, uint32_t *);
    673 int kasan__ufetch_32(const uint32_t *, uint32_t *);
    674 int
    675 kasan__ufetch_32(const uint32_t *uaddr, uint32_t *valp)
    676 {
    677 	kasan_shadow_check((unsigned long)valp, sizeof(*valp), true,
    678 	    __RET_ADDR);
    679 	return _ufetch_32(uaddr, valp);
    680 }
    681 
    682 #ifdef _LP64
    683 int _ufetch_64(const uint64_t *, uint64_t *);
    684 int kasan__ufetch_64(const uint64_t *, uint64_t *);
    685 int
    686 kasan__ufetch_64(const uint64_t *uaddr, uint64_t *valp)
    687 {
    688 	kasan_shadow_check((unsigned long)valp, sizeof(*valp), true,
    689 	    __RET_ADDR);
    690 	return _ufetch_64(uaddr, valp);
    691 }
    692 #endif
    693 
    694 /* -------------------------------------------------------------------------- */
    695 
    696 #undef atomic_add_32
    697 #undef atomic_add_int
    698 #undef atomic_add_long
    699 #undef atomic_add_ptr
    700 #undef atomic_add_64
    701 #undef atomic_add_32_nv
    702 #undef atomic_add_int_nv
    703 #undef atomic_add_long_nv
    704 #undef atomic_add_ptr_nv
    705 #undef atomic_add_64_nv
    706 #undef atomic_and_32
    707 #undef atomic_and_uint
    708 #undef atomic_and_ulong
    709 #undef atomic_and_64
    710 #undef atomic_and_32_nv
    711 #undef atomic_and_uint_nv
    712 #undef atomic_and_ulong_nv
    713 #undef atomic_and_64_nv
    714 #undef atomic_or_32
    715 #undef atomic_or_uint
    716 #undef atomic_or_ulong
    717 #undef atomic_or_64
    718 #undef atomic_or_32_nv
    719 #undef atomic_or_uint_nv
    720 #undef atomic_or_ulong_nv
    721 #undef atomic_or_64_nv
    722 #undef atomic_cas_32
    723 #undef atomic_cas_uint
    724 #undef atomic_cas_ulong
    725 #undef atomic_cas_ptr
    726 #undef atomic_cas_64
    727 #undef atomic_cas_32_ni
    728 #undef atomic_cas_uint_ni
    729 #undef atomic_cas_ulong_ni
    730 #undef atomic_cas_ptr_ni
    731 #undef atomic_cas_64_ni
    732 #undef atomic_swap_32
    733 #undef atomic_swap_uint
    734 #undef atomic_swap_ulong
    735 #undef atomic_swap_ptr
    736 #undef atomic_swap_64
    737 #undef atomic_dec_32
    738 #undef atomic_dec_uint
    739 #undef atomic_dec_ulong
    740 #undef atomic_dec_ptr
    741 #undef atomic_dec_64
    742 #undef atomic_dec_32_nv
    743 #undef atomic_dec_uint_nv
    744 #undef atomic_dec_ulong_nv
    745 #undef atomic_dec_ptr_nv
    746 #undef atomic_dec_64_nv
    747 #undef atomic_inc_32
    748 #undef atomic_inc_uint
    749 #undef atomic_inc_ulong
    750 #undef atomic_inc_ptr
    751 #undef atomic_inc_64
    752 #undef atomic_inc_32_nv
    753 #undef atomic_inc_uint_nv
    754 #undef atomic_inc_ulong_nv
    755 #undef atomic_inc_ptr_nv
    756 #undef atomic_inc_64_nv
    757 
    758 #define ASAN_ATOMIC_FUNC_ADD(name, tret, targ1, targ2) \
    759 	void atomic_add_##name(volatile targ1 *, targ2); \
    760 	void kasan_atomic_add_##name(volatile targ1 *, targ2); \
    761 	void kasan_atomic_add_##name(volatile targ1 *ptr, targ2 val) \
    762 	{ \
    763 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    764 		    __RET_ADDR); \
    765 		atomic_add_##name(ptr, val); \
    766 	} \
    767 	tret atomic_add_##name##_nv(volatile targ1 *, targ2); \
    768 	tret kasan_atomic_add_##name##_nv(volatile targ1 *, targ2); \
    769 	tret kasan_atomic_add_##name##_nv(volatile targ1 *ptr, targ2 val) \
    770 	{ \
    771 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    772 		    __RET_ADDR); \
    773 		return atomic_add_##name##_nv(ptr, val); \
    774 	}
    775 
    776 #define ASAN_ATOMIC_FUNC_AND(name, tret, targ1, targ2) \
    777 	void atomic_and_##name(volatile targ1 *, targ2); \
    778 	void kasan_atomic_and_##name(volatile targ1 *, targ2); \
    779 	void kasan_atomic_and_##name(volatile targ1 *ptr, targ2 val) \
    780 	{ \
    781 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    782 		    __RET_ADDR); \
    783 		atomic_and_##name(ptr, val); \
    784 	} \
    785 	tret atomic_and_##name##_nv(volatile targ1 *, targ2); \
    786 	tret kasan_atomic_and_##name##_nv(volatile targ1 *, targ2); \
    787 	tret kasan_atomic_and_##name##_nv(volatile targ1 *ptr, targ2 val) \
    788 	{ \
    789 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    790 		    __RET_ADDR); \
    791 		return atomic_and_##name##_nv(ptr, val); \
    792 	}
    793 
    794 #define ASAN_ATOMIC_FUNC_OR(name, tret, targ1, targ2) \
    795 	void atomic_or_##name(volatile targ1 *, targ2); \
    796 	void kasan_atomic_or_##name(volatile targ1 *, targ2); \
    797 	void kasan_atomic_or_##name(volatile targ1 *ptr, targ2 val) \
    798 	{ \
    799 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    800 		    __RET_ADDR); \
    801 		atomic_or_##name(ptr, val); \
    802 	} \
    803 	tret atomic_or_##name##_nv(volatile targ1 *, targ2); \
    804 	tret kasan_atomic_or_##name##_nv(volatile targ1 *, targ2); \
    805 	tret kasan_atomic_or_##name##_nv(volatile targ1 *ptr, targ2 val) \
    806 	{ \
    807 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    808 		    __RET_ADDR); \
    809 		return atomic_or_##name##_nv(ptr, val); \
    810 	}
    811 
    812 #define ASAN_ATOMIC_FUNC_CAS(name, tret, targ1, targ2) \
    813 	tret atomic_cas_##name(volatile targ1 *, targ2, targ2); \
    814 	tret kasan_atomic_cas_##name(volatile targ1 *, targ2, targ2); \
    815 	tret kasan_atomic_cas_##name(volatile targ1 *ptr, targ2 exp, targ2 new) \
    816 	{ \
    817 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    818 		    __RET_ADDR); \
    819 		return atomic_cas_##name(ptr, exp, new); \
    820 	} \
    821 	tret atomic_cas_##name##_ni(volatile targ1 *, targ2, targ2); \
    822 	tret kasan_atomic_cas_##name##_ni(volatile targ1 *, targ2, targ2); \
    823 	tret kasan_atomic_cas_##name##_ni(volatile targ1 *ptr, targ2 exp, targ2 new) \
    824 	{ \
    825 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    826 		    __RET_ADDR); \
    827 		return atomic_cas_##name##_ni(ptr, exp, new); \
    828 	}
    829 
    830 #define ASAN_ATOMIC_FUNC_SWAP(name, tret, targ1, targ2) \
    831 	tret atomic_swap_##name(volatile targ1 *, targ2); \
    832 	tret kasan_atomic_swap_##name(volatile targ1 *, targ2); \
    833 	tret kasan_atomic_swap_##name(volatile targ1 *ptr, targ2 val) \
    834 	{ \
    835 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    836 		    __RET_ADDR); \
    837 		return atomic_swap_##name(ptr, val); \
    838 	}
    839 
    840 #define ASAN_ATOMIC_FUNC_DEC(name, tret, targ1) \
    841 	void atomic_dec_##name(volatile targ1 *); \
    842 	void kasan_atomic_dec_##name(volatile targ1 *); \
    843 	void kasan_atomic_dec_##name(volatile targ1 *ptr) \
    844 	{ \
    845 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    846 		    __RET_ADDR); \
    847 		atomic_dec_##name(ptr); \
    848 	} \
    849 	tret atomic_dec_##name##_nv(volatile targ1 *); \
    850 	tret kasan_atomic_dec_##name##_nv(volatile targ1 *); \
    851 	tret kasan_atomic_dec_##name##_nv(volatile targ1 *ptr) \
    852 	{ \
    853 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    854 		    __RET_ADDR); \
    855 		return atomic_dec_##name##_nv(ptr); \
    856 	}
    857 
    858 #define ASAN_ATOMIC_FUNC_INC(name, tret, targ1) \
    859 	void atomic_inc_##name(volatile targ1 *); \
    860 	void kasan_atomic_inc_##name(volatile targ1 *); \
    861 	void kasan_atomic_inc_##name(volatile targ1 *ptr) \
    862 	{ \
    863 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    864 		    __RET_ADDR); \
    865 		atomic_inc_##name(ptr); \
    866 	} \
    867 	tret atomic_inc_##name##_nv(volatile targ1 *); \
    868 	tret kasan_atomic_inc_##name##_nv(volatile targ1 *); \
    869 	tret kasan_atomic_inc_##name##_nv(volatile targ1 *ptr) \
    870 	{ \
    871 		kasan_shadow_check((uintptr_t)ptr, sizeof(tret), true, \
    872 		    __RET_ADDR); \
    873 		return atomic_inc_##name##_nv(ptr); \
    874 	}
    875 
    876 ASAN_ATOMIC_FUNC_ADD(32, uint32_t, uint32_t, int32_t);
    877 ASAN_ATOMIC_FUNC_ADD(64, uint64_t, uint64_t, int64_t);
    878 ASAN_ATOMIC_FUNC_ADD(int, unsigned int, unsigned int, int);
    879 ASAN_ATOMIC_FUNC_ADD(long, unsigned long, unsigned long, long);
    880 ASAN_ATOMIC_FUNC_ADD(ptr, void *, void, ssize_t);
    881 
    882 ASAN_ATOMIC_FUNC_AND(32, uint32_t, uint32_t, uint32_t);
    883 ASAN_ATOMIC_FUNC_AND(64, uint64_t, uint64_t, uint64_t);
    884 ASAN_ATOMIC_FUNC_AND(uint, unsigned int, unsigned int, unsigned int);
    885 ASAN_ATOMIC_FUNC_AND(ulong, unsigned long, unsigned long, unsigned long);
    886 
    887 ASAN_ATOMIC_FUNC_OR(32, uint32_t, uint32_t, uint32_t);
    888 ASAN_ATOMIC_FUNC_OR(64, uint64_t, uint64_t, uint64_t);
    889 ASAN_ATOMIC_FUNC_OR(uint, unsigned int, unsigned int, unsigned int);
    890 ASAN_ATOMIC_FUNC_OR(ulong, unsigned long, unsigned long, unsigned long);
    891 
    892 ASAN_ATOMIC_FUNC_CAS(32, uint32_t, uint32_t, uint32_t);
    893 ASAN_ATOMIC_FUNC_CAS(64, uint64_t, uint64_t, uint64_t);
    894 ASAN_ATOMIC_FUNC_CAS(uint, unsigned int, unsigned int, unsigned int);
    895 ASAN_ATOMIC_FUNC_CAS(ulong, unsigned long, unsigned long, unsigned long);
    896 ASAN_ATOMIC_FUNC_CAS(ptr, void *, void, void *);
    897 
    898 ASAN_ATOMIC_FUNC_SWAP(32, uint32_t, uint32_t, uint32_t);
    899 ASAN_ATOMIC_FUNC_SWAP(64, uint64_t, uint64_t, uint64_t);
    900 ASAN_ATOMIC_FUNC_SWAP(uint, unsigned int, unsigned int, unsigned int);
    901 ASAN_ATOMIC_FUNC_SWAP(ulong, unsigned long, unsigned long, unsigned long);
    902 ASAN_ATOMIC_FUNC_SWAP(ptr, void *, void, void *);
    903 
    904 ASAN_ATOMIC_FUNC_DEC(32, uint32_t, uint32_t)
    905 ASAN_ATOMIC_FUNC_DEC(64, uint64_t, uint64_t)
    906 ASAN_ATOMIC_FUNC_DEC(uint, unsigned int, unsigned int);
    907 ASAN_ATOMIC_FUNC_DEC(ulong, unsigned long, unsigned long);
    908 ASAN_ATOMIC_FUNC_DEC(ptr, void *, void);
    909 
    910 ASAN_ATOMIC_FUNC_INC(32, uint32_t, uint32_t)
    911 ASAN_ATOMIC_FUNC_INC(64, uint64_t, uint64_t)
    912 ASAN_ATOMIC_FUNC_INC(uint, unsigned int, unsigned int);
    913 ASAN_ATOMIC_FUNC_INC(ulong, unsigned long, unsigned long);
    914 ASAN_ATOMIC_FUNC_INC(ptr, void *, void);
    915 
    916 /* -------------------------------------------------------------------------- */
    917 
    918 #ifdef __HAVE_KASAN_INSTR_BUS
    919 
    920 #include <sys/bus.h>
    921 
    922 #undef bus_space_read_multi_1
    923 #undef bus_space_read_multi_2
    924 #undef bus_space_read_multi_4
    925 #undef bus_space_read_multi_8
    926 #undef bus_space_read_multi_stream_1
    927 #undef bus_space_read_multi_stream_2
    928 #undef bus_space_read_multi_stream_4
    929 #undef bus_space_read_multi_stream_8
    930 #undef bus_space_read_region_1
    931 #undef bus_space_read_region_2
    932 #undef bus_space_read_region_4
    933 #undef bus_space_read_region_8
    934 #undef bus_space_read_region_stream_1
    935 #undef bus_space_read_region_stream_2
    936 #undef bus_space_read_region_stream_4
    937 #undef bus_space_read_region_stream_8
    938 #undef bus_space_write_multi_1
    939 #undef bus_space_write_multi_2
    940 #undef bus_space_write_multi_4
    941 #undef bus_space_write_multi_8
    942 #undef bus_space_write_multi_stream_1
    943 #undef bus_space_write_multi_stream_2
    944 #undef bus_space_write_multi_stream_4
    945 #undef bus_space_write_multi_stream_8
    946 #undef bus_space_write_region_1
    947 #undef bus_space_write_region_2
    948 #undef bus_space_write_region_4
    949 #undef bus_space_write_region_8
    950 #undef bus_space_write_region_stream_1
    951 #undef bus_space_write_region_stream_2
    952 #undef bus_space_write_region_stream_4
    953 #undef bus_space_write_region_stream_8
    954 
    955 #define ASAN_BUS_READ_FUNC(bytes, bits) \
    956 	void bus_space_read_multi_##bytes(bus_space_tag_t, bus_space_handle_t,	\
    957 	    bus_size_t, uint##bits##_t *, bus_size_t);				\
    958 	void kasan_bus_space_read_multi_##bytes(bus_space_tag_t,		\
    959 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    960 	void kasan_bus_space_read_multi_##bytes(bus_space_tag_t tag,		\
    961 	    bus_space_handle_t hnd, bus_size_t size, uint##bits##_t *buf,	\
    962 	    bus_size_t count)							\
    963 	{									\
    964 		kasan_shadow_check((uintptr_t)buf,				\
    965 		    sizeof(uint##bits##_t) * count, false, __RET_ADDR);		\
    966 		bus_space_read_multi_##bytes(tag, hnd, size, buf, count);	\
    967 	}									\
    968 	void bus_space_read_multi_stream_##bytes(bus_space_tag_t,		\
    969 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    970 	void kasan_bus_space_read_multi_stream_##bytes(bus_space_tag_t,		\
    971 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    972 	void kasan_bus_space_read_multi_stream_##bytes(bus_space_tag_t tag,	\
    973 	    bus_space_handle_t hnd, bus_size_t size, uint##bits##_t *buf,	\
    974 	    bus_size_t count)							\
    975 	{									\
    976 		kasan_shadow_check((uintptr_t)buf,				\
    977 		    sizeof(uint##bits##_t) * count, false, __RET_ADDR);		\
    978 		bus_space_read_multi_stream_##bytes(tag, hnd, size, buf, count);\
    979 	}									\
    980 	void bus_space_read_region_##bytes(bus_space_tag_t, bus_space_handle_t,	\
    981 	    bus_size_t, uint##bits##_t *, bus_size_t);				\
    982 	void kasan_bus_space_read_region_##bytes(bus_space_tag_t,		\
    983 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    984 	void kasan_bus_space_read_region_##bytes(bus_space_tag_t tag,		\
    985 	    bus_space_handle_t hnd, bus_size_t size, uint##bits##_t *buf,	\
    986 	    bus_size_t count)							\
    987 	{									\
    988 		kasan_shadow_check((uintptr_t)buf,				\
    989 		    sizeof(uint##bits##_t) * count, false, __RET_ADDR);		\
    990 		bus_space_read_region_##bytes(tag, hnd, size, buf, count);	\
    991 	}									\
    992 	void bus_space_read_region_stream_##bytes(bus_space_tag_t,		\
    993 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    994 	void kasan_bus_space_read_region_stream_##bytes(bus_space_tag_t,	\
    995 	    bus_space_handle_t, bus_size_t, uint##bits##_t *, bus_size_t);	\
    996 	void kasan_bus_space_read_region_stream_##bytes(bus_space_tag_t tag,	\
    997 	    bus_space_handle_t hnd, bus_size_t size, uint##bits##_t *buf,	\
    998 	    bus_size_t count)							\
    999 	{									\
   1000 		kasan_shadow_check((uintptr_t)buf,				\
   1001 		    sizeof(uint##bits##_t) * count, false, __RET_ADDR);		\
   1002 		bus_space_read_region_stream_##bytes(tag, hnd, size, buf, count);\
   1003 	}
   1004 
   1005 #define ASAN_BUS_WRITE_FUNC(bytes, bits) \
   1006 	void bus_space_write_multi_##bytes(bus_space_tag_t, bus_space_handle_t,	\
   1007 	    bus_size_t, const uint##bits##_t *, bus_size_t);			\
   1008 	void kasan_bus_space_write_multi_##bytes(bus_space_tag_t,		\
   1009 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1010 	void kasan_bus_space_write_multi_##bytes(bus_space_tag_t tag,		\
   1011 	    bus_space_handle_t hnd, bus_size_t size, const uint##bits##_t *buf,	\
   1012 	    bus_size_t count)							\
   1013 	{									\
   1014 		kasan_shadow_check((uintptr_t)buf,				\
   1015 		    sizeof(uint##bits##_t) * count, true, __RET_ADDR);		\
   1016 		bus_space_write_multi_##bytes(tag, hnd, size, buf, count);	\
   1017 	}									\
   1018 	void bus_space_write_multi_stream_##bytes(bus_space_tag_t,		\
   1019 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1020 	void kasan_bus_space_write_multi_stream_##bytes(bus_space_tag_t,	\
   1021 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1022 	void kasan_bus_space_write_multi_stream_##bytes(bus_space_tag_t tag,	\
   1023 	    bus_space_handle_t hnd, bus_size_t size, const uint##bits##_t *buf,	\
   1024 	    bus_size_t count)							\
   1025 	{									\
   1026 		kasan_shadow_check((uintptr_t)buf,				\
   1027 		    sizeof(uint##bits##_t) * count, true, __RET_ADDR);		\
   1028 		bus_space_write_multi_stream_##bytes(tag, hnd, size, buf, count);\
   1029 	}									\
   1030 	void bus_space_write_region_##bytes(bus_space_tag_t, bus_space_handle_t,\
   1031 	    bus_size_t, const uint##bits##_t *, bus_size_t);			\
   1032 	void kasan_bus_space_write_region_##bytes(bus_space_tag_t,		\
   1033 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1034 	void kasan_bus_space_write_region_##bytes(bus_space_tag_t tag,		\
   1035 	    bus_space_handle_t hnd, bus_size_t size, const uint##bits##_t *buf,	\
   1036 	    bus_size_t count)							\
   1037 	{									\
   1038 		kasan_shadow_check((uintptr_t)buf,				\
   1039 		    sizeof(uint##bits##_t) * count, true, __RET_ADDR);		\
   1040 		bus_space_write_region_##bytes(tag, hnd, size, buf, count);	\
   1041 	}									\
   1042 	void bus_space_write_region_stream_##bytes(bus_space_tag_t,		\
   1043 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1044 	void kasan_bus_space_write_region_stream_##bytes(bus_space_tag_t,	\
   1045 	    bus_space_handle_t, bus_size_t, const uint##bits##_t *, bus_size_t);\
   1046 	void kasan_bus_space_write_region_stream_##bytes(bus_space_tag_t tag,	\
   1047 	    bus_space_handle_t hnd, bus_size_t size, const uint##bits##_t *buf,	\
   1048 	    bus_size_t count)							\
   1049 	{									\
   1050 		kasan_shadow_check((uintptr_t)buf,				\
   1051 		    sizeof(uint##bits##_t) * count, true, __RET_ADDR);		\
   1052 		bus_space_write_region_stream_##bytes(tag, hnd, size, buf, count);\
   1053 	}
   1054 
   1055 ASAN_BUS_READ_FUNC(1, 8)
   1056 ASAN_BUS_READ_FUNC(2, 16)
   1057 ASAN_BUS_READ_FUNC(4, 32)
   1058 ASAN_BUS_READ_FUNC(8, 64)
   1059 
   1060 ASAN_BUS_WRITE_FUNC(1, 8)
   1061 ASAN_BUS_WRITE_FUNC(2, 16)
   1062 ASAN_BUS_WRITE_FUNC(4, 32)
   1063 ASAN_BUS_WRITE_FUNC(8, 64)
   1064 
   1065 #endif /* __HAVE_KASAN_INSTR_BUS */
   1066 
   1067 /* -------------------------------------------------------------------------- */
   1068 
   1069 #include <sys/mbuf.h>
   1070 
   1071 static void
   1072 kasan_dma_sync_linear(uint8_t *buf, bus_addr_t offset, bus_size_t len,
   1073     bool write, uintptr_t pc)
   1074 {
   1075 	kasan_shadow_check((uintptr_t)(buf + offset), len, write, pc);
   1076 }
   1077 
   1078 static void
   1079 kasan_dma_sync_mbuf(struct mbuf *m, bus_addr_t offset, bus_size_t len,
   1080     bool write, uintptr_t pc)
   1081 {
   1082 	bus_addr_t minlen;
   1083 
   1084 	for (; m != NULL && len != 0; m = m->m_next) {
   1085 		kasan_shadow_check((uintptr_t)m, sizeof(*m), false, pc);
   1086 
   1087 		if (offset >= m->m_len) {
   1088 			offset -= m->m_len;
   1089 			continue;
   1090 		}
   1091 
   1092 		minlen = MIN(len, m->m_len - offset);
   1093 		kasan_shadow_check((uintptr_t)(mtod(m, char *) + offset),
   1094 		    minlen, write, pc);
   1095 
   1096 		offset = 0;
   1097 		len -= minlen;
   1098 	}
   1099 }
   1100 
   1101 static void
   1102 kasan_dma_sync_uio(struct uio *uio, bus_addr_t offset, bus_size_t len,
   1103     bool write, uintptr_t pc)
   1104 {
   1105 	bus_size_t minlen, resid;
   1106 	struct iovec *iov;
   1107 	int i;
   1108 
   1109 	kasan_shadow_check((uintptr_t)uio, sizeof(struct uio), false, pc);
   1110 
   1111 	if (!VMSPACE_IS_KERNEL_P(uio->uio_vmspace))
   1112 		return;
   1113 
   1114 	resid = uio->uio_resid;
   1115 	iov = uio->uio_iov;
   1116 
   1117 	for (i = 0; i < uio->uio_iovcnt && resid != 0; i++) {
   1118 		kasan_shadow_check((uintptr_t)&iov[i], sizeof(iov[i]),
   1119 		    false, pc);
   1120 		minlen = MIN(resid, iov[i].iov_len);
   1121 		kasan_shadow_check((uintptr_t)iov[i].iov_base, minlen,
   1122 		    write, pc);
   1123 		resid -= minlen;
   1124 	}
   1125 }
   1126 
   1127 void
   1128 kasan_dma_sync(bus_dmamap_t map, bus_addr_t offset, bus_size_t len, int ops)
   1129 {
   1130 	bool write = (ops & (BUS_DMASYNC_PREWRITE|BUS_DMASYNC_POSTWRITE)) != 0;
   1131 
   1132 	switch (map->dm_buftype) {
   1133 	case KASAN_DMA_LINEAR:
   1134 		kasan_dma_sync_linear(map->dm_buf, offset, len, write,
   1135 		    __RET_ADDR);
   1136 		break;
   1137 	case KASAN_DMA_MBUF:
   1138 		kasan_dma_sync_mbuf(map->dm_buf, offset, len, write,
   1139 		    __RET_ADDR);
   1140 		break;
   1141 	case KASAN_DMA_UIO:
   1142 		kasan_dma_sync_uio(map->dm_buf, offset, len, write,
   1143 		    __RET_ADDR);
   1144 		break;
   1145 	case KASAN_DMA_RAW:
   1146 		break;
   1147 	default:
   1148 		panic("%s: impossible", __func__);
   1149 	}
   1150 }
   1151 
   1152 void
   1153 kasan_dma_load(bus_dmamap_t map, void *buf, bus_size_t buflen, int type)
   1154 {
   1155 	map->dm_buf = buf;
   1156 	map->dm_buflen = buflen;
   1157 	map->dm_buftype = type;
   1158 }
   1159 
   1160 /* -------------------------------------------------------------------------- */
   1161 
   1162 void __asan_register_globals(struct __asan_global *, size_t);
   1163 void __asan_unregister_globals(struct __asan_global *, size_t);
   1164 
   1165 void
   1166 __asan_register_globals(struct __asan_global *globals, size_t n)
   1167 {
   1168 	size_t i;
   1169 
   1170 	for (i = 0; i < n; i++) {
   1171 		kasan_mark(globals[i].beg, globals[i].size,
   1172 		    globals[i].size_with_redzone, KASAN_GENERIC_REDZONE);
   1173 	}
   1174 }
   1175 
   1176 void
   1177 __asan_unregister_globals(struct __asan_global *globals, size_t n)
   1178 {
   1179 	/* never called */
   1180 }
   1181 
   1182 #define ASAN_LOAD_STORE(size)					\
   1183 	void __asan_load##size(unsigned long);			\
   1184 	void __asan_load##size(unsigned long addr)		\
   1185 	{							\
   1186 		kasan_shadow_check(addr, size, false, __RET_ADDR);\
   1187 	} 							\
   1188 	void __asan_load##size##_noabort(unsigned long);	\
   1189 	void __asan_load##size##_noabort(unsigned long addr)	\
   1190 	{							\
   1191 		kasan_shadow_check(addr, size, false, __RET_ADDR);\
   1192 	}							\
   1193 	void __asan_store##size(unsigned long);			\
   1194 	void __asan_store##size(unsigned long addr)		\
   1195 	{							\
   1196 		kasan_shadow_check(addr, size, true, __RET_ADDR);\
   1197 	}							\
   1198 	void __asan_store##size##_noabort(unsigned long);	\
   1199 	void __asan_store##size##_noabort(unsigned long addr)	\
   1200 	{							\
   1201 		kasan_shadow_check(addr, size, true, __RET_ADDR);\
   1202 	}
   1203 
   1204 ASAN_LOAD_STORE(1);
   1205 ASAN_LOAD_STORE(2);
   1206 ASAN_LOAD_STORE(4);
   1207 ASAN_LOAD_STORE(8);
   1208 ASAN_LOAD_STORE(16);
   1209 
   1210 void __asan_loadN(unsigned long, size_t);
   1211 void __asan_loadN_noabort(unsigned long, size_t);
   1212 void __asan_storeN(unsigned long, size_t);
   1213 void __asan_storeN_noabort(unsigned long, size_t);
   1214 void __asan_handle_no_return(void);
   1215 
   1216 void
   1217 __asan_loadN(unsigned long addr, size_t size)
   1218 {
   1219 	kasan_shadow_check(addr, size, false, __RET_ADDR);
   1220 }
   1221 
   1222 void
   1223 __asan_loadN_noabort(unsigned long addr, size_t size)
   1224 {
   1225 	kasan_shadow_check(addr, size, false, __RET_ADDR);
   1226 }
   1227 
   1228 void
   1229 __asan_storeN(unsigned long addr, size_t size)
   1230 {
   1231 	kasan_shadow_check(addr, size, true, __RET_ADDR);
   1232 }
   1233 
   1234 void
   1235 __asan_storeN_noabort(unsigned long addr, size_t size)
   1236 {
   1237 	kasan_shadow_check(addr, size, true, __RET_ADDR);
   1238 }
   1239 
   1240 void
   1241 __asan_handle_no_return(void)
   1242 {
   1243 	/* nothing */
   1244 }
   1245 
   1246 #define ASAN_SET_SHADOW(byte) \
   1247 	void __asan_set_shadow_##byte(void *, size_t);			\
   1248 	void __asan_set_shadow_##byte(void *addr, size_t size)		\
   1249 	{								\
   1250 		__builtin_memset((void *)addr, 0x##byte, size);		\
   1251 	}
   1252 
   1253 ASAN_SET_SHADOW(00);
   1254 ASAN_SET_SHADOW(f1);
   1255 ASAN_SET_SHADOW(f2);
   1256 ASAN_SET_SHADOW(f3);
   1257 ASAN_SET_SHADOW(f5);
   1258 ASAN_SET_SHADOW(f8);
   1259 
   1260 void __asan_poison_stack_memory(const void *, size_t);
   1261 void __asan_unpoison_stack_memory(const void *, size_t);
   1262 
   1263 void
   1264 __asan_poison_stack_memory(const void *addr, size_t size)
   1265 {
   1266 	size = roundup(size, KASAN_SHADOW_SCALE_SIZE);
   1267 	kasan_shadow_Nbyte_fill(addr, size, KASAN_USE_AFTER_SCOPE);
   1268 }
   1269 
   1270 void
   1271 __asan_unpoison_stack_memory(const void *addr, size_t size)
   1272 {
   1273 	kasan_shadow_Nbyte_markvalid(addr, size);
   1274 }
   1275 
   1276 void __asan_alloca_poison(const void *, size_t);
   1277 void __asan_allocas_unpoison(const void *, const void *);
   1278 
   1279 void __asan_alloca_poison(const void *addr, size_t size)
   1280 {
   1281 	const void *l, *r;
   1282 
   1283 	KASSERT((vaddr_t)addr % KASAN_ALLOCA_SCALE_SIZE == 0);
   1284 
   1285 	l = (const uint8_t *)addr - KASAN_ALLOCA_SCALE_SIZE;
   1286 	r = (const uint8_t *)addr + roundup(size, KASAN_ALLOCA_SCALE_SIZE);
   1287 
   1288 	kasan_shadow_Nbyte_fill(l, KASAN_ALLOCA_SCALE_SIZE, KASAN_STACK_LEFT);
   1289 	kasan_mark(addr, size, roundup(size, KASAN_ALLOCA_SCALE_SIZE),
   1290 	    KASAN_STACK_MID);
   1291 	kasan_shadow_Nbyte_fill(r, KASAN_ALLOCA_SCALE_SIZE, KASAN_STACK_RIGHT);
   1292 }
   1293 
   1294 void __asan_allocas_unpoison(const void *stkbegin, const void *stkend)
   1295 {
   1296 	size_t size;
   1297 
   1298 	if (__predict_false(!stkbegin))
   1299 		return;
   1300 	if (__predict_false((uintptr_t)stkbegin > (uintptr_t)stkend))
   1301 		return;
   1302 	size = (uintptr_t)stkend - (uintptr_t)stkbegin;
   1303 
   1304 	kasan_shadow_Nbyte_fill(stkbegin, size, 0);
   1305 }
   1306