/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <netdb.h>
#include "ldap_common.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/tsol/tndb.h>

/* tnrhdb attributes filters */
#define	_TNRHDB_ADDR		"ipTnetNumber"
#define	_TNRHDB_TNAME		"ipTnetTemplateName"
#define	_F_GETTNDBBYADDR	"(&(objectClass=ipTnetHost)(ipTnetNumber=%s))"
#define	_F_GETTNDBBYADDR_SSD	"(&(%%s)(ipTnetNumber=%s))"

static const char *tnrhdb_attrs[] = {
	_TNRHDB_ADDR,
	_TNRHDB_TNAME,
	NULL
};

static void
escape_colon(char *in, char *out) {
	int i, j;
	for (i = 0, j = 0; in[i] != '\0'; i++) {
		if (in[i] == ':') {
			out[j++] = '\\';
			out[j++] = in[i];
		} else
			out[j++] = in[i];
	}
	out[j] = '\0';
}

/*
 * _nss_ldap_tnrhdb2str is the data marshaling method for the tnrhdb
 * (tsol_getrhbyaddr()/tsol_getrhent()) backend processes.
 * This method is called after a successful ldap search has been performed.
 * This method will parse the ldap search values into the file format.
 *
 * e.g.
 *
 * 192.168.120.6:public
 * fec0\:\:a00\:20ff\:fea0\:21f7:cipso
 *
 */
static int
_nss_ldap_tnrhdb2str(ldap_backend_ptr be, nss_XbyY_args_t *argp)
{
	int			nss_result = NSS_STR_PARSE_SUCCESS;
	int			len = 0;
	char			*buffer = NULL;
	char			**addr, **template, *addr_out;
	ns_ldap_result_t	*result = be->result;
	char addr6[INET6_ADDRSTRLEN + 5]; /* 5 '\' for ':' at most */

	if (result == NULL)
		return (NSS_STR_PARSE_PARSE);

	addr = __ns_ldap_getAttr(result->entry, _TNRHDB_ADDR);
	if (addr == NULL || addr[0] == NULL || (strlen(addr[0]) < 1)) {
		nss_result = NSS_STR_PARSE_PARSE;
		goto result_tnrhdb2str;
	}

	/*
	 * Escape ':' in IPV6.
	 * The value is stored in LDAP directory without escape charaters.
	 */
	if (strchr(addr[0], ':') != NULL) {
		escape_colon(addr[0], addr6);
		addr_out = addr6;
	} else
		addr_out = addr[0];

	template = __ns_ldap_getAttr(result->entry, _TNRHDB_TNAME);
	if (template == NULL || template[0] == NULL ||
			(strlen(template[0]) < 1)) {
		nss_result = NSS_STR_PARSE_PARSE;
		goto result_tnrhdb2str;
	}
	/* "addr:template" */
	len = strlen(addr_out) + strlen(template[0]) + 2;

	if (argp->buf.result != NULL) {
		if ((be->buffer = calloc(1, len)) == NULL) {
			nss_result = NSS_STR_PARSE_PARSE;
			goto result_tnrhdb2str;
		}
		be->buflen = len - 1;
		buffer = be->buffer;
	} else
		buffer = argp->buf.buffer;

	(void) snprintf(buffer, len, "%s:%s", addr_out, template[0]);

result_tnrhdb2str:
	(void) __ns_ldap_freeResult(&be->result);
	return (nss_result);
}


static nss_status_t
getbyaddr(ldap_backend_ptr be, void *a)
{
	char		searchfilter[SEARCHFILTERLEN];
	char		userdata[SEARCHFILTERLEN];
	nss_XbyY_args_t	*argp = (nss_XbyY_args_t *)a;

	if (argp->key.hostaddr.addr == NULL ||
		(argp->key.hostaddr.type != AF_INET &&
		argp->key.hostaddr.type != AF_INET6))
			return (NSS_NOTFOUND);
	if (strchr(argp->key.hostaddr.addr, ':') != NULL) {
		/* IPV6 */
		if (argp->key.hostaddr.type == AF_INET)
			return (NSS_NOTFOUND);
	} else {
		/* IPV4 */
		if (argp->key.hostaddr.type == AF_INET6)
			return (NSS_NOTFOUND);
	}

	/*
	 * The IPV6 addresses are saved in the directory without '\'s.
	 * So don't need to escape colons in IPV6 addresses.
	 */
	if (snprintf(searchfilter, sizeof (searchfilter), _F_GETTNDBBYADDR,
	    argp->key.hostaddr.addr) < 0)
		return ((nss_status_t)NSS_NOTFOUND);

	if (snprintf(userdata, sizeof (userdata), _F_GETTNDBBYADDR_SSD,
	    argp->key.hostaddr.addr) < 0)
		return ((nss_status_t)NSS_NOTFOUND);

	return (_nss_ldap_lookup(be, argp, _TNRHDB, searchfilter, NULL,
	    _merge_SSD_filter, userdata));
}


static ldap_backend_op_t tnrhdb_ops[] = {
	_nss_ldap_destr,
	_nss_ldap_endent,
	_nss_ldap_setent,
	_nss_ldap_getent,
	getbyaddr
};


/* ARGSUSED */
nss_backend_t *
_nss_ldap_tnrhdb_constr(const char *dummy1,
    const char *dummy2,
    const char *dummy3,
    const char *dummy4,
    const char *dummy5)
{
	return ((nss_backend_t *)_nss_ldap_constr(tnrhdb_ops,
		sizeof (tnrhdb_ops)/sizeof (tnrhdb_ops[0]), _TNRHDB,
		tnrhdb_attrs, _nss_ldap_tnrhdb2str));
}