f1113b466572a80cdfb5ccc347f55384a9670b21
[tinc] / src / nolegacy / crypto.c
1 /*
2     crypto.c -- Cryptographic miscellaneous functions and initialisation
3     Copyright (C) 2007-2014 Guus Sliepen <guus@tinc-vpn.org>
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 #ifndef HAVE_MINGW
25
26 static int random_fd = -1;
27
28 static void random_init(void) {
29         random_fd = open("/dev/urandom", O_RDONLY);
30         if(random_fd < 0)
31                 random_fd = open("/dev/random", O_RDONLY);
32         if(random_fd < 0) {
33                 fprintf(stderr, "Could not open source of random numbers: %s\n", strerror(errno));
34                 abort();
35         }
36 }
37
38 static void random_exit(void) {
39         close(random_fd);
40 }
41
42 void randomize(void *out, size_t outlen) {
43         while(outlen) {
44                 size_t len = read(random_fd, out, outlen);
45                 if(len <= 0) {
46                         if(errno == EAGAIN || errno == EINTR)
47                                 continue;
48                         fprintf(stderr, "Could not read random numbers: %s\n", strerror(errno));
49                         abort();
50                 }
51                 out += len;
52                 outlen -= len;
53         }
54 }
55
56 #else
57
58 #include <wincrypt.h>
59 HCRYPTPROV prov;
60
61 void random_init(void) {
62         if(!CryptAcquireContext(&prov, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT)) {
63                 fprintf(stderr, "CryptAcquireContext() failed!\n");
64                 abort();
65         }
66 }
67
68 void random_exit(void) {
69         CryptReleaseContext(prov, 0);
70 }
71
72 void randomize(void *out, size_t outlen) {
73         if(!CryptGenRandom(prov, outlen, out)) {
74                 fprintf(stderr, "CryptGenRandom() failed\n");
75                 abort();
76         }
77 }
78
79 #endif
80
81 void crypto_init(void) {
82         random_init();
83 }
84
85 void crypto_exit(void) {
86         random_exit();
87 }