/*
 * 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 <ctype.h>
#include <netdb.h>
#include <stdlib.h>
#include <string.h>
#include <nss_dbdefs.h>

int str2protoent(const char *, int, void *,
		char *, int);

static int proto_stayopen;
/*
 * Unsynchronized, but it affects only
 * efficiency, not correctness
 */

static DEFINE_NSS_DB_ROOT(db_root);
static DEFINE_NSS_GETENT(context);

void
_nss_initf_proto(nss_db_params_t *p)
{
	p->name	= NSS_DBNAM_PROTOCOLS;
	p->default_config = NSS_DEFCONF_PROTOCOLS;
}

struct protoent *
getprotobyname_r(const char *name, struct protoent *result,
	char *buffer, int buflen)
{
	nss_XbyY_args_t arg;
	nss_status_t	res;

	if (name == (const char *)NULL) {
		errno = ERANGE;
		return (NULL);
	}
	NSS_XbyY_INIT(&arg, result, buffer, buflen, str2protoent);
	arg.key.name	= name;
	arg.stayopen	= proto_stayopen;
	res = nss_search(&db_root, _nss_initf_proto,
		NSS_DBOP_PROTOCOLS_BYNAME, &arg);
	arg.status = res;
	(void) NSS_XbyY_FINI(&arg);
	return ((struct protoent *)arg.returnval);
}

struct protoent *
getprotobynumber_r(int proto, struct protoent *result, char *buffer, int buflen)
{
	nss_XbyY_args_t arg;
	nss_status_t	res;

	NSS_XbyY_INIT(&arg, result, buffer, buflen, str2protoent);
	arg.key.number = proto;
	arg.stayopen	= proto_stayopen;
	res = nss_search(&db_root, _nss_initf_proto,
		NSS_DBOP_PROTOCOLS_BYNUMBER, &arg);
	arg.status = res;
	(void) NSS_XbyY_FINI(&arg);
	return ((struct protoent *)arg.returnval);
}

int
setprotoent(int stay)
{
	proto_stayopen = stay;
	nss_setent(&db_root, _nss_initf_proto, &context);
	return (0);
}

int
endprotoent()
{
	proto_stayopen = 0;
	nss_endent(&db_root, _nss_initf_proto, &context);
	nss_delete(&db_root);
	return (0);
}

struct protoent *
getprotoent_r(struct protoent *result, char *buffer, int buflen)
{
	nss_XbyY_args_t arg;
	nss_status_t	res;

	NSS_XbyY_INIT(&arg, result, buffer, buflen, str2protoent);
	/* No stayopen flag;  of course you stay open for iteration */
	res = nss_getent(&db_root, _nss_initf_proto, &context, &arg);
	arg.status = res;
	(void) NSS_XbyY_FINI(&arg);
	return ((struct protoent *)arg.returnval);
}

/*
 * Return values: 0 = success, 1 = parse error, 2 = erange ...
 * The structure pointer passed in is a structure in the caller's space
 * wherein the field pointers would be set to areas in the buffer if
 * need be. instring and buffer should be separate areas. Let's not
 * fight over crumbs.
 */
int
str2protoent(const char *instr, int lenstr,
	void *ent /* it is really (struct protoent *) */,
	char *buffer, int buflen)
{
	struct	protoent *proto = (struct protoent *)ent;
	const char	*p, *numstart, *namestart, *limit;
	int		numlen, namelen = 0;
	char		numbuf[16];
	char		*numend;

	if ((instr >= buffer && (buffer + buflen) > instr) ||
	    (buffer >= instr && (instr + lenstr) > buffer)) {
		return (NSS_STR_PARSE_PARSE);
	}

	p = instr;
	limit = p + lenstr;

	while (p < limit && isspace(*p)) {
		p++;
	}
	namestart = p;
	while (p < limit && !isspace(*p)) {
		p++;		/* Skip over the canonical name */
	}
	namelen = (int)(p - namestart);

	if (buflen <= namelen) { /* not enough buffer */
		return (NSS_STR_PARSE_ERANGE);
	}
	(void) memcpy(buffer, namestart, namelen);
	buffer[namelen] = '\0';
	proto->p_name = buffer;

	while (p < limit && isspace(*p)) {
		p++;
	}
	if (p >= limit) {
		/* Syntax error -- no proto number */
		return (NSS_STR_PARSE_PARSE);
	}
	numstart = p;
	do {
		p++;		/* Find the end of the proto number */
	} while (p < limit && !isspace(*p));
	numlen = (int)(p - numstart);
	if (numlen >= (int)sizeof (numbuf)) {
		/* Syntax error -- supposed number is too long */
		return (NSS_STR_PARSE_PARSE);
	}
	(void) memcpy(numbuf, numstart, (size_t)numlen);
	numbuf[numlen] = '\0';
	proto->p_proto = (int)strtol(numbuf, &numend, 10);
	if (*numend != '\0') {
		/* Syntax error -- protocol number isn't a number */
		return (NSS_STR_PARSE_PARSE);
	}

	while (p < limit && isspace(*p)) {
		p++;
	}
	/*
	 * Although nss_files_XY_all calls us with # stripped,
	 * we should be able to deal with it here in order to
	 * be more useful.
	 */
	if (p >= limit || *p == '#') { /* no aliases, no problem */
		char **ptr;

		ptr = (char **)ROUND_UP(buffer + namelen + 1,
							sizeof (char *));
		if ((char *)ptr >= buffer + buflen) {
			/* hope they don't try to peek in */
			proto->p_aliases = 0;
			return (NSS_STR_PARSE_ERANGE);
		} else {
			*ptr = 0;
			proto->p_aliases = ptr;
			return (NSS_STR_PARSE_SUCCESS);
		}
	}
	proto->p_aliases = _nss_netdb_aliases(p, lenstr - (int)(p - instr),
				buffer + namelen + 1, buflen - namelen - 1);
	return (NSS_STR_PARSE_SUCCESS);
}