d93bcd357b6303b2cc8fdace66aa0c9f5a9198dd
[openssl.git] / crypto / cryptlib.c
1 /*
2  * Copyright 1998-2018 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 /* ====================================================================
11  * Copyright 2002 Sun Microsystems, Inc. ALL RIGHTS RESERVED.
12  * ECDH support in OpenSSL originally developed by
13  * SUN MICROSYSTEMS, INC., and contributed to the OpenSSL project.
14  */
15
16 #include "internal/cryptlib_int.h"
17 #include <openssl/safestack.h>
18
19 #if     defined(__i386)   || defined(__i386__)   || defined(_M_IX86) || \
20         defined(__x86_64) || defined(__x86_64__) || \
21         defined(_M_AMD64) || defined(_M_X64)
22
23 extern unsigned int OPENSSL_ia32cap_P[4];
24
25 # if defined(OPENSSL_CPUID_OBJ) && !defined(OPENSSL_NO_ASM) && !defined(I386_ONLY)
26 #include <stdio.h>
27 #  define OPENSSL_CPUID_SETUP
28 typedef uint64_t IA32CAP;
29 void OPENSSL_cpuid_setup(void)
30 {
31     static int trigger = 0;
32     IA32CAP OPENSSL_ia32_cpuid(unsigned int *);
33     IA32CAP vec;
34     char *env;
35
36     if (trigger)
37         return;
38
39     trigger = 1;
40     if ((env = getenv("OPENSSL_ia32cap"))) {
41         int off = (env[0] == '~') ? 1 : 0;
42 #  if defined(_WIN32)
43         if (!sscanf(env + off, "%I64i", &vec))
44             vec = strtoul(env + off, NULL, 0);
45 #  else
46         if (!sscanf(env + off, "%lli", (long long *)&vec))
47             vec = strtoul(env + off, NULL, 0);
48 #  endif
49         if (off) {
50             IA32CAP mask = vec;
51             vec = OPENSSL_ia32_cpuid(OPENSSL_ia32cap_P) & ~mask;
52             if (mask & (1<<24)) {
53                 /*
54                  * User disables FXSR bit, mask even other capabilities
55                  * that operate exclusively on XMM, so we don't have to
56                  * double-check all the time. We mask PCLMULQDQ, AMD XOP,
57                  * AES-NI and AVX. Formally speaking we don't have to
58                  * do it in x86_64 case, but we can safely assume that
59                  * x86_64 users won't actually flip this flag.
60                  */
61                 vec &= ~((IA32CAP)(1<<1|1<<11|1<<25|1<<28) << 32);
62             }
63         } else if (env[0] == ':') {
64             vec = OPENSSL_ia32_cpuid(OPENSSL_ia32cap_P);
65         }
66
67         if ((env = strchr(env, ':'))) {
68             unsigned int vecx;
69             env++;
70             off = (env[0] == '~') ? 1 : 0;
71             vecx = strtoul(env + off, NULL, 0);
72             if (off)
73                 OPENSSL_ia32cap_P[2] &= ~vecx;
74             else
75                 OPENSSL_ia32cap_P[2] = vecx;
76         } else {
77             OPENSSL_ia32cap_P[2] = 0;
78         }
79     } else {
80         vec = OPENSSL_ia32_cpuid(OPENSSL_ia32cap_P);
81     }
82
83     /*
84      * |(1<<10) sets a reserved bit to signal that variable
85      * was initialized already... This is to avoid interference
86      * with cpuid snippets in ELF .init segment.
87      */
88     OPENSSL_ia32cap_P[0] = (unsigned int)vec | (1 << 10);
89     OPENSSL_ia32cap_P[1] = (unsigned int)(vec >> 32);
90 }
91 # else
92 unsigned int OPENSSL_ia32cap_P[4];
93 # endif
94 #endif
95 int OPENSSL_NONPIC_relocated = 0;
96 #if !defined(OPENSSL_CPUID_SETUP) && !defined(OPENSSL_CPUID_OBJ)
97 void OPENSSL_cpuid_setup(void)
98 {
99 }
100 #endif
101
102 #if defined(_WIN32)
103 # include <tchar.h>
104 # include <signal.h>
105 # ifdef __WATCOMC__
106 #  if defined(_UNICODE) || defined(__UNICODE__)
107 #   define _vsntprintf _vsnwprintf
108 #  else
109 #   define _vsntprintf _vsnprintf
110 #  endif
111 # endif
112 # ifdef _MSC_VER
113 #  define alloca _alloca
114 # endif
115
116 # if defined(_WIN32_WINNT) && _WIN32_WINNT>=0x0333
117 int OPENSSL_isservice(void)
118 {
119     HWINSTA h;
120     DWORD len;
121     WCHAR *name;
122     static union {
123         void *p;
124         FARPROC f;
125     } _OPENSSL_isservice = {
126         NULL
127     };
128
129     if (_OPENSSL_isservice.p == NULL) {
130         HANDLE mod = GetModuleHandle(NULL);
131         if (mod != NULL)
132             _OPENSSL_isservice.f = GetProcAddress(mod, "_OPENSSL_isservice");
133         if (_OPENSSL_isservice.p == NULL)
134             _OPENSSL_isservice.p = (void *)-1;
135     }
136
137     if (_OPENSSL_isservice.p != (void *)-1)
138         return (*_OPENSSL_isservice.f) ();
139
140     h = GetProcessWindowStation();
141     if (h == NULL)
142         return -1;
143
144     if (GetUserObjectInformationW(h, UOI_NAME, NULL, 0, &len) ||
145         GetLastError() != ERROR_INSUFFICIENT_BUFFER)
146         return -1;
147
148     if (len > 512)
149         return -1;              /* paranoia */
150     len++, len &= ~1;           /* paranoia */
151     name = (WCHAR *)alloca(len + sizeof(WCHAR));
152     if (!GetUserObjectInformationW(h, UOI_NAME, name, len, &len))
153         return -1;
154
155     len++, len &= ~1;           /* paranoia */
156     name[len / sizeof(WCHAR)] = L'\0'; /* paranoia */
157 #  if 1
158     /*
159      * This doesn't cover "interactive" services [working with real
160      * WinSta0's] nor programs started non-interactively by Task Scheduler
161      * [those are working with SAWinSta].
162      */
163     if (wcsstr(name, L"Service-0x"))
164         return 1;
165 #  else
166     /* This covers all non-interactive programs such as services. */
167     if (!wcsstr(name, L"WinSta0"))
168         return 1;
169 #  endif
170     else
171         return 0;
172 }
173 # else
174 int OPENSSL_isservice(void)
175 {
176     return 0;
177 }
178 # endif
179
180 void OPENSSL_showfatal(const char *fmta, ...)
181 {
182     va_list ap;
183     TCHAR buf[256];
184     const TCHAR *fmt;
185 # ifdef STD_ERROR_HANDLE        /* what a dirty trick! */
186     HANDLE h;
187
188     if ((h = GetStdHandle(STD_ERROR_HANDLE)) != NULL &&
189         GetFileType(h) != FILE_TYPE_UNKNOWN) {
190         /* must be console application */
191         int len;
192         DWORD out;
193
194         va_start(ap, fmta);
195         len = _vsnprintf((char *)buf, sizeof(buf), fmta, ap);
196         WriteFile(h, buf, len < 0 ? sizeof(buf) : (DWORD) len, &out, NULL);
197         va_end(ap);
198         return;
199     }
200 # endif
201
202     if (sizeof(TCHAR) == sizeof(char))
203         fmt = (const TCHAR *)fmta;
204     else
205         do {
206             int keepgoing;
207             size_t len_0 = strlen(fmta) + 1, i;
208             WCHAR *fmtw;
209
210             fmtw = (WCHAR *)alloca(len_0 * sizeof(WCHAR));
211             if (fmtw == NULL) {
212                 fmt = (const TCHAR *)L"no stack?";
213                 break;
214             }
215             if (!MultiByteToWideChar(CP_ACP, 0, fmta, len_0, fmtw, len_0))
216                 for (i = 0; i < len_0; i++)
217                     fmtw[i] = (WCHAR)fmta[i];
218             for (i = 0; i < len_0; i++) {
219                 if (fmtw[i] == L'%')
220                     do {
221                         keepgoing = 0;
222                         switch (fmtw[i + 1]) {
223                         case L'0':
224                         case L'1':
225                         case L'2':
226                         case L'3':
227                         case L'4':
228                         case L'5':
229                         case L'6':
230                         case L'7':
231                         case L'8':
232                         case L'9':
233                         case L'.':
234                         case L'*':
235                         case L'-':
236                             i++;
237                             keepgoing = 1;
238                             break;
239                         case L's':
240                             fmtw[i + 1] = L'S';
241                             break;
242                         case L'S':
243                             fmtw[i + 1] = L's';
244                             break;
245                         case L'c':
246                             fmtw[i + 1] = L'C';
247                             break;
248                         case L'C':
249                             fmtw[i + 1] = L'c';
250                             break;
251                         }
252                     } while (keepgoing);
253             }
254             fmt = (const TCHAR *)fmtw;
255         } while (0);
256
257     va_start(ap, fmta);
258     _vsntprintf(buf, OSSL_NELEM(buf) - 1, fmt, ap);
259     buf[OSSL_NELEM(buf) - 1] = _T('\0');
260     va_end(ap);
261
262 # if defined(_WIN32_WINNT) && _WIN32_WINNT>=0x0333
263     /* this -------------v--- guards NT-specific calls */
264     if (check_winnt() && OPENSSL_isservice() > 0) {
265         HANDLE hEventLog = RegisterEventSource(NULL, _T("OpenSSL"));
266
267         if (hEventLog != NULL) {
268             const TCHAR *pmsg = buf;
269
270             if (!ReportEvent(hEventLog, EVENTLOG_ERROR_TYPE, 0, 0, NULL,
271                              1, 0, &pmsg, NULL)) {
272 #if defined(DEBUG)
273                 /*
274                  * We are in a situation where we tried to report a critical
275                  * error and this failed for some reason. As a last resort,
276                  * in debug builds, send output to the debugger or any other
277                  * tool like DebugView which can monitor the output.
278                  */
279                 OutputDebugString(pmsg);
280 #endif
281             }
282
283             (void)DeregisterEventSource(hEventLog);
284         }
285     } else
286 # endif
287         MessageBox(NULL, buf, _T("OpenSSL: FATAL"), MB_OK | MB_ICONERROR);
288 }
289 #else
290 void OPENSSL_showfatal(const char *fmta, ...)
291 {
292 #ifndef OPENSSL_NO_STDIO
293     va_list ap;
294
295     va_start(ap, fmta);
296     vfprintf(stderr, fmta, ap);
297     va_end(ap);
298 #endif
299 }
300
301 int OPENSSL_isservice(void)
302 {
303     return 0;
304 }
305 #endif
306
307 void OPENSSL_die(const char *message, const char *file, int line)
308 {
309     OPENSSL_showfatal("%s:%d: OpenSSL internal error: %s\n",
310                       file, line, message);
311 #if !defined(_WIN32)
312     abort();
313 #else
314     /*
315      * Win32 abort() customarily shows a dialog, but we just did that...
316      */
317 # if !defined(_WIN32_WCE)
318     raise(SIGABRT);
319 # endif
320     _exit(3);
321 #endif
322 }
323
324 #if !defined(OPENSSL_CPUID_OBJ)
325 /* volatile unsigned char* pointers are there because
326  * 1. Accessing a variable declared volatile via a pointer
327  *    that lacks a volatile qualifier causes undefined behavior.
328  * 2. When the variable itself is not volatile the compiler is
329  *    not required to keep all those reads and can convert
330  *    this into canonical memcmp() which doesn't read the whole block.
331  * Pointers to volatile resolve the first problem fully. The second
332  * problem cannot be resolved in any Standard-compliant way but this
333  * works the problem around. Compilers typically react to
334  * pointers to volatile by preserving the reads and writes through them.
335  * The latter is not required by the Standard if the memory pointed to
336  * is not volatile.
337  * Pointers themselves are volatile in the function signature to work
338  * around a subtle bug in gcc 4.6+ which causes writes through
339  * pointers to volatile to not be emitted in some rare,
340  * never needed in real life, pieces of code.
341  */
342 int CRYPTO_memcmp(const volatile void * volatile in_a,
343                   const volatile void * volatile in_b,
344                   size_t len)
345 {
346     size_t i;
347     const volatile unsigned char *a = in_a;
348     const volatile unsigned char *b = in_b;
349     unsigned char x = 0;
350
351     for (i = 0; i < len; i++)
352         x |= a[i] ^ b[i];
353
354     return x;
355 }
356 #endif