/mandos/trunk : revision 355

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() */

45

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

47

srand(), strtof() */

48

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

46

49

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

47

50

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

48

#include <sys/ioctl.h> /* ioctl */

51

#include <sys/ioctl.h> /* ioctl */

49

52

#include <sys/types.h> /* socket(), inet_pton(), sockaddr,

50

53

sockaddr_in6, PF_INET6,

51

SOCK_STREAM, INET6_ADDRSTRLEN,

52

uid_t, gid_t, open(), opendir(),

53

DIR */

54

SOCK_STREAM, uid_t, gid_t, open(),

55

opendir(), DIR */

54

56

#include <sys/stat.h> /* open() */

55

57

#include <sys/socket.h> /* socket(), struct sockaddr_in6,

56

struct in6_addr, inet_pton(),

57

connect() */

58

inet_pton(), connect() */

58

59

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

59

60

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

60

61

*/

61

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

62

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

63

strtoimax() */

62

64

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

63

65

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

64

66

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

65

67

#include <net/if.h> /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,

66

68

SIOCSIFFLAGS, if_indextoname(),

67

69

if_nametoindex(), IF_NAMESIZE */

68

#include <netinet/in.h>

70

#include <netinet/in.h> /* IN6_IS_ADDR_LINKLOCAL,

71

INET_ADDRSTRLEN, INET6_ADDRSTRLEN

72

*/

69

73

#include <unistd.h> /* close(), SEEK_SET, off_t, write(),

70

74

getuid(), getgid(), setuid(),

71

75

setgid() */

72

76

#include <arpa/inet.h> /* inet_pton(), htons */

73

#include <iso646.h> /* not, and, or */

77

#include <iso646.h> /* not, or, and */

74

78

#include <argp.h> /* struct argp_option, error_t, struct

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

* from

414

* -Wunreachable-code

415

*/

422

from

423

-Wunreachable-code

424

*/

416

425

safer_gnutls_strerror(ret));

417

426

gnutls_global_deinit();

418

427

