205 lines
2.7 KiB
C
205 lines
2.7 KiB
C
#include "types.h"
|
|
#include "riscv.h"
|
|
#include "param.h"
|
|
#include "defs.h"
|
|
#include "memlayout.h"
|
|
#include "spinlock.h"
|
|
#include "proc.h"
|
|
|
|
uint64
|
|
sys_exit(void)
|
|
{
|
|
int n;
|
|
argint(0, &n);
|
|
exit(n);
|
|
return 0; // not reached
|
|
}
|
|
|
|
uint64
|
|
sys_getpid(void)
|
|
{
|
|
return myproc()->pid;
|
|
}
|
|
|
|
uint64
|
|
sys_fork(void)
|
|
{
|
|
return fork();
|
|
}
|
|
|
|
uint64
|
|
sys_wait(void)
|
|
{
|
|
uint64 p;
|
|
argaddr(0, &p);
|
|
return wait(p);
|
|
}
|
|
|
|
uint64
|
|
sys_sbrk(void)
|
|
{
|
|
uint64 addr;
|
|
int n;
|
|
|
|
argint(0, &n);
|
|
addr = myproc()->sz;
|
|
if(growproc(n) < 0)
|
|
return -1;
|
|
return addr;
|
|
}
|
|
|
|
uint64
|
|
sys_sleep(void)
|
|
{
|
|
int n;
|
|
uint ticks0;
|
|
|
|
|
|
argint(0, &n);
|
|
if(n < 0)
|
|
n = 0;
|
|
acquire(&tickslock);
|
|
ticks0 = ticks;
|
|
while(ticks - ticks0 < n){
|
|
if(killed(myproc())){
|
|
release(&tickslock);
|
|
return -1;
|
|
}
|
|
sleep(&ticks, &tickslock);
|
|
}
|
|
release(&tickslock);
|
|
return 0;
|
|
}
|
|
|
|
|
|
#ifdef LAB_PGTBL
|
|
int
|
|
sys_pgpte(void)
|
|
{
|
|
uint64 va;
|
|
struct proc *p;
|
|
|
|
p = myproc();
|
|
argaddr(0, &va);
|
|
pte_t *pte = pgpte(p->pagetable, va);
|
|
if(pte != 0) {
|
|
return (uint64) *pte;
|
|
}
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
#ifdef LAB_PGTBL
|
|
int
|
|
sys_kpgtbl(void)
|
|
{
|
|
struct proc *p;
|
|
|
|
p = myproc();
|
|
vmprint(p->pagetable);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
|
|
uint64
|
|
sys_kill(void)
|
|
{
|
|
int pid;
|
|
|
|
argint(0, &pid);
|
|
return kill(pid);
|
|
}
|
|
|
|
// return how many clock tick interrupts have occurred
|
|
// since start.
|
|
uint64
|
|
sys_uptime(void)
|
|
{
|
|
uint xticks;
|
|
|
|
acquire(&tickslock);
|
|
xticks = ticks;
|
|
release(&tickslock);
|
|
return xticks;
|
|
}
|
|
|
|
uint64
|
|
sys_pgaccess(void)
|
|
{
|
|
// lab pgtbl: your code here.
|
|
struct proc *p = myproc();
|
|
unsigned int abits=0;
|
|
|
|
uint64 addr;
|
|
argaddr(0, &addr);
|
|
|
|
int num;
|
|
argint(1,&num);
|
|
|
|
uint64 dest;
|
|
argaddr(2, &dest);
|
|
|
|
|
|
for(int i=0;i<num;i++){
|
|
uint64 query_addr = addr + i * PGSIZE ;
|
|
|
|
|
|
pte_t * pte=walk(p->pagetable, query_addr, 0);
|
|
if(*pte&PTE_A)
|
|
{
|
|
abits=abits|(1<<i);
|
|
*pte=(*pte)&(~PTE_A);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if(copyout(p->pagetable,dest,(char*)&abits, sizeof(abits)) < 0)
|
|
return -1;
|
|
|
|
|
|
|
|
return 0;
|
|
}
|
|
|
|
#ifdef LAB_PGTBL
|
|
uint64
|
|
sys_dirtypages(void)
|
|
{
|
|
struct proc *p = myproc();
|
|
unsigned int dbits = 0;
|
|
|
|
uint64 addr;
|
|
argaddr(0, &addr);
|
|
|
|
int num;
|
|
argint(1, &num);
|
|
|
|
uint64 dest;
|
|
argaddr(2, &dest);
|
|
|
|
// Check each page in the range
|
|
for(int i = 0; i < num; i++){
|
|
uint64 query_addr = addr + i * PGSIZE;
|
|
|
|
pte_t *pte = walk(p->pagetable, query_addr, 0);
|
|
if(pte == 0)
|
|
continue; // Skip pages that don't exist
|
|
|
|
if(*pte & PTE_D) {
|
|
dbits = dbits | (1 << i);
|
|
// Clear the dirty bit after reading it
|
|
*pte = (*pte) & (~PTE_D);
|
|
}
|
|
}
|
|
|
|
// Copy the result back to user space
|
|
if(copyout(p->pagetable, dest, (char*)&dbits, sizeof(dbits)) < 0)
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
#endif
|