/mandos/trunk : revision 266

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

14

*

14

15

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

15

16

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

35

36

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

36

37

38

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

38

stdout, ferror() */

39

stdout, ferror(), sscanf */

39

40

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

40

41

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

41

42

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

47

48

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

48

49

sockaddr_in6, PF_INET6,

49

50

SOCK_STREAM, INET6_ADDRSTRLEN,

50

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

51

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

52

DIR */

51

53

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

52

54

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

53

55

struct in6_addr, inet_pton(),

54

56

connect() */

55

57

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

56

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

57

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

58

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

59

*/

60

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

58

61

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

59

62

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

60

63

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

66

69

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

67

70

setgid() */

68

71

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

69

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

72

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

70

73

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

71

74

argp_state, struct argp,

72

75

argp_parse(), ARGP_KEY_ARG,

89

92

gnutls_*

90

93

init_gnutls_session(),

91

94

GNUTLS_* */

92

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

95

#include <gnutls/openpgp.h>

96

/* gnutls_certificate_set_openpgp_key_file(),

93

97

GNUTLS_OPENPGP_FMT_BASE64 */

94

98

95

99

/* GPGME */

101

105

102

106

#define BUFFER_SIZE 256

103

107

104

/*

105

#define PATHDIR "/conf/conf.d/mandos"

106

*/

107

108

#define PATHDIR "/conf/conf.d/mandos"

109

#define SECKEY "seckey.txt"

110

#define PUBKEY "pubkey.txt"

111

112

bool debug = false;

113

static const char mandos_protocol_version[] = "1";

114

const char *argp_program_version = "mandos-client 1.0";

114

const char *argp_program_version = "mandos-client " VERSION;

115

const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";

116

117

/* Used for passing in values through the Avahi callback functions */

132

*/

133

