/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
80
80
                                   argp_parse(), ARGP_KEY_ARG,
81
81
                                   ARGP_KEY_END, ARGP_ERR_UNKNOWN */
82
82
#include <signal.h>             /* sigemptyset(), sigaddset(),
83
 
                                   sigaction(), SIGTERM, sigaction,
84
 
                                   sig_atomic_t */
 
83
                                   sigaction(), SIGTERM, sig_atomic_t,
 
84
                                   raise() */
85
85
 
86
86
#ifdef __linux__
87
87
#include <sys/klog.h>           /* klogctl() */
249
249
    return false;
250
250
  }
251
251
  
252
 
  return true; 
 
252
  return true;
253
253
}
254
254
 
255
255
/* 
309
309
        }
310
310
        gpgme_recipient_t recipient;
311
311
        recipient = result->recipients;
312
 
        if(recipient){
313
 
          while(recipient != NULL){
314
 
            fprintf(stderr, "Public key algorithm: %s\n",
315
 
                    gpgme_pubkey_algo_name(recipient->pubkey_algo));
316
 
            fprintf(stderr, "Key ID: %s\n", recipient->keyid);
317
 
            fprintf(stderr, "Secret key available: %s\n",
318
 
                    recipient->status == GPG_ERR_NO_SECKEY
319
 
                    ? "No" : "Yes");
320
 
            recipient = recipient->next;
321
 
          }
 
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;
322
320
        }
323
321
      }
324
322
    }
516
514
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
517
515
                      __attribute__((unused)) const char *txt){}
518
516
 
 
517
sig_atomic_t quit_now = 0;
 
518
int signal_received = 0;
 
519
 
519
520
/* Called when a Mandos server is found */
520
521
static int start_mandos_communication(const char *ip, uint16_t port,
521
522
                                      AvahiIfIndex if_index,
566
567
  
567
568
  memset(&to, 0, sizeof(to));
568
569
  if(af == AF_INET6){
569
 
    to.in6.sin6_family = (uint16_t)af;
 
570
    to.in6.sin6_family = (sa_family_t)af;
570
571
    ret = inet_pton(af, ip, &to.in6.sin6_addr);
571
572
  } else {                      /* IPv4 */
572
573
    to.in.sin_family = (sa_family_t)af;
801
802
  /* Called whenever a service has been resolved successfully or
802
803
     timed out */
803
804
  
 
805
  if(quit_now){
 
806
    return;
 
807
  }
 
808
  
804
809
  switch(event){
805
810
  default:
806
811
  case AVAHI_RESOLVER_FAILURE:
877
882
  }
878
883
}
879
884
 
880
 
sig_atomic_t quit_now = 0;
881
 
 
882
885
/* stop main loop after sigterm has been called */
883
 
static void handle_sigterm(__attribute__((unused)) int sig){
 
886
static void handle_sigterm(int sig){
884
887
  if(quit_now){
885
888
    return;
886
889
  }
887
890
  quit_now = 1;
 
891
  signal_received = sig;
888
892
  int old_errno = errno;
889
893
  if(mc.simple_poll != NULL){
890
894
    avahi_simple_poll_quit(mc.simple_poll);
901
905
  int exitcode = EXIT_SUCCESS;
902
906
  const char *interface = "eth0";
903
907
  struct ifreq network;
904
 
  int sd;
 
908
  int sd = -1;
 
909
  bool take_down_interface = false;
905
910
  uid_t uid;
906
911
  gid_t gid;
907
912
  char *connect_to = NULL;
1050
1055
    exitcode = EXIT_FAILURE;
1051
1056
    goto end;
1052
1057
  }
1053
 
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1054
 
  if(ret == -1){
1055
 
    perror("sigaction");
1056
 
    exitcode = EXIT_FAILURE;
1057
 
    goto end;
1058
 
  }  
 
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
  }
1059
1101
  
1060
1102
  /* If the interface is down, bring it up */
1061
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
    
1062
1115
#ifdef __linux__
1063
1116
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
1064
1117
       messages to mess up the prompt */
1101
1154
    }
1102
1155
    if((network.ifr_flags & IFF_UP) == 0){
1103
1156
      network.ifr_flags |= IFF_UP;
 
1157
      take_down_interface = true;
1104
1158
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
1105
1159
      if(ret == -1){
 
1160
        take_down_interface = false;
1106
1161
        perror("ioctl SIOCSIFFLAGS");
1107
1162
        exitcode = EXIT_FAILURE;
1108
1163
#ifdef __linux__
1130
1185
        perror("nanosleep");
1131
1186
      }
1132
1187
    }
1133
 
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
1134
 
    if(ret == -1){
1135
 
      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
      }
1136
1194
    }
