/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-15 09:28:06 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090215092806-l0uq42v3j1rq3gmi
* INSTALL: Changed "Python 2.4" to "Python 2.5".
* mandos.lsm: - '' -

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
 
#include <sys/ioctl.h>          // ioctl, ifreq, SIOCGIFFLAGS, IFF_UP, SIOCSIFFLAGS
43
 
#include <net/if.h>             // ioctl, ifreq, SIOCGIFFLAGS, IFF_UP, SIOCSIFFLAGS
44
 
 
 
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_* */
45
94
#include <avahi-core/core.h>
46
95
#include <avahi-core/lookup.h>
47
96
#include <avahi-core/log.h>
49
98
#include <avahi-common/malloc.h>
50
99
#include <avahi-common/error.h>
51
100
 
52
 
//mandos client part
53
 
#include <sys/types.h>          /* socket(), inet_pton() */
54
 
#include <sys/socket.h>         /* socket(), struct sockaddr_in6,
55
 
                                   struct in6_addr, inet_pton() */
56
 
#include <gnutls/gnutls.h>      /* All GnuTLS stuff */
57
 
#include <gnutls/openpgp.h>     /* GnuTLS with openpgp stuff */
58
 
 
59
 
#include <unistd.h>             /* close() */
60
 
#include <netinet/in.h>
61
 
#include <stdbool.h>            /* true */
62
 
#include <string.h>             /* memset */
63
 
#include <arpa/inet.h>          /* inet_pton() */
64
 
#include <iso646.h>             /* not */
65
 
 
66
 
// gpgme
67
 
#include <errno.h>              /* perror() */
68
 
#include <gpgme.h>
69
 
 
70
 
// getopt long
71
 
#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_* */
72
117
 
73
118
#define BUFFER_SIZE 256
74
 
#define DH_BITS 1024
75
119
 
76
 
const char *certdir = "/conf/conf.d/cryptkeyreq/";
77
 
const char *certfile = "openpgp-client.txt";
78
 
const char *certkey = "openpgp-client-key.txt";
 
120
#define PATHDIR "/conf/conf.d/mandos"
 
121
#define SECKEY "seckey.txt"
 
122
#define PUBKEY "pubkey.txt"
79
123
 
80
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>";
81
128
 
 
129
/* Used for passing in values through the Avahi callback functions */
82
130
typedef struct {
83
 
  gnutls_session_t session;
 
131
  AvahiSimplePoll *simple_poll;
 
132
  AvahiServer *server;
84
133
  gnutls_certificate_credentials_t cred;
 
134
  unsigned int dh_bits;
85
135
  gnutls_dh_params_t dh_params;
86
 
} encrypted_session;
87
 
 
88
 
 
89
 
ssize_t pgp_packet_decrypt (char *packet, size_t packet_size,
90
 
                            char **new_packet, const char *homedir){
91
 
  gpgme_data_t dh_crypto, dh_plain;
 
136
  const char *priority;
92
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;
93
168
  gpgme_error_t rc;
94
 
  ssize_t ret;
95
 
  ssize_t new_packet_capacity = 0;
96
 
  ssize_t new_packet_length = 0;
97
169
  gpgme_engine_info_t engine_info;
98
 
 
99
 
  if (debug){
100
 
    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");
101
209
  }
102
210
  
103
211
  /* Init GPGME */
104
212
  gpgme_check_version(NULL);
105
213
  rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
106
 
  if (rc != GPG_ERR_NO_ERROR){
 
214
  if(rc != GPG_ERR_NO_ERROR){
107
215
    fprintf(stderr, "bad gpgme_engine_check_version: %s: %s\n",
108
216
            gpgme_strsource(rc), gpgme_strerror(rc));
109
 
    return -1;
 
217
    return false;
110
218
  }
111
219
  
112
 
  /* Set GPGME home directory */
113
 
  rc = gpgme_get_engine_info (&engine_info);
114
 
  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){
115
223
    fprintf(stderr, "bad gpgme_get_engine_info: %s: %s\n",
116
224
            gpgme_strsource(rc), gpgme_strerror(rc));
117
 
    return -1;
 
225
    return false;
118
226
  }
119
227
  while(engine_info != NULL){
120
228
    if(engine_info->protocol == GPGME_PROTOCOL_OpenPGP){
121
229
      gpgme_set_engine_info(GPGME_PROTOCOL_OpenPGP,
122
 
                            engine_info->file_name, homedir);
 
230
                            engine_info->file_name, tempdir);
123
231
      break;
124
232
    }
125
233
    engine_info = engine_info->next;
126
234
  }
