/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to plugins.d/mandosclient.c

  • Committer: Teddy Hogeborn
  • Date: 2008-08-03 16:05:52 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080803160552-mvqy5klry3wn57x1
* plugins.d/mandosclient.c (initgnutls): Moved "err" variable into its
                                         own code block.
  (start_mandos_communication): Moved call to "initgnutls" to
                                beginning.  Renamed label "exit" to
                                "mandos_end".  Bug fix: set return
                                code if failure to realloc buffer.
                                Close TLS session as early as
                                possible.
  (resolve_callback): Also print interface number when debugging.
  (main): Moved "debug_int" and "config" into their own respective
          code blocks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandosclient.c
32
32
#define _LARGEFILE_SOURCE
33
33
#define _FILE_OFFSET_BITS 64
34
34
 
35
 
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
36
 
 
37
 
#include <stdio.h>              /* fprintf(), stderr, fwrite(),
38
 
                                   stdout, ferror() */
39
 
#include <stdint.h>             /* uint16_t, uint32_t */
40
 
#include <stddef.h>             /* NULL, size_t, ssize_t */
41
 
#include <stdlib.h>             /* free(), EXIT_SUCCESS, EXIT_FAILURE,
42
 
                                   srand() */
43
 
#include <stdbool.h>            /* bool, true */
44
 
#include <string.h>             /* memset(), strcmp(), strlen(),
45
 
                                   strerror(), asprintf(), strcpy() */
46
 
#include <sys/ioctl.h>          /* ioctl */
47
 
#include <sys/types.h>          /* socket(), inet_pton(), sockaddr,
48
 
                                   sockaddr_in6, PF_INET6,
49
 
                                   SOCK_STREAM, INET6_ADDRSTRLEN,
50
 
                                   uid_t, gid_t */
51
 
#include <inttypes.h>           /* PRIu16 */
52
 
#include <sys/socket.h>         /* socket(), struct sockaddr_in6,
53
 
                                   struct in6_addr, inet_pton(),
54
 
                                   connect() */
55
 
#include <assert.h>             /* assert() */
56
 
#include <errno.h>              /* perror(), errno */
57
 
#include <time.h>               /* time() */
 
35
#include <stdio.h>
 
36
#include <assert.h>
 
37
#include <stdlib.h>
 
38
#include <time.h>
 
39
#include <net/if.h>             /* if_nametoindex */
 
40
#include <sys/ioctl.h>          /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
 
41
                                   SIOCSIFFLAGS */
58
42
#include <net/if.h>             /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
59
 
                                   SIOCSIFFLAGS, if_indextoname(),
60
 
                                   if_nametoindex(), IF_NAMESIZE */
61
 
#include <unistd.h>             /* close(), SEEK_SET, off_t, write(),
62
 
                                   getuid(), getgid(), setuid(),
63
 
                                   setgid() */
64
 
#include <netinet/in.h>
65
 
#include <arpa/inet.h>          /* inet_pton(), htons */
66
 
#include <iso646.h>             /* not, and */
67
 
#include <argp.h>               /* struct argp_option, error_t, struct
68
 
                                   argp_state, struct argp,
69
 
                                   argp_parse(), ARGP_KEY_ARG,
70
 
                                   ARGP_KEY_END, ARGP_ERR_UNKNOWN */
 
43
                                   SIOCSIFFLAGS */
71
44
 
72
 
/* Avahi */
73
 
/* All Avahi types, constants and functions
74
 
 Avahi*, avahi_*,
75
 
 AVAHI_* */
76
45
#include <avahi-core/core.h>
77
46
#include <avahi-core/lookup.h>
78
47
#include <avahi-core/log.h>
80
49
#include <avahi-common/malloc.h>
81
50
#include <avahi-common/error.h>
82
51
 
83
 
/* GnuTLS */
84
 
#include <gnutls/gnutls.h>      /* All GnuTLS types, constants and
85
 
                                   functions:
86
 
                                   gnutls_*
87
 
                                   init_gnutls_session(),
88
 
                                   GNUTLS_* */
89
 
#include <gnutls/openpgp.h>     /* gnutls_certificate_set_openpgp_key_file(),
90
 
                                   GNUTLS_OPENPGP_FMT_BASE64 */
 
52
/* Mandos client part */
 
53
#include <sys/types.h>          /* socket(), inet_pton() */
 
54
#include <sys/socket.h>         /* socket(), struct sockaddr_in6,
 
55
                                   struct in6_addr, inet_pton() */
 
56
#include <gnutls/gnutls.h>      /* All GnuTLS stuff */
 
57
#include <gnutls/openpgp.h>     /* GnuTLS with openpgp stuff */
 
58
 
 
59
#include <unistd.h>             /* close() */
 
60
#include <netinet/in.h>
 
61
#include <stdbool.h>            /* true */
 
62
#include <string.h>             /* memset */
 
63
#include <arpa/inet.h>          /* inet_pton() */
 
64
#include <iso646.h>             /* not */
 
