1
0
This repository has been archived on 2024-07-22. You can view files and clone it, but cannot push or open issues or pull requests.
TP-Link_Archer-XR500v/EN7526G_3.18Kernel_SDK/apps/public/zebra-0.93a/lib/sockunion.c
2024-07-22 01:58:46 -03:00

757 lines
16 KiB
C
Executable File

/* Socket union related function.
* Copyright (c) 1997, 98 Kunihiro Ishiguro
*
* This file is part of GNU Zebra.
*
* GNU Zebra is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2, or (at your option) any
* later version.
*
* GNU Zebra is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Zebra; see the file COPYING. If not, write to the Free
* Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
* 02111-1307, USA.
*/
#include <zebra.h>
#include "prefix.h"
#include "vty.h"
#include "sockunion.h"
#include "memory.h"
#include "str.h"
#include "log.h"
#ifndef HAVE_INET_ATON
int
inet_aton (const char *cp, struct in_addr *inaddr)
{
int dots = 0;
register u_long addr = 0;
register u_long val = 0, base = 10;
do
{
register char c = *cp;
switch (c)
{
case '0': case '1': case '2': case '3': case '4': case '5':
case '6': case '7': case '8': case '9':
val = (val * base) + (c - '0');
break;
case '.':
if (++dots > 3)
return 0;
case '\0':
if (val > 255)
return 0;
addr = addr << 8 | val;
val = 0;
break;
default:
return 0;
}
} while (*cp++) ;
if (dots < 3)
addr <<= 8 * (3 - dots);
if (inaddr)
inaddr->s_addr = htonl (addr);
return 1;
}
#endif /* ! HAVE_INET_ATON */
#ifndef HAVE_INET_PTON
int
inet_pton (int family, const char *strptr, void *addrptr)
{
if (family == AF_INET)
{
struct in_addr in_val;
if (inet_aton (strptr, &in_val))
{
memcpy (addrptr, &in_val, sizeof (struct in_addr));
return 1;
}
return 0;
}
errno = EAFNOSUPPORT;
return -1;
}
#endif /* ! HAVE_INET_PTON */
#ifndef HAVE_INET_NTOP
const char *
inet_ntop (int family, const void *addrptr, char *strptr, size_t len)
{
unsigned char *p = (unsigned char *) addrptr;
if (family == AF_INET)
{
char temp[INET_ADDRSTRLEN];
snprintf(temp, sizeof(temp), "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
if (strlen(temp) >= len)
{
errno = ENOSPC;
return NULL;
}
strcpy(strptr, temp);
return strptr;
}
errno = EAFNOSUPPORT;
return NULL;
}
#endif /* ! HAVE_INET_NTOP */
const char *
inet_sutop (union sockunion *su, char *str)
{
switch (su->sa.sa_family)
{
case AF_INET:
inet_ntop (AF_INET, &su->sin.sin_addr, str, INET_ADDRSTRLEN);
break;
#ifdef HAVE_IPV6
case AF_INET6:
inet_ntop (AF_INET6, &su->sin6.sin6_addr, str, INET6_ADDRSTRLEN);
break;
#endif /* HAVE_IPV6 */
}
return str;
}
int
str2sockunion (char *str, union sockunion *su)
{
int ret;
memset (su, 0, sizeof (union sockunion));
ret = inet_pton (AF_INET, str, &su->sin.sin_addr);
if (ret > 0) /* Valid IPv4 address format. */
{
su->sin.sin_family = AF_INET;
#ifdef HAVE_SIN_LEN
su->sin.sin_len = sizeof(struct sockaddr_in);
#endif /* HAVE_SIN_LEN */
return 0;
}
#ifdef HAVE_IPV6
ret = inet_pton (AF_INET6, str, &su->sin6.sin6_addr);
if (ret > 0) /* Valid IPv6 address format. */
{
su->sin6.sin6_family = AF_INET6;
#ifdef SIN6_LEN
su->sin6.sin6_len = sizeof(struct sockaddr_in6);
#endif /* SIN6_LEN */
return 0;
}
#endif /* HAVE_IPV6 */
return -1;
}
const char *
sockunion2str (union sockunion *su, char *buf, size_t len)
{
if (su->sa.sa_family == AF_INET)
return inet_ntop (AF_INET, &su->sin.sin_addr, buf, len);
#ifdef HAVE_IPV6
else if (su->sa.sa_family == AF_INET6)
return inet_ntop (AF_INET6, &su->sin6.sin6_addr, buf, len);
#endif /* HAVE_IPV6 */
return NULL;
}
union sockunion *
sockunion_str2su (char *str)
{
int ret;
union sockunion *su;
su = XMALLOC (MTYPE_SOCKUNION, sizeof (union sockunion));
memset (su, 0, sizeof (union sockunion));
ret = inet_pton (AF_INET, str, &su->sin.sin_addr);
if (ret > 0) /* Valid IPv4 address format. */
{
su->sin.sin_family = AF_INET;
#ifdef HAVE_SIN_LEN
su->sin.sin_len = sizeof(struct sockaddr_in);
#endif /* HAVE_SIN_LEN */
return su;
}
#ifdef HAVE_IPV6
ret = inet_pton (AF_INET6, str, &su->sin6.sin6_addr);
if (ret > 0) /* Valid IPv6 address format. */
{
su->sin6.sin6_family = AF_INET6;
#ifdef SIN6_LEN
su->sin6.sin6_len = sizeof(struct sockaddr_in6);
#endif /* SIN6_LEN */
return su;
}
#endif /* HAVE_IPV6 */
XFREE (MTYPE_SOCKUNION, su);
return NULL;
}
char *
sockunion_su2str (union sockunion *su)
{
char str[INET6_ADDRSTRLEN];
switch (su->sa.sa_family)
{
case AF_INET:
inet_ntop (AF_INET, &su->sin.sin_addr, str, sizeof (str));
break;
#ifdef HAVE_IPV6
case AF_INET6:
inet_ntop (AF_INET6, &su->sin6.sin6_addr, str, sizeof (str));
break;
#endif /* HAVE_IPV6 */
}
return strdup (str);
}
/* Return socket of sockunion. */
int
sockunion_socket (union sockunion *su)
{
int sock;
sock = socket (su->sa.sa_family, SOCK_STREAM, 0);
if (sock < 0)
{
zlog (NULL, LOG_WARNING, "Can't make socket : %s", strerror (errno));
return -1;
}
return sock;
}
/* Return accepted new socket file descriptor. */
int
sockunion_accept (int sock, union sockunion *su)
{
socklen_t len;
int client_sock;
len = sizeof (union sockunion);
client_sock = accept (sock, (struct sockaddr *) su, &len);
/* Convert IPv4 compatible IPv6 address to IPv4 address. */
#ifdef HAVE_IPV6
if (su->sa.sa_family == AF_INET6)
{
if (IN6_IS_ADDR_V4MAPPED (&su->sin6.sin6_addr))
{
struct sockaddr_in sin;
memset (&sin, 0, sizeof (struct sockaddr_in));
sin.sin_family = AF_INET;
memcpy (&sin.sin_addr, ((char *)&su->sin6.sin6_addr) + 12, 4);
memcpy (su, &sin, sizeof (struct sockaddr_in));
}
}
#endif /* HAVE_IPV6 */
return client_sock;
}
/* Return sizeof union sockunion. */
int
sockunion_sizeof (union sockunion *su)
{
int ret;
ret = 0;
switch (su->sa.sa_family)
{
case AF_INET:
ret = sizeof (struct sockaddr_in);
break;
#ifdef HAVE_IPV6
case AF_INET6:
ret = sizeof (struct sockaddr_in6);
break;
#endif /* AF_INET6 */
}
return ret;
}
/* return sockunion structure : this function should be revised. */
char *
sockunion_log (union sockunion *su)
{
static char buf[SU_ADDRSTRLEN];
switch (su->sa.sa_family)
{
case AF_INET:
snprintf (buf, BUFSIZ, "%s", inet_ntoa (su->sin.sin_addr));
break;
#ifdef HAVE_IPV6
case AF_INET6:
snprintf (buf, BUFSIZ, "%s",
inet_ntop (AF_INET6, &(su->sin6.sin6_addr), buf, BUFSIZ));
break;
#endif /* HAVE_IPV6 */
default:
snprintf (buf, BUFSIZ, "af_unknown %d ", su->sa.sa_family);
break;
}
return buf;
}
/* sockunion_connect returns
-1 : error occured
0 : connect success
1 : connect is in progress */
enum connect_result
sockunion_connect (int fd, union sockunion *peersu, unsigned short port,
unsigned int ifindex)
{
int ret;
int val;
union sockunion su;
memcpy (&su, peersu, sizeof (union sockunion));
switch (su.sa.sa_family)
{
case AF_INET:
su.sin.sin_port = port;
break;
#ifdef HAVE_IPV6
case AF_INET6:
su.sin6.sin6_port = port;
#ifdef KAME
if (IN6_IS_ADDR_LINKLOCAL(&su.sin6.sin6_addr) && ifindex)
{
#ifdef HAVE_SIN6_SCOPE_ID
/* su.sin6.sin6_scope_id = ifindex; */
#endif /* HAVE_SIN6_SCOPE_ID */
SET_IN6_LINKLOCAL_IFINDEX (su.sin6.sin6_addr, ifindex);
}
#endif /* KAME */
break;
#endif /* HAVE_IPV6 */
}
/* Make socket non-block. */
val = fcntl (fd, F_GETFL, 0);
fcntl (fd, F_SETFL, val|O_NONBLOCK);
/* Call connect function. */
ret = connect (fd, (struct sockaddr *) &su, sockunion_sizeof (&su));
/* Immediate success */
if (ret == 0)
{
fcntl (fd, F_SETFL, val);
return connect_success;
}
/* If connect is in progress then return 1 else it's real error. */
if (ret < 0)
{
if (errno != EINPROGRESS)
{
zlog_info ("can't connect to %s fd %d : %s",
sockunion_log (&su), fd, strerror (errno));
return connect_error;
}
}
fcntl (fd, F_SETFL, val);
return connect_in_progress;
}
/* Make socket from sockunion union. */
int
sockunion_stream_socket (union sockunion *su)
{
int sock;
if (su->sa.sa_family == 0)
su->sa.sa_family = AF_INET_UNION;
sock = socket (su->sa.sa_family, SOCK_STREAM, 0);
if (sock < 0)
zlog (NULL, LOG_WARNING, "can't make socket sockunion_stream_socket");
return sock;
}
/* Bind socket to specified address. */
int
sockunion_bind (int sock, union sockunion *su, unsigned short port,
union sockunion *su_addr)
{
int size = 0;
int ret;
if (su->sa.sa_family == AF_INET)
{
size = sizeof (struct sockaddr_in);
su->sin.sin_port = htons (port);
#ifdef HAVE_SIN_LEN
su->sin.sin_len = size;
#endif /* HAVE_SIN_LEN */
if (su_addr == NULL)
su->sin.sin_addr.s_addr = htonl (INADDR_ANY);
}
#ifdef HAVE_IPV6
else if (su->sa.sa_family == AF_INET6)
{
size = sizeof (struct sockaddr_in6);
su->sin6.sin6_port = htons (port);
#ifdef SIN6_LEN
su->sin6.sin6_len = size;
#endif /* SIN6_LEN */
if (su_addr == NULL)
{
#if defined(LINUX_IPV6) || defined(NRL)
memset (&su->sin6.sin6_addr, 0, sizeof (struct in6_addr));
#else
su->sin6.sin6_addr = in6addr_any;
#endif /* LINUX_IPV6 */
}
}
#endif /* HAVE_IPV6 */
ret = bind (sock, (struct sockaddr *)su, size);
if (ret < 0)
zlog (NULL, LOG_WARNING, "can't bind socket : %s", strerror (errno));
return ret;
}
int
sockopt_reuseaddr (int sock)
{
int ret;
int on = 1;
ret = setsockopt (sock, SOL_SOCKET, SO_REUSEADDR,
(void *) &on, sizeof (on));
if (ret < 0)
{
zlog (NULL, LOG_WARNING, "can't set sockopt SO_REUSEADDR to socket %d", sock);
return -1;
}
return 0;
}
#ifdef SO_REUSEPORT
int
sockopt_reuseport (int sock)
{
int ret;
int on = 1;
ret = setsockopt (sock, SOL_SOCKET, SO_REUSEPORT,
(void *) &on, sizeof (on));
if (ret < 0)
{
zlog (NULL, LOG_WARNING, "can't set sockopt SO_REUSEADDR to socket %d", sock);
return -1;
}
return 0;
}
#else
int
sockopt_reuseport (int sock)
{
return 0;
}
#endif /* 0 */
int
sockopt_ttl (int family, int sock, int ttl)
{
int ret;
#ifdef IP_TTL
if (family == AF_INET)
{
ret = setsockopt (sock, IPPROTO_IP, IP_TTL,
(void *) &ttl, sizeof (int));
if (ret < 0)
{
zlog (NULL, LOG_WARNING, "can't set sockopt IP_TTL %d to socket %d", ttl, sock);
return -1;
}
return 0;
}
#endif /* IP_TTL */
#ifdef HAVE_IPV6
if (family == AF_INET6)
{
ret = setsockopt (sock, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
(void *) &ttl, sizeof (int));
if (ret < 0)
{
zlog (NULL, LOG_WARNING, "can't set sockopt IPV6_UNICAST_HOPS %d to socket %d",
ttl, sock);
return -1;
}
return 0;
}
#endif /* HAVE_IPV6 */
return 0;
}
/* If same family and same prefix return 1. */
int
sockunion_same (union sockunion *su1, union sockunion *su2)
{
int ret = 0;
if (su1->sa.sa_family != su2->sa.sa_family)
return 0;
switch (su1->sa.sa_family)
{
case AF_INET:
ret = memcmp (&su1->sin.sin_addr, &su2->sin.sin_addr,
sizeof (struct in_addr));
break;
#ifdef HAVE_IPV6
case AF_INET6:
ret = memcmp (&su1->sin6.sin6_addr, &su2->sin6.sin6_addr,
sizeof (struct in6_addr));
break;
#endif /* HAVE_IPV6 */
}
if (ret == 0)
return 1;
else
return 0;
}
/* After TCP connection is established. Get local address and port. */
union sockunion *
sockunion_getsockname (int fd)
{
int ret;
int len;
union
{
struct sockaddr sa;
struct sockaddr_in sin;
#ifdef HAVE_IPV6
struct sockaddr_in6 sin6;
#endif /* HAVE_IPV6 */
char tmp_buffer[128];
} name;
union sockunion *su;
memset (&name, 0, sizeof name);
len = sizeof name;
ret = getsockname (fd, (struct sockaddr *)&name, &len);
if (ret < 0)
{
zlog_warn ("Can't get local address and port by getsockname: %s",
strerror (errno));
return NULL;
}
if (name.sa.sa_family == AF_INET)
{
su = XCALLOC (MTYPE_TMP, sizeof (union sockunion));
memcpy (su, &name, sizeof (struct sockaddr_in));
return su;
}
#ifdef HAVE_IPV6
if (name.sa.sa_family == AF_INET6)
{
su = XCALLOC (MTYPE_TMP, sizeof (union sockunion));
memcpy (su, &name, sizeof (struct sockaddr_in6));
if (IN6_IS_ADDR_V4MAPPED (&su->sin6.sin6_addr))
{
struct sockaddr_in sin;
sin.sin_family = AF_INET;
memcpy (&sin.sin_addr, ((char *)&su->sin6.sin6_addr) + 12, 4);
sin.sin_port = su->sin6.sin6_port;
memcpy (su, &sin, sizeof (struct sockaddr_in));
}
return su;
}
#endif /* HAVE_IPV6 */
return NULL;
}
/* After TCP connection is established. Get remote address and port. */
union sockunion *
sockunion_getpeername (int fd)
{
int ret;
int len;
union
{
struct sockaddr sa;
struct sockaddr_in sin;
#ifdef HAVE_IPV6
struct sockaddr_in6 sin6;
#endif /* HAVE_IPV6 */
char tmp_buffer[128];
} name;
union sockunion *su;
memset (&name, 0, sizeof name);
len = sizeof name;
ret = getpeername (fd, (struct sockaddr *)&name, &len);
if (ret < 0)
{
zlog (NULL, LOG_WARNING, "Can't get remote address and port: %s",
strerror (errno));
return NULL;
}
if (name.sa.sa_family == AF_INET)
{
su = XCALLOC (MTYPE_TMP, sizeof (union sockunion));
memcpy (su, &name, sizeof (struct sockaddr_in));
return su;
}
#ifdef HAVE_IPV6
if (name.sa.sa_family == AF_INET6)
{
su = XCALLOC (MTYPE_TMP, sizeof (union sockunion));
memcpy (su, &name, sizeof (struct sockaddr_in6));
if (IN6_IS_ADDR_V4MAPPED (&su->sin6.sin6_addr))
{
struct sockaddr_in sin;
sin.sin_family = AF_INET;
memcpy (&sin.sin_addr, ((char *)&su->sin6.sin6_addr) + 12, 4);
sin.sin_port = su->sin6.sin6_port;
memcpy (su, &sin, sizeof (struct sockaddr_in));
}
return su;
}
#endif /* HAVE_IPV6 */
return NULL;
}
/* Print sockunion structure */
void
sockunion_print (union sockunion *su)
{
if (su == NULL)
return;
switch (su->sa.sa_family)
{
case AF_INET:
printf ("%s\n", inet_ntoa (su->sin.sin_addr));
break;
#ifdef HAVE_IPV6
case AF_INET6:
{
char buf [64];
printf ("%s\n", inet_ntop (AF_INET6, &(su->sin6.sin6_addr),
buf, sizeof (buf)));
}
break;
#endif /* HAVE_IPV6 */
#ifdef AF_LINK
case AF_LINK:
{
struct sockaddr_dl *sdl;
sdl = (struct sockaddr_dl *)&(su->sa);
printf ("link#%d\n", sdl->sdl_index);
}
break;
#endif /* AF_LINK */
default:
printf ("af_unknown %d\n", su->sa.sa_family);
break;
}
}
#ifdef HAVE_IPV6
int
in6addr_cmp (struct in6_addr *addr1, struct in6_addr *addr2)
{
int i;
u_char *p1, *p2;
p1 = (u_char *)addr1;
p2 = (u_char *)addr2;
for (i = 0; i < sizeof (struct in6_addr); i++)
{
if (p1[i] > p2[i])
return 1;
else if (p1[i] < p2[i])
return -1;
}
return 0;
}
#endif /* HAVE_IPV6 */
int
sockunion_cmp (union sockunion *su1, union sockunion *su2)
{
if (su1->sa.sa_family > su2->sa.sa_family)
return 1;
if (su1->sa.sa_family < su2->sa.sa_family)
return -1;
if (su1->sa.sa_family == AF_INET)
{
if (ntohl (su1->sin.sin_addr.s_addr) == ntohl (su2->sin.sin_addr.s_addr))
return 0;
if (ntohl (su1->sin.sin_addr.s_addr) > ntohl (su2->sin.sin_addr.s_addr))
return 1;
else
return -1;
}
#ifdef HAVE_IPV6
if (su1->sa.sa_family == AF_INET6)
return in6addr_cmp (&su1->sin6.sin6_addr, &su2->sin6.sin6_addr);
#endif /* HAVE_IPV6 */
return 0;
}
/* Duplicate sockunion. */
union sockunion *
sockunion_dup (union sockunion *su)
{
union sockunion *dup = XCALLOC (MTYPE_SOCKUNION, sizeof (union sockunion));
memcpy (dup, su, sizeof (union sockunion));
return dup;
}
void
sockunion_free (union sockunion *su)
{
XFREE (MTYPE_SOCKUNION, su);
}