/mandos/release : revision 237.7.755

1

/* -*- mode: c; coding: utf-8; after-save-hook: (lambda () (let* ((find-build-directory (lambda (try-directory &optional base-directory) (let ((base-directory (or base-directory try-directory))) (cond ((equal try-directory "/") base-directory) ((file-readable-p (concat (file-name-as-directory try-directory) "Makefile")) try-directory) ((funcall find-build-directory (directory-file-name (file-name-directory try-directory)) base-directory)))))) (build-directory (funcall find-build-directory (buffer-file-name))) (local-build-directory (if (fboundp 'file-local-name) (file-local-name build-directory) (or (file-remote-p build-directory 'localname) build-directory))) (command (file-relative-name (file-name-sans-extension (buffer-file-name)) build-directory))) (pcase (progn (if (get-buffer "*Test*") (kill-buffer "*Test*")) (process-file-shell-command (let ((qbdir (shell-quote-argument local-build-directory)) (qcmd (shell-quote-argument command))) (format "cd %s && CFLAGS=-Werror make --silent %s && %s --test --verbose" qbdir qcmd qcmd)) nil "*Test*")) (0 (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w)))) (_ (with-current-buffer "*Test*" (compilation-mode) (cd-absolute build-directory)) (display-buffer "*Test*" '(display-buffer-in-side-window)))))); -*- */

2

/*

3

* Mandos password agent - Simple password agent to run Mandos client

4

*

5

6

7

*

8

* This file is part of Mandos.

9

*

10

* Mandos is free software: you can redistribute it and/or modify it

11

* under the terms of the GNU General Public License as published by

12

* the Free Software Foundation, either version 3 of the License, or

13

* (at your option) any later version.

14

*

15

* Mandos is distributed in the hope that it will be useful, but

16

* WITHOUT ANY WARRANTY; without even the implied warranty of

17

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

18

* General Public License for more details.

19

*

20

* You should have received a copy of the GNU General Public License

21

* along with Mandos. If not, see <http://www.gnu.org/licenses/>.

22

*

23

* Contact the authors at <mandos@recompile.se>.

24

*/

25

26

#define _GNU_SOURCE

27

#include <inttypes.h> /* uintmax_t, PRIuMAX, PRIdMAX,

28

intmax_t, uint32_t, SCNx32,

29

SCNuMAX, SCNxMAX */

30

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

31

#include <sys/types.h> /* pid_t, uid_t, gid_t, getuid(),

32

getpid() */

33

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

34

#include <signal.h> /* struct sigaction, sigset_t,

35

sigemptyset(), sigaddset(),

36

SIGCHLD, pthread_sigmask(),

37

SIG_BLOCK, SIG_SETMASK, SA_RESTART,

38

SA_NOCLDSTOP, sigfillset(), kill(),

39

SIGTERM, sigdelset(), SIGKILL,

40

NSIG, sigismember(), SA_ONSTACK,

41

SIG_DFL, SIG_IGN, SIGINT, SIGQUIT,

42

SIGHUP, SIGSTOP, SIG_UNBLOCK */

43

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

44

malloc(), free(), strtoumax(),

45

realloc(), setenv(), calloc(),

46

mkdtemp(), mkostemp() */

47

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

48

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

49

#include <sysexits.h> /* EX_USAGE, EX_OSERR, EX_OSFILE */

50

#include <errno.h> /* errno, error_t, EACCES,

51

ENAMETOOLONG, ENOENT, ENOTDIR,

52

ENOMEM, EEXIST, ECHILD, EPERM,

53

EAGAIN, EINTR, ENOBUFS, EADDRINUSE,

54

ECONNREFUSED, ECONNRESET,

55

ETOOMANYREFS, EMSGSIZE, EBADF,

56

EINVAL */

57

#include <string.h> /* strdup(), memcpy(),

58

explicit_bzero(), memset(),

59

strcmp(), strlen(), strncpy(),

60

memcmp(), basename() */

61

#include <argz.h> /* argz_create(), argz_count(),

62

argz_extract(), argz_next(),

63

argz_add() */

64

#include <sys/epoll.h> /* epoll_create1(), EPOLL_CLOEXEC,

65

epoll_ctl(), EPOLL_CTL_ADD,

66

struct epoll_event, EPOLLIN,

67

EPOLLRDHUP, EPOLLOUT,

68

epoll_pwait() */

69

#include <time.h> /* struct timespec, clock_gettime(),

70

CLOCK_MONOTONIC */

71

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

72

OPTION_ALIAS, struct argp_state,

73

ARGP_ERR_UNKNOWN, ARGP_KEY_ARGS,

74

struct argp, argp_parse(),

75

ARGP_NO_EXIT */

76

#include <stdint.h> /* SIZE_MAX */

77

#include <unistd.h> /* uid_t, gid_t, close(), pipe2(),

78

fork(), _exit(), dup2(),

79

STDOUT_FILENO, setresgid(),

80

setresuid(), execv(), ssize_t,

81

read(), dup3(), getuid(), dup(),

82

STDERR_FILENO, pause(), write(),

83

rmdir(), unlink(), getpid() */

84

#include <sys/mman.h> /* munlock(), mlock() */

85

#include <fcntl.h> /* O_CLOEXEC, O_NONBLOCK, fcntl(),

86

F_GETFD, F_GETFL, FD_CLOEXEC,

87

open(), O_WRONLY, O_NOCTTY,

88

O_RDONLY, O_NOFOLLOW */

89

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

90

WEXITSTATUS() */

91

#include <limits.h> /* PIPE_BUF, NAME_MAX, INT_MAX */

92

#include <sys/inotify.h> /* inotify_init1(), IN_NONBLOCK,

93

IN_CLOEXEC, inotify_add_watch(),

94

IN_CLOSE_WRITE, IN_MOVED_TO,

95

IN_MOVED_FROM, IN_DELETE,

96

IN_EXCL_UNLINK, IN_ONLYDIR,

97

struct inotify_event */

98

#include <fnmatch.h> /* fnmatch(), FNM_FILE_NAME */

99

#include <stdio.h> /* asprintf(), FILE, stderr, fopen(),

100

fclose(), getline(), sscanf(),

101

feof(), ferror(), rename(),

102

fdopen(), fprintf(), fscanf() */

103

#include <glib.h> /* GKeyFile, g_key_file_free(), g_key_file_new(),

104

GError, g_key_file_load_from_file(),

105

G_KEY_FILE_NONE, TRUE, G_FILE_ERROR_NOENT,

106

g_key_file_get_string(), guint64,

107

g_key_file_get_uint64(),

108

G_KEY_FILE_ERROR_KEY_NOT_FOUND, gconstpointer,

109

g_assert_true(), g_assert_nonnull(),

110

g_assert_null(), g_assert_false(),

111

g_assert_cmpint(), g_assert_cmpuint(),

112

g_test_skip(), g_assert_cmpstr(),

113

g_test_init(), g_test_add(), g_test_run(),

114

GOptionContext, g_option_context_new(),

115

g_option_context_set_help_enabled(), FALSE,

116

g_option_context_set_ignore_unknown_options(),

117

gboolean, GOptionEntry, G_OPTION_ARG_NONE,

118

g_option_context_add_main_entries(),

119

g_option_context_parse(),

120

g_option_context_free(), g_error() */

121

#include <sys/un.h> /* struct sockaddr_un, SUN_LEN */

122

#include <sys/socket.h> /* AF_LOCAL, socket(), PF_LOCAL,

123

SOCK_DGRAM, SOCK_NONBLOCK,

124

SOCK_CLOEXEC, connect(),

125

struct sockaddr, socklen_t,

126

shutdown(), SHUT_RD, send(),

127

MSG_NOSIGNAL, bind(), recv(),

128

socketpair() */

129

#include <glob.h> /* globfree(), glob_t, glob(),

130

GLOB_ERR, GLOB_NOSORT, GLOB_MARK,

131

GLOB_ABORTED, GLOB_NOMATCH,

132

GLOB_NOSPACE */

133

134

/* End of includes */

135

136

/* Start of declarations of private types and functions */

137

138

/* microseconds of CLOCK_MONOTONIC absolute time; 0 means unset */

139

typedef uintmax_t mono_microsecs;

140

141

/* "task_queue" - A queue of tasks to be run */

142

typedef struct {

143

struct task_struct *tasks; /* Tasks in this queue */

144

size_t length; /* Number of tasks */

145

/* Memory allocated for "tasks", in bytes */

146

size_t allocated;

147

/* Time when this queue should be run, at the latest */

148

mono_microsecs next_run;

149

} __attribute__((designated_init)) task_queue;

150

151

/* "task_func" - A function type for task functions

152

153

I.e. functions for the code which runs when a task is run, all have

154

this type */

155

typedef void (task_func) (const struct task_struct,

156

task_queue *const)

157

__attribute__((nonnull));

158

159

/* "buffer" - A data buffer for a growing array of bytes

160

161

Used for the "password" variable */

162

typedef struct {

163

char *data;

164

size_t length;

165

size_t allocated;

166

} __attribute__((designated_init)) buffer;

167

168

/* "string_set" - A set type which can contain strings

169

170

Used by the "cancelled_filenames" variable */

171

typedef struct {

172

char *argz; /* Do not access these except in */

173

size_t argz_len; /* the string_set_* functions */

174

} __attribute__((designated_init)) string_set;

175

176

/* "task_context" - local variables for tasks

177

178

This data structure distinguishes between different tasks which are

179

using the same function. This data structure is passed to every

180

task function when each task is run.

181

182

Note that not every task uses every struct member. */

183

typedef struct task_struct {

184

task_func *const func; /* The function run by this task */

185

char *const question_filename; /* The question file */

186

const pid_t pid; /* Mandos client process ID */

187

const int epoll_fd; /* The epoll set file descriptor */

188

bool *const quit_now; /* Set to true on fatal errors */

189

const int fd; /* General purpose file descriptor */

190

bool *const mandos_client_exited; /* Set true when client exits */

191

buffer *const password; /* As read from client process */

192

bool *const password_is_read; /* "password" is done growing */

193

char *filename; /* General purpose file name */

194

/* A set of strings of all the file names of questions which have

195

been cancelled for any reason; tasks pertaining to these question

196

files should not be run */

197

string_set *const cancelled_filenames;

198

const mono_microsecs notafter; /* "NotAfter" from question file */

199

/* Updated before each queue run; is compared with queue.next_run */

200

const mono_microsecs *const current_time;

201

} __attribute__((designated_init)) task_context;

202

203

/* Declare all our functions here so we can define them in any order

204

below. Note: test functions are *not* declared here, they are

205

declared in the test section. */

206

__attribute__((warn_unused_result))

207

static bool should_only_run_tests(int *, char **[]);

208

__attribute__((warn_unused_result, cold))

209

static bool run_tests(int, char *[]);

210

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

211

__attribute__((warn_unused_result, malloc))

212

task_queue *create_queue(void);

213

__attribute__((nonnull, warn_unused_result))

214

bool add_to_queue(task_queue *const, const task_context);

215

__attribute__((nonnull))

216

void cleanup_task(const task_context *const);

217

__attribute__((nonnull))

218

void cleanup_queue(task_queue *const *const);

219

__attribute__((pure, nonnull, warn_unused_result))

220

bool queue_has_question(const task_queue *const);

221

__attribute__((nonnull))

222

void cleanup_close(const int *const);

223

__attribute__((nonnull))

224

void cleanup_string(char *const *const);

225

__attribute__((nonnull))

226

void cleanup_buffer(buffer *const);

227

__attribute__((pure, nonnull, warn_unused_result))

228

bool string_set_contains(const string_set, const char *const);

229

__attribute__((nonnull, warn_unused_result))

230

bool string_set_add(string_set *const, const char *const);

231

__attribute__((nonnull))

232

void string_set_clear(string_set *);

233

void string_set_swap(string_set *const, string_set *const);

234

__attribute__((nonnull, warn_unused_result))

235

bool start_mandos_client(task_queue *const, const int, bool *const,

236

bool *const, buffer *const, bool *const,

237

const struct sigaction *const,

238

const sigset_t, const char *const,

239

const uid_t, const gid_t,

240

const char *const *const);

241

__attribute__((nonnull))

242

task_func wait_for_mandos_client_exit;

243

__attribute__((nonnull))

244

task_func read_mandos_client_output;

245

__attribute__((warn_unused_result))

246

bool add_inotify_dir_watch(task_queue *const, const int, bool *const,

247

buffer *const, const char *const,

248

string_set *, const mono_microsecs *const,

249

bool *const, bool *const);

250

__attribute__((nonnull))

251

task_func read_inotify_event;

252

__attribute__((nonnull))

253

task_func open_and_parse_question;

254

__attribute__((nonnull))

255

task_func cancel_old_question;

256

__attribute__((nonnull))

257

task_func connect_question_socket;

258

__attribute__((nonnull))

259

task_func send_password_to_socket;

260

__attribute__((warn_unused_result))

261

bool add_existing_questions(task_queue *const, const int,

262

buffer *const, string_set *,

263

const mono_microsecs *const,

264

bool *const, bool *const,

265

const char *const);

266

__attribute__((nonnull, warn_unused_result))

267

bool wait_for_event(const int, const mono_microsecs,

268

const mono_microsecs);

269

bool run_queue(task_queue **const, string_set *const, bool *const);

270

bool clear_all_fds_from_epoll_set(const int);

271

mono_microsecs get_current_time(void);

272

__attribute__((nonnull, warn_unused_result))

273

bool setup_signal_handler(struct sigaction *const);

274

__attribute__((nonnull))

275

bool restore_signal_handler(const struct sigaction *const);

276

__attribute__((nonnull, warn_unused_result))

277

bool block_sigchld(sigset_t *const);

278

__attribute__((nonnull))

279

bool restore_sigmask(const sigset_t *const);

280

__attribute__((nonnull))

281

bool parse_arguments(int, char *[], const bool, char **, char **,

282

uid_t *const , gid_t *const, char **, size_t *);

283

284

/* End of declarations of private types and functions */

285

286

/* Start of "main" section; this section LACKS TESTS!

287

288

Code here should be as simple as possible. */

289

290

/* These are required to be global by Argp */

291

const char *argp_program_version = "password-agent " VERSION;

292

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

293

294

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

295

296

/* If the --test option is passed, skip all normal operations and

297

instead only run the run_tests() function, which also does all

298

its own option parsing, so we don't have to do anything here. */

299

if(should_only_run_tests(&argc, &argv)){

300

if(run_tests(argc, argv)){

301

return EXIT_SUCCESS; /* All tests successful */

302

}

303

return EXIT_FAILURE; /* Some test(s) failed */

304

}

305

306

__attribute__((cleanup(cleanup_string)))

307

char *agent_directory = NULL;

308

309

__attribute__((cleanup(cleanup_string)))

310

char *helper_directory = NULL;

311

312

uid_t user = 0;

313

gid_t group = 0;

314

315

__attribute__((cleanup(cleanup_string)))

316

char *mandos_argz = NULL;

317

size_t mandos_argz_length = 0;

318

319

if(not parse_arguments(argc, argv, true, &agent_directory,

320

&helper_directory, &user, &group,

321

&mandos_argz, &mandos_argz_length)){

322

/* This should never happen, since "true" is passed as the third

323

argument to parse_arguments() above, which should make

324

argp_parse() call exit() if any parsing error occurs. */

325

error(EX_USAGE, errno, "Failed to parse arguments");

326

}

327

328

const char default_agent_directory[] = "/run/systemd/ask-password";

329

const char default_helper_directory[]

330

= "/lib/mandos/plugin-helpers";

331

const char *const default_argv[]

332

= {"/lib/mandos/plugins.d/mandos-client", NULL };

333

334

/* Set variables to default values if unset */

335

if(agent_directory == NULL){

336

agent_directory = strdup(default_agent_directory);

337

if(agent_directory == NULL){

338

error(EX_OSERR, errno, "Failed strdup()");

339

}

340

}

341

if(helper_directory == NULL){

342

helper_directory = strdup(default_helper_directory);

343

if(helper_directory == NULL){

344

error(EX_OSERR, errno, "Failed strdup()");

345

}

346

}

347

if(user == 0){

348

user = 65534; /* nobody */

349

}

350

if(group == 0){

351

group = 65534; /* nogroup */

352

}

353

/* If parse_opt did not create an argz vector, create one with

354

default values */

355

if(mandos_argz == NULL){

356

#ifdef __GNUC__

357

#pragma GCC diagnostic push

358

/* argz_create() takes a non-const argv for some unknown reason -

359

argz_create() isn't modifying the strings, just copying them.

360

Therefore, this cast to non-const should be safe. */

361

#pragma GCC diagnostic ignored "-Wcast-qual"

362

#endif

363

errno = argz_create((char *const *)default_argv, &mandos_argz,

364

&mandos_argz_length);

365

#ifdef __GNUC__

366

#pragma GCC diagnostic pop

367

#endif

368

if(errno != 0){

369

error(EX_OSERR, errno, "Failed argz_create()");

370

}

371

}

372

/* Use argz vector to create a normal argv, usable by execv() */

373

374

char **mandos_argv = malloc((argz_count(mandos_argz,

375

mandos_argz_length)

376

+ 1) * sizeof(char *));

377

if(mandos_argv == NULL){

378

error_t saved_errno = errno;

379

free(mandos_argz);

380

error(EX_OSERR, saved_errno, "Failed malloc()");

381

}

382

argz_extract(mandos_argz, mandos_argz_length, mandos_argv);

383

384

sigset_t orig_sigmask;

385

if(not block_sigchld(&orig_sigmask)){

386

return EX_OSERR;

387

}

388

389

struct sigaction old_sigchld_action;

390

if(not setup_signal_handler(&old_sigchld_action)){

391

return EX_OSERR;

392

}

393

394

mono_microsecs current_time = 0;

395

396

bool mandos_client_exited = false;

397

bool quit_now = false;

398

__attribute__((cleanup(cleanup_close)))

399

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

400

if(epoll_fd < 0){

401

error(EX_OSERR, errno, "Failed to create epoll set fd");

402

}

403

__attribute__((cleanup(cleanup_queue)))

404

task_queue *queue = create_queue();

405

if(queue == NULL){

406

error(EX_OSERR, errno, "Failed to create task queue");

407

}

408

409

__attribute__((cleanup(cleanup_buffer)))

410

buffer password = {};

411

bool password_is_read = false;

412

413

__attribute__((cleanup(string_set_clear)))

414

string_set cancelled_filenames = {};

415

416

/* Add tasks to queue */

417

if(not start_mandos_client(queue, epoll_fd, &mandos_client_exited,

418

&quit_now, &password, &password_is_read,

419

&old_sigchld_action, orig_sigmask,

420

helper_directory, user, group,

421

(const char *const *)mandos_argv)){

422

return EX_OSERR; /* Error has already been printed */

423

}

424

/* These variables were only for start_mandos_client() and are not

425

needed anymore */

426

free(mandos_argv);

427

free(mandos_argz);

428

mandos_argz = NULL;

429

if(not add_inotify_dir_watch(queue, epoll_fd, &quit_now, &password,

430

agent_directory, &cancelled_filenames,

431

&current_time, &mandos_client_exited,

432

&password_is_read)){

433

switch(errno){ /* Error has already been printed */

434

case EACCES:

435

case ENAMETOOLONG:

436

case ENOENT:

437

case ENOTDIR:

438

return EX_OSFILE;

439

default:

440

return EX_OSERR;

441

}

442

}

443

if(not add_existing_questions(queue, epoll_fd, &password,

444

&cancelled_filenames, &current_time,

445

&mandos_client_exited,

446

&password_is_read, agent_directory)){

447

return EXIT_FAILURE; /* Error has already been printed */

448

}

449

450

/* Run queue */

451

do {

452

current_time = get_current_time();

453

if(not wait_for_event(epoll_fd, queue->next_run, current_time)){

454

const error_t saved_errno = errno;

455

error(EXIT_FAILURE, saved_errno, "Failure while waiting for"

456

" events");

457

}

458

459

current_time = get_current_time();

460

if(not run_queue(&queue, &cancelled_filenames, &quit_now)){

461

const error_t saved_errno = errno;

462

error(EXIT_FAILURE, saved_errno, "Failure while running queue");

463

}

464

465

/* When no tasks about questions are left in the queue, break out

466

of the loop (and implicitly exit the program) */

467

} while(queue_has_question(queue));

468

469

restore_signal_handler(&old_sigchld_action);

470

restore_sigmask(&orig_sigmask);

471

472

return EXIT_SUCCESS;

473

}

474

475

__attribute__((warn_unused_result))

476

mono_microsecs get_current_time(void){

477

struct timespec currtime;

478

if(clock_gettime(CLOCK_MONOTONIC, &currtime) != 0){

479

error(0, errno, "Failed to get current time");

480

return 0;

481

}

482

return ((mono_microsecs)currtime.tv_sec * 1000000) /* seconds */

483

+ ((mono_microsecs)currtime.tv_nsec / 1000); /* nanoseconds */

484

}

485

486

/* End of "main" section */

487

488

/* Start of regular code section; ALL this code has tests */

489

490

__attribute__((nonnull))

491

bool parse_arguments(int argc, char *argv[], const bool exit_failure,

492

char **agent_directory, char **helper_directory,

493

uid_t *const user, gid_t *const group,

494

char **mandos_argz, size_t *mandos_argz_length){

495

496

const struct argp_option options[] = {

497

{ .name="agent-directory",.key='d', .arg="DIRECTORY",

498

.doc="Systemd password agent directory" },

499

{ .name="helper-directory",.key=128, .arg="DIRECTORY",

500

.doc="Mandos Client password helper directory" },

501

{ .name="plugin-helper-dir", .key=129, /* From plugin-runner */

502

.flags=OPTION_HIDDEN | OPTION_ALIAS },

503

{ .name="user", .key='u', .arg="USERID",

504

.doc="User ID the Mandos Client will use as its unprivileged"

505

" user" },

506

{ .name="userid", .key=130, /* From plugin--runner */

507

.flags=OPTION_HIDDEN | OPTION_ALIAS },

508

{ .name="group", .key='g', .arg="GROUPID",

509

.doc="Group ID the Mandos Client will use as its unprivileged"

510

" group" },

511

{ .name="groupid", .key=131, /* From plugin--runner */

512

.flags=OPTION_HIDDEN | OPTION_ALIAS },

513

{ .name="test", .key=255, /* See should_only_run_tests() */

514

.doc="Skip normal operation, and only run self-tests. See"

515

" --test --help.", .group=10, },

516

{ NULL },

517

};

518

519

__attribute__((nonnull(3)))

520

error_t parse_opt(int key, char *arg, struct argp_state *state){

521

errno = 0;

522

switch(key){

523

case 'd': /* --agent-directory */

524

*agent_directory = strdup(arg);

525

break;

526

case 128: /* --helper-directory */

527

case 129: /* --plugin-helper-dir */

528

*helper_directory = strdup(arg);

529

break;

530

case 'u': /* --user */

531

case 130: /* --userid */

532

{

533

char *tmp;

534

uintmax_t tmp_id = 0;

535

errno = 0;

536

tmp_id = (uid_t)strtoumax(arg, &tmp, 10);

537

if(errno != 0 or tmp == arg or *tmp != '\0'

538

or tmp_id != (uid_t)tmp_id or (uid_t)tmp_id == 0){

539

return ARGP_ERR_UNKNOWN;

540

}

541

*user = (uid_t)tmp_id;

542

errno = 0;

543

break;

544

}

545

case 'g': /* --group */

546

case 131: /* --groupid */

547

{

548

char *tmp;

549

uintmax_t tmp_id = 0;

550

errno = 0;

551

tmp_id = (uid_t)strtoumax(arg, &tmp, 10);

552

if(errno != 0 or tmp == arg or *tmp != '\0'

553

or tmp_id != (gid_t)tmp_id or (gid_t)tmp_id == 0){

554

return ARGP_ERR_UNKNOWN;

555

}

556

*group = (gid_t)tmp_id;

557

errno = 0;

558

break;

559

}

560

case ARGP_KEY_ARGS:

561

/* Copy arguments into argz vector */

562

return argz_create(state->argv + state->next, mandos_argz,

563

mandos_argz_length);

564

default:

565

return ARGP_ERR_UNKNOWN;

566

}

567

return errno;

568

}

569

570

const struct argp argp = {

571

.options=options,

572

.parser=parse_opt,

573

.args_doc="[MANDOS_CLIENT [OPTION...]]\n--test",

574

.doc = "Mandos password agent -- runs Mandos client as a"

575

" systemd password agent",

576

};

577

578

errno = argp_parse(&argp, argc, argv,

579

exit_failure ? 0 : ARGP_NO_EXIT, NULL, NULL);

580

581

return errno == 0;

582

}

583

584

__attribute__((nonnull, warn_unused_result))

585

bool block_sigchld(sigset_t *const orig_sigmask){

586

sigset_t sigchld_sigmask;

587

if(sigemptyset(&sigchld_sigmask) < 0){

588

error(0, errno, "Failed to empty signal set");

589

return false;

590

}

591

if(sigaddset(&sigchld_sigmask, SIGCHLD) < 0){

592

error(0, errno, "Failed to add SIGCHLD to signal set");

593

return false;

594

}

595

if(pthread_sigmask(SIG_BLOCK, &sigchld_sigmask, orig_sigmask) != 0){

596

error(0, errno, "Failed to block SIGCHLD signal");

597

return false;

598

}

599

return true;

600

}

601

602

__attribute__((nonnull, warn_unused_result, const))

603

bool restore_sigmask(const sigset_t *const orig_sigmask){

604

if(pthread_sigmask(SIG_SETMASK, orig_sigmask, NULL) != 0){

605

error(0, errno, "Failed to restore blocked signals");

606

return false;

607

}

608

return true;

609

}

610

611

__attribute__((nonnull, warn_unused_result))

612

bool setup_signal_handler(struct sigaction *const old_sigchld_action){

613

struct sigaction sigchld_action = {

614

.sa_handler=handle_sigchld,

615

.sa_flags=SA_RESTART | SA_NOCLDSTOP,

616

};

617

/* Set all signals in "sa_mask" struct member; this makes all

618

signals automatically blocked during signal handler */

619

if(sigfillset(&sigchld_action.sa_mask) != 0){

620

error(0, errno, "Failed to do sigfillset()");

621

return false;

622

}

623

if(sigaction(SIGCHLD, &sigchld_action, old_sigchld_action) != 0){

624

error(0, errno, "Failed to set SIGCHLD signal handler");

625

return false;

626

}

627

return true;

628

}

629

630

__attribute__((nonnull, warn_unused_result))

631

bool restore_signal_handler(const struct sigaction *const

632

old_sigchld_action){

633

if(sigaction(SIGCHLD, old_sigchld_action, NULL) != 0){

634

error(0, errno, "Failed to restore signal handler");

635

return false;

636

}

637

return true;

638

}

639

640

__attribute__((warn_unused_result, malloc))

641

task_queue *create_queue(void){

642

task_queue *queue = malloc(sizeof(task_queue));

643

if(queue){

644

queue->tasks = NULL;

645

queue->length = 0;

646

queue->allocated = 0;

647

queue->next_run = 0;

648

}

649

return queue;

650

}

651

652

__attribute__((nonnull, warn_unused_result))

653

bool add_to_queue(task_queue *const queue, const task_context task){

654

if((queue->length + 1) > (SIZE_MAX / sizeof(task_context))){

655

/* overflow */

656

error(0, ENOMEM, "Failed to allocate %" PRIuMAX

657

" tasks for queue->tasks", (uintmax_t)(queue->length + 1));

658

errno = ENOMEM;

659

return false;

660

}

661

const size_t needed_size = sizeof(task_context)*(queue->length + 1);

662

if(needed_size > (queue->allocated)){

663

task_context *const new_tasks = realloc(queue->tasks,

664

needed_size);

665

if(new_tasks == NULL){

666

error(0, errno, "Failed to allocate %" PRIuMAX

667

" bytes for queue->tasks", (uintmax_t)needed_size);

668

return false;

669

}

670

queue->tasks = new_tasks;

671

queue->allocated = needed_size;

672

}

673

/* Using memcpy here is necessary because doing */

674

/* queue->tasks[queue->length++] = task; */

675

/* would violate const-ness of task members */

676

memcpy(&(queue->tasks[queue->length++]), &task,

677

sizeof(task_context));

678

return true;

679

}

680

681

__attribute__((nonnull))

682

void cleanup_task(const task_context *const task){

683

const error_t saved_errno = errno;

684

/* free and close all task data */

685

free(task->question_filename);

686

if(task->filename != task->question_filename){

687

free(task->filename);

688

}

689

if(task->pid > 0){

690

kill(task->pid, SIGTERM);

691

}

692

if(task->fd > 0){

693

close(task->fd);

694

}

695

errno = saved_errno;

696

}

697

698

__attribute__((nonnull))

699

void free_queue(task_queue *const queue){

700

free(queue->tasks);

701

free(queue);

702

}

703

704

__attribute__((nonnull))

705

void cleanup_queue(task_queue *const *const queue){

706

if(*queue == NULL){

707

return;

708

}

709

for(size_t i = 0; i < (*queue)->length; i++){

710

const task_context *const task = ((*queue)->tasks)+i;

711

cleanup_task(task);

712

}

713

free_queue(*queue);

714

}

715

716

__attribute__((pure, nonnull, warn_unused_result))

717

bool queue_has_question(const task_queue *const queue){

718

for(size_t i=0; i < queue->length; i++){

719

if(queue->tasks[i].question_filename != NULL){

720

return true;

721

}

722

}

723

return false;

724

}

725

726

__attribute__((nonnull))

727

void cleanup_close(const int *const fd){

728

const error_t saved_errno = errno;

729

close(*fd);

730

errno = saved_errno;

731

}

732

733

__attribute__((nonnull))

734

void cleanup_string(char *const *const ptr){

735

free(*ptr);

736

}

737

738

__attribute__((nonnull))

739

void cleanup_buffer(buffer *buf){

740

if(buf->allocated > 0){

741

#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)

742

explicit_bzero(buf->data, buf->allocated);

743

#else

744

memset(buf->data, '\0', buf->allocated);

745

#endif

746

}

747

if(buf->data != NULL){

748

if(munlock(buf->data, buf->allocated) != 0){

749

error(0, errno, "Failed to unlock memory of old buffer");

750

}

751

free(buf->data);

752

buf->data = NULL;

753

}

754

buf->length = 0;

755

buf->allocated = 0;

756

}

757

758

__attribute__((pure, nonnull, warn_unused_result))

759

bool string_set_contains(const string_set set, const char *const str){

760

for(const char *s = set.argz; s != NULL and set.argz_len > 0;

761

s = argz_next(set.argz, set.argz_len, s)){

762

if(strcmp(s, str) == 0){

763

return true;

764

}

765

}

766

return false;

767

}

768

769

__attribute__((nonnull, warn_unused_result))

770

bool string_set_add(string_set *const set, const char *const str){

771

if(string_set_contains(*set, str)){

772

return true;

773

}

774

error_t error = argz_add(&set->argz, &set->argz_len, str);

775

if(error == 0){

776

return true;

777

}

778

errno = error;

779

return false;

780

}

781

782

__attribute__((nonnull))

783

void string_set_clear(string_set *set){

784

free(set->argz);

785

set->argz = NULL;

786

set->argz_len = 0;

787

}

788

789

__attribute__((nonnull))

790

void string_set_swap(string_set *const set1, string_set *const set2){

791

/* Swap contents of two string sets */

792

{

793

char *const tmp_argz = set1->argz;

794

set1->argz = set2->argz;

795

set2->argz = tmp_argz;

796

}

797

{

798

const size_t tmp_argz_len = set1->argz_len;

799

set1->argz_len = set2->argz_len;

800

set2->argz_len = tmp_argz_len;

801

}

802

}

803

804

__attribute__((nonnull, warn_unused_result))

805

bool start_mandos_client(task_queue *const queue,

806

const int epoll_fd,

807

bool *const mandos_client_exited,

808

bool *const quit_now, buffer *const password,

809

bool *const password_is_read,

810

const struct sigaction *const

811

old_sigchld_action, const sigset_t sigmask,

812

const char *const helper_directory,

813

const uid_t user, const gid_t group,

814

const char *const *const argv){

815

int pipefds[2];

816

if(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK) != 0){

817

error(0, errno, "Failed to pipe2(..., O_CLOEXEC | O_NONBLOCK)");

818

return false;

819

}

820

821

const pid_t pid = fork();

822

if(pid == 0){

823

if(not restore_signal_handler(old_sigchld_action)){

824

_exit(EXIT_FAILURE);

825

}

826

if(not restore_sigmask(&sigmask)){

827

_exit(EXIT_FAILURE);

828

}

829

if(close(pipefds[0]) != 0){

830

error(0, errno, "Failed to close() parent pipe fd");

831

_exit(EXIT_FAILURE);

832

}

833

if(dup2(pipefds[1], STDOUT_FILENO) == -1){

834

error(0, errno, "Failed to dup2() pipe fd to stdout");

835

_exit(EXIT_FAILURE);

836

}

837

if(close(pipefds[1]) != 0){

838

error(0, errno, "Failed to close() old child pipe fd");

839

_exit(EXIT_FAILURE);

840

}

841

if(setenv("MANDOSPLUGINHELPERDIR", helper_directory, 1) != 0){

842

error(0, errno, "Failed to setenv(\"MANDOSPLUGINHELPERDIR\","

843

" \"%s\", 1)", helper_directory);

844

_exit(EXIT_FAILURE);

845

}

846

if(group != 0 and setresgid(group, 0, 0) == -1){

847

error(0, errno, "Failed to setresgid(-1, %" PRIuMAX ", %"

848

PRIuMAX")", (uintmax_t)group, (uintmax_t)group);

849

_exit(EXIT_FAILURE);

850

}

851

if(user != 0 and setresuid(user, 0, 0) == -1){

852

error(0, errno, "Failed to setresuid(-1, %" PRIuMAX ", %"

853

PRIuMAX")", (uintmax_t)user, (uintmax_t)user);

854

_exit(EXIT_FAILURE);

855

}

856

#ifdef __GNUC__

857

#pragma GCC diagnostic push

858

/* For historical reasons, the "argv" argument to execv() is not

859

const, but it is safe to override this. */

860

#pragma GCC diagnostic ignored "-Wcast-qual"

861

#endif

862

execv(argv[0], (char **)argv);

863

#ifdef __GNUC__

864

#pragma GCC diagnostic pop

865

#endif

866

error(0, errno, "execv(\"%s\", ...) failed", argv[0]);

867

_exit(EXIT_FAILURE);

868

}

869

close(pipefds[1]);

870

871

if(not add_to_queue(queue, (task_context){

872

.func=wait_for_mandos_client_exit,

873

.pid=pid,

874

.mandos_client_exited=mandos_client_exited,

875

.quit_now=quit_now,

876

})){

877

error(0, errno, "Failed to add wait_for_mandos_client to queue");

878

close(pipefds[0]);

879

return false;

880

}

881

882

const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, pipefds[0],

883

&(struct epoll_event)

884

{ .events=EPOLLIN | EPOLLRDHUP });

885

if(ret != 0 and errno != EEXIST){

886

error(0, errno, "Failed to add file descriptor to epoll set");

887

close(pipefds[0]);

888

return false;

889

}

890

891

return add_to_queue(queue, (task_context){

892

.func=read_mandos_client_output,

893

.epoll_fd=epoll_fd,

894

.fd=pipefds[0],

895

.quit_now=quit_now,

896

.password=password,

897

.password_is_read=password_is_read,

898

});

899

}

900

901

__attribute__((nonnull))

902

void wait_for_mandos_client_exit(const task_context task,

903

task_queue *const queue){

904

const pid_t pid = task.pid;

905

bool *const mandos_client_exited = task.mandos_client_exited;

906

bool *const quit_now = task.quit_now;

907

908

int status;

909

switch(waitpid(pid, &status, WNOHANG)){

910

case 0: /* Not exited yet */

911

if(not add_to_queue(queue, task)){

912

error(0, errno, "Failed to add myself to queue");

913

*quit_now = true;

914

}

915

break;

916

case -1: /* Error */

917

error(0, errno, "waitpid(%" PRIdMAX ") failed", (intmax_t)pid);

918

if(errno != ECHILD){

919

kill(pid, SIGTERM);

920

}

921

*quit_now = true;

922

break;

923

default: /* Has exited */

924

*mandos_client_exited = true;

925

if((not WIFEXITED(status))

926

or (WEXITSTATUS(status) != EXIT_SUCCESS)){

927

error(0, 0, "Mandos client failed or was killed");

928

*quit_now = true;

929

}

930

}

931

}

932

933

__attribute__((nonnull))

934

void read_mandos_client_output(const task_context task,

935

task_queue *const queue){

936

buffer *const password = task.password;

937

bool *const quit_now = task.quit_now;

938

bool *const password_is_read = task.password_is_read;

939

const int fd = task.fd;

940

const int epoll_fd = task.epoll_fd;

941

942

const size_t new_potential_size = (password->length + PIPE_BUF);

943

if(password->allocated < new_potential_size){

944

char *const new_buffer = calloc(new_potential_size, 1);

945

if(new_buffer == NULL){

946

error(0, errno, "Failed to allocate %" PRIuMAX

947

" bytes for password", (uintmax_t)new_potential_size);

948

*quit_now = true;

949

close(fd);

950

return;

951

}

952

if(mlock(new_buffer, new_potential_size) != 0){

953

/* Warn but do not treat as fatal error */

954

if(errno != EPERM and errno != ENOMEM){

955

error(0, errno, "Failed to lock memory for password");

956

}

957

}

958

if(password->length > 0){

959

memcpy(new_buffer, password->data, password->length);

960

#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)

961

explicit_bzero(password->data, password->allocated);

962

#else

963

memset(password->data, '\0', password->allocated);

964

#endif

965

}

966

if(password->data != NULL){

967

if(munlock(password->data, password->allocated) != 0){

968

error(0, errno, "Failed to unlock memory of old buffer");

969

}

970

free(password->data);

971

}

972

password->data = new_buffer;

973

password->allocated = new_potential_size;

974

}

975

976

const ssize_t read_length = read(fd, password->data

977

+ password->length, PIPE_BUF);

978

979

if(read_length == 0){ /* EOF */

980

*password_is_read = true;

981

close(fd);

982

return;

983

}

984

if(read_length < 0 and errno != EAGAIN){ /* Actual error */

985

error(0, errno, "Failed to read password from Mandos client");

986

*quit_now = true;

987

close(fd);

988

return;

989

}

990

if(read_length > 0){ /* Data has been read */

991

password->length += (size_t)read_length;

992

}

993

994

/* Either data was read, or EAGAIN was indicated, meaning no data

995

available yet */

996

997

/* Re-add the fd to the epoll set */

998

const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,

999

&(struct epoll_event)

1000

{ .events=EPOLLIN | EPOLLRDHUP });