return -1;

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){

523

int af){

513

524

int ret, tcp_sd;

514

525

ssize_t sret;

515

union { struct sockaddr in; struct sockaddr_in6 in6; } to;

526

union {

527

struct sockaddr_in in;

528

struct sockaddr_in6 in6;

529

} to;

516

530

char *buffer = NULL;

517

531

char *decrypted_buffer;

518

532

size_t buffer_length = 0;

520

534

ssize_t decrypted_buffer_size;

521

535

size_t written;

522

536

int retval = 0;

523

char interface[IF_NAMESIZE];

524

537

gnutls_session_t session;

525

526

ret = init_gnutls_session(mc, &session);

538

int pf; /* Protocol family */

539

540

switch(af){

541

case AF_INET6:

542

pf = PF_INET6;

543

break;

544

case AF_INET:

545

pf = PF_INET;

546

break;

547

default:

548

fprintf(stderr, "Bad address family: %d\n", af);

549

return -1;

550

}

551

552

ret = init_gnutls_session(&session);

527

553

if(ret != 0){

528

554

return -1;

529

555

}

530

556

531

557

if(debug){

532

fprintf(stderr, "Setting up a tcp connection to %s, port %" PRIu16

558

fprintf(stderr, "Setting up a TCP connection to %s, port %" PRIu16

533

559

"\n", ip, port);

534

560

}

535

561

536

tcp_sd = socket(PF_INET6, SOCK_STREAM, 0);

562

tcp_sd = socket(pf, SOCK_STREAM, 0);

537

563

if(tcp_sd < 0){

538

564

perror("socket");

539

565

return -1;

540

566

}

541

567

542

if(debug){

543

if(if_indextoname((unsigned int)if_index, interface) == NULL){

544

perror("if_indextoname");

545

return -1;

546

}

547

fprintf(stderr, "Binding to interface %s\n", interface);

548

}

549

550

568

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

551

to.in6.sin6_family = AF_INET6;

552

/* It would be nice to have a way to detect if we were passed an

553

IPv4 address here. Now we assume an IPv6 address. */

554

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

569

if(af == AF_INET6){

570

to.in6.sin6_family = (sa_family_t)af;

571

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

572

} else { /* IPv4 */

573

to.in.sin_family = (sa_family_t)af;

574

ret = inet_pton(af, ip, &to.in.sin_addr);

575

}

555

576

if(ret < 0 ){

556

577

perror("inet_pton");

557

578

return -1;

560

581

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

561

582

return -1;

562

583

}

563

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

584

if(af == AF_INET6){

585

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

586

-Wconversion and

587

-Wunreachable-code */

588

589

if(IN6_IS_ADDR_LINKLOCAL /* Spurious warnings from */

590

(&to.in6.sin6_addr)){ /* -Wstrict-aliasing=2 or lower and

591

-Wunreachable-code*/

592

if(if_index == AVAHI_IF_UNSPEC){

593

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

594

" without a network interface\n");

595

return -1;

596

}

597

/* Set the network interface number as scope */

598

to.in6.sin6_scope_id = (uint32_t)if_index;

599

}

600

} else {

601

to.in.sin_port = htons(port); /* Spurious warnings from

564

602

-Wconversion and

565

603

-Wunreachable-code */

566

567

to.in6.sin6_scope_id = (uint32_t)if_index;

604

}

568

605

569

606

if(debug){

570

fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,

571

port);

572

char addrstr[INET6_ADDRSTRLEN] = "";

573

if(inet_ntop(to.in6.sin6_family, &(to.in6.sin6_addr), addrstr,

574

sizeof(addrstr)) == NULL){

607

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

608

char interface[IF_NAMESIZE];

609

if(if_indextoname((unsigned int)if_index, interface) == NULL){

610

perror("if_indextoname");

611

} else {

612

fprintf(stderr, "Connection to: %s%%%s, port %" PRIu16 "\n",

613

ip, interface, port);

614

}

615

} else {

616

fprintf(stderr, "Connection to: %s, port %" PRIu16 "\n", ip,

617

port);

618

}

619

char addrstr[(INET_ADDRSTRLEN > INET6_ADDRSTRLEN) ?

620

INET_ADDRSTRLEN : INET6_ADDRSTRLEN] = "";

621

const char *pcret;

622

if(af == AF_INET6){

623

pcret = inet_ntop(af, &(to.in6.sin6_addr), addrstr,

624

sizeof(addrstr));

625

} else {

626

pcret = inet_ntop(af, &(to.in.sin_addr), addrstr,

627

sizeof(addrstr));

628

}

629

if(pcret == NULL){

575

630

perror("inet_ntop");

576

631

} else {

577

632

if(strcmp(addrstr, ip) != 0){

580

635

}

581

636

}

582

637

583

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

638

if(af == AF_INET6){

639

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

640

} else {

641

ret = connect(tcp_sd, &to.in, sizeof(to)); /* IPv4 */

642

}

584

643

if(ret < 0){

585

644

perror("connect");

586

645

return -1;

632

691

/* Read OpenPGP packet that contains the wanted password */

633

692

634

693

if(debug){

635

fprintf(stderr, "Retrieving pgp encrypted password from %s\n",

694

fprintf(stderr, "Retrieving OpenPGP encrypted password from %s\n",

636

695

ip);

637

696

}

638

697

639

698

while(true){

640

buffer_capacity = adjustbuffer(&buffer, buffer_length,

699

buffer_capacity = incbuffer(&buffer, buffer_length,

641

700

buffer_capacity);

642

701

if(buffer_capacity == 0){

643

perror("adjustbuffer");

702

perror("incbuffer");

644

703

retval = -1;

645

704

goto mandos_end;

646

705

}

685

744

gnutls_bye(session, GNUTLS_SHUT_RDWR);

686

745

687

746

if(buffer_length > 0){

688

decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,

747

decrypted_buffer_size = pgp_packet_decrypt(buffer,

689

748

buffer_length,

690

749

&decrypted_buffer);

691

750

if(decrypted_buffer_size >= 0){

726

785

727

786

static void resolve_callback(AvahiSServiceResolver *r,

728

787

AvahiIfIndex interface,

729

AVAHI_GCC_UNUSED AvahiProtocol protocol,

788

AvahiProtocol proto,

730

789

AvahiResolverEvent event,

731

790

const char *name,

732

791

const char *type,

737

796

AVAHI_GCC_UNUSED AvahiStringList *txt,

738

797

AVAHI_GCC_UNUSED AvahiLookupResultFlags

739

798

flags,

740

void* userdata){

741

mandos_context *mc = userdata;

799

AVAHI_GCC_UNUSED void* userdata){

742

800

assert(r);

743

801

744

802

/* Called whenever a service has been resolved successfully or

745

803

timed out */

746

804

805

if(quit_now){

806

return;

807

}

808

747

809

switch(event){

748

810

default:

749

811

case AVAHI_RESOLVER_FAILURE:

750

812

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

751

813

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

752

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

814

avahi_strerror(avahi_server_errno(mc.server)));

753

815

break;

754

816

755

817

case AVAHI_RESOLVER_FOUND:

761

823

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

762

824

ip, (intmax_t)interface, port);

763

825

}

764

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

826

int ret = start_mandos_communication(ip, port, interface,

827

avahi_proto_to_af(proto));

765

828

if(ret == 0){

766

avahi_simple_poll_quit(mc->simple_poll);

829

avahi_simple_poll_quit(mc.simple_poll);

767

830

}

768

831

}

769

832

}

779

842

const char *domain,

780

843

AVAHI_GCC_UNUSED AvahiLookupResultFlags

781

844

flags,

782

void* userdata){

783

mandos_context *mc = userdata;

845

AVAHI_GCC_UNUSED void* userdata){

784

846

assert(b);

785

847

786

848

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

791

853

case AVAHI_BROWSER_FAILURE:

792

854

793

855

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

794

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

795

avahi_simple_poll_quit(mc->simple_poll);

856

avahi_strerror(avahi_server_errno(mc.server)));

857

avahi_simple_poll_quit(mc.simple_poll);

796

858

return;

797

859

798

860

case AVAHI_BROWSER_NEW:

801

863

the callback function is called the Avahi server will free the

802

864

resolver for us. */

803

865

804

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

805

protocol, name, type, domain,

806

AVAHI_PROTO_INET6, 0,

807

resolve_callback, mc)))

