/mandos/release

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

« 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() */
36
 
 
37
35
#include <stdio.h>
38
36
#include <assert.h>
39
37
#include <stdlib.h>
65
63
#include <arpa/inet.h>          /* inet_pton() */
66
64
#include <iso646.h>             /* not */
67
65
#include <net/if.h>             /* IF_NAMESIZE */
68
 
#include <argp.h>               /* struct argp_option,
69
 
                                   struct argp_state, struct argp,
70
 
                                   argp_parse() */
 
66
 
71
67
/* GPGME */
72
68
#include <errno.h>              /* perror() */
73
69
#include <gpgme.h>
74
70
 
 
71
/* getopt_long */
 
72
#include <getopt.h>
 
73
 
75
74
#define BUFFER_SIZE 256
76
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
 
77
80
bool debug = false;
78
 
static const char *keydir = "/conf/conf.d/mandos";
79
 
static const char mandos_protocol_version[] = "1";
80
 
const char *argp_program_version = "mandosclient 0.9";
81
 
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
82
81
 
83
 
/* Used for passing in values through the Avahi callback functions */
 
82
/* Used for passing in values through all the callback functions */
84
83
typedef struct {
85
84
  AvahiSimplePoll *simple_poll;
86
85
  AvahiServer *server;
87
86
  gnutls_certificate_credentials_t cred;
88
87
  unsigned int dh_bits;
89
 
  gnutls_dh_params_t dh_params;
90
88
  const char *priority;
91
89
} mandos_context;
92
90
 
93
 
/*
94
 
 * Make room in "buffer" for at least BUFFER_SIZE additional bytes.
95
 
 * "buffer_capacity" is how much is currently allocated,
96
 
 * "buffer_length" is how much is already used.
97
 
 */
98
 
size_t adjustbuffer(char **buffer, size_t buffer_length,
99
 
                  size_t buffer_capacity){
100
 
  if (buffer_length + BUFFER_SIZE > buffer_capacity){
101
 
    *buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
102
 
    if (buffer == NULL){
103
 
      return 0;
104
 
    }
105
 
    buffer_capacity += BUFFER_SIZE;
106
 
  }
107
 
  return buffer_capacity;
108
 
}
109
 
 
110
91
/* 
111
92
 * Decrypt OpenPGP data using keyrings in HOMEDIR.
112
93
 * Returns -1 on error
235
216
  
236
217
  *plaintext = NULL;
237
218
  while(true){
238
 
    plaintext_capacity = adjustbuffer(plaintext,
239
 
                                      (size_t)plaintext_length,
240
 
                                      plaintext_capacity);
241
 
    if (plaintext_capacity == 0){
242
 
        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");
243
225
        plaintext_length = -1;
244
226
        goto decrypt_end;
 
227
      }
 
228
      plaintext_capacity += BUFFER_SIZE;
245
229
    }
246
230
    
247
231
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
290
274
  fprintf(stderr, "GnuTLS: %s", string);
291
275
}
292
276
 
293
 
static int init_gnutls_global(mandos_context *mc,
294
 
                              const char *pubkeyfile,
295
 
                              const char *seckeyfile){
 
277
static int initgnutls(mandos_context *mc, gnutls_session_t *session,
 
278
                      gnutls_dh_params_t *dh_params){
296
279
  int ret;
297
280
  
298
281
  if(debug){
340
323
  }
341
324
  
342
325
  /* GnuTLS server initialization */
343
 
  ret = gnutls_dh_params_init(&mc->dh_params);
 
326
  ret = gnutls_dh_params_init(dh_params);
344
327
  if (ret != GNUTLS_E_SUCCESS) {
345
328
    fprintf (stderr, "Error in GnuTLS DH parameter initialization:"
346
329
             " %s\n", safer_gnutls_strerror(ret));
347
330
    return -1;
348
331
  }
349
 
  ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
 
332
  ret = gnutls_dh_params_generate2(*dh_params, mc->dh_bits);
