src/base/hostinfo.cc

   1 /*
   2  * Copyright (c) 2003-2005 The Regents of The University of Michigan
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are
   7  * met: redistributions of source code must retain the above copyright
   8  * notice, this list of conditions and the following disclaimer;
   9  * redistributions in binary form must reproduce the above copyright
  10  * notice, this list of conditions and the following disclaimer in the
  11  * documentation and/or other materials provided with the distribution;
  12  * neither the name of the copyright holders nor the names of its
  13  * contributors may be used to endorse or promote products derived from
  14  * this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  20  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  *
  28  * Authors: Nathan Binkert
  29  */
  30
  31 #include <unistd.h>
  32
  33 #ifdef __APPLE__
  34 #include <mach/mach_init.h>
  35 #include <mach/shared_region.h>
  36 #include <mach/task.h>
  37 #endif
  38
  39 #include <cctype>
  40 #include <cerrno>
  41 #include <cmath>
  42 #include <cstdio>
  43 #include <cstdlib>
  44 #include <cstring>
  45 #include <string>
  46
  47 #include "base/hostinfo.hh"
  48 #include "base/misc.hh"
  49 #include "base/str.hh"
  50 #include "base/types.hh"
  51
  52 using namespace std;
  53
  54 string
  55 __get_hostname()
  56 {
  57     char host[256];
  58     if (gethostname(host, sizeof host) == -1)
  59         warn("could not get host name!");
  60     return host;
  61 }
  62
  63 string &
  64 hostname()
  65 {
  66     static string hostname = __get_hostname();
  67     return hostname;
  68 }
  69
  70 uint64_t
  71 procInfo(const char *filename, const char *target)
  72 {
  73     int  done = 0;
  74     char line[80];
  75     char format[80];
  76     long usage;
  77
  78     FILE *fp = fopen(filename, "r");
  79
  80     while (fp && !feof(fp) && !done) {
  81         if (fgets(line, 80, fp)) {
  82             if (startswith(line, target)) {
  83                 snprintf(format, sizeof(format), "%s %%ld", target);
  84                 sscanf(line, format, &usage);
  85
  86                 fclose(fp);
  87                 return usage ;
  88             }
  89         }
  90     }
  91
  92     if (fp)
  93         fclose(fp);
  94
  95     return 0;
  96 }
  97
  98 uint64_t
  99 memUsage()
 100 {
 101 // For the Mach-based Darwin kernel, use the task_info of the self task
 102 #ifdef __APPLE__
 103     struct task_basic_info t_info;
 104     mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;
 105
 106     if (KERN_SUCCESS != task_info(mach_task_self(),
 107                                   TASK_BASIC_INFO, (task_info_t)&t_info,
 108                                   &t_info_count)) {
 109         return 0;
 110     }
 111
 112     // Mimic Darwin's implementation of top and subtract
 113     // SHARED_REGION_SIZE from the tasks virtual size to account for the
 114     // shared memory submap that is incorporated into every process.
 115     return (t_info.virtual_size - SHARED_REGION_SIZE) / 1024;
 116 #else
 117     // Linux implementation
 118     return procInfo("/proc/self/status", "VmSize:");
 119 #endif
 120 }