add sg acl check when init

kubeovn · Dec 6, 2021 · e1dfa7b · e1dfa7b
1 parent c8692df
commit e1dfa7b
Showing 1 changed file with 29 additions and 4 deletions.
diff --git a/pkg/ovs/ovn-nbctl.go b/pkg/ovs/ovn-nbctl.go
@@ -1559,14 +1559,26 @@ func (c Client) createSgRuleACL(sgName string, direction AclDirection, rule *kub
 
 func (c Client) CreateSgDenyAllACL() error {
 	portGroupName := GetSgPortGroupName(util.DenyAllSecurityGroup)
-	if _, err := c.ovnNbCommand(MayExist, "--type=port-group", "acl-add", portGroupName, string(SgAclIngressDirection), util.SecurityGroupDropPriority,
-		fmt.Sprintf("outport==@%s && ip", portGroupName), "drop"); err != nil {
+	exist, err := c.AclExists(util.SecurityGroupDropPriority, string(SgAclIngressDirection))
+	if err != nil {
 		return err
 	}
-	if _, err := c.ovnNbCommand(MayExist, "--type=port-group", "acl-add", portGroupName, string(SgAclEgressDirection), util.SecurityGroupDropPriority,
-		fmt.Sprintf("inport==@%s && ip", portGroupName), "drop"); err != nil {
+	if !exist {
+		if _, err := c.ovnNbCommand(MayExist, "--type=port-group", "acl-add", portGroupName, string(SgAclIngressDirection), util.SecurityGroupDropPriority,
+			fmt.Sprintf("outport==@%s && ip", portGroupName), "drop"); err != nil {
+			return err
+		}
+	}
+	exist, err = c.AclExists(util.SecurityGroupDropPriority, string(SgAclEgressDirection))
+	if err != nil {
 		return err
 	}
+	if !exist {
+		if _, err := c.ovnNbCommand(MayExist, "--type=port-group", "acl-add", portGroupName, string(SgAclEgressDirection), util.SecurityGroupDropPriority,
+			fmt.Sprintf("inport==@%s && ip", portGroupName), "drop"); err != nil {
+			return err
+		}
+	}
 	return nil
 }
 
@@ -1644,3 +1656,16 @@ func (c Client) SetLspExternalIds(cmd []string) error {
 	}
 	return nil
 }
+
+func (c *Client) AclExists(priority, direction string) (bool, error) {
+	priorityVal, _ := strconv.Atoi(priority)
+	results, err := c.CustomFindEntity("acl", []string{"match"}, fmt.Sprintf("priority=%d", priorityVal), fmt.Sprintf("direction=%s", direction))
+	if err != nil {
+		klog.Errorf("customFindEntity failed, %v", err)
+		return false, err
+	}
+	if len(results) == 0 {
+		return false, nil
+	}
+	return true, nil
+}