crypto.c 18.5 KB
Newer Older
1
2
3
4
/* Copyright 2001,2002 Roger Dingledine, Matej Pfajfar. */
/* See LICENSE for licensing information */
/* $Id$ */

5
6
7
8
9
10
11
#include <string.h>

#include <openssl/err.h>
#include <openssl/rsa.h>
#include <openssl/pem.h>
#include <openssl/evp.h>
#include <openssl/rand.h>
12
#include <openssl/opensslv.h>
13
14
#include <openssl/bn.h>
#include <openssl/dh.h>
15
16

#include <stdlib.h>
17
#include <assert.h>
18
#include <stdio.h>
19

20
#include "crypto.h"
21
#include "../or/or.h"
22
23
#include "log.h"

24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#if OPENSSL_VERSION_NUMBER < 0x00905000l
#error "We require openssl >= 0.9.5"
#elif OPENSSL_VERSION_NUMBER < 0x00906000l
#define OPENSSL_095
#endif

/*
 * Certain functions that return a success code in OpenSSL 0.9.6 return void
 * (and don't indicate errors) in OpenSSL version 0.9.5.
 *
 * [OpenSSL 0.9.5 matters, because it ships with Redhat 6.2.]
 */
#ifdef OPENSSL_095
#define RETURN_SSL_OUTCOME(exp) (exp); return 0
#else
#define RETURN_SSL_OUTCOME(exp) return !(exp)
#endif

Nick Mathewson's avatar
Nick Mathewson committed
42
43
44
45
static inline const EVP_CIPHER *
crypto_cipher_evp_cipher(int type, int enc);


46
47
static inline int 
crypto_cipher_iv_length(int type) {
Nick Mathewson's avatar
Nick Mathewson committed
48
49
50
51
  /*
  printf("%d -> %d IV\n",type, EVP_CIPHER_iv_length(
						  crypto_cipher_evp_cipher(type,0)));
  */
52
53
54
55
56
57
58
59
60
61
62
63
  switch(type) 
    {
    case CRYPTO_CIPHER_IDENTITY: return 0;
    case CRYPTO_CIPHER_DES: return 8;
    case CRYPTO_CIPHER_RC4: return 16;
    case CRYPTO_CIPHER_3DES: return 8;
    default: assert(0); return -1;
    }
}

static inline int
crypto_cipher_key_length(int type) {
Nick Mathewson's avatar
Nick Mathewson committed
64
65
66
67
  /*
  printf("%d -> %d\n",type, EVP_CIPHER_key_length(
						  crypto_cipher_evp_cipher(type,0)));
  */
68
69
70
71
72
73
74
75
76
77
  switch(type) 
    {
    case CRYPTO_CIPHER_IDENTITY: return 0;
    case CRYPTO_CIPHER_DES: return 8;
    case CRYPTO_CIPHER_RC4: return 16;
    case CRYPTO_CIPHER_3DES: return 16;
    default: assert(0); return -1;
    }
}

78
static inline const EVP_CIPHER *
79
80
81
82
83
84
85
86
87
88
89
crypto_cipher_evp_cipher(int type, int enc) {
  switch(type) 
    {
    case CRYPTO_CIPHER_IDENTITY: return EVP_enc_null();
    case CRYPTO_CIPHER_DES: return EVP_des_ofb();
    case CRYPTO_CIPHER_RC4: return EVP_rc4();
    case CRYPTO_CIPHER_3DES: return EVP_des_ede_ofb();
    default: return NULL;
    }
}

90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
int crypto_global_init() 
{
  ERR_load_crypto_strings();
  return 0;
}

int crypto_global_cleanup()
{
  ERR_free_strings();
  return 0;
}

