mirror of
				https://git.proxmox.com/git/mirror_frr
				synced 2025-10-31 20:23:36 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			3130 lines
		
	
	
		
			78 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			3130 lines
		
	
	
		
			78 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* Zebra's client library.
 | |
|  * Copyright (C) 1999 Kunihiro Ishiguro
 | |
|  * Copyright (C) 2005 Andrew J. Schorr
 | |
|  *
 | |
|  * 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 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 "prefix.h"
 | |
| #include "stream.h"
 | |
| #include "buffer.h"
 | |
| #include "network.h"
 | |
| #include "vrf.h"
 | |
| #include "vrf_int.h"
 | |
| #include "if.h"
 | |
| #include "log.h"
 | |
| #include "thread.h"
 | |
| #include "zclient.h"
 | |
| #include "memory.h"
 | |
| #include "table.h"
 | |
| #include "nexthop.h"
 | |
| #include "mpls.h"
 | |
| #include "sockopt.h"
 | |
| #include "pbr.h"
 | |
| #include "nexthop_group.h"
 | |
| #include "lib_errors.h"
 | |
| 
 | |
| DEFINE_MTYPE_STATIC(LIB, ZCLIENT, "Zclient")
 | |
| DEFINE_MTYPE_STATIC(LIB, REDIST_INST, "Redistribution instance IDs")
 | |
| 
 | |
| /* Zebra client events. */
 | |
| enum event { ZCLIENT_SCHEDULE, ZCLIENT_READ, ZCLIENT_CONNECT };
 | |
| 
 | |
| /* Prototype for event manager. */
 | |
| static void zclient_event(enum event, struct zclient *);
 | |
| 
 | |
| static void zebra_interface_if_set_value(struct stream *s,
 | |
| 					 struct interface *ifp);
 | |
| 
 | |
| struct zclient_options zclient_options_default = {.receive_notify = false};
 | |
| 
 | |
| struct sockaddr_storage zclient_addr;
 | |
| socklen_t zclient_addr_len;
 | |
| 
 | |
| /* This file local debug flag. */
 | |
| static int zclient_debug;
 | |
| 
 | |
| /* Allocate zclient structure. */
 | |
| struct zclient *zclient_new(struct thread_master *master,
 | |
| 			    struct zclient_options *opt)
 | |
| {
 | |
| 	struct zclient *zclient;
 | |
| 	size_t stream_size =
 | |
| 		MAX(ZEBRA_MAX_PACKET_SIZ, sizeof(struct zapi_route));
 | |
| 
 | |
| 	zclient = XCALLOC(MTYPE_ZCLIENT, sizeof(struct zclient));
 | |
| 
 | |
| 	zclient->ibuf = stream_new(stream_size);
 | |
| 	zclient->obuf = stream_new(stream_size);
 | |
| 	zclient->wb = buffer_new(0);
 | |
| 	zclient->master = master;
 | |
| 
 | |
| 	zclient->receive_notify = opt->receive_notify;
 | |
| 
 | |
| 	return zclient;
 | |
| }
 | |
| 
 | |
| /* This function is only called when exiting, because
 | |
|    many parts of the code do not check for I/O errors, so they could
 | |
|    reference an invalid pointer if the structure was ever freed.
 | |
| 
 | |
|    Free zclient structure. */
 | |
| void zclient_free(struct zclient *zclient)
 | |
| {
 | |
| 	if (zclient->ibuf)
 | |
| 		stream_free(zclient->ibuf);
 | |
| 	if (zclient->obuf)
 | |
| 		stream_free(zclient->obuf);
 | |
| 	if (zclient->wb)
 | |
| 		buffer_free(zclient->wb);
 | |
| 
 | |
| 	XFREE(MTYPE_ZCLIENT, zclient);
 | |
| }
 | |
| 
 | |
| unsigned short *redist_check_instance(struct redist_proto *red,
 | |
| 				      unsigned short instance)
 | |
| {
 | |
| 	struct listnode *node;
 | |
| 	unsigned short *id;
 | |
| 
 | |
| 	if (!red->instances)
 | |
| 		return NULL;
 | |
| 
 | |
| 	for (ALL_LIST_ELEMENTS_RO(red->instances, node, id))
 | |
| 		if (*id == instance)
 | |
| 			return id;
 | |
| 
 | |
| 	return NULL;
 | |
| }
 | |
| 
 | |
| void redist_add_instance(struct redist_proto *red, unsigned short instance)
 | |
| {
 | |
| 	unsigned short *in;
 | |
| 
 | |
| 	red->enabled = 1;
 | |
| 
 | |
| 	if (!red->instances)
 | |
| 		red->instances = list_new();
 | |
| 
 | |
| 	in = XMALLOC(MTYPE_REDIST_INST, sizeof(unsigned short));
 | |
| 	*in = instance;
 | |
| 	listnode_add(red->instances, in);
 | |
| }
 | |
| 
 | |
| void redist_del_instance(struct redist_proto *red, unsigned short instance)
 | |
