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 © 2007-2008 Teddy Hogeborn & Björn Påhlsson
14
* This program is free software: you can redistribute it and/or
15
* modify it under the terms of the GNU General Public License as
16
* published by the Free Software Foundation, either version 3 of the
17
* License, or (at your option) any later version.
19
* This program is distributed in the hope that it will be useful, but
20
* WITHOUT ANY WARRANTY; without even the implied warranty of
21
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22
* General Public License for more details.
24
* You should have received a copy of the GNU General Public License
25
* along with this program. If not, see
26
* <http://www.gnu.org/licenses/>.
28
* Contact the authors at <mandos@fukt.bsnet.se>.
2
This file is part of avahi.
4
avahi is free software; you can redistribute it and/or modify it
5
under the terms of the GNU Lesser General Public License as
6
published by the Free Software Foundation; either version 2.1 of the
7
License, or (at your option) any later version.
9
avahi is distributed in the hope that it will be useful, but WITHOUT
10
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12
Public License for more details.
14
You should have received a copy of the GNU Lesser General Public
15
License along with avahi; if not, write to the Free Software
16
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
31
/* Needed by GPGME, specifically gpgme_data_seek() */
32
20
#define _LARGEFILE_SOURCE
33
21
#define _FILE_OFFSET_BITS 64
35
#define _GNU_SOURCE /* TEMP_FAILURE_RETRY() */
37
#include <stdio.h> /* fprintf(), stderr, fwrite(), stdout,
39
#include <stdint.h> /* uint16_t, uint32_t */
40
#include <stddef.h> /* NULL, size_t, ssize_t */
41
#include <stdlib.h> /* free(), EXIT_SUCCESS, EXIT_FAILURE,
43
#include <stdbool.h> /* bool, true */
44
#include <string.h> /* memset(), strcmp(), strlen(),
45
strerror(), memcpy(), strcpy() */
46
#include <sys/ioctl.h> /* ioctl */
47
#include <net/if.h> /* ifreq, SIOCGIFFLAGS, SIOCSIFFLAGS,
49
#include <sys/types.h> /* socket(), inet_pton(), sockaddr,
50
sockaddr_in6, PF_INET6,
51
SOCK_STREAM, INET6_ADDRSTRLEN,
53
#include <inttypes.h> /* PRIu16 */
54
#include <sys/socket.h> /* socket(), struct sockaddr_in6,
55
struct in6_addr, inet_pton(),
57
#include <assert.h> /* assert() */
58
#include <errno.h> /* perror(), errno */
59
#include <time.h> /* time() */
60
#include <net/if.h> /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
61
SIOCSIFFLAGS, if_indextoname(),
62
if_nametoindex(), IF_NAMESIZE */
63
#include <unistd.h> /* close(), SEEK_SET, off_t, write(),
64
getuid(), getgid(), setuid(),
66
#include <netinet/in.h>
67
#include <arpa/inet.h> /* inet_pton(), htons */
68
#include <iso646.h> /* not, and */
69
#include <argp.h> /* struct argp_option, error_t, struct
70
argp_state, struct argp,
71
argp_parse(), ARGP_KEY_ARG,
72
ARGP_KEY_END, ARGP_ERR_UNKNOWN */
75
/* All Avahi types, constants and functions
27
#include <net/if.h> /* if_nametoindex */
78
29
#include <avahi-core/core.h>
79
30
#include <avahi-core/lookup.h>
80
31
#include <avahi-core/log.h>
82
33
#include <avahi-common/malloc.h>
83
34
#include <avahi-common/error.h>
86
#include <gnutls/gnutls.h> /* All GnuTLS types, constants and functions
88
init_gnutls_session(),
90
#include <gnutls/openpgp.h> /* gnutls_certificate_set_openpgp_key_file(),
91
GNUTLS_OPENPGP_FMT_BASE64 */
94
#include <gpgme.h> /* All GPGME types, constants and functions
96
GPGME_PROTOCOL_OpenPGP,
37
#include <sys/types.h> /* socket(), setsockopt(), inet_pton() */
38
#include <sys/socket.h> /* socket(), setsockopt(), struct sockaddr_in6, struct in6_addr, inet_pton() */
39
#include <gnutls/gnutls.h> /* ALL GNUTLS STUFF */
40
#include <gnutls/openpgp.h> /* gnutls with openpgp stuff */
42
#include <unistd.h> /* close() */
43
#include <netinet/in.h>
44
#include <stdbool.h> /* true */
45
#include <string.h> /* memset */
46
#include <arpa/inet.h> /* inet_pton() */
47
#include <iso646.h> /* not */
50
#include <errno.h> /* perror() */
55
#define CERT_ROOT "/conf/conf.d/cryptkeyreq/"
57
#define CERTFILE CERT_ROOT "openpgp-client.txt"
58
#define KEYFILE CERT_ROOT "openpgp-client-key.txt"
99
59
#define BUFFER_SIZE 256
101
62
bool debug = false;
102
static const char *keydir = "/conf/conf.d/mandos";
103
static const char mandos_protocol_version[] = "1";
104
const char *argp_program_version = "password-request 1.0";
105
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
107
/* Used for passing in values through the Avahi callback functions */
109
AvahiSimplePoll *simple_poll;
65
gnutls_session_t session;
111
66
gnutls_certificate_credentials_t cred;
112
unsigned int dh_bits;
113
67
gnutls_dh_params_t dh_params;
114
const char *priority;
118
* Make room in "buffer" for at least BUFFER_SIZE additional bytes.
119
* "buffer_capacity" is how much is currently allocated,
120
* "buffer_length" is how much is already used.
122
size_t adjustbuffer(char **buffer, size_t buffer_length,
123
size_t buffer_capacity){
124
if (buffer_length + BUFFER_SIZE > buffer_capacity){
125
*buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
129
buffer_capacity += BUFFER_SIZE;
131
return buffer_capacity;
135
* Decrypt OpenPGP data using keyrings in HOMEDIR.
136
* Returns -1 on error
138
static ssize_t pgp_packet_decrypt (const char *cryptotext,
141
const char *homedir){
71
ssize_t gpg_packet_decrypt (char *packet, size_t packet_size, char **new_packet, char *homedir){
142
72
gpgme_data_t dh_crypto, dh_plain;
146
size_t plaintext_capacity = 0;
147
ssize_t plaintext_length = 0;
76
size_t new_packet_capacity = 0;
77
size_t new_packet_length = 0;
148
78
gpgme_engine_info_t engine_info;
151
fprintf(stderr, "Trying to decrypt OpenPGP data\n");
81
fprintf(stderr, "Attempting to decrypt password from gpg packet\n");
155
85
gpgme_check_version(NULL);
156
rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
157
if (rc != GPG_ERR_NO_ERROR){
158
fprintf(stderr, "bad gpgme_engine_check_version: %s: %s\n",
159
gpgme_strsource(rc), gpgme_strerror(rc));
86
gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);
163
/* Set GPGME home directory for the OpenPGP engine only */
88
/* Set GPGME home directory */
164
89
rc = gpgme_get_engine_info (&engine_info);
165
90
if (rc != GPG_ERR_NO_ERROR){
166
91
fprintf(stderr, "bad gpgme_get_engine_info: %s: %s\n",
311
/* GnuTLS log function callback */
312
static void debuggnutls(__attribute__((unused)) int level,
314
fprintf(stderr, "GnuTLS: %s", string);
216
void debuggnutls(int level, const char* string){
217
fprintf(stderr, "%s", string);
317
static int init_gnutls_global(mandos_context *mc,
318
const char *pubkeyfile,
319
const char *seckeyfile){
220
int initgnutls(encrypted_session *es){
323
fprintf(stderr, "Initializing GnuTLS\n");
225
fprintf(stderr, "Initializing gnutls\n");
326
ret = gnutls_global_init();
327
if (ret != GNUTLS_E_SUCCESS) {
328
fprintf (stderr, "GnuTLS global_init: %s\n",
329
safer_gnutls_strerror(ret));
229
if ((ret = gnutls_global_init ())
230
!= GNUTLS_E_SUCCESS) {
231
fprintf (stderr, "global_init: %s\n", safer_gnutls_strerror(ret));
334
/* "Use a log level over 10 to enable all debugging options."
337
236
gnutls_global_set_log_level(11);
338
237
gnutls_global_set_log_function(debuggnutls);
341
/* OpenPGP credentials */
342
gnutls_certificate_allocate_credentials(&mc->cred);
343
if (ret != GNUTLS_E_SUCCESS){
344
fprintf (stderr, "GnuTLS memory error: %s\n",
345
safer_gnutls_strerror(ret));
346
gnutls_global_deinit ();
241
/* openpgp credentials */
242
if ((ret = gnutls_certificate_allocate_credentials (&es->cred))
243
!= GNUTLS_E_SUCCESS) {
244
fprintf (stderr, "memory error: %s\n", safer_gnutls_strerror(ret));
351
fprintf(stderr, "Attempting to use OpenPGP certificate %s"
352
" and keyfile %s as GnuTLS credentials\n", pubkeyfile,
249
fprintf(stderr, "Attempting to use openpgp certificate %s"
250
" and keyfile %s as gnutls credentials\n", CERTFILE, KEYFILE);
356
253
ret = gnutls_certificate_set_openpgp_key_file
357
(mc->cred, pubkeyfile, seckeyfile, GNUTLS_OPENPGP_FMT_BASE64);
358
if (ret != GNUTLS_E_SUCCESS) {
360
"Error[%d] while reading the OpenPGP key pair ('%s',"
361
" '%s')\n", ret, pubkeyfile, seckeyfile);
362
fprintf(stdout, "The GnuTLS error is: %s\n",
363
safer_gnutls_strerror(ret));
367
/* GnuTLS server initialization */
368
ret = gnutls_dh_params_init(&mc->dh_params);
369
if (ret != GNUTLS_E_SUCCESS) {
370
fprintf (stderr, "Error in GnuTLS DH parameter initialization:"
371
" %s\n", safer_gnutls_strerror(ret));
374
ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
375
if (ret != GNUTLS_E_SUCCESS) {
376
fprintf (stderr, "Error in GnuTLS prime generation: %s\n",
377
safer_gnutls_strerror(ret));
381
gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
387
gnutls_certificate_free_credentials(mc->cred);
388
gnutls_global_deinit();
393
static int init_gnutls_session(mandos_context *mc,
394
gnutls_session_t *session){
396
/* GnuTLS session creation */
397
ret = gnutls_init(session, GNUTLS_SERVER);
398
if (ret != GNUTLS_E_SUCCESS){
399
fprintf(stderr, "Error in GnuTLS session initialization: %s\n",
400
safer_gnutls_strerror(ret));
405
ret = gnutls_priority_set_direct(*session, mc->priority, &err);
406
if (ret != GNUTLS_E_SUCCESS) {
407
fprintf(stderr, "Syntax error at: %s\n", err);
408
fprintf(stderr, "GnuTLS error: %s\n",
409
safer_gnutls_strerror(ret));
410
gnutls_deinit (*session);
415
ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
417
if (ret != GNUTLS_E_SUCCESS) {
418
fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
419
safer_gnutls_strerror(ret));
420
gnutls_deinit (*session);
254
(es->cred, CERTFILE, KEYFILE, GNUTLS_OPENPGP_FMT_BASE64);
255
if (ret != GNUTLS_E_SUCCESS) {
257
(stderr, "Error[%d] while reading the OpenPGP key pair ('%s', '%s')\n",
258
ret, CERTFILE, KEYFILE);
259
fprintf(stdout, "The Error is: %s\n",
260
safer_gnutls_strerror(ret));
264
//Gnutls server initialization
265
if ((ret = gnutls_dh_params_init (&es->dh_params))
266
!= GNUTLS_E_SUCCESS) {
267
fprintf (stderr, "Error in dh parameter initialization: %s\n",
268
safer_gnutls_strerror(ret));
272
if ((ret = gnutls_dh_params_generate2 (es->dh_params, DH_BITS))
273
!= GNUTLS_E_SUCCESS) {
274
fprintf (stderr, "Error in prime generation: %s\n",
275
safer_gnutls_strerror(ret));
279
gnutls_certificate_set_dh_params (es->cred, es->dh_params);
281
// Gnutls session creation
282
if ((ret = gnutls_init (&es->session, GNUTLS_SERVER))
283
!= GNUTLS_E_SUCCESS){
284
fprintf(stderr, "Error in gnutls session initialization: %s\n",
285
safer_gnutls_strerror(ret));
288
if ((ret = gnutls_priority_set_direct (es->session, "NORMAL", &err))
289
!= GNUTLS_E_SUCCESS) {
290
fprintf(stderr, "Syntax error at: %s\n", err);
291
fprintf(stderr, "Gnutls error: %s\n",
292
safer_gnutls_strerror(ret));
296
if ((ret = gnutls_credentials_set
297
(es->session, GNUTLS_CRD_CERTIFICATE, es->cred))
298
!= GNUTLS_E_SUCCESS) {
299
fprintf(stderr, "Error setting a credentials set: %s\n",
300
safer_gnutls_strerror(ret));
424
304
/* ignore client certificate if any. */
425
gnutls_certificate_server_set_request (*session,
305
gnutls_certificate_server_set_request (es->session, GNUTLS_CERT_IGNORE);
428
gnutls_dh_set_prime_bits (*session, mc->dh_bits);
307
gnutls_dh_set_prime_bits (es->session, DH_BITS);
433
/* Avahi log function callback */
434
static void empty_log(__attribute__((unused)) AvahiLogLevel level,
435
__attribute__((unused)) const char *txt){}
312
void empty_log(AvahiLogLevel level, const char *txt){}
437
/* Called when a Mandos server is found */
438
static int start_mandos_communication(const char *ip, uint16_t port,
439
AvahiIfIndex if_index,
314
int start_mandos_communcation(char *ip, uint16_t port){
442
union { struct sockaddr in; struct sockaddr_in6 in6; } to;
316
struct sockaddr_in6 to;
317
encrypted_session es;
443
318
char *buffer = NULL;
444
319
char *decrypted_buffer;
445
320
size_t buffer_length = 0;
446
321
size_t buffer_capacity = 0;
447
322
ssize_t decrypted_buffer_size;
450
char interface[IF_NAMESIZE];
451
gnutls_session_t session;
453
ret = init_gnutls_session (mc, &session);
324
const char interface[] = "eth0";
459
fprintf(stderr, "Setting up a tcp connection to %s, port %" PRIu16
327
fprintf(stderr, "Setting up a tcp connection to %s\n", ip);
463
330
tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);
470
if(if_indextoname((unsigned int)if_index, interface) == NULL){
471
perror("if_indextoname");
474
337
fprintf(stderr, "Binding to interface %s\n", interface);
340
ret = setsockopt(tcp_sd, SOL_SOCKET, SO_BINDTODEVICE, interface, 5);
342
perror("setsockopt bindtodevice");
477
memset(&to,0,sizeof(to)); /* Spurious warning */
478
to.in6.sin6_family = AF_INET6;
479
/* It would be nice to have a way to detect if we were passed an
480
IPv4 address here. Now we assume an IPv6 address. */
481
ret = inet_pton(AF_INET6, ip, &to.in6.sin6_addr);
346
memset(&to,0,sizeof(to));
347
to.sin6_family = AF_INET6;
348
ret = inet_pton(AF_INET6, ip, &to.sin6_addr);
483
350
perror("inet_pton");
487
354
fprintf(stderr, "Bad address: %s\n", ip);
490
to.in6.sin6_port = htons(port); /* Spurious warning */
492
to.in6.sin6_scope_id = (uint32_t)if_index;
357
to.sin6_port = htons(port);
358
to.sin6_scope_id = if_nametoindex(interface);
495
fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,
497
char addrstr[INET6_ADDRSTRLEN] = "";
498
if(inet_ntop(to.in6.sin6_family, &(to.in6.sin6_addr), addrstr,
499
sizeof(addrstr)) == NULL){
502
if(strcmp(addrstr, ip) != 0){
503
fprintf(stderr, "Canonical address form: %s\n", addrstr);
361
fprintf(stderr, "Connection to: %s\n", ip);
508
ret = connect(tcp_sd, &to.in, sizeof(to));
364
ret = connect(tcp_sd, (struct sockaddr *) &to, sizeof(to));
510
366
perror("connect");
370
ret = initgnutls (&es);
377
gnutls_transport_set_ptr (es.session, (gnutls_transport_ptr_t) tcp_sd);
514
const char *out = mandos_protocol_version;
517
size_t out_size = strlen(out);
518
ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,
519
out_size - written));
525
written += (size_t)ret;
526
if(written < out_size){
529
if (out == mandos_protocol_version){
539
fprintf(stderr, "Establishing TLS session with %s\n", ip);
380
fprintf(stderr, "Establishing tls session with %s\n", ip);
542
gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) tcp_sd);
545
ret = gnutls_handshake (session);
546
} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
384
ret = gnutls_handshake (es.session);
548
386
if (ret != GNUTLS_E_SUCCESS){
550
fprintf(stderr, "*** GnuTLS Handshake failed ***\n");
387
fprintf(stderr, "\n*** Handshake failed ***\n");
557
/* Read OpenPGP packet that contains the wanted password */
393
//Retrieve gpg packet that contains the wanted password
560
fprintf(stderr, "Retrieving pgp encrypted password from %s\n",
396
fprintf(stderr, "Retrieving pgp encrypted password from %s\n", ip);
565
buffer_capacity = adjustbuffer(&buffer, buffer_length,
567
if (buffer_capacity == 0){
568
perror("adjustbuffer");
400
if (buffer_length + BUFFER_SIZE > buffer_capacity){
401
buffer = realloc(buffer, buffer_capacity + BUFFER_SIZE);
406
buffer_capacity += BUFFER_SIZE;
573
ret = gnutls_record_recv(session, buffer+buffer_length,
409
ret = gnutls_record_recv
410
(es.session, buffer+buffer_length, BUFFER_SIZE);
581
417
case GNUTLS_E_AGAIN:
583
419
case GNUTLS_E_REHANDSHAKE:
585
ret = gnutls_handshake (session);
586
} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);
420
ret = gnutls_handshake (es.session);
588
fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");
422
fprintf(stderr, "\n*** Handshake failed ***\n");
589
423
gnutls_perror (ret);
595
fprintf(stderr, "Unknown error while reading data from"
596
" encrypted session with Mandos server\n");
429
fprintf(stderr, "Unknown error while reading data from encrypted session with mandos server\n");
598
gnutls_bye (session, GNUTLS_SHUT_RDWR);
431
gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
602
buffer_length += (size_t) ret;
435
buffer_length += ret;
607
fprintf(stderr, "Closing TLS session\n");
610
gnutls_bye (session, GNUTLS_SHUT_RDWR);
612
439
if (buffer_length > 0){
613
decrypted_buffer_size = pgp_packet_decrypt(buffer,
617
if (decrypted_buffer_size >= 0){
619
while(written < (size_t) decrypted_buffer_size){
620
ret = (int)fwrite (decrypted_buffer + written, 1,
621
(size_t)decrypted_buffer_size - written,
623
if(ret == 0 and ferror(stdout)){
625
fprintf(stderr, "Error writing encrypted data: %s\n",
631
written += (size_t)ret;
440
if ((decrypted_buffer_size = gpg_packet_decrypt(buffer, buffer_length, &decrypted_buffer, CERT_ROOT)) >= 0){
441
fwrite (decrypted_buffer, 1, decrypted_buffer_size, stdout);
633
442
free(decrypted_buffer);
639
/* Shutdown procedure */
451
fprintf(stderr, "Closing tls session\n");
455
gnutls_bye (es.session, GNUTLS_SHUT_RDWR);
644
gnutls_deinit (session);
458
gnutls_deinit (es.session);
459
gnutls_certificate_free_credentials (es.cred);
460
gnutls_global_deinit ();
648
static void resolve_callback(AvahiSServiceResolver *r,
649
AvahiIfIndex interface,
650
AVAHI_GCC_UNUSED AvahiProtocol protocol,
651
AvahiResolverEvent event,
655
const char *host_name,
656
const AvahiAddress *address,
658
AVAHI_GCC_UNUSED AvahiStringList *txt,
659
AVAHI_GCC_UNUSED AvahiLookupResultFlags
662
mandos_context *mc = userdata;
663
assert(r); /* Spurious warning */
665
/* Called whenever a service has been resolved successfully or
670
case AVAHI_RESOLVER_FAILURE:
671
fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
672
" of type '%s' in domain '%s': %s\n", name, type, domain,
673
avahi_strerror(avahi_server_errno(mc->server)));
676
case AVAHI_RESOLVER_FOUND:
678
char ip[AVAHI_ADDRESS_STR_MAX];
679
avahi_address_snprint(ip, sizeof(ip), address);
681
fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"
682
PRIu16 ") on port %d\n", name, host_name, ip,
685
int ret = start_mandos_communication(ip, port, interface, mc);
691
avahi_s_service_resolver_free(r);
694
static void browse_callback( AvahiSServiceBrowser *b,
695
AvahiIfIndex interface,
696
AvahiProtocol protocol,
697
AvahiBrowserEvent event,
701
AVAHI_GCC_UNUSED AvahiLookupResultFlags
704
mandos_context *mc = userdata;
705
assert(b); /* Spurious warning */
707
/* Called whenever a new services becomes available on the LAN or
708
is removed from the LAN */
712
case AVAHI_BROWSER_FAILURE:
714
fprintf(stderr, "(Avahi browser) %s\n",
715
avahi_strerror(avahi_server_errno(mc->server)));
716
avahi_simple_poll_quit(mc->simple_poll);
719
case AVAHI_BROWSER_NEW:
720
/* We ignore the returned Avahi resolver object. In the callback
721
function we free it. If the Avahi server is terminated before
722
the callback function is called the Avahi server will free the
725
if (!(avahi_s_service_resolver_new(mc->server, interface,
726
protocol, name, type, domain,
727
AVAHI_PROTO_INET6, 0,
728
resolve_callback, mc)))
729
fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
730
name, avahi_strerror(avahi_server_errno(mc->server)));
733
case AVAHI_BROWSER_REMOVE:
736
case AVAHI_BROWSER_ALL_FOR_NOW:
737
case AVAHI_BROWSER_CACHE_EXHAUSTED:
739
fprintf(stderr, "No Mandos server found, still searching...\n");
745
/* Combines file name and path and returns the malloced new
746
string. some sane checks could/should be added */
747
static const char *combinepath(const char *first, const char *second){
748
size_t f_len = strlen(first);
749
size_t s_len = strlen(second);
750
char *tmp = malloc(f_len + s_len + 2);
755
memcpy(tmp, first, f_len); /* Spurious warning */
759
memcpy(tmp + f_len + 1, second, s_len); /* Spurious warning */
761
tmp[f_len + 1 + s_len] = '\0';
766
int main(int argc, char *argv[]){
464
static AvahiSimplePoll *simple_poll = NULL;
465
static AvahiServer *server = NULL;
467
static void resolve_callback(
468
AvahiSServiceResolver *r,
469
AVAHI_GCC_UNUSED AvahiIfIndex interface,
470
AVAHI_GCC_UNUSED AvahiProtocol protocol,
471
AvahiResolverEvent event,
475
const char *host_name,
476
const AvahiAddress *address,
478
AvahiStringList *txt,
479
AvahiLookupResultFlags flags,
480
AVAHI_GCC_UNUSED void* userdata) {
484
/* Called whenever a service has been resolved successfully or timed out */
487
case AVAHI_RESOLVER_FAILURE:
488
fprintf(stderr, "(Resolver) Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_server_errno(server)));
491
case AVAHI_RESOLVER_FOUND: {
492
char ip[AVAHI_ADDRESS_STR_MAX];
493
avahi_address_snprint(ip, sizeof(ip), address);
495
fprintf(stderr, "Mandos server found at %s on port %d\n", ip, port);
497
int ret = start_mandos_communcation(ip, port);
505
avahi_s_service_resolver_free(r);
508
static void browse_callback(
509
AvahiSServiceBrowser *b,
510
AvahiIfIndex interface,
511
AvahiProtocol protocol,
512
AvahiBrowserEvent event,
516
AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
519
AvahiServer *s = userdata;
522
/* Called whenever a new services becomes available on the LAN or is removed from the LAN */
526
case AVAHI_BROWSER_FAILURE:
528
fprintf(stderr, "(Browser) %s\n", avahi_strerror(avahi_server_errno(server)));
529
avahi_simple_poll_quit(simple_poll);
532
case AVAHI_BROWSER_NEW:
533
/* We ignore the returned resolver object. In the callback
534
function we free it. If the server is terminated before
535
the callback function is called the server will free
536
the resolver for us. */
538
if (!(avahi_s_service_resolver_new(s, interface, protocol, name, type, domain, AVAHI_PROTO_INET6, 0, resolve_callback, s)))
539
fprintf(stderr, "Failed to resolve service '%s': %s\n", name, avahi_strerror(avahi_server_errno(s)));
543
case AVAHI_BROWSER_REMOVE:
546
case AVAHI_BROWSER_ALL_FOR_NOW:
547
case AVAHI_BROWSER_CACHE_EXHAUSTED:
552
int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char*argv[]) {
553
AvahiServerConfig config;
767
554
AvahiSServiceBrowser *sb = NULL;
555
const char db[] = "--debug";
770
int exitcode = EXIT_SUCCESS;
771
const char *interface = "eth0";
772
struct ifreq network;
776
char *connect_to = NULL;
777
AvahiIfIndex if_index = AVAHI_IF_UNSPEC;
778
const char *pubkeyfile = "pubkey.txt";
779
const char *seckeyfile = "seckey.txt";
780
mandos_context mc = { .simple_poll = NULL, .server = NULL,
781
.dh_bits = 1024, .priority = "SECURE256"};
782
bool gnutls_initalized = false;
558
int returncode = EXIT_SUCCESS;
559
char *basename = rindex(argv[0], '/');
560
if(basename == NULL){
785
struct argp_option options[] = {
786
{ .name = "debug", .key = 128,
787
.doc = "Debug mode", .group = 3 },
788
{ .name = "connect", .key = 'c',
790
.doc = "Connect directly to a sepcified mandos server",
792
{ .name = "interface", .key = 'i',
794
.doc = "Interface that Avahi will conntect through",
796
{ .name = "keydir", .key = 'd',
798
.doc = "Directory where the openpgp keyring is",
800
{ .name = "seckey", .key = 's',
802
.doc = "Secret openpgp key for gnutls authentication",
804
{ .name = "pubkey", .key = 'p',
806
.doc = "Public openpgp key for gnutls authentication",
808
{ .name = "dh-bits", .key = 129,
810
.doc = "dh-bits to use in gnutls communication",
812
{ .name = "priority", .key = 130,
814
.doc = "GNUTLS priority", .group = 1 },
566
char *program_name = malloc(strlen(basename) + sizeof(db));
819
error_t parse_opt (int key, char *arg,
820
struct argp_state *state) {
821
/* Get the INPUT argument from `argp_parse', which we know is
822
a pointer to our plugin list pointer. */
844
mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);
568
if (program_name == NULL){
573
program_name[0] = '\0';
575
for (int i = 1; i < argc; i++){
576
if (not strncmp(argv[i], db, 5)){
577
strcat(strcat(strcat(program_name, db ), "="), basename);
578
if(not strcmp(argv[i], db) or not strcmp(argv[i], program_name)){
859
return ARGP_ERR_UNKNOWN;
864
struct argp argp = { .options = options, .parser = parse_opt,
866
.doc = "Mandos client -- Get and decrypt"
867
" passwords from mandos server" };
868
ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
869
if (ret == ARGP_ERR_UNKNOWN){
870
fprintf(stderr, "Unkown error while parsing arguments\n");
871
exitcode = EXIT_FAILURE;
876
pubkeyfile = combinepath(keydir, pubkeyfile);
877
if (pubkeyfile == NULL){
878
perror("combinepath");
879
exitcode = EXIT_FAILURE;
883
seckeyfile = combinepath(keydir, seckeyfile);
884
if (seckeyfile == NULL){
885
perror("combinepath");
889
ret = init_gnutls_global(&mc, pubkeyfile, seckeyfile);
891
fprintf(stderr, "init_gnutls_global\n");
894
gnutls_initalized = true;
910
if_index = (AvahiIfIndex) if_nametoindex(interface);
912
fprintf(stderr, "No such interface: \"%s\"\n", interface);
916
if(connect_to != NULL){
917
/* Connect directly, do not use Zeroconf */
918
/* (Mainly meant for debugging) */
919
char *address = strrchr(connect_to, ':');
921
fprintf(stderr, "No colon in address\n");
922
exitcode = EXIT_FAILURE;
926
uint16_t port = (uint16_t) strtol(address+1, NULL, 10);
928
perror("Bad port number");
929
exitcode = EXIT_FAILURE;
933
address = connect_to;
934
ret = start_mandos_communication(address, port, if_index, &mc);
936
exitcode = EXIT_FAILURE;
938
exitcode = EXIT_SUCCESS;
943
/* If the interface is down, bring it up */
945
sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
948
exitcode = EXIT_FAILURE;
951
strcpy(network.ifr_name, interface); /* Spurious warning */
952
ret = ioctl(sd, SIOCGIFFLAGS, &network);
954
perror("ioctl SIOCGIFFLAGS");
955
exitcode = EXIT_FAILURE;
958
if((network.ifr_flags & IFF_UP) == 0){
959
network.ifr_flags |= IFF_UP;
960
ret = ioctl(sd, SIOCSIFFLAGS, &network);
962
perror("ioctl SIOCSIFFLAGS");
963
exitcode = EXIT_FAILURE;
971
586
avahi_set_log_function(empty_log);
974
/* Initialize the pseudo-RNG for Avahi */
975
srand((unsigned int) time(NULL));
977
/* Allocate main Avahi loop object */
978
mc.simple_poll = avahi_simple_poll_new();
979
if (mc.simple_poll == NULL) {
980
fprintf(stderr, "Avahi: Failed to create simple poll"
982
exitcode = EXIT_FAILURE;
987
AvahiServerConfig config;
988
/* Do not publish any local Zeroconf records */
989
avahi_server_config_init(&config);
990
config.publish_hinfo = 0;
991
config.publish_addresses = 0;
992
config.publish_workstation = 0;
993
config.publish_domain = 0;
995
/* Allocate a new server */
996
mc.server = avahi_server_new(avahi_simple_poll_get
997
(mc.simple_poll), &config, NULL,
1000
/* Free the Avahi configuration data */
1001
avahi_server_config_free(&config);
1004
/* Check if creating the Avahi server object succeeded */
1005
if (mc.server == NULL) {
1006
fprintf(stderr, "Failed to create Avahi server: %s\n",
1007
avahi_strerror(error));
1008
exitcode = EXIT_FAILURE;
1012
/* Create the Avahi service browser */
1013
sb = avahi_s_service_browser_new(mc.server, if_index,
1015
"_mandos._tcp", NULL, 0,
1016
browse_callback, &mc);
1018
fprintf(stderr, "Failed to create service browser: %s\n",
1019
avahi_strerror(avahi_server_errno(mc.server)));
1020
exitcode = EXIT_FAILURE;
589
/* Initialize the psuedo-RNG */
592
/* Allocate main loop object */
593
if (!(simple_poll = avahi_simple_poll_new())) {
594
fprintf(stderr, "Failed to create simple poll object.\n");
599
/* Do not publish any local records */
600
avahi_server_config_init(&config);
601
config.publish_hinfo = 0;
602
config.publish_addresses = 0;
603
config.publish_workstation = 0;
604
config.publish_domain = 0;
606
/* Allocate a new server */
607
server = avahi_server_new(avahi_simple_poll_get(simple_poll), &config, NULL, NULL, &error);
609
/* Free the configuration data */
610
avahi_server_config_free(&config);
612
/* Check if creating the server object succeeded */
614
fprintf(stderr, "Failed to create server: %s\n", avahi_strerror(error));
615
returncode = EXIT_FAILURE;
619
/* Create the service browser */
620
if (!(sb = avahi_s_service_browser_new(server, if_nametoindex("eth0"), AVAHI_PROTO_INET6, "_mandos._tcp", NULL, 0, browse_callback, server))) {
621
fprintf(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_server_errno(server)));
622
returncode = EXIT_FAILURE;
1024
626
/* Run the main loop */
1027
fprintf(stderr, "Starting Avahi loop search\n");
629
fprintf(stderr, "Starting avahi loop search\n");
1030
avahi_simple_poll_loop(mc.simple_poll);
632
avahi_simple_poll_loop(simple_poll);
1035
637
fprintf(stderr, "%s exiting\n", argv[0]);
1038
640
/* Cleanup things */
1040
642
avahi_s_service_browser_free(sb);
1042
if (mc.server != NULL)
1043
avahi_server_free(mc.server);
1045
if (mc.simple_poll != NULL)
1046
avahi_simple_poll_free(mc.simple_poll);
1050
if (gnutls_initalized){
1051
gnutls_certificate_free_credentials(mc.cred);
1052
gnutls_global_deinit ();
645
avahi_server_free(server);
648
avahi_simple_poll_free(simple_poll);