crypto_pk_env_t *crypto_new_pk_env(int type)
{
  crypto_pk_env_t *env;
  
106
  env = (crypto_pk_env_t *)tor_malloc(sizeof(crypto_pk_env_t));
107
108
  
  env->type = type;
109
  env->refs = 1;
110
111
112
113
114
  env->key = NULL;
  env->aux = NULL;
  
  switch(type) {
    case CRYPTO_PK_RSA:
115
116
117
118
119
120
    env->key = (unsigned char *)RSA_new();
    if (!env->key) {
      free((void *)env);
      return NULL;
    }
    break;
121
    default:
122
123
124
    free((void *)env);
    return NULL;
    break;
125
126
127
128
129
130
131
  }

  return env;
}

void crypto_free_pk_env(crypto_pk_env_t *env)
{
132
  assert(env);
133
134
135

  if(--env->refs > 0)
    return;
136
137
138
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
139
    if (env->key)
140
      RSA_free((RSA *)env->key);
141
    break;
142
    default:
143
    break;
144
145
146
147
148
149
  }
  
  free((void *)env);
  return;
}

150
151
152
153
154
155
156
157
158
159
160
161

/* Create a new crypto_cipher_env_t for a given onion cipher type, key,
 * iv, and encryption flag (1=encrypt, 0=decrypt).  Return the crypto object
 * on success; NULL on failure.
 */
crypto_cipher_env_t *
crypto_create_init_cipher(int cipher_type, char *key, char *iv, int encrypt_mode) 
{
  int r;
  crypto_cipher_env_t *crypto = NULL;

  if (! (crypto = crypto_new_cipher_env(cipher_type))) {
162
    log_fn(LOG_ERR, "Unable to allocate crypto object");
163
164
165
166
    return NULL;
  }

  if (crypto_cipher_set_key(crypto, key)) {
167
    log_fn(LOG_ERR, "Unable to set key: %s", crypto_perror());
168
169
170
171
    goto error;
  }

  if (crypto_cipher_set_iv(crypto, iv)) {
172
    log_fn(LOG_ERR, "Unable to set iv: %s", crypto_perror());
173
174
175
176
177
178
179
180
181
    goto error;
  }
  
  if (encrypt_mode)
    r = crypto_cipher_encrypt_init_cipher(crypto);
  else
    r = crypto_cipher_decrypt_init_cipher(crypto);

  if (r) {
182
    log_fn(LOG_ERR, "Unabble to initialize cipher: %s", crypto_perror());
183
184
185
186
187
188
189
190
191
192
    goto error;
  }
  return crypto;

 error:
  if (crypto)
    crypto_free_cipher_env(crypto);
  return NULL;
}

193
194
195
crypto_cipher_env_t *crypto_new_cipher_env(int type)
{
  crypto_cipher_env_t *env;
196
  int iv_len, key_len;
197
  
198
  env = (crypto_cipher_env_t *)tor_malloc(sizeof(crypto_cipher_env_t));
199
200
201
202
203
  
  env->type = type;
  env->key = NULL;
  env->iv = NULL;
  env->aux = NULL;
204
205
206

  iv_len = crypto_cipher_iv_length(type);
  key_len = crypto_cipher_key_length(type);
207
  
208
209
210
211
  if (! crypto_cipher_evp_cipher(type,0))
    /* This is not an openssl cipher */
    goto err;
  else {
212
    env->aux = (unsigned char *)tor_malloc(sizeof(EVP_CIPHER_CTX));
213
    EVP_CIPHER_CTX_init((EVP_CIPHER_CTX *)env->aux);
214
  }
215

216
217
  if(iv_len)
    env->iv = (unsigned char *)tor_malloc(iv_len);
218

219
220
  if(key_len)
    env->key = (unsigned char *)tor_malloc(key_len);
221
  
222
  return env;
223
224
225
226
227
228
229
230
231
232
err:
  if (env->key)
    free(env->key);
  if (env->iv)
    free(env->iv);
  if (env->aux)
    free(env->aux);
  if (env)
    free(env);
  return NULL;
233
234
235
236
}