866

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

867

name, type, domain, protocol, 0,

868

resolve_callback, NULL) == NULL)

808

869

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

809

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

870

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

810

871

break;

811

872

812

873

case AVAHI_BROWSER_REMOVE:

821

882

}

822

883

}

823

884

885

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

886

static void handle_sigterm(int sig){

887

if(quit_now){

888

return;

889

}

890

quit_now = 1;

891

signal_received = sig;

892

int old_errno = errno;

893

if(mc.simple_poll != NULL){

894

avahi_simple_poll_quit(mc.simple_poll);

895

}

896

errno = old_errno;

897

}

898

824

899

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

825

900

AvahiSServiceBrowser *sb = NULL;

826

901

int error;

827

902

int ret;

828

903

intmax_t tmpmax;

829

int numchars;

904

char *tmp;

830

905

int exitcode = EXIT_SUCCESS;

831

906

const char *interface = "eth0";

832

907

struct ifreq network;

833

int sd;

908

int sd = -1;

909

bool take_down_interface = false;

834

910

uid_t uid;

835

911

gid_t gid;

836

912

char *connect_to = NULL;

840

916

const char *seckey = PATHDIR "/" SECKEY;

841

917

const char *pubkey = PATHDIR "/" PUBKEY;

842

918

843

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

844

.dh_bits = 1024, .priority = "SECURE256"

845

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

846

919

bool gnutls_initialized = false;

847

920

bool gpgme_initialized = false;

848

double delay = 2.5;

921

float delay = 2.5f;

922

923

struct sigaction old_sigterm_action;

924

struct sigaction sigterm_action = { .sa_handler = handle_sigterm };

849

925

850

926

{

851

927

struct argp_option options[] = {

857

933

.group = 1 },

858

934

{ .name = "interface", .key = 'i',

859

935

.arg = "NAME",

860

.doc = "Interface that will be used to search for Mandos"

861

" servers",

936

.doc = "Network interface that will be used to search for"

937

" Mandos servers",

862

938

.group = 1 },

863

939

{ .name = "seckey", .key = 's',

864

940

.arg = "FILE",

903

979

pubkey = arg;

904

980

break;

905

981

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

906

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

907

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

908

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

982

errno = 0;

983

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

984

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

985

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

909

986

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

910

987

exit(EXIT_FAILURE);

911

988

}

915

992

mc.priority = arg;

916

993

break;

917

994

case 131: /* --delay */

918

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

919

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

995

errno = 0;

996

delay = strtof(arg, &tmp);

997

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

920

998

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

921

999

exit(EXIT_FAILURE);

922

1000

}

943

1021

}

944

1022

}

945

1023

1024

if(not debug){

1025

avahi_set_log_function(empty_log);

1026

}

1027

1028

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

1029

from the signal handler */

1030

/* Initialize the pseudo-RNG for Avahi */

1031

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

1032

mc.simple_poll = avahi_simple_poll_new();

1033

if(mc.simple_poll == NULL){

1034

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

1035

exitcode = EXIT_FAILURE;

1036

goto end;

1037

}

1038

1039

sigemptyset(&sigterm_action.sa_mask);

1040

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

1041

if(ret == -1){

1042

perror("sigaddset");

1043

exitcode = EXIT_FAILURE;

1044

goto end;

1045

}

1046

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

1047

if(ret == -1){

1048

perror("sigaddset");

1049

exitcode = EXIT_FAILURE;

1050

goto end;

1051

}

