/* 
   Unix SMB/CIFS implementation.
   LDAP server SIMPLE LDB implementation
   Copyright (C) Stefan Metzmacher 2004
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "includes.h"

/* TODO: samdb_context is not a pulblic struct */
struct samdb_context {
	struct ldb_context *ldb;
	struct samdb_context **static_ptr;
};


#define ALLOC_CHECK(ptr) do {\
	if (!(ptr)) {\
		return NT_STATUS_NO_MEMORY;\
	}\
} while(0)


/*
   fix the DN removing unneded non-significative spaces
   this function ASSUME the string is talloced
 */
static char *sldb_fix_dn(const char *dn)
{
	char *new_dn;
	int i, j, k;

	/* alloc enough room to host the whole dn as multibyte string */
	new_dn = talloc(dn, strlen(dn) + 1);
	if (!new_dn) {
		DEBUG(0, ("sldb_fix_dn: Out of memory!"));
		return NULL;
	}

	i = j = 0;
	while (dn[i] != '\0') {
		/* it is legal to check for ascii chars in utf-8 as it is
		 * guaranted to never contain ascii chars (up to 0x7F) as part
		 * of a multibyte sequence */

		new_dn[j] = dn[i];

		if (dn[i] == ',' || dn[i] == '=') {
			/* skip spaces after ',' or '=' */
			for (++i; dn[i] == ' '; i++) ;
			j++;
			continue;
		}
		if (dn[i] == ' ') {
			/* check if there's a ',' after these spaces */
			for (k = i; dn[k] == ' '; k++) ;
			if (dn[k] == ',') {
				/* skip spaces */
				i = k;
				continue;
			} else {
				/* fill the dest buffer with the spaces */
				for (; dn[i] == ' '; i++, j++) {
					new_dn[j] = ' ';
				}
				continue;
			}
		}
		i++;
		j++;
	}
	new_dn[j] = '\0';

	return new_dn;
}


static NTSTATUS sldb_Search(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
				     struct ldap_SearchRequest *r)
{
	NTSTATUS status;
	struct ldap_Result *done;
	struct ldap_SearchResEntry *ent;
	struct ldapsrv_reply *ent_r, *done_r;
	int result = 80;
	struct samdb_context *samdb;
	struct ldb_message **res;
	int i, j, y, count;
	struct ldb_context *ldb;
	enum ldb_scope scope = LDB_SCOPE_DEFAULT;
	const char **attrs = NULL;
	const char *basedn;
	const char *errstr;

	samdb = samdb_connect(call);
	ldb = samdb->ldb;
	basedn = sldb_fix_dn(r->basedn);
	if (basedn == NULL) {
		return NT_STATUS_NO_MEMORY;
	}

	DEBUG(10, ("sldb_Search: basedn: [%s]\n", basedn));
	DEBUG(10, ("sldb_Search: filter: [%s]\n", r->filter));

	switch (r->scope) {
		case LDAP_SEARCH_SCOPE_BASE:
			DEBUG(10,("sldb_Search: scope: [BASE]\n"));
			scope = LDB_SCOPE_BASE;
			break;
		case LDAP_SEARCH_SCOPE_SINGLE:
			DEBUG(10,("sldb_Search: scope: [ONE]\n"));
			scope = LDB_SCOPE_ONELEVEL;
			break;
		case LDAP_SEARCH_SCOPE_SUB:
			DEBUG(10,("sldb_Search: scope: [SUB]\n"));
			scope = LDB_SCOPE_SUBTREE;
			break;
	}

	if (r->num_attributes >= 1) {
		attrs = talloc_array_p(samdb, const char *, r->num_attributes+1);
		ALLOC_CHECK(attrs);

		for (i=0; i < r->num_attributes; i++) {
			DEBUG(10,("sldb_Search: attrs: [%s]\n",r->attributes[i]));
			attrs[i] = r->attributes[i];
		}
		attrs[i] = NULL;
	}

	ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
	count = ldb_search(ldb, basedn, scope, r->filter, attrs, &res);

	for (i=0; i < count; i++) {
		ent_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultEntry);
		ALLOC_CHECK(ent_r);

