xv6-65oo2/kernel/proc.c

655 lines
14 KiB
C
Raw Normal View History

2006-06-12 15:22:12 +00:00
#include "types.h"
2007-08-27 23:26:33 +00:00
#include "param.h"
#include "memlayout.h"
2019-05-31 13:45:59 +00:00
#include "riscv.h"
#include "spinlock.h"
#include "proc.h"
2019-05-31 13:45:59 +00:00
#include "defs.h"
2019-05-31 13:45:59 +00:00
struct cpu cpus[NCPU];
2019-07-10 12:57:51 +00:00
struct proc proc[NPROC];
2019-05-31 13:45:59 +00:00
struct proc *initproc;
2007-08-23 14:35:28 +00:00
int nextpid = 1;
2019-07-10 12:57:51 +00:00
struct spinlock pid_lock;
extern void forkret(void);
static void wakeup1(struct proc *chan);
extern char trampout[]; // trampoline.S
2019-05-31 13:45:59 +00:00
void
2019-05-31 13:45:59 +00:00
procinit(void)
{
struct proc *p;
initlock(&pid_lock, "nextpid");
for(p = proc; p < &proc[NPROC]; p++)
initlock(&p->lock, "proc");
}
2019-06-05 15:42:03 +00:00
// Must be called with interrupts disabled,
// to prevent race with process being moved
// to a different CPU.
int
2019-06-05 15:42:03 +00:00
cpuid()
{
int id = r_tp();
return id;
}
2019-05-31 13:45:59 +00:00
// Return this core's cpu struct.
2019-06-05 15:42:03 +00:00
// Interrupts must be disabled.
2018-10-10 00:22:48 +00:00
struct cpu*
mycpu(void) {
2019-06-05 15:42:03 +00:00
int id = cpuid();
struct cpu *c = &cpus[id];
2018-10-10 00:22:48 +00:00
return c;
}
2019-06-05 15:42:03 +00:00
// Return the current struct proc *.
struct proc*
myproc(void) {
2019-06-05 18:14:57 +00:00
push_off();
2019-06-05 15:42:03 +00:00
struct cpu *c = mycpu();
struct proc *p = c->proc;
2019-06-05 18:14:57 +00:00
pop_off();
2019-06-05 15:42:03 +00:00
return p;
}
int
allocpid() {
int pid;
acquire(&pid_lock);
2019-07-10 18:54:34 +00:00
pid = nextpid;
nextpid = nextpid + 1;
release(&pid_lock);
2019-07-10 18:54:34 +00:00
return pid;
}
2009-09-03 07:46:15 +00:00
//PAGEBREAK: 32
// Look in the process table for an UNUSED proc.
// If found, initialize state required to run in the kernel,
2019-07-07 18:57:16 +00:00
// and return with p->lock held.
2019-07-10 14:13:08 +00:00
// If there are no free procs, return 0.
static struct proc*
allocproc(void)
2006-06-12 15:22:12 +00:00
{
struct proc *p;
2006-06-12 15:22:12 +00:00
for(p = proc; p < &proc[NPROC]; p++) {
acquire(&p->lock);
if(p->state == UNUSED) {
goto found;
} else {
release(&p->lock);
}
}
return 0;
found:
p->pid = allocpid();
2019-05-31 13:45:59 +00:00
// Allocate a page for the kernel stack.
if((p->kstack = kalloc()) == 0){
2019-07-10 14:13:08 +00:00
release(&p->lock);
return 0;
}
2019-05-31 13:45:59 +00:00
// Allocate a trapframe page.
if((p->tf = (struct trapframe *)kalloc()) == 0){
kfree(p->kstack);
p->kstack = 0;
2019-07-10 14:13:08 +00:00
release(&p->lock);
2019-05-31 13:45:59 +00:00
return 0;
}
2019-05-31 13:45:59 +00:00
// An empty user page table.
p->pagetable = proc_pagetable(p);
// Set up new context to start executing at forkret,
// which returns to user space.
memset(&p->context, 0, sizeof p->context);
p->context.ra = (uint64)forkret;
p->context.sp = (uint64)p->kstack + PGSIZE;
return p;
}
// free a proc structure and the data hanging from it,
// including user pages.
2019-07-07 18:57:16 +00:00
// p->lock must be held.
static void
freeproc(struct proc *p)
{
if(p->kstack)
kfree(p->kstack);
p->kstack = 0;
if(p->tf)
kfree((void*)p->tf);
p->tf = 0;
if(p->pagetable)
proc_freepagetable(p->pagetable, p->sz);
p->pagetable = 0;
p->sz = 0;
p->pid = 0;
p->parent = 0;
p->name[0] = 0;
p->chan = 0;
p->killed = 0;
p->state = UNUSED;
}
// Create a page table for a given process,
2019-07-10 14:13:08 +00:00
// with no user pages, but with trampoline pages.
pagetable_t
proc_pagetable(struct proc *p)
{
pagetable_t pagetable;
2019-07-10 14:13:08 +00:00
// An empty page table.
pagetable = uvmcreate();
2019-05-31 13:45:59 +00:00
// map the trampoline code (for system call return)
// at the highest user virtual address.
// only the supervisor uses it, on the way
// to/from user space, so not PTE_U.
mappages(pagetable, TRAMPOLINE, PGSIZE,
(uint64)trampout, PTE_R | PTE_X);
2019-05-31 13:45:59 +00:00
// map the trapframe, for trampoline.S.
mappages(pagetable, (TRAMPOLINE - PGSIZE), PGSIZE,
2019-05-31 13:45:59 +00:00
(uint64)(p->tf), PTE_R | PTE_W);
2009-07-12 02:28:29 +00:00
return pagetable;
}
// Free a process's page table, and free the
2019-07-10 14:13:08 +00:00
// physical memory it refers to.
void
proc_freepagetable(pagetable_t pagetable, uint64 sz)
{
unmappages(pagetable, TRAMPOLINE, PGSIZE, 0);
unmappages(pagetable, TRAMPOLINE-PGSIZE, PGSIZE, 0);
if(sz > 0)
uvmfree(pagetable, sz);
2006-06-12 15:22:12 +00:00
}
2019-05-31 15:45:42 +00:00
// a user program that calls exec("/init")
// od -t xC initcode
2019-06-05 18:31:13 +00:00
uchar initcode[] = {
2019-07-10 14:13:08 +00:00
0x17, 0x05, 0x00, 0x00, 0x13, 0x05, 0x05, 0x02,
0x97, 0x05, 0x00, 0x00, 0x93, 0x85, 0x05, 0x02,
0x9d, 0x48, 0x73, 0x00, 0x00, 0x00, 0x89, 0x48,
0x73, 0x00, 0x00, 0x00, 0xef, 0xf0, 0xbf, 0xff,
0x2f, 0x69, 0x6e, 0x69, 0x74, 0x00, 0x00, 0x01,
0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
2019-05-31 15:45:42 +00:00
0x00, 0x00, 0x00
2019-05-31 13:45:59 +00:00
};
2009-09-03 07:46:15 +00:00
//PAGEBREAK: 32
// Set up first user process.
void
userinit(void)
{
struct proc *p;
p = allocproc();
initproc = p;
2019-05-31 13:45:59 +00:00
2019-07-10 14:13:08 +00:00
// allocate one user page and copy init's instructions
// and data into it.
2019-05-31 13:45:59 +00:00
uvminit(p->pagetable, initcode, sizeof(initcode));
p->sz = PGSIZE;
2019-05-31 13:45:59 +00:00
// prepare for the very first "return" from kernel to user.
2019-07-10 14:13:08 +00:00
p->tf->epc = 0; // user program counter
p->tf->sp = PGSIZE; // user stack pointer
safestrcpy(p->name, "initcode", sizeof(p->name));
2019-05-31 15:45:42 +00:00
p->cwd = namei("/");
p->state = RUNNABLE;
release(&p->lock);
}
2019-07-10 14:13:08 +00:00
// Grow or shrink user memory by n bytes.
// Return 0 on success, -1 on failure.
2006-09-08 14:26:51 +00:00
int
growproc(int n)
{
uint sz;
2019-05-31 13:45:59 +00:00
struct proc *p = myproc();
2019-05-31 13:45:59 +00:00
sz = p->sz;
if(n > 0){
if((sz = uvmalloc(p->pagetable, sz, sz + n)) == 0) {
return -1;
}
} else if(n < 0){
if((sz = uvmdealloc(p->pagetable, sz, sz + n)) == 0) {
return -1;
}
}
2019-05-31 13:45:59 +00:00
p->sz = sz;
return 0;
2006-09-08 14:26:51 +00:00
}
2019-07-10 14:13:08 +00:00
// Create a new process, copying the parent.
// Sets up child kernel stack to return as if from fork() system call.
2009-05-31 00:38:51 +00:00
int
fork(void)
2006-06-12 15:22:12 +00:00
{
2009-05-31 00:38:51 +00:00
int i, pid;
2006-06-12 15:22:12 +00:00
struct proc *np;
2019-05-31 13:45:59 +00:00
struct proc *p = myproc();
2006-06-12 15:22:12 +00:00
// Allocate process.
if((np = allocproc()) == 0){
2009-05-31 00:38:51 +00:00
return -1;
}
2019-05-31 13:45:59 +00:00
// Copy user memory from parent to child.
if(uvmcopy(p->pagetable, np->pagetable, p->sz) < 0){
freeproc(np);
2019-07-07 18:57:16 +00:00
release(&np->lock);
return -1;
}
2019-05-31 13:45:59 +00:00
np->sz = p->sz;
2006-09-06 17:27:19 +00:00
2019-05-31 13:45:59 +00:00
np->parent = p;
2019-05-31 13:45:59 +00:00
// copy saved user registers.
*(np->tf) = *(p->tf);
// Cause fork to return 0 in the child.
np->tf->a0 = 0;
// increment reference counts on open file descriptors.
2009-05-31 00:38:51 +00:00
for(i = 0; i < NOFILE; i++)
2019-05-31 13:45:59 +00:00
if(p->ofile[i])
np->ofile[i] = filedup(p->ofile[i]);
np->cwd = idup(p->cwd);
2019-05-31 13:45:59 +00:00
safestrcpy(np->name, p->name, sizeof(p->name));
2009-05-31 00:38:51 +00:00
pid = np->pid;
2009-05-31 00:38:51 +00:00
np->state = RUNNABLE;
release(&np->lock);
2009-05-31 00:38:51 +00:00
return pid;
2006-06-12 15:22:12 +00:00
}
2019-07-10 12:57:51 +00:00
// Pass p's abandoned children to init.
// Caller must hold p->lock and parent->lock.
2019-07-07 18:57:16 +00:00
void
reparent(struct proc *p, struct proc *parent) {
struct proc *pp;
2019-07-07 18:57:16 +00:00
int child_of_init = (p->parent == initproc);
for(pp = proc; pp < &proc[NPROC]; pp++){
if (pp != p && pp != parent) {
acquire(&pp->lock);
if(pp->parent == p){
pp->parent = initproc;
if(pp->state == ZOMBIE) {
2019-07-07 18:57:16 +00:00
if(!child_of_init)
acquire(&initproc->lock);
wakeup1(initproc);
2019-07-07 18:57:16 +00:00
if(!child_of_init)
release(&initproc->lock);
}
}
release(&pp->lock);
}
}
}
// Exit the current process. Does not return.
// An exited process remains in the zombie state
2019-05-31 13:45:59 +00:00
// until its parent calls wait().
void
exit(void)
{
2019-05-31 13:45:59 +00:00
struct proc *p = myproc();
int fd;
2019-05-31 13:45:59 +00:00
if(p == initproc)
panic("init exiting");
// Close all open files.
for(fd = 0; fd < NOFILE; fd++){
2019-05-31 13:45:59 +00:00
if(p->ofile[fd]){
struct file *f = p->ofile[fd];
fileclose(f);
2019-05-31 13:45:59 +00:00
p->ofile[fd] = 0;
}
}
2014-08-27 21:15:30 +00:00
begin_op();
iput(p->cwd);
2014-08-27 21:15:30 +00:00
end_op();
p->cwd = 0;
acquire(&p->parent->lock);
acquire(&p->lock);
2019-07-10 14:13:08 +00:00
// Give our children to init.
reparent(p, p->parent);
p->state = ZOMBIE;
// Parent might be sleeping in wait().
wakeup1(p->parent);
release(&p->parent->lock);
// Jump into the scheduler, never to return.
sched();
panic("zombie exit");
}
// Wait for a child process to exit and return its pid.
// Return -1 if this process has no children.
int
wait(void)
{
2019-05-31 13:45:59 +00:00
struct proc *np;
int havekids, pid;
2019-05-31 13:45:59 +00:00
struct proc *p = myproc();
2019-07-10 14:13:08 +00:00
// hold p->lock for the whole time to avoid lost
// wakeups from a child's exit().
acquire(&p->lock);
2019-07-10 14:13:08 +00:00
for(;;){
// Scan through table looking for exited children.
havekids = 0;
for(np = proc; np < &proc[NPROC]; np++){
// this code uses np->parent without holding np->lock.
// acquiring the lock first would cause a deadlock,
// since np might be an ancestor, and we already hold p->lock.
if(np->parent == p){
2019-07-10 14:13:08 +00:00
// np->parent can't change between the check and the acquire()
// because only the parent changes it, and we're the parent.
acquire(&np->lock);
havekids = 1;
if(np->state == ZOMBIE){
// Found one.
pid = np->pid;
freeproc(np);
release(&np->lock);
release(&p->lock);
return pid;
}
release(&np->lock);
}
}
// No point waiting if we don't have any children.
2019-05-31 13:45:59 +00:00
if(!havekids || p->killed){
release(&p->lock);
return -1;
}
2019-07-10 14:13:08 +00:00
// Wait for a child to exit.
sleep(p, &p->lock); //DOC: wait-sleep
}
}
2006-09-07 14:12:30 +00:00
//PAGEBREAK: 42
2006-09-06 17:27:19 +00:00
// Per-CPU process scheduler.
// Each CPU calls scheduler() after setting itself up.
// Scheduler never returns. It loops, doing:
// - choose a process to run.
// - swtch to start running that process.
2007-08-30 17:39:56 +00:00
// - eventually that process transfers control
// via swtch back to the scheduler.
2006-06-12 15:22:12 +00:00
void
2006-07-11 01:07:40 +00:00
scheduler(void)
2006-06-12 15:22:12 +00:00
{
struct proc *p;
struct cpu *c = mycpu();
c->proc = 0;
for(;;){
// Give devices a brief chance to interrupt.
2019-06-03 21:59:17 +00:00
intr_on();
for(p = proc; p < &proc[NPROC]; p++) {
acquire(&p->lock);
if(p->state == RUNNABLE) {
// Switch to chosen process. It is the process's job
// to release its lock and then reacquire it
// before jumping back to us.
p->state = RUNNING;
c->proc = p;
swtch(&c->scheduler, &p->context);
// Process is done running for now.
// It should have changed its p->state before coming back.
c->proc = 0;
}
2019-07-07 18:57:16 +00:00
release(&p->lock);
}
2006-06-12 15:22:12 +00:00
}
}
2006-06-12 15:22:12 +00:00
2019-07-10 14:13:08 +00:00
// Switch to scheduler. Must hold only p->lock
// and have changed proc->state. Saves and restores
// intena because intena is a property of this
// kernel thread, not this CPU. It should
// be proc->intena and proc->noff, but that would
// break in the few places where a lock is held but
// there's no process.
void
sched(void)
{
int intena;
struct proc *p = myproc();
if(!holding(&p->lock))
panic("sched p->lock");
2019-06-05 18:14:57 +00:00
if(mycpu()->noff != 1)
panic("sched locks");
if(p->state == RUNNING)
2009-07-12 02:28:29 +00:00
panic("sched running");
2019-06-05 18:14:57 +00:00
if(intr_get())
panic("sched interruptible");
intena = mycpu()->intena;
2019-05-31 13:45:59 +00:00
swtch(&p->context, &mycpu()->scheduler);
mycpu()->intena = intena;
2006-07-11 01:07:40 +00:00
}
// Give up the CPU for one scheduling round.
2006-07-11 01:07:40 +00:00
void
yield(void)
2006-07-11 01:07:40 +00:00
{
struct proc *p = myproc();
acquire(&p->lock); //DOC: yieldlock
p->state = RUNNABLE;
sched();
release(&p->lock);
2006-06-12 15:22:12 +00:00
}
2006-06-15 19:58:01 +00:00
2006-08-29 21:35:30 +00:00
// A fork child's very first scheduling by scheduler()
2019-05-31 13:45:59 +00:00
// will swtch to forkret.
void
forkret(void)
{
static int first = 1;
2019-06-05 18:31:13 +00:00
// Still holding p->lock from scheduler.
release(&myproc()->lock);
if (first) {
2011-08-23 00:07:18 +00:00
// Some initialization functions must be run in the context
// of a regular process (e.g., they call sleep), and thus cannot
2011-08-23 00:07:18 +00:00
// be run from main().
first = 0;
2019-05-31 15:45:42 +00:00
iinit(ROOTDEV);
initlog(ROOTDEV);
}
2019-05-31 13:45:59 +00:00
usertrapret();
}
// Atomically release lock and sleep on chan.
// Reacquires lock when awakened.
2006-06-15 19:58:01 +00:00
void
sleep(void *chan, struct spinlock *lk)
2006-06-15 19:58:01 +00:00
{
struct proc *p = myproc();
if(p == 0)
2006-07-11 01:07:40 +00:00
panic("sleep");
2006-07-17 05:00:25 +00:00
if(lk == 0)
panic("sleep without lk");
// Must acquire p->lock in order to
// change p->state and then call sched.
// Once we hold p->lock, we can be
// guaranteed that we won't miss any wakeup
// (wakeup locks p->lock),
// so it's okay to release lk.
if(lk != &p->lock){ //DOC: sleeplock0
acquire(&p->lock); //DOC: sleeplock1
release(lk);
}
2019-07-10 14:13:08 +00:00
// Go to sleep.
p->chan = chan;
p->state = SLEEPING;
sched();
// Tidy up.
p->chan = 0;
// Reacquire original lock.
if(lk != &p->lock){ //DOC: sleeplock2
release(&p->lock);
acquire(lk);
}
2006-06-15 19:58:01 +00:00
}
//PAGEBREAK!
2019-07-10 14:13:08 +00:00
// Wake up p, used by exit().
2019-07-10 12:57:51 +00:00
// Caller must hold p->lock.
2007-08-24 20:22:55 +00:00
static void
2019-07-03 19:38:30 +00:00
wakeup1(struct proc *p)
2006-06-15 19:58:01 +00:00
{
2019-07-03 19:38:30 +00:00
if(p->chan == p && p->state == SLEEPING) {
p->state = RUNNABLE;
}
}
2019-07-10 12:57:51 +00:00
// Wake up all processes sleeping on chan.
// Must be called without any p->lock.
void
wakeup(void *chan)
{
struct proc *p;
for(p = proc; p < &proc[NPROC]; p++) {
acquire(&p->lock);
if(p->state == SLEEPING && p->chan == chan) {
p->state = RUNNABLE;
}
release(&p->lock);
}
2006-06-15 19:58:01 +00:00
}
// Kill the process with the given pid.
2019-07-10 14:13:08 +00:00
// The victim won't exit until it tries to return
// to user space (see usertrap() in trap.c).
int
kill(int pid)
{
struct proc *p;
for(p = proc; p < &proc[NPROC]; p++){
acquire(&p->lock);
if(p->pid == pid){
p->killed = 1;
if(p->state == SLEEPING){
// Wake process from sleep().
p->state = RUNNABLE;
}
release(&p->lock);
return 0;
}
release(&p->lock);
}
return -1;
}
// Copy to either a user address, or kernel address,
// depending on usr_dst.
// Returns 0 on success, -1 on error.
int
2019-06-05 18:31:13 +00:00
either_copyout(int user_dst, uint64 dst, void *src, uint64 len)
{
struct proc *p = myproc();
if(user_dst){
return copyout(p->pagetable, dst, src, len);
} else {
memmove((char *)dst, src, len);
2019-06-05 18:31:13 +00:00
return 0;
}
}
// Copy from either a user address, or kernel address,
// depending on usr_src.
// Returns 0 on success, -1 on error.
int
2019-06-05 18:31:13 +00:00
either_copyin(void *dst, int user_src, uint64 src, uint64 len)
{
struct proc *p = myproc();
if(user_src){
return copyin(p->pagetable, dst, src, len);
} else {
memmove(dst, (char*)src, len);
2019-06-05 18:31:13 +00:00
return 0;
}
}
// Print a process listing to console. For debugging.
// Runs when user types ^P on console.
// No lock to avoid wedging a stuck machine further.
void
procdump(void)
{
static char *states[] = {
[UNUSED] "unused",
[SLEEPING] "sleep ",
[RUNNABLE] "runble",
[RUNNING] "run ",
[ZOMBIE] "zombie"
};
struct proc *p;
char *state;
for(p = proc; p < &proc[NPROC]; p++){
if(p->state == UNUSED)
continue;
if(p->state >= 0 && p->state < NELEM(states) && states[p->state])
state = states[p->state];
else
state = "???";
printf("%d %s %s", p->pid, state, p->name);
printf("\n");
}
}