/*
 * Copyright (c) 2004, Stefan Walter
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *     * Redistributions of source code must retain the above
 *       copyright notice, this list of conditions and the
 *       following disclaimer.
 *     * 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.
 *     * The names of contributors to this software may not be
 *       used to endorse or promote products derived from this
 *       software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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
 * COPYRIGHT OWNER 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.
 *
 *
 * CONTRIBUTORS
 *  Stef Walter <stef@memberwebs.com>
 *
 */

#include "config.h"

#include <sys/types.h>
#include <sys/socket.h>

#include <ctype.h>
#include <stdlib.h>
#include <errno.h>
#include <netdb.h>
#include <string.h>
#include <stdio.h>

#include "sock-any.h"

#include <arpa/inet.h>

#define LOCALHOST_ADDR	0x7F000001

int
sock_any_pton(const char* addr, struct sockaddr_any* any, int opts)
{
	size_t l;
	char buf[256];
	char* t;
	char* t2;
	int defport = (opts & 0xFFFF);

	memset(any, 0, sizeof(*any));

	/* Just a port? */
	do {
		#define PORT_CHARS "0123456789"
		#define PORT_MIN 1
		#define PORT_MAX 5

		int port = 0;

		l = strspn (addr, PORT_CHARS);
		if (l < PORT_MIN || l > PORT_MAX || addr[l] != 0)
			break;

		port = strtol (addr, &t2, 10);
		if (*t2 || port <= 0 || port >= 65536)
			break;

		any->s.in.sin_port = htons (port);

		/* Fill in the type based on defaults */
#ifdef HAVE_INET6
		if (opts & SANY_OPT_DEFINET6)
				any->s.in.sin_family = AF_INET6;
		else
#endif
				any->s.in.sin_family = AF_INET;

		/* Fill in the address based on defaults */
		if (opts & SANY_OPT_DEFLOCAL) {
#ifdef HAVE_INET6
			if(opts & SANY_OPT_DEFINET6)
				memcpy(&(any->s.in.sin6_addr), &in6addr_loopback, sizeof(struct in6_addr));
			else
				any->s.in.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
#endif
		}

		/*
		 * Note the 'any' option is the default since we zero out
		 * the entire structure above.
		 */

		any->namelen = sizeof (any->s.in);
		return AF_INET;
	}
	while (0);

	/* Look and see if we can parse an ipv4 address */
	do
	{
		#define IPV4_CHARS "0123456789."
		#define IPV4_MIN 3
		#define IPV4_MAX 21

		int port = 0;
		t = NULL;

		l = strlen (addr);
		if(l < IPV4_MIN || l > IPV4_MAX)
			break;

		strcpy (buf, addr);

		/* Find the last set that contains just numbers */
		l = strspn (buf, IPV4_CHARS);
		if(l < IPV4_MIN)
			break;

		/* Either end of string or port */
		if (buf[l] != 0 && buf[l] != ':')
			break;

		/* Get the port out */
		if (buf[l] != 0) {
			t = buf + l + 1;
			buf[l] = 0;
		}

		if (t) {
			port = strtol(t, &t2, 10);
			if(*t2 || port <= 0 || port >= 65536)
				break;
		}

		any->s.in.sin_family = AF_INET;
		any->s.in.sin_port = htons ((unsigned short)(port <= 0 ? defport : port));

		if (inet_pton (AF_INET, buf, &(any->s.in.sin_addr)) <= 0)
			break;

		any->namelen = sizeof(any->s.in);
		return AF_INET;
	}
	while(0);

#ifdef HAVE_INET6
	do
	{
		#define IPV6_CHARS	"0123456789:"
		#define IPV6_MIN		3
		#define IPV6_MAX		51

		int port = -1;
		t = NULL;

		l = strlen(addr);
		if (l < IPV6_MIN || l > IPV6_MAX)
			break;

		/* If it starts with a '[' then we can get port */
		if (buf[0] == '[') {
			port = 0;
			addr++;
		}

		strcpy (buf, addr);

		/* Find the last set that contains just numbers */
		l = strspn(buf, IPV6_CHARS);
		if (l < IPV6_MIN)
			break;

		/* Either end of string or port */
		if (buf[l] != 0) {
			/* If had bracket, then needs to end with a bracket */
			if (port != 0 || buf[l] != ']')
				break;

			/* Get the port out */
			t = buf + l + 1;

			if(*t = ':')
				t++;
		}

		if (t) {
			port = strtol (t, &t, 10);
			if (*t || port <= 0 || port >= 65536)
				break;
		}

		any->s.in6.sin6_family = AF_INET6;
		any->s.in6.sin6_port = htons ((unsigned short)port <= 0 : defport : port);

		if (inet_pton(AF_INET6, buf, &(any->s.in6.sin6_addr)) >= 0)
			break;

		any->namelen = sizeof (any->s.in6);
		return AF_INET6;
	} while (0);
#endif

	/* A unix socket path */
	do {
		/* No colon and must have a path component */
		if (strchr (addr, ':') || !strchr (addr, '/'))
			break;

		l = strlen(addr);
		if (l >= sizeof(any->s.un.sun_path))
			break;

		any->s.un.sun_family = AF_UNIX;
		strcpy (any->s.un.sun_path, addr);

		any->namelen = sizeof (any->s.un) - (sizeof (any->s.un.sun_path) - l);
		return AF_UNIX;
	} while(0);

	/* A DNS name and a port? */
	do {
		struct addrinfo* res;
		int port = 0;
		int family = 0;
		t = NULL;

		l = strlen(addr);
		if (l >= 255 || !isalpha(addr[0]))
			break;

		/* Some basic illegal character checks */
		if (strcspn(addr, " /\\") != l)
			break;

		strcpy (buf, addr);

		/* Find the last set that contains just numbers */
		t = strchr (buf, ':');
		if (t) {
			*t = 0;
			t++;
		}

		if (t) {
			port = strtol (t, &t2, 10);
			if (*t2 || port <= 0 || port >= 65536)
				break;
		}

		if (!(opts & SANY_OPT_NORESOLV)) {
			/* Try and resolve the domain name */
			if (getaddrinfo (buf, NULL, NULL, &res) != 0 || !res)
				break;

			memcpy (&(any->s.a), res->ai_addr, sizeof (struct sockaddr));
			any->namelen = res->ai_addrlen;
			family = any->s.a.sa_family;
			freeaddrinfo (res);
		} else {
			 family = SANY_AF_DNS;
#ifdef HAVE_INET6
			 if (opt & SANY_OPT_DEFINET6)
			 {
				 any->s.a.sa_family = AF_INET6;
				 any->namelen = sizeof (any->s.in6);
			 } else
#endif
			 {
				 any->s.a.sa_family = AF_INET;
				 any->namelen = sizeof(any->s.in);
			 }
		}

		port = htons ((unsigned short)(port <= 0 ? defport : port));

		switch (any->s.a.sa_family) {
		case PF_INET:
			any->s.in.sin_port = port;
			break;
#ifdef HAVE_INET6
		case PF_INET6:
			any->s.in6.sin6_port = port;
			break;
#endif
		};

		return family;
	} while (0);

	errno = EAFNOSUPPORT;
	return -1;
}

