Add cleanup hook for integration tests
[tinc] / src / openssl / 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 <openssl/obj_mac.h>
23
24 #include "digest.h"
25 #include "../digest.h"
26 #include "../prf.h"
27
28 /* Generate key material from a master secret and a seed, based on RFC 4346 section 5.
29    We use SHA512 instead of MD5 and SHA1.
30  */
31
32 static bool prf_xor(int nid, const char *secret, size_t secretlen, char *seed, size_t seedlen, char *out, size_t outlen) {
33         digest_t *digest = digest_open_by_nid(nid, -1);
34
35         if(!digest) {
36                 return false;
37         }
38
39         if(!digest_set_key(digest, secret, secretlen)) {
40                 digest_close(digest);
41                 return false;
42         }
43
44         size_t len = digest_length(digest);
45
46         /* Data is what the "inner" HMAC function processes.
47            It consists of the previous HMAC result plus the seed.
48          */
49
50         char data[len + seedlen];
51         memset(data, 0, len);
52         memcpy(data + len, seed, seedlen);
53
54         char hash[len];
55
56         while(outlen > 0) {
57                 /* Inner HMAC */
58                 if(!digest_create(digest, data, len + seedlen, data)) {
59                         digest_close(digest);
60                         return false;
61                 }
62
63                 /* Outer HMAC */
64                 if(!digest_create(digest, data, len + seedlen, hash)) {
65                         digest_close(digest);
66                         return false;
67                 }
68
69                 /* XOR the results of the outer HMAC into the out buffer */
70                 size_t i;
71
72                 for(i = 0; i < len && i < outlen; i++) {
73                         *out++ ^= hash[i];
74                 }
75
76                 outlen -= i;
77         }
78
79         digest_close(digest);
80         return true;
81 }
82
83 bool prf(const char *secret, size_t secretlen, char *seed, size_t seedlen, char *out, size_t outlen) {
84         /* This construction allows us to easily switch back to a scheme where the PRF is calculated using two different digest algorithms. */
85         memset(out, 0, outlen);
86
87         return prf_xor(NID_sha512, secret, secretlen, seed, seedlen, out, outlen);
88 }