1137
1195
#ifdef __linux__
1138
1196
    if(restore_loglevel){
1145
1203
#endif  /* __linux__ */
1146
1204
  }
1147
1205
  
 
1206
  if(quit_now){
 
1207
    goto end;
 
1208
  }
 
1209
  
1148
1210
  uid = getuid();
1149
1211
  gid = getgid();
1150
1212
  
1159
1221
    perror("setuid");
1160
1222
  }
1161
1223
  
 
1224
  if(quit_now){
 
1225
    goto end;
 
1226
  }
 
1227
  
1162
1228
  ret = init_gnutls_global(pubkey, seckey);
1163
1229
  if(ret == -1){
1164
1230
    fprintf(stderr, "init_gnutls_global failed\n");
1168
1234
    gnutls_initialized = true;
1169
1235
  }
1170
1236
  
 
1237
  if(quit_now){
 
1238
    goto end;
 
1239
  }
 
1240
  
 
1241
  tempdir_created = true;
1171
1242
  if(mkdtemp(tempdir) == NULL){
 
1243
    tempdir_created = false;
1172
1244
    perror("mkdtemp");
1173
1245
    goto end;
1174
1246
  }
1175
 
  tempdir_created = true;
 
1247
  
 
1248
  if(quit_now){
 
1249
    goto end;
 
1250
  }
1176
1251
  
1177
1252
  if(not init_gpgme(pubkey, seckey, tempdir)){
1178
1253
    fprintf(stderr, "init_gpgme failed\n");
1182
1257
    gpgme_initialized = true;
1183
1258
  }
1184
1259
  
1185
 
  if(interface[0] != '\0'){
1186
 
    if_index = (AvahiIfIndex) if_nametoindex(interface);
1187
 
    if(if_index == 0){
1188
 
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
1189
 
      exitcode = EXIT_FAILURE;
1190
 
      goto end;
1191
 
    }
 
1260
  if(quit_now){
 
1261
    goto end;
1192
1262
  }
1193
1263
  
1194
1264
  if(connect_to != NULL){
1200
1270
      exitcode = EXIT_FAILURE;
1201
1271
      goto end;
1202
1272
    }
 
1273
    
 
1274
    if(quit_now){
 
1275
      goto end;
 
1276
    }
 
1277
    
1203
1278
    uint16_t port;
1204
1279
    errno = 0;
1205
1280
    tmpmax = strtoimax(address+1, &tmp, 10);
1209
1284
      exitcode = EXIT_FAILURE;
1210
1285
      goto end;
1211
1286
    }
 
1287
  
 
1288
    if(quit_now){
 
1289
      goto end;
 
1290
    }
 
1291
    
1212
1292
    port = (uint16_t)tmpmax;
1213
1293
    *address = '\0';
1214
1294
    address = connect_to;
1219
1299
    } else {
1220
1300
      af = AF_INET;
1221
1301
    }
 
1302
    
 
1303
    if(quit_now){
 
1304
      goto end;
 
1305
    }
 
1306
    
1222
1307
    ret = start_mandos_communication(address, port, if_index, af);
1223
1308
    if(ret < 0){
1224
1309
      exitcode = EXIT_FAILURE;
1227
1312
    }
1228
1313
    goto end;
1229
1314
  }
1230
 
    
 
1315
  
 
1316
  if(quit_now){
 
1317
    goto end;
 
1318
  }
 
1319
  
1231
1320
  {
1232
1321
    AvahiServerConfig config;
1233
1322
    /* Do not publish any local Zeroconf records */
1254
1343
    goto end;
1255
1344
  }
1256
1345
  
 
1346
  if(quit_now){
 
1347
    goto end;
 
1348
  }
 
1349
  
1257
1350
  /* Create the Avahi service browser */
1258
1351
  sb = avahi_s_service_browser_new(mc.server, if_index,
1259
1352
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
1265
1358
    goto end;
1266
1359
  }
1267
1360
  
 
1361
  if(quit_now){
 
1362
    goto end;
 
1363
  }
 
1364
  
1268
1365
  /* Run the main loop */
1269
1366
  
1270
1367
  if(debug){
1299
1396
    gpgme_release(mc.ctx);
1300
1397
  }
1301
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
  
1302
1417
  /* Removes the temp directory used by GPGME */
1303
1418
  if(tempdir_created){
1304
1419
    DIR *d;
1343
1458
    }
1344
1459
  }
1345
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
  
1346
1471
  return exitcode;
1347
1472
}