394 	msan_td_t *mtd;  in kmsan_init_arg()  local
401 	mtd = curthread->td_kmsan;  in kmsan_init_arg()
402 	arg = mtd->tls[mtd->ctx].param_shadow;  in kmsan_init_arg()
409 	msan_td_t *mtd;  in kmsan_init_ret()  local
416 	mtd = curthread->td_kmsan;  in kmsan_init_ret()
417 	arg = mtd->tls[mtd->ctx].retval_shadow;  in kmsan_init_ret()
425 	msan_td_t *mtd;  in kmsan_check_arg()  local
433 	mtd = curthread->td_kmsan;  in kmsan_check_arg()
434 	ctx = mtd->ctx;  in kmsan_check_arg()
435 	arg = mtd->tls[ctx].param_shadow;  in kmsan_check_arg()
440 		orig = &mtd->tls[ctx].param_origin[i / sizeof(msan_orig_t)];  in kmsan_check_arg()
449 	msan_td_t *mtd;  in kmsan_thread_alloc()  local
454 	mtd = td->td_kmsan;  in kmsan_thread_alloc()
455 	if (mtd == NULL) {  in kmsan_thread_alloc()
459 		mtd = malloc(sizeof(*mtd), M_KMSAN, M_WAITOK);  in kmsan_thread_alloc()
461 	__builtin_memset(mtd, 0, sizeof(*mtd));  in kmsan_thread_alloc()
462 	mtd->ctx = 0;  in kmsan_thread_alloc()
468 	td->td_kmsan = mtd;  in kmsan_thread_alloc()
474 	msan_td_t *mtd;  in kmsan_thread_free()  local
481 	mtd = td->td_kmsan;  in kmsan_thread_free()
483 	free(mtd, M_KMSAN);  in kmsan_thread_free()
493 	msan_td_t *mtd;  in kmsan_intr_enter()  local
498 	mtd = curthread->td_kmsan;  in kmsan_intr_enter()
499 	mtd->ctx++;  in kmsan_intr_enter()
500 	if (__predict_false(mtd->ctx >= MSAN_NCONTEXT))  in kmsan_intr_enter()
501 		kmsan_panic("%s: mtd->ctx = %zu", __func__, mtd->ctx);  in kmsan_intr_enter()
507 	msan_td_t *mtd;  in kmsan_intr_leave()  local
512 	mtd = curthread->td_kmsan;  in kmsan_intr_leave()
513 	if (__predict_false(mtd->ctx == 0))  in kmsan_intr_leave()
514 		kmsan_panic("%s: mtd->ctx = %zu", __func__, mtd->ctx);  in kmsan_intr_leave()
515 	mtd->ctx--;  in kmsan_intr_leave()
717 	msan_td_t *mtd;  in __msan_get_context_state()  local
725 	mtd = curthread->td_kmsan;  in __msan_get_context_state()
726 	return (&mtd->tls[mtd->ctx]);  in __msan_get_context_state()