#include "portable.h"

#include <stdio.h>
#include <stdlib.h>

#include <ac/socket.h>
#include <ac/string.h>

#include "ldap-int.h"

static const char* features[] = {
#ifdef LDAP_API_FEATURE_THREAD_SAFE
	"THREAD_SAFE",
#endif
#ifdef LDAP_API_FEATURE_SESSION_THREAD_SAFE
	"SESSION_THREAD_SAFE",
#endif
#ifdef LDAP_API_FEATURE_OPERATION_THREAD_SAFE
	"OPERATION_THREAD_SAFE",
#endif
#ifdef LDAP_API_FEATURE_X_OPENLDAP_REEENTRANT
	"X_OPENLDAP_REENTRANT",
#endif
#if defined( LDAP_API_FEATURE_X_OPENLDAP_THREAD_SAFE ) && \
	defined( LDAP_THREAD_SAFE )
	"X_OPENLDAP_THREAD_SAFE",
#endif
#ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_DNS
	"X_OPENLDAP_V2_DNS",
#endif
#ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_REFERRALS
	"X_OPENLDAP_V2_REFERRALS",
#endif
	NULL
};

int
ldap_get_option(
	LDAP	*ld,
	int		option,
	void	*outvalue)
{
	struct ldapoptions *lo;

	if(!openldap_ldap_initialized) {
		openldap_ldap_initialize();
	}

	if(outvalue == NULL) {
		/* no place to get to */
		return -1;
	}

	if(ld == NULL) {
		lo = &openldap_ldap_global_options;
	} else {
		lo = &ld->ld_options;
	}

	switch(option) {
	case LDAP_OPT_API_INFO: {
			struct ldapapiinfo *info = (struct ldapapiinfo *) outvalue;

			if(info == NULL) {
				/* outvalue must point to an apiinfo structure */
				return -1;
			}

			if(info->ldapai_info_version != LDAP_API_INFO_VERSION) {
				/* api info version mismatch */
				info->ldapai_info_version = LDAP_API_INFO_VERSION;
				return -1;
			}

			info->ldapai_api_version = LDAP_API_VERSION;
			info->ldapai_api_version = LDAP_API_VERSION;
			info->ldapai_protocol_version = LDAP_VERSION_MAX;
			if(features[0] == NULL) {
				info->ldapai_extensions = NULL;
			} else {
				int i;
				info->ldapai_extensions = malloc(sizeof(features));

				for(i=0; features[i] != NULL; i++) {
					info->ldapai_extensions[i] =
						ldap_strdup(features[i]);
				}

				info->ldapai_extensions[i] = NULL;
			}

			info->ldapai_vendor_name = ldap_strdup(LDAP_VENDOR_NAME);
			info->ldapai_vendor_version = LDAP_VENDOR_VERSION;

			return 0;
		} break;

	case LDAP_OPT_DESC:
		if(ld == NULL) {
			/* bad param */
			break;
		} 

		* (int *) outvalue = ld->ld_sb.sb_sd;
		return 0;

	case LDAP_OPT_DEREF:
		* (int *) outvalue = lo->ldo_deref;
		return 0;

	case LDAP_OPT_SIZELIMIT:
		* (int *) outvalue = lo->ldo_sizelimit;
		return 0;

	case LDAP_OPT_TIMELIMIT:
		* (int *) outvalue = lo->ldo_timelimit;
		return 0;

	case LDAP_OPT_REFERRALS:
		* (int *) outvalue = (int) LDAP_BOOL_GET(lo, LDAP_BOOL_REFERRALS);
		return 0;
		
	case LDAP_OPT_RESTART:
		* (int *) outvalue = (int) LDAP_BOOL_GET(lo, LDAP_BOOL_RESTART);
		return 0;

	case LDAP_OPT_DNS:	/* LDAPv2 */
		* (int *) outvalue = (int) LDAP_BOOL_GET(lo, LDAP_BOOL_DNS);
		return 0;

	case LDAP_OPT_PROTOCOL_VERSION:
		if ((ld != NULL) && ld->ld_version) {
			* (int *) outvalue = ld->ld_version;
		} else { 
			* (int *) outvalue = lo->ldo_version;
		}
		return 0;

	case LDAP_OPT_SERVER_CONTROLS:
	case LDAP_OPT_CLIENT_CONTROLS:
		/* not yet supported */
		break;

	case LDAP_OPT_HOST_NAME:
		/*
		 * draft-ietf-ldapext-ldap-c-api-01 doesn't state
		 * whether client to have to free host names or no,
		 * we do
		 */

		* (char **) outvalue = ldap_strdup(lo->ldo_defhost);
		return 0;

	case LDAP_OPT_ERROR_NUMBER:
		if(ld == NULL) {
			/* bad param */
			break;
		} 
		* (int *) outvalue = ld->ld_errno;
		return 0;

	case LDAP_OPT_ERROR_STRING:
		/* not yet supported */
		if(ld == NULL) {
			/* bad param */
			break;
		} 

		/*
		 * draft-ietf-ldapext-ldap-c-api-01 doesn't require
		 *	the client to have to free error strings, we do
		 */

		if( ld->ld_error == NULL ) {
			* (char **) outvalue = NULL;
		} else {
			* (char **) outvalue = ldap_strdup(ld->ld_error);
		}
		break;

	default:
		/* bad param */
		break;
	}

	return -1;
}