1001

if(ret != 0 and errno != EEXIST){

1002

error(0, errno, "Failed to re-add file descriptor to epoll set");

1003

*quit_now = true;

1004

close(fd);

1005

return;

1006

}

1007

1008

/* Re-add myself to the queue */

1009

if(not add_to_queue(queue, task)){

1010

error(0, errno, "Failed to add myself to queue");

1011

*quit_now = true;

1012

close(fd);

1013

}

1014

}

1015

1016

__attribute__((nonnull, warn_unused_result))

1017

bool add_inotify_dir_watch(task_queue *const queue,

1018

const int epoll_fd, bool *const quit_now,

1019

buffer *const password,

1020

const char *const dir,

1021

string_set *cancelled_filenames,

1022

const mono_microsecs *const current_time,

1023

bool *const mandos_client_exited,

1024

bool *const password_is_read){

1025

const int fd = inotify_init1(IN_NONBLOCK | IN_CLOEXEC);

1026

if(fd == -1){

1027

error(0, errno, "Failed to create inotify instance");

1028

return false;

1029

}

1030

1031

if(inotify_add_watch(fd, dir, IN_CLOSE_WRITE | IN_MOVED_TO

1032

| IN_MOVED_FROM| IN_DELETE | IN_EXCL_UNLINK

1033

| IN_ONLYDIR)

1034

== -1){

1035

error(0, errno, "Failed to create inotify watch on %s", dir);

1036

return false;

1037

}

1038

1039

/* Add the inotify fd to the epoll set */

1040

const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,

1041

&(struct epoll_event)

1042

{ .events=EPOLLIN | EPOLLRDHUP });

1043

if(ret != 0 and errno != EEXIST){

1044

error(0, errno, "Failed to add file descriptor to epoll set");

1045

close(fd);

1046

return false;

1047

}

1048

1049

const task_context read_inotify_event_task = {

1050

.func=read_inotify_event,

1051

.epoll_fd=epoll_fd,

1052

.quit_now=quit_now,

1053

.password=password,

1054

.fd=fd,

1055

.filename=strdup(dir),

1056

.cancelled_filenames=cancelled_filenames,

1057

.current_time=current_time,

1058

.mandos_client_exited=mandos_client_exited,

1059

.password_is_read=password_is_read,

1060

};

1061

if(read_inotify_event_task.filename == NULL){

1062

error(0, errno, "Failed to strdup(\"%s\")", dir);

1063

close(fd);

1064

return false;

1065

}

1066

1067

return add_to_queue(queue, read_inotify_event_task);

1068

}

1069

1070

__attribute__((nonnull))

1071

void read_inotify_event(const task_context task,

1072

task_queue *const queue){

1073

const int fd = task.fd;

1074

const int epoll_fd = task.epoll_fd;

1075

char *const filename = task.filename;

1076

bool *quit_now = task.quit_now;

1077

buffer *const password = task.password;

1078

string_set *const cancelled_filenames = task.cancelled_filenames;

1079

const mono_microsecs *const current_time = task.current_time;

1080

bool *const mandos_client_exited = task.mandos_client_exited;

1081

bool *const password_is_read = task.password_is_read;

1082

1083

/* "sufficient to read at least one event." - inotify(7) */

1084

const size_t ievent_size = (sizeof(struct inotify_event)

1085

+ NAME_MAX + 1);

1086

struct {

1087

struct inotify_event event;

1088

char name_buffer[NAME_MAX + 1];

1089

} ievent_buffer;

1090

struct inotify_event *const ievent = &ievent_buffer.event;

1091

1092

const ssize_t read_length = read(fd, ievent, ievent_size);

1093

if(read_length == 0){ /* EOF */

1094

error(0, 0, "Got EOF from inotify fd for directory %s", filename);

1095

*quit_now = true;

1096

cleanup_task(&task);

1097

return;

1098

}

1099

if(read_length < 0 and errno != EAGAIN){ /* Actual error */

1100

error(0, errno, "Failed to read from inotify fd for directory %s",

1101

filename);

1102

*quit_now = true;

1103

cleanup_task(&task);

1104

return;

1105

}

1106

if(read_length > 0 /* Data has been read */

1107

and fnmatch("ask.*", ievent->name, FNM_FILE_NAME) == 0){

1108

char *question_filename = NULL;

1109

const ssize_t question_filename_length

1110

= asprintf(&question_filename, "%s/%s", filename, ievent->name);

1111

if(question_filename_length < 0){

1112

error(0, errno, "Failed to create file name from directory name"

1113

" %s and file name %s", filename, ievent->name);

1114

} else {

1115

if(ievent->mask & (IN_CLOSE_WRITE | IN_MOVED_TO)){

1116

if(not add_to_queue(queue, (task_context){

1117

.func=open_and_parse_question,

1118

.epoll_fd=epoll_fd,

1119

.question_filename=question_filename,

1120

.filename=question_filename,

1121

.password=password,

1122

.cancelled_filenames=cancelled_filenames,

1123

.current_time=current_time,

1124

.mandos_client_exited=mandos_client_exited,

1125

.password_is_read=password_is_read,

1126

})){

1127

error(0, errno, "Failed to add open_and_parse_question task"

1128

" for file name %s to queue", filename);

1129

} else {

1130

/* Force the added task (open_and_parse_question) to run

1131

immediately */

1132

queue->next_run = 1;

1133

}

1134

} else if(ievent->mask & (IN_MOVED_FROM | IN_DELETE)){

1135

if(not string_set_add(cancelled_filenames,

1136

question_filename)){

1137

error(0, errno, "Could not add question %s to"

1138

" cancelled_questions", question_filename);

1139

*quit_now = true;

1140

free(question_filename);

1141

cleanup_task(&task);

1142

return;

1143

}

1144

free(question_filename);

1145

}

1146

}

1147

}

1148

1149

/* Either data was read, or EAGAIN was indicated, meaning no data

1150

available yet */

1151

1152

/* Re-add myself to the queue */

1153

if(not add_to_queue(queue, task)){

1154

error(0, errno, "Failed to re-add read_inotify_event(%s) to"

1155

" queue", filename);

1156

*quit_now = true;

1157

cleanup_task(&task);

1158

return;

1159

}

1160

1161

/* Re-add the fd to the epoll set */

1162

const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,

1163

&(struct epoll_event)

1164

{ .events=EPOLLIN | EPOLLRDHUP });

1165

if(ret != 0 and errno != EEXIST){

1166

error(0, errno, "Failed to re-add inotify file descriptor %d for"

1167

" directory %s to epoll set", fd, filename);

1168

/* Force the added task (read_inotify_event) to run again, at most

1169

one second from now */

1170

if((queue->next_run == 0)

1171

or (queue->next_run > (*current_time + 1000000))){

1172

queue->next_run = *current_time + 1000000;

1173

}

1174

}

1175

}

1176

1177

__attribute__((nonnull))

1178

void open_and_parse_question(const task_context task,

1179

task_queue *const queue){

1180

__attribute__((cleanup(cleanup_string)))

1181

char *question_filename = task.question_filename;

1182

const int epoll_fd = task.epoll_fd;

1183

buffer *const password = task.password;

1184

string_set *const cancelled_filenames = task.cancelled_filenames;

1185

const mono_microsecs *const current_time = task.current_time;

1186

bool *const mandos_client_exited = task.mandos_client_exited;

1187

bool *const password_is_read = task.password_is_read;

1188

1189

/* We use the GLib "Key-value file parser" functions to parse the

1190

question file. See <https://www.freedesktop.org/wiki/Software

1191

/systemd/PasswordAgents/> for specification of contents */

1192

__attribute__((nonnull))

1193

void cleanup_g_key_file(GKeyFile **key_file){

1194

if(*key_file != NULL){

1195

g_key_file_free(*key_file);

1196

}

1197

}

1198

1199

__attribute__((cleanup(cleanup_g_key_file)))

1200

GKeyFile *key_file = g_key_file_new();

1201

if(key_file == NULL){

1202

error(0, errno, "Failed g_key_file_new() for \"%s\"",

1203

question_filename);

1204

return;

1205

}

1206

GError *glib_error = NULL;

1207

if(g_key_file_load_from_file(key_file, question_filename,

1208

G_KEY_FILE_NONE, &glib_error) != TRUE){

1209

/* If a file was removed, we should ignore it, so */

1210

/* only show error message if file actually existed */

1211

if(glib_error->code != G_FILE_ERROR_NOENT){

1212

error(0, 0, "Failed to load question data from file \"%s\": %s",

1213

question_filename, glib_error->message);

1214

}

1215

return;

1216

}

1217

1218

__attribute__((cleanup(cleanup_string)))

1219

char *socket_name = g_key_file_get_string(key_file, "Ask",

1220

"Socket",

1221

&glib_error);

1222

if(socket_name == NULL){

1223

error(0, 0, "Question file \"%s\" did not contain \"Socket\": %s",

1224

question_filename, glib_error->message);

1225

return;

1226

}

1227

1228

if(strlen(socket_name) == 0){

1229

error(0, 0, "Question file \"%s\" had empty \"Socket\" value",

1230

question_filename);

1231

return;

1232

}

1233

1234

const guint64 pid = g_key_file_get_uint64(key_file, "Ask", "PID",

1235

&glib_error);

1236

if(glib_error != NULL){

1237

error(0, 0, "Question file \"%s\" contained bad \"PID\": %s",

1238

question_filename, glib_error->message);

1239

return;

1240

}

1241

1242

if((pid != (guint64)((pid_t)pid))

1243

or (kill((pid_t)pid, 0) != 0)){

1244

error(0, 0, "PID %" PRIuMAX " in question file \"%s\" is bad or"

1245

" does not exist", (uintmax_t)pid, question_filename);

1246

return;

1247

}

1248

1249

guint64 notafter = g_key_file_get_uint64(key_file, "Ask",

1250

"NotAfter", &glib_error);

1251

if(glib_error != NULL){

1252

if(glib_error->code != G_KEY_FILE_ERROR_KEY_NOT_FOUND){

1253

error(0, 0, "Question file \"%s\" contained bad \"NotAfter\":"

1254

" %s", question_filename, glib_error->message);

1255

}

1256

notafter = 0;

1257

}

1258

if(notafter != 0){

1259

if(queue->next_run == 0 or (queue->next_run > notafter)){

1260

queue->next_run = notafter;

1261

}

1262

if(*current_time >= notafter){

1263

return;

1264

}

1265

}

1266

1267

const task_context connect_question_socket_task = {

1268

.func=connect_question_socket,

1269

.question_filename=strdup(question_filename),

1270

.epoll_fd=epoll_fd,

1271

.password=password,

1272

.filename=strdup(socket_name),

1273

.cancelled_filenames=task.cancelled_filenames,

1274

.mandos_client_exited=mandos_client_exited,

1275

.password_is_read=password_is_read,

1276

.current_time=current_time,

1277

};

1278

if(connect_question_socket_task.question_filename == NULL

1279

or connect_question_socket_task.filename == NULL

1280

or not add_to_queue(queue, connect_question_socket_task)){

1281

error(0, errno, "Failed to add connect_question_socket for socket"

1282

" %s (from \"%s\") to queue", socket_name,

1283

question_filename);

1284

cleanup_task(&connect_question_socket_task);

1285

return;

1286

}

1287

/* Force the added task (connect_question_socket) to run

1288

immediately */

1289

queue->next_run = 1;

1290

1291

if(notafter > 0){

1292

char *const dup_filename = strdup(question_filename);

1293

const task_context cancel_old_question_task = {

1294

.func=cancel_old_question,

1295

.question_filename=dup_filename,

1296

.notafter=notafter,

1297

.filename=dup_filename,

1298

.cancelled_filenames=cancelled_filenames,

1299

.current_time=current_time,

1300

};

1301

if(cancel_old_question_task.question_filename == NULL

1302

or not add_to_queue(queue, cancel_old_question_task)){

1303

error(0, errno, "Failed to add cancel_old_question for file "

1304

"\"%s\" to queue", question_filename);

1305

cleanup_task(&cancel_old_question_task);

1306

return;

1307

}

1308

}

1309

}

1310

1311

__attribute__((nonnull))

1312

void cancel_old_question(const task_context task,

1313

task_queue *const queue){

1314

char *const question_filename = task.question_filename;

1315

string_set *const cancelled_filenames = task.cancelled_filenames;

1316

const mono_microsecs notafter = task.notafter;

1317

const mono_microsecs *const current_time = task.current_time;

1318

1319

if(*current_time >= notafter){

1320

if(not string_set_add(cancelled_filenames, question_filename)){

1321

error(0, errno, "Failed to cancel question for file %s",

1322

question_filename);

1323

}

1324

cleanup_task(&task);

1325

return;

1326

}

1327

1328

if(not add_to_queue(queue, task)){

1329

error(0, errno, "Failed to add cancel_old_question for file "

1330

"%s to queue", question_filename);

1331

cleanup_task(&task);

1332

return;

1333

}

1334

1335

if((queue->next_run == 0) or (queue->next_run > notafter)){

1336

queue->next_run = notafter;

1337

}

1338

}

1339

1340

__attribute__((nonnull))

1341

void connect_question_socket(const task_context task,

1342

task_queue *const queue){

1343

char *const question_filename = task.question_filename;

1344

char *const filename = task.filename;

1345

const int epoll_fd = task.epoll_fd;

1346

buffer *const password = task.password;

1347

string_set *const cancelled_filenames = task.cancelled_filenames;

1348

bool *const mandos_client_exited = task.mandos_client_exited;

1349

bool *const password_is_read = task.password_is_read;

1350

const mono_microsecs *const current_time = task.current_time;

1351

1352

struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };

1353

1354

if(sizeof(sock_name.sun_path) <= strlen(filename)){

1355

error(0, 0, "Socket filename is larger than"

1356

" sizeof(sockaddr_un.sun_path); %" PRIuMAX ": \"%s\"",

1357

(uintmax_t)sizeof(sock_name.sun_path), filename);

1358

if(not string_set_add(cancelled_filenames, question_filename)){

1359

error(0, errno, "Failed to cancel question for file %s",

1360

question_filename);

1361

}

1362

cleanup_task(&task);

1363

return;

1364

}

1365

1366

const int fd = socket(PF_LOCAL, SOCK_DGRAM

1367

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0);

1368

if(fd < 0){

1369

error(0, errno,

1370

"Failed to create socket(PF_LOCAL, SOCK_DGRAM, 0)");

1371

if(not add_to_queue(queue, task)){

1372

error(0, errno, "Failed to add connect_question_socket for file"

1373

" \"%s\" and socket \"%s\" to queue", question_filename,

1374

filename);

1375

cleanup_task(&task);

1376

} else {

1377

/* Force the added task (connect_question_socket) to run

1378

immediately */

1379

queue->next_run = 1;

1380

}

1381

return;

1382

}

1383

1384

strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));

1385

if(connect(fd, (struct sockaddr *)&sock_name,

1386

(socklen_t)SUN_LEN(&sock_name)) != 0){

1387

error(0, errno, "Failed to connect socket to \"%s\"", filename);

1388

if(not add_to_queue(queue, task)){

1389

error(0, errno, "Failed to add connect_question_socket for file"

1390

" \"%s\" and socket \"%s\" to queue", question_filename,

1391

filename);

1392

cleanup_task(&task);

1393

} else {

1394

/* Force the added task (connect_question_socket) to run again,

1395

at most one second from now */

1396

if((queue->next_run == 0)

1397

or (queue->next_run > (*current_time + 1000000))){

1398

queue->next_run = *current_time + 1000000;

1399

}

1400

}

1401

return;

1402

}

1403

1404

/* Not necessary, but we can try, and merely warn on failure */

1405

if(shutdown(fd, SHUT_RD) != 0){

1406

error(0, errno, "Failed to shutdown reading from socket \"%s\"",

1407

filename);

1408

}

1409

1410

/* Add the fd to the epoll set */

1411

if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,

1412

&(struct epoll_event){ .events=EPOLLOUT })

1413

!= 0){

1414

error(0, errno, "Failed to add inotify file descriptor %d for"

1415

" socket %s to epoll set", fd, filename);

1416

if(not add_to_queue(queue, task)){

1417

error(0, errno, "Failed to add connect_question_socket for file"

1418

" \"%s\" and socket \"%s\" to queue", question_filename,

1419

filename);

1420

cleanup_task(&task);

1421

} else {

1422

/* Force the added task (connect_question_socket) to run again,

1423

at most one second from now */

1424

if((queue->next_run == 0)

1425

or (queue->next_run > (*current_time + 1000000))){

1426

queue->next_run = *current_time + 1000000;

1427

}

1428

}

1429

return;

1430

}

1431

1432

/* add task send_password_to_socket to queue */

1433

const task_context send_password_to_socket_task = {

1434

.func=send_password_to_socket,

1435

.question_filename=question_filename,

1436

.filename=filename,

1437

.epoll_fd=epoll_fd,

1438

.fd=fd,

1439

.password=password,

1440

.cancelled_filenames=cancelled_filenames,

1441

.mandos_client_exited=mandos_client_exited,

1442

.password_is_read=password_is_read,

1443

.current_time=current_time,

1444

};

1445

1446

if(not add_to_queue(queue, send_password_to_socket_task)){

1447

error(0, errno, "Failed to add send_password_to_socket for"

1448

" file \"%s\" and socket \"%s\" to queue",

1449

question_filename, filename);

1450

cleanup_task(&send_password_to_socket_task);

1451

}

1452

}

1453

1454

__attribute__((nonnull))

1455

void send_password_to_socket(const task_context task,

1456

task_queue *const queue){

1457

char *const question_filename=task.question_filename;

1458

char *const filename=task.filename;

1459

const int epoll_fd=task.epoll_fd;

1460

const int fd=task.fd;

1461

buffer *const password=task.password;

1462

string_set *const cancelled_filenames=task.cancelled_filenames;

1463

bool *const mandos_client_exited = task.mandos_client_exited;

1464

bool *const password_is_read = task.password_is_read;

1465

const mono_microsecs *const current_time = task.current_time;

1466

1467

if(*mandos_client_exited and *password_is_read){

1468

1469

const size_t send_buffer_length = password->length + 2;

1470

char *send_buffer = malloc(send_buffer_length);

1471

if(send_buffer == NULL){

1472

error(0, errno, "Failed to allocate send_buffer");

1473

} else {

1474

if(mlock(send_buffer, send_buffer_length) != 0){

1475

/* Warn but do not treat as fatal error */

1476

if(errno != EPERM and errno != ENOMEM){

1477

error(0, errno, "Failed to lock memory for password"

1478

" buffer");

1479

}

1480

}

1481

/* “[…] send a single datagram to the socket consisting of the

1482

password string either prefixed with "+" or with "-"

1483

depending on whether the password entry was successful or

1484

not. You may but don't have to include a final NUL byte in

1485

your message.

1486

1487

— <https://www.freedesktop.org/wiki/Software/systemd/

1488

PasswordAgents/> (Wed 08 Oct 2014 02:14:28 AM UTC)

1489

*/

1490

send_buffer[0] = '+'; /* Prefix with "+" */

1491

/* Always add an extra NUL */

1492

send_buffer[password->length + 1] = '\0';

1493

if(password->length > 0){

1494

memcpy(send_buffer + 1, password->data, password->length);

1495

}

1496

errno = 0;

1497

ssize_t ssret = send(fd, send_buffer, send_buffer_length,

1498

MSG_NOSIGNAL);

1499

const error_t saved_errno = errno;

1500

#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)

1501

explicit_bzero(send_buffer, send_buffer_length);

1502

#else

1503

memset(send_buffer, '\0', send_buffer_length);

1504

#endif

1505

if(munlock(send_buffer, send_buffer_length) != 0){

1506

error(0, errno, "Failed to unlock memory of send buffer");

1507

}

1508

free(send_buffer);

1509

if(ssret < 0 or ssret < (ssize_t)send_buffer_length){

1510

switch(saved_errno){

1511

case EINTR:

1512

case ENOBUFS:

1513

case ENOMEM:

1514

case EADDRINUSE:

1515

case ECONNREFUSED:

1516

case ECONNRESET:

1517

case ENOENT:

1518

case ETOOMANYREFS:

1519

case EAGAIN:

1520

/* Retry, below */

1521

break;

1522

case EMSGSIZE:

1523

error(0, 0, "Password of size %" PRIuMAX " is too big",

1524

(uintmax_t)password->length);

1525

#if __GNUC__ < 7

1526

/* FALLTHROUGH */

1527

#else

1528

__attribute__((fallthrough));

1529

#endif

1530

case 0:

1531

if(ssret >= 0 and ssret < (ssize_t)send_buffer_length){

1532

error(0, 0, "Password only partially sent to socket");

1533

}

1534

#if __GNUC__ < 7

1535

/* FALLTHROUGH */

1536

#else

1537

__attribute__((fallthrough));

1538

#endif

1539

default:

1540

error(0, saved_errno, "Failed to send() to socket %s",

1541

filename);

1542

if(not string_set_add(cancelled_filenames,

1543

question_filename)){

1544

error(0, errno, "Failed to cancel question for file %s",

1545

question_filename);

1546

}

1547

cleanup_task(&task);

1548

return;

1549

}

1550

} else {

1551

/* Success */

1552

cleanup_task(&task);

1553

return;

1554

}

1555

}

1556

}

1557

1558

/* We failed or are not ready yet; retry later */

1559

1560

if(not add_to_queue(queue, task)){

1561

error(0, errno, "Failed to add send_password_to_socket for"

1562

" file %s and socket %s to queue", question_filename,

1563

filename);

1564

cleanup_task(&task);

1565

}

1566

1567

/* Add the fd to the epoll set */

1568

if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,

1569

&(struct epoll_event){ .events=EPOLLOUT })

1570

!= 0){

1571

error(0, errno, "Failed to add socket file descriptor %d for"

1572

" socket %s to epoll set", fd, filename);

1573

/* Force the added task (send_password_to_socket) to run again, at

1574

most one second from now */

1575

if((queue->next_run == 0)

1576

or (queue->next_run > (*current_time + 1000000))){

1577

queue->next_run = *current_time + 1000000;

1578

}

1579

}

1580

}

1581

1582

__attribute__((warn_unused_result))

1583

bool add_existing_questions(task_queue *const queue,

1584

const int epoll_fd,

1585

buffer *const password,

1586

string_set *cancelled_filenames,

1587

const mono_microsecs *const current_time,

1588

bool *const mandos_client_exited,

1589

bool *const password_is_read,

1590

const char *const dirname){

1591

__attribute__((cleanup(cleanup_string)))

1592

char *dir_pattern = NULL;

1593

const int ret = asprintf(&dir_pattern, "%s/ask.*", dirname);

1594

if(ret < 0 or dir_pattern == NULL){

1595

error(0, errno, "Could not create glob pattern for directory %s",

1596

dirname);

1597

return false;

1598

}

1599

__attribute__((cleanup(globfree)))

1600

glob_t question_filenames = {};

1601

switch(glob(dir_pattern, GLOB_ERR | GLOB_NOSORT | GLOB_MARK,

1602

NULL, &question_filenames)){

1603

case GLOB_ABORTED:

1604

default:

1605

error(0, errno, "Failed to open directory %s", dirname);

1606

return false;

1607

case GLOB_NOMATCH:

1608

error(0, errno, "There are no question files in %s", dirname);

1609

return false;

1610

case GLOB_NOSPACE:

1611

error(0, errno, "Could not allocate memory for question file"

1612

" names in %s", dirname);

1613

#if __GNUC__ < 7

1614

/* FALLTHROUGH */

1615

#else

1616

__attribute__((fallthrough));

1617

#endif

1618

case 0:

1619

for(size_t i = 0; i < question_filenames.gl_pathc; i++){

1620

char *const question_filename = strdup(question_filenames

1621

.gl_pathv[i]);

1622

const task_context task = {

1623

.func=open_and_parse_question,

1624

.epoll_fd=epoll_fd,

1625

.question_filename=question_filename,

1626

.filename=question_filename,

1627

.password=password,

1628

.cancelled_filenames=cancelled_filenames,

1629

.current_time=current_time,

1630

.mandos_client_exited=mandos_client_exited,

1631

.password_is_read=password_is_read,

1632

};

1633

1634

if(question_filename == NULL

1635

or not add_to_queue(queue, task)){

1636

error(0, errno, "Failed to add open_and_parse_question for"

1637

" file %s to queue",

1638

question_filenames.gl_pathv[i]);

1639

free(question_filename);

1640

} else {

1641

queue->next_run = 1;

1642

}

1643

}

1644

return true;

1645

}

1646

}

1647

1648

__attribute__((nonnull, warn_unused_result))

1649

bool wait_for_event(const int epoll_fd,

1650

const mono_microsecs queue_next_run,

1651

const mono_microsecs current_time){

1652

__attribute__((const))

1653

int milliseconds_to_wait(const mono_microsecs currtime,

1654

const mono_microsecs nextrun){

1655

if(currtime >= nextrun){

1656

return 0;

1657

}

1658

const uintmax_t wait_time_ms = (nextrun - currtime) / 1000;

1659

if(wait_time_ms > (uintmax_t)INT_MAX){

1660

return INT_MAX;

1661

}

1662

return (int)wait_time_ms;

1663

}

1664

1665

const int wait_time_ms = milliseconds_to_wait(current_time,

1666

queue_next_run);

1667

1668

/* Prepare unblocking of SIGCHLD during epoll_pwait */

1669

sigset_t temporary_unblocked_sigmask;

1670

/* Get current signal mask */

1671

if(pthread_sigmask(-1, NULL, &temporary_unblocked_sigmask) != 0){

1672

return false;

1673

}

1674

/* Remove SIGCHLD from the signal mask */

1675

if(sigdelset(&temporary_unblocked_sigmask, SIGCHLD) != 0){

1676

return false;

1677

}

1678

struct epoll_event events[8]; /* Ignored */

1679

int ret = epoll_pwait(epoll_fd, events,

1680

sizeof(events) / sizeof(struct epoll_event),

1681

queue_next_run == 0 ? -1 : (int)wait_time_ms,

1682

&temporary_unblocked_sigmask);

1683

if(ret < 0 and errno != EINTR){

1684

error(0, errno, "Failed epoll_pwait(epfd=%d, ..., timeout=%d,"

1685

" ...", epoll_fd,

1686

queue_next_run == 0 ? -1 : (int)wait_time_ms);

1687

return false;

1688

}

1689

return clear_all_fds_from_epoll_set(epoll_fd);

1690

}

1691

1692

bool clear_all_fds_from_epoll_set(const int epoll_fd){

1693

/* Create a new empty epoll set */

1694

__attribute__((cleanup(cleanup_close)))

1695

const int new_epoll_fd = epoll_create1(EPOLL_CLOEXEC);

1696

if(new_epoll_fd < 0){

1697

return false;

1698

}

1699

/* dup3() the new epoll set fd over the old one, replacing it */

1700

if(dup3(new_epoll_fd, epoll_fd, O_CLOEXEC) < 0){

1701

return false;

1702

}

1703

return true;

1704

}

1705

1706

__attribute__((nonnull, warn_unused_result))

1707

bool run_queue(task_queue **const queue,

1708

string_set *const cancelled_filenames,

1709

bool *const quit_now){

1710

1711

task_queue *new_queue = create_queue();

1712

if(new_queue == NULL){

1713

return false;

1714

}

1715

1716

__attribute__((cleanup(string_set_clear)))

1717

string_set old_cancelled_filenames = {};

1718

string_set_swap(cancelled_filenames, &old_cancelled_filenames);

1719

1720

/* Declare i outside the for loop, since we might need i after the

1721

loop in case we aborted in the middle */

1722

size_t i;

1723

for(i=0; i < (*queue)->length and not *quit_now; i++){

1724

task_context *const task = &((*queue)->tasks[i]);

1725

const char *const question_filename = task->question_filename;

1726

/* Skip any task referencing a cancelled question filename */

1727

if(question_filename != NULL

1728

and string_set_contains(old_cancelled_filenames,

1729

question_filename)){

1730

cleanup_task(task);

1731

continue;

1732

}

1733

task->func(*task, new_queue);

1734

}

1735

1736

if(*quit_now){

1737

/* we might be in the middle of the queue, so clean up any

1738

remaining tasks in the current queue */

1739

for(; i < (*queue)->length; i++){

1740

cleanup_task(&((*queue)->tasks[i]));

1741

}

1742

free_queue(*queue);

1743

*queue = new_queue;

1744

new_queue = NULL;

1745

return false;

1746

}

1747

free_queue(*queue);

1748

*queue = new_queue;

1749

new_queue = NULL;

1750

1751

return true;

1752

}

1753

1754

/* End of regular code section */

1755

1756

/* Start of tests section; here are the tests for the above code */

1757

1758

/* This "fixture" data structure is used by the test setup and

1759

teardown functions */

1760

typedef struct {

1761

struct sigaction orig_sigaction;

1762

sigset_t orig_sigmask;

1763

} test_fixture;

1764

1765

static void test_setup(test_fixture *fixture,

1766

__attribute__((unused))

1767

gconstpointer user_data){

1768

g_assert_true(setup_signal_handler(&fixture->orig_sigaction));

1769

g_assert_true(block_sigchld(&fixture->orig_sigmask));

1770

}

1771

1772

static void test_teardown(test_fixture *fixture,

1773

__attribute__((unused))

1774

gconstpointer user_data){

1775

g_assert_true(restore_signal_handler(&fixture->orig_sigaction));

1776

g_assert_true(restore_sigmask(&fixture->orig_sigmask));

1777

}

1778

1779

/* Utility function used by tests to search queue for matching task */

1780

__attribute__((pure, nonnull, warn_unused_result))

1781

static task_context *find_matching_task(const task_queue *const queue,

1782

const task_context task){

1783

/* The argument "task" structure is a pattern to match; 0 in any

1784

member means any value matches, otherwise the value must match.

1785

The filename strings are compared by strcmp(), not by pointer. */

1786

for(size_t i = 0; i < queue->length; i++){

1787

task_context *const current_task = queue->tasks+i;

1788

/* Check all members of task_context, if set to a non-zero value.

1789

If a member does not match, continue to next task in queue */

1790

1791

/* task_func *const func */

1792

if(task.func != NULL and current_task->func != task.func){

1793

continue;

1794

}

1795

/* char *const question_filename; */

1796

if(task.question_filename != NULL

1797

and (current_task->question_filename == NULL

1798

or strcmp(current_task->question_filename,

1799

task.question_filename) != 0)){

1800

continue;

1801

}

1802

/* const pid_t pid; */

1803

if(task.pid != 0 and current_task->pid != task.pid){

1804

continue;

1805

}

1806

/* const int epoll_fd; */

1807

if(task.epoll_fd != 0

1808

and current_task->epoll_fd != task.epoll_fd){

1809

continue;

1810

}

1811

/* bool *const quit_now; */

1812

if(task.quit_now != NULL

1813

and current_task->quit_now != task.quit_now){

1814

continue;

1815

}

1816

/* const int fd; */

1817

if(task.fd != 0 and current_task->fd != task.fd){

1818

continue;

1819

}

1820

/* bool *const mandos_client_exited; */

1821

if(task.mandos_client_exited != NULL

1822

and current_task->mandos_client_exited

1823

!= task.mandos_client_exited){

1824

continue;

1825

}

1826

/* buffer *const password; */

1827

if(task.password != NULL

1828

and current_task->password != task.password){

1829

continue;

1830

}

1831

/* bool *const password_is_read; */

1832

if(task.password_is_read != NULL

1833

and current_task->password_is_read != task.password_is_read){

1834

continue;

1835

}

1836

/* char *filename; */

1837

if(task.filename != NULL

1838

and (current_task->filename == NULL

1839

or strcmp(current_task->filename, task.filename) != 0)){

1840

continue;

1841

}

1842

/* string_set *const cancelled_filenames; */

1843

if(task.cancelled_filenames != NULL

1844

and current_task->cancelled_filenames

1845

!= task.cancelled_filenames){

1846

continue;

1847

}

1848

/* const mono_microsecs notafter; */

1849

if(task.notafter != 0

1850

and current_task->notafter != task.notafter){

1851

continue;

1852

}

1853

/* const mono_microsecs *const current_time; */

1854

if(task.current_time != NULL

1855

and current_task->current_time != task.current_time){

1856

continue;

1857

}

1858

/* Current task matches all members; return it */

1859

return current_task;

1860

}

1861

/* No task in queue matches passed pattern task */

1862

return NULL;

1863

}

1864

1865

static void test_create_queue(__attribute__((unused))

1866

test_fixture *fixture,

1867

__attribute__((unused))

1868

gconstpointer user_data){

1869

__attribute__((cleanup(cleanup_queue)))

1870

task_queue *const queue = create_queue();

1871

g_assert_nonnull(queue);

1872

g_assert_null(queue->tasks);

1873

g_assert_true(queue->length == 0);

1874

g_assert_true(queue->next_run == 0);

1875

}

1876

1877

static task_func dummy_func;

1878

1879

static void test_add_to_queue(__attribute__((unused))

1880

test_fixture *fixture,

1881

__attribute__((unused))

1882

gconstpointer user_data){

1883

__attribute__((cleanup(cleanup_queue)))

1884

task_queue *queue = create_queue();

1885

g_assert_nonnull(queue);

1886

1887

g_assert_true(add_to_queue(queue,

1888

(task_context){ .func=dummy_func }));

1889

g_assert_true(queue->length == 1);

1890

g_assert_nonnull(queue->tasks);

1891

g_assert_true(queue->tasks[0].func == dummy_func);

1892

}

1893

1894

static void test_add_to_queue_overflow(__attribute__((unused))

1895

test_fixture *fixture,

1896

__attribute__((unused))

1897

gconstpointer user_data){

1898

__attribute__((cleanup(cleanup_queue)))

1899

task_queue *queue = create_queue();

1900

g_assert_nonnull(queue);

1901

g_assert_true(queue->length == 0);

1902

queue->length = SIZE_MAX / sizeof(task_context); /* fake max size */

1903

1904

FILE *real_stderr = stderr;

1905

FILE *devnull = fopen("/dev/null", "we");

1906

g_assert_nonnull(devnull);

1907

stderr = devnull;

1908

const bool ret = add_to_queue(queue,

1909

(task_context){ .func=dummy_func });

1910

g_assert_true(errno == ENOMEM);

1911

g_assert_false(ret);

1912

stderr = real_stderr;

1913

g_assert_cmpint(fclose(devnull), ==, 0);

1914

queue->length = 0; /* Restore real size */

1915

}

1916

1917

static void dummy_func(__attribute__((unused))

1918

const task_context task,

1919

__attribute__((unused))

1920

task_queue *const queue){

1921

}

1922

1923

static void test_queue_has_question_empty(__attribute__((unused))

1924

test_fixture *fixture,

1925

__attribute__((unused))

1926

gconstpointer user_data){

1927

__attribute__((cleanup(cleanup_queue)))

1928

task_queue *queue = create_queue();

1929

g_assert_nonnull(queue);

1930

g_assert_false(queue_has_question(queue));

1931

}

1932

1933

static void test_queue_has_question_false(__attribute__((unused))

1934

test_fixture *fixture,

1935

__attribute__((unused))

1936

gconstpointer user_data){

1937

__attribute__((cleanup(cleanup_queue)))

1938

task_queue *queue = create_queue();

1939

g_assert_nonnull(queue);

1940

g_assert_true(add_to_queue(queue,

1941

(task_context){ .func=dummy_func }));

1942

g_assert_false(queue_has_question(queue));

1943

}

1944

1945

static void test_queue_has_question_true(__attribute__((unused))

1946

test_fixture *fixture,

1947

__attribute__((unused))

1948

gconstpointer user_data){

1949

__attribute__((cleanup(cleanup_queue)))

1950

task_queue *queue = create_queue();

1951

g_assert_nonnull(queue);

1952

char *const question_filename

1953

= strdup("/nonexistent/question_filename");

1954

g_assert_nonnull(question_filename);

1955

task_context task = {

1956

.func=dummy_func,

1957

.question_filename=question_filename,

1958

};

1959

g_assert_true(add_to_queue(queue, task));

1960

g_assert_true(queue_has_question(queue));

1961

}

1962

1963

static void test_queue_has_question_false2(__attribute__((unused))

1964

test_fixture *fixture,

1965

__attribute__((unused))

1966

gconstpointer user_data){

1967

__attribute__((cleanup(cleanup_queue)))

1968

task_queue *queue = create_queue();

1969

g_assert_nonnull(queue);

1970

task_context task = { .func=dummy_func };

1971

g_assert_true(add_to_queue(queue, task));

1972

g_assert_true(add_to_queue(queue, task));

1973

g_assert_cmpint((int)queue->length, ==, 2);

1974

g_assert_false(queue_has_question(queue));

1975

}

1976

1977

static void test_queue_has_question_true2(__attribute__((unused))

1978

test_fixture *fixture,

1979

__attribute__((unused))

1980

gconstpointer user_data){

1981

__attribute__((cleanup(cleanup_queue)))

1982

task_queue *queue = create_queue();

1983

g_assert_nonnull(queue);

1984

task_context task1 = { .func=dummy_func };

1985

g_assert_true(add_to_queue(queue, task1));

1986

char *const question_filename

1987

= strdup("/nonexistent/question_filename");

1988

g_assert_nonnull(question_filename);

1989

task_context task2 = {

1990

.func=dummy_func,

1991

.question_filename=question_filename,

1992

};

1993

g_assert_true(add_to_queue(queue, task2));

1994

g_assert_cmpint((int)queue->length, ==, 2);

1995

g_assert_true(queue_has_question(queue));

1996

}

1997

1998

static void test_cleanup_buffer(__attribute__((unused))

1999

test_fixture *fixture,

2000

__attribute__((unused))

2001

gconstpointer user_data){

2002

buffer buf = {};

2003

2004

const size_t buffersize = 10;

2005

2006

buf.data = malloc(buffersize);

2007

g_assert_nonnull(buf.data);

2008

if(mlock(buf.data, buffersize) != 0){

2009

g_assert_true(errno == EPERM or errno == ENOMEM);

2010

}

2011

2012

cleanup_buffer(&buf);

2013

g_assert_null(buf.data);

2014

}

2015

2016

static

2017

