/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to dracut-module/password-agent.c

  • Committer: Teddy Hogeborn
  • Date: 2017-02-21 22:15:43 UTC
  • Revision ID: teddy@recompile.se-20170221221543-kkh15uag0luiyz4y
Remove useless code (and use "read -r") in initramfs-tools-hook

* initramfs-tools-hook: When reading module names from network hooks
  given the "modules" argument, use "read -r" and ignore ${target}
  variable (left over from code above it).

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
 
 * Copyright © 2019 Teddy Hogeborn
6
 
 * Copyright © 2019 Björn Påhlsson
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, EEXIST,
52
 
                                   ECHILD, EPERM, ENOMEM, EAGAIN,
53
 
                                   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 <unistd.h>             /* uid_t, gid_t, close(), pipe2(),
77
 
                                   fork(), _exit(), dup2(),
78
 
                                   STDOUT_FILENO, setresgid(),
79
 
                                   setresuid(), execv(), ssize_t,
80
 
                                   read(), dup3(), getuid(), dup(),
81
 
                                   STDERR_FILENO, pause(), write(),
82
 
                                   rmdir(), unlink(), getpid() */
83
 
#include <sys/mman.h>           /* munlock(), mlock() */
84
 
#include <fcntl.h>              /* O_CLOEXEC, O_NONBLOCK, fcntl(),
85
 
                                   F_GETFD, F_GETFL, FD_CLOEXEC,
86
 
                                   open(), O_WRONLY, O_RDONLY */
87
 
#include <sys/wait.h>           /* waitpid(), WNOHANG, WIFEXITED(),
88
 
                                   WEXITSTATUS() */
89
 
#include <limits.h>             /* PIPE_BUF, NAME_MAX, INT_MAX */
90
 
#include <sys/inotify.h>        /* inotify_init1(), IN_NONBLOCK,
91
 
                                   IN_CLOEXEC, inotify_add_watch(),
92
 
                                   IN_CLOSE_WRITE, IN_MOVED_TO,
93
 
                                   IN_DELETE, struct inotify_event */
94
 
#include <fnmatch.h>            /* fnmatch(), FNM_FILE_NAME */
95
 
#include <stdio.h>              /* asprintf(), FILE, fopen(),
96
 
                                   getline(), sscanf(), feof(),
97
 
                                   ferror(), fclose(), stderr,
98
 
                                   rename(), fdopen(), fprintf(),
99
 
                                   fscanf() */
100
 
#include <glib.h>    /* GKeyFile, g_key_file_free(), g_key_file_new(),
101
 
                        GError, g_key_file_load_from_file(),
102
 
                        G_KEY_FILE_NONE, TRUE, G_FILE_ERROR_NOENT,
103
 
                        g_key_file_get_string(), guint64,
104
 
                        g_key_file_get_uint64(),
105
 
                        G_KEY_FILE_ERROR_KEY_NOT_FOUND, gconstpointer,
106
 
                        g_assert_true(), g_assert_nonnull(),
107
 
                        g_assert_null(), g_assert_false(),
108
 
                        g_assert_cmpint(), g_assert_cmpuint(),
109
 
                        g_test_skip(), g_assert_cmpstr(),
110
 
                        g_test_init(), g_test_add(), g_test_run(),
111
 
                        GOptionContext, g_option_context_new(),
112
 
                        g_option_context_set_help_enabled(), FALSE,
113
 
                        g_option_context_set_ignore_unknown_options(),
114
 
                        gboolean, GOptionEntry, G_OPTION_ARG_NONE,
115
 
                        g_option_context_add_main_entries(),
116
 
                        g_option_context_parse(),
117
 
                        g_option_context_free(), g_error() */
118
 
#include <sys/un.h>             /* struct sockaddr_un, SUN_LEN */
119
 
#include <sys/socket.h>         /* AF_LOCAL, socket(), PF_LOCAL,
120
 
                                   SOCK_DGRAM, SOCK_NONBLOCK,
121
 
                                   SOCK_CLOEXEC, connect(),
122
 
                                   struct sockaddr, socklen_t,
123
 
                                   shutdown(), SHUT_RD, send(),
124
 
                                   MSG_NOSIGNAL, bind(), recv(),
125
 
                                   socketpair() */
126
 
#include <glob.h>               /* globfree(), glob_t, glob(),
127
 
                                   GLOB_ERR, GLOB_NOSORT, GLOB_MARK,
128
 
                                   GLOB_ABORTED, GLOB_NOMATCH,
129
 
                                   GLOB_NOSPACE */
130
 
 
131
 
/* End of includes */
132
 
 
133
 
/* Start of declarations of private types and functions */
134
 
 
135
 
/* microseconds of CLOCK_MONOTONIC absolute time; 0 means unset */
136
 
typedef uintmax_t mono_microsecs;
137
 
 
138
 
/* "task_queue" - A queue of tasks to be run */
139
 
typedef struct {
140
 
  struct task_struct *tasks;    /* Tasks in this queue */
141
 
  size_t length;                /* Number of tasks */
142
 
  /* Memory allocated for "tasks", in bytes */
143
 
  size_t allocated;             
144
 
  /* Time when this queue should be run, at the latest */
145
 
  mono_microsecs next_run;
146
 
} __attribute__((designated_init)) task_queue;
147
 
 
148
 
/* "func_type" - A function type for task functions
149
 
 
150
 
   I.e. functions for the code which runs when a task is run, all have
151
 
   this type */
152
 
typedef void (task_func) (const struct task_struct,
153
 
                          task_queue *const)
154
 
  __attribute__((nonnull));
155
 
 
156
 
/* "buffer" - A data buffer for a growing array of bytes
157
 
 
158
 
   Used for the "password" variable */
159
 
typedef struct {
160
 
  char *data;
161
 
  size_t length;
162
 
  size_t allocated;
163
 
} __attribute__((designated_init)) buffer;
164
 
 
165
 
/* "string_set" - A set type which can contain strings
166
 
 
167
 
   Used by the "cancelled_filenames" variable */
168
 
typedef struct {
169
 
  char *argz;                   /* Do not access these except in */
170
 
  size_t argz_len;              /* the string_set_* functions */
171
 
} __attribute__((designated_init)) string_set;
172
 
 
173
 
/* "task_context" - local variables for tasks
174
 
 
175
 
   This data structure distinguishes between different tasks which are
176
 
   using the same function.  This data structure is passed to every
177
 
   task function when each task is run.
178
 
 
179
 
   Note that not every task uses every struct member. */
180
 
typedef struct task_struct {
181
 
  task_func *const func;         /* The function run by this task */
182
 
  char *const question_filename; /* The question file */
183
 
  const pid_t pid;               /* Mandos client process ID */
184
 
  const int epoll_fd;            /* The epoll set file descriptor */
185
 
  bool *const quit_now;          /* Set to true on fatal errors */
186
 
  const int fd;                  /* General purpose file descriptor */
187
 
  bool *const mandos_client_exited; /* Set true when client exits */
188
 
  buffer *const password;           /* As read from client process */
189
 
  bool *const password_is_read;     /* "password" is done growing */
190
 
  char *filename;                   /* General purpose file name */
191
 
  /* A set of strings of all the file names of questions which have
192
 
     been cancelled for any reason; tasks pertaining to these question
193
 
     files should not be run */
194
 
  string_set *const cancelled_filenames;
195
 
  const mono_microsecs notafter; /* "NotAfter" from question file */
196
 
  /* Updated before each queue run; is compared with queue.next_run */
197
 
  const mono_microsecs *const current_time;
198
 
} __attribute__((designated_init)) task_context;
199
 
 
200
 
/* Declare all our functions here so we can define them in any order
201
 
   below.  Note: test functions are *not* declared here, they are
202
 
   declared in the test section. */
203
 
__attribute__((warn_unused_result))
204
 
static bool should_only_run_tests(int *, char **[]);
205
 
__attribute__((warn_unused_result, cold))
206
 
static bool run_tests(int, char *[]);
207
 
static void handle_sigchld(__attribute__((unused)) int sig){}
208
 
__attribute__((warn_unused_result, malloc))
209
 
task_queue *create_queue(void);
210
 
__attribute__((nonnull, warn_unused_result))
211
 
bool add_to_queue(task_queue *const, const task_context);
212
 
__attribute__((nonnull))
213
 
void cleanup_task(const task_context *const);
214
 
__attribute__((nonnull))
215
 
void cleanup_queue(task_queue *const *const);
216
 
__attribute__((pure, nonnull, warn_unused_result))
217
 
bool queue_has_question(const task_queue *const);
218
 
__attribute__((nonnull))
219
 
void cleanup_close(const int *const);
220
 
__attribute__((nonnull))
221
 
void cleanup_string(char *const *const);
222
 
__attribute__((nonnull))
223
 
void cleanup_buffer(buffer *const);
224
 
__attribute__((pure, nonnull, warn_unused_result))
225
 
bool string_set_contains(const string_set, const char *const);
226
 
__attribute__((nonnull, warn_unused_result))
227
 
bool string_set_add(string_set *const, const char *const);
228
 
__attribute__((nonnull))
229
 
void string_set_clear(string_set *);
230
 
void string_set_swap(string_set *const, string_set *const);
231
 
__attribute__((nonnull, warn_unused_result))
232
 
bool start_mandos_client(task_queue *const, const int, bool *const,
233
 
                         bool *const, buffer *const, bool *const,
234
 
                         const struct sigaction *const,
235
 
                         const sigset_t, const char *const,
236
 
                         const uid_t, const gid_t,
237
 
                         const char *const *const);
238
 
__attribute__((nonnull))
239
 
task_func wait_for_mandos_client_exit;
240
 
__attribute__((nonnull))
241
 
task_func read_mandos_client_output;
242
 
__attribute__((warn_unused_result))
243
 
bool add_inotify_dir_watch(task_queue *const, const int, bool *const,
244
 
                           buffer *const, const char *const,
245
 
                           string_set *, const mono_microsecs *const,
246
 
                           bool *const, bool *const);
247
 
__attribute__((nonnull))
248
 
task_func read_inotify_event;
249
 
__attribute__((nonnull))
250
 
task_func open_and_parse_question;
251
 
__attribute__((nonnull))
252
 
task_func cancel_old_question;
253
 
__attribute__((nonnull))
254
 
task_func connect_question_socket;
255
 
__attribute__((nonnull))
256
 
task_func send_password_to_socket;
257
 
__attribute__((warn_unused_result))
258
 
bool add_existing_questions(task_queue *const, const int,
259
 
                            buffer *const, string_set *,
260
 
                            const mono_microsecs *const,
261
 
                            bool *const, bool *const,
262
 
                            const char *const);
263
 
__attribute__((nonnull, warn_unused_result))
264
 
bool wait_for_event(const int, const mono_microsecs,
265
 
                    const mono_microsecs);
266
 
bool run_queue(task_queue **const, string_set *const, bool *const);
267
 
bool clear_all_fds_from_epoll_set(const int);
268
 
mono_microsecs get_current_time(void);
269
 
__attribute__((nonnull, warn_unused_result))
270
 
bool setup_signal_handler(struct sigaction *const);
271
 
__attribute__((nonnull))
272
 
bool restore_signal_handler(const struct sigaction *const);
273
 
__attribute__((nonnull, warn_unused_result))
274
 
bool block_sigchld(sigset_t *const);
275
 
__attribute__((nonnull))
276
 
bool restore_sigmask(const sigset_t *const);
277
 
__attribute__((nonnull))
278
 
bool parse_arguments(int, char *[], const bool, char **, char **,
279
 
                     uid_t *const , gid_t *const, char **, size_t *);
280
 
 
281
 
/* End of declarations of private types and functions */
282
 
 
283
 
/* Start of "main" section; this section LACKS TESTS!
284
 
 
285
 
   Code here should be as simple as possible. */
286
 
 
287
 
/* These are required to be global by Argp */
288
 
const char *argp_program_version = "password-agent " VERSION;
289
 
const char *argp_program_bug_address = "<mandos@recompile.se>";
290
 
 
291
 
int main(int argc, char *argv[]){
292
 
 
293
 
  /* If the --test option is passed, skip all normal operations and
294
 
     instead only run the run_tests() function, which also does all
295
 
     its own option parsing, so we don't have to do anything here. */
296
 
  if(should_only_run_tests(&argc, &argv)){
297
 
    if(run_tests(argc, argv)){
298
 
      return EXIT_SUCCESS;      /* All tests successful */
299
 
    }
300
 
    return EXIT_FAILURE;        /* Some test(s) failed */
301
 
  }
302
 
 
303
 
  __attribute__((cleanup(cleanup_string)))
304
 
    char *agent_directory = NULL;
305
 
 
306
 
  __attribute__((cleanup(cleanup_string)))
307
 
    char *helper_directory = NULL;
308
 
 
309
 
  uid_t user = 0;
310
 
  gid_t group = 0;
311
 
 
312
 
  __attribute__((cleanup(cleanup_string)))
313
 
    char *mandos_argz = NULL;
314
 
  size_t mandos_argz_length = 0;
315
 
 
316
 
  if(not parse_arguments(argc, argv, true, &agent_directory,
317
 
                         &helper_directory, &user, &group,
318
 
                         &mandos_argz, &mandos_argz_length)){
319
 
    /* This should never happen, since "true" is passed as the third
320
 
       argument to parse_arguments() above, which should make
321
 
       argp_parse() call exit() if any parsing error occurs. */
322
 
    error(EX_USAGE, errno, "Failed to parse arguments");
323
 
  }
324
 
 
325
 
  const char default_agent_directory[] = "/run/systemd/ask-password";
326
 
  const char default_helper_directory[]
327
 
    = "/lib/mandos/plugin-helpers";
328
 
  const char *const default_argv[]
329
 
    = {"/lib/mandos/plugins.d/mandos-client", NULL };
330
 
 
331
 
  /* Set variables to default values if unset */
332
 
  if(agent_directory == NULL){
333
 
    agent_directory = strdup(default_agent_directory);
334
 
    if(agent_directory == NULL){
335
 
      error(EX_OSERR, errno, "Failed strdup()");
336
 
    }
337
 
  }
338
 
  if(helper_directory == NULL){
339
 
    helper_directory = strdup(default_helper_directory);
340
 
    if(helper_directory == NULL){
341
 
      error(EX_OSERR, errno, "Failed strdup()");
342
 
    }
343
 
  }
344
 
  if(user == 0){
345
 
    user = 65534;               /* nobody */
346
 
  }
347
 
  if(group == 0){
348
 
    group = 65534;              /* nogroup */
349
 
  }
350
 
  /* If parse_opt did not create an argz vector, create one with
351
 
     default values */
352
 
  if(mandos_argz == NULL){
353
 
#ifdef __GNUC__
354
 
#pragma GCC diagnostic push
355
 
    /* argz_create() takes a non-const argv for some unknown reason -
356
 
       argz_create() isn't modifying the strings, just copying them.
357
 
       Therefore, this cast to non-const should be safe. */
358
 
#pragma GCC diagnostic ignored "-Wcast-qual"
359
 
#endif
360
 
    errno = argz_create((char *const *)default_argv, &mandos_argz,
361
 
                        &mandos_argz_length);
362
 
#ifdef __GNUC__
363
 
#pragma GCC diagnostic pop
364
 
#endif
365
 
    if(errno != 0){
366
 
      error(EX_OSERR, errno, "Failed argz_create()");
367
 
    }
368
 
  }
369
 
  /* Use argz vector to create a normal argv, usable by execv() */
370
 
 
371
 
  char **mandos_argv = malloc((argz_count(mandos_argz,
372
 
                                          mandos_argz_length)
373
 
                               + 1) * sizeof(char *));
374
 
  if(mandos_argv == NULL){
375
 
    error_t saved_errno = errno;
376
 
    free(mandos_argz);
377
 
    error(EX_OSERR, saved_errno, "Failed malloc()");
378
 
  }
379
 
  argz_extract(mandos_argz, mandos_argz_length, mandos_argv);
380
 
 
381
 
  sigset_t orig_sigmask;
382
 
  if(not block_sigchld(&orig_sigmask)){
383
 
    return EX_OSERR;
384
 
  }
385
 
 
386
 
  struct sigaction old_sigchld_action;
387
 
  if(not setup_signal_handler(&old_sigchld_action)){
388
 
    return EX_OSERR;
389
 
  }
390
 
 
391
 
  mono_microsecs current_time = 0;
392
 
 
393
 
  bool mandos_client_exited = false;
394
 
  bool quit_now = false;
395
 
  __attribute__((cleanup(cleanup_close)))
396
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
397
 
  if(epoll_fd < 0){
398
 
    error(EX_OSERR, errno, "Failed to create epoll set fd");
399
 
  }
400
 
  __attribute__((cleanup(cleanup_queue)))
401
 
    task_queue *queue = create_queue();
402
 
  if(queue == NULL){
403
 
    error(EX_OSERR, errno, "Failed to create task queue");
404
 
  }
405
 
 
406
 
  __attribute__((cleanup(cleanup_buffer)))
407
 
    buffer password = {};
408
 
  bool password_is_read = false;
409
 
 
410
 
  __attribute__((cleanup(string_set_clear)))
411
 
    string_set cancelled_filenames = {};
412
 
 
413
 
  /* Add tasks to queue */
414
 
  if(not start_mandos_client(queue, epoll_fd, &mandos_client_exited,
415
 
                             &quit_now, &password, &password_is_read,
416
 
                             &old_sigchld_action, orig_sigmask,
417
 
                             helper_directory, user, group,
418
 
                             (const char *const *)mandos_argv)){
419
 
    return EX_OSERR;            /* Error has already been printed */
420
 
  }
421
 
  /* These variables were only for start_mandos_client() and are not
422
 
     needed anymore */
423
 
  free(mandos_argv);
424
 
  free(mandos_argz);
425
 
  mandos_argz = NULL;
426
 
  if(not add_inotify_dir_watch(queue, epoll_fd, &quit_now, &password,
427
 
                               agent_directory, &cancelled_filenames,
428
 
                               &current_time, &mandos_client_exited,
429
 
                               &password_is_read)){
430
 
    switch(errno){              /* Error has already been printed */
431
 
    case EACCES:
432
 
    case ENAMETOOLONG:
433
 
    case ENOENT:
434
 
      return EX_OSFILE;
435
 
    default:
436
 
      return EX_OSERR;
437
 
    }
438
 
  }
439
 
  if(not add_existing_questions(queue, epoll_fd, &password,
440
 
                                &cancelled_filenames, &current_time,
441
 
                                &mandos_client_exited,
442
 
                                &password_is_read, agent_directory)){
443
 
    return EXIT_FAILURE;        /* Error has already been printed */
444
 
  }
445
 
 
446
 
  /* Run queue */
447
 
  do {
448
 
    current_time = get_current_time();
449
 
    if(not wait_for_event(epoll_fd, queue->next_run, current_time)){
450
 
      const error_t saved_errno = errno;
451
 
      error(EXIT_FAILURE, saved_errno, "Failure while waiting for"
452
 
            " events");
453
 
    }
454
 
 
455
 
    current_time = get_current_time();
456
 
    if(not run_queue(&queue, &cancelled_filenames, &quit_now)){
457
 
      const error_t saved_errno = errno;
458
 
      error(EXIT_FAILURE, saved_errno, "Failure while running queue");
459
 
    }
460
 
 
461
 
    /*  When no tasks about questions are left in the queue, break out
462
 
        of the loop (and implicitly exit the program) */
463
 
  } while(queue_has_question(queue));
464
 
 
465
 
  restore_signal_handler(&old_sigchld_action);
466
 
  restore_sigmask(&orig_sigmask);
467
 
 
468
 
  return EXIT_SUCCESS;
469
 
}
470
 
 
471
 
__attribute__((warn_unused_result))
472
 
mono_microsecs get_current_time(void){
473
 
  struct timespec currtime;
474
 
  if(clock_gettime(CLOCK_MONOTONIC, &currtime) != 0){
475
 
    error(0, errno, "Failed to get current time");
476
 
    return 0;
477
 
  }
478
 
  return ((mono_microsecs)currtime.tv_sec * 1000000) /* seconds */
479
 
    + ((mono_microsecs)currtime.tv_nsec / 1000);     /* nanoseconds */
480
 
}
481
 
 
482
 
/* End of "main" section */
483
 
 
484
 
/* Start of regular code section; ALL this code has tests */
485
 
 
486
 
__attribute__((nonnull))
487
 
bool parse_arguments(int argc, char *argv[], const bool exit_failure,
488
 
                     char **agent_directory, char **helper_directory,
489
 
                     uid_t *const user, gid_t *const group,
490
 
                     char **mandos_argz, size_t *mandos_argz_length){
491
 
 
492
 
  const struct argp_option options[] = {
493
 
    { .name="agent-directory",.key='d', .arg="DIRECTORY",
494
 
      .doc="Systemd password agent directory" },
495
 
    { .name="helper-directory",.key=128, .arg="DIRECTORY",
496
 
      .doc="Mandos Client password helper directory" },
497
 
    { .name="plugin-helper-dir", .key=129, /* From plugin-runner */
498
 
      .flags=OPTION_HIDDEN | OPTION_ALIAS },
499
 
    { .name="user", .key='u', .arg="USERID",
500
 
      .doc="User ID the Mandos Client will use as its unprivileged"
501
 
      " user" },
502
 
    { .name="userid", .key=130, /* From plugin--runner */
503
 
      .flags=OPTION_HIDDEN | OPTION_ALIAS },
504
 
    { .name="group", .key='g', .arg="GROUPID",
505
 
      .doc="Group ID the Mandos Client will use as its unprivileged"
506
 
      " group" },
507
 
    { .name="groupid", .key=131, /* From plugin--runner */
508
 
      .flags=OPTION_HIDDEN | OPTION_ALIAS },
509
 
    { .name="test", .key=255, /* See should_only_run_tests() */
510
 
      .doc="Skip normal operation, and only run self-tests.  See"
511
 
      " --test --help.", .group=10, },
512
 
    { NULL },
513
 
  };
514
 
 
515
 
  __attribute__((nonnull(3)))
516
 
    error_t parse_opt(int key, char *arg, struct argp_state *state){
517
 
    errno = 0;
518
 
    switch(key){
519
 
    case 'd':                   /* --agent-directory */
520
 
      *agent_directory = strdup(arg);
521
 
      break;
522
 
    case 128:                   /* --helper-directory */
523
 
    case 129:                   /* --plugin-helper-dir */
524
 
      *helper_directory = strdup(arg);
525
 
      break;
526
 
    case 'u':                   /* --user */
527
 
    case 130:                   /* --userid */
528
 
      {
529
 
        char *tmp;
530
 
        uintmax_t tmp_id = 0;
531
 
        errno = 0;
532
 
        tmp_id = (uid_t)strtoumax(arg, &tmp, 10);
533
 
        if(errno != 0 or tmp == arg or *tmp != '\0'
534
 
           or tmp_id != (uid_t)tmp_id or (uid_t)tmp_id == 0){
535
 
          return ARGP_ERR_UNKNOWN;
536
 
        }
537
 
        *user = (uid_t)tmp_id;
538
 
        errno = 0;
539
 
        break;
540
 
      }
541
 
    case 'g':                   /* --group */
542
 
    case 131:                   /* --groupid */
543
 
      {
544
 
        char *tmp;
545
 
        uintmax_t tmp_id = 0;
546
 
        errno = 0;
547
 
        tmp_id = (uid_t)strtoumax(arg, &tmp, 10);
548
 
        if(errno != 0 or tmp == arg or *tmp != '\0'
549
 
           or tmp_id != (gid_t)tmp_id or (gid_t)tmp_id == 0){
550
 
          return ARGP_ERR_UNKNOWN;
551
 
        }
552
 
        *group = (gid_t)tmp_id;
553
 
        errno = 0;
554
 
        break;
555
 
      }
556
 
    case ARGP_KEY_ARGS:
557
 
      /* Copy arguments into argz vector */
558
 
      return argz_create(state->argv + state->next, mandos_argz,
559
 
                         mandos_argz_length);
560
 
    default:
561
 
      return ARGP_ERR_UNKNOWN;
562
 
    }
563
 
    return errno;
564
 
  }
565
 
 
566
 
  const struct argp argp = {
567
 
    .options=options,
568
 
    .parser=parse_opt,
569
 
    .args_doc="[MANDOS_CLIENT [OPTION...]]\n--test",
570
 
    .doc = "Mandos password agent -- runs Mandos client as a"
571
 
    " systemd password agent",
572
 
  };
573
 
 
574
 
  errno = argp_parse(&argp, argc, argv,
575
 
                     exit_failure ? 0 : ARGP_NO_EXIT, NULL, NULL);
576
 
 
577
 
  return errno == 0;
578
 
}
579
 
 
580
 
__attribute__((nonnull, warn_unused_result))
581
 
bool block_sigchld(sigset_t *const orig_sigmask){
582
 
  sigset_t sigchld_sigmask;
583
 
  if(sigemptyset(&sigchld_sigmask) < 0){
584
 
    error(0, errno, "Failed to empty signal set");
585
 
    return false;
586
 
  }
587
 
  if(sigaddset(&sigchld_sigmask, SIGCHLD) < 0){
588
 
    error(0, errno, "Failed to add SIGCHLD to signal set");
589
 
    return false;
590
 
  }
591
 
  if(pthread_sigmask(SIG_BLOCK, &sigchld_sigmask, orig_sigmask) != 0){
592
 
    error(0, errno, "Failed to block SIGCHLD signal");
593
 
    return false;
594
 
  }
595
 
  return true;
596
 
}
597
 
 
598
 
__attribute__((nonnull, warn_unused_result, const))
599
 
bool restore_sigmask(const sigset_t *const orig_sigmask){
600
 
  if(pthread_sigmask(SIG_SETMASK, orig_sigmask, NULL) != 0){
601
 
    error(0, errno, "Failed to restore blocked signals");
602
 
    return false;
603
 
  }
604
 
  return true;
605
 
}
606
 
 
607
 
__attribute__((nonnull, warn_unused_result))
608
 
bool setup_signal_handler(struct sigaction *const old_sigchld_action){
609
 
  struct sigaction sigchld_action = {
610
 
    .sa_handler=handle_sigchld,
611
 
    .sa_flags=SA_RESTART | SA_NOCLDSTOP,
612
 
  };
613
 
  /* Set all signals in "sa_mask" struct member; this makes all
614
 
     signals automatically blocked during signal handler */
615
 
  if(sigfillset(&sigchld_action.sa_mask) != 0){
616
 
    error(0, errno, "Failed to do sigfillset()");
617
 
    return false;
618
 
  }
619
 
  if(sigaction(SIGCHLD, &sigchld_action, old_sigchld_action) != 0){
620
 
    error(0, errno, "Failed to set SIGCHLD signal handler");
621
 
    return false;
622
 
  }
623
 
  return true;
624
 
}
625
 
 
626
 
__attribute__((nonnull, warn_unused_result))
627
 
bool restore_signal_handler(const struct sigaction *const
628
 
                            old_sigchld_action){
629
 
  if(sigaction(SIGCHLD, old_sigchld_action, NULL) != 0){
630
 
    error(0, errno, "Failed to restore signal handler");
631
 
    return false;
632
 
  }
633
 
  return true;
634
 
}
635
 
 
636
 
__attribute__((warn_unused_result, malloc))
637
 
task_queue *create_queue(void){
638
 
  task_queue *queue = malloc(sizeof(task_queue));
639
 
  if(queue){
640
 
    queue->tasks = NULL;
641
 
    queue->length = 0;
642
 
    queue->allocated = 0;
643
 
    queue->next_run = 0;
644
 
  }
645
 
  return queue;
646
 
}
647
 
 
648
 
__attribute__((nonnull, warn_unused_result))
649
 
bool add_to_queue(task_queue *const queue, const task_context task){
650
 
  const size_t needed_size = sizeof(task_context)*(queue->length + 1);
651
 
  if(needed_size > (queue->allocated)){
652
 
    task_context *const new_tasks = realloc(queue->tasks,
653
 
                                            needed_size);
654
 
    if(new_tasks == NULL){
655
 
      error(0, errno, "Failed to allocate %" PRIuMAX
656
 
            " bytes for queue->tasks", (uintmax_t)needed_size);
657
 
      return false;
658
 
    }
659
 
    queue->tasks = new_tasks;
660
 
    queue->allocated = needed_size;
661
 
  }
662
 
  /* Using memcpy here is necessary because doing */
663
 
  /* queue->tasks[queue->length++] = task; */
664
 
  /* would violate const-ness of task members */
665
 
  memcpy(&(queue->tasks[queue->length++]), &task,
666
 
         sizeof(task_context));
667
 
  return true;
668
 
}
669
 
 
670
 
__attribute__((nonnull))
671
 
void cleanup_task(const task_context *const task){
672
 
  const error_t saved_errno = errno;
673
 
  /* free and close all task data */
674
 
  free(task->question_filename);
675
 
  if(task->filename != task->question_filename){
676
 
    free(task->filename);
677
 
  }
678
 
  if(task->pid > 0){
679
 
    kill(task->pid, SIGTERM);
680
 
  }
681
 
  if(task->fd > 0){
682
 
    close(task->fd);
683
 
  }
684
 
  errno = saved_errno;
685
 
}
686
 
 
687
 
__attribute__((nonnull))
688
 
void free_queue(task_queue *const queue){
689
 
  free(queue->tasks);
690
 
  free(queue);
691
 
}
692
 
 
693
 
__attribute__((nonnull))
694
 
void cleanup_queue(task_queue *const *const queue){
695
 
  if(*queue == NULL){
696
 
    return;
697
 
  }
698
 
  for(size_t i = 0; i < (*queue)->length; i++){
699
 
    const task_context *const task = ((*queue)->tasks)+i;
700
 
    cleanup_task(task);
701
 
  }
702
 
  free_queue(*queue);
703
 
}
704
 
 
705
 
__attribute__((pure, nonnull, warn_unused_result))
706
 
bool queue_has_question(const task_queue *const queue){
707
 
  for(size_t i=0; i < queue->length; i++){
708
 
    if(queue->tasks[i].question_filename != NULL){
709
 
      return true;
710
 
    }
711
 
  }
712
 
  return false;
713
 
}
714
 
 
715
 
__attribute__((nonnull))
716
 
void cleanup_close(const int *const fd){
717
 
  const error_t saved_errno = errno;
718
 
  close(*fd);
719
 
  errno = saved_errno;
720
 
}
721
 
 
722
 
__attribute__((nonnull))
723
 
void cleanup_string(char *const *const ptr){
724
 
  free(*ptr);
725
 
}
726
 
 
727
 
__attribute__((nonnull))
728
 
void cleanup_buffer(buffer *buf){
729
 
  if(buf->allocated > 0){
730
 
#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)
731
 
    explicit_bzero(buf->data, buf->allocated);
732
 
#else
733
 
    memset(buf->data, '\0', buf->allocated);
734
 
#endif
735
 
  }
736
 
  if(buf->data != NULL){
737
 
    if(munlock(buf->data, buf->allocated) != 0){
738
 
      error(0, errno, "Failed to unlock memory of old buffer");
739
 
    }
740
 
    free(buf->data);
741
 
    buf->data = NULL;
742
 
  }
743
 
  buf->length = 0;
744
 
  buf->allocated = 0;
745
 
}
746
 
 
747
 
__attribute__((pure, nonnull, warn_unused_result))
748
 
bool string_set_contains(const string_set set, const char *const str){
749
 
  for(const char *s = set.argz; s != NULL and set.argz_len > 0;
750
 
      s = argz_next(set.argz, set.argz_len, s)){
751
 
    if(strcmp(s, str) == 0){
752
 
      return true;
753
 
    }
754
 
  }
755
 
  return false;
756
 
}
757
 
 
758
 
__attribute__((nonnull, warn_unused_result))
759
 
bool string_set_add(string_set *const set, const char *const str){
760
 
  if(string_set_contains(*set, str)){
761
 
    return true;
762
 
  }
763
 
  error_t error = argz_add(&set->argz, &set->argz_len, str);
764
 
  if(error == 0){
765
 
    return true;
766
 
  }
767
 
  errno = error;
768
 
  return false;
769
 
}
770
 
 
771
 
__attribute__((nonnull))
772
 
void string_set_clear(string_set *set){
773
 
  free(set->argz);
774
 
  set->argz = NULL;
775
 
  set->argz_len = 0;
776
 
}
777
 
 
778
 
__attribute__((nonnull))
779
 
void string_set_swap(string_set *const set1, string_set *const set2){
780
 
  /* Swap contents of two string sets */
781
 
  {
782
 
    char *const tmp_argz = set1->argz;
783
 
    set1->argz = set2->argz;
784
 
    set2->argz = tmp_argz;
785
 
  }
786
 
  {
787
 
    const size_t tmp_argz_len = set1->argz_len;
788
 
    set1->argz_len = set2->argz_len;
789
 
    set2->argz_len = tmp_argz_len;
790
 
  }
791
 
}
792
 
 
793
 
__attribute__((nonnull, warn_unused_result))
794
 
bool start_mandos_client(task_queue *const queue,
795
 
                         const int epoll_fd,
796
 
                         bool *const mandos_client_exited,
797
 
                         bool *const quit_now, buffer *const password,
798
 
                         bool *const password_is_read,
799
 
                         const struct sigaction *const
800
 
                         old_sigchld_action, const sigset_t sigmask,
801
 
                         const char *const helper_directory,
802
 
                         const uid_t user, const gid_t group,
803
 
                         const char *const *const argv){
804
 
  int pipefds[2];
805
 
  if(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK) != 0){
806
 
    error(0, errno, "Failed to pipe2(..., O_CLOEXEC | O_NONBLOCK)");
807
 
    return false;
808
 
  }
809
 
 
810
 
  const pid_t pid = fork();
811
 
  if(pid == 0){
812
 
    if(not restore_signal_handler(old_sigchld_action)){
813
 
      _exit(EXIT_FAILURE);
814
 
    }
815
 
    if(not restore_sigmask(&sigmask)){
816
 
      _exit(EXIT_FAILURE);
817
 
    }
818
 
    if(close(pipefds[0]) != 0){
819
 
      error(0, errno, "Failed to close() parent pipe fd");
820
 
      _exit(EXIT_FAILURE);
821
 
    }
822
 
    if(dup2(pipefds[1], STDOUT_FILENO) == -1){
823
 
      error(0, errno, "Failed to dup2() pipe fd to stdout");
824
 
      _exit(EXIT_FAILURE);
825
 
    }
826
 
    if(close(pipefds[1]) != 0){
827
 
      error(0, errno, "Failed to close() old child pipe fd");
828
 
      _exit(EXIT_FAILURE);
829
 
    }
830
 
    if(setenv("MANDOSPLUGINHELPERDIR", helper_directory, 1) != 0){
831
 
      error(0, errno, "Failed to setenv(\"MANDOSPLUGINHELPERDIR\","
832
 
            " \"%s\", 1)", helper_directory);
833
 
      _exit(EXIT_FAILURE);
834
 
    }
835
 
    if(group != 0 and setresgid(group, 0, 0) == -1){
836
 
      error(0, errno, "Failed to setresgid(-1, %" PRIuMAX ", %"
837
 
            PRIuMAX")", (uintmax_t)group, (uintmax_t)group);
838
 
      _exit(EXIT_FAILURE);
839
 
    }
840
 
    if(user != 0 and setresuid(user, 0, 0) == -1){
841
 
      error(0, errno, "Failed to setresuid(-1, %" PRIuMAX ", %"
842
 
            PRIuMAX")", (uintmax_t)user, (uintmax_t)user);
843
 
      _exit(EXIT_FAILURE);
844
 
    }
845
 
#ifdef __GNUC__
846
 
#pragma GCC diagnostic push
847
 
    /* For historical reasons, the "argv" argument to execv() is not
848
 
       const, but it is safe to override this. */
849
 
#pragma GCC diagnostic ignored "-Wcast-qual"
850
 
#endif
851
 
    execv(argv[0], (char **)argv);
852
 
#ifdef __GNUC__
853
 
#pragma GCC diagnostic pop
854
 
#endif
855
 
    error(0, errno, "execv(\"%s\", ...) failed", argv[0]);
856
 
    _exit(EXIT_FAILURE);
857
 
  }
858
 
  close(pipefds[1]);
859
 
 
860
 
  if(not add_to_queue(queue, (task_context){
861
 
        .func=wait_for_mandos_client_exit,
862
 
        .pid=pid,
863
 
        .mandos_client_exited=mandos_client_exited,
864
 
        .quit_now=quit_now,
865
 
      })){
866
 
    error(0, errno, "Failed to add wait_for_mandos_client to queue");
867
 
    close(pipefds[0]);
868
 
    return false;
869
 
  }
870
 
 
871
 
  const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, pipefds[0],
872
 
                            &(struct epoll_event)
873
 
                            { .events=EPOLLIN | EPOLLRDHUP });
874
 
  if(ret != 0 and errno != EEXIST){
875
 
    error(0, errno, "Failed to add file descriptor to epoll set");
876
 
    close(pipefds[0]);
877
 
    return false;
878
 
  }
879
 
 
880
 
  return add_to_queue(queue, (task_context){
881
 
      .func=read_mandos_client_output,
882
 
      .epoll_fd=epoll_fd,
883
 
      .fd=pipefds[0],
884
 
      .quit_now=quit_now,
885
 
      .password=password,
886
 
      .password_is_read=password_is_read,
887
 
    });
888
 
}
889
 
 
890
 
__attribute__((nonnull))
891
 
void wait_for_mandos_client_exit(const task_context task,
892
 
                                 task_queue *const queue){
893
 
  const pid_t pid = task.pid;
894
 
  bool *const mandos_client_exited = task.mandos_client_exited;
895
 
  bool *const quit_now = task.quit_now;
896
 
 
897
 
  int status;
898
 
  switch(waitpid(pid, &status, WNOHANG)){
899
 
  case 0:                       /* Not exited yet */
900
 
    if(not add_to_queue(queue, task)){
901
 
      error(0, errno, "Failed to add myself to queue");
902
 
      *quit_now = true;
903
 
    }
904
 
    break;
905
 
  case -1:                      /* Error */
906
 
    error(0, errno, "waitpid(%" PRIdMAX ") failed", (intmax_t)pid);
907
 
    if(errno != ECHILD){
908
 
      kill(pid, SIGTERM);
909
 
    }
910
 
    *quit_now = true;
911
 
    break;
912
 
  default:                      /* Has exited */
913
 
    *mandos_client_exited = true;
914
 
    if((not WIFEXITED(status))
915
 
       or (WEXITSTATUS(status) != EXIT_SUCCESS)){
916
 
      error(0, 0, "Mandos client failed or was killed");
917
 
      *quit_now = true;
918
 
    }
919
 
  }
920
 
}
921
 
 
922
 
__attribute__((nonnull))
923
 
void read_mandos_client_output(const task_context task,
924
 
                               task_queue *const queue){
925
 
  buffer *const password = task.password;
926
 
  bool *const quit_now = task.quit_now;
927
 
  bool *const password_is_read = task.password_is_read;
928
 
  const int fd = task.fd;
929
 
  const int epoll_fd = task.epoll_fd;
930
 
 
931
 
  const size_t new_potential_size = (password->length + PIPE_BUF);
932
 
  if(password->allocated < new_potential_size){
933
 
    char *const new_buffer = calloc(new_potential_size, 1);
934
 
    if(new_buffer == NULL){
935
 
      error(0, errno, "Failed to allocate %" PRIuMAX
936
 
            " bytes for password", (uintmax_t)new_potential_size);
937
 
      *quit_now = true;
938
 
      close(fd);
939
 
      return;
940
 
    }
941
 
    if(mlock(new_buffer, new_potential_size) != 0){
942
 
      /* Warn but do not treat as fatal error */
943
 
      if(errno != EPERM and errno != ENOMEM){
944
 
        error(0, errno, "Failed to lock memory for password");
945
 
      }
946
 
    }
947
 
    if(password->length > 0){
948
 
      memcpy(new_buffer, password->data, password->length);
949
 
#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)
950
 
      explicit_bzero(password->data, password->allocated);
951
 
#else
952
 
      memset(password->data, '\0', password->allocated);
953
 
#endif
954
 
    }
955
 
    if(password->data != NULL){
956
 
      if(munlock(password->data, password->allocated) != 0){
957
 
        error(0, errno, "Failed to unlock memory of old buffer");
958
 
      }
959
 
      free(password->data);
960
 
    }
961
 
    password->data = new_buffer;
962
 
    password->allocated = new_potential_size;
963
 
  }
964
 
 
965
 
  const ssize_t read_length = read(fd, password->data
966
 
                                   + password->length, PIPE_BUF);
967
 
 
968
 
  if(read_length == 0){ /* EOF */
969
 
    *password_is_read = true;
970
 
    close(fd);
971
 
    return;
972
 
  }
973
 
  if(read_length < 0 and errno != EAGAIN){ /* Actual error */
974
 
    error(0, errno, "Failed to read password from Mandos client");
975
 
    *quit_now = true;
976
 
    close(fd);
977
 
    return;
978
 
  }
979
 
  if(read_length > 0){          /* Data has been read */
980
 
    password->length += (size_t)read_length;
981
 
  }
982
 
 
983
 
  /* Either data was read, or EAGAIN was indicated, meaning no data
984
 
     available yet */
985
 
 
986
 
  /* Re-add the fd to the epoll set */
987
 
  const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
988
 
                            &(struct epoll_event)
989
 
                            { .events=EPOLLIN | EPOLLRDHUP });
990
 
  if(ret != 0 and errno != EEXIST){
991
 
    error(0, errno, "Failed to re-add file descriptor to epoll set");
992
 
    *quit_now = true;
993
 
    close(fd);
994
 
    return;
995
 
  }
996
 
 
997
 
  /* Re-add myself to the queue */
998
 
  if(not add_to_queue(queue, task)){
999
 
    error(0, errno, "Failed to add myself to queue");
1000
 
    *quit_now = true;
1001
 
    close(fd);
1002
 
  }
1003
 
}
1004
 
 
1005
 
__attribute__((nonnull, warn_unused_result))
1006
 
bool add_inotify_dir_watch(task_queue *const queue,
1007
 
                           const int epoll_fd, bool *const quit_now,
1008
 
                           buffer *const password,
1009
 
                           const char *const dir,
1010
 
                           string_set *cancelled_filenames,
1011
 
                           const mono_microsecs *const current_time,
1012
 
                           bool *const mandos_client_exited,
1013
 
                           bool *const password_is_read){
1014
 
  const int fd = inotify_init1(IN_NONBLOCK | IN_CLOEXEC);
1015
 
  if(fd == -1){
1016
 
    error(0, errno, "Failed to create inotify instance");
1017
 
    return false;
1018
 
  }
1019
 
 
1020
 
  if(inotify_add_watch(fd, dir, IN_CLOSE_WRITE
1021
 
                       | IN_MOVED_TO | IN_DELETE)
1022
 
     == -1){
1023
 
    error(0, errno, "Failed to create inotify watch on %s", dir);
1024
 
    return false;
1025
 
  }
1026
 
 
1027
 
  /* Add the inotify fd to the epoll set */
1028
 
  const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
1029
 
                            &(struct epoll_event)
1030
 
                            { .events=EPOLLIN | EPOLLRDHUP });
1031
 
  if(ret != 0 and errno != EEXIST){
1032
 
    error(0, errno, "Failed to add file descriptor to epoll set");
1033
 
    close(fd);
1034
 
    return false;
1035
 
  }
1036
 
 
1037
 
  const task_context read_inotify_event_task = {
1038
 
    .func=read_inotify_event,
1039
 
    .epoll_fd=epoll_fd,
1040
 
    .quit_now=quit_now,
1041
 
    .password=password,
1042
 
    .fd=fd,
1043
 
    .filename=strdup(dir),
1044
 
    .cancelled_filenames=cancelled_filenames,
1045
 
    .current_time=current_time,
1046
 
    .mandos_client_exited=mandos_client_exited,
1047
 
    .password_is_read=password_is_read,
1048
 
  };
1049
 
  if(read_inotify_event_task.filename == NULL){
1050
 
    error(0, errno, "Failed to strdup(\"%s\")", dir);
1051
 
    close(fd);
1052
 
    return false;
1053
 
  }
1054
 
 
1055
 
  return add_to_queue(queue, read_inotify_event_task);
1056
 
}
1057
 
 
1058
 
__attribute__((nonnull))
1059
 
void read_inotify_event(const task_context task,
1060
 
                        task_queue *const queue){
1061
 
  const int fd = task.fd;
1062
 
  const int epoll_fd = task.epoll_fd;
1063
 
  char *const filename = task.filename;
1064
 
  bool *quit_now = task.quit_now;
1065
 
  buffer *const password = task.password;
1066
 
  string_set *const cancelled_filenames = task.cancelled_filenames;
1067
 
  const mono_microsecs *const current_time = task.current_time;
1068
 
  bool *const mandos_client_exited = task.mandos_client_exited;
1069
 
  bool *const password_is_read = task.password_is_read;
1070
 
 
1071
 
  /* "sufficient to read at least one event." - inotify(7) */
1072
 
  const size_t ievent_size = (sizeof(struct inotify_event)
1073
 
                              + NAME_MAX + 1);
1074
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
1075
 
  struct inotify_event *ievent = ((struct inotify_event *)
1076
 
                                  ievent_buffer);
1077
 
 
1078
 
  const ssize_t read_length = read(fd, ievent, ievent_size);
1079
 
  if(read_length == 0){ /* EOF */
1080
 
    error(0, 0, "Got EOF from inotify fd for directory %s", filename);
1081
 
    *quit_now = true;
1082
 
    cleanup_task(&task);
1083
 
    return;
1084
 
  }
1085
 
  if(read_length < 0 and errno != EAGAIN){ /* Actual error */
1086
 
    error(0, errno, "Failed to read from inotify fd for directory %s",
1087
 
          filename);
1088
 
    *quit_now = true;
1089
 
    cleanup_task(&task);
1090
 
    return;
1091
 
  }
1092
 
  if(read_length > 0            /* Data has been read */
1093
 
     and fnmatch("ask.*", ievent->name, FNM_FILE_NAME) == 0){
1094
 
    char *question_filename = NULL;
1095
 
    const ssize_t question_filename_length
1096
 
      = asprintf(&question_filename, "%s/%s", filename, ievent->name);
1097
 
    if(question_filename_length < 0){
1098
 
      error(0, errno, "Failed to create file name from directory name"
1099
 
            " %s and file name %s", filename, ievent->name);
1100
 
    } else {
1101
 
      if(ievent->mask & (IN_CLOSE_WRITE | IN_MOVED_TO)){
1102
 
        if(not add_to_queue(queue, (task_context){
1103
 
              .func=open_and_parse_question,
1104
 
              .epoll_fd=epoll_fd,
1105
 
              .question_filename=question_filename,
1106
 
              .filename=question_filename,
1107
 
              .password=password,
1108
 
              .cancelled_filenames=cancelled_filenames,
1109
 
              .current_time=current_time,
1110
 
              .mandos_client_exited=mandos_client_exited,
1111
 
              .password_is_read=password_is_read,
1112
 
            })){
1113
 
          error(0, errno, "Failed to add open_and_parse_question task"
1114
 
                " for file name %s to queue", filename);
1115
 
        } else {
1116
 
          /* Force the added task (open_and_parse_question) to run
1117
 
             immediately */
1118
 
          queue->next_run = 1;
1119
 
        }
1120
 
      } else if(ievent->mask & IN_DELETE){
1121
 
        if(not string_set_add(cancelled_filenames,
1122
 
                              question_filename)){
1123
 
          error(0, errno, "Could not add question %s to"
1124
 
                " cancelled_questions", question_filename);
1125
 
          *quit_now = true;
1126
 
          free(question_filename);
1127
 
          cleanup_task(&task);
1128
 
          return;
1129
 
        }
1130
 
        free(question_filename);
1131
 
      }
1132
 
    }
1133
 
  }
1134
 
 
1135
 
  /* Either data was read, or EAGAIN was indicated, meaning no data
1136
 
     available yet */
1137
 
 
1138
 
  /* Re-add myself to the queue */
1139
 
  if(not add_to_queue(queue, task)){
1140
 
    error(0, errno, "Failed to re-add read_inotify_event(%s) to"
1141
 
          " queue", filename);
1142
 
    *quit_now = true;
1143
 
    cleanup_task(&task);
1144
 
    return;
1145
 
  }
1146
 
 
1147
 
  /* Re-add the fd to the epoll set */
1148
 
  const int ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
1149
 
                            &(struct epoll_event)
1150
 
                            { .events=EPOLLIN | EPOLLRDHUP });
1151
 
  if(ret != 0 and errno != EEXIST){
1152
 
    error(0, errno, "Failed to re-add inotify file descriptor %d for"
1153
 
          " directory %s to epoll set", fd, filename);
1154
 
    /* Force the added task (read_inotify_event) to run again, at most
1155
 
       one second from now */
1156
 
    if((queue->next_run == 0)
1157
 
       or (queue->next_run > (*current_time + 1000000))){
1158
 
      queue->next_run = *current_time + 1000000;
1159
 
    }
1160
 
  }
1161
 
}
1162
 
 
1163
 
__attribute__((nonnull))
1164
 
void open_and_parse_question(const task_context task,
1165
 
                             task_queue *const queue){
1166
 
  __attribute__((cleanup(cleanup_string)))
1167
 
    char *question_filename = task.question_filename;
1168
 
  const int epoll_fd = task.epoll_fd;
1169
 
  buffer *const password = task.password;
1170
 
  string_set *const cancelled_filenames = task.cancelled_filenames;
1171
 
  const mono_microsecs *const current_time = task.current_time;
1172
 
  bool *const mandos_client_exited = task.mandos_client_exited;
1173
 
  bool *const password_is_read = task.password_is_read;
1174
 
 
1175
 
  /* We use the GLib "Key-value file parser" functions to parse the
1176
 
     question file.  See <https://www.freedesktop.org/wiki/Software
1177
 
     /systemd/PasswordAgents/> for specification of contents */
1178
 
  __attribute__((nonnull))
1179
 
    void cleanup_g_key_file(GKeyFile **key_file){
1180
 
    if(*key_file != NULL){
1181
 
      g_key_file_free(*key_file);
1182
 
    }
1183
 
  }
1184
 
 
1185
 
  __attribute__((cleanup(cleanup_g_key_file)))
1186
 
    GKeyFile *key_file = g_key_file_new();
1187
 
  if(key_file == NULL){
1188
 
    error(0, errno, "Failed g_key_file_new() for \"%s\"",
1189
 
          question_filename);
1190
 
    return;
1191
 
  }
1192
 
  GError *glib_error = NULL;
1193
 
  if(g_key_file_load_from_file(key_file, question_filename,
1194
 
                               G_KEY_FILE_NONE, &glib_error) != TRUE){
1195
 
    /* If a file was removed, we should ignore it, so */
1196
 
    /* only show error message if file actually existed */
1197
 
    if(glib_error->code != G_FILE_ERROR_NOENT){
1198
 
      error(0, 0, "Failed to load question data from file \"%s\": %s",
1199
 
            question_filename, glib_error->message);
1200
 
    }
1201
 
    return;
1202
 
  }
1203
 
 
1204
 
  __attribute__((cleanup(cleanup_string)))
1205
 
    char *socket_name = g_key_file_get_string(key_file, "Ask",
1206
 
                                              "Socket",
1207
 
                                              &glib_error);
1208
 
  if(socket_name == NULL){
1209
 
    error(0, 0, "Question file \"%s\" did not contain \"Socket\": %s",
1210
 
          question_filename, glib_error->message);
1211
 
    return;
1212
 
  }
1213
 
 
1214
 
  if(strlen(socket_name) == 0){
1215
 
    error(0, 0, "Question file \"%s\" had empty \"Socket\" value",
1216
 
          question_filename);
1217
 
    return;
1218
 
  }
1219
 
 
1220
 
  const guint64 pid = g_key_file_get_uint64(key_file, "Ask", "PID",
1221
 
                                            &glib_error);
1222
 
  if(glib_error != NULL){
1223
 
    error(0, 0, "Question file \"%s\" contained bad \"PID\": %s",
1224
 
          question_filename, glib_error->message);
1225
 
    return;
1226
 
  }
1227
 
 
1228
 
  if((pid != (guint64)((pid_t)pid))
1229
 
     or (kill((pid_t)pid, 0) != 0)){
1230
 
    error(0, 0, "PID %" PRIuMAX " in question file \"%s\" is bad or"
1231
 
          " does not exist", (uintmax_t)pid, question_filename);
1232
 
    return;
1233
 
  }
1234
 
 
1235
 
  guint64 notafter = g_key_file_get_uint64(key_file, "Ask",
1236
 
                                           "NotAfter", &glib_error);
1237
 
  if(glib_error != NULL){
1238
 
    if(glib_error->code != G_KEY_FILE_ERROR_KEY_NOT_FOUND){
1239
 
      error(0, 0, "Question file \"%s\" contained bad \"NotAfter\":"
1240
 
            " %s", question_filename, glib_error->message);
1241
 
    }
1242
 
    notafter = 0;
1243
 
  }
1244
 
  if(notafter != 0){
1245
 
    if(queue->next_run == 0 or (queue->next_run > notafter)){
1246
 
      queue->next_run = notafter;
1247
 
    }
1248
 
    if(*current_time >= notafter){
1249
 
      return;
1250
 
    }
1251
 
  }
1252
 
 
1253
 
  const task_context connect_question_socket_task = {
1254
 
    .func=connect_question_socket,
1255
 
    .question_filename=strdup(question_filename),
1256
 
    .epoll_fd=epoll_fd,
1257
 
    .password=password,
1258
 
    .filename=strdup(socket_name),
1259
 
    .cancelled_filenames=task.cancelled_filenames,
1260
 
    .mandos_client_exited=mandos_client_exited,
1261
 
    .password_is_read=password_is_read,
1262
 
    .current_time=current_time,
1263
 
  };
1264
 
  if(connect_question_socket_task.question_filename == NULL
1265
 
     or connect_question_socket_task.filename == NULL
1266
 
     or not add_to_queue(queue, connect_question_socket_task)){
1267
 
    error(0, errno, "Failed to add connect_question_socket for socket"
1268
 
          " %s (from \"%s\") to queue", socket_name,
1269
 
          question_filename);
1270
 
    cleanup_task(&connect_question_socket_task);
1271
 
    return;
1272
 
  }
1273
 
  /* Force the added task (connect_question_socket) to run
1274
 
     immediately */
1275
 
  queue->next_run = 1;
1276
 
 
1277
 
  if(notafter > 0){
1278
 
    char *const dup_filename = strdup(question_filename);
1279
 
    const task_context cancel_old_question_task = {
1280
 
      .func=cancel_old_question,
1281
 
      .question_filename=dup_filename,
1282
 
      .notafter=notafter,
1283
 
      .filename=dup_filename,
1284
 
      .cancelled_filenames=cancelled_filenames,
1285
 
      .current_time=current_time,
1286
 
    };
1287
 
    if(cancel_old_question_task.question_filename == NULL
1288
 
       or not add_to_queue(queue, cancel_old_question_task)){
1289
 
      error(0, errno, "Failed to add cancel_old_question for file "
1290
 
            "\"%s\" to queue", question_filename);
1291
 
      cleanup_task(&cancel_old_question_task);
1292
 
      return;
1293
 
    }
1294
 
  }
1295
 
}
1296
 
 
1297
 
__attribute__((nonnull))
1298
 
void cancel_old_question(const task_context task,
1299
 
                         task_queue *const queue){
1300
 
  char *const question_filename = task.question_filename;
1301
 
  string_set *const cancelled_filenames = task.cancelled_filenames;
1302
 
  const mono_microsecs notafter = task.notafter;
1303
 
  const mono_microsecs *const current_time = task.current_time;
1304
 
 
1305
 
  if(*current_time >= notafter){
1306
 
    if(not string_set_add(cancelled_filenames, question_filename)){
1307
 
      error(0, errno, "Failed to cancel question for file %s",
1308
 
            question_filename);
1309
 
    }
1310
 
    cleanup_task(&task);
1311
 
    return;
1312
 
  }
1313
 
 
1314
 
  if(not add_to_queue(queue, task)){
1315
 
    error(0, errno, "Failed to add cancel_old_question for file "
1316
 
          "%s to queue", question_filename);
1317
 
    cleanup_task(&task);
1318
 
    return;
1319
 
  }
1320
 
 
1321
 
  if((queue->next_run == 0) or (queue->next_run > notafter)){
1322
 
    queue->next_run = notafter;
1323
 
  }
1324
 
}
1325
 
 
1326
 
__attribute__((nonnull))
1327
 
void connect_question_socket(const task_context task,
1328
 
                             task_queue *const queue){
1329
 
  char *const question_filename = task.question_filename;
1330
 
  char *const filename = task.filename;
1331
 
  const int epoll_fd = task.epoll_fd;
1332
 
  buffer *const password = task.password;
1333
 
  string_set *const cancelled_filenames = task.cancelled_filenames;
1334
 
  bool *const mandos_client_exited = task.mandos_client_exited;
1335
 
  bool *const password_is_read = task.password_is_read;
1336
 
  const mono_microsecs *const current_time = task.current_time;
1337
 
 
1338
 
  struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };
1339
 
 
1340
 
  if(sizeof(sock_name.sun_path) <= strlen(filename)){
1341
 
    error(0, 0, "Socket filename is larger than"
1342
 
          " sizeof(sockaddr_un.sun_path); %" PRIuMAX ": \"%s\"",
1343
 
          (uintmax_t)sizeof(sock_name.sun_path), filename);
1344
 
    if(not string_set_add(cancelled_filenames, question_filename)){
1345
 
      error(0, errno, "Failed to cancel question for file %s",
1346
 
            question_filename);
1347
 
    }
1348
 
    cleanup_task(&task);
1349
 
    return;
1350
 
  }
1351
 
 
1352
 
  const int fd = socket(PF_LOCAL, SOCK_DGRAM
1353
 
                        | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
1354
 
  if(fd < 0){
1355
 
    error(0, errno,
1356
 
          "Failed to create socket(PF_LOCAL, SOCK_DGRAM, 0)");
1357
 
    if(not add_to_queue(queue, task)){
1358
 
      error(0, errno, "Failed to add connect_question_socket for file"
1359
 
            " \"%s\" and socket \"%s\" to queue", question_filename,
1360
 
            filename);
1361
 
      cleanup_task(&task);
1362
 
    } else {
1363
 
      /* Force the added task (connect_question_socket) to run
1364
 
         immediately */
1365
 
      queue->next_run = 1;
1366
 
    }
1367
 
    return;
1368
 
  }
1369
 
 
1370
 
  strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));
1371
 
  if(connect(fd, (struct sockaddr *)&sock_name,
1372
 
             (socklen_t)SUN_LEN(&sock_name)) != 0){
1373
 
    error(0, errno, "Failed to connect socket to \"%s\"", filename);
1374
 
    if(not add_to_queue(queue, task)){
1375
 
      error(0, errno, "Failed to add connect_question_socket for file"
1376
 
            " \"%s\" and socket \"%s\" to queue", question_filename,
1377
 
            filename);
1378
 
      cleanup_task(&task);
1379
 
    } else {
1380
 
      /* Force the added task (connect_question_socket) to run again,
1381
 
         at most one second from now */
1382
 
      if((queue->next_run == 0)
1383
 
         or (queue->next_run > (*current_time + 1000000))){
1384
 
        queue->next_run = *current_time + 1000000;
1385
 
      }
1386
 
    }
1387
 
    return;
1388
 
  }
1389
 
 
1390
 
  /* Not necessary, but we can try, and merely warn on failure */
1391
 
  if(shutdown(fd, SHUT_RD) != 0){
1392
 
    error(0, errno, "Failed to shutdown reading from socket \"%s\"",
1393
 
          filename);
1394
 
  }
1395
 
 
1396
 
  /* Add the fd to the epoll set */
1397
 
  if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
1398
 
               &(struct epoll_event){ .events=EPOLLOUT })
1399
 
     != 0){
1400
 
    error(0, errno, "Failed to add inotify file descriptor %d for"
1401
 
          " socket %s to epoll set", fd, filename);
1402
 
    if(not add_to_queue(queue, task)){
1403
 
      error(0, errno, "Failed to add connect_question_socket for file"
1404
 
            " \"%s\" and socket \"%s\" to queue", question_filename,
1405
 
            filename);
1406
 
      cleanup_task(&task);
1407
 
    } else {
1408
 
      /* Force the added task (connect_question_socket) to run again,
1409
 
         at most one second from now */
1410
 
      if((queue->next_run == 0)
1411
 
         or (queue->next_run > (*current_time + 1000000))){
1412
 
        queue->next_run = *current_time + 1000000;
1413
 
      }
1414
 
    }
1415
 
    return;
1416
 
  }
1417
 
 
1418
 
  /* add task send_password_to_socket to queue */
1419
 
  const task_context send_password_to_socket_task = {
1420
 
    .func=send_password_to_socket,
1421
 
    .question_filename=question_filename,
1422
 
    .filename=filename,
1423
 
    .epoll_fd=epoll_fd,
1424
 
    .fd=fd,
1425
 
    .password=password,
1426
 
    .cancelled_filenames=cancelled_filenames,
1427
 
    .mandos_client_exited=mandos_client_exited,
1428
 
    .password_is_read=password_is_read,
1429
 
    .current_time=current_time,
1430
 
  };
1431
 
 
1432
 
  if(not add_to_queue(queue, send_password_to_socket_task)){
1433
 
    error(0, errno, "Failed to add send_password_to_socket for"
1434
 
          " file \"%s\" and socket \"%s\" to queue",
1435
 
          question_filename, filename);
1436
 
    cleanup_task(&send_password_to_socket_task);
1437
 
  }
1438
 
}
1439
 
 
1440
 
__attribute__((nonnull))
1441
 
void send_password_to_socket(const task_context task,
1442
 
                             task_queue *const queue){
1443
 
  char *const question_filename=task.question_filename;
1444
 
  char *const filename=task.filename;
1445
 
  const int epoll_fd=task.epoll_fd;
1446
 
  const int fd=task.fd;
1447
 
  buffer *const password=task.password;
1448
 
  string_set *const cancelled_filenames=task.cancelled_filenames;
1449
 
  bool *const mandos_client_exited = task.mandos_client_exited;
1450
 
  bool *const password_is_read = task.password_is_read;
1451
 
  const mono_microsecs *const current_time = task.current_time;
1452
 
 
1453
 
  if(*mandos_client_exited and *password_is_read){
1454
 
 
1455
 
    const size_t send_buffer_length = password->length + 2;
1456
 
    char *send_buffer = malloc(send_buffer_length);
1457
 
    if(send_buffer == NULL){
1458
 
      error(0, errno, "Failed to allocate send_buffer");
1459
 
    } else {
1460
 
      if(mlock(send_buffer, send_buffer_length) != 0){
1461
 
        /* Warn but do not treat as fatal error */
1462
 
        if(errno != EPERM and errno != ENOMEM){
1463
 
          error(0, errno, "Failed to lock memory for password"
1464
 
                " buffer");
1465
 
        }
1466
 
      }
1467
 
      /* “[…] send a single datagram to the socket consisting of the
1468
 
         password string either prefixed with "+" or with "-"
1469
 
         depending on whether the password entry was successful or
1470
 
         not. You may but don't have to include a final NUL byte in
1471
 
         your message.
1472
 
 
1473
 
         — <https://www.freedesktop.org/wiki/Software/systemd/
1474
 
         PasswordAgents/> (Wed 08 Oct 2014 02:14:28 AM UTC)
1475
 
      */
1476
 
      send_buffer[0] = '+';     /* Prefix with "+" */
1477
 
      /* Always add an extra NUL */
1478
 
      send_buffer[password->length + 1] = '\0';
1479
 
      if(password->length > 0){
1480
 
        memcpy(send_buffer + 1, password->data, password->length);
1481
 
      }
1482
 
      errno = 0;
1483
 
      ssize_t ssret = send(fd, send_buffer, send_buffer_length,
1484
 
                           MSG_NOSIGNAL);
1485
 
      const error_t saved_errno = errno;
1486
 
#if defined(__GLIBC_PREREQ) and __GLIBC_PREREQ(2, 25)
1487
 
      explicit_bzero(send_buffer, send_buffer_length);
1488
 
#else
1489
 
      memset(send_buffer, '\0', send_buffer_length);
1490
 
#endif
1491
 
      if(munlock(send_buffer, send_buffer_length) != 0){
1492
 
        error(0, errno, "Failed to unlock memory of send buffer");
1493
 
      }
1494
 
      free(send_buffer);
1495
 
      if(ssret < 0 or ssret < (ssize_t)send_buffer_length){
1496
 
        switch(saved_errno){
1497
 
        case EINTR:
1498
 
        case ENOBUFS:
1499
 
        case ENOMEM:
1500
 
        case EADDRINUSE:
1501
 
        case ECONNREFUSED:
1502
 
        case ECONNRESET:
1503
 
        case ENOENT:
1504
 
        case ETOOMANYREFS:
1505
 
        case EAGAIN:
1506
 
          /* Retry, below */
1507
 
          break;
1508
 
        case EMSGSIZE:
1509
 
          error(0, 0, "Password of size %" PRIuMAX " is too big",
1510
 
                (uintmax_t)password->length);
1511
 
#if __GNUC__ < 7
1512
 
          /* FALLTHROUGH */
1513
 
#else
1514
 
          __attribute__((fallthrough));
1515
 
#endif
1516
 
        case 0:
1517
 
          if(ssret >= 0 and ssret < (ssize_t)send_buffer_length){
1518
 
            error(0, 0, "Password only partially sent to socket");
1519
 
          }
1520
 
#if __GNUC__ < 7
1521
 
          /* FALLTHROUGH */
1522
 
#else
1523
 
          __attribute__((fallthrough));
1524
 
#endif
1525
 
        default:
1526
 
          error(0, saved_errno, "Failed to send() to socket %s",
1527
 
                filename);
1528
 
          if(not string_set_add(cancelled_filenames,
1529
 
                                question_filename)){
1530
 
            error(0, errno, "Failed to cancel question for file %s",
1531
 
                  question_filename);
1532
 
          }
1533
 
          cleanup_task(&task);
1534
 
          return;
1535
 
        }
1536
 
      } else {
1537
 
        /* Success */
1538
 
        cleanup_task(&task);
1539
 
        return;
1540
 
      }
1541
 
    }
1542
 
  }
1543
 
 
1544
 
  /* We failed or are not ready yet; retry later */
1545
 
 
1546
 
  if(not add_to_queue(queue, task)){
1547
 
    error(0, errno, "Failed to add send_password_to_socket for"
1548
 
          " file %s and socket %s to queue", question_filename,
1549
 
          filename);
1550
 
    cleanup_task(&task);
1551
 
  }
1552
 
 
1553
 
  /* Add the fd to the epoll set */
1554
 
  if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
1555
 
               &(struct epoll_event){ .events=EPOLLOUT })
1556
 
     != 0){
1557
 
    error(0, errno, "Failed to add socket file descriptor %d for"
1558
 
          " socket %s to epoll set", fd, filename);
1559
 
    /* Force the added task (send_password_to_socket) to run again, at
1560
 
       most one second from now */
1561
 
    if((queue->next_run == 0)
1562
 
       or (queue->next_run > (*current_time + 1000000))){
1563
 
      queue->next_run = *current_time + 1000000;
1564
 
    }
1565
 
  }
1566
 
}
1567
 
 
1568
 
__attribute__((warn_unused_result))
1569
 
bool add_existing_questions(task_queue *const queue,
1570
 
                            const int epoll_fd,
1571
 
                            buffer *const password,
1572
 
                            string_set *cancelled_filenames,
1573
 
                            const mono_microsecs *const current_time,
1574
 
                            bool *const mandos_client_exited,
1575
 
                            bool *const password_is_read,
1576
 
                            const char *const dirname){
1577
 
  __attribute__((cleanup(cleanup_string)))
1578
 
    char *dir_pattern = NULL;
1579
 
  const int ret = asprintf(&dir_pattern, "%s/ask.*", dirname);
1580
 
  if(ret < 0 or dir_pattern == NULL){
1581
 
    error(0, errno, "Could not create glob pattern for directory %s",
1582
 
          dirname);
1583
 
    return false;
1584
 
  }
1585
 
  __attribute__((cleanup(globfree)))
1586
 
    glob_t question_filenames = {};
1587
 
  switch(glob(dir_pattern, GLOB_ERR | GLOB_NOSORT | GLOB_MARK,
1588
 
              NULL, &question_filenames)){
1589
 
  case GLOB_ABORTED:
1590
 
  default:
1591
 
    error(0, errno, "Failed to open directory %s", dirname);
1592
 
    return false;
1593
 
  case GLOB_NOMATCH:
1594
 
    error(0, errno, "There are no question files in %s", dirname);
1595
 
    return false;
1596
 
  case GLOB_NOSPACE:
1597
 
    error(0, errno, "Could not allocate memory for question file"
1598
 
          " names in %s", dirname);
1599
 
#if __GNUC__ < 7
1600
 
    /* FALLTHROUGH */
1601
 
#else
1602
 
    __attribute__((fallthrough));
1603
 
#endif
1604
 
  case 0:
1605
 
    for(size_t i = 0; i < question_filenames.gl_pathc; i++){
1606
 
      char *const question_filename = strdup(question_filenames
1607
 
                                             .gl_pathv[i]);
1608
 
      const task_context task = {
1609
 
        .func=open_and_parse_question,
1610
 
        .epoll_fd=epoll_fd,
1611
 
        .question_filename=question_filename,
1612
 
        .filename=question_filename,
1613
 
        .password=password,
1614
 
        .cancelled_filenames=cancelled_filenames,
1615
 
        .current_time=current_time,
1616
 
        .mandos_client_exited=mandos_client_exited,
1617
 
        .password_is_read=password_is_read,
1618
 
      };
1619
 
 
1620
 
      if(question_filename == NULL
1621
 
         or not add_to_queue(queue, task)){
1622
 
        error(0, errno, "Failed to add open_and_parse_question for"
1623
 
              " file %s to queue",
1624
 
              question_filenames.gl_pathv[i]);
1625
 
        free(question_filename);
1626
 
      } else {
1627
 
        queue->next_run = 1;
1628
 
      }
1629
 
    }
1630
 
    return true;
1631
 
  }
1632
 
}
1633
 
 
1634
 
__attribute__((nonnull, warn_unused_result))
1635
 
bool wait_for_event(const int epoll_fd,
1636
 
                    const mono_microsecs queue_next_run,
1637
 
                    const mono_microsecs current_time){
1638
 
  __attribute__((const))
1639
 
    int milliseconds_to_wait(const mono_microsecs currtime,
1640
 
                             const mono_microsecs nextrun){
1641
 
    if(currtime >= nextrun){
1642
 
      return 0;
1643
 
    }
1644
 
    const uintmax_t wait_time_ms = (nextrun - currtime) / 1000;
1645
 
    if(wait_time_ms > (uintmax_t)INT_MAX){
1646
 
      return INT_MAX;
1647
 
    }
1648
 
    return (int)wait_time_ms;
1649
 
  }
1650
 
 
1651
 
  const int wait_time_ms = milliseconds_to_wait(current_time,
1652
 
                                                queue_next_run);
1653
 
 
1654
 
  /* Prepare unblocking of SIGCHLD during epoll_pwait */
1655
 
  sigset_t temporary_unblocked_sigmask;
1656
 
  /* Get current signal mask */
1657
 
  if(pthread_sigmask(-1, NULL, &temporary_unblocked_sigmask) != 0){
1658
 
    return false;
1659
 
  }
1660
 
  /* Remove SIGCHLD from the signal mask */
1661
 
  if(sigdelset(&temporary_unblocked_sigmask, SIGCHLD) != 0){
1662
 
    return false;
1663
 
  }
1664
 
  struct epoll_event events[8]; /* Ignored */
1665
 
  int ret = epoll_pwait(epoll_fd, events,
1666
 
                        sizeof(events) / sizeof(struct epoll_event),
1667
 
                        queue_next_run == 0 ? -1 : (int)wait_time_ms,
1668
 
                        &temporary_unblocked_sigmask);
1669
 
  if(ret < 0 and errno != EINTR){
1670
 
    error(0, errno, "Failed epoll_pwait(epfd=%d, ..., timeout=%d,"
1671
 
          " ...", epoll_fd,
1672
 
          queue_next_run == 0 ? -1 : (int)wait_time_ms);
1673
 
    return false;
1674
 
  }
1675
 
  return clear_all_fds_from_epoll_set(epoll_fd);
1676
 
}
1677
 
 
1678
 
bool clear_all_fds_from_epoll_set(const int epoll_fd){
1679
 
  /* Create a new empty epoll set */
1680
 
  __attribute__((cleanup(cleanup_close)))
1681
 
    const int new_epoll_fd = epoll_create1(EPOLL_CLOEXEC);
1682
 
  if(new_epoll_fd < 0){
1683
 
    return false;
1684
 
  }
1685
 
  /* dup3() the new epoll set fd over the old one, replacing it */
1686
 
  if(dup3(new_epoll_fd, epoll_fd, O_CLOEXEC) < 0){
1687
 
    return false;
1688
 
  }
1689
 
  return true;
1690
 
}
1691
 
 
1692
 
__attribute__((nonnull, warn_unused_result))
1693
 
bool run_queue(task_queue **const queue,
1694
 
               string_set *const cancelled_filenames,
1695
 
               bool *const quit_now){
1696
 
 
1697
 
  task_queue *new_queue = create_queue();
1698
 
  if(new_queue == NULL){
1699
 
    return false;
1700
 
  }
1701
 
 
1702
 
  __attribute__((cleanup(string_set_clear)))
1703
 
    string_set old_cancelled_filenames = {};
1704
 
  string_set_swap(cancelled_filenames, &old_cancelled_filenames);
1705
 
 
1706
 
  /* Declare i outside the for loop, since we might need i after the
1707
 
     loop in case we aborted in the middle */
1708
 
  size_t i;
1709
 
  for(i=0; i < (*queue)->length and not *quit_now; i++){
1710
 
    task_context *const task = &((*queue)->tasks[i]);
1711
 
    const char *const question_filename = task->question_filename;
1712
 
    /* Skip any task referencing a cancelled question filename */
1713
 
    if(question_filename != NULL
1714
 
       and string_set_contains(old_cancelled_filenames,
1715
 
                               question_filename)){
1716
 
      cleanup_task(task);
1717
 
      continue;
1718
 
    }
1719
 
    task->func(*task, new_queue);
1720
 
  }
1721
 
 
1722
 
  if(*quit_now){
1723
 
    /* we might be in the middle of the queue, so clean up any
1724
 
       remaining tasks in the current queue */
1725
 
    for(; i < (*queue)->length; i++){
1726
 
      cleanup_task(&((*queue)->tasks[i]));
1727
 
    }
1728
 
    free_queue(*queue);
1729
 
    *queue = new_queue;
1730
 
    new_queue = NULL;
1731
 
    return false;
1732
 
  }
1733
 
  free_queue(*queue);
1734
 
  *queue = new_queue;
1735
 
  new_queue = NULL;
1736
 
 
1737
 
  return true;
1738
 
}
1739
 
 
1740
 
/* End of regular code section */
1741
 
 
1742
 
/* Start of tests section; here are the tests for the above code */
1743
 
 
1744
 
/* This "fixture" data structure is used by the test setup and
1745
 
   teardown functions */
1746
 
typedef struct {
1747
 
  struct sigaction orig_sigaction;
1748
 
  sigset_t orig_sigmask;
1749
 
} test_fixture;
1750
 
 
1751
 
static void test_setup(test_fixture *fixture,
1752
 
                       __attribute__((unused))
1753
 
                       gconstpointer user_data){
1754
 
  g_assert_true(setup_signal_handler(&fixture->orig_sigaction));
1755
 
  g_assert_true(block_sigchld(&fixture->orig_sigmask));
1756
 
}
1757
 
 
1758
 
static void test_teardown(test_fixture *fixture,
1759
 
                          __attribute__((unused))
1760
 
                          gconstpointer user_data){
1761
 
  g_assert_true(restore_signal_handler(&fixture->orig_sigaction));
1762
 
  g_assert_true(restore_sigmask(&fixture->orig_sigmask));
1763
 
}
1764
 
 
1765
 
/* Utility function used by tests to search queue for matching task */
1766
 
__attribute__((pure, nonnull, warn_unused_result))
1767
 
static task_context *find_matching_task(const task_queue *const queue,
1768
 
                                        const task_context task){
1769
 
  /* The argument "task" structure is a pattern to match; 0 in any
1770
 
     member means any value matches, otherwise the value must match.
1771
 
     The filename strings are compared by strcmp(), not by pointer. */
1772
 
  for(size_t i = 0; i < queue->length; i++){
1773
 
    task_context *const current_task = queue->tasks+i;
1774
 
    /* Check all members of task_context, if set to a non-zero value.
1775
 
       If a member does not match, continue to next task in queue */
1776
 
 
1777
 
    /* task_func *const func */
1778
 
    if(task.func != NULL and current_task->func != task.func){
1779
 
      continue;
1780
 
    }
1781
 
    /* char *const question_filename; */
1782
 
    if(task.question_filename != NULL
1783
 
       and (current_task->question_filename == NULL
1784
 
            or strcmp(current_task->question_filename,
1785
 
                      task.question_filename) != 0)){
1786
 
      continue;
1787
 
    }
1788
 
    /* const pid_t pid; */
1789
 
    if(task.pid != 0 and current_task->pid != task.pid){
1790
 
      continue;
1791
 
    }
1792
 
    /* const int epoll_fd; */
1793
 
    if(task.epoll_fd != 0
1794
 
       and current_task->epoll_fd != task.epoll_fd){
1795
 
      continue;
1796
 
    }
1797
 
    /* bool *const quit_now; */
1798
 
    if(task.quit_now != NULL
1799
 
       and current_task->quit_now != task.quit_now){
1800
 
      continue;
1801
 
    }
1802
 
    /* const int fd; */
1803
 
    if(task.fd != 0 and current_task->fd != task.fd){
1804
 
      continue;
1805
 
    }
1806
 
    /* bool *const mandos_client_exited; */
1807
 
    if(task.mandos_client_exited != NULL
1808
 
       and current_task->mandos_client_exited
1809
 
       != task.mandos_client_exited){
1810
 
      continue;
1811
 
    }
1812
 
    /* buffer *const password; */
1813
 
    if(task.password != NULL
1814
 
       and current_task->password != task.password){
1815
 
      continue;
1816
 
    }
1817
 
    /* bool *const password_is_read; */
1818
 
    if(task.password_is_read != NULL
1819
 
       and current_task->password_is_read != task.password_is_read){
1820
 
      continue;
1821
 
    }
1822
 
    /* char *filename; */
1823
 
    if(task.filename != NULL
1824
 
       and (current_task->filename == NULL
1825
 
            or strcmp(current_task->filename, task.filename) != 0)){
1826
 
      continue;
1827
 
    }
1828
 
    /* string_set *const cancelled_filenames; */
1829
 
    if(task.cancelled_filenames != NULL
1830
 
       and current_task->cancelled_filenames
1831
 
       != task.cancelled_filenames){
1832
 
      continue;
1833
 
    }
1834
 
    /* const mono_microsecs notafter; */
1835
 
    if(task.notafter != 0
1836
 
       and current_task->notafter != task.notafter){
1837
 
      continue;
1838
 
    }
1839
 
    /* const mono_microsecs *const current_time; */
1840
 
    if(task.current_time != NULL
1841
 
       and current_task->current_time != task.current_time){
1842
 
      continue;
1843
 
    }
1844
 
    /* Current task matches all members; return it */
1845
 
    return current_task;
1846
 
  }
1847
 
  /* No task in queue matches passed pattern task */
1848
 
  return NULL;
1849
 
}
1850
 
 
1851
 
static void test_create_queue(__attribute__((unused))
1852
 
                              test_fixture *fixture,
1853
 
                              __attribute__((unused))
1854
 
                              gconstpointer user_data){
1855
 
  __attribute__((cleanup(cleanup_queue)))
1856
 
    task_queue *const queue = create_queue();
1857
 
  g_assert_nonnull(queue);
1858
 
  g_assert_null(queue->tasks);
1859
 
  g_assert_true(queue->length == 0);
1860
 
  g_assert_true(queue->next_run == 0);
1861
 
}
1862
 
 
1863
 
static task_func dummy_func;
1864
 
 
1865
 
static void test_add_to_queue(__attribute__((unused))
1866
 
                              test_fixture *fixture,
1867
 
                              __attribute__((unused))
1868
 
                              gconstpointer user_data){
1869
 
  __attribute__((cleanup(cleanup_queue)))
1870
 
    task_queue *queue = create_queue();
1871
 
  g_assert_nonnull(queue);
1872
 
 
1873
 
  g_assert_true(add_to_queue(queue,
1874
 
                             (task_context){ .func=dummy_func }));
1875
 
  g_assert_true(queue->length == 1);
1876
 
  g_assert_nonnull(queue->tasks);
1877
 
  g_assert_true(queue->tasks[0].func == dummy_func);
1878
 
}
1879
 
 
1880
 
static void dummy_func(__attribute__((unused))
1881
 
                       const task_context task,
1882
 
                       __attribute__((unused))
1883
 
                       task_queue *const queue){
1884
 
}
1885
 
 
1886
 
static void test_queue_has_question_empty(__attribute__((unused))
1887
 
                                          test_fixture *fixture,
1888
 
                                          __attribute__((unused))
1889
 
                                          gconstpointer user_data){
1890
 
  __attribute__((cleanup(cleanup_queue)))
1891
 
    task_queue *queue = create_queue();
1892
 
  g_assert_nonnull(queue);
1893
 
  g_assert_false(queue_has_question(queue));
1894
 
}
1895
 
 
1896
 
static void test_queue_has_question_false(__attribute__((unused))
1897
 
                                          test_fixture *fixture,
1898
 
                                          __attribute__((unused))
1899
 
                                          gconstpointer user_data){
1900
 
  __attribute__((cleanup(cleanup_queue)))
1901
 
    task_queue *queue = create_queue();
1902
 
  g_assert_nonnull(queue);
1903
 
  g_assert_true(add_to_queue(queue,
1904
 
                             (task_context){ .func=dummy_func }));
1905
 
  g_assert_false(queue_has_question(queue));
1906
 
}
1907
 
 
1908
 
static void test_queue_has_question_true(__attribute__((unused))
1909
 
                                         test_fixture *fixture,
1910
 
                                         __attribute__((unused))
1911
 
                                         gconstpointer user_data){
1912
 
  __attribute__((cleanup(cleanup_queue)))
1913
 
    task_queue *queue = create_queue();
1914
 
  g_assert_nonnull(queue);
1915
 
  char *const question_filename
1916
 
    = strdup("/nonexistent/question_filename");
1917
 
  g_assert_nonnull(question_filename);
1918
 
  task_context task = {
1919
 
    .func=dummy_func,
1920
 
    .question_filename=question_filename,
1921
 
  };
1922
 
  g_assert_true(add_to_queue(queue, task));
1923
 
  g_assert_true(queue_has_question(queue));
1924
 
}
1925
 
 
1926
 
static void test_queue_has_question_false2(__attribute__((unused))
1927
 
                                           test_fixture *fixture,
1928
 
                                           __attribute__((unused))
1929
 
                                           gconstpointer user_data){
1930
 
  __attribute__((cleanup(cleanup_queue)))
1931
 
    task_queue *queue = create_queue();
1932
 
  g_assert_nonnull(queue);
1933
 
  task_context task = { .func=dummy_func };
1934
 
  g_assert_true(add_to_queue(queue, task));
1935
 
  g_assert_true(add_to_queue(queue, task));
1936
 
  g_assert_cmpint((int)queue->length, ==, 2);
1937
 
  g_assert_false(queue_has_question(queue));
1938
 
}
1939
 
 
1940
 
static void test_queue_has_question_true2(__attribute__((unused))
1941
 
                                          test_fixture *fixture,
1942
 
                                          __attribute__((unused))
1943
 
                                          gconstpointer user_data){
1944
 
  __attribute__((cleanup(cleanup_queue)))
1945
 
    task_queue *queue = create_queue();
1946
 
  g_assert_nonnull(queue);
1947
 
  task_context task1 = { .func=dummy_func };
1948
 
  g_assert_true(add_to_queue(queue, task1));
1949
 
  char *const question_filename
1950
 
    = strdup("/nonexistent/question_filename");
1951
 
  g_assert_nonnull(question_filename);
1952
 
  task_context task2 = {
1953
 
    .func=dummy_func,
1954
 
    .question_filename=question_filename,
1955
 
  };
1956
 
  g_assert_true(add_to_queue(queue, task2));
1957
 
  g_assert_cmpint((int)queue->length, ==, 2);
1958
 
  g_assert_true(queue_has_question(queue));
1959
 
}
1960
 
 
1961
 
static void test_cleanup_buffer(__attribute__((unused))
1962
 
                                test_fixture *fixture,
1963
 
                                __attribute__((unused))
1964
 
                                gconstpointer user_data){
1965
 
  buffer buf = {};
1966
 
 
1967
 
  const size_t buffersize = 10;
1968
 
 
1969
 
  buf.data = malloc(buffersize);
1970
 
  g_assert_nonnull(buf.data);
1971
 
  if(mlock(buf.data, buffersize) != 0){
1972
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
1973
 
  }
1974
 
 
1975
 
  cleanup_buffer(&buf);
1976
 
  g_assert_null(buf.data);
1977
 
}
1978
 
 
1979
 
static
1980
 
void test_string_set_new_set_contains_nothing(__attribute__((unused))
1981
 
                                              test_fixture *fixture,
1982
 
                                              __attribute__((unused))
1983
 
                                              gconstpointer
1984
 
                                              user_data){
1985
 
  __attribute__((cleanup(string_set_clear)))
1986
 
    string_set set = {};
1987
 
  g_assert_false(string_set_contains(set, "")); /* Empty string */
1988
 
  g_assert_false(string_set_contains(set, "test_string"));
1989
 
}
1990
 
 
1991
 
static void
1992
 
test_string_set_with_added_string_contains_it(__attribute__((unused))
1993
 
                                              test_fixture *fixture,
1994
 
                                              __attribute__((unused))
1995
 
                                              gconstpointer
1996
 
                                              user_data){
1997
 
  __attribute__((cleanup(string_set_clear)))
1998
 
    string_set set = {};
1999
 
  g_assert_true(string_set_add(&set, "test_string"));
2000
 
  g_assert_true(string_set_contains(set, "test_string"));
2001
 
}
2002
 
 
2003
 
static void
2004
 
test_string_set_cleared_does_not_contain_str(__attribute__((unused))
2005
 
                                             test_fixture *fixture,
2006
 
                                             __attribute__((unused))
2007
 
                                             gconstpointer user_data){
2008
 
  __attribute__((cleanup(string_set_clear)))
2009
 
    string_set set = {};
2010
 
  g_assert_true(string_set_add(&set, "test_string"));
2011
 
  string_set_clear(&set);
2012
 
  g_assert_false(string_set_contains(set, "test_string"));
2013
 
}
2014
 
 
2015
 
static
2016
 
void test_string_set_swap_one_with_empty(__attribute__((unused))
2017
 
                                         test_fixture *fixture,
2018
 
                                         __attribute__((unused))
2019
 
                                         gconstpointer user_data){
2020
 
  __attribute__((cleanup(string_set_clear)))
2021
 
    string_set set1 = {};
2022
 
  __attribute__((cleanup(string_set_clear)))
2023
 
    string_set set2 = {};
2024
 
  g_assert_true(string_set_add(&set1, "test_string1"));
2025
 
  string_set_swap(&set1, &set2);
2026
 
  g_assert_false(string_set_contains(set1, "test_string1"));
2027
 
  g_assert_true(string_set_contains(set2, "test_string1"));
2028
 
}
2029
 
 
2030
 
static
2031
 
void test_string_set_swap_empty_with_one(__attribute__((unused))
2032
 
                                         test_fixture *fixture,
2033
 
                                         __attribute__((unused))
2034
 
                                         gconstpointer user_data){
2035
 
  __attribute__((cleanup(string_set_clear)))
2036
 
    string_set set1 = {};
2037
 
  __attribute__((cleanup(string_set_clear)))
2038
 
    string_set set2 = {};
2039
 
  g_assert_true(string_set_add(&set2, "test_string2"));
2040
 
  string_set_swap(&set1, &set2);
2041
 
  g_assert_true(string_set_contains(set1, "test_string2"));
2042
 
  g_assert_false(string_set_contains(set2, "test_string2"));
2043
 
}
2044
 
 
2045
 
static void test_string_set_swap_one_with_one(__attribute__((unused))
2046
 
                                              test_fixture *fixture,
2047
 
                                              __attribute__((unused))
2048
 
                                              gconstpointer
2049
 
                                              user_data){
2050
 
  __attribute__((cleanup(string_set_clear)))
2051
 
    string_set set1 = {};
2052
 
  __attribute__((cleanup(string_set_clear)))
2053
 
    string_set set2 = {};
2054
 
  g_assert_true(string_set_add(&set1, "test_string1"));
2055
 
  g_assert_true(string_set_add(&set2, "test_string2"));
2056
 
  string_set_swap(&set1, &set2);
2057
 
  g_assert_false(string_set_contains(set1, "test_string1"));
2058
 
  g_assert_true(string_set_contains(set1, "test_string2"));
2059
 
  g_assert_false(string_set_contains(set2, "test_string2"));
2060
 
  g_assert_true(string_set_contains(set2, "test_string1"));
2061
 
}
2062
 
 
2063
 
static bool fd_has_cloexec_and_nonblock(const int);
2064
 
 
2065
 
static bool epoll_set_contains(int, int, uint32_t);
2066
 
 
2067
 
static void test_start_mandos_client(test_fixture *fixture,
2068
 
                                     __attribute__((unused))
2069
 
                                     gconstpointer user_data){
2070
 
 
2071
 
  bool mandos_client_exited = false;
2072
 
  bool quit_now = false;
2073
 
  __attribute__((cleanup(cleanup_close)))
2074
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2075
 
  g_assert_cmpint(epoll_fd, >=, 0);
2076
 
  __attribute__((cleanup(cleanup_queue)))
2077
 
    task_queue *queue = create_queue();
2078
 
  g_assert_nonnull(queue);
2079
 
  buffer password = {};
2080
 
  bool password_is_read = false;
2081
 
  const char helper_directory[] = "/nonexistent";
2082
 
  const char *const argv[] = { "/bin/true", NULL };
2083
 
 
2084
 
  g_assert_true(start_mandos_client(queue, epoll_fd,
2085
 
                                    &mandos_client_exited, &quit_now,
2086
 
                                    &password, &password_is_read,
2087
 
                                    &fixture->orig_sigaction,
2088
 
                                    fixture->orig_sigmask,
2089
 
                                    helper_directory, 0, 0, argv));
2090
 
 
2091
 
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
2092
 
 
2093
 
  const task_context *const added_wait_task
2094
 
    = find_matching_task(queue, (task_context){
2095
 
        .func=wait_for_mandos_client_exit,
2096
 
        .mandos_client_exited=&mandos_client_exited,
2097
 
        .quit_now=&quit_now,
2098
 
      });
2099
 
  g_assert_nonnull(added_wait_task);
2100
 
  g_assert_cmpint(added_wait_task->pid, >, 0);
2101
 
  g_assert_cmpint(kill(added_wait_task->pid, SIGKILL), ==, 0);
2102
 
  waitpid(added_wait_task->pid, NULL, 0);
2103
 
 
2104
 
  const task_context *const added_read_task
2105
 
    = find_matching_task(queue, (task_context){
2106
 
        .func=read_mandos_client_output,
2107
 
        .epoll_fd=epoll_fd,
2108
 
        .password=&password,
2109
 
        .password_is_read=&password_is_read,
2110
 
        .quit_now=&quit_now,
2111
 
      });
2112
 
  g_assert_nonnull(added_read_task);
2113
 
  g_assert_cmpint(added_read_task->fd, >, 2);
2114
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
2115
 
  g_assert_true(epoll_set_contains(epoll_fd, added_read_task->fd,
2116
 
                                   EPOLLIN | EPOLLRDHUP));
2117
 
}
2118
 
 
2119
 
static bool fd_has_cloexec_and_nonblock(const int fd){
2120
 
  const int socket_fd_flags = fcntl(fd, F_GETFD, 0);
2121
 
  const int socket_file_flags = fcntl(fd, F_GETFL, 0);
2122
 
  return ((socket_fd_flags >= 0)
2123
 
          and (socket_fd_flags & FD_CLOEXEC)
2124
 
          and (socket_file_flags >= 0)
2125
 
          and (socket_file_flags & O_NONBLOCK));
2126
 
}
2127
 
 
2128
 
__attribute__((const))
2129
 
bool is_privileged(void){
2130
 
  uid_t user = getuid() + 1;
2131
 
  if(user == 0){                /* Overflow check */
2132
 
    user++;
2133
 
  }
2134
 
  gid_t group = getuid() + 1;
2135
 
  if(group == 0){               /* Overflow check */
2136
 
    group++;
2137
 
  }
2138
 
  const pid_t pid = fork();
2139
 
  if(pid == 0){                 /* Child */
2140
 
    if(setresgid((uid_t)-1, group, group) == -1){
2141
 
      if(errno != EPERM){
2142
 
        error(EXIT_FAILURE, errno, "Failed to setresgid(-1, %" PRIuMAX
2143
 
              ", %" PRIuMAX")", (uintmax_t)group, (uintmax_t)group);
2144
 
      }
2145
 
      exit(EXIT_FAILURE);
2146
 
    }
2147
 
    if(setresuid((uid_t)-1, user, user) == -1){
2148
 
      if(errno != EPERM){
2149
 
        error(EXIT_FAILURE, errno, "Failed to setresuid(-1, %" PRIuMAX
2150
 
              ", %" PRIuMAX")", (uintmax_t)user, (uintmax_t)user);
2151
 
      }
2152
 
      exit(EXIT_FAILURE);
2153
 
    }
2154
 
    exit(EXIT_SUCCESS);
2155
 
  }
2156
 
  int status;
2157
 
  waitpid(pid, &status, 0);
2158
 
  if(WIFEXITED(status) and (WEXITSTATUS(status) == EXIT_SUCCESS)){
2159
 
    return true;
2160
 
  }
2161
 
  return false;
2162
 
}
2163
 
 
2164
 
static bool epoll_set_contains(int epoll_fd, int fd, uint32_t events){
2165
 
  /* Only scan for events in this eventmask */
2166
 
  const uint32_t eventmask = EPOLLIN | EPOLLOUT | EPOLLRDHUP;
2167
 
  __attribute__((cleanup(cleanup_string)))
2168
 
    char *fdinfo_name = NULL;
2169
 
  int ret = asprintf(&fdinfo_name, "/proc/self/fdinfo/%d", epoll_fd);
2170
 
  g_assert_cmpint(ret, >, 0);
2171
 
  g_assert_nonnull(fdinfo_name);
2172
 
 
2173
 
  FILE *fdinfo = fopen(fdinfo_name, "r");
2174
 
  g_assert_nonnull(fdinfo);
2175
 
  uint32_t reported_events;
2176
 
  buffer line = {};
2177
 
  int found_fd = -1;
2178
 
 
2179
 
  do {
2180
 
    if(getline(&line.data, &line.allocated, fdinfo) < 0){
2181
 
      break;
2182
 
    }
2183
 
    /* See proc(5) for format of /proc/PID/fdinfo/FD for epoll fd's */
2184
 
    if(sscanf(line.data, "tfd: %d events: %" SCNx32 " ",
2185
 
              &found_fd, &reported_events) == 2){
2186
 
      if(found_fd == fd){
2187
 
        break;
2188
 
      }
2189
 
    }
2190
 
  } while(not feof(fdinfo) and not ferror(fdinfo));
2191
 
  g_assert_cmpint(fclose(fdinfo), ==, 0);
2192
 
  free(line.data);
2193
 
  if(found_fd != fd){
2194
 
    return false;
2195
 
  }
2196
 
 
2197
 
  if(events == 0){
2198
 
    /* Don't check events if none are given */
2199
 
    return true;
2200
 
  }
2201
 
  return (reported_events & eventmask) == (events & eventmask);
2202
 
}
2203
 
 
2204
 
static void test_start_mandos_client_execv(test_fixture *fixture,
2205
 
                                           __attribute__((unused))
2206
 
                                           gconstpointer user_data){
2207
 
  bool mandos_client_exited = false;
2208
 
  bool quit_now = false;
2209
 
  __attribute__((cleanup(cleanup_close)))
2210
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2211
 
  g_assert_cmpint(epoll_fd, >=, 0);
2212
 
  __attribute__((cleanup(cleanup_queue)))
2213
 
    task_queue *queue = create_queue();
2214
 
  g_assert_nonnull(queue);
2215
 
  __attribute__((cleanup(cleanup_buffer)))
2216
 
    buffer password = {};
2217
 
  const char helper_directory[] = "/nonexistent";
2218
 
  /* Can't execv("/", ...), so this should fail */
2219
 
  const char *const argv[] = { "/", NULL };
2220
 
 
2221
 
  {
2222
 
    __attribute__((cleanup(cleanup_close)))
2223
 
      const int devnull_fd = open("/dev/null", O_WRONLY | O_CLOEXEC);
2224
 
    g_assert_cmpint(devnull_fd, >=, 0);
2225
 
    __attribute__((cleanup(cleanup_close)))
2226
 
      const int real_stderr_fd = dup(STDERR_FILENO);
2227
 
    g_assert_cmpint(real_stderr_fd, >=, 0);
2228
 
    dup2(devnull_fd, STDERR_FILENO);
2229
 
 
2230
 
    const bool success = start_mandos_client(queue, epoll_fd,
2231
 
                                             &mandos_client_exited,
2232
 
                                             &quit_now,
2233
 
                                             &password,
2234
 
                                             (bool[]){false},
2235
 
                                             &fixture->orig_sigaction,
2236
 
                                             fixture->orig_sigmask,
2237
 
                                             helper_directory, 0, 0,
2238
 
                                             argv);
2239
 
    dup2(real_stderr_fd, STDERR_FILENO);
2240
 
    g_assert_true(success);
2241
 
  }
2242
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 2);
2243
 
 
2244
 
  struct timespec starttime, currtime;
2245
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2246
 
  do {
2247
 
    queue->next_run = 0;
2248
 
    string_set cancelled_filenames = {};
2249
 
 
2250
 
    {
2251
 
      __attribute__((cleanup(cleanup_close)))
2252
 
        const int devnull_fd = open("/dev/null",
2253
 
                                    O_WRONLY | O_CLOEXEC);
2254
 
      g_assert_cmpint(devnull_fd, >=, 0);
2255
 
      __attribute__((cleanup(cleanup_close)))
2256
 
        const int real_stderr_fd = dup(STDERR_FILENO);
2257
 
      g_assert_cmpint(real_stderr_fd, >=, 0);
2258
 
      g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2259
 
      dup2(devnull_fd, STDERR_FILENO);
2260
 
      const bool success = run_queue(&queue, &cancelled_filenames,
2261
 
                                     &quit_now);
2262
 
      dup2(real_stderr_fd, STDERR_FILENO);
2263
 
      if(not success){
2264
 
        break;
2265
 
      }
2266
 
    }
2267
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2268
 
  } while(((queue->length) > 0)
2269
 
          and (not quit_now)
2270
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2271
 
 
2272
 
  g_assert_true(quit_now);
2273
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2274
 
  g_assert_true(mandos_client_exited);
2275
 
}
2276
 
 
2277
 
static void test_start_mandos_client_suid_euid(test_fixture *fixture,
2278
 
                                               __attribute__((unused))
2279
 
                                               gconstpointer
2280
 
                                               user_data){
2281
 
  if(not is_privileged()){
2282
 
    g_test_skip("Not privileged");
2283
 
    return;
2284
 
  }
2285
 
 
2286
 
  bool mandos_client_exited = false;
2287
 
  bool quit_now = false;
2288
 
  __attribute__((cleanup(cleanup_close)))
2289
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2290
 
  g_assert_cmpint(epoll_fd, >=, 0);
2291
 
  __attribute__((cleanup(cleanup_queue)))
2292
 
    task_queue *queue = create_queue();
2293
 
  g_assert_nonnull(queue);
2294
 
  __attribute__((cleanup(cleanup_buffer)))
2295
 
    buffer password = {};
2296
 
  bool password_is_read = false;
2297
 
  const char helper_directory[] = "/nonexistent";
2298
 
  const char *const argv[] = { "/usr/bin/id", "--user", NULL };
2299
 
  uid_t user = 1000;
2300
 
  gid_t group = 1001;
2301
 
 
2302
 
  const bool success = start_mandos_client(queue, epoll_fd,
2303
 
                                           &mandos_client_exited,
2304
 
                                           &quit_now, &password,
2305
 
                                           &password_is_read,
2306
 
                                           &fixture->orig_sigaction,
2307
 
                                           fixture->orig_sigmask,
2308
 
                                           helper_directory, user,
2309
 
                                           group, argv);
2310
 
  g_assert_true(success);
2311
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2312
 
 
2313
 
  struct timespec starttime, currtime;
2314
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2315
 
  do {
2316
 
    queue->next_run = 0;
2317
 
    string_set cancelled_filenames = {};
2318
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2319
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2320
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2321
 
  } while(((queue->length) > 0)
2322
 
          and (not quit_now)
2323
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2324
 
 
2325
 
  g_assert_false(quit_now);
2326
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2327
 
  g_assert_true(mandos_client_exited);
2328
 
 
2329
 
  g_assert_true(password_is_read);
2330
 
  g_assert_nonnull(password.data);
2331
 
 
2332
 
  uintmax_t id;
2333
 
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
2334
 
                  ==, 1);
2335
 
  g_assert_true((uid_t)id == id);
2336
 
 
2337
 
  g_assert_cmpuint((unsigned int)id, ==, 0);
2338
 
}
2339
 
 
2340
 
static void test_start_mandos_client_suid_egid(test_fixture *fixture,
2341
 
                                               __attribute__((unused))
2342
 
                                               gconstpointer
2343
 
                                               user_data){
2344
 
  if(not is_privileged()){
2345
 
    g_test_skip("Not privileged");
2346
 
    return;
2347
 
  }
2348
 
 
2349
 
  bool mandos_client_exited = false;
2350
 
  bool quit_now = false;
2351
 
  __attribute__((cleanup(cleanup_close)))
2352
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2353
 
  g_assert_cmpint(epoll_fd, >=, 0);
2354
 
  __attribute__((cleanup(cleanup_queue)))
2355
 
    task_queue *queue = create_queue();
2356
 
  g_assert_nonnull(queue);
2357
 
  __attribute__((cleanup(cleanup_buffer)))
2358
 
    buffer password = {};
2359
 
  bool password_is_read = false;
2360
 
  const char helper_directory[] = "/nonexistent";
2361
 
  const char *const argv[] = { "/usr/bin/id", "--group", NULL };
2362
 
  uid_t user = 1000;
2363
 
  gid_t group = 1001;
2364
 
 
2365
 
  const bool success = start_mandos_client(queue, epoll_fd,
2366
 
                                           &mandos_client_exited,
2367
 
                                           &quit_now, &password,
2368
 
                                           &password_is_read,
2369
 
                                           &fixture->orig_sigaction,
2370
 
                                           fixture->orig_sigmask,
2371
 
                                           helper_directory, user,
2372
 
                                           group, argv);
2373
 
  g_assert_true(success);
2374
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2375
 
 
2376
 
  struct timespec starttime, currtime;
2377
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2378
 
  do {
2379
 
    queue->next_run = 0;
2380
 
    string_set cancelled_filenames = {};
2381
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2382
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2383
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2384
 
  } while(((queue->length) > 0)
2385
 
          and (not quit_now)
2386
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2387
 
 
2388
 
  g_assert_false(quit_now);
2389
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2390
 
  g_assert_true(mandos_client_exited);
2391
 
 
2392
 
  g_assert_true(password_is_read);
2393
 
  g_assert_nonnull(password.data);
2394
 
 
2395
 
  uintmax_t id;
2396
 
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
2397
 
                  ==, 1);
2398
 
  g_assert_true((gid_t)id == id);
2399
 
 
2400
 
  g_assert_cmpuint((unsigned int)id, ==, 0);
2401
 
}
2402
 
 
2403
 
static void test_start_mandos_client_suid_ruid(test_fixture *fixture,
2404
 
                                               __attribute__((unused))
2405
 
                                               gconstpointer
2406
 
                                               user_data){
2407
 
  if(not is_privileged()){
2408
 
    g_test_skip("Not privileged");
2409
 
    return;
2410
 
  }
2411
 
 
2412
 
  bool mandos_client_exited = false;
2413
 
  bool quit_now = false;
2414
 
  __attribute__((cleanup(cleanup_close)))
2415
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2416
 
  g_assert_cmpint(epoll_fd, >=, 0);
2417
 
  __attribute__((cleanup(cleanup_queue)))
2418
 
    task_queue *queue = create_queue();
2419
 
  g_assert_nonnull(queue);
2420
 
  __attribute__((cleanup(cleanup_buffer)))
2421
 
    buffer password = {};
2422
 
  bool password_is_read = false;
2423
 
  const char helper_directory[] = "/nonexistent";
2424
 
  const char *const argv[] = { "/usr/bin/id", "--user", "--real",
2425
 
    NULL };
2426
 
  uid_t user = 1000;
2427
 
  gid_t group = 1001;
2428
 
 
2429
 
  const bool success = start_mandos_client(queue, epoll_fd,
2430
 
                                           &mandos_client_exited,
2431
 
                                           &quit_now, &password,
2432
 
                                           &password_is_read,
2433
 
                                           &fixture->orig_sigaction,
2434
 
                                           fixture->orig_sigmask,
2435
 
                                           helper_directory, user,
2436
 
                                           group, argv);
2437
 
  g_assert_true(success);
2438
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2439
 
 
2440
 
  struct timespec starttime, currtime;
2441
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2442
 
  do {
2443
 
    queue->next_run = 0;
2444
 
    string_set cancelled_filenames = {};
2445
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2446
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2447
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2448
 
  } while(((queue->length) > 0)
2449
 
          and (not quit_now)
2450
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2451
 
 
2452
 
  g_assert_false(quit_now);
2453
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2454
 
  g_assert_true(mandos_client_exited);
2455
 
 
2456
 
  g_assert_true(password_is_read);
2457
 
  g_assert_nonnull(password.data);
2458
 
 
2459
 
  uintmax_t id;
2460
 
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
2461
 
                  ==, 1);
2462
 
  g_assert_true((uid_t)id == id);
2463
 
 
2464
 
  g_assert_cmpuint((unsigned int)id, ==, user);
2465
 
}
2466
 
 
2467
 
static void test_start_mandos_client_suid_rgid(test_fixture *fixture,
2468
 
                                               __attribute__((unused))
2469
 
                                               gconstpointer
2470
 
                                               user_data){
2471
 
  if(not is_privileged()){
2472
 
    g_test_skip("Not privileged");
2473
 
    return;
2474
 
  }
2475
 
 
2476
 
  bool mandos_client_exited = false;
2477
 
  bool quit_now = false;
2478
 
  __attribute__((cleanup(cleanup_close)))
2479
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2480
 
  g_assert_cmpint(epoll_fd, >=, 0);
2481
 
  __attribute__((cleanup(cleanup_queue)))
2482
 
    task_queue *queue = create_queue();
2483
 
  g_assert_nonnull(queue);
2484
 
  __attribute__((cleanup(cleanup_buffer)))
2485
 
    buffer password = {};
2486
 
  bool password_is_read = false;
2487
 
  const char helper_directory[] = "/nonexistent";
2488
 
  const char *const argv[] = { "/usr/bin/id", "--group", "--real",
2489
 
    NULL };
2490
 
  uid_t user = 1000;
2491
 
  gid_t group = 1001;
2492
 
 
2493
 
  const bool success = start_mandos_client(queue, epoll_fd,
2494
 
                                           &mandos_client_exited,
2495
 
                                           &quit_now, &password,
2496
 
                                           &password_is_read,
2497
 
                                           &fixture->orig_sigaction,
2498
 
                                           fixture->orig_sigmask,
2499
 
                                           helper_directory, user,
2500
 
                                           group, argv);
2501
 
  g_assert_true(success);
2502
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2503
 
 
2504
 
  struct timespec starttime, currtime;
2505
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2506
 
  do {
2507
 
    queue->next_run = 0;
2508
 
    string_set cancelled_filenames = {};
2509
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2510
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2511
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2512
 
  } while(((queue->length) > 0)
2513
 
          and (not quit_now)
2514
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2515
 
 
2516
 
  g_assert_false(quit_now);
2517
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2518
 
  g_assert_true(mandos_client_exited);
2519
 
 
2520
 
  g_assert_true(password_is_read);
2521
 
  g_assert_nonnull(password.data);
2522
 
 
2523
 
  uintmax_t id;
2524
 
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
2525
 
                  ==, 1);
2526
 
  g_assert_true((gid_t)id == id);
2527
 
 
2528
 
  g_assert_cmpuint((unsigned int)id, ==, group);
2529
 
}
2530
 
 
2531
 
static void test_start_mandos_client_read(test_fixture *fixture,
2532
 
                                          __attribute__((unused))
2533
 
                                          gconstpointer user_data){
2534
 
  bool mandos_client_exited = false;
2535
 
  bool quit_now = false;
2536
 
  __attribute__((cleanup(cleanup_close)))
2537
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2538
 
  g_assert_cmpint(epoll_fd, >=, 0);
2539
 
  __attribute__((cleanup(cleanup_queue)))
2540
 
    task_queue *queue = create_queue();
2541
 
  g_assert_nonnull(queue);
2542
 
  __attribute__((cleanup(cleanup_buffer)))
2543
 
    buffer password = {};
2544
 
  bool password_is_read = false;
2545
 
  const char dummy_test_password[] = "dummy test password";
2546
 
  const char helper_directory[] = "/nonexistent";
2547
 
  const char *const argv[] = { "/bin/echo", "-n", dummy_test_password,
2548
 
    NULL };
2549
 
 
2550
 
  const bool success = start_mandos_client(queue, epoll_fd,
2551
 
                                           &mandos_client_exited,
2552
 
                                           &quit_now, &password,
2553
 
                                           &password_is_read,
2554
 
                                           &fixture->orig_sigaction,
2555
 
                                           fixture->orig_sigmask,
2556
 
                                           helper_directory, 0, 0,
2557
 
                                           argv);
2558
 
  g_assert_true(success);
2559
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2560
 
 
2561
 
  struct timespec starttime, currtime;
2562
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2563
 
  do {
2564
 
    queue->next_run = 0;
2565
 
    string_set cancelled_filenames = {};
2566
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2567
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2568
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2569
 
  } while(((queue->length) > 0)
2570
 
          and (not quit_now)
2571
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2572
 
 
2573
 
  g_assert_false(quit_now);
2574
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2575
 
  g_assert_true(mandos_client_exited);
2576
 
 
2577
 
  g_assert_true(password_is_read);
2578
 
  g_assert_cmpint((int)password.length, ==,
2579
 
                  sizeof(dummy_test_password)-1);
2580
 
  g_assert_nonnull(password.data);
2581
 
  g_assert_cmpint(memcmp(dummy_test_password, password.data,
2582
 
                         sizeof(dummy_test_password)-1), ==, 0);
2583
 
}
2584
 
 
2585
 
static
2586
 
void test_start_mandos_client_helper_directory(test_fixture *fixture,
2587
 
                                               __attribute__((unused))
2588
 
                                               gconstpointer
2589
 
                                               user_data){
2590
 
  bool mandos_client_exited = false;
2591
 
  bool quit_now = false;
2592
 
  __attribute__((cleanup(cleanup_close)))
2593
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2594
 
  g_assert_cmpint(epoll_fd, >=, 0);
2595
 
  __attribute__((cleanup(cleanup_queue)))
2596
 
    task_queue *queue = create_queue();
2597
 
  g_assert_nonnull(queue);
2598
 
  __attribute__((cleanup(cleanup_buffer)))
2599
 
    buffer password = {};
2600
 
  bool password_is_read = false;
2601
 
  const char helper_directory[] = "/nonexistent";
2602
 
  const char *const argv[] = { "/bin/sh", "-c",
2603
 
    "echo -n ${MANDOSPLUGINHELPERDIR}", NULL };
2604
 
 
2605
 
  const bool success = start_mandos_client(queue, epoll_fd,
2606
 
                                           &mandos_client_exited,
2607
 
                                           &quit_now, &password,
2608
 
                                           &password_is_read,
2609
 
                                           &fixture->orig_sigaction,
2610
 
                                           fixture->orig_sigmask,
2611
 
                                           helper_directory, 0, 0,
2612
 
                                           argv);
2613
 
  g_assert_true(success);
2614
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
2615
 
 
2616
 
  struct timespec starttime, currtime;
2617
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2618
 
  do {
2619
 
    queue->next_run = 0;
2620
 
    string_set cancelled_filenames = {};
2621
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2622
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2623
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2624
 
  } while(((queue->length) > 0)
2625
 
          and (not quit_now)
2626
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2627
 
 
2628
 
  g_assert_false(quit_now);
2629
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2630
 
  g_assert_true(mandos_client_exited);
2631
 
 
2632
 
  g_assert_true(password_is_read);
2633
 
  g_assert_cmpint((int)password.length, ==,
2634
 
                  sizeof(helper_directory)-1);
2635
 
  g_assert_nonnull(password.data);
2636
 
  g_assert_cmpint(memcmp(helper_directory, password.data,
2637
 
                         sizeof(helper_directory)-1), ==, 0);
2638
 
}
2639
 
 
2640
 
__attribute__((nonnull, warn_unused_result))
2641
 
static bool proc_status_sigblk_to_sigset(const char *const,
2642
 
                                         sigset_t *const);
2643
 
 
2644
 
static void test_start_mandos_client_sigmask(test_fixture *fixture,
2645
 
                                             __attribute__((unused))
2646
 
                                             gconstpointer user_data){
2647
 
  bool mandos_client_exited = false;
2648
 
  bool quit_now = false;
2649
 
  __attribute__((cleanup(cleanup_close)))
2650
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2651
 
  g_assert_cmpint(epoll_fd, >=, 0);
2652
 
  __attribute__((cleanup(cleanup_queue)))
2653
 
    task_queue *queue = create_queue();
2654
 
  g_assert_nonnull(queue);
2655
 
  __attribute__((cleanup(cleanup_buffer)))
2656
 
    buffer password = {};
2657
 
  bool password_is_read = false;
2658
 
  const char helper_directory[] = "/nonexistent";
2659
 
  /* see proc(5) for format of /proc/self/status */
2660
 
  const char *const argv[] = { "/usr/bin/awk",
2661
 
    "$1==\"SigBlk:\"{ print $2 }", "/proc/self/status", NULL };
2662
 
 
2663
 
  g_assert_true(start_mandos_client(queue, epoll_fd,
2664
 
                                    &mandos_client_exited, &quit_now,
2665
 
                                    &password, &password_is_read,
2666
 
                                    &fixture->orig_sigaction,
2667
 
                                    fixture->orig_sigmask,
2668
 
                                    helper_directory, 0, 0, argv));
2669
 
 
2670
 
  struct timespec starttime, currtime;
2671
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2672
 
  do {
2673
 
    queue->next_run = 0;
2674
 
    string_set cancelled_filenames = {};
2675
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2676
 
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
2677
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2678
 
  } while((not (mandos_client_exited and password_is_read))
2679
 
          and (not quit_now)
2680
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2681
 
  g_assert_true(mandos_client_exited);
2682
 
  g_assert_true(password_is_read);
2683
 
 
2684
 
  sigset_t parsed_sigmask;
2685
 
  g_assert_true(proc_status_sigblk_to_sigset(password.data,
2686
 
                                             &parsed_sigmask));
2687
 
 
2688
 
  for(int signum = 1; signum < NSIG; signum++){
2689
 
    const bool has_signal = sigismember(&parsed_sigmask, signum);
2690
 
    if(sigismember(&fixture->orig_sigmask, signum)){
2691
 
      g_assert_true(has_signal);
2692
 
    } else {
2693
 
      g_assert_false(has_signal);
2694
 
    }
2695
 
  }
2696
 
}
2697
 
 
2698
 
__attribute__((nonnull, warn_unused_result))
2699
 
static bool proc_status_sigblk_to_sigset(const char *const sigblk,
2700
 
                                         sigset_t *const sigmask){
2701
 
  /* parse /proc/PID/status SigBlk value and convert to a sigset_t */
2702
 
  uintmax_t scanned_sigmask;
2703
 
  if(sscanf(sigblk, "%" SCNxMAX " ", &scanned_sigmask) != 1){
2704
 
    return false;
2705
 
  }
2706
 
  if(sigemptyset(sigmask) != 0){
2707
 
    return false;
2708
 
  }
2709
 
  for(int signum = 1; signum < NSIG; signum++){
2710
 
    if(scanned_sigmask & ((uintmax_t)1 << (signum-1))){
2711
 
      if(sigaddset(sigmask, signum) != 0){
2712
 
        return false;
2713
 
      }
2714
 
    }
2715
 
  }
2716
 
  return true;
2717
 
}
2718
 
 
2719
 
static void run_task_with_stderr_to_dev_null(const task_context task,
2720
 
                                             task_queue *const queue);
2721
 
 
2722
 
static
2723
 
void test_wait_for_mandos_client_exit_badpid(__attribute__((unused))
2724
 
                                             test_fixture *fixture,
2725
 
                                             __attribute__((unused))
2726
 
                                             gconstpointer user_data){
2727
 
 
2728
 
  bool mandos_client_exited = false;
2729
 
  bool quit_now = false;
2730
 
 
2731
 
  __attribute__((cleanup(cleanup_queue)))
2732
 
    task_queue *queue = create_queue();
2733
 
  g_assert_nonnull(queue);
2734
 
  const task_context task = {
2735
 
    .func=wait_for_mandos_client_exit,
2736
 
    .pid=1,
2737
 
    .mandos_client_exited=&mandos_client_exited,
2738
 
    .quit_now=&quit_now,
2739
 
  };
2740
 
  run_task_with_stderr_to_dev_null(task, queue);
2741
 
 
2742
 
  g_assert_false(mandos_client_exited);
2743
 
  g_assert_true(quit_now);
2744
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2745
 
}
2746
 
 
2747
 
static void run_task_with_stderr_to_dev_null(const task_context task,
2748
 
                                             task_queue *const queue){
2749
 
  FILE *real_stderr = stderr;
2750
 
  FILE *devnull = fopen("/dev/null", "we");
2751
 
  g_assert_nonnull(devnull);
2752
 
 
2753
 
  stderr = devnull;
2754
 
  task.func(task, queue);
2755
 
  stderr = real_stderr;
2756
 
 
2757
 
  g_assert_cmpint(fclose(devnull), ==, 0);
2758
 
}
2759
 
 
2760
 
static
2761
 
void test_wait_for_mandos_client_exit_noexit(test_fixture *fixture,
2762
 
                                             __attribute__((unused))
2763
 
                                             gconstpointer user_data){
2764
 
  bool mandos_client_exited = false;
2765
 
  bool quit_now = false;
2766
 
 
2767
 
  pid_t create_eternal_process(void){
2768
 
    const pid_t pid = fork();
2769
 
    if(pid == 0){               /* Child */
2770
 
      if(not restore_signal_handler(&fixture->orig_sigaction)){
2771
 
        _exit(EXIT_FAILURE);
2772
 
      }
2773
 
      if(not restore_sigmask(&fixture->orig_sigmask)){
2774
 
        _exit(EXIT_FAILURE);
2775
 
      }
2776
 
      while(true){
2777
 
        pause();
2778
 
      }
2779
 
    }
2780
 
    return pid;
2781
 
  }
2782
 
  pid_t pid = create_eternal_process();
2783
 
  g_assert_true(pid != -1);
2784
 
 
2785
 
  __attribute__((cleanup(cleanup_queue)))
2786
 
    task_queue *queue = create_queue();
2787
 
  g_assert_nonnull(queue);
2788
 
  const task_context task = {
2789
 
    .func=wait_for_mandos_client_exit,
2790
 
    .pid=pid,
2791
 
    .mandos_client_exited=&mandos_client_exited,
2792
 
    .quit_now=&quit_now,
2793
 
  };
2794
 
  task.func(task, queue);
2795
 
 
2796
 
  g_assert_false(mandos_client_exited);
2797
 
  g_assert_false(quit_now);
2798
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
2799
 
 
2800
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
2801
 
        .func=wait_for_mandos_client_exit,
2802
 
        .pid=task.pid,
2803
 
        .mandos_client_exited=&mandos_client_exited,
2804
 
        .quit_now=&quit_now,
2805
 
      }));
2806
 
}
2807
 
 
2808
 
static
2809
 
void test_wait_for_mandos_client_exit_success(test_fixture *fixture,
2810
 
                                              __attribute__((unused))
2811
 
                                              gconstpointer
2812
 
                                              user_data){
2813
 
  bool mandos_client_exited = false;
2814
 
  bool quit_now = false;
2815
 
 
2816
 
  pid_t create_successful_process(void){
2817
 
    const pid_t pid = fork();
2818
 
    if(pid == 0){               /* Child */
2819
 
      if(not restore_signal_handler(&fixture->orig_sigaction)){
2820
 
        _exit(EXIT_FAILURE);
2821
 
      }
2822
 
      if(not restore_sigmask(&fixture->orig_sigmask)){
2823
 
        _exit(EXIT_FAILURE);
2824
 
      }
2825
 
      exit(EXIT_SUCCESS);
2826
 
    }
2827
 
    return pid;
2828
 
  }
2829
 
  const pid_t pid = create_successful_process();
2830
 
  g_assert_true(pid != -1);
2831
 
 
2832
 
  __attribute__((cleanup(cleanup_queue)))
2833
 
    task_queue *queue = create_queue();
2834
 
  g_assert_nonnull(queue);
2835
 
  const task_context initial_task = {
2836
 
    .func=wait_for_mandos_client_exit,
2837
 
    .pid=pid,
2838
 
    .mandos_client_exited=&mandos_client_exited,
2839
 
    .quit_now=&quit_now,
2840
 
  };
2841
 
  g_assert_true(add_to_queue(queue, initial_task));
2842
 
 
2843
 
  struct timespec starttime, currtime;
2844
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2845
 
  __attribute__((cleanup(cleanup_close)))
2846
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2847
 
  do {
2848
 
    queue->next_run = 0;
2849
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2850
 
    g_assert_true(run_queue(&queue, (string_set[]){{}}, &quit_now));
2851
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2852
 
  } while((not mandos_client_exited)
2853
 
          and (not quit_now)
2854
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2855
 
 
2856
 
  g_assert_true(mandos_client_exited);
2857
 
  g_assert_false(quit_now);
2858
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2859
 
}
2860
 
 
2861
 
static
2862
 
void test_wait_for_mandos_client_exit_failure(test_fixture *fixture,
2863
 
                                              __attribute__((unused))
2864
 
                                              gconstpointer
2865
 
                                              user_data){
2866
 
  bool mandos_client_exited = false;
2867
 
  bool quit_now = false;
2868
 
 
2869
 
  pid_t create_failing_process(void){
2870
 
    const pid_t pid = fork();
2871
 
    if(pid == 0){               /* Child */
2872
 
      if(not restore_signal_handler(&fixture->orig_sigaction)){
2873
 
        _exit(EXIT_FAILURE);
2874
 
      }
2875
 
      if(not restore_sigmask(&fixture->orig_sigmask)){
2876
 
        _exit(EXIT_FAILURE);
2877
 
      }
2878
 
      exit(EXIT_FAILURE);
2879
 
    }
2880
 
    return pid;
2881
 
  }
2882
 
  const pid_t pid = create_failing_process();
2883
 
  g_assert_true(pid != -1);
2884
 
 
2885
 
  __attribute__((cleanup(string_set_clear)))
2886
 
    string_set cancelled_filenames = {};
2887
 
  __attribute__((cleanup(cleanup_close)))
2888
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2889
 
  g_assert_cmpint(epoll_fd, >=, 0);
2890
 
  __attribute__((cleanup(cleanup_queue)))
2891
 
    task_queue *queue = create_queue();
2892
 
  g_assert_nonnull(queue);
2893
 
  g_assert_true(add_to_queue(queue, (task_context){
2894
 
        .func=wait_for_mandos_client_exit,
2895
 
        .pid=pid,
2896
 
        .mandos_client_exited=&mandos_client_exited,
2897
 
        .quit_now=&quit_now,
2898
 
      }));
2899
 
 
2900
 
  g_assert_true(sigismember(&fixture->orig_sigmask, SIGCHLD) == 0);
2901
 
 
2902
 
  __attribute__((cleanup(cleanup_close)))
2903
 
    const int devnull_fd = open("/dev/null",
2904
 
                                O_WRONLY | O_CLOEXEC);
2905
 
  g_assert_cmpint(devnull_fd, >=, 0);
2906
 
  __attribute__((cleanup(cleanup_close)))
2907
 
    const int real_stderr_fd = dup(STDERR_FILENO);
2908
 
  g_assert_cmpint(real_stderr_fd, >=, 0);
2909
 
 
2910
 
  struct timespec starttime, currtime;
2911
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2912
 
  do {
2913
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2914
 
    dup2(devnull_fd, STDERR_FILENO);
2915
 
    const bool success = run_queue(&queue, &cancelled_filenames,
2916
 
                                   &quit_now);
2917
 
    dup2(real_stderr_fd, STDERR_FILENO);
2918
 
    if(not success){
2919
 
      break;
2920
 
    }
2921
 
 
2922
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2923
 
  } while((not mandos_client_exited)
2924
 
          and (not quit_now)
2925
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2926
 
 
2927
 
  g_assert_true(quit_now);
2928
 
  g_assert_true(mandos_client_exited);
2929
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
2930
 
}
2931
 
 
2932
 
static
2933
 
void test_wait_for_mandos_client_exit_killed(test_fixture *fixture,
2934
 
                                             __attribute__((unused))
2935
 
                                             gconstpointer user_data){
2936
 
  bool mandos_client_exited = false;
2937
 
  bool quit_now = false;
2938
 
 
2939
 
  pid_t create_killed_process(void){
2940
 
    const pid_t pid = fork();
2941
 
    if(pid == 0){               /* Child */
2942
 
      if(not restore_signal_handler(&fixture->orig_sigaction)){
2943
 
        _exit(EXIT_FAILURE);
2944
 
      }
2945
 
      if(not restore_sigmask(&fixture->orig_sigmask)){
2946
 
        _exit(EXIT_FAILURE);
2947
 
      }
2948
 
      while(true){
2949
 
        pause();
2950
 
      }
2951
 
    }
2952
 
    kill(pid, SIGKILL);
2953
 
    return pid;
2954
 
  }
2955
 
  const pid_t pid = create_killed_process();
2956
 
  g_assert_true(pid != -1);
2957
 
 
2958
 
  __attribute__((cleanup(string_set_clear)))
2959
 
    string_set cancelled_filenames = {};
2960
 
  __attribute__((cleanup(cleanup_close)))
2961
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
2962
 
  g_assert_cmpint(epoll_fd, >=, 0);
2963
 
  __attribute__((cleanup(cleanup_queue)))
2964
 
    task_queue *queue = create_queue();
2965
 
  g_assert_nonnull(queue);
2966
 
  g_assert_true(add_to_queue(queue, (task_context){
2967
 
        .func=wait_for_mandos_client_exit,
2968
 
        .pid=pid,
2969
 
        .mandos_client_exited=&mandos_client_exited,
2970
 
        .quit_now=&quit_now,
2971
 
      }));
2972
 
 
2973
 
  __attribute__((cleanup(cleanup_close)))
2974
 
    const int devnull_fd = open("/dev/null",
2975
 
                                O_WRONLY | O_CLOEXEC);
2976
 
  g_assert_cmpint(devnull_fd, >=, 0);
2977
 
  __attribute__((cleanup(cleanup_close)))
2978
 
    const int real_stderr_fd = dup(STDERR_FILENO);
2979
 
  g_assert_cmpint(real_stderr_fd, >=, 0);
2980
 
 
2981
 
  struct timespec starttime, currtime;
2982
 
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
2983
 
  do {
2984
 
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
2985
 
    dup2(devnull_fd, STDERR_FILENO);
2986
 
    const bool success = run_queue(&queue, &cancelled_filenames,
2987
 
                                   &quit_now);
2988
 
    dup2(real_stderr_fd, STDERR_FILENO);
2989
 
    if(not success){
2990
 
      break;
2991
 
    }
2992
 
 
2993
 
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
2994
 
  } while((not mandos_client_exited)
2995
 
          and (not quit_now)
2996
 
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
2997
 
 
2998
 
  g_assert_true(mandos_client_exited);
2999
 
  g_assert_true(quit_now);
3000
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3001
 
}
3002
 
 
3003
 
static bool epoll_set_does_not_contain(int, int);
3004
 
 
3005
 
static
3006
 
void test_read_mandos_client_output_readerror(__attribute__((unused))
3007
 
                                              test_fixture *fixture,
3008
 
                                              __attribute__((unused))
3009
 
                                              gconstpointer
3010
 
                                              user_data){
3011
 
  __attribute__((cleanup(cleanup_close)))
3012
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3013
 
  g_assert_cmpint(epoll_fd, >=, 0);
3014
 
 
3015
 
  __attribute__((cleanup(cleanup_buffer)))
3016
 
    buffer password = {};
3017
 
 
3018
 
  /* Reading /proc/self/mem from offset 0 will always give EIO */
3019
 
  const int fd = open("/proc/self/mem", O_RDONLY | O_CLOEXEC);
3020
 
 
3021
 
  bool password_is_read = false;
3022
 
  bool quit_now = false;
3023
 
  __attribute__((cleanup(cleanup_queue)))
3024
 
    task_queue *queue = create_queue();
3025
 
  g_assert_nonnull(queue);
3026
 
 
3027
 
  task_context task = {
3028
 
    .func=read_mandos_client_output,
3029
 
    .epoll_fd=epoll_fd,
3030
 
    .fd=fd,
3031
 
    .password=&password,
3032
 
    .password_is_read=&password_is_read,
3033
 
    .quit_now=&quit_now,
3034
 
  };
3035
 
  run_task_with_stderr_to_dev_null(task, queue);
3036
 
  g_assert_false(password_is_read);
3037
 
  g_assert_cmpint((int)password.length, ==, 0);
3038
 
  g_assert_true(quit_now);
3039
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3040
 
 
3041
 
  g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));
3042
 
 
3043
 
  g_assert_cmpint(close(fd), ==, -1);
3044
 
}
3045
 
 
3046
 
static bool epoll_set_does_not_contain(int epoll_fd, int fd){
3047
 
  return not epoll_set_contains(epoll_fd, fd, 0);
3048
 
}
3049
 
 
3050
 
static
3051
 
void test_read_mandos_client_output_nodata(__attribute__((unused))
3052
 
                                           test_fixture *fixture,
3053
 
                                           __attribute__((unused))
3054
 
                                           gconstpointer user_data){
3055
 
  __attribute__((cleanup(cleanup_close)))
3056
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3057
 
  g_assert_cmpint(epoll_fd, >=, 0);
3058
 
 
3059
 
  int pipefds[2];
3060
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3061
 
 
3062
 
  __attribute__((cleanup(cleanup_buffer)))
3063
 
    buffer password = {};
3064
 
 
3065
 
  bool password_is_read = false;
3066
 
  bool quit_now = false;
3067
 
  __attribute__((cleanup(cleanup_queue)))
3068
 
    task_queue *queue = create_queue();
3069
 
  g_assert_nonnull(queue);
3070
 
 
3071
 
  task_context task = {
3072
 
    .func=read_mandos_client_output,
3073
 
    .epoll_fd=epoll_fd,
3074
 
    .fd=pipefds[0],
3075
 
    .password=&password,
3076
 
    .password_is_read=&password_is_read,
3077
 
    .quit_now=&quit_now,
3078
 
  };
3079
 
  task.func(task, queue);
3080
 
  g_assert_false(password_is_read);
3081
 
  g_assert_cmpint((int)password.length, ==, 0);
3082
 
  g_assert_false(quit_now);
3083
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
3084
 
 
3085
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3086
 
        .func=read_mandos_client_output,
3087
 
        .epoll_fd=epoll_fd,
3088
 
        .fd=pipefds[0],
3089
 
        .password=&password,
3090
 
        .password_is_read=&password_is_read,
3091
 
        .quit_now=&quit_now,
3092
 
      }));
3093
 
 
3094
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3095
 
                                   EPOLLIN | EPOLLRDHUP));
3096
 
 
3097
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3098
 
}
3099
 
 
3100
 
static void test_read_mandos_client_output_eof(__attribute__((unused))
3101
 
                                               test_fixture *fixture,
3102
 
                                               __attribute__((unused))
3103
 
                                               gconstpointer
3104
 
                                               user_data){
3105
 
  __attribute__((cleanup(cleanup_close)))
3106
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3107
 
  g_assert_cmpint(epoll_fd, >=, 0);
3108
 
 
3109
 
  int pipefds[2];
3110
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3111
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3112
 
 
3113
 
  __attribute__((cleanup(cleanup_buffer)))
3114
 
    buffer password = {};
3115
 
 
3116
 
  bool password_is_read = false;
3117
 
  bool quit_now = false;
3118
 
  __attribute__((cleanup(cleanup_queue)))
3119
 
    task_queue *queue = create_queue();
3120
 
  g_assert_nonnull(queue);
3121
 
 
3122
 
  task_context task = {
3123
 
    .func=read_mandos_client_output,
3124
 
    .epoll_fd=epoll_fd,
3125
 
    .fd=pipefds[0],
3126
 
    .password=&password,
3127
 
    .password_is_read=&password_is_read,
3128
 
    .quit_now=&quit_now,
3129
 
  };
3130
 
  task.func(task, queue);
3131
 
  g_assert_true(password_is_read);
3132
 
  g_assert_cmpint((int)password.length, ==, 0);
3133
 
  g_assert_false(quit_now);
3134
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3135
 
 
3136
 
  g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));
3137
 
 
3138
 
  g_assert_cmpint(close(pipefds[0]), ==, -1);
3139
 
}
3140
 
 
3141
 
static
3142
 
void test_read_mandos_client_output_once(__attribute__((unused))
3143
 
                                         test_fixture *fixture,
3144
 
                                         __attribute__((unused))
3145
 
                                         gconstpointer user_data){
3146
 
  __attribute__((cleanup(cleanup_close)))
3147
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3148
 
  g_assert_cmpint(epoll_fd, >=, 0);
3149
 
 
3150
 
  int pipefds[2];
3151
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3152
 
 
3153
 
  const char dummy_test_password[] = "dummy test password";
3154
 
  /* Start with a pre-allocated buffer */
3155
 
  __attribute__((cleanup(cleanup_buffer)))
3156
 
    buffer password = {
3157
 
    .data=malloc(sizeof(dummy_test_password)),
3158
 
    .length=0,
3159
 
    .allocated=sizeof(dummy_test_password),
3160
 
  };
3161
 
  g_assert_nonnull(password.data);
3162
 
  if(mlock(password.data, password.allocated) != 0){
3163
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
3164
 
  }
3165
 
 
3166
 
  bool password_is_read = false;
3167
 
  bool quit_now = false;
3168
 
  __attribute__((cleanup(cleanup_queue)))
3169
 
    task_queue *queue = create_queue();
3170
 
  g_assert_nonnull(queue);
3171
 
 
3172
 
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
3173
 
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
3174
 
                             sizeof(dummy_test_password)),
3175
 
                  ==, (int)sizeof(dummy_test_password));
3176
 
 
3177
 
  task_context task = {
3178
 
    .func=read_mandos_client_output,
3179
 
    .epoll_fd=epoll_fd,
3180
 
    .fd=pipefds[0],
3181
 
    .password=&password,
3182
 
    .password_is_read=&password_is_read,
3183
 
    .quit_now=&quit_now,
3184
 
  };
3185
 
  task.func(task, queue);
3186
 
 
3187
 
  g_assert_false(password_is_read);
3188
 
  g_assert_cmpint((int)password.length, ==,
3189
 
                  (int)sizeof(dummy_test_password));
3190
 
  g_assert_nonnull(password.data);
3191
 
  g_assert_cmpint(memcmp(password.data, dummy_test_password,
3192
 
                         sizeof(dummy_test_password)), ==, 0);
3193
 
 
3194
 
  g_assert_false(quit_now);
3195
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
3196
 
 
3197
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3198
 
        .func=read_mandos_client_output,
3199
 
        .epoll_fd=epoll_fd,
3200
 
        .fd=pipefds[0],
3201
 
        .password=&password,
3202
 
        .password_is_read=&password_is_read,
3203
 
        .quit_now=&quit_now,
3204
 
      }));
3205
 
 
3206
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3207
 
                                   EPOLLIN | EPOLLRDHUP));
3208
 
 
3209
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3210
 
}
3211
 
 
3212
 
static
3213
 
void test_read_mandos_client_output_malloc(__attribute__((unused))
3214
 
                                           test_fixture *fixture,
3215
 
                                           __attribute__((unused))
3216
 
                                           gconstpointer user_data){
3217
 
  __attribute__((cleanup(cleanup_close)))
3218
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3219
 
  g_assert_cmpint(epoll_fd, >=, 0);
3220
 
 
3221
 
  int pipefds[2];
3222
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3223
 
 
3224
 
  const char dummy_test_password[] = "dummy test password";
3225
 
  /* Start with an empty buffer */
3226
 
  __attribute__((cleanup(cleanup_buffer)))
3227
 
    buffer password = {};
3228
 
 
3229
 
  bool password_is_read = false;
3230
 
  bool quit_now = false;
3231
 
  __attribute__((cleanup(cleanup_queue)))
3232
 
    task_queue *queue = create_queue();
3233
 
  g_assert_nonnull(queue);
3234
 
 
3235
 
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
3236
 
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
3237
 
                             sizeof(dummy_test_password)),
3238
 
                  ==, (int)sizeof(dummy_test_password));
3239
 
 
3240
 
  task_context task = {
3241
 
    .func=read_mandos_client_output,
3242
 
    .epoll_fd=epoll_fd,
3243
 
    .fd=pipefds[0],
3244
 
    .password=&password,
3245
 
    .password_is_read=&password_is_read,
3246
 
    .quit_now=&quit_now,
3247
 
  };
3248
 
  task.func(task, queue);
3249
 
 
3250
 
  g_assert_false(password_is_read);
3251
 
  g_assert_cmpint((int)password.length, ==,
3252
 
                  (int)sizeof(dummy_test_password));
3253
 
  g_assert_nonnull(password.data);
3254
 
  g_assert_cmpint(memcmp(password.data, dummy_test_password,
3255
 
                         sizeof(dummy_test_password)), ==, 0);
3256
 
 
3257
 
  g_assert_false(quit_now);
3258
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
3259
 
 
3260
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3261
 
        .func=read_mandos_client_output,
3262
 
        .epoll_fd=epoll_fd,
3263
 
        .fd=pipefds[0],
3264
 
        .password=&password,
3265
 
        .password_is_read=&password_is_read,
3266
 
        .quit_now=&quit_now,
3267
 
      }));
3268
 
 
3269
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3270
 
                                   EPOLLIN | EPOLLRDHUP));
3271
 
 
3272
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3273
 
}
3274
 
 
3275
 
static
3276
 
void test_read_mandos_client_output_append(__attribute__((unused))
3277
 
                                           test_fixture *fixture,
3278
 
                                           __attribute__((unused))
3279
 
                                           gconstpointer user_data){
3280
 
  __attribute__((cleanup(cleanup_close)))
3281
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3282
 
  g_assert_cmpint(epoll_fd, >=, 0);
3283
 
 
3284
 
  int pipefds[2];
3285
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3286
 
 
3287
 
  const char dummy_test_password[] = "dummy test password";
3288
 
  __attribute__((cleanup(cleanup_buffer)))
3289
 
    buffer password = {
3290
 
    .data=malloc(PIPE_BUF),
3291
 
    .length=PIPE_BUF,
3292
 
    .allocated=PIPE_BUF,
3293
 
  };
3294
 
  g_assert_nonnull(password.data);
3295
 
  if(mlock(password.data, password.allocated) != 0){
3296
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
3297
 
  }
3298
 
 
3299
 
  memset(password.data, 'x', PIPE_BUF);
3300
 
  char password_expected[PIPE_BUF];
3301
 
  memcpy(password_expected, password.data, PIPE_BUF);
3302
 
 
3303
 
  bool password_is_read = false;
3304
 
  bool quit_now = false;
3305
 
  __attribute__((cleanup(cleanup_queue)))
3306
 
    task_queue *queue = create_queue();
3307
 
  g_assert_nonnull(queue);
3308
 
 
3309
 
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
3310
 
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
3311
 
                             sizeof(dummy_test_password)),
3312
 
                  ==, (int)sizeof(dummy_test_password));
3313
 
 
3314
 
  task_context task = {
3315
 
    .func=read_mandos_client_output,
3316
 
    .epoll_fd=epoll_fd,
3317
 
    .fd=pipefds[0],
3318
 
    .password=&password,
3319
 
    .password_is_read=&password_is_read,
3320
 
    .quit_now=&quit_now,
3321
 
  };
3322
 
  task.func(task, queue);
3323
 
 
3324
 
  g_assert_false(password_is_read);
3325
 
  g_assert_cmpint((int)password.length, ==,
3326
 
                  PIPE_BUF + sizeof(dummy_test_password));
3327
 
  g_assert_nonnull(password.data);
3328
 
  g_assert_cmpint(memcmp(password_expected, password.data, PIPE_BUF),
3329
 
                  ==, 0);
3330
 
  g_assert_cmpint(memcmp(password.data + PIPE_BUF,
3331
 
                         dummy_test_password,
3332
 
                         sizeof(dummy_test_password)), ==, 0);
3333
 
  g_assert_false(quit_now);
3334
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
3335
 
 
3336
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3337
 
        .func=read_mandos_client_output,
3338
 
        .epoll_fd=epoll_fd,
3339
 
        .fd=pipefds[0],
3340
 
        .password=&password,
3341
 
        .password_is_read=&password_is_read,
3342
 
        .quit_now=&quit_now,
3343
 
      }));
3344
 
 
3345
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3346
 
                                   EPOLLIN | EPOLLRDHUP));
3347
 
}
3348
 
 
3349
 
static char *make_temporary_directory(void);
3350
 
 
3351
 
static void test_add_inotify_dir_watch(__attribute__((unused))
3352
 
                                       test_fixture *fixture,
3353
 
                                       __attribute__((unused))
3354
 
                                       gconstpointer user_data){
3355
 
  __attribute__((cleanup(cleanup_close)))
3356
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3357
 
  g_assert_cmpint(epoll_fd, >=, 0);
3358
 
  __attribute__((cleanup(cleanup_queue)))
3359
 
    task_queue *queue = create_queue();
3360
 
  g_assert_nonnull(queue);
3361
 
  __attribute__((cleanup(string_set_clear)))
3362
 
    string_set cancelled_filenames = {};
3363
 
  const mono_microsecs current_time = 0;
3364
 
 
3365
 
  bool quit_now = false;
3366
 
  buffer password = {};
3367
 
  bool mandos_client_exited = false;
3368
 
  bool password_is_read = false;
3369
 
 
3370
 
  __attribute__((cleanup(cleanup_string)))
3371
 
    char *tempdir = make_temporary_directory();
3372
 
  g_assert_nonnull(tempdir);
3373
 
 
3374
 
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3375
 
                                      &password, tempdir,
3376
 
                                      &cancelled_filenames,
3377
 
                                      &current_time,
3378
 
                                      &mandos_client_exited,
3379
 
                                      &password_is_read));
3380
 
 
3381
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
3382
 
 
3383
 
  const task_context *const added_read_task
3384
 
    = find_matching_task(queue, (task_context){
3385
 
        .func=read_inotify_event,
3386
 
        .epoll_fd=epoll_fd,
3387
 
        .quit_now=&quit_now,
3388
 
        .password=&password,
3389
 
        .filename=tempdir,
3390
 
        .cancelled_filenames=&cancelled_filenames,
3391
 
        .current_time=&current_time,
3392
 
        .mandos_client_exited=&mandos_client_exited,
3393
 
        .password_is_read=&password_is_read,
3394
 
      });
3395
 
  g_assert_nonnull(added_read_task);
3396
 
 
3397
 
  g_assert_cmpint(added_read_task->fd, >, 2);
3398
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
3399
 
  g_assert_true(epoll_set_contains(added_read_task->epoll_fd,
3400
 
                                   added_read_task->fd,
3401
 
                                   EPOLLIN | EPOLLRDHUP));
3402
 
 
3403
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
3404
 
}
3405
 
 
3406
 
static char *make_temporary_directory(void){
3407
 
  char *name = strdup("/tmp/mandosXXXXXX");
3408
 
  g_assert_nonnull(name);
3409
 
  char *result = mkdtemp(name);
3410
 
  if(result == NULL){
3411
 
    free(name);
3412
 
  }
3413
 
  return result;
3414
 
}
3415
 
 
3416
 
static void test_add_inotify_dir_watch_fail(__attribute__((unused))
3417
 
                                            test_fixture *fixture,
3418
 
                                            __attribute__((unused))
3419
 
                                            gconstpointer user_data){
3420
 
  __attribute__((cleanup(cleanup_close)))
3421
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3422
 
  g_assert_cmpint(epoll_fd, >=, 0);
3423
 
  __attribute__((cleanup(cleanup_queue)))
3424
 
    task_queue *queue = create_queue();
3425
 
  g_assert_nonnull(queue);
3426
 
  __attribute__((cleanup(string_set_clear)))
3427
 
    string_set cancelled_filenames = {};
3428
 
  const mono_microsecs current_time = 0;
3429
 
 
3430
 
  bool quit_now = false;
3431
 
  buffer password = {};
3432
 
  bool mandos_client_exited = false;
3433
 
  bool password_is_read = false;
3434
 
 
3435
 
  const char nonexistent_dir[] = "/nonexistent";
3436
 
 
3437
 
  FILE *real_stderr = stderr;
3438
 
  FILE *devnull = fopen("/dev/null", "we");
3439
 
  g_assert_nonnull(devnull);
3440
 
  stderr = devnull;
3441
 
  g_assert_false(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3442
 
                                       &password, nonexistent_dir,
3443
 
                                       &cancelled_filenames,
3444
 
                                       &current_time,
3445
 
                                       &mandos_client_exited,
3446
 
                                       &password_is_read));
3447
 
  stderr = real_stderr;
3448
 
  g_assert_cmpint(fclose(devnull), ==, 0);
3449
 
 
3450
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3451
 
}
3452
 
 
3453
 
static void test_add_inotify_dir_watch_EAGAIN(__attribute__((unused))
3454
 
                                              test_fixture *fixture,
3455
 
                                              __attribute__((unused))
3456
 
                                              gconstpointer
3457
 
                                              user_data){
3458
 
  __attribute__((cleanup(cleanup_close)))
3459
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3460
 
  g_assert_cmpint(epoll_fd, >=, 0);
3461
 
  __attribute__((cleanup(cleanup_queue)))
3462
 
    task_queue *queue = create_queue();
3463
 
  g_assert_nonnull(queue);
3464
 
  __attribute__((cleanup(string_set_clear)))
3465
 
    string_set cancelled_filenames = {};
3466
 
  const mono_microsecs current_time = 0;
3467
 
 
3468
 
  bool quit_now = false;
3469
 
  buffer password = {};
3470
 
  bool mandos_client_exited = false;
3471
 
  bool password_is_read = false;
3472
 
 
3473
 
  __attribute__((cleanup(cleanup_string)))
3474
 
    char *tempdir = make_temporary_directory();
3475
 
  g_assert_nonnull(tempdir);
3476
 
 
3477
 
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3478
 
                                      &password, tempdir,
3479
 
                                      &cancelled_filenames,
3480
 
                                      &current_time,
3481
 
                                      &mandos_client_exited,
3482
 
                                      &password_is_read));
3483
 
 
3484
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
3485
 
 
3486
 
  const task_context *const added_read_task
3487
 
    = find_matching_task(queue,
3488
 
                         (task_context){ .func=read_inotify_event });
3489
 
  g_assert_nonnull(added_read_task);
3490
 
 
3491
 
  g_assert_cmpint(added_read_task->fd, >, 2);
3492
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
3493
 
 
3494
 
  /* "sufficient to read at least one event." - inotify(7) */
3495
 
  const size_t ievent_size = (sizeof(struct inotify_event)
3496
 
                              + NAME_MAX + 1);
3497
 
  struct inotify_event *ievent = malloc(ievent_size);
3498
 
  g_assert_nonnull(ievent);
3499
 
 
3500
 
  g_assert_cmpint(read(added_read_task->fd, ievent, ievent_size), ==,
3501
 
                  -1);
3502
 
  g_assert_cmpint(errno, ==, EAGAIN);
3503
 
 
3504
 
  free(ievent);
3505
 
 
3506
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
3507
 
}
3508
 
 
3509
 
static char *make_temporary_file_in_directory(const char
3510
 
                                              *const dir);
3511
 
 
3512
 
static
3513
 
void test_add_inotify_dir_watch_IN_CLOSE_WRITE(__attribute__((unused))
3514
 
                                               test_fixture *fixture,
3515
 
                                               __attribute__((unused))
3516
 
                                               gconstpointer
3517
 
                                               user_data){
3518
 
  __attribute__((cleanup(cleanup_close)))
3519
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3520
 
  g_assert_cmpint(epoll_fd, >=, 0);
3521
 
  __attribute__((cleanup(cleanup_queue)))
3522
 
    task_queue *queue = create_queue();
3523
 
  g_assert_nonnull(queue);
3524
 
  __attribute__((cleanup(string_set_clear)))
3525
 
    string_set cancelled_filenames = {};
3526
 
  const mono_microsecs current_time = 0;
3527
 
 
3528
 
  bool quit_now = false;
3529
 
  buffer password = {};
3530
 
  bool mandos_client_exited = false;
3531
 
  bool password_is_read = false;
3532
 
 
3533
 
  __attribute__((cleanup(cleanup_string)))
3534
 
    char *tempdir = make_temporary_directory();
3535
 
  g_assert_nonnull(tempdir);
3536
 
 
3537
 
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3538
 
                                      &password, tempdir,
3539
 
                                      &cancelled_filenames,
3540
 
                                      &current_time,
3541
 
                                      &mandos_client_exited,
3542
 
                                      &password_is_read));
3543
 
 
3544
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
3545
 
 
3546
 
  const task_context *const added_read_task
3547
 
    = find_matching_task(queue,
3548
 
                         (task_context){ .func=read_inotify_event });
3549
 
  g_assert_nonnull(added_read_task);
3550
 
 
3551
 
  g_assert_cmpint(added_read_task->fd, >, 2);
3552
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
3553
 
 
3554
 
  __attribute__((cleanup(cleanup_string)))
3555
 
    char *filename = make_temporary_file_in_directory(tempdir);
3556
 
  g_assert_nonnull(filename);
3557
 
 
3558
 
  /* "sufficient to read at least one event." - inotify(7) */
3559
 
  const size_t ievent_size = (sizeof(struct inotify_event)
3560
 
                              + NAME_MAX + 1);
3561
 
  struct inotify_event *ievent = malloc(ievent_size);
3562
 
  g_assert_nonnull(ievent);
3563
 
 
3564
 
  ssize_t read_size = 0;
3565
 
  read_size = read(added_read_task->fd, ievent, ievent_size);
3566
 
 
3567
 
  g_assert_cmpint((int)read_size, >, 0);
3568
 
  g_assert_true(ievent->mask & IN_CLOSE_WRITE);
3569
 
  g_assert_cmpstr(ievent->name, ==, basename(filename));
3570
 
 
3571
 
  free(ievent);
3572
 
 
3573
 
  g_assert_cmpint(unlink(filename), ==, 0);
3574
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
3575
 
}
3576
 
 
3577
 
static char *make_temporary_prefixed_file_in_directory(const char
3578
 
                                                       *const prefix,
3579
 
                                                       const char
3580
 
                                                       *const dir){
3581
 
  char *filename = NULL;
3582
 
  g_assert_cmpint(asprintf(&filename, "%s/%sXXXXXX", dir, prefix),
3583
 
                  >, 0);
3584
 
  g_assert_nonnull(filename);
3585
 
  const int fd = mkostemp(filename, O_CLOEXEC);
3586
 
  g_assert_cmpint(fd, >=, 0);
3587
 
  g_assert_cmpint(close(fd), ==, 0);
3588
 
  return filename;
3589
 
}
3590
 
 
3591
 
static char *make_temporary_file_in_directory(const char
3592
 
                                              *const dir){
3593
 
  return make_temporary_prefixed_file_in_directory("temp", dir);
3594
 
}
3595
 
 
3596
 
static
3597
 
void test_add_inotify_dir_watch_IN_MOVED_TO(__attribute__((unused))
3598
 
                                            test_fixture *fixture,
3599
 
                                            __attribute__((unused))
3600
 
                                            gconstpointer user_data){
3601
 
  __attribute__((cleanup(cleanup_close)))
3602
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3603
 
  g_assert_cmpint(epoll_fd, >=, 0);
3604
 
  __attribute__((cleanup(cleanup_queue)))
3605
 
    task_queue *queue = create_queue();
3606
 
  g_assert_nonnull(queue);
3607
 
  __attribute__((cleanup(string_set_clear)))
3608
 
    string_set cancelled_filenames = {};
3609
 
  const mono_microsecs current_time = 0;
3610
 
 
3611
 
  bool quit_now = false;
3612
 
  buffer password = {};
3613
 
  bool mandos_client_exited = false;
3614
 
  bool password_is_read = false;
3615
 
 
3616
 
  __attribute__((cleanup(cleanup_string)))
3617
 
    char *watchdir = make_temporary_directory();
3618
 
  g_assert_nonnull(watchdir);
3619
 
 
3620
 
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3621
 
                                      &password, watchdir,
3622
 
                                      &cancelled_filenames,
3623
 
                                      &current_time,
3624
 
                                      &mandos_client_exited,
3625
 
                                      &password_is_read));
3626
 
 
3627
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
3628
 
 
3629
 
  const task_context *const added_read_task
3630
 
    = find_matching_task(queue,
3631
 
                         (task_context){ .func=read_inotify_event });
3632
 
  g_assert_nonnull(added_read_task);
3633
 
 
3634
 
  g_assert_cmpint(added_read_task->fd, >, 2);
3635
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
3636
 
 
3637
 
  char *sourcedir = make_temporary_directory();
3638
 
  g_assert_nonnull(sourcedir);
3639
 
 
3640
 
  __attribute__((cleanup(cleanup_string)))
3641
 
    char *filename = make_temporary_file_in_directory(sourcedir);
3642
 
  g_assert_nonnull(filename);
3643
 
 
3644
 
  __attribute__((cleanup(cleanup_string)))
3645
 
    char *targetfilename = NULL;
3646
 
  g_assert_cmpint(asprintf(&targetfilename, "%s/%s", watchdir,
3647
 
                           basename(filename)), >, 0);
3648
 
  g_assert_nonnull(targetfilename);
3649
 
 
3650
 
  g_assert_cmpint(rename(filename, targetfilename), ==, 0);
3651
 
  g_assert_cmpint(rmdir(sourcedir), ==, 0);
3652
 
  free(sourcedir);
3653
 
 
3654
 
  /* "sufficient to read at least one event." - inotify(7) */
3655
 
  const size_t ievent_size = (sizeof(struct inotify_event)
3656
 
                              + NAME_MAX + 1);
3657
 
  struct inotify_event *ievent = malloc(ievent_size);
3658
 
  g_assert_nonnull(ievent);
3659
 
 
3660
 
  ssize_t read_size = read(added_read_task->fd, ievent, ievent_size);
3661
 
 
3662
 
  g_assert_cmpint((int)read_size, >, 0);
3663
 
  g_assert_true(ievent->mask & IN_MOVED_TO);
3664
 
  g_assert_cmpstr(ievent->name, ==, basename(targetfilename));
3665
 
 
3666
 
  free(ievent);
3667
 
 
3668
 
  g_assert_cmpint(unlink(targetfilename), ==, 0);
3669
 
  g_assert_cmpint(rmdir(watchdir), ==, 0);
3670
 
}
3671
 
 
3672
 
static
3673
 
void test_add_inotify_dir_watch_IN_DELETE(__attribute__((unused))
3674
 
                                          test_fixture *fixture,
3675
 
                                          __attribute__((unused))
3676
 
                                          gconstpointer user_data){
3677
 
  __attribute__((cleanup(cleanup_close)))
3678
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3679
 
  g_assert_cmpint(epoll_fd, >=, 0);
3680
 
  __attribute__((cleanup(cleanup_queue)))
3681
 
    task_queue *queue = create_queue();
3682
 
  g_assert_nonnull(queue);
3683
 
  __attribute__((cleanup(string_set_clear)))
3684
 
    string_set cancelled_filenames = {};
3685
 
  const mono_microsecs current_time = 0;
3686
 
 
3687
 
  bool quit_now = false;
3688
 
  buffer password = {};
3689
 
  bool mandos_client_exited = false;
3690
 
  bool password_is_read = false;
3691
 
 
3692
 
  __attribute__((cleanup(cleanup_string)))
3693
 
    char *tempdir = make_temporary_directory();
3694
 
  g_assert_nonnull(tempdir);
3695
 
 
3696
 
  __attribute__((cleanup(cleanup_string)))
3697
 
    char *tempfile = make_temporary_file_in_directory(tempdir);
3698
 
  g_assert_nonnull(tempfile);
3699
 
 
3700
 
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
3701
 
                                      &password, tempdir,
3702
 
                                      &cancelled_filenames,
3703
 
                                      &current_time,
3704
 
                                      &mandos_client_exited,
3705
 
                                      &password_is_read));
3706
 
  g_assert_cmpint(unlink(tempfile), ==, 0);
3707
 
 
3708
 
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
3709
 
 
3710
 
  const task_context *const added_read_task
3711
 
    = find_matching_task(queue,
3712
 
                         (task_context){ .func=read_inotify_event });
3713
 
  g_assert_nonnull(added_read_task);
3714
 
 
3715
 
  g_assert_cmpint(added_read_task->fd, >, 2);
3716
 
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
3717
 
 
3718
 
  /* "sufficient to read at least one event." - inotify(7) */
3719
 
  const size_t ievent_size = (sizeof(struct inotify_event)
3720
 
                              + NAME_MAX + 1);
3721
 
  struct inotify_event *ievent = malloc(ievent_size);
3722
 
  g_assert_nonnull(ievent);
3723
 
 
3724
 
  ssize_t read_size = 0;
3725
 
  read_size = read(added_read_task->fd, ievent, ievent_size);
3726
 
 
3727
 
  g_assert_cmpint((int)read_size, >, 0);
3728
 
  g_assert_true(ievent->mask & IN_DELETE);
3729
 
  g_assert_cmpstr(ievent->name, ==, basename(tempfile));
3730
 
 
3731
 
  free(ievent);
3732
 
 
3733
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
3734
 
}
3735
 
 
3736
 
static void test_read_inotify_event_readerror(__attribute__((unused))
3737
 
                                              test_fixture *fixture,
3738
 
                                              __attribute__((unused))
3739
 
                                              gconstpointer
3740
 
                                              user_data){
3741
 
  __attribute__((cleanup(cleanup_close)))
3742
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3743
 
  g_assert_cmpint(epoll_fd, >=, 0);
3744
 
  const mono_microsecs current_time = 0;
3745
 
 
3746
 
  /* Reading /proc/self/mem from offset 0 will always result in EIO */
3747
 
  const int fd = open("/proc/self/mem", O_RDONLY | O_CLOEXEC);
3748
 
 
3749
 
  bool quit_now = false;
3750
 
  __attribute__((cleanup(cleanup_queue)))
3751
 
    task_queue *queue = create_queue();
3752
 
  g_assert_nonnull(queue);
3753
 
 
3754
 
  task_context task = {
3755
 
    .func=read_inotify_event,
3756
 
    .epoll_fd=epoll_fd,
3757
 
    .fd=fd,
3758
 
    .quit_now=&quit_now,
3759
 
    .filename=strdup("/nonexistent"),
3760
 
    .cancelled_filenames = &(string_set){},
3761
 
    .notafter=0,
3762
 
    .current_time=&current_time,
3763
 
  };
3764
 
  g_assert_nonnull(task.filename);
3765
 
  run_task_with_stderr_to_dev_null(task, queue);
3766
 
  g_assert_true(quit_now);
3767
 
  g_assert_true(queue->next_run == 0);
3768
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3769
 
 
3770
 
  g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));
3771
 
 
3772
 
  g_assert_cmpint(close(fd), ==, -1);
3773
 
}
3774
 
 
3775
 
static void test_read_inotify_event_bad_epoll(__attribute__((unused))
3776
 
                                              test_fixture *fixture,
3777
 
                                              __attribute__((unused))
3778
 
                                              gconstpointer
3779
 
                                              user_data){
3780
 
  const mono_microsecs current_time = 17;
3781
 
 
3782
 
  int pipefds[2];
3783
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3784
 
  const int epoll_fd = pipefds[0]; /* This will obviously fail */
3785
 
 
3786
 
  bool quit_now = false;
3787
 
  buffer password = {};
3788
 
  bool mandos_client_exited = false;
3789
 
  bool password_is_read = false;
3790
 
  __attribute__((cleanup(cleanup_queue)))
3791
 
    task_queue *queue = create_queue();
3792
 
  g_assert_nonnull(queue);
3793
 
 
3794
 
  task_context task = {
3795
 
    .func=read_inotify_event,
3796
 
    .epoll_fd=epoll_fd,
3797
 
    .fd=pipefds[0],
3798
 
    .quit_now=&quit_now,
3799
 
    .password=&password,
3800
 
    .filename=strdup("/nonexistent"),
3801
 
    .cancelled_filenames = &(string_set){},
3802
 
    .notafter=0,
3803
 
    .current_time=&current_time,
3804
 
    .mandos_client_exited=&mandos_client_exited,
3805
 
    .password_is_read=&password_is_read,
3806
 
  };
3807
 
  g_assert_nonnull(task.filename);
3808
 
  run_task_with_stderr_to_dev_null(task, queue);
3809
 
 
3810
 
  g_assert_nonnull(find_matching_task(queue, task));
3811
 
  g_assert_true(queue->next_run == 1000000 + current_time);
3812
 
 
3813
 
  g_assert_cmpint(close(pipefds[0]), ==, 0);
3814
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3815
 
}
3816
 
 
3817
 
static void test_read_inotify_event_nodata(__attribute__((unused))
3818
 
                                           test_fixture *fixture,
3819
 
                                           __attribute__((unused))
3820
 
                                           gconstpointer user_data){
3821
 
  __attribute__((cleanup(cleanup_close)))
3822
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3823
 
  g_assert_cmpint(epoll_fd, >=, 0);
3824
 
  const mono_microsecs current_time = 0;
3825
 
 
3826
 
  int pipefds[2];
3827
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3828
 
 
3829
 
  bool quit_now = false;
3830
 
  buffer password = {};
3831
 
  bool mandos_client_exited = false;
3832
 
  bool password_is_read = false;
3833
 
  __attribute__((cleanup(cleanup_queue)))
3834
 
    task_queue *queue = create_queue();
3835
 
  g_assert_nonnull(queue);
3836
 
 
3837
 
  task_context task = {
3838
 
    .func=read_inotify_event,
3839
 
    .epoll_fd=epoll_fd,
3840
 
    .fd=pipefds[0],
3841
 
    .quit_now=&quit_now,
3842
 
    .password=&password,
3843
 
    .filename=strdup("/nonexistent"),
3844
 
    .cancelled_filenames = &(string_set){},
3845
 
    .notafter=0,
3846
 
    .current_time=&current_time,
3847
 
    .mandos_client_exited=&mandos_client_exited,
3848
 
    .password_is_read=&password_is_read,
3849
 
  };
3850
 
  g_assert_nonnull(task.filename);
3851
 
  task.func(task, queue);
3852
 
  g_assert_false(quit_now);
3853
 
  g_assert_true(queue->next_run == 0);
3854
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
3855
 
 
3856
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3857
 
        .func=read_inotify_event,
3858
 
        .epoll_fd=epoll_fd,
3859
 
        .fd=pipefds[0],
3860
 
        .quit_now=&quit_now,
3861
 
        .password=&password,
3862
 
        .filename=task.filename,
3863
 
        .cancelled_filenames=task.cancelled_filenames,
3864
 
        .current_time=&current_time,
3865
 
        .mandos_client_exited=&mandos_client_exited,
3866
 
        .password_is_read=&password_is_read,
3867
 
      }));
3868
 
 
3869
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3870
 
                                   EPOLLIN | EPOLLRDHUP));
3871
 
 
3872
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3873
 
}
3874
 
 
3875
 
static void test_read_inotify_event_eof(__attribute__((unused))
3876
 
                                        test_fixture *fixture,
3877
 
                                        __attribute__((unused))
3878
 
                                        gconstpointer user_data){
3879
 
  __attribute__((cleanup(cleanup_close)))
3880
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3881
 
  g_assert_cmpint(epoll_fd, >=, 0);
3882
 
  const mono_microsecs current_time = 0;
3883
 
 
3884
 
  int pipefds[2];
3885
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3886
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3887
 
 
3888
 
  bool quit_now = false;
3889
 
  buffer password = {};
3890
 
  __attribute__((cleanup(cleanup_queue)))
3891
 
    task_queue *queue = create_queue();
3892
 
  g_assert_nonnull(queue);
3893
 
 
3894
 
  task_context task = {
3895
 
    .func=read_inotify_event,
3896
 
    .epoll_fd=epoll_fd,
3897
 
    .fd=pipefds[0],
3898
 
    .quit_now=&quit_now,
3899
 
    .password=&password,
3900
 
    .filename=strdup("/nonexistent"),
3901
 
    .cancelled_filenames = &(string_set){},
3902
 
    .notafter=0,
3903
 
    .current_time=&current_time,
3904
 
  };
3905
 
  run_task_with_stderr_to_dev_null(task, queue);
3906
 
  g_assert_true(quit_now);
3907
 
  g_assert_true(queue->next_run == 0);
3908
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
3909
 
 
3910
 
  g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));
3911
 
 
3912
 
  g_assert_cmpint(close(pipefds[0]), ==, -1);
3913
 
}
3914
 
 
3915
 
static
3916
 
void test_read_inotify_event_IN_CLOSE_WRITE(__attribute__((unused))
3917
 
                                            test_fixture *fixture,
3918
 
                                            __attribute__((unused))
3919
 
                                            gconstpointer user_data){
3920
 
  __attribute__((cleanup(cleanup_close)))
3921
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
3922
 
  g_assert_cmpint(epoll_fd, >=, 0);
3923
 
  const mono_microsecs current_time = 0;
3924
 
 
3925
 
  int pipefds[2];
3926
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
3927
 
 
3928
 
  /* "sufficient to read at least one event." - inotify(7) */
3929
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
3930
 
                                  + NAME_MAX + 1);
3931
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
3932
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
3933
 
  struct inotify_event *ievent = ((struct inotify_event *)
3934
 
                                  ievent_buffer);
3935
 
 
3936
 
  const char dummy_file_name[] = "ask.dummy_file_name";
3937
 
  ievent->mask = IN_CLOSE_WRITE;
3938
 
  ievent->len = sizeof(dummy_file_name);
3939
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
3940
 
  const size_t ievent_size = (sizeof(struct inotify_event)
3941
 
                              + sizeof(dummy_file_name));
3942
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
3943
 
                  ==, ievent_size);
3944
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
3945
 
 
3946
 
  bool quit_now = false;
3947
 
  buffer password = {};
3948
 
  bool mandos_client_exited = false;
3949
 
  bool password_is_read = false;
3950
 
  __attribute__((cleanup(cleanup_queue)))
3951
 
    task_queue *queue = create_queue();
3952
 
  g_assert_nonnull(queue);
3953
 
 
3954
 
  task_context task = {
3955
 
    .func=read_inotify_event,
3956
 
    .epoll_fd=epoll_fd,
3957
 
    .fd=pipefds[0],
3958
 
    .quit_now=&quit_now,
3959
 
    .password=&password,
3960
 
    .filename=strdup("/nonexistent"),
3961
 
    .cancelled_filenames = &(string_set){},
3962
 
    .notafter=0,
3963
 
    .current_time=&current_time,
3964
 
    .mandos_client_exited=&mandos_client_exited,
3965
 
    .password_is_read=&password_is_read,
3966
 
  };
3967
 
  task.func(task, queue);
3968
 
  g_assert_false(quit_now);
3969
 
  g_assert_true(queue->next_run != 0);
3970
 
  g_assert_cmpuint((unsigned int)queue->length, >=, 1);
3971
 
 
3972
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3973
 
        .func=read_inotify_event,
3974
 
        .epoll_fd=epoll_fd,
3975
 
        .fd=pipefds[0],
3976
 
        .quit_now=&quit_now,
3977
 
        .password=&password,
3978
 
        .filename=task.filename,
3979
 
        .cancelled_filenames=task.cancelled_filenames,
3980
 
        .current_time=&current_time,
3981
 
        .mandos_client_exited=&mandos_client_exited,
3982
 
        .password_is_read=&password_is_read,
3983
 
      }));
3984
 
 
3985
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
3986
 
                                   EPOLLIN | EPOLLRDHUP));
3987
 
 
3988
 
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
3989
 
 
3990
 
  __attribute__((cleanup(cleanup_string)))
3991
 
    char *filename = NULL;
3992
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
3993
 
                           dummy_file_name), >, 0);
3994
 
  g_assert_nonnull(filename);
3995
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
3996
 
        .func=open_and_parse_question,
3997
 
        .epoll_fd=epoll_fd,
3998
 
        .filename=filename,
3999
 
        .question_filename=filename,
4000
 
        .password=&password,
4001
 
        .cancelled_filenames=task.cancelled_filenames,
4002
 
        .current_time=&current_time,
4003
 
        .mandos_client_exited=&mandos_client_exited,
4004
 
        .password_is_read=&password_is_read,
4005
 
      }));
4006
 
}
4007
 
 
4008
 
static
4009
 
void test_read_inotify_event_IN_MOVED_TO(__attribute__((unused))
4010
 
                                         test_fixture *fixture,
4011
 
                                         __attribute__((unused))
4012
 
                                         gconstpointer user_data){
4013
 
  __attribute__((cleanup(cleanup_close)))
4014
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4015
 
  g_assert_cmpint(epoll_fd, >=, 0);
4016
 
  const mono_microsecs current_time = 0;
4017
 
 
4018
 
  int pipefds[2];
4019
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
4020
 
 
4021
 
  /* "sufficient to read at least one event." - inotify(7) */
4022
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
4023
 
                                  + NAME_MAX + 1);
4024
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
4025
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
4026
 
  struct inotify_event *ievent = ((struct inotify_event *)
4027
 
                                  ievent_buffer);
4028
 
 
4029
 
  const char dummy_file_name[] = "ask.dummy_file_name";
4030
 
  ievent->mask = IN_MOVED_TO;
4031
 
  ievent->len = sizeof(dummy_file_name);
4032
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
4033
 
  const size_t ievent_size = (sizeof(struct inotify_event)
4034
 
                              + sizeof(dummy_file_name));
4035
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
4036
 
                  ==, ievent_size);
4037
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
4038
 
 
4039
 
  bool quit_now = false;
4040
 
  buffer password = {};
4041
 
  bool mandos_client_exited = false;
4042
 
  bool password_is_read = false;
4043
 
  __attribute__((cleanup(cleanup_queue)))
4044
 
    task_queue *queue = create_queue();
4045
 
  g_assert_nonnull(queue);
4046
 
 
4047
 
  task_context task = {
4048
 
    .func=read_inotify_event,
4049
 
    .epoll_fd=epoll_fd,
4050
 
    .fd=pipefds[0],
4051
 
    .quit_now=&quit_now,
4052
 
    .password=&password,
4053
 
    .filename=strdup("/nonexistent"),
4054
 
    .cancelled_filenames = &(string_set){},
4055
 
    .notafter=0,
4056
 
    .current_time=&current_time,
4057
 
    .mandos_client_exited=&mandos_client_exited,
4058
 
    .password_is_read=&password_is_read,
4059
 
  };
4060
 
  task.func(task, queue);
4061
 
  g_assert_false(quit_now);
4062
 
  g_assert_true(queue->next_run != 0);
4063
 
  g_assert_cmpuint((unsigned int)queue->length, >=, 1);
4064
 
 
4065
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4066
 
        .func=read_inotify_event,
4067
 
        .epoll_fd=epoll_fd,
4068
 
        .fd=pipefds[0],
4069
 
        .quit_now=&quit_now,
4070
 
        .password=&password,
4071
 
        .filename=task.filename,
4072
 
        .cancelled_filenames=task.cancelled_filenames,
4073
 
        .current_time=&current_time,
4074
 
        .mandos_client_exited=&mandos_client_exited,
4075
 
        .password_is_read=&password_is_read,
4076
 
      }));
4077
 
 
4078
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
4079
 
                                   EPOLLIN | EPOLLRDHUP));
4080
 
 
4081
 
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
4082
 
 
4083
 
  __attribute__((cleanup(cleanup_string)))
4084
 
    char *filename = NULL;
4085
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
4086
 
                           dummy_file_name), >, 0);
4087
 
  g_assert_nonnull(filename);
4088
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4089
 
        .func=open_and_parse_question,
4090
 
        .epoll_fd=epoll_fd,
4091
 
        .filename=filename,
4092
 
        .question_filename=filename,
4093
 
        .password=&password,
4094
 
        .cancelled_filenames=task.cancelled_filenames,
4095
 
        .current_time=&current_time,
4096
 
        .mandos_client_exited=&mandos_client_exited,
4097
 
        .password_is_read=&password_is_read,
4098
 
      }));
4099
 
}
4100
 
 
4101
 
static void test_read_inotify_event_IN_DELETE(__attribute__((unused))
4102
 
                                              test_fixture *fixture,
4103
 
                                              __attribute__((unused))
4104
 
                                              gconstpointer
4105
 
                                              user_data){
4106
 
  __attribute__((cleanup(cleanup_close)))
4107
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4108
 
  g_assert_cmpint(epoll_fd, >=, 0);
4109
 
  __attribute__((cleanup(string_set_clear)))
4110
 
    string_set cancelled_filenames = {};
4111
 
  const mono_microsecs current_time = 0;
4112
 
 
4113
 
  int pipefds[2];
4114
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
4115
 
 
4116
 
  /* "sufficient to read at least one event." - inotify(7) */
4117
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
4118
 
                                  + NAME_MAX + 1);
4119
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
4120
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
4121
 
  struct inotify_event *ievent = ((struct inotify_event *)
4122
 
                                  ievent_buffer);
4123
 
 
4124
 
  const char dummy_file_name[] = "ask.dummy_file_name";
4125
 
  ievent->mask = IN_DELETE;
4126
 
  ievent->len = sizeof(dummy_file_name);
4127
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
4128
 
  const size_t ievent_size = (sizeof(struct inotify_event)
4129
 
                              + sizeof(dummy_file_name));
4130
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
4131
 
                  ==, ievent_size);
4132
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
4133
 
 
4134
 
  bool quit_now = false;
4135
 
  buffer password = {};
4136
 
  bool mandos_client_exited = false;
4137
 
  bool password_is_read = false;
4138
 
  __attribute__((cleanup(cleanup_queue)))
4139
 
    task_queue *queue = create_queue();
4140
 
  g_assert_nonnull(queue);
4141
 
 
4142
 
  task_context task = {
4143
 
    .func=read_inotify_event,
4144
 
    .epoll_fd=epoll_fd,
4145
 
    .fd=pipefds[0],
4146
 
    .quit_now=&quit_now,
4147
 
    .password=&password,
4148
 
    .filename=strdup("/nonexistent"),
4149
 
    .cancelled_filenames=&cancelled_filenames,
4150
 
    .current_time=&current_time,
4151
 
    .mandos_client_exited=&mandos_client_exited,
4152
 
    .password_is_read=&password_is_read,
4153
 
  };
4154
 
  task.func(task, queue);
4155
 
  g_assert_false(quit_now);
4156
 
  g_assert_true(queue->next_run == 0);
4157
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4158
 
 
4159
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4160
 
        .func=read_inotify_event,
4161
 
        .epoll_fd=epoll_fd,
4162
 
        .fd=pipefds[0],
4163
 
        .quit_now=&quit_now,
4164
 
        .password=&password,
4165
 
        .filename=task.filename,
4166
 
        .cancelled_filenames=&cancelled_filenames,
4167
 
        .current_time=&current_time,
4168
 
        .mandos_client_exited=&mandos_client_exited,
4169
 
        .password_is_read=&password_is_read,
4170
 
      }));
4171
 
 
4172
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
4173
 
                                   EPOLLIN | EPOLLRDHUP));
4174
 
 
4175
 
  __attribute__((cleanup(cleanup_string)))
4176
 
    char *filename = NULL;
4177
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
4178
 
                           dummy_file_name), >, 0);
4179
 
  g_assert_nonnull(filename);
4180
 
  g_assert_true(string_set_contains(*task.cancelled_filenames,
4181
 
                                    filename));
4182
 
}
4183
 
 
4184
 
static void
4185
 
test_read_inotify_event_IN_CLOSE_WRITE_badname(__attribute__((unused))
4186
 
                                               test_fixture *fixture,
4187
 
                                               __attribute__((unused))
4188
 
                                               gconstpointer
4189
 
                                               user_data){
4190
 
  __attribute__((cleanup(cleanup_close)))
4191
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4192
 
  g_assert_cmpint(epoll_fd, >=, 0);
4193
 
  const mono_microsecs current_time = 0;
4194
 
 
4195
 
  int pipefds[2];
4196
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
4197
 
 
4198
 
  /* "sufficient to read at least one event." - inotify(7) */
4199
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
4200
 
                                  + NAME_MAX + 1);
4201
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
4202
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
4203
 
  struct inotify_event *ievent = ((struct inotify_event *)
4204
 
                                  ievent_buffer);
4205
 
 
4206
 
  const char dummy_file_name[] = "ignored.dummy_file_name";
4207
 
  ievent->mask = IN_CLOSE_WRITE;
4208
 
  ievent->len = sizeof(dummy_file_name);
4209
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
4210
 
  const size_t ievent_size = (sizeof(struct inotify_event)
4211
 
                              + sizeof(dummy_file_name));
4212
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
4213
 
                  ==, ievent_size);
4214
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
4215
 
 
4216
 
  bool quit_now = false;
4217
 
  buffer password = {};
4218
 
  bool mandos_client_exited = false;
4219
 
  bool password_is_read = false;
4220
 
  __attribute__((cleanup(cleanup_queue)))
4221
 
    task_queue *queue = create_queue();
4222
 
  g_assert_nonnull(queue);
4223
 
 
4224
 
  task_context task = {
4225
 
    .func=read_inotify_event,
4226
 
    .epoll_fd=epoll_fd,
4227
 
    .fd=pipefds[0],
4228
 
    .quit_now=&quit_now,
4229
 
    .password=&password,
4230
 
    .filename=strdup("/nonexistent"),
4231
 
    .cancelled_filenames = &(string_set){},
4232
 
    .notafter=0,
4233
 
    .current_time=&current_time,
4234
 
    .mandos_client_exited=&mandos_client_exited,
4235
 
    .password_is_read=&password_is_read,
4236
 
  };
4237
 
  task.func(task, queue);
4238
 
  g_assert_false(quit_now);
4239
 
  g_assert_true(queue->next_run == 0);
4240
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4241
 
 
4242
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4243
 
        .func=read_inotify_event,
4244
 
        .epoll_fd=epoll_fd,
4245
 
        .fd=pipefds[0],
4246
 
        .quit_now=&quit_now,
4247
 
        .password=&password,
4248
 
        .filename=task.filename,
4249
 
        .cancelled_filenames=task.cancelled_filenames,
4250
 
        .current_time=&current_time,
4251
 
        .mandos_client_exited=&mandos_client_exited,
4252
 
        .password_is_read=&password_is_read,
4253
 
      }));
4254
 
 
4255
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
4256
 
                                   EPOLLIN | EPOLLRDHUP));
4257
 
}
4258
 
 
4259
 
static void
4260
 
test_read_inotify_event_IN_MOVED_TO_badname(__attribute__((unused))
4261
 
                                            test_fixture *fixture,
4262
 
                                            __attribute__((unused))
4263
 
                                            gconstpointer user_data){
4264
 
  __attribute__((cleanup(cleanup_close)))
4265
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4266
 
  g_assert_cmpint(epoll_fd, >=, 0);
4267
 
  const mono_microsecs current_time = 0;
4268
 
 
4269
 
  int pipefds[2];
4270
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
4271
 
 
4272
 
  /* "sufficient to read at least one event." - inotify(7) */
4273
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
4274
 
                                  + NAME_MAX + 1);
4275
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
4276
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
4277
 
  struct inotify_event *ievent = ((struct inotify_event *)
4278
 
                                  ievent_buffer);
4279
 
 
4280
 
  const char dummy_file_name[] = "ignored.dummy_file_name";
4281
 
  ievent->mask = IN_MOVED_TO;
4282
 
  ievent->len = sizeof(dummy_file_name);
4283
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
4284
 
  const size_t ievent_size = (sizeof(struct inotify_event)
4285
 
                              + sizeof(dummy_file_name));
4286
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
4287
 
                  ==, ievent_size);
4288
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
4289
 
 
4290
 
  bool quit_now = false;
4291
 
  buffer password = {};
4292
 
  bool mandos_client_exited = false;
4293
 
  bool password_is_read = false;
4294
 
  __attribute__((cleanup(cleanup_queue)))
4295
 
    task_queue *queue = create_queue();
4296
 
  g_assert_nonnull(queue);
4297
 
 
4298
 
  task_context task = {
4299
 
    .func=read_inotify_event,
4300
 
    .epoll_fd=epoll_fd,
4301
 
    .fd=pipefds[0],
4302
 
    .quit_now=&quit_now,
4303
 
    .password=&password,
4304
 
    .filename=strdup("/nonexistent"),
4305
 
    .cancelled_filenames = &(string_set){},
4306
 
    .notafter=0,
4307
 
    .current_time=&current_time,
4308
 
    .mandos_client_exited=&mandos_client_exited,
4309
 
    .password_is_read=&password_is_read,
4310
 
  };
4311
 
  task.func(task, queue);
4312
 
  g_assert_false(quit_now);
4313
 
  g_assert_true(queue->next_run == 0);
4314
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4315
 
 
4316
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4317
 
        .func=read_inotify_event,
4318
 
        .epoll_fd=epoll_fd,
4319
 
        .fd=pipefds[0],
4320
 
        .quit_now=&quit_now,
4321
 
        .password=&password,
4322
 
        .filename=task.filename,
4323
 
        .cancelled_filenames=task.cancelled_filenames,
4324
 
        .current_time=&current_time,
4325
 
        .mandos_client_exited=&mandos_client_exited,
4326
 
        .password_is_read=&password_is_read,
4327
 
      }));
4328
 
 
4329
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
4330
 
                                   EPOLLIN | EPOLLRDHUP));
4331
 
}
4332
 
 
4333
 
static
4334
 
void test_read_inotify_event_IN_DELETE_badname(__attribute__((unused))
4335
 
                                               test_fixture *fixture,
4336
 
                                               __attribute__((unused))
4337
 
                                               gconstpointer
4338
 
                                               user_data){
4339
 
  __attribute__((cleanup(cleanup_close)))
4340
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4341
 
  g_assert_cmpint(epoll_fd, >=, 0);
4342
 
  __attribute__((cleanup(string_set_clear)))
4343
 
    string_set cancelled_filenames = {};
4344
 
  const mono_microsecs current_time = 0;
4345
 
 
4346
 
  int pipefds[2];
4347
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
4348
 
 
4349
 
  /* "sufficient to read at least one event." - inotify(7) */
4350
 
  const size_t ievent_max_size = (sizeof(struct inotify_event)
4351
 
                                  + NAME_MAX + 1);
4352
 
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
4353
 
  char ievent_buffer[sizeof(struct inotify_event) + NAME_MAX + 1];
4354
 
  struct inotify_event *ievent = ((struct inotify_event *)
4355
 
                                  ievent_buffer);
4356
 
 
4357
 
  const char dummy_file_name[] = "ignored.dummy_file_name";
4358
 
  ievent->mask = IN_DELETE;
4359
 
  ievent->len = sizeof(dummy_file_name);
4360
 
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
4361
 
  const size_t ievent_size = (sizeof(struct inotify_event)
4362
 
                              + sizeof(dummy_file_name));
4363
 
  g_assert_cmpint(write(pipefds[1], ievent_buffer, ievent_size),
4364
 
                  ==, ievent_size);
4365
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
4366
 
 
4367
 
  bool quit_now = false;
4368
 
  buffer password = {};
4369
 
  bool mandos_client_exited = false;
4370
 
  bool password_is_read = false;
4371
 
  __attribute__((cleanup(cleanup_queue)))
4372
 
    task_queue *queue = create_queue();
4373
 
  g_assert_nonnull(queue);
4374
 
 
4375
 
  task_context task = {
4376
 
    .func=read_inotify_event,
4377
 
    .epoll_fd=epoll_fd,
4378
 
    .fd=pipefds[0],
4379
 
    .quit_now=&quit_now,
4380
 
    .password=&password,
4381
 
    .filename=strdup("/nonexistent"),
4382
 
    .cancelled_filenames=&cancelled_filenames,
4383
 
    .current_time=&current_time,
4384
 
    .mandos_client_exited=&mandos_client_exited,
4385
 
    .password_is_read=&password_is_read,
4386
 
  };
4387
 
  task.func(task, queue);
4388
 
  g_assert_false(quit_now);
4389
 
  g_assert_true(queue->next_run == 0);
4390
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4391
 
 
4392
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4393
 
        .func=read_inotify_event,
4394
 
        .epoll_fd=epoll_fd,
4395
 
        .fd=pipefds[0],
4396
 
        .quit_now=&quit_now,
4397
 
        .password=&password,
4398
 
        .filename=task.filename,
4399
 
        .cancelled_filenames=&cancelled_filenames,
4400
 
        .current_time=&current_time,
4401
 
        .mandos_client_exited=&mandos_client_exited,
4402
 
        .password_is_read=&password_is_read,
4403
 
      }));
4404
 
 
4405
 
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
4406
 
                                   EPOLLIN | EPOLLRDHUP));
4407
 
 
4408
 
  __attribute__((cleanup(cleanup_string)))
4409
 
    char *filename = NULL;
4410
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
4411
 
                           dummy_file_name), >, 0);
4412
 
  g_assert_nonnull(filename);
4413
 
  g_assert_false(string_set_contains(cancelled_filenames, filename));
4414
 
}
4415
 
 
4416
 
static
4417
 
void test_open_and_parse_question_ENOENT(__attribute__((unused))
4418
 
                                         test_fixture *fixture,
4419
 
                                         __attribute__((unused))
4420
 
                                         gconstpointer 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
 
  bool mandos_client_exited = false;
4427
 
  bool password_is_read = false;
4428
 
  __attribute__((cleanup(cleanup_queue)))
4429
 
    task_queue *queue = create_queue();
4430
 
  g_assert_nonnull(queue);
4431
 
 
4432
 
  char *const filename = strdup("/nonexistent");
4433
 
  g_assert_nonnull(filename);
4434
 
  task_context task = {
4435
 
    .func=open_and_parse_question,
4436
 
    .question_filename=filename,
4437
 
    .epoll_fd=epoll_fd,
4438
 
    .password=(buffer[]){{}},
4439
 
    .filename=filename,
4440
 
    .cancelled_filenames=&cancelled_filenames,
4441
 
    .current_time=(mono_microsecs[]){0},
4442
 
    .mandos_client_exited=&mandos_client_exited,
4443
 
    .password_is_read=&password_is_read,
4444
 
  };
4445
 
  task.func(task, queue);
4446
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4447
 
}
4448
 
 
4449
 
static void test_open_and_parse_question_EIO(__attribute__((unused))
4450
 
                                             test_fixture *fixture,
4451
 
                                             __attribute__((unused))
4452
 
                                             gconstpointer user_data){
4453
 
  __attribute__((cleanup(cleanup_close)))
4454
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4455
 
  g_assert_cmpint(epoll_fd, >=, 0);
4456
 
  __attribute__((cleanup(string_set_clear)))
4457
 
    string_set cancelled_filenames = {};
4458
 
  buffer password = {};
4459
 
  bool mandos_client_exited = false;
4460
 
  bool password_is_read = false;
4461
 
  __attribute__((cleanup(cleanup_queue)))
4462
 
    task_queue *queue = create_queue();
4463
 
  g_assert_nonnull(queue);
4464
 
  const mono_microsecs current_time = 0;
4465
 
 
4466
 
  char *filename = strdup("/proc/self/mem");
4467
 
  task_context task = {
4468
 
    .func=open_and_parse_question,
4469
 
    .question_filename=filename,
4470
 
    .epoll_fd=epoll_fd,
4471
 
    .password=&password,
4472
 
    .filename=filename,
4473
 
    .cancelled_filenames=&cancelled_filenames,
4474
 
    .current_time=&current_time,
4475
 
    .mandos_client_exited=&mandos_client_exited,
4476
 
    .password_is_read=&password_is_read,
4477
 
  };
4478
 
  run_task_with_stderr_to_dev_null(task, queue);
4479
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4480
 
}
4481
 
 
4482
 
static void
4483
 
test_open_and_parse_question_parse_error(__attribute__((unused))
4484
 
                                         test_fixture *fixture,
4485
 
                                         __attribute__((unused))
4486
 
                                         gconstpointer user_data){
4487
 
  __attribute__((cleanup(cleanup_close)))
4488
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4489
 
  g_assert_cmpint(epoll_fd, >=, 0);
4490
 
  __attribute__((cleanup(string_set_clear)))
4491
 
    string_set cancelled_filenames = {};
4492
 
  __attribute__((cleanup(cleanup_queue)))
4493
 
    task_queue *queue = create_queue();
4494
 
  g_assert_nonnull(queue);
4495
 
 
4496
 
  __attribute__((cleanup(cleanup_string)))
4497
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4498
 
  g_assert_nonnull(tempfilename);
4499
 
  int tempfile = mkostemp(tempfilename, O_CLOEXEC);
4500
 
  g_assert_cmpint(tempfile, >, 0);
4501
 
  const char bad_data[] = "this is bad syntax\n";
4502
 
  g_assert_cmpint(write(tempfile, bad_data, sizeof(bad_data)),
4503
 
                  ==, sizeof(bad_data));
4504
 
  g_assert_cmpint(close(tempfile), ==, 0);
4505
 
 
4506
 
  char *const filename = strdup(tempfilename);
4507
 
  g_assert_nonnull(filename);
4508
 
  task_context task = {
4509
 
    .func=open_and_parse_question,
4510
 
    .question_filename=filename,
4511
 
    .epoll_fd=epoll_fd,
4512
 
    .password=(buffer[]){{}},
4513
 
    .filename=filename,
4514
 
    .cancelled_filenames=&cancelled_filenames,
4515
 
    .current_time=(mono_microsecs[]){0},
4516
 
    .mandos_client_exited=(bool[]){false},
4517
 
    .password_is_read=(bool[]){false},
4518
 
  };
4519
 
  run_task_with_stderr_to_dev_null(task, queue);
4520
 
 
4521
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4522
 
 
4523
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4524
 
}
4525
 
 
4526
 
static
4527
 
void test_open_and_parse_question_nosocket(__attribute__((unused))
4528
 
                                           test_fixture *fixture,
4529
 
                                           __attribute__((unused))
4530
 
                                           gconstpointer user_data){
4531
 
  __attribute__((cleanup(cleanup_close)))
4532
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4533
 
  g_assert_cmpint(epoll_fd, >=, 0);
4534
 
  __attribute__((cleanup(string_set_clear)))
4535
 
    string_set cancelled_filenames = {};
4536
 
  __attribute__((cleanup(cleanup_queue)))
4537
 
    task_queue *queue = create_queue();
4538
 
  g_assert_nonnull(queue);
4539
 
 
4540
 
  __attribute__((cleanup(cleanup_string)))
4541
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4542
 
  g_assert_nonnull(tempfilename);
4543
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4544
 
  g_assert_cmpint(questionfile, >, 0);
4545
 
  FILE *qf = fdopen(questionfile, "w");
4546
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nPID=1\n"), >, 0);
4547
 
  g_assert_cmpint(fclose(qf), ==, 0);
4548
 
 
4549
 
  char *const filename = strdup(tempfilename);
4550
 
  g_assert_nonnull(filename);
4551
 
  task_context task = {
4552
 
    .func=open_and_parse_question,
4553
 
    .question_filename=filename,
4554
 
    .epoll_fd=epoll_fd,
4555
 
    .password=(buffer[]){{}},
4556
 
    .filename=filename,
4557
 
    .cancelled_filenames=&cancelled_filenames,
4558
 
    .current_time=(mono_microsecs[]){0},
4559
 
    .mandos_client_exited=(bool[]){false},
4560
 
    .password_is_read=(bool[]){false},
4561
 
  };
4562
 
  run_task_with_stderr_to_dev_null(task, queue);
4563
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4564
 
 
4565
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4566
 
}
4567
 
 
4568
 
static
4569
 
void test_open_and_parse_question_badsocket(__attribute__((unused))
4570
 
                                            test_fixture *fixture,
4571
 
                                            __attribute__((unused))
4572
 
                                            gconstpointer user_data){
4573
 
  __attribute__((cleanup(cleanup_close)))
4574
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4575
 
  g_assert_cmpint(epoll_fd, >=, 0);
4576
 
  __attribute__((cleanup(string_set_clear)))
4577
 
    string_set cancelled_filenames = {};
4578
 
  __attribute__((cleanup(cleanup_queue)))
4579
 
    task_queue *queue = create_queue();
4580
 
  g_assert_nonnull(queue);
4581
 
 
4582
 
  __attribute__((cleanup(cleanup_string)))
4583
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4584
 
  g_assert_nonnull(tempfilename);
4585
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4586
 
  g_assert_cmpint(questionfile, >, 0);
4587
 
  FILE *qf = fdopen(questionfile, "w");
4588
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=\nPID=1\n"), >, 0);
4589
 
  g_assert_cmpint(fclose(qf), ==, 0);
4590
 
 
4591
 
  char *const filename = strdup(tempfilename);
4592
 
  g_assert_nonnull(filename);
4593
 
  task_context task = {
4594
 
    .func=open_and_parse_question,
4595
 
    .question_filename=filename,
4596
 
    .epoll_fd=epoll_fd,
4597
 
    .password=(buffer[]){{}},
4598
 
    .filename=filename,
4599
 
    .cancelled_filenames=&cancelled_filenames,
4600
 
    .current_time=(mono_microsecs[]){0},
4601
 
    .mandos_client_exited=(bool[]){false},
4602
 
    .password_is_read=(bool[]){false},
4603
 
  };
4604
 
  run_task_with_stderr_to_dev_null(task, queue);
4605
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4606
 
 
4607
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4608
 
}
4609
 
 
4610
 
static
4611
 
void test_open_and_parse_question_nopid(__attribute__((unused))
4612
 
                                        test_fixture *fixture,
4613
 
                                        __attribute__((unused))
4614
 
                                        gconstpointer user_data){
4615
 
  __attribute__((cleanup(cleanup_close)))
4616
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4617
 
  g_assert_cmpint(epoll_fd, >=, 0);
4618
 
  __attribute__((cleanup(string_set_clear)))
4619
 
    string_set cancelled_filenames = {};
4620
 
  __attribute__((cleanup(cleanup_queue)))
4621
 
    task_queue *queue = create_queue();
4622
 
  g_assert_nonnull(queue);
4623
 
 
4624
 
  __attribute__((cleanup(cleanup_string)))
4625
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4626
 
  g_assert_nonnull(tempfilename);
4627
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4628
 
  g_assert_cmpint(questionfile, >, 0);
4629
 
  FILE *qf = fdopen(questionfile, "w");
4630
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\n"), >, 0);
4631
 
  g_assert_cmpint(fclose(qf), ==, 0);
4632
 
 
4633
 
  char *const filename = strdup(tempfilename);
4634
 
  g_assert_nonnull(filename);
4635
 
  task_context task = {
4636
 
    .func=open_and_parse_question,
4637
 
    .question_filename=filename,
4638
 
    .epoll_fd=epoll_fd,
4639
 
    .password=(buffer[]){{}},
4640
 
    .filename=filename,
4641
 
    .cancelled_filenames=&cancelled_filenames,
4642
 
    .current_time=(mono_microsecs[]){0},
4643
 
    .mandos_client_exited=(bool[]){false},
4644
 
    .password_is_read=(bool[]){false},
4645
 
  };
4646
 
  run_task_with_stderr_to_dev_null(task, queue);
4647
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4648
 
 
4649
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4650
 
}
4651
 
 
4652
 
static
4653
 
void test_open_and_parse_question_badpid(__attribute__((unused))
4654
 
                                         test_fixture *fixture,
4655
 
                                         __attribute__((unused))
4656
 
                                         gconstpointer user_data){
4657
 
  __attribute__((cleanup(cleanup_close)))
4658
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4659
 
  g_assert_cmpint(epoll_fd, >=, 0);
4660
 
  __attribute__((cleanup(string_set_clear)))
4661
 
    string_set cancelled_filenames = {};
4662
 
  __attribute__((cleanup(cleanup_queue)))
4663
 
    task_queue *queue = create_queue();
4664
 
  g_assert_nonnull(queue);
4665
 
 
4666
 
  __attribute__((cleanup(cleanup_string)))
4667
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4668
 
  g_assert_nonnull(tempfilename);
4669
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4670
 
  g_assert_cmpint(questionfile, >, 0);
4671
 
  FILE *qf = fdopen(questionfile, "w");
4672
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=\n"),
4673
 
                  >, 0);
4674
 
  g_assert_cmpint(fclose(qf), ==, 0);
4675
 
 
4676
 
  char *const filename = strdup(tempfilename);
4677
 
  g_assert_nonnull(filename);
4678
 
  task_context task = {
4679
 
    .func=open_and_parse_question,
4680
 
    .question_filename=filename,
4681
 
    .epoll_fd=epoll_fd,
4682
 
    .password=(buffer[]){{}},
4683
 
    .filename=filename,
4684
 
    .cancelled_filenames=&cancelled_filenames,
4685
 
    .current_time=(mono_microsecs[]){0},
4686
 
    .mandos_client_exited=(bool[]){false},
4687
 
    .password_is_read=(bool[]){false},
4688
 
  };
4689
 
  run_task_with_stderr_to_dev_null(task, queue);
4690
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4691
 
 
4692
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4693
 
}
4694
 
 
4695
 
static void
4696
 
test_open_and_parse_question_noexist_pid(__attribute__((unused))
4697
 
                                         test_fixture *fixture,
4698
 
                                         __attribute__((unused))
4699
 
                                         gconstpointer user_data){
4700
 
  __attribute__((cleanup(cleanup_close)))
4701
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4702
 
  g_assert_cmpint(epoll_fd, >=, 0);
4703
 
  __attribute__((cleanup(string_set_clear)))
4704
 
    string_set cancelled_filenames = {};
4705
 
  buffer password = {};
4706
 
  bool mandos_client_exited = false;
4707
 
  bool password_is_read = false;
4708
 
  __attribute__((cleanup(cleanup_queue)))
4709
 
    task_queue *queue = create_queue();
4710
 
  g_assert_nonnull(queue);
4711
 
  const mono_microsecs current_time = 0;
4712
 
 
4713
 
  /* Find value of sysctl kernel.pid_max */
4714
 
  uintmax_t pid_max = 0;
4715
 
  FILE *sysctl_pid_max = fopen("/proc/sys/kernel/pid_max", "r");
4716
 
  g_assert_nonnull(sysctl_pid_max);
4717
 
  g_assert_cmpint(fscanf(sysctl_pid_max, "%" PRIuMAX, &pid_max),
4718
 
                  ==, 1);
4719
 
  g_assert_cmpint(fclose(sysctl_pid_max), ==, 0);
4720
 
 
4721
 
  pid_t nonexisting_pid = ((pid_t)pid_max)+1;
4722
 
  g_assert_true(nonexisting_pid > 0); /* Overflow check */
4723
 
 
4724
 
  __attribute__((cleanup(cleanup_string)))
4725
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4726
 
  g_assert_nonnull(tempfilename);
4727
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4728
 
  g_assert_cmpint(questionfile, >, 0);
4729
 
  FILE *qf = fdopen(questionfile, "w");
4730
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
4731
 
                          PRIuMAX"\n", (uintmax_t)nonexisting_pid),
4732
 
                  >, 0);
4733
 
  g_assert_cmpint(fclose(qf), ==, 0);
4734
 
 
4735
 
  char *const question_filename = strdup(tempfilename);
4736
 
  g_assert_nonnull(question_filename);
4737
 
  task_context task = {
4738
 
    .func=open_and_parse_question,
4739
 
    .question_filename=question_filename,
4740
 
    .epoll_fd=epoll_fd,
4741
 
    .password=&password,
4742
 
    .filename=question_filename,
4743
 
    .cancelled_filenames=&cancelled_filenames,
4744
 
    .current_time=&current_time,
4745
 
    .mandos_client_exited=&mandos_client_exited,
4746
 
    .password_is_read=&password_is_read,
4747
 
  };
4748
 
  run_task_with_stderr_to_dev_null(task, queue);
4749
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4750
 
 
4751
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4752
 
}
4753
 
 
4754
 
static void
4755
 
test_open_and_parse_question_no_notafter(__attribute__((unused))
4756
 
                                         test_fixture *fixture,
4757
 
                                         __attribute__((unused))
4758
 
                                         gconstpointer user_data){
4759
 
  __attribute__((cleanup(cleanup_close)))
4760
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4761
 
  g_assert_cmpint(epoll_fd, >=, 0);
4762
 
  __attribute__((cleanup(string_set_clear)))
4763
 
    string_set cancelled_filenames = {};
4764
 
  buffer password = {};
4765
 
  bool mandos_client_exited = false;
4766
 
  bool password_is_read = false;
4767
 
  __attribute__((cleanup(cleanup_queue)))
4768
 
    task_queue *queue = create_queue();
4769
 
  g_assert_nonnull(queue);
4770
 
  const mono_microsecs current_time = 0;
4771
 
 
4772
 
  __attribute__((cleanup(cleanup_string)))
4773
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4774
 
  g_assert_nonnull(tempfilename);
4775
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4776
 
  g_assert_cmpint(questionfile, >, 0);
4777
 
  FILE *qf = fdopen(questionfile, "w");
4778
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
4779
 
                          PRIuMAX "\n", (uintmax_t)getpid()), >, 0);
4780
 
  g_assert_cmpint(fclose(qf), ==, 0);
4781
 
 
4782
 
  char *const filename = strdup(tempfilename);
4783
 
  g_assert_nonnull(filename);
4784
 
  task_context task = {
4785
 
    .func=open_and_parse_question,
4786
 
    .question_filename=filename,
4787
 
    .epoll_fd=epoll_fd,
4788
 
    .password=&password,
4789
 
    .filename=filename,
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_cmpuint((unsigned int)queue->length, ==, 1);
4797
 
 
4798
 
  __attribute__((cleanup(cleanup_string)))
4799
 
    char *socket_filename = strdup("/nonexistent");
4800
 
  g_assert_nonnull(socket_filename);
4801
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4802
 
        .func=connect_question_socket,
4803
 
        .question_filename=tempfilename,
4804
 
        .filename=socket_filename,
4805
 
        .epoll_fd=epoll_fd,
4806
 
        .password=&password,
4807
 
        .current_time=&current_time,
4808
 
        .mandos_client_exited=&mandos_client_exited,
4809
 
        .password_is_read=&password_is_read,
4810
 
      }));
4811
 
 
4812
 
  g_assert_true(queue->next_run != 0);
4813
 
 
4814
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4815
 
}
4816
 
 
4817
 
static void
4818
 
test_open_and_parse_question_bad_notafter(__attribute__((unused))
4819
 
                                          test_fixture *fixture,
4820
 
                                          __attribute__((unused))
4821
 
                                          gconstpointer user_data){
4822
 
  __attribute__((cleanup(cleanup_close)))
4823
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4824
 
  g_assert_cmpint(epoll_fd, >=, 0);
4825
 
  __attribute__((cleanup(string_set_clear)))
4826
 
    string_set cancelled_filenames = {};
4827
 
  buffer password = {};
4828
 
  bool mandos_client_exited = false;
4829
 
  bool password_is_read = false;
4830
 
  __attribute__((cleanup(cleanup_queue)))
4831
 
    task_queue *queue = create_queue();
4832
 
  g_assert_nonnull(queue);
4833
 
  const mono_microsecs current_time = 0;
4834
 
 
4835
 
  __attribute__((cleanup(cleanup_string)))
4836
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4837
 
  g_assert_nonnull(tempfilename);
4838
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4839
 
  g_assert_cmpint(questionfile, >, 0);
4840
 
  FILE *qf = fdopen(questionfile, "w");
4841
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
4842
 
                          PRIuMAX "\nNotAfter=\n",
4843
 
                          (uintmax_t)getpid()), >, 0);
4844
 
  g_assert_cmpint(fclose(qf), ==, 0);
4845
 
 
4846
 
  char *const filename = strdup(tempfilename);
4847
 
  g_assert_nonnull(filename);
4848
 
  task_context task = {
4849
 
    .func=open_and_parse_question,
4850
 
    .question_filename=filename,
4851
 
    .epoll_fd=epoll_fd,
4852
 
    .password=&password,
4853
 
    .filename=filename,
4854
 
    .cancelled_filenames=&cancelled_filenames,
4855
 
    .current_time=&current_time,
4856
 
    .mandos_client_exited=&mandos_client_exited,
4857
 
    .password_is_read=&password_is_read,
4858
 
  };
4859
 
  run_task_with_stderr_to_dev_null(task, queue);
4860
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4861
 
 
4862
 
  __attribute__((cleanup(cleanup_string)))
4863
 
    char *socket_filename = strdup("/nonexistent");
4864
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
4865
 
        .func=connect_question_socket,
4866
 
        .question_filename=tempfilename,
4867
 
        .filename=socket_filename,
4868
 
        .epoll_fd=epoll_fd,
4869
 
        .password=&password,
4870
 
        .current_time=&current_time,
4871
 
        .mandos_client_exited=&mandos_client_exited,
4872
 
        .password_is_read=&password_is_read,
4873
 
      }));
4874
 
  g_assert_true(queue->next_run != 0);
4875
 
 
4876
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4877
 
}
4878
 
 
4879
 
static
4880
 
void assert_open_and_parse_question_with_notafter(const mono_microsecs
4881
 
                                                  current_time,
4882
 
                                                  const mono_microsecs
4883
 
                                                  notafter,
4884
 
                                                  const mono_microsecs
4885
 
                                                  next_queue_run){
4886
 
  __attribute__((cleanup(cleanup_close)))
4887
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
4888
 
  g_assert_cmpint(epoll_fd, >=, 0);
4889
 
  __attribute__((cleanup(string_set_clear)))
4890
 
    string_set cancelled_filenames = {};
4891
 
  buffer password = {};
4892
 
  bool mandos_client_exited = false;
4893
 
  bool password_is_read = false;
4894
 
  __attribute__((cleanup(cleanup_queue)))
4895
 
    task_queue *queue = create_queue();
4896
 
  g_assert_nonnull(queue);
4897
 
  queue->next_run = next_queue_run;
4898
 
 
4899
 
  __attribute__((cleanup(cleanup_string)))
4900
 
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
4901
 
  g_assert_nonnull(tempfilename);
4902
 
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
4903
 
  g_assert_cmpint(questionfile, >, 0);
4904
 
  FILE *qf = fdopen(questionfile, "w");
4905
 
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
4906
 
                          PRIuMAX "\nNotAfter=%" PRIuMAX "\n",
4907
 
                          (uintmax_t)getpid(), notafter), >, 0);
4908
 
  g_assert_cmpint(fclose(qf), ==, 0);
4909
 
 
4910
 
  char *const filename = strdup(tempfilename);
4911
 
  g_assert_nonnull(filename);
4912
 
  task_context task = {
4913
 
    .func=open_and_parse_question,
4914
 
    .question_filename=filename,
4915
 
    .epoll_fd=epoll_fd,
4916
 
    .password=&password,
4917
 
    .filename=filename,
4918
 
    .cancelled_filenames=&cancelled_filenames,
4919
 
    .current_time=&current_time,
4920
 
    .mandos_client_exited=&mandos_client_exited,
4921
 
    .password_is_read=&password_is_read,
4922
 
  };
4923
 
  task.func(task, queue);
4924
 
 
4925
 
  if(queue->length >= 1){
4926
 
    __attribute__((cleanup(cleanup_string)))
4927
 
      char *socket_filename = strdup("/nonexistent");
4928
 
    g_assert_nonnull(find_matching_task(queue, (task_context){
4929
 
          .func=connect_question_socket,
4930
 
          .filename=socket_filename,
4931
 
          .epoll_fd=epoll_fd,
4932
 
          .password=&password,
4933
 
          .current_time=&current_time,
4934
 
          .cancelled_filenames=&cancelled_filenames,
4935
 
          .mandos_client_exited=&mandos_client_exited,
4936
 
          .password_is_read=&password_is_read,
4937
 
        }));
4938
 
    g_assert_true(queue->next_run != 0);
4939
 
  }
4940
 
 
4941
 
  if(notafter == 0){
4942
 
    g_assert_cmpuint((unsigned int)queue->length, ==, 1);
4943
 
  } else if(current_time >= notafter) {
4944
 
    g_assert_cmpuint((unsigned int)queue->length, ==, 0);
4945
 
  } else {
4946
 
    g_assert_nonnull(find_matching_task(queue, (task_context){
4947
 
          .func=cancel_old_question,
4948
 
          .question_filename=tempfilename,
4949
 
          .filename=tempfilename,
4950
 
          .notafter=notafter,
4951
 
          .cancelled_filenames=&cancelled_filenames,
4952
 
          .current_time=&current_time,
4953
 
        }));
4954
 
  }
4955
 
  g_assert_true(queue->next_run == 1);
4956
 
 
4957
 
  g_assert_cmpint(unlink(tempfilename), ==, 0);
4958
 
}
4959
 
 
4960
 
static void
4961
 
test_open_and_parse_question_notafter_0(__attribute__((unused))
4962
 
                                        test_fixture *fixture,
4963
 
                                        __attribute__((unused))
4964
 
                                        gconstpointer user_data){
4965
 
  /* current_time, notafter, next_queue_run */
4966
 
  assert_open_and_parse_question_with_notafter(0, 0, 0);
4967
 
}
4968
 
 
4969
 
static void
4970
 
test_open_and_parse_question_notafter_1(__attribute__((unused))
4971
 
                                        test_fixture *fixture,
4972
 
                                        __attribute__((unused))
4973
 
                                        gconstpointer user_data){
4974
 
  /* current_time, notafter, next_queue_run */
4975
 
  assert_open_and_parse_question_with_notafter(0, 1, 0);
4976
 
}
4977
 
 
4978
 
static void
4979
 
test_open_and_parse_question_notafter_1_1(__attribute__((unused))
4980
 
                                          test_fixture *fixture,
4981
 
                                          __attribute__((unused))
4982
 
                                          gconstpointer user_data){
4983
 
  /* current_time, notafter, next_queue_run */
4984
 
  assert_open_and_parse_question_with_notafter(0, 1, 1);
4985
 
}
4986
 
 
4987
 
static void
4988
 
test_open_and_parse_question_notafter_1_2(__attribute__((unused))
4989
 
                                          test_fixture *fixture,
4990
 
                                          __attribute__((unused))
4991
 
                                          gconstpointer user_data){
4992
 
  /* current_time, notafter, next_queue_run */
4993
 
  assert_open_and_parse_question_with_notafter(0, 1, 2);
4994
 
}
4995
 
 
4996
 
static void
4997
 
test_open_and_parse_question_equal_notafter(__attribute__((unused))
4998
 
                                            test_fixture *fixture,
4999
 
                                            __attribute__((unused))
5000
 
                                            gconstpointer user_data){
5001
 
  /* current_time, notafter, next_queue_run */
5002
 
  assert_open_and_parse_question_with_notafter(1, 1, 0);
5003
 
}
5004
 
 
5005
 
static void
5006
 
test_open_and_parse_question_late_notafter(__attribute__((unused))
5007
 
                                           test_fixture *fixture,
5008
 
                                           __attribute__((unused))
5009
 
                                           gconstpointer user_data){
5010
 
  /* current_time, notafter, next_queue_run */
5011
 
  assert_open_and_parse_question_with_notafter(2, 1, 0);
5012
 
}
5013
 
 
5014
 
static void assert_cancel_old_question_param(const mono_microsecs
5015
 
                                             next_queue_run,
5016
 
                                             const mono_microsecs
5017
 
                                             notafter,
5018
 
                                             const mono_microsecs
5019
 
                                             current_time,
5020
 
                                             const mono_microsecs
5021
 
                                             next_set_to){
5022
 
  __attribute__((cleanup(string_set_clear)))
5023
 
    string_set cancelled_filenames = {};
5024
 
  __attribute__((cleanup(cleanup_queue)))
5025
 
    task_queue *queue = create_queue();
5026
 
  g_assert_nonnull(queue);
5027
 
  queue->next_run = next_queue_run;
5028
 
 
5029
 
  char *const question_filename = strdup("/nonexistent");
5030
 
  g_assert_nonnull(question_filename);
5031
 
  task_context task = {
5032
 
    .func=cancel_old_question,
5033
 
    .question_filename=question_filename,
5034
 
    .filename=question_filename,
5035
 
    .notafter=notafter,
5036
 
    .cancelled_filenames=&cancelled_filenames,
5037
 
    .current_time=&current_time,
5038
 
  };
5039
 
  task.func(task, queue);
5040
 
 
5041
 
  if(current_time >= notafter){
5042
 
    g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5043
 
    g_assert_true(string_set_contains(cancelled_filenames,
5044
 
                                      "/nonexistent"));
5045
 
  } else {
5046
 
    g_assert_nonnull(find_matching_task(queue, (task_context){
5047
 
          .func=cancel_old_question,
5048
 
          .question_filename=question_filename,
5049
 
          .filename=question_filename,
5050
 
          .notafter=notafter,
5051
 
          .cancelled_filenames=&cancelled_filenames,
5052
 
          .current_time=&current_time,
5053
 
        }));
5054
 
 
5055
 
    g_assert_false(string_set_contains(cancelled_filenames,
5056
 
                                       question_filename));
5057
 
  }
5058
 
  g_assert_cmpuint((unsigned int)queue->next_run, ==,
5059
 
                   (unsigned int)next_set_to);
5060
 
}
5061
 
 
5062
 
static void test_cancel_old_question_0_1_2(__attribute__((unused))
5063
 
                                           test_fixture *fixture,
5064
 
                                           __attribute__((unused))
5065
 
                                           gconstpointer user_data){
5066
 
  /* next_queue_run unset,
5067
 
     cancellation should happen because time has come,
5068
 
     next_queue_run should be unchanged */
5069
 
  /* next_queue_run, notafter, current_time, next_set_to */
5070
 
  assert_cancel_old_question_param(0, 1, 2, 0);
5071
 
}
5072
 
 
5073
 
static void test_cancel_old_question_0_2_1(__attribute__((unused))
5074
 
                                           test_fixture *fixture,
5075
 
                                           __attribute__((unused))
5076
 
                                           gconstpointer user_data){
5077
 
  /* If next_queue_run is 0, meaning unset, and notafter is 2,
5078
 
     and current_time is not yet notafter or greater,
5079
 
     update value of next_queue_run to value of notafter */
5080
 
  /* next_queue_run, notafter, current_time, next_set_to */
5081
 
  assert_cancel_old_question_param(0, 2, 1, 2);
5082
 
}
5083
 
 
5084
 
static void test_cancel_old_question_1_2_3(__attribute__((unused))
5085
 
                                           test_fixture *fixture,
5086
 
                                           __attribute__((unused))
5087
 
                                           gconstpointer user_data){
5088
 
  /* next_queue_run 1,
5089
 
     cancellation should happen because time has come,
5090
 
     next_queue_run should be unchanged */
5091
 
  /* next_queue_run, notafter, current_time, next_set_to */
5092
 
  assert_cancel_old_question_param(1, 2, 3, 1);
5093
 
}
5094
 
 
5095
 
static void test_cancel_old_question_1_3_2(__attribute__((unused))
5096
 
                                           test_fixture *fixture,
5097
 
                                           __attribute__((unused))
5098
 
                                           gconstpointer user_data){
5099
 
  /* If next_queue_run is set,
5100
 
     and current_time is not yet notafter or greater,
5101
 
     and notafter is larger than next_queue_run
5102
 
     next_queue_run should be unchanged */
5103
 
  /* next_queue_run, notafter, current_time, next_set_to */
5104
 
  assert_cancel_old_question_param(1, 3, 2, 1);
5105
 
}
5106
 
 
5107
 
static void test_cancel_old_question_2_1_3(__attribute__((unused))
5108
 
                                           test_fixture *fixture,
5109
 
                                           __attribute__((unused))
5110
 
                                           gconstpointer user_data){
5111
 
  /* next_queue_run 2,
5112
 
     cancellation should happen because time has come,
5113
 
     next_queue_run should be unchanged */
5114
 
  /* next_queue_run, notafter, current_time, next_set_to */
5115
 
  assert_cancel_old_question_param(2, 1, 3, 2);
5116
 
}
5117
 
 
5118
 
static void test_cancel_old_question_2_3_1(__attribute__((unused))
5119
 
                                           test_fixture *fixture,
5120
 
                                           __attribute__((unused))
5121
 
                                           gconstpointer user_data){
5122
 
  /* If next_queue_run is set,
5123
 
     and current_time is not yet notafter or greater,
5124
 
     and notafter is larger than next_queue_run
5125
 
     next_queue_run should be unchanged */
5126
 
  /* next_queue_run, notafter, current_time, next_set_to */
5127
 
  assert_cancel_old_question_param(2, 3, 1, 2);
5128
 
}
5129
 
 
5130
 
static void test_cancel_old_question_3_1_2(__attribute__((unused))
5131
 
                                           test_fixture *fixture,
5132
 
                                           __attribute__((unused))
5133
 
                                           gconstpointer user_data){
5134
 
  /* next_queue_run 3,
5135
 
     cancellation should happen because time has come,
5136
 
     next_queue_run should be unchanged */
5137
 
  /* next_queue_run, notafter, current_time, next_set_to */
5138
 
  assert_cancel_old_question_param(3, 1, 2, 3);
5139
 
}
5140
 
 
5141
 
static void test_cancel_old_question_3_2_1(__attribute__((unused))
5142
 
                                           test_fixture *fixture,
5143
 
                                           __attribute__((unused))
5144
 
                                           gconstpointer user_data){
5145
 
  /* If next_queue_run is set,
5146
 
     and current_time is not yet notafter or greater,
5147
 
     and notafter is smaller than next_queue_run
5148
 
     update value of next_queue_run to value of notafter */
5149
 
  /* next_queue_run, notafter, current_time, next_set_to */
5150
 
  assert_cancel_old_question_param(3, 2, 1, 2);
5151
 
}
5152
 
 
5153
 
static void
5154
 
test_connect_question_socket_name_too_long(__attribute__((unused))
5155
 
                                           test_fixture *fixture,
5156
 
                                           __attribute__((unused))
5157
 
                                           gconstpointer user_data){
5158
 
  __attribute__((cleanup(cleanup_close)))
5159
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5160
 
  g_assert_cmpint(epoll_fd, >=, 0);
5161
 
  const char question_filename[] = "/nonexistent/question";
5162
 
  __attribute__((cleanup(string_set_clear)))
5163
 
    string_set cancelled_filenames = {};
5164
 
  __attribute__((cleanup(cleanup_queue)))
5165
 
    task_queue *queue = create_queue();
5166
 
  g_assert_nonnull(queue);
5167
 
  __attribute__((cleanup(cleanup_string)))
5168
 
    char *tempdir = make_temporary_directory();
5169
 
  g_assert_nonnull(tempdir);
5170
 
  struct sockaddr_un unix_socket = { .sun_family=AF_LOCAL };
5171
 
  char socket_name[sizeof(unix_socket.sun_path)];
5172
 
  memset(socket_name, 'x', sizeof(socket_name));
5173
 
  socket_name[sizeof(socket_name)-1] = '\0';
5174
 
  char *filename = NULL;
5175
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
5176
 
                  >, 0);
5177
 
  g_assert_nonnull(filename);
5178
 
 
5179
 
  task_context task = {
5180
 
    .func=connect_question_socket,
5181
 
    .question_filename=strdup(question_filename),
5182
 
    .epoll_fd=epoll_fd,
5183
 
    .password=(buffer[]){{}},
5184
 
    .filename=filename,
5185
 
    .cancelled_filenames=&cancelled_filenames,
5186
 
    .mandos_client_exited=(bool[]){false},
5187
 
    .password_is_read=(bool[]){false},
5188
 
    .current_time=(mono_microsecs[]){0},
5189
 
  };
5190
 
  g_assert_nonnull(task.question_filename);
5191
 
  run_task_with_stderr_to_dev_null(task, queue);
5192
 
 
5193
 
  g_assert_true(string_set_contains(cancelled_filenames,
5194
 
                                    question_filename));
5195
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5196
 
  g_assert_true(queue->next_run == 0);
5197
 
 
5198
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
5199
 
}
5200
 
 
5201
 
static
5202
 
void test_connect_question_socket_connect_fail(__attribute__((unused))
5203
 
                                               test_fixture *fixture,
5204
 
                                               __attribute__((unused))
5205
 
                                               gconstpointer
5206
 
                                               user_data){
5207
 
  __attribute__((cleanup(cleanup_close)))
5208
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5209
 
  g_assert_cmpint(epoll_fd, >=, 0);
5210
 
  const char question_filename[] = "/nonexistent/question";
5211
 
  __attribute__((cleanup(string_set_clear)))
5212
 
    string_set cancelled_filenames = {};
5213
 
  const mono_microsecs current_time = 3;
5214
 
  __attribute__((cleanup(cleanup_queue)))
5215
 
    task_queue *queue = create_queue();
5216
 
  g_assert_nonnull(queue);
5217
 
  __attribute__((cleanup(cleanup_string)))
5218
 
    char *tempdir = make_temporary_directory();
5219
 
  g_assert_nonnull(tempdir);
5220
 
  char socket_name[] = "nonexistent";
5221
 
  char *filename = NULL;
5222
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
5223
 
                  >, 0);
5224
 
  g_assert_nonnull(filename);
5225
 
 
5226
 
  task_context task = {
5227
 
    .func=connect_question_socket,
5228
 
    .question_filename=strdup(question_filename),
5229
 
    .epoll_fd=epoll_fd,
5230
 
    .password=(buffer[]){{}},
5231
 
    .filename=filename,
5232
 
    .cancelled_filenames=&cancelled_filenames,
5233
 
    .mandos_client_exited=(bool[]){false},
5234
 
    .password_is_read=(bool[]){false},
5235
 
    .current_time=&current_time,
5236
 
  };
5237
 
  g_assert_nonnull(task.question_filename);
5238
 
  run_task_with_stderr_to_dev_null(task, queue);
5239
 
 
5240
 
  g_assert_nonnull(find_matching_task(queue, task));
5241
 
 
5242
 
  g_assert_false(string_set_contains(cancelled_filenames,
5243
 
                                     question_filename));
5244
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5245
 
  g_assert_true(queue->next_run == 1000000 + current_time);
5246
 
 
5247
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
5248
 
}
5249
 
 
5250
 
static
5251
 
void test_connect_question_socket_bad_epoll(__attribute__((unused))
5252
 
                                            test_fixture *fixture,
5253
 
                                            __attribute__((unused))
5254
 
                                            gconstpointer user_data){
5255
 
  __attribute__((cleanup(cleanup_close)))
5256
 
    const int epoll_fd = open("/dev/null", O_WRONLY | O_CLOEXEC);
5257
 
  __attribute__((cleanup(cleanup_string)))
5258
 
    char *const question_filename = strdup("/nonexistent/question");
5259
 
  g_assert_nonnull(question_filename);
5260
 
  __attribute__((cleanup(string_set_clear)))
5261
 
    string_set cancelled_filenames = {};
5262
 
  const mono_microsecs current_time = 5;
5263
 
  __attribute__((cleanup(cleanup_queue)))
5264
 
    task_queue *queue = create_queue();
5265
 
  g_assert_nonnull(queue);
5266
 
  __attribute__((cleanup(cleanup_string)))
5267
 
    char *tempdir = make_temporary_directory();
5268
 
  g_assert_nonnull(tempdir);
5269
 
  __attribute__((cleanup(cleanup_close)))
5270
 
    const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM
5271
 
                               | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
5272
 
  g_assert_cmpint(sock_fd, >=, 0);
5273
 
  struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };
5274
 
  const char socket_name[] = "socket_name";
5275
 
  __attribute__((cleanup(cleanup_string)))
5276
 
    char *filename = NULL;
5277
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
5278
 
                  >, 0);
5279
 
  g_assert_nonnull(filename);
5280
 
  g_assert_cmpint((int)strlen(filename), <,
5281
 
                  (int)sizeof(sock_name.sun_path));
5282
 
  strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));
5283
 
  sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';
5284
 
  g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,
5285
 
                            (socklen_t)SUN_LEN(&sock_name)), >=, 0);
5286
 
  task_context task = {
5287
 
    .func=connect_question_socket,
5288
 
    .question_filename=strdup(question_filename),
5289
 
    .epoll_fd=epoll_fd,
5290
 
    .password=(buffer[]){{}},
5291
 
    .filename=strdup(filename),
5292
 
    .cancelled_filenames=&cancelled_filenames,
5293
 
    .mandos_client_exited=(bool[]){false},
5294
 
    .password_is_read=(bool[]){false},
5295
 
    .current_time=&current_time,
5296
 
  };
5297
 
  g_assert_nonnull(task.question_filename);
5298
 
  run_task_with_stderr_to_dev_null(task, queue);
5299
 
 
5300
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5301
 
  const task_context *const added_task
5302
 
    = find_matching_task(queue, task);
5303
 
  g_assert_nonnull(added_task);
5304
 
  g_assert_true(queue->next_run == 1000000 + current_time);
5305
 
 
5306
 
  g_assert_cmpint(unlink(filename), ==, 0);
5307
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
5308
 
}
5309
 
 
5310
 
static
5311
 
void test_connect_question_socket_usable(__attribute__((unused))
5312
 
                                         test_fixture *fixture,
5313
 
                                         __attribute__((unused))
5314
 
                                         gconstpointer user_data){
5315
 
  __attribute__((cleanup(cleanup_close)))
5316
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5317
 
  g_assert_cmpint(epoll_fd, >=, 0);
5318
 
  __attribute__((cleanup(cleanup_string)))
5319
 
    char *const question_filename = strdup("/nonexistent/question");
5320
 
  g_assert_nonnull(question_filename);
5321
 
  __attribute__((cleanup(string_set_clear)))
5322
 
    string_set cancelled_filenames = {};
5323
 
  buffer password = {};
5324
 
  bool mandos_client_exited = false;
5325
 
  bool password_is_read = false;
5326
 
  const mono_microsecs current_time = 0;
5327
 
  __attribute__((cleanup(cleanup_queue)))
5328
 
    task_queue *queue = create_queue();
5329
 
  g_assert_nonnull(queue);
5330
 
  __attribute__((cleanup(cleanup_string)))
5331
 
    char *tempdir = make_temporary_directory();
5332
 
  g_assert_nonnull(tempdir);
5333
 
  __attribute__((cleanup(cleanup_close)))
5334
 
    const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM
5335
 
                               | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
5336
 
  g_assert_cmpint(sock_fd, >=, 0);
5337
 
  struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };
5338
 
  const char socket_name[] = "socket_name";
5339
 
  __attribute__((cleanup(cleanup_string)))
5340
 
    char *filename = NULL;
5341
 
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
5342
 
                  >, 0);
5343
 
  g_assert_nonnull(filename);
5344
 
  g_assert_cmpint((int)strlen(filename), <,
5345
 
                  (int)sizeof(sock_name.sun_path));
5346
 
  strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));
5347
 
  sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';
5348
 
  g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,
5349
 
                            (socklen_t)SUN_LEN(&sock_name)), >=, 0);
5350
 
  task_context task = {
5351
 
    .func=connect_question_socket,
5352
 
    .question_filename=strdup(question_filename),
5353
 
    .epoll_fd=epoll_fd,
5354
 
    .password=&password,
5355
 
    .filename=strdup(filename),
5356
 
    .cancelled_filenames=&cancelled_filenames,
5357
 
    .mandos_client_exited=&mandos_client_exited,
5358
 
    .password_is_read=&password_is_read,
5359
 
    .current_time=&current_time,
5360
 
  };
5361
 
  g_assert_nonnull(task.question_filename);
5362
 
  task.func(task, queue);
5363
 
 
5364
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5365
 
  const task_context *const added_task
5366
 
    = find_matching_task(queue, (task_context){
5367
 
        .func=send_password_to_socket,
5368
 
        .question_filename=question_filename,
5369
 
        .filename=filename,
5370
 
        .epoll_fd=epoll_fd,
5371
 
        .password=&password,
5372
 
        .cancelled_filenames=&cancelled_filenames,
5373
 
        .mandos_client_exited=&mandos_client_exited,
5374
 
        .password_is_read=&password_is_read,
5375
 
        .current_time=&current_time,
5376
 
      });
5377
 
  g_assert_nonnull(added_task);
5378
 
  g_assert_cmpint(added_task->fd, >, 0);
5379
 
 
5380
 
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
5381
 
                                   EPOLLOUT));
5382
 
 
5383
 
  const int fd = added_task->fd;
5384
 
  g_assert_cmpint(fd, >, 0);
5385
 
  g_assert_true(fd_has_cloexec_and_nonblock(fd));
5386
 
 
5387
 
  /* write to fd */
5388
 
  char write_data[PIPE_BUF];
5389
 
  {
5390
 
    /* Construct test password buffer */
5391
 
    /* Start with + since that is what the real procotol uses */
5392
 
    write_data[0] = '+';
5393
 
    /* Set a special character at string end just to mark the end */
5394
 
    write_data[sizeof(write_data)-2] = 'y';
5395
 
    /* Set NUL at buffer end, as suggested by the protocol */
5396
 
    write_data[sizeof(write_data)-1] = '\0';
5397
 
    /* Fill rest of password with 'x' */
5398
 
    memset(write_data+1, 'x', sizeof(write_data)-3);
5399
 
    g_assert_cmpint((int)send(fd, write_data, sizeof(write_data),
5400
 
                              MSG_NOSIGNAL), ==, sizeof(write_data));
5401
 
  }
5402
 
 
5403
 
  /* read from sock_fd */
5404
 
  char read_data[sizeof(write_data)];
5405
 
  g_assert_cmpint((int)read(sock_fd, read_data, sizeof(read_data)),
5406
 
                  ==, sizeof(read_data));
5407
 
 
5408
 
  g_assert_true(memcmp(write_data, read_data, sizeof(write_data))
5409
 
                == 0);
5410
 
 
5411
 
  /* writing to sock_fd should fail */
5412
 
  g_assert_cmpint(send(sock_fd, write_data, sizeof(write_data),
5413
 
                       MSG_NOSIGNAL), <, 0);
5414
 
 
5415
 
  /* reading from fd should fail */
5416
 
  g_assert_cmpint((int)recv(fd, read_data, sizeof(read_data),
5417
 
                            MSG_NOSIGNAL), <, 0);
5418
 
 
5419
 
  g_assert_cmpint(unlink(filename), ==, 0);
5420
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
5421
 
}
5422
 
 
5423
 
static void
5424
 
test_send_password_to_socket_client_not_exited(__attribute__((unused))
5425
 
                                               test_fixture *fixture,
5426
 
                                               __attribute__((unused))
5427
 
                                               gconstpointer
5428
 
                                               user_data){
5429
 
  __attribute__((cleanup(cleanup_close)))
5430
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5431
 
  g_assert_cmpint(epoll_fd, >=, 0);
5432
 
  __attribute__((cleanup(cleanup_string)))
5433
 
    char *const question_filename = strdup("/nonexistent/question");
5434
 
  g_assert_nonnull(question_filename);
5435
 
  __attribute__((cleanup(cleanup_string)))
5436
 
    char *const filename = strdup("/nonexistent/socket");
5437
 
  g_assert_nonnull(filename);
5438
 
  __attribute__((cleanup(string_set_clear)))
5439
 
    string_set cancelled_filenames = {};
5440
 
  buffer password = {};
5441
 
  bool password_is_read = true;
5442
 
  __attribute__((cleanup(cleanup_queue)))
5443
 
    task_queue *queue = create_queue();
5444
 
  g_assert_nonnull(queue);
5445
 
  int socketfds[2];
5446
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5447
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5448
 
                             socketfds), ==, 0);
5449
 
  __attribute__((cleanup(cleanup_close)))
5450
 
    const int read_socket = socketfds[0];
5451
 
  const int write_socket = socketfds[1];
5452
 
  task_context task = {
5453
 
    .func=send_password_to_socket,
5454
 
    .question_filename=strdup(question_filename),
5455
 
    .filename=strdup(filename),
5456
 
    .epoll_fd=epoll_fd,
5457
 
    .fd=write_socket,
5458
 
    .password=&password,
5459
 
    .cancelled_filenames=&cancelled_filenames,
5460
 
    .mandos_client_exited=(bool[]){false},
5461
 
    .password_is_read=&password_is_read,
5462
 
    .current_time=(mono_microsecs[]){0},
5463
 
  };
5464
 
  g_assert_nonnull(task.question_filename);
5465
 
 
5466
 
  task.func(task, queue);
5467
 
 
5468
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5469
 
 
5470
 
  const task_context *const added_task
5471
 
    = find_matching_task(queue, task);
5472
 
  g_assert_nonnull(added_task);
5473
 
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
5474
 
  g_assert_true(password_is_read);
5475
 
 
5476
 
  g_assert_cmpint(added_task->fd, >, 0);
5477
 
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
5478
 
                                   EPOLLOUT));
5479
 
}
5480
 
 
5481
 
static void
5482
 
test_send_password_to_socket_password_not_read(__attribute__((unused))
5483
 
                                               test_fixture *fixture,
5484
 
                                               __attribute__((unused))
5485
 
                                               gconstpointer
5486
 
                                               user_data){
5487
 
  __attribute__((cleanup(cleanup_close)))
5488
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5489
 
  g_assert_cmpint(epoll_fd, >=, 0);
5490
 
  __attribute__((cleanup(cleanup_string)))
5491
 
    char *const question_filename = strdup("/nonexistent/question");
5492
 
  g_assert_nonnull(question_filename);
5493
 
  __attribute__((cleanup(cleanup_string)))
5494
 
    char *const filename = strdup("/nonexistent/socket");
5495
 
  __attribute__((cleanup(string_set_clear)))
5496
 
    string_set cancelled_filenames = {};
5497
 
  buffer password = {};
5498
 
  __attribute__((cleanup(cleanup_queue)))
5499
 
    task_queue *queue = create_queue();
5500
 
  g_assert_nonnull(queue);
5501
 
  int socketfds[2];
5502
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5503
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5504
 
                             socketfds), ==, 0);
5505
 
  __attribute__((cleanup(cleanup_close)))
5506
 
    const int read_socket = socketfds[0];
5507
 
  const int write_socket = socketfds[1];
5508
 
  task_context task = {
5509
 
    .func=send_password_to_socket,
5510
 
    .question_filename=strdup(question_filename),
5511
 
    .filename=strdup(filename),
5512
 
    .epoll_fd=epoll_fd,
5513
 
    .fd=write_socket,
5514
 
    .password=&password,
5515
 
    .cancelled_filenames=&cancelled_filenames,
5516
 
    .mandos_client_exited=(bool[]){false},
5517
 
    .password_is_read=(bool[]){false},
5518
 
    .current_time=(mono_microsecs[]){0},
5519
 
  };
5520
 
  g_assert_nonnull(task.question_filename);
5521
 
 
5522
 
  task.func(task, queue);
5523
 
 
5524
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5525
 
 
5526
 
  const task_context *const added_task = find_matching_task(queue,
5527
 
                                                            task);
5528
 
  g_assert_nonnull(added_task);
5529
 
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
5530
 
  g_assert_true(queue->next_run == 0);
5531
 
 
5532
 
  g_assert_cmpint(added_task->fd, >, 0);
5533
 
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
5534
 
                                   EPOLLOUT));
5535
 
}
5536
 
 
5537
 
static
5538
 
void test_send_password_to_socket_EMSGSIZE(__attribute__((unused))
5539
 
                                           test_fixture *fixture,
5540
 
                                           __attribute__((unused))
5541
 
                                           gconstpointer user_data){
5542
 
  __attribute__((cleanup(cleanup_close)))
5543
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5544
 
  g_assert_cmpint(epoll_fd, >=, 0);
5545
 
  const char question_filename[] = "/nonexistent/question";
5546
 
  char *const filename = strdup("/nonexistent/socket");
5547
 
  __attribute__((cleanup(string_set_clear)))
5548
 
    string_set cancelled_filenames = {};
5549
 
  const size_t oversized = 1024*1024; /* Limit seems to be 212960 */
5550
 
  __attribute__((cleanup(cleanup_buffer)))
5551
 
    buffer password = {
5552
 
    .data=malloc(oversized),
5553
 
    .length=oversized,
5554
 
    .allocated=oversized,
5555
 
  };
5556
 
  g_assert_nonnull(password.data);
5557
 
  if(mlock(password.data, password.allocated) != 0){
5558
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
5559
 
  }
5560
 
  /* Construct test password buffer */
5561
 
  /* Start with + since that is what the real procotol uses */
5562
 
  password.data[0] = '+';
5563
 
  /* Set a special character at string end just to mark the end */
5564
 
  password.data[oversized-3] = 'y';
5565
 
  /* Set NUL at buffer end, as suggested by the protocol */
5566
 
  password.data[oversized-2] = '\0';
5567
 
  /* Fill rest of password with 'x' */
5568
 
  memset(password.data+1, 'x', oversized-3);
5569
 
 
5570
 
  __attribute__((cleanup(cleanup_queue)))
5571
 
    task_queue *queue = create_queue();
5572
 
  g_assert_nonnull(queue);
5573
 
  int socketfds[2];
5574
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5575
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5576
 
                             socketfds), ==, 0);
5577
 
  __attribute__((cleanup(cleanup_close)))
5578
 
    const int read_socket = socketfds[0];
5579
 
  __attribute__((cleanup(cleanup_close)))
5580
 
    const int write_socket = socketfds[1];
5581
 
  task_context task = {
5582
 
    .func=send_password_to_socket,
5583
 
    .question_filename=strdup(question_filename),
5584
 
    .filename=filename,
5585
 
    .epoll_fd=epoll_fd,
5586
 
    .fd=write_socket,
5587
 
    .password=&password,
5588
 
    .cancelled_filenames=&cancelled_filenames,
5589
 
    .mandos_client_exited=(bool[]){true},
5590
 
    .password_is_read=(bool[]){true},
5591
 
    .current_time=(mono_microsecs[]){0},
5592
 
  };
5593
 
  g_assert_nonnull(task.question_filename);
5594
 
 
5595
 
  run_task_with_stderr_to_dev_null(task, queue);
5596
 
 
5597
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5598
 
  g_assert_true(string_set_contains(cancelled_filenames,
5599
 
                                    question_filename));
5600
 
}
5601
 
 
5602
 
static void test_send_password_to_socket_retry(__attribute__((unused))
5603
 
                                               test_fixture *fixture,
5604
 
                                               __attribute__((unused))
5605
 
                                               gconstpointer
5606
 
                                               user_data){
5607
 
  __attribute__((cleanup(cleanup_close)))
5608
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5609
 
  g_assert_cmpint(epoll_fd, >=, 0);
5610
 
  __attribute__((cleanup(cleanup_string)))
5611
 
    char *const question_filename = strdup("/nonexistent/question");
5612
 
  g_assert_nonnull(question_filename);
5613
 
  __attribute__((cleanup(cleanup_string)))
5614
 
    char *const filename = strdup("/nonexistent/socket");
5615
 
  g_assert_nonnull(filename);
5616
 
  __attribute__((cleanup(string_set_clear)))
5617
 
    string_set cancelled_filenames = {};
5618
 
  __attribute__((cleanup(cleanup_buffer)))
5619
 
    buffer password = {};
5620
 
 
5621
 
  __attribute__((cleanup(cleanup_queue)))
5622
 
    task_queue *queue = create_queue();
5623
 
  g_assert_nonnull(queue);
5624
 
  int socketfds[2];
5625
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5626
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5627
 
                             socketfds), ==, 0);
5628
 
  __attribute__((cleanup(cleanup_close)))
5629
 
    const int read_socket = socketfds[0];
5630
 
  const int write_socket = socketfds[1];
5631
 
  /* Close the server side socket to force ECONNRESET on client */
5632
 
  g_assert_cmpint(close(read_socket), ==, 0);
5633
 
  task_context task = {
5634
 
    .func=send_password_to_socket,
5635
 
    .question_filename=strdup(question_filename),
5636
 
    .filename=strdup(filename),
5637
 
    .epoll_fd=epoll_fd,
5638
 
    .fd=write_socket,
5639
 
    .password=&password,
5640
 
    .cancelled_filenames=&cancelled_filenames,
5641
 
    .mandos_client_exited=(bool[]){true},
5642
 
    .password_is_read=(bool[]){true},
5643
 
    .current_time=(mono_microsecs[]){0},
5644
 
  };
5645
 
  g_assert_nonnull(task.question_filename);
5646
 
 
5647
 
  task.func(task, queue);
5648
 
 
5649
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5650
 
 
5651
 
  const task_context *const added_task = find_matching_task(queue,
5652
 
                                                            task);
5653
 
  g_assert_nonnull(added_task);
5654
 
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
5655
 
 
5656
 
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
5657
 
                                   EPOLLOUT));
5658
 
}
5659
 
 
5660
 
static
5661
 
void test_send_password_to_socket_bad_epoll(__attribute__((unused))
5662
 
                                            test_fixture *fixture,
5663
 
                                            __attribute__((unused))
5664
 
                                            gconstpointer user_data){
5665
 
  __attribute__((cleanup(cleanup_close)))
5666
 
    const int epoll_fd = open("/dev/null", O_WRONLY | O_CLOEXEC);
5667
 
  __attribute__((cleanup(cleanup_string)))
5668
 
    char *const question_filename = strdup("/nonexistent/question");
5669
 
  g_assert_nonnull(question_filename);
5670
 
  __attribute__((cleanup(cleanup_string)))
5671
 
    char *const filename = strdup("/nonexistent/socket");
5672
 
  g_assert_nonnull(filename);
5673
 
  __attribute__((cleanup(string_set_clear)))
5674
 
    string_set cancelled_filenames = {};
5675
 
  __attribute__((cleanup(cleanup_buffer)))
5676
 
    buffer password = {};
5677
 
 
5678
 
  const mono_microsecs current_time = 11;
5679
 
  __attribute__((cleanup(cleanup_queue)))
5680
 
    task_queue *queue = create_queue();
5681
 
  g_assert_nonnull(queue);
5682
 
  int socketfds[2];
5683
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5684
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5685
 
                             socketfds), ==, 0);
5686
 
  __attribute__((cleanup(cleanup_close)))
5687
 
    const int read_socket = socketfds[0];
5688
 
  const int write_socket = socketfds[1];
5689
 
  /* Close the server side socket to force ECONNRESET on client */
5690
 
  g_assert_cmpint(close(read_socket), ==, 0);
5691
 
  task_context task = {
5692
 
    .func=send_password_to_socket,
5693
 
    .question_filename=strdup(question_filename),
5694
 
    .filename=strdup(filename),
5695
 
    .epoll_fd=epoll_fd,
5696
 
    .fd=write_socket,
5697
 
    .password=&password,
5698
 
    .cancelled_filenames=&cancelled_filenames,
5699
 
    .mandos_client_exited=(bool[]){true},
5700
 
    .password_is_read=(bool[]){true},
5701
 
    .current_time=&current_time,
5702
 
  };
5703
 
  g_assert_nonnull(task.question_filename);
5704
 
 
5705
 
  run_task_with_stderr_to_dev_null(task, queue);
5706
 
 
5707
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
5708
 
 
5709
 
  const task_context *const added_task = find_matching_task(queue,
5710
 
                                                            task);
5711
 
  g_assert_nonnull(added_task);
5712
 
  g_assert_true(queue->next_run == current_time + 1000000);
5713
 
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
5714
 
}
5715
 
 
5716
 
static void assert_send_password_to_socket_password(buffer password){
5717
 
  __attribute__((cleanup(cleanup_close)))
5718
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5719
 
  g_assert_cmpint(epoll_fd, >=, 0);
5720
 
  char *const question_filename = strdup("/nonexistent/question");
5721
 
  g_assert_nonnull(question_filename);
5722
 
  char *const filename = strdup("/nonexistent/socket");
5723
 
  g_assert_nonnull(filename);
5724
 
  __attribute__((cleanup(string_set_clear)))
5725
 
    string_set cancelled_filenames = {};
5726
 
 
5727
 
  __attribute__((cleanup(cleanup_queue)))
5728
 
    task_queue *queue = create_queue();
5729
 
  g_assert_nonnull(queue);
5730
 
  int socketfds[2];
5731
 
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
5732
 
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
5733
 
                             socketfds), ==, 0);
5734
 
  __attribute__((cleanup(cleanup_close)))
5735
 
    const int read_socket = socketfds[0];
5736
 
  const int write_socket = socketfds[1];
5737
 
  task_context task = {
5738
 
    .func=send_password_to_socket,
5739
 
    .question_filename=question_filename,
5740
 
    .filename=filename,
5741
 
    .epoll_fd=epoll_fd,
5742
 
    .fd=write_socket,
5743
 
    .password=&password,
5744
 
    .cancelled_filenames=&cancelled_filenames,
5745
 
    .mandos_client_exited=(bool[]){true},
5746
 
    .password_is_read=(bool[]){true},
5747
 
    .current_time=(mono_microsecs[]){0},
5748
 
  };
5749
 
 
5750
 
  char *expected_written_data = malloc(password.length + 2);
5751
 
  g_assert_nonnull(expected_written_data);
5752
 
  expected_written_data[0] = '+';
5753
 
  expected_written_data[password.length + 1] = '\0';
5754
 
  if(password.length > 0){
5755
 
    g_assert_nonnull(password.data);
5756
 
    memcpy(expected_written_data + 1, password.data, password.length);
5757
 
  }
5758
 
 
5759
 
  task.func(task, queue);
5760
 
 
5761
 
  char buf[PIPE_BUF];
5762
 
  g_assert_cmpint((int)read(read_socket, buf, PIPE_BUF), ==,
5763
 
                  (int)(password.length + 2));
5764
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5765
 
 
5766
 
  g_assert_true(memcmp(expected_written_data, buf,
5767
 
                       password.length + 2) == 0);
5768
 
 
5769
 
  g_assert_true(epoll_set_does_not_contain(epoll_fd, write_socket));
5770
 
 
5771
 
  free(expected_written_data);
5772
 
}
5773
 
 
5774
 
static void
5775
 
test_send_password_to_socket_null_password(__attribute__((unused))
5776
 
                                           test_fixture *fixture,
5777
 
                                           __attribute__((unused))
5778
 
                                           gconstpointer user_data){
5779
 
  __attribute__((cleanup(cleanup_buffer)))
5780
 
    buffer password = {};
5781
 
  assert_send_password_to_socket_password(password);
5782
 
}
5783
 
 
5784
 
static void
5785
 
test_send_password_to_socket_empty_password(__attribute__((unused))
5786
 
                                            test_fixture *fixture,
5787
 
                                            __attribute__((unused))
5788
 
                                            gconstpointer user_data){
5789
 
  __attribute__((cleanup(cleanup_buffer)))
5790
 
    buffer password = {
5791
 
    .data=malloc(1),           /* because malloc(0) may return NULL */
5792
 
    .length=0,
5793
 
    .allocated=0,               /* deliberate lie */
5794
 
  };
5795
 
  g_assert_nonnull(password.data);
5796
 
  assert_send_password_to_socket_password(password);
5797
 
}
5798
 
 
5799
 
static void
5800
 
test_send_password_to_socket_empty_str_pass(__attribute__((unused))
5801
 
                                            test_fixture *fixture,
5802
 
                                            __attribute__((unused))
5803
 
                                            gconstpointer user_data){
5804
 
  __attribute__((cleanup(cleanup_buffer)))
5805
 
    buffer password = {
5806
 
    .data=strdup(""),
5807
 
    .length=0,
5808
 
    .allocated=1,
5809
 
  };
5810
 
  if(mlock(password.data, password.allocated) != 0){
5811
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
5812
 
  }
5813
 
  assert_send_password_to_socket_password(password);
5814
 
}
5815
 
 
5816
 
static void
5817
 
test_send_password_to_socket_text_password(__attribute__((unused))
5818
 
                                           test_fixture *fixture,
5819
 
                                           __attribute__((unused))
5820
 
                                           gconstpointer user_data){
5821
 
  const char dummy_test_password[] = "dummy test password";
5822
 
  __attribute__((cleanup(cleanup_buffer)))
5823
 
    buffer password = {
5824
 
    .data = strdup(dummy_test_password),
5825
 
    .length = strlen(dummy_test_password),
5826
 
    .allocated = sizeof(dummy_test_password),
5827
 
  };
5828
 
  if(mlock(password.data, password.allocated) != 0){
5829
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
5830
 
  }
5831
 
  assert_send_password_to_socket_password(password);
5832
 
}
5833
 
 
5834
 
static void
5835
 
test_send_password_to_socket_binary_password(__attribute__((unused))
5836
 
                                             test_fixture *fixture,
5837
 
                                             __attribute__((unused))
5838
 
                                             gconstpointer user_data){
5839
 
  __attribute__((cleanup(cleanup_buffer)))
5840
 
    buffer password = {
5841
 
    .data=malloc(255),
5842
 
    .length=255,
5843
 
    .allocated=255,
5844
 
  };
5845
 
  g_assert_nonnull(password.data);
5846
 
  if(mlock(password.data, password.allocated) != 0){
5847
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
5848
 
  }
5849
 
  char c = 1;                   /* Start at 1, avoiding NUL */
5850
 
  for(int i=0; i < 255; i++){
5851
 
    password.data[i] = c++;
5852
 
  }
5853
 
  assert_send_password_to_socket_password(password);
5854
 
}
5855
 
 
5856
 
static void
5857
 
test_send_password_to_socket_nuls_in_password(__attribute__((unused))
5858
 
                                              test_fixture *fixture,
5859
 
                                              __attribute__((unused))
5860
 
                                              gconstpointer
5861
 
                                              user_data){
5862
 
  char test_password[] = {'\0', 'a', '\0', 'b', '\0', 'c', '\0'};
5863
 
  __attribute__((cleanup(cleanup_buffer)))
5864
 
    buffer password = {
5865
 
    .data=malloc(sizeof(test_password)),
5866
 
    .length=sizeof(test_password),
5867
 
    .allocated=sizeof(test_password),
5868
 
  };
5869
 
  g_assert_nonnull(password.data);
5870
 
  if(mlock(password.data, password.allocated) !=0){
5871
 
    g_assert_true(errno == EPERM or errno == ENOMEM);
5872
 
  }
5873
 
  memcpy(password.data, test_password, password.allocated);
5874
 
  assert_send_password_to_socket_password(password);
5875
 
}
5876
 
 
5877
 
static bool assert_add_existing_questions_to_devnull(task_queue
5878
 
                                                     *const,
5879
 
                                                     const int,
5880
 
                                                     buffer *const,
5881
 
                                                     string_set *,
5882
 
                                                     const
5883
 
                                                     mono_microsecs
5884
 
                                                     *const,
5885
 
                                                     bool *const,
5886
 
                                                     bool *const,
5887
 
                                                     const char
5888
 
                                                     *const);
5889
 
 
5890
 
static void test_add_existing_questions_ENOENT(__attribute__((unused))
5891
 
                                               test_fixture *fixture,
5892
 
                                               __attribute__((unused))
5893
 
                                               gconstpointer
5894
 
                                               user_data){
5895
 
  __attribute__((cleanup(cleanup_queue)))
5896
 
    task_queue *queue = create_queue();
5897
 
  g_assert_nonnull(queue);
5898
 
  __attribute__((cleanup(cleanup_close)))
5899
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5900
 
  g_assert_cmpint(epoll_fd, >=, 0);
5901
 
  __attribute__((cleanup(string_set_clear)))
5902
 
    string_set cancelled_filenames = {};
5903
 
 
5904
 
  g_assert_false(assert_add_existing_questions_to_devnull
5905
 
                 (queue,
5906
 
                  epoll_fd,
5907
 
                  (buffer[]){{}}, /* password */
5908
 
                  &cancelled_filenames,
5909
 
                  (mono_microsecs[]){0}, /* current_time */
5910
 
                  (bool[]){false},       /* mandos_client_exited */
5911
 
                  (bool[]){false},       /* password_is_read */
5912
 
                  "/nonexistent"));      /* dirname */
5913
 
 
5914
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5915
 
}
5916
 
 
5917
 
static
5918
 
bool assert_add_existing_questions_to_devnull(task_queue
5919
 
                                              *const queue,
5920
 
                                              const int
5921
 
                                              epoll_fd,
5922
 
                                              buffer *const
5923
 
                                              password,
5924
 
                                              string_set
5925
 
                                              *cancelled_filenames,
5926
 
                                              const mono_microsecs
5927
 
                                              *const current_time,
5928
 
                                              bool *const
5929
 
                                              mandos_client_exited,
5930
 
                                              bool *const
5931
 
                                              password_is_read,
5932
 
                                              const char *const
5933
 
                                              dirname){
5934
 
  __attribute__((cleanup(cleanup_close)))
5935
 
    const int devnull_fd = open("/dev/null", O_WRONLY | O_CLOEXEC);
5936
 
  g_assert_cmpint(devnull_fd, >=, 0);
5937
 
  __attribute__((cleanup(cleanup_close)))
5938
 
    const int real_stderr_fd = dup(STDERR_FILENO);
5939
 
  g_assert_cmpint(real_stderr_fd, >=, 0);
5940
 
  dup2(devnull_fd, STDERR_FILENO);
5941
 
  const bool ret = add_existing_questions(queue, epoll_fd, password,
5942
 
                                          cancelled_filenames,
5943
 
                                          current_time,
5944
 
                                          mandos_client_exited,
5945
 
                                          password_is_read, dirname);
5946
 
  dup2(real_stderr_fd, STDERR_FILENO);
5947
 
  return ret;
5948
 
}
5949
 
 
5950
 
static
5951
 
void test_add_existing_questions_no_questions(__attribute__((unused))
5952
 
                                              test_fixture *fixture,
5953
 
                                              __attribute__((unused))
5954
 
                                              gconstpointer
5955
 
                                              user_data){
5956
 
  __attribute__((cleanup(cleanup_queue)))
5957
 
    task_queue *queue = create_queue();
5958
 
  g_assert_nonnull(queue);
5959
 
  __attribute__((cleanup(cleanup_close)))
5960
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5961
 
  g_assert_cmpint(epoll_fd, >=, 0);
5962
 
  __attribute__((cleanup(string_set_clear)))
5963
 
    string_set cancelled_filenames = {};
5964
 
  __attribute__((cleanup(cleanup_string)))
5965
 
    char *tempdir = make_temporary_directory();
5966
 
  g_assert_nonnull(tempdir);
5967
 
 
5968
 
  g_assert_false(assert_add_existing_questions_to_devnull
5969
 
                 (queue,
5970
 
                  epoll_fd,
5971
 
                  (buffer[]){{}}, /* password */
5972
 
                  &cancelled_filenames,
5973
 
                  (mono_microsecs[]){0}, /* current_time */
5974
 
                  (bool[]){false},       /* mandos_client_exited */
5975
 
                  (bool[]){false},       /* password_is_read */
5976
 
                  tempdir));
5977
 
 
5978
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
5979
 
 
5980
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
5981
 
}
5982
 
 
5983
 
static char *make_question_file_in_directory(const char *const);
5984
 
 
5985
 
static
5986
 
void test_add_existing_questions_one_question(__attribute__((unused))
5987
 
                                              test_fixture *fixture,
5988
 
                                              __attribute__((unused))
5989
 
                                              gconstpointer
5990
 
                                              user_data){
5991
 
  __attribute__((cleanup(cleanup_queue)))
5992
 
    task_queue *queue = create_queue();
5993
 
  g_assert_nonnull(queue);
5994
 
  __attribute__((cleanup(cleanup_close)))
5995
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5996
 
  g_assert_cmpint(epoll_fd, >=, 0);
5997
 
  __attribute__((cleanup(cleanup_buffer)))
5998
 
    buffer password = {};
5999
 
  __attribute__((cleanup(string_set_clear)))
6000
 
    string_set cancelled_filenames = {};
6001
 
  const mono_microsecs current_time = 0;
6002
 
  bool mandos_client_exited = false;
6003
 
  bool password_is_read = false;
6004
 
  __attribute__((cleanup(cleanup_string)))
6005
 
    char *tempdir = make_temporary_directory();
6006
 
  g_assert_nonnull(tempdir);
6007
 
  __attribute__((cleanup(cleanup_string)))
6008
 
    char *question_filename
6009
 
    = make_question_file_in_directory(tempdir);
6010
 
  g_assert_nonnull(question_filename);
6011
 
 
6012
 
  g_assert_true(assert_add_existing_questions_to_devnull
6013
 
                (queue,
6014
 
                 epoll_fd,
6015
 
                 &password,
6016
 
                 &cancelled_filenames,
6017
 
                 &current_time,
6018
 
                 &mandos_client_exited,
6019
 
                 &password_is_read,
6020
 
                 tempdir));
6021
 
 
6022
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
6023
 
 
6024
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
6025
 
        .func=open_and_parse_question,
6026
 
        .epoll_fd=epoll_fd,
6027
 
        .filename=question_filename,
6028
 
        .question_filename=question_filename,
6029
 
        .password=&password,
6030
 
        .cancelled_filenames=&cancelled_filenames,
6031
 
        .current_time=&current_time,
6032
 
        .mandos_client_exited=&mandos_client_exited,
6033
 
        .password_is_read=&password_is_read,
6034
 
      }));
6035
 
 
6036
 
  g_assert_true(queue->next_run == 1);
6037
 
 
6038
 
  g_assert_cmpint(unlink(question_filename), ==, 0);
6039
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
6040
 
}
6041
 
 
6042
 
static char *make_question_file_in_directory(const char
6043
 
                                             *const dir){
6044
 
  return make_temporary_prefixed_file_in_directory("ask.", dir);
6045
 
}
6046
 
 
6047
 
static
6048
 
void test_add_existing_questions_two_questions(__attribute__((unused))
6049
 
                                               test_fixture *fixture,
6050
 
                                               __attribute__((unused))
6051
 
                                               gconstpointer
6052
 
                                               user_data){
6053
 
  __attribute__((cleanup(cleanup_queue)))
6054
 
    task_queue *queue = create_queue();
6055
 
  g_assert_nonnull(queue);
6056
 
  __attribute__((cleanup(cleanup_close)))
6057
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6058
 
  g_assert_cmpint(epoll_fd, >=, 0);
6059
 
  __attribute__((cleanup(cleanup_buffer)))
6060
 
    buffer password = {};
6061
 
  __attribute__((cleanup(string_set_clear)))
6062
 
    string_set cancelled_filenames = {};
6063
 
  const mono_microsecs current_time = 0;
6064
 
  bool mandos_client_exited = false;
6065
 
  bool password_is_read = false;
6066
 
  __attribute__((cleanup(cleanup_string)))
6067
 
    char *tempdir = make_temporary_directory();
6068
 
  g_assert_nonnull(tempdir);
6069
 
  __attribute__((cleanup(cleanup_string)))
6070
 
    char *question_filename1
6071
 
    = make_question_file_in_directory(tempdir);
6072
 
  g_assert_nonnull(question_filename1);
6073
 
  __attribute__((cleanup(cleanup_string)))
6074
 
    char *question_filename2
6075
 
    = make_question_file_in_directory(tempdir);
6076
 
  g_assert_nonnull(question_filename2);
6077
 
 
6078
 
  g_assert_true(assert_add_existing_questions_to_devnull
6079
 
                (queue,
6080
 
                 epoll_fd,
6081
 
                 &password,
6082
 
                 &cancelled_filenames,
6083
 
                 &current_time,
6084
 
                 &mandos_client_exited,
6085
 
                 &password_is_read,
6086
 
                 tempdir));
6087
 
 
6088
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 2);
6089
 
 
6090
 
  g_assert_true(queue->next_run == 1);
6091
 
 
6092
 
  __attribute__((cleanup(string_set_clear)))
6093
 
    string_set seen_questions = {};
6094
 
 
6095
 
  bool queue_contains_question_opener(char *const question_filename){
6096
 
    return(find_matching_task(queue, (task_context){
6097
 
          .func=open_and_parse_question,
6098
 
          .epoll_fd=epoll_fd,
6099
 
          .question_filename=question_filename,
6100
 
          .password=&password,
6101
 
          .cancelled_filenames=&cancelled_filenames,
6102
 
          .current_time=&current_time,
6103
 
          .mandos_client_exited=&mandos_client_exited,
6104
 
          .password_is_read=&password_is_read,
6105
 
        }) != NULL);
6106
 
  }
6107
 
 
6108
 
  g_assert_true(queue_contains_question_opener(question_filename1));
6109
 
  g_assert_true(queue_contains_question_opener(question_filename2));
6110
 
 
6111
 
  g_assert_true(queue->next_run == 1);
6112
 
 
6113
 
  g_assert_cmpint(unlink(question_filename1), ==, 0);
6114
 
  g_assert_cmpint(unlink(question_filename2), ==, 0);
6115
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
6116
 
}
6117
 
 
6118
 
static void
6119
 
test_add_existing_questions_non_questions(__attribute__((unused))
6120
 
                                          test_fixture *fixture,
6121
 
                                          __attribute__((unused))
6122
 
                                          gconstpointer user_data){
6123
 
  __attribute__((cleanup(cleanup_queue)))
6124
 
    task_queue *queue = create_queue();
6125
 
  g_assert_nonnull(queue);
6126
 
  __attribute__((cleanup(cleanup_close)))
6127
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6128
 
  g_assert_cmpint(epoll_fd, >=, 0);
6129
 
  __attribute__((cleanup(string_set_clear)))
6130
 
    string_set cancelled_filenames = {};
6131
 
  __attribute__((cleanup(cleanup_string)))
6132
 
    char *tempdir = make_temporary_directory();
6133
 
  g_assert_nonnull(tempdir);
6134
 
  __attribute__((cleanup(cleanup_string)))
6135
 
    char *question_filename1
6136
 
    = make_temporary_file_in_directory(tempdir);
6137
 
  g_assert_nonnull(question_filename1);
6138
 
  __attribute__((cleanup(cleanup_string)))
6139
 
    char *question_filename2
6140
 
    = make_temporary_file_in_directory(tempdir);
6141
 
  g_assert_nonnull(question_filename2);
6142
 
 
6143
 
  g_assert_false(assert_add_existing_questions_to_devnull
6144
 
                 (queue,
6145
 
                  epoll_fd,
6146
 
                  (buffer[]){{}}, /* password */
6147
 
                  &cancelled_filenames,
6148
 
                  (mono_microsecs[]){0}, /* current_time */
6149
 
                  (bool[]){false},       /* mandos_client_exited */
6150
 
                  (bool[]){false},       /* password_is_read */
6151
 
                  tempdir));
6152
 
 
6153
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
6154
 
 
6155
 
  g_assert_cmpint(unlink(question_filename1), ==, 0);
6156
 
  g_assert_cmpint(unlink(question_filename2), ==, 0);
6157
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
6158
 
}
6159
 
 
6160
 
static void
6161
 
test_add_existing_questions_both_types(__attribute__((unused))
6162
 
                                       test_fixture *fixture,
6163
 
                                       __attribute__((unused))
6164
 
                                       gconstpointer user_data){
6165
 
  __attribute__((cleanup(cleanup_queue)))
6166
 
    task_queue *queue = create_queue();
6167
 
  g_assert_nonnull(queue);
6168
 
  __attribute__((cleanup(cleanup_close)))
6169
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6170
 
  g_assert_cmpint(epoll_fd, >=, 0);
6171
 
  __attribute__((cleanup(cleanup_buffer)))
6172
 
    buffer password = {};
6173
 
  __attribute__((cleanup(string_set_clear)))
6174
 
    string_set cancelled_filenames = {};
6175
 
  const mono_microsecs current_time = 0;
6176
 
  bool mandos_client_exited = false;
6177
 
  bool password_is_read = false;
6178
 
  __attribute__((cleanup(cleanup_string)))
6179
 
    char *tempdir = make_temporary_directory();
6180
 
  g_assert_nonnull(tempdir);
6181
 
  __attribute__((cleanup(cleanup_string)))
6182
 
    char *tempfilename1 = make_temporary_file_in_directory(tempdir);
6183
 
  g_assert_nonnull(tempfilename1);
6184
 
  __attribute__((cleanup(cleanup_string)))
6185
 
    char *tempfilename2 = make_temporary_file_in_directory(tempdir);
6186
 
  g_assert_nonnull(tempfilename2);
6187
 
  __attribute__((cleanup(cleanup_string)))
6188
 
    char *question_filename
6189
 
    = make_question_file_in_directory(tempdir);
6190
 
  g_assert_nonnull(question_filename);
6191
 
 
6192
 
  g_assert_true(assert_add_existing_questions_to_devnull
6193
 
                (queue,
6194
 
                 epoll_fd,
6195
 
                 &password,
6196
 
                 &cancelled_filenames,
6197
 
                 &current_time,
6198
 
                 &mandos_client_exited,
6199
 
                 &password_is_read,
6200
 
                 tempdir));
6201
 
 
6202
 
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
6203
 
 
6204
 
  g_assert_nonnull(find_matching_task(queue, (task_context){
6205
 
        .func=open_and_parse_question,
6206
 
        .epoll_fd=epoll_fd,
6207
 
        .filename=question_filename,
6208
 
        .question_filename=question_filename,
6209
 
        .password=&password,
6210
 
        .cancelled_filenames=&cancelled_filenames,
6211
 
        .current_time=&current_time,
6212
 
        .mandos_client_exited=&mandos_client_exited,
6213
 
        .password_is_read=&password_is_read,
6214
 
      }));
6215
 
 
6216
 
  g_assert_true(queue->next_run == 1);
6217
 
 
6218
 
  g_assert_cmpint(unlink(tempfilename1), ==, 0);
6219
 
  g_assert_cmpint(unlink(tempfilename2), ==, 0);
6220
 
  g_assert_cmpint(unlink(question_filename), ==, 0);
6221
 
  g_assert_cmpint(rmdir(tempdir), ==, 0);
6222
 
}
6223
 
 
6224
 
static void test_wait_for_event_timeout(__attribute__((unused))
6225
 
                                        test_fixture *fixture,
6226
 
                                        __attribute__((unused))
6227
 
                                        gconstpointer user_data){
6228
 
  __attribute__((cleanup(cleanup_close)))
6229
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6230
 
  g_assert_cmpint(epoll_fd, >=, 0);
6231
 
 
6232
 
  g_assert_true(wait_for_event(epoll_fd, 1, 0));
6233
 
}
6234
 
 
6235
 
static void test_wait_for_event_event(__attribute__((unused))
6236
 
                                      test_fixture *fixture,
6237
 
                                      __attribute__((unused))
6238
 
                                      gconstpointer user_data){
6239
 
  __attribute__((cleanup(cleanup_close)))
6240
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6241
 
  g_assert_cmpint(epoll_fd, >=, 0);
6242
 
  int pipefds[2];
6243
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
6244
 
  __attribute__((cleanup(cleanup_close)))
6245
 
    const int read_pipe = pipefds[0];
6246
 
  __attribute__((cleanup(cleanup_close)))
6247
 
    const int write_pipe = pipefds[1];
6248
 
  g_assert_cmpint(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, read_pipe,
6249
 
                            &(struct epoll_event)
6250
 
                            { .events=EPOLLIN | EPOLLRDHUP }), ==, 0);
6251
 
  g_assert_cmpint((int)write(write_pipe, "x", 1), ==, 1);
6252
 
 
6253
 
  g_assert_true(wait_for_event(epoll_fd, 0, 0));
6254
 
}
6255
 
 
6256
 
static void test_wait_for_event_sigchld(test_fixture *fixture,
6257
 
                                        __attribute__((unused))
6258
 
                                        gconstpointer user_data){
6259
 
  const pid_t pid = fork();
6260
 
  if(pid == 0){         /* Child */
6261
 
    if(not restore_signal_handler(&fixture->orig_sigaction)){
6262
 
      _exit(EXIT_FAILURE);
6263
 
    }
6264
 
    if(not restore_sigmask(&fixture->orig_sigmask)){
6265
 
      _exit(EXIT_FAILURE);
6266
 
    }
6267
 
    exit(EXIT_SUCCESS);
6268
 
  }
6269
 
  g_assert_true(pid != -1);
6270
 
  __attribute__((cleanup(cleanup_close)))
6271
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6272
 
  g_assert_cmpint(epoll_fd, >=, 0);
6273
 
 
6274
 
  g_assert_true(wait_for_event(epoll_fd, 0, 0));
6275
 
 
6276
 
  int status;
6277
 
  g_assert_true(waitpid(pid, &status, 0) == pid);
6278
 
  g_assert_true(WIFEXITED(status));
6279
 
  g_assert_cmpint(WEXITSTATUS(status), ==, EXIT_SUCCESS);
6280
 
}
6281
 
 
6282
 
static void test_run_queue_zeroes_next_run(__attribute__((unused))
6283
 
                                           test_fixture *fixture,
6284
 
                                           __attribute__((unused))
6285
 
                                           gconstpointer user_data){
6286
 
  __attribute__((cleanup(cleanup_queue)))
6287
 
    task_queue *queue = create_queue();
6288
 
  g_assert_nonnull(queue);
6289
 
  queue->next_run = 1;
6290
 
  __attribute__((cleanup(cleanup_close)))
6291
 
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
6292
 
  __attribute__((cleanup(string_set_clear)))
6293
 
    string_set cancelled_filenames = {};
6294
 
  bool quit_now = false;
6295
 
 
6296
 
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
6297
 
  g_assert_false(quit_now);
6298
 
  g_assert_cmpuint((unsigned int)queue->next_run, ==, 0);
6299
 
}
6300
 
 
6301
 
static
6302
 
void test_run_queue_clears_cancelled_filenames(__attribute__((unused))
6303
 
                                               test_fixture *fixture,
6304
 
                                               __attribute__((unused))
6305
 
                                               gconstpointer
6306
 
                                               user_data){
6307
 
  __attribute__((cleanup(cleanup_queue)))
6308
 
    task_queue *queue = create_queue();
6309
 
  g_assert_nonnull(queue);
6310
 
  __attribute__((cleanup(string_set_clear)))
6311
 
    string_set cancelled_filenames = {};
6312
 
  bool quit_now = false;
6313
 
  const char question_filename[] = "/nonexistent/question_filename";
6314
 
  g_assert_true(string_set_add(&cancelled_filenames,
6315
 
                               question_filename));
6316
 
 
6317
 
  g_assert_true(add_to_queue(queue,
6318
 
                             (task_context){ .func=dummy_func }));
6319
 
 
6320
 
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
6321
 
  g_assert_false(quit_now);
6322
 
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
6323
 
  g_assert_false(string_set_contains(cancelled_filenames,
6324
 
                                     question_filename));
6325
 
}
6326
 
 
6327
 
static
6328
 
void test_run_queue_skips_cancelled_filenames(__attribute__((unused))
6329
 
                                              test_fixture *fixture,
6330
 
                                              __attribute__((unused))
6331
 
                                              gconstpointer
6332
 
                                              user_data){
6333
 
  __attribute__((cleanup(cleanup_queue)))
6334
 
    task_queue *queue = create_queue();
6335
 
  g_assert_nonnull(queue);
6336
 
  __attribute__((cleanup(string_set_clear)))
6337
 
    string_set cancelled_filenames = {};
6338
 
  bool quit_now = false;
6339
 
  int pipefds[2];
6340
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
6341
 
  __attribute__((cleanup(cleanup_close)))
6342
 
    const int read_pipe = pipefds[0];
6343
 
  g_assert_cmpint(close(pipefds[1]), ==, 0);
6344
 
  const char question_filename[] = "/nonexistent/question_filename";
6345
 
  g_assert_true(string_set_add(&cancelled_filenames,
6346
 
                               question_filename));
6347
 
  __attribute__((nonnull))
6348
 
    void quit_func(const task_context task,
6349
 
                   __attribute__((unused)) task_queue *const q){
6350
 
    g_assert_nonnull(task.quit_now);
6351
 
    *task.quit_now = true;
6352
 
  }
6353
 
  task_context task = {
6354
 
    .func=quit_func,
6355
 
    .question_filename=strdup(question_filename),
6356
 
    .quit_now=&quit_now,
6357
 
    .fd=read_pipe,
6358
 
  };
6359
 
  g_assert_nonnull(task.question_filename);
6360
 
 
6361
 
  g_assert_true(add_to_queue(queue, task));
6362
 
 
6363
 
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
6364
 
  g_assert_false(quit_now);
6365
 
 
6366
 
  /* read_pipe should be closed already */
6367
 
  errno = 0;
6368
 
  bool read_pipe_closed = (close(read_pipe) == -1);
6369
 
  read_pipe_closed &= (errno == EBADF);
6370
 
  g_assert_true(read_pipe_closed);
6371
 
}
6372
 
 
6373
 
static void test_run_queue_one_task(__attribute__((unused))
6374
 
                                    test_fixture *fixture,
6375
 
                                    __attribute__((unused))
6376
 
                                    gconstpointer user_data){
6377
 
  __attribute__((cleanup(cleanup_queue)))
6378
 
    task_queue *queue = create_queue();
6379
 
  g_assert_nonnull(queue);
6380
 
  __attribute__((cleanup(string_set_clear)))
6381
 
    string_set cancelled_filenames = {};
6382
 
  bool quit_now = false;
6383
 
 
6384
 
  __attribute__((nonnull))
6385
 
    void next_run_func(__attribute__((unused))
6386
 
                       const task_context task,
6387
 
                       task_queue *const q){
6388
 
    q->next_run = 1;
6389
 
  }
6390
 
 
6391
 
  task_context task = {
6392
 
    .func=next_run_func,
6393
 
  };
6394
 
  g_assert_true(add_to_queue(queue, task));
6395
 
 
6396
 
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
6397
 
  g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);
6398
 
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
6399
 
}
6400
 
 
6401
 
static void test_run_queue_two_tasks(__attribute__((unused))
6402
 
                                     test_fixture *fixture,
6403
 
                                     __attribute__((unused))
6404
 
                                     gconstpointer user_data){
6405
 
  __attribute__((cleanup(cleanup_queue)))
6406
 
    task_queue *queue = create_queue();
6407
 
  g_assert_nonnull(queue);
6408
 
  queue->next_run = 1;
6409
 
  __attribute__((cleanup(string_set_clear)))
6410
 
    string_set cancelled_filenames = {};
6411
 
  bool quit_now = false;
6412
 
  bool mandos_client_exited = false;
6413
 
 
6414
 
  __attribute__((nonnull))
6415
 
    void next_run_func(__attribute__((unused))
6416
 
                       const task_context task,
6417
 
                       task_queue *const q){
6418
 
    q->next_run = 1;
6419
 
  }
6420
 
 
6421
 
  __attribute__((nonnull))
6422
 
    void exited_func(const task_context task,
6423
 
                     __attribute__((unused)) task_queue *const q){
6424
 
    *task.mandos_client_exited = true;
6425
 
  }
6426
 
 
6427
 
  task_context task1 = {
6428
 
    .func=next_run_func,
6429
 
  };
6430
 
  g_assert_true(add_to_queue(queue, task1));
6431
 
 
6432
 
  task_context task2 = {
6433
 
    .func=exited_func,
6434
 
    .mandos_client_exited=&mandos_client_exited,
6435
 
  };
6436
 
  g_assert_true(add_to_queue(queue, task2));
6437
 
 
6438
 
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
6439
 
  g_assert_false(quit_now);
6440
 
  g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);
6441
 
  g_assert_true(mandos_client_exited);
6442
 
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
6443
 
}
6444
 
 
6445
 
static void test_run_queue_two_tasks_quit(__attribute__((unused))
6446
 
                                          test_fixture *fixture,
6447
 
                                          __attribute__((unused))
6448
 
                                          gconstpointer user_data){
6449
 
  __attribute__((cleanup(cleanup_queue)))
6450
 
    task_queue *queue = create_queue();
6451
 
  g_assert_nonnull(queue);
6452
 
  __attribute__((cleanup(string_set_clear)))
6453
 
    string_set cancelled_filenames = {};
6454
 
  bool quit_now = false;
6455
 
  bool mandos_client_exited = false;
6456
 
  bool password_is_read = false;
6457
 
 
6458
 
  __attribute__((nonnull))
6459
 
    void set_exited_func(const task_context task,
6460
 
                         __attribute__((unused)) task_queue *const q){
6461
 
    *task.mandos_client_exited = true;
6462
 
    *task.quit_now = true;
6463
 
  }
6464
 
  task_context task1 = {
6465
 
    .func=set_exited_func,
6466
 
    .quit_now=&quit_now,
6467
 
    .mandos_client_exited=&mandos_client_exited,
6468
 
  };
6469
 
  g_assert_true(add_to_queue(queue, task1));
6470
 
 
6471
 
  __attribute__((nonnull))
6472
 
    void set_read_func(const task_context task,
6473
 
                       __attribute__((unused)) task_queue *const q){
6474
 
    *task.quit_now = true;
6475
 
    *task.password_is_read = true;
6476
 
  }
6477
 
  task_context task2 = {
6478
 
    .func=set_read_func,
6479
 
    .quit_now=&quit_now,
6480
 
    .password_is_read=&password_is_read,
6481
 
  };
6482
 
  g_assert_true(add_to_queue(queue, task2));
6483
 
 
6484
 
  g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));
6485
 
  g_assert_true(quit_now);
6486
 
  g_assert_true(mandos_client_exited xor password_is_read);
6487
 
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
6488
 
}
6489
 
 
6490
 
static void test_run_queue_two_tasks_cleanup(__attribute__((unused))
6491
 
                                             test_fixture *fixture,
6492
 
                                             __attribute__((unused))
6493
 
                                             gconstpointer user_data){
6494
 
  __attribute__((cleanup(cleanup_queue)))
6495
 
    task_queue *queue = create_queue();
6496
 
  g_assert_nonnull(queue);
6497
 
  __attribute__((cleanup(string_set_clear)))
6498
 
    string_set cancelled_filenames = {};
6499
 
  int pipefds[2];
6500
 
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
6501
 
  __attribute__((cleanup(cleanup_close)))
6502
 
    const int read_pipe = pipefds[0];
6503
 
  __attribute__((cleanup(cleanup_close)))
6504
 
    const int write_pipe = pipefds[1];
6505
 
  bool quit_now = false;
6506
 
 
6507
 
  __attribute__((nonnull))
6508
 
    void read_func(const task_context task,
6509
 
                   __attribute__((unused)) task_queue *const q){
6510
 
    *task.quit_now = true;
6511
 
  }
6512
 
  task_context task1 = {
6513
 
    .func=read_func,
6514
 
    .quit_now=&quit_now,
6515
 
    .fd=read_pipe,
6516
 
  };
6517
 
  g_assert_true(add_to_queue(queue, task1));
6518
 
 
6519
 
  __attribute__((nonnull))
6520
 
    void write_func(const task_context task,
6521
 
                    __attribute__((unused)) task_queue *const q){
6522
 
    *task.quit_now = true;
6523
 
  }
6524
 
  task_context task2 = {
6525
 
    .func=write_func,
6526
 
    .quit_now=&quit_now,
6527
 
    .fd=write_pipe,
6528
 
  };
6529
 
  g_assert_true(add_to_queue(queue, task2));
6530
 
 
6531
 
  g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));
6532
 
  g_assert_true(quit_now);
6533
 
 
6534
 
  /* Either read_pipe or write_pipe should be closed already */
6535
 
  errno = 0;
6536
 
  bool close_read_pipe = (close(read_pipe) == -1);
6537
 
  close_read_pipe &= (errno == EBADF);
6538
 
  errno = 0;
6539
 
  bool close_write_pipe = (close(write_pipe) == -1);
6540
 
  close_write_pipe &= (errno == EBADF);
6541
 
  g_assert_true(close_read_pipe xor close_write_pipe);
6542
 
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
6543
 
}
6544
 
 
6545
 
static void test_setup_signal_handler(__attribute__((unused))
6546
 
                                      test_fixture *fixture,
6547
 
                                      __attribute__((unused))
6548
 
                                      gconstpointer user_data){
6549
 
  /* Save current SIGCHLD action, whatever it is */
6550
 
  struct sigaction expected_sigchld_action;
6551
 
  g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),
6552
 
                  ==, 0);
6553
 
 
6554
 
  /* Act; i.e. run the setup_signal_handler() function */
6555
 
  struct sigaction actual_old_sigchld_action;
6556
 
  g_assert_true(setup_signal_handler(&actual_old_sigchld_action));
6557
 
 
6558
 
  /* Check that the function correctly set "actual_old_sigchld_action"
6559
 
     to the same values as the previously saved
6560
 
     "expected_sigchld_action" */
6561
 
  /* Check member sa_handler */
6562
 
  g_assert_true(actual_old_sigchld_action.sa_handler
6563
 
                == expected_sigchld_action.sa_handler);
6564
 
  /* Check member sa_mask */
6565
 
  for(int signum = 1; signum < NSIG; signum++){
6566
 
    const int expected_old_block_state
6567
 
      = sigismember(&expected_sigchld_action.sa_mask, signum);
6568
 
    g_assert_cmpint(expected_old_block_state, >=, 0);
6569
 
    const int actual_old_block_state
6570
 
      = sigismember(&actual_old_sigchld_action.sa_mask, signum);
6571
 
    g_assert_cmpint(actual_old_block_state, >=, 0);
6572
 
    g_assert_cmpint(actual_old_block_state,
6573
 
                    ==, expected_old_block_state);
6574
 
  }
6575
 
  /* Check member sa_flags */
6576
 
  g_assert_true((actual_old_sigchld_action.sa_flags
6577
 
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
6578
 
                == (expected_sigchld_action.sa_flags
6579
 
                    & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));
6580
 
 
6581
 
  /* Retrieve the current signal handler for SIGCHLD as set by
6582
 
     setup_signal_handler() */
6583
 
  struct sigaction actual_new_sigchld_action;
6584
 
  g_assert_cmpint(sigaction(SIGCHLD, NULL,
6585
 
                            &actual_new_sigchld_action), ==, 0);
6586
 
  /* Check that the signal handler (member sa_handler) is correctly
6587
 
     set to the "handle_sigchld" function */
6588
 
  g_assert_true(actual_new_sigchld_action.sa_handler != SIG_DFL);
6589
 
  g_assert_true(actual_new_sigchld_action.sa_handler != SIG_IGN);
6590
 
  g_assert_true(actual_new_sigchld_action.sa_handler
6591
 
                == handle_sigchld);
6592
 
  /* Check (in member sa_mask) that at least a handful of signals are
6593
 
     actually blocked during the signal handler */
6594
 
  for(int signum = 1; signum < NSIG; signum++){
6595
 
    int actual_new_block_state;
6596
 
    switch(signum){
6597
 
    case SIGTERM:
6598
 
    case SIGINT:
6599
 
    case SIGQUIT:
6600
 
    case SIGHUP:
6601
 
      actual_new_block_state
6602
 
        = sigismember(&actual_new_sigchld_action.sa_mask, signum);
6603
 
      g_assert_cmpint(actual_new_block_state, ==, 1);
6604
 
      continue;
6605
 
    case SIGKILL:               /* non-blockable */
6606
 
    case SIGSTOP:               /* non-blockable */
6607
 
    case SIGCHLD:               /* always blocked */
6608
 
    default:
6609
 
      continue;
6610
 
    }
6611
 
  }
6612
 
  /* Check member sa_flags */
6613
 
  g_assert_true((actual_new_sigchld_action.sa_flags
6614
 
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
6615
 
                == (SA_NOCLDSTOP | SA_RESTART));
6616
 
 
6617
 
  /* Restore signal handler */
6618
 
  g_assert_cmpint(sigaction(SIGCHLD, &expected_sigchld_action, NULL),
6619
 
                  ==, 0);
6620
 
}
6621
 
 
6622
 
static void test_restore_signal_handler(__attribute__((unused))
6623
 
                                        test_fixture *fixture,
6624
 
                                        __attribute__((unused))
6625
 
                                        gconstpointer user_data){
6626
 
  /* Save current SIGCHLD action, whatever it is */
6627
 
  struct sigaction expected_sigchld_action;
6628
 
  g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),
6629
 
                  ==, 0);
6630
 
  /* Since we haven't established a signal handler yet, there should
6631
 
     not be one established.  But another test may have relied on
6632
 
     restore_signal_handler() to restore the signal handler, and if
6633
 
     restore_signal_handler() is buggy (which we should be prepared
6634
 
     for in this test) the signal handler may not have been restored
6635
 
     properly; check for this: */
6636
 
  g_assert_true(expected_sigchld_action.sa_handler != handle_sigchld);
6637
 
 
6638
 
  /* Establish a signal handler */
6639
 
  struct sigaction sigchld_action = {
6640
 
    .sa_handler=handle_sigchld,
6641
 
    .sa_flags=SA_RESTART | SA_NOCLDSTOP,
6642
 
  };
6643
 
  g_assert_cmpint(sigfillset(&sigchld_action.sa_mask), ==, 0);
6644
 
  g_assert_cmpint(sigaction(SIGCHLD, &sigchld_action, NULL), ==, 0);
6645
 
 
6646
 
  /* Act; i.e. run the restore_signal_handler() function */
6647
 
  g_assert_true(restore_signal_handler(&expected_sigchld_action));
6648
 
 
6649
 
  /* Retrieve the restored signal handler data */
6650
 
  struct sigaction actual_restored_sigchld_action;
6651
 
  g_assert_cmpint(sigaction(SIGCHLD, NULL,
6652
 
                            &actual_restored_sigchld_action), ==, 0);
6653
 
 
6654
 
  /* Check that the function correctly restored the signal action, as
6655
 
     saved in "actual_restored_sigchld_action", to the same values as
6656
 
     the previously saved "expected_sigchld_action" */
6657
 
  /* Check member sa_handler */
6658
 
  g_assert_true(actual_restored_sigchld_action.sa_handler
6659
 
                == expected_sigchld_action.sa_handler);
6660
 
  /* Check member sa_mask */
6661
 
  for(int signum = 1; signum < NSIG; signum++){
6662
 
    const int expected_old_block_state
6663
 
      = sigismember(&expected_sigchld_action.sa_mask, signum);
6664
 
    g_assert_cmpint(expected_old_block_state, >=, 0);
6665
 
    const int actual_restored_block_state
6666
 
      = sigismember(&actual_restored_sigchld_action.sa_mask, signum);
6667
 
    g_assert_cmpint(actual_restored_block_state, >=, 0);
6668
 
    g_assert_cmpint(actual_restored_block_state,
6669
 
                    ==, expected_old_block_state);
6670
 
  }
6671
 
  /* Check member sa_flags */
6672
 
  g_assert_true((actual_restored_sigchld_action.sa_flags
6673
 
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
6674
 
                == (expected_sigchld_action.sa_flags
6675
 
                    & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));
6676
 
}
6677
 
 
6678
 
static void test_block_sigchld(__attribute__((unused))
6679
 
                               test_fixture *fixture,
6680
 
                               __attribute__((unused))
6681
 
                               gconstpointer user_data){
6682
 
  /* Save original signal mask */
6683
 
  sigset_t expected_sigmask;
6684
 
  g_assert_cmpint(pthread_sigmask(-1, NULL, &expected_sigmask),
6685
 
                  ==, 0);
6686
 
 
6687
 
  /* Make sure SIGCHLD is unblocked for this test */
6688
 
  sigset_t sigchld_sigmask;
6689
 
  g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);
6690
 
  g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);
6691
 
  g_assert_cmpint(pthread_sigmask(SIG_UNBLOCK, &sigchld_sigmask,
6692
 
                                  NULL), ==, 0);
6693
 
 
6694
 
  /* Act; i.e. run the block_sigchld() function */
6695
 
  sigset_t actual_old_sigmask;
6696
 
  g_assert_true(block_sigchld(&actual_old_sigmask));
6697
 
 
6698
 
  /* Check the actual_old_sigmask; it should be the same as the
6699
 
     previously saved signal mask "expected_sigmask". */
6700
 
  for(int signum = 1; signum < NSIG; signum++){
6701
 
    const int expected_old_block_state
6702
 
      = sigismember(&expected_sigmask, signum);
6703
 
    g_assert_cmpint(expected_old_block_state, >=, 0);
6704
 
    const int actual_old_block_state
6705
 
      = sigismember(&actual_old_sigmask, signum);
6706
 
    g_assert_cmpint(actual_old_block_state, >=, 0);
6707
 
    g_assert_cmpint(actual_old_block_state,
6708
 
                    ==, expected_old_block_state);
6709
 
  }
6710
 
 
6711
 
  /* Retrieve the newly set signal mask */
6712
 
  sigset_t actual_sigmask;
6713
 
  g_assert_cmpint(pthread_sigmask(-1, NULL, &actual_sigmask), ==, 0);
6714
 
 
6715
 
  /* SIGCHLD should be blocked */
6716
 
  g_assert_cmpint(sigismember(&actual_sigmask, SIGCHLD), ==, 1);
6717
 
 
6718
 
  /* Restore signal mask */
6719
 
  g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &expected_sigmask,
6720
 
                                  NULL), ==, 0);
6721
 
}
6722
 
 
6723
 
static void test_restore_sigmask(__attribute__((unused))
6724
 
                                 test_fixture *fixture,
6725
 
                                 __attribute__((unused))
6726
 
                                 gconstpointer user_data){
6727
 
  /* Save original signal mask */
6728
 
  sigset_t orig_sigmask;
6729
 
  g_assert_cmpint(pthread_sigmask(-1, NULL, &orig_sigmask), ==, 0);
6730
 
 
6731
 
  /* Make sure SIGCHLD is blocked for this test */
6732
 
  sigset_t sigchld_sigmask;
6733
 
  g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);
6734
 
  g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);
6735
 
  g_assert_cmpint(pthread_sigmask(SIG_BLOCK, &sigchld_sigmask,
6736
 
                                  NULL), ==, 0);
6737
 
 
6738
 
  /* Act; i.e. run the restore_sigmask() function */
6739
 
  g_assert_true(restore_sigmask(&orig_sigmask));
6740
 
 
6741
 
  /* Retrieve the newly restored signal mask */
6742
 
  sigset_t restored_sigmask;
6743
 
  g_assert_cmpint(pthread_sigmask(-1, NULL, &restored_sigmask),
6744
 
                  ==, 0);
6745
 
 
6746
 
  /* Check the restored_sigmask; it should be the same as the
6747
 
     previously saved signal mask "orig_sigmask". */
6748
 
  for(int signum = 1; signum < NSIG; signum++){
6749
 
    const int orig_block_state = sigismember(&orig_sigmask, signum);
6750
 
    g_assert_cmpint(orig_block_state, >=, 0);
6751
 
    const int restored_block_state = sigismember(&restored_sigmask,
6752
 
                                                 signum);
6753
 
    g_assert_cmpint(restored_block_state, >=, 0);
6754
 
    g_assert_cmpint(restored_block_state, ==, orig_block_state);
6755
 
  }
6756
 
 
6757
 
  /* Restore signal mask */
6758
 
  g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &orig_sigmask,
6759
 
                                  NULL), ==, 0);
6760
 
}
6761
 
 
6762
 
static void test_parse_arguments_noargs(__attribute__((unused))
6763
 
                                        test_fixture *fixture,
6764
 
                                        __attribute__((unused))
6765
 
                                        gconstpointer user_data){
6766
 
  char *argv[] = {
6767
 
    strdup("prgname"),
6768
 
    NULL };
6769
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6770
 
 
6771
 
  char *agent_directory = NULL;
6772
 
  char *helper_directory = NULL;
6773
 
  uid_t user = 0;
6774
 
  gid_t group = 0;
6775
 
  char *mandos_argz = NULL;
6776
 
  size_t mandos_argz_length = 0;
6777
 
 
6778
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
6779
 
                                &helper_directory, &user, &group,
6780
 
                                &mandos_argz, &mandos_argz_length));
6781
 
  g_assert_null(agent_directory);
6782
 
  g_assert_null(helper_directory);
6783
 
  g_assert_true(user == 0);
6784
 
  g_assert_true(group == 0);
6785
 
  g_assert_null(mandos_argz);
6786
 
  g_assert_true(mandos_argz_length == 0);
6787
 
 
6788
 
  for(char **arg = argv; *arg != NULL; arg++){
6789
 
    free(*arg);
6790
 
  }
6791
 
}
6792
 
 
6793
 
__attribute__((nonnull))
6794
 
static bool parse_arguments_devnull(int argc, char *argv[],
6795
 
                                    const bool exit_failure,
6796
 
                                    char **agent_directory,
6797
 
                                    char **helper_directory,
6798
 
                                    uid_t *const user,
6799
 
                                    gid_t *const group,
6800
 
                                    char **mandos_argz,
6801
 
                                    size_t *mandos_argz_length){
6802
 
 
6803
 
  FILE *real_stderr = stderr;
6804
 
  FILE *devnull = fopen("/dev/null", "we");
6805
 
  g_assert_nonnull(devnull);
6806
 
  stderr = devnull;
6807
 
 
6808
 
  const bool ret = parse_arguments(argc, argv, exit_failure,
6809
 
                                   agent_directory,
6810
 
                                   helper_directory, user, group,
6811
 
                                   mandos_argz, mandos_argz_length);
6812
 
  const error_t saved_errno = errno;
6813
 
 
6814
 
  stderr = real_stderr;
6815
 
  g_assert_cmpint(fclose(devnull), ==, 0);
6816
 
 
6817
 
  errno = saved_errno;
6818
 
 
6819
 
  return ret;
6820
 
}
6821
 
 
6822
 
static void test_parse_arguments_invalid(__attribute__((unused))
6823
 
                                         test_fixture *fixture,
6824
 
                                         __attribute__((unused))
6825
 
                                         gconstpointer user_data){
6826
 
  char *argv[] = {
6827
 
    strdup("prgname"),
6828
 
    strdup("--invalid"),
6829
 
    NULL };
6830
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6831
 
 
6832
 
  char *agent_directory = NULL;
6833
 
  char *helper_directory = NULL;
6834
 
  uid_t user = 0;
6835
 
  gid_t group = 0;
6836
 
  char *mandos_argz = NULL;
6837
 
  size_t mandos_argz_length = 0;
6838
 
 
6839
 
  g_assert_false(parse_arguments_devnull(argc, argv, false,
6840
 
                                         &agent_directory,
6841
 
                                         &helper_directory, &user,
6842
 
                                         &group, &mandos_argz,
6843
 
                                         &mandos_argz_length));
6844
 
 
6845
 
  g_assert_true(errno == EINVAL);
6846
 
  g_assert_null(agent_directory);
6847
 
  g_assert_null(helper_directory);
6848
 
  g_assert_null(mandos_argz);
6849
 
  g_assert_true(mandos_argz_length == 0);
6850
 
 
6851
 
  for(char **arg = argv; *arg != NULL; arg++){
6852
 
    free(*arg);
6853
 
  }
6854
 
}
6855
 
 
6856
 
static void test_parse_arguments_long_dir(__attribute__((unused))
6857
 
                                          test_fixture *fixture,
6858
 
                                          __attribute__((unused))
6859
 
                                          gconstpointer user_data){
6860
 
  char *argv[] = {
6861
 
    strdup("prgname"),
6862
 
    strdup("--agent-directory"),
6863
 
    strdup("/tmp"),
6864
 
    NULL };
6865
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6866
 
 
6867
 
  __attribute__((cleanup(cleanup_string)))
6868
 
    char *agent_directory = NULL;
6869
 
  char *helper_directory = NULL;
6870
 
  uid_t user = 0;
6871
 
  gid_t group = 0;
6872
 
  __attribute__((cleanup(cleanup_string)))
6873
 
    char *mandos_argz = NULL;
6874
 
  size_t mandos_argz_length = 0;
6875
 
 
6876
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
6877
 
                                &helper_directory, &user, &group,
6878
 
                                &mandos_argz, &mandos_argz_length));
6879
 
 
6880
 
  g_assert_cmpstr(agent_directory, ==, "/tmp");
6881
 
  g_assert_null(helper_directory);
6882
 
  g_assert_true(user == 0);
6883
 
  g_assert_true(group == 0);
6884
 
  g_assert_null(mandos_argz);
6885
 
  g_assert_true(mandos_argz_length == 0);
6886
 
 
6887
 
  for(char **arg = argv; *arg != NULL; arg++){
6888
 
    free(*arg);
6889
 
  }
6890
 
}
6891
 
 
6892
 
static void test_parse_arguments_short_dir(__attribute__((unused))
6893
 
                                           test_fixture *fixture,
6894
 
                                           __attribute__((unused))
6895
 
                                           gconstpointer user_data){
6896
 
  char *argv[] = {
6897
 
    strdup("prgname"),
6898
 
    strdup("-d"),
6899
 
    strdup("/tmp"),
6900
 
    NULL };
6901
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6902
 
 
6903
 
  __attribute__((cleanup(cleanup_string)))
6904
 
    char *agent_directory = NULL;
6905
 
  char *helper_directory = NULL;
6906
 
  uid_t user = 0;
6907
 
  gid_t group = 0;
6908
 
  __attribute__((cleanup(cleanup_string)))
6909
 
    char *mandos_argz = NULL;
6910
 
  size_t mandos_argz_length = 0;
6911
 
 
6912
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
6913
 
                                &helper_directory, &user, &group,
6914
 
                                &mandos_argz, &mandos_argz_length));
6915
 
 
6916
 
  g_assert_cmpstr(agent_directory, ==, "/tmp");
6917
 
  g_assert_null(helper_directory);
6918
 
  g_assert_true(user == 0);
6919
 
  g_assert_true(group == 0);
6920
 
  g_assert_null(mandos_argz);
6921
 
  g_assert_true(mandos_argz_length == 0);
6922
 
 
6923
 
  for(char **arg = argv; *arg != NULL; arg++){
6924
 
    free(*arg);
6925
 
  }
6926
 
}
6927
 
 
6928
 
static
6929
 
void test_parse_arguments_helper_directory(__attribute__((unused))
6930
 
                                           test_fixture *fixture,
6931
 
                                           __attribute__((unused))
6932
 
                                           gconstpointer user_data){
6933
 
  char *argv[] = {
6934
 
    strdup("prgname"),
6935
 
    strdup("--helper-directory"),
6936
 
    strdup("/tmp"),
6937
 
    NULL };
6938
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6939
 
 
6940
 
  char *agent_directory = NULL;
6941
 
  __attribute__((cleanup(cleanup_string)))
6942
 
    char *helper_directory = NULL;
6943
 
  uid_t user = 0;
6944
 
  gid_t group = 0;
6945
 
  __attribute__((cleanup(cleanup_string)))
6946
 
    char *mandos_argz = NULL;
6947
 
  size_t mandos_argz_length = 0;
6948
 
 
6949
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
6950
 
                                &helper_directory, &user, &group,
6951
 
                                &mandos_argz, &mandos_argz_length));
6952
 
 
6953
 
  g_assert_cmpstr(helper_directory, ==, "/tmp");
6954
 
  g_assert_null(agent_directory);
6955
 
  g_assert_true(user == 0);
6956
 
  g_assert_true(group == 0);
6957
 
  g_assert_null(mandos_argz);
6958
 
  g_assert_true(mandos_argz_length == 0);
6959
 
 
6960
 
  for(char **arg = argv; *arg != NULL; arg++){
6961
 
    free(*arg);
6962
 
  }
6963
 
}
6964
 
 
6965
 
static
6966
 
void test_parse_arguments_plugin_helper_dir(__attribute__((unused))
6967
 
                                            test_fixture *fixture,
6968
 
                                            __attribute__((unused))
6969
 
                                            gconstpointer user_data){
6970
 
  char *argv[] = {
6971
 
    strdup("prgname"),
6972
 
    strdup("--plugin-helper-dir"),
6973
 
    strdup("/tmp"),
6974
 
    NULL };
6975
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
6976
 
 
6977
 
  char *agent_directory = NULL;
6978
 
  __attribute__((cleanup(cleanup_string)))
6979
 
    char *helper_directory = NULL;
6980
 
  uid_t user = 0;
6981
 
  gid_t group = 0;
6982
 
  __attribute__((cleanup(cleanup_string)))
6983
 
    char *mandos_argz = NULL;
6984
 
  size_t mandos_argz_length = 0;
6985
 
 
6986
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
6987
 
                                &helper_directory, &user, &group,
6988
 
                                &mandos_argz, &mandos_argz_length));
6989
 
 
6990
 
  g_assert_cmpstr(helper_directory, ==, "/tmp");
6991
 
  g_assert_null(agent_directory);
6992
 
  g_assert_true(user == 0);
6993
 
  g_assert_true(group == 0);
6994
 
  g_assert_null(mandos_argz);
6995
 
  g_assert_true(mandos_argz_length == 0);
6996
 
 
6997
 
  for(char **arg = argv; *arg != NULL; arg++){
6998
 
    free(*arg);
6999
 
  }
7000
 
}
7001
 
 
7002
 
static void test_parse_arguments_user(__attribute__((unused))
7003
 
                                      test_fixture *fixture,
7004
 
                                      __attribute__((unused))
7005
 
                                      gconstpointer user_data){
7006
 
  char *argv[] = {
7007
 
    strdup("prgname"),
7008
 
    strdup("--user"),
7009
 
    strdup("1000"),
7010
 
    NULL };
7011
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7012
 
 
7013
 
  char *agent_directory = NULL;
7014
 
  __attribute__((cleanup(cleanup_string)))
7015
 
    char *helper_directory = NULL;
7016
 
  uid_t user = 0;
7017
 
  gid_t group = 0;
7018
 
  __attribute__((cleanup(cleanup_string)))
7019
 
    char *mandos_argz = NULL;
7020
 
  size_t mandos_argz_length = 0;
7021
 
 
7022
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7023
 
                                &helper_directory, &user, &group,
7024
 
                                &mandos_argz, &mandos_argz_length));
7025
 
 
7026
 
  g_assert_null(helper_directory);
7027
 
  g_assert_null(agent_directory);
7028
 
  g_assert_cmpuint((unsigned int)user, ==, 1000);
7029
 
  g_assert_true(group == 0);
7030
 
  g_assert_null(mandos_argz);
7031
 
  g_assert_true(mandos_argz_length == 0);
7032
 
 
7033
 
  for(char **arg = argv; *arg != NULL; arg++){
7034
 
    free(*arg);
7035
 
  }
7036
 
}
7037
 
 
7038
 
static void test_parse_arguments_user_invalid(__attribute__((unused))
7039
 
                                              test_fixture *fixture,
7040
 
                                              __attribute__((unused))
7041
 
                                              gconstpointer
7042
 
                                              user_data){
7043
 
  char *argv[] = {
7044
 
    strdup("prgname"),
7045
 
    strdup("--user"),
7046
 
    strdup("invalid"),
7047
 
    NULL };
7048
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7049
 
 
7050
 
  char *agent_directory = NULL;
7051
 
  __attribute__((cleanup(cleanup_string)))
7052
 
    char *helper_directory = NULL;
7053
 
  uid_t user = 0;
7054
 
  gid_t group = 0;
7055
 
  __attribute__((cleanup(cleanup_string)))
7056
 
    char *mandos_argz = NULL;
7057
 
  size_t mandos_argz_length = 0;
7058
 
 
7059
 
  g_assert_false(parse_arguments_devnull(argc, argv, false,
7060
 
                                         &agent_directory,
7061
 
                                         &helper_directory, &user,
7062
 
                                         &group, &mandos_argz,
7063
 
                                         &mandos_argz_length));
7064
 
 
7065
 
  g_assert_null(helper_directory);
7066
 
  g_assert_null(agent_directory);
7067
 
  g_assert_cmpuint((unsigned int)user, ==, 0);
7068
 
  g_assert_true(group == 0);
7069
 
  g_assert_null(mandos_argz);
7070
 
  g_assert_true(mandos_argz_length == 0);
7071
 
 
7072
 
  for(char **arg = argv; *arg != NULL; arg++){
7073
 
    free(*arg);
7074
 
  }
7075
 
}
7076
 
 
7077
 
static
7078
 
void test_parse_arguments_user_zero_invalid(__attribute__((unused))
7079
 
                                            test_fixture *fixture,
7080
 
                                            __attribute__((unused))
7081
 
                                            gconstpointer user_data){
7082
 
  char *argv[] = {
7083
 
    strdup("prgname"),
7084
 
    strdup("--user"),
7085
 
    strdup("0"),
7086
 
    NULL };
7087
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7088
 
 
7089
 
  char *agent_directory = NULL;
7090
 
  __attribute__((cleanup(cleanup_string)))
7091
 
    char *helper_directory = NULL;
7092
 
  uid_t user = 0;
7093
 
  gid_t group = 0;
7094
 
  __attribute__((cleanup(cleanup_string)))
7095
 
    char *mandos_argz = NULL;
7096
 
  size_t mandos_argz_length = 0;
7097
 
 
7098
 
  g_assert_false(parse_arguments_devnull(argc, argv, false,
7099
 
                                         &agent_directory,
7100
 
                                         &helper_directory, &user,
7101
 
                                         &group, &mandos_argz,
7102
 
                                         &mandos_argz_length));
7103
 
 
7104
 
  g_assert_null(helper_directory);
7105
 
  g_assert_null(agent_directory);
7106
 
  g_assert_cmpuint((unsigned int)user, ==, 0);
7107
 
  g_assert_true(group == 0);
7108
 
  g_assert_null(mandos_argz);
7109
 
  g_assert_true(mandos_argz_length == 0);
7110
 
 
7111
 
  for(char **arg = argv; *arg != NULL; arg++){
7112
 
    free(*arg);
7113
 
  }
7114
 
}
7115
 
 
7116
 
static void test_parse_arguments_group(__attribute__((unused))
7117
 
                                       test_fixture *fixture,
7118
 
                                       __attribute__((unused))
7119
 
                                       gconstpointer user_data){
7120
 
  char *argv[] = {
7121
 
    strdup("prgname"),
7122
 
    strdup("--group"),
7123
 
    strdup("1000"),
7124
 
    NULL };
7125
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7126
 
 
7127
 
  char *agent_directory = NULL;
7128
 
  __attribute__((cleanup(cleanup_string)))
7129
 
    char *helper_directory = NULL;
7130
 
  uid_t user = 0;
7131
 
  gid_t group = 0;
7132
 
  __attribute__((cleanup(cleanup_string)))
7133
 
    char *mandos_argz = NULL;
7134
 
  size_t mandos_argz_length = 0;
7135
 
 
7136
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7137
 
                                &helper_directory, &user, &group,
7138
 
                                &mandos_argz, &mandos_argz_length));
7139
 
 
7140
 
  g_assert_null(helper_directory);
7141
 
  g_assert_null(agent_directory);
7142
 
  g_assert_true(user == 0);
7143
 
  g_assert_cmpuint((unsigned int)group, ==, 1000);
7144
 
  g_assert_null(mandos_argz);
7145
 
  g_assert_true(mandos_argz_length == 0);
7146
 
 
7147
 
  for(char **arg = argv; *arg != NULL; arg++){
7148
 
    free(*arg);
7149
 
  }
7150
 
}
7151
 
 
7152
 
static void test_parse_arguments_group_invalid(__attribute__((unused))
7153
 
                                               test_fixture *fixture,
7154
 
                                               __attribute__((unused))
7155
 
                                               gconstpointer
7156
 
                                               user_data){
7157
 
  char *argv[] = {
7158
 
    strdup("prgname"),
7159
 
    strdup("--group"),
7160
 
    strdup("invalid"),
7161
 
    NULL };
7162
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7163
 
 
7164
 
  char *agent_directory = NULL;
7165
 
  __attribute__((cleanup(cleanup_string)))
7166
 
    char *helper_directory = NULL;
7167
 
  uid_t user = 0;
7168
 
  gid_t group = 0;
7169
 
  __attribute__((cleanup(cleanup_string)))
7170
 
    char *mandos_argz = NULL;
7171
 
  size_t mandos_argz_length = 0;
7172
 
 
7173
 
  g_assert_false(parse_arguments_devnull(argc, argv, false,
7174
 
                                         &agent_directory,
7175
 
                                         &helper_directory, &user,
7176
 
                                         &group, &mandos_argz,
7177
 
                                         &mandos_argz_length));
7178
 
 
7179
 
  g_assert_null(helper_directory);
7180
 
  g_assert_null(agent_directory);
7181
 
  g_assert_true(user == 0);
7182
 
  g_assert_true(group == 0);
7183
 
  g_assert_null(mandos_argz);
7184
 
  g_assert_true(mandos_argz_length == 0);
7185
 
 
7186
 
  for(char **arg = argv; *arg != NULL; arg++){
7187
 
    free(*arg);
7188
 
  }
7189
 
}
7190
 
 
7191
 
static
7192
 
void test_parse_arguments_group_zero_invalid(__attribute__((unused))
7193
 
                                             test_fixture *fixture,
7194
 
                                             __attribute__((unused))
7195
 
                                             gconstpointer user_data){
7196
 
  char *argv[] = {
7197
 
    strdup("prgname"),
7198
 
    strdup("--group"),
7199
 
    strdup("0"),
7200
 
    NULL };
7201
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7202
 
 
7203
 
  char *agent_directory = NULL;
7204
 
  __attribute__((cleanup(cleanup_string)))
7205
 
    char *helper_directory = NULL;
7206
 
  uid_t user = 0;
7207
 
  gid_t group = 0;
7208
 
  __attribute__((cleanup(cleanup_string)))
7209
 
    char *mandos_argz = NULL;
7210
 
  size_t mandos_argz_length = 0;
7211
 
 
7212
 
  g_assert_false(parse_arguments_devnull(argc, argv, false,
7213
 
                                         &agent_directory,
7214
 
                                         &helper_directory, &user,
7215
 
                                         &group, &mandos_argz,
7216
 
                                         &mandos_argz_length));
7217
 
 
7218
 
  g_assert_null(helper_directory);
7219
 
  g_assert_null(agent_directory);
7220
 
  g_assert_cmpuint((unsigned int)group, ==, 0);
7221
 
  g_assert_true(group == 0);
7222
 
  g_assert_null(mandos_argz);
7223
 
  g_assert_true(mandos_argz_length == 0);
7224
 
 
7225
 
  for(char **arg = argv; *arg != NULL; arg++){
7226
 
    free(*arg);
7227
 
  }
7228
 
}
7229
 
 
7230
 
static void test_parse_arguments_mandos_noargs(__attribute__((unused))
7231
 
                                               test_fixture *fixture,
7232
 
                                               __attribute__((unused))
7233
 
                                               gconstpointer
7234
 
                                               user_data){
7235
 
  char *argv[] = {
7236
 
    strdup("prgname"),
7237
 
    strdup("mandos-client"),
7238
 
    NULL };
7239
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7240
 
 
7241
 
  __attribute__((cleanup(cleanup_string)))
7242
 
    char *agent_directory = NULL;
7243
 
  __attribute__((cleanup(cleanup_string)))
7244
 
    char *helper_directory = NULL;
7245
 
  uid_t user = 0;
7246
 
  gid_t group = 0;
7247
 
  __attribute__((cleanup(cleanup_string)))
7248
 
    char *mandos_argz = NULL;
7249
 
  size_t mandos_argz_length = 0;
7250
 
 
7251
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7252
 
                                &helper_directory, &user, &group,
7253
 
                                &mandos_argz, &mandos_argz_length));
7254
 
 
7255
 
  g_assert_null(agent_directory);
7256
 
  g_assert_null(helper_directory);
7257
 
  g_assert_true(user == 0);
7258
 
  g_assert_true(group == 0);
7259
 
  g_assert_cmpstr(mandos_argz, ==, "mandos-client");
7260
 
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
7261
 
                                            mandos_argz_length),
7262
 
                   ==, 1);
7263
 
 
7264
 
  for(char **arg = argv; *arg != NULL; arg++){
7265
 
    free(*arg);
7266
 
  }
7267
 
}
7268
 
 
7269
 
static void test_parse_arguments_mandos_args(__attribute__((unused))
7270
 
                                             test_fixture *fixture,
7271
 
                                             __attribute__((unused))
7272
 
                                             gconstpointer user_data){
7273
 
  char *argv[] = {
7274
 
    strdup("prgname"),
7275
 
    strdup("mandos-client"),
7276
 
    strdup("one"),
7277
 
    strdup("two"),
7278
 
    strdup("three"),
7279
 
    NULL };
7280
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7281
 
 
7282
 
  __attribute__((cleanup(cleanup_string)))
7283
 
    char *agent_directory = NULL;
7284
 
  __attribute__((cleanup(cleanup_string)))
7285
 
    char *helper_directory = NULL;
7286
 
  uid_t user = 0;
7287
 
  gid_t group = 0;
7288
 
  __attribute__((cleanup(cleanup_string)))
7289
 
    char *mandos_argz = NULL;
7290
 
  size_t mandos_argz_length = 0;
7291
 
 
7292
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7293
 
                                &helper_directory, &user, &group,
7294
 
                                &mandos_argz, &mandos_argz_length));
7295
 
 
7296
 
  g_assert_null(agent_directory);
7297
 
  g_assert_null(helper_directory);
7298
 
  g_assert_true(user == 0);
7299
 
  g_assert_true(group == 0);
7300
 
  char *marg = mandos_argz;
7301
 
  g_assert_cmpstr(marg, ==, "mandos-client");
7302
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7303
 
  g_assert_cmpstr(marg, ==, "one");
7304
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7305
 
  g_assert_cmpstr(marg, ==, "two");
7306
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7307
 
  g_assert_cmpstr(marg, ==, "three");
7308
 
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
7309
 
                                            mandos_argz_length),
7310
 
                   ==, 4);
7311
 
 
7312
 
  for(char **arg = argv; *arg != NULL; arg++){
7313
 
    free(*arg);
7314
 
  }
7315
 
}
7316
 
 
7317
 
static void test_parse_arguments_all_args(__attribute__((unused))
7318
 
                                          test_fixture *fixture,
7319
 
                                          __attribute__((unused))
7320
 
                                          gconstpointer user_data){
7321
 
  char *argv[] = {
7322
 
    strdup("prgname"),
7323
 
    strdup("--agent-directory"),
7324
 
    strdup("/tmp"),
7325
 
    strdup("--helper-directory"),
7326
 
    strdup("/var/tmp"),
7327
 
    strdup("--user"),
7328
 
    strdup("1"),
7329
 
    strdup("--group"),
7330
 
    strdup("2"),
7331
 
    strdup("mandos-client"),
7332
 
    strdup("one"),
7333
 
    strdup("two"),
7334
 
    strdup("three"),
7335
 
    NULL };
7336
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7337
 
 
7338
 
  __attribute__((cleanup(cleanup_string)))
7339
 
    char *agent_directory = NULL;
7340
 
  __attribute__((cleanup(cleanup_string)))
7341
 
    char *helper_directory = NULL;
7342
 
  uid_t user = 0;
7343
 
  gid_t group = 0;
7344
 
  __attribute__((cleanup(cleanup_string)))
7345
 
    char *mandos_argz = NULL;
7346
 
  size_t mandos_argz_length = 0;
7347
 
 
7348
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7349
 
                                &helper_directory, &user, &group,
7350
 
                                &mandos_argz, &mandos_argz_length));
7351
 
 
7352
 
  g_assert_cmpstr(agent_directory, ==, "/tmp");
7353
 
  g_assert_cmpstr(helper_directory, ==, "/var/tmp");
7354
 
  g_assert_true(user == 1);
7355
 
  g_assert_true(group == 2);
7356
 
  char *marg = mandos_argz;
7357
 
  g_assert_cmpstr(marg, ==, "mandos-client");
7358
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7359
 
  g_assert_cmpstr(marg, ==, "one");
7360
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7361
 
  g_assert_cmpstr(marg, ==, "two");
7362
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7363
 
  g_assert_cmpstr(marg, ==, "three");
7364
 
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
7365
 
                                            mandos_argz_length),
7366
 
                   ==, 4);
7367
 
 
7368
 
  for(char **arg = argv; *arg != NULL; arg++){
7369
 
    free(*arg);
7370
 
  }
7371
 
}
7372
 
 
7373
 
static void test_parse_arguments_mixed(__attribute__((unused))
7374
 
                                       test_fixture *fixture,
7375
 
                                       __attribute__((unused))
7376
 
                                       gconstpointer user_data){
7377
 
  char *argv[] = {
7378
 
    strdup("prgname"),
7379
 
    strdup("mandos-client"),
7380
 
    strdup("--user"),
7381
 
    strdup("1"),
7382
 
    strdup("one"),
7383
 
    strdup("--agent-directory"),
7384
 
    strdup("/tmp"),
7385
 
    strdup("two"),
7386
 
    strdup("three"),
7387
 
    strdup("--helper-directory=/var/tmp"),
7388
 
    NULL };
7389
 
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
7390
 
 
7391
 
  __attribute__((cleanup(cleanup_string)))
7392
 
    char *agent_directory = NULL;
7393
 
  __attribute__((cleanup(cleanup_string)))
7394
 
    char *helper_directory = NULL;
7395
 
  uid_t user = 0;
7396
 
  gid_t group = 0;
7397
 
  __attribute__((cleanup(cleanup_string)))
7398
 
    char *mandos_argz = NULL;
7399
 
  size_t mandos_argz_length = 0;
7400
 
 
7401
 
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
7402
 
                                &helper_directory, &user, &group,
7403
 
                                &mandos_argz, &mandos_argz_length));
7404
 
 
7405
 
  g_assert_cmpstr(agent_directory, ==, "/tmp");
7406
 
  g_assert_cmpstr(helper_directory, ==, "/var/tmp");
7407
 
  g_assert_true(user == 1);
7408
 
  g_assert_true(group == 0);
7409
 
  char *marg = mandos_argz;
7410
 
  g_assert_cmpstr(marg, ==, "mandos-client");
7411
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7412
 
  g_assert_cmpstr(marg, ==, "one");
7413
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7414
 
  g_assert_cmpstr(marg, ==, "two");
7415
 
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
7416
 
  g_assert_cmpstr(marg, ==, "three");
7417
 
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
7418
 
                                            mandos_argz_length),
7419
 
                   ==, 4);
7420
 
 
7421
 
  for(char **arg = argv; *arg != NULL; arg++){
7422
 
    free(*arg);
7423
 
  }
7424
 
}
7425
 
 
7426
 
/* End of tests section */
7427
 
 
7428
 
/* Test boilerplate section; New tests should be added to the test
7429
 
   suite definition here, in the "run_tests" function.
7430
 
 
7431
 
   Finally, this section also contains the should_only_run_tests()
7432
 
   function used by main() for deciding if tests should be run or to
7433
 
   start normally. */
7434
 
 
7435
 
__attribute__((cold))
7436
 
static bool run_tests(int argc, char *argv[]){
7437
 
  g_test_init(&argc, &argv, NULL);
7438
 
 
7439
 
  /* A macro to add a test with no setup or teardown functions */
7440
 
#define test_add(testpath, testfunc)                    \
7441
 
  do {                                                  \
7442
 
    g_test_add((testpath), test_fixture, NULL, NULL,    \
7443
 
               (testfunc), NULL);                       \
7444
 
  } while(false)
7445
 
 
7446
 
  /* Test the signal-related functions first, since some other tests
7447
 
     depend on these functions in their setups and teardowns */
7448
 
  test_add("/signal-handling/setup", test_setup_signal_handler);
7449
 
  test_add("/signal-handling/restore", test_restore_signal_handler);
7450
 
  test_add("/signal-handling/block", test_block_sigchld);
7451
 
  test_add("/signal-handling/restore-sigmask", test_restore_sigmask);
7452
 
 
7453
 
  /* Regular non-signal-related tests; these use no setups or
7454
 
     teardowns */
7455
 
  test_add("/parse_arguments/noargs", test_parse_arguments_noargs);
7456
 
  test_add("/parse_arguments/invalid", test_parse_arguments_invalid);
7457
 
  test_add("/parse_arguments/long-dir",
7458
 
           test_parse_arguments_long_dir);
7459
 
  test_add("/parse_arguments/short-dir",
7460
 
           test_parse_arguments_short_dir);
7461
 
  test_add("/parse_arguments/helper-directory",
7462
 
           test_parse_arguments_helper_directory);
7463
 
  test_add("/parse_arguments/plugin-helper-dir",
7464
 
           test_parse_arguments_plugin_helper_dir);
7465
 
  test_add("/parse_arguments/user", test_parse_arguments_user);
7466
 
  test_add("/parse_arguments/user-invalid",
7467
 
           test_parse_arguments_user_invalid);
7468
 
  test_add("/parse_arguments/user-zero-invalid",
7469
 
           test_parse_arguments_user_zero_invalid);
7470
 
  test_add("/parse_arguments/group", test_parse_arguments_group);
7471
 
  test_add("/parse_arguments/group-invalid",
7472
 
           test_parse_arguments_group_invalid);
7473
 
  test_add("/parse_arguments/group-zero-invalid",
7474
 
           test_parse_arguments_group_zero_invalid);
7475
 
  test_add("/parse_arguments/mandos-noargs",
7476
 
           test_parse_arguments_mandos_noargs);
7477
 
  test_add("/parse_arguments/mandos-args",
7478
 
           test_parse_arguments_mandos_args);
7479
 
  test_add("/parse_arguments/all-args",
7480
 
           test_parse_arguments_all_args);
7481
 
  test_add("/parse_arguments/mixed", test_parse_arguments_mixed);
7482
 
  test_add("/queue/create", test_create_queue);
7483
 
  test_add("/queue/add", test_add_to_queue);
7484
 
  test_add("/queue/has_question/empty",
7485
 
           test_queue_has_question_empty);
7486
 
  test_add("/queue/has_question/false",
7487
 
           test_queue_has_question_false);
7488
 
  test_add("/queue/has_question/true", test_queue_has_question_true);
7489
 
  test_add("/queue/has_question/false2",
7490
 
           test_queue_has_question_false2);
7491
 
  test_add("/queue/has_question/true2",
7492
 
           test_queue_has_question_true2);
7493
 
  test_add("/buffer/cleanup", test_cleanup_buffer);
7494
 
  test_add("/string_set/net-set-contains-nothing",
7495
 
           test_string_set_new_set_contains_nothing);
7496
 
  test_add("/string_set/with-added-string-contains-it",
7497
 
           test_string_set_with_added_string_contains_it);
7498
 
  test_add("/string_set/cleared-does-not-contain-string",
7499
 
           test_string_set_cleared_does_not_contain_str);
7500
 
  test_add("/string_set/swap/one-with-empty",
7501
 
           test_string_set_swap_one_with_empty);
7502
 
  test_add("/string_set/swap/empty-with-one",
7503
 
           test_string_set_swap_empty_with_one);
7504
 
  test_add("/string_set/swap/one-with-one",
7505
 
           test_string_set_swap_one_with_one);
7506
 
 
7507
 
  /* A macro to add a test using the setup and teardown functions */
7508
 
#define test_add_st(path, func)                                 \
7509
 
  do {                                                          \
7510
 
    g_test_add((path), test_fixture, NULL, test_setup, (func),  \
7511
 
               test_teardown);                                  \
7512
 
  } while(false)
7513
 
 
7514
 
  /* Signal-related tests; these use setups and teardowns which
7515
 
     establish, during each test run, a signal handler for, and a
7516
 
     signal mask blocking, the SIGCHLD signal, just like main() */
7517
 
  test_add_st("/wait_for_event/timeout", test_wait_for_event_timeout);
7518
 
  test_add_st("/wait_for_event/event", test_wait_for_event_event);
7519
 
  test_add_st("/wait_for_event/sigchld", test_wait_for_event_sigchld);
7520
 
  test_add_st("/run_queue/zeroes-next-run",
7521
 
              test_run_queue_zeroes_next_run);
7522
 
  test_add_st("/run_queue/clears-cancelled_filenames",
7523
 
              test_run_queue_clears_cancelled_filenames);
7524
 
  test_add_st("/run_queue/skips-cancelled-filenames",
7525
 
              test_run_queue_skips_cancelled_filenames);
7526
 
  test_add_st("/run_queue/one-task", test_run_queue_one_task);
7527
 
  test_add_st("/run_queue/two-tasks", test_run_queue_two_tasks);
7528
 
  test_add_st("/run_queue/two-tasks/quit",
7529
 
              test_run_queue_two_tasks_quit);
7530
 
  test_add_st("/run_queue/two-tasks-cleanup",
7531
 
              test_run_queue_two_tasks_cleanup);
7532
 
  test_add_st("/task-creators/start_mandos_client",
7533
 
              test_start_mandos_client);
7534
 
  test_add_st("/task-creators/start_mandos_client/execv",
7535
 
              test_start_mandos_client_execv);
7536
 
  test_add_st("/task-creators/start_mandos_client/suid/euid",
7537
 
              test_start_mandos_client_suid_euid);
7538
 
  test_add_st("/task-creators/start_mandos_client/suid/egid",
7539
 
              test_start_mandos_client_suid_egid);
7540
 
  test_add_st("/task-creators/start_mandos_client/suid/ruid",
7541
 
              test_start_mandos_client_suid_ruid);
7542
 
  test_add_st("/task-creators/start_mandos_client/suid/rgid",
7543
 
              test_start_mandos_client_suid_rgid);
7544
 
  test_add_st("/task-creators/start_mandos_client/read",
7545
 
              test_start_mandos_client_read);
7546
 
  test_add_st("/task-creators/start_mandos_client/helper-directory",
7547
 
              test_start_mandos_client_helper_directory);
7548
 
  test_add_st("/task-creators/start_mandos_client/sigmask",
7549
 
              test_start_mandos_client_sigmask);
7550
 
  test_add_st("/task/wait_for_mandos_client_exit/badpid",
7551
 
              test_wait_for_mandos_client_exit_badpid);
7552
 
  test_add_st("/task/wait_for_mandos_client_exit/noexit",
7553
 
              test_wait_for_mandos_client_exit_noexit);
7554
 
  test_add_st("/task/wait_for_mandos_client_exit/success",
7555
 
              test_wait_for_mandos_client_exit_success);
7556
 
  test_add_st("/task/wait_for_mandos_client_exit/failure",
7557
 
              test_wait_for_mandos_client_exit_failure);
7558
 
  test_add_st("/task/wait_for_mandos_client_exit/killed",
7559
 
              test_wait_for_mandos_client_exit_killed);
7560
 
  test_add_st("/task/read_mandos_client_output/readerror",
7561
 
              test_read_mandos_client_output_readerror);
7562
 
  test_add_st("/task/read_mandos_client_output/nodata",
7563
 
              test_read_mandos_client_output_nodata);
7564
 
  test_add_st("/task/read_mandos_client_output/eof",
7565
 
              test_read_mandos_client_output_eof);
7566
 
  test_add_st("/task/read_mandos_client_output/once",
7567
 
              test_read_mandos_client_output_once);
7568
 
  test_add_st("/task/read_mandos_client_output/malloc",
7569
 
              test_read_mandos_client_output_malloc);
7570
 
  test_add_st("/task/read_mandos_client_output/append",
7571
 
              test_read_mandos_client_output_append);
7572
 
  test_add_st("/task-creators/add_inotify_dir_watch",
7573
 
              test_add_inotify_dir_watch);
7574
 
  test_add_st("/task-creators/add_inotify_dir_watch/fail",
7575
 
              test_add_inotify_dir_watch_fail);
7576
 
  test_add_st("/task-creators/add_inotify_dir_watch/EAGAIN",
7577
 
              test_add_inotify_dir_watch_EAGAIN);
7578
 
  test_add_st("/task-creators/add_inotify_dir_watch/IN_CLOSE_WRITE",
7579
 
              test_add_inotify_dir_watch_IN_CLOSE_WRITE);
7580
 
  test_add_st("/task-creators/add_inotify_dir_watch/IN_MOVED_TO",
7581
 
              test_add_inotify_dir_watch_IN_MOVED_TO);
7582
 
  test_add_st("/task-creators/add_inotify_dir_watch/IN_DELETE",
7583
 
              test_add_inotify_dir_watch_IN_DELETE);
7584
 
  test_add_st("/task/read_inotify_event/readerror",
7585
 
              test_read_inotify_event_readerror);
7586
 
  test_add_st("/task/read_inotify_event/bad-epoll",
7587
 
              test_read_inotify_event_bad_epoll);
7588
 
  test_add_st("/task/read_inotify_event/nodata",
7589
 
              test_read_inotify_event_nodata);
7590
 
  test_add_st("/task/read_inotify_event/eof",
7591
 
              test_read_inotify_event_eof);
7592
 
  test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE",
7593
 
              test_read_inotify_event_IN_CLOSE_WRITE);
7594
 
  test_add_st("/task/read_inotify_event/IN_MOVED_TO",
7595
 
              test_read_inotify_event_IN_MOVED_TO);
7596
 
  test_add_st("/task/read_inotify_event/IN_DELETE",
7597
 
              test_read_inotify_event_IN_DELETE);
7598
 
  test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE/badname",
7599
 
              test_read_inotify_event_IN_CLOSE_WRITE_badname);
7600
 
  test_add_st("/task/read_inotify_event/IN_MOVED_TO/badname",
7601
 
              test_read_inotify_event_IN_MOVED_TO_badname);
7602
 
  test_add_st("/task/read_inotify_event/IN_DELETE/badname",
7603
 
              test_read_inotify_event_IN_DELETE_badname);
7604
 
  test_add_st("/task/open_and_parse_question/ENOENT",
7605
 
              test_open_and_parse_question_ENOENT);
7606
 
  test_add_st("/task/open_and_parse_question/EIO",
7607
 
              test_open_and_parse_question_EIO);
7608
 
  test_add_st("/task/open_and_parse_question/parse-error",
7609
 
              test_open_and_parse_question_parse_error);
7610
 
  test_add_st("/task/open_and_parse_question/nosocket",
7611
 
              test_open_and_parse_question_nosocket);
7612
 
  test_add_st("/task/open_and_parse_question/badsocket",
7613
 
              test_open_and_parse_question_badsocket);
7614
 
  test_add_st("/task/open_and_parse_question/nopid",
7615
 
              test_open_and_parse_question_nopid);
7616
 
  test_add_st("/task/open_and_parse_question/badpid",
7617
 
              test_open_and_parse_question_badpid);
7618
 
  test_add_st("/task/open_and_parse_question/noexist_pid",
7619
 
              test_open_and_parse_question_noexist_pid);
7620
 
  test_add_st("/task/open_and_parse_question/no-notafter",
7621
 
              test_open_and_parse_question_no_notafter);
7622
 
  test_add_st("/task/open_and_parse_question/bad-notafter",
7623
 
              test_open_and_parse_question_bad_notafter);
7624
 
  test_add_st("/task/open_and_parse_question/notafter-0",
7625
 
              test_open_and_parse_question_notafter_0);
7626
 
  test_add_st("/task/open_and_parse_question/notafter-1",
7627
 
              test_open_and_parse_question_notafter_1);
7628
 
  test_add_st("/task/open_and_parse_question/notafter-1-1",
7629
 
              test_open_and_parse_question_notafter_1_1);
7630
 
  test_add_st("/task/open_and_parse_question/notafter-1-2",
7631
 
              test_open_and_parse_question_notafter_1_2);
7632
 
  test_add_st("/task/open_and_parse_question/equal-notafter",
7633
 
              test_open_and_parse_question_equal_notafter);
7634
 
  test_add_st("/task/open_and_parse_question/late-notafter",
7635
 
              test_open_and_parse_question_late_notafter);
7636
 
  test_add_st("/task/cancel_old_question/0-1-2",
7637
 
              test_cancel_old_question_0_1_2);
7638
 
  test_add_st("/task/cancel_old_question/0-2-1",
7639
 
              test_cancel_old_question_0_2_1);
7640
 
  test_add_st("/task/cancel_old_question/1-2-3",
7641
 
              test_cancel_old_question_1_2_3);
7642
 
  test_add_st("/task/cancel_old_question/1-3-2",
7643
 
              test_cancel_old_question_1_3_2);
7644
 
  test_add_st("/task/cancel_old_question/2-1-3",
7645
 
              test_cancel_old_question_2_1_3);
7646
 
  test_add_st("/task/cancel_old_question/2-3-1",
7647
 
              test_cancel_old_question_2_3_1);
7648
 
  test_add_st("/task/cancel_old_question/3-1-2",
7649
 
              test_cancel_old_question_3_1_2);
7650
 
  test_add_st("/task/cancel_old_question/3-2-1",
7651
 
              test_cancel_old_question_3_2_1);
7652
 
  test_add_st("/task/connect_question_socket/name-too-long",
7653
 
              test_connect_question_socket_name_too_long);
7654
 
  test_add_st("/task/connect_question_socket/connect-fail",
7655
 
              test_connect_question_socket_connect_fail);
7656
 
  test_add_st("/task/connect_question_socket/bad-epoll",
7657
 
              test_connect_question_socket_bad_epoll);
7658
 
  test_add_st("/task/connect_question_socket/usable",
7659
 
              test_connect_question_socket_usable);
7660
 
  test_add_st("/task/send_password_to_socket/client-not-exited",
7661
 
              test_send_password_to_socket_client_not_exited);
7662
 
  test_add_st("/task/send_password_to_socket/password-not-read",
7663
 
              test_send_password_to_socket_password_not_read);
7664
 
  test_add_st("/task/send_password_to_socket/EMSGSIZE",
7665
 
              test_send_password_to_socket_EMSGSIZE);
7666
 
  test_add_st("/task/send_password_to_socket/retry",
7667
 
              test_send_password_to_socket_retry);
7668
 
  test_add_st("/task/send_password_to_socket/bad-epoll",
7669
 
              test_send_password_to_socket_bad_epoll);
7670
 
  test_add_st("/task/send_password_to_socket/null-password",
7671
 
              test_send_password_to_socket_null_password);
7672
 
  test_add_st("/task/send_password_to_socket/empty-password",
7673
 
              test_send_password_to_socket_empty_password);
7674
 
  test_add_st("/task/send_password_to_socket/empty-str-password",
7675
 
              test_send_password_to_socket_empty_str_pass);
7676
 
  test_add_st("/task/send_password_to_socket/text-password",
7677
 
              test_send_password_to_socket_text_password);
7678
 
  test_add_st("/task/send_password_to_socket/binary-password",
7679
 
              test_send_password_to_socket_binary_password);
7680
 
  test_add_st("/task/send_password_to_socket/nuls-in-password",
7681
 
              test_send_password_to_socket_nuls_in_password);
7682
 
  test_add_st("/task-creators/add_existing_questions/ENOENT",
7683
 
              test_add_existing_questions_ENOENT);
7684
 
  test_add_st("/task-creators/add_existing_questions/no-questions",
7685
 
              test_add_existing_questions_no_questions);
7686
 
  test_add_st("/task-creators/add_existing_questions/one-question",
7687
 
              test_add_existing_questions_one_question);
7688
 
  test_add_st("/task-creators/add_existing_questions/two-questions",
7689
 
              test_add_existing_questions_two_questions);
7690
 
  test_add_st("/task-creators/add_existing_questions/non-questions",
7691
 
              test_add_existing_questions_non_questions);
7692
 
  test_add_st("/task-creators/add_existing_questions/both-types",
7693
 
              test_add_existing_questions_both_types);
7694
 
 
7695
 
  return g_test_run() == 0;
7696
 
}
7697
 
 
7698
 
static bool should_only_run_tests(int *argc_p, char **argv_p[]){
7699
 
  GOptionContext *context = g_option_context_new("");
7700
 
 
7701
 
  g_option_context_set_help_enabled(context, FALSE);
7702
 
  g_option_context_set_ignore_unknown_options(context, TRUE);
7703
 
 
7704
 
  gboolean run_tests = FALSE;
7705
 
  GOptionEntry entries[] = {
7706
 
    { "test", 0, 0, G_OPTION_ARG_NONE,
7707
 
      &run_tests, "Run tests", NULL },
7708
 
    { NULL }
7709
 
  };
7710
 
  g_option_context_add_main_entries(context, entries, NULL);
7711
 
 
7712
 
  GError *error = NULL;
7713
 
 
7714
 
  if(g_option_context_parse(context, argc_p, argv_p, &error) != TRUE){
7715
 
    g_option_context_free(context);
7716
 
    g_error("Failed to parse options: %s", error->message);
7717
 
  }
7718
 
 
7719
 
  g_option_context_free(context);
7720
 
  return run_tests != FALSE;
7721
 
}