src/util/os_misc.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2008-2010 Vmware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28
  29 #include "os_misc.h"
  30 #include "os_file.h"
  31 #include "macros.h"
  32
  33 #include <stdarg.h>
  34
  35
  36 #if DETECT_OS_WINDOWS
  37
  38 #ifndef WIN32_LEAN_AND_MEAN
  39 #define WIN32_LEAN_AND_MEAN      // Exclude rarely-used stuff from Windows headers
  40 #endif
  41 #include <windows.h>
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44
  45 #else
  46
  47 #include <stdio.h>
  48 #include <stdlib.h>
  49 #include <string.h>
  50 #include <inttypes.h>
  51
  52 #endif
  53
  54
  55 #if DETECT_OS_ANDROID
  56 #  define LOG_TAG "MESA"
  57 #  include <unistd.h>
  58 #  include <log/log.h>
  59 #elif DETECT_OS_LINUX || DETECT_OS_CYGWIN || DETECT_OS_SOLARIS || DETECT_OS_HURD
  60 #  include <unistd.h>
  61 #elif DETECT_OS_OPENBSD
  62 #  include <sys/resource.h>
  63 #  include <sys/sysctl.h>
  64 #elif DETECT_OS_APPLE || DETECT_OS_BSD
  65 #  include <sys/sysctl.h>
  66 #elif DETECT_OS_HAIKU
  67 #  include <kernel/OS.h>
  68 #elif DETECT_OS_WINDOWS
  69 #  include <windows.h>
  70 #else
  71 #error unexpected platform in os_sysinfo.c
  72 #endif
  73
  74
  75 void
  76 os_log_message(const char *message)
  77 {
  78    /* If the GALLIUM_LOG_FILE environment variable is set to a valid filename,
  79     * write all messages to that file.
  80     */
  81    static FILE *fout = NULL;
  82
  83    if (!fout) {
  84 #ifdef DEBUG
  85       /* one-time init */
  86       const char *filename = os_get_option("GALLIUM_LOG_FILE");
  87       if (filename) {
  88          const char *mode = "w";
  89          if (filename[0] == '+') {
  90             /* If the filename is prefixed with '+' then open the file for
  91              * appending instead of normal writing.
  92              */
  93             mode = "a";
  94             filename++; /* skip the '+' */
  95          }
  96          fout = fopen(filename, mode);
  97       }
  98 #endif
  99       if (!fout)
 100          fout = stderr;
 101    }
 102
 103 #if DETECT_OS_WINDOWS
 104    OutputDebugStringA(message);
 105    if(GetConsoleWindow() && !IsDebuggerPresent()) {
 106       fflush(stdout);
 107       fputs(message, fout);
 108       fflush(fout);
 109    }
 110    else if (fout != stderr) {
 111       fputs(message, fout);
 112       fflush(fout);
 113    }
 114 #else /* !DETECT_OS_WINDOWS */
 115    fflush(stdout);
 116    fputs(message, fout);
 117    fflush(fout);
 118 #  if DETECT_OS_ANDROID
 119    LOG_PRI(ANDROID_LOG_ERROR, LOG_TAG, "%s", message);
 120 #  endif
 121 #endif
 122 }
 123
 124
 125 #if !defined(EMBEDDED_DEVICE)
 126 const char *
 127 os_get_option(const char *name)
 128 {
 129    return getenv(name);
 130 }
 131 #endif /* !EMBEDDED_DEVICE */
 132
 133
 134 /**
 135  * Return the size of the total physical memory.
 136  * \param size returns the size of the total physical memory
 137  * \return true for success, or false on failure
 138  */
 139 bool
 140 os_get_total_physical_memory(uint64_t *size)
 141 {
 142 #if DETECT_OS_LINUX || DETECT_OS_CYGWIN || DETECT_OS_SOLARIS || DETECT_OS_HURD
 143    const long phys_pages = sysconf(_SC_PHYS_PAGES);
 144    const long page_size = sysconf(_SC_PAGE_SIZE);
 145
 146    if (phys_pages <= 0 || page_size <= 0)
 147       return false;
 148
 149    *size = (uint64_t)phys_pages * (uint64_t)page_size;
 150    return true;
 151 #elif DETECT_OS_APPLE || DETECT_OS_BSD
 152    size_t len = sizeof(*size);
 153    int mib[2];
 154
 155    mib[0] = CTL_HW;
 156 #if DETECT_OS_APPLE
 157    mib[1] = HW_MEMSIZE;
 158 #elif DETECT_OS_NETBSD || DETECT_OS_OPENBSD
 159    mib[1] = HW_PHYSMEM64;
 160 #elif DETECT_OS_FREEBSD
 161    mib[1] = HW_REALMEM;
 162 #elif DETECT_OS_DRAGONFLY
 163    mib[1] = HW_PHYSMEM;
 164 #else
 165 #error Unsupported *BSD
 166 #endif
 167
 168    return (sysctl(mib, 2, size, &len, NULL, 0) == 0);
 169 #elif DETECT_OS_HAIKU
 170    system_info info;
 171    status_t ret;
 172
 173    ret = get_system_info(&info);
 174    if (ret != B_OK || info.max_pages <= 0)
 175       return false;
 176
 177    *size = (uint64_t)info.max_pages * (uint64_t)B_PAGE_SIZE;
 178    return true;
 179 #elif DETECT_OS_WINDOWS
 180    MEMORYSTATUSEX status;
 181    BOOL ret;
 182
 183    status.dwLength = sizeof(status);
 184    ret = GlobalMemoryStatusEx(&status);
 185    *size = status.ullTotalPhys;
 186    return (ret == TRUE);
 187 #else
 188 #error unexpected platform in os_sysinfo.c
 189    return false;
 190 #endif
 191 }
 192
 193 bool
 194 os_get_available_system_memory(uint64_t *size)
 195 {
 196 #if DETECT_OS_LINUX
 197    char *meminfo = os_read_file("/proc/meminfo", NULL);
 198    if (!meminfo)
 199       return false;
 200
 201    char *str = strstr(meminfo, "MemAvailable:");
 202    if (!str) {
 203       free(meminfo);
 204       return false;
 205    }
 206
 207    uint64_t kb_mem_available;
 208    if (sscanf(str, "MemAvailable: %" PRIx64, &kb_mem_available) == 1) {
 209       free(meminfo);
 210       *size = kb_mem_available << 10;
 211       return true;
 212    }
 213
 214    free(meminfo);
 215    return false;
 216 #elif DETECT_OS_OPENBSD
 217    struct rlimit rl;
 218    int mib[] = { CTL_HW, HW_USERMEM64 };
 219    int64_t mem_available;
 220    size_t len = sizeof(mem_available);
 221
 222    /* physmem - wired */
 223    if (sysctl(mib, 2, &mem_available, &len, NULL, 0) == -1)
 224       return false;
 225
 226    /* static login.conf limit */
 227    if (getrlimit(RLIMIT_DATA, &rl) == -1)
 228       return false;
 229
 230    *size = MIN2(mem_available, rl.rlim_cur);
 231    return true;
 232 #else
 233    return false;
 234 #endif
 235 }