/* Copyright 2004 IBM Corporation
 * All rights reserved.
 * Redisribution and use in source and binary forms, with or without
 * modification, are permitted only as authorizd by the OpenLADP
 * Public License.
 */
/* ACKNOWLEDGEMENTS
 * This work originally developed by Sang Seok Lim
 * 2004/06/18	03:20:00	slim@OpenLDAP.org
 */

#include "portable.h"
#include <ac/string.h>
#include <ac/socket.h>
#include <ldap_pvt.h>
#include "lutil.h"
#include <ldap.h>
#include "slap.h"

#include "component.h"
#include "asn.h"
#include <asn-gser.h>

#include <string.h>

#ifndef SLAPD_COMP_MATCH
#define SLAPD_COMP_MATCH SLAPD_MOD_DYNAMIC
#endif


OD_entry* gOD_table = NULL;
AsnTypetoMatchingRuleTable* gATMR_table = NULL;

int
load_derived_matching_rule ( char* cfg_path ){
}

MatchingRule*
retrieve_matching_rule( char* mr_oid, AsnTypeId type ) {
	char* tmp;
	struct berval mr_name = BER_BVNULL;
	AsnTypetoMatchingRuleTable* atmr;

	for ( atmr = gATMR_table ; atmr ; atmr = atmr->atmr_table_next ) {
		if ( strcmp( atmr->atmr_oid, mr_oid ) == 0 ) {
			tmp = atmr->atmr_table[type].atmr_mr_name;
			if ( tmp ) {
				mr_name.bv_val = tmp;
				mr_name.bv_len = strlen( tmp );
				return mr_bvfind ( &mr_name );
			}
		}
	}
	return (MatchingRule*)NULL;
}

OD_entry*
retrieve_oid_decoder_table ( char* oid ) {
	OD_entry* curr_entry;
	for ( curr_entry = gOD_table ; curr_entry != NULL ;
				curr_entry = curr_entry->oe_next ) {
		if ( strcmp ( curr_entry->oe_oid , oid ) == 0 )
			return curr_entry;
	}
	return (OD_entry*) NULL;
}

int
add_OD_entry ( char* oid, gser_decoder_func* gser_decoder ,
		ber_decoder_func ber_decoder, converter_func* converter ) {
	OD_entry* new_entry;

	if ( !gOD_table ) {
		gOD_table = new_entry = (OD_entry*) malloc( sizeof ( OD_entry ) );
		gOD_table->oe_next = NULL;
		gOD_table->oe_prev = NULL;
	}
	else {
		new_entry = (OD_entry*) malloc( sizeof ( OD_entry ) );
		if ( !new_entry ) return -1;
		gOD_table->oe_prev = new_entry;
		new_entry->oe_next = gOD_table;
		new_entry->oe_prev = NULL;
		gOD_table = new_entry;
	}

	strcpy ( new_entry->oe_oid ,oid );
	new_entry->oe_gser_decoder = gser_decoder;
	new_entry->oe_ber_decoder = ber_decoder;
	new_entry->oe_converter = converter;

	return 1;
}

int
remove_OD_entry ( char* oid ) {
	OD_entry* curr_entry;
	for ( curr_entry = gOD_table ; curr_entry != NULL ;
				curr_entry = curr_entry->oe_next ) {
		if ( strcmp ( curr_entry->oe_oid , oid ) == 0 ) {
			if ( !curr_entry->oe_next ) {
				if ( curr_entry->oe_prev ) {
					curr_entry->oe_prev->oe_next = NULL;
				}
			} else {
				curr_entry->oe_prev->oe_next = curr_entry->oe_next;
				curr_entry->oe_next->oe_prev = curr_entry->oe_prev;
			}
			free ( curr_entry );
			return 1;
		}
	}
	return -1;
}

