/*
 * $OpenBSD: search.c,v 1.1 2020/09/12 15:06:12 martijn Exp $
 * Copyright (c) 2002 Institute for Open Systems Technology Australia (IFOST)
 * Copyright (c) 2007 Michael Erdely <merdely@openbsd.org>
 * Copyright (c) 2019 Martijn van Duren <martijn@openbsd.org>
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
 * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include <netinet/in.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <login_cap.h>
#include <unistd.h>

#include "aldap.h"
#include "login_ldap.h"

int q = 0;

char *
search(struct auth_ctx *ctx, char *base, char *flt, enum scope scp)
{
	struct aldap_message *m;
	int mcode;
	const char *errstr;
	char *userdn = NULL;

	dlog(1, "%d: search (%s, %s)", q, base ? base : "no base", flt);

	if (aldap_search(ctx->ld, base, scp, flt, NULL, 0, 0, ctx->timeout,
	    NULL) == -1) {
		aldap_get_errno(ctx->ld, &errstr);
		dlog(1, "search result: %s", errstr);
		return NULL;
	}

	q++;

	while ((m = aldap_parse(ctx->ld)) != NULL) {
		dlog(1, "%d: msgid %d, type %02x",q, m->msgid, m->message_type);

		if (m->message_type == LDAP_RES_SEARCH_RESULT) {
			mcode = aldap_get_resultcode(m);
			/*
			 * if its not a referral we're done with this message
			 */
			if (mcode != LDAP_SUCCESS) {
				dlog(0, "%d: unhandled search result %x %s",
				    q, mcode, ldap_resultcode(mcode));
				aldap_freemsg(m);
				return NULL;
			}
			aldap_freemsg(m);
			break;
		} else if (m->message_type == LDAP_RES_SEARCH_ENTRY) {
			userdn = aldap_get_dn(m);
			dlog(1, "%d: SEARCH_ENTRY userdn %s", q, userdn ? userdn : "none");
		}
		aldap_freemsg(m);
	}

	dlog(1, "%d: returning userdn = %s", q, userdn ? userdn : "no user dn");
	q--;
	return userdn;
}