/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/mandos-client.c

  • Committer: Teddy Hogeborn
  • Date: 2009-09-17 01:21:27 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090917012127-4kwzquwuvudwucuv
* debian/control (Standards-Version): Updated to "2.8.3".

* mandos-client.conf.xml (OPTIONS/timeout): Add that a successful
                                            client request will also
                                            reset the timeout.
* mandos.xml (CHECKING): - '' -

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandos-client.c
44
44
#include <stdint.h>             /* uint16_t, uint32_t */
45
45
#include <stddef.h>             /* NULL, size_t, ssize_t */
46
46
#include <stdlib.h>             /* free(), EXIT_SUCCESS, EXIT_FAILURE,
47
 
                                   srand(), strtof(), abort() */
 
47
                                   srand(), strtof() */
48
48
#include <stdbool.h>            /* bool, false, true */
49
49
#include <string.h>             /* memset(), strcmp(), strlen(),
50
50
                                   strerror(), asprintf(), strcpy() */
72
72
                                */
73
73
#include <unistd.h>             /* close(), SEEK_SET, off_t, write(),
74
74
                                   getuid(), getgid(), seteuid(),
75
 
                                   setgid(), pause() */
 
75
                                   setgid() */
76
76
#include <arpa/inet.h>          /* inet_pton(), htons */
77
77
#include <iso646.h>             /* not, or, and */
78
78
#include <argp.h>               /* struct argp_option, error_t, struct
125
125
static const char mandos_protocol_version[] = "1";
126
126
const char *argp_program_version = "mandos-client " VERSION;
127
127
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
128
 
static const char sys_class_net[] = "/sys/class/net";
129
 
char *connect_to = NULL;
130
128
 
131
129
/* Used for passing in values through the Avahi callback functions */
132
130
typedef struct {
144
142
                      .dh_bits = 1024, .priority = "SECURE256"
145
143
                      ":!CTYPE-X.509:+CTYPE-OPENPGP" };
146
144
 
147
 
sig_atomic_t quit_now = 0;
148
 
int signal_received = 0;
149
 
 
150
145
/*
151
146
 * Make additional room in "buffer" for at least BUFFER_SIZE more
152
147
 * bytes. "buffer_capacity" is how much is currently allocated,
481
476
  /* GnuTLS session creation */
482
477
  do {
483
478
    ret = gnutls_init(session, GNUTLS_SERVER);
484
 
    if(quit_now){
485
 
      return -1;
486
 
    }
487
479
  } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
488
480
  if(ret != GNUTLS_E_SUCCESS){
489
481
    fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
494
486
    const char *err;
495
487
    do {
496
488
      ret = gnutls_priority_set_direct(*session, mc.priority, &err);
497
 
      if(quit_now){
498
 
        gnutls_deinit(*session);
499
 
        return -1;
500
 
      }
501
489
    } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
502
490
    if(ret != GNUTLS_E_SUCCESS){
503
491
      fprintf(stderr, "Syntax error at: %s\n", err);
511
499
  do {
512
500
    ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
513
501
                                 mc.cred);
514
 
    if(quit_now){
515
 
      gnutls_deinit(*session);
516
 
      return -1;
517
 
    }
518
502
  } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
519
503
  if(ret != GNUTLS_E_SUCCESS){
520
504
    fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
535
519
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
536
520
                      __attribute__((unused)) const char *txt){}
537
521
 
 
522
sig_atomic_t quit_now = 0;
 
523
int signal_received = 0;
 
524
 
538
525
/* Called when a Mandos server is found */
539
526
static int start_mandos_communication(const char *ip, uint16_t port,
540
527
                                      AvahiIfIndex if_index,
546
533
    struct sockaddr_in6 in6;
547
534
  } to;
548
535
  char *buffer = NULL;
549
 
  char *decrypted_buffer = NULL;
 
536
  char *decrypted_buffer;
550
537
  size_t buffer_length = 0;
551
538
  size_t buffer_capacity = 0;
552
539
  size_t written;
553
 
  int retval = -1;
 
540
  int retval = 0;
554
541
  gnutls_session_t session;
555
542
  int pf;                       /* Protocol family */
556
543
  
583
570
  tcp_sd = socket(pf, SOCK_STREAM, 0);
584
571
  if(tcp_sd < 0){
585
572
    perror("socket");
 
573
    retval = -1;
586
574
    goto mandos_end;
587
575
  }
588
576
  
600
588
  }
601
589
  if(ret < 0 ){
602
590
    perror("inet_pton");
 
591
    retval = -1;
603
592
    goto mandos_end;
604
593
  }
605
594
  if(ret == 0){
606
595
    fprintf(stderr, "Bad address: %s\n", ip);
 
596
    retval = -1;
607
597
    goto mandos_end;
608
598
  }
