/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,
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
309
    fprintf(stderr, "Attempting to use OpenPGP certificate %s"
352
 
            " and keyfile %s as GnuTLS credentials\n", pubkeyfilename,
353
 
            seckeyfilename);
 
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);
 
319
            " '%s')\n", ret, pubkeyfile, seckeyfile);
363
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
 
 
394
 
static int init_gnutls_session(mandos_context *mc,
395
 
                               gnutls_session_t *session){
396
 
  int ret;
 
339
  gnutls_certificate_set_dh_params(mc->cred, *dh_params);
 
340
  
397
341
  /* GnuTLS session creation */
398
342
  ret = gnutls_init(session, GNUTLS_SERVER);
399
343
  if (ret != GNUTLS_E_SUCCESS){
408
352
      fprintf(stderr, "Syntax error at: %s\n", err);
409
353
      fprintf(stderr, "GnuTLS error: %s\n",
410
354
              safer_gnutls_strerror(ret));
411
 
      gnutls_deinit (*session);
412
355
      return -1;
413
356
    }
414
357
  }
418
361
  if (ret != GNUTLS_E_SUCCESS) {
419
362
    fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
420
363
            safer_gnutls_strerror(ret));
421
 
    gnutls_deinit (*session);
422
364
    return -1;
423
365
  }
424
366
  
440
382
                                      AvahiIfIndex if_index,
441
383
                                      mandos_context *mc){
442
384
  int ret, tcp_sd;
443
 
  union { struct sockaddr in; struct sockaddr_in6 in6; } to;
 
385
  struct sockaddr_in6 to;
444
386
  char *buffer = NULL;
445
387
  char *decrypted_buffer;
446
388
  size_t buffer_length = 0;
447
389
  size_t buffer_capacity = 0;
448
390
  ssize_t decrypted_buffer_size;
449
 
  size_t written;
 
391
  size_t written = 0;
450
392
  int retval = 0;
451
393
  char interface[IF_NAMESIZE];
452
394
  gnutls_session_t session;
 
395
  gnutls_dh_params_t dh_params;
453
396
  
454
 
  ret = init_gnutls_session (mc, &session);
 
397
  ret = initgnutls (mc, &session, &dh_params);
455
398
  if (ret != 0){
456
399
    return -1;
457
400
  }
458
401
  
459
402
  if(debug){
460
 
    fprintf(stderr, "Setting up a tcp connection to %s, port %" PRIu16
461
 
            "\n", ip, port);
 
403
    fprintf(stderr, "Setting up a tcp connection to %s, port %d\n",
 
404
            ip, port);
462
405
  }
463
406
  
464
407
  tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
475
418
    fprintf(stderr, "Binding to interface %s\n", interface);
476
419
  }
477
420
  
478
 
  memset(&to, 0, sizeof(to));
479
 
  to.in6.sin6_family = AF_INET6;
 
421
  memset(&to,0,sizeof(to));     /* Spurious warning */
 
422
  to.sin6_family = AF_INET6;
480
423
  /* It would be nice to have a way to detect if we were passed an
481
424
     IPv4 address here.   Now we assume an IPv6 address. */
482
 
  ret = inet_pton(AF_INET6, ip, &to.in6.sin6_addr);
 
425
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
483
426
  if (ret < 0 ){
484
427
    perror("inet_pton");
485
428
    return -1;
488
431
    fprintf(stderr, "Bad address: %s\n", ip);
489
432
    return -1;
490
433
  }
491
 
  to.in6.sin6_port = htons(port); /* Spurious warning */
 
434
  to.sin6_port = htons(port);   /* Spurious warning */
492
435
  
493
 
  to.in6.sin6_scope_id = (uint32_t)if_index;
 
436
  to.sin6_scope_id = (uint32_t)if_index;
494
437
  
495
438
  if(debug){
496
 
    fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
497
 
            port);
 
439
    fprintf(stderr, "Connection to: %s, port %d\n", ip, port);
498
440
    char addrstr[INET6_ADDRSTRLEN] = "";
