working on vpn rpc

Signed-off-by: Serguei Bezverkhi <[email protected]>

Author: sbezverk

cmd/gateway-client/jalapeno-client.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"flag"
 	"fmt"
 	"io"
 	"math/rand"
@@ -23,6 +24,9 @@ const (
 )
 
 func main() {
+	flag.Parse()
+	flag.Set("logtostderr", "true")
+
 	client := os.Getenv("CLIENT_IP")
 	if client == "" {
 		glog.Errorf("env variable \"CLIENT_IP\" is not defined, cannor proceed further, exiting...")
@@ -37,7 +41,7 @@ func main() {
 		Addr: &net.IPAddr{
 			IP: net.ParseIP(client),
 		}})
-	conn, err := grpc.DialContext(ctx, jalapenoGateway)
+	conn, err := grpc.DialContext(ctx, jalapenoGateway, grpc.WithInsecure())
 	if err != nil {
 		glog.Errorf("failed to connect to Jalapeno Gateway at the address: %s with error: %+v", jalapenoGateway, err)
 		os.Exit(1)
@@ -72,17 +76,19 @@ func main() {
 		i := rand.Intn(3)
 		stream, err := gwclient.VPN(ctx, requests[i])
 		if err != nil {
-			glog.Errorf("failed to request VPN label for request %+v with error: %+v", requests[i])
-		}
-		for {
-			entry, err := stream.Recv()
-			if err == io.EOF {
-				break
-			}
-			if err != nil {
-				glog.Errorf("failed to receive a message from the stream with error: %+v", err)
+			glog.Errorf("failed to request VPN label for request %+v with error: %+v", requests[i], err)
+		} else {
+			for {
+				entry, err := stream.Recv()
+				if err == io.EOF {
+					break
+				}
+				if err != nil {
+					glog.Errorf("failed to receive a message from the stream with error: %+v", err)
+					break
+				}
+				fmt.Printf("Received message: %+v\n", *entry)
 			}
-			fmt.Printf("Received message: %+v\n", *entry)
 		}
 		select {
 		case <-ticker.C:

deployment/jalapeno-client.yaml

@@ -15,7 +15,7 @@ spec:
     spec:
       containers:
         - name: jalapeno-client
-          image: 192.168.80.240:4000/jalapeno-client/jalapeno-client:0.0.0
+          image: docker.io/sbezverk/jalapeno-client:0.0.0
           imagePullPolicy: Always
           env:
           - name: CLIENT_IP

deployment/jalapeno-gateway.yaml

@@ -39,7 +39,7 @@ spec:
     spec:
       containers:
         - name: jalapeno-gateway
-          image: 192.168.80.240:4000/jalapeno-gateway/jalapeno-gateway-debug:0.0.0
+          image: docker.io/sbezverk/jalapeno-gateway-debug:0.0.0
           imagePullPolicy: Always
           ports:
             - containerPort: 40040

pkg/dbclient/mock/dbmock.go

@@ -19,7 +19,6 @@ var (
 type dbMock struct {
 	mu  sync.Mutex
 	qoe map[int32]*pbapi.Qoe
-	vpn map[pbapi.RouteDistinguisherType]map[uint32][]uint32
 }
 
 // GetQoE is required method by DB interface, it takes requested QoE and searches through

pkg/gateway/server.go

@@ -2,13 +2,15 @@ package gateway
 
 import (
 	"context"
+	"fmt"
 	"net"
 	"time"
 
 	pbapi "github.com/cisco-ie/jalapeno-go-gateway/pkg/apis"
 	"github.com/cisco-ie/jalapeno-go-gateway/pkg/bgpclient"
 	"github.com/cisco-ie/jalapeno-go-gateway/pkg/dbclient"
 	"github.com/golang/glog"
+	"github.com/golang/protobuf/ptypes"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/peer"
 )
@@ -42,6 +44,36 @@ func (g *gateway) Stop() {
 }
 
 func (g *gateway) VPN(reqVPN *pbapi.RequestVPN, stream pbapi.GatewayService_VPNServer) error {
+	var rdValue ptypes.DynamicAny
+	if err := ptypes.UnmarshalAny(reqVPN.Rd, &rdValue); err != nil {
+		return fmt.Errorf("failed to unmarshal route distinguisher with error: %+v", err)
+	}
+	rd := "RD: "
+	switch v := rdValue.Message.(type) {
+	case *pbapi.RouteDistinguisherTwoOctetAS:
+		rd = fmt.Sprintf("%d:%d ", uint16(v.Admin), v.Assigned)
+	case *pbapi.RouteDistinguisherIPAddress:
+		rd = fmt.Sprintf("%s:%d ", v.Admin, uint16(v.Assigned))
+	case *pbapi.RouteDistinguisherFourOctetAS:
+		rd = fmt.Sprintf("%d:%d ", v.Admin, uint16(v.Assigned))
+	}
+	var rt string
+	for i := 0; i < len(reqVPN.Rt); i++ {
+		var rtValue ptypes.DynamicAny
+		if err := ptypes.UnmarshalAny(reqVPN.Rt[i], &rtValue); err != nil {
+			return fmt.Errorf("failed to unmarshal route target with error: %+v", err)
+		}
+		rt += "RT: "
+		switch v := rtValue.Message.(type) {
+		case *pbapi.TwoOctetAsSpecificExtended:
+			rt += fmt.Sprintf("Subtype:%d 2 bytes AS:%d LocalAdmin:%d Transitive:%t", v.SubType, uint16(v.As), v.LocalAdmin, v.IsTransitive)
+		case *pbapi.IPv4AddressSpecificExtended:
+			rt += fmt.Sprintf("Subtype:%d Address:%d LocalAdmin:%d Transitive:%t", v.SubType, v.Address, uint16(v.LocalAdmin), v.IsTransitive)
+		case *pbapi.FourOctetAsSpecificExtended:
+			rt += fmt.Sprintf("Subtype:%d 4 bytes AS:%d LocalAdmin:%d Transitive:%t", v.SubType, v.As, uint16(v.LocalAdmin), v.IsTransitive)
+		}
+	}
+	glog.Infof("Request for VPN with %s %s", rd, rt)
 	if err := stream.Send(&pbapi.ResponseVPNEntry{}); err != nil {
 		return err
 	}