127
235
  if(engine_info == NULL){
128
 
    fprintf(stderr, "Could not set home dir to %s\n", homedir);
129
 
    return -1;
130
 
  }
131
 
  
132
 
  /* Create new GPGME data buffer from packet buffer */
133
 
  rc = gpgme_data_new_from_mem(&dh_crypto, packet, packet_size, 0);
134
 
  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){
135
276
    fprintf(stderr, "bad gpgme_data_new_from_mem: %s: %s\n",
136
277
            gpgme_strsource(rc), gpgme_strerror(rc));
137
278
    return -1;
139
280
  
140
281
  /* Create new empty GPGME data buffer for the plaintext */
141
282
  rc = gpgme_data_new(&dh_plain);
142
 
  if (rc != GPG_ERR_NO_ERROR){
 
283
  if(rc != GPG_ERR_NO_ERROR){
143
284
    fprintf(stderr, "bad gpgme_data_new: %s: %s\n",
144
285
            gpgme_strsource(rc), gpgme_strerror(rc));
145
 
    return -1;
146
 
  }
147
 
  
148
 
  /* Create new GPGME "context" */
149
 
  rc = gpgme_new(&ctx);
150
 
  if (rc != GPG_ERR_NO_ERROR){
151
 
    fprintf(stderr, "bad gpgme_new: %s: %s\n",
152
 
            gpgme_strsource(rc), gpgme_strerror(rc));
153
 
    return -1;
154
 
  }
155
 
  
156
 
  /* Decrypt data from the FILE pointer to the plaintext data
157
 
     buffer */
158
 
  rc = gpgme_op_decrypt(ctx, dh_crypto, dh_plain);
159
 
  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){
160
294
    fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
161
295
            gpgme_strsource(rc), gpgme_strerror(rc));
162
 
    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;
163
326
  }
164
 
 
 
327
  
165
328
  if(debug){
166
 
    fprintf(stderr, "Decryption of OpenPGP packet succeeded\n");
167
 
  }
168
 
 
169
 
  if (debug){
170
 
    gpgme_decrypt_result_t result;
171
 
    result = gpgme_op_decrypt_result(ctx);
172
 
    if (result == NULL){
173
 
      fprintf(stderr, "gpgme_op_decrypt_result failed\n");
174
 
    } else {
175
 
      fprintf(stderr, "Unsupported algorithm: %s\n",
176
 
              result->unsupported_algorithm);
177
 
      fprintf(stderr, "Wrong key usage: %d\n",
178
 
              result->wrong_key_usage);
179
 
      if(result->file_name != NULL){
180
 
        fprintf(stderr, "File name: %s\n", result->file_name);
181
 
      }
182
 
      gpgme_recipient_t recipient;
183
 
      recipient = result->recipients;
184
 
      if(recipient){
185
 
        while(recipient != NULL){
186
 
          fprintf(stderr, "Public key algorithm: %s\n",
187
 
                  gpgme_pubkey_algo_name(recipient->pubkey_algo));
188
 
          fprintf(stderr, "Key ID: %s\n", recipient->keyid);
189
 
          fprintf(stderr, "Secret key available: %s\n",
190
 
                  recipient->status == GPG_ERR_NO_SECKEY
191
 
                  ? "No" : "Yes");
192
 
          recipient = recipient->next;
193
 
        }
194
 
      }
195
 
    }
196
 
  }
197
 
  
198
 
  /* Delete the GPGME FILE pointer cryptotext data buffer */
199
 
  gpgme_data_release(dh_crypto);
 
329
    fprintf(stderr, "Decryption of OpenPGP data succeeded\n");
 
330
  }
200
331
  
201
332
  /* Seek back to the beginning of the GPGME plaintext data buffer */
