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