static void disconnect(TSRMLS_D)
{
if (APM_G(sqlite3_event_db) != NULL) {
sqlite3_close(APM_G(sqlite3_event_db));
APM_G(sqlite3_event_db) = NULL;
}
}
void apm_driver_mysql_process_stats(TSRMLS_D)
{
APM_DEBUG("[MySQL driver] ======apm_driver_mysql_process_stats \n");
char *sql = NULL;
MYSQL *connection;
apm_driver_mysql_insert_request(TSRMLS_C);
MYSQL_INSTANCE_INIT
sql = emalloc(170);
sprintf(
sql,
"INSERT INTO stats (request_id, duration, user_cpu, sys_cpu, mem_peak_usage) VALUES (@request_id, %f, %f, %f, %ld)",
USEC_TO_SEC(APM_G(duration)),
USEC_TO_SEC(APM_G(user_cpu)),
USEC_TO_SEC(APM_G(sys_cpu)),
APM_G(mem_peak_usage)
);
APM_DEBUG("[MySQL driver] Sending: %s\n", sql);
if (mysql_query(connection, sql) != 0)
APM_DEBUG("[MySQL driver] Error: %s\n", mysql_error(APM_G(mysql_event_db)));
efree(sql);
}
static void mysql_destroy(TSRMLS_D) {
APM_DEBUG("[MySQL driver] Closing connection\n");
mysql_close(APM_G(mysql_event_db));
free(APM_G(mysql_event_db));
APM_G(mysql_event_db) = NULL;
mysql_library_end();
}
/* {{{ void apm_error(int type, const char *format, ...)
This function provides a hook for error */
void apm_error_cb(int type, const char *error_filename, const uint error_lineno, const char *format, va_list args)
{
TSRMLS_FETCH();
char *msg;
va_list args_copy;
zend_module_entry tmp_mod_entry;
/* A copy of args is needed to be used for the old_error_cb */
va_copy(args_copy, args);
vspprintf(&msg, 0, format, args_copy);
va_end(args_copy);
if (APM_G(event_enabled)) {
/* We need to see if we have an uncaught exception fatal error now */
if (type == E_ERROR && strncmp(msg, "Uncaught exception", 18) == 0) {
} else {
insert_event(type, (char *) error_filename, error_lineno, msg TSRMLS_CC);
}
}
efree(msg);
/* Calling saved callback function for error handling, unless xdebug is loaded */
if (zend_hash_find(&module_registry, "xdebug", 7, (void**) &tmp_mod_entry) != SUCCESS) {
old_error_cb(type, error_filename, error_lineno, format, args);
}
}
int apm_driver_mysql_mshutdown(SHUTDOWN_FUNC_ARGS)
{
if (APM_G(mysql_event_db) != NULL) {
mysql_destroy(TSRMLS_C);
}
return SUCCESS;
}
static int apm_end_silence_opcode_handler(ZEND_OPCODE_HANDLER_ARGS)
{
APM_G(currently_silenced) = 0;
if (_orig_end_silence_opcode_handler)
return _orig_end_silence_opcode_handler(ZEND_OPCODE_HANDLER_ARGS_PASSTHRU);
return ZEND_USER_OPCODE_DISPATCH;
}
/* Insert an event in the backend */
void apm_driver_mysql_process_event(PROCESS_EVENT_ARGS)
{
APM_DEBUG("[MySQL driver] ======apm_driver_mysql_process_event \n");
char *filename_esc = NULL, *msg_esc = NULL, *trace_esc = NULL, *sql = NULL;
int filename_len = 0, msg_len = 0, trace_len = 0;
MYSQL *connection;
apm_driver_mysql_insert_request(TSRMLS_C);
MYSQL_INSTANCE_INIT
if (error_filename) {
filename_len = strlen(error_filename);
filename_esc = emalloc(filename_len * 2 + 1);
filename_len = mysql_real_escape_string(connection, filename_esc, error_filename, filename_len);
}
if (msg) {
msg_len = strlen(msg);
msg_esc = emalloc(msg_len * 2 + 1);
msg_len = mysql_real_escape_string(connection, msg_esc, msg, msg_len);
}
if (trace) {
trace_len = strlen(trace);
trace_esc = emalloc(trace_len * 2 + 1);
trace_len = mysql_real_escape_string(connection, trace_esc, trace, trace_len);
}
sql = emalloc(135 + filename_len + msg_len + trace_len);
sprintf(
sql,
"INSERT INTO event (request_id, type, file, line, message, backtrace) VALUES (@request_id, %d, '%s', %u, '%s', '%s')",
type, error_filename ? filename_esc : "", error_lineno, msg ? msg_esc : "", trace ? trace_esc : "");
APM_DEBUG("[MySQL driver] Sending: %s\n", sql);
if (mysql_query(connection, sql) != 0)
APM_DEBUG("[MySQL driver] Error: %s\n", mysql_error(APM_G(mysql_event_db)));
efree(sql);
efree(filename_esc);
efree(msg_esc);
efree(trace_esc);
}
/* {{{ void apm_error(int type, const char *format, ...)
This function provides a hook for error */
void apm_error_cb(int type, const char *error_filename, const uint error_lineno, const char *format, va_list args)
{
char *msg;
va_list args_copy;
zend_module_entry tmp_mod_entry;
TSRMLS_FETCH();
/* A copy of args is needed to be used for the old_error_cb */
va_copy(args_copy, args);
vspprintf(&msg, 0, format, args_copy);
va_end(args_copy);
if (APM_G(event_enabled)) {
process_event(APM_EVENT_ERROR, type, (char *) error_filename, error_lineno, msg TSRMLS_CC);
}
efree(msg);
/* Calling saved callback function for error handling, unless xdebug is loaded */
if (zend_hash_find(&module_registry, "xdebug", 7, (void**) &tmp_mod_entry) != SUCCESS) {
old_error_cb(type, error_filename, error_lineno, format, args);
}
}
/* Returns the MYSQL instance (singleton) */
MYSQL * mysql_get_instance(TSRMLS_D) {
my_bool reconnect = 1;
if (APM_G(mysql_event_db) == NULL) {
APM_DEBUG("[MySQL driver] mysql_event_db is null (mysql handler)... \n");
mysql_library_init(0, NULL, NULL);
APM_G(mysql_event_db) = malloc(sizeof(MYSQL));
mysql_init(APM_G(mysql_event_db));
mysql_options(APM_G(mysql_event_db), MYSQL_OPT_RECONNECT, &reconnect);
APM_DEBUG("[MySQL driver] Connecting to server...");
if (mysql_real_connect(APM_G(mysql_event_db), APM_G(mysql_db_host), APM_G(mysql_db_user), APM_G(mysql_db_pass), APM_G(mysql_db_name), APM_G(mysql_db_port), NULL, 0) == NULL) {
APM_DEBUG("FAILED! Message: %s\n", mysql_error(APM_G(mysql_event_db)));
mysql_destroy(TSRMLS_C);
return NULL;
}
APM_DEBUG("OK\n");
mysql_set_character_set(APM_G(mysql_event_db), "utf8");
mysql_query(
APM_G(mysql_event_db),
"\
CREATE TABLE IF NOT EXISTS request (\
id INTEGER UNSIGNED PRIMARY KEY auto_increment,\
application VARCHAR(255) NOT NULL,\
ts TIMESTAMP NOT NULL,\
script TEXT NOT NULL,\
uri TEXT NOT NULL,\
host TEXT NOT NULL,\
ip INTEGER UNSIGNED NOT NULL,\
cookies TEXT NOT NULL,\
post_vars TEXT NOT NULL,\
referer TEXT NOT NULL,\
method TEXT NOT NULL,\
status TEXT NOT NULL\
)"
);
mysql_query(
APM_G(mysql_event_db),
"\
CREATE TABLE IF NOT EXISTS event (\
id INTEGER UNSIGNED PRIMARY KEY auto_increment,\
request_id INTEGER UNSIGNED,\
ts TIMESTAMP NOT NULL,\
type SMALLINT UNSIGNED NOT NULL,\
file TEXT NOT NULL,\
line MEDIUMINT UNSIGNED NOT NULL,\
message TEXT NOT NULL,\
backtrace BLOB NOT NULL,\
KEY request (request_id)\
)"
);
mysql_query(
APM_G(mysql_event_db),
"\
CREATE TABLE IF NOT EXISTS stats (\
id INTEGER UNSIGNED PRIMARY KEY auto_increment,\
request_id INTEGER UNSIGNED,\
duration FLOAT UNSIGNED NOT NULL,\
user_cpu FLOAT UNSIGNED NOT NULL,\
sys_cpu FLOAT UNSIGNED NOT NULL,\
mem_peak_usage INTEGER UNSIGNED NOT NULL,\
KEY request (request_id)\
)"
);
}
return APM_G(mysql_event_db);
}
int apm_driver_mysql_rinit(TSRMLS_D)
{
APM_G(mysql_is_request_created) = 0;
return SUCCESS;
}
/* Insert a request in the backend */
static void apm_driver_mysql_insert_request(TSRMLS_D)
{
APM_DEBUG("[MySQL driver] ======apm_driver_mysql_insert_request \n");
char *application_esc = NULL, *script_esc = NULL, *uri_esc = NULL, *host_esc = NULL, *cookies_esc = NULL, *post_vars_esc = NULL, *referer_esc = NULL, *method_esc = NULL, *status_esc = NULL, *sql = NULL;
unsigned int application_len = 0, script_len = 0, uri_len = 0, host_len = 0, ip_int = 0, cookies_len = 0, post_vars_len = 0, referer_len = 0, method_len = 0, status_len = 0;
struct in_addr ip_addr;
MYSQL *connection;
extract_data();
APM_DEBUG("[MySQL driver] Begin insert request\n");
if (APM_G(mysql_is_request_created)) {
APM_DEBUG("[MySQL driver] SKIPPED, request already created.\n");
return;
}
MYSQL_INSTANCE_INIT
if (APM_G(application_id)) {
application_len = strlen(APM_G(application_id));
application_esc = emalloc(application_len * 2 + 1);
application_len = mysql_real_escape_string(connection, application_esc, APM_G(application_id), application_len);
}
APM_MYSQL_ESCAPE_STR(script);
APM_MYSQL_ESCAPE_STR(uri);
APM_MYSQL_ESCAPE_STR(host);
APM_MYSQL_ESCAPE_STR(referer);
APM_MYSQL_ESCAPE_STR(method);
APM_MYSQL_ESCAPE_STR(status);
APM_MYSQL_ESCAPE_SMART_STR(cookies);
APM_MYSQL_ESCAPE_SMART_STR(post_vars);
if (APM_RD(ip_found) && (inet_pton(AF_INET, APM_RD_STRVAL(ip), &ip_addr) == 1)) {
ip_int = ntohl(ip_addr.s_addr);
}
sql = emalloc(166 + application_len + script_len + uri_len + host_len + cookies_len + post_vars_len + referer_len + method_len);
sprintf(
sql,
"INSERT INTO request (application, script, uri, host, ip, cookies, post_vars, referer, method, status) VALUES ('%s', '%s', '%s', '%s', %u, '%s', '%s', '%s', '%s', '%s')",
application_esc ? application_esc : "",
APM_RD(script_found) ? script_esc : "",
APM_RD(uri_found) ? uri_esc : "",
APM_RD(host_found) ? host_esc : "",
ip_int, APM_RD(cookies_found) ? cookies_esc : "",
APM_RD(post_vars_found) ? post_vars_esc : "",
APM_RD(referer_found) ? referer_esc : "",
APM_RD(method_found) ? method_esc : "",
APM_RD(status_found) ? status_esc : "");
APM_DEBUG("[MySQL driver] Sending: %s\n", sql);
if (mysql_query(connection, sql) != 0)
APM_DEBUG("[MySQL driver] Error: %s\n", mysql_error(APM_G(mysql_event_db)));
mysql_query(connection, "SET @request_id = LAST_INSERT_ID()");
efree(sql);
if (application_esc)
efree(application_esc);
if (script_esc)
efree(script_esc);
if (uri_esc)
efree(uri_esc);
if (host_esc)
efree(host_esc);
if (cookies_esc)
efree(cookies_esc);
if (post_vars_esc)
efree(post_vars_esc);
if (referer_esc)
efree(referer_esc);
if (method_esc)
efree(method_esc);
if (status_esc)
efree(status_esc);
APM_G(mysql_is_request_created) = 1;
APM_DEBUG("[MySQL driver] End insert request\n");
}
void apm_driver_statsd_process_stats()
{
int socketDescriptor;
char data[1024];
if (
(socketDescriptor = socket(APM_SD_G(servinfo)->ai_family, APM_SD_G(servinfo)->ai_socktype, APM_SD_G(servinfo)->ai_protocol)) != -1
) {
APM_DEBUG("Sending data to StatsD");
sprintf(data, "%1$s.duration:%2$f|ms\n%1$s.user_cpu:%3$f|ms\n%1$s.sys_cpu:%4$f|ms\n%1$s.mem_peak_usage:%5$ld|g\n%1$s.response.code.%6$d:1|c", APM_SD_G(prefix), APM_G(duration) / 1000, APM_G(user_cpu) / 1000, APM_G(sys_cpu) / 1000, APM_G(mem_peak_usage), SG(sapi_headers).http_response_code);
if (sendto(socketDescriptor, data, strlen(data), 0, APM_SD_G(servinfo)->ai_addr, APM_SD_G(servinfo)->ai_addrlen) == -1) {/* cannot send */ }
close(socketDescriptor);
}
}
static int apm_write(const char *str, uint length) {
TSRMLS_FETCH();
smart_str_appendl(APM_G(buffer), str, length);
smart_str_0(APM_G(buffer));
return length;
}
/* Insert a request in the backend */
void apm_driver_mysql_insert_request(TSRMLS_D)
{
char *script = NULL, *application_esc, *script_esc = NULL, *uri_esc = NULL, *host_esc = NULL, *cookies_esc = NULL, *post_vars_esc = NULL, *referer_esc = NULL, *sql = NULL;
unsigned int application_len = 0, script_len = 0, uri_len = 0, host_len = 0, ip_int = 0, cookies_len = 0, post_vars_len = 0, referer_len = 0;
struct in_addr ip_addr;
MYSQL *connection;
zval *tmp;
EXTRACT_DATA();
APM_DEBUG("[MySQL driver] Begin insert request\n");
if (APM_MY_G(is_request_created)) {
APM_DEBUG("[MySQL driver] SKIPPED, request already created.\n");
return;
}
MYSQL_INSTANCE_INIT
if (APM_G(application_id)) {
application_len = strlen(APM_G(application_id));
application_esc = emalloc(application_len * 2 + 1);
application_len = mysql_real_escape_string(connection, application_esc, APM_G(application_id), application_len);
}
get_script(&script);
if (script) {
script_len = strlen(script);
script_esc = emalloc(script_len * 2 + 1);
script_len = mysql_real_escape_string(connection, script_esc, script, script_len);
}
if (APM_RD(uri_found)) {
uri_len = strlen(Z_STRVAL_PP(APM_RD(uri)));
uri_esc = emalloc(uri_len * 2 + 1);
uri_len = mysql_real_escape_string(connection, uri_esc, Z_STRVAL_PP(APM_RD(uri)), uri_len);
}
if (APM_RD(host_found)) {
host_len = strlen(Z_STRVAL_PP(APM_RD(host)));
host_esc = emalloc(host_len * 2 + 1);
host_len = mysql_real_escape_string(connection, host_esc, Z_STRVAL_PP(APM_RD(host)), host_len);
}
if (APM_RD(ip_found) && (inet_pton(AF_INET, Z_STRVAL_PP(APM_RD(ip)), &ip_addr) == 1)) {
ip_int = ntohl(ip_addr.s_addr);
}
if (APM_RD(cookies_found)) {
cookies_len = strlen(APM_RD(cookies).c);
cookies_esc = emalloc(cookies_len * 2 + 1);
cookies_len = mysql_real_escape_string(connection, cookies_esc, APM_RD(cookies).c, cookies_len);
}
if (APM_RD(post_vars_found)) {
post_vars_len = strlen(APM_RD(post_vars).c);
post_vars_esc = emalloc(post_vars_len * 2 + 1);
post_vars_len = mysql_real_escape_string(connection, post_vars_esc, APM_RD(post_vars).c, post_vars_len);
}
if (APM_RD(referer_found)) {
referer_len = strlen(Z_STRVAL_PP(APM_RD(referer)));
referer_esc = emalloc(referer_len * 2 + 1);
referer_len = mysql_real_escape_string(connection, referer_esc, Z_STRVAL_PP(APM_RD(referer)), referer_len);
}
sql = emalloc(154 + application_len + script_len + uri_len + host_len + cookies_len + post_vars_len + referer_len);
sprintf(
sql,
"INSERT INTO request (application, script, uri, host, ip, cookies, post_vars, referer) VALUES ('%s', '%s', '%s', '%s', %u, '%s', '%s', '%s')",
APM_G(application_id) ? application_esc : "", script ? script_esc : "", APM_RD(uri_found) ? uri_esc : "", APM_RD(host_found) ? host_esc : "", ip_int, APM_RD(cookies_found) ? cookies_esc : "", APM_RD(post_vars_found) ? post_vars_esc : "", APM_RD(referer_found) ? referer_esc : "");
APM_DEBUG("[MySQL driver] Sending: %s\n", sql);
if (mysql_query(connection, sql) != 0)
APM_DEBUG("[MySQL driver] Error: %s\n", mysql_error(APM_MY_G(event_db)));
mysql_query(connection, "SET @request_id = LAST_INSERT_ID()");
efree(sql);
efree(script_esc);
efree(uri_esc);
efree(host_esc);
efree(cookies_esc);
efree(post_vars_esc);
efree(referer_esc);
APM_MY_G(is_request_created) = 1;
APM_DEBUG("[MySQL driver] End insert request\n");
}