202
 
  if (gpgme_data_seek(dh_plain, (off_t) 0, SEEK_SET) == -1){
203
 
    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;
204
337
  }
205
338
  
206
 
  *new_packet = 0;
 
339
  *plaintext = NULL;
207
340
  while(true){
208
 
    if (new_packet_length + BUFFER_SIZE > new_packet_capacity){
209
 
      *new_packet = realloc(*new_packet,
210
 
                            (unsigned int)new_packet_capacity
211
 
                            + BUFFER_SIZE);
212
 
      if (*new_packet == NULL){
213
 
        perror("realloc");
214
 
        return -1;
215
 
      }
216
 
      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;
217
348
    }
218
349
    
219
 
    ret = gpgme_data_read(dh_plain, *new_packet + new_packet_length,
 
350
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
220
351
                          BUFFER_SIZE);
221
352
    /* Print the data, if any */
222
 
    if (ret == 0){
 
353
    if(ret == 0){
 
354
      /* EOF */
223
355
      break;
224
356
    }
225
357
    if(ret < 0){
226
358
      perror("gpgme_data_read");
227
 
      return -1;
228
 
    }
229
 
    new_packet_length += ret;
230
 
  }
231
 
 
232
 
  /* FIXME: check characters before printing to screen so to not print
233
 
     terminal control characters */
234
 
  /*   if(debug){ */
235
 
  /*     fprintf(stderr, "decrypted password is: "); */
236
 
  /*     fwrite(*new_packet, 1, new_packet_length, stderr); */
237
 
  /*     fprintf(stderr, "\n"); */
238
 
  /*   } */
 
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);
239
377
  
240
378
  /* Delete the GPGME plaintext data buffer */
241
379
  gpgme_data_release(dh_plain);
242
 
  return new_packet_length;
 
380
  return plaintext_length;
243
381
}
244
382
 
245
 
static const char * safer_gnutls_strerror (int value) {
246
 
  const char *ret = gnutls_strerror (value);
247
 
  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)
248
387
    ret = "(unknown)";
249
388
  return ret;
250
389
}
251
390
 
252
 
void debuggnutls(__attribute__((unused)) int level,
253
 
                 const char* string){
254
 
  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);
255
395
}
256
396
 
257
 
int initgnutls(encrypted_session *es){
258
 
  const char *err;
 
397
static int init_gnutls_global(const char *pubkeyfilename,
 
398
                              const char *seckeyfilename){
259
399
  int ret;
260
400
  
261
401
  if(debug){
262
402
    fprintf(stderr, "Initializing GnuTLS\n");
263
403
  }
264
 
 
265
 
  if ((ret = gnutls_global_init ())
266
 
      != GNUTLS_E_SUCCESS) {
267
 
    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));
268
409
    return -1;
269
410
  }
