/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 at recompile
  • Date: 2020-07-04 13:39:36 UTC
  • mfrom: (401 release)
  • mto: This revision was merged to the branch mainline in revision 402.
  • Revision ID: teddy@recompile.se-20200704133936-lfk4sy1d48m53iqu
Merge from release branch

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://www.freedesktop.org/wiki/Software
 
1197
     /systemd/PasswordAgents/> for 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://www.freedesktop.org/wiki/Software/systemd/
 
1494
         PasswordAgents/> (Wed 08 Oct 2014 02:14:28 AM UTC)
 
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 = errno;
 
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, 0, "Password of size %" PRIuMAX " is too big",
 
1530
                (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");
 
1539
          }
 
1540
#if __GNUC__ < 7
 
1541
          /* FALLTHROUGH */
 
1542
#else
 
1543
          __attribute__((fallthrough));
 
1544
#endif
 
1545
        default:
 
1546
          error(0, saved_errno, "Failed to send() to socket %s",
 
1547
                filename);
 
1548
          if(not string_set_add(cancelled_filenames,
 
1549
                                question_filename)){
 
1550
            error(0, errno, "Failed to cancel question for file %s",
 
1551
                  question_filename);
 
1552
          }
 
1553
          cleanup_task(&task);
 
1554
          return;
 
1555
        }
 
1556
      } else {
 
1557
        /* Success */
 
1558
        cleanup_task(&task);
 
1559
        return;
 
1560
      }
 
1561
    }
 
1562
  }
 
1563
 
 
1564
  /* We failed or are not ready yet; retry later */
 
1565
 
 
1566
  if(not add_to_queue(queue, task)){
 
1567
    error(0, errno, "Failed to add send_password_to_socket for"
 
1568
          " file %s and socket %s to queue", question_filename,
 
1569
          filename);
 
1570
    cleanup_task(&task);
 
1571
  }
 
1572
 
 
1573
  /* Add the fd to the epoll set */
 
1574
  if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd,
 
1575
               &(struct epoll_event){ .events=EPOLLOUT })
 
1576
     != 0){
 
1577
    error(0, errno, "Failed to add socket file descriptor %d for"
 
1578
          " socket %s to epoll set", fd, filename);
 
1579
    /* Force the added task (send_password_to_socket) to run again, at
 
1580
       most one second from now */
 
1581
    if((queue->next_run == 0)
 
1582
       or (queue->next_run > (*current_time + 1000000))){
 
1583
      queue->next_run = *current_time + 1000000;
 
1584
    }
 
1585
  }
 
1586
}
 
1587
 
 
1588
__attribute__((warn_unused_result))
 
1589
bool add_existing_questions(task_queue *const queue,
 
1590
                            const int epoll_fd,
 
1591
                            buffer *const password,
 
1592
                            string_set *cancelled_filenames,
 
1593
                            const mono_microsecs *const current_time,
 
1594
                            bool *const mandos_client_exited,
 
1595
                            bool *const password_is_read,
 
1596
                            const char *const dirname){
 
1597
  __attribute__((cleanup(cleanup_string)))
 
1598
    char *dir_pattern = NULL;
 
1599
  const int ret = asprintf(&dir_pattern, "%s/ask.*", dirname);
 
1600
  if(ret < 0 or dir_pattern == NULL){
 
1601
    error(0, errno, "Could not create glob pattern for directory %s",
 
1602
          dirname);
 
1603
    return false;
 
1604
  }
 
1605
  __attribute__((cleanup(globfree)))
 
1606
    glob_t question_filenames = {};
 
1607
  switch(glob(dir_pattern, GLOB_ERR | GLOB_NOSORT | GLOB_MARK,
 
1608
              NULL, &question_filenames)){
 
1609
  case GLOB_ABORTED:
 
1610
  default:
 
1611
    error(0, errno, "Failed to open directory %s", dirname);
 
1612
    return false;
 
1613
  case GLOB_NOMATCH:
 
1614
    error(0, errno, "There are no question files in %s", dirname);
 
1615
    return false;
 
1616
  case GLOB_NOSPACE:
 
1617
    error(0, errno, "Could not allocate memory for question file"
 
1618
          " names in %s", dirname);
 
1619
#if __GNUC__ < 7
 
1620
    /* FALLTHROUGH */
 
1621
#else
 
1622
    __attribute__((fallthrough));
 
1623
#endif
 
1624
  case 0:
 
1625
    for(size_t i = 0; i < question_filenames.gl_pathc; i++){
 
1626
      char *const question_filename = strdup(question_filenames
 
1627
                                             .gl_pathv[i]);
 
1628
      const task_context task = {
 
1629
        .func=open_and_parse_question,
 
1630
        .epoll_fd=epoll_fd,
 
1631
        .question_filename=question_filename,
 
1632
        .filename=question_filename,
 
1633
        .password=password,
 
1634
        .cancelled_filenames=cancelled_filenames,
 
1635
        .current_time=current_time,
 
1636
        .mandos_client_exited=mandos_client_exited,
 
1637
        .password_is_read=password_is_read,
 
1638
      };
 
1639
 
 
1640
      if(question_filename == NULL
 
1641
         or not add_to_queue(queue, task)){
 
1642
        error(0, errno, "Failed to add open_and_parse_question for"
 
1643
              " file %s to queue",
 
1644
              question_filenames.gl_pathv[i]);
 
1645
        free(question_filename);
 
1646
      } else {
 
1647
        queue->next_run = 1;
 
1648
      }
 
1649
    }
 
1650
    return true;
 
1651
  }
 
1652
}
 
1653
 
 
1654
__attribute__((nonnull, warn_unused_result))
 
1655
bool wait_for_event(const int epoll_fd,
 
1656
                    const mono_microsecs queue_next_run,
 
1657
                    const mono_microsecs current_time){
 
1658
  __attribute__((const))
 
1659
    int milliseconds_to_wait(const mono_microsecs currtime,
 
1660
                             const mono_microsecs nextrun){
 
1661
    if(currtime >= nextrun){
 
1662
      return 0;
 
1663
    }
 
1664
    const uintmax_t wait_time_ms = (nextrun - currtime) / 1000;
 
1665
    if(wait_time_ms > (uintmax_t)INT_MAX){
 
1666
      return INT_MAX;
 
1667
    }
 
1668
    return (int)wait_time_ms;
 
1669
  }
 
1670
 
 
1671
  const int wait_time_ms = milliseconds_to_wait(current_time,
 
1672
                                                queue_next_run);
 
1673
 
 
1674
  /* Prepare unblocking of SIGCHLD during epoll_pwait */
 
1675
  sigset_t temporary_unblocked_sigmask;
 
1676
  /* Get current signal mask */
 
1677
  if(pthread_sigmask(-1, NULL, &temporary_unblocked_sigmask) != 0){
 
1678
    return false;
 
1679
  }
 
1680
  /* Remove SIGCHLD from the signal mask */
 
1681
  if(sigdelset(&temporary_unblocked_sigmask, SIGCHLD) != 0){
 
1682
    return false;
 
1683
  }
 
1684
  struct epoll_event events[8]; /* Ignored */
 
1685
  int ret = epoll_pwait(epoll_fd, events,
 
1686
                        sizeof(events) / sizeof(struct epoll_event),
 
1687
                        queue_next_run == 0 ? -1 : (int)wait_time_ms,
 
1688
                        &temporary_unblocked_sigmask);
 
1689
  if(ret < 0 and errno != EINTR){
 
1690
    error(0, errno, "Failed epoll_pwait(epfd=%d, ..., timeout=%d,"
 
1691
          " ...", epoll_fd,
 
1692
          queue_next_run == 0 ? -1 : (int)wait_time_ms);
 
1693
    return false;
 
1694
  }
 
1695
  return clear_all_fds_from_epoll_set(epoll_fd);
 
1696
}
 
1697
 
 
1698
bool clear_all_fds_from_epoll_set(const int epoll_fd){
 
1699
  /* Create a new empty epoll set */
 
1700
  __attribute__((cleanup(cleanup_close)))
 
1701
    const int new_epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
1702
  if(new_epoll_fd < 0){
 
1703
    return false;
 
1704
  }
 
1705
  /* dup3() the new epoll set fd over the old one, replacing it */
 
1706
  if(dup3(new_epoll_fd, epoll_fd, O_CLOEXEC) < 0){
 
1707
    return false;
 
1708
  }
 
1709
  return true;
 
1710
}
 
1711
 
 
1712
__attribute__((nonnull, warn_unused_result))
 
1713
bool run_queue(task_queue **const queue,
 
1714
               string_set *const cancelled_filenames,
 
1715
               bool *const quit_now){
 
1716
 
 
1717
  task_queue *new_queue = create_queue();
 
1718
  if(new_queue == NULL){
 
1719
    return false;
 
1720
  }
 
1721
 
 
1722
  __attribute__((cleanup(string_set_clear)))
 
1723
    string_set old_cancelled_filenames = {};
 
1724
  string_set_swap(cancelled_filenames, &old_cancelled_filenames);
 
1725
 
 
1726
  /* Declare i outside the for loop, since we might need i after the
 
1727
     loop in case we aborted in the middle */
 
1728
  size_t i;
 
1729
  for(i=0; i < (*queue)->length and not *quit_now; i++){
 
1730
    task_context *const task = &((*queue)->tasks[i]);
 
1731
    const char *const question_filename = task->question_filename;
 
1732
    /* Skip any task referencing a cancelled question filename */
 
1733
    if(question_filename != NULL
 
1734
       and string_set_contains(old_cancelled_filenames,
 
1735
                               question_filename)){
 
1736
      cleanup_task(task);
 
1737
      continue;
 
1738
    }
 
1739
    task->func(*task, new_queue);
 
1740
  }
 
1741
 
 
1742
  if(*quit_now){
 
1743
    /* we might be in the middle of the queue, so clean up any
 
1744
       remaining tasks in the current queue */
 
1745
    for(; i < (*queue)->length; i++){
 
1746
      cleanup_task(&((*queue)->tasks[i]));
 
1747
    }
 
1748
    free_queue(*queue);
 
1749
    *queue = new_queue;
 
1750
    new_queue = NULL;
 
1751
    return false;
 
1752
  }
 
1753
  free_queue(*queue);
 
1754
  *queue = new_queue;
 
1755
  new_queue = NULL;
 
1756
 
 
1757
  return true;
 
1758
}
 
1759
 
 
1760
/* End of regular code section */
 
1761
 
 
1762
/* Start of tests section; here are the tests for the above code */
 
1763
 
 
1764
/* This "fixture" data structure is used by the test setup and
 
1765
   teardown functions */
 
1766
typedef struct {
 
1767
  struct sigaction orig_sigaction;
 
1768
  sigset_t orig_sigmask;
 
1769
} test_fixture;
 
1770
 
 
1771
static void test_setup(test_fixture *fixture,
 
1772
                       __attribute__((unused))
 
1773
                       gconstpointer user_data){
 
1774
  g_assert_true(setup_signal_handler(&fixture->orig_sigaction));
 
1775
  g_assert_true(block_sigchld(&fixture->orig_sigmask));
 
1776
}
 
1777
 
 
1778
static void test_teardown(test_fixture *fixture,
 
1779
                          __attribute__((unused))
 
1780
                          gconstpointer user_data){
 
1781
  g_assert_true(restore_signal_handler(&fixture->orig_sigaction));
 
1782
  g_assert_true(restore_sigmask(&fixture->orig_sigmask));
 
1783
}
 
1784
 
 
1785
/* Utility function used by tests to search queue for matching task */
 
1786
__attribute__((pure, nonnull, warn_unused_result))
 
1787
static task_context *find_matching_task(const task_queue *const queue,
 
1788
                                        const task_context task){
 
1789
  /* The argument "task" structure is a pattern to match; 0 in any
 
1790
     member means any value matches, otherwise the value must match.
 
1791
     The filename strings are compared by strcmp(), not by pointer. */
 
1792
  for(size_t i = 0; i < queue->length; i++){
 
1793
    task_context *const current_task = queue->tasks+i;
 
1794
    /* Check all members of task_context, if set to a non-zero value.
 
1795
       If a member does not match, continue to next task in queue */
 
1796
 
 
1797
    /* task_func *const func */
 
1798
    if(task.func != NULL and current_task->func != task.func){
 
1799
      continue;
 
1800
    }
 
1801
    /* char *const question_filename; */
 
1802
    if(task.question_filename != NULL
 
1803
       and (current_task->question_filename == NULL
 
1804
            or strcmp(current_task->question_filename,
 
1805
                      task.question_filename) != 0)){
 
1806
      continue;
 
1807
    }
 
1808
    /* const pid_t pid; */
 
1809
    if(task.pid != 0 and current_task->pid != task.pid){
 
1810
      continue;
 
1811
    }
 
1812
    /* const int epoll_fd; */
 
1813
    if(task.epoll_fd != 0
 
1814
       and current_task->epoll_fd != task.epoll_fd){
 
1815
      continue;
 
1816
    }
 
1817
    /* bool *const quit_now; */
 
1818
    if(task.quit_now != NULL
 
1819
       and current_task->quit_now != task.quit_now){
 
1820
      continue;
 
1821
    }
 
1822
    /* const int fd; */
 
1823
    if(task.fd != 0 and current_task->fd != task.fd){
 
1824
      continue;
 
1825
    }
 
1826
    /* bool *const mandos_client_exited; */
 
1827
    if(task.mandos_client_exited != NULL
 
1828
       and current_task->mandos_client_exited
 
1829
       != task.mandos_client_exited){
 
1830
      continue;
 
1831
    }
 
1832
    /* buffer *const password; */
 
1833
    if(task.password != NULL
 
1834
       and current_task->password != task.password){
 
1835
      continue;
 
1836
    }
 
1837
    /* bool *const password_is_read; */
 
1838
    if(task.password_is_read != NULL
 
1839
       and current_task->password_is_read != task.password_is_read){
 
1840
      continue;
 
1841
    }
 
1842
    /* char *filename; */
 
1843
    if(task.filename != NULL
 
1844
       and (current_task->filename == NULL
 
1845
            or strcmp(current_task->filename, task.filename) != 0)){
 
1846
      continue;
 
1847
    }
 
1848
    /* string_set *const cancelled_filenames; */
 
1849
    if(task.cancelled_filenames != NULL
 
1850
       and current_task->cancelled_filenames
 
1851
       != task.cancelled_filenames){
 
1852
      continue;
 
1853
    }
 
1854
    /* const mono_microsecs notafter; */
 
1855
    if(task.notafter != 0
 
1856
       and current_task->notafter != task.notafter){
 
1857
      continue;
 
1858
    }
 
1859
    /* const mono_microsecs *const current_time; */
 
1860
    if(task.current_time != NULL
 
1861
       and current_task->current_time != task.current_time){
 
1862
      continue;
 
1863
    }
 
1864
    /* Current task matches all members; return it */
 
1865
    return current_task;
 
1866
  }
 
1867
  /* No task in queue matches passed pattern task */
 
1868
  return NULL;
 
1869
}
 
1870
 
 
1871
static void test_create_queue(__attribute__((unused))
 
1872
                              test_fixture *fixture,
 
1873
                              __attribute__((unused))
 
1874
                              gconstpointer user_data){
 
1875
  __attribute__((cleanup(cleanup_queue)))
 
1876
    task_queue *const queue = create_queue();
 
1877
  g_assert_nonnull(queue);
 
1878
  g_assert_null(queue->tasks);
 
1879
  g_assert_true(queue->length == 0);
 
1880
  g_assert_true(queue->next_run == 0);
 
1881
}
 
1882
 
 
1883
static task_func dummy_func;
 
1884
 
 
1885
static void test_add_to_queue(__attribute__((unused))
 
1886
                              test_fixture *fixture,
 
1887
                              __attribute__((unused))
 
1888
                              gconstpointer user_data){
 
1889
  __attribute__((cleanup(cleanup_queue)))
 
1890
    task_queue *queue = create_queue();
 
1891
  g_assert_nonnull(queue);
 
1892
 
 
1893
  g_assert_true(add_to_queue(queue,
 
1894
                             (task_context){ .func=dummy_func }));
 
1895
  g_assert_true(queue->length == 1);
 
1896
  g_assert_nonnull(queue->tasks);
 
1897
  g_assert_true(queue->tasks[0].func == dummy_func);
 
1898
}
 
1899
 
 
1900
static void test_add_to_queue_overflow(__attribute__((unused))
 
1901
                                       test_fixture *fixture,
 
1902
                                       __attribute__((unused))
 
1903
                                       gconstpointer user_data){
 
1904
  __attribute__((cleanup(cleanup_queue)))
 
1905
    task_queue *queue = create_queue();
 
1906
  g_assert_nonnull(queue);
 
1907
  g_assert_true(queue->length == 0);
 
1908
  queue->length = SIZE_MAX / sizeof(task_context); /* fake max size */
 
1909
 
 
1910
  FILE *real_stderr = stderr;
 
1911
  FILE *devnull = fopen("/dev/null", "we");
 
1912
  g_assert_nonnull(devnull);
 
1913
  stderr = devnull;
 
1914
  const bool ret = add_to_queue(queue,
 
1915
                                (task_context){ .func=dummy_func });
 
1916
  g_assert_true(errno == ENOMEM);
 
1917
  g_assert_false(ret);
 
1918
  stderr = real_stderr;
 
1919
  g_assert_cmpint(fclose(devnull), ==, 0);
 
1920
  queue->length = 0;            /* Restore real size */
 
1921
}
 
1922
 
 
1923
static void dummy_func(__attribute__((unused))
 
1924
                       const task_context task,
 
1925
                       __attribute__((unused))
 
1926
                       task_queue *const queue){
 
1927
}
 
1928
 
 
1929
static void test_queue_has_question_empty(__attribute__((unused))
 
1930
                                          test_fixture *fixture,
 
1931
                                          __attribute__((unused))
 
1932
                                          gconstpointer user_data){
 
1933
  __attribute__((cleanup(cleanup_queue)))
 
1934
    task_queue *queue = create_queue();
 
1935
  g_assert_nonnull(queue);
 
1936
  g_assert_false(queue_has_question(queue));
 
1937
}
 
1938
 
 
1939
static void test_queue_has_question_false(__attribute__((unused))
 
1940
                                          test_fixture *fixture,
 
1941
                                          __attribute__((unused))
 
1942
                                          gconstpointer user_data){
 
1943
  __attribute__((cleanup(cleanup_queue)))
 
1944
    task_queue *queue = create_queue();
 
1945
  g_assert_nonnull(queue);
 
1946
  g_assert_true(add_to_queue(queue,
 
1947
                             (task_context){ .func=dummy_func }));
 
1948
  g_assert_false(queue_has_question(queue));
 
1949
}
 
1950
 
 
1951
static void test_queue_has_question_true(__attribute__((unused))
 
1952
                                         test_fixture *fixture,
 
1953
                                         __attribute__((unused))
 
1954
                                         gconstpointer user_data){
 
1955
  __attribute__((cleanup(cleanup_queue)))
 
1956
    task_queue *queue = create_queue();
 
1957
  g_assert_nonnull(queue);
 
1958
  char *const question_filename
 
1959
    = strdup("/nonexistent/question_filename");
 
1960
  g_assert_nonnull(question_filename);
 
1961
  task_context task = {
 
1962
    .func=dummy_func,
 
1963
    .question_filename=question_filename,
 
1964
  };
 
1965
  g_assert_true(add_to_queue(queue, task));
 
1966
  g_assert_true(queue_has_question(queue));
 
1967
}
 
1968
 
 
1969
static void test_queue_has_question_false2(__attribute__((unused))
 
1970
                                           test_fixture *fixture,
 
1971
                                           __attribute__((unused))
 
1972
                                           gconstpointer user_data){
 
1973
  __attribute__((cleanup(cleanup_queue)))
 
1974
    task_queue *queue = create_queue();
 
1975
  g_assert_nonnull(queue);
 
1976
  task_context task = { .func=dummy_func };
 
1977
  g_assert_true(add_to_queue(queue, task));
 
1978
  g_assert_true(add_to_queue(queue, task));
 
1979
  g_assert_cmpint((int)queue->length, ==, 2);
 
1980
  g_assert_false(queue_has_question(queue));
 
1981
}
 
1982
 
 
1983
static void test_queue_has_question_true2(__attribute__((unused))
 
1984
                                          test_fixture *fixture,
 
1985
                                          __attribute__((unused))
 
1986
                                          gconstpointer user_data){
 
1987
  __attribute__((cleanup(cleanup_queue)))
 
1988
    task_queue *queue = create_queue();
 
1989
  g_assert_nonnull(queue);
 
1990
  task_context task1 = { .func=dummy_func };
 
1991
  g_assert_true(add_to_queue(queue, task1));
 
1992
  char *const question_filename
 
1993
    = strdup("/nonexistent/question_filename");
 
1994
  g_assert_nonnull(question_filename);
 
1995
  task_context task2 = {
 
1996
    .func=dummy_func,
 
1997
    .question_filename=question_filename,
 
1998
  };
 
1999
  g_assert_true(add_to_queue(queue, task2));
 
2000
  g_assert_cmpint((int)queue->length, ==, 2);
 
2001
  g_assert_true(queue_has_question(queue));
 
2002
}
 
2003
 
 
2004
static void test_cleanup_buffer(__attribute__((unused))
 
2005
                                test_fixture *fixture,
 
2006
                                __attribute__((unused))
 
2007
                                gconstpointer user_data){
 
2008
  buffer buf = {};
 
2009
 
 
2010
  const size_t buffersize = 10;
 
2011
 
 
2012
  buf.data = malloc(buffersize);
 
2013
  g_assert_nonnull(buf.data);
 
2014
  if(mlock(buf.data, buffersize) != 0){
 
2015
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
2016
  }
 
2017
 
 
2018
  cleanup_buffer(&buf);
 
2019
  g_assert_null(buf.data);
 
2020
}
 
2021
 
 
2022
static
 
2023
void test_string_set_new_set_contains_nothing(__attribute__((unused))
 
2024
                                              test_fixture *fixture,
 
2025
                                              __attribute__((unused))
 
2026
                                              gconstpointer
 
2027
                                              user_data){
 
2028
  __attribute__((cleanup(string_set_clear)))
 
2029
    string_set set = {};
 
2030
  g_assert_false(string_set_contains(set, "")); /* Empty string */
 
2031
  g_assert_false(string_set_contains(set, "test_string"));
 
2032
}
 
2033
 
 
2034
static void
 
2035
test_string_set_with_added_string_contains_it(__attribute__((unused))
 
2036
                                              test_fixture *fixture,
 
2037
                                              __attribute__((unused))
 
2038
                                              gconstpointer
 
2039
                                              user_data){
 
2040
  __attribute__((cleanup(string_set_clear)))
 
2041
    string_set set = {};
 
2042
  g_assert_true(string_set_add(&set, "test_string"));
 
2043
  g_assert_true(string_set_contains(set, "test_string"));
 
2044
}
 
2045
 
 
2046
static void
 
2047
test_string_set_cleared_does_not_contain_str(__attribute__((unused))
 
2048
                                             test_fixture *fixture,
 
2049
                                             __attribute__((unused))
 
2050
                                             gconstpointer user_data){
 
2051
  __attribute__((cleanup(string_set_clear)))
 
2052
    string_set set = {};
 
2053
  g_assert_true(string_set_add(&set, "test_string"));
 
2054
  string_set_clear(&set);
 
2055
  g_assert_false(string_set_contains(set, "test_string"));
 
2056
}
 
2057
 
 
2058
static
 
2059
void test_string_set_swap_one_with_empty(__attribute__((unused))
 
2060
                                         test_fixture *fixture,
 
2061
                                         __attribute__((unused))
 
2062
                                         gconstpointer user_data){
 
2063
  __attribute__((cleanup(string_set_clear)))
 
2064
    string_set set1 = {};
 
2065
  __attribute__((cleanup(string_set_clear)))
 
2066
    string_set set2 = {};
 
2067
  g_assert_true(string_set_add(&set1, "test_string1"));
 
2068
  string_set_swap(&set1, &set2);
 
2069
  g_assert_false(string_set_contains(set1, "test_string1"));
 
2070
  g_assert_true(string_set_contains(set2, "test_string1"));
 
2071
}
 
2072
 
 
2073
static
 
2074
void test_string_set_swap_empty_with_one(__attribute__((unused))
 
2075
                                         test_fixture *fixture,
 
2076
                                         __attribute__((unused))
 
2077
                                         gconstpointer user_data){
 
2078
  __attribute__((cleanup(string_set_clear)))
 
2079
    string_set set1 = {};
 
2080
  __attribute__((cleanup(string_set_clear)))
 
2081
    string_set set2 = {};
 
2082
  g_assert_true(string_set_add(&set2, "test_string2"));
 
2083
  string_set_swap(&set1, &set2);
 
2084
  g_assert_true(string_set_contains(set1, "test_string2"));
 
2085
  g_assert_false(string_set_contains(set2, "test_string2"));
 
2086
}
 
2087
 
 
2088
static void test_string_set_swap_one_with_one(__attribute__((unused))
 
2089
                                              test_fixture *fixture,
 
2090
                                              __attribute__((unused))
 
2091
                                              gconstpointer
 
2092
                                              user_data){
 
2093
  __attribute__((cleanup(string_set_clear)))
 
2094
    string_set set1 = {};
 
2095
  __attribute__((cleanup(string_set_clear)))
 
2096
    string_set set2 = {};
 
2097
  g_assert_true(string_set_add(&set1, "test_string1"));
 
2098
  g_assert_true(string_set_add(&set2, "test_string2"));
 
2099
  string_set_swap(&set1, &set2);
 
2100
  g_assert_false(string_set_contains(set1, "test_string1"));
 
2101
  g_assert_true(string_set_contains(set1, "test_string2"));
 
2102
  g_assert_false(string_set_contains(set2, "test_string2"));
 
2103
  g_assert_true(string_set_contains(set2, "test_string1"));
 
2104
}
 
2105
 
 
2106
static bool fd_has_cloexec_and_nonblock(const int);
 
2107
 
 
2108
static bool epoll_set_contains(int, int, uint32_t);
 
2109
 
 
2110
static void test_start_mandos_client(test_fixture *fixture,
 
2111
                                     __attribute__((unused))
 
2112
                                     gconstpointer user_data){
 
2113
 
 
2114
  bool mandos_client_exited = false;
 
2115
  bool quit_now = false;
 
2116
  __attribute__((cleanup(cleanup_close)))
 
2117
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2118
  g_assert_cmpint(epoll_fd, >=, 0);
 
2119
  __attribute__((cleanup(cleanup_queue)))
 
2120
    task_queue *queue = create_queue();
 
2121
  g_assert_nonnull(queue);
 
2122
  buffer password = {};
 
2123
  bool password_is_read = false;
 
2124
  const char helper_directory[] = "/nonexistent";
 
2125
  const char *const argv[] = { "/bin/true", NULL };
 
2126
 
 
2127
  g_assert_true(start_mandos_client(queue, epoll_fd,
 
2128
                                    &mandos_client_exited, &quit_now,
 
2129
                                    &password, &password_is_read,
 
2130
                                    &fixture->orig_sigaction,
 
2131
                                    fixture->orig_sigmask,
 
2132
                                    helper_directory, 0, 0, argv));
 
2133
 
 
2134
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
 
2135
 
 
2136
  const task_context *const added_wait_task
 
2137
    = find_matching_task(queue, (task_context){
 
2138
        .func=wait_for_mandos_client_exit,
 
2139
        .mandos_client_exited=&mandos_client_exited,
 
2140
        .quit_now=&quit_now,
 
2141
      });
 
2142
  g_assert_nonnull(added_wait_task);
 
2143
  g_assert_cmpint(added_wait_task->pid, >, 0);
 
2144
  g_assert_cmpint(kill(added_wait_task->pid, SIGKILL), ==, 0);
 
2145
  waitpid(added_wait_task->pid, NULL, 0);
 
2146
 
 
2147
  const task_context *const added_read_task
 
2148
    = find_matching_task(queue, (task_context){
 
2149
        .func=read_mandos_client_output,
 
2150
        .epoll_fd=epoll_fd,
 
2151
        .password=&password,
 
2152
        .password_is_read=&password_is_read,
 
2153
        .quit_now=&quit_now,
 
2154
      });
 
2155
  g_assert_nonnull(added_read_task);
 
2156
  g_assert_cmpint(added_read_task->fd, >, 2);
 
2157
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
2158
  g_assert_true(epoll_set_contains(epoll_fd, added_read_task->fd,
 
2159
                                   EPOLLIN | EPOLLRDHUP));
 
2160
}
 
2161
 
 
2162
static bool fd_has_cloexec_and_nonblock(const int fd){
 
2163
  const int socket_fd_flags = fcntl(fd, F_GETFD, 0);
 
2164
  const int socket_file_flags = fcntl(fd, F_GETFL, 0);
 
2165
  return ((socket_fd_flags >= 0)
 
2166
          and (socket_fd_flags & FD_CLOEXEC)
 
2167
          and (socket_file_flags >= 0)
 
2168
          and (socket_file_flags & O_NONBLOCK));
 
2169
}
 
2170
 
 
2171
__attribute__((const))
 
2172
bool is_privileged(void){
 
2173
  uid_t user = getuid() + 1;
 
2174
  if(user == 0){                /* Overflow check */
 
2175
    user++;
 
2176
  }
 
2177
  gid_t group = getuid() + 1;
 
2178
  if(group == 0){               /* Overflow check */
 
2179
    group++;
 
2180
  }
 
2181
  const pid_t pid = fork();
 
2182
  if(pid == 0){                 /* Child */
 
2183
    if(setresgid((uid_t)-1, group, group) == -1){
 
2184
      if(errno != EPERM){
 
2185
        error(EXIT_FAILURE, errno, "Failed to setresgid(-1, %" PRIuMAX
 
2186
              ", %" PRIuMAX")", (uintmax_t)group, (uintmax_t)group);
 
2187
      }
 
2188
      exit(EXIT_FAILURE);
 
2189
    }
 
2190
    if(setresuid((uid_t)-1, user, user) == -1){
 
2191
      if(errno != EPERM){
 
2192
        error(EXIT_FAILURE, errno, "Failed to setresuid(-1, %" PRIuMAX
 
2193
              ", %" PRIuMAX")", (uintmax_t)user, (uintmax_t)user);
 
2194
      }
 
2195
      exit(EXIT_FAILURE);
 
2196
    }
 
2197
    exit(EXIT_SUCCESS);
 
2198
  }
 
2199
  if(pid == -1){
 
2200
    error(EXIT_FAILURE, errno, "Failed to fork()");
 
2201
  }
 
2202
 
 
2203
  int status;
 
2204
  waitpid(pid, &status, 0);
 
2205
  if(WIFEXITED(status) and (WEXITSTATUS(status) == EXIT_SUCCESS)){
 
2206
    return true;
 
2207
  }
 
2208
  return false;
 
2209
}
 
2210
 
 
2211
static bool epoll_set_contains(int epoll_fd, int fd, uint32_t events){
 
2212
  /* Only scan for events in this eventmask */
 
2213
  const uint32_t eventmask = EPOLLIN | EPOLLOUT | EPOLLRDHUP;
 
2214
  __attribute__((cleanup(cleanup_string)))
 
2215
    char *fdinfo_name = NULL;
 
2216
  int ret = asprintf(&fdinfo_name, "/proc/self/fdinfo/%d", epoll_fd);
 
2217
  g_assert_cmpint(ret, >, 0);
 
2218
  g_assert_nonnull(fdinfo_name);
 
2219
 
 
2220
  FILE *fdinfo = fopen(fdinfo_name, "r");
 
2221
  g_assert_nonnull(fdinfo);
 
2222
  uint32_t reported_events;
 
2223
  buffer line = {};
 
2224
  int found_fd = -1;
 
2225
 
 
2226
  do {
 
2227
    if(getline(&line.data, &line.allocated, fdinfo) < 0){
 
2228
      break;
 
2229
    }
 
2230
    /* See proc(5) for format of /proc/PID/fdinfo/FD for epoll fd's */
 
2231
    if(sscanf(line.data, "tfd: %d events: %" SCNx32 " ",
 
2232
              &found_fd, &reported_events) == 2){
 
2233
      if(found_fd == fd){
 
2234
        break;
 
2235
      }
 
2236
    }
 
2237
  } while(not feof(fdinfo) and not ferror(fdinfo));
 
2238
  g_assert_cmpint(fclose(fdinfo), ==, 0);
 
2239
  free(line.data);
 
2240
  if(found_fd != fd){
 
2241
    return false;
 
2242
  }
 
2243
 
 
2244
  if(events == 0){
 
2245
    /* Don't check events if none are given */
 
2246
    return true;
 
2247
  }
 
2248
  return (reported_events & eventmask) == (events & eventmask);
 
2249
}
 
