/mandos/trunk : revision 234

1

/* -*- coding: utf-8 -*- */

2

/*

3

* Mandos-client - get and decrypt data from a Mandos server

3

* Mandos client - get and decrypt data from a Mandos server

4

*

5

* This program is partly derived from an example program for an Avahi

6

* service browser, downloaded from

9

* "browse_callback", and parts of "main".

10

*

11

* Everything else is

12

13

12

14

13

*

15

14

* This program is free software: you can redistribute it and/or

16

15

* modify it under the terms of the GNU General Public License as

36

35

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

37

36

38

37

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

39

stdout, ferror(), sscanf(),

40

remove() */

38

stdout, ferror() */

41

39

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

42

40

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

43

41

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

49

47

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

50

48

sockaddr_in6, PF_INET6,

51

49

SOCK_STREAM, INET6_ADDRSTRLEN,

52

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

53

DIR */

50

uid_t, gid_t, open(), opendir(), DIR */

54

51

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

55

52

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

56

53

struct in6_addr, inet_pton(),

57

54

connect() */

58

55

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

59

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

60

*/

61

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

56

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

57

#include <inttypes.h> /* PRIu16 */

62

58

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

63

59

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

64

60

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

70

66

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

71

67

setgid() */

72

68

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

73

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

69

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

74

70

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

75

71

argp_state, struct argp,

76

72

argp_parse(), ARGP_KEY_ARG,

93

89

gnutls_*

94

90

init_gnutls_session(),

95

91

GNUTLS_* */

96

#include <gnutls/openpgp.h>

97

/* gnutls_certificate_set_openpgp_key_file(),

92

#include <gnutls/openpgp.h> /* gnutls_certificate_set_openpgp_key_file(),

98

93

GNUTLS_OPENPGP_FMT_BASE64 */

99

94

100

95

/* GPGME */

133

128

*/

134

129

