/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-05 01:05:25 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090905010525-7qeq05ztwc2ddhuv
* plugins.d/mandos-client.c (main): Do not handle ignored signals.
                                    Bug fix: remove signal handler
                                    before re-raising signal.

* plugins.d/password-prompt.c (main): Check return values from
                                      "sigaddset()".

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandos-client.c
30
30
 */
31
31
 
32
32
/* Needed by GPGME, specifically gpgme_data_seek() */
 
33
#ifndef _LARGEFILE_SOURCE
33
34
#define _LARGEFILE_SOURCE
 
35
#endif
 
36
#ifndef _FILE_OFFSET_BITS
34
37
#define _FILE_OFFSET_BITS 64
 
38
#endif
35
39
 
36
40
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
37
41
 
76
80
                                   argp_parse(), ARGP_KEY_ARG,
77
81
                                   ARGP_KEY_END, ARGP_ERR_UNKNOWN */
78
82
#include <signal.h>             /* sigemptyset(), sigaddset(),
79
 
                                   sigaction(), SIGTERM, sigaction,
80
 
                                   sig_atomic_t */
 
83
                                   sigaction(), SIGTERM, sig_atomic_t,
 
84
                                   raise() */
81
85
 
82
86
#ifdef __linux__
83
87
#include <sys/klog.h>           /* klogctl() */
245
249
    return false;
246
250
  }
247
251
  
248
 
  return true; 
 
252
  return true;
249
253
}
250
254
 
251
255
/* 
305
309
        }
306
310
        gpgme_recipient_t recipient;
307
311
        recipient = result->recipients;
308
 
        if(recipient){
309
 
          while(recipient != NULL){
310
 
            fprintf(stderr, "Public key algorithm: %s\n",
311
 
                    gpgme_pubkey_algo_name(recipient->pubkey_algo));
312
 
            fprintf(stderr, "Key ID: %s\n", recipient->keyid);
313
 
            fprintf(stderr, "Secret key available: %s\n",
314
 
                    recipient->status == GPG_ERR_NO_SECKEY
315
 
                    ? "No" : "Yes");
316
 
            recipient = recipient->next;
317
 
          }
 
312
        while(recipient != NULL){
 
313
          fprintf(stderr, "Public key algorithm: %s\n",
 
314
                  gpgme_pubkey_algo_name(recipient->pubkey_algo));
 
315
          fprintf(stderr, "Key ID: %s\n", recipient->keyid);
 
316
          fprintf(stderr, "Secret key available: %s\n",
 
317
                  recipient->status == GPG_ERR_NO_SECKEY
 
318
                  ? "No" : "Yes");
 
319
          recipient = recipient->next;
318
320
        }
319
321
      }
320
322
    }
512
514
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
513
515
                      __attribute__((unused)) const char *txt){}
514
516
 
 
517
sig_atomic_t quit_now = 0;
 
518
int signal_received = 0;
 
519
 
515
520
/* Called when a Mandos server is found */
516
521
static int start_mandos_communication(const char *ip, uint16_t port,
517
522
                                      AvahiIfIndex if_index,
562
567
  
563
568
  memset(&to, 0, sizeof(to));
564
569
  if(af == AF_INET6){
565
 
    to.in6.sin6_family = (uint16_t)af;
 
570
    to.in6.sin6_family = (sa_family_t)af;
566
571
    ret = inet_pton(af, ip, &to.in6.sin6_addr);
567
572
  } else {                      /* IPv4 */
568
573
    to.in.sin_family = (sa_family_t)af;
797
802
  /* Called whenever a service has been resolved successfully or
798
803
     timed out */
799
804
  
 
805
  if(quit_now){
 
806
    return;
 
807
  }
 
808
  
800
809
  switch(event){
801
810
  default:
802
811
  case AVAHI_RESOLVER_FAILURE:
854
863
       the callback function is called the Avahi server will free the
855
864
       resolver for us. */
856
865
    
857
 
    if(!(avahi_s_service_resolver_new(mc.server, interface,
858
 
                                       protocol, name, type, domain,
859
 
                                       AVAHI_PROTO_INET6, 0,
860
 
                                       resolve_callback, NULL)))
 
866
    if(avahi_s_service_resolver_new(mc.server, interface, protocol,
 
867
                                    name, type, domain, protocol, 0,
 
868
                                    resolve_callback, NULL) == NULL)
861
869
      fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
862
870
              name, avahi_strerror(avahi_server_errno(mc.server)));
863
871
    break;
874
882
  }
875
883
}
876
884
 
877
 
sig_atomic_t quit_now = 0;
878
 
 
879
885
/* stop main loop after sigterm has been called */
880
 
