/mandos/trunk : revision 24.1.62

21

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

22

*/

23

24

#define _GNU_SOURCE /* TEMP_FAILURE_RETRY() */

25

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

51

close() */

52

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

53

FD_CLOEXEC */

54

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

55

strcat() */

54

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

56

55

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

57

56

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

58

57

argp_state, struct argp,

65

64

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

66

65

67

66

#define BUFFER_SIZE 256

67

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

68

69

const char *argp_program_version = "mandos-client 1.0";

69

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

70

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

71

72

struct process;

87

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

88

char **argv;

89

int argc;

90

char **environ;

91

int envc;

90

92

bool disabled;

91

93

struct plugin *next;

92

94

} plugin;

101

103

/* Create a new plugin */

102

104

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

103

105

if (new_plugin == NULL){

104

perror("malloc");

105

exit(EXIT_FAILURE);

106

}

107

new_plugin->name = name;

106

return NULL;

107

}

108

char *copy_name = NULL;

109

if(name != NULL){

110

copy_name = strdup(name);

111

}

112

if(copy_name == NULL){

113

return NULL;

114

}

115

116

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

117

.argc = 1,

118

.envc = 0,

119

.disabled = false,

120

.next = *plugin_list };

121

108

122

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

109

123

if (new_plugin->argv == NULL){

110

perror("malloc");

111

exit(EXIT_FAILURE);

124

free(copy_name);

125

free(new_plugin);

126

return NULL;

112

127

}

113

new_plugin->argv[0] = name;

128

new_plugin->argv[0] = copy_name;

114

129

new_plugin->argv[1] = NULL;

115

new_plugin->argc = 1;

116

new_plugin->disabled = false;

117

new_plugin->next = *plugin_list;

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;

118

139

/* Append the new plugin to the list */

119

140

*plugin_list = new_plugin;

120

141

return new_plugin;

121

142

}

122

143

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

}

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

133

183

134

184

/*

135

185

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

185

235

return true;

186

236

}

187

237

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; plugin_list != NULL; plugin_list = next){

261

next = plugin_list->next;

262

free(plugin_list->name);

263

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

264

free(*arg);

265

}

266

free(plugin_list->argv);

267

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

268

free(*env);

269

}

270

free(plugin_list->environ);

271

free(plugin_list);

272

}

273

}

274

188

275

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

189

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

276

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

277

const char *argfile = ARGFILE;

278

FILE *conffp;

190

279

size_t d_name_len;

191

280

DIR *dir = NULL;

192

281

struct dirent *dirst;

200

289

struct sigaction old_sigchld_action;

201

290

struct sigaction sigchld_action = { .sa_handler = handle_sigchld,

202

291

.sa_flags = SA_NOCLDSTOP };

203

char *plus_options = NULL;

204

char **plus_argv = NULL;

205

292

char **custom_argv = NULL;

293

int custom_argc = 0;

294

206

295

/* Establish a signal handler */

207

296

sigemptyset(&sigchld_action.sa_mask);

208

297

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

209

if(ret < 0){

298

if(ret == -1){

210

299

perror("sigaddset");

211

exit(EXIT_FAILURE);

300

exitstatus = EXIT_FAILURE;

301

goto fallback;

212

302

}

213

303

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

214

if(ret < 0){

304

if(ret == -1){

215

305

perror("sigaction");

216

exit(EXIT_FAILURE);

306

exitstatus = EXIT_FAILURE;

307

goto fallback;

217

308

}

218

309

219

310

/* The options we understand. */

221

312

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

222

313

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

223

314

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

315

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

316

.arg = "VAR=value",

317

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

224

318

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

225

319

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

226

320

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

321

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

322

.arg = "PLUGIN:ENV=value",

323

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

227

324

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

228

325

.arg = "PLUGIN",

229

326

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

248

345

switch (key) {

249

346

case 'g':

250

347

if (arg != NULL){

251

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

252

do{

253

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

254

p = strtok(NULL, ",");

255

} while (p != NULL);

348

char *p;

349

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

350

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

351

continue;

352

}

353

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

354

perror("add_argument");

355

return ARGP_ERR_UNKNOWN;

356

}

357

}

358

}

359

break;

360

case 'e':

361

if(arg == NULL){

362

break;

363

}

364

{

365

char *envdef = strdup(arg);

366

if(envdef == NULL){

367

break;

368

}

369

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

370

perror("add_environment");

371

}

256

372

}

257

373

break;

258

374

case 'o':

259

375

if (arg != NULL){

260

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

261

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

262

if(p != NULL){

263

p = strtok(p, ",");

264

do{

265

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

266

p = strtok(NULL, ",");

267

} while (p != NULL);

376

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

377

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

378

break;

379

}

380

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

381

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

382

break;

383

}

