/mandos/trunk : revision 24.1.7

21

* Contact the authors at <mandos@fukt.bsnet.se>.

22

*/

23

24

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

25

asprintf() */

26

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

27

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

28

EXIT_SUCCESS, realloc() */

29

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

30

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

31

fprintf(), stderr, STDOUT_FILENO */

32

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

33

stat, waitpid(), WIFEXITED(),

34

WEXITSTATUS(), wait(), pid_t,

35

uid_t, gid_t, getuid(), getgid(),

36

dirfd() */

37

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

38

FD_SET(), FD_ISSET(), FD_CLR */

39

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

40

WEXITSTATUS() */

41

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

24

#define _GNU_SOURCE /* TEMP_FAILURE_RETRY() */

25

26

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

27

stderr, STDOUT_FILENO */

42

28

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

29

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

30

struct stat, waitpid(),

31

WIFEXITED(), WEXITSTATUS(),

32

wait() */

33

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

43

34

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

44

readdir(), closedir(), dirfd() */

35

readdir(), closedir() */

36

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

45

37

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

46

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

47

F_GETFD, F_SETFD, FD_CLOEXEC,

48

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

49

dup2, STDOUT_FILENO, _exit(),

50

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

51

close() */

52

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

53

FD_CLOEXEC */

54

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

38

fcntl() */

39

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

40

#include <stddef.h> /* NULL */

41

#include <stdlib.h> /* EXIT_FAILURE */

42

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

43

FD_SET(), FD_ISSET() */

44

#include <string.h> /* strlen(), strcpy(), strcat() */

45

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

46

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

47

WEXITSTATUS() */

55

48

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

56

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

57

argp_state, struct argp,

58

argp_parse(), ARGP_ERR_UNKNOWN,

59

ARGP_KEY_END, ARGP_KEY_ARG,

60

error_t */

61

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

62

sigaddset(), sigaction(),

63

sigprocmask(), SIG_BLOCK, SIGCHLD,

64

SIG_UNBLOCK, kill() */

65

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

66

67

#define BUFFER_SIZE 256

68

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

69

70

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

71

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

49

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

50

struct argp_state, struct argp,

51

argp_parse() */

72

52

73

53

struct process;

74

54

79

59

size_t buffer_size;

80

60

size_t buffer_length;

81

61

bool eof;

82

volatile bool completed;

83

volatile int status;

62

bool completed;

63

int status;

84

64

struct process *next;

85

65

} process;

86

66

88

68

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

89

69

char **argv;

90

70

int argc;

91

char **environ;

92

int envc;

93

71

bool disabled;

94

72

struct plugin *next;

95

73

} plugin;

96

74

97

static plugin *getplugin(char *name, plugin **plugin_list){

75

plugin *getplugin(char *name, plugin **plugin_list){

98

76

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

99

77

if ((p->name == name)

100

78

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

104

82

/* Create a new plugin */

105

83

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

106

84

if (new_plugin == NULL){

107

return NULL;

108

}

109

char *copy_name = NULL;

110

if(name != NULL){

111

copy_name = strdup(name);

112

if(copy_name == NULL){

113

return NULL;

114

}

115

}

116

117

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

118

.argc = 1,

119

.envc = 0,

120

.disabled = false,

121

.next = *plugin_list };

122

85

perror("malloc");

86

exit(EXIT_FAILURE);

87

}

88

new_plugin->name = name;

123

89

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

124

90

if (new_plugin->argv == NULL){

125

free(copy_name);

126

free(new_plugin);

127

return NULL;

91

perror("malloc");

92

exit(EXIT_FAILURE);

128

93

}

129

new_plugin->argv[0] = copy_name;

94

new_plugin->argv[0] = name;

130

95

new_plugin->argv[1] = NULL;

131

132

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

133

if(new_plugin->environ == NULL){

134

free(copy_name);

135

free(new_plugin->argv);

136

free(new_plugin);

137

return NULL;

138

}

139

new_plugin->environ[0] = NULL;

96

new_plugin->argc = 1;

97

new_plugin->disabled = false;

98

new_plugin->next = *plugin_list;

140

99

/* Append the new plugin to the list */

141

100

*plugin_list = new_plugin;

142

101

return new_plugin;

143

102

}

