/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

* initramfs-tools-hook: Bug fix: Add "--userid" and "--groupid" to
                        start of "plugin-runner.conf" file instead of
                        appending, to allow any preexisting options to
                        override.
* plugin-runner.conf: Improved wording.

Show diffs side-by-side

added added

removed removed

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