Merge from vendor branch BIND:
[dragonfly.git] / sys / platform / vkernel / i386 / cpu_regs.c
1 /*-
2  * Copyright (c) 1992 Terrence R. Lambert.
3  * Copyright (C) 1994, David Greenman
4  * Copyright (c) 1982, 1987, 1990, 1993
5  *      The Regents of the University of California.  All rights reserved.
6  *
7  * This code is derived from software contributed to Berkeley by
8  * William Jolitz.
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 University of
21  *      California, Berkeley and its contributors.
22  * 4. Neither the name of the University nor the names of its contributors
23  *    may be used to endorse or promote products derived from this software
24  *    without specific prior written permission.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
27  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
30  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36  * SUCH DAMAGE.
37  *
38  *      from: @(#)machdep.c     7.4 (Berkeley) 6/3/91
39  * $FreeBSD: src/sys/i386/i386/machdep.c,v 1.385.2.30 2003/05/31 08:48:05 alc Exp $
40  * $DragonFly: src/sys/platform/vkernel/i386/cpu_regs.c,v 1.13 2007/02/03 17:05:58 corecode Exp $
41  */
42
43 #include "use_ether.h"
44 #include "use_npx.h"
45 #include "use_isa.h"
46 #include "opt_atalk.h"
47 #include "opt_compat.h"
48 #include "opt_ddb.h"
49 #include "opt_directio.h"
50 #include "opt_inet.h"
51 #include "opt_ipx.h"
52 #include "opt_msgbuf.h"
53 #include "opt_swap.h"
54
55 #include <sys/param.h>
56 #include <sys/systm.h>
57 #include <sys/sysproto.h>
58 #include <sys/signalvar.h>
59 #include <sys/kernel.h>
60 #include <sys/linker.h>
61 #include <sys/malloc.h>
62 #include <sys/proc.h>
63 #include <sys/buf.h>
64 #include <sys/reboot.h>
65 #include <sys/mbuf.h>
66 #include <sys/msgbuf.h>
67 #include <sys/sysent.h>
68 #include <sys/sysctl.h>
69 #include <sys/vmmeter.h>
70 #include <sys/bus.h>
71 #include <sys/upcall.h>
72 #include <sys/usched.h>
73 #include <sys/reg.h>
74
75 #include <vm/vm.h>
76 #include <vm/vm_param.h>
77 #include <sys/lock.h>
78 #include <vm/vm_kern.h>
79 #include <vm/vm_object.h>
80 #include <vm/vm_page.h>
81 #include <vm/vm_map.h>
82 #include <vm/vm_pager.h>
83 #include <vm/vm_extern.h>
84
85 #include <sys/thread2.h>
86
87 #include <sys/user.h>
88 #include <sys/exec.h>
89 #include <sys/cons.h>
90
91 #include <ddb/ddb.h>
92
93 #include <machine/cpu.h>
94 #include <machine/clock.h>
95 #include <machine/specialreg.h>
96 #include <machine/md_var.h>
97 #include <machine/pcb_ext.h>            /* pcb.h included via sys/user.h */
98 #include <machine/globaldata.h>         /* CPU_prvspace */
99 #include <machine/smp.h>
100 #ifdef PERFMON
101 #include <machine/perfmon.h>
102 #endif
103 #include <machine/cputypes.h>
104
105 #include <bus/isa/rtc.h>
106 #include <machine/vm86.h>
107 #include <sys/random.h>
108 #include <sys/ptrace.h>
109 #include <machine/sigframe.h>
110
111 extern void dblfault_handler (void);
112
113 #ifndef CPU_DISABLE_SSE
114 static void set_fpregs_xmm (struct save87 *, struct savexmm *);
115 static void fill_fpregs_xmm (struct savexmm *, struct save87 *);
116 #endif /* CPU_DISABLE_SSE */
117 #ifdef DIRECTIO
118 extern void ffs_rawread_setup(void);
119 #endif /* DIRECTIO */
120
121 #ifdef SMP
122 int64_t tsc_offsets[MAXCPU];
123 #else
124 int64_t tsc_offsets[1];
125 #endif
126
127 #if defined(SWTCH_OPTIM_STATS)
128 extern int swtch_optim_stats;
129 SYSCTL_INT(_debug, OID_AUTO, swtch_optim_stats,
130         CTLFLAG_RD, &swtch_optim_stats, 0, "");
131 SYSCTL_INT(_debug, OID_AUTO, tlb_flush_count,
132         CTLFLAG_RD, &tlb_flush_count, 0, "");
133 #endif
134
135 static int
136 sysctl_hw_physmem(SYSCTL_HANDLER_ARGS)
137 {
138         int error = sysctl_handle_int(oidp, 0, ctob((int)Maxmem), req);
139         return (error);
140 }
141
142 SYSCTL_PROC(_hw, HW_PHYSMEM, physmem, CTLTYPE_INT|CTLFLAG_RD,
143         0, 0, sysctl_hw_physmem, "IU", "");
144
145 static int
146 sysctl_hw_usermem(SYSCTL_HANDLER_ARGS)
147 {
148         int error = sysctl_handle_int(oidp, 0,
149                 ctob((int)Maxmem - vmstats.v_wire_count), req);
150         return (error);
151 }
152
153 SYSCTL_PROC(_hw, HW_USERMEM, usermem, CTLTYPE_INT|CTLFLAG_RD,
154         0, 0, sysctl_hw_usermem, "IU", "");
155
156 SYSCTL_ULONG(_hw, OID_AUTO, availpages, CTLFLAG_RD, &Maxmem, NULL, "");
157
158 #if 0
159
160 static int
161 sysctl_machdep_msgbuf(SYSCTL_HANDLER_ARGS)
162 {
163         int error;
164
165         /* Unwind the buffer, so that it's linear (possibly starting with
166          * some initial nulls).
167          */
168         error=sysctl_handle_opaque(oidp,msgbufp->msg_ptr+msgbufp->msg_bufr,
169                 msgbufp->msg_size-msgbufp->msg_bufr,req);
170         if(error) return(error);
171         if(msgbufp->msg_bufr>0) {
172                 error=sysctl_handle_opaque(oidp,msgbufp->msg_ptr,
173                         msgbufp->msg_bufr,req);
174         }
175         return(error);
176 }
177
178 SYSCTL_PROC(_machdep, OID_AUTO, msgbuf, CTLTYPE_STRING|CTLFLAG_RD,
179         0, 0, sysctl_machdep_msgbuf, "A","Contents of kernel message buffer");
180
181 static int msgbuf_clear;
182
183 static int
184 sysctl_machdep_msgbuf_clear(SYSCTL_HANDLER_ARGS)
185 {
186         int error;
187         error = sysctl_handle_int(oidp, oidp->oid_arg1, oidp->oid_arg2,
188                 req);
189         if (!error && req->newptr) {
190                 /* Clear the buffer and reset write pointer */
191                 bzero(msgbufp->msg_ptr,msgbufp->msg_size);
192                 msgbufp->msg_bufr=msgbufp->msg_bufx=0;
193                 msgbuf_clear=0;
194         }
195         return (error);
196 }
197
198 SYSCTL_PROC(_machdep, OID_AUTO, msgbuf_clear, CTLTYPE_INT|CTLFLAG_RW,
199         &msgbuf_clear, 0, sysctl_machdep_msgbuf_clear, "I",
200         "Clear kernel message buffer");
201
202 #endif
203
204 /*
205  * Send an interrupt to process.
206  *
207  * Stack is set up to allow sigcode stored
208  * at top to call routine, followed by kcall
209  * to sigreturn routine below.  After sigreturn
210  * resets the signal mask, the stack, and the
211  * frame pointer, it returns to the user
212  * specified pc, psl.
213  */
214 void
215 sendsig(sig_t catcher, int sig, sigset_t *mask, u_long code)
216 {
217         struct lwp *lp = curthread->td_lwp;
218         struct proc *p = lp->lwp_proc;
219         struct trapframe *regs;
220         struct sigacts *psp = p->p_sigacts;
221         struct sigframe sf, *sfp;
222         int oonstack;
223
224         regs = lp->lwp_md.md_regs;
225         oonstack = (lp->lwp_sigstk.ss_flags & SS_ONSTACK) ? 1 : 0;
226
227         /* save user context */
228         bzero(&sf, sizeof(struct sigframe));
229         sf.sf_uc.uc_sigmask = *mask;
230         sf.sf_uc.uc_stack = lp->lwp_sigstk;
231         sf.sf_uc.uc_mcontext.mc_onstack = oonstack;
232         bcopy(regs, &sf.sf_uc.uc_mcontext.mc_gs, sizeof(struct trapframe));
233
234         /* make the size of the saved context visible to userland */
235         sf.sf_uc.uc_mcontext.mc_len = sizeof(sf.sf_uc.uc_mcontext); 
236
237         /* save mailbox pending state for syscall interlock semantics */
238         if (p->p_flag & P_MAILBOX)
239                 sf.sf_uc.uc_mcontext.mc_xflags |= PGEX_MAILBOX;
240
241
242         /* Allocate and validate space for the signal handler context. */
243         /* XXX lwp flags */
244         if ((lp->lwp_flag & LWP_ALTSTACK) != 0 && !oonstack &&
245             SIGISMEMBER(psp->ps_sigonstack, sig)) {
246                 sfp = (struct sigframe *)(lp->lwp_sigstk.ss_sp +
247                     lp->lwp_sigstk.ss_size - sizeof(struct sigframe));
248                 lp->lwp_sigstk.ss_flags |= SS_ONSTACK;
249         }
250         else
251                 sfp = (struct sigframe *)regs->tf_esp - 1;
252
253         /* Translate the signal is appropriate */
254         if (p->p_sysent->sv_sigtbl) {
255                 if (sig <= p->p_sysent->sv_sigsize)
256                         sig = p->p_sysent->sv_sigtbl[_SIG_IDX(sig)];
257         }
258
259         /* Build the argument list for the signal handler. */
260         sf.sf_signum = sig;
261         sf.sf_ucontext = (register_t)&sfp->sf_uc;
262         if (SIGISMEMBER(psp->ps_siginfo, sig)) {
263                 /* Signal handler installed with SA_SIGINFO. */
264                 sf.sf_siginfo = (register_t)&sfp->sf_si;
265                 sf.sf_ahu.sf_action = (__siginfohandler_t *)catcher;
266
267                 /* fill siginfo structure */
268                 sf.sf_si.si_signo = sig;
269                 sf.sf_si.si_code = code;
270                 sf.sf_si.si_addr = (void*)regs->tf_err;
271         }
272         else {
273                 /* Old FreeBSD-style arguments. */
274                 sf.sf_siginfo = code;
275                 sf.sf_addr = regs->tf_err;
276                 sf.sf_ahu.sf_handler = catcher;
277         }
278
279 #if 0
280         /*
281          * If we're a vm86 process, we want to save the segment registers.
282          * We also change eflags to be our emulated eflags, not the actual
283          * eflags.
284          */
285         if (regs->tf_eflags & PSL_VM) {
286                 struct trapframe_vm86 *tf = (struct trapframe_vm86 *)regs;
287                 struct vm86_kernel *vm86 = &lp->lwp_thread->td_pcb->pcb_ext->ext_vm86;
288
289                 sf.sf_uc.uc_mcontext.mc_gs = tf->tf_vm86_gs;
290                 sf.sf_uc.uc_mcontext.mc_fs = tf->tf_vm86_fs;
291                 sf.sf_uc.uc_mcontext.mc_es = tf->tf_vm86_es;
292                 sf.sf_uc.uc_mcontext.mc_ds = tf->tf_vm86_ds;
293
294                 if (vm86->vm86_has_vme == 0)
295                         sf.sf_uc.uc_mcontext.mc_eflags =
296                             (tf->tf_eflags & ~(PSL_VIF | PSL_VIP)) |
297                             (vm86->vm86_eflags & (PSL_VIF | PSL_VIP));
298
299                 /*
300                  * Clear PSL_NT to inhibit T_TSSFLT faults on return from
301                  * syscalls made by the signal handler.  This just avoids
302                  * wasting time for our lazy fixup of such faults.  PSL_NT
303                  * does nothing in vm86 mode, but vm86 programs can set it
304                  * almost legitimately in probes for old cpu types.
305                  */
306                 tf->tf_eflags &= ~(PSL_VM | PSL_NT | PSL_VIF | PSL_VIP);
307         }
308 #endif
309
310         /*
311          * Copy the sigframe out to the user's stack.
312          */
313         if (copyout(&sf, sfp, sizeof(struct sigframe)) != 0) {
314                 /*
315                  * Something is wrong with the stack pointer.
316                  * ...Kill the process.
317                  */
318                 sigexit(p, SIGILL);
319         }
320
321         regs->tf_esp = (int)sfp;
322         regs->tf_eip = PS_STRINGS - *(p->p_sysent->sv_szsigcode);
323         regs->tf_eflags &= ~PSL_T;
324         regs->tf_cs = _ucodesel;
325         regs->tf_ds = _udatasel;
326         regs->tf_es = _udatasel;
327         if (regs->tf_trapno == T_PROTFLT) {
328                 regs->tf_fs = _udatasel;
329                 regs->tf_gs = _udatasel;
330         }
331         regs->tf_ss = _udatasel;
332 }
333
334 /*
335  * Sanitize the trapframe for a virtual kernel passing control to a custom
336  * VM context.
337  *
338  * Allow userland to set or maintain PSL_RF, the resume flag.  This flag
339  * basically controls whether the return PC should skip the first instruction
340  * (as in an explicit system call) or re-execute it (as in an exception).
341  */
342 int
343 cpu_sanitize_frame(struct trapframe *frame)
344 {
345         frame->tf_cs = _ucodesel;
346         frame->tf_ds = _udatasel;
347         frame->tf_es = _udatasel;
348 #if 0
349         frame->tf_fs = _udatasel;
350         frame->tf_gs = _udatasel;
351 #endif
352         frame->tf_ss = _udatasel;
353         frame->tf_eflags &= (PSL_RF | PSL_USERCHANGE);
354         frame->tf_eflags |= PSL_RESERVED_DEFAULT | PSL_I;
355         return(0);
356 }
357
358 int
359 cpu_sanitize_tls(struct savetls *tls)
360 {
361          struct segment_descriptor *desc;
362          int i;
363
364          for (i = 0; i < NGTLS; ++i) {
365                 desc = &tls->tls[i];
366                 if (desc->sd_dpl == 0 && desc->sd_type == 0)
367                         continue;
368                 if (desc->sd_def32 == 0)
369                         return(ENXIO);
370                 if (desc->sd_type != SDT_MEMRWA)
371                         return(ENXIO);
372                 if (desc->sd_dpl != SEL_UPL)
373                         return(ENXIO);
374                 if (desc->sd_xx != 0 || desc->sd_p != 1)
375                         return(ENXIO);
376          }
377          return(0);
378 }
379
380 /*
381  * sigreturn(ucontext_t *sigcntxp)
382  *
383  * System call to cleanup state after a signal
384  * has been taken.  Reset signal mask and
385  * stack state from context left by sendsig (above).
386  * Return to previous pc and psl as specified by
387  * context left by sendsig. Check carefully to
388  * make sure that the user has not modified the
389  * state to gain improper privileges.
390  */
391 #define EFL_SECURE(ef, oef)     ((((ef) ^ (oef)) & ~PSL_USERCHANGE) == 0)
392 #define CS_SECURE(cs)           (ISPL(cs) == SEL_UPL)
393
394 int
395 sys_sigreturn(struct sigreturn_args *uap)
396 {
397         struct lwp *lp = curthread->td_lwp;
398         struct proc *p = lp->lwp_proc;
399         struct trapframe *regs;
400         ucontext_t ucp;
401         int cs;
402         int eflags;
403         int error;
404
405         error = copyin(uap->sigcntxp, &ucp, sizeof(ucp));
406         if (error)
407                 return (error);
408
409         regs = lp->lwp_md.md_regs;
410         eflags = ucp.uc_mcontext.mc_eflags;
411
412 #if 0
413         if (eflags & PSL_VM) {
414                 struct trapframe_vm86 *tf = (struct trapframe_vm86 *)regs;
415                 struct vm86_kernel *vm86;
416
417                 /*
418                  * if pcb_ext == 0 or vm86_inited == 0, the user hasn't
419                  * set up the vm86 area, and we can't enter vm86 mode.
420                  */
421                 if (lp->lwp_thread->td_pcb->pcb_ext == 0)
422                         return (EINVAL);
423                 vm86 = &lp->lwp_thread->td_pcb->pcb_ext->ext_vm86;
424                 if (vm86->vm86_inited == 0)
425                         return (EINVAL);
426
427                 /* go back to user mode if both flags are set */
428                 if ((eflags & PSL_VIP) && (eflags & PSL_VIF))
429                         trapsignal(lp->lwp_proc, SIGBUS, 0);
430
431                 if (vm86->vm86_has_vme) {
432                         eflags = (tf->tf_eflags & ~VME_USERCHANGE) |
433                             (eflags & VME_USERCHANGE) | PSL_VM;
434                 } else {
435                         vm86->vm86_eflags = eflags;     /* save VIF, VIP */
436                         eflags = (tf->tf_eflags & ~VM_USERCHANGE) |                                         (eflags & VM_USERCHANGE) | PSL_VM;
437                 }
438                 bcopy(&ucp.uc_mcontext.mc_gs, tf, sizeof(struct trapframe));
439                 tf->tf_eflags = eflags;
440                 tf->tf_vm86_ds = tf->tf_ds;
441                 tf->tf_vm86_es = tf->tf_es;
442                 tf->tf_vm86_fs = tf->tf_fs;
443                 tf->tf_vm86_gs = tf->tf_gs;
444                 tf->tf_ds = _udatasel;
445                 tf->tf_es = _udatasel;
446 #if 0
447                 tf->tf_fs = _udatasel;
448                 tf->tf_gs = _udatasel;
449 #endif
450         } else 
451 #endif
452         {
453                 /*
454                  * Don't allow users to change privileged or reserved flags.
455                  */
456                 /*
457                  * XXX do allow users to change the privileged flag PSL_RF.
458                  * The cpu sets PSL_RF in tf_eflags for faults.  Debuggers
459                  * should sometimes set it there too.  tf_eflags is kept in
460                  * the signal context during signal handling and there is no
461                  * other place to remember it, so the PSL_RF bit may be
462                  * corrupted by the signal handler without us knowing.
463                  * Corruption of the PSL_RF bit at worst causes one more or
464                  * one less debugger trap, so allowing it is fairly harmless.
465                  */
466                 if (!EFL_SECURE(eflags & ~PSL_RF, regs->tf_eflags & ~PSL_RF)) {
467                         kprintf("sigreturn: eflags = 0x%x\n", eflags);
468                         return(EINVAL);
469                 }
470
471                 /*
472                  * Don't allow users to load a valid privileged %cs.  Let the
473                  * hardware check for invalid selectors, excess privilege in
474                  * other selectors, invalid %eip's and invalid %esp's.
475                  */
476                 cs = ucp.uc_mcontext.mc_cs;
477                 if (!CS_SECURE(cs)) {
478                         kprintf("sigreturn: cs = 0x%x\n", cs);
479                         trapsignal(lp, SIGBUS, T_PROTFLT);
480                         return(EINVAL);
481                 }
482                 bcopy(&ucp.uc_mcontext.mc_gs, regs, sizeof(struct trapframe));
483         }
484
485         /*
486          * Merge saved signal mailbox pending flag to maintain interlock
487          * semantics against system calls.
488          */
489         if (ucp.uc_mcontext.mc_xflags & PGEX_MAILBOX)
490                 p->p_flag |= P_MAILBOX;
491
492         if (ucp.uc_mcontext.mc_onstack & 1)
493                 lp->lwp_sigstk.ss_flags |= SS_ONSTACK;
494         else
495                 lp->lwp_sigstk.ss_flags &= ~SS_ONSTACK;
496
497         lp->lwp_sigmask = ucp.uc_sigmask;
498         SIG_CANTMASK(lp->lwp_sigmask);
499         return(EJUSTRETURN);
500 }
501
502 /*
503  * Stack frame on entry to function.  %eax will contain the function vector,
504  * %ecx will contain the function data.  flags, ecx, and eax will have 
505  * already been pushed on the stack.
506  */
507 struct upc_frame {
508         register_t      eax;
509         register_t      ecx;
510         register_t      edx;
511         register_t      flags;
512         register_t      oldip;
513 };
514
515 void
516 sendupcall(struct vmupcall *vu, int morepending)
517 {
518         struct lwp *lp = curthread->td_lwp;
519         struct proc *p = lp->lwp_proc;
520         struct trapframe *regs;
521         struct upcall upcall;
522         struct upc_frame upc_frame;
523         int     crit_count = 0;
524
525         /*
526          * If we are a virtual kernel running an emulated user process
527          * context, switch back to the virtual kernel context before
528          * trying to post the signal.
529          */
530         if (p->p_vkernel && p->p_vkernel->vk_current) {
531                 lp->lwp_md.md_regs->tf_trapno = 0;
532                 vkernel_trap(p, lp->lwp_md.md_regs);
533         }
534
535         /*
536          * Get the upcall data structure
537          */
538         if (copyin(lp->lwp_upcall, &upcall, sizeof(upcall)) ||
539             copyin((char *)upcall.upc_uthread + upcall.upc_critoff, &crit_count, sizeof(int))
540         ) {
541                 vu->vu_pending = 0;
542                 kprintf("bad upcall address\n");
543                 return;
544         }
545
546         /*
547          * If the data structure is already marked pending or has a critical
548          * section count, mark the data structure as pending and return 
549          * without doing an upcall.  vu_pending is left set.
550          */
551         if (upcall.upc_pending || crit_count >= vu->vu_pending) {
552                 if (upcall.upc_pending < vu->vu_pending) {
553                         upcall.upc_pending = vu->vu_pending;
554                         copyout(&upcall.upc_pending, &lp->lwp_upcall->upc_pending,
555                                 sizeof(upcall.upc_pending));
556                 }
557                 return;
558         }
559
560         /*
561          * We can run this upcall now, clear vu_pending.
562          *
563          * Bump our critical section count and set or clear the
564          * user pending flag depending on whether more upcalls are
565          * pending.  The user will be responsible for calling 
566          * upc_dispatch(-1) to process remaining upcalls.
567          */
568         vu->vu_pending = 0;
569         upcall.upc_pending = morepending;
570         crit_count += TDPRI_CRIT;
571         copyout(&upcall.upc_pending, &lp->lwp_upcall->upc_pending, 
572                 sizeof(upcall.upc_pending));
573         copyout(&crit_count, (char *)upcall.upc_uthread + upcall.upc_critoff,
574                 sizeof(int));
575
576         /*
577          * Construct a stack frame and issue the upcall
578          */
579         regs = lp->lwp_md.md_regs;
580         upc_frame.eax = regs->tf_eax;
581         upc_frame.ecx = regs->tf_ecx;
582         upc_frame.edx = regs->tf_edx;
583         upc_frame.flags = regs->tf_eflags;
584         upc_frame.oldip = regs->tf_eip;
585         if (copyout(&upc_frame, (void *)(regs->tf_esp - sizeof(upc_frame)),
586             sizeof(upc_frame)) != 0) {
587                 kprintf("bad stack on upcall\n");
588         } else {
589                 regs->tf_eax = (register_t)vu->vu_func;
590                 regs->tf_ecx = (register_t)vu->vu_data;
591                 regs->tf_edx = (register_t)lp->lwp_upcall;
592                 regs->tf_eip = (register_t)vu->vu_ctx;
593                 regs->tf_esp -= sizeof(upc_frame);
594         }
595 }
596
597 /*
598  * fetchupcall occurs in the context of a system call, which means that
599  * we have to return EJUSTRETURN in order to prevent eax and edx from
600  * being overwritten by the syscall return value.
601  *
602  * if vu is not NULL we return the new context in %edx, the new data in %ecx,
603  * and the function pointer in %eax.  
604  */
605 int
606 fetchupcall (struct vmupcall *vu, int morepending, void *rsp)
607 {
608         struct upc_frame upc_frame;
609         struct lwp *lp = curthread->td_lwp;
610         struct trapframe *regs;
611         int error;
612         struct upcall upcall;
613         int crit_count;
614
615         regs = lp->lwp_md.md_regs;
616
617         error = copyout(&morepending, &lp->lwp_upcall->upc_pending, sizeof(int));
618         if (error == 0) {
619             if (vu) {
620                 /*
621                  * This jumps us to the next ready context.
622                  */
623                 vu->vu_pending = 0;
624                 error = copyin(lp->lwp_upcall, &upcall, sizeof(upcall));
625                 crit_count = 0;
626                 if (error == 0)
627                         error = copyin((char *)upcall.upc_uthread + upcall.upc_critoff, &crit_count, sizeof(int));
628                 crit_count += TDPRI_CRIT;
629                 if (error == 0)
630                         error = copyout(&crit_count, (char *)upcall.upc_uthread + upcall.upc_critoff, sizeof(int));
631                 regs->tf_eax = (register_t)vu->vu_func;
632                 regs->tf_ecx = (register_t)vu->vu_data;
633                 regs->tf_edx = (register_t)lp->lwp_upcall;
634                 regs->tf_eip = (register_t)vu->vu_ctx;
635                 regs->tf_esp = (register_t)rsp;
636             } else {
637                 /*
638                  * This returns us to the originally interrupted code.
639                  */
640                 error = copyin(rsp, &upc_frame, sizeof(upc_frame));
641                 regs->tf_eax = upc_frame.eax;
642                 regs->tf_ecx = upc_frame.ecx;
643                 regs->tf_edx = upc_frame.edx;
644                 regs->tf_eflags = (regs->tf_eflags & ~PSL_USERCHANGE) |
645                                 (upc_frame.flags & PSL_USERCHANGE);
646                 regs->tf_eip = upc_frame.oldip;
647                 regs->tf_esp = (register_t)((char *)rsp + sizeof(upc_frame));
648             }
649         }
650         if (error == 0)
651                 error = EJUSTRETURN;
652         return(error);
653 }
654
655 /*
656  * cpu_idle() represents the idle LWKT.  You cannot return from this function
657  * (unless you want to blow things up!).  Instead we look for runnable threads
658  * and loop or halt as appropriate.  Giant is not held on entry to the thread.
659  *
660  * The main loop is entered with a critical section held, we must release
661  * the critical section before doing anything else.  lwkt_switch() will
662  * check for pending interrupts due to entering and exiting its own 
663  * critical section.
664  *
665  * Note on cpu_idle_hlt:  On an SMP system we rely on a scheduler IPI
666  * to wake a HLTed cpu up.  However, there are cases where the idlethread
667  * will be entered with the possibility that no IPI will occur and in such
668  * cases lwkt_switch() sets TDF_IDLE_NOHLT.
669  */
670 static int      cpu_idle_hlt = 1;
671 static int      cpu_idle_hltcnt;
672 static int      cpu_idle_spincnt;
673 SYSCTL_INT(_machdep, OID_AUTO, cpu_idle_hlt, CTLFLAG_RW,
674     &cpu_idle_hlt, 0, "Idle loop HLT enable");
675 SYSCTL_INT(_machdep, OID_AUTO, cpu_idle_hltcnt, CTLFLAG_RW,
676     &cpu_idle_hltcnt, 0, "Idle loop entry halts");
677 SYSCTL_INT(_machdep, OID_AUTO, cpu_idle_spincnt, CTLFLAG_RW,
678     &cpu_idle_spincnt, 0, "Idle loop entry spins");
679
680 void
681 cpu_idle(void)
682 {
683         struct thread *td = curthread;
684
685         crit_exit();
686         KKASSERT(td->td_pri < TDPRI_CRIT);
687         for (;;) {
688                 /*
689                  * See if there are any LWKTs ready to go.
690                  */
691                 lwkt_switch();
692
693                 /*
694                  * If we are going to halt call splz unconditionally after
695                  * CLIing to catch any interrupt races.  Note that we are
696                  * at SPL0 and interrupts are enabled.
697                  *
698                  * We must poll our mailbox signals prior to calling 
699                  * sigpause() in order to properly interlock with them.
700                  */
701                 if (cpu_idle_hlt && !lwkt_runnable() &&
702                     (td->td_flags & TDF_IDLE_NOHLT) == 0) {
703                         splz();
704                         signalmailbox(NULL);
705                         if (!lwkt_runnable()) {
706                                 sigpause(0);
707                         }
708 #ifdef SMP
709                         else {
710                             __asm __volatile("pause");
711                         }
712 #endif
713                         ++cpu_idle_hltcnt;
714                 } else {
715                         td->td_flags &= ~TDF_IDLE_NOHLT;
716                         splz();
717                         signalmailbox(NULL);
718 #ifdef SMP
719                         /*__asm __volatile("sti; pause");*/
720                         __asm __volatile("pause");
721 #else
722                         /*__asm __volatile("sti");*/
723 #endif
724                         ++cpu_idle_spincnt;
725                 }
726         }
727 }
728
729 /*
730  * Clear registers on exec
731  */
732 void
733 exec_setregs(u_long entry, u_long stack, u_long ps_strings)
734 {
735         struct thread *td = curthread;
736         struct lwp *lp = td->td_lwp;
737         struct trapframe *regs = lp->lwp_md.md_regs;
738         struct pcb *pcb = lp->lwp_thread->td_pcb;
739
740         /* was i386_user_cleanup() in NetBSD */
741         user_ldt_free(pcb);
742   
743         bzero((char *)regs, sizeof(struct trapframe));
744         regs->tf_eip = entry;
745         regs->tf_esp = stack;
746         regs->tf_eflags = PSL_USER | (regs->tf_eflags & PSL_T);
747         regs->tf_ss = 0;
748         regs->tf_ds = 0;
749         regs->tf_es = 0;
750         regs->tf_fs = 0;
751         regs->tf_gs = 0;
752         regs->tf_cs = 0;
753
754         /* PS_STRINGS value for BSD/OS binaries.  It is 0 for non-BSD/OS. */
755         regs->tf_ebx = ps_strings;
756
757         /*
758          * Reset the hardware debug registers if they were in use.
759          * They won't have any meaning for the newly exec'd process.  
760          */
761         if (pcb->pcb_flags & PCB_DBREGS) {
762                 pcb->pcb_dr0 = 0;
763                 pcb->pcb_dr1 = 0;
764                 pcb->pcb_dr2 = 0;
765                 pcb->pcb_dr3 = 0;
766                 pcb->pcb_dr6 = 0;
767                 pcb->pcb_dr7 = 0;
768                 if (pcb == td->td_pcb) {
769                         /*
770                          * Clear the debug registers on the running
771                          * CPU, otherwise they will end up affecting
772                          * the next process we switch to.
773                          */
774                         reset_dbregs();
775                 }
776                 pcb->pcb_flags &= ~PCB_DBREGS;
777         }
778
779         /*
780          * Initialize the math emulator (if any) for the current process.
781          * Actually, just clear the bit that says that the emulator has
782          * been initialized.  Initialization is delayed until the process
783          * traps to the emulator (if it is done at all) mainly because
784          * emulators don't provide an entry point for initialization.
785          */
786         pcb->pcb_flags &= ~FP_SOFTFP;
787
788         /*
789          * note: do not set CR0_TS here.  npxinit() must do it after clearing
790          * gd_npxthread.  Otherwise a preemptive interrupt thread may panic
791          * in npxdna().
792          */
793         crit_enter();
794 #if 0
795         load_cr0(rcr0() | CR0_MP);
796 #endif
797
798 #if NNPX > 0
799         /* Initialize the npx (if any) for the current process. */
800         npxinit(__INITIAL_NPXCW__);
801 #endif
802         crit_exit();
803
804         /*
805          * note: linux emulator needs edx to be 0x0 on entry, which is
806          * handled in execve simply by setting the 64 bit syscall
807          * return value to 0.
808          */
809 }
810
811 void
812 cpu_setregs(void)
813 {
814 #if 0
815         unsigned int cr0;
816
817         cr0 = rcr0();
818         cr0 |= CR0_NE;                  /* Done by npxinit() */
819         cr0 |= CR0_MP | CR0_TS;         /* Done at every execve() too. */
820 #ifdef I386_CPU
821         if (cpu_class != CPUCLASS_386)
822 #endif
823                 cr0 |= CR0_WP | CR0_AM;
824         load_cr0(cr0);
825         load_gs(_udatasel);
826 #endif
827 }
828
829 static int
830 sysctl_machdep_adjkerntz(SYSCTL_HANDLER_ARGS)
831 {
832         int error;
833         error = sysctl_handle_int(oidp, oidp->oid_arg1, oidp->oid_arg2,
834                 req);
835         if (!error && req->newptr)
836                 resettodr();
837         return (error);
838 }
839
840 SYSCTL_PROC(_machdep, CPU_ADJKERNTZ, adjkerntz, CTLTYPE_INT|CTLFLAG_RW,
841         &adjkerntz, 0, sysctl_machdep_adjkerntz, "I", "");
842
843 extern u_long bootdev;          /* not a cdev_t - encoding is different */
844 SYSCTL_ULONG(_machdep, OID_AUTO, guessed_bootdev,
845         CTLFLAG_RD, &bootdev, 0, "Boot device (not in cdev_t format)");
846
847 /*
848  * Initialize 386 and configure to run kernel
849  */
850
851 /*
852  * Initialize segments & interrupt table
853  */
854
855 extern  struct user *proc0paddr;
856
857 #if 0
858
859 extern inthand_t
860         IDTVEC(div), IDTVEC(dbg), IDTVEC(nmi), IDTVEC(bpt), IDTVEC(ofl),
861         IDTVEC(bnd), IDTVEC(ill), IDTVEC(dna), IDTVEC(fpusegm),
862         IDTVEC(tss), IDTVEC(missing), IDTVEC(stk), IDTVEC(prot),
863         IDTVEC(page), IDTVEC(mchk), IDTVEC(fpu), IDTVEC(align),
864         IDTVEC(xmm), IDTVEC(syscall),
865         IDTVEC(rsvd0);
866 extern inthand_t
867         IDTVEC(int0x80_syscall);
868
869 #endif
870
871 #ifdef DEBUG_INTERRUPTS
872 extern inthand_t *Xrsvdary[256];
873 #endif
874
875 int
876 ptrace_set_pc(struct lwp *lp, unsigned long addr)
877 {
878         lp->lwp_md.md_regs->tf_eip = addr;
879         return (0);
880 }
881
882 int
883 ptrace_single_step(struct lwp *lp)
884 {
885         lp->lwp_md.md_regs->tf_eflags |= PSL_T;
886         return (0);
887 }
888
889 int
890 fill_regs(struct lwp *lp, struct reg *regs)
891 {
892         struct trapframe *tp;
893
894         tp = lp->lwp_md.md_regs;
895         regs->r_gs = tp->tf_gs;
896         regs->r_fs = tp->tf_fs;
897         regs->r_es = tp->tf_es;
898         regs->r_ds = tp->tf_ds;
899         regs->r_edi = tp->tf_edi;
900         regs->r_esi = tp->tf_esi;
901         regs->r_ebp = tp->tf_ebp;
902         regs->r_ebx = tp->tf_ebx;
903         regs->r_edx = tp->tf_edx;
904         regs->r_ecx = tp->tf_ecx;
905         regs->r_eax = tp->tf_eax;
906         regs->r_eip = tp->tf_eip;
907         regs->r_cs = tp->tf_cs;
908         regs->r_eflags = tp->tf_eflags;
909         regs->r_esp = tp->tf_esp;
910         regs->r_ss = tp->tf_ss;
911         return (0);
912 }
913
914 int
915 set_regs(struct lwp *lp, struct reg *regs)
916 {
917         struct trapframe *tp;
918
919         tp = lp->lwp_md.md_regs;
920         if (!EFL_SECURE(regs->r_eflags, tp->tf_eflags) ||
921             !CS_SECURE(regs->r_cs))
922                 return (EINVAL);
923         tp->tf_gs = regs->r_gs;
924         tp->tf_fs = regs->r_fs;
925         tp->tf_es = regs->r_es;
926         tp->tf_ds = regs->r_ds;
927         tp->tf_edi = regs->r_edi;
928         tp->tf_esi = regs->r_esi;
929         tp->tf_ebp = regs->r_ebp;
930         tp->tf_ebx = regs->r_ebx;
931         tp->tf_edx = regs->r_edx;
932         tp->tf_ecx = regs->r_ecx;
933         tp->tf_eax = regs->r_eax;
934         tp->tf_eip = regs->r_eip;
935         tp->tf_cs = regs->r_cs;
936         tp->tf_eflags = regs->r_eflags;
937         tp->tf_esp = regs->r_esp;
938         tp->tf_ss = regs->r_ss;
939         return (0);
940 }
941
942 #ifndef CPU_DISABLE_SSE
943 static void
944 fill_fpregs_xmm(struct savexmm *sv_xmm, struct save87 *sv_87)
945 {
946         struct env87 *penv_87 = &sv_87->sv_env;
947         struct envxmm *penv_xmm = &sv_xmm->sv_env;
948         int i;
949
950         /* FPU control/status */
951         penv_87->en_cw = penv_xmm->en_cw;
952         penv_87->en_sw = penv_xmm->en_sw;
953         penv_87->en_tw = penv_xmm->en_tw;
954         penv_87->en_fip = penv_xmm->en_fip;
955         penv_87->en_fcs = penv_xmm->en_fcs;
956         penv_87->en_opcode = penv_xmm->en_opcode;
957         penv_87->en_foo = penv_xmm->en_foo;
958         penv_87->en_fos = penv_xmm->en_fos;
959
960         /* FPU registers */
961         for (i = 0; i < 8; ++i)
962                 sv_87->sv_ac[i] = sv_xmm->sv_fp[i].fp_acc;
963
964         sv_87->sv_ex_sw = sv_xmm->sv_ex_sw;
965 }
966
967 static void
968 set_fpregs_xmm(struct save87 *sv_87, struct savexmm *sv_xmm)
969 {
970         struct env87 *penv_87 = &sv_87->sv_env;
971         struct envxmm *penv_xmm = &sv_xmm->sv_env;
972         int i;
973
974         /* FPU control/status */
975         penv_xmm->en_cw = penv_87->en_cw;
976         penv_xmm->en_sw = penv_87->en_sw;
977         penv_xmm->en_tw = penv_87->en_tw;
978         penv_xmm->en_fip = penv_87->en_fip;
979         penv_xmm->en_fcs = penv_87->en_fcs;
980         penv_xmm->en_opcode = penv_87->en_opcode;
981         penv_xmm->en_foo = penv_87->en_foo;
982         penv_xmm->en_fos = penv_87->en_fos;
983
984         /* FPU registers */
985         for (i = 0; i < 8; ++i)
986                 sv_xmm->sv_fp[i].fp_acc = sv_87->sv_ac[i];
987
988         sv_xmm->sv_ex_sw = sv_87->sv_ex_sw;
989 }
990 #endif /* CPU_DISABLE_SSE */
991
992 int
993 fill_fpregs(struct lwp *lp, struct fpreg *fpregs)
994 {
995 #ifndef CPU_DISABLE_SSE
996         if (cpu_fxsr) {
997                 fill_fpregs_xmm(&lp->lwp_thread->td_pcb->pcb_save.sv_xmm,
998                                 (struct save87 *)fpregs);
999                 return (0);
1000         }
1001 #endif /* CPU_DISABLE_SSE */
1002         bcopy(&lp->lwp_thread->td_pcb->pcb_save.sv_87, fpregs, sizeof *fpregs);
1003         return (0);
1004 }
1005
1006 int
1007 set_fpregs(struct lwp *lp, struct fpreg *fpregs)
1008 {
1009 #ifndef CPU_DISABLE_SSE
1010         if (cpu_fxsr) {
1011                 set_fpregs_xmm((struct save87 *)fpregs,
1012                                &lp->lwp_thread->td_pcb->pcb_save.sv_xmm);
1013                 return (0);
1014         }
1015 #endif /* CPU_DISABLE_SSE */
1016         bcopy(fpregs, &lp->lwp_thread->td_pcb->pcb_save.sv_87, sizeof *fpregs);
1017         return (0);
1018 }
1019
1020 int
1021 fill_dbregs(struct lwp *lp, struct dbreg *dbregs)
1022 {
1023         if (lp == NULL) {
1024                 dbregs->dr0 = rdr0();
1025                 dbregs->dr1 = rdr1();
1026                 dbregs->dr2 = rdr2();
1027                 dbregs->dr3 = rdr3();
1028                 dbregs->dr4 = rdr4();
1029                 dbregs->dr5 = rdr5();
1030                 dbregs->dr6 = rdr6();
1031                 dbregs->dr7 = rdr7();
1032         } else {
1033                 struct pcb *pcb;
1034
1035                 pcb = lp->lwp_thread->td_pcb;
1036                 dbregs->dr0 = pcb->pcb_dr0;
1037                 dbregs->dr1 = pcb->pcb_dr1;
1038                 dbregs->dr2 = pcb->pcb_dr2;
1039                 dbregs->dr3 = pcb->pcb_dr3;
1040                 dbregs->dr4 = 0;
1041                 dbregs->dr5 = 0;
1042                 dbregs->dr6 = pcb->pcb_dr6;
1043                 dbregs->dr7 = pcb->pcb_dr7;
1044         }
1045         return (0);
1046 }
1047
1048 int
1049 set_dbregs(struct lwp *lp, struct dbreg *dbregs)
1050 {
1051         if (lp == NULL) {
1052                 load_dr0(dbregs->dr0);
1053                 load_dr1(dbregs->dr1);
1054                 load_dr2(dbregs->dr2);
1055                 load_dr3(dbregs->dr3);
1056                 load_dr4(dbregs->dr4);
1057                 load_dr5(dbregs->dr5);
1058                 load_dr6(dbregs->dr6);
1059                 load_dr7(dbregs->dr7);
1060         } else {
1061                 struct pcb *pcb;
1062                 struct ucred *ucred;
1063                 int i;
1064                 uint32_t mask1, mask2;
1065
1066                 /*
1067                  * Don't let an illegal value for dr7 get set.  Specifically,
1068                  * check for undefined settings.  Setting these bit patterns
1069                  * result in undefined behaviour and can lead to an unexpected
1070                  * TRCTRAP.
1071                  */
1072                 for (i = 0, mask1 = 0x3<<16, mask2 = 0x2<<16; i < 8; 
1073                      i++, mask1 <<= 2, mask2 <<= 2)
1074                         if ((dbregs->dr7 & mask1) == mask2)
1075                                 return (EINVAL);
1076                 
1077                 pcb = lp->lwp_thread->td_pcb;
1078                 ucred = lp->lwp_proc->p_ucred;
1079
1080                 /*
1081                  * Don't let a process set a breakpoint that is not within the
1082                  * process's address space.  If a process could do this, it
1083                  * could halt the system by setting a breakpoint in the kernel
1084                  * (if ddb was enabled).  Thus, we need to check to make sure
1085                  * that no breakpoints are being enabled for addresses outside
1086                  * process's address space, unless, perhaps, we were called by
1087                  * uid 0.
1088                  *
1089                  * XXX - what about when the watched area of the user's
1090                  * address space is written into from within the kernel
1091                  * ... wouldn't that still cause a breakpoint to be generated
1092                  * from within kernel mode?
1093                  */
1094
1095                 if (suser_cred(ucred, 0) != 0) {
1096                         if (dbregs->dr7 & 0x3) {
1097                                 /* dr0 is enabled */
1098                                 if (dbregs->dr0 >= VM_MAX_USER_ADDRESS)
1099                                         return (EINVAL);
1100                         }
1101
1102                         if (dbregs->dr7 & (0x3<<2)) {
1103                                 /* dr1 is enabled */
1104                                 if (dbregs->dr1 >= VM_MAX_USER_ADDRESS)
1105                                         return (EINVAL);
1106                         }
1107
1108                         if (dbregs->dr7 & (0x3<<4)) {
1109                                 /* dr2 is enabled */
1110                                 if (dbregs->dr2 >= VM_MAX_USER_ADDRESS)
1111                                         return (EINVAL);
1112                         }
1113
1114                         if (dbregs->dr7 & (0x3<<6)) {
1115                                 /* dr3 is enabled */
1116                                 if (dbregs->dr3 >= VM_MAX_USER_ADDRESS)
1117                                         return (EINVAL);
1118                         }
1119                 }
1120
1121                 pcb->pcb_dr0 = dbregs->dr0;
1122                 pcb->pcb_dr1 = dbregs->dr1;
1123                 pcb->pcb_dr2 = dbregs->dr2;
1124                 pcb->pcb_dr3 = dbregs->dr3;
1125                 pcb->pcb_dr6 = dbregs->dr6;
1126                 pcb->pcb_dr7 = dbregs->dr7;
1127
1128                 pcb->pcb_flags |= PCB_DBREGS;
1129         }
1130
1131         return (0);
1132 }
1133
1134 #if 0
1135 /*
1136  * Return > 0 if a hardware breakpoint has been hit, and the
1137  * breakpoint was in user space.  Return 0, otherwise.
1138  */
1139 int
1140 user_dbreg_trap(void)
1141 {
1142         u_int32_t dr7, dr6; /* debug registers dr6 and dr7 */
1143         u_int32_t bp;       /* breakpoint bits extracted from dr6 */
1144         int nbp;            /* number of breakpoints that triggered */
1145         caddr_t addr[4];    /* breakpoint addresses */
1146         int i;
1147         
1148         dr7 = rdr7();
1149         if ((dr7 & 0x000000ff) == 0) {
1150                 /*
1151                  * all GE and LE bits in the dr7 register are zero,
1152                  * thus the trap couldn't have been caused by the
1153                  * hardware debug registers
1154                  */
1155                 return 0;
1156         }
1157
1158         nbp = 0;
1159         dr6 = rdr6();
1160         bp = dr6 & 0x0000000f;
1161
1162         if (!bp) {
1163                 /*
1164                  * None of the breakpoint bits are set meaning this
1165                  * trap was not caused by any of the debug registers
1166                  */
1167                 return 0;
1168         }
1169
1170         /*
1171          * at least one of the breakpoints were hit, check to see
1172          * which ones and if any of them are user space addresses
1173          */
1174
1175         if (bp & 0x01) {
1176                 addr[nbp++] = (caddr_t)rdr0();
1177         }
1178         if (bp & 0x02) {
1179                 addr[nbp++] = (caddr_t)rdr1();
1180         }
1181         if (bp & 0x04) {
1182                 addr[nbp++] = (caddr_t)rdr2();
1183         }
1184         if (bp & 0x08) {
1185                 addr[nbp++] = (caddr_t)rdr3();
1186         }
1187
1188         for (i=0; i<nbp; i++) {
1189                 if (addr[i] <
1190                     (caddr_t)VM_MAX_USER_ADDRESS) {
1191                         /*
1192                          * addr[i] is in user space
1193                          */
1194                         return nbp;
1195                 }
1196         }
1197
1198         /*
1199          * None of the breakpoints are in user space.
1200          */
1201         return 0;
1202 }
1203
1204 #endif
1205
1206
1207 #ifndef DDB
1208 void
1209 Debugger(const char *msg)
1210 {
1211         kprintf("Debugger(\"%s\") called.\n", msg);
1212 }
1213 #endif /* no DDB */
1214
1215 #include <sys/disklabel.h>
1216
1217 /*
1218  * Determine the size of the transfer, and make sure it is
1219  * within the boundaries of the partition. Adjust transfer
1220  * if needed, and signal errors or early completion.
1221  *
1222  * On success a new bio layer is pushed with the translated
1223  * block number, and returned.
1224  */
1225 struct bio *
1226 bounds_check_with_label(cdev_t dev, struct bio *bio,
1227                         struct disklabel *lp, int wlabel)
1228 {
1229         struct bio *nbio;
1230         struct buf *bp = bio->bio_buf;
1231         struct partition *p = lp->d_partitions + dkpart(dev);
1232         int labelsect = lp->d_partitions[0].p_offset;
1233         int maxsz = p->p_size,
1234                 sz = (bp->b_bcount + DEV_BSIZE - 1) >> DEV_BSHIFT;
1235         daddr_t blkno = (daddr_t)(bio->bio_offset >> DEV_BSHIFT);
1236
1237         /* overwriting disk label ? */
1238         /* XXX should also protect bootstrap in first 8K */
1239         if (blkno + p->p_offset <= LABELSECTOR + labelsect &&
1240 #if LABELSECTOR != 0
1241             blkno + p->p_offset + sz > LABELSECTOR + labelsect &&
1242 #endif
1243             bp->b_cmd != BUF_CMD_READ && wlabel == 0) {
1244                 bp->b_error = EROFS;
1245                 goto error;
1246         }
1247
1248 #if     defined(DOSBBSECTOR) && defined(notyet)
1249         /* overwriting master boot record? */
1250         if (blkno + p->p_offset <= DOSBBSECTOR &&
1251             bp->b_cmd != BUF_CMD_READ && wlabel == 0) {
1252                 bp->b_error = EROFS;
1253                 goto error;
1254         }
1255 #endif
1256
1257         /*
1258          * Check for out of bounds, EOF, and EOF clipping.
1259          */
1260         if (bio->bio_offset < 0)
1261                 goto bad;
1262         if (blkno + sz > maxsz) {
1263                 /*
1264                  * Past EOF or B_BNOCLIP flag was set, the request is bad.
1265                  */
1266                 if (blkno > maxsz || (bp->b_flags & B_BNOCLIP))
1267                         goto bad;
1268
1269                 /*
1270                  * If exactly on EOF just complete the I/O with no bytes
1271                  * transfered.  B_INVAL must be set to throw away the
1272                  * contents of the buffer.  Otherwise clip b_bcount.
1273                  */
1274                 if (blkno == maxsz) {
1275                         bp->b_resid = bp->b_bcount;
1276                         bp->b_flags |= B_INVAL;
1277                         goto done;
1278                 }
1279                 bp->b_bcount = (maxsz - blkno) << DEV_BSHIFT;
1280         }
1281         nbio = push_bio(bio);
1282         nbio->bio_offset = bio->bio_offset + ((off_t)p->p_offset << DEV_BSHIFT);
1283         return (nbio);
1284
1285         /*
1286          * The caller is responsible for calling biodone() on the passed bio
1287          * when we return NULL.
1288          */
1289 bad:
1290         bp->b_error = EINVAL;
1291 error:
1292         bp->b_resid = bp->b_bcount;
1293         bp->b_flags |= B_ERROR | B_INVAL;
1294 done:
1295         return (NULL);
1296 }
1297