xref: /freebsd/contrib/wireguard-tools/curve25519-hacl64.h (revision adf376485712c8fffbf3be330d505a969647f479)
1*adf37648SKyle Evans // SPDX-License-Identifier: GPL-2.0 OR MIT
2*adf37648SKyle Evans /*
3*adf37648SKyle Evans  * Copyright (C) 2016-2017 INRIA and Microsoft Corporation.
4*adf37648SKyle Evans  * Copyright (C) 2018-2020 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
5*adf37648SKyle Evans  *
6*adf37648SKyle Evans  * This is a machine-generated formally verified implementation of Curve25519
7*adf37648SKyle Evans  * ECDH from: <https://github.com/mitls/hacl-star>. Though originally machine
8*adf37648SKyle Evans  * generated, it has been tweaked to be suitable for use in the kernel. It is
9*adf37648SKyle Evans  * optimized for 64-bit machines that can efficiently work with 128-bit
10*adf37648SKyle Evans  * integer types.
11*adf37648SKyle Evans  */
12*adf37648SKyle Evans 
13*adf37648SKyle Evans typedef __uint128_t u128;
14*adf37648SKyle Evans 
u64_eq_mask(u64 a,u64 b)15*adf37648SKyle Evans static __always_inline u64 u64_eq_mask(u64 a, u64 b)
16*adf37648SKyle Evans {
17*adf37648SKyle Evans 	u64 x = a ^ b;
18*adf37648SKyle Evans 	u64 minus_x = ~x + (u64)1U;
19*adf37648SKyle Evans 	u64 x_or_minus_x = x | minus_x;
20*adf37648SKyle Evans 	u64 xnx = x_or_minus_x >> (u32)63U;
21*adf37648SKyle Evans 	u64 c = xnx - (u64)1U;
22*adf37648SKyle Evans 	return c;
23*adf37648SKyle Evans }
24*adf37648SKyle Evans 
u64_gte_mask(u64 a,u64 b)25*adf37648SKyle Evans static __always_inline u64 u64_gte_mask(u64 a, u64 b)
26*adf37648SKyle Evans {
27*adf37648SKyle Evans 	u64 x = a;
28*adf37648SKyle Evans 	u64 y = b;
29*adf37648SKyle Evans 	u64 x_xor_y = x ^ y;
30*adf37648SKyle Evans 	u64 x_sub_y = x - y;
31*adf37648SKyle Evans 	u64 x_sub_y_xor_y = x_sub_y ^ y;
32*adf37648SKyle Evans 	u64 q = x_xor_y | x_sub_y_xor_y;
33*adf37648SKyle Evans 	u64 x_xor_q = x ^ q;
34*adf37648SKyle Evans 	u64 x_xor_q_ = x_xor_q >> (u32)63U;
35*adf37648SKyle Evans 	u64 c = x_xor_q_ - (u64)1U;
36*adf37648SKyle Evans 	return c;
37*adf37648SKyle Evans }
38*adf37648SKyle Evans 
modulo_carry_top(u64 * b)39*adf37648SKyle Evans static __always_inline void modulo_carry_top(u64 *b)
40*adf37648SKyle Evans {
41*adf37648SKyle Evans 	u64 b4 = b[4];
42*adf37648SKyle Evans 	u64 b0 = b[0];
43*adf37648SKyle Evans 	u64 b4_ = b4 & 0x7ffffffffffffLLU;
44*adf37648SKyle Evans 	u64 b0_ = b0 + 19 * (b4 >> 51);
45*adf37648SKyle Evans 	b[4] = b4_;
46*adf37648SKyle Evans 	b[0] = b0_;
47*adf37648SKyle Evans }
48*adf37648SKyle Evans 
fproduct_copy_from_wide_(u64 * output,u128 * input)49*adf37648SKyle Evans static __always_inline void fproduct_copy_from_wide_(u64 *output, u128 *input)
50*adf37648SKyle Evans {
51*adf37648SKyle Evans 	{
52*adf37648SKyle Evans 		u128 xi = input[0];
53*adf37648SKyle Evans 		output[0] = ((u64)(xi));
54*adf37648SKyle Evans 	}
55*adf37648SKyle Evans 	{
56*adf37648SKyle Evans 		u128 xi = input[1];
57*adf37648SKyle Evans 		output[1] = ((u64)(xi));
58*adf37648SKyle Evans 	}
59*adf37648SKyle Evans 	{
60*adf37648SKyle Evans 		u128 xi = input[2];
61*adf37648SKyle Evans 		output[2] = ((u64)(xi));
62*adf37648SKyle Evans 	}
63*adf37648SKyle Evans 	{
64*adf37648SKyle Evans 		u128 xi = input[3];
65*adf37648SKyle Evans 		output[3] = ((u64)(xi));
66*adf37648SKyle Evans 	}
67*adf37648SKyle Evans 	{
68*adf37648SKyle Evans 		u128 xi = input[4];
69*adf37648SKyle Evans 		output[4] = ((u64)(xi));
70*adf37648SKyle Evans 	}
71*adf37648SKyle Evans }
72*adf37648SKyle Evans 
73*adf37648SKyle Evans static __always_inline void
fproduct_sum_scalar_multiplication_(u128 * output,u64 * input,u64 s)74*adf37648SKyle Evans fproduct_sum_scalar_multiplication_(u128 *output, u64 *input, u64 s)
75*adf37648SKyle Evans {
76*adf37648SKyle Evans 	output[0] += (u128)input[0] * s;
77*adf37648SKyle Evans 	output[1] += (u128)input[1] * s;
78*adf37648SKyle Evans 	output[2] += (u128)input[2] * s;
79*adf37648SKyle Evans 	output[3] += (u128)input[3] * s;
80*adf37648SKyle Evans 	output[4] += (u128)input[4] * s;
81*adf37648SKyle Evans }
82*adf37648SKyle Evans 
fproduct_carry_wide_(u128 * tmp)83*adf37648SKyle Evans static __always_inline void fproduct_carry_wide_(u128 *tmp)
84*adf37648SKyle Evans {
85*adf37648SKyle Evans 	{
86*adf37648SKyle Evans 		u32 ctr = 0;
87*adf37648SKyle Evans 		u128 tctr = tmp[ctr];
88*adf37648SKyle Evans 		u128 tctrp1 = tmp[ctr + 1];
89*adf37648SKyle Evans 		u64 r0 = ((u64)(tctr)) & 0x7ffffffffffffLLU;
90*adf37648SKyle Evans 		u128 c = ((tctr) >> (51));
91*adf37648SKyle Evans 		tmp[ctr] = ((u128)(r0));
92*adf37648SKyle Evans 		tmp[ctr + 1] = ((tctrp1) + (c));
93*adf37648SKyle Evans 	}
94*adf37648SKyle Evans 	{
95*adf37648SKyle Evans 		u32 ctr = 1;
96*adf37648SKyle Evans 		u128 tctr = tmp[ctr];
97*adf37648SKyle Evans 		u128 tctrp1 = tmp[ctr + 1];
98*adf37648SKyle Evans 		u64 r0 = ((u64)(tctr)) & 0x7ffffffffffffLLU;
99*adf37648SKyle Evans 		u128 c = ((tctr) >> (51));
100*adf37648SKyle Evans 		tmp[ctr] = ((u128)(r0));
101*adf37648SKyle Evans 		tmp[ctr + 1] = ((tctrp1) + (c));
102*adf37648SKyle Evans 	}
103*adf37648SKyle Evans 
104*adf37648SKyle Evans 	{
105*adf37648SKyle Evans 		u32 ctr = 2;
106*adf37648SKyle Evans 		u128 tctr = tmp[ctr];
107*adf37648SKyle Evans 		u128 tctrp1 = tmp[ctr + 1];
108*adf37648SKyle Evans 		u64 r0 = ((u64)(tctr)) & 0x7ffffffffffffLLU;
109*adf37648SKyle Evans 		u128 c = ((tctr) >> (51));
110*adf37648SKyle Evans 		tmp[ctr] = ((u128)(r0));
111*adf37648SKyle Evans 		tmp[ctr + 1] = ((tctrp1) + (c));
112*adf37648SKyle Evans 	}
113*adf37648SKyle Evans 	{
114*adf37648SKyle Evans 		u32 ctr = 3;
115*adf37648SKyle Evans 		u128 tctr = tmp[ctr];
116*adf37648SKyle Evans 		u128 tctrp1 = tmp[ctr + 1];
117*adf37648SKyle Evans 		u64 r0 = ((u64)(tctr)) & 0x7ffffffffffffLLU;
118*adf37648SKyle Evans 		u128 c = ((tctr) >> (51));
119*adf37648SKyle Evans 		tmp[ctr] = ((u128)(r0));
120*adf37648SKyle Evans 		tmp[ctr + 1] = ((tctrp1) + (c));
121*adf37648SKyle Evans 	}
122*adf37648SKyle Evans }
123*adf37648SKyle Evans 
fmul_shift_reduce(u64 * output)124*adf37648SKyle Evans static __always_inline void fmul_shift_reduce(u64 *output)
125*adf37648SKyle Evans {
126*adf37648SKyle Evans 	u64 tmp = output[4];
127*adf37648SKyle Evans 	u64 b0;
128*adf37648SKyle Evans 	{
129*adf37648SKyle Evans 		u32 ctr = 5 - 0 - 1;
130*adf37648SKyle Evans 		u64 z = output[ctr - 1];
131*adf37648SKyle Evans 		output[ctr] = z;
132*adf37648SKyle Evans 	}
133*adf37648SKyle Evans 	{
134*adf37648SKyle Evans 		u32 ctr = 5 - 1 - 1;
135*adf37648SKyle Evans 		u64 z = output[ctr - 1];
136*adf37648SKyle Evans 		output[ctr] = z;
137*adf37648SKyle Evans 	}
138*adf37648SKyle Evans 	{
139*adf37648SKyle Evans 		u32 ctr = 5 - 2 - 1;
140*adf37648SKyle Evans 		u64 z = output[ctr - 1];
141*adf37648SKyle Evans 		output[ctr] = z;
142*adf37648SKyle Evans 	}
143*adf37648SKyle Evans 	{
144*adf37648SKyle Evans 		u32 ctr = 5 - 3 - 1;
145*adf37648SKyle Evans 		u64 z = output[ctr - 1];
146*adf37648SKyle Evans 		output[ctr] = z;
147*adf37648SKyle Evans 	}
148*adf37648SKyle Evans 	output[0] = tmp;
149*adf37648SKyle Evans 	b0 = output[0];
150*adf37648SKyle Evans 	output[0] = 19 * b0;
151*adf37648SKyle Evans }
152*adf37648SKyle Evans 
fmul_mul_shift_reduce_(u128 * output,u64 * input,u64 * input21)153*adf37648SKyle Evans static __always_inline void fmul_mul_shift_reduce_(u128 *output, u64 *input,
154*adf37648SKyle Evans 						   u64 *input21)
155*adf37648SKyle Evans {
156*adf37648SKyle Evans 	u32 i;
157*adf37648SKyle Evans 	u64 input2i;
158*adf37648SKyle Evans 	{
159*adf37648SKyle Evans 		u64 input2i = input21[0];
160*adf37648SKyle Evans 		fproduct_sum_scalar_multiplication_(output, input, input2i);
161*adf37648SKyle Evans 		fmul_shift_reduce(input);
162*adf37648SKyle Evans 	}
163*adf37648SKyle Evans 	{
164*adf37648SKyle Evans 		u64 input2i = input21[1];
165*adf37648SKyle Evans 		fproduct_sum_scalar_multiplication_(output, input, input2i);
166*adf37648SKyle Evans 		fmul_shift_reduce(input);
167*adf37648SKyle Evans 	}
168*adf37648SKyle Evans 	{
169*adf37648SKyle Evans 		u64 input2i = input21[2];
170*adf37648SKyle Evans 		fproduct_sum_scalar_multiplication_(output, input, input2i);
171*adf37648SKyle Evans 		fmul_shift_reduce(input);
172*adf37648SKyle Evans 	}
173*adf37648SKyle Evans 	{
174*adf37648SKyle Evans 		u64 input2i = input21[3];
175*adf37648SKyle Evans 		fproduct_sum_scalar_multiplication_(output, input, input2i);
176*adf37648SKyle Evans 		fmul_shift_reduce(input);
177*adf37648SKyle Evans 	}
178*adf37648SKyle Evans 	i = 4;
179*adf37648SKyle Evans 	input2i = input21[i];
180*adf37648SKyle Evans 	fproduct_sum_scalar_multiplication_(output, input, input2i);
181*adf37648SKyle Evans }
182*adf37648SKyle Evans 
fmul_fmul(u64 * output,u64 * input,u64 * input21)183*adf37648SKyle Evans static __always_inline void fmul_fmul(u64 *output, u64 *input, u64 *input21)
184*adf37648SKyle Evans {
185*adf37648SKyle Evans 	u64 tmp[5] = { input[0], input[1], input[2], input[3], input[4] };
186*adf37648SKyle Evans 	{
187*adf37648SKyle Evans 		u128 b4;
188*adf37648SKyle Evans 		u128 b0;
189*adf37648SKyle Evans 		u128 b4_;
190*adf37648SKyle Evans 		u128 b0_;
191*adf37648SKyle Evans 		u64 i0;
192*adf37648SKyle Evans 		u64 i1;
193*adf37648SKyle Evans 		u64 i0_;
194*adf37648SKyle Evans 		u64 i1_;
195*adf37648SKyle Evans 		u128 t[5] = { 0 };
196*adf37648SKyle Evans 		fmul_mul_shift_reduce_(t, tmp, input21);
197*adf37648SKyle Evans 		fproduct_carry_wide_(t);
198*adf37648SKyle Evans 		b4 = t[4];
199*adf37648SKyle Evans 		b0 = t[0];
200*adf37648SKyle Evans 		b4_ = ((b4) & (((u128)(0x7ffffffffffffLLU))));
201*adf37648SKyle Evans 		b0_ = ((b0) + (((u128)(19) * (((u64)(((b4) >> (51))))))));
202*adf37648SKyle Evans 		t[4] = b4_;
203*adf37648SKyle Evans 		t[0] = b0_;
204*adf37648SKyle Evans 		fproduct_copy_from_wide_(output, t);
205*adf37648SKyle Evans 		i0 = output[0];
206*adf37648SKyle Evans 		i1 = output[1];
207*adf37648SKyle Evans 		i0_ = i0 & 0x7ffffffffffffLLU;
208*adf37648SKyle Evans 		i1_ = i1 + (i0 >> 51);
209*adf37648SKyle Evans 		output[0] = i0_;
210*adf37648SKyle Evans 		output[1] = i1_;
211*adf37648SKyle Evans 	}
212*adf37648SKyle Evans }
213*adf37648SKyle Evans 
fsquare_fsquare__(u128 * tmp,u64 * output)214*adf37648SKyle Evans static __always_inline void fsquare_fsquare__(u128 *tmp, u64 *output)
215*adf37648SKyle Evans {
216*adf37648SKyle Evans 	u64 r0 = output[0];
217*adf37648SKyle Evans 	u64 r1 = output[1];
218*adf37648SKyle Evans 	u64 r2 = output[2];
219*adf37648SKyle Evans 	u64 r3 = output[3];
220*adf37648SKyle Evans 	u64 r4 = output[4];
221*adf37648SKyle Evans 	u64 d0 = r0 * 2;
222*adf37648SKyle Evans 	u64 d1 = r1 * 2;
223*adf37648SKyle Evans 	u64 d2 = r2 * 2 * 19;
224*adf37648SKyle Evans 	u64 d419 = r4 * 19;
225*adf37648SKyle Evans 	u64 d4 = d419 * 2;
226*adf37648SKyle Evans 	u128 s0 = ((((((u128)(r0) * (r0))) + (((u128)(d4) * (r1))))) +
227*adf37648SKyle Evans 		   (((u128)(d2) * (r3))));
228*adf37648SKyle Evans 	u128 s1 = ((((((u128)(d0) * (r1))) + (((u128)(d4) * (r2))))) +
229*adf37648SKyle Evans 		   (((u128)(r3 * 19) * (r3))));
230*adf37648SKyle Evans 	u128 s2 = ((((((u128)(d0) * (r2))) + (((u128)(r1) * (r1))))) +
231*adf37648SKyle Evans 		   (((u128)(d4) * (r3))));
232*adf37648SKyle Evans 	u128 s3 = ((((((u128)(d0) * (r3))) + (((u128)(d1) * (r2))))) +
233*adf37648SKyle Evans 		   (((u128)(r4) * (d419))));
234*adf37648SKyle Evans 	u128 s4 = ((((((u128)(d0) * (r4))) + (((u128)(d1) * (r3))))) +
235*adf37648SKyle Evans 		   (((u128)(r2) * (r2))));
236*adf37648SKyle Evans 	tmp[0] = s0;
237*adf37648SKyle Evans 	tmp[1] = s1;
238*adf37648SKyle Evans 	tmp[2] = s2;
239*adf37648SKyle Evans 	tmp[3] = s3;
240*adf37648SKyle Evans 	tmp[4] = s4;
241*adf37648SKyle Evans }
242*adf37648SKyle Evans 
fsquare_fsquare_(u128 * tmp,u64 * output)243*adf37648SKyle Evans static __always_inline void fsquare_fsquare_(u128 *tmp, u64 *output)
244*adf37648SKyle Evans {
245*adf37648SKyle Evans 	u128 b4;
246*adf37648SKyle Evans 	u128 b0;
247*adf37648SKyle Evans 	u128 b4_;
248*adf37648SKyle Evans 	u128 b0_;
249*adf37648SKyle Evans 	u64 i0;
250*adf37648SKyle Evans 	u64 i1;
251*adf37648SKyle Evans 	u64 i0_;
252*adf37648SKyle Evans 	u64 i1_;
253*adf37648SKyle Evans 	fsquare_fsquare__(tmp, output);
254*adf37648SKyle Evans 	fproduct_carry_wide_(tmp);
255*adf37648SKyle Evans 	b4 = tmp[4];
256*adf37648SKyle Evans 	b0 = tmp[0];
257*adf37648SKyle Evans 	b4_ = ((b4) & (((u128)(0x7ffffffffffffLLU))));
258*adf37648SKyle Evans 	b0_ = ((b0) + (((u128)(19) * (((u64)(((b4) >> (51))))))));
259*adf37648SKyle Evans 	tmp[4] = b4_;
260*adf37648SKyle Evans 	tmp[0] = b0_;
261*adf37648SKyle Evans 	fproduct_copy_from_wide_(output, tmp);
262*adf37648SKyle Evans 	i0 = output[0];
263*adf37648SKyle Evans 	i1 = output[1];
264*adf37648SKyle Evans 	i0_ = i0 & 0x7ffffffffffffLLU;
265*adf37648SKyle Evans 	i1_ = i1 + (i0 >> 51);
266*adf37648SKyle Evans 	output[0] = i0_;
267*adf37648SKyle Evans 	output[1] = i1_;
268*adf37648SKyle Evans }
269*adf37648SKyle Evans 
fsquare_fsquare_times_(u64 * output,u128 * tmp,u32 count1)270*adf37648SKyle Evans static __always_inline void fsquare_fsquare_times_(u64 *output, u128 *tmp,
271*adf37648SKyle Evans 						   u32 count1)
272*adf37648SKyle Evans {
273*adf37648SKyle Evans 	u32 i;
274*adf37648SKyle Evans 	fsquare_fsquare_(tmp, output);
275*adf37648SKyle Evans 	for (i = 1; i < count1; ++i)
276*adf37648SKyle Evans 		fsquare_fsquare_(tmp, output);
277*adf37648SKyle Evans }
278*adf37648SKyle Evans 
fsquare_fsquare_times(u64 * output,u64 * input,u32 count1)279*adf37648SKyle Evans static __always_inline void fsquare_fsquare_times(u64 *output, u64 *input,
280*adf37648SKyle Evans 						  u32 count1)
281*adf37648SKyle Evans {
282*adf37648SKyle Evans 	u128 t[5];
283*adf37648SKyle Evans 	memcpy(output, input, 5 * sizeof(*input));
284*adf37648SKyle Evans 	fsquare_fsquare_times_(output, t, count1);
285*adf37648SKyle Evans }
286*adf37648SKyle Evans 
fsquare_fsquare_times_inplace(u64 * output,u32 count1)287*adf37648SKyle Evans static __always_inline void fsquare_fsquare_times_inplace(u64 *output,
288*adf37648SKyle Evans 							  u32 count1)
289*adf37648SKyle Evans {
290*adf37648SKyle Evans 	u128 t[5];
291*adf37648SKyle Evans 	fsquare_fsquare_times_(output, t, count1);
292*adf37648SKyle Evans }
293*adf37648SKyle Evans 
crecip_crecip(u64 * out,u64 * z)294*adf37648SKyle Evans static __always_inline void crecip_crecip(u64 *out, u64 *z)
295*adf37648SKyle Evans {
296*adf37648SKyle Evans 	u64 buf[20] = { 0 };
297*adf37648SKyle Evans 	u64 *a0 = buf;
298*adf37648SKyle Evans 	u64 *t00 = buf + 5;
299*adf37648SKyle Evans 	u64 *b0 = buf + 10;
300*adf37648SKyle Evans 	u64 *t01;
301*adf37648SKyle Evans 	u64 *b1;
302*adf37648SKyle Evans 	u64 *c0;
303*adf37648SKyle Evans 	u64 *a;
304*adf37648SKyle Evans 	u64 *t0;
305*adf37648SKyle Evans 	u64 *b;
306*adf37648SKyle Evans 	u64 *c;
307*adf37648SKyle Evans 	fsquare_fsquare_times(a0, z, 1);
308*adf37648SKyle Evans 	fsquare_fsquare_times(t00, a0, 2);
309*adf37648SKyle Evans 	fmul_fmul(b0, t00, z);
310*adf37648SKyle Evans 	fmul_fmul(a0, b0, a0);
311*adf37648SKyle Evans 	fsquare_fsquare_times(t00, a0, 1);
312*adf37648SKyle Evans 	fmul_fmul(b0, t00, b0);
313*adf37648SKyle Evans 	fsquare_fsquare_times(t00, b0, 5);
314*adf37648SKyle Evans 	t01 = buf + 5;
315*adf37648SKyle Evans 	b1 = buf + 10;
316*adf37648SKyle Evans 	c0 = buf + 15;
317*adf37648SKyle Evans 	fmul_fmul(b1, t01, b1);
318*adf37648SKyle Evans 	fsquare_fsquare_times(t01, b1, 10);
319*adf37648SKyle Evans 	fmul_fmul(c0, t01, b1);
320*adf37648SKyle Evans 	fsquare_fsquare_times(t01, c0, 20);
321*adf37648SKyle Evans 	fmul_fmul(t01, t01, c0);
322*adf37648SKyle Evans 	fsquare_fsquare_times_inplace(t01, 10);
323*adf37648SKyle Evans 	fmul_fmul(b1, t01, b1);
324*adf37648SKyle Evans 	fsquare_fsquare_times(t01, b1, 50);
325*adf37648SKyle Evans 	a = buf;
326*adf37648SKyle Evans 	t0 = buf + 5;
327*adf37648SKyle Evans 	b = buf + 10;
328*adf37648SKyle Evans 	c = buf + 15;
329*adf37648SKyle Evans 	fmul_fmul(c, t0, b);
330*adf37648SKyle Evans 	fsquare_fsquare_times(t0, c, 100);
331*adf37648SKyle Evans 	fmul_fmul(t0, t0, c);
332*adf37648SKyle Evans 	fsquare_fsquare_times_inplace(t0, 50);
333*adf37648SKyle Evans 	fmul_fmul(t0, t0, b);
334*adf37648SKyle Evans 	fsquare_fsquare_times_inplace(t0, 5);
335*adf37648SKyle Evans 	fmul_fmul(out, t0, a);
336*adf37648SKyle Evans }
337*adf37648SKyle Evans 
fsum(u64 * a,u64 * b)338*adf37648SKyle Evans static __always_inline void fsum(u64 *a, u64 *b)
339*adf37648SKyle Evans {
340*adf37648SKyle Evans 	a[0] += b[0];
341*adf37648SKyle Evans 	a[1] += b[1];
342*adf37648SKyle Evans 	a[2] += b[2];
343*adf37648SKyle Evans 	a[3] += b[3];
344*adf37648SKyle Evans 	a[4] += b[4];
345*adf37648SKyle Evans }
346*adf37648SKyle Evans 
fdifference(u64 * a,u64 * b)347*adf37648SKyle Evans static __always_inline void fdifference(u64 *a, u64 *b)
348*adf37648SKyle Evans {
349*adf37648SKyle Evans 	u64 tmp[5] = { 0 };
350*adf37648SKyle Evans 	u64 b0;
351*adf37648SKyle Evans 	u64 b1;
352*adf37648SKyle Evans 	u64 b2;
353*adf37648SKyle Evans 	u64 b3;
354*adf37648SKyle Evans 	u64 b4;
355*adf37648SKyle Evans 	memcpy(tmp, b, 5 * sizeof(*b));
356*adf37648SKyle Evans 	b0 = tmp[0];
357*adf37648SKyle Evans 	b1 = tmp[1];
358*adf37648SKyle Evans 	b2 = tmp[2];
359*adf37648SKyle Evans 	b3 = tmp[3];
360*adf37648SKyle Evans 	b4 = tmp[4];
361*adf37648SKyle Evans 	tmp[0] = b0 + 0x3fffffffffff68LLU;
362*adf37648SKyle Evans 	tmp[1] = b1 + 0x3ffffffffffff8LLU;
363*adf37648SKyle Evans 	tmp[2] = b2 + 0x3ffffffffffff8LLU;
364*adf37648SKyle Evans 	tmp[3] = b3 + 0x3ffffffffffff8LLU;
365*adf37648SKyle Evans 	tmp[4] = b4 + 0x3ffffffffffff8LLU;
366*adf37648SKyle Evans 	{
367*adf37648SKyle Evans 		u64 xi = a[0];
368*adf37648SKyle Evans 		u64 yi = tmp[0];
369*adf37648SKyle Evans 		a[0] = yi - xi;
370*adf37648SKyle Evans 	}
371*adf37648SKyle Evans 	{
372*adf37648SKyle Evans 		u64 xi = a[1];
373*adf37648SKyle Evans 		u64 yi = tmp[1];
374*adf37648SKyle Evans 		a[1] = yi - xi;
375*adf37648SKyle Evans 	}
376*adf37648SKyle Evans 	{
377*adf37648SKyle Evans 		u64 xi = a[2];
378*adf37648SKyle Evans 		u64 yi = tmp[2];
379*adf37648SKyle Evans 		a[2] = yi - xi;
380*adf37648SKyle Evans 	}
381*adf37648SKyle Evans 	{
382*adf37648SKyle Evans 		u64 xi = a[3];
383*adf37648SKyle Evans 		u64 yi = tmp[3];
384*adf37648SKyle Evans 		a[3] = yi - xi;
385*adf37648SKyle Evans 	}
386*adf37648SKyle Evans 	{
387*adf37648SKyle Evans 		u64 xi = a[4];
388*adf37648SKyle Evans 		u64 yi = tmp[4];
389*adf37648SKyle Evans 		a[4] = yi - xi;
390*adf37648SKyle Evans 	}
391*adf37648SKyle Evans }
392*adf37648SKyle Evans 
fscalar(u64 * output,u64 * b,u64 s)393*adf37648SKyle Evans static __always_inline void fscalar(u64 *output, u64 *b, u64 s)
394*adf37648SKyle Evans {
395*adf37648SKyle Evans 	u128 tmp[5];
396*adf37648SKyle Evans 	u128 b4;
397*adf37648SKyle Evans 	u128 b0;
398*adf37648SKyle Evans 	u128 b4_;
399*adf37648SKyle Evans 	u128 b0_;
400*adf37648SKyle Evans 	{
401*adf37648SKyle Evans 		u64 xi = b[0];
402*adf37648SKyle Evans 		tmp[0] = ((u128)(xi) * (s));
403*adf37648SKyle Evans 	}
404*adf37648SKyle Evans 	{
405*adf37648SKyle Evans 		u64 xi = b[1];
406*adf37648SKyle Evans 		tmp[1] = ((u128)(xi) * (s));
407*adf37648SKyle Evans 	}
408*adf37648SKyle Evans 	{
409*adf37648SKyle Evans 		u64 xi = b[2];
410*adf37648SKyle Evans 		tmp[2] = ((u128)(xi) * (s));
411*adf37648SKyle Evans 	}
412*adf37648SKyle Evans 	{
413*adf37648SKyle Evans 		u64 xi = b[3];
414*adf37648SKyle Evans 		tmp[3] = ((u128)(xi) * (s));
415*adf37648SKyle Evans 	}
416*adf37648SKyle Evans 	{
417*adf37648SKyle Evans 		u64 xi = b[4];
418*adf37648SKyle Evans 		tmp[4] = ((u128)(xi) * (s));
419*adf37648SKyle Evans 	}
420*adf37648SKyle Evans 	fproduct_carry_wide_(tmp);
421*adf37648SKyle Evans 	b4 = tmp[4];
422*adf37648SKyle Evans 	b0 = tmp[0];
423*adf37648SKyle Evans 	b4_ = ((b4) & (((u128)(0x7ffffffffffffLLU))));
424*adf37648SKyle Evans 	b0_ = ((b0) + (((u128)(19) * (((u64)(((b4) >> (51))))))));
425*adf37648SKyle Evans 	tmp[4] = b4_;
426*adf37648SKyle Evans 	tmp[0] = b0_;
427*adf37648SKyle Evans 	fproduct_copy_from_wide_(output, tmp);
428*adf37648SKyle Evans }
429*adf37648SKyle Evans 
fmul(u64 * output,u64 * a,u64 * b)430*adf37648SKyle Evans static __always_inline void fmul(u64 *output, u64 *a, u64 *b)
431*adf37648SKyle Evans {
432*adf37648SKyle Evans 	fmul_fmul(output, a, b);
433*adf37648SKyle Evans }
434*adf37648SKyle Evans 
crecip(u64 * output,u64 * input)435*adf37648SKyle Evans static __always_inline void crecip(u64 *output, u64 *input)
436*adf37648SKyle Evans {
437*adf37648SKyle Evans 	crecip_crecip(output, input);
438*adf37648SKyle Evans }
439*adf37648SKyle Evans 
point_swap_conditional_step(u64 * a,u64 * b,u64 swap1,u32 ctr)440*adf37648SKyle Evans static __always_inline void point_swap_conditional_step(u64 *a, u64 *b,
441*adf37648SKyle Evans 							u64 swap1, u32 ctr)
442*adf37648SKyle Evans {
443*adf37648SKyle Evans 	u32 i = ctr - 1;
444*adf37648SKyle Evans 	u64 ai = a[i];
445*adf37648SKyle Evans 	u64 bi = b[i];
446*adf37648SKyle Evans 	u64 x = swap1 & (ai ^ bi);
447*adf37648SKyle Evans 	u64 ai1 = ai ^ x;
448*adf37648SKyle Evans 	u64 bi1 = bi ^ x;
449*adf37648SKyle Evans 	a[i] = ai1;
450*adf37648SKyle Evans 	b[i] = bi1;
451*adf37648SKyle Evans }
452*adf37648SKyle Evans 
point_swap_conditional5(u64 * a,u64 * b,u64 swap1)453*adf37648SKyle Evans static __always_inline void point_swap_conditional5(u64 *a, u64 *b, u64 swap1)
454*adf37648SKyle Evans {
455*adf37648SKyle Evans 	point_swap_conditional_step(a, b, swap1, 5);
456*adf37648SKyle Evans 	point_swap_conditional_step(a, b, swap1, 4);
457*adf37648SKyle Evans 	point_swap_conditional_step(a, b, swap1, 3);
458*adf37648SKyle Evans 	point_swap_conditional_step(a, b, swap1, 2);
459*adf37648SKyle Evans 	point_swap_conditional_step(a, b, swap1, 1);
460*adf37648SKyle Evans }
461*adf37648SKyle Evans 
point_swap_conditional(u64 * a,u64 * b,u64 iswap)462*adf37648SKyle Evans static __always_inline void point_swap_conditional(u64 *a, u64 *b, u64 iswap)
463*adf37648SKyle Evans {
464*adf37648SKyle Evans 	u64 swap1 = 0 - iswap;
465*adf37648SKyle Evans 	point_swap_conditional5(a, b, swap1);
466*adf37648SKyle Evans 	point_swap_conditional5(a + 5, b + 5, swap1);
467*adf37648SKyle Evans }
468*adf37648SKyle Evans 
point_copy(u64 * output,u64 * input)469*adf37648SKyle Evans static __always_inline void point_copy(u64 *output, u64 *input)
470*adf37648SKyle Evans {
471*adf37648SKyle Evans 	memcpy(output, input, 5 * sizeof(*input));
472*adf37648SKyle Evans 	memcpy(output + 5, input + 5, 5 * sizeof(*input));
473*adf37648SKyle Evans }
474*adf37648SKyle Evans 
addanddouble_fmonty(u64 * pp,u64 * ppq,u64 * p,u64 * pq,u64 * qmqp)475*adf37648SKyle Evans static __always_inline void addanddouble_fmonty(u64 *pp, u64 *ppq, u64 *p,
476*adf37648SKyle Evans 						u64 *pq, u64 *qmqp)
477*adf37648SKyle Evans {
478*adf37648SKyle Evans 	u64 *qx = qmqp;
479*adf37648SKyle Evans 	u64 *x2 = pp;
480*adf37648SKyle Evans 	u64 *z2 = pp + 5;
481*adf37648SKyle Evans 	u64 *x3 = ppq;
482*adf37648SKyle Evans 	u64 *z3 = ppq + 5;
483*adf37648SKyle Evans 	u64 *x = p;
484*adf37648SKyle Evans 	u64 *z = p + 5;
485*adf37648SKyle Evans 	u64 *xprime = pq;
486*adf37648SKyle Evans 	u64 *zprime = pq + 5;
487*adf37648SKyle Evans 	u64 buf[40] = { 0 };
488*adf37648SKyle Evans 	u64 *origx = buf;
489*adf37648SKyle Evans 	u64 *origxprime0 = buf + 5;
490*adf37648SKyle Evans 	u64 *xxprime0;
491*adf37648SKyle Evans 	u64 *zzprime0;
492*adf37648SKyle Evans 	u64 *origxprime;
493*adf37648SKyle Evans 	xxprime0 = buf + 25;
494*adf37648SKyle Evans 	zzprime0 = buf + 30;
495*adf37648SKyle Evans 	memcpy(origx, x, 5 * sizeof(*x));
496*adf37648SKyle Evans 	fsum(x, z);
497*adf37648SKyle Evans 	fdifference(z, origx);
498*adf37648SKyle Evans 	memcpy(origxprime0, xprime, 5 * sizeof(*xprime));
499*adf37648SKyle Evans 	fsum(xprime, zprime);
500*adf37648SKyle Evans 	fdifference(zprime, origxprime0);
501*adf37648SKyle Evans 	fmul(xxprime0, xprime, z);
502*adf37648SKyle Evans 	fmul(zzprime0, x, zprime);
503*adf37648SKyle Evans 	origxprime = buf + 5;
504*adf37648SKyle Evans 	{
505*adf37648SKyle Evans 		u64 *xx0;
506*adf37648SKyle Evans 		u64 *zz0;
507*adf37648SKyle Evans 		u64 *xxprime;
508*adf37648SKyle Evans 		u64 *zzprime;
509*adf37648SKyle Evans 		u64 *zzzprime;
510*adf37648SKyle Evans 		xx0 = buf + 15;
511*adf37648SKyle Evans 		zz0 = buf + 20;
512*adf37648SKyle Evans 		xxprime = buf + 25;
513*adf37648SKyle Evans 		zzprime = buf + 30;
514*adf37648SKyle Evans 		zzzprime = buf + 35;
515*adf37648SKyle Evans 		memcpy(origxprime, xxprime, 5 * sizeof(*xxprime));
516*adf37648SKyle Evans 		fsum(xxprime, zzprime);
517*adf37648SKyle Evans 		fdifference(zzprime, origxprime);
518*adf37648SKyle Evans 		fsquare_fsquare_times(x3, xxprime, 1);
519*adf37648SKyle Evans 		fsquare_fsquare_times(zzzprime, zzprime, 1);
520*adf37648SKyle Evans 		fmul(z3, zzzprime, qx);
521*adf37648SKyle Evans 		fsquare_fsquare_times(xx0, x, 1);
522*adf37648SKyle Evans 		fsquare_fsquare_times(zz0, z, 1);
523*adf37648SKyle Evans 		{
524*adf37648SKyle Evans 			u64 *zzz;
525*adf37648SKyle Evans 			u64 *xx;
526*adf37648SKyle Evans 			u64 *zz;
527*adf37648SKyle Evans 			u64 scalar;
528*adf37648SKyle Evans 			zzz = buf + 10;
529*adf37648SKyle Evans 			xx = buf + 15;
530*adf37648SKyle Evans 			zz = buf + 20;
531*adf37648SKyle Evans 			fmul(x2, xx, zz);
532*adf37648SKyle Evans 			fdifference(zz, xx);
533*adf37648SKyle Evans 			scalar = 121665;
534*adf37648SKyle Evans 			fscalar(zzz, zz, scalar);
535*adf37648SKyle Evans 			fsum(zzz, xx);
536*adf37648SKyle Evans 			fmul(z2, zzz, zz);
537*adf37648SKyle Evans 		}
538*adf37648SKyle Evans 	}
539*adf37648SKyle Evans }
540*adf37648SKyle Evans 
541*adf37648SKyle Evans static __always_inline void
ladder_smallloop_cmult_small_loop_step(u64 * nq,u64 * nqpq,u64 * nq2,u64 * nqpq2,u64 * q,u8 byt)542*adf37648SKyle Evans ladder_smallloop_cmult_small_loop_step(u64 *nq, u64 *nqpq, u64 *nq2, u64 *nqpq2,
543*adf37648SKyle Evans 				       u64 *q, u8 byt)
544*adf37648SKyle Evans {
545*adf37648SKyle Evans 	u64 bit0 = (u64)(byt >> 7);
546*adf37648SKyle Evans 	u64 bit;
547*adf37648SKyle Evans 	point_swap_conditional(nq, nqpq, bit0);
548*adf37648SKyle Evans 	addanddouble_fmonty(nq2, nqpq2, nq, nqpq, q);
549*adf37648SKyle Evans 	bit = (u64)(byt >> 7);
550*adf37648SKyle Evans 	point_swap_conditional(nq2, nqpq2, bit);
551*adf37648SKyle Evans }
552*adf37648SKyle Evans 
553*adf37648SKyle Evans static __always_inline void
ladder_smallloop_cmult_small_loop_double_step(u64 * nq,u64 * nqpq,u64 * nq2,u64 * nqpq2,u64 * q,u8 byt)554*adf37648SKyle Evans ladder_smallloop_cmult_small_loop_double_step(u64 *nq, u64 *nqpq, u64 *nq2,
555*adf37648SKyle Evans 					      u64 *nqpq2, u64 *q, u8 byt)
556*adf37648SKyle Evans {
557*adf37648SKyle Evans 	u8 byt1;
558*adf37648SKyle Evans 	ladder_smallloop_cmult_small_loop_step(nq, nqpq, nq2, nqpq2, q, byt);
559*adf37648SKyle Evans 	byt1 = byt << 1;
560*adf37648SKyle Evans 	ladder_smallloop_cmult_small_loop_step(nq2, nqpq2, nq, nqpq, q, byt1);
561*adf37648SKyle Evans }
562*adf37648SKyle Evans 
563*adf37648SKyle Evans static __always_inline void
ladder_smallloop_cmult_small_loop(u64 * nq,u64 * nqpq,u64 * nq2,u64 * nqpq2,u64 * q,u8 byt,u32 i)564*adf37648SKyle Evans ladder_smallloop_cmult_small_loop(u64 *nq, u64 *nqpq, u64 *nq2, u64 *nqpq2,
565*adf37648SKyle Evans 				  u64 *q, u8 byt, u32 i)
566*adf37648SKyle Evans {
567*adf37648SKyle Evans 	while (i--) {
568*adf37648SKyle Evans 		ladder_smallloop_cmult_small_loop_double_step(nq, nqpq, nq2,
569*adf37648SKyle Evans 							      nqpq2, q, byt);
570*adf37648SKyle Evans 		byt <<= 2;
571*adf37648SKyle Evans 	}
572*adf37648SKyle Evans }
573*adf37648SKyle Evans 
ladder_bigloop_cmult_big_loop(u8 * n1,u64 * nq,u64 * nqpq,u64 * nq2,u64 * nqpq2,u64 * q,u32 i)574*adf37648SKyle Evans static __always_inline void ladder_bigloop_cmult_big_loop(u8 *n1, u64 *nq,
575*adf37648SKyle Evans 							  u64 *nqpq, u64 *nq2,
576*adf37648SKyle Evans 							  u64 *nqpq2, u64 *q,
577*adf37648SKyle Evans 							  u32 i)
578*adf37648SKyle Evans {
579*adf37648SKyle Evans 	while (i--) {
580*adf37648SKyle Evans 		u8 byte = n1[i];
581*adf37648SKyle Evans 		ladder_smallloop_cmult_small_loop(nq, nqpq, nq2, nqpq2, q,
582*adf37648SKyle Evans 						  byte, 4);
583*adf37648SKyle Evans 	}
584*adf37648SKyle Evans }
585*adf37648SKyle Evans 
ladder_cmult(u64 * result,u8 * n1,u64 * q)586*adf37648SKyle Evans static void ladder_cmult(u64 *result, u8 *n1, u64 *q)
587*adf37648SKyle Evans {
588*adf37648SKyle Evans 	u64 point_buf[40] = { 0 };
589*adf37648SKyle Evans 	u64 *nq = point_buf;
590*adf37648SKyle Evans 	u64 *nqpq = point_buf + 10;
591*adf37648SKyle Evans 	u64 *nq2 = point_buf + 20;
592*adf37648SKyle Evans 	u64 *nqpq2 = point_buf + 30;
593*adf37648SKyle Evans 	point_copy(nqpq, q);
594*adf37648SKyle Evans 	nq[0] = 1;
595*adf37648SKyle Evans 	ladder_bigloop_cmult_big_loop(n1, nq, nqpq, nq2, nqpq2, q, 32);
596*adf37648SKyle Evans 	point_copy(result, nq);
597*adf37648SKyle Evans }
598*adf37648SKyle Evans 
format_fexpand(u64 * output,const u8 * input)599*adf37648SKyle Evans static __always_inline void format_fexpand(u64 *output, const u8 *input)
600*adf37648SKyle Evans {
601*adf37648SKyle Evans 	const u8 *x00 = input + 6;
602*adf37648SKyle Evans 	const u8 *x01 = input + 12;
603*adf37648SKyle Evans 	const u8 *x02 = input + 19;
604*adf37648SKyle Evans 	const u8 *x0 = input + 24;
605*adf37648SKyle Evans 	u64 i0, i1, i2, i3, i4, output0, output1, output2, output3, output4;
606*adf37648SKyle Evans 	i0 = get_unaligned_le64(input);
607*adf37648SKyle Evans 	i1 = get_unaligned_le64(x00);
608*adf37648SKyle Evans 	i2 = get_unaligned_le64(x01);
609*adf37648SKyle Evans 	i3 = get_unaligned_le64(x02);
610*adf37648SKyle Evans 	i4 = get_unaligned_le64(x0);
611*adf37648SKyle Evans 	output0 = i0 & 0x7ffffffffffffLLU;
612*adf37648SKyle Evans 	output1 = i1 >> 3 & 0x7ffffffffffffLLU;
613*adf37648SKyle Evans 	output2 = i2 >> 6 & 0x7ffffffffffffLLU;
614*adf37648SKyle Evans 	output3 = i3 >> 1 & 0x7ffffffffffffLLU;
615*adf37648SKyle Evans 	output4 = i4 >> 12 & 0x7ffffffffffffLLU;
616*adf37648SKyle Evans 	output[0] = output0;
617*adf37648SKyle Evans 	output[1] = output1;
618*adf37648SKyle Evans 	output[2] = output2;
619*adf37648SKyle Evans 	output[3] = output3;
620*adf37648SKyle Evans 	output[4] = output4;
621*adf37648SKyle Evans }
622*adf37648SKyle Evans 
format_fcontract_first_carry_pass(u64 * input)623*adf37648SKyle Evans static __always_inline void format_fcontract_first_carry_pass(u64 *input)
624*adf37648SKyle Evans {
625*adf37648SKyle Evans 	u64 t0 = input[0];
626*adf37648SKyle Evans 	u64 t1 = input[1];
627*adf37648SKyle Evans 	u64 t2 = input[2];
628*adf37648SKyle Evans 	u64 t3 = input[3];
629*adf37648SKyle Evans 	u64 t4 = input[4];
630*adf37648SKyle Evans 	u64 t1_ = t1 + (t0 >> 51);
631*adf37648SKyle Evans 	u64 t0_ = t0 & 0x7ffffffffffffLLU;
632*adf37648SKyle Evans 	u64 t2_ = t2 + (t1_ >> 51);
633*adf37648SKyle Evans 	u64 t1__ = t1_ & 0x7ffffffffffffLLU;
634*adf37648SKyle Evans 	u64 t3_ = t3 + (t2_ >> 51);
635*adf37648SKyle Evans 	u64 t2__ = t2_ & 0x7ffffffffffffLLU;
636*adf37648SKyle Evans 	u64 t4_ = t4 + (t3_ >> 51);
637*adf37648SKyle Evans 	u64 t3__ = t3_ & 0x7ffffffffffffLLU;
638*adf37648SKyle Evans 	input[0] = t0_;
639*adf37648SKyle Evans 	input[1] = t1__;
640*adf37648SKyle Evans 	input[2] = t2__;
641*adf37648SKyle Evans 	input[3] = t3__;
642*adf37648SKyle Evans 	input[4] = t4_;
643*adf37648SKyle Evans }
644*adf37648SKyle Evans 
format_fcontract_first_carry_full(u64 * input)645*adf37648SKyle Evans static __always_inline void format_fcontract_first_carry_full(u64 *input)
646*adf37648SKyle Evans {
647*adf37648SKyle Evans 	format_fcontract_first_carry_pass(input);
648*adf37648SKyle Evans 	modulo_carry_top(input);
649*adf37648SKyle Evans }
650*adf37648SKyle Evans 
format_fcontract_second_carry_pass(u64 * input)651*adf37648SKyle Evans static __always_inline void format_fcontract_second_carry_pass(u64 *input)
652*adf37648SKyle Evans {
653*adf37648SKyle Evans 	u64 t0 = input[0];
654*adf37648SKyle Evans 	u64 t1 = input[1];
655*adf37648SKyle Evans 	u64 t2 = input[2];
656*adf37648SKyle Evans 	u64 t3 = input[3];
657*adf37648SKyle Evans 	u64 t4 = input[4];
658*adf37648SKyle Evans 	u64 t1_ = t1 + (t0 >> 51);
659*adf37648SKyle Evans 	u64 t0_ = t0 & 0x7ffffffffffffLLU;
660*adf37648SKyle Evans 	u64 t2_ = t2 + (t1_ >> 51);
661*adf37648SKyle Evans 	u64 t1__ = t1_ & 0x7ffffffffffffLLU;
662*adf37648SKyle Evans 	u64 t3_ = t3 + (t2_ >> 51);
663*adf37648SKyle Evans 	u64 t2__ = t2_ & 0x7ffffffffffffLLU;
664*adf37648SKyle Evans 	u64 t4_ = t4 + (t3_ >> 51);
665*adf37648SKyle Evans 	u64 t3__ = t3_ & 0x7ffffffffffffLLU;
666*adf37648SKyle Evans 	input[0] = t0_;
667*adf37648SKyle Evans 	input[1] = t1__;
668*adf37648SKyle Evans 	input[2] = t2__;
669*adf37648SKyle Evans 	input[3] = t3__;
670*adf37648SKyle Evans 	input[4] = t4_;
671*adf37648SKyle Evans }
672*adf37648SKyle Evans 
format_fcontract_second_carry_full(u64 * input)673*adf37648SKyle Evans static __always_inline void format_fcontract_second_carry_full(u64 *input)
674*adf37648SKyle Evans {
675*adf37648SKyle Evans 	u64 i0;
676*adf37648SKyle Evans 	u64 i1;
677*adf37648SKyle Evans 	u64 i0_;
678*adf37648SKyle Evans 	u64 i1_;
679*adf37648SKyle Evans 	format_fcontract_second_carry_pass(input);
680*adf37648SKyle Evans 	modulo_carry_top(input);
681*adf37648SKyle Evans 	i0 = input[0];
682*adf37648SKyle Evans 	i1 = input[1];
683*adf37648SKyle Evans 	i0_ = i0 & 0x7ffffffffffffLLU;
684*adf37648SKyle Evans 	i1_ = i1 + (i0 >> 51);
685*adf37648SKyle Evans 	input[0] = i0_;
686*adf37648SKyle Evans 	input[1] = i1_;
687*adf37648SKyle Evans }
688*adf37648SKyle Evans 
format_fcontract_trim(u64 * input)689*adf37648SKyle Evans static __always_inline void format_fcontract_trim(u64 *input)
690*adf37648SKyle Evans {
691*adf37648SKyle Evans 	u64 a0 = input[0];
692*adf37648SKyle Evans 	u64 a1 = input[1];
693*adf37648SKyle Evans 	u64 a2 = input[2];
694*adf37648SKyle Evans 	u64 a3 = input[3];
695*adf37648SKyle Evans 	u64 a4 = input[4];
696*adf37648SKyle Evans 	u64 mask0 = u64_gte_mask(a0, 0x7ffffffffffedLLU);
697*adf37648SKyle Evans 	u64 mask1 = u64_eq_mask(a1, 0x7ffffffffffffLLU);
698*adf37648SKyle Evans 	u64 mask2 = u64_eq_mask(a2, 0x7ffffffffffffLLU);
699*adf37648SKyle Evans 	u64 mask3 = u64_eq_mask(a3, 0x7ffffffffffffLLU);
700*adf37648SKyle Evans 	u64 mask4 = u64_eq_mask(a4, 0x7ffffffffffffLLU);
701*adf37648SKyle Evans 	u64 mask = (((mask0 & mask1) & mask2) & mask3) & mask4;
702*adf37648SKyle Evans 	u64 a0_ = a0 - (0x7ffffffffffedLLU & mask);
703*adf37648SKyle Evans 	u64 a1_ = a1 - (0x7ffffffffffffLLU & mask);
704*adf37648SKyle Evans 	u64 a2_ = a2 - (0x7ffffffffffffLLU & mask);
705*adf37648SKyle Evans 	u64 a3_ = a3 - (0x7ffffffffffffLLU & mask);
706*adf37648SKyle Evans 	u64 a4_ = a4 - (0x7ffffffffffffLLU & mask);
707*adf37648SKyle Evans 	input[0] = a0_;
708*adf37648SKyle Evans 	input[1] = a1_;
709*adf37648SKyle Evans 	input[2] = a2_;
710*adf37648SKyle Evans 	input[3] = a3_;
711*adf37648SKyle Evans 	input[4] = a4_;
712*adf37648SKyle Evans }
713*adf37648SKyle Evans 
format_fcontract_store(u8 * output,u64 * input)714*adf37648SKyle Evans static __always_inline void format_fcontract_store(u8 *output, u64 *input)
715*adf37648SKyle Evans {
716*adf37648SKyle Evans 	u64 t0 = input[0];
717*adf37648SKyle Evans 	u64 t1 = input[1];
718*adf37648SKyle Evans 	u64 t2 = input[2];
719*adf37648SKyle Evans 	u64 t3 = input[3];
720*adf37648SKyle Evans 	u64 t4 = input[4];
721*adf37648SKyle Evans 	u64 o0 = t1 << 51 | t0;
722*adf37648SKyle Evans 	u64 o1 = t2 << 38 | t1 >> 13;
723*adf37648SKyle Evans 	u64 o2 = t3 << 25 | t2 >> 26;
724*adf37648SKyle Evans 	u64 o3 = t4 << 12 | t3 >> 39;
725*adf37648SKyle Evans 	u8 *b0 = output;
726*adf37648SKyle Evans 	u8 *b1 = output + 8;
727*adf37648SKyle Evans 	u8 *b2 = output + 16;
728*adf37648SKyle Evans 	u8 *b3 = output + 24;
729*adf37648SKyle Evans 	put_unaligned_le64(o0, b0);
730*adf37648SKyle Evans 	put_unaligned_le64(o1, b1);
731*adf37648SKyle Evans 	put_unaligned_le64(o2, b2);
732*adf37648SKyle Evans 	put_unaligned_le64(o3, b3);
733*adf37648SKyle Evans }
734*adf37648SKyle Evans 
format_fcontract(u8 * output,u64 * input)735*adf37648SKyle Evans static __always_inline void format_fcontract(u8 *output, u64 *input)
736*adf37648SKyle Evans {
737*adf37648SKyle Evans 	format_fcontract_first_carry_full(input);
738*adf37648SKyle Evans 	format_fcontract_second_carry_full(input);
739*adf37648SKyle Evans 	format_fcontract_trim(input);
740*adf37648SKyle Evans 	format_fcontract_store(output, input);
741*adf37648SKyle Evans }
742*adf37648SKyle Evans 
format_scalar_of_point(u8 * scalar,u64 * point)743*adf37648SKyle Evans static __always_inline void format_scalar_of_point(u8 *scalar, u64 *point)
744*adf37648SKyle Evans {
745*adf37648SKyle Evans 	u64 *x = point;
746*adf37648SKyle Evans 	u64 *z = point + 5;
747*adf37648SKyle Evans 	u64 buf[10] __aligned(32) = { 0 };
748*adf37648SKyle Evans 	u64 *zmone = buf;
749*adf37648SKyle Evans 	u64 *sc = buf + 5;
750*adf37648SKyle Evans 	crecip(zmone, z);
751*adf37648SKyle Evans 	fmul(sc, x, zmone);
752*adf37648SKyle Evans 	format_fcontract(scalar, sc);
753*adf37648SKyle Evans }
754*adf37648SKyle Evans 
curve25519_generic(u8 mypublic[CURVE25519_KEY_SIZE],const u8 secret[CURVE25519_KEY_SIZE],const u8 basepoint[CURVE25519_KEY_SIZE])755*adf37648SKyle Evans static void curve25519_generic(u8 mypublic[CURVE25519_KEY_SIZE],
756*adf37648SKyle Evans 			       const u8 secret[CURVE25519_KEY_SIZE],
757*adf37648SKyle Evans 			       const u8 basepoint[CURVE25519_KEY_SIZE])
758*adf37648SKyle Evans {
759*adf37648SKyle Evans 	u64 buf0[10] __aligned(32) = { 0 };
760*adf37648SKyle Evans 	u64 *x0 = buf0;
761*adf37648SKyle Evans 	u64 *z = buf0 + 5;
762*adf37648SKyle Evans 	u64 *q;
763*adf37648SKyle Evans 	format_fexpand(x0, basepoint);
764*adf37648SKyle Evans 	z[0] = 1;
765*adf37648SKyle Evans 	q = buf0;
766*adf37648SKyle Evans 	{
767*adf37648SKyle Evans 		u8 e[32] __aligned(32) = { 0 };
768*adf37648SKyle Evans 		u8 *scalar;
769*adf37648SKyle Evans 		memcpy(e, secret, 32);
770*adf37648SKyle Evans 		curve25519_clamp_secret(e);
771*adf37648SKyle Evans 		scalar = e;
772*adf37648SKyle Evans 		{
773*adf37648SKyle Evans 			u64 buf[15] = { 0 };
774*adf37648SKyle Evans 			u64 *nq = buf;
775*adf37648SKyle Evans 			u64 *x = nq;
776*adf37648SKyle Evans 			x[0] = 1;
777*adf37648SKyle Evans 			ladder_cmult(nq, scalar, q);
778*adf37648SKyle Evans 			format_scalar_of_point(mypublic, nq);
779*adf37648SKyle Evans 			memzero_explicit(buf, sizeof(buf));
780*adf37648SKyle Evans 		}
781*adf37648SKyle Evans 		memzero_explicit(e, sizeof(e));
782*adf37648SKyle Evans 	}
783*adf37648SKyle Evans 	memzero_explicit(buf0, sizeof(buf0));
784*adf37648SKyle Evans }
785