gboolean
afsocket_dd_init(LogPipe *s)
{
AFSocketDestDriver *self = (AFSocketDestDriver *) s;
GlobalConfig *cfg = log_pipe_get_config(s);
if (!log_dest_driver_init_method(s))
return FALSE;
if (!afsocket_dd_apply_transport(self))
return FALSE;
/* these fields must be set up by apply_transport, so let's check if it indeed did */
g_assert(self->transport);
g_assert(self->bind_addr);
g_assert(self->hostname);
g_assert(self->dest_name);
if (cfg)
{
self->time_reopen = cfg->time_reopen;
}
log_writer_options_init(&self->writer_options, cfg, 0);
self->writer = cfg_persist_config_fetch(cfg, afsocket_dd_format_persist_name(self, FALSE));
if (!self->writer)
{
/* NOTE: we open our writer with no fd, so we can send messages down there
* even while the connection is not established */
self->writer = log_writer_new(LW_FORMAT_PROTO |
#if ENABLE_SSL
(((self->flags & AFSOCKET_STREAM) && !self->tls_context) ? LW_DETECT_EOF : 0) |
#else
((self->flags & AFSOCKET_STREAM) ? LW_DETECT_EOF : 0) |
#endif
(self->flags & AFSOCKET_SYSLOG_PROTOCOL ? LW_SYSLOG_PROTOCOL : 0));
}
log_writer_set_options((LogWriter *) self->writer, &self->super.super.super, &self->writer_options, 0, afsocket_dd_stats_source(self), self->super.super.id, afsocket_dd_stats_instance(self));
log_writer_set_queue(self->writer, log_dest_driver_acquire_queue(&self->super, afsocket_dd_format_persist_name(self, TRUE)));
log_pipe_init(self->writer, NULL);
log_pipe_append(&self->super.super.super, self->writer);
if (!log_writer_opened((LogWriter *) self->writer))
afsocket_dd_reconnect(self);
return TRUE;
}
static gboolean
afsocket_dd_setup_writer(AFSocketDestDriver *self)
{
afsocket_dd_restore_writer(self);
if (!self->writer)
{
/* NOTE: we open our writer with no fd, so we can send messages down there
* even while the connection is not established */
self->writer = afsocket_dd_construct_writer(self);
}
log_writer_set_options(self->writer, &self->super.super.super,
&self->writer_options,
STATS_LEVEL0,
self->transport_mapper->stats_source,
self->super.super.id,
afsocket_dd_stats_instance(self));
log_writer_set_queue(self->writer, log_dest_driver_acquire_queue(&self->super, afsocket_dd_format_persist_name(self, TRUE)));
if (!log_pipe_init((LogPipe *) self->writer))
{
log_pipe_unref((LogPipe *) self->writer);
return FALSE;
}
log_pipe_append(&self->super.super.super, (LogPipe *) self->writer);
return TRUE;
}
static void
afsocket_dd_restore_connection(AFSocketDestDriver *self)
{
GlobalConfig *cfg = log_pipe_get_config(&self->super.super.super);
self->writer = cfg_persist_config_fetch(cfg, afsocket_dd_format_persist_name(self, FALSE));
}
gboolean
afsocket_dd_deinit(LogPipe *s)
{
AFSocketDestDriver *self = (AFSocketDestDriver *) s;
GlobalConfig *cfg = log_pipe_get_config(s);
if (self->reconnect_timer)
g_source_remove(self->reconnect_timer);
if (self->source_id && g_source_remove(self->source_id))
{
msg_verbose("Closing connecting fd",
evt_tag_int("fd", self->fd),
NULL);
close(self->fd);
}
if (self->writer)
{
log_pipe_deinit(self->writer);
}
if (self->flags & AFSOCKET_KEEP_ALIVE)
{
cfg_persist_config_add(cfg, afsocket_dd_format_persist_name(self, self->dest_name, FALSE), self->writer, -1, (GDestroyNotify) log_pipe_unref, FALSE);
self->writer = NULL;
}
return TRUE;
}
static void
afsocket_dd_save_connection(AFSocketDestDriver *self)
{
GlobalConfig *cfg = log_pipe_get_config(&self->super.super.super);
if (self->connections_kept_alive_accross_reloads)
{
cfg_persist_config_add(cfg, afsocket_dd_format_persist_name(self, FALSE), self->writer, (GDestroyNotify) log_pipe_unref, FALSE);
self->writer = NULL;
}
}
static void
afsocket_dd_save_connection(AFSocketDestDriver *self)
{
GlobalConfig *cfg = log_pipe_get_config(&self->super.super.super);
if (self->connections_kept_alive_accross_reloads)
{
ReloadStoreItem *item = _reload_store_item_new(self);
cfg_persist_config_add(cfg, afsocket_dd_format_persist_name(self, FALSE), item, (GDestroyNotify) _reload_store_item_free, FALSE);
self->writer = NULL;
}
}
static void
afsocket_dd_restore_writer(AFSocketDestDriver *self)
{
GlobalConfig *cfg;
ReloadStoreItem *item;
g_assert(self->writer == NULL);
cfg = log_pipe_get_config(&self->super.super.super);
item = cfg_persist_config_fetch(cfg, afsocket_dd_format_persist_name(self, FALSE));
if (item && !_is_protocol_type_changed_during_reload(self, item))
self->writer = _reload_store_item_release_writer(item);
_reload_store_item_free(item);
}
gboolean
afsocket_dd_init(LogPipe *s)
{
AFSocketDestDriver *self = (AFSocketDestDriver *) s;
GlobalConfig *cfg = log_pipe_get_config(s);
#if ENABLE_SSL
if (self->flags & AFSOCKET_REQUIRE_TLS && !self->tls_context)
{
msg_error("Transport TLS was specified, but TLS related parameters missing", NULL);
return FALSE;
}
#endif
self->error_logged = FALSE;
if (cfg)
{
self->time_reopen = cfg->time_reopen;
}
if (!self->writer)
{
log_writer_options_init(&self->writer_options, cfg, 0);
/* NOTE: we open our writer with no fd, so we can send messages down there
* even while the connection is not established */
if ((self->flags & AFSOCKET_KEEP_ALIVE))
self->writer = cfg_persist_config_fetch(cfg, afsocket_dd_format_persist_name(self, self->dest_name, FALSE));
if (!self->writer)
self->writer = log_writer_new(LW_FORMAT_PROTO |
#if ENABLE_SSL
(((self->flags & AFSOCKET_STREAM) && !self->tls_context) ? LW_DETECT_EOF : 0) |
#else
((self->flags & AFSOCKET_STREAM) ? LW_DETECT_EOF : 0) |
#endif
(self->flags & AFSOCKET_SYSLOG_PROTOCOL ? LW_SYSLOG_PROTOCOL : 0));
log_writer_set_options((LogWriter *) self->writer, &self->super.super, &self->writer_options, 0, afsocket_dd_stats_source(self), self->super.id, afsocket_dd_stats_instance(self));
log_pipe_init(self->writer, NULL);
log_pipe_append(&self->super.super, self->writer);
}
afsocket_dd_reconnect(self);
return TRUE;
}
gboolean
afsocket_dd_deinit(LogPipe *s)
{
AFSocketDestDriver *self = (AFSocketDestDriver *) s;
GlobalConfig *cfg = log_pipe_get_config(s);
afsocket_dd_stop_watches(self);
if (self->writer)
log_pipe_deinit(self->writer);
if (self->flags & AFSOCKET_KEEP_ALIVE)
{
cfg_persist_config_add(cfg, afsocket_dd_format_persist_name(self, FALSE), self->writer, (GDestroyNotify) log_pipe_unref, FALSE);
self->writer = NULL;
}
if (!log_dest_driver_deinit_method(s))
return FALSE;
return TRUE;
}
