redirect_cache.c

/*
  +----------------------------------------------------------------------+
  | PHP Version 7                                                        |
  +----------------------------------------------------------------------+
  | Copyright (c) The PHP Group                                          |
  +----------------------------------------------------------------------+
  | This source file is subject to version 3.01 of the PHP license,      |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.php.net/license/3_01.txt                                  |
  | If you did not receive a copy of the PHP license and are unable to   |
  | obtain it through the world-wide-web, please send a note to          |
  | license@php.net so we can mail you a copy immediately.               |
  +----------------------------------------------------------------------+
  | Authors: Qianqian Bu <qianqian.bu@microsoft.com>                     |
  +----------------------------------------------------------------------+
*/


#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "php_mysqlnd_azure.h"
#include "mysqlnd_azure.h"
#include "ext/standard/info.h"
#include "ext/mysqlnd/mysqlnd_ext_plugin.h"
#include "ext/mysqlnd/mysqlnd_structs.h"
#include "ext/mysqlnd/mysqlnd_statistics.h"
#include "ext/mysqlnd/mysqlnd_connection.h"

/* {{{ mysqlnd_azure_redirect_info_dtor */
static void mysqlnd_azure_redirect_info_dtor(zval *zv)
{
    MYSQLND_AZURE_REDIRECT_INFO *redirect_info = (MYSQLND_AZURE_REDIRECT_INFO*)Z_PTR_P(zv);

    if (redirect_info != NULL) {

        if (redirect_info->redirect_user) {
            mnd_pefree(redirect_info->redirect_user, 1);
            redirect_info->redirect_user = NULL;
        }
        if (redirect_info->redirect_host) {
            mnd_pefree(redirect_info->redirect_host, 1);
            redirect_info->redirect_host = NULL;
        }

        mnd_pefree(redirect_info, 1);
        redirect_info = NULL;
      
    }

}
/* }}} */

/* {{{ mysqlnd_azure_add_redirect_cache */
enum_func_status mysqlnd_azure_add_redirect_cache(const char* user, const char* host, int port, const char* redirect_user, const char* redirect_host, int redirect_port)
{
    if (MYSQLND_AZURE_G(redirectCache) == NULL) {
        MYSQLND_AZURE_G(redirectCache) = mnd_pemalloc(sizeof(HashTable), 1);
        if(MYSQLND_AZURE_G(redirectCache) == NULL) {
            return FAIL;
        }
        zend_hash_init(MYSQLND_AZURE_G(redirectCache), 0, NULL, mysqlnd_azure_redirect_info_dtor, 1);
    }

    char* key = NULL;
    mnd_sprintf(&key, MAX_REDIRECT_HOST_LEN+ MAX_REDIRECT_USER_LEN+8, "%s_%s_%d", user, host, port);
    if(!key) {
        return FAIL;
    }

    MYSQLND_AZURE_REDIRECT_INFO* redirect_info = mnd_pemalloc(sizeof(MYSQLND_AZURE_REDIRECT_INFO), 1);
    if(redirect_info == NULL) {
        return FAIL;
    }
    redirect_info->redirect_user = mnd_pestrndup(redirect_user, strlen(redirect_user), 1);
    redirect_info->redirect_host = mnd_pestrndup(redirect_host, strlen(redirect_host), 1);
    if(redirect_info->redirect_user == NULL || redirect_info->redirect_host == NULL) {
        return FAIL;
    }
    redirect_info->redirect_port = redirect_port;

    zend_hash_str_update_ptr(MYSQLND_AZURE_G(redirectCache), key, strlen(key), redirect_info);

    mnd_sprintf_free(key);

    return PASS;
}
/* }}} */

/* {{{ mysqlnd_azure_remove_redirect_cache */
enum_func_status mysqlnd_azure_remove_redirect_cache(const char* user, const char* host, int port)
{
    if (MYSQLND_AZURE_G(redirectCache) != NULL) {
        char *key = NULL;
        mnd_sprintf(&key, MAX_REDIRECT_HOST_LEN + MAX_REDIRECT_USER_LEN + 8, "%s_%s_%d", user, host, port);
        if(!key) {
            return FAIL;
        }
        zend_hash_str_del(MYSQLND_AZURE_G(redirectCache), key, strlen(key));

        mnd_sprintf_free(key);
    }

    return PASS;
}
/* }}} */

/* {{{ mysqlnd_azure_find_redirect_cache */
MYSQLND_AZURE_REDIRECT_INFO* mysqlnd_azure_find_redirect_cache(const char* user, const char* host, int port)
{
    if (MYSQLND_AZURE_G(redirectCache) != NULL) {
        char *key = NULL;
        mnd_sprintf(&key, MAX_REDIRECT_HOST_LEN + MAX_REDIRECT_USER_LEN + 8, "%s_%s_%d", user, host, port);
        if(!key) {
            return NULL;
        }

        void* zv_dest = zend_hash_str_find_ptr(MYSQLND_AZURE_G(redirectCache), key, strlen(key));
        mnd_sprintf_free(key);

        return (MYSQLND_AZURE_REDIRECT_INFO*)zv_dest;
    }

    return NULL;
}
/* }}} */