270
 
 
271
 
  if (debug){
 
411
  
 
412
  if(debug){
 
413
    /* "Use a log level over 10 to enable all debugging options."
 
414
     * - GnuTLS manual
 
415
     */
272
416
    gnutls_global_set_log_level(11);
273
417
    gnutls_global_set_log_function(debuggnutls);
274
418
  }
275
419
  
276
 
  /* openpgp credentials */
277
 
  if ((ret = gnutls_certificate_allocate_credentials (&es->cred))
278
 
      != GNUTLS_E_SUCCESS) {
279
 
    fprintf (stderr, "memory error: %s\n",
280
 
             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();
281
429
    return -1;
282
430
  }
283
431
  
284
432
  if(debug){
285
 
    fprintf(stderr, "Attempting to use OpenPGP certificate %s"
286
 
            " and keyfile %s as GnuTLS credentials\n", certfile,
287
 
            certkey);
 
433
    fprintf(stderr, "Attempting to use OpenPGP public key %s and"
 
434
            " secret key %s as GnuTLS credentials\n", pubkeyfilename,
 
435
            seckeyfilename);
288
436
  }
289
437
  
290
438
  ret = gnutls_certificate_set_openpgp_key_file
291
 
    (es->cred, certfile, certkey, GNUTLS_OPENPGP_FMT_BASE64);
292
 
  if (ret != GNUTLS_E_SUCCESS) {
293
 
    fprintf
294
 
      (stderr, "Error[%d] while reading the OpenPGP key pair ('%s',"
295
 
       " '%s')\n",
296
 
       ret, certfile, certkey);
297
 
    fprintf(stdout, "The Error is: %s\n",
298
 
            safer_gnutls_strerror(ret));
299
 
    return -1;
300
 
  }
301
 
  
302
 
  //GnuTLS server initialization
303
 
  if ((ret = gnutls_dh_params_init (&es->dh_params))
304
 
      != GNUTLS_E_SUCCESS) {
305
 
    fprintf (stderr, "Error in dh parameter initialization: %s\n",
306
 
             safer_gnutls_strerror(ret));
307
 
    return -1;
308
 
  }
309
 
  
310
 
  if ((ret = gnutls_dh_params_generate2 (es->dh_params, DH_BITS))
311
 
      != GNUTLS_E_SUCCESS) {
312
 
    fprintf (stderr, "Error in prime generation: %s\n",
313
 
             safer_gnutls_strerror(ret));
314
 
    return -1;
315
 
  }
316
 
  
317
 
  gnutls_certificate_set_dh_params (es->cred, es->dh_params);
318
 
  
319
 
  // GnuTLS session creation
320
 
  if ((ret = gnutls_init (&es->session, GNUTLS_SERVER))
321
 
      != 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){
322
481
    fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
323
482
            safer_gnutls_strerror(ret));
324
483
  }
325
484
  
326
 
  if ((ret = gnutls_priority_set_direct (es->session, "NORMAL", &err))
327
 
      != GNUTLS_E_SUCCESS) {
328
 
    fprintf(stderr, "Syntax error at: %s\n", err);
329
 
    fprintf(stderr, "GnuTLS error: %s\n",
330
 
            safer_gnutls_strerror(ret));
331
 
    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
    }
332
495
  }
333
496
  
334
 
  if ((ret = gnutls_credentials_set
335
 
       (es->session, GNUTLS_CRD_CERTIFICATE, es->cred))
336
 
      != GNUTLS_E_SUCCESS) {
337
 
    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",
338
501
            safer_gnutls_strerror(ret));
 
502
    gnutls_deinit(*session);
339
503
    return -1;
340
504
  }
341
505
  
342
506
  /* ignore client certificate if any. */
343
 
  gnutls_certificate_server_set_request (es->session,
344
 
                                         GNUTLS_CERT_IGNORE);
 
507
  gnutls_certificate_server_set_request(*session,
 
508
                                        GNUTLS_CERT_IGNORE);
345
509
  
346
 
  gnutls_dh_set_prime_bits (es->session, DH_BITS);
 
510
  gnutls_dh_set_prime_bits(*session, mc.dh_bits);
347
511
  
348
512
  return 0;
349
513
}
350
514
 
351
 
void empty_log(__attribute__((unused)) AvahiLogLevel level,
352
 
               __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){}
353
518
 
354
 
int start_mandos_communication(const char *ip, uint16_t port,
355
 
                               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){
356
523
  int ret, tcp_sd;
357
 
  struct sockaddr_in6 to;
358
 
  encrypted_session es;
 
524
  ssize_t sret;
 
525
  union {
 
526
    struct sockaddr_in in;
 
527
    struct sockaddr_in6 in6;
 
528
  } to;
359
529
  char *buffer = NULL;
360
530
  char *decrypted_buffer;
361
531
  size_t buffer_length = 0;
362
532
  size_t buffer_capacity = 0;
363
533
  ssize_t decrypted_buffer_size;
364
 
  size_t written = 0;
 
534
  size_t written;
365
535
  int retval = 0;
366
 
  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
  }
367
555
  
368
556
  if(debug){
369
 
    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);
370
559
  }
371
560
  
372
 
  tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
