hostinfo.cc revision 8655:e4001326a5ba
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/misc.hh" 48#include "base/types.hh" 49 50using namespace std; 51 52string 53__get_hostname() 54{ 55 char host[256]; 56 if (gethostname(host, sizeof host) == -1) 57 warn("could not get host name!"); 58 return host; 59} 60 61string & 62hostname() 63{ 64 static string hostname = __get_hostname(); 65 return hostname; 66} 67 68uint64_t 69procInfo(const char *filename, const char *target) 70{ 71 int done = 0; 72 char line[80]; 73 char format[80]; 74 long usage; 75 76 FILE *fp = fopen(filename, "r"); 77 78 while (fp && !feof(fp) && !done) { 79 if (fgets(line, 80, fp)) { 80 if (strncmp(line, target, strlen(target)) == 0) { 81 snprintf(format, sizeof(format), "%s %%ld", target); 82 sscanf(line, format, &usage); 83 84 fclose(fp); 85 return usage ; 86 } 87 } 88 } 89 90 if (fp) 91 fclose(fp); 92 93 return 0; 94} 95 96uint64_t 97memUsage() 98{ 99// For the Mach-based Darwin kernel, use the task_info of the self task 100#ifdef __APPLE__ 101 struct task_basic_info t_info; 102 mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT; 103 104 if (KERN_SUCCESS != task_info(mach_task_self(), 105 TASK_BASIC_INFO, (task_info_t)&t_info, 106 &t_info_count)) { 107 return 0; 108 } 109 110 // Mimic Darwin's implementation of top and subtract 111 // SHARED_REGION_SIZE from the tasks virtual size to account for the 112 // shared memory submap that is incorporated into every process. 113 return (t_info.virtual_size - SHARED_REGION_SIZE) / 1024; 114#else 115 // Linux implementation 116 return procInfo("/proc/self/status", "VmSize:"); 117#endif 118} 119