void test_string_set_new_set_contains_nothing(__attribute__((unused))

2018

test_fixture *fixture,

2019

__attribute__((unused))

2020

gconstpointer

2021

user_data){

2022

__attribute__((cleanup(string_set_clear)))

2023

string_set set = {};

2024

g_assert_false(string_set_contains(set, "")); /* Empty string */

2025

g_assert_false(string_set_contains(set, "test_string"));

2026

}

2027

2028

static void

2029

test_string_set_with_added_string_contains_it(__attribute__((unused))

2030

test_fixture *fixture,

2031

__attribute__((unused))

2032

gconstpointer

2033

user_data){

2034

__attribute__((cleanup(string_set_clear)))

2035

string_set set = {};

2036

g_assert_true(string_set_add(&set, "test_string"));

2037

g_assert_true(string_set_contains(set, "test_string"));

2038

}

2039

2040

static void

2041

test_string_set_cleared_does_not_contain_str(__attribute__((unused))

2042

test_fixture *fixture,

2043

__attribute__((unused))

2044

gconstpointer user_data){

2045

__attribute__((cleanup(string_set_clear)))

2046

string_set set = {};

2047

g_assert_true(string_set_add(&set, "test_string"));

2048

string_set_clear(&set);

2049

g_assert_false(string_set_contains(set, "test_string"));

2050

}

2051

2052

static

2053

void test_string_set_swap_one_with_empty(__attribute__((unused))

2054

test_fixture *fixture,

2055

__attribute__((unused))

2056

gconstpointer user_data){

2057

__attribute__((cleanup(string_set_clear)))

2058

string_set set1 = {};

2059

__attribute__((cleanup(string_set_clear)))

2060

string_set set2 = {};

2061

g_assert_true(string_set_add(&set1, "test_string1"));

2062

string_set_swap(&set1, &set2);

2063

g_assert_false(string_set_contains(set1, "test_string1"));

2064

g_assert_true(string_set_contains(set2, "test_string1"));

2065

}

2066

2067

static

2068

void test_string_set_swap_empty_with_one(__attribute__((unused))

2069

test_fixture *fixture,

2070

__attribute__((unused))

2071

gconstpointer user_data){

2072

__attribute__((cleanup(string_set_clear)))

2073

string_set set1 = {};

2074

__attribute__((cleanup(string_set_clear)))

2075

string_set set2 = {};

2076

g_assert_true(string_set_add(&set2, "test_string2"));

2077

string_set_swap(&set1, &set2);

2078

g_assert_true(string_set_contains(set1, "test_string2"));

2079

g_assert_false(string_set_contains(set2, "test_string2"));

2080

}

2081

2082

static void test_string_set_swap_one_with_one(__attribute__((unused))

2083

test_fixture *fixture,

2084

__attribute__((unused))

2085

gconstpointer

2086

user_data){

2087

__attribute__((cleanup(string_set_clear)))

2088

string_set set1 = {};

2089

__attribute__((cleanup(string_set_clear)))

2090

string_set set2 = {};

2091

g_assert_true(string_set_add(&set1, "test_string1"));

2092

g_assert_true(string_set_add(&set2, "test_string2"));

2093

string_set_swap(&set1, &set2);

2094

g_assert_false(string_set_contains(set1, "test_string1"));

2095

g_assert_true(string_set_contains(set1, "test_string2"));

2096

g_assert_false(string_set_contains(set2, "test_string2"));

2097

g_assert_true(string_set_contains(set2, "test_string1"));

2098

}

2099

2100

static bool fd_has_cloexec_and_nonblock(const int);

2101

2102

static bool epoll_set_contains(int, int, uint32_t);

2103

2104

static void test_start_mandos_client(test_fixture *fixture,

2105

__attribute__((unused))

2106

gconstpointer user_data){

2107

2108

bool mandos_client_exited = false;

2109

bool quit_now = false;

2110

__attribute__((cleanup(cleanup_close)))

2111

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2112

g_assert_cmpint(epoll_fd, >=, 0);

2113

__attribute__((cleanup(cleanup_queue)))

2114

task_queue *queue = create_queue();

2115

g_assert_nonnull(queue);

2116

buffer password = {};

2117

bool password_is_read = false;

2118

const char helper_directory[] = "/nonexistent";

2119

const char *const argv[] = { "/bin/true", NULL };

2120

2121

g_assert_true(start_mandos_client(queue, epoll_fd,

2122

&mandos_client_exited, &quit_now,

2123

&password, &password_is_read,

2124

&fixture->orig_sigaction,

2125

fixture->orig_sigmask,

2126

helper_directory, 0, 0, argv));

2127

2128

g_assert_cmpuint((unsigned int)queue->length, >=, 2);

2129

2130

const task_context *const added_wait_task

2131

= find_matching_task(queue, (task_context){

2132

.func=wait_for_mandos_client_exit,

2133

.mandos_client_exited=&mandos_client_exited,

2134

.quit_now=&quit_now,

2135

});

2136

g_assert_nonnull(added_wait_task);

2137

g_assert_cmpint(added_wait_task->pid, >, 0);

2138

g_assert_cmpint(kill(added_wait_task->pid, SIGKILL), ==, 0);

2139

waitpid(added_wait_task->pid, NULL, 0);

2140

2141

const task_context *const added_read_task

2142

= find_matching_task(queue, (task_context){

2143

.func=read_mandos_client_output,

2144

.epoll_fd=epoll_fd,

2145

.password=&password,

2146

.password_is_read=&password_is_read,

2147

.quit_now=&quit_now,

2148

});

2149

g_assert_nonnull(added_read_task);

2150

g_assert_cmpint(added_read_task->fd, >, 2);

2151

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

2152

g_assert_true(epoll_set_contains(epoll_fd, added_read_task->fd,

2153

EPOLLIN | EPOLLRDHUP));

2154

}

2155

2156

static bool fd_has_cloexec_and_nonblock(const int fd){

2157

const int socket_fd_flags = fcntl(fd, F_GETFD, 0);

2158

const int socket_file_flags = fcntl(fd, F_GETFL, 0);

2159

return ((socket_fd_flags >= 0)

2160

and (socket_fd_flags & FD_CLOEXEC)

2161

and (socket_file_flags >= 0)

2162

and (socket_file_flags & O_NONBLOCK));

2163

}

2164

2165

__attribute__((const))

2166

bool is_privileged(void){

2167

uid_t user = getuid() + 1;

2168

if(user == 0){ /* Overflow check */

2169

user++;

2170

}

2171

gid_t group = getuid() + 1;

2172

if(group == 0){ /* Overflow check */

2173

group++;

2174

}

2175

const pid_t pid = fork();

2176

if(pid == 0){ /* Child */

2177

if(setresgid((uid_t)-1, group, group) == -1){

2178

if(errno != EPERM){

2179

error(EXIT_FAILURE, errno, "Failed to setresgid(-1, %" PRIuMAX

2180

", %" PRIuMAX")", (uintmax_t)group, (uintmax_t)group);

2181

}

2182

exit(EXIT_FAILURE);

2183

}

2184

if(setresuid((uid_t)-1, user, user) == -1){

2185

if(errno != EPERM){

2186

error(EXIT_FAILURE, errno, "Failed to setresuid(-1, %" PRIuMAX

2187

", %" PRIuMAX")", (uintmax_t)user, (uintmax_t)user);

2188

}

2189

exit(EXIT_FAILURE);

2190

}

2191

exit(EXIT_SUCCESS);

2192

}

2193

int status;

2194

waitpid(pid, &status, 0);

2195

if(WIFEXITED(status) and (WEXITSTATUS(status) == EXIT_SUCCESS)){

2196

return true;

2197

}

2198

return false;

2199

}

2200

2201

static bool epoll_set_contains(int epoll_fd, int fd, uint32_t events){

2202

/* Only scan for events in this eventmask */

2203

const uint32_t eventmask = EPOLLIN | EPOLLOUT | EPOLLRDHUP;

2204

__attribute__((cleanup(cleanup_string)))

2205

char *fdinfo_name = NULL;

2206

int ret = asprintf(&fdinfo_name, "/proc/self/fdinfo/%d", epoll_fd);

2207

g_assert_cmpint(ret, >, 0);

2208

g_assert_nonnull(fdinfo_name);

2209

2210

FILE *fdinfo = fopen(fdinfo_name, "r");

2211

g_assert_nonnull(fdinfo);

2212

uint32_t reported_events;

2213

buffer line = {};

2214

int found_fd = -1;

2215

2216

do {

2217

if(getline(&line.data, &line.allocated, fdinfo) < 0){

2218

break;

2219

}

2220

/* See proc(5) for format of /proc/PID/fdinfo/FD for epoll fd's */

2221

if(sscanf(line.data, "tfd: %d events: %" SCNx32 " ",

2222

&found_fd, &reported_events) == 2){

2223

if(found_fd == fd){

2224

break;

2225

}

2226

}

2227

} while(not feof(fdinfo) and not ferror(fdinfo));

2228

g_assert_cmpint(fclose(fdinfo), ==, 0);

2229

free(line.data);

2230

if(found_fd != fd){

2231

return false;

2232

}

2233

2234

if(events == 0){

2235

/* Don't check events if none are given */

2236

return true;

2237

}

2238

return (reported_events & eventmask) == (events & eventmask);

2239

}

2240

2241

static void test_start_mandos_client_execv(test_fixture *fixture,

2242

__attribute__((unused))

2243

gconstpointer user_data){

2244

bool mandos_client_exited = false;

2245

bool quit_now = false;

2246

__attribute__((cleanup(cleanup_close)))

2247

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2248

g_assert_cmpint(epoll_fd, >=, 0);

2249

__attribute__((cleanup(cleanup_queue)))

2250

task_queue *queue = create_queue();

2251

g_assert_nonnull(queue);

2252

__attribute__((cleanup(cleanup_buffer)))

2253

buffer password = {};

2254

const char helper_directory[] = "/nonexistent";

2255

/* Can't execv("/", ...), so this should fail */

2256

const char *const argv[] = { "/", NULL };

2257

2258

{

2259

__attribute__((cleanup(cleanup_close)))

2260

const int devnull_fd = open("/dev/null",

2261

O_WRONLY | O_CLOEXEC | O_NOCTTY);

2262

g_assert_cmpint(devnull_fd, >=, 0);

2263

__attribute__((cleanup(cleanup_close)))

2264

const int real_stderr_fd = dup(STDERR_FILENO);

2265

g_assert_cmpint(real_stderr_fd, >=, 0);

2266

dup2(devnull_fd, STDERR_FILENO);

2267

2268

const bool success = start_mandos_client(queue, epoll_fd,

2269

&mandos_client_exited,

2270

&quit_now,

2271

&password,

2272

(bool[]){false},

2273

&fixture->orig_sigaction,

2274

fixture->orig_sigmask,

2275

helper_directory, 0, 0,

2276

argv);

2277

dup2(real_stderr_fd, STDERR_FILENO);

2278

g_assert_true(success);

2279

}

2280

g_assert_cmpuint((unsigned int)queue->length, ==, 2);

2281

2282

struct timespec starttime, currtime;

2283

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2284

do {

2285

queue->next_run = 0;

2286

string_set cancelled_filenames = {};

2287

2288

{

2289

__attribute__((cleanup(cleanup_close)))

2290

const int devnull_fd = open("/dev/null",

2291

O_WRONLY | O_CLOEXEC | O_NOCTTY);

2292

g_assert_cmpint(devnull_fd, >=, 0);

2293

__attribute__((cleanup(cleanup_close)))

2294

const int real_stderr_fd = dup(STDERR_FILENO);

2295

g_assert_cmpint(real_stderr_fd, >=, 0);

2296

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2297

dup2(devnull_fd, STDERR_FILENO);

2298

const bool success = run_queue(&queue, &cancelled_filenames,

2299

&quit_now);

2300

dup2(real_stderr_fd, STDERR_FILENO);

2301

if(not success){

2302

break;

2303

}

2304

}

2305

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2306

} while(((queue->length) > 0)

2307

and (not quit_now)

2308

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2309

2310

g_assert_true(quit_now);

2311

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2312

g_assert_true(mandos_client_exited);

2313

}

2314

2315

static void test_start_mandos_client_suid_euid(test_fixture *fixture,

2316

__attribute__((unused))

2317

gconstpointer

2318

user_data){

2319

if(not is_privileged()){

2320

g_test_skip("Not privileged");

2321

return;

2322

}

2323

2324

bool mandos_client_exited = false;

2325

bool quit_now = false;

2326

__attribute__((cleanup(cleanup_close)))

2327

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2328

g_assert_cmpint(epoll_fd, >=, 0);

2329

__attribute__((cleanup(cleanup_queue)))

2330

task_queue *queue = create_queue();

2331

g_assert_nonnull(queue);

2332

__attribute__((cleanup(cleanup_buffer)))

2333

buffer password = {};

2334

bool password_is_read = false;

2335

const char helper_directory[] = "/nonexistent";

2336

const char *const argv[] = { "/usr/bin/id", "--user", NULL };

2337

uid_t user = 1000;

2338

gid_t group = 1001;

2339

2340

const bool success = start_mandos_client(queue, epoll_fd,

2341

&mandos_client_exited,

2342

&quit_now, &password,

2343

&password_is_read,

2344

&fixture->orig_sigaction,

2345

fixture->orig_sigmask,

2346

helper_directory, user,

2347

group, argv);

2348

g_assert_true(success);

2349

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2350

2351

struct timespec starttime, currtime;

2352

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2353

do {

2354

queue->next_run = 0;

2355

string_set cancelled_filenames = {};

2356

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2357

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2358

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2359

} while(((queue->length) > 0)

2360

and (not quit_now)

2361

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2362

2363

g_assert_false(quit_now);

2364

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2365

g_assert_true(mandos_client_exited);

2366

2367

g_assert_true(password_is_read);

2368

g_assert_nonnull(password.data);

2369

2370

uintmax_t id;

2371

g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),

2372

==, 1);

2373

g_assert_true((uid_t)id == id);

2374

2375

g_assert_cmpuint((unsigned int)id, ==, 0);

2376

}

2377

2378

static void test_start_mandos_client_suid_egid(test_fixture *fixture,

2379

__attribute__((unused))

2380

gconstpointer

2381

user_data){

2382

if(not is_privileged()){

2383

g_test_skip("Not privileged");

2384

return;

2385

}

2386

2387

bool mandos_client_exited = false;

2388

bool quit_now = false;

2389

__attribute__((cleanup(cleanup_close)))

2390

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2391

g_assert_cmpint(epoll_fd, >=, 0);

2392

__attribute__((cleanup(cleanup_queue)))

2393

task_queue *queue = create_queue();

2394

g_assert_nonnull(queue);

2395

__attribute__((cleanup(cleanup_buffer)))

2396

buffer password = {};

2397

bool password_is_read = false;

2398

const char helper_directory[] = "/nonexistent";

2399

const char *const argv[] = { "/usr/bin/id", "--group", NULL };

2400

uid_t user = 1000;

2401

gid_t group = 1001;

2402

2403

const bool success = start_mandos_client(queue, epoll_fd,

2404

&mandos_client_exited,

2405

&quit_now, &password,

2406

&password_is_read,

2407

&fixture->orig_sigaction,

2408

fixture->orig_sigmask,

2409

helper_directory, user,

2410

group, argv);

2411

g_assert_true(success);

2412

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2413

2414

struct timespec starttime, currtime;

2415

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2416

do {

2417

queue->next_run = 0;

2418

string_set cancelled_filenames = {};

2419

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2420

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2421

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2422

} while(((queue->length) > 0)

2423

and (not quit_now)

2424

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2425

2426

g_assert_false(quit_now);

2427

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2428

g_assert_true(mandos_client_exited);

2429

2430

g_assert_true(password_is_read);

2431

g_assert_nonnull(password.data);

2432

2433

uintmax_t id;

2434

g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),

2435

==, 1);

2436

g_assert_true((gid_t)id == id);

2437

2438

g_assert_cmpuint((unsigned int)id, ==, 0);

2439

}

2440

2441

static void test_start_mandos_client_suid_ruid(test_fixture *fixture,

2442

__attribute__((unused))

2443

gconstpointer

2444

user_data){

2445

if(not is_privileged()){

2446

g_test_skip("Not privileged");

2447

return;

2448

}

2449

2450

bool mandos_client_exited = false;

2451

bool quit_now = false;

2452

__attribute__((cleanup(cleanup_close)))

2453

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2454

g_assert_cmpint(epoll_fd, >=, 0);

2455

__attribute__((cleanup(cleanup_queue)))

2456

task_queue *queue = create_queue();

2457

g_assert_nonnull(queue);

2458

__attribute__((cleanup(cleanup_buffer)))

2459

buffer password = {};

2460

bool password_is_read = false;

2461

const char helper_directory[] = "/nonexistent";

2462

const char *const argv[] = { "/usr/bin/id", "--user", "--real",

2463

NULL };

2464

uid_t user = 1000;

2465

gid_t group = 1001;

2466

2467

const bool success = start_mandos_client(queue, epoll_fd,

2468

&mandos_client_exited,

2469

&quit_now, &password,

2470

&password_is_read,

2471

&fixture->orig_sigaction,

2472

fixture->orig_sigmask,

2473

helper_directory, user,

2474

group, argv);

2475

g_assert_true(success);

2476

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2477

2478

struct timespec starttime, currtime;

2479

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2480

do {

2481

queue->next_run = 0;

2482

string_set cancelled_filenames = {};

2483

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2484

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2485

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2486

} while(((queue->length) > 0)

2487

and (not quit_now)

2488

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2489

2490

g_assert_false(quit_now);

2491

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2492

g_assert_true(mandos_client_exited);

2493

2494

g_assert_true(password_is_read);

2495

g_assert_nonnull(password.data);

2496

2497

uintmax_t id;

2498

g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),

2499

==, 1);

2500

g_assert_true((uid_t)id == id);

2501

2502

g_assert_cmpuint((unsigned int)id, ==, user);

2503

}

2504

2505

static void test_start_mandos_client_suid_rgid(test_fixture *fixture,

2506

__attribute__((unused))

2507

gconstpointer

2508

user_data){

2509

if(not is_privileged()){

2510

g_test_skip("Not privileged");

2511

return;

2512

}

2513

2514

bool mandos_client_exited = false;

2515

bool quit_now = false;

2516

__attribute__((cleanup(cleanup_close)))

2517

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2518

g_assert_cmpint(epoll_fd, >=, 0);

2519

__attribute__((cleanup(cleanup_queue)))

2520

task_queue *queue = create_queue();

2521

g_assert_nonnull(queue);

2522

__attribute__((cleanup(cleanup_buffer)))

2523

buffer password = {};

2524

bool password_is_read = false;

2525

const char helper_directory[] = "/nonexistent";

2526

const char *const argv[] = { "/usr/bin/id", "--group", "--real",

2527

NULL };

2528

uid_t user = 1000;

2529

gid_t group = 1001;

2530

2531

const bool success = start_mandos_client(queue, epoll_fd,

2532

&mandos_client_exited,

2533

&quit_now, &password,

2534

&password_is_read,

2535

&fixture->orig_sigaction,

2536

fixture->orig_sigmask,

2537

helper_directory, user,

2538

group, argv);

2539

g_assert_true(success);

2540

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2541

2542

struct timespec starttime, currtime;

2543

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2544

do {

2545

queue->next_run = 0;

2546

string_set cancelled_filenames = {};

2547

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2548

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2549

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2550

} while(((queue->length) > 0)

2551

and (not quit_now)

2552

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2553

2554

g_assert_false(quit_now);

2555

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2556

g_assert_true(mandos_client_exited);

2557

2558

g_assert_true(password_is_read);

2559

g_assert_nonnull(password.data);

2560

2561

uintmax_t id;

2562

g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),

2563

==, 1);

2564

g_assert_true((gid_t)id == id);

2565

2566

g_assert_cmpuint((unsigned int)id, ==, group);

2567

}

2568

2569

static void test_start_mandos_client_read(test_fixture *fixture,

2570

__attribute__((unused))

2571

gconstpointer user_data){

2572

bool mandos_client_exited = false;

2573

bool quit_now = false;

2574

__attribute__((cleanup(cleanup_close)))

2575

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2576

g_assert_cmpint(epoll_fd, >=, 0);

2577

__attribute__((cleanup(cleanup_queue)))

2578

task_queue *queue = create_queue();

2579

g_assert_nonnull(queue);

2580

__attribute__((cleanup(cleanup_buffer)))

2581

buffer password = {};

2582

bool password_is_read = false;

2583

const char dummy_test_password[] = "dummy test password";

2584

const char helper_directory[] = "/nonexistent";

2585

const char *const argv[] = { "/bin/echo", "-n", dummy_test_password,

2586

NULL };

2587

2588

const bool success = start_mandos_client(queue, epoll_fd,

2589

&mandos_client_exited,

2590

&quit_now, &password,

2591

&password_is_read,

2592

&fixture->orig_sigaction,

2593

fixture->orig_sigmask,

2594

helper_directory, 0, 0,

2595

argv);

2596

g_assert_true(success);

2597

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2598

2599

struct timespec starttime, currtime;

2600

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2601

do {

2602

queue->next_run = 0;

2603

string_set cancelled_filenames = {};

2604

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2605

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2606

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2607

} while(((queue->length) > 0)

2608

and (not quit_now)

2609

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2610

2611

g_assert_false(quit_now);

2612

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2613

g_assert_true(mandos_client_exited);

2614

2615

g_assert_true(password_is_read);

2616

g_assert_cmpint((int)password.length, ==,

2617

sizeof(dummy_test_password)-1);

2618

g_assert_nonnull(password.data);

2619

g_assert_cmpint(memcmp(dummy_test_password, password.data,

2620

sizeof(dummy_test_password)-1), ==, 0);

2621

}

2622

2623

static

2624

void test_start_mandos_client_helper_directory(test_fixture *fixture,

2625

__attribute__((unused))

2626

gconstpointer

2627

user_data){

2628

bool mandos_client_exited = false;

2629

bool quit_now = false;

2630

__attribute__((cleanup(cleanup_close)))

2631

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2632

g_assert_cmpint(epoll_fd, >=, 0);

2633

__attribute__((cleanup(cleanup_queue)))

2634

task_queue *queue = create_queue();

2635

g_assert_nonnull(queue);

2636

__attribute__((cleanup(cleanup_buffer)))

2637

buffer password = {};

2638

bool password_is_read = false;

2639

const char helper_directory[] = "/nonexistent";

2640

const char *const argv[] = { "/bin/sh", "-c",

2641

"echo -n ${MANDOSPLUGINHELPERDIR}", NULL };

2642

2643

const bool success = start_mandos_client(queue, epoll_fd,

2644

&mandos_client_exited,

2645

&quit_now, &password,

2646

&password_is_read,

2647

&fixture->orig_sigaction,

2648

fixture->orig_sigmask,

2649

helper_directory, 0, 0,

2650

argv);

2651

g_assert_true(success);

2652

g_assert_cmpuint((unsigned int)queue->length, >, 0);

2653

2654

struct timespec starttime, currtime;

2655

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2656

do {

2657

queue->next_run = 0;

2658

string_set cancelled_filenames = {};

2659

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2660

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2661

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2662

} while(((queue->length) > 0)

2663

and (not quit_now)

2664

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2665

2666

g_assert_false(quit_now);

2667

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2668

g_assert_true(mandos_client_exited);

2669

2670

g_assert_true(password_is_read);

2671

g_assert_cmpint((int)password.length, ==,

2672

sizeof(helper_directory)-1);

2673

g_assert_nonnull(password.data);

2674

g_assert_cmpint(memcmp(helper_directory, password.data,

2675

sizeof(helper_directory)-1), ==, 0);

2676

}

2677

2678

__attribute__((nonnull, warn_unused_result))

2679

static bool proc_status_sigblk_to_sigset(const char *const,

2680

sigset_t *const);

2681

2682

static void test_start_mandos_client_sigmask(test_fixture *fixture,

2683

__attribute__((unused))

2684

gconstpointer user_data){

2685

bool mandos_client_exited = false;

2686

bool quit_now = false;

2687

__attribute__((cleanup(cleanup_close)))

2688

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2689

g_assert_cmpint(epoll_fd, >=, 0);

2690

__attribute__((cleanup(cleanup_queue)))

2691

task_queue *queue = create_queue();

2692

g_assert_nonnull(queue);

2693

__attribute__((cleanup(cleanup_buffer)))

2694

buffer password = {};

2695

bool password_is_read = false;

2696

const char helper_directory[] = "/nonexistent";

2697

/* see proc(5) for format of /proc/self/status */

2698

const char *const argv[] = { "/usr/bin/awk",

2699

"$1==\"SigBlk:\"{ print $2 }", "/proc/self/status", NULL };

2700

2701

g_assert_true(start_mandos_client(queue, epoll_fd,

2702

&mandos_client_exited, &quit_now,

2703

&password, &password_is_read,

2704

&fixture->orig_sigaction,

2705

fixture->orig_sigmask,

2706

helper_directory, 0, 0, argv));

2707

2708

struct timespec starttime, currtime;

2709

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2710

do {

2711

queue->next_run = 0;

2712

string_set cancelled_filenames = {};

2713

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2714

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

2715

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2716

} while((not (mandos_client_exited and password_is_read))

2717

and (not quit_now)

2718

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2719

g_assert_true(mandos_client_exited);

2720

g_assert_true(password_is_read);

2721

2722

sigset_t parsed_sigmask;

2723

g_assert_true(proc_status_sigblk_to_sigset(password.data,

2724

&parsed_sigmask));

2725

2726

for(int signum = 1; signum < NSIG; signum++){

2727

const bool has_signal = sigismember(&parsed_sigmask, signum);

2728

if(sigismember(&fixture->orig_sigmask, signum)){

2729

g_assert_true(has_signal);

2730

} else {

2731

g_assert_false(has_signal);

2732

}

2733

}

2734

}

2735

2736

__attribute__((nonnull, warn_unused_result))

2737

static bool proc_status_sigblk_to_sigset(const char *const sigblk,

2738

sigset_t *const sigmask){

2739

/* parse /proc/PID/status SigBlk value and convert to a sigset_t */

2740

uintmax_t scanned_sigmask;

2741

if(sscanf(sigblk, "%" SCNxMAX " ", &scanned_sigmask) != 1){

2742

return false;

2743

}

2744

if(sigemptyset(sigmask) != 0){

2745

return false;

2746

}

2747

for(int signum = 1; signum < NSIG; signum++){

2748

if(scanned_sigmask & ((uintmax_t)1 << (signum-1))){

2749

if(sigaddset(sigmask, signum) != 0){

2750

return false;

2751

}

2752

}

2753

}

2754

return true;

2755

}

2756

2757

static void run_task_with_stderr_to_dev_null(const task_context task,

2758

task_queue *const queue);

2759

2760

static

2761

void test_wait_for_mandos_client_exit_badpid(__attribute__((unused))

2762

test_fixture *fixture,

2763

__attribute__((unused))

2764

gconstpointer user_data){

2765

2766

bool mandos_client_exited = false;

2767

bool quit_now = false;

2768

2769

__attribute__((cleanup(cleanup_queue)))

2770

task_queue *queue = create_queue();

2771

g_assert_nonnull(queue);

2772

const task_context task = {

2773

.func=wait_for_mandos_client_exit,

2774

.pid=1,

2775

.mandos_client_exited=&mandos_client_exited,

2776

.quit_now=&quit_now,

2777

};

2778

run_task_with_stderr_to_dev_null(task, queue);

2779

2780

g_assert_false(mandos_client_exited);

2781

g_assert_true(quit_now);

2782

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2783

}

2784

2785

static void run_task_with_stderr_to_dev_null(const task_context task,

2786

task_queue *const queue){

2787

FILE *real_stderr = stderr;

2788

FILE *devnull = fopen("/dev/null", "we");

2789

g_assert_nonnull(devnull);

2790

2791

stderr = devnull;

2792

task.func(task, queue);

2793

stderr = real_stderr;

2794

2795

g_assert_cmpint(fclose(devnull), ==, 0);

2796

}

2797

2798

static

2799

void test_wait_for_mandos_client_exit_noexit(test_fixture *fixture,

2800

__attribute__((unused))

2801

gconstpointer user_data){

2802

bool mandos_client_exited = false;

2803

bool quit_now = false;

2804

2805

pid_t create_eternal_process(void){

2806

const pid_t pid = fork();

2807

if(pid == 0){ /* Child */

2808

if(not restore_signal_handler(&fixture->orig_sigaction)){

2809

_exit(EXIT_FAILURE);

2810

}

2811

if(not restore_sigmask(&fixture->orig_sigmask)){

2812

_exit(EXIT_FAILURE);

2813

}

2814

while(true){

2815

pause();

2816

}

2817

}

2818

return pid;

2819

}

2820

pid_t pid = create_eternal_process();

2821

g_assert_true(pid != -1);

2822

2823

__attribute__((cleanup(cleanup_queue)))

2824

task_queue *queue = create_queue();

2825

g_assert_nonnull(queue);

2826

const task_context task = {

2827

.func=wait_for_mandos_client_exit,

2828

.pid=pid,

2829

.mandos_client_exited=&mandos_client_exited,

2830

.quit_now=&quit_now,

2831

};

2832

task.func(task, queue);

2833

2834

g_assert_false(mandos_client_exited);

2835

g_assert_false(quit_now);

2836

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

2837

2838

g_assert_nonnull(find_matching_task(queue, (task_context){

2839

.func=wait_for_mandos_client_exit,

2840

.pid=task.pid,

2841

.mandos_client_exited=&mandos_client_exited,

2842

.quit_now=&quit_now,

2843

}));

2844

}

2845

2846

static

2847

void test_wait_for_mandos_client_exit_success(test_fixture *fixture,

2848

__attribute__((unused))

2849

gconstpointer

2850

user_data){

2851

bool mandos_client_exited = false;

2852

bool quit_now = false;

2853

2854

pid_t create_successful_process(void){

2855

const pid_t pid = fork();

2856

if(pid == 0){ /* Child */

2857

if(not restore_signal_handler(&fixture->orig_sigaction)){

2858

_exit(EXIT_FAILURE);

2859

}

2860

if(not restore_sigmask(&fixture->orig_sigmask)){

2861

_exit(EXIT_FAILURE);

2862

}

2863

exit(EXIT_SUCCESS);

2864

}

2865

return pid;

2866

}

2867

const pid_t pid = create_successful_process();

2868

g_assert_true(pid != -1);

2869

2870

__attribute__((cleanup(cleanup_queue)))

2871

task_queue *queue = create_queue();

2872

g_assert_nonnull(queue);

2873

const task_context initial_task = {

2874

.func=wait_for_mandos_client_exit,

2875

.pid=pid,

2876

.mandos_client_exited=&mandos_client_exited,

2877

.quit_now=&quit_now,

2878

};

2879

g_assert_true(add_to_queue(queue, initial_task));

2880

2881

struct timespec starttime, currtime;

2882

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2883

__attribute__((cleanup(cleanup_close)))

2884

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2885

do {

2886

queue->next_run = 0;

2887

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2888

g_assert_true(run_queue(&queue, (string_set[]){{}}, &quit_now));

2889

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2890

} while((not mandos_client_exited)

2891

and (not quit_now)

2892

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2893

2894

g_assert_true(mandos_client_exited);

2895

g_assert_false(quit_now);

2896

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2897

}

2898

2899

static

2900

void test_wait_for_mandos_client_exit_failure(test_fixture *fixture,

2901

__attribute__((unused))

2902

gconstpointer

2903

user_data){

2904

bool mandos_client_exited = false;

2905

bool quit_now = false;

2906

2907

pid_t create_failing_process(void){

2908

const pid_t pid = fork();

2909

if(pid == 0){ /* Child */

2910

if(not restore_signal_handler(&fixture->orig_sigaction)){

2911

_exit(EXIT_FAILURE);

2912

}

2913

if(not restore_sigmask(&fixture->orig_sigmask)){

2914

_exit(EXIT_FAILURE);

2915

}

2916

exit(EXIT_FAILURE);

2917

}

2918

return pid;

2919

}

2920

const pid_t pid = create_failing_process();

2921

g_assert_true(pid != -1);

2922

2923

__attribute__((cleanup(string_set_clear)))

2924

string_set cancelled_filenames = {};

2925

__attribute__((cleanup(cleanup_close)))

2926

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

2927

g_assert_cmpint(epoll_fd, >=, 0);

2928

__attribute__((cleanup(cleanup_queue)))

2929

task_queue *queue = create_queue();

2930

g_assert_nonnull(queue);

2931

g_assert_true(add_to_queue(queue, (task_context){

2932

.func=wait_for_mandos_client_exit,

2933

.pid=pid,

2934

.mandos_client_exited=&mandos_client_exited,

2935

.quit_now=&quit_now,

2936

}));

2937

2938

g_assert_true(sigismember(&fixture->orig_sigmask, SIGCHLD) == 0);

2939

2940

__attribute__((cleanup(cleanup_close)))

2941

const int devnull_fd = open("/dev/null",

2942

O_WRONLY | O_CLOEXEC | O_NOCTTY);

2943

g_assert_cmpint(devnull_fd, >=, 0);

2944

__attribute__((cleanup(cleanup_close)))

2945

const int real_stderr_fd = dup(STDERR_FILENO);

2946

g_assert_cmpint(real_stderr_fd, >=, 0);

2947

2948

struct timespec starttime, currtime;

2949

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

2950

do {

2951

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

2952

dup2(devnull_fd, STDERR_FILENO);

2953

const bool success = run_queue(&queue, &cancelled_filenames,

2954

&quit_now);

2955

dup2(real_stderr_fd, STDERR_FILENO);

2956

if(not success){

2957

break;

2958

}

2959

2960

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

2961

} while((not mandos_client_exited)

2962

and (not quit_now)

2963

and ((currtime.tv_sec - starttime.tv_sec) < 10));

2964

2965

g_assert_true(quit_now);

2966

g_assert_true(mandos_client_exited);

2967

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

2968

}

2969

2970

static

2971

void test_wait_for_mandos_client_exit_killed(test_fixture *fixture,

2972

__attribute__((unused))

2973

gconstpointer user_data){

2974

bool mandos_client_exited = false;

2975

bool quit_now = false;

2976

2977

pid_t create_killed_process(void){

2978

const pid_t pid = fork();

2979

if(pid == 0){ /* Child */

2980

if(not restore_signal_handler(&fixture->orig_sigaction)){

2981

_exit(EXIT_FAILURE);

2982

}

2983

if(not restore_sigmask(&fixture->orig_sigmask)){

2984

_exit(EXIT_FAILURE);

2985

}

2986

while(true){

2987

pause();

2988

}

2989

}

2990

kill(pid, SIGKILL);

2991

return pid;

2992

}

2993

const pid_t pid = create_killed_process();

2994

g_assert_true(pid != -1);

2995

2996

__attribute__((cleanup(string_set_clear)))

2997

string_set cancelled_filenames = {};

2998

__attribute__((cleanup(cleanup_close)))

2999

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3000

g_assert_cmpint(epoll_fd, >=, 0);

3001

__attribute__((cleanup(cleanup_queue)))

3002

task_queue *queue = create_queue();

3003

g_assert_nonnull(queue);

3004

g_assert_true(add_to_queue(queue, (task_context){

3005

.func=wait_for_mandos_client_exit,

3006

.pid=pid,

3007

.mandos_client_exited=&mandos_client_exited,

3008

.quit_now=&quit_now,

3009

}));

3010

3011

__attribute__((cleanup(cleanup_close)))

3012

const int devnull_fd = open("/dev/null",

3013

O_WRONLY | O_CLOEXEC, O_NOCTTY);

3014

g_assert_cmpint(devnull_fd, >=, 0);

3015

__attribute__((cleanup(cleanup_close)))

3016

const int real_stderr_fd = dup(STDERR_FILENO);

3017

g_assert_cmpint(real_stderr_fd, >=, 0);

3018

3019

struct timespec starttime, currtime;

3020

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);

3021

do {

3022

g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));

3023

dup2(devnull_fd, STDERR_FILENO);

3024

const bool success = run_queue(&queue, &cancelled_filenames,

3025

&quit_now);

3026

dup2(real_stderr_fd, STDERR_FILENO);

3027

if(not success){

3028

break;

3029

}

3030

3031

g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);

3032

} while((not mandos_client_exited)

3033

and (not quit_now)

3034

and ((currtime.tv_sec - starttime.tv_sec) < 10));

3035

3036

g_assert_true(mandos_client_exited);

3037

g_assert_true(quit_now);

3038

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3039

}

3040

3041

static bool epoll_set_does_not_contain(int, int);

3042

3043

static

3044

void test_read_mandos_client_output_readerror(__attribute__((unused))

3045

test_fixture *fixture,

3046

__attribute__((unused))

3047

gconstpointer

3048

user_data){

3049

__attribute__((cleanup(cleanup_close)))

3050

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3051

g_assert_cmpint(epoll_fd, >=, 0);

3052

3053

__attribute__((cleanup(cleanup_buffer)))

3054

buffer password = {};

3055

3056

/* Reading /proc/self/mem from offset 0 will always give EIO */

3057

const int fd = open("/proc/self/mem",

3058

O_RDONLY | O_CLOEXEC | O_NOCTTY);

3059

3060

bool password_is_read = false;

3061

bool quit_now = false;

3062

__attribute__((cleanup(cleanup_queue)))

3063

task_queue *queue = create_queue();

3064

g_assert_nonnull(queue);

3065

3066

task_context task = {

3067

.func=read_mandos_client_output,

3068

.epoll_fd=epoll_fd,

3069

.fd=fd,

3070

.password=&password,

3071

.password_is_read=&password_is_read,

3072

.quit_now=&quit_now,

3073

};

3074

run_task_with_stderr_to_dev_null(task, queue);

3075

g_assert_false(password_is_read);

3076

g_assert_cmpint((int)password.length, ==, 0);

3077

g_assert_true(quit_now);

3078

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3079

3080

g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));

3081

3082

g_assert_cmpint(close(fd), ==, -1);

3083

}

3084

3085

static bool epoll_set_does_not_contain(int epoll_fd, int fd){

3086

return not epoll_set_contains(epoll_fd, fd, 0);

3087

}

3088

3089

static

3090

