Skip to content

in_tcp: add source address even when format is none #10264

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 17 additions & 1 deletion plugins/in_tcp/tcp_conn.c
Original file line number Diff line number Diff line change
Expand Up @@ -189,6 +189,7 @@ static ssize_t parse_payload_none(struct tcp_conn *conn)
char *buf;
char *s;
char *separator;
char *source_address;
struct flb_in_tcp_config *ctx;

ctx = conn->ctx;
Expand All @@ -214,10 +215,25 @@ static ssize_t parse_payload_none(struct tcp_conn *conn)
}

if (ret == FLB_EVENT_ENCODER_SUCCESS) {
ret = flb_log_event_encoder_append_body_values(
source_address = NULL;
if (ctx->source_address_key != NULL) {
source_address = flb_connection_get_remote_address(conn->connection);
}

if (ctx->source_address_key != NULL && source_address != NULL) {
ret = flb_log_event_encoder_append_body_values(
ctx->log_encoder,
FLB_LOG_EVENT_CSTRING_VALUE("log"),
FLB_LOG_EVENT_STRING_VALUE(buf, len),
FLB_LOG_EVENT_CSTRING_VALUE(ctx->source_address_key),
FLB_LOG_EVENT_CSTRING_VALUE(source_address));
}
else {
ret = flb_log_event_encoder_append_body_values(
ctx->log_encoder,
FLB_LOG_EVENT_CSTRING_VALUE("log"),
FLB_LOG_EVENT_STRING_VALUE(buf, len));
}
}

if (ret == FLB_EVENT_ENCODER_SUCCESS) {
Expand Down
56 changes: 56 additions & 0 deletions tests/runtime/in_tcp.c
Original file line number Diff line number Diff line change
Expand Up @@ -547,6 +547,61 @@ void flb_test_format_none_separator()
test_ctx_destroy(ctx);
}

void flb_test_format_none_with_source_address()
{
struct flb_lib_out_cb cb_data;
struct test_ctx *ctx;
flb_sockfd_t fd;
int ret;
int num;
ssize_t w_size;
char *buf = "message\n";
size_t size = strlen(buf);
clear_output_num();
cb_data.cb = cb_check_result_json;
cb_data.data = "\"log\":\"message\",\"source_host\":\"tcp://";
ctx = test_ctx_create(&cb_data);
if (!TEST_CHECK(ctx != NULL)) {
TEST_MSG("test_ctx_create failed");
exit(EXIT_FAILURE);
}
ret = flb_input_set(ctx->flb, ctx->i_ffd,
"source_address_key", "source_host",
NULL);
TEST_CHECK(ret == 0);
ret = flb_output_set(ctx->flb, ctx->o_ffd,
"match", "*",
"format", "json",
NULL);
TEST_CHECK(ret == 0);
ret = flb_input_set(ctx->flb, ctx->i_ffd,
"format", "none",
NULL);
TEST_CHECK(ret == 0);
/* Start the engine */
ret = flb_start(ctx->flb);
TEST_CHECK(ret == 0);
/* use default host/port */
fd = connect_tcp(NULL, -1);
if (!TEST_CHECK(fd >= 0)) {
exit(EXIT_FAILURE);
}
w_size = send(fd, buf, size, 0);
if (!TEST_CHECK(w_size == size)) {
TEST_MSG("failed to send, errno=%d", errno);
flb_socket_close(fd);
exit(EXIT_FAILURE);
}
/* waiting to flush */
flb_time_msleep(1500);
num = get_output_num();
if (!TEST_CHECK(num > 0)) {
TEST_MSG("no outputs");
}
flb_socket_close(fd);
test_ctx_destroy(ctx);
}

/*
* Ingest 64k records.
* https://github.com/fluent/fluent-bit/issues/5336
Expand Down Expand Up @@ -619,6 +674,7 @@ TEST_LIST = {
{"tcp_with_tls", flb_test_tcp_with_tls},
{"format_none", flb_test_format_none},
{"format_none_separator", flb_test_format_none_separator},
{"format_none_with_source_address", flb_test_format_none_with_source_address},
{"65535_records_issue_5336", flb_test_issue_5336},
{NULL, NULL}
};
Expand Down