144

103

145

/* Helper function for add_argument and add_environment */

146

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

147

int *len){

148

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

149

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

150

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

151

/* Malloc check */

152

if(*array == NULL){

153

return false;

154

}

155

/* Make a copy of the new string */

156

char *copy = strdup(new);

157

if(copy == NULL){

158

return false;

159

}

160

/* Insert the copy */

161

(*array)[*len] = copy;

162

(*len)++;

163

/* Add a new terminating NULL pointer to the last element */

164

(*array)[*len] = NULL;

165

return true;

166

}

167

168

/* Add to a plugin's argument vector */

169

static bool add_argument(plugin *p, const char *arg){

170

if(p == NULL){

171

return false;

172

}

173

return add_to_char_array(arg, &(p->argv), &(p->argc));

174

}

175

176

/* Add to a plugin's environment */

177

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

178

if(p == NULL){

179

return false;

180

}

181

return add_to_char_array(def, &(p->environ), &(p->envc));

182

}

183

104

void addargument(plugin *p, char *arg){

105

p->argv[p->argc] = arg;

106

p->argv = realloc(p->argv, sizeof(char *) * (size_t)(p->argc + 2));

107

if (p->argv == NULL){

108

perror("malloc");

109

exit(EXIT_FAILURE);

110

}

111

p->argc++;

112

p->argv[p->argc] = NULL;

113

}

184

114

185

115

/*

186

116

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

187

117

* Descriptor Flags".

188

118

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

189

119

*/

190

static int set_cloexec_flag(int fd)

120

int set_cloexec_flag(int fd)

191

121

{

192

122

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

193

123

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

198

128

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

199

129

}

200

130

131

#define BUFFER_SIZE 256

132

133

const char *argp_program_version = "plugbasedclient 0.9";

134

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

135

201

136

process *process_list = NULL;

202

137

203

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

138

/* Mark a process as completed when it exits, and save its exit

204

139

status. */

205

140

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

206

while(true){

207

process *proc = process_list;

208

int status;

209

pid_t pid = waitpid(-1, &status, WNOHANG);

210

if(pid == 0){

211

/* Only still running child processes */

212

break;

213

}

214

if(pid == -1){

215

if (errno != ECHILD){

216

perror("waitpid");

217

}

218

/* No child processes */

219

break;

220

}

221

222

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

223

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

224

proc = proc->next;

225

}

226

if(proc == NULL){

227

/* Process not found in process list */

228

continue;

229

}

230

proc->status = status;

231

proc->completed = true;

232

}

233

}

234

235

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

236

ssize_t ret;

237

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

238

length--;

239

}

240

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

241

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

242

length - written));

243

if(ret < 0){

244

return false;

245

}

246

}

247

return true;

248

}

249

250

char **add_to_argv(char **argv, int *argc, char *arg){

251

if (argv == NULL){

252

*argc = 1;

253

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

254

if(argv == NULL){

255

return NULL;

256

}

257

argv[0] = NULL; /* Will be set to argv[0] in main before

258

parsing */

259

argv[1] = NULL;

260

}

261

*argc += 1;

262

argv = realloc(argv, sizeof(char *)

263

* ((unsigned int) *argc + 1));

264

if(argv == NULL){

265

return NULL;

266

}

267

argv[*argc-1] = arg;

268

argv[*argc] = NULL;

269

return argv;

270

}

271

272

static void free_plugin_list(plugin *plugin_list){

273

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

274

next = plugin_list->next;

275

for(char **arg = plugin_list->argv; *arg != NULL; arg++){

276

free(*arg);

277

}

278

free(plugin_list->argv);

279

for(char **env = plugin_list->environ; *env != NULL; env++){

280

free(*env);

281

}

282

free(plugin_list->environ);

283

free(plugin_list);

284

}

141

process *proc = process_list;

142

int status;

143

pid_t pid = wait(&status);

144

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

145

proc = proc->next;

146

}

147

if(proc == NULL){

148

/* Process not found in process list */

149

return;

150

}

151

proc->status = status;

