1
/* -*- coding: utf-8 -*- */
3
* Mandos-client - get and decrypt data from a Mandos server
5
* This program is partly derived from an example program for an Avahi
6
* service browser, downloaded from
7
* <http://avahi.org/browser/examples/core-browse-services.c>. This
8
* includes the following functions: "resolve_callback",
9
* "browse_callback", and parts of "main".
12
* Copyright © 2008,2009 Teddy Hogeborn
13
* Copyright © 2008,2009 Björn Påhlsson
15
* This program is free software: you can redistribute it and/or
16
* modify it under the terms of the GNU General Public License as
17
* published by the Free Software Foundation, either version 3 of the
18
* License, or (at your option) any later version.
20
* This program is distributed in the hope that it will be useful, but
21
* WITHOUT ANY WARRANTY; without even the implied warranty of
22
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
23
* General Public License for more details.
25
* You should have received a copy of the GNU General Public License
26
* along with this program. If not, see
27
* <http://www.gnu.org/licenses/>.
29
* Contact the authors at <mandos@fukt.bsnet.se>.
32
/* Needed by GPGME, specifically gpgme_data_seek() */
33
#ifndef _LARGEFILE_SOURCE
34
#define _LARGEFILE_SOURCE
36
#ifndef _FILE_OFFSET_BITS
37
#define _FILE_OFFSET_BITS 64
40
#define _GNU_SOURCE /* TEMP_FAILURE_RETRY(), asprintf() */
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,
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(),
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()
62
#include <inttypes.h> /* PRIu16, PRIdMAX, intmax_t,
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
73
#include <unistd.h> /* close(), SEEK_SET, off_t, write(),
74
getuid(), getgid(), seteuid(),
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,
87
#include <sys/klog.h> /* klogctl() */
88
#endif /* __linux__ */
91
/* All Avahi types, constants and functions
94
#include <avahi-core/core.h>
95
#include <avahi-core/lookup.h>
96
#include <avahi-core/log.h>
97
#include <avahi-common/simple-watch.h>
98
#include <avahi-common/malloc.h>
99
#include <avahi-common/error.h>
102
#include <gnutls/gnutls.h> /* All GnuTLS types, constants and
105
init_gnutls_session(),
107
#include <gnutls/openpgp.h>
108
/* gnutls_certificate_set_openpgp_key_file(),
109
GNUTLS_OPENPGP_FMT_BASE64 */
112
#include <gpgme.h> /* All GPGME types, constants and
115
GPGME_PROTOCOL_OpenPGP,
118
#define BUFFER_SIZE 256
120
#define PATHDIR "/conf/conf.d/mandos"
121
#define SECKEY "seckey.txt"
122
#define PUBKEY "pubkey.txt"
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>";
129
/* Used for passing in values through the Avahi callback functions */
131
AvahiSimplePoll *simple_poll;
133
gnutls_certificate_credentials_t cred;
134
unsigned int dh_bits;
135
gnutls_dh_params_t dh_params;
136
const char *priority;
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" };
145
sig_atomic_t quit_now = 0;
146
int signal_received = 0;
149
* Make additional room in "buffer" for at least BUFFER_SIZE more
150
* bytes. "buffer_capacity" is how much is currently allocated,
151
* "buffer_length" is how much is already used.
153
size_t incbuffer(char **buffer, size_t buffer_length,
154
size_t buffer_capacity){
155
if(buffer_length + BUFFER_SIZE > buffer_capacity){
156
*buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
160
buffer_capacity += BUFFER_SIZE;
162
return buffer_capacity;
168
static bool init_gpgme(const char *seckey,
169
const char *pubkey, const char *tempdir){
171
gpgme_engine_info_t engine_info;
175
* Helper function to insert pub and seckey to the engine keyring.
177
bool import_key(const char *filename){
180
gpgme_data_t pgp_data;
182
fd = (int)TEMP_FAILURE_RETRY(open(filename, O_RDONLY));
188
rc = gpgme_data_new_from_fd(&pgp_data, fd);
189
if(rc != GPG_ERR_NO_ERROR){
190
fprintf(stderr, "bad gpgme_data_new_from_fd: %s: %s\n",
191
gpgme_strsource(rc), gpgme_strerror(rc));
195
rc = gpgme_op_import(mc.ctx, pgp_data);
196
if(rc != GPG_ERR_NO_ERROR){
197
fprintf(stderr, "bad gpgme_op_import: %s: %s\n",
198
gpgme_strsource(rc), gpgme_strerror(rc));
202
ret = (int)TEMP_FAILURE_RETRY(close(fd));
206
gpgme_data_release(pgp_data);
211
fprintf(stderr, "Initializing GPGME\n");
215
gpgme_check_version(NULL);
216
rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
217
if(rc != GPG_ERR_NO_ERROR){
218
fprintf(stderr, "bad gpgme_engine_check_version: %s: %s\n",
219
gpgme_strsource(rc), gpgme_strerror(rc));
223
/* Set GPGME home directory for the OpenPGP engine only */
224
rc = gpgme_get_engine_info(&engine_info);
225
if(rc != GPG_ERR_NO_ERROR){
226
fprintf(stderr, "bad gpgme_get_engine_info: %s: %s\n",
227
gpgme_strsource(rc), gpgme_strerror(rc));
230
while(engine_info != NULL){
231
if(engine_info->protocol == GPGME_PROTOCOL_OpenPGP){
232
gpgme_set_engine_info(GPGME_PROTOCOL_OpenPGP,
233
engine_info->file_name, tempdir);
236
engine_info = engine_info->next;
238
if(engine_info == NULL){
239
fprintf(stderr, "Could not set GPGME home dir to %s\n", tempdir);
243
/* Create new GPGME "context" */
244
rc = gpgme_new(&(mc.ctx));
245
if(rc != GPG_ERR_NO_ERROR){
246
fprintf(stderr, "bad gpgme_new: %s: %s\n",
247
gpgme_strsource(rc), gpgme_strerror(rc));
251
if(not import_key(pubkey) or not import_key(seckey)){
259
* Decrypt OpenPGP data.
260
* Returns -1 on error
262
static ssize_t pgp_packet_decrypt(const char *cryptotext,
265
gpgme_data_t dh_crypto, dh_plain;
268
size_t plaintext_capacity = 0;
269
ssize_t plaintext_length = 0;
272
fprintf(stderr, "Trying to decrypt OpenPGP data\n");
275
/* Create new GPGME data buffer from memory cryptotext */
276
rc = gpgme_data_new_from_mem(&dh_crypto, cryptotext, crypto_size,
278
if(rc != GPG_ERR_NO_ERROR){
279
fprintf(stderr, "bad gpgme_data_new_from_mem: %s: %s\n",
280
gpgme_strsource(rc), gpgme_strerror(rc));
284
/* Create new empty GPGME data buffer for the plaintext */
285
rc = gpgme_data_new(&dh_plain);
286
if(rc != GPG_ERR_NO_ERROR){
287
fprintf(stderr, "bad gpgme_data_new: %s: %s\n",
288
gpgme_strsource(rc), gpgme_strerror(rc));
289
gpgme_data_release(dh_crypto);
293
/* Decrypt data from the cryptotext data buffer to the plaintext
295
rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);
296
if(rc != GPG_ERR_NO_ERROR){
297
fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
298
gpgme_strsource(rc), gpgme_strerror(rc));
299
plaintext_length = -1;
301
gpgme_decrypt_result_t result;
302
result = gpgme_op_decrypt_result(mc.ctx);
304
fprintf(stderr, "gpgme_op_decrypt_result failed\n");
306
fprintf(stderr, "Unsupported algorithm: %s\n",
307
result->unsupported_algorithm);
308
fprintf(stderr, "Wrong key usage: %u\n",
309
result->wrong_key_usage);
310
if(result->file_name != NULL){
311
fprintf(stderr, "File name: %s\n", result->file_name);
313
gpgme_recipient_t recipient;
314
recipient = result->recipients;
315
while(recipient != NULL){
316
fprintf(stderr, "Public key algorithm: %s\n",
317
gpgme_pubkey_algo_name(recipient->pubkey_algo));
318
fprintf(stderr, "Key ID: %s\n", recipient->keyid);
319
fprintf(stderr, "Secret key available: %s\n",
320
recipient->status == GPG_ERR_NO_SECKEY
322
recipient = recipient->next;
330
fprintf(stderr, "Decryption of OpenPGP data succeeded\n");
333
/* Seek back to the beginning of the GPGME plaintext data buffer */
334
if(gpgme_data_seek(dh_plain, (off_t)0, SEEK_SET) == -1){
335
perror("gpgme_data_seek");
336
plaintext_length = -1;
342
plaintext_capacity = incbuffer(plaintext,
343
(size_t)plaintext_length,
345
if(plaintext_capacity == 0){
347
plaintext_length = -1;
351
ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,
353
/* Print the data, if any */
359
perror("gpgme_data_read");
360
plaintext_length = -1;
363
plaintext_length += ret;
367
fprintf(stderr, "Decrypted password is: ");
368
for(ssize_t i = 0; i < plaintext_length; i++){
369
fprintf(stderr, "%02hhX ", (*plaintext)[i]);
371
fprintf(stderr, "\n");
376
/* Delete the GPGME cryptotext data buffer */
377
gpgme_data_release(dh_crypto);
379
/* Delete the GPGME plaintext data buffer */
380
gpgme_data_release(dh_plain);
381
return plaintext_length;
384
static const char * safer_gnutls_strerror(int value){
385
const char *ret = gnutls_strerror(value); /* Spurious warning from
386
-Wunreachable-code */
392
/* GnuTLS log function callback */
393
static void debuggnutls(__attribute__((unused)) int level,
395
fprintf(stderr, "GnuTLS: %s", string);
398
static int init_gnutls_global(const char *pubkeyfilename,
399
const char *seckeyfilename){
403
fprintf(stderr, "Initializing GnuTLS\n");
406
ret = gnutls_global_init();
407
if(ret != GNUTLS_E_SUCCESS){
408
fprintf(stderr, "GnuTLS global_init: %s\n",
409
safer_gnutls_strerror(ret));
414
/* "Use a log level over 10 to enable all debugging options."
417
gnutls_global_set_log_level(11);
418
gnutls_global_set_log_function(debuggnutls);
421
/* OpenPGP credentials */
422
gnutls_certificate_allocate_credentials(&mc.cred);
423
if(ret != GNUTLS_E_SUCCESS){
424
fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning
428
safer_gnutls_strerror(ret));
429
gnutls_global_deinit();
434
fprintf(stderr, "Attempting to use OpenPGP public key %s and"
435
" secret key %s as GnuTLS credentials\n", pubkeyfilename,
439
ret = gnutls_certificate_set_openpgp_key_file
440
(mc.cred, pubkeyfilename, seckeyfilename,
441
GNUTLS_OPENPGP_FMT_BASE64);
442
if(ret != GNUTLS_E_SUCCESS){
444
"Error[%d] while reading the OpenPGP key pair ('%s',"
445
" '%s')\n", ret, pubkeyfilename, seckeyfilename);
446
fprintf(stderr, "The GnuTLS error is: %s\n",
447
safer_gnutls_strerror(ret));
451
/* GnuTLS server initialization */
452
ret = gnutls_dh_params_init(&mc.dh_params);
453
if(ret != GNUTLS_E_SUCCESS){
454
fprintf(stderr, "Error in GnuTLS DH parameter initialization:"
455
" %s\n", safer_gnutls_strerror(ret));
458
ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);
459
if(ret != GNUTLS_E_SUCCESS){
460
fprintf(stderr, "Error in GnuTLS prime generation: %s\n",
461
safer_gnutls_strerror(ret));
465
gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);
471
gnutls_certificate_free_credentials(mc.cred);
472
gnutls_global_deinit();
473
gnutls_dh_params_deinit(mc.dh_params);
477
static int init_gnutls_session(gnutls_session_t *session){
479
/* GnuTLS session creation */
481
ret = gnutls_init(session, GNUTLS_SERVER);
485
} while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
486
if(ret != GNUTLS_E_SUCCESS){
487
fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
488
safer_gnutls_strerror(ret));
494
ret = gnutls_priority_set_direct(*session, mc.priority, &err);
496
gnutls_deinit(*session);
499
} while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
500
if(ret != GNUTLS_E_SUCCESS){
501
fprintf(stderr, "Syntax error at: %s\n", err);
502
fprintf(stderr, "GnuTLS error: %s\n",
503
safer_gnutls_strerror(ret));
504
gnutls_deinit(*session);
510
ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
513
gnutls_deinit(*session);
516
} while(ret == GNUTLS_E_INTERRUPTED or ret == GNUTLS_E_AGAIN);
517
if(ret != GNUTLS_E_SUCCESS){
518
fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
519
safer_gnutls_strerror(ret));
520
gnutls_deinit(*session);
524
/* ignore client certificate if any. */
525
gnutls_certificate_server_set_request(*session, GNUTLS_CERT_IGNORE);
527
gnutls_dh_set_prime_bits(*session, mc.dh_bits);
532
/* Avahi log function callback */
533
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
534
__attribute__((unused)) const char *txt){}
536
/* Called when a Mandos server is found */
537
static int start_mandos_communication(const char *ip, uint16_t port,
538
AvahiIfIndex if_index,
540
int ret, tcp_sd = -1;
543
struct sockaddr_in in;
544
struct sockaddr_in6 in6;
547
char *decrypted_buffer;
548
size_t buffer_length = 0;
549
size_t buffer_capacity = 0;
552
gnutls_session_t session;
553
int pf; /* Protocol family */
567
fprintf(stderr, "Bad address family: %d\n", af);
571
ret = init_gnutls_session(&session);
577
fprintf(stderr, "Setting up a TCP connection to %s, port %" PRIu16
581
tcp_sd = socket(pf, SOCK_STREAM, 0);
593
memset(&to, 0, sizeof(to));
595
to.in6.sin6_family = (sa_family_t)af;
596
ret = inet_pton(af, ip, &to.in6.sin6_addr);
598
to.in.sin_family = (sa_family_t)af;
599
ret = inet_pton(af, ip, &to.in.sin_addr);
607
fprintf(stderr, "Bad address: %s\n", ip);
612
to.in6.sin6_port = htons(port); /* Spurious warnings from
614
-Wunreachable-code */
616
if(IN6_IS_ADDR_LINKLOCAL /* Spurious warnings from */
617
(&to.in6.sin6_addr)){ /* -Wstrict-aliasing=2 or lower and
619
if(if_index == AVAHI_IF_UNSPEC){
620
fprintf(stderr, "An IPv6 link-local address is incomplete"
621
" without a network interface\n");
625
/* Set the network interface number as scope */
626
to.in6.sin6_scope_id = (uint32_t)if_index;
629
to.in.sin_port = htons(port); /* Spurious warnings from
631
-Wunreachable-code */
640
if(af == AF_INET6 and if_index != AVAHI_IF_UNSPEC){
641
char interface[IF_NAMESIZE];
642
if(if_indextoname((unsigned int)if_index, interface) == NULL){
643
perror("if_indextoname");
645
fprintf(stderr, "Connection to: %s%%%s, port %" PRIu16 "\n",
646
ip, interface, port);
649
fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
652
char addrstr[(INET_ADDRSTRLEN > INET6_ADDRSTRLEN) ?
653
INET_ADDRSTRLEN : INET6_ADDRSTRLEN] = "";
656
pcret = inet_ntop(af, &(to.in6.sin6_addr), addrstr,
659
pcret = inet_ntop(af, &(to.in.sin_addr), addrstr,
665
if(strcmp(addrstr, ip) != 0){
666
fprintf(stderr, "Canonical address form: %s\n", addrstr);
677
ret = connect(tcp_sd, &to.in6, sizeof(to));
679
ret = connect(tcp_sd, &to.in, sizeof(to)); /* IPv4 */
692
const char *out = mandos_protocol_version;
695
size_t out_size = strlen(out);
696
ret = (int)TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
697
out_size - written));
703
written += (size_t)ret;
704
if(written < out_size){
707
if(out == mandos_protocol_version){
722
fprintf(stderr, "Establishing TLS session with %s\n", ip);
730
gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);
738
ret = gnutls_handshake(session);
743
} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
745
if(ret != GNUTLS_E_SUCCESS){
747
fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
754
/* Read OpenPGP packet that contains the wanted password */
757
fprintf(stderr, "Retrieving OpenPGP encrypted password from %s\n",
768
buffer_capacity = incbuffer(&buffer, buffer_length,
770
if(buffer_capacity == 0){
781
sret = gnutls_record_recv(session, buffer+buffer_length,
788
case GNUTLS_E_INTERRUPTED:
791
case GNUTLS_E_REHANDSHAKE:
793
ret = gnutls_handshake(session);
799
} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
801
fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
808
fprintf(stderr, "Unknown error while reading data from"
809
" encrypted session with Mandos server\n");
811
gnutls_bye(session, GNUTLS_SHUT_RDWR);
815
buffer_length += (size_t) sret;
820
fprintf(stderr, "Closing TLS session\n");
829
ret = gnutls_bye(session, GNUTLS_SHUT_RDWR);
834
} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
836
if(buffer_length > 0){
837
ssize_t decrypted_buffer_size;
838
decrypted_buffer_size = pgp_packet_decrypt(buffer,
841
if(decrypted_buffer_size >= 0){
844
while(written < (size_t) decrypted_buffer_size){
850
ret = (int)fwrite(decrypted_buffer + written, 1,
851
(size_t)decrypted_buffer_size - written,
853
if(ret == 0 and ferror(stdout)){
855
fprintf(stderr, "Error writing encrypted data: %s\n",
861
written += (size_t)ret;
863
free(decrypted_buffer);
871
/* Shutdown procedure */
876
ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));
881
gnutls_deinit(session);
888
static void resolve_callback(AvahiSServiceResolver *r,
889
AvahiIfIndex interface,
891
AvahiResolverEvent event,
895
const char *host_name,
896
const AvahiAddress *address,
898
AVAHI_GCC_UNUSED AvahiStringList *txt,
899
AVAHI_GCC_UNUSED AvahiLookupResultFlags
901
AVAHI_GCC_UNUSED void* userdata){
904
/* Called whenever a service has been resolved successfully or
913
case AVAHI_RESOLVER_FAILURE:
914
fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
915
" of type '%s' in domain '%s': %s\n", name, type, domain,
916
avahi_strerror(avahi_server_errno(mc.server)));
919
case AVAHI_RESOLVER_FOUND:
921
char ip[AVAHI_ADDRESS_STR_MAX];
922
avahi_address_snprint(ip, sizeof(ip), address);
924
fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
925
PRIdMAX ") on port %" PRIu16 "\n", name, host_name,
926
ip, (intmax_t)interface, port);
928
int ret = start_mandos_communication(ip, port, interface,
929
avahi_proto_to_af(proto));
931
avahi_simple_poll_quit(mc.simple_poll);
935
avahi_s_service_resolver_free(r);
938
static void browse_callback(AvahiSServiceBrowser *b,
939
AvahiIfIndex interface,
940
AvahiProtocol protocol,
941
AvahiBrowserEvent event,
945
AVAHI_GCC_UNUSED AvahiLookupResultFlags
947
AVAHI_GCC_UNUSED void* userdata){
950
/* Called whenever a new services becomes available on the LAN or
951
is removed from the LAN */
959
case AVAHI_BROWSER_FAILURE:
961
fprintf(stderr, "(Avahi browser) %s\n",
962
avahi_strerror(avahi_server_errno(mc.server)));
963
avahi_simple_poll_quit(mc.simple_poll);
966
case AVAHI_BROWSER_NEW:
967
/* We ignore the returned Avahi resolver object. In the callback
968
function we free it. If the Avahi server is terminated before
969
the callback function is called the Avahi server will free the
972
if(avahi_s_service_resolver_new(mc.server, interface, protocol,
973
name, type, domain, protocol, 0,
974
resolve_callback, NULL) == NULL)
975
fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
976
name, avahi_strerror(avahi_server_errno(mc.server)));
979
case AVAHI_BROWSER_REMOVE:
982
case AVAHI_BROWSER_ALL_FOR_NOW:
983
case AVAHI_BROWSER_CACHE_EXHAUSTED:
985
fprintf(stderr, "No Mandos server found, still searching...\n");
991
/* stop main loop after sigterm has been called */
992
static void handle_sigterm(int sig){
997
signal_received = sig;
998
int old_errno = errno;
999
if(mc.simple_poll != NULL){
1000
avahi_simple_poll_quit(mc.simple_poll);
1005
int main(int argc, char *argv[]){
1006
AvahiSServiceBrowser *sb = NULL;
1011
int exitcode = EXIT_SUCCESS;
1012
const char *interface = "eth0";
1013
struct ifreq network;
1015
bool take_down_interface = false;
1018
char *connect_to = NULL;
1019
char tempdir[] = "/tmp/mandosXXXXXX";
1020
bool tempdir_created = false;
1021
AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
1022
const char *seckey = PATHDIR "/" SECKEY;
1023
const char *pubkey = PATHDIR "/" PUBKEY;
1025
bool gnutls_initialized = false;
1026
bool gpgme_initialized = false;
1029
struct sigaction old_sigterm_action = { .sa_handler = SIG_DFL };
1030
struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
1035
/* Lower any group privileges we might have, just to be safe */
1042
/* Lower user privileges (temporarily) */
1054
struct argp_option options[] = {
1055
{ .name = "debug", .key = 128,
1056
.doc = "Debug mode", .group = 3 },
1057
{ .name = "connect", .key = 'c',
1058
.arg = "ADDRESS:PORT",
1059
.doc = "Connect directly to a specific Mandos server",
1061
{ .name = "interface", .key = 'i',
1063
.doc = "Network interface that will be used to search for"
1066
{ .name = "seckey", .key = 's',
1068
.doc = "OpenPGP secret key file base name",
1070
{ .name = "pubkey", .key = 'p',
1072
.doc = "OpenPGP public key file base name",
1074
{ .name = "dh-bits", .key = 129,
1076
.doc = "Bit length of the prime number used in the"
1077
" Diffie-Hellman key exchange",
1079
{ .name = "priority", .key = 130,
1081
.doc = "GnuTLS priority string for the TLS handshake",
1083
{ .name = "delay", .key = 131,
1085
.doc = "Maximum delay to wait for interface startup",
1090
error_t parse_opt(int key, char *arg,
1091
struct argp_state *state){
1093
case 128: /* --debug */
1096
case 'c': /* --connect */
1099
case 'i': /* --interface */
1102
case 's': /* --seckey */
1105
case 'p': /* --pubkey */
1108
case 129: /* --dh-bits */
1110
tmpmax = strtoimax(arg, &tmp, 10);
1111
if(errno != 0 or tmp == arg or *tmp != '\0'
1112
or tmpmax != (typeof(mc.dh_bits))tmpmax){
1113
fprintf(stderr, "Bad number of DH bits\n");
1116
mc.dh_bits = (typeof(mc.dh_bits))tmpmax;
1118
case 130: /* --priority */
1121
case 131: /* --delay */
1123
delay = strtof(arg, &tmp);
1124
if(errno != 0 or tmp == arg or *tmp != '\0'){
1125
fprintf(stderr, "Bad delay\n");
1134
return ARGP_ERR_UNKNOWN;
1139
struct argp argp = { .options = options, .parser = parse_opt,
1141
.doc = "Mandos client -- Get and decrypt"
1142
" passwords from a Mandos server" };
1143
ret = argp_parse(&argp, argc, argv, 0, 0, NULL);
1144
if(ret == ARGP_ERR_UNKNOWN){
1145
fprintf(stderr, "Unknown error while parsing arguments\n");
1146
exitcode = EXIT_FAILURE;
1152
avahi_set_log_function(empty_log);
1155
/* Initialize Avahi early so avahi_simple_poll_quit() can be called
1156
from the signal handler */
1157
/* Initialize the pseudo-RNG for Avahi */
1158
srand((unsigned int) time(NULL));
1159
mc.simple_poll = avahi_simple_poll_new();
1160
if(mc.simple_poll == NULL){
1161
fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
1162
exitcode = EXIT_FAILURE;
1166
sigemptyset(&sigterm_action.sa_mask);
1167
ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
1169
perror("sigaddset");
1170
exitcode = EXIT_FAILURE;
1173
ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
1175
perror("sigaddset");
1176
exitcode = EXIT_FAILURE;
1179
ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
1181
perror("sigaddset");
1182
exitcode = EXIT_FAILURE;
1185
/* Need to check if the handler is SIG_IGN before handling:
1186
| [[info:libc:Initial Signal Actions]] |
1187
| [[info:libc:Basic Signal Handling]] |
1189
ret = sigaction(SIGINT, NULL, &old_sigterm_action);
1191
perror("sigaction");
1192
return EXIT_FAILURE;
1194
if(old_sigterm_action.sa_handler != SIG_IGN){
1195
ret = sigaction(SIGINT, &sigterm_action, NULL);
1197
perror("sigaction");
1198
exitcode = EXIT_FAILURE;
1202
ret = sigaction(SIGHUP, NULL, &old_sigterm_action);
1204
perror("sigaction");
1205
return EXIT_FAILURE;
1207
if(old_sigterm_action.sa_handler != SIG_IGN){
1208
ret = sigaction(SIGHUP, &sigterm_action, NULL);
1210
perror("sigaction");
1211
exitcode = EXIT_FAILURE;
1215
ret = sigaction(SIGTERM, NULL, &old_sigterm_action);
1217
perror("sigaction");
1218
return EXIT_FAILURE;
1220
if(old_sigterm_action.sa_handler != SIG_IGN){
1221
ret = sigaction(SIGTERM, &sigterm_action, NULL);
1223
perror("sigaction");
1224
exitcode = EXIT_FAILURE;
1229
/* If the interface is down, bring it up */
1230
if(interface[0] != '\0'){
1231
if_index = (AvahiIfIndex) if_nametoindex(interface);
1233
fprintf(stderr, "No such interface: \"%s\"\n", interface);
1234
exitcode = EXIT_FAILURE;
1242
/* Re-raise priviliges */
1250
/* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO
1251
messages to mess up the prompt */
1252
ret = klogctl(8, NULL, 5);
1253
bool restore_loglevel = true;
1255
restore_loglevel = false;
1258
#endif /* __linux__ */
1260
sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
1263
exitcode = EXIT_FAILURE;
1265
if(restore_loglevel){
1266
ret = klogctl(7, NULL, 0);
1271
#endif /* __linux__ */
1272
/* Lower privileges */
1280
strcpy(network.ifr_name, interface);
1281
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1283
perror("ioctl SIOCGIFFLAGS");
1285
if(restore_loglevel){
1286
ret = klogctl(7, NULL, 0);
1291
#endif /* __linux__ */
1292
exitcode = EXIT_FAILURE;
1293
/* Lower privileges */
1301
if((network.ifr_flags & IFF_UP) == 0){
1302
network.ifr_flags |= IFF_UP;
1303
take_down_interface = true;
1304
ret = ioctl(sd, SIOCSIFFLAGS, &network);
1306
take_down_interface = false;
1307
perror("ioctl SIOCSIFFLAGS");
1308
exitcode = EXIT_FAILURE;
1310
if(restore_loglevel){
1311
ret = klogctl(7, NULL, 0);
1316
#endif /* __linux__ */
1317
/* Lower privileges */
1326
/* sleep checking until interface is running */
1327
for(int i=0; i < delay * 4; i++){
1328
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1330
perror("ioctl SIOCGIFFLAGS");
1331
} else if(network.ifr_flags & IFF_RUNNING){
1334
struct timespec sleeptime = { .tv_nsec = 250000000 };
1335
ret = nanosleep(&sleeptime, NULL);
1336
if(ret == -1 and errno != EINTR){
1337
perror("nanosleep");
1340
if(not take_down_interface){
1341
/* We won't need the socket anymore */
1342
ret = (int)TEMP_FAILURE_RETRY(close(sd));
1348
if(restore_loglevel){
1349
/* Restores kernel loglevel to default */
1350
ret = klogctl(7, NULL, 0);
1355
#endif /* __linux__ */
1356
/* Lower privileges */
1358
if(take_down_interface){
1359
/* Lower privileges */
1365
/* Lower privileges permanently */
1377
ret = init_gnutls_global(pubkey, seckey);
1379
fprintf(stderr, "init_gnutls_global failed\n");
1380
exitcode = EXIT_FAILURE;
1383
gnutls_initialized = true;
1390
tempdir_created = true;
1391
if(mkdtemp(tempdir) == NULL){
1392
tempdir_created = false;
1401
if(not init_gpgme(pubkey, seckey, tempdir)){
1402
fprintf(stderr, "init_gpgme failed\n");
1403
exitcode = EXIT_FAILURE;
1406
gpgme_initialized = true;
1413
if(connect_to != NULL){
1414
/* Connect directly, do not use Zeroconf */
1415
/* (Mainly meant for debugging) */
1416
char *address = strrchr(connect_to, ':');
1417
if(address == NULL){
1418
fprintf(stderr, "No colon in address\n");
1419
exitcode = EXIT_FAILURE;
1429
tmpmax = strtoimax(address+1, &tmp, 10);
1430
if(errno != 0 or tmp == address+1 or *tmp != '\0'
1431
or tmpmax != (uint16_t)tmpmax){
1432
fprintf(stderr, "Bad port number\n");
1433
exitcode = EXIT_FAILURE;
1441
port = (uint16_t)tmpmax;
1443
address = connect_to;
1444
/* Colon in address indicates IPv6 */
1446
if(strchr(address, ':') != NULL){
1456
ret = start_mandos_communication(address, port, if_index, af);
1458
exitcode = EXIT_FAILURE;
1460
exitcode = EXIT_SUCCESS;
1470
AvahiServerConfig config;
1471
/* Do not publish any local Zeroconf records */
1472
avahi_server_config_init(&config);
1473
config.publish_hinfo = 0;
1474
config.publish_addresses = 0;
1475
config.publish_workstation = 0;
1476
config.publish_domain = 0;
1478
/* Allocate a new server */
1479
mc.server = avahi_server_new(avahi_simple_poll_get
1480
(mc.simple_poll), &config, NULL,
1483
/* Free the Avahi configuration data */
1484
avahi_server_config_free(&config);
1487
/* Check if creating the Avahi server object succeeded */
1488
if(mc.server == NULL){
1489
fprintf(stderr, "Failed to create Avahi server: %s\n",
1490
avahi_strerror(error));
1491
exitcode = EXIT_FAILURE;
1499
/* Create the Avahi service browser */
1500
sb = avahi_s_service_browser_new(mc.server, if_index,
1501
AVAHI_PROTO_UNSPEC, "_mandos._tcp",
1502
NULL, 0, browse_callback, NULL);
1504
fprintf(stderr, "Failed to create service browser: %s\n",
1505
avahi_strerror(avahi_server_errno(mc.server)));
1506
exitcode = EXIT_FAILURE;
1514
/* Run the main loop */
1517
fprintf(stderr, "Starting Avahi loop search\n");
1520
avahi_simple_poll_loop(mc.simple_poll);
1525
fprintf(stderr, "%s exiting\n", argv[0]);
1528
/* Cleanup things */
1530
avahi_s_service_browser_free(sb);
1532
if(mc.server != NULL)
1533
avahi_server_free(mc.server);
1535
if(mc.simple_poll != NULL)
1536
avahi_simple_poll_free(mc.simple_poll);
1538
if(gnutls_initialized){
1539
gnutls_certificate_free_credentials(mc.cred);
1540
gnutls_global_deinit();
1541
gnutls_dh_params_deinit(mc.dh_params);
1544
if(gpgme_initialized){
1545
gpgme_release(mc.ctx);
1548
/* Take down the network interface */
1549
if(take_down_interface){
1550
/* Re-raise priviliges */
1557
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1559
perror("ioctl SIOCGIFFLAGS");
1560
} else if(network.ifr_flags & IFF_UP) {
1561
network.ifr_flags &= ~IFF_UP; /* clear flag */
1562
ret = ioctl(sd, SIOCSIFFLAGS, &network);
1564
perror("ioctl SIOCSIFFLAGS");
1567
ret = (int)TEMP_FAILURE_RETRY(close(sd));
1571
/* Lower privileges permanently */
1580
/* Removes the temp directory used by GPGME */
1581
if(tempdir_created){
1583
struct dirent *direntry;
1584
d = opendir(tempdir);
1586
if(errno != ENOENT){
1591
direntry = readdir(d);
1592
if(direntry == NULL){
1595
/* Skip "." and ".." */
1596
if(direntry->d_name[0] == '.'
1597
and (direntry->d_name[1] == '\0'
1598
or (direntry->d_name[1] == '.'
1599
and direntry->d_name[2] == '\0'))){
1602
char *fullname = NULL;
1603
ret = asprintf(&fullname, "%s/%s", tempdir,
1609
ret = remove(fullname);
1611
fprintf(stderr, "remove(\"%s\"): %s\n", fullname,
1618
ret = rmdir(tempdir);
1619
if(ret == -1 and errno != ENOENT){
1625
sigemptyset(&old_sigterm_action.sa_mask);
1626
old_sigterm_action.sa_handler = SIG_DFL;
1627
ret = sigaction(signal_received, &old_sigterm_action, NULL);
1629
perror("sigaction");
1631
raise(signal_received);