1 /* ec_dh.c - TinyCrypt implementation of EC-DH */
2 
3 /*
4  * Copyright (c) 2014, Kenneth MacKay
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are met:
9  *  * Redistributions of source code must retain the above copyright notice,
10  *    this list of conditions and the following disclaimer.
11  *  * Redistributions in binary form must reproduce the above copyright notice,
12  *    this list of conditions and the following disclaimer in the documentation
13  *    and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
19  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25  * POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 /*
29  *  Copyright (C) 2017 by Intel Corporation, All Rights Reserved.
30  *
31  *  Redistribution and use in source and binary forms, with or without
32  *  modification, are permitted provided that the following conditions are met:
33  *
34  *    - Redistributions of source code must retain the above copyright notice,
35  *     this list of conditions and the following disclaimer.
36  *
37  *    - Redistributions in binary form must reproduce the above copyright
38  *    notice, this list of conditions and the following disclaimer in the
39  *    documentation and/or other materials provided with the distribution.
40  *
41  *    - Neither the name of Intel Corporation nor the names of its contributors
42  *    may be used to endorse or promote products derived from this software
43  *    without specific prior written permission.
44  *
45  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
46  *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
47  *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
48  *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
49  *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
50  *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
51  *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
52  *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
53  *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
54  *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
55  *  POSSIBILITY OF SUCH DAMAGE.
56  */
57 #include <tinycrypt/constants.h>
58 #include <tinycrypt/ecc.h>
59 #include <tinycrypt/ecc_dh.h>
60 #include <string.h>
61 
62 #if default_RNG_defined
63 static uECC_RNG_Function g_rng_function = &default_CSPRNG;
64 #else
65 static uECC_RNG_Function g_rng_function = 0;
66 #endif
67 
uECC_make_key_with_d(uint8_t * public_key,uint8_t * private_key,unsigned int * d,uECC_Curve curve)68 int uECC_make_key_with_d(uint8_t *public_key, uint8_t *private_key,
69 			 unsigned int *d, uECC_Curve curve)
70 {
71 
72 	uECC_word_t _private[NUM_ECC_WORDS];
73 	uECC_word_t _public[NUM_ECC_WORDS * 2];
74 
75 	/* This function is designed for test purposes-only (such as validating NIST
76 	 * test vectors) as it uses a provided value for d instead of generating
77 	 * it uniformly at random. */
78 	memcpy (_private, d, NUM_ECC_BYTES);
79 
80 	/* Computing public-key from private: */
81 	if (EccPoint_compute_public_key(_public, _private, curve)) {
82 
83 		/* Converting buffers to correct bit order: */
84 		uECC_vli_nativeToBytes(private_key,
85 				       BITS_TO_BYTES(curve->num_n_bits),
86 				       _private);
87 		uECC_vli_nativeToBytes(public_key,
88 				       curve->num_bytes,
89 				       _public);
90 		uECC_vli_nativeToBytes(public_key + curve->num_bytes,
91 				       curve->num_bytes,
92 				       _public + curve->num_words);
93 
94 		/* erasing temporary buffer used to store secret: */
95 		memset(_private, 0, NUM_ECC_BYTES);
96 
97 		return 1;
98 	}
99 	return 0;
100 }
101 
uECC_make_key(uint8_t * public_key,uint8_t * private_key,uECC_Curve curve)102 int uECC_make_key(uint8_t *public_key, uint8_t *private_key, uECC_Curve curve)
103 {
104 
105 	uECC_word_t _random[NUM_ECC_WORDS * 2];
106 	uECC_word_t _private[NUM_ECC_WORDS];
107 	uECC_word_t _public[NUM_ECC_WORDS * 2];
108 	uECC_word_t tries;
109 
110 	for (tries = 0; tries < uECC_RNG_MAX_TRIES; ++tries) {
111 		/* Generating _private uniformly at random: */
112 		uECC_RNG_Function rng_function = uECC_get_rng();
113 		if (!rng_function ||
114 			!rng_function((uint8_t *)_random, 2 * NUM_ECC_WORDS*uECC_WORD_SIZE)) {
115         		return 0;
116 		}
117 
118 		/* computing modular reduction of _random (see FIPS 186.4 B.4.1): */
119 		uECC_vli_mmod(_private, _random, curve->n, BITS_TO_WORDS(curve->num_n_bits));
120 
121 		/* Computing public-key from private: */
122 		if (EccPoint_compute_public_key(_public, _private, curve)) {
123 
124 			/* Converting buffers to correct bit order: */
125 			uECC_vli_nativeToBytes(private_key,
126 					       BITS_TO_BYTES(curve->num_n_bits),
127 					       _private);
128 			uECC_vli_nativeToBytes(public_key,
129 					       curve->num_bytes,
130 					       _public);
131 			uECC_vli_nativeToBytes(public_key + curve->num_bytes,
132  					       curve->num_bytes,
133 					       _public + curve->num_words);
134 
135 			/* erasing temporary buffer that stored secret: */
136 			memset(_private, 0, NUM_ECC_BYTES);
137 
138       			return 1;
139     		}
140   	}
141 	return 0;
142 }
143 
uECC_shared_secret(const uint8_t * public_key,const uint8_t * private_key,uint8_t * secret,uECC_Curve curve)144 int uECC_shared_secret(const uint8_t *public_key, const uint8_t *private_key,
145 		       uint8_t *secret, uECC_Curve curve)
146 {
147 
148 	uECC_word_t _public[NUM_ECC_WORDS * 2];
149 	uECC_word_t _private[NUM_ECC_WORDS];
150 
151 	uECC_word_t tmp[NUM_ECC_WORDS];
152 	uECC_word_t *p2[2] = {_private, tmp};
153 	uECC_word_t *initial_Z = 0;
154 	uECC_word_t carry;
155 	wordcount_t num_words = curve->num_words;
156 	wordcount_t num_bytes = curve->num_bytes;
157 	int r;
158 
159 	/* Converting buffers to correct bit order: */
160 	uECC_vli_bytesToNative(_private,
161       			       private_key,
162 			       BITS_TO_BYTES(curve->num_n_bits));
163 	uECC_vli_bytesToNative(_public,
164       			       public_key,
165 			       num_bytes);
166 	uECC_vli_bytesToNative(_public + num_words,
167 			       public_key + num_bytes,
168 			       num_bytes);
169 
170 	/* Regularize the bitcount for the private key so that attackers cannot use a
171 	 * side channel attack to learn the number of leading zeros. */
172 	carry = regularize_k(_private, _private, tmp, curve);
173 
174 	/* If an RNG function was specified, try to get a random initial Z value to
175 	 * improve protection against side-channel attacks. */
176 	if (g_rng_function) {
177 		if (!uECC_generate_random_int(p2[carry], curve->p, num_words)) {
178 			r = 0;
179 			goto clear_and_out;
180     		}
181     		initial_Z = p2[carry];
182   	}
183 
184 	EccPoint_mult(_public, _public, p2[!carry], initial_Z, curve->num_n_bits + 1,
185 		      curve);
186 
187 	uECC_vli_nativeToBytes(secret, num_bytes, _public);
188 	r = !EccPoint_isZero(_public, curve);
189 
190 clear_and_out:
191 	/* erasing temporary buffer used to store secret: */
192 	memset(p2, 0, sizeof(p2));
193 	__asm__ __volatile__("" :: "g"(p2) : "memory");
194 	memset(tmp, 0, sizeof(tmp));
195 	__asm__ __volatile__("" :: "g"(tmp) : "memory");
196 	memset(_private, 0, sizeof(_private));
197 	__asm__ __volatile__("" :: "g"(_private) : "memory");
198 
199 	return r;
200 }
201