/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-05-21 20:04:06 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090521200406-2d5yqwuelzai5si8
* Makefile (GNUTLS_CFLAGS, GNUTLS_LIBS): Use "pkg-config" instead of
                                         "libgnutls-config".

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
8
8
 * includes the following functions: "resolve_callback",
9
9
 * "browse_callback", and parts of "main".
10
10
 * 
11
 
 * Everything else is Copyright © 2007-2008 Teddy Hogeborn and Björn
12
 
 * Påhlsson.
 
11
 * Everything else is
 
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
25
26
 * along with this program.  If not, see
26
27
 * <http://www.gnu.org/licenses/>.
27
28
 * 
28
 
 * Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
29
 
 * <https://www.fukt.bsnet.se/~teddy/>.
 
29
 * Contact the authors at <mandos@fukt.bsnet.se>.
30
30
 */
31
31
 
32
 
#define _FORTIFY_SOURCE 2
33
 
 
 
32
/* Needed by GPGME, specifically gpgme_data_seek() */
 
33
#ifndef _LARGEFILE_SOURCE
34
34
#define _LARGEFILE_SOURCE
 
35
#endif
 
36
#ifndef _FILE_OFFSET_BITS
35
37
#define _FILE_OFFSET_BITS 64
36
 
 
37
 
#include <stdio.h>
38
 
#include <assert.h>
39
 
#include <stdlib.h>
40
 
#include <time.h>
41
 
#include <net/if.h>             /* if_nametoindex */
42
 
 
 
38
#endif
 
39
 
 
40
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
 
41
 
 
42
#include <stdio.h>              /* fprintf(), stderr, fwrite(),
 
43
                                   stdout, ferror(), remove() */
 
44
#include <stdint.h>             /* uint16_t, uint32_t */
 
45
#include <stddef.h>             /* NULL, size_t, ssize_t */
 
46
#include <stdlib.h>             /* free(), EXIT_SUCCESS, EXIT_FAILURE,
 
47
                                   srand(), strtof() */
 
48
#include <stdbool.h>            /* bool, false, true */
 
49
#include <string.h>             /* memset(), strcmp(), strlen(),
 
50
                                   strerror(), asprintf(), strcpy() */
 
51
#include <sys/ioctl.h>          /* ioctl */
 
52
#include <sys/types.h>          /* socket(), inet_pton(), sockaddr,
 
53
                                   sockaddr_in6, PF_INET6,
 
54
                                   SOCK_STREAM, uid_t, gid_t, open(),
 
55
                                   opendir(), DIR */
 
56
#include <sys/stat.h>           /* open() */
 
57
#include <sys/socket.h>         /* socket(), struct sockaddr_in6,
 
58
                                   inet_pton(), connect() */
 
59
#include <fcntl.h>              /* open() */
 
60
#include <dirent.h>             /* opendir(), struct dirent, readdir()
 
61
                                 */
 
62
#include <inttypes.h>           /* PRIu16, PRIdMAX, intmax_t,
 
63
                                   strtoimax() */
 
64
#include <assert.h>             /* assert() */
 
65
#include <errno.h>              /* perror(), errno */
 
66
#include <time.h>               /* nanosleep(), time() */
 
67
#include <net/if.h>             /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
 
68
                                   SIOCSIFFLAGS, if_indextoname(),
 
69
                                   if_nametoindex(), IF_NAMESIZE */
 
70
#include <netinet/in.h>         /* IN6_IS_ADDR_LINKLOCAL,
 
71
                                   INET_ADDRSTRLEN, INET6_ADDRSTRLEN
 
72
                                */
 
73
#include <unistd.h>             /* close(), SEEK_SET, off_t, write(),
 
74
                                   getuid(), getgid(), setuid(),
 
75
                                   setgid() */
 
76
#include <arpa/inet.h>          /* inet_pton(), htons */
 
77
#include <iso646.h>             /* not, or, and */
 
78
#include <argp.h>               /* struct argp_option, error_t, struct
 
79
                                   argp_state, struct argp,
 
80
                                   argp_parse(), ARGP_KEY_ARG,
 
81
                                   ARGP_KEY_END, ARGP_ERR_UNKNOWN */
 
82
#include <signal.h>             /* sigemptyset(), sigaddset(),
 
83
                                   sigaction(), SIGTERM, sigaction,
 
84
                                   sig_atomic_t */
 
85
 
 
86
#ifdef __linux__
 
87
#include <sys/klog.h>           /* klogctl() */
 
88
#endif  /* __linux__ */
 
89
 
 
90
/* Avahi */
 
91
/* All Avahi types, constants and functions
 
92
 Avahi*, avahi_*,
 
93
 AVAHI_* */
43
94
#include <avahi-core/core.h>
44
95
#include <avahi-core/lookup.h>
45
96
#include <avahi-core/log.h>
47
98
#include <avahi-common/malloc.h>
48
99
#include <avahi-common/error.h>
49
100
 
50
 
//mandos client part
51
 
#include <sys/types.h>          /* socket(), inet_pton() */
52
 
#include <sys/socket.h>         /* socket(), struct sockaddr_in6,
53
 
                                   struct in6_addr, inet_pton() */
54
 
#include <gnutls/gnutls.h>      /* All GnuTLS stuff */
55
 
#include <gnutls/openpgp.h>     /* GnuTLS with openpgp stuff */
56
 
 
57
 
#include <unistd.h>             /* close() */
58
 
#include <netinet/in.h>
59
 
#include <stdbool.h>            /* true */
60
 
#include <string.h>             /* memset */
61
 
#include <arpa/inet.h>          /* inet_pton() */
62
 
#include <iso646.h>             /* not */
63
 
 
64
 
// gpgme
65
 
#include <errno.h>              /* perror() */
66
 
#include <gpgme.h>
67
 
 
68
 
// getopt long
69
 
#include <getopt.h>
 
101
/* GnuTLS */
 
102
#include <gnutls/gnutls.h>      /* All GnuTLS types, constants and
 
103
                                   functions:
 
104
                                   gnutls_*
 
105
                                   init_gnutls_session(),
 
106
                                   GNUTLS_* */
 
107
#include <gnutls/openpgp.h>
 
108
                          /* gnutls_certificate_set_openpgp_key_file(),
 
109
                                   GNUTLS_OPENPGP_FMT_BASE64 */
 
110
 
 
111
/* GPGME */
 
112
#include <gpgme.h>              /* All GPGME types, constants and
 
113
                                   functions:
 
114
                                   gpgme_*
 
115
                                   GPGME_PROTOCOL_OpenPGP,
 
116
                                   GPG_ERR_NO_* */
70
117
 
71
118
#define BUFFER_SIZE 256
72
 
#define DH_BITS 1024
73
119
 
74
 
const char *certdir = "/conf/conf.d/cryptkeyreq/";
75
 
const char *certfile = "openpgp-client.txt";
76
 
const char *certkey = "openpgp-client-key.txt";
 
120
#define PATHDIR "/conf/conf.d/mandos"
 
121
#define SECKEY "seckey.txt"
 
122
#define PUBKEY "pubkey.txt"
77
123
 
78
124
bool debug = false;
 
125
static const char mandos_protocol_version[] = "1";
 
126
const char *argp_program_version = "mandos-client " VERSION;
 
127
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
79
128
 
 
129
/* Used for passing in values through the Avahi callback functions */
80
130
typedef struct {
81
 
  gnutls_session_t session;
 
131
  AvahiSimplePoll *simple_poll;
 
132
  AvahiServer *server;
82
133
  gnutls_certificate_credentials_t cred;
 
134
  unsigned int dh_bits;
83
135
  gnutls_dh_params_t dh_params;
84
 
} encrypted_session;
85
 
 
86
 
 
87
 
ssize_t pgp_packet_decrypt (char *packet, size_t packet_size,
88
 
                            char **new_packet, const char *homedir){
89
 
  gpgme_data_t dh_crypto, dh_plain;
 
136
  const char *priority;
90
137
  gpgme_ctx_t ctx;
 
138
} mandos_context;
 
139
 
 
140
/* global context so signal handler can reach it*/
 
141
mandos_context mc = { .simple_poll = NULL, .server = NULL,
 
142
                      .dh_bits = 1024, .priority = "SECURE256"
 
143
                      ":!CTYPE-X.509:+CTYPE-OPENPGP" };
 
144
 
 
145
/*
 
146
 * Make additional room in "buffer" for at least BUFFER_SIZE more
 
147
 * bytes. "buffer_capacity" is how much is currently allocated,
 
148
 * "buffer_length" is how much is already used.
 
149
 */
 
150
size_t incbuffer(char **buffer, size_t buffer_length,
 
151
                  size_t buffer_capacity){
 
152
  if(buffer_length + BUFFER_SIZE > buffer_capacity){
 
153
    *buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
 
154
    if(buffer == NULL){
 
155
      return 0;
 
156
    }
 
157
    buffer_capacity += BUFFER_SIZE;
 
158
  }
 
159
  return buffer_capacity;
 
160
}
 
161
 
 
162
/* 
 
163
 * Initialize GPGME.
 
164
 */
 
