Merge pull request #1103 from andyzhangx/aznfs-fix

fix: make aznfs protocol consistent with nfs

Author: andyzhangx

pkg/blob/blob.go

@@ -728,6 +728,12 @@ func isSupportedContainerNamePrefix(prefix string) bool {
 	return true
 }
 
+// isNFSProtocol checks if the protocol is NFS or AZNFS
+func isNFSProtocol(protocol string) bool {
+	protocol = strings.ToLower(protocol)
+	return protocol == NFS || protocol == AZNFS
+}
+
 // get storage account from secrets map
 func getStorageAccount(secrets map[string]string) (string, string, error) {
 	if secrets == nil {

pkg/blob/blob_test.go

@@ -1661,3 +1661,46 @@ func TestIsSupportedAccessTier(t *testing.T) {
 		}
 	}
 }
+
+func TestIsNFSProtocol(t *testing.T) {
+	tests := []struct {
+		protocol       string
+		expectedResult bool
+	}{
+		{
+			protocol:       "",
+			expectedResult: false,
+		},
+		{
+			protocol:       "NFS",
+			expectedResult: true,
+		},
+		{
+			protocol:       "nfs",
+			expectedResult: true,
+		},
+		{
+			protocol:       "Nfs",
+			expectedResult: true,
+		},
+		{
+			protocol:       "NFSv3",
+			expectedResult: false,
+		},
+		{
+			protocol:       "aznfs",
+			expectedResult: true,
+		},
+		{
+			protocol:       "azNfs",
+			expectedResult: true,
+		},
+	}
+
+	for _, test := range tests {
+		result := isNFSProtocol(test.protocol)
+		if result != test.expectedResult {
+			t.Errorf("isNFSVolume(%s) returned with %v, not equal to %v", test.protocol, result, test.expectedResult)
+		}
+	}
+}

pkg/blob/controllerserver.go

@@ -207,7 +207,7 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 	}
 
 	if pointer.BoolDeref(enableBlobVersioning, false) {
-		if protocol == NFS || pointer.BoolDeref(isHnsEnabled, false) {
+		if isNFSProtocol(protocol) || pointer.BoolDeref(isHnsEnabled, false) {
 			return nil, status.Errorf(codes.InvalidArgument, "enableBlobVersioning is not supported for NFS protocol or HNS enabled account")
 		}
 	}
@@ -217,7 +217,7 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 	}
 
 	if subsID != "" && subsID != d.cloud.SubscriptionID {
-		if protocol == NFS {
+		if isNFSProtocol(protocol) {
 			return nil, status.Errorf(codes.InvalidArgument, fmt.Sprintf("NFS protocol is not supported in cross subscription(%s)", subsID))
 		}
 		if !storeAccountKey {
@@ -259,7 +259,7 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 		createPrivateEndpoint = pointer.BoolPtr(true)
 	}
 	accountKind := string(storage.KindStorageV2)
-	if protocol == NFS {
+	if isNFSProtocol(protocol) {
 		isHnsEnabled = pointer.Bool(true)
 		enableNfsV3 = pointer.Bool(true)
 		// NFS protocol does not need account key
@@ -378,7 +378,7 @@ func (d *Driver) CreateVolume(ctx context.Context, req *csi.CreateVolumeRequest)
 		}
 	}
 
-	if pointer.BoolDeref(createPrivateEndpoint, false) && protocol == NFS {
+	if pointer.BoolDeref(createPrivateEndpoint, false) && isNFSProtocol(protocol) {
 		// As for blobfuse/blobfuse2, serverName, i.e.,AZURE_STORAGE_BLOB_ENDPOINT env variable can't include
 		// "privatelink", issue: https://github.com/Azure/azure-storage-fuse/issues/1014
 		//

pkg/blob/nodeserver.go

@@ -318,7 +318,7 @@ func (d *Driver) NodeStageVolume(ctx context.Context, req *csi.NodeStageVolumeRe
 		serverAddress = fmt.Sprintf("%s.blob.%s", accountName, storageEndpointSuffix)
 	}
 
-	if protocol == NFS {
+	if isNFSProtocol(protocol) {
 		klog.V(2).Infof("target %v\nprotocol %v\n\nvolumeId %v\ncontext %v\nmountflags %v\nserverAddress %v",
 			targetPath, protocol, volumeID, attrib, mountFlags, serverAddress)