int
ldap_set_option(
	LDAP	*ld,
	int		option,
	void	*invalue)
{
	struct ldapoptions *lo;

	if(!openldap_ldap_initialized) {
		openldap_ldap_initialize();
	}

	if(invalue == NULL) {
		/* no place to set from */
		return -1;
	}

	if(ld == NULL) {
		lo = &openldap_ldap_global_options;
	} else {
		lo = &ld->ld_options;
	}

	switch(option) {
	case LDAP_OPT_API_INFO:
	case LDAP_OPT_DESC:
		/* READ ONLY */
		break;

	case LDAP_OPT_DEREF:
		lo->ldo_deref = * (int *) invalue;
		return 0;

	case LDAP_OPT_SIZELIMIT:
		lo->ldo_sizelimit = * (int *) invalue;
		return 0;

	case LDAP_OPT_TIMELIMIT:
		lo->ldo_timelimit = * (int *) invalue;
		return 0;

	case LDAP_OPT_REFERRALS:
		if((int) invalue == (int) LDAP_OPT_ON) {
			LDAP_BOOL_SET(lo, LDAP_BOOL_REFERRALS);
		} else {
			LDAP_BOOL_CLR(lo, LDAP_BOOL_REFERRALS);
		}
		return 0;

	case LDAP_OPT_RESTART:
		if((int) invalue == (int) LDAP_OPT_ON) {
			LDAP_BOOL_SET(lo, LDAP_BOOL_RESTART);
		} else {
			LDAP_BOOL_CLR(lo, LDAP_BOOL_RESTART);
		}
		return 0;

	case LDAP_OPT_PROTOCOL_VERSION: {
			int vers = * (int *) invalue;
			if (vers < LDAP_VERSION_MIN || vers > LDAP_VERSION_MAX) {
				/* not supported */
				break;
			}
			ld->ld_version = vers;
		} return 0;

	case LDAP_OPT_SERVER_CONTROLS:
	case LDAP_OPT_CLIENT_CONTROLS:
		/* not yet supported */
		break;

	case LDAP_OPT_HOST_NAME: {
			char* host = * (char **) invalue;

			if(lo->ldo_defhost != NULL) {
				free(lo->ldo_defhost);
				lo->ldo_defhost = NULL;
			}

			if(host != NULL) {
				lo->ldo_defhost = ldap_strdup(host);
				return 0;
			}

			if(ld == NULL) {
				/*
				 * must want global default returned
				 * to initial condition.
				 */
				lo->ldo_defhost = ldap_strdup("localhost");

			} else {
				/*
				 * must want the session default
				 *   updated to the current global default
				 */
				lo->ldo_defhost = ldap_strdup(
					openldap_ldap_global_options.ldo_defhost);
			}
		} return 0;

	case LDAP_OPT_ERROR_NUMBER: {
			int err = * (int *) invalue;

			if (err != 0 ) {
				/* not supported */
				/* we only allow ld_errno to be cleared. */
				break;
			}

			if(ld == NULL) {
				/* need a struct ldap */
				break;
			}

			ld->ld_errno = err;
		} return 0;

	case LDAP_OPT_ERROR_STRING: {
			char* err = * (char **) invalue;

			if (err != NULL ) {
				/* not supported */
				/* we only allow ld_error to be cleared. */
				break;
			}

			if(ld == NULL) {
				/* need a struct ldap */
				break;
			}

			ld->ld_error = err;
		} return 0;

	default:
		/* bad param */
		break;
	}
	return -1;
}