/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-01-31 10:33:17 UTC
  • mfrom: (24.1.129 mandos)
  • Revision ID: teddy@fukt.bsnet.se-20090131103317-wzqvyr532sjcjt7u
Merge from Björn:

* mandos-ctl: New option "--remove-client".  Only default to listing
              clients if no clients were given on the command line.
* plugins.d/mandos-client.c: Lower kernel log level while bringing up
                             network interface.  New option "--delay"
                             to control the maximum delay to wait for
                             running interface.
* plugins.d/mandos-client.xml (SYNOPSIS, OPTIONS): New option
                                                   "--delay".

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandos-client.c
36
36
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
37
37
 
38
38
#include <stdio.h>              /* fprintf(), stderr, fwrite(),
39
 
                                   stdout, ferror(), sscanf */
 
39
                                   stdout, ferror(), sscanf(),
 
40
                                   remove() */
40
41
#include <stdint.h>             /* uint16_t, uint32_t */
41
42
#include <stddef.h>             /* NULL, size_t, ssize_t */
42
43
#include <stdlib.h>             /* free(), EXIT_SUCCESS, EXIT_FAILURE,
60
61
#include <inttypes.h>           /* PRIu16, intmax_t, SCNdMAX */
61
62
#include <assert.h>             /* assert() */
62
63
#include <errno.h>              /* perror(), errno */
63
 
#include <time.h>               /* time() */
 
64
#include <time.h>               /* nanosleep(), time() */
64
65
#include <net/if.h>             /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
65
66
                                   SIOCSIFFLAGS, if_indextoname(),
66
67
                                   if_nametoindex(), IF_NAMESIZE */
74
75
                                   argp_state, struct argp,
75
76
                                   argp_parse(), ARGP_KEY_ARG,
76
77
                                   ARGP_KEY_END, ARGP_ERR_UNKNOWN */
 
78
#include <sys/klog.h>           /* klogctl() */
77
79
 
78
80
/* Avahi */
79
81
/* All Avahi types, constants and functions
153
155
  
154
156
  
155
157
  /*
156
 
   * Helper function to insert pub and seckey to the enigne keyring.
 
158
   * Helper function to insert pub and seckey to the engine keyring.
157
159
   */
158
160
  bool import_key(const char *filename){
159
161
    int fd;
831
833
    gid_t gid;
832
834
    char *connect_to = NULL;
833
835
    char tempdir[] = "/tmp/mandosXXXXXX";
 
836
    bool tempdir_created = false;
834
837
    AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
835
838
    const char *seckey = PATHDIR "/" SECKEY;
836
839
    const char *pubkey = PATHDIR "/" PUBKEY;
838
841
    mandos_context mc = { .simple_poll = NULL, .server = NULL,
839
842
                          .dh_bits = 1024, .priority = "SECURE256"
840
843
                          ":!CTYPE-X.509:+CTYPE-OPENPGP" };
841
 
    bool gnutls_initalized = false;
842
 
    bool gpgme_initalized = false;
 
844
    bool gnutls_initialized = false;
 
845
    bool gpgme_initialized = false;
 
846
    double delay = 2.5;
843
847
    
844
848
    {
845
849
      struct argp_option options[] = {
871
875
          .arg = "STRING",
872
876
          .doc = "GnuTLS priority string for the TLS handshake",
873
877
          .group = 1 },
 
878
        { .name = "delay", .key = 131,
 
879
          .arg = "SECONDS",
 
880
          .doc = "Maximum delay to wait for interface startup",
 
881
          .group = 2 },
874
882
        { .name = NULL }
875
883
      };
876
884
      
904
912
        case 130:               /* --priority */
905
913
          mc.priority = arg;
906
914
          break;
 
915
        case 131:               /* --delay */
 
916
          ret = sscanf(arg, "%lf%n", &delay, &numchars);
 
917
          if(ret < 1 or arg[numchars] != '\0'){
 
918
            fprintf(stderr, "Bad delay\n");
 
919
            exit(EXIT_FAILURE);
 
920
          }
 
921
          break;
907
922
        case ARGP_KEY_ARG:
908
923
          argp_usage(state);
909
924
        case ARGP_KEY_END:
928
943
    
929
944
    /* If the interface is down, bring it up */
930
945
    {
 
946
      /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
 
947
         messages to mess up the prompt */
 
948
      ret = klogctl(8, NULL, 5);
 
949
      if(ret == -1){
 
950
        perror("klogctl");
 
951
      }
 
952
      
931
953
      sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
932
954
      if(sd < 0) {
933
955
        perror("socket");
934
956
        exitcode = EXIT_FAILURE;
 
957
        ret = klogctl(7, NULL, 0);
 
958
        if(ret == -1){
 
959
          perror("klogctl");
 
960
        }
935
961
        goto end;
936
962
      }
937
963
      strcpy(network.ifr_name, interface);
938
964
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
939
965
      if(ret == -1){
940
966
        perror("ioctl SIOCGIFFLAGS");
 
967
        ret = klogctl(7, NULL, 0);
 
968
        if(ret == -1){
 
969
          perror("klogctl");
 
970
        }
941
971
        exitcode = EXIT_FAILURE;
942
972
        goto end;
943
973
      }
947
977
        if(ret == -1){
948
978
          perror("ioctl SIOCSIFFLAGS");
949
979
          exitcode = EXIT_FAILURE;
 
980
          ret = klogctl(7, NULL, 0);
 
981
          if(ret == -1){
 
982
            perror("klogctl");
 
983
          }
950
984
          goto end;
951
985
        }
952
986
      }
 
987
      /* sleep checking until interface is running */
 
988
      for(int i=0; i < delay * 4; i++){
 
989
        ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
990
        if(ret == -1){
 
991
          perror("ioctl SIOCGIFFLAGS");
 
992
        } else if(network.ifr_flags & IFF_RUNNING){
 
993
          break;
 
994
        }
 
995
        struct timespec sleeptime = { .tv_nsec = 250000000 };
 
996
        nanosleep(&sleeptime, NULL);
 
997
      }
953
998
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
954
999
      if(ret == -1){
955
1000
        perror("close");
956
1001
      }
 
1002
      /* Restores kernel loglevel to default */
 
1003
      ret = klogctl(7, NULL, 0);
 
1004
      if(ret == -1){
 
1005
        perror("klogctl");
 
1006
      }
957
1007
    }
