/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to plugins.d/mandos-client.c

* plugins.d/mandos-client.c (signal_received): New.
  (handle_sigterm): Save signal received.
  (main): Also set handler for SIGINT and SIGHUP.  If exiting due to
          signal, re-raise signal received instead of returning.

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() */
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,
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
  ret = sigaction(SIGINT, &sigterm_action, &old_sigterm_action);
 
1059
  if(ret == -1){
 
1060
    perror("sigaction");
 
1061
    exitcode = EXIT_FAILURE;
 
1062
    goto end;
 
1063
  }
 
1064
  ret = sigaction(SIGHUP, &sigterm_action, NULL);
 
1065
  if(ret == -1){
 
1066
    perror("sigaction");
 
1067
    exitcode = EXIT_FAILURE;
 
1068
    goto end;
 
1069
  }
 
1070
  ret = sigaction(SIGTERM, &sigterm_action, NULL);
 
1071
  if(ret == -1){
 
1072
    perror("sigaction");
 
1073
    exitcode = EXIT_FAILURE;
 
1074
    goto end;
 
1075
  }
1059
1076
  
1060
1077
  /* If the interface is down, bring it up */
1061
1078
  if(interface[0] != '\0'){
 
1079
    if_index = (AvahiIfIndex) if_nametoindex(interface);
 
1080
    if(if_index == 0){
 
1081
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
 
1082
      exitcode = EXIT_FAILURE;
 
1083
      goto end;
 
1084
    }
 
1085
    
 
1086
    if(quit_now){
 
1087
      goto end;
 
1088
    }
 
1089
    
1062
1090
#ifdef __linux__
1063
1091
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
1064
1092
       messages to mess up the prompt */
1101
1129
    }
1102
1130
    if((network.ifr_flags & IFF_UP) == 0){
1103
1131
      network.ifr_flags |= IFF_UP;
 
1132
      take_down_interface = true;
1104
1133
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
1105
1134
      if(ret == -1){
 
1135
        take_down_interface = false;
1106
1136
        perror("ioctl SIOCSIFFLAGS");
1107
1137
        exitcode = EXIT_FAILURE;
1108
1138
#ifdef __linux__
1130
1160
        perror("nanosleep");
1131
1161
      }
1132
1162
    }
1133
 
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
1134
 
    if(ret == -1){
1135
 
      perror("close");
 
1163
    if(not take_down_interface){
 
1164
      /* We won't need the socket anymore */
 
1165
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1166
      if(ret == -1){
 
1167
        perror("close");
 
1168
      }
1136
1169
    }
1137
1170
#ifdef __linux__
1138
1171
    if(restore_loglevel){
1145
1178
#endif  /* __linux__ */
1146
1179
  }
1147
1180
  
 
1181
  if(quit_now){
 
1182
    goto end;
 
1183
  }
 
1184
  
1148
1185
  uid = getuid();
1149
1186
  gid = getgid();
1150
1187
  
1159
1196
    perror("setuid");
1160
1197
  }
1161
1198
  
 
1199
  if(quit_now){
 
1200
    goto end;
 
1201
  }
 
1202
  
1162
1203
  ret = init_gnutls_global(pubkey, seckey);
1163
1204
  if(ret == -1){
1164
1205
    fprintf(stderr, "init_gnutls_global failed\n");
1168
1209
    gnutls_initialized = true;
1169
1210
  }
1170
1211
  
 
1212
  if(quit_now){
 
1213
    goto end;
 
1214
  }
 
1215
  
 
1216
  tempdir_created = true;
1171
1217
  if(mkdtemp(tempdir) == NULL){
 
1218
    tempdir_created = false;
1172
1219
    perror("mkdtemp");
1173
1220
    goto end;
1174
1221
  }
1175
 
  tempdir_created = true;
 
1222
  
 
1223
  if(quit_now){
 
1224
    goto end;
 
1225
  }
1176
1226
  
1177
1227
  if(not init_gpgme(pubkey, seckey, tempdir)){
1178
1228
    fprintf(stderr, "init_gpgme failed\n");
1182
1232
    gpgme_initialized = true;
1183
1233
  }
1184
1234
  
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
 
    }
 
1235
  if(quit_now){
 
1236
    goto end;
1192
1237
  }
1193
1238
  
1194
1239
  if(connect_to != NULL){
1200
1245
      exitcode = EXIT_FAILURE;
1201
1246
      goto end;
1202
1247
    }
 
1248
    
 
1249
    if(quit_now){
 
1250
      goto end;
 
1251
    }
 
1252
    
1203
1253
    uint16_t port;
1204
1254
    errno = 0;
1205
1255
    tmpmax = strtoimax(address+1, &tmp, 10);
1209
1259
      exitcode = EXIT_FAILURE;
1210
1260
      goto end;
1211
1261
    }
 
1262
  
 
1263
    if(quit_now){
 
1264
      goto end;
 
1265
    }
 
1266
    
1212
1267
    port = (uint16_t)tmpmax;
1213
1268
    *address = '\0';
1214
1269
    address = connect_to;
1219
1274
    } else {
1220
1275
      af = AF_INET;
1221
1276
    }
 
1277
    
 
1278
    if(quit_now){
 
1279
      goto end;
 
1280
    }
 
1281
    
1222
1282
    ret = start_mandos_communication(address, port, if_index, af);
1223
1283
    if(ret < 0){
1224
1284
      exitcode = EXIT_FAILURE;
1227
1287
    }
1228
1288
    goto end;
1229
1289
  }
1230
 
    
 
1290
  
 
1291
  if(quit_now){
 
1292
    goto end;
 
1293
  }
 
1294
  
1231
1295
  {
1232
1296
    AvahiServerConfig config;
1233
1297
    /* Do not publish any local Zeroconf records */
1254
1318
    goto end;
1255
1319
  }
1256
1320
  
 
1321
  if(quit_now){
 
1322
    goto end;
 
1323
  }
 
1324
  
1257
1325
  /* Create the Avahi service browser */
1258
1326
  sb = avahi_s_service_browser_new(mc.server, if_index,
1259
1327
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
1265
1333
    goto end;
1266
1334
  }
1267
1335
  
 
1336
  if(quit_now){
 
1337
    goto end;
 
1338
  }
 
1339
  
1268
1340
  /* Run the main loop */
1269
1341
  
1270
1342
  if(debug){
1299
1371
    gpgme_release(mc.ctx);
1300
1372
  }
1301
1373
  
 
1374
  /* Take down the network interface */
 
1375
  if(take_down_interface){
 
1376
    ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1377
    if(ret == -1){
 
1378
      perror("ioctl SIOCGIFFLAGS");
 
1379
    } else if(network.ifr_flags & IFF_UP) {
 
1380
      network.ifr_flags &= ~IFF_UP; /* clear flag */
 
1381
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
1382
      if(ret == -1){
 
1383
        perror("ioctl SIOCSIFFLAGS");
 
1384
      }
 
1385
    }
 
1386
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1387
    if(ret == -1){
 
1388
      perror("close");
 
1389
    }
 
1390
  }
 
1391
  
1302
1392
  /* Removes the temp directory used by GPGME */
1303
1393
  if(tempdir_created){
1304
1394
    DIR *d;
1343
1433
    }
1344
1434
  }
1345
1435
  
 
1436
  if(quit_now){
 
1437
    ret = sigaction(signal_received, &old_sigterm_action, NULL);
 
1438
    if(ret == -1){
 
1439
      perror("sigaction");
 
1440
    }
 
1441
    raise(signal_received);
 
1442
  }
 
1443
  
1346
1444
  return exitcode;
1347
1445
}