[mirror_frr.git] / ldpd / socket.c

/*	$OpenBSD$ */

/*
 * Copyright (c) 2016 Renato Westphal <renato@openbsd.org>
 * Copyright (c) 2009 Michele Marchetto <michele@openbsd.org>
 * Copyright (c) 2005 Claudio Jeker <claudio@openbsd.org>
 * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <zebra.h>

#include "ldpd.h"
#include "ldpe.h"
#include "log.h"

#include "lib/log.h"
#include "privs.h"
#include "sockopt.h"

extern struct zebra_privs_t	 ldpd_privs;
extern struct zebra_privs_t	 ldpe_privs;

int
ldp_create_socket(int af, enum socket_type type)
{
	int			 fd, domain, proto;
	union ldpd_addr		 addr;
	union sockunion		 local_su;
#ifdef __OpenBSD__
	int			 opt;
#endif
	int			 save_errno;

	/* create socket */
	switch (type) {
	case LDP_SOCKET_DISC:
	case LDP_SOCKET_EDISC:
		domain = SOCK_DGRAM;
		proto = IPPROTO_UDP;
		break;
	case LDP_SOCKET_SESSION:
		domain = SOCK_STREAM;
		proto = IPPROTO_TCP;
		break;
	default:
		fatalx("ldp_create_socket: unknown socket type");
	}
	fd = socket(af, domain, proto);
	if (fd == -1) {
		log_warn("%s: error creating socket", __func__);
		return (-1);
	}
	sock_set_nonblock(fd);
	sockopt_v6only(af, fd);

	/* bind to a local address/port */
	switch (type) {
	case LDP_SOCKET_DISC:
		/* listen on all addresses */
		memset(&addr, 0, sizeof(addr));
		addr2sa(af, &addr, LDP_PORT, &local_su);
		break;
	case LDP_SOCKET_EDISC:
	case LDP_SOCKET_SESSION:
		addr = (ldp_af_conf_get(ldpd_conf, af))->trans_addr;
		addr2sa(af, &addr, LDP_PORT, &local_su);
		/* ignore any possible error */
		sock_set_bindany(fd, 1);
		break;
	}
	if (ldpd_privs.change(ZPRIVS_RAISE))
		log_warn("%s: could not raise privs", __func__);
	if (sock_set_reuse(fd, 1) == -1) {
		if (ldpd_privs.change(ZPRIVS_LOWER))
			log_warn("%s: could not lower privs", __func__);
		close(fd);
		return (-1);
	}
	if (bind(fd, &local_su.sa, sockaddr_len(&local_su.sa)) == -1) {
		save_errno = errno;
		if (ldpd_privs.change(ZPRIVS_LOWER))
			log_warn("%s: could not lower privs", __func__);
		log_warnx("%s: error binding socket: %s", __func__,
		    safe_strerror(save_errno));
		close(fd);
		return (-1);
	}
	if (ldpd_privs.change(ZPRIVS_LOWER))
		log_warn("%s: could not lower privs", __func__);

	/* set options */
	switch (af) {
	case AF_INET:
		if (sock_set_ipv4_tos(fd, IPTOS_PREC_INTERNETCONTROL) == -1) {
			close(fd);
			return (-1);
		}
		if (type == LDP_SOCKET_DISC) {
			if (sock_set_ipv4_mcast_ttl(fd,
			    IP_DEFAULT_MULTICAST_TTL) == -1) {
				close(fd);
				return (-1);
			}
			if (sock_set_ipv4_mcast_loop(fd) == -1) {
				close(fd);
				return (-1);
			}
		}
		if (type == LDP_SOCKET_DISC || type == LDP_SOCKET_EDISC) {
			if (sock_set_ipv4_recvif(fd, 1) == -1) {
				close(fd);
				return (-1);
			}
#ifndef MSG_MCAST
#if defined(HAVE_IP_PKTINFO)
			if (sock_set_ipv4_pktinfo(fd, 1) == -1) {
				close(fd);
				return (-1);
			}
#elif defined(HAVE_IP_RECVDSTADDR)
			if (sock_set_ipv4_recvdstaddr(fd, 1) == -1) {
				close(fd);
				return (-1);
			}
#else
#error "Unsupported socket API"
#endif
#endif /* MSG_MCAST */
		}
		if (type == LDP_SOCKET_SESSION) {
			if (sock_set_ipv4_ucast_ttl(fd, 255) == -1) {
				close(fd);
				return (-1);
			}
		}
		break;
	case AF_INET6:
		if (sock_set_ipv6_dscp(fd, IPTOS_PREC_INTERNETCONTROL) == -1) {
			close(fd);
			return (-1);
		}
		if (type == LDP_SOCKET_DISC) {
			if (sock_set_ipv6_mcast_loop(fd) == -1) {
				close(fd);
				return (-1);
			}
			if (sock_set_ipv6_mcast_hops(fd, 255) == -1) {
				close(fd);
				return (-1);
			}
			if (!(ldpd_conf->ipv6.flags & F_LDPD_AF_NO_GTSM)) {
				/* ignore any possible error */
				sock_set_ipv6_minhopcount(fd, 255);
			}
		}
		if (type == LDP_SOCKET_DISC || type == LDP_SOCKET_EDISC) {
			if (sock_set_ipv6_pktinfo(fd, 1) == -1) {
				close(fd);
				return (-1);
			}
		}
		if (type == LDP_SOCKET_SESSION) {
			if (sock_set_ipv6_ucast_hops(fd, 255) == -1) {
				close(fd);
				return (-1);
			}
		}
		break;
	}
	switch (type) {
	case LDP_SOCKET_DISC:
	case LDP_SOCKET_EDISC:
		sock_set_recvbuf(fd);
		break;
	case LDP_SOCKET_SESSION:
		if (listen(fd, LDP_BACKLOG) == -1)
			log_warn("%s: error listening on socket", __func__);

#ifdef __OpenBSD__
		opt = 1;
		if (setsockopt(fd, IPPROTO_TCP, TCP_MD5SIG, &opt,
		    sizeof(opt)) == -1) {
			if (errno == ENOPROTOOPT) {	/* system w/o md5sig */
				log_warnx("md5sig not available, disabling");
				sysdep.no_md5sig = 1;
			} else {
				close(fd);
				return (-1);
			}
		}
#endif
		break;
	}

	return (fd);
}

