From 527bfcea5c9ca5c8414620a022f097d4e53af784 Mon Sep 17 00:00:00 2001 From: Rainer Gerhards Date: Tue, 1 Jun 2010 18:51:55 +0200 Subject: first implementation of strgen interface and a first built-in strgen module. Some tweaks and more default strgens are needed, but the code doesn't look too bad ;) --- tools/smtradfile.c | 154 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 154 insertions(+) create mode 100644 tools/smtradfile.c (limited to 'tools/smtradfile.c') diff --git a/tools/smtradfile.c b/tools/smtradfile.c new file mode 100644 index 00000000..dcbc1d47 --- /dev/null +++ b/tools/smtradfile.c @@ -0,0 +1,154 @@ +/* smtradfile.c + * This is a strgen module for the traditional file format. + * + * NOTE: read comments in module-template.h to understand how this file + * works! + * + * File begun on 2010-06-01 by RGerhards + * + * Copyright 2010 Rainer Gerhards and Adiscon GmbH. + * + * This file is part of rsyslog. + * + * Rsyslog is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * Rsyslog 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 Rsyslog. If not, see . + * + * A copy of the GPL can be found in the file "COPYING" in this distribution. + */ +#include "config.h" +#include "rsyslog.h" +#include +#include +#include +#include +#include "syslogd.h" +#include "conf.h" +#include "syslogd-types.h" +#include "template.h" +#include "msg.h" +#include "module-template.h" +#include "glbl.h" +#include "errmsg.h" +#include "parser.h" +#include "datetime.h" +#include "unicode-helper.h" + +MODULE_TYPE_STRGEN +STRGEN_NAME("RSYSLOG_TraditionalFileFormat") + +/* internal structures + */ +DEF_SMOD_STATIC_DATA +DEFobjCurrIf(errmsg) +DEFobjCurrIf(glbl) +DEFobjCurrIf(parser) +DEFobjCurrIf(datetime) + + +/* config data */ + + +#warning TODO: ExtendBuf via object interface! +extern rsRetVal ExtendBuf(uchar **pBuf, size_t *pLenBuf, size_t iMinSize); +extern char *getTimeReported(msg_t *pM, enum tplFormatTypes eFmt); + +BEGINstrgen + int iBuf; + uchar *pVal; + size_t iLenVal; +CODESTARTstrgen + dbgprintf("XXX: strgen module, *ppBuf = %p\n", *ppBuf); + iBuf = 0; + dbgprintf("TIMESTAMP\n"); + /* TIMESTAMP + ' ' */ + dbgprintf("getTimeReported\n"); + pVal = (uchar*) getTimeReported(pMsg, tplFmtDefault); + dbgprintf("obtain iLenVal ptr %p\n", pVal); + iLenVal = ustrlen(pVal); + dbgprintf("TIMESTAMP pVal='%p', iLenVal=%d\n", pVal, iLenVal); + if(iBuf + iLenVal + 1 >= *pLenBuf) /* we reserve one char for the final \0! */ + CHKiRet(ExtendBuf(ppBuf, pLenBuf, iBuf + iLenVal + 1)); + memcpy(*ppBuf + iBuf, pVal, iLenVal); + iBuf += iLenVal; + *(*ppBuf + iBuf++) = ' '; + + dbgprintf("HOSTNAME\n"); + /* HOSTNAME + ' ' */ + pVal = (uchar*) getHOSTNAME(pMsg); + iLenVal = getHOSTNAMELen(pMsg); + if(iBuf + iLenVal + 1 >= *pLenBuf) /* we reserve one char for the ' '! */ + CHKiRet(ExtendBuf(ppBuf, pLenBuf, iBuf + iLenVal + 1)); + memcpy(*ppBuf + iBuf, pVal, iLenVal); + iBuf += iLenVal; + *(*ppBuf + iBuf++) = ' '; + + dbgprintf("TAG\n"); + /* syslogtag */ + /* max size for TAG assumed 200 * TODO: check! */ + if(iBuf + 200 >= *pLenBuf) + CHKiRet(ExtendBuf(ppBuf, pLenBuf, iBuf + 200)); + getTAG(pMsg, &pVal, &iLenVal); + memcpy(*ppBuf + iBuf, pVal, iLenVal); + iBuf += iLenVal; + + dbgprintf("MSG\n"); + /* MSG, plus leading space if necessary */ + pVal = getMSG(pMsg); + iLenVal = getMSGLen(pMsg); + if(iBuf + iLenVal + 1 >= *pLenBuf) /* we reserve one char for the leading SP*/ + CHKiRet(ExtendBuf(ppBuf, pLenBuf, iBuf + iLenVal + 1)); + if(pVal[0] != ' ') + *(*ppBuf + iBuf++) = ' '; + memcpy(*ppBuf + iBuf, pVal, iLenVal); + iBuf += iLenVal; + + dbgprintf("Trailer\n"); + /* end sequence */ + iLenVal = 2; + if(iBuf + iLenVal >= *pLenBuf) /* we reserve one char for the final \0! */ + CHKiRet(ExtendBuf(ppBuf, pLenBuf, iBuf + iLenVal + 1)); + *(*ppBuf + iBuf++) = '\n'; + *(*ppBuf + iBuf) = '\0'; + +finalize_it: +ENDstrgen + + +BEGINmodExit +CODESTARTmodExit + /* release what we no longer need */ + objRelease(errmsg, CORE_COMPONENT); + objRelease(glbl, CORE_COMPONENT); + objRelease(parser, CORE_COMPONENT); + objRelease(datetime, CORE_COMPONENT); +ENDmodExit + + +BEGINqueryEtryPt +CODESTARTqueryEtryPt +CODEqueryEtryPt_STD_SMOD_QUERIES +ENDqueryEtryPt + + +BEGINmodInit(smtradfile) +CODESTARTmodInit + *ipIFVersProvided = CURR_MOD_IF_VERSION; /* we only support the current interface specification */ +CODEmodInit_QueryRegCFSLineHdlr + CHKiRet(objUse(glbl, CORE_COMPONENT)); + CHKiRet(objUse(errmsg, CORE_COMPONENT)); + CHKiRet(objUse(parser, CORE_COMPONENT)); + CHKiRet(objUse(datetime, CORE_COMPONENT)); + + dbgprintf("traditional file format strgen init called, compiled with version %s\n", VERSION); + dbgprintf("GetStrgenName addr %p\n", GetStrgenName); +ENDmodInit -- cgit