/mandos/trunk : revision 75

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

69

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

70

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

87

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

88

char **argv;

89

int argc;

90

char **environ;

91

int envc;

92

90

bool disabled;

93

91

struct plugin *next;

94

92

} plugin;

103

101

/* Create a new plugin */

104

102

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

105

103

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

104

perror("malloc");

105

exit(EXIT_FAILURE);

106

}

107

new_plugin->name = name;

122

108

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

123

109

if (new_plugin->argv == NULL){

124

free(copy_name);

125

free(new_plugin);

126

return NULL;

110

perror("malloc");

111

exit(EXIT_FAILURE);

127

112

}

128

new_plugin->argv[0] = copy_name;

113

new_plugin->argv[0] = name;

129

114

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;

115

new_plugin->argc = 1;

116

new_plugin->disabled = false;

117

new_plugin->next = *plugin_list;

139

118

/* Append the new plugin to the list */

140

119

*plugin_list = new_plugin;

141

120

return new_plugin;

142

121

}

143

122

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

123

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

124

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

125

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

126

if (p->argv == NULL){

127

perror("malloc");

128

exit(EXIT_FAILURE);

129

}

130

p->argc++;

131

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

132

}

183

133

184

134

/*

185

135

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

235

185

return true;

236

186

}

237

187

238

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

239

if (argv == NULL){

240

*argc = 1;

241

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

242

if(argv == NULL){

243

return NULL;

244

}

245

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

246

argv[1] = NULL;

247

}

248

*argc += 1;

249

argv = realloc(argv, sizeof(char *)

250

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

251

if(argv == NULL){

252

return NULL;

253

}

254

argv[*argc-1] = arg;

255

argv[*argc] = NULL;

256

return argv;

257

}

258

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

188

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

275

189

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

276

const char *argfile = ARGFILE;

277

FILE *conffp;

278

190

size_t d_name_len;

279

191

DIR *dir = NULL;

280

192

struct dirent *dirst;

288

200

struct sigaction old_sigchld_action;

289

201

struct sigaction sigchld_action = { .sa_handler = handle_sigchld,

290

202

.sa_flags = SA_NOCLDSTOP };

291

char **custom_argv = NULL;

292

int custom_argc = 0;

293

203

char *plus_options = NULL;

204

char **plus_argv = NULL;

205

294

206

/* Establish a signal handler */

295

207

sigemptyset(&sigchld_action.sa_mask);

296

208

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

297

if(ret == -1){

209

if(ret < 0){

298

210

perror("sigaddset");

299

exitstatus = EXIT_FAILURE;

300

goto fallback;

211

exit(EXIT_FAILURE);

301

212

}

302

213

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

303

if(ret == -1){

214

if(ret < 0){

304

215

perror("sigaction");

305

exitstatus = EXIT_FAILURE;

306

goto fallback;

216

exit(EXIT_FAILURE);

307

217

}

308

218

309

219

/* The options we understand. */

311

221

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

312

222

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

313

223

.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

224

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

318

225

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

319

226

.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

227

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

324

228

.arg = "PLUGIN",

325

229

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

346

250

if (arg != NULL){

347

251

char *p;

348

252

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

349

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

253

if(strcmp(p, "") == 0){

350

254

continue;

351

255

}

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");

256

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

370

257

}

371

258

}

372

259

break;

373

260

case 'o':

374

261

if (arg != NULL){

375

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

376

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

262

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

263

if(strcmp(name, "") == 0){

377

264

break;

378

265

}

379

266

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

380

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

267

if(strcmp(opt, "") == 0){

381

268

break;

382

269

}

383

270

if(opt != NULL){

384

271

char *p;

385

272

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

386

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

273

if(strcmp(p, "") == 0){

387

274

continue;

388

275

}

389

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

390

perror("add_argument");

391

return ARGP_ERR_UNKNOWN;

392

}

276

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

393

277

}

394

278

}

395

279

}

396

280

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

281

case 'd':

417

282

if (arg != NULL){

418

plugin *p = getplugin(arg, plugins);

419

if(p == NULL){

420

return ARGP_ERR_UNKNOWN;

421

}

422

p->disabled = true;

283

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

423

284

}