609
599
  if(af == AF_INET6){
617
607
      if(if_index == AVAHI_IF_UNSPEC){
618
608
        fprintf(stderr, "An IPv6 link-local address is incomplete"
619
609
                " without a network interface\n");
 
610
        retval = -1;
620
611
        goto mandos_end;
621
612
      }
622
613
      /* Set the network interface number as scope */
675
666
  }
676
667
  if(ret < 0){
677
668
    perror("connect");
 
669
    retval = -1;
678
670
    goto mandos_end;
679
671
  }
680
672
  
690
682
                                   out_size - written));
691
683
    if(ret == -1){
692
684
      perror("write");
 
685
      retval = -1;
693
686
      goto mandos_end;
694
687
    }
695
688
    written += (size_t)ret;
735
728
      fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
736
729
      gnutls_perror(ret);
737
730
    }
 
731
    retval = -1;
738
732
    goto mandos_end;
739
733
  }
740
734
  
755
749
                                   buffer_capacity);
756
750
    if(buffer_capacity == 0){
757
751
      perror("incbuffer");
 
752
      retval = -1;
758
753
      goto mandos_end;
759
754
    }
760
755
    
783
778
        if(ret < 0){
784
779
          fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
785
780
          gnutls_perror(ret);
 
781
          retval = -1;
786
782
          goto mandos_end;
787
783
        }
788
784
        break;
789
785
      default:
790
786
        fprintf(stderr, "Unknown error while reading data from"
791
787
                " encrypted session with Mandos server\n");
 
788
        retval = -1;
792
789
        gnutls_bye(session, GNUTLS_SHUT_RDWR);
793
790
        goto mandos_end;
794
791
      }
805
802
    goto mandos_end;
806
803
  }
807
804
  
808
 
  do {
809
 
    ret = gnutls_bye(session, GNUTLS_SHUT_RDWR);
810
 
    if(quit_now){
811
 
      goto mandos_end;
812
 
    }
813
 
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
805
  gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
806
  
 
807
  if(quit_now){
 
808
    goto mandos_end;
 
809
  }
814
810
  
815
811
  if(buffer_length > 0){
816
812
    ssize_t decrypted_buffer_size;
833
829
            fprintf(stderr, "Error writing encrypted data: %s\n",
834
830
                    strerror(errno));
835
831
          }
836
 
          goto mandos_end;
 
832
          retval = -1;
 
833
          break;
837
834
        }
838
835
        written += (size_t)ret;
839
836
      }
840
 
      retval = 0;
 
837
      free(decrypted_buffer);
 
838
    } else {
 
839
      retval = -1;
841
840
    }
 
841
  } else {
 
842
    retval = -1;
842
843
  }
843
844
  
844
845
  /* Shutdown procedure */
845
846
  
846
847
 mandos_end:
847
 
  free(decrypted_buffer);
848
848
  free(buffer);
849
849
  if(tcp_sd >= 0){
850
850
    ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
976
976
  errno = old_errno;
977
977
}
978
978
 
979
 
/* 
980
 
 * This function determines if a directory entry in /sys/class/net
981
 
 * corresponds to an acceptable network device.
982
 
 * (This function is passed to scandir(3) as a filter function.)
983
 
 */
984
 
