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