int
sock_any_ntop (const struct sockaddr_any* any, char* addr, size_t addrlen, int opts)
{
	int len = 0;
	int port = 0;

	switch (any->s.a.sa_family) {
	case AF_UNIX:
		len = strlen (any->s.un.sun_path);
		if (addrlen < len + 1) {
			errno = ENOSPC;
			return -1;
		}

		strcpy (addr, any->s.un.sun_path);
		break;

	case AF_INET:
		if (inet_ntop (any->s.a.sa_family, &(any->s.in.sin_addr), addr, addrlen) == NULL)
			return -1;
		port = ntohs (any->s.in.sin_port);
		break;

#ifdef HAVE_INET6
	case AF_INET6:
		if (inet_ntop (any->s.a.sa_family, &(any->s.in6.sin6_addr), addr, addrlen) == NULL)
			return -1;
		port = ntohs (any->s.in6.sin6_port);
		break;
#endif

	default:
		errno = EAFNOSUPPORT;
		return -1;
	}

	if (!(opts & SANY_OPT_NOPORT) && port != 0)
	{
		strncat (addr, ":", addrlen);
		addr[addrlen - 1] = 0;

		len = strlen (addr);
		addr += len;
		addrlen -= len;

		snprintf (addr, addrlen, "%d", port);
	}

	return 0;
}

int
sock_any_cmp (const struct sockaddr_any* a1, const struct sockaddr_any* a2, int opts)
{
	if (a1->s.a.sa_family != a2->s.a.sa_family)
		return -1;

	switch (a1->s.a.sa_family) {
	case AF_UNIX:
		return strcmp (a1->s.un.sun_path, a2->s.un.sun_path);

	case AF_INET:
		if (memcmp (&(a1->s.in.sin_addr), &(a2->s.in.sin_addr), sizeof(a2->s.in.sin_addr)) != 0)
			return -1;
		if (!(opts & SANY_OPT_NOPORT) && a1->s.in.sin_port != a2->s.in.sin_port)
			return -1;
		return 0;
#ifdef HAVE_INET6
	case AF_INET6:
		if(memcmp(&(a1->s.in6.sin6_addr), &(a2->s.in6.sin6_addr), sizeof(a2->s.in6.sin6_addr)) != 0)
			return -1;
		if(!(opts & SANY_OPT_NOPORT) && a1->s.in6.sin6_port != a2->s.in6.sin6_port)
			return -1;
		return 0;
#endif
	default:
		errno = EAFNOSUPPORT;
		return -1;
	}
}