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*/
140
* Make additional room in "buffer" for at least BUFFER_SIZE more
141
* bytes. "buffer_capacity" is how much is currently allocated,
133
* Make room in "buffer" for at least BUFFER_SIZE additional bytes.
134
* "buffer_capacity" is how much is currently allocated,
142
135
* "buffer_length" is how much is already used.
144
size_t incbuffer(char **buffer, size_t buffer_length,
137
size_t adjustbuffer(char **buffer, size_t buffer_length,
145
138
size_t buffer_capacity){
146
139
if(buffer_length + BUFFER_SIZE > buffer_capacity){
147
140
*buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);
157
150
* Initialize GPGME.
159
static bool init_gpgme(const char *seckey,
152
static bool init_gpgme(mandos_context *mc, const char *seckey,
160
153
const char *pubkey, const char *tempdir){
162
155
gpgme_error_t rc;
186
rc = gpgme_op_import(mc.ctx, pgp_data);
179
rc = gpgme_op_import(mc->ctx, pgp_data);
187
180
if(rc != GPG_ERR_NO_ERROR){
188
181
fprintf(stderr, "bad gpgme_op_import: %s: %s\n",
189
182
gpgme_strsource(rc), gpgme_strerror(rc));
234
227
/* Create new GPGME "context" */
235
rc = gpgme_new(&(mc.ctx));
228
rc = gpgme_new(&(mc->ctx));
236
229
if(rc != GPG_ERR_NO_ERROR){
237
230
fprintf(stderr, "bad gpgme_new: %s: %s\n",
238
231
gpgme_strsource(rc), gpgme_strerror(rc));
250
243
* Decrypt OpenPGP data.
251
244
* Returns -1 on error
253
static ssize_t pgp_packet_decrypt(const char *cryptotext,
246
static ssize_t pgp_packet_decrypt(const mandos_context *mc,
247
const char *cryptotext,
254
248
size_t crypto_size,
255
249
char **plaintext){
256
250
gpgme_data_t dh_crypto, dh_plain;
284
278
/* Decrypt data from the cryptotext data buffer to the plaintext
286
rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);
280
rc = gpgme_op_decrypt(mc->ctx, dh_crypto, dh_plain);
287
281
if(rc != GPG_ERR_NO_ERROR){
288
282
fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",
289
283
gpgme_strsource(rc), gpgme_strerror(rc));
290
284
plaintext_length = -1;
292
286
gpgme_decrypt_result_t result;
293
result = gpgme_op_decrypt_result(mc.ctx);
287
result = gpgme_op_decrypt_result(mc->ctx);
294
288
if(result == NULL){
295
289
fprintf(stderr, "gpgme_op_decrypt_result failed\n");
333
327
*plaintext = NULL;
335
plaintext_capacity = incbuffer(plaintext,
329
plaintext_capacity = adjustbuffer(plaintext,
336
330
(size_t)plaintext_length,
337
331
plaintext_capacity);
338
332
if(plaintext_capacity == 0){
333
perror("adjustbuffer");
340
334
plaintext_length = -1;
341
335
goto decrypt_end;
388
382
fprintf(stderr, "GnuTLS: %s", string);
391
static int init_gnutls_global(const char *pubkeyfilename,
385
static int init_gnutls_global(mandos_context *mc,
386
const char *pubkeyfilename,
392
387
const char *seckeyfilename){
414
409
/* OpenPGP credentials */
415
gnutls_certificate_allocate_credentials(&mc.cred);
410
gnutls_certificate_allocate_credentials(&mc->cred);
416
411
if(ret != GNUTLS_E_SUCCESS){
417
412
fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning
432
427
ret = gnutls_certificate_set_openpgp_key_file
433
(mc.cred, pubkeyfilename, seckeyfilename,
428
(mc->cred, pubkeyfilename, seckeyfilename,
434
429
GNUTLS_OPENPGP_FMT_BASE64);
435
430
if(ret != GNUTLS_E_SUCCESS){
444
439
/* GnuTLS server initialization */
445
ret = gnutls_dh_params_init(&mc.dh_params);
440
ret = gnutls_dh_params_init(&mc->dh_params);
446
441
if(ret != GNUTLS_E_SUCCESS){
447
442
fprintf(stderr, "Error in GnuTLS DH parameter initialization:"
448
443
" %s\n", safer_gnutls_strerror(ret));
451
ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);
446
ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);
452
447
if(ret != GNUTLS_E_SUCCESS){
453
448
fprintf(stderr, "Error in GnuTLS prime generation: %s\n",
454
449
safer_gnutls_strerror(ret));
458
gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);
453
gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);
464
gnutls_certificate_free_credentials(mc.cred);
459
gnutls_certificate_free_credentials(mc->cred);
465
460
gnutls_global_deinit();
466
gnutls_dh_params_deinit(mc.dh_params);
461
gnutls_dh_params_deinit(mc->dh_params);
470
static int init_gnutls_session(gnutls_session_t *session){
465
static int init_gnutls_session(mandos_context *mc,
466
gnutls_session_t *session){
472
468
/* GnuTLS session creation */
473
469
ret = gnutls_init(session, GNUTLS_SERVER);
481
ret = gnutls_priority_set_direct(*session, mc.priority, &err);
477
ret = gnutls_priority_set_direct(*session, mc->priority, &err);
482
478
if(ret != GNUTLS_E_SUCCESS){
483
479
fprintf(stderr, "Syntax error at: %s\n", err);
484
480
fprintf(stderr, "GnuTLS error: %s\n",
491
487
ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,
493
489
if(ret != GNUTLS_E_SUCCESS){
494
490
fprintf(stderr, "Error setting GnuTLS credentials: %s\n",
495
491
safer_gnutls_strerror(ret));
501
497
gnutls_certificate_server_set_request(*session,
502
498
GNUTLS_CERT_IGNORE);
504
gnutls_dh_set_prime_bits(*session, mc.dh_bits);
500
gnutls_dh_set_prime_bits(*session, mc->dh_bits);
513
509
/* Called when a Mandos server is found */
514
510
static int start_mandos_communication(const char *ip, uint16_t port,
515
511
AvahiIfIndex if_index,
512
mandos_context *mc, int af){
692
buffer_capacity = incbuffer(&buffer, buffer_length,
688
buffer_capacity = adjustbuffer(&buffer, buffer_length,
693
689
buffer_capacity);
694
690
if(buffer_capacity == 0){
691
perror("adjustbuffer");
737
733
gnutls_bye(session, GNUTLS_SHUT_RDWR);
739
735
if(buffer_length > 0){
740
decrypted_buffer_size = pgp_packet_decrypt(buffer,
736
decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,
742
738
&decrypted_buffer);
743
739
if(decrypted_buffer_size >= 0){
789
785
AVAHI_GCC_UNUSED AvahiStringList *txt,
790
786
AVAHI_GCC_UNUSED AvahiLookupResultFlags
792
AVAHI_GCC_UNUSED void* userdata){
789
mandos_context *mc = userdata;
795
792
/* Called whenever a service has been resolved successfully or
800
797
case AVAHI_RESOLVER_FAILURE:
801
798
fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"
802
799
" of type '%s' in domain '%s': %s\n", name, type, domain,
803
avahi_strerror(avahi_server_errno(mc.server)));
800
avahi_strerror(avahi_server_errno(mc->server)));
806
803
case AVAHI_RESOLVER_FOUND:
812
809
PRIdMAX ") on port %" PRIu16 "\n", name, host_name,
813
810
ip, (intmax_t)interface, port);
815
int ret = start_mandos_communication(ip, port, interface,
812
int ret = start_mandos_communication(ip, port, interface, mc,
816
813
avahi_proto_to_af(proto));
818
avahi_simple_poll_quit(mc.simple_poll);
815
avahi_simple_poll_quit(mc->simple_poll);
842
840
case AVAHI_BROWSER_FAILURE:
844
842
fprintf(stderr, "(Avahi browser) %s\n",
845
avahi_strerror(avahi_server_errno(mc.server)));
846
avahi_simple_poll_quit(mc.simple_poll);
843
avahi_strerror(avahi_server_errno(mc->server)));
844
avahi_simple_poll_quit(mc->simple_poll);
849
847
case AVAHI_BROWSER_NEW:
852
850
the callback function is called the Avahi server will free the
853
851
resolver for us. */
855
if(!(avahi_s_service_resolver_new(mc.server, interface,
853
if(!(avahi_s_service_resolver_new(mc->server, interface,
856
854
protocol, name, type, domain,
857
855
AVAHI_PROTO_INET6, 0,
858
resolve_callback, NULL)))
856
resolve_callback, mc)))
859
857
fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",
860
name, avahi_strerror(avahi_server_errno(mc.server)));
858
name, avahi_strerror(avahi_server_errno(mc->server)));
863
861
case AVAHI_BROWSER_REMOVE:
875
sig_atomic_t quit_now = 0;
877
static void handle_sigterm(__attribute__((unused)) int sig){
882
int old_errno = errno;
883
if(mc.simple_poll != NULL){
884
avahi_simple_poll_quit(mc.simple_poll);
889
873
int main(int argc, char *argv[]){
890
874
AvahiSServiceBrowser *sb = NULL;
905
889
const char *seckey = PATHDIR "/" SECKEY;
906
890
const char *pubkey = PATHDIR "/" PUBKEY;
908
/* Initialize Mandos context */
909
mc = (mandos_context){ .simple_poll = NULL, .server = NULL,
910
.dh_bits = 1024, .priority = "SECURE256"
911
":!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" };
912
895
bool gnutls_initialized = false;
913
896
bool gpgme_initialized = false;
914
897
double delay = 2.5;
916
struct sigaction old_sigterm_action;
917
struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
920
900
struct argp_option options[] = {
1114
1094
perror("setuid");
1117
ret = init_gnutls_global(pubkey, seckey);
1097
ret = init_gnutls_global(&mc, pubkey, seckey);
1119
1099
fprintf(stderr, "init_gnutls_global failed\n");
1120
1100
exitcode = EXIT_FAILURE;
1130
1110
tempdir_created = true;
1132
if(not init_gpgme(pubkey, seckey, tempdir)){
1112
if(not init_gpgme(&mc, pubkey, seckey, tempdir)){
1133
1113
fprintf(stderr, "init_gpgme failed\n");
1134
1114
exitcode = EXIT_FAILURE;
1176
ret = start_mandos_communication(address, port, if_index, af);
1156
ret = start_mandos_communication(address, port, if_index, &mc,
1178
1159
exitcode = EXIT_FAILURE;
1226
1207
/* Create the Avahi service browser */
1227
1208
sb = avahi_s_service_browser_new(mc.server, if_index,
1228
1209
AVAHI_PROTO_INET6, "_mandos._tcp",
1229
NULL, 0, browse_callback, NULL);
1210
NULL, 0, browse_callback, &mc);
1230
1211
if(sb == NULL){
1231
1212
fprintf(stderr, "Failed to create service browser: %s\n",
1232
1213
avahi_strerror(avahi_server_errno(mc.server)));
1237
sigemptyset(&sigterm_action.sa_mask);
1238
ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
1240
perror("sigaddset");
1241
exitcode = EXIT_FAILURE;
1244
ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
1246
perror("sigaddset");
1247
exitcode = EXIT_FAILURE;
1250
ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
1252
perror("sigaddset");
1253
exitcode = EXIT_FAILURE;
1256
ret = sigaction(SIGTERM, &sigterm_action, &old_sigterm_action);
1258
perror("sigaction");
1259
exitcode = EXIT_FAILURE;
1263
1218
/* Run the main loop */