958
1008
    
959
1009
    uid = getuid();
960
1010
    gid = getgid();
961
1011
    
 
1012
    setgid(gid);
 
1013
    if(ret == -1){
 
1014
      perror("setgid");
 
1015
    }
 
1016
    
962
1017
    ret = setuid(uid);
963
1018
    if(ret == -1){
964
1019
      perror("setuid");
965
1020
    }
966
1021
    
967
 
    setgid(gid);
968
 
    if(ret == -1){
969
 
      perror("setgid");
970
 
    }
971
 
    
972
1022
    ret = init_gnutls_global(&mc, pubkey, seckey);
973
1023
    if(ret == -1){
974
1024
      fprintf(stderr, "init_gnutls_global failed\n");
975
1025
      exitcode = EXIT_FAILURE;
976
1026
      goto end;
977
1027
    } else {
978
 
      gnutls_initalized = true;
 
1028
      gnutls_initialized = true;
979
1029
    }
980
1030
    
981
1031
    if(mkdtemp(tempdir) == NULL){
982
1032
      perror("mkdtemp");
983
 
      tempdir[0] = '\0';
984
1033
      goto end;
985
1034
    }
 
1035
    tempdir_created = true;
986
1036
    
987
1037
    if(not init_gpgme(&mc, pubkey, seckey, tempdir)){
988
 
      fprintf(stderr, "gpgme_initalized failed\n");
 
1038
      fprintf(stderr, "init_gpgme failed\n");
989
1039
      exitcode = EXIT_FAILURE;
990
1040
      goto end;
991
1041
    } else {
992
 
      gpgme_initalized = true;
 
1042
      gpgme_initialized = true;
993
1043
    }
994
1044
    
995
1045
    if_index = (AvahiIfIndex) if_nametoindex(interface);
996
1046
    if(if_index == 0){
997
1047
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
998
 
      exit(EXIT_FAILURE);
 
1048
      exitcode = EXIT_FAILURE;
 
1049
      goto end;
999
1050
    }
1000
1051
    
1001
1052
    if(connect_to != NULL){
1086
1137
    if(debug){
1087
1138
      fprintf(stderr, "Starting Avahi loop search\n");
1088
1139
    }
1089
 
    
 
1140
 
1090
1141
    avahi_simple_poll_loop(mc.simple_poll);
1091
1142
    
1092
1143
 end:
1105
1156
    if(mc.simple_poll != NULL)
1106
1157
        avahi_simple_poll_free(mc.simple_poll);
1107
1158
    
1108
 
    if(gnutls_initalized){
 
1159
    if(gnutls_initialized){
1109
1160
      gnutls_certificate_free_credentials(mc.cred);
1110
1161
      gnutls_global_deinit();
1111
1162
      gnutls_dh_params_deinit(mc.dh_params);
1112
1163
    }
1113
1164
    
1114
 
    if(gpgme_initalized){
 
1165
    if(gpgme_initialized){
1115
1166
      gpgme_release(mc.ctx);
1116
1167
    }
1117
1168
    
1118
1169
    /* Removes the temp directory used by GPGME */
1119
 
    if(tempdir[0] != '\0'){
 
1170
    if(tempdir_created){
1120
1171
      DIR *d;
1121
1172
      struct dirent *direntry;
1122
1173
      d = opendir(tempdir);
1130
1181
          if(direntry == NULL){
1131
1182
            break;
1132
1183
          }
1133
 
          if(direntry->d_type == DT_REG){
1134
 
            char *fullname = NULL;
1135
 
            ret = asprintf(&fullname, "%s/%s", tempdir,
1136
 
                           direntry->d_name);
1137
 
            if(ret < 0){
1138
 
              perror("asprintf");
1139
 
              continue;
1140
 
            }
1141
 
            ret = unlink(fullname);
1142
 
            if(ret == -1){
1143
 
              fprintf(stderr, "unlink(\"%s\"): %s",
1144
 
                      fullname, strerror(errno));
1145
 
            }
1146
 
            free(fullname);
1147
 
          }
 
1184
          /* Skip "." and ".." */
 
1185
          if(direntry->d_name[0] == '.'
 
1186
             and (direntry->d_name[1] == '\0'
 
1187
                  or (direntry->d_name[1] == '.'
 
1188
                      and direntry->d_name[2] == '\0'))){
 
1189
            continue;
 
1190
          }
 
1191
          char *fullname = NULL;
 
1192
          ret = asprintf(&fullname, "%s/%s", tempdir,
 
1193
                         direntry->d_name);
 
1194
          if(ret < 0){
 
1195
            perror("asprintf");
 
1196
            continue;
 
1197
          }
 
1198
          ret = remove(fullname);
 
1199
          if(ret == -1){
 
1200
            fprintf(stderr, "remove(\"%s\"): %s\n", fullname,
 
1201
                    strerror(errno));
 
1202
          }
 
1203
          free(fullname);
1148
1204
        }
1149
1205
        closedir(d);
1150
1206
      }
1153
1209
        perror("rmdir");
1154
1210
      }
1155
1211
    }
1156
 
          
 
1212
    
1157
1213
    return exitcode;
1158
1214
}