Remove inconsistency in ARM support.
[openssl.git] / crypto / armcap.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <setjmp.h>
5 #include <signal.h>
6 #include <crypto.h>
7
8 #include "arm_arch.h"
9
10 unsigned int OPENSSL_armcap_P=0;
11
12 #if __ARM_MAX_ARCH__<7
13 void OPENSSL_cpuid_setup(void) {}
14 unsigned long OPENSSL_rdtsc(void) { return 0; }
15 #else
16 static sigset_t all_masked;
17
18 static sigjmp_buf ill_jmp;
19 static void ill_handler (int sig) { siglongjmp(ill_jmp,sig); }
20
21 /*
22  * Following subroutines could have been inlined, but it's not all
23  * ARM compilers support inline assembler...
24  */
25 void _armv7_neon_probe(void);
26 void _armv8_aes_probe(void);
27 void _armv8_sha1_probe(void);
28 void _armv8_sha256_probe(void);
29 void _armv8_pmull_probe(void);
30 unsigned long _armv7_tick(void);
31
32 unsigned long OPENSSL_rdtsc(void)
33         {
34         if (OPENSSL_armcap_P & ARMV7_TICK)
35                 return _armv7_tick();
36         else
37                 return 0;
38         }
39
40 /*
41  * Use a weak reference to getauxval() so we can use it if it is available but
42  * don't break the build if it is not.
43  */
44 #if defined(__GNUC__) && __GNUC__>=2
45 void OPENSSL_cpuid_setup(void) __attribute__((constructor));
46 extern unsigned long getauxval(unsigned long type) __attribute__((weak));
47 #else
48 static unsigned long (*getauxval)(unsigned long) = NULL;
49 #endif
50
51 /*
52  * ARM puts the the feature bits for Crypto Extensions in AT_HWCAP2, whereas
53  * AArch64 used AT_HWCAP.
54  */
55 #if defined(__arm__) || defined (__arm)
56 # define HWCAP                  16      /* AT_HWCAP */
57 # define HWCAP_NEON             (1 << 12)
58
59 # define HWCAP_CE               26      /* AT_HWCAP2 */
60 # define HWCAP_CE_AES           (1 << 0)
61 # define HWCAP_CE_PMULL         (1 << 1)
62 # define HWCAP_CE_SHA1          (1 << 2)
63 # define HWCAP_CE_SHA256        (1 << 3)
64 #elif defined(__aarch64__)
65 # define HWCAP                  16      /* AT_HWCAP */
66 # define HWCAP_NEON             (1 << 1)
67
68 # define HWCAP_CE               HWCAP
69 # define HWCAP_CE_AES           (1 << 3)
70 # define HWCAP_CE_PMULL         (1 << 4)
71 # define HWCAP_CE_SHA1          (1 << 5)
72 # define HWCAP_CE_SHA256        (1 << 6)
73 #endif
74
75 void OPENSSL_cpuid_setup(void)
76         {
77         char *e;
78         struct sigaction        ill_oact,ill_act;
79         sigset_t                oset;
80         static int trigger=0;
81
82         if (trigger) return;
83         trigger=1;
84  
85         if ((e=getenv("OPENSSL_armcap")))
86                 {
87                 OPENSSL_armcap_P=(unsigned int)strtoul(e,NULL,0);
88                 return;
89                 }
90
91         sigfillset(&all_masked);
92         sigdelset(&all_masked,SIGILL);
93         sigdelset(&all_masked,SIGTRAP);
94         sigdelset(&all_masked,SIGFPE);
95         sigdelset(&all_masked,SIGBUS);
96         sigdelset(&all_masked,SIGSEGV);
97
98         OPENSSL_armcap_P = 0;
99
100         memset(&ill_act,0,sizeof(ill_act));
101         ill_act.sa_handler = ill_handler;
102         ill_act.sa_mask    = all_masked;
103
104         sigprocmask(SIG_SETMASK,&ill_act.sa_mask,&oset);
105         sigaction(SIGILL,&ill_act,&ill_oact);
106
107         if (getauxval != NULL)
108                 {
109                 if (getauxval(HWCAP) & HWCAP_NEON)
110                         {
111                         unsigned long hwcap = getauxval(HWCAP_CE);
112
113                         OPENSSL_armcap_P |= ARMV7_NEON;
114
115                         if (hwcap & HWCAP_CE_AES)
116                                 OPENSSL_armcap_P |= ARMV8_AES;
117
118                         if (hwcap & HWCAP_CE_PMULL)
119                                 OPENSSL_armcap_P |= ARMV8_PMULL;
120
121                         if (hwcap & HWCAP_CE_SHA1)
122                                 OPENSSL_armcap_P |= ARMV8_SHA1;
123
124                         if (hwcap & HWCAP_CE_SHA256)
125                                 OPENSSL_armcap_P |= ARMV8_SHA256;
126                         }
127                 }
128         else if (sigsetjmp(ill_jmp,1) == 0)
129                 {
130                 _armv7_neon_probe();
131                 OPENSSL_armcap_P |= ARMV7_NEON;
132                 if (sigsetjmp(ill_jmp,1) == 0)
133                         {
134                         _armv8_pmull_probe();
135                         OPENSSL_armcap_P |= ARMV8_PMULL|ARMV8_AES;
136                         }
137                 else if (sigsetjmp(ill_jmp,1) == 0)
138                         {
139                         _armv8_aes_probe();
140                         OPENSSL_armcap_P |= ARMV8_AES;
141                         }
142                 if (sigsetjmp(ill_jmp,1) == 0)
143                         {
144                         _armv8_sha1_probe();
145                         OPENSSL_armcap_P |= ARMV8_SHA1;
146                         }
147                 if (sigsetjmp(ill_jmp,1) == 0)
148                         {
149                         _armv8_sha256_probe();
150                         OPENSSL_armcap_P |= ARMV8_SHA256;
151                         }
152                 }
153         if (sigsetjmp(ill_jmp,1) == 0)
154                 {
155                 _armv7_tick();
156                 OPENSSL_armcap_P |= ARMV7_TICK;
157                 }
158
159         sigaction (SIGILL,&ill_oact,NULL);
160         sigprocmask(SIG_SETMASK,&oset,NULL);
161         }
162 #endif