size_t adjustbuffer(char **buffer, size_t buffer_length,

134

size_t buffer_capacity){

135

if (buffer_length + BUFFER_SIZE > buffer_capacity){

135

if(buffer_length + BUFFER_SIZE > buffer_capacity){

136

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

137

if (buffer == NULL){

137

if(buffer == NULL){

138

return 0;

139

}

140

buffer_capacity += BUFFER_SIZE;

159

int fd;

160

gpgme_data_t pgp_data;

161

162

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

162

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

163

if(fd == -1){

164

perror("open");

165

return false;

166

}

167

168

rc = gpgme_data_new_from_fd(&pgp_data, fd);

169

if (rc != GPG_ERR_NO_ERROR){

169

if(rc != GPG_ERR_NO_ERROR){

170

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

171

gpgme_strsource(rc), gpgme_strerror(rc));

172

return false;

173

}

174

175

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

176

if (rc != GPG_ERR_NO_ERROR){

176

if(rc != GPG_ERR_NO_ERROR){

177

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

178

gpgme_strsource(rc), gpgme_strerror(rc));

179

return false;

180

}

181

182

ret = TEMP_FAILURE_RETRY(close(fd));

182

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

183

if(ret == -1){

184

perror("close");

185

}

187

return true;

188

}

189

190

if (debug){

190

if(debug){

191

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

192

}

193

194

/* Init GPGME */

195

gpgme_check_version(NULL);

196

rc = gpgme_engine_check_version(GPGME_PROTOCOL_OpenPGP);

197

if (rc != GPG_ERR_NO_ERROR){

197

if(rc != GPG_ERR_NO_ERROR){

198

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

199

gpgme_strsource(rc), gpgme_strerror(rc));

200

return false;

201

}

202

203

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

204

rc = gpgme_get_engine_info (&engine_info);

205

if (rc != GPG_ERR_NO_ERROR){

204

rc = gpgme_get_engine_info(&engine_info);

205

if(rc != GPG_ERR_NO_ERROR){

206

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

207

gpgme_strsource(rc), gpgme_strerror(rc));

208

return false;

222

223

/* Create new GPGME "context" */

224

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

225

if (rc != GPG_ERR_NO_ERROR){

225

if(rc != GPG_ERR_NO_ERROR){

226

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

227

gpgme_strsource(rc), gpgme_strerror(rc));

228

return false;

229

}

230

231

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

231

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

232

return false;

233

}

234

239

* Decrypt OpenPGP data.

240

* Returns -1 on error

241

*/

242

static ssize_t pgp_packet_decrypt (const mandos_context *mc,

243

const char *cryptotext,

244

size_t crypto_size,

245

char **plaintext){

242

static ssize_t pgp_packet_decrypt(const mandos_context *mc,

243

const char *cryptotext,

244

size_t crypto_size,

245

char **plaintext){

246

gpgme_data_t dh_crypto, dh_plain;

247

gpgme_error_t rc;

248

ssize_t ret;

249

size_t plaintext_capacity = 0;

250

ssize_t plaintext_length = 0;

251

252

if (debug){

252

if(debug){

253

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

254

}

255

256

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

257

rc = gpgme_data_new_from_mem(&dh_crypto, cryptotext, crypto_size,

258

0);

259

if (rc != GPG_ERR_NO_ERROR){

259

if(rc != GPG_ERR_NO_ERROR){

260

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

261

gpgme_strsource(rc), gpgme_strerror(rc));

262

return -1;

264

265

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

266

rc = gpgme_data_new(&dh_plain);

267

if (rc != GPG_ERR_NO_ERROR){

267

if(rc != GPG_ERR_NO_ERROR){

268

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

269

gpgme_strsource(rc), gpgme_strerror(rc));

270

gpgme_data_release(dh_crypto);

274

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

275

data buffer */

276

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

277

if (rc != GPG_ERR_NO_ERROR){

277

if(rc != GPG_ERR_NO_ERROR){

278

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

279

gpgme_strsource(rc), gpgme_strerror(rc));

280

plaintext_length = -1;

281

if (debug){

281

if(debug){

282

gpgme_decrypt_result_t result;

283

result = gpgme_op_decrypt_result(mc->ctx);

284

if (result == NULL){

284

if(result == NULL){

285

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

286

} else {

287

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

314

}

315

316

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

317

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

317

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

318

perror("gpgme_data_seek");

319

plaintext_length = -1;

320

goto decrypt_end;

325

plaintext_capacity = adjustbuffer(plaintext,

326

(size_t)plaintext_length,

327

plaintext_capacity);

328

if (plaintext_capacity == 0){

328

if(plaintext_capacity == 0){

329

perror("adjustbuffer");

330

plaintext_length = -1;

331

goto decrypt_end;

334

ret = gpgme_data_read(dh_plain, *plaintext + plaintext_length,

335

BUFFER_SIZE);

336

/* Print the data, if any */

337

if (ret == 0){

337

if(ret == 0){

338

/* EOF */

339

break;

340

}

364

return plaintext_length;

365

}

366

367

static const char * safer_gnutls_strerror (int value) {

368

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

369

if (ret == NULL)

367

static const char * safer_gnutls_strerror(int value) {

368

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

369

if(ret == NULL)

370

ret = "(unknown)";

371

return ret;

372

}

387

}

388

389

ret = gnutls_global_init();

390

if (ret != GNUTLS_E_SUCCESS) {

391

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

392

safer_gnutls_strerror(ret));

390

if(ret != GNUTLS_E_SUCCESS) {

391

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

392

safer_gnutls_strerror(ret));

393

return -1;

394

}

395

396

if (debug){

396

if(debug){

397

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

398

* - GnuTLS manual

399

*/

403

404

/* OpenPGP credentials */

405

gnutls_certificate_allocate_credentials(&mc->cred);

406

if (ret != GNUTLS_E_SUCCESS){

407

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

408

warning */

409

safer_gnutls_strerror(ret));

410

gnutls_global_deinit ();

406

if(ret != GNUTLS_E_SUCCESS){

407

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

408

warning */

409

safer_gnutls_strerror(ret));

410

gnutls_global_deinit();

411

return -1;

412

}

413

420

ret = gnutls_certificate_set_openpgp_key_file

421

(mc->cred, pubkeyfilename, seckeyfilename,

422

GNUTLS_OPENPGP_FMT_BASE64);

423

if (ret != GNUTLS_E_SUCCESS) {

423

if(ret != GNUTLS_E_SUCCESS) {

424

fprintf(stderr,

425

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

426

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

431

432

/* GnuTLS server initialization */

433

ret = gnutls_dh_params_init(&mc->dh_params);

434

if (ret != GNUTLS_E_SUCCESS) {

435

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

436

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

434

if(ret != GNUTLS_E_SUCCESS) {

435

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

436

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

437

goto globalfail;

438

}

439

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

440

if (ret != GNUTLS_E_SUCCESS) {

441

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

442

safer_gnutls_strerror(ret));

440

if(ret != GNUTLS_E_SUCCESS) {

441

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

442

safer_gnutls_strerror(ret));

443

goto globalfail;

444

}

445

460

int ret;

461

/* GnuTLS session creation */

462

ret = gnutls_init(session, GNUTLS_SERVER);

463

if (ret != GNUTLS_E_SUCCESS){

463

if(ret != GNUTLS_E_SUCCESS){

464

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

465

safer_gnutls_strerror(ret));

466

}

468

{

469

const char *err;

470

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

471

if (ret != GNUTLS_E_SUCCESS) {

471

if(ret != GNUTLS_E_SUCCESS) {

472

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

473

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

474

safer_gnutls_strerror(ret));

475

gnutls_deinit (*session);

475

gnutls_deinit(*session);

476

return -1;

477

}

478

}

479

480

ret = gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE,

481

mc->cred);

482

if (ret != GNUTLS_E_SUCCESS) {

482

if(ret != GNUTLS_E_SUCCESS) {

483

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

484

safer_gnutls_strerror(ret));

485

gnutls_deinit (*session);

485

gnutls_deinit(*session);

486

return -1;

487

}

488

489

/* ignore client certificate if any. */

490

gnutls_certificate_server_set_request (*session,

491

GNUTLS_CERT_IGNORE);

490

gnutls_certificate_server_set_request(*session,

491

GNUTLS_CERT_IGNORE);

492

493

gnutls_dh_set_prime_bits (*session, mc->dh_bits);

493

gnutls_dh_set_prime_bits(*session, mc->dh_bits);

494

495

return 0;

496

}

504

AvahiIfIndex if_index,

505

mandos_context *mc){

506

int ret, tcp_sd;

507

ssize_t sret;

507

508

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

508

509

char *buffer = NULL;

509

510

char *decrypted_buffer;

515

516

char interface[IF_NAMESIZE];

516

517

gnutls_session_t session;

517

518

ret = init_gnutls_session (mc, &session);

519

if (ret != 0){

519

ret = init_gnutls_session(mc, &session);

520

if(ret != 0){

520

521

return -1;

521

522

}

522

523

544

545

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

545

546

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

546

547

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

547

if (ret < 0 ){

548

if(ret < 0 ){

548

549

perror("inet_pton");

549

550

return -1;

550

551

}

571

572

}

572

573

574

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

574

if (ret < 0){

575

if(ret < 0){

575

576

perror("connect");

576

577

return -1;

577

578

}

578

579

580

const char *out = mandos_protocol_version;

580

581

written = 0;

581

while (true){

582

while(true){

582

583

size_t out_size = strlen(out);

583

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

584

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

584

585

out_size - written));

585

if (ret == -1){

586

if(ret == -1){

586

587

perror("write");

587

588

retval = -1;

588

589

goto mandos_end;

591

592

if(written < out_size){

592

593

continue;

593

594

} else {

594

if (out == mandos_protocol_version){

595

if(out == mandos_protocol_version){

595

596

written = 0;

596

597

out = "\r\n";

597

598

} else {

604

605

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

605

606

}

606

607

gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) tcp_sd);

608

gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) tcp_sd);

608

609

610

do{

610

ret = gnutls_handshake (session);

611

ret = gnutls_handshake(session);

611

612

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

612

613

if (ret != GNUTLS_E_SUCCESS){

614

if(ret != GNUTLS_E_SUCCESS){

614

615

if(debug){

615

616

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

616

gnutls_perror (ret);

617

gnutls_perror(ret);

617

618

}

618

619

retval = -1;

619

620

goto mandos_end;

629

630

while(true){

630

631

buffer_capacity = adjustbuffer(&buffer, buffer_length,

631

632

buffer_capacity);

632

if (buffer_capacity == 0){

633

if(buffer_capacity == 0){

633

634

perror("adjustbuffer");

634

635

retval = -1;

635

636

goto mandos_end;

636

637

}

637

638

ret = gnutls_record_recv(session, buffer+buffer_length,

639

BUFFER_SIZE);

640

if (ret == 0){

639

sret = gnutls_record_recv(session, buffer+buffer_length,

640

BUFFER_SIZE);

641

if(sret == 0){

641

642

break;

642

643

}

643

if (ret < 0){

644

switch(ret){

644

if(sret < 0){

645

switch(sret){

645

646

case GNUTLS_E_INTERRUPTED:

646

647

case GNUTLS_E_AGAIN:

647

648

break;

648

649

case GNUTLS_E_REHANDSHAKE:

649

650

do{

650

ret = gnutls_handshake (session);

651

ret = gnutls_handshake(session);

651

652

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

652

if (ret < 0){

653

if(ret < 0){

653

654

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

654

gnutls_perror (ret);

655

gnutls_perror(ret);

655

656

retval = -1;

656

657

goto mandos_end;

657

658

}

660

661

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

661

662

" encrypted session with Mandos server\n");

662

663

retval = -1;

663

gnutls_bye (session, GNUTLS_SHUT_RDWR);

664

gnutls_bye(session, GNUTLS_SHUT_RDWR);

664

665

goto mandos_end;

665

666

}

666

667

} else {

667

buffer_length += (size_t) ret;

668

buffer_length += (size_t) sret;

668

669

}

669

670

}

670

671

672

673

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

673

674

}

674

675

gnutls_bye (session, GNUTLS_SHUT_RDWR);

676

gnutls_bye(session, GNUTLS_SHUT_RDWR);

676

677

if (buffer_length > 0){

678

if(buffer_length > 0){

678

679

decrypted_buffer_size = pgp_packet_decrypt(mc, buffer,

679

680

buffer_length,

680

681

&decrypted_buffer);

681

if (decrypted_buffer_size >= 0){

682

if(decrypted_buffer_size >= 0){

682

683

written = 0;

683

684

while(written < (size_t) decrypted_buffer_size){

684

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

685

(size_t)decrypted_buffer_size - written,

686

stdout);

685

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

686

(size_t)decrypted_buffer_size - written,

687

stdout);

687

688

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

688

689

if(debug){

689

690

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

706

707

708

mandos_end:

708

709

free(buffer);

709

ret = TEMP_FAILURE_RETRY(close(tcp_sd));

710

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

710

711

if(ret == -1){

711

712

perror("close");

712

713

}

713

gnutls_deinit (session);

714

gnutls_deinit(session);

714

715

return retval;

715

716

}

716

717

734

735

/* Called whenever a service has been resolved successfully or

735

736

timed out */

736

737

switch (event) {

738

switch(event) {

738

739

default:

739

740

case AVAHI_RESOLVER_FAILURE:

740

741

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

752

753

interface, port);

753

754

}

754

755

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

755

if (ret == 0){

756

if(ret == 0){

756

757

avahi_simple_poll_quit(mc->simple_poll);

757

758

}

758

759

}

776

777

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

777

778

is removed from the LAN */

778

779

switch (event) {

780

switch(event) {

780

781

default:

781

782

case AVAHI_BROWSER_FAILURE:

782

783

791

792

the callback function is called the Avahi server will free the

792

793

resolver for us. */

793

794

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

795

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

795

796

protocol, name, type, domain,

796

797

AVAHI_PROTO_INET6, 0,

797

798

resolve_callback, mc)))

866

867

{ .name = NULL }

867

868

};

868

869

error_t parse_opt (int key, char *arg,

870

struct argp_state *state) {

871

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

872

a pointer to our plugin list pointer. */

873

switch (key) {

870

error_t parse_opt(int key, char *arg,

871

struct argp_state *state) {

872

switch(key) {

874

873

case 128: /* --debug */

875

874

debug = true;

876

875

break;

887

886

pubkey = arg;

888

887

break;

889

888

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

890

errno = 0;

891

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

892

if (errno){

893

perror("strtol");

889

ret = sscanf(arg, "%u", &mc.dh_bits);

890

if(ret != 1){

891

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

894

892

exit(EXIT_FAILURE);

895

893

}

896

894

break;

898

896

mc.priority = arg;

899

897

break;

900

898

case ARGP_KEY_ARG:

901

argp_usage (state);

899

argp_usage(state);

902

900

case ARGP_KEY_END:

903

901

break;

904

902

default:

911

909

.args_doc = "",

912

910

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

913

911

" passwords from a Mandos server" };

914

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

915

if (ret == ARGP_ERR_UNKNOWN){

912

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

913

if(ret == ARGP_ERR_UNKNOWN){

916

914

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

917

915

exitcode = EXIT_FAILURE;

918

916

goto end;

943

941

goto end;

944

942

}

945

943

}

946

ret = TEMP_FAILURE_RETRY(close(sd));

944

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

947

945

if(ret == -1){

948

946

perror("close");

949

947

}

953

951

gid = getgid();

954

952

955

953

ret = setuid(uid);

956

if (ret == -1){

954

if(ret == -1){

957

955

perror("setuid");

958

956

}

959

957

960

958

setgid(gid);

961

if (ret == -1){

959

if(ret == -1){

962

960

perror("setgid");

963

961

}

964

962

965

963

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

966

if (ret == -1){

964

if(ret == -1){

967

965

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

968

966

exitcode = EXIT_FAILURE;

969

967

goto end;

1000

998

exitcode = EXIT_FAILURE;

1001

999

goto end;

1002

1000

}

1003

errno = 0;

1004

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

1005

if(errno){

1006

perror("Bad port number");

1001

uint16_t port;

1002

ret = sscanf(address+1, "%" SCNu16, &port);

1003

if(ret != 1){

1004

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

1007

1005

exitcode = EXIT_FAILURE;

1008

1006

goto end;

1009

1007

}

1018

1016

goto end;

1019

1017

}

1020

1018

1021

if (not debug){

1019

if(not debug){

1022

1020

avahi_set_log_function(empty_log);

1023

1021

}

1024

1022

1027

1025

1028

1026

/* Allocate main Avahi loop object */

1029

1027

mc.simple_poll = avahi_simple_poll_new();

1030

if (mc.simple_poll == NULL) {

1028

if(mc.simple_poll == NULL) {

1031

1029

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

1032

1030

" object.\n");

1033

1031

exitcode = EXIT_FAILURE;

1053

1051

}

1054

1052

1055

1053

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

1056

if (mc.server == NULL) {

1054

if(mc.server == NULL) {

1057

1055

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

1058

1056

avahi_strerror(error));

1059

1057

exitcode = EXIT_FAILURE;

1065

1063

AVAHI_PROTO_INET6,

1066

1064

"_mandos._tcp", NULL, 0,

1067

1065

browse_callback, &mc);

1068

if (sb == NULL) {

1066

if(sb == NULL) {

1069

1067

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

1070

1068

avahi_strerror(avahi_server_errno(mc.server)));

1071

1069

exitcode = EXIT_FAILURE;

1074

1072

1075

1073

/* Run the main loop */

1076

1074

1077

if (debug){

1075

if(debug){

1078

1076

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

1079

1077

}

1080

1078

1082

1080

1083

1081

end:

1084

1082

1085

if (debug){

1083

if(debug){

1086

1084

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

1087

1085

}

1088

1086

1089

1087

/* Cleanup things */

1090

if (sb != NULL)

1088

if(sb != NULL)

1091

1089

avahi_s_service_browser_free(sb);

1092

1090

1093

if (mc.server != NULL)

1091

if(mc.server != NULL)

1094

1092

avahi_server_free(mc.server);

1095

1093

1096

if (mc.simple_poll != NULL)

1094

if(mc.simple_poll != NULL)

1097

1095

avahi_simple_poll_free(mc.simple_poll);

1098

1096

1099

if (gnutls_initalized){

1097

if(gnutls_initalized){

1100

1098

gnutls_certificate_free_credentials(mc.cred);

1101

gnutls_global_deinit ();

1099

gnutls_global_deinit();

1102

1100

gnutls_dh_params_deinit(mc.dh_params);

1103

1101

}

1104

1102

1112

1110

struct dirent *direntry;

1113

1111

d = opendir(tempdir);

1114

1112

if(d == NULL){

1115

perror("opendir");

1113

if(errno != ENOENT){

1114

perror("opendir");

1115

}

1116

} else {

1117

while(true){

1118

direntry = readdir(d);

1119

if(direntry == NULL){

1120

break;

1121

}

1122

if (direntry->d_type == DT_REG){

1122

if(direntry->d_type == DT_REG){

1123

char *fullname = NULL;

1124

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

1125

direntry->d_name);

1138

closedir(d);

1139

}

1140

ret = rmdir(tempdir);

1141

if(ret == -1){

1141

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

1142

perror("rmdir");

1143

}

1144

}