Home | History | Annotate | Line # | Download | only in kern
kern_ras.c revision 1.18
      1 /*	$NetBSD: kern_ras.c,v 1.18 2007/03/12 18:18:33 ad Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2002, 2006 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Gregory McGarry.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  * 3. All advertising materials mentioning features or use of this software
     19  *    must display the following acknowledgement:
     20  *        This product includes software developed by the NetBSD
     21  *        Foundation, Inc. and its contributors.
     22  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  *    contributors may be used to endorse or promote products derived
     24  *    from this software without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36  * POSSIBILITY OF SUCH DAMAGE.
     37  */
     38 
     39 #include <sys/cdefs.h>
     40 __KERNEL_RCSID(0, "$NetBSD: kern_ras.c,v 1.18 2007/03/12 18:18:33 ad Exp $");
     41 
     42 #include <sys/param.h>
     43 #include <sys/lock.h>
     44 #include <sys/systm.h>
     45 #include <sys/pool.h>
     46 #include <sys/proc.h>
     47 #include <sys/ras.h>
     48 
     49 #include <sys/mount.h>
     50 #include <sys/syscallargs.h>
     51 
     52 #include <uvm/uvm_extern.h>
     53 
     54 POOL_INIT(ras_pool, sizeof(struct ras), 0, 0, 0, "raspl",
     55     &pool_allocator_nointr, IPL_NONE);
     56 
     57 #define MAX_RAS_PER_PROC	16
     58 
     59 u_int ras_per_proc = MAX_RAS_PER_PROC;
     60 
     61 #ifdef DEBUG
     62 int ras_debug = 0;
     63 #define DPRINTF(x)	if (ras_debug) printf x
     64 #else
     65 #define DPRINTF(x)	/* nothing */
     66 #endif
     67 
     68 /*
     69  * Check the specified address to see if it is within the
     70  * sequence.  If it is found, we return the restart address,
     71  * otherwise we return -1.  If we do perform a restart, we
     72  * mark the sequence as hit.
     73  */
     74 void *
     75 ras_lookup(struct proc *p, void *addr)
     76 {
     77 	struct ras *rp;
     78 	void *startaddr;
     79 
     80 	startaddr = (void *)-1;
     81 
     82 #ifdef DIAGNOSTIC
     83 	if (addr < (void *)VM_MIN_ADDRESS ||
     84 	    addr > (void *)VM_MAXUSER_ADDRESS)
     85 		return (startaddr);
     86 #endif
     87 
     88 	mutex_enter(&p->p_rasmutex);
     89 	LIST_FOREACH(rp, &p->p_raslist, ras_list) {
     90 		if (addr > rp->ras_startaddr && addr < rp->ras_endaddr) {
     91 			rp->ras_hits++;
     92 			startaddr = rp->ras_startaddr;
     93 #ifdef DIAGNOSTIC
     94 			DPRINTF(("RAS hit: p=%p %p\n", p, addr));
     95 #endif
     96 			break;
     97 		}
     98 	}
     99 	mutex_exit(&p->p_rasmutex);
    100 
    101 	return (startaddr);
    102 }
    103 
    104 /*
    105  * During a fork, we copy all of the sequences from parent p1 to
    106  * the child p2.
    107  */
    108 int
    109 ras_fork(struct proc *p1, struct proc *p2)
    110 {
    111 	struct ras *rp, *nrp;
    112 	int nras;
    113 
    114 again:
    115 	/*
    116 	 * first, try to shortcut.
    117 	 */
    118 
    119 	if (LIST_EMPTY(&p1->p_raslist))
    120 		return (0);
    121 
    122 	/*
    123 	 * count entries.
    124 	 */
    125 
    126 	nras = 0;
    127 	mutex_enter(&p1->p_rasmutex);
    128 	LIST_FOREACH(rp, &p1->p_raslist, ras_list)
    129 		nras++;
    130 	mutex_exit(&p1->p_rasmutex);
    131 
    132 	/*
    133 	 * allocate entries.
    134 	 */
    135 
    136 	for ( ; nras > 0; nras--) {
    137 		nrp = pool_get(&ras_pool, PR_WAITOK);
    138 		nrp->ras_hits = 0;
    139 		LIST_INSERT_HEAD(&p2->p_raslist, nrp, ras_list);
    140 	}
    141 
    142 	/*
    143 	 * copy entries.
    144 	 */
    145 
    146 	mutex_enter(&p1->p_rasmutex);
    147 	nrp = LIST_FIRST(&p2->p_raslist);
    148 	LIST_FOREACH(rp, &p1->p_raslist, ras_list) {
    149 		if (nrp == NULL)
    150 			break;
    151 		nrp->ras_startaddr = rp->ras_startaddr;
    152 		nrp->ras_endaddr = rp->ras_endaddr;
    153 		nrp = LIST_NEXT(nrp, ras_list);
    154 	}
    155 	mutex_exit(&p1->p_rasmutex);
    156 
    157 	/*
    158 	 * if we lose a race, retry.
    159 	 */
    160 
    161 	if (rp != NULL || nrp != NULL) {
    162 		ras_purgeall(p2);
    163 		goto again;
    164 	}
    165 
    166 	DPRINTF(("ras_fork: p1=%p, p2=%p, nras=%d\n", p1, p2, nras));
    167 
    168 	return (0);
    169 }
    170 
    171 /*
    172  * Nuke all sequences for this process.
    173  */
    174 int
    175 ras_purgeall(struct proc *p)
    176 {
    177 	struct ras *rp;
    178 
    179 	mutex_enter(&p->p_rasmutex);
    180 	while (!LIST_EMPTY(&p->p_raslist)) {
    181 		rp = LIST_FIRST(&p->p_raslist);
    182                 DPRINTF(("RAS %p-%p, hits %d\n", rp->ras_startaddr,
    183                     rp->ras_endaddr, rp->ras_hits));
    184 		LIST_REMOVE(rp, ras_list);
    185 		mutex_exit(&p->p_rasmutex);
    186 		pool_put(&ras_pool, rp);
    187 		mutex_enter(&p->p_rasmutex);
    188 	}
    189 	mutex_exit(&p->p_rasmutex);
    190 
    191 	return (0);
    192 }
    193 
    194 #if defined(__HAVE_RAS)
    195 
    196 /*
    197  * Install the new sequence.  If it already exists, return
    198  * an error.
    199  */
    200 static int
    201 ras_install(struct proc *p, void *addr, size_t len)
    202 {
    203 	struct ras *rp;
    204 	struct ras *newrp;
    205 	void *endaddr = (char *)addr + len;
    206 	int nras = 0;
    207 
    208 	if (addr < (void *)VM_MIN_ADDRESS ||
    209 	    endaddr > (void *)VM_MAXUSER_ADDRESS)
    210 		return (EINVAL);
    211 
    212 	if (len <= 0)
    213 		return (EINVAL);
    214 
    215 	newrp = NULL;
    216 again:
    217 	mutex_enter(&p->p_rasmutex);
    218 	LIST_FOREACH(rp, &p->p_raslist, ras_list) {
    219 		if (++nras >= ras_per_proc ||
    220 		    (addr < rp->ras_endaddr && endaddr > rp->ras_startaddr)) {
    221 			mutex_exit(&p->p_rasmutex);
    222 			return (EINVAL);
    223 		}
    224 	}
    225 	if (newrp == NULL) {
    226 		mutex_exit(&p->p_rasmutex);
    227 		newrp = pool_get(&ras_pool, PR_WAITOK);
    228 		goto again;
    229 	}
    230 	newrp->ras_startaddr = addr;
    231 	newrp->ras_endaddr = endaddr;
    232 	newrp->ras_hits = 0;
    233 	LIST_INSERT_HEAD(&p->p_raslist, newrp, ras_list);
    234 	mutex_exit(&p->p_rasmutex);
    235 
    236 	return (0);
    237 }
    238 
    239 /*
    240  * Nuke the specified sequence.  Both address and len must
    241  * match, otherwise we return an error.
    242  */
    243 static int
    244 ras_purge(struct proc *p, void *addr, size_t len)
    245 {
    246 	struct ras *rp;
    247 	void *endaddr = (char *)addr + len;
    248 	int error = ESRCH;
    249 
    250 	mutex_enter(&p->p_rasmutex);
    251 	LIST_FOREACH(rp, &p->p_raslist, ras_list) {
    252 		if (addr == rp->ras_startaddr && endaddr == rp->ras_endaddr) {
    253 			LIST_REMOVE(rp, ras_list);
    254 			break;
    255 		}
    256 	}
    257 	mutex_exit(&p->p_rasmutex);
    258 
    259 	if (rp != NULL) {
    260 		pool_put(&ras_pool, rp);
    261 		error = 0;
    262 	}
    263 
    264 	return (error);
    265 }
    266 
    267 #endif /* defined(__HAVE_RAS) */
    268 
    269 /*ARGSUSED*/
    270 int
    271 sys_rasctl(struct lwp *l, void *v, register_t *retval)
    272 {
    273 
    274 #if defined(__HAVE_RAS)
    275 
    276 	struct sys_rasctl_args /* {
    277 		syscallarg(void *) addr;
    278 		syscallarg(size_t) len;
    279 		syscallarg(int) op;
    280 	} */ *uap = v;
    281 	struct proc *p = l->l_proc;
    282 	void *addr;
    283 	size_t len;
    284 	int op;
    285 	int error;
    286 
    287 	/*
    288 	 * first, extract syscall args from the uap.
    289 	 */
    290 
    291 	addr = (void *)SCARG(uap, addr);
    292 	len = (size_t)SCARG(uap, len);
    293 	op = SCARG(uap, op);
    294 
    295 	DPRINTF(("sys_rasctl: p=%p addr=%p, len=%ld, op=0x%x\n",
    296 	    p, addr, (long)len, op));
    297 
    298 	switch (op) {
    299 	case RAS_INSTALL:
    300 		error = ras_install(p, addr, len);
    301 		break;
    302 	case RAS_PURGE:
    303 		error = ras_purge(p, addr, len);
    304 		break;
    305 	case RAS_PURGE_ALL:
    306 		error = ras_purgeall(p);
    307 		break;
    308 	default:
    309 		error = EINVAL;
    310 		break;
    311 	}
    312 
    313 	return (error);
    314 
    315 #else
    316 
    317 	return (EOPNOTSUPP);
    318 
    319 #endif
    320 
    321 }
    322