75
75
argp_state, struct argp,
76
76
argp_parse(), ARGP_KEY_ARG,
77
77
ARGP_KEY_END, ARGP_ERR_UNKNOWN */
78
#include <signal.h> /* sigemptyset(), sigaddset(),
79
sigaction(), SIGTERM, sigaction,
83
79
#include <sys/klog.h> /* klogctl() */
84
#endif /* __linux__ */
87
83
/* All Avahi types, constants and functions
134
130
} mandos_context;
136
/* global context so signal handler can reach it*/
137
mandos_context mc = { .simple_poll = NULL, .server = NULL,
138
.dh_bits = 1024, .priority = "SECURE256"
139
":!CTYPE-X.509:+CTYPE-OPENPGP" };
142
* Make additional room in "buffer" for at least BUFFER_SIZE more
143
* bytes. "buffer_capacity" is how much is currently allocated,
144
* "buffer_length" is how much is already used.
133
* Make additional room in "buffer" for at least BUFFER_SIZE
134
* additional bytes. "buffer_capacity" is how much is currently
135
* allocated, "buffer_length" is how much is already used.
146
137
size_t incbuffer(char **buffer, size_t buffer_length,
147
138
size_t buffer_capacity){
159
150
* Initialize GPGME.
161
static bool init_gpgme(const char *seckey,
152
static bool init_gpgme(mandos_context *mc, const char *seckey,
162
153
const char *pubkey, const char *tempdir){
164
155
gpgme_error_t rc;
188
rc = gpgme_op_import(mc.ctx, pgp_data);
179
rc = gpgme_op_import(mc->ctx, pgp_data);
189
180
if(rc != GPG_ERR_NO_ERROR){
190
181
fprintf(stderr, "bad gpgme_op_import: %s: %s\n",
191
182
gpgme_strsource(rc), gpgme_strerror(rc));
236
227
/* Create new GPGME "context" */
237
rc = gpgme_new(&(mc.ctx));
228
rc = gpgme_new(&(mc->ctx));
238
229
if(rc != GPG_ERR_NO_ERROR){
239
230
fprintf(stderr, "bad gpgme_new: %s: %s\n",
240
231
gpgme_strsource(rc), gpgme_strerror(rc));
252
243
* Decrypt OpenPGP data.
253
244
* Returns -1 on error
255
static ssize_t pgp_packet_decrypt(const char *cryptotext,
246
static ssize_t pgp_packet_decrypt(const mandos_context *mc,
247
const char *cryptotext,
256
248
size_t crypto_size,
257
249
char **plaintext){
258
250
gpgme_data_t dh_crypto, dh_plain;
286
278
/* Decrypt data from the cryptotext data buffer to the plaintext
288
rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);
280
rc = gpgme_op_decrypt(mc->ctx, dh_crypto, dh_plain);
289
281
if(rc != GPG_ERR_NO_ERROR){
290
282
fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
291
283
gpgme_strsource(rc), gpgme_strerror(rc));
292
284
plaintext_length = -1;
294
286
gpgme_decrypt_result_t result;
295
result = gpgme_op_decrypt_result(mc.ctx);
287
result = gpgme_op_decrypt_result(mc->ctx);
296
288
if(result == NULL){
297
289
fprintf(stderr, "gpgme_op_decrypt_result failed\n");
390
382
fprintf(stderr, "GnuTLS: %s", string);
393
static int init_gnutls_global(const char *pubkeyfilename,
385
static int init_gnutls_global(mandos_context *mc,
386
const char *pubkeyfilename,
394
387
const char *seckeyfilename){
416
409
/* OpenPGP credentials */
417
gnutls_certificate_allocate_credentials(&mc.cred);
410
gnutls_certificate_allocate_credentials(&mc->cred);
418
411
if(ret != GNUTLS_E_SUCCESS){
419
412
fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning
434
427
ret = gnutls_certificate_set_openpgp_key_file
435
(mc.cred, pubkeyfilename, seckeyfilename,
428
(mc->cred, pubkeyfilename, seckeyfilename,
436
429
GNUTLS_OPENPGP_FMT_BASE64);
437
430
if(ret != GNUTLS_E_SUCCESS){
446
439
/* GnuTLS server initialization */
447
ret = gnutls_dh_params_init(&mc.dh_params);
440
ret = gnutls_dh_params_init(&mc->dh_params);
448
441
if(ret != GNUTLS_E_SUCCESS){
449
442
fprintf(stderr, "Error in GnuTLS DH parameter initialization:"
450
443
" %s\n", safer_gnutls_strerror(ret));
453
ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);
446
ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
454
447
if(ret != GNUTLS_E_SUCCESS){
455
448
fprintf(stderr, "Error in GnuTLS prime generation: %s\n",
456
449
safer_gnutls_strerror(ret));
460
gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);
453
gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
466
gnutls_certificate_free_credentials(mc.cred);
459
gnutls_certificate_free_credentials(mc->cred);
467
460
gnutls_global_deinit();
468
gnutls_dh_params_deinit(mc.dh_params);
461
gnutls_dh_params_deinit(mc->dh_params);
472
static int init_gnutls_session(gnutls_session_t *session){
465
static int init_gnutls_session(mandos_context *mc,
466
gnutls_session_t *session){
474
468
/* GnuTLS session creation */
475
469
ret = gnutls_init(session, GNUTLS_SERVER);
483
ret = gnutls_priority_set_direct(*session, mc.priority, &err);
477
ret = gnutls_priority_set_direct(*session, mc->priority, &err);
484
478
if(ret != GNUTLS_E_SUCCESS){
485
479
fprintf(stderr, "Syntax error at: %s\n", err);
486
480
fprintf(stderr, "GnuTLS error: %s\n",
493
487
ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
495
489
if(ret != GNUTLS_E_SUCCESS){
496
490
fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
497
491
safer_gnutls_strerror(ret));
503
497
gnutls_certificate_server_set_request(*session,
504
498
GNUTLS_CERT_IGNORE);
506
gnutls_dh_set_prime_bits(*session, mc.dh_bits);
500
gnutls_dh_set_prime_bits(*session, mc->dh_bits);
515
509
/* Called when a Mandos server is found */
516
510
static int start_mandos_communication(const char *ip, uint16_t port,
517
511
AvahiIfIndex if_index,
512
mandos_context *mc, int af){
739
733
gnutls_bye(session, GNUTLS_SHUT_RDWR);
741
735
if(buffer_length > 0){
742
decrypted_buffer_size = pgp_packet_decrypt(buffer,
736
decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,
744
738
&decrypted_buffer);
745
739
if(decrypted_buffer_size >= 0){
791
785
AVAHI_GCC_UNUSED AvahiStringList *txt,
792
786
AVAHI_GCC_UNUSED AvahiLookupResultFlags
794
AVAHI_GCC_UNUSED void* userdata){
789
mandos_context *mc = userdata;
797
792
/* Called whenever a service has been resolved successfully or
802
797
case AVAHI_RESOLVER_FAILURE:
803
798
fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
804
799
" of type '%s' in domain '%s': %s\n", name, type, domain,
805
avahi_strerror(avahi_server_errno(mc.server)));
800
avahi_strerror(avahi_server_errno(mc->server)));
808
803
case AVAHI_RESOLVER_FOUND:
814
809
PRIdMAX ") on port %" PRIu16 "\n", name, host_name,
815
810
ip, (intmax_t)interface, port);
817
int ret = start_mandos_communication(ip, port, interface,
812
int ret = start_mandos_communication(ip, port, interface, mc,
818
813
avahi_proto_to_af(proto));
820
avahi_simple_poll_quit(mc.simple_poll);
815
avahi_simple_poll_quit(mc->simple_poll);
844
840
case AVAHI_BROWSER_FAILURE:
846
842
fprintf(stderr, "(Avahi browser) %s\n",
847
avahi_strerror(avahi_server_errno(mc.server)));
848
avahi_simple_poll_quit(mc.simple_poll);
843
avahi_strerror(avahi_server_errno(mc->server)));
844
avahi_simple_poll_quit(mc->simple_poll);
851
847
case AVAHI_BROWSER_NEW:
854
850
the callback function is called the Avahi server will free the
855
851
resolver for us. */
857
if(!(avahi_s_service_resolver_new(mc.server, interface,
853
if(!(avahi_s_service_resolver_new(mc->server, interface,
858
854
protocol, name, type, domain,
859
855
AVAHI_PROTO_INET6, 0,
860
resolve_callback, NULL)))
856
resolve_callback, mc)))
861
857
fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
862
name, avahi_strerror(avahi_server_errno(mc.server)));
858
name, avahi_strerror(avahi_server_errno(mc->server)));
865
861
case AVAHI_BROWSER_REMOVE:
877
sig_atomic_t quit_now = 0;
879
/* stop main loop after sigterm has been called */
880
static void handle_sigterm(__attribute__((unused)) int sig){
885
int old_errno = errno;
886
if(mc.simple_poll != NULL){
887
avahi_simple_poll_quit(mc.simple_poll);
892
873
int main(int argc, char *argv[]){
893
874
AvahiSServiceBrowser *sb = NULL;
908
889
const char *seckey = PATHDIR "/" SECKEY;
909
890
const char *pubkey = PATHDIR "/" PUBKEY;
911
/* Initialize Mandos context */
912
mc = (mandos_context){ .simple_poll = NULL, .server = NULL,
913
.dh_bits = 1024, .priority = "SECURE256"
914
":!CTYPE-X.509:+CTYPE-OPENPGP" };
892
mandos_context mc = { .simple_poll = NULL, .server = NULL,
893
.dh_bits = 1024, .priority = "SECURE256"
894
":!CTYPE-X.509:+CTYPE-OPENPGP" };
915
895
bool gnutls_initialized = false;
916
896
bool gpgme_initialized = false;
917
897
double delay = 2.5;
919
struct sigaction old_sigterm_action;
920
struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
923
900
struct argp_option options[] = {
1019
avahi_set_log_function(empty_log);
1022
/* Initialize Avahi early so avahi_simple_poll_quit() can be called
1023
from the signal handler */
1024
/* Initialize the pseudo-RNG for Avahi */
1025
srand((unsigned int) time(NULL));
1026
mc.simple_poll = avahi_simple_poll_new();
1027
if(mc.simple_poll == NULL){
1028
fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
1029
exitcode = EXIT_FAILURE;
1033
sigemptyset(&sigterm_action.sa_mask);
1034
ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
1036
perror("sigaddset");
1037
exitcode = EXIT_FAILURE;
1040
ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
1042
perror("sigaddset");
1043
exitcode = EXIT_FAILURE;
1046
ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
1048
perror("sigaddset");
1049
exitcode = EXIT_FAILURE;
1052
ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1054
perror("sigaction");
1055
exitcode = EXIT_FAILURE;
1059
995
/* If the interface is down, bring it up */
1060
996
if(interface[0] != '\0'){
1061
997
#ifdef __linux__
1158
1094
perror("setuid");
1161
ret = init_gnutls_global(pubkey, seckey);
1097
ret = init_gnutls_global(&mc, pubkey, seckey);
1163
1099
fprintf(stderr, "init_gnutls_global failed\n");
1164
1100
exitcode = EXIT_FAILURE;
1174
1110
tempdir_created = true;
1176
if(not init_gpgme(pubkey, seckey, tempdir)){
1112
if(not init_gpgme(&mc, pubkey, seckey, tempdir)){
1177
1113
fprintf(stderr, "init_gpgme failed\n");
1178
1114
exitcode = EXIT_FAILURE;
1220
ret = start_mandos_communication(address, port, if_index, af);
1156
ret = start_mandos_communication(address, port, if_index, &mc,
1222
1159
exitcode = EXIT_FAILURE;
1167
avahi_set_log_function(empty_log);
1170
/* Initialize the pseudo-RNG for Avahi */
1171
srand((unsigned int) time(NULL));
1173
/* Allocate main Avahi loop object */
1174
mc.simple_poll = avahi_simple_poll_new();
1175
if(mc.simple_poll == NULL){
1176
fprintf(stderr, "Avahi: Failed to create simple poll object.\n");
1177
exitcode = EXIT_FAILURE;
1230
1182
AvahiServerConfig config;
1231
1183
/* Do not publish any local Zeroconf records */
1255
1207
/* Create the Avahi service browser */
1256
1208
sb = avahi_s_service_browser_new(mc.server, if_index,
1257
1209
AVAHI_PROTO_INET6, "_mandos._tcp",
1258
NULL, 0, browse_callback, NULL);
1210
NULL, 0, browse_callback, &mc);
1259
1211
if(sb == NULL){
1260
1212
fprintf(stderr, "Failed to create service browser: %s\n",
1261
1213
avahi_strerror(avahi_server_errno(mc.server)));