void crypto_free_cipher_env(crypto_cipher_env_t *env)
{
237
  assert(env);
238
239
240
241
242

  if (crypto_cipher_evp_cipher(env->type,0)) {
    /* This is an openssl cipher */
    assert(env->aux);
    EVP_CIPHER_CTX_cleanup((EVP_CIPHER_CTX *)env->aux);
243
  }
244
245
246
247
248
249
250
251

  if (env->aux)
    free((void *)env->aux);
  if (env->iv) 
    free((void *)env->iv);
  if (env->key)
    free((void *)env->key);

252
253
254
255
256
257
  free((void *)env);
}

/* public key crypto */
int crypto_pk_generate_key(crypto_pk_env_t *env)
{
258
259
260
261
262
263
  assert(env);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
    if (env->key)
      RSA_free((RSA *)env->key);
264
    env->key = (unsigned char *)RSA_generate_key(1024,65537, NULL, NULL);
265
266
267
268
269
270
271
    if (!env->key)
      return -1;
    break;
    default:
    return -1;
  }
  
272
273
274
  return 0;
}

275
int crypto_pk_read_private_key_from_file(crypto_pk_env_t *env, FILE *src)
276
{
277
278
279
280
281
  assert(env && src);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
    if (env->key)
Roger Dingledine's avatar
Roger Dingledine committed
282
283
      RSA_free((RSA *)env->key);
    env->key = (unsigned char *)PEM_read_RSAPrivateKey(src, NULL, NULL, NULL);
284
285
286
287
288
289
290
    if (!env->key)
      return -1;
    break;
    default :
    return -1;
  }
  
291
292
  return 0;
}
293

294
int crypto_pk_read_private_key_from_filename(crypto_pk_env_t *env, unsigned char *keyfile)
295
296
297
298
299
300
301
302
303
{
  FILE *f_pr;
  int retval = 0;
 
  assert(env && keyfile);
  
  if (strspn(keyfile,CONFIG_LEGAL_FILENAME_CHARACTERS) == strlen(keyfile)) /* filename contains legal characters only */
  {
    /* open the keyfile */
304
    f_pr=fopen(keyfile,"rb");
305
306
    if (!f_pr)
      return -1;
307
    
308
    /* read the private key */
309
    retval = crypto_pk_read_private_key_from_file(env, f_pr);
310
311
312
    fclose(f_pr);
    if (retval == -1)
    {
313
      log_fn(LOG_ERR,"Error reading private key : %s",crypto_perror());
314
315
316
317
318
319
320
      return -1;
    }
      
    /* check the private key */
    retval = crypto_pk_check_key(env);
    if (retval == 0)
    {
321
      log_fn(LOG_ERR,"Private key read but is invalid : %s.", crypto_perror());
322
323
324
325
      return -1;
    }
    else if (retval == -1)
    {
326
      log_fn(LOG_ERR,"Private key read but validity checking failed : %s",crypto_perror());
327
328
329
330
331
332
333
334
335
336
337
      return -1;
    }
    else if (retval == 1)
    {
      return 0;
    }
  } /* filename contains legal characters only */
  
  return -1; /* report error */
}

338
int crypto_pk_read_public_key_from_file(crypto_pk_env_t *env, FILE *src)
339
{
340
341
342
343
  assert(env && src);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
Roger Dingledine's avatar
Roger Dingledine committed
344
345
346
    if(env->key)
      RSA_free((RSA *)env->key); 
    env->key = (unsigned char *)PEM_read_RSAPublicKey(src, NULL, NULL, NULL);
347
348
349
350
351
352
353
    if (!env->key)
      return -1;
    break;
    default :
    return -1;
  }
  
354
355
  return 0;
}
356

357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
int crypto_pk_write_public_key_to_string(crypto_pk_env_t *env, char **dest, int *len) {
  BUF_MEM *buf;
  BIO *b; 

  assert(env && env->key && dest);

  switch(env->type) {
    case CRYPTO_PK_RSA:

        b = BIO_new(BIO_s_mem()); /* Create a memory BIO */

        /* Now you can treat b as if it were a file.  Just use the
         *          * PEM_*_bio_* functions instead of the non-bio variants.
         *                   */
        if(!PEM_write_bio_RSAPublicKey(b, (RSA *)env->key))
          return -1;

        BIO_get_mem_ptr(b, &buf);
        BIO_set_close(b, BIO_NOCLOSE); /* so BIO_free doesn't free buf */
        BIO_free(b);

378
        *dest = tor_malloc(buf->length+1);
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
        memcpy(*dest, buf->data, buf->length);
        (*dest)[buf->length] = 0; /* null terminate it */
        *len = buf->length;
        BUF_MEM_free(buf);

      break;
    default:
      return -1;
  }
  
  return 0;
}