424

285

break;

425

286

case 128:

435

296

debug = true;

436

297

break;

437

298

case ARGP_KEY_ARG:

438

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

299

if(plus_options != NULL or arg == NULL or arg[0] != '+'){

300

argp_usage (state);

301

}

302

plus_options = arg;

439

303

break;

440

304

case ARGP_KEY_END:

441

305

break;

455

319

if (ret == ARGP_ERR_UNKNOWN){

456

320

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

457

321

exitstatus = EXIT_FAILURE;

458

goto fallback;

322

goto end;

459

323

}

460

461

conffp = fopen(argfile, "r");

462

if(conffp != NULL){

463

char *org_line = NULL;

464

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

465

size_t size = 0;

466

ssize_t sret;

467

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

468

const char comment_delim[] = "#";

469

470

while(true){

471

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

472

if(sret == -1){

473

break;

474

}

475

476

line = org_line;

477

arg = strsep(&line, comment_delim);

478

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

479

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

480

continue;

481

}

482

new_arg = strdup(p);

483

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

484

if (custom_argv == NULL){

485

perror("add_to_argv");

486

exitstatus = EXIT_FAILURE;

487

goto fallback;

488

}

489

}

490

}

491

free(org_line);

492

} else{

493

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

494

not affect opening plugins */

495

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

496

perror("fopen");

324

325

if(plus_options){

326

/* This is a mangled argument in the form of

327

"+--option+--other-option=parameter+--yet-another-option", etc */

328

/* Make new argc and argv vars, and call argp_parse() again. */

329

plus_options++; /* skip the first '+' character */

330

const char delims[] = "+";

331

char *arg;

332

int new_argc = 1;

333

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

334

if(plus_argv == NULL){

335

perror("malloc");

497

336

exitstatus = EXIT_FAILURE;

498

goto fallback;

337

goto end;

499

338

}

500

}

339

plus_argv[0] = argv[0];

340

plus_argv[1] = NULL;

501

341

502

if(custom_argv != NULL){

503

custom_argv[0] = argv[0];

504

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

342

while((arg = strsep(&plus_options, delims)) != NULL){

343

new_argc++;

344

plus_argv = realloc(plus_argv, sizeof(char *)

345

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

346

if(plus_argv == NULL){

347

perror("realloc");

348

exitstatus = EXIT_FAILURE;

349

goto end;

350

}

351

plus_argv[new_argc-1] = arg;

352

plus_argv[new_argc] = NULL;

353

}

354

355

ret = argp_parse (&argp, new_argc, plus_argv, 0, 0, &plugin_list);

505

356

if (ret == ARGP_ERR_UNKNOWN){

506

357

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

507

358

exitstatus = EXIT_FAILURE;

508

goto fallback;

359

goto end;

509

360

}

510

361

}

511

362

516

367

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

517

368

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

518

369

}

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

370

}

524

371

}

525

372

537

384

if(dir == NULL){

538

385

perror("Could not open plugin dir");

539

386

exitstatus = EXIT_FAILURE;

540

goto fallback;

387

goto end;

541

388

}

542

389

543

390

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

548

395

if(ret < 0){

549

396

perror("set_cloexec_flag");

550

397

exitstatus = EXIT_FAILURE;

551

goto fallback;

398

goto end;

552

399

}

553

400

}

554

401

}

563

410

if (errno == EBADF){

564

411

perror("readdir");

565

412

exitstatus = EXIT_FAILURE;

566

goto fallback;

413

goto end;

567

414

}

568

415

break;

569

416

}

614

461

continue;

615

462

}

616

463

}

617

618

char *filename;

619

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

620

if(ret < 0){

621

perror("asprintf");

464

465

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

466

if (filename == NULL){

467

perror("malloc");

622

468

continue;

623

469

}

470

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

471

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

472

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

624

473

625

474

ret = stat(filename, &st);

626

475

if (ret == -1){

637

486

free(filename);

638

487

continue;

639

488

}

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

489

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

647

490

if(debug){

648

491

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

649

492

dirst->d_name);

651

494

free(filename);

652

495

continue;

653

496

}