static void handle_sigterm(__attribute__((unused)) int sig){
 
886
static void handle_sigterm(int sig){
881
887
  if(quit_now){
882
888
    return;
883
889
  }
884
890
  quit_now = 1;
 
891
  signal_received = sig;
885
892
  int old_errno = errno;
886
893
  if(mc.simple_poll != NULL){
887
894
    avahi_simple_poll_quit(mc.simple_poll);
898
905
  int exitcode = EXIT_SUCCESS;
899
906
  const char *interface = "eth0";
900
907
  struct ifreq network;
901
 
  int sd;
 
908
  int sd = -1;
 
909
  bool take_down_interface = false;
902
910
  uid_t uid;
903
911
  gid_t gid;
904
912
  char *connect_to = NULL;
908
916
  const char *seckey = PATHDIR "/" SECKEY;
909
917
  const char *pubkey = PATHDIR "/" PUBKEY;
910
918
  
911
 
  /* Initialize Mandos context */
912
 
  mc = (mandos_context){ .simple_poll = NULL, .server = NULL,
913
 
                         .dh_bits = 1024, .priority = "SECURE256"
914
 
                         ":!CTYPE-X.509:+CTYPE-OPENPGP" };
915
919
  bool gnutls_initialized = false;
916
920
  bool gpgme_initialized = false;
917
921
  float delay = 2.5f;
1051
1055
    exitcode = EXIT_FAILURE;
1052
1056
    goto end;
1053
1057
  }
1054
 
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1055
 
  if(ret == -1){
1056
 
    perror("sigaction");
1057
 
    exitcode = EXIT_FAILURE;
1058
 
    goto end;
1059
 
  }  
 
1058
  /* Need to check if the handler is SIG_IGN before handling:
 
1059
     | [[info:libc:Initial Signal Actions]] |
 
1060
     | [[info:libc:Basic Signal Handling]]  |
 
1061
  */
 
1062
  ret = sigaction(SIGINT, NULL, &old_sigterm_action);
 
1063
  if(ret == -1){
 
1064
    perror("sigaction");
 
1065
    return EXIT_FAILURE;
 
1066
  }
 
1067
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1068
    ret = sigaction(SIGINT, &sigterm_action, NULL);
 
1069
    if(ret == -1){
 
1070
      perror("sigaction");
 
1071
      exitcode = EXIT_FAILURE;
 
1072
      goto end;
 
1073
    }
 
1074
  }
 
1075
  ret = sigaction(SIGHUP, NULL, &old_sigterm_action);
 
1076
  if(ret == -1){
 
1077
    perror("sigaction");
 
1078
    return EXIT_FAILURE;
 
1079
  }
 
1080
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1081
    ret = sigaction(SIGHUP, &sigterm_action, NULL);
 
1082
    if(ret == -1){
 
1083
      perror("sigaction");
 
1084
      exitcode = EXIT_FAILURE;
 
1085
      goto end;
 
1086
    }
 
1087
  }
 
1088
  ret = sigaction(SIGTERM, NULL, &old_sigterm_action);
 
1089
  if(ret == -1){
 
1090
    perror("sigaction");
 
1091
    return EXIT_FAILURE;
 
1092
  }
 
1093
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1094
    ret = sigaction(SIGTERM, &sigterm_action, NULL);
 
1095
    if(ret == -1){
 
1096
      perror("sigaction");
 
1097
      exitcode = EXIT_FAILURE;
 
1098
      goto end;
 
1099
    }
 
1100
  }
1060
1101
  
1061
1102
  /* If the interface is down, bring it up */
1062
1103
  if(interface[0] != '\0'){
 
1104
    if_index = (AvahiIfIndex) if_nametoindex(interface);
 
1105
    if(if_index == 0){
 
1106
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
 
1107
      exitcode = EXIT_FAILURE;
 
1108
      goto end;
 
1109
    }
 
1110
    
 
1111
    if(quit_now){
 
1112
      goto end;
 
1113
    }
 
1114
    
1063
1115
#ifdef __linux__
1064
1116
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
1065
1117
       messages to mess up the prompt */
1102
1154
    }
1103
1155
    if((network.ifr_flags & IFF_UP) == 0){
1104
1156
      network.ifr_flags |= IFF_UP;
 
1157
      take_down_interface = true;
1105
1158
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
1106
1159
      if(ret == -1){
 
1160
        take_down_interface = false;
1107
1161
        perror("ioctl SIOCSIFFLAGS");
1108
1162
        exitcode = EXIT_FAILURE;
1109
1163
#ifdef __linux__
1131
1185
        perror("nanosleep");
1132
1186
      }
1133
1187
    }
1134
 
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
1135
 
    if(ret == -1){
1136
 
      perror("close");
 
1188
    if(not take_down_interface){
 
1189
      /* We won't need the socket anymore */
 
1190
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1191
      if(ret == -1){
 
1192
        perror("close");
 
1193
      }
1137
1194
    }
1138
1195
#ifdef __linux__
1139
1196
    if(restore_loglevel){
1146
1203
#endif  /* __linux__ */
1147
1204
  }
1148
1205
  
 
1206
  if(quit_now){
 
1207
    goto end;
 
1208
  }
 
1209
  
1149
1210
  uid = getuid();
1150
1211
  gid = getgid();
1151
1212
  
1160
1221
    perror("setuid");
1161
1222
  }
1162
1223
  
 
1224
  if(quit_now){
 
1225
    goto end;
 
1226
  }
 
1227
  
1163
1228
  ret = init_gnutls_global(pubkey, seckey);