int crypto_pk_read_public_key_from_string(crypto_pk_env_t *env, char *src, int len) {
  BIO *b; 

  assert(env && src);

  switch(env->type) {
    case CRYPTO_PK_RSA:
      b = BIO_new(BIO_s_mem()); /* Create a memory BIO */

      BIO_write(b, src, len);

      RSA_free((RSA *)env->key); 
      env->key = (unsigned char *)PEM_read_bio_RSAPublicKey(b, NULL, NULL, NULL);
      if(!env->key)
        return -1;

      BIO_free(b);
      break;
    default:
      return -1;
  }
  
  return 0;
}

int crypto_pk_write_private_key_to_file(crypto_pk_env_t *env, FILE *dest)
418
{
419
420
421
422
423
424
425
426
427
428
429
430
431
  assert(env && dest);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
    if (!env->key)
      return -1;
    if (PEM_write_RSAPrivateKey(dest, (RSA *)env->key, NULL, NULL, 0,0, NULL) == 0)
      return -1;
    break;
    default :
    return -1;
  }
  
432
433
  return 0;
}
434
int crypto_pk_write_public_key_to_file(crypto_pk_env_t *env, FILE *dest)
435
{
436
437
438
439
440
441
442
443
444
445
446
447
448
  assert(env && dest);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
    if (!env->key)
      return -1;
    if (PEM_write_RSAPublicKey(dest, (RSA *)env->key) == 0)
      return -1;
    break;
    default :
    return -1;
  }
  
449
450
451
  return 0;
}

452
453
454
455
456
457
int crypto_pk_check_key(crypto_pk_env_t *env)
{
  assert(env);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
458
      return RSA_check_key((RSA *)env->key);
459
    default:
460
      return -1;
461
462
463
  }
}

464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
int crypto_pk_cmp_keys(crypto_pk_env_t *a, crypto_pk_env_t *b) {
  int result; 
  
  if (!a || !b)
    return -1;
  
  if (!a->key || !b->key)
    return -1;
  
  if (a->type != b->type)
    return -1;
  
  switch(a->type) {
    case CRYPTO_PK_RSA:
    assert(((RSA *)a->key)->n && ((RSA *)a->key)->e && ((RSA *)b->key)->n && ((RSA *)b->key)->e);
    result = BN_cmp(((RSA *)a->key)->n, ((RSA *)b->key)->n);
    if (result)
      return result;
    return BN_cmp(((RSA *)a->key)->e, ((RSA *)b->key)->e);
    default:
    return -1;
  }
}

int crypto_pk_keysize(crypto_pk_env_t *env)
{
  assert(env && env->key);
  
  return RSA_size((RSA *)env->key);
}
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508

crypto_pk_env_t *crypto_pk_dup_key(crypto_pk_env_t *env) {
  assert(env && env->key);

  switch(env->type) {
    case CRYPTO_PK_RSA:
      env->refs++; 
      break;
    default:
      return NULL; 
  }
  
  return env;
}

509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
int crypto_pk_public_encrypt(crypto_pk_env_t *env, unsigned char *from, int fromlen, unsigned char *to, int padding)
{
  assert(env && from && to);
  
  switch(env->type) {
    case CRYPTO_PK_RSA:
    return RSA_public_encrypt(fromlen, from, to, (RSA *)env->key, padding);
    default:
    return -1;
  }
}

