feat: parse ros ospf lsa network

Author: BaiMeow

backend/graph/fetch/ros/ros.go

@@ -1,19 +1,14 @@
 package ros
 
 import (
-	"bytes"
-	"encoding/gob"
 	"fmt"
 	"github.com/BaiMeow/NetworkMonitor/graph/fetch"
 	"github.com/go-routeros/routeros"
-	"github.com/go-routeros/routeros/proto"
 	"net"
 	"time"
 )
 
 func init() {
-	gob.Register(&proto.Sentence{})
-
 	fetch.Register("ros", func(config map[string]any) (fetch.Fetcher, error) {
 		addr, ok := config["address"].(string)
 		if !ok {
@@ -57,8 +52,5 @@ func (R *ROS) GetData() (any, error) {
 	if err != nil {
 		return nil, err
 	}
-
-	buf := &bytes.Buffer{}
-	gob.NewEncoder(buf).Encode(reply.Re)
-	return buf.Bytes(), nil
+	return reply.Re, nil
 }

backend/graph/fetch/ros/ros_test.go

@@ -1,7 +1,10 @@
 package ros
 
 import (
+	"bytes"
 	"encoding/base64"
+	"encoding/gob"
+	"github.com/go-routeros/routeros/proto"
 	"testing"
 )
 
@@ -15,7 +18,11 @@ func TestROS_GetData(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	t.Log(base64.StdEncoding.EncodeToString(resp.([]byte)))
+	sentences := resp.([]*proto.Sentence)
+	gob.Register(sentences)
+	var buf bytes.Buffer
+	gob.NewEncoder(&buf).Encode(sentences)
+	t.Log(base64.StdEncoding.EncodeToString(buf.Bytes()))
 
 	ros7 := &ROS{
 		Address:  "10.28.0.1:8728",
@@ -26,5 +33,8 @@ func TestROS_GetData(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	t.Log(base64.StdEncoding.EncodeToString(resp.([]byte)))
+	sentences = resp.([]*proto.Sentence)
+	gob.Register(sentences)
+	gob.NewEncoder(&buf).Encode(sentences)
+	t.Log(base64.StdEncoding.EncodeToString(buf.Bytes()))
 }

backend/graph/parse/rosospf/ros_ospf.go

@@ -1,7 +1,6 @@
 package rosospf
 
 import (
-	"bytes"
 	"encoding/gob"
 	"github.com/BaiMeow/NetworkMonitor/graph/entity"
 	"log"
@@ -22,8 +21,9 @@ func init() {
 }
 
 var (
-	ros6BodyPtpReg = regexp.MustCompile(`Point-To-Point ((?:[0-9]{1,3}\.){3}[0-9]{1,3}) (?:[0-9]{1,3}\.){3}[0-9]{1,3} (\d+)`)
-	ros7BodyPtpReg = regexp.MustCompile(`type=p2p id=((?:[0-9]{1,3}\.){3}[0-9]{1,3}) data=(?:[0-9]{1,3}\.){3}[0-9]{1,3} metric=(\d+)`)
+	ros6BodyPtpReg     = regexp.MustCompile(`Point-To-Point ((?:[0-9]{1,3}\.){3}[0-9]{1,3}) (?:[0-9]{1,3}\.){3}[0-9]{1,3} (\d+)`)
+	ros7BodyPtpReg     = regexp.MustCompile(`type=p2p id=((?:[0-9]{1,3}\.){3}[0-9]{1,3}) data=(?:[0-9]{1,3}\.){3}[0-9]{1,3} metric=(\d+)`)
+	ros7BodyNetworkReg = regexp.MustCompile(`type=network id=((?:[0-9]{1,3}\.){3}[0-9]{1,3}) data=(?:[0-9]{1,3}\.){3}[0-9]{1,3} metric=(\d+)`)
 )
 
 var _ parse.Parser[*entity.OSPF] = (*RosOSPF)(nil)
@@ -32,17 +32,21 @@ type RosOSPF struct {
 	parse.Base[*entity.OSPF]
 }
 
+type network struct {
+	cost    int
+	area    string
+	routers []string
+}
+
 func (p *RosOSPF) Parse(input any) (*entity.OSPF, error) {
-	raw, ok := input.([]byte)
+	sentences, ok := input.([]*proto.Sentence)
 	if !ok {
 		log.Fatalf("invalid input data type: %s\n", reflect.TypeOf(input).Elem())
 	}
 
-	var graph entity.OSPF
-
-	var sentences []*proto.Sentence
-	gob.NewDecoder(bytes.NewReader(raw)).Decode(&sentences) // 这里本来应该在初始化就直接处理了，但是因为Init没有抛异常，所以这一步在这里做
+	networks := make(map[string]*network)
 
+	var graph entity.OSPF
 	for _, sentence := range sentences {
 		if sentence.Word == "!done" { // 这个判断可有可无 因为fetcher已经做了处理
 			break
@@ -74,6 +78,36 @@ func (p *RosOSPF) Parse(input any) (*entity.OSPF, error) {
 			}
 			graph.AddLink(sentence.Map["area"], sentence.Map["id"], field[1], cost)
 		}
+		nw := ros7BodyNetworkReg.FindAllStringSubmatch(sentence.Map["body"], -1)
+		for _, field := range nw {
+			if len(field) != 3 {
+				continue
+			}
+			cost, err := strconv.Atoi(field[2])
+			if err != nil {
+				continue
+			}
+			nw := networks[field[1]]
+			if nw == nil {
+				nw = &network{
+					cost: cost,
+					area: sentence.Map["area"],
+				}
+				networks[field[1]] = nw
+			}
+			nw.routers = append(nw.routers, sentence.Map["id"])
+		}
 	}
+
+	for _, network := range networks {
+		area := graph.GetArea(network.area)
+		for i := 0; i < len(network.routers); i++ {
+			for j := i + 1; j < len(network.routers); j++ {
+				area.AddLink(network.routers[i], network.routers[j], network.cost)
+				area.AddLink(network.routers[j], network.routers[i], network.cost)
+			}
+		}
+	}
+
 	return &graph, nil
 }

backend/graph/parse/rosospf/ros_ospf_test.go

@@ -1,9 +1,12 @@
 package rosospf
 
 import (
+	"bytes"
 	"encoding/base64"
+	"encoding/gob"
 	"encoding/json"
 	"github.com/BaiMeow/NetworkMonitor/graph/entity"
+	"github.com/go-routeros/routeros/proto"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -30,8 +33,10 @@ func TestROSOSPF(t *testing.T) {
 	for _, v := range testcases {
 		t.Run(v.name, func(t *testing.T) {
 			gobOutput, err := base64.StdEncoding.DecodeString(v.feterOutput)
+			var sentences []*proto.Sentence
+			gob.NewDecoder(bytes.NewReader(gobOutput)).Decode(&sentences)
 			p := RosOSPF{}
-			res, err := p.Parse(gobOutput)
+			res, err := p.Parse(sentences)
 			assert.NoError(t, err)
 			o := &entity.OSPF{}
 			_ = json.Unmarshal([]byte(v.ospfData), o)