epan/app_mem_usage.c

   1 /*
   2  * app_mem_usage.c
   3  *
   4  * Wireshark - Network traffic analyzer
   5  * By Gerald Combs <gerald@wireshark.org>
   6  * Copyright 1998 Gerald Combs
   7  *
   8  * SPDX-License-Identifier: GPL-2.0-or-later
   9  */
  10
  11 #if defined(__linux__)
  12  #define _XOPEN_SOURCE 500
  13 #endif
  14
  15 #include "config.h"
  16
  17 #include <stdio.h>
  18
  19 #ifdef _WIN32
  20 #include <windows.h>
  21 #include <psapi.h>
  22 #endif /*  _WIN32 */
  23
  24 #if defined(__linux__)
  25 # include <sys/types.h>
  26 # include <sys/stat.h>
  27 # include <unistd.h>
  28 # include <fcntl.h>
  29 #endif
  30
  31 #include "wsutil/file_util.h"
  32 #include "app_mem_usage.h"
  33
  34 #define MAX_COMPONENTS 16
  35
  36 #if defined(_WIN32)
  37
  38 static size_t
  39 win32_get_total_mem_used_by_app(void)
  40 {
  41         HANDLE pHandle;
  42         PROCESS_MEMORY_COUNTERS pmc;
  43         SIZE_T workingSize = 0;
  44
  45         pHandle = GetCurrentProcess();
  46
  47         if (GetProcessMemoryInfo(pHandle, &pmc, sizeof(pmc))){
  48                 workingSize = pmc.WorkingSetSize;
  49         }
  50
  51         CloseHandle(pHandle);
  52
  53         if(workingSize == 0){
  54                 return -1;
  55         }else{
  56                 return (int)workingSize;
  57         }
  58 }
  59
  60 static const ws_mem_usage_t total_usage = { "Total", win32_get_total_mem_used_by_app, NULL };
  61
  62 static const ws_mem_usage_t *memory_components[MAX_COMPONENTS] = {
  63         &total_usage,
  64 };
  65
  66 static unsigned memory_register_num = 1;
  67
  68 #elif defined(__linux__)
  69
  70 static bool
  71 linux_get_memory(size_t *ptotal, size_t *prss)
  72 {
  73         static int fd = -1;
  74         static intptr_t pagesize = 0;
  75
  76         char buf[128];
  77         unsigned long total, rss;
  78         ssize_t ret;
  79
  80         if (!pagesize)
  81                 pagesize = sysconf(_SC_PAGESIZE);
  82
  83         if (pagesize == -1)
  84                 return false;
  85
  86         if (fd < 0) {
  87                 char path[64];
  88
  89                 snprintf(path, sizeof(path), "/proc/%d/statm", getpid());
  90
  91                 fd = ws_open(path, O_RDONLY);
  92
  93                 /* XXX, fallback to some other /proc file ? */
  94         }
  95
  96         if (fd < 0)
  97                 return false;
  98
  99         ret = pread(fd, buf, sizeof(buf)-1, 0);
 100         if (ret <= 0)
 101                 return false;
 102
 103         buf[ret] = '\0';
 104
 105         if (sscanf(buf, "%lu %lu", &total, &rss) != 2)
 106                 return false;
 107
 108         if (ptotal)
 109                 *ptotal = pagesize * (size_t) total;
 110         if (prss)
 111                 *prss = pagesize * (size_t) rss;
 112
 113         return true;
 114 }
 115
 116 static size_t
 117 linux_get_total_mem_used_by_app(void)
 118 {
 119         size_t total;
 120
 121         if (!linux_get_memory(&total, NULL))
 122                 total = 0;
 123
 124         return total;
 125 }
 126
 127 static size_t
 128 linux_get_rss_mem_used_by_app(void)
 129 {
 130         size_t rss;
 131
 132         if (!linux_get_memory(NULL, &rss))
 133                 rss = 0;
 134
 135         return rss;
 136 }
 137
 138 static const ws_mem_usage_t total_usage = { "Total", linux_get_total_mem_used_by_app, NULL };
 139 static const ws_mem_usage_t rss_usage = { "RSS", linux_get_rss_mem_used_by_app, NULL };
 140
 141 static const ws_mem_usage_t *memory_components[MAX_COMPONENTS] = {
 142         &total_usage,
 143         &rss_usage,
 144 };
 145
 146 static unsigned memory_register_num = 2;
 147
 148 #else
 149
 150 /*
 151  * macOS: task_info()?
 152  *
 153  * *BSD: getrusage() -> ru_ixrss ?  Note that there are three
 154  * current-RSS components in struct rusage, but those date
 155  * back to the days when you had just text, data, and stack,
 156  * and kernels might not even bother supplying them.
 157  */
 158
 159 static const ws_mem_usage_t *memory_components[MAX_COMPONENTS];
 160
 161 static unsigned memory_register_num;
 162
 163 #endif
 164
 165 /* public API */
 166
 167 void
 168 memory_usage_component_register(const ws_mem_usage_t *component)
 169 {
 170         if (memory_register_num >= MAX_COMPONENTS)
 171                 return;
 172
 173         memory_components[memory_register_num++] = component;
 174 }
 175
 176 const char *
 177 memory_usage_get(unsigned idx, size_t *value)
 178 {
 179         if (idx >= memory_register_num)
 180                 return NULL;
 181
 182         if (value)
 183                 *value = memory_components[idx]->fetch();
 184
 185         return memory_components[idx]->name;
 186 }
 187
 188 void
 189 memory_usage_gc(void)
 190 {
 191         unsigned i;
 192
 193         for (i = 0; i < memory_register_num; i++) {
 194                 if (memory_components[i]->gc)
 195                         memory_components[i]->gc();
 196         }
 197 }
 198
 199
 200 /*
 201  * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
 202  *
 203  * Local variables:
 204  * c-basic-offset: 8
 205  * tab-width: 8
 206  * indent-tabs-mode: t
 207  * End:
 208  *
 209  * vi: set shiftwidth=8 tabstop=8 noexpandtab:
 210  * :indentSize=8:tabSize=8:noTabs=false:
 211  */