int crypto_pk_private_decrypt(crypto_pk_env_t *env, unsigned char *from, int fromlen, unsigned char *to, int padding)
{
  assert(env && from && to);
524

525
  switch(env->type) {
526
527
528
  case CRYPTO_PK_RSA:
    if (!(((RSA*)env->key)->p))
      return -1;
529
530
531
532
533
534
    return RSA_private_decrypt(fromlen, from, to, (RSA *)env->key, padding);
    default:
    return -1;
  }
}

535
536
537
538
539
540
541
int crypto_pk_public_checksig(crypto_pk_env_t *env, unsigned char *from, int fromlen, unsigned char *to)
{
  assert(env && from && to);

  switch(env->type) {
  case CRYPTO_PK_RSA:
    return RSA_public_decrypt(fromlen, from, to, (RSA *)env->key, 
542
			      RSA_PKCS1_PADDING);
543
544
545
546
547
548
549
550
551
552
553
554
555
556
    default:
    return -1;
  }
}

int crypto_pk_private_sign(crypto_pk_env_t *env, unsigned char *from, int fromlen, unsigned char *to)
{
  assert(env && from && to);

  switch(env->type) {
  case CRYPTO_PK_RSA:
    if (!(((RSA*)env->key)->p))
      return -1;
    return RSA_private_encrypt(fromlen, from, to, (RSA *)env->key, 
557
			       RSA_PKCS1_PADDING);
558
559
560
561
562
    default:
    return -1;
  }
}

563
/* symmetric crypto */
564
565
int crypto_cipher_generate_key(crypto_cipher_env_t *env)
{
566
  int key_len;
567
  assert(env);
568
569
570
571
572
573

  key_len = crypto_cipher_key_length(env->type);

  if (key_len > 0)
    return crypto_rand(key_len, env->key);
  else if (key_len == 0)
574
    return 0;
575
  else
576
577
578
    return -1;
}

579
580
int crypto_cipher_set_iv(crypto_cipher_env_t *env, unsigned char *iv)
{
581
  int iv_len;
582
583
  assert(env && iv);
  
584
585
586
587
588
  iv_len = crypto_cipher_iv_length(env->type);
  if (!iv_len)
    return 0;

  if (!env->iv)
589
    return -1;
590
591

  memcpy((void*)env->iv, (void*)iv, iv_len);
592
  
593
594
  return 0;
}
595

596
597
int crypto_cipher_set_key(crypto_cipher_env_t *env, unsigned char *key)
{
598
  int key_len;
599
  assert(env && key);
600
601
602
603
604
605

  key_len = crypto_cipher_key_length(env->type);
  if (!key_len)
    return 0;

  if (!env->key)
606
    return -1;
607
608

  memcpy((void*)env->key, (void*)key, key_len);
609
  
610
611
  return 0;
}
612
613

int crypto_cipher_encrypt_init_cipher(crypto_cipher_env_t *env)
614
{
615
616
  assert(env);
  
617
618
  if (crypto_cipher_evp_cipher(env->type, 1)) {
    RETURN_SSL_OUTCOME(EVP_EncryptInit((EVP_CIPHER_CTX *)env->aux,
Roger Dingledine's avatar
Roger Dingledine committed
619
620
                                       crypto_cipher_evp_cipher(env->type, 1),
                                       env->key, env->iv));
621
622
623
  } else {
    return -1;
  }
624
625
}

626
627
628
int crypto_cipher_decrypt_init_cipher(crypto_cipher_env_t *env)
{
  assert(env);
629
630
631

  if (crypto_cipher_evp_cipher(env->type, 0)) {
    RETURN_SSL_OUTCOME(EVP_EncryptInit((EVP_CIPHER_CTX *)env->aux,
Roger Dingledine's avatar
Roger Dingledine committed
632
633
                                       crypto_cipher_evp_cipher(env->type, 0),
                                       env->key, env->iv));
634
  } else {
635
636
637
638
    return -1;
  }
}

Nick Mathewson's avatar
Nick Mathewson committed
639
int crypto_cipher_encrypt(crypto_cipher_env_t *env, unsigned char *from, unsigned int fromlen, unsigned char *to)
640
{
641
642
643
644
  int tolen;
  
  assert(env && from && to);
  
645
  RETURN_SSL_OUTCOME(EVP_EncryptUpdate((EVP_CIPHER_CTX *)env->aux, to, &tolen, from, fromlen));
646
647
}

