/mandos/release

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

« back to all changes in this revision

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

  • Committer: Teddy Hogeborn
  • Date: 2020-12-06 12:02:04 UTC
  • mto: This revision was merged to the branch mainline in revision 404.
  • Revision ID: teddy@recompile.se-20201206120204-pkx839mdn2x2v2m7
Minor code cleanup

* dracut-module/password-agent.c
  (test_send_password_to_socket_EMSGSIZE): Replace a free()/malloc()
  pair with single realloc() call.

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