/mandos/trunk : revision 24.1.52

21

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

22

*/

23

24

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

25

asprintf() */

24

#define _GNU_SOURCE /* TEMP_FAILURE_RETRY() */

25

26

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

27

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

28

EXIT_SUCCESS, realloc() */

51

close() */

52

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

53

FD_CLOEXEC */

54

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

54

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

55

strcat() */

55

56

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

56

57

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

57

58

argp_state, struct argp,

64

65

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

65

66

67

#define BUFFER_SIZE 256

67

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

68

#define ARGFILE "/conf/conf.d/mandos/plugins.conf"

68

69

70

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

70

71

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

87

88

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

88

89

char **argv;

89

90

int argc;

90

char **environ;

91

int envc;

92

91

bool disabled;

93

92

struct plugin *next;

94

93

} plugin;

103

102

/* Create a new plugin */

104

103

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

105

104

if (new_plugin == NULL){

106

return NULL;

107

}

108

char *copy_name = NULL;

109

if(name != NULL){

110

copy_name = strdup(name);

111

if(copy_name == NULL){

112

return NULL;

113

}

114

}

115

116

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

117

.argc = 1,

118

.envc = 0,

119

.disabled = false,

120

.next = *plugin_list };

121

105

perror("malloc");

106

exit(EXIT_FAILURE);

107

}

108

new_plugin->name = name;

122

109

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

123

110

if (new_plugin->argv == NULL){

124

free(copy_name);

125

free(new_plugin);

126

return NULL;

111

perror("malloc");

112

exit(EXIT_FAILURE);

127

113

}

128

new_plugin->argv[0] = copy_name;

114

new_plugin->argv[0] = name;

129

115

new_plugin->argv[1] = NULL;

130

131

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

132

if(new_plugin->environ == NULL){

133

free(copy_name);

134

free(new_plugin->argv);

135

free(new_plugin);

136

return NULL;

137

}

138

new_plugin->environ[0] = NULL;

116

new_plugin->argc = 1;

117

new_plugin->disabled = false;

118

new_plugin->next = *plugin_list;

139

119

/* Append the new plugin to the list */

140

120

*plugin_list = new_plugin;

141

121

return new_plugin;

142

122

}

143

123

144

/* Helper function for add_argument and add_environment */

145

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

146

int *len){

147

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

148

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

149

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

150

/* Malloc check */

151

if(*array == NULL){

152

return false;

153

}

154

/* Make a copy of the new string */

155

char *copy = strdup(new);

156

if(copy == NULL){

157

return false;

158

}

159

/* Insert the copy */

160

(*array)[*len] = copy;

161

(*len)++;

162

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

163

(*array)[*len] = NULL;

164

return true;

165

}

166

167

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

168

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

169

if(p == NULL){

170

return false;

171

}

172

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

173

}

174

175

/* Add to a plugin's environment */

176

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

177

if(p == NULL){

178

return false;

179

}

180

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

181

}

182

124

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

125

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

126

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

127

if (p->argv == NULL){

128

perror("malloc");

129

exit(EXIT_FAILURE);

130

}

131

p->argc++;

132

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

133

}

183

134

184

135

/*

185

136

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

235

186

return true;

236

187

}

237

188

238

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

189

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

190

239

191

if (argv == NULL){

240

192

*argc = 1;

241

193

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

252

204

return NULL;

253

205

}

254

206

argv[*argc-1] = arg;

255

argv[*argc] = NULL;

207

argv[*argc] = NULL;

256

208

return argv;

257

209

}

258

210

259

static void free_plugin_list(plugin *plugin_list){

260

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

261

next = plugin_list->next;

262

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

263

free(*arg);

264

}

265

free(plugin_list->argv);

266

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

267

free(*env);

268

}

269

free(plugin_list->environ);

270

free(plugin_list);

271

}

272

}

273

274

211

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

275

212

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

276

213

const char *argfile = ARGFILE;

294

231

/* Establish a signal handler */

295

232

sigemptyset(&sigchld_action.sa_mask);

296

233

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

297

if(ret == -1){

234

if(ret < 0){

298

235

perror("sigaddset");

299

exitstatus = EXIT_FAILURE;

300

goto fallback;

236

exit(EXIT_FAILURE);

301

237

}

302

238

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

303

if(ret == -1){

239

if(ret < 0){

304

240

perror("sigaction");

305

exitstatus = EXIT_FAILURE;

306

goto fallback;

241

exit(EXIT_FAILURE);

307

242

}

308

243

309

244

/* The options we understand. */

311

246

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

312

247

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

313

248

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

314

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

315

.arg = "VAR=value",

316

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

317

249

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

318

250

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

319

251

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

320

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

321

.arg = "PLUGIN:ENV=value",

322

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

323

252

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

324

253

.arg = "PLUGIN",

325

254

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

349

278

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

350

279

continue;

351

280

}

352

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

353

perror("add_argument");

354

return ARGP_ERR_UNKNOWN;

355

}

