rest: use []byte instead of interface{} for REST response

Some Marshall method is not thread-safe. They can't be called by
a goroutine for a REST client.

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

2 files changed, 7 insertions, 3 deletions

diff --git a/server/peer.go b/server/peer.go
index 1d7ec4a8..b3261dbd 100644
--- a/server/peer.go
+++ b/server/peer.go
@@ -106,7 +106,8 @@ func (peer *Peer) path2update(pathList []table.Path) []*bgp.BGPMessage {
 
 func (peer *Peer) handleREST(restReq *api.RestRequest) {
 	result := &api.RestResponse{}
-	result.Data = peer.rib.Tables[bgp.RF_IPv4_UC]
+	j, _ := json.Marshal(peer.rib.Tables[bgp.RF_IPv4_UC])
+	result.Data = j
 	restReq.ResponseCh <- result
 	close(restReq.ResponseCh)
 }
diff --git a/server/server.go b/server/server.go
index ac1e4f81..e97d31c4 100644
--- a/server/server.go
+++ b/server/server.go
@@ -16,6 +16,7 @@
 package server
 
 import (
+	"encoding/json"
 	"fmt"
 	log "github.com/Sirupsen/logrus"
 	"github.com/osrg/gobgp/api"
@@ -157,7 +158,8 @@ func (server *BgpServer) handleRest(restReq *api.RestRequest) {
 		for _, peer := range server.peerMap {
 			peerList = append(peerList, peer)
 		}
-		result.Data = peerList
+		j, _ := json.Marshal(peerList)
+		result.Data = j
 		restReq.ResponseCh <- result
 		close(restReq.ResponseCh)
 
@@ -167,7 +169,8 @@ func (server *BgpServer) handleRest(restReq *api.RestRequest) {
 		result := &api.RestResponse{}
 		peer, found := server.peerMap[remoteAddr]
 		if found {
-			result.Data = peer
+			j, _ := json.Marshal(peer)
+			result.Data = j
 		} else {
 			result.ResponseErr = fmt.Errorf("Neighbor that has %v does not exist.", remoteAddr)
 		}