chnages done at the train
[gnupg.git] / cipher / elgamal.c
1 /* elgamal.c  -  ElGamal Public Key encryption
2  *      Copyright (C) 1998 Free Software Foundation, Inc.
3  *
4  * For a description of the algorithm, see:
5  *   Bruce Schneier: Applied Cryptography. John Wiley & Sons, 1996.
6  *   ISBN 0-471-11709-9. Pages 476 ff.
7  *
8  * This file is part of GNUPG.
9  *
10  * GNUPG is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * GNUPG is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
23  */
24
25 #include <config.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include "util.h"
30 #include "mpi.h"
31 #include "cipher.h"
32 #include "elgamal.h"
33
34 typedef struct {
35     MPI p;          /* prime */
36     MPI g;          /* group generator */
37     MPI y;          /* g^x mod p */
38 } ELG_public_key;
39
40
41 typedef struct {
42     MPI p;          /* prime */
43     MPI g;          /* group generator */
44     MPI y;          /* g^x mod p */
45     MPI x;          /* secret exponent */
46 } ELG_secret_key;
47
48
49 static void test_keys( ELG_secret_key *sk, unsigned nbits );
50 static MPI gen_k( MPI p );
51 static void generate( ELG_secret_key *sk, unsigned nbits, MPI **factors );
52 static int  check_secret_key( ELG_secret_key *sk );
53 static void encrypt(MPI a, MPI b, MPI input, ELG_public_key *pkey );
54 static void decrypt(MPI output, MPI a, MPI b, ELG_secret_key *skey );
55 static void sign(MPI a, MPI b, MPI input, ELG_secret_key *skey);
56 static int  verify(MPI a, MPI b, MPI input, ELG_public_key *pkey);
57
58
59 static void
60 test_keys( ELG_secret_key *sk, unsigned nbits )
61 {
62     ELG_public_key pk;
63     MPI test = mpi_alloc( 0 );
64     MPI out1_a = mpi_alloc( nbits / BITS_PER_MPI_LIMB );
65     MPI out1_b = mpi_alloc( nbits / BITS_PER_MPI_LIMB );
66     MPI out2 = mpi_alloc( nbits / BITS_PER_MPI_LIMB );
67
68     pk.p = sk->p;
69     pk.g = sk->g;
70     pk.y = sk->y;
71
72     /*mpi_set_bytes( test, nbits, get_random_byte, 0 );*/
73     {   char *p = get_random_bits( nbits, 0, 0 );
74         mpi_set_buffer( test, p, (nbits+7)/8, 0 );
75         m_free(p);
76     }
77
78     encrypt( out1_a, out1_b, test, &pk );
79     decrypt( out2, out1_a, out1_b, sk );
80     if( mpi_cmp( test, out2 ) )
81         log_fatal("ElGamal operation: encrypt, decrypt failed\n");
82
83     sign( out1_a, out1_b, test, sk );
84     if( !verify( out1_a, out1_b, test, &pk ) )
85         log_fatal("ElGamal operation: sign, verify failed\n");
86
87     mpi_free( test );
88     mpi_free( out1_a );
89     mpi_free( out1_b );
90     mpi_free( out2 );
91 }
92
93
94 /****************
95  * generate a random secret exponent k from prime p, so
96  * that k is relatively prime to p-1
97  */
98 static MPI
99 gen_k( MPI p )
100 {
101     MPI k = mpi_alloc_secure( 0 );
102     MPI temp = mpi_alloc( mpi_get_nlimbs(p) );
103     MPI p_1 = mpi_copy(p);
104     unsigned nbits = mpi_get_nbits(p);
105
106     if( DBG_CIPHER )
107         log_debug("choosing a random k ");
108     mpi_sub_ui( p_1, p, 1);
109     for(;;) {
110         if( DBG_CIPHER )
111             fputc('.', stderr);
112         {   char *p = get_random_bits( nbits, 1, 1 );
113             mpi_set_buffer( k, p, (nbits+7)/8, 0 );
114             m_free(p);
115             /* make sure that the number is of the exact lenght */
116             if( mpi_test_bit( k, nbits-1 ) )
117                 mpi_set_highbit( k, nbits-1 );
118             else {
119                 mpi_set_highbit( k, nbits-1 );
120                 mpi_clear_bit( k, nbits-1 );
121             }
122         }
123         if( !(mpi_cmp( k, p_1 ) < 0) )  /* check: k < (p-1) */
124             continue; /* no  */
125         if( !(mpi_cmp_ui( k, 0 ) > 0) ) /* check: k > 0 */
126             continue; /* no */
127         if( mpi_gcd( temp, k, p_1 ) )
128             break;  /* okay, k is relatively prime to (p-1) */
129     }
130     if( DBG_CIPHER )
131         fputc('\n', stderr);
132     mpi_free(p_1);
133     mpi_free(temp);
134
135     return k;
136 }
137
138 /****************
139  * Generate a key pair with a key of size NBITS
140  * Returns: 2 structures filles with all needed values
141  *          and an array with n-1 factors of (p-1)
142  */
143 static void
144 generate(  ELG_secret_key *sk, unsigned nbits, MPI **ret_factors )
145 {
146     MPI p;    /* the prime */
147     MPI p_min1;
148     MPI g;
149     MPI x;    /* the secret exponent */
150     MPI y;
151     MPI temp;
152     unsigned qbits;
153     byte *rndbuf;
154
155     p_min1 = mpi_alloc( (nbits+BITS_PER_MPI_LIMB-1)/BITS_PER_MPI_LIMB );
156     temp   = mpi_alloc( (nbits+BITS_PER_MPI_LIMB-1)/BITS_PER_MPI_LIMB );
157     if( nbits < 512 )
158         qbits = 120;
159     else if( nbits <= 1024 )
160         qbits = 160;
161     else if( nbits <= 2048 )
162         qbits = 200;
163     else
164         qbits = 240;
165     g = mpi_alloc(1);
166     p = generate_elg_prime( 0, nbits, qbits, g, ret_factors );
167     mpi_sub_ui(p_min1, p, 1);
168
169
170     /* select a random number which has these properties:
171      *   0 < x < p-1
172      * This must be a very good random number because this is the
173      * secret part.  The prime is public and may be shared anyway,
174      * so a random generator level of 1 is used for the prime.
175      */
176     x = mpi_alloc_secure( nbits/BITS_PER_MPI_LIMB );
177     if( DBG_CIPHER )
178         log_debug("choosing a random x ");
179     rndbuf = NULL;
180     do {
181         if( DBG_CIPHER )
182             fputc('.', stderr);
183         if( rndbuf ) { /* change only some of the higher bits */
184             if( nbits < 16 ) {/* should never happen ... */
185                 m_free(rndbuf);
186                 rndbuf = get_random_bits( nbits, 2, 1 );
187             }
188             else {
189                 char *r = get_random_bits( 16, 2, 1 );
190                 memcpy(rndbuf, r, 16/8 );
191                 m_free(r);
192             }
193         }
194         else
195             rndbuf = get_random_bits( nbits, 2, 1 );
196         mpi_set_buffer( x, rndbuf, (nbits+7)/8, 0 );
197         mpi_clear_highbit( x, nbits+1 );
198     } while( !( mpi_cmp_ui( x, 0 )>0 && mpi_cmp( x, p_min1 )<0 ) );
199     m_free(rndbuf);
200
201     y = mpi_alloc(nbits/BITS_PER_MPI_LIMB);
202     mpi_powm( y, g, x, p );
203
204     if( DBG_CIPHER ) {
205         fputc('\n', stderr);
206         log_mpidump("elg  p= ", p );
207         log_mpidump("elg  g= ", g );
208         log_mpidump("elg  y= ", y );
209         log_mpidump("elg  x= ", x );
210     }
211
212     /* copy the stuff to the key structures */
213     sk->p = p;
214     sk->g = g;
215     sk->y = y;
216     sk->x = x;
217
218     /* now we can test our keys (this should never fail!) */
219     test_keys( sk, nbits - 64 );
220
221     mpi_free( p_min1 );
222     mpi_free( temp   );
223 }
224
225
226 /****************
227  * Test whether the secret key is valid.
228  * Returns: if this is a valid key.
229  */
230 static int
231 check_secret_key( ELG_secret_key *sk )
232 {
233     int rc;
234     MPI y = mpi_alloc( mpi_get_nlimbs(sk->y) );
235
236     mpi_powm( y, sk->g, sk->x, sk->p );
237     rc = !mpi_cmp( y, sk->y );
238     mpi_free( y );
239     return rc;
240 }
241
242
243 static void
244 encrypt(MPI a, MPI b, MPI input, ELG_public_key *pkey )
245 {
246     MPI k;
247
248     k = gen_k( pkey->p );
249     mpi_powm( a, pkey->g, k, pkey->p );
250     /* b = (y^k * input) mod p
251      *   = ((y^k mod p) * (input mod p)) mod p
252      * and because input is < p  (FIXME: check this!)
253      *   = ((y^k mod p) * input) mod p
254      */
255     mpi_powm( b, pkey->y, k, pkey->p );
256     mpi_mulm( b, b, input, pkey->p );
257   #if 0
258     if( DBG_CIPHER ) {
259         log_mpidump("elg encrypted y= ", pkey->y);
260         log_mpidump("elg encrypted p= ", pkey->p);
261         log_mpidump("elg encrypted k= ", k);
262         log_mpidump("elg encrypted M= ", input);
263         log_mpidump("elg encrypted a= ", a);
264         log_mpidump("elg encrypted b= ", b);
265     }
266   #endif
267     mpi_free(k);
268 }
269
270
271
272
273 static void
274 decrypt(MPI output, MPI a, MPI b, ELG_secret_key *skey )
275 {
276     MPI t1 = mpi_alloc_secure( mpi_get_nlimbs( skey->p ) );
277
278     /* output = b/(a^x) mod p */
279
280     mpi_powm( t1, a, skey->x, skey->p );
281     mpi_invm( t1, t1, skey->p );
282     mpi_mulm( output, b, t1, skey->p );
283   #if 0
284     if( DBG_CIPHER ) {
285         log_mpidump("elg decrypted x= ", skey->x);
286         log_mpidump("elg decrypted p= ", skey->p);
287         log_mpidump("elg decrypted a= ", a);
288         log_mpidump("elg decrypted b= ", b);
289         log_mpidump("elg decrypted M= ", output);
290     }
291   #endif
292     mpi_free(t1);
293 }
294
295
296 /****************
297  * Make an Elgamal signature out of INPUT
298  */
299
300 static void
301 sign(MPI a, MPI b, MPI input, ELG_secret_key *skey )
302 {
303     MPI k;
304     MPI t   = mpi_alloc( mpi_get_nlimbs(a) );
305     MPI inv = mpi_alloc( mpi_get_nlimbs(a) );
306     MPI p_1 = mpi_copy(skey->p);
307
308    /*
309     * b = (t * inv) mod (p-1)
310     * b = (t * inv(k,(p-1),(p-1)) mod (p-1)
311     * b = (((M-x*a) mod (p-1)) * inv(k,(p-1),(p-1))) mod (p-1)
312     *
313     */
314     mpi_sub_ui(p_1, p_1, 1);
315     k = gen_k( skey->p );
316     mpi_powm( a, skey->g, k, skey->p );
317     mpi_mul(t, skey->x, a );
318     mpi_subm(t, input, t, p_1 );
319     while( mpi_is_neg(t) )
320         mpi_add(t, t, p_1);
321     mpi_invm(inv, k, p_1 );
322     mpi_mulm(b, t, inv, p_1 );
323
324   #if 0
325     if( DBG_CIPHER ) {
326         log_mpidump("elg sign p= ", skey->p);
327         log_mpidump("elg sign g= ", skey->g);
328         log_mpidump("elg sign y= ", skey->y);
329         log_mpidump("elg sign x= ", skey->x);
330         log_mpidump("elg sign k= ", k);
331         log_mpidump("elg sign M= ", input);
332         log_mpidump("elg sign a= ", a);
333         log_mpidump("elg sign b= ", b);
334     }
335   #endif
336     mpi_free(k);
337     mpi_free(t);
338     mpi_free(inv);
339     mpi_free(p_1);
340 }
341
342
343 /****************
344  * Returns true if the signature composed of A and B is valid.
345  */
346 static int
347 verify(MPI a, MPI b, MPI input, ELG_public_key *pkey )
348 {
349     int rc;
350     MPI t1;
351     MPI t2;
352     MPI base[4];
353     MPI exp[4];
354
355     if( !(mpi_cmp_ui( a, 0 ) > 0 && mpi_cmp( a, pkey->p ) < 0) )
356         return 0; /* assertion  0 < a < p  failed */
357
358     t1 = mpi_alloc( mpi_get_nlimbs(a) );
359     t2 = mpi_alloc( mpi_get_nlimbs(a) );
360
361   #if 0
362     /* t1 = (y^a mod p) * (a^b mod p) mod p */
363     mpi_powm( t1, pkey->y, a, pkey->p );
364     mpi_powm( t2, a, b, pkey->p );
365     mpi_mulm( t1, t1, t2, pkey->p );
366
367     /* t2 = g ^ input mod p */
368     mpi_powm( t2, pkey->g, input, pkey->p );
369
370     rc = !mpi_cmp( t1, t2 );
371   #elif 0
372     /* t1 = (y^a mod p) * (a^b mod p) mod p */
373     base[0] = pkey->y; exp[0] = a;
374     base[1] = a;       exp[1] = b;
375     base[2] = NULL;    exp[2] = NULL;
376     mpi_mulpowm( t1, base, exp, pkey->p );
377
378     /* t2 = g ^ input mod p */
379     mpi_powm( t2, pkey->g, input, pkey->p );
380
381     rc = !mpi_cmp( t1, t2 );
382   #else
383     /* t1 = g ^ - input * y ^ a * a ^ b  mod p */
384     mpi_invm(t2, pkey->g, pkey->p );
385     base[0] = t2     ; exp[0] = input;
386     base[1] = pkey->y; exp[1] = a;
387     base[2] = a;       exp[2] = b;
388     base[3] = NULL;    exp[3] = NULL;
389     mpi_mulpowm( t1, base, exp, pkey->p );
390     rc = !mpi_cmp_ui( t1, 1 );
391
392   #endif
393
394     mpi_free(t1);
395     mpi_free(t2);
396     return rc;
397 }
398
399 /*********************************************
400  **************  interface  ******************
401  *********************************************/
402
403 int
404 elg_generate( int algo, unsigned nbits, MPI *skey, MPI **retfactors )
405 {
406     ELG_secret_key sk;
407
408     if( !is_ELGAMAL(algo) )
409         return G10ERR_PUBKEY_ALGO;
410
411     generate( &sk, nbits, retfactors );
412     skey[0] = sk.p;
413     skey[1] = sk.g;
414     skey[2] = sk.y;
415     skey[3] = sk.x;
416     return 0;
417 }
418
419
420 int
421 elg_check_secret_key( int algo, MPI *skey )
422 {
423     ELG_secret_key sk;
424
425     if( !is_ELGAMAL(algo) )
426         return G10ERR_PUBKEY_ALGO;
427
428     sk.p = skey[0];
429     sk.g = skey[1];
430     sk.y = skey[2];
431     sk.x = skey[3];
432     if( !check_secret_key( &sk ) )
433         return G10ERR_BAD_SECKEY;
434
435     return 0;
436 }
437
438
439
440 int
441 elg_encrypt( int algo, MPI *resarr, MPI data, MPI *pkey )
442 {
443     ELG_public_key pk;
444
445     if( !is_ELGAMAL(algo) )
446         return G10ERR_PUBKEY_ALGO;
447
448     pk.p = pkey[0];
449     pk.g = pkey[1];
450     pk.y = pkey[2];
451     resarr[0] = mpi_alloc( mpi_get_nlimbs( pk.p ) );
452     resarr[1] = mpi_alloc( mpi_get_nlimbs( pk.p ) );
453     encrypt( resarr[0], resarr[1], data, &pk );
454     return 0;
455 }
456
457 int
458 elg_decrypt( int algo, MPI *result, MPI *data, MPI *skey )
459 {
460     ELG_secret_key sk;
461
462     if( !is_ELGAMAL(algo) )
463         return G10ERR_PUBKEY_ALGO;
464
465     sk.p = skey[0];
466     sk.g = skey[1];
467     sk.y = skey[2];
468     sk.x = skey[3];
469     *result = mpi_alloc_secure( mpi_get_nlimbs( sk.p ) );
470     decrypt( *result, data[0], data[1], &sk );
471     return 0;
472 }
473
474 int
475 elg_sign( int algo, MPI *resarr, MPI data, MPI *skey )
476 {
477     ELG_secret_key sk;
478
479     if( !is_ELGAMAL(algo) )
480         return G10ERR_PUBKEY_ALGO;
481
482     sk.p = skey[0];
483     sk.g = skey[1];
484     sk.y = skey[2];
485     sk.x = skey[3];
486     resarr[0] = mpi_alloc( mpi_get_nlimbs( sk.p ) );
487     resarr[1] = mpi_alloc( mpi_get_nlimbs( sk.p ) );
488     sign( resarr[0], resarr[1], data, &sk );
489     return 0;
490 }
491
492 int
493 elg_verify( int algo, MPI hash, MPI *data, MPI *pkey,
494                     int (*cmp)(void *, MPI), void *opaquev )
495 {
496     ELG_public_key pk;
497
498     if( !is_ELGAMAL(algo) )
499         return G10ERR_PUBKEY_ALGO;
500
501     pk.p = pkey[0];
502     pk.g = pkey[1];
503     pk.y = pkey[2];
504     if( !verify( data[0], data[1], hash, &pk ) )
505         return G10ERR_BAD_SIGN;
506     return 0;
507 }
508
509
510
511 unsigned
512 elg_get_nbits( int algo, MPI *pkey )
513 {
514     if( !is_ELGAMAL(algo) )
515         return 0;
516     return mpi_get_nbits( pkey[0] );
517 }
518
519
520 /****************
521  * Return some information about the algorithm.  We need algo here to
522  * distinguish different flavors of the algorithm.
523  * Returns: A pointer to string describing the algorithm or NULL if
524  *          the ALGO is invalid.
525  * Usage: Bit 0 set : allows signing
526  *            1 set : allows encryption
527  * NOTE: This function allows signing also for ELG-E, which is not
528  * okay but a bad hack to allow to work with old gpg keys. The real check
529  * is done in the gnupg ocde depending on the packet version.
530  */
531 const char *
532 elg_get_info( int algo, int *npkey, int *nskey, int *nenc, int *nsig,
533                                                          int *usage )
534 {
535     *npkey = 3;
536     *nskey = 4;
537     *nenc = 2;
538     *nsig = 2;
539
540     switch( algo ) {
541       case PUBKEY_ALGO_ELGAMAL:
542         *usage = PUBKEY_USAGE_SIG|PUBKEY_USAGE_ENC;
543         return "ELG";
544       case PUBKEY_ALGO_ELGAMAL_E:
545         *usage = PUBKEY_USAGE_SIG|PUBKEY_USAGE_ENC;
546         return "ELG-E";
547       default: *usage = 0; return NULL;
548     }
549 }
550
551