1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
|
/*
* Copyright (c) 2002-2012 Balabit
* Copyright (c) 1998-2012 Balázs Scheidler
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 as published
* by the Free Software Foundation, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* As an additional exemption you are allowed to compile & link against the
* OpenSSL libraries as published by the OpenSSL project. See the file
* COPYING for details.
*
*/
#ifndef AFSOCKET_SOURCE_H_INCLUDED
#define AFSOCKET_SOURCE_H_INCLUDED
#include "afsocket.h"
#include "socket-options.h"
#include "transport-mapper.h"
#include "driver.h"
#include "logreader.h"
#include "dynamic-window-pool.h"
#include "atomic-gssize.h"
#include "stats/stats-counter.h"
#include <iv.h>
typedef struct _AFSocketSourceDriver AFSocketSourceDriver;
struct _AFSocketSourceDriver
{
LogSrcDriver super;
guint32 connections_kept_alive_across_reloads:1,
window_size_initialized:1,
activate_listener:1;
struct iv_fd listen_fd;
struct iv_timer dynamic_window_timer;
gsize dynamic_window_size;
gsize dynamic_window_timer_tick;
glong dynamic_window_stats_freq;
gint dynamic_window_realloc_ticks;
gint fd;
LogReaderOptions reader_options;
DynamicWindowPool *dynamic_window_pool;
LogProtoServerFactory *proto_factory;
struct
{
struct iv_timer packet_stats_timer;
StatsCounterItem *socket_dropped_packets;
StatsCounterItem *socket_receive_buffer_max;
StatsCounterItem *socket_receive_buffer_used;
StatsCounterItem *rejected_connections;
} metrics;
GSockAddr *bind_addr;
atomic_gssize max_connections;
atomic_gssize num_connections;
gint listen_backlog;
GList *connections;
SocketOptions *socket_options;
TransportMapper *transport_mapper;
/*
* Apply transport options, set up bind_addr based on the
* information processed during parse time. This used to be
* constructed during the parser, however that made the ordering of
* various options matter and behave incorrectly when the port() was
* specified _after_ transport(). Now, it collects the information,
* and then applies them with a separate call to apply_transport()
* during init().
*/
gboolean (*setup_addresses)(AFSocketSourceDriver *s);
/* optionally acquire a socket from the runtime environment (e.g. systemd) */
gboolean (*acquire_socket)(AFSocketSourceDriver *s, gint *fd);
};
void afsocket_sd_set_keep_alive(LogDriver *self, gint enable);
void afsocket_sd_set_max_connections(LogDriver *self, gint max_connections);
void afsocket_sd_set_listen_backlog(LogDriver *self, gint listen_backlog);
void afsocket_sd_set_dynamic_window_size(LogDriver *self, gint dynamic_window_size);
void afsocket_sd_set_dynamic_window_stats_freq(LogDriver *self, gdouble stats_freq);
void afsocket_sd_set_dynamic_window_realloc_ticks(LogDriver *self, gint realloc_ticks);
static inline gboolean
afsocket_sd_acquire_socket(AFSocketSourceDriver *s, gint *fd)
{
if (s->acquire_socket)
return s->acquire_socket(s, fd);
*fd = -1;
return TRUE;
}
static inline gboolean
afsocket_sd_setup_addresses(AFSocketSourceDriver *s)
{
return s->setup_addresses(s);
}
gboolean afsocket_sd_setup_addresses_method(AFSocketSourceDriver *self);
gboolean afsocket_sd_init_method(LogPipe *s);
gboolean afsocket_sd_deinit_method(LogPipe *s);
void afsocket_sd_free_method(LogPipe *self);
void afsocket_sd_init_instance(AFSocketSourceDriver *self, SocketOptions *socket_options,
TransportMapper *transport_mapper, GlobalConfig *cfg);
#endif
|