[mirror_frr.git] / pimd / pim_msdp_socket.c

/*
 * IP MSDP socket management
 * Copyright (C) 2016 Cumulus Networks, Inc.
 *
 * This program 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 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; see the file COPYING; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <zebra.h>

#include <lib/log.h>
#include <lib/network.h>
#include <lib/sockunion.h>
#include <lib/thread.h>
#include <lib/vty.h>
#include <lib/if.h>
#include <lib/vrf.h>

#include "pimd.h"
#include "pim_sock.h"

#include "pim_msdp.h"
#include "pim_msdp_socket.h"

/* increase socket send buffer size */
static void pim_msdp_update_sock_send_buffer_size(int fd)
{
	int size = PIM_MSDP_SOCKET_SNDBUF_SIZE;
	int optval;
	socklen_t optlen = sizeof(optval);

	if (getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &optval, &optlen) < 0) {
		zlog_err("getsockopt of SO_SNDBUF failed %s\n",
			 safe_strerror(errno));
		return;
	}

	if (optval < size) {
		if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &size, sizeof(size))
		    < 0) {
			zlog_err("Couldn't increase send buffer: %s\n",
				 safe_strerror(errno));
		}
	}
}

/* passive peer socket accept */
static int pim_msdp_sock_accept(struct thread *thread)
{
	union sockunion su;
	struct pim_instance *pim = THREAD_ARG(thread);
	int accept_sock;
	int msdp_sock;
	struct pim_msdp_peer *mp;
	char buf[SU_ADDRSTRLEN];

	sockunion_init(&su);

	/* re-register accept thread */
	accept_sock = THREAD_FD(thread);
	if (accept_sock < 0) {
		zlog_err("accept_sock is negative value %d", accept_sock);
		return -1;
	}
	pim->msdp.listener.thread = NULL;
	thread_add_read(master, pim_msdp_sock_accept, pim, accept_sock,
			&pim->msdp.listener.thread);

	/* accept client connection. */
	msdp_sock = sockunion_accept(accept_sock, &su);
	if (msdp_sock < 0) {
		zlog_err("pim_msdp_sock_accept failed (%s)",
			 safe_strerror(errno));
		return -1;
	}

	/* see if have peer config for this */
	mp = pim_msdp_peer_find(pim, su.sin.sin_addr);
	if (!mp || !PIM_MSDP_PEER_IS_LISTENER(mp)) {
		++pim->msdp.rejected_accepts;
		if (PIM_DEBUG_MSDP_EVENTS) {
			zlog_err("msdp peer connection refused from %s",
				 sockunion2str(&su, buf, SU_ADDRSTRLEN));
		}
		close(msdp_sock);
		return -1;
	}

	if (PIM_DEBUG_MSDP_INTERNAL) {
		zlog_debug("MSDP peer %s accept success%s", mp->key_str,
			   mp->fd >= 0 ? "(dup)" : "");
	}

	/* if we have an existing connection we need to kill that one
	 * with this one */
	if (mp->fd >= 0) {
		if (PIM_DEBUG_MSDP_EVENTS) {
			zlog_err(
				"msdp peer new connection from %s stop old connection",
				sockunion2str(&su, buf, SU_ADDRSTRLEN));
		}
		pim_msdp_peer_stop_tcp_conn(mp, true /* chg_state */);
	}
	mp->fd = msdp_sock;
	set_nonblocking(mp->fd);
	pim_msdp_update_sock_send_buffer_size(mp->fd);
	pim_msdp_peer_established(mp);
	return 0;
}

/* global listener for the MSDP well know TCP port */
int pim_msdp_sock_listen(struct pim_instance *pim)
{
	int sock;
	int socklen;
	struct sockaddr_in sin;
	int rc;
	struct pim_msdp_listener *listener = &pim->msdp.listener;

	if (pim->msdp.flags & PIM_MSDPF_LISTENER) {
		/* listener already setup */
		return 0;
	}

	sock = socket(AF_INET, SOCK_STREAM, 0);
	if (sock < 0) {
		zlog_err("socket: %s", safe_strerror(errno));
		return sock;
	}

	memset(&sin, 0, sizeof(struct sockaddr_in));
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PIM_MSDP_TCP_PORT);
	socklen = sizeof(struct sockaddr_in);
