pdo_snowflake.c

/* Copyright (c) 2017-2019 Snowflake Computing Inc. All right reserved.  */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include "php.h"
#include "ext/standard/info.h"
#include "pdo/php_pdo.h"
#include "pdo/php_pdo_driver.h"
#include "php_pdo_snowflake.h"
#include "php_pdo_snowflake_int.h"

ZEND_DECLARE_MODULE_GLOBALS(pdo_snowflake)

#ifdef COMPILE_DL_PDO_SNOWFLAKE
#   if (PHP_VERSION_ID >= 70000)
#       ifdef ZTS
ZEND_TSRMLS_CACHE_DEFINE()
#       endif /* ZTS */
#   endif /* PHP_VERSION_ID */

ZEND_GET_MODULE(pdo_snowflake)

#endif /* COMPILE_DL_PDO_SNOWFLAKE */

/* {{{ PHP_INI_BEGIN
*/
PHP_INI_BEGIN()
    STD_PHP_INI_ENTRY
    ("pdo_snowflake.cacert", NULL, PHP_INI_SYSTEM, OnUpdateString, cacert,
     zend_pdo_snowflake_globals, pdo_snowflake_globals)
    STD_PHP_INI_ENTRY
    ("pdo_snowflake.logdir", NULL, PHP_INI_SYSTEM, OnUpdateString, logdir,
     zend_pdo_snowflake_globals, pdo_snowflake_globals)
    STD_PHP_INI_ENTRY
    ("pdo_snowflake.loglevel", NULL, PHP_INI_SYSTEM, OnUpdateString, loglevel,
     zend_pdo_snowflake_globals, pdo_snowflake_globals)
    STD_PHP_INI_ENTRY
    ("pdo_snowflake.debug", NULL, PHP_INI_SYSTEM, OnUpdateString, debug,
     zend_pdo_snowflake_globals, pdo_snowflake_globals)

PHP_INI_END()
/* }}} */

/* {{{ PHP_MINIT_FUNCTION
 */
static PHP_MINIT_FUNCTION(pdo_snowflake) {
    REGISTER_INI_ENTRIES();

    char *cacert = PDO_SNOWFLAKE_G(cacert);
    char *logdir = PDO_SNOWFLAKE_G(logdir);
    char* loglevel = PDO_SNOWFLAKE_G(loglevel);
    char* debug = PDO_SNOWFLAKE_G(debug);
    SF_USER_MEM_HOOKS php_hooks = {
        .alloc_fn = _pdo_snowflake_user_malloc,
        .calloc_fn = _pdo_snowflake_user_calloc,
        .realloc_fn = _pdo_snowflake_user_realloc,
        .dealloc_fn = _pdo_snowflake_user_dealloc
    };

    snowflake_global_init(logdir, log_from_str_to_level(loglevel), &php_hooks);
    snowflake_global_set_attribute(SF_GLOBAL_CA_BUNDLE_FILE, cacert);
    sf_bool debug_bool =
        (debug && strncasecmp(debug, "true", 4) == 0) ?
        SF_BOOLEAN_TRUE : SF_BOOLEAN_FALSE;
    snowflake_global_set_attribute(SF_GLOBAL_DEBUG, &debug_bool);

    REGISTER_PDO_CLASS_CONST_LONG("SNOWFLAKE_ATTR_SSL_CAPATH",
                                  (zend_long) PDO_SNOWFLAKE_ATTR_SSL_CAPATH);
    REGISTER_PDO_CLASS_CONST_LONG(
      "SNOWFLAKE_ATTR_SSL_VERIFY_CERTIFICATE_REVOCATION_STATUS",
      (zend_long) PDO_SNOWFLAKE_ATTR_SSL_VERIFY_CERTIFICATE_REVOCATION_STATUS);

    REGISTER_PDO_CLASS_CONST_LONG(
      "SNOWFLAKE_ATTR_QUERY_ID",
      (zend_long) PDO_SNOWFLAKE_ATTR_QUERY_ID);

    return php_pdo_register_driver(&pdo_snowflake_driver);
}
/* }}} */

/* {{{ PHP_MSHUTDOWN_FUNCTION
 */
static PHP_MSHUTDOWN_FUNCTION(pdo_snowflake) {
    snowflake_global_term();
    php_pdo_unregister_driver(&pdo_snowflake_driver);
    UNREGISTER_INI_ENTRIES();

    return SUCCESS;
}
/* }}} */

/* {{{ PHP_MINFO_FUNCTION
 */
static PHP_MINFO_FUNCTION(pdo_snowflake) {
    php_info_print_table_start();
    php_info_print_table_header(2, "PDO Driver for Snowflake", "enabled");
    /* TODO: get Snowflake Driver version, etc*/
    php_info_print_table_row(2, "Version", PDO_SNOWFLAKE_VERSION);

    php_info_print_table_end();

    DISPLAY_INI_ENTRIES();
}
/* }}} */

/* {{{ PHP_GINIT_FUNCTION */
static PHP_GINIT_FUNCTION(pdo_snowflake) {
#if (PHP_VERSION_ID >= 70000)
#if defined(COMPILE_DL_PDO_SNOWFLAKE) && defined(ZTS)
    ZEND_TSRMLS_CACHE_UPDATE();
#endif
#endif
    pdo_snowflake_globals->logdir = NULL;
    pdo_snowflake_globals->loglevel = "FATAL";
    pdo_snowflake_globals->cacert = NULL;
    pdo_snowflake_globals->debug = NULL;
}
/* }}} */

/* {{{ pdo_snowflake_functions[] */
const zend_function_entry pdo_snowflake_functions[] = {
  PHP_FE_END
};
/* }}} */

/* {{{ pdo_snowflake_deps[] */
static const zend_module_dep pdo_snowflake_deps[] = {
  ZEND_MOD_REQUIRED("pdo")
  ZEND_MOD_END
};
/* }}} */

/* {{{ pdo_snowflake_module_entry */
zend_module_entry pdo_snowflake_module_entry = {
  STANDARD_MODULE_HEADER_EX, NULL,
  pdo_snowflake_deps,
  "pdo_snowflake",
  pdo_snowflake_functions,
  PHP_MINIT(pdo_snowflake),
  PHP_MSHUTDOWN(pdo_snowflake),
  NULL,
  NULL,
  PHP_MINFO(pdo_snowflake),
  PDO_SNOWFLAKE_VERSION,
  PHP_MODULE_GLOBALS(pdo_snowflake),
  PHP_GINIT(pdo_snowflake),
  NULL,
  NULL,
  STANDARD_MODULE_PROPERTIES_EX
};
/* }}} */