356

}

357

}

358

break;

359

case 'e':

360

if(arg == NULL){

361

break;

362

}

363

{

364

char *envdef = strdup(arg);

365

if(envdef == NULL){

366

break;

367

}

368

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

369

perror("add_environment");

281

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

370

282

}

371

283

}

372

284

break;

373

285

case 'o':

374

286

if (arg != NULL){

375

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

376

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

287

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

288

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

377

289

break;

378

290

}

379

291

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

386

298

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

387

299

continue;

388

300

}

389

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

390

perror("add_argument");

391

return ARGP_ERR_UNKNOWN;

392

}

301

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

393

302

}

394

303

}

395

304

}

396

305

break;

397

case 'f':

398

if(arg == NULL){

399

break;

400

}

401

{

402

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

403

if(envdef == NULL){

404

break;

405

}

406

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

407

if(p_name == NULL){

408

break;

409

}

410

envdef++;

411

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

412

perror("add_environment");

413

}

414

}

415

break;

416

306

case 'd':

417

307

if (arg != NULL){

418

plugin *p = getplugin(arg, plugins);

419

if(p == NULL){

420

return ARGP_ERR_UNKNOWN;

421

}

422

p->disabled = true;

308

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

423

309

}

424

310

break;

425

311

case 128:

435

321

debug = true;

436

322

break;

437

323

case ARGP_KEY_ARG:

438

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

324

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

439

325

break;

440

326

case ARGP_KEY_END:

441

327

break;

455

341

if (ret == ARGP_ERR_UNKNOWN){

456

342

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

457

343

exitstatus = EXIT_FAILURE;

458

goto fallback;

344

goto end;

459

345

}

460

346

461

347

conffp = fopen(argfile, "r");

462

348

if(conffp != NULL){

463

349

char *org_line = NULL;

464

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

465

350

size_t size = 0;

466

351

ssize_t sret;

352

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

467

353

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

468

354

const char comment_delim[] = "#";

469

355

480

366

continue;

481

367

}

482

368

new_arg = strdup(p);

483

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

369

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

484

370

if (custom_argv == NULL){

485

perror("add_to_argv");

371

perror("addcustomargument");

486

372

exitstatus = EXIT_FAILURE;

487

goto fallback;

373

goto end;

488

374

}

489

375

}

490

376

}

495

381

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

496

382

perror("fopen");

497

383

exitstatus = EXIT_FAILURE;

498

goto fallback;

384

goto end;

499

385

}

500

386

}

501

387

505

391

if (ret == ARGP_ERR_UNKNOWN){

506

392

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

507

393

exitstatus = EXIT_FAILURE;

508

goto fallback;

394

goto end;

509

395

}

510

396

}

511

397

516

402

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

517

403

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

518

404

}

519

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

520

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

521

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

522

}

523

405

}

524

406

}

525

407

537

419

if(dir == NULL){

538

420

perror("Could not open plugin dir");

539

421

exitstatus = EXIT_FAILURE;

540

goto fallback;

422

goto end;

541

423

}

542

424

543

425

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

548

430

if(ret < 0){

549

431

perror("set_cloexec_flag");

550

432

exitstatus = EXIT_FAILURE;

551

goto fallback;

433

goto end;

552

434

}

553

435

}

554

436

}

563

445

if (errno == EBADF){

564

446

perror("readdir");

565

447

exitstatus = EXIT_FAILURE;

566

goto fallback;

448

goto end;

567

449

}

568

450

break;

569

451

}

614

496

continue;

615

497

}

616

498

}

617

618

char *filename;

619

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

620

if(ret < 0){

621

perror("asprintf");

499

500

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

501

if (filename == NULL){

502

perror("malloc");

622

503

continue;

623

504

}

505

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

506

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

507

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

624

508

625

509

ret = stat(filename, &st);

626

510

if (ret == -1){

637

521

free(filename);

638

522

continue;

639

523

}

640

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

641

if(p == NULL){

642

perror("getplugin");

643

free(filename);

644

continue;

645

}

646

if(p->disabled){

524

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

647

525

if(debug){

648

526

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

649

527

dirst->d_name);

651

529

free(filename);

652

530

continue;

653

531

}

532

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

654

533

{

655

534

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

656

535

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

657

if(g != NULL){

658

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

659

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

660

perror("add_argument");

661

}

662

}

663

/* Add global environment variables */

664

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

665

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

666

perror("add_environment");

667

}

668

}

669

}

670

}

671

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

672