| {
 | |
| 	unsigned short *id;
 | |
| 
 | |
| 	id = redist_check_instance(red, instance);
 | |
| 	if (!id)
 | |
| 		return;
 | |
| 
 | |
| 	listnode_delete(red->instances, id);
 | |
| 	XFREE(MTYPE_REDIST_INST, id);
 | |
| 	if (!red->instances->count) {
 | |
| 		red->enabled = 0;
 | |
| 		list_delete(&red->instances);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| /* Stop zebra client services. */
 | |
| void zclient_stop(struct zclient *zclient)
 | |
| {
 | |
| 	afi_t afi;
 | |
| 	int i;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("zclient stopped");
 | |
| 
 | |
| 	/* Stop threads. */
 | |
| 	THREAD_OFF(zclient->t_read);
 | |
| 	THREAD_OFF(zclient->t_connect);
 | |
| 	THREAD_OFF(zclient->t_write);
 | |
| 
 | |
| 	/* Reset streams. */
 | |
| 	stream_reset(zclient->ibuf);
 | |
| 	stream_reset(zclient->obuf);
 | |
| 
 | |
| 	/* Empty the write buffer. */
 | |
| 	buffer_reset(zclient->wb);
 | |
| 
 | |
| 	/* Close socket. */
 | |
| 	if (zclient->sock >= 0) {
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 	}
 | |
| 	zclient->fail = 0;
 | |
| 
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 		for (i = 0; i < ZEBRA_ROUTE_MAX; i++) {
 | |
| 			vrf_bitmap_free(zclient->redist[afi][i]);
 | |
| 			zclient->redist[afi][i] = VRF_BITMAP_NULL;
 | |
| 		}
 | |
| 		redist_del_instance(
 | |
| 			&zclient->mi_redist[afi][zclient->redist_default],
 | |
| 			zclient->instance);
 | |
| 
 | |
| 		vrf_bitmap_free(zclient->default_information[afi]);
 | |
| 		zclient->default_information[afi] = VRF_BITMAP_NULL;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void zclient_reset(struct zclient *zclient)
 | |
| {
 | |
| 	afi_t afi;
 | |
| 
 | |
| 	zclient_stop(zclient);
 | |
| 
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++)
 | |
| 		redist_del_instance(
 | |
| 			&zclient->mi_redist[afi][zclient->redist_default],
 | |
| 			zclient->instance);
 | |
| 
 | |
| 	zclient_init(zclient, zclient->redist_default, zclient->instance,
 | |
| 		     zclient->privs);
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Connect to zebra daemon.
 | |
|  * @param zclient a pointer to zclient structure
 | |
|  * @return socket fd just to make sure that connection established
 | |
|  * @see zclient_init
 | |
|  * @see zclient_new
 | |
|  */
 | |
| int zclient_socket_connect(struct zclient *zclient)
 | |
| {
 | |
| 	int sock;
 | |
| 	int ret;
 | |
| 
 | |
| 	/* We should think about IPv6 connection. */
 | |
| 	sock = socket(zclient_addr.ss_family, SOCK_STREAM, 0);
 | |
| 	if (sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	set_cloexec(sock);
 | |
| 	setsockopt_so_sendbuf(sock, 1048576);
 | |
| 
 | |
| 	/* Connect to zebra. */
 | |
| 	ret = connect(sock, (struct sockaddr *)&zclient_addr, zclient_addr_len);
 | |
| 	if (ret < 0) {
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("%s connect failure: %d(%s)",
 | |
| 				   __PRETTY_FUNCTION__, errno,
 | |
| 				   safe_strerror(errno));
 | |
| 		close(sock);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	zclient->sock = sock;
 | |
| 	return sock;
 | |
| }
 | |
| 
 | |
| static int zclient_failed(struct zclient *zclient)
 | |
| {
 | |
| 	zclient->fail++;
 | |
| 	zclient_stop(zclient);
 | |
| 	zclient_event(ZCLIENT_CONNECT, zclient);
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| static int zclient_flush_data(struct thread *thread)
 | |
| {
 | |
| 	struct zclient *zclient = THREAD_ARG(thread);
 | |
| 
 | |
| 	zclient->t_write = NULL;
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 	switch (buffer_flush_available(zclient->wb, zclient->sock)) {
 | |
| 	case BUFFER_ERROR:
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_SOCKET,
 | |
| 			"%s: buffer_flush_available failed on zclient fd %d, closing",
 | |
| 			__func__, zclient->sock);
 | |
| 		return zclient_failed(zclient);
 | |
| 		break;
 | |
| 	case BUFFER_PENDING:
 | |
| 		zclient->t_write = NULL;
 | |
| 		thread_add_write(zclient->master, zclient_flush_data, zclient,
 | |
| 				 zclient->sock, &zclient->t_write);
 | |
| 		break;
 | |
| 	case BUFFER_EMPTY:
 | |
| 		break;
 | |
| 	}
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| int zclient_send_message(struct zclient *zclient)
 | |
| {
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 	switch (buffer_write(zclient->wb, zclient->sock,
 | |
| 			     STREAM_DATA(zclient->obuf),
 | |
| 			     stream_get_endp(zclient->obuf))) {
 | |
| 	case BUFFER_ERROR:
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET,
 | |
| 			 "%s: buffer_write failed to zclient fd %d, closing",
 | |
| 			 __func__, zclient->sock);
 | |
| 		return zclient_failed(zclient);
 | |
| 		break;
 | |
| 	case BUFFER_EMPTY:
 | |
| 		THREAD_OFF(zclient->t_write);
 | |
| 		break;
 | |
| 	case BUFFER_PENDING:
 | |
| 		thread_add_write(zclient->master, zclient_flush_data, zclient,
 | |
| 				 zclient->sock, &zclient->t_write);
 | |
| 		break;
 | |
| 	}
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| void zclient_create_header(struct stream *s, uint16_t command, vrf_id_t vrf_id)
 | |
| {
 | |
| 	/* length placeholder, caller can update */
 | |
| 	stream_putw(s, ZEBRA_HEADER_SIZE);
 | |
| 	stream_putc(s, ZEBRA_HEADER_MARKER);
 | |
| 	stream_putc(s, ZSERV_VERSION);
 | |
| 	stream_putl(s, vrf_id);
 | |
| 	stream_putw(s, command);
 | |
| }
 | |
| 
 | |
| int zclient_read_header(struct stream *s, int sock, uint16_t *size,
 | |
| 			uint8_t *marker, uint8_t *version, vrf_id_t *vrf_id,
 | |
| 			uint16_t *cmd)
 | |
| {
 | |
| 	if (stream_read(s, sock, ZEBRA_HEADER_SIZE) != ZEBRA_HEADER_SIZE)
 | |
| 		return -1;
 | |
| 
 | |
| 	STREAM_GETW(s, *size);
 | |
| 	*size -= ZEBRA_HEADER_SIZE;
 | |
| 	STREAM_GETC(s, *marker);
 | |
| 	STREAM_GETC(s, *version);
 | |
| 	STREAM_GETL(s, *vrf_id);
 | |
| 	STREAM_GETW(s, *cmd);
 | |
| 
 | |
| 	if (*version != ZSERV_VERSION || *marker != ZEBRA_HEADER_MARKER) {
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_MISSMATCH,
 | |
| 			"%s: socket %d version mismatch, marker %d, version %d",
 | |
| 			__func__, sock, *marker, *version);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	if (*size && stream_read(s, sock, *size) != *size)
 | |
| 		return -1;
 | |
| 
 | |
| 	return 0;
 | |
| stream_failure:
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| bool zapi_parse_header(struct stream *zmsg, struct zmsghdr *hdr)
 | |
| {
 | |
| 	STREAM_GETW(zmsg, hdr->length);
 | |
| 	STREAM_GETC(zmsg, hdr->marker);
 | |
| 	STREAM_GETC(zmsg, hdr->version);
 | |
| 	STREAM_GETL(zmsg, hdr->vrf_id);
 | |
| 	STREAM_GETW(zmsg, hdr->command);
 | |
| 	return true;
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| /* Send simple Zebra message. */
 | |
| static int zebra_message_send(struct zclient *zclient, int command,
 | |
| 			      vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	/* Get zclient output buffer. */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	/* Send very simple command only Zebra message. */
 | |
| 	zclient_create_header(s, command, vrf_id);
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| static int zebra_hello_send(struct zclient *zclient)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient->redist_default) {
 | |
| 		s = zclient->obuf;
 | |
| 		stream_reset(s);
 | |
| 
 | |
| 		/* The VRF ID in the HELLO message is always 0. */
 | |
| 		zclient_create_header(s, ZEBRA_HELLO, VRF_DEFAULT);
 | |
| 		stream_putc(s, zclient->redist_default);
 | |
| 		stream_putw(s, zclient->instance);
 | |
| 		if (zclient->receive_notify)
 | |
| 			stream_putc(s, 1);
 | |
| 		else
 | |
| 			stream_putc(s, 0);
 | |
| 
 | |
| 		stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 		return zclient_send_message(zclient);
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| void zclient_send_vrf_label(struct zclient *zclient, vrf_id_t vrf_id, afi_t afi,
 | |
| 			    mpls_label_t label, enum lsp_types_t ltype)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, ZEBRA_VRF_LABEL, vrf_id);
 | |
| 	stream_putl(s, label);
 | |
| 	stream_putc(s, afi);
 | |
| 	stream_putc(s, ltype);
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 	zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| /* Send register requests to zebra daemon for the information in a VRF. */
 | |
| void zclient_send_reg_requests(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	int i;
 | |
| 	afi_t afi;
 | |
| 
 | |
| 	/* If not connected to the zebra yet. */
 | |
| 	if (zclient->sock < 0)
 | |
| 		return;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: send register messages for VRF %u", __func__,
 | |
| 			   vrf_id);
 | |
| 
 | |
| 	/* We need router-id information. */
 | |
| 	zebra_message_send(zclient, ZEBRA_ROUTER_ID_ADD, vrf_id);
 | |
| 
 | |
| 	/* We need interface information. */
 | |
| 	zebra_message_send(zclient, ZEBRA_INTERFACE_ADD, vrf_id);
 | |
| 
 | |
| 	/* Set unwanted redistribute route. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++)
 | |
| 		vrf_bitmap_set(zclient->redist[afi][zclient->redist_default],
 | |
| 			       vrf_id);
 | |
| 
 | |
| 	/* Flush all redistribute request. */
 | |
| 	if (vrf_id == VRF_DEFAULT) {
 | |
| 		for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 			for (i = 0; i < ZEBRA_ROUTE_MAX; i++) {
 | |
| 				if (!zclient->mi_redist[afi][i].enabled)
 | |
| 					continue;
 | |
| 
 | |
| 				struct listnode *node;
 | |
| 				unsigned short *id;
 | |
| 
 | |
| 				for (ALL_LIST_ELEMENTS_RO(
 | |
| 					     zclient->mi_redist[afi][i]
 | |
| 						     .instances,
 | |
| 					     node, id))
 | |
| 					if (!(i == zclient->redist_default
 | |
| 					      && *id == zclient->instance))
 | |
| 						zebra_redistribute_send(
 | |
| 							ZEBRA_REDISTRIBUTE_ADD,
 | |
| 							zclient, afi, i, *id,
 | |
| 							VRF_DEFAULT);
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/* Resend all redistribute request. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 		for (i = 0; i < ZEBRA_ROUTE_MAX; i++)
 | |
| 			if (i != zclient->redist_default
 | |
| 			    && vrf_bitmap_check(zclient->redist[afi][i],
 | |
| 						vrf_id))
 | |
| 				zebra_redistribute_send(ZEBRA_REDISTRIBUTE_ADD,
 | |
| 							zclient, afi, i, 0,
 | |
| 							vrf_id);
 | |
| 
 | |
| 		/* If default information is needed. */
 | |
| 		if (vrf_bitmap_check(zclient->default_information[afi], vrf_id))
 | |
| 			zebra_redistribute_default_send(
 | |
| 				ZEBRA_REDISTRIBUTE_DEFAULT_ADD, zclient, afi,
 | |
| 				vrf_id);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| /* Send unregister requests to zebra daemon for the information in a VRF. */
 | |
| void zclient_send_dereg_requests(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	int i;
 | |
| 	afi_t afi;
 | |
| 
 | |
| 	/* If not connected to the zebra yet. */
 | |
| 	if (zclient->sock < 0)
 | |
| 		return;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: send deregister messages for VRF %u", __func__,
 | |
| 			   vrf_id);
 | |
| 
 | |
| 	/* We need router-id information. */
 | |
| 	zebra_message_send(zclient, ZEBRA_ROUTER_ID_DELETE, vrf_id);
 | |
| 
 | |
| 	zebra_message_send(zclient, ZEBRA_INTERFACE_DELETE, vrf_id);
 | |
| 
 | |
| 	/* Set unwanted redistribute route. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++)
 | |
| 		vrf_bitmap_unset(zclient->redist[afi][zclient->redist_default],
 | |
| 				 vrf_id);
 | |
| 
 | |
| 	/* Flush all redistribute request. */
 | |
| 	if (vrf_id == VRF_DEFAULT) {
 | |
| 		for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 			for (i = 0; i < ZEBRA_ROUTE_MAX; i++) {
 | |
| 				if (!zclient->mi_redist[afi][i].enabled)
 | |
| 					continue;
 | |
| 
 | |
| 				struct listnode *node;
 | |
| 				unsigned short *id;
 | |
| 
 | |
| 				for (ALL_LIST_ELEMENTS_RO(
 | |
| 					     zclient->mi_redist[afi][i]
 | |
| 						     .instances,
 | |
| 					     node, id))
 | |
| 					if (!(i == zclient->redist_default
 | |
| 					      && *id == zclient->instance))
 | |
| 						zebra_redistribute_send(
 | |
| 							ZEBRA_REDISTRIBUTE_DELETE,
 | |
| 							zclient, afi, i, *id,
 | |
| 							VRF_DEFAULT);
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/* Flush all redistribute request. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 		for (i = 0; i < ZEBRA_ROUTE_MAX; i++)
 | |
| 			if (i != zclient->redist_default
 | |
| 			    && vrf_bitmap_check(zclient->redist[afi][i],
 | |
| 						vrf_id))
 | |
| 				zebra_redistribute_send(
 | |
| 					ZEBRA_REDISTRIBUTE_DELETE, zclient, afi,
 | |
| 					i, 0, vrf_id);
 | |
| 
 | |
| 		/* If default information is needed. */
 | |
| 		if (vrf_bitmap_check(zclient->default_information[afi], vrf_id))
 | |
| 			zebra_redistribute_default_send(
 | |
| 				ZEBRA_REDISTRIBUTE_DEFAULT_DELETE, zclient, afi,
 | |
| 				vrf_id);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| /* Send request to zebra daemon to start or stop RA. */
 | |
| void zclient_send_interface_radv_req(struct zclient *zclient, vrf_id_t vrf_id,
 | |
| 				     struct interface *ifp, int enable,
 | |
| 				     int ra_interval)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	/* If not connected to the zebra yet. */
 | |
| 	if (zclient->sock < 0)
 | |
| 		return;
 | |
| 
 | |
| 	/* Form and send message. */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	if (enable)
 | |
| 		zclient_create_header(s, ZEBRA_INTERFACE_ENABLE_RADV, vrf_id);
 | |
| 	else
 | |
| 		zclient_create_header(s, ZEBRA_INTERFACE_DISABLE_RADV, vrf_id);
 | |
| 
 | |
| 	stream_putl(s, ifp->ifindex);
 | |
| 	stream_putl(s, ra_interval);
 | |
| 
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| int zclient_send_interface_protodown(struct zclient *zclient, vrf_id_t vrf_id,
 | |
| 				     struct interface *ifp, bool down)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_INTERFACE_SET_PROTODOWN, vrf_id);
 | |
| 	stream_putl(s, ifp->ifindex);
 | |
| 	stream_putc(s, !!down);
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 	zclient_send_message(zclient);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| /* Make connection to zebra daemon. */
 | |
| int zclient_start(struct zclient *zclient)
 | |
| {
 | |
| 	if (zclient_debug)
 | |
| 		zlog_info("zclient_start is called");
 | |
| 
 | |
| 	/* If already connected to the zebra. */
 | |
| 	if (zclient->sock >= 0)
 | |
| 		return 0;
 | |
| 
 | |
| 	/* Check connect thread. */
 | |
| 	if (zclient->t_connect)
 | |
| 		return 0;
 | |
| 
 | |
| 	if (zclient_socket_connect(zclient) < 0) {
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("zclient connection fail");
 | |
| 		zclient->fail++;
 | |
| 		zclient_event(ZCLIENT_CONNECT, zclient);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	if (set_nonblocking(zclient->sock) < 0)
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "%s: set_nonblocking(%d) failed",
 | |
| 			 __func__, zclient->sock);
 | |
| 
 | |
| 	/* Clear fail count. */
 | |
| 	zclient->fail = 0;
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("zclient connect success with socket [%d]",
 | |
| 			   zclient->sock);
 | |
| 
 | |
| 	/* Create read thread. */
 | |
| 	zclient_event(ZCLIENT_READ, zclient);
 | |
| 
 | |
| 	zebra_hello_send(zclient);
 | |
| 
 | |
| 	zebra_message_send(zclient, ZEBRA_INTERFACE_ADD, VRF_DEFAULT);
 | |
| 
 | |
| 	/* Inform the successful connection. */
 | |
| 	if (zclient->zebra_connected)
 | |
| 		(*zclient->zebra_connected)(zclient);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| /* Initialize zebra client.  Argument redist_default is unwanted
 | |
|    redistribute route type. */
 | |
| void zclient_init(struct zclient *zclient, int redist_default,
 | |
| 		  unsigned short instance, struct zebra_privs_t *privs)
 | |
| {
 | |
| 	int afi, i;
 | |
| 
 | |
| 	/* Set -1 to the default socket value. */
 | |
| 	zclient->sock = -1;
 | |
| 	zclient->privs = privs;
 | |
| 
 | |
| 	/* Clear redistribution flags. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++)
 | |
| 		for (i = 0; i < ZEBRA_ROUTE_MAX; i++)
 | |
| 			zclient->redist[afi][i] = vrf_bitmap_init();
 | |
| 
 | |
| 	/* Set unwanted redistribute route.  bgpd does not need BGP route
 | |
| 	   redistribution. */
 | |
| 	zclient->redist_default = redist_default;
 | |
| 	zclient->instance = instance;
 | |
| 	/* Pending: make afi(s) an arg. */
 | |
| 	for (afi = AFI_IP; afi < AFI_MAX; afi++) {
 | |
| 		redist_add_instance(&zclient->mi_redist[afi][redist_default],
 | |
| 				    instance);
 | |
| 
 | |
| 		/* Set default-information redistribute to zero. */
 | |
| 		zclient->default_information[afi] = vrf_bitmap_init();
 | |
| 	}
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("scheduling zclient connection");
 | |
| 
 | |
| 	zclient_event(ZCLIENT_SCHEDULE, zclient);
 | |
| }
 | |
| 
 | |
| /* This function is a wrapper function for calling zclient_start from
 | |
|    timer or event thread. */
 | |
| static int zclient_connect(struct thread *t)
 | |
| {
 | |
| 	struct zclient *zclient;
 | |
| 
 | |
| 	zclient = THREAD_ARG(t);
 | |
| 	zclient->t_connect = NULL;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("zclient_connect is called");
 | |
| 
 | |
| 	return zclient_start(zclient);
 | |
| }
 | |
| 
 | |
| int zclient_send_rnh(struct zclient *zclient, int command, struct prefix *p,
 | |
| 		     bool exact_match, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, command, vrf_id);
 | |
| 	stream_putc(s, (exact_match) ? 1 : 0);
 | |
| 
 | |
| 	stream_putw(s, PREFIX_FAMILY(p));
 | |
| 	stream_putc(s, p->prefixlen);
 | |
| 	switch (PREFIX_FAMILY(p)) {
 | |
| 	case AF_INET:
 | |
| 		stream_put_in_addr(s, &p->u.prefix4);
 | |
| 		break;
 | |
| 	case AF_INET6:
 | |
| 		stream_put(s, &(p->u.prefix6), 16);
 | |
| 		break;
 | |
| 	default:
 | |
| 		break;
 | |
| 	}
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * "xdr_encode"-like interface that allows daemon (client) to send
 | |
|  * a message to zebra server for a route that needs to be
 | |
|  * added/deleted to the kernel. Info about the route is specified
 | |
|  * by the caller in a struct zapi_route. zapi_route_encode() then writes
 | |
|  * the info down the zclient socket using the stream_* functions.
 | |
|  *
 | |
|  * The corresponding read ("xdr_decode") function on the server
 | |
|  * side is zapi_route_decode().
 | |
|  *
 | |
|  *  0 1 2 3 4 5 6 7 8 9 A B C D E F 0 1 2 3 4 5 6 7 8 9 A B C D E F
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |            Length (2)         |    Command    | Route Type    |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * | ZEBRA Flags   | Message Flags | Prefix length |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * | Destination IPv4 Prefix for route                             |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * | Nexthop count |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  *
 | |
|  *
 | |
|  * A number of IPv4 nexthop(s) or nexthop interface index(es) are then
 | |
|  * described, as per the Nexthop count. Each nexthop described as:
 | |
|  *
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * | Nexthop Type  |  Set to one of ZEBRA_NEXTHOP_*
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |       IPv4 Nexthop address or Interface Index number          |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  *
 | |
|  * Alternatively, if the route is a blackhole route, then Nexthop count
 | |
|  * is set to 1 and a nexthop of type NEXTHOP_TYPE_BLACKHOLE is the sole
 | |
|  * nexthop.
 | |
|  *
 | |
|  * The original struct zapi_route_*() infrastructure was built around
 | |
|  * the traditional (32-bit "gate OR ifindex") nexthop data unit.
 | |
|  * A special encoding can be used to feed onlink (64-bit "gate AND ifindex")
 | |
|  * nexthops into zapi_route_encode() using the same zapi_route structure.
 | |
|  * This is done by setting zapi_route fields as follows:
 | |
|  *  - .message |= ZAPI_MESSAGE_NEXTHOP | ZAPI_MESSAGE_ONLINK
 | |
|  *  - .nexthop_num == .ifindex_num
 | |
|  *  - .nexthop and .ifindex are filled with gate and ifindex parts of
 | |
|  *    each compound nexthop, both in the same order
 | |
|  *
 | |
|  * If ZAPI_MESSAGE_DISTANCE is set, the distance value is written as a 1
 | |
|  * byte value.
 | |
|  *
 | |
|  * If ZAPI_MESSAGE_METRIC is set, the metric value is written as an 8
 | |
|  * byte value.
 | |
|  *
 | |
|  * If ZAPI_MESSAGE_TAG is set, the tag value is written as a 4 byte value
 | |
|  *
 | |
|  * If ZAPI_MESSAGE_MTU is set, the mtu value is written as a 4 byte value
 | |
|  *
 | |
|  * XXX: No attention paid to alignment.
 | |
|  */
 | |
| int zclient_route_send(uint8_t cmd, struct zclient *zclient,
 | |
| 		       struct zapi_route *api)
 | |
| {
 | |
| 	if (zapi_route_encode(cmd, zclient->obuf, api) < 0)
 | |
| 		return -1;
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| static int zapi_nexthop_labels_cmp(const struct zapi_nexthop *next1,
 | |
| 				   const struct zapi_nexthop *next2)
 | |
| {
 | |
| 	if (next1->label_num > next2->label_num)
 | |
| 		return 1;
 | |
| 
 | |
| 	if (next1->label_num < next2->label_num)
 | |
| 		return -1;
 | |
| 
 | |
| 	return memcmp(next1->labels, next2->labels, next1->label_num);
 | |
| }
 | |
| 
 | |
| static int zapi_nexthop_cmp_no_labels(const struct zapi_nexthop *next1,
 | |
| 				      const struct zapi_nexthop *next2)
 | |
| {
 | |
| 	int ret = 0;
 | |
| 
 | |
| 	if (next1->vrf_id < next2->vrf_id)
 | |
| 		return -1;
 | |
| 
 | |
| 	if (next1->vrf_id > next2->vrf_id)
 | |
| 		return 1;
 | |
| 
 | |
| 	if (next1->type < next2->type)
 | |
| 		return -1;
 | |
| 
 | |
| 	if (next1->type > next2->type)
 | |
| 		return 1;
 | |
| 
 | |
| 	switch (next1->type) {
 | |
| 	case NEXTHOP_TYPE_IPV4:
 | |
| 	case NEXTHOP_TYPE_IPV6:
 | |
| 		ret = nexthop_g_addr_cmp(next1->type, &next1->gate,
 | |
| 					 &next2->gate);
 | |
| 		if (ret != 0)
 | |
| 			return ret;
 | |
| 		break;
 | |
| 	case NEXTHOP_TYPE_IPV4_IFINDEX:
 | |
| 	case NEXTHOP_TYPE_IPV6_IFINDEX:
 | |
| 		ret = nexthop_g_addr_cmp(next1->type, &next1->gate,
 | |
| 					 &next2->gate);
 | |
| 		if (ret != 0)
 | |
| 			return ret;
 | |
| 		/* Intentional Fall-Through */
 | |
| 	case NEXTHOP_TYPE_IFINDEX:
 | |
| 		if (next1->ifindex < next2->ifindex)
 | |
| 			return -1;
 | |
| 
 | |
| 		if (next1->ifindex > next2->ifindex)
 | |
| 			return 1;
 | |
| 		break;
 | |
| 	case NEXTHOP_TYPE_BLACKHOLE:
 | |
| 		if (next1->bh_type < next2->bh_type)
 | |
| 			return -1;
 | |
| 
 | |
| 		if (next1->bh_type > next2->bh_type)
 | |
| 			return 1;
 | |
| 		break;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int zapi_nexthop_cmp(const void *item1, const void *item2)
 | |
| {
 | |
| 	int ret = 0;
 | |
| 
 | |
| 	const struct zapi_nexthop *next1 = item1;
 | |
| 	const struct zapi_nexthop *next2 = item2;
 | |
| 
 | |
| 	ret = zapi_nexthop_cmp_no_labels(next1, next2);
 | |
| 	if (ret != 0)
 | |
| 		return ret;
 | |
| 
 | |
| 	ret = zapi_nexthop_labels_cmp(next1, next2);
 | |
| 
 | |
| 	return ret;
 | |
| }
 | |
| 
 | |
| static void zapi_nexthop_group_sort(struct zapi_nexthop *nh_grp,
 | |
| 				    uint16_t nexthop_num)
 | |
| {
 | |
| 	qsort(nh_grp, nexthop_num, sizeof(struct zapi_nexthop),
 | |
| 	      &zapi_nexthop_cmp);
 | |
| }
 | |
| 
 | |
| int zapi_route_encode(uint8_t cmd, struct stream *s, struct zapi_route *api)
 | |
| {
 | |
| 	struct zapi_nexthop *api_nh;
 | |
| 	int i;
 | |
| 	int psize;
 | |
| 
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, cmd, api->vrf_id);
 | |
| 
 | |
| 	if (api->type >= ZEBRA_ROUTE_MAX) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: Specified route type (%u) is not a legal value\n",
 | |
| 			 __PRETTY_FUNCTION__, api->type);
 | |
| 		return -1;
 | |
| 	}
 | |
| 	stream_putc(s, api->type);
 | |
| 
 | |
| 	stream_putw(s, api->instance);
 | |
| 	stream_putl(s, api->flags);
 | |
| 	stream_putc(s, api->message);
 | |
| 
 | |
| 	if (api->safi < SAFI_UNICAST || api->safi >= SAFI_MAX) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: Specified route SAFI (%u) is not a legal value\n",
 | |
| 			 __PRETTY_FUNCTION__, api->safi);
 | |
| 		return -1;
 | |
| 	}
 | |
| 	stream_putc(s, api->safi);
 | |
| 
 | |
| 	/* Put prefix information. */
 | |
| 	stream_putc(s, api->prefix.family);
 | |
| 	psize = PSIZE(api->prefix.prefixlen);
 | |
| 	stream_putc(s, api->prefix.prefixlen);
 | |
| 	stream_write(s, (uint8_t *)&api->prefix.u.prefix, psize);
 | |
| 
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_SRCPFX)) {
 | |
| 		psize = PSIZE(api->src_prefix.prefixlen);
 | |
| 		stream_putc(s, api->src_prefix.prefixlen);
 | |
| 		stream_write(s, (uint8_t *)&api->src_prefix.prefix, psize);
 | |
| 	}
 | |
| 
 | |
| 	/* Nexthops.  */
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_NEXTHOP)) {
 | |
| 		/* limit the number of nexthops if necessary */
 | |
| 		if (api->nexthop_num > MULTIPATH_NUM) {
 | |
| 			char buf[PREFIX2STR_BUFFER];
 | |
| 
 | |
| 			prefix2str(&api->prefix, buf, sizeof(buf));
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_ENCODE,
 | |
| 				"%s: prefix %s: can't encode %u nexthops (maximum is %u)",
 | |
| 				__func__, buf, api->nexthop_num, MULTIPATH_NUM);
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		zapi_nexthop_group_sort(api->nexthops, api->nexthop_num);
 | |
| 
 | |
| 		stream_putw(s, api->nexthop_num);
 | |
| 
 | |
| 		for (i = 0; i < api->nexthop_num; i++) {
 | |
| 			api_nh = &api->nexthops[i];
 | |
| 
 | |
| 			stream_putl(s, api_nh->vrf_id);
 | |
| 			stream_putc(s, api_nh->type);
 | |
| 			stream_putc(s, api_nh->onlink);
 | |
| 			switch (api_nh->type) {
 | |
| 			case NEXTHOP_TYPE_BLACKHOLE:
 | |
| 				stream_putc(s, api_nh->bh_type);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV4:
 | |
| 				stream_put_in_addr(s, &api_nh->gate.ipv4);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV4_IFINDEX:
 | |
| 				stream_put_in_addr(s, &api_nh->gate.ipv4);
 | |
| 				stream_putl(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IFINDEX:
 | |
| 				stream_putl(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV6:
 | |
| 				stream_write(s, (uint8_t *)&api_nh->gate.ipv6,
 | |
| 					     16);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV6_IFINDEX:
 | |
| 				stream_write(s, (uint8_t *)&api_nh->gate.ipv6,
 | |
| 					     16);
 | |
| 				stream_putl(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			}
 | |
| 
 | |
| 			/* MPLS labels for BGP-LU or Segment Routing */
 | |
| 			if (CHECK_FLAG(api->message, ZAPI_MESSAGE_LABEL)) {
 | |
| 				if (api_nh->label_num > MPLS_MAX_LABELS) {
 | |
| 					char buf[PREFIX2STR_BUFFER];
 | |
| 					prefix2str(&api->prefix, buf,
 | |
| 						   sizeof(buf));
 | |
| 					flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 						 "%s: prefix %s: can't encode "
 | |
| 						 "%u labels (maximum is %u)",
 | |
| 						 __func__, buf,
 | |
| 						 api_nh->label_num,
 | |
| 						 MPLS_MAX_LABELS);
 | |
| 					return -1;
 | |
| 				}
 | |
| 
 | |
| 				stream_putc(s, api_nh->label_num);
 | |
| 				stream_put(s, &api_nh->labels[0],
 | |
| 					   api_nh->label_num
 | |
| 						   * sizeof(mpls_label_t));
 | |
| 			}
 | |
| 
 | |
| 			/* Router MAC for EVPN routes. */
 | |
| 			if (CHECK_FLAG(api->flags, ZEBRA_FLAG_EVPN_ROUTE))
 | |
| 				stream_put(s, &(api_nh->rmac),
 | |
| 					   sizeof(struct ethaddr));
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/* Attributes. */
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_DISTANCE))
 | |
| 		stream_putc(s, api->distance);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_METRIC))
 | |
| 		stream_putl(s, api->metric);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_TAG))
 | |
| 		stream_putl(s, api->tag);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_MTU))
 | |
| 		stream_putl(s, api->mtu);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_TABLEID))
 | |
| 		stream_putl(s, api->tableid);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| int zapi_route_decode(struct stream *s, struct zapi_route *api)
 | |
| {
 | |
| 	struct zapi_nexthop *api_nh;
 | |
| 	int i;
 | |
| 
 | |
| 	memset(api, 0, sizeof(*api));
 | |
| 
 | |
| 	/* Type, flags, message. */
 | |
| 	STREAM_GETC(s, api->type);
 | |
| 	if (api->type >= ZEBRA_ROUTE_MAX) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: Specified route type: %d is not a legal value\n",
 | |
| 			 __PRETTY_FUNCTION__, api->type);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	STREAM_GETW(s, api->instance);
 | |
| 	STREAM_GETL(s, api->flags);
 | |
| 	STREAM_GETC(s, api->message);
 | |
| 	STREAM_GETC(s, api->safi);
 | |
| 	if (api->safi < SAFI_UNICAST || api->safi >= SAFI_MAX) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: Specified route SAFI (%u) is not a legal value\n",
 | |
| 			 __PRETTY_FUNCTION__, api->safi);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	/* Prefix. */
 | |
