#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