/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-01-13 01:36:46 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090113013646-1u21ie2oaotcekdz
* plugin-runner.c (main): Use "sscanf" instead of "strtol"; using the
                          correct type instead of casting.
* plugins.d/mandos-cliend.c (main): Detect an empty string when
                                    parsing DH bits and port number.
* plugins.d/splashy.c (main): Use "sscanf" instead of "strtoul"; using
                              the correct type instead of casting.
* plugins.d/usplash.c (main): - '' -

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandos-client.c
1
1
/*  -*- coding: utf-8 -*- */
2
2
/*
3
 
 * Mandos client - get and decrypt data from a Mandos server
 
3
 * Mandos-client - get and decrypt data from a Mandos server
4
4
 *
5
5
 * This program is partly derived from an example program for an Avahi
6
6
 * service browser, downloaded from
9
9
 * "browse_callback", and parts of "main".
10
10
 * 
11
11
 * Everything else is
12
 
 * Copyright © 2008 Teddy Hogeborn & Björn Påhlsson
 
12
 * Copyright © 2008,2009 Teddy Hogeborn
 
13
 * Copyright © 2008,2009 Björn Påhlsson
13
14
 * 
14
15
 * This program is free software: you can redistribute it and/or
15
16
 * modify it under the terms of the GNU General Public License as
35
36
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
36
37
 
37
38
#include <stdio.h>              /* fprintf(), stderr, fwrite(),
38
 
                                   stdout, ferror() */
 
39
                                   stdout, ferror(), sscanf */
39
40
#include <stdint.h>             /* uint16_t, uint32_t */
40
41
#include <stddef.h>             /* NULL, size_t, ssize_t */
41
42
#include <stdlib.h>             /* free(), EXIT_SUCCESS, EXIT_FAILURE,
54
55
                                   connect() */
55
56
#include <fcntl.h>              /* open() */
56
57
#include <dirent.h>             /* opendir(), struct dirent, readdir() */
57
 
#include <inttypes.h>           /* PRIu16 */
 
58
#include <inttypes.h>           /* PRIu16, SCNu16 */
58
59
#include <assert.h>             /* assert() */
59
60
#include <errno.h>              /* perror(), errno */
60
61
#include <time.h>               /* time() */
66
67
                                   getuid(), getgid(), setuid(),
67
68
                                   setgid() */
68
69
#include <arpa/inet.h>          /* inet_pton(), htons */
69
 
#include <iso646.h>             /* not, and */
 
70
#include <iso646.h>             /* not, and, or */
70
71
#include <argp.h>               /* struct argp_option, error_t, struct
71
72
                                   argp_state, struct argp,
72
73
                                   argp_parse(), ARGP_KEY_ARG,
101
102
 
102
103
#define BUFFER_SIZE 256
103
104
 
104
 
/*
105
 
  #define PATHDIR "/conf/conf.d/mandos"
106
 
*/
107
 
 
108
105
#define PATHDIR "/conf/conf.d/mandos"
109
106
#define SECKEY "seckey.txt"
110
107
#define PUBKEY "pubkey.txt"
111
108
 
112
109
bool debug = false;
113
110
static const char mandos_protocol_version[] = "1";
114
 
const char *argp_program_version = "mandos-client 1.0";
 
111
const char *argp_program_version = "mandos-client " VERSION;
115
112
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
116
113
 
117
114
/* Used for passing in values through the Avahi callback functions */
159
156
    int fd;
160
157
    gpgme_data_t pgp_data;
161
158
    
162
 
    fd = TEMP_FAILURE_RETRY(open(filename, O_RDONLY));
 
159
    fd = (int)TEMP_FAILURE_RETRY(open(filename, O_RDONLY));
163
160
    if(fd == -1){
164
161
      perror("open");
165
162
      return false;
179
176
      return false;
180
177
    }
181
178
    
182
 
    ret = TEMP_FAILURE_RETRY(close(fd));
 
179
    ret = (int)TEMP_FAILURE_RETRY(close(fd));
183
180
    if(ret == -1){
184
181
      perror("close");
185
182
    }
