glue/vmsize-glue.c

   1 /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2
   3 /*
   4  * vmsize-glue.c
   5  *
   6  * Copyright (C) 2004 Novell, Inc.
   7  *
   8  */
   9
  10 /*
  11  * Permission is hereby granted, free of charge, to any person obtaining a
  12  * copy of this software and associated documentation files (the "Software"),
  13  * to deal in the Software without restriction, including without limitation
  14  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  15  * and/or sell copies of the Software, and to permit persons to whom the
  16  * Software is furnished to do so, subject to the following conditions:
  17  *
  18  * The above copyright notice and this permission notice shall be included in
  19  * all copies or substantial portions of the Software.
  20  *
  21  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  24  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  26  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  27  * DEALINGS IN THE SOFTWARE.
  28  */
  29
  30 #include <string.h>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33 #include <fcntl.h>
  34 #include <unistd.h>
  35 #include <stdlib.h>
  36 #include <stdio.h>
  37 #include <ctype.h>
  38
  39 /*
  40   FIXME: It is not safe to call this function from multiple threads.
  41 */
  42
  43 int
  44 get_vmsize (void)
  45 {
  46     static char proc_filename[64] = {'\0'};
  47     static char buffer [1024];
  48     int fd;
  49     int vmsize = -1;
  50
  51     if (proc_filename[0] == '\0')
  52         snprintf (proc_filename, 64, "/proc/%d/status", getpid ());
  53
  54     fd = open (proc_filename, O_RDONLY);
  55     if (fd >= 0) {
  56         if (read (fd, buffer, sizeof (buffer)) > 0) {
  57             char *pos = strstr (buffer, "VmSize:");
  58             char *endpos = NULL;
  59             if (pos != NULL && strlen (pos) > 7) {
  60                 pos += 7;
  61                 while (*pos && isspace (*pos))
  62                     ++pos;
  63                 if (*pos != '\0') {
  64                     vmsize = (int) strtol (pos, &endpos, 10);
  65                     if (pos == endpos || *endpos != ' ')
  66                         vmsize = -1;
  67                 }
  68             }
  69         }
  70         close (fd);
  71     }
  72
  73     return vmsize;
  74 }
  75
  76 /* a stupid cut&paste */
  77 int
  78 get_vmrss (void)
  79 {
  80     static char proc_filename[64] = {'\0'};
  81     static char buffer [1024];
  82     int fd;
  83     int vmsize = -1;
  84
  85     if (proc_filename[0] == '\0')
  86         snprintf (proc_filename, 64, "/proc/%d/status", getpid ());
  87
  88     fd = open (proc_filename, O_RDONLY);
  89     if (fd >= 0) {
  90         if (read (fd, buffer, sizeof (buffer)) > 0) {
  91             char *pos = strstr (buffer, "VmRSS:");
  92             char *endpos = NULL;
  93             if (pos != NULL && strlen (pos) > 7) {
  94                 pos += 7;
  95                 while (*pos && isspace (*pos))
  96                     ++pos;
  97                 if (*pos != '\0') {
  98                     vmsize = (int) strtol (pos, &endpos, 10);
  99                     if (pos == endpos || *endpos != ' ')
 100                         vmsize = -1;
 101                 }
 102             }
 103         }
 104         close (fd);
 105     }
 106
 107     return vmsize;
 108 }