server: add mrt bgp4mp support

You can enable the feature like:

[Global]
  [Global.GlobalConfig]
    As = 64512
    RouterId = "10.0.255.254"
  [Global.Mrt]
    FileName = "update.dump"

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

1 files changed, 71 insertions, 0 deletions

diff --git a/server/dumper.go b/server/dumper.go
new file mode 100644
index 00000000..46281ca6
--- /dev/null
+++ b/server/dumper.go
@@ -0,0 +1,71 @@
+// Copyright (C) 2015 Nippon Telegraph and Telephone Corporation.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//    http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+// implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package server
+
+import (
+	log "github.com/Sirupsen/logrus"
+	"github.com/osrg/gobgp/packet"
+	"os"
+	"time"
+)
+
+type dumper struct {
+	ch chan *broadcastBGPMsg
+}
+
+func (d *dumper) sendCh() chan *broadcastBGPMsg {
+	return d.ch
+}
+
+func newDumper(filename string) (*dumper, error) {
+	f, err := os.Create(filename)
+	if err != nil {
+		return nil, err
+	}
+
+	ch := make(chan *broadcastBGPMsg, 16)
+
+	go func() {
+		for {
+			m := <-ch
+			subtype := bgp.MESSAGE_AS4
+			mp := bgp.NewBGP4MPMessage(m.peerAS, m.localAS, 0, m.peerAddress.String(), m.localAddress.String(), m.fourBytesAs, m.message)
+			if m.fourBytesAs == false {
+				subtype = bgp.MESSAGE
+			}
+			bm, err := bgp.NewMRTMessage(uint32(time.Now().Unix()), bgp.BGP4MP, subtype, mp)
+			if err != nil {
+				log.WithFields(log.Fields{
+					"Topic": "mrt",
+					"Data":  m,
+				}).Warn(err)
+				continue
+			}
+			buf, err := bm.Serialize()
+			if err != nil {
+				log.WithFields(log.Fields{
+					"Topic": "mrt",
+					"Data":  m,
+				}).Warn(err)
+			} else {
+				f.Write(buf)
+			}
+		}
+	}()
+	return &dumper{
+		ch: ch,
+	}, nil
+}