11 static struct maps_list *alloc_maplist(void)
13 struct maps_list *map;
15 map = malloc(sizeof *map);
19 memset(map, 0, sizeof(*map));
20 INIT_LIST_HEAD(&map->list);
25 static struct maps *alloc_map(void)
29 map = malloc(sizeof *map);
33 memset(map, 0, sizeof(*map));
34 INIT_LIST_HEAD(&map->list);
39 static struct maps *parse_maps(FILE *file, int pid)
41 struct maps *the_map = NULL;
45 while(fgets(line, sizeof(line), file)) {
46 struct maps *map = alloc_map();
47 unsigned long start, end;
56 ret = sscanf(line, "%lx-%lx %*s %*s %*s %*s %s",
60 printf("Error reading input: %s\n", line);
66 map->size = end - start;
70 strncpy(map->name, name, sizeof(map->name));
72 list_add_tail(&map->list, &the_map->list);
78 static void clear_pageframe(struct pageframe *pf)
80 memset(pf, 0, sizeof(*pf));
83 static struct pageframe *alloc_pageframe(void)
85 struct pageframe *pageframe;
87 pageframe = malloc(sizeof *pageframe);
88 if (pageframe == NULL)
91 clear_pageframe(pageframe);
96 #define BITRANGE(first, last) (((2ll << (last - first)) - 1) << first)
98 static void pageframe_to_struct(unsigned long long p, struct pageframe *pf)
100 pf->page_present = !!(BITRANGE(63, 63) & p);
101 pf->page_swapped = !!(BITRANGE(62, 62) & p);
102 pf->page_shift = (BITRANGE(55, 60) & p) >> 55;
103 pf->pfn = (BITRANGE(0, 54) & p);
104 pf->swap_type = (BITRANGE(0, 4) & p);
105 pf->swap_offset = (BITRANGE(5, 54) & p) >> 5;
107 printf("pfn: %lx shift: %d present: %d swapped %d\n",
108 pf->pfn, pf->page_shift, pf->page_present, pf->page_swapped);
112 static int compare_pageframe(struct bintree *at, struct bintree *bt)
114 struct pageframe *a, *b;
115 a = tree_to_pageframe(at);
116 b = tree_to_pageframe(bt);
118 return a->pfn - b->pfn;
121 struct bintree_ops pageframe_ops = {
122 .compare = compare_pageframe,
125 /* Read data from the /proc/pid/pagemap file */
126 static int parse_pageframe(FILE *file, struct pageframe *pf_tree,
127 struct maps *maps, int add_to_tree)
130 struct maps_list *tmp;
131 struct pageframe *match, *pageframe = NULL;
133 unsigned long long pf[10240];
139 /* Go through the list of allocated memory areas */
140 list_for_each_entry(map, &maps->list, list) {
141 start = map->start >> (PAGE_SHIFT - 3);
142 len = map->size >> (PAGE_SHIFT - 3);
144 ret = fseek(file, start, SEEK_SET);
147 fprintf(stderr, "Error seeking to %lx: %s\n", start,
152 for (i = 0; i < len; i++) {
155 MIN(sizeof(pf), len - i), file);
161 pageframe = alloc_pageframe();
162 ret -= sizeof(pf[0]);
163 pageframe_to_struct(pf[ret / sizeof(pf[0])], pageframe);
166 match = tree_to_pageframe(
167 bintree_add(&pf_tree->tree,
171 match = tree_to_pageframe(
172 bintree_find(&pf_tree->tree,
180 if (match == pageframe)
185 * Add a link from the physical page to this
189 match->ml = alloc_maplist();
190 match->ml->map = map;
192 tmp = alloc_maplist();
194 list_add(&match->ml->list, &tmp->list);
197 if (match->page_present) {
198 map->pages_present++;
199 } else if (match->page_swapped) {
200 map->pages_swapped++;
208 void read_pageframe(int pid, struct pageframe *pageframe,
209 struct process **process_list, int add_to_tree)
212 struct process *process;
217 process = malloc(sizeof(*process));
218 memset(process, 0, sizeof(*process));
219 INIT_LIST_HEAD(&process->list);
221 if (*process_list == NULL)
222 *process_list = process;
226 list_add_tail(&process->list, &(*process_list)->list);
228 snprintf(path, sizeof(path), "/proc/%d/maps", pid);
229 file = fopen(path, "rb");
234 maps = parse_maps(file, pid);
236 process->maps = maps;
238 snprintf(path, sizeof(path), "/proc/%d/pagemap", pid);
239 file = fopen(path, "rb");
244 parse_pageframe(file, pageframe, maps, add_to_tree);
247 snprintf(path, sizeof(path), "/proc/%d/cmdline", pid);
248 file = fopen(path, "rb");
253 ret = fread(process->name, 1, sizeof(process->name), file);
255 process->name[ret - 1] = 0;
261 list_for_each_entry(maps, &process->maps->list, list) {
262 process->pages_present += maps->pages_present;
263 process->pages_swapped += maps->pages_swapped;
269 static int get_next_pid(void)
271 static DIR *dir = NULL;
272 struct dirent *dirent;
276 dir = opendir("/proc");
279 printf("Failed to open /proc directory: %s\n",
286 dirent = readdir(dir);
294 printf("Failed to read /proc directory: %s\n", strerror(error));
298 printf("%s\n", dirent->d_name);
299 if (dirent->d_name[0] < '0' || dirent->d_name[0] > '9')
302 return atoi(dirent->d_name);
305 void scan_all_pids(struct pageframe *pf, struct process **process_list,
310 read_pageframe(interesting_pid, pf, process_list, 1);
313 pid = get_next_pid();
316 read_pageframe(pid, pf, process_list, 0);