49
49
#include <sysexits.h> /* EX_USAGE, EX_OSERR, EX_OSFILE */
50
50
#include <errno.h> /* errno, error_t, EACCES,
51
51
ENAMETOOLONG, ENOENT, ENOTDIR,
52
ENOMEM, EEXIST, ECHILD, EPERM,
52
EEXIST, ECHILD, EPERM, ENOMEM,
53
53
EAGAIN, EINTR, ENOBUFS, EADDRINUSE,
54
54
ECONNREFUSED, ECONNRESET,
55
55
ETOOMANYREFS, EMSGSIZE, EBADF,
73
73
ARGP_ERR_UNKNOWN, ARGP_KEY_ARGS,
74
74
struct argp, argp_parse(),
76
#include <stdint.h> /* SIZE_MAX */
77
76
#include <unistd.h> /* uid_t, gid_t, close(), pipe2(),
78
77
fork(), _exit(), dup2(),
79
78
STDOUT_FILENO, setresgid(),
96
95
IN_EXCL_UNLINK, IN_ONLYDIR,
97
96
struct inotify_event */
98
97
#include <fnmatch.h> /* fnmatch(), FNM_FILE_NAME */
99
#include <stdio.h> /* asprintf(), FILE, stderr, fopen(),
100
fclose(), getline(), sscanf(),
101
feof(), ferror(), rename(),
102
fdopen(), fprintf(), fscanf() */
98
#include <stdio.h> /* asprintf(), FILE, fopen(),
99
getline(), sscanf(), feof(),
100
ferror(), fclose(), stderr,
101
rename(), fdopen(), fprintf(),
103
103
#include <glib.h> /* GKeyFile, g_key_file_free(), g_key_file_new(),
104
104
GError, g_key_file_load_from_file(),
105
105
G_KEY_FILE_NONE, TRUE, G_FILE_ERROR_NOENT,
148
148
mono_microsecs next_run;
149
149
} __attribute__((designated_init)) task_queue;
151
/* "task_func" - A function type for task functions
151
/* "func_type" - A function type for task functions
153
153
I.e. functions for the code which runs when a task is run, all have
652
652
__attribute__((nonnull, warn_unused_result))
653
653
bool add_to_queue(task_queue *const queue, const task_context task){
654
if((queue->length + 1) > (SIZE_MAX / sizeof(task_context))){
656
error(0, ENOMEM, "Failed to allocate %" PRIuMAX
657
" tasks for queue->tasks", (uintmax_t)(queue->length + 1));
661
654
const size_t needed_size = sizeof(task_context)*(queue->length + 1);
662
655
if(needed_size > (queue->allocated)){
663
656
task_context *const new_tasks = realloc(queue->tasks,
869
862
close(pipefds[1]);
872
error(0, errno, "Failed to fork()");
877
864
if(not add_to_queue(queue, (task_context){
878
865
.func=wait_for_mandos_client_exit,
1897
1884
g_assert_true(queue->tasks[0].func == dummy_func);
1900
static void test_add_to_queue_overflow(__attribute__((unused))
1901
test_fixture *fixture,
1902
__attribute__((unused))
1903
gconstpointer user_data){
1904
__attribute__((cleanup(cleanup_queue)))
1905
task_queue *queue = create_queue();
1906
g_assert_nonnull(queue);
1907
g_assert_true(queue->length == 0);
1908
queue->length = SIZE_MAX / sizeof(task_context); /* fake max size */
1910
FILE *real_stderr = stderr;
1911
FILE *devnull = fopen("/dev/null", "we");
1912
g_assert_nonnull(devnull);
1914
const bool ret = add_to_queue(queue,
1915
(task_context){ .func=dummy_func });
1916
g_assert_true(errno == ENOMEM);
1917
g_assert_false(ret);
1918
stderr = real_stderr;
1919
g_assert_cmpint(fclose(devnull), ==, 0);
1920
queue->length = 0; /* Restore real size */
1923
1887
static void dummy_func(__attribute__((unused))
1924
1888
const task_context task,
1925
1889
__attribute__((unused))
2197
2161
exit(EXIT_SUCCESS);
2200
error(EXIT_FAILURE, errno, "Failed to fork()");
2204
2164
waitpid(pid, &status, 0);
2205
2165
if(WIFEXITED(status) and (WEXITSTATUS(status) == EXIT_SUCCESS)){
5958
5918
test_fixture *fixture,
5959
5919
__attribute__((unused))
5960
5920
gconstpointer user_data){
5962
g_test_skip("Skipping EMSGSIZE test on non-AMD64 platform");
5964
5921
__attribute__((cleanup(cleanup_close)))
5965
5922
const int epoll_fd = epoll_create1(EPOLL_CLOEXEC);
5966
5923
g_assert_cmpint(epoll_fd, >=, 0);
6019
5976
g_assert_cmpuint((unsigned int)queue->length, ==, 0);
6020
5977
g_assert_true(string_set_contains(cancelled_filenames,
6021
5978
question_filename));
6025
5981
static void test_send_password_to_socket_retry(__attribute__((unused))
7906
7862
test_add("/parse_arguments/mixed", test_parse_arguments_mixed);
7907
7863
test_add("/queue/create", test_create_queue);
7908
7864
test_add("/queue/add", test_add_to_queue);
7909
test_add("/queue/add/overflow", test_add_to_queue_overflow);
7910
7865
test_add("/queue/has_question/empty",
7911
7866
test_queue_has_question_empty);
7912
7867
test_add("/queue/has_question/false",
8137
8092
g_option_context_set_help_enabled(context, FALSE);
8138
8093
g_option_context_set_ignore_unknown_options(context, TRUE);
8140
gboolean should_run_tests = FALSE;
8095
gboolean run_tests = FALSE;
8141
8096
GOptionEntry entries[] = {
8142
8097
{ "test", 0, 0, G_OPTION_ARG_NONE,
8143
&should_run_tests, "Run tests", NULL },
8098
&run_tests, "Run tests", NULL },
8146
8101
g_option_context_add_main_entries(context, entries, NULL);