crypto/rand/rand_win.c

   1 /* crypto/rand/rand_win.c */
   2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
   3  * All rights reserved.
   4  *
   5  * This package is an SSL implementation written
   6  * by Eric Young (eay@cryptsoft.com).
   7  * The implementation was written so as to conform with Netscapes SSL.
   8  *
   9  * This library is free for commercial and non-commercial use as long as
  10  * the following conditions are aheared to.  The following conditions
  11  * apply to all code found in this distribution, be it the RC4, RSA,
  12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13  * included with this distribution is covered by the same copyright terms
  14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15  *
  16  * Copyright remains Eric Young's, and as such any Copyright notices in
  17  * the code are not to be removed.
  18  * If this package is used in a product, Eric Young should be given attribution
  19  * as the author of the parts of the library used.
  20  * This can be in the form of a textual message at program startup or
  21  * in documentation (online or textual) provided with the package.
  22  *
  23  * Redistribution and use in source and binary forms, with or without
  24  * modification, are permitted provided that the following conditions
  25  * are met:
  26  * 1. Redistributions of source code must retain the copyright
  27  *    notice, this list of conditions and the following disclaimer.
  28  * 2. Redistributions in binary form must reproduce the above copyright
  29  *    notice, this list of conditions and the following disclaimer in the
  30  *    documentation and/or other materials provided with the distribution.
  31  * 3. All advertising materials mentioning features or use of this software
  32  *    must display the following acknowledgement:
  33  *    "This product includes cryptographic software written by
  34  *     Eric Young (eay@cryptsoft.com)"
  35  *    The word 'cryptographic' can be left out if the rouines from the library
  36  *    being used are not cryptographic related :-).
  37  * 4. If you include any Windows specific code (or a derivative thereof) from
  38  *    the apps directory (application code) you must include an acknowledgement:
  39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40  *
  41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51  * SUCH DAMAGE.
  52  *
  53  * The licence and distribution terms for any publically available version or
  54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55  * copied and put under another distribution licence
  56  * [including the GNU Public Licence.]
  57  */
  58 /* ====================================================================
  59  * Copyright (c) 1998-2000 The OpenSSL Project.  All rights reserved.
  60  *
  61  * Redistribution and use in source and binary forms, with or without
  62  * modification, are permitted provided that the following conditions
  63  * are met:
  64  *
  65  * 1. Redistributions of source code must retain the above copyright
  66  *    notice, this list of conditions and the following disclaimer.
  67  *
  68  * 2. Redistributions in binary form must reproduce the above copyright
  69  *    notice, this list of conditions and the following disclaimer in
  70  *    the documentation and/or other materials provided with the
  71  *    distribution.
  72  *
  73  * 3. All advertising materials mentioning features or use of this
  74  *    software must display the following acknowledgment:
  75  *    "This product includes software developed by the OpenSSL Project
  76  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
  77  *
  78  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  79  *    endorse or promote products derived from this software without
  80  *    prior written permission. For written permission, please contact
  81  *    openssl-core@openssl.org.
  82  *
  83  * 5. Products derived from this software may not be called "OpenSSL"
  84  *    nor may "OpenSSL" appear in their names without prior written
  85  *    permission of the OpenSSL Project.
  86  *
  87  * 6. Redistributions of any form whatsoever must retain the following
  88  *    acknowledgment:
  89  *    "This product includes software developed by the OpenSSL Project
  90  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
  91  *
  92  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  93  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  94  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  95  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  96  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  97  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  98  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  99  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 100  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 101  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 102  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 103  * OF THE POSSIBILITY OF SUCH DAMAGE.
 104  * ====================================================================
 105  *
 106  * This product includes cryptographic software written by Eric Young
 107  * (eay@cryptsoft.com).  This product includes software written by Tim
 108  * Hudson (tjh@cryptsoft.com).
 109  *
 110  */
 111
 112 #include "cryptlib.h"
 113 #include <openssl/rand.h>
 114 #include "rand_lcl.h"
 115
 116 #if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
 117 #include <windows.h>
 118 #ifndef _WIN32_WINNT
 119 # define _WIN32_WINNT 0x0400
 120 #endif
 121 #include <wincrypt.h>
 122 #include <tlhelp32.h>
 123
 124 /* Limit the time spent walking through the heap, processes, threads and modules to
 125    a maximum of 1000 miliseconds each, unless CryptoGenRandom failed */
 126 #define MAXDELAY 1000
 127
 128 /* Intel hardware RNG CSP -- available from
 129  * http://developer.intel.com/design/security/rng/redist_license.htm
 130  */
 131 #define PROV_INTEL_SEC 22
 132 #define INTEL_DEF_PROV L"Intel Hardware Cryptographic Service Provider"
 133
 134 static void readtimer(void);
 135 static void readscreen(void);
 136
 137 /* It appears like CURSORINFO, PCURSORINFO and LPCURSORINFO are only defined
 138    when WINVER is 0x0500 and up, which currently only happens on Win2000.
 139    Unfortunately, those are typedefs, so they're a little bit difficult to
 140    detect properly.  On the other hand, the macro CURSOR_SHOWING is defined
 141    within the same conditional, so it can be use to detect the absence of said
 142    typedefs. */
 143
 144 #ifndef CURSOR_SHOWING
 145 /*
 146  * Information about the global cursor.
 147  */
 148 typedef struct tagCURSORINFO
 149 {
 150     DWORD   cbSize;
 151     DWORD   flags;
 152     HCURSOR hCursor;
 153     POINT   ptScreenPos;
 154 } CURSORINFO, *PCURSORINFO, *LPCURSORINFO;
 155
 156 #define CURSOR_SHOWING     0x00000001
 157 #endif /* CURSOR_SHOWING */
 158
 159 typedef BOOL (WINAPI *CRYPTACQUIRECONTEXTW)(HCRYPTPROV *, LPCWSTR, LPCWSTR,
 160                                     DWORD, DWORD);
 161 typedef BOOL (WINAPI *CRYPTGENRANDOM)(HCRYPTPROV, DWORD, BYTE *);
 162 typedef BOOL (WINAPI *CRYPTRELEASECONTEXT)(HCRYPTPROV, DWORD);
 163
 164 typedef HWND (WINAPI *GETFOREGROUNDWINDOW)(VOID);
 165 typedef BOOL (WINAPI *GETCURSORINFO)(PCURSORINFO);
 166 typedef DWORD (WINAPI *GETQUEUESTATUS)(UINT);
 167
 168 typedef HANDLE (WINAPI *CREATETOOLHELP32SNAPSHOT)(DWORD, DWORD);
 169 typedef BOOL (WINAPI *CLOSETOOLHELP32SNAPSHOT)(HANDLE);
 170 typedef BOOL (WINAPI *HEAP32FIRST)(LPHEAPENTRY32, DWORD, DWORD);
 171 typedef BOOL (WINAPI *HEAP32NEXT)(LPHEAPENTRY32);
 172 typedef BOOL (WINAPI *HEAP32LIST)(HANDLE, LPHEAPLIST32);
 173 typedef BOOL (WINAPI *PROCESS32)(HANDLE, LPPROCESSENTRY32);
 174 typedef BOOL (WINAPI *THREAD32)(HANDLE, LPTHREADENTRY32);
 175 typedef BOOL (WINAPI *MODULE32)(HANDLE, LPMODULEENTRY32);
 176
 177 #include <lmcons.h>
 178 #ifndef OPENSSL_SYS_WINCE
 179 #include <lmstats.h>
 180 #endif
 181 #if 1 /* The NET API is Unicode only.  It requires the use of the UNICODE
 182        * macro.  When UNICODE is defined LPTSTR becomes LPWSTR.  LMSTR was
 183        * was added to the Platform SDK to allow the NET API to be used in
 184        * non-Unicode applications provided that Unicode strings were still
 185        * used for input.  LMSTR is defined as LPWSTR.
 186        */
 187 typedef NET_API_STATUS (NET_API_FUNCTION * NETSTATGET)
 188         (LPWSTR, LPWSTR, DWORD, DWORD, LPBYTE*);
 189 typedef NET_API_STATUS (NET_API_FUNCTION * NETFREE)(LPBYTE);
 190 #endif /* 1 */
 191
 192 int RAND_poll(void)
 193 {
 194         MEMORYSTATUS m;
 195         HCRYPTPROV hProvider = 0;
 196         BYTE buf[64];
 197         DWORD w;
 198         HWND h;
 199         int good = 0;
 200
 201         HMODULE advapi, kernel, user, netapi;
 202         CRYPTACQUIRECONTEXTW acquire = 0;
 203         CRYPTGENRANDOM gen = 0;
 204         CRYPTRELEASECONTEXT release = 0;
 205 #if 1 /* There was previously a problem with NETSTATGET.  Currently, this
 206        * section is still experimental, but if all goes well, this conditional
 207        * will be removed
 208        */
 209         NETSTATGET netstatget = 0;
 210         NETFREE netfree = 0;
 211 #endif /* 1 */
 212
 213         /* Determine the OS version we are on so we can turn off things
 214          * that do not work properly.
 215          */
 216         OSVERSIONINFO osverinfo ;
 217         osverinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO) ;
 218         GetVersionEx( &osverinfo ) ;
 219
 220 #if defined(OPENSSL_SYS_WINCE)
 221 # if defined(_WIN32_WCE) && _WIN32_WCE>=210
 222 # ifndef CryptAcquireContext
 223    /* reserve for broken header... */
 224 #  define CryptAcquireContext CryptAcquireContextW
 225 # endif
 226         /* poll the CryptoAPI PRNG */
 227         /* The CryptoAPI returns sizeof(buf) bytes of randomness */
 228         if (CryptAcquireContext(&hProvider, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
 229                 {
 230                 if (CryptGenRandom(hProvider, sizeof(buf), buf))
 231                         RAND_add(buf, sizeof(buf), sizeof(buf));
 232                 CryptReleaseContext(hProvider, 0);
 233                 }
 234 # endif
 235 #else   /* OPENSSL_SYS_WINCE */
 236         /*
 237          * None of below libraries are present on Windows CE, which is
 238          * why we #ifndef the whole section. This also excuses us from
 239          * handling the GetProcAddress issue. The trouble is that in
 240          * real Win32 API GetProcAddress is available in ANSI flavor
 241          * only. In WinCE on the other hand GetProcAddress is a macro
 242          * most commonly defined as GetProcAddressW, which accepts
 243          * Unicode argument. If we were to call GetProcAddress under
 244          * WinCE, I'd recommend to either redefine GetProcAddress as
 245          * GetProcAddressA (there seem to be one in common CE spec) or
 246          * implement own shim routine, which would accept ANSI argument
 247          * and expand it to Unicode.
 248          */
 249
 250         /* load functions dynamically - not available on all systems */
 251         advapi = LoadLibrary(TEXT("ADVAPI32.DLL"));
 252         kernel = LoadLibrary(TEXT("KERNEL32.DLL"));
 253         user = LoadLibrary(TEXT("USER32.DLL"));
 254         netapi = LoadLibrary(TEXT("NETAPI32.DLL"));
 255
 256 #if 1 /* There was previously a problem with NETSTATGET.  Currently, this
 257        * section is still experimental, but if all goes well, this conditional
 258        * will be removed
 259        */
 260         if (netapi)
 261                 {
 262                 netstatget = (NETSTATGET) GetProcAddress(netapi,"NetStatisticsGet");
 263                 netfree = (NETFREE) GetProcAddress(netapi,"NetApiBufferFree");
 264                 }
 265
 266         if (netstatget && netfree)
 267                 {
 268                 LPBYTE outbuf;
 269                 /* NetStatisticsGet() is a Unicode only function
 270                  * STAT_WORKSTATION_0 contains 45 fields and STAT_SERVER_0
 271                  * contains 17 fields.  We treat each field as a source of
 272                  * one byte of entropy.
 273                  */
 274
 275                 if (netstatget(NULL, L"LanmanWorkstation", 0, 0, &outbuf) == 0)
 276                         {
 277                         RAND_add(outbuf, sizeof(STAT_WORKSTATION_0), 45);
 278                         netfree(outbuf);
 279                         }
 280                 if (netstatget(NULL, L"LanmanServer", 0, 0, &outbuf) == 0)
 281                         {
 282                         RAND_add(outbuf, sizeof(STAT_SERVER_0), 17);
 283                         netfree(outbuf);
 284                         }
 285                 }
 286
 287         if (netapi)
 288                 FreeLibrary(netapi);
 289 #endif /* 1 */
 290
 291         /* It appears like this can cause an exception deep within ADVAPI32.DLL
 292          * at random times on Windows 2000.  Reported by Jeffrey Altman.
 293          * Only use it on NT.
 294          */
 295         /* Wolfgang Marczy <WMarczy@topcall.co.at> reports that
 296          * the RegQueryValueEx call below can hang on NT4.0 (SP6).
 297          * So we don't use this at all for now. */
 298 #if 0
 299         if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
 300                 osverinfo.dwMajorVersion < 5)
 301                 {
 302                 /* Read Performance Statistics from NT/2000 registry
 303                  * The size of the performance data can vary from call
 304                  * to call so we must guess the size of the buffer to use
 305                  * and increase its size if we get an ERROR_MORE_DATA
 306                  * return instead of ERROR_SUCCESS.
 307                  */
 308                 LONG   rc=ERROR_MORE_DATA;
 309                 char * buf=NULL;
 310                 DWORD bufsz=0;
 311                 DWORD length;
 312
 313                 while (rc == ERROR_MORE_DATA)
 314                         {
 315                         buf = realloc(buf,bufsz+8192);
 316                         if (!buf)
 317                                 break;
 318                         bufsz += 8192;
 319
 320                         length = bufsz;
 321                         rc = RegQueryValueEx(HKEY_PERFORMANCE_DATA, TEXT("Global"),
 322                                 NULL, NULL, buf, &length);
 323                         }
 324                 if (rc == ERROR_SUCCESS)
 325                         {
 326                         /* For entropy count assume only least significant
 327                          * byte of each DWORD is random.
 328                          */
 329                         RAND_add(&length, sizeof(length), 0);
 330                         RAND_add(buf, length, length / 4.0);
 331
 332                         /* Close the Registry Key to allow Windows to cleanup/close
 333                          * the open handle
 334                          * Note: The 'HKEY_PERFORMANCE_DATA' key is implicitly opened
 335                          *       when the RegQueryValueEx above is done.  However, if
 336                          *       it is not explicitly closed, it can cause disk
 337                          *       partition manipulation problems.
 338                          */
 339                         RegCloseKey(HKEY_PERFORMANCE_DATA);
 340                         }
 341                 if (buf)
 342                         free(buf);
 343                 }
 344 #endif
 345
 346         if (advapi)
 347                 {
 348                 /*
 349                  * If it's available, then it's available in both ANSI
 350                  * and UNICODE flavors even in Win9x, documentation says.
 351                  * We favor Unicode...
 352                  */
 353                 acquire = (CRYPTACQUIRECONTEXTW) GetProcAddress(advapi,
 354                         "CryptAcquireContextW");
 355                 gen = (CRYPTGENRANDOM) GetProcAddress(advapi,
 356                         "CryptGenRandom");
 357                 release = (CRYPTRELEASECONTEXT) GetProcAddress(advapi,
 358                         "CryptReleaseContext");
 359                 }
 360
 361         if (acquire && gen && release)
 362                 {
 363                 /* poll the CryptoAPI PRNG */
 364                 /* The CryptoAPI returns sizeof(buf) bytes of randomness */
 365                 if (acquire(&hProvider, NULL, NULL, PROV_RSA_FULL,
 366                         CRYPT_VERIFYCONTEXT))
 367                         {
 368                         if (gen(hProvider, sizeof(buf), buf) != 0)
 369                                 {
 370                                 RAND_add(buf, sizeof(buf), 0);
 371                                 good = 1;
 372 #if 0
 373                                 printf("randomness from PROV_RSA_FULL\n");
 374 #endif
 375                                 }
 376                         release(hProvider, 0);
 377                         }
 378
 379                 /* poll the Pentium PRG with CryptoAPI */
 380                 if (acquire(&hProvider, 0, INTEL_DEF_PROV, PROV_INTEL_SEC, 0))
 381                         {
 382                         if (gen(hProvider, sizeof(buf), buf) != 0)
 383                                 {
 384                                 RAND_add(buf, sizeof(buf), sizeof(buf));
 385                                 good = 1;
 386 #if 0
 387                                 printf("randomness from PROV_INTEL_SEC\n");
 388 #endif
 389                                 }
 390                         release(hProvider, 0);
 391                         }
 392                 }
 393
 394         if (advapi)
 395                 FreeLibrary(advapi);
 396
 397         if (user)
 398                 {
 399                 GETCURSORINFO cursor;
 400                 GETFOREGROUNDWINDOW win;
 401                 GETQUEUESTATUS queue;
 402
 403                 win = (GETFOREGROUNDWINDOW) GetProcAddress(user, "GetForegroundWindow");
 404                 cursor = (GETCURSORINFO) GetProcAddress(user, "GetCursorInfo");
 405                 queue = (GETQUEUESTATUS) GetProcAddress(user, "GetQueueStatus");
 406
 407                 if (win)
 408                         {
 409                         /* window handle */
 410                         h = win();
 411                         RAND_add(&h, sizeof(h), 0);
 412                         }
 413                 if (cursor)
 414                         {
 415                         /* unfortunately, its not safe to call GetCursorInfo()
 416                          * on NT4 even though it exists in SP3 (or SP6) and
 417                          * higher.
 418                          */
 419                         if ( osverinfo.dwPlatformId == VER_PLATFORM_WIN32_NT &&
 420                                 osverinfo.dwMajorVersion < 5)
 421                                 cursor = 0;
 422                         }
 423                 if (cursor)
 424                         {
 425                         /* cursor position */
 426                         /* assume 2 bytes of entropy */
 427                         CURSORINFO ci;
 428                         ci.cbSize = sizeof(CURSORINFO);
 429                         if (cursor(&ci))
 430                                 RAND_add(&ci, ci.cbSize, 2);
 431                         }
 432
 433                 if (queue)
 434                         {
 435                         /* message queue status */
 436                         /* assume 1 byte of entropy */
 437                         w = queue(QS_ALLEVENTS);
 438                         RAND_add(&w, sizeof(w), 1);
 439                         }
 440
 441                 FreeLibrary(user);
 442                 }
 443
 444         /* Toolhelp32 snapshot: enumerate processes, threads, modules and heap
 445          * http://msdn.microsoft.com/library/psdk/winbase/toolhelp_5pfd.htm
 446          * (Win 9x and 2000 only, not available on NT)
 447          *
 448          * This seeding method was proposed in Peter Gutmann, Software
 449          * Generation of Practically Strong Random Numbers,
 450          * http://www.usenix.org/publications/library/proceedings/sec98/gutmann.html
 451          * revised version at http://www.cryptoengines.com/~peter/06_random.pdf
 452          * (The assignment of entropy estimates below is arbitrary, but based
 453          * on Peter's analysis the full poll appears to be safe. Additional
 454          * interactive seeding is encouraged.)
 455          */
 456
 457         if (kernel)
 458                 {
 459                 CREATETOOLHELP32SNAPSHOT snap;
 460                 CLOSETOOLHELP32SNAPSHOT close_snap;
 461                 HANDLE handle;
 462
 463                 HEAP32FIRST heap_first;
 464                 HEAP32NEXT heap_next;
 465                 HEAP32LIST heaplist_first, heaplist_next;
 466                 PROCESS32 process_first, process_next;
 467                 THREAD32 thread_first, thread_next;
 468                 MODULE32 module_first, module_next;
 469
 470                 HEAPLIST32 hlist;
 471                 HEAPENTRY32 hentry;
 472                 PROCESSENTRY32 p;
 473                 THREADENTRY32 t;
 474                 MODULEENTRY32 m;
 475                 DWORD stoptime = 0;
 476
 477                 snap = (CREATETOOLHELP32SNAPSHOT)
 478                         GetProcAddress(kernel, "CreateToolhelp32Snapshot");
 479                 close_snap = (CLOSETOOLHELP32SNAPSHOT)
 480                         GetProcAddress(kernel, "CloseToolhelp32Snapshot");
 481                 heap_first = (HEAP32FIRST) GetProcAddress(kernel, "Heap32First");
 482                 heap_next = (HEAP32NEXT) GetProcAddress(kernel, "Heap32Next");
 483                 heaplist_first = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListFirst");
 484                 heaplist_next = (HEAP32LIST) GetProcAddress(kernel, "Heap32ListNext");
 485                 process_first = (PROCESS32) GetProcAddress(kernel, "Process32First");
 486                 process_next = (PROCESS32) GetProcAddress(kernel, "Process32Next");
 487                 thread_first = (THREAD32) GetProcAddress(kernel, "Thread32First");
 488                 thread_next = (THREAD32) GetProcAddress(kernel, "Thread32Next");
 489                 module_first = (MODULE32) GetProcAddress(kernel, "Module32First");
 490                 module_next = (MODULE32) GetProcAddress(kernel, "Module32Next");
 491
 492                 if (snap && heap_first && heap_next && heaplist_first &&
 493                         heaplist_next && process_first && process_next &&
 494                         thread_first && thread_next && module_first &&
 495                         module_next && (handle = snap(TH32CS_SNAPALL,0))
 496                         != INVALID_HANDLE_VALUE)
 497                         {
 498                         /* heap list and heap walking */
 499                         /* HEAPLIST32 contains 3 fields that will change with
 500                          * each entry.  Consider each field a source of 1 byte
 501                          * of entropy.
 502                          * HEAPENTRY32 contains 5 fields that will change with
 503                          * each entry.  Consider each field a source of 1 byte
 504                          * of entropy.
 505                          */
 506                         hlist.dwSize = sizeof(HEAPLIST32);
 507                         if (good) stoptime = GetTickCount() + MAXDELAY;
 508                         if (heaplist_first(handle, &hlist))
 509                                 do
 510                                         {
 511                                         RAND_add(&hlist, hlist.dwSize, 3);
 512                                         hentry.dwSize = sizeof(HEAPENTRY32);
 513                                         if (heap_first(&hentry,
 514                                                 hlist.th32ProcessID,
 515                                                 hlist.th32HeapID))
 516                                                 {
 517                                                 int entrycnt = 80;
 518                                                 do
 519                                                         RAND_add(&hentry,
 520                                                                 hentry.dwSize, 5);
 521                                                 while (heap_next(&hentry)
 522                                                         && --entrycnt > 0);
 523                                                 }
 524                                         } while (heaplist_next(handle,
 525                                                 &hlist) && GetTickCount() < stoptime);
 526
 527                         /* process walking */
 528                         /* PROCESSENTRY32 contains 9 fields that will change
 529                          * with each entry.  Consider each field a source of
 530                          * 1 byte of entropy.
 531                          */
 532                         p.dwSize = sizeof(PROCESSENTRY32);
 533
 534                         if (good) stoptime = GetTickCount() + MAXDELAY;
 535                         if (process_first(handle, &p))
 536                                 do
 537                                         RAND_add(&p, p.dwSize, 9);
 538                                 while (process_next(handle, &p) && GetTickCount() < stoptime);
 539
 540                         /* thread walking */
 541                         /* THREADENTRY32 contains 6 fields that will change
 542                          * with each entry.  Consider each field a source of
 543                          * 1 byte of entropy.
 544                          */
 545                         t.dwSize = sizeof(THREADENTRY32);
 546                         if (good) stoptime = GetTickCount() + MAXDELAY;
 547                         if (thread_first(handle, &t))
 548                                 do
 549                                         RAND_add(&t, t.dwSize, 6);
 550                                 while (thread_next(handle, &t) && GetTickCount() < stoptime);
 551
 552                         /* module walking */
 553                         /* MODULEENTRY32 contains 9 fields that will change
 554                          * with each entry.  Consider each field a source of
 555                          * 1 byte of entropy.
 556                          */
 557                         m.dwSize = sizeof(MODULEENTRY32);
 558                         if (good) stoptime = GetTickCount() + MAXDELAY;
 559                         if (module_first(handle, &m))
 560                                 do
 561                                         RAND_add(&m, m.dwSize, 9);
 562                                 while (module_next(handle, &m)
 563                                                 && (GetTickCount() < stoptime));
 564                         if (close_snap)
 565                                 close_snap(handle);
 566                         else
 567                                 CloseHandle(handle);
 568
 569                         }
 570
 571                 FreeLibrary(kernel);
 572                 }
 573 #endif /* !OPENSSL_SYS_WINCE */
 574
 575         /* timer data */
 576         readtimer();
 577
 578         /* memory usage statistics */
 579         GlobalMemoryStatus(&m);
 580         RAND_add(&m, sizeof(m), 1);
 581
 582         /* process ID */
 583         w = GetCurrentProcessId();
 584         RAND_add(&w, sizeof(w), 1);
 585
 586 #if 0
 587         printf("Exiting RAND_poll\n");
 588 #endif
 589
 590         return(1);
 591 }
 592
 593 int RAND_event(UINT iMsg, WPARAM wParam, LPARAM lParam)
 594         {
 595         double add_entropy=0;
 596
 597         switch (iMsg)
 598                 {
 599         case WM_KEYDOWN:
 600                         {
 601                         static WPARAM key;
 602                         if (key != wParam)
 603                                 add_entropy = 0.05;
 604                         key = wParam;
 605                         }
 606                         break;
 607         case WM_MOUSEMOVE:
 608                         {
 609                         static int lastx,lasty,lastdx,lastdy;
 610                         int x,y,dx,dy;
 611
 612                         x=LOWORD(lParam);
 613                         y=HIWORD(lParam);
 614                         dx=lastx-x;
 615                         dy=lasty-y;
 616                         if (dx != 0 && dy != 0 && dx-lastdx != 0 && dy-lastdy != 0)
 617                                 add_entropy=.2;
 618                         lastx=x, lasty=y;
 619                         lastdx=dx, lastdy=dy;
 620                         }
 621                 break;
 622                 }
 623
 624         readtimer();
 625         RAND_add(&iMsg, sizeof(iMsg), add_entropy);
 626         RAND_add(&wParam, sizeof(wParam), 0);
 627         RAND_add(&lParam, sizeof(lParam), 0);
 628
 629         return (RAND_status());
 630         }
 631
 632
 633 void RAND_screen(void) /* function available for backward compatibility */
 634 {
 635         RAND_poll();
 636         if (GetVersion() >= 0x80000000 || !OPENSSL_isservice())
 637                 readscreen();
 638 }
 639
 640
 641 /* feed timing information to the PRNG */
 642 static void readtimer(void)
 643 {
 644         DWORD w;
 645         LARGE_INTEGER l;
 646         static int have_perfc = 1;
 647 #if defined(_MSC_VER) && defined(_M_X86)
 648         static int have_tsc = 1;
 649         DWORD cyclecount;
 650
 651         if (have_tsc) {
 652           __try {
 653             __asm {
 654               _emit 0x0f
 655               _emit 0x31
 656               mov cyclecount, eax
 657               }
 658             RAND_add(&cyclecount, sizeof(cyclecount), 1);
 659           } __except(EXCEPTION_EXECUTE_HANDLER) {
 660             have_tsc = 0;
 661           }
 662         }
 663 #else
 664 # define have_tsc 0
 665 #endif
 666
 667         if (have_perfc) {
 668           if (QueryPerformanceCounter(&l) == 0)
 669             have_perfc = 0;
 670           else
 671             RAND_add(&l, sizeof(l), 0);
 672         }
 673
 674         if (!have_tsc && !have_perfc) {
 675           w = GetTickCount();
 676           RAND_add(&w, sizeof(w), 0);
 677         }
 678 }
 679
 680 /* feed screen contents to PRNG */
 681 /*****************************************************************************
 682  *
 683  * Created 960901 by Gertjan van Oosten, gertjan@West.NL, West Consulting B.V.
 684  *
 685  * Code adapted from
 686  * <URL:http://support.microsoft.com/default.aspx?scid=kb;[LN];97193>;
 687  * the original copyright message is:
 688  *
 689  *   (C) Copyright Microsoft Corp. 1993.  All rights reserved.
 690  *
 691  *   You have a royalty-free right to use, modify, reproduce and
 692  *   distribute the Sample Files (and/or any modified version) in
 693  *   any way you find useful, provided that you agree that
 694  *   Microsoft has no warranty obligations or liability for any
 695  *   Sample Application Files which are modified.
 696  */
 697
 698 static void readscreen(void)
 699 {
 700 #if !defined(OPENSSL_SYS_WINCE) && !defined(OPENSSL_SYS_WIN32_CYGWIN)
 701   HDC           hScrDC;         /* screen DC */
 702   HDC           hMemDC;         /* memory DC */
 703   HBITMAP       hBitmap;        /* handle for our bitmap */
 704   HBITMAP       hOldBitmap;     /* handle for previous bitmap */
 705   BITMAP        bm;             /* bitmap properties */
 706   unsigned int  size;           /* size of bitmap */
 707   char          *bmbits;        /* contents of bitmap */
 708   int           w;              /* screen width */
 709   int           h;              /* screen height */
 710   int           y;              /* y-coordinate of screen lines to grab */
 711   int           n = 16;         /* number of screen lines to grab at a time */
 712
 713   /* Create a screen DC and a memory DC compatible to screen DC */
 714   hScrDC = CreateDC(TEXT("DISPLAY"), NULL, NULL, NULL);
 715   hMemDC = CreateCompatibleDC(hScrDC);
 716
 717   /* Get screen resolution */
 718   w = GetDeviceCaps(hScrDC, HORZRES);
 719   h = GetDeviceCaps(hScrDC, VERTRES);
 720
 721   /* Create a bitmap compatible with the screen DC */
 722   hBitmap = CreateCompatibleBitmap(hScrDC, w, n);
 723
 724   /* Select new bitmap into memory DC */
 725   hOldBitmap = SelectObject(hMemDC, hBitmap);
 726
 727   /* Get bitmap properties */
 728   GetObject(hBitmap, sizeof(BITMAP), (LPSTR)&bm);
 729   size = (unsigned int)bm.bmWidthBytes * bm.bmHeight * bm.bmPlanes;
 730
 731   bmbits = OPENSSL_malloc(size);
 732   if (bmbits) {
 733     /* Now go through the whole screen, repeatedly grabbing n lines */
 734     for (y = 0; y < h-n; y += n)
 735         {
 736         unsigned char md[MD_DIGEST_LENGTH];
 737
 738         /* Bitblt screen DC to memory DC */
 739         BitBlt(hMemDC, 0, 0, w, n, hScrDC, 0, y, SRCCOPY);
 740
 741         /* Copy bitmap bits from memory DC to bmbits */
 742         GetBitmapBits(hBitmap, size, bmbits);
 743
 744         /* Get the hash of the bitmap */
 745         MD(bmbits,size,md);
 746
 747         /* Seed the random generator with the hash value */
 748         RAND_add(md, MD_DIGEST_LENGTH, 0);
 749         }
 750
 751     OPENSSL_free(bmbits);
 752   }
 753
 754   /* Select old bitmap back into memory DC */
 755   hBitmap = SelectObject(hMemDC, hOldBitmap);
 756
 757   /* Clean up */
 758   DeleteObject(hBitmap);
 759   DeleteDC(hMemDC);
 760   DeleteDC(hScrDC);
 761 #endif /* !OPENSSL_SYS_WINCE */
 762 }
 763
 764 #endif