497

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

654

498

{

655

499

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

656

500

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];

501

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

502

addargument(p, *a);

503

}

504

}

505

int pipefd[2];

688

506

ret = pipe(pipefd);

689

507

if (ret == -1){

690

508

perror("pipe");

691

509

exitstatus = EXIT_FAILURE;

692

goto fallback;

510

goto end;

693

511

}

694

512

ret = set_cloexec_flag(pipefd[0]);

695

513

if(ret < 0){

696

514

perror("set_cloexec_flag");

697

515

exitstatus = EXIT_FAILURE;

698

goto fallback;

516

goto end;

699

517

}

700

518

ret = set_cloexec_flag(pipefd[1]);

701

519

if(ret < 0){

702

520

perror("set_cloexec_flag");

703

521

exitstatus = EXIT_FAILURE;

704

goto fallback;

522

goto end;

705

523

}

706

524

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

707

525

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

708

526

if(ret < 0){

709

527

perror("sigprocmask");

710

528

exitstatus = EXIT_FAILURE;

711

goto fallback;

529

goto end;

712

530

}

713

531

// Starting a new process to be watched

714

532

pid_t pid = fork();

715

533

if(pid == -1){

716

534

perror("fork");

717

535

exitstatus = EXIT_FAILURE;

718

goto fallback;

536

goto end;

719

537

}

720

538

if(pid == 0){

721

539

/* this is the child process */

741

559

above and must now close it manually here. */

742

560

closedir(dir);

743

561

}

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

}

562

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

563

perror("execv");

564

_exit(EXIT_FAILURE);

754

565

}

755

566

/* no return */

756

567

}

765

576

perror("sigprocmask");

766

577

}

767

578

exitstatus = EXIT_FAILURE;

768

goto fallback;

579

goto end;

769

580

}

770

581

771

582

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

779

590

if(ret < 0){

780

591

perror("sigprocmask");

781

592

exitstatus = EXIT_FAILURE;

782

goto fallback;

593

goto end;

783

594

}

784

595

785

596

FD_SET(new_process->fd, &rfds_all);

790

601

791

602

}

792

603

793

free_plugin_list(plugin_list);

794

plugin_list = NULL;

604

/* Free the plugin list */

605

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

606

next = plugin_list->next;

607

free(plugin_list->argv);

608

free(plugin_list);

609

}

795

610

796

611

closedir(dir);

797

612

dir = NULL;

807

622

if (select_ret == -1){

808

623

perror("select");

809

624

exitstatus = EXIT_FAILURE;

810

goto fallback;

625

goto end;

811

626

}

812

627

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

813

628

from one of them */

839

654

if(ret < 0){

840

655

perror("sigprocmask");

841

656

exitstatus = EXIT_FAILURE;

842

goto fallback;

657

goto end;

843

658

}

844

659

/* Delete this process entry from the list */

845

660

if(process_list == proc){

874

689

perror("print_out_password");

875

690

exitstatus = EXIT_FAILURE;

876

691

}

877

goto fallback;

692

goto end;

878

693

}

879

694

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

880

695

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

888

703

if (proc->buffer == NULL){

889

704

perror("malloc");

890

705

exitstatus = EXIT_FAILURE;

891

goto fallback;

706

goto end;

892

707

}

893

708

proc->buffer_size += BUFFER_SIZE;

894

709

}

909

724

}

910

725

911

726

912

fallback:

727

end:

913

728

914

729

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

915

730

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

920

735

if(not bret){

921

736

perror("print_out_password");

922

737

exitstatus = EXIT_FAILURE;

738

goto end;

923

739

}

924

740

}

925

741

926

742

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

743

sigaction(SIGCHLD, &old_sigchld_action, NULL);

744

745

free(plus_argv);

746

747

/* Free the plugin list */

748

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

749

next = plugin_list->next;

750

free(plugin_list->argv);

751

free(plugin_list);

752

}

940

753

941

754

if(dir != NULL){

942

755

closedir(dir);