void* 
comp_convert_attr_to_comp LDAP_P (( Attribute* a, Syntax *syn, struct berval* bv ))
{

	char* peek_head;
        int mode, bytesDecoded, size, rc;
        void* component;
	char* oid = a->a_desc->ad_type->sat_atype.at_oid ;
        GenBuf* b;
        ExpBuf* buf;
	OD_entry* od_entry;
	
	/* look for the decoder registered for the given attribute */
	od_entry =  retrieve_oid_decoder_table ( oid );
	if ( !od_entry || !od_entry->oe_ber_decoder ) return (void*)NULL;
	if ( od_entry->oe_converter ) {
		size = (*od_entry->oe_converter)( bv );
		if ( size <= 0 ) return (void*)NULL;
	}

	buf = ExpBufAllocBuf();
	ExpBuftoGenBuf( buf, &b );
	ExpBufInstallDataInBuf ( buf, bv->bv_val, bv->bv_len );
	BufResetInReadMode( b );

	mode = DEC_ALLOC_MODE_2;
	/*
	 * How can we decide which decoder will be called, GSER or BER?
	 * Currently BER decoder is called for a certificate.
	 * The flag of Attribute will say something about it in the future
	 */
	if ( slap_syntax_is_ber ( syn ) ) {
		rc =BDecComponentTop(od_entry->oe_ber_decoder, a->a_comp_data->cd_mem_op, b, 0,0, &component,&bytesDecoded,mode ) ;
	}
	else {
		rc = od_entry->oe_gser_decoder( a->a_comp_data->cd_mem_op, b, component,&bytesDecoded,mode);
	}

	ExpBufFreeBuf( buf );
	if ( rc == -1 ) {
		ShutdownNibbleMemLocal ( a->a_comp_data->cd_mem_op );
		a->a_comp_data->cd_mem_op = NULL;
		return (void*)NULL;
	}
	else {
		return component;
	}
}

#include <nibble-alloc.h>
void
comp_free_component ( void* mem_op ) {
	ShutdownNibbleMemLocal( (NibbleMem*)mem_op );
	return;
}

int
comp_convert_assert_to_comp (
	void* mem_op,
	ComponentSyntaxInfo *csi_attr,
	struct berval* bv,
	ComponentSyntaxInfo** csi, int* len, int mode )
{
	int rc;
	GenBuf* genBuf;
	ExpBuf* buf;
	gser_decoder_func *decoder = csi_attr->csi_comp_desc->cd_gser_decoder;

	buf = ExpBufAllocBuf();
	ExpBuftoGenBuf( buf, &genBuf );
	ExpBufInstallDataInBuf ( buf, bv->bv_val, bv->bv_len );
	BufResetInReadMode( genBuf );

	if ( csi_attr->csi_comp_desc->cd_type_id == BASICTYPE_ANY )
		decoder = ((ComponentAny*)csi_attr)->cai->GSER_Decode;

	rc = (*decoder)( mem_op, genBuf, csi, len, mode );
	ExpBufFreeBuf ( buf );
	return rc;
}

int intToAscii( int value, char* buf ) {
	int minus=0,i,temp;
	int total_num_digits;

	if ( value == 0 ){
		buf[0] = '0';
		return 1;
	}

	if ( value < 0 ){
		minus = 1;
		value = value*(-1);
		buf[0] = '-';
	}
	
	/* How many digits */
	for ( temp = value, total_num_digits=0 ; temp ; total_num_digits++ )
		temp = temp/10;

	total_num_digits += minus;

	for ( i = minus ; value ; i++ ) {
		buf[ total_num_digits - i - 1 ]= (char)(value%10 + '0');
		value = value/10;
	}
	return i;
}

int
comp_convert_asn_to_ldap ( MatchingRule* mr, ComponentSyntaxInfo* csi, struct berval* bv, int *allocated )
{
	int value, rc;
	Syntax* syn;

	AsnTypetoSyntax* asn_to_syn =
		&asn_to_syntax_mapping_tbl[csi->csi_comp_desc->cd_type_id];
	if ( asn_to_syn->ats_syn_oid )
		csi->csi_syntax = syn_find ( asn_to_syn->ats_syn_oid );
	else 
		csi->csi_syntax = NULL;


        switch ( csi->csi_comp_desc->cd_type_id ) {
          case BASICTYPE_BOOLEAN :
		bv->bv_val = (char*)malloc( 5 );
		*allocated = 1;
		bv->bv_len = 5;
		if ( ((ComponentBool*)csi)->value > 0 ) {
			strcpy ( bv->bv_val , "TRUE" );
			bv->bv_len = 4;
		}
		else {
			strcpy ( bv->bv_val , "FALSE" );
			bv->bv_len = 5;
		}
                break ;
          case BASICTYPE_NULL :
                bv->bv_len = 0;
                break;
          case BASICTYPE_INTEGER :
		bv->bv_val = (char*)malloc( INITIAL_ATTR_SIZE );
		*allocated = 1;
		bv->bv_len = INITIAL_ATTR_SIZE;
		bv->bv_len = intToAscii(((ComponentInt*)csi)->value, bv->bv_val );
		if ( bv->bv_len <= 0 )
			return LDAP_INVALID_SYNTAX;
                break;
          case BASICTYPE_REAL :
		return LDAP_INVALID_SYNTAX;
          case BASICTYPE_ENUMERATED :
		bv->bv_val = (char*)malloc( INITIAL_ATTR_SIZE );
		*allocated = 1;
		bv->bv_len = INITIAL_ATTR_SIZE;
		bv->bv_len = intToAscii(((ComponentEnum*)csi)->value, bv->bv_val );
		if ( bv->bv_len <= 0 )
			return LDAP_INVALID_SYNTAX;
                break;
          case BASICTYPE_OID :
          case BASICTYPE_OCTETSTRING :
          case BASICTYPE_BITSTRING :
          case BASICTYPE_NUMERIC_STR :
          case BASICTYPE_PRINTABLE_STR :
          case BASICTYPE_UNIVERSAL_STR :
          case BASICTYPE_IA5_STR :
          case BASICTYPE_BMP_STR :
          case BASICTYPE_UTF8_STR :
          case BASICTYPE_UTCTIME :
          case BASICTYPE_GENERALIZEDTIME :
          case BASICTYPE_GRAPHIC_STR :
          case BASICTYPE_VISIBLE_STR :
          case BASICTYPE_GENERAL_STR :
          case BASICTYPE_OBJECTDESCRIPTOR :
          case BASICTYPE_VIDEOTEX_STR :
          case BASICTYPE_T61_STR :
          case BASICTYPE_OCTETCONTAINING :
          case BASICTYPE_BITCONTAINING :
          case BASICTYPE_RELATIVE_OID :
		bv->bv_val = ((ComponentOcts*)csi)->value.octs;
		bv->bv_len = ((ComponentOcts*)csi)->value.octetLen;
                break;
	  case BASICTYPE_ANY :
		csi = ((ComponentAny*)csi)->value;
		if ( csi->csi_comp_desc->cd_type != ASN_BASIC ||
			csi->csi_comp_desc->cd_type_id == BASICTYPE_ANY )
			return LDAP_INVALID_SYNTAX;
		return comp_convert_asn_to_ldap( mr, csi, bv, allocated );
          case COMPOSITE_ASN1_TYPE :
          case RDNSequence :
		/*dnMatch*/
		if( strncmp( mr->smr_mrule.mr_oid, DN_MATCH_OID, strlen(DN_MATCH_OID) ) != 0 )
			return LDAP_INVALID_SYNTAX;
		*allocated = 1;
		rc = ConvertRDNSequence2RFC2253( (irRDNSequence*)csi, bv );
		return rc;
          case RelativeDistinguishedName :
		/*rdnMatch*/
		if( strncmp( mr->smr_mrule.mr_oid, RDN_MATCH_OID, strlen(RDN_MATCH_OID) ) != 0 )
			return LDAP_INVALID_SYNTAX;
		*allocated = 1;
		rc = ConvertRDN2RFC2253((irRelativeDistinguishedName*)csi,bv);
		return rc;
          case TelephoneNumber :
          case FacsimileTelephoneNumber__telephoneNumber :
		break;
          case DirectoryString :
		return LDAP_INVALID_SYNTAX;
          case ASN_COMP_CERTIFICATE :
          case ASNTYPE_END :
		break;
          default :
                /*Only ASN Basic Type can be converted into LDAP string*/
		return LDAP_INVALID_SYNTAX;
        }

	if ( csi->csi_syntax && csi->csi_syntax->ssyn_validate ) {
		if ( csi->csi_syntax->ssyn_validate(csi->csi_syntax, bv) != LDAP_SUCCESS )
			return LDAP_INVALID_SYNTAX;
	}

	return LDAP_SUCCESS;
}