165
static bool init_gpgme(const char *seckey,
 
166
                       const char *pubkey, const char *tempdir){
 
167
  int ret;
91
168
  gpgme_error_t rc;
92
 
  ssize_t ret;
93
 
  ssize_t new_packet_capacity = 0;
94
 
  ssize_t new_packet_length = 0;
95
169
  gpgme_engine_info_t engine_info;
96
 
 
97
 
  if (debug){
98
 
    fprintf(stderr, "Trying to decrypt OpenPGP packet\n");
 
170
  
 
171
  
 
172
  /*
 
173
   * Helper function to insert pub and seckey to the engine keyring.
 
174
   */
 
175
  bool import_key(const char *filename){
 
176
    int fd;
 
177
    gpgme_data_t pgp_data;
 
178
    
 
179
    fd = (int)TEMP_FAILURE_RETRY(open(filename, O_RDONLY));
 
180
    if(fd == -1){
 
181
      perror("open");
 
182
      return false;
 
183
    }
 
184
    
 
185
    rc = gpgme_data_new_from_fd(&pgp_data, fd);
 
186
    if(rc != GPG_ERR_NO_ERROR){
 
187
      fprintf(stderr, "bad gpgme_data_new_from_fd: %s: %s\n",
 
188
              gpgme_strsource(rc), gpgme_strerror(rc));
 
189
      return false;
 
190
    }
 
191
    
 
192
    rc = gpgme_op_import(mc.ctx, pgp_data);
 
193
    if(rc != GPG_ERR_NO_ERROR){
 
194
      fprintf(stderr, "bad gpgme_op_import: %s: %s\n",
 
195
              gpgme_strsource(rc), gpgme_strerror(rc));
 
196
      return false;
 
197
    }
 
198
    
 
199
    ret = (int)TEMP_FAILURE_RETRY(close(fd));
 
200
    if(ret == -1){
 
201
      perror("close");
 
202
    }
 
203
    gpgme_data_release(pgp_data);
 
204
    return true;
 
205
  }
 
206
  
 
207
  if(debug){
 
208
    fprintf(stderr, "Initializing GPGME\n");
99
209
  }
100
210
  
101
211
  /* Init GPGME */
102
212
  gpgme_check_version(NULL);
103
213
  rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
104
 
  if (rc != GPG_ERR_NO_ERROR){
 
214
  if(rc != GPG_ERR_NO_ERROR){
105
215
    fprintf(stderr, "bad gpgme_engine_check_version: %s: %s\n",
106
216
            gpgme_strsource(rc), gpgme_strerror(rc));
107
 
    return -1;
 
217
    return false;
108
218
  }
109
219
  
110
 
  /* Set GPGME home directory */
111
 
  rc = gpgme_get_engine_info (&engine_info);
112
 
  if (rc != GPG_ERR_NO_ERROR){
 
220
    /* Set GPGME home directory for the OpenPGP engine only */
 
221
  rc = gpgme_get_engine_info(&engine_info);
 
222
  if(rc != GPG_ERR_NO_ERROR){
113
223
    fprintf(stderr, "bad gpgme_get_engine_info: %s: %s\n",
114
224
            gpgme_strsource(rc), gpgme_strerror(rc));
115
 
    return -1;
 
225
    return false;
116
226
  }
117
227
  while(engine_info != NULL){
118
228
    if(engine_info->protocol == GPGME_PROTOCOL_OpenPGP){
119
229
      gpgme_set_engine_info(GPGME_PROTOCOL_OpenPGP,
120
 
                            engine_info->file_name, homedir);
 
230
                            engine_info->file_name, tempdir);
121
231
      break;
122
232
    }
123
233
    engine_info = engine_info->next;
124
234
  }
125
235
  if(engine_info == NULL){
126
 
    fprintf(stderr, "Could not set home dir to %s\n", homedir);
127
 
    return -1;
128
 
  }
129
 
  
130
 
  /* Create new GPGME data buffer from packet buffer */
131
 
  rc = gpgme_data_new_from_mem(&dh_crypto, packet, packet_size, 0);
132
 
  if (rc != GPG_ERR_NO_ERROR){
 
236
    fprintf(stderr, "Could not set GPGME home dir to %s\n", tempdir);
 
237
    return false;
 
238
  }
 
239
  
 
240
  /* Create new GPGME "context" */
 
241
  rc = gpgme_new(&(mc.ctx));
 
242
  if(rc != GPG_ERR_NO_ERROR){
 
243
    fprintf(stderr, "bad gpgme_new: %s: %s\n",
 
244
            gpgme_strsource(rc), gpgme_strerror(rc));
 
245
    return false;
 
246
  }
 
247
  
 
248
  if(not import_key(pubkey) or not import_key(seckey)){
 
249
    return false;
 
250
  }
 
251
  
 
252
  return true; 
 
253
}
 
254
 
 
255
/* 
 
256
 * Decrypt OpenPGP data.
 
257
 * Returns -1 on error
 
258
 */
 
259
static ssize_t pgp_packet_decrypt(const char *cryptotext,
 
260
                                  size_t crypto_size,
 
261
                                  char **plaintext){
 
262
  gpgme_data_t dh_crypto, dh_plain;
 
263
  gpgme_error_t rc;
 
264
  ssize_t ret;
 
265
  size_t plaintext_capacity = 0;
 
266
  ssize_t plaintext_length = 0;
 
267
  
 
268
  if(debug){
 
269
    fprintf(stderr, "Trying to decrypt OpenPGP data\n");
 
270
  }
 
271
  
 
272
  /* Create new GPGME data buffer from memory cryptotext */
 
273
  rc = gpgme_data_new_from_mem(&dh_crypto, cryptotext, crypto_size,
 
274
                               0);
 
275
  if(rc != GPG_ERR_NO_ERROR){
133
276
    fprintf(stderr, "bad gpgme_data_new_from_mem: %s: %s\n",
134
277
            gpgme_strsource(rc), gpgme_strerror(rc));
135
278
    return -1;
137
280
  
138
281
  /* Create new empty GPGME data buffer for the plaintext */
139
282
  rc = gpgme_data_new(&dh_plain);
140
 
  if (rc != GPG_ERR_NO_ERROR){
 
283
  if(rc != GPG_ERR_NO_ERROR){
141
284
    fprintf(stderr, "bad gpgme_data_new: %s: %s\n",
142
285
            gpgme_strsource(rc), gpgme_strerror(rc));
143
 
    return -1;
144
 
  }
145
 
  
146
 
  /* Create new GPGME "context" */
147
 
  rc = gpgme_new(&ctx);
148
 
  if (rc != GPG_ERR_NO_ERROR){
149
 
    fprintf(stderr, "bad gpgme_new: %s: %s\n",
150
 
            gpgme_strsource(rc), gpgme_strerror(rc));
151
 
    return -1;
152
 
  }
153
 
  
154
 
  /* Decrypt data from the FILE pointer to the plaintext data
155
 
     buffer */
156
 
  rc = gpgme_op_decrypt(ctx, dh_crypto, dh_plain);
157
 
  if (rc != GPG_ERR_NO_ERROR){
 
286
    gpgme_data_release(dh_crypto);
 
287
    return -1;
 
288
  }
 
289
  
 
290
  /* Decrypt data from the cryptotext data buffer to the plaintext
 
291
     data buffer */
 
292
  rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);
 
293
  if(rc != GPG_ERR_NO_ERROR){
158
294
    fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
159
295
            gpgme_strsource(rc), gpgme_strerror(rc));
160
 
    return -1;
 
296
    plaintext_length = -1;
 
297
    if(debug){
 
298
      gpgme_decrypt_result_t result;
 
299
      result = gpgme_op_decrypt_result(mc.ctx);
 
300
      if(result == NULL){
 
301
        fprintf(stderr, "gpgme_op_decrypt_result failed\n");
 
302
      } else {
 
303
        fprintf(stderr, "Unsupported algorithm: %s\n",
 
304
                result->unsupported_algorithm);
 
305
        fprintf(stderr, "Wrong key usage: %u\n",
 
306
                result->wrong_key_usage);
 
307
        if(result->file_name != NULL){
 
308
          fprintf(stderr, "File name: %s\n", result->file_name);
 
309
        }
 
310
        gpgme_recipient_t recipient;
 
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
          }
 
322
        }
 
323
      }
 
324
    }
 
325
    goto decrypt_end;
161
326
  }
162
 
 
 
327
  
163
328
  if(debug){
164
 
    fprintf(stderr, "Decryption of OpenPGP packet succeeded\n");
165
 
  }
166
 
 
167
 
  if (debug){
168
 
    gpgme_decrypt_result_t result;
169
 
    result = gpgme_op_decrypt_result(ctx);
170
 
    if (result == NULL){
171
 
      fprintf(stderr, "gpgme_op_decrypt_result failed\n");
172
 
    } else {
173
 
      fprintf(stderr, "Unsupported algorithm: %s\n",
174
 
              result->unsupported_algorithm);
175
 
      fprintf(stderr, "Wrong key usage: %d\n",
176
 
              result->wrong_key_usage);
177
 
      if(result->file_name != NULL){
178
 
        fprintf(stderr, "File name: %s\n", result->file_name);
179
 
      }
180
 
      gpgme_recipient_t recipient;
181
 
      recipient = result->recipients;
182
 
      if(recipient){
183
 
        while(recipient != NULL){
184
 
          fprintf(stderr, "Public key algorithm: %s\n",
185
 
                  gpgme_pubkey_algo_name(recipient->pubkey_algo));
186
 
          fprintf(stderr, "Key ID: %s\n", recipient->keyid);
187
 
          fprintf(stderr, "Secret key available: %s\n",
188
 
                  recipient->status == GPG_ERR_NO_SECKEY
189
 
                  ? "No" : "Yes");
190
 
          recipient = recipient->next;
191
 
        }
192
 
      }
193
 
    }
194
 
  }
195
 
  
196
 
  /* Delete the GPGME FILE pointer cryptotext data buffer */
197
 
  gpgme_data_release(dh_crypto);
 
329
    fprintf(stderr, "Decryption of OpenPGP data succeeded\n");
 
330
  }
