Add support for building tinc with MSVC
[tinc] / src / nolegacy / prf.c
1 /*
2     prf.c -- Pseudo-Random Function for key material generation
3     Copyright (C) 2011-2013 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 "../prf.h"
23 #include "../ed25519/sha512.h"
24
25 static void memxor(uint8_t *buf, uint8_t c, size_t len) {
26         for(size_t i = 0; i < len; i++) {
27                 buf[i] ^= c;
28         }
29 }
30
31 static const size_t mdlen = 64;
32 static const size_t blklen = 128;
33
34 static bool hmac_sha512(const uint8_t *key, size_t keylen, const uint8_t *msg, size_t msglen, uint8_t *out) {
35         const size_t tmplen = blklen + mdlen;
36         uint8_t *tmp = alloca(tmplen);
37         sha512_context md;
38
39         if(keylen <= blklen) {
40                 memcpy(tmp, key, keylen);
41                 memset(tmp + keylen, 0, blklen - keylen);
42         } else {
43                 if(sha512(key, keylen, tmp) != 0) {
44                         return false;
45                 }
46
47                 memset(tmp + mdlen, 0, blklen - mdlen);
48         }
49
50         if(sha512_init(&md) != 0) {
51                 return false;
52         }
53
54         // ipad
55         memxor(tmp, 0x36, blklen);
56
57         if(sha512_update(&md, tmp, blklen) != 0) {
58                 return false;
59         }
60
61         // message
62         if(sha512_update(&md, msg, msglen) != 0) {
63                 return false;
64         }
65
66         if(sha512_final(&md, tmp + blklen) != 0) {
67                 return false;
68         }
69
70         // opad
71         memxor(tmp, 0x36 ^ 0x5c, blklen);
72
73         if(sha512(tmp, tmplen, out) != 0) {
74                 return false;
75         }
76
77         return true;
78 }
79
80
81 /* Generate key material from a master secret and a seed, based on RFC 4346 section 5.
82    We use SHA512 instead of MD5 and SHA1.
83  */
84
85 bool prf(const uint8_t *secret, size_t secretlen, uint8_t *seed, size_t seedlen, uint8_t *out, size_t outlen) {
86         /* Data is what the "inner" HMAC function processes.
87            It consists of the previous HMAC result plus the seed.
88          */
89
90         const size_t datalen = mdlen + seedlen;
91         uint8_t *data = alloca(datalen);
92         memset(data, 0, mdlen);
93         memcpy(data + mdlen, seed, seedlen);
94
95         uint8_t *hash = alloca(mdlen);
96
97         while(outlen > 0) {
98                 /* Inner HMAC */
99                 if(!hmac_sha512(secret, secretlen, data, datalen, data)) {
100                         return false;
101                 }
102
103                 /* Outer HMAC */
104                 if(outlen >= mdlen) {
105                         if(!hmac_sha512(secret, secretlen, data, datalen, out)) {
106                                 return false;
107                         }
108
109                         out += mdlen;
110                         outlen -= mdlen;
111                 } else {
112                         if(!hmac_sha512(secret, secretlen, data, datalen, hash)) {
113                                 return false;
114                         }
115
116                         memcpy(out, hash, outlen);
117                         out += outlen;
118                         outlen = 0;
119                 }
120         }
121
122         return true;
123 }