373
 
  if(tcp_sd < 0) {
 
561
  tcp_sd = socket(pf, SOCK_STREAM, 0);
 
562
  if(tcp_sd < 0){
374
563
    perror("socket");
375
564
    return -1;
376
565
  }
377
 
 
378
 
  if(debug){
379
 
    if(if_indextoname(if_index, interface) == NULL){
380
 
      if(debug){
381
 
        perror("if_indextoname");
382
 
      }
383
 
      return -1;
384
 
    }
385
 
    
386
 
    fprintf(stderr, "Binding to interface %s\n", interface);
387
 
  }
388
566
  
389
 
  memset(&to,0,sizeof(to));     /* Spurious warning */
390
 
  to.sin6_family = AF_INET6;
391
 
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
392
 
  if (ret < 0 ){
 
567
  memset(&to, 0, sizeof(to));
 
568
  if(af == AF_INET6){
 
569
    to.in6.sin6_family = (uint16_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 ){
393
576
    perror("inet_pton");
394
577
    return -1;
395
 
  }  
 
578
  }
396
579
  if(ret == 0){
397
580
    fprintf(stderr, "Bad address: %s\n", ip);
398
581
    return -1;
399
582
  }
400
 
  to.sin6_port = htons(port);   /* Spurious warning */
401
 
  
402
 
  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
  }
403
604
  
404
605
  if(debug){
405
 
    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
    }
406
635
  }
407
636
  
408
 
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
409
 
  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){
410
643
    perror("connect");
411
644
    return -1;
412
645
  }
413
646
  
414
 
  ret = initgnutls (&es);
415
 
  if (ret != 0){
416
 
    retval = -1;
417
 
    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
    }
418
669
  }
419
670
  
420
 
  gnutls_transport_set_ptr (es.session,
421
 
                            (gnutls_transport_ptr_t) tcp_sd);
422
 
  
423
671
  if(debug){
424
672
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
425
673
  }
426
674
  
427
 
  ret = gnutls_handshake (es.session);
428
 
  
429
 
  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){
430
682
    if(debug){
431
 
      fprintf(stderr, "\n*** Handshake failed ***\n");
432
 
      gnutls_perror (ret);
 
683
      fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
 
684
      gnutls_perror(ret);
433
685
    }
434
686
    retval = -1;
435
 
    goto exit;
 
687
    goto mandos_end;
436
688
  }
437
689
  
438
 
  //Retrieve OpenPGP packet that contains the wanted password
 
690
  /* Read OpenPGP packet that contains the wanted password */
439
691
  
