Commit b552ecc4 authored by Martin Mareš's avatar Martin Mareš

Connection state machine works.

parent ce0603a6
......@@ -42,13 +42,11 @@ bgp_init(struct proto_config *C)
p->local_as = c->local_as;
p->remote_as = c->remote_as;
p->is_internal = (c->local_as == c->remote_as);
p->conn.state = BS_IDLE;
p->incoming_conn.state = BS_IDLE;
p->local_id = C->global->router_id;
return P;
}
static void
void
bgp_close(struct bgp_proto *p)
{
rem_node(&p->bgp_node);
......@@ -62,13 +60,6 @@ bgp_close(struct bgp_proto *p)
/* FIXME: Automatic restart after errors? */
}
static void /* FIXME: Nobody uses */
bgp_reset(struct bgp_proto *p)
{
bgp_close(p);
proto_notify_state(&p->p, PS_DOWN);
}
void
bgp_start_timer(timer *t, int value)
{
......@@ -76,6 +67,32 @@ bgp_start_timer(timer *t, int value)
/* FIXME: Check if anybody uses tm_start directly */
if (value)
tm_start(t, value);
else
tm_stop(t);
}
void
bgp_close_conn(struct bgp_conn *conn)
{
struct bgp_proto *p = conn->bgp;
DBG("BGP: Closing connection\n");
conn->packets_to_send = 0;
rfree(conn->connect_retry_timer);
conn->connect_retry_timer = NULL;
rfree(conn->keepalive_timer);
conn->keepalive_timer = NULL;
rfree(conn->hold_timer);
conn->hold_timer = NULL;
sk_close(conn->sk);
conn->sk = NULL;
conn->state = BS_IDLE;
if (conn->primary)
{
bgp_close(p);
p->conn = NULL;
proto_notify_state(&p->p, PS_DOWN);
}
}
static void
......@@ -83,6 +100,7 @@ bgp_send_open(struct bgp_conn *conn)
{
DBG("BGP: Sending open\n");
conn->sk->rx_hook = bgp_rx;
conn->sk->tx_hook = bgp_tx;
tm_stop(conn->connect_retry_timer);
bgp_schedule_packet(conn, PKT_OPEN);
conn->state = BS_OPENSENT;
......@@ -123,7 +141,7 @@ bgp_sock_err(sock *sk, int err)
break;
case BS_OPENCONFIRM:
case BS_ESTABLISHED:
/* FIXME */
break;
default:
bug("bgp_sock_err called in invalid state %d", conn->state);
}
......@@ -162,7 +180,7 @@ bgp_hold_timeout(timer *t)
{
struct bgp_conn *conn = t->data;
DBG("BGP: Hold timeout, closing connection\n"); /* FIXME: Check states? */
DBG("BGP: Hold timeout, closing connection\n");
bgp_error(conn, 4, 0, 0, 0);
}
......@@ -184,13 +202,14 @@ bgp_setup_sk(struct bgp_proto *p, struct bgp_conn *conn, sock *s)
s->ttl = p->cf->multihop ? : 1;
s->rbsize = BGP_RX_BUFFER_SIZE;
s->tbsize = BGP_TX_BUFFER_SIZE;
s->tx_hook = bgp_tx;
s->err_hook = bgp_sock_err;
s->tos = IP_PREC_INTERNET_CONTROL;
conn->bgp = p;
conn->sk = s;
conn->packets_to_send = 0;
conn->error_flag = 0;
conn->primary = 0;
t = conn->connect_retry_timer = tm_new(p->p.pool);
t->hook = bgp_connect_timeout;
......@@ -203,27 +222,11 @@ bgp_setup_sk(struct bgp_proto *p, struct bgp_conn *conn, sock *s)
t->data = conn;
}
void
bgp_close_conn(struct bgp_conn *conn)
{
DBG("BGP: Closing connection\n");
conn->packets_to_send = 0;
rfree(conn->connect_retry_timer);
conn->connect_retry_timer = NULL;
rfree(conn->keepalive_timer);
conn->keepalive_timer = NULL;
rfree(conn->hold_timer);
conn->hold_timer = NULL;
sk_close(conn->sk);
conn->sk = NULL;
conn->state = BS_IDLE;
}
static void
bgp_connect(struct bgp_proto *p) /* Enter Connect state and start establishing connection */
{
sock *s;
struct bgp_conn *conn = &p->conn;
struct bgp_conn *conn = &p->outgoing_conn;
DBG("BGP: Connecting\n");
s = sk_new(p->p.pool);
......@@ -282,13 +285,16 @@ bgp_start(struct proto *P)
struct bgp_proto *p = (struct bgp_proto *) P;
struct object_lock *lock;
DBG("BGP: Startup.\n");
p->outgoing_conn.state = BS_IDLE;
p->incoming_conn.state = BS_IDLE;
/*
* Before attempting to create the connection, we need to lock the
* port, so that are sure we're the only instance attempting to talk
* with that neighbor.
*/
DBG("BGP: Startup. Acquiring lock.\n");
lock = p->lock = olock_new(P->pool);
lock->addr = p->cf->remote_ip;
lock->type = OBJLOCK_TCP;
......@@ -300,6 +306,27 @@ bgp_start(struct proto *P)
return PS_START;
}
static int
bgp_graceful_close(struct bgp_conn *c)
{
switch (c->state)
{
case BS_IDLE:
return 0;
case BS_CONNECT:
case BS_ACTIVE:
bgp_close_conn(c);
return 1;
case BS_OPENSENT:
case BS_OPENCONFIRM:
case BS_ESTABLISHED:
bgp_error(c, 6, 0, 0, 0);
return 1;
default:
bug("bgp_graceful_close: Unknown state %d", c->state);
}
}
static int
bgp_shutdown(struct proto *P)
{
......@@ -307,8 +334,30 @@ bgp_shutdown(struct proto *P)
DBG("BGP: Explicit shutdown\n");
bgp_close(p);
return PS_DOWN;
/*
* We want to send the Cease notification message to all connections
* we have open, but we don't want to wait for all of them to complete.
* We are willing to handle the primary connection carefully, but for
* the others we just try to send the packet and if there is no buffer
* space free, we'll gracefully finish.
*/
proto_notify_state(&p->p, PS_STOP);
if (!p->conn)
{
if (p->outgoing_conn.state != BS_IDLE)
p->outgoing_conn.primary = 1; /* Shuts protocol down after connection close */
else if (p->incoming_conn.state != BS_IDLE)
p->incoming_conn.primary = 1;
}
if (bgp_graceful_close(&p->outgoing_conn) || bgp_graceful_close(&p->incoming_conn))
return p->p.proto_state;
else
{
/* No connections open, shutdown automatically */
bgp_close(p);
return PS_DOWN;
}
}
void
......@@ -322,7 +371,8 @@ bgp_error(struct bgp_conn *c, unsigned code, unsigned subcode, unsigned data, un
c->notify_subcode = subcode;
c->notify_arg = data;
c->notify_arg_size = len;
proto_notify_state(&c->bgp->p, PS_STOP);
if (c->primary)
proto_notify_state(&c->bgp->p, PS_STOP);
bgp_schedule_packet(c, PKT_NOTIFICATION);
}
......
......@@ -29,6 +29,7 @@ struct bgp_conn {
int packets_to_send; /* Bitmap of packet types to be sent */
int notify_code, notify_subcode, notify_arg, notify_arg_size;
int error_flag; /* Error state, ignore all input */
int primary; /* This connection is primary */
unsigned hold_time, keepalive_time; /* Times calculated from my and neighbor's requirements */
};
......@@ -40,7 +41,8 @@ struct bgp_proto {
int is_internal; /* Internal BGP connection (local_as == remote_as) */
u32 local_id; /* BGP identifier of this router */
u32 remote_id; /* BGP identifier of the neighbor */
struct bgp_conn conn; /* Our primary connection */
struct bgp_conn *conn; /* Connection we have established */
struct bgp_conn outgoing_conn; /* Outgoing connection we're working with */
struct bgp_conn incoming_conn; /* Incoming connection we have neither accepted nor rejected yet */
struct object_lock *lock; /* Lock for neighbor connection */
};
......@@ -55,7 +57,7 @@ struct bgp_proto {
void bgp_start_timer(struct timer *t, int value);
void bgp_check(struct bgp_config *c);
void bgp_error(struct bgp_conn *c, unsigned code, unsigned subcode, unsigned data, unsigned len);
void bgp_close_conn(struct bgp_conn *conn);
void bgp_close_conn(struct bgp_conn *c);
/* attrs.c */
......
......@@ -26,7 +26,7 @@ bgp_proto_start: proto_start BGP {
this_proto->preference = DEF_PREF_BGP;
BGP_CFG->hold_time = 240;
BGP_CFG->connect_retry_time = 120;
BGP_CFG->initial_hold_time = 300;
BGP_CFG->initial_hold_time = 240;
}
;
......
......@@ -26,14 +26,11 @@ bgp_create_notification(struct bgp_conn *conn, byte *buf)
buf[1] = conn->notify_subcode;
switch (conn->notify_arg_size)
{
case 1:
buf[2] = conn->notify_arg; return buf+3;
case 2:
put_u16(buf+2, conn->notify_arg); return buf+4;
case 4:
put_u32(buf+2, conn->notify_arg); return buf+6;
default:
bug("bgp_create_notification: unknown error code size");
case 0: return buf + 1;
case 1: buf[2] = conn->notify_arg; return buf+3;
case 2: put_u16(buf+2, conn->notify_arg); return buf+4;
case 4: put_u32(buf+2, conn->notify_arg); return buf+6;
default: bug("bgp_create_notification: unknown error code size");
}
}
......@@ -140,6 +137,7 @@ bgp_tx(sock *sk)
static void
bgp_rx_open(struct bgp_conn *conn, byte *pkt, int len)
{
struct bgp_conn *other;
struct bgp_proto *p = conn->bgp;
struct bgp_config *cf = p->cf;
unsigned as, hold;
......@@ -168,8 +166,39 @@ bgp_rx_open(struct bgp_conn *conn, byte *pkt, int len)
if (!id || id == 0xffffffff || id == p->local_id)
{ bgp_error(conn, 2, 3, id, 0); return; }
/* FIXME: What to do with the other connection??? */
ASSERT(conn == &p->conn);
/* Check the other connection */
other = (conn == &p->outgoing_conn) ? &p->incoming_conn : &p->outgoing_conn;
switch (other->state)
{
case BS_IDLE:
break;
case BS_CONNECT:
case BS_ACTIVE:
case BS_OPENSENT:
DBG("BGP: Collision, closing the other connection\n");
bgp_close_conn(other);
break;
case BS_OPENCONFIRM:
if ((p->local_id < id) == (conn == &p->incoming_conn))
{
/* Should close the other connection */
DBG("BGP: Collision, closing the other connection\n");
bgp_error(other, 6, 0, 0, 0);
break;
}
/* Fall thru */
case BS_ESTABLISHED:
/* Should close this connection */
DBG("BGP: Collision, closing this connection\n");
bgp_error(conn, 6, 0, 0, 0);
return;
default:
bug("bgp_rx_open: Unknown state");
}
/* Make this connection primary */
conn->primary = 1;
p->conn = conn;
/* Update our local variables */
if (hold < p->cf->hold_time)
......@@ -216,7 +245,8 @@ bgp_rx_notification(struct bgp_conn *conn, byte *pkt, int len)
}
DBG("BGP: NOTIFICATION %d.%d %08x\n", pkt[19], pkt[20], arg); /* FIXME: Better reporting */
conn->error_flag = 1;
proto_notify_state(&conn->bgp->p, PS_STOP);
if (conn->primary)
proto_notify_state(&conn->bgp->p, PS_STOP);
bgp_schedule_packet(conn, PKT_SCHEDULE_CLOSE);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment