kernel SMP interruptibility fixes.

Last year, right before I sent xv6 to the printer, I changed the
SETGATE calls so that interrupts would be disabled on entry to
interrupt handlers, and I added the nlock++ / nlock-- in trap()
so that interrupts would stay disabled while the hw handlers
(but not the syscall handler) did their work.  I did this because
the kernel was otherwise causing Bochs to triple-fault in SMP
mode, and time was short.

Robert observed yesterday that something was keeping the SMP
preemption user test from working.  It turned out that when I
simplified the lapic code I swapped the order of two register
writes that I didn't realize were order dependent.  I fixed that
and then since I had everything paged in kept going and tried
to figure out why you can't leave interrupts on during interrupt
handlers.  There are a few issues.

First, there must be some way to keep interrupts from "stacking
up" and overflowing the stack.  Keeping interrupts off the whole
time solves this problem -- even if the clock tick handler runs
long enough that the next clock tick is waiting when it finishes,
keeping interrupts off means that the handler runs all the way
through the "iret" before the next handler begins.  This is not
really a problem unless you are putting too many prints in trap
-- if the OS is doing its job right, the handlers should run
quickly and not stack up.

Second, if xv6 had page faults, then it would be important to
keep interrupts disabled between the start of the interrupt and
the time that cr2 was read, to avoid a scenario like:

   p1 page faults [cr2 set to faulting address]
   p1 starts executing trapasm.S
   clock interrupt, p1 preempted, p2 starts executing
   p2 page faults [cr2 set to another faulting address]
   p2 starts, finishes fault handler
   p1 rescheduled, reads cr2, sees wrong fault address

Alternately p1 could be rescheduled on the other cpu, in which
case it would still see the wrong cr2.  That said, I think cr2
is the only interrupt state that isn't pushed onto the interrupt
stack atomically at fault time, and xv6 doesn't care.  (This isn't
entirely hypothetical -- I debugged this problem on Plan 9.)

Third, and this is the big one, it is not safe to call cpu()
unless interrupts are disabled.  If interrupts are enabled then
there is no guarantee that, between the time cpu() looks up the
cpu id and the time that it the result gets used, the process
has not been rescheduled to the other cpu.  For example, the
very commonly-used expression curproc[cpu()] (aka the macro cp)
can end up referring to the wrong proc: the code stores the
result of cpu() in %eax, gets rescheduled to the other cpu at
just the wrong instant, and then reads curproc[%eax].

We use curproc[cpu()] to get the current process a LOT.  In that
particular case, if we arranged for the current curproc entry
to be addressed by %fs:0 and just use a different %fs on each
CPU, then we could safely get at curproc even with interrupts
disabled, since the read of %fs would be atomic with the read
of %fs:0.  Alternately, we could have a curproc() function that
disables interrupts while computing curproc[cpu()].  I've done
that last one.

Even in the current kernel, with interrupts off on entry to trap,
interrupts are enabled inside release if there are no locks held.
Also, the scheduler's idle loop must be interruptible at times
so that the clock and disk interrupts (which might make processes
runnable) can be handled.

In addition to the rampant use of curproc[cpu()], this little
snippet from acquire is wrong on smp:

  if(cpus[cpu()].nlock == 0)
    cli();
  cpus[cpu()].nlock++;

because if interrupts are off then we might call cpu(), get
rescheduled to a different cpu, look at cpus[oldcpu].nlock, and
wrongly decide not to disable interrupts on the new cpu.  The
fix is to always call cli().  But this is wrong too:

  if(holding(lock))
    panic("acquire");
  cli();
  cpus[cpu()].nlock++;

because holding looks at cpu().  The fix is:

  cli();
  if(holding(lock))
    panic("acquire");
  cpus[cpu()].nlock++;

I've done that, and I changed cpu() to complain the first time
it gets called with interrupts disabled.  (It gets called too
much to complain every time.)

I added new functions splhi and spllo that are like acquire and
release but without the locking:

  void
  splhi(void)
  {
    cli();
    cpus[cpu()].nsplhi++;
  }

  void
  spllo(void)
  {
    if(--cpus[cpu()].nsplhi == 0)
      sti();
  }

and I've used those to protect other sections of code that refer
to cpu() when interrupts would otherwise be disabled (basically
just curproc and setupsegs).  I also use them in acquire/release
and got rid of nlock.

I'm not thrilled with the names, but I think the concept -- a
counted cli/sti -- is sound.  Having them also replaces the
nlock++/nlock-- in trap.c and main.c, which is nice.


