/mandos/trunk : revision 63

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> /* strtok, 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";

69

const char *argp_program_version = "mandos-client 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;

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 = strdup(name);

109

if(copy_name == NULL){

110

return NULL;

111

}

112

113

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

114

.argc = 1,

115

.envc = 0,

116

.disabled = false,

117

.next = *plugin_list };

118

104

perror("malloc");

105

exit(EXIT_FAILURE);

106

}

107

new_plugin->name = name;

119

108

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

120

109

if (new_plugin->argv == NULL){

121

free(new_plugin);

122

return NULL;

110

perror("malloc");

111

exit(EXIT_FAILURE);

123

112

}

124

new_plugin->argv[0] = copy_name;

113

new_plugin->argv[0] = name;

125

114

new_plugin->argv[1] = NULL;

126

127

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

128

if(new_plugin->environ == NULL){

129

free(new_plugin->argv);

130

free(new_plugin);

131

return NULL;

132

}

133

new_plugin->environ[0] = NULL;

115

new_plugin->argc = 1;

116

new_plugin->disabled = false;

117

new_plugin->next = *plugin_list;

134

118

/* Append the new plugin to the list */

135

119

*plugin_list = new_plugin;

136

120

return new_plugin;

137

121

}

138

122

139

/* Helper function for add_argument and add_environment */

140

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

141

int *len){

142

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

143

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

144

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

145

/* Malloc check */

146

if(*array == NULL){

147

return false;

148

}

149

/* Make a copy of the new string */

150

char *copy = strdup(new);

151

if(copy == NULL){

152

return false;

153

}

154

/* Insert the copy */

155

(*array)[*len] = copy;

156

(*len)++;

157

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

158

(*array)[*len] = NULL;

159

return true;

160

}

161

162

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

163

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

164

if(p == NULL){

165

return false;

166

}

167

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

168

}

169

170

/* Add to a plugin's environment */

171

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

172

if(p == NULL){

173

return false;

174

}

175

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

176

}

177

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

}

178

133

179

134

/*

180

135

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

216

171

}

217

172

218

173

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

219

ssize_t ret;

220

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

221

length--;

222

}

223

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

174

size_t ret;

175

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

224

176

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

225

177

length - written));

226

178

if(ret < 0){

230

182

return true;

231

183

}

232

184

233

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

234

if (argv == NULL){

235

*argc = 1;

236

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

237

if(argv == NULL){

238

return NULL;

239

}

240

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

241

argv[1] = NULL;

242

}

243

*argc += 1;

244

argv = realloc(argv, sizeof(char *)

245

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

246

if(argv == NULL){

247

return NULL;

248

}

249

argv[*argc-1] = arg;

250

argv[*argc] = NULL;

251

return argv;

252

}

253

254

static void free_plugin_list(plugin *plugin_list){

255

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

256

next = plugin_list->next;

257

free(plugin_list->name);

258

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

259

free(*arg);

260

}

261

free(plugin_list->argv);

262

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

263

free(*env);

264

}

265

free(plugin_list->environ);

266

free(plugin_list);

267

}

268

}

269

270

185

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

271

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

272

const char *argfile = ARGFILE;

273

FILE *conffp;

186

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

274

187

size_t d_name_len;

275

188

DIR *dir = NULL;

276

189

struct dirent *dirst;

284

197

struct sigaction old_sigchld_action;

285

198

struct sigaction sigchld_action = { .sa_handler = handle_sigchld,

286

199

.sa_flags = SA_NOCLDSTOP };

287

char **custom_argv = NULL;

288

int custom_argc = 0;

200

char *plus_options = NULL;

201

char **plus_argv = NULL;

289

202

290

203

/* Establish a signal handler */

291

204

sigemptyset(&sigchld_action.sa_mask);

292

205

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

293

