/mandos/trunk : revision 360

30

*/

31

32

/* Needed by GPGME, specifically gpgme_data_seek() */

33

#ifndef _LARGEFILE_SOURCE

33

34

#define _LARGEFILE_SOURCE

35

#endif

36

#ifndef _FILE_OFFSET_BITS

34

37

#define _FILE_OFFSET_BITS 64

38

#endif

35

39

36

40

#define _GNU_SOURCE /* TEMP_FAILURE_RETRY(), asprintf() */

37

41

38

42

#include <stdio.h> /* fprintf(), stderr, fwrite(),

39

stdout, ferror(), sscanf(),

40

remove() */

43

stdout, ferror(), remove() */

41

44

#include <stdint.h> /* uint16_t, uint32_t */

42

45

#include <stddef.h> /* NULL, size_t, ssize_t */

43

46

#include <stdlib.h> /* free(), EXIT_SUCCESS, EXIT_FAILURE,

44

srand() */

47

srand(), strtof() */

45

48

#include <stdbool.h> /* bool, false, true */

46

49

#include <string.h> /* memset(), strcmp(), strlen(),

47

50

strerror(), asprintf(), strcpy() */

56

59

#include <fcntl.h> /* open() */

57

60

#include <dirent.h> /* opendir(), struct dirent, readdir()

58

61

*/

59

#include <inttypes.h> /* PRIu16, intmax_t, SCNdMAX */

62

#include <inttypes.h> /* PRIu16, PRIdMAX, intmax_t,

63

strtoimax() */

60

64

#include <assert.h> /* assert() */

61

65

#include <errno.h> /* perror(), errno */

62

66

#include <time.h> /* nanosleep(), time() */

75

79

argp_state, struct argp,

76

80

argp_parse(), ARGP_KEY_ARG,

77

81

ARGP_KEY_END, ARGP_ERR_UNKNOWN */

82

#include <signal.h> /* sigemptyset(), sigaddset(),

83

sigaction(), SIGTERM, sig_atomic_t,

84

raise() */

85

78

86

#ifdef __linux__

79

87

#include <sys/klog.h> /* klogctl() */

80

#endif

88

#endif /* __linux__ */

81

89

82

90

/* Avahi */

83

91

/* All Avahi types, constants and functions

129

137

gpgme_ctx_t ctx;

130

138

} mandos_context;

131

139

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" };

144

132

145

/*

133

* Make room in "buffer" for at least BUFFER_SIZE additional bytes.

134

* "buffer_capacity" is how much is currently allocated,

146

* Make additional room in "buffer" for at least BUFFER_SIZE more

147

* bytes. "buffer_capacity" is how much is currently allocated,

135

148

* "buffer_length" is how much is already used.

136

149

*/

137