499
 
    if(inet_ntop(to.in6.sin6_family, &(to.in6.sin6_addr), addrstr,
 
441
    if(inet_ntop(to.sin6_family, &(to.sin6_addr), addrstr,
500
442
                 sizeof(addrstr)) == NULL){
501
443
      perror("inet_ntop");
502
444
    } else {
506
448
    }
507
449
  }
508
450
  
509
 
  ret = connect(tcp_sd, &to.in, sizeof(to));
 
451
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
510
452
  if (ret < 0){
511
453
    perror("connect");
512
454
    return -1;
513
455
  }
514
 
 
515
 
  const char *out = mandos_protocol_version;
516
 
  written = 0;
517
 
  while (true){
518
 
    size_t out_size = strlen(out);
519
 
    ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
520
 
                                   out_size - written));
521
 
    if (ret == -1){
522
 
      perror("write");
523
 
      retval = -1;
524
 
      goto mandos_end;
525
 
    }
526
 
    written += (size_t)ret;
527
 
    if(written < out_size){
528
 
      continue;
529
 
    } else {
530
 
      if (out == mandos_protocol_version){
531
 
        written = 0;
532
 
        out = "\r\n";
533
 
      } else {
534
 
        break;
535
 
      }
536
 
    }
537
 
  }
538
 
 
 
456
  
539
457
  if(debug){
540
458
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
541
459
  }
542
460
  
543
461
  gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) tcp_sd);
544
 
 
545
 
  do{
546
 
    ret = gnutls_handshake (session);
547
 
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
462
  
 
463
  ret = gnutls_handshake (session);
548
464
  
549
465
  if (ret != GNUTLS_E_SUCCESS){
550
466
    if(debug){
563
479
  }
564
480
 
565
481
  while(true){
566
 
    buffer_capacity = adjustbuffer(&buffer, buffer_length,
567
 
                                   buffer_capacity);
568
 
    if (buffer_capacity == 0){
569
 
      perror("adjustbuffer");
570
 
      retval = -1;
571
 
      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;
572
490
    }
573
491
    
574
492
    ret = gnutls_record_recv(session, buffer+buffer_length,
582
500
      case GNUTLS_E_AGAIN:
583
501
        break;
584
502
      case GNUTLS_E_REHANDSHAKE:
585
 
        do{
586
 
          ret = gnutls_handshake (session);
587
 
        } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
503
        ret = gnutls_handshake (session);
588
504
        if (ret < 0){
589
505
          fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
590
506
          gnutls_perror (ret);
616
532
                                               &decrypted_buffer,
617
533
                                               keydir);
618
534
    if (decrypted_buffer_size >= 0){
619
 
      written = 0;
620
535
      while(written < (size_t) decrypted_buffer_size){
621
536
        ret = (int)fwrite (decrypted_buffer + written, 1,
622
537
                           (size_t)decrypted_buffer_size - written,
635
550
    } else {
636
551
      retval = -1;
637
552
    }
638
 
  } else {
639
 
    retval = -1;
640
553
  }
641
554
  
642
555
  /* Shutdown procedure */
645
558
  free(buffer);
646
559
  close(tcp_sd);
647
560
  gnutls_deinit (session);
 
561
  gnutls_certificate_free_credentials (mc->cred);
 
562
  gnutls_global_deinit ();
648
563
  return retval;
649
564
}
650
565
 
663
578
                             flags,
664
579
                             void* userdata) {
665
580
  mandos_context *mc = userdata;
666
 
  assert(r);
 
581
  assert(r);                    /* Spurious warning */
667
582
  
668
583
  /* Called whenever a service has been resolved successfully or
669
584
     timed out */
681
596
      char ip[AVAHI_ADDRESS_STR_MAX];
682
597
      avahi_address_snprint(ip, sizeof(ip), address);
683
598
      if(debug){
684
 
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
685
 
                PRIu16 ") on port %d\n", name, host_name, ip,
686
 
                interface, port);
 
599
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %d) on"
 
600
                " port %d\n", name, host_name, ip, interface, port);
687
601
      }
688
602
      int ret = start_mandos_communication(ip, port, interface, mc);
689
603
      if (ret == 0){
690
 
        avahi_simple_poll_quit(mc->simple_poll);
 
604
        exit(EXIT_SUCCESS);
691
605
      }
692
606
    }
693
607
  }
705
619
                             flags,