void test_read_mandos_client_output_nodata(__attribute__((unused))

3091

test_fixture *fixture,

3092

__attribute__((unused))

3093

gconstpointer user_data){

3094

__attribute__((cleanup(cleanup_close)))

3095

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3096

g_assert_cmpint(epoll_fd, >=, 0);

3097

3098

int pipefds[2];

3099

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

3100

3101

__attribute__((cleanup(cleanup_buffer)))

3102

buffer password = {};

3103

3104

bool password_is_read = false;

3105

bool quit_now = false;

3106

__attribute__((cleanup(cleanup_queue)))

3107

task_queue *queue = create_queue();

3108

g_assert_nonnull(queue);

3109

3110

task_context task = {

3111

.func=read_mandos_client_output,

3112

.epoll_fd=epoll_fd,

3113

.fd=pipefds[0],

3114

.password=&password,

3115

.password_is_read=&password_is_read,

3116

.quit_now=&quit_now,

3117

};

3118

task.func(task, queue);

3119

g_assert_false(password_is_read);

3120

g_assert_cmpint((int)password.length, ==, 0);

3121

g_assert_false(quit_now);

3122

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

3123

3124

g_assert_nonnull(find_matching_task(queue, (task_context){

3125

.func=read_mandos_client_output,

3126

.epoll_fd=epoll_fd,

3127

.fd=pipefds[0],

3128

.password=&password,

3129

.password_is_read=&password_is_read,

3130

.quit_now=&quit_now,

3131

}));

3132

3133

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

3134

EPOLLIN | EPOLLRDHUP));

3135

3136

g_assert_cmpint(close(pipefds[1]), ==, 0);

3137

}

3138

3139

static void test_read_mandos_client_output_eof(__attribute__((unused))

3140

test_fixture *fixture,

3141

__attribute__((unused))

3142

gconstpointer

3143

user_data){

3144

__attribute__((cleanup(cleanup_close)))

3145

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3146

g_assert_cmpint(epoll_fd, >=, 0);

3147

3148

int pipefds[2];

3149

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

3150

g_assert_cmpint(close(pipefds[1]), ==, 0);

3151

3152

__attribute__((cleanup(cleanup_buffer)))

3153

buffer password = {};

3154

3155

bool password_is_read = false;

3156

bool quit_now = false;

3157

__attribute__((cleanup(cleanup_queue)))

3158

task_queue *queue = create_queue();

3159

g_assert_nonnull(queue);

3160

3161

task_context task = {

3162

.func=read_mandos_client_output,

3163

.epoll_fd=epoll_fd,

3164

.fd=pipefds[0],

3165

.password=&password,

3166

.password_is_read=&password_is_read,

3167

.quit_now=&quit_now,

3168

};

3169

task.func(task, queue);

3170

g_assert_true(password_is_read);

3171

g_assert_cmpint((int)password.length, ==, 0);

3172

g_assert_false(quit_now);

3173

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3174

3175

g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));

3176

3177

g_assert_cmpint(close(pipefds[0]), ==, -1);

3178

}

3179

3180

static

3181

void test_read_mandos_client_output_once(__attribute__((unused))

3182

test_fixture *fixture,

3183

__attribute__((unused))

3184

gconstpointer user_data){

3185

__attribute__((cleanup(cleanup_close)))

3186

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3187

g_assert_cmpint(epoll_fd, >=, 0);

3188

3189

int pipefds[2];

3190

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

3191

3192

const char dummy_test_password[] = "dummy test password";

3193

/* Start with a pre-allocated buffer */

3194

__attribute__((cleanup(cleanup_buffer)))

3195

buffer password = {

3196

.data=malloc(sizeof(dummy_test_password)),

3197

.length=0,

3198

.allocated=sizeof(dummy_test_password),

3199

};

3200

g_assert_nonnull(password.data);

3201

if(mlock(password.data, password.allocated) != 0){

3202

g_assert_true(errno == EPERM or errno == ENOMEM);

3203

}

3204

3205

bool password_is_read = false;

3206

bool quit_now = false;

3207

__attribute__((cleanup(cleanup_queue)))

3208

task_queue *queue = create_queue();

3209

g_assert_nonnull(queue);

3210

3211

g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);

3212

g_assert_cmpint((int)write(pipefds[1], dummy_test_password,

3213

sizeof(dummy_test_password)),

3214

==, (int)sizeof(dummy_test_password));

3215

3216

task_context task = {

3217

.func=read_mandos_client_output,

3218

.epoll_fd=epoll_fd,

3219

.fd=pipefds[0],

3220

.password=&password,

3221

.password_is_read=&password_is_read,

3222

.quit_now=&quit_now,

3223

};

3224

task.func(task, queue);

3225

3226

g_assert_false(password_is_read);

3227

g_assert_cmpint((int)password.length, ==,

3228

(int)sizeof(dummy_test_password));

3229

g_assert_nonnull(password.data);

3230

g_assert_cmpint(memcmp(password.data, dummy_test_password,

3231

sizeof(dummy_test_password)), ==, 0);

3232

3233

g_assert_false(quit_now);

3234

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

3235

3236

g_assert_nonnull(find_matching_task(queue, (task_context){

3237

.func=read_mandos_client_output,

3238

.epoll_fd=epoll_fd,

3239

.fd=pipefds[0],

3240

.password=&password,

3241

.password_is_read=&password_is_read,

3242

.quit_now=&quit_now,

3243

}));

3244

3245

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

3246

EPOLLIN | EPOLLRDHUP));

3247

3248

g_assert_cmpint(close(pipefds[1]), ==, 0);

3249

}

3250

3251

static

3252

void test_read_mandos_client_output_malloc(__attribute__((unused))

3253

test_fixture *fixture,

3254

__attribute__((unused))

3255

gconstpointer user_data){

3256

__attribute__((cleanup(cleanup_close)))

3257

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3258

g_assert_cmpint(epoll_fd, >=, 0);

3259

3260

int pipefds[2];

3261

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

3262

3263

const char dummy_test_password[] = "dummy test password";

3264

/* Start with an empty buffer */

3265

__attribute__((cleanup(cleanup_buffer)))

3266

buffer password = {};

3267

3268

bool password_is_read = false;

3269

bool quit_now = false;

3270

__attribute__((cleanup(cleanup_queue)))

3271

task_queue *queue = create_queue();

3272

g_assert_nonnull(queue);

3273

3274

g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);

3275

g_assert_cmpint((int)write(pipefds[1], dummy_test_password,

3276

sizeof(dummy_test_password)),

3277

==, (int)sizeof(dummy_test_password));

3278

3279

task_context task = {

3280

.func=read_mandos_client_output,

3281

.epoll_fd=epoll_fd,

3282

.fd=pipefds[0],

3283

.password=&password,

3284

.password_is_read=&password_is_read,

3285

.quit_now=&quit_now,

3286

};

3287

task.func(task, queue);

3288

3289

g_assert_false(password_is_read);

3290

g_assert_cmpint((int)password.length, ==,

3291

(int)sizeof(dummy_test_password));

3292

g_assert_nonnull(password.data);

3293

g_assert_cmpint(memcmp(password.data, dummy_test_password,

3294

sizeof(dummy_test_password)), ==, 0);

3295

3296

g_assert_false(quit_now);

3297

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

3298

3299

g_assert_nonnull(find_matching_task(queue, (task_context){

3300

.func=read_mandos_client_output,

3301

.epoll_fd=epoll_fd,

3302

.fd=pipefds[0],

3303

.password=&password,

3304

.password_is_read=&password_is_read,

3305

.quit_now=&quit_now,

3306

}));

3307

3308

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

3309

EPOLLIN | EPOLLRDHUP));

3310

3311

g_assert_cmpint(close(pipefds[1]), ==, 0);

3312

}

3313

3314

static

3315

void test_read_mandos_client_output_append(__attribute__((unused))

3316

test_fixture *fixture,

3317

__attribute__((unused))

3318

gconstpointer user_data){

3319

__attribute__((cleanup(cleanup_close)))

3320

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3321

g_assert_cmpint(epoll_fd, >=, 0);

3322

3323

int pipefds[2];

3324

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

3325

3326

const char dummy_test_password[] = "dummy test password";

3327

__attribute__((cleanup(cleanup_buffer)))

3328

buffer password = {

3329

.data=malloc(PIPE_BUF),

3330

.length=PIPE_BUF,

3331

.allocated=PIPE_BUF,

3332

};

3333

g_assert_nonnull(password.data);

3334

if(mlock(password.data, password.allocated) != 0){

3335

g_assert_true(errno == EPERM or errno == ENOMEM);

3336

}

3337

3338

memset(password.data, 'x', PIPE_BUF);

3339

char password_expected[PIPE_BUF];

3340

memcpy(password_expected, password.data, PIPE_BUF);

3341

3342

bool password_is_read = false;

3343

bool quit_now = false;

3344

__attribute__((cleanup(cleanup_queue)))

3345

task_queue *queue = create_queue();

3346

g_assert_nonnull(queue);

3347

3348

g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);

3349

g_assert_cmpint((int)write(pipefds[1], dummy_test_password,

3350

sizeof(dummy_test_password)),

3351

==, (int)sizeof(dummy_test_password));

3352

3353

task_context task = {

3354

.func=read_mandos_client_output,

3355

.epoll_fd=epoll_fd,

3356

.fd=pipefds[0],

3357

.password=&password,

3358

.password_is_read=&password_is_read,

3359

.quit_now=&quit_now,

3360

};

3361

task.func(task, queue);

3362

3363

g_assert_false(password_is_read);

3364

g_assert_cmpint((int)password.length, ==,

3365

PIPE_BUF + sizeof(dummy_test_password));

3366

g_assert_nonnull(password.data);

3367

g_assert_cmpint(memcmp(password_expected, password.data, PIPE_BUF),

3368

==, 0);

3369

g_assert_cmpint(memcmp(password.data + PIPE_BUF,

3370

dummy_test_password,

3371

sizeof(dummy_test_password)), ==, 0);

3372

g_assert_false(quit_now);

3373

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

3374

3375

g_assert_nonnull(find_matching_task(queue, (task_context){

3376

.func=read_mandos_client_output,

3377

.epoll_fd=epoll_fd,

3378

.fd=pipefds[0],

3379

.password=&password,

3380

.password_is_read=&password_is_read,

3381

.quit_now=&quit_now,

3382

}));

3383

3384

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

3385

EPOLLIN | EPOLLRDHUP));

3386

}

3387

3388

static char *make_temporary_directory(void);

3389

3390

static void test_add_inotify_dir_watch(__attribute__((unused))

3391

test_fixture *fixture,

3392

__attribute__((unused))

3393

gconstpointer user_data){

3394

__attribute__((cleanup(cleanup_close)))

3395

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3396

g_assert_cmpint(epoll_fd, >=, 0);

3397

__attribute__((cleanup(cleanup_queue)))

3398

task_queue *queue = create_queue();

3399

g_assert_nonnull(queue);

3400

__attribute__((cleanup(string_set_clear)))

3401

string_set cancelled_filenames = {};

3402

const mono_microsecs current_time = 0;

3403

3404

bool quit_now = false;

3405

buffer password = {};

3406

bool mandos_client_exited = false;

3407

bool password_is_read = false;

3408

3409

__attribute__((cleanup(cleanup_string)))

3410

char *tempdir = make_temporary_directory();

3411

g_assert_nonnull(tempdir);

3412

3413

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3414

&password, tempdir,

3415

&cancelled_filenames,

3416

&current_time,

3417

&mandos_client_exited,

3418

&password_is_read));

3419

3420

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3421

3422

const task_context *const added_read_task

3423

= find_matching_task(queue, (task_context){

3424

.func=read_inotify_event,

3425

.epoll_fd=epoll_fd,

3426

.quit_now=&quit_now,

3427

.password=&password,

3428

.filename=tempdir,

3429

.cancelled_filenames=&cancelled_filenames,

3430

.current_time=&current_time,

3431

.mandos_client_exited=&mandos_client_exited,

3432

.password_is_read=&password_is_read,

3433

});

3434

g_assert_nonnull(added_read_task);

3435

3436

g_assert_cmpint(added_read_task->fd, >, 2);

3437

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3438

g_assert_true(epoll_set_contains(added_read_task->epoll_fd,

3439

added_read_task->fd,

3440

EPOLLIN | EPOLLRDHUP));

3441

3442

g_assert_cmpint(rmdir(tempdir), ==, 0);

3443

}

3444

3445

static char *make_temporary_directory(void){

3446

char *name = strdup("/tmp/mandosXXXXXX");

3447

g_assert_nonnull(name);

3448

char *result = mkdtemp(name);

3449

if(result == NULL){

3450

free(name);

3451

}

3452

return result;

3453

}

3454

3455

static void test_add_inotify_dir_watch_fail(__attribute__((unused))

3456

test_fixture *fixture,

3457

__attribute__((unused))

3458

gconstpointer user_data){

3459

__attribute__((cleanup(cleanup_close)))

3460

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3461

g_assert_cmpint(epoll_fd, >=, 0);

3462

__attribute__((cleanup(cleanup_queue)))

3463

task_queue *queue = create_queue();

3464

g_assert_nonnull(queue);

3465

__attribute__((cleanup(string_set_clear)))

3466

string_set cancelled_filenames = {};

3467

const mono_microsecs current_time = 0;

3468

3469

bool quit_now = false;

3470

buffer password = {};

3471

bool mandos_client_exited = false;

3472

bool password_is_read = false;

3473

3474

const char nonexistent_dir[] = "/nonexistent";

3475

3476

FILE *real_stderr = stderr;

3477

FILE *devnull = fopen("/dev/null", "we");

3478

g_assert_nonnull(devnull);

3479

stderr = devnull;

3480

g_assert_false(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3481

&password, nonexistent_dir,

3482

&cancelled_filenames,

3483

&current_time,

3484

&mandos_client_exited,

3485

&password_is_read));

3486

stderr = real_stderr;

3487

g_assert_cmpint(fclose(devnull), ==, 0);

3488

3489

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3490

}

3491

3492

static void test_add_inotify_dir_watch_nondir(__attribute__((unused))

3493

test_fixture *fixture,

3494

__attribute__((unused))

3495

gconstpointer

3496

user_data){

3497

__attribute__((cleanup(cleanup_close)))

3498

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3499

g_assert_cmpint(epoll_fd, >=, 0);

3500

__attribute__((cleanup(cleanup_queue)))

3501

task_queue *queue = create_queue();

3502

g_assert_nonnull(queue);

3503

__attribute__((cleanup(string_set_clear)))

3504

string_set cancelled_filenames = {};

3505

const mono_microsecs current_time = 0;

3506

3507

bool quit_now = false;

3508

buffer password = {};

3509

bool mandos_client_exited = false;

3510

bool password_is_read = false;

3511

3512

const char not_a_directory[] = "/dev/tty";

3513

3514

FILE *real_stderr = stderr;

3515

FILE *devnull = fopen("/dev/null", "we");

3516

g_assert_nonnull(devnull);

3517

stderr = devnull;

3518

g_assert_false(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3519

&password, not_a_directory,

3520

&cancelled_filenames,

3521

&current_time,

3522

&mandos_client_exited,

3523

&password_is_read));

3524

stderr = real_stderr;

3525

g_assert_cmpint(fclose(devnull), ==, 0);

3526

3527

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3528

}

3529

3530

static void test_add_inotify_dir_watch_EAGAIN(__attribute__((unused))

3531

test_fixture *fixture,

3532

__attribute__((unused))

3533

gconstpointer

3534

user_data){

3535

__attribute__((cleanup(cleanup_close)))

3536

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3537

g_assert_cmpint(epoll_fd, >=, 0);

3538

__attribute__((cleanup(cleanup_queue)))

3539

task_queue *queue = create_queue();

3540

g_assert_nonnull(queue);

3541

__attribute__((cleanup(string_set_clear)))

3542

string_set cancelled_filenames = {};

3543

const mono_microsecs current_time = 0;

3544

3545

bool quit_now = false;

3546

buffer password = {};

3547

bool mandos_client_exited = false;

3548

bool password_is_read = false;

3549

3550

__attribute__((cleanup(cleanup_string)))

3551

char *tempdir = make_temporary_directory();

3552

g_assert_nonnull(tempdir);

3553

3554

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3555

&password, tempdir,

3556

&cancelled_filenames,

3557

&current_time,

3558

&mandos_client_exited,

3559

&password_is_read));

3560

3561

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3562

3563

const task_context *const added_read_task

3564

= find_matching_task(queue,

3565

(task_context){ .func=read_inotify_event });

3566

g_assert_nonnull(added_read_task);

3567

3568

g_assert_cmpint(added_read_task->fd, >, 2);

3569

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3570

3571

/* "sufficient to read at least one event." - inotify(7) */

3572

const size_t ievent_size = (sizeof(struct inotify_event)

3573

+ NAME_MAX + 1);

3574

struct inotify_event *ievent = malloc(ievent_size);

3575

g_assert_nonnull(ievent);

3576

3577

g_assert_cmpint(read(added_read_task->fd, ievent, ievent_size), ==,

3578

-1);

3579

g_assert_cmpint(errno, ==, EAGAIN);

3580

3581

free(ievent);

3582

3583

g_assert_cmpint(rmdir(tempdir), ==, 0);

3584

}

3585

3586

static char *make_temporary_file_in_directory(const char

3587

*const dir);

3588

3589

static

3590

void test_add_inotify_dir_watch_IN_CLOSE_WRITE(__attribute__((unused))

3591

test_fixture *fixture,

3592

__attribute__((unused))

3593

gconstpointer

3594

user_data){

3595

__attribute__((cleanup(cleanup_close)))

3596

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3597

g_assert_cmpint(epoll_fd, >=, 0);

3598

__attribute__((cleanup(cleanup_queue)))

3599

task_queue *queue = create_queue();

3600

g_assert_nonnull(queue);

3601

__attribute__((cleanup(string_set_clear)))

3602

string_set cancelled_filenames = {};

3603

const mono_microsecs current_time = 0;

3604

3605

bool quit_now = false;

3606

buffer password = {};

3607

bool mandos_client_exited = false;

3608

bool password_is_read = false;

3609

3610

__attribute__((cleanup(cleanup_string)))

3611

char *tempdir = make_temporary_directory();

3612

g_assert_nonnull(tempdir);

3613

3614

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3615

&password, tempdir,

3616

&cancelled_filenames,

3617

&current_time,

3618

&mandos_client_exited,

3619

&password_is_read));

3620

3621

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3622

3623

const task_context *const added_read_task

3624

= find_matching_task(queue,

3625

(task_context){ .func=read_inotify_event });

3626

g_assert_nonnull(added_read_task);

3627

3628

g_assert_cmpint(added_read_task->fd, >, 2);

3629

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3630

3631

__attribute__((cleanup(cleanup_string)))

3632

char *filename = make_temporary_file_in_directory(tempdir);

3633

g_assert_nonnull(filename);

3634

3635

/* "sufficient to read at least one event." - inotify(7) */

3636

const size_t ievent_size = (sizeof(struct inotify_event)

3637

+ NAME_MAX + 1);

3638

struct inotify_event *ievent = malloc(ievent_size);

3639

g_assert_nonnull(ievent);

3640

3641

ssize_t read_size = 0;

3642

read_size = read(added_read_task->fd, ievent, ievent_size);

3643

3644

g_assert_cmpint((int)read_size, >, 0);

3645

g_assert_true(ievent->mask & IN_CLOSE_WRITE);

3646

g_assert_cmpstr(ievent->name, ==, basename(filename));

3647

3648

free(ievent);

3649

3650

g_assert_cmpint(unlink(filename), ==, 0);

3651

g_assert_cmpint(rmdir(tempdir), ==, 0);

3652

}

3653

3654

static char *make_temporary_prefixed_file_in_directory(const char

3655

*const prefix,

3656

const char

3657

*const dir){

3658

char *filename = NULL;

3659

g_assert_cmpint(asprintf(&filename, "%s/%sXXXXXX", dir, prefix),

3660

>, 0);

3661

g_assert_nonnull(filename);

3662

const int fd = mkostemp(filename, O_CLOEXEC);

3663

g_assert_cmpint(fd, >=, 0);

3664

g_assert_cmpint(close(fd), ==, 0);

3665

return filename;

3666

}

3667

3668

static char *make_temporary_file_in_directory(const char

3669

*const dir){

3670

return make_temporary_prefixed_file_in_directory("temp", dir);

3671

}

3672

3673

static

3674

void test_add_inotify_dir_watch_IN_MOVED_TO(__attribute__((unused))

3675

test_fixture *fixture,

3676

__attribute__((unused))

3677

gconstpointer user_data){

3678

__attribute__((cleanup(cleanup_close)))

3679

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3680

g_assert_cmpint(epoll_fd, >=, 0);

3681

__attribute__((cleanup(cleanup_queue)))

3682

task_queue *queue = create_queue();

3683

g_assert_nonnull(queue);

3684

__attribute__((cleanup(string_set_clear)))

3685

string_set cancelled_filenames = {};

3686

const mono_microsecs current_time = 0;

3687

3688

bool quit_now = false;

3689

buffer password = {};

3690

bool mandos_client_exited = false;

3691

bool password_is_read = false;

3692

3693

__attribute__((cleanup(cleanup_string)))

3694

char *watchdir = make_temporary_directory();

3695

g_assert_nonnull(watchdir);

3696

3697

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3698

&password, watchdir,

3699

&cancelled_filenames,

3700

&current_time,

3701

&mandos_client_exited,

3702

&password_is_read));

3703

3704

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3705

3706

const task_context *const added_read_task

3707

= find_matching_task(queue,

3708

(task_context){ .func=read_inotify_event });

3709

g_assert_nonnull(added_read_task);

3710

3711

g_assert_cmpint(added_read_task->fd, >, 2);

3712

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3713

3714

char *sourcedir = make_temporary_directory();

3715

g_assert_nonnull(sourcedir);

3716

3717

__attribute__((cleanup(cleanup_string)))

3718

char *filename = make_temporary_file_in_directory(sourcedir);

3719

g_assert_nonnull(filename);

3720

3721

__attribute__((cleanup(cleanup_string)))

3722

char *targetfilename = NULL;

3723

g_assert_cmpint(asprintf(&targetfilename, "%s/%s", watchdir,

3724

basename(filename)), >, 0);

3725

g_assert_nonnull(targetfilename);

3726

3727

g_assert_cmpint(rename(filename, targetfilename), ==, 0);

3728

g_assert_cmpint(rmdir(sourcedir), ==, 0);

3729

free(sourcedir);

3730

3731

/* "sufficient to read at least one event." - inotify(7) */

3732

const size_t ievent_size = (sizeof(struct inotify_event)

3733

+ NAME_MAX + 1);

3734

struct inotify_event *ievent = malloc(ievent_size);

3735

g_assert_nonnull(ievent);

3736

3737

ssize_t read_size = read(added_read_task->fd, ievent, ievent_size);

3738

3739

g_assert_cmpint((int)read_size, >, 0);

3740

g_assert_true(ievent->mask & IN_MOVED_TO);

3741

g_assert_cmpstr(ievent->name, ==, basename(targetfilename));

3742

3743

free(ievent);

3744

3745

g_assert_cmpint(unlink(targetfilename), ==, 0);

3746

g_assert_cmpint(rmdir(watchdir), ==, 0);

3747

}

3748

3749

static

3750

void test_add_inotify_dir_watch_IN_MOVED_FROM(__attribute__((unused))

3751

test_fixture *fixture,

3752

__attribute__((unused))

3753

gconstpointer

3754

user_data){

3755

__attribute__((cleanup(cleanup_close)))

3756

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3757

g_assert_cmpint(epoll_fd, >=, 0);

3758

__attribute__((cleanup(cleanup_queue)))

3759

task_queue *queue = create_queue();

3760

g_assert_nonnull(queue);

3761

__attribute__((cleanup(string_set_clear)))

3762

string_set cancelled_filenames = {};

3763

const mono_microsecs current_time = 0;

3764

3765

bool quit_now = false;

3766

buffer password = {};

3767

bool mandos_client_exited = false;

3768

bool password_is_read = false;

3769

3770

__attribute__((cleanup(cleanup_string)))

3771

char *tempdir = make_temporary_directory();

3772

g_assert_nonnull(tempdir);

3773

3774

__attribute__((cleanup(cleanup_string)))

3775

char *tempfilename = make_temporary_file_in_directory(tempdir);

3776

g_assert_nonnull(tempfilename);

3777

3778

__attribute__((cleanup(cleanup_string)))

3779

char *targetdir = make_temporary_directory();

3780

g_assert_nonnull(targetdir);

3781

3782

__attribute__((cleanup(cleanup_string)))

3783

char *targetfilename = NULL;

3784

g_assert_cmpint(asprintf(&targetfilename, "%s/%s", targetdir,

3785

basename(tempfilename)), >, 0);

3786

g_assert_nonnull(targetfilename);

3787

3788

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3789

&password, tempdir,

3790

&cancelled_filenames,

3791

&current_time,

3792

&mandos_client_exited,

3793

&password_is_read));

3794

3795

g_assert_cmpint(rename(tempfilename, targetfilename), ==, 0);

3796

3797

const task_context *const added_read_task

3798

= find_matching_task(queue,

3799

(task_context){ .func=read_inotify_event });

3800

g_assert_nonnull(added_read_task);

3801

3802

/* "sufficient to read at least one event." - inotify(7) */

3803

const size_t ievent_size = (sizeof(struct inotify_event)

3804

+ NAME_MAX + 1);

3805

struct inotify_event *ievent = malloc(ievent_size);

3806

g_assert_nonnull(ievent);

3807

3808

ssize_t read_size = read(added_read_task->fd, ievent, ievent_size);

3809

3810

g_assert_cmpint((int)read_size, >, 0);

3811

g_assert_true(ievent->mask & IN_MOVED_FROM);

3812

g_assert_cmpstr(ievent->name, ==, basename(tempfilename));

3813

3814

free(ievent);

3815

3816

g_assert_cmpint(unlink(targetfilename), ==, 0);

3817

g_assert_cmpint(rmdir(targetdir), ==, 0);

3818

g_assert_cmpint(rmdir(tempdir), ==, 0);

3819

}

3820

3821

static

3822

void test_add_inotify_dir_watch_IN_DELETE(__attribute__((unused))

3823

test_fixture *fixture,

3824

__attribute__((unused))

3825

gconstpointer user_data){

3826

__attribute__((cleanup(cleanup_close)))

3827

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3828

g_assert_cmpint(epoll_fd, >=, 0);

3829

__attribute__((cleanup(cleanup_queue)))

3830

task_queue *queue = create_queue();

3831

g_assert_nonnull(queue);

3832

__attribute__((cleanup(string_set_clear)))

3833

string_set cancelled_filenames = {};

3834

const mono_microsecs current_time = 0;

3835

3836

bool quit_now = false;

3837

buffer password = {};

3838

bool mandos_client_exited = false;

3839

bool password_is_read = false;

3840

3841

__attribute__((cleanup(cleanup_string)))

3842

char *tempdir = make_temporary_directory();

3843

g_assert_nonnull(tempdir);

3844

3845

__attribute__((cleanup(cleanup_string)))

3846

char *tempfile = make_temporary_file_in_directory(tempdir);

3847

g_assert_nonnull(tempfile);

3848

3849

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3850

&password, tempdir,

3851

&cancelled_filenames,

3852

&current_time,

3853

&mandos_client_exited,

3854

&password_is_read));

3855

g_assert_cmpint(unlink(tempfile), ==, 0);

3856

3857

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3858

3859

const task_context *const added_read_task

3860

= find_matching_task(queue,

3861

(task_context){ .func=read_inotify_event });

3862

g_assert_nonnull(added_read_task);

3863

3864

g_assert_cmpint(added_read_task->fd, >, 2);

3865

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3866

3867

/* "sufficient to read at least one event." - inotify(7) */

3868

const size_t ievent_size = (sizeof(struct inotify_event)

3869

+ NAME_MAX + 1);

3870

struct inotify_event *ievent = malloc(ievent_size);

3871

g_assert_nonnull(ievent);

3872

3873

ssize_t read_size = 0;

3874

read_size = read(added_read_task->fd, ievent, ievent_size);

3875

3876

g_assert_cmpint((int)read_size, >, 0);

3877

g_assert_true(ievent->mask & IN_DELETE);

3878

g_assert_cmpstr(ievent->name, ==, basename(tempfile));

3879

3880

free(ievent);

3881

3882

g_assert_cmpint(rmdir(tempdir), ==, 0);

3883

}

3884

3885

static

3886

void test_add_inotify_dir_watch_IN_EXCL_UNLINK(__attribute__((unused))

3887

test_fixture *fixture,

3888

__attribute__((unused))

3889

gconstpointer

3890

user_data){

3891

__attribute__((cleanup(cleanup_close)))

3892

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3893

g_assert_cmpint(epoll_fd, >=, 0);

3894

__attribute__((cleanup(cleanup_queue)))

3895

task_queue *queue = create_queue();

3896

g_assert_nonnull(queue);

3897

__attribute__((cleanup(string_set_clear)))

3898

string_set cancelled_filenames = {};

3899

const mono_microsecs current_time = 0;

3900

3901

bool quit_now = false;

3902

buffer password = {};

3903

bool mandos_client_exited = false;

3904

bool password_is_read = false;

3905

3906

__attribute__((cleanup(cleanup_string)))

3907

char *tempdir = make_temporary_directory();

3908

g_assert_nonnull(tempdir);

3909

3910

__attribute__((cleanup(cleanup_string)))

3911

char *tempfile = make_temporary_file_in_directory(tempdir);

3912

g_assert_nonnull(tempfile);

3913

int tempfile_fd = open(tempfile, O_WRONLY | O_CLOEXEC | O_NOCTTY

3914

| O_NOFOLLOW);

3915

g_assert_cmpint(tempfile_fd, >, 2);

3916

3917

g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,

3918

&password, tempdir,

3919

&cancelled_filenames,

3920

&current_time,

3921

&mandos_client_exited,

3922

&password_is_read));

3923

g_assert_cmpint(unlink(tempfile), ==, 0);

3924

3925

g_assert_cmpuint((unsigned int)queue->length, >, 0);

3926

3927

const task_context *const added_read_task

3928

= find_matching_task(queue,

3929

(task_context){ .func=read_inotify_event });

3930

g_assert_nonnull(added_read_task);

3931

3932

g_assert_cmpint(added_read_task->fd, >, 2);

3933

g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));

3934

3935

/* "sufficient to read at least one event." - inotify(7) */

3936

const size_t ievent_size = (sizeof(struct inotify_event)

3937

+ NAME_MAX + 1);

3938

struct inotify_event *ievent = malloc(ievent_size);

3939

g_assert_nonnull(ievent);

3940

3941

ssize_t read_size = 0;

3942

read_size = read(added_read_task->fd, ievent, ievent_size);

3943

3944

g_assert_cmpint((int)read_size, >, 0);

3945

g_assert_true(ievent->mask & IN_DELETE);

3946

g_assert_cmpstr(ievent->name, ==, basename(tempfile));

3947

3948

g_assert_cmpint(close(tempfile_fd), ==, 0);

3949

3950

/* IN_EXCL_UNLINK should make the closing of the previously unlinked

3951

file not appear as an ievent, so we should not see it now. */

3952

read_size = read(added_read_task->fd, ievent, ievent_size);

3953

g_assert_cmpint((int)read_size, ==, -1);

3954

g_assert_true(errno == EAGAIN);

3955

3956

free(ievent);

3957

3958

g_assert_cmpint(rmdir(tempdir), ==, 0);

3959

}

3960

3961

static void test_read_inotify_event_readerror(__attribute__((unused))

3962

test_fixture *fixture,

3963

__attribute__((unused))

3964

gconstpointer

3965

user_data){

3966

__attribute__((cleanup(cleanup_close)))

3967

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

3968

g_assert_cmpint(epoll_fd, >=, 0);

3969

const mono_microsecs current_time = 0;

3970

3971

/* Reading /proc/self/mem from offset 0 will always result in EIO */

3972

const int fd = open("/proc/self/mem",

3973

O_RDONLY | O_CLOEXEC | O_NOCTTY);

3974

3975

bool quit_now = false;

3976

__attribute__((cleanup(cleanup_queue)))

3977

task_queue *queue = create_queue();

3978

g_assert_nonnull(queue);

3979

3980

task_context task = {

3981

.func=read_inotify_event,

3982

.epoll_fd=epoll_fd,

3983

.fd=fd,

3984

.quit_now=&quit_now,

3985

.filename=strdup("/nonexistent"),

3986

.cancelled_filenames = &(string_set){},

3987

.notafter=0,

3988

.current_time=&current_time,

3989

};

3990

g_assert_nonnull(task.filename);

3991

run_task_with_stderr_to_dev_null(task, queue);

3992

g_assert_true(quit_now);

3993

g_assert_true(queue->next_run == 0);

3994

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

3995

3996

g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));

3997

3998

g_assert_cmpint(close(fd), ==, -1);

3999

}

4000

4001

static void test_read_inotify_event_bad_epoll(__attribute__((unused))

4002

test_fixture *fixture,

4003

__attribute__((unused))

4004

gconstpointer

4005

user_data){

4006

const mono_microsecs current_time = 17;

4007

4008

int pipefds[2];

4009

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4010

const int epoll_fd = pipefds[0]; /* This will obviously fail */

4011

4012

bool quit_now = false;

4013

buffer password = {};

4014

bool mandos_client_exited = false;

4015

bool password_is_read = false;

4016

__attribute__((cleanup(cleanup_queue)))

4017

task_queue *queue = create_queue();

4018

g_assert_nonnull(queue);

4019

4020

task_context task = {

4021

.func=read_inotify_event,

4022

.epoll_fd=epoll_fd,

4023

.fd=pipefds[0],

4024

.quit_now=&quit_now,

4025

.password=&password,

4026

.filename=strdup("/nonexistent"),

4027

.cancelled_filenames = &(string_set){},

4028

.notafter=0,

4029

.current_time=&current_time,

4030

.mandos_client_exited=&mandos_client_exited,

4031

.password_is_read=&password_is_read,

4032

};

4033

g_assert_nonnull(task.filename);

4034

run_task_with_stderr_to_dev_null(task, queue);

4035

4036

g_assert_nonnull(find_matching_task(queue, task));

4037

g_assert_true(queue->next_run == 1000000 + current_time);

4038

4039

g_assert_cmpint(close(pipefds[0]), ==, 0);

4040

g_assert_cmpint(close(pipefds[1]), ==, 0);

4041

}

4042

4043

static void test_read_inotify_event_nodata(__attribute__((unused))

4044

test_fixture *fixture,

4045

__attribute__((unused))

4046

gconstpointer user_data){

4047

__attribute__((cleanup(cleanup_close)))

4048

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4049

g_assert_cmpint(epoll_fd, >=, 0);

4050

const mono_microsecs current_time = 0;

4051

4052

int pipefds[2];

4053

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4054

4055

bool quit_now = false;

4056

buffer password = {};

4057

bool mandos_client_exited = false;

4058

bool password_is_read = false;

4059

__attribute__((cleanup(cleanup_queue)))

4060

task_queue *queue = create_queue();

4061

g_assert_nonnull(queue);

4062

4063

task_context task = {

4064

.func=read_inotify_event,

4065

.epoll_fd=epoll_fd,

4066

.fd=pipefds[0],

4067

.quit_now=&quit_now,

4068

.password=&password,

4069

.filename=strdup("/nonexistent"),

4070

.cancelled_filenames = &(string_set){},

4071

.notafter=0,

4072

.current_time=&current_time,

4073

.mandos_client_exited=&mandos_client_exited,

4074

.password_is_read=&password_is_read,

4075

};

4076

g_assert_nonnull(task.filename);

4077

task.func(task, queue);

4078

g_assert_false(quit_now);

4079

g_assert_true(queue->next_run == 0);

4080

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4081

4082

g_assert_nonnull(find_matching_task(queue, (task_context){

4083

.func=read_inotify_event,

4084

.epoll_fd=epoll_fd,

4085

.fd=pipefds[0],

4086

.quit_now=&quit_now,

4087

.password=&password,

4088

.filename=task.filename,

4089

.cancelled_filenames=task.cancelled_filenames,

4090

.current_time=&current_time,

4091

.mandos_client_exited=&mandos_client_exited,

4092

.password_is_read=&password_is_read,

4093

}));

4094

4095

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4096

EPOLLIN | EPOLLRDHUP));

4097

4098

g_assert_cmpint(close(pipefds[1]), ==, 0);

4099

}

4100

4101

static void test_read_inotify_event_eof(__attribute__((unused))

4102

test_fixture *fixture,

4103

__attribute__((unused))

4104

gconstpointer user_data){

4105

__attribute__((cleanup(cleanup_close)))

4106

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4107

g_assert_cmpint(epoll_fd, >=, 0);

4108

const mono_microsecs current_time = 0;

4109

4110

int pipefds[2];

4111

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4112

g_assert_cmpint(close(pipefds[1]), ==, 0);

4113

4114

bool quit_now = false;

4115

buffer password = {};

4116

__attribute__((cleanup(cleanup_queue)))

4117

task_queue *queue = create_queue();

4118

g_assert_nonnull(queue);

4119

4120

task_context task = {

4121

.func=read_inotify_event,

4122

.epoll_fd=epoll_fd,

4123

.fd=pipefds[0],

4124

.quit_now=&quit_now,

4125

.password=&password,

4126

.filename=strdup("/nonexistent"),

4127

.cancelled_filenames = &(string_set){},

4128

.notafter=0,

4129

.current_time=&current_time,

4130

};

4131

run_task_with_stderr_to_dev_null(task, queue);

4132

g_assert_true(quit_now);

4133

g_assert_true(queue->next_run == 0);

4134

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

4135

4136

g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));

4137

4138

g_assert_cmpint(close(pipefds[0]), ==, -1);

4139

}

4140

4141

static

4142

void test_read_inotify_event_IN_CLOSE_WRITE(__attribute__((unused))

4143

test_fixture *fixture,

4144

__attribute__((unused))

4145

gconstpointer user_data){

4146

__attribute__((cleanup(cleanup_close)))

4147

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4148

g_assert_cmpint(epoll_fd, >=, 0);

4149

const mono_microsecs current_time = 0;

4150

4151

int pipefds[2];

4152

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4153

4154

/* "sufficient to read at least one event." - inotify(7) */

4155

const size_t ievent_max_size = (sizeof(struct inotify_event)

4156

+ NAME_MAX + 1);

4157

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4158

struct {

4159

struct inotify_event event;

4160

char name_buffer[NAME_MAX + 1];

4161

} ievent_buffer;

4162

struct inotify_event *const ievent = &ievent_buffer.event;

4163

4164

const char dummy_file_name[] = "ask.dummy_file_name";

4165

ievent->mask = IN_CLOSE_WRITE;

4166

ievent->len = sizeof(dummy_file_name);

4167

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4168

const size_t ievent_size = (sizeof(struct inotify_event)

4169

+ sizeof(dummy_file_name));

4170

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4171

==, ievent_size);