198
331
  
199
332
  /* Seek back to the beginning of the GPGME plaintext data buffer */
200
 
  if (gpgme_data_seek(dh_plain, (off_t) 0, SEEK_SET) == -1){
201
 
    perror("pgpme_data_seek");
 
333
  if(gpgme_data_seek(dh_plain, (off_t)0, SEEK_SET) == -1){
 
334
    perror("gpgme_data_seek");
 
335
    plaintext_length = -1;
 
336
    goto decrypt_end;
202
337
  }
203
338
  
204
 
  *new_packet = 0;
 
339
  *plaintext = NULL;
205
340
  while(true){
206
 
    if (new_packet_length + BUFFER_SIZE > new_packet_capacity){
207
 
      *new_packet = realloc(*new_packet,
208
 
                            (unsigned int)new_packet_capacity
209
 
                            + BUFFER_SIZE);
210
 
      if (*new_packet == NULL){
211
 
        perror("realloc");
212
 
        return -1;
213
 
      }
214
 
      new_packet_capacity += BUFFER_SIZE;
 
341
    plaintext_capacity = incbuffer(plaintext,
 
342
                                      (size_t)plaintext_length,
 
343
                                      plaintext_capacity);
 
344
    if(plaintext_capacity == 0){
 
345
        perror("incbuffer");
 
346
        plaintext_length = -1;
 
347
        goto decrypt_end;
215
348
    }
216
349
    
217
 
    ret = gpgme_data_read(dh_plain, *new_packet + new_packet_length,
 
350
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
218
351
                          BUFFER_SIZE);
219
352
    /* Print the data, if any */
220
 
    if (ret == 0){
 
353
    if(ret == 0){
 
354
      /* EOF */
221
355
      break;
222
356
    }
223
357
    if(ret < 0){
224
358
      perror("gpgme_data_read");
225
 
      return -1;
226
 
    }
227
 
    new_packet_length += ret;
228
 
  }
229
 
 
230
 
  /* FIXME: check characters before printing to screen so to not print
231
 
     terminal control characters */
232
 
  /*   if(debug){ */
233
 
  /*     fprintf(stderr, "decrypted password is: "); */
234
 
  /*     fwrite(*new_packet, 1, new_packet_length, stderr); */
235
 
  /*     fprintf(stderr, "\n"); */
236
 
  /*   } */
 
359
      plaintext_length = -1;
 
360
      goto decrypt_end;
 
361
    }
 
362
    plaintext_length += ret;
 
363
  }
 
364
  
 
365
  if(debug){
 
366
    fprintf(stderr, "Decrypted password is: ");
 
367
    for(ssize_t i = 0; i < plaintext_length; i++){
 
368
      fprintf(stderr, "%02hhX ", (*plaintext)[i]);
 
369
    }
 
370
    fprintf(stderr, "\n");
 
371
  }
 
372
  
 
373
 decrypt_end:
 
374
  
 
375
  /* Delete the GPGME cryptotext data buffer */
 
376
  gpgme_data_release(dh_crypto);
237
377
  
238
378
  /* Delete the GPGME plaintext data buffer */
239
379
  gpgme_data_release(dh_plain);
240
 
  return new_packet_length;
 
380
  return plaintext_length;
241
381
}
242
382
 
243
 
