diff options
author | Richard Braun <rbraun@sceen.net> | 2018-08-31 01:15:43 +0200 |
---|---|---|
committer | Richard Braun <rbraun@sceen.net> | 2018-08-31 01:15:43 +0200 |
commit | b0969d077818a74f2fd26b5a7bec2263a4517b07 (patch) | |
tree | 67ba719aeb48682db21274e19e0d9c3ef7a547b8 /test | |
parent | 863bc0bbb53e27178a041aad563ecc96e5f076ce (diff) |
mbuf: new module
Diffstat (limited to 'test')
-rw-r--r-- | test/test_mbuf.c | 222 |
1 files changed, 222 insertions, 0 deletions
diff --git a/test/test_mbuf.c b/test/test_mbuf.c new file mode 100644 index 0000000..6b3f90f --- /dev/null +++ b/test/test_mbuf.c @@ -0,0 +1,222 @@ +/* + * Copyright (c) 2018 Richard Braun. + * + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. + */ + +#include <errno.h> +#include <stddef.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include <check.h> +#include <macros.h> +#include <mbuf.h> + +static void +test_regular(void) +{ + char mbuf_buf[32], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + mbuf_init(&mbuf, mbuf_buf, sizeof(mbuf_buf)); + +#define STRING "abcdef" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, false); + check(!error); +#undef STRING + +#define STRING "xyz" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, false); + check(!error); +#undef STRING + +#define STRING "abcdef" + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(!error); + check(size == (STRLEN(STRING) + 1)); +#undef STRING + +#define STRING "xyz" + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(!error); + check(size == (STRLEN(STRING) + 1)); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +static void +test_write_full(void) +{ + char mbuf_buf[16], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + mbuf_init(&mbuf, mbuf_buf, sizeof(mbuf_buf)); + +#define STRING "abcdef" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, false); + check(!error); +#undef STRING + +#define STRING "xyz" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, false); + check(error == EMSGSIZE); +#undef STRING + +#define STRING "abcdef" + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(!error); + check(size == (STRLEN(STRING) + 1)); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +static void +test_overwrite(void) +{ + char mbuf_buf[16], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + mbuf_init(&mbuf, mbuf_buf, sizeof(mbuf_buf)); + +#define STRING "abcdef" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, true); + check(!error); +#undef STRING + +#define STRING "xyz" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, true); + check(!error); +#undef STRING + +#define STRING "xyz" + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(!error); + check(size == (STRLEN(STRING) + 1)); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +static void +test_msg_too_big(void) +{ + char mbuf_buf[8], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + mbuf_init(&mbuf, mbuf_buf, sizeof(mbuf_buf)); + +#define STRING "abcdef" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, true); + check(error == EMSGSIZE); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +static void +test_msg_far_too_big(void) +{ + char mbuf_buf[8], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + if (sizeof(size_t) <= sizeof(uint32_t)) { + fprintf(stderr, "warning: %s disabled\n", __func__); + return; + } + + mbuf_init(&mbuf, mbuf_buf, (size_t)1 << 36); + +#define STRING "abcdef" + size = (size_t)1 << 32; + error = mbuf_push(&mbuf, STRING, size, true); + check(error == EMSGSIZE); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +static void +test_peak(void) +{ + char mbuf_buf[16], buffer[8]; + struct mbuf mbuf; + size_t size; + int error; + + mbuf_init(&mbuf, mbuf_buf, sizeof(mbuf_buf)); + +#define STRING "abcdef" + error = mbuf_push(&mbuf, STRING, STRLEN(STRING) + 1, true); + check(!error); +#undef STRING + +#define STRING "abcdef" + size = 0; + error = mbuf_pop(&mbuf, NULL, &size); + check((error == EMSGSIZE) && (size == (STRLEN(STRING) + 1))); + error = mbuf_pop(&mbuf, buffer, &size); + check(!error && (size == (STRLEN(STRING) + 1))); +#undef STRING + + size = sizeof(buffer); + error = mbuf_pop(&mbuf, buffer, &size); + check(error == EAGAIN); +} + +int +main(void) +{ + test_regular(); + test_write_full(); + test_overwrite(); + test_msg_too_big(); + test_msg_far_too_big(); + test_peak(); + + return EXIT_SUCCESS; +} |