384

if(opt != NULL){

385

char *p;

386

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

387

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

388

continue;

389

}

390

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

391

perror("add_argument");

392

return ARGP_ERR_UNKNOWN;

393

}

394

}

395

}

396

}

397

break;

398

case 'f':

399

if(arg == NULL){

400

break;

401

}

402

{

403

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

404

if(envdef == NULL){

405

break;

406

}

407

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

408

if(p_name == NULL){

409

break;

410

}

411

envdef++;

412

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

413

perror("add_environment");

268

414

}

269

415

}

270

416

break;

271

417

case 'd':

272

418

if (arg != NULL){

273

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

419

plugin *p = getplugin(arg, plugins);

420

if(p == NULL){

421

return ARGP_ERR_UNKNOWN;

422

}

423

p->disabled = true;

274

424

}

275

425

break;

276

426

case 128:

286

436

debug = true;

287

437

break;

288

438

case ARGP_KEY_ARG:

289

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

290

argp_usage (state);

291

}

292

plus_options = arg;

439

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

293

440

break;

294

441

case ARGP_KEY_END:

295

442

break;

309

456

if (ret == ARGP_ERR_UNKNOWN){

310

457

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

311

458

exitstatus = EXIT_FAILURE;

312

goto end;

459

goto fallback;

313

460

}

314

315

if(plus_options){

316

/* This is a mangled argument in the form of

317

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

318

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

319

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

320

const char delims[] = "+";

321

char *arg;

322

int new_argc = 1;

323

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

324

if(plus_argv == NULL){

325

perror("malloc");

461

462

conffp = fopen(argfile, "r");

463

if(conffp != NULL){

464

char *org_line = NULL;

465

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

466

size_t size = 0;

467

ssize_t sret;

468

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

469

const char comment_delim[] = "#";

470

471

while(true){

472

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

473

if(sret == -1){

474

break;

475

}

476

477

line = org_line;

478

arg = strsep(&line, comment_delim);

479

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

480

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

481

continue;

482

}

483

new_arg = strdup(p);

484

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

485

if (custom_argv == NULL){

486

perror("add_to_argv");

487

exitstatus = EXIT_FAILURE;

488

goto fallback;

489

}

490

}

491

}

492

free(org_line);

493

} else{

494

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

495

not affect opening plugins */

496

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

497

perror("fopen");

326

498

exitstatus = EXIT_FAILURE;

327

goto end;

328

}

329

plus_argv[0] = argv[0];

330

plus_argv[1] = NULL;

331

arg = strtok(plus_options, delims); /* Get first argument */

332

while(arg != NULL){

333

new_argc++;

334

plus_argv = realloc(plus_argv, sizeof(char *)

335

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

336

if(plus_argv == NULL){

337

perror("realloc");

338

exitstatus = EXIT_FAILURE;

339

goto end;

340

}

341

plus_argv[new_argc-1] = arg;

342

plus_argv[new_argc] = NULL;

343

arg = strtok(NULL, delims); /* Get next argument */

344

}

345

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

499

goto fallback;

500

}

501

}

502

503

if(custom_argv != NULL){

504

custom_argv[0] = argv[0];

505

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

346

506

if (ret == ARGP_ERR_UNKNOWN){

347

507

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

348

508

exitstatus = EXIT_FAILURE;

349

goto end;

509

goto fallback;

350

510

}

351

511

}

352

512

357

517

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

358

518

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

359

519

}

520

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

521

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

522

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

523

}

360

524

}

361

525

}

362

526

374

538

if(dir == NULL){

375

539

perror("Could not open plugin dir");

376

540

exitstatus = EXIT_FAILURE;

377

goto end;

541

goto fallback;

378

542

}

379

543

380

544

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

385

549

if(ret < 0){

386

550

perror("set_cloexec_flag");

387

551

exitstatus = EXIT_FAILURE;

388

goto end;

552

goto fallback;

389

553

}

390

554

}

391

555

}

400

564

if (errno == EBADF){

401

565

perror("readdir");

402

566

exitstatus = EXIT_FAILURE;

403

goto end;

567

goto fallback;

404

568

}

405

569

break;

406

570

}

451

615

continue;

452

616

}

453

617

}

454

455

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

456

if (filename == NULL){

457

perror("malloc");

618

619

char *filename;

620

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

621

if(ret < 0){

622

perror("asprintf");

458

623

continue;

459

624

}

460

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

461

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

462

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

463

625

464

626

ret = stat(filename, &st);

465

627

if (ret == -1){

476

638

free(filename);

477

639

continue;

478

640

}

479

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

641

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

642

if(p == NULL){

643

perror("getplugin");

644

free(filename);

645

continue;

646

}

647

if(p->disabled){

480

648

if(debug){

481

649

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

482

650

dirst->d_name);

484

652

free(filename);

485

653

continue;

486

654

}

487

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

488

655

{

489

656

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

490

657

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

491

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

492

addargument(p, *a);

493

}

494

}