4172

g_assert_cmpint(close(pipefds[1]), ==, 0);

4173

4174

bool quit_now = false;

4175

buffer password = {};

4176

bool mandos_client_exited = false;

4177

bool password_is_read = false;

4178

__attribute__((cleanup(cleanup_queue)))

4179

task_queue *queue = create_queue();

4180

g_assert_nonnull(queue);

4181

4182

task_context task = {

4183

.func=read_inotify_event,

4184

.epoll_fd=epoll_fd,

4185

.fd=pipefds[0],

4186

.quit_now=&quit_now,

4187

.password=&password,

4188

.filename=strdup("/nonexistent"),

4189

.cancelled_filenames = &(string_set){},

4190

.notafter=0,

4191

.current_time=&current_time,

4192

.mandos_client_exited=&mandos_client_exited,

4193

.password_is_read=&password_is_read,

4194

};

4195

task.func(task, queue);

4196

g_assert_false(quit_now);

4197

g_assert_true(queue->next_run != 0);

4198

g_assert_cmpuint((unsigned int)queue->length, >=, 1);

4199

4200

g_assert_nonnull(find_matching_task(queue, (task_context){

4201

.func=read_inotify_event,

4202

.epoll_fd=epoll_fd,

4203

.fd=pipefds[0],

4204

.quit_now=&quit_now,

4205

.password=&password,

4206

.filename=task.filename,

4207

.cancelled_filenames=task.cancelled_filenames,

4208

.current_time=&current_time,

4209

.mandos_client_exited=&mandos_client_exited,

4210

.password_is_read=&password_is_read,

4211

}));

4212

4213

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4214

EPOLLIN | EPOLLRDHUP));

4215

4216

g_assert_cmpuint((unsigned int)queue->length, >=, 2);

4217

4218

__attribute__((cleanup(cleanup_string)))

4219

char *filename = NULL;

4220

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4221

dummy_file_name), >, 0);

4222

g_assert_nonnull(filename);

4223

g_assert_nonnull(find_matching_task(queue, (task_context){

4224

.func=open_and_parse_question,

4225

.epoll_fd=epoll_fd,

4226

.filename=filename,

4227

.question_filename=filename,

4228

.password=&password,

4229

.cancelled_filenames=task.cancelled_filenames,

4230

.current_time=&current_time,

4231

.mandos_client_exited=&mandos_client_exited,

4232

.password_is_read=&password_is_read,

4233

}));

4234

}

4235

4236

static

4237

void test_read_inotify_event_IN_MOVED_TO(__attribute__((unused))

4238

test_fixture *fixture,

4239

__attribute__((unused))

4240

gconstpointer user_data){

4241

__attribute__((cleanup(cleanup_close)))

4242

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4243

g_assert_cmpint(epoll_fd, >=, 0);

4244

const mono_microsecs current_time = 0;

4245

4246

int pipefds[2];

4247

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4248

4249

/* "sufficient to read at least one event." - inotify(7) */

4250

const size_t ievent_max_size = (sizeof(struct inotify_event)

4251

+ NAME_MAX + 1);

4252

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4253

struct {

4254

struct inotify_event event;

4255

char name_buffer[NAME_MAX + 1];

4256

} ievent_buffer;

4257

struct inotify_event *const ievent = &ievent_buffer.event;

4258

4259

const char dummy_file_name[] = "ask.dummy_file_name";

4260

ievent->mask = IN_MOVED_TO;

4261

ievent->len = sizeof(dummy_file_name);

4262

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4263

const size_t ievent_size = (sizeof(struct inotify_event)

4264

+ sizeof(dummy_file_name));

4265

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4266

==, ievent_size);

4267

g_assert_cmpint(close(pipefds[1]), ==, 0);

4268

4269

bool quit_now = false;

4270

buffer password = {};

4271

bool mandos_client_exited = false;

4272

bool password_is_read = false;

4273

__attribute__((cleanup(cleanup_queue)))

4274

task_queue *queue = create_queue();

4275

g_assert_nonnull(queue);

4276

4277

task_context task = {

4278

.func=read_inotify_event,

4279

.epoll_fd=epoll_fd,

4280

.fd=pipefds[0],

4281

.quit_now=&quit_now,

4282

.password=&password,

4283

.filename=strdup("/nonexistent"),

4284

.cancelled_filenames = &(string_set){},

4285

.notafter=0,

4286

.current_time=&current_time,

4287

.mandos_client_exited=&mandos_client_exited,

4288

.password_is_read=&password_is_read,

4289

};

4290

task.func(task, queue);

4291

g_assert_false(quit_now);

4292

g_assert_true(queue->next_run != 0);

4293

g_assert_cmpuint((unsigned int)queue->length, >=, 1);

4294

4295

g_assert_nonnull(find_matching_task(queue, (task_context){

4296

.func=read_inotify_event,

4297

.epoll_fd=epoll_fd,

4298

.fd=pipefds[0],

4299

.quit_now=&quit_now,

4300

.password=&password,

4301

.filename=task.filename,

4302

.cancelled_filenames=task.cancelled_filenames,

4303

.current_time=&current_time,

4304

.mandos_client_exited=&mandos_client_exited,

4305

.password_is_read=&password_is_read,

4306

}));

4307

4308

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4309

EPOLLIN | EPOLLRDHUP));

4310

4311

g_assert_cmpuint((unsigned int)queue->length, >=, 2);

4312

4313

__attribute__((cleanup(cleanup_string)))

4314

char *filename = NULL;

4315

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4316

dummy_file_name), >, 0);

4317

g_assert_nonnull(filename);

4318

g_assert_nonnull(find_matching_task(queue, (task_context){

4319

.func=open_and_parse_question,

4320

.epoll_fd=epoll_fd,

4321

.filename=filename,

4322

.question_filename=filename,

4323

.password=&password,

4324

.cancelled_filenames=task.cancelled_filenames,

4325

.current_time=&current_time,

4326

.mandos_client_exited=&mandos_client_exited,

4327

.password_is_read=&password_is_read,

4328

}));

4329

}

4330

4331

static

4332

void test_read_inotify_event_IN_MOVED_FROM(__attribute__((unused))

4333

test_fixture *fixture,

4334

__attribute__((unused))

4335

gconstpointer user_data){

4336

__attribute__((cleanup(cleanup_close)))

4337

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4338

g_assert_cmpint(epoll_fd, >=, 0);

4339

__attribute__((cleanup(string_set_clear)))

4340

string_set cancelled_filenames = {};

4341

const mono_microsecs current_time = 0;

4342

4343

int pipefds[2];

4344

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4345

4346

/* "sufficient to read at least one event." - inotify(7) */

4347

const size_t ievent_max_size = (sizeof(struct inotify_event)

4348

+ NAME_MAX + 1);

4349

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4350

struct {

4351

struct inotify_event event;

4352

char name_buffer[NAME_MAX + 1];

4353

} ievent_buffer;

4354

struct inotify_event *const ievent = &ievent_buffer.event;

4355

4356

const char dummy_file_name[] = "ask.dummy_file_name";

4357

ievent->mask = IN_MOVED_FROM;

4358

ievent->len = sizeof(dummy_file_name);

4359

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4360

const size_t ievent_size = (sizeof(struct inotify_event)

4361

+ sizeof(dummy_file_name));

4362

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4363

==, ievent_size);

4364

g_assert_cmpint(close(pipefds[1]), ==, 0);

4365

4366

bool quit_now = false;

4367

buffer password = {};

4368

bool mandos_client_exited = false;

4369

bool password_is_read = false;

4370

__attribute__((cleanup(cleanup_queue)))

4371

task_queue *queue = create_queue();

4372

g_assert_nonnull(queue);

4373

4374

task_context task = {

4375

.func=read_inotify_event,

4376

.epoll_fd=epoll_fd,

4377

.fd=pipefds[0],

4378

.quit_now=&quit_now,

4379

.password=&password,

4380

.filename=strdup("/nonexistent"),

4381

.cancelled_filenames=&cancelled_filenames,

4382

.current_time=&current_time,

4383

.mandos_client_exited=&mandos_client_exited,

4384

.password_is_read=&password_is_read,

4385

};

4386

task.func(task, queue);

4387

g_assert_false(quit_now);

4388

g_assert_true(queue->next_run == 0);

4389

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4390

4391

g_assert_nonnull(find_matching_task(queue, (task_context){

4392

.func=read_inotify_event,

4393

.epoll_fd=epoll_fd,

4394

.fd=pipefds[0],

4395

.quit_now=&quit_now,

4396

.password=&password,

4397

.filename=task.filename,

4398

.cancelled_filenames=&cancelled_filenames,

4399

.current_time=&current_time,

4400

.mandos_client_exited=&mandos_client_exited,

4401

.password_is_read=&password_is_read,

4402

}));

4403

4404

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4405

EPOLLIN | EPOLLRDHUP));

4406

4407

__attribute__((cleanup(cleanup_string)))

4408

char *filename = NULL;

4409

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4410

dummy_file_name), >, 0);

4411

g_assert_nonnull(filename);

4412

g_assert_true(string_set_contains(*task.cancelled_filenames,

4413

filename));

4414

}

4415

4416

static void test_read_inotify_event_IN_DELETE(__attribute__((unused))

4417

test_fixture *fixture,

4418

__attribute__((unused))

4419

gconstpointer

4420

user_data){

4421

__attribute__((cleanup(cleanup_close)))

4422

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4423

g_assert_cmpint(epoll_fd, >=, 0);

4424

__attribute__((cleanup(string_set_clear)))

4425

string_set cancelled_filenames = {};

4426

const mono_microsecs current_time = 0;

4427

4428

int pipefds[2];

4429

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4430

4431

/* "sufficient to read at least one event." - inotify(7) */

4432

const size_t ievent_max_size = (sizeof(struct inotify_event)

4433

+ NAME_MAX + 1);

4434

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4435

struct {

4436

struct inotify_event event;

4437

char name_buffer[NAME_MAX + 1];

4438

} ievent_buffer;

4439

struct inotify_event *const ievent = &ievent_buffer.event;

4440

4441

const char dummy_file_name[] = "ask.dummy_file_name";

4442

ievent->mask = IN_DELETE;

4443

ievent->len = sizeof(dummy_file_name);

4444

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4445

const size_t ievent_size = (sizeof(struct inotify_event)

4446

+ sizeof(dummy_file_name));

4447

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4448

==, ievent_size);

4449

g_assert_cmpint(close(pipefds[1]), ==, 0);

4450

4451

bool quit_now = false;

4452

buffer password = {};

4453

bool mandos_client_exited = false;

4454

bool password_is_read = false;

4455

__attribute__((cleanup(cleanup_queue)))

4456

task_queue *queue = create_queue();

4457

g_assert_nonnull(queue);

4458

4459

task_context task = {

4460

.func=read_inotify_event,

4461

.epoll_fd=epoll_fd,

4462

.fd=pipefds[0],

4463

.quit_now=&quit_now,

4464

.password=&password,

4465

.filename=strdup("/nonexistent"),

4466

.cancelled_filenames=&cancelled_filenames,

4467

.current_time=&current_time,

4468

.mandos_client_exited=&mandos_client_exited,

4469

.password_is_read=&password_is_read,

4470

};

4471

task.func(task, queue);

4472

g_assert_false(quit_now);

4473

g_assert_true(queue->next_run == 0);

4474

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4475

4476

g_assert_nonnull(find_matching_task(queue, (task_context){

4477

.func=read_inotify_event,

4478

.epoll_fd=epoll_fd,

4479

.fd=pipefds[0],

4480

.quit_now=&quit_now,

4481

.password=&password,

4482

.filename=task.filename,

4483

.cancelled_filenames=&cancelled_filenames,

4484

.current_time=&current_time,

4485

.mandos_client_exited=&mandos_client_exited,

4486

.password_is_read=&password_is_read,

4487

}));

4488

4489

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4490

EPOLLIN | EPOLLRDHUP));

4491

4492

__attribute__((cleanup(cleanup_string)))

4493

char *filename = NULL;

4494

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4495

dummy_file_name), >, 0);

4496

g_assert_nonnull(filename);

4497

g_assert_true(string_set_contains(*task.cancelled_filenames,

4498

filename));

4499

}

4500

4501

static void

4502

test_read_inotify_event_IN_CLOSE_WRITE_badname(__attribute__((unused))

4503

test_fixture *fixture,

4504

__attribute__((unused))

4505

gconstpointer

4506

user_data){

4507

__attribute__((cleanup(cleanup_close)))

4508

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4509

g_assert_cmpint(epoll_fd, >=, 0);

4510

const mono_microsecs current_time = 0;

4511

4512

int pipefds[2];

4513

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4514

4515

/* "sufficient to read at least one event." - inotify(7) */

4516

const size_t ievent_max_size = (sizeof(struct inotify_event)

4517

+ NAME_MAX + 1);

4518

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4519

struct {

4520

struct inotify_event event;

4521

char name_buffer[NAME_MAX + 1];

4522

} ievent_buffer;

4523

struct inotify_event *const ievent = &ievent_buffer.event;

4524

4525

const char dummy_file_name[] = "ignored.dummy_file_name";

4526

ievent->mask = IN_CLOSE_WRITE;

4527

ievent->len = sizeof(dummy_file_name);

4528

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4529

const size_t ievent_size = (sizeof(struct inotify_event)

4530

+ sizeof(dummy_file_name));

4531

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4532

==, ievent_size);

4533

g_assert_cmpint(close(pipefds[1]), ==, 0);

4534

4535

bool quit_now = false;

4536

buffer password = {};

4537

bool mandos_client_exited = false;

4538

bool password_is_read = false;

4539

__attribute__((cleanup(cleanup_queue)))

4540

task_queue *queue = create_queue();

4541

g_assert_nonnull(queue);

4542

4543

task_context task = {

4544

.func=read_inotify_event,

4545

.epoll_fd=epoll_fd,

4546

.fd=pipefds[0],

4547

.quit_now=&quit_now,

4548

.password=&password,

4549

.filename=strdup("/nonexistent"),

4550

.cancelled_filenames = &(string_set){},

4551

.notafter=0,

4552

.current_time=&current_time,

4553

.mandos_client_exited=&mandos_client_exited,

4554

.password_is_read=&password_is_read,

4555

};

4556

task.func(task, queue);

4557

g_assert_false(quit_now);

4558

g_assert_true(queue->next_run == 0);

4559

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4560

4561

g_assert_nonnull(find_matching_task(queue, (task_context){

4562

.func=read_inotify_event,

4563

.epoll_fd=epoll_fd,

4564

.fd=pipefds[0],

4565

.quit_now=&quit_now,

4566

.password=&password,

4567

.filename=task.filename,

4568

.cancelled_filenames=task.cancelled_filenames,

4569

.current_time=&current_time,

4570

.mandos_client_exited=&mandos_client_exited,

4571

.password_is_read=&password_is_read,

4572

}));

4573

4574

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4575

EPOLLIN | EPOLLRDHUP));

4576

}

4577

4578

static void

4579

test_read_inotify_event_IN_MOVED_TO_badname(__attribute__((unused))

4580

test_fixture *fixture,

4581

__attribute__((unused))

4582

gconstpointer user_data){

4583

__attribute__((cleanup(cleanup_close)))

4584

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4585

g_assert_cmpint(epoll_fd, >=, 0);

4586

const mono_microsecs current_time = 0;

4587

4588

int pipefds[2];

4589

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4590

4591

/* "sufficient to read at least one event." - inotify(7) */

4592

const size_t ievent_max_size = (sizeof(struct inotify_event)

4593

+ NAME_MAX + 1);

4594

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4595

struct {

4596

struct inotify_event event;

4597

char name_buffer[NAME_MAX + 1];

4598

} ievent_buffer;

4599

struct inotify_event *const ievent = &ievent_buffer.event;

4600

4601

const char dummy_file_name[] = "ignored.dummy_file_name";

4602

ievent->mask = IN_MOVED_TO;

4603

ievent->len = sizeof(dummy_file_name);

4604

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4605

const size_t ievent_size = (sizeof(struct inotify_event)

4606

+ sizeof(dummy_file_name));

4607

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4608

==, ievent_size);

4609

g_assert_cmpint(close(pipefds[1]), ==, 0);

4610

4611

bool quit_now = false;

4612

buffer password = {};

4613

bool mandos_client_exited = false;

4614

bool password_is_read = false;

4615

__attribute__((cleanup(cleanup_queue)))

4616

task_queue *queue = create_queue();

4617

g_assert_nonnull(queue);

4618

4619

task_context task = {

4620

.func=read_inotify_event,

4621

.epoll_fd=epoll_fd,

4622

.fd=pipefds[0],

4623

.quit_now=&quit_now,

4624

.password=&password,

4625

.filename=strdup("/nonexistent"),

4626

.cancelled_filenames = &(string_set){},

4627

.notafter=0,

4628

.current_time=&current_time,

4629

.mandos_client_exited=&mandos_client_exited,

4630

.password_is_read=&password_is_read,

4631

};

4632

task.func(task, queue);

4633

g_assert_false(quit_now);

4634

g_assert_true(queue->next_run == 0);

4635

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4636

4637

g_assert_nonnull(find_matching_task(queue, (task_context){

4638

.func=read_inotify_event,

4639

.epoll_fd=epoll_fd,

4640

.fd=pipefds[0],

4641

.quit_now=&quit_now,

4642

.password=&password,

4643

.filename=task.filename,

4644

.cancelled_filenames=task.cancelled_filenames,

4645

.current_time=&current_time,

4646

.mandos_client_exited=&mandos_client_exited,

4647

.password_is_read=&password_is_read,

4648

}));

4649

4650

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4651

EPOLLIN | EPOLLRDHUP));

4652

}

4653

4654

static void

4655

test_read_inotify_event_IN_MOVED_FROM_badname(__attribute__((unused))

4656

test_fixture *fixture,

4657

__attribute__((unused))

4658

gconstpointer

4659

user_data){

4660

__attribute__((cleanup(cleanup_close)))

4661

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4662

g_assert_cmpint(epoll_fd, >=, 0);

4663

__attribute__((cleanup(string_set_clear)))

4664

string_set cancelled_filenames = {};

4665

const mono_microsecs current_time = 0;

4666

4667

int pipefds[2];

4668

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4669

4670

/* "sufficient to read at least one event." - inotify(7) */

4671

const size_t ievent_max_size = (sizeof(struct inotify_event)

4672

+ NAME_MAX + 1);

4673

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4674

struct {

4675

struct inotify_event event;

4676

char name_buffer[NAME_MAX + 1];

4677

} ievent_buffer;

4678

struct inotify_event *const ievent = &ievent_buffer.event;

4679

4680

const char dummy_file_name[] = "ignored.dummy_file_name";

4681

ievent->mask = IN_MOVED_FROM;

4682

ievent->len = sizeof(dummy_file_name);

4683

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4684

const size_t ievent_size = (sizeof(struct inotify_event)

4685

+ sizeof(dummy_file_name));

4686

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4687

==, ievent_size);

4688

g_assert_cmpint(close(pipefds[1]), ==, 0);

4689

4690

bool quit_now = false;

4691

buffer password = {};

4692

bool mandos_client_exited = false;

4693

bool password_is_read = false;

4694

__attribute__((cleanup(cleanup_queue)))

4695

task_queue *queue = create_queue();

4696

g_assert_nonnull(queue);

4697

4698

task_context task = {

4699

.func=read_inotify_event,

4700

.epoll_fd=epoll_fd,

4701

.fd=pipefds[0],

4702

.quit_now=&quit_now,

4703

.password=&password,

4704

.filename=strdup("/nonexistent"),

4705

.cancelled_filenames=&cancelled_filenames,

4706

.current_time=&current_time,

4707

.mandos_client_exited=&mandos_client_exited,

4708

.password_is_read=&password_is_read,

4709

};

4710

task.func(task, queue);

4711

g_assert_false(quit_now);

4712

g_assert_true(queue->next_run == 0);

4713

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4714

4715

g_assert_nonnull(find_matching_task(queue, (task_context){

4716

.func=read_inotify_event,

4717

.epoll_fd=epoll_fd,

4718

.fd=pipefds[0],

4719

.quit_now=&quit_now,

4720

.password=&password,

4721

.filename=task.filename,

4722

.cancelled_filenames=&cancelled_filenames,

4723

.current_time=&current_time,

4724

.mandos_client_exited=&mandos_client_exited,

4725

.password_is_read=&password_is_read,

4726

}));

4727

4728

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4729

EPOLLIN | EPOLLRDHUP));

4730

4731

__attribute__((cleanup(cleanup_string)))

4732

char *filename = NULL;

4733

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4734

dummy_file_name), >, 0);

4735

g_assert_nonnull(filename);

4736

g_assert_false(string_set_contains(cancelled_filenames, filename));

4737

}

4738

4739

static

4740

void test_read_inotify_event_IN_DELETE_badname(__attribute__((unused))

4741

test_fixture *fixture,

4742

__attribute__((unused))

4743

gconstpointer

4744

user_data){

4745

__attribute__((cleanup(cleanup_close)))

4746

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4747

g_assert_cmpint(epoll_fd, >=, 0);

4748

__attribute__((cleanup(string_set_clear)))

4749

string_set cancelled_filenames = {};

4750

const mono_microsecs current_time = 0;

4751

4752

int pipefds[2];

4753

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

4754

4755

/* "sufficient to read at least one event." - inotify(7) */

4756

const size_t ievent_max_size = (sizeof(struct inotify_event)

4757

+ NAME_MAX + 1);

4758

g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);

4759

struct {

4760

struct inotify_event event;

4761

char name_buffer[NAME_MAX + 1];

4762

} ievent_buffer;

4763

struct inotify_event *const ievent = &ievent_buffer.event;

4764

4765

const char dummy_file_name[] = "ignored.dummy_file_name";

4766

ievent->mask = IN_DELETE;

4767

ievent->len = sizeof(dummy_file_name);

4768

memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));

4769

const size_t ievent_size = (sizeof(struct inotify_event)

4770

+ sizeof(dummy_file_name));

4771

g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),

4772

==, ievent_size);

4773

g_assert_cmpint(close(pipefds[1]), ==, 0);

4774

4775

bool quit_now = false;

4776

buffer password = {};

4777

bool mandos_client_exited = false;

4778

bool password_is_read = false;

4779

__attribute__((cleanup(cleanup_queue)))

4780

task_queue *queue = create_queue();

4781

g_assert_nonnull(queue);

4782

4783

task_context task = {

4784

.func=read_inotify_event,

4785

.epoll_fd=epoll_fd,

4786

.fd=pipefds[0],

4787

.quit_now=&quit_now,

4788

.password=&password,

4789

.filename=strdup("/nonexistent"),

4790

.cancelled_filenames=&cancelled_filenames,

4791

.current_time=&current_time,

4792

.mandos_client_exited=&mandos_client_exited,

4793

.password_is_read=&password_is_read,

4794

};

4795

task.func(task, queue);

4796

g_assert_false(quit_now);

4797

g_assert_true(queue->next_run == 0);

4798

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

4799

4800

g_assert_nonnull(find_matching_task(queue, (task_context){

4801

.func=read_inotify_event,

4802

.epoll_fd=epoll_fd,

4803

.fd=pipefds[0],

4804

.quit_now=&quit_now,

4805

.password=&password,

4806

.filename=task.filename,

4807

.cancelled_filenames=&cancelled_filenames,

4808

.current_time=&current_time,

4809

.mandos_client_exited=&mandos_client_exited,

4810

.password_is_read=&password_is_read,

4811

}));

4812

4813

g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],

4814

EPOLLIN | EPOLLRDHUP));

4815

4816

__attribute__((cleanup(cleanup_string)))

4817

char *filename = NULL;

4818

g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,

4819

dummy_file_name), >, 0);

4820

g_assert_nonnull(filename);

4821

g_assert_false(string_set_contains(cancelled_filenames, filename));

4822

}

4823

4824

static

4825

void test_open_and_parse_question_ENOENT(__attribute__((unused))

4826

test_fixture *fixture,

4827

__attribute__((unused))

4828

gconstpointer user_data){

4829

__attribute__((cleanup(cleanup_close)))

4830

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4831

g_assert_cmpint(epoll_fd, >=, 0);

4832

__attribute__((cleanup(string_set_clear)))

4833

string_set cancelled_filenames = {};

4834

bool mandos_client_exited = false;

4835

bool password_is_read = false;

4836

__attribute__((cleanup(cleanup_queue)))

4837

task_queue *queue = create_queue();

4838

g_assert_nonnull(queue);

4839

4840

char *const filename = strdup("/nonexistent");

4841

g_assert_nonnull(filename);

4842

task_context task = {

4843

.func=open_and_parse_question,

4844

.question_filename=filename,

4845

.epoll_fd=epoll_fd,

4846

.password=(buffer[]){{}},

4847

.filename=filename,

4848

.cancelled_filenames=&cancelled_filenames,

4849

.current_time=(mono_microsecs[]){0},

4850

.mandos_client_exited=&mandos_client_exited,

4851

.password_is_read=&password_is_read,

4852

};

4853

task.func(task, queue);

4854

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

4855

}

4856

4857

static void test_open_and_parse_question_EIO(__attribute__((unused))

4858

test_fixture *fixture,

4859

__attribute__((unused))

4860

gconstpointer user_data){

4861

__attribute__((cleanup(cleanup_close)))

4862

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4863

g_assert_cmpint(epoll_fd, >=, 0);

4864

__attribute__((cleanup(string_set_clear)))

4865

string_set cancelled_filenames = {};

4866

buffer password = {};

4867

bool mandos_client_exited = false;

4868

bool password_is_read = false;

4869

__attribute__((cleanup(cleanup_queue)))

4870

task_queue *queue = create_queue();

4871

g_assert_nonnull(queue);

4872

const mono_microsecs current_time = 0;

4873

4874

char *filename = strdup("/proc/self/mem");

4875

task_context task = {

4876

.func=open_and_parse_question,

4877

.question_filename=filename,

4878

.epoll_fd=epoll_fd,

4879

.password=&password,

4880

.filename=filename,

4881

.cancelled_filenames=&cancelled_filenames,

4882

.current_time=&current_time,

4883

.mandos_client_exited=&mandos_client_exited,

4884

.password_is_read=&password_is_read,

4885

};

4886

run_task_with_stderr_to_dev_null(task, queue);

4887

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

4888

}

4889

4890

static void

4891

test_open_and_parse_question_parse_error(__attribute__((unused))

4892

test_fixture *fixture,

4893

__attribute__((unused))

4894

gconstpointer user_data){

4895

__attribute__((cleanup(cleanup_close)))

4896

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4897

g_assert_cmpint(epoll_fd, >=, 0);

4898

__attribute__((cleanup(string_set_clear)))

4899

string_set cancelled_filenames = {};

4900

__attribute__((cleanup(cleanup_queue)))

4901

task_queue *queue = create_queue();

4902

g_assert_nonnull(queue);

4903

4904

__attribute__((cleanup(cleanup_string)))

4905

char *tempfilename = strdup("/tmp/mandosXXXXXX");

4906

g_assert_nonnull(tempfilename);

4907

int tempfile = mkostemp(tempfilename, O_CLOEXEC);

4908

g_assert_cmpint(tempfile, >, 0);

4909

const char bad_data[] = "this is bad syntax\n";

4910

g_assert_cmpint(write(tempfile, bad_data, sizeof(bad_data)),

4911

==, sizeof(bad_data));

4912

g_assert_cmpint(close(tempfile), ==, 0);

4913

4914

char *const filename = strdup(tempfilename);

4915

g_assert_nonnull(filename);

4916

task_context task = {

4917

.func=open_and_parse_question,

4918

.question_filename=filename,

4919

.epoll_fd=epoll_fd,

4920

.password=(buffer[]){{}},

4921

.filename=filename,

4922

.cancelled_filenames=&cancelled_filenames,

4923

.current_time=(mono_microsecs[]){0},

4924

.mandos_client_exited=(bool[]){false},

4925

.password_is_read=(bool[]){false},

4926

};

4927

run_task_with_stderr_to_dev_null(task, queue);

4928

4929

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

4930

4931

g_assert_cmpint(unlink(tempfilename), ==, 0);

4932

}

4933

4934

static

4935

void test_open_and_parse_question_nosocket(__attribute__((unused))

4936

test_fixture *fixture,

4937

__attribute__((unused))

4938

gconstpointer user_data){

4939

__attribute__((cleanup(cleanup_close)))

4940

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4941

g_assert_cmpint(epoll_fd, >=, 0);

4942

__attribute__((cleanup(string_set_clear)))

4943

string_set cancelled_filenames = {};

4944

__attribute__((cleanup(cleanup_queue)))

4945

task_queue *queue = create_queue();

4946

g_assert_nonnull(queue);

4947

4948

__attribute__((cleanup(cleanup_string)))

4949

char *tempfilename = strdup("/tmp/mandosXXXXXX");

4950

g_assert_nonnull(tempfilename);

4951

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

4952

g_assert_cmpint(questionfile, >, 0);

4953

FILE *qf = fdopen(questionfile, "w");

4954

g_assert_cmpint(fprintf(qf, "[Ask]\nPID=1\n"), >, 0);

4955

g_assert_cmpint(fclose(qf), ==, 0);

4956

4957

char *const filename = strdup(tempfilename);

4958

g_assert_nonnull(filename);

4959

task_context task = {

4960

.func=open_and_parse_question,

4961

.question_filename=filename,

4962

.epoll_fd=epoll_fd,

4963

.password=(buffer[]){{}},

4964

.filename=filename,

4965

.cancelled_filenames=&cancelled_filenames,

4966

.current_time=(mono_microsecs[]){0},

4967

.mandos_client_exited=(bool[]){false},

4968

.password_is_read=(bool[]){false},

4969

};

4970

run_task_with_stderr_to_dev_null(task, queue);

4971

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

4972

4973

g_assert_cmpint(unlink(tempfilename), ==, 0);

4974

}

4975

4976

static

4977

void test_open_and_parse_question_badsocket(__attribute__((unused))

4978

test_fixture *fixture,

4979

__attribute__((unused))

4980

gconstpointer user_data){

4981

__attribute__((cleanup(cleanup_close)))

4982

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

4983

g_assert_cmpint(epoll_fd, >=, 0);

4984

__attribute__((cleanup(string_set_clear)))

4985

string_set cancelled_filenames = {};

4986

__attribute__((cleanup(cleanup_queue)))

4987

task_queue *queue = create_queue();

4988

g_assert_nonnull(queue);

4989

4990

__attribute__((cleanup(cleanup_string)))

4991

char *tempfilename = strdup("/tmp/mandosXXXXXX");

4992

g_assert_nonnull(tempfilename);

4993

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

4994

g_assert_cmpint(questionfile, >, 0);

4995

FILE *qf = fdopen(questionfile, "w");

4996

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=\nPID=1\n"), >, 0);

4997

g_assert_cmpint(fclose(qf), ==, 0);

4998

4999

char *const filename = strdup(tempfilename);

5000

g_assert_nonnull(filename);

5001

task_context task = {

5002

.func=open_and_parse_question,

5003

.question_filename=filename,

5004

.epoll_fd=epoll_fd,

5005

.password=(buffer[]){{}},

5006

.filename=filename,

5007

.cancelled_filenames=&cancelled_filenames,

5008

.current_time=(mono_microsecs[]){0},

5009

.mandos_client_exited=(bool[]){false},

5010

.password_is_read=(bool[]){false},

5011

};

5012

run_task_with_stderr_to_dev_null(task, queue);

5013

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5014

5015

g_assert_cmpint(unlink(tempfilename), ==, 0);

5016

}

5017

5018

static

5019

void test_open_and_parse_question_nopid(__attribute__((unused))

5020

test_fixture *fixture,

5021

__attribute__((unused))

5022

gconstpointer user_data){

5023

__attribute__((cleanup(cleanup_close)))

5024

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5025

g_assert_cmpint(epoll_fd, >=, 0);

5026

__attribute__((cleanup(string_set_clear)))

5027

string_set cancelled_filenames = {};

5028

__attribute__((cleanup(cleanup_queue)))

5029

task_queue *queue = create_queue();

5030

g_assert_nonnull(queue);

5031

5032

__attribute__((cleanup(cleanup_string)))

5033

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5034

g_assert_nonnull(tempfilename);

5035

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5036

g_assert_cmpint(questionfile, >, 0);

5037

FILE *qf = fdopen(questionfile, "w");

5038

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\n"), >, 0);

5039

g_assert_cmpint(fclose(qf), ==, 0);

5040

5041

char *const filename = strdup(tempfilename);

5042

g_assert_nonnull(filename);

5043

task_context task = {

5044

.func=open_and_parse_question,

5045

.question_filename=filename,

5046

.epoll_fd=epoll_fd,

5047

.password=(buffer[]){{}},

5048

.filename=filename,

5049

.cancelled_filenames=&cancelled_filenames,

5050

.current_time=(mono_microsecs[]){0},

5051

.mandos_client_exited=(bool[]){false},

5052

.password_is_read=(bool[]){false},

5053

};

5054

run_task_with_stderr_to_dev_null(task, queue);

5055

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5056

5057

g_assert_cmpint(unlink(tempfilename), ==, 0);

5058

}

5059

5060

static

5061

void test_open_and_parse_question_badpid(__attribute__((unused))

5062

test_fixture *fixture,

5063

__attribute__((unused))

5064

gconstpointer user_data){

5065

__attribute__((cleanup(cleanup_close)))

5066

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5067

g_assert_cmpint(epoll_fd, >=, 0);

5068

__attribute__((cleanup(string_set_clear)))

5069

string_set cancelled_filenames = {};

5070

__attribute__((cleanup(cleanup_queue)))

5071

task_queue *queue = create_queue();

5072

g_assert_nonnull(queue);

5073

5074

__attribute__((cleanup(cleanup_string)))

5075

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5076

g_assert_nonnull(tempfilename);

5077

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5078

g_assert_cmpint(questionfile, >, 0);

5079

FILE *qf = fdopen(questionfile, "w");

5080

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=\n"),

5081

>, 0);

5082

g_assert_cmpint(fclose(qf), ==, 0);

5083

5084

char *const filename = strdup(tempfilename);

5085

g_assert_nonnull(filename);

5086

task_context task = {

5087

.func=open_and_parse_question,

5088

.question_filename=filename,

5089

.epoll_fd=epoll_fd,

5090

.password=(buffer[]){{}},

5091

.filename=filename,

5092

.cancelled_filenames=&cancelled_filenames,

5093

.current_time=(mono_microsecs[]){0},

5094

.mandos_client_exited=(bool[]){false},

5095

.password_is_read=(bool[]){false},

5096

};

5097

run_task_with_stderr_to_dev_null(task, queue);

5098

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5099

5100

g_assert_cmpint(unlink(tempfilename), ==, 0);

5101

}

5102

5103

static void

5104

test_open_and_parse_question_noexist_pid(__attribute__((unused))

5105

test_fixture *fixture,

5106

__attribute__((unused))

5107

gconstpointer user_data){

5108

__attribute__((cleanup(cleanup_close)))

5109

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5110

g_assert_cmpint(epoll_fd, >=, 0);

5111

__attribute__((cleanup(string_set_clear)))

5112

string_set cancelled_filenames = {};

5113

buffer password = {};

5114

bool mandos_client_exited = false;

5115

bool password_is_read = false;

5116

__attribute__((cleanup(cleanup_queue)))

5117

task_queue *queue = create_queue();

5118

g_assert_nonnull(queue);

5119

const mono_microsecs current_time = 0;

5120

5121

/* Find value of sysctl kernel.pid_max */

5122

uintmax_t pid_max = 0;

5123

FILE *sysctl_pid_max = fopen("/proc/sys/kernel/pid_max", "r");

5124

g_assert_nonnull(sysctl_pid_max);

5125

g_assert_cmpint(fscanf(sysctl_pid_max, "%" PRIuMAX, &pid_max),

5126

==, 1);

5127

g_assert_cmpint(fclose(sysctl_pid_max), ==, 0);

5128

5129

pid_t nonexisting_pid = ((pid_t)pid_max)+1;

5130

g_assert_true(nonexisting_pid > 0); /* Overflow check */

5131

5132

__attribute__((cleanup(cleanup_string)))

5133

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5134

g_assert_nonnull(tempfilename);

5135

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5136

g_assert_cmpint(questionfile, >, 0);

5137

FILE *qf = fdopen(questionfile, "w");

5138

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"

5139

PRIuMAX"\n", (uintmax_t)nonexisting_pid),

5140

>, 0);

5141

g_assert_cmpint(fclose(qf), ==, 0);

5142

5143

char *const question_filename = strdup(tempfilename);

5144

g_assert_nonnull(question_filename);

5145

task_context task = {

5146

.func=open_and_parse_question,

5147

.question_filename=question_filename,

5148

.epoll_fd=epoll_fd,

5149

.password=&password,

5150

.filename=question_filename,

5151

.cancelled_filenames=&cancelled_filenames,

5152

.current_time=&current_time,

5153

.mandos_client_exited=&mandos_client_exited,

5154

.password_is_read=&password_is_read,

5155

};

5156

run_task_with_stderr_to_dev_null(task, queue);

5157

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5158

5159

g_assert_cmpint(unlink(tempfilename), ==, 0);

5160

}

5161

5162

static void

5163

test_open_and_parse_question_no_notafter(__attribute__((unused))

5164

test_fixture *fixture,

5165

__attribute__((unused))

5166

gconstpointer user_data){

5167

__attribute__((cleanup(cleanup_close)))

5168

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5169

g_assert_cmpint(epoll_fd, >=, 0);

5170

__attribute__((cleanup(string_set_clear)))

5171

string_set cancelled_filenames = {};

5172

buffer password = {};

5173

bool mandos_client_exited = false;

5174

bool password_is_read = false;

5175

__attribute__((cleanup(cleanup_queue)))

5176

task_queue *queue = create_queue();

5177

g_assert_nonnull(queue);

5178

const mono_microsecs current_time = 0;

5179

5180

__attribute__((cleanup(cleanup_string)))

5181

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5182

g_assert_nonnull(tempfilename);

5183

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5184

g_assert_cmpint(questionfile, >, 0);

5185

FILE *qf = fdopen(questionfile, "w");

5186

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"

5187

PRIuMAX "\n", (uintmax_t)getpid()), >, 0);

5188

g_assert_cmpint(fclose(qf), ==, 0);

