internal/pkg/ovn/common.go

   1 package ovn
   2
   3 import (
   4         "encoding/json"
   5         "fmt"
   6         "math/big"
   7         "math/rand"
   8         "net"
   9         logf "sigs.k8s.io/controller-runtime/pkg/runtime/log"
  10         "strings"
  11         "time"
  12 )
  13
  14 var log = logf.Log.WithName("ovn")
  15
  16 func parseOvnNetworkObject(ovnnetwork string) ([]map[string]interface{}, error) {
  17         var ovnNet []map[string]interface{}
  18
  19         if ovnnetwork == "" {
  20                 return nil, fmt.Errorf("parseOvnNetworkObject:error")
  21         }
  22
  23         if err := json.Unmarshal([]byte(ovnnetwork), &ovnNet); err != nil {
  24                 return nil, fmt.Errorf("parseOvnNetworkObject: failed to load ovn network err: %v | ovn network: %v", err, ovnnetwork)
  25         }
  26
  27         return ovnNet, nil
  28 }
  29
  30 func setupDistributedRouter(name string) error {
  31
  32         // Create a single common distributed router for the cluster.
  33         stdout, stderr, err := RunOVNNbctl("--", "--may-exist", "lr-add", name, "--", "set", "logical_router", name, "external_ids:ovn4nfv-cluster-router=yes")
  34         if err != nil {
  35                 log.Error(err, "Failed to create a single common distributed router for the cluster", "stdout", stdout, "stderr", stderr)
  36                 return err
  37         }
  38         // Create a logical switch called "ovn4nfv-join" that will be used to connect gateway routers to the distributed router.
  39         // The "ovn4nfv-join" will be allocated IP addresses in the range 100.64.1.0/24.
  40         stdout, stderr, err = RunOVNNbctl("--may-exist", "ls-add", "ovn4nfv-join")
  41         if err != nil {
  42                 log.Error(err, "Failed to create logical switch called \"ovn4nfv-join\"", "stdout", stdout, "stderr", stderr)
  43                 return err
  44         }
  45         // Connect the distributed router to "ovn4nfv-join".
  46         routerMac, stderr, err := RunOVNNbctl("--if-exist", "get", "logical_router_port", "rtoj-"+name, "mac")
  47         if err != nil {
  48                 log.Error(err, "Failed to get logical router port rtoj-", "name", name, "stdout", stdout, "stderr", stderr)
  49                 return err
  50         }
  51         if routerMac == "" {
  52                 routerMac = generateMac()
  53                 stdout, stderr, err = RunOVNNbctl("--", "--may-exist", "lrp-add", name, "rtoj-"+name, routerMac, "100.64.1.1/24", "--", "set", "logical_router_port", "rtoj-"+name, "external_ids:connect_to_ovn4nfvjoin=yes")
  54                 if err != nil {
  55                         log.Error(err, "Failed to add logical router port rtoj", "name", name, "stdout", stdout, "stderr", stderr)
  56                         return err
  57                 }
  58         }
  59         // Connect the switch "ovn4nfv-join" to the router.
  60         stdout, stderr, err = RunOVNNbctl("--", "--may-exist", "lsp-add", "ovn4nfv-join", "jtor-"+name, "--", "set", "logical_switch_port", "jtor-"+name, "type=router", "options:router-port=rtoj-"+name, "addresses="+"\""+routerMac+"\"")
  61         if err != nil {
  62                 log.Error(err, "Failed to add logical switch port to logical router", "stdout", stdout, "stderr", stderr)
  63                 return err
  64         }
  65         return nil
  66 }
  67
  68 // Find if switch exists
  69 func findLogicalSwitch(name string) bool {
  70         // get logical switch from OVN
  71         output, stderr, err := RunOVNNbctl("--data=bare", "--no-heading",
  72                 "--columns=name", "find", "logical_switch", "name="+name)
  73         if err != nil {
  74                 log.Error(err, "Error in obtaining list of logical switch", "stderr", stderr)
  75                 return false
  76         }
  77
  78         if strings.Compare(name, output) == 0 {
  79                 return true
  80         }
  81         return false
  82 }
  83
  84 // generateMac generates mac address.
  85 func generateMac() string {
  86         prefix := "00:00:00"
  87         newRand := rand.New(rand.NewSource(time.Now().UnixNano()))
  88         mac := fmt.Sprintf("%s:%02x:%02x:%02x", prefix, newRand.Intn(255), newRand.Intn(255), newRand.Intn(255))
  89         return mac
  90 }
  91
  92 // NextIP returns IP incremented by 1
  93 func NextIP(ip net.IP) net.IP {
  94         i := ipToInt(ip)
  95         return intToIP(i.Add(i, big.NewInt(1)))
  96 }
  97
  98 func ipToInt(ip net.IP) *big.Int {
  99         if v := ip.To4(); v != nil {
 100                 return big.NewInt(0).SetBytes(v)
 101         }
 102         return big.NewInt(0).SetBytes(ip.To16())
 103 }
 104
 105 func intToIP(i *big.Int) net.IP {
 106         return net.IP(i.Bytes())
 107 }