350
333
  if (ret != GNUTLS_E_SUCCESS) {
351
334
    fprintf (stderr, "Error in GnuTLS prime generation: %s\n",
352
335
             safer_gnutls_strerror(ret));
353
336
    return -1;
354
337
  }
355
338
  
356
 
  gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
357
 
 
358
 
  return 0;
359
 
}
360
 
 
361
 
static int init_gnutls_session(mandos_context *mc,
362
 
                               gnutls_session_t *session){
363
 
  int ret;
 
339
  gnutls_certificate_set_dh_params(mc->cred, *dh_params);
 
340
  
364
341
  /* GnuTLS session creation */
365
342
  ret = gnutls_init(session, GNUTLS_SERVER);
366
343
  if (ret != GNUTLS_E_SUCCESS){
405
382
                                      AvahiIfIndex if_index,
406
383
                                      mandos_context *mc){
407
384
  int ret, tcp_sd;
408
 
  union { struct sockaddr in; struct sockaddr_in6 in6; } to;
 
385
  struct sockaddr_in6 to;
409
386
  char *buffer = NULL;
410
387
  char *decrypted_buffer;
411
388
  size_t buffer_length = 0;
412
389
  size_t buffer_capacity = 0;
413
390
  ssize_t decrypted_buffer_size;
414
 
  size_t written;
 
391
  size_t written = 0;
415
392
  int retval = 0;
416
393
  char interface[IF_NAMESIZE];
417
394
  gnutls_session_t session;
 
395
  gnutls_dh_params_t dh_params;
418
396
  
419
 
  ret = init_gnutls_session (mc, &session);
 
397
  ret = initgnutls (mc, &session, &dh_params);
420
398
  if (ret != 0){
421
399
    return -1;
422
400
  }
441
419
  }
442
420
  
443
421
  memset(&to,0,sizeof(to));     /* Spurious warning */
444
 
  to.in6.sin6_family = AF_INET6;
 
422
  to.sin6_family = AF_INET6;
445
423
  /* It would be nice to have a way to detect if we were passed an
446
424
     IPv4 address here.   Now we assume an IPv6 address. */
447
 
  ret = inet_pton(AF_INET6, ip, &to.in6.sin6_addr);
 
425
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
448
426
  if (ret < 0 ){
449
427
    perror("inet_pton");
450
428
    return -1;
453
431
    fprintf(stderr, "Bad address: %s\n", ip);
454
432
    return -1;
455
433
  }
456
 
  to.in6.sin6_port = htons(port);       /* Spurious warning */
 
434
  to.sin6_port = htons(port);   /* Spurious warning */
457
435
  
458
 
  to.in6.sin6_scope_id = (uint32_t)if_index;
 
436
  to.sin6_scope_id = (uint32_t)if_index;
459
437
  
460
438
  if(debug){
461
439
    fprintf(stderr, "Connection to: %s, port %d\n", ip, port);
462
440
    char addrstr[INET6_ADDRSTRLEN] = "";
463
 
    if(inet_ntop(to.in6.sin6_family, &(to.in6.sin6_addr), addrstr,
 
441
    if(inet_ntop(to.sin6_family, &(to.sin6_addr), addrstr,
464
442
                 sizeof(addrstr)) == NULL){
465
443
      perror("inet_ntop");
466
444
    } else {
470
448
    }
471
449
  }
472
450
  
473
 
  ret = connect(tcp_sd, &to.in, sizeof(to));
 
451
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
474
452
  if (ret < 0){
475
453
    perror("connect");
476
454
    return -1;
477
455
  }
478
 
 
479
 
  const char *out = mandos_protocol_version;
480
 
  written = 0;
481
 
  while (true){
482
 
    size_t out_size = strlen(out);
483
 
    ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
484
 
                                   out_size - written));
485
 
    if (ret == -1){
486
 
      perror("write");
487
 
      retval = -1;
488
 
      goto mandos_end;
489
 
    }
490
 
    written += (size_t)ret;
491
 
    if(written < out_size){
492
 
      continue;
493
 
    } else {
494
 
      if (out == mandos_protocol_version){
495
 
        written = 0;
496
 
        out = "\r\n";
497
 
      } else {
498
 
        break;
499
 
      }
500
 
    }
501
 
  }
502
 
 
 
456
  
503
457
  if(debug){
504
458
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
505
459
  }
525
479
  }
