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