/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-02-09 19:26:19 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090209192619-7rse82x9pypq4ihk
* plugin-runner.c: Comment change.

* plugins.d/mandos-client.c: Comment changes.
  (quit_now): New global flag.
  (handle_sigterm): Only call avahi_simple_poll_quit once.
  (main): Also handle SIGINT and SIGHUP.

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(), remove() */
 
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,
43
 
                                   srand(), strtof() */
 
44
                                   srand() */
44
45
#include <stdbool.h>            /* bool, false, true */
45
46
#include <string.h>             /* memset(), strcmp(), strlen(),
46
47
                                   strerror(), asprintf(), strcpy() */
55
56
#include <fcntl.h>              /* open() */
56
57
#include <dirent.h>             /* opendir(), struct dirent, readdir()
57
58
                                 */
58
 
#include <inttypes.h>           /* PRIu16, PRIdMAX, intmax_t,
59
 
                                   strtoimax() */
 
59
#include <inttypes.h>           /* PRIu16, intmax_t, SCNdMAX */
60
60
#include <assert.h>             /* assert() */
61
61
#include <errno.h>              /* perror(), errno */
62
62
#include <time.h>               /* nanosleep(), time() */
134
134
} mandos_context;
135
135
 
136
136
/* global context so signal handler can reach it*/
137
 
mandos_context mc = { .simple_poll = NULL, .server = NULL,
138
 
                      .dh_bits = 1024, .priority = "SECURE256"
139
 
                      ":!CTYPE-X.509:+CTYPE-OPENPGP" };
 
137
mandos_context mc;
140
138
 
141
139
/*
142
140
 * Make additional room in "buffer" for at least BUFFER_SIZE more
854
852
       the callback function is called the Avahi server will free the
855
853
       resolver for us. */
856
854
    
857
 
    if(avahi_s_service_resolver_new(mc.server, interface, protocol,
858
 
                                    name, type, domain, protocol, 0,
859
 
                                    resolve_callback, NULL) == NULL)
 
855
    if(!(avahi_s_service_resolver_new(mc.server, interface,
 
856
                                       protocol, name, type, domain,
 
857
                                       AVAHI_PROTO_INET6, 0,
 
858
                                       resolve_callback, NULL)))
860
859
      fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
861
860
              name, avahi_strerror(avahi_server_errno(mc.server)));
862
861
    break;
875
874
 
876
875
sig_atomic_t quit_now = 0;
877
876
 
878
 