5189

5190

char *const filename = strdup(tempfilename);

5191

g_assert_nonnull(filename);

5192

task_context task = {

5193

.func=open_and_parse_question,

5194

.question_filename=filename,

5195

.epoll_fd=epoll_fd,

5196

.password=&password,

5197

.filename=filename,

5198

.cancelled_filenames=&cancelled_filenames,

5199

.current_time=&current_time,

5200

.mandos_client_exited=&mandos_client_exited,

5201

.password_is_read=&password_is_read,

5202

};

5203

task.func(task, queue);

5204

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5205

5206

__attribute__((cleanup(cleanup_string)))

5207

char *socket_filename = strdup("/nonexistent");

5208

g_assert_nonnull(socket_filename);

5209

g_assert_nonnull(find_matching_task(queue, (task_context){

5210

.func=connect_question_socket,

5211

.question_filename=tempfilename,

5212

.filename=socket_filename,

5213

.epoll_fd=epoll_fd,

5214

.password=&password,

5215

.current_time=&current_time,

5216

.mandos_client_exited=&mandos_client_exited,

5217

.password_is_read=&password_is_read,

5218

}));

5219

5220

g_assert_true(queue->next_run != 0);

5221

5222

g_assert_cmpint(unlink(tempfilename), ==, 0);

5223

}

5224

5225

static void

5226

test_open_and_parse_question_bad_notafter(__attribute__((unused))

5227

test_fixture *fixture,

5228

__attribute__((unused))

5229

gconstpointer user_data){

5230

__attribute__((cleanup(cleanup_close)))

5231

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5232

g_assert_cmpint(epoll_fd, >=, 0);

5233

__attribute__((cleanup(string_set_clear)))

5234

string_set cancelled_filenames = {};

5235

buffer password = {};

5236

bool mandos_client_exited = false;

5237

bool password_is_read = false;

5238

__attribute__((cleanup(cleanup_queue)))

5239

task_queue *queue = create_queue();

5240

g_assert_nonnull(queue);

5241

const mono_microsecs current_time = 0;

5242

5243

__attribute__((cleanup(cleanup_string)))

5244

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5245

g_assert_nonnull(tempfilename);

5246

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5247

g_assert_cmpint(questionfile, >, 0);

5248

FILE *qf = fdopen(questionfile, "w");

5249

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"

5250

PRIuMAX "\nNotAfter=\n",

5251

(uintmax_t)getpid()), >, 0);

5252

g_assert_cmpint(fclose(qf), ==, 0);

5253

5254

char *const filename = strdup(tempfilename);

5255

g_assert_nonnull(filename);

5256

task_context task = {

5257

.func=open_and_parse_question,

5258

.question_filename=filename,

5259

.epoll_fd=epoll_fd,

5260

.password=&password,

5261

.filename=filename,

5262

.cancelled_filenames=&cancelled_filenames,

5263

.current_time=&current_time,

5264

.mandos_client_exited=&mandos_client_exited,

5265

.password_is_read=&password_is_read,

5266

};

5267

run_task_with_stderr_to_dev_null(task, queue);

5268

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5269

5270

__attribute__((cleanup(cleanup_string)))

5271

char *socket_filename = strdup("/nonexistent");

5272

g_assert_nonnull(find_matching_task(queue, (task_context){

5273

.func=connect_question_socket,

5274

.question_filename=tempfilename,

5275

.filename=socket_filename,

5276

.epoll_fd=epoll_fd,

5277

.password=&password,

5278

.current_time=&current_time,

5279

.mandos_client_exited=&mandos_client_exited,

5280

.password_is_read=&password_is_read,

5281

}));

5282

g_assert_true(queue->next_run != 0);

5283

5284

g_assert_cmpint(unlink(tempfilename), ==, 0);

5285

}

5286

5287

static

5288

void assert_open_and_parse_question_with_notafter(const mono_microsecs

5289

current_time,

5290

const mono_microsecs

5291

notafter,

5292

const mono_microsecs

5293

next_queue_run){

5294

__attribute__((cleanup(cleanup_close)))

5295

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5296

g_assert_cmpint(epoll_fd, >=, 0);

5297

__attribute__((cleanup(string_set_clear)))

5298

string_set cancelled_filenames = {};

5299

buffer password = {};

5300

bool mandos_client_exited = false;

5301

bool password_is_read = false;

5302

__attribute__((cleanup(cleanup_queue)))

5303

task_queue *queue = create_queue();

5304

g_assert_nonnull(queue);

5305

queue->next_run = next_queue_run;

5306

5307

__attribute__((cleanup(cleanup_string)))

5308

char *tempfilename = strdup("/tmp/mandosXXXXXX");

5309

g_assert_nonnull(tempfilename);

5310

int questionfile = mkostemp(tempfilename, O_CLOEXEC);

5311

g_assert_cmpint(questionfile, >, 0);

5312

FILE *qf = fdopen(questionfile, "w");

5313

g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"

5314

PRIuMAX "\nNotAfter=%" PRIuMAX "\n",

5315

(uintmax_t)getpid(), notafter), >, 0);

5316

g_assert_cmpint(fclose(qf), ==, 0);

5317

5318

char *const filename = strdup(tempfilename);

5319

g_assert_nonnull(filename);

5320

task_context task = {

5321

.func=open_and_parse_question,

5322

.question_filename=filename,

5323

.epoll_fd=epoll_fd,

5324

.password=&password,

5325

.filename=filename,

5326

.cancelled_filenames=&cancelled_filenames,

5327

.current_time=&current_time,

5328

.mandos_client_exited=&mandos_client_exited,

5329

.password_is_read=&password_is_read,

5330

};

5331

task.func(task, queue);

5332

5333

if(queue->length >= 1){

5334

__attribute__((cleanup(cleanup_string)))

5335

char *socket_filename = strdup("/nonexistent");

5336

g_assert_nonnull(find_matching_task(queue, (task_context){

5337

.func=connect_question_socket,

5338

.filename=socket_filename,

5339

.epoll_fd=epoll_fd,

5340

.password=&password,

5341

.current_time=&current_time,

5342

.cancelled_filenames=&cancelled_filenames,

5343

.mandos_client_exited=&mandos_client_exited,

5344

.password_is_read=&password_is_read,

5345

}));

5346

g_assert_true(queue->next_run != 0);

5347

}

5348

5349

if(notafter == 0){

5350

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5351

} else if(current_time >= notafter) {

5352

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5353

} else {

5354

g_assert_nonnull(find_matching_task(queue, (task_context){

5355

.func=cancel_old_question,

5356

.question_filename=tempfilename,

5357

.filename=tempfilename,

5358

.notafter=notafter,

5359

.cancelled_filenames=&cancelled_filenames,

5360

.current_time=&current_time,

5361

}));

5362

}

5363

g_assert_true(queue->next_run == 1);

5364

5365

g_assert_cmpint(unlink(tempfilename), ==, 0);

5366

}

5367

5368

static void

5369

test_open_and_parse_question_notafter_0(__attribute__((unused))

5370

test_fixture *fixture,

5371

__attribute__((unused))

5372

gconstpointer user_data){

5373

/* current_time, notafter, next_queue_run */

5374

assert_open_and_parse_question_with_notafter(0, 0, 0);

5375

}

5376

5377

static void

5378

test_open_and_parse_question_notafter_1(__attribute__((unused))

5379

test_fixture *fixture,

5380

__attribute__((unused))

5381

gconstpointer user_data){

5382

/* current_time, notafter, next_queue_run */

5383

assert_open_and_parse_question_with_notafter(0, 1, 0);

5384

}

5385

5386

static void

5387

test_open_and_parse_question_notafter_1_1(__attribute__((unused))

5388

test_fixture *fixture,

5389

__attribute__((unused))

5390

gconstpointer user_data){

5391

/* current_time, notafter, next_queue_run */

5392

assert_open_and_parse_question_with_notafter(0, 1, 1);

5393

}

5394

5395

static void

5396

test_open_and_parse_question_notafter_1_2(__attribute__((unused))

5397

test_fixture *fixture,

5398

__attribute__((unused))

5399

gconstpointer user_data){

5400

/* current_time, notafter, next_queue_run */

5401

assert_open_and_parse_question_with_notafter(0, 1, 2);

5402

}

5403

5404

static void

5405

test_open_and_parse_question_equal_notafter(__attribute__((unused))

5406

test_fixture *fixture,

5407

__attribute__((unused))

5408

gconstpointer user_data){

5409

/* current_time, notafter, next_queue_run */

5410

assert_open_and_parse_question_with_notafter(1, 1, 0);

5411

}

5412

5413

static void

5414

test_open_and_parse_question_late_notafter(__attribute__((unused))

5415

test_fixture *fixture,

5416

__attribute__((unused))

5417

gconstpointer user_data){

5418

/* current_time, notafter, next_queue_run */

5419

assert_open_and_parse_question_with_notafter(2, 1, 0);

5420

}

5421

5422

static void assert_cancel_old_question_param(const mono_microsecs

5423

next_queue_run,

5424

const mono_microsecs

5425

notafter,

5426

const mono_microsecs

5427

current_time,

5428

const mono_microsecs

5429

next_set_to){

5430

__attribute__((cleanup(string_set_clear)))

5431

string_set cancelled_filenames = {};

5432

__attribute__((cleanup(cleanup_queue)))

5433

task_queue *queue = create_queue();

5434

g_assert_nonnull(queue);

5435

queue->next_run = next_queue_run;

5436

5437

char *const question_filename = strdup("/nonexistent");

5438

g_assert_nonnull(question_filename);

5439

task_context task = {

5440

.func=cancel_old_question,

5441

.question_filename=question_filename,

5442

.filename=question_filename,

5443

.notafter=notafter,

5444

.cancelled_filenames=&cancelled_filenames,

5445

.current_time=&current_time,

5446

};

5447

task.func(task, queue);

5448

5449

if(current_time >= notafter){

5450

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5451

g_assert_true(string_set_contains(cancelled_filenames,

5452

"/nonexistent"));

5453

} else {

5454

g_assert_nonnull(find_matching_task(queue, (task_context){

5455

.func=cancel_old_question,

5456

.question_filename=question_filename,

5457

.filename=question_filename,

5458

.notafter=notafter,

5459

.cancelled_filenames=&cancelled_filenames,

5460

.current_time=&current_time,

5461

}));

5462

5463

g_assert_false(string_set_contains(cancelled_filenames,

5464

question_filename));

5465

}

5466

g_assert_cmpuint((unsigned int)queue->next_run, ==,

5467

(unsigned int)next_set_to);

5468

}

5469

5470

static void test_cancel_old_question_0_1_2(__attribute__((unused))

5471

test_fixture *fixture,

5472

__attribute__((unused))

5473

gconstpointer user_data){

5474

/* next_queue_run unset,

5475

cancellation should happen because time has come,

5476

next_queue_run should be unchanged */

5477

/* next_queue_run, notafter, current_time, next_set_to */

5478

assert_cancel_old_question_param(0, 1, 2, 0);

5479

}

5480

5481

static void test_cancel_old_question_0_2_1(__attribute__((unused))

5482

test_fixture *fixture,

5483

__attribute__((unused))

5484

gconstpointer user_data){

5485

/* If next_queue_run is 0, meaning unset, and notafter is 2,

5486

and current_time is not yet notafter or greater,

5487

update value of next_queue_run to value of notafter */

5488

/* next_queue_run, notafter, current_time, next_set_to */

5489

assert_cancel_old_question_param(0, 2, 1, 2);

5490

}

5491

5492

static void test_cancel_old_question_1_2_3(__attribute__((unused))

5493

test_fixture *fixture,

5494

__attribute__((unused))

5495

gconstpointer user_data){

5496

/* next_queue_run 1,

5497

cancellation should happen because time has come,

5498

next_queue_run should be unchanged */

5499

/* next_queue_run, notafter, current_time, next_set_to */

5500

assert_cancel_old_question_param(1, 2, 3, 1);

5501

}

5502

5503

static void test_cancel_old_question_1_3_2(__attribute__((unused))

5504

test_fixture *fixture,

5505

__attribute__((unused))

5506

gconstpointer user_data){

5507

/* If next_queue_run is set,

5508

and current_time is not yet notafter or greater,

5509

and notafter is larger than next_queue_run

5510

next_queue_run should be unchanged */

5511

/* next_queue_run, notafter, current_time, next_set_to */

5512

assert_cancel_old_question_param(1, 3, 2, 1);

5513

}

5514

5515

static void test_cancel_old_question_2_1_3(__attribute__((unused))

5516

test_fixture *fixture,

5517

__attribute__((unused))

5518

gconstpointer user_data){

5519

/* next_queue_run 2,

5520

cancellation should happen because time has come,

5521

next_queue_run should be unchanged */

5522

/* next_queue_run, notafter, current_time, next_set_to */

5523

assert_cancel_old_question_param(2, 1, 3, 2);

5524

}

5525

5526

static void test_cancel_old_question_2_3_1(__attribute__((unused))

5527

test_fixture *fixture,

5528

__attribute__((unused))

5529

gconstpointer user_data){

5530

/* If next_queue_run is set,

5531

and current_time is not yet notafter or greater,

5532

and notafter is larger than next_queue_run

5533

next_queue_run should be unchanged */

5534

/* next_queue_run, notafter, current_time, next_set_to */

5535

assert_cancel_old_question_param(2, 3, 1, 2);

5536

}

5537

5538

static void test_cancel_old_question_3_1_2(__attribute__((unused))

5539

test_fixture *fixture,

5540

__attribute__((unused))

5541

gconstpointer user_data){

5542

/* next_queue_run 3,

5543

cancellation should happen because time has come,

5544

next_queue_run should be unchanged */

5545

/* next_queue_run, notafter, current_time, next_set_to */

5546

assert_cancel_old_question_param(3, 1, 2, 3);

5547

}

5548

5549

static void test_cancel_old_question_3_2_1(__attribute__((unused))

5550

test_fixture *fixture,

5551

__attribute__((unused))

5552

gconstpointer user_data){

5553

/* If next_queue_run is set,

5554

and current_time is not yet notafter or greater,

5555

and notafter is smaller than next_queue_run

5556

update value of next_queue_run to value of notafter */

5557

/* next_queue_run, notafter, current_time, next_set_to */

5558

assert_cancel_old_question_param(3, 2, 1, 2);

5559

}

5560

5561

static void

5562

test_connect_question_socket_name_too_long(__attribute__((unused))

5563

test_fixture *fixture,

5564

__attribute__((unused))

5565

gconstpointer user_data){

5566

__attribute__((cleanup(cleanup_close)))

5567

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5568

g_assert_cmpint(epoll_fd, >=, 0);

5569

const char question_filename[] = "/nonexistent/question";

5570

__attribute__((cleanup(string_set_clear)))

5571

string_set cancelled_filenames = {};

5572

__attribute__((cleanup(cleanup_queue)))

5573

task_queue *queue = create_queue();

5574

g_assert_nonnull(queue);

5575

__attribute__((cleanup(cleanup_string)))

5576

char *tempdir = make_temporary_directory();

5577

g_assert_nonnull(tempdir);

5578

struct sockaddr_un unix_socket = { .sun_family=AF_LOCAL };

5579

char socket_name[sizeof(unix_socket.sun_path)];

5580

memset(socket_name, 'x', sizeof(socket_name));

5581

socket_name[sizeof(socket_name)-1] = '\0';

5582

char *filename = NULL;

5583

g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),

5584

>, 0);

5585

g_assert_nonnull(filename);

5586

5587

task_context task = {

5588

.func=connect_question_socket,

5589

.question_filename=strdup(question_filename),

5590

.epoll_fd=epoll_fd,

5591

.password=(buffer[]){{}},

5592

.filename=filename,

5593

.cancelled_filenames=&cancelled_filenames,

5594

.mandos_client_exited=(bool[]){false},

5595

.password_is_read=(bool[]){false},

5596

.current_time=(mono_microsecs[]){0},

5597

};

5598

g_assert_nonnull(task.question_filename);

5599

run_task_with_stderr_to_dev_null(task, queue);

5600

5601

g_assert_true(string_set_contains(cancelled_filenames,

5602

question_filename));

5603

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

5604

g_assert_true(queue->next_run == 0);

5605

5606

g_assert_cmpint(rmdir(tempdir), ==, 0);

5607

}

5608

5609

static

5610

void test_connect_question_socket_connect_fail(__attribute__((unused))

5611

test_fixture *fixture,

5612

__attribute__((unused))

5613

gconstpointer

5614

user_data){

5615

__attribute__((cleanup(cleanup_close)))

5616

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5617

g_assert_cmpint(epoll_fd, >=, 0);

5618

const char question_filename[] = "/nonexistent/question";

5619

__attribute__((cleanup(string_set_clear)))

5620

string_set cancelled_filenames = {};

5621

const mono_microsecs current_time = 3;

5622

__attribute__((cleanup(cleanup_queue)))

5623

task_queue *queue = create_queue();

5624

g_assert_nonnull(queue);

5625

__attribute__((cleanup(cleanup_string)))

5626

char *tempdir = make_temporary_directory();

5627

g_assert_nonnull(tempdir);

5628

char socket_name[] = "nonexistent";

5629

char *filename = NULL;

5630

g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),

5631

>, 0);

5632

g_assert_nonnull(filename);

5633

5634

task_context task = {

5635

.func=connect_question_socket,

5636

.question_filename=strdup(question_filename),

5637

.epoll_fd=epoll_fd,

5638

.password=(buffer[]){{}},

5639

.filename=filename,

5640

.cancelled_filenames=&cancelled_filenames,

5641

.mandos_client_exited=(bool[]){false},

5642

.password_is_read=(bool[]){false},

5643

.current_time=&current_time,

5644

};

5645

g_assert_nonnull(task.question_filename);

5646

run_task_with_stderr_to_dev_null(task, queue);

5647

5648

g_assert_nonnull(find_matching_task(queue, task));

5649

5650

g_assert_false(string_set_contains(cancelled_filenames,

5651

question_filename));

5652

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5653

g_assert_true(queue->next_run == 1000000 + current_time);

5654

5655

g_assert_cmpint(rmdir(tempdir), ==, 0);

5656

}

5657

5658

static

5659

void test_connect_question_socket_bad_epoll(__attribute__((unused))

5660

test_fixture *fixture,

5661

__attribute__((unused))

5662

gconstpointer user_data){

5663

__attribute__((cleanup(cleanup_close)))

5664

const int epoll_fd = open("/dev/null",

5665

O_WRONLY | O_CLOEXEC | O_NOCTTY);

5666

__attribute__((cleanup(cleanup_string)))

5667

char *const question_filename = strdup("/nonexistent/question");

5668

g_assert_nonnull(question_filename);

5669

__attribute__((cleanup(string_set_clear)))

5670

string_set cancelled_filenames = {};

5671

const mono_microsecs current_time = 5;

5672

__attribute__((cleanup(cleanup_queue)))

5673

task_queue *queue = create_queue();

5674

g_assert_nonnull(queue);

5675

__attribute__((cleanup(cleanup_string)))

5676

char *tempdir = make_temporary_directory();

5677

g_assert_nonnull(tempdir);

5678

__attribute__((cleanup(cleanup_close)))

5679

const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM

5680

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0);

5681

g_assert_cmpint(sock_fd, >=, 0);

5682

struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };

5683

const char socket_name[] = "socket_name";

5684

__attribute__((cleanup(cleanup_string)))

5685

char *filename = NULL;

5686

g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),

5687

>, 0);

5688

g_assert_nonnull(filename);

5689

g_assert_cmpint((int)strlen(filename), <,

5690

(int)sizeof(sock_name.sun_path));

5691

strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));

5692

sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';

5693

g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,

5694

(socklen_t)SUN_LEN(&sock_name)), >=, 0);

5695

task_context task = {

5696

.func=connect_question_socket,

5697

.question_filename=strdup(question_filename),

5698

.epoll_fd=epoll_fd,

5699

.password=(buffer[]){{}},

5700

.filename=strdup(filename),

5701

.cancelled_filenames=&cancelled_filenames,

5702

.mandos_client_exited=(bool[]){false},

5703

.password_is_read=(bool[]){false},

5704

.current_time=&current_time,

5705

};

5706

g_assert_nonnull(task.question_filename);

5707

run_task_with_stderr_to_dev_null(task, queue);

5708

5709

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5710

const task_context *const added_task

5711

= find_matching_task(queue, task);

5712

g_assert_nonnull(added_task);

5713

g_assert_true(queue->next_run == 1000000 + current_time);

5714

5715

g_assert_cmpint(unlink(filename), ==, 0);

5716

g_assert_cmpint(rmdir(tempdir), ==, 0);

5717

}

5718

5719

static

5720

void test_connect_question_socket_usable(__attribute__((unused))

5721

test_fixture *fixture,

5722

__attribute__((unused))

5723

gconstpointer user_data){

5724

__attribute__((cleanup(cleanup_close)))

5725

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5726

g_assert_cmpint(epoll_fd, >=, 0);

5727

__attribute__((cleanup(cleanup_string)))

5728

char *const question_filename = strdup("/nonexistent/question");

5729

g_assert_nonnull(question_filename);

5730

__attribute__((cleanup(string_set_clear)))

5731

string_set cancelled_filenames = {};

5732

buffer password = {};

5733

bool mandos_client_exited = false;

5734

bool password_is_read = false;

5735

const mono_microsecs current_time = 0;

5736

__attribute__((cleanup(cleanup_queue)))

5737

task_queue *queue = create_queue();

5738

g_assert_nonnull(queue);

5739

__attribute__((cleanup(cleanup_string)))

5740

char *tempdir = make_temporary_directory();

5741

g_assert_nonnull(tempdir);

5742

__attribute__((cleanup(cleanup_close)))

5743

const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM

5744

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0);

5745

g_assert_cmpint(sock_fd, >=, 0);

5746

struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };

5747

const char socket_name[] = "socket_name";

5748

__attribute__((cleanup(cleanup_string)))

5749

char *filename = NULL;

5750

g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),

5751

>, 0);

5752

g_assert_nonnull(filename);

5753

g_assert_cmpint((int)strlen(filename), <,

5754

(int)sizeof(sock_name.sun_path));

5755

strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));

5756

sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';

5757

g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,

5758

(socklen_t)SUN_LEN(&sock_name)), >=, 0);

5759

task_context task = {

5760

.func=connect_question_socket,

5761

.question_filename=strdup(question_filename),

5762

.epoll_fd=epoll_fd,

5763

.password=&password,

5764

.filename=strdup(filename),

5765

.cancelled_filenames=&cancelled_filenames,

5766

.mandos_client_exited=&mandos_client_exited,

5767

.password_is_read=&password_is_read,

5768

.current_time=&current_time,

5769

};

5770

g_assert_nonnull(task.question_filename);

5771

task.func(task, queue);

5772

5773

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5774

const task_context *const added_task

5775

= find_matching_task(queue, (task_context){

5776

.func=send_password_to_socket,

5777

.question_filename=question_filename,

5778

.filename=filename,

5779

.epoll_fd=epoll_fd,

5780

.password=&password,

5781

.cancelled_filenames=&cancelled_filenames,

5782

.mandos_client_exited=&mandos_client_exited,

5783

.password_is_read=&password_is_read,

5784

.current_time=&current_time,

5785

});

5786

g_assert_nonnull(added_task);

5787

g_assert_cmpint(added_task->fd, >, 0);

5788

5789

g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,

5790

EPOLLOUT));

5791

5792

const int fd = added_task->fd;

5793

g_assert_cmpint(fd, >, 0);

5794

g_assert_true(fd_has_cloexec_and_nonblock(fd));

5795

5796

/* write to fd */

5797

char write_data[PIPE_BUF];

5798

{

5799

/* Construct test password buffer */

5800

/* Start with + since that is what the real procotol uses */

5801

write_data[0] = '+';

5802

/* Set a special character at string end just to mark the end */

5803

write_data[sizeof(write_data)-2] = 'y';

5804

/* Set NUL at buffer end, as suggested by the protocol */

5805

write_data[sizeof(write_data)-1] = '\0';

5806

/* Fill rest of password with 'x' */

5807

memset(write_data+1, 'x', sizeof(write_data)-3);

5808

g_assert_cmpint((int)send(fd, write_data, sizeof(write_data),

5809

MSG_NOSIGNAL), ==, sizeof(write_data));

5810

}

5811

5812

/* read from sock_fd */

5813

char read_data[sizeof(write_data)];

5814

g_assert_cmpint((int)read(sock_fd, read_data, sizeof(read_data)),

5815

==, sizeof(read_data));

5816

5817

g_assert_true(memcmp(write_data, read_data, sizeof(write_data))

5818

== 0);

5819

5820

/* writing to sock_fd should fail */

5821

g_assert_cmpint(send(sock_fd, write_data, sizeof(write_data),

5822

MSG_NOSIGNAL), <, 0);

5823

5824

/* reading from fd should fail */

5825

g_assert_cmpint((int)recv(fd, read_data, sizeof(read_data),

5826

MSG_NOSIGNAL), <, 0);

5827

5828

g_assert_cmpint(unlink(filename), ==, 0);

5829

g_assert_cmpint(rmdir(tempdir), ==, 0);

5830

}

5831

5832

static void

5833

test_send_password_to_socket_client_not_exited(__attribute__((unused))

5834

test_fixture *fixture,

5835

__attribute__((unused))

5836

gconstpointer

5837

user_data){

5838

__attribute__((cleanup(cleanup_close)))

5839

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5840

g_assert_cmpint(epoll_fd, >=, 0);

5841

__attribute__((cleanup(cleanup_string)))

5842

char *const question_filename = strdup("/nonexistent/question");

5843

g_assert_nonnull(question_filename);

5844

__attribute__((cleanup(cleanup_string)))

5845

char *const filename = strdup("/nonexistent/socket");

5846

g_assert_nonnull(filename);

5847

__attribute__((cleanup(string_set_clear)))

5848

string_set cancelled_filenames = {};

5849

buffer password = {};

5850

bool password_is_read = true;

5851

__attribute__((cleanup(cleanup_queue)))

5852

task_queue *queue = create_queue();

5853

g_assert_nonnull(queue);

5854

int socketfds[2];

5855

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

5856

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

5857

socketfds), ==, 0);

5858

__attribute__((cleanup(cleanup_close)))

5859

const int read_socket = socketfds[0];

5860

const int write_socket = socketfds[1];

5861

task_context task = {

5862

.func=send_password_to_socket,

5863

.question_filename=strdup(question_filename),

5864

.filename=strdup(filename),

5865

.epoll_fd=epoll_fd,

5866

.fd=write_socket,

5867

.password=&password,

5868

.cancelled_filenames=&cancelled_filenames,

5869

.mandos_client_exited=(bool[]){false},

5870

.password_is_read=&password_is_read,

5871

.current_time=(mono_microsecs[]){0},

5872

};

5873

g_assert_nonnull(task.question_filename);

5874

5875

task.func(task, queue);

5876

5877

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5878

5879

const task_context *const added_task

5880

= find_matching_task(queue, task);

5881

g_assert_nonnull(added_task);

5882

g_assert_cmpuint((unsigned int)password.length, ==, 0);

5883

g_assert_true(password_is_read);

5884

5885

g_assert_cmpint(added_task->fd, >, 0);

5886

g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,

5887

EPOLLOUT));

5888

}

5889

5890

static void

5891

test_send_password_to_socket_password_not_read(__attribute__((unused))

5892

test_fixture *fixture,

5893

__attribute__((unused))

5894

gconstpointer

5895

user_data){

5896

__attribute__((cleanup(cleanup_close)))

5897

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5898

g_assert_cmpint(epoll_fd, >=, 0);

5899

__attribute__((cleanup(cleanup_string)))

5900

char *const question_filename = strdup("/nonexistent/question");

5901

g_assert_nonnull(question_filename);

5902

__attribute__((cleanup(cleanup_string)))

5903

char *const filename = strdup("/nonexistent/socket");

5904

__attribute__((cleanup(string_set_clear)))

5905

string_set cancelled_filenames = {};

5906

buffer password = {};

5907

__attribute__((cleanup(cleanup_queue)))

5908

task_queue *queue = create_queue();

5909

g_assert_nonnull(queue);

5910

int socketfds[2];

5911

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

5912

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

5913

socketfds), ==, 0);

5914

__attribute__((cleanup(cleanup_close)))

5915

const int read_socket = socketfds[0];

5916

const int write_socket = socketfds[1];

5917

task_context task = {

5918

.func=send_password_to_socket,

5919

.question_filename=strdup(question_filename),

5920

.filename=strdup(filename),

5921

.epoll_fd=epoll_fd,

5922

.fd=write_socket,

5923

.password=&password,

5924

.cancelled_filenames=&cancelled_filenames,

5925

.mandos_client_exited=(bool[]){false},

5926

.password_is_read=(bool[]){false},

5927

.current_time=(mono_microsecs[]){0},

5928

};

5929

g_assert_nonnull(task.question_filename);

5930

5931

task.func(task, queue);

5932

5933

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

5934

5935

const task_context *const added_task = find_matching_task(queue,

5936

task);

5937

g_assert_nonnull(added_task);

5938

g_assert_cmpuint((unsigned int)password.length, ==, 0);

5939

g_assert_true(queue->next_run == 0);

5940

5941

g_assert_cmpint(added_task->fd, >, 0);

5942

g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,

5943

EPOLLOUT));

5944

}

5945

5946

static

5947

void test_send_password_to_socket_EMSGSIZE(__attribute__((unused))

5948

test_fixture *fixture,

5949

__attribute__((unused))

5950

gconstpointer user_data){

5951

__attribute__((cleanup(cleanup_close)))

5952

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

5953

g_assert_cmpint(epoll_fd, >=, 0);

5954

const char question_filename[] = "/nonexistent/question";

5955

char *const filename = strdup("/nonexistent/socket");

5956

__attribute__((cleanup(string_set_clear)))

5957

string_set cancelled_filenames = {};

5958

const size_t oversized = 1024*1024; /* Limit seems to be 212960 */

5959

__attribute__((cleanup(cleanup_buffer)))

5960

buffer password = {

5961

.data=malloc(oversized),

5962

.length=oversized,

5963

.allocated=oversized,

5964

};

5965

g_assert_nonnull(password.data);

5966

if(mlock(password.data, password.allocated) != 0){

5967

g_assert_true(errno == EPERM or errno == ENOMEM);

5968

}

5969

/* Construct test password buffer */

5970

/* Start with + since that is what the real procotol uses */

5971

password.data[0] = '+';

5972

/* Set a special character at string end just to mark the end */

5973

password.data[oversized-3] = 'y';

5974

/* Set NUL at buffer end, as suggested by the protocol */

5975

password.data[oversized-2] = '\0';

5976

/* Fill rest of password with 'x' */

5977

memset(password.data+1, 'x', oversized-3);

5978

5979

__attribute__((cleanup(cleanup_queue)))

5980

task_queue *queue = create_queue();

5981

g_assert_nonnull(queue);

5982

int socketfds[2];

5983

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

5984

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

5985

socketfds), ==, 0);

5986

__attribute__((cleanup(cleanup_close)))

5987

const int read_socket = socketfds[0];

5988

__attribute__((cleanup(cleanup_close)))

5989

const int write_socket = socketfds[1];

5990

task_context task = {

5991

.func=send_password_to_socket,

5992

.question_filename=strdup(question_filename),

5993

.filename=filename,

5994

.epoll_fd=epoll_fd,

5995

.fd=write_socket,

5996

.password=&password,

5997

.cancelled_filenames=&cancelled_filenames,

5998

.mandos_client_exited=(bool[]){true},

5999

.password_is_read=(bool[]){true},

6000

.current_time=(mono_microsecs[]){0},

6001

};

6002

g_assert_nonnull(task.question_filename);

6003

6004

run_task_with_stderr_to_dev_null(task, queue);

6005

6006

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

6007

g_assert_true(string_set_contains(cancelled_filenames,

6008

question_filename));

6009

}

6010

6011

static void test_send_password_to_socket_retry(__attribute__((unused))

6012

test_fixture *fixture,

6013

__attribute__((unused))

6014

gconstpointer

6015

user_data){

6016

__attribute__((cleanup(cleanup_close)))

6017

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6018

g_assert_cmpint(epoll_fd, >=, 0);

6019

__attribute__((cleanup(cleanup_string)))

6020

char *const question_filename = strdup("/nonexistent/question");

6021

g_assert_nonnull(question_filename);

6022

__attribute__((cleanup(cleanup_string)))

6023

char *const filename = strdup("/nonexistent/socket");

6024

g_assert_nonnull(filename);

6025

__attribute__((cleanup(string_set_clear)))

6026

string_set cancelled_filenames = {};

6027

__attribute__((cleanup(cleanup_buffer)))

6028

buffer password = {};

6029

6030

__attribute__((cleanup(cleanup_queue)))

6031

task_queue *queue = create_queue();

6032

g_assert_nonnull(queue);

6033

int socketfds[2];

6034

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

6035

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

6036

socketfds), ==, 0);

6037

__attribute__((cleanup(cleanup_close)))

6038

const int read_socket = socketfds[0];

6039

const int write_socket = socketfds[1];

6040

/* Close the server side socket to force ECONNRESET on client */

6041

g_assert_cmpint(close(read_socket), ==, 0);

6042

task_context task = {

6043

.func=send_password_to_socket,

6044

.question_filename=strdup(question_filename),

6045

.filename=strdup(filename),

6046

.epoll_fd=epoll_fd,

6047

.fd=write_socket,

6048

.password=&password,

6049

.cancelled_filenames=&cancelled_filenames,

6050

.mandos_client_exited=(bool[]){true},

6051

.password_is_read=(bool[]){true},

6052

.current_time=(mono_microsecs[]){0},

6053

};

6054

g_assert_nonnull(task.question_filename);

6055

6056

task.func(task, queue);

6057

6058

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

6059

6060

const task_context *const added_task = find_matching_task(queue,

6061

task);

6062

g_assert_nonnull(added_task);

6063

g_assert_cmpuint((unsigned int)password.length, ==, 0);

6064

6065

g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,

6066

EPOLLOUT));

6067

}

6068

6069

static

6070

void test_send_password_to_socket_bad_epoll(__attribute__((unused))

6071

test_fixture *fixture,

6072

__attribute__((unused))

6073

gconstpointer user_data){

6074

__attribute__((cleanup(cleanup_close)))

6075

const int epoll_fd = open("/dev/null",

6076

O_WRONLY | O_CLOEXEC | O_NOCTTY);

6077

__attribute__((cleanup(cleanup_string)))

6078

char *const question_filename = strdup("/nonexistent/question");

6079

g_assert_nonnull(question_filename);

6080

__attribute__((cleanup(cleanup_string)))

6081

char *const filename = strdup("/nonexistent/socket");

6082

g_assert_nonnull(filename);

6083

__attribute__((cleanup(string_set_clear)))

6084

string_set cancelled_filenames = {};

6085

__attribute__((cleanup(cleanup_buffer)))

6086

buffer password = {};

6087

6088

const mono_microsecs current_time = 11;

6089

__attribute__((cleanup(cleanup_queue)))

6090

task_queue *queue = create_queue();

6091

g_assert_nonnull(queue);

6092

int socketfds[2];

6093

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

6094

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

6095

socketfds), ==, 0);

6096

__attribute__((cleanup(cleanup_close)))

6097

const int read_socket = socketfds[0];

6098

const int write_socket = socketfds[1];

6099

/* Close the server side socket to force ECONNRESET on client */

6100

g_assert_cmpint(close(read_socket), ==, 0);

6101

task_context task = {

6102

.func=send_password_to_socket,

6103

.question_filename=strdup(question_filename),

6104

.filename=strdup(filename),

6105

.epoll_fd=epoll_fd,

6106

.fd=write_socket,

6107

.password=&password,

6108

.cancelled_filenames=&cancelled_filenames,

6109

.mandos_client_exited=(bool[]){true},

6110

.password_is_read=(bool[]){true},

6111

.current_time=&current_time,

6112

};

6113

g_assert_nonnull(task.question_filename);

6114

6115

run_task_with_stderr_to_dev_null(task, queue);

6116

6117

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

6118

6119

const task_context *const added_task = find_matching_task(queue,

6120

task);

6121

g_assert_nonnull(added_task);

6122

g_assert_true(queue->next_run == current_time + 1000000);

6123

g_assert_cmpuint((unsigned int)password.length, ==, 0);

6124

}

6125

6126

static void assert_send_password_to_socket_password(buffer password){

6127

__attribute__((cleanup(cleanup_close)))

6128

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6129

g_assert_cmpint(epoll_fd, >=, 0);

6130

char *const question_filename = strdup("/nonexistent/question");

6131

g_assert_nonnull(question_filename);

6132

char *const filename = strdup("/nonexistent/socket");

6133

g_assert_nonnull(filename);

6134

__attribute__((cleanup(string_set_clear)))

6135

string_set cancelled_filenames = {};

6136

6137

__attribute__((cleanup(cleanup_queue)))

6138

task_queue *queue = create_queue();

6139

g_assert_nonnull(queue);

6140

int socketfds[2];

6141

g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM

6142

| SOCK_NONBLOCK | SOCK_CLOEXEC, 0,

6143

socketfds), ==, 0);

6144

__attribute__((cleanup(cleanup_close)))

6145

const int read_socket = socketfds[0];

6146

const int write_socket = socketfds[1];

6147

task_context task = {

6148

.func=send_password_to_socket,

6149

.question_filename=question_filename,

6150

.filename=filename,

6151

.epoll_fd=epoll_fd,

6152

.fd=write_socket,

6153

.password=&password,

6154

.cancelled_filenames=&cancelled_filenames,

6155

.mandos_client_exited=(bool[]){true},

6156

.password_is_read=(bool[]){true},

6157

.current_time=(mono_microsecs[]){0},

6158

};

6159

6160

char *expected_written_data = malloc(password.length + 2);

6161

g_assert_nonnull(expected_written_data);

6162

expected_written_data[0] = '+';

6163

expected_written_data[password.length + 1] = '\0';

6164

if(password.length > 0){

6165

g_assert_nonnull(password.data);

6166

memcpy(expected_written_data + 1, password.data, password.length);

6167

}

6168

6169

task.func(task, queue);

6170

6171

char buf[PIPE_BUF];

6172

g_assert_cmpint((int)read(read_socket, buf, PIPE_BUF), ==,

6173

(int)(password.length + 2));

6174

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

6175

6176

g_assert_true(memcmp(expected_written_data, buf,

6177

password.length + 2) == 0);

6178

6179