/*
 * If <all> type component referenced is used
 * more than one component will be tested
 */
#define IS_TERMINAL_COMPREF(cr) (cr->cr_curr->ci_next == NULL)
int
comp_test_all_components (
	void* mem_op,
	ComponentSyntaxInfo *csi_attr,
	ComponentAssertion* ca )
{
	int rc;
	ComponentSyntaxInfo *csi_temp = NULL, *csi_assert = NULL, *comp_elmt = NULL;
	ComponentReference *cr = ca->ca_comp_ref;
	struct berval *ca_val = &ca->ca_ma_value;

	switch ( cr->cr_curr->ci_type ) {
	    case LDAP_COMPREF_IDENTIFIER:
	    case LDAP_COMPREF_FROM_BEGINNING:
	    case LDAP_COMPREF_FROM_END:
		csi_temp = (ComponentSyntaxInfo*)csi_attr->csi_comp_desc->cd_extract_i( mem_op, cr, csi_attr );
		if ( cr->cr_curr->ci_type == LDAP_COMPREF_ALL ) {
			rc = comp_test_all_components ( mem_op, csi_temp, ca );
		} else {
			rc = comp_test_one_component( mem_op, csi_temp, ca );
		}
		break;
	    case LDAP_COMPREF_COUNT:
		/* "count" component reference should be the last component id */
		if ( IS_TERMINAL_COMPREF(cr) ) {
			ComponentInt *k;
			k = (ComponentInt*)CompAlloc( mem_op, sizeof(ComponentInt) );
			k->comp_desc = CompAlloc( mem_op, sizeof( ComponentDesc ) );
			k->comp_desc->cd_tag = 0;
			k->comp_desc->cd_gser_decoder = (gser_decoder_func*)GDecComponentInt;
			k->comp_desc->cd_ber_decoder = (ber_decoder_func*)BDecComponentInt;
			k->comp_desc->cd_extract_i = (extract_component_from_id_func*)NULL;
			k->comp_desc->cd_type = ASN_BASIC;
			k->comp_desc->cd_type_id = BASICTYPE_INTEGER;
			k->comp_desc->cd_all_match = (allcomponent_matching_func*)MatchingComponentInt;
			k->value = AsnListCount(&((ComponentList*)csi_attr)->comp_list);
			rc = comp_test_one_component( mem_op, k, ca );
		} else {
			rc = LDAP_INVALID_SYNTAX;
		}
		break;
	    case LDAP_COMPREF_ALL:
		if ( IS_TERMINAL_COMPREF(cr) ) {
			FOR_EACH_LIST_ELMT( comp_elmt, &((ComponentList*)csi_attr)->comp_list )
			{
				rc = comp_test_one_component( mem_op, comp_elmt, ca );
				if ( rc == LDAP_COMPARE_TRUE ) {
					break;
				}
			}
		} else {
			ComponentId *start_compid = ca->ca_comp_ref->cr_curr->ci_next;
			FOR_EACH_LIST_ELMT( comp_elmt, &((ComponentList*)csi_attr)->comp_list )
			{
				cr->cr_curr = start_compid;
				csi_temp = comp_elmt->csi_comp_desc->cd_extract_i( mem_op, cr, comp_elmt );
				if ( cr->cr_curr->ci_type == LDAP_COMPREF_ALL ) {
					rc = comp_test_all_components ( mem_op, csi_temp, ca );
				} else {
					rc = comp_test_one_component ( mem_op, csi_temp, ca );
				}

				if ( rc == LDAP_COMPARE_TRUE ) {
					break;
				}
			}
		}
		break;
	    case LDAP_COMPREF_CONTENT:
	    case LDAP_COMPREF_SELECT:
	    case LDAP_COMPREF_DEFINED:
	    case LDAP_COMPREF_UNDEFINED:
		rc = LDAP_OPERATIONS_ERROR;
		break;
	    default:
		rc = LDAP_OPERATIONS_ERROR;
	}
	return rc;
}

void
eat_bv_whsp ( struct berval* in )
{
	char* end = in->bv_val + in->bv_len;
        for ( ; ( *in->bv_val == ' ' ) && ( in->bv_val < end ) ; ) {
                in->bv_val++;
        }
}

int
get_primitive_GSER_value ( struct berval* in )
{
	int count, sequent_dquote, unclosed_brace, succeed;
	char* ptr = in->bv_val;
	char* end = in->bv_val + in->bv_len;

	eat_bv_whsp( in );
	/*
 	 * Four cases of GSER <Values>
	 * 1) "..." :
	 *      StringVal, GeneralizedTimeVal, UTCTimeVal, ObjectDescriptorVal
	 * 2) '...'B or '...'H :
	 *      BitStringVal, OctetStringVal
	 * 3) {...} :
	 *      SEQUENCE, SEQUENCEOF, SETOF, SET, CHOICE, BIT STRING(bit list)
	 * 4) Between two white spaces
	 *      INTEGER, BOOLEAN, NULL,ENUMERATE, REAL
	 */

	if ( in->bv_len <= 0 )
		return LDAP_INVALID_SYNTAX;

	succeed = 0;
	if ( ptr[0] == '"' ) {
		for( count = 1, sequent_dquote = 0 ; ; count++ ) {
			/* In order to find escaped double quote */
			if ( ptr[count] == '"' ) sequent_dquote++;
			else sequent_dquote = 0;

			if ( ptr[count] == '\0' || (ptr + count) > end ) {
				break;
			}

			if ( ( ptr[count] == '"' && ptr[count-1] != '"') ||
			( sequent_dquote > 2 && (sequent_dquote%2) == 1 ) ) {
				succeed = 1;
				break;
			}
		}

        	if ( !succeed || ptr[count] != '"' )
			return LDAP_FILTER_ERROR;

		in->bv_val = ptr+1; /*the next to '"'*/
		in->bv_len = count - 1; /* exclude '"' */
	}
	else if ( ptr[0] == '\'' ) {
		for( count = 1 ; ; count++ ) {
			if ( ptr[count] == '\0' || (ptr+count) > end ) {
				break;
			}
			if ((ptr[count-1] == '\'' && ptr[count] == 'B')||
			(ptr[count-1] == '\'' && ptr[count] == 'H') ) {
				succeed = 1;
				break;
			}
		}

        	if ( !succeed || !(ptr[count] == 'H' || ptr[count] == 'B') )
			return LDAP_FILTER_ERROR;

		in->bv_val = ptr+1; /* the next to '"' */
		in->bv_len = count - 2; /* exclude "'H" or "'B" */

	}
	else if ( ptr[0] == '{' ) {
		for( count = 1, unclosed_brace = 1 ; ; count++ ) {
			if ( ptr[count] == '{' ) unclosed_brace++;
			if ( ptr[count] == '}' ) unclosed_brace--;

			if ( ptr[count] == '\0' || (ptr+count) > end )
				break;
			if ( unclosed_brace == 0 ) {
				succeed = 1;
				break;
			}
		}

        	if ( !succeed || ptr[count] != '}' )
			return LDAP_FILTER_ERROR;

		in->bv_val = ptr+1; /*the next to '"'*/
		in->bv_len = count - 1; /* exclude '"' */
	}
        else {
                /*Find  following white space where the value is ended*/
                for( count = 1 ; ; count++ ) {
                        if ( ptr[count] == '\0' || ptr[count] == ' ' ||
					(ptr+count) >end ) {
                                break;
                        }
                }
        	if ( ptr[count] != ' ' )
			return LDAP_FILTER_ERROR;

		in->bv_val = ptr; /*the next to '"'*/
		in->bv_len = count; /* exclude '"' */
        }

        return LDAP_SUCCESS;
}

/*
 * Perform matching one referenced component against assertion
 * If the matching rule in a component filter is allComponentsMatch
 * or its derivatives the extracted component's ASN.1 specification
 * is applied to the assertion value as its syntax
 * Otherwise, the matching rule's syntax is applied to the assertion value
 * By RFC 3687
 */