440
692
  if(debug){
441
 
    fprintf(stderr, "Retrieving pgp encrypted password from %s\n",
 
693
    fprintf(stderr, "Retrieving OpenPGP encrypted password from %s\n",
442
694
            ip);
443
695
  }
444
 
 
 
696
  
445
697
  while(true){
446
 
    if (buffer_length + BUFFER_SIZE > buffer_capacity){
447
 
      buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
448
 
      if (buffer == NULL){
449
 
        perror("realloc");
450
 
        goto exit;
451
 
      }
452
 
      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;
453
704
    }
454
705
    
455
 
    ret = gnutls_record_recv
456
 
      (es.session, buffer+buffer_length, BUFFER_SIZE);
457
 
    if (ret == 0){
 
706
    sret = gnutls_record_recv(session, buffer+buffer_length,
 
707
                              BUFFER_SIZE);
 
708
    if(sret == 0){
458
709
      break;
459
710
    }
460
 
    if (ret < 0){
461
 
      switch(ret){
 
711
    if(sret < 0){
 
712
      switch(sret){
462
713
      case GNUTLS_E_INTERRUPTED:
463
714
      case GNUTLS_E_AGAIN:
464
715
        break;
465
716
      case GNUTLS_E_REHANDSHAKE:
466
 
        ret = gnutls_handshake (es.session);
467
 
        if (ret < 0){
468
 
          fprintf(stderr, "\n*** Handshake failed ***\n");
469
 
          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);
470
723
          retval = -1;
471
 
          goto exit;
 
724
          goto mandos_end;
472
725
        }
473
726
        break;
474
727
      default:
475
728
        fprintf(stderr, "Unknown error while reading data from"
476
 
                " encrypted session with mandos server\n");
 
729
                " encrypted session with Mandos server\n");
477
730
        retval = -1;
478
 
        gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
479
 
        goto exit;
 
731
        gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
732
        goto mandos_end;
480
733
      }
481
734
    } else {
482
 
      buffer_length += (size_t) ret;
 
735
      buffer_length += (size_t) sret;
483
736
    }
484
737
  }
485
738
  
486
 
  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){
487
746
    decrypted_buffer_size = pgp_packet_decrypt(buffer,
488
747
                                               buffer_length,
489
 
                                               &decrypted_buffer,
490
 
                                               certdir);
491
 
    if (decrypted_buffer_size >= 0){
492
 
      while(written < (size_t)decrypted_buffer_size){
493
 
        ret = (int)fwrite (decrypted_buffer + written, 1,
494
 
                           (size_t)decrypted_buffer_size - written,
495
 
                           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);
496
755
        if(ret == 0 and ferror(stdout)){
497
756
          if(debug){
498
757
            fprintf(stderr, "Error writing encrypted data: %s\n",
507
766
    } else {
508
767
      retval = -1;
509
768
    }
510
 
  }
511
 
 
512
 
  //shutdown procedure
513
 
 
514
 
  if(debug){
515
 
    fprintf(stderr, "Closing TLS session\n");
516
 
  }
517
 
 
 
769
  } else {
 
770
    retval = -1;
 
771
  }
 
772
  
 
773
  /* Shutdown procedure */
 
774
  
 
775
 mandos_end:
518
776
  free(buffer);
519
 
  gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
520
 
 exit:
521
 
  close(tcp_sd);
522
 
  gnutls_deinit (es.session);
523
 
  gnutls_certificate_free_credentials (es.cred);
524
 
  gnutls_global_deinit ();
 
777
  ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
 
778
  if(ret == -1){
 
779
    perror("close");
 
780
  }
 
781
  gnutls_deinit(session);
525
782
  return retval;
526
783
}
527
784
 
528
 
static AvahiSimplePoll *simple_poll = NULL;
529
 
static AvahiServer *server = NULL;
530
 
 
531
 
static void resolve_callback(
532
 
    AvahiSServiceResolver *r,
533
 
    AvahiIfIndex interface,
534
 
    AVAHI_GCC_UNUSED AvahiProtocol protocol,
535
 
    AvahiResolverEvent event,
536
 
    const char *name,
537
 
    const char *type,
538
 
    const char *domain,
539
 
    const char *host_name,
540
 
    const AvahiAddress *address,
541
 
    uint16_t port,
542
 
    AVAHI_GCC_UNUSED AvahiStringList *txt,
543
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
544
 
    AVAHI_GCC_UNUSED void* userdata) {
545
 
    
546
 
  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);
547
800
  
548
801
  /* Called whenever a service has been resolved successfully or
549
802
     timed out */
550
803
  
551
 
  switch (event) {
 
804
  switch(event){
552
805
  default:
553
806
  case AVAHI_RESOLVER_FAILURE:
554
 
    fprintf(stderr, "(Resolver) Failed to resolve service '%s' of"
555
 
            " type '%s' in domain '%s': %s\n", name, type, domain,
556
 
            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)));
557
810
    break;
558
811
    
559
812
  case AVAHI_RESOLVER_FOUND:
561
814
      char ip[AVAHI_ADDRESS_STR_MAX];
562
815
      avahi_address_snprint(ip, sizeof(ip), address);
563
816
      if(debug){
564
 
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s) on"
565
 
                " 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);
566
820
      }
567
 
      int ret = start_mandos_communication(ip, port,
568
 
                                           (unsigned int) interface);
569
 
      if (ret == 0){
570
 
        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);
571
825
      }
572
826
    }
573
827
  }
574
828
  avahi_s_service_resolver_free(r);
575
829
}
576
830
 
577
 