1164
1229
  if(ret == -1){
1165
1230
    fprintf(stderr, "init_gnutls_global failed\n");
1169
1234
    gnutls_initialized = true;
1170
1235
  }
1171
1236
  
 
1237
  if(quit_now){
 
1238
    goto end;
 
1239
  }
 
1240
  
 
1241
  tempdir_created = true;
1172
1242
  if(mkdtemp(tempdir) == NULL){
 
1243
    tempdir_created = false;
1173
1244
    perror("mkdtemp");
1174
1245
    goto end;
1175
1246
  }
1176
 
  tempdir_created = true;
 
1247
  
 
1248
  if(quit_now){
 
1249
    goto end;
 
1250
  }
1177
1251
  
1178
1252
  if(not init_gpgme(pubkey, seckey, tempdir)){
1179
1253
    fprintf(stderr, "init_gpgme failed\n");
1183
1257
    gpgme_initialized = true;
1184
1258
  }
1185
1259
  
1186
 
  if(interface[0] != '\0'){
1187
 
    if_index = (AvahiIfIndex) if_nametoindex(interface);
1188
 
    if(if_index == 0){
1189
 
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
1190
 
      exitcode = EXIT_FAILURE;
1191
 
      goto end;
1192
 
    }
 
1260
  if(quit_now){
 
1261
    goto end;
1193
1262
  }
1194
1263
  
1195
1264
  if(connect_to != NULL){
1201
1270
      exitcode = EXIT_FAILURE;
1202
1271
      goto end;
1203
1272
    }
 
1273
    
 
1274
    if(quit_now){
 
1275
      goto end;
 
1276
    }
 
1277
    
1204
1278
    uint16_t port;
1205
1279
    errno = 0;
1206
1280
    tmpmax = strtoimax(address+1, &tmp, 10);
1210
1284
      exitcode = EXIT_FAILURE;
1211
1285
      goto end;
1212
1286
    }
 
1287
  
 
1288
    if(quit_now){
 
1289
      goto end;
 
1290
    }
 
1291
    
1213
1292
    port = (uint16_t)tmpmax;
1214
1293
    *address = '\0';
1215
1294
    address = connect_to;
1220
1299
    } else {
1221
1300
      af = AF_INET;
1222
1301
    }
 
1302
    
 
1303
    if(quit_now){
 
1304
      goto end;
 
1305
    }
 
1306
    
1223
1307
    ret = start_mandos_communication(address, port, if_index, af);
1224
1308
    if(ret < 0){
1225
1309
      exitcode = EXIT_FAILURE;
1228
1312
    }
1229
1313
    goto end;
1230
1314
  }
1231
 
    
 
1315
  
 
1316
  if(quit_now){
 
1317
    goto end;
 
1318
  }
 
1319
  
1232
1320
  {
1233
1321
    AvahiServerConfig config;
1234
1322
    /* Do not publish any local Zeroconf records */
1255
1343
    goto end;
1256
1344
  }
1257
1345
  
 
1346
  if(quit_now){
 
1347
    goto end;
 
1348
  }
 
1349
  
1258
1350
  /* Create the Avahi service browser */
1259
1351
  sb = avahi_s_service_browser_new(mc.server, if_index,
1260
 
                                   AVAHI_PROTO_INET6, "_mandos._tcp",
 
1352
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
1261
1353
                                   NULL, 0, browse_callback, NULL);
1262
1354
  if(sb == NULL){
1263
1355
    fprintf(stderr, "Failed to create service browser: %s\n",
1266
1358
    goto end;
1267
1359
  }
1268
1360
  
 
1361
  if(quit_now){
 
1362
    goto end;
 
1363
  }
 
1364
  
1269
1365
  /* Run the main loop */
1270
1366
  
1271
1367
  if(debug){
1300
1396
    gpgme_release(mc.ctx);
1301
1397
  }
1302
1398
  
 
1399
  /* Take down the network interface */
 
1400
  if(take_down_interface){
 
1401
    ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1402
    if(ret == -1){
 
1403
      perror("ioctl SIOCGIFFLAGS");
 
1404
    } else if(network.ifr_flags & IFF_UP) {
 
1405
      network.ifr_flags &= ~IFF_UP; /* clear flag */
 
1406
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
1407
      if(ret == -1){
 
1408
        perror("ioctl SIOCSIFFLAGS");
 
1409
      }
 
1410
    }
 
1411
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1412
    if(ret == -1){
 
1413
      perror("close");
 
1414
    }
 
1415
  }
 
1416
  
1303
1417
  /* Removes the temp directory used by GPGME */
1304
1418
  if(tempdir_created){
1305
1419
    DIR *d;
1344
1458
    }
1345
1459
  }
1346
1460
  
 
1461
  if(quit_now){
 
1462
    sigemptyset(&old_sigterm_action.sa_mask);
 
1463
    old_sigterm_action.sa_handler = SIG_DFL;
 
1464
    ret = sigaction(signal_received, &old_sigterm_action, NULL);
 
1465
    if(ret == -1){
 
1466
      perror("sigaction");
 
1467
    }
 
1468
    raise(signal_received);
 
1469
  }
 
1470
  
1347
1471
  return exitcode;
1348
1472
}