/mandos/trunk

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

« back to all changes in this revision

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

  • Committer: Teddy Hogeborn
  • Date: 2009-01-17 00:26:13 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090117002613-lg5x2b8mnmi3alqy
Fix problem found using cppcheck, thanks to George Danchev
<danchev@spnet.net> for the report and fix:

* plugin-runner.c (getplugin): Bug fix: Free the plugin struct in case
                               of strdup failure.

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