152

proc->completed = true;

285

153

}

286

154

287

155

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

288

const char *plugindir = "/lib/mandos/plugins.d";

289

const char *argfile = ARGFILE;

290

FILE *conffp;

156

const char *plugindir = "/conf/conf.d/mandos/plugins.d";

291

157

size_t d_name_len;

292

158

DIR *dir = NULL;

293

159

struct dirent *dirst;

301

167

struct sigaction old_sigchld_action;

302

168

struct sigaction sigchld_action = { .sa_handler = handle_sigchld,

303

169

.sa_flags = SA_NOCLDSTOP };

304

char **custom_argv = NULL;

305

int custom_argc = 0;

306

170

307

171

/* Establish a signal handler */

308

172

sigemptyset(&sigchld_action.sa_mask);

309

173

ret = sigaddset(&sigchld_action.sa_mask, SIGCHLD);

310

if(ret == -1){

174

if(ret < 0){

311

175

perror("sigaddset");

312

exitstatus = EXIT_FAILURE;

313

goto fallback;

176

exit(EXIT_FAILURE);

314

177

}

315

178

ret = sigaction(SIGCHLD, &sigchld_action, &old_sigchld_action);

316

if(ret == -1){

179

if(ret < 0){

317

180

perror("sigaction");

318

exitstatus = EXIT_FAILURE;

319

goto fallback;

181

exit(EXIT_FAILURE);

320

182

}

321

183

322

184

/* The options we understand. */

324

186

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

325

187

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

326

188

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

327

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

328

.arg = "VAR=value",

329

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

330

189

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

331

190

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

332

191

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

333

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

334

.arg = "PLUGIN:ENV=value",

335

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

336

192

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

337

193

.arg = "PLUGIN",

338

194

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

357

213

switch (key) {

358

214

case 'g':

359

215

if (arg != NULL){

360

char *p;

361

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

362

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

363

continue;

364

}

365

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

366

perror("add_argument");

367

return ARGP_ERR_UNKNOWN;

368

}

369

}

370

}

371

break;

372

case 'e':

373

if(arg == NULL){

374

break;

375

}

376

{

377

char *envdef = strdup(arg);

378

if(envdef == NULL){

379

break;

380

}

381

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

382

perror("add_environment");

383

}

216

char *p = strtok(arg, ",");

217

do{

218

addargument(getplugin(NULL, plugins), p);

219

p = strtok(NULL, ",");

220

} while (p);

384

221

}

385

222

break;

386

223

case 'o':

387

224

if (arg != NULL){

388

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

389

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

390

break;

391

}

392

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

393

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

394

break;

395

}

396

if(opt != NULL){

397

char *p;

398

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

399

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

400

continue;

401

}

402

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

403

perror("add_argument");

404

return ARGP_ERR_UNKNOWN;

405

}

406

}

407

}

408

}

409

break;

410

case 'f':

411

if(arg == NULL){

412

break;

413

}

414

{

415

char *envdef = strchr(arg, ':');

416

if(envdef == NULL){

417

break;

418

}

419

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

420

if(p_name == NULL){

421

break;

422

}

423

envdef++;

424

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

425

perror("add_environment");

225

char *name = strtok(arg, ":");

226

char *p = strtok(NULL, ":");

227

if(p){

228

p = strtok(p, ",");

229

do{

230

addargument(getplugin(name, plugins), p);

231

p = strtok(NULL, ",");

232

} while (p);

426

233

}

427

234

}

428

235

break;

429

236

case 'd':

430

237

if (arg != NULL){

431

plugin *p = getplugin(arg, plugins);

432

if(p == NULL){

433

return ARGP_ERR_UNKNOWN;

434

}

435

p->disabled = true;

238

getplugin(arg, plugins)->disabled = true;

436

239

}

437

240

break;

438

241

case 128:

448

251

debug = true;

449

252

break;

450

253

case ARGP_KEY_ARG:

451

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

254

argp_usage (state);

452

255

break;

453

256

case ARGP_KEY_END:

454

257

break;

461

264

plugin *plugin_list = NULL;

462

265

463

266

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

464

