Merge and close elliptic curve branch.
[u/mdw/catacomb] / ec-prime.c
1 /* -*-c-*-
2 *
3 * $Id: ec-prime.c,v 1.4 2004/03/21 22:52:06 mdw Exp $
4 *
5 * Elliptic curves over prime fields
6 *
7 * (c) 2001 Straylight/Edgeware
8 */
9
10 /*----- Licensing notice --------------------------------------------------*
11 *
12 * This file is part of Catacomb.
13 *
14 * Catacomb is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU Library General Public License as
16 * published by the Free Software Foundation; either version 2 of the
17 * License, or (at your option) any later version.
18 *
19 * Catacomb is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU Library General Public License for more details.
23 *
24 * You should have received a copy of the GNU Library General Public
25 * License along with Catacomb; if not, write to the Free
26 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
27 * MA 02111-1307, USA.
28 */
29
30 /*----- Revision history --------------------------------------------------*
31 *
32 * $Log: ec-prime.c,v $
33 * Revision 1.4 2004/03/21 22:52:06 mdw
34 * Merge and close elliptic curve branch.
35 *
36 * Revision 1.3.4.3 2004/03/21 22:39:46 mdw
37 * Elliptic curves on binary fields work.
38 *
39 * Revision 1.3.4.2 2004/03/20 00:13:31 mdw
40 * Projective coordinates for prime curves
41 *
42 * Revision 1.3.4.1 2003/06/10 13:43:53 mdw
43 * Simple (non-projective) curves over prime fields now seem to work.
44 *
45 * Revision 1.3 2003/05/15 23:25:59 mdw
46 * Make elliptic curve stuff build.
47 *
48 * Revision 1.2 2002/01/13 13:48:44 mdw
49 * Further progress.
50 *
51 * Revision 1.1 2001/04/29 18:12:33 mdw
52 * Prototype version.
53 *
54 */
55
56 /*----- Header files ------------------------------------------------------*/
57
58 #include <mLib/sub.h>
59
60 #include "ec.h"
61
62 /*----- Data structures ---------------------------------------------------*/
63
64 typedef struct ecctx {
65 ec_curve c;
66 mp *a, *b;
67 } ecctx;
68
69 /*----- Simple prime curves -----------------------------------------------*/
70
71 static const ec_ops ec_primeops, ec_primeprojops, ec_primeprojxops;
72
73 static ec *ecneg(ec_curve *c, ec *d, const ec *p)
74 {
75 EC_COPY(d, p);
76 if (d->y)
77 d->y = F_NEG(c->f, d->y, d->y);
78 return (d);
79 }
80
81 static ec *ecfind(ec_curve *c, ec *d, mp *x)
82 {
83 mp *p, *q;
84 ecctx *cc = (ecctx *)c;
85 field *f = c->f;
86
87 q = F_SQR(f, MP_NEW, x);
88 p = F_MUL(f, MP_NEW, x, q);
89 q = F_MUL(f, q, x, cc->a);
90 p = F_ADD(f, p, p, q);
91 p = F_ADD(f, p, p, cc->b);
92 MP_DROP(q);
93 p = F_SQRT(f, p, p);
94 if (!p)
95 return (0);
96 EC_DESTROY(d);
97 d->x = MP_COPY(x);
98 d->y = p;
99 d->z = MP_COPY(f->one);
100 return (d);
101 }
102
103 static ec *ecdbl(ec_curve *c, ec *d, const ec *a)
104 {
105 if (EC_ATINF(a))
106 EC_SETINF(d);
107 else if (F_ZEROP(c->f, a->y))
108 EC_COPY(d, a);
109 else {
110 field *f = c->f;
111 ecctx *cc = (ecctx *)c;
112 mp *lambda;
113 mp *dy, *dx;
114
115 dx = F_SQR(f, MP_NEW, a->x); /* %$x^2$% */
116 dy = F_DBL(f, MP_NEW, a->y); /* %$2 y$% */
117 dx = F_TPL(f, dx, dx); /* %$3 x^2$% */
118 dx = F_ADD(f, dx, dx, cc->a); /* %$3 x^2 + A$% */
119 dy = F_INV(f, dy, dy); /* %$(2 y)^{-1}$% */
120 lambda = F_MUL(f, MP_NEW, dx, dy); /* %$\lambda = (3 x^2 + A)/(2 y)$% */
121
122 dx = F_SQR(f, dx, lambda); /* %$\lambda^2$% */
123 dy = F_DBL(f, dy, a->x); /* %$2 x$% */
124 dx = F_SUB(f, dx, dx, dy); /* %$x' = \lambda^2 - 2 x */
125 dy = F_SUB(f, dy, a->x, dx); /* %$x - x'$% */
126 dy = F_MUL(f, dy, lambda, dy); /* %$\lambda (x - x')$% */
127 dy = F_SUB(f, dy, dy, a->y); /* %$y' = \lambda (x - x') - y$% */
128
129 EC_DESTROY(d);
130 d->x = dx;
131 d->y = dy;
132 d->z = 0;
133 MP_DROP(lambda);
134 }
135 return (d);
136 }
137
138 static ec *ecprojdbl(ec_curve *c, ec *d, const ec *a)
139 {
140 if (EC_ATINF(a))
141 EC_SETINF(d);
142 else if (F_ZEROP(c->f, a->y))
143 EC_COPY(d, a);
144 else {
145 field *f = c->f;
146 ecctx *cc = (ecctx *)c;
147 mp *p, *q, *m, *s, *dx, *dy, *dz;
148
149 p = F_SQR(f, MP_NEW, a->z); /* %$z^2$% */
150 q = F_SQR(f, MP_NEW, p); /* %$z^4$% */
151 p = F_MUL(f, p, q, cc->a); /* %$A z^4$% */
152 m = F_SQR(f, MP_NEW, a->x); /* %$x^2$% */
153 m = F_TPL(f, m, m); /* %$3 x^2$% */
154 m = F_ADD(f, m, m, p); /* %$m = 3 x^2 + A z^4$% */
155
156 q = F_DBL(f, q, a->y); /* %$2 y$% */
157 dz = F_MUL(f, MP_NEW, q, a->z); /* %$z' = 2 y z$% */
158
159 p = F_SQR(f, p, q); /* %$4 y^2$% */
160 s = F_MUL(f, MP_NEW, p, a->x); /* %$s = 4 x y^2$% */
161 q = F_SQR(f, q, p); /* %$16 y^4$% */
162 q = F_HLV(f, q, q); /* %$t = 8 y^4$% */
163
164 p = F_DBL(f, p, s); /* %$2 s$% */
165 dx = F_SQR(f, MP_NEW, m); /* %$m^2$% */
166 dx = F_SUB(f, dx, dx, p); /* %$x' = m^2 - 2 s$% */
167
168 s = F_SUB(f, s, s, dx); /* %$s - x'$% */
169 dy = F_MUL(f, p, m, s); /* %$m (s - x')$% */
170 dy = F_SUB(f, dy, dy, q); /* %$y' = m (s - x') - t$% */
171
172 EC_DESTROY(d);
173 d->x = dx;
174 d->y = dy;
175 d->z = dz;
176 MP_DROP(m);
177 MP_DROP(q);
178 MP_DROP(s);
179 }
180 return (d);
181 }
182
183 static ec *ecprojxdbl(ec_curve *c, ec *d, const ec *a)
184 {
185 if (EC_ATINF(a))
186 EC_SETINF(d);
187 else if (F_ZEROP(c->f, a->y))
188 EC_COPY(d, a);
189 else {
190 field *f = c->f;
191 mp *p, *q, *m, *s, *dx, *dy, *dz;
192
193 m = F_SQR(f, MP_NEW, a->z); /* %$z^2$% */
194 p = F_SUB(f, MP_NEW, a->x, m); /* %$x - z^2$% */
195 q = F_ADD(f, MP_NEW, a->x, m); /* %$x + z^2$% */
196 m = F_MUL(f, m, p, q); /* %$x^2 - z^4$% */
197 m = F_TPL(f, m, m); /* %$m = 3 x^2 - 3 z^4$% */
198
199 q = F_DBL(f, q, a->y); /* %$2 y$% */
200 dz = F_MUL(f, MP_NEW, q, a->z); /* %$z' = 2 y z$% */
201
202 p = F_SQR(f, p, q); /* %$4 y^2$% */
203 s = F_MUL(f, MP_NEW, p, a->x); /* %$s = 4 x y^2$% */
204 q = F_SQR(f, q, p); /* %$16 y^4$% */
205 q = F_HLV(f, q, q); /* %$t = 8 y^4$% */
206
207 p = F_DBL(f, p, s); /* %$2 s$% */
208 dx = F_SQR(f, MP_NEW, m); /* %$m^2$% */
209 dx = F_SUB(f, dx, dx, p); /* %$x' = m^2 - 2 s$% */
210
211 s = F_SUB(f, s, s, dx); /* %$s - x'$% */
212 dy = F_MUL(f, p, m, s); /* %$m (s - x')$% */
213 dy = F_SUB(f, dy, dy, q); /* %$y' = m (s - x') - t$% */
214
215 EC_DESTROY(d);
216 d->x = dx;
217 d->y = dy;
218 d->z = dz;
219 MP_DROP(m);
220 MP_DROP(q);
221 MP_DROP(s);
222 }
223 return (d);
224 }
225
226 static ec *ecadd(ec_curve *c, ec *d, const ec *a, const ec *b)
227 {
228 if (a == b)
229 ecdbl(c, d, a);
230 else if (EC_ATINF(a))
231 EC_COPY(d, b);
232 else if (EC_ATINF(b))
233 EC_COPY(d, a);
234 else {
235 field *f = c->f;
236 mp *lambda;
237 mp *dy, *dx;
238
239 if (!MP_EQ(a->x, b->x)) {
240 dy = F_SUB(f, MP_NEW, a->y, b->y); /* %$y_0 - y_1$% */
241 dx = F_SUB(f, MP_NEW, a->x, b->x); /* %$x_0 - x_1$% */
242 dx = F_INV(f, dx, dx); /* %$(x_0 - x_1)^{-1}$% */
243 lambda = F_MUL(f, MP_NEW, dy, dx);
244 /* %$\lambda = (y_0 - y1)/(x_0 - x_1)$% */
245 } else if (F_ZEROP(c->f, a->y) || !MP_EQ(a->y, b->y)) {
246 EC_SETINF(d);
247 return (d);
248 } else {
249 ecctx *cc = (ecctx *)c;
250 dx = F_SQR(f, MP_NEW, a->x); /* %$x_0^2$% */
251 dx = F_TPL(f, dx, dx); /* %$3 x_0^2$% */
252 dx = F_ADD(f, dx, dx, cc->a); /* %$3 x_0^2 + A$% */
253 dy = F_DBL(f, MP_NEW, a->y); /* %$2 y_0$% */
254 dy = F_INV(f, dy, dy); /* %$(2 y_0)^{-1}$% */
255 lambda = F_MUL(f, MP_NEW, dx, dy);
256 /* %$\lambda = (3 x_0^2 + A)/(2 y_0)$% */
257 }
258
259 dx = F_SQR(f, dx, lambda); /* %$\lambda^2$% */
260 dx = F_SUB(f, dx, dx, a->x); /* %$\lambda^2 - x_0$% */
261 dx = F_SUB(f, dx, dx, b->x); /* %$x' = \lambda^2 - x_0 - x_1$% */
262 dy = F_SUB(f, dy, b->x, dx); /* %$x_1 - x'$% */
263 dy = F_MUL(f, dy, lambda, dy); /* %$\lambda (x_1 - x')$% */
264 dy = F_SUB(f, dy, dy, b->y); /* %$y' = \lambda (x_1 - x') - y_1$% */
265
266 EC_DESTROY(d);
267 d->x = dx;
268 d->y = dy;
269 d->z = 0;
270 MP_DROP(lambda);
271 }
272 return (d);
273 }
274
275 static ec *ecprojadd(ec_curve *c, ec *d, const ec *a, const ec *b)
276 {
277 if (a == b)
278 c->ops->dbl(c, d, a);
279 else if (EC_ATINF(a))
280 EC_COPY(d, b);
281 else if (EC_ATINF(b))
282 EC_COPY(d, a);
283 else {
284 field *f = c->f;
285 mp *p, *q, *r, *w, *u, *s, *dx, *dy, *dz;
286
287 q = F_SQR(f, MP_NEW, a->z); /* %$z_0^2$% */
288 u = F_MUL(f, MP_NEW, q, b->x); /* %$u = x_1 z_0^2$% */
289 p = F_MUL(f, MP_NEW, q, b->y); /* %$y_1 z_0^2$% */
290 s = F_MUL(f, q, p, a->z); /* %$s = y_1 z_0^3$% */
291
292 w = F_SUB(f, p, a->x, u); /* %$w = x_0 - u$% */
293 r = F_SUB(f, MP_NEW, a->y, s); /* %$r = y_0 - s$% */
294 if (F_ZEROP(f, w)) {
295 MP_DROP(w);
296 MP_DROP(u);
297 MP_DROP(s);
298 if (F_ZEROP(f, r)) {
299 MP_DROP(r);
300 return (c->ops->dbl(c, d, a));
301 } else {
302 MP_DROP(r);
303 EC_SETINF(d);
304 return (d);
305 }
306 }
307 u = F_ADD(f, u, u, a->x); /* %$t = x_0 + u$% */
308 s = F_ADD(f, s, s, a->y); /* %$m = y_0 + r$% */
309
310 dz = F_MUL(f, MP_NEW, a->z, w); /* %$z' = z_0 w$% */
311
312 p = F_SQR(f, MP_NEW, w); /* %$w^2$% */
313 q = F_MUL(f, MP_NEW, p, u); /* %$t w^2$% */
314 u = F_MUL(f, u, p, w); /* %$w^3$% */
315 p = F_MUL(f, p, u, s); /* %$m w^3$% */
316
317 dx = F_SQR(f, u, r); /* %$r^2$% */
318 dx = F_SUB(f, dx, dx, q); /* %$x' = r^2 - t w^2$% */
319
320 s = F_DBL(f, s, dx); /* %$2 x'$% */
321 q = F_SUB(f, q, q, s); /* %$v = t w^2 - 2 x'$% */
322 dy = F_MUL(f, s, q, r); /* %$v r$% */
323 dy = F_SUB(f, dy, dy, p); /* %$v r - m w^3$% */
324 dy = F_HLV(f, dy, dy); /* %$y' = (v r - m w^3)/2$% */
325
326 EC_DESTROY(d);
327 d->x = dx;
328 d->y = dy;
329 d->z = dz;
330 MP_DROP(p);
331 MP_DROP(q);
332 MP_DROP(r);
333 MP_DROP(w);
334 }
335 return (d);
336 }
337
338 static int eccheck(ec_curve *c, const ec *p)
339 {
340 ecctx *cc = (ecctx *)c;
341 field *f = c->f;
342 int rc;
343 mp *l = F_SQR(f, MP_NEW, p->y);
344 mp *x = F_SQR(f, MP_NEW, p->x);
345 mp *r = F_MUL(f, MP_NEW, x, p->x);
346 x = F_MUL(f, x, cc->a, p->x);
347 r = F_ADD(f, r, r, x);
348 r = F_ADD(f, r, r, cc->b);
349 rc = MP_EQ(l, r) ? 0 : -1;
350 mp_drop(l);
351 mp_drop(x);
352 mp_drop(r);
353 return (rc);
354 }
355
356 static int ecprojcheck(ec_curve *c, const ec *p)
357 {
358 ec t = EC_INIT;
359 int rc;
360
361 c->ops->fix(c, &t, p);
362 rc = eccheck(c, &t);
363 EC_DESTROY(&t);
364 return (rc);
365 }
366
367 static void ecdestroy(ec_curve *c)
368 {
369 ecctx *cc = (ecctx *)c;
370 MP_DROP(cc->a);
371 MP_DROP(cc->b);
372 DESTROY(cc);
373 }
374
375 /* --- @ec_prime@, @ec_primeproj@ --- *
376 *
377 * Arguments: @field *f@ = the underlying field for this elliptic curve
378 * @mp *a, *b@ = the coefficients for this curve
379 *
380 * Returns: A pointer to the curve.
381 *
382 * Use: Creates a curve structure for an elliptic curve defined over
383 * a prime field. The @primeproj@ variant uses projective
384 * coordinates, which can be a win.
385 */
386
387 extern ec_curve *ec_prime(field *f, mp *a, mp *b)
388 {
389 ecctx *cc = CREATE(ecctx);
390 cc->c.ops = &ec_primeops;
391 cc->c.f = f;
392 cc->a = F_IN(f, MP_NEW, a);
393 cc->b = F_IN(f, MP_NEW, b);
394 return (&cc->c);
395 }
396
397 extern ec_curve *ec_primeproj(field *f, mp *a, mp *b)
398 {
399 ecctx *cc = CREATE(ecctx);
400 mp *ax;
401
402 ax = mp_add(MP_NEW, a, MP_THREE);
403 ax = F_IN(f, ax, ax);
404 if (F_ZEROP(f, ax))
405 cc->c.ops = &ec_primeprojxops;
406 else
407 cc->c.ops = &ec_primeprojops;
408 MP_DROP(ax);
409 cc->c.f = f;
410 cc->a = F_IN(f, MP_NEW, a);
411 cc->b = F_IN(f, MP_NEW, b);
412 return (&cc->c);
413 }
414
415 static const ec_ops ec_primeops = {
416 ecdestroy, ec_idin, ec_idout, ec_idfix,
417 0, ecneg, ecadd, ec_stdsub, ecdbl, eccheck
418 };
419
420 static const ec_ops ec_primeprojops = {
421 ecdestroy, ec_projin, ec_projout, ec_projfix,
422 0, ecneg, ecprojadd, ec_stdsub, ecprojdbl, ecprojcheck
423 };
424
425 static const ec_ops ec_primeprojxops = {
426 ecdestroy, ec_projin, ec_projout, ec_projfix,
427 0, ecneg, ecprojadd, ec_stdsub, ecprojxdbl, ecprojcheck
428 };
429
430 /*----- Test rig ----------------------------------------------------------*/
431
432 #ifdef TEST_RIG
433
434 #define MP(x) mp_readstring(MP_NEW, #x, 0, 0)
435
436 int main(int argc, char *argv[])
437 {
438 field *f;
439 ec_curve *c;
440 ec g = EC_INIT, d = EC_INIT;
441 mp *p, *a, *b, *r;
442 int i, n = argc == 1 ? 1 : atoi(argv[1]);
443
444 printf("ec-prime: ");
445 fflush(stdout);
446 a = MP(-3);
447 b = MP(0x64210519e59c80e70fa7e9ab72243049feb8deecc146b9b1);
448 p = MP(6277101735386680763835789423207666416083908700390324961279);
449 r = MP(6277101735386680763835789423176059013767194773182842284080);
450
451 f = field_prime(p);
452 c = ec_primeproj(f, a, b);
453
454 g.x = MP(0x188da80eb03090f67cbf20eb43a18800f4ff0afd82ff1012);
455 g.y = MP(0x07192b95ffc8da78631011ed6b24cdd573f977a11e794811);
456
457 for (i = 0; i < n; i++) {
458 ec_mul(c, &d, &g, r);
459 if (EC_ATINF(&d)) {
460 fprintf(stderr, "zero too early\n");
461 return (1);
462 }
463 ec_add(c, &d, &d, &g);
464 if (!EC_ATINF(&d)) {
465 fprintf(stderr, "didn't reach zero\n");
466 MP_EPRINT("d.x", d.x);
467 MP_EPRINT("d.y", d.y);
468 return (1);
469 }
470 ec_destroy(&d);
471 }
472 ec_destroy(&g);
473 ec_destroycurve(c);
474 F_DESTROY(f);
475 MP_DROP(p); MP_DROP(a); MP_DROP(b); MP_DROP(r);
476 assert(!mparena_count(&mparena_global));
477 printf("ok\n");
478 return (0);
479 }
480
481 #endif
482
483 /*----- That's all, folks -------------------------------------------------*/