2250
 
 
2251
static void test_start_mandos_client_execv(test_fixture *fixture,
 
2252
                                           __attribute__((unused))
 
2253
                                           gconstpointer user_data){
 
2254
  bool mandos_client_exited = false;
 
2255
  bool quit_now = false;
 
2256
  __attribute__((cleanup(cleanup_close)))
 
2257
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2258
  g_assert_cmpint(epoll_fd, >=, 0);
 
2259
  __attribute__((cleanup(cleanup_queue)))
 
2260
    task_queue *queue = create_queue();
 
2261
  g_assert_nonnull(queue);
 
2262
  __attribute__((cleanup(cleanup_buffer)))
 
2263
    buffer password = {};
 
2264
  const char helper_directory[] = "/nonexistent";
 
2265
  /* Can't execv("/", ...), so this should fail */
 
2266
  const char *const argv[] = { "/", NULL };
 
2267
 
 
2268
  {
 
2269
    __attribute__((cleanup(cleanup_close)))
 
2270
      const int devnull_fd = open("/dev/null",
 
2271
                                  O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
2272
    g_assert_cmpint(devnull_fd, >=, 0);
 
2273
    __attribute__((cleanup(cleanup_close)))
 
2274
      const int real_stderr_fd = dup(STDERR_FILENO);
 
2275
    g_assert_cmpint(real_stderr_fd, >=, 0);
 
2276
    dup2(devnull_fd, STDERR_FILENO);
 
2277
 
 
2278
    const bool success = start_mandos_client(queue, epoll_fd,
 
2279
                                             &mandos_client_exited,
 
2280
                                             &quit_now,
 
2281
                                             &password,
 
2282
                                             (bool[]){false},
 
2283
                                             &fixture->orig_sigaction,
 
2284
                                             fixture->orig_sigmask,
 
2285
                                             helper_directory, 0, 0,
 
2286
                                             argv);
 
2287
    dup2(real_stderr_fd, STDERR_FILENO);
 
2288
    g_assert_true(success);
 
2289
  }
 
2290
  g_assert_cmpuint((unsigned int)queue->length, ==, 2);
 
2291
 
 
2292
  struct timespec starttime, currtime;
 
2293
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2294
  do {
 
2295
    queue->next_run = 0;
 
2296
    string_set cancelled_filenames = {};
 
2297
 
 
2298
    {
 
2299
      __attribute__((cleanup(cleanup_close)))
 
2300
        const int devnull_fd = open("/dev/null",
 
2301
                                    O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
2302
      g_assert_cmpint(devnull_fd, >=, 0);
 
2303
      __attribute__((cleanup(cleanup_close)))
 
2304
        const int real_stderr_fd = dup(STDERR_FILENO);
 
2305
      g_assert_cmpint(real_stderr_fd, >=, 0);
 
2306
      g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2307
      dup2(devnull_fd, STDERR_FILENO);
 
2308
      const bool success = run_queue(&queue, &cancelled_filenames,
 
2309
                                     &quit_now);
 
2310
      dup2(real_stderr_fd, STDERR_FILENO);
 
2311
      if(not success){
 
2312
        break;
 
2313
      }
 
2314
    }
 
2315
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2316
  } while(((queue->length) > 0)
 
2317
          and (not quit_now)
 
2318
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2319
 
 
2320
  g_assert_true(quit_now);
 
2321
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2322
  g_assert_true(mandos_client_exited);
 
2323
}
 
2324
 
 
2325
static void test_start_mandos_client_suid_euid(test_fixture *fixture,
 
2326
                                               __attribute__((unused))
 
2327
                                               gconstpointer
 
2328
                                               user_data){
 
2329
  if(not is_privileged()){
 
2330
    g_test_skip("Not privileged");
 
2331
    return;
 
2332
  }
 
2333
 
 
2334
  bool mandos_client_exited = false;
 
2335
  bool quit_now = false;
 
2336
  __attribute__((cleanup(cleanup_close)))
 
2337
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2338
  g_assert_cmpint(epoll_fd, >=, 0);
 
2339
  __attribute__((cleanup(cleanup_queue)))
 
2340
    task_queue *queue = create_queue();
 
2341
  g_assert_nonnull(queue);
 
2342
  __attribute__((cleanup(cleanup_buffer)))
 
2343
    buffer password = {};
 
2344
  bool password_is_read = false;
 
2345
  const char helper_directory[] = "/nonexistent";
 
2346
  const char *const argv[] = { "/usr/bin/id", "--user", NULL };
 
2347
  uid_t user = 1000;
 
2348
  gid_t group = 1001;
 
2349
 
 
2350
  const bool success = start_mandos_client(queue, epoll_fd,
 
2351
                                           &mandos_client_exited,
 
2352
                                           &quit_now, &password,
 
2353
                                           &password_is_read,
 
2354
                                           &fixture->orig_sigaction,
 
2355
                                           fixture->orig_sigmask,
 
2356
                                           helper_directory, user,
 
2357
                                           group, argv);
 
2358
  g_assert_true(success);
 
2359
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2360
 
 
2361
  struct timespec starttime, currtime;
 
2362
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2363
  do {
 
2364
    queue->next_run = 0;
 
2365
    string_set cancelled_filenames = {};
 
2366
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2367
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2368
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2369
  } while(((queue->length) > 0)
 
2370
          and (not quit_now)
 
2371
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2372
 
 
2373
  g_assert_false(quit_now);
 
2374
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2375
  g_assert_true(mandos_client_exited);
 
2376
 
 
2377
  g_assert_true(password_is_read);
 
2378
  g_assert_nonnull(password.data);
 
2379
 
 
2380
  uintmax_t id;
 
2381
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
 
2382
                  ==, 1);
 
2383
  g_assert_true((uid_t)id == id);
 
2384
 
 
2385
  g_assert_cmpuint((unsigned int)id, ==, 0);
 
2386
}
 
2387
 
 
2388
static void test_start_mandos_client_suid_egid(test_fixture *fixture,
 
2389
                                               __attribute__((unused))
 
2390
                                               gconstpointer
 
2391
                                               user_data){
 
2392
  if(not is_privileged()){
 
2393
    g_test_skip("Not privileged");
 
2394
    return;
 
2395
  }
 
2396
 
 
2397
  bool mandos_client_exited = false;
 
2398
  bool quit_now = false;
 
2399
  __attribute__((cleanup(cleanup_close)))
 
2400
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2401
  g_assert_cmpint(epoll_fd, >=, 0);
 
2402
  __attribute__((cleanup(cleanup_queue)))
 
2403
    task_queue *queue = create_queue();
 
2404
  g_assert_nonnull(queue);
 
2405
  __attribute__((cleanup(cleanup_buffer)))
 
2406
    buffer password = {};
 
2407
  bool password_is_read = false;
 
2408
  const char helper_directory[] = "/nonexistent";
 
2409
  const char *const argv[] = { "/usr/bin/id", "--group", NULL };
 
2410
  uid_t user = 1000;
 
2411
  gid_t group = 1001;
 
2412
 
 
2413
  const bool success = start_mandos_client(queue, epoll_fd,
 
2414
                                           &mandos_client_exited,
 
2415
                                           &quit_now, &password,
 
2416
                                           &password_is_read,
 
2417
                                           &fixture->orig_sigaction,
 
2418
                                           fixture->orig_sigmask,
 
2419
                                           helper_directory, user,
 
2420
                                           group, argv);
 
2421
  g_assert_true(success);
 
2422
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2423
 
 
2424
  struct timespec starttime, currtime;
 
2425
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2426
  do {
 
2427
    queue->next_run = 0;
 
2428
    string_set cancelled_filenames = {};
 
2429
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2430
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2431
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2432
  } while(((queue->length) > 0)
 
2433
          and (not quit_now)
 
2434
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2435
 
 
2436
  g_assert_false(quit_now);
 
2437
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2438
  g_assert_true(mandos_client_exited);
 
2439
 
 
2440
  g_assert_true(password_is_read);
 
2441
  g_assert_nonnull(password.data);
 
2442
 
 
2443
  uintmax_t id;
 
2444
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
 
2445
                  ==, 1);
 
2446
  g_assert_true((gid_t)id == id);
 
2447
 
 
2448
  g_assert_cmpuint((unsigned int)id, ==, 0);
 
2449
}
 
2450
 
 
2451
static void test_start_mandos_client_suid_ruid(test_fixture *fixture,
 
2452
                                               __attribute__((unused))
 
2453
                                               gconstpointer
 
2454
                                               user_data){
 
2455
  if(not is_privileged()){
 
2456
    g_test_skip("Not privileged");
 
2457
    return;
 
2458
  }
 
2459
 
 
2460
  bool mandos_client_exited = false;
 
2461
  bool quit_now = false;
 
2462
  __attribute__((cleanup(cleanup_close)))
 
2463
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2464
  g_assert_cmpint(epoll_fd, >=, 0);
 
2465
  __attribute__((cleanup(cleanup_queue)))
 
2466
    task_queue *queue = create_queue();
 
2467
  g_assert_nonnull(queue);
 
2468
  __attribute__((cleanup(cleanup_buffer)))
 
2469
    buffer password = {};
 
2470
  bool password_is_read = false;
 
2471
  const char helper_directory[] = "/nonexistent";
 
2472
  const char *const argv[] = { "/usr/bin/id", "--user", "--real",
 
2473
    NULL };
 
2474
  uid_t user = 1000;
 
2475
  gid_t group = 1001;
 
2476
 
 
2477
  const bool success = start_mandos_client(queue, epoll_fd,
 
2478
                                           &mandos_client_exited,
 
2479
                                           &quit_now, &password,
 
2480
                                           &password_is_read,
 
2481
                                           &fixture->orig_sigaction,
 
2482
                                           fixture->orig_sigmask,
 
2483
                                           helper_directory, user,
 
2484
                                           group, argv);
 
2485
  g_assert_true(success);
 
2486
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2487
 
 
2488
  struct timespec starttime, currtime;
 
2489
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2490
  do {
 
2491
    queue->next_run = 0;
 
2492
    string_set cancelled_filenames = {};
 
2493
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2494
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2495
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2496
  } while(((queue->length) > 0)
 
2497
          and (not quit_now)
 
2498
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2499
 
 
2500
  g_assert_false(quit_now);
 
2501
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2502
  g_assert_true(mandos_client_exited);
 
2503
 
 
2504
  g_assert_true(password_is_read);
 
2505
  g_assert_nonnull(password.data);
 
2506
 
 
2507
  uintmax_t id;
 
2508
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
 
2509
                  ==, 1);
 
2510
  g_assert_true((uid_t)id == id);
 
2511
 
 
2512
  g_assert_cmpuint((unsigned int)id, ==, user);
 
2513
}
 
2514
 
 
2515
static void test_start_mandos_client_suid_rgid(test_fixture *fixture,
 
2516
                                               __attribute__((unused))
 
2517
                                               gconstpointer
 
2518
                                               user_data){
 
2519
  if(not is_privileged()){
 
2520
    g_test_skip("Not privileged");
 
2521
    return;
 
2522
  }
 
2523
 
 
2524
  bool mandos_client_exited = false;
 
2525
  bool quit_now = false;
 
2526
  __attribute__((cleanup(cleanup_close)))
 
2527
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2528
  g_assert_cmpint(epoll_fd, >=, 0);
 
2529
  __attribute__((cleanup(cleanup_queue)))
 
2530
    task_queue *queue = create_queue();
 
2531
  g_assert_nonnull(queue);
 
2532
  __attribute__((cleanup(cleanup_buffer)))
 
2533
    buffer password = {};
 
2534
  bool password_is_read = false;
 
2535
  const char helper_directory[] = "/nonexistent";
 
2536
  const char *const argv[] = { "/usr/bin/id", "--group", "--real",
 
2537
    NULL };
 
2538
  uid_t user = 1000;
 
2539
  gid_t group = 1001;
 
2540
 
 
2541
  const bool success = start_mandos_client(queue, epoll_fd,
 
2542
                                           &mandos_client_exited,
 
2543
                                           &quit_now, &password,
 
2544
                                           &password_is_read,
 
2545
                                           &fixture->orig_sigaction,
 
2546
                                           fixture->orig_sigmask,
 
2547
                                           helper_directory, user,
 
2548
                                           group, argv);
 
2549
  g_assert_true(success);
 
2550
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2551
 
 
2552
  struct timespec starttime, currtime;
 
2553
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2554
  do {
 
2555
    queue->next_run = 0;
 
2556
    string_set cancelled_filenames = {};
 
2557
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2558
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2559
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2560
  } while(((queue->length) > 0)
 
2561
          and (not quit_now)
 
2562
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2563
 
 
2564
  g_assert_false(quit_now);
 
2565
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2566
  g_assert_true(mandos_client_exited);
 
2567
 
 
2568
  g_assert_true(password_is_read);
 
2569
  g_assert_nonnull(password.data);
 
2570
 
 
2571
  uintmax_t id;
 
2572
  g_assert_cmpint(sscanf(password.data, "%" SCNuMAX "\n", &id),
 
2573
                  ==, 1);
 
2574
  g_assert_true((gid_t)id == id);
 
2575
 
 
2576
  g_assert_cmpuint((unsigned int)id, ==, group);
 
2577
}
 
2578
 
 
2579
static void test_start_mandos_client_read(test_fixture *fixture,
 
2580
                                          __attribute__((unused))
 
2581
                                          gconstpointer user_data){
 
2582
  bool mandos_client_exited = false;
 
2583
  bool quit_now = false;
 
2584
  __attribute__((cleanup(cleanup_close)))
 
2585
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2586
  g_assert_cmpint(epoll_fd, >=, 0);
 
2587
  __attribute__((cleanup(cleanup_queue)))
 
2588
    task_queue *queue = create_queue();
 
2589
  g_assert_nonnull(queue);
 
2590
  __attribute__((cleanup(cleanup_buffer)))
 
2591
    buffer password = {};
 
2592
  bool password_is_read = false;
 
2593
  const char dummy_test_password[] = "dummy test password";
 
2594
  const char helper_directory[] = "/nonexistent";
 
2595
  const char *const argv[] = { "/bin/echo", "-n", dummy_test_password,
 
2596
    NULL };
 
2597
 
 
2598
  const bool success = start_mandos_client(queue, epoll_fd,
 
2599
                                           &mandos_client_exited,
 
2600
                                           &quit_now, &password,
 
2601
                                           &password_is_read,
 
2602
                                           &fixture->orig_sigaction,
 
2603
                                           fixture->orig_sigmask,
 
2604
                                           helper_directory, 0, 0,
 
2605
                                           argv);
 
2606
  g_assert_true(success);
 
2607
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2608
 
 
2609
  struct timespec starttime, currtime;
 
2610
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2611
  do {
 
2612
    queue->next_run = 0;
 
2613
    string_set cancelled_filenames = {};
 
2614
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2615
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2616
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2617
  } while(((queue->length) > 0)
 
2618
          and (not quit_now)
 
2619
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2620
 
 
2621
  g_assert_false(quit_now);
 
2622
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2623
  g_assert_true(mandos_client_exited);
 
2624
 
 
2625
  g_assert_true(password_is_read);
 
2626
  g_assert_cmpint((int)password.length, ==,
 
2627
                  sizeof(dummy_test_password)-1);
 
2628
  g_assert_nonnull(password.data);
 
2629
  g_assert_cmpint(memcmp(dummy_test_password, password.data,
 
2630
                         sizeof(dummy_test_password)-1), ==, 0);
 
2631
}
 
2632
 
 
2633
static
 
2634
void test_start_mandos_client_helper_directory(test_fixture *fixture,
 
2635
                                               __attribute__((unused))
 
2636
                                               gconstpointer
 
2637
                                               user_data){
 
2638
  bool mandos_client_exited = false;
 
2639
  bool quit_now = false;
 
2640
  __attribute__((cleanup(cleanup_close)))
 
2641
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2642
  g_assert_cmpint(epoll_fd, >=, 0);
 
2643
  __attribute__((cleanup(cleanup_queue)))
 
2644
    task_queue *queue = create_queue();
 
2645
  g_assert_nonnull(queue);
 
2646
  __attribute__((cleanup(cleanup_buffer)))
 
2647
    buffer password = {};
 
2648
  bool password_is_read = false;
 
2649
  const char helper_directory[] = "/nonexistent";
 
2650
  const char *const argv[] = { "/bin/sh", "-c",
 
2651
    "echo -n ${MANDOSPLUGINHELPERDIR}", NULL };
 
2652
 
 
2653
  const bool success = start_mandos_client(queue, epoll_fd,
 
2654
                                           &mandos_client_exited,
 
2655
                                           &quit_now, &password,
 
2656
                                           &password_is_read,
 
2657
                                           &fixture->orig_sigaction,
 
2658
                                           fixture->orig_sigmask,
 
2659
                                           helper_directory, 0, 0,
 
2660
                                           argv);
 
2661
  g_assert_true(success);
 
2662
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
2663
 
 
2664
  struct timespec starttime, currtime;
 
2665
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2666
  do {
 
2667
    queue->next_run = 0;
 
2668
    string_set cancelled_filenames = {};
 
2669
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2670
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2671
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2672
  } while(((queue->length) > 0)
 
2673
          and (not quit_now)
 
2674
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2675
 
 
2676
  g_assert_false(quit_now);
 
2677
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2678
  g_assert_true(mandos_client_exited);
 
2679
 
 
2680
  g_assert_true(password_is_read);
 
2681
  g_assert_cmpint((int)password.length, ==,
 
2682
                  sizeof(helper_directory)-1);
 
2683
  g_assert_nonnull(password.data);
 
2684
  g_assert_cmpint(memcmp(helper_directory, password.data,
 
2685
                         sizeof(helper_directory)-1), ==, 0);
 
2686
}
 
2687
 
 
2688
__attribute__((nonnull, warn_unused_result))
 
2689
static bool proc_status_sigblk_to_sigset(const char *const,
 
2690
                                         sigset_t *const);
 
2691
 
 
2692
static void test_start_mandos_client_sigmask(test_fixture *fixture,
 
2693
                                             __attribute__((unused))
 
2694
                                             gconstpointer user_data){
 
2695
  bool mandos_client_exited = false;
 
2696
  bool quit_now = false;
 
2697
  __attribute__((cleanup(cleanup_close)))
 
2698
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2699
  g_assert_cmpint(epoll_fd, >=, 0);
 
2700
  __attribute__((cleanup(cleanup_queue)))
 
2701
    task_queue *queue = create_queue();
 
2702
  g_assert_nonnull(queue);
 
2703
  __attribute__((cleanup(cleanup_buffer)))
 
2704
    buffer password = {};
 
2705
  bool password_is_read = false;
 
2706
  const char helper_directory[] = "/nonexistent";
 
2707
  /* see proc(5) for format of /proc/self/status */
 
2708
  const char *const argv[] = { "/usr/bin/awk",
 
2709
    "$1==\"SigBlk:\"{ print $2 }", "/proc/self/status", NULL };
 
2710
 
 
2711
  g_assert_true(start_mandos_client(queue, epoll_fd,
 
2712
                                    &mandos_client_exited, &quit_now,
 
2713
                                    &password, &password_is_read,
 
2714
                                    &fixture->orig_sigaction,
 
2715
                                    fixture->orig_sigmask,
 
2716
                                    helper_directory, 0, 0, argv));
 
2717
 
 
2718
  struct timespec starttime, currtime;
 
2719
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2720
  do {
 
2721
    queue->next_run = 0;
 
2722
    string_set cancelled_filenames = {};
 
2723
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2724
    g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
2725
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2726
  } while((not (mandos_client_exited and password_is_read))
 
2727
          and (not quit_now)
 
2728
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2729
  g_assert_true(mandos_client_exited);
 
2730
  g_assert_true(password_is_read);
 
2731
 
 
2732
  sigset_t parsed_sigmask;
 
2733
  g_assert_true(proc_status_sigblk_to_sigset(password.data,
 
2734
                                             &parsed_sigmask));
 
2735
 
 
2736
  for(int signum = 1; signum < NSIG; signum++){
 
2737
    const bool has_signal = sigismember(&parsed_sigmask, signum);
 
2738
    if(sigismember(&fixture->orig_sigmask, signum)){
 
2739
      g_assert_true(has_signal);
 
2740
    } else {
 
2741
      g_assert_false(has_signal);
 
2742
    }
 
2743
  }
 
2744
}
 
2745
 
 
2746
__attribute__((nonnull, warn_unused_result))
 
2747
static bool proc_status_sigblk_to_sigset(const char *const sigblk,
 
2748
                                         sigset_t *const sigmask){
 
2749
  /* parse /proc/PID/status SigBlk value and convert to a sigset_t */
 
2750
  uintmax_t scanned_sigmask;
 
2751
  if(sscanf(sigblk, "%" SCNxMAX " ", &scanned_sigmask) != 1){
 
2752
    return false;
 
2753
  }
 
2754
  if(sigemptyset(sigmask) != 0){
 
2755
    return false;
 
2756
  }
 
2757
  for(int signum = 1; signum < NSIG; signum++){
 
2758
    if(scanned_sigmask & ((uintmax_t)1 << (signum-1))){
 
2759
      if(sigaddset(sigmask, signum) != 0){
 
2760
        return false;
 
2761
      }
 
2762
    }
 
2763
  }
 
2764
  return true;
 
2765
}
 
2766
 
 
2767
static void run_task_with_stderr_to_dev_null(const task_context task,
 
2768
                                             task_queue *const queue);
 
2769
 
 
2770
static
 
2771
void test_wait_for_mandos_client_exit_badpid(__attribute__((unused))
 
2772
                                             test_fixture *fixture,
 
2773
                                             __attribute__((unused))
 
2774
                                             gconstpointer user_data){
 
2775
 
 
2776
  bool mandos_client_exited = false;
 
2777
  bool quit_now = false;
 
2778
 
 
2779
  __attribute__((cleanup(cleanup_queue)))
 
2780
    task_queue *queue = create_queue();
 
2781
  g_assert_nonnull(queue);
 
2782
  const task_context task = {
 
2783
    .func=wait_for_mandos_client_exit,
 
2784
    .pid=1,
 
2785
    .mandos_client_exited=&mandos_client_exited,
 
2786
    .quit_now=&quit_now,
 
2787
  };
 
2788
  run_task_with_stderr_to_dev_null(task, queue);
 
2789
 
 
2790
  g_assert_false(mandos_client_exited);
 
2791
  g_assert_true(quit_now);
 
2792
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2793
}
 
2794
 
 
2795
static void run_task_with_stderr_to_dev_null(const task_context task,
 
2796
                                             task_queue *const queue){
 
2797
  FILE *real_stderr = stderr;
 
2798
  FILE *devnull = fopen("/dev/null", "we");
 
2799
  g_assert_nonnull(devnull);
 
2800
 
 
2801
  stderr = devnull;
 
2802
  task.func(task, queue);
 
2803
  stderr = real_stderr;
 
2804
 
 
2805
  g_assert_cmpint(fclose(devnull), ==, 0);
 
2806
}
 
2807
 
 
2808
static
 
2809
void test_wait_for_mandos_client_exit_noexit(test_fixture *fixture,
 
2810
                                             __attribute__((unused))
 
2811
                                             gconstpointer user_data){
 
2812
  bool mandos_client_exited = false;
 
2813
  bool quit_now = false;
 
2814
 
 
2815
  pid_t create_eternal_process(void){
 
2816
    const pid_t pid = fork();
 
2817
    if(pid == 0){               /* Child */
 
2818
      if(not restore_signal_handler(&fixture->orig_sigaction)){
 
2819
        _exit(EXIT_FAILURE);
 
2820
      }
 
2821
      if(not restore_sigmask(&fixture->orig_sigmask)){
 
2822
        _exit(EXIT_FAILURE);
 
2823
      }
 
2824
      while(true){
 
2825
        pause();
 
2826
      }
 
2827
    }
 
2828
    return pid;
 
2829
  }
 
2830
  pid_t pid = create_eternal_process();
 
2831
  g_assert_true(pid != -1);
 
2832
 
 
2833
  __attribute__((cleanup(cleanup_queue)))
 
2834
    task_queue *queue = create_queue();
 
2835
  g_assert_nonnull(queue);
 
2836
  const task_context task = {
 
2837
    .func=wait_for_mandos_client_exit,
 
2838
    .pid=pid,
 
2839
    .mandos_client_exited=&mandos_client_exited,
 
2840
    .quit_now=&quit_now,
 
2841
  };
 
2842
  task.func(task, queue);
 
2843
 
 
2844
  g_assert_false(mandos_client_exited);
 
2845
  g_assert_false(quit_now);
 
2846
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
2847
 
 
2848
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
2849
        .func=wait_for_mandos_client_exit,
 
2850
        .pid=task.pid,
 
2851
        .mandos_client_exited=&mandos_client_exited,
 
2852
        .quit_now=&quit_now,
 
2853
      }));
 
2854
}
 
2855
 
 
2856
static
 
2857
void test_wait_for_mandos_client_exit_success(test_fixture *fixture,
 
2858
                                              __attribute__((unused))
 
2859
                                              gconstpointer
 
2860
                                              user_data){
 
2861
  bool mandos_client_exited = false;
 
2862
  bool quit_now = false;
 
2863
 
 
2864
  pid_t create_successful_process(void){
 
2865
    const pid_t pid = fork();
 
2866
    if(pid == 0){               /* Child */
 
2867
      if(not restore_signal_handler(&fixture->orig_sigaction)){
 
2868
        _exit(EXIT_FAILURE);
 
2869
      }
 
2870
      if(not restore_sigmask(&fixture->orig_sigmask)){
 
2871
        _exit(EXIT_FAILURE);
 
2872
      }
 
2873
      exit(EXIT_SUCCESS);
 
2874
    }
 
2875
    return pid;
 
2876
  }
 
2877
  const pid_t pid = create_successful_process();
 
2878
  g_assert_true(pid != -1);
 
2879
 
 
2880
  __attribute__((cleanup(cleanup_queue)))
 
2881
    task_queue *queue = create_queue();
 
2882
  g_assert_nonnull(queue);
 
2883
  const task_context initial_task = {
 
2884
    .func=wait_for_mandos_client_exit,
 
2885
    .pid=pid,
 
2886
    .mandos_client_exited=&mandos_client_exited,
 
2887
    .quit_now=&quit_now,
 
2888
  };
 
2889
  g_assert_true(add_to_queue(queue, initial_task));
 
2890
 
 
2891
  struct timespec starttime, currtime;
 
2892
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2893
  __attribute__((cleanup(cleanup_close)))
 
2894
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2895
  do {
 
2896
    queue->next_run = 0;
 
2897
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2898
    g_assert_true(run_queue(&queue, (string_set[]){{}}, &quit_now));
 
2899
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2900
  } while((not mandos_client_exited)
 
2901
          and (not quit_now)
 
2902
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2903
 
 
2904
  g_assert_true(mandos_client_exited);
 
2905
  g_assert_false(quit_now);
 
2906
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2907
}
 
2908
 
 
2909
static
 
2910
void test_wait_for_mandos_client_exit_failure(test_fixture *fixture,
 
2911
                                              __attribute__((unused))
 
2912
                                              gconstpointer
 
2913
                                              user_data){
 
2914
  bool mandos_client_exited = false;
 
2915
  bool quit_now = false;
 
2916
 
 
2917
  pid_t create_failing_process(void){
 
2918
    const pid_t pid = fork();
 
2919
    if(pid == 0){               /* Child */
 
2920
      if(not restore_signal_handler(&fixture->orig_sigaction)){
 
2921
        _exit(EXIT_FAILURE);
 
2922
      }
 
2923
      if(not restore_sigmask(&fixture->orig_sigmask)){
 
2924
        _exit(EXIT_FAILURE);
 
2925
      }
 
2926
      exit(EXIT_FAILURE);
 
2927
    }
 
2928
    return pid;
 
2929
  }
 
2930
  const pid_t pid = create_failing_process();
 
2931
  g_assert_true(pid != -1);
 
2932
 
 
2933
  __attribute__((cleanup(string_set_clear)))
 
2934
    string_set cancelled_filenames = {};
 
2935
  __attribute__((cleanup(cleanup_close)))
 
2936
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
2937
  g_assert_cmpint(epoll_fd, >=, 0);
 
2938
  __attribute__((cleanup(cleanup_queue)))
 
2939
    task_queue *queue = create_queue();
 
2940
  g_assert_nonnull(queue);
 
2941
  g_assert_true(add_to_queue(queue, (task_context){
 
2942
        .func=wait_for_mandos_client_exit,
 
2943
        .pid=pid,
 
2944
        .mandos_client_exited=&mandos_client_exited,
 
2945
        .quit_now=&quit_now,
 
2946
      }));
 
2947
 
 
2948
  g_assert_true(sigismember(&fixture->orig_sigmask, SIGCHLD) == 0);
 
2949
 
 
2950
  __attribute__((cleanup(cleanup_close)))
 
2951
    const int devnull_fd = open("/dev/null",
 
2952
                                O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
2953
  g_assert_cmpint(devnull_fd, >=, 0);
 
2954
  __attribute__((cleanup(cleanup_close)))
 
2955
    const int real_stderr_fd = dup(STDERR_FILENO);
 
2956
  g_assert_cmpint(real_stderr_fd, >=, 0);
 
2957
 
 
2958
  struct timespec starttime, currtime;
 
2959
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
2960
  do {
 
2961
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
2962
    dup2(devnull_fd, STDERR_FILENO);
 
2963
    const bool success = run_queue(&queue, &cancelled_filenames,
 
2964
                                   &quit_now);
 
2965
    dup2(real_stderr_fd, STDERR_FILENO);
 
2966
    if(not success){
 
2967
      break;
 
2968
    }
 
2969
 
 
2970
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
2971
  } while((not mandos_client_exited)
 
2972
          and (not quit_now)
 
2973
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
2974
 
 
2975
  g_assert_true(quit_now);
 
2976
  g_assert_true(mandos_client_exited);
 
2977
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
2978
}
 
2979
 
 
2980
static
 
2981
void test_wait_for_mandos_client_exit_killed(test_fixture *fixture,
 
2982
                                             __attribute__((unused))
 
2983
                                             gconstpointer user_data){
 
2984
  bool mandos_client_exited = false;
 
2985
  bool quit_now = false;
 
2986
 
 
2987
  pid_t create_killed_process(void){
 
2988
    const pid_t pid = fork();
 
2989
    if(pid == 0){               /* Child */
 
2990
      if(not restore_signal_handler(&fixture->orig_sigaction)){
 
2991
        _exit(EXIT_FAILURE);
 
2992
      }
 
2993
      if(not restore_sigmask(&fixture->orig_sigmask)){
 
2994
        _exit(EXIT_FAILURE);
 
2995
      }
 
2996
      while(true){
 
2997
        pause();
 
2998
      }
 
2999
    }
 
3000
    kill(pid, SIGKILL);
 
3001
    return pid;
 
3002
  }
 
3003
  const pid_t pid = create_killed_process();
 
3004
  g_assert_true(pid != -1);
 
3005
 
 
3006
  __attribute__((cleanup(string_set_clear)))
 
3007
    string_set cancelled_filenames = {};
 
3008
  __attribute__((cleanup(cleanup_close)))
 
3009
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3010
  g_assert_cmpint(epoll_fd, >=, 0);
 
3011
  __attribute__((cleanup(cleanup_queue)))
 
3012
    task_queue *queue = create_queue();
 
3013
  g_assert_nonnull(queue);
 
3014
  g_assert_true(add_to_queue(queue, (task_context){
 
3015
        .func=wait_for_mandos_client_exit,
 
3016
        .pid=pid,
 
3017
        .mandos_client_exited=&mandos_client_exited,
 
3018
        .quit_now=&quit_now,
 
3019
      }));
 
3020
 
 
3021
  __attribute__((cleanup(cleanup_close)))
 
3022
    const int devnull_fd = open("/dev/null",
 
3023
                                O_WRONLY | O_CLOEXEC, O_NOCTTY);
 
3024
  g_assert_cmpint(devnull_fd, >=, 0);
 
3025
  __attribute__((cleanup(cleanup_close)))
 
3026
    const int real_stderr_fd = dup(STDERR_FILENO);
 
3027
  g_assert_cmpint(real_stderr_fd, >=, 0);
 
3028
 
 
3029
  struct timespec starttime, currtime;
 
3030
  g_assert_true(clock_gettime(CLOCK_MONOTONIC, &starttime) == 0);
 
3031
  do {
 
3032
    g_assert_true(wait_for_event(epoll_fd, queue->next_run, 0));
 
3033
    dup2(devnull_fd, STDERR_FILENO);
 
3034
    const bool success = run_queue(&queue, &cancelled_filenames,
 
3035
                                   &quit_now);
 
3036
    dup2(real_stderr_fd, STDERR_FILENO);
 
3037
    if(not success){
 
3038
      break;
 
3039
    }
 
3040
 
 
3041
    g_assert_true(clock_gettime(CLOCK_MONOTONIC, &currtime) == 0);
 
3042
  } while((not mandos_client_exited)
 
3043
          and (not quit_now)
 
3044
          and ((currtime.tv_sec - starttime.tv_sec) < 10));
 
3045
 
 
3046
  g_assert_true(mandos_client_exited);
 
3047
  g_assert_true(quit_now);
 
3048
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
3049
}
 
3050
 
 
3051
static bool epoll_set_does_not_contain(int, int);
 
3052
 
 
3053
static
 
3054
void test_read_mandos_client_output_readerror(__attribute__((unused))
 
3055
                                              test_fixture *fixture,
 
3056
                                              __attribute__((unused))
 
3057
                                              gconstpointer
 
3058
                                              user_data){
 
3059
  __attribute__((cleanup(cleanup_close)))
 
3060
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3061
  g_assert_cmpint(epoll_fd, >=, 0);
 
3062
 
 
3063
  __attribute__((cleanup(cleanup_buffer)))
 
3064
    buffer password = {};
 
3065
 
 
3066
  /* Reading /proc/self/mem from offset 0 will always give EIO */
 
3067
  const int fd = open("/proc/self/mem",
 
3068
                      O_RDONLY | O_CLOEXEC | O_NOCTTY);
 
3069
 
 
3070
  bool password_is_read = false;
 
3071
  bool quit_now = false;
 
3072
  __attribute__((cleanup(cleanup_queue)))
 
3073
    task_queue *queue = create_queue();
 
3074
  g_assert_nonnull(queue);
 
3075
 
 
3076
  task_context task = {
 
3077
    .func=read_mandos_client_output,
 
3078
    .epoll_fd=epoll_fd,
 
3079
    .fd=fd,
 
3080
    .password=&password,
 
3081
    .password_is_read=&password_is_read,
 
3082
    .quit_now=&quit_now,
 
3083
  };
 
3084
  run_task_with_stderr_to_dev_null(task, queue);
 
3085
  g_assert_false(password_is_read);
 
3086
  g_assert_cmpint((int)password.length, ==, 0);
 
3087
  g_assert_true(quit_now);
 
3088
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
3089
 
 
3090
  g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));
 
3091
 
 
3092
  g_assert_cmpint(close(fd), ==, -1);
 
3093
}
 
3094
 
 
3095
static bool epoll_set_does_not_contain(int epoll_fd, int fd){
 
3096
  return not epoll_set_contains(epoll_fd, fd, 0);
 
3097
}
 
3098
 
 
3099
static
 
3100
void test_read_mandos_client_output_nodata(__attribute__((unused))
 
3101
                                           test_fixture *fixture,
 
3102
                                           __attribute__((unused))
 
3103
                                           gconstpointer user_data){
 
3104
  __attribute__((cleanup(cleanup_close)))
 
3105
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3106
  g_assert_cmpint(epoll_fd, >=, 0);
 
3107
 
 
3108
  int pipefds[2];
 
3109
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
3110
 
 
3111
  __attribute__((cleanup(cleanup_buffer)))
 
3112
    buffer password = {};
 
3113
 
 
3114
  bool password_is_read = false;
 
3115
  bool quit_now = false;
 
3116
  __attribute__((cleanup(cleanup_queue)))
 
3117
    task_queue *queue = create_queue();
 
3118
  g_assert_nonnull(queue);
 
3119
 
 
3120
  task_context task = {
 
3121
    .func=read_mandos_client_output,
 
3122
    .epoll_fd=epoll_fd,
 
3123
    .fd=pipefds[0],
 
3124
    .password=&password,
 
3125
    .password_is_read=&password_is_read,
 
3126
    .quit_now=&quit_now,
 
3127
  };
 
3128
  task.func(task, queue);
 
3129
  g_assert_false(password_is_read);
 
3130
  g_assert_cmpint((int)password.length, ==, 0);
 
3131
  g_assert_false(quit_now);
 
3132
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
3133
 
 
3134
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
3135
        .func=read_mandos_client_output,
 
3136
        .epoll_fd=epoll_fd,
 
3137
        .fd=pipefds[0],
 
3138
        .password=&password,
 
3139
        .password_is_read=&password_is_read,
 
3140
        .quit_now=&quit_now,
 
3141
      }));
 
3142
 
 
3143
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
3144
                                   EPOLLIN | EPOLLRDHUP));
 
3145
 
 
3146
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
3147
}
 
3148
 
 
3149
static void test_read_mandos_client_output_eof(__attribute__((unused))
 
3150
                                               test_fixture *fixture,
 
3151
                                               __attribute__((unused))
 
3152
                                               gconstpointer
 
3153
                                               user_data){
 
3154
  __attribute__((cleanup(cleanup_close)))
 
3155
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3156
  g_assert_cmpint(epoll_fd, >=, 0);
 
3157
 
 
3158
  int pipefds[2];
 
3159
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
3160
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
3161
 
 
3162
  __attribute__((cleanup(cleanup_buffer)))
 
3163
    buffer password = {};
 
3164
 
 
3165
  bool password_is_read = false;
 
3166
  bool quit_now = false;
 
3167
  __attribute__((cleanup(cleanup_queue)))
 
3168
    task_queue *queue = create_queue();
 
3169
  g_assert_nonnull(queue);
 
3170
 
 
3171
  task_context task = {
 
3172
    .func=read_mandos_client_output,
 
3173
    .epoll_fd=epoll_fd,
 
3174
    .fd=pipefds[0],
 
3175
    .password=&password,
 
3176
    .password_is_read=&password_is_read,
 
3177
    .quit_now=&quit_now,
 
3178
  };
 
3179
  task.func(task, queue);
 
3180
  g_assert_true(password_is_read);
 
3181
  g_assert_cmpint((int)password.length, ==, 0);
 
3182
  g_assert_false(quit_now);
 
3183
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
3184
 
 
3185
  g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));
 
3186
 
 
3187
  g_assert_cmpint(close(pipefds[0]), ==, -1);
 
3188
}
 
3189
 
 
3190
static
 
3191
void test_read_mandos_client_output_once(__attribute__((unused))
 
3192
                                         test_fixture *fixture,
 
3193
                                         __attribute__((unused))
 
3194
                                         gconstpointer user_data){
 
3195
  __attribute__((cleanup(cleanup_close)))
 
3196
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3197
  g_assert_cmpint(epoll_fd, >=, 0);
 
3198
 
 
3199
  int pipefds[2];
 
3200
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
3201
 
 
3202
  const char dummy_test_password[] = "dummy test password";
 
3203
  /* Start with a pre-allocated buffer */
 
3204
  __attribute__((cleanup(cleanup_buffer)))
 
3205
    buffer password = {
 
3206
    .data=malloc(sizeof(dummy_test_password)),
 
3207
    .length=0,
 
3208
    .allocated=sizeof(dummy_test_password),
 
3209
  };
 
3210
  g_assert_nonnull(password.data);
 
3211
  if(mlock(password.data, password.allocated) != 0){
 
3212
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
3213
  }
 
3214
 
 
3215
  bool password_is_read = false;
 
3216
  bool quit_now = false;
 
3217
  __attribute__((cleanup(cleanup_queue)))
 
3218
    task_queue *queue = create_queue();
 
3219
  g_assert_nonnull(queue);
 
3220
 
 
3221
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
 
3222
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
 
3223
                             sizeof(dummy_test_password)),
 
3224
                  ==, (int)sizeof(dummy_test_password));
 
3225
 
 
3226
  task_context task = {
 
3227
    .func=read_mandos_client_output,
 
3228
    .epoll_fd=epoll_fd,
 
3229
    .fd=pipefds[0],
 
3230
    .password=&password,
 
3231
    .password_is_read=&password_is_read,
 
3232
    .quit_now=&quit_now,
 
3233
  };
 
3234
  task.func(task, queue);
 
3235
 
 
3236
  g_assert_false(password_is_read);
 
3237
  g_assert_cmpint((int)password.length, ==,
 
3238
                  (int)sizeof(dummy_test_password));
 
3239
  g_assert_nonnull(password.data);
 
3240
  g_assert_cmpint(memcmp(password.data, dummy_test_password,
 
3241
                         sizeof(dummy_test_password)), ==, 0);
 