| 	STREAM_GETC(s, api->prefix.family);
 | |
| 	STREAM_GETC(s, api->prefix.prefixlen);
 | |
| 	switch (api->prefix.family) {
 | |
| 	case AF_INET:
 | |
| 		if (api->prefix.prefixlen > IPV4_MAX_PREFIXLEN) {
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_ENCODE,
 | |
| 				"%s: V4 prefixlen is %d which should not be more than 32",
 | |
| 				__PRETTY_FUNCTION__, api->prefix.prefixlen);
 | |
| 			return -1;
 | |
| 		}
 | |
| 		break;
 | |
| 	case AF_INET6:
 | |
| 		if (api->prefix.prefixlen > IPV6_MAX_PREFIXLEN) {
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_ENCODE,
 | |
| 				"%s: v6 prefixlen is %d which should not be more than 128",
 | |
| 				__PRETTY_FUNCTION__, api->prefix.prefixlen);
 | |
| 			return -1;
 | |
| 		}
 | |
| 		break;
 | |
| 	default:
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: Specified family %d is not v4 or v6",
 | |
| 			 __PRETTY_FUNCTION__, api->prefix.family);
 | |
| 		return -1;
 | |
| 	}
 | |
| 	STREAM_GET(&api->prefix.u.prefix, s, PSIZE(api->prefix.prefixlen));
 | |
| 
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_SRCPFX)) {
 | |
| 		api->src_prefix.family = AF_INET6;
 | |
| 		STREAM_GETC(s, api->src_prefix.prefixlen);
 | |
| 		if (api->src_prefix.prefixlen > IPV6_MAX_PREFIXLEN) {
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_ENCODE,
 | |
| 				"%s: SRC Prefix prefixlen received: %d is too large",
 | |
| 				__PRETTY_FUNCTION__, api->src_prefix.prefixlen);
 | |
| 			return -1;
 | |
| 		}
 | |
| 		STREAM_GET(&api->src_prefix.prefix, s,
 | |
| 			   PSIZE(api->src_prefix.prefixlen));
 | |
| 
 | |
| 		if (api->prefix.family != AF_INET6
 | |
| 		    || api->src_prefix.prefixlen == 0) {
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_ENCODE,
 | |
| 				"%s: SRC prefix specified in some manner that makes no sense",
 | |
| 				__PRETTY_FUNCTION__);
 | |
| 			return -1;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/* Nexthops. */
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_NEXTHOP)) {
 | |
| 		STREAM_GETW(s, api->nexthop_num);
 | |
| 		if (api->nexthop_num > MULTIPATH_NUM) {
 | |
| 			flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 				 "%s: invalid number of nexthops (%u)",
 | |
| 				 __func__, api->nexthop_num);
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		for (i = 0; i < api->nexthop_num; i++) {
 | |
| 			api_nh = &api->nexthops[i];
 | |
| 
 | |
| 			STREAM_GETL(s, api_nh->vrf_id);
 | |
| 			STREAM_GETC(s, api_nh->type);
 | |
| 			STREAM_GETC(s, api_nh->onlink);
 | |
| 			switch (api_nh->type) {
 | |
| 			case NEXTHOP_TYPE_BLACKHOLE:
 | |
| 				STREAM_GETC(s, api_nh->bh_type);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV4:
 | |
| 				STREAM_GET(&api_nh->gate.ipv4.s_addr, s,
 | |
| 					   IPV4_MAX_BYTELEN);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV4_IFINDEX:
 | |
| 				STREAM_GET(&api_nh->gate.ipv4.s_addr, s,
 | |
| 					   IPV4_MAX_BYTELEN);
 | |
| 				STREAM_GETL(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IFINDEX:
 | |
| 				STREAM_GETL(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV6:
 | |
| 				STREAM_GET(&api_nh->gate.ipv6, s, 16);
 | |
| 				break;
 | |
| 			case NEXTHOP_TYPE_IPV6_IFINDEX:
 | |
| 				STREAM_GET(&api_nh->gate.ipv6, s, 16);
 | |
| 				STREAM_GETL(s, api_nh->ifindex);
 | |
| 				break;
 | |
| 			}
 | |
| 
 | |
| 			/* MPLS labels for BGP-LU or Segment Routing */
 | |
| 			if (CHECK_FLAG(api->message, ZAPI_MESSAGE_LABEL)) {
 | |
| 				STREAM_GETC(s, api_nh->label_num);
 | |
| 
 | |
| 				if (api_nh->label_num > MPLS_MAX_LABELS) {
 | |
| 					flog_err(
 | |
| 						EC_LIB_ZAPI_ENCODE,
 | |
| 						"%s: invalid number of MPLS labels (%u)",
 | |
| 						__func__, api_nh->label_num);
 | |
| 					return -1;
 | |
| 				}
 | |
| 
 | |
| 				STREAM_GET(&api_nh->labels[0], s,
 | |
| 					   api_nh->label_num
 | |
| 						   * sizeof(mpls_label_t));
 | |
| 			}
 | |
| 
 | |
| 			/* Router MAC for EVPN routes. */
 | |
| 			if (CHECK_FLAG(api->flags, ZEBRA_FLAG_EVPN_ROUTE))
 | |
| 				stream_get(&(api_nh->rmac), s,
 | |
| 					   sizeof(struct ethaddr));
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/* Attributes. */
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_DISTANCE))
 | |
| 		STREAM_GETC(s, api->distance);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_METRIC))
 | |
| 		STREAM_GETL(s, api->metric);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_TAG))
 | |