658

if(g != NULL){

659

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

660

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

661

perror("add_argument");

662

}

663

}

664

/* Add global environment variables */

665

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

666

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

667

perror("add_environment");

668

}

669

}

670

}

671

}

672

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

673

using execve and need to duplicate the environment from this

674

process, too. */

675

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

676

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

677

char *copy = strdup(*e);

678

if(copy == NULL){

679

perror("strdup");

680

continue;

681

}

682

if(not add_environment(p, copy)){

683

perror("add_environment");

684

}

685

}

686

}

687

495

688

int pipefd[2];

496

689

ret = pipe(pipefd);

497

690

if (ret == -1){

498

691

perror("pipe");

499

692

exitstatus = EXIT_FAILURE;

500

goto end;

693

goto fallback;

501

694

}

502

695

ret = set_cloexec_flag(pipefd[0]);

503

696

if(ret < 0){

504

697

perror("set_cloexec_flag");

505

698

exitstatus = EXIT_FAILURE;

506

goto end;

699

goto fallback;

507

700

}

508

701

ret = set_cloexec_flag(pipefd[1]);

509

702

if(ret < 0){

510

703

perror("set_cloexec_flag");

511

704

exitstatus = EXIT_FAILURE;

512

goto end;

705

goto fallback;

513

706

}

514

707

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

515

708

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

516

709

if(ret < 0){

517

710

perror("sigprocmask");

518

711

exitstatus = EXIT_FAILURE;

519

goto end;

712

goto fallback;

520

713

}

521

714

// Starting a new process to be watched

522

715

pid_t pid = fork();

523

716

if(pid == -1){

524

717

perror("fork");

525

718

exitstatus = EXIT_FAILURE;

526

goto end;

719

goto fallback;

527

720

}

528

721

if(pid == 0){

529

722

/* this is the child process */

549

742

above and must now close it manually here. */

550

743

closedir(dir);

551

744

}

552

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

553

perror("execv");

554

_exit(EXIT_FAILURE);

745

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

746

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

747

perror("execv");

748

_exit(EXIT_FAILURE);

749

}

750

} else {

751

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

752

perror("execve");

753

_exit(EXIT_FAILURE);

754

}

555

755

}

556

756

/* no return */

557

757

}

566

766

perror("sigprocmask");

567

767

}

568

768

exitstatus = EXIT_FAILURE;

569

goto end;

769

goto fallback;

570

770

}

571

771

572

772

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

580

780

if(ret < 0){

581

781

perror("sigprocmask");

582

782

exitstatus = EXIT_FAILURE;

583

goto end;

783

goto fallback;

584

784

}

585

785

586

786

FD_SET(new_process->fd, &rfds_all);

591

791

592

792

}

593

793

594

/* Free the plugin list */

595

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

596

next = plugin_list->next;

597

free(plugin_list->argv);

598

free(plugin_list);

599

}

794

free_plugin_list(plugin_list);

600

795

601

796

closedir(dir);

602

797

dir = NULL;

612

807

if (select_ret == -1){

613

808

perror("select");

614

809

exitstatus = EXIT_FAILURE;

615

goto end;

810

goto fallback;

616

811

}

617

812

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

618

813

from one of them */

644

839

if(ret < 0){

645

840

perror("sigprocmask");

646

841

exitstatus = EXIT_FAILURE;

647

goto end;

842

goto fallback;

648

843

}

649

844

/* Delete this process entry from the list */

650

845

if(process_list == proc){

679

874

perror("print_out_password");

680

875

exitstatus = EXIT_FAILURE;

681

876

}

682

goto end;

877

goto fallback;

683

878

}

684

879

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

685

880

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

693

888

if (proc->buffer == NULL){

694

889

perror("malloc");

695

890

exitstatus = EXIT_FAILURE;

696

goto end;

891

goto fallback;

697

892

}

698

893

proc->buffer_size += BUFFER_SIZE;

699

894

}

714

909

}

715

910

716

911

717

end:

912

fallback:

718

913

719

914

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

720

915

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

725

920

if(not bret){

726

921

perror("print_out_password");

727

922

exitstatus = EXIT_FAILURE;

728

goto end;

729

923

}

730

924

}

731

925

732

926

/* Restore old signal handler */

733

sigaction(SIGCHLD, &old_sigchld_action, NULL);

734

735

free(plus_argv);

736

737

/* Free the plugin list */

738

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

739

next = plugin_list->next;

740

free(plugin_list->argv);

741

free(plugin_list);

742

}

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

940

744

941

if(dir != NULL){

745

942

closedir(dir);