sriovnet: Skip SRIOV enablement if it is already enabled

Skip enabling sriov if it is already enabled. Signed-off-by: Parav Pandit <parav@mellanox.com>
k8snetworkplumbingwg · Apr 7, 2018 · 2607506 · 2607506
1 parent 13dedb2
commit 2607506
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/sriovnet.go b/sriovnet.go
@@ -61,11 +61,19 @@ func EnableSriov(pfNetdevName string) error {
 		return err
 	}
 
-	if maxVfCount != 0 {
-		return setMaxVfCount(pfNetdevName, maxVfCount)
-	} else {
+	if maxVfCount == 0 {
 		return fmt.Errorf("sriov unsupported for device: ", pfNetdevName)
 	}
+
+	curVfCount, err2 := netdevGetEnabledVfCount(pfNetdevName)
+	if err2 != nil {
+		fmt.Println("Fail to read current vf count of PF %v", pfNetdevName)
+		return err
+	}
+	if curVfCount == 0 {
+		return setMaxVfCount(pfNetdevName, maxVfCount)
+	}
+	return nil
 }
 
 func DisableSriov(pfNetdevName string) error {

diff --git a/sriovnet_test.go b/sriovnet_test.go
@@ -1,7 +1,6 @@
 package sriovnet
 
 import (
-	"fmt"
 	"testing"
 )