		ent = &ent_r->msg.r.SearchResultEntry;
		ent->dn = talloc_steal(ent_r, res[i]->dn);
		ent->num_attributes = 0;
		ent->attributes = NULL;
		if (res[i]->num_elements == 0) {
			goto queue_reply;
		}
		ent->num_attributes = res[i]->num_elements;
		ent->attributes = talloc_array_p(ent_r, struct ldap_attribute, ent->num_attributes);
		ALLOC_CHECK(ent->attributes);
		for (j=0; j < ent->num_attributes; j++) {
			ent->attributes[j].name = talloc_steal(ent->attributes, res[i]->elements[j].name);
			ent->attributes[j].num_values = 0;
			ent->attributes[j].values = NULL;
			if (r->attributesonly && (res[i]->elements[j].num_values == 0)) {
				continue;
			}
			ent->attributes[j].num_values = res[i]->elements[j].num_values;
			ent->attributes[j].values = talloc_array_p(ent->attributes,
							DATA_BLOB, ent->attributes[j].num_values);
			ALLOC_CHECK(ent->attributes[j].values);
			for (y=0; y < ent->attributes[j].num_values; y++) {
				ent->attributes[j].values[y].length = res[i]->elements[j].values[y].length;
				ent->attributes[j].values[y].data = talloc_steal(ent->attributes[j].values,
									res[i]->elements[j].values[y].data);
			}
		}
queue_reply:
		status = ldapsrv_queue_reply(call, ent_r);
		if (!NT_STATUS_IS_OK(status)) {
			return status;
		}
	}

	done_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultDone);
	ALLOC_CHECK(done_r);

	if (count > 0) {
		DEBUG(10,("sldb_Search: results: [%d]\n",count));
		result = 0;
		errstr = NULL;
	} else if (count == 0) {
		DEBUG(10,("sldb_Search: no results\n"));
		result = 32;
		errstr = talloc_strdup(done_r, ldb_errstring(ldb));
	} else if (count == -1) {
		DEBUG(10,("sldb_Search: error\n"));
		result = 1;
		errstr = talloc_strdup(done_r, ldb_errstring(ldb));
	}

	done = &done_r->msg.r.SearchResultDone;
	done->resultcode = result;
	done->dn = NULL;
	done->errormessage = errstr;
	done->referral = NULL;

	talloc_free(samdb);

	return ldapsrv_queue_reply(call, done_r);
}

static NTSTATUS sldb_Add(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
				     struct ldap_AddRequest *r)
{
	struct ldap_Result *add_result;
	struct ldapsrv_reply *add_reply;
	int ldb_ret;
	struct samdb_context *samdb;
	struct ldb_context *ldb;
	const char *dn;
	struct ldb_message *msg;
	int result = LDAP_SUCCESS;
	const char *errstr = NULL;
	int i,j;

	samdb = samdb_connect(call);
	ldb = samdb->ldb;
	dn = sldb_fix_dn(r->dn);
	if (dn == NULL) {
		return NT_STATUS_NO_MEMORY;
	}

	DEBUG(10, ("sldb_add: dn: [%s]\n", dn));

	msg = talloc_p(samdb, struct ldb_message);
	ALLOC_CHECK(msg);

	msg->dn = discard_const_p(char, dn);
	msg->private_data = NULL;
	msg->num_elements = 0;
	msg->elements = NULL;

	if (r->num_attributes > 0) {
		msg->num_elements = r->num_attributes;
		msg->elements = talloc_array_p(msg, struct ldb_message_element, msg->num_elements);
		ALLOC_CHECK(msg->elements);

		for (i=0; i < msg->num_elements; i++) {
			msg->elements[i].name = discard_const_p(char, r->attributes[i].name);
			msg->elements[i].flags = 0;
			msg->elements[i].num_values = 0;
			msg->elements[i].values = NULL;
			
			if (r->attributes[i].num_values > 0) {
				msg->elements[i].num_values = r->attributes[i].num_values;
				msg->elements[i].values = talloc_array_p(msg, struct ldb_val, msg->elements[i].num_values);
				ALLOC_CHECK(msg->elements[i].values);

				for (j=0; j < msg->elements[i].num_values; j++) {
					if (!(r->attributes[i].values[j].length > 0)) {
						result = 80;
						goto invalid_input;
					}
					msg->elements[i].values[j].length = r->attributes[i].values[j].length;
					msg->elements[i].values[j].data = r->attributes[i].values[j].data;			
				}
			} else {
				result = 80;
				goto invalid_input;
			}
		}
	} else {
		result = 80;
		goto invalid_input;
	}

invalid_input:

	add_reply = ldapsrv_init_reply(call, LDAP_TAG_AddResponse);
	ALLOC_CHECK(add_reply);

