/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to plugins.d/mandos-client.c

* plugins.d/mandos-client.c (good_interface): Check if the interface
                                              name starts with "."
                                              before anything else.
                                              Bug fix: Fix memory
                                              leak.

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-2011 Teddy Hogeborn
 
13
 * Copyright © 2008-2011 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, srand(),
 
47
                                   strtof(), abort() */
 
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(), sleep() */
 
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(), seteuid(),
 
75
                                   setgid(), pause() */
 
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, sig_atomic_t,
 
84
                                   raise() */
 
85
#include <sysexits.h>           /* EX_OSERR, EX_USAGE, EX_UNAVAILABLE,
 
86
                                   EX_NOHOST, EX_IOERR, EX_PROTOCOL */
 
87
 
 
88
#ifdef __linux__
 
89
#include <sys/klog.h>           /* klogctl() */
 
90
#endif  /* __linux__ */
 
91
 
 
92
/* Avahi */
 
93
/* All Avahi types, constants and functions
 
94
 Avahi*, avahi_*,
 
95
 AVAHI_* */
43
96
#include <avahi-core/core.h>
44
97
#include <avahi-core/lookup.h>
45
98
#include <avahi-core/log.h>
47
100
#include <avahi-common/malloc.h>
48
101
#include <avahi-common/error.h>
49
102
 
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>
70
 
 
71
 
#ifndef CERT_ROOT
72
 
#define CERT_ROOT "/conf/conf.d/cryptkeyreq/"
73
 
#endif
74
 
#define CERTFILE CERT_ROOT "openpgp-client.txt"
75
 
#define KEYFILE CERT_ROOT "openpgp-client-key.txt"
 
103
/* GnuTLS */
 
104
#include <gnutls/gnutls.h>      /* All GnuTLS types, constants and
 
105
                                   functions:
 
106
                                   gnutls_*
 
107
                                   init_gnutls_session(),
 
108
                                   GNUTLS_* */
 
109
#include <gnutls/openpgp.h>
 
110
                          /* gnutls_certificate_set_openpgp_key_file(),
 
111
                                   GNUTLS_OPENPGP_FMT_BASE64 */
 
112
 
 
113
/* GPGME */
 
114
#include <gpgme.h>              /* All GPGME types, constants and
 
115
                                   functions:
 
116
                                   gpgme_*
 
117
                                   GPGME_PROTOCOL_OpenPGP,
 
118
                                   GPG_ERR_NO_* */
 
119
 
76
120
#define BUFFER_SIZE 256
77
 
#define DH_BITS 1024
 
121
 
 
122
#define PATHDIR "/conf/conf.d/mandos"
 
123
#define SECKEY "seckey.txt"
 
124
#define PUBKEY "pubkey.txt"
78
125
 
79
126
bool debug = false;
 
127
static const char mandos_protocol_version[] = "1";
 
128
const char *argp_program_version = "mandos-client " VERSION;
 
129
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
 
130
static const char sys_class_net[] = "/sys/class/net";
 
131
char *connect_to = NULL;
80
132
 
 
133
/* Used for passing in values through the Avahi callback functions */
81
134
typedef struct {
82
 
  gnutls_session_t session;
 
135
  AvahiSimplePoll *simple_poll;
 
136
  AvahiServer *server;
83
137
  gnutls_certificate_credentials_t cred;
 
138
  unsigned int dh_bits;
84
139
  gnutls_dh_params_t dh_params;
85
 
} encrypted_session;
86
 
 
87
 
 
88
 
ssize_t pgp_packet_decrypt (char *packet, size_t packet_size,
89
 
                            char **new_packet, const char *homedir){
90
 
  gpgme_data_t dh_crypto, dh_plain;
 
140
  const char *priority;
91
141
  gpgme_ctx_t ctx;
 
142
} mandos_context;
 
143
 
 
144
/* global context so signal handler can reach it*/
 
145
mandos_context mc = { .simple_poll = NULL, .server = NULL,
 
146
                      .dh_bits = 1024, .priority = "SECURE256"
 
147
                      ":!CTYPE-X.509:+CTYPE-OPENPGP" };
 
148
 
 
149
sig_atomic_t quit_now = 0;
 
150
int signal_received = 0;
 
151
 
 
152
/*
 
153
 * Make additional room in "buffer" for at least BUFFER_SIZE more
 
154
 * bytes. "buffer_capacity" is how much is currently allocated,
 
155
 * "buffer_length" is how much is already used.
 
156
 */
 
157
size_t incbuffer(char **buffer, size_t buffer_length,
 
158
                  size_t buffer_capacity){
 
159
  if(buffer_length + BUFFER_SIZE > buffer_capacity){
 
160
    *buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
 
161
    if(buffer == NULL){
 
162
      return 0;
 
163
    }
 
164
    buffer_capacity += BUFFER_SIZE;
 
165
  }
 
166
  return buffer_capacity;
 
167
}
 
168
 
 
169
/* 
 
170
 * Initialize GPGME.
 
171
 */
 
