/mandos/trunk : revision 181

1

/* -*- coding: utf-8; mode: c; mode: orgtbl -*- */

1

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

2

/*

3

* Mandos plugin runner - Run Mandos plugins

4

*

5

6

5

7

6

*

8

7

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

9

8

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

28

27

#include <stdlib.h> /* malloc(), exit(), EXIT_FAILURE,

29

28

EXIT_SUCCESS, realloc() */

30

29

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

31

#include <stdio.h> /* perror, fileno(), fprintf(),

32

stderr, STDOUT_FILENO */

30

#include <stdio.h> /* perror, popen(), fileno(),

31

fprintf(), stderr, STDOUT_FILENO */

33

32

#include <sys/types.h> /* DIR, opendir(), stat(), struct

34

33

stat, waitpid(), WIFEXITED(),

35

34

WEXITSTATUS(), wait(), pid_t,

38

37

#include <sys/select.h> /* fd_set, select(), FD_ZERO(),

39

38

FD_SET(), FD_ISSET(), FD_CLR */

40

39

#include <sys/wait.h> /* wait(), waitpid(), WIFEXITED(),

41

WEXITSTATUS(), WTERMSIG(),

42

WCOREDUMP() */

40

WEXITSTATUS() */

43

41

#include <sys/stat.h> /* struct stat, stat(), S_ISREG() */

44

42

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

45

43

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

48

46

fcntl(), setuid(), setgid(),

49

47

F_GETFD, F_SETFD, FD_CLOEXEC,

50

48

access(), pipe(), fork(), close()

51

dup2(), STDOUT_FILENO, _exit(),

49

dup2, STDOUT_FILENO, _exit(),

52

50

execv(), write(), read(),

53

51

close() */

54

52

#include <fcntl.h> /* fcntl(), F_GETFD, F_SETFD,

55

53

FD_CLOEXEC */

56

#include <string.h> /* strsep, strlen(), asprintf(),

57

strsignal(), strcmp(), strncmp() */

54

#include <string.h> /* strsep, strlen(), asprintf() */

58

55

#include <errno.h> /* errno */

59

56

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

60

57

argp_state, struct argp,

64

61

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

65

62

sigaddset(), sigaction(),

66

63

sigprocmask(), SIG_BLOCK, SIGCHLD,

67

SIG_UNBLOCK, kill(), sig_atomic_t

68

*/

64

SIG_UNBLOCK, kill() */

69

65

#include <errno.h> /* errno, EBADF */

70

#include <inttypes.h> /* intmax_t, PRIdMAX, strtoimax() */

71

66

72

67

#define BUFFER_SIZE 256

73

68

74

69

#define PDIR "/lib/mandos/plugins.d"

75

70

#define AFILE "/conf/conf.d/mandos/plugin-runner.conf"

76

71

77

const char *argp_program_version = "plugin-runner " VERSION;

72

const char *argp_program_version = "plugin-runner 1.0";

78

73

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

79

74

75

struct plugin;

76

80

77

typedef struct plugin{

81

78

char *name; /* can be NULL or any plugin name */

82

79

char **argv;

84

81

char **environ;

85

82

int envc;

86

83

bool disabled;

87

84

88

85

/* Variables used for running processes*/

89

86

pid_t pid;

90

87

int fd;

92

89

size_t buffer_size;

93

90

size_t buffer_length;

94

91

bool eof;

95

volatile sig_atomic_t completed;

96

int status;

92

volatile bool completed;

93

volatile int status;

97

94

struct plugin *next;

98

95

} plugin;

99

96

103

100

or if none is found, creates a new one */

104

101

