Module Name:	src
Committed By:	maxv
Date:		Thu Aug 23 12:18:02 UTC 2018
Modified Files:
	src/sys/kern: subr_pool.c
Log Message:
Add kASan redzones on pools and pool_caches. Also enable POOL_REDZONE
on DIAGNOSTIC.
To generate a diff of this commit:
cvs rdiff -u -r1.223 -r1.224 src/sys/kern/subr_pool.c
Please note that diffs are not public domain; they are subject to the
copyright notices on the relevant files.
Modified files:
Index: src/sys/kern/subr_pool.c
diff -u src/sys/kern/subr_pool.c:1.223 src/sys/kern/subr_pool.c:1.224
--- src/sys/kern/subr_pool.c:1.223	Wed Jul  4 02:19:02 2018
+++ src/sys/kern/subr_pool.c	Thu Aug 23 12:18:02 2018
@@ -1,4 +1,4 @@
-/*	$NetBSD: subr_pool.c,v 1.223 2018/07/04 02:19:02 kamil Exp $	*/
+/*	$NetBSD: subr_pool.c,v 1.224 2018/08/23 12:18:02 maxv Exp $	*/
  
  /*-
   * Copyright (c) 1997, 1999, 2000, 2002, 2007, 2008, 2010, 2014, 2015
@@ -33,7 +33,7 @@
   */
  
  #include <sys/cdefs.h>
-__KERNEL_RCSID(0, "$NetBSD: subr_pool.c,v 1.223 2018/07/04 02:19:02 kamil Exp $");
+__KERNEL_RCSID(0, "$NetBSD: subr_pool.c,v 1.224 2018/08/23 12:18:02 maxv Exp $");
  
  #ifdef _KERNEL_OPT
  #include "opt_ddb.h"
@@ -55,6 +55,7 @@ __KERNEL_RCSID(0, "$NetBSD: subr_pool.c,
  #include <sys/xcall.h>
  #include <sys/cpu.h>
  #include <sys/atomic.h>
+#include <sys/asan.h>
  
  #include <uvm/uvm_extern.h>
  
@@ -85,8 +86,16 @@ static struct pool phpool[PHPOOL_MAX];
  static struct pool psppool;
  #endif
  
+#if defined(DIAGNOSTIC) || defined(KASAN)
+#define POOL_REDZONE
+#endif
+
  #ifdef POOL_REDZONE
-# define POOL_REDZONE_SIZE 2
+# ifdef KASAN
+#  define POOL_REDZONE_SIZE 8
+# else
+#  define POOL_REDZONE_SIZE 2
+# endif
  static void pool_redzone_init(struct pool *, size_t);
  static void pool_redzone_fill(struct pool *, void *);
  static void pool_redzone_check(struct pool *, void *);
@@ -2728,12 +2737,14 @@ pool_page_free_meta(struct pool *pp, voi
  #define STATIC_BYTE	0xFE
  CTASSERT(POOL_REDZONE_SIZE > 1);
  
+#ifndef KASAN
  static inline uint8_t
  pool_pattern_generate(const void *p)
  {
  	return (uint8_t)(((uintptr_t)p) * PRIME
  	   >> ((sizeof(uintptr_t) - sizeof(uint8_t))) * CHAR_BIT);
  }
+#endif
  
  static void
  pool_redzone_init(struct pool *pp, size_t requested_size)
@@ -2777,11 +2788,13 @@ pool_redzone_init(struct pool *pp, size_
  static void
  pool_redzone_fill(struct pool *pp, void *p)
  {
-	uint8_t *cp, pat;
-	const uint8_t *ep;
-
  	if (!pp->pr_redzone)
  		return;
+#ifdef KASAN
+	kasan_alloc(p, pp->pr_reqsize, pp->pr_reqsize + POOL_REDZONE_SIZE);
+#else
+	uint8_t *cp, pat;
+	const uint8_t *ep;
  
  	cp = (uint8_t *)p + pp->pr_reqsize;
  	ep = cp + POOL_REDZONE_SIZE;
@@ -2798,16 +2811,19 @@ pool_redzone_fill(struct pool *pp, void
  		*cp = pool_pattern_generate(cp);
  		cp++;
  	}
+#endif
  }
  
  static void
  pool_redzone_check(struct pool *pp, void *p)
  {
-	uint8_t *cp, pat, expected;
-	const uint8_t *ep;
-
  	if (!pp->pr_redzone)
  		return;
+#ifdef KASAN
+	kasan_free(p, pp->pr_reqsize + POOL_REDZONE_SIZE);
+#else
+	uint8_t *cp, pat, expected;
+	const uint8_t *ep;
  
  	cp = (uint8_t *)p + pp->pr_reqsize;
  	ep = cp + POOL_REDZONE_SIZE;
@@ -2828,6 +2844,7 @@ pool_redzone_check(struct pool *pp, void
  		}
  		cp++;
  	}
+#endif
  }
  
  #endif /* POOL_REDZONE */