bgpd: add packet send hook

Unlike MRT dumps, BMP also provides packets sent by the router. Add another hook for that. Signed-off-by: David Lamparter <equinox@opensourcerouting.org>
idryzhov · Jul 3, 2019 · 6fd0459 · 6fd0459
1 parent 584470f
commit 6fd0459
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/bgpd/bgp_packet.c b/bgpd/bgp_packet.c
@@ -68,6 +68,11 @@ DEFINE_HOOK(bgp_packet_dump,
 			struct stream *s),
 		(peer, type, size, s))
 
+DEFINE_HOOK(bgp_packet_send,
+		(struct peer *peer, uint8_t type, bgp_size_t size,
+			struct stream *s),
+		(peer, type, size, s))
+
 /**
  * Sets marker and type fields for a BGP message.
  *
@@ -547,6 +552,7 @@ void bgp_open_send(struct peer *peer)
 
 	/* Dump packet if debug option is set. */
 	/* bgp_packet_dump (s); */
+	hook_call(bgp_packet_send, peer, BGP_MSG_OPEN, stream_get_endp(s), s);
 
 	/* Add packet to the peer. */
 	bgp_packet_add(peer, s);

diff --git a/bgpd/bgp_packet.h b/bgpd/bgp_packet.h
@@ -28,6 +28,11 @@ DECLARE_HOOK(bgp_packet_dump,
 			struct stream *s),
 		(peer, type, size, s))
 
+DECLARE_HOOK(bgp_packet_send,
+		(struct peer *peer, uint8_t type, bgp_size_t size,
+			struct stream *s),
+		(peer, type, size, s))
+
 #define BGP_NLRI_LENGTH       1U
 #define BGP_TOTAL_ATTR_LEN    2U
 #define BGP_UNFEASIBLE_LEN    2U