.args_doc = "[+PLUS_SEPARATED_OPTIONS]",

267

.args_doc = "",

465

268

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

466

269

467

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

468

if (ret == ARGP_ERR_UNKNOWN){

469

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

470

exitstatus = EXIT_FAILURE;

471

goto fallback;

472

}

473

474

conffp = fopen(argfile, "r");

475

if(conffp != NULL){

476

char *org_line = NULL;

477

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

478

size_t size = 0;

479

ssize_t sret;

480

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

481

const char comment_delim[] = "#";

482

483

while(true){

484

sret = getline(&org_line, &size, conffp);

485

if(sret == -1){

486

break;

487

}

488

489

line = org_line;

490

arg = strsep(&line, comment_delim);

491

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

492

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

493

continue;

494

}

495

new_arg = strdup(p);

496

custom_argv = add_to_argv(custom_argv, &custom_argc, new_arg);

497

if (custom_argv == NULL){

498

perror("add_to_argv");

499

exitstatus = EXIT_FAILURE;

500

goto fallback;

501

}

502

}

503

}

504

free(org_line);

505

} else{

506

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

507

not affect opening plugins */

508

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

509

perror("fopen");

510

exitstatus = EXIT_FAILURE;

511

goto fallback;

512

}

513

}

514

515

if(custom_argv != NULL){

516

custom_argv[0] = argv[0];

517

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

518

&plugin_list);

519

if (ret == ARGP_ERR_UNKNOWN){

520

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

521

exitstatus = EXIT_FAILURE;

522

goto fallback;

523

}

524

}

525

270

argp_parse (&argp, argc, argv, 0, 0, &plugin_list);

271

526

272

if(debug){

527

273

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

528

274

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

530

276

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

531

277

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

532

278

}

533

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

534

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

535

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

536

}

537

279

}

538

280

}

539

281

540

282

ret = setuid(uid);

541

283

if (ret == -1){

542

284

perror("setuid");

544

286

545

287

setgid(gid);

546

288

if (ret == -1){

547

perror("setgid");

289

perror("setuid");

548

290

}

549

291

550

292

dir = opendir(plugindir);

551

293

if(dir == NULL){

552

294

perror("Could not open plugin dir");

553

295

exitstatus = EXIT_FAILURE;

554

goto fallback;

296

goto end;

555

297

}

556

298

557

299

/* Set the FD_CLOEXEC flag on the directory, if possible */

562

304

if(ret < 0){

563

305

perror("set_cloexec_flag");

564

306

exitstatus = EXIT_FAILURE;

565

goto fallback;

307

goto end;

566

308

}

567

309

}

568

310

}

574

316

575

317

// All directory entries have been processed

576

318

if(dirst == NULL){

577

if (errno == EBADF){

578

perror("readdir");

579

exitstatus = EXIT_FAILURE;

580

goto fallback;

581

}

582

319

break;

583

320

}

584

321

628

365

continue;

629

366

}

630

367

}

631

632

char *filename;

633

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

634

if(ret < 0){

635

perror("asprintf");

636

continue;

637

}

638

639

ret = stat(filename, &st);

640

if (ret == -1){

641

perror("stat");

642

free(filename);

643

continue;

644

}

368

369

char *filename = malloc(d_name_len + strlen(plugindir) + 2);

370

if (filename == NULL){

371

perror("malloc");

372

exitstatus = EXIT_FAILURE;

373

goto end;

374

}

375

strcpy(filename, plugindir); /* Spurious warning */

376

strcat(filename, "/"); /* Spurious warning */

377

strcat(filename, dirst->d_name); /* Spurious warning */

378

379

stat(filename, &st);

645

380

646

381

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

647

382

if(debug){

651

386

free(filename);

652

387

continue;

653

388

}

654

plugin *p = getplugin(dirst->d_name, &plugin_list);

655

if(p == NULL){

656

perror("getplugin");

657

free(filename);

658

continue;

659

}

660

