mirror of
https://github.com/rsyslog/rsyslog.git
synced 2025-12-20 09:40:42 +01:00
Merge branch 'sock-abstract' into tls
This commit is contained in:
commit
961f2956e8
@ -1,6 +1,8 @@
|
||||
---------------------------------------------------------------------------
|
||||
Version 3.19.0 (rgerhards), 2008-04-??
|
||||
- begins new devel branch version
|
||||
- partly rewritten and improved omfwd among others, now loads TCP
|
||||
code only if this is actually necessary
|
||||
- implemented im3195, the RFC3195 input as a plugin
|
||||
- split of a "runtime library" for rsyslog - this is not yet a clean
|
||||
model, because some modularization is still outstanding. In theory,
|
||||
|
||||
@ -58,17 +58,6 @@
|
||||
|
||||
MODULE_TYPE_OUTPUT
|
||||
|
||||
#define INET_SUSPEND_TIME 60
|
||||
/* equal to 1 minute - TODO: see if we can get rid of this now that we have
|
||||
* the retry intervals in the engine -- rgerhards, 2008-03-12
|
||||
*/
|
||||
|
||||
#define INET_RETRY_MAX 30 /* maximum of retries for gethostbyname() */
|
||||
/* was 10, changed to 30 because we reduced INET_SUSPEND_TIME by one third. So
|
||||
* this "fixes" some of implications of it (see comment on INET_SUSPEND_TIME).
|
||||
* rgerhards, 2005-07-26
|
||||
* TODO: this needs to be reviewed in spite of the new engine, too -- rgerhards, 2008-03-12
|
||||
*/
|
||||
|
||||
/* internal structures
|
||||
*/
|
||||
@ -86,11 +75,9 @@ typedef struct _instanceData {
|
||||
eDestFORW_SUSP,
|
||||
eDestFORW_UNKN
|
||||
} eDestState;
|
||||
int iRtryCnt;
|
||||
struct addrinfo *f_addr;
|
||||
int compressionLevel; /* 0 - no compression, else level for zlib */
|
||||
char *port;
|
||||
time_t ttSuspend; /* time selector was suspended */
|
||||
tcpclt_t *pTCPClt; /* our tcpclt object */
|
||||
gss_ctx_id_t gss_context;
|
||||
OM_uint32 gss_flags;
|
||||
@ -174,8 +161,6 @@ CODESTARTdbgPrintInstInfo
|
||||
ENDdbgPrintInstInfo
|
||||
|
||||
|
||||
/* CODE FOR SENDING TCP MESSAGES */
|
||||
|
||||
/* This function is called immediately before a send retry is attempted.
|
||||
* It shall clean up whatever makes sense.
|
||||
* rgerhards, 2007-12-28
|
||||
@ -207,9 +192,7 @@ static rsRetVal TCPSendGSSInit(void *pvData)
|
||||
|
||||
base = (gss_base_service_name == NULL) ? "host" : gss_base_service_name;
|
||||
out_tok.length = strlen(pData->f_hname) + strlen(base) + 2;
|
||||
if ((out_tok.value = malloc(out_tok.length)) == NULL) {
|
||||
ABORT_FINALIZE(RS_RET_OUT_OF_MEMORY);
|
||||
}
|
||||
CHKmalloc(out_tok.value = malloc(out_tok.length));
|
||||
strcpy(out_tok.value, base);
|
||||
strcat(out_tok.value, "@");
|
||||
strcat(out_tok.value, pData->f_hname);
|
||||
@ -371,7 +354,6 @@ static rsRetVal doTryResume(instanceData *pData)
|
||||
getFwdSyslogPt(pData), &hints, &res)) == 0) {
|
||||
dbgprintf("%s found, resuming.\n", pData->f_hname);
|
||||
pData->f_addr = res;
|
||||
pData->iRtryCnt = 0;
|
||||
pData->eDestState = eDestFORW;
|
||||
} else {
|
||||
iRet = RS_RET_SUSPENDED;
|
||||
@ -410,7 +392,6 @@ CODESTARTdoAction
|
||||
|
||||
case eDestFORW:
|
||||
dbgprintf(" %s:%s/%s\n", pData->f_hname, getFwdSyslogPt(pData), "tcp-gssapi");
|
||||
pData->ttSuspend = time(NULL);
|
||||
psz = (char*) ppString[0];
|
||||
l = strlen((char*) psz);
|
||||
if (l > MAXLINE)
|
||||
@ -613,8 +594,6 @@ CODE_STD_STRING_REQUESTparseSelectorAct(1)
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
if( (error = getaddrinfo(pData->f_hname, getFwdSyslogPt(pData), &hints, &res)) != 0) {
|
||||
pData->eDestState = eDestFORW_UNKN;
|
||||
pData->iRtryCnt = INET_RETRY_MAX;
|
||||
pData->ttSuspend = time(NULL);
|
||||
} else {
|
||||
pData->eDestState = eDestFORW;
|
||||
pData->f_addr = res;
|
||||
|
||||
@ -11,6 +11,7 @@ librsyslog_la_SOURCES = \
|
||||
syslogd-types.h \
|
||||
module-template.h \
|
||||
obj-types.h \
|
||||
nsd.h \
|
||||
glbl.h \
|
||||
glbl.c \
|
||||
msg.c \
|
||||
@ -75,18 +76,32 @@ if ENABLE_REGEXP
|
||||
pkglib_LTLIBRARIES += lmregexp.la
|
||||
lmregexp_la_SOURCES = regexp.c regexp.h
|
||||
lmregexp_la_CPPFLAGS = $(pthreads_cflags) $(rsrt_cflags)
|
||||
lmregexp_la_LDFLAGS = -module -avoid-version $(rsrt_libs)
|
||||
lmregexp_la_LDFLAGS = -module -avoid-version
|
||||
lmregexp_la_LIBADD =
|
||||
endif
|
||||
|
||||
if ENABLE_INET
|
||||
pkglib_LTLIBRARIES += lmnet.la
|
||||
pkglib_LTLIBRARIES += lmnet.la lmnetstrm.la
|
||||
#
|
||||
# network support
|
||||
#
|
||||
lmnet_la_SOURCES = net.c net.h
|
||||
lmnet_la_CPPFLAGS = $(pthreads_cflags) $(rsrt_cflags)
|
||||
lmnet_la_LDFLAGS = -module -avoid-version $(rsrt_libs)
|
||||
lmnet_la_LDFLAGS = -module -avoid-version
|
||||
lmnet_la_LIBADD =
|
||||
|
||||
# network streams
|
||||
lmnetstrm_la_SOURCES = netstrm.c netstrm.h
|
||||
lmnetstrm_la_CPPFLAGS = $(pthreads_cflags) $(rsrt_cflags)
|
||||
lmnetstrm_la_LDFLAGS = -module -avoid-version
|
||||
lmnetstrm_la_LIBADD =
|
||||
|
||||
# netstream drivers
|
||||
|
||||
# plain tcp driver
|
||||
pkglib_LTLIBRARIES += lmnsd_ptcp.la
|
||||
lmnsd_ptcp_la_SOURCES = nsd_ptcp.c nsd_ptcp.h
|
||||
lmnsd_ptcp_la_CPPFLAGS = $(pthreads_cflags) $(rsrt_cflags)
|
||||
lmnsd_ptcp_la_LDFLAGS = -module -avoid-version
|
||||
lmnsd_ptcp_la_LIBADD =
|
||||
endif # if ENABLE_INET
|
||||
|
||||
@ -38,6 +38,11 @@
|
||||
#include "cfsysline.h"
|
||||
#include "glbl.h"
|
||||
|
||||
/* some defaults */
|
||||
#ifndef DFLT_NETSTRM_DRVR
|
||||
# define DFLT_NETSTRM_DRVR ((uchar*)"lmnsd_ptcp")
|
||||
#endif
|
||||
|
||||
/* static data */
|
||||
DEFobjStaticHelpers
|
||||
|
||||
@ -54,6 +59,7 @@ static uchar *LocalHostName = NULL;/* our hostname - read-only after startup */
|
||||
static uchar *LocalDomain; /* our local domain name - read-only after startup */
|
||||
static char **StripDomains = NULL;/* these domains may be stripped before writing logs - r/o after s.u., never touched by init */
|
||||
static char **LocalHosts = NULL;/* these hosts are logged with their hostname - read-only after startup, never touched by init */
|
||||
static uchar *pszDfltNetstrmDrvr = NULL; /* module name of default netstream driver */
|
||||
|
||||
|
||||
/* define a macro for the simple properties' set and get functions
|
||||
@ -84,6 +90,7 @@ SIMP_PROP(StripDomains, StripDomains, char**)
|
||||
SIMP_PROP(LocalHosts, LocalHosts, char**)
|
||||
|
||||
SIMP_PROP_SET(LocalHostName, LocalHostName, uchar*)
|
||||
SIMP_PROP_SET(DfltNetstrmDrvr, pszDfltNetstrmDrvr, uchar*) // TODO: use custom function which frees existing value
|
||||
|
||||
#undef SIMP_PROP
|
||||
#undef SIMP_PROP_SET
|
||||
@ -99,8 +106,7 @@ GetLocalHostName(void)
|
||||
}
|
||||
|
||||
|
||||
/* return the current working directory.
|
||||
*/
|
||||
/* return the current working directory */
|
||||
static uchar*
|
||||
GetWorkDir(void)
|
||||
{
|
||||
@ -108,6 +114,14 @@ GetWorkDir(void)
|
||||
}
|
||||
|
||||
|
||||
/* return the current default netstream driver */
|
||||
static uchar*
|
||||
GetDfltNetstrmDrvr(void)
|
||||
{
|
||||
return(pszDfltNetstrmDrvr == NULL ? DFLT_NETSTRM_DRVR : pszWorkDir);
|
||||
}
|
||||
|
||||
|
||||
/* queryInterface function
|
||||
* rgerhards, 2008-02-21
|
||||
*/
|
||||
@ -134,6 +148,7 @@ CODESTARTobjQueryInterface(glbl)
|
||||
SIMP_PROP(LocalDomain)
|
||||
SIMP_PROP(StripDomains)
|
||||
SIMP_PROP(LocalHosts)
|
||||
SIMP_PROP(DfltNetstrmDrvr)
|
||||
#undef SIMP_PROP
|
||||
finalize_it:
|
||||
ENDobjQueryInterface(glbl)
|
||||
@ -144,6 +159,10 @@ ENDobjQueryInterface(glbl)
|
||||
*/
|
||||
static rsRetVal resetConfigVariables(uchar __attribute__((unused)) *pp, void __attribute__((unused)) *pVal)
|
||||
{
|
||||
if(pszDfltNetstrmDrvr != NULL) {
|
||||
free(pszDfltNetstrmDrvr);
|
||||
pszDfltNetstrmDrvr = NULL;
|
||||
}
|
||||
if(pszWorkDir != NULL) {
|
||||
free(pszWorkDir);
|
||||
pszWorkDir = NULL;
|
||||
@ -172,6 +191,8 @@ ENDObjClassInit(glbl)
|
||||
* rgerhards, 2008-04-17
|
||||
*/
|
||||
BEGINObjClassExit(glbl, OBJ_IS_CORE_MODULE) /* class, version */
|
||||
if(pszDfltNetstrmDrvr != NULL)
|
||||
free(pszDfltNetstrmDrvr);
|
||||
if(pszWorkDir != NULL)
|
||||
free(pszWorkDir);
|
||||
if(LocalHostName != NULL)
|
||||
|
||||
@ -48,6 +48,7 @@ BEGINinterface(glbl) /* name must also be changed in ENDinterface macro! */
|
||||
SIMP_PROP(LocalDomain, uchar*)
|
||||
SIMP_PROP(StripDomains, char**)
|
||||
SIMP_PROP(LocalHosts, char**)
|
||||
SIMP_PROP(DfltNetstrmDrvr, uchar*)
|
||||
#undef SIMP_PROP
|
||||
ENDinterface(glbl)
|
||||
#define glblCURR_IF_VERSION 1 /* increment whenever you change the interface structure! */
|
||||
|
||||
@ -666,7 +666,6 @@ gethname(struct sockaddr_storage *f, uchar *pszHostFQDN)
|
||||
if (error == 0) {
|
||||
memset (&hints, 0, sizeof (struct addrinfo));
|
||||
hints.ai_flags = AI_NUMERICHOST;
|
||||
hints.ai_socktype = SOCK_DGRAM;
|
||||
|
||||
/* we now do a lookup once again. This one should fail,
|
||||
* because we should not have obtained a non-numeric address. If
|
||||
|
||||
344
runtime/netstrm.c
Normal file
344
runtime/netstrm.c
Normal file
@ -0,0 +1,344 @@
|
||||
/* netstrmstrm.c
|
||||
*
|
||||
* This class implements a generic netstrmwork stream class. It supports
|
||||
* sending and receiving data streams over a netstrmwork. The class abstracts
|
||||
* the transport, though it is a safe assumption that TCP is being used.
|
||||
* The class has a number of properties, among which are also ones to
|
||||
* select privacy settings, eg by enabling TLS and/or GSSAPI. In the
|
||||
* long run, this class shall provide all stream-oriented netstrmwork
|
||||
* functionality inside rsyslog.
|
||||
*
|
||||
* It is a high-level class, which uses a number of helper objects
|
||||
* to carry out its work (including, and most importantly, transport
|
||||
* drivers).
|
||||
*
|
||||
* Work on this module begun 2008-04-17 by Rainer Gerhards. This code
|
||||
* borrows from librelp's tcp.c/.h code. librelp is dual licensed and
|
||||
* Rainer Gerhards and Adiscon GmbH have agreed to permit using the code
|
||||
* under the terms of the GNU Lesser General Public License.
|
||||
*
|
||||
* Copyright 2007, 2008 Rainer Gerhards and Adiscon GmbH.
|
||||
*
|
||||
* This file is part of the rsyslog runtime library.
|
||||
*
|
||||
* The rsyslog runtime library is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* The rsyslog runtime library 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 Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with the rsyslog runtime library. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
* A copy of the LGPL can be found in the file "COPYING.LESSER" in this distribution.
|
||||
*/
|
||||
#include "config.h"
|
||||
|
||||
#include "rsyslog.h"
|
||||
#include <stdio.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <assert.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <signal.h>
|
||||
#include <ctype.h>
|
||||
#include <netdb.h>
|
||||
#include <fnmatch.h>
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "syslogd-types.h"
|
||||
#include "module-template.h"
|
||||
#include "parse.h"
|
||||
#include "srUtils.h"
|
||||
#include "obj.h"
|
||||
#include "errmsg.h"
|
||||
#include "net.h"
|
||||
#include "nsd.h"
|
||||
#include "netstrm.h"
|
||||
|
||||
MODULE_TYPE_LIB
|
||||
|
||||
/* static data */
|
||||
DEFobjStaticHelpers
|
||||
DEFobjCurrIf(errmsg)
|
||||
DEFobjCurrIf(glbl)
|
||||
DEFobjCurrIf(net)
|
||||
|
||||
|
||||
/* load our low-level driver. This must be done before any
|
||||
* driver-specific functions (allmost all...) can be carried
|
||||
* out. Note that the driver's .ifIsLoaded is correctly
|
||||
* initialized by calloc() and we depend on that.
|
||||
* rgerhards, 2008-04-18
|
||||
*/
|
||||
static rsRetVal
|
||||
loadDrvr(netstrm_t *pThis)
|
||||
{
|
||||
uchar *pDrvrName;
|
||||
DEFiRet;
|
||||
|
||||
pDrvrName = pThis->pDrvrName;
|
||||
if(pDrvrName == NULL) /* if no drvr name is set, use system default */
|
||||
pDrvrName = glbl.GetDfltNetstrmDrvr();
|
||||
|
||||
pThis->Drvr.ifVersion = nsdCURR_IF_VERSION;
|
||||
/* The pDrvrName+2 below is a hack to obtain the object name. It
|
||||
* safes us to have yet another variable with the name without "lm" in
|
||||
* front of it. If we change the module load interface, we may re-think
|
||||
* about this hack, but for the time being it is efficient and clean
|
||||
* enough. -- rgerhards, 2008-04-18
|
||||
*/
|
||||
CHKiRet(obj.UseObj(__FILE__, pDrvrName+2, pDrvrName, (void*) &pThis->Drvr));
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* Standard-Constructor */
|
||||
BEGINobjConstruct(netstrm) /* be sure to specify the object type also in END macro! */
|
||||
ENDobjConstruct(netstrm)
|
||||
|
||||
|
||||
/* destructor for the netstrm object */
|
||||
BEGINobjDestruct(netstrm) /* be sure to specify the object type also in END and CODESTART macros! */
|
||||
CODESTARTobjDestruct(netstrm)
|
||||
if(pThis->pDrvrData != NULL)
|
||||
iRet = pThis->Drvr.Destruct(&pThis->pDrvrData);
|
||||
ENDobjDestruct(netstrm)
|
||||
|
||||
|
||||
/* ConstructionFinalizer */
|
||||
static rsRetVal
|
||||
netstrmConstructFinalize(netstrm_t *pThis)
|
||||
{
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, netstrm);
|
||||
CHKiRet(loadDrvr(pThis));
|
||||
CHKiRet(pThis->Drvr.Construct(&pThis->pDrvrData));
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
/* abort a connection. This is much like Destruct(), but tries
|
||||
* to discard any unsent data. -- rgerhards, 2008-03-24
|
||||
*/
|
||||
static rsRetVal
|
||||
AbortDestruct(netstrm_t **ppThis)
|
||||
{
|
||||
DEFiRet;
|
||||
assert(ppThis != NULL);
|
||||
ISOBJ_TYPE_assert((*ppThis), netstrm);
|
||||
|
||||
/* we do NOT exit on error, because that would make things worse */
|
||||
(*ppThis)->Drvr.Abort((*ppThis)->pDrvrData);
|
||||
iRet = netstrmDestruct(ppThis);
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
#if 0
|
||||
This is not yet working - wait until we arrive at the receiver side (distracts too much at the moment)
|
||||
|
||||
/* accept an incoming connection request, pNsdLstn provides the "listen socket" on which we can
|
||||
* accept the new session.
|
||||
* rgerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
AcceptConnReq(netstrm_t **ppThis, nsd_t *pNsdLstn)
|
||||
{
|
||||
netstrm_t *pThis = NULL;
|
||||
nsd_t *pNsd;
|
||||
DEFiRet;
|
||||
|
||||
assert(ppThis != NULL);
|
||||
|
||||
/* construct our object so that we can use it... */
|
||||
CHKiRet(netstrmConstruct(&pThis));
|
||||
|
||||
/* TODO: obtain hostname, normalize (callback?), save it */
|
||||
CHKiRet(FillRemHost(pThis, (struct sockaddr*) &addr));
|
||||
|
||||
/* set the new socket to non-blocking IO */
|
||||
if((sockflags = fcntl(iNewSock, F_GETFL)) != -1) {
|
||||
sockflags |= O_NONBLOCK;
|
||||
/* SETFL could fail too, so get it caught by the subsequent
|
||||
* error check.
|
||||
*/
|
||||
sockflags = fcntl(iNewSock, F_SETFL, sockflags);
|
||||
}
|
||||
if(sockflags == -1) {
|
||||
dbgprintf("error %d setting fcntl(O_NONBLOCK) on tcp socket %d", errno, iNewSock);
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
}
|
||||
|
||||
pThis->sock = iNewSock;
|
||||
|
||||
*ppThis = pThis;
|
||||
|
||||
finalize_it:
|
||||
if(iRet != RS_RET_OK) {
|
||||
if(pThis != NULL)
|
||||
netstrmDestruct(&pThis);
|
||||
/* the close may be redundant, but that doesn't hurt... */
|
||||
if(iNewSock >= 0)
|
||||
close(iNewSock);
|
||||
}
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
/* initialize the tcp socket for a listner
|
||||
* pLstnPort must point to a port name or number. NULL is NOT permitted
|
||||
* (hint: we need to be careful when we use this module together with librelp,
|
||||
* there NULL indicates the default port
|
||||
* default is used.
|
||||
* gerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
LstnInit(netstrm_t *pThis, uchar *pLstnPort)
|
||||
{
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, netstrm);
|
||||
assert(pLstnPort != NULL);
|
||||
CHKiRet(pThis->Drvr.LstnInit(pThis->pDrvrData, pLstnPort));
|
||||
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* receive data from a tcp socket
|
||||
* The lenBuf parameter must contain the max buffer size on entry and contains
|
||||
* the number of octets read (or -1 in case of error) on exit. This function
|
||||
* never blocks, not even when called on a blocking socket. That is important
|
||||
* for client sockets, which are set to block during send, but should not
|
||||
* block when trying to read data. If *pLenBuf is -1, an error occured and
|
||||
* errno holds the exact error cause.
|
||||
* rgerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
Rcv(netstrm_t *pThis, uchar *pBuf, ssize_t *pLenBuf)
|
||||
{
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, netstrm);
|
||||
iRet = pThis->Drvr.Rcv(pThis->pDrvrData, pBuf, pLenBuf);
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* send a buffer. On entry, pLenBuf contains the number of octets to
|
||||
* write. On exit, it contains the number of octets actually written.
|
||||
* If this number is lower than on entry, only a partial buffer has
|
||||
* been written.
|
||||
* rgerhards, 2008-03-19
|
||||
*/
|
||||
static rsRetVal
|
||||
Send(netstrm_t *pThis, uchar *pBuf, ssize_t *pLenBuf)
|
||||
{
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, netstrm);
|
||||
iRet = pThis->Drvr.Send(pThis->pDrvrData, pBuf, pLenBuf);
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* open a connection to a remote host (server).
|
||||
* rgerhards, 2008-03-19
|
||||
*/
|
||||
static rsRetVal
|
||||
Connect(netstrm_t *pThis, int family, uchar *port, uchar *host)
|
||||
{
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, netstrm);
|
||||
assert(port != NULL);
|
||||
assert(host != NULL);
|
||||
iRet = pThis->Drvr.Connect(pThis->pDrvrData, family, port, host);
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* queryInterface function
|
||||
*/
|
||||
BEGINobjQueryInterface(netstrm)
|
||||
CODESTARTobjQueryInterface(netstrm)
|
||||
if(pIf->ifVersion != netstrmCURR_IF_VERSION) {/* check for current version, increment on each change */
|
||||
ABORT_FINALIZE(RS_RET_INTERFACE_NOT_SUPPORTED);
|
||||
}
|
||||
|
||||
/* ok, we have the right interface, so let's fill it
|
||||
* Please note that we may also do some backwards-compatibility
|
||||
* work here (if we can support an older interface version - that,
|
||||
* of course, also affects the "if" above).
|
||||
*/
|
||||
pIf->Construct = netstrmConstruct;
|
||||
pIf->ConstructFinalize = netstrmConstructFinalize;
|
||||
pIf->Destruct = netstrmDestruct;
|
||||
pIf->AbortDestruct = AbortDestruct;
|
||||
pIf->LstnInit = LstnInit;
|
||||
// TODO: add later: pIf->AcceptConnReq = AcceptConnReq;
|
||||
pIf->Rcv = Rcv;
|
||||
pIf->Send = Send;
|
||||
pIf->Connect = Connect;
|
||||
finalize_it:
|
||||
ENDobjQueryInterface(netstrm)
|
||||
|
||||
|
||||
/* exit our class
|
||||
*/
|
||||
BEGINObjClassExit(netstrm, OBJ_IS_LOADABLE_MODULE) /* CHANGE class also in END MACRO! */
|
||||
CODESTARTObjClassExit(netstrm)
|
||||
/* release objects we no longer need */
|
||||
objRelease(net, CORE_COMPONENT);
|
||||
objRelease(glbl, CORE_COMPONENT);
|
||||
objRelease(errmsg, CORE_COMPONENT);
|
||||
ENDObjClassExit(netstrm)
|
||||
|
||||
|
||||
/* Initialize the netstrm class. Must be called as the very first method
|
||||
* before anything else is called inside this class.
|
||||
* rgerhards, 2008-02-19
|
||||
*/
|
||||
BEGINAbstractObjClassInit(netstrm, 1, OBJ_IS_CORE_MODULE) /* class, version */
|
||||
/* request objects we use */
|
||||
CHKiRet(objUse(errmsg, CORE_COMPONENT));
|
||||
CHKiRet(objUse(glbl, CORE_COMPONENT));
|
||||
CHKiRet(objUse(net, CORE_COMPONENT));
|
||||
|
||||
/* set our own handlers */
|
||||
ENDObjClassInit(netstrm)
|
||||
|
||||
|
||||
/* --------------- here now comes the plumbing that makes as a library module --------------- */
|
||||
|
||||
|
||||
BEGINmodExit
|
||||
CODESTARTmodExit
|
||||
netstrmClassExit();
|
||||
ENDmodExit
|
||||
|
||||
|
||||
BEGINqueryEtryPt
|
||||
CODESTARTqueryEtryPt
|
||||
CODEqueryEtryPt_STD_LIB_QUERIES
|
||||
ENDqueryEtryPt
|
||||
|
||||
|
||||
BEGINmodInit()
|
||||
CODESTARTmodInit
|
||||
*ipIFVersProvided = CURR_MOD_IF_VERSION; /* we only support the current interface specification */
|
||||
|
||||
/* Initialize all classes that are in our module - this includes ourselfs */
|
||||
CHKiRet(netstrmClassInit(pModInfo)); /* must be done after tcps_sess, as we use it */
|
||||
ENDmodInit
|
||||
/* vi:set ai:
|
||||
*/
|
||||
58
runtime/netstrm.h
Normal file
58
runtime/netstrm.h
Normal file
@ -0,0 +1,58 @@
|
||||
/* Definitions for the stream-based netstrmworking class.
|
||||
*
|
||||
* Copyright 2007, 2008 Rainer Gerhards and Adiscon GmbH.
|
||||
*
|
||||
* This file is part of the rsyslog runtime library.
|
||||
*
|
||||
* The rsyslog runtime library is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* The rsyslog runtime library 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 Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with the rsyslog runtime library. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
* A copy of the LGPL can be found in the file "COPYING.LESSER" in this distribution.
|
||||
*/
|
||||
|
||||
#ifndef INCLUDED_NETSTRM_H
|
||||
#define INCLUDED_NETSTRM_H
|
||||
|
||||
#include "nsd.h" /* we need our driver interface to be defined */
|
||||
|
||||
/* the netstrm object */
|
||||
struct netstrm_s {
|
||||
BEGINobjInstance; /* Data to implement generic object - MUST be the first data element! */
|
||||
nsd_if_t Drvr; /**< our stream driver */
|
||||
nsd_t *pDrvrData; /**< the driver's data elements */
|
||||
uchar *pDrvrName; /**< nsd driver name to use, or NULL if system default */
|
||||
};
|
||||
|
||||
|
||||
/* interface */
|
||||
BEGINinterface(netstrm) /* name must also be changed in ENDinterface macro! */
|
||||
rsRetVal (*Construct)(netstrm_t **ppThis);
|
||||
rsRetVal (*ConstructFinalize)(netstrm_t *pThis);
|
||||
rsRetVal (*Destruct)(netstrm_t **ppThis);
|
||||
rsRetVal (*AbortDestruct)(netstrm_t **ppThis);
|
||||
rsRetVal (*LstnInit)(netstrm_t *pThis, unsigned char *pLstnPort);
|
||||
rsRetVal (*AcceptConnReq)(netstrm_t **ppThis, int sock);
|
||||
rsRetVal (*Rcv)(netstrm_t *pThis, uchar *pRcvBuf, ssize_t *pLenBuf);
|
||||
rsRetVal (*Send)(netstrm_t *pThis, uchar *pBuf, ssize_t *pLenBuf);
|
||||
rsRetVal (*Connect)(netstrm_t *pThis, int family, unsigned char *port, unsigned char *host);
|
||||
ENDinterface(netstrm)
|
||||
#define netstrmCURR_IF_VERSION 1 /* increment whenever you change the interface structure! */
|
||||
|
||||
/* prototypes */
|
||||
PROTOTYPEObj(netstrm);
|
||||
|
||||
/* the name of our library binary */
|
||||
#define LM_NETSTRM_FILENAME "lmnetstrm"
|
||||
|
||||
#endif /* #ifndef INCLUDED_NETSTRM_H */
|
||||
48
runtime/nsd.h
Normal file
48
runtime/nsd.h
Normal file
@ -0,0 +1,48 @@
|
||||
/* The interface definition for "NetStream Drivers" (nsd).
|
||||
*
|
||||
* This is just an abstract driver interface, which needs to be
|
||||
* implemented by concrete classes. As such, no nsd data type itself
|
||||
* is defined.
|
||||
*
|
||||
* Copyright 2008 Rainer Gerhards and Adiscon GmbH.
|
||||
*
|
||||
* This file is part of the rsyslog runtime library.
|
||||
*
|
||||
* The rsyslog runtime library is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* The rsyslog runtime library 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 Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with the rsyslog runtime library. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
* A copy of the LGPL can be found in the file "COPYING.LESSER" in this distribution.
|
||||
*/
|
||||
|
||||
#ifndef INCLUDED_NSD_H
|
||||
#define INCLUDED_NSD_H
|
||||
|
||||
/* nsd_t is actually obj_t (which is somewhat better than void* but in essence
|
||||
* much the same).
|
||||
*/
|
||||
|
||||
/* interface */
|
||||
BEGINinterface(nsd) /* name must also be changed in ENDinterface macro! */
|
||||
rsRetVal (*Construct)(nsd_t **ppThis);
|
||||
rsRetVal (*Destruct)(nsd_t **ppThis);
|
||||
rsRetVal (*Abort)(nsd_t *pThis);
|
||||
rsRetVal (*LstnInit)(nsd_t *pThis, unsigned char *pLstnPort);
|
||||
rsRetVal (*AcceptConnReq)(nsd_t **ppThis, int sock);
|
||||
rsRetVal (*Rcv)(nsd_t *pThis, uchar *pRcvBuf, ssize_t *pLenBuf);
|
||||
rsRetVal (*Send)(nsd_t *pThis, uchar *pBuf, ssize_t *pLenBuf);
|
||||
rsRetVal (*Connect)(nsd_t *pThis, int family, unsigned char *port, unsigned char *host);
|
||||
ENDinterface(nsd)
|
||||
#define nsdCURR_IF_VERSION 1 /* increment whenever you change the interface structure! */
|
||||
|
||||
#endif /* #ifndef INCLUDED_NSD_H */
|
||||
579
runtime/nsd_ptcp.c
Normal file
579
runtime/nsd_ptcp.c
Normal file
@ -0,0 +1,579 @@
|
||||
/* nsd_ptcp.c
|
||||
*
|
||||
* An implementation of the nsd interface for plain tcp sockets.
|
||||
*
|
||||
* Copyright 2007, 2008 Rainer Gerhards and Adiscon GmbH.
|
||||
*
|
||||
* This file is part of the rsyslog runtime library.
|
||||
*
|
||||
* The rsyslog runtime library is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* The rsyslog runtime library 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 Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with the rsyslog runtime library. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
* A copy of the LGPL can be found in the file "COPYING.LESSER" in this distribution.
|
||||
*/
|
||||
#include "config.h"
|
||||
|
||||
#include "rsyslog.h"
|
||||
#include <stdio.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <assert.h>
|
||||
#include <errno.h>
|
||||
#include <string.h>
|
||||
#include <signal.h>
|
||||
#include <ctype.h>
|
||||
#include <netdb.h>
|
||||
#include <fnmatch.h>
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "syslogd-types.h"
|
||||
#include "module-template.h"
|
||||
#include "parse.h"
|
||||
#include "srUtils.h"
|
||||
#include "obj.h"
|
||||
#include "errmsg.h"
|
||||
#include "net.h"
|
||||
#include "nsd_ptcp.h"
|
||||
|
||||
MODULE_TYPE_LIB
|
||||
|
||||
/* static data */
|
||||
DEFobjStaticHelpers
|
||||
DEFobjCurrIf(errmsg)
|
||||
DEFobjCurrIf(glbl)
|
||||
DEFobjCurrIf(net)
|
||||
|
||||
|
||||
/* Standard-Constructor
|
||||
*/
|
||||
BEGINobjConstruct(nsd_ptcp) /* be sure to specify the object type also in END macro! */
|
||||
pThis->sock = -1;
|
||||
pThis->iSessMax = 500; /* default max nbr of sessions -TODO:make configurable--rgerhards, 2008-04-17*/
|
||||
ENDobjConstruct(nsd_ptcp)
|
||||
|
||||
|
||||
/* destructor for the nsd_ptcp object */
|
||||
BEGINobjDestruct(nsd_ptcp) /* be sure to specify the object type also in END and CODESTART macros! */
|
||||
int i;
|
||||
CODESTARTobjDestruct(nsd_pctp)
|
||||
if(pThis->sock != -1) {
|
||||
close(pThis->sock);
|
||||
pThis->sock = -1;
|
||||
}
|
||||
|
||||
if(pThis->socks != NULL) {
|
||||
/* if we have some sockets at this stage, we need to close them */
|
||||
for(i = 1 ; i <= pThis->socks[0] ; ++i)
|
||||
close(pThis->socks[i]);
|
||||
free(pThis->socks);
|
||||
}
|
||||
|
||||
if(pThis->pRemHostIP != NULL)
|
||||
free(pThis->pRemHostIP);
|
||||
if(pThis->pRemHostName != NULL)
|
||||
free(pThis->pRemHostName);
|
||||
ENDobjDestruct(nsd_ptcp)
|
||||
|
||||
|
||||
/* abort a connection. This is meant to be called immediately
|
||||
* before the Destruct call. -- rgerhards, 2008-03-24
|
||||
*/
|
||||
static rsRetVal
|
||||
Abort(nsd_t *pNsd)
|
||||
{
|
||||
struct linger ling;
|
||||
nsd_ptcp_t *pThis = (nsd_ptcp_t*) pNsd;
|
||||
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert((pThis), nsd_ptcp);
|
||||
|
||||
if((pThis)->sock != -1) {
|
||||
ling.l_onoff = 1;
|
||||
ling.l_linger = 0;
|
||||
if(setsockopt((pThis)->sock, SOL_SOCKET, SO_LINGER, &ling, sizeof(ling)) < 0 ) {
|
||||
dbgprintf("could not set SO_LINGER, errno %d\n", errno);
|
||||
}
|
||||
}
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* Set pRemHost based on the address provided. This is to be called upon accept()ing
|
||||
* a connection request. It must be provided by the socket we received the
|
||||
* message on as well as a NI_MAXHOST size large character buffer for the FQDN.
|
||||
* Please see http://www.hmug.org/man/3/getnameinfo.php (under Caveats)
|
||||
* for some explanation of the code found below. If we detect a malicious
|
||||
* hostname, we return RS_RET_MALICIOUS_HNAME and let the caller decide
|
||||
* on how to deal with that.
|
||||
* rgerhards, 2008-03-31
|
||||
*/
|
||||
static rsRetVal
|
||||
FillRemHost(nsd_ptcp_t *pThis, struct sockaddr *pAddr)
|
||||
{
|
||||
int error;
|
||||
uchar szIP[NI_MAXHOST] = "";
|
||||
uchar szHname[NI_MAXHOST] = "";
|
||||
struct addrinfo hints, *res;
|
||||
size_t len;
|
||||
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, nsd_ptcp);
|
||||
assert(pAddr != NULL);
|
||||
|
||||
error = getnameinfo(pAddr, SALEN(pAddr), (char*)szIP, sizeof(szIP), NULL, 0, NI_NUMERICHOST);
|
||||
|
||||
if(error) {
|
||||
dbgprintf("Malformed from address %s\n", gai_strerror(error));
|
||||
strcpy((char*)szHname, "???");
|
||||
strcpy((char*)szIP, "???");
|
||||
ABORT_FINALIZE(RS_RET_INVALID_HNAME);
|
||||
}
|
||||
|
||||
if(!glbl.GetDisableDNS()) {
|
||||
error = getnameinfo(pAddr, SALEN(pAddr), (char*)szHname, NI_MAXHOST, NULL, 0, NI_NAMEREQD);
|
||||
if(error == 0) {
|
||||
memset (&hints, 0, sizeof (struct addrinfo));
|
||||
hints.ai_flags = AI_NUMERICHOST;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
/* we now do a lookup once again. This one should fail,
|
||||
* because we should not have obtained a non-numeric address. If
|
||||
* we got a numeric one, someone messed with DNS!
|
||||
*/
|
||||
if(getaddrinfo((char*)szHname, NULL, &hints, &res) == 0) {
|
||||
freeaddrinfo (res);
|
||||
/* OK, we know we have evil, so let's indicate this to our caller */
|
||||
snprintf((char*)szHname, NI_MAXHOST, "[MALICIOUS:IP=%s]", szIP);
|
||||
dbgprintf("Malicious PTR record, IP = \"%s\" HOST = \"%s\"", szIP, szHname);
|
||||
iRet = RS_RET_MALICIOUS_HNAME;
|
||||
}
|
||||
} else {
|
||||
strcpy((char*)szHname, (char*)szIP);
|
||||
}
|
||||
} else {
|
||||
strcpy((char*)szHname, (char*)szIP);
|
||||
}
|
||||
|
||||
/* We now have the names, so now let's allocate memory and store them permanently.
|
||||
* (side note: we may hold on to these values for quite a while, thus we trim their
|
||||
* memory consumption)
|
||||
*/
|
||||
len = strlen((char*)szIP) + 1; /* +1 for \0 byte */
|
||||
if((pThis->pRemHostIP = malloc(len)) == NULL)
|
||||
ABORT_FINALIZE(RS_RET_OUT_OF_MEMORY);
|
||||
memcpy(pThis->pRemHostIP, szIP, len);
|
||||
|
||||
len = strlen((char*)szHname) + 1; /* +1 for \0 byte */
|
||||
if((pThis->pRemHostName = malloc(len)) == NULL) {
|
||||
free(pThis->pRemHostIP); /* prevent leak */
|
||||
pThis->pRemHostIP = NULL;
|
||||
ABORT_FINALIZE(RS_RET_OUT_OF_MEMORY);
|
||||
}
|
||||
memcpy(pThis->pRemHostName, szHname, len);
|
||||
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/* accept an incoming connection request, sock provides the socket on which we can
|
||||
* accept the new session.
|
||||
* rgerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
AcceptConnReq(nsd_t **ppThis, int sock)
|
||||
{
|
||||
int sockflags;
|
||||
struct sockaddr_storage addr;
|
||||
socklen_t addrlen = sizeof(addr);
|
||||
nsd_ptcp_t *pThis = NULL;
|
||||
int iNewSock = -1;
|
||||
|
||||
DEFiRet;
|
||||
assert(ppThis != NULL);
|
||||
|
||||
iNewSock = accept(sock, (struct sockaddr*) &addr, &addrlen);
|
||||
if(iNewSock < 0) {
|
||||
ABORT_FINALIZE(RS_RET_ACCEPT_ERR);
|
||||
}
|
||||
|
||||
/* construct our object so that we can use it... */
|
||||
CHKiRet(nsd_ptcpConstruct(&pThis));
|
||||
|
||||
CHKiRet(FillRemHost(pThis, (struct sockaddr*) &addr));
|
||||
|
||||
/* set the new socket to non-blocking IO */
|
||||
if((sockflags = fcntl(iNewSock, F_GETFL)) != -1) {
|
||||
sockflags |= O_NONBLOCK;
|
||||
/* SETFL could fail too, so get it caught by the subsequent
|
||||
* error check.
|
||||
*/
|
||||
sockflags = fcntl(iNewSock, F_SETFL, sockflags);
|
||||
}
|
||||
if(sockflags == -1) {
|
||||
dbgprintf("error %d setting fcntl(O_NONBLOCK) on tcp socket %d", errno, iNewSock);
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
}
|
||||
|
||||
pThis->sock = iNewSock;
|
||||
|
||||
*ppThis = (nsd_t*) pThis;
|
||||
|
||||
finalize_it:
|
||||
if(iRet != RS_RET_OK) {
|
||||
if(pThis != NULL)
|
||||
nsd_ptcpDestruct(&pThis);
|
||||
/* the close may be redundant, but that doesn't hurt... */
|
||||
if(iNewSock >= 0)
|
||||
close(iNewSock);
|
||||
}
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* initialize the tcp socket for a listner
|
||||
* pLstnPort must point to a port name or number. NULL is NOT permitted
|
||||
* (hint: we need to be careful when we use this module together with librelp,
|
||||
* there NULL indicates the default port
|
||||
* default is used.
|
||||
* gerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
LstnInit(nsd_t *pNsd, uchar *pLstnPort)
|
||||
{
|
||||
nsd_ptcp_t *pThis = (nsd_ptcp_t*) pNsd;
|
||||
struct addrinfo hints, *res, *r;
|
||||
int error, maxs, *s, on = 1;
|
||||
int sockflags;
|
||||
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, nsd_ptcp);
|
||||
assert(pLstnPort != NULL);
|
||||
|
||||
dbgprintf("creating tcp listen socket on port %s\n", pLstnPort);
|
||||
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
hints.ai_flags = AI_PASSIVE;
|
||||
hints.ai_family = glbl.GetDefPFFamily();
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
|
||||
error = getaddrinfo(NULL, (char*) pLstnPort, &hints, &res);
|
||||
if(error) {
|
||||
dbgprintf("error %d querying port '%s'\n", error, pLstnPort);
|
||||
ABORT_FINALIZE(RS_RET_INVALID_PORT);
|
||||
}
|
||||
|
||||
/* Count max number of sockets we may open */
|
||||
for(maxs = 0, r = res; r != NULL ; r = r->ai_next, maxs++)
|
||||
/* EMPTY */;
|
||||
pThis->socks = malloc((maxs+1) * sizeof(int));
|
||||
if (pThis->socks == NULL) {
|
||||
dbgprintf("couldn't allocate memory for TCP listen sockets, suspending RELP message reception.");
|
||||
freeaddrinfo(res);
|
||||
ABORT_FINALIZE(RS_RET_OUT_OF_MEMORY);
|
||||
}
|
||||
|
||||
*pThis->socks = 0; /* num of sockets counter at start of array */
|
||||
s = pThis->socks + 1;
|
||||
for(r = res; r != NULL ; r = r->ai_next) {
|
||||
*s = socket(r->ai_family, r->ai_socktype, r->ai_protocol);
|
||||
if (*s < 0) {
|
||||
if(!(r->ai_family == PF_INET6 && errno == EAFNOSUPPORT))
|
||||
dbgprintf("creating tcp listen socket");
|
||||
/* it is debatable if PF_INET with EAFNOSUPPORT should
|
||||
* also be ignored...
|
||||
*/
|
||||
continue;
|
||||
}
|
||||
|
||||
#ifdef IPV6_V6ONLY
|
||||
if (r->ai_family == AF_INET6) {
|
||||
int iOn = 1;
|
||||
if (setsockopt(*s, IPPROTO_IPV6, IPV6_V6ONLY,
|
||||
(char *)&iOn, sizeof (iOn)) < 0) {
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
if(setsockopt(*s, SOL_SOCKET, SO_REUSEADDR, (char *) &on, sizeof(on)) < 0 ) {
|
||||
dbgprintf("error %d setting tcp socket option\n", errno);
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
|
||||
/* We use non-blocking IO! */
|
||||
if((sockflags = fcntl(*s, F_GETFL)) != -1) {
|
||||
sockflags |= O_NONBLOCK;
|
||||
/* SETFL could fail too, so get it caught by the subsequent
|
||||
* error check.
|
||||
*/
|
||||
sockflags = fcntl(*s, F_SETFL, sockflags);
|
||||
}
|
||||
if(sockflags == -1) {
|
||||
dbgprintf("error %d setting fcntl(O_NONBLOCK) on tcp socket", errno);
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/* We need to enable BSD compatibility. Otherwise an attacker
|
||||
* could flood our log files by sending us tons of ICMP errors.
|
||||
*/
|
||||
#ifndef BSD
|
||||
if(net.should_use_so_bsdcompat()) {
|
||||
if (setsockopt(*s, SOL_SOCKET, SO_BSDCOMPAT,
|
||||
(char *) &on, sizeof(on)) < 0) {
|
||||
errmsg.LogError(NO_ERRCODE, "TCP setsockopt(BSDCOMPAT)");
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
if( (bind(*s, r->ai_addr, r->ai_addrlen) < 0)
|
||||
#ifndef IPV6_V6ONLY
|
||||
&& (errno != EADDRINUSE)
|
||||
#endif
|
||||
) {
|
||||
dbgprintf("error %d while binding tcp socket", errno);
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if(listen(*s,pThis->iSessMax / 10 + 5) < 0) {
|
||||
/* If the listen fails, it most probably fails because we ask
|
||||
* for a too-large backlog. So in this case we first set back
|
||||
* to a fixed, reasonable, limit that should work. Only if
|
||||
* that fails, too, we give up.
|
||||
*/
|
||||
dbgprintf("listen with a backlog of %d failed - retrying with default of 32.",
|
||||
pThis->iSessMax / 10 + 5);
|
||||
if(listen(*s, 32) < 0) {
|
||||
dbgprintf("tcp listen error %d, suspending\n", errno);
|
||||
close(*s);
|
||||
*s = -1;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
(*pThis->socks)++;
|
||||
s++;
|
||||
}
|
||||
|
||||
if(res != NULL)
|
||||
freeaddrinfo(res);
|
||||
|
||||
if(*pThis->socks != maxs)
|
||||
dbgprintf("We could initialize %d RELP TCP listen sockets out of %d we received "
|
||||
"- this may or may not be an error indication.\n", *pThis->socks, maxs);
|
||||
|
||||
if(*pThis->socks == 0) {
|
||||
dbgprintf("No RELP TCP listen socket could successfully be initialized, "
|
||||
"message reception via RELP disabled.\n");
|
||||
free(pThis->socks);
|
||||
ABORT_FINALIZE(RS_RET_COULD_NOT_BIND);
|
||||
}
|
||||
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* receive data from a tcp socket
|
||||
* The lenBuf parameter must contain the max buffer size on entry and contains
|
||||
* the number of octets read (or -1 in case of error) on exit. This function
|
||||
* never blocks, not even when called on a blocking socket. That is important
|
||||
* for client sockets, which are set to block during send, but should not
|
||||
* block when trying to read data. If *pLenBuf is -1, an error occured and
|
||||
* errno holds the exact error cause.
|
||||
* rgerhards, 2008-03-17
|
||||
*/
|
||||
static rsRetVal
|
||||
Rcv(nsd_t *pNsd, uchar *pRcvBuf, ssize_t *pLenBuf)
|
||||
{
|
||||
DEFiRet;
|
||||
nsd_ptcp_t *pThis = (nsd_ptcp_t*) pNsd;
|
||||
ISOBJ_TYPE_assert(pThis, nsd_ptcp);
|
||||
|
||||
*pLenBuf = recv(pThis->sock, pRcvBuf, *pLenBuf, MSG_DONTWAIT);
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* send a buffer. On entry, pLenBuf contains the number of octets to
|
||||
* write. On exit, it contains the number of octets actually written.
|
||||
* If this number is lower than on entry, only a partial buffer has
|
||||
* been written.
|
||||
* rgerhards, 2008-03-19
|
||||
*/
|
||||
static rsRetVal
|
||||
Send(nsd_t *pNsd, uchar *pBuf, ssize_t *pLenBuf)
|
||||
{
|
||||
nsd_ptcp_t *pThis = (nsd_ptcp_t*) pNsd;
|
||||
ssize_t written;
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, nsd_ptcp);
|
||||
|
||||
written = send(pThis->sock, pBuf, *pLenBuf, 0);
|
||||
|
||||
if(written == -1) {
|
||||
switch(errno) {
|
||||
case EAGAIN:
|
||||
case EINTR:
|
||||
/* this is fine, just retry... */
|
||||
written = 0;
|
||||
break;
|
||||
default:
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
*pLenBuf = written;
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* open a connection to a remote host (server).
|
||||
* rgerhards, 2008-03-19
|
||||
*/
|
||||
static rsRetVal
|
||||
Connect(nsd_t *pNsd, int family, uchar *port, uchar *host)
|
||||
{
|
||||
nsd_ptcp_t *pThis = (nsd_ptcp_t*) pNsd;
|
||||
struct addrinfo *res = NULL;
|
||||
struct addrinfo hints;
|
||||
|
||||
DEFiRet;
|
||||
ISOBJ_TYPE_assert(pThis, nsd_ptcp);
|
||||
assert(port != NULL);
|
||||
assert(host != NULL);
|
||||
assert(pThis->sock == -1);
|
||||
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
hints.ai_family = family;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
if(getaddrinfo((char*)host, (char*)port, &hints, &res) != 0) {
|
||||
dbgprintf("error %d in getaddrinfo\n", errno);
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
}
|
||||
|
||||
if((pThis->sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol)) == -1) {
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
}
|
||||
|
||||
if(connect(pThis->sock, res->ai_addr, res->ai_addrlen) != 0) {
|
||||
ABORT_FINALIZE(RS_RET_IO_ERROR);
|
||||
}
|
||||
|
||||
finalize_it:
|
||||
if(res != NULL)
|
||||
freeaddrinfo(res);
|
||||
|
||||
if(iRet != RS_RET_OK) {
|
||||
if(pThis->sock != -1) {
|
||||
close(pThis->sock);
|
||||
pThis->sock = -1;
|
||||
}
|
||||
}
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* queryInterface function */
|
||||
BEGINobjQueryInterface(nsd_ptcp)
|
||||
CODESTARTobjQueryInterface(nsd_ptcp)
|
||||
if(pIf->ifVersion != nsdCURR_IF_VERSION) {/* check for current version, increment on each change */
|
||||
ABORT_FINALIZE(RS_RET_INTERFACE_NOT_SUPPORTED);
|
||||
}
|
||||
|
||||
/* ok, we have the right interface, so let's fill it
|
||||
* Please note that we may also do some backwards-compatibility
|
||||
* work here (if we can support an older interface version - that,
|
||||
* of course, also affects the "if" above).
|
||||
*/
|
||||
pIf->Construct = (rsRetVal(*)(nsd_t**)) nsd_ptcpConstruct;
|
||||
pIf->Destruct = (rsRetVal(*)(nsd_t**)) nsd_ptcpDestruct;
|
||||
pIf->Abort = Abort;
|
||||
pIf->LstnInit = LstnInit;
|
||||
pIf->AcceptConnReq = AcceptConnReq;
|
||||
pIf->Rcv = Rcv;
|
||||
pIf->Send = Send;
|
||||
pIf->Connect = Connect;
|
||||
finalize_it:
|
||||
ENDobjQueryInterface(nsd_ptcp)
|
||||
|
||||
|
||||
/* exit our class
|
||||
*/
|
||||
BEGINObjClassExit(nsd_ptcp, OBJ_IS_LOADABLE_MODULE) /* CHANGE class also in END MACRO! */
|
||||
CODESTARTObjClassExit(nsd_ptcp)
|
||||
/* release objects we no longer need */
|
||||
objRelease(net, CORE_COMPONENT);
|
||||
objRelease(glbl, CORE_COMPONENT);
|
||||
objRelease(errmsg, CORE_COMPONENT);
|
||||
ENDObjClassExit(nsd_ptcp)
|
||||
|
||||
|
||||
/* Initialize the nsd_ptcp class. Must be called as the very first method
|
||||
* before anything else is called inside this class.
|
||||
* rgerhards, 2008-02-19
|
||||
*/
|
||||
BEGINObjClassInit(nsd_ptcp, 1, OBJ_IS_LOADABLE_MODULE) /* class, version */
|
||||
/* request objects we use */
|
||||
CHKiRet(objUse(errmsg, CORE_COMPONENT));
|
||||
CHKiRet(objUse(glbl, CORE_COMPONENT));
|
||||
CHKiRet(objUse(net, CORE_COMPONENT));
|
||||
|
||||
/* set our own handlers */
|
||||
ENDObjClassInit(nsd_ptcp)
|
||||
|
||||
|
||||
/* --------------- here now comes the plumbing that makes as a library module --------------- */
|
||||
|
||||
|
||||
BEGINmodExit
|
||||
CODESTARTmodExit
|
||||
nsd_ptcpClassExit();
|
||||
ENDmodExit
|
||||
|
||||
|
||||
BEGINqueryEtryPt
|
||||
CODESTARTqueryEtryPt
|
||||
CODEqueryEtryPt_STD_LIB_QUERIES
|
||||
ENDqueryEtryPt
|
||||
|
||||
|
||||
BEGINmodInit()
|
||||
CODESTARTmodInit
|
||||
*ipIFVersProvided = CURR_MOD_IF_VERSION; /* we only support the current interface specification */
|
||||
|
||||
/* Initialize all classes that are in our module - this includes ourselfs */
|
||||
CHKiRet(nsd_ptcpClassInit(pModInfo)); /* must be done after tcps_sess, as we use it */
|
||||
ENDmodInit
|
||||
/* vi:set ai:
|
||||
*/
|
||||
48
runtime/nsd_ptcp.h
Normal file
48
runtime/nsd_ptcp.h
Normal file
@ -0,0 +1,48 @@
|
||||
/* An implementation of the nsd interface for plain tcp sockets.
|
||||
*
|
||||
* Copyright 2007, 2008 Rainer Gerhards and Adiscon GmbH.
|
||||
*
|
||||
* This file is part of the rsyslog runtime library.
|
||||
*
|
||||
* The rsyslog runtime library is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* The rsyslog runtime library 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 Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with the rsyslog runtime library. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
* A copy of the LGPL can be found in the file "COPYING.LESSER" in this distribution.
|
||||
*/
|
||||
|
||||
#ifndef INCLUDED_NSD_PTCP_H
|
||||
#define INCLUDED_NSD_PTCP_H
|
||||
|
||||
#include "nsd.h"
|
||||
typedef nsd_if_t nsd_ptcp_if_t; /* we just *implement* this interface */
|
||||
|
||||
/* the nsd_ptcp object */
|
||||
struct nsd_ptcp_s {
|
||||
BEGINobjInstance; /* Data to implement generic object - MUST be the first data element! */
|
||||
uchar *pRemHostIP; /**< IP address of remote peer (currently used in server mode, only) */
|
||||
uchar *pRemHostName; /**< host name of remote peer (currently used in server mode, only) */
|
||||
int sock; /**< the socket we use for regular, single-socket, operations */
|
||||
int *socks; /**< the socket(s) we use for listeners, element 0 has nbr of socks */
|
||||
int iSessMax; /**< maximum number of sessions permitted */
|
||||
};
|
||||
|
||||
/* interface is defined in nsd.h, we just implement it! */
|
||||
|
||||
/* prototypes */
|
||||
PROTOTYPEObj(nsd_ptcp);
|
||||
|
||||
/* the name of our library binary */
|
||||
#define LM_NSD_PTCP_FILENAME "lmnsd_ptcp"
|
||||
|
||||
#endif /* #ifndef INCLUDED_NSD_PTCP_H */
|
||||
@ -81,13 +81,13 @@ struct objInfo_s {
|
||||
};
|
||||
|
||||
|
||||
typedef struct obj { /* the dummy struct that each derived class can be casted to */
|
||||
struct obj_s { /* the dummy struct that each derived class can be casted to */
|
||||
objInfo_t *pObjInfo;
|
||||
#ifndef NDEBUG /* this means if debug... */
|
||||
unsigned int iObjCooCKiE; /* must always be 0xBADEFEE for a valid object */
|
||||
#endif
|
||||
uchar *pszName; /* the name of *this* specific object instance */
|
||||
} obj_t;
|
||||
};
|
||||
|
||||
|
||||
/* macros which must be gloablly-visible (because they are used during definition of
|
||||
|
||||
@ -61,8 +61,12 @@
|
||||
/* define some base data types */
|
||||
typedef unsigned char uchar;/* get rid of the unhandy "unsigned char" */
|
||||
typedef struct thrdInfo thrdInfo_t;
|
||||
typedef struct filed selector_t; /* TODO: this so far resides in syslogd.c, think about modularization */
|
||||
typedef struct obj_s obj_t;
|
||||
typedef struct filed selector_t;/* TODO: this so far resides in syslogd.c, think about modularization */
|
||||
typedef struct NetAddr netAddr_t;
|
||||
typedef struct netstrm_s netstrm_t;
|
||||
typedef struct nsd_ptcp_s nsd_ptcp_t;
|
||||
typedef obj_t nsd_t;
|
||||
typedef struct msg msg_t;
|
||||
typedef struct interface_s interface_t;
|
||||
typedef struct objInfo_s objInfo_t;
|
||||
@ -197,6 +201,11 @@ enum rsRetVal_ /** return value. All methods return this if not specified oth
|
||||
RS_RET_MAIL_NO_TO = -2071, /**< recipient for mail destination is missing */
|
||||
RS_RET_MAIL_NO_FROM = -2072, /**< sender for mail destination is missing */
|
||||
RS_RET_INVALID_PRI = -2073, /**< PRI value is invalid */
|
||||
RS_RET_MALICIOUS_HNAME = -2074, /**< remote peer is trying malicious things with its hostname */
|
||||
RS_RET_ACCEPT_ERR = -2074, /**< error during accept() system call */
|
||||
RS_RET_INVALID_HNAME = -2075, /**< remote peer's hostname invalid or unobtainable */
|
||||
RS_RET_INVALID_PORT = -2076, /**< invalid port value */
|
||||
RS_RET_COULD_NOT_BIND = -2077, /**< could not bind socket, defunct */
|
||||
|
||||
/* RainerScript error messages (range 1000.. 1999) */
|
||||
RS_RET_SYSVAR_NOT_FOUND = 1001, /**< system variable could not be found (maybe misspelled) */
|
||||
|
||||
1
tcpclt.c
1
tcpclt.c
@ -51,7 +51,6 @@ MODULE_TYPE_LIB
|
||||
DEFobjStaticHelpers
|
||||
|
||||
/* Initialize TCP sockets (for sender)
|
||||
* This is done once per selector line, if not yet initialized.
|
||||
*/
|
||||
static int
|
||||
CreateSocket(struct addrinfo *addrDest)
|
||||
|
||||
1
tcpclt.h
1
tcpclt.h
@ -26,7 +26,6 @@
|
||||
#ifndef TCPCLT_H_INCLUDED
|
||||
#define TCPCLT_H_INCLUDED 1
|
||||
|
||||
//#include "tcpsyslog.h"
|
||||
#include "obj.h"
|
||||
|
||||
/* the tcpclt object */
|
||||
|
||||
1
tcpsrv.c
1
tcpsrv.c
@ -457,7 +457,6 @@ SessAccept(tcpsrv_t *pThis, tcps_sess_t **ppSess, int fd)
|
||||
*/
|
||||
close (newConn);
|
||||
ABORT_FINALIZE(RS_RET_ERR); // TODO: better error code
|
||||
//was: return -1;
|
||||
}
|
||||
|
||||
/* Here we check if a host is permitted to send us
|
||||
|
||||
215
tools/omfwd.c
215
tools/omfwd.c
@ -30,7 +30,6 @@
|
||||
* A copy of the GPL can be found in the file "COPYING" in this distribution.
|
||||
*/
|
||||
#include "config.h"
|
||||
#ifdef SYSLOG_INET
|
||||
#include "rsyslog.h"
|
||||
#include <stdio.h>
|
||||
#include <stdarg.h>
|
||||
@ -52,6 +51,7 @@
|
||||
#include "syslogd-types.h"
|
||||
#include "srUtils.h"
|
||||
#include "net.h"
|
||||
#include "netstrm.h"
|
||||
#include "omfwd.h"
|
||||
#include "template.h"
|
||||
#include "msg.h"
|
||||
@ -69,17 +69,14 @@ DEF_OMOD_STATIC_DATA
|
||||
DEFobjCurrIf(errmsg)
|
||||
DEFobjCurrIf(glbl)
|
||||
DEFobjCurrIf(net)
|
||||
DEFobjCurrIf(netstrm)
|
||||
DEFobjCurrIf(tcpclt)
|
||||
|
||||
typedef struct _instanceData {
|
||||
char *f_hname;
|
||||
short sock; /* file descriptor */
|
||||
netstrm_t *pNetstrm; /* our output netstream */
|
||||
int *pSockArray; /* sockets to use for UDP */
|
||||
enum { /* TODO: we shoud revisit these definitions */
|
||||
eDestFORW,
|
||||
eDestFORW_SUSP,
|
||||
eDestFORW_UNKN
|
||||
} eDestState;
|
||||
int bIsConnected; /* are we connected to remote host? 0 - no, 1 - yes, UDP means addr resolved */
|
||||
struct addrinfo *f_addr;
|
||||
int compressionLevel; /* 0 - no compression, else level for zlib */
|
||||
char *port;
|
||||
@ -87,7 +84,6 @@ typedef struct _instanceData {
|
||||
# define FORW_UDP 0
|
||||
# define FORW_TCP 1
|
||||
/* following fields for TCP-based delivery */
|
||||
time_t ttSuspend; /* time selector was suspended */
|
||||
tcpclt_t *pTCPClt; /* our tcpclt object */
|
||||
} instanceData;
|
||||
|
||||
@ -101,7 +97,7 @@ static uchar *pszTplName = NULL; /* name of the default template to use */
|
||||
* We may change the implementation to try to lookup the port
|
||||
* if it is unspecified. So far, we use the IANA default auf 514.
|
||||
*/
|
||||
static char *getFwdSyslogPt(instanceData *pData)
|
||||
static char *getFwdPt(instanceData *pData)
|
||||
{
|
||||
assert(pData != NULL);
|
||||
if(pData->port == NULL)
|
||||
@ -112,7 +108,6 @@ static char *getFwdSyslogPt(instanceData *pData)
|
||||
|
||||
BEGINcreateInstance
|
||||
CODESTARTcreateInstance
|
||||
pData->sock = -1;
|
||||
ENDcreateInstance
|
||||
|
||||
|
||||
@ -125,20 +120,16 @@ ENDisCompatibleWithFeature
|
||||
|
||||
BEGINfreeInstance
|
||||
CODESTARTfreeInstance
|
||||
switch (pData->eDestState) {
|
||||
case eDestFORW:
|
||||
case eDestFORW_SUSP:
|
||||
if(pData->f_addr != NULL) { /* TODO: is the check ok? */
|
||||
freeaddrinfo(pData->f_addr);
|
||||
/* fall through */
|
||||
case eDestFORW_UNKN:
|
||||
pData->f_addr = NULL;
|
||||
}
|
||||
if(pData->port != NULL)
|
||||
free(pData->port);
|
||||
break;
|
||||
}
|
||||
|
||||
/* final cleanup */
|
||||
if(pData->sock >= 0)
|
||||
close(pData->sock);
|
||||
if(pData->pNetstrm != NULL)
|
||||
netstrm.Destruct(&pData->pNetstrm);
|
||||
if(pData->pSockArray != NULL)
|
||||
net.closeUDPListenSockets(pData->pSockArray);
|
||||
|
||||
@ -175,8 +166,7 @@ static rsRetVal UDPSend(instanceData *pData, char *msg, size_t len)
|
||||
* succeeding. We track this be bSendSuccess. We can not simply
|
||||
* rely on lsent, as a call might initially work, but a later
|
||||
* call fails. Then, lsent has the error status, even though
|
||||
* the sendto() succeeded.
|
||||
* rgerhards, 2007-06-22
|
||||
* the sendto() succeeded. -- rgerhards, 2007-06-22
|
||||
*/
|
||||
bSendSuccess = FALSE;
|
||||
for (r = pData->f_addr; r; r = r->ai_next) {
|
||||
@ -205,6 +195,7 @@ static rsRetVal UDPSend(instanceData *pData, char *msg, size_t len)
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* CODE FOR SENDING TCP MESSAGES */
|
||||
|
||||
|
||||
@ -217,22 +208,11 @@ static rsRetVal TCPSendFrame(void *pvData, char *msg, size_t len)
|
||||
ssize_t lenSend;
|
||||
instanceData *pData = (instanceData *) pvData;
|
||||
|
||||
lenSend = send(pData->sock, msg, len, 0);
|
||||
lenSend = len;
|
||||
CHKiRet(netstrm.Send(pData->pNetstrm, (uchar*)msg, &lenSend));
|
||||
dbgprintf("TCP sent %ld bytes, requested %ld\n", (long) lenSend, (long) len);
|
||||
|
||||
if(lenSend == -1) {
|
||||
/* we have an error case - check what we can live with */
|
||||
switch(errno) {
|
||||
case EMSGSIZE:
|
||||
dbgprintf("message not (tcp)send, too large\n");
|
||||
/* This is not a real error, so it is not flagged as one */
|
||||
break;
|
||||
default:
|
||||
dbgprintf("message not (tcp)send");
|
||||
iRet = RS_RET_TCP_SEND_ERROR;
|
||||
break;
|
||||
}
|
||||
} else if(lenSend != (ssize_t) len) {
|
||||
if(lenSend != (ssize_t) len) {
|
||||
/* no real error, could "just" not send everything...
|
||||
* For the time being, we ignore this...
|
||||
* rgerhards, 2005-10-25
|
||||
@ -242,6 +222,7 @@ static rsRetVal TCPSendFrame(void *pvData, char *msg, size_t len)
|
||||
/* TODO: we need to revisit this code -- rgerhards, 2007-12-28 */
|
||||
}
|
||||
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
@ -256,13 +237,12 @@ static rsRetVal TCPSendPrepRetry(void *pvData)
|
||||
instanceData *pData = (instanceData *) pvData;
|
||||
|
||||
assert(pData != NULL);
|
||||
close(pData->sock);
|
||||
pData->sock = -1;
|
||||
netstrm.Destruct(&pData->pNetstrm);
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
/* initialies everything so that TCPSend can work.
|
||||
/* initializes everything so that TCPSend can work.
|
||||
* rgerhards, 2007-12-28
|
||||
*/
|
||||
static rsRetVal TCPSendInit(void *pvData)
|
||||
@ -271,11 +251,18 @@ static rsRetVal TCPSendInit(void *pvData)
|
||||
instanceData *pData = (instanceData *) pvData;
|
||||
|
||||
assert(pData != NULL);
|
||||
if(pData->sock < 0) {
|
||||
if((pData->sock = tcpclt.CreateSocket(pData->f_addr)) < 0)
|
||||
iRet = RS_RET_TCP_SOCKCREATE_ERR;
|
||||
if(pData->pNetstrm == NULL) {
|
||||
CHKiRet(netstrm.Construct(&pData->pNetstrm));
|
||||
/* here we may set another netstream driver (e.g. to do TLS) */
|
||||
CHKiRet(netstrm.ConstructFinalize(pData->pNetstrm));
|
||||
CHKiRet(netstrm.Connect(pData->pNetstrm, glbl.GetDefPFFamily(),
|
||||
(uchar*)pData->port, (uchar*)pData->f_hname));
|
||||
}
|
||||
|
||||
finalize_it:
|
||||
if(iRet != RS_RET_OK)
|
||||
netstrm.Destruct(&pData->pNetstrm);
|
||||
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
@ -288,39 +275,38 @@ static rsRetVal doTryResume(instanceData *pData)
|
||||
DEFiRet;
|
||||
struct addrinfo *res;
|
||||
struct addrinfo hints;
|
||||
unsigned e;
|
||||
|
||||
switch (pData->eDestState) {
|
||||
case eDestFORW_SUSP:
|
||||
iRet = RS_RET_OK; /* the actual check happens during doAction() only */
|
||||
pData->eDestState = eDestFORW;
|
||||
break;
|
||||
if(pData->bIsConnected)
|
||||
FINALIZE;
|
||||
|
||||
case eDestFORW_UNKN:
|
||||
/* The remote address is not yet known and needs to be obtained */
|
||||
dbgprintf(" %s\n", pData->f_hname);
|
||||
if(pData->protocol == FORW_UDP) {
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
/* port must be numeric, because config file syntax requests this */
|
||||
/* TODO: this code is a duplicate from cfline() - we should later create
|
||||
* a common function.
|
||||
*/
|
||||
/* port must be numeric, because config file syntax requires this */
|
||||
hints.ai_flags = AI_NUMERICSERV;
|
||||
hints.ai_family = glbl.GetDefPFFamily();
|
||||
hints.ai_socktype = pData->protocol == FORW_UDP ? SOCK_DGRAM : SOCK_STREAM;
|
||||
if((e = getaddrinfo(pData->f_hname,
|
||||
getFwdSyslogPt(pData), &hints, &res)) == 0) {
|
||||
hints.ai_socktype = pData->protocol == SOCK_DGRAM;
|
||||
if((getaddrinfo(pData->f_hname, getFwdPt(pData), &hints, &res)) != 0) {
|
||||
ABORT_FINALIZE(RS_RET_SUSPENDED);
|
||||
}
|
||||
dbgprintf("%s found, resuming.\n", pData->f_hname);
|
||||
pData->f_addr = res;
|
||||
pData->eDestState = eDestFORW;
|
||||
} else {
|
||||
iRet = RS_RET_SUSPENDED;
|
||||
pData->bIsConnected = 1;
|
||||
if(pData->pSockArray == NULL) {
|
||||
pData->pSockArray = net.create_udp_socket((uchar*)pData->f_hname, NULL, 0);
|
||||
}
|
||||
break;
|
||||
case eDestFORW:
|
||||
/* rgerhards, 2007-09-11: this can not happen, but I've included it to
|
||||
* a) make the compiler happy, b) detect any logic errors */
|
||||
assert(0);
|
||||
break;
|
||||
} else {
|
||||
CHKiRet(TCPSendInit((void*)pData));
|
||||
}
|
||||
|
||||
finalize_it:
|
||||
if(iRet != RS_RET_OK) {
|
||||
if(pData->f_addr != NULL) {
|
||||
freeaddrinfo(pData->f_addr);
|
||||
pData->f_addr = NULL;
|
||||
}
|
||||
iRet = RS_RET_SUSPENDED;
|
||||
}
|
||||
|
||||
RETiRet;
|
||||
@ -336,30 +322,11 @@ BEGINdoAction
|
||||
char *psz; /* temporary buffering */
|
||||
register unsigned l;
|
||||
CODESTARTdoAction
|
||||
switch (pData->eDestState) {
|
||||
case eDestFORW_SUSP:
|
||||
dbgprintf("internal error in omfwd.c, eDestFORW_SUSP in doAction()!\n");
|
||||
iRet = RS_RET_SUSPENDED;
|
||||
break;
|
||||
CHKiRet(doTryResume(pData));
|
||||
|
||||
case eDestFORW_UNKN:
|
||||
dbgprintf("doAction eDestFORW_UNKN\n");
|
||||
iRet = doTryResume(pData);
|
||||
break;
|
||||
|
||||
case eDestFORW:
|
||||
dbgprintf(" %s:%s/%s\n", pData->f_hname, getFwdSyslogPt(pData),
|
||||
dbgprintf(" %s:%s/%s\n", pData->f_hname, getFwdPt(pData),
|
||||
pData->protocol == FORW_UDP ? "udp" : "tcp");
|
||||
/* with UDP, check if the socket is there and, if not, alloc
|
||||
* it. TODO: there should be a better place for that code.
|
||||
* rgerhards, 2007-12-26
|
||||
*/
|
||||
if(pData->protocol == FORW_UDP) {
|
||||
if(pData->pSockArray == NULL) {
|
||||
pData->pSockArray = net.create_udp_socket((uchar*)pData->f_hname, NULL, 0);
|
||||
}
|
||||
}
|
||||
pData->ttSuspend = time(NULL);
|
||||
|
||||
psz = (char*) ppString[0];
|
||||
l = strlen((char*) psz);
|
||||
if (l > MAXLINE)
|
||||
@ -412,30 +379,53 @@ CODESTARTdoAction
|
||||
if(ret != RS_RET_OK) {
|
||||
/* error! */
|
||||
dbgprintf("error forwarding via tcp, suspending\n");
|
||||
pData->eDestState = eDestFORW_SUSP;
|
||||
iRet = RS_RET_SUSPENDED;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
finalize_it:
|
||||
ENDdoAction
|
||||
|
||||
|
||||
/* This function loads TCP support, if not already loaded. It will be called
|
||||
* during config processing. To server ressources, TCP support will only
|
||||
* be loaded if it actually is used. -- rgerhard, 2008-04-17
|
||||
*/
|
||||
static rsRetVal
|
||||
loadTCPSupport(void)
|
||||
{
|
||||
DEFiRet;
|
||||
if(!netstrm.ifIsLoaded)
|
||||
CHKiRet(objUse(netstrm, LM_NETSTRM_FILENAME));
|
||||
if(!tcpclt.ifIsLoaded)
|
||||
CHKiRet(objUse(tcpclt, LM_TCPCLT_FILENAME));
|
||||
|
||||
finalize_it:
|
||||
RETiRet;
|
||||
}
|
||||
|
||||
|
||||
BEGINparseSelectorAct
|
||||
uchar *q;
|
||||
int i;
|
||||
int error;
|
||||
int bErr;
|
||||
struct addrinfo hints, *res;
|
||||
rsRetVal localRet;
|
||||
struct addrinfo;
|
||||
TCPFRAMINGMODE tcp_framing = TCP_FRAMING_OCTET_STUFFING;
|
||||
CODESTARTparseSelectorAct
|
||||
CODE_STD_STRING_REQUESTparseSelectorAct(1)
|
||||
if(*p == '@') {
|
||||
if((iRet = createInstance(&pData)) != RS_RET_OK)
|
||||
goto finalize_it;
|
||||
if(*p != '@')
|
||||
ABORT_FINALIZE(RS_RET_CONFLINE_UNPROCESSED);
|
||||
|
||||
CHKiRet(createInstance(&pData));
|
||||
|
||||
++p; /* eat '@' */
|
||||
if(*p == '@') { /* indicator for TCP! */
|
||||
localRet = loadTCPSupport();
|
||||
if(localRet != RS_RET_OK) {
|
||||
errmsg.LogError(NO_ERRCODE, "could not activate network stream modules for TCP "
|
||||
"(internal error %d) - are modules missing?", localRet);
|
||||
ABORT_FINALIZE(localRet);
|
||||
}
|
||||
pData->protocol = FORW_TCP;
|
||||
++p; /* eat this '@', too */
|
||||
} else {
|
||||
@ -518,9 +508,7 @@ CODE_STD_STRING_REQUESTparseSelectorAct(1)
|
||||
if(pData->port == NULL) {
|
||||
errmsg.LogError(NO_ERRCODE, "Could not get memory to store syslog forwarding port, "
|
||||
"using default port, results may not be what you intend\n");
|
||||
/* we leave f_forw.port set to NULL, this is then handled by
|
||||
* getFwdSyslogPt().
|
||||
*/
|
||||
/* we leave f_forw.port set to NULL, this is then handled by getFwdPt(). */
|
||||
} else {
|
||||
memcpy(pData->port, tmp, i);
|
||||
*(pData->port + i) = '\0';
|
||||
@ -554,25 +542,6 @@ CODE_STD_STRING_REQUESTparseSelectorAct(1)
|
||||
CHKiRet(cflineParseTemplateName(&p, *ppOMSR, 0, OMSR_NO_RQD_TPL_OPTS,
|
||||
(pszTplName == NULL) ? (uchar*)"RSYSLOG_TraditionalForwardFormat" : pszTplName));
|
||||
|
||||
/* first set the pData->eDestState */
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
/* port must be numeric, because config file syntax requests this */
|
||||
hints.ai_flags = AI_NUMERICSERV;
|
||||
hints.ai_family = glbl.GetDefPFFamily();
|
||||
hints.ai_socktype = pData->protocol == FORW_UDP ? SOCK_DGRAM : SOCK_STREAM;
|
||||
if( (error = getaddrinfo(pData->f_hname, getFwdSyslogPt(pData), &hints, &res)) != 0) {
|
||||
pData->eDestState = eDestFORW_UNKN;
|
||||
pData->ttSuspend = time(NULL);
|
||||
} else {
|
||||
pData->eDestState = eDestFORW;
|
||||
pData->f_addr = res;
|
||||
}
|
||||
/*
|
||||
* Otherwise the host might be unknown due to an
|
||||
* inaccessible nameserver (perhaps on the same
|
||||
* host). We try to get the ip number later, like
|
||||
* FORW_SUSP.
|
||||
*/
|
||||
if(pData->protocol == FORW_TCP) {
|
||||
/* create our tcpclt */
|
||||
CHKiRet(tcpclt.Construct(&pData->pTCPClt));
|
||||
@ -583,13 +552,6 @@ CODE_STD_STRING_REQUESTparseSelectorAct(1)
|
||||
CHKiRet(tcpclt.SetFraming(pData->pTCPClt, tcp_framing));
|
||||
}
|
||||
|
||||
} else {
|
||||
iRet = RS_RET_CONFLINE_UNPROCESSED;
|
||||
}
|
||||
|
||||
/* TODO: do we need to call freeInstance if we failed - this is a general question for
|
||||
* all output modules. I'll address it lates as the interface evolves. rgerhards, 2007-07-25
|
||||
*/
|
||||
CODE_STD_FINALIZERparseSelectorAct
|
||||
ENDparseSelectorAct
|
||||
|
||||
@ -600,6 +562,9 @@ CODESTARTmodExit
|
||||
objRelease(errmsg, CORE_COMPONENT);
|
||||
objRelease(glbl, CORE_COMPONENT);
|
||||
objRelease(net, LM_NET_FILENAME);
|
||||
if(netstrm.ifIsLoaded)
|
||||
objRelease(netstrm, LM_NETSTRM_FILENAME);
|
||||
if(!tcpclt.ifIsLoaded)
|
||||
objRelease(tcpclt, LM_TCPCLT_FILENAME);
|
||||
|
||||
if(pszTplName != NULL) {
|
||||
@ -635,13 +600,11 @@ CODESTARTmodInit
|
||||
CODEmodInit_QueryRegCFSLineHdlr
|
||||
CHKiRet(objUse(glbl, CORE_COMPONENT));
|
||||
CHKiRet(objUse(errmsg, CORE_COMPONENT));
|
||||
CHKiRet(objUse(net, LM_NET_FILENAME));
|
||||
CHKiRet(objUse(tcpclt, LM_TCPCLT_FILENAME));
|
||||
CHKiRet(objUse(net,LM_NET_FILENAME));
|
||||
|
||||
CHKiRet(regCfSysLineHdlr((uchar *)"actionforwarddefaulttemplate", 0, eCmdHdlrGetWord, NULL, &pszTplName, NULL));
|
||||
CHKiRet(omsdRegCFSLineHdlr((uchar *)"resetconfigvariables", 1, eCmdHdlrCustomHandler, resetConfigVariables, NULL, STD_LOADABLE_MODULE_ID));
|
||||
ENDmodInit
|
||||
|
||||
#endif /* #ifdef SYSLOG_INET */
|
||||
/* vim:set ai:
|
||||
*/
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user