504
501
                                      AvahiIfIndex if_index,
505
502
                                      mandos_context *mc){
506
503
  int ret, tcp_sd;
 
504
  ssize_t sret;
507
505
  union { struct sockaddr in; struct sockaddr_in6 in6; } to;
508
506
  char *buffer = NULL;
509
507
  char *decrypted_buffer;
580
578
  written = 0;
581
579
  while (true){
582
580
    size_t out_size = strlen(out);
583
 
    ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
 
581
    ret = (int)TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
584
582
                                   out_size - written));
585
583
    if (ret == -1){
586
584
      perror("write");
635
633
      goto mandos_end;
636
634
    }
637
635
    
638
 
    ret = gnutls_record_recv(session, buffer+buffer_length,
639
 
                             BUFFER_SIZE);
640
 
    if (ret == 0){
 
636
    sret = gnutls_record_recv(session, buffer+buffer_length,
 
637
                              BUFFER_SIZE);
 
638
    if (sret == 0){
641
639
      break;
642
640
    }
643
 
    if (ret < 0){
644
 
      switch(ret){
 
641
    if (sret < 0){
 
642
      switch(sret){
645
643
      case GNUTLS_E_INTERRUPTED:
646
644
      case GNUTLS_E_AGAIN:
647
645
        break;
664
662
        goto mandos_end;
665
663
      }
666
664
    } else {
667
 
      buffer_length += (size_t) ret;
 
665
      buffer_length += (size_t) sret;
668
666
    }
669
667
  }
670
668
  
706
704
  
707
705
 mandos_end:
708
706
  free(buffer);
709
 
  ret = TEMP_FAILURE_RETRY(close(tcp_sd));
 
707
  ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
710
708
  if(ret == -1){
711
709
    perror("close");
712
710
  }
868
866
      
869
867
      error_t parse_opt (int key, char *arg,
870
868
                         struct argp_state *state) {
871
 
        /* Get the INPUT argument from `argp_parse', which we know is
872
 
           a pointer to our plugin list pointer. */
873
869
        switch (key) {
874
870
        case 128:               /* --debug */
875
871
          debug = true;
887
883
          pubkey = arg;
888
884
          break;
889
885
        case 129:               /* --dh-bits */
890
 
          errno = 0;
891
 
          mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);
892
 
          if (errno){
893
 
            perror("strtol");
 
886
          ret = sscanf(arg, "%u", &mc.dh_bits);
 
887
          if(ret != 1){
 
888
            fprintf(stderr, "Bad number of DH bits\n");
894
889
            exit(EXIT_FAILURE);
895
890
          }
896
891
          break;
943
938
          goto end;
944
939
        }
945
940
      }
946
 
      ret = TEMP_FAILURE_RETRY(close(sd));
 
941
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
947
942
      if(ret == -1){
948
943
        perror("close");
949
944
      }
1000
995
        exitcode = EXIT_FAILURE;
1001
996
        goto end;
1002
997
      }
1003
 
      errno = 0;
1004
 
      uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
1005
 
      if(errno){
1006
 
        perror("Bad port number");
 
998
      uint16_t port;
 
999
      ret = sscanf(address+1, "%" SCNu16, &port);
 
1000
      if(ret != 1){
 
1001
        fprintf(stderr, "Bad port number\n");
1007
1002
        exitcode = EXIT_FAILURE;
1008
1003
        goto end;
1009
1004
      }
1112
1107
      struct dirent *direntry;
1113
1108
      d = opendir(tempdir);
1114
1109
      if(d == NULL){
1115
 
        perror("opendir");
 
1110
        if(errno != ENOENT){
 
1111
          perror("opendir");
 
1112
        }
1116
1113
      } else {
1117
1114
        while(true){
1118
1115
          direntry = readdir(d);
1138
1135
        closedir(d);
1139
1136
      }
1140
1137
      ret = rmdir(tempdir);
1141
 
      if(ret == -1){
 
1138
      if(ret == -1 and errno != ENOENT){
1142
1139
        perror("rmdir");
1143
1140
      }
1144
1141
    }