/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: 2010-09-11 18:34:55 UTC
  • mfrom: (24.1.162 mandos)
  • Revision ID: teddy@fukt.bsnet.se-20100911183455-kmzttvbywdbfbepm
Merge from Björn.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandos-client.c
127
127
static const char mandos_protocol_version[] = "1";
128
128
const char *argp_program_version = "mandos-client " VERSION;
129
129
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
 
130
static const char sys_class_net[] = "/sys/class/net";
 
131
char *connect_to = NULL;
130
132
 
131
133
/* Used for passing in values through the Avahi callback functions */
132
134
typedef struct {
1020
1022
  errno = old_errno;
1021
1023
}
1022
1024
 
 
1025
/* 
 
1026
 * This function determines if a directory entry in /sys/class/net
 
1027
 * corresponds to an acceptable network device.
 
1028
 * (This function is passed to scandir(3) as a filter function.)
 
1029
 */
 
1030
int good_interface(const struct dirent *if_entry){
 
1031
  ssize_t ssret;
 
1032
  char *flagname = NULL;
 
1033
  int ret = asprintf(&flagname, "%s/%s/flags", sys_class_net,
 
1034
                     if_entry->d_name);
 
1035
  if(ret < 0){
 
1036
    perror("asprintf");
 
1037
    return 0;
 
1038
  }
 
1039
  if(if_entry->d_name[0] == '.'){
 
1040
    return 0;
 
1041
  }
 
1042
  int flags_fd = (int)TEMP_FAILURE_RETRY(open(flagname, O_RDONLY));
 
1043
  if(flags_fd == -1){
 
1044
    perror("open");
 
1045
    return 0;
 
1046
  }
 
1047
  typedef short ifreq_flags;    /* ifreq.ifr_flags in netdevice(7) */
 
1048
  /* read line from flags_fd */
 
1049
  ssize_t to_read = (sizeof(ifreq_flags)*2)+3; /* "0x1003\n" */
 
1050
  char *flagstring = malloc((size_t)to_read+1); /* +1 for final \0 */
 
1051
  flagstring[(size_t)to_read] = '\0';
 
1052
  if(flagstring == NULL){
 
1053
    perror("malloc");
 
1054
    close(flags_fd);
 
1055
    return 0;
 
1056
  }
 
1057
  while(to_read > 0){
 
1058
    ssret = (ssize_t)TEMP_FAILURE_RETRY(read(flags_fd, flagstring,
 
1059
                                             (size_t)to_read));
 
1060
    if(ssret == -1){
 
1061
      perror("read");
 
1062
      free(flagstring);
 
1063
      close(flags_fd);
 
1064
      return 0;
 
1065
    }
 
1066
    to_read -= ssret;
 
1067
    if(ssret == 0){
 
1068
      break;
 
1069
    }
 
1070
  }
 
1071
  close(flags_fd);
 
1072
  intmax_t tmpmax;
 
1073
  char *tmp;
 
1074
  errno = 0;
 
1075
  tmpmax = strtoimax(flagstring, &tmp, 0);
 
1076
  if(errno != 0 or tmp == flagstring or (*tmp != '\0'
 
1077
                                         and not (isspace(*tmp)))
 
1078
     or tmpmax != (ifreq_flags)tmpmax){
 
1079
    if(debug){
 
1080
      fprintf(stderr, "Invalid flags \"%s\" for interface \"%s\"\n",
 
1081
              flagstring, if_entry->d_name);
 
1082
    }
 
1083
    free(flagstring);
 
1084
    return 0;
 
1085
  }
 
1086
  free(flagstring);
 
1087
  ifreq_flags flags = (ifreq_flags)tmpmax;
 
1088
  /* Reject the loopback device */
 
1089
  if(flags & IFF_LOOPBACK){
 
1090
    if(debug){
 
1091
      fprintf(stderr, "Rejecting loopback interface \"%s\"\n",
 
1092
              if_entry->d_name);
 
1093
    }
 
1094
    return 0;
 
1095
  }
 
1096
  /* Accept point-to-point devices only if connect_to is specified */
 
1097
  if(connect_to != NULL and (flags & IFF_POINTOPOINT)){
 
1098
    if(debug){
 
1099
      fprintf(stderr, "Accepting point-to-point interface \"%s\"\n",
 
1100
              if_entry->d_name);
 
1101
    }
 
1102
    return 1;
 
1103
  }
 
1104
  /* Otherwise, reject non-broadcast-capable devices */
 
1105
  if(not (flags & IFF_BROADCAST)){
 
1106
    if(debug){
 
1107
      fprintf(stderr, "Rejecting non-broadcast interface \"%s\"\n",
 
1108
              if_entry->d_name);
 
1109
    }
 
1110
    return 0;
 
1111
  }
 
1112
  /* Accept this device */
 
1113
  if(debug){
 
1114
    fprintf(stderr, "Interface \"%s\" is acceptable\n",
 
1115
            if_entry->d_name);
 
1116
  }
 
1117
  return 1;
 
1118
}
 
