Adding a fix to allow encryption only on data volumes for linux VMs

Author: moisesac

src/ResourceManager/Compute/Commands.Compute/Extension/AzureDiskEncryption/SetAzureDiskEncryptionExtension.cs

@@ -411,6 +411,20 @@ public override void ExecuteCmdlet()
                         this.ResourceGroupName, VMName).Body;
 
                     currentOSType = virtualMachineResponse.StorageProfile.OsDisk.OsType;
+                    
+                    if (OperatingSystemTypes.Linux.Equals(currentOSType) &&
+                        !AzureDiskEncryptionExtensionContext.VolumeTypeData.Equals(VolumeType, StringComparison.InvariantCultureIgnoreCase))
+                    {
+                        ThrowTerminatingError(
+                            new ErrorRecord(
+                                new ArgumentException(
+                                    string.Format(
+                                        CultureInfo.CurrentUICulture,
+                                        "Enabling encryption is only allowed on Data volumes for Linux VMs.")),
+                                "InvalidType",
+                                ErrorCategory.NotImplemented,
+                                null));
+                    }
 
                     if (OperatingSystemTypes.Linux.Equals(currentOSType))
                     {