#ifdef HAVE_STRUCT_SOCKADDR_IN_SIN_LEN
	sin.sin_len = socklen;
#endif /* HAVE_STRUCT_SOCKADDR_IN_SIN_LEN */

	sockopt_reuseaddr(sock);
	sockopt_reuseport(sock);

	if (pim->vrf_id != VRF_DEFAULT) {
		struct interface *ifp =
			if_lookup_by_name(pim->vrf->name, pim->vrf_id);
		if (!ifp) {
			zlog_err("%s: Unable to lookup vrf interface: %s",
				 __PRETTY_FUNCTION__, pim->vrf->name);
			close(sock);
			return -1;
		}
		if (pim_socket_bind(sock, ifp)) {
			zlog_err("%s: Unable to bind to socket: %s",
				 __PRETTY_FUNCTION__, safe_strerror(errno));
			close(sock);
			return -1;
		}
	}

	if (pimd_privs.change(ZPRIVS_RAISE)) {
		zlog_err("pim_msdp_socket: could not raise privs, %s",
			 safe_strerror(errno));
	}

	/* bind to well known TCP port */
	rc = bind(sock, (struct sockaddr *)&sin, socklen);

	if (pimd_privs.change(ZPRIVS_LOWER)) {
		zlog_err("pim_msdp_socket: could not lower privs, %s",
			 safe_strerror(errno));
	}

	if (rc < 0) {
		zlog_err("pim_msdp_socket bind to port %d: %s",
			 ntohs(sin.sin_port), safe_strerror(errno));
		close(sock);
		return rc;
	}

	rc = listen(sock, 3 /* backlog */);
	if (rc < 0) {
		zlog_err("pim_msdp_socket listen: %s", safe_strerror(errno));
		close(sock);
		return rc;
	}

	/* add accept thread */
	listener->fd = sock;
	memcpy(&listener->su, &sin, socklen);
	listener->thread = NULL;
	thread_add_read(pim->msdp.master, pim_msdp_sock_accept, pim, sock,
			&listener->thread);

	pim->msdp.flags |= PIM_MSDPF_LISTENER;
	return 0;
}

/* active peer socket setup */
int pim_msdp_sock_connect(struct pim_msdp_peer *mp)
{
	int rc;

	if (PIM_DEBUG_MSDP_INTERNAL) {
		zlog_debug("MSDP peer %s attempt connect%s", mp->key_str,
			   mp->fd < 0 ? "" : "(dup)");
	}

	/* if we have an existing connection we need to kill that one
	 * with this one */
	if (mp->fd >= 0) {
		if (PIM_DEBUG_MSDP_EVENTS) {
			zlog_err(
				"msdp duplicate connect to %s nuke old connection",
				mp->key_str);
		}
		pim_msdp_peer_stop_tcp_conn(mp, false /* chg_state */);
	}

	/* Make socket for the peer. */
	mp->fd = sockunion_socket(&mp->su_peer);
	if (mp->fd < 0) {
		zlog_err("pim_msdp_socket socket failure: %s",
			 safe_strerror(errno));
		return -1;
	}

	if (mp->pim->vrf_id != VRF_DEFAULT) {
		struct interface *ifp =
			if_lookup_by_name(mp->pim->vrf->name, mp->pim->vrf_id);
		if (!ifp) {
			zlog_err("%s: Unable to lookup vrf interface: %s",
				 __PRETTY_FUNCTION__, mp->pim->vrf->name);
			return -1;
		}
		if (pim_socket_bind(mp->fd, ifp)) {
			zlog_err("%s: Unable to bind to socket: %s",
				 __PRETTY_FUNCTION__, safe_strerror(errno));
			close(mp->fd);
			mp->fd = -1;
			return -1;
		}
	}

	set_nonblocking(mp->fd);

	/* Set socket send buffer size */
	pim_msdp_update_sock_send_buffer_size(mp->fd);
	sockopt_reuseaddr(mp->fd);
	sockopt_reuseport(mp->fd);

	/* source bind */
	rc = sockunion_bind(mp->fd, &mp->su_local, 0, &mp->su_local);
	if (rc < 0) {
		zlog_err("pim_msdp_socket connect bind failure: %s",
			 safe_strerror(errno));
		close(mp->fd);
		mp->fd = -1;
		return rc;
	}

	/* Connect to the remote mp. */
	return (sockunion_connect(mp->fd, &mp->su_peer,
				  htons(PIM_MSDP_TCP_PORT), 0));
}
Commit	Line	Data
2a333e0f	1	/*
	2	* IP MSDP socket management
	3	* Copyright (C) 2016 Cumulus Networks, Inc.
	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; either version 2 of the License, or
	8	* (at your option) any later version.
	9	*
	10	* This program is distributed in the hope that it will be useful, but
	11	* WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	* General Public License for more details.
	14	*
896014f4 DL	15	* You should have received a copy of the GNU General Public License along
	16	* with this program; see the file COPYING; if not, write to the Free Software
	17	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
2a333e0f	18	*/
	19
	20	#include <zebra.h>
	21
	22	#include <lib/log.h>
	23	#include <lib/network.h>