g_assert_true(epoll_set_does_not_contain(epoll_fd, write_socket));

6180

6181

free(expected_written_data);

6182

}

6183

6184

static void

6185

test_send_password_to_socket_null_password(__attribute__((unused))

6186

test_fixture *fixture,

6187

__attribute__((unused))

6188

gconstpointer user_data){

6189

__attribute__((cleanup(cleanup_buffer)))

6190

buffer password = {};

6191

assert_send_password_to_socket_password(password);

6192

}

6193

6194

static void

6195

test_send_password_to_socket_empty_password(__attribute__((unused))

6196

test_fixture *fixture,

6197

__attribute__((unused))

6198

gconstpointer user_data){

6199

__attribute__((cleanup(cleanup_buffer)))

6200

buffer password = {

6201

.data=malloc(1), /* because malloc(0) may return NULL */

6202

.length=0,

6203

.allocated=0, /* deliberate lie */

6204

};

6205

g_assert_nonnull(password.data);

6206

assert_send_password_to_socket_password(password);

6207

}

6208

6209

static void

6210

test_send_password_to_socket_empty_str_pass(__attribute__((unused))

6211

test_fixture *fixture,

6212

__attribute__((unused))

6213

gconstpointer user_data){

6214

__attribute__((cleanup(cleanup_buffer)))

6215

buffer password = {

6216

.data=strdup(""),

6217

.length=0,

6218

.allocated=1,

6219

};

6220

if(mlock(password.data, password.allocated) != 0){

6221

g_assert_true(errno == EPERM or errno == ENOMEM);

6222

}

6223

assert_send_password_to_socket_password(password);

6224

}

6225

6226

static void

6227

test_send_password_to_socket_text_password(__attribute__((unused))

6228

test_fixture *fixture,

6229

__attribute__((unused))

6230

gconstpointer user_data){

6231

const char dummy_test_password[] = "dummy test password";

6232

__attribute__((cleanup(cleanup_buffer)))

6233

buffer password = {

6234

.data = strdup(dummy_test_password),

6235

.length = strlen(dummy_test_password),

6236

.allocated = sizeof(dummy_test_password),

6237

};

6238

if(mlock(password.data, password.allocated) != 0){

6239

g_assert_true(errno == EPERM or errno == ENOMEM);

6240

}

6241

assert_send_password_to_socket_password(password);

6242

}

6243

6244

static void

6245

test_send_password_to_socket_binary_password(__attribute__((unused))

6246

test_fixture *fixture,

6247

__attribute__((unused))

6248

gconstpointer user_data){

6249

__attribute__((cleanup(cleanup_buffer)))

6250

buffer password = {

6251

.data=malloc(255),

6252

.length=255,

6253

.allocated=255,

6254

};

6255

g_assert_nonnull(password.data);

6256

if(mlock(password.data, password.allocated) != 0){

6257

g_assert_true(errno == EPERM or errno == ENOMEM);

6258

}

6259

char c = 1; /* Start at 1, avoiding NUL */

6260

for(int i=0; i < 255; i++){

6261

password.data[i] = c++;

6262

}

6263

assert_send_password_to_socket_password(password);

6264

}

6265

6266

static void

6267

test_send_password_to_socket_nuls_in_password(__attribute__((unused))

6268

test_fixture *fixture,

6269

__attribute__((unused))

6270

gconstpointer

6271

user_data){

6272

char test_password[] = {'\0', 'a', '\0', 'b', '\0', 'c', '\0'};

6273

__attribute__((cleanup(cleanup_buffer)))

6274

buffer password = {

6275

.data=malloc(sizeof(test_password)),

6276

.length=sizeof(test_password),

6277

.allocated=sizeof(test_password),

6278

};

6279

g_assert_nonnull(password.data);

6280

if(mlock(password.data, password.allocated) !=0){

6281

g_assert_true(errno == EPERM or errno == ENOMEM);

6282

}

6283

memcpy(password.data, test_password, password.allocated);

6284

assert_send_password_to_socket_password(password);

6285

}

6286

6287

static bool assert_add_existing_questions_to_devnull(task_queue

6288

*const,

6289

const int,

6290

buffer *const,

6291

string_set *,

6292

const

6293

mono_microsecs

6294

*const,

6295

bool *const,

6296

bool *const,

6297

const char

6298

*const);

6299

6300

static void test_add_existing_questions_ENOENT(__attribute__((unused))

6301

test_fixture *fixture,

6302

__attribute__((unused))

6303

gconstpointer

6304

user_data){

6305

__attribute__((cleanup(cleanup_queue)))

6306

task_queue *queue = create_queue();

6307

g_assert_nonnull(queue);

6308

__attribute__((cleanup(cleanup_close)))

6309

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6310

g_assert_cmpint(epoll_fd, >=, 0);

6311

__attribute__((cleanup(string_set_clear)))

6312

string_set cancelled_filenames = {};

6313

6314

g_assert_false(assert_add_existing_questions_to_devnull

6315

(queue,

6316

epoll_fd,

6317

(buffer[]){{}}, /* password */

6318

&cancelled_filenames,

6319

(mono_microsecs[]){0}, /* current_time */

6320

(bool[]){false}, /* mandos_client_exited */

6321

(bool[]){false}, /* password_is_read */

6322

"/nonexistent")); /* dirname */

6323

6324

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

6325

}

6326

6327

static

6328

bool assert_add_existing_questions_to_devnull(task_queue

6329

*const queue,

6330

const int

6331

epoll_fd,

6332

buffer *const

6333

password,

6334

string_set

6335

*cancelled_filenames,

6336

const mono_microsecs

6337

*const current_time,

6338

bool *const

6339

mandos_client_exited,

6340

bool *const

6341

password_is_read,

6342

const char *const

6343

dirname){

6344

__attribute__((cleanup(cleanup_close)))

6345

const int devnull_fd = open("/dev/null",

6346

O_WRONLY | O_CLOEXEC | O_NOCTTY);

6347

g_assert_cmpint(devnull_fd, >=, 0);

6348

__attribute__((cleanup(cleanup_close)))

6349

const int real_stderr_fd = dup(STDERR_FILENO);

6350

g_assert_cmpint(real_stderr_fd, >=, 0);

6351

dup2(devnull_fd, STDERR_FILENO);

6352

const bool ret = add_existing_questions(queue, epoll_fd, password,

6353

cancelled_filenames,

6354

current_time,

6355

mandos_client_exited,

6356

password_is_read, dirname);

6357

dup2(real_stderr_fd, STDERR_FILENO);

6358

return ret;

6359

}

6360

6361

static

6362

void test_add_existing_questions_no_questions(__attribute__((unused))

6363

test_fixture *fixture,

6364

__attribute__((unused))

6365

gconstpointer

6366

user_data){

6367

__attribute__((cleanup(cleanup_queue)))

6368

task_queue *queue = create_queue();

6369

g_assert_nonnull(queue);

6370

__attribute__((cleanup(cleanup_close)))

6371

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6372

g_assert_cmpint(epoll_fd, >=, 0);

6373

__attribute__((cleanup(string_set_clear)))

6374

string_set cancelled_filenames = {};

6375

__attribute__((cleanup(cleanup_string)))

6376

char *tempdir = make_temporary_directory();

6377

g_assert_nonnull(tempdir);

6378

6379

g_assert_false(assert_add_existing_questions_to_devnull

6380

(queue,

6381

epoll_fd,

6382

(buffer[]){{}}, /* password */

6383

&cancelled_filenames,

6384

(mono_microsecs[]){0}, /* current_time */

6385

(bool[]){false}, /* mandos_client_exited */

6386

(bool[]){false}, /* password_is_read */

6387

tempdir));

6388

6389

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

6390

6391

g_assert_cmpint(rmdir(tempdir), ==, 0);

6392

}

6393

6394

static char *make_question_file_in_directory(const char *const);

6395

6396

static

6397

void test_add_existing_questions_one_question(__attribute__((unused))

6398

test_fixture *fixture,

6399

__attribute__((unused))

6400

gconstpointer

6401

user_data){

6402

__attribute__((cleanup(cleanup_queue)))

6403

task_queue *queue = create_queue();

6404

g_assert_nonnull(queue);

6405

__attribute__((cleanup(cleanup_close)))

6406

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6407

g_assert_cmpint(epoll_fd, >=, 0);

6408

__attribute__((cleanup(cleanup_buffer)))

6409

buffer password = {};

6410

__attribute__((cleanup(string_set_clear)))

6411

string_set cancelled_filenames = {};

6412

const mono_microsecs current_time = 0;

6413

bool mandos_client_exited = false;

6414

bool password_is_read = false;

6415

__attribute__((cleanup(cleanup_string)))

6416

char *tempdir = make_temporary_directory();

6417

g_assert_nonnull(tempdir);

6418

__attribute__((cleanup(cleanup_string)))

6419

char *question_filename

6420

= make_question_file_in_directory(tempdir);

6421

g_assert_nonnull(question_filename);

6422

6423

g_assert_true(assert_add_existing_questions_to_devnull

6424

(queue,

6425

epoll_fd,

6426

&password,

6427

&cancelled_filenames,

6428

&current_time,

6429

&mandos_client_exited,

6430

&password_is_read,

6431

tempdir));

6432

6433

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

6434

6435

g_assert_nonnull(find_matching_task(queue, (task_context){

6436

.func=open_and_parse_question,

6437

.epoll_fd=epoll_fd,

6438

.filename=question_filename,

6439

.question_filename=question_filename,

6440

.password=&password,

6441

.cancelled_filenames=&cancelled_filenames,

6442

.current_time=&current_time,

6443

.mandos_client_exited=&mandos_client_exited,

6444

.password_is_read=&password_is_read,

6445

}));

6446

6447

g_assert_true(queue->next_run == 1);

6448

6449

g_assert_cmpint(unlink(question_filename), ==, 0);

6450

g_assert_cmpint(rmdir(tempdir), ==, 0);

6451

}

6452

6453

static char *make_question_file_in_directory(const char

6454

*const dir){

6455

return make_temporary_prefixed_file_in_directory("ask.", dir);

6456

}

6457

6458

static

6459

void test_add_existing_questions_two_questions(__attribute__((unused))

6460

test_fixture *fixture,

6461

__attribute__((unused))

6462

gconstpointer

6463

user_data){

6464

__attribute__((cleanup(cleanup_queue)))

6465

task_queue *queue = create_queue();

6466

g_assert_nonnull(queue);

6467

__attribute__((cleanup(cleanup_close)))

6468

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6469

g_assert_cmpint(epoll_fd, >=, 0);

6470

__attribute__((cleanup(cleanup_buffer)))

6471

buffer password = {};

6472

__attribute__((cleanup(string_set_clear)))

6473

string_set cancelled_filenames = {};

6474

const mono_microsecs current_time = 0;

6475

bool mandos_client_exited = false;

6476

bool password_is_read = false;

6477

__attribute__((cleanup(cleanup_string)))

6478

char *tempdir = make_temporary_directory();

6479

g_assert_nonnull(tempdir);

6480

__attribute__((cleanup(cleanup_string)))

6481

char *question_filename1

6482

= make_question_file_in_directory(tempdir);

6483

g_assert_nonnull(question_filename1);

6484

__attribute__((cleanup(cleanup_string)))

6485

char *question_filename2

6486

= make_question_file_in_directory(tempdir);

6487

g_assert_nonnull(question_filename2);

6488

6489

g_assert_true(assert_add_existing_questions_to_devnull

6490

(queue,

6491

epoll_fd,

6492

&password,

6493

&cancelled_filenames,

6494

&current_time,

6495

&mandos_client_exited,

6496

&password_is_read,

6497

tempdir));

6498

6499

g_assert_cmpuint((unsigned int)queue->length, ==, 2);

6500

6501

g_assert_true(queue->next_run == 1);

6502

6503

__attribute__((cleanup(string_set_clear)))

6504

string_set seen_questions = {};

6505

6506

bool queue_contains_question_opener(char *const question_filename){

6507

return(find_matching_task(queue, (task_context){

6508

.func=open_and_parse_question,

6509

.epoll_fd=epoll_fd,

6510

.question_filename=question_filename,

6511

.password=&password,

6512

.cancelled_filenames=&cancelled_filenames,

6513

.current_time=&current_time,

6514

.mandos_client_exited=&mandos_client_exited,

6515

.password_is_read=&password_is_read,

6516

}) != NULL);

6517

}

6518

6519

g_assert_true(queue_contains_question_opener(question_filename1));

6520

g_assert_true(queue_contains_question_opener(question_filename2));

6521

6522

g_assert_true(queue->next_run == 1);

6523

6524

g_assert_cmpint(unlink(question_filename1), ==, 0);

6525

g_assert_cmpint(unlink(question_filename2), ==, 0);

6526

g_assert_cmpint(rmdir(tempdir), ==, 0);

6527

}

6528

6529

static void

6530

test_add_existing_questions_non_questions(__attribute__((unused))

6531

test_fixture *fixture,

6532

__attribute__((unused))

6533

gconstpointer user_data){

6534

__attribute__((cleanup(cleanup_queue)))

6535

task_queue *queue = create_queue();

6536

g_assert_nonnull(queue);

6537

__attribute__((cleanup(cleanup_close)))

6538

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6539

g_assert_cmpint(epoll_fd, >=, 0);

6540

__attribute__((cleanup(string_set_clear)))

6541

string_set cancelled_filenames = {};

6542

__attribute__((cleanup(cleanup_string)))

6543

char *tempdir = make_temporary_directory();

6544

g_assert_nonnull(tempdir);

6545

__attribute__((cleanup(cleanup_string)))

6546

char *question_filename1

6547

= make_temporary_file_in_directory(tempdir);

6548

g_assert_nonnull(question_filename1);

6549

__attribute__((cleanup(cleanup_string)))

6550

char *question_filename2

6551

= make_temporary_file_in_directory(tempdir);

6552

g_assert_nonnull(question_filename2);

6553

6554

g_assert_false(assert_add_existing_questions_to_devnull

6555

(queue,

6556

epoll_fd,

6557

(buffer[]){{}}, /* password */

6558

&cancelled_filenames,

6559

(mono_microsecs[]){0}, /* current_time */

6560

(bool[]){false}, /* mandos_client_exited */

6561

(bool[]){false}, /* password_is_read */

6562

tempdir));

6563

6564

g_assert_cmpuint((unsigned int)queue->length, ==, 0);

6565

6566

g_assert_cmpint(unlink(question_filename1), ==, 0);

6567

g_assert_cmpint(unlink(question_filename2), ==, 0);

6568

g_assert_cmpint(rmdir(tempdir), ==, 0);

6569

}

6570

6571

static void

6572

test_add_existing_questions_both_types(__attribute__((unused))

6573

test_fixture *fixture,

6574

__attribute__((unused))

6575

gconstpointer user_data){

6576

__attribute__((cleanup(cleanup_queue)))

6577

task_queue *queue = create_queue();

6578

g_assert_nonnull(queue);

6579

__attribute__((cleanup(cleanup_close)))

6580

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6581

g_assert_cmpint(epoll_fd, >=, 0);

6582

__attribute__((cleanup(cleanup_buffer)))

6583

buffer password = {};

6584

__attribute__((cleanup(string_set_clear)))

6585

string_set cancelled_filenames = {};

6586

const mono_microsecs current_time = 0;

6587

bool mandos_client_exited = false;

6588

bool password_is_read = false;

6589

__attribute__((cleanup(cleanup_string)))

6590

char *tempdir = make_temporary_directory();

6591

g_assert_nonnull(tempdir);

6592

__attribute__((cleanup(cleanup_string)))

6593

char *tempfilename1 = make_temporary_file_in_directory(tempdir);

6594

g_assert_nonnull(tempfilename1);

6595

__attribute__((cleanup(cleanup_string)))

6596

char *tempfilename2 = make_temporary_file_in_directory(tempdir);

6597

g_assert_nonnull(tempfilename2);

6598

__attribute__((cleanup(cleanup_string)))

6599

char *question_filename

6600

= make_question_file_in_directory(tempdir);

6601

g_assert_nonnull(question_filename);

6602

6603

g_assert_true(assert_add_existing_questions_to_devnull

6604

(queue,

6605

epoll_fd,

6606

&password,

6607

&cancelled_filenames,

6608

&current_time,

6609

&mandos_client_exited,

6610

&password_is_read,

6611

tempdir));

6612

6613

g_assert_cmpuint((unsigned int)queue->length, ==, 1);

6614

6615

g_assert_nonnull(find_matching_task(queue, (task_context){

6616

.func=open_and_parse_question,

6617

.epoll_fd=epoll_fd,

6618

.filename=question_filename,

6619

.question_filename=question_filename,

6620

.password=&password,

6621

.cancelled_filenames=&cancelled_filenames,

6622

.current_time=&current_time,

6623

.mandos_client_exited=&mandos_client_exited,

6624

.password_is_read=&password_is_read,

6625

}));

6626

6627

g_assert_true(queue->next_run == 1);

6628

6629

g_assert_cmpint(unlink(tempfilename1), ==, 0);

6630

g_assert_cmpint(unlink(tempfilename2), ==, 0);

6631

g_assert_cmpint(unlink(question_filename), ==, 0);

6632

g_assert_cmpint(rmdir(tempdir), ==, 0);

6633

}

6634

6635

static void test_wait_for_event_timeout(__attribute__((unused))

6636

test_fixture *fixture,

6637

__attribute__((unused))

6638

gconstpointer user_data){

6639

__attribute__((cleanup(cleanup_close)))

6640

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6641

g_assert_cmpint(epoll_fd, >=, 0);

6642

6643

g_assert_true(wait_for_event(epoll_fd, 1, 0));

6644

}

6645

6646

static void test_wait_for_event_event(__attribute__((unused))

6647

test_fixture *fixture,

6648

__attribute__((unused))

6649

gconstpointer user_data){

6650

__attribute__((cleanup(cleanup_close)))

6651

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6652

g_assert_cmpint(epoll_fd, >=, 0);

6653

int pipefds[2];

6654

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

6655

__attribute__((cleanup(cleanup_close)))

6656

const int read_pipe = pipefds[0];

6657

__attribute__((cleanup(cleanup_close)))

6658

const int write_pipe = pipefds[1];

6659

g_assert_cmpint(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, read_pipe,

6660

&(struct epoll_event)

6661

{ .events=EPOLLIN | EPOLLRDHUP }), ==, 0);

6662

g_assert_cmpint((int)write(write_pipe, "x", 1), ==, 1);

6663

6664

g_assert_true(wait_for_event(epoll_fd, 0, 0));

6665

}

6666

6667

static void test_wait_for_event_sigchld(test_fixture *fixture,

6668

__attribute__((unused))

6669

gconstpointer user_data){

6670

const pid_t pid = fork();

6671

if(pid == 0){ /* Child */

6672

if(not restore_signal_handler(&fixture->orig_sigaction)){

6673

_exit(EXIT_FAILURE);

6674

}

6675

if(not restore_sigmask(&fixture->orig_sigmask)){

6676

_exit(EXIT_FAILURE);

6677

}

6678

exit(EXIT_SUCCESS);

6679

}

6680

g_assert_true(pid != -1);

6681

__attribute__((cleanup(cleanup_close)))

6682

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6683

g_assert_cmpint(epoll_fd, >=, 0);

6684

6685

g_assert_true(wait_for_event(epoll_fd, 0, 0));

6686

6687

int status;

6688

g_assert_true(waitpid(pid, &status, 0) == pid);

6689

g_assert_true(WIFEXITED(status));

6690

g_assert_cmpint(WEXITSTATUS(status), ==, EXIT_SUCCESS);

6691

}

6692

6693

static void test_run_queue_zeroes_next_run(__attribute__((unused))

6694

test_fixture *fixture,

6695

__attribute__((unused))

6696

gconstpointer user_data){

6697

__attribute__((cleanup(cleanup_queue)))

6698

task_queue *queue = create_queue();

6699

g_assert_nonnull(queue);

6700

queue->next_run = 1;

6701

__attribute__((cleanup(cleanup_close)))

6702

const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);

6703

__attribute__((cleanup(string_set_clear)))

6704

string_set cancelled_filenames = {};

6705

bool quit_now = false;

6706

6707

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

6708

g_assert_false(quit_now);

6709

g_assert_cmpuint((unsigned int)queue->next_run, ==, 0);

6710

}

6711

6712

static

6713

void test_run_queue_clears_cancelled_filenames(__attribute__((unused))

6714

test_fixture *fixture,

6715

__attribute__((unused))

6716

gconstpointer

6717

user_data){

6718

__attribute__((cleanup(cleanup_queue)))

6719

task_queue *queue = create_queue();

6720

g_assert_nonnull(queue);

6721

__attribute__((cleanup(string_set_clear)))

6722

string_set cancelled_filenames = {};

6723

bool quit_now = false;

6724

const char question_filename[] = "/nonexistent/question_filename";

6725

g_assert_true(string_set_add(&cancelled_filenames,

6726

question_filename));

6727

6728

g_assert_true(add_to_queue(queue,

6729

(task_context){ .func=dummy_func }));

6730

6731

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

6732

g_assert_false(quit_now);

6733

g_assert_cmpuint((unsigned int)(queue->length), ==, 0);

6734

g_assert_false(string_set_contains(cancelled_filenames,

6735

question_filename));

6736

}

6737

6738

static

6739

void test_run_queue_skips_cancelled_filenames(__attribute__((unused))

6740

test_fixture *fixture,

6741

__attribute__((unused))

6742

gconstpointer

6743

user_data){

6744

__attribute__((cleanup(cleanup_queue)))

6745

task_queue *queue = create_queue();

6746

g_assert_nonnull(queue);

6747

__attribute__((cleanup(string_set_clear)))

6748

string_set cancelled_filenames = {};

6749

bool quit_now = false;

6750

int pipefds[2];

6751

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

6752

__attribute__((cleanup(cleanup_close)))

6753

const int read_pipe = pipefds[0];

6754

g_assert_cmpint(close(pipefds[1]), ==, 0);

6755

const char question_filename[] = "/nonexistent/question_filename";

6756

g_assert_true(string_set_add(&cancelled_filenames,

6757

question_filename));

6758

__attribute__((nonnull))

6759

void quit_func(const task_context task,

6760

__attribute__((unused)) task_queue *const q){

6761

g_assert_nonnull(task.quit_now);

6762

*task.quit_now = true;

6763

}

6764

task_context task = {

6765

.func=quit_func,

6766

.question_filename=strdup(question_filename),

6767

.quit_now=&quit_now,

6768

.fd=read_pipe,

6769

};

6770

g_assert_nonnull(task.question_filename);

6771

6772

g_assert_true(add_to_queue(queue, task));

6773

6774

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

6775

g_assert_false(quit_now);

6776

6777

/* read_pipe should be closed already */

6778

errno = 0;

6779

bool read_pipe_closed = (close(read_pipe) == -1);

6780

read_pipe_closed &= (errno == EBADF);

6781

g_assert_true(read_pipe_closed);

6782

}

6783

6784

static void test_run_queue_one_task(__attribute__((unused))

6785

test_fixture *fixture,

6786

__attribute__((unused))

6787

gconstpointer user_data){

6788

__attribute__((cleanup(cleanup_queue)))

6789

task_queue *queue = create_queue();

6790

g_assert_nonnull(queue);

6791

__attribute__((cleanup(string_set_clear)))

6792

string_set cancelled_filenames = {};

6793

bool quit_now = false;

6794

6795

__attribute__((nonnull))

6796

void next_run_func(__attribute__((unused))

6797

const task_context task,

6798

task_queue *const q){

6799

q->next_run = 1;

6800

}

6801

6802

task_context task = {

6803

.func=next_run_func,

6804

};

6805

g_assert_true(add_to_queue(queue, task));

6806

6807

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

6808

g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);

6809

g_assert_cmpuint((unsigned int)(queue->length), ==, 0);

6810

}

6811

6812

static void test_run_queue_two_tasks(__attribute__((unused))

6813

test_fixture *fixture,

6814

__attribute__((unused))

6815

gconstpointer user_data){

6816

__attribute__((cleanup(cleanup_queue)))

6817

task_queue *queue = create_queue();

6818

g_assert_nonnull(queue);

6819

queue->next_run = 1;

6820

__attribute__((cleanup(string_set_clear)))

6821

string_set cancelled_filenames = {};

6822

bool quit_now = false;

6823

bool mandos_client_exited = false;

6824

6825

__attribute__((nonnull))

6826

void next_run_func(__attribute__((unused))

6827

const task_context task,

6828

task_queue *const q){

6829

q->next_run = 1;

6830

}

6831

6832

__attribute__((nonnull))

6833

void exited_func(const task_context task,

6834

__attribute__((unused)) task_queue *const q){

6835

*task.mandos_client_exited = true;

6836

}

6837

6838

task_context task1 = {

6839

.func=next_run_func,

6840

};

6841

g_assert_true(add_to_queue(queue, task1));

6842

6843

task_context task2 = {

6844

.func=exited_func,

6845

.mandos_client_exited=&mandos_client_exited,

6846

};

6847

g_assert_true(add_to_queue(queue, task2));

6848

6849

g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));

6850

g_assert_false(quit_now);

6851

g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);

6852

g_assert_true(mandos_client_exited);

6853

g_assert_cmpuint((unsigned int)(queue->length), ==, 0);

6854

}

6855

6856

static void test_run_queue_two_tasks_quit(__attribute__((unused))

6857

test_fixture *fixture,

6858

__attribute__((unused))

6859

gconstpointer user_data){

6860

__attribute__((cleanup(cleanup_queue)))

6861

task_queue *queue = create_queue();

6862

g_assert_nonnull(queue);

6863

__attribute__((cleanup(string_set_clear)))

6864

string_set cancelled_filenames = {};

6865

bool quit_now = false;

6866

bool mandos_client_exited = false;

6867

bool password_is_read = false;

6868

6869

__attribute__((nonnull))

6870

void set_exited_func(const task_context task,

6871

__attribute__((unused)) task_queue *const q){

6872

*task.mandos_client_exited = true;

6873

*task.quit_now = true;

6874

}

6875

task_context task1 = {

6876

.func=set_exited_func,

6877

.quit_now=&quit_now,

6878

.mandos_client_exited=&mandos_client_exited,

6879

};

6880

g_assert_true(add_to_queue(queue, task1));

6881

6882

__attribute__((nonnull))

6883

void set_read_func(const task_context task,

6884

__attribute__((unused)) task_queue *const q){

6885

*task.quit_now = true;

6886

*task.password_is_read = true;

6887

}

6888

task_context task2 = {

6889

.func=set_read_func,

6890

.quit_now=&quit_now,

6891

.password_is_read=&password_is_read,

6892

};

6893

g_assert_true(add_to_queue(queue, task2));

6894

6895

g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));

6896

g_assert_true(quit_now);

6897

g_assert_true(mandos_client_exited xor password_is_read);

6898

g_assert_cmpuint((unsigned int)(queue->length), ==, 0);

6899

}

6900

6901

static void test_run_queue_two_tasks_cleanup(__attribute__((unused))

6902

test_fixture *fixture,

6903

__attribute__((unused))

6904

gconstpointer user_data){

6905

__attribute__((cleanup(cleanup_queue)))

6906

task_queue *queue = create_queue();

6907

g_assert_nonnull(queue);

6908

__attribute__((cleanup(string_set_clear)))

6909

string_set cancelled_filenames = {};

6910

int pipefds[2];

6911

g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);

6912

__attribute__((cleanup(cleanup_close)))

6913

const int read_pipe = pipefds[0];

6914

__attribute__((cleanup(cleanup_close)))

6915

const int write_pipe = pipefds[1];

6916

bool quit_now = false;

6917

6918

__attribute__((nonnull))

6919

void read_func(const task_context task,

6920

__attribute__((unused)) task_queue *const q){

6921

*task.quit_now = true;

6922

}

6923

task_context task1 = {

6924

.func=read_func,

6925

.quit_now=&quit_now,

6926

.fd=read_pipe,

6927

};

6928

g_assert_true(add_to_queue(queue, task1));

6929

6930

__attribute__((nonnull))

6931

void write_func(const task_context task,

6932

__attribute__((unused)) task_queue *const q){

6933

*task.quit_now = true;

6934

}

6935

task_context task2 = {

6936

.func=write_func,

6937

.quit_now=&quit_now,

6938

.fd=write_pipe,

6939

};

6940

g_assert_true(add_to_queue(queue, task2));

6941

6942

g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));

6943

g_assert_true(quit_now);

6944

6945

/* Either read_pipe or write_pipe should be closed already */

6946

errno = 0;

6947

bool close_read_pipe = (close(read_pipe) == -1);

6948

close_read_pipe &= (errno == EBADF);

6949

errno = 0;

6950

bool close_write_pipe = (close(write_pipe) == -1);

6951

close_write_pipe &= (errno == EBADF);

6952

g_assert_true(close_read_pipe xor close_write_pipe);

6953

g_assert_cmpuint((unsigned int)(queue->length), ==, 0);

6954

}

6955

6956

static void test_setup_signal_handler(__attribute__((unused))

6957

test_fixture *fixture,

6958

__attribute__((unused))

6959

gconstpointer user_data){

6960

/* Save current SIGCHLD action, whatever it is */

6961

struct sigaction expected_sigchld_action;

6962

g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),

6963

==, 0);

6964

6965

/* Act; i.e. run the setup_signal_handler() function */

6966

struct sigaction actual_old_sigchld_action;

6967

g_assert_true(setup_signal_handler(&actual_old_sigchld_action));

6968

6969

/* Check that the function correctly set "actual_old_sigchld_action"

6970

to the same values as the previously saved

6971

"expected_sigchld_action" */

6972

/* Check member sa_handler */

6973

g_assert_true(actual_old_sigchld_action.sa_handler

6974

== expected_sigchld_action.sa_handler);

6975

/* Check member sa_mask */

6976

for(int signum = 1; signum < NSIG; signum++){

6977

const int expected_old_block_state

6978

= sigismember(&expected_sigchld_action.sa_mask, signum);

6979

g_assert_cmpint(expected_old_block_state, >=, 0);

6980

const int actual_old_block_state

6981

= sigismember(&actual_old_sigchld_action.sa_mask, signum);

6982

g_assert_cmpint(actual_old_block_state, >=, 0);

6983

g_assert_cmpint(actual_old_block_state,

6984

==, expected_old_block_state);

6985

}

6986

/* Check member sa_flags */

6987

g_assert_true((actual_old_sigchld_action.sa_flags

6988

& (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))

6989

== (expected_sigchld_action.sa_flags

6990

& (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));

6991

6992

/* Retrieve the current signal handler for SIGCHLD as set by

6993

setup_signal_handler() */

6994

struct sigaction actual_new_sigchld_action;

6995

g_assert_cmpint(sigaction(SIGCHLD, NULL,

6996

&actual_new_sigchld_action), ==, 0);

6997

/* Check that the signal handler (member sa_handler) is correctly

6998

set to the "handle_sigchld" function */

6999

g_assert_true(actual_new_sigchld_action.sa_handler != SIG_DFL);

7000

g_assert_true(actual_new_sigchld_action.sa_handler != SIG_IGN);

7001

g_assert_true(actual_new_sigchld_action.sa_handler

7002

== handle_sigchld);

7003

/* Check (in member sa_mask) that at least a handful of signals are

7004

actually blocked during the signal handler */

7005

for(int signum = 1; signum < NSIG; signum++){

7006

int actual_new_block_state;

7007

switch(signum){

7008

case SIGTERM:

7009

case SIGINT:

7010

case SIGQUIT:

7011

case SIGHUP:

7012

actual_new_block_state

7013

= sigismember(&actual_new_sigchld_action.sa_mask, signum);

7014

g_assert_cmpint(actual_new_block_state, ==, 1);

7015

continue;

7016

case SIGKILL: /* non-blockable */

7017

case SIGSTOP: /* non-blockable */

7018

case SIGCHLD: /* always blocked */

7019

default:

7020

continue;

7021

}

7022

}

7023

/* Check member sa_flags */

7024

g_assert_true((actual_new_sigchld_action.sa_flags

7025

& (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))

7026

== (SA_NOCLDSTOP | SA_RESTART));

7027

7028

/* Restore signal handler */

7029

g_assert_cmpint(sigaction(SIGCHLD, &expected_sigchld_action, NULL),

7030

==, 0);

7031

}

7032

7033

static void test_restore_signal_handler(__attribute__((unused))

7034

test_fixture *fixture,

7035

__attribute__((unused))

7036

gconstpointer user_data){

7037

/* Save current SIGCHLD action, whatever it is */

7038

struct sigaction expected_sigchld_action;

7039

g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),

7040

==, 0);

7041

/* Since we haven't established a signal handler yet, there should

7042

not be one established. But another test may have relied on

7043

restore_signal_handler() to restore the signal handler, and if

7044

restore_signal_handler() is buggy (which we should be prepared

7045

for in this test) the signal handler may not have been restored

7046

properly; check for this: */

7047

g_assert_true(expected_sigchld_action.sa_handler != handle_sigchld);

7048

7049

/* Establish a signal handler */

7050

struct sigaction sigchld_action = {

7051

.sa_handler=handle_sigchld,

7052

.sa_flags=SA_RESTART | SA_NOCLDSTOP,

7053

};

7054

g_assert_cmpint(sigfillset(&sigchld_action.sa_mask), ==, 0);

7055

g_assert_cmpint(sigaction(SIGCHLD, &sigchld_action, NULL), ==, 0);

7056

7057

/* Act; i.e. run the restore_signal_handler() function */

7058

g_assert_true(restore_signal_handler(&expected_sigchld_action));

7059

7060

/* Retrieve the restored signal handler data */

7061

struct sigaction actual_restored_sigchld_action;

7062

g_assert_cmpint(sigaction(SIGCHLD, NULL,

7063

&actual_restored_sigchld_action), ==, 0);

7064

7065

/* Check that the function correctly restored the signal action, as

7066

saved in "actual_restored_sigchld_action", to the same values as

7067

the previously saved "expected_sigchld_action" */

7068

/* Check member sa_handler */

7069

g_assert_true(actual_restored_sigchld_action.sa_handler

7070

== expected_sigchld_action.sa_handler);

7071

/* Check member sa_mask */

7072

for(int signum = 1; signum < NSIG; signum++){

7073

const int expected_old_block_state

7074

= sigismember(&expected_sigchld_action.sa_mask, signum);

7075

g_assert_cmpint(expected_old_block_state, >=, 0);

7076

const int actual_restored_block_state

7077

= sigismember(&actual_restored_sigchld_action.sa_mask, signum);

7078

g_assert_cmpint(actual_restored_block_state, >=, 0);

7079

g_assert_cmpint(actual_restored_block_state,

7080

==, expected_old_block_state);

7081

}

7082

/* Check member sa_flags */

7083

g_assert_true((actual_restored_sigchld_action.sa_flags

7084

& (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))

7085

== (expected_sigchld_action.sa_flags

7086

& (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));

7087

}

7088

7089

static void test_block_sigchld(__attribute__((unused))

7090

test_fixture *fixture,

7091

__attribute__((unused))

7092

gconstpointer user_data){

7093

/* Save original signal mask */

7094

sigset_t expected_sigmask;

7095

g_assert_cmpint(pthread_sigmask(-1, NULL, &expected_sigmask),

7096

==, 0);

7097

7098

/* Make sure SIGCHLD is unblocked for this test */

7099

sigset_t sigchld_sigmask;

7100

g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);

7101

g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);

7102

g_assert_cmpint(pthread_sigmask(SIG_UNBLOCK, &sigchld_sigmask,

7103

NULL), ==, 0);

7104

7105

/* Act; i.e. run the block_sigchld() function */

7106

sigset_t actual_old_sigmask;

7107

g_assert_true(block_sigchld(&actual_old_sigmask));

7108

7109

/* Check the actual_old_sigmask; it should be the same as the

7110

previously saved signal mask "expected_sigmask". */

7111

for(int signum = 1; signum < NSIG; signum++){

7112

const int expected_old_block_state

7113

= sigismember(&expected_sigmask, signum);

7114

g_assert_cmpint(expected_old_block_state, >=, 0);

7115

const int actual_old_block_state

7116

= sigismember(&actual_old_sigmask, signum);

7117

g_assert_cmpint(actual_old_block_state, >=, 0);

7118

g_assert_cmpint(actual_old_block_state,

7119

==, expected_old_block_state);

7120

}

7121

7122

/* Retrieve the newly set signal mask */

7123

sigset_t actual_sigmask;

7124

g_assert_cmpint(pthread_sigmask(-1, NULL, &actual_sigmask), ==, 0);

7125

7126

/* SIGCHLD should be blocked */

7127

g_assert_cmpint(sigismember(&actual_sigmask, SIGCHLD), ==, 1);

7128

7129

/* Restore signal mask */

7130

g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &expected_sigmask,

7131

NULL), ==, 0);

7132

}

7133

7134

static void test_restore_sigmask(__attribute__((unused))

7135

test_fixture *fixture,

7136

__attribute__((unused))

7137

gconstpointer user_data){

7138

/* Save original signal mask */

7139

sigset_t orig_sigmask;

7140

g_assert_cmpint(pthread_sigmask(-1, NULL, &orig_sigmask), ==, 0);

7141

7142

/* Make sure SIGCHLD is blocked for this test */

7143

sigset_t sigchld_sigmask;

7144

g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);

7145

g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);

7146

g_assert_cmpint(pthread_sigmask(SIG_BLOCK, &sigchld_sigmask,

7147

NULL), ==, 0);

7148

7149

/* Act; i.e. run the restore_sigmask() function */

7150

g_assert_true(restore_sigmask(&orig_sigmask));

7151

7152

/* Retrieve the newly restored signal mask */

7153

sigset_t restored_sigmask;

7154

g_assert_cmpint(pthread_sigmask(-1, NULL, &restored_sigmask),

7155

==, 0);

7156

7157

/* Check the restored_sigmask; it should be the same as the

7158

previously saved signal mask "orig_sigmask". */

7159

for(int signum = 1; signum < NSIG; signum++){

7160

const int orig_block_state = sigismember(&orig_sigmask, signum);

7161

g_assert_cmpint(orig_block_state, >=, 0);

7162

const int restored_block_state = sigismember(&restored_sigmask,

7163

signum);

7164

g_assert_cmpint(restored_block_state, >=, 0);

7165

g_assert_cmpint(restored_block_state, ==, orig_block_state);

7166

}

7167

7168

/* Restore signal mask */

7169

g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &orig_sigmask,

7170

NULL), ==, 0);

7171

}

7172

7173