1052

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

1053

if(ret == -1){

1054

perror("sigaddset");

1055

exitcode = EXIT_FAILURE;

1056

goto end;

1057

}

1058

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

1059

if(ret == -1){

1060

perror("sigaction");

1061

exitcode = EXIT_FAILURE;

1062

goto end;

1063

}

1064

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

1065

if(ret == -1){

1066

perror("sigaction");

1067

exitcode = EXIT_FAILURE;

1068

goto end;

1069

}

1070

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

1071

if(ret == -1){

1072

perror("sigaction");

1073

exitcode = EXIT_FAILURE;

1074

goto end;

1075

}

1076

946

1077

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

947

{

1078

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

1079

if_index = (AvahiIfIndex) if_nametoindex(interface);

1080

if(if_index == 0){

1081

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

1082

exitcode = EXIT_FAILURE;

1083

goto end;

1084

}

1085

1086

if(quit_now){

1087

goto end;

1088

}

1089

948

1090

#ifdef __linux__

949

1091

/* Lower kernel loglevel to KERN_NOTICE to avoid KERN_INFO

950

1092

messages to mess up the prompt */

951

1093

ret = klogctl(8, NULL, 5);

1094

bool restore_loglevel = true;

952

1095

if(ret == -1){

1096

restore_loglevel = false;

953

1097

perror("klogctl");

954

1098

}

955

#endif

1099

#endif /* __linux__ */

956

1100

957

1101

sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);

958

1102

if(sd < 0){

959

1103

perror("socket");

960

1104

exitcode = EXIT_FAILURE;

961

1105

#ifdef __linux__

962

ret = klogctl(7, NULL, 0);

963

if(ret == -1){

964

perror("klogctl");

1106

if(restore_loglevel){

1107

ret = klogctl(7, NULL, 0);

1108

if(ret == -1){

1109

perror("klogctl");

1110

}

965

1111

}

966

#endif

1112

#endif /* __linux__ */

967

1113

goto end;

968

1114

}

969

1115

strcpy(network.ifr_name, interface);

971

1117

if(ret == -1){

972

1118

perror("ioctl SIOCGIFFLAGS");

973

1119

#ifdef __linux__

974

ret = klogctl(7, NULL, 0);

975

if(ret == -1){

976

perror("klogctl");

1120

if(restore_loglevel){

1121

ret = klogctl(7, NULL, 0);

1122

if(ret == -1){

1123

perror("klogctl");

1124

}

977

1125

}

978

#endif

1126

#endif /* __linux__ */

979

1127

exitcode = EXIT_FAILURE;

980

1128

goto end;

981

1129

}

982

1130

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

983

1131

network.ifr_flags |= IFF_UP;

1132

take_down_interface = true;

984

1133

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

985

1134

if(ret == -1){

1135

take_down_interface = false;

986

1136

perror("ioctl SIOCSIFFLAGS");

987

1137

exitcode = EXIT_FAILURE;

988

1138

#ifdef __linux__

989

ret = klogctl(7, NULL, 0);

990

if(ret == -1){

991

perror("klogctl");

1139

if(restore_loglevel){

1140

ret = klogctl(7, NULL, 0);

1141

if(ret == -1){

1142

perror("klogctl");

1143

}

992

1144

}

993

#endif

1145

#endif /* __linux__ */

994

1146

goto end;

995

1147

}

996

1148

}

1008

1160

perror("nanosleep");

1009

1161

}

1010

1162

}

1011

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

1012

if(ret == -1){

1013

perror("close");

1163

if(not take_down_interface){

1164

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

1165

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

1166

if(ret == -1){

1167

perror("close");

1168

}

1014

1169

}

1015

1170

#ifdef __linux__

1016

/* Restores kernel loglevel to default */

1017

ret = klogctl(7, NULL, 0);

1018

if(ret == -1){

1019

perror("klogctl");

1171

if(restore_loglevel){

1172

/* Restores kernel loglevel to default */

1173

ret = klogctl(7, NULL, 0);

1174

if(ret == -1){

1175

perror("klogctl");

1176

}

1020

1177

}

1021

#endif

1178

#endif /* __linux__ */

1179

}

1180

1181

if(quit_now){

1182

goto end;

1022

1183

}

1023

1184

1024

1185

uid = getuid();

1025

1186

gid = getgid();

1026

1187

1188

errno = 0;

1027

1189

setgid(gid);

1028

1190

if(ret == -1){

1029

1191

perror("setgid");

1034

1196

perror("setuid");

1035

1197

}

1036

1198

1037

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

1199

if(quit_now){

1200

goto end;

1201

}

1202

1203

ret = init_gnutls_global(pubkey, seckey);

1038

1204

if(ret == -1){

1039

1205

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

1040

1206

exitcode = EXIT_FAILURE;

1043

1209

gnutls_initialized = true;

1044

1210

}

1045

1211

1212

if(quit_now){

1213

goto end;

1214

}

1215

1216

tempdir_created = true;

1046

1217

if(mkdtemp(tempdir) == NULL){

1218

tempdir_created = false;

1047

1219

perror("mkdtemp");

1048

1220

goto end;

1049

1221

}

1050

tempdir_created = true;

1051

1052

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

1222

1223

if(quit_now){

1224

goto end;

1225

}

1226

1227

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

1053

1228

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

1054

1229

exitcode = EXIT_FAILURE;

1055

1230

goto end;

1057

1232

gpgme_initialized = true;

1058

1233

}

