$//$

$//$ Created by chang on $10/15/23.$

$//$

#include

#include

#include

$//$ TODO $1$: Define your input parameter $($pid $-$ int$)$ here

$//$ Then use module$\_$param to pass them from insmod command line. $(--$Assignment $2)$

struct task$\_$struct $*$task $=$ NULL;

$//$ Initialize memory statistics variables

unsigned long total$\_$rss $=0$;

unsigned long total$\_$swap $=0$;

unsigned long total$\_$wss $=0$;

static void parse$\_$vma$($void$)$

$\{$

struct vm$\_$area$\_$struct $*$vma $=$ NULL;

struct mm$\_$struct $*$mm $=$ NULL;

if$($pid $>0)\{$

task $=$ pid$\_$task$($find$\_$vpid$($pid$),$ PIDTYPE$\_$PID$)$;

if$($task && task$->$mm$)\{$

mm $=$ task$->$mm;

$//$ TODO $2$: mm$\_$struct to initialize the VMA$\_$ITERATOR $(--$ Assignment $4)$

$//$ Hint: Only one line of code is needed

$//$ TODO $3$: Iterate through the VMA linked list with for$\_$each$\_$vma $(--$ Assignment $4)$

$//$ Hint: Only one line of code is needed

$//$ TODO $4$: Iterate through each page of the VMA

$//$ declear "page" as unsigned long, start from vma$->$vm$\_$start to vma$->$vm$\_$end with PAGE$\_$SIZE step

$//$ Hint: Only one line of code is needed

$//$ TODO $5$: Use pgd$\_$offset, p$4$d$\_$offset, pud$\_$offset, pmd$\_$offset, pte$\_$offset$\_$map to get the page table entry

$//$ Hint: Copy from Background Knowledge in the instruction

$//$ Hint: change the address in the instruction to "page" variable you defined above

$//$ TODO $6$: use pte$\_$none$($pte$)$ to check if the page table entry is valid

$//$ Hint: one if statement here

$//$ TODO $7$: use pte$\_$present$($pte$)$ to check if the page is in memory, otherwise it is in swap

$//$ TODO $8$: use pte$\_$young$($pte$)$ to check if the page is actively used

$//$ if it is actively used, update wss and clear the accessed bit by: "test$\_$and$\_$clear$\_$bit$(\_$PAGE$\_$BIT$\_$ACCESSED,$($unsigned long $*)$ppte$)$;$"$

$\}$

$\}$

$\}$

$\}$

$\}$

unsigned long timer$\_$interval$\_$ns $=10$e$9$; $//10$ sec timer

static struct hrtimer hr$\_$timer;

enum hrtimer$\_$restart timer$\_$callback$($ struct hrtimer $*$timer$\_$for$\_$restart $)$

$\{$

ktime$\_$t currtime $,$ interval;

currtime $=$ ktime$\_$get$()$;

interval $=$ ktime$\_$set$(0,$timer$\_$interval$\_$ns$)$;

hrtimer$\_$forward$($timer$\_$for$\_$restart, currtime $,$ interval$)$;

total$\_$rss $=0$;

total$\_$swap $=0$;

total$\_$wss $=0$;

parse$\_$vma$()$;

printk$("[$PID$-\%$i$]$:$[$RSS:$\%$lu MB$][$Swap:$\%$lu MB$][$WSS:$\%$lu MB$]$

$",$ pid, total$\_$rss$*4/1024,$ total$\_$swap$*4/1024,$ total$\_$wss$*4/1024)$;

return HRTIMER$\_$RESTART;

$\}$

int memory$\_$init$($void$)\{$

printk$("$CSE$330$ Project $2$ Kernel Module Inserted

$")$;

ktime$\_$t ktime;

ktime $=$ ktime$\_$set$(0,$ timer$\_$interval$\_$ns $)$;

hrtimer$\_$init$($ &hr$\_$timer, CLOCK$\_$MONOTONIC, HRTIMER$\_$MODE$\_$REL $)$;

hr$\_$timer.function $=$ &timer$\_$callback;

hrtimer$\_$start$($ &hr$\_$timer, ktime, HRTIMER$\_$MODE$\_$REL $)$;

return $0$;

$\}$

void memory$\_$cleanup$($void$)\{$

int ret;

ret $=$ hrtimer$\_$cancel$($ &hr$\_$timer $)$;

if $($ret$)$ printk$("$HR Timer cancelled $...$

$")$;

printk$("$CSE$330$ Project $2$ Kernel Module Removed

$")$;

$\}$

module$\_$init$($memory$\_$init$)$;

module$\_$exit$($memory$\_$cleanup$)$;

MODULE$\_$VERSION$("0\mathrm{.}1")$;

MODULE$\_$LICENSE$("$GPL$")$;

MODULE$\_$AUTHOR$("$Your Name"$)$;

MODULE$\_$DESCRIPTION$("$CSE$330$ Project $2$ Memory Management

$")$;