| 		STREAM_GETL(s, api->tag);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_MTU))
 | |
| 		STREAM_GETL(s, api->mtu);
 | |
| 	if (CHECK_FLAG(api->message, ZAPI_MESSAGE_TABLEID))
 | |
| 		STREAM_GETL(s, api->tableid);
 | |
| 
 | |
| 	return 0;
 | |
| stream_failure:
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| static void zapi_encode_prefix(struct stream *s, struct prefix *p,
 | |
| 			       uint8_t family)
 | |
| {
 | |
| 	struct prefix any;
 | |
| 
 | |
| 	if (!p) {
 | |
| 		memset(&any, 0, sizeof(any));
 | |
| 		any.family = family;
 | |
| 		p = &any;
 | |
| 	}
 | |
| 
 | |
| 	stream_putc(s, p->family);
 | |
| 	stream_putc(s, p->prefixlen);
 | |
| 	stream_put(s, &p->u.prefix, prefix_blen(p));
 | |
| }
 | |
| 
 | |
| int zapi_pbr_rule_encode(uint8_t cmd, struct stream *s, struct pbr_rule *zrule)
 | |
| {
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, cmd, zrule->vrf_id);
 | |
| 
 | |
| 	/*
 | |
| 	 * We are sending one item at a time at the moment
 | |
| 	 */
 | |
| 	stream_putl(s, 1);
 | |
| 
 | |
| 	stream_putl(s, zrule->seq);
 | |
| 	stream_putl(s, zrule->priority);
 | |
| 	stream_putl(s, zrule->unique);
 | |
| 
 | |
| 	zapi_encode_prefix(s, &(zrule->filter.src_ip),
 | |
| 			   zrule->filter.src_ip.family);
 | |
| 	stream_putw(s, zrule->filter.src_port); /* src port */
 | |
| 	zapi_encode_prefix(s, &(zrule->filter.dst_ip),
 | |
| 			   zrule->filter.src_ip.family);
 | |
| 	stream_putw(s, zrule->filter.dst_port); /* dst port */
 | |
| 	stream_putw(s, zrule->filter.fwmark);   /* fwmark */
 | |
| 
 | |
| 	stream_putl(s, zrule->action.table);
 | |
| 	stream_putl(s, zrule->ifindex);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| bool zapi_route_notify_decode(struct stream *s, struct prefix *p,
 | |
| 			      uint32_t *tableid,
 | |
| 			      enum zapi_route_notify_owner *note)
 | |