if(ret == -1){

206

if(ret < 0){

294

207

perror("sigaddset");

295

exitstatus = EXIT_FAILURE;

296

goto fallback;

208

exit(EXIT_FAILURE);

297

209

}

298

210

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

299

if(ret == -1){

211

if(ret < 0){

300

212

perror("sigaction");

301

exitstatus = EXIT_FAILURE;

302

goto fallback;

213

exit(EXIT_FAILURE);

303

214

}

304

215

305

216

/* The options we understand. */

307

218

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

308

219

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

309

220

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

310

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

311

.arg = "VAR=value",

312

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

313

221

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

314

222

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

315

223

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

316

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

317

.arg = "PLUGIN:ENV=value",

318

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

319

224

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

320

225

.arg = "PLUGIN",

321

226

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

340

245

switch (key) {

341

246

case 'g':

342

247

if (arg != NULL){

343

char *p;

344

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

345

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

346

continue;

347

}

348

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

349

perror("add_argument");

350

return ARGP_ERR_UNKNOWN;

351

}

352

}

353

}

354

break;

355

case 'e':

356

if(arg == NULL){

357

break;

358

}

359

{

360

char *envdef = strdup(arg);

361

if(envdef == NULL){

362

break;

363

}

364

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

365

perror("add_environment");

366

}

248

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

249

do{

250

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

251

p = strtok(NULL, ",");

252

} while (p != NULL);

367

253

}

368

254

break;

369

255

case 'o':

370

256

if (arg != NULL){

371

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

372

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

373

break;

374

}

375

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

376

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

377

break;

378

}

379

if(opt != NULL){

380

char *p;

381

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

382

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

383

continue;

384

}

385

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

386

perror("add_argument");

387

return ARGP_ERR_UNKNOWN;

388

}

389

}

390

}

391

}

392

break;

393

case 'f':

394

if(arg == NULL){

395

break;

396

}

397

{

398

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

399

if(envdef == NULL){

400

break;

401

}

402

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

403

if(p_name == NULL){

404

break;

405

}

406

envdef++;

407

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

408

perror("add_environment");

257

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

258

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

259

if(p != NULL){

260

p = strtok(p, ",");

261

do{

262

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

263

p = strtok(NULL, ",");

264

} while (p != NULL);

409

265

}

410

266

}

411

267

break;

412

268

case 'd':

413

269

if (arg != NULL){

414

plugin *p = getplugin(arg, plugins);

415

if(p == NULL){

416

return ARGP_ERR_UNKNOWN;

417

}

418

p->disabled = true;

270

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

419

271

}

420

272

break;

421

273

case 128:

431

283

debug = true;

432

284

break;

433

285

case ARGP_KEY_ARG:

434

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

286

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

287

argp_usage (state);

288

}

289

plus_options = arg;

435

290

break;

436

291

case ARGP_KEY_END:

437

292

break;

449

304

450

305

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

451

306

if (ret == ARGP_ERR_UNKNOWN){

452

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

307

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

453

308

exitstatus = EXIT_FAILURE;

454

goto fallback;

309

goto end;

455

310

}

456

457

conffp = fopen(argfile, "r");

458

if(conffp != NULL){

459

char *org_line = NULL;

460

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

461

size_t size = 0;

462

ssize_t sret;

463

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

464

const char comment_delim[] = "#";

465

466

while(true){

467

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

468

if(sret == -1){

469

break;

470

}

471

472

line = org_line;

473

arg = strsep(&line, comment_delim);

474

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

475

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

476

continue;

477

}

478

new_arg = strdup(p);

479

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

480

if (custom_argv == NULL){

481

perror("add_to_argv");

482

exitstatus = EXIT_FAILURE;

483

goto fallback;

484

}

485

}

486

}

487

free(org_line);

