add checks against router and bridge options

Author: msagheer

plugin/driver/driver.go

@@ -138,7 +138,12 @@ func (driver *driver) createNetwork(w http.ResponseWriter, r *http.Request) {
 	neName := create.Options[netlabel.GenericData].(map[string]interface{})["bridge"]
 
 	if neName != nil {
-		AddNetworkInfo(create.NetworkID, neName.(string), domainid.(string))
+		if err := AddNetworkInfo(create.NetworkID, neName.(string), domainid.(string)); err != nil {
+			Log.Error(err)
+			DomainDelete(domainid.(string))
+			errorResponse(w, fmt.Sprintf("Bridge (%s) doesnot exist in doamin (%s).", neName.(string), domainid.(string)))
+			return
+		}
 		AddGatewayInfo(create.NetworkID, domainid.(string), gatewayip)
 		emptyResponse(w)
 
@@ -154,7 +159,13 @@ func (driver *driver) createNetwork(w http.ResponseWriter, r *http.Request) {
 		tm, _ := hex.DecodeString(ipnet.Mask.String())
 		netmask := fmt.Sprintf("%v.%v.%v.%v", tm[0], tm[1], tm[2], tm[3])
 		Log.Infof("Adding router interface for : ", router, gatewayip, netmask)
-		CreateNetworkLink(router.(string), domainid.(string), create.NetworkID, gatewayip, netmask)
+		if err := CreateNetworkLink(router.(string), domainid.(string), create.NetworkID, gatewayip, netmask); err != nil {
+			Log.Error(err)
+			BridgeDelete(create.NetworkID, domainid.(string))
+			DomainDelete(domainid.(string))
+			errorResponse(w, fmt.Sprintf("Router (%s) doesnot exist in doamin (%s).", router.(string), domainid.(string)))
+			return
+		}
 	}
 
 	emptyResponse(w)
@@ -276,6 +287,7 @@ func (driver *driver) joinEndpoint(w http.ResponseWriter, r *http.Request) {
 	if err := ifc_ctl.PgIfUp(if_local_name, "cont_"+endID[:8], domainid, bridgeID, mac, "gateway"); err != nil {
 		Log.Error(err)
 		errorResponse(w, "Unable to on-board container onto PLUMgrid")
+		return
 	}
 
 	if netlink.LinkSetUp(local) != nil {
@@ -316,6 +328,7 @@ func (driver *driver) leaveEndpoint(w http.ResponseWriter, r *http.Request) {
 	if err := ifc_ctl.PgIfDown(if_local_name, "gateway"); err != nil {
 		Log.Error(err)
 		errorResponse(w, "Unable to off-board container from PLUMgrid")
+		return
 	}
 
 	RemoveMetaconfig(domainid, bridgeID, l.EndpointID)

plugin/driver/pg_helper.go

@@ -16,6 +16,7 @@ package driver
 
 import (
 	"encoding/json"
+	"fmt"
 	"strings"
 )
 
@@ -77,14 +78,19 @@ func BridgeDelete(ID string, domainid string) {
 	RestCall("DELETE", url, nil)
 }
 
-func AddNetworkInfo(dNetwork string, neName string, domain string) {
+func AddNetworkInfo(dNetwork string, neName string, domain string) error {
+
 	id := GetNeId(neName, domain)
 	var neId string
 	if id == "" {
 		neId = neName
 	} else {
 		neId = id
 	}
+	if !CheckNeExists(neId, domain) {
+		return fmt.Errorf("Bridge (%s) doesnot exist in doamin (%s).", neName, domain)
+	}
+
 	url := "/0/connectivity/domain/" + domain + "/ne/" + neId
 	body, _ := RestCall("GET", url+"?configonly=true", nil)
 	var netData map[string]interface{}
@@ -95,6 +101,8 @@ func AddNetworkInfo(dNetwork string, neName string, domain string) {
 	netData["metadata"] = dNetwork
 	finalData, _ := json.Marshal(netData)
 	RestCall("PUT", url, finalData)
+
+	return nil
 }
 
 func FindDomainFromNetwork(ID string) (domainid string, netid string) {
@@ -238,9 +246,13 @@ func GetNeId(NeName string, DomainID string) (NeID string) {
 	return ""
 }
 
-func CreateNetworkLink(NeName string, DomainID string, NetworkID string, IP string, Netmask string) {
+func CreateNetworkLink(NeName string, DomainID string, NetworkID string, IP string, Netmask string) error {
 
 	ne_ID := GetNeId(NeName, DomainID)
+	if !CheckNeExists(ne_ID, DomainID) {
+		return fmt.Errorf("Router (%s) doesnot exist in doamin (%s).", NeName, DomainID)
+	}
+
 	CheckNeChildList(ne_ID, DomainID, "ifc")
 	CheckNeChildList(NetworkName(NetworkID), DomainID, "ifc")
 	ne_ifc := NetworkID
@@ -271,6 +283,8 @@ func CreateNetworkLink(NeName string, DomainID string, NetworkID string, IP stri
                          "attachment1": "/ne/` + ne_ID + `/ifc/` + ne_ifc + `",
                          "attachment2": "/ne/` + NetworkName(NetworkID) + `/ifc/` + net_ifc + `"}`)
 	RestCall("PUT", url, data)
+
+	return nil
 }
 
 func DeleteNetworkLinks(DomainID string, NetworkID string) {
@@ -409,3 +423,23 @@ func DeleteAttachedLB(DomainID string, NetworkID string) {
 		}
 	}
 }
+
+func CheckNeExists(ne_name, domain string) bool {
+
+	url := "/0/connectivity/domain/" + domain + "/ne?configonly=true"
+
+	body, _ := RestCall("GET", url+"?configonly=true", nil)
+	var domain_data map[string]interface{}
+	err := json.Unmarshal([]byte(body), &domain_data)
+	if err != nil {
+		panic(err)
+	}
+
+	for ne, _ := range domain_data {
+		if ne_name == ne {
+			return true
+		}
+	}
+
+	return false
+}