size_t adjustbuffer(char **buffer, size_t buffer_length,

135

130

size_t buffer_capacity){

136

if(buffer_length + BUFFER_SIZE > buffer_capacity){

131

if (buffer_length + BUFFER_SIZE > buffer_capacity){

137

132

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

138

if(buffer == NULL){

133

if (buffer == NULL){

139

134

return 0;

140

135

}

141

136

buffer_capacity += BUFFER_SIZE;

154

149

155

150

156

151

/*

157

* Helper function to insert pub and seckey to the engine keyring.

152

* Helper function to insert pub and seckey to the enigne keyring.

158

153

*/

159

154

bool import_key(const char *filename){

160

155

int fd;

161

156

gpgme_data_t pgp_data;

162

157

163

fd = (int)TEMP_FAILURE_RETRY(open(filename, O_RDONLY));

158

fd = TEMP_FAILURE_RETRY(open(filename, O_RDONLY));

164

159

if(fd == -1){

165

160

perror("open");

166

161

return false;

167

162

}

168

163

169

164

rc = gpgme_data_new_from_fd(&pgp_data, fd);

170

if(rc != GPG_ERR_NO_ERROR){

165

if (rc != GPG_ERR_NO_ERROR){

171

166

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

172

167

gpgme_strsource(rc), gpgme_strerror(rc));

173

168

return false;

174

169

}

175

170

176

171

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

177

if(rc != GPG_ERR_NO_ERROR){

172

if (rc != GPG_ERR_NO_ERROR){

178

173

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

179

174

gpgme_strsource(rc), gpgme_strerror(rc));

180

175

return false;

181

176

}

182

177

183

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

178

ret = TEMP_FAILURE_RETRY(close(fd));

184

179

if(ret == -1){

185

180

perror("close");

186

181

}

188

183

return true;

189

184

}

190

185

191

if(debug){

186

if (debug){

192

187

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

193

188

}

194

189

195

190

/* Init GPGME */

196

191

gpgme_check_version(NULL);

197

192

rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);

198

if(rc != GPG_ERR_NO_ERROR){

193

if (rc != GPG_ERR_NO_ERROR){

199

194

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

200

195

gpgme_strsource(rc), gpgme_strerror(rc));

201

196

return false;

202

197

}

203

198

204

199

/* Set GPGME home directory for the OpenPGP engine only */

205

rc = gpgme_get_engine_info(&engine_info);

206

if(rc != GPG_ERR_NO_ERROR){

200

rc = gpgme_get_engine_info (&engine_info);

201

if (rc != GPG_ERR_NO_ERROR){

207

202

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

208

203

gpgme_strsource(rc), gpgme_strerror(rc));

209

204

return false;

223

218

224

219

/* Create new GPGME "context" */

225

220

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

226

if(rc != GPG_ERR_NO_ERROR){

221

if (rc != GPG_ERR_NO_ERROR){

227

222

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

228

223

gpgme_strsource(rc), gpgme_strerror(rc));

229

224

return false;

230

225

}

231

226

232

if(not import_key(pubkey) or not import_key(seckey)){

227

if (not import_key(pubkey) or not import_key(seckey)){

233

228

return false;

234

229

}

235

230

240

235

* Decrypt OpenPGP data.

241

236

* Returns -1 on error

242

237

*/

243

static ssize_t pgp_packet_decrypt(const mandos_context *mc,

244

const char *cryptotext,

245

size_t crypto_size,

246

char **plaintext){

238

static ssize_t pgp_packet_decrypt (const mandos_context *mc,

239

const char *cryptotext,

240

size_t crypto_size,

241

char **plaintext){

247

242

gpgme_data_t dh_crypto, dh_plain;

248

243

gpgme_error_t rc;

249

244

ssize_t ret;

250

245

size_t plaintext_capacity = 0;

251

246

ssize_t plaintext_length = 0;

252

247

253

if(debug){

248

if (debug){

254

249

fprintf(stderr, "Trying to decrypt OpenPGP data\n");

255

250

}

256

251

257

252

/* Create new GPGME data buffer from memory cryptotext */

258

253

rc = gpgme_data_new_from_mem(&dh_crypto, cryptotext, crypto_size,

259

254

0);

260

if(rc != GPG_ERR_NO_ERROR){

255

if (rc != GPG_ERR_NO_ERROR){

261

256

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

262

257

gpgme_strsource(rc), gpgme_strerror(rc));

263

258

return -1;

265

260

266

261

/* Create new empty GPGME data buffer for the plaintext */

267

262

rc = gpgme_data_new(&dh_plain);

268

if(rc != GPG_ERR_NO_ERROR){

263

if (rc != GPG_ERR_NO_ERROR){

269

264

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

270

265

gpgme_strsource(rc), gpgme_strerror(rc));

271

266

gpgme_data_release(dh_crypto);

275

270

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

276

271

data buffer */

277

272

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

278

if(rc != GPG_ERR_NO_ERROR){

273

if (rc != GPG_ERR_NO_ERROR){

279

274

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

280

275

gpgme_strsource(rc), gpgme_strerror(rc));

281

276

plaintext_length = -1;

282

if(debug){

277

if (debug){

283

278

gpgme_decrypt_result_t result;

284

279

result = gpgme_op_decrypt_result(mc->ctx);

285

if(result == NULL){

280

if (result == NULL){

286

281

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

287

282

} else {

288

283

fprintf(stderr, "Unsupported algorithm: %s\n",

315

310

}

316

311

317

312

/* Seek back to the beginning of the GPGME plaintext data buffer */

318

if(gpgme_data_seek(dh_plain, (off_t)0, SEEK_SET) == -1){

313

if (gpgme_data_seek(dh_plain, (off_t) 0, SEEK_SET) == -1){

319

314

perror("gpgme_data_seek");

320

315

plaintext_length = -1;

321

316

goto decrypt_end;

326

321

plaintext_capacity = adjustbuffer(plaintext,

327

322

(size_t)plaintext_length,

328

323

plaintext_capacity);

329

if(plaintext_capacity == 0){

324

if (plaintext_capacity == 0){

330

325

perror("adjustbuffer");

331

326

plaintext_length = -1;

332

327

goto decrypt_end;

335

330

ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,

336

331

BUFFER_SIZE);

337

332

/* Print the data, if any */

338

if(ret == 0){

333

if (ret == 0){

339

334

/* EOF */

340

335

break;

341

336

}

365

360

return plaintext_length;

366

361

}

367

362

368

static const char * safer_gnutls_strerror(int value) {

369

const char *ret = gnutls_strerror(value); /* Spurious warning from

370

-Wunreachable-code */

371

if(ret == NULL)

363

static const char * safer_gnutls_strerror (int value) {

364

const char *ret = gnutls_strerror (value); /* Spurious warning */

365

if (ret == NULL)

372

366

ret = "(unknown)";

373

367

return ret;

374

368

}

389

383

}

390

384

391

385

ret = gnutls_global_init();

392

if(ret != GNUTLS_E_SUCCESS) {

393

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

394

safer_gnutls_strerror(ret));

386

if (ret != GNUTLS_E_SUCCESS) {

387

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

388

safer_gnutls_strerror(ret));

395

389

return -1;

396

390

}

397

391

398

if(debug){

392

if (debug){

399

393

/* "Use a log level over 10 to enable all debugging options."

400

394

* - GnuTLS manual

401

395

*/

405

399

406

400

/* OpenPGP credentials */

407

401

gnutls_certificate_allocate_credentials(&mc->cred);

408

if(ret != GNUTLS_E_SUCCESS){

409

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

410

* from

411

* -Wunreachable-code

412

*/

413

safer_gnutls_strerror(ret));

414

gnutls_global_deinit();

402

if (ret != GNUTLS_E_SUCCESS){

403

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

404

warning */

405

safer_gnutls_strerror(ret));

406

gnutls_global_deinit ();

415

407

return -1;

416

408

}

417

409

424

416

ret = gnutls_certificate_set_openpgp_key_file

425

417

(mc->cred, pubkeyfilename, seckeyfilename,

426

418

GNUTLS_OPENPGP_FMT_BASE64);

427

if(ret != GNUTLS_E_SUCCESS) {

419

if (ret != GNUTLS_E_SUCCESS) {

428

420

fprintf(stderr,

429

421

"Error[%d] while reading the OpenPGP key pair ('%s',"

430

422

" '%s')\n", ret, pubkeyfilename, seckeyfilename);

435

427

436

428

/* GnuTLS server initialization */

437

429

ret = gnutls_dh_params_init(&mc->dh_params);

438

if(ret != GNUTLS_E_SUCCESS) {

439

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

440

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

430

if (ret != GNUTLS_E_SUCCESS) {

431

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

432

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

441

433

goto globalfail;

442

434

}

443

435

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

444

if(ret != GNUTLS_E_SUCCESS) {

445

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

446

safer_gnutls_strerror(ret));

436

if (ret != GNUTLS_E_SUCCESS) {

437

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

438

safer_gnutls_strerror(ret));

447

439

goto globalfail;

448

440

}

449

441

464

456

int ret;

465

457

/* GnuTLS session creation */

466

458

ret = gnutls_init(session, GNUTLS_SERVER);

467

if(ret != GNUTLS_E_SUCCESS){

459

if (ret != GNUTLS_E_SUCCESS){

468

460

fprintf(stderr, "Error in GnuTLS session initialization: %s\n",

469

461

safer_gnutls_strerror(ret));

470

462

}

472

464

{

473

465

const char *err;

474

466

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

475

if(ret != GNUTLS_E_SUCCESS) {

467

if (ret != GNUTLS_E_SUCCESS) {

476

468

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

477

469

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

478

470

safer_gnutls_strerror(ret));

479

gnutls_deinit(*session);

471

gnutls_deinit (*session);

480

472

return -1;

481

473

}

482

474

}

483

475

484

476

ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,

485

477

mc->cred);

486

if(ret != GNUTLS_E_SUCCESS) {

478

if (ret != GNUTLS_E_SUCCESS) {

487

479

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

488

480

safer_gnutls_strerror(ret));

489

gnutls_deinit(*session);

481

gnutls_deinit (*session);

490

482

return -1;

491

483

}

492

484

493

485

/* ignore client certificate if any. */

494

gnutls_certificate_server_set_request(*session,

495

GNUTLS_CERT_IGNORE);

486

gnutls_certificate_server_set_request (*session,

487

GNUTLS_CERT_IGNORE);

496

488

497

gnutls_dh_set_prime_bits(*session, mc->dh_bits);

489

gnutls_dh_set_prime_bits (*session, mc->dh_bits);

498

490

499

491

return 0;

500

492

}

508

500

AvahiIfIndex if_index,

509

501

mandos_context *mc){

510

502

int ret, tcp_sd;

511

ssize_t sret;

512

503

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

513

504

char *buffer = NULL;

514

505

char *decrypted_buffer;

520

511

char interface[IF_NAMESIZE];

521

512

gnutls_session_t session;

522

513

523

ret = init_gnutls_session(mc, &session);

524

if(ret != 0){

514

ret = init_gnutls_session (mc, &session);

515

if (ret != 0){

525

516

return -1;

526

517

}

527

518

549

540

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

550

541

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

551

542

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

552

if(ret < 0 ){

543

if (ret < 0 ){

553

544

perror("inet_pton");

554

545

return -1;

555

546

}

557

548

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

558

549

return -1;

559

550

}

560

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

561

-Wconversion and

562

-Wunreachable-code */

551

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

563

552

564

553

to.in6.sin6_scope_id = (uint32_t)if_index;

565

554

578

567

}

579

568

580

569

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

581

if(ret < 0){

570

if (ret < 0){

582

571

perror("connect");

583

572

return -1;

584

573

}

585

574

586

575

const char *out = mandos_protocol_version;

587

576

written = 0;

588

while(true){

577

while (true){

589

578

size_t out_size = strlen(out);

590

ret = (int)TEMP_FAILURE_RETRY(write(tcp_sd, out + written,

579

ret = TEMP_FAILURE_RETRY(write(tcp_sd, out + written,

591

580

out_size - written));

592

if(ret == -1){

581

if (ret == -1){

593

582

perror("write");

594

583

retval = -1;

595

584

goto mandos_end;

598

587

if(written < out_size){

599

588

continue;

600

589

} else {

601

if(out == mandos_protocol_version){

590

if (out == mandos_protocol_version){

602

591

written = 0;

603

592

out = "\r\n";

604

593

} else {

611

600

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

612

601

}

613

602

614

gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);

603

gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) tcp_sd);

615

604

616

605

do{

617

ret = gnutls_handshake(session);

606

ret = gnutls_handshake (session);

618

607

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

619

608

620

if(ret != GNUTLS_E_SUCCESS){

609

if (ret != GNUTLS_E_SUCCESS){

621

610

if(debug){

622

611

fprintf(stderr, "*** GnuTLS Handshake failed ***\n");

623

gnutls_perror(ret);

612

gnutls_perror (ret);

624

613

}

625

614

retval = -1;

626

615

goto mandos_end;

636

625

while(true){

637

626

buffer_capacity = adjustbuffer(&buffer, buffer_length,

638

627

buffer_capacity);

639

if(buffer_capacity == 0){

628

if (buffer_capacity == 0){

640

629

perror("adjustbuffer");

641

630

retval = -1;

642

631

goto mandos_end;

643

632

}

644

633

645

sret = gnutls_record_recv(session, buffer+buffer_length,

646

BUFFER_SIZE);

647

if(sret == 0){

634

ret = gnutls_record_recv(session, buffer+buffer_length,

635

BUFFER_SIZE);

636

if (ret == 0){

648

637

break;

649

638

}

650

if(sret < 0){

651

switch(sret){

639

if (ret < 0){

640

switch(ret){

652

641

case GNUTLS_E_INTERRUPTED:

653

642

case GNUTLS_E_AGAIN:

654

643

break;

655

644

case GNUTLS_E_REHANDSHAKE:

656

645

do{

657

ret = gnutls_handshake(session);

646

ret = gnutls_handshake (session);

658

647

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

659

if(ret < 0){

648

if (ret < 0){

660

649

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

661

gnutls_perror(ret);

650

gnutls_perror (ret);

662

651

retval = -1;

663

652

goto mandos_end;

664

653

}

667

656

fprintf(stderr, "Unknown error while reading data from"

668

657

" encrypted session with Mandos server\n");

669

658

retval = -1;

670

gnutls_bye(session, GNUTLS_SHUT_RDWR);

659

gnutls_bye (session, GNUTLS_SHUT_RDWR);

671

660

goto mandos_end;

672

661

}

673

662

} else {

674

buffer_length += (size_t) sret;

663

buffer_length += (size_t) ret;

675

664

}

676

665

}

677

666

679

668

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

680

669

}

681

670

682

gnutls_bye(session, GNUTLS_SHUT_RDWR);

671

gnutls_bye (session, GNUTLS_SHUT_RDWR);

683

672

684

if(buffer_length > 0){

673

if (buffer_length > 0){

685

674

decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,

686

675

buffer_length,

687

676

&decrypted_buffer);

688

if(decrypted_buffer_size >= 0){

677

if (decrypted_buffer_size >= 0){

689

678

written = 0;

690

679

while(written < (size_t) decrypted_buffer_size){

691

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

692

(size_t)decrypted_buffer_size - written,

693

stdout);

680

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

681

(size_t)decrypted_buffer_size - written,

682

stdout);

694

683

if(ret == 0 and ferror(stdout)){

695

684

if(debug){

696

685

fprintf(stderr, "Error writing encrypted data: %s\n",

713

702

714

703

mandos_end:

715

704

free(buffer);

716

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

705

ret = TEMP_FAILURE_RETRY(close(tcp_sd));

717

706

if(ret == -1){

718

707

perror("close");

719

708

}

720

gnutls_deinit(session);

709

gnutls_deinit (session);

721

710

return retval;

722

711

}

723

712

741

730

/* Called whenever a service has been resolved successfully or

742

731

timed out */

743

732

744

switch(event) {

733

switch (event) {

745

734

default:

746

735

case AVAHI_RESOLVER_FAILURE:

747

736

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

755

744

avahi_address_snprint(ip, sizeof(ip), address);

756

745

if(debug){

757

746

fprintf(stderr, "Mandos server \"%s\" found on %s (%s, %"

758

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

759

ip, (intmax_t)interface, port);

747

PRIu16 ") on port %d\n", name, host_name, ip,

748

interface, port);

760

749

}

761

750

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

762

if(ret == 0){

751

if (ret == 0){

763

752

avahi_simple_poll_quit(mc->simple_poll);

764

753

}

765

754

}

783

772

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

784

773

is removed from the LAN */

785

774

786

switch(event) {

775

switch (event) {

787

776

default:

788

777

case AVAHI_BROWSER_FAILURE:

789

778

798

787

the callback function is called the Avahi server will free the

799

788

resolver for us. */

800

789

801

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

790

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

802

791

protocol, name, type, domain,

803

792

AVAHI_PROTO_INET6, 0,

804

793

resolve_callback, mc)))

822

811

AvahiSServiceBrowser *sb = NULL;

823

812

int error;

824

813

int ret;

825

intmax_t tmpmax;

826

int numchars;

827

814

int exitcode = EXIT_SUCCESS;

828

815

const char *interface = "eth0";

829

816

struct ifreq network;

832

819

gid_t gid;

833

820

char *connect_to = NULL;

834

821

char tempdir[] = "/tmp/mandosXXXXXX";

835

bool tempdir_created = false;

836

822

AvahiIfIndex if_index = AVAHI_IF_UNSPEC;

837

823

const char *seckey = PATHDIR "/" SECKEY;

838

824

const char *pubkey = PATHDIR "/" PUBKEY;

840

826

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

841

827

.dh_bits = 1024, .priority = "SECURE256"

842

828

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

843

bool gnutls_initialized = false;

844

bool gpgme_initialized = false;

829

bool gnutls_initalized = false;

830

bool gpgme_initalized = false;

845

831

846

832

{

847

833

struct argp_option options[] = {

876

862

{ .name = NULL }

877

863

};

878

864

879

error_t parse_opt(int key, char *arg,

880

struct argp_state *state) {

881

switch(key) {

865

error_t parse_opt (int key, char *arg,

866

struct argp_state *state) {

867

/* Get the INPUT argument from `argp_parse', which we know is

868

a pointer to our plugin list pointer. */

869

switch (key) {

882

870

case 128: /* --debug */

883

871

debug = true;

884

872

break;

895

883

pubkey = arg;

896

884

break;

897

885

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

898

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

899

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

900

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

901

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

886

errno = 0;

887

mc.dh_bits = (unsigned int) strtol(arg, NULL, 10);

888

if (errno){

889

perror("strtol");

902

890

exit(EXIT_FAILURE);

903

891

}

904

mc.dh_bits = (typeof(mc.dh_bits))tmpmax;

905

892

break;

906

893

case 130: /* --priority */

907

894

mc.priority = arg;

908

895

break;

909

896

case ARGP_KEY_ARG:

910

argp_usage(state);

897

argp_usage (state);

911

898

case ARGP_KEY_END:

912

899

break;

913

900

default:

920

907

.args_doc = "",

921

908

.doc = "Mandos client -- Get and decrypt"

922

909

" passwords from a Mandos server" };

923

ret = argp_parse(&argp, argc, argv, 0, 0, NULL);

924

if(ret == ARGP_ERR_UNKNOWN){

910

ret = argp_parse (&argp, argc, argv, 0, 0, NULL);

911

if (ret == ARGP_ERR_UNKNOWN){

925

912

fprintf(stderr, "Unknown error while parsing arguments\n");

926

913

exitcode = EXIT_FAILURE;

927

914

goto end;

952

939

goto end;

953

940

}

954

941

}

955

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

942

ret = TEMP_FAILURE_RETRY(close(sd));

956

943

if(ret == -1){

957

944

perror("close");

958

945

}

961

948

uid = getuid();

962

949

gid = getgid();

963

950

951

ret = setuid(uid);

952

if (ret == -1){

953

perror("setuid");

954

}

955

964

956

setgid(gid);

965

if(ret == -1){

957

if (ret == -1){

966

958

perror("setgid");

967

959

}

968

960

969

ret = setuid(uid);

970

if(ret == -1){

971

perror("setuid");

972

}

973

974

961

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

975

if(ret == -1){

962

if (ret == -1){

976

963

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

977

964

exitcode = EXIT_FAILURE;

978

965

goto end;

979

966

} else {

980

gnutls_initialized = true;

967

gnutls_initalized = true;

981

968

}

982

969

983

970

if(mkdtemp(tempdir) == NULL){

984

971

perror("mkdtemp");

972

tempdir[0] = '\0';

985

973

goto end;

986

974

}

987

tempdir_created = true;

988

975

989

976

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

990

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

977

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

991

978

exitcode = EXIT_FAILURE;

992

979

goto end;

993

980

} else {

994

gpgme_initialized = true;

981

gpgme_initalized = true;

995

982

}

996

983

997

984

if_index = (AvahiIfIndex) if_nametoindex(interface);

998

985

if(if_index == 0){

999

986

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

1000

exitcode = EXIT_FAILURE;

1001

goto end;

987

exit(EXIT_FAILURE);

1002

988

}

1003

989

1004

990

if(connect_to != NULL){

1010

996

exitcode = EXIT_FAILURE;

1011

997

goto end;

1012

998

}

1013

uint16_t port;

1014

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

1015

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

1016

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

1017

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

999

errno = 0;

1000

uint16_t port = (uint16_t) strtol(address+1, NULL, 10);

1001

if(errno){

1002

perror("Bad port number");

1018

1003

exitcode = EXIT_FAILURE;

1019

1004

goto end;

1020

1005

}

1021

port = (uint16_t)tmpmax;

1022

1006

*address = '\0';

1023

1007

address = connect_to;

1024

1008

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

1030

1014

goto end;

1031

1015

}

1032

1016

1033

if(not debug){

1017

if (not debug){

1034

1018

avahi_set_log_function(empty_log);

1035

1019

}

1036

1020

1039

1023

1040

1024

/* Allocate main Avahi loop object */

1041

1025

mc.simple_poll = avahi_simple_poll_new();

1042

if(mc.simple_poll == NULL) {

1026

if (mc.simple_poll == NULL) {

1043

1027

fprintf(stderr, "Avahi: Failed to create simple poll"

1044

1028

" object.\n");

1045

1029

exitcode = EXIT_FAILURE;

1065

1049

}

1066

1050

1067

1051

/* Check if creating the Avahi server object succeeded */

1068

if(mc.server == NULL) {

1052

if (mc.server == NULL) {

1069

1053

fprintf(stderr, "Failed to create Avahi server: %s\n",

1070

1054

avahi_strerror(error));

1071

1055

exitcode = EXIT_FAILURE;

1077

1061

AVAHI_PROTO_INET6,

1078

1062

"_mandos._tcp", NULL, 0,

1079

1063

browse_callback, &mc);

1080

if(sb == NULL) {

1064

if (sb == NULL) {

1081

1065

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

1082

1066

avahi_strerror(avahi_server_errno(mc.server)));

1083

1067

exitcode = EXIT_FAILURE;

1086

1070

1087

1071

/* Run the main loop */

1088

1072

1089

if(debug){

1073

if (debug){

1090

1074

fprintf(stderr, "Starting Avahi loop search\n");

1091

1075

}

1092

1076

1094

1078

1095

1079

end:

1096

1080

1097

if(debug){

1081

if (debug){

1098

1082

fprintf(stderr, "%s exiting\n", argv[0]);

1099

1083

}

1100

1084

1101

1085

/* Cleanup things */

1102

if(sb != NULL)

1086

if (sb != NULL)

1103

1087

avahi_s_service_browser_free(sb);

1104

1088

1105

if(mc.server != NULL)

1089

if (mc.server != NULL)

1106

1090

avahi_server_free(mc.server);

1107

1091

1108

if(mc.simple_poll != NULL)

1092

if (mc.simple_poll != NULL)

1109

1093

avahi_simple_poll_free(mc.simple_poll);

1110

1094

1111

if(gnutls_initialized){

1095

if (gnutls_initalized){

1112

1096

gnutls_certificate_free_credentials(mc.cred);

1113

gnutls_global_deinit();

1097

gnutls_global_deinit ();

1114

1098

gnutls_dh_params_deinit(mc.dh_params);

1115

1099

}

1116

1100

1117

if(gpgme_initialized){

1101

if(gpgme_initalized){

1118

1102

gpgme_release(mc.ctx);

1119

1103

}

1120

1104

1121

1105

/* Removes the temp directory used by GPGME */

1122

if(tempdir_created){

1106

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

1123

1107

DIR *d;

1124

1108

struct dirent *direntry;

1125

1109

d = opendir(tempdir);

1126

1110

if(d == NULL){

1127

if(errno != ENOENT){

1128

perror("opendir");

1129

}

1111

perror("opendir");

1130

1112

} else {

1131

1113

while(true){

1132

1114

direntry = readdir(d);

1133

1115

if(direntry == NULL){

1134

1116

break;

1135

1117

}

1136

/* Skip "." and ".." */

1137

if(direntry->d_name[0] == '.'

1138

and (direntry->d_name[1] == '\0'

1139

or (direntry->d_name[1] == '.'

1140

and direntry->d_name[2] == '\0'))){

1141

continue;

1142

}

1143

char *fullname = NULL;

1144

ret = asprintf(&fullname, "%s/%s", tempdir,

1145

direntry->d_name);

1146

if(ret < 0){

1147

perror("asprintf");

1148

continue;

1149

}

1150

ret = remove(fullname);

1151

if(ret == -1){

1152

fprintf(stderr, "remove(\"%s\"): %s\n", fullname,

1153

strerror(errno));

1154

}

1155

free(fullname);

1118

if (direntry->d_type == DT_REG){

1119

char *fullname = NULL;

1120

ret = asprintf(&fullname, "%s/%s", tempdir,

1121

direntry->d_name);

1122

if(ret < 0){

1123

perror("asprintf");

1124

continue;

1125

}

1126

ret = unlink(fullname);

1127

if(ret == -1){

1128

fprintf(stderr, "unlink(\"%s\"): %s",

1129

fullname, strerror(errno));

1130

}

1131

free(fullname);

1132

}

1156

1133

}

1157

1134

closedir(d);

1158

1135

}

1159

1136

ret = rmdir(tempdir);

1160

if(ret == -1 and errno != ENOENT){

1137

if(ret == -1){

1161

1138

perror("rmdir");

1162

1139

}

1163

1140

}

1164

1141

1165

1142

return exitcode;

1166

1143

}