size_t adjustbuffer(char **buffer, size_t buffer_length,

150

size_t incbuffer(char **buffer, size_t buffer_length,

138

151

size_t buffer_capacity){

139

152

if(buffer_length + BUFFER_SIZE > buffer_capacity){

140

153

*buffer = realloc(*buffer, buffer_capacity + BUFFER_SIZE);

149

162

/*

150

163

* Initialize GPGME.

151

164

*/

152

static bool init_gpgme(mandos_context *mc, const char *seckey,

165

static bool init_gpgme(const char *seckey,

153

166

const char *pubkey, const char *tempdir){

154

167

int ret;

155

168

gpgme_error_t rc;

176

189

return false;

177

190

}

178

191

179

rc = gpgme_op_import(mc->ctx, pgp_data);

192

rc = gpgme_op_import(mc.ctx, pgp_data);

180

193

if(rc != GPG_ERR_NO_ERROR){

181

194

fprintf(stderr, "bad gpgme_op_import: %s: %s\n",

182

195

gpgme_strsource(rc), gpgme_strerror(rc));

192

205

}

193

206

194

207

if(debug){

195

fprintf(stderr, "Initialize gpgme\n");

208

fprintf(stderr, "Initializing GPGME\n");

196

209

}

197

210

198

211

/* Init GPGME */

225

238

}

226

239

227

240

/* Create new GPGME "context" */

228

rc = gpgme_new(&(mc->ctx));

241

rc = gpgme_new(&(mc.ctx));

229

242

if(rc != GPG_ERR_NO_ERROR){

230

243

fprintf(stderr, "bad gpgme_new: %s: %s\n",

231

244

gpgme_strsource(rc), gpgme_strerror(rc));

236

249

return false;

237

250

}

238

251

239

return true;

252

return true;

240

253

}

241

254

242

255

/*

243

256

* Decrypt OpenPGP data.

244

257

* Returns -1 on error

245

258

*/

246

static ssize_t pgp_packet_decrypt(const mandos_context *mc,

247

const char *cryptotext,

259

static ssize_t pgp_packet_decrypt(const char *cryptotext,

248

260

size_t crypto_size,

249

261

char **plaintext){

250

262

gpgme_data_t dh_crypto, dh_plain;

277

289

278

290

/* Decrypt data from the cryptotext data buffer to the plaintext

279

291

data buffer */

280

rc = gpgme_op_decrypt(mc->ctx, dh_crypto, dh_plain);

292

rc = gpgme_op_decrypt(mc.ctx, dh_crypto, dh_plain);

281

293

if(rc != GPG_ERR_NO_ERROR){

282

294

fprintf(stderr, "bad gpgme_op_decrypt: %s: %s\n",

283

295

gpgme_strsource(rc), gpgme_strerror(rc));

284

296

plaintext_length = -1;

285

297

if(debug){

286

298

gpgme_decrypt_result_t result;

287

result = gpgme_op_decrypt_result(mc->ctx);

299

result = gpgme_op_decrypt_result(mc.ctx);

288

300

if(result == NULL){

289

301

fprintf(stderr, "gpgme_op_decrypt_result failed\n");

290

302

} else {

297

309

}

298

310

gpgme_recipient_t recipient;

299

311

recipient = result->recipients;

300

if(recipient){

301

while(recipient != NULL){

302

fprintf(stderr, "Public key algorithm: %s\n",

303

gpgme_pubkey_algo_name(recipient->pubkey_algo));

304

fprintf(stderr, "Key ID: %s\n", recipient->keyid);

305

fprintf(stderr, "Secret key available: %s\n",

306

recipient->status == GPG_ERR_NO_SECKEY

307

? "No" : "Yes");

308

recipient = recipient->next;

309

}

312

while(recipient != NULL){

313

fprintf(stderr, "Public key algorithm: %s\n",

314

gpgme_pubkey_algo_name(recipient->pubkey_algo));

315

fprintf(stderr, "Key ID: %s\n", recipient->keyid);

316

fprintf(stderr, "Secret key available: %s\n",

317

recipient->status == GPG_ERR_NO_SECKEY

318

? "No" : "Yes");

319

recipient = recipient->next;

310

320

}

311

321

}

312

322

}

326

336

327

337

*plaintext = NULL;

328

338

while(true){

329

plaintext_capacity = adjustbuffer(plaintext,

339

plaintext_capacity = incbuffer(plaintext,

330

340

(size_t)plaintext_length,

331

341

plaintext_capacity);

332

342

if(plaintext_capacity == 0){

333

perror("adjustbuffer");

343

perror("incbuffer");

334

344

plaintext_length = -1;

335

345

goto decrypt_end;

336

346

}

382

392

fprintf(stderr, "GnuTLS: %s", string);

383

393

}

384

394

385

static int init_gnutls_global(mandos_context *mc,

386

const char *pubkeyfilename,

395

static int init_gnutls_global(const char *pubkeyfilename,

387

396

const char *seckeyfilename){

388

397

int ret;

389

398

407

416

}

408

417

409

418

/* OpenPGP credentials */

410

gnutls_certificate_allocate_credentials(&mc->cred);

419

gnutls_certificate_allocate_credentials(&mc.cred);

411

420

if(ret != GNUTLS_E_SUCCESS){

412

421

fprintf(stderr, "GnuTLS memory error: %s\n", /* Spurious warning

413

422

from

425

434

}

426

435

427

436

ret = gnutls_certificate_set_openpgp_key_file

428

(mc->cred, pubkeyfilename, seckeyfilename,

437

(mc.cred, pubkeyfilename, seckeyfilename,

429

438

GNUTLS_OPENPGP_FMT_BASE64);

430

439

if(ret != GNUTLS_E_SUCCESS){

431

440

fprintf(stderr,

437

446

}

438

447

439

448

/* GnuTLS server initialization */

440

ret = gnutls_dh_params_init(&mc->dh_params);

449

ret = gnutls_dh_params_init(&mc.dh_params);

441

450

if(ret != GNUTLS_E_SUCCESS){

442

451

fprintf(stderr, "Error in GnuTLS DH parameter initialization:"

443

452

" %s\n", safer_gnutls_strerror(ret));

444

453

goto globalfail;

445

454

}

446

ret = gnutls_dh_params_generate2(mc->dh_params, mc->dh_bits);

455

ret = gnutls_dh_params_generate2(mc.dh_params, mc.dh_bits);

447

456

if(ret != GNUTLS_E_SUCCESS){

448

457

fprintf(stderr, "Error in GnuTLS prime generation: %s\n",

449

458

safer_gnutls_strerror(ret));

450

459

goto globalfail;

451

460

}

452

461

453

gnutls_certificate_set_dh_params(mc->cred, mc->dh_params);

462

gnutls_certificate_set_dh_params(mc.cred, mc.dh_params);

454

463

455

464

return 0;

456

465

457

466

globalfail:

458

467

459

gnutls_certificate_free_credentials(mc->cred);

468

gnutls_certificate_free_credentials(mc.cred);

460

469

gnutls_global_deinit();

461

gnutls_dh_params_deinit(mc->dh_params);

470

gnutls_dh_params_deinit(mc.dh_params);

462

471

return -1;

463

472

}

464

473

465

static int init_gnutls_session(mandos_context *mc,

466

gnutls_session_t *session){

474

static int init_gnutls_session(gnutls_session_t *session){

467

475

int ret;

468

476

/* GnuTLS session creation */

469

477

ret = gnutls_init(session, GNUTLS_SERVER);

474

482

475

483

{

476

484

const char *err;

477

ret = gnutls_priority_set_direct(*session, mc->priority, &err);

485

ret = gnutls_priority_set_direct(*session, mc.priority, &err);

478

486

if(ret != GNUTLS_E_SUCCESS){

479

487

fprintf(stderr, "Syntax error at: %s\n", err);

480

488

fprintf(stderr, "GnuTLS error: %s\n",

485

493

}

486

494

487

495

ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,

488

mc->cred);

496

mc.cred);

489

497

if(ret != GNUTLS_E_SUCCESS){

490

498

fprintf(stderr, "Error setting GnuTLS credentials: %s\n",

491

499

safer_gnutls_strerror(ret));

497

505

gnutls_certificate_server_set_request(*session,

498

506

GNUTLS_CERT_IGNORE);

499

507

500

gnutls_dh_set_prime_bits(*session, mc->dh_bits);

508

gnutls_dh_set_prime_bits(*session, mc.dh_bits);

501

509

502

510

return 0;

503

511

}

506

514

static void empty_log(__attribute__((unused)) AvahiLogLevel level,

507

515

__attribute__((unused)) const char *txt){}

508

516

517

sig_atomic_t quit_now = 0;

518

int signal_received = 0;

519

509

520

/* Called when a Mandos server is found */

510

521

static int start_mandos_communication(const char *ip, uint16_t port,

511

522

AvahiIfIndex if_index,

512

mandos_context *mc, int af){

513

int ret, tcp_sd;

523

int af){

524

int ret, tcp_sd = -1;

514

525

ssize_t sret;

515

526

union {

516

527

struct sockaddr_in in;

526

537

gnutls_session_t session;

527

538

int pf; /* Protocol family */

528

539

540

if(quit_now){

541

return -1;

542

}

543

529

544

switch(af){

530

545

case AF_INET6:

531

546

pf = PF_INET6;

538

553

return -1;

539

554

}

540

555

541

ret = init_gnutls_session(mc, &session);

556

ret = init_gnutls_session(&session);

542

557

if(ret != 0){

543

558

return -1;

544

559

}

551

566

tcp_sd = socket(pf, SOCK_STREAM, 0);

552

567

if(tcp_sd < 0){

553

568

perror("socket");

554

return -1;

569

retval = -1;

570

goto mandos_end;

571

}

572

573

if(quit_now){

574

goto mandos_end;

555

575

}

556

576

557

577

memset(&to, 0, sizeof(to));

558

578

if(af == AF_INET6){

559

to.in6.sin6_family = (uint16_t)af;

579

to.in6.sin6_family = (sa_family_t)af;

560

580

ret = inet_pton(af, ip, &to.in6.sin6_addr);

561

581

} else { /* IPv4 */

562

582

to.in.sin_family = (sa_family_t)af;

564

584

}

565

585

if(ret < 0 ){

566

586

perror("inet_pton");

567

return -1;

587

retval = -1;

588

goto mandos_end;

568

589

}

569

590

if(ret == 0){

570

591

fprintf(stderr, "Bad address: %s\n", ip);

571

return -1;

592

retval = -1;

593

goto mandos_end;

572

594

}

573

595

if(af == AF_INET6){

574

596

to.in6.sin6_port = htons(port); /* Spurious warnings from

581

603

if(if_index == AVAHI_IF_UNSPEC){

582

604

fprintf(stderr, "An IPv6 link-local address is incomplete"

583

605

" without a network interface\n");

584

return -1;

606

retval = -1;

607

goto mandos_end;

585

608

}

586

609

/* Set the network interface number as scope */

587

610

to.in6.sin6_scope_id = (uint32_t)if_index;

592

615

-Wunreachable-code */

593

616

}

594

617

618

if(quit_now){

619

goto mandos_end;

620

}

621

595

622

if(debug){

596

623

if(af == AF_INET6 and if_index != AVAHI_IF_UNSPEC){

597

624

char interface[IF_NAMESIZE];

624

651

}

625

652

}

626

653

654

if(quit_now){

655

goto mandos_end;

656

}

657

627

658

if(af == AF_INET6){

628

659

ret = connect(tcp_sd, &to.in6, sizeof(to));

629

660

} else {

631

662

}

632

663

if(ret < 0){

633

664

perror("connect");

634

return -1;

665

retval = -1;

666

goto mandos_end;

667

}

668

669

if(quit_now){

670

goto mandos_end;

635

671

}

636

672

637

673

const char *out = mandos_protocol_version;

656

692

break;

657

693

}

658

694

}

695

696

if(quit_now){

697

goto mandos_end;

698

}

659

699

}

660

700

661

701

if(debug){

662

702

fprintf(stderr, "Establishing TLS session with %s\n", ip);

663

703

}

664

704

705

if(quit_now){

706

goto mandos_end;

707

}

708

665

709

gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);

666

710

711

if(quit_now){

712

goto mandos_end;

713

}

714

667

715

do{

668

716

ret = gnutls_handshake(session);

717

if(quit_now){

718

goto mandos_end;

719

}

669

720

} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);

670

721

671

722

if(ret != GNUTLS_E_SUCCESS){

685

736

}

686

737

687

738

while(true){

688

buffer_capacity = adjustbuffer(&buffer, buffer_length,

739

740

if(quit_now){

741

goto mandos_end;

742

}

743

744

buffer_capacity = incbuffer(&buffer, buffer_length,

689

745

buffer_capacity);

690

746

if(buffer_capacity == 0){

691

perror("adjustbuffer");

747

perror("incbuffer");

692

748

retval = -1;

693

749

goto mandos_end;

694

750

}

695

751

752

if(quit_now){

753

goto mandos_end;

754

}

755

696

756

sret = gnutls_record_recv(session, buffer+buffer_length,

697

757

BUFFER_SIZE);

698

758

if(sret == 0){

706

766

case GNUTLS_E_REHANDSHAKE:

707

767

do{

708

768

ret = gnutls_handshake(session);

769

770

if(quit_now){

771

goto mandos_end;

772

}

709

773

} while(ret == GNUTLS_E_AGAIN or ret == GNUTLS_E_INTERRUPTED);

710

774

if(ret < 0){

711

775

fprintf(stderr, "*** GnuTLS Re-handshake failed ***\n");

730

794

fprintf(stderr, "Closing TLS session\n");

731

795

}

732

796

797

if(quit_now){

798

goto mandos_end;

799

}

800

733

801

gnutls_bye(session, GNUTLS_SHUT_RDWR);

734

802

803

if(quit_now){

804

goto mandos_end;

805

}

806

735

807

if(buffer_length > 0){

736

decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,

808

decrypted_buffer_size = pgp_packet_decrypt(buffer,

737

809

buffer_length,

738

810

&decrypted_buffer);

739

811

if(decrypted_buffer_size >= 0){

740

812

written = 0;

741

813

while(written < (size_t) decrypted_buffer_size){

814

if(quit_now){

815

goto mandos_end;

816

}

817

742

818

ret = (int)fwrite(decrypted_buffer + written, 1,

743

819

(size_t)decrypted_buffer_size - written,

744

820

stdout);

764

840

765

841

mandos_end:

766

842

free(buffer);

767

ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));

843

if(tcp_sd >= 0){

844

ret = (int)TEMP_FAILURE_RETRY(close(tcp_sd));

845

}

768

846

if(ret == -1){

769

847

perror("close");

770

848

}

771

849

gnutls_deinit(session);

850

if(quit_now){

851

retval = -1;

852

}

772

853

return retval;

773

854

}

774

855

785

866

AVAHI_GCC_UNUSED AvahiStringList *txt,

786

867

AVAHI_GCC_UNUSED AvahiLookupResultFlags

787

868

flags,

788

void* userdata){

789

mandos_context *mc = userdata;

869

AVAHI_GCC_UNUSED void* userdata){

790

870

assert(r);

791

871

792

872

/* Called whenever a service has been resolved successfully or

793

873

timed out */

794

874

875

if(quit_now){

876

return;

877

}

878

795

879

switch(event){

796

880

default:

797

881

case AVAHI_RESOLVER_FAILURE:

798

882

fprintf(stderr, "(Avahi Resolver) Failed to resolve service '%s'"

799

883

" of type '%s' in domain '%s': %s\n", name, type, domain,

800

avahi_strerror(avahi_server_errno(mc->server)));

884

avahi_strerror(avahi_server_errno(mc.server)));

801

885

break;

802

886

803

887

case AVAHI_RESOLVER_FOUND:

809

893

PRIdMAX ") on port %" PRIu16 "\n", name, host_name,

810

894

ip, (intmax_t)interface, port);

811

895

}

812

int ret = start_mandos_communication(ip, port, interface, mc,

896

int ret = start_mandos_communication(ip, port, interface,

813

897

avahi_proto_to_af(proto));

814

898

if(ret == 0){

815

avahi_simple_poll_quit(mc->simple_poll);

899

avahi_simple_poll_quit(mc.simple_poll);

816

900

}

817

901

}

818

902

}

828

912

const char *domain,

829

913

AVAHI_GCC_UNUSED AvahiLookupResultFlags

830

914

flags,

831

void* userdata){

832

mandos_context *mc = userdata;

915

AVAHI_GCC_UNUSED void* userdata){

833

916

assert(b);

834

917

835

918

/* Called whenever a new services becomes available on the LAN or

836

919

is removed from the LAN */

837

920

921

if(quit_now){

922

return;

923

}

924

838

925

switch(event){

839

926

default:

840

927

case AVAHI_BROWSER_FAILURE:

841

928

842

929

fprintf(stderr, "(Avahi browser) %s\n",

843

avahi_strerror(avahi_server_errno(mc->server)));

844

avahi_simple_poll_quit(mc->simple_poll);

930

avahi_strerror(avahi_server_errno(mc.server)));

931

avahi_simple_poll_quit(mc.simple_poll);

845

932

return;

846

933

847

934

case AVAHI_BROWSER_NEW:

850

937

the callback function is called the Avahi server will free the

851

938

resolver for us. */

852

939

853

if(!(avahi_s_service_resolver_new(mc->server, interface,

854

protocol, name, type, domain,

855

AVAHI_PROTO_INET6, 0,

856

resolve_callback, mc)))

940

if(avahi_s_service_resolver_new(mc.server, interface, protocol,

941

name, type, domain, protocol, 0,

942

resolve_callback, NULL) == NULL)

857

943

fprintf(stderr, "Avahi: Failed to resolve service '%s': %s\n",

858

name, avahi_strerror(avahi_server_errno(mc->server)));

944

name, avahi_strerror(avahi_server_errno(mc.server)));

859

945

break;

860

946

861

947

case AVAHI_BROWSER_REMOVE:

870

956

}

871

957

}

872

958

959

/* stop main loop after sigterm has been called */

960

static void handle_sigterm(int sig){

961

if(quit_now){

962

return;

963

}

964

quit_now = 1;

965

signal_received = sig;

966

int old_errno = errno;

967

if(mc.simple_poll != NULL){

968

avahi_simple_poll_quit(mc.simple_poll);

969

}

970

errno = old_errno;

971

}

972

873

973

int main(int argc, char *argv[]){

874

974

AvahiSServiceBrowser *sb = NULL;

875

975

int error;

876

976

int ret;

877

977

intmax_t tmpmax;

878

int numchars;

978

char *tmp;

879

979

int exitcode = EXIT_SUCCESS;

880

980

const char *interface = "eth0";

881

981

struct ifreq network;

882

int sd;

982

int sd = -1;

983

bool take_down_interface = false;

883

984

uid_t uid;

884

985

gid_t gid;

885

986

char *connect_to = NULL;

889

990

const char *seckey = PATHDIR "/" SECKEY;

890

991

const char *pubkey = PATHDIR "/" PUBKEY;

891

992

892

mandos_context mc = { .simple_poll = NULL, .server = NULL,

893

.dh_bits = 1024, .priority = "SECURE256"

894

":!CTYPE-X.509:+CTYPE-OPENPGP" };

895

993

bool gnutls_initialized = false;

896

994

bool gpgme_initialized = false;

897

double delay = 2.5;

995

float delay = 2.5f;

996

997

struct sigaction old_sigterm_action;

998

struct sigaction sigterm_action = { .sa_handler = handle_sigterm };

898

999

899

1000

{

900

1001

struct argp_option options[] = {

952

1053

pubkey = arg;

953

1054

break;

954

1055

case 129: /* --dh-bits */

955

ret = sscanf(arg, "%" SCNdMAX "%n", &tmpmax, &numchars);

956

if(ret < 1 or tmpmax != (typeof(mc.dh_bits))tmpmax

957

or arg[numchars] != '\0'){

1056

errno = 0;

1057

tmpmax = strtoimax(arg, &tmp, 10);

1058

if(errno != 0 or tmp == arg or *tmp != '\0'

1059

or tmpmax != (typeof(mc.dh_bits))tmpmax){

958

1060

fprintf(stderr, "Bad number of DH bits\n");

959

1061

exit(EXIT_FAILURE);

960

1062

}

964

1066

mc.priority = arg;

965

1067

break;

966

1068

case 131: /* --delay */

967

ret = sscanf(arg, "%lf%n", &delay, &numchars);

968

if(ret < 1 or arg[numchars] != '\0'){

1069

errno = 0;

1070

delay = strtof(arg, &tmp);

1071

if(errno != 0 or tmp == arg or *tmp != '\0'){

969

1072

fprintf(stderr, "Bad delay\n");

970

1073

exit(EXIT_FAILURE);

971

1074

}

992

1095

}

993

1096

}

994

1097

1098

if(not debug){

1099

avahi_set_log_function(empty_log);

1100

}

1101

1102

/* Initialize Avahi early so avahi_simple_poll_quit() can be called

1103

from the signal handler */

1104

/* Initialize the pseudo-RNG for Avahi */

1105

srand((unsigned int) time(NULL));

1106

mc.simple_poll = avahi_simple_poll_new();

1107

if(mc.simple_poll == NULL){

1108

fprintf(stderr, "Avahi: Failed to create simple poll object.\n");

1109

exitcode = EXIT_FAILURE;

1110

goto end;

1111

}

1112

1113

sigemptyset(&sigterm_action.sa_mask);

1114

ret = sigaddset(&sigterm_action.sa_mask, SIGINT);

1115

if(ret == -1){

1116

perror("sigaddset");

1117

exitcode = EXIT_FAILURE;

1118

goto end;

1119

}

1120

ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);

1121

if(ret == -1){

1122

perror("sigaddset");

1123

exitcode = EXIT_FAILURE;

1124

goto end;

1125

}

1126

ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);

1127

if(ret == -1){

1128

perror("sigaddset");

1129

exitcode = EXIT_FAILURE;

1130

goto end;

1131

}

1132

/* Need to check if the handler is SIG_IGN before handling:

1133

| [[info:libc:Initial Signal Actions]] |

1134

| [[info:libc:Basic Signal Handling]] |

1135

*/

1136

ret = sigaction(SIGINT, NULL, &old_sigterm_action);

1137

if(ret == -1){

1138

perror("sigaction");

1139

return EXIT_FAILURE;

1140

}

1141

if(old_sigterm_action.sa_handler != SIG_IGN){

1142

ret = sigaction(SIGINT, &sigterm_action, NULL);

1143

if(ret == -1){

1144

perror("sigaction");

1145

exitcode = EXIT_FAILURE;

1146

goto end;

1147

}

1148

}

1149

ret = sigaction(SIGHUP, NULL, &old_sigterm_action);

1150

if(ret == -1){

1151

perror("sigaction");

1152

return EXIT_FAILURE;

1153

}

1154

if(old_sigterm_action.sa_handler != SIG_IGN){

1155

ret = sigaction(SIGHUP, &sigterm_action, NULL);

1156

if(ret == -1){

1157

perror("sigaction");

1158

exitcode = EXIT_FAILURE;

1159

goto end;

1160

}

1161

}

1162

ret = sigaction(SIGTERM, NULL, &old_sigterm_action);

1163

if(ret == -1){

1164

perror("sigaction");

1165

return EXIT_FAILURE;

1166

}

1167

if(old_sigterm_action.sa_handler != SIG_IGN){

1168

ret = sigaction(SIGTERM, &sigterm_action, NULL);

1169

if(ret == -1){

1170

perror("sigaction");

1171

exitcode = EXIT_FAILURE;

1172

goto end;

1173

}

1174

}

1175

995

1176

/* If the interface is down, bring it up */

996

1177

if(interface[0] != '\0'){

1178

if_index = (AvahiIfIndex) if_nametoindex(interface);

1179

if(if_index == 0){

1180

fprintf(stderr, "No such interface: \"%s\"\n", interface);

1181

exitcode = EXIT_FAILURE;

1182

goto end;

1183

}

1184

1185

if(quit_now){

1186

goto end;

1187

}

1188

997

1189

#ifdef __linux__

998

1190

/* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO

999

1191

messages to mess up the prompt */

1003

1195

restore_loglevel = false;

1004

1196

perror("klogctl");

1005

1197

}

1006

#endif

1198

#endif /* __linux__ */

1007

1199

1008

1200

sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);

1009

1201

if(sd < 0){

1016

1208

perror("klogctl");

1017

1209

}

1018

1210

}

1019

#endif

1211

#endif /* __linux__ */

1020

1212

goto end;

1021

1213

}

1022

1214

strcpy(network.ifr_name, interface);

1030

1222

perror("klogctl");

1031

1223

}

1032

1224

}