static void browse_callback(
578
 
    AvahiSServiceBrowser *b,
579
 
    AvahiIfIndex interface,
580
 
    AvahiProtocol protocol,
581
 
    AvahiBrowserEvent event,
582
 
    const char *name,
583
 
    const char *type,
584
 
    const char *domain,
585
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
586
 
    void* userdata) {
587
 
    
588
 
    AvahiServer *s = userdata;
589
 
    assert(b);                  /* Spurious warning */
590
 
    
591
 
    /* Called whenever a new services becomes available on the LAN or
592
 
       is removed from the LAN */
593
 
    
594
 
    switch (event) {
595
 
    default:
596
 
    case AVAHI_BROWSER_FAILURE:
597
 
      
598
 
      fprintf(stderr, "(Browser) %s\n",
599
 
              avahi_strerror(avahi_server_errno(server)));
600
 
      avahi_simple_poll_quit(simple_poll);
601
 
      return;
602
 
      
603
 
    case AVAHI_BROWSER_NEW:
604
 
      /* We ignore the returned resolver object. In the callback
605
 
         function we free it. If the server is terminated before
606
 
         the callback function is called the server will free
607
 
         the resolver for us. */
608
 
      
609
 
      if (!(avahi_s_service_resolver_new(s, interface, protocol, name,
610
 
                                         type, domain,
611
 
                                         AVAHI_PROTO_INET6, 0,
612
 
                                         resolve_callback, s)))
613
 
        fprintf(stderr, "Failed to resolve service '%s': %s\n", name,
614
 
                avahi_strerror(avahi_server_errno(s)));
615
 
      break;
616
 
      
617
 
    case AVAHI_BROWSER_REMOVE:
618
 
      break;
619
 
      
620
 
    case AVAHI_BROWSER_ALL_FOR_NOW:
621
 
    case AVAHI_BROWSER_CACHE_EXHAUSTED:
622
 
      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");
623
875
    }
624
 
}
625
 
 
626
 
/* combinds file name and path and returns the malloced new string. som sane checks could/should be added */
627
 
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;
628
900
  char *tmp;
629
 
  tmp = malloc(strlen(first) + strlen(second) + 2);
630
 
  if (tmp == NULL){
631
 
    perror("malloc");
632
 
    return NULL;
633
 
  }
634
 
  strcpy(tmp, first);
635
 
  if (first[0] != '\0' and first[strlen(first) - 1] != '/'){
636
 
    strcat(tmp, "/");
637
 
  }
638
 
  strcat(tmp, second);
639
 
  return tmp;
640
 
}
641
 
 
642
 
 
643
 
int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char*argv[]) {
644
 
    AvahiServerConfig config;
645
 
    AvahiSServiceBrowser *sb = NULL;
646
 
    int error;
647
 
    int ret;
648
 
    int returncode = EXIT_SUCCESS;
649
 
    const char *interface = "eth0";
650
 
    struct ifreq network;
651
 
    int sd;
 
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
    };
652
957
    
653
 
    while (true){
654
 
      static struct option long_options[] = {
655
 
        {"debug", no_argument, (int *)&debug, 1},
656
 
        {"interface", required_argument, 0, 'i'},
657
 
        {"certdir", required_argument, 0, 'd'},
658
 
        {"certkey", required_argument, 0, 'c'},
659
 
        {"certfile", required_argument, 0, 'k'},
660
 
        {0, 0, 0, 0} };
661
 
      
662
 
      int option_index = 0;
663
 
      ret = getopt_long (argc, argv, "i:", long_options,
664
 
                         &option_index);
665
 
      
666
 
      if (ret == -1){
667
 
        break;
668
 
      }
669
 
      
670
 
      switch(ret){
671
 
      case 0:
672
 
        break;
673
 
      case 'i':
674
 
        interface = optarg;
675
 
        break;
676
 
      case 'd':
677
 
        certdir = optarg;
678
 
        break;
679
 
      case 'c':
680
 
        certfile = optarg;
681
 
        break;
682
 
      case 'k':
683
 
        certkey = optarg;
 
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:
684
1000
        break;
685
1001
      default:
686
 
        exit(EXIT_FAILURE);
 
1002
        return ARGP_ERR_UNKNOWN;
687
1003
      }
688
 
    }
689
 
 
690
 
    certfile = combinepath(certdir, certfile);
691
 
    if (certfile == NULL){
692
 
      returncode = EXIT_FAILURE;
693
 
      goto exit;
694
 
    }
695
 
    
696
 
    certkey = combinepath(certdir, certkey);
697
 
    if (certkey == NULL){
698
 
      returncode = EXIT_FAILURE;
699
 
      goto exit;
700
 
    }
701
 
 
 
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
    
702
1073
    sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
