xref: /illumos-gate/usr/src/tools/smatch/src/smatch_mem_tracker.c (revision c85f09cc92abd00c84e58ec9f0f5d942906cb713)
1 /*
2  * Copyright (C) 2018 Oracle.
3  * Copyright 2019 Joyent, Inc.
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version 2
8  * of the License, or (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
17  */
18 
19 #include "smatch.h"
20 #include <fcntl.h>
21 #include <unistd.h>
22 #ifdef __sun
23 #include <sys/procfs.h>
24 #endif
25 
26 static int my_id;
27 
28 static unsigned long max_size;
29 
30 #ifdef __sun
get_mem_kb(void)31 unsigned long get_mem_kb(void)
32 {
33 	static int my_fd = -2;
34 	prpsinfo_t pbuf;
35 
36 	if (my_fd == -2) {
37 		/* Do not repeatedly attempt this if it fails. */
38 		my_fd = open("/proc/self/psinfo", O_RDONLY);
39 	}
40 	if (my_fd == -1) {
41 		return (0);
42 	}
43 
44 	if (pread(my_fd, &pbuf, sizeof (pbuf), 0) != sizeof (pbuf)) {
45 		return (0);
46 	}
47 
48 	return (pbuf.pr_rssize);
49 }
50 #else
get_mem_kb(void)51 unsigned long get_mem_kb(void)
52 {
53 	FILE *file;
54 	char buf[1024] = "0";
55 	unsigned long size;
56 
57 	file = fopen("/proc/self/statm", "r");
58 	if (!file)
59 	        return 0;
60 	fread(buf, 1, sizeof(buf), file);
61 	fclose(file);
62 
63 	size = strtoul(buf, NULL, 10);
64 	size = size * sysconf(_SC_PAGESIZE) / 1024;
65 	return size;
66 }
67 #endif
68 
match_end_func(struct symbol * sym)69 static void match_end_func(struct symbol *sym)
70 {
71 	unsigned long size;
72 
73 	if (option_mem) {
74 		size = get_mem_kb();
75 		if (size > max_size)
76 			max_size = size;
77 	}
78 }
79 
get_max_memory(void)80 unsigned long get_max_memory(void)
81 {
82 	return max_size;
83 }
84 
register_mem_tracker(int id)85 void register_mem_tracker(int id)
86 {
87 	my_id = id;
88 
89 	add_hook(&match_end_func, END_FUNC_HOOK);
90 }
91