/mandos/trunk : revision 24.1.78

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

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

100

97

static plugin *plugin_list = NULL;

101

98

102

/* Gets an existing plugin based on name,

99

/* Gets a existing plugin based on name,

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

131

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

132

.argc = 1,

133

.disabled = false,

134

.next = plugin_list };

135

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

121

122

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

123

.argc = 1,

124

.disabled = false,

125

.next = plugin_list };

126

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

}

197

179

}

198

180

199

181

/* Add to a plugin's environment */

200

static bool add_environment(plugin *p, const char *def, bool replace){

182

static bool add_environment(plugin *p, const char *def){

201

183

if(p == NULL){

202

184

return false;

203

185

}

205

187

size_t namelen = (size_t)(strchrnul(def, '=') - def);

206

188

/* Search for this environment variable */

207

189

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

208

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

209

/* It already exists */

210

if(replace){

211

char *new;

212

do{

213

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

214

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

215

if(new == NULL){

216

return false;

217

}

218

*e = new;

219

strcpy(*e, def);

220

}

190

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

191

/* Refuse to add an existing variable */

221

192

return true;

222

193

}

223

194

}

227

198

/*

228

199

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

229

200

* Descriptor Flags".

230

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

201

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

231

202

*/

232

static int set_cloexec_flag(int fd){

233

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

203

static int set_cloexec_flag(int fd)

204

{

205

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

234

206

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

235

207

if(ret < 0){

236

208

return ret;

237

209

}

238

210

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

239

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

211

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

240

212

}

241

213

242

214

243

215

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

244

216

status. */

245

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

246

int old_errno = errno;

217

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

247

218

while(true){

248

219

plugin *proc = plugin_list;

249

220

int status;

253

224

break;

254

225

}

255

226

if(pid == -1){

256

if(errno == ECHILD){

257

/* No child processes */

258

break;

227

if (errno != ECHILD){

228

perror("waitpid");

259

229

}

260

perror("waitpid");

230

/* No child processes */

231

break;

261

232

}

262

233

263

234

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

264

235

while(proc != NULL and proc->pid != pid){

265

236

proc = proc->next;

269

240

continue;

270

241

}

271

242

proc->status = status;

272

proc->completed = 1;

243

proc->completed = true;

273

244

}

274

errno = old_errno;

275

245

}

276

246

277

247

/* Prints out a password to stdout */

278

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

248

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

279

249

ssize_t ret;

250

if(length>0 and buffer[length-1] == '\n'){

251

length--;

252

}

280

253

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

281

254

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

282

255

length - written));

299

272

}

300

273

free(plugin_node->environ);

301

274

free(plugin_node->buffer);

302

275

303

276

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

304

277

if(plugin_node == plugin_list){

305

278

/* First one - simple */

333

306

struct stat st;

334

307

fd_set rfds_all;

335

308

int ret, maxfd = 0;

336

ssize_t sret;

337

309

uid_t uid = 65534;

338

310

gid_t gid = 65534;

339

311

bool debug = false;

364

336

{ .name = "global-options", .key = 'g',

365

337

.arg = "OPTION[,OPTION[,...]]",

366

338

.doc = "Options passed to all plugins" },

367

{ .name = "global-env", .key = 'G',

339

{ .name = "global-env", .key = 'e',

368

340

.arg = "VAR=value",

369

341

.doc = "Environment variable passed to all plugins" },

370

342

{ .name = "options-for", .key = 'o',

371

343

.arg = "PLUGIN:OPTION[,OPTION[,...]]",

372

344

.doc = "Options passed only to specified plugin" },

373

{ .name = "env-for", .key = 'E',

345

{ .name = "env-for", .key = 'f',

374

346

.arg = "PLUGIN:ENV=value",

375

347

.doc = "Environment variable passed to specified plugin" },

376

348

{ .name = "disable", .key = 'd',

377

349

.arg = "PLUGIN",

378

350

.doc = "Disable a specific plugin", .group = 1 },

379

{ .name = "enable", .key = 'e',

380

.arg = "PLUGIN",

381

.doc = "Enable a specific plugin", .group = 1 },

382

351

{ .name = "plugin-dir", .key = 128,

383

352

.arg = "DIRECTORY",

384

353

.doc = "Specify a different plugin directory", .group = 2 },

396

365

{ .name = NULL }

397

366

};

398

367

399

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

400

struct argp_state *state){

401

switch(key){

402

char *tmp;

403

intmax_t tmpmax;

368

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

369

struct argp_state *state) {

370

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

371

pointer to our plugin list pointer. */

372

switch (key) {

404

373

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

405

if(arg != NULL){

406

char *plugin_option;

407

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

408

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

374

if (arg != NULL){

375

char *p;

376

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

377

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

409

378

continue;

410

379

}

411

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

380

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

412

381

perror("add_argument");

413

382

return ARGP_ERR_UNKNOWN;

414

383

}

415

384

}

416

385

}

417

386

break;

418

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

387

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

419

388

if(arg == NULL){

420

389

break;

421

390

}

422

if(not add_environment(getplugin(NULL), arg, true)){

423

perror("add_environment");

391

{

392

char *envdef = strdup(arg);

393

if(envdef == NULL){

394

break;

395

}

396

if(not add_environment(getplugin(NULL), envdef)){

397

perror("add_environment");

398

}

424

399

}

425

400

break;

426

401

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

427

if(arg != NULL){

428

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

429

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

430

break;

431

}

432

char *plugin_option;

433

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

434

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

435

perror("add_argument");

436

return ARGP_ERR_UNKNOWN;

402

if (arg != NULL){

403

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

404

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

405

break;

406

}

407

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

408

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

409

break;

410

}

411

if(opt != NULL){

412

char *p;

413

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

414

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

415

continue;

416

}

417

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

418

perror("add_argument");

419

return ARGP_ERR_UNKNOWN;

420

}

437

421

}

438

422

}

439

423

}

440

424

break;

441

case 'E': /* --env-for */

425

case 'f': /* --env-for */

442

426

if(arg == NULL){

443

427

break;

444

428

}

447

431

if(envdef == NULL){

448

432

break;

449

433

}

450

*envdef = '\0';

451

if(not add_environment(getplugin(arg), envdef+1, true)){

434

char *p_name = strndup(arg, (size_t) (envdef-arg));

435

if(p_name == NULL){

436

break;

437

}

438

envdef++;

439

if(not add_environment(getplugin(p_name), envdef)){

452

440

perror("add_environment");

453

441

}

454

442

}

455

443

break;

456

444

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

457

if(arg != NULL){

445

if (arg != NULL){

458

446

plugin *p = getplugin(arg);

459

447

if(p == NULL){

460

448

return ARGP_ERR_UNKNOWN;

462

450

p->disabled = true;

463

451

}

464

452

break;

465

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

466

if(arg != NULL){

467

plugin *p = getplugin(arg);

468

if(p == NULL){

469

return ARGP_ERR_UNKNOWN;

470

}

471

p->disabled = false;

472

}

473

break;

474

453

case 128: /* --plugin-dir */

475

free(plugindir);

476

454

plugindir = strdup(arg);

477

455

if(plugindir == NULL){

478

456

perror("strdup");

479

}

480

break;

481

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

482

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

483

break;

484

case 130: /* --userid */

485

errno = 0;

486

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

487

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

488

or tmpmax != (uid_t)tmpmax){

489

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

490

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

491

} else {

492

uid = (uid_t)tmpmax;

493

}

494

break;

495

case 131: /* --groupid */

496

errno = 0;

497

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

498

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

499

or tmpmax != (gid_t)tmpmax){

500

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

501

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

502

} else {

503

gid = (gid_t)tmpmax;

504

}

505

break;

506

