From 6615dfcda3514d440c7450b29d9af94f0007139f Mon Sep 17 00:00:00 2001
From: huangyunpeng <huangyp@yusur.tech>
Date: Thu, 30 Nov 2023 14:01:05 +0800
Subject: [PATCH] Fix can't get pfname for virtio pci path

Fixed issue with virtio driver by adding a conditional check for the virtio path under the PCI address,
preventing errors when the pf name cannot be found. This ensures successful acquisition of vf resources.

Fixes #507

Signed-off-by: JasinlikeEatingOrange <jasin_smith@outlook.com>
---
 pkg/utils/utils.go      | 11 +++++++++++
 pkg/utils/utils_test.go | 14 +++++++++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
index 8ee7f78ab..8c6513575 100644
--- a/pkg/utils/utils.go
+++ b/pkg/utils/utils.go
@@ -92,6 +92,17 @@ func GetPfName(pciAddr string) (string, error) {
 	}
 
 	path := filepath.Join(sysBusPci, pciAddr, "physfn", "net")
+	virtIOpath := filepath.Join(sysBusPci, pciAddr, "physfn")
+	dir, err := os.ReadDir(virtIOpath)
+	if err != nil {
+		return "", fmt.Errorf("can't open path %s: %w", virtIOpath, err)
+	}
+	for _, file := range dir {
+		if strings.Contains(file.Name(), "virtio") {
+			path = filepath.Join(virtIOpath, file.Name(), "net")
+			break
+		}
+	}
 	files, err := os.ReadDir(path)
 	if err != nil {
 		if os.IsNotExist(err) {
diff --git a/pkg/utils/utils_test.go b/pkg/utils/utils_test.go
index 51b6d2c66..e74f0a6b1 100644
--- a/pkg/utils/utils_test.go
+++ b/pkg/utils/utils_test.go
@@ -432,13 +432,25 @@ var _ = Describe("In the utils package", func() {
 			&FakeFilesystem{Dirs: []string{"sys/bus/pci/devices/0000:01:10.0/physfn/net/"}},
 			"0000:01:10.0", "", true,
 		),
-		Entry("pf net is not a directory at all",
+		Entry("PF net is not a directory at all",
 			&FakeFilesystem{
 				Dirs:  []string{"sys/bus/pci/devices/0000:01:10.0/physfn"},
 				Files: map[string][]byte{"sys/bus/pci/devices/0000:01:10.0/physfn/net/": []byte("junk")},
 			},
 			"0000:01:10.0", "", true,
 		),
+		Entry("PF net is in virtio path",
+			&FakeFilesystem{
+				Dirs: []string{
+					"sys/bus/pci/devices/0000:01:00.0/",
+					"sys/bus/pci/devices/0000:01:10.0/",
+					"sys/bus/pci/devices/0000:01:00.0/virtio1/net/fakePF"},
+				Symlinks: map[string]string{
+					"sys/bus/pci/devices/0000:01:10.0/physfn": "../0000:01:00.0",
+				},
+			},
+			"0000:01:10.0", "fakePF", false,
+		),
 	)
 
 	DescribeTable("getting PF names switchdev",