1033

#endif

1225

#endif /* __linux__ */

1034

1226

exitcode = EXIT_FAILURE;

1035

1227

goto end;

1036

1228

}

1037

1229

if((network.ifr_flags & IFF_UP) == 0){

1038

1230

network.ifr_flags |= IFF_UP;

1231

take_down_interface = true;

1039

1232

ret = ioctl(sd, SIOCSIFFLAGS, &network);

1040

1233

if(ret == -1){

1234

take_down_interface = false;

1041

1235

perror("ioctl SIOCSIFFLAGS");

1042

1236

exitcode = EXIT_FAILURE;

1043

1237

#ifdef __linux__

1047

1241

perror("klogctl");

1048

1242

}

1049

1243

}

1050

#endif

1244

#endif /* __linux__ */

1051

1245

goto end;

1052

1246

}

1053

1247

}

1065

1259

perror("nanosleep");

1066

1260

}

1067

1261

}

1068

ret = (int)TEMP_FAILURE_RETRY(close(sd));

1069

if(ret == -1){

1070

perror("close");

1262

if(not take_down_interface){

1263

/* We won't need the socket anymore */

1264

ret = (int)TEMP_FAILURE_RETRY(close(sd));

1265

if(ret == -1){

1266

perror("close");

1267

}

1071

1268

}

1072

1269

#ifdef __linux__

1073

1270

if(restore_loglevel){

1077

1274

perror("klogctl");

1078

1275

}

1079

1276

}

