Merge branch 'master' into blender2.8
[blender.git] / source / blender / blenlib / intern / system.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * ***** END GPL LICENSE BLOCK *****
19  */
20
21 /** \file blender/blenlib/intern/system.c
22  *  \ingroup bli
23  */
24
25 #include <stdio.h>
26 #include <stdlib.h>
27
28 #include "BLI_utildefines.h"
29 #include "BLI_system.h"
30 #include "BLI_string.h"
31
32 #include "MEM_guardedalloc.h"
33
34 /* for backtrace and gethostname/GetComputerName */
35 #if defined(WIN32)
36 #  include <windows.h>
37 #  include <dbghelp.h>
38 #else
39 #  include <execinfo.h>
40 #  include <unistd.h>
41 #endif
42
43 int BLI_cpu_support_sse2(void)
44 {
45 #if defined(__x86_64__) || defined(_M_X64)
46         /* x86_64 always has SSE2 instructions */
47         return 1;
48 #elif defined(__GNUC__) && defined(i386)
49         /* for GCC x86 we check cpuid */
50         unsigned int d;
51         __asm__(
52             "pushl %%ebx\n\t"
53             "cpuid\n\t"
54             "popl %%ebx\n\t"
55             : "=d" (d)
56             : "a" (1));
57         return (d & 0x04000000) != 0;
58 #elif (defined(_MSC_VER) && defined(_M_IX86))
59         /* also check cpuid for MSVC x86 */
60         unsigned int d;
61         __asm {
62                 xor     eax, eax
63                 inc eax
64                 push ebx
65                 cpuid
66                 pop ebx
67                 mov d, edx
68         }
69         return (d & 0x04000000) != 0;
70 #else
71         return 0;
72 #endif
73 }
74
75 /**
76  * Write a backtrace into a file for systems which support it.
77  */
78 void BLI_system_backtrace(FILE *fp)
79 {
80         /* ------------- */
81         /* Linux / Apple */
82 #if defined(__linux__) || defined(__APPLE__)
83
84 #define SIZE 100
85         void *buffer[SIZE];
86         int nptrs;
87         char **strings;
88         int i;
89
90         /* include a backtrace for good measure */
91         nptrs = backtrace(buffer, SIZE);
92         strings = backtrace_symbols(buffer, nptrs);
93         for (i = 0; i < nptrs; i++) {
94                 fputs(strings[i], fp);
95                 fputc('\n', fp);
96         }
97
98         free(strings);
99 #undef SIZE
100
101         /* -------- */
102         /* Windows  */
103 #elif defined(_MSC_VER)
104
105 #ifndef NDEBUG
106 #define MAXSYMBOL 256
107 #define SIZE 100
108         unsigned short i;
109         void *stack[SIZE];
110         unsigned short nframes;
111         SYMBOL_INFO *symbolinfo;
112         HANDLE process;
113
114         process = GetCurrentProcess();
115
116         SymInitialize(process, NULL, TRUE);
117
118         nframes = CaptureStackBackTrace(0, SIZE, stack, NULL);
119         symbolinfo = MEM_callocN(sizeof(SYMBOL_INFO) + MAXSYMBOL * sizeof(char), "crash Symbol table");
120         symbolinfo->MaxNameLen = MAXSYMBOL - 1;
121         symbolinfo->SizeOfStruct = sizeof(SYMBOL_INFO);
122
123         for (i = 0; i < nframes; i++) {
124                 SymFromAddr(process, (DWORD64)(stack[i]), 0, symbolinfo);
125
126                 fprintf(fp, "%u: %s - 0x%0X\n", nframes - i - 1, symbolinfo->Name, symbolinfo->Address);
127         }
128
129         MEM_freeN(symbolinfo);
130 #undef MAXSYMBOL
131 #undef SIZE
132 #else
133         fprintf(fp, "Crash backtrace not supported on release builds\n");
134 #endif /* NDEBUG */
135 #else /* _MSC_VER */
136         /* ------------------ */
137         /* non msvc/osx/linux */
138         (void)fp;
139 #endif
140
141 }
142 /* end BLI_system_backtrace */
143
144 /* NOTE: The code for CPU brand string is adopted from Cycles. */
145
146 #if !defined(_WIN32) || defined(FREE_WINDOWS)
147 static void __cpuid(int data[4], int selector)
148 {
149 #if defined(__x86_64__)
150         asm("cpuid" : "=a" (data[0]), "=b" (data[1]), "=c" (data[2]), "=d" (data[3]) : "a"(selector));
151 #elif defined(__i386__)
152         asm("pushl %%ebx    \n\t"
153                 "cpuid          \n\t"
154                 "movl %%ebx, %1 \n\t"
155                 "popl %%ebx     \n\t"
156                 : "=a" (data[0]), "=r" (data[1]), "=c" (data[2]), "=d" (data[3])
157                 : "a"(selector)
158                 : "ebx");
159 #else
160         data[0] = data[1] = data[2] = data[3] = 0;
161 #endif
162 }
163 #endif
164
165 char *BLI_cpu_brand_string(void)
166 {
167         char buf[48] = { 0 };
168         int result[4] = { 0 };
169         __cpuid(result, 0x80000000);
170         if (result[0] >= (int)0x80000004) {
171                 __cpuid((int *)(buf + 0), 0x80000002);
172                 __cpuid((int *)(buf + 16), 0x80000003);
173                 __cpuid((int *)(buf + 32), 0x80000004);
174                 char *brand = BLI_strdup(buf);
175                 /* TODO(sergey): Make it a bit more presentable by removing trademark. */
176                 return brand;
177         }
178         return NULL;
179 }
180
181 void BLI_hostname_get(char *buffer, size_t bufsize)
182 {
183 #ifndef WIN32
184         if (gethostname(buffer, bufsize - 1) < 0) {
185                 BLI_strncpy(buffer, "-unknown-", bufsize);
186         }
187         /* When gethostname() truncates, it doesn't guarantee the trailing \0. */
188         buffer[bufsize - 1] = '\0';
189 #else
190         DWORD bufsize_inout = bufsize;
191         if (!GetComputerName(buffer, &bufsize_inout)) {
192                 strncpy(buffer, "-unknown-", bufsize);
193         }
194 #endif
195 }