static int o_shutdown (void) /* {{{ */
{
size_t i;
for (i = 0; i < databases_num; i++)
if (databases[i]->oci_service_context != NULL)
{
OCIHandleFree (databases[i]->oci_service_context, OCI_HTYPE_SVCCTX);
databases[i]->oci_service_context = NULL;
}
for (i = 0; i < queries_num; i++)
{
OCIStmt *oci_statement;
oci_statement = udb_query_get_user_data (queries[i]);
if (oci_statement != NULL)
{
OCIHandleFree (oci_statement, OCI_HTYPE_STMT);
udb_query_set_user_data (queries[i], NULL);
}
}
OCIHandleFree (oci_env, OCI_HTYPE_ENV);
oci_env = NULL;
udb_query_free (queries, queries_num);
queries = NULL;
queries_num = 0;
return (0);
} /* }}} int o_shutdown */
static int o_config (oconfig_item_t *ci) /* {{{ */
{
int i;
for (i = 0; i < ci->children_num; i++)
{
oconfig_item_t *child = ci->children + i;
if (strcasecmp ("Query", child->key) == 0)
udb_query_create (&queries, &queries_num, child,
/* callback = */ NULL);
else if (strcasecmp ("Database", child->key) == 0)
o_config_add_database (child);
else
{
WARNING ("oracle plugin: Ignoring unknown config option `%s'.", child->key);
}
if (queries_num > 0)
{
DEBUG ("oracle plugin: o_config: queries_num = %zu; queries[0] = %p; udb_query_get_user_data (queries[0]) = %p;",
queries_num, (void *) queries[0], udb_query_get_user_data (queries[0]));
}
} /* for (ci->children) */
return (0);
} /* }}} int o_config */
static int c_psql_config_database (oconfig_item_t *ci)
{
c_psql_database_t *db;
int i;
if ((1 != ci->values_num)
|| (OCONFIG_TYPE_STRING != ci->values[0].type)) {
log_err ("<Database> expects a single string argument.");
return 1;
}
db = c_psql_database_new (ci->values[0].value.string);
for (i = 0; i < ci->children_num; ++i) {
oconfig_item_t *c = ci->children + i;
if (0 == strcasecmp (c->key, "Host"))
config_set_s ("Host", &db->host, c);
else if (0 == strcasecmp (c->key, "Port"))
config_set_s ("Port", &db->port, c);
else if (0 == strcasecmp (c->key, "User"))
config_set_s ("User", &db->user, c);
else if (0 == strcasecmp (c->key, "Password"))
config_set_s ("Password", &db->password, c);
else if (0 == strcasecmp (c->key, "SSLMode"))
config_set_s ("SSLMode", &db->sslmode, c);
else if (0 == strcasecmp (c->key, "KRBSrvName"))
config_set_s ("KRBSrvName", &db->krbsrvname, c);
else if (0 == strcasecmp (c->key, "Service"))
config_set_s ("Service", &db->service, c);
else if (0 == strcasecmp (c->key, "Query"))
udb_query_pick_from_list (c, queries, queries_num,
&db->queries, &db->queries_num);
else
log_warn ("Ignoring unknown config key \"%s\".", c->key);
}
/* If no `Query' options were given, add the default queries.. */
if (db->queries_num == 0)
{
for (i = 0; i < def_queries_num; i++)
udb_query_pick_from_list_by_name (def_queries[i],
queries, queries_num,
&db->queries, &db->queries_num);
}
for (i = 0; (size_t)i < db->queries_num; ++i) {
c_psql_user_data_t *data;
data = udb_query_get_user_data (db->queries[i]);
if ((data != NULL) && (data->params_num > db->max_params_num))
db->max_params_num = data->params_num;
}
return 0;
} /* c_psql_config_database */
static int config_query_param_add (udb_query_t *q, oconfig_item_t *ci)
{
c_psql_user_data_t *data;
const char *param_str;
c_psql_param_t *tmp;
data = udb_query_get_user_data (q);
if (NULL == data) {
data = (c_psql_user_data_t *) smalloc (sizeof (*data));
if (NULL == data) {
log_err ("Out of memory.");
return -1;
}
memset (data, 0, sizeof (*data));
data->params = NULL;
}
tmp = (c_psql_param_t *) realloc (data->params,
(data->params_num + 1) * sizeof (c_psql_param_t));
if (NULL == tmp) {
log_err ("Out of memory.");
return -1;
}
data->params = tmp;
param_str = ci->values[0].value.string;
if (0 == strcasecmp (param_str, "hostname"))
data->params[data->params_num] = C_PSQL_PARAM_HOST;
else if (0 == strcasecmp (param_str, "database"))
data->params[data->params_num] = C_PSQL_PARAM_DB;
else if (0 == strcasecmp (param_str, "username"))
data->params[data->params_num] = C_PSQL_PARAM_USER;
else if (0 == strcasecmp (param_str, "interval"))
data->params[data->params_num] = C_PSQL_PARAM_INTERVAL;
else if (0 == strcasecmp (param_str, "instance"))
data->params[data->params_num] = C_PSQL_PARAM_INSTANCE;
else {
log_err ("Invalid parameter \"%s\".", param_str);
return 1;
}
data->params_num++;
udb_query_set_user_data (q, data);
return (0);
} /* config_query_param_add */
static int c_psql_config_database (oconfig_item_t *ci)
{
c_psql_database_t *db;
char cb_name[DATA_MAX_NAME_LEN];
struct timespec cb_interval = { 0, 0 };
user_data_t ud;
int i;
if ((1 != ci->values_num)
|| (OCONFIG_TYPE_STRING != ci->values[0].type)) {
log_err ("<Database> expects a single string argument.");
return 1;
}
memset (&ud, 0, sizeof (ud));
db = c_psql_database_new (ci->values[0].value.string);
if (db == NULL)
return -1;
for (i = 0; i < ci->children_num; ++i) {
oconfig_item_t *c = ci->children + i;
if (0 == strcasecmp (c->key, "Host"))
cf_util_get_string (c, &db->host);
else if (0 == strcasecmp (c->key, "Port"))
cf_util_get_service (c, &db->port);
else if (0 == strcasecmp (c->key, "User"))
cf_util_get_string (c, &db->user);
else if (0 == strcasecmp (c->key, "Password"))
cf_util_get_string (c, &db->password);
else if (0 == strcasecmp (c->key, "Instance"))
cf_util_get_string (c, &db->instance);
else if (0 == strcasecmp (c->key, "SSLMode"))
cf_util_get_string (c, &db->sslmode);
else if (0 == strcasecmp (c->key, "KRBSrvName"))
cf_util_get_string (c, &db->krbsrvname);
else if (0 == strcasecmp (c->key, "Service"))
cf_util_get_string (c, &db->service);
else if (0 == strcasecmp (c->key, "Query"))
udb_query_pick_from_list (c, queries, queries_num,
&db->queries, &db->queries_num);
else if (0 == strcasecmp (c->key, "Interval"))
cf_util_get_cdtime (c, &db->interval);
else
log_warn ("Ignoring unknown config key \"%s\".", c->key);
}
/* If no `Query' options were given, add the default queries.. */
if (db->queries_num == 0) {
for (i = 0; i < def_queries_num; i++)
udb_query_pick_from_list_by_name (def_queries[i],
queries, queries_num,
&db->queries, &db->queries_num);
}
if (db->queries_num > 0) {
db->q_prep_areas = (udb_query_preparation_area_t **) calloc (
db->queries_num, sizeof (*db->q_prep_areas));
if (db->q_prep_areas == NULL) {
log_err ("Out of memory.");
c_psql_database_delete (db);
return -1;
}
}
for (i = 0; (size_t)i < db->queries_num; ++i) {
c_psql_user_data_t *data;
data = udb_query_get_user_data (db->queries[i]);
if ((data != NULL) && (data->params_num > db->max_params_num))
db->max_params_num = data->params_num;
db->q_prep_areas[i]
= udb_query_allocate_preparation_area (db->queries[i]);
if (db->q_prep_areas[i] == NULL) {
log_err ("Out of memory.");
c_psql_database_delete (db);
return -1;
}
}
ud.data = db;
ud.free_func = c_psql_database_delete;
ssnprintf (cb_name, sizeof (cb_name), "postgresql-%s", db->instance);
CDTIME_T_TO_TIMESPEC (db->interval, &cb_interval);
plugin_register_complex_read ("postgresql", cb_name, c_psql_read,
/* interval = */ (db->interval > 0) ? &cb_interval : NULL,
&ud);
return 0;
} /* c_psql_config_database */
static int c_psql_exec_query (c_psql_database_t *db, udb_query_t *q,
udb_query_preparation_area_t *prep_area)
{
PGresult *res;
c_psql_user_data_t *data;
const char *host;
char **column_names;
char **column_values;
int column_num;
int rows_num;
int status;
int row, col;
/* The user data may hold parameter information, but may be NULL. */
data = udb_query_get_user_data (q);
/* Versions up to `3' don't know how to handle parameters. */
if (3 <= db->proto_version)
res = c_psql_exec_query_params (db, q, data);
else if ((NULL == data) || (0 == data->params_num))
res = c_psql_exec_query_noparams (db, q);
else {
log_err ("Connection to database \"%s\" (%s) does not support "
"parameters (protocol version %d) - "
"cannot execute query \"%s\".",
db->database, db->instance, db->proto_version,
udb_query_get_name (q));
return -1;
}
column_names = NULL;
column_values = NULL;
#define BAIL_OUT(status) \
sfree (column_names); \
sfree (column_values); \
PQclear (res); \
return status
if (PGRES_TUPLES_OK != PQresultStatus (res)) {
log_err ("Failed to execute SQL query: %s",
PQerrorMessage (db->conn));
log_info ("SQL query was: %s",
udb_query_get_statement (q));
BAIL_OUT (-1);
}
rows_num = PQntuples (res);
if (1 > rows_num) {
BAIL_OUT (0);
}
column_num = PQnfields (res);
column_names = (char **) calloc (column_num, sizeof (char *));
if (NULL == column_names) {
log_err ("calloc failed.");
BAIL_OUT (-1);
}
column_values = (char **) calloc (column_num, sizeof (char *));
if (NULL == column_values) {
log_err ("calloc failed.");
BAIL_OUT (-1);
}
for (col = 0; col < column_num; ++col) {
/* Pointers returned by `PQfname' are freed by `PQclear' via
* `BAIL_OUT'. */
column_names[col] = PQfname (res, col);
if (NULL == column_names[col]) {
log_err ("Failed to resolve name of column %i.", col);
BAIL_OUT (-1);
}
}
if (C_PSQL_IS_UNIX_DOMAIN_SOCKET (db->host)
|| (0 == strcmp (db->host, "localhost")))
host = hostname_g;
else
host = db->host;
status = udb_query_prepare_result (q, prep_area, host, "postgresql",
db->instance, column_names, (size_t) column_num, db->interval);
if (0 != status) {
log_err ("udb_query_prepare_result failed with status %i.",
status);
BAIL_OUT (-1);
}
for (row = 0; row < rows_num; ++row) {
for (col = 0; col < column_num; ++col) {
/* Pointers returned by `PQgetvalue' are freed by `PQclear' via
* `BAIL_OUT'. */
column_values[col] = PQgetvalue (res, row, col);
if (NULL == column_values[col]) {
log_err ("Failed to get value at (row = %i, col = %i).",
row, col);
break;
}
}
/* check for an error */
if (col < column_num)
continue;
status = udb_query_handle_result (q, prep_area, column_values);
if (status != 0) {
log_err ("udb_query_handle_result failed with status %i.",
status);
}
} /* for (row = 0; row < rows_num; ++row) */
udb_query_finish_result (q, prep_area);
BAIL_OUT (0);
#undef BAIL_OUT
} /* c_psql_exec_query */
static int o_read_database_query (o_database_t *db, /* {{{ */
udb_query_t *q, udb_query_preparation_area_t *prep_area)
{
char **column_names;
char **column_values;
size_t column_num;
OCIStmt *oci_statement;
/* List of `OCIDefine' pointers. These defines map columns to the buffer
* space declared above. */
OCIDefine **oci_defines;
int status;
size_t i;
oci_statement = udb_query_get_user_data (q);
/* Prepare the statement */
if (oci_statement == NULL) /* {{{ */
{
const char *statement;
statement = udb_query_get_statement (q);
assert (statement != NULL);
status = OCIHandleAlloc (oci_env, (void *) &oci_statement,
OCI_HTYPE_STMT, /* user_data_size = */ 0, /* user_data = */ NULL);
if (status != OCI_SUCCESS)
{
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIHandleAlloc", oci_error);
oci_statement = NULL;
return (-1);
}
status = OCIStmtPrepare (oci_statement, oci_error,
(text *) statement, (ub4) strlen (statement),
/* language = */ OCI_NTV_SYNTAX,
/* mode = */ OCI_DEFAULT);
if (status != OCI_SUCCESS)
{
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIStmtPrepare", oci_error);
OCIHandleFree (oci_statement, OCI_HTYPE_STMT);
oci_statement = NULL;
return (-1);
}
udb_query_set_user_data (q, oci_statement);
DEBUG ("oracle plugin: o_read_database_query (%s, %s): "
"Successfully allocated statement handle.",
db->name, udb_query_get_name (q));
} /* }}} */
assert (oci_statement != NULL);
/* Execute the statement */
status = OCIStmtExecute (db->oci_service_context, /* {{{ */
oci_statement,
oci_error,
/* iters = */ 0,
/* rowoff = */ 0,
/* snap_in = */ NULL, /* snap_out = */ NULL,
/* mode = */ OCI_DEFAULT);
if (status != OCI_SUCCESS)
{
o_report_error ("o_read_database_query", db->name, udb_query_get_name (q),
"OCIStmtExecute", oci_error);
return (-1);
} /* }}} */
/* Acquire the number of columns returned. */
do /* {{{ */
{
ub4 param_counter = 0;
status = OCIAttrGet (oci_statement, OCI_HTYPE_STMT, /* {{{ */
¶m_counter, /* size pointer = */ NULL,
OCI_ATTR_PARAM_COUNT, oci_error);
if (status != OCI_SUCCESS)
{
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIAttrGet", oci_error);
return (-1);
} /* }}} */
column_num = (size_t) param_counter;
} while (0); /* }}} */
/* Allocate the following buffers:
*
* +---------------+-----------------------------------+
* ! Name ! Size !
* +---------------+-----------------------------------+
* ! column_names ! column_num x DATA_MAX_NAME_LEN !
* ! column_values ! column_num x DATA_MAX_NAME_LEN !
* ! oci_defines ! column_num x sizeof (OCIDefine *) !
* +---------------+-----------------------------------+
*
* {{{ */
#define NUMBER_BUFFER_SIZE 64
#define FREE_ALL \
if (column_names != NULL) { \
sfree (column_names[0]); \
sfree (column_names); \
} \
if (column_values != NULL) { \
sfree (column_values[0]); \
sfree (column_values); \
} \
sfree (oci_defines)
#define ALLOC_OR_FAIL(ptr, ptr_size) \
do { \
size_t alloc_size = (size_t) ((ptr_size)); \
(ptr) = calloc (1, alloc_size); \
if ((ptr) == NULL) { \
FREE_ALL; \
ERROR ("oracle plugin: o_read_database_query: calloc failed."); \
return (-1); \
} \
} while (0)
/* Initialize everything to NULL so the above works. */
column_names = NULL;
column_values = NULL;
oci_defines = NULL;
ALLOC_OR_FAIL (column_names, column_num * sizeof (char *));
ALLOC_OR_FAIL (column_names[0], column_num * DATA_MAX_NAME_LEN
* sizeof (char));
for (i = 1; i < column_num; i++)
column_names[i] = column_names[i - 1] + DATA_MAX_NAME_LEN;
ALLOC_OR_FAIL (column_values, column_num * sizeof (char *));
ALLOC_OR_FAIL (column_values[0], column_num * DATA_MAX_NAME_LEN
* sizeof (char));
for (i = 1; i < column_num; i++)
column_values[i] = column_values[i - 1] + DATA_MAX_NAME_LEN;
ALLOC_OR_FAIL (oci_defines, column_num * sizeof (OCIDefine *));
/* }}} End of buffer allocations. */
/* ``Define'' the returned data, i. e. bind the columns to the buffers
* allocated above. */
for (i = 0; i < column_num; i++) /* {{{ */
{
char *column_name;
ub4 column_name_length;
OCIParam *oci_param;
oci_param = NULL;
status = OCIParamGet (oci_statement, OCI_HTYPE_STMT, oci_error,
(void *) &oci_param, (ub4) (i + 1));
if (status != OCI_SUCCESS)
{
/* This is probably alright */
DEBUG ("oracle plugin: o_read_database_query: status = %#x (= %i);",
status, status);
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIParamGet", oci_error);
status = OCI_SUCCESS;
break;
}
column_name = NULL;
column_name_length = 0;
status = OCIAttrGet (oci_param, OCI_DTYPE_PARAM,
&column_name, &column_name_length, OCI_ATTR_NAME, oci_error);
if (status != OCI_SUCCESS)
{
OCIDescriptorFree (oci_param, OCI_DTYPE_PARAM);
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIAttrGet (OCI_ATTR_NAME)", oci_error);
continue;
}
OCIDescriptorFree (oci_param, OCI_DTYPE_PARAM);
oci_param = NULL;
/* Copy the name to column_names. Warning: The ``string'' returned by OCI
* may not be null terminated! */
memset (column_names[i], 0, DATA_MAX_NAME_LEN);
if (column_name_length >= DATA_MAX_NAME_LEN)
column_name_length = DATA_MAX_NAME_LEN - 1;
memcpy (column_names[i], column_name, column_name_length);
column_names[i][column_name_length] = 0;
DEBUG ("oracle plugin: o_read_database_query: column_names[%zu] = %s; "
"column_name_length = %"PRIu32";",
i, column_names[i], (uint32_t) column_name_length);
status = OCIDefineByPos (oci_statement,
&oci_defines[i], oci_error, (ub4) (i + 1),
column_values[i], DATA_MAX_NAME_LEN, SQLT_STR,
NULL, NULL, NULL, OCI_DEFAULT);
if (status != OCI_SUCCESS)
{
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIDefineByPos", oci_error);
continue;
}
} /* for (j = 1; j <= param_counter; j++) */
/* }}} End of the ``define'' stuff. */
status = udb_query_prepare_result (q, prep_area,
(db->host != NULL) ? db->host : hostname_g,
/* plugin = */ "oracle", db->name, column_names, column_num,
/* interval = */ 0);
if (status != 0)
{
ERROR ("oracle plugin: o_read_database_query (%s, %s): "
"udb_query_prepare_result failed.",
db->name, udb_query_get_name (q));
FREE_ALL;
return (-1);
}
/* Fetch and handle all the rows that matched the query. */
while (42) /* {{{ */
{
status = OCIStmtFetch2 (oci_statement, oci_error,
/* nrows = */ 1, /* orientation = */ OCI_FETCH_NEXT,
/* fetch offset = */ 0, /* mode = */ OCI_DEFAULT);
if (status == OCI_NO_DATA)
{
status = OCI_SUCCESS;
break;
}
else if ((status != OCI_SUCCESS) && (status != OCI_SUCCESS_WITH_INFO))
{
o_report_error ("o_read_database_query", db->name,
udb_query_get_name (q), "OCIStmtFetch2", oci_error);
break;
}
status = udb_query_handle_result (q, prep_area, column_values);
if (status != 0)
{
WARNING ("oracle plugin: o_read_database_query (%s, %s): "
"udb_query_handle_result failed.",
db->name, udb_query_get_name (q));
}
} /* }}} while (42) */
/* DEBUG ("oracle plugin: o_read_database_query: This statement succeeded: %s", q->statement); */
FREE_ALL;
return (0);
#undef FREE_ALL
#undef ALLOC_OR_FAIL
} /* }}} int o_read_database_query */