	add_result = &add_reply->msg.r.AddResponse;
	add_result->dn = talloc_steal(add_reply, dn);

	if (result == LDAP_SUCCESS) {
		ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
		ldb_ret = ldb_add(ldb, msg);
		if (ldb_ret == 0) {
			result = LDAP_SUCCESS;
			errstr = NULL;
		} else {
			/* currently we have no way to tell if there was an internal ldb error
		 	 * or if the object was not found, return the most probable error
		 	 */
			result = 1;
			errstr = talloc_strdup(add_reply, ldb_errstring(ldb));
		}
	} else {
		errstr = talloc_strdup(add_reply,"invalid input data");
	}

	add_result->resultcode = result;
	add_result->errormessage = errstr;
	add_result->referral = NULL;

	talloc_free(samdb);

	return ldapsrv_queue_reply(call, add_reply);
}

static NTSTATUS sldb_Del(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
				     struct ldap_DelRequest *r)
{
	struct ldap_Result *del_result;
	struct ldapsrv_reply *del_reply;
	int ldb_ret;
	struct samdb_context *samdb;
	struct ldb_context *ldb;
	const char *dn;
	const char *errstr = NULL;
	int result = LDAP_SUCCESS;

	samdb = samdb_connect(call);
	ldb = samdb->ldb;
	dn = sldb_fix_dn(r->dn);
	if (dn == NULL) {
		return NT_STATUS_NO_MEMORY;
	}

	DEBUG(10, ("sldb_Del: dn: [%s]\n", dn));

	ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
	ldb_ret = ldb_delete(ldb, dn);

	del_reply = ldapsrv_init_reply(call, LDAP_TAG_DelResponse);
	ALLOC_CHECK(del_reply);

	del_result = &del_reply->msg.r.DelResponse;
	del_result->dn = talloc_steal(del_reply, dn);

	if (ldb_ret == 0) {
		result = LDAP_SUCCESS;
		errstr = NULL;
	} else {
		/* currently we have no way to tell if there was an internal ldb error
		 * or if the object was not found, return the most probable error
		 */
		result = LDAP_NO_SUCH_OBJECT;
		errstr = talloc_strdup(del_reply, ldb_errstring(ldb));
	}

	del_result->resultcode = result;
	del_result->errormessage = errstr;
	del_result->referral = NULL;

	talloc_free(samdb);

	return ldapsrv_queue_reply(call, del_reply);
}

static NTSTATUS sldb_Modify(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
				     struct ldap_ModifyRequest *r)
{
	struct ldap_Result *modify_result;
	struct ldapsrv_reply *modify_reply;
	int ldb_ret;
	struct samdb_context *samdb;
	struct ldb_context *ldb;
	const char *dn;
	struct ldb_message *msg;
	int result = LDAP_SUCCESS;
	const char *errstr = NULL;
	int i,j;

	samdb = samdb_connect(call);
	ldb = samdb->ldb;
	dn = sldb_fix_dn(r->dn);
	if (dn == NULL) {
		return NT_STATUS_NO_MEMORY;
	}

	DEBUG(10, ("sldb_modify: dn: [%s]\n", dn));

	msg = talloc_p(samdb, struct ldb_message);
	ALLOC_CHECK(msg);

	msg->dn = discard_const_p(char, dn);
	msg->private_data = NULL;
	msg->num_elements = 0;
	msg->elements = NULL;

	if (r->num_mods > 0) {
		msg->num_elements = r->num_mods;
		msg->elements = talloc_array_p(msg, struct ldb_message_element, r->num_mods);
		ALLOC_CHECK(msg->elements);

		for (i=0; i < msg->num_elements; i++) {
			msg->elements[i].name = discard_const_p(char, r->mods[i].attrib.name);
			msg->elements[i].num_values = 0;
			msg->elements[i].values = NULL;

			switch (r->mods[i].type) {
			default:
				result = 2;
				goto invalid_input;
			case LDAP_MODIFY_ADD:
				msg->elements[i].flags = LDB_FLAG_MOD_ADD;
				break;
			case LDAP_MODIFY_DELETE:
				msg->elements[i].flags = LDB_FLAG_MOD_DELETE;
				break;
			case LDAP_MODIFY_REPLACE:
				msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
				break;
			}

			if (r->mods[i].attrib.num_values > 0) {
				msg->elements[i].num_values = r->mods[i].attrib.num_values;
				msg->elements[i].values = talloc_array_p(msg, struct ldb_val, msg->elements[i].num_values);
				ALLOC_CHECK(msg->elements[i].values);

				for (j=0; j < msg->elements[i].num_values; j++) {
					if (!(r->mods[i].attrib.values[j].length > 0)) {
						result = 80;
						goto invalid_input;
					}
					msg->elements[i].values[j].length = r->mods[i].attrib.values[j].length;
					msg->elements[i].values[j].data = r->mods[i].attrib.values[j].data;			
				}
			} else {
				/* TODO: test what we should do here 
				 *
				 *       LDAP_MODIFY_DELETE is ok to pass here
				 */
			}
		}
	} else {
		result = 80;
		goto invalid_input;
	}

invalid_input:

