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