Project 3: Virtual Memory - Part A
Due: Friday, April 12th by 11 pm
By now you should have some familiarity with the inner workings of Pintos. Your OS can properly handle multiple threads of execution with proper synchronization, and can load multiple user programs at once. However, the number and size of programs that can run is limited by the machine's main memory size. In this assignment, you will remove that limitation.
You will build this assignment on top of the last one. Test programs from project 2 should also work with project 3. You should take care to fix any bugs in your project 2 submission before you start work on project 3, because those bugs will most likely cause the same problems in project 3.
You will continue to handle Pintos disks and file systems the same way you did in the previous assignment (see section 4.1.2 Using the File System).
Background
B.1 Source Files
You will work in the “vm
” directory for this project. The “vm
”
directory contains only “Makefile
”s. The only change from “userprog
”
is that this new “Makefile
” turns on the setting “-DVM”.
All code you write will be in new files or in files introduced in
earlier projects.
You will probably be encountering just a few files for the first time:
“devices/block.h
”
- “
devices/block.c
” - Provides sector-based read and write access to block device. You will use this interface to access the swap partition as a block device.
B.2 Memory Terminology
Careful definitions are needed to keep discussion of virtual memory from being confusing. Thus, we begin by presenting some terminology for memory and storage. Some of these terms should be familiar from project 2 (see section 4.1.4 Virtual Memory Layout), but much of it is new.
B.2.1 Pages
A page, sometimes called a virtual page, is a continuous region of virtual memory 4,096 bytes (the page size) in length. A page must be page-aligned, that is, start on a virtual address evenly divisible by the page size. Thus, a 32-bit virtual address can be divided into a 20-bit page number and a 12-bit page offset (or just offset), like this:
31 12 11 0
+-------------------+-----------+
| Page Number | Offset |
+-------------------+-----------+
Virtual Address
Each process has an independent set of user (virtual) pages, which are
those pages below virtual address PHYS_BASE
, typically 0xc0000000
(3
GB). The set of kernel (virtual) pages, on the other hand, is global,
remaining the same regardless of what thread or process is active. The
kernel may access both user and kernel pages, but a user process may
access only its own user pages. See section 4.1.4 Virtual Memory
Layout, for more information.
Pintos provides several useful functions for working with virtual addresses. See section A.6 Virtual Addresses, for details.
B.2.2 Frames
A frame, sometimes called a physical frame or a page frame, is a continuous region of physical memory. Like pages, frames must be page-size and page-aligned. Thus, a 32-bit physical address can be divided into a 20-bit frame number and a 12-bit frame offset (or just offset), like this:
31 12 11 0
+-------------------+-----------+
| Frame Number | Offset |
+-------------------+-----------+
Physical Address
The 80x86 doesn't provide any way to directly access memory at a physical address. Pintos works around this by mapping kernel virtual memory directly to physical memory: the first page of kernel virtual memory is mapped to the first frame of physical memory, the second page to the second frame, and so on. Thus, frames can be accessed through kernel virtual memory.
Pintos provides functions for translating between physical addresses and kernel virtual addresses. See section A.6 Virtual Addresses, for details.
B.2.3 Page Tables
In Pintos, a page table is a data structure that the CPU uses to
translate a virtual address to a physical address, that is, from a page
to a frame. The page table format is dictated by the 80x86
architecture. Pintos provides page table management code in
“pagedir.c
” (see section A.7 Page Table).
The diagram below illustrates the relationship between pages and frames. The virtual address, on the left, consists of a page number and an offset. The page table translates the page number into a frame number, which is combined with the unmodified offset to obtain the physical address, on the right.
+----------+
.--------------->|Page Table|---------.
/ +----------+ |
31 | 12 11 0 31 V 12 11 0
+-----------+-------+ +------------+-------+
| Page Nr | Ofs | | Frame Nr | Ofs |
+-----------+-------+ +------------+-------+
Virt Addr | Phys Addr ^
\_____________________________________/
B.2.4 Swap Slots
A swap slot is a continuous, page-size region of disk space in the swap partition. Although hardware limitations dictating the placement of slots are looser than for pages and frames, swap slots should be page-aligned because there is no downside in doing so.
B.3 Resource Management Overview
You will need to design the following data structures:
Supplemental page table
Enables page fault handling by supplementing the hadrware page table. See section B.4 Managing the Supplemental Page Table.
Frame table
Allows efficient implementation of eviction policy. See section B.5 Managing the Frame Table.
Swap table
Tracks usage of swap slots. See section B.6 Managing the Swap Table.
You do not necessarily need to implement three completely distinct data structures: it may be convenient to wholly or partially merge related resources into a unified data structure.
For each data structure, you need to determine what information each element should contain. You also need to decide on the data structure's scope, either local (per-process) or global (applying to the whole system), and how many instances are required within its scope.
To simplify your design, you may store these data structures in non-pageable memory. That means that you can be sure that pointers among them will remain valid.
Possible choices of data structures include arrays, lists, bitmaps, and hash tables. An array is often the simplest approach, but a sparsely populated array wastes memory. Lists are also simple, but traversing a long list to find a particular position wastes time. Both arrays and lists can be resized, but lists more efficiently support insertion and deletion in the middle.
Pintos includes a bitmap data structure in “lib/kernel/bitmap.c
” and
“lib/kernel/bitmap.h
”. A bitmap is an array of bits, each of which can
be true or false. Bitmaps are typically used to track usage in a set of
(identical) resources: if resource n is in use, then bit
n of the bitmap is true. Pintos bitmaps are fixed in size,
although you could extend their implementation to support resizing.
Pintos also includes a hash table data structure (see section A.8 Hash Table). Pintos hash tables efficiently support insertions and deletions over a wide range of table sizes.
Although more complex data structures may yield performance or other benefits, they may also needlessly complicate your implementation. Thus, we do not recommend implementing any advanced data structure (e.g. a balanced binary tree) as part of your design.
B.4 Managing the Supplemental Page Table
The supplemental page table supplements the page table with additional data about each page. It is needed because of the limitations imposed by the page table's format. Such a data structure is often called a "page table" also; we add the word "supplemental" to reduce confusion.
The supplemental page table is used for at least two purposes. Most importantly, on a page fault, the kernel looks up the virtual page that faulted in the supplemental page table to find out what data should be there. Second, the kernel consults the supplemental page table when a process terminates, to decide what resources to free.
You may organize the supplemental page table as you wish. There are at
least two basic approaches to its organization: in terms of segments or
in terms of pages. Optionally, you may use the page table itself as an
index to track the members of the supplemental page table. You will have
to modify the Pintos page table implementation in “pagedir.c
” to do
so. We recommend this approach for advanced students only. See section
A.7.4.2 Page Table Entry Format, for more
information.
The most important user of the supplemental page table is the page fault
handler. In project 2, a page fault always indicated a bug in the kernel
or a user program. In project 3, this is no longer true. Now, a page
fault might only indicate that the page must be brought in from a file
or swap. You will have to implement a more sophisticated page fault
handler to handle these cases. Your page fault handler, which you should
implement by modifying page_fault()
in “userprog/exception.c
”, needs
to do roughly the following:
-
Locate the page that faulted in the supplemental page table. If the memory reference is valid, use the supplemental page table entry to locate the data that goes in the page, which might be in the file system, or in a swap slot, or it might simply be an all-zero page. If you implement sharing, the page's data might even already be in a page frame, but not in the page table.
If the supplemental page table indicates that the user process should not expect any data at the address it was trying to access, or if the page lies within kernel virtual memory, or if the access is an attempt to write to a read-only page, then the access is invalid. Any invalid access terminates the process and thereby frees all of its resources.
-
Obtain a frame to store the page. See section B.5 Managing the Frame Table, for details.
If you implement sharing, the data you need may already be in a frame, in which case you must be able to locate that frame.
-
Fetch the data into the frame, by reading it from the file system or swap, zeroing it, etc.
If you implement sharing, the page you need may already be in a frame, in which case no action is necessary in this step.
-
Point the page table entry for the faulting virtual address to the physical page. You can use the functions in “
userprog/pagedir.c
”.
B.5 Managing the Frame Table
The frame table contains one entry for each frame that contains a user page. Each entry in the frame table contains a pointer to the page, if any, that currently occupies it, and other data of your choice. The frame table allows Pintos to efficiently implement an eviction policy, by choosing a page to evict when no frames are free.
The frames used for user pages should be obtained from the "user
pool," by calling palloc_get_page(PAL_USER)
. You must use PAL_USER
to avoid allocating from the "kernel pool," which could cause some
test cases to fail unexpectedly (see Why PAL_USER?). If you modify “palloc.c
”
as part of your frame table implementation, be sure to retain the
distinction between the two pools.
The most important operation on the frame table is obtaining an unused frame. This is easy when a frame is free. When none is free, a frame must be made free by evicting some page from its frame.
If no frame can be evicted without allocating a swap slot, but swap is full, panic the kernel. Real OSes apply a wide range of policies to recover from or prevent such situations, but these policies are beyond the scope of this project.
The process of eviction comprises roughly the following steps:
-
Choose a frame to evict, using your page replacement algorithm. The "accessed" and "dirty" bits in the page table, described below, will come in handy.
-
Remove references to the frame from any page table that refers to it.
Unless you have implemented sharing, only a single page should refer to a frame at any given time.
-
If necessary, write the page to the file system or to swap.
The evicted frame may then be used to store a different page.
B.5.1 Accessed and Dirty Bits
80x86 hardware provides some assistance for implementing page replacement algorithms, through a pair of bits in the page table entry (PTE) for each page. On any read or write to a page, the CPU sets the accessed bit to 1 in the page's PTE, and on any write, the CPU sets the dirty bit to 1. The CPU never resets these bits to 0, but the OS may do so.
You need to be aware of aliases, that is, two (or more) pages that refer to the same frame. When an aliased frame is accessed, the accessed and dirty bits are updated in only one page table entry (the one for the page used for access). The accessed and dirty bits for the other aliases are not updated.
In Pintos, every user virtual page is aliased to its kernel virtual page. You must manage these aliases somehow. For example, your code could check and update the accessed and dirty bits for both addresses. Alternatively, the kernel could avoid the problem by only accessing user data through the user virtual address.
Other aliases should only arise if you implement sharing for extra credit (see VM Extra Credit), or if there is a bug in your code.
See section A.7.3 Accessed and Dirty Bits, for details of the functions to work with accessed and dirty bits.
B.6 Managing the Swap Table
The swap table tracks in-use and free swap slots. It should allow picking an unused swap slot for evicting a page from its frame to the swap partition. It should allow freeing a swap slot when its page is read back or the process whose page was swapped is terminated.
You may use the BLOCK_SWAP
block device for swapping, obtaining the
struct block
that represents it by calling block_get_role()
. From
the “vm/build
” directory, use the command
pintos-mkdisk swap.dsk --swap-size=
n to create an disk
named “swap.dsk
” that contains a n-MB swap partition.
Afterward, “swap.dsk
” will automatically be attached as an extra disk
when you run pintos
. Alternatively, you can tell pintos
to use a
temporary n-MB swap disk for a single run with
“–swap-size=n”.
Swap slots should be allocated lazily, that is, only when they are actually required by eviction. Reading data pages from the executable and writing them to swap immediately at process startup is not lazy. Swap slots should not be reserved to store particular pages.
Free a swap slot when its contents are read back into a frame.
Suggested Order of Implementation
We suggest the following initial order of implementation:
-
Frame table (see section B.5 Managing the Frame Table). Change “
process.c
” to use your frame table allocator.Do not implement swapping yet. If you run out of frames, fail the allocator or panic the kernel.
After this step, your kernel should still pass all the project 2 test cases.
-
Supplemental page table and page fault handler (see section B.4 Managing the Supplemental Page Table). Change “
process.c
” to record the necessary information in the supplemental page table when loading an executable and setting up its stack. Implement loading of code and data segments in the page fault handler. For now, consider only valid accesses.After this step, your kernel should pass all of the project 2 functionality test cases, but only some of the robustness tests.
From here, you can implement page reclamation on process exit.
The next step is to implement eviction (see section B.5 Managing the Frame Table). Initially you could choose the page to evict randomly. At this point, you need to consider how to manage accessed and dirty bits and aliasing of user and kernel pages. Synchronization is also a concern: how do you deal with it if process A faults on a page whose frame process B is in the process of evicting? Finally, implement a eviction strategy such as the clock algorithm.
Requirements
This assignment is an open-ended design problem. We are going to say as little as possible about how to do things. Instead we will focus on what functionality we require your OS to support. We will expect you to come up with a design that makes sense. You will have the freedom to choose how to handle page faults, how to organize the swap partition, how to implement paging, etc.
The test Grading and Rubric files are split into to two parts for this assignment. The stack growth and mmap related tests (listed in Rubric_part2.functionality and Rubric_part2.robustness) are for project 3b. Thus, you can ignore them in project 3a. To directly use the Grading file for this project 3a:
0. Design Document
Before you turn in your project, you must copy the project 3a design document template into your source tree under the name
“pintos/src/vm/PART1_DESIGNDOC
” and fill it in. We recommend that you
read the design document template before you start working on the
project. See section D. Project Documentation,
for a sample design document that goes along with a fictitious project.
1. Paging
Implement paging for segments loaded from executables. All of these pages should be loaded lazily, that is, only as the kernel intercepts page faults for them. Upon eviction, pages modified since load (e.g. as indicated by the "dirty bit") should be written to swap. Unmodified pages, including read-only pages, should never be written to swap because they can always be read back from the executable.
Implement a global page replacement algorithm that approximates LRU. Your algorithm should perform at least as well as the simple variant of the "second chance" or "clock" algorithm.
Your design should allow for parallelism. If one page fault requires I/O, in the meantime processes that do not fault should continue executing and other page faults that do not require I/O should be able to complete. This will require some synchronization effort.
You'll need to modify the core of the program loader, which is the loop
in load_segment()
in “userprog/process.c
”. Each time around the
loop, page_read_bytes
receives the number of bytes to read from the
executable file and page_zero_bytes
receives the number of bytes to
initialize to zero following the bytes read. The two always sum to
PGSIZE
(4,096). The handling of a page depends on these variables'
values:
- If
page_read_bytes
equalsPGSIZE
, the page should be demand paged from the underlying file on its first access. - If
page_zero_bytes
equalsPGSIZE
, the page does not need to be read from disk at all because it is all zeroes. You should handle such pages by creating a new page consisting of all zeroes at the first page fault. - Otherwise, neither
page_read_bytes
norpage_zero_bytes
equalsPGSIZE
. In this case, an initial part of the page is to be read from the underlying file and the remainder zeroed.
In order for demand paging to work, you need to record metadata for each
lazily-loaded page, which allows you to know what location to read its
content from disk later. In particular, if before demand paging a
page's content comes from reading offset X
of the executable file at
loading time, after demand paging, you should still read the content
from offset X
of the executable file during page fault handling.
The supplementary page table keeps track of relationship of memory pages
and their backing store locations. You should consider filling in the
supplementary page table in load_segment
.
If you would like to retain the previous file-reading code in
load_segment
, you can use macros like this to select the behavior of
load_segment
at compilation time:
static bool load_segment(...)
{
#ifndef VM
file_seek (file, ofs);
...
#else
... // fill in code for demand paging behavior in lab 3.
#endif
}
If you compile Pintos under lab 1 (threads
directory) or lab 2
(userprog
directory), the #ifndef VM
section will be selected. If
you compile Pintos under lab 3 or lab 4, the #else
section will be
selected.
You can use the “-ul” kernel command-line option to limit the
size of the user pool, which makes it easy to test your VM
implementation with various user memory sizes. For example,
pintos --swap-size=2 --filesys-size=2 -p ../../examples/echo -a echo -- -ul=4 -f -q run 'echo hello world'
will test Pintos with 4 page frames for user program.
Debugging in this lab can be challenging since the root cause (bug) can be far away from the symptom point, sometimes even many page faults away that make it difficult to track it down using backtrace.
So when you encounter some strange error (e.g., kernel panic due to dereferencing some invalid pointer), do not limit yourself to just inspect the failure code region (e.g., the invalid mem access per se). For example, the bug could be because of a one-off bug when you calculate the swap address, which much later cause a incorrect page content to be fetched in; the garbage content may be interpreted as a code page and the CPU will execute invalid instructions. Sometimes, it could be even caused by one boolean flag in the some page struct being incorrectly set (the course instructor was bitten by such a bug!).
One gdb
command that is particularly useful for this lab is
watchpoint
. Different from breakpoint
, which has to be set to a
particular code location, watchpoint
allows you to pause the execution
whenever the value of a specified expression changes without being bound
to a specific location. In other words, if there are N places that can
possibly change the value an expression, you will be able to track down
who made the change without setting breakpoints everywhere. See this
reference
for how to use watchpoint.
Besides using gdb to debug, we also suggest a "pair-programming" style debugging: explain the core code logic line by line to your teammate and pay special attention to logic such as calculating the address, setting flags/enum statuses, resetting offsets, locking, etc. It may take much shorter time to localize the bugs compared to directly debugging the symptom head-on.
2. Accessing User Memory
Adjust user memory access code in system call handling to deal with potential page faults.
You will need to adapt your code to access user memory (see section 3 Accessing User Memory in project 2) while handling a system call. Just as user processes may access pages whose content is currently in a file or in swap space, so can they pass addresses that refer to such non-resident pages to system calls. Moreover, unless your kernel takes measures to prevent this, a page may be evicted from its frame even while it is being accessed by kernel code. If kernel code accesses such non-resident user pages, a page fault will result.
While accessing user memory, your kernel must either be prepared to
handle such page faults, or it must prevent them from occurring. The
kernel must prevent such page faults while it is holding resources it
would need to acquire to handle these faults. In Pintos, such resources
include locks acquired by the device driver(s) that control the
device(s) containing the file system and swap space. As a concrete
example, you must not allow page faults to occur while a device driver
accesses a user buffer passed to file_read
, because you would not be
able to invoke the driver while handling such faults.
Preventing such page faults requires cooperation between the code within which the access occurs and your page eviction code. For instance, you could extend your frame table to record when a page contained in a frame must not be evicted. (This is also referred to as "pinning" or "locking" the page in its frame.) Pinning restricts your page replacement algorithm's choices when looking for pages to evict, so be sure to pin pages no longer than necessary, and avoid pinning pages when it is not necessary.
Submission Instructions
The submission process is essentially the same as Assignments 1 and 2:
- Download the submission script into the root of your development repository
- Run the script
- Enter the base commit (the hash of the last commit prior to starting work on this assignment)
- Enter the names of the team members
- Upload the resulting zipfile to Gradescope as “Assignment 3a”
Commands (where $REPODIR
is the path to the root directory of your development
repository):
cd $REPODIR
curl -O https://jhuopsys.github.io/spring2024/assign/create_submission_a3a.sh
chmod a+x create_submission_a3a.sh
./create_submission_a3a.sh
Make sure you are switched to the branch containing your work before running the submission script.
FAQ
- How much code will I need to write?
-
Here's a summary of our reference solution, produced by the
diffstat
program. The final row gives total lines inserted and deleted; a changed line counts as both an insertion and a deletion.
This summary is relative to the Pintos base code, but the reference solution for project 3 starts from the reference solution to project 2. See section 4.4 FAQ, for the summary of project 2.
The reference solution represents just one possible solution. Many other solutions are also possible and many of those differ greatly from the reference solution. Some excellent solutions may not modify all the files modified by the reference solution, and some may modify files not modified by the reference solution.
Makefile.build | 4
devices/timer.c | 42 ++
threads/init.c | 5
threads/interrupt.c | 2
threads/thread.c | 31 +
threads/thread.h | 37 +-
userprog/exception.c | 12
userprog/pagedir.c | 10
userprog/process.c | 319 +++++++++++++-----
userprog/syscall.c | 545 ++++++++++++++++++++++++++++++-
userprog/syscall.h | 1
vm/frame.c | 162 +++++++++
vm/frame.h | 23 +
vm/page.c | 297 ++++++++++++++++
vm/page.h | 50 ++
vm/swap.c | 85 ++++
vm/swap.h | 11
17 files changed, 1532 insertions(+), 104 deletions(-)
- Do we need a working Project 2 to implement Project 3?
-
Yes.
- What extra credit is available?
-
You may implement sharing: when multiple processes are created that use the same executable file, share read-only pages among those processes instead of creating separate copies of read-only segments for each process. If you carefully designed your data structures, sharing of read-only pages should not make this part significantly harder.
- How do we resume a process after we have handled a page fault?
-
Returning from
page_fault()
resumes the current user process (see section A.4.2 Internal Interrupt Handling). It will then retry the instruction to which the instruction pointer points. - Does the virtual memory system need to support data segment growth?
-
No. The size of the data segment is determined by the linker. We still have no dynamic allocation in Pintos (although it is possible to "fake" it at the user level by using memory-mapped files). Supporting data segment growth should add little additional complexity to a well-designed system.
- Why should I use
PAL_USER
for allocating page frames? -
Passing
PAL_USER
topalloc_get_page()
causes it to allocate memory from the user pool, instead of the main kernel pool. Running out of pages in the user pool just causes user programs to page, but running out of pages in the kernel pool will cause many failures because so many kernel functions need to obtain memory. You can layer some other allocator on top ofpalloc_get_page()
if you like, but it should be the underlying mechanism.