3242
 
 
3243
  g_assert_false(quit_now);
 
3244
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
3245
 
 
3246
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
3247
        .func=read_mandos_client_output,
 
3248
        .epoll_fd=epoll_fd,
 
3249
        .fd=pipefds[0],
 
3250
        .password=&password,
 
3251
        .password_is_read=&password_is_read,
 
3252
        .quit_now=&quit_now,
 
3253
      }));
 
3254
 
 
3255
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
3256
                                   EPOLLIN | EPOLLRDHUP));
 
3257
 
 
3258
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
3259
}
 
3260
 
 
3261
static
 
3262
void test_read_mandos_client_output_malloc(__attribute__((unused))
 
3263
                                           test_fixture *fixture,
 
3264
                                           __attribute__((unused))
 
3265
                                           gconstpointer user_data){
 
3266
  __attribute__((cleanup(cleanup_close)))
 
3267
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3268
  g_assert_cmpint(epoll_fd, >=, 0);
 
3269
 
 
3270
  int pipefds[2];
 
3271
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
3272
 
 
3273
  const char dummy_test_password[] = "dummy test password";
 
3274
  /* Start with an empty buffer */
 
3275
  __attribute__((cleanup(cleanup_buffer)))
 
3276
    buffer password = {};
 
3277
 
 
3278
  bool password_is_read = false;
 
3279
  bool quit_now = false;
 
3280
  __attribute__((cleanup(cleanup_queue)))
 
3281
    task_queue *queue = create_queue();
 
3282
  g_assert_nonnull(queue);
 
3283
 
 
3284
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
 
3285
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
 
3286
                             sizeof(dummy_test_password)),
 
3287
                  ==, (int)sizeof(dummy_test_password));
 
3288
 
 
3289
  task_context task = {
 
3290
    .func=read_mandos_client_output,
 
3291
    .epoll_fd=epoll_fd,
 
3292
    .fd=pipefds[0],
 
3293
    .password=&password,
 
3294
    .password_is_read=&password_is_read,
 
3295
    .quit_now=&quit_now,
 
3296
  };
 
3297
  task.func(task, queue);
 
3298
 
 
3299
  g_assert_false(password_is_read);
 
3300
  g_assert_cmpint((int)password.length, ==,
 
3301
                  (int)sizeof(dummy_test_password));
 
3302
  g_assert_nonnull(password.data);
 
3303
  g_assert_cmpint(memcmp(password.data, dummy_test_password,
 
3304
                         sizeof(dummy_test_password)), ==, 0);
 
3305
 
 
3306
  g_assert_false(quit_now);
 
3307
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
3308
 
 
3309
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
3310
        .func=read_mandos_client_output,
 
3311
        .epoll_fd=epoll_fd,
 
3312
        .fd=pipefds[0],
 
3313
        .password=&password,
 
3314
        .password_is_read=&password_is_read,
 
3315
        .quit_now=&quit_now,
 
3316
      }));
 
3317
 
 
3318
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
3319
                                   EPOLLIN | EPOLLRDHUP));
 
3320
 
 
3321
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
3322
}
 
3323
 
 
3324
static
 
3325
void test_read_mandos_client_output_append(__attribute__((unused))
 
3326
                                           test_fixture *fixture,
 
3327
                                           __attribute__((unused))
 
3328
                                           gconstpointer user_data){
 
3329
  __attribute__((cleanup(cleanup_close)))
 
3330
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3331
  g_assert_cmpint(epoll_fd, >=, 0);
 
3332
 
 
3333
  int pipefds[2];
 
3334
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
3335
 
 
3336
  const char dummy_test_password[] = "dummy test password";
 
3337
  __attribute__((cleanup(cleanup_buffer)))
 
3338
    buffer password = {
 
3339
    .data=malloc(PIPE_BUF),
 
3340
    .length=PIPE_BUF,
 
3341
    .allocated=PIPE_BUF,
 
3342
  };
 
3343
  g_assert_nonnull(password.data);
 
3344
  if(mlock(password.data, password.allocated) != 0){
 
3345
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
3346
  }
 
3347
 
 
3348
  memset(password.data, 'x', PIPE_BUF);
 
3349
  char password_expected[PIPE_BUF];
 
3350
  memcpy(password_expected, password.data, PIPE_BUF);
 
3351
 
 
3352
  bool password_is_read = false;
 
3353
  bool quit_now = false;
 
3354
  __attribute__((cleanup(cleanup_queue)))
 
3355
    task_queue *queue = create_queue();
 
3356
  g_assert_nonnull(queue);
 
3357
 
 
3358
  g_assert_true(sizeof(dummy_test_password) <= PIPE_BUF);
 
3359
  g_assert_cmpint((int)write(pipefds[1], dummy_test_password,
 
3360
                             sizeof(dummy_test_password)),
 
3361
                  ==, (int)sizeof(dummy_test_password));
 
3362
 
 
3363
  task_context task = {
 
3364
    .func=read_mandos_client_output,
 
3365
    .epoll_fd=epoll_fd,
 
3366
    .fd=pipefds[0],
 
3367
    .password=&password,
 
3368
    .password_is_read=&password_is_read,
 
3369
    .quit_now=&quit_now,
 
3370
  };
 
3371
  task.func(task, queue);
 
3372
 
 
3373
  g_assert_false(password_is_read);
 
3374
  g_assert_cmpint((int)password.length, ==,
 
3375
                  PIPE_BUF + sizeof(dummy_test_password));
 
3376
  g_assert_nonnull(password.data);
 
3377
  g_assert_cmpint(memcmp(password_expected, password.data, PIPE_BUF),
 
3378
                  ==, 0);
 
3379
  g_assert_cmpint(memcmp(password.data + PIPE_BUF,
 
3380
                         dummy_test_password,
 
3381
                         sizeof(dummy_test_password)), ==, 0);
 
3382
  g_assert_false(quit_now);
 
3383
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
3384
 
 
3385
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
3386
        .func=read_mandos_client_output,
 
3387
        .epoll_fd=epoll_fd,
 
3388
        .fd=pipefds[0],
 
3389
        .password=&password,
 
3390
        .password_is_read=&password_is_read,
 
3391
        .quit_now=&quit_now,
 
3392
      }));
 
3393
 
 
3394
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
3395
                                   EPOLLIN | EPOLLRDHUP));
 
3396
}
 
3397
 
 
3398
static char *make_temporary_directory(void);
 
3399
 
 
3400
static void test_add_inotify_dir_watch(__attribute__((unused))
 
3401
                                       test_fixture *fixture,
 
3402
                                       __attribute__((unused))
 
3403
                                       gconstpointer user_data){
 
3404
  __attribute__((cleanup(cleanup_close)))
 
3405
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3406
  g_assert_cmpint(epoll_fd, >=, 0);
 
3407
  __attribute__((cleanup(cleanup_queue)))
 
3408
    task_queue *queue = create_queue();
 
3409
  g_assert_nonnull(queue);
 
3410
  __attribute__((cleanup(string_set_clear)))
 
3411
    string_set cancelled_filenames = {};
 
3412
  const mono_microsecs current_time = 0;
 
3413
 
 
3414
  bool quit_now = false;
 
3415
  buffer password = {};
 
3416
  bool mandos_client_exited = false;
 
3417
  bool password_is_read = false;
 
3418
 
 
3419
  __attribute__((cleanup(cleanup_string)))
 
3420
    char *tempdir = make_temporary_directory();
 
3421
  g_assert_nonnull(tempdir);
 
3422
 
 
3423
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3424
                                      &password, tempdir,
 
3425
                                      &cancelled_filenames,
 
3426
                                      &current_time,
 
3427
                                      &mandos_client_exited,
 
3428
                                      &password_is_read));
 
3429
 
 
3430
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3431
 
 
3432
  const task_context *const added_read_task
 
3433
    = find_matching_task(queue, (task_context){
 
3434
        .func=read_inotify_event,
 
3435
        .epoll_fd=epoll_fd,
 
3436
        .quit_now=&quit_now,
 
3437
        .password=&password,
 
3438
        .filename=tempdir,
 
3439
        .cancelled_filenames=&cancelled_filenames,
 
3440
        .current_time=&current_time,
 
3441
        .mandos_client_exited=&mandos_client_exited,
 
3442
        .password_is_read=&password_is_read,
 
3443
      });
 
3444
  g_assert_nonnull(added_read_task);
 
3445
 
 
3446
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3447
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3448
  g_assert_true(epoll_set_contains(added_read_task->epoll_fd,
 
3449
                                   added_read_task->fd,
 
3450
                                   EPOLLIN | EPOLLRDHUP));
 
3451
 
 
3452
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3453
}
 
3454
 
 
3455
static char *make_temporary_directory(void){
 
3456
  char *name = strdup("/tmp/mandosXXXXXX");
 
3457
  g_assert_nonnull(name);
 
3458
  char *result = mkdtemp(name);
 
3459
  if(result == NULL){
 
3460
    free(name);
 
3461
  }
 
3462
  return result;
 
3463
}
 
3464
 
 
3465
static void test_add_inotify_dir_watch_fail(__attribute__((unused))
 
3466
                                            test_fixture *fixture,
 
3467
                                            __attribute__((unused))
 
3468
                                            gconstpointer user_data){
 
3469
  __attribute__((cleanup(cleanup_close)))
 
3470
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3471
  g_assert_cmpint(epoll_fd, >=, 0);
 
3472
  __attribute__((cleanup(cleanup_queue)))
 
3473
    task_queue *queue = create_queue();
 
3474
  g_assert_nonnull(queue);
 
3475
  __attribute__((cleanup(string_set_clear)))
 
3476
    string_set cancelled_filenames = {};
 
3477
  const mono_microsecs current_time = 0;
 
3478
 
 
3479
  bool quit_now = false;
 
3480
  buffer password = {};
 
3481
  bool mandos_client_exited = false;
 
3482
  bool password_is_read = false;
 
3483
 
 
3484
  const char nonexistent_dir[] = "/nonexistent";
 
3485
 
 
3486
  FILE *real_stderr = stderr;
 
3487
  FILE *devnull = fopen("/dev/null", "we");
 
3488
  g_assert_nonnull(devnull);
 
3489
  stderr = devnull;
 
3490
  g_assert_false(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3491
                                       &password, nonexistent_dir,
 
3492
                                       &cancelled_filenames,
 
3493
                                       &current_time,
 
3494
                                       &mandos_client_exited,
 
3495
                                       &password_is_read));
 
3496
  stderr = real_stderr;
 
3497
  g_assert_cmpint(fclose(devnull), ==, 0);
 
3498
 
 
3499
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
3500
}
 
3501
 
 
3502
static void test_add_inotify_dir_watch_nondir(__attribute__((unused))
 
3503
                                              test_fixture *fixture,
 
3504
                                            __attribute__((unused))
 
3505
                                              gconstpointer
 
3506
                                              user_data){
 
3507
  __attribute__((cleanup(cleanup_close)))
 
3508
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3509
  g_assert_cmpint(epoll_fd, >=, 0);
 
3510
  __attribute__((cleanup(cleanup_queue)))
 
3511
    task_queue *queue = create_queue();
 
3512
  g_assert_nonnull(queue);
 
3513
  __attribute__((cleanup(string_set_clear)))
 
3514
    string_set cancelled_filenames = {};
 
3515
  const mono_microsecs current_time = 0;
 
3516
 
 
3517
  bool quit_now = false;
 
3518
  buffer password = {};
 
3519
  bool mandos_client_exited = false;
 
3520
  bool password_is_read = false;
 
3521
 
 
3522
  const char not_a_directory[] = "/dev/tty";
 
3523
 
 
3524
  FILE *real_stderr = stderr;
 
3525
  FILE *devnull = fopen("/dev/null", "we");
 
3526
  g_assert_nonnull(devnull);
 
3527
  stderr = devnull;
 
3528
  g_assert_false(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3529
                                       &password, not_a_directory,
 
3530
                                       &cancelled_filenames,
 
3531
                                       &current_time,
 
3532
                                       &mandos_client_exited,
 
3533
                                       &password_is_read));
 
3534
  stderr = real_stderr;
 
3535
  g_assert_cmpint(fclose(devnull), ==, 0);
 
3536
 
 
3537
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
3538
}
 
3539
 
 
3540
static void test_add_inotify_dir_watch_EAGAIN(__attribute__((unused))
 
3541
                                              test_fixture *fixture,
 
3542
                                              __attribute__((unused))
 
3543
                                              gconstpointer
 
3544
                                              user_data){
 
3545
  __attribute__((cleanup(cleanup_close)))
 
3546
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3547
  g_assert_cmpint(epoll_fd, >=, 0);
 
3548
  __attribute__((cleanup(cleanup_queue)))
 
3549
    task_queue *queue = create_queue();
 
3550
  g_assert_nonnull(queue);
 
3551
  __attribute__((cleanup(string_set_clear)))
 
3552
    string_set cancelled_filenames = {};
 
3553
  const mono_microsecs current_time = 0;
 
3554
 
 
3555
  bool quit_now = false;
 
3556
  buffer password = {};
 
3557
  bool mandos_client_exited = false;
 
3558
  bool password_is_read = false;
 
3559
 
 
3560
  __attribute__((cleanup(cleanup_string)))
 
3561
    char *tempdir = make_temporary_directory();
 
3562
  g_assert_nonnull(tempdir);
 
3563
 
 
3564
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3565
                                      &password, tempdir,
 
3566
                                      &cancelled_filenames,
 
3567
                                      &current_time,
 
3568
                                      &mandos_client_exited,
 
3569
                                      &password_is_read));
 
3570
 
 
3571
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3572
 
 
3573
  const task_context *const added_read_task
 
3574
    = find_matching_task(queue,
 
3575
                         (task_context){ .func=read_inotify_event });
 
3576
  g_assert_nonnull(added_read_task);
 
3577
 
 
3578
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3579
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3580
 
 
3581
  /* "sufficient to read at least one event." - inotify(7) */
 
3582
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3583
                              + NAME_MAX + 1);
 
3584
  struct inotify_event *ievent = malloc(ievent_size);
 
3585
  g_assert_nonnull(ievent);
 
3586
 
 
3587
  g_assert_cmpint(read(added_read_task->fd, ievent, ievent_size), ==,
 
3588
                  -1);
 
3589
  g_assert_cmpint(errno, ==, EAGAIN);
 
3590
 
 
3591
  free(ievent);
 
3592
 
 
3593
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3594
}
 
3595
 
 
3596
static char *make_temporary_file_in_directory(const char
 
3597
                                              *const dir);
 
3598
 
 
3599
static
 
3600
void test_add_inotify_dir_watch_IN_CLOSE_WRITE(__attribute__((unused))
 
3601
                                               test_fixture *fixture,
 
3602
                                               __attribute__((unused))
 
3603
                                               gconstpointer
 
3604
                                               user_data){
 
3605
  __attribute__((cleanup(cleanup_close)))
 
3606
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3607
  g_assert_cmpint(epoll_fd, >=, 0);
 
3608
  __attribute__((cleanup(cleanup_queue)))
 
3609
    task_queue *queue = create_queue();
 
3610
  g_assert_nonnull(queue);
 
3611
  __attribute__((cleanup(string_set_clear)))
 
3612
    string_set cancelled_filenames = {};
 
3613
  const mono_microsecs current_time = 0;
 
3614
 
 
3615
  bool quit_now = false;
 
3616
  buffer password = {};
 
3617
  bool mandos_client_exited = false;
 
3618
  bool password_is_read = false;
 
3619
 
 
3620
  __attribute__((cleanup(cleanup_string)))
 
3621
    char *tempdir = make_temporary_directory();
 
3622
  g_assert_nonnull(tempdir);
 
3623
 
 
3624
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3625
                                      &password, tempdir,
 
3626
                                      &cancelled_filenames,
 
3627
                                      &current_time,
 
3628
                                      &mandos_client_exited,
 
3629
                                      &password_is_read));
 
3630
 
 
3631
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3632
 
 
3633
  const task_context *const added_read_task
 
3634
    = find_matching_task(queue,
 
3635
                         (task_context){ .func=read_inotify_event });
 
3636
  g_assert_nonnull(added_read_task);
 
3637
 
 
3638
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3639
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3640
 
 
3641
  __attribute__((cleanup(cleanup_string)))
 
3642
    char *filename = make_temporary_file_in_directory(tempdir);
 
3643
  g_assert_nonnull(filename);
 
3644
 
 
3645
  /* "sufficient to read at least one event." - inotify(7) */
 
3646
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3647
                              + NAME_MAX + 1);
 
3648
  struct inotify_event *ievent = malloc(ievent_size);
 
3649
  g_assert_nonnull(ievent);
 
3650
 
 
3651
  ssize_t read_size = 0;
 
3652
  read_size = read(added_read_task->fd, ievent, ievent_size);
 
3653
 
 
3654
  g_assert_cmpint((int)read_size, >, 0);
 
3655
  g_assert_true(ievent->mask & IN_CLOSE_WRITE);
 
3656
  g_assert_cmpstr(ievent->name, ==, basename(filename));
 
3657
 
 
3658
  free(ievent);
 
3659
 
 
3660
  g_assert_cmpint(unlink(filename), ==, 0);
 
3661
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3662
}
 
3663
 
 
3664
static char *make_temporary_prefixed_file_in_directory(const char
 
3665
                                                       *const prefix,
 
3666
                                                       const char
 
3667
                                                       *const dir){
 
3668
  char *filename = NULL;
 
3669
  g_assert_cmpint(asprintf(&filename, "%s/%sXXXXXX", dir, prefix),
 
3670
                  >, 0);
 
3671
  g_assert_nonnull(filename);
 
3672
  const int fd = mkostemp(filename, O_CLOEXEC);
 
3673
  g_assert_cmpint(fd, >=, 0);
 
3674
  g_assert_cmpint(close(fd), ==, 0);
 
3675
  return filename;
 
3676
}
 
3677
 
 
3678
static char *make_temporary_file_in_directory(const char
 
3679
                                              *const dir){
 
3680
  return make_temporary_prefixed_file_in_directory("temp", dir);
 
3681
}
 
3682
 
 
3683
static
 
3684
void test_add_inotify_dir_watch_IN_MOVED_TO(__attribute__((unused))
 
3685
                                            test_fixture *fixture,
 
3686
                                            __attribute__((unused))
 
3687
                                            gconstpointer user_data){
 
3688
  __attribute__((cleanup(cleanup_close)))
 
3689
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3690
  g_assert_cmpint(epoll_fd, >=, 0);
 
3691
  __attribute__((cleanup(cleanup_queue)))
 
3692
    task_queue *queue = create_queue();
 
3693
  g_assert_nonnull(queue);
 
3694
  __attribute__((cleanup(string_set_clear)))
 
3695
    string_set cancelled_filenames = {};
 
3696
  const mono_microsecs current_time = 0;
 
3697
 
 
3698
  bool quit_now = false;
 
3699
  buffer password = {};
 
3700
  bool mandos_client_exited = false;
 
3701
  bool password_is_read = false;
 
3702
 
 
3703
  __attribute__((cleanup(cleanup_string)))
 
3704
    char *watchdir = make_temporary_directory();
 
3705
  g_assert_nonnull(watchdir);
 
3706
 
 
3707
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3708
                                      &password, watchdir,
 
3709
                                      &cancelled_filenames,
 
3710
                                      &current_time,
 
3711
                                      &mandos_client_exited,
 
3712
                                      &password_is_read));
 
3713
 
 
3714
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3715
 
 
3716
  const task_context *const added_read_task
 
3717
    = find_matching_task(queue,
 
3718
                         (task_context){ .func=read_inotify_event });
 
3719
  g_assert_nonnull(added_read_task);
 
3720
 
 
3721
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3722
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3723
 
 
3724
  char *sourcedir = make_temporary_directory();
 
3725
  g_assert_nonnull(sourcedir);
 
3726
 
 
3727
  __attribute__((cleanup(cleanup_string)))
 
3728
    char *filename = make_temporary_file_in_directory(sourcedir);
 
3729
  g_assert_nonnull(filename);
 
3730
 
 
3731
  __attribute__((cleanup(cleanup_string)))
 
3732
    char *targetfilename = NULL;
 
3733
  g_assert_cmpint(asprintf(&targetfilename, "%s/%s", watchdir,
 
3734
                           basename(filename)), >, 0);
 
3735
  g_assert_nonnull(targetfilename);
 
3736
 
 
3737
  g_assert_cmpint(rename(filename, targetfilename), ==, 0);
 
3738
  g_assert_cmpint(rmdir(sourcedir), ==, 0);
 
3739
  free(sourcedir);
 
3740
 
 
3741
  /* "sufficient to read at least one event." - inotify(7) */
 
3742
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3743
                              + NAME_MAX + 1);
 
3744
  struct inotify_event *ievent = malloc(ievent_size);
 
3745
  g_assert_nonnull(ievent);
 
3746
 
 
3747
  ssize_t read_size = read(added_read_task->fd, ievent, ievent_size);
 
3748
 
 
3749
  g_assert_cmpint((int)read_size, >, 0);
 
3750
  g_assert_true(ievent->mask & IN_MOVED_TO);
 
3751
  g_assert_cmpstr(ievent->name, ==, basename(targetfilename));
 
3752
 
 
3753
  free(ievent);
 
3754
 
 
3755
  g_assert_cmpint(unlink(targetfilename), ==, 0);
 
3756
  g_assert_cmpint(rmdir(watchdir), ==, 0);
 
3757
}
 
3758
 
 
3759
static
 
3760
void test_add_inotify_dir_watch_IN_MOVED_FROM(__attribute__((unused))
 
3761
                                              test_fixture *fixture,
 
3762
                                              __attribute__((unused))
 
3763
                                              gconstpointer
 
3764
                                              user_data){
 
3765
  __attribute__((cleanup(cleanup_close)))
 
3766
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3767
  g_assert_cmpint(epoll_fd, >=, 0);
 
3768
  __attribute__((cleanup(cleanup_queue)))
 
3769
    task_queue *queue = create_queue();
 
3770
  g_assert_nonnull(queue);
 
3771
  __attribute__((cleanup(string_set_clear)))
 
3772
    string_set cancelled_filenames = {};
 
3773
  const mono_microsecs current_time = 0;
 
3774
 
 
3775
  bool quit_now = false;
 
3776
  buffer password = {};
 
3777
  bool mandos_client_exited = false;
 
3778
  bool password_is_read = false;
 
3779
 
 
3780
  __attribute__((cleanup(cleanup_string)))
 
3781
    char *tempdir = make_temporary_directory();
 
3782
  g_assert_nonnull(tempdir);
 
3783
 
 
3784
  __attribute__((cleanup(cleanup_string)))
 
3785
    char *tempfilename = make_temporary_file_in_directory(tempdir);
 
3786
  g_assert_nonnull(tempfilename);
 
3787
 
 
3788
  __attribute__((cleanup(cleanup_string)))
 
3789
    char *targetdir = make_temporary_directory();
 
3790
  g_assert_nonnull(targetdir);
 
3791
 
 
3792
  __attribute__((cleanup(cleanup_string)))
 
3793
    char *targetfilename = NULL;
 
3794
  g_assert_cmpint(asprintf(&targetfilename, "%s/%s", targetdir,
 
3795
                           basename(tempfilename)), >, 0);
 
3796
  g_assert_nonnull(targetfilename);
 
3797
 
 
3798
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3799
                                      &password, tempdir,
 
3800
                                      &cancelled_filenames,
 
3801
                                      &current_time,
 
3802
                                      &mandos_client_exited,
 
3803
                                      &password_is_read));
 
3804
 
 
3805
  g_assert_cmpint(rename(tempfilename, targetfilename), ==, 0);
 
3806
 
 
3807
  const task_context *const added_read_task
 
3808
    = find_matching_task(queue,
 
3809
                         (task_context){ .func=read_inotify_event });
 
3810
  g_assert_nonnull(added_read_task);
 
3811
 
 
3812
  /* "sufficient to read at least one event." - inotify(7) */
 
3813
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3814
                              + NAME_MAX + 1);
 
3815
  struct inotify_event *ievent = malloc(ievent_size);
 
3816
  g_assert_nonnull(ievent);
 
3817
 
 
3818
  ssize_t read_size = read(added_read_task->fd, ievent, ievent_size);
 
3819
 
 
3820
  g_assert_cmpint((int)read_size, >, 0);
 
3821
  g_assert_true(ievent->mask & IN_MOVED_FROM);
 
3822
  g_assert_cmpstr(ievent->name, ==, basename(tempfilename));
 
3823
 
 
3824
  free(ievent);
 
3825
 
 
3826
  g_assert_cmpint(unlink(targetfilename), ==, 0);
 
3827
  g_assert_cmpint(rmdir(targetdir), ==, 0);
 
3828
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3829
}
 
3830
 
 
3831
static
 
3832
void test_add_inotify_dir_watch_IN_DELETE(__attribute__((unused))
 
3833
                                          test_fixture *fixture,
 
3834
                                          __attribute__((unused))
 
3835
                                          gconstpointer user_data){
 
3836
  __attribute__((cleanup(cleanup_close)))
 
3837
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3838
  g_assert_cmpint(epoll_fd, >=, 0);
 
3839
  __attribute__((cleanup(cleanup_queue)))
 
3840
    task_queue *queue = create_queue();
 
3841
  g_assert_nonnull(queue);
 
3842
  __attribute__((cleanup(string_set_clear)))
 
3843
    string_set cancelled_filenames = {};
 
3844
  const mono_microsecs current_time = 0;
 
3845
 
 
3846
  bool quit_now = false;
 
3847
  buffer password = {};
 
3848
  bool mandos_client_exited = false;
 
3849
  bool password_is_read = false;
 
3850
 
 
3851
  __attribute__((cleanup(cleanup_string)))
 
3852
    char *tempdir = make_temporary_directory();
 
3853
  g_assert_nonnull(tempdir);
 
3854
 
 
3855
  __attribute__((cleanup(cleanup_string)))
 
3856
    char *tempfile = make_temporary_file_in_directory(tempdir);
 
3857
  g_assert_nonnull(tempfile);
 
3858
 
 
3859
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3860
                                      &password, tempdir,
 
3861
                                      &cancelled_filenames,
 
3862
                                      &current_time,
 
3863
                                      &mandos_client_exited,
 
3864
                                      &password_is_read));
 
3865
  g_assert_cmpint(unlink(tempfile), ==, 0);
 
3866
 
 
3867
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3868
 
 
3869
  const task_context *const added_read_task
 
3870
    = find_matching_task(queue,
 
3871
                         (task_context){ .func=read_inotify_event });
 
3872
  g_assert_nonnull(added_read_task);
 
3873
 
 
3874
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3875
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3876
 
 
3877
  /* "sufficient to read at least one event." - inotify(7) */
 
3878
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3879
                              + NAME_MAX + 1);
 
3880
  struct inotify_event *ievent = malloc(ievent_size);
 
3881
  g_assert_nonnull(ievent);
 
3882
 
 
3883
  ssize_t read_size = 0;
 
3884
  read_size = read(added_read_task->fd, ievent, ievent_size);
 
3885
 
 
3886
  g_assert_cmpint((int)read_size, >, 0);
 
3887
  g_assert_true(ievent->mask & IN_DELETE);
 
3888
  g_assert_cmpstr(ievent->name, ==, basename(tempfile));
 
3889
 
 
3890
  free(ievent);
 
3891
 
 
3892
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3893
}
 
3894
 
 
3895
static
 
3896
void test_add_inotify_dir_watch_IN_EXCL_UNLINK(__attribute__((unused))
 
3897
                                               test_fixture *fixture,
 
3898
                                               __attribute__((unused))
 
3899
                                               gconstpointer
 
3900
                                               user_data){
 
3901
  __attribute__((cleanup(cleanup_close)))
 
3902
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3903
  g_assert_cmpint(epoll_fd, >=, 0);
 
3904
  __attribute__((cleanup(cleanup_queue)))
 
3905
    task_queue *queue = create_queue();
 
3906
  g_assert_nonnull(queue);
 
3907
  __attribute__((cleanup(string_set_clear)))
 
3908
    string_set cancelled_filenames = {};
 
3909
  const mono_microsecs current_time = 0;
 
3910
 
 
3911
  bool quit_now = false;
 
3912
  buffer password = {};
 
3913
  bool mandos_client_exited = false;
 
3914
  bool password_is_read = false;
 
3915
 
 
3916
  __attribute__((cleanup(cleanup_string)))
 
3917
    char *tempdir = make_temporary_directory();
 
3918
  g_assert_nonnull(tempdir);
 
3919
 
 
3920
  __attribute__((cleanup(cleanup_string)))
 
3921
    char *tempfile = make_temporary_file_in_directory(tempdir);
 
3922
  g_assert_nonnull(tempfile);
 
3923
  int tempfile_fd = open(tempfile, O_WRONLY | O_CLOEXEC | O_NOCTTY
 
3924
                         | O_NOFOLLOW);
 
3925
  g_assert_cmpint(tempfile_fd, >, 2);
 
3926
 
 
3927
  g_assert_true(add_inotify_dir_watch(queue, epoll_fd, &quit_now,
 
3928
                                      &password, tempdir,
 
3929
                                      &cancelled_filenames,
 
3930
                                      &current_time,
 
3931
                                      &mandos_client_exited,
 
3932
                                      &password_is_read));
 
3933
  g_assert_cmpint(unlink(tempfile), ==, 0);
 
3934
 
 
3935
  g_assert_cmpuint((unsigned int)queue->length, >, 0);
 
3936
 
 
3937
  const task_context *const added_read_task
 
3938
    = find_matching_task(queue,
 
3939
                         (task_context){ .func=read_inotify_event });
 
3940
  g_assert_nonnull(added_read_task);
 
3941
 
 
3942
  g_assert_cmpint(added_read_task->fd, >, 2);
 
3943
  g_assert_true(fd_has_cloexec_and_nonblock(added_read_task->fd));
 
3944
 
 
3945
  /* "sufficient to read at least one event." - inotify(7) */
 
3946
  const size_t ievent_size = (sizeof(struct inotify_event)
 
3947
                              + NAME_MAX + 1);
 
3948
  struct inotify_event *ievent = malloc(ievent_size);
 
3949
  g_assert_nonnull(ievent);
 
3950
 
 
3951
  ssize_t read_size = 0;
 
3952
  read_size = read(added_read_task->fd, ievent, ievent_size);
 
3953
 
 
3954
  g_assert_cmpint((int)read_size, >, 0);
 
3955
  g_assert_true(ievent->mask & IN_DELETE);
 
3956
  g_assert_cmpstr(ievent->name, ==, basename(tempfile));
 
3957
 
 
3958
  g_assert_cmpint(close(tempfile_fd), ==, 0);
 
3959
 
 
3960
  /* IN_EXCL_UNLINK should make the closing of the previously unlinked
 
3961
     file not appear as an ievent, so we should not see it now. */
 
3962
  read_size = read(added_read_task->fd, ievent, ievent_size);
 
3963
  g_assert_cmpint((int)read_size, ==, -1);
 
3964
  g_assert_true(errno == EAGAIN);
 
3965
 
 
3966
  free(ievent);
 
3967
 
 
3968
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
3969
}
 
3970
 
 
3971
static void test_read_inotify_event_readerror(__attribute__((unused))
 
3972
                                              test_fixture *fixture,
 
3973
                                              __attribute__((unused))
 
3974
                                              gconstpointer
 
3975
                                              user_data){
 
3976
  __attribute__((cleanup(cleanup_close)))
 
3977
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
3978
  g_assert_cmpint(epoll_fd, >=, 0);
 
3979
  const mono_microsecs current_time = 0;
 
3980
 
 
3981
  /* Reading /proc/self/mem from offset 0 will always result in EIO */
 
3982
  const int fd = open("/proc/self/mem",
 
3983
                      O_RDONLY | O_CLOEXEC | O_NOCTTY);
 
3984
 
 
3985
  bool quit_now = false;
 
3986
  __attribute__((cleanup(cleanup_queue)))
 
3987
    task_queue *queue = create_queue();
 
3988
  g_assert_nonnull(queue);
 
3989
 
 
3990
  task_context task = {
 
3991
    .func=read_inotify_event,
 
3992
    .epoll_fd=epoll_fd,
 
3993
    .fd=fd,
 
3994
    .quit_now=&quit_now,
 
3995
    .filename=strdup("/nonexistent"),
 
3996
    .cancelled_filenames = &(string_set){},
 
3997
    .notafter=0,
 
3998
    .current_time=&current_time,
 
3999
  };
 
4000
  g_assert_nonnull(task.filename);
 
4001
  run_task_with_stderr_to_dev_null(task, queue);
 
4002
  g_assert_true(quit_now);
 
4003
  g_assert_true(queue->next_run == 0);
 
4004
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4005
 
 
4006
  g_assert_true(epoll_set_does_not_contain(epoll_fd, fd));
 
4007
 
 
4008
  g_assert_cmpint(close(fd), ==, -1);
 
4009
}
 
4010
 
 
4011
static void test_read_inotify_event_bad_epoll(__attribute__((unused))
 
4012
                                              test_fixture *fixture,
 
4013
                                              __attribute__((unused))
 
4014
                                              gconstpointer
 
4015
                                              user_data){
 
4016
  const mono_microsecs current_time = 17;
 
4017
 
 
4018
  int pipefds[2];
 
4019
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4020
  const int epoll_fd = pipefds[0]; /* This will obviously fail */
 
4021
 
 
4022
  bool quit_now = false;
 
4023
  buffer password = {};
 
4024
  bool mandos_client_exited = false;
 
4025
  bool password_is_read = false;
 
4026
  __attribute__((cleanup(cleanup_queue)))
 
4027
    task_queue *queue = create_queue();
 
4028
  g_assert_nonnull(queue);
 
4029
 
 
4030
  task_context task = {
 
4031
    .func=read_inotify_event,
 
4032
    .epoll_fd=epoll_fd,
 
4033
    .fd=pipefds[0],
 
4034
    .quit_now=&quit_now,
 
4035
    .password=&password,
 
4036
    .filename=strdup("/nonexistent"),
 
4037
    .cancelled_filenames = &(string_set){},
 
4038
    .notafter=0,
 
4039
    .current_time=&current_time,
 
4040
    .mandos_client_exited=&mandos_client_exited,
 
4041
    .password_is_read=&password_is_read,
 
4042
  };
 
4043
  g_assert_nonnull(task.filename);
 
4044
  run_task_with_stderr_to_dev_null(task, queue);
 
4045
 
 
4046
  g_assert_nonnull(find_matching_task(queue, task));
 
4047
  g_assert_true(queue->next_run == 1000000 + current_time);
 
4048
 
 
4049
  g_assert_cmpint(close(pipefds[0]), ==, 0);
 
4050
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4051
}
 
4052
 
 
4053
static void test_read_inotify_event_nodata(__attribute__((unused))
 
4054
                                           test_fixture *fixture,
 
4055
                                           __attribute__((unused))
 
4056
                                           gconstpointer user_data){
 
4057
  __attribute__((cleanup(cleanup_close)))
 
4058
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4059
  g_assert_cmpint(epoll_fd, >=, 0);
 
4060
  const mono_microsecs current_time = 0;
 
4061
 
 
4062
  int pipefds[2];
 
4063
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4064
 
 
4065
  bool quit_now = false;
 
4066
  buffer password = {};
 
4067
  bool mandos_client_exited = false;
 
4068
  bool password_is_read = false;
 
4069
  __attribute__((cleanup(cleanup_queue)))
 
4070
    task_queue *queue = create_queue();
 
4071
  g_assert_nonnull(queue);
 
4072
 
 
4073
  task_context task = {
 
4074
    .func=read_inotify_event,
 
4075
    .epoll_fd=epoll_fd,
 
4076
    .fd=pipefds[0],
 
4077
    .quit_now=&quit_now,
 
4078
    .password=&password,
 
4079
    .filename=strdup("/nonexistent"),
 
4080
    .cancelled_filenames = &(string_set){},
 
4081
    .notafter=0,
 
4082
    .current_time=&current_time,
 
4083
    .mandos_client_exited=&mandos_client_exited,
 
4084
    .password_is_read=&password_is_read,
 
4085
  };
 
4086
  g_assert_nonnull(task.filename);
 
4087
  task.func(task, queue);
 
4088
  g_assert_false(quit_now);
 
4089
  g_assert_true(queue->next_run == 0);
 
4090
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4091
 
 
4092
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4093
        .func=read_inotify_event,
 
4094
        .epoll_fd=epoll_fd,
 
4095
        .fd=pipefds[0],
 
4096
        .quit_now=&quit_now,
 
4097
        .password=&password,
 
4098
        .filename=task.filename,
 
4099
        .cancelled_filenames=task.cancelled_filenames,
 
4100
        .current_time=&current_time,
 
4101
        .mandos_client_exited=&mandos_client_exited,
 
4102
        .password_is_read=&password_is_read,
 
4103
      }));
 