1080

#endif

1277

#endif /* __linux__ */

1278

}

1279

1280

if(quit_now){

1281

goto end;

1081

1282

}

1082

1283

1083

1284

uid = getuid();

1094

1295

perror("setuid");

1095

1296

}

1096

1297

1097

ret = init_gnutls_global(&mc, pubkey, seckey);

1298

if(quit_now){

1299

goto end;

1300

}

1301

1302

ret = init_gnutls_global(pubkey, seckey);

1098

1303

if(ret == -1){

1099

1304

fprintf(stderr, "init_gnutls_global failed\n");

1100

1305

exitcode = EXIT_FAILURE;

1103

1308

gnutls_initialized = true;

1104

1309

}

1105

1310

1311

if(quit_now){

1312

goto end;

1313

}

1314

1315

tempdir_created = true;

1106

1316

if(mkdtemp(tempdir) == NULL){

1317

tempdir_created = false;

1107

1318

perror("mkdtemp");

1108

1319

goto end;

1109

1320

}

1110

tempdir_created = true;

1111

1112

if(not init_gpgme(&mc, pubkey, seckey, tempdir)){

1321

1322

if(quit_now){

1323

goto end;

1324

}

1325

1326

if(not init_gpgme(pubkey, seckey, tempdir)){

1113

1327

fprintf(stderr, "init_gpgme failed\n");

1114

1328

exitcode = EXIT_FAILURE;

1115

1329

goto end;

1117

1331

gpgme_initialized = true;

1118

1332

}

1119

1333

1120

if(interface[0] != '\0'){

1121

if_index = (AvahiIfIndex) if_nametoindex(interface);

1122

if(if_index == 0){

1123

fprintf(stderr, "No such interface: \"%s\"\n", interface);

1124

exitcode = EXIT_FAILURE;

1125

goto end;

1126

}

1334

if(quit_now){

1335

goto end;

1127

1336

}

1128

1337

1129

1338

if(connect_to != NULL){

1135

1344

exitcode = EXIT_FAILURE;

1136

1345

goto end;

1137

1346

}

1347

1348

if(quit_now){

1349

goto end;

1350

}

1351

1138

1352

uint16_t port;

1139

ret = sscanf(address+1, "%" SCNdMAX "%n", &tmpmax, &numchars);

1140

if(ret < 1 or tmpmax != (uint16_t)tmpmax

1141

or address[numchars+1] != '\0'){

1353

errno = 0;

1354

tmpmax = strtoimax(address+1, &tmp, 10);

1355

if(errno != 0 or tmp == address+1 or *tmp != '\0'

1356

or tmpmax != (uint16_t)tmpmax){

1142

1357

fprintf(stderr, "Bad port number\n");

1143

1358

exitcode = EXIT_FAILURE;

1144

1359

goto end;

1145

1360

}

1361

1362

if(quit_now){

1363

goto end;

1364

}

1365

1146

1366

port = (uint16_t)tmpmax;

1147

1367

*address = '\0';

1148

1368

address = connect_to;

1153

1373

} else {

1154

1374

af = AF_INET;

1155

1375

}

1156

ret = start_mandos_communication(address, port, if_index, &mc,

1157

af);

1376

1377

if(quit_now){

1378

goto end;

1379

}

1380

1381

ret = start_mandos_communication(address, port, if_index, af);

1158

1382

if(ret < 0){

1159

1383

exitcode = EXIT_FAILURE;

1160

1384

} else {

1163

1387

goto end;

1164

1388

}

1165

1389

1166

if(not debug){

1167

avahi_set_log_function(empty_log);

1168

}

1169

1170

/* Initialize the pseudo-RNG for Avahi */

1171

srand((unsigned int) time(NULL));

1172

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;

1390

if(quit_now){

1178

1391

goto end;

1179

1392

}

1180

1393

1204

1417

goto end;

1205

1418

}

1206

1419

1420

if(quit_now){

1421

goto end;

1422

}

1423

1207

1424

/* Create the Avahi service browser */

1208

1425

sb = avahi_s_service_browser_new(mc.server, if_index,

1209

AVAHI_PROTO_INET6, "_mandos._tcp",

1210

NULL, 0, browse_callback, &mc);

1426

AVAHI_PROTO_UNSPEC, "_mandos._tcp",

1427

NULL, 0, browse_callback, NULL);