	modify_reply = ldapsrv_init_reply(call, LDAP_TAG_ModifyResponse);
	ALLOC_CHECK(modify_reply);

	modify_result = &modify_reply->msg.r.AddResponse;
	modify_result->dn = talloc_steal(modify_reply, dn);

	if (result == LDAP_SUCCESS) {
		ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
		ldb_ret = ldb_modify(ldb, msg);
		if (ldb_ret == 0) {
			result = LDAP_SUCCESS;
			errstr = NULL;
		} else {
			/* currently we have no way to tell if there was an internal ldb error
		 	 * or if the object was not found, return the most probable error
		 	 */
			result = 1;
			errstr = talloc_strdup(modify_reply, ldb_errstring(ldb));
		}
	} else {
		errstr = talloc_strdup(modify_reply,"invalid input data");
	}

	modify_result->resultcode = result;
	modify_result->errormessage = errstr;
	modify_result->referral = NULL;

	talloc_free(samdb);

	return ldapsrv_queue_reply(call, modify_reply);
}

static NTSTATUS sldb_Compare(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
				     struct ldap_CompareRequest *r)
{
	struct ldap_Result *compare;
	struct ldapsrv_reply *compare_r;
	int result = 80;
	struct samdb_context *samdb;
	struct ldb_message **res;
	struct ldb_context *ldb;
	const char *attrs[1];
	const char *errstr;
	const char *dn;
	const char *filter;
	int count;

	samdb = samdb_connect(call);
	ldb = samdb->ldb;
	dn = sldb_fix_dn(r->dn);
	if (dn == NULL) {
		return NT_STATUS_NO_MEMORY;
	}

	DEBUG(10, ("sldb_Compare: dn: [%s]\n", dn));
	filter = talloc_asprintf(samdb, "(%s=%*s)", r->attribute, r->value.length, r->value.data);
	ALLOC_CHECK(filter);
	DEBUGADD(10, ("sldb_Compare: attribute: [%s]\n", filter));

	attrs[0] = NULL;

	ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
	count = ldb_search(ldb, dn, LDB_SCOPE_BASE, filter, attrs, &res);

	compare_r = ldapsrv_init_reply(call, LDAP_TAG_CompareResponse);
	ALLOC_CHECK(compare_r);

	if (count == 1) {
		DEBUG(10,("sldb_Compare: matched\n"));
		result = 0;
		errstr = NULL;
	} else if (count == 0) {
		result = 32;
		errstr = talloc_strdup(compare_r, ldb_errstring(ldb));
		DEBUG(10,("sldb_Compare: no results: %s\n", errstr));
	} else if (count > 1) {
		result = 80;
		errstr = talloc_strdup(compare_r, "too many objects match");
		DEBUG(10,("sldb_Compare: %d results: %s\n", count, errstr));
	} else if (count == -1) {
		result = 1;
		errstr = talloc_strdup(compare_r, ldb_errstring(ldb));
		DEBUG(10,("sldb_Compare: error: %s\n", errstr));
	}

	compare = &compare_r->msg.r.CompareResponse;
	compare->resultcode = result;
	compare->dn = NULL;
	compare->errormessage = errstr;
	compare->referral = NULL;

	talloc_free(samdb);

	return ldapsrv_queue_reply(call, compare_r);
}

static const struct ldapsrv_partition_ops sldb_ops = {
	.Search		= sldb_Search,
	.Add		= sldb_Add,
	.Del		= sldb_Del,
	.Modify		= sldb_Modify,
	.Compare	= sldb_Compare
};

const struct ldapsrv_partition_ops *ldapsrv_get_sldb_partition_ops(void)
{
	return &sldb_ops;
}