Fixbug: check driver timeout

cmd/csi-attacher/main.go

@@ -120,7 +120,9 @@ func main() {
 			os.Exit(1)
 		}
 
-		supportsService, err := csiConn.SupportsPluginControllerService(ctx)
+		ctxController, cancelController := context.WithTimeout(context.Background(), csiTimeout)
+		defer cancelController()
+		supportsService, err := csiConn.SupportsPluginControllerService(ctxController)
 		if err != nil {
 			glog.Error(err.Error())
 			os.Exit(1)
@@ -130,7 +132,9 @@ func main() {
 			glog.V(2).Infof("CSI driver does not support Plugin Controller Service, using trivial handler")
 		} else {
 			// Find out if the driver supports attach/detach.
-			supportsAttach, err := csiConn.SupportsControllerPublish(ctx)
+			ctxPublish, cancelPublish := context.WithTimeout(context.Background(), csiTimeout)
+			defer cancelPublish()
+			supportsAttach, err := csiConn.SupportsControllerPublish(ctxPublish)
 			if err != nil {
 				glog.Error(err.Error())
 				os.Exit(1)
@@ -195,7 +199,7 @@ func waitForDriverReady(csiConn connection.CSIConnection, timeout time.Duration)
 	finish := now.Add(timeout)
 	var err error
 	for {
-		ctx, cancel := context.WithTimeout(context.Background(), csiTimeout)
+		ctx, cancel := context.WithTimeout(context.Background(), timeout)
 		defer cancel()
 		err = csiConn.Probe(ctx)
 		if err == nil {