4104
 
 
4105
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4106
                                   EPOLLIN | EPOLLRDHUP));
 
4107
 
 
4108
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4109
}
 
4110
 
 
4111
static void test_read_inotify_event_eof(__attribute__((unused))
 
4112
                                        test_fixture *fixture,
 
4113
                                        __attribute__((unused))
 
4114
                                        gconstpointer user_data){
 
4115
  __attribute__((cleanup(cleanup_close)))
 
4116
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4117
  g_assert_cmpint(epoll_fd, >=, 0);
 
4118
  const mono_microsecs current_time = 0;
 
4119
 
 
4120
  int pipefds[2];
 
4121
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4122
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4123
 
 
4124
  bool quit_now = false;
 
4125
  buffer password = {};
 
4126
  __attribute__((cleanup(cleanup_queue)))
 
4127
    task_queue *queue = create_queue();
 
4128
  g_assert_nonnull(queue);
 
4129
 
 
4130
  task_context task = {
 
4131
    .func=read_inotify_event,
 
4132
    .epoll_fd=epoll_fd,
 
4133
    .fd=pipefds[0],
 
4134
    .quit_now=&quit_now,
 
4135
    .password=&password,
 
4136
    .filename=strdup("/nonexistent"),
 
4137
    .cancelled_filenames = &(string_set){},
 
4138
    .notafter=0,
 
4139
    .current_time=&current_time,
 
4140
  };
 
4141
  run_task_with_stderr_to_dev_null(task, queue);
 
4142
  g_assert_true(quit_now);
 
4143
  g_assert_true(queue->next_run == 0);
 
4144
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4145
 
 
4146
  g_assert_true(epoll_set_does_not_contain(epoll_fd, pipefds[0]));
 
4147
 
 
4148
  g_assert_cmpint(close(pipefds[0]), ==, -1);
 
4149
}
 
4150
 
 
4151
static
 
4152
void test_read_inotify_event_IN_CLOSE_WRITE(__attribute__((unused))
 
4153
                                            test_fixture *fixture,
 
4154
                                            __attribute__((unused))
 
4155
                                            gconstpointer user_data){
 
4156
  __attribute__((cleanup(cleanup_close)))
 
4157
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4158
  g_assert_cmpint(epoll_fd, >=, 0);
 
4159
  const mono_microsecs current_time = 0;
 
4160
 
 
4161
  int pipefds[2];
 
4162
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4163
 
 
4164
  /* "sufficient to read at least one event." - inotify(7) */
 
4165
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4166
                                  + NAME_MAX + 1);
 
4167
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4168
  struct {
 
4169
    struct inotify_event event;
 
4170
    char name_buffer[NAME_MAX + 1];
 
4171
  } ievent_buffer;
 
4172
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4173
 
 
4174
  const char dummy_file_name[] = "ask.dummy_file_name";
 
4175
  ievent->mask = IN_CLOSE_WRITE;
 
4176
  ievent->len = sizeof(dummy_file_name);
 
4177
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4178
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4179
                              + sizeof(dummy_file_name));
 
4180
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4181
                  ==, ievent_size);
 
4182
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4183
 
 
4184
  bool quit_now = false;
 
4185
  buffer password = {};
 
4186
  bool mandos_client_exited = false;
 
4187
  bool password_is_read = false;
 
4188
  __attribute__((cleanup(cleanup_queue)))
 
4189
    task_queue *queue = create_queue();
 
4190
  g_assert_nonnull(queue);
 
4191
 
 
4192
  task_context task = {
 
4193
    .func=read_inotify_event,
 
4194
    .epoll_fd=epoll_fd,
 
4195
    .fd=pipefds[0],
 
4196
    .quit_now=&quit_now,
 
4197
    .password=&password,
 
4198
    .filename=strdup("/nonexistent"),
 
4199
    .cancelled_filenames = &(string_set){},
 
4200
    .notafter=0,
 
4201
    .current_time=&current_time,
 
4202
    .mandos_client_exited=&mandos_client_exited,
 
4203
    .password_is_read=&password_is_read,
 
4204
  };
 
4205
  task.func(task, queue);
 
4206
  g_assert_false(quit_now);
 
4207
  g_assert_true(queue->next_run != 0);
 
4208
  g_assert_cmpuint((unsigned int)queue->length, >=, 1);
 
4209
 
 
4210
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4211
        .func=read_inotify_event,
 
4212
        .epoll_fd=epoll_fd,
 
4213
        .fd=pipefds[0],
 
4214
        .quit_now=&quit_now,
 
4215
        .password=&password,
 
4216
        .filename=task.filename,
 
4217
        .cancelled_filenames=task.cancelled_filenames,
 
4218
        .current_time=&current_time,
 
4219
        .mandos_client_exited=&mandos_client_exited,
 
4220
        .password_is_read=&password_is_read,
 
4221
      }));
 
4222
 
 
4223
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4224
                                   EPOLLIN | EPOLLRDHUP));
 
4225
 
 
4226
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
 
4227
 
 
4228
  __attribute__((cleanup(cleanup_string)))
 
4229
    char *filename = NULL;
 
4230
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4231
                           dummy_file_name), >, 0);
 
4232
  g_assert_nonnull(filename);
 
4233
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4234
        .func=open_and_parse_question,
 
4235
        .epoll_fd=epoll_fd,
 
4236
        .filename=filename,
 
4237
        .question_filename=filename,
 
4238
        .password=&password,
 
4239
        .cancelled_filenames=task.cancelled_filenames,
 
4240
        .current_time=&current_time,
 
4241
        .mandos_client_exited=&mandos_client_exited,
 
4242
        .password_is_read=&password_is_read,
 
4243
      }));
 
4244
}
 
4245
 
 
4246
static
 
4247
void test_read_inotify_event_IN_MOVED_TO(__attribute__((unused))
 
4248
                                         test_fixture *fixture,
 
4249
                                         __attribute__((unused))
 
4250
                                         gconstpointer user_data){
 
4251
  __attribute__((cleanup(cleanup_close)))
 
4252
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4253
  g_assert_cmpint(epoll_fd, >=, 0);
 
4254
  const mono_microsecs current_time = 0;
 
4255
 
 
4256
  int pipefds[2];
 
4257
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4258
 
 
4259
  /* "sufficient to read at least one event." - inotify(7) */
 
4260
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4261
                                  + NAME_MAX + 1);
 
4262
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4263
  struct {
 
4264
    struct inotify_event event;
 
4265
    char name_buffer[NAME_MAX + 1];
 
4266
  } ievent_buffer;
 
4267
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4268
 
 
4269
  const char dummy_file_name[] = "ask.dummy_file_name";
 
4270
  ievent->mask = IN_MOVED_TO;
 
4271
  ievent->len = sizeof(dummy_file_name);
 
4272
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4273
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4274
                              + sizeof(dummy_file_name));
 
4275
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4276
                  ==, ievent_size);
 
4277
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4278
 
 
4279
  bool quit_now = false;
 
4280
  buffer password = {};
 
4281
  bool mandos_client_exited = false;
 
4282
  bool password_is_read = false;
 
4283
  __attribute__((cleanup(cleanup_queue)))
 
4284
    task_queue *queue = create_queue();
 
4285
  g_assert_nonnull(queue);
 
4286
 
 
4287
  task_context task = {
 
4288
    .func=read_inotify_event,
 
4289
    .epoll_fd=epoll_fd,
 
4290
    .fd=pipefds[0],
 
4291
    .quit_now=&quit_now,
 
4292
    .password=&password,
 
4293
    .filename=strdup("/nonexistent"),
 
4294
    .cancelled_filenames = &(string_set){},
 
4295
    .notafter=0,
 
4296
    .current_time=&current_time,
 
4297
    .mandos_client_exited=&mandos_client_exited,
 
4298
    .password_is_read=&password_is_read,
 
4299
  };
 
4300
  task.func(task, queue);
 
4301
  g_assert_false(quit_now);
 
4302
  g_assert_true(queue->next_run != 0);
 
4303
  g_assert_cmpuint((unsigned int)queue->length, >=, 1);
 
4304
 
 
4305
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4306
        .func=read_inotify_event,
 
4307
        .epoll_fd=epoll_fd,
 
4308
        .fd=pipefds[0],
 
4309
        .quit_now=&quit_now,
 
4310
        .password=&password,
 
4311
        .filename=task.filename,
 
4312
        .cancelled_filenames=task.cancelled_filenames,
 
4313
        .current_time=&current_time,
 
4314
        .mandos_client_exited=&mandos_client_exited,
 
4315
        .password_is_read=&password_is_read,
 
4316
      }));
 
4317
 
 
4318
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4319
                                   EPOLLIN | EPOLLRDHUP));
 
4320
 
 
4321
  g_assert_cmpuint((unsigned int)queue->length, >=, 2);
 
4322
 
 
4323
  __attribute__((cleanup(cleanup_string)))
 
4324
    char *filename = NULL;
 
4325
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4326
                           dummy_file_name), >, 0);
 
4327
  g_assert_nonnull(filename);
 
4328
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4329
        .func=open_and_parse_question,
 
4330
        .epoll_fd=epoll_fd,
 
4331
        .filename=filename,
 
4332
        .question_filename=filename,
 
4333
        .password=&password,
 
4334
        .cancelled_filenames=task.cancelled_filenames,
 
4335
        .current_time=&current_time,
 
4336
        .mandos_client_exited=&mandos_client_exited,
 
4337
        .password_is_read=&password_is_read,
 
4338
      }));
 
4339
}
 
4340
 
 
4341
static
 
4342
void test_read_inotify_event_IN_MOVED_FROM(__attribute__((unused))
 
4343
                                           test_fixture *fixture,
 
4344
                                           __attribute__((unused))
 
4345
                                           gconstpointer user_data){
 
4346
  __attribute__((cleanup(cleanup_close)))
 
4347
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4348
  g_assert_cmpint(epoll_fd, >=, 0);
 
4349
  __attribute__((cleanup(string_set_clear)))
 
4350
    string_set cancelled_filenames = {};
 
4351
  const mono_microsecs current_time = 0;
 
4352
 
 
4353
  int pipefds[2];
 
4354
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4355
 
 
4356
  /* "sufficient to read at least one event." - inotify(7) */
 
4357
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4358
                                  + NAME_MAX + 1);
 
4359
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4360
  struct {
 
4361
    struct inotify_event event;
 
4362
    char name_buffer[NAME_MAX + 1];
 
4363
  } ievent_buffer;
 
4364
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4365
 
 
4366
  const char dummy_file_name[] = "ask.dummy_file_name";
 
4367
  ievent->mask = IN_MOVED_FROM;
 
4368
  ievent->len = sizeof(dummy_file_name);
 
4369
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4370
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4371
                              + sizeof(dummy_file_name));
 
4372
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4373
                  ==, ievent_size);
 
4374
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4375
 
 
4376
  bool quit_now = false;
 
4377
  buffer password = {};
 
4378
  bool mandos_client_exited = false;
 
4379
  bool password_is_read = false;
 
4380
  __attribute__((cleanup(cleanup_queue)))
 
4381
    task_queue *queue = create_queue();
 
4382
  g_assert_nonnull(queue);
 
4383
 
 
4384
  task_context task = {
 
4385
    .func=read_inotify_event,
 
4386
    .epoll_fd=epoll_fd,
 
4387
    .fd=pipefds[0],
 
4388
    .quit_now=&quit_now,
 
4389
    .password=&password,
 
4390
    .filename=strdup("/nonexistent"),
 
4391
    .cancelled_filenames=&cancelled_filenames,
 
4392
    .current_time=&current_time,
 
4393
    .mandos_client_exited=&mandos_client_exited,
 
4394
    .password_is_read=&password_is_read,
 
4395
  };
 
4396
  task.func(task, queue);
 
4397
  g_assert_false(quit_now);
 
4398
  g_assert_true(queue->next_run == 0);
 
4399
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4400
 
 
4401
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4402
        .func=read_inotify_event,
 
4403
        .epoll_fd=epoll_fd,
 
4404
        .fd=pipefds[0],
 
4405
        .quit_now=&quit_now,
 
4406
        .password=&password,
 
4407
        .filename=task.filename,
 
4408
        .cancelled_filenames=&cancelled_filenames,
 
4409
        .current_time=&current_time,
 
4410
        .mandos_client_exited=&mandos_client_exited,
 
4411
        .password_is_read=&password_is_read,
 
4412
      }));
 
4413
 
 
4414
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4415
                                   EPOLLIN | EPOLLRDHUP));
 
4416
 
 
4417
  __attribute__((cleanup(cleanup_string)))
 
4418
    char *filename = NULL;
 
4419
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4420
                           dummy_file_name), >, 0);
 
4421
  g_assert_nonnull(filename);
 
4422
  g_assert_true(string_set_contains(*task.cancelled_filenames,
 
4423
                                    filename));
 
4424
}
 
4425
 
 
4426
static void test_read_inotify_event_IN_DELETE(__attribute__((unused))
 
4427
                                              test_fixture *fixture,
 
4428
                                              __attribute__((unused))
 
4429
                                              gconstpointer
 
4430
                                              user_data){
 
4431
  __attribute__((cleanup(cleanup_close)))
 
4432
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4433
  g_assert_cmpint(epoll_fd, >=, 0);
 
4434
  __attribute__((cleanup(string_set_clear)))
 
4435
    string_set cancelled_filenames = {};
 
4436
  const mono_microsecs current_time = 0;
 
4437
 
 
4438
  int pipefds[2];
 
4439
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4440
 
 
4441
  /* "sufficient to read at least one event." - inotify(7) */
 
4442
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4443
                                  + NAME_MAX + 1);
 
4444
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4445
  struct {
 
4446
    struct inotify_event event;
 
4447
    char name_buffer[NAME_MAX + 1];
 
4448
  } ievent_buffer;
 
4449
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4450
 
 
4451
  const char dummy_file_name[] = "ask.dummy_file_name";
 
4452
  ievent->mask = IN_DELETE;
 
4453
  ievent->len = sizeof(dummy_file_name);
 
4454
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4455
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4456
                              + sizeof(dummy_file_name));
 
4457
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4458
                  ==, ievent_size);
 
4459
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4460
 
 
4461
  bool quit_now = false;
 
4462
  buffer password = {};
 
4463
  bool mandos_client_exited = false;
 
4464
  bool password_is_read = false;
 
4465
  __attribute__((cleanup(cleanup_queue)))
 
4466
    task_queue *queue = create_queue();
 
4467
  g_assert_nonnull(queue);
 
4468
 
 
4469
  task_context task = {
 
4470
    .func=read_inotify_event,
 
4471
    .epoll_fd=epoll_fd,
 
4472
    .fd=pipefds[0],
 
4473
    .quit_now=&quit_now,
 
4474
    .password=&password,
 
4475
    .filename=strdup("/nonexistent"),
 
4476
    .cancelled_filenames=&cancelled_filenames,
 
4477
    .current_time=&current_time,
 
4478
    .mandos_client_exited=&mandos_client_exited,
 
4479
    .password_is_read=&password_is_read,
 
4480
  };
 
4481
  task.func(task, queue);
 
4482
  g_assert_false(quit_now);
 
4483
  g_assert_true(queue->next_run == 0);
 
4484
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4485
 
 
4486
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4487
        .func=read_inotify_event,
 
4488
        .epoll_fd=epoll_fd,
 
4489
        .fd=pipefds[0],
 
4490
        .quit_now=&quit_now,
 
4491
        .password=&password,
 
4492
        .filename=task.filename,
 
4493
        .cancelled_filenames=&cancelled_filenames,
 
4494
        .current_time=&current_time,
 
4495
        .mandos_client_exited=&mandos_client_exited,
 
4496
        .password_is_read=&password_is_read,
 
4497
      }));
 
4498
 
 
4499
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4500
                                   EPOLLIN | EPOLLRDHUP));
 
4501
 
 
4502
  __attribute__((cleanup(cleanup_string)))
 
4503
    char *filename = NULL;
 
4504
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4505
                           dummy_file_name), >, 0);
 
4506
  g_assert_nonnull(filename);
 
4507
  g_assert_true(string_set_contains(*task.cancelled_filenames,
 
4508
                                    filename));
 
4509
}
 
4510
 
 
4511
static void
 
4512
test_read_inotify_event_IN_CLOSE_WRITE_badname(__attribute__((unused))
 
4513
                                               test_fixture *fixture,
 
4514
                                               __attribute__((unused))
 
4515
                                               gconstpointer
 
4516
                                               user_data){
 
4517
  __attribute__((cleanup(cleanup_close)))
 
4518
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4519
  g_assert_cmpint(epoll_fd, >=, 0);
 
4520
  const mono_microsecs current_time = 0;
 
4521
 
 
4522
  int pipefds[2];
 
4523
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4524
 
 
4525
  /* "sufficient to read at least one event." - inotify(7) */
 
4526
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4527
                                  + NAME_MAX + 1);
 
4528
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4529
  struct {
 
4530
    struct inotify_event event;
 
4531
    char name_buffer[NAME_MAX + 1];
 
4532
  } ievent_buffer;
 
4533
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4534
 
 
4535
  const char dummy_file_name[] = "ignored.dummy_file_name";
 
4536
  ievent->mask = IN_CLOSE_WRITE;
 
4537
  ievent->len = sizeof(dummy_file_name);
 
4538
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4539
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4540
                              + sizeof(dummy_file_name));
 
4541
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4542
                  ==, ievent_size);
 
4543
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4544
 
 
4545
  bool quit_now = false;
 
4546
  buffer password = {};
 
4547
  bool mandos_client_exited = false;
 
4548
  bool password_is_read = false;
 
4549
  __attribute__((cleanup(cleanup_queue)))
 
4550
    task_queue *queue = create_queue();
 
4551
  g_assert_nonnull(queue);
 
4552
 
 
4553
  task_context task = {
 
4554
    .func=read_inotify_event,
 
4555
    .epoll_fd=epoll_fd,
 
4556
    .fd=pipefds[0],
 
4557
    .quit_now=&quit_now,
 
4558
    .password=&password,
 
4559
    .filename=strdup("/nonexistent"),
 
4560
    .cancelled_filenames = &(string_set){},
 
4561
    .notafter=0,
 
4562
    .current_time=&current_time,
 
4563
    .mandos_client_exited=&mandos_client_exited,
 
4564
    .password_is_read=&password_is_read,
 
4565
  };
 
4566
  task.func(task, queue);
 
4567
  g_assert_false(quit_now);
 
4568
  g_assert_true(queue->next_run == 0);
 
4569
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4570
 
 
4571
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4572
        .func=read_inotify_event,
 
4573
        .epoll_fd=epoll_fd,
 
4574
        .fd=pipefds[0],
 
4575
        .quit_now=&quit_now,
 
4576
        .password=&password,
 
4577
        .filename=task.filename,
 
4578
        .cancelled_filenames=task.cancelled_filenames,
 
4579
        .current_time=&current_time,
 
4580
        .mandos_client_exited=&mandos_client_exited,
 
4581
        .password_is_read=&password_is_read,
 
4582
      }));
 
4583
 
 
4584
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4585
                                   EPOLLIN | EPOLLRDHUP));
 
4586
}
 
4587
 
 
4588
static void
 
4589
test_read_inotify_event_IN_MOVED_TO_badname(__attribute__((unused))
 
4590
                                            test_fixture *fixture,
 
4591
                                            __attribute__((unused))
 
4592
                                            gconstpointer user_data){
 
4593
  __attribute__((cleanup(cleanup_close)))
 
4594
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4595
  g_assert_cmpint(epoll_fd, >=, 0);
 
4596
  const mono_microsecs current_time = 0;
 
4597
 
 
4598
  int pipefds[2];
 
4599
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4600
 
 
4601
  /* "sufficient to read at least one event." - inotify(7) */
 
4602
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4603
                                  + NAME_MAX + 1);
 
4604
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4605
  struct {
 
4606
    struct inotify_event event;
 
4607
    char name_buffer[NAME_MAX + 1];
 
4608
  } ievent_buffer;
 
4609
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4610
 
 
4611
  const char dummy_file_name[] = "ignored.dummy_file_name";
 
4612
  ievent->mask = IN_MOVED_TO;
 
4613
  ievent->len = sizeof(dummy_file_name);
 
4614
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4615
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4616
                              + sizeof(dummy_file_name));
 
4617
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4618
                  ==, ievent_size);
 
4619
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4620
 
 
4621
  bool quit_now = false;
 
4622
  buffer password = {};
 
4623
  bool mandos_client_exited = false;
 
4624
  bool password_is_read = false;
 
4625
  __attribute__((cleanup(cleanup_queue)))
 
4626
    task_queue *queue = create_queue();
 
4627
  g_assert_nonnull(queue);
 
4628
 
 
4629
  task_context task = {
 
4630
    .func=read_inotify_event,
 
4631
    .epoll_fd=epoll_fd,
 
4632
    .fd=pipefds[0],
 
4633
    .quit_now=&quit_now,
 
4634
    .password=&password,
 
4635
    .filename=strdup("/nonexistent"),
 
4636
    .cancelled_filenames = &(string_set){},
 
4637
    .notafter=0,
 
4638
    .current_time=&current_time,
 
4639
    .mandos_client_exited=&mandos_client_exited,
 
4640
    .password_is_read=&password_is_read,
 
4641
  };
 
4642
  task.func(task, queue);
 
4643
  g_assert_false(quit_now);
 
4644
  g_assert_true(queue->next_run == 0);
 
4645
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4646
 
 
4647
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4648
        .func=read_inotify_event,
 
4649
        .epoll_fd=epoll_fd,
 
4650
        .fd=pipefds[0],
 
4651
        .quit_now=&quit_now,
 
4652
        .password=&password,
 
4653
        .filename=task.filename,
 
4654
        .cancelled_filenames=task.cancelled_filenames,
 
4655
        .current_time=&current_time,
 
4656
        .mandos_client_exited=&mandos_client_exited,
 
4657
        .password_is_read=&password_is_read,
 
4658
      }));
 
4659
 
 
4660
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4661
                                   EPOLLIN | EPOLLRDHUP));
 
4662
}
 
4663
 
 
4664
static void
 
4665
test_read_inotify_event_IN_MOVED_FROM_badname(__attribute__((unused))
 
4666
                                              test_fixture *fixture,
 
4667
                                              __attribute__((unused))
 
4668
                                              gconstpointer
 
4669
                                              user_data){
 
4670
  __attribute__((cleanup(cleanup_close)))
 
4671
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4672
  g_assert_cmpint(epoll_fd, >=, 0);
 
4673
  __attribute__((cleanup(string_set_clear)))
 
4674
    string_set cancelled_filenames = {};
 
4675
  const mono_microsecs current_time = 0;
 
4676
 
 
4677
  int pipefds[2];
 
4678
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4679
 
 
4680
  /* "sufficient to read at least one event." - inotify(7) */
 
4681
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4682
                                  + NAME_MAX + 1);
 
4683
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4684
  struct {
 
4685
    struct inotify_event event;
 
4686
    char name_buffer[NAME_MAX + 1];
 
4687
  } ievent_buffer;
 
4688
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4689
 
 
4690
  const char dummy_file_name[] = "ignored.dummy_file_name";
 
4691
  ievent->mask = IN_MOVED_FROM;
 
4692
  ievent->len = sizeof(dummy_file_name);
 
4693
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4694
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4695
                              + sizeof(dummy_file_name));
 
4696
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4697
                  ==, ievent_size);
 
4698
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4699
 
 
4700
  bool quit_now = false;
 
4701
  buffer password = {};
 
4702
  bool mandos_client_exited = false;
 
4703
  bool password_is_read = false;
 
4704
  __attribute__((cleanup(cleanup_queue)))
 
4705
    task_queue *queue = create_queue();
 
4706
  g_assert_nonnull(queue);
 
4707
 
 
4708
  task_context task = {
 
4709
    .func=read_inotify_event,
 
4710
    .epoll_fd=epoll_fd,
 
4711
    .fd=pipefds[0],
 
4712
    .quit_now=&quit_now,
 
4713
    .password=&password,
 
4714
    .filename=strdup("/nonexistent"),
 
4715
    .cancelled_filenames=&cancelled_filenames,
 
4716
    .current_time=&current_time,
 
4717
    .mandos_client_exited=&mandos_client_exited,
 
4718
    .password_is_read=&password_is_read,
 
4719
  };
 
4720
  task.func(task, queue);
 
4721
  g_assert_false(quit_now);
 
4722
  g_assert_true(queue->next_run == 0);
 
4723
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4724
 
 
4725
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4726
        .func=read_inotify_event,
 
4727
        .epoll_fd=epoll_fd,
 
4728
        .fd=pipefds[0],
 
4729
        .quit_now=&quit_now,
 
4730
        .password=&password,
 
4731
        .filename=task.filename,
 
4732
        .cancelled_filenames=&cancelled_filenames,
 
4733
        .current_time=&current_time,
 
4734
        .mandos_client_exited=&mandos_client_exited,
 
4735
        .password_is_read=&password_is_read,
 
4736
      }));
 
4737
 
 
4738
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4739
                                   EPOLLIN | EPOLLRDHUP));
 
4740
 
 
4741
  __attribute__((cleanup(cleanup_string)))
 
4742
    char *filename = NULL;
 
4743
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4744
                           dummy_file_name), >, 0);
 
4745
  g_assert_nonnull(filename);
 
4746
  g_assert_false(string_set_contains(cancelled_filenames, filename));
 
4747
}
 
4748
 
 
4749
static
 
4750
void test_read_inotify_event_IN_DELETE_badname(__attribute__((unused))
 
4751
                                               test_fixture *fixture,
 
4752
                                               __attribute__((unused))
 
4753
                                               gconstpointer
 
4754
                                               user_data){
 
4755
  __attribute__((cleanup(cleanup_close)))
 
4756
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4757
  g_assert_cmpint(epoll_fd, >=, 0);
 
4758
  __attribute__((cleanup(string_set_clear)))
 
4759
    string_set cancelled_filenames = {};
 
4760
  const mono_microsecs current_time = 0;
 
4761
 
 
4762
  int pipefds[2];
 
4763
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
4764
 
 
4765
  /* "sufficient to read at least one event." - inotify(7) */
 
4766
  const size_t ievent_max_size = (sizeof(struct inotify_event)
 
4767
                                  + NAME_MAX + 1);
 
4768
  g_assert_cmpint(ievent_max_size, <=, PIPE_BUF);
 
4769
  struct {
 
4770
    struct inotify_event event;
 
4771
    char name_buffer[NAME_MAX + 1];
 
4772
  } ievent_buffer;
 
4773
  struct inotify_event *const ievent = &ievent_buffer.event;
 
4774
 
 
4775
  const char dummy_file_name[] = "ignored.dummy_file_name";
 
4776
  ievent->mask = IN_DELETE;
 
4777
  ievent->len = sizeof(dummy_file_name);
 
4778
  memcpy(ievent->name, dummy_file_name, sizeof(dummy_file_name));
 
4779
  const size_t ievent_size = (sizeof(struct inotify_event)
 
4780
                              + sizeof(dummy_file_name));
 
4781
  g_assert_cmpint(write(pipefds[1], (char *)ievent, ievent_size),
 
4782
                  ==, ievent_size);
 
4783
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
4784
 
 
4785
  bool quit_now = false;
 
4786
  buffer password = {};
 
4787
  bool mandos_client_exited = false;
 
4788
  bool password_is_read = false;
 
4789
  __attribute__((cleanup(cleanup_queue)))
 
4790
    task_queue *queue = create_queue();
 
4791
  g_assert_nonnull(queue);
 
4792
 
 
4793
  task_context task = {
 
4794
    .func=read_inotify_event,
 
4795
    .epoll_fd=epoll_fd,
 
4796
    .fd=pipefds[0],
 
4797
    .quit_now=&quit_now,
 
4798
    .password=&password,
 
4799
    .filename=strdup("/nonexistent"),
 
4800
    .cancelled_filenames=&cancelled_filenames,
 
4801
    .current_time=&current_time,
 
4802
    .mandos_client_exited=&mandos_client_exited,
 
4803
    .password_is_read=&password_is_read,
 
4804
  };
 
4805
  task.func(task, queue);
 
4806
  g_assert_false(quit_now);
 
4807
  g_assert_true(queue->next_run == 0);
 
4808
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
4809
 
 
4810
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
4811
        .func=read_inotify_event,
 
4812
        .epoll_fd=epoll_fd,
 
4813
        .fd=pipefds[0],
 
4814
        .quit_now=&quit_now,
 
4815
        .password=&password,
 
4816
        .filename=task.filename,
 
4817
        .cancelled_filenames=&cancelled_filenames,
 
4818
        .current_time=&current_time,
 
4819
        .mandos_client_exited=&mandos_client_exited,
 
4820
        .password_is_read=&password_is_read,
 
4821
      }));
 
4822
 
 
4823
  g_assert_true(epoll_set_contains(epoll_fd, pipefds[0],
 
4824
                                   EPOLLIN | EPOLLRDHUP));
 
4825
 
 
4826
  __attribute__((cleanup(cleanup_string)))
 
4827
    char *filename = NULL;
 
4828
  g_assert_cmpint(asprintf(&filename, "%s/%s", task.filename,
 
4829
                           dummy_file_name), >, 0);
 
4830
  g_assert_nonnull(filename);
 
4831
  g_assert_false(string_set_contains(cancelled_filenames, filename));
 
4832
}
 
4833
 
 
4834
static
 
4835
void test_open_and_parse_question_ENOENT(__attribute__((unused))
 
4836
                                         test_fixture *fixture,
 
4837
                                         __attribute__((unused))
 
4838
                                         gconstpointer user_data){
 
4839
  __attribute__((cleanup(cleanup_close)))
 
4840
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4841
  g_assert_cmpint(epoll_fd, >=, 0);
 
4842
  __attribute__((cleanup(string_set_clear)))
 
4843
    string_set cancelled_filenames = {};
 
4844
  bool mandos_client_exited = false;
 
4845
  bool password_is_read = false;
 
4846
  __attribute__((cleanup(cleanup_queue)))
 
4847
    task_queue *queue = create_queue();
 
4848
  g_assert_nonnull(queue);
 
4849
 
 
4850
  char *const filename = strdup("/nonexistent");
 
4851
  g_assert_nonnull(filename);
 
4852
  task_context task = {
 
4853
    .func=open_and_parse_question,
 
4854
    .question_filename=filename,
 
4855
    .epoll_fd=epoll_fd,
 
4856
    .password=(buffer[]){{}},
 
4857
    .filename=filename,
 
4858
    .cancelled_filenames=&cancelled_filenames,
 
4859
    .current_time=(mono_microsecs[]){0},
 
4860
    .mandos_client_exited=&mandos_client_exited,
 
4861
    .password_is_read=&password_is_read,
 
4862
  };
 
4863
  task.func(task, queue);
 
4864
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4865
}
 
4866
 
 
4867
static void test_open_and_parse_question_EIO(__attribute__((unused))
 
4868
                                             test_fixture *fixture,
 
4869
                                             __attribute__((unused))
 
4870
                                             gconstpointer user_data){
 
4871
  __attribute__((cleanup(cleanup_close)))
 
4872
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4873
  g_assert_cmpint(epoll_fd, >=, 0);
 
4874
  __attribute__((cleanup(string_set_clear)))
 
4875
    string_set cancelled_filenames = {};
 
4876
  buffer password = {};
 
4877
  bool mandos_client_exited = false;
 
4878
  bool password_is_read = false;
 
4879
  __attribute__((cleanup(cleanup_queue)))
 
4880
    task_queue *queue = create_queue();
 
4881
  g_assert_nonnull(queue);
 
4882
  const mono_microsecs current_time = 0;
 
4883
 
 
4884
  char *filename = strdup("/proc/self/mem");
 
4885
  task_context task = {
 
4886
    .func=open_and_parse_question,
 
4887
    .question_filename=filename,
 
4888
    .epoll_fd=epoll_fd,
 
4889
    .password=&password,
 
4890
    .filename=filename,
 
4891
    .cancelled_filenames=&cancelled_filenames,
 
4892
    .current_time=&current_time,
 
4893
    .mandos_client_exited=&mandos_client_exited,
 
4894
    .password_is_read=&password_is_read,
 
4895
  };
 
4896
  run_task_with_stderr_to_dev_null(task, queue);
 
4897
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4898
}
 
4899
 
 
4900
static void
 
4901
test_open_and_parse_question_parse_error(__attribute__((unused))
 
4902
                                         test_fixture *fixture,
 
4903
                                         __attribute__((unused))
 
4904
                                         gconstpointer user_data){
 
4905
  __attribute__((cleanup(cleanup_close)))
 
4906
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4907
  g_assert_cmpint(epoll_fd, >=, 0);
 
4908
  __attribute__((cleanup(string_set_clear)))
 
4909
    string_set cancelled_filenames = {};
 
4910
  __attribute__((cleanup(cleanup_queue)))
 
4911
    task_queue *queue = create_queue();
 
4912
  g_assert_nonnull(queue);
 
4913
 
 
4914
  __attribute__((cleanup(cleanup_string)))
 
4915
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
4916
  g_assert_nonnull(tempfilename);
 
4917
  int tempfile = mkostemp(tempfilename, O_CLOEXEC);
 
4918
  g_assert_cmpint(tempfile, >, 0);
 
4919
  const char bad_data[] = "this is bad syntax\n";
 
4920
  g_assert_cmpint(write(tempfile, bad_data, sizeof(bad_data)),
 
4921
                  ==, sizeof(bad_data));
 
4922
  g_assert_cmpint(close(tempfile), ==, 0);
 
4923
 
 
4924
  char *const filename = strdup(tempfilename);
 
4925
  g_assert_nonnull(filename);
 
4926
  task_context task = {
 
4927
    .func=open_and_parse_question,
 
4928
    .question_filename=filename,
 
4929
    .epoll_fd=epoll_fd,
 
4930
    .password=(buffer[]){{}},
 
4931
    .filename=filename,
 
4932
    .cancelled_filenames=&cancelled_filenames,
 
4933
    .current_time=(mono_microsecs[]){0},
 
4934
    .mandos_client_exited=(bool[]){false},
 
4935
    .password_is_read=(bool[]){false},
 
4936
  };
 
4937
  run_task_with_stderr_to_dev_null(task, queue);
 
4938
 
 
4939
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4940
 
 
4941
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
4942
}
 
4943
 
 
4944
static
 
4945
void test_open_and_parse_question_nosocket(__attribute__((unused))
 
4946
                                           test_fixture *fixture,
 
4947
                                           __attribute__((unused))
 
4948
                                           gconstpointer user_data){
 
4949
  __attribute__((cleanup(cleanup_close)))
 
4950
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4951
  g_assert_cmpint(epoll_fd, >=, 0);
 
4952
  __attribute__((cleanup(string_set_clear)))
 
4953
    string_set cancelled_filenames = {};
 
4954
  __attribute__((cleanup(cleanup_queue)))
 
4955
    task_queue *queue = create_queue();
 
4956
  g_assert_nonnull(queue);
 
4957
 
 
4958
  __attribute__((cleanup(cleanup_string)))
 
4959
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
4960
  g_assert_nonnull(tempfilename);
 
4961
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
4962
  g_assert_cmpint(questionfile, >, 0);
 
4963
  FILE *qf = fdopen(questionfile, "w");
 
4964
  g_assert_cmpint(fprintf(qf, "[Ask]\nPID=1\n"), >, 0);
 
4965
  g_assert_cmpint(fclose(qf), ==, 0);
 
4966
 
 
4967
  char *const filename = strdup(tempfilename);
 
4968
  g_assert_nonnull(filename);
 
4969
  task_context task = {
 
4970
    .func=open_and_parse_question,
 
4971
    .question_filename=filename,
 
4972
    .epoll_fd=epoll_fd,
 
4973
    .password=(buffer[]){{}},
 
4974
    .filename=filename,
 
4975
    .cancelled_filenames=&cancelled_filenames,
 
4976
    .current_time=(mono_microsecs[]){0},
 
4977
    .mandos_client_exited=(bool[]){false},
 
4978
    .password_is_read=(bool[]){false},
 
4979
  };
 
4980
  run_task_with_stderr_to_dev_null(task, queue);
 
4981
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
4982
 
 
4983
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
4984
}
 
4985
 
 
4986
static
 