706
620
                             void* userdata) {
707
621
  mandos_context *mc = userdata;
708
 
  assert(b);
 
622
  assert(b);                    /* Spurious warning */
709
623
  
710
624
  /* Called whenever a new services becomes available on the LAN or
711
625
     is removed from the LAN */
747
661
 
748
662
/* Combines file name and path and returns the malloced new
749
663
   string. some sane checks could/should be added */
750
 
static char *combinepath(const char *first, const char *second){
751
 
  char *tmp;
752
 
  int ret = asprintf(&tmp, "%s/%s", first, second);
753
 
  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){
754
669
    return NULL;
755
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';
756
679
  return tmp;
757
680
}
758
681
 
765
688
    const char *interface = "eth0";
766
689
    struct ifreq network;
767
690
    int sd;
768
 
    uid_t uid;
769
 
    gid_t gid;
770
691
    char *connect_to = NULL;
771
692
    AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
772
 
    char *pubkeyfilename = NULL;
773
 
    char *seckeyfilename = NULL;
774
 
    const char *pubkeyname = "pubkey.txt";
775
 
    const char *seckeyname = "seckey.txt";
776
693
    mandos_context mc = { .simple_poll = NULL, .server = NULL,
777
694
                          .dh_bits = 1024, .priority = "SECURE256"};
778
 
    bool gnutls_initalized = false;
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 = "IP",
786
 
          .doc = "Connect directly to a sepcified mandos server",
787
 
          .group = 1 },
788
 
        { .name = "interface", .key = 'i',
789
 
          .arg = "INTERFACE",
790
 
          .doc = "Interface that Avahi will conntect through",
791
 
          .group = 1 },
792
 
        { .name = "keydir", .key = 'd',
793
 
          .arg = "KEYDIR",
794
 
          .doc = "Directory where the openpgp keyring is",
795
 
          .group = 1 },
796
 
        { .name = "seckey", .key = 's',
797
 
          .arg = "SECKEY",
798
 
          .doc = "Secret openpgp key for gnutls authentication",
799
 
          .group = 1 },
800
 
        { .name = "pubkey", .key = 'p',
801
 
          .arg = "PUBKEY",
802
 
          .doc = "Public openpgp key for gnutls authentication",
803
 
          .group = 2 },
804
 
        { .name = "dh-bits", .key = 129,
805
 
          .arg = "BITS",
806
 
          .doc = "dh-bits to use in gnutls communication",
807
 
          .group = 2 },
808
 
        { .name = "priority", .key = 130,
809
 
          .arg = "PRIORITY",
810
 
          .doc = "GNUTLS priority", .group = 1 },
811
 
        { .name = NULL }
812
 
      };
813
 
 
814
 
      
815
 
      error_t parse_opt (int key, char *arg,
816
 
                         struct argp_state *state) {
817
 
        /* Get the INPUT argument from `argp_parse', which we know is
818
 
           a pointer to our plugin list pointer. */
819
 
        switch (key) {
820
 
        case 128:
821
 
          debug = true;
 
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;
822
724
          break;
823
725
        case 'c':
824
 
          connect_to = arg;
825
 
          break;
826
 
        case 'i':
827
 
          interface = arg;
 
726
          connect_to = optarg;
828
727
          break;
829
728
        case 'd':
830
 
          keydir = arg;
 
729
          keydir = optarg;
 
730
          break;
 
731
        case 'p':
 
732
          pubkeyfile = optarg;
831
733
          break;
832
734
        case 's':
833
 
          seckeyname = arg;
834
 
          break;
835
 
        case 'p':
836
 
          pubkeyname = arg;
837
 
          break;
838
 
        case 129:
 
735
          seckeyfile = optarg;
 
736
          break;
 
737
        case 'D':
839
738
          errno = 0;
840
 
          mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);
 
739
          mc.dh_bits = (unsigned int) strtol(optarg, NULL, 10);
841
740
          if (errno){
842
741
            perror("strtol");
843
742
            exit(EXIT_FAILURE);
844
743
          }
845
744
          break;
846
 
        case 130:
847
 
          mc.priority = arg;
848
 
          break;
849
 
        case ARGP_KEY_ARG:
850
 
          argp_usage (state);
851
 
        case ARGP_KEY_END:
852
 
          break;
 
745
        case 'P':
 
746
          mc.priority = optarg;
 
747
          break;
 
748
        case '?':
853
749
        default:
854
 
          return ARGP_ERR_UNKNOWN;
855
 
        }
856
 
        return 0;
857
 
      }
858
 
 
859
 
      struct argp argp = { .options = options, .parser = parse_opt,
860
 
                           .args_doc = "",
861
 
                           .doc = "Mandos client -- Get and decrypt"
862
 
                           " passwords from mandos server" };