526
480
 
527
481
  while(true){
528
 
    buffer_capacity = adjustbuffer(&buffer, buffer_length,
529
 
                                   buffer_capacity);
530
 
    if (buffer_capacity == 0){
531
 
      perror("adjustbuffer");
532
 
      retval = -1;
533
 
      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;
534
490
    }
535
491
    
536
492
    ret = gnutls_record_recv(session, buffer+buffer_length,
576
532
                                               &decrypted_buffer,
577
533
                                               keydir);
578
534
    if (decrypted_buffer_size >= 0){
579
 
      written = 0;
580
535
      while(written < (size_t) decrypted_buffer_size){
581
536
        ret = (int)fwrite (decrypted_buffer + written, 1,
582
537
                           (size_t)decrypted_buffer_size - written,
733
688
    const char *interface = "eth0";
734
689
    struct ifreq network;
735
690
    int sd;
736
 
    uid_t uid;
737
 
    gid_t gid;
738
691
    char *connect_to = NULL;
739
692
    AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
740
 
    const char *pubkeyfile = "pubkey.txt";
741
 
    const char *seckeyfile = "seckey.txt";
742
693
    mandos_context mc = { .simple_poll = NULL, .server = NULL,
743
694
                          .dh_bits = 1024, .priority = "SECURE256"};
744
695
    
745
696
    {
746
 
      struct argp_option options[] = {
747
 
        { .name = "debug", .key = 128,
748
 
          .doc = "Debug mode", .group = 3 },
749
 
        { .name = "connect", .key = 'c',
750
 
          .arg = "IP",
751
 
          .doc = "Connect directly to a sepcified mandos server",
752
 
          .group = 1 },
753
 
        { .name = "interface", .key = 'i',
754
 
          .arg = "INTERFACE",
755
 
          .doc = "Interface that Avahi will conntect through",
756
 
          .group = 1 },
757
 
        { .name = "keydir", .key = 'd',
758
 
          .arg = "KEYDIR",
759
 
          .doc = "Directory where the openpgp keyring is",
760
 
          .group = 1 },
761
 
        { .name = "seckey", .key = 's',
762
 
          .arg = "SECKEY",
763
 
          .doc = "Secret openpgp key for gnutls authentication",
764
 
          .group = 1 },
765
 
        { .name = "pubkey", .key = 'p',
766
 
          .arg = "PUBKEY",
767
 
          .doc = "Public openpgp key for gnutls authentication",
768
 
          .group = 2 },
769
 
        { .name = "dh-bits", .key = 129,
770
 
          .arg = "BITS",
771
 
          .doc = "dh-bits to use in gnutls communication",
772
 
          .group = 2 },
773
 
        { .name = "priority", .key = 130,
774
 
          .arg = "PRIORITY",
775
 
          .doc = "GNUTLS priority", .group = 1 },
776
 
        { .name = NULL }
777
 
      };
778
 
 
779
 
      
780
 
      error_t parse_opt (int key, char *arg,
781
 
                         struct argp_state *state) {
782
 
        /* Get the INPUT argument from `argp_parse', which we know is
783
 
           a pointer to our plugin list pointer. */
784
 
        switch (key) {
785
 
        case 128:
786
 
          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;
787
724
          break;
788
725
        case 'c':
789
 
          connect_to = arg;
790
 
          break;
791
 
        case 'i':
792
 
          interface = arg;
 
726
          connect_to = optarg;
793
727
          break;
794
728
        case 'd':
795
 
          keydir = arg;
 
729
          keydir = optarg;
 
730
          break;
 
731
        case 'p':
 
732
          pubkeyfile = optarg;
796
733
          break;
797
734
        case 's':
798
 
          seckeyfile = arg;
799
 
          break;
800
 
        case 'p':
801
 
          pubkeyfile = arg;
802
 
          break;
803
 
        case 129:
 
735
          seckeyfile = optarg;
 
736
          break;
 
737
        case 'D':
804
738
          errno = 0;
805
 
          mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);
 
739
          mc.dh_bits = (unsigned int) strtol(optarg, NULL, 10);
806
740
          if (errno){
807
741
            perror("strtol");
808
742
            exit(EXIT_FAILURE);
809
743
          }
810
744
          break;
811
 
        case 130:
812
 
          mc.priority = arg;
813
 
          break;
814
 
        case ARGP_KEY_ARG:
815
 
          argp_usage (state);
816
 
          break;
817
 
          case ARGP_KEY_END:
818
 
            break;
 
745
        case 'P':
 
746
          mc.priority = optarg;
 
747
          break;
 
748
        case '?':
819
749
        default:
820
 
          return ARGP_ERR_UNKNOWN;
 
750
          /* getopt_long() has already printed a message about the
 
751
             unrcognized option, so just exit. */
 
752
          exit(EXIT_FAILURE);
821
753
        }
822
 
        return 0;
823
754
      }
824
 
 
825
 
      struct argp argp = { .options = options, .parser = parse_opt,
826
 
                           .args_doc = "",
827
 
                           .doc = "Mandos client -- Get and decrypt"
828
 
                           " passwords from mandos server" };
829
 
      argp_parse (&argp, argc, argv, 0, 0, NULL);
 
755
      /* Set the global debug flag from the temporary int */
 
756
      debug = debug_int ? true : false;
830
757
    }
831
 
      
 
758
    
832
759
    pubkeyfile = combinepath(keydir, pubkeyfile);
833
760
    if (pubkeyfile == NULL){
834
761
      perror("combinepath");
841
768
      perror("combinepath");
842
769
      goto end;
843
770
    }
844
 
 
845
 
    ret = init_gnutls_global(&mc, pubkeyfile, seckeyfile);
846
 
    if (ret == -1){
847
 
      fprintf(stderr, "init_gnutls_global\n");
848
 
      goto end;
849
 
    }
850
 
 
851
 
    uid = getuid();
852
 
    gid = getgid();
853
 
 
854
 
    ret = setuid(uid);
855
 
    if (ret == -1){
856
 
      perror("setuid");
857
 
    }
858
 
    
859
 
    setgid(gid);
860
 
    if (ret == -1){
861
 
      perror("setgid");
862
 
    }
863
771
    
864
772
    if_index = (AvahiIfIndex) if_nametoindex(interface);
865
773
    if(if_index == 0){
873
781
      char *address = strrchr(connect_to, ':');
874
782
      if(address == NULL){
875
783
        fprintf(stderr, "No colon in address\n");
876
 
        exitcode = EXIT_FAILURE;
877
 
        goto end;
 
784
        exit(EXIT_FAILURE);
878
785
      }
879
786
      errno = 0;
880
787
      uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
881
788
      if(errno){
882
789
        perror("Bad port number");
883
 
        exitcode = EXIT_FAILURE;
884
 
        goto end;
 
790
        exit(EXIT_FAILURE);
885
791
      }
886
792
      *address = '\0';
887
793
      address = connect_to;
888
794
      ret = start_mandos_communication(address, port, if_index, &mc);
889
795
      if(ret < 0){
890
 
        exitcode = EXIT_FAILURE;
 
796
        exit(EXIT_FAILURE);
891
797
      } else {
892
 
        exitcode = EXIT_SUCCESS;
 
798
        exit(EXIT_SUCCESS);
893
799
      }
894
 
      goto end;
895
800
    }
896
801
    
897
802
    /* If the interface is down, bring it up */