/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/mandosclient.c

  • Committer: Teddy Hogeborn
  • Date: 2008-08-03 16:05:52 UTC
  • Revision ID: teddy@fukt.bsnet.se-20080803160552-mvqy5klry3wn57x1
* plugins.d/mandosclient.c (initgnutls): Moved "err" variable into its
                                         own code block.
  (start_mandos_communication): Moved call to "initgnutls" to
                                beginning.  Renamed label "exit" to
                                "mandos_end".  Bug fix: set return
                                code if failure to realloc buffer.
                                Close TLS session as early as
                                possible.
  (resolve_callback): Also print interface number when debugging.
  (main): Moved "debug_int" and "config" into their own respective
          code blocks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/mandosclient.c
32
32
#define _LARGEFILE_SOURCE
33
33
#define _FILE_OFFSET_BITS 64
34
34
 
35
 
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY() */
36
 
 
37
35
#include <stdio.h>
38
36
#include <assert.h>
39
37
#include <stdlib.h>
81
79
 
82
80
bool debug = false;
83
81
 
84
 
const char mandos_protocol_version[] = "1";
85
 
 
86
82
/* Used for passing in values through all the callback functions */
87
83
typedef struct {
88
84
  AvahiSimplePoll *simple_poll;
89
85
  AvahiServer *server;
90
86
  gnutls_certificate_credentials_t cred;
91
87
  unsigned int dh_bits;
92
 
  gnutls_dh_params_t dh_params;
93
88
  const char *priority;
94
89
} mandos_context;
95
90
 
96
 
size_t adjustbuffer(char **buffer, size_t buffer_length,
97
 
                  size_t buffer_capacity){
98
 
  if (buffer_length + BUFFER_SIZE > buffer_capacity){
99
 
    *buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
100
 
    if (buffer == NULL){
101
 
      return 0;
102
 
    }
103
 
    buffer_capacity += BUFFER_SIZE;
104
 
  }
105
 
  return buffer_capacity;
106
 
}
107
 
 
108
91
/* 
109
92
 * Decrypt OpenPGP data using keyrings in HOMEDIR.
110
93
 * Returns -1 on error
233
216
  
234
217
  *plaintext = NULL;
235
218
  while(true){
236
 
    plaintext_capacity = adjustbuffer(plaintext, (size_t)plaintext_length,
237
 
                                      plaintext_capacity);
238
 
    if (plaintext_capacity == 0){
239
 
        perror("adjustbuffer");
 
219
    if (plaintext_length + BUFFER_SIZE
 
220
        > (ssize_t) plaintext_capacity){
 
221
      *plaintext = realloc(*plaintext, plaintext_capacity
 
222
                            + BUFFER_SIZE);
 
223
      if (*plaintext == NULL){
 
224
        perror("realloc");
240
225
        plaintext_length = -1;
241
226
        goto decrypt_end;
 
227
      }
 
228
      plaintext_capacity += BUFFER_SIZE;
242
229
    }
243
230
    
244
231
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
287
274
  fprintf(stderr, "GnuTLS: %s", string);
288
275
}
289
276
 
290
 
static int init_gnutls_global(mandos_context *mc){
 
277
static int initgnutls(mandos_context *mc, gnutls_session_t *session,
 
278
                      gnutls_dh_params_t *dh_params){
291
279
  int ret;
292
280
  
293
281
  if(debug){
335
323
  }
336
324
  
337
325
  /* GnuTLS server initialization */
338
 
  ret = gnutls_dh_params_init(&mc->dh_params);
 
326
  ret = gnutls_dh_params_init(dh_params);
339
327
  if (ret != GNUTLS_E_SUCCESS) {
340
328
    fprintf (stderr, "Error in GnuTLS DH parameter initialization:"
341
329
             " %s\n", safer_gnutls_strerror(ret));
342
330
    return -1;
343
331
  }
344
 
  ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
 
332
  ret = gnutls_dh_params_generate2(*dh_params, mc->dh_bits);
345
333
  if (ret != GNUTLS_E_SUCCESS) {
346
334
    fprintf (stderr, "Error in GnuTLS prime generation: %s\n",
347
335
             safer_gnutls_strerror(ret));
348
336
    return -1;
349
337
  }
350
338
  
351
 
  gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
352
 
 
353
 
  return 0;
354
 
}
355
 
 
356
 
