static void
test_log_proto_dgram_server_invalid_ucs4(void)
{
LogProtoServer *proto;
proto_server_options.max_msg_size = 32;
log_proto_server_options_set_encoding(&proto_server_options, "ucs-4");
proto = log_proto_dgram_server_new(
/* 31 bytes record size */
log_transport_mock_endless_records_new(
/* invalid ucs4, trailing zeroes at the end */
"\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", 31, /* |...z...t...ű...r| */
LTM_EOF),
get_inited_proto_server_options());
assert_proto_server_fetch_failure(proto, LPS_ERROR, "Byte sequence too short, cannot convert an individual frame in its entirety");
log_proto_server_free(proto);
}
static void
test_log_proto_dgram_server_eof_handling(void)
{
LogProtoServer *proto;
proto_server_options.max_msg_size = 32;
proto = log_proto_dgram_server_new(
log_transport_mock_endless_records_new(
/* no eol before EOF */
"01234567", -1,
LTM_EOF),
get_inited_proto_server_options());
assert_proto_server_fetch(proto, "01234567", -1);
assert_proto_server_fetch_ignored_eof(proto);
assert_proto_server_fetch_ignored_eof(proto);
assert_proto_server_fetch_ignored_eof(proto);
log_proto_server_free(proto);
}
static void
test_log_proto_dgram_server_no_encoding(void)
{
LogProtoServer *proto;
proto_server_options.max_msg_size = 32;
proto = log_proto_dgram_server_new(
log_transport_mock_endless_records_new(
"0123456789ABCDEF0123456789ABCDEF", -1,
"01234567\n", -1,
"01234567\0", 9,
/* utf8 */
"árvíztűrőtükörfúrógép\n\n", 32,
/* 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| */
/* 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", 32, /* |...z...t...ű...r| */
"01234", 5,
LTM_EOF),
get_inited_proto_server_options());
assert_proto_server_fetch(proto, "0123456789ABCDEF0123456789ABCDEF", -1);
assert_proto_server_fetch(proto, "01234567\n", -1);
assert_proto_server_fetch(proto, "01234567\0", 9);
/* no encoding: utf8 remains utf8 */
assert_proto_server_fetch(proto, "árvíztűrőtükörfúrógép\n\n", -1);
/* no encoding: iso-8859-2 remains iso-8859-2 */
assert_proto_server_fetch(proto, "\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", /* |r.g.p| */
-1);
/* no encoding, ucs4 becomes a string with embedded NULs */
assert_proto_server_fetch(proto, "\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", 32); /* |...z...t...ű...r| */
assert_proto_server_fetch(proto, "01234", -1);
log_proto_server_free(proto);
}
static void
test_log_proto_dgram_server_iso_8859_2(void)
{
LogProtoServer *proto;
proto_server_options.max_msg_size = 32;
log_proto_server_options_set_encoding(&proto_server_options, "iso-8859-2");
proto = log_proto_dgram_server_new(
log_transport_mock_endless_records_new(
/* iso-8859-2, deliberately contains
* accented chars so utf8 representation
* becomes longer than the record size */
"\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\xe9\xe9\xe9\xe9\xe9\xe9\xe9\xe9\xe9\xe9\xe9", -1, /* |rógépééééééééééé| */
LTM_EOF),
get_inited_proto_server_options());
assert_proto_server_fetch(proto, "árvíztűrőtükörfúrógépééééééééééé", -1);
assert_proto_server_fetch_ignored_eof(proto);
log_proto_server_free(proto);
}
static void
test_log_proto_dgram_server_ucs4(void)
{
LogProtoServer *proto;
proto_server_options.max_msg_size = 32;
log_proto_server_options_set_encoding(&proto_server_options, "ucs-4");
proto = log_proto_dgram_server_new(
log_transport_mock_endless_records_new(
/* ucs4, terminated by record size */
"\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", 32, /* |...z...t...ű...r| */
/* ucs4, terminated by ucs4 encododed NL at the end */
"\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\n", 32, /* |...z...t...ű| */
LTM_EOF),
get_inited_proto_server_options());
assert_proto_server_fetch(proto, "árvíztűr", -1);
assert_proto_server_fetch(proto, "árvíztű\n", -1);
log_proto_server_free(proto);
}
static LogProtoServer *
affile_sd_construct_proto(AFFileSourceDriver *self, gint fd)
{
LogProtoServerOptions *proto_options = &self->reader_options.proto_options.super;
LogTransport *transport;
MsgFormatHandler *format_handler;
transport = affile_sd_construct_transport(self, fd);
format_handler = self->reader_options.parse_options.format_handler;
if ((format_handler && format_handler->construct_proto))
{
return format_handler->construct_proto(&self->reader_options.parse_options, transport, proto_options);
}
if (self->pad_size)
return log_proto_padded_record_server_new(transport, proto_options, self->pad_size);
else if (affile_is_linux_proc_kmsg(self->filename->str))
return log_proto_linux_proc_kmsg_reader_new(transport, proto_options);
else if (affile_is_linux_dev_kmsg(self->filename->str))
return log_proto_dgram_server_new(transport, proto_options);
else
{
switch (self->multi_line_mode)
{
case MLM_INDENTED:
return log_proto_indented_multiline_server_new(transport, proto_options);
case MLM_PREFIX_GARBAGE:
return log_proto_prefix_garbage_multiline_server_new(transport, proto_options, self->multi_line_prefix, self->multi_line_garbage);
case MLM_PREFIX_SUFFIX:
return log_proto_prefix_suffix_multiline_server_new(transport, proto_options, self->multi_line_prefix, self->multi_line_garbage);
default:
return log_proto_text_server_new(transport, proto_options);
}
}
}
static gboolean
afstreams_sd_init(LogPipe *s)
{
AFStreamsSourceDriver *self = (AFStreamsSourceDriver *) s;
GlobalConfig *cfg = log_pipe_get_config(s);
gint fd;
if (!log_src_driver_init_method(s))
return FALSE;
log_reader_options_init(&self->reader_options, cfg, self->super.super.group);
fd = open(self->dev_filename->str, O_RDONLY | O_NOCTTY | O_NONBLOCK);
if (fd != -1)
{
struct strioctl ioc;
g_fd_set_cloexec(fd, TRUE);
memset(&ioc, 0, sizeof(ioc));
ioc.ic_cmd = I_CONSLOG;
if (ioctl(fd, I_STR, &ioc) < 0)
{
msg_error("Error in ioctl(I_STR, I_CONSLOG)",
evt_tag_str(EVT_TAG_FILENAME, self->dev_filename->str),
evt_tag_errno(EVT_TAG_OSERROR, errno));
close(fd);
return FALSE;
}
g_fd_set_nonblock(fd, TRUE);
self->reader = log_reader_new(cfg);
log_reader_reopen(self->reader, log_proto_dgram_server_new(log_transport_streams_new(fd), &self->reader_options.proto_options.super), poll_fd_events_new(fd));
log_reader_set_options(self->reader,
s,
&self->reader_options,
STATS_LEVEL1,
SCS_SUN_STREAMS,
self->super.super.id,
self->dev_filename->str);
log_pipe_append((LogPipe *) self->reader, s);
if (self->door_filename)
{
/* door creation is deferred, because it creates threads which is
* not inherited through forks, and syslog-ng forks during
* startup, but _after_ the configuration was initialized */
register_application_hook(AH_POST_DAEMONIZED, afstreams_init_door, self);
}
if (!log_pipe_init((LogPipe *) self->reader))
{
msg_error("Error initializing log_reader, closing fd",
evt_tag_int("fd", fd));
log_pipe_unref((LogPipe *) self->reader);
self->reader = NULL;
close(fd);
return FALSE;
}
}
else
{
msg_error("Error opening syslog device",
evt_tag_str(EVT_TAG_FILENAME, self->dev_filename->str),
evt_tag_errno(EVT_TAG_OSERROR, errno));
return FALSE;
}
return TRUE;
}
static gboolean
afsocket_sc_init(LogPipe *s)
{
AFSocketSourceConnection *self = (AFSocketSourceConnection *) s;
gint read_flags;
LogTransport *transport;
LogProto *proto;
read_flags = ((self->owner->flags & AFSOCKET_DGRAM) ? LTF_RECV : 0);
if (!self->reader)
{
#if ENABLE_SSL
if (self->owner->tls_context)
{
TLSSession *tls_session = tls_context_setup_session(self->owner->tls_context);
if (!tls_session)
return FALSE;
transport = log_transport_tls_new(tls_session, self->sock, read_flags);
}
else
#endif
transport = log_transport_plain_new(self->sock, read_flags);
if ((self->owner->flags & AFSOCKET_SYSLOG_PROTOCOL) == 0)
{
/* plain protocol */
if (self->owner->flags & AFSOCKET_DGRAM)
proto = log_proto_dgram_server_new(transport, self->owner->reader_options.msg_size, 0);
else if (self->owner->reader_options.padding)
proto = log_proto_record_server_new(transport, self->owner->reader_options.padding, 0);
else
proto = log_proto_text_server_new(transport, self->owner->reader_options.msg_size, 0);
}
else
{
if (self->owner->flags & AFSOCKET_DGRAM)
{
/* plain protocol */
proto = log_proto_dgram_server_new(transport, self->owner->reader_options.msg_size, 0);
}
else
{
/* framed protocol */
proto = log_proto_framed_server_new(transport, self->owner->reader_options.msg_size);
}
}
self->reader = log_reader_new(proto);
}
log_reader_set_options(self->reader, s, &self->owner->reader_options, 1, afsocket_sc_stats_source(self), self->owner->super.super.id, afsocket_sc_stats_instance(self));
log_reader_set_peer_addr(self->reader, self->peer_addr);
log_pipe_append(self->reader, s);
if (log_pipe_init(self->reader, NULL))
{
return TRUE;
}
else
{
log_pipe_unref(self->reader);
self->reader = NULL;
}
return FALSE;
}