int
comp_test_one_component (
	void* mem_op,
	ComponentSyntaxInfo *csi_attr,
	ComponentAssertion *ca )
{
	int len, rc;
	ComponentSyntaxInfo *csi_assert = NULL;
	char* oid = NULL;
	MatchingRule* mr = ca->ca_ma_rule;

	if ( mr->smr_usage & SLAP_MR_COMPONENT ) {
		/* If allComponentsMatch or its derivatives */
		if ( !ca->ca_comp_data.cd_tree ) {
			comp_convert_assert_to_comp( mem_op, csi_attr, &ca->ca_ma_value, &csi_assert, &len, DEC_ALLOC_MODE_0 );
			ca->ca_comp_data.cd_tree = (void*)csi_assert;
		} else {
			csi_assert = ca->ca_comp_data.cd_tree;
		}

		if ( !csi_assert )
			return LDAP_PROTOCOL_ERROR;

		if ( strcmp( mr->smr_mrule.mr_oid, OID_ALL_COMP_MATCH ) != 0 )
                {
                        /* allComponentMatch's derivatives */
			oid =  mr->smr_mrule.mr_oid;
                }
                        return csi_attr->csi_comp_desc->cd_all_match(
                               			 oid, csi_attr, csi_assert );

	} else {
		/* LDAP existing matching rules */
		struct berval attr_bv = BER_BVNULL;
		struct berval* assert_bv = &ca->ca_ma_value;
		int allocated = 0;
		/*Attribute is converted to compatible LDAP encodings*/
		if ( comp_convert_asn_to_ldap( mr, csi_attr, &attr_bv, &allocated ) != LDAP_SUCCESS )
			return LDAP_INAPPROPRIATE_MATCHING;

		/*Assertion value is validated by MR's syntax*/
		if ( !ca->ca_comp_data.cd_tree ) {
			if ( get_primitive_GSER_value( assert_bv ) != LDAP_SUCCESS ) 
				return LDAP_INVALID_SYNTAX;
			assert_bv->bv_val[assert_bv->bv_len] = '\0';
			if ( mr->smr_syntax->ssyn_validate( mr->smr_syntax, assert_bv ) != LDAP_SUCCESS ) {
				return LDAP_INVALID_SYNTAX;
			}
			ca->ca_comp_data.cd_tree = assert_bv;
		}
		else {
			assert_bv = ca->ca_comp_data.cd_tree;
		}

		rc = csi_value_match( mr, &attr_bv, assert_bv );
		if ( allocated ) free (attr_bv.bv_val);
		return rc;
	}
}

void*
comp_nibble_memory_allocator ( int init_mem, int inc_mem ) {
	void* nm;
	nm = InitNibbleMemLocal( init_mem, inc_mem );
	if ( !nm ) return NULL;
	else return (void*)nm;
}

void
comp_nibble_memory_free ( void* nm ) {
	ShutdownNibbleMemLocal( nm );
}

#if SLAPD_COMP_MATCH == SLAPD_MOD_DYNAMIC

#include "certificate.h"

extern convert_attr_to_comp_func* attr_converter;
extern convert_assert_to_comp_func* assert_converter;
extern convert_asn_to_ldap_func* csi_converter;
extern free_component_func* component_destructor;
extern test_component_func* test_one_component;
extern test_component_func* test_all_components;
extern alloc_nibble_func* nibble_mem_allocator;
extern free_nibble_func* nibble_mem_free;


int init_module(int argc, char *argv[]) {
	/*
	 * Initialize function pointers in slapd
	 */
	attr_converter = comp_convert_attr_to_comp;
	assert_converter = comp_convert_assert_to_comp;
	component_destructor = comp_free_component;
	test_one_component = comp_test_one_component;
	test_all_components = comp_test_all_components;
	nibble_mem_allocator = comp_nibble_memory_allocator;
	nibble_mem_free = comp_nibble_memory_free;

	/* file path needs to be */
	load_derived_matching_rule ("derived_mr.cfg");

	/* the initialization for example X.509 certificate */
	init_module_AuthenticationFramework();
	return 0;
}

#endif /* SLAPD_PASSWD */