static gboolean
afprogram_dd_reopen(AFProgramDestDriver *self)
{
int fd;
if (self->pid != -1)
{
msg_verbose("Sending destination program a TERM signal",
evt_tag_str("cmdline", self->cmdline->str),
evt_tag_int("child_pid", self->pid),
NULL);
kill(self->pid, SIGTERM);
self->pid = -1;
}
msg_verbose("Starting destination program",
evt_tag_str("cmdline", self->cmdline->str),
NULL);
if (!afprogram_popen(self->cmdline->str, G_IO_OUT, &self->pid, &fd))
return FALSE;
child_manager_register(self->pid, afprogram_dd_exit, log_pipe_ref(&self->super.super.super), (GDestroyNotify) log_pipe_unref);
g_fd_set_nonblock(fd, TRUE);
log_writer_reopen(self->writer, log_proto_text_client_new(log_transport_plain_new(fd, 0)));
return TRUE;
}
static void
afsocket_dd_notify(LogPipe *s, LogPipe *sender, gint notify_code, gpointer user_data)
{
AFSocketDestDriver *self = (AFSocketDestDriver *) s;
gchar buf[MAX_SOCKADDR_STRING];
switch (notify_code)
{
case NC_CLOSE:
case NC_WRITE_ERROR:
log_writer_reopen(self->writer, NULL);
msg_notice("Syslog connection broken",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf, sizeof(buf), GSA_FULL)),
evt_tag_int("time_reopen", self->time_reopen),
NULL);
if (self->reconnect_timer)
{
g_source_remove(self->reconnect_timer);
self->reconnect_timer = 0;
}
self->reconnect_timer = g_timeout_add(self->time_reopen * 1000, afsocket_dd_reconnect_timer, self);
break;
}
}
static gboolean
afsocket_dd_connected(AFSocketDestDriver *self)
{
gchar buf1[256], buf2[256];
int error = 0;
socklen_t errorlen = sizeof(error);
LogTransport *transport;
LogProtoClient *proto;
main_loop_assert_main_thread();
if (iv_fd_registered(&self->connect_fd))
iv_fd_unregister(&self->connect_fd);
if (self->transport_mapper->sock_type == SOCK_STREAM)
{
if (getsockopt(self->fd, SOL_SOCKET, SO_ERROR, &error, &errorlen) == -1)
{
msg_error("getsockopt(SOL_SOCKET, SO_ERROR) failed for connecting socket",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_errno(EVT_TAG_OSERROR, errno),
evt_tag_int("time_reopen", self->time_reopen),
NULL);
goto error_reconnect;
}
if (error)
{
msg_error("Syslog connection failed",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_errno(EVT_TAG_OSERROR, error),
evt_tag_int("time_reopen", self->time_reopen),
NULL);
goto error_reconnect;
}
}
msg_notice("Syslog connection established",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_str("local", g_sockaddr_format(self->bind_addr, buf1, sizeof(buf1), GSA_FULL)),
NULL);
transport = afsocket_dd_construct_transport(self, self->fd);
if (!transport)
goto error_reconnect;
proto = log_proto_client_factory_construct(self->proto_factory, transport, &self->writer_options.proto_options.super);
log_writer_reopen(self->writer, proto);
return TRUE;
error_reconnect:
close(self->fd);
self->fd = -1;
afsocket_dd_start_reconnect_timer(self);
return FALSE;
}
static gboolean
afprogram_dd_reopen(AFProgramDestDriver *self)
{
int fd;
afprogram_dd_kill_child(self);
if (!afprogram_dd_open_program(self, &fd))
return FALSE;
log_writer_reopen(self->writer, log_proto_text_client_new(log_transport_pipe_new(fd), &self->writer_options.proto_options.super));
return TRUE;
}
static gboolean
afprogram_dd_reopen(AFProgramDestDriver *self)
{
int fd;
afprogram_dd_kill_child(self);
msg_verbose("Starting destination program",
evt_tag_str("cmdline", self->cmdline->str),
NULL);
if (!afprogram_popen(self->cmdline->str, G_IO_OUT, &self->pid, &fd))
return FALSE;
child_manager_register(self->pid, afprogram_dd_exit, log_pipe_ref(&self->super.super.super), (GDestroyNotify) log_pipe_unref);
g_fd_set_nonblock(fd, TRUE);
log_writer_reopen(self->writer, log_proto_text_client_new(log_transport_pipe_new(fd), &self->writer_options.proto_options.super));
return TRUE;
}
static gboolean
afprogram_dd_init(LogPipe *s, GlobalConfig *cfg, PersistentConfig *persist)
{
AFProgramDestDriver *self = (AFProgramDestDriver *) s;
int msg_pipe[2];
if (cfg)
log_writer_options_init(&self->writer_options, cfg, 0, afprogram_dd_format_stats_name(self));
msg_verbose("Starting destination program",
evt_tag_str("cmdline", self->cmdline->str),
NULL);
if (pipe(msg_pipe) == -1)
{
msg_error("Error creating program pipe",
evt_tag_str("cmdline", self->cmdline->str),
evt_tag_errno(EVT_TAG_OSERROR, errno),
NULL);
return FALSE;
}
if ((self->pid = fork()) < 0)
{
msg_error("Error in fork()",
evt_tag_errno(EVT_TAG_OSERROR, errno),
NULL);
close(msg_pipe[0]);
close(msg_pipe[1]);
return FALSE;
}
if (self->pid == 0)
{
/* child */
int devnull = open("/dev/null", O_WRONLY);
if (devnull == -1)
{
_exit(127);
}
dup2(msg_pipe[0], 0);
dup2(devnull, 1);
dup2(devnull, 2);
close(devnull);
close(msg_pipe[0]);
close(msg_pipe[1]);
execl("/bin/sh", "/bin/sh", "-c", self->cmdline->str, NULL);
_exit(127);
}
else
{
/* parent */
child_manager_register(self->pid, afprogram_dd_exit, log_pipe_ref(&self->super.super), (GDestroyNotify) log_pipe_unref);
close(msg_pipe[0]);
g_fd_set_nonblock(msg_pipe[1], TRUE);
if (!self->writer)
self->writer = log_writer_new(LW_FORMAT_FILE, s, &self->writer_options);
#if 0/*start dongshu*/
log_writer_reopen(self->writer, fd_write_new(msg_pipe[1]));
#else
log_writer_reopen(self->writer, fd_write_new(msg_pipe[1],NULL));
#endif/*end*/
log_pipe_init(self->writer, NULL, NULL);
log_pipe_append(&self->super.super, self->writer);
}
return TRUE;
}
static gboolean
afsocket_dd_connected(AFSocketDestDriver *self)
{
gchar buf1[256], buf2[256];
int error = 0;
socklen_t errorlen = sizeof(error);
LogTransport *transport;
LogProto *proto;
guint32 transport_flags = 0;
main_loop_assert_main_thread();
if (iv_fd_registered(&self->connect_fd))
iv_fd_unregister(&self->connect_fd);
if (self->flags & AFSOCKET_STREAM)
{
transport_flags |= LTF_SHUTDOWN;
if (getsockopt(self->fd, SOL_SOCKET, SO_ERROR, &error, &errorlen) == -1)
{
msg_error("getsockopt(SOL_SOCKET, SO_ERROR) failed for connecting socket",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_errno(EVT_TAG_OSERROR, errno),
evt_tag_int("time_reopen", self->time_reopen),
NULL);
goto error_reconnect;
}
if (error)
{
msg_error("Syslog connection failed",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_errno(EVT_TAG_OSERROR, error),
evt_tag_int("time_reopen", self->time_reopen),
NULL);
goto error_reconnect;
}
}
msg_notice("Syslog connection established",
evt_tag_int("fd", self->fd),
evt_tag_str("server", g_sockaddr_format(self->dest_addr, buf2, sizeof(buf2), GSA_FULL)),
evt_tag_str("local", g_sockaddr_format(self->bind_addr, buf1, sizeof(buf1), GSA_FULL)),
NULL);
#if ENABLE_SSL
if (self->tls_context)
{
TLSSession *tls_session;
tls_session = tls_context_setup_session(self->tls_context);
if (!tls_session)
{
goto error_reconnect;
}
tls_session_set_verify(tls_session, afsocket_dd_tls_verify_callback, self, NULL);
transport = log_transport_tls_new(tls_session, self->fd, transport_flags);
}
else
#endif
transport = log_transport_plain_new(self->fd, transport_flags);
if (self->flags & AFSOCKET_SYSLOG_PROTOCOL)
{
if (self->flags & AFSOCKET_STREAM)
proto = log_proto_framed_client_new(transport);
else
proto = log_proto_text_client_new(transport);
}
else
{
proto = log_proto_text_client_new(transport);
}
log_writer_reopen(self->writer, proto);
return TRUE;
error_reconnect:
close(self->fd);
self->fd = -1;
afsocket_dd_start_reconnect_timer(self);
return FALSE;
}