4987
void test_open_and_parse_question_badsocket(__attribute__((unused))
 
4988
                                            test_fixture *fixture,
 
4989
                                            __attribute__((unused))
 
4990
                                            gconstpointer user_data){
 
4991
  __attribute__((cleanup(cleanup_close)))
 
4992
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
4993
  g_assert_cmpint(epoll_fd, >=, 0);
 
4994
  __attribute__((cleanup(string_set_clear)))
 
4995
    string_set cancelled_filenames = {};
 
4996
  __attribute__((cleanup(cleanup_queue)))
 
4997
    task_queue *queue = create_queue();
 
4998
  g_assert_nonnull(queue);
 
4999
 
 
5000
  __attribute__((cleanup(cleanup_string)))
 
5001
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5002
  g_assert_nonnull(tempfilename);
 
5003
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5004
  g_assert_cmpint(questionfile, >, 0);
 
5005
  FILE *qf = fdopen(questionfile, "w");
 
5006
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=\nPID=1\n"), >, 0);
 
5007
  g_assert_cmpint(fclose(qf), ==, 0);
 
5008
 
 
5009
  char *const filename = strdup(tempfilename);
 
5010
  g_assert_nonnull(filename);
 
5011
  task_context task = {
 
5012
    .func=open_and_parse_question,
 
5013
    .question_filename=filename,
 
5014
    .epoll_fd=epoll_fd,
 
5015
    .password=(buffer[]){{}},
 
5016
    .filename=filename,
 
5017
    .cancelled_filenames=&cancelled_filenames,
 
5018
    .current_time=(mono_microsecs[]){0},
 
5019
    .mandos_client_exited=(bool[]){false},
 
5020
    .password_is_read=(bool[]){false},
 
5021
  };
 
5022
  run_task_with_stderr_to_dev_null(task, queue);
 
5023
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5024
 
 
5025
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5026
}
 
5027
 
 
5028
static
 
5029
void test_open_and_parse_question_nopid(__attribute__((unused))
 
5030
                                        test_fixture *fixture,
 
5031
                                        __attribute__((unused))
 
5032
                                        gconstpointer user_data){
 
5033
  __attribute__((cleanup(cleanup_close)))
 
5034
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5035
  g_assert_cmpint(epoll_fd, >=, 0);
 
5036
  __attribute__((cleanup(string_set_clear)))
 
5037
    string_set cancelled_filenames = {};
 
5038
  __attribute__((cleanup(cleanup_queue)))
 
5039
    task_queue *queue = create_queue();
 
5040
  g_assert_nonnull(queue);
 
5041
 
 
5042
  __attribute__((cleanup(cleanup_string)))
 
5043
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5044
  g_assert_nonnull(tempfilename);
 
5045
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5046
  g_assert_cmpint(questionfile, >, 0);
 
5047
  FILE *qf = fdopen(questionfile, "w");
 
5048
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\n"), >, 0);
 
5049
  g_assert_cmpint(fclose(qf), ==, 0);
 
5050
 
 
5051
  char *const filename = strdup(tempfilename);
 
5052
  g_assert_nonnull(filename);
 
5053
  task_context task = {
 
5054
    .func=open_and_parse_question,
 
5055
    .question_filename=filename,
 
5056
    .epoll_fd=epoll_fd,
 
5057
    .password=(buffer[]){{}},
 
5058
    .filename=filename,
 
5059
    .cancelled_filenames=&cancelled_filenames,
 
5060
    .current_time=(mono_microsecs[]){0},
 
5061
    .mandos_client_exited=(bool[]){false},
 
5062
    .password_is_read=(bool[]){false},
 
5063
  };
 
5064
  run_task_with_stderr_to_dev_null(task, queue);
 
5065
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5066
 
 
5067
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5068
}
 
5069
 
 
5070
static
 
5071
void test_open_and_parse_question_badpid(__attribute__((unused))
 
5072
                                         test_fixture *fixture,
 
5073
                                         __attribute__((unused))
 
5074
                                         gconstpointer user_data){
 
5075
  __attribute__((cleanup(cleanup_close)))
 
5076
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5077
  g_assert_cmpint(epoll_fd, >=, 0);
 
5078
  __attribute__((cleanup(string_set_clear)))
 
5079
    string_set cancelled_filenames = {};
 
5080
  __attribute__((cleanup(cleanup_queue)))
 
5081
    task_queue *queue = create_queue();
 
5082
  g_assert_nonnull(queue);
 
5083
 
 
5084
  __attribute__((cleanup(cleanup_string)))
 
5085
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5086
  g_assert_nonnull(tempfilename);
 
5087
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5088
  g_assert_cmpint(questionfile, >, 0);
 
5089
  FILE *qf = fdopen(questionfile, "w");
 
5090
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=\n"),
 
5091
                  >, 0);
 
5092
  g_assert_cmpint(fclose(qf), ==, 0);
 
5093
 
 
5094
  char *const filename = strdup(tempfilename);
 
5095
  g_assert_nonnull(filename);
 
5096
  task_context task = {
 
5097
    .func=open_and_parse_question,
 
5098
    .question_filename=filename,
 
5099
    .epoll_fd=epoll_fd,
 
5100
    .password=(buffer[]){{}},
 
5101
    .filename=filename,
 
5102
    .cancelled_filenames=&cancelled_filenames,
 
5103
    .current_time=(mono_microsecs[]){0},
 
5104
    .mandos_client_exited=(bool[]){false},
 
5105
    .password_is_read=(bool[]){false},
 
5106
  };
 
5107
  run_task_with_stderr_to_dev_null(task, queue);
 
5108
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5109
 
 
5110
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5111
}
 
5112
 
 
5113
static void
 
5114
test_open_and_parse_question_noexist_pid(__attribute__((unused))
 
5115
                                         test_fixture *fixture,
 
5116
                                         __attribute__((unused))
 
5117
                                         gconstpointer user_data){
 
5118
  __attribute__((cleanup(cleanup_close)))
 
5119
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5120
  g_assert_cmpint(epoll_fd, >=, 0);
 
5121
  __attribute__((cleanup(string_set_clear)))
 
5122
    string_set cancelled_filenames = {};
 
5123
  buffer password = {};
 
5124
  bool mandos_client_exited = false;
 
5125
  bool password_is_read = false;
 
5126
  __attribute__((cleanup(cleanup_queue)))
 
5127
    task_queue *queue = create_queue();
 
5128
  g_assert_nonnull(queue);
 
5129
  const mono_microsecs current_time = 0;
 
5130
 
 
5131
  /* Find value of sysctl kernel.pid_max */
 
5132
  uintmax_t pid_max = 0;
 
5133
  FILE *sysctl_pid_max = fopen("/proc/sys/kernel/pid_max", "r");
 
5134
  g_assert_nonnull(sysctl_pid_max);
 
5135
  g_assert_cmpint(fscanf(sysctl_pid_max, "%" PRIuMAX, &pid_max),
 
5136
                  ==, 1);
 
5137
  g_assert_cmpint(fclose(sysctl_pid_max), ==, 0);
 
5138
 
 
5139
  pid_t nonexisting_pid = ((pid_t)pid_max)+1;
 
5140
  g_assert_true(nonexisting_pid > 0); /* Overflow check */
 
5141
 
 
5142
  __attribute__((cleanup(cleanup_string)))
 
5143
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5144
  g_assert_nonnull(tempfilename);
 
5145
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5146
  g_assert_cmpint(questionfile, >, 0);
 
5147
  FILE *qf = fdopen(questionfile, "w");
 
5148
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
 
5149
                          PRIuMAX"\n", (uintmax_t)nonexisting_pid),
 
5150
                  >, 0);
 
5151
  g_assert_cmpint(fclose(qf), ==, 0);
 
5152
 
 
5153
  char *const question_filename = strdup(tempfilename);
 
5154
  g_assert_nonnull(question_filename);
 
5155
  task_context task = {
 
5156
    .func=open_and_parse_question,
 
5157
    .question_filename=question_filename,
 
5158
    .epoll_fd=epoll_fd,
 
5159
    .password=&password,
 
5160
    .filename=question_filename,
 
5161
    .cancelled_filenames=&cancelled_filenames,
 
5162
    .current_time=&current_time,
 
5163
    .mandos_client_exited=&mandos_client_exited,
 
5164
    .password_is_read=&password_is_read,
 
5165
  };
 
5166
  run_task_with_stderr_to_dev_null(task, queue);
 
5167
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5168
 
 
5169
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5170
}
 
5171
 
 
5172
static void
 
5173
test_open_and_parse_question_no_notafter(__attribute__((unused))
 
5174
                                         test_fixture *fixture,
 
5175
                                         __attribute__((unused))
 
5176
                                         gconstpointer user_data){
 
5177
  __attribute__((cleanup(cleanup_close)))
 
5178
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5179
  g_assert_cmpint(epoll_fd, >=, 0);
 
5180
  __attribute__((cleanup(string_set_clear)))
 
5181
    string_set cancelled_filenames = {};
 
5182
  buffer password = {};
 
5183
  bool mandos_client_exited = false;
 
5184
  bool password_is_read = false;
 
5185
  __attribute__((cleanup(cleanup_queue)))
 
5186
    task_queue *queue = create_queue();
 
5187
  g_assert_nonnull(queue);
 
5188
  const mono_microsecs current_time = 0;
 
5189
 
 
5190
  __attribute__((cleanup(cleanup_string)))
 
5191
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5192
  g_assert_nonnull(tempfilename);
 
5193
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5194
  g_assert_cmpint(questionfile, >, 0);
 
5195
  FILE *qf = fdopen(questionfile, "w");
 
5196
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
 
5197
                          PRIuMAX "\n", (uintmax_t)getpid()), >, 0);
 
5198
  g_assert_cmpint(fclose(qf), ==, 0);
 
5199
 
 
5200
  char *const filename = strdup(tempfilename);
 
5201
  g_assert_nonnull(filename);
 
5202
  task_context task = {
 
5203
    .func=open_and_parse_question,
 
5204
    .question_filename=filename,
 
5205
    .epoll_fd=epoll_fd,
 
5206
    .password=&password,
 
5207
    .filename=filename,
 
5208
    .cancelled_filenames=&cancelled_filenames,
 
5209
    .current_time=&current_time,
 
5210
    .mandos_client_exited=&mandos_client_exited,
 
5211
    .password_is_read=&password_is_read,
 
5212
  };
 
5213
  task.func(task, queue);
 
5214
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5215
 
 
5216
  __attribute__((cleanup(cleanup_string)))
 
5217
    char *socket_filename = strdup("/nonexistent");
 
5218
  g_assert_nonnull(socket_filename);
 
5219
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
5220
        .func=connect_question_socket,
 
5221
        .question_filename=tempfilename,
 
5222
        .filename=socket_filename,
 
5223
        .epoll_fd=epoll_fd,
 
5224
        .password=&password,
 
5225
        .current_time=&current_time,
 
5226
        .mandos_client_exited=&mandos_client_exited,
 
5227
        .password_is_read=&password_is_read,
 
5228
      }));
 
5229
 
 
5230
  g_assert_true(queue->next_run != 0);
 
5231
 
 
5232
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5233
}
 
5234
 
 
5235
static void
 
5236
test_open_and_parse_question_bad_notafter(__attribute__((unused))
 
5237
                                          test_fixture *fixture,
 
5238
                                          __attribute__((unused))
 
5239
                                          gconstpointer user_data){
 
5240
  __attribute__((cleanup(cleanup_close)))
 
5241
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5242
  g_assert_cmpint(epoll_fd, >=, 0);
 
5243
  __attribute__((cleanup(string_set_clear)))
 
5244
    string_set cancelled_filenames = {};
 
5245
  buffer password = {};
 
5246
  bool mandos_client_exited = false;
 
5247
  bool password_is_read = false;
 
5248
  __attribute__((cleanup(cleanup_queue)))
 
5249
    task_queue *queue = create_queue();
 
5250
  g_assert_nonnull(queue);
 
5251
  const mono_microsecs current_time = 0;
 
5252
 
 
5253
  __attribute__((cleanup(cleanup_string)))
 
5254
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5255
  g_assert_nonnull(tempfilename);
 
5256
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5257
  g_assert_cmpint(questionfile, >, 0);
 
5258
  FILE *qf = fdopen(questionfile, "w");
 
5259
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
 
5260
                          PRIuMAX "\nNotAfter=\n",
 
5261
                          (uintmax_t)getpid()), >, 0);
 
5262
  g_assert_cmpint(fclose(qf), ==, 0);
 
5263
 
 
5264
  char *const filename = strdup(tempfilename);
 
5265
  g_assert_nonnull(filename);
 
5266
  task_context task = {
 
5267
    .func=open_and_parse_question,
 
5268
    .question_filename=filename,
 
5269
    .epoll_fd=epoll_fd,
 
5270
    .password=&password,
 
5271
    .filename=filename,
 
5272
    .cancelled_filenames=&cancelled_filenames,
 
5273
    .current_time=&current_time,
 
5274
    .mandos_client_exited=&mandos_client_exited,
 
5275
    .password_is_read=&password_is_read,
 
5276
  };
 
5277
  run_task_with_stderr_to_dev_null(task, queue);
 
5278
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5279
 
 
5280
  __attribute__((cleanup(cleanup_string)))
 
5281
    char *socket_filename = strdup("/nonexistent");
 
5282
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
5283
        .func=connect_question_socket,
 
5284
        .question_filename=tempfilename,
 
5285
        .filename=socket_filename,
 
5286
        .epoll_fd=epoll_fd,
 
5287
        .password=&password,
 
5288
        .current_time=&current_time,
 
5289
        .mandos_client_exited=&mandos_client_exited,
 
5290
        .password_is_read=&password_is_read,
 
5291
      }));
 
5292
  g_assert_true(queue->next_run != 0);
 
5293
 
 
5294
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5295
}
 
5296
 
 
5297
static
 
5298
void assert_open_and_parse_question_with_notafter(const mono_microsecs
 
5299
                                                  current_time,
 
5300
                                                  const mono_microsecs
 
5301
                                                  notafter,
 
5302
                                                  const mono_microsecs
 
5303
                                                  next_queue_run){
 
5304
  __attribute__((cleanup(cleanup_close)))
 
5305
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5306
  g_assert_cmpint(epoll_fd, >=, 0);
 
5307
  __attribute__((cleanup(string_set_clear)))
 
5308
    string_set cancelled_filenames = {};
 
5309
  buffer password = {};
 
5310
  bool mandos_client_exited = false;
 
5311
  bool password_is_read = false;
 
5312
  __attribute__((cleanup(cleanup_queue)))
 
5313
    task_queue *queue = create_queue();
 
5314
  g_assert_nonnull(queue);
 
5315
  queue->next_run = next_queue_run;
 
5316
 
 
5317
  __attribute__((cleanup(cleanup_string)))
 
5318
    char *tempfilename = strdup("/tmp/mandosXXXXXX");
 
5319
  g_assert_nonnull(tempfilename);
 
5320
  int questionfile = mkostemp(tempfilename, O_CLOEXEC);
 
5321
  g_assert_cmpint(questionfile, >, 0);
 
5322
  FILE *qf = fdopen(questionfile, "w");
 
5323
  g_assert_cmpint(fprintf(qf, "[Ask]\nSocket=/nonexistent\nPID=%"
 
5324
                          PRIuMAX "\nNotAfter=%" PRIuMAX "\n",
 
5325
                          (uintmax_t)getpid(), notafter), >, 0);
 
5326
  g_assert_cmpint(fclose(qf), ==, 0);
 
5327
 
 
5328
  char *const filename = strdup(tempfilename);
 
5329
  g_assert_nonnull(filename);
 
5330
  task_context task = {
 
5331
    .func=open_and_parse_question,
 
5332
    .question_filename=filename,
 
5333
    .epoll_fd=epoll_fd,
 
5334
    .password=&password,
 
5335
    .filename=filename,
 
5336
    .cancelled_filenames=&cancelled_filenames,
 
5337
    .current_time=&current_time,
 
5338
    .mandos_client_exited=&mandos_client_exited,
 
5339
    .password_is_read=&password_is_read,
 
5340
  };
 
5341
  task.func(task, queue);
 
5342
 
 
5343
  if(queue->length >= 1){
 
5344
    __attribute__((cleanup(cleanup_string)))
 
5345
      char *socket_filename = strdup("/nonexistent");
 
5346
    g_assert_nonnull(find_matching_task(queue, (task_context){
 
5347
          .func=connect_question_socket,
 
5348
          .filename=socket_filename,
 
5349
          .epoll_fd=epoll_fd,
 
5350
          .password=&password,
 
5351
          .current_time=&current_time,
 
5352
          .cancelled_filenames=&cancelled_filenames,
 
5353
          .mandos_client_exited=&mandos_client_exited,
 
5354
          .password_is_read=&password_is_read,
 
5355
        }));
 
5356
    g_assert_true(queue->next_run != 0);
 
5357
  }
 
5358
 
 
5359
  if(notafter == 0){
 
5360
    g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5361
  } else if(current_time >= notafter) {
 
5362
    g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5363
  } else {
 
5364
    g_assert_nonnull(find_matching_task(queue, (task_context){
 
5365
          .func=cancel_old_question,
 
5366
          .question_filename=tempfilename,
 
5367
          .filename=tempfilename,
 
5368
          .notafter=notafter,
 
5369
          .cancelled_filenames=&cancelled_filenames,
 
5370
          .current_time=&current_time,
 
5371
        }));
 
5372
  }
 
5373
  g_assert_true(queue->next_run == 1);
 
5374
 
 
5375
  g_assert_cmpint(unlink(tempfilename), ==, 0);
 
5376
}
 
5377
 
 
5378
static void
 
5379
test_open_and_parse_question_notafter_0(__attribute__((unused))
 
5380
                                        test_fixture *fixture,
 
5381
                                        __attribute__((unused))
 
5382
                                        gconstpointer user_data){
 
5383
  /* current_time, notafter, next_queue_run */
 
5384
  assert_open_and_parse_question_with_notafter(0, 0, 0);
 
5385
}
 
5386
 
 
5387
static void
 
5388
test_open_and_parse_question_notafter_1(__attribute__((unused))
 
5389
                                        test_fixture *fixture,
 
5390
                                        __attribute__((unused))
 
5391
                                        gconstpointer user_data){
 
5392
  /* current_time, notafter, next_queue_run */
 
5393
  assert_open_and_parse_question_with_notafter(0, 1, 0);
 
5394
}
 
5395
 
 
5396
static void
 
5397
test_open_and_parse_question_notafter_1_1(__attribute__((unused))
 
5398
                                          test_fixture *fixture,
 
5399
                                          __attribute__((unused))
 
5400
                                          gconstpointer user_data){
 
5401
  /* current_time, notafter, next_queue_run */
 
5402
  assert_open_and_parse_question_with_notafter(0, 1, 1);
 
5403
}
 
5404
 
 
5405
static void
 
5406
test_open_and_parse_question_notafter_1_2(__attribute__((unused))
 
5407
                                          test_fixture *fixture,
 
5408
                                          __attribute__((unused))
 
5409
                                          gconstpointer user_data){
 
5410
  /* current_time, notafter, next_queue_run */
 
5411
  assert_open_and_parse_question_with_notafter(0, 1, 2);
 
5412
}
 
5413
 
 
5414
static void
 
5415
test_open_and_parse_question_equal_notafter(__attribute__((unused))
 
5416
                                            test_fixture *fixture,
 
5417
                                            __attribute__((unused))
 
5418
                                            gconstpointer user_data){
 
5419
  /* current_time, notafter, next_queue_run */
 
5420
  assert_open_and_parse_question_with_notafter(1, 1, 0);
 
5421
}
 
5422
 
 
5423
static void
 
5424
test_open_and_parse_question_late_notafter(__attribute__((unused))
 
5425
                                           test_fixture *fixture,
 
5426
                                           __attribute__((unused))
 
5427
                                           gconstpointer user_data){
 
5428
  /* current_time, notafter, next_queue_run */
 
5429
  assert_open_and_parse_question_with_notafter(2, 1, 0);
 
5430
}
 
5431
 
 
5432
static void assert_cancel_old_question_param(const mono_microsecs
 
5433
                                             next_queue_run,
 
5434
                                             const mono_microsecs
 
5435
                                             notafter,
 
5436
                                             const mono_microsecs
 
5437
                                             current_time,
 
5438
                                             const mono_microsecs
 
5439
                                             next_set_to){
 
5440
  __attribute__((cleanup(string_set_clear)))
 
5441
    string_set cancelled_filenames = {};
 
5442
  __attribute__((cleanup(cleanup_queue)))
 
5443
    task_queue *queue = create_queue();
 
5444
  g_assert_nonnull(queue);
 
5445
  queue->next_run = next_queue_run;
 
5446
 
 
5447
  char *const question_filename = strdup("/nonexistent");
 
5448
  g_assert_nonnull(question_filename);
 
5449
  task_context task = {
 
5450
    .func=cancel_old_question,
 
5451
    .question_filename=question_filename,
 
5452
    .filename=question_filename,
 
5453
    .notafter=notafter,
 
5454
    .cancelled_filenames=&cancelled_filenames,
 
5455
    .current_time=&current_time,
 
5456
  };
 
5457
  task.func(task, queue);
 
5458
 
 
5459
  if(current_time >= notafter){
 
5460
    g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5461
    g_assert_true(string_set_contains(cancelled_filenames,
 
5462
                                      "/nonexistent"));
 
5463
  } else {
 
5464
    g_assert_nonnull(find_matching_task(queue, (task_context){
 
5465
          .func=cancel_old_question,
 
5466
          .question_filename=question_filename,
 
5467
          .filename=question_filename,
 
5468
          .notafter=notafter,
 
5469
          .cancelled_filenames=&cancelled_filenames,
 
5470
          .current_time=&current_time,
 
5471
        }));
 
5472
 
 
5473
    g_assert_false(string_set_contains(cancelled_filenames,
 
5474
                                       question_filename));
 
5475
  }
 
5476
  g_assert_cmpuint((unsigned int)queue->next_run, ==,
 
5477
                   (unsigned int)next_set_to);
 
5478
}
 
5479
 
 
5480
static void test_cancel_old_question_0_1_2(__attribute__((unused))
 
5481
                                           test_fixture *fixture,
 
5482
                                           __attribute__((unused))
 
5483
                                           gconstpointer user_data){
 
5484
  /* next_queue_run unset,
 
5485
     cancellation should happen because time has come,
 
5486
     next_queue_run should be unchanged */
 
5487
  /* next_queue_run, notafter, current_time, next_set_to */
 
5488
  assert_cancel_old_question_param(0, 1, 2, 0);
 
5489
}
 
5490
 
 
5491
static void test_cancel_old_question_0_2_1(__attribute__((unused))
 
5492
                                           test_fixture *fixture,
 
5493
                                           __attribute__((unused))
 
5494
                                           gconstpointer user_data){
 
5495
  /* If next_queue_run is 0, meaning unset, and notafter is 2,
 
5496
     and current_time is not yet notafter or greater,
 
5497
     update value of next_queue_run to value of notafter */
 
5498
  /* next_queue_run, notafter, current_time, next_set_to */
 
5499
  assert_cancel_old_question_param(0, 2, 1, 2);
 
5500
}
 
5501
 
 
5502
static void test_cancel_old_question_1_2_3(__attribute__((unused))
 
5503
                                           test_fixture *fixture,
 
5504
                                           __attribute__((unused))
 
5505
                                           gconstpointer user_data){
 
5506
  /* next_queue_run 1,
 
5507
     cancellation should happen because time has come,
 
5508
     next_queue_run should be unchanged */
 
5509
  /* next_queue_run, notafter, current_time, next_set_to */
 
5510
  assert_cancel_old_question_param(1, 2, 3, 1);
 
5511
}
 
5512
 
 
5513
static void test_cancel_old_question_1_3_2(__attribute__((unused))
 
5514
                                           test_fixture *fixture,
 
5515
                                           __attribute__((unused))
 
5516
                                           gconstpointer user_data){
 
5517
  /* If next_queue_run is set,
 
5518
     and current_time is not yet notafter or greater,
 
5519
     and notafter is larger than next_queue_run
 
5520
     next_queue_run should be unchanged */
 
5521
  /* next_queue_run, notafter, current_time, next_set_to */
 
5522
  assert_cancel_old_question_param(1, 3, 2, 1);
 
5523
}
 
5524
 
 
5525
static void test_cancel_old_question_2_1_3(__attribute__((unused))
 
5526
                                           test_fixture *fixture,
 
5527
                                           __attribute__((unused))
 
5528
                                           gconstpointer user_data){
 
5529
  /* next_queue_run 2,
 
5530
     cancellation should happen because time has come,
 
5531
     next_queue_run should be unchanged */
 
5532
  /* next_queue_run, notafter, current_time, next_set_to */
 
5533
  assert_cancel_old_question_param(2, 1, 3, 2);
 
5534
}
 
5535
 
 
5536
static void test_cancel_old_question_2_3_1(__attribute__((unused))
 
5537
                                           test_fixture *fixture,
 
5538
                                           __attribute__((unused))
 
5539
                                           gconstpointer user_data){
 
5540
  /* If next_queue_run is set,
 
5541
     and current_time is not yet notafter or greater,
 
5542
     and notafter is larger than next_queue_run
 
5543
     next_queue_run should be unchanged */
 
5544
  /* next_queue_run, notafter, current_time, next_set_to */
 
5545
  assert_cancel_old_question_param(2, 3, 1, 2);
 
5546
}
 
5547
 
 
5548
static void test_cancel_old_question_3_1_2(__attribute__((unused))
 
5549
                                           test_fixture *fixture,
 
5550
                                           __attribute__((unused))
 
5551
                                           gconstpointer user_data){
 
5552
  /* next_queue_run 3,
 
5553
     cancellation should happen because time has come,
 
5554
     next_queue_run should be unchanged */
 
5555
  /* next_queue_run, notafter, current_time, next_set_to */
 
5556
  assert_cancel_old_question_param(3, 1, 2, 3);
 
5557
}
 
5558
 
 
5559
static void test_cancel_old_question_3_2_1(__attribute__((unused))
 
5560
                                           test_fixture *fixture,
 
5561
                                           __attribute__((unused))
 
5562
                                           gconstpointer user_data){
 
5563
  /* If next_queue_run is set,
 
5564
     and current_time is not yet notafter or greater,
 
5565
     and notafter is smaller than next_queue_run
 
5566
     update value of next_queue_run to value of notafter */
 
5567
  /* next_queue_run, notafter, current_time, next_set_to */
 
5568
  assert_cancel_old_question_param(3, 2, 1, 2);
 
5569
}
 
5570
 
 
5571
static void
 
5572
test_connect_question_socket_name_too_long(__attribute__((unused))
 
5573
                                           test_fixture *fixture,
 
5574
                                           __attribute__((unused))
 
5575
                                           gconstpointer user_data){
 
5576
  __attribute__((cleanup(cleanup_close)))
 
5577
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5578
  g_assert_cmpint(epoll_fd, >=, 0);
 
5579
  const char question_filename[] = "/nonexistent/question";
 
5580
  __attribute__((cleanup(string_set_clear)))
 
5581
    string_set cancelled_filenames = {};
 
5582
  __attribute__((cleanup(cleanup_queue)))
 
5583
    task_queue *queue = create_queue();
 
5584
  g_assert_nonnull(queue);
 
5585
  __attribute__((cleanup(cleanup_string)))
 
5586
    char *tempdir = make_temporary_directory();
 
5587
  g_assert_nonnull(tempdir);
 
5588
  struct sockaddr_un unix_socket = { .sun_family=AF_LOCAL };
 
5589
  char socket_name[sizeof(unix_socket.sun_path)];
 
5590
  memset(socket_name, 'x', sizeof(socket_name));
 
5591
  socket_name[sizeof(socket_name)-1] = '\0';
 
5592
  char *filename = NULL;
 
5593
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
 
5594
                  >, 0);
 
5595
  g_assert_nonnull(filename);
 
5596
 
 
5597
  task_context task = {
 
5598
    .func=connect_question_socket,
 
5599
    .question_filename=strdup(question_filename),
 
5600
    .epoll_fd=epoll_fd,
 
5601
    .password=(buffer[]){{}},
 
5602
    .filename=filename,
 
5603
    .cancelled_filenames=&cancelled_filenames,
 
5604
    .mandos_client_exited=(bool[]){false},
 
5605
    .password_is_read=(bool[]){false},
 
5606
    .current_time=(mono_microsecs[]){0},
 
5607
  };
 
5608
  g_assert_nonnull(task.question_filename);
 
5609
  run_task_with_stderr_to_dev_null(task, queue);
 
5610
 
 
5611
  g_assert_true(string_set_contains(cancelled_filenames,
 
5612
                                    question_filename));
 
5613
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
5614
  g_assert_true(queue->next_run == 0);
 
5615
 
 
5616
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
5617
}
 
5618
 
 
5619
static
 
5620
void test_connect_question_socket_connect_fail(__attribute__((unused))
 
5621
                                               test_fixture *fixture,
 
5622
                                               __attribute__((unused))
 
5623
                                               gconstpointer
 
5624
                                               user_data){
 
5625
  __attribute__((cleanup(cleanup_close)))
 
5626
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5627
  g_assert_cmpint(epoll_fd, >=, 0);
 
5628
  const char question_filename[] = "/nonexistent/question";
 
5629
  __attribute__((cleanup(string_set_clear)))
 
5630
    string_set cancelled_filenames = {};
 
5631
  const mono_microsecs current_time = 3;
 
5632
  __attribute__((cleanup(cleanup_queue)))
 
5633
    task_queue *queue = create_queue();
 
5634
  g_assert_nonnull(queue);
 
5635
  __attribute__((cleanup(cleanup_string)))
 
5636
    char *tempdir = make_temporary_directory();
 
5637
  g_assert_nonnull(tempdir);
 
5638
  char socket_name[] = "nonexistent";
 
5639
  char *filename = NULL;
 
5640
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
 
5641
                  >, 0);
 
5642
  g_assert_nonnull(filename);
 
5643
 
 
5644
  task_context task = {
 
5645
    .func=connect_question_socket,
 
5646
    .question_filename=strdup(question_filename),
 
5647
    .epoll_fd=epoll_fd,
 
5648
    .password=(buffer[]){{}},
 
5649
    .filename=filename,
 
5650
    .cancelled_filenames=&cancelled_filenames,
 
5651
    .mandos_client_exited=(bool[]){false},
 
5652
    .password_is_read=(bool[]){false},
 
5653
    .current_time=&current_time,
 
5654
  };
 
5655
  g_assert_nonnull(task.question_filename);
 
5656
  run_task_with_stderr_to_dev_null(task, queue);
 
5657
 
 
5658
  g_assert_nonnull(find_matching_task(queue, task));
 
5659
 
 
5660
  g_assert_false(string_set_contains(cancelled_filenames,
 
5661
                                     question_filename));
 
5662
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5663
  g_assert_true(queue->next_run == 1000000 + current_time);
 
5664
 
 
5665
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
5666
}
 
5667
 
 
5668
static
 
5669
void test_connect_question_socket_bad_epoll(__attribute__((unused))
 
5670
                                            test_fixture *fixture,
 
5671
                                            __attribute__((unused))
 
5672
                                            gconstpointer user_data){
 
5673
  __attribute__((cleanup(cleanup_close)))
 
5674
    const int epoll_fd = open("/dev/null",
 
5675
                              O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
5676
  __attribute__((cleanup(cleanup_string)))
 
5677
    char *const question_filename = strdup("/nonexistent/question");
 
5678
  g_assert_nonnull(question_filename);
 
5679
  __attribute__((cleanup(string_set_clear)))
 
5680
    string_set cancelled_filenames = {};
 
5681
  const mono_microsecs current_time = 5;
 
5682
  __attribute__((cleanup(cleanup_queue)))
 
5683
    task_queue *queue = create_queue();
 
5684
  g_assert_nonnull(queue);
 
5685
  __attribute__((cleanup(cleanup_string)))
 
5686
    char *tempdir = make_temporary_directory();
 
5687
  g_assert_nonnull(tempdir);
 
5688
  __attribute__((cleanup(cleanup_close)))
 
5689
    const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM
 
5690
                               | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
 
5691
  g_assert_cmpint(sock_fd, >=, 0);
 
5692
  struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };
 
5693
  const char socket_name[] = "socket_name";
 
5694
  __attribute__((cleanup(cleanup_string)))
 
5695
    char *filename = NULL;
 
5696
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
 
5697
                  >, 0);
 
5698
  g_assert_nonnull(filename);
 
5699
  g_assert_cmpint((int)strlen(filename), <,
 
5700
                  (int)sizeof(sock_name.sun_path));
 
5701
  strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));
 
5702
  sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';
 
5703
  g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,
 
5704
                            (socklen_t)SUN_LEN(&sock_name)), >=, 0);
 
5705
  task_context task = {
 
5706
    .func=connect_question_socket,
 
5707
    .question_filename=strdup(question_filename),
 
5708
    .epoll_fd=epoll_fd,
 
5709
    .password=(buffer[]){{}},
 
5710
    .filename=strdup(filename),
 
5711
    .cancelled_filenames=&cancelled_filenames,
 
5712
    .mandos_client_exited=(bool[]){false},
 
5713
    .password_is_read=(bool[]){false},
 
5714
    .current_time=&current_time,
 
5715
  };
 
5716
  g_assert_nonnull(task.question_filename);
 
5717
  run_task_with_stderr_to_dev_null(task, queue);
 
5718
 
 
5719
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5720
  const task_context *const added_task
 
5721
    = find_matching_task(queue, task);
 
5722
  g_assert_nonnull(added_task);
 
5723
  g_assert_true(queue->next_run == 1000000 + current_time);
 
5724
 
 
5725
  g_assert_cmpint(unlink(filename), ==, 0);
 
5726
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
5727
}
 
5728
 
 
5729
static
 
5730
void test_connect_question_socket_usable(__attribute__((unused))
 
5731
                                         test_fixture *fixture,
 
5732
                                         __attribute__((unused))
 
5733
                                         gconstpointer user_data){
 
5734
  __attribute__((cleanup(cleanup_close)))
 
5735
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5736
  g_assert_cmpint(epoll_fd, >=, 0);
 
5737
  __attribute__((cleanup(cleanup_string)))
 
5738
    char *const question_filename = strdup("/nonexistent/question");
 
5739
  g_assert_nonnull(question_filename);
 
5740
  __attribute__((cleanup(string_set_clear)))
 
5741
    string_set cancelled_filenames = {};
 
5742
  buffer password = {};
 
5743
  bool mandos_client_exited = false;
 
5744
  bool password_is_read = false;
 
5745
  const mono_microsecs current_time = 0;
 
5746
  __attribute__((cleanup(cleanup_queue)))
 
5747
    task_queue *queue = create_queue();
 
5748
  g_assert_nonnull(queue);
 
5749
  __attribute__((cleanup(cleanup_string)))
 
5750
    char *tempdir = make_temporary_directory();
 
5751
  g_assert_nonnull(tempdir);
 
5752
  __attribute__((cleanup(cleanup_close)))
 
5753
    const int sock_fd = socket(PF_LOCAL, SOCK_DGRAM
 
5754
                               | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
 
5755
  g_assert_cmpint(sock_fd, >=, 0);
 
5756
  struct sockaddr_un sock_name = { .sun_family=AF_LOCAL };
 
5757
  const char socket_name[] = "socket_name";
 
5758
  __attribute__((cleanup(cleanup_string)))
 
5759
    char *filename = NULL;
 
5760
  g_assert_cmpint(asprintf(&filename, "%s/%s", tempdir, socket_name),
 
5761
                  >, 0);
 
5762
  g_assert_nonnull(filename);
 
5763
  g_assert_cmpint((int)strlen(filename), <,
 
5764
                  (int)sizeof(sock_name.sun_path));
 
5765
  strncpy(sock_name.sun_path, filename, sizeof(sock_name.sun_path));
 
5766
  sock_name.sun_path[sizeof(sock_name.sun_path)-1] = '\0';
 
5767
  g_assert_cmpint((int)bind(sock_fd, (struct sockaddr *)&sock_name,
 
5768
                            (socklen_t)SUN_LEN(&sock_name)), >=, 0);
 
5769
  task_context task = {
 
5770
    .func=connect_question_socket,
 
5771
    .question_filename=strdup(question_filename),
 
5772
    .epoll_fd=epoll_fd,
 
5773
    .password=&password,
 
5774
    .filename=strdup(filename),
 
5775
    .cancelled_filenames=&cancelled_filenames,
 
5776
    .mandos_client_exited=&mandos_client_exited,
 
5777
    .password_is_read=&password_is_read,
 
5778
    .current_time=&current_time,
 
5779
  };
 
5780
  g_assert_nonnull(task.question_filename);
 
5781
  task.func(task, queue);
 
5782
 
 
5783
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5784
  const task_context *const added_task
 
5785
    = find_matching_task(queue, (task_context){
 
5786
        .func=send_password_to_socket,
 
5787
        .question_filename=question_filename,
 
5788
        .filename=filename,
 
5789
        .epoll_fd=epoll_fd,
 
5790
        .password=&password,
 
5791
        .cancelled_filenames=&cancelled_filenames,
 
5792
        .mandos_client_exited=&mandos_client_exited,
 
5793
        .password_is_read=&password_is_read,
 
5794
        .current_time=&current_time,
 
5795
      });
 
5796
  g_assert_nonnull(added_task);
 
5797
  g_assert_cmpint(added_task->fd, >, 0);
 
5798
 
 
5799
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
 
5800
                                   EPOLLOUT));
 