1119
 
1023
1120
int main(int argc, char *argv[]){
1024
1121
  AvahiSServiceBrowser *sb = NULL;
1025
1122
  int error;
1027
1124
  intmax_t tmpmax;
1028
1125
  char *tmp;
1029
1126
  int exitcode = EXIT_SUCCESS;
1030
 
  const char *interface = "eth0";
 
1127
  const char *interface = "";
1031
1128
  struct ifreq network;
1032
1129
  int sd = -1;
1033
1130
  bool take_down_interface = false;
1034
1131
  uid_t uid;
1035
1132
  gid_t gid;
1036
 
  char *connect_to = NULL;
1037
1133
  char tempdir[] = "/tmp/mandosXXXXXX";
1038
1134
  bool tempdir_created = false;
1039
1135
  AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
1196
1292
  if(not debug){
1197
1293
    avahi_set_log_function(empty_log);
1198
1294
  }
 
1295
 
 
1296
  if(interface[0] == '\0'){
 
1297
    struct dirent **direntries;
 
1298
    ret = scandir(sys_class_net, &direntries, good_interface,
 
1299
                  alphasort);
 
1300
    if(ret >= 1){
 
1301
      /* Pick the first good interface */
 
1302
      interface = strdup(direntries[0]->d_name);
 
1303
      if(debug){
 
1304
        fprintf(stderr, "Using interface \"%s\"\n", interface);
 
1305
      }
 
1306
      if(interface == NULL){
 
1307
        perror("malloc");
 
1308
        free(direntries);
 
1309
        exitcode = EXIT_FAILURE;
 
1310
        goto end;
 
1311
      }
 
1312
      free(direntries);
 
1313
    } else {
 
1314
      free(direntries);
 
1315
      fprintf(stderr, "Could not find a network interface\n");
 
1316
      exitcode = EXIT_FAILURE;
 
1317
      goto end;
 
1318
    }
 
1319
  }
1199
1320
  
1200
1321
  /* Initialize Avahi early so avahi_simple_poll_quit() can be called
1201
1322
     from the signal handler */
1272
1393
  }
1273
1394
  
1274
1395
  /* If the interface is down, bring it up */
1275
 
  if(interface[0] != '\0'){
 
1396
  if(strcmp(interface, "none") != 0){
1276
1397
    if_index = (AvahiIfIndex) if_nametoindex(interface);
1277
1398
    if(if_index == 0){
1278
1399
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
1349
1470
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
1350
1471
      if(ret == -1){
1351
1472
        take_down_interface = false;
1352
 
        perror("ioctl SIOCSIFFLAGS");
 
1473
        perror("ioctl SIOCSIFFLAGS +IFF_UP");
1353
1474
        exitcode = EX_OSERR;
1354
1475
#ifdef __linux__
1355
1476
        if(restore_loglevel){
1624
1745
        network.ifr_flags &= ~(short)IFF_UP; /* clear flag */
1625
1746
        ret = ioctl(sd, SIOCSIFFLAGS, &network);
1626
1747
        if(ret == -1){
1627
 
          perror("ioctl SIOCSIFFLAGS");
 
1748
          perror("ioctl SIOCSIFFLAGS -IFF_UP");
1628
1749
        }
1629
1750
      }
1630
1751
      ret = (int)TEMP_FAILURE_RETRY(close(sd));