int good_interface(const struct dirent *if_entry){
985
 
  ssize_t ssret;
986
 
  char *flagname = NULL;
987
 
  int ret = asprintf(&flagname, "%s/%s/flags", sys_class_net,
988
 
                     if_entry->d_name);
989
 
  if(ret < 0){
990
 
    perror("asprintf");
991
 
    return 0;
992
 
  }
993
 
  if(if_entry->d_name[0] == '.'){
994
 
    return 0;
995
 
  }
996
 
  int flags_fd = (int)TEMP_FAILURE_RETRY(open(flagname, O_RDONLY));
997
 
  if(flags_fd == -1){
998
 
    perror("open");
999
 
    return 0;
1000
 
  }
1001
 
  typedef short ifreq_flags;    /* ifreq.ifr_flags in netdevice(7) */
1002
 
  /* read line from flags_fd */
1003
 
  ssize_t to_read = (sizeof(ifreq_flags)*2)+3; /* "0x1003\n" */
1004
 
  char *flagstring = malloc((size_t)to_read);
1005
 
  if(flagstring == NULL){
1006
 
    perror("malloc");
1007
 
    close(flags_fd);
1008
 
    return 0;
1009
 
  }
1010
 
  while(to_read > 0){
1011
 
    ssret = (ssize_t)TEMP_FAILURE_RETRY(read(flags_fd, flagstring,
1012
 
                                             (size_t)to_read));
1013
 
    if(ssret == -1){
1014
 
      perror("read");
1015
 
      free(flagstring);
1016
 
      close(flags_fd);
1017
 
      return 0;
1018
 
    }
1019
 
    to_read -= ssret;
1020
 
    if(ssret == 0){
1021
 
      break;
1022
 
    }
1023
 
  }
1024
 
  close(flags_fd);
1025
 
  intmax_t tmpmax;
1026
 
  char *tmp;
1027
 
  errno = 0;
1028
 
  tmpmax = strtoimax(flagstring, &tmp, 0);
1029
 
  if(errno != 0 or tmp == flagstring or (*tmp != '\0'
1030
 
                                         and not (isspace(*tmp)))
1031
 
     or tmpmax != (ifreq_flags)tmpmax){
1032
 
    free(flagstring);
1033
 
    return 0;
1034
 
  }
1035
 
  free(flagstring);
1036
 
  ifreq_flags flags = (ifreq_flags)tmpmax;
1037
 
  /* Reject the loopback device */
1038
 
  if(flags & IFF_LOOPBACK){
1039
 
    return 0;
1040
 
  }
1041
 
  /* Accept point-to-point devices only if connect_to is specified */
1042
 
  if(connect_to != NULL and (flags & IFF_POINTOPOINT)){
1043
 
    return 1;
1044
 
  }
1045
 
  /* Otherwise, reject non-broadcast-capable devices */
1046
 
  if(not (flags & IFF_BROADCAST)){
1047
 
    return 0;
1048
 
  }
1049
 
  /* Accept this device */
1050
 
  return 1;
1051
 
}
1052
 
 
1053
979
int main(int argc, char *argv[]){
1054
980
  AvahiSServiceBrowser *sb = NULL;
1055
981
  int error;
1057
983
  intmax_t tmpmax;
1058
984
  char *tmp;
1059
985
  int exitcode = EXIT_SUCCESS;
1060
 
  const char *interface = "";
 
986
  const char *interface = "eth0";
1061
987
  struct ifreq network;
1062
988
  int sd = -1;
1063
989
  bool take_down_interface = false;
1064
990
  uid_t uid;
1065
991
  gid_t gid;
 
992
  char *connect_to = NULL;
1066
993
  char tempdir[] = "/tmp/mandosXXXXXX";
1067
994
  bool tempdir_created = false;
1068
995
  AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
1198
1125
  if(not debug){
1199
1126
    avahi_set_log_function(empty_log);
1200
1127
  }
1201
 
 
1202
 
  if(interface[0] == '\0'){
1203
 
    struct dirent **direntries;
1204
 
    ret = scandir(sys_class_net, &direntries, good_interface,
1205
 
                  alphasort);
1206
 
    if(ret >= 1){
1207
 
      /* Pick the first good interface */
1208
 
      interface = strdup(direntries[0]->d_name);
1209
 
      if(interface == NULL){
1210
 
        perror("malloc");
1211
 
        free(direntries);
1212
 
        exitcode = EXIT_FAILURE;
1213
 
        goto end;
1214
 
      }
1215
 
      free(direntries);
1216
 
    } else {
1217
 
      free(direntries);
1218
 
      fprintf(stderr, "Could not find a network interface\n");
1219
 
      exitcode = EXIT_FAILURE;
1220
 
      goto end;
1221
 
    }
1222
 
  }
1223
1128
  
1224
1129
  /* Initialize Avahi early so avahi_simple_poll_quit() can be called
1225
1130
     from the signal handler */
1296
1201
  }
1297
1202
  
1298
1203
  /* If the interface is down, bring it up */
1299
 
  if(strcmp(interface, "none") != 0){
 
1204
  if(interface[0] != '\0'){
1300
1205
    if_index = (AvahiIfIndex) if_nametoindex(interface);
1301
1206
    if(if_index == 0){
1302
1207
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
1693
1598
  if(quit_now){
1694
1599
    sigemptyset(&old_sigterm_action.sa_mask);
1695
1600
    old_sigterm_action.sa_handler = SIG_DFL;
1696
 
    ret = (int)TEMP_FAILURE_RETRY(sigaction(signal_received,
1697
 
                                            &old_sigterm_action,
1698
 
                                            NULL));
 
1601
    ret = sigaction(signal_received, &old_sigterm_action, NULL);
1699
1602
    if(ret == -1){
1700
1603
      perror("sigaction");
1701
1604
    }
1702
 
    do {
1703
 
      ret = raise(signal_received);
1704
 
    } while(ret != 0 and errno == EINTR);
1705
 
    if(ret != 0){
1706
 
      perror("raise");
1707
 
      abort();
1708
 
    }
1709
 
    TEMP_FAILURE_RETRY(pause());
 
1605
    raise(signal_received);
1710
1606
  }
1711
1607
  
1712
1608
  return exitcode;