5801
 
 
5802
  const int fd = added_task->fd;
 
5803
  g_assert_cmpint(fd, >, 0);
 
5804
  g_assert_true(fd_has_cloexec_and_nonblock(fd));
 
5805
 
 
5806
  /* write to fd */
 
5807
  char write_data[PIPE_BUF];
 
5808
  {
 
5809
    /* Construct test password buffer */
 
5810
    /* Start with + since that is what the real procotol uses */
 
5811
    write_data[0] = '+';
 
5812
    /* Set a special character at string end just to mark the end */
 
5813
    write_data[sizeof(write_data)-2] = 'y';
 
5814
    /* Set NUL at buffer end, as suggested by the protocol */
 
5815
    write_data[sizeof(write_data)-1] = '\0';
 
5816
    /* Fill rest of password with 'x' */
 
5817
    memset(write_data+1, 'x', sizeof(write_data)-3);
 
5818
    g_assert_cmpint((int)send(fd, write_data, sizeof(write_data),
 
5819
                              MSG_NOSIGNAL), ==, sizeof(write_data));
 
5820
  }
 
5821
 
 
5822
  /* read from sock_fd */
 
5823
  char read_data[sizeof(write_data)];
 
5824
  g_assert_cmpint((int)read(sock_fd, read_data, sizeof(read_data)),
 
5825
                  ==, sizeof(read_data));
 
5826
 
 
5827
  g_assert_true(memcmp(write_data, read_data, sizeof(write_data))
 
5828
                == 0);
 
5829
 
 
5830
  /* writing to sock_fd should fail */
 
5831
  g_assert_cmpint(send(sock_fd, write_data, sizeof(write_data),
 
5832
                       MSG_NOSIGNAL), <, 0);
 
5833
 
 
5834
  /* reading from fd should fail */
 
5835
  g_assert_cmpint((int)recv(fd, read_data, sizeof(read_data),
 
5836
                            MSG_NOSIGNAL), <, 0);
 
5837
 
 
5838
  g_assert_cmpint(unlink(filename), ==, 0);
 
5839
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
5840
}
 
5841
 
 
5842
static void
 
5843
test_send_password_to_socket_client_not_exited(__attribute__((unused))
 
5844
                                               test_fixture *fixture,
 
5845
                                               __attribute__((unused))
 
5846
                                               gconstpointer
 
5847
                                               user_data){
 
5848
  __attribute__((cleanup(cleanup_close)))
 
5849
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5850
  g_assert_cmpint(epoll_fd, >=, 0);
 
5851
  __attribute__((cleanup(cleanup_string)))
 
5852
    char *const question_filename = strdup("/nonexistent/question");
 
5853
  g_assert_nonnull(question_filename);
 
5854
  __attribute__((cleanup(cleanup_string)))
 
5855
    char *const filename = strdup("/nonexistent/socket");
 
5856
  g_assert_nonnull(filename);
 
5857
  __attribute__((cleanup(string_set_clear)))
 
5858
    string_set cancelled_filenames = {};
 
5859
  buffer password = {};
 
5860
  bool password_is_read = true;
 
5861
  __attribute__((cleanup(cleanup_queue)))
 
5862
    task_queue *queue = create_queue();
 
5863
  g_assert_nonnull(queue);
 
5864
  int socketfds[2];
 
5865
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
5866
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
5867
                             socketfds), ==, 0);
 
5868
  __attribute__((cleanup(cleanup_close)))
 
5869
    const int read_socket = socketfds[0];
 
5870
  const int write_socket = socketfds[1];
 
5871
  task_context task = {
 
5872
    .func=send_password_to_socket,
 
5873
    .question_filename=strdup(question_filename),
 
5874
    .filename=strdup(filename),
 
5875
    .epoll_fd=epoll_fd,
 
5876
    .fd=write_socket,
 
5877
    .password=&password,
 
5878
    .cancelled_filenames=&cancelled_filenames,
 
5879
    .mandos_client_exited=(bool[]){false},
 
5880
    .password_is_read=&password_is_read,
 
5881
    .current_time=(mono_microsecs[]){0},
 
5882
  };
 
5883
  g_assert_nonnull(task.question_filename);
 
5884
 
 
5885
  task.func(task, queue);
 
5886
 
 
5887
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5888
 
 
5889
  const task_context *const added_task
 
5890
    = find_matching_task(queue, task);
 
5891
  g_assert_nonnull(added_task);
 
5892
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
 
5893
  g_assert_true(password_is_read);
 
5894
 
 
5895
  g_assert_cmpint(added_task->fd, >, 0);
 
5896
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
 
5897
                                   EPOLLOUT));
 
5898
}
 
5899
 
 
5900
static void
 
5901
test_send_password_to_socket_password_not_read(__attribute__((unused))
 
5902
                                               test_fixture *fixture,
 
5903
                                               __attribute__((unused))
 
5904
                                               gconstpointer
 
5905
                                               user_data){
 
5906
  __attribute__((cleanup(cleanup_close)))
 
5907
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5908
  g_assert_cmpint(epoll_fd, >=, 0);
 
5909
  __attribute__((cleanup(cleanup_string)))
 
5910
    char *const question_filename = strdup("/nonexistent/question");
 
5911
  g_assert_nonnull(question_filename);
 
5912
  __attribute__((cleanup(cleanup_string)))
 
5913
    char *const filename = strdup("/nonexistent/socket");
 
5914
  __attribute__((cleanup(string_set_clear)))
 
5915
    string_set cancelled_filenames = {};
 
5916
  buffer password = {};
 
5917
  __attribute__((cleanup(cleanup_queue)))
 
5918
    task_queue *queue = create_queue();
 
5919
  g_assert_nonnull(queue);
 
5920
  int socketfds[2];
 
5921
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
5922
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
5923
                             socketfds), ==, 0);
 
5924
  __attribute__((cleanup(cleanup_close)))
 
5925
    const int read_socket = socketfds[0];
 
5926
  const int write_socket = socketfds[1];
 
5927
  task_context task = {
 
5928
    .func=send_password_to_socket,
 
5929
    .question_filename=strdup(question_filename),
 
5930
    .filename=strdup(filename),
 
5931
    .epoll_fd=epoll_fd,
 
5932
    .fd=write_socket,
 
5933
    .password=&password,
 
5934
    .cancelled_filenames=&cancelled_filenames,
 
5935
    .mandos_client_exited=(bool[]){false},
 
5936
    .password_is_read=(bool[]){false},
 
5937
    .current_time=(mono_microsecs[]){0},
 
5938
  };
 
5939
  g_assert_nonnull(task.question_filename);
 
5940
 
 
5941
  task.func(task, queue);
 
5942
 
 
5943
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
5944
 
 
5945
  const task_context *const added_task = find_matching_task(queue,
 
5946
                                                            task);
 
5947
  g_assert_nonnull(added_task);
 
5948
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
 
5949
  g_assert_true(queue->next_run == 0);
 
5950
 
 
5951
  g_assert_cmpint(added_task->fd, >, 0);
 
5952
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
 
5953
                                   EPOLLOUT));
 
5954
}
 
5955
 
 
5956
static
 
5957
void test_send_password_to_socket_EMSGSIZE(__attribute__((unused))
 
5958
                                           test_fixture *fixture,
 
5959
                                           __attribute__((unused))
 
5960
                                           gconstpointer user_data){
 
5961
#ifndef __amd64__
 
5962
  g_test_skip("Skipping EMSGSIZE test on non-AMD64 platform");
 
5963
#else
 
5964
  __attribute__((cleanup(cleanup_close)))
 
5965
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
5966
  g_assert_cmpint(epoll_fd, >=, 0);
 
5967
  const char question_filename[] = "/nonexistent/question";
 
5968
  char *const filename = strdup("/nonexistent/socket");
 
5969
  __attribute__((cleanup(string_set_clear)))
 
5970
    string_set cancelled_filenames = {};
 
5971
  const size_t oversized = 1024*1024; /* Limit seems to be 212960 */
 
5972
  __attribute__((cleanup(cleanup_buffer)))
 
5973
    buffer password = {
 
5974
    .data=malloc(oversized),
 
5975
    .length=oversized,
 
5976
    .allocated=oversized,
 
5977
  };
 
5978
  g_assert_nonnull(password.data);
 
5979
  if(mlock(password.data, password.allocated) != 0){
 
5980
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
5981
  }
 
5982
  /* Construct test password buffer */
 
5983
  /* Start with + since that is what the real procotol uses */
 
5984
  password.data[0] = '+';
 
5985
  /* Set a special character at string end just to mark the end */
 
5986
  password.data[oversized-3] = 'y';
 
5987
  /* Set NUL at buffer end, as suggested by the protocol */
 
5988
  password.data[oversized-2] = '\0';
 
5989
  /* Fill rest of password with 'x' */
 
5990
  memset(password.data+1, 'x', oversized-3);
 
5991
 
 
5992
  __attribute__((cleanup(cleanup_queue)))
 
5993
    task_queue *queue = create_queue();
 
5994
  g_assert_nonnull(queue);
 
5995
  int socketfds[2];
 
5996
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
5997
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
5998
                             socketfds), ==, 0);
 
5999
  __attribute__((cleanup(cleanup_close)))
 
6000
    const int read_socket = socketfds[0];
 
6001
  __attribute__((cleanup(cleanup_close)))
 
6002
    const int write_socket = socketfds[1];
 
6003
  task_context task = {
 
6004
    .func=send_password_to_socket,
 
6005
    .question_filename=strdup(question_filename),
 
6006
    .filename=filename,
 
6007
    .epoll_fd=epoll_fd,
 
6008
    .fd=write_socket,
 
6009
    .password=&password,
 
6010
    .cancelled_filenames=&cancelled_filenames,
 
6011
    .mandos_client_exited=(bool[]){true},
 
6012
    .password_is_read=(bool[]){true},
 
6013
    .current_time=(mono_microsecs[]){0},
 
6014
  };
 
6015
  g_assert_nonnull(task.question_filename);
 
6016
 
 
6017
  run_task_with_stderr_to_dev_null(task, queue);
 
6018
 
 
6019
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
6020
  g_assert_true(string_set_contains(cancelled_filenames,
 
6021
                                    question_filename));
 
6022
#endif
 
6023
}
 
6024
 
 
6025
static void test_send_password_to_socket_retry(__attribute__((unused))
 
6026
                                               test_fixture *fixture,
 
6027
                                               __attribute__((unused))
 
6028
                                               gconstpointer
 
6029
                                               user_data){
 
6030
  __attribute__((cleanup(cleanup_close)))
 
6031
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6032
  g_assert_cmpint(epoll_fd, >=, 0);
 
6033
  __attribute__((cleanup(cleanup_string)))
 
6034
    char *const question_filename = strdup("/nonexistent/question");
 
6035
  g_assert_nonnull(question_filename);
 
6036
  __attribute__((cleanup(cleanup_string)))
 
6037
    char *const filename = strdup("/nonexistent/socket");
 
6038
  g_assert_nonnull(filename);
 
6039
  __attribute__((cleanup(string_set_clear)))
 
6040
    string_set cancelled_filenames = {};
 
6041
  __attribute__((cleanup(cleanup_buffer)))
 
6042
    buffer password = {};
 
6043
 
 
6044
  __attribute__((cleanup(cleanup_queue)))
 
6045
    task_queue *queue = create_queue();
 
6046
  g_assert_nonnull(queue);
 
6047
  int socketfds[2];
 
6048
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
6049
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
6050
                             socketfds), ==, 0);
 
6051
  __attribute__((cleanup(cleanup_close)))
 
6052
    const int read_socket = socketfds[0];
 
6053
  const int write_socket = socketfds[1];
 
6054
  /* Close the server side socket to force ECONNRESET on client */
 
6055
  g_assert_cmpint(close(read_socket), ==, 0);
 
6056
  task_context task = {
 
6057
    .func=send_password_to_socket,
 
6058
    .question_filename=strdup(question_filename),
 
6059
    .filename=strdup(filename),
 
6060
    .epoll_fd=epoll_fd,
 
6061
    .fd=write_socket,
 
6062
    .password=&password,
 
6063
    .cancelled_filenames=&cancelled_filenames,
 
6064
    .mandos_client_exited=(bool[]){true},
 
6065
    .password_is_read=(bool[]){true},
 
6066
    .current_time=(mono_microsecs[]){0},
 
6067
  };
 
6068
  g_assert_nonnull(task.question_filename);
 
6069
 
 
6070
  task.func(task, queue);
 
6071
 
 
6072
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
6073
 
 
6074
  const task_context *const added_task = find_matching_task(queue,
 
6075
                                                            task);
 
6076
  g_assert_nonnull(added_task);
 
6077
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
 
6078
 
 
6079
  g_assert_true(epoll_set_contains(epoll_fd, added_task->fd,
 
6080
                                   EPOLLOUT));
 
6081
}
 
6082
 
 
6083
static
 
6084
void test_send_password_to_socket_bad_epoll(__attribute__((unused))
 
6085
                                            test_fixture *fixture,
 
6086
                                            __attribute__((unused))
 
6087
                                            gconstpointer user_data){
 
6088
  __attribute__((cleanup(cleanup_close)))
 
6089
    const int epoll_fd = open("/dev/null",
 
6090
                              O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
6091
  __attribute__((cleanup(cleanup_string)))
 
6092
    char *const question_filename = strdup("/nonexistent/question");
 
6093
  g_assert_nonnull(question_filename);
 
6094
  __attribute__((cleanup(cleanup_string)))
 
6095
    char *const filename = strdup("/nonexistent/socket");
 
6096
  g_assert_nonnull(filename);
 
6097
  __attribute__((cleanup(string_set_clear)))
 
6098
    string_set cancelled_filenames = {};
 
6099
  __attribute__((cleanup(cleanup_buffer)))
 
6100
    buffer password = {};
 
6101
 
 
6102
  const mono_microsecs current_time = 11;
 
6103
  __attribute__((cleanup(cleanup_queue)))
 
6104
    task_queue *queue = create_queue();
 
6105
  g_assert_nonnull(queue);
 
6106
  int socketfds[2];
 
6107
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
6108
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
6109
                             socketfds), ==, 0);
 
6110
  __attribute__((cleanup(cleanup_close)))
 
6111
    const int read_socket = socketfds[0];
 
6112
  const int write_socket = socketfds[1];
 
6113
  /* Close the server side socket to force ECONNRESET on client */
 
6114
  g_assert_cmpint(close(read_socket), ==, 0);
 
6115
  task_context task = {
 
6116
    .func=send_password_to_socket,
 
6117
    .question_filename=strdup(question_filename),
 
6118
    .filename=strdup(filename),
 
6119
    .epoll_fd=epoll_fd,
 
6120
    .fd=write_socket,
 
6121
    .password=&password,
 
6122
    .cancelled_filenames=&cancelled_filenames,
 
6123
    .mandos_client_exited=(bool[]){true},
 
6124
    .password_is_read=(bool[]){true},
 
6125
    .current_time=&current_time,
 
6126
  };
 
6127
  g_assert_nonnull(task.question_filename);
 
6128
 
 
6129
  run_task_with_stderr_to_dev_null(task, queue);
 
6130
 
 
6131
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
6132
 
 
6133
  const task_context *const added_task = find_matching_task(queue,
 
6134
                                                            task);
 
6135
  g_assert_nonnull(added_task);
 
6136
  g_assert_true(queue->next_run == current_time + 1000000);
 
6137
  g_assert_cmpuint((unsigned int)password.length, ==, 0);
 
6138
}
 
6139
 
 
6140
static void assert_send_password_to_socket_password(buffer password){
 
6141
  __attribute__((cleanup(cleanup_close)))
 
6142
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6143
  g_assert_cmpint(epoll_fd, >=, 0);
 
6144
  char *const question_filename = strdup("/nonexistent/question");
 
6145
  g_assert_nonnull(question_filename);
 
6146
  char *const filename = strdup("/nonexistent/socket");
 
6147
  g_assert_nonnull(filename);
 
6148
  __attribute__((cleanup(string_set_clear)))
 
6149
    string_set cancelled_filenames = {};
 
6150
 
 
6151
  __attribute__((cleanup(cleanup_queue)))
 
6152
    task_queue *queue = create_queue();
 
6153
  g_assert_nonnull(queue);
 
6154
  int socketfds[2];
 
6155
  g_assert_cmpint(socketpair(PF_LOCAL, SOCK_DGRAM
 
6156
                             | SOCK_NONBLOCK | SOCK_CLOEXEC, 0,
 
6157
                             socketfds), ==, 0);
 
6158
  __attribute__((cleanup(cleanup_close)))
 
6159
    const int read_socket = socketfds[0];
 
6160
  const int write_socket = socketfds[1];
 
6161
  task_context task = {
 
6162
    .func=send_password_to_socket,
 
6163
    .question_filename=question_filename,
 
6164
    .filename=filename,
 
6165
    .epoll_fd=epoll_fd,
 
6166
    .fd=write_socket,
 
6167
    .password=&password,
 
6168
    .cancelled_filenames=&cancelled_filenames,
 
6169
    .mandos_client_exited=(bool[]){true},
 
6170
    .password_is_read=(bool[]){true},
 
6171
    .current_time=(mono_microsecs[]){0},
 
6172
  };
 
6173
 
 
6174
  char *expected_written_data = malloc(password.length + 2);
 
6175
  g_assert_nonnull(expected_written_data);
 
6176
  expected_written_data[0] = '+';
 
6177
  expected_written_data[password.length + 1] = '\0';
 
6178
  if(password.length > 0){
 
6179
    g_assert_nonnull(password.data);
 
6180
    memcpy(expected_written_data + 1, password.data, password.length);
 
6181
  }
 
6182
 
 
6183
  task.func(task, queue);
 
6184
 
 
6185
  char buf[PIPE_BUF];
 
6186
  g_assert_cmpint((int)read(read_socket, buf, PIPE_BUF), ==,
 
6187
                  (int)(password.length + 2));
 
6188
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
6189
 
 
6190
  g_assert_true(memcmp(expected_written_data, buf,
 
6191
                       password.length + 2) == 0);
 
6192
 
 
6193
  g_assert_true(epoll_set_does_not_contain(epoll_fd, write_socket));
 
6194
 
 
6195
  free(expected_written_data);
 
6196
}
 
6197
 
 
6198
static void
 
6199
test_send_password_to_socket_null_password(__attribute__((unused))
 
6200
                                           test_fixture *fixture,
 
6201
                                           __attribute__((unused))
 
6202
                                           gconstpointer user_data){
 
6203
  __attribute__((cleanup(cleanup_buffer)))
 
6204
    buffer password = {};
 
6205
  assert_send_password_to_socket_password(password);
 
6206
}
 
6207
 
 
6208
static void
 
6209
test_send_password_to_socket_empty_password(__attribute__((unused))
 
6210
                                            test_fixture *fixture,
 
6211
                                            __attribute__((unused))
 
6212
                                            gconstpointer user_data){
 
6213
  __attribute__((cleanup(cleanup_buffer)))
 
6214
    buffer password = {
 
6215
    .data=malloc(1),           /* because malloc(0) may return NULL */
 
6216
    .length=0,
 
6217
    .allocated=0,               /* deliberate lie */
 
6218
  };
 
6219
  g_assert_nonnull(password.data);
 
6220
  assert_send_password_to_socket_password(password);
 
6221
}
 
6222
 
 
6223
static void
 
6224
test_send_password_to_socket_empty_str_pass(__attribute__((unused))
 
6225
                                            test_fixture *fixture,
 
6226
                                            __attribute__((unused))
 
6227
                                            gconstpointer user_data){
 
6228
  __attribute__((cleanup(cleanup_buffer)))
 
6229
    buffer password = {
 
6230
    .data=strdup(""),
 
6231
    .length=0,
 
6232
    .allocated=1,
 
6233
  };
 
6234
  if(mlock(password.data, password.allocated) != 0){
 
6235
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
6236
  }
 
6237
  assert_send_password_to_socket_password(password);
 
6238
}
 
6239
 
 
6240
static void
 
6241
test_send_password_to_socket_text_password(__attribute__((unused))
 
6242
                                           test_fixture *fixture,
 
6243
                                           __attribute__((unused))
 
6244
                                           gconstpointer user_data){
 
6245
  const char dummy_test_password[] = "dummy test password";
 
6246
  __attribute__((cleanup(cleanup_buffer)))
 
6247
    buffer password = {
 
6248
    .data = strdup(dummy_test_password),
 
6249
    .length = strlen(dummy_test_password),
 
6250
    .allocated = sizeof(dummy_test_password),
 
6251
  };
 
6252
  if(mlock(password.data, password.allocated) != 0){
 
6253
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
6254
  }
 
6255
  assert_send_password_to_socket_password(password);
 
6256
}
 
6257
 
 
6258
static void
 
6259
test_send_password_to_socket_binary_password(__attribute__((unused))
 
6260
                                             test_fixture *fixture,
 
6261
                                             __attribute__((unused))
 
6262
                                             gconstpointer user_data){
 
6263
  __attribute__((cleanup(cleanup_buffer)))
 
6264
    buffer password = {
 
6265
    .data=malloc(255),
 
6266
    .length=255,
 
6267
    .allocated=255,
 
6268
  };
 
6269
  g_assert_nonnull(password.data);
 
6270
  if(mlock(password.data, password.allocated) != 0){
 
6271
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
6272
  }
 
6273
  char c = 1;                   /* Start at 1, avoiding NUL */
 
6274
  for(int i=0; i < 255; i++){
 
6275
    password.data[i] = c++;
 
6276
  }
 
6277
  assert_send_password_to_socket_password(password);
 
6278
}
 
6279
 
 
6280
static void
 
6281
test_send_password_to_socket_nuls_in_password(__attribute__((unused))
 
6282
                                              test_fixture *fixture,
 
6283
                                              __attribute__((unused))
 
6284
                                              gconstpointer
 
6285
                                              user_data){
 
6286
  char test_password[] = {'\0', 'a', '\0', 'b', '\0', 'c', '\0'};
 
6287
  __attribute__((cleanup(cleanup_buffer)))
 
6288
    buffer password = {
 
6289
    .data=malloc(sizeof(test_password)),
 
6290
    .length=sizeof(test_password),
 
6291
    .allocated=sizeof(test_password),
 
6292
  };
 
6293
  g_assert_nonnull(password.data);
 
6294
  if(mlock(password.data, password.allocated) !=0){
 
6295
    g_assert_true(errno == EPERM or errno == ENOMEM);
 
6296
  }
 
6297
  memcpy(password.data, test_password, password.allocated);
 
6298
  assert_send_password_to_socket_password(password);
 
6299
}
 
6300
 
 
6301
static bool assert_add_existing_questions_to_devnull(task_queue
 
6302
                                                     *const,
 
6303
                                                     const int,
 
6304
                                                     buffer *const,
 
6305
                                                     string_set *,
 
6306
                                                     const
 
6307
                                                     mono_microsecs
 
6308
                                                     *const,
 
6309
                                                     bool *const,
 
6310
                                                     bool *const,
 
6311
                                                     const char
 
6312
                                                     *const);
 
6313
 
 
6314
static void test_add_existing_questions_ENOENT(__attribute__((unused))
 
6315
                                               test_fixture *fixture,
 
6316
                                               __attribute__((unused))
 
6317
                                               gconstpointer
 
6318
                                               user_data){
 
6319
  __attribute__((cleanup(cleanup_queue)))
 
6320
    task_queue *queue = create_queue();
 
6321
  g_assert_nonnull(queue);
 
6322
  __attribute__((cleanup(cleanup_close)))
 
6323
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6324
  g_assert_cmpint(epoll_fd, >=, 0);
 
6325
  __attribute__((cleanup(string_set_clear)))
 
6326
    string_set cancelled_filenames = {};
 
6327
 
 
6328
  g_assert_false(assert_add_existing_questions_to_devnull
 
6329
                 (queue,
 
6330
                  epoll_fd,
 
6331
                  (buffer[]){{}}, /* password */
 
6332
                  &cancelled_filenames,
 
6333
                  (mono_microsecs[]){0}, /* current_time */
 
6334
                  (bool[]){false},       /* mandos_client_exited */
 
6335
                  (bool[]){false},       /* password_is_read */
 
6336
                  "/nonexistent"));      /* dirname */
 
6337
 
 
6338
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
6339
}
 
6340
 
 
6341
static
 
6342
bool assert_add_existing_questions_to_devnull(task_queue
 
6343
                                              *const queue,
 
6344
                                              const int
 
6345
                                              epoll_fd,
 
6346
                                              buffer *const
 
6347
                                              password,
 
6348
                                              string_set
 
6349
                                              *cancelled_filenames,
 
6350
                                              const mono_microsecs
 
6351
                                              *const current_time,
 
6352
                                              bool *const
 
6353
                                              mandos_client_exited,
 
6354
                                              bool *const
 
6355
                                              password_is_read,
 
6356
                                              const char *const
 
6357
                                              dirname){
 
6358
  __attribute__((cleanup(cleanup_close)))
 
6359
    const int devnull_fd = open("/dev/null",
 
6360
                                O_WRONLY | O_CLOEXEC | O_NOCTTY);
 
6361
  g_assert_cmpint(devnull_fd, >=, 0);
 
6362
  __attribute__((cleanup(cleanup_close)))
 
6363
    const int real_stderr_fd = dup(STDERR_FILENO);
 
6364
  g_assert_cmpint(real_stderr_fd, >=, 0);
 
6365
  dup2(devnull_fd, STDERR_FILENO);
 
6366
  const bool ret = add_existing_questions(queue, epoll_fd, password,
 
6367
                                          cancelled_filenames,
 
6368
                                          current_time,
 
6369
                                          mandos_client_exited,
 
6370
                                          password_is_read, dirname);
 
6371
  dup2(real_stderr_fd, STDERR_FILENO);
 
6372
  return ret;
 
6373
}
 
6374
 
 
6375
static
 
6376
void test_add_existing_questions_no_questions(__attribute__((unused))
 
6377
                                              test_fixture *fixture,
 
6378
                                              __attribute__((unused))
 
6379
                                              gconstpointer
 
6380
                                              user_data){
 
6381
  __attribute__((cleanup(cleanup_queue)))
 
6382
    task_queue *queue = create_queue();
 
6383
  g_assert_nonnull(queue);
 
6384
  __attribute__((cleanup(cleanup_close)))
 
6385
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6386
  g_assert_cmpint(epoll_fd, >=, 0);
 
6387
  __attribute__((cleanup(string_set_clear)))
 
6388
    string_set cancelled_filenames = {};
 
6389
  __attribute__((cleanup(cleanup_string)))
 
6390
    char *tempdir = make_temporary_directory();
 
6391
  g_assert_nonnull(tempdir);
 
6392
 
 
6393
  g_assert_false(assert_add_existing_questions_to_devnull
 
6394
                 (queue,
 
6395
                  epoll_fd,
 
6396
                  (buffer[]){{}}, /* password */
 
6397
                  &cancelled_filenames,
 
6398
                  (mono_microsecs[]){0}, /* current_time */
 
6399
                  (bool[]){false},       /* mandos_client_exited */
 
6400
                  (bool[]){false},       /* password_is_read */
 
6401
                  tempdir));
 
6402
 
 
6403
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
6404
 
 
6405
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
6406
}
 
6407
 
 
6408
static char *make_question_file_in_directory(const char *const);
 
6409
 
 
6410
static
 
6411
void test_add_existing_questions_one_question(__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(cleanup_buffer)))
 
6423
    buffer password = {};
 
6424
  __attribute__((cleanup(string_set_clear)))
 
6425
    string_set cancelled_filenames = {};
 
6426
  const mono_microsecs current_time = 0;
 
6427
  bool mandos_client_exited = false;
 
6428
  bool password_is_read = false;
 
6429
  __attribute__((cleanup(cleanup_string)))
 
6430
    char *tempdir = make_temporary_directory();
 
6431
  g_assert_nonnull(tempdir);
 
6432
  __attribute__((cleanup(cleanup_string)))
 
6433
    char *question_filename
 
6434
    = make_question_file_in_directory(tempdir);
 
6435
  g_assert_nonnull(question_filename);
 
6436
 
 
6437
  g_assert_true(assert_add_existing_questions_to_devnull
 
6438
                (queue,
 
6439
                 epoll_fd,
 
6440
                 &password,
 
6441
                 &cancelled_filenames,
 
6442
                 &current_time,
 
6443
                 &mandos_client_exited,
 
6444
                 &password_is_read,
 
6445
                 tempdir));
 
6446
 
 
6447
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
6448
 
 
6449
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
6450
        .func=open_and_parse_question,
 
6451
        .epoll_fd=epoll_fd,
 
6452
        .filename=question_filename,
 
6453
        .question_filename=question_filename,
 
6454
        .password=&password,
 
6455
        .cancelled_filenames=&cancelled_filenames,
 
6456
        .current_time=&current_time,
 
6457
        .mandos_client_exited=&mandos_client_exited,
 
6458
        .password_is_read=&password_is_read,
 
6459
      }));
 
6460
 
 
6461
  g_assert_true(queue->next_run == 1);
 
6462
 
 
6463
  g_assert_cmpint(unlink(question_filename), ==, 0);
 
6464
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
6465
}
 
6466
 
 
6467
static char *make_question_file_in_directory(const char
 
6468
                                             *const dir){
 
6469
  return make_temporary_prefixed_file_in_directory("ask.", dir);
 
6470
}
 
6471
 
 
6472
static
 
6473
void test_add_existing_questions_two_questions(__attribute__((unused))
 
6474
                                               test_fixture *fixture,
 
6475
                                               __attribute__((unused))
 
6476
                                               gconstpointer
 
6477
                                               user_data){
 
6478
  __attribute__((cleanup(cleanup_queue)))
 
6479
    task_queue *queue = create_queue();
 
6480
  g_assert_nonnull(queue);
 
6481
  __attribute__((cleanup(cleanup_close)))
 
6482
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6483
  g_assert_cmpint(epoll_fd, >=, 0);
 
6484
  __attribute__((cleanup(cleanup_buffer)))
 
6485
    buffer password = {};
 
6486
  __attribute__((cleanup(string_set_clear)))
 
6487
    string_set cancelled_filenames = {};
 
6488
  const mono_microsecs current_time = 0;
 
6489
  bool mandos_client_exited = false;
 
6490
  bool password_is_read = false;
 
6491
  __attribute__((cleanup(cleanup_string)))
 
6492
    char *tempdir = make_temporary_directory();
 
6493
  g_assert_nonnull(tempdir);
 
6494
  __attribute__((cleanup(cleanup_string)))
 
6495
    char *question_filename1
 
6496
    = make_question_file_in_directory(tempdir);
 
6497
  g_assert_nonnull(question_filename1);
 
6498
  __attribute__((cleanup(cleanup_string)))
 
6499
    char *question_filename2
 
6500
    = make_question_file_in_directory(tempdir);
 
6501
  g_assert_nonnull(question_filename2);
 
6502
 
 
6503
  g_assert_true(assert_add_existing_questions_to_devnull
 
6504
                (queue,
 
6505
                 epoll_fd,
 
6506
                 &password,
 
6507
                 &cancelled_filenames,
 
6508
                 &current_time,
 
6509
                 &mandos_client_exited,
 
6510
                 &password_is_read,
 
6511
                 tempdir));
 
6512
 
 
6513
  g_assert_cmpuint((unsigned int)queue->length, ==, 2);
 
6514
 
 
6515
  g_assert_true(queue->next_run == 1);
 
6516
 
 
6517
  __attribute__((cleanup(string_set_clear)))
 
6518
    string_set seen_questions = {};
 
6519
 
 
6520
  bool queue_contains_question_opener(char *const question_filename){
 
6521
    return(find_matching_task(queue, (task_context){
 
6522
          .func=open_and_parse_question,
 
6523
          .epoll_fd=epoll_fd,
 
6524
          .question_filename=question_filename,
 
6525
          .password=&password,
 
6526
          .cancelled_filenames=&cancelled_filenames,
 
6527
          .current_time=&current_time,
 
6528
          .mandos_client_exited=&mandos_client_exited,
 
6529
          .password_is_read=&password_is_read,
 
6530
        }) != NULL);
 
6531
  }
 
6532
 
 
6533
  g_assert_true(queue_contains_question_opener(question_filename1));
 
6534
  g_assert_true(queue_contains_question_opener(question_filename2));
 
6535
 
 
6536
  g_assert_true(queue->next_run == 1);
 
6537
 
 
6538
  g_assert_cmpint(unlink(question_filename1), ==, 0);
 
6539
  g_assert_cmpint(unlink(question_filename2), ==, 0);
 
6540
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
6541
}
 
6542
 
 
6543
static void
 
6544
test_add_existing_questions_non_questions(__attribute__((unused))
 
6545
                                          test_fixture *fixture,
 
6546
                                          __attribute__((unused))
 
6547
                                          gconstpointer user_data){
 
6548
  __attribute__((cleanup(cleanup_queue)))
 
6549
    task_queue *queue = create_queue();
 
6550
  g_assert_nonnull(queue);
 
6551
  __attribute__((cleanup(cleanup_close)))
 
6552
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6553
  g_assert_cmpint(epoll_fd, >=, 0);
 
6554
  __attribute__((cleanup(string_set_clear)))
 
6555
    string_set cancelled_filenames = {};
 
6556
  __attribute__((cleanup(cleanup_string)))
 
6557
    char *tempdir = make_temporary_directory();
 
6558
  g_assert_nonnull(tempdir);
 
6559
  __attribute__((cleanup(cleanup_string)))
 
6560
    char *question_filename1
 
6561
    = make_temporary_file_in_directory(tempdir);
 
6562
  g_assert_nonnull(question_filename1);
 
6563
  __attribute__((cleanup(cleanup_string)))
 
6564
    char *question_filename2
 
6565
    = make_temporary_file_in_directory(tempdir);
 
6566
  g_assert_nonnull(question_filename2);
 
6567
 
 
6568
  g_assert_false(assert_add_existing_questions_to_devnull
 
6569
                 (queue,
 
6570
                  epoll_fd,
 
6571
                  (buffer[]){{}}, /* password */
 
6572
                  &cancelled_filenames,
 
6573
                  (mono_microsecs[]){0}, /* current_time */
 
6574
                  (bool[]){false},       /* mandos_client_exited */
 
6575
                  (bool[]){false},       /* password_is_read */
 
6576
                  tempdir));
 
6577
 
 
6578
  g_assert_cmpuint((unsigned int)queue->length, ==, 0);
 
6579
 
 
6580
  g_assert_cmpint(unlink(question_filename1), ==, 0);
 
6581
  g_assert_cmpint(unlink(question_filename2), ==, 0);
 
6582
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
6583
}
 
6584
 
 
6585
static void
 
6586
test_add_existing_questions_both_types(__attribute__((unused))
 
6587
                                       test_fixture *fixture,
 
6588
                                       __attribute__((unused))
 
6589
                                       gconstpointer user_data){
 
6590
  __attribute__((cleanup(cleanup_queue)))
 
6591
    task_queue *queue = create_queue();
 
6592
  g_assert_nonnull(queue);
 
6593
  __attribute__((cleanup(cleanup_close)))
 
6594
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6595
  g_assert_cmpint(epoll_fd, >=, 0);
 
6596
  __attribute__((cleanup(cleanup_buffer)))
 
6597
    buffer password = {};
 
6598
  __attribute__((cleanup(string_set_clear)))
 
6599
    string_set cancelled_filenames = {};
 
6600
  const mono_microsecs current_time = 0;
 
6601
  bool mandos_client_exited = false;
 
6602
  bool password_is_read = false;
 
6603
  __attribute__((cleanup(cleanup_string)))
 
6604
    char *tempdir = make_temporary_directory();
 
6605
  g_assert_nonnull(tempdir);
 
6606
  __attribute__((cleanup(cleanup_string)))
 
6607
    char *tempfilename1 = make_temporary_file_in_directory(tempdir);
 
6608
  g_assert_nonnull(tempfilename1);
 
6609
  __attribute__((cleanup(cleanup_string)))
 
6610
    char *tempfilename2 = make_temporary_file_in_directory(tempdir);
 
6611
  g_assert_nonnull(tempfilename2);
 
6612
  __attribute__((cleanup(cleanup_string)))
 
6613
    char *question_filename
 
6614
    = make_question_file_in_directory(tempdir);
 
6615
  g_assert_nonnull(question_filename);
 
6616
 
 
6617
  g_assert_true(assert_add_existing_questions_to_devnull
 
6618
                (queue,
 
6619
                 epoll_fd,
 
6620
                 &password,
 
6621
                 &cancelled_filenames,
 
6622
                 &current_time,
 
6623
                 &mandos_client_exited,
 
6624
                 &password_is_read,
 
6625
                 tempdir));
 
6626
 
 
6627
  g_assert_cmpuint((unsigned int)queue->length, ==, 1);
 
6628
 
 
6629
  g_assert_nonnull(find_matching_task(queue, (task_context){
 
6630
        .func=open_and_parse_question,
 
6631
        .epoll_fd=epoll_fd,
 
6632
        .filename=question_filename,
 
6633
        .question_filename=question_filename,
 
6634
        .password=&password,
 
6635
        .cancelled_filenames=&cancelled_filenames,
 
6636
        .current_time=&current_time,
 
6637
        .mandos_client_exited=&mandos_client_exited,
 
6638
        .password_is_read=&password_is_read,
 
6639
      }));
 
6640
 
 
6641
  g_assert_true(queue->next_run == 1);
 
6642
 
 
6643
  g_assert_cmpint(unlink(tempfilename1), ==, 0);
 
6644
  g_assert_cmpint(unlink(tempfilename2), ==, 0);
 
6645
  g_assert_cmpint(unlink(question_filename), ==, 0);
 
6646
  g_assert_cmpint(rmdir(tempdir), ==, 0);
 
6647
}
 
