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
  33         if(random_fd < 0) {
  34                 random_fd = open("/dev/random", O_RDONLY);
  35         }
  36
  37         if(random_fd < 0) {
  38                 fprintf(stderr, "Could not open source of random numbers: %s\n", strerror(errno));
  39                 abort();
  40         }
  41 }
  42
  43 void crypto_exit(void) {
  44         close(random_fd);
  45 }
  46
  47 void randomize(void *out, size_t outlen) {
  48         char *ptr = out;
  49
  50         while(outlen) {
  51                 size_t len = read(random_fd, ptr, outlen);
  52
  53                 if(len <= 0) {
  54                         if(errno == EAGAIN || errno == EINTR) {
  55                                 continue;
  56                         }
  57
  58                         fprintf(stderr, "Could not read random numbers: %s\n", strerror(errno));
  59                         abort();
  60                 }
  61
  62                 ptr += len;
  63                 outlen -= len;
  64         }
  65 }
  66
  67 #else
  68
  69 #include <wincrypt.h>
  70 HCRYPTPROV prov;
  71
  72 void crypto_init(void) {
  73         if(!CryptAcquireContext(&prov, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) {
  74                 fprintf(stderr, "CryptAcquireContext() failed!\n");
  75                 abort();
  76         }
  77 }
  78
  79 void crypto_exit(void) {
  80         CryptReleaseContext(prov, 0);
  81 }
  82
  83 void randomize(void *out, size_t outlen) {
  84         if(!CryptGenRandom(prov, outlen, out)) {
  85                 fprintf(stderr, "CryptGenRandom() failed\n");
  86                 abort();
  87         }
  88 }
  89
  90 #endif