static int init_gnutls_session(mandos_context *mc, gnutls_session_t *session){
357
 
  int ret;
 
339
  gnutls_certificate_set_dh_params(mc->cred, *dh_params);
 
340
  
358
341
  /* GnuTLS session creation */
359
342
  ret = gnutls_init(session, GNUTLS_SERVER);
360
343
  if (ret != GNUTLS_E_SUCCESS){
405
388
  size_t buffer_length = 0;
406
389
  size_t buffer_capacity = 0;
407
390
  ssize_t decrypted_buffer_size;
408
 
  size_t written;
 
391
  size_t written = 0;
409
392
  int retval = 0;
410
393
  char interface[IF_NAMESIZE];
411
394
  gnutls_session_t session;
412
395
  gnutls_dh_params_t dh_params;
413
396
  
414
 
  ret = init_gnutls_session (mc, &session);
 
397
  ret = initgnutls (mc, &session, &dh_params);
415
398
  if (ret != 0){
416
399
    return -1;
417
400
  }
470
453
    perror("connect");
471
454
    return -1;
472
455
  }
473
 
 
474
 
  const char *out = mandos_protocol_version;
475
 
  written = 0;
476
 
  while (true){
477
 
    size_t out_size = strlen(out);
478
 
    ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
479
 
                                   out_size - written));
480
 
    if (ret == -1){
481
 
      perror("write");
482
 
      retval = -1;
483
 
      goto mandos_end;
484
 
    }
485
 
    written += (size_t)ret;
486
 
    if(written < out_size){
487
 
      continue;
488
 
    } else {
489
 
      if (out == mandos_protocol_version){
490
 
        written = 0;
491
 
        out = "\r\n";
492
 
      } else {
493
 
        break;
494
 
      }
495
 
    }
496
 
  }
497
 
 
 
456
  
498
457
  if(debug){
499
458
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
500
459
  }
520
479
  }
521
480
 
522
481
  while(true){
523
 
    buffer_capacity = adjustbuffer(&buffer, buffer_length, buffer_capacity);
524
 
    if (buffer_capacity == 0){
525
 
      perror("adjustbuffer");
526
 
      retval = -1;
527
 
      goto mandos_end;
 
482
    if (buffer_length + BUFFER_SIZE > buffer_capacity){
 
483
      buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
 
484
      if (buffer == NULL){
 
485
        perror("realloc");
 
486
        retval = -1;
 
487
        goto mandos_end;
 
488
      }
 
489
      buffer_capacity += BUFFER_SIZE;
528
490
    }
529
491
    
530
492
    ret = gnutls_record_recv(session, buffer+buffer_length,
570
532
                                               &decrypted_buffer,
571
533
                                               keydir);
572
534
    if (decrypted_buffer_size >= 0){
573
 
      written = 0;
574
535
      while(written < (size_t) decrypted_buffer_size){
575
536
        ret = (int)fwrite (decrypted_buffer + written, 1,
576
537
                           (size_t)decrypted_buffer_size - written,
727
688
    const char *interface = "eth0";
728
689
    struct ifreq network;
729
690
    int sd;
730
 
    uid_t uid;
731
 
    gid_t gid;
732
691
    char *connect_to = NULL;
733
692
    AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
734
693
    mandos_context mc = { .simple_poll = NULL, .server = NULL,
809
768
      perror("combinepath");
810
769
      goto end;
811
770
    }
812
 
 
813
 
    ret = init_gnutls_global(&mc);
814
 
    if (ret == -1){
815
 
      fprintf(stderr, "init_gnutls_global\n");
816
 
      goto end;
817
 
    }
818
 
 
819
 
    uid = getuid();
820
 
    gid = getgid();
821
 
 
822
 
    ret = setuid(uid);
823
 
    if (ret == -1){
824
 
      perror("setuid");
825
 
    }
826
 
    
827
 
    setgid(gid);
828
 
    if (ret == -1){
829
 
      perror("setgid");
830
 
    }
831
771
    
832
772
    if_index = (AvahiIfIndex) if_nametoindex(interface);
833
773
    if(if_index == 0){
841
781
      char *address = strrchr(connect_to, ':');
842
782
      if(address == NULL){
843
783
        fprintf(stderr, "No colon in address\n");
844
 
        exitcode = EXIT_FAILURE;
845
 
        goto end;
 
784
        exit(EXIT_FAILURE);
846
785
      }
847
786
      errno = 0;
848
787
      uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
849
788
      if(errno){
850
789
        perror("Bad port number");
851
 
        exitcode = EXIT_FAILURE;
852
 
        goto end;
 
790
        exit(EXIT_FAILURE);
853
791
      }
854
792
      *address = '\0';
855
793
      address = connect_to;
856
794
      ret = start_mandos_communication(address, port, if_index, &mc);
857
795
      if(ret < 0){
858
 
        exitcode = EXIT_FAILURE;
 
796
        exit(EXIT_FAILURE);
859
797
      } else {
860
 
        exitcode = EXIT_SUCCESS;
 
798
        exit(EXIT_SUCCESS);
861
799
      }
862
 
      goto end;
863
800
    }
864
801
    
865
802
    /* If the interface is down, bring it up */