From b485aab7e71c1625cfc27e0f92c9509f42378458 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 5 May 2024 13:19:16 +0200 Subject: Adding upstream version 1.45.3+dfsg. Signed-off-by: Daniel Baumann --- src/fluent-bit/lib/monkey/fuzz/CMakeLists.txt | 11 ++ src/fluent-bit/lib/monkey/fuzz/mk_check.c | 113 ++++++++++++++++++++ src/fluent-bit/lib/monkey/fuzz/mk_fuzz_me.c | 143 ++++++++++++++++++++++++++ 3 files changed, 267 insertions(+) create mode 100644 src/fluent-bit/lib/monkey/fuzz/CMakeLists.txt create mode 100644 src/fluent-bit/lib/monkey/fuzz/mk_check.c create mode 100644 src/fluent-bit/lib/monkey/fuzz/mk_fuzz_me.c (limited to 'src/fluent-bit/lib/monkey/fuzz') diff --git a/src/fluent-bit/lib/monkey/fuzz/CMakeLists.txt b/src/fluent-bit/lib/monkey/fuzz/CMakeLists.txt new file mode 100644 index 000000000..48100f5f3 --- /dev/null +++ b/src/fluent-bit/lib/monkey/fuzz/CMakeLists.txt @@ -0,0 +1,11 @@ +set(src + mk_fuzz_me.c) + +add_executable(mk_fuzz_me ${src}) +target_link_libraries(mk_fuzz_me monkey-core-static) + +set(src + mk_check.c) + +add_executable(mk_check ${src}) +target_link_libraries(mk_check monkey-core-static) diff --git a/src/fluent-bit/lib/monkey/fuzz/mk_check.c b/src/fluent-bit/lib/monkey/fuzz/mk_check.c new file mode 100644 index 000000000..acae6cba0 --- /dev/null +++ b/src/fluent-bit/lib/monkey/fuzz/mk_check.c @@ -0,0 +1,113 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +#include + +#include +#include +#include +#include + +#define API_ADDR "127.0.0.1" +#define API_PORT "8080" + +/* Main context set as global so the signal handler can use it */ +mk_ctx_t *ctx; + +void cb_main(mk_request_t *request, void *data) +{ + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + + mk_http_send(request, ":)\n", 3, NULL); + mk_http_done(request); +} + +void cb_test_chunks(mk_request_t *request, void *data) +{ + int i = 0; + int len; + char tmp[32]; + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + + for (i = 0; i < 1000; i++) { + len = snprintf(tmp, sizeof(tmp) -1, "test-chunk %6i\n ", i); + mk_http_send(request, tmp, len, NULL); + } + mk_http_done(request); +} + +void cb_test_big_chunk(mk_request_t *request, void *data) +{ + size_t chunk_size = 1024000000; + char *chunk; + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + + chunk = calloc(1, chunk_size); + mk_http_send(request, chunk, chunk_size, NULL); + free(chunk); + mk_http_done(request); +} + + +static void signal_handler(int signal) +{ + write(STDERR_FILENO, "[engine] caught signal\n", 23); + + switch (signal) { + case SIGTERM: + case SIGINT: + mk_stop(ctx); + mk_destroy(ctx); + _exit(EXIT_SUCCESS); + default: + break; + } +} + +static void signal_init() +{ + signal(SIGINT, &signal_handler); + signal(SIGTERM, &signal_handler); +} + +int main() +{ + int vid; + + signal_init(); + + ctx = mk_create(); + if (!ctx) { + return -1; + } + + mk_config_set(ctx, + "Listen", API_PORT, + NULL); + + vid = mk_vhost_create(ctx, NULL); + mk_vhost_set(ctx, vid, + "Name", "monotop", + NULL); + mk_vhost_handler(ctx, vid, "/test_chunks", cb_test_chunks, NULL); + mk_vhost_handler(ctx, vid, "/test_big_chunk", cb_test_big_chunk, NULL); + mk_vhost_handler(ctx, vid, "/", cb_main, NULL); + + mk_info("Service: http://%s:%s/test_chunks", API_ADDR, API_PORT); + mk_start(ctx); + + sleep(3600); + + mk_stop(ctx); + mk_destroy(ctx); + + return 0; +} diff --git a/src/fluent-bit/lib/monkey/fuzz/mk_fuzz_me.c b/src/fluent-bit/lib/monkey/fuzz/mk_fuzz_me.c new file mode 100644 index 000000000..0a88d3f1f --- /dev/null +++ b/src/fluent-bit/lib/monkey/fuzz/mk_fuzz_me.c @@ -0,0 +1,143 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +#include + +#include +#include +#include +#include + +#define API_ADDR "127.0.0.1" +#define API_PORT "8080" + +/* Main context set as global so the signal handler can use it */ +mk_ctx_t *ctx; + +void cb_main(mk_request_t *request, void *data) +{ + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + mk_http_send(request, ":)\n", 3, NULL); + mk_http_done(request); +} + +void cb_test_chunks(mk_request_t *request, void *data) +{ + int i = 0; + int len; + char tmp[32]; + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + + for (i = 0; i < 1000; i++) { + len = snprintf(tmp, sizeof(tmp) -1, "test-chunk %6i\n ", i); + mk_http_send(request, tmp, len, NULL); + } + mk_http_done(request); +} + +void cb_test_big_chunk(mk_request_t *request, void *data) +{ + size_t chunk_size = 1024000000; + char *chunk; + (void) data; + + mk_http_status(request, 200); + mk_http_header(request, "X-Monkey", 8, "OK", 2); + + chunk = calloc(1, chunk_size); + mk_http_send(request, chunk, chunk_size, NULL); + free(chunk); + mk_http_done(request); +} + + +static void signal_handler(int signal) +{ + write(STDERR_FILENO, "[engine] caught signal\n", 23); + + switch (signal) { + case SIGTERM: + case SIGINT: + mk_stop(ctx); + mk_destroy(ctx); + _exit(EXIT_SUCCESS); + default: + break; + } +} + +static void signal_init() +{ + signal(SIGINT, &signal_handler); + signal(SIGTERM, &signal_handler); +} + +static void cb_queue_message(mk_mq_t *queue, void *data, size_t size, void *ctx) +{ + size_t i; + char *buf; + (void) ctx; + (void) queue; + + printf("=== cb queue message === \n"); + printf(" => %lu bytes\n", size); + printf(" => "); + + buf = data; + for (i = 0; i < size; i++) { + printf("%c", buf[i]); + } + printf("\n\n"); +} + + +HFND_FUZZING_ENTRY_FUNCTION(int argc, const char *const *argv) +{ + int i = 0; + int len; + int vid; + int qid; + char msg[800000]; + + signal_init(); + + ctx = mk_create(); + if (!ctx) { + return -1; + } + + /* Create a message queue and a callback for each message */ + qid = mk_mq_create(ctx, "/data", cb_queue_message, NULL); + + mk_config_set(ctx, + "Listen", API_PORT, + NULL); + + vid = mk_vhost_create(ctx, NULL); + mk_vhost_set(ctx, vid, + "Name", "monotop", + NULL); + mk_vhost_handler(ctx, vid, "/test_chunks", cb_test_chunks, NULL); + mk_vhost_handler(ctx, vid, "/test_big_chunk", cb_test_big_chunk, NULL); + mk_vhost_handler(ctx, vid, "/", cb_main, NULL); + + mk_info("Service: http://%s:%s/test_chunks", API_ADDR, API_PORT); + mk_start(ctx); + + for (i = 0; i < 5; i++) { + len = snprintf(msg, sizeof(msg) - 1, "[...] message ID: %i\n", i); + mk_mq_send(ctx, qid, &msg, len); + } + + sleep(3600); + + mk_stop(ctx); + mk_destroy(ctx); + + return 0; +} -- cgit v1.2.3