| {
 | |
| 	uint32_t t;
 | |
| 
 | |
| 	STREAM_GET(note, s, sizeof(*note));
 | |
| 
 | |
| 	STREAM_GETC(s, p->family);
 | |
| 	STREAM_GETC(s, p->prefixlen);
 | |
| 	STREAM_GET(&p->u.prefix, s, prefix_blen(p));
 | |
| 	STREAM_GETL(s, t);
 | |
| 
 | |
| 	*tableid = t;
 | |
| 
 | |
| 	return true;
 | |
| 
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| bool zapi_rule_notify_decode(struct stream *s, uint32_t *seqno,
 | |
| 			     uint32_t *priority, uint32_t *unique,
 | |
| 			     ifindex_t *ifindex,
 | |
| 			     enum zapi_rule_notify_owner *note)
 | |
| {
 | |
| 	uint32_t prio, seq, uni;
 | |
| 	ifindex_t ifi;
 | |
| 
 | |
| 	STREAM_GET(note, s, sizeof(*note));
 | |
| 
 | |
| 	STREAM_GETL(s, seq);
 | |
| 	STREAM_GETL(s, prio);
 | |
| 	STREAM_GETL(s, uni);
 | |
| 	STREAM_GETL(s, ifi);
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: %u %u %u %u", __PRETTY_FUNCTION__, seq, prio,
 | |
| 			   uni, ifi);
 | |
| 	*seqno = seq;
 | |
| 	*priority = prio;
 | |
| 	*unique = uni;
 | |
| 	*ifindex = ifi;
 | |
| 
 | |
| 	return true;
 | |
| 
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| bool zapi_ipset_notify_decode(struct stream *s, uint32_t *unique,
 | |
| 			      enum zapi_ipset_notify_owner *note)
 | |
| {
 | |
| 	uint32_t uni;
 | |
| 
 | |
| 	STREAM_GET(note, s, sizeof(*note));
 | |
| 
 | |
| 	STREAM_GETL(s, uni);
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: %u", __PRETTY_FUNCTION__, uni);
 | |
| 	*unique = uni;
 | |
| 
 | |
| 	return true;
 | |
| 
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| bool zapi_ipset_entry_notify_decode(struct stream *s, uint32_t *unique,
 | |
| 				    char *ipset_name,
 | |
| 				    enum zapi_ipset_entry_notify_owner *note)
 | |
| {
 | |
| 	uint32_t uni;
 | |
| 
 | |
| 	STREAM_GET(note, s, sizeof(*note));
 | |
| 
 | |
| 	STREAM_GETL(s, uni);
 | |
| 
 | |
| 	STREAM_GET(ipset_name, s, ZEBRA_IPSET_NAME_SIZE);
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: %u", __PRETTY_FUNCTION__, uni);
 | |
| 	*unique = uni;
 | |
| 
 | |
| 	return true;
 | |
| 
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| bool zapi_iptable_notify_decode(struct stream *s,
 | |
| 		uint32_t *unique,
 | |
| 		enum zapi_iptable_notify_owner *note)
 | |
| {
 | |
| 	uint32_t uni;
 | |
| 
 | |
| 	STREAM_GET(note, s, sizeof(*note));
 | |
| 
 | |
| 	STREAM_GETL(s, uni);
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: %u", __PRETTY_FUNCTION__, uni);
 | |
| 	*unique = uni;
 | |
| 
 | |
| 	return true;
 | |
| 
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| struct nexthop *nexthop_from_zapi_nexthop(struct zapi_nexthop *znh)
 | |
| {
 | |
| 	struct nexthop *n = nexthop_new();
 | |
| 
 | |
| 	n->type = znh->type;
 | |
| 	n->vrf_id = znh->vrf_id;
 | |
| 	n->ifindex = znh->ifindex;
 | |
| 	n->gate = znh->gate;
 | |
| 
 | |
| 	/*
 | |
| 	 * This function currently handles labels
 | |
| 	 */
 | |
| 	if (znh->label_num) {
 | |
| 		nexthop_add_labels(n, ZEBRA_LSP_NONE, znh->label_num,
 | |
| 				   znh->labels);
 | |
| 	}
 | |
| 
 | |
| 	return n;
 | |
| }
 | |
| 
 | |
| bool zapi_nexthop_update_decode(struct stream *s, struct zapi_route *nhr)
 | |
| {
 | |
| 	uint32_t i;
 | |
| 
 | |
| 	memset(nhr, 0, sizeof(*nhr));
 | |
| 
 | |
| 	STREAM_GETW(s, nhr->prefix.family);
 | |
| 	STREAM_GETC(s, nhr->prefix.prefixlen);
 | |
| 	switch (nhr->prefix.family) {
 | |
| 	case AF_INET:
 | |
| 		STREAM_GET(&nhr->prefix.u.prefix4.s_addr, s, IPV4_MAX_BYTELEN);
 | |
| 		break;
 | |
| 	case AF_INET6:
 | |
| 		STREAM_GET(&nhr->prefix.u.prefix6, s, IPV6_MAX_BYTELEN);
 | |
| 		break;
 | |
| 	default:
 | |
| 		break;
 | |
| 	}
 | |
| 
 | |
| 	STREAM_GETC(s, nhr->type);
 | |
| 	STREAM_GETW(s, nhr->instance);
 | |
| 	STREAM_GETC(s, nhr->distance);
 | |
| 	STREAM_GETL(s, nhr->metric);
 | |
| 	STREAM_GETC(s, nhr->nexthop_num);
 | |
| 
 | |
| 	for (i = 0; i < nhr->nexthop_num; i++) {
 | |
| 		STREAM_GETL(s, nhr->nexthops[i].vrf_id);
 | |
| 		STREAM_GETC(s, nhr->nexthops[i].type);
 | |
| 		switch (nhr->nexthops[i].type) {
 | |
| 		case NEXTHOP_TYPE_IPV4:
 | |
| 		case NEXTHOP_TYPE_IPV4_IFINDEX:
 | |
| 			STREAM_GET(&nhr->nexthops[i].gate.ipv4.s_addr, s,
 | |
| 				   IPV4_MAX_BYTELEN);
 | |
| 			STREAM_GETL(s, nhr->nexthops[i].ifindex);
 | |
| 			break;
 | |
| 		case NEXTHOP_TYPE_IFINDEX:
 | |
| 			STREAM_GETL(s, nhr->nexthops[i].ifindex);
 | |
| 			break;
 | |
| 		case NEXTHOP_TYPE_IPV6:
 | |
| 		case NEXTHOP_TYPE_IPV6_IFINDEX:
 | |
| 			STREAM_GET(&nhr->nexthops[i].gate.ipv6, s,
 | |
| 				   IPV6_MAX_BYTELEN);
 | |
| 			STREAM_GETL(s, nhr->nexthops[i].ifindex);
 | |
| 			break;
 | |
| 		case NEXTHOP_TYPE_BLACKHOLE:
 | |
| 			break;
 | |
| 		}
 | |
| 		STREAM_GETC(s, nhr->nexthops[i].label_num);
 | |
| 		if (nhr->nexthops[i].label_num > MPLS_MAX_LABELS) {
 | |
| 			flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 				 "%s: invalid number of MPLS labels (%u)",
 | |
| 				 __func__, nhr->nexthops[i].label_num);
 | |
| 			return false;
 | |
| 		}
 | |
| 		if (nhr->nexthops[i].label_num)
 | |
| 			STREAM_GET(&nhr->nexthops[i].labels[0], s,
 | |
| 				   nhr->nexthops[i].label_num
 | |
| 					   * sizeof(mpls_label_t));
 | |
| 	}
 | |
| 
 | |
| 	return true;
 | |
| stream_failure:
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * send a ZEBRA_REDISTRIBUTE_ADD or ZEBRA_REDISTRIBUTE_DELETE
 | |
|  * for the route type (ZEBRA_ROUTE_KERNEL etc.). The zebra server will
 | |
|  * then set/unset redist[type] in the client handle (a struct zserv) for the
 | |
|  * sending client
 | |
|  */
 | |
| int zebra_redistribute_send(int command, struct zclient *zclient, afi_t afi,
 | |
| 			    int type, unsigned short instance, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, command, vrf_id);
 | |
| 	stream_putc(s, afi);
 | |
| 	stream_putc(s, type);
 | |
| 	stream_putw(s, instance);
 | |
| 
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| int zebra_redistribute_default_send(int command, struct zclient *zclient,
 | |
| 				    afi_t afi, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, command, vrf_id);
 | |
| 	stream_putc(s, afi);
 | |
| 
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| /* Get prefix in ZServ format; family should be filled in on prefix */
 | |
| static void zclient_stream_get_prefix(struct stream *s, struct prefix *p)
 | |
| {
 | |
| 	size_t plen = prefix_blen(p);
 | |
| 	uint8_t c;
 | |
| 	p->prefixlen = 0;
 | |
| 
 | |
| 	if (plen == 0)
 | |
| 		return;
 | |
| 
 | |
| 	stream_get(&p->u.prefix, s, plen);
 | |
| 	STREAM_GETC(s, c);
 | |
| 	p->prefixlen = MIN(plen * 8, c);
 | |
| 
 | |
| stream_failure:
 | |
| 	return;
 | |
| }
 | |
| 
 | |
| /* Router-id update from zebra daemon. */
 | |
| void zebra_router_id_update_read(struct stream *s, struct prefix *rid)
 | |
| {
 | |
| 	/* Fetch interface address. */
 | |
| 	STREAM_GETC(s, rid->family);
 | |
| 
 | |
| 	zclient_stream_get_prefix(s, rid);
 | |
| 
 | |
| stream_failure:
 | |
| 	return;
 | |
| }
 | |
| 
 | |
| /* Interface addition from zebra daemon. */
 | |
| /*
 | |
|  * The format of the message sent with type ZEBRA_INTERFACE_ADD or
 | |
|  * ZEBRA_INTERFACE_DELETE from zebra to the client is:
 | |
|  *     0                   1                   2                   3
 | |
|  *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  ifname                                                       |
 | |
|  * |                                                               |
 | |
|  * |                                                               |
 | |
|  * |                                                               |
 | |
|  * |                                                               |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  ifindex                                                      |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  status       |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  if_flags                                                     |
 | |
|  * |                                                               |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  metric                                                       |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  speed                                                        |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  ifmtu                                                        |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  ifmtu6                                                       |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  bandwidth                                                    |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  parent ifindex                                               |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  Link Layer Type                                              |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  Harware Address Length                                       |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  Hardware Address      if HW lenght different from 0          |
 | |
|  * |   ...                  max INTERFACE_HWADDR_MAX               |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  Link_params? |  Whether a link-params follows: 1 or 0.
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  * |  Link_params    0 or 1 INTERFACE_LINK_PARAMS_SIZE sized       |
 | |
|  * |   ....          (struct if_link_params).                      |
 | |
|  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 | |
|  */
 | |
| 
 | |
| static void zclient_vrf_add(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct vrf *vrf;
 | |
| 	char vrfname_tmp[VRF_NAMSIZ];
 | |
| 	struct vrf_data data;
 | |
| 
 | |
| 	stream_get(&data, zclient->ibuf, sizeof(struct vrf_data));
 | |
| 	/* Read interface name. */
 | |
| 	stream_get(vrfname_tmp, zclient->ibuf, VRF_NAMSIZ);
 | |
| 
 | |
| 	/* Lookup/create vrf by vrf_id. */
 | |
| 	vrf = vrf_get(vrf_id, vrfname_tmp);
 | |
| 	vrf->data.l.table_id = data.l.table_id;
 | |
| 	memcpy(vrf->data.l.netns_name, data.l.netns_name, NS_NAMSIZ);
 | |
| 	/* overwrite default vrf */
 | |
| 	if (vrf_id == VRF_DEFAULT)
 | |
| 		vrf_set_default_name(vrfname_tmp, false);
 | |
| 	vrf_enable(vrf);
 | |
| }
 | |
| 
 | |
| static void zclient_vrf_delete(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct vrf *vrf;
 | |
| 
 | |
| 	/* Lookup vrf by vrf_id. */
 | |
| 	vrf = vrf_lookup_by_id(vrf_id);
 | |
| 
 | |
| 	/*
 | |
| 	 * If a routing protocol doesn't know about a
 | |
| 	 * vrf that is about to be deleted.  There is
 | |
| 	 * no point in attempting to delete it.
 | |
| 	 */
 | |
| 	if (!vrf)
 | |
| 		return;
 | |
| 
 | |
| 	vrf_delete(vrf);
 | |
| }
 | |
| 
 | |
| static void zclient_interface_add(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct interface *ifp;
 | |
| 	char ifname_tmp[INTERFACE_NAMSIZ];
 | |
| 	struct stream *s = zclient->ibuf;
 | |
| 
 | |
| 	/* Read interface name. */
 | |
| 	stream_get(ifname_tmp, s, INTERFACE_NAMSIZ);
 | |
| 
 | |
| 	/* Lookup/create interface by name. */
 | |
| 	ifp = if_get_by_name(ifname_tmp, vrf_id);
 | |
| 
 | |
| 	zebra_interface_if_set_value(s, ifp);
 | |
| 
 | |
| 	if_new_via_zapi(ifp);
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Read interface up/down msg (ZEBRA_INTERFACE_UP/ZEBRA_INTERFACE_DOWN)
 | |
|  * from zebra server.  The format of this message is the same as
 | |
|  * that sent for ZEBRA_INTERFACE_ADD/ZEBRA_INTERFACE_DELETE,
 | |
|  * except that no sockaddr_dl is sent at the tail of the message.
 | |
|  */
 | |
| struct interface *zebra_interface_state_read(struct stream *s, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct interface *ifp;
 | |
| 	char ifname_tmp[INTERFACE_NAMSIZ];
 | |
| 
 | |
| 	/* Read interface name. */
 | |
| 	stream_get(ifname_tmp, s, INTERFACE_NAMSIZ);
 | |
| 
 | |
| 	/* Lookup this by interface index. */
 | |
| 	ifp = if_lookup_by_name(ifname_tmp, vrf_id);
 | |
| 	if (ifp == NULL) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "INTERFACE_STATE: Cannot find IF %s in VRF %d",
 | |
| 			 ifname_tmp, vrf_id);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	zebra_interface_if_set_value(s, ifp);
 | |
| 
 | |
| 	return ifp;
 | |
| }
 | |
| 
 | |
| static void zclient_interface_delete(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct interface *ifp;
 | |
| 	struct stream *s = zclient->ibuf;
 | |
| 
 | |
| 	ifp = zebra_interface_state_read(s, vrf_id);
 | |
| 
 | |
| 	if (ifp == NULL)
 | |
| 		return;
 | |
| 
 | |
| 	if_destroy_via_zapi(ifp);
 | |
| 	return;
 | |
| }
 | |
| 
 | |
| static void zclient_interface_up(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct interface *ifp;
 | |
| 	struct stream *s = zclient->ibuf;
 | |
| 
 | |
| 	ifp = zebra_interface_state_read(s, vrf_id);
 | |
| 
 | |
| 	if (!ifp)
 | |
| 		return;
 | |
| 
 | |
| 	if_up_via_zapi(ifp);
 | |
| }
 | |
| 
 | |
| static void zclient_interface_down(struct zclient *zclient, vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct interface *ifp;
 | |
| 	struct stream *s = zclient->ibuf;
 | |
| 
 | |
| 	ifp = zebra_interface_state_read(s, vrf_id);
 | |
| 
 | |
| 	if (!ifp)
 | |
| 		return;
 | |
| 
 | |
| 	if_down_via_zapi(ifp);
 | |
| }
 | |
| 
 | |
| static void link_params_set_value(struct stream *s, struct if_link_params *iflp)
 | |
| {
 | |
| 
 | |
| 	if (iflp == NULL)
 | |
| 		return;
 | |
| 
 | |
| 	iflp->lp_status = stream_getl(s);
 | |
| 	iflp->te_metric = stream_getl(s);
 | |
| 	iflp->max_bw = stream_getf(s);
 | |
| 	iflp->max_rsv_bw = stream_getf(s);
 | |
| 	uint32_t bwclassnum = stream_getl(s);
 | |
| 	{
 | |
| 		unsigned int i;
 | |
| 		for (i = 0; i < bwclassnum && i < MAX_CLASS_TYPE; i++)
 | |
| 			iflp->unrsv_bw[i] = stream_getf(s);
 | |
| 		if (i < bwclassnum)
 | |
| 			flog_err(
 | |
| 				EC_LIB_ZAPI_MISSMATCH,
 | |
| 				"%s: received %d > %d (MAX_CLASS_TYPE) bw entries"
 | |
| 				" - outdated library?",
 | |
| 				__func__, bwclassnum, MAX_CLASS_TYPE);
 | |
| 	}
 | |
| 	iflp->admin_grp = stream_getl(s);
 | |
| 	iflp->rmt_as = stream_getl(s);
 | |
| 	iflp->rmt_ip.s_addr = stream_get_ipv4(s);
 | |
| 
 | |
| 	iflp->av_delay = stream_getl(s);
 | |
| 	iflp->min_delay = stream_getl(s);
 | |
| 	iflp->max_delay = stream_getl(s);
 | |
| 	iflp->delay_var = stream_getl(s);
 | |
| 
 | |
| 	iflp->pkt_loss = stream_getf(s);
 | |
| 	iflp->res_bw = stream_getf(s);
 | |
| 	iflp->ava_bw = stream_getf(s);
 | |
| 	iflp->use_bw = stream_getf(s);
 | |
| }
 | |
| 
 | |
| struct interface *zebra_interface_link_params_read(struct stream *s,
 | |
| 						   vrf_id_t vrf_id)
 | |
| {
 | |
| 	struct if_link_params *iflp;
 | |
| 	ifindex_t ifindex;
 | |
| 
 | |
| 	assert(s);
 | |
| 
 | |
| 	ifindex = stream_getl(s);
 | |
| 
 | |
| 	struct interface *ifp = if_lookup_by_index(ifindex, vrf_id);
 | |
| 
 | |
| 	if (ifp == NULL) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "%s: unknown ifindex %u, shouldn't happen", __func__,
 | |
| 			 ifindex);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	if ((iflp = if_link_params_get(ifp)) == NULL)
 | |
| 		return NULL;
 | |
| 
 | |
| 	link_params_set_value(s, iflp);
 | |
| 
 | |
| 	return ifp;
 | |
| }
 | |
| 
 | |
| static void zebra_interface_if_set_value(struct stream *s,
 | |
| 					 struct interface *ifp)
 | |
| {
 | |
| 	uint8_t link_params_status = 0;
 | |
| 	ifindex_t old_ifindex;
 | |
| 
 | |
| 	old_ifindex = ifp->ifindex;
 | |
| 	/* Read interface's index. */
 | |
| 	if_set_index(ifp, stream_getl(s));
 | |
| 	ifp->status = stream_getc(s);
 | |
| 
 | |
| 	/* Read interface's value. */
 | |
| 	ifp->flags = stream_getq(s);
 | |
| 	ifp->ptm_enable = stream_getc(s);
 | |
| 	ifp->ptm_status = stream_getc(s);
 | |
| 	ifp->metric = stream_getl(s);
 | |
| 	ifp->speed = stream_getl(s);
 | |
| 	ifp->mtu = stream_getl(s);
 | |
| 	ifp->mtu6 = stream_getl(s);
 | |
| 	ifp->bandwidth = stream_getl(s);
 | |
| 	ifp->link_ifindex = stream_getl(s);
 | |
| 	ifp->ll_type = stream_getl(s);
 | |
| 	ifp->hw_addr_len = stream_getl(s);
 | |
| 	if (ifp->hw_addr_len)
 | |
| 		stream_get(ifp->hw_addr, s,
 | |
| 			   MIN(ifp->hw_addr_len, INTERFACE_HWADDR_MAX));
 | |
| 
 | |
| 	/* Read Traffic Engineering status */
 | |
| 	link_params_status = stream_getc(s);
 | |
| 	/* Then, Traffic Engineering parameters if any */
 | |
| 	if (link_params_status) {
 | |
| 		struct if_link_params *iflp = if_link_params_get(ifp);
 | |
| 		link_params_set_value(s, iflp);
 | |
| 	}
 | |
| 
 | |
| 	nexthop_group_interface_state_change(ifp, old_ifindex);
 | |
| }
 | |
| 
 | |
| size_t zebra_interface_link_params_write(struct stream *s,
 | |
| 					 struct interface *ifp)
 | |
| {
 | |
| 	size_t w;
 | |
| 	struct if_link_params *iflp;
 | |
| 	int i;
 | |
| 
 | |
| 	if (s == NULL || ifp == NULL || ifp->link_params == NULL)
 | |
| 		return 0;
 | |
| 
 | |
| 	iflp = ifp->link_params;
 | |
| 	w = 0;
 | |
| 
 | |
| 	w += stream_putl(s, iflp->lp_status);
 | |
| 
 | |
| 	w += stream_putl(s, iflp->te_metric);
 | |
| 	w += stream_putf(s, iflp->max_bw);
 | |
| 	w += stream_putf(s, iflp->max_rsv_bw);
 | |
| 
 | |
| 	w += stream_putl(s, MAX_CLASS_TYPE);
 | |
| 	for (i = 0; i < MAX_CLASS_TYPE; i++)
 | |
| 		w += stream_putf(s, iflp->unrsv_bw[i]);
 | |
| 
 | |
| 	w += stream_putl(s, iflp->admin_grp);
 | |
| 	w += stream_putl(s, iflp->rmt_as);
 | |
| 	w += stream_put_in_addr(s, &iflp->rmt_ip);
 | |
| 
 | |
| 	w += stream_putl(s, iflp->av_delay);
 | |
| 	w += stream_putl(s, iflp->min_delay);
 | |
| 	w += stream_putl(s, iflp->max_delay);
 | |
| 	w += stream_putl(s, iflp->delay_var);
 | |
| 
 | |
| 	w += stream_putf(s, iflp->pkt_loss);
 | |
| 	w += stream_putf(s, iflp->res_bw);
 | |
| 	w += stream_putf(s, iflp->ava_bw);
 | |
| 	w += stream_putf(s, iflp->use_bw);
 | |
| 
 | |
| 	return w;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * format of message for address additon is:
 | |
|  *    0
 | |
|  *  0 1 2 3 4 5 6 7
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |   type        |  ZEBRA_INTERFACE_ADDRESS_ADD or
 | |
|  * +-+-+-+-+-+-+-+-+  ZEBRA_INTERFACE_ADDRES_DELETE
 | |
|  * |               |
 | |
|  * +               +
 | |
|  * |   ifindex     |
 | |
|  * +               +
 | |
|  * |               |
 | |
|  * +               +
 | |
|  * |               |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |   ifc_flags   |  flags for connected address
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |  addr_family  |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |    addr...    |
 | |
|  * :               :
 | |
|  * |               |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |    addr_len   |  len of addr. E.g., addr_len = 4 for ipv4 addrs.
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |     daddr..   |
 | |
|  * :               :
 | |
|  * |               |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  */
 | |
| 
 | |
| static int memconstant(const void *s, int c, size_t n)
 | |
| {
 | |
| 	const uint8_t *p = s;
 | |
| 
 | |
| 	while (n-- > 0)
 | |
| 		if (*p++ != c)
 | |
| 			return 0;
 | |
| 	return 1;
 | |
| }
 | |
| 
 | |
| 
 | |
| struct connected *zebra_interface_address_read(int type, struct stream *s,
 | |
| 					       vrf_id_t vrf_id)
 | |
| {
 | |
| 	ifindex_t ifindex;
 | |
| 	struct interface *ifp;
 | |
| 	struct connected *ifc;
 | |
| 	struct prefix p, d, *dp;
 | |
| 	int plen;
 | |
| 	uint8_t ifc_flags;
 | |
| 
 | |
| 	memset(&p, 0, sizeof(p));
 | |
| 	memset(&d, 0, sizeof(d));
 | |
| 
 | |
| 	/* Get interface index. */
 | |
| 	ifindex = stream_getl(s);
 | |
| 
 | |
| 	/* Lookup index. */
 | |
| 	ifp = if_lookup_by_index(ifindex, vrf_id);
 | |
| 	if (ifp == NULL) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "INTERFACE_ADDRESS_%s: Cannot find IF %u in VRF %d",
 | |
| 			 (type == ZEBRA_INTERFACE_ADDRESS_ADD) ? "ADD" : "DEL",
 | |
| 			 ifindex, vrf_id);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	/* Fetch flag. */
 | |
| 	ifc_flags = stream_getc(s);
 | |
| 
 | |
| 	/* Fetch interface address. */
 | |
| 	d.family = p.family = stream_getc(s);
 | |
| 	plen = prefix_blen(&d);
 | |
| 
 | |
| 	zclient_stream_get_prefix(s, &p);
 | |
| 
 | |
| 	/* Fetch destination address. */
 | |
| 	stream_get(&d.u.prefix, s, plen);
 | |
| 
 | |
| 	/* N.B. NULL destination pointers are encoded as all zeroes */
 | |
| 	dp = memconstant(&d.u.prefix, 0, plen) ? NULL : &d;
 | |
| 
 | |
| 	if (type == ZEBRA_INTERFACE_ADDRESS_ADD) {
 | |
| 		ifc = connected_lookup_prefix_exact(ifp, &p);
 | |
| 		if (!ifc) {
 | |
| 			/* N.B. NULL destination pointers are encoded as all
 | |
| 			 * zeroes */
 | |
| 			ifc = connected_add_by_prefix(ifp, &p, dp);
 | |
| 		}
 | |
| 		if (ifc) {
 | |
| 			ifc->flags = ifc_flags;
 | |
| 			if (ifc->destination)
 | |
| 				ifc->destination->prefixlen =
 | |
| 					ifc->address->prefixlen;
 | |
| 			else if (CHECK_FLAG(ifc->flags, ZEBRA_IFA_PEER)) {
 | |
| 				/* carp interfaces on OpenBSD with 0.0.0.0/0 as
 | |
| 				 * "peer" */
 | |
| 				char buf[PREFIX_STRLEN];
 | |
| 				flog_err(
 | |
| 					EC_LIB_ZAPI_ENCODE,
 | |
| 					"warning: interface %s address %s with peer flag set, but no peer address!",
 | |
| 					ifp->name,
 | |
| 					prefix2str(ifc->address, buf,
 | |
| 						   sizeof buf));
 | |
| 				UNSET_FLAG(ifc->flags, ZEBRA_IFA_PEER);
 | |
| 			}
 | |
| 		}
 | |
| 	} else {
 | |
| 		assert(type == ZEBRA_INTERFACE_ADDRESS_DELETE);
 | |
| 		ifc = connected_delete_by_prefix(ifp, &p);
 | |
| 	}
 | |
| 
 | |
| 	return ifc;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * format of message for neighbor connected address is:
 | |
|  *    0
 | |
|  *  0 1 2 3 4 5 6 7
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |   type        |  ZEBRA_INTERFACE_NBR_ADDRESS_ADD or
 | |
|  * +-+-+-+-+-+-+-+-+  ZEBRA_INTERFACE_NBR_ADDRES_DELETE
 | |
|  * |               |
 | |
|  * +               +
 | |
|  * |   ifindex     |
 | |
|  * +               +
 | |
|  * |               |
 | |
|  * +               +
 | |
|  * |               |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |  addr_family  |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |    addr...    |
 | |
|  * :               :
 | |
|  * |               |
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  * |    addr_len   |  len of addr.
 | |
|  * +-+-+-+-+-+-+-+-+
 | |
|  */
 | |
| struct nbr_connected *
 | |
| zebra_interface_nbr_address_read(int type, struct stream *s, vrf_id_t vrf_id)
 | |
| {
 | |
| 	unsigned int ifindex;
 | |
| 	struct interface *ifp;
 | |
| 	struct prefix p;
 | |
| 	struct nbr_connected *ifc;
 | |
| 
 | |
| 	/* Get interface index. */
 | |
| 	ifindex = stream_getl(s);
 | |
| 
 | |
| 	/* Lookup index. */
 | |
| 	ifp = if_lookup_by_index(ifindex, vrf_id);
 | |
| 	if (ifp == NULL) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "INTERFACE_NBR_%s: Cannot find IF %u in VRF %d",
 | |
| 			 (type == ZEBRA_INTERFACE_NBR_ADDRESS_ADD) ? "ADD"
 | |
| 								   : "DELETE",
 | |
| 			 ifindex, vrf_id);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	p.family = stream_getc(s);
 | |
| 	stream_get(&p.u.prefix, s, prefix_blen(&p));
 | |
| 	p.prefixlen = stream_getc(s);
 | |
| 
 | |
| 	if (type == ZEBRA_INTERFACE_NBR_ADDRESS_ADD) {
 | |
| 		/* Currently only supporting P2P links, so any new RA source
 | |
| 		   address is
 | |
| 		   considered as the replacement of the previously learnt
 | |
| 		   Link-Local address. */
 | |
| 		if (!(ifc = listnode_head(ifp->nbr_connected))) {
 | |
| 			ifc = nbr_connected_new();
 | |
| 			ifc->address = prefix_new();
 | |
| 			ifc->ifp = ifp;
 | |
| 			listnode_add(ifp->nbr_connected, ifc);
 | |
| 		}
 | |
| 
 | |
| 		prefix_copy(ifc->address, &p);
 | |
| 	} else {
 | |
| 		assert(type == ZEBRA_INTERFACE_NBR_ADDRESS_DELETE);
 | |
| 
 | |
| 		ifc = nbr_connected_check(ifp, &p);
 | |
| 		if (ifc)
 | |
| 			listnode_delete(ifp->nbr_connected, ifc);
 | |
| 	}
 | |
| 
 | |
| 	return ifc;
 | |
| }
 | |
| 
 | |
| struct interface *zebra_interface_vrf_update_read(struct stream *s,
 | |
| 						  vrf_id_t vrf_id,
 | |
| 						  vrf_id_t *new_vrf_id)
 | |
| {
 | |
| 	char ifname[INTERFACE_NAMSIZ];
 | |
| 	struct interface *ifp;
 | |
| 	vrf_id_t new_id;
 | |
| 
 | |
| 	/* Read interface name. */
 | |
| 	stream_get(ifname, s, INTERFACE_NAMSIZ);
 | |
| 
 | |
| 	/* Lookup interface. */
 | |
| 	ifp = if_lookup_by_name(ifname, vrf_id);
 | |
| 	if (ifp == NULL) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "INTERFACE_VRF_UPDATE: Cannot find IF %s in VRF %d",
 | |
| 			 ifname, vrf_id);
 | |
| 		return NULL;
 | |
| 	}
 | |
| 
 | |
| 	/* Fetch new VRF Id. */
 | |
| 	new_id = stream_getl(s);
 | |
| 
 | |
| 	*new_vrf_id = new_id;
 | |
| 	return ifp;
 | |
| }
 | |
| 
 | |
| /* filter unwanted messages until the expected one arrives */
 | |
| static int zclient_read_sync_response(struct zclient *zclient,
 | |
| 				      uint16_t expected_cmd)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 	uint16_t size = -1;
 | |
| 	uint8_t marker;
 | |
| 	uint8_t version;
 | |
| 	vrf_id_t vrf_id;
 | |
| 	uint16_t cmd;
 | |
| 	fd_set readfds;
 | |
| 	int ret;
 | |
| 
 | |
| 	ret = 0;
 | |
| 	cmd = expected_cmd + 1;
 | |
| 	while (ret == 0 && cmd != expected_cmd) {
 | |
| 		s = zclient->ibuf;
 | |
| 		stream_reset(s);
 | |
| 
 | |
| 		/* wait until response arrives */
 | |
| 		FD_ZERO(&readfds);
 | |
| 		FD_SET(zclient->sock, &readfds);
 | |
| 		select(zclient->sock + 1, &readfds, NULL, NULL, NULL);
 | |
| 		if (!FD_ISSET(zclient->sock, &readfds))
 | |
| 			continue;
 | |
| 		/* read response */
 | |
| 		ret = zclient_read_header(s, zclient->sock, &size, &marker,
 | |
| 					  &version, &vrf_id, &cmd);
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("%s: Response (%d bytes) received", __func__,
 | |
| 				   size);
 | |
| 	}
 | |
| 	if (ret != 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE, "%s: Invalid Sync Message Reply",
 | |
| 			 __func__);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| /**
 | |
|  * Connect to label manager in a syncronous way
 | |
|  *
 | |
|  * It first writes the request to zcient output buffer and then
 | |
|  * immediately reads the answer from the input buffer.
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to label manager (zebra)
 | |
|  * @param async Synchronous (0) or asynchronous (1) operation
 | |
|  * @result Result of response
 | |
|  */
 | |
| int lm_label_manager_connect(struct zclient *zclient, int async)
 | |
| {
 | |
| 	int ret;
 | |
| 	struct stream *s;
 | |
| 	uint8_t result;
 | |
| 	uint16_t cmd = async ? ZEBRA_LABEL_MANAGER_CONNECT_ASYNC :
 | |
| 			       ZEBRA_LABEL_MANAGER_CONNECT;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Connecting to Label Manager (LM)");
 | |
| 
 | |
| 	if (zclient->sock < 0) {
 | |
| 		zlog_debug("%s: invalid zclient socket", __func__);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, cmd, VRF_DEFAULT);
 | |
| 
 | |
| 	/* proto */
 | |
| 	stream_putc(s, zclient->redist_default);
 | |
| 	/* instance */
 | |
| 	stream_putw(s, zclient->instance);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	ret = writen(zclient->sock, s->data, stream_get_endp(s));
 | |
| 	if (ret < 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Can't write to zclient sock");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (ret == 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Zclient sock closed");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("LM connect request sent (%d bytes)", ret);
 | |
| 
 | |
| 	if (async)
 | |
| 		return 0;
 | |
| 
 | |
| 	/* read response */
 | |
| 	if (zclient_read_sync_response(zclient, cmd)
 | |
| 	    != 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	s = zclient->ibuf;
 | |
| 
 | |
| 	/* read instance and proto */
 | |
| 	uint8_t proto = stream_getc(s);
 | |
| 	uint16_t instance = stream_getw(s);
 | |
| 
 | |
| 	/* sanity */
 | |
| 	if (proto != zclient->redist_default)
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_ENCODE,
 | |
| 			"Wrong proto (%u) in LM connect response. Should be %u",
 | |
| 			proto, zclient->redist_default);
 | |
| 	if (instance != zclient->instance)
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_ENCODE,
 | |
| 			"Wrong instId (%u) in LM connect response. Should be %u",
 | |
| 			instance, zclient->instance);
 | |
| 
 | |
| 	/* result code */
 | |
| 	result = stream_getc(s);
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("LM connect-response received, result %u", result);
 | |
| 
 | |
| 	return (int)result;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Asynchronous label chunk request
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to label manager (zebra)
 | |
|  * @param keep Avoid garbage collection
 | |
|  * @param chunk_size Amount of labels requested
 | |
|  * @param base Base for the label chunk. if MPLS_LABEL_BASE_ANY we do not care
 | |
|  * @result 0 on success, -1 otherwise
 | |
|  */
 | |
| int zclient_send_get_label_chunk(struct zclient *zclient, uint8_t keep,
 | |
| 				 uint32_t chunk_size, uint32_t base)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Getting Label Chunk");
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, ZEBRA_GET_LABEL_CHUNK, VRF_DEFAULT);
 | |
| 	/* proto */
 | |
| 	stream_putc(s, zclient->redist_default);
 | |
| 	/* instance */
 | |
| 	stream_putw(s, zclient->instance);
 | |
| 	stream_putc(s, keep);
 | |
| 	stream_putl(s, chunk_size);
 | |
| 	stream_putl(s, base);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Function to request a label chunk in a syncronous way
 | |
|  *
 | |
|  * It first writes the request to zlcient output buffer and then
 | |
|  * immediately reads the answer from the input buffer.
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to label manager (zebra)
 | |
|  * @param keep Avoid garbage collection
 | |
|  * @param chunk_size Amount of labels requested
 | |
|  * @param start To write first assigned chunk label to
 | |
|  * @param end To write last assigned chunk label to
 | |
|  * @result 0 on success, -1 otherwise
 | |
|  */
 | |
| int lm_get_label_chunk(struct zclient *zclient, uint8_t keep, uint32_t base,
 | |
| 		       uint32_t chunk_size, uint32_t *start, uint32_t *end)
 | |
| {
 | |
| 	int ret;
 | |
| 	struct stream *s;
 | |
| 	uint8_t response_keep;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Getting Label Chunk");
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_GET_LABEL_CHUNK, VRF_DEFAULT);
 | |
| 	/* proto */
 | |
| 	stream_putc(s, zclient->redist_default);
 | |
| 	/* instance */
 | |
| 	stream_putw(s, zclient->instance);
 | |
| 	/* keep */
 | |
| 	stream_putc(s, keep);
 | |
| 	/* chunk size */
 | |
| 	stream_putl(s, chunk_size);
 | |
| 	/* requested chunk base */
 | |
| 	stream_putl(s, base);
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	ret = writen(zclient->sock, s->data, stream_get_endp(s));
 | |
| 	if (ret < 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Can't write to zclient sock");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (ret == 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Zclient sock closed");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Label chunk request (%d bytes) sent", ret);
 | |
| 
 | |
| 	/* read response */
 | |
| 	if (zclient_read_sync_response(zclient, ZEBRA_GET_LABEL_CHUNK) != 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* parse response */
 | |
| 	s = zclient->ibuf;
 | |
| 
 | |
| 	/* read proto and instance */
 | |
| 	uint8_t proto = stream_getc(s);
 | |
| 	uint16_t instance = stream_getw(s);
 | |
| 
 | |
| 	/* sanities */
 | |
| 	if (proto != zclient->redist_default)
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "Wrong proto (%u) in get chunk response. Should be %u",
 | |
| 			 proto, zclient->redist_default);
 | |
| 	if (instance != zclient->instance)
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 			 "Wrong instId (%u) in get chunk response Should be %u",
 | |
| 			 instance, zclient->instance);
 | |
| 
 | |
| 	/* if we requested a specific chunk and it could not be allocated, the
 | |
| 	 * response message will end here
 | |
| 	 */
 | |
| 	if (!STREAM_READABLE(s)) {
 | |
| 		zlog_info("Unable to assign Label Chunk to %s instance %u",
 | |
| 			  zebra_route_string(proto), instance);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	/* keep */
 | |
| 	response_keep = stream_getc(s);
 | |
| 	/* start and end labels */
 | |
| 	*start = stream_getl(s);
 | |
| 	*end = stream_getl(s);
 | |
| 
 | |
| 	/* not owning this response */
 | |
| 	if (keep != response_keep) {
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_ENCODE,
 | |
| 			"Invalid Label chunk: %u - %u, keeps mismatch %u != %u",
 | |
| 			*start, *end, keep, response_keep);
 | |
| 	}
 | |
| 	/* sanity */
 | |
| 	if (*start > *end || *start < MPLS_LABEL_UNRESERVED_MIN
 | |
| 	    || *end > MPLS_LABEL_UNRESERVED_MAX) {
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE, "Invalid Label chunk: %u - %u",
 | |
| 			 *start, *end);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Label Chunk assign: %u - %u (%u)", *start, *end,
 | |
| 			   response_keep);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Function to release a label chunk
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to label manager (zebra)
 | |
|  * @param start First label of chunk
 | |
|  * @param end Last label of chunk
 | |
|  * @result 0 on success, -1 otherwise
 | |
|  */
 | |
| int lm_release_label_chunk(struct zclient *zclient, uint32_t start,
 | |
| 			   uint32_t end)
 | |
| {
 | |
| 	int ret;
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Releasing Label Chunk %u - %u", start, end);
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_RELEASE_LABEL_CHUNK, VRF_DEFAULT);
 | |
| 
 | |
| 	/* proto */
 | |
| 	stream_putc(s, zclient->redist_default);
 | |
| 	/* instance */
 | |
| 	stream_putw(s, zclient->instance);
 | |
| 	/* start */
 | |
| 	stream_putl(s, start);
 | |
| 	/* end */
 | |
| 	stream_putl(s, end);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	ret = writen(zclient->sock, s->data, stream_get_endp(s));
 | |
| 	if (ret < 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Can't write to zclient sock");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (ret == 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "Zclient sock connection closed");
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Connect to table manager in a syncronous way
 | |
|  *
 | |
|  * It first writes the request to zcient output buffer and then
 | |
|  * immediately reads the answer from the input buffer.
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to table manager (zebra)
 | |
|  * @result Result of response
 | |
|  */
 | |
| int tm_table_manager_connect(struct zclient *zclient)
 | |
| {
 | |
| 	int ret;
 | |
| 	struct stream *s;
 | |
| 	uint8_t result;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Connecting to Table Manager");
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_TABLE_MANAGER_CONNECT, VRF_DEFAULT);
 | |
| 
 | |
| 	/* proto */
 | |
| 	stream_putc(s, zclient->redist_default);
 | |
| 	/* instance */
 | |
| 	stream_putw(s, zclient->instance);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	ret = zclient_send_message(zclient);
 | |
| 	if (ret < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: Table manager connect request sent", __func__);
 | |
| 
 | |
| 	/* read response */
 | |
| 	if (zclient_read_sync_response(zclient, ZEBRA_TABLE_MANAGER_CONNECT)
 | |
| 	    != 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* result */
 | |
| 	s = zclient->ibuf;
 | |
| 	STREAM_GETC(s, result);
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug(
 | |
| 			"%s: Table Manager connect response received, result %u",
 | |
| 			__func__, result);
 | |
| 
 | |
| 	return (int)result;
 | |
| stream_failure:
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Function to request a table chunk in a syncronous way
 | |
|  *
 | |
|  * It first writes the request to zclient output buffer and then
 | |
|  * immediately reads the answer from the input buffer.
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to table manager (zebra)
 | |
|  * @param chunk_size Amount of table requested
 | |
|  * @param start to write first assigned chunk table RT ID to
 | |
|  * @param end To write last assigned chunk table RT ID to
 | |
|  * @result 0 on success, -1 otherwise
 | |
|  */
 | |
| int tm_get_table_chunk(struct zclient *zclient, uint32_t chunk_size,
 | |
| 		       uint32_t *start, uint32_t *end)
 | |
| {
 | |
| 	int ret;
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Getting Table Chunk");
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_GET_TABLE_CHUNK, VRF_DEFAULT);
 | |
| 	/* chunk size */
 | |
| 	stream_putl(s, chunk_size);
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	ret = writen(zclient->sock, s->data, stream_get_endp(s));
 | |
| 	if (ret < 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET, "%s: can't write to zclient->sock",
 | |
| 			 __func__);
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (ret == 0) {
 | |
| 		flog_err(EC_LIB_ZAPI_SOCKET,
 | |
| 			 "%s: zclient->sock connection closed", __func__);
 | |
| 		close(zclient->sock);
 | |
| 		zclient->sock = -1;
 | |
| 		return -1;
 | |
| 	}
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("%s: Table chunk request (%d bytes) sent", __func__,
 | |
| 			   ret);
 | |
| 
 | |
| 	/* read response */
 | |
| 	if (zclient_read_sync_response(zclient, ZEBRA_GET_TABLE_CHUNK) != 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	s = zclient->ibuf;
 | |
| 	/* start and end table IDs */
 | |
| 	STREAM_GETL(s, *start);
 | |
| 	STREAM_GETL(s, *end);
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Table Chunk assign: %u - %u ", *start, *end);
 | |
| 
 | |
| 	return 0;
 | |
| stream_failure:
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Function to release a table chunk
 | |
|  *
 | |
|  * @param zclient Zclient used to connect to table manager (zebra)
 | |
|  * @param start First label of table
 | |
|  * @param end Last label of chunk
 | |
|  * @result 0 on success, -1 otherwise
 | |
|  */
 | |
| int tm_release_table_chunk(struct zclient *zclient, uint32_t start,
 | |
| 			   uint32_t end)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("Releasing Table Chunk");
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		return -1;
 | |
| 
 | |
| 	/* send request */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 	zclient_create_header(s, ZEBRA_RELEASE_TABLE_CHUNK, VRF_DEFAULT);
 | |
| 
 | |
| 	/* start */
 | |
| 	stream_putl(s, start);
 | |
| 	/* end */
 | |
| 	stream_putl(s, end);
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| int zebra_send_mpls_labels(struct zclient *zclient, int cmd,
 | |
| 			   struct zapi_labels *zl)
 | |
| {
 | |
| 	if (zapi_labels_encode(zclient->obuf, cmd, zl) < 0)
 | |
| 		return -1;
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| int zapi_labels_encode(struct stream *s, int cmd, struct zapi_labels *zl)
 | |
| {
 | |
| 	struct zapi_nexthop_label *znh;
 | |
| 
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, cmd, VRF_DEFAULT);
 | |
| 	stream_putc(s, zl->message);
 | |
| 	stream_putc(s, zl->type);
 | |
| 	stream_putl(s, zl->local_label);
 | |
| 
 | |
| 	if (CHECK_FLAG(zl->message, ZAPI_LABELS_FTN)) {
 | |
| 		stream_putw(s, zl->route.prefix.family);
 | |
| 		stream_put_prefix(s, &zl->route.prefix);
 | |
| 		stream_putc(s, zl->route.type);
 | |
| 		stream_putw(s, zl->route.instance);
 | |
| 	}
 | |
| 
 | |
| 	if (zl->nexthop_num > MULTIPATH_NUM) {
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_ENCODE,
 | |
| 			"%s: label %u: can't encode %u nexthops (maximum is %u)",
 | |
| 			__func__, zl->local_label, zl->nexthop_num,
 | |
| 			MULTIPATH_NUM);
 | |
| 		return -1;
 | |
| 	}
 | |
| 	stream_putw(s, zl->nexthop_num);
 | |
| 
 | |
| 	for (int i = 0; i < zl->nexthop_num; i++) {
 | |
| 		znh = &zl->nexthops[i];
 | |
| 
 | |
| 		stream_putc(s, znh->type);
 | |
| 		stream_putw(s, znh->family);
 | |
| 		switch (znh->family) {
 | |
| 		case AF_INET:
 | |
| 			stream_put_in_addr(s, &znh->address.ipv4);
 | |
| 			break;
 | |
| 		case AF_INET6:
 | |
| 			stream_write(s, (uint8_t *)&znh->address.ipv6, 16);
 | |
| 			break;
 | |
| 		default:
 | |
| 			break;
 | |
| 		}
 | |
| 		stream_putl(s, znh->ifindex);
 | |
| 		stream_putl(s, znh->label);
 | |
| 	}
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| int zapi_labels_decode(struct stream *s, struct zapi_labels *zl)
 | |
| {
 | |
| 	struct zapi_nexthop_label *znh;
 | |
| 
 | |
| 	memset(zl, 0, sizeof(*zl));
 | |
| 
 | |
| 	/* Get data. */
 | |
| 	STREAM_GETC(s, zl->message);
 | |
| 	STREAM_GETC(s, zl->type);
 | |
| 	STREAM_GETL(s, zl->local_label);
 | |
| 
 | |
| 	if (CHECK_FLAG(zl->message, ZAPI_LABELS_FTN)) {
 | |
| 		size_t psize;
 | |
| 
 | |
| 		STREAM_GETW(s, zl->route.prefix.family);
 | |
| 		STREAM_GETC(s, zl->route.prefix.prefixlen);
 | |
| 
 | |
| 		psize = PSIZE(zl->route.prefix.prefixlen);
 | |
| 		switch (zl->route.prefix.family) {
 | |
| 		case AF_INET:
 | |
| 			if (zl->route.prefix.prefixlen > IPV4_MAX_BITLEN) {
 | |
| 				zlog_debug(
 | |
| 					"%s: Specified prefix length %d is greater than a v4 address can support",
 | |
| 					__PRETTY_FUNCTION__,
 | |
| 					zl->route.prefix.prefixlen);
 | |
| 				return -1;
 | |
| 			}
 | |
| 			STREAM_GET(&zl->route.prefix.u.prefix4.s_addr, s,
 | |
| 				   psize);
 | |
| 			break;
 | |
| 		case AF_INET6:
 | |
| 			if (zl->route.prefix.prefixlen > IPV6_MAX_BITLEN) {
 | |
| 				zlog_debug(
 | |
| 					"%s: Specified prefix length %d is greater than a v6 address can support",
 | |
| 					__PRETTY_FUNCTION__,
 | |
| 					zl->route.prefix.prefixlen);
 | |
| 				return -1;
 | |
| 			}
 | |
| 			STREAM_GET(&zl->route.prefix.u.prefix6, s, psize);
 | |
| 			break;
 | |
| 		default:
 | |
| 			flog_err(EC_LIB_ZAPI_ENCODE,
 | |
| 				 "%s: Specified family %u is not v4 or v6",
 | |
| 				 __PRETTY_FUNCTION__, zl->route.prefix.family);
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		STREAM_GETC(s, zl->route.type);
 | |
| 		STREAM_GETW(s, zl->route.instance);
 | |
| 	}
 | |
| 
 | |
| 	STREAM_GETW(s, zl->nexthop_num);
 | |
| 	for (int i = 0; i < zl->nexthop_num; i++) {
 | |
| 		znh = &zl->nexthops[i];
 | |
| 
 | |
| 		STREAM_GETC(s, znh->type);
 | |
| 		STREAM_GETW(s, znh->family);
 | |
| 		switch (znh->family) {
 | |
| 		case AF_INET:
 | |
| 			STREAM_GET(&znh->address.ipv4.s_addr, s,
 | |
| 				   IPV4_MAX_BYTELEN);
 | |
| 			break;
 | |
| 		case AF_INET6:
 | |
| 			STREAM_GET(&znh->address.ipv6, s, 16);
 | |
| 			break;
 | |
| 		default:
 | |
| 			break;
 | |
| 		}
 | |
| 		STREAM_GETL(s, znh->ifindex);
 | |
| 		STREAM_GETL(s, znh->label);
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| stream_failure:
 | |
| 	return -1;
 | |
| }
 | |
| 
 | |
| int zebra_send_pw(struct zclient *zclient, int command, struct zapi_pw *pw)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	/* Reset stream. */
 | |
| 	s = zclient->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, command, VRF_DEFAULT);
 | |
| 	stream_write(s, pw->ifname, IF_NAMESIZE);
 | |
| 	stream_putl(s, pw->ifindex);
 | |
| 
 | |
| 	/* Put type */
 | |
| 	stream_putl(s, pw->type);
 | |
| 
 | |
| 	/* Put nexthop */
 | |
| 	stream_putl(s, pw->af);
 | |
| 	switch (pw->af) {
 | |
| 	case AF_INET:
 | |
| 		stream_put_in_addr(s, &pw->nexthop.ipv4);
 | |
| 		break;
 | |
| 	case AF_INET6:
 | |
| 		stream_write(s, (uint8_t *)&pw->nexthop.ipv6, 16);
 | |
| 		break;
 | |
| 	default:
 | |
| 		flog_err(EC_LIB_ZAPI_ENCODE, "%s: unknown af", __func__);
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	/* Put labels */
 | |
| 	stream_putl(s, pw->local_label);
 | |
| 	stream_putl(s, pw->remote_label);
 | |
| 
 | |
| 	/* Put flags */
 | |
| 	stream_putc(s, pw->flags);
 | |
| 
 | |
| 	/* Protocol specific fields */
 | |
| 	stream_write(s, &pw->data, sizeof(union pw_protocol_fields));
 | |
| 
 | |
| 	/* Put length at the first point of the stream. */
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 
 | |
| 	return zclient_send_message(zclient);
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Receive PW status update from Zebra and send it to LDE process.
 | |
|  */
 | |
| void zebra_read_pw_status_update(ZAPI_CALLBACK_ARGS, struct zapi_pw_status *pw)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	memset(pw, 0, sizeof(struct zapi_pw_status));
 | |
| 	s = zclient->ibuf;
 | |
| 
 | |
| 	/* Get data. */
 | |
| 	stream_get(pw->ifname, s, IF_NAMESIZE);
 | |
| 	pw->ifindex = stream_getl(s);
 | |
| 	pw->status = stream_getl(s);
 | |
| }
 | |
| 
 | |
| static void zclient_capability_decode(ZAPI_CALLBACK_ARGS)
 | |
| {
 | |
| 	struct zclient_capabilities cap;
 | |
| 	struct stream *s = zclient->ibuf;
 | |
| 	int vrf_backend;
 | |
| 	uint8_t mpls_enabled;
 | |
| 
 | |
| 	STREAM_GETL(s, vrf_backend);
 | |
| 	vrf_configure_backend(vrf_backend);
 | |
| 
 | |
| 	memset(&cap, 0, sizeof(cap));
 | |
| 	STREAM_GETC(s, mpls_enabled);
 | |
| 	cap.mpls_enabled = !!mpls_enabled;
 | |
| 	STREAM_GETL(s, cap.ecmp);
 | |
| 	STREAM_GETC(s, cap.role);
 | |
| 
 | |
| 	if (zclient->zebra_capabilities)
 | |
| 		(*zclient->zebra_capabilities)(&cap);
 | |
| 
 | |
| stream_failure:
 | |
| 	return;
 | |
| }
 | |
| 
 | |
| /* Zebra client message read function. */
 | |
| static int zclient_read(struct thread *thread)
 | |
| {
 | |
| 	size_t already;
 | |
| 	uint16_t length, command;
 | |
| 	uint8_t marker, version;
 | |
| 	vrf_id_t vrf_id;
 | |
| 	struct zclient *zclient;
 | |
| 
 | |
| 	/* Get socket to zebra. */
 | |
| 	zclient = THREAD_ARG(thread);
 | |
| 	zclient->t_read = NULL;
 | |
| 
 | |
| 	/* Read zebra header (if we don't have it already). */
 | |
| 	if ((already = stream_get_endp(zclient->ibuf)) < ZEBRA_HEADER_SIZE) {
 | |
| 		ssize_t nbyte;
 | |
| 		if (((nbyte = stream_read_try(zclient->ibuf, zclient->sock,
 | |
| 					      ZEBRA_HEADER_SIZE - already))
 | |
| 		     == 0)
 | |
| 		    || (nbyte == -1)) {
 | |
| 			if (zclient_debug)
 | |
| 				zlog_debug(
 | |
| 					"zclient connection closed socket [%d].",
 | |
| 					zclient->sock);
 | |
| 			return zclient_failed(zclient);
 | |
| 		}
 | |
| 		if (nbyte != (ssize_t)(ZEBRA_HEADER_SIZE - already)) {
 | |
| 			/* Try again later. */
 | |
| 			zclient_event(ZCLIENT_READ, zclient);
 | |
| 			return 0;
 | |
| 		}
 | |
| 		already = ZEBRA_HEADER_SIZE;
 | |
| 	}
 | |
| 
 | |
| 	/* Reset to read from the beginning of the incoming packet. */
 | |
| 	stream_set_getp(zclient->ibuf, 0);
 | |
| 
 | |
| 	/* Fetch header values. */
 | |
| 	length = stream_getw(zclient->ibuf);
 | |
| 	marker = stream_getc(zclient->ibuf);
 | |
| 	version = stream_getc(zclient->ibuf);
 | |
| 	vrf_id = stream_getl(zclient->ibuf);
 | |
| 	command = stream_getw(zclient->ibuf);
 | |
| 
 | |
| 	if (marker != ZEBRA_HEADER_MARKER || version != ZSERV_VERSION) {
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_MISSMATCH,
 | |
| 			"%s: socket %d version mismatch, marker %d, version %d",
 | |
| 			__func__, zclient->sock, marker, version);
 | |
| 		return zclient_failed(zclient);
 | |
| 	}
 | |
| 
 | |
| 	if (length < ZEBRA_HEADER_SIZE) {
 | |
| 		flog_err(EC_LIB_ZAPI_MISSMATCH,
 | |
| 			 "%s: socket %d message length %u is less than %d ",
 | |
| 			 __func__, zclient->sock, length, ZEBRA_HEADER_SIZE);
 | |
| 		return zclient_failed(zclient);
 | |
| 	}
 | |
| 
 | |
| 	/* Length check. */
 | |
| 	if (length > STREAM_SIZE(zclient->ibuf)) {
 | |
| 		struct stream *ns;
 | |
| 		flog_err(
 | |
| 			EC_LIB_ZAPI_ENCODE,
 | |
| 			"%s: message size %u exceeds buffer size %lu, expanding...",
 | |
| 			__func__, length,
 | |
| 			(unsigned long)STREAM_SIZE(zclient->ibuf));
 | |
| 		ns = stream_new(length);
 | |
| 		stream_copy(ns, zclient->ibuf);
 | |
| 		stream_free(zclient->ibuf);
 | |
| 		zclient->ibuf = ns;
 | |
| 	}
 | |
| 
 | |
| 	/* Read rest of zebra packet. */
 | |
| 	if (already < length) {
 | |
| 		ssize_t nbyte;
 | |
| 		if (((nbyte = stream_read_try(zclient->ibuf, zclient->sock,
 | |
| 					      length - already))
 | |
| 		     == 0)
 | |
| 		    || (nbyte == -1)) {
 | |
| 			if (zclient_debug)
 | |
| 				zlog_debug(
 | |
| 					"zclient connection closed socket [%d].",
 | |
| 					zclient->sock);
 | |
| 			return zclient_failed(zclient);
 | |
| 		}
 | |
| 		if (nbyte != (ssize_t)(length - already)) {
 | |
| 			/* Try again later. */
 | |
| 			zclient_event(ZCLIENT_READ, zclient);
 | |
| 			return 0;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	length -= ZEBRA_HEADER_SIZE;
 | |
| 
 | |
| 	if (zclient_debug)
 | |
| 		zlog_debug("zclient 0x%p command %s VRF %u",
 | |
| 			   (void *)zclient, zserv_command_string(command),
 | |
| 			   vrf_id);
 | |
| 
 | |
| 	switch (command) {
 | |
| 	case ZEBRA_CAPABILITIES:
 | |
| 		zclient_capability_decode(command, zclient, length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_ROUTER_ID_UPDATE:
 | |
| 		if (zclient->router_id_update)
 | |
| 			(*zclient->router_id_update)(command, zclient, length,
 | |
| 						     vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VRF_ADD:
 | |
| 		zclient_vrf_add(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VRF_DELETE:
 | |
| 		zclient_vrf_delete(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_ADD:
 | |
| 		zclient_interface_add(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_DELETE:
 | |
| 		zclient_interface_delete(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_ADDRESS_ADD:
 | |
| 		if (zclient->interface_address_add)
 | |
| 			(*zclient->interface_address_add)(command, zclient,
 | |
| 							  length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_ADDRESS_DELETE:
 | |
| 		if (zclient->interface_address_delete)
 | |
| 			(*zclient->interface_address_delete)(command, zclient,
 | |
| 							     length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_BFD_DEST_UPDATE:
 | |
| 		if (zclient->interface_bfd_dest_update)
 | |
| 			(*zclient->interface_bfd_dest_update)(command, zclient,
 | |
| 							      length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_NBR_ADDRESS_ADD:
 | |
| 		if (zclient->interface_nbr_address_add)
 | |
| 			(*zclient->interface_nbr_address_add)(command, zclient,
 | |
| 							      length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_NBR_ADDRESS_DELETE:
 | |
| 		if (zclient->interface_nbr_address_delete)
 | |
| 			(*zclient->interface_nbr_address_delete)(
 | |
| 				command, zclient, length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_UP:
 | |
| 		zclient_interface_up(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_DOWN:
 | |
| 		zclient_interface_down(zclient, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_VRF_UPDATE:
 | |
| 		if (zclient->interface_vrf_update)
 | |
| 			(*zclient->interface_vrf_update)(command, zclient,
 | |
| 							 length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_NEXTHOP_UPDATE:
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("zclient rcvd nexthop update");
 | |
| 		if (zclient->nexthop_update)
 | |
| 			(*zclient->nexthop_update)(command, zclient, length,
 | |
| 						   vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IMPORT_CHECK_UPDATE:
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("zclient rcvd import check update");
 | |
| 		if (zclient->import_check_update)
 | |
| 			(*zclient->import_check_update)(command, zclient,
 | |
| 							length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_BFD_DEST_REPLAY:
 | |
| 		if (zclient->bfd_dest_replay)
 | |
| 			(*zclient->bfd_dest_replay)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_REDISTRIBUTE_ROUTE_ADD:
 | |
| 		if (zclient->redistribute_route_add)
 | |
| 			(*zclient->redistribute_route_add)(command, zclient,
 | |
| 							   length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_REDISTRIBUTE_ROUTE_DEL:
 | |
| 		if (zclient->redistribute_route_del)
 | |
| 			(*zclient->redistribute_route_del)(command, zclient,
 | |
| 							   length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_INTERFACE_LINK_PARAMS:
 | |
| 		if (zclient->interface_link_params)
 | |
| 			(*zclient->interface_link_params)(command, zclient,
 | |
| 							  length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_FEC_UPDATE:
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug("zclient rcvd fec update");
 | |
| 		if (zclient->fec_update)
 | |
| 			(*zclient->fec_update)(command, zclient, length);
 | |
| 		break;
 | |
| 	case ZEBRA_LOCAL_ES_ADD:
 | |
| 		if (zclient->local_es_add)
 | |
| 			(*zclient->local_es_add)(command, zclient, length,
 | |
| 						 vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_LOCAL_ES_DEL:
 | |
| 		if (zclient->local_es_del)
 | |
| 			(*zclient->local_es_del)(command, zclient, length,
 | |
| 						 vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VNI_ADD:
 | |
| 		if (zclient->local_vni_add)
 | |
| 			(*zclient->local_vni_add)(command, zclient, length,
 | |
| 						  vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VNI_DEL:
 | |
| 		if (zclient->local_vni_del)
 | |
| 			(*zclient->local_vni_del)(command, zclient, length,
 | |
| 						  vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_L3VNI_ADD:
 | |
| 		if (zclient->local_l3vni_add)
 | |
| 			(*zclient->local_l3vni_add)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_L3VNI_DEL:
 | |
| 		if (zclient->local_l3vni_del)
 | |
| 			(*zclient->local_l3vni_del)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_MACIP_ADD:
 | |
| 		if (zclient->local_macip_add)
 | |
| 			(*zclient->local_macip_add)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_MACIP_DEL:
 | |
| 		if (zclient->local_macip_del)
 | |
| 			(*zclient->local_macip_del)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IP_PREFIX_ROUTE_ADD:
 | |
| 		if (zclient->local_ip_prefix_add)
 | |
| 			(*zclient->local_ip_prefix_add)(command, zclient,
 | |
| 							length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IP_PREFIX_ROUTE_DEL:
 | |
| 		if (zclient->local_ip_prefix_del)
 | |
| 			(*zclient->local_ip_prefix_del)(command, zclient,
 | |
| 							length, vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_PW_STATUS_UPDATE:
 | |
| 		if (zclient->pw_status_update)
 | |
| 			(*zclient->pw_status_update)(command, zclient, length,
 | |
| 						     vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_ROUTE_NOTIFY_OWNER:
 | |
| 		if (zclient->route_notify_owner)
 | |
| 			(*zclient->route_notify_owner)(command, zclient, length,
 | |
| 						       vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_RULE_NOTIFY_OWNER:
 | |
| 		if (zclient->rule_notify_owner)
 | |
| 			(*zclient->rule_notify_owner)(command, zclient, length,
 | |
| 						      vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_GET_LABEL_CHUNK:
 | |
| 		if (zclient->label_chunk)
 | |
| 			(*zclient->label_chunk)(command, zclient, length,
 | |
| 						vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IPSET_NOTIFY_OWNER:
 | |
| 		if (zclient->ipset_notify_owner)
 | |
| 			(*zclient->ipset_notify_owner)(command, zclient, length,
 | |
| 						      vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IPSET_ENTRY_NOTIFY_OWNER:
 | |
| 		if (zclient->ipset_entry_notify_owner)
 | |
| 			(*zclient->ipset_entry_notify_owner)(command,
 | |
| 						     zclient, length,
 | |
| 						     vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_IPTABLE_NOTIFY_OWNER:
 | |
| 		if (zclient->iptable_notify_owner)
 | |
| 			(*zclient->iptable_notify_owner)(command,
 | |
| 						 zclient, length,
 | |
| 						 vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VXLAN_SG_ADD:
 | |
| 		if (zclient->vxlan_sg_add)
 | |
| 			(*zclient->vxlan_sg_add)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	case ZEBRA_VXLAN_SG_DEL:
 | |
| 		if (zclient->vxlan_sg_del)
 | |
| 			(*zclient->vxlan_sg_del)(command, zclient, length,
 | |
| 						    vrf_id);
 | |
| 		break;
 | |
| 	default:
 | |
| 		break;
 | |
| 	}
 | |
| 
 | |
| 	if (zclient->sock < 0)
 | |
| 		/* Connection was closed during packet processing. */
 | |
| 		return -1;
 | |
| 
 | |
| 	/* Register read thread. */
 | |
| 	stream_reset(zclient->ibuf);
 | |
| 	zclient_event(ZCLIENT_READ, zclient);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| void zclient_redistribute(int command, struct zclient *zclient, afi_t afi,
 | |
| 			  int type, unsigned short instance, vrf_id_t vrf_id)
 | |
| {
 | |
| 
 | |
| 	if (instance) {
 | |
| 		if (command == ZEBRA_REDISTRIBUTE_ADD) {
 | |
| 			if (redist_check_instance(
 | |
| 				    &zclient->mi_redist[afi][type], instance))
 | |
| 				return;
 | |
| 			redist_add_instance(&zclient->mi_redist[afi][type],
 | |
| 					    instance);
 | |
| 		} else {
 | |
| 			if (!redist_check_instance(
 | |
| 				    &zclient->mi_redist[afi][type], instance))
 | |
| 				return;
 | |
| 			redist_del_instance(&zclient->mi_redist[afi][type],
 | |
| 					    instance);
 | |
| 		}
 | |
| 
 | |
| 	} else {
 | |
| 		if (command == ZEBRA_REDISTRIBUTE_ADD) {
 | |
| 			if (vrf_bitmap_check(zclient->redist[afi][type],
 | |
| 					     vrf_id))
 | |
| 				return;
 | |
| 			vrf_bitmap_set(zclient->redist[afi][type], vrf_id);
 | |
| 		} else {
 | |
| 			if (!vrf_bitmap_check(zclient->redist[afi][type],
 | |
| 					      vrf_id))
 | |
| 				return;
 | |
| 			vrf_bitmap_unset(zclient->redist[afi][type], vrf_id);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if (zclient->sock > 0)
 | |
| 		zebra_redistribute_send(command, zclient, afi, type, instance,
 | |
| 					vrf_id);
 | |
| }
 | |
| 
 | |
| 
 | |
| void zclient_redistribute_default(int command, struct zclient *zclient,
 | |
| 				  afi_t afi, vrf_id_t vrf_id)
 | |
| {
 | |
| 
 | |
| 	if (command == ZEBRA_REDISTRIBUTE_DEFAULT_ADD) {
 | |
| 		if (vrf_bitmap_check(zclient->default_information[afi], vrf_id))
 | |
| 			return;
 | |
| 		vrf_bitmap_set(zclient->default_information[afi], vrf_id);
 | |
| 	} else {
 | |
| 		if (!vrf_bitmap_check(zclient->default_information[afi],
 | |
| 				      vrf_id))
 | |
| 			return;
 | |
| 		vrf_bitmap_unset(zclient->default_information[afi], vrf_id);
 | |
| 	}
 | |
| 
 | |
| 	if (zclient->sock > 0)
 | |
| 		zebra_redistribute_default_send(command, zclient, afi, vrf_id);
 | |
| }
 | |
| 
 | |
| static void zclient_event(enum event event, struct zclient *zclient)
 | |
| {
 | |
| 	switch (event) {
 | |
| 	case ZCLIENT_SCHEDULE:
 | |
| 		thread_add_event(zclient->master, zclient_connect, zclient, 0,
 | |
| 				 &zclient->t_connect);
 | |
| 		break;
 | |
| 	case ZCLIENT_CONNECT:
 | |
| 		if (zclient_debug)
 | |
| 			zlog_debug(
 | |
| 				"zclient connect failures: %d schedule interval is now %d",
 | |
| 				zclient->fail, zclient->fail < 3 ? 10 : 60);
 | |
| 		thread_add_timer(zclient->master, zclient_connect, zclient,
 | |
| 				 zclient->fail < 3 ? 10 : 60,
 | |
| 				 &zclient->t_connect);
 | |
| 		break;
 | |
| 	case ZCLIENT_READ:
 | |
| 		zclient->t_read = NULL;
 | |
| 		thread_add_read(zclient->master, zclient_read, zclient,
 | |
| 				zclient->sock, &zclient->t_read);
 | |
| 		break;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void zclient_interface_set_master(struct zclient *client,
 | |
| 				  struct interface *master,
 | |
| 				  struct interface *slave)
 | |
| {
 | |
| 	struct stream *s;
 | |
| 
 | |
| 	s = client->obuf;
 | |
| 	stream_reset(s);
 | |
| 
 | |
| 	zclient_create_header(s, ZEBRA_INTERFACE_SET_MASTER, master->vrf_id);
 | |
| 
 | |
| 	stream_putl(s, master->vrf_id);
 | |
| 	stream_putl(s, master->ifindex);
 | |
| 	stream_putl(s, slave->vrf_id);
 | |
| 	stream_putl(s, slave->ifindex);
 | |
| 
 | |
| 	stream_putw_at(s, 0, stream_get_endp(s));
 | |
| 	zclient_send_message(client);
 | |
| }
 | 