65
#include <net/if.h>             /* IF_NAMESIZE */
91
66
 
92
67
/* GPGME */
93
 
#include <gpgme.h>              /* All GPGME types, constants and
94
 
                                   functions:
95
 
                                   gpgme_*
96
 
                                   GPGME_PROTOCOL_OpenPGP,
97
 
                                   GPG_ERR_NO_* */
 
68
#include <errno.h>              /* perror() */
 
69
#include <gpgme.h>
 
70
 
 
71
/* getopt_long */
 
72
#include <getopt.h>
98
73
 
99
74
#define BUFFER_SIZE 256
100
75
 
 
76
static const char *keydir = "/conf/conf.d/mandos";
 
77
static const char *pubkeyfile = "pubkey.txt";
 
78
static const char *seckeyfile = "seckey.txt";
 
79
 
101
80
bool debug = false;
102
 
static const char *keydir = "/conf/conf.d/mandos";
103
 
static const char mandos_protocol_version[] = "1";
104
 
const char *argp_program_version = "password-request 1.0";
105
 
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
106
81
 
107
 
/* Used for passing in values through the Avahi callback functions */
 
82
/* Used for passing in values through all the callback functions */
108
83
typedef struct {
109
84
  AvahiSimplePoll *simple_poll;
110
85
  AvahiServer *server;
111
86
  gnutls_certificate_credentials_t cred;
112
87
  unsigned int dh_bits;
113
 
  gnutls_dh_params_t dh_params;
114
88
  const char *priority;
115
89
} mandos_context;
116
90
 
117
 
/*
118
 
 * Make room in "buffer" for at least BUFFER_SIZE additional bytes.
119
 
 * "buffer_capacity" is how much is currently allocated,
120
 
 * "buffer_length" is how much is already used.
121
 
 */
122
 
size_t adjustbuffer(char **buffer, size_t buffer_length,
123
 
                  size_t buffer_capacity){
124
 
  if (buffer_length + BUFFER_SIZE > buffer_capacity){
125
 
    *buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
126
 
    if (buffer == NULL){
127
 
      return 0;
128
 
    }
129
 
    buffer_capacity += BUFFER_SIZE;
130
 
  }
131
 
  return buffer_capacity;
132
 
}
133
 
 
134
91
/* 
135
92
 * Decrypt OpenPGP data using keyrings in HOMEDIR.
136
93
 * Returns -1 on error
214
171
    fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
215
172
            gpgme_strsource(rc), gpgme_strerror(rc));
216
173
    plaintext_length = -1;
217
 
    if (debug){
218
 
      gpgme_decrypt_result_t result;
219
 
      result = gpgme_op_decrypt_result(ctx);
220
 
      if (result == NULL){
221
 
        fprintf(stderr, "gpgme_op_decrypt_result failed\n");
222
 
      } else {
223
 
        fprintf(stderr, "Unsupported algorithm: %s\n",
224
 
                result->unsupported_algorithm);
225
 
        fprintf(stderr, "Wrong key usage: %u\n",
226
 
                result->wrong_key_usage);
227
 
        if(result->file_name != NULL){
228
 
          fprintf(stderr, "File name: %s\n", result->file_name);
229
 
        }
230
 
        gpgme_recipient_t recipient;
231
 
        recipient = result->recipients;
232
 
        if(recipient){
233
 
          while(recipient != NULL){
234
 
            fprintf(stderr, "Public key algorithm: %s\n",
235
 
                    gpgme_pubkey_algo_name(recipient->pubkey_algo));
236
 
            fprintf(stderr, "Key ID: %s\n", recipient->keyid);
237
 
            fprintf(stderr, "Secret key available: %s\n",
238
 
                    recipient->status == GPG_ERR_NO_SECKEY
239
 
                    ? "No" : "Yes");
240
 
            recipient = recipient->next;
241
 
          }
242
 
        }
243
 
      }
244
 
    }
245
174
    goto decrypt_end;
246
175
  }
247
176
  
249
178
    fprintf(stderr, "Decryption of OpenPGP data succeeded\n");
250
179
  }
251
180
  
 
181
  if (debug){
 
182
    gpgme_decrypt_result_t result;
 
183
    result = gpgme_op_decrypt_result(ctx);
 
184
    if (result == NULL){
 
185
      fprintf(stderr, "gpgme_op_decrypt_result failed\n");
 
186
    } else {
 
187
      fprintf(stderr, "Unsupported algorithm: %s\n",
 
188
              result->unsupported_algorithm);
 
189
      fprintf(stderr, "Wrong key usage: %d\n",
 
190
              result->wrong_key_usage);
 
191
      if(result->file_name != NULL){
 
192
        fprintf(stderr, "File name: %s\n", result->file_name);
 
193
      }
 
194
      gpgme_recipient_t recipient;
 
195
      recipient = result->recipients;
 
196
      if(recipient){
 
197
        while(recipient != NULL){
 
198
          fprintf(stderr, "Public key algorithm: %s\n",
 
199
                  gpgme_pubkey_algo_name(recipient->pubkey_algo));
 
200
          fprintf(stderr, "Key ID: %s\n", recipient->keyid);
 
201
          fprintf(stderr, "Secret key available: %s\n",
 
202
                  recipient->status == GPG_ERR_NO_SECKEY
 
203
                  ? "No" : "Yes");
 
204
          recipient = recipient->next;
 
205
        }
 
206
      }
 
207
    }
 
208
  }
 
209
  
252
210
  /* Seek back to the beginning of the GPGME plaintext data buffer */