static plugin *getplugin(char *name){

105

102

/* Check for exiting plugin with that name */

106

for(plugin *p = plugin_list; p != NULL; p = p->next){

107

if((p->name == name)

108

or (p->name and name and (strcmp(p->name, name) == 0))){

103

for (plugin *p = plugin_list; p != NULL; p = p->next){

104

if ((p->name == name)

105

or (p->name and name and (strcmp(p->name, name) == 0))){

109

106

return p;

110

107

}

111

108

}

112

109

/* Create a new plugin */

113

plugin *new_plugin = NULL;

114

do {

115

new_plugin = malloc(sizeof(plugin));

116

} while(new_plugin == NULL and errno == EINTR);

117

if(new_plugin == NULL){

110

plugin *new_plugin = malloc(sizeof(plugin));

111

if (new_plugin == NULL){

118

112

return NULL;

119

113

}

120

114

char *copy_name = NULL;

121

115

if(name != NULL){

122

do {

123

copy_name = strdup(name);

124

} while(copy_name == NULL and errno == EINTR);

116

copy_name = strdup(name);

125

117

if(copy_name == NULL){

126

free(new_plugin);

127

118

return NULL;

128

119

}

129

120

}

130

121

131

*new_plugin = (plugin){ .name = copy_name,

132

.argc = 1,

133

.disabled = false,

134

.next = plugin_list };

122

*new_plugin = (plugin) { .name = copy_name,

123

.argc = 1,

124

.disabled = false,

125

.next = plugin_list };

135

126

136

do {

137

new_plugin->argv = malloc(sizeof(char *) * 2);

138

} while(new_plugin->argv == NULL and errno == EINTR);

139

if(new_plugin->argv == NULL){

127

new_plugin->argv = malloc(sizeof(char *) * 2);

128

if (new_plugin->argv == NULL){

140

129

free(copy_name);

141

130

free(new_plugin);

142

131

return NULL;

144

133

new_plugin->argv[0] = copy_name;

145

134

new_plugin->argv[1] = NULL;

146

135

147

do {

148

new_plugin->environ = malloc(sizeof(char *));

149

} while(new_plugin->environ == NULL and errno == EINTR);

136

new_plugin->environ = malloc(sizeof(char *));

150

137

if(new_plugin->environ == NULL){

151

138

free(copy_name);

152

139

free(new_plugin->argv);

164

151

static bool add_to_char_array(const char *new, char ***array,

165

152

int *len){

166

153

/* Resize the pointed-to array to hold one more pointer */

167

do {

168

*array = realloc(*array, sizeof(char *)

169

* (size_t) ((*len) + 2));

170

} while(*array == NULL and errno == EINTR);

154

*array = realloc(*array, sizeof(char *)

155

* (size_t) ((*len) + 2));

171

156

/* Malloc check */

172

157

if(*array == NULL){

173

158

return false;

174

159

}

175

160

/* Make a copy of the new string */

176

char *copy;

177

do {

178

copy = strdup(new);

179

} while(copy == NULL and errno == EINTR);

161

char *copy = strdup(new);

180

162

if(copy == NULL){

181

163

return false;

182

164

}

208

190

if(strncmp(*e, def, namelen + 1) == 0){

209

191

/* It already exists */

210

192

if(replace){

211

char *new;

212

do {

213

new = realloc(*e, strlen(def) + 1);

214

} while(new == NULL and errno == EINTR);

193

char *new = realloc(*e, strlen(def) + 1);

215

194

if(new == NULL){

216

195

return false;

217

196

}

227

206

/*

228

207

* Based on the example in the GNU LibC manual chapter 13.13 "File

229

208

* Descriptor Flags".

230

| [[info:libc:Descriptor%20Flags][File Descriptor Flags]] |

209

* *Note File Descriptor Flags:(libc)Descriptor Flags.

231

210

*/

232

static int set_cloexec_flag(int fd){

233

int ret = (int)TEMP_FAILURE_RETRY(fcntl(fd, F_GETFD, 0));

211

static int set_cloexec_flag(int fd)

212

{

213

int ret = fcntl(fd, F_GETFD, 0);

234

214

/* If reading the flags failed, return error indication now. */

235

215

if(ret < 0){

236

216

return ret;

237

217

}

238

218

/* Store modified flag word in the descriptor. */

239

return (int)TEMP_FAILURE_RETRY(fcntl(fd, F_SETFD,

240

ret | FD_CLOEXEC));

219

return fcntl(fd, F_SETFD, ret | FD_CLOEXEC);

241

220

}

242

221

243

222

244

223

/* Mark processes as completed when they exit, and save their exit

245

224

status. */

246

static void handle_sigchld(__attribute__((unused)) int sig){

247

int old_errno = errno;

225

void handle_sigchld(__attribute__((unused)) int sig){

248

226

while(true){

249

227

plugin *proc = plugin_list;

250

228

int status;

254

232

break;

255

233

}

256

234

if(pid == -1){

257

if(errno == ECHILD){

258

/* No child processes */

259

break;

235

if (errno != ECHILD){

236

perror("waitpid");

260

237

}

261

perror("waitpid");

238

/* No child processes */

239

break;

262

240

}

263

241

264

242

/* A child exited, find it in process_list */

270

248

continue;

271

249

}

272

250

proc->status = status;

273

proc->completed = 1;

251

proc->completed = true;

274

252

}

275

errno = old_errno;

276

253

}

277

254

278

255

/* Prints out a password to stdout */

279

static bool print_out_password(const char *buffer, size_t length){

256

bool print_out_password(const char *buffer, size_t length){

280

257

ssize_t ret;

281

258

for(size_t written = 0; written < length; written += (size_t)ret){

282

259

ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO, buffer + written,

300

277

}

301

278

free(plugin_node->environ);

302

279

free(plugin_node->buffer);

303

280

304

281

/* Removes the plugin from the singly-linked list */

305

282

if(plugin_node == plugin_list){

306

283

/* First one - simple */

334

311

struct stat st;

335

312

fd_set rfds_all;

336

313

int ret, maxfd = 0;

337

ssize_t sret;

338

314

uid_t uid = 65534;

339

315

gid_t gid = 65534;

340

316

bool debug = false;

397

373

{ .name = NULL }

398

374

};

399

375

400

error_t parse_opt(int key, char *arg, __attribute__((unused))

401

struct argp_state *state){

402

switch(key){

403

char *tmp;

404

intmax_t tmpmax;

376

error_t parse_opt (int key, char *arg, __attribute__((unused))

377

struct argp_state *state) {

378

switch (key) {

405

379

case 'g': /* --global-options */

406

if(arg != NULL){

407

char *plugin_option;

408

while((plugin_option = strsep(&arg, ",")) != NULL){

409

if(plugin_option[0] == '\0'){

380

if (arg != NULL){

381

char *p;

382

while((p = strsep(&arg, ",")) != NULL){

383

if(p[0] == '\0'){

410

384

continue;

411

385

}

412

if(not add_argument(getplugin(NULL), plugin_option)){

386

if(not add_argument(getplugin(NULL), p)){

413

387

perror("add_argument");

414

388

return ARGP_ERR_UNKNOWN;

415

389

}

425

399

}

426

400

break;

427

401

case 'o': /* --options-for */

428

if(arg != NULL){

429

char *plugin_name = strsep(&arg, ":");

430

if(plugin_name[0] == '\0'){

431

break;

432

}

433

char *plugin_option;

434

while((plugin_option = strsep(&arg, ",")) != NULL){

435

if(not add_argument(getplugin(plugin_name), plugin_option)){

402

if (arg != NULL){

403

char *p_name = strsep(&arg, ":");

404

if(p_name[0] == '\0' or arg == NULL){

405

break;

406

}

407

char *opt = strsep(&arg, ":");

408

if(opt[0] == '\0' or opt == NULL){

409

break;

410

}

411

char *p;

412

while((p = strsep(&opt, ",")) != NULL){

413

if(p[0] == '\0'){

414

continue;

415

}

416

if(not add_argument(getplugin(p_name), p)){

436

417

perror("add_argument");

437

418

return ARGP_ERR_UNKNOWN;

438

419

}

455

436

}

456

437

break;

457

438

case 'd': /* --disable */

458

if(arg != NULL){

439

if (arg != NULL){

459

440

plugin *p = getplugin(arg);

460

441

if(p == NULL){

461

442

return ARGP_ERR_UNKNOWN;

464

445

}

465

446

break;

466

447

case 'e': /* --enable */

467

if(arg != NULL){

448

if (arg != NULL){

468

449

plugin *p = getplugin(arg);

469

450

if(p == NULL){

470

451

return ARGP_ERR_UNKNOWN;

477

458

plugindir = strdup(arg);

478

459

if(plugindir == NULL){

479

460

perror("strdup");

480

}

461

}

481

462

break;

482

463

case 129: /* --config-file */

483

464

/* This is already done by parse_opt_config_file() */

484

465

break;

485

466

case 130: /* --userid */

486

errno = 0;

487

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

488

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

489

or tmpmax != (uid_t)tmpmax){

490

fprintf(stderr, "Bad user ID number: \"%s\", using %"

491

PRIdMAX "\n", arg, (intmax_t)uid);

492

} else {

493

uid = (uid_t)tmpmax;

494

}

467

uid = (uid_t)strtol(arg, NULL, 10);

495

468

break;

496

469

case 131: /* --groupid */

497

errno = 0;

498

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

499

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

500

or tmpmax != (gid_t)tmpmax){

501

fprintf(stderr, "Bad group ID number: \"%s\", using %"

502

PRIdMAX "\n", arg, (intmax_t)gid);

503

} else {

504

gid = (gid_t)tmpmax;

505

}

470

gid = (gid_t)strtol(arg, NULL, 10);

506

471

break;

507

472

case 132: /* --debug */

508

473

debug = true;

509

474

break;

510

/*

511

* When adding more options before this line, remember to also add a

512

* "case" to the "parse_opt_config_file" function below.

513

*/

514

475

case ARGP_KEY_ARG:

515

476

/* Cryptsetup always passes an argument, which is an empty

516

477

string if "none" was specified in /etc/crypttab. So if

529

490

530

491

/* This option parser is the same as parse_opt() above, except it

531

492

ignores everything but the --config-file option. */

532

error_t parse_opt_config_file(int key, char *arg,

533

__attribute__((unused))

534

struct argp_state *state){

535

switch(key){

493

error_t parse_opt_config_file (int key, char *arg,

494

__attribute__((unused))

495

struct argp_state *state) {

496

switch (key) {

536

497

case 'g': /* --global-options */

537

498

case 'G': /* --global-env */

538

499

case 'o': /* --options-for */

547

508

if(argfile == NULL){

548

509

perror("strdup");

549

510

}

550

break;

511

break;

551

512

case 130: /* --userid */

552

513

case 131: /* --groupid */

553

514

case 132: /* --debug */

565

526

.args_doc = "",

566

527

.doc = "Mandos plugin runner -- Run plugins" };

567

528

568

/* Parse using parse_opt_config_file() in order to get the custom

529

/* Parse using the parse_opt_config_file in order to get the custom

569

530

config file location, if any. */

570

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

571

if(ret == ARGP_ERR_UNKNOWN){

531

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

532

if (ret == ARGP_ERR_UNKNOWN){

572

533

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

573

534

exitstatus = EXIT_FAILURE;

574

535

goto fallback;

578

539

argp.parser = parse_opt;

579

540

580

541

/* Open the configfile if available */

581

if(argfile == NULL){

542

if (argfile == NULL){

582

543

conffp = fopen(AFILE, "r");

583

544

} else {

584

545

conffp = fopen(argfile, "r");

585

}

546

}

586

547

if(conffp != NULL){

587

548

char *org_line = NULL;

588

549

char *p, *arg, *new_arg, *line;

589

550

size_t size = 0;

551

ssize_t sret;

590

552

const char whitespace_delims[] = " \r\t\f\v\n";

591

553

const char comment_delim[] = "#";

592

554

593

555

custom_argc = 1;

594

556

custom_argv = malloc(sizeof(char*) * 2);

595

557

if(custom_argv == NULL){

599

561

}

600

562

custom_argv[0] = argv[0];

601

563

custom_argv[1] = NULL;

602

564

603

565

/* for each line in the config file, strip whitespace and ignore

604

566

commented text */

605

567

while(true){

607

569

if(sret == -1){

608

570

break;

609

571

}

610

572

611

573

line = org_line;

612

574

arg = strsep(&line, comment_delim);

613

575

while((p = strsep(&arg, whitespace_delims)) != NULL){

632

594

goto fallback;

633

595

}

634

596

custom_argv[custom_argc-1] = new_arg;

635

custom_argv[custom_argc] = NULL;

597

custom_argv[custom_argc] = NULL;

636

598

}

637

599

}

638

do {

639

ret = fclose(conffp);

640

} while(ret == EOF and errno == EINTR);

641

if(ret == EOF){

642

perror("fclose");

643

exitstatus = EXIT_FAILURE;

644

goto fallback;

645

}

646

600

free(org_line);

647

601

} else {

648

602

/* Check for harmful errors and go to fallback. Other errors might

649

603

not affect opening plugins */

650

if(errno == EMFILE or errno == ENFILE or errno == ENOMEM){

604

if (errno == EMFILE or errno == ENFILE or errno == ENOMEM){

651

605

perror("fopen");

652

606

exitstatus = EXIT_FAILURE;

653

607

goto fallback;

656

610

/* If there was any arguments from configuration file,

657

611

pass them to parser as command arguments */

658

612

if(custom_argv != NULL){

659

ret = argp_parse(&argp, custom_argc, custom_argv, ARGP_IN_ORDER,

660

0, NULL);

661

if(ret == ARGP_ERR_UNKNOWN){

613

ret = argp_parse (&argp, custom_argc, custom_argv, ARGP_IN_ORDER,

614

0, NULL);

615

if (ret == ARGP_ERR_UNKNOWN){

662

616

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

663

617

exitstatus = EXIT_FAILURE;

664

618

goto fallback;

667

621

668

622

/* Parse actual command line arguments, to let them override the

669

623

config file */

670

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

671

if(ret == ARGP_ERR_UNKNOWN){

624

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

625

if (ret == ARGP_ERR_UNKNOWN){

672

626

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

673

627

exitstatus = EXIT_FAILURE;

674

628

goto fallback;

681

635

for(char **a = p->argv; *a != NULL; a++){

682

636

fprintf(stderr, "\tArg: %s\n", *a);

683

637

}

684

fprintf(stderr, "...and %d environment variables\n", p->envc);

638

fprintf(stderr, "...and %u environment variables\n", p->envc);

685

639

for(char **a = p->environ; *a != NULL; a++){

686

640

fprintf(stderr, "\t%s\n", *a);

687

641

}

689

643

}

690

644

691

645

/* Strip permissions down to nobody */

646

ret = setuid(uid);

647

if (ret == -1){

648

perror("setuid");

649

}

692

650

setgid(gid);

693

if(ret == -1){

651

if (ret == -1){

694

652

perror("setgid");

695

653

}

696

ret = setuid(uid);

697

if(ret == -1){

698

perror("setuid");

699

}

700

654

701

if(plugindir == NULL){

655

if (plugindir == NULL){

702

656

dir = opendir(PDIR);

703

657

} else {

704

658

dir = opendir(plugindir);

727

681

728

682

/* Read and execute any executable in the plugin directory*/

729

683

while(true){

730

do {

731

dirst = readdir(dir);

732

} while(dirst == NULL and errno == EINTR);

684

dirst = readdir(dir);

733

685

734

686

/* All directory entries have been processed */

735

687

if(dirst == NULL){

736

if(errno == EBADF){

688

if (errno == EBADF){

737

689

perror("readdir");

738

690

exitstatus = EXIT_FAILURE;

739

691

goto fallback;

751

703

752

704

const char const *bad_suffixes[] = { "~", "#", ".dpkg-new",

753

705

".dpkg-old",

754

".dpkg-bak",

755

706

".dpkg-divert", NULL };

756

707

for(const char **pre = bad_prefixes; *pre != NULL; pre++){

757

708

size_t pre_len = strlen(*pre);

771

722

for(const char **suf = bad_suffixes; *suf != NULL; suf++){

772

723

size_t suf_len = strlen(*suf);

773

724

if((d_name_len >= suf_len)

774

and (strcmp((dirst->d_name) + d_name_len-suf_len, *suf)

725

and (strcmp((dirst->d_name)+d_name_len-suf_len, *suf)

775

726

== 0)){

776

727

if(debug){

777

728

fprintf(stderr, "Ignoring plugin dir entry \"%s\""

786

737

continue;

787

738

}

788

739

}

789

740

790

741

char *filename;

791

742

if(plugindir == NULL){

792

ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, PDIR "/%s",

793

dirst->d_name));

743

ret = asprintf(&filename, PDIR "/%s", dirst->d_name);

794

744

} else {

795

ret = (int)TEMP_FAILURE_RETRY(asprintf(&filename, "%s/%s",

796

plugindir,

797

dirst->d_name));

745

ret = asprintf(&filename, "%s/%s", plugindir, dirst->d_name);

798

746

}

799

747

if(ret < 0){

800

748

perror("asprintf");

801

749

continue;

802

750

}

803

751

804

ret = (int)TEMP_FAILURE_RETRY(stat(filename, &st));

805

if(ret == -1){

752

ret = stat(filename, &st);

753

if (ret == -1){

806

754

perror("stat");

807

755

free(filename);

808

756

continue;

809

757

}

810

758

811

759

/* Ignore non-executable files */

812

if(not S_ISREG(st.st_mode)

813

or (TEMP_FAILURE_RETRY(access(filename, X_OK)) != 0)){

760

if (not S_ISREG(st.st_mode) or (access(filename, X_OK) != 0)){

814

761

if(debug){

815

762

fprintf(stderr, "Ignoring plugin dir entry \"%s\""

816

763

" with bad type or mode\n", filename);

862

809

}

863

810

864

811

int pipefd[2];

865

ret = (int)TEMP_FAILURE_RETRY(pipe(pipefd));

866

if(ret == -1){

812

ret = pipe(pipefd);

813

if (ret == -1){

867

814

perror("pipe");

868

815

exitstatus = EXIT_FAILURE;

869

816

goto fallback;

882

829

goto fallback;

883

830

}

884

831

/* Block SIGCHLD until process is safely in process list */

885

ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,

886

&sigchld_action.sa_mask,

887

NULL));

832

ret = sigprocmask (SIG_BLOCK, &sigchld_action.sa_mask, NULL);

888

833

if(ret < 0){

889

834

perror("sigprocmask");

890

835

exitstatus = EXIT_FAILURE;

891

836

goto fallback;

892

837

}

893

838

/* Starting a new process to be watched */

894

pid_t pid;

895

do {

896

pid = fork();

897

} while(pid == -1 and errno == EINTR);

839

pid_t pid = fork();

898

840

if(pid == -1){

899

841

perror("fork");

900

842

exitstatus = EXIT_FAILURE;

907

849

perror("sigaction");

908

850

_exit(EXIT_FAILURE);

909

851

}

910

ret = sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);

852

ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);

911

853

if(ret < 0){

912

854

perror("sigprocmask");

913

855

_exit(EXIT_FAILURE);

914

856

}

915

857

916

858

ret = dup2(pipefd[1], STDOUT_FILENO); /* replace our stdout */

917

859

if(ret == -1){

918

860

perror("dup2");

938

880

/* no return */

939

881

}

940

882

/* Parent process */

941

TEMP_FAILURE_RETRY(close(pipefd[1])); /* Close unused write end of

942

pipe */

883

close(pipefd[1]); /* Close unused write end of pipe */

943

884

free(filename);

944

885

plugin *new_plugin = getplugin(dirst->d_name);

945

if(new_plugin == NULL){

886

if (new_plugin == NULL){

946

887

perror("getplugin");

947

ret = (int)(TEMP_FAILURE_RETRY

948

(sigprocmask(SIG_UNBLOCK, &sigchld_action.sa_mask,

949

NULL)));

888

ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);

950

889

if(ret < 0){

951

890

perror("sigprocmask");

952

891

}

959

898

960

899

/* Unblock SIGCHLD so signal handler can be run if this process

961

900

has already completed */

962

ret = (int)TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,

963

&sigchld_action.sa_mask,

964

NULL));

901

ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask, NULL);

965

902

if(ret < 0){

966

903

perror("sigprocmask");

967

904

exitstatus = EXIT_FAILURE;

968

905

goto fallback;

969

906

}

970

907

971

FD_SET(new_plugin->fd, &rfds_all); /* Spurious warning from

972

-Wconversion */

908

FD_SET(new_plugin->fd, &rfds_all);

973

909

974

if(maxfd < new_plugin->fd){

910

if (maxfd < new_plugin->fd){

975

911

maxfd = new_plugin->fd;

976

912

}

913

977

914

}

978

915

979

TEMP_FAILURE_RETRY(closedir(dir));

916

closedir(dir);

980

917

dir = NULL;

981

free_plugin(getplugin(NULL));

982

918

983

919

for(plugin *p = plugin_list; p != NULL; p = p->next){

984

920

if(p->pid != 0){

985

921

break;

990

926

free_plugin_list();

991

927

}

992

928

}

993

929

994

930

/* Main loop while running plugins exist */

995

931

while(plugin_list){

996

932

fd_set rfds = rfds_all;

997

933

int select_ret = select(maxfd+1, &rfds, NULL, NULL, NULL);

998

if(select_ret == -1 and errno != EINTR){

934

if (select_ret == -1){

999

935

perror("select");

1000

936

exitstatus = EXIT_FAILURE;

1001

937

goto fallback;

1004

940

from one of them */

1005

941

for(plugin *proc = plugin_list; proc != NULL;){

1006

942

/* Is this process completely done? */

1007

if(proc->completed and proc->eof){

943

if(proc->eof and proc->completed){

1008

944

/* Only accept the plugin output if it exited cleanly */

1009

945

if(not WIFEXITED(proc->status)

1010

946

or WEXITSTATUS(proc->status) != 0){

1011

947

/* Bad exit by plugin */

1012

948

1013

949

if(debug){

1014

950

if(WIFEXITED(proc->status)){

1015

fprintf(stderr, "Plugin %s [%" PRIdMAX "] exited with"

1016

" status %d\n", proc->name,

1017

(intmax_t) (proc->pid),

951

fprintf(stderr, "Plugin %u exited with status %d\n",

952

(unsigned int) (proc->pid),

1018

953

WEXITSTATUS(proc->status));

1019

} else if(WIFSIGNALED(proc->status)){

1020

fprintf(stderr, "Plugin %s [%" PRIdMAX "] killed by"

1021

" signal %d: %s\n", proc->name,

1022

(intmax_t) (proc->pid),

1023

WTERMSIG(proc->status),

1024

strsignal(WTERMSIG(proc->status)));

954

} else if(WIFSIGNALED(proc->status)) {

955

fprintf(stderr, "Plugin %u killed by signal %d\n",

956

(unsigned int) (proc->pid),

957

WTERMSIG(proc->status));

1025

958

} else if(WCOREDUMP(proc->status)){

1026

fprintf(stderr, "Plugin %s [%" PRIdMAX "] dumped"

1027

" core\n", proc->name, (intmax_t) (proc->pid));

959

fprintf(stderr, "Plugin %d dumped core\n",

960

(unsigned int) (proc->pid));

1028

961

}

1029

962

}

1030

963

1031

964

/* Remove the plugin */

1032

FD_CLR(proc->fd, &rfds_all); /* Spurious warning from

1033

-Wconversion */

1034

965

FD_CLR(proc->fd, &rfds_all);

966

1035

967

/* Block signal while modifying process_list */

1036

ret = (int)TEMP_FAILURE_RETRY(sigprocmask

1037

(SIG_BLOCK,

1038

&sigchld_action.sa_mask,

1039

NULL));

1040

if(ret < 0){

1041

perror("sigprocmask");

1042

exitstatus = EXIT_FAILURE;

1043

goto fallback;

968

ret = sigprocmask(SIG_BLOCK, &sigchld_action.sa_mask, NULL);

969

if(ret < 0){

970

perror("sigprocmask");

971

exitstatus = EXIT_FAILURE;

972

goto fallback;

973

}

974

975

/* We are done modifying process list, so unblock signal */

976

ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,

977

NULL);

978

if(ret < 0){

979

perror("sigprocmask");

980

exitstatus = EXIT_FAILURE;

981

goto fallback;

982

}

983

984

if(plugin_list == NULL){

985

break;

1044

986

}

1045

987

1046

988

plugin *next_plugin = proc->next;

1047

989

free_plugin(proc);

1048

990

proc = next_plugin;

1049

1050

/* We are done modifying process list, so unblock signal */

1051

ret = (int)(TEMP_FAILURE_RETRY

1052

(sigprocmask(SIG_UNBLOCK,

1053

&sigchld_action.sa_mask, NULL)));

1054

if(ret < 0){

1055

perror("sigprocmask");

1056

exitstatus = EXIT_FAILURE;

1057

goto fallback;

1058

}

1059

1060

if(plugin_list == NULL){

1061

break;

1062

}

1063

1064

991

continue;

1065

992

}

1066

993

1067

994

/* This process exited nicely, so print its buffer */

1068

995

1069

996

bool bret = print_out_password(proc->buffer,

1070

997

proc->buffer_length);

1071

998

if(not bret){

1076

1003

}

1077

1004

1078

1005

/* This process has not completed. Does it have any output? */

1079

if(proc->eof or not FD_ISSET(proc->fd, &rfds)){ /* Spurious

1080

warning from

1081

-Wconversion */

1006

if(proc->eof or not FD_ISSET(proc->fd, &rfds)){

1082

1007

/* This process had nothing to say at this time */

1083

1008

proc = proc->next;

1084

1009

continue;

1087

1012

if(proc->buffer_length + BUFFER_SIZE > proc->buffer_size){

1088

1013

proc->buffer = realloc(proc->buffer, proc->buffer_size

1089

1014

+ (size_t) BUFFER_SIZE);

1090

if(proc->buffer == NULL){

1015

if (proc->buffer == NULL){

1091

1016

perror("malloc");

1092

1017

exitstatus = EXIT_FAILURE;

1093

1018

goto fallback;

1095

1020

proc->buffer_size += BUFFER_SIZE;

1096

1021

}

1097

1022

/* Read from the process */

1098

sret = TEMP_FAILURE_RETRY(read(proc->fd,

1099

proc->buffer

1100

+ proc->buffer_length,

1101

BUFFER_SIZE));

1102

if(sret < 0){

1023

ret = read(proc->fd, proc->buffer + proc->buffer_length,

1024

BUFFER_SIZE);

1025

if(ret < 0){

1103

1026

/* Read error from this process; ignore the error */

1104

1027

proc = proc->next;

1105

1028

continue;

1106

1029

}

1107

if(sret == 0){

1030

if(ret == 0){

1108

1031

/* got EOF */

1109

1032

proc->eof = true;

1110

1033

} else {

1111

proc->buffer_length += (size_t) sret;

1034

proc->buffer_length += (size_t) ret;

1112

1035

}

1113

1036

}

1114

1037

}

1115

1116

1038

1039

1117

1040

fallback:

1118

1041

1119

1042

if(plugin_list == NULL or exitstatus != EXIT_SUCCESS){

1141

1064

perror("sigaction");

1142

1065

exitstatus = EXIT_FAILURE;

1143

1066

}

1144

1067

1145

1068

if(custom_argv != NULL){

1146

1069

for(char **arg = custom_argv+1; *arg != NULL; arg++){

1147

1070

free(*arg);

1153

1076

closedir(dir);

1154

1077

}

1155

1078

1156

/* Kill the processes */

1079

/* Free the process list and kill the processes */

1157

1080

for(plugin *p = plugin_list; p != NULL; p = p->next){

1158

1081

if(p->pid != 0){

1159

1082

close(p->fd);

1166

1089

}

1167

1090

1168

1091

/* Wait for any remaining child processes to terminate */

1169

do {

1092

do{

1170

1093

ret = wait(NULL);

1171

1094

} while(ret >= 0);

1172

1095

if(errno != ECHILD){

1173

1096

perror("wait");

1174

1097

}

1175

1098

1176

1099

free_plugin_list();

1177

1100

1178

1101

free(plugindir);