static int rc_config(oconfig_item_t *ci) {
for (int i = 0; i < ci->children_num; i++) {
oconfig_item_t const *child = ci->children + i;
const char *key = child->key;
int status = 0;
if (strcasecmp("DataDir", key) == 0) {
status = cf_util_get_string(child, &datadir);
if (status == 0) {
int len = strlen(datadir);
while ((len > 0) && (datadir[len - 1] == '/')) {
len--;
datadir[len] = 0;
}
if (len <= 0)
sfree(datadir);
}
} else if (strcasecmp("DaemonAddress", key) == 0)
status = cf_util_get_string(child, &daemon_address);
else if (strcasecmp("CreateFiles", key) == 0)
status = cf_util_get_boolean(child, &config_create_files);
else if (strcasecmp("CreateFilesAsync", key) == 0)
status = cf_util_get_boolean(child, &rrdcreate_config.async);
else if (strcasecmp("CollectStatistics", key) == 0)
status = cf_util_get_boolean(child, &config_collect_stats);
else if (strcasecmp("StepSize", key) == 0) {
int tmp = -1;
status = rc_config_get_int_positive(child, &tmp);
if (status == 0)
rrdcreate_config.stepsize = (unsigned long)tmp;
} else if (strcasecmp("HeartBeat", key) == 0)
status = rc_config_get_int_positive(child, &rrdcreate_config.heartbeat);
else if (strcasecmp("RRARows", key) == 0)
status = rc_config_get_int_positive(child, &rrdcreate_config.rrarows);
else if (strcasecmp("RRATimespan", key) == 0) {
int tmp = -1;
status = rc_config_get_int_positive(child, &tmp);
if (status == 0)
status = rc_config_add_timespan(tmp);
} else if (strcasecmp("XFF", key) == 0)
status = rc_config_get_xff(child, &rrdcreate_config.xff);
else {
WARNING("rrdcached plugin: Ignoring invalid option %s.", key);
continue;
}
if (status != 0)
WARNING("rrdcached plugin: Handling the \"%s\" option failed.", key);
}
if (daemon_address != NULL) {
plugin_register_write("rrdcached", rc_write, /* user_data = */ NULL);
plugin_register_flush("rrdcached", rc_flush, /* user_data = */ NULL);
}
return 0;
} /* int rc_config */
static int lua_cb_register_write(lua_State *L) /* {{{ */
{
int nargs = lua_gettop(L);
if (nargs != 1)
return luaL_error(L, "Invalid number of arguments (%d != 1)", nargs);
luaL_checktype(L, 1, LUA_TFUNCTION);
char function_name[DATA_MAX_NAME_LEN] = "";
ssnprintf(function_name, sizeof(function_name), "lua/%s", lua_tostring(L, 1));
int callback_id = clua_store_callback(L, 1);
if (callback_id < 0)
return luaL_error(L, "%s", "Storing callback function failed");
lua_State *thread = lua_newthread(L);
if (thread == NULL)
return luaL_error(L, "%s", "lua_newthread failed");
clua_store_thread(L, -1);
lua_pop(L, 1);
clua_callback_data_t *cb = calloc(1, sizeof(*cb));
if (cb == NULL)
return luaL_error(L, "%s", "calloc failed");
cb->lua_state = thread;
cb->callback_id = callback_id;
cb->lua_function_name = strdup(function_name);
pthread_mutex_init(&cb->lock, NULL);
user_data_t ud = {
.data = cb
};
int status = plugin_register_write(/* name = */ function_name,
/* callback = */ clua_write,
/* user_data = */ &ud);
if (status != 0)
return luaL_error(L, "%s", "plugin_register_write failed");
return 0;
} /* }}} int lua_cb_register_write */
static int wr_config_node (oconfig_item_t *ci) /* {{{ */
{
wr_node_t *node;
int timeout;
int status;
int i;
node = calloc (1, sizeof (*node));
if (node == NULL)
return (ENOMEM);
node->host = NULL;
node->port = 0;
node->timeout.tv_sec = 0;
node->timeout.tv_usec = 1000;
node->conn = NULL;
node->prefix = NULL;
node->database = 0;
node->max_set_size = -1;
node->store_rates = 1;
pthread_mutex_init (&node->lock, /* attr = */ NULL);
status = cf_util_get_string_buffer (ci, node->name, sizeof (node->name));
if (status != 0)
{
sfree (node);
return (status);
}
for (i = 0; i < ci->children_num; i++)
{
oconfig_item_t *child = ci->children + i;
if (strcasecmp ("Host", child->key) == 0)
status = cf_util_get_string (child, &node->host);
else if (strcasecmp ("Port", child->key) == 0)
{
status = cf_util_get_port_number (child);
if (status > 0)
{
node->port = status;
status = 0;
}
}
else if (strcasecmp ("Timeout", child->key) == 0) {
status = cf_util_get_int (child, &timeout);
if (status == 0) node->timeout.tv_usec = timeout;
}
else if (strcasecmp ("Prefix", child->key) == 0) {
status = cf_util_get_string (child, &node->prefix);
}
else if (strcasecmp ("Database", child->key) == 0) {
status = cf_util_get_int (child, &node->database);
}
else if (strcasecmp ("MaxSetSize", child->key) == 0) {
status = cf_util_get_int (child, &node->max_set_size);
}
else if (strcasecmp ("StoreRates", child->key) == 0) {
status = cf_util_get_boolean (child, &node->store_rates);
}
else
WARNING ("write_redis plugin: Ignoring unknown config option \"%s\".",
child->key);
if (status != 0)
break;
} /* for (i = 0; i < ci->children_num; i++) */
if (status == 0)
{
char cb_name[DATA_MAX_NAME_LEN];
user_data_t ud;
ssnprintf (cb_name, sizeof (cb_name), "write_redis/%s", node->name);
ud.data = node;
ud.free_func = wr_config_free;
status = plugin_register_write (cb_name, wr_write, &ud);
}
if (status != 0)
wr_config_free (node);
return (status);
} /* }}} int wr_config_node */
static int camqp_config_connection (oconfig_item_t *ci, /* {{{ */
_Bool publish)
{
camqp_config_t *conf;
int status;
int i;
conf = malloc (sizeof (*conf));
if (conf == NULL)
{
ERROR ("amqp plugin: malloc failed.");
return (ENOMEM);
}
/* Initialize "conf" {{{ */
memset (conf, 0, sizeof (*conf));
conf->publish = publish;
conf->name = NULL;
conf->format = CAMQP_FORMAT_COMMAND;
conf->host = NULL;
conf->port = 5672;
conf->vhost = NULL;
conf->user = NULL;
conf->password = NULL;
conf->exchange = NULL;
conf->routing_key = NULL;
/* publish only */
conf->delivery_mode = CAMQP_DM_VOLATILE;
conf->store_rates = 0;
/* publish & graphite only */
conf->prefix = NULL;
conf->postfix = NULL;
conf->escape_char = '_';
/* subscribe only */
conf->exchange_type = NULL;
conf->queue = NULL;
/* general */
conf->connection = NULL;
pthread_mutex_init (&conf->lock, /* attr = */ NULL);
/* }}} */
status = cf_util_get_string (ci, &conf->name);
if (status != 0)
{
sfree (conf);
return (status);
}
for (i = 0; i < ci->children_num; i++)
{
oconfig_item_t *child = ci->children + i;
if (strcasecmp ("Host", child->key) == 0)
status = cf_util_get_string (child, &conf->host);
else if (strcasecmp ("Port", child->key) == 0)
{
status = cf_util_get_port_number (child);
if (status > 0)
{
conf->port = status;
status = 0;
}
}
else if (strcasecmp ("VHost", child->key) == 0)
status = cf_util_get_string (child, &conf->vhost);
else if (strcasecmp ("User", child->key) == 0)
status = cf_util_get_string (child, &conf->user);
else if (strcasecmp ("Password", child->key) == 0)
status = cf_util_get_string (child, &conf->password);
else if (strcasecmp ("Exchange", child->key) == 0)
status = cf_util_get_string (child, &conf->exchange);
else if ((strcasecmp ("ExchangeType", child->key) == 0) && !publish)
status = cf_util_get_string (child, &conf->exchange_type);
else if ((strcasecmp ("Queue", child->key) == 0) && !publish)
status = cf_util_get_string (child, &conf->queue);
else if (strcasecmp ("RoutingKey", child->key) == 0)
status = cf_util_get_string (child, &conf->routing_key);
else if ((strcasecmp ("Persistent", child->key) == 0) && publish)
{
_Bool tmp = 0;
status = cf_util_get_boolean (child, &tmp);
if (tmp)
conf->delivery_mode = CAMQP_DM_PERSISTENT;
else
conf->delivery_mode = CAMQP_DM_VOLATILE;
}
else if ((strcasecmp ("StoreRates", child->key) == 0) && publish)
status = cf_util_get_boolean (child, &conf->store_rates);
else if ((strcasecmp ("Format", child->key) == 0) && publish)
status = camqp_config_set_format (child, conf);
else if ((strcasecmp ("GraphitePrefix", child->key) == 0) && publish)
status = cf_util_get_string (child, &conf->prefix);
else if ((strcasecmp ("GraphitePostfix", child->key) == 0) && publish)
status = cf_util_get_string (child, &conf->postfix);
else if ((strcasecmp ("GraphiteEscapeChar", child->key) == 0) && publish)
{
char *tmp_buff = NULL;
status = cf_util_get_string (child, &tmp_buff);
if (strlen (tmp_buff) > 1)
WARNING ("amqp plugin: The option \"GraphiteEscapeChar\" handles "
"only one character. Others will be ignored.");
conf->escape_char = tmp_buff[0];
sfree (tmp_buff);
}
else
WARNING ("amqp plugin: Ignoring unknown "
"configuration option \"%s\".", child->key);
if (status != 0)
break;
} /* for (i = 0; i < ci->children_num; i++) */
if ((status == 0) && (conf->exchange == NULL))
{
if (conf->exchange_type != NULL)
WARNING ("amqp plugin: The option \"ExchangeType\" was given "
"without the \"Exchange\" option. It will be ignored.");
if (!publish && (conf->routing_key != NULL))
WARNING ("amqp plugin: The option \"RoutingKey\" was given "
"without the \"Exchange\" option. It will be ignored.");
}
if (status != 0)
{
camqp_config_free (conf);
return (status);
}
if (conf->exchange != NULL)
{
DEBUG ("amqp plugin: camqp_config_connection: exchange = %s;",
conf->exchange);
}
if (publish)
{
char cbname[128];
user_data_t ud = { conf, camqp_config_free };
ssnprintf (cbname, sizeof (cbname), "amqp/%s", conf->name);
status = plugin_register_write (cbname, camqp_write, &ud);
if (status != 0)
{
camqp_config_free (conf);
return (status);
}
}
else
{
status = camqp_subscribe_init (conf);
if (status != 0)
{
camqp_config_free (conf);
return (status);
}
}
return (0);
} /* }}} int camqp_config_connection */
static void kafka_config_topic(rd_kafka_conf_t *conf, oconfig_item_t *ci) /* {{{ */
{
int status;
int i;
struct kafka_topic_context *tctx;
char *key;
char *val;
char callback_name[DATA_MAX_NAME_LEN];
char errbuf[1024];
user_data_t ud;
oconfig_item_t *child;
rd_kafka_conf_res_t ret;
if ((tctx = calloc(1, sizeof (*tctx))) == NULL) {
ERROR ("write_kafka plugin: calloc failed.");
return;
}
tctx->escape_char = '.';
tctx->store_rates = 1;
rd_kafka_conf_set_log_cb(conf, kafka_log);
if ((tctx->kafka = rd_kafka_new(RD_KAFKA_PRODUCER, conf,
errbuf, sizeof(errbuf))) == NULL) {
sfree(tctx);
ERROR("write_kafka plugin: cannot create kafka handle.");
return;
}
conf = NULL;
if ((tctx->conf = rd_kafka_topic_conf_new()) == NULL) {
rd_kafka_destroy(tctx->kafka);
sfree(tctx);
ERROR ("write_kafka plugin: cannot create topic configuration.");
return;
}
if (ci->values_num != 1) {
WARNING("kafka topic name needed.");
goto errout;
}
if (ci->values[0].type != OCONFIG_TYPE_STRING) {
WARNING("kafka topic needs a string argument.");
goto errout;
}
if ((tctx->topic_name = strdup(ci->values[0].value.string)) == NULL) {
ERROR("write_kafka plugin: cannot copy topic name.");
goto errout;
}
for (i = 0; i < ci->children_num; i++) {
/*
* The code here could be simplified but makes room
* for easy adding of new options later on.
*/
child = &ci->children[i];
status = 0;
if (strcasecmp ("Property", child->key) == 0) {
if (child->values_num != 2) {
WARNING("kafka properties need both a key and a value.");
goto errout;
}
if (child->values[0].type != OCONFIG_TYPE_STRING ||
child->values[1].type != OCONFIG_TYPE_STRING) {
WARNING("kafka properties needs string arguments.");
goto errout;
}
key = child->values[0].value.string;
val = child->values[0].value.string;
ret = rd_kafka_topic_conf_set(tctx->conf,key, val,
errbuf, sizeof(errbuf));
if (ret != RD_KAFKA_CONF_OK) {
WARNING("cannot set kafka topic property %s to %s: %s.",
key, val, errbuf);
goto errout;
}
} else if (strcasecmp ("Key", child->key) == 0) {
char *tmp_buf = NULL;
status = cf_util_get_string(child, &tmp_buf);
if (status != 0) {
WARNING("write_kafka plugin: invalid key supplied");
break;
}
if (strcasecmp(tmp_buf, "Random") != 0) {
tctx->has_key = 1;
tctx->key = crc32_buffer((u_char *)tmp_buf, strlen(tmp_buf));
}
sfree(tmp_buf);
} else if (strcasecmp ("Format", child->key) == 0) {
status = cf_util_get_string(child, &key);
if (status != 0)
goto errout;
assert(key != NULL);
if (strcasecmp(key, "Command") == 0) {
tctx->format = KAFKA_FORMAT_COMMAND;
} else if (strcasecmp(key, "Graphite") == 0) {
tctx->format = KAFKA_FORMAT_GRAPHITE;
} else if (strcasecmp(key, "Json") == 0) {
tctx->format = KAFKA_FORMAT_JSON;
} else {
WARNING ("write_kafka plugin: Invalid format string: %s",
key);
}
sfree(key);
} else if (strcasecmp ("StoreRates", child->key) == 0) {
status = cf_util_get_boolean (child, &tctx->store_rates);
(void) cf_util_get_flag (child, &tctx->graphite_flags,
GRAPHITE_STORE_RATES);
} else if (strcasecmp ("GraphiteSeparateInstances", child->key) == 0) {
status = cf_util_get_flag (child, &tctx->graphite_flags,
GRAPHITE_SEPARATE_INSTANCES);
} else if (strcasecmp ("GraphiteAlwaysAppendDS", child->key) == 0) {
status = cf_util_get_flag (child, &tctx->graphite_flags,
GRAPHITE_ALWAYS_APPEND_DS);
} else if (strcasecmp ("GraphitePrefix", child->key) == 0) {
status = cf_util_get_string (child, &tctx->prefix);
} else if (strcasecmp ("GraphitePostfix", child->key) == 0) {
status = cf_util_get_string (child, &tctx->postfix);
} else if (strcasecmp ("GraphiteEscapeChar", child->key) == 0) {
char *tmp_buff = NULL;
status = cf_util_get_string (child, &tmp_buff);
if (strlen (tmp_buff) > 1)
WARNING ("write_kafka plugin: The option \"GraphiteEscapeChar\" handles "
"only one character. Others will be ignored.");
tctx->escape_char = tmp_buff[0];
sfree (tmp_buff);
} else {
WARNING ("write_kafka plugin: Invalid directive: %s.", child->key);
}
if (status != 0)
break;
}
rd_kafka_topic_conf_set_partitioner_cb(tctx->conf, kafka_partition);
rd_kafka_topic_conf_set_opaque(tctx->conf, tctx);
if ((tctx->topic = rd_kafka_topic_new(tctx->kafka, tctx->topic_name,
tctx->conf)) == NULL) {
ERROR("write_kafka plugin: cannot create topic.");
goto errout;
}
tctx->conf = NULL;
ssnprintf(callback_name, sizeof(callback_name),
"write_kafka/%s", tctx->topic_name);
ud.data = tctx;
ud.free_func = kafka_topic_context_free;
status = plugin_register_write (callback_name, kafka_write, &ud);
if (status != 0) {
WARNING ("write_kafka plugin: plugin_register_write (\"%s\") "
"failed with status %i.",
callback_name, status);
goto errout;
}
return;
errout:
if (conf != NULL)
rd_kafka_conf_destroy(conf);
if (tctx->kafka != NULL)
rd_kafka_destroy(tctx->kafka);
if (tctx->topic != NULL)
rd_kafka_topic_destroy(tctx->topic);
if (tctx->topic_name != NULL)
free(tctx->topic_name);
if (tctx->conf != NULL)
rd_kafka_topic_conf_destroy(tctx->conf);
sfree(tctx);
} /* }}} int kafka_config_topic */
static int
riemann_config_node(oconfig_item_t *ci)
{
struct riemann_host *host = NULL;
int status = 0;
int i;
oconfig_item_t *child;
char callback_name[DATA_MAX_NAME_LEN];
user_data_t ud;
if ((host = calloc(1, sizeof (*host))) == NULL) {
ERROR ("write_riemann plugin: calloc failed.");
return ENOMEM;
}
pthread_mutex_init (&host->lock, NULL);
host->reference_count = 1;
host->node = NULL;
host->service = NULL;
host->store_rates = 1;
host->always_append_ds = 0;
host->use_tcp = 0;
status = cf_util_get_string (ci, &host->name);
if (status != 0) {
WARNING("write_riemann plugin: Required host name is missing.");
riemann_free (host);
return -1;
}
for (i = 0; i < ci->children_num; i++) {
/*
* The code here could be simplified but makes room
* for easy adding of new options later on.
*/
child = &ci->children[i];
status = 0;
if (strcasecmp ("Host", child->key) == 0) {
status = cf_util_get_string (child, &host->node);
if (status != 0)
break;
} else if (strcasecmp ("Port", child->key) == 0) {
status = cf_util_get_service (child, &host->service);
if (status != 0) {
ERROR ("write_riemann plugin: Invalid argument "
"configured for the \"Port\" "
"option.");
break;
}
} else if (strcasecmp ("Protocol", child->key) == 0) {
char tmp[16];
status = cf_util_get_string_buffer (child,
tmp, sizeof (tmp));
if (status != 0)
{
ERROR ("write_riemann plugin: cf_util_get_"
"string_buffer failed with "
"status %i.", status);
break;
}
if (strcasecmp ("UDP", tmp) == 0)
host->use_tcp = 0;
else if (strcasecmp ("TCP", tmp) == 0)
host->use_tcp = 1;
else
WARNING ("write_riemann plugin: The value "
"\"%s\" is not valid for the "
"\"Protocol\" option. Use "
"either \"UDP\" or \"TCP\".",
tmp);
} else if (strcasecmp ("StoreRates", child->key) == 0) {
status = cf_util_get_boolean (child, &host->store_rates);
if (status != 0)
break;
} else if (strcasecmp ("AlwaysAppendDS", child->key) == 0) {
status = cf_util_get_boolean (child,
&host->always_append_ds);
if (status != 0)
break;
} else {
WARNING("write_riemann plugin: ignoring unknown config "
"option: \"%s\"", child->key);
}
}
if (status != 0) {
riemann_free (host);
return status;
}
ssnprintf (callback_name, sizeof (callback_name), "write_riemann/%s",
host->name);
ud.data = host;
ud.free_func = riemann_free;
pthread_mutex_lock (&host->lock);
status = plugin_register_write (callback_name, riemann_write, &ud);
if (status != 0)
WARNING ("write_riemann plugin: plugin_register_write (\"%s\") "
"failed with status %i.",
callback_name, status);
else /* success */
host->reference_count++;
status = plugin_register_notification (callback_name,
riemann_notification, &ud);
if (status != 0)
WARNING ("write_riemann plugin: plugin_register_notification (\"%s\") "
"failed with status %i.",
callback_name, status);
else /* success */
host->reference_count++;
if (host->reference_count <= 1)
{
/* Both callbacks failed => free memory.
* We need to unlock here, because riemann_free() will lock.
* This is not a race condition, because we're the only one
* holding a reference. */
pthread_mutex_unlock (&host->lock);
riemann_free (host);
return (-1);
}
host->reference_count--;
pthread_mutex_unlock (&host->lock);
return status;
}
static int wm_config_node (oconfig_item_t *ci) /* {{{ */
{
wm_node_t *node;
int status;
node = calloc (1, sizeof (*node));
if (node == NULL)
return (ENOMEM);
mongo_init (node->conn);
node->host = NULL;
node->store_rates = 1;
pthread_mutex_init (&node->lock, /* attr = */ NULL);
status = cf_util_get_string_buffer (ci, node->name, sizeof (node->name));
if (status != 0)
{
sfree (node);
return (status);
}
for (int i = 0; i < ci->children_num; i++)
{
oconfig_item_t *child = ci->children + i;
if (strcasecmp ("Host", child->key) == 0)
status = cf_util_get_string (child, &node->host);
else if (strcasecmp ("Port", child->key) == 0)
{
status = cf_util_get_port_number (child);
if (status > 0)
{
node->port = status;
status = 0;
}
}
else if (strcasecmp ("Timeout", child->key) == 0)
status = cf_util_get_int (child, &node->timeout);
else if (strcasecmp ("StoreRates", child->key) == 0)
status = cf_util_get_boolean (child, &node->store_rates);
else if (strcasecmp ("Database", child->key) == 0)
status = cf_util_get_string (child, &node->db);
else if (strcasecmp ("User", child->key) == 0)
status = cf_util_get_string (child, &node->user);
else if (strcasecmp ("Password", child->key) == 0)
status = cf_util_get_string (child, &node->passwd);
else
WARNING ("write_mongodb plugin: Ignoring unknown config option \"%s\".",
child->key);
if (status != 0)
break;
} /* for (i = 0; i < ci->children_num; i++) */
if ((node->db != NULL) || (node->user != NULL) || (node->passwd != NULL))
{
if ((node->db == NULL) || (node->user == NULL) || (node->passwd == NULL))
{
WARNING ("write_mongodb plugin: Authentication requires the "
"\"Database\", \"User\" and \"Password\" options to be specified, "
"but at last one of them is missing. Authentication will NOT be "
"used.");
sfree (node->db);
sfree (node->user);
sfree (node->passwd);
}
}
if (status == 0)
{
char cb_name[DATA_MAX_NAME_LEN];
user_data_t ud;
ssnprintf (cb_name, sizeof (cb_name), "write_mongodb/%s", node->name);
ud.data = node;
ud.free_func = wm_config_free;
status = plugin_register_write (cb_name, wm_write, &ud);
INFO ("write_mongodb plugin: registered write plugin %s %d",cb_name,status);
}
if (status != 0)
wm_config_free (node);
return (status);
} /* }}} int wm_config_node */
static int wm_config_node (oconfig_item_t *ci) /* {{{ */
{
wm_node_t *node;
int status;
int i;
node = malloc (sizeof (*node));
if (node == NULL)
return (ENOMEM);
memset (node, 0, sizeof (*node));
mongo_init (node->conn);
node->host = NULL;
node->store_rates = 1;
pthread_mutex_init (&node->lock, /* attr = */ NULL);
status = cf_util_get_string_buffer (ci, node->name, sizeof (node->name));
if (status != 0)
{
sfree (node);
return (status);
}
for (i = 0; i < ci->children_num; i++)
{
oconfig_item_t *child = ci->children + i;
if (strcasecmp ("Host", child->key) == 0)
status = cf_util_get_string (child, &node->host);
else if (strcasecmp ("Port", child->key) == 0)
{
status = cf_util_get_port_number (child);
if (status > 0)
{
node->port = status;
status = 0;
}
}
else if (strcasecmp ("Timeout", child->key) == 0)
status = cf_util_get_int (child, &node->timeout);
else if (strcasecmp ("StoreRates", child->key) == 0)
status = cf_util_get_boolean (child, &node->store_rates);
else
WARNING ("write_mongodb plugin: Ignoring unknown config option \"%s\".",
child->key);
if (status != 0)
break;
} /* for (i = 0; i < ci->children_num; i++) */
if (status == 0)
{
char cb_name[DATA_MAX_NAME_LEN];
user_data_t ud;
ssnprintf (cb_name, sizeof (cb_name), "write_mongodb/%s", node->name);
ud.data = node;
ud.free_func = wm_config_free;
status = plugin_register_write (cb_name, wm_write, &ud);
INFO ("write_mongodb plugin: registered write plugin %s %d",cb_name,status);
}
if (status != 0)
wm_config_free (node);
return (status);
} /* }}} int wm_config_node */
