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 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 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 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 80 unsigned long get_max_memory(void) 81 { 82 return max_size; 83 } 84 85 void register_mem_tracker(int id) 86 { 87 my_id = id; 88 89 add_hook(&match_end_func, END_FUNC_HOOK); 90 }