Newer
Older
/*
* Asterisk -- An open source telephony toolkit.
*
* Copyright (C) 1999-2005, Digium, Inc.
*
* Mark Spencer <markster@digium.com> - Asterisk Author
* Matthew Boehm <mboehm@cytelcom.com> - MySQL RealTime Driver Author
*
* See http://www.asterisk.org for more information about
* the Asterisk project. Please do not directly contact
* any of the maintainers of this project for assistance;
* the project provides a web site, mailing lists and IRC
* channels for your use.
*
* This program is free software, distributed under the terms of
* the GNU General Public License Version 2. See the LICENSE file
* at the top of the source tree.
*/
/*!
* \file
* \brief MySQL CDR backend
*/
/*** MODULEINFO
<depend>mysqlclient</depend>
<defaultenabled>no</defaultenabled>
<support_level>extended</support_level>
***/
#include "asterisk.h"
#include <sys/stat.h>
#include <mysql/mysql.h>
#include <mysql/errmsg.h>
#include "asterisk/channel.h"
#include "asterisk/logger.h"
#include "asterisk/config.h"
#include "asterisk/module.h"
#include "asterisk/lock.h"
#include "asterisk/options.h"
#include "asterisk/cli.h"
#include "asterisk/utils.h"
#include "asterisk/threadstorage.h"
#include "asterisk/strings.h"
#define RES_CONFIG_MYSQL_CONF "res_config_mysql.conf"
#define RES_CONFIG_MYSQL_CONF_OLD "res_mysql.conf"
#define READHANDLE 0
#define WRITEHANDLE 1
#define ESCAPE_STRING(buf, var) \
do { \
struct ast_str *semi = ast_str_thread_get(&scratch2_buf, strlen(var) * 3 + 1); \
const char *chunk = var; \
ast_str_reset(semi); \
for (; *chunk; chunk++) { \
if (strchr(";^", *chunk)) { \
ast_str_append(&semi, 0, "^%02hhX", *chunk); \
} else { \
ast_str_append(&semi, 0, "%c", *chunk); \
} \
if (ast_str_strlen(semi) * 2 + 1 > ast_str_size(buf)) { \
ast_str_make_space(&(buf), ast_str_strlen(semi) * 2 + 1); \
} \
mysql_real_escape_string(&dbh->handle, ast_str_buffer(buf), ast_str_buffer(semi), ast_str_strlen(semi)); \
} while (0)
AST_THREADSTORAGE(sql_buf);
AST_THREADSTORAGE(sql2_buf);
AST_THREADSTORAGE(find_buf);
AST_THREADSTORAGE(scratch_buf);
AST_THREADSTORAGE(scratch2_buf);
AST_THREADSTORAGE(modify_buf);
AST_THREADSTORAGE(modify2_buf);
AST_THREADSTORAGE(modify3_buf);
enum requirements { RQ_WARN, RQ_CREATECLOSE, RQ_CREATECHAR };
struct mysql_conn {
AST_RWLIST_ENTRY(mysql_conn) list;
ast_mutex_t lock;
MYSQL handle;
char host[50];
char name[50];
char user[50];
char pass[50];
char sock[50];
Erin Spiceland
committed
char charset[50];
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
int port;
int connected;
time_t connect_time;
enum requirements requirements;
char unique_name[0];
};
struct columns {
char *name;
char *type;
char *dflt;
char null;
int len;
AST_LIST_ENTRY(columns) list;
};
struct tables {
ast_mutex_t lock;
AST_LIST_HEAD_NOLOCK(mysql_columns, columns) columns;
AST_LIST_ENTRY(tables) list;
struct mysql_conn *database;
char name[0];
};
static AST_LIST_HEAD_STATIC(mysql_tables, tables);
static AST_RWLIST_HEAD_STATIC(databases, mysql_conn);
static int parse_config(int reload);
static int mysql_reconnect(struct mysql_conn *conn);
static char *handle_cli_realtime_mysql_status(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a);
static char *handle_cli_realtime_mysql_cache(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a);
static int load_mysql_config(struct ast_config *config, const char *category, struct mysql_conn *conn);
static int require_mysql(const char *database, const char *tablename, va_list ap);
static struct ast_cli_entry cli_realtime_mysql_status[] = {
AST_CLI_DEFINE(handle_cli_realtime_mysql_status, "Shows connection information for the MySQL RealTime driver"),
AST_CLI_DEFINE(handle_cli_realtime_mysql_cache, "Shows cached tables within the MySQL realtime driver"),
};
static struct mysql_conn *find_database(const char *database, int for_write)
{
char *whichdb;
const char *ptr;
struct mysql_conn *cur;
if ((ptr = strchr(database, '/'))) {
/* Multiple databases encoded within string */
if (for_write) {
whichdb = ast_strdupa(ptr + 1);
} else {
whichdb = ast_alloca(ptr - database + 1);
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
strncpy(whichdb, database, ptr - database);
whichdb[ptr - database] = '\0';
}
} else {
whichdb = ast_strdupa(database);
}
AST_RWLIST_RDLOCK(&databases);
AST_RWLIST_TRAVERSE(&databases, cur, list) {
if (!strcmp(cur->unique_name, whichdb)) {
ast_mutex_lock(&cur->lock);
break;
}
}
AST_RWLIST_UNLOCK(&databases);
return cur;
}
#define release_database(a) ast_mutex_unlock(&(a)->lock)
static void destroy_table(struct tables *table)
{
struct columns *column;
ast_mutex_lock(&table->lock);
while ((column = AST_LIST_REMOVE_HEAD(&table->columns, list))) {
ast_free(column);
}
ast_mutex_unlock(&table->lock);
ast_mutex_destroy(&table->lock);
ast_free(table);
}
static struct tables *find_table(const char *database, const char *tablename)
{
struct columns *column;
struct tables *table;
struct ast_str *sql = ast_str_thread_get(&find_buf, 30);
char *fname, *ftype, *flen, *fdflt, *fnull;
struct mysql_conn *dbh;
MYSQL_RES *result;
MYSQL_ROW row;
if (!(dbh = find_database(database, 1))) {
return NULL;
}
AST_LIST_LOCK(&mysql_tables);
AST_LIST_TRAVERSE(&mysql_tables, table, list) {
if (!strcasecmp(table->name, tablename)) {
ast_mutex_lock(&table->lock);
AST_LIST_UNLOCK(&mysql_tables);
release_database(dbh);
return table;
}
}
/* Not found, scan the table */
ast_str_set(&sql, 0, "DESC %s", tablename);
if (!mysql_reconnect(dbh)) {
release_database(dbh);
AST_LIST_UNLOCK(&mysql_tables);
return NULL;
}
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_ERROR, "Failed to query database '%s', table '%s' columns: %s\n", database, tablename, mysql_error(&dbh->handle));
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
release_database(dbh);
AST_LIST_UNLOCK(&mysql_tables);
return NULL;
}
if (!(table = ast_calloc(1, sizeof(*table) + strlen(tablename) + 1))) {
ast_log(LOG_ERROR, "Unable to allocate memory for new table structure\n");
release_database(dbh);
AST_LIST_UNLOCK(&mysql_tables);
return NULL;
}
strcpy(table->name, tablename); /* SAFE */
table->database = dbh;
ast_mutex_init(&table->lock);
AST_LIST_HEAD_INIT_NOLOCK(&table->columns);
if ((result = mysql_store_result(&dbh->handle))) {
while ((row = mysql_fetch_row(result))) {
fname = row[0];
ftype = row[1];
fnull = row[2];
fdflt = row[4];
ast_verb(4, "Found column '%s' of type '%s'\n", fname, ftype);
if (fdflt == NULL) {
fdflt = "";
}
if (!(column = ast_calloc(1, sizeof(*column) + strlen(fname) + strlen(ftype) + strlen(fdflt) + 3))) {
ast_log(LOG_ERROR, "Unable to allocate column element %s for %s\n", fname, tablename);
destroy_table(table);
release_database(dbh);
AST_LIST_UNLOCK(&mysql_tables);
return NULL;
}
if ((flen = strchr(ftype, '('))) {
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
} else {
/* Columns like dates, times, and timestamps don't have a length */
column->len = -1;
}
column->name = (char *)column + sizeof(*column);
column->type = (char *)column + sizeof(*column) + strlen(fname) + 1;
column->dflt = (char *)column + sizeof(*column) + strlen(fname) + 1 + strlen(ftype) + 1;
strcpy(column->name, fname);
strcpy(column->type, ftype);
strcpy(column->dflt, fdflt);
column->null = (strcmp(fnull, "YES") == 0 ? 1 : 0);
AST_LIST_INSERT_TAIL(&table->columns, column, list);
}
mysql_free_result(result);
}
AST_LIST_INSERT_TAIL(&mysql_tables, table, list);
ast_mutex_lock(&table->lock);
AST_LIST_UNLOCK(&mysql_tables);
release_database(dbh);
return table;
}
static void release_table(struct tables *table)
{
if (table) {
ast_mutex_unlock(&table->lock);
}
}
static struct columns *find_column(struct tables *table, const char *colname)
{
struct columns *column;
AST_LIST_TRAVERSE(&table->columns, column, list) {
if (strcmp(column->name, colname) == 0) {
break;
}
}
return column;
}
static char *decode_chunk(char *chunk)
{
char *orig = chunk;
for (; *chunk; chunk++) {
if (*chunk == '^' && strchr("0123456789ABCDEFabcdef", chunk[1]) && strchr("0123456789ABCDEFabcdef", chunk[2])) {
sscanf(chunk + 1, "%02hhX", chunk);
memmove(chunk + 1, chunk + 3, strlen(chunk + 3) + 1);
}
}
return orig;
}
#define IS_SQL_LIKE_CLAUSE(x) ((x) && ast_ends_with(x, " LIKE"))
/* MySQL requires us to escape the escape... yo dawg */
static char *ESCAPE_CLAUSE = " ESCAPE '\\\\'";
static struct ast_variable *realtime_mysql(const char *database, const char *table, const struct ast_variable *rt_fields)
{
struct mysql_conn *dbh;
MYSQL_RES *result;
MYSQL_ROW row;
MYSQL_FIELD *fields;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 16);
struct ast_str *buf = ast_str_thread_get(&scratch_buf, 16);
char *stringp;
char *chunk;
char *op;
const struct ast_variable *field = rt_fields;
struct ast_variable *var=NULL, *prev=NULL;
if (!(dbh = find_database(database, 0))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: %s (check res_mysql.conf)\n", database);
return NULL;
}
if (!table) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
release_database(dbh);
return NULL;
}
/* Get the first parameter and first value in our list of passed paramater/value pairs */
ast_log(LOG_WARNING, "MySQL RealTime: Realtime retrieval requires at least 1 parameter and 1 value to search on.\n");
release_database(dbh);
return NULL;
}
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_database(dbh);
return NULL;
}
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
if (!strchr(field->name, ' ')) {
op = " =";
} else {
op = "";
if (IS_SQL_LIKE_CLAUSE(field->name)) {
escape = ESCAPE_CLAUSE;
}
}
ESCAPE_STRING(buf, field->value);
ast_str_set(&sql, 0, "SELECT * FROM %s WHERE %s%s '%s'%s", table, field->name, op, ast_str_buffer(buf), escape);
while ((field = field->next)) {
escape = "";
if (!strchr(field->name, ' ')) {
op = " =";
} else {
op = "";
if (IS_SQL_LIKE_CLAUSE(field->name)) {
escape = ESCAPE_CLAUSE;
}
}
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, " AND %s%s '%s'%s", field->name, op, ast_str_buffer(buf), escape);
}
ast_debug(1, "MySQL RealTime: Retrieve SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to query database: %s\n", mysql_error(&dbh->handle));
release_database(dbh);
return NULL;
}
if ((result = mysql_store_result(&dbh->handle))) {
numFields = mysql_num_fields(result);
fields = mysql_fetch_fields(result);
while ((row = mysql_fetch_row(result))) {
for (i = 0; i < numFields; i++) {
Tilghman Lesher
committed
/* Encode NULL values separately from blank values, for the Realtime API */
if (row[i] == NULL) {
row[i] = "";
} else if (ast_strlen_zero(row[i])) {
row[i] = " ";
}
for (stringp = row[i], chunk = strsep(&stringp, ";"); chunk; chunk = strsep(&stringp, ";")) {
if (prev) {
if ((prev->next = ast_variable_new(fields[i].name, decode_chunk(chunk), ""))) {
prev = prev->next;
}
} else {
prev = var = ast_variable_new(fields[i].name, decode_chunk(chunk), "");
}
}
}
}
} else {
ast_debug(1, "MySQL RealTime: Could not find any rows in table %s.\n", table);
}
release_database(dbh);
mysql_free_result(result);
return var;
}
static struct ast_config *realtime_multi_mysql(const char *database, const char *table, const struct ast_variable *rt_fields)
{
struct mysql_conn *dbh;
MYSQL_RES *result;
MYSQL_ROW row;
MYSQL_FIELD *fields;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 16);
struct ast_str *buf = ast_str_thread_get(&scratch_buf, 16);
const char *initfield = NULL;
char *stringp;
char *chunk;
char *op;
const struct ast_variable *field = rt_fields;
struct ast_variable *var = NULL;
struct ast_config *cfg = NULL;
struct ast_category *cat = NULL;
if (!(dbh = find_database(database, 0))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: '%s' (check res_mysql.conf)\n", database);
return NULL;
}
if (!table) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
release_database(dbh);
return NULL;
}
if (!(cfg = ast_config_new())) {
/* If I can't alloc memory at this point, why bother doing anything else? */
ast_log(LOG_WARNING, "Out of memory!\n");
release_database(dbh);
return NULL;
}
/* Get the first parameter and first value in our list of passed paramater/value pairs */
ast_log(LOG_WARNING, "MySQL RealTime: Realtime retrieval requires at least 1 parameter and 1 value to search on.\n");
ast_config_destroy(cfg);
release_database(dbh);
return NULL;
}
initfield = ast_strdupa(field->name);
if ((op = strchr(initfield, ' '))) {
*op = '\0';
}
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_database(dbh);
ast_config_destroy(cfg);
return NULL;
}
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
if (!strchr(field->name, ' ')) {
op = " =";
op = "";
if (IS_SQL_LIKE_CLAUSE(field->name)) {
escape = ESCAPE_CLAUSE;
}
}
ESCAPE_STRING(buf, field->value);
ast_str_set(&sql, 0, "SELECT * FROM %s WHERE %s%s '%s'%s", table, field->name, op, ast_str_buffer(buf), escape);
while ((field = field->next)) {
escape = "";
if (!strchr(field->name, ' ')) {
op = " =";
} else {
op = "";
if (IS_SQL_LIKE_CLAUSE(field->name)) {
escape = ESCAPE_CLAUSE;
}
}
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, " AND %s%s '%s'%s", field->name, op, ast_str_buffer(buf), escape);
}
if (initfield) {
ast_str_append(&sql, 0, " ORDER BY %s", initfield);
}
ast_debug(1, "MySQL RealTime: Retrieve SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to query database: %s\n", mysql_error(&dbh->handle));
release_database(dbh);
ast_config_destroy(cfg);
return NULL;
}
if ((result = mysql_store_result(&dbh->handle))) {
numFields = mysql_num_fields(result);
fields = mysql_fetch_fields(result);
while ((row = mysql_fetch_row(result))) {
var = NULL;
cat = ast_category_new_anonymous();
if (!cat) {
continue;
}
for (i = 0; i < numFields; i++) {
if (ast_strlen_zero(row[i]))
continue;
for (stringp = row[i], chunk = strsep(&stringp, ";"); chunk; chunk = strsep(&stringp, ";")) {
if (chunk && !ast_strlen_zero(decode_chunk(ast_strip(chunk)))) {
if (initfield && !strcmp(initfield, fields[i].name)) {
ast_category_rename(cat, chunk);
}
var = ast_variable_new(fields[i].name, chunk, "");
ast_variable_append(cat, var);
}
}
}
ast_category_append(cfg, cat);
}
} else {
ast_debug(1, "MySQL RealTime: Could not find any rows in table %s.\n", table);
}
release_database(dbh);
mysql_free_result(result);
return cfg;
}
static int update_mysql(const char *database, const char *tablename, const char *keyfield, const char *lookup, const struct ast_variable *rt_fields)
{
struct mysql_conn *dbh;
uint64_t numrows;
const struct ast_variable *field = rt_fields;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 100), *buf = ast_str_thread_get(&scratch_buf, 100);
struct tables *table;
struct columns *column = NULL;
if (!(dbh = find_database(database, 1))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: '%s' (check res_mysql.conf)\n", database);
return -1;
}
if (!tablename) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
release_database(dbh);
return -1;
}
if (!(table = find_table(database, tablename))) {
ast_log(LOG_ERROR, "Table '%s' does not exist!!\n", tablename);
release_database(dbh);
return -1;
}
if (!(column = find_column(table, keyfield))) {
ast_log(LOG_ERROR, "MySQL RealTime: Updating on column '%s', but that column does not exist within the table '%s' (db '%s')!\n", keyfield, tablename, database);
release_table(table);
release_database(dbh);
return -1;
}
/* Get the first parameter and first value in our list of passed paramater/value pairs */
ast_log(LOG_WARNING, "MySQL RealTime: Realtime update requires at least 1 parameter and 1 value to update.\n");
release_table(table);
release_database(dbh);
return -1;
}
/* Check that the column exists in the table */
if (!(column = find_column(table, field->name))) {
ast_log(LOG_ERROR, "MySQL RealTime: Updating column '%s', but that column does not exist within the table '%s' (first pair MUST exist)!\n", field->name, tablename);
release_table(table);
release_database(dbh);
return -1;
}
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_table(table);
release_database(dbh);
return -1;
}
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
ESCAPE_STRING(buf, field->value);
ast_str_set(&sql, 0, "UPDATE %s SET `%s` = '%s'", tablename, field->name, ast_str_buffer(buf));
while ((field = field->next)) {
/* If the column is not within the table, then skip it */
if (!(column = find_column(table, field->name))) {
ast_log(LOG_WARNING, "Attempted to update column '%s' in table '%s', but column does not exist!\n", field->name, tablename);
continue;
}
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, ", `%s` = '%s'", field->name, ast_str_buffer(buf));
}
ESCAPE_STRING(buf, lookup);
ast_str_append(&sql, 0, " WHERE `%s` = '%s'", keyfield, ast_str_buffer(buf));
ast_debug(1, "MySQL RealTime: Update SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to update database: %s\n", mysql_error(&dbh->handle));
release_table(table);
release_database(dbh);
return -1;
}
numrows = mysql_affected_rows(&dbh->handle);
release_table(table);
release_database(dbh);
ast_debug(1, "MySQL RealTime: Updated %" PRIu64 " rows on table: %s\n", numrows, tablename);
/* From http://dev.mysql.com/doc/mysql/en/mysql-affected-rows.html
* An integer greater than zero indicates the number of rows affected
* Zero indicates that no records were updated
* -1 indicates that the query returned an error (although, if the query failed, it should have been caught above.)
*/
return (int)numrows;
}
static int update2_mysql(const char *database, const char *tablename, const struct ast_variable *lookup_fields, const struct ast_variable *update_fields)
{
struct mysql_conn *dbh;
uint64_t numrows;
const struct ast_variable *field;
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
struct ast_str *sql = ast_str_thread_get(&sql_buf, 100), *buf = ast_str_thread_get(&scratch_buf, 100);
struct ast_str *where = ast_str_thread_get(&sql2_buf, 100);
struct tables *table;
struct columns *column = NULL;
if (!tablename) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
return -1;
}
if (!(dbh = find_database(database, 1))) {
ast_log(LOG_ERROR, "Invalid database specified: %s\n", database);
return -1;
}
if (!(table = find_table(database, tablename))) {
ast_log(LOG_ERROR, "Table '%s' does not exist!!\n", tablename);
release_database(dbh);
return -1;
}
if (!sql || !buf || !where) {
release_database(dbh);
release_table(table);
return -1;
}
ast_str_set(&sql, 0, "UPDATE %s SET", tablename);
ast_str_set(&where, 0, "WHERE");
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_table(table);
release_database(dbh);
return -1;
}
for (field = lookup_fields; field; field = field->next) {
if (!(column = find_column(table, field->name))) {
ast_log(LOG_ERROR, "Updating on column '%s', but that column does not exist within the table '%s'!\n", field->name, tablename);
release_table(table);
release_database(dbh);
return -1;
}
ESCAPE_STRING(buf, field->value);
ast_str_append(&where, 0, "%s `%s` = '%s'", first ? "" : " AND", field->name, ast_str_buffer(buf));
first = 0;
}
first = 1;
for (field = update_fields; field; field = field->next) {
/* If the column is not within the table, then skip it */
if (!(column = find_column(table, field->name))) {
ast_log(LOG_WARNING, "Attempted to update column '%s' in table '%s', but column does not exist!\n", field->name, tablename);
continue;
}
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, "%s `%s` = '%s'", first ? "" : ",", field->name, ast_str_buffer(buf));
release_table(table);
ast_str_append(&sql, 0, " %s", ast_str_buffer(where));
ast_debug(1, "MySQL RealTime: Update SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to update database: %s\n", mysql_error(&dbh->handle));
release_table(table);
release_database(dbh);
return -1;
}
numrows = mysql_affected_rows(&dbh->handle);
release_database(dbh);
ast_debug(1, "MySQL RealTime: Updated %" PRIu64 " rows on table: %s\n", numrows, tablename);
/* From http://dev.mysql.com/doc/mysql/en/mysql-affected-rows.html
* An integer greater than zero indicates the number of rows affected
* Zero indicates that no records were updated
* -1 indicates that the query returned an error (although, if the query failed, it should have been caught above.)
*/
return (int)numrows;
}
static int store_mysql(const char *database, const char *table, const struct ast_variable *rt_fields)
{
struct mysql_conn *dbh;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 16);
struct ast_str *sql2 = ast_str_thread_get(&sql2_buf, 16);
struct ast_str *buf = ast_str_thread_get(&scratch_buf, 16);
const struct ast_variable *field = rt_fields;
if (!(dbh = find_database(database, 1))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: '%s' (check res_mysql.conf)\n", database);
return -1;
}
if (!table) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
release_database(dbh);
return -1;
}
/* Get the first parameter and first value in our list of passed paramater/value pairs */
ast_log(LOG_WARNING, "MySQL RealTime: Realtime storage requires at least 1 parameter and 1 value to search on.\n");
release_database(dbh);
return -1;
}
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_database(dbh);
return -1;
}
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
ESCAPE_STRING(buf, field->value);
ast_str_set(&sql, 0, "INSERT INTO %s (`%s`", table, field->name);
ast_str_set(&sql2, 0, ") VALUES ('%s'", ast_str_buffer(buf));
while ((field = field->next)) {
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, ", `%s`", field->name);
ast_str_append(&sql2, 0, ", '%s'", ast_str_buffer(buf));
}
ast_str_append(&sql, 0, "%s)", ast_str_buffer(sql2));
ast_debug(1,"MySQL RealTime: Insert SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to insert into database: %s\n", mysql_error(&dbh->handle));
release_database(dbh);
return -1;
}
release_database(dbh);
ast_debug(1, "MySQL RealTime: row inserted on table: %s\n", table);
return 1;
static int destroy_mysql(const char *database, const char *table, const char *keyfield, const char *lookup, const struct ast_variable *rt_fields)
{
struct mysql_conn *dbh;
uint64_t numrows;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 16);
struct ast_str *buf = ast_str_thread_get(&scratch_buf, 16);
const struct ast_variable *field;
if (!(dbh = find_database(database, 1))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: '%s' (check res_mysql.conf)\n", database);
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
return -1;
}
if (!table) {
ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
release_database(dbh);
return -1;
}
/* Get the first parameter and first value in our list of passed paramater/value pairs */
/* newparam = va_arg(ap, const char *);
newval = va_arg(ap, const char *);*/
if (ast_strlen_zero(keyfield) || ast_strlen_zero(lookup)) {
ast_log(LOG_WARNING, "MySQL RealTime: Realtime destroying requires at least 1 parameter and 1 value to search on.\n");
release_database(dbh);
return -1;
}
/* Must connect to the server before anything else, as the escape function requires the mysql handle. */
if (!mysql_reconnect(dbh)) {
release_database(dbh);
return -1;
}
/* Create the first part of the query using the first parameter/value pairs we just extracted
If there is only 1 set, then we have our query. Otherwise, loop thru the list and concat */
ESCAPE_STRING(buf, lookup);
ast_str_set(&sql, 0, "DELETE FROM %s WHERE `%s` = '%s'", table, keyfield, ast_str_buffer(buf));
for (field = rt_fields; field; field = field->next) {
ESCAPE_STRING(buf, field->value);
ast_str_append(&sql, 0, " AND `%s` = '%s'", field->name, ast_str_buffer(buf));
}
ast_debug(1, "MySQL RealTime: Delete SQL: %s\n", ast_str_buffer(sql));
/* Execution. */
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to delete from database: %s\n", mysql_error(&dbh->handle));
release_database(dbh);
return -1;
}
numrows = mysql_affected_rows(&dbh->handle);
release_database(dbh);
ast_debug(1, "MySQL RealTime: Deleted %" PRIu64 " rows on table: %s\n", numrows, table);
/* From http://dev.mysql.com/doc/mysql/en/mysql-affected-rows.html
* An integer greater than zero indicates the number of rows affected
* Zero indicates that no records were updated
* -1 indicates that the query returned an error (although, if the query failed, it should have been caught above.)
*/
return (int)numrows;
}
static struct ast_config *config_mysql(const char *database, const char *table, const char *file, struct ast_config *cfg, struct ast_flags config_flags, const char *unused, const char *who_asked)
{
struct mysql_conn *dbh;
MYSQL_RES *result;
MYSQL_ROW row;
uint64_t num_rows;
struct ast_variable *new_v;
struct ast_category *cur_cat = NULL;
struct ast_str *sql = ast_str_thread_get(&sql_buf, 200);
char last[80] = "";
int last_cat_metric = 0;
ast_clear_flag(&config_flags, CONFIG_FLAG_FILEUNCHANGED);
if (!file || !strcmp(file, RES_CONFIG_MYSQL_CONF)) {
ast_log(LOG_WARNING, "MySQL RealTime: Cannot configure myself.\n");
return NULL;
}
if (!(dbh = find_database(database, 0))) {
ast_log(LOG_WARNING, "MySQL RealTime: Invalid database specified: '%s' (check res_mysql.conf)\n", database);
return NULL;
}
ast_str_set(&sql, 0, "SELECT category, var_name, var_val, cat_metric FROM %s WHERE filename='%s' and commented=0 ORDER BY filename, category, cat_metric desc, var_metric asc, var_name, var_val, id", table, file);
ast_debug(1, "MySQL RealTime: Static SQL: %s\n", ast_str_buffer(sql));
/* We now have our complete statement; Lets connect to the server and execute it. */
if (!mysql_reconnect(dbh)) {
return NULL;
}
if (mysql_real_query(&dbh->handle, ast_str_buffer(sql), ast_str_strlen(sql))) {
ast_log(LOG_WARNING, "MySQL RealTime: Failed to query database. Check debug for more info.\n");
ast_debug(1, "MySQL RealTime: Query: %s\n", ast_str_buffer(sql));
ast_debug(1, "MySQL RealTime: Query Failed because: %s\n", mysql_error(&dbh->handle));
release_database(dbh);
return NULL;
}
if ((result = mysql_store_result(&dbh->handle))) {
num_rows = mysql_num_rows(result);
ast_debug(1, "MySQL RealTime: Found %" PRIu64 " rows.\n", num_rows);
/* There might exist a better way to access the column names other than counting,
* but I believe that would require another loop that we don't need. */
while ((row = mysql_fetch_row(result))) {
if (!strcmp(row[1], "#include")) {
if (!ast_config_internal_load(row[2], cfg, config_flags, "", who_asked)) {
mysql_free_result(result);
release_database(dbh);
return NULL;
}
continue;
}
if (strcmp(last, row[0]) || last_cat_metric != atoi(row[3])) {
cur_cat = ast_category_new_dynamic(row[0]);
if (!cur_cat) {
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
break;
}
strcpy(last, row[0]);
last_cat_metric = atoi(row[3]);
ast_category_append(cfg, cur_cat);
}
new_v = ast_variable_new(row[1], row[2], "");
if (cur_cat)
ast_variable_append(cur_cat, new_v);
}
} else {
ast_log(LOG_WARNING, "MySQL RealTime: Could not find config '%s' in database.\n", file);
}
mysql_free_result(result);
release_database(dbh);
return cfg;
}
static int unload_mysql(const char *database, const char *tablename)
{
struct tables *cur;
AST_LIST_LOCK(&mysql_tables);
AST_LIST_TRAVERSE_SAFE_BEGIN(&mysql_tables, cur, list) {
if (strcmp(cur->name, tablename) == 0) {
AST_LIST_REMOVE_CURRENT(list);
destroy_table(cur);
break;
}
}
AST_LIST_TRAVERSE_SAFE_END
AST_LIST_UNLOCK(&mysql_tables);
return cur ? 0 : -1;
}
static int require_mysql(const char *database, const char *tablename, va_list ap)
{
struct columns *column;
struct tables *table = find_table(database, tablename);
char *elm;
int type;
int size;
int res = 0;
if (!table) {
ast_log(LOG_WARNING, "Table %s not found in database. This table should exist if you're using realtime.\n", tablename);
return -1;
}
while ((elm = va_arg(ap, char *))) {
type = va_arg(ap, require_type);
size = va_arg(ap, int);
AST_LIST_TRAVERSE(&table->columns, column, list) {
if (strcmp(column->name, elm) == 0) {
/* Char can hold anything, as long as it is large enough */
if (strncmp(column->type, "char", 4) == 0 || strncmp(column->type, "varchar", 7) == 0) {
if ((size > column->len) && column->len != -1) {
ast_log(LOG_WARNING, "Realtime table %s@%s: Column '%s' should be at least %d long, but is only %d long.\n", database, tablename, column->name, size, column->len);
res = -1;
}
} else if (strcasestr(column->type, "unsigned")) {
if (!ast_rq_is_int(type)) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' cannot be type '%s' (need %s)\n",
database, tablename, column->name, column->type,
type == RQ_CHAR ? "char" : type == RQ_FLOAT ? "float" :
type == RQ_DATETIME ? "datetime" : type == RQ_DATE ? "date" : "a rather stiff drink");
res = -1;
} else if (strncasecmp(column->type, "tinyint", 1) == 0) {
if (type != RQ_UINTEGER1) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "smallint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 && type != RQ_UINTEGER2) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "mediumint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_UINTEGER3) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "int", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_UINTEGER3 && type != RQ_INTEGER3 &&
type != RQ_UINTEGER4) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "bigint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_UINTEGER3 && type != RQ_INTEGER3 &&
type != RQ_UINTEGER4 && type != RQ_INTEGER4 &&
type != RQ_UINTEGER8) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
}
} else if (strcasestr(column->type, "int")) {
if (!ast_rq_is_int(type)) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' cannot be type '%s' (need %s)\n",
database, tablename, column->name, column->type,
type == RQ_CHAR ? "char" : type == RQ_FLOAT ? "float" :
type == RQ_DATETIME ? "datetime" : type == RQ_DATE ? "date" :
"to get a life, rather than writing silly error messages");
res = -1;
} else if (strncasecmp(column->type, "tinyint", 1) == 0) {
if (type != RQ_INTEGER1) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "smallint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 && type != RQ_INTEGER2) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "mediumint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_INTEGER3) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "int", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_UINTEGER3 && type != RQ_INTEGER3 &&
type != RQ_INTEGER4) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
} else if (strncasecmp(column->type, "bigint", 1) == 0) {
if (type != RQ_UINTEGER1 && type != RQ_INTEGER1 &&
type != RQ_UINTEGER2 && type != RQ_INTEGER2 &&
type != RQ_UINTEGER3 && type != RQ_INTEGER3 &&
type != RQ_UINTEGER4 && type != RQ_INTEGER4 &&
type != RQ_INTEGER8) {
ast_log(LOG_WARNING, "Realtime table %s@%s: column '%s' may not be large enough for " \
"the required data length: %d (detected stringtype)\n", \
tablename, database, column->name, size); \
res = -1; \
}
}
} else if (strncmp(column->type, "float", 5) == 0) {
if (!ast_rq_is_int(type) && type != RQ_FLOAT) {
ast_log(LOG_WARNING, "Realtime table %s@%s: Column %s cannot be a %s\n", tablename, database, column->name, column->type);
res = -1;
} else if (strncmp(column->type, "datetime", 8) == 0 || strncmp(column->type, "timestamp", 9) == 0) {
if (type != RQ_DATETIME) {
ast_log(LOG_WARNING, "Realtime table %s@%s: Column %s cannot be a %s\n", tablename, database, column->name, column->type);
res = -1;
} else if (strncmp(column->type, "date", 4) == 0) {
if (type != RQ_DATE) {
ast_log(LOG_WARNING, "Realtime table %s@%s: Column %s cannot be a %s\n", tablename, database, column->name, column->type);
res = -1;
}
} else { /* Other, possibly unsupported types? */
ast_log(LOG_WARNING, "Possibly unsupported column type '%s' on column '%s'\n", column->type, column->name);
res = -1;
}
break;
}
}
if (!column) {
ast_log(LOG_WARNING, "Table %s requires a column '%s' of size '%d', but no such column exists.\n", tablename, elm, size);
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
}
}
release_table(table);
return res;
}
static struct ast_config_engine mysql_engine = {
.name = "mysql",
.load_func = config_mysql,
.realtime_func = realtime_mysql,
.realtime_multi_func = realtime_multi_mysql,
.store_func = store_mysql,
.destroy_func = destroy_mysql,
.update_func = update_mysql,
.update2_func = update2_mysql,
.require_func = require_mysql,
.unload_func = unload_mysql,
};
static int load_module(void)
{
parse_config(0);
ast_config_engine_register(&mysql_engine);
Tilghman Lesher
committed
ast_verb(2, "MySQL RealTime driver loaded.\n");
ast_cli_register_multiple(cli_realtime_mysql_status, sizeof(cli_realtime_mysql_status) / sizeof(struct ast_cli_entry));
return 0;
}
static int unload_module(void)
{
struct mysql_conn *cur;
struct tables *table;
ast_cli_unregister_multiple(cli_realtime_mysql_status, sizeof(cli_realtime_mysql_status) / sizeof(struct ast_cli_entry));
ast_config_engine_deregister(&mysql_engine);
Tilghman Lesher
committed
ast_verb(2, "MySQL RealTime unloaded.\n");
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
AST_RWLIST_WRLOCK(&databases);
while ((cur = AST_RWLIST_REMOVE_HEAD(&databases, list))) {
mysql_close(&cur->handle);
ast_mutex_destroy(&cur->lock);
ast_free(cur);
}
AST_RWLIST_UNLOCK(&databases);
/* Destroy cached table info */
AST_LIST_LOCK(&mysql_tables);
while ((table = AST_LIST_REMOVE_HEAD(&mysql_tables, list))) {
destroy_table(table);
}
AST_LIST_UNLOCK(&mysql_tables);
return 0;
}
static int reload(void)
{
parse_config(1);
ast_verb(2, "MySQL RealTime reloaded.\n");
return 0;
}
static int parse_config(int reload)
{
struct ast_config *config = NULL;
struct ast_flags config_flags = { reload ? CONFIG_FLAG_FILEUNCHANGED : 0 };
const char *catg;
struct mysql_conn *cur;
if ((config = ast_config_load(RES_CONFIG_MYSQL_CONF, config_flags)) == CONFIG_STATUS_FILEMISSING) {
/* Support old config file name */
config = ast_config_load(RES_CONFIG_MYSQL_CONF_OLD, config_flags);
}
if (config == CONFIG_STATUS_FILEMISSING) {
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
return 0;
} else if (config == CONFIG_STATUS_FILEUNCHANGED) {
return 0;
} else if (config == CONFIG_STATUS_FILEINVALID) {
ast_log(LOG_ERROR, "Not %sloading " RES_CONFIG_MYSQL_CONF "\n", reload ? "re" : "");
}
AST_RWLIST_WRLOCK(&databases);
for (catg = ast_category_browse(config, NULL); catg; catg = ast_category_browse(config, catg)) {
/* Does this category already exist? */
AST_RWLIST_TRAVERSE(&databases, cur, list) {
if (!strcmp(cur->unique_name, catg)) {
break;
}
}
if (!cur) {
if (!(cur = ast_calloc(1, sizeof(*cur) + strlen(catg) + 1))) {
ast_log(LOG_WARNING, "Could not allocate space for MySQL database '%s'\n", catg);
continue;
}
strcpy(cur->unique_name, catg); /* SAFE */
ast_mutex_init(&cur->lock);
AST_RWLIST_INSERT_TAIL(&databases, cur, list);
}
load_mysql_config(config, catg, cur);
}
AST_RWLIST_UNLOCK(&databases);
ast_config_destroy(config);
return 0;
}
static int load_mysql_config(struct ast_config *config, const char *category, struct mysql_conn *conn)
{
const char *s;
if (!(s = ast_variable_retrieve(config, category, "dbuser"))) {
ast_log(LOG_WARNING, "MySQL RealTime: No database user found, using 'asterisk' as default.\n");
s = "asterisk";
}
ast_copy_string(conn->user, s, sizeof(conn->user));
if (!(s = ast_variable_retrieve(config, category, "dbpass"))) {
ast_log(LOG_WARNING, "MySQL RealTime: No database password found, using 'asterisk' as default.\n");
s = "asterisk";
}
ast_copy_string(conn->pass, s, sizeof(conn->pass));
if (!(s = ast_variable_retrieve(config, category, "dbhost"))) {
ast_log(LOG_WARNING, "MySQL RealTime: No database host found, using localhost via socket.\n");
s = "";
}
ast_copy_string(conn->host, s, sizeof(conn->host));
if (!(s = ast_variable_retrieve(config, category, "dbname"))) {
ast_log(LOG_WARNING, "MySQL RealTime: No database name found, using 'asterisk' as default.\n");
s = "asterisk";
}
ast_copy_string(conn->name, s, sizeof(conn->name));
if (!(s = ast_variable_retrieve(config, category, "dbport"))) {
ast_log(LOG_WARNING, "MySQL RealTime: No database port found, using 3306 as default.\n");
conn->port = 3306;
} else
conn->port = atoi(s);
if (!(s = ast_variable_retrieve(config, category, "dbsock"))) {
if (ast_strlen_zero(conn->host)) {
char *paths[3] = { "/tmp/mysql.sock", "/var/lib/mysql/mysql.sock", "/var/run/mysqld/mysqld.sock" };
struct stat st;
int i;
for (i = 0; i < 3; i++) {
if (!stat(paths[i], &st)) {
ast_log(LOG_WARNING, "MySQL RealTime: No database socket found, using '%s' as default.\n", paths[i]);
ast_copy_string(conn->sock, paths[i], sizeof(conn->sock));
}
}
if (i == 3) {
ast_log(LOG_WARNING, "MySQL RealTime: No database socket found (and unable to detect a suitable path).\n");
return 0;
}
}
} else
ast_copy_string(conn->sock, s, sizeof(conn->sock));
Erin Spiceland
committed
if ((s = ast_variable_retrieve(config, category, "dbcharset"))) {
ast_copy_string(conn->charset, s, sizeof(conn->charset));
}
1299
1300
1301
1302
1303
1304
1305
1306
1307
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
1319
1320
if (!(s = ast_variable_retrieve(config, category, "requirements"))) {
ast_log(LOG_WARNING, "MySQL realtime: no requirements setting found, using 'warn' as default.\n");
conn->requirements = RQ_WARN;
} else if (!strcasecmp(s, "createclose")) {
conn->requirements = RQ_CREATECLOSE;
} else if (!strcasecmp(s, "createchar")) {
conn->requirements = RQ_CREATECHAR;
} else if (!strcasecmp(s, "warn")) {
conn->requirements = RQ_WARN;
} else {
ast_log(LOG_WARNING, "MySQL realtime: unrecognized requirements setting '%s', using 'warn'\n", s);
conn->requirements = RQ_WARN;
}
if (!ast_strlen_zero(conn->host)) {
ast_debug(1, "MySQL RealTime host: %s\n", conn->host);
ast_debug(1, "MySQL RealTime port: %i\n", conn->port);
} else
ast_debug(1, "MySQL RealTime socket: %s\n", conn->sock);
ast_debug(1, "MySQL RealTime database name: %s\n", conn->name);
ast_debug(1, "MySQL RealTime user: %s\n", conn->user);
ast_debug(1, "MySQL RealTime password: %s\n", conn->pass);
if(!ast_strlen_zero(conn->charset))
Erin Spiceland
committed
ast_debug(1, "MySQL RealTime charset: %s\n", conn->charset);
return 1;
}
static int mysql_reconnect(struct mysql_conn *conn)
{
#ifdef MYSQL_OPT_RECONNECT
my_bool trueval = 1;
#endif
/* mutex lock should have been locked before calling this function. */
reconnect_tryagain:
if ((!conn->connected) && (!ast_strlen_zero(conn->host) || !ast_strlen_zero(conn->sock)) && !ast_strlen_zero(conn->user) && !ast_strlen_zero(conn->name)) {
if (!mysql_init(&conn->handle)) {
ast_log(LOG_WARNING, "MySQL RealTime: Insufficient memory to allocate MySQL resource.\n");
conn->connected = 0;
return 0;
}
if(strlen(conn->charset) > 2){
Erin Spiceland
committed
char set_names[255];
char statement[512];
snprintf(set_names, sizeof(set_names), "SET NAMES %s", conn->charset);
mysql_real_escape_string(&conn->handle, statement, set_names, sizeof(set_names));
mysql_options(&conn->handle, MYSQL_INIT_COMMAND, set_names);
mysql_options(&conn->handle, MYSQL_SET_CHARSET_NAME, conn->charset);
}
if (mysql_real_connect(&conn->handle, conn->host, conn->user, conn->pass, conn->name, conn->port, conn->sock, 0)) {
#ifdef MYSQL_OPT_RECONNECT
/* The default is no longer to automatically reconnect on failure,
* (as of 5.0.3) so we have to set that option here. */
mysql_options(&conn->handle, MYSQL_OPT_RECONNECT, &trueval);
#endif
ast_debug(1, "MySQL RealTime: Successfully connected to database.\n");
conn->connected = 1;
conn->connect_time = time(NULL);
return 1;
} else {
ast_log(LOG_ERROR, "MySQL RealTime: Failed to connect database server %s on %s (err %d). Check debug for more info.\n", conn->name, !ast_strlen_zero(conn->host) ? conn->host : conn->sock, mysql_errno(&conn->handle));
ast_debug(1, "MySQL RealTime: Cannot Connect (%d): %s\n", mysql_errno(&conn->handle), mysql_error(&conn->handle));
conn->connected = 0;
return 0;
}
} else {
/* MySQL likes to return an error, even if it reconnects successfully.
* So the postman pings twice. */
if (mysql_ping(&conn->handle) != 0 && (usleep(1) + 2 > 0) && mysql_ping(&conn->handle) != 0) {
conn->connected = 0;
ast_log(LOG_ERROR, "MySQL RealTime: Ping failed (%d). Trying an explicit reconnect.\n", mysql_errno(&conn->handle));
ast_debug(1, "MySQL RealTime: Server Error (%d): %s\n", mysql_errno(&conn->handle), mysql_error(&conn->handle));
goto reconnect_tryagain;
}
if (!conn->connected) {
conn->connected = 1;
conn->connect_time = time(NULL);
}
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461
1462
1463
1464
1465
if (mysql_select_db(&conn->handle, conn->name) != 0) {
ast_log(LOG_WARNING, "MySQL RealTime: Unable to select database: %s. Still Connected (%u) - %s.\n", conn->name, mysql_errno(&conn->handle), mysql_error(&conn->handle));
return 0;
}
ast_debug(1, "MySQL RealTime: Connection okay.\n");
return 1;
}
}
static char *handle_cli_realtime_mysql_cache(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
{
struct tables *cur;
int l, which;
char *ret = NULL;
switch (cmd) {
case CLI_INIT:
e->command = "realtime mysql cache";
e->usage =
"Usage: realtime mysql cache [<database> <table>]\n"
" Shows table cache for the MySQL RealTime driver\n";
return NULL;
case CLI_GENERATE:
if (a->argc < 4 || a->argc > 5) {
return NULL;
}
l = strlen(a->word);
which = 0;
if (a->argc == 5) {
AST_LIST_LOCK(&mysql_tables);
AST_LIST_TRAVERSE(&mysql_tables, cur, list) {
if (!strcasecmp(a->argv[3], cur->database->unique_name) && !strncasecmp(a->word, cur->name, l) && ++which > a->n) {
ret = ast_strdup(cur->name);
break;
}
}
AST_LIST_UNLOCK(&mysql_tables);
} else {
struct mysql_conn *cur;
AST_RWLIST_RDLOCK(&databases);
AST_RWLIST_TRAVERSE(&databases, cur, list) {
if (!strncasecmp(a->word, cur->unique_name, l) && ++which > a->n) {
ret = ast_strdup(cur->unique_name);
break;
}
}
AST_RWLIST_UNLOCK(&databases);
}
return ret;
}
if (a->argc == 3) {
/* List of tables */
AST_LIST_LOCK(&mysql_tables);
AST_LIST_TRAVERSE(&mysql_tables, cur, list) {
ast_cli(a->fd, "%20.20s %s\n", cur->database->unique_name, cur->name);
}
AST_LIST_UNLOCK(&mysql_tables);
} else if (a->argc == 4) {
int found = 0;
/* List of tables */
AST_LIST_LOCK(&mysql_tables);
AST_LIST_TRAVERSE(&mysql_tables, cur, list) {
if (!strcasecmp(cur->database->unique_name, a->argv[3])) {
ast_cli(a->fd, "%s\n", cur->name);
found = 1;
}
}
AST_LIST_UNLOCK(&mysql_tables);
if (!found) {
ast_cli(a->fd, "No tables cached within %s database\n", a->argv[3]);
}
} else if (a->argc == 5) {
/* List of columns */
if ((cur = find_table(a->argv[3], a->argv[4]))) {
struct columns *col;
ast_cli(a->fd, "Columns for Table Cache '%s':\n", a->argv[3]);
ast_cli(a->fd, "%-20.20s %-20.20s %-3.3s\n", "Name", "Type", "Len");
AST_LIST_TRAVERSE(&cur->columns, col, list) {
ast_cli(a->fd, "%-20.20s %-20.20s %3d\n", col->name, col->type, col->len);
}
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
1490
1491
1492
1493
1494
1495
1496
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
1516
1517
1518
1519
1520
1521
1522
1523
1524
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
1535
1536
1537
1538
1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
1550
1551
1552
1553
1554
1555
} else {
ast_cli(a->fd, "No such table '%s'\n", a->argv[3]);
}
}
return CLI_SUCCESS;
}
static char *handle_cli_realtime_mysql_status(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
{
char status[256], status2[100] = "", type[20];
char *ret = NULL;
int ctime = 0, found = 0;
struct mysql_conn *cur;
int l = 0, which = 0;
switch (cmd) {
case CLI_INIT:
e->command = "realtime mysql status";
e->usage =
"Usage: realtime mysql status [<database>]\n"
" Shows connection information for the MySQL RealTime driver\n";
return NULL;
case CLI_GENERATE:
if (a->argc == 4) {
AST_RWLIST_RDLOCK(&databases);
AST_RWLIST_TRAVERSE(&databases, cur, list) {
if (!strncasecmp(a->word, cur->unique_name, l) && ++which > a->n) {
ret = ast_strdup(cur->unique_name);
break;
}
}
AST_RWLIST_UNLOCK(&databases);
}
return ret;
}
if (a->argc != 3)
return CLI_SHOWUSAGE;
AST_RWLIST_RDLOCK(&databases);
AST_RWLIST_TRAVERSE(&databases, cur, list) {
if (a->argc == 3 || (a->argc == 4 && !strcasecmp(a->argv[3], cur->unique_name))) {
found = 1;
if (mysql_reconnect(cur)) {
snprintf(type, sizeof(type), "connected to");
ctime = time(NULL) - cur->connect_time;
} else {
snprintf(type, sizeof(type), "configured for");
ctime = -1;
}
if (!ast_strlen_zero(cur->host)) {
snprintf(status, sizeof(status), "%s %s %s@%s, port %d", cur->unique_name, type, cur->name, cur->host, cur->port);
} else {
snprintf(status, sizeof(status), "%s %s %s on socket file %s", cur->unique_name, type, cur->name, cur->sock);
}
if (!ast_strlen_zero(cur->user)) {
snprintf(status2, sizeof(status2), " with username %s", cur->user);
} else {
status2[0] = '\0';
}
if (ctime > 31536000) {
ast_cli(a->fd, "%s%s for %.1f years.\n", status, status2, (double)ctime / 31536000.0);
} else if (ctime > 86400 * 30) {
ast_cli(a->fd, "%s%s for %d days.\n", status, status2, ctime / 86400);
} else if (ctime > 86400) {
ast_cli(a->fd, "%s%s for %d days, %d hours.\n", status, status2, ctime / 86400, (ctime % 86400) / 3600);
} else if (ctime > 3600) {
ast_cli(a->fd, "%s%s for %d hours, %d minutes.\n", status, status2, ctime / 3600, (ctime % 3600) / 60);
} else if (ctime > 60) {
ast_cli(a->fd, "%s%s for %d minutes.\n", status, status2, ctime / 60);
} else if (ctime > -1) {
ast_cli(a->fd, "%s%s for %d seconds.\n", status, status2, ctime);
} else {
ast_cli(a->fd, "%s%s.\n", status, status2);
}
}
}
AST_RWLIST_UNLOCK(&databases);
if (!found) {
ast_cli(a->fd, "No connections configured.\n");
}
return CLI_SUCCESS;
}
Tilghman Lesher
committed
AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "MySQL RealTime Configuration Driver",
Rodrigo Ramírez Norambuena
committed
.support_level = AST_MODULE_SUPPORT_EXTENDED,
.load = load_module,
.unload = unload_module,
.reload = reload,
.load_pri = AST_MODPRI_REALTIME_DRIVER,
.requires = "extconfig",