-#include "x25519.h"
-
-#define FOR(i, start, end) for (size_t i = start; i < end; i++)
-#define sv static void
-typedef int64_t gf[16];
+// Simplified version of the the ref10 implementation.
+// Its bigger than TweetNaCl, but nearly twice as fast.
-static const uint8_t _0[16];
-static const uint8_t _9[32] = { 9 };
-static const gf _121665 = { 0xDB41, 1 };
+#include "x25519.h"
-/* static int vn(const uint8_t *x, const uint8_t *y, size_t n) */
-/* { */
-/* uint32_t d = 0; */
-/* FOR(i, 0, n) d |= x[i] ^ y[i]; */
-/* return (1 & ((d - 1) >> 8)) - 1; */
-/* } */
+typedef int32_t fe[10];
+void fe_0(fe h){ for (int i = 0; i < 10; i++) h[i] = 0; }
+void fe_1(fe h){ h[0] = 1; for (int i = 1; i < 10; i++) h[i] = 0; }
-// needed for signatures (not here)
-/* sv set_25519(gf r, const gf a) */
-/* { */
-/* FOR(i, 0, 16) r[i] = a[i]; */
-/* } */
+void fe_add (fe h, fe f, fe g) { for (int i = 0; i < 10; i++) h[i] = f[i] + g[i];}
+void fe_sub (fe h, fe f, fe g) { for (int i = 0; i < 10; i++) h[i] = f[i] - g[i];}
+void fe_copy(fe h, fe f ) { for (int i = 0; i < 10; i++) h[i] = f[i]; }
-sv car_25519(gf o)
+void fe_cswap(fe f, fe g, unsigned int b)
{
- FOR(i, 0, 16) {
- o[i] += 1LL << 16;
- int64_t c = o[i] >> 16;
- o[(i+1) * (i<15)] += c - 1 + (37 * (c-1) * (i==15));
- o[i] -= c << 16;
+ for (int i = 0; i < 10; i++){
+ int32_t x = (f[i] ^ g[i]) & -b;
+ f[i] = f[i] ^ x;
+ g[i] = g[i] ^ x;
}
}
-sv sel_25519(gf p, gf q, int b)
+static uint64_t load_3(const unsigned char *in)
{
- int64_t c = ~(b-1);
- FOR(i, 0, 16) {
- int64_t t = c & (p[i] ^ q[i]);
- p[i] ^= t;
- q[i] ^= t;
- }
+ uint64_t result;
+ result = (uint64_t) in[0];
+ result |= ((uint64_t) in[1]) << 8;
+ result |= ((uint64_t) in[2]) << 16;
+ return result;
}
-sv pack_25519(uint8_t *o, const gf n)
+static uint64_t load_4(const unsigned char *in)
{
- gf t;
- FOR(i, 0, 16) t[i] = n[i];
- car_25519(t);
- car_25519(t);
- car_25519(t);
- FOR(j, 0, 2) {
- gf m;
- m[0] = t[0] - 0xffed;
- FOR(i, 1, 15) {
- m[i ] = t[i] - 0xffff - ((m[i-1] >> 16) & 1);
- m[i-1] &= 0xffff;
- }
- m[15] = t[15] - 0x7fff - ((m[14] >> 16) & 1);
- int b = (m[15] >> 16) & 1;
- m[14] &= 0xffff;
- sel_25519(t, m, 1-b);
- }
- FOR(i, 0, 16) {
- o[2*i ] = t[i] & 0xff;
- o[2*i + 1] = t[i] >> 8;
- }
+ uint64_t result;
+ result = (uint64_t) in[0];
+ result |= ((uint64_t) in[1]) << 8;
+ result |= ((uint64_t) in[2]) << 16;
+ result |= ((uint64_t) in[3]) << 24;
+ return result;
}
-// needed for signatures (not here)
-/* static int neq_25519(const gf a, const gf b) */
-/* { */
-/* uint8_t c[32],d[32]; */
-/* pack_25519(c, a); */
-/* pack_25519(d, b); */
-/* return vn(c, d, 32); */
-/* } */
-// needed for signatures (not here)
-/* static uint8_t par_25519(const gf a) */
-/* { */
-/* uint8_t d[32]; */
-/* pack_25519(d, a); */
-/* return d[0] & 1; */
-/* } */
-
-sv unpack_25519(gf o, const uint8_t *n)
+void fe_carry(fe h, int64_t t[10])
{
- FOR(i, 0, 16) o[i] = n[2*i] + ((int64_t)n[2*i + 1] << 8);
- o[15] &= 0x7fff;
+ int64_t c[10]; // carry
+ c[9] = (t[9] + (1ll << 24)) >> 25; t[0] += c[9] * 19; t[9] -= c[9] << 25;
+ c[1] = (t[1] + (1ll << 24)) >> 25; t[2] += c[1]; t[1] -= c[1] << 25;
+ c[3] = (t[3] + (1ll << 24)) >> 25; t[4] += c[3]; t[3] -= c[3] << 25;
+ c[5] = (t[5] + (1ll << 24)) >> 25; t[6] += c[5]; t[5] -= c[5] << 25;
+ c[7] = (t[7] + (1ll << 24)) >> 25; t[8] += c[7]; t[7] -= c[7] << 25;
+ c[0] = (t[0] + (1ll << 25)) >> 26; t[1] += c[0]; t[0] -= c[0] << 26;
+ c[2] = (t[2] + (1ll << 25)) >> 26; t[3] += c[2]; t[2] -= c[2] << 26;
+ c[4] = (t[4] + (1ll << 25)) >> 26; t[5] += c[4]; t[4] -= c[4] << 26;
+ c[6] = (t[6] + (1ll << 25)) >> 26; t[7] += c[6]; t[6] -= c[6] << 26;
+ c[8] = (t[8] + (1ll << 25)) >> 26; t[9] += c[8]; t[8] -= c[8] << 26;
+ for (int i = 0; i < 10; i++)
+ h[i] = t[i];
}
-sv A(gf o, const gf a, const gf b)
+void fe_frombytes(fe h, const unsigned char *s)
{
- FOR(i, 0, 16) o[i] = a[i] + b[i];
+ int64_t t[10]; // intermediate result (may overflow 32 bits)
+ t[0] = load_4(s);
+ t[1] = load_3(s + 4) << 6;
+ t[2] = load_3(s + 7) << 5;
+ t[3] = load_3(s + 10) << 3;
+ t[4] = load_3(s + 13) << 2;
+ t[5] = load_4(s + 16);
+ t[6] = load_3(s + 20) << 7;
+ t[7] = load_3(s + 23) << 5;
+ t[8] = load_3(s + 26) << 4;
+ t[9] = (load_3(s + 29) & 8388607) << 2;
+ fe_carry(h, t);
}
-sv Z(gf o, const gf a, const gf b)
+void fe_mul121666(fe h,fe f)
{
- FOR(i, 0, 16) o[i] = a[i] - b[i];
+ int64_t t[10];
+ for (int i = 0; i < 10; i++)
+ t[i] = f[i] * (int64_t) 121666;
+ fe_carry(h, t);
}
-sv M(gf o, const gf a, const gf b)
+// Rolled version of ref10. A bit slower, but much simpler.
+void fe_mul(fe h,fe f,fe g)
{
- int64_t t[31];
- FOR(i, 0, 31) t[i] = 0;
- FOR(i, 0, 16) FOR(j, 0, 16) t[i+j] += a[i] * b[j];
- FOR(i, 0, 15) t[i] += 38 * t[i+16];
- FOR(i, 0, 16) o[i] = t[i];
- car_25519(o);
- car_25519(o);
+ int64_t t[10];
+ for (int i = 0; i < 10; i++) {
+ int64_t acc = 0;
+ int64_t even_i = 1 - (i & 1);
+ for (int j = 0; j < i+1; j++) {
+ int64_t m1_2 = (even_i & j) + 1;
+ acc += (f[j] * m1_2) * (int64_t) g[i-j];
+ }
+ for (int j = i+1; j < 10; j++) {
+ int64_t m1_2 = (even_i & j) + 1;
+ acc += (f[j] * m1_2) * (int64_t) (g[i-j+10] * 19);
+ }
+ t[i] = acc;
+ }
+
+ int64_t c[10]; // carry
+ c[0] = (t[0] + (1ll << 25)) >> 26; t[1] += c[0]; t[0] -= c[0] << 26;
+ c[4] = (t[4] + (1ll << 25)) >> 26; t[5] += c[4]; t[4] -= c[4] << 26;
+ c[1] = (t[1] + (1ll << 24)) >> 25; t[2] += c[1]; t[1] -= c[1] << 25;
+ c[5] = (t[5] + (1ll << 24)) >> 25; t[6] += c[5]; t[5] -= c[5] << 25;
+ c[2] = (t[2] + (1ll << 25)) >> 26; t[3] += c[2]; t[2] -= c[2] << 26;
+ c[6] = (t[6] + (1ll << 25)) >> 26; t[7] += c[6]; t[6] -= c[6] << 26;
+ c[3] = (t[3] + (1ll << 24)) >> 25; t[4] += c[3]; t[3] -= c[3] << 25;
+ c[7] = (t[7] + (1ll << 24)) >> 25; t[8] += c[7]; t[7] -= c[7] << 25;
+ c[4] = (t[4] + (1ll << 25)) >> 26; t[5] += c[4]; t[4] -= c[4] << 26;
+ c[8] = (t[8] + (1ll << 25)) >> 26; t[9] += c[8]; t[8] -= c[8] << 26;
+ c[9] = (t[9] + (1ll << 24)) >> 25; t[0] += c[9] * 19; t[9] -= c[9] << 25;
+ c[0] = (t[0] + (1ll << 25)) >> 26; t[1] += c[0]; t[0] -= c[0] << 26;
+
+ for (int i = 0; i < 10; i++)
+ h[i] = t[i];
}
-sv S(gf o,const gf a)
+// The specialised square function from ref10 is much faster, but takes
+// way too much source code.
+// Replace with the original ref10 implementation for 30% speedup.
+void fe_sq(fe h, fe f) { fe_mul(h, f, f); }
+
+void fe_invert(fe out,fe z)
{
- M(o, a, a);
+ fe t0, t1, t2, t3;
+ fe_sq(t0, z); for (int i = 1; i < 1; i++) fe_sq(t0,t0);
+ fe_sq(t1, t0); for (int i = 1; i < 2; i++) fe_sq(t1,t1);
+ fe_mul(t1, z, t1);
+ fe_mul(t0, t0, t1);
+ fe_sq(t2, t0); for (int i = 1; i < 1; i++) fe_sq(t2,t2); fe_mul(t1 , t1, t2);
+ fe_sq(t2, t1); for (int i = 1; i < 5; i++) fe_sq(t2,t2); fe_mul(t1 , t2, t1);
+ fe_sq(t2, t1); for (int i = 1; i < 10; i++) fe_sq(t2,t2); fe_mul(t2 , t2, t1);
+ fe_sq(t3, t2); for (int i = 1; i < 20; i++) fe_sq(t3,t3); fe_mul(t2 , t3, t2);
+ fe_sq(t2, t2); for (int i = 1; i < 10; i++) fe_sq(t2,t2); fe_mul(t1 , t2, t1);
+ fe_sq(t2, t1); for (int i = 1; i < 50; i++) fe_sq(t2,t2); fe_mul(t2 , t2, t1);
+ fe_sq(t3, t2); for (int i = 1; i < 100; i++) fe_sq(t3,t3); fe_mul(t2 , t3, t2);
+ fe_sq(t2, t2); for (int i = 1; i < 50; i++) fe_sq(t2,t2); fe_mul(t1 , t2, t1);
+ fe_sq(t1, t1); for (int i = 1; i < 5; i++) fe_sq(t1,t1); fe_mul(out, t1, t0);
}
-sv inv_25519(gf o,const gf i)
+void fe_tobytes(unsigned char *s, fe h)
{
- gf c;
- FOR(a, 0, 16) c[a] = i[a];
- for(int a = 253; a >= 0; a--) {
- S(c, c);
- if(a != 2 && a != 4)
- M(c, c, i);
+ int32_t t[11];
+ for (int i = 0; i < 10; i++)
+ t[i] = h[i];
+
+ int32_t q = (19 * t[9] + (((int32_t) 1) << 24)) >> 25;
+ for (int i = 0; i < 10; i += 2) {
+ q += t[i ]; q >>= 26;
+ q += t[i+1]; q >>= 25;
+ }
+
+ // Goal: Output t-(2^255-19)q, which is between 0 and 2^255-20.
+ t[0] += 19 * q;
+ // Goal: Output t-2^255 q, which is between 0 and 2^255-20.
+ int32_t carry[10];
+ for (int i = 0; i < 10; i += 2) {
+ carry[i ] = t [i ] >> 26;
+ t [i+1] += carry[i ] ;
+ t [i ] -= carry[i ] << 26;
+
+ carry[i+1] = t [i+1] >> 25;
+ t [i+2] += carry[i+1] ;
+ t [i+1] -= carry[i+1] << 25;
}
- FOR(a, 0, 16) o[a] = c[a];
+ // Goal: Output t0+...+2^255 t10-2^255 q, which is between 0 and 2^255-20.
+ // Have t0+...+2^230 t9 between 0 and 2^255-1;
+ // evidently 2^255 t10-2^255 q = 0.
+ // Goal: Output t0+...+2^230 t9.
+ s[ 0] = t[0] >> 0;
+ s[ 1] = t[0] >> 8;
+ s[ 2] = t[0] >> 16;
+ s[ 3] = (t[0] >> 24) | (t[1] << 2);
+ s[ 4] = t[1] >> 6;
+ s[ 5] = t[1] >> 14;
+ s[ 6] = (t[1] >> 22) | (t[2] << 3);
+ s[ 7] = t[2] >> 5;
+ s[ 8] = t[2] >> 13;
+ s[ 9] = (t[2] >> 21) | (t[3] << 5);
+ s[10] = t[3] >> 3;
+ s[11] = t[3] >> 11;
+ s[12] = (t[3] >> 19) | (t[4] << 6);
+ s[13] = t[4] >> 2;
+ s[14] = t[4] >> 10;
+ s[15] = t[4] >> 18;
+ s[16] = t[5] >> 0;
+ s[17] = t[5] >> 8;
+ s[18] = t[5] >> 16;
+ s[19] = (t[5] >> 24) | (t[6] << 1);
+ s[20] = t[6] >> 7;
+ s[21] = t[6] >> 15;
+ s[22] = (t[6] >> 23) | (t[7] << 3);
+ s[23] = t[7] >> 5;
+ s[24] = t[7] >> 13;
+ s[25] = (t[7] >> 21) | (t[8] << 4);
+ s[26] = t[8] >> 4;
+ s[27] = t[8] >> 12;
+ s[28] = (t[8] >> 20) | (t[9] << 6);
+ s[29] = t[9] >> 2;
+ s[30] = t[9] >> 10;
+ s[31] = t[9] >> 18;
}
-// needed for signatures (not here)
-/* sv pow2523(gf o,const gf i) */
-/* { */
-/* gf c; */
-/* FOR(a, 0, 16) c[a] = i[a]; */
-/* for(int a = 250; a >= 0; a--) { */
-/* S(c, c); */
-/* if(a != 1) M(c, c, i); */
-/* } */
-/* FOR(a, 0, 16) o[a] = c[a]; */
-/* } */
-
-void crypto_x25519(uint8_t q[32], const uint8_t n[32], const uint8_t p[32])
+
+void crypto_x25519(uint8_t shared_secret [32],
+ const uint8_t your_secret_key [32],
+ const uint8_t their_public_key[32])
{
- uint8_t z[32];
- int64_t x[80];
- int64_t r;
- gf a, b, c, d, e, f;
- FOR(i, 0, 31) z[i] = n[i];
- z[31] = (n[31] & 127) | 64;
- z[0 ] &= 248;
- unpack_25519(x, p);
- FOR(i, 0, 16) {
- b[i] = x[i];
- d[i] = a[i] = c[i] = 0;
- }
- a[0] = d[0] = 1;
- for(int i = 254; i>=0; i--) {
- r = (z[i>>3] >> (i & 7)) & 1;
- sel_25519(a, b, r);
- sel_25519(c, d, r);
- A(e, a, c);
- Z(a, a, c);
- A(c, b, d);
- Z(b, b, d);
- S(d, e);
- S(f, a);
- M(a, c, a);
- M(c, b, e);
- A(e, a, c);
- Z(a, a, c);
- S(b, a);
- Z(c, d, f);
- M(a, c, _121665);
- A(a, a, d);
- M(c, c, a);
- M(a, d, f);
- M(d, b, x);
- S(b, e);
- sel_25519(a, b, r);
- sel_25519(c, d, r);
- }
- FOR(i, 0, 16) {
- x[i+16] = a[i];
- x[i+32] = c[i];
- x[i+48] = b[i];
- x[i+64] = d[i];
+ unsigned char e[32];
+ unsigned int i;
+ fe x1;
+ fe x2;
+ fe z2;
+ fe x3;
+ fe z3;
+ fe tmp0;
+ fe tmp1;
+ int pos;
+ unsigned int swap;
+ unsigned int b;
+
+ for (i = 0;i < 32;++i) e[i] = your_secret_key[i];
+ e[0] &= 248;
+ e[31] &= 127;
+ e[31] |= 64;
+ fe_frombytes(x1,their_public_key);
+ fe_1(x2);
+ fe_0(z2);
+ fe_copy(x3,x1);
+ fe_1(z3);
+
+ swap = 0;
+ for (pos = 254;pos >= 0;--pos) {
+ b = e[pos / 8] >> (pos & 7);
+ b &= 1;
+ swap ^= b;
+ fe_cswap(x2,x3,swap);
+ fe_cswap(z2,z3,swap);
+ swap = b;
+ fe_sub(tmp0,x3,z3);
+ fe_sub(tmp1,x2,z2);
+ fe_add(x2,x2,z2);
+ fe_add(z2,x3,z3);
+ fe_mul(z3,tmp0,x2);
+ fe_mul(z2,z2,tmp1);
+ fe_sq(tmp0,tmp1);
+ fe_sq(tmp1,x2);
+ fe_add(x3,z3,z2);
+ fe_sub(z2,z3,z2);
+ fe_mul(x2,tmp1,tmp0);
+ fe_sub(tmp1,tmp1,tmp0);
+ fe_sq(z2,z2);
+ fe_mul121666(z3,tmp1);
+ fe_sq(x3,x3);
+ fe_add(tmp0,tmp0,z3);
+ fe_mul(z3,x1,z2);
+ fe_mul(z2,tmp1,tmp0);
}
- inv_25519(x+32, x+32);
- M(x+16, x+16, x+32);
- pack_25519(q, x+16);
+ fe_cswap(x2,x3,swap);
+ fe_cswap(z2,z3,swap);
+
+ fe_invert(z2, z2);
+ fe_mul(x2, x2, z2);
+ fe_tobytes(shared_secret,x2);
}
-void crypto_x25519_base(uint8_t q[32], const uint8_t n[32])
+void crypto_x25519_base(uint8_t public_key[32], const uint8_t secret_key[32])
{
- crypto_x25519(q, n, _9);
+ static const uint8_t base_point [32] = {9};
+ crypto_x25519(public_key, secret_key, base_point);
}