172
static bool init_gpgme(const char *seckey,
 
173
                       const char *pubkey, const char *tempdir){
92
174
  gpgme_error_t rc;
93
 
  ssize_t ret;
94
 
  ssize_t new_packet_capacity = 0;
95
 
  ssize_t new_packet_length = 0;
96
175
  gpgme_engine_info_t engine_info;
97
 
 
98
 
  if (debug){
99
 
    fprintf(stderr, "Trying to decrypt OpenPGP packet\n");
 
176
  
 
177
  
 
178
  /*
 
179
   * Helper function to insert pub and seckey to the engine keyring.
 
180
   */
 
181
  bool import_key(const char *filename){
 
182
    int ret;
 
183
    int fd;
 
184
    gpgme_data_t pgp_data;
 
185
    
 
186
    fd = (int)TEMP_FAILURE_RETRY(open(filename, O_RDONLY));
 
187
    if(fd == -1){
 
188
      perror("open");
 
189
      return false;
 
190
    }
 
191
    
 
192
    rc = gpgme_data_new_from_fd(&pgp_data, fd);
 
193
    if(rc != GPG_ERR_NO_ERROR){
 
194
      fprintf(stderr, "bad gpgme_data_new_from_fd: %s: %s\n",
 
195
              gpgme_strsource(rc), gpgme_strerror(rc));
 
196
      return false;
 
197
    }
 
198
    
 
199
    rc = gpgme_op_import(mc.ctx, pgp_data);
 
200
    if(rc != GPG_ERR_NO_ERROR){
 
201
      fprintf(stderr, "bad gpgme_op_import: %s: %s\n",
 
202
              gpgme_strsource(rc), gpgme_strerror(rc));
 
203
      return false;
 
204
    }
 
205
    
 
206
    ret = (int)TEMP_FAILURE_RETRY(close(fd));
 
207
    if(ret == -1){
 
208
      perror("close");
 
209
    }
 
210
    gpgme_data_release(pgp_data);
 
211
    return true;
 
212
  }
 
213
  
 
214
  if(debug){
 
215
    fprintf(stderr, "Initializing GPGME\n");
100
216
  }
101
217
  
102
218
  /* Init GPGME */
103
219
  gpgme_check_version(NULL);
104
 
  gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
 
220
  rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
 
221
  if(rc != GPG_ERR_NO_ERROR){
 
222
    fprintf(stderr, "bad gpgme_engine_check_version: %s: %s\n",
 
223
            gpgme_strsource(rc), gpgme_strerror(rc));
 
224
    return false;
 
225
  }
105
226
  
106
 
  /* Set GPGME home directory */
107
 
  rc = gpgme_get_engine_info (&engine_info);
108
 
  if (rc != GPG_ERR_NO_ERROR){
 
227
    /* Set GPGME home directory for the OpenPGP engine only */
 
228
  rc = gpgme_get_engine_info(&engine_info);
 
229
  if(rc != GPG_ERR_NO_ERROR){
109
230
    fprintf(stderr, "bad gpgme_get_engine_info: %s: %s\n",
110
231
            gpgme_strsource(rc), gpgme_strerror(rc));
111
 
    return -1;
 
232
    return false;
112
233
  }
113
234
  while(engine_info != NULL){
114
235
    if(engine_info->protocol == GPGME_PROTOCOL_OpenPGP){
115
236
      gpgme_set_engine_info(GPGME_PROTOCOL_OpenPGP,
116
 
                            engine_info->file_name, homedir);
 
237
                            engine_info->file_name, tempdir);
117
238
      break;
118
239
    }
119
240
    engine_info = engine_info->next;
120
241
  }
121
242
  if(engine_info == NULL){
122
 
    fprintf(stderr, "Could not set home dir to %s\n", homedir);
123
 
    return -1;
124
 
  }
125
 
  
126
 
  /* Create new GPGME data buffer from packet buffer */
127
 
  rc = gpgme_data_new_from_mem(&dh_crypto, packet, packet_size, 0);
128
 
  if (rc != GPG_ERR_NO_ERROR){
 
243
    fprintf(stderr, "Could not set GPGME home dir to %s\n", tempdir);
 
244
    return false;
 
245
  }
 
246
  
 
247
  /* Create new GPGME "context" */
 
248
  rc = gpgme_new(&(mc.ctx));
 
249
  if(rc != GPG_ERR_NO_ERROR){
 
250
    fprintf(stderr, "bad gpgme_new: %s: %s\n",
 
251
            gpgme_strsource(rc), gpgme_strerror(rc));
 
252
    return false;
 
253
  }
 
254
  
 
255
  if(not import_key(pubkey) or not import_key(seckey)){
 
256
    return false;
 
257
  }
 
258
  
 
259
  return true;
 
260
}
 
261
 
 
262
/* 
 
263
 * Decrypt OpenPGP data.
 
264
 * Returns -1 on error
 
265
 */
 
266
static ssize_t pgp_packet_decrypt(const char *cryptotext,
 
267
                                  size_t crypto_size,
 
268
                                  char **plaintext){
 
269
  gpgme_data_t dh_crypto, dh_plain;
 
270
  gpgme_error_t rc;
 
271
  ssize_t ret;
 
272
  size_t plaintext_capacity = 0;
 
273
  ssize_t plaintext_length = 0;
 
274
  
 
275
  if(debug){
 
276
    fprintf(stderr, "Trying to decrypt OpenPGP data\n");
 
277
  }
 
278
  
 
279
  /* Create new GPGME data buffer from memory cryptotext */
 
280
  rc = gpgme_data_new_from_mem(&dh_crypto, cryptotext, crypto_size,
 
281
                               0);
 
282
  if(rc != GPG_ERR_NO_ERROR){
129
283
    fprintf(stderr, "bad gpgme_data_new_from_mem: %s: %s\n",
130
284
            gpgme_strsource(rc), gpgme_strerror(rc));
131
285
    return -1;
133
287
  
134
288
  /* Create new empty GPGME data buffer for the plaintext */
135
289
  rc = gpgme_data_new(&dh_plain);
136
 
  if (rc != GPG_ERR_NO_ERROR){
 
290
  if(rc != GPG_ERR_NO_ERROR){
137
291
    fprintf(stderr, "bad gpgme_data_new: %s: %s\n",
138
292
            gpgme_strsource(rc), gpgme_strerror(rc));
139
 
    return -1;
140
 
  }
141
 
  
142
 
  /* Create new GPGME "context" */
143
 
  rc = gpgme_new(&ctx);
144
 
  if (rc != GPG_ERR_NO_ERROR){
145
 
    fprintf(stderr, "bad gpgme_new: %s: %s\n",
146
 
            gpgme_strsource(rc), gpgme_strerror(rc));
147
 
    return -1;
148
 
  }
149
 
  
150
 
  /* Decrypt data from the FILE pointer to the plaintext data
151
 
     buffer */
152
 
  rc = gpgme_op_decrypt(ctx, dh_crypto, dh_plain);
153
 
  if (rc != GPG_ERR_NO_ERROR){
 
293
    gpgme_data_release(dh_crypto);
 
294
    return -1;
 
295
  }
 
296
  
 
297
  /* Decrypt data from the cryptotext data buffer to the plaintext
 
298
     data buffer */
 
299
  rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);
 
300
  if(rc != GPG_ERR_NO_ERROR){
154
301
    fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
155
302
            gpgme_strsource(rc), gpgme_strerror(rc));
156
 
    return -1;
157
 
  }
158
 
 
159
 
  if(debug){
160
 
    fprintf(stderr, "Decryption of OpenPGP packet succeeded\n");
161
 
  }
162
 
 
163
 
  if (debug){
164
 
    gpgme_decrypt_result_t result;
165
 
    result = gpgme_op_decrypt_result(ctx);
166
 
    if (result == NULL){
167
 
      fprintf(stderr, "gpgme_op_decrypt_result failed\n");
168
 
    } else {
169
 
      fprintf(stderr, "Unsupported algorithm: %s\n",
170
 
              result->unsupported_algorithm);
171
 
      fprintf(stderr, "Wrong key usage: %d\n",
172
 
              result->wrong_key_usage);
173
 
      if(result->file_name != NULL){
174
 
        fprintf(stderr, "File name: %s\n", result->file_name);
175
 
      }
176
 
      gpgme_recipient_t recipient;
177
 
      recipient = result->recipients;
178
 
      if(recipient){
 
303
    plaintext_length = -1;
 
304
    if(debug){
 
305
      gpgme_decrypt_result_t result;
 
306
      result = gpgme_op_decrypt_result(mc.ctx);
 
307
      if(result == NULL){
 
308
        fprintf(stderr, "gpgme_op_decrypt_result failed\n");
 
309
      } else {
 
310
        fprintf(stderr, "Unsupported algorithm: %s\n",
 
311
                result->unsupported_algorithm);
 
312
        fprintf(stderr, "Wrong key usage: %u\n",
 
313
                result->wrong_key_usage);
 
314
        if(result->file_name != NULL){
 
315
          fprintf(stderr, "File name: %s\n", result->file_name);
 
316
        }
 
317
        gpgme_recipient_t recipient;
 
318
        recipient = result->recipients;
179
319
        while(recipient != NULL){
180
320
          fprintf(stderr, "Public key algorithm: %s\n",
181
321
                  gpgme_pubkey_algo_name(recipient->pubkey_algo));
187
327
        }
188
328
      }
189
329
    }
 
330
    goto decrypt_end;
190
331
  }
191
332
  
192
 
  /* Delete the GPGME FILE pointer cryptotext data buffer */
193
 
  gpgme_data_release(dh_crypto);
 
333
  if(debug){
 
334
    fprintf(stderr, "Decryption of OpenPGP data succeeded\n");
 
335
  }
194
336
  
195
337
  /* Seek back to the beginning of the GPGME plaintext data buffer */
196
 
  gpgme_data_seek(dh_plain, (off_t) 0, SEEK_SET);
197
 
 
198
 
  *new_packet = 0;
 
338
  if(gpgme_data_seek(dh_plain, (off_t)0, SEEK_SET) == -1){
 
339
    perror("gpgme_data_seek");
 
340
    plaintext_length = -1;
 
341
    goto decrypt_end;
 
342
  }
 
343
  
 
344
  *plaintext = NULL;
199
345
  while(true){
200
 
    if (new_packet_length + BUFFER_SIZE > new_packet_capacity){
201
 
      *new_packet = realloc(*new_packet,
202
 
                            (unsigned int)new_packet_capacity
203
 
                            + BUFFER_SIZE);
204
 
      if (*new_packet == NULL){
205
 
        perror("realloc");
206
 
        return -1;
207
 
      }
208
 
      new_packet_capacity += BUFFER_SIZE;
 
346
    plaintext_capacity = incbuffer(plaintext,
 
347
                                      (size_t)plaintext_length,
 
348
                                      plaintext_capacity);
 
349
    if(plaintext_capacity == 0){
 
350
        perror("incbuffer");
 
351
        plaintext_length = -1;
 
352
        goto decrypt_end;
209
353
    }
210
354
    
211
 
    ret = gpgme_data_read(dh_plain, *new_packet + new_packet_length,
 
355
    ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
212
356
                          BUFFER_SIZE);
213
357
    /* Print the data, if any */
214
 
    if (ret == 0){
 
358
    if(ret == 0){
 
359
      /* EOF */
215
360
      break;
216
361
    }
217
362
    if(ret < 0){
218
363
      perror("gpgme_data_read");
219
 
      return -1;
220
 
    }
221
 
    new_packet_length += ret;
222
 
  }
223
 
 
224
 
  /* FIXME: check characters before printing to screen so to not print
225
 
     terminal control characters */
226
 
  /*   if(debug){ */
227
 
  /*     fprintf(stderr, "decrypted password is: "); */
228
 
  /*     fwrite(*new_packet, 1, new_packet_length, stderr); */
229
 
  /*     fprintf(stderr, "\n"); */
230
 
  /*   } */
 
364
      plaintext_length = -1;
 
365
      goto decrypt_end;
 
366
    }
 
367
    plaintext_length += ret;
 
368
  }
 
369
  
 
370
  if(debug){
 
371
    fprintf(stderr, "Decrypted password is: ");
 
372
    for(ssize_t i = 0; i < plaintext_length; i++){
 
373
      fprintf(stderr, "%02hhX ", (*plaintext)[i]);
 
374
    }
 
375
    fprintf(stderr, "\n");
 
376
  }
 
377
  
 
378
 decrypt_end:
 
379
  
 
380
  /* Delete the GPGME cryptotext data buffer */
 
381
  gpgme_data_release(dh_crypto);
231
382
  
232
383
  /* Delete the GPGME plaintext data buffer */
233
384
  gpgme_data_release(dh_plain);
234
 
  return new_packet_length;
 
385
  return plaintext_length;
235
386
}
236
387
 
237
 
static const char * safer_gnutls_strerror (int value) {
238
 
  const char *ret = gnutls_strerror (value);
239
 
  if (ret == NULL)
 
388
static const char * safer_gnutls_strerror(int value){
 
389
  const char *ret = gnutls_strerror(value); /* Spurious warning from
 
390
                                               -Wunreachable-code */
 
391
  if(ret == NULL)
240
392
    ret = "(unknown)";
241
393
  return ret;
242
394
}
243
395
 
244
 
void debuggnutls(__attribute__((unused)) int level,
245
 
                 const char* string){
246
 
  fprintf(stderr, "%s", string);
 
396
/* GnuTLS log function callback */
 
397
static void debuggnutls(__attribute__((unused)) int level,
 
398
                        const char* string){
 
399
  fprintf(stderr, "GnuTLS: %s", string);
247
400
}
248
401
 
249
 
int initgnutls(encrypted_session *es){
250
 
  const char *err;
 
402
static int init_gnutls_global(const char *pubkeyfilename,
 
403
                              const char *seckeyfilename){
251
404
  int ret;
252
405
  
253
406
  if(debug){
254
407
    fprintf(stderr, "Initializing GnuTLS\n");
255
408
  }
256
409
  
257
 
  if ((ret = gnutls_global_init ())
258
 
      != GNUTLS_E_SUCCESS) {
259
 
    fprintf (stderr, "global_init: %s\n", safer_gnutls_strerror(ret));
 
410
  ret = gnutls_global_init();
 
411
  if(ret != GNUTLS_E_SUCCESS){
 
412
    fprintf(stderr, "GnuTLS global_init: %s\n",
 
413
            safer_gnutls_strerror(ret));
260
414
    return -1;
261
415
  }
262
 
 
263
 
  if (debug){
 
416
  
 
417
  if(debug){
 
418
    /* "Use a log level over 10 to enable all debugging options."
 
419
     * - GnuTLS manual
 
420
     */
264
421
    gnutls_global_set_log_level(11);
265
422
    gnutls_global_set_log_function(debuggnutls);
266
423
  }
267
424
  
268
 
  /* openpgp credentials */
269
 
  if ((ret = gnutls_certificate_allocate_credentials (&es->cred))
270
 
      != GNUTLS_E_SUCCESS) {
271
 
    fprintf (stderr, "memory error: %s\n",
272
 
             safer_gnutls_strerror(ret));
 
425
  /* OpenPGP credentials */
 
426
  gnutls_certificate_allocate_credentials(&mc.cred);
 
427
  if(ret != GNUTLS_E_SUCCESS){
 
428
    fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning
 
429
                                                    from
 
430
                                                    -Wunreachable-code
 
431
                                                 */
 
432
            safer_gnutls_strerror(ret));
 
433
    gnutls_global_deinit();
273
434
    return -1;
274
435
  }
275
436
  
276
437
  if(debug){
277
 
    fprintf(stderr, "Attempting to use OpenPGP certificate %s"
278
 
            " and keyfile %s as GnuTLS credentials\n", CERTFILE,
279
 
            KEYFILE);
 
438
    fprintf(stderr, "Attempting to use OpenPGP public key %s and"
 
439
            " secret key %s as GnuTLS credentials\n", pubkeyfilename,
 
440
            seckeyfilename);
280
441
  }
281
442
  
282
443
  ret = gnutls_certificate_set_openpgp_key_file
283
 
    (es->cred, CERTFILE, KEYFILE, GNUTLS_OPENPGP_FMT_BASE64);
284
 
  if (ret != GNUTLS_E_SUCCESS) {
285
 
    fprintf
286
 
      (stderr, "Error[%d] while reading the OpenPGP key pair ('%s',"
287
 
       " '%s')\n",
288
 
       ret, CERTFILE, KEYFILE);
289
 
    fprintf(stdout, "The Error is: %s\n",
290
 
            safer_gnutls_strerror(ret));
291
 
    return -1;
292
 
  }
293
 
  
294
 
  //GnuTLS server initialization
295
 
  if ((ret = gnutls_dh_params_init (&es->dh_params))
296
 
      != GNUTLS_E_SUCCESS) {
297
 
    fprintf (stderr, "Error in dh parameter initialization: %s\n",
298
 
             safer_gnutls_strerror(ret));
299
 
    return -1;
300
 
  }
301
 
  
302
 
  if ((ret = gnutls_dh_params_generate2 (es->dh_params, DH_BITS))
303
 
      != GNUTLS_E_SUCCESS) {
304
 
    fprintf (stderr, "Error in prime generation: %s\n",
305
 
             safer_gnutls_strerror(ret));
306
 
    return -1;
307
 
  }
308
 
  
309
 
  gnutls_certificate_set_dh_params (es->cred, es->dh_params);
310
 
  
311
 
  // GnuTLS session creation
312
 
  if ((ret = gnutls_init (&es->session, GNUTLS_SERVER))
313
 
      != GNUTLS_E_SUCCESS){
 
444
    (mc.cred, pubkeyfilename, seckeyfilename,
 
445
     GNUTLS_OPENPGP_FMT_BASE64);
 
446
  if(ret != GNUTLS_E_SUCCESS){
 
447
    fprintf(stderr,
 
448
            "Error[%d] while reading the OpenPGP key pair ('%s',"
 
449
            " '%s')\n", ret, pubkeyfilename, seckeyfilename);
 
450
    fprintf(stderr, "The GnuTLS error is: %s\n",
 
451
            safer_gnutls_strerror(ret));
 
452
    goto globalfail;
 
453
  }
 
454
  
 
455
  /* GnuTLS server initialization */
 
456
  ret = gnutls_dh_params_init(&mc.dh_params);
 
457
  if(ret != GNUTLS_E_SUCCESS){
 
458
    fprintf(stderr, "Error in GnuTLS DH parameter initialization:"
 
459
            " %s\n", safer_gnutls_strerror(ret));
 
460
    goto globalfail;
 
461
  }
 
462
  ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);
 
463
  if(ret != GNUTLS_E_SUCCESS){
 
464
    fprintf(stderr, "Error in GnuTLS prime generation: %s\n",
 
465
            safer_gnutls_strerror(ret));
 
466
    goto globalfail;
 
467
  }
 
468
  
 
469
  gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);
 
470
  
 
471
  return 0;
 
472
  
 
473
 globalfail:
 
474
  
 
475
  gnutls_certificate_free_credentials(mc.cred);
 
476
  gnutls_global_deinit();
 
477
  gnutls_dh_params_deinit(mc.dh_params);
 
478
  return -1;
 
479
}
 
480
 
 
481
static int init_gnutls_session(gnutls_session_t *session){
 
482
  int ret;
 
483
  /* GnuTLS session creation */
 
484
  do {
 
485
    ret = gnutls_init(session, GNUTLS_SERVER);
 
486
    if(quit_now){
 
487
      return -1;
 
488
    }
 
489
  } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
 
490
  if(ret != GNUTLS_E_SUCCESS){
314
491
    fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
315
492
            safer_gnutls_strerror(ret));
316
493
  }
317
494
  
318
 
  if ((ret = gnutls_priority_set_direct (es->session, "NORMAL", &err))
319
 
      != GNUTLS_E_SUCCESS) {
320
 
    fprintf(stderr, "Syntax error at: %s\n", err);
321
 
    fprintf(stderr, "GnuTLS error: %s\n",
322
 
            safer_gnutls_strerror(ret));
323
 
    return -1;
 
495
  {
 
496
    const char *err;
 
497
    do {
 
498
      ret = gnutls_priority_set_direct(*session, mc.priority, &err);
 
499
      if(quit_now){
 
500
        gnutls_deinit(*session);
 
501
        return -1;
 
502
      }
 
503
    } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
 
504
    if(ret != GNUTLS_E_SUCCESS){
 
505
      fprintf(stderr, "Syntax error at: %s\n", err);
 
506
      fprintf(stderr, "GnuTLS error: %s\n",
 
507
              safer_gnutls_strerror(ret));
 
508
      gnutls_deinit(*session);
 
509
      return -1;
 
510
    }
324
511
  }
325
512
  
326
 
  if ((ret = gnutls_credentials_set
327
 
       (es->session, GNUTLS_CRD_CERTIFICATE, es->cred))
328
 
      != GNUTLS_E_SUCCESS) {
329
 
    fprintf(stderr, "Error setting a credentials set: %s\n",
 
513
  do {
 
514
    ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
 
515
                                 mc.cred);
 
516
    if(quit_now){
 
517
      gnutls_deinit(*session);
 
518
      return -1;
 
519
    }
 
520
  } while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
 
521
  if(ret != GNUTLS_E_SUCCESS){
 
522
    fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
330
523
            safer_gnutls_strerror(ret));
 
524
    gnutls_deinit(*session);
331
525
    return -1;
332
526
  }
333
527
  
334
528
  /* ignore client certificate if any. */
335
 
  gnutls_certificate_server_set_request (es->session,
336
 
                                         GNUTLS_CERT_IGNORE);
 
529
  gnutls_certificate_server_set_request(*session, GNUTLS_CERT_IGNORE);
337
530
  
338
 
  gnutls_dh_set_prime_bits (es->session, DH_BITS);
 
531
  gnutls_dh_set_prime_bits(*session, mc.dh_bits);
339
532
  
340
533
  return 0;
341
534
}
342
535
 
343
 
void empty_log(__attribute__((unused)) AvahiLogLevel level,
344
 
               __attribute__((unused)) const char *txt){}
 
536
/* Avahi log function callback */
 
537
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
 
538
                      __attribute__((unused)) const char *txt){}
345
539
 
346
 
int start_mandos_communication(const char *ip, uint16_t port,
347
 
                               unsigned int if_index){
348
 
  int ret, tcp_sd;
349
 
  struct sockaddr_in6 to;
350
 
  encrypted_session es;
 
540
/* Called when a Mandos server is found */
 
541
static int start_mandos_communication(const char *ip, uint16_t port,
 
542
                                      AvahiIfIndex if_index,
 
543
                                      int af){
 
544
  int ret, tcp_sd = -1;
 
545
  ssize_t sret;
 
546
  union {
 
547
    struct sockaddr_in in;
 
548
    struct sockaddr_in6 in6;
 
549
  } to;
351
550
  char *buffer = NULL;
352
 
  char *decrypted_buffer;
 
551
  char *decrypted_buffer = NULL;
353
552
  size_t buffer_length = 0;
354
553
  size_t buffer_capacity = 0;
355
 
  ssize_t decrypted_buffer_size;
356
 
  size_t written = 0;
357
 
  int retval = 0;
358
 
  char interface[IF_NAMESIZE];
 
554
  size_t written;
 
555
  int retval = -1;
 
556
  gnutls_session_t session;
 
557
  int pf;                       /* Protocol family */
 
558
  
 
559
  errno = 0;
 
560
  
 
561
  if(quit_now){
 
562
    errno = EINTR;
 
563
    return -1;
 
564
  }
 
565
  
 
566
  switch(af){
 
567
  case AF_INET6:
 
568
    pf = PF_INET6;
 
569
    break;
 
570
  case AF_INET:
 
571
    pf = PF_INET;
 
572
    break;
 
573
  default:
 
574
    fprintf(stderr, "Bad address family: %d\n", af);
 
575
    errno = EINVAL;
 
576
    return -1;
 
577
  }
 
578
  
 
579
  ret = init_gnutls_session(&session);
 
580
  if(ret != 0){
 
581
    return -1;
 
582
  }
359
583
  
360
584
  if(debug){
361
 
    fprintf(stderr, "Setting up a tcp connection to %s\n", ip);
 
585
    fprintf(stderr, "Setting up a TCP connection to %s, port %" PRIu16
 
586
            "\n", ip, port);
362
587
  }
363
588
  
364
 
  tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
365
 
  if(tcp_sd < 0) {
 
589
  tcp_sd = socket(pf, SOCK_STREAM, 0);
 
590
  if(tcp_sd < 0){
 
591
    int e = errno;
366
592
    perror("socket");
367
 
    return -1;
368
 
  }
369
 
  
370
 
  if(if_indextoname(if_index, interface) == NULL){
371
 
    if(debug){
372
 
      perror("if_indextoname");
373
 
    }
374
 
    return -1;
375
 
  }
376
 
  
377
 
  if(debug){
378
 
    fprintf(stderr, "Binding to interface %s\n", interface);
379
 
  }
380
 
  
381
 
  memset(&to,0,sizeof(to));     /* Spurious warning */
382
 
  to.sin6_family = AF_INET6;
383
 
  ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
384
 
  if (ret < 0 ){
 
593
    errno = e;
 
594
    goto mandos_end;
 
595
  }
 
596
  
 
597
  if(quit_now){
 
598
    errno = EINTR;
 
599
    goto mandos_end;
 
600
  }
 
601
  
 
602
  memset(&to, 0, sizeof(to));
 
603
  if(af == AF_INET6){
 
604
    to.in6.sin6_family = (sa_family_t)af;
 
605
    ret = inet_pton(af, ip, &to.in6.sin6_addr);
 
606
  } else {                      /* IPv4 */
 
607
    to.in.sin_family = (sa_family_t)af;
 
608
    ret = inet_pton(af, ip, &to.in.sin_addr);
 
609
  }
 
610
  if(ret < 0 ){
 
611
    int e = errno;
385
612
    perror("inet_pton");
386
 
    return -1;
387
 
  }  
 
613
    errno = e;
 
614
    goto mandos_end;
 
615
  }
388
616
  if(ret == 0){
 
617
    int e = errno;
389
618
    fprintf(stderr, "Bad address: %s\n", ip);
390
 
    return -1;
391
 
  }
392
 
  to.sin6_port = htons(port);   /* Spurious warning */
 
619
    errno = e;
 
620
    goto mandos_end;
 
621
  }
 
622
  if(af == AF_INET6){
 
623
    to.in6.sin6_port = htons(port); /* Spurious warnings from
 
624
                                       -Wconversion and
 
625
                                       -Wunreachable-code */
 
626
    
 
627
    if(IN6_IS_ADDR_LINKLOCAL /* Spurious warnings from */
 
628
       (&to.in6.sin6_addr)){ /* -Wstrict-aliasing=2 or lower and
 
629
                              -Wunreachable-code*/
 
630
      if(if_index == AVAHI_IF_UNSPEC){
 
631
        fprintf(stderr, "An IPv6 link-local address is incomplete"
 
632
                " without a network interface\n");
 
633
        errno = EINVAL;
 
634
        goto mandos_end;
 
635
      }
 
636
      /* Set the network interface number as scope */
 
637
      to.in6.sin6_scope_id = (uint32_t)if_index;
 
638
    }
 
639
  } else {
 
640
    to.in.sin_port = htons(port); /* Spurious warnings from
 
641
                                     -Wconversion and
 
642
                                     -Wunreachable-code */
 
643
  }
393
644
  
394
 
  to.sin6_scope_id = (uint32_t)if_index;
 
645
  if(quit_now){
 
646
    errno = EINTR;
 
647
    goto mandos_end;
 
648
  }
395
649
  
396
650
  if(debug){
397
 
    fprintf(stderr, "Connection to: %s\n", ip);
398
 
  }
399
 
  
400
 
  ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
401
 
  if (ret < 0){
402
 
    perror("connect");
403
 
    return -1;
404
 
  }
405
 
  
406
 
  ret = initgnutls (&es);
407
 
  if (ret != 0){
408
 
    retval = -1;
409
 
    return -1;
410
 
  }
411
 
  
412
 
  gnutls_transport_set_ptr (es.session,
413
 
                            (gnutls_transport_ptr_t) tcp_sd);
 
651
    if(af == AF_INET6 and if_index != AVAHI_IF_UNSPEC){
 
652
      char interface[IF_NAMESIZE];
 
653
      if(if_indextoname((unsigned int)if_index, interface) == NULL){
 
654
        perror("if_indextoname");
 
655
      } else {
 
656
        fprintf(stderr, "Connection to: %s%%%s, port %" PRIu16 "\n",
 
657
                ip, interface, port);
 
658
      }
 
659
    } else {
 
660
      fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
 
661
              port);
 
662
    }
 
663
    char addrstr[(INET_ADDRSTRLEN > INET6_ADDRSTRLEN) ?
 
664
                 INET_ADDRSTRLEN : INET6_ADDRSTRLEN] = "";
 
665
    const char *pcret;
 
666
    if(af == AF_INET6){
 
667
      pcret = inet_ntop(af, &(to.in6.sin6_addr), addrstr,
 
668
                        sizeof(addrstr));
 
669
    } else {
 
670
      pcret = inet_ntop(af, &(to.in.sin_addr), addrstr,
 
671
                        sizeof(addrstr));
 
672
    }
 
673
    if(pcret == NULL){
 
674
      perror("inet_ntop");
 
675
    } else {
 
676
      if(strcmp(addrstr, ip) != 0){
 
677
        fprintf(stderr, "Canonical address form: %s\n", addrstr);
 
678
      }
 
679
    }
 
680
  }
 
681
  
 
682
  if(quit_now){
 
683
    errno = EINTR;
 
684
    goto mandos_end;
 
685
  }
 
686
  
 
687
  if(af == AF_INET6){
 
688
    ret = connect(tcp_sd, &to.in6, sizeof(to));
 
689
  } else {
 
690
    ret = connect(tcp_sd, &to.in, sizeof(to)); /* IPv4 */
 
691
  }
 
692
  if(ret < 0){
 
693
    if ((errno != ECONNREFUSED and errno != ENETUNREACH) or debug){
 
694
      int e = errno;
 
695
      perror("connect");
 
696
      errno = e;
 
697
    }
 
698
    goto mandos_end;
 
699
  }
 
700
  
 
701
  if(quit_now){
 
702
    errno = EINTR;
 
703
    goto mandos_end;
 
704
  }
 
705
  
 
706
  const char *out = mandos_protocol_version;
 
707
  written = 0;
 
708
  while(true){
 
709
    size_t out_size = strlen(out);
 
710
    ret = (int)TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
 
711
                                   out_size - written));
 
712
    if(ret == -1){
 
713
      int e = errno;
 
714
      perror("write");
 
715
      errno = e;
 
716
      goto mandos_end;
 
717
    }
 
718
    written += (size_t)ret;
 
719
    if(written < out_size){
 
720
      continue;
 
721
    } else {
 
722
      if(out == mandos_protocol_version){
 
723
        written = 0;
 
724
        out = "\r\n";
 
725
      } else {
 
726
        break;
 
727
      }
 
728
    }
 
729
  
 
730
    if(quit_now){
 
731
      errno = EINTR;
 
732
      goto mandos_end;
 
733
    }
 
734
  }
414
735
  
415
736
  if(debug){
416
737
    fprintf(stderr, "Establishing TLS session with %s\n", ip);
417
738
  }
418
739
  
419
 
  ret = gnutls_handshake (es.session);
420
 
  
421
 
  if (ret != GNUTLS_E_SUCCESS){
 
740
  if(quit_now){
 
741
    errno = EINTR;
 
742
    goto mandos_end;
 
743
  }
 
744
  
 
745
  gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);
 
746
  
 
747
  if(quit_now){
 
748
    errno = EINTR;
 
749
    goto mandos_end;
 
750
  }
 
751
  
 
752
  do {
 
753
    ret = gnutls_handshake(session);
 
754
    if(quit_now){
 
755
      errno = EINTR;
 
756
      goto mandos_end;
 
757
    }
 
758
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
759
  
 
760
  if(ret != GNUTLS_E_SUCCESS){
422
761
    if(debug){
423
 
      fprintf(stderr, "\n*** Handshake failed ***\n");
424
 
      gnutls_perror (ret);
 
762
      fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
 
763
      gnutls_perror(ret);
425
764
    }
426
 
    retval = -1;
427
 
    goto exit;
 
765
    errno = EPROTO;
 
766
    goto mandos_end;
428
767
  }
429
768
  
430
 
  //Retrieve OpenPGP packet that contains the wanted password
 
769
  /* Read OpenPGP packet that contains the wanted password */
431
770
  
432
771
  if(debug){
433
 
    fprintf(stderr, "Retrieving pgp encrypted password from %s\n",
 
772
    fprintf(stderr, "Retrieving OpenPGP encrypted password from %s\n",
434
773
            ip);
435
774
  }
436
 
 
 
775
  
437
776
  while(true){
438
 
    if (buffer_length + BUFFER_SIZE > buffer_capacity){
439
 
      buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
440
 
      if (buffer == NULL){
441
 
        perror("realloc");
442
 
        goto exit;
443
 
      }
444
 
      buffer_capacity += BUFFER_SIZE;
445
 
    }
446
 
    
447
 
    ret = gnutls_record_recv
448
 
      (es.session, buffer+buffer_length, BUFFER_SIZE);
449
 
    if (ret == 0){
 
777
    
 
778
    if(quit_now){
 
779
      errno = EINTR;
 
780
      goto mandos_end;
 
781
    }
 
782
    
 
783
    buffer_capacity = incbuffer(&buffer, buffer_length,
 
784
                                   buffer_capacity);
 
785
    if(buffer_capacity == 0){
 
786
      int e = errno;
 
787
      perror("incbuffer");
 
788
      errno = e;
 
789
      goto mandos_end;
 
790
    }
 
791
    
 
792
    if(quit_now){
 
793
      errno = EINTR;
 
794
      goto mandos_end;
 
795
    }
 
796
    
 
797
    sret = gnutls_record_recv(session, buffer+buffer_length,
 
798
                              BUFFER_SIZE);
 
799
    if(sret == 0){
450
800
      break;
451
801
    }
452
 
    if (ret < 0){
453
 
      switch(ret){
 
802
    if(sret < 0){
 
803
      switch(sret){
454
804
      case GNUTLS_E_INTERRUPTED:
455
805
      case GNUTLS_E_AGAIN:
456
806
        break;
457
807
      case GNUTLS_E_REHANDSHAKE:
458
 
        ret = gnutls_handshake (es.session);
459
 
        if (ret < 0){
460
 
          fprintf(stderr, "\n*** Handshake failed ***\n");
461
 
          gnutls_perror (ret);
462
 
          retval = -1;
463
 
          goto exit;
 
808
        do {
 
809
          ret = gnutls_handshake(session);
 
810
          
 
811
          if(quit_now){
 
812
            errno = EINTR;
 
813
            goto mandos_end;
 
814
          }
 
815
        } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
816
        if(ret < 0){
 
817
          fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
 
818
          gnutls_perror(ret);
 
819
          errno = EPROTO;
 
820
          goto mandos_end;
464
821
        }
465
822
        break;
466
823
      default:
467
824
        fprintf(stderr, "Unknown error while reading data from"
468
 
                " encrypted session with mandos server\n");
469
 
        retval = -1;
470
 
        gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
471
 
        goto exit;
 
825
                " encrypted session with Mandos server\n");
 
826
        gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
827
        errno = EIO;
 
828
        goto mandos_end;
472
829
      }
473
830
    } else {
474
 
      buffer_length += (size_t) ret;
475
 
    }
476
 
  }
477
 
  
478
 
  if (buffer_length > 0){
 
831
      buffer_length += (size_t) sret;
 
832
    }
 
833
  }
 
834
  
 
835
  if(debug){
 
836
    fprintf(stderr, "Closing TLS session\n");
 
837
  }
 
838
  
 
839
  if(quit_now){
 
840
    errno = EINTR;
 
841
    goto mandos_end;
 
842
  }
 
843
  
 
844
  do {
 
845
    ret = gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
846
    if(quit_now){
 
847
      errno = EINTR;
 
848
      goto mandos_end;
 
849
    }
 
850
  } while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
 
851
  
 
852
  if(buffer_length > 0){
 
853
    ssize_t decrypted_buffer_size;
479
854
    decrypted_buffer_size = pgp_packet_decrypt(buffer,
480
855
                                               buffer_length,
481
 
                                               &decrypted_buffer,
482
 
                                               CERT_ROOT);
483
 
    if (decrypted_buffer_size >= 0){
484
 
      while(written < decrypted_buffer_size){
485
 
        ret = (int)fwrite (decrypted_buffer + written, 1,
486
 
                           (size_t)decrypted_buffer_size - written,
487
 
                           stdout);
 
856
                                               &decrypted_buffer);
 
857
    if(decrypted_buffer_size >= 0){
 
858
      
 
859
      written = 0;
 
860
      while(written < (size_t) decrypted_buffer_size){
 
861
        if(quit_now){
 
862
          errno = EINTR;
 
863
          goto mandos_end;
 
864
        }
 
865
        
 
866
        ret = (int)fwrite(decrypted_buffer + written, 1,
 
867
                          (size_t)decrypted_buffer_size - written,
 
868
                          stdout);
488
869
        if(ret == 0 and ferror(stdout)){
 
870
          int e = errno;
489
871
          if(debug){
490
872
            fprintf(stderr, "Error writing encrypted data: %s\n",
491
873
                    strerror(errno));
492
874
          }
493
 
          retval = -1;
494
 
          break;
 
875
          errno = e;
 
876
          goto mandos_end;
495
877
        }
496
878
        written += (size_t)ret;
497
879
      }
498
 
      free(decrypted_buffer);
499
 
    } else {
 
880
      retval = 0;
 
881
    }
 
882
  }
 
883
  
 
884
  /* Shutdown procedure */
 
885
  
 
886
 mandos_end:
 
887
  {
 
888
    int e = errno;
 
889
    free(decrypted_buffer);
 
890
    free(buffer);
 
891
    if(tcp_sd >= 0){
 
892
      ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
 
893
    }
 
894
    if(ret == -1){
 
895
      if(e == 0){
 
896
        e = errno;
 
897
      }
 
898
      perror("close");
 
899
    }
 
900
    gnutls_deinit(session);
 
901
    if(quit_now){
 
902
      e = EINTR;
500
903
      retval = -1;
501
904
    }
502
 
  }
503
 
 
504
 
  //shutdown procedure
505
 
 
506
 
  if(debug){
507
 
    fprintf(stderr, "Closing TLS session\n");
508
 
  }
509
 
 
510
 
  free(buffer);
511
 
  gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
512
 
 exit:
513
 
  close(tcp_sd);
514
 
  gnutls_deinit (es.session);
515
 
  gnutls_certificate_free_credentials (es.cred);
516
 
  gnutls_global_deinit ();
 
905
    errno = e;
 
906
  }
517
907
  return retval;
518
908
}
519
909
 
520
 
static AvahiSimplePoll *simple_poll = NULL;
521
 
static AvahiServer *server = NULL;
522
 
 
523
 
static void resolve_callback(
524
 
    AvahiSServiceResolver *r,
525
 
    AvahiIfIndex interface,
526
 
    AVAHI_GCC_UNUSED AvahiProtocol protocol,
527
 
    AvahiResolverEvent event,
528
 
    const char *name,
529
 
    const char *type,
530
 
    const char *domain,
531
 
    const char *host_name,
532
 
    const AvahiAddress *address,
533
 
    uint16_t port,
534
 
    AVAHI_GCC_UNUSED AvahiStringList *txt,
535
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
536
 
    AVAHI_GCC_UNUSED void* userdata) {
537
 
    
538
 
  assert(r);                    /* Spurious warning */
 
910
static void resolve_callback(AvahiSServiceResolver *r,
 
911
                             AvahiIfIndex interface,
 
912
                             AvahiProtocol proto,
 
913
                             AvahiResolverEvent event,
 
914
                             const char *name,
 
915
                             const char *type,
 
916
                             const char *domain,
 
917
                             const char *host_name,
 
918
                             const AvahiAddress *address,
 
919
                             uint16_t port,
 
920
                             AVAHI_GCC_UNUSED AvahiStringList *txt,
 
921
                             AVAHI_GCC_UNUSED AvahiLookupResultFlags
 
922
                             flags,
 
923
                             AVAHI_GCC_UNUSED void* userdata){
 
924
  assert(r);
539
925
  
540
926
  /* Called whenever a service has been resolved successfully or
541
927
     timed out */
542
928
  
543
 
  switch (event) {
 
929
  if(quit_now){
 
930
    return;
 
931
  }
 
932
  
 
933
  switch(event){
544
934
  default:
545
935
  case AVAHI_RESOLVER_FAILURE:
546
 
    fprintf(stderr, "(Resolver) Failed to resolve service '%s' of"
547
 
            " type '%s' in domain '%s': %s\n", name, type, domain,
548
 
            avahi_strerror(avahi_server_errno(server)));
 
936
    fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
 
937
            " of type '%s' in domain '%s': %s\n", name, type, domain,
 
938
            avahi_strerror(avahi_server_errno(mc.server)));
549
939
    break;
550
940
    
551
941
  case AVAHI_RESOLVER_FOUND:
553
943
      char ip[AVAHI_ADDRESS_STR_MAX];
554
944
      avahi_address_snprint(ip, sizeof(ip), address);
555
945
      if(debug){
556
 
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s) on"
557
 
                " port %d\n", name, host_name, ip, port);
 
946
        fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
 
947
                PRIdMAX ") on port %" PRIu16 "\n", name, host_name,
 
948
                ip, (intmax_t)interface, port);
558
949
      }
559
 
      int ret = start_mandos_communication(ip, port,
560
 
                                           (unsigned int) interface);
561
 
      if (ret == 0){
562
 
        exit(EXIT_SUCCESS);
 
950
      int ret = start_mandos_communication(ip, port, interface,
 
951
                                           avahi_proto_to_af(proto));
 
952
      if(ret == 0){
 
953
        avahi_simple_poll_quit(mc.simple_poll);
563
954
      }
564
955
    }
565
956
  }
566
957
  avahi_s_service_resolver_free(r);
567
958
}
568
959
 
569
 
static void browse_callback(
570
 
    AvahiSServiceBrowser *b,
571
 
    AvahiIfIndex interface,
572
 
    AvahiProtocol protocol,
573
 
    AvahiBrowserEvent event,
574
 
    const char *name,
575
 
    const char *type,
576
 
    const char *domain,
577
 
    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
578
 
    void* userdata) {
579
 
    
580
 
    AvahiServer *s = userdata;
581
 
    assert(b);                  /* Spurious warning */
582
 
    
583
 
    /* Called whenever a new services becomes available on the LAN or
584
 
       is removed from the LAN */
585
 
    
586
 
    switch (event) {
 
960
static void browse_callback(AvahiSServiceBrowser *b,
 
961
                            AvahiIfIndex interface,
 
962
                            AvahiProtocol protocol,
 
963
                            AvahiBrowserEvent event,
 
964
                            const char *name,
 
965
                            const char *type,
 
966
                            const char *domain,
 
967
                            AVAHI_GCC_UNUSED AvahiLookupResultFlags
 
968
                            flags,
 
969
                            AVAHI_GCC_UNUSED void* userdata){
 
970
  assert(b);
 
971
  
 
972
  /* Called whenever a new services becomes available on the LAN or
 
973
     is removed from the LAN */
 
974
  
 
975
  if(quit_now){
 
976
    return;
 
977
  }
 
978
  
 
979
  switch(event){
 
980
  default:
 
981
  case AVAHI_BROWSER_FAILURE:
 
982
    
 
983
    fprintf(stderr, "(Avahi browser) %s\n",
 
984
            avahi_strerror(avahi_server_errno(mc.server)));
 
985
    avahi_simple_poll_quit(mc.simple_poll);
 
986
    return;
 
987
    
 
988
  case AVAHI_BROWSER_NEW:
 
989
    /* We ignore the returned Avahi resolver object. In the callback
 
990
       function we free it. If the Avahi server is terminated before
 
991
       the callback function is called the Avahi server will free the
 
992
       resolver for us. */
 
993
    
 
994
    if(avahi_s_service_resolver_new(mc.server, interface, protocol,
 
995
                                    name, type, domain, protocol, 0,
 
996
                                    resolve_callback, NULL) == NULL)
 
997
      fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
 
998
              name, avahi_strerror(avahi_server_errno(mc.server)));
 
999
    break;
 
1000
    
 
1001
  case AVAHI_BROWSER_REMOVE:
 
1002
    break;
 
1003
    
 
1004
  case AVAHI_BROWSER_ALL_FOR_NOW:
 
1005
  case AVAHI_BROWSER_CACHE_EXHAUSTED:
 
1006
    if(debug){
 
1007
      fprintf(stderr, "No Mandos server found, still searching...\n");
 
1008
    }
 
1009
    break;
 
1010
  }
 
1011
}
 
1012
 
 
1013
/* stop main loop after sigterm has been called */
 
1014
static void handle_sigterm(int sig){
 
1015
  if(quit_now){
 
1016
    return;
 
1017
  }
 
1018
  quit_now = 1;
 
1019
  signal_received = sig;
 
1020
  int old_errno = errno;
 
1021
  if(mc.simple_poll != NULL){
 
1022
    avahi_simple_poll_quit(mc.simple_poll);
 
1023
  }
 
1024
  errno = old_errno;
 
1025
}
 
1026
 
 
1027
/* 
 
1028
 * This function determines if a directory entry in /sys/class/net
 
1029
 * corresponds to an acceptable network device.
 
1030
 * (This function is passed to scandir(3) as a filter function.)
 
1031
 */
 
1032
int good_interface(const struct dirent *if_entry){
 
1033
  ssize_t ssret;
 
1034
  char *flagname = NULL;
 
1035
  if(if_entry->d_name[0] == '.'){
 
1036
    return 0;
 
1037
  }
 
1038
  int ret = asprintf(&flagname, "%s/%s/flags", sys_class_net,
 
1039
                     if_entry->d_name);
 
1040
  if(ret < 0){
 
1041
    perror("asprintf");
 
1042
    return 0;
 
1043
  }
 
1044
  int flags_fd = (int)TEMP_FAILURE_RETRY(open(flagname, O_RDONLY));
 
1045
  if(flags_fd == -1){
 
1046
    perror("open");
 
1047
    free(flagname);
 
1048
    return 0;
 
1049
  }
 
1050
  free(flagname);
 
1051
  typedef short ifreq_flags;    /* ifreq.ifr_flags in netdevice(7) */
 
1052
  /* read line from flags_fd */
 
1053
  ssize_t to_read = (sizeof(ifreq_flags)*2)+3; /* "0x1003\n" */
 
1054
  char *flagstring = malloc((size_t)to_read+1); /* +1 for final \0 */
 
1055
  flagstring[(size_t)to_read] = '\0';
 
1056
  if(flagstring == NULL){
 
1057
    perror("malloc");
 
1058
    close(flags_fd);
 
1059
    return 0;
 
1060
  }
 
1061
  while(to_read > 0){
 
1062
    ssret = (ssize_t)TEMP_FAILURE_RETRY(read(flags_fd, flagstring,
 
1063
                                             (size_t)to_read));
 
1064
    if(ssret == -1){
 
1065
      perror("read");
 
1066
      free(flagstring);
 
1067
      close(flags_fd);
 
1068
      return 0;
 
1069
    }
 
1070
    to_read -= ssret;
 
1071
    if(ssret == 0){
 
1072
      break;
 
1073
    }
 
1074
  }
 
1075
  close(flags_fd);
 
1076
  intmax_t tmpmax;
 
1077
  char *tmp;
 
1078
  errno = 0;
 
1079
  tmpmax = strtoimax(flagstring, &tmp, 0);
 
1080
  if(errno != 0 or tmp == flagstring or (*tmp != '\0'
 
1081
                                         and not (isspace(*tmp)))
 
1082
     or tmpmax != (ifreq_flags)tmpmax){
 
1083
    if(debug){
 
1084
      fprintf(stderr, "Invalid flags \"%s\" for interface \"%s\"\n",
 
1085
              flagstring, if_entry->d_name);
 
1086
    }
 
1087
    free(flagstring);
 
1088
    return 0;
 
1089
  }
 
1090
  free(flagstring);
 
1091
  ifreq_flags flags = (ifreq_flags)tmpmax;
 
1092
  /* Reject the loopback device */
 
1093
  if(flags & IFF_LOOPBACK){
 
1094
    if(debug){
 
1095
      fprintf(stderr, "Rejecting loopback interface \"%s\"\n",
 
1096
              if_entry->d_name);
 
1097
    }
 
1098
    return 0;
 
1099
  }
 
1100
  /* Accept point-to-point devices only if connect_to is specified */
 
1101
  if(connect_to != NULL and (flags & IFF_POINTOPOINT)){
 
1102
    if(debug){
 
1103
      fprintf(stderr, "Accepting point-to-point interface \"%s\"\n",
 
1104
              if_entry->d_name);
 
1105
    }
 
1106
    return 1;
 
1107
  }
 
1108
  /* Otherwise, reject non-broadcast-capable devices */
 
1109
  if(not (flags & IFF_BROADCAST)){
 
1110
    if(debug){
 
1111
      fprintf(stderr, "Rejecting non-broadcast interface \"%s\"\n",
 
1112
              if_entry->d_name);
 
1113
    }
 
1114
    return 0;
 
1115
  }
 
1116
  /* Accept this device */
 
1117
  if(debug){
 
1118
    fprintf(stderr, "Interface \"%s\" is acceptable\n",
 
1119
            if_entry->d_name);
 
1120
  }
 
1121
  return 1;
 
1122
}
 
1123
 
 
1124
int main(int argc, char *argv[]){
 
1125
  AvahiSServiceBrowser *sb = NULL;
 
1126
  int error;
 
1127
  int ret;
 
1128
  intmax_t tmpmax;
 
1129
  char *tmp;
 
1130
  int exitcode = EXIT_SUCCESS;
 
1131
  const char *interface = "";
 
1132
  struct ifreq network;
 
1133
  int sd = -1;
 
1134
  bool take_down_interface = false;
 
1135
  uid_t uid;
 
1136
  gid_t gid;
 
1137
  char tempdir[] = "/tmp/mandosXXXXXX";
 
1138
  bool tempdir_created = false;
 
1139
  AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
 
1140
  const char *seckey = PATHDIR "/" SECKEY;
 
1141
  const char *pubkey = PATHDIR "/" PUBKEY;
 
1142
  
 
1143
  bool gnutls_initialized = false;
 
1144
  bool gpgme_initialized = false;
 
1145
  float delay = 2.5f;
 
1146
  
 
1147
  struct sigaction old_sigterm_action = { .sa_handler = SIG_DFL };
 
1148
  struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
 
1149
  
 
1150
  uid = getuid();
 
1151
  gid = getgid();
 
1152
  
 
1153
  /* Lower any group privileges we might have, just to be safe */
 
1154
  errno = 0;
 
1155
  ret = setgid(gid);
 
1156
  if(ret == -1){
 
1157
    perror("setgid");
 
1158
  }
 
1159
  
 
1160
  /* Lower user privileges (temporarily) */
 
1161
  errno = 0;
 
1162
  ret = seteuid(uid);
 
1163
  if(ret == -1){
 
1164
    perror("seteuid");
 
1165
  }
 
1166
  
 
1167
  if(quit_now){
 
1168
    goto end;
 
1169
  }
 
1170
  
 
1171
  {
 
1172
    struct argp_option options[] = {
 
1173
      { .name = "debug", .key = 128,
 
1174
        .doc = "Debug mode", .group = 3 },
 
1175
      { .name = "connect", .key = 'c',
 
1176
        .arg = "ADDRESS:PORT",
 
1177
        .doc = "Connect directly to a specific Mandos server",
 
1178
        .group = 1 },
 
1179
      { .name = "interface", .key = 'i',
 
1180
        .arg = "NAME",
 
1181
        .doc = "Network interface that will be used to search for"
 
1182
        " Mandos servers",
 
1183
        .group = 1 },
 
1184
      { .name = "seckey", .key = 's',
 
1185
        .arg = "FILE",
 
1186
        .doc = "OpenPGP secret key file base name",
 
1187
        .group = 1 },
 
1188
      { .name = "pubkey", .key = 'p',
 
1189
        .arg = "FILE",
 
1190
        .doc = "OpenPGP public key file base name",
 
1191
        .group = 2 },
 
1192
      { .name = "dh-bits", .key = 129,
 
1193
        .arg = "BITS",
 
1194
        .doc = "Bit length of the prime number used in the"
 
1195
        " Diffie-Hellman key exchange",
 
1196
        .group = 2 },
 
1197
      { .name = "priority", .key = 130,
 
1198
        .arg = "STRING",
 
1199
        .doc = "GnuTLS priority string for the TLS handshake",
 
1200
        .group = 1 },
 
1201
      { .name = "delay", .key = 131,
 
1202
        .arg = "SECONDS",
 
1203
        .doc = "Maximum delay to wait for interface startup",
 
1204
        .group = 2 },
 
1205
      /*
 
1206
       * These reproduce what we would get without ARGP_NO_HELP
 
1207
       */
 
1208
      { .name = "help", .key = '?',
 
1209
        .doc = "Give this help list", .group = -1 },
 
1210
      { .name = "usage", .key = -3,
 
1211
        .doc = "Give a short usage message", .group = -1 },
 
1212
      { .name = "version", .key = 'V',
 
1213
        .doc = "Print program version", .group = -1 },
 
1214
      { .name = NULL }
 
1215
    };
 
1216
    
 
1217
    error_t parse_opt(int key, char *arg,
 
1218
                      struct argp_state *state){
 
1219
      errno = 0;
 
1220
      switch(key){
 
1221
      case 128:                 /* --debug */
 
1222
        debug = true;
 
1223
        break;
 
1224
      case 'c':                 /* --connect */
 
1225
        connect_to = arg;
 
1226
        break;
 
1227
      case 'i':                 /* --interface */
 
1228
        interface = arg;
 
1229
        break;
 
1230
      case 's':                 /* --seckey */
 
1231
        seckey = arg;
 
1232
        break;
 
1233
      case 'p':                 /* --pubkey */
 
1234
        pubkey = arg;
 
1235
        break;
 
1236
      case 129:                 /* --dh-bits */
 
1237
        errno = 0;
 
1238
        tmpmax = strtoimax(arg, &tmp, 10);
 
1239
        if(errno != 0 or tmp == arg or *tmp != '\0'
 
1240
           or tmpmax != (typeof(mc.dh_bits))tmpmax){
 
1241
          argp_error(state, "Bad number of DH bits");
 
1242
        }
 
1243
        mc.dh_bits = (typeof(mc.dh_bits))tmpmax;
 
1244
        break;
 
1245
      case 130:                 /* --priority */
 
1246
        mc.priority = arg;
 
1247
        break;
 
1248
      case 131:                 /* --delay */
 
1249
        errno = 0;
 
1250
        delay = strtof(arg, &tmp);
 
1251
        if(errno != 0 or tmp == arg or *tmp != '\0'){
 
1252
          argp_error(state, "Bad delay");
 
1253
        }
 
1254
        break;
 
1255
        /*
 
1256
         * These reproduce what we would get without ARGP_NO_HELP
 
1257
         */
 
1258
      case '?':                 /* --help */
 
1259
        argp_state_help(state, state->out_stream,
 
1260
                        (ARGP_HELP_STD_HELP | ARGP_HELP_EXIT_ERR)
 
1261
                        & ~(unsigned int)ARGP_HELP_EXIT_OK);
 
1262
      case -3:                  /* --usage */
 
1263
        argp_state_help(state, state->out_stream,
 
1264
                        ARGP_HELP_USAGE | ARGP_HELP_EXIT_ERR);
 
1265
      case 'V':                 /* --version */
 
1266
        fprintf(state->out_stream, "%s\n", argp_program_version);
 
1267
        exit(argp_err_exit_status);
 
1268
        break;
 
1269
      default:
 
1270
        return ARGP_ERR_UNKNOWN;
 
1271
      }
 
1272
      return errno;
 
1273
    }
 
1274
    
 
1275
    struct argp argp = { .options = options, .parser = parse_opt,
 
1276
                         .args_doc = "",
 
1277
                         .doc = "Mandos client -- Get and decrypt"
 
1278
                         " passwords from a Mandos server" };
 
1279
    ret = argp_parse(&argp, argc, argv,
 
1280
                     ARGP_IN_ORDER | ARGP_NO_HELP, 0, NULL);
 
1281
    switch(ret){
 
1282
    case 0:
 
1283
      break;
 
1284
    case ENOMEM:
587
1285
    default:
588
 
    case AVAHI_BROWSER_FAILURE:
589
 
      
590
 
      fprintf(stderr, "(Browser) %s\n",
591
 
              avahi_strerror(avahi_server_errno(server)));
592
 
      avahi_simple_poll_quit(simple_poll);
593
 
      return;
594
 
      
595
 
    case AVAHI_BROWSER_NEW:
596
 
      /* We ignore the returned resolver object. In the callback
597
 
         function we free it. If the server is terminated before
598
 
         the callback function is called the server will free
599
 
         the resolver for us. */
600
 
      
601
 
      if (!(avahi_s_service_resolver_new(s, interface, protocol, name,
602
 
                                         type, domain,
603
 
                                         AVAHI_PROTO_INET6, 0,
604
 
                                         resolve_callback, s)))
605
 
        fprintf(stderr, "Failed to resolve service '%s': %s\n", name,
606
 
                avahi_strerror(avahi_server_errno(s)));
607
 
      break;
608
 
      
609
 
    case AVAHI_BROWSER_REMOVE:
610
 
      break;
611
 
      
612
 
    case AVAHI_BROWSER_ALL_FOR_NOW:
613
 
    case AVAHI_BROWSER_CACHE_EXHAUSTED:
614
 
      break;
615
 
    }
616
 
}
617
 
 
618
 
int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char*argv[]) {
 
1286
      errno = ret;
 
1287
      perror("argp_parse");
 
1288
      exitcode = EX_OSERR;
 
1289
      goto end;
 
1290
    case EINVAL:
 
1291
      exitcode = EX_USAGE;
 
1292
      goto end;
 
1293
    }
 
1294
  }
 
1295
  
 
1296
  if(not debug){
 
1297
    avahi_set_log_function(empty_log);
 
1298
  }
 
1299
 
 
1300
  if(interface[0] == '\0'){
 
1301
    struct dirent **direntries;
 
1302
    ret = scandir(sys_class_net, &direntries, good_interface,
 
1303
                  alphasort);
 
1304
    if(ret >= 1){
 
1305
      /* Pick the first good interface */
 
1306
      interface = strdup(direntries[0]->d_name);
 
1307
      if(debug){
 
1308
        fprintf(stderr, "Using interface \"%s\"\n", interface);
 
1309
      }
 
1310
      if(interface == NULL){
 
1311
        perror("malloc");
 
1312
        free(direntries);
 
1313
        exitcode = EXIT_FAILURE;
 
1314
        goto end;
 
1315
      }
 
1316
      free(direntries);
 
1317
    } else {
 
1318
      free(direntries);
 
1319
      fprintf(stderr, "Could not find a network interface\n");
 
1320
      exitcode = EXIT_FAILURE;
 
1321
      goto end;
 
1322
    }
 
1323
  }
 
1324
  
 
1325
  /* Initialize Avahi early so avahi_simple_poll_quit() can be called
 
1326
     from the signal handler */
 
1327
  /* Initialize the pseudo-RNG for Avahi */
 
1328
  srand((unsigned int) time(NULL));
 
1329
  mc.simple_poll = avahi_simple_poll_new();
 
1330
  if(mc.simple_poll == NULL){
 
1331
    fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
 
1332
    exitcode = EX_UNAVAILABLE;
 
1333
    goto end;
 
1334
  }
 
1335
  
 
1336
  sigemptyset(&sigterm_action.sa_mask);
 
1337
  ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
 
1338
  if(ret == -1){
 
1339
    perror("sigaddset");
 
1340
    exitcode = EX_OSERR;
 
1341
    goto end;
 
1342
  }
 
1343
  ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
 
1344
  if(ret == -1){
 
1345
    perror("sigaddset");
 
1346
    exitcode = EX_OSERR;
 
1347
    goto end;
 
1348
  }
 
1349
  ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
 
1350
  if(ret == -1){
 
1351
    perror("sigaddset");
 
1352
    exitcode = EX_OSERR;
 
1353
    goto end;
 
1354
  }
 
1355
  /* Need to check if the handler is SIG_IGN before handling:
 
1356
     | [[info:libc:Initial Signal Actions]] |
 
1357
     | [[info:libc:Basic Signal Handling]]  |
 
1358
  */
 
1359
  ret = sigaction(SIGINT, NULL, &old_sigterm_action);
 
1360
  if(ret == -1){
 
1361
    perror("sigaction");
 
1362
    return EX_OSERR;
 
1363
  }
 
1364
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1365
    ret = sigaction(SIGINT, &sigterm_action, NULL);
 
1366
    if(ret == -1){
 
1367
      perror("sigaction");
 
1368
      exitcode = EX_OSERR;
 
1369
      goto end;
 
1370
    }
 
1371
  }
 
1372
  ret = sigaction(SIGHUP, NULL, &old_sigterm_action);
 
1373
  if(ret == -1){
 
1374
    perror("sigaction");
 
1375
    return EX_OSERR;
 
1376
  }
 
1377
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1378
    ret = sigaction(SIGHUP, &sigterm_action, NULL);
 
1379
    if(ret == -1){
 
1380
      perror("sigaction");
 
1381
      exitcode = EX_OSERR;
 
1382
      goto end;
 
1383
    }
 
1384
  }
 
1385
  ret = sigaction(SIGTERM, NULL, &old_sigterm_action);
 
1386
  if(ret == -1){
 
1387
    perror("sigaction");
 
1388
    return EX_OSERR;
 
1389
  }
 
1390
  if(old_sigterm_action.sa_handler != SIG_IGN){
 
1391
    ret = sigaction(SIGTERM, &sigterm_action, NULL);
 
1392
    if(ret == -1){
 
1393
      perror("sigaction");
 
1394
      exitcode = EX_OSERR;
 
1395
      goto end;
 
1396
    }
 
1397
  }
 
1398
  
 
1399
  /* If the interface is down, bring it up */
 
1400
  if(strcmp(interface, "none") != 0){
 
1401
    if_index = (AvahiIfIndex) if_nametoindex(interface);
 
1402
    if(if_index == 0){
 
1403
      fprintf(stderr, "No such interface: \"%s\"\n", interface);
 
1404
      exitcode = EX_UNAVAILABLE;
 
1405
      goto end;
 
1406
    }
 
1407
    
 
1408
    if(quit_now){
 
1409
      goto end;
 
1410
    }
 
1411
    
 
1412
    /* Re-raise priviliges */
 
1413
    errno = 0;
 
1414
    ret = seteuid(0);
 
1415
    if(ret == -1){
 
1416
      perror("seteuid");
 
1417
    }
 
1418
    
 
1419
#ifdef __linux__
 
1420
    /* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
 
1421
       messages about the network interface to mess up the prompt */
 
1422
    ret = klogctl(8, NULL, 5);
 
1423
    bool restore_loglevel = true;
 
1424
    if(ret == -1){
 
1425
      restore_loglevel = false;
 
1426
      perror("klogctl");
 
1427
    }
 
1428
#endif  /* __linux__ */
 
1429
    
 
1430
    sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
 
1431
    if(sd < 0){
 
1432
      perror("socket");
 
1433
      exitcode = EX_OSERR;
 
1434
#ifdef __linux__
 
1435
      if(restore_loglevel){
 
1436
        ret = klogctl(7, NULL, 0);
 
1437
        if(ret == -1){
 
1438
          perror("klogctl");
 
1439
        }
 
1440
      }
 
1441
#endif  /* __linux__ */
 
1442
      /* Lower privileges */
 
1443
      errno = 0;
 
1444
      ret = seteuid(uid);
 
1445
      if(ret == -1){
 
1446
        perror("seteuid");
 
1447
      }
 
1448
      goto end;
 
1449
    }
 
1450
    strcpy(network.ifr_name, interface);
 
1451
    ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1452
    if(ret == -1){
 
1453
      perror("ioctl SIOCGIFFLAGS");
 
1454
#ifdef __linux__
 
1455
      if(restore_loglevel){
 
1456
        ret = klogctl(7, NULL, 0);
 
1457
        if(ret == -1){
 
1458
          perror("klogctl");
 
1459
        }
 
1460
      }
 
1461
#endif  /* __linux__ */
 
1462
      exitcode = EX_OSERR;
 
1463
      /* Lower privileges */
 
1464
      errno = 0;
 
1465
      ret = seteuid(uid);
 
1466
      if(ret == -1){
 
1467
        perror("seteuid");
 
1468
      }
 
1469
      goto end;
 
1470
    }
 
1471
    if((network.ifr_flags & IFF_UP) == 0){
 
1472
      network.ifr_flags |= IFF_UP;
 
1473
      take_down_interface = true;
 
1474
      ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
1475
      if(ret == -1){
 
1476
        take_down_interface = false;
 
1477
        perror("ioctl SIOCSIFFLAGS +IFF_UP");
 
1478
        exitcode = EX_OSERR;
 
1479
#ifdef __linux__
 
1480
        if(restore_loglevel){
 
1481
          ret = klogctl(7, NULL, 0);
 
1482
          if(ret == -1){
 
1483
            perror("klogctl");
 
1484
          }
 
1485
        }
 
1486
#endif  /* __linux__ */
 
1487
        /* Lower privileges */
 
1488
        errno = 0;
 
1489
        ret = seteuid(uid);
 
1490
        if(ret == -1){
 
1491
          perror("seteuid");
 
1492
        }
 
1493
        goto end;
 
1494
      }
 
1495
    }
 
1496
    /* sleep checking until interface is running */
 
1497
    for(int i=0; i < delay * 4; i++){
 
1498
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1499
      if(ret == -1){
 
1500
        perror("ioctl SIOCGIFFLAGS");
 
1501
      } else if(network.ifr_flags & IFF_RUNNING){
 
1502
        break;
 
1503
      }
 
1504
      struct timespec sleeptime = { .tv_nsec = 250000000 };
 
1505
      ret = nanosleep(&sleeptime, NULL);
 
1506
      if(ret == -1 and errno != EINTR){
 
1507
        perror("nanosleep");
 
1508
      }
 
1509
    }
 
1510
    if(not take_down_interface){
 
1511
      /* We won't need the socket anymore */
 
1512
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1513
      if(ret == -1){
 
1514
        perror("close");
 
1515
      }
 
1516
    }
 
1517
#ifdef __linux__
 
1518
    if(restore_loglevel){
 
1519
      /* Restores kernel loglevel to default */
 
1520
      ret = klogctl(7, NULL, 0);
 
1521
      if(ret == -1){
 
1522
        perror("klogctl");
 
1523
      }
 
1524
    }
 
1525
#endif  /* __linux__ */
 
1526
    /* Lower privileges */
 
1527
    errno = 0;
 
1528
    if(take_down_interface){
 
1529
      /* Lower privileges */
 
1530
      ret = seteuid(uid);
 
1531
      if(ret == -1){
 
1532
        perror("seteuid");
 
1533
      }
 
1534
    } else {
 
1535
      /* Lower privileges permanently */
 
1536
      ret = setuid(uid);
 
1537
      if(ret == -1){
 
1538
        perror("setuid");
 
1539
      }
 
1540
    }
 
1541
  }
 
1542
  
 
1543
  if(quit_now){
 
1544
    goto end;
 
1545
  }
 
1546
  
 
1547
  ret = init_gnutls_global(pubkey, seckey);
 
1548
  if(ret == -1){
 
1549
    fprintf(stderr, "init_gnutls_global failed\n");
 
1550
    exitcode = EX_UNAVAILABLE;
 
1551
    goto end;
 
1552
  } else {
 
1553
    gnutls_initialized = true;
 
1554
  }
 
1555
  
 
1556
  if(quit_now){
 
1557
    goto end;
 
1558
  }
 
1559
  
 
1560
  tempdir_created = true;
 
1561
  if(mkdtemp(tempdir) == NULL){
 
1562
    tempdir_created = false;
 
1563
    perror("mkdtemp");
 
1564
    goto end;
 
1565
  }
 
1566
  
 
1567
  if(quit_now){
 
1568
    goto end;
 
1569
  }
 
1570
  
 
1571
  if(not init_gpgme(pubkey, seckey, tempdir)){
 
1572
    fprintf(stderr, "init_gpgme failed\n");
 
1573
    exitcode = EX_UNAVAILABLE;
 
1574
    goto end;
 
1575
  } else {
 
1576
    gpgme_initialized = true;
 
1577
  }
 
1578
  
 
1579
  if(quit_now){
 
1580
    goto end;
 
1581
  }
 
1582
  
 
1583
  if(connect_to != NULL){
 
1584
    /* Connect directly, do not use Zeroconf */
 
1585
    /* (Mainly meant for debugging) */
 
1586
    char *address = strrchr(connect_to, ':');
 
1587
    if(address == NULL){
 
1588
      fprintf(stderr, "No colon in address\n");
 
1589
      exitcode = EX_USAGE;
 
1590
      goto end;
 
1591
    }
 
1592
    
 
1593
    if(quit_now){
 
1594
      goto end;
 
1595
    }
 
1596
    
 
1597
    uint16_t port;
 
1598
    errno = 0;
 
1599
    tmpmax = strtoimax(address+1, &tmp, 10);
 
1600
    if(errno != 0 or tmp == address+1 or *tmp != '\0'
 
1601
       or tmpmax != (uint16_t)tmpmax){
 
1602
      fprintf(stderr, "Bad port number\n");
 
1603
      exitcode = EX_USAGE;
 
1604
      goto end;
 
1605
    }
 
1606
  
 
1607
    if(quit_now){
 
1608
      goto end;
 
1609
    }
 
1610
    
 
1611
    port = (uint16_t)tmpmax;
 
1612
    *address = '\0';
 
1613
    address = connect_to;
 
1614
    /* Colon in address indicates IPv6 */
 
1615
    int af;
 
1616
    if(strchr(address, ':') != NULL){
 
1617
      af = AF_INET6;
 
1618
    } else {
 
1619
      af = AF_INET;
 
1620
    }
 
1621
    
 
1622
    if(quit_now){
 
1623
      goto end;
 
1624
    }
 
1625
 
 
1626
    while(not quit_now){
 
1627
      ret = start_mandos_communication(address, port, if_index, af);
 
1628
      if(quit_now or ret == 0){
 
1629
        break;
 
1630
      }
 
1631
      sleep(15);
 
1632
    };
 
1633
 
 
1634
    if (not quit_now){
 
1635
      exitcode = EXIT_SUCCESS;
 
1636
    }
 
1637
 
 
1638
    goto end;
 
1639
  }
 
1640
  
 
1641
  if(quit_now){
 
1642
    goto end;
 
1643
  }
 
1644
  
 
1645
  {
619
1646
    AvahiServerConfig config;
620
 
    AvahiSServiceBrowser *sb = NULL;
621
 
    int error;
622
 
    int ret;
623
 
    int returncode = EXIT_SUCCESS;
624
 
    const char *interface = "eth0";
625
 
    
626
 
    while (true){
627
 
      static struct option long_options[] = {
628
 
        {"debug", no_argument, (int *)&debug, 1},
629
 
        {"interface", required_argument, 0, 'i'},
630
 
        {0, 0, 0, 0} };
631
 
      
632
 
      int option_index = 0;
633
 
      ret = getopt_long (argc, argv, "i:", long_options,
634
 
                         &option_index);
635
 
      
636
 
      if (ret == -1){
637
 
        break;
638
 
      }
639
 
      
640
 
      switch(ret){
641
 
      case 0:
642
 
        break;
643
 
      case 'i':
644
 
        interface = optarg;
645
 
        break;
646
 
      default:
647
 
        exit(EXIT_FAILURE);
648
 
      }
649
 
    }
650
 
    
651
 
    if (not debug){
652
 
      avahi_set_log_function(empty_log);
653
 
    }
654
 
    
655
 
    /* Initialize the psuedo-RNG */
656
 
    srand((unsigned int) time(NULL));
657
 
 
658
 
    /* Allocate main loop object */
659
 
    if (!(simple_poll = avahi_simple_poll_new())) {
660
 
        fprintf(stderr, "Failed to create simple poll object.\n");
661
 
        
662
 
        goto exit;
663
 
    }
664
 
 
665
 
    /* Do not publish any local records */
 
1647
    /* Do not publish any local Zeroconf records */
666
1648
    avahi_server_config_init(&config);
667
1649
    config.publish_hinfo = 0;
668
1650
    config.publish_addresses = 0;
669
1651
    config.publish_workstation = 0;
670
1652
    config.publish_domain = 0;
671
 
 
 
1653
    
672
1654
    /* Allocate a new server */
673
 
    server = avahi_server_new(avahi_simple_poll_get(simple_poll),
674
 
                              &config, NULL, NULL, &error);
675
 
 
676
 
    /* Free the configuration data */
 
1655
    mc.server = avahi_server_new(avahi_simple_poll_get
 
1656
                                 (mc.simple_poll), &config, NULL,
 
1657
                                 NULL, &error);
 
1658
    
 
1659
    /* Free the Avahi configuration data */
677
1660
    avahi_server_config_free(&config);
678
 
 
679
 
    /* Check if creating the server object succeeded */
680
 
    if (!server) {
681
 
        fprintf(stderr, "Failed to create server: %s\n",
682
 
                avahi_strerror(error));
683
 
        returncode = EXIT_FAILURE;
684
 
        goto exit;
685
 
    }
686
 
    
687
 
    /* Create the service browser */
688
 
    sb = avahi_s_service_browser_new(server,
689
 
                                     (AvahiIfIndex)
690
 
                                     if_nametoindex(interface),
691
 
                                     AVAHI_PROTO_INET6,
692
 
                                     "_mandos._tcp", NULL, 0,
693
 
                                     browse_callback, server);
694
 
    if (!sb) {
695
 
        fprintf(stderr, "Failed to create service browser: %s\n",
696
 
                avahi_strerror(avahi_server_errno(server)));
697
 
        returncode = EXIT_FAILURE;
698
 
        goto exit;
699
 
    }
700
 
    
701
 
    /* Run the main loop */
702
 
 
703
 
    if (debug){
704
 
      fprintf(stderr, "Starting avahi loop search\n");
705
 
    }
706
 
    
707
 
    avahi_simple_poll_loop(simple_poll);
708
 
    
709
 
 exit:
710
 
 
711
 
    if (debug){
712
 
      fprintf(stderr, "%s exiting\n", argv[0]);
713
 
    }
714
 
    
715
 
    /* Cleanup things */
716
 
    if (sb)
717
 
        avahi_s_service_browser_free(sb);
718
 
    
719
 
    if (server)
720
 
        avahi_server_free(server);
721
 
 
722
 
    if (simple_poll)
723
 
        avahi_simple_poll_free(simple_poll);
724
 
 
725
 
    return returncode;
 
1661
  }
 
1662
  
 
1663
  /* Check if creating the Avahi server object succeeded */
 
1664
  if(mc.server == NULL){
 
1665
    fprintf(stderr, "Failed to create Avahi server: %s\n",
 
1666
            avahi_strerror(error));
 
1667
    exitcode = EX_UNAVAILABLE;
 
1668
    goto end;
 
1669
  }
 
1670
  
 
1671
  if(quit_now){
 
1672
    goto end;
 
1673
  }
 
1674
  
 
1675
  /* Create the Avahi service browser */
 
1676
  sb = avahi_s_service_browser_new(mc.server, if_index,
 
1677
                                   AVAHI_PROTO_UNSPEC, "_mandos._tcp",
 
1678
                                   NULL, 0, browse_callback, NULL);
 
1679
  if(sb == NULL){
 
1680
    fprintf(stderr, "Failed to create service browser: %s\n",
 
1681
            avahi_strerror(avahi_server_errno(mc.server)));
 
1682
    exitcode = EX_UNAVAILABLE;
 
1683
    goto end;
 
1684
  }
 
1685
  
 
1686
  if(quit_now){
 
1687
    goto end;
 
1688
  }
 
1689
  
 
1690
  /* Run the main loop */
 
1691
  
 
1692
  if(debug){
 
1693
    fprintf(stderr, "Starting Avahi loop search\n");
 
1694
  }
 
1695
  
 
1696
  avahi_simple_poll_loop(mc.simple_poll);
 
1697
  
 
1698
 end:
 
1699
  
 
1700
  if(debug){
 
1701
    fprintf(stderr, "%s exiting\n", argv[0]);
 
1702
  }
 
1703
  
 
1704
  /* Cleanup things */
 
1705
  if(sb != NULL)
 
1706
    avahi_s_service_browser_free(sb);
 
1707
  
 
1708
  if(mc.server != NULL)
 
1709
    avahi_server_free(mc.server);
 
1710
  
 
1711
  if(mc.simple_poll != NULL)
 
1712
    avahi_simple_poll_free(mc.simple_poll);
 
1713
  
 
1714
  if(gnutls_initialized){
 
1715
    gnutls_certificate_free_credentials(mc.cred);
 
1716
    gnutls_global_deinit();
 
1717
    gnutls_dh_params_deinit(mc.dh_params);
 
1718
  }
 
1719
  
 
1720
  if(gpgme_initialized){
 
1721
    gpgme_release(mc.ctx);
 
1722
  }
 
1723
  
 
1724
  /* Take down the network interface */
 
1725
  if(take_down_interface){
 
1726
    /* Re-raise priviliges */
 
1727
    errno = 0;
 
1728
    ret = seteuid(0);
 
1729
    if(ret == -1){
 
1730
      perror("seteuid");
 
1731
    }
 
1732
    if(geteuid() == 0){
 
1733
      ret = ioctl(sd, SIOCGIFFLAGS, &network);
 
1734
      if(ret == -1){
 
1735
        perror("ioctl SIOCGIFFLAGS");
 
1736
      } else if(network.ifr_flags & IFF_UP) {
 
1737
        network.ifr_flags &= ~(short)IFF_UP; /* clear flag */
 
1738
        ret = ioctl(sd, SIOCSIFFLAGS, &network);
 
1739
        if(ret == -1){
 
1740
          perror("ioctl SIOCSIFFLAGS -IFF_UP");
 
1741
        }
 
1742
      }
 
1743
      ret = (int)TEMP_FAILURE_RETRY(close(sd));
 
1744
      if(ret == -1){
 
1745
        perror("close");
 
1746
      }
 
1747
      /* Lower privileges permanently */
 
1748
      errno = 0;
 
1749
      ret = setuid(uid);
 
1750
      if(ret == -1){
 
1751
        perror("setuid");
 
1752
      }
 
1753
    }
 
1754
  }
 
1755
  
 
1756
  /* Removes the temp directory used by GPGME */
 
1757
  if(tempdir_created){
 
1758
    DIR *d;
 
1759
    struct dirent *direntry;
 
1760
    d = opendir(tempdir);
 
1761
    if(d == NULL){
 
1762
      if(errno != ENOENT){
 
1763
        perror("opendir");
 
1764
      }
 
1765
    } else {
 
1766
      while(true){
 
1767
        direntry = readdir(d);
 
1768
        if(direntry == NULL){
 
1769
          break;
 
1770
        }
 
1771
        /* Skip "." and ".." */
 
1772
        if(direntry->d_name[0] == '.'
 
1773
           and (direntry->d_name[1] == '\0'
 
1774
                or (direntry->d_name[1] == '.'
 
1775
                    and direntry->d_name[2] == '\0'))){
 
1776
          continue;
 
1777
        }
 
1778
        char *fullname = NULL;
 
1779
        ret = asprintf(&fullname, "%s/%s", tempdir,
 
1780
                       direntry->d_name);
 
1781
        if(ret < 0){
 
1782
          perror("asprintf");
 
1783
          continue;
 
1784
        }
 
1785
        ret = remove(fullname);
 
1786
        if(ret == -1){
 
1787
          fprintf(stderr, "remove(\"%s\"): %s\n", fullname,
 
1788
                  strerror(errno));
 
1789
        }
 
1790
        free(fullname);
 
1791
      }
 
1792
      closedir(d);
 
1793
    }
 
1794
    ret = rmdir(tempdir);
 
1795
    if(ret == -1 and errno != ENOENT){
 
1796
      perror("rmdir");
 
1797
    }
 
1798
  }
 
1799
  
 
1800
  if(quit_now){
 
1801
    sigemptyset(&old_sigterm_action.sa_mask);
 
1802
    old_sigterm_action.sa_handler = SIG_DFL;
 
1803
    ret = (int)TEMP_FAILURE_RETRY(sigaction(signal_received,
 
1804
                                            &old_sigterm_action,
 
1805
                                            NULL));
 
1806
    if(ret == -1){
 
1807
      perror("sigaction");
 
1808
    }
 
1809
    do {
 
1810
      ret = raise(signal_received);
 
1811
    } while(ret != 0 and errno == EINTR);
 
1812
    if(ret != 0){
 
1813
      perror("raise");
 
1814
      abort();
 
1815
    }
 
1816
    TEMP_FAILURE_RETRY(pause());
 
1817
  }
 
1818
  
 
1819
  return exitcode;
726
1820
}