fips/*: extend fipsro segmenting to all _MSC_VER builds (including WinCE).
[openssl.git] / fips / fips_premain.c
1 /* ====================================================================
2  * Copyright (c) 2005 The OpenSSL Project. Rights for redistribution
3  * and usage in source and binary forms are granted according to the
4  * OpenSSL license.
5  */
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <string.h>
10 #if defined(__unix) || defined(__unix__) || defined(__vxworks) || defined(__ANDROID__)
11 #include <unistd.h>
12 #endif
13
14 #ifndef FINGERPRINT_PREMAIN_DSO_LOAD
15
16 #if defined(__GNUC__) && __GNUC__>=2
17   void FINGERPRINT_premain(void) __attribute__((constructor));
18   /* Most commonly this results in pointer to premain to be dropped
19    * to .ctors segment, which is traversed by GCC crtbegin.o upon
20    * program startup. Except on a.out OpenBSD where it results in
21    * _GLOBAL_$I$premain() {premain();} being auto-generated by
22    * compiler... But one way or another this is believed to cover
23    * *all* GCC targets. */
24 #elif defined(_MSC_VER)
25 # ifdef _WINDLL
26   __declspec(dllexport) /* this is essentially cosmetics... */
27 # endif
28   void FINGERPRINT_premain(void);
29   static int premain_wrapper(void) { FINGERPRINT_premain(); return 0; }
30 # ifdef _WIN64
31 # pragma section(".CRT$XCU",read)
32   __declspec(allocate(".CRT$XCU"))
33 # else
34 # pragma data_seg(".CRT$XCU")
35 # endif
36   static int (*p)(void) = premain_wrapper;
37   /* This results in pointer to premain to appear in .CRT segment,
38    * which is traversed by Visual C run-time initialization code.
39    * This applies to both Win32 and [all flavors of] Win64. */
40 # pragma data_seg()
41 #elif defined(__SUNPRO_C)
42   void FINGERPRINT_premain(void);
43 # pragma init(FINGERPRINT_premain)
44   /* This results in a call to premain to appear in .init segment. */
45 #elif defined(__DECC) && (defined(__VMS) || defined(VMS))
46   void FINGERPRINT_premain(void);
47 # pragma __nostandard
48   globaldef { "LIB$INITIALIZ" } readonly _align (LONGWORD)
49         int spare[8] = {0};
50   globaldef { "LIB$INITIALIZE" } readonly _align (LONGWORD)
51         void (*x_FINGERPRINT_premain)(void) = FINGERPRINT_premain;
52   /* Refer to LIB$INITIALIZE to ensure it exists in the image. */
53   int lib$initialize();
54   globaldef int (*lib_init_ref)() = lib$initialize;
55 # pragma __standard
56 #elif 0
57   The rest has to be taken care of through command line:
58
59         -Wl,-init,FINGERPRINT_premain           on OSF1 and IRIX
60         -Wl,+init,FINGERPRINT_premain           on HP-UX
61         -Wl,-binitfini:FINGERPRINT_premain      on AIX
62
63   On ELF platforms this results in a call to premain to appear in
64   .init segment...
65 #endif
66
67 #ifndef HMAC_SHA1_SIG
68 #define HMAC_SHA1_SIG "?have to make sure this string is unique"
69 #endif
70
71 #if defined(_MSC_VER)
72 # pragma section("fipsro",read)
73 # __declspec(allocate("fipsro"))
74 #endif
75 static const unsigned char FINGERPRINT_ascii_value[41] = HMAC_SHA1_SIG;
76
77 #define atox(c) ((c)>='a'?((c)-'a'+10):((c)>='A'?(c)-'A'+10:(c)-'0'))
78
79 extern const void         *FIPS_text_start(),  *FIPS_text_end();
80 extern const unsigned char FIPS_rodata_start[], FIPS_rodata_end[];
81 extern unsigned char       FIPS_signature[20];
82 extern unsigned int        FIPS_incore_fingerprint(unsigned char *,unsigned int);
83
84 /*
85  * As name suggests this code is executed prior main(). We use this
86  * opportunity to fingerprint sequestered code in virtual address
87  * space of target application.
88  */
89 void FINGERPRINT_premain(void)
90 { unsigned char sig[sizeof(FIPS_signature)];
91   const unsigned char * volatile p=FINGERPRINT_ascii_value;
92   unsigned int len=sizeof(sig),i;
93
94     /* "volatilization" is done to disengage unwanted optimization... */
95     if (*((volatile unsigned char *)p)=='?')
96     {   if (FIPS_text_start()==NULL)
97         {   fprintf(stderr,"FIPS_text_start() returns NULL\n");
98             _exit(1);
99         }
100 #if defined(DEBUG_FINGERPRINT_PREMAIN)
101         fprintf(stderr,".text:%p+%d=%p\n",FIPS_text_start(),
102                 (int)((size_t)FIPS_text_end()-(size_t)FIPS_text_start()),
103                 FIPS_text_end());
104         fprintf(stderr,".rodata:%p+%d=%p\n",FIPS_rodata_start,
105                 (int)((size_t)FIPS_rodata_end-(size_t)FIPS_rodata_start),
106                 FIPS_rodata_end);
107 #endif
108
109         len=FIPS_incore_fingerprint(sig,sizeof(sig));
110
111         if (len!=sizeof(sig))
112         {   fprintf(stderr,"fingerprint length mismatch: %u\n",len);
113             _exit(1);
114         }
115
116         for (i=0;i<len;i++) printf("%02x",sig[i]);
117         printf("\n");
118         fflush(stdout);
119         _exit(0);
120     }
121     else if (FIPS_signature[0]=='\0') do
122     {   for (i=0;i<sizeof(FIPS_signature);i++,p+=2)
123             FIPS_signature[i] = (atox(p[0])<<4)|atox(p[1]);
124
125 #if defined(DEBUG_FINGERPRINT_PREMAIN)
126         if (getenv("OPENSSL_FIPS")==NULL) break;
127
128         len=FIPS_incore_fingerprint(sig,sizeof(sig));
129
130         if (memcmp(FIPS_signature,sig,sizeof(FIPS_signature)))
131         {   fprintf(stderr,"FINGERPRINT_premain: FIPS_signature mismatch\n");
132             _exit(1);
133         }
134 #endif
135     } while(0);
136 }
137
138 #else
139
140 #include <openssl/bio.h>
141 #include <openssl/dso.h>
142 #include <openssl/err.h>
143
144 int main(int argc,char *argv[])
145 { DSO *dso;
146   DSO_FUNC_TYPE func;
147   BIO *bio_err;
148
149     if (argc < 2)
150     {   fprintf (stderr,"usage: %s libcrypto.dso\n",argv[0]);
151         return 1;
152     }
153
154     if ((bio_err=BIO_new(BIO_s_file())) == NULL)
155     {   fprintf (stderr,"unable to allocate BIO\n");
156         return 1;
157     }
158     BIO_set_fp(bio_err,stderr,BIO_NOCLOSE|BIO_FP_TEXT);
159     ERR_load_crypto_strings();
160
161     dso = DSO_load(NULL,argv[1],NULL,DSO_FLAG_NO_NAME_TRANSLATION);
162     if (dso == NULL)
163     {   ERR_print_errors(bio_err);
164         return 1;
165     }
166
167     /* This is not normally reached, because FINGERPRINT_premain should
168      * have executed and terminated application already upon DSO_load... */
169     func = DSO_bind_func(dso,"FINGERPRINT_premain");
170     if (func == NULL)
171     {   ERR_print_errors(bio_err);
172         return 1;
173     }
174
175     (*func)();
176
177   return 0;
178 }
179
180 #endif