/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:
873
882
  }
874
883
}
875
884
 
876
 
sig_atomic_t quit_now = 0;
877
 
 
878
885
/* stop main loop after sigterm has been called */
879
 
static void handle_sigterm(__attribute__((unused)) int sig){
 
886
static void handle_sigterm(int sig){
880
887
  if(quit_now){
881
888
    return;
882
889
  }
883
890
  quit_now = 1;
 
891
  signal_received = sig;
884
892
  int old_errno = errno;
885
893
  if(mc.simple_poll != NULL){
886
894
    avahi_simple_poll_quit(mc.simple_poll);
897
905
  int exitcode = EXIT_SUCCESS;
898
906
  const char *interface = "eth0";
899
907
  struct ifreq network;
900
 
  int sd;
 
908
  int sd = -1;
 
909
  bool take_down_interface = false;
901
910
  uid_t uid;
902
911
  gid_t gid;
903
912
  char *connect_to = NULL;
1046
1055
    exitcode = EXIT_FAILURE;
1047
1056
    goto end;
1048
1057
  }
1049
 
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1050
 
  if(ret == -1){
1051
 
    perror("sigaction");
1052
 
    exitcode = EXIT_FAILURE;
1053
 
    goto end;
1054
 
  }  
 
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
  }
1055
1101
  
1056
1102
  /* If the interface is down, bring it up */
1057
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
    
1058
1115
#ifdef __linux__
1059
1116
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
1060
1117
       messages to mess up the prompt */
1097
1154
    }
1098
1155
    if((network.ifr_flags & IFF_UP) == 0){
1099
1156
      network.ifr_flags |= IFF_UP;
 
1157
      take_down_interface = true;
1100
1158
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
1101
1159
      if(ret == -1){
 
1160
        take_down_interface = false;
1102
1161
        perror("ioctl SIOCSIFFLAGS");
1103
1162
        exitcode = EXIT_FAILURE;
1104
1163
#ifdef __linux__
1126
1185
        perror("nanosleep");
1127
1186
      }
1128
1187
    }
1129
 
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
1130
 
    if(ret == -1){
1131
 
      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
      }
1132
1194
    }
1133
1195
#ifdef __linux__
1134
1196
    if(restore_loglevel){
1141
1203
#endif  /* __linux__ */
1142
1204
  }
1143
1205
  
 
1206
  if(quit_now){
 
1207
    goto end;
 
1208
  }
 
1209
  
1144
1210
  uid = getuid();
1145
1211
  gid = getgid();
1146
1212
  
1155
1221
    perror("setuid");
1156
1222
  }
1157
1223
  
 
1224
  if(quit_now){
 
1225
    goto end;
 
1226
  }
 
1227
  
1158
1228
  ret = init_gnutls_global(pubkey, seckey);
1159
1229
  if(ret == -1){
1160
1230
    fprintf(stderr, "init_gnutls_global failed\n");
1164
1234
    gnutls_initialized = true;
1165
1235
  }
1166
1236
  
 
1237
  if(quit_now){
 
1238
    goto end;
 
1239
  }
 
1240
  
 
1241
  tempdir_created = true;
1167
1242
  if(mkdtemp(tempdir) == NULL){
 
1243
    tempdir_created = false;
1168
1244
    perror("mkdtemp");
1169
1245
    goto end;
1170
1246
  }
1171
 
  tempdir_created = true;
 
1247
  
 
1248
  if(quit_now){
 
1249
    goto end;
 
1250
  }
1172
1251
  
1173
1252
  if(not init_gpgme(pubkey, seckey, tempdir)){
1174
1253
    fprintf(stderr, "init_gpgme failed\n");
1178
1257
    gpgme_initialized = true;
1179
1258
  }
1180
1259
  
1181
 
  if(interface[0] != '\0'){
1182
 
    if_index = (AvahiIfIndex) if_nametoindex(interface);
1183
 
    if(if_index == 0){
1184
 
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
1185
 
      exitcode = EXIT_FAILURE;
1186
 
      goto end;
1187
 
    }
 
1260
  if(quit_now){
 
1261
    goto end;
1188
1262
  }
1189
1263
  
1190
1264
  if(connect_to != NULL){
1196
1270
      exitcode = EXIT_FAILURE;
1197
1271
      goto end;
1198
1272
    }
 
1273
    
 
1274
    if(quit_now){
 
1275
      goto end;
 
1276
    }
 
1277
    
1199
1278
    uint16_t port;
1200
1279
    errno = 0;
1201
1280
    tmpmax = strtoimax(address+1, &tmp, 10);
1205
1284
      exitcode = EXIT_FAILURE;
1206
1285
      goto end;
1207
1286
    }
 
1287
  
 
1288
    if(quit_now){
 
1289
      goto end;
 
1290
    }
 
1291
    
1208
1292
    port = (uint16_t)tmpmax;
1209
1293
    *address = '\0';
1210
1294
    address = connect_to;
1215
1299
    } else {
1216
1300
      af = AF_INET;
1217
1301
    }
 
1302
    
 
1303
    if(quit_now){
 
1304
      goto end;
 
1305
    }
 
1306
    
1218
1307
    ret = start_mandos_communication(address, port, if_index, af);
1219
1308
    if(ret < 0){
1220
1309
      exitcode = EXIT_FAILURE;
1223
1312
    }
1224
1313
    goto end;
1225
1314
  }
1226
 
    
 
1315
  
 
1316
  if(quit_now){
 
1317
    goto end;
 
1318
  }
 
1319
  
1227
1320
  {
1228
1321
    AvahiServerConfig config;
1229
1322
    /* Do not publish any local Zeroconf records */
1250
1343
    goto end;
1251
1344
  }
1252
1345
  
 
1346
  if(quit_now){
 
1347
    goto end;
 
1348
  }
 
1349
  
1253
1350
  /* Create the Avahi service browser */
1254
1351
  sb = avahi_s_service_browser_new(mc.server, if_index,
1255
1352
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
1261
1358
    goto end;
1262
1359
  }
1263
1360
  
 
1361
  if(quit_now){
 
1362
    goto end;
 
1363
  }
 
1364
  
1264
1365
  /* Run the main loop */
1265
1366
  
1266
1367
  if(debug){
1295
1396
    gpgme_release(mc.ctx);
1296
1397
  }
1297
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
  
1298
1417
  /* Removes the temp directory used by GPGME */
1299
1418
  if(tempdir_created){
1300
1419
    DIR *d;
1339
1458
    }
1340
1459
  }
1341
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
  
1342
1471
  return exitcode;
1343
1472
}