Move malloc fail checks closer to malloc
[openssl.git] / apps / winrand.c
1 /* apps/winrand.c */
2 /* ====================================================================
3  * Copyright (c) 1998-2000 The OpenSSL Project.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in
14  *    the documentation and/or other materials provided with the
15  *    distribution.
16  *
17  * 3. All advertising materials mentioning features or use of this
18  *    software must display the following acknowledgment:
19  *    "This product includes software developed by the OpenSSL Project
20  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
21  *
22  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
23  *    endorse or promote products derived from this software without
24  *    prior written permission. For written permission, please contact
25  *    openssl-core@openssl.org.
26  *
27  * 5. Products derived from this software may not be called "OpenSSL"
28  *    nor may "OpenSSL" appear in their names without prior written
29  *    permission of the OpenSSL Project.
30  *
31  * 6. Redistributions of any form whatsoever must retain the following
32  *    acknowledgment:
33  *    "This product includes software developed by the OpenSSL Project
34  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
35  *
36  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
37  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
39  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
40  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
41  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
42  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
43  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
44  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
45  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
46  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
47  * OF THE POSSIBILITY OF SUCH DAMAGE.
48  * ====================================================================
49  *
50  * This product includes cryptographic software written by Eric Young
51  * (eay@cryptsoft.com).  This product includes software written by Tim
52  * Hudson (tjh@cryptsoft.com).
53  *
54  */
55
56 /*-
57  * Usage: winrand [filename]
58  *
59  * Collects entropy from mouse movements and other events and writes
60  * random data to filename or .rnd
61  */
62
63 #include <windows.h>
64 #include <openssl/opensslv.h>
65 #include <openssl/rand.h>
66
67 LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
68 const char *filename;
69
70 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
71                    PSTR cmdline, int iCmdShow)
72 {
73     static char appname[] = "OpenSSL";
74     HWND hwnd;
75     MSG msg;
76     WNDCLASSEX wndclass;
77     char buffer[200];
78
79     if (cmdline[0] == '\0')
80         filename = RAND_file_name(buffer, sizeof buffer);
81     else
82         filename = cmdline;
83
84     RAND_load_file(filename, -1);
85
86     wndclass.cbSize = sizeof(wndclass);
87     wndclass.style = CS_HREDRAW | CS_VREDRAW;
88     wndclass.lpfnWndProc = WndProc;
89     wndclass.cbClsExtra = 0;
90     wndclass.cbWndExtra = 0;
91     wndclass.hInstance = hInstance;
92     wndclass.hIcon = LoadIcon(NULL, IDI_APPLICATION);
93     wndclass.hCursor = LoadCursor(NULL, IDC_ARROW);
94     wndclass.hbrBackground = (HBRUSH) GetStockObject(WHITE_BRUSH);
95     wndclass.lpszMenuName = NULL;
96     wndclass.lpszClassName = appname;
97     wndclass.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
98     RegisterClassEx(&wndclass);
99
100     hwnd = CreateWindow(appname, OPENSSL_VERSION_TEXT,
101                         WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT,
102                         CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, hInstance,
103                         NULL);
104
105     ShowWindow(hwnd, iCmdShow);
106     UpdateWindow(hwnd);
107
108     while (GetMessage(&msg, NULL, 0, 0)) {
109         TranslateMessage(&msg);
110         DispatchMessage(&msg);
111     }
112
113     return msg.wParam;
114 }
115
116 LRESULT CALLBACK WndProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
117 {
118     HDC hdc;
119     PAINTSTRUCT ps;
120     RECT rect;
121     static int seeded = 0;
122
123     switch (iMsg) {
124     case WM_PAINT:
125         hdc = BeginPaint(hwnd, &ps);
126         GetClientRect(hwnd, &rect);
127         DrawText(hdc, "Seeding the PRNG. Please move the mouse!", -1,
128                  &rect, DT_SINGLELINE | DT_CENTER | DT_VCENTER);
129         EndPaint(hwnd, &ps);
130         return 0;
131
132     case WM_DESTROY:
133         PostQuitMessage(0);
134         return 0;
135     }
136
137     if (RAND_event(iMsg, wParam, lParam) == 1 && seeded == 0) {
138         seeded = 1;
139         if (RAND_write_file(filename) <= 0)
140             MessageBox(hwnd, "Couldn't write random file!",
141                        "OpenSSL", MB_OK | MB_ICONERROR);
142         PostQuitMessage(0);
143     }
144
145     return DefWindowProc(hwnd, iMsg, wParam, lParam);
146 }