Started work on BGP. Wrote main part of the connection handling code.

1 files changed, 259 insertions, 1 deletions

diff --git a/proto/bgp/bgp.c b/proto/bgp/bgp.c
index 26e3906..40915a4 100644
--- a/proto/bgp/bgp.c
+++ b/proto/bgp/bgp.c
@@ -12,10 +12,20 @@
 #include "nest/iface.h"
 #include "nest/protocol.h"
 #include "nest/route.h"
+#include "nest/locks.h"
 #include "conf/conf.h"
+#include "lib/socket.h"
 
 #include "bgp.h"
 
+static sock *bgp_listen_sk;		/* Global listening socket */
+static int bgp_counter;			/* Number of protocol instances using the listening socket */
+static list bgp_list;			/* List of active BGP instances */
+
+static void bgp_close_conn(struct bgp_conn *conn);
+static void bgp_connect(struct bgp_proto *p);
+static void bgp_setup_sk(struct bgp_proto *p, struct bgp_conn *conn, sock *s);
+
 static void
 bgp_rt_notify(struct proto *P, net *n, rte *new, rte *old, ea_list *tmpa)
 {
@@ -29,18 +39,266 @@ bgp_init(struct proto_config *C)
   struct bgp_proto *p = (struct bgp_proto *) P;
 
   P->rt_notify = bgp_rt_notify;
+  p->cf = 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
+bgp_close(struct bgp_proto *p)
+{
+  rem_node(&p->bgp_node);
+  ASSERT(bgp_counter);
+  bgp_counter--;
+  if (!bgp_counter)
+    {
+      rfree(bgp_listen_sk);
+      bgp_listen_sk = NULL;
+    }
+  /* FIXME: Automatic restart after errors? */
+}
+
+static void
+bgp_reset(struct bgp_proto *p)
+{
+  bgp_close(p);
+  proto_notify_state(&p->p, PS_DOWN);
+}
+
+static void
+bgp_start_timer(timer *t, int value)
+{
+  /* FIXME: Randomize properly */
+  /* FIXME: Check if anybody uses tm_start directly */
+  tm_start(t, value);
+}
+
+static int
+bgp_rx(sock *sk, int size)
+{
+  DBG("BGP: Got %d bytes\n", size);
+
+  return 1;				/* Start from the beginning */
+}
+
+static void
+bgp_send_open(struct bgp_conn *conn)
+{
+  DBG("BGP: Sending open\n");
+  conn->sk->rx_hook = bgp_rx;
+  tm_stop(conn->connect_retry_timer);
+  /* FIXME */
+  conn->state = BS_OPENSENT;
+}
+
+static int
+bgp_connected(sock *sk, int dummy)
+{
+  struct bgp_conn *conn = sk->data;
+
+  DBG("BGP: Connected\n");
+  bgp_send_open(conn);
+  return 0;
+}
+
+static void
+bgp_connect_timeout(timer *t)
+{
+  struct bgp_proto *p = t->data;
+  struct bgp_conn *conn = &p->conn;
+
+  DBG("BGP: Connect timeout, retrying\n");
+  bgp_close_conn(conn);
+  bgp_connect(p);
+}
+
+static void
+bgp_err(sock *sk, int err)
+{
+  struct bgp_conn *conn = sk->data;
+
+  DBG("BGP: Socket error %d in state %d\n", err, conn->state);
+  sk->type = SK_DELETED;		/* FIXME: Need to do this always! */
+  switch (conn->state)
+    {
+    case BS_CONNECT:
+    case BS_OPENSENT:
+      conn->state = BS_ACTIVE;
+      bgp_start_timer(conn->connect_retry_timer, conn->bgp->cf->connect_retry_time);
+      break;
+    case BS_OPENCONFIRM:
+    case BS_ESTABLISHED:
+      /* FIXME: Should close the connection and go to Idle state */
+    default:
+      bug("bgp_err called in invalid state %d", conn->state);
+    }
+}
+
+static int
+bgp_incoming_connection(sock *sk, int dummy)
+{
+  node *n;
+
+  DBG("BGP: Incoming connection from %I port %d\n", sk->daddr, sk->dport);
+  WALK_LIST(n, bgp_list)
+    {
+      struct bgp_proto *p = SKIP_BACK(struct bgp_proto, bgp_node, n);
+      if (ipa_equal(p->cf->remote_ip, sk->daddr) && sk->dport == BGP_PORT)
+	{
+	  DBG("BGP: Authorized\n");
+	  if (p->incoming_conn.sk)
+	    {
+	      DBG("BGP: But one incoming connection already exists, how is that possible?\n");
+	      break;
+	    }
+	  bgp_setup_sk(p, &p->incoming_conn, sk);
+	  bgp_send_open(&p->incoming_conn);
+	  return 0;
+	}
+    }
+  DBG("BGP: Unauthorized\n");
+  rfree(sk);
+  return 0;
+}
+
+static void
+bgp_setup_sk(struct bgp_proto *p, struct bgp_conn *conn, sock *s)
+{
+  timer *t;
+
+  s->data = conn;
+  s->ttl = p->cf->multihop ? : 1;
+  s->rbsize = BGP_RX_BUFFER_SIZE;
+#if 0
+  s->tx_hook = bgp_tx;
+#endif
+  s->err_hook = bgp_err;
+  s->tos = IP_PREC_INTERNET_CONTROL;
+
+  conn->bgp = p;
+  conn->sk = s;
+
+  t = conn->connect_retry_timer = tm_new(p->p.pool);
+  t->hook = bgp_connect_timeout;
+  t->data = p;
+#if 0
+  t = p->hold_timer = tm_new(p->p.pool);
+  t->hook = bgp_hold_timeout;
+  t->data = p;
+  t = p->keepalive_timer = tm_new(p->p.pool);
+  t->hook = bgp_keepalive_timeout;
+  t->data = p;
+#endif
+}
+
+static void
+bgp_close_conn(struct bgp_conn *conn)
+{
+  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;
+  rfree(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;
+
+  DBG("BGP: Connecting\n");
+  s = sk_new(p->p.pool);
+  s->type = SK_TCP_ACTIVE;
+  s->saddr = _MI(0x3ea80001);		/* FIXME: Hack */
+  s->daddr = p->cf->remote_ip;
+#if 0
+  s->sport =				/* FIXME */
+#endif
+  s->dport = BGP_PORT;
+  s->rx_hook = bgp_connected;
+  bgp_setup_sk(p, conn, s);
+  conn->state = BS_CONNECT;
+  if (sk_open(s))
+    {
+      bgp_err(s, 0);
+      return;
+    }
+  DBG("BGP: Waiting for connect success\n");
+  bgp_start_timer(conn->connect_retry_timer, p->cf->connect_retry_time);
+}
+
+static void
+bgp_start_locked(struct object_lock *lock)
+{
+  struct bgp_proto *p = lock->data;
+
+  DBG("BGP: Got lock\n");
+  if (!bgp_counter++)
+    init_list(&bgp_list);
+  if (!bgp_listen_sk)
+    {
+      sock *s = sk_new(&root_pool);
+      DBG("BGP: Creating incoming socket\n");
+      s->type = SK_TCP_PASSIVE;
+      s->sport = BGP_PORT;
+      s->tos = IP_PREC_INTERNET_CONTROL;
+      s->ttl = 1;
+      s->rbsize = BGP_RX_BUFFER_SIZE;
+      s->rx_hook = bgp_incoming_connection;
+      if (sk_open(s))
+	{
+	  log(L_ERR "Unable to open incoming BGP socket");
+	  rfree(s);
+	}
+      else
+	bgp_listen_sk = s;
+    }
+  add_tail(&bgp_list, &p->bgp_node);
+  bgp_connect(p);			/* FIXME: Use neighbor cache for fast up/down transitions? */
+}
+
 static int
 bgp_start(struct proto *P)
 {
-  return PS_UP;
+  struct bgp_proto *p = (struct bgp_proto *) P;
+  struct object_lock *lock;
+
+  /*
+   *  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;
+  lock->port = BGP_PORT;
+  lock->iface = NULL;
+  lock->hook = bgp_start_locked;
+  lock->data = p;
+  olock_acquire(lock);
+  return PS_START;
 }
 
 static int
 bgp_shutdown(struct proto *P)
 {
+  struct bgp_proto *p = (struct bgp_proto *) P;
+
+  DBG("BGP: Explicit shutdown\n");
+
+  bgp_close(p);
   return PS_DOWN;
 }