if(p->disabled){

389

if(getplugin(dirst->d_name, &plugin_list)->disabled){

661

390

if(debug){

662

391

fprintf(stderr, "Ignoring disabled plugin \"%s\"\n",

663

392

dirst->d_name);

665

394

free(filename);

666

395

continue;

667

396

}

397

plugin *p = getplugin(dirst->d_name, &plugin_list);

668

398

{

669

399

/* Add global arguments to argument list for this plugin */

670

400

plugin *g = getplugin(NULL, &plugin_list);

671

if(g != NULL){

672

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

673

if(not add_argument(p, *a)){

674

perror("add_argument");

675

}

676

}

677

/* Add global environment variables */

678

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

679

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

680

perror("add_environment");

681

}

682

}

683

}

684

}

685

/* If this plugin has any environment variables, we will call

686

using execve and need to duplicate the environment from this

687

process, too. */

688

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

689

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

690

char *copy = strdup(*e);

691

if(copy == NULL){

692

perror("strdup");

693

continue;

694

}

695

if(not add_environment(p, copy)){

696

perror("add_environment");

697

}

698

}

699

}

700

701

int pipefd[2];

401

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

402

addargument(p, *a);

403

}

404

}

405

int pipefd[2];

702

406

ret = pipe(pipefd);

703

407

if (ret == -1){

704

408

perror("pipe");

705

409

exitstatus = EXIT_FAILURE;

706

goto fallback;

410

goto end;

707

411

}

708

412

ret = set_cloexec_flag(pipefd[0]);

709

413

if(ret < 0){

710

414

perror("set_cloexec_flag");

711

415

exitstatus = EXIT_FAILURE;

712

goto fallback;

416

goto end;

713

417

}

714

418

ret = set_cloexec_flag(pipefd[1]);

715

419

if(ret < 0){

716

420

perror("set_cloexec_flag");

717

421

exitstatus = EXIT_FAILURE;

718

goto fallback;

422

goto end;

719

423

}

720

424

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

721

425

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

722

426

if(ret < 0){

723

427

perror("sigprocmask");

724

428

exitstatus = EXIT_FAILURE;

725

goto fallback;

429

goto end;

726

430

}

727

431

// Starting a new process to be watched

728

432

pid_t pid = fork();

729

if(pid == -1){

730

perror("fork");

731

exitstatus = EXIT_FAILURE;

732

goto fallback;

733

}

734

433

if(pid == 0){

735

434

/* this is the child process */

736

435

ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);

743

442

perror("sigprocmask");

744

443

_exit(EXIT_FAILURE);

745

444

}

746

747

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

748

if(ret == -1){

749

perror("dup2");

750

_exit(EXIT_FAILURE);

751

}

445

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

752

446

753

447

if(dirfd(dir) < 0){

754

448

/* If dir has no file descriptor, we could not set FD_CLOEXEC

755

449

above and must now close it manually here. */

756

450

closedir(dir);

757

451

}

758

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

759

if(execv(filename, p->argv) < 0){

760

perror("execv");

761

_exit(EXIT_FAILURE);

762

}

763

} else {

764

if(execve(filename, p->argv, p->environ) < 0){

765

perror("execve");

766

_exit(EXIT_FAILURE);

767

}

452

if(execv(filename, p->argv) < 0){

453

perror("execv");

454

_exit(EXIT_FAILURE);

768

455

}

769

456

/* no return */

770

457

}

779

466

perror("sigprocmask");

780

467

}

781

468

exitstatus = EXIT_FAILURE;

782

goto fallback;

469

goto end;

783

470

}

784

471

785

472

*new_process = (struct process){ .pid = pid,

793

480

if(ret < 0){

794

481

perror("sigprocmask");

795

482

exitstatus = EXIT_FAILURE;

796

goto fallback;

483

goto end;

797

484

}

798

485

799

486

FD_SET(new_process->fd, &rfds_all);

804

491

805

492

}

806

493

807

free_plugin_list(plugin_list);

808

plugin_list = NULL;

494

/* Free the plugin list */

495

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

496

next = plugin_list->next;

497

free(plugin_list->argv);

498

free(plugin_list);

499

}

809

500

810

501

closedir(dir);

811

502

dir = NULL;

812

503

813

504