488

} else{

489

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

490

not affect opening plugins */

491

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

492

perror("fopen");

311

312

if(plus_options){

313

/* This is a mangled argument in the form of

314

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

315

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

316

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

317

const char delims[] = "+";

318

char *arg;

319

int new_argc = 1;

320

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

321

if(plus_argv == NULL){

322

perror("malloc");

493

323

exitstatus = EXIT_FAILURE;

494

goto fallback;

495

}

496

}

497

498

if(custom_argv != NULL){

499

custom_argv[0] = argv[0];

500

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

324

goto end;

325

}

326

plus_argv[0] = argv[0];

327

plus_argv[1] = NULL;

328

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

329

while(arg != NULL){

330

new_argc++;

331

plus_argv = realloc(plus_argv, sizeof(char *)

332

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

333

if(plus_argv == NULL){

334

perror("realloc");

335

exitstatus = EXIT_FAILURE;

336

goto end;

337

}

338

plus_argv[new_argc-1] = arg;

339

plus_argv[new_argc] = NULL;

340

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

341

}

342

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

501

343

if (ret == ARGP_ERR_UNKNOWN){

502

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

344

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

503

345

exitstatus = EXIT_FAILURE;

504

goto fallback;

346

goto end;

505

347

}

506

348

}

507

349

512

354

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

513

355

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

514

356

}

515

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

516

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

517

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

518

}

519

357

}

520

358

}

521

359

533

371

if(dir == NULL){

534

372

perror("Could not open plugin dir");

535

373

exitstatus = EXIT_FAILURE;

536

goto fallback;

374

goto end;

537

375

}

538

376

539

377

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

544

382

if(ret < 0){

545

383

perror("set_cloexec_flag");

546

384

exitstatus = EXIT_FAILURE;

547

goto fallback;

385

goto end;

548

386

}

549

387

}

550

388

}

559

397

if (errno == EBADF){

560

398

perror("readdir");

561

399

exitstatus = EXIT_FAILURE;

562

goto fallback;

400

goto end;

563

401

}

564

402

break;

565

403

}

610

448

continue;

611

449

}

612

450

}

613

614

char *filename;

615

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

616

if(ret < 0){

617

perror("asprintf");

618

continue;

451

452

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

453

if (filename == NULL){

454

perror("malloc");

455

exitstatus = EXIT_FAILURE;

456

goto end;

619

457

}

458

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

459

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

460

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

620

461

621

462

ret = stat(filename, &st);

622

463

if (ret == -1){

623

464

perror("stat");

624

free(filename);

625

continue;

465

exitstatus = EXIT_FAILURE;

466

goto end;

626

467

}

627

468

628

469

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

633

474

free(filename);

634

475

continue;

635

476

}

636

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

637

if(p == NULL){

638

perror("getplugin");

639

free(filename);

640

continue;

641

}

642

if(p->disabled){

477

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

643

478

if(debug){

644

479

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

645

480

dirst->d_name);

647

482

free(filename);

648

483

continue;

649

484

}

485

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

650

486

{

651

487

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

652

488

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

653

if(g != NULL){

654

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

655

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

656

perror("add_argument");

657

}

658

}

659

/* Add global environment variables */

660

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

661

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

662

perror("add_environment");

663

}

664

}

665

}

666

}

667

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

668

using execve and need to duplicate the environment from this

669

process, too. */

670

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

671

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

672

char *copy = strdup(*e);

673

if(copy == NULL){

674

perror("strdup");

675

continue;

676

}

677

if(not add_environment(p, copy)){

678

perror("add_environment");

679

}

680

}

681

}

682

489

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

490

addargument(p, *a);

491

}

492

}

683

493

int pipefd[2];

684

494

ret = pipe(pipefd);

685

495

if (ret == -1){

686

496

perror("pipe");

687

497

exitstatus = EXIT_FAILURE;

688

goto fallback;

498

goto end;

689

499

}

690

500

ret = set_cloexec_flag(pipefd[0]);

691

501

if(ret < 0){

692

502

perror("set_cloexec_flag");

693

503

exitstatus = EXIT_FAILURE;

694

goto fallback;

504

goto end;

695

505

}

696

506

ret = set_cloexec_flag(pipefd[1]);

697

507