2a333e0f	24	#include <lib/sockunion.h>
3c72d654	25	#include <lib/thread.h>
3c72d654	26	#include <lib/vty.h>
62fde409 DS	27	#include <lib/if.h>
62fde409 DS	28	#include <lib/vrf.h>
2a333e0f	29
2a333e0f	30	#include "pimd.h"
62fde409	31	#include "pim_sock.h"
2a333e0f	32
	33	#include "pim_msdp.h"
	34	#include "pim_msdp_socket.h"
	35
2a333e0f	36	/* increase socket send buffer size */
d62a17ae	37	static void pim_msdp_update_sock_send_buffer_size(int fd)
2a333e0f	38	{
d62a17ae	39	int size = PIM_MSDP_SOCKET_SNDBUF_SIZE;
	40	int optval;
	41	socklen_t optlen = sizeof(optval);
	42
	43	if (getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &optval, &optlen) < 0) {
	44	zlog_err("getsockopt of SO_SNDBUF failed %s\n",
	45	safe_strerror(errno));
	46	return;
	47	}
	48
	49	if (optval < size) {
	50	if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &size, sizeof(size))
	51	< 0) {
	52	zlog_err("Couldn't increase send buffer: %s\n",
	53	safe_strerror(errno));
	54	}
	55	}
2a333e0f	56	}
	57
	58	/* passive peer socket accept */
