abrt.git - Unnamed repository; edit this file 'description' to name the repository.

diff options

Diffstat (limited to 'lib/Plugins/Python.cpp')

0 files changed, 0 insertions, 0 deletions

/* ommongodb.c * Output module for mongodb. * Note: this module uses the libmongo-client library. The original 10gen * mongodb C interface is crap. Obtain the library here: * https://github.com/algernon/libmongo-client * * Copyright 2007-2012 Rainer Gerhards and Adiscon GmbH. * * This file is part of rsyslog. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * -or- * see COPYING.ASL20 in the source distribution * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include "config.h" #include <stdio.h> #include <string.h> #include <stdlib.h> #include <ctype.h> #include <errno.h> #include <assert.h> #include <signal.h> #include <time.h> #include <mongo.h> #include "rsyslog.h" #include "conf.h" #include "syslogd-types.h" #include "srUtils.h" #include "template.h" #include "module-template.h" #include "datetime.h" #include "errmsg.h" #include "cfsysline.h" MODULE_TYPE_OUTPUT MODULE_TYPE_NOKEEP MODULE_CNFNAME("ommongodb") /* internal structures */ DEF_OMOD_STATIC_DATA DEFobjCurrIf(errmsg) DEFobjCurrIf(datetime) typedef struct _instanceData { mongo_sync_connection *conn; uchar *server; int port; uchar *db; uchar *collection; uchar *uid; uchar *pwd; uchar *dbNcoll; uchar *tplName; } instanceData; /* tables for interfacing with the v6 config system */ /* action (instance) parameters */ static struct cnfparamdescr actpdescr[] = { { "server", eCmdHdlrGetWord, 0 }, { "serverport", eCmdHdlrInt, 0 }, { "db", eCmdHdlrGetWord, 0 }, { "collection", eCmdHdlrGetWord, 0 }, { "uid", eCmdHdlrGetWord, 0 }, { "pwd", eCmdHdlrGetWord, 0 }, { "template", eCmdHdlrGetWord, 1 } }; static struct cnfparamblk actpblk = { CNFPARAMBLK_VERSION, sizeof(actpdescr)/sizeof(struct cnfparamdescr), actpdescr }; BEGINcreateInstance CODESTARTcreateInstance ENDcreateInstance BEGINisCompatibleWithFeature CODESTARTisCompatibleWithFeature /* use this to specify if select features are supported by this * plugin. If not, the framework will handle that. Currently, only * RepeatedMsgReduction ("last message repeated n times") is optional. */ if(eFeat == sFEATURERepeatedMsgReduction) iRet = RS_RET_OK; ENDisCompatibleWithFeature static void closeMongoDB(instanceData *pData) { if(pData->conn != NULL) { mongo_sync_disconnect(pData->conn); pData->conn = NULL; } } BEGINfreeInstance CODESTARTfreeInstance closeMongoDB(pData); free(pData->server); free(pData->db); free(pData->collection); free(pData->uid); free(pData->pwd); free(pData->tplName); ENDfreeInstance BEGINdbgPrintInstInfo CODESTARTdbgPrintInstInfo /* nothing special here */ ENDdbgPrintInstInfo /* report error that occured during *last* operation */ static void reportMongoError(instanceData *pData) { char errStr[1024]; errmsg.LogError(0, RS_RET_ERR, "ommongodb: error: %s", rs_strerror_r(errno, errStr, sizeof(errStr))); #if 0 gchar *err; if(mongo_sync_cmd_get_last_error(pData->conn, (gchar*)pData->db, &err) == TRUE) { errmsg.LogError(0, RS_RET_ERR, "ommongodb: error: %s", err); } else { errmsg.LogError(0, RS_RET_ERR, "ommongodb: we had an error, but can " "not obtain specifics"); } #endif } /* The following function is responsible for initializing a * MySQL connection. * Initially added 2004-10-28 mmeckelein */ static rsRetVal initMongoDB(instanceData *pData, int bSilent) { char *server; DEFiRet; server = (pData->server == NULL) ? "127.0.0.1" : (char*) pData->server; DBGPRINTF("ommongodb: trying connect to '%s' at port %d\n", server, pData->port); pData->conn = mongo_sync_connect(server, pData->port, TRUE); if(pData->conn == NULL) { if(!bSilent) { reportMongoError(pData); dbgprintf("ommongodb: can not initialize MongoDB handle"); } ABORT_FINALIZE(RS_RET_SUSPENDED); } finalize_it: RETiRet; } /* map syslog severity to lumberjack level * TODO: consider moving this to msg.c - make some dirty "friend" references... * rgerhards, 2012-03-19 */ static inline char * getLumberjackLevel(short severity) { switch(severity) { case 0: return "FATAL"; case 1: case 2: case 3: return "ERROR"; case 4: return "WARN"; case 5: case 6: return "INFO"; case 7: return "DEBUG"; default:DBGPRINTF("ommongodb: invalid syslog severity %u\n", severity); return "INVLD"; } } /* small helper: get integer power of 10 */ static inline int i10pow(int exp) { int r = 1; while(exp > 0) { r *= 10; exp--; } return r; } /* write to mongodb in MSG passing mode, that is without a template. * In this mode, we use the standard document format, which is somewhat * aligned to cee (as described in project lumberjack). Note that this is * a moving target, so we may run out of sync (and stay so to retain * backward compatibility, which we consider pretty important). */ rsRetVal writeMongoDB_msg(msg_t *pMsg, instanceData *pData) { bson *doc = NULL; uchar *procid; short unsigned procid_free; size_t procid_len; uchar *tag; short unsigned tag_free; size_t tag_len; uchar *pid; short unsigned pid_free; size_t pid_len; uchar *sys; short unsigned sys_free; size_t sys_len; uchar *msg; short unsigned msg_free; size_t msg_len; int severity, facil; gint64 ts_gen, ts_rcv; /* timestamps: generated, received */ int secfrac; DEFiRet; /* see if we are ready to proceed */ if(pData->conn == NULL) { CHKiRet(initMongoDB(pData, 0)); } procid = MsgGetProp(pMsg, NULL, PROP_PROGRAMNAME, NULL, &procid_len, &procid_free); tag = MsgGetProp(pMsg, NULL, PROP_SYSLOGTAG, NULL, &tag_len, &tag_free); pid = MsgGetProp(pMsg, NULL, PROP_PROCID, NULL, &pid_len, &pid_free); sys = MsgGetProp(pMsg, NULL, PROP_HOSTNAME, NULL, &sys_len, &sys_free); msg = MsgGetProp(pMsg, NULL, PROP_MSG, NULL, &msg_len, &msg_free); // TODO: move to datetime? Refactor in any case! rgerhards, 2012-03-30 ts_gen = (gint64) datetime.syslogTime2time_t(&pMsg->tTIMESTAMP) * 1000; /* ms! */ dbgprintf("ommongodb: ts_gen is %lld\n", (long long) ts_gen); dbgprintf("ommongodb: secfrac is %d, precision %d\n", pMsg->tTIMESTAMP.secfrac, pMsg->tTIMESTAMP.secfracPrecision); if(pMsg->tTIMESTAMP.secfracPrecision > 3) { secfrac = pMsg->tTIMESTAMP.secfrac / i10pow(pMsg->tTIMESTAMP.secfracPrecision - 3); } else if(pMsg->tTIMESTAMP.secfracPrecision < 3) { secfrac = pMsg->tTIMESTAMP.secfrac * i10pow(3 - pMsg->tTIMESTAMP.secfracPrecision); } else { secfrac = pMsg->tTIMESTAMP.secfrac; } ts_gen += secfrac; ts_rcv = (gint64) datetime.syslogTime2time_t(&pMsg->tRcvdAt) * 1000; /* ms! */ if(pMsg->tRcvdAt.secfracPrecision > 3) { secfrac = pMsg->tRcvdAt.secfrac / i10pow(pMsg->tRcvdAt.secfracPrecision - 3); } else if(pMsg->tRcvdAt.secfracPrecision < 3) { secfrac = pMsg->tRcvdAt.secfrac * i10pow(3 - pMsg->tRcvdAt.secfracPrecision); } else { secfrac = pMsg->tRcvdAt.secfrac; } ts_rcv += secfrac; /* the following need to be int, but are short, so we need to xlat */ severity = pMsg->iSeverity; facil = pMsg->iFacility; doc = bson_build(BSON_TYPE_STRING, "sys", sys, sys_len, BSON_TYPE_UTC_DATETIME, "time", ts_gen, BSON_TYPE_UTC_DATETIME, "time_rcvd", ts_rcv, BSON_TYPE_STRING, "msg", msg, msg_len, BSON_TYPE_INT32, "syslog_fac", facil, BSON_TYPE_INT32, "syslog_sever", severity, BSON_TYPE_STRING, "syslog_tag", tag, tag_len, BSON_TYPE_STRING, "procid", procid, procid_len, BSON_TYPE_STRING, "pid", pid, pid_len, BSON_TYPE_STRING, "level", getLumberjackLevel(pMsg->iSeverity), -1, BSON_TYPE_NONE); if(procid_free) free(procid); if(tag_free) free(tag); if(pid_free) free(pid); if(sys_free) free(sys); if(msg_free) free(msg); if(doc == NULL) { reportMongoError(pData); dbgprintf("ommongodb: error creating BSON doc\n"); ABORT_FINALIZE(RS_RET_SUSPENDED); } bson_finish(doc); if(!mongo_sync_cmd_insert(pData->conn, (char*)pData->dbNcoll, doc, NULL)) { reportMongoError(pData); dbgprintf("ommongodb: insert error\n"); ABORT_FINALIZE(RS_RET_SUSPENDED); } finalize_it: if(doc != NULL) bson_free(doc); RETiRet; } BEGINtryResume CODESTARTtryResume if(pData->conn == NULL) { iRet = initMongoDB(pData, 1); } ENDtryResume BEGINdoAction CODESTARTdoAction if(pData->tplName == NULL) { iRet = writeMongoDB_msg((msg_t*)ppString[0], pData); } ENDdoAction static inline void setInstParamDefaults(instanceData *pData) { pData->server = NULL; pData->port = 27017; pData->db = NULL; pData->collection= NULL; pData->uid = NULL; pData->pwd = NULL; pData->tplName = NULL; } BEGINnewActInst struct cnfparamvals *pvals; int i; unsigned lendb, lencoll; CODESTARTnewActInst if((pvals = nvlstGetParams(lst, &actpblk, NULL)) == NULL) { ABORT_FINALIZE(RS_RET_MISSING_CNFPARAMS); }


context:
space:
mode: