summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--proto/bgp/Makefile2
-rw-r--r--proto/bgp/bgp.c14
-rw-r--r--proto/bgp/bgp.h18
-rw-r--r--proto/bgp/packets.c141
4 files changed, 160 insertions, 15 deletions
diff --git a/proto/bgp/Makefile b/proto/bgp/Makefile
index 0f0ba27..a634cf0 100644
--- a/proto/bgp/Makefile
+++ b/proto/bgp/Makefile
@@ -1,4 +1,4 @@
-source=bgp.c attrs.c
+source=bgp.c attrs.c packets.c
root-rel=../../
dir-name=proto/bgp
diff --git a/proto/bgp/bgp.c b/proto/bgp/bgp.c
index 40915a4..c52e8bd 100644
--- a/proto/bgp/bgp.c
+++ b/proto/bgp/bgp.c
@@ -78,21 +78,13 @@ bgp_start_timer(timer *t, int value)
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 */
+ bgp_schedule_packet(conn, PKT_OPEN);
conn->state = BS_OPENSENT;
}
@@ -174,14 +166,14 @@ bgp_setup_sk(struct bgp_proto *p, struct bgp_conn *conn, sock *s)
s->data = conn;
s->ttl = p->cf->multihop ? : 1;
s->rbsize = BGP_RX_BUFFER_SIZE;
-#if 0
+ s->tbsize = BGP_TX_BUFFER_SIZE;
s->tx_hook = bgp_tx;
-#endif
s->err_hook = bgp_err;
s->tos = IP_PREC_INTERNET_CONTROL;
conn->bgp = p;
conn->sk = s;
+ conn->packets_to_send = 0;
t = conn->connect_retry_timer = tm_new(p->p.pool);
t->hook = bgp_connect_timeout;
diff --git a/proto/bgp/bgp.h b/proto/bgp/bgp.h
index 8e2e8df..3ddf619 100644
--- a/proto/bgp/bgp.h
+++ b/proto/bgp/bgp.h
@@ -22,10 +22,13 @@ struct bgp_config {
struct bgp_conn {
struct bgp_proto *bgp;
struct birdsock *sk;
- int state; /* State of connection state machine */
+ unsigned int state; /* State of connection state machine */
struct timer *connect_retry_timer;
struct timer *hold_timer;
struct timer *keepalive_timer;
+ unsigned int packets_to_send; /* Bitmap of packet types to be sent */
+ unsigned int notify_code, notify_subcode, notify_arg, notify_arg_size;
+ unsigned int error_flag; /* Error state, ignore all input */
};
struct bgp_proto {
@@ -42,8 +45,12 @@ struct bgp_proto {
struct object_lock *lock; /* Lock for neighbor connection */
};
-#define BGP_PORT 179
-#define BGP_RX_BUFFER_SIZE 4096
+#define BGP_PORT 179
+#define BGP_VERSION 4
+#define BGP_HEADER_LENGTH 19
+#define BGP_MAX_PACKET_LENGTH 4096
+#define BGP_RX_BUFFER_SIZE 4096
+#define BGP_TX_BUFFER_SIZE BGP_MAX_PACKET_LENGTH
void bgp_check(struct bgp_config *c);
@@ -51,12 +58,17 @@ void bgp_check(struct bgp_config *c);
/* packets.c */
+void bgp_schedule_packet(struct bgp_conn *conn, int type);
+void bgp_tx(struct birdsock *sk);
+int bgp_rx(struct birdsock *sk, int size);
+
/* Packet types */
#define PKT_OPEN 0x01
#define PKT_UPDATE 0x02
#define PKT_NOTIFICATION 0x03
#define PKT_KEEPALIVE 0x04
+#define PKT_SCHEDULE_CLOSE 0x1f /* Used internally to schedule socket close */
/* Attributes */
diff --git a/proto/bgp/packets.c b/proto/bgp/packets.c
index a5f52b8..b3b25f4 100644
--- a/proto/bgp/packets.c
+++ b/proto/bgp/packets.c
@@ -6,10 +6,151 @@
* Can be freely distributed and used under the terms of the GNU GPL.
*/
+#define LOCAL_DEBUG
+
#include "nest/bird.h"
#include "nest/iface.h"
#include "nest/protocol.h"
#include "nest/route.h"
#include "conf/conf.h"
+#include "lib/unaligned.h"
+#include "lib/socket.h"
#include "bgp.h"
+
+static byte *
+bgp_create_notification(struct bgp_conn *conn, byte *buf)
+{
+ DBG("BGP: Sending notification: code=%d, sub=%d, arg=%d:%d\n", conn->notify_code, conn->notify_subcode, conn->notify_arg, conn->notify_arg_size);
+ buf[0] = conn->notify_code;
+ 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");
+ }
+}
+
+static byte *
+bgp_create_open(struct bgp_conn *conn, byte *buf)
+{
+ DBG("BGP: Sending open\n");
+ buf[0] = BGP_VERSION;
+ put_u16(buf+1, conn->bgp->local_as);
+ put_u16(buf+3, conn->bgp->cf->hold_time);
+ put_u32(buf+5, conn->bgp->local_id);
+ buf[9] = 0; /* No optional parameters */
+ return buf+10;
+}
+
+static byte *
+bgp_create_update(struct bgp_conn *conn, byte *buf)
+{
+ DBG("BGP: Sending update\n");
+ bug("Don't know how to create updates");
+}
+
+static void
+bgp_create_header(byte *buf, unsigned int len, unsigned int type)
+{
+ memset(buf, 0xff, 16); /* Marker */
+ put_u16(buf+16, len);
+ buf[18] = type;
+}
+
+static void
+bgp_fire_tx(struct bgp_conn *conn)
+{
+ unsigned int s = conn->packets_to_send;
+ sock *sk = conn->sk;
+ byte *buf = sk->tbuf;
+ byte *pkt = buf + BGP_HEADER_LENGTH;
+ byte *end;
+ int type;
+
+ if (s & (1 << PKT_SCHEDULE_CLOSE))
+ {
+ conn->packets_to_send = 0;
+ bug("Scheduled close"); /* FIXME */
+ }
+ if (s & (1 << PKT_NOTIFICATION))
+ {
+ s = 1 << PKT_SCHEDULE_CLOSE;
+ type = PKT_NOTIFICATION;
+ end = bgp_create_notification(conn, pkt);
+ }
+ else if (s & (1 << PKT_KEEPALIVE))
+ {
+ s &= ~(1 << PKT_KEEPALIVE);
+ type = PKT_KEEPALIVE;
+ end = pkt; /* Keepalives carry no data */
+ DBG("BGP: Sending keepalive\n");
+ }
+ else if (s & (1 << PKT_OPEN))
+ {
+ s &= ~(1 << PKT_OPEN);
+ type = PKT_OPEN;
+ end = bgp_create_open(conn, pkt);
+ }
+ else if (s & (1 << PKT_UPDATE))
+ {
+ end = bgp_create_update(conn, pkt);
+ type = PKT_UPDATE;
+ if (!end)
+ {
+ conn->packets_to_send = 0;
+ return;
+ }
+ }
+ else
+ return;
+ conn->packets_to_send = s;
+ bgp_create_header(buf, end - buf, type);
+ sk_send(sk, end - buf);
+}
+
+void
+bgp_schedule_packet(struct bgp_conn *conn, int type)
+{
+ DBG("BGP: Scheduling packet type %d\n", type);
+ conn->packets_to_send |= 1 << type;
+ if (conn->sk->tpos != conn->sk->tbuf)
+ bgp_fire_tx(conn);
+}
+
+void
+bgp_tx(sock *sk)
+{
+ struct bgp_conn *conn = sk->data;
+
+ DBG("BGP: TX hook\n");
+ bgp_fire_tx(conn);
+}
+
+int
+bgp_rx(sock *sk, int size)
+{
+ struct bgp_conn *conn = sk->data;
+ byte *pkt_start = sk->rbuf;
+ byte *end = pkt_start + size;
+
+ DBG("BGP: RX hook: Got %d bytes\n", size);
+ while (end >= pkt_start + BGP_HEADER_LENGTH)
+ {
+ if (conn->error_flag)
+ return 1;
+ bug("Incoming packets not handled"); /* FIXME */
+ }
+ if (pkt_start != sk->rbuf)
+ {
+ memmove(sk->rbuf, pkt_start, end - pkt_start);
+ sk->rpos = sk->rbuf + (end - pkt_start);
+ }
+ return 0;
+}