d62a17ae	59	static int pim_msdp_sock_accept(struct thread *thread)
2a333e0f	60	{
d62a17ae	61	union sockunion su;
472ad383	62	struct pim_instance *pim = THREAD_ARG(thread);
d62a17ae	63	int accept_sock;
	64	int msdp_sock;
	65	struct pim_msdp_peer *mp;
	66	char buf[SU_ADDRSTRLEN];
	67
	68	sockunion_init(&su);
	69
	70	/* re-register accept thread */
	71	accept_sock = THREAD_FD(thread);
	72	if (accept_sock < 0) {
	73	zlog_err("accept_sock is negative value %d", accept_sock);
	74	return -1;
	75	}
472ad383 DS	76	pim->msdp.listener.thread = NULL;
	77	thread_add_read(master, pim_msdp_sock_accept, pim, accept_sock,
	78	&pim->msdp.listener.thread);
d62a17ae	79
	80	/* accept client connection. */
	81	msdp_sock = sockunion_accept(accept_sock, &su);
	82	if (msdp_sock < 0) {
	83	zlog_err("pim_msdp_sock_accept failed (%s)",
	84	safe_strerror(errno));
	85	return -1;
	86	}
	87
	88	/* see if have peer config for this */
472ad383	89	mp = pim_msdp_peer_find(pim, su.sin.sin_addr);
d62a17ae	90	if (!mp \|\| !PIM_MSDP_PEER_IS_LISTENER(mp)) {
472ad383	91	++pim->msdp.rejected_accepts;
d62a17ae	92	if (PIM_DEBUG_MSDP_EVENTS) {
	93	zlog_err("msdp peer connection refused from %s",
	94	sockunion2str(&su, buf, SU_ADDRSTRLEN));
	95	}
	96	close(msdp_sock);
	97	return -1;
	98	}
	99
	100	if (PIM_DEBUG_MSDP_INTERNAL) {
	101	zlog_debug("MSDP peer %s accept success%s", mp->key_str,
	102	mp->fd >= 0 ? "(dup)" : "");
	103	}
	104
	105	/* if we have an existing connection we need to kill that one
	106	* with this one */
	107	if (mp->fd >= 0) {
	108	if (PIM_DEBUG_MSDP_EVENTS) {
	109	zlog_err(
	110	"msdp peer new connection from %s stop old connection",
	111	sockunion2str(&su, buf, SU_ADDRSTRLEN));
	112	}
	113	pim_msdp_peer_stop_tcp_conn(mp, true /* chg_state */);
	114	}
	115	mp->fd = msdp_sock;
	116	set_nonblocking(mp->fd);
	117	pim_msdp_update_sock_send_buffer_size(mp->fd);
	118	pim_msdp_peer_established(mp);
	119	return 0;
2a333e0f	120	}
	121
	122	/* global listener for the MSDP well know TCP port */
472ad383	123	int pim_msdp_sock_listen(struct pim_instance *pim)
2a333e0f	124	{
d62a17ae	125	int sock;
	126	int socklen;
	127	struct sockaddr_in sin;
	128	int rc;
472ad383	129	struct pim_msdp_listener *listener = &pim->msdp.listener;
d62a17ae	130
472ad383	131	if (pim->msdp.flags & PIM_MSDPF_LISTENER) {
d62a17ae	132	/* listener already setup */
	133	return 0;
	134	}
	135
	136	sock = socket(AF_INET, SOCK_STREAM, 0);
	137	if (sock < 0) {
	138	zlog_err("socket: %s", safe_strerror(errno));
	139	return sock;
	140	}
	141
	142	memset(&sin, 0, sizeof(struct sockaddr_in));
	143	sin.sin_family = AF_INET;
	144	sin.sin_port = htons(PIM_MSDP_TCP_PORT);
	145	socklen = sizeof(struct sockaddr_in);
2a333e0f	146	#ifdef HAVE_STRUCT_SOCKADDR_IN_SIN_LEN
d62a17ae	147	sin.sin_len = socklen;
2a333e0f	148	#endif /* HAVE_STRUCT_SOCKADDR_IN_SIN_LEN */
2a333e0f	149
d62a17ae	150	sockopt_reuseaddr(sock);
	151	sockopt_reuseport(sock);
	152
2267994c DS	153	if (pim->vrf_id != VRF_DEFAULT) {
	154	struct interface *ifp =
	155	if_lookup_by_name(pim->vrf->name, pim->vrf_id);
449a3f5b DS	156	if (!ifp) {
	157	zlog_err("%s: Unable to lookup vrf interface: %s",
	158	__PRETTY_FUNCTION__, pim->vrf->name);
e691f179 DS	159	close(sock);
	160	return -1;
	161	}
	162	if (pim_socket_bind(sock, ifp)) {
	163	zlog_err("%s: Unable to bind to socket: %s",
	164	__PRETTY_FUNCTION__, safe_strerror(errno));
	165	close(sock);
449a3f5b DS	166	return -1;
449a3f5b DS	167	}
2267994c DS	168	}
2267994c DS	169
d62a17ae	170	if (pimd_privs.change(ZPRIVS_RAISE)) {
	171	zlog_err("pim_msdp_socket: could not raise privs, %s",
	172	safe_strerror(errno));
	173	}
	174
	175	/* bind to well known TCP port */
	176	rc = bind(sock, (struct sockaddr *)&sin, socklen);
	177
	178	if (pimd_privs.change(ZPRIVS_LOWER)) {
	179	zlog_err("pim_msdp_socket: could not lower privs, %s",
	180	safe_strerror(errno));
	181	}
	182
	183	if (rc < 0) {
	184	zlog_err("pim_msdp_socket bind to port %d: %s",
	185	ntohs(sin.sin_port), safe_strerror(errno));
	186	close(sock);
	187	return rc;
	188	}
	189
	190	rc = listen(sock, 3 /* backlog */);
	191	if (rc < 0) {
	192	zlog_err("pim_msdp_socket listen: %s", safe_strerror(errno));
	193	close(sock);
	194	return rc;
	195	}
	196
	197	/* add accept thread */
	198	listener->fd = sock;
	199	memcpy(&listener->su, &sin, socklen);
	200	listener->thread = NULL;
472ad383	201	thread_add_read(pim->msdp.master, pim_msdp_sock_accept, pim, sock,
d62a17ae	202	&listener->thread);
d62a17ae	203
472ad383	204	pim->msdp.flags \|= PIM_MSDPF_LISTENER;
d62a17ae	205	return 0;
2a333e0f	206	}
	207
	208	/* active peer socket setup */