6648
 
 
6649
static void test_wait_for_event_timeout(__attribute__((unused))
 
6650
                                        test_fixture *fixture,
 
6651
                                        __attribute__((unused))
 
6652
                                        gconstpointer user_data){
 
6653
  __attribute__((cleanup(cleanup_close)))
 
6654
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6655
  g_assert_cmpint(epoll_fd, >=, 0);
 
6656
 
 
6657
  g_assert_true(wait_for_event(epoll_fd, 1, 0));
 
6658
}
 
6659
 
 
6660
static void test_wait_for_event_event(__attribute__((unused))
 
6661
                                      test_fixture *fixture,
 
6662
                                      __attribute__((unused))
 
6663
                                      gconstpointer user_data){
 
6664
  __attribute__((cleanup(cleanup_close)))
 
6665
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6666
  g_assert_cmpint(epoll_fd, >=, 0);
 
6667
  int pipefds[2];
 
6668
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
6669
  __attribute__((cleanup(cleanup_close)))
 
6670
    const int read_pipe = pipefds[0];
 
6671
  __attribute__((cleanup(cleanup_close)))
 
6672
    const int write_pipe = pipefds[1];
 
6673
  g_assert_cmpint(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, read_pipe,
 
6674
                            &(struct epoll_event)
 
6675
                            { .events=EPOLLIN | EPOLLRDHUP }), ==, 0);
 
6676
  g_assert_cmpint((int)write(write_pipe, "x", 1), ==, 1);
 
6677
 
 
6678
  g_assert_true(wait_for_event(epoll_fd, 0, 0));
 
6679
}
 
6680
 
 
6681
static void test_wait_for_event_sigchld(test_fixture *fixture,
 
6682
                                        __attribute__((unused))
 
6683
                                        gconstpointer user_data){
 
6684
  const pid_t pid = fork();
 
6685
  if(pid == 0){         /* Child */
 
6686
    if(not restore_signal_handler(&fixture->orig_sigaction)){
 
6687
      _exit(EXIT_FAILURE);
 
6688
    }
 
6689
    if(not restore_sigmask(&fixture->orig_sigmask)){
 
6690
      _exit(EXIT_FAILURE);
 
6691
    }
 
6692
    exit(EXIT_SUCCESS);
 
6693
  }
 
6694
  g_assert_true(pid != -1);
 
6695
  __attribute__((cleanup(cleanup_close)))
 
6696
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6697
  g_assert_cmpint(epoll_fd, >=, 0);
 
6698
 
 
6699
  g_assert_true(wait_for_event(epoll_fd, 0, 0));
 
6700
 
 
6701
  int status;
 
6702
  g_assert_true(waitpid(pid, &status, 0) == pid);
 
6703
  g_assert_true(WIFEXITED(status));
 
6704
  g_assert_cmpint(WEXITSTATUS(status), ==, EXIT_SUCCESS);
 
6705
}
 
6706
 
 
6707
static void test_run_queue_zeroes_next_run(__attribute__((unused))
 
6708
                                           test_fixture *fixture,
 
6709
                                           __attribute__((unused))
 
6710
                                           gconstpointer user_data){
 
6711
  __attribute__((cleanup(cleanup_queue)))
 
6712
    task_queue *queue = create_queue();
 
6713
  g_assert_nonnull(queue);
 
6714
  queue->next_run = 1;
 
6715
  __attribute__((cleanup(cleanup_close)))
 
6716
    const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
 
6717
  __attribute__((cleanup(string_set_clear)))
 
6718
    string_set cancelled_filenames = {};
 
6719
  bool quit_now = false;
 
6720
 
 
6721
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6722
  g_assert_false(quit_now);
 
6723
  g_assert_cmpuint((unsigned int)queue->next_run, ==, 0);
 
6724
}
 
6725
 
 
6726
static
 
6727
void test_run_queue_clears_cancelled_filenames(__attribute__((unused))
 
6728
                                               test_fixture *fixture,
 
6729
                                               __attribute__((unused))
 
6730
                                               gconstpointer
 
6731
                                               user_data){
 
6732
  __attribute__((cleanup(cleanup_queue)))
 
6733
    task_queue *queue = create_queue();
 
6734
  g_assert_nonnull(queue);
 
6735
  __attribute__((cleanup(string_set_clear)))
 
6736
    string_set cancelled_filenames = {};
 
6737
  bool quit_now = false;
 
6738
  const char question_filename[] = "/nonexistent/question_filename";
 
6739
  g_assert_true(string_set_add(&cancelled_filenames,
 
6740
                               question_filename));
 
6741
 
 
6742
  g_assert_true(add_to_queue(queue,
 
6743
                             (task_context){ .func=dummy_func }));
 
6744
 
 
6745
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6746
  g_assert_false(quit_now);
 
6747
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
 
6748
  g_assert_false(string_set_contains(cancelled_filenames,
 
6749
                                     question_filename));
 
6750
}
 
6751
 
 
6752
static
 
6753
void test_run_queue_skips_cancelled_filenames(__attribute__((unused))
 
6754
                                              test_fixture *fixture,
 
6755
                                              __attribute__((unused))
 
6756
                                              gconstpointer
 
6757
                                              user_data){
 
6758
  __attribute__((cleanup(cleanup_queue)))
 
6759
    task_queue *queue = create_queue();
 
6760
  g_assert_nonnull(queue);
 
6761
  __attribute__((cleanup(string_set_clear)))
 
6762
    string_set cancelled_filenames = {};
 
6763
  bool quit_now = false;
 
6764
  int pipefds[2];
 
6765
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
6766
  __attribute__((cleanup(cleanup_close)))
 
6767
    const int read_pipe = pipefds[0];
 
6768
  g_assert_cmpint(close(pipefds[1]), ==, 0);
 
6769
  const char question_filename[] = "/nonexistent/question_filename";
 
6770
  g_assert_true(string_set_add(&cancelled_filenames,
 
6771
                               question_filename));
 
6772
  __attribute__((nonnull))
 
6773
    void quit_func(const task_context task,
 
6774
                   __attribute__((unused)) task_queue *const q){
 
6775
    g_assert_nonnull(task.quit_now);
 
6776
    *task.quit_now = true;
 
6777
  }
 
6778
  task_context task = {
 
6779
    .func=quit_func,
 
6780
    .question_filename=strdup(question_filename),
 
6781
    .quit_now=&quit_now,
 
6782
    .fd=read_pipe,
 
6783
  };
 
6784
  g_assert_nonnull(task.question_filename);
 
6785
 
 
6786
  g_assert_true(add_to_queue(queue, task));
 
6787
 
 
6788
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6789
  g_assert_false(quit_now);
 
6790
 
 
6791
  /* read_pipe should be closed already */
 
6792
  errno = 0;
 
6793
  bool read_pipe_closed = (close(read_pipe) == -1);
 
6794
  read_pipe_closed &= (errno == EBADF);
 
6795
  g_assert_true(read_pipe_closed);
 
6796
}
 
6797
 
 
6798
static void test_run_queue_one_task(__attribute__((unused))
 
6799
                                    test_fixture *fixture,
 
6800
                                    __attribute__((unused))
 
6801
                                    gconstpointer user_data){
 
6802
  __attribute__((cleanup(cleanup_queue)))
 
6803
    task_queue *queue = create_queue();
 
6804
  g_assert_nonnull(queue);
 
6805
  __attribute__((cleanup(string_set_clear)))
 
6806
    string_set cancelled_filenames = {};
 
6807
  bool quit_now = false;
 
6808
 
 
6809
  __attribute__((nonnull))
 
6810
    void next_run_func(__attribute__((unused))
 
6811
                       const task_context task,
 
6812
                       task_queue *const q){
 
6813
    q->next_run = 1;
 
6814
  }
 
6815
 
 
6816
  task_context task = {
 
6817
    .func=next_run_func,
 
6818
  };
 
6819
  g_assert_true(add_to_queue(queue, task));
 
6820
 
 
6821
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6822
  g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);
 
6823
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
 
6824
}
 
6825
 
 
6826
static void test_run_queue_two_tasks(__attribute__((unused))
 
6827
                                     test_fixture *fixture,
 
6828
                                     __attribute__((unused))
 
6829
                                     gconstpointer user_data){
 
6830
  __attribute__((cleanup(cleanup_queue)))
 
6831
    task_queue *queue = create_queue();
 
6832
  g_assert_nonnull(queue);
 
6833
  queue->next_run = 1;
 
6834
  __attribute__((cleanup(string_set_clear)))
 
6835
    string_set cancelled_filenames = {};
 
6836
  bool quit_now = false;
 
6837
  bool mandos_client_exited = false;
 
6838
 
 
6839
  __attribute__((nonnull))
 
6840
    void next_run_func(__attribute__((unused))
 
6841
                       const task_context task,
 
6842
                       task_queue *const q){
 
6843
    q->next_run = 1;
 
6844
  }
 
6845
 
 
6846
  __attribute__((nonnull))
 
6847
    void exited_func(const task_context task,
 
6848
                     __attribute__((unused)) task_queue *const q){
 
6849
    *task.mandos_client_exited = true;
 
6850
  }
 
6851
 
 
6852
  task_context task1 = {
 
6853
    .func=next_run_func,
 
6854
  };
 
6855
  g_assert_true(add_to_queue(queue, task1));
 
6856
 
 
6857
  task_context task2 = {
 
6858
    .func=exited_func,
 
6859
    .mandos_client_exited=&mandos_client_exited,
 
6860
  };
 
6861
  g_assert_true(add_to_queue(queue, task2));
 
6862
 
 
6863
  g_assert_true(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6864
  g_assert_false(quit_now);
 
6865
  g_assert_cmpuint((unsigned int)(queue->next_run), ==, 1);
 
6866
  g_assert_true(mandos_client_exited);
 
6867
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
 
6868
}
 
6869
 
 
6870
static void test_run_queue_two_tasks_quit(__attribute__((unused))
 
6871
                                          test_fixture *fixture,
 
6872
                                          __attribute__((unused))
 
6873
                                          gconstpointer user_data){
 
6874
  __attribute__((cleanup(cleanup_queue)))
 
6875
    task_queue *queue = create_queue();
 
6876
  g_assert_nonnull(queue);
 
6877
  __attribute__((cleanup(string_set_clear)))
 
6878
    string_set cancelled_filenames = {};
 
6879
  bool quit_now = false;
 
6880
  bool mandos_client_exited = false;
 
6881
  bool password_is_read = false;
 
6882
 
 
6883
  __attribute__((nonnull))
 
6884
    void set_exited_func(const task_context task,
 
6885
                         __attribute__((unused)) task_queue *const q){
 
6886
    *task.mandos_client_exited = true;
 
6887
    *task.quit_now = true;
 
6888
  }
 
6889
  task_context task1 = {
 
6890
    .func=set_exited_func,
 
6891
    .quit_now=&quit_now,
 
6892
    .mandos_client_exited=&mandos_client_exited,
 
6893
  };
 
6894
  g_assert_true(add_to_queue(queue, task1));
 
6895
 
 
6896
  __attribute__((nonnull))
 
6897
    void set_read_func(const task_context task,
 
6898
                       __attribute__((unused)) task_queue *const q){
 
6899
    *task.quit_now = true;
 
6900
    *task.password_is_read = true;
 
6901
  }
 
6902
  task_context task2 = {
 
6903
    .func=set_read_func,
 
6904
    .quit_now=&quit_now,
 
6905
    .password_is_read=&password_is_read,
 
6906
  };
 
6907
  g_assert_true(add_to_queue(queue, task2));
 
6908
 
 
6909
  g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6910
  g_assert_true(quit_now);
 
6911
  g_assert_true(mandos_client_exited xor password_is_read);
 
6912
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
 
6913
}
 
6914
 
 
6915
static void test_run_queue_two_tasks_cleanup(__attribute__((unused))
 
6916
                                             test_fixture *fixture,
 
6917
                                             __attribute__((unused))
 
6918
                                             gconstpointer user_data){
 
6919
  __attribute__((cleanup(cleanup_queue)))
 
6920
    task_queue *queue = create_queue();
 
6921
  g_assert_nonnull(queue);
 
6922
  __attribute__((cleanup(string_set_clear)))
 
6923
    string_set cancelled_filenames = {};
 
6924
  int pipefds[2];
 
6925
  g_assert_cmpint(pipe2(pipefds, O_CLOEXEC | O_NONBLOCK), ==, 0);
 
6926
  __attribute__((cleanup(cleanup_close)))
 
6927
    const int read_pipe = pipefds[0];
 
6928
  __attribute__((cleanup(cleanup_close)))
 
6929
    const int write_pipe = pipefds[1];
 
6930
  bool quit_now = false;
 
6931
 
 
6932
  __attribute__((nonnull))
 
6933
    void read_func(const task_context task,
 
6934
                   __attribute__((unused)) task_queue *const q){
 
6935
    *task.quit_now = true;
 
6936
  }
 
6937
  task_context task1 = {
 
6938
    .func=read_func,
 
6939
    .quit_now=&quit_now,
 
6940
    .fd=read_pipe,
 
6941
  };
 
6942
  g_assert_true(add_to_queue(queue, task1));
 
6943
 
 
6944
  __attribute__((nonnull))
 
6945
    void write_func(const task_context task,
 
6946
                    __attribute__((unused)) task_queue *const q){
 
6947
    *task.quit_now = true;
 
6948
  }
 
6949
  task_context task2 = {
 
6950
    .func=write_func,
 
6951
    .quit_now=&quit_now,
 
6952
    .fd=write_pipe,
 
6953
  };
 
6954
  g_assert_true(add_to_queue(queue, task2));
 
6955
 
 
6956
  g_assert_false(run_queue(&queue, &cancelled_filenames, &quit_now));
 
6957
  g_assert_true(quit_now);
 
6958
 
 
6959
  /* Either read_pipe or write_pipe should be closed already */
 
6960
  errno = 0;
 
6961
  bool close_read_pipe = (close(read_pipe) == -1);
 
6962
  close_read_pipe &= (errno == EBADF);
 
6963
  errno = 0;
 
6964
  bool close_write_pipe = (close(write_pipe) == -1);
 
6965
  close_write_pipe &= (errno == EBADF);
 
6966
  g_assert_true(close_read_pipe xor close_write_pipe);
 
6967
  g_assert_cmpuint((unsigned int)(queue->length), ==, 0);
 
6968
}
 
6969
 
 
6970
static void test_setup_signal_handler(__attribute__((unused))
 
6971
                                      test_fixture *fixture,
 
6972
                                      __attribute__((unused))
 
6973
                                      gconstpointer user_data){
 
6974
  /* Save current SIGCHLD action, whatever it is */
 
6975
  struct sigaction expected_sigchld_action;
 
6976
  g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),
 
6977
                  ==, 0);
 
6978
 
 
6979
  /* Act; i.e. run the setup_signal_handler() function */
 
6980
  struct sigaction actual_old_sigchld_action;
 
6981
  g_assert_true(setup_signal_handler(&actual_old_sigchld_action));
 
6982
 
 
6983
  /* Check that the function correctly set "actual_old_sigchld_action"
 
6984
     to the same values as the previously saved
 
6985
     "expected_sigchld_action" */
 
6986
  /* Check member sa_handler */
 
6987
  g_assert_true(actual_old_sigchld_action.sa_handler
 
6988
                == expected_sigchld_action.sa_handler);
 
6989
  /* Check member sa_mask */
 
6990
  for(int signum = 1; signum < NSIG; signum++){
 
6991
    const int expected_old_block_state
 
6992
      = sigismember(&expected_sigchld_action.sa_mask, signum);
 
6993
    g_assert_cmpint(expected_old_block_state, >=, 0);
 
6994
    const int actual_old_block_state
 
6995
      = sigismember(&actual_old_sigchld_action.sa_mask, signum);
 
6996
    g_assert_cmpint(actual_old_block_state, >=, 0);
 
6997
    g_assert_cmpint(actual_old_block_state,
 
6998
                    ==, expected_old_block_state);
 
6999
  }
 
7000
  /* Check member sa_flags */
 
7001
  g_assert_true((actual_old_sigchld_action.sa_flags
 
7002
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
 
7003
                == (expected_sigchld_action.sa_flags
 
7004
                    & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));
 
7005
 
 
7006
  /* Retrieve the current signal handler for SIGCHLD as set by
 
7007
     setup_signal_handler() */
 
7008
  struct sigaction actual_new_sigchld_action;
 
7009
  g_assert_cmpint(sigaction(SIGCHLD, NULL,
 
7010
                            &actual_new_sigchld_action), ==, 0);
 
7011
  /* Check that the signal handler (member sa_handler) is correctly
 
7012
     set to the "handle_sigchld" function */
 
7013
  g_assert_true(actual_new_sigchld_action.sa_handler != SIG_DFL);
 
7014
  g_assert_true(actual_new_sigchld_action.sa_handler != SIG_IGN);
 
7015
  g_assert_true(actual_new_sigchld_action.sa_handler
 
7016
                == handle_sigchld);
 
7017
  /* Check (in member sa_mask) that at least a handful of signals are
 
7018
     actually blocked during the signal handler */
 
7019
  for(int signum = 1; signum < NSIG; signum++){
 
7020
    int actual_new_block_state;
 
7021
    switch(signum){
 
7022
    case SIGTERM:
 
7023
    case SIGINT:
 
7024
    case SIGQUIT:
 
7025
    case SIGHUP:
 
7026
      actual_new_block_state
 
7027
        = sigismember(&actual_new_sigchld_action.sa_mask, signum);
 
7028
      g_assert_cmpint(actual_new_block_state, ==, 1);
 
7029
      continue;
 
7030
    case SIGKILL:               /* non-blockable */
 
7031
    case SIGSTOP:               /* non-blockable */
 
7032
    case SIGCHLD:               /* always blocked */
 
7033
    default:
 
7034
      continue;
 
7035
    }
 
7036
  }
 
7037
  /* Check member sa_flags */
 
7038
  g_assert_true((actual_new_sigchld_action.sa_flags
 
7039
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
 
7040
                == (SA_NOCLDSTOP | SA_RESTART));
 
7041
 
 
7042
  /* Restore signal handler */
 
7043
  g_assert_cmpint(sigaction(SIGCHLD, &expected_sigchld_action, NULL),
 
7044
                  ==, 0);
 
7045
}
 
7046
 
 
7047
static void test_restore_signal_handler(__attribute__((unused))
 
7048
                                        test_fixture *fixture,
 
7049
                                        __attribute__((unused))
 
7050
                                        gconstpointer user_data){
 
7051
  /* Save current SIGCHLD action, whatever it is */
 
7052
  struct sigaction expected_sigchld_action;
 
7053
  g_assert_cmpint(sigaction(SIGCHLD, NULL, &expected_sigchld_action),
 
7054
                  ==, 0);
 
7055
  /* Since we haven't established a signal handler yet, there should
 
7056
     not be one established.  But another test may have relied on
 
7057
     restore_signal_handler() to restore the signal handler, and if
 
7058
     restore_signal_handler() is buggy (which we should be prepared
 
7059
     for in this test) the signal handler may not have been restored
 
7060
     properly; check for this: */
 
7061
  g_assert_true(expected_sigchld_action.sa_handler != handle_sigchld);
 
7062
 
 
7063
  /* Establish a signal handler */
 
7064
  struct sigaction sigchld_action = {
 
7065
    .sa_handler=handle_sigchld,
 
7066
    .sa_flags=SA_RESTART | SA_NOCLDSTOP,
 
7067
  };
 
7068
  g_assert_cmpint(sigfillset(&sigchld_action.sa_mask), ==, 0);
 
7069
  g_assert_cmpint(sigaction(SIGCHLD, &sigchld_action, NULL), ==, 0);
 
7070
 
 
7071
  /* Act; i.e. run the restore_signal_handler() function */
 
7072
  g_assert_true(restore_signal_handler(&expected_sigchld_action));
 
7073
 
 
7074
  /* Retrieve the restored signal handler data */
 
7075
  struct sigaction actual_restored_sigchld_action;
 
7076
  g_assert_cmpint(sigaction(SIGCHLD, NULL,
 
7077
                            &actual_restored_sigchld_action), ==, 0);
 
7078
 
 
7079
  /* Check that the function correctly restored the signal action, as
 
7080
     saved in "actual_restored_sigchld_action", to the same values as
 
7081
     the previously saved "expected_sigchld_action" */
 
7082
  /* Check member sa_handler */
 
7083
  g_assert_true(actual_restored_sigchld_action.sa_handler
 
7084
                == expected_sigchld_action.sa_handler);
 
7085
  /* Check member sa_mask */
 
7086
  for(int signum = 1; signum < NSIG; signum++){
 
7087
    const int expected_old_block_state
 
7088
      = sigismember(&expected_sigchld_action.sa_mask, signum);
 
7089
    g_assert_cmpint(expected_old_block_state, >=, 0);
 
7090
    const int actual_restored_block_state
 
7091
      = sigismember(&actual_restored_sigchld_action.sa_mask, signum);
 
7092
    g_assert_cmpint(actual_restored_block_state, >=, 0);
 
7093
    g_assert_cmpint(actual_restored_block_state,
 
7094
                    ==, expected_old_block_state);
 
7095
  }
 
7096
  /* Check member sa_flags */
 
7097
  g_assert_true((actual_restored_sigchld_action.sa_flags
 
7098
                 & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART))
 
7099
                == (expected_sigchld_action.sa_flags
 
7100
                    & (SA_NOCLDSTOP | SA_ONSTACK | SA_RESTART)));
 
7101
}
 
7102
 
 
7103
static void test_block_sigchld(__attribute__((unused))
 
7104
                               test_fixture *fixture,
 
7105
                               __attribute__((unused))
 
7106
                               gconstpointer user_data){
 
7107
  /* Save original signal mask */
 
7108
  sigset_t expected_sigmask;
 
7109
  g_assert_cmpint(pthread_sigmask(-1, NULL, &expected_sigmask),
 
7110
                  ==, 0);
 
7111
 
 
7112
  /* Make sure SIGCHLD is unblocked for this test */
 
7113
  sigset_t sigchld_sigmask;
 
7114
  g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);
 
7115
  g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);
 
7116
  g_assert_cmpint(pthread_sigmask(SIG_UNBLOCK, &sigchld_sigmask,
 
7117
                                  NULL), ==, 0);
 
7118
 
 
7119
  /* Act; i.e. run the block_sigchld() function */
 
7120
  sigset_t actual_old_sigmask;
 
7121
  g_assert_true(block_sigchld(&actual_old_sigmask));
 
7122
 
 
7123
  /* Check the actual_old_sigmask; it should be the same as the
 
7124
     previously saved signal mask "expected_sigmask". */
 
7125
  for(int signum = 1; signum < NSIG; signum++){
 
7126
    const int expected_old_block_state
 
7127
      = sigismember(&expected_sigmask, signum);
 
7128
    g_assert_cmpint(expected_old_block_state, >=, 0);
 
7129
    const int actual_old_block_state
 
7130
      = sigismember(&actual_old_sigmask, signum);
 
7131
    g_assert_cmpint(actual_old_block_state, >=, 0);
 
7132
    g_assert_cmpint(actual_old_block_state,
 
7133
                    ==, expected_old_block_state);
 
7134
  }
 
7135
 
 
7136
  /* Retrieve the newly set signal mask */
 
7137
  sigset_t actual_sigmask;
 
7138
  g_assert_cmpint(pthread_sigmask(-1, NULL, &actual_sigmask), ==, 0);
 
7139
 
 
7140
  /* SIGCHLD should be blocked */
 
7141
  g_assert_cmpint(sigismember(&actual_sigmask, SIGCHLD), ==, 1);
 
7142
 
 
7143
  /* Restore signal mask */
 
7144
  g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &expected_sigmask,
 
7145
                                  NULL), ==, 0);
 
7146
}
 
7147
 
 
7148
static void test_restore_sigmask(__attribute__((unused))
 
7149
                                 test_fixture *fixture,
 
7150
                                 __attribute__((unused))
 
7151
                                 gconstpointer user_data){
 
7152
  /* Save original signal mask */
 
7153
  sigset_t orig_sigmask;
 
7154
  g_assert_cmpint(pthread_sigmask(-1, NULL, &orig_sigmask), ==, 0);
 
7155
 
 
7156
  /* Make sure SIGCHLD is blocked for this test */
 
7157
  sigset_t sigchld_sigmask;
 
7158
  g_assert_cmpint(sigemptyset(&sigchld_sigmask), ==, 0);
 
7159
  g_assert_cmpint(sigaddset(&sigchld_sigmask, SIGCHLD), ==, 0);
 
7160
  g_assert_cmpint(pthread_sigmask(SIG_BLOCK, &sigchld_sigmask,
 
7161
                                  NULL), ==, 0);
 
7162
 
 
7163
  /* Act; i.e. run the restore_sigmask() function */
 
7164
  g_assert_true(restore_sigmask(&orig_sigmask));
 
7165
 
 
7166
  /* Retrieve the newly restored signal mask */
 
7167
  sigset_t restored_sigmask;
 
7168
  g_assert_cmpint(pthread_sigmask(-1, NULL, &restored_sigmask),
 
7169
                  ==, 0);
 
7170
 
 
7171
  /* Check the restored_sigmask; it should be the same as the
 
7172
     previously saved signal mask "orig_sigmask". */
 
7173
  for(int signum = 1; signum < NSIG; signum++){
 
7174
    const int orig_block_state = sigismember(&orig_sigmask, signum);
 
7175
    g_assert_cmpint(orig_block_state, >=, 0);
 
7176
    const int restored_block_state = sigismember(&restored_sigmask,
 
7177
                                                 signum);
 
7178
    g_assert_cmpint(restored_block_state, >=, 0);
 
7179
    g_assert_cmpint(restored_block_state, ==, orig_block_state);
 
7180
  }
 
7181
 
 
7182
  /* Restore signal mask */
 
7183
  g_assert_cmpint(pthread_sigmask(SIG_SETMASK, &orig_sigmask,
 
7184
                                  NULL), ==, 0);
 
7185
}
 
7186
 
 
7187
static void test_parse_arguments_noargs(__attribute__((unused))
 
7188
                                        test_fixture *fixture,
 
7189
                                        __attribute__((unused))
 
7190
                                        gconstpointer user_data){
 
7191
  char *argv[] = {
 
7192
    strdup("prgname"),
 
7193
    NULL };
 
7194
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7195
 
 
7196
  char *agent_directory = NULL;
 
7197
  char *helper_directory = NULL;
 
7198
  uid_t user = 0;
 
7199
  gid_t group = 0;
 
7200
  char *mandos_argz = NULL;
 
7201
  size_t mandos_argz_length = 0;
 
7202
 
 
7203
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7204
                                &helper_directory, &user, &group,
 
7205
                                &mandos_argz, &mandos_argz_length));
 
7206
  g_assert_null(agent_directory);
 
7207
  g_assert_null(helper_directory);
 
7208
  g_assert_true(user == 0);
 
7209
  g_assert_true(group == 0);
 
7210
  g_assert_null(mandos_argz);
 
7211
  g_assert_true(mandos_argz_length == 0);
 
7212
 
 
7213
  for(char **arg = argv; *arg != NULL; arg++){
 
7214
    free(*arg);
 
7215
  }
 
7216
}
 
7217
 
 
7218
__attribute__((nonnull))
 
7219
static bool parse_arguments_devnull(int argc, char *argv[],
 
7220
                                    const bool exit_failure,
 
7221
                                    char **agent_directory,
 
7222
                                    char **helper_directory,
 
7223
                                    uid_t *const user,
 
7224
                                    gid_t *const group,
 
7225
                                    char **mandos_argz,
 
7226
                                    size_t *mandos_argz_length){
 
7227
 
 
7228
  FILE *real_stderr = stderr;
 
7229
  FILE *devnull = fopen("/dev/null", "we");
 
7230
  g_assert_nonnull(devnull);
 
7231
  stderr = devnull;
 
7232
 
 
7233
  const bool ret = parse_arguments(argc, argv, exit_failure,
 
7234
                                   agent_directory,
 
7235
                                   helper_directory, user, group,
 
7236
                                   mandos_argz, mandos_argz_length);
 
7237
  const error_t saved_errno = errno;
 
7238
 
 
7239
  stderr = real_stderr;
 
7240
  g_assert_cmpint(fclose(devnull), ==, 0);
 
7241
 
 
7242
  errno = saved_errno;
 
7243
 
 
7244
  return ret;
 
7245
}
 
7246
 
 
7247
static void test_parse_arguments_invalid(__attribute__((unused))
 
7248
                                         test_fixture *fixture,
 
7249
                                         __attribute__((unused))
 
7250
                                         gconstpointer user_data){
 
7251
  char *argv[] = {
 
7252
    strdup("prgname"),
 
7253
    strdup("--invalid"),
 
7254
    NULL };
 
7255
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7256
 
 
7257
  char *agent_directory = NULL;
 
7258
  char *helper_directory = NULL;
 
7259
  uid_t user = 0;
 
7260
  gid_t group = 0;
 
7261
  char *mandos_argz = NULL;
 
7262
  size_t mandos_argz_length = 0;
 
7263
 
 
7264
  g_assert_false(parse_arguments_devnull(argc, argv, false,
 
7265
                                         &agent_directory,
 
7266
                                         &helper_directory, &user,
 
7267
                                         &group, &mandos_argz,
 
7268
                                         &mandos_argz_length));
 
7269
 
 
7270
  g_assert_true(errno == EINVAL);
 
7271
  g_assert_null(agent_directory);
 
7272
  g_assert_null(helper_directory);
 
7273
  g_assert_null(mandos_argz);
 
7274
  g_assert_true(mandos_argz_length == 0);
 
7275
 
 
7276
  for(char **arg = argv; *arg != NULL; arg++){
 
7277
    free(*arg);
 
7278
  }
 
7279
}
 
7280
 
 
7281
static void test_parse_arguments_long_dir(__attribute__((unused))
 
7282
                                          test_fixture *fixture,
 
7283
                                          __attribute__((unused))
 
7284
                                          gconstpointer user_data){
 
7285
  char *argv[] = {
 
7286
    strdup("prgname"),
 
7287
    strdup("--agent-directory"),
 
7288
    strdup("/tmp"),
 
7289
    NULL };
 
7290
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7291
 
 
7292
  __attribute__((cleanup(cleanup_string)))
 
7293
    char *agent_directory = NULL;
 
7294
  char *helper_directory = NULL;
 
7295
  uid_t user = 0;
 
7296
  gid_t group = 0;
 
7297
  __attribute__((cleanup(cleanup_string)))
 
7298
    char *mandos_argz = NULL;
 
7299
  size_t mandos_argz_length = 0;
 
7300
 
 
7301
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7302
                                &helper_directory, &user, &group,
 
7303
                                &mandos_argz, &mandos_argz_length));
 
7304
 
 
7305
  g_assert_cmpstr(agent_directory, ==, "/tmp");
 
7306
  g_assert_null(helper_directory);
 
7307
  g_assert_true(user == 0);
 
7308
  g_assert_true(group == 0);
 
7309
  g_assert_null(mandos_argz);
 
7310
  g_assert_true(mandos_argz_length == 0);
 
7311
 
 
7312
  for(char **arg = argv; *arg != NULL; arg++){
 
7313
    free(*arg);
 
7314
  }
 
7315
}
 
7316
 
 
7317
static void test_parse_arguments_short_dir(__attribute__((unused))
 
7318
                                           test_fixture *fixture,
 
7319
                                           __attribute__((unused))
 
7320
                                           gconstpointer user_data){
 
7321
  char *argv[] = {
 
7322
    strdup("prgname"),
 
7323
    strdup("-d"),
 
7324
    strdup("/tmp"),
 
7325
    NULL };
 
7326
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7327
 
 
7328
  __attribute__((cleanup(cleanup_string)))
 
7329
    char *agent_directory = NULL;
 
7330
  char *helper_directory = NULL;
 
7331
  uid_t user = 0;
 
7332
  gid_t group = 0;
 
7333
  __attribute__((cleanup(cleanup_string)))
 
7334
    char *mandos_argz = NULL;
 
7335
  size_t mandos_argz_length = 0;
 
7336
 
 
7337
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7338
                                &helper_directory, &user, &group,
 
7339
                                &mandos_argz, &mandos_argz_length));
 
7340
 
 
7341
  g_assert_cmpstr(agent_directory, ==, "/tmp");
 
7342
  g_assert_null(helper_directory);
 
7343
  g_assert_true(user == 0);
 
7344
  g_assert_true(group == 0);
 
7345
  g_assert_null(mandos_argz);
 
7346
  g_assert_true(mandos_argz_length == 0);
 
7347
 
 
7348
  for(char **arg = argv; *arg != NULL; arg++){
 
7349
    free(*arg);
 
7350
  }
 
7351
}
 
7352
 
 
7353
static
 
7354
void test_parse_arguments_helper_directory(__attribute__((unused))
 
7355
                                           test_fixture *fixture,
 
7356
                                           __attribute__((unused))
 
7357
                                           gconstpointer user_data){
 
7358
  char *argv[] = {
 
7359
    strdup("prgname"),
 
7360
    strdup("--helper-directory"),
 
7361
    strdup("/tmp"),
 
7362
    NULL };
 
7363
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7364
 
 
7365
  char *agent_directory = NULL;
 
7366
  __attribute__((cleanup(cleanup_string)))
 
7367
    char *helper_directory = NULL;
 
7368
  uid_t user = 0;
 
7369
  gid_t group = 0;
 
7370
  __attribute__((cleanup(cleanup_string)))
 
7371
    char *mandos_argz = NULL;
 
7372
  size_t mandos_argz_length = 0;
 
7373
 
 
7374
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7375
                                &helper_directory, &user, &group,
 
7376
                                &mandos_argz, &mandos_argz_length));
 
7377
 
 
7378
  g_assert_cmpstr(helper_directory, ==, "/tmp");
 
7379
  g_assert_null(agent_directory);
 
7380
  g_assert_true(user == 0);
 
7381
  g_assert_true(group == 0);
 
7382
  g_assert_null(mandos_argz);
 
7383
  g_assert_true(mandos_argz_length == 0);
 
7384
 
 
7385
  for(char **arg = argv; *arg != NULL; arg++){
 
7386
    free(*arg);
 
7387
  }
 
7388
}
 
7389
 
 
7390
static
 
7391
void test_parse_arguments_plugin_helper_dir(__attribute__((unused))
 
7392
                                            test_fixture *fixture,
 
7393
                                            __attribute__((unused))
 
7394
                                            gconstpointer user_data){
 
7395
  char *argv[] = {
 
7396
    strdup("prgname"),
 
7397
    strdup("--plugin-helper-dir"),
 
7398
    strdup("/tmp"),
 
7399
    NULL };
 
7400
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7401
 
 
7402
  char *agent_directory = NULL;
 
7403
  __attribute__((cleanup(cleanup_string)))
 
7404
    char *helper_directory = NULL;
 
7405
  uid_t user = 0;
 
7406
  gid_t group = 0;
 
7407
  __attribute__((cleanup(cleanup_string)))
 
7408
    char *mandos_argz = NULL;
 
7409
  size_t mandos_argz_length = 0;
 
7410
 
 
7411
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7412
                                &helper_directory, &user, &group,
 
7413
                                &mandos_argz, &mandos_argz_length));
 
7414
 
 
7415
  g_assert_cmpstr(helper_directory, ==, "/tmp");
 
7416
  g_assert_null(agent_directory);
 
7417
  g_assert_true(user == 0);
 
7418
  g_assert_true(group == 0);
 
7419
  g_assert_null(mandos_argz);
 
7420
  g_assert_true(mandos_argz_length == 0);
 
7421
 
 
7422
  for(char **arg = argv; *arg != NULL; arg++){
 
7423
    free(*arg);
 
7424
  }
 
7425
}
 
