diff options
| -rw-r--r-- | meson.build | 9 | ||||
| -rw-r--r-- | tests/connection.c | 176 |
2 files changed, 185 insertions, 0 deletions
diff --git a/meson.build b/meson.build index 1ab3d87..a1c758c 100644 --- a/meson.build +++ b/meson.build @@ -203,6 +203,15 @@ test( ) ) +test( + 'connection', + executable( + 'connection_test', + ['common/connection.c', 'tests/connection.c'], + include_directories: [include_directories('.', 'include')], + ) +) + scdoc = dependency('scdoc', required: get_option('man-pages'), version: '>= 1.9.7', native: true) if scdoc.found() diff --git a/tests/connection.c b/tests/connection.c new file mode 100644 index 0000000..551d799 --- /dev/null +++ b/tests/connection.c @@ -0,0 +1,176 @@ +#include <errno.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/socket.h> +#include <sys/types.h> +#include <unistd.h> + +#include "connection.h" +#include "test.h" + +static void test_send_one_byte(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + char in = 85, out = 0; + + test_assert(connection_put(&c1, &in, sizeof in) == 0); + test_assert(connection_flush(&c1) == sizeof in); + + test_assert(connection_read(&c2) == sizeof out); + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out) == sizeof out); + test_assert(out == in); + + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +static void test_short_read(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + char in = 85, out = 0; + int out_large = 0; + + test_assert(connection_put(&c1, &in, sizeof in) == 0); + test_assert(connection_flush(&c1) > 0); + + test_assert(connection_read(&c2) > 0); + + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out_large) == -1); + test_assert(errno == EAGAIN); + + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out) == sizeof out); + + test_assert(out == in); + + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +static void test_long_write(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + char in[CONNECTION_BUFFER_SIZE + 1]; + memset(in, 0, sizeof in); + + test_assert(connection_put(&c1, &in, sizeof in) == -1); + test_assert(errno = EAGAIN); + + test_assert(connection_read(&c2) == -1 && errno == EAGAIN); + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +static void test_send_one_int(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + int in = 0xDEADBEEF, out = 0; + + test_assert(connection_put(&c1, &in, sizeof in) == 0); + test_assert(connection_flush(&c1) == sizeof in); + + test_assert(connection_read(&c2) == sizeof out); + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out) == sizeof out); + + test_assert(out == in); + + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +static void test_restore(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + int in = 0xDEADBEEF, out = 0; + + test_assert(connection_put(&c1, &in, sizeof in) == 0); + test_assert(connection_flush(&c1) == sizeof in); + + test_assert(connection_read(&c2) == sizeof out); + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out) == sizeof out); + + test_assert(out == in); + + test_assert(connection_pending(&c2) == 0); + + connection_restore(&c2, sizeof out); + test_assert(connection_pending(&c2) == sizeof out); + test_assert(connection_get(&c2, &out, sizeof out) == sizeof out); + + test_assert(out == in); + + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +static void test_send_variable_sequence(void) { + int fds[2]; + socketpair(AF_UNIX, SOCK_STREAM, 0, fds); + struct connection c1 = {.fd = fds[0]}; + struct connection c2 = {.fd = fds[1]}; + + char in1 = 85, out1 = 0; + int in2 = 0xDEADBEEF, out2 = 0; + uint64_t in3 = 0xCAFEDEADBEEF, out3 = 0; + char in4 = 85, out4 = 0; + + test_assert(connection_put(&c1, &in1, sizeof in1) == 0); + test_assert(connection_put(&c1, &in2, sizeof in2) == 0); + test_assert(connection_put(&c1, &in3, sizeof in3) == 0); + test_assert(connection_put(&c1, &in4, sizeof in4) == 0); + test_assert(connection_flush(&c1) > 0); + + test_assert(connection_read(&c2) > 0); + test_assert(connection_pending(&c2) > 0); + test_assert(connection_get(&c2, &out1, sizeof out1) == sizeof out1); + test_assert(connection_get(&c2, &out2, sizeof out2) == sizeof out2); + test_assert(connection_get(&c2, &out3, sizeof out3) == sizeof out3); + test_assert(connection_get(&c2, &out4, sizeof out4) == sizeof out4); + + test_assert(out1 == in1); + test_assert(out2 == in2); + test_assert(out3 == in3); + test_assert(out4 == in4); + + test_assert(connection_pending(&c2) == 0); + close(fds[0]); + close(fds[1]); +} + +int main(int argc, char *argv[]) { + (void)argc; + (void)argv; + + test_run(test_send_one_byte); + test_run(test_short_read); + test_run(test_long_write); + test_run(test_send_one_int); + test_run(test_restore); + test_run(test_send_variable_sequence); +} |