1059

1234

1060

if_index = (AvahiIfIndex) if_nametoindex(interface);

1061

if(if_index == 0){

1062

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

1063

exitcode = EXIT_FAILURE;

1235

if(quit_now){

1064

1236

goto end;

1065

1237

}

1066

1238

1073

1245

exitcode = EXIT_FAILURE;

1074

1246

goto end;

1075

1247

}

1248

1249

if(quit_now){

1250

goto end;

1251

}

1252

1076

1253

uint16_t port;

1077

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

1078

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

1079

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

1254

errno = 0;

1255

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

1256

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

1257

or tmpmax != (uint16_t)tmpmax){

1080

1258

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

1081

1259

exitcode = EXIT_FAILURE;

1082

1260

goto end;

1083

1261

}

1262

1263

if(quit_now){

1264

goto end;

1265

}

1266

1084

1267

port = (uint16_t)tmpmax;

1085

1268

*address = '\0';

1086

1269

address = connect_to;

1087

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

1270

/* Colon in address indicates IPv6 */

1271

int af;

1272

if(strchr(address, ':') != NULL){

1273

af = AF_INET6;

1274

} else {

1275

af = AF_INET;

1276

}

1277

1278

if(quit_now){

1279

goto end;

1280

}

1281

1282

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

1088

1283

if(ret < 0){

1089

1284

exitcode = EXIT_FAILURE;

1090

1285

} else {

1093

1288

goto end;

1094

1289

}

1095

1290

1096

if(not debug){

1097

avahi_set_log_function(empty_log);

1098

}

1099

1100

/* Initialize the pseudo-RNG for Avahi */

1101

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

1102

1103

/* Allocate main Avahi loop object */

1104

mc.simple_poll = avahi_simple_poll_new();

1105

if(mc.simple_poll == NULL){

1106

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

1107

exitcode = EXIT_FAILURE;

1291

if(quit_now){

1108

1292

goto end;

1109

1293

}

1110

1294

1134

1318

goto end;

1135

1319

}

1136

1320

1321

if(quit_now){

1322

goto end;

1323

}

1324

1137

1325

/* Create the Avahi service browser */

1138

1326

sb = avahi_s_service_browser_new(mc.server, if_index,

1139

AVAHI_PROTO_INET6, "_mandos._tcp",

1140

NULL, 0, browse_callback, &mc);

1327

AVAHI_PROTO_UNSPEC, "_mandos._tcp",

1328

NULL, 0, browse_callback, NULL);

1141

1329

if(sb == NULL){

1142

1330

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

1143

1331

avahi_strerror(avahi_server_errno(mc.server)));

1145

1333

goto end;

1146

1334

}

1147

1335

1336

if(quit_now){

1337

goto end;

1338

}

1339

1148

1340

/* Run the main loop */

1149

1341

1150

1342

if(debug){

1179

1371

gpgme_release(mc.ctx);

1180

1372

}

1181

1373

1374

/* Take down the network interface */

1375

if(take_down_interface){

1376

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

1377

if(ret == -1){

1378

perror("ioctl SIOCGIFFLAGS");

1379

} else if(network.ifr_flags & IFF_UP) {

1380

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

1381

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

1382

if(ret == -1){

1383

perror("ioctl SIOCSIFFLAGS");

1384

}

1385

}

1386

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

1387

if(ret == -1){

1388

perror("close");

1389

}

1390

}

1391

1182

1392

/* Removes the temp directory used by GPGME */

1183

1393

if(tempdir_created){

1184

1394

DIR *d;

1223

1433

}

1224

1434

}

1225

1435

1436

if(quit_now){

1437

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

1438

if(ret == -1){

1439

perror("sigaction");

1440

}

1441

raise(signal_received);

1442

}

1443

1226

1444

return exitcode;

1227

1445

}