using execve and need to duplicate the environment from this

673

process, too. */

674

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

675

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

676

char *copy = strdup(*e);

677

if(copy == NULL){

678

perror("strdup");

679

continue;

680

}

681

if(not add_environment(p, copy)){

682

perror("add_environment");

683

}

684

}

685

}

686

687

int pipefd[2];

536

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

537

addargument(p, *a);

538

}

539

}

540

int pipefd[2];

688

541

ret = pipe(pipefd);

689

542

if (ret == -1){

690

543

perror("pipe");

691

544

exitstatus = EXIT_FAILURE;

692

goto fallback;

545

goto end;

693

546

}

694

547

ret = set_cloexec_flag(pipefd[0]);

695

548

if(ret < 0){

696

549

perror("set_cloexec_flag");

697

550

exitstatus = EXIT_FAILURE;

698

goto fallback;

551

goto end;

699

552

}

700

553

ret = set_cloexec_flag(pipefd[1]);

701

554

if(ret < 0){

702

555

perror("set_cloexec_flag");

703

556

exitstatus = EXIT_FAILURE;

704

goto fallback;

557

goto end;

705

558

}

706

559

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

707

560

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

708

561

if(ret < 0){

709

562

perror("sigprocmask");

710

563

exitstatus = EXIT_FAILURE;

711

goto fallback;

564

goto end;

712

565

}

713

566

// Starting a new process to be watched

714

567

pid_t pid = fork();

715

568

if(pid == -1){

716

569

perror("fork");

717

570

exitstatus = EXIT_FAILURE;

718

goto fallback;

571

goto end;

719

572

}

720

573

if(pid == 0){

721

574

/* this is the child process */

741

594

above and must now close it manually here. */

742

595

closedir(dir);

743

596

}

744

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

745

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

746

perror("execv");

747

_exit(EXIT_FAILURE);

748

}

749

} else {

750

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

751

perror("execve");

752

_exit(EXIT_FAILURE);

753

}

597

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

598

perror("execv");

599

_exit(EXIT_FAILURE);

754

600

}

755

601

/* no return */

756

602

}

765

611

perror("sigprocmask");

766

612

}

767

613

exitstatus = EXIT_FAILURE;

768

goto fallback;

614

goto end;

769

615

}

770

616

771

617

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

779

625

if(ret < 0){

780

626

perror("sigprocmask");

781

627

exitstatus = EXIT_FAILURE;

782

goto fallback;

628

goto end;

783

629

}

784

630

785

631

FD_SET(new_process->fd, &rfds_all);

790

636

791

637

}

792

638

793

free_plugin_list(plugin_list);

794

plugin_list = NULL;

639

/* Free the plugin list */

640

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

641

next = plugin_list->next;

642

free(plugin_list->argv);

643

free(plugin_list);

644

}

795

645

796

646

closedir(dir);

797

647

dir = NULL;

807

657

if (select_ret == -1){

808

658

perror("select");

809

659

exitstatus = EXIT_FAILURE;

810

goto fallback;

660

goto end;

811

661

}

812

662

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

813

663

from one of them */

839

689

if(ret < 0){

840

690

perror("sigprocmask");

841

691

exitstatus = EXIT_FAILURE;

842

goto fallback;

692

goto end;

843

693

}

844

694

/* Delete this process entry from the list */

845

695

if(process_list == proc){

874

724

perror("print_out_password");

875

725

exitstatus = EXIT_FAILURE;

876

726

}

877

goto fallback;

727

goto end;

878

728

}

879

729

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

880

730

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

888

738

if (proc->buffer == NULL){

889

739

perror("malloc");

890

740

exitstatus = EXIT_FAILURE;

891

goto fallback;

741

goto end;

892

742

}

893

743

proc->buffer_size += BUFFER_SIZE;

894

744

}

909

759

}

910

760

911

761

912

fallback:

762

end:

913

763

914

764

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

915

765

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

920

770

if(not bret){

921

771

perror("print_out_password");

922

772

exitstatus = EXIT_FAILURE;

773

goto end;

923

774

}

924

775

}

925

776

926

777

/* Restore old signal handler */

927

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

928

if(ret == -1){

929

perror("sigaction");

930

exitstatus = EXIT_FAILURE;

931

}

932

933

if(custom_argv != NULL){

934

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

935

free(*arg);

936

}

937

free(custom_argv);

938

}

939

free_plugin_list(plugin_list);

778

sigaction(SIGCHLD, &old_sigchld_action, NULL);

779

780

free(custom_argv);

781

782

/* Free the plugin list */

783

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

784

next = plugin_list->next;

785

free(plugin_list->argv);

786

free(plugin_list);

787

}

940

788

941

789

if(dir != NULL){

942

790

closedir(dir);