static gboolean
log_reader_init(LogPipe *s)
{
LogReader *self = (LogReader *) s;
if (!log_source_init(s))
return FALSE;
if (!log_proto_server_validate_options(self->proto))
return FALSE;
if (!self->options->parse_options.format_handler)
{
msg_error("Unknown format plugin specified",
evt_tag_str("format", self->options->parse_options.format));
return FALSE;
}
poll_events_set_callback(self->poll_events, log_reader_io_process_input, self);
log_reader_update_watches(self);
iv_event_register(&self->schedule_wakeup);
iv_event_register(&self->last_msg_sent_event);
return TRUE;
}
static void
test_log_proto_text_server_partial_chars_before_eof(void)
{
LogProtoServer *proto;
log_proto_server_options_set_encoding(&proto_server_options, "utf-8");
proto = construct_test_proto(
log_transport_mock_stream_new(
/* utf8 */
"\xc3", -1,
LTM_EOF));
assert_true(log_proto_server_validate_options(proto), "validate_options() returned failure but it should have succeeded");
assert_proto_server_fetch_failure(proto, LPS_EOF, "EOF read on a channel with leftovers from previous character conversion, dropping input");
log_proto_server_free(proto);
}
static void
test_log_proto_text_server_invalid_encoding(void)
{
LogProtoServer *proto;
gboolean success;
log_proto_server_options_set_encoding(&proto_server_options, "never-ever-is-going-to-be-such-an-encoding");
proto = construct_test_proto(
log_transport_mock_stream_new(
"", -1,
LTM_EOF));
start_grabbing_messages();
success = log_proto_server_validate_options(proto);
assert_grabbed_messages_contain("Unknown character set name specified; encoding='never-ever-is-going-to-be-such-an-encoding'", "message about unknown charset missing");
assert_false(success, "validate_options() returned success but it should have failed");
log_proto_server_free(proto);
}
static void
test_log_proto_text_server_iso8859_2(void)
{
LogProtoServer *proto;
log_proto_server_options_set_encoding(&proto_server_options, "iso-8859-2");
proto = construct_test_proto(
log_transport_mock_stream_new(
/* iso-8859-2 */
"\xe1\x72\x76\xed\x7a\x74\xfb\x72\xf5\x74\xfc\x6b\xf6\x72\x66\xfa" /* |árvíztűrőtükörfú| */
"\x72\xf3\x67\xe9\x70\n", -1, /* |rógép| */
LTM_EOF));
assert_true(log_proto_server_validate_options(proto), "validate_options() returned failure but it should have succeeded");
assert_proto_server_fetch(proto, "árvíztűrőtükörfúrógép", -1);
assert_proto_server_fetch_failure(proto, LPS_EOF, NULL);
log_proto_server_free(proto);
}
static void
test_log_proto_text_server_not_fixed_encoding(void)
{
LogProtoServer *proto;
log_proto_server_options_set_encoding(&proto_server_options, "utf-8");
/* to test whether a non-easily-reversable charset works too */
proto = construct_test_proto(
log_transport_mock_stream_new(
/* utf8 */
"árvíztűrőtükörfúrógép\n", -1,
LTM_EOF));
assert_true(log_proto_server_validate_options(proto), "validate_options() returned failure but it should have succeeded");
assert_proto_server_fetch(proto, "árvíztűrőtükörfúrógép", -1);
assert_proto_server_fetch_failure(proto, LPS_EOF, NULL);
log_proto_server_free(proto);
}
static void
test_log_proto_text_server_ucs4(void)
{
LogProtoServer *proto;
log_proto_server_options_set_encoding(&proto_server_options, "ucs-4");
proto = construct_test_proto(
log_transport_mock_stream_new(
/* ucs4 */
"\x00\x00\x00\xe1\x00\x00\x00\x72\x00\x00\x00\x76\x00\x00\x00\xed" /* |...á...r...v...í| */
"\x00\x00\x00\x7a\x00\x00\x00\x74\x00\x00\x01\x71\x00\x00\x00\x72" /* |...z...t...ű...r| */
"\x00\x00\x01\x51\x00\x00\x00\x74\x00\x00\x00\xfc\x00\x00\x00\x6b" /* |...Q...t.......k| */
"\x00\x00\x00\xf6\x00\x00\x00\x72\x00\x00\x00\x66\x00\x00\x00\xfa" /* |.......r...f....| */
"\x00\x00\x00\x72\x00\x00\x00\xf3\x00\x00\x00\x67\x00\x00\x00\xe9" /* |...r.......g....| */
"\x00\x00\x00\x70\x00\x00\x00\x0a", 88, /* |...p....| */
LTM_EOF));
assert_true(log_proto_server_validate_options(proto), "validate_options() returned failure but it should have succeeded");
assert_proto_server_fetch(proto, "árvíztűrőtükörfúrógép", -1);
assert_proto_server_fetch_failure(proto, LPS_EOF, NULL);
log_proto_server_free(proto);
}
static gboolean
log_reader_init(LogPipe *s)
{
LogReader *self = (LogReader *) s;
if (!log_source_init(s))
return FALSE;
if (!log_proto_server_validate_options(self->proto))
return FALSE;
if (!self->options->parse_options.format_handler)
{
msg_error("Unknown format plugin specified",
evt_tag_str("format", self->options->parse_options.format),
NULL);
return FALSE;
}
if (!log_reader_start_watches(self))
return FALSE;
iv_event_register(&self->schedule_wakeup);
return TRUE;
}