253
211
  if (gpgme_data_seek(dh_plain, (off_t) 0, SEEK_SET) == -1){
254
212
    perror("pgpme_data_seek");
258
216
  
259
217
  *plaintext = NULL;
260
218
  while(true){
261
 
    plaintext_capacity = adjustbuffer(plaintext,
262
 
                                      (size_t)plaintext_length,
263
 
                                      plaintext_capacity);
264
 
    if (plaintext_capacity == 0){
265
 
        perror("adjustbuffer");
 
219
    if (plaintext_length + BUFFER_SIZE
 
220
        > (ssize_t) plaintext_capacity){
 
221
      *plaintext = realloc(*plaintext, plaintext_capacity
 
222
                            + BUFFER_SIZE);
 
223
      if (*plaintext == NULL){
 
224
        perror("realloc");
266
225
        plaintext_length = -1;
267
226
        goto decrypt_end;
 
227
      }
 
228
      plaintext_capacity += BUFFER_SIZE;
268
229
    }
269
230
    
270
231
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
281
242
    }
282
243
    plaintext_length += ret;
283
244
  }
284
 
  
 
245
 
285
246
  if(debug){
286
247
    fprintf(stderr, "Decrypted password is: ");
287
248
    for(ssize_t i = 0; i < plaintext_length; i++){
301
262
}
302
263
 
303
264
static const char * safer_gnutls_strerror (int value) {
304
 
  const char *ret = gnutls_strerror (value); /* Spurious warning */
 
265
  const char *ret = gnutls_strerror (value);
305
266
  if (ret == NULL)
306
267
    ret = "(unknown)";
307
268
  return ret;
313
274
  fprintf(stderr, "GnuTLS: %s", string);
314
275
}
315
276
 
316
 
static int init_gnutls_global(mandos_context *mc,
317
 
                              const char *pubkeyfilename,
318
 
                              const char *seckeyfilename){
 
277
static int initgnutls(mandos_context *mc, gnutls_session_t *session,
 
278
                      gnutls_dh_params_t *dh_params){
319
279
  int ret;
320
280
  
321
281
  if(debug){
322
282
    fprintf(stderr, "Initializing GnuTLS\n");
323
283
  }
324
 
  
325
 
  ret = gnutls_global_init();
326
 
  if (ret != GNUTLS_E_SUCCESS) {
 
284
 
 
285
  if ((ret = gnutls_global_init ())
 
286
      != GNUTLS_E_SUCCESS) {
327
287
    fprintf (stderr, "GnuTLS global_init: %s\n",
328
288
             safer_gnutls_strerror(ret));
329
289
    return -1;
338
298
  }
339
299
  
340
300
  /* OpenPGP credentials */
341
 
  gnutls_certificate_allocate_credentials(&mc->cred);
342
 
  if (ret != GNUTLS_E_SUCCESS){
343
 
    fprintf (stderr, "GnuTLS memory error: %s\n", /* Spurious
344
 
                                                     warning */
 
301
  if ((ret = gnutls_certificate_allocate_credentials (&mc->cred))
 
302
      != GNUTLS_E_SUCCESS) {
 
303
    fprintf (stderr, "GnuTLS memory error: %s\n",
345
304
             safer_gnutls_strerror(ret));
346
 
    gnutls_global_deinit ();
347
305
    return -1;
348
306
  }
349
307
  
350
308
  if(debug){
351
 
    fprintf(stderr, "Attempting to use OpenPGP public key %s and"
352
 
            " secret key %s as GnuTLS credentials\n", pubkeyfilename,
353
 
            seckeyfilename);
 
309
    fprintf(stderr, "Attempting to use OpenPGP certificate %s"
 
310
            " and keyfile %s as GnuTLS credentials\n", pubkeyfile,
 
311
            seckeyfile);
354
312
  }
355
313
  
356
314
  ret = gnutls_certificate_set_openpgp_key_file
357
 
    (mc->cred, pubkeyfilename, seckeyfilename,
358
 
     GNUTLS_OPENPGP_FMT_BASE64);
 
315
    (mc->cred, pubkeyfile, seckeyfile, GNUTLS_OPENPGP_FMT_BASE64);
359
316
  if (ret != GNUTLS_E_SUCCESS) {
360
317
    fprintf(stderr,
361
318
            "Error[%d] while reading the OpenPGP key pair ('%s',"
362
 
            " '%s')\n", ret, pubkeyfilename, seckeyfilename);
363
 
    fprintf(stderr, "The GnuTLS error is: %s\n",
 
319
            " '%s')\n", ret, pubkeyfile, seckeyfile);
 
320
    fprintf(stdout, "The GnuTLS error is: %s\n",
364
321
            safer_gnutls_strerror(ret));
365
 
    goto globalfail;
 
322
    return -1;
366
323
  }
367
324
  
368
325
  /* GnuTLS server initialization */
369
 
  ret = gnutls_dh_params_init(&mc->dh_params);
 
326
  ret = gnutls_dh_params_init(dh_params);
370
327
  if (ret != GNUTLS_E_SUCCESS) {
371
328
    fprintf (stderr, "Error in GnuTLS DH parameter initialization:"
372
329
             " %s\n", safer_gnutls_strerror(ret));
373
 
    goto globalfail;
 
330
    return -1;
374
331
  }
375
 
  ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
 
332
  ret = gnutls_dh_params_generate2(*dh_params, mc->dh_bits);
376
333
  if (ret != GNUTLS_E_SUCCESS) {
377
334
    fprintf (stderr, "Error in GnuTLS prime generation: %s\n",
378
335
             safer_gnutls_strerror(ret));
379
 
    goto globalfail;
 
336
    return -1;
380
337
  }
381
338
  
382
 
  gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
383
 
  
384
 
  return 0;
385
 
  
386
 
 globalfail:
387
 
  
388
 
  gnutls_certificate_free_credentials(mc->cred);
389
 
  gnutls_global_deinit();
390
 
  return -1;
391
 
}
392
 
 
393
 
static int init_gnutls_session(mandos_context *mc,
394
 
                               gnutls_session_t *session){
395
 
  int ret;
 
339
  gnutls_certificate_set_dh_params(mc->cred, *dh_params);
 
340
  
396
341
  /* GnuTLS session creation */
397
342
  ret = gnutls_init(session, GNUTLS_SERVER);
398
343
  if (ret != GNUTLS_E_SUCCESS){
407
352
      fprintf(stderr, "Syntax error at: %s\n", err);
408
353
      fprintf(stderr, "GnuTLS error: %s\n",
409
354
              safer_gnutls_strerror(ret));
410
 
      gnutls_deinit (*session);
411
355
      return -1;
412
356
    }
413
357
  }
417
361
  if (ret != GNUTLS_E_SUCCESS) {
418
362
    fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
419
363
            safer_gnutls_strerror(ret));
420
 
    gnutls_deinit (*session);
421
364
    return -1;
422
365
  }
423
366
  
439
382
                                      AvahiIfIndex if_index,
440
383
                                      mandos_context *mc){
441
384
  int ret, tcp_sd;
442
 
  union { struct sockaddr in; struct sockaddr_in6 in6; } to;
 
385
  struct sockaddr_in6 to;
443
386
  char *buffer = NULL;
444
387
  char *decrypted_buffer;
445
388
  size_t buffer_length = 0;
446
389
  size_t buffer_capacity = 0;
447
390
  ssize_t decrypted_buffer_size;
448
 
  size_t written;
 
391
  size_t written = 0;
449
392
  int retval = 0;
450
393
  char interface[IF_NAMESIZE];
451
394
  gnutls_session_t session;
 
395
  gnutls_dh_params_t dh_params;
452
396
  
453
 
  ret = init_gnutls_session (mc, &session);
 
397
  ret = initgnutls (mc, &session, &dh_params);
454
398
  if (ret != 0){
455
399
    return -1;
456
400
  }
457
401
  
458
402
  if(debug){
459
 
    fprintf(stderr, "Setting up a tcp connection to %s, port %" PRIu16
460
 
            "\n", ip, port);
 
403
    fprintf(stderr, "Setting up a tcp connection to %s, port %d\n",
 
404
            ip, port);
461
405
  }
462
406
  
463
407
  tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
465
409
    perror("socket");
466
410
    return -1;
467
411
  }
468
 
  
 
412
 
469
413
  if(debug){
470
414
    if(if_indextoname((unsigned int)if_index, interface) == NULL){
471
415
      perror("if_indextoname");
474
418
    fprintf(stderr, "Binding to interface %s\n", interface);
475
419
  }
476
420
  
477
 
  memset(&to, 0, sizeof(to));
478
 
  to.in6.sin6_family = AF_INET6;
 
421
  memset(&to,0,sizeof(to));     /* Spurious warning */
 
422
  to.sin6_family = AF_INET6;
479
423
  /* It would be nice to have a way to detect if we were passed an
480
424
     IPv4 address here.   Now we assume an IPv6 address. */
481
 
  ret = inet_pton(AF_INET6, ip, &to.in6.sin6_addr);
 
425
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
482
426
  if (ret < 0 ){
483
427
    perror("inet_pton");
484
428
    return -1;
487
431
    fprintf(stderr, "Bad address: %s\n", ip);
488
432
    return -1;
489
433
  }
490
 
  to.in6.sin6_port = htons(port); /* Spurious warning */
 
434
  to.sin6_port = htons(port);   /* Spurious warning */
491
435
  
492
 
  to.in6.sin6_scope_id = (uint32_t)if_index;
 
436
  to.sin6_scope_id = (uint32_t)if_index;
493
437
  
494
438
  if(debug){
495
 
    fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
496
 
            port);
 
439
    fprintf(stderr, "Connection to: %s, port %d\n", ip, port);
497
440
    char addrstr[INET6_ADDRSTRLEN] = "";
498
 
    if(inet_ntop(to.in6.sin6_family, &(to.in6.sin6_addr), addrstr,
 
441
    if(inet_ntop(to.sin6_family, &(to.sin6_addr), addrstr,
499
442
                 sizeof(addrstr)) == NULL){
500
443
      perror("inet_ntop");
501
444
    } else {
505
448
    }
506
449
  }
507
450
  
508
 
  ret = connect(tcp_sd, &to.in, sizeof(to));
 
451
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
509
452
  if (ret < 0){
510
453
    perror("connect");
511
454
    return -1;
512
455
  }
513
456
  
514
 
  const char *out = mandos_protocol_version;
515
 
  written = 0;
516
 
  while (true){
517
 
    size_t out_size = strlen(out);
518
 
    ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
519
 
                                   out_size - written));
520
 
    if (ret == -1){
521
 
      perror("write");
522
 
      retval = -1;
523
 
      goto mandos_end;
524
 
    }
525
 
    written += (size_t)ret;
526
 
    if(written < out_size){
527
 
      continue;
528
 
    } else {
529
 
      if (out == mandos_protocol_version){
530
 
        written = 0;
531
 
        out = "\r\n";
532
 
      } else {
533
 
        break;
534
 
      }
535
 
    }
536
 
  }
537
 
  
538
457
  if(debug){
539
458
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
540
459
  }
541
460
  
542
461
  gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) tcp_sd);
543
462
  
544
 
  do{
545
 
    ret = gnutls_handshake (session);
546
 
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
463
  ret = gnutls_handshake (session);
547
464
  
548
465
  if (ret != GNUTLS_E_SUCCESS){
549
466
    if(debug){
560
477
    fprintf(stderr, "Retrieving pgp encrypted password from %s\n",
561
478
            ip);
562
479
  }
563
 
  
 
480
 
564
481
  while(true){
565
 
    buffer_capacity = adjustbuffer(&buffer, buffer_length,
566
 
                                   buffer_capacity);
567
 
    if (buffer_capacity == 0){
568
 
      perror("adjustbuffer");
569
 
      retval = -1;
570
 
      goto mandos_end;
 
482
    if (buffer_length + BUFFER_SIZE > buffer_capacity){
 
483
      buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
 
484
      if (buffer == NULL){
 
485
        perror("realloc");
 
486
        retval = -1;
 
487
        goto mandos_end;
 
488
      }
 
489
      buffer_capacity += BUFFER_SIZE;
571
490
    }
572
491
    
573
492
    ret = gnutls_record_recv(session, buffer+buffer_length,
581
500
      case GNUTLS_E_AGAIN:
582
501
        break;
583
502
      case GNUTLS_E_REHANDSHAKE:
584
 
        do{
585
 
          ret = gnutls_handshake (session);
586
 
        } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
503
        ret = gnutls_handshake (session);
587
504
        if (ret < 0){
588
505
          fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
589
506
          gnutls_perror (ret);
615
532
                                               &decrypted_buffer,
616
533
                                               keydir);
617
534
    if (decrypted_buffer_size >= 0){
618
 
      written = 0;
619
535
      while(written < (size_t) decrypted_buffer_size){
620
536
        ret = (int)fwrite (decrypted_buffer + written, 1,
621
537
                           (size_t)decrypted_buffer_size - written,
634
550
    } else {
635
551
      retval = -1;
636
552
    }
637
 
  } else {
638
 
    retval = -1;
639
553
  }
640
554
  
641
555
  /* Shutdown procedure */
644
558
  free(buffer);
645
559
  close(tcp_sd);
646
560
  gnutls_deinit (session);
 
561
  gnutls_certificate_free_credentials (mc->cred);
 
562
  gnutls_global_deinit ();
647
563
  return retval;
648
564
}
649
565
 
662
578
                             flags,
663
579
                             void* userdata) {
664
580
  mandos_context *mc = userdata;
665
 
  assert(r);
 
581
  assert(r);                    /* Spurious warning */
666
582
  
667
583
  /* Called whenever a service has been resolved successfully or
668
584
     timed out */
680
596
      char ip[AVAHI_ADDRESS_STR_MAX];
681
597
      avahi_address_snprint(ip, sizeof(ip), address);
682
598
      if(debug){
683
 
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
684
 
                PRIu16 ") on port %d\n", name, host_name, ip,
685
 
                interface, port);
 
599
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %d) on"
 
600
                " port %d\n", name, host_name, ip, interface, port);
686
601
      }
687
602
      int ret = start_mandos_communication(ip, port, interface, mc);
688
603
      if (ret == 0){
689
 
        avahi_simple_poll_quit(mc->simple_poll);
 
604
        exit(EXIT_SUCCESS);
690
605
      }
691
606
    }
692
607
  }
704
619
                             flags,
705
620
                             void* userdata) {
706
621
  mandos_context *mc = userdata;
707
 
  assert(b);
 
622
  assert(b);                    /* Spurious warning */
708
623
  
709
624
  /* Called whenever a new services becomes available on the LAN or
710
625
     is removed from the LAN */
746
661
 
747
662
/* Combines file name and path and returns the malloced new
748
663
   string. some sane checks could/should be added */
749
 
static char *combinepath(const char *first, const char *second){
750
 
  char *tmp;
751
 
  int ret = asprintf(&tmp, "%s/%s", first, second);
752
 
  if(ret < 0){
 
664
static const char *combinepath(const char *first, const char *second){
 
665
  size_t f_len = strlen(first);
 
666
  size_t s_len = strlen(second);
 
667
  char *tmp = malloc(f_len + s_len + 2);
 
668
  if (tmp == NULL){
753
669
    return NULL;
754
670
  }
 
671
  if(f_len > 0){
 
672
    memcpy(tmp, first, f_len);  /* Spurious warning */
 
673
  }
 
674
  tmp[f_len] = '/';
 
675
  if(s_len > 0){
 
676
    memcpy(tmp + f_len + 1, second, s_len); /* Spurious warning */
 
677
  }
 
678
  tmp[f_len + 1 + s_len] = '\0';
755
679
  return tmp;
756
680
}
757
681
 
764
688
    const char *interface = "eth0";
765
689
    struct ifreq network;
766
690
    int sd;
767
 
    uid_t uid;
768
 
    gid_t gid;
769
691
    char *connect_to = NULL;
770
692
    AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
771
 
    char *pubkeyfilename = NULL;
772
 
    char *seckeyfilename = NULL;
773
 
    const char *pubkeyname = "pubkey.txt";
774
 
    const char *seckeyname = "seckey.txt";
775
693
    mandos_context mc = { .simple_poll = NULL, .server = NULL,
776
 
                          .dh_bits = 1024, .priority = "SECURE256"
777
 
                          ":!CTYPE-X.509:+CTYPE-OPENPGP" };
778
 
    bool gnutls_initalized = false;
 
694
                          .dh_bits = 1024, .priority = "SECURE256"};
779
695
    
780
696
    {
781
 
      struct argp_option options[] = {
782
 
        { .name = "debug", .key = 128,
783
 
          .doc = "Debug mode", .group = 3 },
784
 
        { .name = "connect", .key = 'c',
785
 
          .arg = "ADDRESS:PORT",
786
 
          .doc = "Connect directly to a specific Mandos server",
787
 
          .group = 1 },
788
 
        { .name = "interface", .key = 'i',
789
 
          .arg = "NAME",
790
 
          .doc = "Interface that will be used to search for Mandos"
791
 
          " servers",
792
 
          .group = 1 },
793
 
        { .name = "keydir", .key = 'd',
794
 
          .arg = "DIRECTORY",
795
 
          .doc = "Directory to read the OpenPGP key files from",
796
 
          .group = 1 },
797
 
        { .name = "seckey", .key = 's',
798
 
          .arg = "FILE",
799
 
          .doc = "OpenPGP secret key file base name",
800
 
          .group = 1 },
801
 
        { .name = "pubkey", .key = 'p',
802
 
          .arg = "FILE",
803
 
          .doc = "OpenPGP public key file base name",
804
 
          .group = 2 },
805
 
        { .name = "dh-bits", .key = 129,
806
 
          .arg = "BITS",
807
 
          .doc = "Bit length of the prime number used in the"
808
 
          " Diffie-Hellman key exchange",
809
 
          .group = 2 },
810
 
        { .name = "priority", .key = 130,
811
 
          .arg = "STRING",
812
 
          .doc = "GnuTLS priority string for the TLS handshake",
813
 
          .group = 1 },
814
 
        { .name = NULL }
815
 
      };
816
 
      
817
 
      error_t parse_opt (int key, char *arg,
818
 
                         struct argp_state *state) {
819
 
        /* Get the INPUT argument from `argp_parse', which we know is
820
 
           a pointer to our plugin list pointer. */
821
 
        switch (key) {
822
 
        case 128:               /* --debug */
823
 
          debug = true;
824
 
          break;
825
 
        case 'c':               /* --connect */
826
 
          connect_to = arg;
827
 
          break;
828
 
        case 'i':               /* --interface */
829
 
          interface = arg;
830
 
          break;
831
 
        case 'd':               /* --keydir */
832
 
          keydir = arg;
833
 
          break;
834
 
        case 's':               /* --seckey */
835
 
          seckeyname = arg;
836
 
          break;
837
 
        case 'p':               /* --pubkey */
838
 
          pubkeyname = arg;
839
 
          break;
840
 
        case 129:               /* --dh-bits */
 
697
      /* Temporary int to get the address of for getopt_long */
 
698
      int debug_int = debug ? 1 : 0;
 
699
      while (true){
 
700
        struct option long_options[] = {
 
701
          {"debug", no_argument, &debug_int, 1},
 
702
          {"connect", required_argument, NULL, 'c'},
 
703
          {"interface", required_argument, NULL, 'i'},
 
704
          {"keydir", required_argument, NULL, 'd'},
 
705
          {"seckey", required_argument, NULL, 's'},
 
706
          {"pubkey", required_argument, NULL, 'p'},
 
707
          {"dh-bits", required_argument, NULL, 'D'},
 
708
          {"priority", required_argument, NULL, 'P'},
 
709
          {0, 0, 0, 0} };
 
710
      
 
711
        int option_index = 0;
 
712
        ret = getopt_long (argc, argv, "i:", long_options,
 
713
                           &option_index);
 
714
      
 
715
        if (ret == -1){
 
716
          break;
 
717
        }
 
718
      
 
719
        switch(ret){
 
720
        case 0:
 
721
          break;
 
722
        case 'i':
 
723
          interface = optarg;
 
724
          break;
 
725
        case 'c':
 
726
          connect_to = optarg;
 
727
          break;
 
728
        case 'd':
 
729
          keydir = optarg;
 
730
          break;
 
731
        case 'p':
 
732
          pubkeyfile = optarg;
 
733
          break;
 
734
        case 's':
 
735
          seckeyfile = optarg;
 
736
          break;
 
737
        case 'D':
841
738
          errno = 0;
842
 
          mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);
 
739
          mc.dh_bits = (unsigned int) strtol(optarg, NULL, 10);
843
740
          if (errno){
844
741
            perror("strtol");
845
742
            exit(EXIT_FAILURE);
846
743
          }
847
744
          break;
848
 
        case 130:               /* --priority */
849
 
          mc.priority = arg;
850
 
          break;
851
 
        case ARGP_KEY_ARG:
852
 
          argp_usage (state);
853
 
        case ARGP_KEY_END:
854
 
          break;
 
745
        case 'P':
 
746
          mc.priority = optarg;
 
747
          break;
 
748
        case '?':
855
749
        default:
856
 
          return ARGP_ERR_UNKNOWN;
857
 
        }
858
 
        return 0;
859
 
      }
860
 
      
861
 
      struct argp argp = { .options = options, .parser = parse_opt,
862
 
                           .args_doc = "",
863
 
                           .doc = "Mandos client -- Get and decrypt"
864
 
                           " passwords from a Mandos server" };
865
 
      ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
866
 
      if (ret == ARGP_ERR_UNKNOWN){
867
 
        fprintf(stderr, "Unknown error while parsing arguments\n");
868
 
        exitcode = EXIT_FAILURE;
869
 
        goto end;
870
 
      }
871
 
    }
872
 
    
873
 
    pubkeyfilename = combinepath(keydir, pubkeyname);
874
 
    if (pubkeyfilename == NULL){
875
 
      perror("combinepath");
876
 
      exitcode = EXIT_FAILURE;
877
 
      goto end;
878
 
    }
879
 
    
880
 
    seckeyfilename = combinepath(keydir, seckeyname);
881
 
    if (seckeyfilename == NULL){
882
 
      perror("combinepath");
883
 
      exitcode = EXIT_FAILURE;
884
 
      goto end;
885
 
    }
886
 
    
887
 
    ret = init_gnutls_global(&mc, pubkeyfilename, seckeyfilename);
888
 
    if (ret == -1){
889
 
      fprintf(stderr, "init_gnutls_global failed\n");
890
 
      exitcode = EXIT_FAILURE;
891
 
      goto end;
892
 
    } else {
893
 
      gnutls_initalized = true;
894
 
    }
895
 
    
896
 
    /* If the interface is down, bring it up */
897
 
    {
898
 
      sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
899
 
      if(sd < 0) {
900
 
        perror("socket");
901
 
        exitcode = EXIT_FAILURE;
902
 
        goto end;
903
 
      }
904
 
      strcpy(network.ifr_name, interface);
905
 
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
906
 
      if(ret == -1){
907
 
        perror("ioctl SIOCGIFFLAGS");
908
 
        exitcode = EXIT_FAILURE;
909
 
        goto end;
910
 
      }
911
 
      if((network.ifr_flags & IFF_UP) == 0){
912
 
        network.ifr_flags |= IFF_UP;
913
 
        ret = ioctl(sd, SIOCSIFFLAGS, &network);
914
 
        if(ret == -1){
915
 
          perror("ioctl SIOCSIFFLAGS");
916
 
          exitcode = EXIT_FAILURE;
917
 
          goto end;
918
 
        }
919
 
      }
920
 
      close(sd);
921
 
    }
922
 
    
923
 
    uid = getuid();
924
 
    gid = getgid();
925
 
    
926
 
    ret = setuid(uid);
927
 
    if (ret == -1){
928
 
      perror("setuid");
929
 
    }
930
 
    
931
 
    setgid(gid);
932
 
    if (ret == -1){
933
 
      perror("setgid");
 
750
          /* getopt_long() has already printed a message about the
 
751
             unrcognized option, so just exit. */
 
752
          exit(EXIT_FAILURE);
 
753
        }
 
754
      }
 
755
      /* Set the global debug flag from the temporary int */
 
756
      debug = debug_int ? true : false;
 
757
    }
 
758
    
 
759
    pubkeyfile = combinepath(keydir, pubkeyfile);
 
760
    if (pubkeyfile == NULL){
 
761
      perror("combinepath");
 
762
      exitcode = EXIT_FAILURE;
 
763
      goto end;
 
764
    }
 
765
    
 
766
    seckeyfile = combinepath(keydir, seckeyfile);
 
767
    if (seckeyfile == NULL){
 
768
      perror("combinepath");
 
769
      goto end;
934
770
    }
935
771
    
936
772
    if_index = (AvahiIfIndex) if_nametoindex(interface);
945
781
      char *address = strrchr(connect_to, ':');
946
782
      if(address == NULL){
947
783
        fprintf(stderr, "No colon in address\n");
948
 
        exitcode = EXIT_FAILURE;
949
 
        goto end;
 
784
        exit(EXIT_FAILURE);
950
785
      }
951
786
      errno = 0;
952
787
      uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
953
788
      if(errno){
954
789
        perror("Bad port number");
955
 
        exitcode = EXIT_FAILURE;
956
 
        goto end;
 
790
        exit(EXIT_FAILURE);
957
791
      }
958
792
      *address = '\0';
959
793
      address = connect_to;
960
794
      ret = start_mandos_communication(address, port, if_index, &mc);
961
795
      if(ret < 0){
962
 
        exitcode = EXIT_FAILURE;
 
796
        exit(EXIT_FAILURE);
963
797
      } else {
964
 
        exitcode = EXIT_SUCCESS;
965
 
      }
966
 
      goto end;
 
798
        exit(EXIT_SUCCESS);
 
799
      }
 
800
    }
 