Nick Mathewson's avatar
Nick Mathewson committed
648
int crypto_cipher_decrypt(crypto_cipher_env_t *env, unsigned char *from, unsigned int fromlen, unsigned char *to)
649
{
650
651
652
653
  int tolen;
  
  assert(env && from && to);
  
654
  RETURN_SSL_OUTCOME(EVP_DecryptUpdate((EVP_CIPHER_CTX *)env->aux, to, &tolen, from, fromlen));
655
656
657
}

/* SHA-1 */
658
int crypto_SHA_digest(unsigned char *m, int len, unsigned char *digest)
659
{
660
661
  assert(m && digest);
  return (SHA1(m,len,digest) == NULL);
662
663
}

664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682

struct crypto_dh_env_st {
  DH *dh;
};

static BIGNUM *dh_param_p = NULL;
static BIGNUM *dh_param_g = NULL;


static void init_dh_param() {
  BIGNUM *p, *g;
  int r;
  if (dh_param_p && dh_param_g)
    return;
  
  p = BN_new();
  g = BN_new();
  assert(p && g);

683
#if 0 
684
685
686
687
688
689
690
691
692
693
694
695
696
  /* This is from draft-ietf-ipsec-ike-modp-groups-05.txt.  It's a safe
     prime, and supposedly it equals:
      2^1536 - 2^1472 - 1 + 2^64 * { [2^1406 pi] + 741804 }
  */
  r = BN_hex2bn(&p, 
		"FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1"
		"29024E088A67CC74020BBEA63B139B22514A08798E3404DD"
		"EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245"
		"E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED"
		"EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D"
		"C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F"
		"83655D23DCA3AD961C62F356208552BB9ED529077096966D"
		"670C354E4ABC9804F1746C08CA237327FFFFFFFFFFFFFFFF");
697
698
699
700
701
702
703
704
705
706
707
708
#endif

  /* This is from rfc2409, section 6.2.  It's a safe prime, and
     supposedly it equals:
        2^1024 - 2^960 - 1 + 2^64 * { [2^894 pi] + 129093 }.
  */
  r = BN_hex2bn(&p,
		"FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD129024E08"
		"8A67CC74020BBEA63B139B22514A08798E3404DDEF9519B3CD3A431B"
		"302B0A6DF25F14374FE1356D6D51C245E485B576625E7EC6F44C42E9"
		"A637ED6B0BFF5CB6F406B7EDEE386BFB5A899FA5AE9F24117C4B1FE6"
		"49286651ECE65381FFFFFFFFFFFFFFFF");
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
  assert(r);

  r = BN_set_word(g, 2);
  assert(r);
  dh_param_p = p;
  dh_param_g = g;
}

crypto_dh_env_t *crypto_dh_new()
{
  crypto_dh_env_t *res = NULL;

  if (!dh_param_p)
    init_dh_param();
    
724
  res = tor_malloc(sizeof(crypto_dh_env_t));
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
  res->dh = NULL;

  if (!(res->dh = DH_new()))
    goto err;

  if (!(res->dh->p = BN_dup(dh_param_p)))
    goto err;

  if (!(res->dh->g = BN_dup(dh_param_g)))
    goto err;

  return res;
 err:
  if (res && res->dh) DH_free(res->dh); /* frees p and g too */
  if (res) free(res);
  return NULL;
}
int crypto_dh_get_bytes(crypto_dh_env_t *dh)
{
  assert(dh);
  return DH_size(dh->dh);
}
int crypto_dh_get_public(crypto_dh_env_t *dh, char *pubkey, int pubkey_len)
{
  int bytes;
  assert(dh);
  if (!DH_generate_key(dh->dh))
    return -1;
  
  assert(dh->dh->pub_key);
  bytes = BN_num_bytes(dh->dh->pub_key);
  if (pubkey_len < bytes)
    return -1;
  
  memset(pubkey, 0, pubkey_len);
  BN_bn2bin(dh->dh->pub_key, pubkey+(pubkey_len-bytes));

  return 0;
}
int crypto_dh_compute_secret(crypto_dh_env_t *dh, 
			     char *pubkey, int pubkey_len,
			     char *secret_out)
{
  BIGNUM *pubkey_bn;
  int secret_len;
  assert(dh);
  
  if (!(pubkey_bn = BN_bin2bn(pubkey, pubkey_len, NULL)))
    return -1;
  
  secret_len = DH_compute_key(secret_out, pubkey_bn, dh->dh);  
  BN_free(pubkey_bn);
  if (secret_len == -1)
    return -1;

780
  return secret_len;
781
782
783
784
785
786
787
788
}
void crypto_dh_free(crypto_dh_env_t *dh)
{
  assert(dh && dh->dh);
  DH_free(dh->dh);
  free(dh);
}