/* stop main loop after sigterm has been called */
879
877
static void handle_sigterm(__attribute__((unused)) int sig){
880
878
  if(quit_now){
881
879
    return;
893
891
  int error;
894
892
  int ret;
895
893
  intmax_t tmpmax;
896
 
  char *tmp;
 
894
  int numchars;
897
895
  int exitcode = EXIT_SUCCESS;
898
896
  const char *interface = "eth0";
899
897
  struct ifreq network;
907
905
  const char *seckey = PATHDIR "/" SECKEY;
908
906
  const char *pubkey = PATHDIR "/" PUBKEY;
909
907
  
 
908
  /* Initialize Mandos context */
 
909
  mc = (mandos_context){ .simple_poll = NULL, .server = NULL,
 
910
                         .dh_bits = 1024, .priority = "SECURE256"
 
911
                         ":!CTYPE-X.509:+CTYPE-OPENPGP" };
910
912
  bool gnutls_initialized = false;
911
913
  bool gpgme_initialized = false;
912
 
  float delay = 2.5f;
913
 
  
 
914
  double delay = 2.5;
 
915
 
914
916
  struct sigaction old_sigterm_action;
915
917
  struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
916
918
  
970
972
        pubkey = arg;
971
973
        break;
972
974
      case 129:                 /* --dh-bits */
973
 
        errno = 0;
974
 
        tmpmax = strtoimax(arg, &tmp, 10);
975
 
        if(errno != 0 or tmp == arg or *tmp != '\0'
976
 
           or tmpmax != (typeof(mc.dh_bits))tmpmax){
 
975
        ret = sscanf(arg, "%" SCNdMAX "%n", &tmpmax, &numchars);
 
976
        if(ret < 1 or tmpmax != (typeof(mc.dh_bits))tmpmax
 
977
           or arg[numchars] != '\0'){
977
978
          fprintf(stderr, "Bad number of DH bits\n");
978
979
          exit(EXIT_FAILURE);
979
980
        }
983
984
        mc.priority = arg;
984
985
        break;
985
986
      case 131:                 /* --delay */
986
 
        errno = 0;
987
 
        delay = strtof(arg, &tmp);
988
 
        if(errno != 0 or tmp == arg or *tmp != '\0'){
 
987
        ret = sscanf(arg, "%lf%n", &delay, &numchars);
 
988
        if(ret < 1 or arg[numchars] != '\0'){
989
989
          fprintf(stderr, "Bad delay\n");
990
990
          exit(EXIT_FAILURE);
991
991
        }
1012
1012
    }
1013
1013
  }
1014
1014
  
1015
 
  if(not debug){
1016
 
    avahi_set_log_function(empty_log);
1017
 
  }
1018
 
  
1019
 
  /* Initialize Avahi early so avahi_simple_poll_quit() can be called
1020
 
     from the signal handler */
1021
 
  /* Initialize the pseudo-RNG for Avahi */
1022
 
  srand((unsigned int) time(NULL));
1023
 
  mc.simple_poll = avahi_simple_poll_new();
1024
 
  if(mc.simple_poll == NULL){
1025
 
    fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
1026
 
    exitcode = EXIT_FAILURE;
1027
 
    goto end;
1028
 
  }
1029
 
  
1030
 
  sigemptyset(&sigterm_action.sa_mask);
1031
 
  ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
1032
 
  if(ret == -1){
1033
 
    perror("sigaddset");
1034
 
    exitcode = EXIT_FAILURE;
1035
 
    goto end;
1036
 
  }
1037
 
  ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
1038
 
  if(ret == -1){
1039
 
    perror("sigaddset");
1040
 
    exitcode = EXIT_FAILURE;
1041
 
    goto end;
1042
 
  }
1043
 
  ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
1044
 
  if(ret == -1){
1045
 
    perror("sigaddset");
1046
 
    exitcode = EXIT_FAILURE;
1047
 
    goto end;
1048
 
  }
1049
 
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1050
 
  if(ret == -1){
1051
 
    perror("sigaction");
1052
 
    exitcode = EXIT_FAILURE;
1053
 
    goto end;
1054
 
  }  
1055
 
  
1056
1015
  /* If the interface is down, bring it up */
1057
1016
  if(interface[0] != '\0'){
1058
1017
#ifdef __linux__
1197
1156
      goto end;
1198
1157
    }
1199
1158
    uint16_t port;
1200
 
    errno = 0;
1201
 
    tmpmax = strtoimax(address+1, &tmp, 10);
1202
 
    if(errno != 0 or tmp == address+1 or *tmp != '\0'
1203
 
       or tmpmax != (uint16_t)tmpmax){
 
1159
    ret = sscanf(address+1, "%" SCNdMAX "%n", &tmpmax, &numchars);
 
1160
    if(ret < 1 or tmpmax != (uint16_t)tmpmax
 
1161
       or address[numchars+1] != '\0'){
1204
1162
      fprintf(stderr, "Bad port number\n");
1205
1163
      exitcode = EXIT_FAILURE;
1206
1164
      goto end;
1223
1181
    }
1224
1182
    goto end;
1225
1183
  }
1226
 
    
 
1184
  
 
1185
  if(not debug){
 
1186
    avahi_set_log_function(empty_log);
 
1187
  }
 
1188
  
 
1189
  /* Initialize the pseudo-RNG for Avahi */
 
1190
  srand((unsigned int) time(NULL));
 
1191
  
 
1192
  /* Allocate main Avahi loop object */
 
1193
  mc.simple_poll = avahi_simple_poll_new();
 
1194
  if(mc.simple_poll == NULL){
 
1195
    fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
 
1196
    exitcode = EXIT_FAILURE;
 
1197
    goto end;
 
1198
  }
 
1199
  
1227
1200
  {
1228
1201
    AvahiServerConfig config;
1229
1202
    /* Do not publish any local Zeroconf records */
1252
1225
  
1253
1226
  /* Create the Avahi service browser */
1254
1227
  sb = avahi_s_service_browser_new(mc.server, if_index,
1255
 
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
 
1228
                                   AVAHI_PROTO_INET6, "_mandos._tcp",
1256
1229
                                   NULL, 0, browse_callback, NULL);
1257
1230
  if(sb == NULL){
1258
1231
    fprintf(stderr, "Failed to create service browser: %s\n",
1261
1234
    goto end;
1262
1235
  }
1263
1236
  
 
1237
  sigemptyset(&sigterm_action.sa_mask);
 
1238
  ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
 
1239
  if(ret == -1){
 
1240
    perror("sigaddset");
 
1241
    exitcode = EXIT_FAILURE;
 
1242
    goto end;
 
1243
  }
 
1244
  ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
 
1245
  if(ret == -1){
 
1246
    perror("sigaddset");
 
1247
    exitcode = EXIT_FAILURE;
 
1248
    goto end;
 
1249
  }
 
1250
  ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
 
1251
  if(ret == -1){
 
1252
    perror("sigaddset");
 
1253
    exitcode = EXIT_FAILURE;
 
1254
    goto end;
 
1255
  }
 
1256
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
 
1257
  if(ret == -1){
 
1258
    perror("sigaction");
 
1259
    exitcode = EXIT_FAILURE;
 
1260
    goto end;
 
1261
  }  
 
1262
  
1264
1263
  /* Run the main loop */
1265
1264
  
1266
1265
  if(debug){