/*
 * Copyright (c) 2000, Boris Popov
 * 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. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    This product includes software developed by Boris Popov.
 * 4. Neither the name of the author nor the names of any co-contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.
 *
 * $Id: nb_net.c,v 1.8 2004/03/19 01:49:47 lindak Exp $
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <sys/param.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <sys/sockio.h>
#include <net/if.h>
#include <ctype.h>
#include <netdb.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <stdio.h>
#include <unistd.h>

#include <err.h>

#include <netsmb/netbios.h>
#include <netsmb/smb_lib.h>
#include <netsmb/nb_lib.h>

int
nb_getlocalname(char *name, size_t maxlen)
{
	char buf[1024], *cp;

	if (gethostname(buf, sizeof (buf)) != 0)
		return (errno);
	cp = strchr(buf, '.');
	if (cp)
		*cp = 0;
	strlcpy(name, buf, maxlen);
	return (0);
}

int
nb_resolvehost_in(const char *name, struct sockaddr **dest)
{
	struct hostent *h;
	struct sockaddr_in *sinp;
	in_addr_t	addr;
	struct in_addr in;
	int len;
	char **p;


	h = gethostbyname(name);
	if (!h) {
#ifdef DEBUG
		warnx("can't get server address `%s': ", name);
#endif
		return (ENETDOWN);
	}
	if (h->h_addrtype != AF_INET) {
#ifdef DEBUG
		warnx("address for `%s' is not in the AF_INET family", name);
#endif
		return (EAFNOSUPPORT);
	}
	if (h->h_length != 4) {
#ifdef DEBUG
		warnx("address for `%s' has invalid length", name);
#endif
		return (EAFNOSUPPORT);
	}
	len = sizeof (struct sockaddr_in);
	sinp = malloc(len);
	if (sinp == NULL)
		return (ENOMEM);
	bzero(sinp, len);
	/*
	 * There is no sin_len in sockaddr_in structure on Solaris.
	 * sinp->sin_len = len;
	 */
	sinp->sin_family = h->h_addrtype;
	memcpy(&sinp->sin_addr.s_addr, *h->h_addr_list,\
	    sizeof (sinp->sin_addr.s_addr));
	sinp->sin_port = htons(SMB_TCP_PORT);
	*dest = (struct sockaddr *)sinp;
	return (0);
}

#ifdef NOT_DEFINED
int
nb_enum_if(struct nb_ifdesc **iflist) {
	struct lifconf ifc;
	struct lifreq *ifrqp;
	struct nb_ifdesc *ifd;
	struct in_addr iaddr, imask;
	struct lifnum ifn;
	char *ifrdata, *iname;
	int s, rdlen, ifcnt, error, iflags, i;

	*iflist = NULL;
	s = socket(AF_INET, SOCK_DGRAM, 0);
	if (s == -1)
		return (errno);

	/* Get number of interfaces. */
	ifn.lifn_family = AF_INET;
	ifn.lifn_flags = 0;
	ifn.lifn_count = 0;
	if (ioctl(s, SIOCGLIFNUM, &ifn) != 0) {
		error = errno;
		goto bad;
	}

	rdlen = ifn.lifn_count * sizeof (struct lifreq);
	ifrdata = malloc(rdlen);
	if (ifrdata == NULL) {
		error = ENOMEM;
		goto bad;
	}
	ifc.lifc_flags = 0;
	ifc.lifc_family = AF_INET;
	ifc.lifc_len = rdlen;
	ifc.lifc_buf = ifrdata;
	if (ioctl(s, SIOCGLIFCONF, &ifc) != 0) {
		error = errno;
		goto bad;
	}
	ifrqp = ifc.lifc_req;
	ifcnt = ifc.lifc_len / sizeof (struct lifreq);
	error = 0;
	for (i = 0; i < ifcnt; i++, ifrqp++) {
		/* XXX for now, avoid IP6 broadcast performance costs */
		if (ifrqp->lifr_addr.ss_family != AF_INET)
			continue;
		if (ioctl(s, SIOCGLIFFLAGS, ifrqp) != 0)
			continue;
		iflags = ifrqp->lifr_flags;
		if ((iflags & IFF_UP) == 0 || (iflags & IFF_BROADCAST) == 0)
			continue;

		if (ioctl(s, SIOCGLIFADDR, ifrqp) != 0 ||
		    ifrqp->lifr_addr.ss_family != AF_INET) {
			continue;
		}
		iname = ifrqp->lifr_name;
		if (strlen(iname) >= sizeof (ifd->id_name))
			continue;
		iaddr = (*(struct sockaddr_in *)&ifrqp->lifr_addr).sin_addr;

		if (ioctl(s, SIOCGLIFNETMASK, ifrqp) != 0)
			continue;
		imask = ((struct sockaddr_in *)&ifrqp->lifr_addr)->sin_addr;

		ifd = malloc(sizeof (struct nb_ifdesc));
		if (ifd == NULL)
			return (ENOMEM);
		bzero(ifd, sizeof (struct nb_ifdesc));
		strcpy(ifd->id_name, iname);
		ifd->id_flags = iflags;
		ifd->id_addr = iaddr;
		ifd->id_mask = imask;
		ifd->id_next = *iflist;
		*iflist = ifd;
	}
bad:
	free(ifrdata);
	close(s);
	return (error);
}

/*ARGSUSED*/
int
nbns_resolvename(const char *name, struct sockaddr **dest)
{
	printf("NetBIOS name resolver is not included in this distribution.\n");
	printf("Please use '-I' option to specify an IP address of server.\n");
	return (EHOSTUNREACH);
}

int
nb_hostlookup(struct nb_name *np, const char *server, const char *hint,
	struct sockaddr_nb **dst)
{
	struct sockaddr_nb *snb;
	int error;

	error = nb_sockaddr(NULL, np, &snb);
	if (error)
		return (error);
	do {
		if (hint) {
			error = nb_resolvehost_in(host, snb);
			if (error)
				break;
		} else {
			error = nb_resolvename(server);
		}
	} while (0);
	if (!error) {
		*dst = snb;
	} else
		nb_snbfree(snb);
	return (error);
}
#endif