netsukuku/src/inet.c

1147 lines
26 KiB
C
Raw Normal View History

2013-09-16 09:53:25 +00:00
/* This file is part of Netsukuku
* (c) Copyright 2005 Andrea Lo Pumo aka AlpT <alpt@freaknet.org>
*
* This source code is free software; you can redistribute it and/or
2014-05-03 19:10:51 +00:00
* modify it under the terms of the GNU General Public License as published
2013-09-16 09:53:25 +00:00
* by the Free Software Foundation; either version 2 of the License,
* or (at your option) any later version.
*
* This source code 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.
* Please refer to the GNU Public License for more details.
*
* You should have received a copy of the GNU Public License along with
* this source code; if not, write to:
* Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include "includes.h"
#include "common.h"
#include "ipv6-gmp.h"
#include "libnetlink.h"
#include "ll_map.h"
#include "inet.h"
#include "endianness.h"
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_ntohl: Converts each element of `data' from network to host order. If
* `family' is equal to AF_INET6, the array is swapped too (on big endian
* machine).
*/
void inet_ntohl(u_int *data, int family)
{
#if BYTE_ORDER == LITTLE_ENDIAN
if(family==AF_INET) {
data[0]=ntohl(data[0]);
} else {
int i;
swap_ints(MAX_IP_INT, data, data);
for(i=0; i<MAX_IP_INT; i++)
data[i]=ntohl(data[i]);
}
#endif
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_htonl: Converts each element of `data' from host to network order. If
* `family' is equal to AF_INET6, the array is swapped too (on big endian
* machine).
*/
void inet_htonl(u_int *data, int family)
{
#if BYTE_ORDER == LITTLE_ENDIAN
if(family==AF_INET) {
data[0]=htonl(data[0]);
} else {
int i;
swap_ints(MAX_IP_INT, data, data);
for(i=0; i<MAX_IP_INT; i++)
data[i]=htonl(data[i]);
}
#endif
}
/*
* inet_setip_raw: fills the `ip' inet_prefix struct with `data' and `family'.
*/
int inet_setip_raw(inet_prefix *ip, u_int *data, int family)
{
ip->family=family;
setzero(ip->data, sizeof(ip->data));
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(family==AF_INET) {
ip->data[0]=data[0];
ip->len=4;
} else if(family==AF_INET6) {
memcpy(ip->data, data, sizeof(ip->data));
ip->len=16;
2014-05-03 19:10:51 +00:00
} else
2013-09-16 09:53:25 +00:00
fatal(ERROR_MSG "family not supported", ERROR_POS);
ip->bits=ip->len<<3; /* bits=len*8 */
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
return 0;
}
/*
* inet_setip: fills the `ip' inet_prefix struct with `data' and `family'.
* Note that it does a network to host order conversion on `data'.
*/
int inet_setip(inet_prefix *ip, u_int *data, int family)
{
inet_setip_raw(ip, data, family);
inet_ntohl(ip->data, ip->family);
return 0;
}
int inet_setip_bcast(inet_prefix *ip, int family)
{
if(family==AF_INET) {
u_int data[MAX_IP_INT]={0, 0, 0, 0};
data[0]=INADDR_BROADCAST;
inet_setip(ip, data, family);
} else if(family==AF_INET6) {
u_int data[MAX_IP_INT]=IPV6_ADDR_BROADCAST;
inet_setip(ip, data, family);
2014-05-03 19:10:51 +00:00
} else
2013-09-16 09:53:25 +00:00
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
int inet_setip_anyaddr(inet_prefix *ip, int family)
{
if(family==AF_INET) {
u_int data[MAX_IP_INT]={0, 0, 0, 0};
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
data[0]=INADDR_ANY;
inet_setip(ip, data, family);
} else if(family==AF_INET6) {
struct in6_addr ipv6=IN6ADDR_ANY_INIT;
inet_setip(ip, (u_int *)(&ipv6), family);
2014-05-03 19:10:51 +00:00
} else
2013-09-16 09:53:25 +00:00
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
int inet_setip_loopback(inet_prefix *ip, int family)
{
if(family==AF_INET) {
u_int data[MAX_IP_INT]={0, 0, 0, 0};
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
data[0]=LOOPBACK_IP;
inet_setip(ip, data, family);
inet_htonl(ip->data, ip->family);
} else if(family==AF_INET6) {
u_int data[MAX_IP_INT]=LOOPBACK_IPV6;
inet_setip(ip, data, family);
2014-05-03 19:10:51 +00:00
} else
2013-09-16 09:53:25 +00:00
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_setip_localaddr: Restrict the `ip' to a local private class changing the
2014-05-03 19:10:51 +00:00
* first byte of the `ip'. `class' specifies what restricted class is currently
2013-09-16 09:53:25 +00:00
* being used (10.x.x.x or 172.16.x.x). In ipv6 the site local class is the
* default.
*/
int inet_setip_localaddr(inet_prefix *ip, int family, int class)
{
if(family==AF_INET) {
if(class == RESTRICTED_10)
ip->data[0] = NTK_RESTRICTED_10_MASK(ip->data[0]);
2014-05-03 19:10:51 +00:00
else
2013-09-16 09:53:25 +00:00
ip->data[0] = NTK_RESTRICTED_172_MASK(ip->data[0]);
} else if(family==AF_INET6) {
ip->data[0] = NTK_RESTRICTED_IPV6_MASK(ip->data[0]);
2014-05-03 19:10:51 +00:00
} else
2013-09-16 09:53:25 +00:00
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
/*
* inet_is_ip_local: verifies if `ip' is a local address. If it is, 1 is
2014-05-03 19:10:51 +00:00
* returned. `class' specifies what restricted class is currently
2013-09-16 09:53:25 +00:00
* being used (10.x.x.x or 172.16.x.x). In ipv6 the site local class is the
* default.
*/
int inet_is_ip_local(inet_prefix *ip, int class)
{
if(ip->family==AF_INET) {
if(class == RESTRICTED_10)
return ip->data[0] == NTK_RESTRICTED_10_MASK(ip->data[0]);
else
return ip->data[0] == NTK_RESTRICTED_172_MASK(ip->data[0]);
} else if(ip->family==AF_INET6)
return ip->data[0] == NTK_RESTRICTED_IPV6_MASK(ip->data[0]);
else
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
void inet_copy(inet_prefix *dst, inet_prefix *src)
{
memcpy(dst, src, sizeof(inet_prefix));
}
/*
* inet_copy_ipdata_raw: copies `ip'->data in `dst_data'.
*/
void inet_copy_ipdata_raw(u_int *dst_data, inet_prefix *ip)
{
memcpy(dst_data, ip->data, MAX_IP_SZ);
}
/*
* inet_copy_ipdata: copies `ip'->data in `dst_data' and converts it in network
* order.
*/
void inet_copy_ipdata(u_int *dst_data, inet_prefix *ip)
{
inet_prefix tmp_ip;
inet_copy(&tmp_ip, ip);
inet_htonl(tmp_ip.data, tmp_ip.family);
memcpy(dst_data, tmp_ip.data, MAX_IP_SZ);
}
/*
* pack_inet_prefix: packs the `ip' inet_prefix struct and stores it in
* `pack', which must be INET_PREFIX_PACK_SZ bytes big. `pack' will be in
* network order.
*/
void pack_inet_prefix(inet_prefix *ip, char *pack)
{
char *buf;
buf=pack;
memcpy(buf, &ip->family, sizeof(u_char));
buf+=sizeof(u_char);
memcpy(buf, &ip->len, sizeof(u_short));
buf+=sizeof(u_short);
memcpy(buf, &ip->bits, sizeof(u_char));
buf+=sizeof(u_char);
memcpy(buf, ip->data, MAX_IP_SZ);
inet_htonl((u_int *)buf, ip->family);
buf+=MAX_IP_SZ;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
ints_host_to_network(pack, inet_prefix_iinfo);
}
/*
* unpack_inet_prefix: restores in `ip' the inet_prefix struct contained in `pack'.
* Note that `pack' will be modified during the restoration.
*/
void unpack_inet_prefix(inet_prefix *ip, char *pack)
{
char *buf;
buf=pack;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
ints_network_to_host(pack, inet_prefix_iinfo);
memcpy(&ip->family, buf, sizeof(u_char));
buf+=sizeof(u_char);
memcpy(&ip->len, buf, sizeof(u_short));
buf+=sizeof(u_short);
memcpy(&ip->bits, buf, sizeof(u_char));
buf+=sizeof(u_char);
memcpy(ip->data, buf, MAX_IP_SZ);
inet_ntohl(ip->data, ip->family);
buf+=MAX_IP_SZ;
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_addr_match: without hesitating this function was robbed from iproute2.
* It compares a->data wih b->data matching `bits'# bits.
*/
int inet_addr_match(const inet_prefix *a, const inet_prefix *b, int bits)
{
2014-05-03 19:10:51 +00:00
const uint32_t *a1 = a->data;
const uint32_t *a2 = b->data;
2013-09-16 09:53:25 +00:00
int words = bits >> 0x05;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
bits &= 0x1f;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if (words)
if (memcmp(a1, a2, words << 2))
return -1;
if (bits) {
uint32_t w1, w2;
uint32_t mask;
w1 = a1[words];
w2 = a2[words];
mask = htonl((0xffffffff) << (0x20 - bits));
if ((w1 ^ w2) & mask)
return 1;
}
return 0;
}
int ipv6_addr_type(inet_prefix addr)
{
int type;
u_int st;
st = htonl(addr.data[0]);
if ((st & htonl(0xFF000000)) == htonl(0xFF000000)) {
type = IPV6_ADDR_MULTICAST;
switch((st & htonl(0x00FF0000))) {
case __constant_htonl(0x00010000):
type |= IPV6_ADDR_LOOPBACK;
break;
case __constant_htonl(0x00020000):
type |= IPV6_ADDR_LINKLOCAL;
break;
case __constant_htonl(0x00050000):
type |= IPV6_ADDR_SITELOCAL;
break;
};
return type;
}
type = IPV6_ADDR_UNICAST;
/* Consider all addresses with the first three bits different of
000 and 111 as finished.
*/
if ((st & htonl(0xE0000000)) != htonl(0x00000000) &&
(st & htonl(0xE0000000)) != htonl(0xE0000000))
return type;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if ((st & htonl(0xFFC00000)) == htonl(0xFE800000))
return (IPV6_ADDR_LINKLOCAL | type);
if ((st & htonl(0xFFC00000)) == htonl(0xFEC00000))
return (IPV6_ADDR_SITELOCAL | type);
if ((addr.data[0] | addr.data[1]) == 0) {
if (addr.data[2] == 0) {
if (addr.data[3] == 0)
return IPV6_ADDR_ANY;
if (htonl(addr.data[3]) == htonl(0x00000001))
return (IPV6_ADDR_LOOPBACK | type);
return (IPV6_ADDR_COMPATv4 | type);
}
if (htonl(addr.data[2]) == htonl(0x0000ffff))
return IPV6_ADDR_MAPPED;
}
st &= htonl(0xFF000000);
if (st == 0)
return IPV6_ADDR_RESERVED;
st &= htonl(0xFE000000);
if (st == htonl(0x02000000))
return IPV6_ADDR_RESERVED; /* for NSAP */
if (st == htonl(0x04000000))
return IPV6_ADDR_RESERVED; /* for IPX */
return type;
}
/*
* inet_validate_ip: returns 0 is `ip' a valid IP which can be set by
* Netsukuku to a network interface
*/
int inet_validate_ip(inet_prefix ip)
{
int type, ipv4;
if(ip.family==AF_INET) {
ipv4=htonl(ip.data[0]);
2014-05-03 19:10:51 +00:00
if(MULTICAST(ipv4) || BADCLASS(ipv4) || ZERONET(ipv4)
2013-09-16 09:53:25 +00:00
|| LOOPBACK(ipv4) || NTK_PRIVATE_C(ipv4) ||
(!restricted_mode && NTK_PRIVATE_B(ipv4)))
return -EINVAL;
} else if(ip.family==AF_INET6) {
type=ipv6_addr_type(ip);
2014-05-03 19:10:51 +00:00
if( (type & IPV6_ADDR_MULTICAST) || (type & IPV6_ADDR_RESERVED) ||
2013-09-16 09:53:25 +00:00
(type & IPV6_ADDR_LOOPBACK))
return -EINVAL;
}
if(is_bufzero((char *)ip.data, MAX_IP_SZ))
return -EINVAL;
return 0;
}
/*\
*
2014-05-03 19:10:51 +00:00
* * * Conversion functions... * *
*
2013-09-16 09:53:25 +00:00
\*/
/*
* ipraw_to_str: It returns the string which represents the given ip in host
* order.
*/
const char *ipraw_to_str(u_int ip[MAX_IP_INT], int family)
{
struct in_addr src;
struct in6_addr src6;
static char dst[INET_ADDRSTRLEN], dst6[INET6_ADDRSTRLEN];
if(family==AF_INET) {
src.s_addr=htonl(ip[0]);
inet_ntop(family, &src, dst, INET_ADDRSTRLEN);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
return dst;
} else if(family==AF_INET6) {
inet_htonl(ip, family);
memcpy(&src6, ip, MAX_IP_SZ);
inet_ntop(family, &src6, dst6, INET6_ADDRSTRLEN);
return dst6;
}
return 0;
}
/*
* inet_to_str: returns the string rapresentation of `ip'
*/
const char *inet_to_str(inet_prefix ip)
{
return ipraw_to_str(ip.data, ip.family);
}
/*
* str_to_inet: it converts the IP address string contained in `src' and
* terminated by a `\0' char to an inet_prefix struct. The result is stored in
* `ip'. On error -1 is returned.
*/
int str_to_inet(const char *src, inet_prefix *ip)
{
struct in_addr dst;
struct in6_addr dst6;
int family,res;
u_int *data;
setzero(ip, sizeof(inet_prefix));
if(strstr(src, ":")) {
family=AF_INET6;
data=(u_int *)&dst6;
} else {
family=AF_INET;
data=(u_int *)&dst;
}
if((res=inet_pton(family, src, (void *)data)) < 0) {
2014-05-03 19:10:51 +00:00
debug(DBG_NORMAL, ERROR_MSG "error -> %s.",
2013-09-16 09:53:25 +00:00
ERROR_FUNC, strerror(errno));
return -1;
}
if (!res) {
debug(DBG_NORMAL, ERROR_MSG "impossible to convert \"%s\":"
" invalid address.", ERROR_FUNC, src);
return -1;
}
inet_setip(ip, data, family);
return 0;
}
/*
* inet_to_sockaddr: Converts a inet_prefix struct to a sockaddr struct
*/
int inet_to_sockaddr(inet_prefix *ip, u_short port, struct sockaddr *dst,
socklen_t *dstlen)
{
port=htons(port);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(ip->family==AF_INET) {
struct sockaddr_in sin;
setzero(&sin, sizeof(struct sockaddr_in));
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
sin.sin_family = ip->family;
sin.sin_port = port;
sin.sin_addr.s_addr = htonl(ip->data[0]);
memcpy(dst, &sin, sizeof(struct sockaddr_in));
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(dstlen)
*dstlen=sizeof(struct sockaddr_in);
} else if(ip->family==AF_INET6) {
struct sockaddr_in6 sin6;
setzero(&sin6, sizeof(struct sockaddr_in6));
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
sin6.sin6_family = ip->family;
sin6.sin6_port = port;
sin6.sin6_flowinfo = 0;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
memcpy(&sin6.sin6_addr, ip->data, MAX_IP_SZ);
inet_htonl((u_int *)&sin6.sin6_addr, ip->family);
memcpy(dst, &sin6, sizeof(struct sockaddr_in6));
if(dstlen)
*dstlen=sizeof(struct sockaddr_in6);
} else
fatal(ERROR_MSG "family not supported", ERROR_POS);
return 0;
}
int sockaddr_to_inet(struct sockaddr *ip, inet_prefix *dst, u_short *port)
{
u_short po;
char *p;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
setzero(dst, sizeof(inet_prefix));
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
dst->family=ip->sa_family;
memcpy(&po, &ip->sa_data, sizeof(u_short));
if(port)
*port=ntohs(po);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(ip->sa_family==AF_INET)
p=(char *)ip->sa_data+sizeof(u_short);
else if(ip->sa_family==AF_INET6)
p=(char *)ip->sa_data+sizeof(u_short)+sizeof(int);
else {
error(ERROR_MSG "family not supported", ERROR_POS);
return -1;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
inet_setip(dst, (u_int *)p, ip->sa_family);
return 0;
}
/*\
*
* * * Socket operations * *
*
\*/
int new_socket(int sock_type)
{
int sockfd;
if((sockfd=socket(sock_type, SOCK_STREAM, 0)) == -1 ) {
error("Socket SOCK_STREAM creation failed: %s", strerror(errno));
return -1;
}
2014-05-09 23:23:50 +00:00
printf("Socket FD: %i ", sockfd);
2013-09-16 09:53:25 +00:00
return sockfd;
}
int new_dgram_socket(int sock_type)
{
int sockfd;
if((sockfd=socket(sock_type, SOCK_DGRAM, 0)) == -1 ) {
error("Socket SOCK_DGRAM creation failed: %s", strerror(errno));
return -1;
}
return sockfd;
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_close
*
* It closes the `*sk' socket and sets it to zero.
* It always returns 0;
*/
int inet_close(int *sk)
{
close(*sk);
return (*sk=0);
}
int inet_getpeername(int sk, inet_prefix *ip, short *port)
{
struct sockaddr_storage saddr_sto;
struct sockaddr *sa=(struct sockaddr *)&saddr_sto;
socklen_t alen;
alen = sizeof(saddr_sto);
setzero(sa, alen);
if(getpeername(sk, sa, &alen) == -1) {
error("Cannot getpeername: %s", strerror(errno));
return -1;
}
return sockaddr_to_inet(sa, ip, port);
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* join_ipv6_multicast: It adds the membership to the IPV6_ADDR_BROADCAST
2014-05-03 19:10:51 +00:00
* multicast group. The device with index `idx' will be used.
2013-09-16 09:53:25 +00:00
*/
int join_ipv6_multicast(int socket, int idx)
{
struct ipv6_mreq mreq6;
const int addr[MAX_IP_INT]=IPV6_ADDR_BROADCAST;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
setzero(&mreq6, sizeof(struct ipv6_mreq));
memcpy(&mreq6.ipv6mr_multiaddr, addr, sizeof(struct in6_addr));
mreq6.ipv6mr_interface=idx;
2014-05-03 19:10:51 +00:00
if(setsockopt(socket, IPPROTO_IPV6, IPV6_JOIN_GROUP, &mreq6,
2013-09-16 09:53:25 +00:00
sizeof(mreq6)) < 0) {
error("Cannot set IPV6_JOIN_GROUP: %s", strerror(errno));
close(socket);
return -1;
}
return socket;
}
int set_multicast_if(int socket, int idx)
{
/* man ipv6 */
if (setsockopt(socket, IPPROTO_IPV6, IPV6_MULTICAST_IF,
&idx, sizeof(int)) < 0) {
error("set_multicast_if(): cannot set IPV6_MULTICAST_IF: %s",
strerror(errno));
close(socket);
return -1;
}
return 0;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
int set_nonblock_sk(int fd)
{
if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) {
2014-05-03 19:10:51 +00:00
error("set_nonblock_sk(): cannot set O_NONBLOCK: %s",
2013-09-16 09:53:25 +00:00
strerror(errno));
close(fd);
return -1;
}
return 0;
}
int unset_nonblock_sk(int fd)
{
if (fcntl(fd, F_SETFL, 0) < 0) {
2014-05-03 19:10:51 +00:00
error("unset_nonblock_sk(): cannot unset O_NONBLOCK: %s",
2013-09-16 09:53:25 +00:00
strerror(errno));
close(fd);
return -1;
}
return 0;
}
int set_reuseaddr_sk(int socket)
{
int reuseaddr=1, ret;
/*
* SO_REUSEADDR: <<Go ahead and reuse that port even if it is in
* TIME_WAIT state.>>
*/
ret=setsockopt(socket, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(int));
if(ret < 0)
error("setsockopt SO_REUSEADDR: %s", strerror(errno));
return ret;
}
int set_bindtodevice_sk(int socket, char *dev)
{
struct ifreq ifr;
int ret=0;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
setzero(&ifr, sizeof(ifr));
strncpy(ifr.ifr_name, dev, IFNAMSIZ-1);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
ret=setsockopt(socket, SOL_SOCKET, SO_BINDTODEVICE, dev, strlen(dev)+1);
if(ret < 0)
error("setsockopt SO_BINDTODEVICE: %s", strerror(errno));
return ret;
}
/*
2014-05-03 19:10:51 +00:00
* `loop': 0 = disable, 1 = enable (default)
2013-09-16 09:53:25 +00:00
*/
int set_multicast_loop_sk(int family, int socket, u_char loop)
{
int ret=0;
/*
* <<The IPV6_MULTICAST_LOOP option gives the sender explicit control
* over whether or not subsequent datagrams are looped bac.>>
*/
if(family==AF_INET6)
ret=setsockopt(socket, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &loop, sizeof(loop));
if(ret < 0)
error("setsockopt IP_MULTICAST_LOOP: %s", strerror(errno));
return ret;
}
int set_broadcast_sk(int socket, int family, inet_prefix *host, short port,
int dev_idx)
{
struct sockaddr_storage saddr_sto;
struct sockaddr *sa=(struct sockaddr *)&saddr_sto;
socklen_t alen;
int broadcast=1;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(family == AF_INET) {
if (setsockopt(socket, SOL_SOCKET, SO_BROADCAST, &broadcast,
sizeof(broadcast)) < 0) {
error("Cannot set SO_BROADCAST to socket: %s", strerror(errno));
close(socket);
return -1;
}
} else if(family == AF_INET6) {
if(join_ipv6_multicast(socket, dev_idx) < 0)
return -1;
if(set_multicast_loop_sk(family, socket, 0) < 0)
return -1;
set_multicast_if(socket, dev_idx);
} else
fatal(ERROR_MSG "family not supported", ERROR_POS);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
/* What's my name ? */
alen = sizeof(saddr_sto);
setzero(sa, alen);
if (getsockname(socket, sa, &alen) == -1) {
error("Cannot getsockname: %s", strerror(errno));
close(socket);
return -1;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
/* Let's bind it! */
if(bind(socket, sa, alen) < 0) {
error("Cannot bind the broadcast socket: %s", strerror(errno));
close(socket);
return -1;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
return socket;
}
int unset_broadcast_sk(int socket, int family)
{
int broadcast=0;
if(family == AF_INET) {
if (setsockopt(socket, SOL_SOCKET, SO_BROADCAST, &broadcast, sizeof(broadcast)) < 0) {
error ("Cannot unset broadcasting: %s", strerror(errno));
return -1;
}
}
return 0;
}
int set_keepalive_sk(int socket)
{
int on=1;
2014-05-03 19:10:51 +00:00
if(setsockopt(socket, SOL_SOCKET, SO_KEEPALIVE, (void *)&on,
2013-09-16 09:53:25 +00:00
sizeof(on)) < 0){
error("Cannot set keepalive socket: %s", strerror(errno));
return -1;
}
return 0;
}
int unset_keepalive_sk(int socket)
{
int off=0;
2014-05-03 19:10:51 +00:00
if(setsockopt(socket, SOL_SOCKET, SO_KEEPALIVE, (void *)&off,
2013-09-16 09:53:25 +00:00
sizeof(off)) < 0){
error("Cannot unset keepalive socket: %s", strerror(errno));
return -1;
}
return 0;
}
int set_tos_sk(int socket, int lowdelay)
{
int tos = lowdelay ? IPTOS_LOWDELAY : IPTOS_THROUGHPUT;
/* Only for Ipv4 */
if (setsockopt(socket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)) < 0) {
error("setsockopt IP_TOS %d: %s", tos, strerror(errno));
return -1;
}
return 0;
}
/*\
*
2014-05-03 19:10:51 +00:00
* * * Connection functions * *
2013-09-16 09:53:25 +00:00
*
\*/
int new_tcp_conn(inet_prefix *host, short port, char *dev)
{
int sk;
socklen_t sa_len;
struct sockaddr_storage saddr_sto;
struct sockaddr *sa=(struct sockaddr *)&saddr_sto;
const char *ntop;
ntop=inet_to_str(*host);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(inet_to_sockaddr(host, port, sa, &sa_len)) {
error("Cannot new_tcp_connect(): %d Family not supported", host->family);
ERROR_FINISH(sk, -1, finish);
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if((sk = new_socket(host->family)) == -1)
ERROR_FINISH(sk, -1, finish);
if(dev) /* if `dev' is not null bind the socket to it */
if(set_bindtodevice_sk(sk, dev) < 0)
ERROR_FINISH(sk, -1, finish);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if (connect(sk, sa, sa_len) == -1) {
error("Cannot tcp_connect() to %s: %s", ntop, strerror(errno));
ERROR_FINISH(sk, -1, finish);
}
finish:
return sk;
}
int new_udp_conn(inet_prefix *host, short port, char *dev)
2014-05-03 19:10:51 +00:00
{
2013-09-16 09:53:25 +00:00
int sk;
socklen_t sa_len;
struct sockaddr_storage saddr_sto;
struct sockaddr *sa=(struct sockaddr *)&saddr_sto;
const char *ntop;
ntop=inet_to_str(*host);
if(inet_to_sockaddr(host, port, sa, &sa_len)) {
error("Cannot new_udp_connect(): %d Family not supported", host->family);
ERROR_FINISH(sk, -1, finish);
}
2014-05-03 19:10:51 +00:00
if((sk = new_dgram_socket(host->family)) == -1)
2013-09-16 09:53:25 +00:00
ERROR_FINISH(sk, -1, finish);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(dev) /* if `dev' is not null bind the socket to it */
if(set_bindtodevice_sk(sk, dev) < 0)
ERROR_FINISH(sk, -1, finish);
if (connect(sk, sa, sa_len) == -1) {
error("Cannot connect to %s: %s", ntop, strerror(errno));
ERROR_FINISH(sk, -1, finish);
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
finish:
return sk;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
int new_bcast_conn(inet_prefix *host, short port, int dev_idx)
2014-05-03 19:10:51 +00:00
{
2013-09-16 09:53:25 +00:00
struct sockaddr_storage saddr_sto;
struct sockaddr *sa=(struct sockaddr *)&saddr_sto;
socklen_t alen;
int sk;
const char *ntop;
if((sk = new_dgram_socket(host->family)) == -1)
return -1;
sk=set_broadcast_sk(sk, host->family, host, port, dev_idx);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
/*
2014-05-03 19:10:51 +00:00
* Connect
2013-09-16 09:53:25 +00:00
*/
if(inet_to_sockaddr(host, port, sa, &alen)) {
error("set_broadcast_sk: %d Family not supported", host->family);
return -1;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(host->family == AF_INET6) {
struct sockaddr_in6 *sin6=(struct sockaddr_in6 *)sa;
sin6->sin6_scope_id = dev_idx;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(set_bindtodevice_sk(sk, (char *)ll_index_to_name(dev_idx)) < 0)
return -1;
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
if(connect(sk, sa, alen) == -1) {
ntop=inet_to_str(*host);
2014-05-03 19:10:51 +00:00
error("Cannot connect to the broadcast (%s): %s", ntop,
2013-09-16 09:53:25 +00:00
strerror(errno));
return -1;
}
return sk;
}
/*\
*
* * * Recv/Send functions * *
*
\*/
ssize_t inet_recv(int s, void *buf, size_t len, int flags)
{
ssize_t err;
fd_set fdset;
int ret;
if((err=recv(s, buf, len, flags))==-1) {
2014-05-03 19:10:51 +00:00
switch(errno)
2013-09-16 09:53:25 +00:00
{
default:
/* Probably connection was closed */
debug(DBG_NORMAL, "inet_recv: Cannot recv(): %s",
strerror(errno));
return err;
break;
}
}
return err;
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_recv_timeout
2014-05-03 19:10:51 +00:00
*
2013-09-16 09:53:25 +00:00
* is the same as inet_recv() but if no reply is received for `timeout'
* seconds it returns -1.
*/
ssize_t inet_recv_timeout(int s, void *buf, size_t len, int flags, u_int timeout)
{
struct timeval timeout_t;
fd_set fdset;
int ret;
MILLISEC_TO_TV(timeout*1000, timeout_t);
FD_ZERO(&fdset);
FD_SET(s, &fdset);
ret = select(s+1, &fdset, NULL, NULL, &timeout_t);
if (ret == -1) {
error(ERROR_MSG "select error: %s", ERROR_FUNC, strerror(errno));
return ret;
}
return FD_ISSET(s, &fdset) ? inet_recv(s, buf, len, flags) : -1;
}
ssize_t inet_recvfrom(int s, void *buf, size_t len, int flags, struct sockaddr *from, socklen_t *fromlen)
{
ssize_t err;
fd_set fdset;
int ret;
if((err=recvfrom(s, buf, len, flags, from, fromlen)) < 0) {
2014-05-03 19:10:51 +00:00
switch(errno)
2013-09-16 09:53:25 +00:00
{
default:
error("inet_recvfrom: Cannot recv(): %s", strerror(errno));
return err;
break;
}
}
return err;
}
2014-05-03 19:10:51 +00:00
/*
2013-09-16 09:53:25 +00:00
* inet_recvfrom_timeout: is the same as inet_recvfrom() but if no reply is
* received for `timeout' seconds it returns -1.
*/
2014-05-03 19:10:51 +00:00
ssize_t inet_recvfrom_timeout(int s, void *buf, size_t len, int flags,
2013-09-16 09:53:25 +00:00
struct sockaddr *from, socklen_t *fromlen, u_int timeout)
{
struct timeval timeout_t;
fd_set fdset;
int ret;
MILLISEC_TO_TV(timeout*1000, timeout_t);
FD_ZERO(&fdset);
FD_SET(s, &fdset);
ret = select(s+1, &fdset, NULL, NULL, &timeout_t);
if (ret == -1) {
error(ERROR_MSG "select error: %s", ERROR_FUNC, strerror(errno));
return ret;
}
if(FD_ISSET(s, &fdset))
return inet_recvfrom(s, buf, len, flags, from, fromlen);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
return -1;
}
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
ssize_t inet_send(int s, const void *msg, size_t len, int flags)
{
ssize_t err;
fd_set fdset;
int ret;
if((err=send(s, msg, len, flags)) < 0) {
2014-05-03 19:10:51 +00:00
switch(errno)
2013-09-16 09:53:25 +00:00
{
case EMSGSIZE:
inet_send(s, msg, len/2, flags);
2014-05-03 19:10:51 +00:00
err=inet_send(s, (const char *)msg+(len/2),
2013-09-16 09:53:25 +00:00
len-(len/2), flags);
break;
default:
error("inet_send: Cannot send(): %s", strerror(errno));
return err;
break;
}
}
return err;
}
/*
* inet_send_timeout: is the same as inet_send() but if the packet isn't sent
* in `timeout' seconds it timeouts and returns -1.
*/
ssize_t inet_send_timeout(int s, const void *msg, size_t len, int flags, u_int timeout)
{
struct timeval timeout_t;
fd_set fdset;
int ret;
MILLISEC_TO_TV(timeout*1000, timeout_t);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
FD_ZERO(&fdset);
FD_SET(s, &fdset);
ret = select(s+1, NULL, &fdset, NULL, &timeout_t);
if (ret == -1) {
error(ERROR_MSG "select error: %s", ERROR_FUNC, strerror(errno));
return ret;
}
if(FD_ISSET(s, &fdset))
return inet_send(s, msg, len, flags);
return -1;
}
2014-05-03 19:10:51 +00:00
ssize_t inet_sendto(int s, const void *msg, size_t len, int flags,
2013-09-16 09:53:25 +00:00
const struct sockaddr *to, socklen_t tolen)
{
ssize_t err;
fd_set fdset;
int ret;
int sendbuf;
socklen_t optlen;
optlen = sizeof(sendbuf);
int res = getsockopt(s, SOL_SOCKET, SO_SNDBUF, &sendbuf, &optlen);
error("GetSockOpt: %i \n", res);
error("Socket: %i Data: %p Data Length: %u Flags: %i Address: %p Address Length: %u", s, msg, len, flags, to, tolen);
2013-09-16 09:53:25 +00:00
if((err=sendto(s, msg, len, flags, to, tolen))==-1) {
error("sendto errno: %d err is: %d", errno, err);
switch(errno)
2013-09-16 09:53:25 +00:00
{
case EMSGSIZE:
error("Packet artificially fragmented: %d", stderr);
error("\nData Length: %u", len);
int bytesleft = len;
2014-05-09 23:23:50 +00:00
int sendbuf;
socklen_t optlen;
optlen = sizeof(sendbuf);
int res = getsockopt(s, SOL_SOCKET, SO_SNDBUF, &sendbuf, &optlen);
error("GetSockOpt: %i \n", res);
while(bytesleft != 0) {
if(bytesleft > 64000) {
inet_sendto(s, msg, 64000, flags, to, tolen);
bytesleft -= 64000;
msg += 64000
2014-05-09 23:23:50 +00:00
//err=inet_sendto(s, ((const char *)msg+(len/2)),
//len-(len/2), flags, to, tolen);
}
else {
inet_sendto(s, msg, bytesleft, flags, to, tolen);
bytesleft = 0;
}
}
2013-09-16 09:53:25 +00:00
break;
case EFAULT:
error("The value of to is: %d", to);
2013-09-16 09:53:25 +00:00
default:
error("inet_sendto: Cannot send(): %s", strerror(errno));
return err;
break;
}
}
return err;
}
/*
* inet_sendto_timeout: is the same as inet_sendto() but if the packet isn't sent
* in `timeout' seconds it timeouts and returns -1.
*/
2014-05-03 19:10:51 +00:00
ssize_t inet_sendto_timeout(int s, const void *msg, size_t len, int flags,
2013-09-16 09:53:25 +00:00
const struct sockaddr *to, socklen_t tolen, u_int timeout)
{
struct timeval timeout_t;
fd_set fdset;
int ret;
MILLISEC_TO_TV(timeout*1000, timeout_t);
2014-05-03 19:10:51 +00:00
2013-09-16 09:53:25 +00:00
FD_ZERO(&fdset);
FD_SET(s, &fdset);
ret = select(s+1, NULL, &fdset, NULL, &timeout_t);
if (ret == -1) {
error(ERROR_MSG "select error: %s", ERROR_FUNC, strerror(errno));
return ret;
}
if(FD_ISSET(s, &fdset))
return inet_sendto(s, msg, len, flags, to, tolen);
return -1;
}
ssize_t inet_sendfile(int out_fd, int in_fd, off_t *offset, size_t count)
{
ssize_t err;
fd_set fdset;
int ret;
if((err=sendfile(out_fd, in_fd, offset, count))==-1)
error("inet_sendfile: Cannot sendfile(): %s", strerror(errno));
return err;
}