863
 
      ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
864
 
      if (ret == ARGP_ERR_UNKNOWN){
865
 
        fprintf(stderr, "Unknown error while parsing arguments\n");
866
 
        exitcode = EXIT_FAILURE;
867
 
        goto end;
868
 
      }
869
 
    }
870
 
      
871
 
    pubkeyfilename = combinepath(keydir, pubkeyname);
872
 
    if (pubkeyfilename == NULL){
873
 
      perror("combinepath");
874
 
      exitcode = EXIT_FAILURE;
875
 
      goto end;
876
 
    }
877
 
    
878
 
    seckeyfilename = combinepath(keydir, seckeyname);
879
 
    if (seckeyfilename == NULL){
880
 
      perror("combinepath");
881
 
      exitcode = EXIT_FAILURE;
882
 
      goto end;
883
 
    }
884
 
 
885
 
    ret = init_gnutls_global(&mc, pubkeyfilename, seckeyfilename);
886
 
    if (ret == -1){
887
 
      fprintf(stderr, "init_gnutls_global failed\n");
888
 
      exitcode = EXIT_FAILURE;
889
 
      goto end;
890
 
    } else {
891
 
      gnutls_initalized = true;
892
 
    }
893
 
    
894
 
    /* If the interface is down, bring it up */
895
 
    {
896
 
      sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
897
 
      if(sd < 0) {
898
 
        perror("socket");
899
 
        exitcode = EXIT_FAILURE;
900
 
        goto end;
901
 
      }
902
 
      strcpy(network.ifr_name, interface);
903
 
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
904
 
      if(ret == -1){
905
 
        perror("ioctl SIOCGIFFLAGS");
906
 
        exitcode = EXIT_FAILURE;
907
 
        goto end;
908
 
      }
909
 
      if((network.ifr_flags & IFF_UP) == 0){
910
 
        network.ifr_flags |= IFF_UP;
911
 
        ret = ioctl(sd, SIOCSIFFLAGS, &network);
912
 
        if(ret == -1){
913
 
          perror("ioctl SIOCSIFFLAGS");
914
 
          exitcode = EXIT_FAILURE;
915
 
          goto end;
916
 
        }
917
 
      }
918
 
      close(sd);
919
 
    }
920
 
    
921
 
    uid = getuid();
922
 
    gid = getgid();
923
 
    
924
 
    ret = setuid(uid);
925
 
    if (ret == -1){
926
 
      perror("setuid");
927
 
    }
928
 
    
929
 
    setgid(gid);
930
 
    if (ret == -1){
931
 
      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;
932
770
    }
933
771
    
934
772
    if_index = (AvahiIfIndex) if_nametoindex(interface);
943
781
      char *address = strrchr(connect_to, ':');
944
782
      if(address == NULL){
945
783
        fprintf(stderr, "No colon in address\n");
946
 
        exitcode = EXIT_FAILURE;
947
 
        goto end;
 
784
        exit(EXIT_FAILURE);
948
785
      }
949
786
      errno = 0;
950
787
      uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
951
788
      if(errno){
952
789
        perror("Bad port number");
953
 
        exitcode = EXIT_FAILURE;
954
 
        goto end;
 
790
        exit(EXIT_FAILURE);
955
791
      }
956
792
      *address = '\0';
957
793
      address = connect_to;
958
794
      ret = start_mandos_communication(address, port, if_index, &mc);
959
795
      if(ret < 0){
960
 
        exitcode = EXIT_FAILURE;
 
796
        exit(EXIT_FAILURE);
961
797
      } else {
962
 
        exitcode = EXIT_SUCCESS;
963
 
      }
964
 
      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);
965
827
    }
966
828
    
967
829
    if (not debug){
1041
903
 
1042
904
    if (mc.simple_poll != NULL)
1043
905
        avahi_simple_poll_free(mc.simple_poll);
1044
 
    free(pubkeyfilename);
1045
 
    free(seckeyfilename);
1046
 
 
1047
 
    if (gnutls_initalized){
1048
 
      gnutls_certificate_free_credentials(mc.cred);
1049
 
      gnutls_global_deinit ();
1050
 
    }
 
906
    free(pubkeyfile);
 
907
    free(seckeyfile);
1051
908
    
1052
909
    return exitcode;
1053
910
}