Final note: it's still not safe to enable interrupts in
the middle of trap() between lapic_eoi and returning
to user space.  I don't understand why, but we get a
fault on pop %es because 0x10 is a bad segment
descriptor (!) and then the fault faults trying to go into
a new interrupt because 0x8 is a bad segment descriptor too!
Triple fault.  I haven't debugged this yet.
This commit is contained in:
rsc 2007-09-27 12:58:42 +00:00
parent 75506c6655
commit c8919e6537
10 changed files with 107 additions and 33 deletions

3
defs.h
View file

@ -92,6 +92,7 @@ int pipewrite(struct pipe*, char*, int);
// proc.c // proc.c
struct proc* copyproc(struct proc*); struct proc* copyproc(struct proc*);
struct proc* curproc();
void exit(void); void exit(void);
int growproc(int); int growproc(int);
int kill(int); int kill(int);
@ -114,6 +115,8 @@ void getcallerpcs(void*, uint*);
int holding(struct spinlock*); int holding(struct spinlock*);
void initlock(struct spinlock*, char*); void initlock(struct spinlock*, char*);
void release(struct spinlock*); void release(struct spinlock*);
void splhi();
void spllo();
// string.c // string.c
int memcmp(const void*, const void*, uint); int memcmp(const void*, const void*, uint);

View file

@ -107,7 +107,7 @@ romimage: file=$BXSHARE/BIOS-bochs-latest
# 650Mhz Athlon K-7 with Linux 2.4.4/egcs-2.91.66 2 to 2.5 Mips # 650Mhz Athlon K-7 with Linux 2.4.4/egcs-2.91.66 2 to 2.5 Mips
# 400Mhz Pentium II with Linux 2.0.36/egcs-1.0.3 1 to 1.8 Mips # 400Mhz Pentium II with Linux 2.0.36/egcs-1.0.3 1 to 1.8 Mips
#======================================================================= #=======================================================================
cpu: count=2, ips=10000000, quantum=1 cpu: count=2, ips=10000000, quantum=1, reset_on_triple_fault=0
#======================================================================= #=======================================================================
# MEGS # MEGS

22
lapic.c
View file

@ -2,7 +2,10 @@
// See Chapter 8 & Appendix C of Intel processor manual volume 3. // See Chapter 8 & Appendix C of Intel processor manual volume 3.
#include "types.h" #include "types.h"
#include "defs.h"
#include "traps.h" #include "traps.h"
#include "mmu.h"
#include "x86.h"
// Local APIC registers, divided by 4 for use as uint[] indices. // Local APIC registers, divided by 4 for use as uint[] indices.
#define ID (0x0020/4) // ID #define ID (0x0020/4) // ID
@ -84,6 +87,25 @@ lapic_init(int c)
int int
cpu(void) cpu(void)
{ {
// Cannot call cpu when interrupts are enabled:
// result not guaranteed to last long enough to be used!
// Would prefer to panic but even printing is chancy here:
// everything, including cprintf, calls cpu, at least indirectly
// through acquire and release.
if(read_eflags()&FL_IF){
static int n;
int i;
uint pcs[10];
if(n++%999 == 0){
getcallerpcs((uint*)read_ebp() + 2, pcs);
cprintf("cpu called from %x with interrupts enabled: stk");
for(i=0; i<10 && pcs[i] && pcs[i] != -1; i++)
cprintf(" %x", pcs[i]);
cprintf("\n");
}
}
if(lapic) if(lapic)
return lapic[ID]>>24; return lapic[ID]>>24;
return 0; return 0;

15
main.c
View file

@ -18,9 +18,9 @@ main(void)
// clear BSS // clear BSS
memset(edata, 0, end - edata); memset(edata, 0, end - edata);
// Prevent release() from enabling interrupts. // splhi() every processor during bootstrap.
for(i=0; i<NCPU; i++) for(i=0; i<NCPU; i++)
cpus[i].nlock = 1; cpus[i].nsplhi = 1;
mp_init(); // collect info about this machine mp_init(); // collect info about this machine
bcpu = mp_bcpu(); bcpu = mp_bcpu();
@ -47,12 +47,14 @@ main(void)
bootothers(); // boot other CPUs bootothers(); // boot other CPUs
if(!ismp) if(!ismp)
timer_init(); // uniprocessor timer timer_init(); // uniprocessor timer
cprintf("ismp %d\n", ismp);
cprintf("userinit\n");
userinit(); // first user process userinit(); // first user process
// enable interrupts on this processor. // enable interrupts on this processor.
cpus[cpu()].nlock--; spllo();
sti();
cprintf("scheduler\n");
scheduler(); scheduler();
} }
@ -66,10 +68,7 @@ mpmain(void)
setupsegs(0); setupsegs(0);
cpuid(0, 0, 0, 0, 0); // memory barrier cpuid(0, 0, 0, 0, 0); // memory barrier
cpus[cpu()].booted = 1; cpus[cpu()].booted = 1;
spllo();
// Enable interrupts on this processor.
cpus[cpu()].nlock--;
sti();
scheduler(); scheduler();
} }

