From 6ad24b2b8c363d21eaf1f204b8d90db9ab878299 Mon Sep 17 00:00:00 2001
From: fanhy36 <fanhy36@chinaunicom.cn>
Date: Tue, 11 Mar 2025 10:22:01 +0800
Subject: [PATCH] fix issue75: seg fault and listen volcano.sock fail

---
 pkg/plugin/nvidia/server.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/pkg/plugin/nvidia/server.go b/pkg/plugin/nvidia/server.go
index cf312a89a..cb6eb622b 100644
--- a/pkg/plugin/nvidia/server.go
+++ b/pkg/plugin/nvidia/server.go
@@ -186,9 +186,20 @@ func (m *NvidiaDevicePlugin) DevicesNum() int {
 
 // Serve starts the gRPC server of the device plugin.
 func (m *NvidiaDevicePlugin) Serve() error {
+
 	sock, err := net.Listen("unix", m.socket)
 	if err != nil {
-		return err
+		log.Printf("Listen sock fail and retry for '%s': %s", m.resourceName, err)
+		err = os.Remove(m.socket)
+		if err != nil {
+			log.Printf("Error deleting file: %s, %v\n", m.socket, err)
+			return err
+		}
+		sock, err = net.Listen("unix", m.socket)
+		if err != nil {
+			log.Printf("Retry Listen sock fail '%s': %s", m.resourceName, err)
+			return err
+		}
 	}
 
 	pluginapi.RegisterDevicePluginServer(m.server, m)
@@ -343,6 +354,7 @@ func (m *NvidiaDevicePlugin) Allocate(ctx context.Context, reqs *pluginapi.Alloc
 	}
 
 	sort.Sort(availablePods)
+	util.UseClient(m.kubeInteractor.clientset)
 
 	var candidatePod *v1.Pod
 	for _, pod := range availablePods {
@@ -406,7 +418,6 @@ Allocate:
 		return nil, fmt.Errorf("failed to update pod annotation %v", err)
 	}
 
-	util.UseClient(m.kubeInteractor.clientset)
 	klog.V(3).Infoln("Releasing lock: nodeName=", m.kubeInteractor.nodeName)
 	err = util.ReleaseNodeLock(m.kubeInteractor.nodeName, "gpu")
 	if err != nil {