1211

1428

if(sb == NULL){

1212

1429

fprintf(stderr, "Failed to create service browser: %s\n",

1213

1430

avahi_strerror(avahi_server_errno(mc.server)));

1215

1432

goto end;

1216

1433

}

1217

1434

1435

if(quit_now){

1436

goto end;

1437

}

1438

1218

1439

/* Run the main loop */

1219

1440

1220

1441

if(debug){

1249

1470

gpgme_release(mc.ctx);

1250

1471

}

1251

1472

1473

/* Take down the network interface */

1474

if(take_down_interface){

1475

ret = ioctl(sd, SIOCGIFFLAGS, &network);

1476

if(ret == -1){

1477

perror("ioctl SIOCGIFFLAGS");

1478

} else if(network.ifr_flags & IFF_UP) {

1479

network.ifr_flags &= ~IFF_UP; /* clear flag */

1480

ret = ioctl(sd, SIOCSIFFLAGS, &network);

1481

if(ret == -1){

1482

perror("ioctl SIOCSIFFLAGS");

1483

}

1484

}

1485

ret = (int)TEMP_FAILURE_RETRY(close(sd));

1486

if(ret == -1){

1487

perror("close");

1488

}

1489

}

1490

1252

1491

/* Removes the temp directory used by GPGME */

1253

1492

if(tempdir_created){

1254

1493

DIR *d;

1293

1532

}

1294

1533

}

1295

1534

1535

if(quit_now){

1536

sigemptyset(&old_sigterm_action.sa_mask);

1537

old_sigterm_action.sa_handler = SIG_DFL;

1538

ret = sigaction(signal_received, &old_sigterm_action, NULL);

1539

if(ret == -1){

1540

perror("sigaction");

1541

}

1542

raise(signal_received);

1543

}

1544

1296

1545

return exitcode;

1297

1546

}