#include <sqlite3.h>

#include <time.h>

#include "php_apm.h"

#include "php_ini.h"

#include "ext/standard/php_smart_str.h"

#include "ext/standard/php_filestat.h"

#include "ext/json/php_json.h"

#include "driver_sqlite3.h"

#ifdef NETWARE

#include <netinet/in.h>

#if HAVE_ARPA_INET_H

#include <arpa/inet.h>

ZEND_EXTERN_MODULE_GLOBALS(apm)

ZEND_DECLARE_MODULE_GLOBALS(apm_sqlite3)

static int perform_db_access_checks(const char *path TSRMLS_DC)

{

zend_bool is_dir;

zval *stat;

MAKE_STD_ZVAL(stat);

php_stat(path, strlen(path), FS_IS_DIR, stat TSRMLS_CC);

is_dir = Z_BVAL_P(stat);

zval_dtor(stat);

FREE_ZVAL(stat);

/* Does db_path exists ? */

if (!is_dir) {

zend_error(E_CORE_WARNING, "APM cannot be enabled, '%s' is not directory", path);

return FAILURE;

}

if (VCWD_ACCESS(path, W_OK | R_OK | X_OK) != SUCCESS) {

zend_error(E_CORE_WARNING, "APM cannot be enabled, %s needs to readable, writable and executable", path);

return FAILURE;

}

return SUCCESS;

}

static PHP_INI_MH(OnUpdateDBFile)

{

if (new_value && new_value_length > 0) {

snprintf(APM_S3_G(db_file), MAXPATHLEN, "%s/%s", new_value, DB_FILE);

if (perform_db_access_checks(new_value TSRMLS_CC) == FAILURE) {

APM_S3_G(enabled) = 0;

}

} else {

APM_S3_G(enabled) = 0;

}

return OnUpdateString(entry, new_value, new_value_length, mh_arg1, mh_arg2, mh_arg3, stage TSRMLS_CC);

}

APM_DRIVER_CREATE(sqlite3)

PHP_INI_BEGIN()

/* Boolean controlling whether the driver is active or not */

STD_PHP_INI_BOOLEAN("apm.sqlite_enabled", "1", PHP_INI_PERDIR, OnUpdateBool, enabled, zend_apm_sqlite3_globals, apm_sqlite3_globals)

/* error_reporting of the driver */

STD_PHP_INI_ENTRY("apm.sqlite_error_reporting", NULL, PHP_INI_ALL, OnUpdateAPMsqlite3ErrorReporting, error_reporting, zend_apm_sqlite3_globals, apm_sqlite3_globals)

sqlite3 * sqlite_get_instance() {

if (APM_S3_G(event_db) == NULL) {

/* Opening the sqlite database file */

APM_DEBUG("[SQLite driver] Connecting to db...");

if (sqlite3_open(APM_S3_G(db_file), &APM_S3_G(event_db))) {

APM_DEBUG("FAILED!\n");

/*

sqlite3_close(APM_S3_G(event_db));

return NULL;

}

APM_DEBUG("OK\n");

sqlite3_busy_timeout(APM_S3_G(event_db), APM_S3_G(timeout));

/* Making the connection asynchronous, not waiting for data being really written to the disk */

sqlite3_exec(APM_S3_G(event_db), "PRAGMA synchronous = OFF", NULL, NULL, NULL);

}

return APM_S3_G(event_db);

}

void apm_driver_sqlite3_insert_request(char * uri, char * host, char * ip, char * cookies, char * post_vars, char * referer TSRMLS_DC)

{

char *sql, *script;

int ip_int = 0, code;

struct in_addr ip_addr;

sqlite3 *connection;

APM_DEBUG("[SQLite driver] Begin insert request\n", sql);

if (APM_S3_G(is_request_created)) {

APM_DEBUG("[SQLite driver] SKIPPED, request already created.\n", sql);

return;

}

SQLITE_INSTANCE_INIT

get_script(&script);

if (ip && (inet_pton(AF_INET, ip, &ip_addr) == 1)) {

ip_int = ntohl(ip_addr.s_addr);

}

/* Builing SQL insert query */

sql = sqlite3_mprintf("INSERT INTO request (ts, script, uri, host, ip, cookies, post_vars, referer) VALUES (%d, %Q, %Q, %Q, %d, %Q, %Q, %Q)",

(long)time(NULL), script ? script : "", uri ? uri : "", host ? host : "", ip_int, cookies ? cookies : "", post_vars ? post_vars : "", referer ? referer : "");

/* Executing SQL insert query */

APM_DEBUG("[SQLite driver] Sending: %s\n", sql);

if ((code = sqlite3_exec(connection, sql, NULL, NULL, NULL)) != SQLITE_OK)

APM_DEBUG("[SQLite driver] Error occured with previous query. Error code: %d\n", code);

sqlite3_free(sql);

APM_S3_G(request_id) = sqlite3_last_insert_rowid(connection);

APM_S3_G(is_request_created) = 1;

APM_DEBUG("[SQLite driver] End insert request\n", sql);

}

void apm_driver_sqlite3_insert_event(int type, char * error_filename, uint error_lineno, char * msg, char * trace TSRMLS_DC)

{

char *sql;

struct in_addr ip_addr;

sqlite3 *connection;

SQLITE_INSTANCE_INIT

/* Builing SQL insert query */

sql = sqlite3_mprintf("INSERT INTO event (request_id, ts, type, file, line, message, backtrace) VALUES (%d, %d, %d, %Q, %d, %Q, %Q)",

APM_S3_G(request_id), (long)time(NULL), type, error_filename ? error_filename : "", error_lineno, msg ? msg : "", trace ? trace : "");

/* Executing SQL insert query */

APM_DEBUG("[SQLite driver] Sending: %s\n", sql);

if (sqlite3_exec(connection, sql, NULL, NULL, NULL) != SQLITE_OK)

APM_DEBUG("[SQLite driver] Error occured with previous query\n");

sqlite3_free(sql);

}

int apm_driver_sqlite3_minit(int module_number)

{

REGISTER_INI_ENTRIES();

return SUCCESS;

}

int apm_driver_sqlite3_rinit()

{

APM_S3_G(is_request_created) = 0;

return SUCCESS;

}

int apm_driver_sqlite3_mshutdown()

{

return SUCCESS;

}

int apm_driver_sqlite3_rshutdown()

{

if (APM_S3_G(event_db) != NULL) {

APM_DEBUG("[SQLite driver] Closing connection\n");

sqlite3_close(APM_S3_G(event_db));

APM_S3_G(event_db) = NULL;

}

return SUCCESS;

}

void apm_driver_sqlite3_insert_slow_request(float duration TSRMLS_DC)

{

char *sql;

sqlite3 *connection;

SQLITE_INSTANCE_INIT

/* Building SQL insert query */

sql = sqlite3_mprintf("INSERT INTO slow_request (request_id, duration) VALUES (%d, %f)",

APM_S3_G(request_id), USEC_TO_SEC(duration));

/* Executing SQL insert query */

APM_DEBUG("[SQLite driver] Sending: %s\n", sql);

if (sqlite3_exec(connection, sql, NULL, NULL, NULL) != SQLITE_OK)

APM_DEBUG("[SQLite driver] Error occured with previous query\n");

sqlite3_free(sql);

}