/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */


#include <rampart_util.h>

/* Load a .dll or .so module */
static void*
rampart_load_module(
    const axutil_env_t *env,
    axis2_char_t *module_name,
    axutil_param_t **param)
{
    axutil_dll_desc_t *dll_desc = NULL;
    axutil_param_t *impl_info_param = NULL;
    void *ptr = NULL;

    AXIS2_LOG_DEBUG(env->log, AXIS2_LOG_SI,  "[rampart]Trying to load module %s", module_name);
    dll_desc = axutil_dll_desc_create(env);
    axutil_dll_desc_set_name(dll_desc, env, module_name);
    impl_info_param = axutil_param_create(env, NULL, dll_desc);
    axutil_param_set_value_free(impl_info_param, env, axutil_dll_desc_free_void_arg);
    axutil_class_loader_init(env);
    ptr = axutil_class_loader_create_dll(env, impl_info_param);

    if (!ptr)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart] Unable to load the module %s.", module_name);
        axutil_param_free(impl_info_param, env);
    }
    else
    {
        AXIS2_LOG_DEBUG(env->log, AXIS2_LOG_SI, 
            "[rampart]Successfully loaded module %s", module_name);
        *param = impl_info_param;
    }

    return ptr;
}

/**
 * Load the credentials module
 * User MUST free memory
 * @param env pointer to environment struct
 * @param cred_module_name name of the credentails module to be loaded
 * @return the loaded credentails module
 */
AXIS2_EXTERN rampart_credentials_t* AXIS2_CALL
rampart_load_credentials_module(
    const axutil_env_t *env,
    axis2_char_t *cred_module_name)
{
    rampart_credentials_t *cred = NULL;
    axutil_param_t *param = NULL;

    cred = (rampart_credentials_t*)rampart_load_module(env, cred_module_name, &param);
    if(!cred)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Unable to identify the credentials  module %s.", cred_module_name);
    }
    else if(param)
    {
        cred->param = param;
    }

    return cred;
}

/**
 * Call credentials module
 * User MUST free memory of username and password
 * @param env pointer to environment struct
 * @param cred_module the credentails module
 * @param ctx the message context
 * @param username reference to the returned username
 * @param password reference to the returned password
 * @return the status of the operation
 */
AXIS2_EXTERN rampart_credentials_status_t AXIS2_CALL
rampart_call_credentials(
    const axutil_env_t *env,
    rampart_credentials_t *cred_module,
    axis2_msg_ctx_t *msg_ctx,
    axis2_char_t **username,
    axis2_char_t **password)
{
    rampart_credentials_status_t cred_status = RAMPART_CREDENTIALS_GENERAL_ERROR;
    cred_status = RAMPART_CREDENTIALS_USERNAME_GET(cred_module, env, msg_ctx, username, password);
    return cred_status;
}

/**
 * Load authentication module
 * User MUST free memory
 * @param env pointer to environment struct
 * @param auth_module_name name of the authentication module
 * @return created athenticaiton module
 */
AXIS2_EXTERN rampart_authn_provider_t* AXIS2_CALL
rampart_load_auth_module(
    const axutil_env_t *env,
    axis2_char_t *auth_module_name)
{
    rampart_authn_provider_t *authp = NULL;
    axutil_param_t *param = NULL;

    authp = (rampart_authn_provider_t*)rampart_load_module(env, auth_module_name, &param);
    if(!authp)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Unable to identify the authentication module %s.", auth_module_name);
    }
    else if(param)
    {
        authp->param = param;
    }

    return authp;
}

/**
 * Load replay detection module
 * User MUST free memory
 * @param env pointer to environment struct
 * @param replay_detector_name name of the replay detection module
 * @return created replay detection module
 */
AXIS2_EXTERN rampart_replay_detector_t* AXIS2_CALL
rampart_load_replay_detector(
    const axutil_env_t *env,
    axis2_char_t *replay_detector_name)
{
    rampart_replay_detector_t *rd = NULL;
    axutil_param_t *param = NULL;

    rd = (rampart_replay_detector_t*)rampart_load_module(env, replay_detector_name, &param);
    if(!rd)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Unable to identify the replay detection  module %s.", replay_detector_name);
    }
    else if(param)
    {
        rd->param = param;
    }

    return rd;
}

/**
 * Load security context token provider
 * User MUST free memory
 * @param env pointer to environment struct
 * @param sct_provider_name name of the security context token provider 
 * @return created security context token provider module
 */
AXIS2_EXTERN rampart_sct_provider_t* AXIS2_CALL
rampart_load_sct_provider(
    const axutil_env_t *env,
    axis2_char_t *sct_provider_name)
{
    rampart_sct_provider_t *sct_provider = NULL;
    axutil_param_t *param = NULL;

    sct_provider = (rampart_sct_provider_t*)rampart_load_module(env, sct_provider_name, &param);
    if(!sct_provider)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Unable to identify security context token provider module %s.", 
            sct_provider_name);
    }
    else if(param)
    {
        sct_provider->param = param;
    }

    return sct_provider;
}

/**
 * Load the password callback module
 * User MUST free memory
 * @param env pointer to environment struct
 * @callback_module_name the name of the callback module
 * @return the loaded callback module
 */
AXIS2_EXTERN rampart_callback_t* AXIS2_CALL
rampart_load_pwcb_module(
    const axutil_env_t *env,
    axis2_char_t *callback_module_name)
{
    rampart_callback_t *cb = NULL;
    axutil_param_t *param = NULL;

    cb = (rampart_callback_t*)rampart_load_module(env, callback_module_name, &param);
    if(!cb)
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Unable to identify the callback module %s.", callback_module_name);
    }
    else if(param)
    {
        cb->param = param;
    }

    return cb;
}