static const char * safer_gnutls_strerror (int value) {
244
 
  const char *ret = gnutls_strerror (value);
245
 
  if (ret == NULL)
 
383
static const char * safer_gnutls_strerror(int value){
 
384
  const char *ret = gnutls_strerror(value); /* Spurious warning from
 
385
                                               -Wunreachable-code */
 
386
  if(ret == NULL)
246
387
    ret = "(unknown)";
247
388
  return ret;
248
389
}
249
390
 
250
 
void debuggnutls(__attribute__((unused)) int level,
251
 
                 const char* string){
252
 
  fprintf(stderr, "%s", string);
 
391
/* GnuTLS log function callback */
 
392
static void debuggnutls(__attribute__((unused)) int level,
 
393
                        const char* string){
 
394
  fprintf(stderr, "GnuTLS: %s", string);
253
395
}
254
396
 
255
 
int initgnutls(encrypted_session *es){
256
 
  const char *err;
 
397
static int init_gnutls_global(const char *pubkeyfilename,
 
398
                              const char *seckeyfilename){
257
399
  int ret;
258
400
  
259
401
  if(debug){
260
402
    fprintf(stderr, "Initializing GnuTLS\n");
261
403
  }
262
 
 
263
 
  if ((ret = gnutls_global_init ())
264
 
      != GNUTLS_E_SUCCESS) {
265
 
    fprintf (stderr, "global_init: %s\n", safer_gnutls_strerror(ret));
 
404
  
 
405
  ret = gnutls_global_init();
 
406
  if(ret != GNUTLS_E_SUCCESS){
 
407
    fprintf(stderr, "GnuTLS global_init: %s\n",
 
408
            safer_gnutls_strerror(ret));
266
409
    return -1;
267
410
  }
268
 
 
269
 
  if (debug){
 
411
  
 
412
  if(debug){
 
413
    /* "Use a log level over 10 to enable all debugging options."
 
414
     * - GnuTLS manual
 
415
     */
270
416
    gnutls_global_set_log_level(11);
271
417
    gnutls_global_set_log_function(debuggnutls);
272
418
  }
273
419
  
274
 
  /* openpgp credentials */
275
 
  if ((ret = gnutls_certificate_allocate_credentials (&es->cred))
276
 
      != GNUTLS_E_SUCCESS) {
277
 
    fprintf (stderr, "memory error: %s\n",
278
 
             safer_gnutls_strerror(ret));
 
420
  /* OpenPGP credentials */
 
421
  gnutls_certificate_allocate_credentials(&mc.cred);
 
422
  if(ret != GNUTLS_E_SUCCESS){
 
423
    fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning
 
424
                                                    from
 
425
                                                    -Wunreachable-code
 
426
                                                 */
 
427
            safer_gnutls_strerror(ret));
 
428
    gnutls_global_deinit();
279
429
    return -1;
280
430
  }
281
431
  
282
432
  if(debug){
283
 
    fprintf(stderr, "Attempting to use OpenPGP certificate %s"
284
 
            " and keyfile %s as GnuTLS credentials\n", certfile,
285
 
            certkey);
 
433
    fprintf(stderr, "Attempting to use OpenPGP public key %s and"
 
434
            " secret key %s as GnuTLS credentials\n", pubkeyfilename,
 
435
            seckeyfilename);
286
436
  }
287
437
  
288
438
  ret = gnutls_certificate_set_openpgp_key_file
289
 
    (es->cred, certfile, certkey, GNUTLS_OPENPGP_FMT_BASE64);
290
 
  if (ret != GNUTLS_E_SUCCESS) {
291
 
    fprintf
292
 
      (stderr, "Error[%d] while reading the OpenPGP key pair ('%s',"
293
 
       " '%s')\n",
294
 
       ret, certfile, certkey);
295
 
    fprintf(stdout, "The Error is: %s\n",
296
 
            safer_gnutls_strerror(ret));
297
 
    return -1;
298
 
  }
299
 
  
300
 
  //GnuTLS server initialization
301
 
  if ((ret = gnutls_dh_params_init (&es->dh_params))
302
 
      != GNUTLS_E_SUCCESS) {
303
 
    fprintf (stderr, "Error in dh parameter initialization: %s\n",
304
 
             safer_gnutls_strerror(ret));
305
 
    return -1;
306
 
  }
307
 
  
308
 
  if ((ret = gnutls_dh_params_generate2 (es->dh_params, DH_BITS))
309
 
      != GNUTLS_E_SUCCESS) {
310
 
    fprintf (stderr, "Error in prime generation: %s\n",
311
 
             safer_gnutls_strerror(ret));
312
 
    return -1;
313
 
  }
314
 
  
315
 
  gnutls_certificate_set_dh_params (es->cred, es->dh_params);
316
 
  
317
 
  // GnuTLS session creation
318
 
  if ((ret = gnutls_init (&es->session, GNUTLS_SERVER))
319
 
      != GNUTLS_E_SUCCESS){
 
439
    (mc.cred, pubkeyfilename, seckeyfilename,
 
440
     GNUTLS_OPENPGP_FMT_BASE64);
 
441
  if(ret != GNUTLS_E_SUCCESS){
 
442
    fprintf(stderr,
 
443
            "Error[%d] while reading the OpenPGP key pair ('%s',"
 
444
            " '%s')\n", ret, pubkeyfilename, seckeyfilename);
 
445
    fprintf(stderr, "The GnuTLS error is: %s\n",
 
446
            safer_gnutls_strerror(ret));
 
447
    goto globalfail;
 
448
  }
 
449
  
 
450
  /* GnuTLS server initialization */
 
451
  ret = gnutls_dh_params_init(&mc.dh_params);
 
452
  if(ret != GNUTLS_E_SUCCESS){
 
453
    fprintf(stderr, "Error in GnuTLS DH parameter initialization:"
 
454
            " %s\n", safer_gnutls_strerror(ret));
 
455
    goto globalfail;
 
456
  }
 
457
  ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);
 
458
  if(ret != GNUTLS_E_SUCCESS){
 
459
    fprintf(stderr, "Error in GnuTLS prime generation: %s\n",
 
460
            safer_gnutls_strerror(ret));
 
461
    goto globalfail;
 
462
  }
 
463
  
 
464
  gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);
 
465
  
 
466
  return 0;
 
467
  
 
468
 globalfail:
 
469
  
 
470
  gnutls_certificate_free_credentials(mc.cred);
 
471
  gnutls_global_deinit();
 
472
  gnutls_dh_params_deinit(mc.dh_params);
 
473
  return -1;
 
474
}
 
475
 
 
476
static int init_gnutls_session(gnutls_session_t *session){
 
477
  int ret;
 
478
  /* GnuTLS session creation */
 
479
  ret = gnutls_init(session, GNUTLS_SERVER);
 
480
  if(ret != GNUTLS_E_SUCCESS){
320
481
    fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
321
482
            safer_gnutls_strerror(ret));
322
483
  }
323
484
  
324
 
  if ((ret = gnutls_priority_set_direct (es->session, "NORMAL", &err))
325
 
      != GNUTLS_E_SUCCESS) {
326
 
    fprintf(stderr, "Syntax error at: %s\n", err);
327
 
    fprintf(stderr, "GnuTLS error: %s\n",
328
 
            safer_gnutls_strerror(ret));
329
 
    return -1;
 
485
  {
 
486
    const char *err;
 
487
    ret = gnutls_priority_set_direct(*session, mc.priority, &err);
 
488
    if(ret != GNUTLS_E_SUCCESS){
 
489
      fprintf(stderr, "Syntax error at: %s\n", err);
 
490
      fprintf(stderr, "GnuTLS error: %s\n",
 
491
              safer_gnutls_strerror(ret));
 
492
      gnutls_deinit(*session);
 
493
      return -1;
 
494
    }
330
495
  }
331
496
  
332
 
  if ((ret = gnutls_credentials_set
333
 
       (es->session, GNUTLS_CRD_CERTIFICATE, es->cred))
334
 
      != GNUTLS_E_SUCCESS) {
335
 
    fprintf(stderr, "Error setting a credentials set: %s\n",
 
497
  ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
 
498
                               mc.cred);
 
499
  if(ret != GNUTLS_E_SUCCESS){
 
500
    fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
336
501
            safer_gnutls_strerror(ret));
 
502
    gnutls_deinit(*session);
337
503
    return -1;
338
504
  }
339
505
  
340
506
  /* ignore client certificate if any. */
341
 
  gnutls_certificate_server_set_request (es->session,
342
 
                                         GNUTLS_CERT_IGNORE);
 
507
  gnutls_certificate_server_set_request(*session,
 
508
                                        GNUTLS_CERT_IGNORE);
343
509
  
344
 
  gnutls_dh_set_prime_bits (es->session, DH_BITS);
 
510
  gnutls_dh_set_prime_bits(*session, mc.dh_bits);
345
511
  
346
512
  return 0;
347
513
}
348
514
 
349
 
void empty_log(__attribute__((unused)) AvahiLogLevel level,
350
 
               __attribute__((unused)) const char *txt){}
 
515
/* Avahi log function callback */
 
516
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
 
517
                      __attribute__((unused)) const char *txt){}
351
518
 
352
 
int start_mandos_communication(const char *ip, uint16_t port,
353
 
                               unsigned int if_index){
 
519
/* Called when a Mandos server is found */
 
520
static int start_mandos_communication(const char *ip, uint16_t port,
 
521
                                      AvahiIfIndex if_index,
 
522
                                      int af){
354
523
  int ret, tcp_sd;
355
 
  struct sockaddr_in6 to;
356
 
  encrypted_session es;
 
524
  ssize_t sret;
 
525
  union {
 
526
    struct sockaddr_in in;
 
527
    struct sockaddr_in6 in6;
 
528
  } to;
357
529
  char *buffer = NULL;
358
530
  char *decrypted_buffer;
359
531
  size_t buffer_length = 0;
360
532
  size_t buffer_capacity = 0;
361
533
  ssize_t decrypted_buffer_size;
362
 
  size_t written = 0;
 
534
  size_t written;
363
535
  int retval = 0;
364
 
  char interface[IF_NAMESIZE];
 
536
  gnutls_session_t session;
 
537
  int pf;                       /* Protocol family */
 
538
  
 
539
  switch(af){
 
540
  case AF_INET6:
 
541
    pf = PF_INET6;
 
542
    break;
 
543
  case AF_INET:
 
544
    pf = PF_INET;
 
545
    break;
 
546
  default:
 
547
    fprintf(stderr, "Bad address family: %d\n", af);
 
548
    return -1;
 
549
  }
 
550
  
 
551
  ret = init_gnutls_session(&session);
 
552
  if(ret != 0){
 
553
    return -1;
 
554
  }
365
555
  
366
556
  if(debug){
367
 
    fprintf(stderr, "Setting up a tcp connection to %s\n", ip);
 
557
    fprintf(stderr, "Setting up a TCP connection to %s, port %" PRIu16
 
558
            "\n", ip, port);
368
559
  }
369
560
  
370
 
  tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
371
 
  if(tcp_sd < 0) {
 
561
  tcp_sd = socket(pf, SOCK_STREAM, 0);
 
562
  if(tcp_sd < 0){
372
563
    perror("socket");
373
564
    return -1;
374
565
  }
375
566
  
376
 
  if(if_indextoname(if_index, interface) == NULL){
377
 
    if(debug){
378
 
      perror("if_indextoname");
379
 
    }
380
 
    return -1;
381
 
  }
382
 
  
383
 
  if(debug){
384
 
    fprintf(stderr, "Binding to interface %s\n", interface);
385
 
  }
386
 
  
387
 
  memset(&to,0,sizeof(to));     /* Spurious warning */
388
 
  to.sin6_family = AF_INET6;
389
 
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
390
 
  if (ret < 0 ){
 
567
  memset(&to, 0, sizeof(to));
 
568
  if(af == AF_INET6){
 
569
    to.in6.sin6_family = (sa_family_t)af;
 
570
    ret = inet_pton(af, ip, &to.in6.sin6_addr);
 
571
  } else {                      /* IPv4 */
 
572
    to.in.sin_family = (sa_family_t)af;
 
573
    ret = inet_pton(af, ip, &to.in.sin_addr);
 
574
  }
 
575
  if(ret < 0 ){
391
576
    perror("inet_pton");
392
577
    return -1;
393
 
  }  
 
578
  }
394
579
  if(ret == 0){
395
580
    fprintf(stderr, "Bad address: %s\n", ip);
396
581
    return -1;
397
582
  }
398
 
  to.sin6_port = htons(port);   /* Spurious warning */
399
 
  
400
 
  to.sin6_scope_id = (uint32_t)if_index;
 
583
  if(af == AF_INET6){
 
584
    to.in6.sin6_port = htons(port); /* Spurious warnings from
 
585
                                       -Wconversion and
 
586
                                       -Wunreachable-code */
 
587
    
 
588
    if(IN6_IS_ADDR_LINKLOCAL /* Spurious warnings from */
 
589
       (&to.in6.sin6_addr)){ /* -Wstrict-aliasing=2 or lower and
 
590
                              -Wunreachable-code*/
 
591
      if(if_index == AVAHI_IF_UNSPEC){
 
592
        fprintf(stderr, "An IPv6 link-local address is incomplete"
 
593
                " without a network interface\n");
 
594
        return -1;
 
595
      }
 
596
      /* Set the network interface number as scope */
 
597
      to.in6.sin6_scope_id = (uint32_t)if_index;
 
598
    }
 
599
  } else {
 
600
    to.in.sin_port = htons(port); /* Spurious warnings from
 
601
                                     -Wconversion and
 
602
                                     -Wunreachable-code */
 
603
  }
401
604
  
402
605
  if(debug){
403
 
    fprintf(stderr, "Connection to: %s\n", ip);
 
606
    if(af == AF_INET6 and if_index != AVAHI_IF_UNSPEC){
 
607
      char interface[IF_NAMESIZE];
 
608
      if(if_indextoname((unsigned int)if_index, interface) == NULL){
 
609
        perror("if_indextoname");
 
610
      } else {
 
611
        fprintf(stderr, "Connection to: %s%%%s, port %" PRIu16 "\n",
 
612
                ip, interface, port);
 
613
      }
 
614
    } else {
 
615
      fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
 
616
              port);
 
617
    }
 
618
    char addrstr[(INET_ADDRSTRLEN > INET6_ADDRSTRLEN) ?
 
619
                 INET_ADDRSTRLEN : INET6_ADDRSTRLEN] = "";
 
620
    const char *pcret;
 
621
    if(af == AF_INET6){
 
622
      pcret = inet_ntop(af, &(to.in6.sin6_addr), addrstr,
 
623
                        sizeof(addrstr));
 
624
    } else {
 
625
      pcret = inet_ntop(af, &(to.in.sin_addr), addrstr,
 
626
                        sizeof(addrstr));
 
627
    }
 
628
    if(pcret == NULL){
 
629
      perror("inet_ntop");
 
630
    } else {
 
631
      if(strcmp(addrstr, ip) != 0){
 
632
        fprintf(stderr, "Canonical address form: %s\n", addrstr);
 
633
      }
 
634
    }
404
635
  }
405
636
  
406
 
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
407
 
  if (ret < 0){
 
637
  if(af == AF_INET6){
 
638
    ret = connect(tcp_sd, &to.in6, sizeof(to));
 
639
  } else {
 
640
    ret = connect(tcp_sd, &to.in, sizeof(to)); /* IPv4 */
 
641
  }
 
642
  if(ret < 0){
408
643
    perror("connect");
409
644
    return -1;
410
645
  }
411
646
  
412
 
  ret = initgnutls (&es);
413
 
  if (ret != 0){
414
 
    retval = -1;
415
 
    return -1;
 
647
  const char *out = mandos_protocol_version;
 
648
  written = 0;
 
649
  while(true){
 
650
    size_t out_size = strlen(out);
 
651
    ret = (int)TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
 
652
                                   out_size - written));
 
653
    if(ret == -1){
 
654
      perror("write");
 
655
      retval = -1;
 
656
      goto mandos_end;
 
657
    }
 
658
    written += (size_t)ret;
 
659
    if(written < out_size){
 
660
      continue;
 
661
    } else {
 
662
      if(out == mandos_protocol_version){
 
663
        written = 0;
 
664
        out = "\r\n";
 
665
      } else {
 
666
        break;
 
667
      }
 
668
    }
416
669
  }
417
670
  
418
 
  gnutls_transport_set_ptr (es.session,
419
 
                            (gnutls_transport_ptr_t) tcp_sd);
420
 
  
421
671
  if(debug){
422
672
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
423
673
  }
424
674
  
425
 
  ret = gnutls_handshake (es.session);
426
 
  
427
 
  if (ret != GNUTLS_E_SUCCESS){
 
675
  gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);
 
676
  
 
677
  do{
 
678
    ret = gnutls_handshake(session);
 
679
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
680
  
 
681
  if(ret != GNUTLS_E_SUCCESS){
428
682
    if(debug){
429
 
      fprintf(stderr, "\n*** Handshake failed ***\n");
430
 
      gnutls_perror (ret);
 
683
      fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
 
684
      gnutls_perror(ret);
431
685
    }
432
686
    retval = -1;
433
 
    goto exit;
 
687
    goto mandos_end;
434
688
  }
435
689
  
436
 
  //Retrieve OpenPGP packet that contains the wanted password
 
690
  /* Read OpenPGP packet that contains the wanted password */
437
691
  
438
692
  if(debug){
439
 
    fprintf(stderr, "Retrieving pgp encrypted password from %s\n",
 
693
    fprintf(stderr, "Retrieving OpenPGP encrypted password from %s\n",
440
694
            ip);
441
695
  }
442
 
 
 
696
  
443
697
  while(true){
444
 
    if (buffer_length + BUFFER_SIZE > buffer_capacity){
445
 
      buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
446
 
      if (buffer == NULL){
447
 
        perror("realloc");
448
 
        goto exit;
449
 
      }
450
 
      buffer_capacity += BUFFER_SIZE;
 
698
    buffer_capacity = incbuffer(&buffer, buffer_length,
 
699
                                   buffer_capacity);
 
700
    if(buffer_capacity == 0){
 
701
      perror("incbuffer");
 
702
      retval = -1;
 
703
      goto mandos_end;
451
704
    }
452
705
    
453
 
    ret = gnutls_record_recv
454
 
      (es.session, buffer+buffer_length, BUFFER_SIZE);
455
 
    if (ret == 0){
 
706
    sret = gnutls_record_recv(session, buffer+buffer_length,
 
707
                              BUFFER_SIZE);
 
708
    if(sret == 0){
456
709
      break;
457
710
    }
458
 
    if (ret < 0){
459
 
      switch(ret){
 
711
    if(sret < 0){
 
712
      switch(sret){
460
713
      case GNUTLS_E_INTERRUPTED:
461
714
      case GNUTLS_E_AGAIN:
462
715
        break;
463
716
      case GNUTLS_E_REHANDSHAKE:
464
 
        ret = gnutls_handshake (es.session);
465
 
        if (ret < 0){
466
 
          fprintf(stderr, "\n*** Handshake failed ***\n");
467
 
          gnutls_perror (ret);
 
717
        do{
 
718
          ret = gnutls_handshake(session);
 
719
        } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
720
        if(ret < 0){
 
721
          fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
 
722
          gnutls_perror(ret);
468
723
          retval = -1;
469
 
          goto exit;
 
724
          goto mandos_end;
470
725
        }
471
726
        break;
472
727
      default:
473
728
        fprintf(stderr, "Unknown error while reading data from"
474
 
                " encrypted session with mandos server\n");
 
729
                " encrypted session with Mandos server\n");
475
730
        retval = -1;
476
 
        gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
477
 
        goto exit;
 
731
        gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
732
        goto mandos_end;
478
733
      }
479
734
    } else {
480
 
      buffer_length += (size_t) ret;
 
735
      buffer_length += (size_t) sret;
481
736
    }
482
737
  }
483
738
  
484
 
  if (buffer_length > 0){
 
739
  if(debug){
 
740
    fprintf(stderr, "Closing TLS session\n");
 
741
  }
 
742
  
 
743
  gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
744
  
 
745
  if(buffer_length > 0){
485
746
    decrypted_buffer_size = pgp_packet_decrypt(buffer,
486
747
                                               buffer_length,
487
 
                                               &decrypted_buffer,
488
 
                                               certdir);
489
 
    if (decrypted_buffer_size >= 0){
490
 
      while(written < decrypted_buffer_size){
491
 
        ret = (int)fwrite (decrypted_buffer + written, 1,
492
 
                           (size_t)decrypted_buffer_size - written,
493
 
                           stdout);
 
748
                                               &decrypted_buffer);
 
749
    if(decrypted_buffer_size >= 0){
 
750
      written = 0;
 
751
      while(written < (size_t) decrypted_buffer_size){
 
752
        ret = (int)fwrite(decrypted_buffer + written, 1,
 
753
                          (size_t)decrypted_buffer_size - written,
 
754
                          stdout);
494
755
        if(ret == 0 and ferror(stdout)){
495
756
          if(debug){
496
757
            fprintf(stderr, "Error writing encrypted data: %s\n",
505
766
    } else {
506
767
      retval = -1;
507
768
    }
508
 
  }
509
 
 
510
 
  //shutdown procedure
511
 
 
512
 
  if(debug){
513
 
    fprintf(stderr, "Closing TLS session\n");
514
 
  }
515
 
 
 
769
  } else {
 
770
    retval = -1;
 
771
  }
 
772
  
 
773
  /* Shutdown procedure */
 
774
  
 
775
 mandos_end:
516
776
  free(buffer);
517
 
  gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
518
 
 exit:
519
 
  close(tcp_sd);
520
 
  gnutls_deinit (es.session);
521
 
  gnutls_certificate_free_credentials (es.cred);
522
 
  gnutls_global_deinit ();
 
777
  ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
 
778
  if(ret == -1){
 
779
    perror("close");
 
780
  }
 
781
  gnutls_deinit(session);
523
782
  return retval;
524
783
}
525
784
 
526
 
static AvahiSimplePoll *simple_poll = NULL;
527
 
static AvahiServer *server = NULL;
528
 
 
529
 
static void resolve_callback(
530
 
    AvahiSServiceResolver *r,
531
 
    AvahiIfIndex interface,
532
 
    AVAHI_GCC_UNUSED AvahiProtocol protocol,
533
 
    AvahiResolverEvent event,
534
 
    const char *name,
535
 
    const char *type,
536
 
    const char *domain,
537
 
    const char *host_name,
538
 
    const AvahiAddress *address,
539
 
    uint16_t port,
540
 
    AVAHI_GCC_UNUSED AvahiStringList *txt,
541
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
542
 
    AVAHI_GCC_UNUSED void* userdata) {
543
 
    
544
 
  assert(r);                    /* Spurious warning */
 
785
static void resolve_callback(AvahiSServiceResolver *r,
 
786
                             AvahiIfIndex interface,
 
787
                             AvahiProtocol proto,
 
788
                             AvahiResolverEvent event,
 
789
                             const char *name,
 
790
                             const char *type,
 
791
                             const char *domain,
 
792
                             const char *host_name,
 
793
                             const AvahiAddress *address,
 
794
                             uint16_t port,
 
795
                             AVAHI_GCC_UNUSED AvahiStringList *txt,
 
796
                             AVAHI_GCC_UNUSED AvahiLookupResultFlags
 
797
                             flags,
 
798
                             AVAHI_GCC_UNUSED void* userdata){
 
799
  assert(r);
545
800
  
546
801
  /* Called whenever a service has been resolved successfully or
547
802
     timed out */
548
803
  
549
 
  switch (event) {
 
804
  switch(event){
550
805
  default:
551
806
  case AVAHI_RESOLVER_FAILURE:
552
 
    fprintf(stderr, "(Resolver) Failed to resolve service '%s' of"
553
 
            " type '%s' in domain '%s': %s\n", name, type, domain,
554
 
            avahi_strerror(avahi_server_errno(server)));
 
807
    fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
 
808
            " of type '%s' in domain '%s': %s\n", name, type, domain,
 
809
            avahi_strerror(avahi_server_errno(mc.server)));
555
810
    break;
556
811
    
557
812
  case AVAHI_RESOLVER_FOUND:
559
814
      char ip[AVAHI_ADDRESS_STR_MAX];
560
815
      avahi_address_snprint(ip, sizeof(ip), address);
561
816
      if(debug){
562
 
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s) on"
563
 
                " port %d\n", name, host_name, ip, port);
 
817
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
 
818
                PRIdMAX ") on port %" PRIu16 "\n", name, host_name,
 
819
                ip, (intmax_t)interface, port);
564
820
      }
565
 
      int ret = start_mandos_communication(ip, port,
566
 
                                           (unsigned int) interface);
567
 
      if (ret == 0){
568
 
        exit(EXIT_SUCCESS);
 
821
      int ret = start_mandos_communication(ip, port, interface,
 
822
                                           avahi_proto_to_af(proto));
 
823
      if(ret == 0){
 
824
        avahi_simple_poll_quit(mc.simple_poll);
569
825
      }
570
826
    }
571
827
  }
572
828
  avahi_s_service_resolver_free(r);
573
829
}
574
830
 
575
 
static void browse_callback(
576
 
    AvahiSServiceBrowser *b,
577
 
    AvahiIfIndex interface,
578
 
    AvahiProtocol protocol,
579
 
    AvahiBrowserEvent event,
580
 
    const char *name,
581
 
    const char *type,
582
 
    const char *domain,
583
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
584
 
    void* userdata) {
585
 
    
586
 
    AvahiServer *s = userdata;
587
 
    assert(b);                  /* Spurious warning */
588
 
    
589
 
    /* Called whenever a new services becomes available on the LAN or
590
 
       is removed from the LAN */
591
 
    
592
 
    switch (event) {
593
 
    default:
594
 
    case AVAHI_BROWSER_FAILURE:
595
 
      
596
 
      fprintf(stderr, "(Browser) %s\n",
597
 
              avahi_strerror(avahi_server_errno(server)));
598
 
      avahi_simple_poll_quit(simple_poll);
599
 
      return;
600
 
      
601
 
    case AVAHI_BROWSER_NEW:
602
 
      /* We ignore the returned resolver object. In the callback
603
 
         function we free it. If the server is terminated before
604
 
         the callback function is called the server will free
605
 
         the resolver for us. */
606
 
      
607
 
      if (!(avahi_s_service_resolver_new(s, interface, protocol, name,
608
 
                                         type, domain,
609
 
                                         AVAHI_PROTO_INET6, 0,
610
 
                                         resolve_callback, s)))
611
 
        fprintf(stderr, "Failed to resolve service '%s': %s\n", name,
612
 
                avahi_strerror(avahi_server_errno(s)));
613
 
      break;
614
 
      
615
 
    case AVAHI_BROWSER_REMOVE:
616
 
      break;
617
 
      
618
 
    case AVAHI_BROWSER_ALL_FOR_NOW:
619
 
    case AVAHI_BROWSER_CACHE_EXHAUSTED:
620
 
      break;
 
831
static void browse_callback(AvahiSServiceBrowser *b,
 
832
                            AvahiIfIndex interface,
 
833
                            AvahiProtocol protocol,
 
834
                            AvahiBrowserEvent event,
 
835
                            const char *name,
 
836
                            const char *type,
 
837
                            const char *domain,
 
838
                            AVAHI_GCC_UNUSED AvahiLookupResultFlags
 
839
                            flags,
 
840
                            AVAHI_GCC_UNUSED void* userdata){
 
841
  assert(b);
 
842
  
 
843
  /* Called whenever a new services becomes available on the LAN or
 
844
     is removed from the LAN */
 
845
  
 
846
  switch(event){
 
847
  default:
 
848
  case AVAHI_BROWSER_FAILURE:
 
849
    
 
850
    fprintf(stderr, "(Avahi browser) %s\n",
 
851
            avahi_strerror(avahi_server_errno(mc.server)));
 
852
    avahi_simple_poll_quit(mc.simple_poll);
 
853
    return;
 
854
    
 
855
  case AVAHI_BROWSER_NEW:
 
856
    /* We ignore the returned Avahi resolver object. In the callback
 
857
       function we free it. If the Avahi server is terminated before
 
858
       the callback function is called the Avahi server will free the
 
859
       resolver for us. */
 
860
    
 
861
    if(avahi_s_service_resolver_new(mc.server, interface, protocol,
 
862
                                    name, type, domain, protocol, 0,
 
863
                                    resolve_callback, NULL) == NULL)
 
864
      fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
 
865
              name, avahi_strerror(avahi_server_errno(mc.server)));
 
866
    break;
 
867
    
 
868
  case AVAHI_BROWSER_REMOVE:
 
869
    break;
 
870
    
 
871
  case AVAHI_BROWSER_ALL_FOR_NOW:
 
872
  case AVAHI_BROWSER_CACHE_EXHAUSTED:
 
873
    if(debug){
 
874
      fprintf(stderr, "No Mandos server found, still searching...\n");
621
875
    }
622
 
}
623
 
 
624
 
/* combinds file name and path and returns the malloced new string. som sane checks could/should be added */
625
 
const char *combinepath(const char *first, const char *second){
 
876
    break;
 
877
  }
 
878
}
 
879
 
 
880
sig_atomic_t quit_now = 0;
 
881
 
 
882
/* stop main loop after sigterm has been called */
 
883
static void handle_sigterm(__attribute__((unused)) int sig){
 
884
  if(quit_now){
 
885
    return;
 
886
  }
 
887
  quit_now = 1;
 
888
  int old_errno = errno;
 
889
  if(mc.simple_poll != NULL){
 
890
    avahi_simple_poll_quit(mc.simple_poll);
 
891
  }
 
892
  errno = old_errno;
 
893
}
 
894
 
 
895
int main(int argc, char *argv[]){
 
896
  AvahiSServiceBrowser *sb = NULL;
 
897
  int error;
 
898
  int ret;
 
899
  intmax_t tmpmax;
626
900
  char *tmp;
627
 
  tmp = malloc(strlen(first) + strlen(second) + 2);
628
 
  if (tmp == NULL){
629
 
    perror("malloc");
630
 
    return NULL;
631
 
  }
632
 
  strcpy(tmp, first);
633
 
  if (first[0] != '\0' and first[strlen(first) - 1] != '/'){
634
 
    strcat(tmp, "/");
635
 
  }
636
 
  strcat(tmp, second);
637
 
  return tmp;
638
 
}
639
 
 
640
 
 
641
 
int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char*argv[]) {
 
901
  int exitcode = EXIT_SUCCESS;
 
902
  const char *interface = "eth0";
 
903
  struct ifreq network;
 
904
  int sd;
 
905
  uid_t uid;
 
906
  gid_t gid;
 
907
  char *connect_to = NULL;
 
908
  char tempdir[] = "/tmp/mandosXXXXXX";
 
909
  bool tempdir_created = false;
 
910
  AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
 
911
  const char *seckey = PATHDIR "/" SECKEY;
 
912
  const char *pubkey = PATHDIR "/" PUBKEY;
 
913
  
 
914
  bool gnutls_initialized = false;
 
915
  bool gpgme_initialized = false;
 
916
  float delay = 2.5f;
 
917
  
 
918
  struct sigaction old_sigterm_action;
 
919
  struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
 
920
  
 
921
  {
 
922
    struct argp_option options[] = {
 
923
      { .name = "debug", .key = 128,
 
924
        .doc = "Debug mode", .group = 3 },
 
925
      { .name = "connect", .key = 'c',
 
926
        .arg = "ADDRESS:PORT",
 
927
        .doc = "Connect directly to a specific Mandos server",
 
928
        .group = 1 },
 
929
      { .name = "interface", .key = 'i',
 
930
        .arg = "NAME",
 
931
        .doc = "Network interface that will be used to search for"
 
932
        " Mandos servers",
 
933
        .group = 1 },
 
934
      { .name = "seckey", .key = 's',
 
935
        .arg = "FILE",
 
936
        .doc = "OpenPGP secret key file base name",
 
937
        .group = 1 },
 
938
      { .name = "pubkey", .key = 'p',
 
939
        .arg = "FILE",
 
940
        .doc = "OpenPGP public key file base name",
 
941
        .group = 2 },
 
942
      { .name = "dh-bits", .key = 129,
 
943
        .arg = "BITS",
 
944
        .doc = "Bit length of the prime number used in the"
 
945
        " Diffie-Hellman key exchange",
 
946
        .group = 2 },
 
947
      { .name = "priority", .key = 130,
 
948
        .arg = "STRING",
 
949
        .doc = "GnuTLS priority string for the TLS handshake",
 
950
        .group = 1 },
 
951
      { .name = "delay", .key = 131,
 
952
        .arg = "SECONDS",
 
953
        .doc = "Maximum delay to wait for interface startup",
 
954
        .group = 2 },
 
955
      { .name = NULL }
 
956
    };
 
957
    
 
958
    error_t parse_opt(int key, char *arg,
 
959
                      struct argp_state *state){
 
960
      switch(key){
 
961
      case 128:                 /* --debug */
 
962
        debug = true;
 
963
        break;
 
964
      case 'c':                 /* --connect */
 
965
        connect_to = arg;
 
966
        break;
 
967
      case 'i':                 /* --interface */
 
968
        interface = arg;
 
969
        break;
 
970
      case 's':                 /* --seckey */
 
971
        seckey = arg;
 
972
        break;
 
973
      case 'p':                 /* --pubkey */
 
974
        pubkey = arg;
 
975
        break;
 
976
      case 129:                 /* --dh-bits */
 
977
        errno = 0;
 
978
        tmpmax = strtoimax(arg, &tmp, 10);
 
979
        if(errno != 0 or tmp == arg or *tmp != '\0'
 
980
           or tmpmax != (typeof(mc.dh_bits))tmpmax){
 
981
          fprintf(stderr, "Bad number of DH bits\n");
 
982
          exit(EXIT_FAILURE);
 
983
        }
 
984
        mc.dh_bits = (typeof(mc.dh_bits))tmpmax;
 
985
        break;
 
986
      case 130:                 /* --priority */
 
987
        mc.priority = arg;
 
988
        break;
 
989
      case 131:                 /* --delay */
 
990
        errno = 0;
 
991
        delay = strtof(arg, &tmp);
 
992
        if(errno != 0 or tmp == arg or *tmp != '\0'){
 
993
          fprintf(stderr, "Bad delay\n");
 
994
          exit(EXIT_FAILURE);
 
995
        }
 
996
        break;
 
997
      case ARGP_KEY_ARG:
 
998
        argp_usage(state);
 
999
      case ARGP_KEY_END:
 
1000
        break;
 
1001
      default:
 
1002
        return ARGP_ERR_UNKNOWN;
 
1003
      }
 
1004
      return 0;
 
1005
    }
 
1006
    
 
1007
    struct argp argp = { .options = options, .parser = parse_opt,
 
1008
                         .args_doc = "",
 
1009
                         .doc = "Mandos client -- Get and decrypt"
 
1010
                         " passwords from a Mandos server" };
 
1011
    ret = argp_parse(&argp, argc, argv, 0, 0, NULL);
 
1012
    if(ret == ARGP_ERR_UNKNOWN){
 
1013
      fprintf(stderr, "Unknown error while parsing arguments\n");
 
1014
      exitcode = EXIT_FAILURE;
 
1015
      goto end;
 
1016
    }
 
1017
  }
 
1018
  
 
1019
  if(not debug){
 
1020
    avahi_set_log_function(empty_log);
 
1021
  }
 
1022
  
 
1023
  /* Initialize Avahi early so avahi_simple_poll_quit() can be called
 
1024
     from the signal handler */
 
1025
  /* Initialize the pseudo-RNG for Avahi */
 
1026
  srand((unsigned int) time(NULL));
 
1027
  mc.simple_poll = avahi_simple_poll_new();
 
1028
  if(mc.simple_poll == NULL){
 
1029
    fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
 
1030
    exitcode = EXIT_FAILURE;
 
1031
    goto end;
 
1032
  }
 
1033
  
 
1034
  sigemptyset(&sigterm_action.sa_mask);
 
1035
  ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
 
1036
  if(ret == -1){
 
1037
    perror("sigaddset");
 
1038
    exitcode = EXIT_FAILURE;
 
1039
    goto end;
 
1040
  }
 
1041
  ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
 
1042
  if(ret == -1){
 
1043
    perror("sigaddset");
 
1044
    exitcode = EXIT_FAILURE;
 
1045
    goto end;
 
1046
  }
 
1047
  ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
 
1048
  if(ret == -1){
 
1049
    perror("sigaddset");
 
1050
    exitcode = EXIT_FAILURE;
 
1051
    goto end;
 
1052
  }
 
1053
  ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
 
1054
  if(ret == -1){
 
1055
    perror("sigaction");
 
1056
    exitcode = EXIT_FAILURE;
 
1057
    goto end;
 
1058
  }  
 
1059
  
 
1060
  /* If the interface is down, bring it up */
 
1061
  if(interface[0] != '\0'){
 
1062
#ifdef __linux__
 
1063
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
 
1064
       messages to mess up the prompt */
 
1065
    ret = klogctl(8, NULL, 5);
 
1066
    bool restore_loglevel = true;
 
1067
    if(ret == -1){
 
1068
      restore_loglevel = false;
 
1069
      perror("klogctl");
 
1070
    }
 
1071
#endif  /* __linux__ */
 
1072
    
 
1073
    sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
 
1074
    if(sd < 0){
 
1075
      perror("socket");
 
1076
      exitcode = EXIT_FAILURE;
 
1077
#ifdef __linux__
 
1078
      if(restore_loglevel){
 
1079
        ret = klogctl(7, NULL, 0);
 
1080
        if(ret == -1){
 
1081
          perror("klogctl");
 
1082
        }
 
1083
      }
 
1084
#endif  /* __linux__ */
 
1085
      goto end;
 
1086
    }
 
1087
    strcpy(network.ifr_name, interface);
 
1088
    ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1089
    if(ret == -1){
 
1090
      perror("ioctl SIOCGIFFLAGS");
 
1091
#ifdef __linux__
 
1092
      if(restore_loglevel){
 
1093
        ret = klogctl(7, NULL, 0);
 
1094
        if(ret == -1){
 
1095
          perror("klogctl");
 
1096
        }
 
1097
      }
 
1098
#endif  /* __linux__ */
 
1099
      exitcode = EXIT_FAILURE;
 
1100
      goto end;
 
1101
    }
 
1102
    if((network.ifr_flags & IFF_UP) == 0){
 
1103
      network.ifr_flags |= IFF_UP;
 
1104
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
1105
      if(ret == -1){
 
1106
        perror("ioctl SIOCSIFFLAGS");
 
1107
        exitcode = EXIT_FAILURE;
 
1108
#ifdef __linux__
 
1109
        if(restore_loglevel){
 
1110
          ret = klogctl(7, NULL, 0);
 
1111
          if(ret == -1){
 
1112
            perror("klogctl");
 
1113
          }
 
1114
        }
 
1115
#endif  /* __linux__ */
 
1116
        goto end;
 
1117
      }
 
1118
    }
 
1119
    /* sleep checking until interface is running */
 
1120
    for(int i=0; i < delay * 4; i++){
 
1121
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1122
      if(ret == -1){
 
1123
        perror("ioctl SIOCGIFFLAGS");
 
1124
      } else if(network.ifr_flags & IFF_RUNNING){
 
1125
        break;
 
1126
      }
 
1127
      struct timespec sleeptime = { .tv_nsec = 250000000 };
 
1128
      ret = nanosleep(&sleeptime, NULL);
 
1129
      if(ret == -1 and errno != EINTR){
 
1130
        perror("nanosleep");
 
1131
      }
 
1132
    }
 
1133
    ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1134
    if(ret == -1){
 
1135
      perror("close");
 
1136
    }
 
1137
#ifdef __linux__
 
1138
    if(restore_loglevel){
 
1139
      /* Restores kernel loglevel to default */
 
1140
      ret = klogctl(7, NULL, 0);
 
1141
      if(ret == -1){
 
1142
        perror("klogctl");
 
1143
      }
 
1144
    }
 
1145
#endif  /* __linux__ */
 
1146
  }
 
1147
  
 
1148
  uid = getuid();
 
1149
  gid = getgid();
 
1150
  
 
1151
  errno = 0;
 
1152
  setgid(gid);
 
1153
  if(ret == -1){
 
1154
    perror("setgid");
 
1155
  }
 
1156
  
 
1157
  ret = setuid(uid);
 
1158
  if(ret == -1){
 
1159
    perror("setuid");
 
1160
  }
 
1161
  
 
1162
  ret = init_gnutls_global(pubkey, seckey);
 
1163
  if(ret == -1){
 
1164
    fprintf(stderr, "init_gnutls_global failed\n");
 
1165
    exitcode = EXIT_FAILURE;
 
1166
    goto end;
 
1167
  } else {
 
1168
    gnutls_initialized = true;
 
1169
  }
 
1170
  
 
1171
  if(mkdtemp(tempdir) == NULL){
 
1172
    perror("mkdtemp");
 
1173
    goto end;
 
1174
  }
 
1175
  tempdir_created = true;
 
1176
  
 
1177
  if(not init_gpgme(pubkey, seckey, tempdir)){
 
1178
    fprintf(stderr, "init_gpgme failed\n");
 
1179
    exitcode = EXIT_FAILURE;
 
1180
    goto end;
 
1181
  } else {
 
1182
    gpgme_initialized = true;
 
1183
  }
 
1184
  
 
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
    }
 
1192
  }
 
1193
  
 
1194
  if(connect_to != NULL){
 
1195
    /* Connect directly, do not use Zeroconf */
 
1196
    /* (Mainly meant for debugging) */
 
1197
    char *address = strrchr(connect_to, ':');
 
1198
    if(address == NULL){
 
1199
      fprintf(stderr, "No colon in address\n");
 
1200
      exitcode = EXIT_FAILURE;
 
1201
      goto end;
 
1202
    }
 
1203
    uint16_t port;
 
1204
    errno = 0;
 
1205
    tmpmax = strtoimax(address+1, &tmp, 10);
 
1206
    if(errno != 0 or tmp == address+1 or *tmp != '\0'
 
1207
       or tmpmax != (uint16_t)tmpmax){
 
1208
      fprintf(stderr, "Bad port number\n");
 
1209
      exitcode = EXIT_FAILURE;
 
1210
      goto end;
 
1211
    }
 
1212
    port = (uint16_t)tmpmax;
 
1213
    *address = '\0';
 
1214
    address = connect_to;
 
1215
    /* Colon in address indicates IPv6 */
 
1216
    int af;
 
1217
    if(strchr(address, ':') != NULL){
 
1218
      af = AF_INET6;
 
1219
    } else {
 
1220
      af = AF_INET;
 
1221
    }
 
1222
    ret = start_mandos_communication(address, port, if_index, af);
 
1223
    if(ret < 0){
 
1224
      exitcode = EXIT_FAILURE;
 
1225
    } else {
 
1226
      exitcode = EXIT_SUCCESS;
 
1227
    }
 
1228
    goto end;
 
1229
  }
 
1230
    
 
1231
  {
642
1232
    AvahiServerConfig config;
643
 
    AvahiSServiceBrowser *sb = NULL;
644
 
    int error;
645
 
    int ret;
646
 
    int returncode = EXIT_SUCCESS;
647
 
    const char *interface = "eth0";
648
 
    
649
 
    while (true){
650
 
      static struct option long_options[] = {
651
 
        {"debug", no_argument, (int *)&debug, 1},
652
 
        {"interface", required_argument, 0, 'i'},
653
 
        {"certdir", required_argument, 0, 'd'},
654
 
        {"certkey", required_argument, 0, 'c'},
655
 
        {"certfile", required_argument, 0, 'k'},
656
 
        {0, 0, 0, 0} };
657
 
      
658
 
      int option_index = 0;
659
 
      ret = getopt_long (argc, argv, "i:", long_options,
660
 
                         &option_index);
661
 
      
662
 
      if (ret == -1){
663
 
        break;
664
 
      }
665
 
      
666
 
      switch(ret){
667
 
      case 0:
668
 
        break;
669
 
      case 'i':
670
 
        interface = optarg;
671
 
        break;
672
 
      case 'd':
673
 
        certdir = optarg;
674
 
        break;
675
 
      case 'c':
676
 
        certfile = optarg;
677
 
        break;
678
 
      case 'k':
679
 
        certkey = optarg;
680
 
        break;
681
 
      default:
682
 
        exit(EXIT_FAILURE);
683
 
      }
684
 
    }
685
 
 
686
 
    certfile = combinepath(certdir, certfile);
687
 
    if (certfile == NULL){
688
 
      goto exit;
689
 
    }
690
 
    
691
 
    certkey = combinepath(certdir, certkey);
692
 
    if (certkey == NULL){
693
 
      goto exit;
694
 
    }
695
 
        
696
 
    if (not debug){
697
 
      avahi_set_log_function(empty_log);
698
 
    }
699
 
    
700
 
    /* Initialize the psuedo-RNG */
701
 
    srand((unsigned int) time(NULL));
702
 
 
703
 
    /* Allocate main loop object */
704
 
    if (!(simple_poll = avahi_simple_poll_new())) {
705
 
        fprintf(stderr, "Failed to create simple poll object.\n");
706
 
        
707
 
        goto exit;
708
 
    }
709
 
 
710
 
    /* Do not publish any local records */
 
1233
    /* Do not publish any local Zeroconf records */
711
1234
    avahi_server_config_init(&config);
712
1235
    config.publish_hinfo = 0;
713
1236
    config.publish_addresses = 0;
714
1237
    config.publish_workstation = 0;
715
1238
    config.publish_domain = 0;
716
 
 
 
1239
    
717
1240
    /* Allocate a new server */
718
 
    server = avahi_server_new(avahi_simple_poll_get(simple_poll),
719
 
                              &config, NULL, NULL, &error);
720
 
 
721
 
    /* Free the configuration data */
 
1241
    mc.server = avahi_server_new(avahi_simple_poll_get
 
1242
                                 (mc.simple_poll), &config, NULL,
 
1243
                                 NULL, &error);
 
1244
    
 
1245
    /* Free the Avahi configuration data */
722
1246
    avahi_server_config_free(&config);
723
 
 
724
 
    /* Check if creating the server object succeeded */
725
 
    if (!server) {
726
 
        fprintf(stderr, "Failed to create server: %s\n",
727
 
                avahi_strerror(error));
728
 
        returncode = EXIT_FAILURE;
729
 
        goto exit;
730
 
    }
731
 
    
732
 
    /* Create the service browser */
733
 
    sb = avahi_s_service_browser_new(server,
734
 
                                     (AvahiIfIndex)
735
 
                                     if_nametoindex(interface),
736
 
                                     AVAHI_PROTO_INET6,
737
 
                                     "_mandos._tcp", NULL, 0,
738
 
                                     browse_callback, server);
739
 
    if (!sb) {
740
 
        fprintf(stderr, "Failed to create service browser: %s\n",
741
 
                avahi_strerror(avahi_server_errno(server)));
742
 
        returncode = EXIT_FAILURE;
743
 
        goto exit;
744
 
    }
745
 
    
746
 
    /* Run the main loop */
747
 
 
748
 
    if (debug){
749
 
      fprintf(stderr, "Starting avahi loop search\n");
750
 
    }
751
 
    
752
 
    avahi_simple_poll_loop(simple_poll);
753
 
    
754
 
 exit:
755
 
 
756
 
    if (debug){
757
 
      fprintf(stderr, "%s exiting\n", argv[0]);
758
 
    }
759
 
    
760
 
    /* Cleanup things */
761
 
    if (sb)
762
 
        avahi_s_service_browser_free(sb);
763
 
    
764
 
    if (server)
765
 
        avahi_server_free(server);
766
 
 
767
 
    if (simple_poll)
768
 
        avahi_simple_poll_free(simple_poll);
769
 
    free(certfile);
770
 
    free(certkey);
771
 
    
772
 
    return returncode;
 
1247
  }
 
1248
  
 
1249
  /* Check if creating the Avahi server object succeeded */
 
1250
  if(mc.server == NULL){
 
1251
    fprintf(stderr, "Failed to create Avahi server: %s\n",
 
1252
            avahi_strerror(error));
 
1253
    exitcode = EXIT_FAILURE;
 
1254
    goto end;
 
1255
  }
 
1256
  
 
1257
  /* Create the Avahi service browser */
 
1258
  sb = avahi_s_service_browser_new(mc.server, if_index,
 
1259
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
 
1260
                                   NULL, 0, browse_callback, NULL);
 
1261
  if(sb == NULL){
 
1262
    fprintf(stderr, "Failed to create service browser: %s\n",
 
1263
            avahi_strerror(avahi_server_errno(mc.server)));
 
1264
    exitcode = EXIT_FAILURE;
 
1265
    goto end;
 
1266
  }
 
1267
  
 
1268
  /* Run the main loop */
 
1269
  
 
1270
  if(debug){
 
1271
    fprintf(stderr, "Starting Avahi loop search\n");
 
1272
  }
 
1273
  
 
1274
  avahi_simple_poll_loop(mc.simple_poll);
 
1275
  
 
1276
 end:
 
1277
  
 
1278
  if(debug){
 
1279
    fprintf(stderr, "%s exiting\n", argv[0]);
 
1280
  }
 
1281
  
 
1282
  /* Cleanup things */
 
1283
  if(sb != NULL)
 
1284
    avahi_s_service_browser_free(sb);
 
1285
  
 
1286
  if(mc.server != NULL)
 
1287
    avahi_server_free(mc.server);
 
1288
  
 
1289
  if(mc.simple_poll != NULL)
 
1290
    avahi_simple_poll_free(mc.simple_poll);
 
1291
  
 
1292
  if(gnutls_initialized){
 
1293
    gnutls_certificate_free_credentials(mc.cred);
 
1294
    gnutls_global_deinit();
 
1295
    gnutls_dh_params_deinit(mc.dh_params);
 
1296
  }
 
1297
  
 
1298
  if(gpgme_initialized){
 
1299
    gpgme_release(mc.ctx);
 
1300
  }
 
1301
  
 
1302
  /* Removes the temp directory used by GPGME */
 
1303
  if(tempdir_created){
 
1304
    DIR *d;
 
1305
    struct dirent *direntry;
 
1306
    d = opendir(tempdir);
 
1307
    if(d == NULL){
 
1308
      if(errno != ENOENT){
 
1309
        perror("opendir");
 
1310
      }
 
1311
    } else {
 
1312
      while(true){
 
1313
        direntry = readdir(d);
 
1314
        if(direntry == NULL){
 
1315
          break;
 
1316
        }
 
1317
        /* Skip "." and ".." */
 
1318
        if(direntry->d_name[0] == '.'
 
1319
           and (direntry->d_name[1] == '\0'
 
1320
                or (direntry->d_name[1] == '.'
 
1321
                    and direntry->d_name[2] == '\0'))){
 
1322
          continue;
 
1323
        }
 
1324
        char *fullname = NULL;
 
1325
        ret = asprintf(&fullname, "%s/%s", tempdir,
 
1326
                       direntry->d_name);
 
1327
        if(ret < 0){
 
1328
          perror("asprintf");
 
1329
          continue;
 
1330
        }
 
1331
        ret = remove(fullname);
 
1332
        if(ret == -1){
 
1333
          fprintf(stderr, "remove(\"%s\"): %s\n", fullname,
 
1334
                  strerror(errno));
 
1335
        }
 
1336
        free(fullname);
 
1337
      }
 
1338
      closedir(d);
 
1339
    }
 
1340
    ret = rmdir(tempdir);
 
1341
    if(ret == -1 and errno != ENOENT){
 
1342
      perror("rmdir");
 
1343
    }
 
1344
  }
 
1345
  
 
1346
  return exitcode;
773
1347
}