7426
 
 
7427
static void test_parse_arguments_user(__attribute__((unused))
 
7428
                                      test_fixture *fixture,
 
7429
                                      __attribute__((unused))
 
7430
                                      gconstpointer user_data){
 
7431
  char *argv[] = {
 
7432
    strdup("prgname"),
 
7433
    strdup("--user"),
 
7434
    strdup("1000"),
 
7435
    NULL };
 
7436
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7437
 
 
7438
  char *agent_directory = NULL;
 
7439
  __attribute__((cleanup(cleanup_string)))
 
7440
    char *helper_directory = NULL;
 
7441
  uid_t user = 0;
 
7442
  gid_t group = 0;
 
7443
  __attribute__((cleanup(cleanup_string)))
 
7444
    char *mandos_argz = NULL;
 
7445
  size_t mandos_argz_length = 0;
 
7446
 
 
7447
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7448
                                &helper_directory, &user, &group,
 
7449
                                &mandos_argz, &mandos_argz_length));
 
7450
 
 
7451
  g_assert_null(helper_directory);
 
7452
  g_assert_null(agent_directory);
 
7453
  g_assert_cmpuint((unsigned int)user, ==, 1000);
 
7454
  g_assert_true(group == 0);
 
7455
  g_assert_null(mandos_argz);
 
7456
  g_assert_true(mandos_argz_length == 0);
 
7457
 
 
7458
  for(char **arg = argv; *arg != NULL; arg++){
 
7459
    free(*arg);
 
7460
  }
 
7461
}
 
7462
 
 
7463
static void test_parse_arguments_user_invalid(__attribute__((unused))
 
7464
                                              test_fixture *fixture,
 
7465
                                              __attribute__((unused))
 
7466
                                              gconstpointer
 
7467
                                              user_data){
 
7468
  char *argv[] = {
 
7469
    strdup("prgname"),
 
7470
    strdup("--user"),
 
7471
    strdup("invalid"),
 
7472
    NULL };
 
7473
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7474
 
 
7475
  char *agent_directory = NULL;
 
7476
  __attribute__((cleanup(cleanup_string)))
 
7477
    char *helper_directory = NULL;
 
7478
  uid_t user = 0;
 
7479
  gid_t group = 0;
 
7480
  __attribute__((cleanup(cleanup_string)))
 
7481
    char *mandos_argz = NULL;
 
7482
  size_t mandos_argz_length = 0;
 
7483
 
 
7484
  g_assert_false(parse_arguments_devnull(argc, argv, false,
 
7485
                                         &agent_directory,
 
7486
                                         &helper_directory, &user,
 
7487
                                         &group, &mandos_argz,
 
7488
                                         &mandos_argz_length));
 
7489
 
 
7490
  g_assert_null(helper_directory);
 
7491
  g_assert_null(agent_directory);
 
7492
  g_assert_cmpuint((unsigned int)user, ==, 0);
 
7493
  g_assert_true(group == 0);
 
7494
  g_assert_null(mandos_argz);
 
7495
  g_assert_true(mandos_argz_length == 0);
 
7496
 
 
7497
  for(char **arg = argv; *arg != NULL; arg++){
 
7498
    free(*arg);
 
7499
  }
 
7500
}
 
7501
 
 
7502
static
 
7503
void test_parse_arguments_user_zero_invalid(__attribute__((unused))
 
7504
                                            test_fixture *fixture,
 
7505
                                            __attribute__((unused))
 
7506
                                            gconstpointer user_data){
 
7507
  char *argv[] = {
 
7508
    strdup("prgname"),
 
7509
    strdup("--user"),
 
7510
    strdup("0"),
 
7511
    NULL };
 
7512
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7513
 
 
7514
  char *agent_directory = NULL;
 
7515
  __attribute__((cleanup(cleanup_string)))
 
7516
    char *helper_directory = NULL;
 
7517
  uid_t user = 0;
 
7518
  gid_t group = 0;
 
7519
  __attribute__((cleanup(cleanup_string)))
 
7520
    char *mandos_argz = NULL;
 
7521
  size_t mandos_argz_length = 0;
 
7522
 
 
7523
  g_assert_false(parse_arguments_devnull(argc, argv, false,
 
7524
                                         &agent_directory,
 
7525
                                         &helper_directory, &user,
 
7526
                                         &group, &mandos_argz,
 
7527
                                         &mandos_argz_length));
 
7528
 
 
7529
  g_assert_null(helper_directory);
 
7530
  g_assert_null(agent_directory);
 
7531
  g_assert_cmpuint((unsigned int)user, ==, 0);
 
7532
  g_assert_true(group == 0);
 
7533
  g_assert_null(mandos_argz);
 
7534
  g_assert_true(mandos_argz_length == 0);
 
7535
 
 
7536
  for(char **arg = argv; *arg != NULL; arg++){
 
7537
    free(*arg);
 
7538
  }
 
7539
}
 
7540
 
 
7541
static void test_parse_arguments_group(__attribute__((unused))
 
7542
                                       test_fixture *fixture,
 
7543
                                       __attribute__((unused))
 
7544
                                       gconstpointer user_data){
 
7545
  char *argv[] = {
 
7546
    strdup("prgname"),
 
7547
    strdup("--group"),
 
7548
    strdup("1000"),
 
7549
    NULL };
 
7550
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7551
 
 
7552
  char *agent_directory = NULL;
 
7553
  __attribute__((cleanup(cleanup_string)))
 
7554
    char *helper_directory = NULL;
 
7555
  uid_t user = 0;
 
7556
  gid_t group = 0;
 
7557
  __attribute__((cleanup(cleanup_string)))
 
7558
    char *mandos_argz = NULL;
 
7559
  size_t mandos_argz_length = 0;
 
7560
 
 
7561
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7562
                                &helper_directory, &user, &group,
 
7563
                                &mandos_argz, &mandos_argz_length));
 
7564
 
 
7565
  g_assert_null(helper_directory);
 
7566
  g_assert_null(agent_directory);
 
7567
  g_assert_true(user == 0);
 
7568
  g_assert_cmpuint((unsigned int)group, ==, 1000);
 
7569
  g_assert_null(mandos_argz);
 
7570
  g_assert_true(mandos_argz_length == 0);
 
7571
 
 
7572
  for(char **arg = argv; *arg != NULL; arg++){
 
7573
    free(*arg);
 
7574
  }
 
7575
}
 
7576
 
 
7577
static void test_parse_arguments_group_invalid(__attribute__((unused))
 
7578
                                               test_fixture *fixture,
 
7579
                                               __attribute__((unused))
 
7580
                                               gconstpointer
 
7581
                                               user_data){
 
7582
  char *argv[] = {
 
7583
    strdup("prgname"),
 
7584
    strdup("--group"),
 
7585
    strdup("invalid"),
 
7586
    NULL };
 
7587
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7588
 
 
7589
  char *agent_directory = NULL;
 
7590
  __attribute__((cleanup(cleanup_string)))
 
7591
    char *helper_directory = NULL;
 
7592
  uid_t user = 0;
 
7593
  gid_t group = 0;
 
7594
  __attribute__((cleanup(cleanup_string)))
 
7595
    char *mandos_argz = NULL;
 
7596
  size_t mandos_argz_length = 0;
 
7597
 
 
7598
  g_assert_false(parse_arguments_devnull(argc, argv, false,
 
7599
                                         &agent_directory,
 
7600
                                         &helper_directory, &user,
 
7601
                                         &group, &mandos_argz,
 
7602
                                         &mandos_argz_length));
 
7603
 
 
7604
  g_assert_null(helper_directory);
 
7605
  g_assert_null(agent_directory);
 
7606
  g_assert_true(user == 0);
 
7607
  g_assert_true(group == 0);
 
7608
  g_assert_null(mandos_argz);
 
7609
  g_assert_true(mandos_argz_length == 0);
 
7610
 
 
7611
  for(char **arg = argv; *arg != NULL; arg++){
 
7612
    free(*arg);
 
7613
  }
 
7614
}
 
7615
 
 
7616
static
 
7617
void test_parse_arguments_group_zero_invalid(__attribute__((unused))
 
7618
                                             test_fixture *fixture,
 
7619
                                             __attribute__((unused))
 
7620
                                             gconstpointer user_data){
 
7621
  char *argv[] = {
 
7622
    strdup("prgname"),
 
7623
    strdup("--group"),
 
7624
    strdup("0"),
 
7625
    NULL };
 
7626
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7627
 
 
7628
  char *agent_directory = NULL;
 
7629
  __attribute__((cleanup(cleanup_string)))
 
7630
    char *helper_directory = NULL;
 
7631
  uid_t user = 0;
 
7632
  gid_t group = 0;
 
7633
  __attribute__((cleanup(cleanup_string)))
 
7634
    char *mandos_argz = NULL;
 
7635
  size_t mandos_argz_length = 0;
 
7636
 
 
7637
  g_assert_false(parse_arguments_devnull(argc, argv, false,
 
7638
                                         &agent_directory,
 
7639
                                         &helper_directory, &user,
 
7640
                                         &group, &mandos_argz,
 
7641
                                         &mandos_argz_length));
 
7642
 
 
7643
  g_assert_null(helper_directory);
 
7644
  g_assert_null(agent_directory);
 
7645
  g_assert_cmpuint((unsigned int)group, ==, 0);
 
7646
  g_assert_true(group == 0);
 
7647
  g_assert_null(mandos_argz);
 
7648
  g_assert_true(mandos_argz_length == 0);
 
7649
 
 
7650
  for(char **arg = argv; *arg != NULL; arg++){
 
7651
    free(*arg);
 
7652
  }
 
7653
}
 
7654
 
 
7655
static void test_parse_arguments_mandos_noargs(__attribute__((unused))
 
7656
                                               test_fixture *fixture,
 
7657
                                               __attribute__((unused))
 
7658
                                               gconstpointer
 
7659
                                               user_data){
 
7660
  char *argv[] = {
 
7661
    strdup("prgname"),
 
7662
    strdup("mandos-client"),
 
7663
    NULL };
 
7664
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7665
 
 
7666
  __attribute__((cleanup(cleanup_string)))
 
7667
    char *agent_directory = NULL;
 
7668
  __attribute__((cleanup(cleanup_string)))
 
7669
    char *helper_directory = NULL;
 
7670
  uid_t user = 0;
 
7671
  gid_t group = 0;
 
7672
  __attribute__((cleanup(cleanup_string)))
 
7673
    char *mandos_argz = NULL;
 
7674
  size_t mandos_argz_length = 0;
 
7675
 
 
7676
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7677
                                &helper_directory, &user, &group,
 
7678
                                &mandos_argz, &mandos_argz_length));
 
7679
 
 
7680
  g_assert_null(agent_directory);
 
7681
  g_assert_null(helper_directory);
 
7682
  g_assert_true(user == 0);
 
7683
  g_assert_true(group == 0);
 
7684
  g_assert_cmpstr(mandos_argz, ==, "mandos-client");
 
7685
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
 
7686
                                            mandos_argz_length),
 
7687
                   ==, 1);
 
7688
 
 
7689
  for(char **arg = argv; *arg != NULL; arg++){
 
7690
    free(*arg);
 
7691
  }
 
7692
}
 
7693
 
 
7694
static void test_parse_arguments_mandos_args(__attribute__((unused))
 
7695
                                             test_fixture *fixture,
 
7696
                                             __attribute__((unused))
 
7697
                                             gconstpointer user_data){
 
7698
  char *argv[] = {
 
7699
    strdup("prgname"),
 
7700
    strdup("mandos-client"),
 
7701
    strdup("one"),
 
7702
    strdup("two"),
 
7703
    strdup("three"),
 
7704
    NULL };
 
7705
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7706
 
 
7707
  __attribute__((cleanup(cleanup_string)))
 
7708
    char *agent_directory = NULL;
 
7709
  __attribute__((cleanup(cleanup_string)))
 
7710
    char *helper_directory = NULL;
 
7711
  uid_t user = 0;
 
7712
  gid_t group = 0;
 
7713
  __attribute__((cleanup(cleanup_string)))
 
7714
    char *mandos_argz = NULL;
 
7715
  size_t mandos_argz_length = 0;
 
7716
 
 
7717
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7718
                                &helper_directory, &user, &group,
 
7719
                                &mandos_argz, &mandos_argz_length));
 
7720
 
 
7721
  g_assert_null(agent_directory);
 
7722
  g_assert_null(helper_directory);
 
7723
  g_assert_true(user == 0);
 
7724
  g_assert_true(group == 0);
 
7725
  char *marg = mandos_argz;
 
7726
  g_assert_cmpstr(marg, ==, "mandos-client");
 
7727
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7728
  g_assert_cmpstr(marg, ==, "one");
 
7729
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7730
  g_assert_cmpstr(marg, ==, "two");
 
7731
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7732
  g_assert_cmpstr(marg, ==, "three");
 
7733
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
 
7734
                                            mandos_argz_length),
 
7735
                   ==, 4);
 
7736
 
 
7737
  for(char **arg = argv; *arg != NULL; arg++){
 
7738
    free(*arg);
 
7739
  }
 
7740
}
 
7741
 
 
7742
static void test_parse_arguments_all_args(__attribute__((unused))
 
7743
                                          test_fixture *fixture,
 
7744
                                          __attribute__((unused))
 
7745
                                          gconstpointer user_data){
 
7746
  char *argv[] = {
 
7747
    strdup("prgname"),
 
7748
    strdup("--agent-directory"),
 
7749
    strdup("/tmp"),
 
7750
    strdup("--helper-directory"),
 
7751
    strdup("/var/tmp"),
 
7752
    strdup("--user"),
 
7753
    strdup("1"),
 
7754
    strdup("--group"),
 
7755
    strdup("2"),
 
7756
    strdup("mandos-client"),
 
7757
    strdup("one"),
 
7758
    strdup("two"),
 
7759
    strdup("three"),
 
7760
    NULL };
 
7761
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7762
 
 
7763
  __attribute__((cleanup(cleanup_string)))
 
7764
    char *agent_directory = NULL;
 
7765
  __attribute__((cleanup(cleanup_string)))
 
7766
    char *helper_directory = NULL;
 
7767
  uid_t user = 0;
 
7768
  gid_t group = 0;
 
7769
  __attribute__((cleanup(cleanup_string)))
 
7770
    char *mandos_argz = NULL;
 
7771
  size_t mandos_argz_length = 0;
 
7772
 
 
7773
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7774
                                &helper_directory, &user, &group,
 
7775
                                &mandos_argz, &mandos_argz_length));
 
7776
 
 
7777
  g_assert_cmpstr(agent_directory, ==, "/tmp");
 
7778
  g_assert_cmpstr(helper_directory, ==, "/var/tmp");
 
7779
  g_assert_true(user == 1);
 
7780
  g_assert_true(group == 2);
 
7781
  char *marg = mandos_argz;
 
7782
  g_assert_cmpstr(marg, ==, "mandos-client");
 
7783
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7784
  g_assert_cmpstr(marg, ==, "one");
 
7785
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7786
  g_assert_cmpstr(marg, ==, "two");
 
7787
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7788
  g_assert_cmpstr(marg, ==, "three");
 
7789
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
 
7790
                                            mandos_argz_length),
 
7791
                   ==, 4);
 
7792
 
 
7793
  for(char **arg = argv; *arg != NULL; arg++){
 
7794
    free(*arg);
 
7795
  }
 
7796
}
 
7797
 
 
7798
static void test_parse_arguments_mixed(__attribute__((unused))
 
7799
                                       test_fixture *fixture,
 
7800
                                       __attribute__((unused))
 
7801
                                       gconstpointer user_data){
 
7802
  char *argv[] = {
 
7803
    strdup("prgname"),
 
7804
    strdup("mandos-client"),
 
7805
    strdup("--user"),
 
7806
    strdup("1"),
 
7807
    strdup("one"),
 
7808
    strdup("--agent-directory"),
 
7809
    strdup("/tmp"),
 
7810
    strdup("two"),
 
7811
    strdup("three"),
 
7812
    strdup("--helper-directory=/var/tmp"),
 
7813
    NULL };
 
7814
  const int argc = (sizeof(argv) / sizeof(char *)) - 1;
 
7815
 
 
7816
  __attribute__((cleanup(cleanup_string)))
 
7817
    char *agent_directory = NULL;
 
7818
  __attribute__((cleanup(cleanup_string)))
 
7819
    char *helper_directory = NULL;
 
7820
  uid_t user = 0;
 
7821
  gid_t group = 0;
 
7822
  __attribute__((cleanup(cleanup_string)))
 
7823
    char *mandos_argz = NULL;
 
7824
  size_t mandos_argz_length = 0;
 
7825
 
 
7826
  g_assert_true(parse_arguments(argc, argv, false, &agent_directory,
 
7827
                                &helper_directory, &user, &group,
 
7828
                                &mandos_argz, &mandos_argz_length));
 
7829
 
 
7830
  g_assert_cmpstr(agent_directory, ==, "/tmp");
 
7831
  g_assert_cmpstr(helper_directory, ==, "/var/tmp");
 
7832
  g_assert_true(user == 1);
 
7833
  g_assert_true(group == 0);
 
7834
  char *marg = mandos_argz;
 
7835
  g_assert_cmpstr(marg, ==, "mandos-client");
 
7836
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7837
  g_assert_cmpstr(marg, ==, "one");
 
7838
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7839
  g_assert_cmpstr(marg, ==, "two");
 
7840
  marg = argz_next(mandos_argz, mandos_argz_length, marg);
 
7841
  g_assert_cmpstr(marg, ==, "three");
 
7842
  g_assert_cmpuint((unsigned int)argz_count(mandos_argz,
 
7843
                                            mandos_argz_length),
 
7844
                   ==, 4);
 
7845
 
 
7846
  for(char **arg = argv; *arg != NULL; arg++){
 
7847
    free(*arg);
 
7848
  }
 
7849
}
 
7850
 
 
7851
/* End of tests section */
 
7852
 
 
7853
/* Test boilerplate section; New tests should be added to the test
 
7854
   suite definition here, in the "run_tests" function.
 
7855
 
 
7856
   Finally, this section also contains the should_only_run_tests()
 
7857
   function used by main() for deciding if tests should be run or to
 
7858
   start normally. */
 
7859
 
 
7860
__attribute__((cold))
 
7861
static bool run_tests(int argc, char *argv[]){
 
7862
  g_test_init(&argc, &argv, NULL);
 
7863
 
 
7864
  /* A macro to add a test with no setup or teardown functions */
 
7865
#define test_add(testpath, testfunc)                    \
 
7866
  do {                                                  \
 
7867
    g_test_add((testpath), test_fixture, NULL, NULL,    \
 
7868
               (testfunc), NULL);                       \
 
7869
  } while(false)
 
7870
 
 
7871
  /* Test the signal-related functions first, since some other tests
 
7872
     depend on these functions in their setups and teardowns */
 
7873
  test_add("/signal-handling/setup", test_setup_signal_handler);
 
7874
  test_add("/signal-handling/restore", test_restore_signal_handler);
 
7875
  test_add("/signal-handling/block", test_block_sigchld);
 
7876
  test_add("/signal-handling/restore-sigmask", test_restore_sigmask);
 
7877
 
 
7878
  /* Regular non-signal-related tests; these use no setups or
 
7879
     teardowns */
 
7880
  test_add("/parse_arguments/noargs", test_parse_arguments_noargs);
 
7881
  test_add("/parse_arguments/invalid", test_parse_arguments_invalid);
 
7882
  test_add("/parse_arguments/long-dir",
 
7883
           test_parse_arguments_long_dir);
 
7884
  test_add("/parse_arguments/short-dir",
 
7885
           test_parse_arguments_short_dir);
 
7886
  test_add("/parse_arguments/helper-directory",
 
7887
           test_parse_arguments_helper_directory);
 
7888
  test_add("/parse_arguments/plugin-helper-dir",
 
7889
           test_parse_arguments_plugin_helper_dir);
 
7890
  test_add("/parse_arguments/user", test_parse_arguments_user);
 
7891
  test_add("/parse_arguments/user-invalid",
 
7892
           test_parse_arguments_user_invalid);
 
7893
  test_add("/parse_arguments/user-zero-invalid",
 
7894
           test_parse_arguments_user_zero_invalid);
 
7895
  test_add("/parse_arguments/group", test_parse_arguments_group);
 
7896
  test_add("/parse_arguments/group-invalid",
 
7897
           test_parse_arguments_group_invalid);
 
7898
  test_add("/parse_arguments/group-zero-invalid",
 
7899
           test_parse_arguments_group_zero_invalid);
 
7900
  test_add("/parse_arguments/mandos-noargs",
 
7901
           test_parse_arguments_mandos_noargs);
 
7902
  test_add("/parse_arguments/mandos-args",
 
7903
           test_parse_arguments_mandos_args);
 
7904
  test_add("/parse_arguments/all-args",
 
7905
           test_parse_arguments_all_args);
 
7906
  test_add("/parse_arguments/mixed", test_parse_arguments_mixed);
 
7907
  test_add("/queue/create", test_create_queue);
 
7908
  test_add("/queue/add", test_add_to_queue);
 
7909
  test_add("/queue/add/overflow", test_add_to_queue_overflow);
 
7910
  test_add("/queue/has_question/empty",
 
7911
           test_queue_has_question_empty);
 
7912
  test_add("/queue/has_question/false",
 
7913
           test_queue_has_question_false);
 
7914
  test_add("/queue/has_question/true", test_queue_has_question_true);
 
7915
  test_add("/queue/has_question/false2",
 
7916
           test_queue_has_question_false2);
 
7917
  test_add("/queue/has_question/true2",
 
7918
           test_queue_has_question_true2);
 
7919
  test_add("/buffer/cleanup", test_cleanup_buffer);
 
7920
  test_add("/string_set/net-set-contains-nothing",
 
7921
           test_string_set_new_set_contains_nothing);
 
7922
  test_add("/string_set/with-added-string-contains-it",
 
7923
           test_string_set_with_added_string_contains_it);
 
7924
  test_add("/string_set/cleared-does-not-contain-string",
 
7925
           test_string_set_cleared_does_not_contain_str);
 
7926
  test_add("/string_set/swap/one-with-empty",
 
7927
           test_string_set_swap_one_with_empty);
 
7928
  test_add("/string_set/swap/empty-with-one",
 
7929
           test_string_set_swap_empty_with_one);
 
7930
  test_add("/string_set/swap/one-with-one",
 
7931
           test_string_set_swap_one_with_one);
 
7932
 
 
7933
  /* A macro to add a test using the setup and teardown functions */
 
7934
#define test_add_st(path, func)                                 \
 
7935
  do {                                                          \
 
7936
    g_test_add((path), test_fixture, NULL, test_setup, (func),  \
 
7937
               test_teardown);                                  \
 
7938
  } while(false)
 
7939
 
 
7940
  /* Signal-related tests; these use setups and teardowns which
 
7941
     establish, during each test run, a signal handler for, and a
 
7942
     signal mask blocking, the SIGCHLD signal, just like main() */
 
7943
  test_add_st("/wait_for_event/timeout", test_wait_for_event_timeout);
 
7944
  test_add_st("/wait_for_event/event", test_wait_for_event_event);
 
7945
  test_add_st("/wait_for_event/sigchld", test_wait_for_event_sigchld);
 
7946
  test_add_st("/run_queue/zeroes-next-run",
 
7947
              test_run_queue_zeroes_next_run);
 
7948
  test_add_st("/run_queue/clears-cancelled_filenames",
 
7949
              test_run_queue_clears_cancelled_filenames);
 
7950
  test_add_st("/run_queue/skips-cancelled-filenames",
 
7951
              test_run_queue_skips_cancelled_filenames);
 
7952
  test_add_st("/run_queue/one-task", test_run_queue_one_task);
 
7953
  test_add_st("/run_queue/two-tasks", test_run_queue_two_tasks);
 
7954
  test_add_st("/run_queue/two-tasks/quit",
 
7955
              test_run_queue_two_tasks_quit);
 
7956
  test_add_st("/run_queue/two-tasks-cleanup",
 
7957
              test_run_queue_two_tasks_cleanup);
 
7958
  test_add_st("/task-creators/start_mandos_client",
 
7959
              test_start_mandos_client);
 
7960
  test_add_st("/task-creators/start_mandos_client/execv",
 
7961
              test_start_mandos_client_execv);
 
7962
  test_add_st("/task-creators/start_mandos_client/suid/euid",
 
7963
              test_start_mandos_client_suid_euid);
 
7964
  test_add_st("/task-creators/start_mandos_client/suid/egid",
 
7965
              test_start_mandos_client_suid_egid);
 
7966
  test_add_st("/task-creators/start_mandos_client/suid/ruid",
 
7967
              test_start_mandos_client_suid_ruid);
 
7968
  test_add_st("/task-creators/start_mandos_client/suid/rgid",
 
7969
              test_start_mandos_client_suid_rgid);
 
7970
  test_add_st("/task-creators/start_mandos_client/read",
 
7971
              test_start_mandos_client_read);
 
7972
  test_add_st("/task-creators/start_mandos_client/helper-directory",
 
7973
              test_start_mandos_client_helper_directory);
 
7974
  test_add_st("/task-creators/start_mandos_client/sigmask",
 
7975
              test_start_mandos_client_sigmask);
 
7976
  test_add_st("/task/wait_for_mandos_client_exit/badpid",
 
7977
              test_wait_for_mandos_client_exit_badpid);
 
7978
  test_add_st("/task/wait_for_mandos_client_exit/noexit",
 
7979
              test_wait_for_mandos_client_exit_noexit);
 
7980
  test_add_st("/task/wait_for_mandos_client_exit/success",
 
7981
              test_wait_for_mandos_client_exit_success);
 
7982
  test_add_st("/task/wait_for_mandos_client_exit/failure",
 
7983
              test_wait_for_mandos_client_exit_failure);
 
7984
  test_add_st("/task/wait_for_mandos_client_exit/killed",
 
7985
              test_wait_for_mandos_client_exit_killed);
 
7986
  test_add_st("/task/read_mandos_client_output/readerror",
 
7987
              test_read_mandos_client_output_readerror);
 
7988
  test_add_st("/task/read_mandos_client_output/nodata",
 
7989
              test_read_mandos_client_output_nodata);
 
7990
  test_add_st("/task/read_mandos_client_output/eof",
 
7991
              test_read_mandos_client_output_eof);
 
7992
  test_add_st("/task/read_mandos_client_output/once",
 
7993
              test_read_mandos_client_output_once);
 
7994
  test_add_st("/task/read_mandos_client_output/malloc",
 
7995
              test_read_mandos_client_output_malloc);
 
7996
  test_add_st("/task/read_mandos_client_output/append",
 
7997
              test_read_mandos_client_output_append);
 
7998
  test_add_st("/task-creators/add_inotify_dir_watch",
 
7999
              test_add_inotify_dir_watch);
 
8000
  test_add_st("/task-creators/add_inotify_dir_watch/fail",
 
8001
              test_add_inotify_dir_watch_fail);
 
8002
  test_add_st("/task-creators/add_inotify_dir_watch/not-a-directory",
 
8003
              test_add_inotify_dir_watch_nondir);
 
8004
  test_add_st("/task-creators/add_inotify_dir_watch/EAGAIN",
 
8005
              test_add_inotify_dir_watch_EAGAIN);
 
8006
  test_add_st("/task-creators/add_inotify_dir_watch/IN_CLOSE_WRITE",
 
8007
              test_add_inotify_dir_watch_IN_CLOSE_WRITE);
 
8008
  test_add_st("/task-creators/add_inotify_dir_watch/IN_MOVED_TO",
 
8009
              test_add_inotify_dir_watch_IN_MOVED_TO);
 
8010
  test_add_st("/task-creators/add_inotify_dir_watch/IN_MOVED_FROM",
 
8011
              test_add_inotify_dir_watch_IN_MOVED_FROM);
 
8012
  test_add_st("/task-creators/add_inotify_dir_watch/IN_EXCL_UNLINK",
 
8013
              test_add_inotify_dir_watch_IN_EXCL_UNLINK);
 
8014
  test_add_st("/task-creators/add_inotify_dir_watch/IN_DELETE",
 
8015
              test_add_inotify_dir_watch_IN_DELETE);
 
8016
  test_add_st("/task/read_inotify_event/readerror",
 
8017
              test_read_inotify_event_readerror);
 
8018
  test_add_st("/task/read_inotify_event/bad-epoll",
 
8019
              test_read_inotify_event_bad_epoll);
 
8020
  test_add_st("/task/read_inotify_event/nodata",
 
8021
              test_read_inotify_event_nodata);
 
8022
  test_add_st("/task/read_inotify_event/eof",
 
8023
              test_read_inotify_event_eof);
 
8024
  test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE",
 
8025
              test_read_inotify_event_IN_CLOSE_WRITE);
 
8026
  test_add_st("/task/read_inotify_event/IN_MOVED_TO",
 
8027
              test_read_inotify_event_IN_MOVED_TO);
 
8028
  test_add_st("/task/read_inotify_event/IN_MOVED_FROM",
 
8029
              test_read_inotify_event_IN_MOVED_FROM);
 
8030
  test_add_st("/task/read_inotify_event/IN_DELETE",
 
8031
              test_read_inotify_event_IN_DELETE);
 
8032
  test_add_st("/task/read_inotify_event/IN_CLOSE_WRITE/badname",
 
8033
              test_read_inotify_event_IN_CLOSE_WRITE_badname);
 
8034
  test_add_st("/task/read_inotify_event/IN_MOVED_TO/badname",
 
8035
              test_read_inotify_event_IN_MOVED_TO_badname);
 
8036
  test_add_st("/task/read_inotify_event/IN_MOVED_FROM/badname",
 
8037
              test_read_inotify_event_IN_MOVED_FROM_badname);
 
8038
  test_add_st("/task/read_inotify_event/IN_DELETE/badname",
 
8039
              test_read_inotify_event_IN_DELETE_badname);
 
8040
  test_add_st("/task/open_and_parse_question/ENOENT",
 
8041
              test_open_and_parse_question_ENOENT);
 
8042
  test_add_st("/task/open_and_parse_question/EIO",
 
8043
              test_open_and_parse_question_EIO);
 
8044
  test_add_st("/task/open_and_parse_question/parse-error",
 
8045
              test_open_and_parse_question_parse_error);
 
8046
  test_add_st("/task/open_and_parse_question/nosocket",
 
8047
              test_open_and_parse_question_nosocket);
 
8048
  test_add_st("/task/open_and_parse_question/badsocket",
 
8049
              test_open_and_parse_question_badsocket);
 
8050
  test_add_st("/task/open_and_parse_question/nopid",
 
8051
              test_open_and_parse_question_nopid);
 
8052
  test_add_st("/task/open_and_parse_question/badpid",
 
8053
              test_open_and_parse_question_badpid);
 
8054
  test_add_st("/task/open_and_parse_question/noexist_pid",
 
8055
              test_open_and_parse_question_noexist_pid);
 
8056
  test_add_st("/task/open_and_parse_question/no-notafter",
 
8057
              test_open_and_parse_question_no_notafter);
 
8058
  test_add_st("/task/open_and_parse_question/bad-notafter",
 
8059
              test_open_and_parse_question_bad_notafter);
 
8060
  test_add_st("/task/open_and_parse_question/notafter-0",
 
8061
              test_open_and_parse_question_notafter_0);
 
8062
  test_add_st("/task/open_and_parse_question/notafter-1",
 
8063
              test_open_and_parse_question_notafter_1);
 
8064
  test_add_st("/task/open_and_parse_question/notafter-1-1",
 
8065
              test_open_and_parse_question_notafter_1_1);
 
8066
  test_add_st("/task/open_and_parse_question/notafter-1-2",
 
8067
              test_open_and_parse_question_notafter_1_2);
 
8068
  test_add_st("/task/open_and_parse_question/equal-notafter",
 
8069
              test_open_and_parse_question_equal_notafter);
 
8070
  test_add_st("/task/open_and_parse_question/late-notafter",
 
8071
              test_open_and_parse_question_late_notafter);
 
8072
  test_add_st("/task/cancel_old_question/0-1-2",
 
8073
              test_cancel_old_question_0_1_2);
 
8074
  test_add_st("/task/cancel_old_question/0-2-1",
 
8075
              test_cancel_old_question_0_2_1);
 
8076
  test_add_st("/task/cancel_old_question/1-2-3",
 
8077
              test_cancel_old_question_1_2_3);
 
8078
  test_add_st("/task/cancel_old_question/1-3-2",
 
8079
              test_cancel_old_question_1_3_2);
 
8080
  test_add_st("/task/cancel_old_question/2-1-3",
 
8081
              test_cancel_old_question_2_1_3);
 
8082
  test_add_st("/task/cancel_old_question/2-3-1",
 
8083
              test_cancel_old_question_2_3_1);
 
8084
  test_add_st("/task/cancel_old_question/3-1-2",
 
8085
              test_cancel_old_question_3_1_2);
 
8086
  test_add_st("/task/cancel_old_question/3-2-1",
 
8087
              test_cancel_old_question_3_2_1);
 
8088
  test_add_st("/task/connect_question_socket/name-too-long",
 
8089
              test_connect_question_socket_name_too_long);
 
8090
  test_add_st("/task/connect_question_socket/connect-fail",
 
8091
              test_connect_question_socket_connect_fail);
 
8092
  test_add_st("/task/connect_question_socket/bad-epoll",
 
8093
              test_connect_question_socket_bad_epoll);
 
8094
  test_add_st("/task/connect_question_socket/usable",
 
8095
              test_connect_question_socket_usable);
 
8096
  test_add_st("/task/send_password_to_socket/client-not-exited",
 
8097
              test_send_password_to_socket_client_not_exited);
 
8098
  test_add_st("/task/send_password_to_socket/password-not-read",
 
8099
              test_send_password_to_socket_password_not_read);
 
8100
  test_add_st("/task/send_password_to_socket/EMSGSIZE",
 
8101
              test_send_password_to_socket_EMSGSIZE);
 
8102
  test_add_st("/task/send_password_to_socket/retry",
 
8103
              test_send_password_to_socket_retry);
 
8104
  test_add_st("/task/send_password_to_socket/bad-epoll",
 
8105
              test_send_password_to_socket_bad_epoll);
 
8106
  test_add_st("/task/send_password_to_socket/null-password",
 
8107
              test_send_password_to_socket_null_password);
 
8108
  test_add_st("/task/send_password_to_socket/empty-password",
 
8109
              test_send_password_to_socket_empty_password);
 
8110
  test_add_st("/task/send_password_to_socket/empty-str-password",
 
8111
              test_send_password_to_socket_empty_str_pass);
 
8112
  test_add_st("/task/send_password_to_socket/text-password",
 
8113
              test_send_password_to_socket_text_password);
 
8114
  test_add_st("/task/send_password_to_socket/binary-password",
 
8115
              test_send_password_to_socket_binary_password);
 
8116
  test_add_st("/task/send_password_to_socket/nuls-in-password",
 
8117
              test_send_password_to_socket_nuls_in_password);
 
8118
  test_add_st("/task-creators/add_existing_questions/ENOENT",
 
8119
              test_add_existing_questions_ENOENT);
 
8120
  test_add_st("/task-creators/add_existing_questions/no-questions",
 
8121
              test_add_existing_questions_no_questions);
 
8122
  test_add_st("/task-creators/add_existing_questions/one-question",
 
8123
              test_add_existing_questions_one_question);
 
8124
  test_add_st("/task-creators/add_existing_questions/two-questions",
 
8125
              test_add_existing_questions_two_questions);
 
8126
  test_add_st("/task-creators/add_existing_questions/non-questions",
 
8127
              test_add_existing_questions_non_questions);
 
8128
  test_add_st("/task-creators/add_existing_questions/both-types",
 
8129
              test_add_existing_questions_both_types);
 
8130
 
 
8131
  return g_test_run() == 0;
 
8132
}
 
8133
 
 
8134
static bool should_only_run_tests(int *argc_p, char **argv_p[]){
 
8135
  GOptionContext *context = g_option_context_new("");
 
8136
 
 
8137
  g_option_context_set_help_enabled(context, FALSE);
 
8138
  g_option_context_set_ignore_unknown_options(context, TRUE);
 
8139
 
 
8140
  gboolean should_run_tests = FALSE;
 
8141
  GOptionEntry entries[] = {
 
8142
    { "test", 0, 0, G_OPTION_ARG_NONE,
 
8143
      &should_run_tests, "Run tests", NULL },
 
8144
    { NULL }
 
8145
  };
 
8146
  g_option_context_add_main_entries(context, entries, NULL);
 
8147
 
 
8148
  GError *error = NULL;
 
8149
 
 
8150
  if(g_option_context_parse(context, argc_p, argv_p, &error) != TRUE){
 
8151
    g_option_context_free(context);
 
8152
    g_error("Failed to parse options: %s", error->message);
 
8153
  }
 
8154
 
 
8155
  g_option_context_free(context);
 
8156
  return should_run_tests != FALSE;
 
8157
}