case 132: /* --debug */

507

debug = true;

508

break;

509

/*

510

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

511

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

512

*/

513

case ARGP_KEY_ARG:

514

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

515

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

516

argument was empty, we ignore it silently. */

517

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

518

fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);

519

}

520

break;

521

case ARGP_KEY_END:

522

break;

523

default:

524

return ARGP_ERR_UNKNOWN;

525

}

526

return 0;

527

}

528

529

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

530

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

531

error_t parse_opt_config_file(int key, char *arg,

532

__attribute__((unused))

533

struct argp_state *state){

534

switch(key){

535

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

536

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

537

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

538

case 'E': /* --env-for */

539

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

540

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

541

case 128: /* --plugin-dir */

542

break;

543

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

544

free(argfile);

457

}

458

break;

459

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

545

460

argfile = strdup(arg);

546

461

if(argfile == NULL){

547

462

perror("strdup");

548

463

}

549

break;

464

break;

550

465

case 130: /* --userid */

466

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

467

break;

551

468

case 131: /* --groupid */

469

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

470

break;

552

471

case 132: /* --debug */

472

debug = true;

473

break;

553

474

case ARGP_KEY_ARG:

475

fprintf(stderr, "Ignoring unknown argument \"%s\"\n", arg);

476

break;

554

477

case ARGP_KEY_END:

555

478

break;

556

479

default:

559

482

return 0;

560

483

}

561

484

562

