From a01b549fffd428f075b8732fde244c47baf28ecd Mon Sep 17 00:00:00 2001
From: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>
Date: Wed, 5 Aug 2015 17:16:31 +0900
Subject: *: kill protobuf path structure

Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>
---
 server/server.go | 38 ++++++++++++++++++++++++--------------
 1 file changed, 24 insertions(+), 14 deletions(-)

(limited to 'server/server.go')

diff --git a/server/server.go b/server/server.go
index 56ef8c68..cb301fef 100644
--- a/server/server.go
+++ b/server/server.go
@@ -507,7 +507,10 @@ func (server *BgpServer) broadcastBests(bests []*table.Path) {
 		}
 
 		result := &GrpcResponse{
-			Data: path.ToApiStruct(),
+			Data: &api.Destination{
+				Prefix: path.GetNlri().String(),
+				Paths:  []*api.Path{path.ToApiStruct()},
+			},
 		}
 		remainReqs := make([]*GrpcRequest, 0, len(server.broadcastReqs))
 		for _, req := range server.broadcastReqs {
@@ -543,7 +546,7 @@ func (server *BgpServer) broadcastPeerState(peer *Peer) {
 		default:
 		}
 		ignore := req.RequestType != REQ_MONITOR_NEIGHBOR_PEER_STATE
-		ignore = ignore || (req.RemoteAddr != "" && req.RemoteAddr != peer.conf.NeighborConfig.NeighborAddress.String())
+		ignore = ignore || (req.Name != "" && req.Name != peer.conf.NeighborConfig.NeighborAddress.String())
 		if ignore {
 			remainReqs = append(remainReqs, req)
 			continue
@@ -748,7 +751,7 @@ func (server *BgpServer) handlePolicy(pl config.RoutingPolicy) {
 }
 
 func (server *BgpServer) checkNeighborRequest(grpcReq *GrpcRequest) (*Peer, error) {
-	remoteAddr := grpcReq.RemoteAddr
+	remoteAddr := grpcReq.Name
 	peer, found := server.neighborMap[remoteAddr]
 	if !found {
 		result := &GrpcResponse{}
@@ -1022,11 +1025,11 @@ func (server *BgpServer) handleVrfRequest(req *GrpcRequest) []*table.Path {
 
 	switch req.RequestType {
 	case REQ_VRF:
-		arg := req.Data.(*api.Arguments)
+		name := req.Name
 		rib := server.localRibMap[GLOBAL_RIB_NAME].rib
 		vrfs := rib.Vrfs
-		if _, ok := vrfs[arg.Name]; !ok {
-			result.ResponseErr = fmt.Errorf("vrf %s not found", arg.Name)
+		if _, ok := vrfs[name]; !ok {
+			result.ResponseErr = fmt.Errorf("vrf %s not found", name)
 			break
 		}
 		var rf bgp.RouteFamily
@@ -1042,12 +1045,15 @@ func (server *BgpServer) handleVrfRequest(req *GrpcRequest) []*table.Path {
 			break
 		}
 		for _, path := range rib.GetPathList(rf) {
-			ok := policy.CanImportToVrf(vrfs[arg.Name], path)
+			ok := policy.CanImportToVrf(vrfs[name], path)
 			if !ok {
 				continue
 			}
 			req.ResponseCh <- &GrpcResponse{
-				Data: path.ToApiStruct(),
+				Data: &api.Destination{
+					Prefix: path.GetNlri().String(),
+					Paths:  []*api.Path{path.ToApiStruct()},
+				},
 			}
 		}
 		goto END
@@ -1124,7 +1130,7 @@ func (server *BgpServer) handleGrpc(grpcReq *GrpcRequest) []*SenderMsg {
 			break
 		}
 		if peer.fsm.adminState != ADMIN_STATE_DOWN {
-			remoteAddr := grpcReq.RemoteAddr
+			remoteAddr := grpcReq.Name
 			if t, ok := server.localRibMap[remoteAddr].rib.Tables[grpcReq.RouteFamily]; ok {
 				for _, dst := range t.GetDestinations() {
 					result := &GrpcResponse{}
@@ -1152,8 +1158,12 @@ func (server *BgpServer) handleGrpc(grpcReq *GrpcRequest) []*SenderMsg {
 		}
 
 		for _, p := range paths {
-			result := &GrpcResponse{}
-			result.Data = p.ToApiStruct()
+			result := &GrpcResponse{
+				Data: &api.Destination{
+					Prefix: p.GetNlri().String(),
+					Paths:  []*api.Path{p.ToApiStruct()},
+				},
+			}
 			grpcReq.ResponseCh <- result
 		}
 		close(grpcReq.ResponseCh)
@@ -2028,15 +2038,15 @@ func (server *BgpServer) handleMrt(grpcReq *GrpcRequest) {
 		if err != nil {
 			return
 		}
-		loc, ok := server.localRibMap[grpcReq.RemoteAddr]
+		loc, ok := server.localRibMap[grpcReq.Name]
 		if !ok {
-			result.ResponseErr = fmt.Errorf("no local rib for %s", grpcReq.RemoteAddr)
+			result.ResponseErr = fmt.Errorf("no local rib for %s", grpcReq.Name)
 			grpcReq.ResponseCh <- result
 			close(grpcReq.ResponseCh)
 			return
 		}
 		manager = loc.rib
-		view = grpcReq.RemoteAddr
+		view = grpcReq.Name
 	}
 
 	msg, err := server.mkMrtPeerIndexTableMsg(now, view)
-- 
cgit v1.2.3