/**
 * Call auth module
 * @param env pointer to environment struct
 * @param authp the authentication module
 * @param  username the username in the UsernameToken
 * @param  password the password in the UsernameToken
 * @param  nonce the nonce in the UsernameToken. Can be NULL if plain text password is used.
 * @param  created created time in UsernameToken. Can be NULL if plain text password is used.
 * @param password_type  the type of the password. either plain text of digest
 * @param msg_ctx the message context
 * @return status of the operation
 */
AXIS2_EXTERN rampart_authn_provider_status_t AXIS2_CALL
rampart_authenticate_un_pw(
    const axutil_env_t *env,
    rampart_authn_provider_t *authp,
    const axis2_char_t *username,
    const axis2_char_t *password,
    const axis2_char_t *nonce,
    const axis2_char_t *created,
    const axis2_char_t *password_type,
    axis2_msg_ctx_t *msg_ctx)
{
    rampart_authn_provider_status_t auth_status = RAMPART_AUTHN_PROVIDER_GENERAL_ERROR;

    if(authp)
    {
        if(!axutil_strcmp(password_type, RAMPART_PASSWORD_DIGEST_URI))
        {
            auth_status = RAMPART_AUTHN_PROVIDER_CHECK_PASSWORD_DIGEST(
                authp, env, msg_ctx, username, nonce, created, password);
        }
        else
        {
            auth_status = RAMPART_AUTHN_PROVIDER_CHECK_PASSWORD(
                authp, env, msg_ctx, username, password);
        }
    }
    else
    {
        AXIS2_LOG_ERROR(env->log, AXIS2_LOG_SI, 
            "[rampart]Cannot authenticate user. Authentication module is not valid");
    }

    return auth_status;
}

/**
 * Gets the password of given user.
 * @env the environment
 * @callback_module callback module structure
 * @username the name of the user to get the password
 * @return the password for the user or NULL if failed
 */
AXIS2_EXTERN axis2_char_t* AXIS2_CALL
rampart_callback_password(
    const axutil_env_t *env,
    rampart_callback_t *callback_module,
    const axis2_char_t *username)
{
    axis2_char_t *password = NULL;
    void *cb_prop_val= NULL;

    /*Get the password thru the callback*/
    password = RAMPART_CALLBACK_CALLBACK_PASSWORD(callback_module, env, username, cb_prop_val);
    return password;
}

/**
 * Get the password for pkcs12 key store.
 * @env pointer to environment struct
 * @callback pointer to rampart callback module
 * @username name of the pkcs12 owner
 * @return the password for the user or NULL if username is incorrect
 */
AXIS2_EXTERN axis2_char_t * AXIS2_CALL
rampart_callback_pkcs12_password(
	const axutil_env_t *env,
	rampart_callback_t *callback_module,
	const axis2_char_t *username)
{
	axis2_char_t *password = NULL;
	void *cb_prop_val = NULL;

	/*Get the password through the callback module*/
	password = RAMPART_CALLBACK_CALLBACK_PKCS12_PASSWORD(callback_module, env, username, cb_prop_val);
	return password;
}

/**
 * Generates time.
 * User MUST free memory
 * @param ttl Time to live. The time difference between created and expired in mili seconds.
 * @param with_millisecond  shows whether millisecond precision is needed or not
 * @return generated time
 **/
AXIS2_EXTERN axis2_char_t* AXIS2_CALL
rampart_generate_time(
    const axutil_env_t *env, 
    int ttl, 
    axis2_bool_t with_millisecond)
{
    axutil_date_time_t *dt = NULL;
    axis2_char_t *dt_str = NULL;

    dt = axutil_date_time_create_with_offset(env, ttl);
    if(with_millisecond)
    {
        dt_str =  axutil_date_time_serialize_date_time(dt, env);
    }
    else
    {
        dt_str = axutil_date_time_serialize_date_time_without_millisecond(dt, env);
    }
    axutil_date_time_free(dt, env);
    return dt_str;
}

/**
 * Check if @dt1 < @dt2. if not returns a false
 * @param env pointer to environment struct
 * @param dt1 date time 1.
 * @param dt2 date time 2.
 * @return AXIS2_SUCCESS if dt1 < dt2. AXIS2_FALSE otherwise
 */
AXIS2_EXTERN axis2_status_t AXIS2_CALL
rampart_compare_date_time(
    const axutil_env_t *env, 
    axis2_char_t *dt1_str, 
    axis2_char_t *dt2_str)
{
    axis2_status_t status = AXIS2_FAILURE;
    axutil_date_time_t *dt1 = NULL;
    axutil_date_time_t *dt2 = NULL;
    axutil_date_time_comp_result_t res = AXIS2_DATE_TIME_COMP_RES_UNKNOWN;

    dt1 = axutil_date_time_create(env);
    status =  axutil_date_time_deserialize_date_time(dt1, env, dt1_str);
    if(status != AXIS2_SUCCESS)
    {
        axutil_date_time_free(dt1, env);
        return AXIS2_FAILURE;
    }

    dt2 = axutil_date_time_create(env);
    status =  axutil_date_time_deserialize_date_time(dt2, env, dt2_str);
    if (status != AXIS2_SUCCESS)
    {
        axutil_date_time_free(dt1, env);
        axutil_date_time_free(dt2, env);
        return AXIS2_FAILURE;
    }

    /* dt1<dt2 for SUCCESS */
    res = axutil_date_time_compare(dt1, env, dt2);
    axutil_date_time_free(dt1, env);
    axutil_date_time_free(dt2, env);
    if(AXIS2_DATE_TIME_COMP_RES_NOT_EXPIRED == res)
    {
        return AXIS2_SUCCESS;
    }
    else
    {
        return AXIS2_FAILURE;
    }
}