801
    
 
802
    /* If the interface is down, bring it up */
 
803
    {
 
804
      sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
 
805
      if(sd < 0) {
 
806
        perror("socket");
 
807
        exitcode = EXIT_FAILURE;
 
808
        goto end;
 
809
      }
 
810
      strcpy(network.ifr_name, interface); /* Spurious warning */
 
811
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
812
      if(ret == -1){
 
813
        perror("ioctl SIOCGIFFLAGS");
 
814
        exitcode = EXIT_FAILURE;
 
815
        goto end;
 
816
      }
 
817
      if((network.ifr_flags & IFF_UP) == 0){
 
818
        network.ifr_flags |= IFF_UP;
 
819
        ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
820
        if(ret == -1){
 
821
          perror("ioctl SIOCSIFFLAGS");
 
822
          exitcode = EXIT_FAILURE;
 
823
          goto end;
 
824
        }
 
825
      }
 
826
      close(sd);
967
827
    }
968
828
    
969
829
    if (not debug){
981
841
        exitcode = EXIT_FAILURE;
982
842
        goto end;
983
843
    }
984
 
    
 
844
 
985
845
    {
986
846
      AvahiServerConfig config;
987
847
      /* Do not publish any local Zeroconf records */
990
850
      config.publish_addresses = 0;
991
851
      config.publish_workstation = 0;
992
852
      config.publish_domain = 0;
993
 
      
 
853
 
994
854
      /* Allocate a new server */
995
855
      mc.server = avahi_server_new(avahi_simple_poll_get
996
856
                                   (mc.simple_poll), &config, NULL,
997
857
                                   NULL, &error);
998
 
      
 
858
    
999
859
      /* Free the Avahi configuration data */
1000
860
      avahi_server_config_free(&config);
1001
861
    }
1021
881
    }
1022
882
    
1023
883
    /* Run the main loop */
1024
 
    
 
884
 
1025
885
    if (debug){
1026
886
      fprintf(stderr, "Starting Avahi loop search\n");
1027
887
    }
1029
889
    avahi_simple_poll_loop(mc.simple_poll);
1030
890
    
1031
891
 end:
1032
 
    
 
892
 
1033
893
    if (debug){
1034
894
      fprintf(stderr, "%s exiting\n", argv[0]);
1035
895
    }
1040
900
    
1041
901
    if (mc.server != NULL)
1042
902
        avahi_server_free(mc.server);
1043
 
    
 
903
 
1044
904
    if (mc.simple_poll != NULL)
1045
905
        avahi_simple_poll_free(mc.simple_poll);
1046
 
    free(pubkeyfilename);
1047
 
    free(seckeyfilename);
1048
 
    
1049
 
    if (gnutls_initalized){
1050
 
      gnutls_certificate_free_credentials(mc.cred);
1051
 
      gnutls_global_deinit ();
1052
 
    }
 
906
    free(pubkeyfile);
 
907
    free(seckeyfile);
1053
908
    
1054
909
    return exitcode;
1055
910
}