git.meshlink.io Git - meshlink/blob - src/crypto.c

   1 /*
   2     crypto.c -- Cryptographic miscellaneous functions and initialisation
   3     Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License along
  16     with this program; if not, write to the Free Software Foundation, Inc.,
  17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #include "system.h"
  21
  22 #include "crypto.h"
  23
  24 //TODO: use a strict random source once to seed a PRNG?
  25
  26 #ifndef HAVE_MINGW
  27
  28 static int random_fd = -1;
  29
  30 void crypto_init(void) {
  31         random_fd = open("/dev/urandom", O_RDONLY);
  32         if(random_fd < 0)
  33                 random_fd = open("/dev/random", O_RDONLY);
  34         if(random_fd < 0) {
  35                 fprintf(stderr, "Could not open source of random numbers: %s\n", strerror(errno));
  36                 abort();
  37         }
  38 }
  39
  40 void crypto_exit(void) {
  41         close(random_fd);
  42 }
  43
  44 void randomize(void *out, size_t outlen) {
  45         while(outlen) {
  46                 size_t len = read(random_fd, out, outlen);
  47                 if(len <= 0) {
  48                         if(errno == EAGAIN || errno == EINTR)
  49                                 continue;
  50                         fprintf(stderr, "Could not read random numbers: %s\n", strerror(errno));
  51                         abort();
  52                 }
  53                 out += len;
  54                 outlen -= len;
  55         }
  56 }
  57
  58 #else
  59
  60 #include <wincrypt.h>
  61 HCRYPTPROV prov;
  62
  63 void crypto_init(void) {
  64         if(!CryptAcquireContext(&prov, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) {
  65                 fprintf(stderr, "CryptAcquireContext() failed!\n");
  66                 abort();
  67         }
  68 }
  69
  70 void crypto_exit(void) {
  71         CryptReleaseContext(prov, 0);
  72 }
  73
  74 void randomize(void *out, size_t outlen) {
  75         if(!CryptGenRandom(prov, outlen, out)) {
  76                 fprintf(stderr, "CryptGenRandom() failed\n");
  77                 abort();
  78         }
  79 }
  80
  81 #endif