if (process_list == NULL){

814

fprintf(stderr, "No plugin processes started. Incorrect plugin"

815

" directory?\n");

816

process_list = NULL;

505

fprintf(stderr, "No plugin processes started, exiting\n");

506

exitstatus = EXIT_FAILURE;

507

goto end;

817

508

}

818

509

while(process_list){

819

510

fd_set rfds = rfds_all;

821

512

if (select_ret == -1){

822

513

perror("select");

823

514

exitstatus = EXIT_FAILURE;

824

goto fallback;

515

goto end;

825

516

}

826

517

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

827

518

from one of them */

834

525

/* Bad exit by plugin */

835

526

if(debug){

836

527

if(WIFEXITED(proc->status)){

837

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

838

(unsigned int) (proc->pid),

839

WEXITSTATUS(proc->status));

528

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

529

proc->pid, WEXITSTATUS(proc->status));

840

530

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

841

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

842

(unsigned int) (proc->pid),

843

WTERMSIG(proc->status));

531

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

532

proc->pid, WTERMSIG(proc->status));

844

533

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

845

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

846

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

534

fprintf(stderr, "Plugin %d dumped core\n", proc->pid);

847

535

}

848

536

}

849

537

/* Remove the plugin */

850

538

FD_CLR(proc->fd, &rfds_all);

851

539

/* Block signal while modifying process_list */

852

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

540

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

853

541

if(ret < 0){

854

542

perror("sigprocmask");

855

543

exitstatus = EXIT_FAILURE;

856

goto fallback;

544

goto end;

857

545

}

858

546

/* Delete this process entry from the list */

859

547

if(process_list == proc){

882

570

break;

883

571

}

884

572

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

885

886

bool bret = print_out_password(proc->buffer,

887

proc->buffer_length);

888

if(not bret){

889

perror("print_out_password");

890

exitstatus = EXIT_FAILURE;

573

for(size_t written = 0; written < proc->buffer_length;

574

written += (size_t)ret){

575

ret = TEMP_FAILURE_RETRY(write(STDOUT_FILENO,

576

proc->buffer + written,

577

proc->buffer_length

578

- written));

579

if(ret < 0){

580

perror("write");

581

exitstatus = EXIT_FAILURE;

582

goto end;

583

}

891

584

}

892

goto fallback;

585

goto end;

893

586

}

894

587

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

895

588

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

903

596

if (proc->buffer == NULL){

904

597

perror("malloc");

905

598

exitstatus = EXIT_FAILURE;

906

goto fallback;

599

goto end;

907

600

}

908

601

proc->buffer_size += BUFFER_SIZE;

909

602

}

922

615

}

923

616

}

924

617

}

925

926

927

fallback:

928

929

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

930

/* Fallback if all plugins failed, none are found or an error

931

occured */

932

bool bret;

933

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

934

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

935

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

936

if(not bret){

937

perror("print_out_password");

938

exitstatus = EXIT_FAILURE;

939

}

618

if(process_list == NULL){

619

fprintf(stderr, "All plugin processes failed, exiting\n");

620

exitstatus = EXIT_FAILURE;

940

621

}

941

622

623

end:

942

624

/* Restore old signal handler */

943

ret = sigaction(SIGCHLD, &old_sigchld_action, NULL);

944

if(ret == -1){

945

perror("sigaction");

946

exitstatus = EXIT_FAILURE;

947

}

948

949

if(custom_argv != NULL){

950

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

951

free(*arg);

952

}

953

free(custom_argv);

954

}

955

free_plugin_list(plugin_list);

625

sigaction(SIGCHLD, &old_sigchld_action, NULL);

626

627

/* Free the plugin list */

628

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

629

next = plugin_list->next;

630

free(plugin_list->argv);

631

free(plugin_list);

632

}

956

633

957

634

if(dir != NULL){

958

635

closedir(dir);

962

639

for(process *next; process_list != NULL; process_list = next){

963

640

next = process_list->next;

964

641

close(process_list->fd);

965

ret = kill(process_list->pid, SIGTERM);

966

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

967

/* set-uid proccesses migth not get closed */

968

perror("kill");

969

}

642

kill(process_list->pid, SIGTERM);

970

643

free(process_list->buffer);

971

644

free(process_list);

972

645

}