789
/* random numbers */
Nick Mathewson's avatar
Nick Mathewson committed
790
791
792
793
794
int crypto_seed_rng()
{
  static char *filenames[] = { 
    "/dev/srandom", "/dev/urandom", "/dev/random", NULL
  };
795
  int i, n;
Nick Mathewson's avatar
Nick Mathewson committed
796
797
798
799
800
801
  char buf[21];
  FILE *f;
  
  for (i = 0; filenames[i]; ++i) {
    f = fopen(filenames[i], "rb");
    if (!f) continue;
802
    log_fn(LOG_INFO, "Seeding RNG from %s", filenames[i]);
Nick Mathewson's avatar
Nick Mathewson committed
803
    buf[20]='\xff';
804
    n = fread(buf, 1, 20, f);
Nick Mathewson's avatar
Nick Mathewson committed
805
    fclose(f);
806
    if (n != 20) {
807
      log_fn(LOG_INFO, "Error reading from entropy source");
Nick Mathewson's avatar
Nick Mathewson committed
808
809
810
811
812
813
      return -1;
    }
    RAND_seed(buf, 20);
    return 0;
  }

814
  log_fn(LOG_INFO, "Cannot seed RNG -- no entropy source found.");
Nick Mathewson's avatar
Nick Mathewson committed
815
816
817
  return -1;
}

818
819
int crypto_rand(unsigned int n, unsigned char *to)
{
820
  assert(to);
821
  return (RAND_bytes(to, n) != 1);
822
}
823

824
825
int crypto_pseudo_rand(unsigned int n, unsigned char *to)
{
826
827
828
829
830
831
832
833
  assert(to);
  return (RAND_pseudo_bytes(to, n) == -1);
}

/* errors */
char *crypto_perror()
{
  return (char *)ERR_reason_error_string(ERR_get_error());
834
}
835

836
837
838
839
840
841
842
843
844
845
846
847
848
849
int 
base64_encode(char *dest, int destlen, char *src, int srclen)
{
  EVP_ENCODE_CTX ctx;
  int len, ret;
  
  /* 48 bytes of input -> 64 bytes of output plus newline. 
     Plus one more byte, in case I'm wrong.
  */
  if (destlen < ((srclen/48)+1)*66)
    return -1;

  EVP_EncodeInit(&ctx);
  EVP_EncodeUpdate(&ctx, dest, &len, src, srclen);
850
  EVP_EncodeFinal(&ctx, dest+len, &ret);
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
  ret += len;
  return ret;
}
int 
base64_decode(char *dest, int destlen, char *src, int srclen)
{
  EVP_ENCODE_CTX ctx;
  int len, ret;
  /* 64 bytes of input -> *up to* 48 bytes of output.
     Plus one more byte, in caes I'm wrong.
  */
  if (destlen < ((srclen/64)+1)*49)
    return -1;

  EVP_DecodeInit(&ctx);
  EVP_DecodeUpdate(&ctx, dest, &len, src, srclen);
  EVP_DecodeFinal(&ctx, dest, &ret);
  ret += len;
  return ret;
}