/*
* 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 <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
#endif
any->s.in.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
}
/*
* 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_PORT_CHARS
#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;
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;
}
/* 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;
freeaddrinfo(res);
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 any->s.a.sa_family;
}
while(0);
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;
}
}
syntax highlighted by Code2HTML, v. 0.9.1