static void test_parse_arguments_noargs(__attribute__((unused))

7174

test_fixture *fixture,

7175

__attribute__((unused))

7176

gconstpointer user_data){

7177

char *argv[] = {

7178

strdup("prgname"),

7179

NULL };

7180

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7181

7182

char *agent_directory = NULL;

7183

char *helper_directory = NULL;

7184

uid_t user = 0;

7185

gid_t group = 0;

7186

char *mandos_argz = NULL;

7187

size_t mandos_argz_length = 0;

7188

7189

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7190

&helper_directory, &user, &group,

7191

&mandos_argz, &mandos_argz_length));

7192

g_assert_null(agent_directory);

7193

g_assert_null(helper_directory);

7194

g_assert_true(user == 0);

7195

g_assert_true(group == 0);

7196

g_assert_null(mandos_argz);

7197

g_assert_true(mandos_argz_length == 0);

7198

7199

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

7200

free(*arg);

7201

}

7202

}

7203

7204

__attribute__((nonnull))

7205

static bool parse_arguments_devnull(int argc, char *argv[],

7206

const bool exit_failure,

7207

char **agent_directory,

7208

char **helper_directory,

7209

uid_t *const user,

7210

gid_t *const group,

7211

char **mandos_argz,

7212

size_t *mandos_argz_length){

7213

7214

FILE *real_stderr = stderr;

7215

FILE *devnull = fopen("/dev/null", "we");

7216

g_assert_nonnull(devnull);

7217

stderr = devnull;

7218

7219

const bool ret = parse_arguments(argc, argv, exit_failure,

7220

agent_directory,

7221

helper_directory, user, group,

7222

mandos_argz, mandos_argz_length);

7223

const error_t saved_errno = errno;

7224

7225

stderr = real_stderr;

7226

g_assert_cmpint(fclose(devnull), ==, 0);

7227

7228

errno = saved_errno;

7229

7230

return ret;

7231

}

7232

7233

static void test_parse_arguments_invalid(__attribute__((unused))

7234

test_fixture *fixture,

7235

__attribute__((unused))

7236

gconstpointer user_data){

7237

char *argv[] = {

7238

strdup("prgname"),

7239

strdup("--invalid"),

7240

NULL };

7241

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7242

7243

char *agent_directory = NULL;

7244

char *helper_directory = NULL;

7245

uid_t user = 0;

7246

gid_t group = 0;

7247

char *mandos_argz = NULL;

7248

size_t mandos_argz_length = 0;

7249

7250

g_assert_false(parse_arguments_devnull(argc, argv, false,

7251

&agent_directory,

7252

&helper_directory, &user,

7253

&group, &mandos_argz,

7254

&mandos_argz_length));

7255

7256

g_assert_true(errno == EINVAL);

7257

g_assert_null(agent_directory);

7258

g_assert_null(helper_directory);

7259

g_assert_null(mandos_argz);

7260

g_assert_true(mandos_argz_length == 0);

7261

7262

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

7263

free(*arg);

7264

}

7265

}

7266

7267

static void test_parse_arguments_long_dir(__attribute__((unused))

7268

test_fixture *fixture,

7269

__attribute__((unused))

7270

gconstpointer user_data){

7271

char *argv[] = {

7272

strdup("prgname"),

7273

strdup("--agent-directory"),

7274

strdup("/tmp"),

7275

NULL };

7276

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7277

7278

__attribute__((cleanup(cleanup_string)))

7279

char *agent_directory = NULL;

7280

char *helper_directory = NULL;

7281

uid_t user = 0;

7282

gid_t group = 0;

7283

__attribute__((cleanup(cleanup_string)))

7284

char *mandos_argz = NULL;

7285

size_t mandos_argz_length = 0;

7286

7287

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7288

&helper_directory, &user, &group,

7289

&mandos_argz, &mandos_argz_length));

7290

7291

g_assert_cmpstr(agent_directory, ==, "/tmp");

7292

g_assert_null(helper_directory);

7293

g_assert_true(user == 0);

7294

g_assert_true(group == 0);

7295

g_assert_null(mandos_argz);

7296

g_assert_true(mandos_argz_length == 0);

7297

7298

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

7299

free(*arg);

7300

}

7301

}

7302

7303

static void test_parse_arguments_short_dir(__attribute__((unused))

7304

test_fixture *fixture,

7305

__attribute__((unused))

7306

gconstpointer user_data){

7307

char *argv[] = {

7308

strdup("prgname"),

7309

strdup("-d"),

7310

strdup("/tmp"),

7311

NULL };

7312

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7313

7314

__attribute__((cleanup(cleanup_string)))

7315

char *agent_directory = NULL;

7316

char *helper_directory = NULL;

7317

uid_t user = 0;

7318

gid_t group = 0;

7319

__attribute__((cleanup(cleanup_string)))

7320

char *mandos_argz = NULL;

7321

size_t mandos_argz_length = 0;

7322

7323

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7324

&helper_directory, &user, &group,

7325

&mandos_argz, &mandos_argz_length));

7326

7327

g_assert_cmpstr(agent_directory, ==, "/tmp");

7328

g_assert_null(helper_directory);

7329

g_assert_true(user == 0);

7330

g_assert_true(group == 0);

7331

g_assert_null(mandos_argz);

7332

g_assert_true(mandos_argz_length == 0);

7333

7334

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

7335

free(*arg);

7336

}

7337

}

7338

7339

static

7340

void test_parse_arguments_helper_directory(__attribute__((unused))

7341

test_fixture *fixture,

7342

__attribute__((unused))

7343

gconstpointer user_data){

7344

char *argv[] = {

7345

strdup("prgname"),

7346

strdup("--helper-directory"),

7347

strdup("/tmp"),

7348

NULL };

7349

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7350

7351

char *agent_directory = NULL;

7352

__attribute__((cleanup(cleanup_string)))

7353

char *helper_directory = NULL;

7354

uid_t user = 0;

7355

gid_t group = 0;

7356

__attribute__((cleanup(cleanup_string)))

7357

char *mandos_argz = NULL;

7358

size_t mandos_argz_length = 0;

7359

7360

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7361

&helper_directory, &user, &group,

7362

&mandos_argz, &mandos_argz_length));

7363

7364

g_assert_cmpstr(helper_directory, ==, "/tmp");

7365

g_assert_null(agent_directory);

7366

g_assert_true(user == 0);

7367

g_assert_true(group == 0);

7368

g_assert_null(mandos_argz);

7369

g_assert_true(mandos_argz_length == 0);

7370

7371

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

7372

free(*arg);

7373

}

7374

}

7375

7376

static

7377

void test_parse_arguments_plugin_helper_dir(__attribute__((unused))

7378

test_fixture *fixture,

7379

__attribute__((unused))

7380

gconstpointer user_data){

7381

char *argv[] = {

7382

strdup("prgname"),

7383

strdup("--plugin-helper-dir"),

7384

strdup("/tmp"),

7385

NULL };

7386

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7387

7388

char *agent_directory = NULL;

7389

__attribute__((cleanup(cleanup_string)))

7390

char *helper_directory = NULL;

7391

uid_t user = 0;

7392

gid_t group = 0;

7393

__attribute__((cleanup(cleanup_string)))

7394

char *mandos_argz = NULL;

7395

size_t mandos_argz_length = 0;

7396

7397

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7398

&helper_directory, &user, &group,

7399

&mandos_argz, &mandos_argz_length));

7400

7401

g_assert_cmpstr(helper_directory, ==, "/tmp");

7402

g_assert_null(agent_directory);

7403

g_assert_true(user == 0);

7404

g_assert_true(group == 0);

7405

g_assert_null(mandos_argz);

7406

g_assert_true(mandos_argz_length == 0);

7407

7408

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

7409

free(*arg);

7410

}

7411

}

7412

7413

static void test_parse_arguments_user(__attribute__((unused))

7414

test_fixture *fixture,

7415

__attribute__((unused))

7416

gconstpointer user_data){

7417

char *argv[] = {

7418

strdup("prgname"),

7419

strdup("--user"),

7420

strdup("1000"),

7421

NULL };

7422

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7423

7424

char *agent_directory = NULL;

7425

__attribute__((cleanup(cleanup_string)))

7426

char *helper_directory = NULL;

7427

uid_t user = 0;

7428

gid_t group = 0;

7429

__attribute__((cleanup(cleanup_string)))

7430

char *mandos_argz = NULL;

7431

size_t mandos_argz_length = 0;

7432

7433

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7434

&helper_directory, &user, &group,

7435

&mandos_argz, &mandos_argz_length));

7436

7437

g_assert_null(helper_directory);

7438

g_assert_null(agent_directory);

7439

g_assert_cmpuint((unsigned int)user, ==, 1000);

7440

g_assert_true(group == 0);

7441

g_assert_null(mandos_argz);

7442

g_assert_true(mandos_argz_length == 0);

7443

7444

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

7445

free(*arg);

7446

}

7447

}

7448

7449

static void test_parse_arguments_user_invalid(__attribute__((unused))

7450

test_fixture *fixture,

7451

__attribute__((unused))

7452

gconstpointer

7453

user_data){

7454

char *argv[] = {

7455

strdup("prgname"),

7456

strdup("--user"),

7457

strdup("invalid"),

7458

NULL };

7459

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7460

7461

char *agent_directory = NULL;

7462

__attribute__((cleanup(cleanup_string)))

7463

char *helper_directory = NULL;

7464

uid_t user = 0;

7465

gid_t group = 0;

7466

__attribute__((cleanup(cleanup_string)))

7467

char *mandos_argz = NULL;

7468

size_t mandos_argz_length = 0;

7469

7470

g_assert_false(parse_arguments_devnull(argc, argv, false,

7471

&agent_directory,

7472

&helper_directory, &user,

7473

&group, &mandos_argz,

7474

&mandos_argz_length));

7475

7476

g_assert_null(helper_directory);

7477

g_assert_null(agent_directory);

7478

g_assert_cmpuint((unsigned int)user, ==, 0);

7479

g_assert_true(group == 0);

7480

g_assert_null(mandos_argz);

7481

g_assert_true(mandos_argz_length == 0);

7482

7483

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

7484

free(*arg);

7485

}

7486

}

7487

7488

static

7489

void test_parse_arguments_user_zero_invalid(__attribute__((unused))

7490

test_fixture *fixture,

7491

__attribute__((unused))

7492

gconstpointer user_data){

7493

char *argv[] = {

7494

strdup("prgname"),

7495

strdup("--user"),

7496

strdup("0"),

7497

NULL };

7498

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7499

7500

char *agent_directory = NULL;

7501

__attribute__((cleanup(cleanup_string)))

7502

char *helper_directory = NULL;

7503

uid_t user = 0;

7504

gid_t group = 0;

7505

__attribute__((cleanup(cleanup_string)))

7506

char *mandos_argz = NULL;

7507

size_t mandos_argz_length = 0;

7508

7509

g_assert_false(parse_arguments_devnull(argc, argv, false,

7510

&agent_directory,

7511

&helper_directory, &user,

7512

&group, &mandos_argz,

7513

&mandos_argz_length));

7514

7515

g_assert_null(helper_directory);

7516

g_assert_null(agent_directory);

7517

g_assert_cmpuint((unsigned int)user, ==, 0);

7518

g_assert_true(group == 0);

7519

g_assert_null(mandos_argz);

7520

g_assert_true(mandos_argz_length == 0);

7521

7522

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

7523

free(*arg);

7524

}

7525

}

7526

7527

static void test_parse_arguments_group(__attribute__((unused))

7528

test_fixture *fixture,

7529

__attribute__((unused))

7530

gconstpointer user_data){

7531

char *argv[] = {

7532

strdup("prgname"),

7533

strdup("--group"),

7534

strdup("1000"),

7535

NULL };

7536

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7537

7538

char *agent_directory = NULL;

7539

__attribute__((cleanup(cleanup_string)))

7540

char *helper_directory = NULL;

7541

uid_t user = 0;

7542

gid_t group = 0;

7543

__attribute__((cleanup(cleanup_string)))

7544

char *mandos_argz = NULL;

7545

size_t mandos_argz_length = 0;

7546

7547

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7548

&helper_directory, &user, &group,

7549

&mandos_argz, &mandos_argz_length));

7550

7551

g_assert_null(helper_directory);

7552

g_assert_null(agent_directory);

7553

g_assert_true(user == 0);

7554

g_assert_cmpuint((unsigned int)group, ==, 1000);

7555

g_assert_null(mandos_argz);

7556

g_assert_true(mandos_argz_length == 0);

7557

7558

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

7559

free(*arg);

7560

}

7561

}

7562

7563

static void test_parse_arguments_group_invalid(__attribute__((unused))

7564

test_fixture *fixture,

7565

__attribute__((unused))

7566

gconstpointer

7567

user_data){

7568

char *argv[] = {

7569

strdup("prgname"),

7570

strdup("--group"),

7571

strdup("invalid"),

7572

NULL };

7573

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7574

7575

char *agent_directory = NULL;

7576

__attribute__((cleanup(cleanup_string)))

7577

char *helper_directory = NULL;

7578

uid_t user = 0;

7579

gid_t group = 0;

7580

__attribute__((cleanup(cleanup_string)))

7581

char *mandos_argz = NULL;

7582

size_t mandos_argz_length = 0;

7583

7584

g_assert_false(parse_arguments_devnull(argc, argv, false,

7585

&agent_directory,

7586

&helper_directory, &user,

7587

&group, &mandos_argz,

7588

&mandos_argz_length));

7589

7590

g_assert_null(helper_directory);

7591

g_assert_null(agent_directory);

7592

g_assert_true(user == 0);

7593

g_assert_true(group == 0);

7594

g_assert_null(mandos_argz);

7595

g_assert_true(mandos_argz_length == 0);

7596

7597

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

7598

free(*arg);

7599

}

7600

}

7601

7602

static

7603

void test_parse_arguments_group_zero_invalid(__attribute__((unused))

7604

test_fixture *fixture,

7605

__attribute__((unused))

7606

gconstpointer user_data){

7607

char *argv[] = {

7608

strdup("prgname"),

7609

strdup("--group"),

7610

strdup("0"),

7611

NULL };

7612

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7613

7614

char *agent_directory = NULL;

7615

__attribute__((cleanup(cleanup_string)))

7616

char *helper_directory = NULL;

7617

uid_t user = 0;

7618

gid_t group = 0;

7619

__attribute__((cleanup(cleanup_string)))

7620

char *mandos_argz = NULL;

7621

size_t mandos_argz_length = 0;

7622

7623

g_assert_false(parse_arguments_devnull(argc, argv, false,

7624

&agent_directory,

7625

&helper_directory, &user,

7626

&group, &mandos_argz,

7627

&mandos_argz_length));

7628

7629

g_assert_null(helper_directory);

7630

g_assert_null(agent_directory);

7631

g_assert_cmpuint((unsigned int)group, ==, 0);

7632

g_assert_true(group == 0);

7633

g_assert_null(mandos_argz);

7634

g_assert_true(mandos_argz_length == 0);

7635

7636

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

7637

free(*arg);

7638

}

7639

}

7640

7641

static void test_parse_arguments_mandos_noargs(__attribute__((unused))

7642

test_fixture *fixture,

7643

__attribute__((unused))

7644

gconstpointer

7645

user_data){

7646

char *argv[] = {

7647

strdup("prgname"),

7648

strdup("mandos-client"),

7649

NULL };

7650

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7651

7652

__attribute__((cleanup(cleanup_string)))

7653

char *agent_directory = NULL;

7654

__attribute__((cleanup(cleanup_string)))

7655

char *helper_directory = NULL;

7656

uid_t user = 0;

7657

gid_t group = 0;

7658

__attribute__((cleanup(cleanup_string)))

7659

char *mandos_argz = NULL;

7660

size_t mandos_argz_length = 0;

7661

7662

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7663

&helper_directory, &user, &group,

7664

&mandos_argz, &mandos_argz_length));

7665

7666

g_assert_null(agent_directory);

7667

g_assert_null(helper_directory);

7668

g_assert_true(user == 0);

7669

g_assert_true(group == 0);

7670

g_assert_cmpstr(mandos_argz, ==, "mandos-client");

7671

g_assert_cmpuint((unsigned int)argz_count(mandos_argz,

7672

mandos_argz_length),

7673

==, 1);

7674

7675

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

7676

free(*arg);

7677

}

7678

}

7679

7680

static void test_parse_arguments_mandos_args(__attribute__((unused))

7681

test_fixture *fixture,

7682

__attribute__((unused))

7683

gconstpointer user_data){

7684

char *argv[] = {

7685

strdup("prgname"),

7686

strdup("mandos-client"),

7687

strdup("one"),

7688

strdup("two"),

7689

strdup("three"),

7690

NULL };

7691

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7692

7693

__attribute__((cleanup(cleanup_string)))

7694

char *agent_directory = NULL;

7695

__attribute__((cleanup(cleanup_string)))

7696

char *helper_directory = NULL;

7697

uid_t user = 0;

7698

gid_t group = 0;

7699

__attribute__((cleanup(cleanup_string)))

7700

char *mandos_argz = NULL;

7701

size_t mandos_argz_length = 0;

7702

7703

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7704

&helper_directory, &user, &group,

7705

&mandos_argz, &mandos_argz_length));

7706

7707

g_assert_null(agent_directory);

7708

g_assert_null(helper_directory);

7709

g_assert_true(user == 0);

7710

g_assert_true(group == 0);

7711

char *marg = mandos_argz;

7712

g_assert_cmpstr(marg, ==, "mandos-client");

7713

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7714

g_assert_cmpstr(marg, ==, "one");

7715

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7716

g_assert_cmpstr(marg, ==, "two");

7717

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7718

g_assert_cmpstr(marg, ==, "three");

7719

g_assert_cmpuint((unsigned int)argz_count(mandos_argz,

7720

mandos_argz_length),

7721

==, 4);

7722

7723

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

7724

free(*arg);

7725

}

7726

}

7727

7728

static void test_parse_arguments_all_args(__attribute__((unused))

7729

test_fixture *fixture,

7730

__attribute__((unused))

7731

gconstpointer user_data){

7732

char *argv[] = {

7733

strdup("prgname"),

7734

strdup("--agent-directory"),

7735

strdup("/tmp"),

7736

strdup("--helper-directory"),

7737

strdup("/var/tmp"),

7738

strdup("--user"),

7739

strdup("1"),

7740

strdup("--group"),

7741

strdup("2"),

7742

strdup("mandos-client"),

7743

strdup("one"),

7744

strdup("two"),

7745

strdup("three"),

7746

NULL };

7747

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7748

7749

__attribute__((cleanup(cleanup_string)))

7750

char *agent_directory = NULL;

7751

__attribute__((cleanup(cleanup_string)))

7752

char *helper_directory = NULL;

7753

uid_t user = 0;

7754

gid_t group = 0;

7755

__attribute__((cleanup(cleanup_string)))

7756

char *mandos_argz = NULL;

7757

size_t mandos_argz_length = 0;

7758

7759

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7760

&helper_directory, &user, &group,

7761

&mandos_argz, &mandos_argz_length));

7762

7763

g_assert_cmpstr(agent_directory, ==, "/tmp");

7764

g_assert_cmpstr(helper_directory, ==, "/var/tmp");

7765

g_assert_true(user == 1);

7766

g_assert_true(group == 2);

7767

char *marg = mandos_argz;

7768

g_assert_cmpstr(marg, ==, "mandos-client");

7769

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7770

g_assert_cmpstr(marg, ==, "one");

7771

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7772

g_assert_cmpstr(marg, ==, "two");

7773

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7774

g_assert_cmpstr(marg, ==, "three");

7775

g_assert_cmpuint((unsigned int)argz_count(mandos_argz,

7776

mandos_argz_length),

7777

==, 4);

7778

7779

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

7780

free(*arg);

7781

}

7782

}

7783

7784

static void test_parse_arguments_mixed(__attribute__((unused))

7785

test_fixture *fixture,

7786

__attribute__((unused))

7787

gconstpointer user_data){

7788

char *argv[] = {

7789

strdup("prgname"),

7790

strdup("mandos-client"),

7791

strdup("--user"),

7792

strdup("1"),

7793

strdup("one"),

7794

strdup("--agent-directory"),

7795

strdup("/tmp"),

7796

strdup("two"),

7797

strdup("three"),

7798

strdup("--helper-directory=/var/tmp"),

7799

NULL };

7800

const int argc = (sizeof(argv) / sizeof(char *)) - 1;

7801

7802

__attribute__((cleanup(cleanup_string)))

7803

char *agent_directory = NULL;

7804

__attribute__((cleanup(cleanup_string)))

7805

char *helper_directory = NULL;

7806

uid_t user = 0;

7807

gid_t group = 0;

7808

__attribute__((cleanup(cleanup_string)))

7809

char *mandos_argz = NULL;

7810

size_t mandos_argz_length = 0;

7811

7812

g_assert_true(parse_arguments(argc, argv, false, &agent_directory,

7813

&helper_directory, &user, &group,

7814

&mandos_argz, &mandos_argz_length));

7815

7816

g_assert_cmpstr(agent_directory, ==, "/tmp");

7817

g_assert_cmpstr(helper_directory, ==, "/var/tmp");

7818

g_assert_true(user == 1);

7819

g_assert_true(group == 0);

7820

char *marg = mandos_argz;

7821

g_assert_cmpstr(marg, ==, "mandos-client");

7822

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7823

g_assert_cmpstr(marg, ==, "one");

7824

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7825

g_assert_cmpstr(marg, ==, "two");

7826

marg = argz_next(mandos_argz, mandos_argz_length, marg);

7827

g_assert_cmpstr(marg, ==, "three");

7828

g_assert_cmpuint((unsigned int)argz_count(mandos_argz,

7829

mandos_argz_length),

7830

==, 4);

7831

7832

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

7833

free(*arg);

7834

}

7835

}

7836

7837

/* End of tests section */

7838

7839

/* Test boilerplate section; New tests should be added to the test

7840

suite definition here, in the "run_tests" function.

7841

7842

Finally, this section also contains the should_only_run_tests()

7843

function used by main() for deciding if tests should be run or to

7844

start normally. */

7845

7846

__attribute__((cold))

7847

static bool run_tests(int argc, char *argv[]){

7848

g_test_init(&argc, &argv, NULL);

7849

7850

/* A macro to add a test with no setup or teardown functions */

7851

#define test_add(testpath, testfunc) \

7852

do { \

7853

g_test_add((testpath), test_fixture, NULL, NULL, \

7854

(testfunc), NULL); \

7855

} while(false)

7856

7857

/* Test the signal-related functions first, since some other tests

7858

depend on these functions in their setups and teardowns */

7859

test_add("/signal-handling/setup", test_setup_signal_handler);

7860

test_add("/signal-handling/restore", test_restore_signal_handler);

7861

test_add("/signal-handling/block", test_block_sigchld);

7862

test_add("/signal-handling/restore-sigmask", test_restore_sigmask);

7863

7864

/* Regular non-signal-related tests; these use no setups or

7865

teardowns */

7866

test_add("/parse_arguments/noargs", test_parse_arguments_noargs);

7867

test_add("/parse_arguments/invalid", test_parse_arguments_invalid);

7868

test_add("/parse_arguments/long-dir",

7869

test_parse_arguments_long_dir);

7870

test_add("/parse_arguments/short-dir",

7871

test_parse_arguments_short_dir);

7872

test_add("/parse_arguments/helper-directory",

7873

test_parse_arguments_helper_directory);

7874

test_add("/parse_arguments/plugin-helper-dir",

7875

test_parse_arguments_plugin_helper_dir);

7876

test_add("/parse_arguments/user", test_parse_arguments_user);

7877

test_add("/parse_arguments/user-invalid",

7878

test_parse_arguments_user_invalid);

7879

test_add("/parse_arguments/user-zero-invalid",

7880

test_parse_arguments_user_zero_invalid);

7881

test_add("/parse_arguments/group", test_parse_arguments_group);

7882

test_add("/parse_arguments/group-invalid",

7883

test_parse_arguments_group_invalid);

7884

test_add("/parse_arguments/group-zero-invalid",

7885

test_parse_arguments_group_zero_invalid);

7886

test_add("/parse_arguments/mandos-noargs",

7887

test_parse_arguments_mandos_noargs);

7888

test_add("/parse_arguments/mandos-args",

7889

test_parse_arguments_mandos_args);

7890

test_add("/parse_arguments/all-args",

7891

test_parse_arguments_all_args);

7892

test_add("/parse_arguments/mixed", test_parse_arguments_mixed);

7893

test_add("/queue/create", test_create_queue);

7894

test_add("/queue/add", test_add_to_queue);

7895

test_add("/queue/add/overflow", test_add_to_queue_overflow);

7896

test_add("/queue/has_question/empty",

7897

test_queue_has_question_empty);

7898

test_add("/queue/has_question/false",

7899

test_queue_has_question_false);

7900

test_add("/queue/has_question/true", test_queue_has_question_true);

7901

test_add("/queue/has_question/false2",

7902

test_queue_has_question_false2);

7903

test_add("/queue/has_question/true2",

7904

test_queue_has_question_true2);

7905

test_add("/buffer/cleanup", test_cleanup_buffer);

7906

test_add("/string_set/net-set-contains-nothing",

7907

test_string_set_new_set_contains_nothing);

7908

test_add("/string_set/with-added-string-contains-it",

7909

test_string_set_with_added_string_contains_it);

7910

test_add("/string_set/cleared-does-not-contain-string",

7911

test_string_set_cleared_does_not_contain_str);

7912

test_add("/string_set/swap/one-with-empty",

7913

test_string_set_swap_one_with_empty);

7914

test_add("/string_set/swap/empty-with-one",

7915

test_string_set_swap_empty_with_one);

7916

test_add("/string_set/swap/one-with-one",

7917

test_string_set_swap_one_with_one);

7918

7919

/* A macro to add a test using the setup and teardown functions */

7920

#define test_add_st(path, func) \

7921

do { \

7922

g_test_add((path), test_fixture, NULL, test_setup, (func), \

7923

test_teardown); \

7924

} while(false)

7925

7926

/* Signal-related tests; these use setups and teardowns which

7927

establish, during each test run, a signal handler for, and a

7928

signal mask blocking, the SIGCHLD signal, just like main() */

7929

test_add_st("/wait_for_event/timeout", test_wait_for_event_timeout);

7930

test_add_st("/wait_for_event/event", test_wait_for_event_event);

7931

test_add_st("/wait_for_event/sigchld", test_wait_for_event_sigchld);

7932

test_add_st("/run_queue/zeroes-next-run",

7933

test_run_queue_zeroes_next_run);

7934

test_add_st("/run_queue/clears-cancelled_filenames",

7935

test_run_queue_clears_cancelled_filenames);

7936

test_add_st("/run_queue/skips-cancelled-filenames",

7937

test_run_queue_skips_cancelled_filenames);

7938

test_add_st("/run_queue/one-task", test_run_queue_one_task);

7939

test_add_st("/run_queue/two-tasks", test_run_queue_two_tasks);

7940

test_add_st("/run_queue/two-tasks/quit",

7941

test_run_queue_two_tasks_quit);

7942

test_add_st("/run_queue/two-tasks-cleanup",

7943

test_run_queue_two_tasks_cleanup);

7944

test_add_st("/task-creators/start_mandos_client",

7945

test_start_mandos_client);

7946

test_add_st("/task-creators/start_mandos_client/execv",

7947

test_start_mandos_client_execv);

7948

test_add_st("/task-creators/start_mandos_client/suid/euid",

7949

test_start_mandos_client_suid_euid);

7950

test_add_st("/task-creators/start_mandos_client/suid/egid",

7951

test_start_mandos_client_suid_egid);

7952

test_add_st("/task-creators/start_mandos_client/suid/ruid",

7953

test_start_mandos_client_suid_ruid);

7954

test_add_st("/task-creators/start_mandos_client/suid/rgid",

7955

test_start_mandos_client_suid_rgid);

7956

test_add_st("/task-creators/start_mandos_client/read",

7957

test_start_mandos_client_read);

7958

test_add_st("/task-creators/start_mandos_client/helper-directory",

7959

test_start_mandos_client_helper_directory);

7960

test_add_st("/task-creators/start_mandos_client/sigmask",

7961

test_start_mandos_client_sigmask);

7962

test_add_st("/task/wait_for_mandos_client_exit/badpid",

7963

test_wait_for_mandos_client_exit_badpid);

7964

test_add_st("/task/wait_for_mandos_client_exit/noexit",

7965

test_wait_for_mandos_client_exit_noexit);

7966

test_add_st("/task/wait_for_mandos_client_exit/success",

7967

test_wait_for_mandos_client_exit_success);

7968

test_add_st("/task/wait_for_mandos_client_exit/failure",

7969

test_wait_for_mandos_client_exit_failure);

7970

test_add_st("/task/wait_for_mandos_client_exit/killed",

7971

test_wait_for_mandos_client_exit_killed);

7972

test_add_st("/task/read_mandos_client_output/readerror",

7973

test_read_mandos_client_output_readerror);

7974

test_add_st("/task/read_mandos_client_output/nodata",

7975

test_read_mandos_client_output_nodata);

7976

test_add_st("/task/read_mandos_client_output/eof",

7977

test_read_mandos_client_output_eof);

7978

test_add_st("/task/read_mandos_client_output/once",

7979

test_read_mandos_client_output_once);

7980

test_add_st("/task/read_mandos_client_output/malloc",

7981

test_read_mandos_client_output_malloc);

7982

test_add_st("/task/read_mandos_client_output/append",

7983

test_read_mandos_client_output_append);

7984

test_add_st("/task-creators/add_inotify_dir_watch",

7985

test_add_inotify_dir_watch);

7986

test_add_st("/task-creators/add_inotify_dir_watch/fail",

7987

test_add_inotify_dir_watch_fail);

7988

test_add_st("/task-creators/add_inotify_dir_watch/not-a-directory",

7989

test_add_inotify_dir_watch_nondir);

7990

test_add_st("/task-creators/add_inotify_dir_watch/EAGAIN",

7991

test_add_inotify_dir_watch_EAGAIN);

7992

test_add_st("/task-creators/add_inotify_dir_watch/IN_CLOSE_WRITE",

7993

test_add_inotify_dir_watch_IN_CLOSE_WRITE);

7994

test_add_st("/task-creators/add_inotify_dir_watch/IN_MOVED_TO",

7995

test_add_inotify_dir_watch_IN_MOVED_TO);

7996

test_add_st("/task-creators/add_inotify_dir_watch/IN_MOVED_FROM",

7997

test_add_inotify_dir_watch_IN_MOVED_FROM);

7998

test_add_st("/task-creators/add_inotify_dir_watch/IN_EXCL_UNLINK",

7999

test_add_inotify_dir_watch_IN_EXCL_UNLINK);

8000

test_add_st("/task-creators/add_inotify_dir_watch/IN_DELETE",

8001

test_add_inotify_dir_watch_IN_DELETE);

8002

test_add_st("/task/read_inotify_event/readerror",

8003

test_read_inotify_event_readerror);

8004

test_add_st("/task/read_inotify_event/bad-epoll",

8005

test_read_inotify_event_bad_epoll);

8006

test_add_st("/task/read_inotify_event/nodata",

8007

test_read_inotify_event_nodata);

8008

test_add_st("/task/read_inotify_event/eof",

8009

test_read_inotify_event_eof);

8010

test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE",

8011

test_read_inotify_event_IN_CLOSE_WRITE);

8012

test_add_st("/task/read_inotify_event/IN_MOVED_TO",

8013

test_read_inotify_event_IN_MOVED_TO);

8014

test_add_st("/task/read_inotify_event/IN_MOVED_FROM",

8015

test_read_inotify_event_IN_MOVED_FROM);

8016

test_add_st("/task/read_inotify_event/IN_DELETE",

8017

test_read_inotify_event_IN_DELETE);

8018

test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE/badname",

8019

test_read_inotify_event_IN_CLOSE_WRITE_badname);

8020

test_add_st("/task/read_inotify_event/IN_MOVED_TO/badname",

8021

test_read_inotify_event_IN_MOVED_TO_badname);

8022

test_add_st("/task/read_inotify_event/IN_MOVED_FROM/badname",

8023

test_read_inotify_event_IN_MOVED_FROM_badname);

8024

test_add_st("/task/read_inotify_event/IN_DELETE/badname",

8025

test_read_inotify_event_IN_DELETE_badname);

8026

test_add_st("/task/open_and_parse_question/ENOENT",

8027

test_open_and_parse_question_ENOENT);

8028

test_add_st("/task/open_and_parse_question/EIO",

8029

test_open_and_parse_question_EIO);

8030

test_add_st("/task/open_and_parse_question/parse-error",

8031

test_open_and_parse_question_parse_error);

8032

test_add_st("/task/open_and_parse_question/nosocket",

8033

test_open_and_parse_question_nosocket);

8034

test_add_st("/task/open_and_parse_question/badsocket",

8035

test_open_and_parse_question_badsocket);

8036

test_add_st("/task/open_and_parse_question/nopid",

8037

test_open_and_parse_question_nopid);

8038

test_add_st("/task/open_and_parse_question/badpid",

8039

test_open_and_parse_question_badpid);

8040

test_add_st("/task/open_and_parse_question/noexist_pid",

8041

test_open_and_parse_question_noexist_pid);

8042

test_add_st("/task/open_and_parse_question/no-notafter",

8043

test_open_and_parse_question_no_notafter);

8044

test_add_st("/task/open_and_parse_question/bad-notafter",

8045

test_open_and_parse_question_bad_notafter);

8046

test_add_st("/task/open_and_parse_question/notafter-0",

8047

test_open_and_parse_question_notafter_0);

8048

test_add_st("/task/open_and_parse_question/notafter-1",

8049

test_open_and_parse_question_notafter_1);

8050

test_add_st("/task/open_and_parse_question/notafter-1-1",

8051

test_open_and_parse_question_notafter_1_1);

8052

test_add_st("/task/open_and_parse_question/notafter-1-2",

8053

test_open_and_parse_question_notafter_1_2);

8054

test_add_st("/task/open_and_parse_question/equal-notafter",

8055

test_open_and_parse_question_equal_notafter);

8056

test_add_st("/task/open_and_parse_question/late-notafter",

8057

test_open_and_parse_question_late_notafter);

8058

test_add_st("/task/cancel_old_question/0-1-2",

8059

test_cancel_old_question_0_1_2);

8060

test_add_st("/task/cancel_old_question/0-2-1",

8061

test_cancel_old_question_0_2_1);

8062

test_add_st("/task/cancel_old_question/1-2-3",

8063

test_cancel_old_question_1_2_3);

8064

test_add_st("/task/cancel_old_question/1-3-2",

8065

test_cancel_old_question_1_3_2);

8066

test_add_st("/task/cancel_old_question/2-1-3",

8067

test_cancel_old_question_2_1_3);

8068

test_add_st("/task/cancel_old_question/2-3-1",

8069

test_cancel_old_question_2_3_1);

8070

test_add_st("/task/cancel_old_question/3-1-2",

8071

test_cancel_old_question_3_1_2);

8072

test_add_st("/task/cancel_old_question/3-2-1",

8073

test_cancel_old_question_3_2_1);

8074

test_add_st("/task/connect_question_socket/name-too-long",

8075

test_connect_question_socket_name_too_long);

8076

test_add_st("/task/connect_question_socket/connect-fail",

8077

test_connect_question_socket_connect_fail);

8078

test_add_st("/task/connect_question_socket/bad-epoll",

8079

test_connect_question_socket_bad_epoll);

8080

test_add_st("/task/connect_question_socket/usable",

8081

test_connect_question_socket_usable);

8082

test_add_st("/task/send_password_to_socket/client-not-exited",

8083

test_send_password_to_socket_client_not_exited);

8084

test_add_st("/task/send_password_to_socket/password-not-read",

8085

test_send_password_to_socket_password_not_read);

8086

test_add_st("/task/send_password_to_socket/EMSGSIZE",

8087

test_send_password_to_socket_EMSGSIZE);

8088

test_add_st("/task/send_password_to_socket/retry",

8089

test_send_password_to_socket_retry);

8090

test_add_st("/task/send_password_to_socket/bad-epoll",

8091

test_send_password_to_socket_bad_epoll);

8092

test_add_st("/task/send_password_to_socket/null-password",

8093

test_send_password_to_socket_null_password);

8094

test_add_st("/task/send_password_to_socket/empty-password",

8095

test_send_password_to_socket_empty_password);

8096

test_add_st("/task/send_password_to_socket/empty-str-password",

8097

test_send_password_to_socket_empty_str_pass);

8098

test_add_st("/task/send_password_to_socket/text-password",

8099

test_send_password_to_socket_text_password);

8100

test_add_st("/task/send_password_to_socket/binary-password",

8101

test_send_password_to_socket_binary_password);

8102

test_add_st("/task/send_password_to_socket/nuls-in-password",

8103

test_send_password_to_socket_nuls_in_password);

8104

test_add_st("/task-creators/add_existing_questions/ENOENT",

8105

test_add_existing_questions_ENOENT);

8106

test_add_st("/task-creators/add_existing_questions/no-questions",

8107

test_add_existing_questions_no_questions);

8108

test_add_st("/task-creators/add_existing_questions/one-question",

8109

test_add_existing_questions_one_question);

8110

test_add_st("/task-creators/add_existing_questions/two-questions",

8111

test_add_existing_questions_two_questions);

8112

test_add_st("/task-creators/add_existing_questions/non-questions",

8113

test_add_existing_questions_non_questions);

8114

test_add_st("/task-creators/add_existing_questions/both-types",

8115

test_add_existing_questions_both_types);

8116

8117

return g_test_run() == 0;

8118

}

8119

8120

static bool should_only_run_tests(int *argc_p, char **argv_p[]){

8121

GOptionContext *context = g_option_context_new("");

8122

8123

g_option_context_set_help_enabled(context, FALSE);

8124

g_option_context_set_ignore_unknown_options(context, TRUE);

8125

8126

gboolean should_run_tests = FALSE;

8127

GOptionEntry entries[] = {

8128

{ "test", 0, 0, G_OPTION_ARG_NONE,

8129

&should_run_tests, "Run tests", NULL },

8130

{ NULL }

8131

};

8132

g_option_context_add_main_entries(context, entries, NULL);

8133

8134

GError *error = NULL;

8135

8136

if(g_option_context_parse(context, argc_p, argv_p, &error) != TRUE){

8137

g_option_context_free(context);

8138

g_error("Failed to parse options: %s", error->message);

8139

}

8140

8141

g_option_context_free(context);

8142

return should_run_tests != FALSE;

8143

}