struct argp argp = { .options = options,

563

.parser = parse_opt_config_file,

564

.args_doc = "",

485

struct argp argp = { .options = options, .parser = parse_opt,

486

.args_doc = "[+PLUS_SEPARATED_OPTIONS]",

565

487

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

566

488

567

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

568

config file location, if any. */

569

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

570

if(ret == ARGP_ERR_UNKNOWN){

489

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

490

if (ret == ARGP_ERR_UNKNOWN){

571

491

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

572

492

exitstatus = EXIT_FAILURE;

573

493

goto fallback;

574

494

}

575

576

/* Reset to the normal argument parser */

577

argp.parser = parse_opt;

578

579

/* Open the configfile if available */

580

if(argfile == NULL){

495

496

/* Opens the configfile if aviable */

497

if (argfile == NULL){

581

498

conffp = fopen(AFILE, "r");

582

499

} else {

583

500

conffp = fopen(argfile, "r");

584

}

501

}

585

502

if(conffp != NULL){

586

503

char *org_line = NULL;

587

504

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

588

505

size_t size = 0;

506

ssize_t sret;

589

507

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

590

508

const char comment_delim[] = "#";

591

509

592

510

custom_argc = 1;

593

511

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

594

512

if(custom_argv == NULL){

598

516

}

599

517

custom_argv[0] = argv[0];

600

518

custom_argv[1] = NULL;

601

519

602

520

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

603

521

commented text */

604

522

while(true){

606

524

if(sret == -1){

607

525

break;

608

526

}

609

527

610

528

line = org_line;

611

529

arg = strsep(&line, comment_delim);

612

530

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

631

549

goto fallback;

632

550

}

633

551

custom_argv[custom_argc-1] = new_arg;

634

custom_argv[custom_argc] = NULL;

552

custom_argv[custom_argc] = NULL;

635

553

}

636

554

}

637

do{

638

ret = fclose(conffp);

639

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

640

if(ret == EOF){

641

perror("fclose");

642

exitstatus = EXIT_FAILURE;

643

goto fallback;

644

}

645

555

free(org_line);

646

} else {

556

} else{

647

557

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

648

558

not affect opening plugins */

649

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

559

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

650

560

perror("fopen");

651

561

exitstatus = EXIT_FAILURE;

652

562

goto fallback;

655

565

/* If there was any arguments from configuration file,

656

566

pass them to parser as command arguments */

657

567

if(custom_argv != NULL){

658

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

659

0, NULL);

660

if(ret == ARGP_ERR_UNKNOWN){

568

ret = argp_parse (&argp, custom_argc, custom_argv, 0, 0, NULL);

569

if (ret == ARGP_ERR_UNKNOWN){

661

570

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

662

571

exitstatus = EXIT_FAILURE;

663

572

goto fallback;

664

573

}

665

574

}

666

575

667

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

668

config file */

669

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

670

if(ret == ARGP_ERR_UNKNOWN){

671

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

672

exitstatus = EXIT_FAILURE;

673

goto fallback;

674

}

675

676

576

if(debug){

677

577

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

678

578

fprintf(stderr, "Plugin: %s has %d arguments\n",

680

580

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

681

581

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

682

582

}

683

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

583

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

684

584

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

685

585

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

686

586

}

687

587

}

688

588

}

689

589

690

590

/* Strip permissions down to nobody */

591

ret = setuid(uid);

592

if (ret == -1){

593

perror("setuid");

594

}

691

595

setgid(gid);

692

if(ret == -1){

596

if (ret == -1){

693

597

perror("setgid");

694

598

}

695

ret = setuid(uid);

696

if(ret == -1){

697

perror("setuid");

698

}

699

700

if(plugindir == NULL){

599

600

if (plugindir == NULL){

701

601

dir = opendir(PDIR);

702

602

} else {

703

603

dir = opendir(plugindir);

723

623

}

724

624

725

625

FD_ZERO(&rfds_all);

726

626

727

627

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

728

628

while(true){

729

do{

730

dirst = readdir(dir);

731

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

629

dirst = readdir(dir);

732

630

733

/* All directory entries have been processed */

631

// All directory entries have been processed

734

632

if(dirst == NULL){

735

if(errno == EBADF){

633

if (errno == EBADF){

736

634

perror("readdir");

737

635

exitstatus = EXIT_FAILURE;

738

636

goto fallback;

742

640

743

641

d_name_len = strlen(dirst->d_name);

744

642

745

/* Ignore dotfiles, backup files and other junk */

643

// Ignore dotfiles, backup files and other junk

746

644

{

747

645

bool bad_name = false;

748

646

750

648

751

649

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

752

650

".dpkg-old",

753

".dpkg-bak",

754

651

".dpkg-divert", NULL };

755

652

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

756

653

size_t pre_len = strlen(*pre);

785

682

continue;

786

683

}

787

684

}

788

685

789

686

char *filename;

790

if(plugindir == NULL){

791

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

792

dirst->d_name));

793

} else {

794

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

795

dirst->d_name));

796

}

687

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

797

688

if(ret < 0){

798

689

perror("asprintf");

799

690

continue;

800

691

}

801

692

802

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

803

if(ret == -1){

693

ret = stat(filename, &st);

694

if (ret == -1){

804

695

perror("stat");

805

696

free(filename);

806

697

continue;

807

698

}

808

699

809

700

/* Ignore non-executable files */

810

if(not S_ISREG(st.st_mode)

811

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

701

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

812

702

if(debug){

813

703

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

814

704

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

842

732

}

843

733

/* Add global environment variables */

844

734

for(char **e = g->environ; *e != NULL; e++){

845

if(not add_environment(p, *e, false)){

735

if(not add_environment(p, *e)){

846

736

perror("add_environment");

847

737

}

848

738

}

853

743

process, too. */

854

744

if(p->environ[0] != NULL){

855

745

for(char **e = environ; *e != NULL; e++){

856

if(not add_environment(p, *e, false)){

746

char *copy = strdup(*e);

747

if(copy == NULL){

748

perror("strdup");

749

continue;

750

}

751

if(not add_environment(p, copy)){

857

752

perror("add_environment");

858

753

}

859

754

}

860

755

}

861

756

862

757

int pipefd[2];

863

ret = TEMP_FAILURE_RETRY(pipe(pipefd));

864

if(ret == -1){

758

ret = pipe(pipefd);

759

if (ret == -1){

865

760

perror("pipe");

866

761

exitstatus = EXIT_FAILURE;

867

762

goto fallback;

880

775

goto fallback;

881

776

}

882

777

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

883

ret = TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,

884

&sigchld_action.sa_mask,

885

NULL));

778

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

886

779

if(ret < 0){

887

780

perror("sigprocmask");

888

781

exitstatus = EXIT_FAILURE;

889

782

goto fallback;

890

783

}

891

/* Starting a new process to be watched */

892

pid_t pid;

893

do{

894

pid = fork();

895

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

784

// Starting a new process to be watched

785

pid_t pid = fork();

896

786

if(pid == -1){

897

787

perror("fork");

898

788

exitstatus = EXIT_FAILURE;

905

795

perror("sigaction");

906

796

_exit(EXIT_FAILURE);

907

797

}

908

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

798

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

909

799

if(ret < 0){

910

800

perror("sigprocmask");

911

801

_exit(EXIT_FAILURE);

912

802

}

913

803

914

804

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

915

805

if(ret == -1){

916

806

perror("dup2");

936

826

/* no return */

937

827

}

938

828

/* Parent process */

939

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

940

pipe */

829

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

941

830

free(filename);

942

831

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

943

if(new_plugin == NULL){

832

if (new_plugin == NULL){

944

833

perror("getplugin");

945

ret = TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,

946

&sigchld_action.sa_mask,

947

NULL));

834

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

948

835

if(ret < 0){

949

836

perror("sigprocmask");

950

837

}

957

844

958

845

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

959

846

has already completed */

960

ret = TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,

961

&sigchld_action.sa_mask,

962

NULL));

847

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

963

848

if(ret < 0){

964

849

perror("sigprocmask");

965

850

exitstatus = EXIT_FAILURE;

968

853

969

854

FD_SET(new_plugin->fd, &rfds_all);

970

855

971

if(maxfd < new_plugin->fd){

856

if (maxfd < new_plugin->fd){

972

857

maxfd = new_plugin->fd;

973

858

}

859

974

860

}

975

861

976

TEMP_FAILURE_RETRY(closedir(dir));

862

closedir(dir);

977

863

dir = NULL;

978

free_plugin(getplugin(NULL));

979

864

980

865

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

981

866

if(p->pid != 0){

982

867

break;

987

872

free_plugin_list();

988

873

}

989

874

}

990

875

991

876

/* Main loop while running plugins exist */

992

877

while(plugin_list){

993

878

fd_set rfds = rfds_all;

994

879

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

995

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

880

if (select_ret == -1){

996

881

perror("select");

997

882

exitstatus = EXIT_FAILURE;

998

883

goto fallback;

999

884

}

1000

885

/* OK, now either a process completed, or something can be read

1001

886

from one of them */

1002

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

887

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

1003

888

/* Is this process completely done? */

1004

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

889

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

1005

890

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

1006

891

if(not WIFEXITED(proc->status)

1007

892

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

1008

893

/* Bad exit by plugin */

1009

894

1010

895

if(debug){

1011

896

if(WIFEXITED(proc->status)){

1012

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

1013

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

1014

(intmax_t) (proc->pid),

897

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

898

(unsigned int) (proc->pid),

1015

899

WEXITSTATUS(proc->status));

1016

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

1017

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

1018

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

1019

(intmax_t) (proc->pid),

1020

WTERMSIG(proc->status),

1021

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

900

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

901

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

902

(unsigned int) (proc->pid),

903

WTERMSIG(proc->status));

1022

904

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

1023

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

1024

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

905

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

906

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

1025

907

}

1026

908

}

1027

909

1028

910

/* Remove the plugin */

1029

911

FD_CLR(proc->fd, &rfds_all);

1030

912

1031

913

/* Block signal while modifying process_list */

1032

ret = TEMP_FAILURE_RETRY(sigprocmask(SIG_BLOCK,

1033

&sigchld_action.sa_mask,

1034

NULL));

914

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

1035

915

if(ret < 0){

1036

916

perror("sigprocmask");

1037

917

exitstatus = EXIT_FAILURE;

1038

918

goto fallback;

1039

919

}

1040

1041

plugin *next_plugin = proc->next;

1042

920

free_plugin(proc);

1043

proc = next_plugin;

1044

1045

921

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

1046

ret = TEMP_FAILURE_RETRY(sigprocmask(SIG_UNBLOCK,

1047

&sigchld_action.sa_mask,

1048

NULL));

922

ret = sigprocmask (SIG_UNBLOCK, &sigchld_action.sa_mask,

923

NULL);

1049

924

if(ret < 0){

1050

925

perror("sigprocmask");

1051

926

exitstatus = EXIT_FAILURE;

1055

930

if(plugin_list == NULL){

1056

931

break;

1057

932

}

1058

1059

933

continue;

1060

934

}

1061

935

1062

936

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

1063

937

1064

938

bool bret = print_out_password(proc->buffer,

1065

939

proc->buffer_length);

1066

940

if(not bret){

1073

947

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

1074

948

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

1075

949

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

1076

proc = proc->next;

1077

950

continue;

1078

951

}

1079

952

/* Before reading, make the process' data buffer large enough */

1080

953

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

1081

954

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

1082

955

+ (size_t) BUFFER_SIZE);

1083

if(proc->buffer == NULL){

956

if (proc->buffer == NULL){

1084

957

perror("malloc");

1085

958

exitstatus = EXIT_FAILURE;

1086

959

goto fallback;

1088

961

proc->buffer_size += BUFFER_SIZE;

1089

962

}

1090

963

/* Read from the process */

1091

sret = TEMP_FAILURE_RETRY(read(proc->fd,

1092

proc->buffer

1093

+ proc->buffer_length,

1094

BUFFER_SIZE));

1095

if(sret < 0){

964

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

965

BUFFER_SIZE);

966

if(ret < 0){

1096

967

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

1097

proc = proc->next;

1098

968

continue;

1099

969

}

1100

if(sret == 0){

970

if(ret == 0){

1101

971

/* got EOF */

1102

972

proc->eof = true;

1103

973

} else {

1104

proc->buffer_length += (size_t) sret;

974

proc->buffer_length += (size_t) ret;

1105

975

}

1106

976

}

1107

977

}

1108

1109

978

979

1110

980

fallback:

1111

981

1112

982

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

1115

985

bool bret;

1116

986

fprintf(stderr, "Going to fallback mode using getpass(3)\n");

1117

987

char *passwordbuffer = getpass("Password: ");

1118

size_t len = strlen(passwordbuffer);

1119

/* Strip trailing newline */

1120

if(len > 0 and passwordbuffer[len-1] == '\n'){

1121

passwordbuffer[len-1] = '\0'; /* not strictly necessary */

1122

len--;

1123

}

1124

bret = print_out_password(passwordbuffer, len);

988

bret = print_out_password(passwordbuffer, strlen(passwordbuffer));

1125

989

if(not bret){

1126

990

perror("print_out_password");

1127

991

exitstatus = EXIT_FAILURE;

1134

998

perror("sigaction");

1135

999

exitstatus = EXIT_FAILURE;

1136

1000

}

1137

1001

1138

1002

if(custom_argv != NULL){

1139

1003

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

1140

1004

free(*arg);

1146

1010

closedir(dir);

1147

1011

}

1148

1012

1149

/* Kill the processes */

1013

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

1150

1014

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

1151

1015

if(p->pid != 0){

1152

1016

close(p->fd);

1165

1029

if(errno != ECHILD){

1166

1030

perror("wait");

1167

1031

}

1168

1032

1169

1033

free_plugin_list();

1170

1034

1171

1035

free(plugindir);