31
proc.c
View file

@ -9,7 +9,6 @@
struct spinlock proc_table_lock; struct spinlock proc_table_lock;
struct proc proc[NPROC]; struct proc proc[NPROC];
struct proc *curproc[NCPU];
static struct proc *initproc; static struct proc *initproc;
int nextpid = 1; int nextpid = 1;
@ -61,6 +60,7 @@ growproc(int n)
cp->mem = newmem; cp->mem = newmem;
kfree(oldmem, cp->sz); kfree(oldmem, cp->sz);
cp->sz += n; cp->sz += n;
setupsegs(cp);
return cp->sz - n; return cp->sz - n;
} }
@ -71,6 +71,7 @@ setupsegs(struct proc *p)
{ {
struct cpu *c; struct cpu *c;
splhi();
c = &cpus[cpu()]; c = &cpus[cpu()];
c->ts.ss0 = SEG_KDATA << 3; c->ts.ss0 = SEG_KDATA << 3;
if(p) if(p)
@ -93,6 +94,7 @@ setupsegs(struct proc *p)
lgdt(c->gdt, sizeof(c->gdt)); lgdt(c->gdt, sizeof(c->gdt));
ltr(SEG_TSS << 3); ltr(SEG_TSS << 3);
spllo();
} }
// Create a new process copying p as the parent. // Create a new process copying p as the parent.
@ -176,6 +178,19 @@ userinit(void)
initproc = p; initproc = p;
} }
// Return currently running process.
// XXX comment better
struct proc*
curproc(void)
{
struct proc *p;
splhi();
p = cpus[cpu()].curproc;
spllo();
return p;
}
//PAGEBREAK: 42 //PAGEBREAK: 42
// Per-CPU process scheduler. // Per-CPU process scheduler.
// Each CPU calls scheduler() after setting itself up. // Each CPU calls scheduler() after setting itself up.
@ -188,12 +203,14 @@ void
scheduler(void) scheduler(void)
{ {
struct proc *p; struct proc *p;
struct cpu *c;
int i; int i;
for(;;){ for(;;){
// Loop over process table looking for process to run. // Loop over process table looking for process to run.
acquire(&proc_table_lock); acquire(&proc_table_lock);
c = &cpus[cpu()];
for(i = 0; i < NPROC; i++){ for(i = 0; i < NPROC; i++){
p = &proc[i]; p = &proc[i];
if(p->state != RUNNABLE) if(p->state != RUNNABLE)
@ -202,14 +219,14 @@ scheduler(void)
// Switch to chosen process. It is the process's job // Switch to chosen process. It is the process's job
// to release proc_table_lock and then reacquire it // to release proc_table_lock and then reacquire it
// before jumping back to us. // before jumping back to us.
cp = p; c->curproc = p;
setupsegs(p); setupsegs(p);
p->state = RUNNING; p->state = RUNNING;
swtch(&cpus[cpu()].context, &p->context); swtch(&c->context, &p->context);
// Process is done running for now. // Process is done running for now.
// It should have changed its p->state before coming back. // It should have changed its p->state before coming back.
cp = 0; c->curproc = 0;
setupsegs(0); setupsegs(0);
} }
@ -222,11 +239,13 @@ scheduler(void)
void void
sched(void) sched(void)
{ {
if(read_eflags()&FL_IF)
panic("sched interruptible");
if(cp->state == RUNNING) if(cp->state == RUNNING)
panic("sched running"); panic("sched running");
if(!holding(&proc_table_lock)) if(!holding(&proc_table_lock))
panic("sched proc_table_lock"); panic("sched proc_table_lock");
if(cpus[cpu()].nlock != 1) if(cpus[cpu()].nsplhi != 1)
panic("sched locks"); panic("sched locks");
swtch(&cp->context, &cpus[cpu()].context); swtch(&cp->context, &cpus[cpu()].context);

14
proc.h
View file

@ -49,25 +49,23 @@ struct proc {
// fixed-size stack // fixed-size stack
// expandable heap // expandable heap
// Arrange that cp point to the struct proc that this
// CPU is currently running. Such preprocessor
// subterfuge can be confusing, but saves a lot of typing.
extern struct proc *curproc[NCPU]; // Current (running) process per CPU
#define cp (curproc[cpu()]) // Current process on this CPU
#define MPSTACK 512 #define MPSTACK 512
// Per-CPU state // Per-CPU state
struct cpu { struct cpu {
uchar apicid; // Local APIC ID uchar apicid; // Local APIC ID
struct proc *curproc; // Process currently running.
struct context context; // Switch here to enter scheduler struct context context; // Switch here to enter scheduler
struct taskstate ts; // Used by x86 to find stack for interrupt struct taskstate ts; // Used by x86 to find stack for interrupt
struct segdesc gdt[NSEGS]; // x86 global descriptor table struct segdesc gdt[NSEGS]; // x86 global descriptor table
char mpstack[MPSTACK]; // Per-CPU startup stack char mpstack[MPSTACK]; // Per-CPU startup stack
volatile int booted; // Has the CPU started? volatile int booted; // Has the CPU started?
int nlock; // Number of locks currently held int nsplhi; // Depth of splhi nesting.
}; };
extern struct cpu cpus[NCPU]; extern struct cpu cpus[NCPU];
extern int ncpu; extern int ncpu;
// "cp" is a short alias for curproc().
// It gets used enough to make this worthwhile.
#define cp curproc()

View file

@ -25,13 +25,10 @@ initlock(struct spinlock *lock, char *name)
void void
acquire(struct spinlock *lock) acquire(struct spinlock *lock)
{ {
splhi();
if(holding(lock)) if(holding(lock))
panic("acquire"); panic("acquire");
if(cpus[cpu()].nlock == 0)
cli();
cpus[cpu()].nlock++;
while(cmpxchg(0, 1, &lock->locked) == 1) while(cmpxchg(0, 1, &lock->locked) == 1)
; ;
@ -62,8 +59,7 @@ release(struct spinlock *lock)
cpuid(0, 0, 0, 0, 0); // memory barrier (see Ch 7, IA-32 manual vol 3) cpuid(0, 0, 0, 0, 0); // memory barrier (see Ch 7, IA-32 manual vol 3)
lock->locked = 0; lock->locked = 0;
if(--cpus[cpu()].nlock == 0) spllo();
sti();
} }
// Record the current call stack in pcs[] by following the %ebp chain. // Record the current call stack in pcs[] by following the %ebp chain.
@ -91,3 +87,26 @@ holding(struct spinlock *lock)
return lock->locked && lock->cpu == cpu() + 10; return lock->locked && lock->cpu == cpu() + 10;
} }
// XXX!
// Better names? Better functions?
void
splhi(void)
{
cli();
cpus[cpu()].nsplhi++;
}
void
spllo(void)
{
if(read_eflags()&FL_IF)
panic("spllo - interruptible");
if(--cpus[cpu()].nsplhi < 0)
panic("spllo");
if(cpus[cpu()].nsplhi == 0)
sti();
}

View file

@ -54,7 +54,6 @@ sys_sbrk(void)
return -1; return -1;
if((addr = growproc(n)) < 0) if((addr = growproc(n)) < 0)
return -1; return -1;
setupsegs(cp);
return addr; return addr;
} }

12
trap.c
View file

@ -44,8 +44,8 @@ trap(struct trapframe *tf)
return; return;
} }
// Make sure interrupts stay off during handler. // No interrupts during interrupt handling.
cpus[cpu()].nlock++; splhi();
switch(tf->trapno){ switch(tf->trapno){
case IRQ_OFFSET + IRQ_TIMER: case IRQ_OFFSET + IRQ_TIMER:
@ -84,7 +84,13 @@ trap(struct trapframe *tf)
cp->killed = 1; cp->killed = 1;
} }
cpus[cpu()].nlock--; // Undo splhi but do not enable interrupts.
// If you change this to spllo() you can get a
// triple fault by just typing too fast at the prompt.
// An interrupt stops us right here, and when that
// interrupt tries to return, somehow the segment
// registers are all invalid.
--cpus[cpu()].nsplhi;
// Force process exit if it has been killed and is in user space. // Force process exit if it has been killed and is in user space.
// (If it is still executing in the kernel, let it keep running // (If it is still executing in the kernel, let it keep running

9
x86.h
View file

@ -39,6 +39,15 @@ outsl(int port, const void *addr, int cnt)
"cc"); "cc");
} }
static inline uint
read_ebp(void)
{
uint ebp;
asm volatile("movl %%ebp, %0" : "=a" (ebp));
return ebp;
}
struct segdesc; struct segdesc;
static inline void static inline void