Newer
Older
"github.com/bio-routing/bio-rd/config"
"github.com/bio-routing/bio-rd/protocols/bgp/packet"
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
log "github.com/sirupsen/logrus"
)
const (
uint16max = 65535
BGPVersion = 4
)
type BGPServer struct {
listeners []*TCPListener
acceptCh chan *net.TCPConn
peers map[string]*Peer
routerID uint32
}
func NewBgpServer() *BGPServer {
return &BGPServer{
peers: make(map[string]*Peer),
}
}
func (b *BGPServer) RouterID() uint32 {
return b.routerID
}
func (b *BGPServer) Start(c *config.Global) error {
if err := c.SetDefaultGlobalConfigValues(); err != nil {
return fmt.Errorf("Failed to load defaults: %v", err)
}
fmt.Printf("ROUTER ID: %d\n", c.RouterID)
b.routerID = c.RouterID
if c.Listen {
acceptCh := make(chan *net.TCPConn, 4096)
for _, addr := range c.LocalAddressList {
l, err := NewTCPListener(addr, c.Port, acceptCh)
if err != nil {
return fmt.Errorf("Failed to start TCPListener for %s: %v", addr.String(), err)
}
b.listeners = append(b.listeners, l)
}
b.acceptCh = acceptCh
go b.incomingConnectionWorker()
}
return nil
}
func (b *BGPServer) incomingConnectionWorker() {
for {
c := <-b.acceptCh
fmt.Printf("Incoming connection!\n")
fmt.Printf("Connection from: %v\n", c.RemoteAddr())
peerAddr := strings.Split(c.RemoteAddr().String(), ":")[0]
if _, ok := b.peers[peerAddr]; !ok {
c.Close()
log.WithFields(log.Fields{
"source": c.RemoteAddr(),
}).Warning("TCP connection from unknown source")
continue
}
log.WithFields(log.Fields{
"source": c.RemoteAddr(),
}).Info("Incoming TCP connection")
fmt.Printf("DEBUG: Sending incoming TCP connection to fsm for peer %s\n", peerAddr)
b.peers[peerAddr].fsm.conCh <- c
fmt.Printf("DEBUG: Sending done\n")
}
}
func (b *BGPServer) AddPeer(c config.Peer, rib *locRIB.LocRIB) error {
if c.LocalAS > uint16max || c.PeerAS > uint16max {
return fmt.Errorf("32bit ASNs are not supported yet")
}
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
if err != nil {
return err
}
peer.routerID = c.RouterID
peerAddr := peer.GetAddr().String()
b.peers[peerAddr] = peer
b.peers[peerAddr].Start()
return nil
}
func recvMsg(c *net.TCPConn) (msg []byte, err error) {
buffer := make([]byte, packet.MaxLen)
_, err = io.ReadFull(c, buffer[0:packet.MinLen])
if err != nil {
return nil, fmt.Errorf("Read failed: %v", err)
}
l := int(buffer[16])*256 + int(buffer[17])
toRead := l
_, err = io.ReadFull(c, buffer[packet.MinLen:toRead])
if err != nil {
return nil, fmt.Errorf("Read failed: %v", err)
}
return buffer, nil
}