mirrors/bird
mirrors/bird
0
0
Fork 0
mirror of https://gitlab.nic.cz/labs/bird.git synced 2024-11-10 05:08:42 +00:00
Code Releases Activity

BMP: Integrate bmp_conn to bmp_proto

Browse Source 
There is only one socket per BMP instance, no need to have separate
struct (like in BGP).
This commit is contained in:
Ondrej Zajicek (work) 2021-03-28 16:41:53 +02:00 committed by Igor Putovny
parent 767c820e57
commit e7789317b2
2 changed files with 29 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
proto/bmp/bmp.c
View File

@ -258,10 +258,10 @@ bmp_schedule_tx_packet(struct bmp_proto *p, const byte *payload, const size_t si
tx_data->data_size = size; tx_data->data_size = size;
add_tail(&p->tx_queue, &tx_data->n); add_tail(&p->tx_queue, &tx_data->n);
if (sk_tx_buffer_empty(p->conn->sk) if (sk_tx_buffer_empty(p->sk)
&& !ev_active(p->conn->tx_ev)) && !ev_active(p->tx_ev))
{ {
ev_schedule(p->conn->tx_ev); ev_schedule(p->tx_ev);
} }
} }
@ -281,18 +281,11 @@ bmp_startup(struct bmp_proto *p)
p->started = true; p->started = true;
} }
void static void
bmp_fire_tx(void *vconn) bmp_fire_tx(void *p_)
{ {
struct bmp_conn *conn = (void *) vconn; struct bmp_proto *p = p_;
struct bmp_proto *p = conn->bmp; byte *buf = p->sk->tbuf;
IF_PTR_IS_NULL_PRINT_ERR_MSG_AND_RETURN_OPT_VAL(
conn->sk,
"Socket is null"
);
byte *buf = conn->sk->tbuf;
IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL( IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL(
EMPTY_LIST(p->tx_queue), EMPTY_LIST(p->tx_queue),
"Called BMP TX event handler when there is not any data to send" "Called BMP TX event handler when there is not any data to send"
@ -303,9 +296,9 @@ bmp_fire_tx(void *vconn)
struct bmp_data_node *tx_data_next; struct bmp_data_node *tx_data_next;
WALK_LIST_DELSAFE(tx_data, tx_data_next, p->tx_queue) WALK_LIST_DELSAFE(tx_data, tx_data_next, p->tx_queue)
{ {
if (tx_data->data_size > conn->sk->tbsize) if (tx_data->data_size > p->sk->tbsize)
{ {
sk_set_tbsize(conn->sk, tx_data->data_size); sk_set_tbsize(p->sk, tx_data->data_size);
} }
size_t data_size = tx_data->data_size; size_t data_size = tx_data->data_size;
@ -314,7 +307,7 @@ bmp_fire_tx(void *vconn)
rem_node((node *) tx_data); rem_node((node *) tx_data);
mb_free(tx_data); mb_free(tx_data);
IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL( IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL(
(sk_send(conn->sk, data_size) <= 0), (sk_send(p->sk, data_size) <= 0),
"Failed to send BMP packet" "Failed to send BMP packet"
); );
@ -323,9 +316,9 @@ bmp_fire_tx(void *vconn)
// call // call
if (++cnt > 32) if (++cnt > 32)
{ {
if (!ev_active(conn->tx_ev)) if (!ev_active(p->tx_ev))
{ {
ev_schedule(conn->tx_ev); ev_schedule(p->tx_ev);
} }
return; return;
@ -342,7 +335,7 @@ bmp_tx(struct birdsock *sk)
static inline int static inline int
bmp_open_socket(struct bmp_proto *p) bmp_open_socket(struct bmp_proto *p)
{ {
sock *s = p->conn->sk; sock *s = p->sk;
s->daddr = p->station_ip; s->daddr = p->station_ip;
s->dport = p->station_port; s->dport = p->station_port;
s->err_hook = bmp_sock_err; s->err_hook = bmp_sock_err;
@ -358,7 +351,7 @@ bmp_open_socket(struct bmp_proto *p)
static void static void
bmp_connection_retry(timer *t) bmp_connection_retry(timer *t)
{ {
struct bmp_proto *p = (void *) t->data; struct bmp_proto *p = t->data;
if (bmp_open_socket(p) < 0) if (bmp_open_socket(p) < 0)
{ {
@ -373,8 +366,8 @@ bmp_connection_retry(timer *t)
void void
bmp_sock_err(sock *sk, int err) bmp_sock_err(sock *sk, int err)
{ {
struct bmp_conn *conn = (void *) sk->data; struct bmp_proto *p = sk->data;
log(L_WARN "[BMP:%s] Socket error: %M", conn->bmp->p.name, err); log(L_WARN "[BMP:%s] Socket error: %M", p->p.name, err);
} }
static inline void static inline void
@ -510,7 +503,7 @@ bmp_open(const struct proto *P)
p->map_mem_pool = rp_new(P->pool, "BMP Map"); p->map_mem_pool = rp_new(P->pool, "BMP Map");
p->tx_mem_pool = rp_new(P->pool, "BMP Tx"); p->tx_mem_pool = rp_new(P->pool, "BMP Tx");
p->update_msg_mem_pool = rp_new(P->pool, "BMP Update"); p->update_msg_mem_pool = rp_new(P->pool, "BMP Update");
p->conn->tx_ev = ev_new_init(p->tx_mem_pool, bmp_fire_tx, p->conn); p->tx_ev = ev_new_init(p->tx_mem_pool, bmp_fire_tx, p);
bmp_peer_map_init(&p->peer_open_msg.tx_msg, p->map_mem_pool); bmp_peer_map_init(&p->peer_open_msg.tx_msg, p->map_mem_pool);
bmp_peer_map_init(&p->peer_open_msg.rx_msg, p->map_mem_pool); bmp_peer_map_init(&p->peer_open_msg.rx_msg, p->map_mem_pool);
@ -1040,15 +1033,14 @@ bmp_send_termination_msg(struct bmp_proto *p,
+ BMP_TERM_INFO_LEN_FIELD_SIZE + BMP_TERM_INFO_LEN_FIELD_SIZE
+ BMP_TERM_REASON_CODE_SIZE; + BMP_TERM_REASON_CODE_SIZE;
const size_t term_msg_size = BMP_COMMON_HDR_SIZE + term_msg_hdr_size; const size_t term_msg_size = BMP_COMMON_HDR_SIZE + term_msg_hdr_size;
buffer stream buffer stream = bmp_buffer_alloc(p->buffer_mpool, term_msg_size);
= bmp_buffer_alloc(p->buffer_mpool, term_msg_size);
bmp_common_hdr_serialize(&stream, BMP_TERM_MSG, term_msg_hdr_size); bmp_common_hdr_serialize(&stream, BMP_TERM_MSG, term_msg_hdr_size);
bmp_put_u16(&stream, BMP_TERM_INFO_REASON); bmp_put_u16(&stream, BMP_TERM_INFO_REASON);
bmp_put_u16(&stream, BMP_TERM_REASON_CODE_SIZE); // 2-byte code indication the reason bmp_put_u16(&stream, BMP_TERM_REASON_CODE_SIZE); // 2-byte code indication the reason
bmp_put_u16(&stream, reason); bmp_put_u16(&stream, reason);
memcpy(p->conn->sk->tbuf, bmp_buffer_data(&stream), bmp_buffer_pos(&stream)); memcpy(p->sk->tbuf, bmp_buffer_data(&stream), bmp_buffer_pos(&stream));
IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL( IF_COND_TRUE_PRINT_ERR_MSG_AND_RETURN_OPT_VAL(
sk_send(p->conn->sk, bmp_buffer_pos(&stream)) < 0, sk_send(p->sk, bmp_buffer_pos(&stream)) < 0,
"Failed to send BMP termination message" "Failed to send BMP termination message"
); );
@ -1058,10 +1050,9 @@ bmp_send_termination_msg(struct bmp_proto *p,
static void static void
bmp_station_connected(struct birdsock *sk) bmp_station_connected(struct birdsock *sk)
{ {
struct bmp_conn *conn = (void *) sk->data; struct bmp_proto *p = (void *) sk->data;
struct bmp_proto *p = conn->bmp;
conn->sk->tx_hook = bmp_tx; sk->tx_hook = bmp_tx;
p->station_connected = true; p->station_connected = true;
bmp_startup(p); bmp_startup(p);
@ -1073,17 +1064,17 @@ bmp_station_connected(struct birdsock *sk)
} }
static inline void static inline void
bmp_setup_socket(struct bmp_conn *conn) bmp_setup_socket(struct bmp_proto *p)
{ {
sock *sk = sk_new(proto_pool); sock *sk = sk_new(p->tx_mem_pool);
sk->type = SK_TCP_ACTIVE; sk->type = SK_TCP_ACTIVE;
sk->ttl = IP4_MAX_TTL; sk->ttl = IP4_MAX_TTL;
sk->tos = IP_PREC_INTERNET_CONTROL; sk->tos = IP_PREC_INTERNET_CONTROL;
sk->tbsize = BGP_TX_BUFFER_EXT_SIZE; sk->tbsize = BGP_TX_BUFFER_EXT_SIZE;
sk->tx_hook = bmp_station_connected; sk->tx_hook = bmp_station_connected;
conn->sk = sk; p->sk = sk;
sk->data = conn; sk->data = p;
} }
/** Configuration handle section **/ /** Configuration handle section **/
@ -1111,9 +1102,7 @@ bmp_start(struct proto *P)
{ {
struct bmp_proto *p = (void *) P; struct bmp_proto *p = (void *) P;
p->conn = mb_allocz(P->pool, sizeof (struct bmp_conn)); bmp_setup_socket(p);
p->conn->bmp = p;
bmp_setup_socket(p->conn);
bmp_open(P); bmp_open(P);
g_bmp = p; g_bmp = p;

9
proto/bmp/bmp.h
View File

@ -46,12 +46,6 @@ struct bmp_config {
struct bgp_proto; struct bgp_proto;
struct bmp_proto; struct bmp_proto;
struct bmp_conn {
struct bmp_proto *bmp;
struct birdsock *sk;
event *tx_ev;
};
// Stores sent and received BGP OPEN MSGs // Stores sent and received BGP OPEN MSGs
struct bmp_peer_open_msg { struct bmp_peer_open_msg {
struct bmp_peer_map tx_msg; struct bmp_peer_map tx_msg;
@ -70,7 +64,8 @@ struct rt_table_info {
struct bmp_proto { struct bmp_proto {
struct proto p; // Parent proto struct proto p; // Parent proto
const struct bmp_config *cf; // Shortcut to BMP configuration const struct bmp_config *cf; // Shortcut to BMP configuration
struct bmp_conn *conn; // Connection we have established sock *sk; // TCP connection
event *tx_ev; // TX event
char sys_descr[MIB_II_STR_LEN]; // sysDescr MIB-II [RFC1213] object char sys_descr[MIB_II_STR_LEN]; // sysDescr MIB-II [RFC1213] object
char sys_name[MIB_II_STR_LEN]; // sysName MIB-II [RFC1213] object char sys_name[MIB_II_STR_LEN]; // sysName MIB-II [RFC1213] object
ip_addr station_ip; // Monitoring station IP address ip_addr station_ip; // Monitoring station IP address