if(ret < 0){

698

508

perror("set_cloexec_flag");

699

509

exitstatus = EXIT_FAILURE;

700

goto fallback;

510

goto end;

701

511

}

702

512

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

703

513

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

704

514

if(ret < 0){

705

515

perror("sigprocmask");

706

516

exitstatus = EXIT_FAILURE;

707

goto fallback;

517

goto end;

708

518

}

709

519

// Starting a new process to be watched

710

520

pid_t pid = fork();

711

if(pid == -1){

712

perror("fork");

713

exitstatus = EXIT_FAILURE;

714

goto fallback;

715

}

716

521

if(pid == 0){

717

522

/* this is the child process */

718

523

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

737

542

above and must now close it manually here. */

738

543

closedir(dir);

739

544

}

740

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

741

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

742

perror("execv");

743

_exit(EXIT_FAILURE);

744

}

745

} else {

746

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

747

perror("execve");

748

_exit(EXIT_FAILURE);

749

}

545

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

546

perror("execv");

547

_exit(EXIT_FAILURE);

750

548

}

751

549

/* no return */

752

550

}

761

559

perror("sigprocmask");

762

560

}

763

561

exitstatus = EXIT_FAILURE;

764

goto fallback;

562

goto end;

765

563

}

766

564

767

565

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

775

573

if(ret < 0){

776

574

perror("sigprocmask");

777

575

exitstatus = EXIT_FAILURE;

778

goto fallback;

576

goto end;

779

577

}

780

578

781

579

FD_SET(new_process->fd, &rfds_all);

786

584

787

585

}

788

586

789

free_plugin_list(plugin_list);

587

/* Free the plugin list */

588

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

589

next = plugin_list->next;

590

free(plugin_list->argv);

591

free(plugin_list);

592

}

790

593

791

594

closedir(dir);

792

595

dir = NULL;

802

605

if (select_ret == -1){

803

606

perror("select");

804

607

exitstatus = EXIT_FAILURE;

805

goto fallback;

608

goto end;

806

609

}

807

610

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

808

611

from one of them */

834

637

if(ret < 0){

835

638

perror("sigprocmask");

836

639

exitstatus = EXIT_FAILURE;

837

goto fallback;

640

goto end;

838

641

}

839

642

/* Delete this process entry from the list */

840

643

if(process_list == proc){

869

672

perror("print_out_password");

870

673

exitstatus = EXIT_FAILURE;

871

674

}

872

goto fallback;

675

goto end;

873

676

}

874

677

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

875

678

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

883

686

if (proc->buffer == NULL){

884

687

perror("malloc");

885

688

exitstatus = EXIT_FAILURE;

886

goto fallback;

689

goto end;

887

690

}

888

691

proc->buffer_size += BUFFER_SIZE;

889

692

}

902

705

}

903

706

}

904

707

}

905

906

907

fallback:

908

708

909

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

910

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

709

if(process_list == NULL){

911

710

bool bret;

912

711

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

913

712

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

915

714

if(not bret){

916

715

perror("print_out_password");

917

716

exitstatus = EXIT_FAILURE;

717

goto end;

718

}

719

bret = print_out_password("\n", 1);

720

if(not bret){

721

perror("print_out_password");

722

exitstatus = EXIT_FAILURE;

918

723

}

919

724

}

725

726

end:

920

727

921

728

/* Restore old signal handler */

922

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

923

if(ret == -1){

924

perror("sigaction");

925

exitstatus = EXIT_FAILURE;

926

}

927

928

if(custom_argv != NULL){

929

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

930

free(*arg);

931

}

932

free(custom_argv);

933

}

934

free_plugin_list(plugin_list);

729

sigaction(SIGCHLD, &old_sigchld_action, NULL);

730

731

free(plus_argv);

732

733

/* Free the plugin list */

734

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

735

next = plugin_list->next;

736

free(plugin_list->argv);

737

free(plugin_list);

738

}

935

739

936

740

if(dir != NULL){

937

741

closedir(dir);