void
sock_set_nonblock(int fd)
{
	int	flags;

	if ((flags = fcntl(fd, F_GETFL, 0)) == -1)
		fatal("fcntl F_GETFL");

	flags |= O_NONBLOCK;

	if ((flags = fcntl(fd, F_SETFL, flags)) == -1)
		fatal("fcntl F_SETFL");
}

void
sock_set_cloexec(int fd)
{
	int	flags;

	if ((flags = fcntl(fd, F_GETFD, 0)) == -1)
		fatal("fcntl F_GETFD");

	flags |= FD_CLOEXEC;

	if ((flags = fcntl(fd, F_SETFD, flags)) == -1)
		fatal("fcntl F_SETFD");
}

void
sock_set_recvbuf(int fd)
{
	int	bsize;

	bsize = 65535;
	while (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &bsize,
	    sizeof(bsize)) == -1)
		bsize /= 2;
}

int
sock_set_reuse(int fd, int enable)
{
	if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &enable,
	    sizeof(int)) < 0) {
		log_warn("%s: error setting SO_REUSEADDR", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_bindany(int fd, int enable)
{
#ifdef HAVE_SO_BINDANY
	frr_elevate_privs(&ldpd_privs) {
		if (setsockopt(fd, SOL_SOCKET, SO_BINDANY, &enable,
			       sizeof(int)) < 0) {
			log_warn("%s: error setting SO_BINDANY", __func__);
			return (-1);
		}
	}
	return (0);
#elif defined(HAVE_IP_FREEBIND)
	if (setsockopt(fd, IPPROTO_IP, IP_FREEBIND, &enable, sizeof(int)) < 0) {
		log_warn("%s: error setting IP_FREEBIND", __func__);
		return (-1);
	}
	return (0);
#else
	log_warnx("%s: missing SO_BINDANY and IP_FREEBIND, unable to bind "
	    "to a nonlocal IP address", __func__);
	return (-1);
#endif /* HAVE_SO_BINDANY */
}

#ifndef __OpenBSD__
/*
 * Set MD5 key for the socket, for the given peer address. If the password
 * is NULL or zero-length, the option will be disabled.
 */
int
sock_set_md5sig(int fd, int af, union ldpd_addr *addr, const char *password)
{
	int		 ret = -1;
	int		 save_errno = ENOSYS;
#if HAVE_DECL_TCP_MD5SIG
	union sockunion	 su;
#endif

	if (fd == -1)
		return (0);
#if HAVE_DECL_TCP_MD5SIG
	addr2sa(af, addr, 0, &su);

	if (ldpe_privs.change(ZPRIVS_RAISE)) {
		log_warn("%s: could not raise privs", __func__);
		return (-1);
	}
	ret = sockopt_tcp_signature(fd, &su, password);
	save_errno = errno;
	if (ldpe_privs.change(ZPRIVS_LOWER))
		log_warn("%s: could not lower privs", __func__);
#endif /* HAVE_TCP_MD5SIG */
	if (ret < 0)
		log_warnx("%s: can't set TCP_MD5SIG option on fd %d: %s",
		    __func__, fd, safe_strerror(save_errno));

	return (ret);
}
#endif

int
sock_set_ipv4_tos(int fd, int tos)
{
	if (setsockopt(fd, IPPROTO_IP, IP_TOS, (int *)&tos, sizeof(tos)) < 0) {
		log_warn("%s: error setting IP_TOS to 0x%x", __func__, tos);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv4_recvif(int fd, int enable)
{
	return (setsockopt_ifindex(AF_INET, fd, enable));
}

int
sock_set_ipv4_minttl(int fd, int ttl)
{
	return (sockopt_minttl(AF_INET, fd, ttl));
}

int
sock_set_ipv4_ucast_ttl(int fd, int ttl)
{
	if (setsockopt(fd, IPPROTO_IP, IP_TTL, &ttl, sizeof(ttl)) < 0) {
		log_warn("%s: error setting IP_TTL", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv4_mcast_ttl(int fd, uint8_t ttl)
{
	if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL,
	    (char *)&ttl, sizeof(ttl)) < 0) {
		log_warn("%s: error setting IP_MULTICAST_TTL to %d",
		    __func__, ttl);
		return (-1);
	}

	return (0);
}

#ifndef MSG_MCAST
#if defined(HAVE_IP_PKTINFO)
int
sock_set_ipv4_pktinfo(int fd, int enable)
{
	if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &enable,
	    sizeof(enable)) < 0) {
		log_warn("%s: error setting IP_PKTINFO", __func__);
		return (-1);
	}

	return (0);
}
#elif defined(HAVE_IP_RECVDSTADDR)
int
sock_set_ipv4_recvdstaddr(int fd, int enable)
{
	if (setsockopt(fd, IPPROTO_IP, IP_RECVDSTADDR, &enable,
	    sizeof(enable)) < 0) {
		log_warn("%s: error setting IP_RECVDSTADDR", __func__);
		return (-1);
	}

	return (0);
}
#else
#error "Unsupported socket API"
#endif
#endif /* MSG_MCAST */

int
sock_set_ipv4_mcast(struct iface *iface)
{
	struct in_addr		 if_addr;

	if_addr.s_addr = if_get_ipv4_addr(iface);

	if (setsockopt_ipv4_multicast_if(global.ipv4.ldp_disc_socket,
	    if_addr, iface->ifindex) < 0) {
		log_warn("%s: error setting IP_MULTICAST_IF, interface %s",
		    __func__, iface->name);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv4_mcast_loop(int fd)
{
	return (setsockopt_ipv4_multicast_loop(fd, 0));
}

int
sock_set_ipv6_dscp(int fd, int dscp)
{
	if (setsockopt(fd, IPPROTO_IPV6, IPV6_TCLASS, &dscp,
	    sizeof(dscp)) < 0) {
		log_warn("%s: error setting IPV6_TCLASS", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv6_pktinfo(int fd, int enable)
{
	if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &enable,
	    sizeof(enable)) < 0) {
		log_warn("%s: error setting IPV6_RECVPKTINFO", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv6_minhopcount(int fd, int hoplimit)
{
	return (sockopt_minttl(AF_INET6, fd, hoplimit));
}

int
sock_set_ipv6_ucast_hops(int fd, int hoplimit)
{
	if (setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
	    &hoplimit, sizeof(hoplimit)) < 0) {
		log_warn("%s: error setting IPV6_UNICAST_HOPS", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv6_mcast_hops(int fd, int hoplimit)
{
	if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
	    &hoplimit, sizeof(hoplimit)) < 0) {
		log_warn("%s: error setting IPV6_MULTICAST_HOPS", __func__);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv6_mcast(struct iface *iface)
{
	if (setsockopt(global.ipv6.ldp_disc_socket, IPPROTO_IPV6,
	    IPV6_MULTICAST_IF, &iface->ifindex, sizeof(iface->ifindex)) < 0) {
		log_warn("%s: error setting IPV6_MULTICAST_IF, interface %s",
		    __func__, iface->name);
		return (-1);
	}

	return (0);
}

int
sock_set_ipv6_mcast_loop(int fd)
{
	unsigned int	loop = 0;

	if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
	    &loop, sizeof(loop)) < 0) {
		log_warn("%s: error setting IPV6_MULTICAST_LOOP", __func__);
		return (-1);
	}

	return (0);
}
Commit	Line	Data
8429abe0 RW	1	/* $OpenBSD$ */
	2
	3	/*
	4	* Copyright (c) 2016 Renato Westphal <renato@openbsd.org>
	5	* Copyright (c) 2009 Michele Marchetto <michele@openbsd.org>
	6	* Copyright (c) 2005 Claudio Jeker <claudio@openbsd.org>
	7	* Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
	8	*
	9	* Permission to use, copy, modify, and distribute this software for any
	10	* purpose with or without fee is hereby granted, provided that the above
	11	* copyright notice and this permission notice appear in all copies.
	12	*
	13	* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
	14	* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
	15	* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
	16	* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
	17	* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
	18	* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
	19	* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
	20	*/
	21
eac6e3f0	22	#include <zebra.h>
8429abe0 RW	23
	24	#include "ldpd.h"
	25	#include "ldpe.h"
	26	#include "log.h"
	27
eac6e3f0 RW	28	#include "lib/log.h"
	29	#include "privs.h"
	30	#include "sockopt.h"
	31
	32	extern struct zebra_privs_t ldpd_privs;
	33	extern struct zebra_privs_t ldpe_privs;
	34
8429abe0 RW	35	int
	36	ldp_create_socket(int af, enum socket_type type)
	37	{
	38	int fd, domain, proto;
	39	union ldpd_addr addr;
4149ef7c	40	union sockunion local_su;
eac6e3f0	41	#ifdef __OpenBSD__
8429abe0	42	int opt;
eac6e3f0 RW	43	#endif
eac6e3f0 RW	44	int save_errno;
8429abe0 RW	45
	46	/* create socket */
	47	switch (type) {
	48	case LDP_SOCKET_DISC:
	49	case LDP_SOCKET_EDISC:
	50	domain = SOCK_DGRAM;
	51	proto = IPPROTO_UDP;
	52	break;
	53	case LDP_SOCKET_SESSION:
	54	domain = SOCK_STREAM;
	55	proto = IPPROTO_TCP;
	56	break;
	57	default:
	58	fatalx("ldp_create_socket: unknown socket type");
	59	}
eac6e3f0	60	fd = socket(af, domain, proto);
8429abe0 RW	61	if (fd == -1) {
	62	log_warn("%s: error creating socket", __func__);
	63	return (-1);
	64	}
eac6e3f0 RW	65	sock_set_nonblock(fd);
eac6e3f0 RW	66	sockopt_v6only(af, fd);
8429abe0 RW	67
	68	/* bind to a local address/port */
	69	switch (type) {
	70	case LDP_SOCKET_DISC:
	71	/* listen on all addresses */
	72	memset(&addr, 0, sizeof(addr));
4149ef7c	73	addr2sa(af, &addr, LDP_PORT, &local_su);
8429abe0 RW	74	break;
	75	case LDP_SOCKET_EDISC:
	76	case LDP_SOCKET_SESSION:
	77	addr = (ldp_af_conf_get(ldpd_conf, af))->trans_addr;
4149ef7c	78	addr2sa(af, &addr, LDP_PORT, &local_su);
eac6e3f0 RW	79	/* ignore any possible error */
eac6e3f0 RW	80	sock_set_bindany(fd, 1);
8429abe0 RW	81	break;
8429abe0 RW	82	}
eac6e3f0 RW	83	if (ldpd_privs.change(ZPRIVS_RAISE))
eac6e3f0 RW	84	log_warn("%s: could not raise privs", __func__);
8429abe0	85	if (sock_set_reuse(fd, 1) == -1) {
07bc3833 RW	86	if (ldpd_privs.change(ZPRIVS_LOWER))
07bc3833 RW	87	log_warn("%s: could not lower privs", __func__);
8429abe0 RW	88	close(fd);
	89	return (-1);
	90	}
4149ef7c	91	if (bind(fd, &local_su.sa, sockaddr_len(&local_su.sa)) == -1) {
eac6e3f0 RW	92	save_errno = errno;
	93	if (ldpd_privs.change(ZPRIVS_LOWER))
	94	log_warn("%s: could not lower privs", __func__);
	95	log_warnx("%s: error binding socket: %s", __func__,
	96	safe_strerror(save_errno));
8429abe0 RW	97	close(fd);
	98	return (-1);
	99	}
eac6e3f0 RW	100	if (ldpd_privs.change(ZPRIVS_LOWER))
eac6e3f0 RW	101	log_warn("%s: could not lower privs", __func__);
8429abe0 RW	102
	103	/* set options */
	104	switch (af) {
	105	case AF_INET:
	106	if (sock_set_ipv4_tos(fd, IPTOS_PREC_INTERNETCONTROL) == -1) {
	107	close(fd);
	108	return (-1);
	109	}
	110	if (type == LDP_SOCKET_DISC) {
	111	if (sock_set_ipv4_mcast_ttl(fd,
	112	IP_DEFAULT_MULTICAST_TTL) == -1) {
	113	close(fd);
	114	return (-1);
	115	}
	116	if (sock_set_ipv4_mcast_loop(fd) == -1) {
	117	close(fd);
	118	return (-1);
	119	}
	120	}
	121	if (type == LDP_SOCKET_DISC \|\| type == LDP_SOCKET_EDISC) {
	122	if (sock_set_ipv4_recvif(fd, 1) == -1) {
	123	close(fd);
	124	return (-1);
	125	}
eac6e3f0 RW	126	#ifndef MSG_MCAST
	127	#if defined(HAVE_IP_PKTINFO)
	128	if (sock_set_ipv4_pktinfo(fd, 1) == -1) {
	129	close(fd);
	130	return (-1);
	131	}
	132	#elif defined(HAVE_IP_RECVDSTADDR)
	133	if (sock_set_ipv4_recvdstaddr(fd, 1) == -1) {
	134	close(fd);
	135	return (-1);
	136	}
	137	#else
	138	#error "Unsupported socket API"
	139	#endif
	140	#endif /* MSG_MCAST */
8429abe0 RW	141	}
	142	if (type == LDP_SOCKET_SESSION) {
	143	if (sock_set_ipv4_ucast_ttl(fd, 255) == -1) {
	144	close(fd);
	145	return (-1);
	146	}
	147	}
	148	break;
	149	case AF_INET6:
	150	if (sock_set_ipv6_dscp(fd, IPTOS_PREC_INTERNETCONTROL) == -1) {
	151	close(fd);
	152	return (-1);
	153	}
	154	if (type == LDP_SOCKET_DISC) {
	155	if (sock_set_ipv6_mcast_loop(fd) == -1) {
	156	close(fd);
	157	return (-1);
	158	}
	159	if (sock_set_ipv6_mcast_hops(fd, 255) == -1) {
	160	close(fd);
	161	return (-1);
	162	}
	163	if (!(ldpd_conf->ipv6.flags & F_LDPD_AF_NO_GTSM)) {
eac6e3f0 RW	164	/* ignore any possible error */
eac6e3f0 RW	165	sock_set_ipv6_minhopcount(fd, 255);
8429abe0 RW	166	}
	167	}
	168	if (type == LDP_SOCKET_DISC \|\| type == LDP_SOCKET_EDISC) {
	169	if (sock_set_ipv6_pktinfo(fd, 1) == -1) {
	170	close(fd);
	171	return (-1);
	172	}
	173	}
	174	if (type == LDP_SOCKET_SESSION) {
	175	if (sock_set_ipv6_ucast_hops(fd, 255) == -1) {
	176	close(fd);
	177	return (-1);
	178	}
	179	}
	180	break;
	181	}
	182	switch (type) {
	183	case LDP_SOCKET_DISC:
	184	case LDP_SOCKET_EDISC:
	185	sock_set_recvbuf(fd);
	186	break;
	187	case LDP_SOCKET_SESSION:
	188	if (listen(fd, LDP_BACKLOG) == -1)
	189	log_warn("%s: error listening on socket", __func__);
	190
eac6e3f0	191	#ifdef __OpenBSD__
8429abe0 RW	192	opt = 1;
	193	if (setsockopt(fd, IPPROTO_TCP, TCP_MD5SIG, &opt,
	194	sizeof(opt)) == -1) {
	195	if (errno == ENOPROTOOPT) { /* system w/o md5sig */
	196	log_warnx("md5sig not available, disabling");
	197	sysdep.no_md5sig = 1;
	198	} else {
	199	close(fd);
	200	return (-1);
	201	}
	202	}
eac6e3f0	203	#endif
8429abe0 RW	204	break;
	205	}
	206
	207	return (fd);
	208	}
	209
eac6e3f0 RW	210	void
	211	sock_set_nonblock(int fd)
	212	{
	213	int flags;
	214
	215	if ((flags = fcntl(fd, F_GETFL, 0)) == -1)
	216	fatal("fcntl F_GETFL");
	217
	218	flags \|= O_NONBLOCK;
	219
	220	if ((flags = fcntl(fd, F_SETFL, flags)) == -1)
	221	fatal("fcntl F_SETFL");
	222	}
	223
	224	void
	225	sock_set_cloexec(int fd)
	226	{
	227	int flags;
	228
	229	if ((flags = fcntl(fd, F_GETFD, 0)) == -1)
	230	fatal("fcntl F_GETFD");
	231
	232	flags \|= FD_CLOEXEC;
	233
	234	if ((flags = fcntl(fd, F_SETFD, flags)) == -1)
	235	fatal("fcntl F_SETFD");
	236	}
	237
8429abe0 RW	238	void
	239	sock_set_recvbuf(int fd)
	240	{
	241	int bsize;
	242
	243	bsize = 65535;
	244	while (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &bsize,
	245	sizeof(bsize)) == -1)
	246	bsize /= 2;
	247	}
	248
	249	int
	250	sock_set_reuse(int fd, int enable)
	251	{
	252	if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &enable,
	253	sizeof(int)) < 0) {
	254	log_warn("%s: error setting SO_REUSEADDR", __func__);
	255	return (-1);
	256	}
	257
	258	return (0);
	259	}
	260
	261	int
	262	sock_set_bindany(int fd, int enable)
	263	{
eac6e3f0	264	#ifdef HAVE_SO_BINDANY
01b9e3fd DL	265	frr_elevate_privs(&ldpd_privs) {
	266	if (setsockopt(fd, SOL_SOCKET, SO_BINDANY, &enable,
	267	sizeof(int)) < 0) {
	268	log_warn("%s: error setting SO_BINDANY", __func__);
	269	return (-1);
	270	}
8429abe0	271	}
eac6e3f0 RW	272	return (0);
	273	#elif defined(HAVE_IP_FREEBIND)
	274	if (setsockopt(fd, IPPROTO_IP, IP_FREEBIND, &enable, sizeof(int)) < 0) {
	275	log_warn("%s: error setting IP_FREEBIND", __func__);
	276	return (-1);
	277	}
8429abe0	278	return (0);
eac6e3f0 RW	279	#else
	280	log_warnx("%s: missing SO_BINDANY and IP_FREEBIND, unable to bind "
	281	"to a nonlocal IP address", __func__);
	282	return (-1);
	283	#endif /* HAVE_SO_BINDANY */
	284	}
	285
	286	#ifndef __OpenBSD__
	287	/*
	288	* Set MD5 key for the socket, for the given peer address. If the password
	289	* is NULL or zero-length, the option will be disabled.
	290	*/
	291	int
	292	sock_set_md5sig(int fd, int af, union ldpd_addr addr, const char password)
	293	{
	294	int ret = -1;
	295	int save_errno = ENOSYS;
	296	#if HAVE_DECL_TCP_MD5SIG
	297	union sockunion su;
	298	#endif
	299
	300	if (fd == -1)
	301	return (0);
	302	#if HAVE_DECL_TCP_MD5SIG
4149ef7c	303	addr2sa(af, addr, 0, &su);
eac6e3f0 RW	304
	305	if (ldpe_privs.change(ZPRIVS_RAISE)) {
	306	log_warn("%s: could not raise privs", __func__);
	307	return (-1);
	308	}
	309	ret = sockopt_tcp_signature(fd, &su, password);
	310	save_errno = errno;
	311	if (ldpe_privs.change(ZPRIVS_LOWER))
	312	log_warn("%s: could not lower privs", __func__);
	313	#endif /* HAVE_TCP_MD5SIG */
	314	if (ret < 0)
	315	log_warnx("%s: can't set TCP_MD5SIG option on fd %d: %s",
	316	__func__, fd, safe_strerror(save_errno));
	317
	318	return (ret);
8429abe0	319	}
eac6e3f0	320	#endif
8429abe0 RW	321
	322	int
	323	sock_set_ipv4_tos(int fd, int tos)
	324	{
	325	if (setsockopt(fd, IPPROTO_IP, IP_TOS, (int *)&tos, sizeof(tos)) < 0) {
	326	log_warn("%s: error setting IP_TOS to 0x%x", __func__, tos);
	327	return (-1);
	328	}
	329
	330	return (0);
	331	}
	332
	333	int
	334	sock_set_ipv4_recvif(int fd, int enable)
	335	{
eac6e3f0	336	return (setsockopt_ifindex(AF_INET, fd, enable));
8429abe0 RW	337	}
	338
	339	int
	340	sock_set_ipv4_minttl(int fd, int ttl)
	341	{
eac6e3f0	342	return (sockopt_minttl(AF_INET, fd, ttl));
8429abe0 RW	343	}
	344
	345	int
	346	sock_set_ipv4_ucast_ttl(int fd, int ttl)
	347	{
	348	if (setsockopt(fd, IPPROTO_IP, IP_TTL, &ttl, sizeof(ttl)) < 0) {
	349	log_warn("%s: error setting IP_TTL", __func__);
	350	return (-1);
	351	}
	352
	353	return (0);
	354	}
	355
	356	int
	357	sock_set_ipv4_mcast_ttl(int fd, uint8_t ttl)
	358	{
	359	if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL,
	360	(char *)&ttl, sizeof(ttl)) < 0) {
	361	log_warn("%s: error setting IP_MULTICAST_TTL to %d",
	362	__func__, ttl);
	363	return (-1);
	364	}
	365
	366	return (0);
	367	}
	368
eac6e3f0 RW	369	#ifndef MSG_MCAST
	370	#if defined(HAVE_IP_PKTINFO)
	371	int
	372	sock_set_ipv4_pktinfo(int fd, int enable)
	373	{
	374	if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &enable,
	375	sizeof(enable)) < 0) {
	376	log_warn("%s: error setting IP_PKTINFO", __func__);
	377	return (-1);
	378	}
	379
	380	return (0);
	381	}
	382	#elif defined(HAVE_IP_RECVDSTADDR)
	383	int
	384	sock_set_ipv4_recvdstaddr(int fd, int enable)
	385	{
	386	if (setsockopt(fd, IPPROTO_IP, IP_RECVDSTADDR, &enable,
	387	sizeof(enable)) < 0) {
	388	log_warn("%s: error setting IP_RECVDSTADDR", __func__);
	389	return (-1);
	390	}
	391
	392	return (0);
	393	}
	394	#else
	395	#error "Unsupported socket API"
	396	#endif
	397	#endif /* MSG_MCAST */
	398
8429abe0 RW	399	int
	400	sock_set_ipv4_mcast(struct iface *iface)
	401	{
eac6e3f0	402	struct in_addr if_addr;
8429abe0	403
eac6e3f0	404	if_addr.s_addr = if_get_ipv4_addr(iface);
8429abe0	405
eac6e3f0 RW	406	if (setsockopt_ipv4_multicast_if(global.ipv4.ldp_disc_socket,
eac6e3f0 RW	407	if_addr, iface->ifindex) < 0) {
8429abe0 RW	408	log_warn("%s: error setting IP_MULTICAST_IF, interface %s",
	409	__func__, iface->name);
	410	return (-1);
	411	}
	412
	413	return (0);
	414	}
	415
	416	int
	417	sock_set_ipv4_mcast_loop(int fd)
	418	{
c5bdb09f	419	return (setsockopt_ipv4_multicast_loop(fd, 0));
8429abe0 RW	420	}
	421
	422	int
	423	sock_set_ipv6_dscp(int fd, int dscp)
	424	{
	425	if (setsockopt(fd, IPPROTO_IPV6, IPV6_TCLASS, &dscp,
	426	sizeof(dscp)) < 0) {
	427	log_warn("%s: error setting IPV6_TCLASS", __func__);
	428	return (-1);
	429	}
	430
	431	return (0);
	432	}
	433
	434	int
	435	sock_set_ipv6_pktinfo(int fd, int enable)
	436	{
	437	if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &enable,
	438	sizeof(enable)) < 0) {
	439	log_warn("%s: error setting IPV6_RECVPKTINFO", __func__);
	440	return (-1);
	441	}
	442
	443	return (0);
	444	}
	445
	446	int
	447	sock_set_ipv6_minhopcount(int fd, int hoplimit)
	448	{
eac6e3f0	449	return (sockopt_minttl(AF_INET6, fd, hoplimit));
8429abe0 RW	450	}
	451
	452	int
	453	sock_set_ipv6_ucast_hops(int fd, int hoplimit)
	454	{
	455	if (setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
	456	&hoplimit, sizeof(hoplimit)) < 0) {
	457	log_warn("%s: error setting IPV6_UNICAST_HOPS", __func__);
	458	return (-1);
	459	}
	460
	461	return (0);
	462	}
	463
	464	int
	465	sock_set_ipv6_mcast_hops(int fd, int hoplimit)
	466	{
	467	if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
	468	&hoplimit, sizeof(hoplimit)) < 0) {
	469	log_warn("%s: error setting IPV6_MULTICAST_HOPS", __func__);
	470	return (-1);
	471	}
	472
	473	return (0);
	474	}
	475
	476	int
	477	sock_set_ipv6_mcast(struct iface *iface)
	478	{
	479	if (setsockopt(global.ipv6.ldp_disc_socket, IPPROTO_IPV6,
	480	IPV6_MULTICAST_IF, &iface->ifindex, sizeof(iface->ifindex)) < 0) {
	481	log_warn("%s: error setting IPV6_MULTICAST_IF, interface %s",
	482	__func__, iface->name);
	483	return (-1);
	484	}
	485
	486	return (0);
	487	}
	488
	489	int
	490	sock_set_ipv6_mcast_loop(int fd)
	491	{
	492	unsigned int loop = 0;
	493
	494	if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
	495	&loop, sizeof(loop)) < 0) {
	496	log_warn("%s: error setting IPV6_MULTICAST_LOOP", __func__);
	497	return (-1);
	498	}
	499
	500	return (0);
	501	}