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 || DETECT_OS_FREEBSD
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
os_log_message(const char * message)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 *
os_get_option(const char * name)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
os_get_total_physical_memory(uint64_t * size)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
os_get_available_system_memory(uint64_t * size)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 || DETECT_OS_FREEBSD
217 struct rlimit rl;
218 #if DETECT_OS_OPENBSD
219 int mib[] = { CTL_HW, HW_USERMEM64 };
220 #elif DETECT_OS_FREEBSD
221 int mib[] = { CTL_HW, HW_USERMEM };
222 #endif
223 int64_t mem_available;
224 size_t len = sizeof(mem_available);
225
226 /* physmem - wired */
227 if (sysctl(mib, 2, &mem_available, &len, NULL, 0) == -1)
228 return false;
229
230 /* static login.conf limit */
231 if (getrlimit(RLIMIT_DATA, &rl) == -1)
232 return false;
233
234 *size = MIN2(mem_available, rl.rlim_cur);
235 return true;
236 #else
237 return false;
238 #endif
239 }
240