703
 
    if(sd < 0) {
 
1074
    if(sd < 0){
704
1075
      perror("socket");
705
 
      returncode = EXIT_FAILURE;
706
 
      goto exit;
 
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;
707
1086
    }
708
 
    strcpy(network.ifr_name, interface);    
 
1087
    strcpy(network.ifr_name, interface);
709
1088
    ret = ioctl(sd, SIOCGIFFLAGS, &network);
710
1089
    if(ret == -1){
711
 
      
712
1090
      perror("ioctl SIOCGIFFLAGS");
713
 
      returncode = EXIT_FAILURE;
714
 
      goto exit;
 
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;
715
1101
    }
716
1102
    if((network.ifr_flags & IFF_UP) == 0){
717
1103
      network.ifr_flags |= IFF_UP;
718
1104
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
719
1105
      if(ret == -1){
720
1106
        perror("ioctl SIOCSIFFLAGS");
721
 
        returncode = EXIT_FAILURE;
722
 
        goto exit;
723
 
      }
724
 
    }
725
 
    close(sd);
726
 
    
727
 
    if (not debug){
728
 
      avahi_set_log_function(empty_log);
729
 
    }
730
 
    
731
 
    /* Initialize the psuedo-RNG */
732
 
    srand((unsigned int) time(NULL));
733
 
 
734
 
    /* Allocate main loop object */
735
 
    if (!(simple_poll = avahi_simple_poll_new())) {
736
 
        fprintf(stderr, "Failed to create simple poll object.\n");
737
 
        returncode = EXIT_FAILURE;      
738
 
        goto exit;
739
 
    }
740
 
 
741
 
    /* Do not publish any local records */
 
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
  {
 
1232
    AvahiServerConfig config;
 
1233
    /* Do not publish any local Zeroconf records */
742
1234
    avahi_server_config_init(&config);
743
1235
    config.publish_hinfo = 0;
744
1236
    config.publish_addresses = 0;
745
1237
    config.publish_workstation = 0;
746
1238
    config.publish_domain = 0;
747
 
 
 
1239
    
748
1240
    /* Allocate a new server */
749
 
    server = avahi_server_new(avahi_simple_poll_get(simple_poll),
750
 
                              &config, NULL, NULL, &error);
751
 
 
752
 
    /* 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 */
753
1246
    avahi_server_config_free(&config);
754
 
 
755
 
    /* Check if creating the server object succeeded */
756
 
    if (!server) {
757
 
        fprintf(stderr, "Failed to create server: %s\n",
758
 
                avahi_strerror(error));
759
 
        returncode = EXIT_FAILURE;
760
 
        goto exit;
761
 
    }
762
 
    
763
 
    /* Create the service browser */
764
 
    sb = avahi_s_service_browser_new(server,
765
 
                                     (AvahiIfIndex)
766
 
                                     if_nametoindex(interface),
767
 
                                     AVAHI_PROTO_INET6,
768
 
                                     "_mandos._tcp", NULL, 0,
769
 
                                     browse_callback, server);
770
 
    if (!sb) {
771
 
        fprintf(stderr, "Failed to create service browser: %s\n",
772
 
                avahi_strerror(avahi_server_errno(server)));
773
 
        returncode = EXIT_FAILURE;
774
 
        goto exit;
775
 
    }
776
 
    
777
 
    /* Run the main loop */
778
 
 
779
 
    if (debug){
780
 
      fprintf(stderr, "Starting avahi loop search\n");
781
 
    }
782
 
    
783
 
    avahi_simple_poll_loop(simple_poll);
784
 
    
785
 
 exit:
786
 
 
787
 
    if (debug){
788
 
      fprintf(stderr, "%s exiting\n", argv[0]);
789
 
    }
790
 
    
791
 
    /* Cleanup things */
792
 
    if (sb)
793
 
        avahi_s_service_browser_free(sb);
794
 
    
795
 
    if (server)
796
 
        avahi_server_free(server);
797
 
 
798
 
    if (simple_poll)
799
 
        avahi_simple_poll_free(simple_poll);
800
 
    free(certfile);
801
 
    free(certkey);
802
 
    
803
 
    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;
804
1347
}