d62a17ae	209	int pim_msdp_sock_connect(struct pim_msdp_peer *mp)
2a333e0f	210	{
d62a17ae	211	int rc;
	212
	213	if (PIM_DEBUG_MSDP_INTERNAL) {
	214	zlog_debug("MSDP peer %s attempt connect%s", mp->key_str,
	215	mp->fd < 0 ? "" : "(dup)");
	216	}
	217
	218	/* if we have an existing connection we need to kill that one
	219	* with this one */
	220	if (mp->fd >= 0) {
	221	if (PIM_DEBUG_MSDP_EVENTS) {
	222	zlog_err(
	223	"msdp duplicate connect to %s nuke old connection",
	224	mp->key_str);
	225	}
	226	pim_msdp_peer_stop_tcp_conn(mp, false /* chg_state */);
	227	}
	228
	229	/* Make socket for the peer. */
	230	mp->fd = sockunion_socket(&mp->su_peer);
	231	if (mp->fd < 0) {
	232	zlog_err("pim_msdp_socket socket failure: %s",
	233	safe_strerror(errno));
	234	return -1;
	235	}
	236
62fde409 DS	237	if (mp->pim->vrf_id != VRF_DEFAULT) {
	238	struct interface *ifp =
	239	if_lookup_by_name(mp->pim->vrf->name, mp->pim->vrf_id);
449a3f5b DS	240	if (!ifp) {
	241	zlog_err("%s: Unable to lookup vrf interface: %s",
	242	__PRETTY_FUNCTION__, mp->pim->vrf->name);
	243	return -1;
	244	}
e691f179 DS	245	if (pim_socket_bind(mp->fd, ifp)) {
	246	zlog_err("%s: Unable to bind to socket: %s",
	247	__PRETTY_FUNCTION__, safe_strerror(errno));
	248	close(mp->fd);
	249	mp->fd = -1;
	250	return -1;
	251	}
62fde409 DS	252	}
62fde409 DS	253
d62a17ae	254	set_nonblocking(mp->fd);
	255
	256	/* Set socket send buffer size */
	257	pim_msdp_update_sock_send_buffer_size(mp->fd);
	258	sockopt_reuseaddr(mp->fd);
	259	sockopt_reuseport(mp->fd);
	260
	261	/* source bind */
	262	rc = sockunion_bind(mp->fd, &mp->su_local, 0, &mp->su_local);
	263	if (rc < 0) {
	264	zlog_err("pim_msdp_socket connect bind failure: %s",
	265	safe_strerror(errno));
	266	close(mp->fd);
	267	mp->fd = -1;
	268	return rc;
	269	}
	270
	271	/* Connect to the remote mp. */
	272	return (sockunion_connect(mp->fd, &mp->su_peer,
	273	htons(PIM_MSDP_TCP_PORT), 0));
2a333e0f	274	}