commit: add switch to auto set RamDisk letter to Z

SvdCli/CliProgram.cs

@@ -6,6 +6,7 @@ using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using System.Management;
 using System.Net;
 using System.Reflection;
 using System.Security;
@@ -20,7 +21,7 @@ namespace SvdCli
     {
         private static void Main(string[] args)
         {
-            if (args.Length < 3)
+            if (args.Length < 2)
             {
                 Console.WriteLine(@"Usage:");
                 Console.WriteLine(@"");
@@ -36,6 +37,7 @@ namespace SvdCli
                 Console.WriteLine(@"[Install ServiceName|Service] <Mount|Server 0.0.0.0 2333> Image C:\Path\To\Your\Image.img [ReadOnly]");
                 Console.WriteLine(@"[Install ServiceName|Service] <Mount|Server 0.0.0.0 2333> Bdd C:\Path\To\Your\Bdd.bdd [ReadOnly]");
                 Console.WriteLine(@"[Install ServiceName|Service] Mount Net 192.168.233.233 2333");
+                Console.WriteLine(@"[Install ServiceName|Service] Mount ZTempNtfsRamDisk 10GB");
                 Console.WriteLine(@"[Install ServiceName|Service] Server 192.168.233.233 2333 Dispatch 64KB C:\Path\To\Your\Image.img C:\Path\To\Your\Dir\");
                 Console.WriteLine(@"");
                 Console.WriteLine(@"Uninstall ServiceName           Safe:Restrict to register by this program");
@@ -200,6 +202,7 @@ namespace SvdCli
                 case "ramdisk":
                 case "ntfsramdisk":
                 case "tempntfsramdisk":
+                case "ztempntfsramdisk":
                     if (ParseDiskSize(args[1], out var value, out var unit))
                     {
                         if (false == string.IsNullOrWhiteSpace(unit) && false == unit.StartsWith("g"))
@@ -212,8 +215,27 @@ namespace SvdCli
                         if (args.Length > 3 && args[3].ToLower() == "readonly") storage.WriteProtect = true;
 
 
-                        if (op == "ntfsramdisk") FsMaker.MakeNtfsRamDisk(storage, false);
-                        if (op == "tempntfsramdisk") FsMaker.MakeNtfsRamDisk(storage, true);
+                        if (op == "ntfsramdisk") FsMaker.MakeNtfs(storage, "RamDisk");
+                        if (op == "tempntfsramdisk") FsMaker.MakeNtfs(storage, "RamDisk", n => n.CreateDirectory("Temp"));
+                        if (op == "ztempntfsramdisk")
+                        {
+                            FsMaker.MakeNtfs(storage, "RamDisk", n => n.CreateDirectory("Temp"));
+                            storage.Mounted += delegate
+                            {
+                                var volumes = new ManagementObjectSearcher("root\\CIMV2", "SELECT * FROM Win32_Volume");
+                                foreach (ManagementObject volume in volumes.Get())
+                                {
+                                    volume.Get();
+                                    var letter = volume["DriveLetter"];
+                                    var label = volume["Label"];
+                                    if (label is string str && str == "RamDisk")
+                                    {
+                                        volume["DriveLetter"] = "Z:";
+                                        volume.Put();
+                                    }
+                                }
+                            };
+                        }
 
                         return storage;
                     }

SvdCli/DriverAdapt/VirtualDrive.cs

@@ -39,6 +39,7 @@ namespace SvdCli.DriverAdapt
         {
             _storage.Init();
             _host.Start();
+            _storage.FireMountedEvent();
         }
 
         public void Detach()

SvdCli/ImageUtils/BlockDifferencingDiskImage/BddImage.cs

@@ -0,0 +1,40 @@
+using SvdCli.Storage;
+using System.IO;
+
+namespace SvdCli.ImageUtils.BlockDifferencingDiskImage
+{
+    internal static class BddImage
+    {
+        public static void CreateStandaloneBddImage(string path, uint blockSize, uint blockCount)
+        {
+            using var stream = File.Create(path);
+
+            const int kb16 = CapacityUnits.KiloByte * 16;
+            const int kb64 = CapacityUnits.KiloByte * 64;
+
+            stream.WriteByte((byte)'B'); // Signature BDD2
+            stream.WriteByte((byte)'D');
+            stream.WriteByte((byte)'D');
+            stream.WriteByte((byte)'2');
+            stream.WriteByte(0); // Standalone
+            stream.WriteLeu32(blockSize);
+            stream.WriteLeu32(blockCount);
+            stream.Position = kb16 + 8 * blockCount;
+            //align to 64kb
+            stream.SetLength(stream.Position + (kb64 - (stream.Position % kb64)));
+        }
+
+        private static void WriteLeu32(this Stream stream, uint value)
+        {
+            var buf = new[]
+            {
+                (byte)(value),
+                (byte)(value>>8),
+                (byte)(value>>16),
+                (byte)(value>>24),
+            };
+
+            stream.Write(buf, 0, buf.Length);
+        }
+    }
+}

SvdCli/BlockDifferencingDiskImage/Format.txt

@@ -2,19 +2,19 @@
 ======
 16KB info area
 16KB Aligned index entries block
-16KB ** Data blocks
+**   Data blocks
 
 Info area
 ------
 4: Fixed `BDD2' signature
-1: LEU16 prefix of base image file path UTF-8
+1: LEU16 prefix of base image file path UTF-8, 0 for Standalone
 0-255: ...
 4: LEU32 Size of block
 4: LEU32 Number of block
-...: Padding to 64KB
+...: Padding to 16KB
 
 Index entry
 ------
-8: LEU32 Offset to file
+7: LEU56 Offset to file
 1: Flags [Allocated,Trimed,...]
 

SvdCli/DiskUtils/FsMaker.cs

@@ -10,7 +10,7 @@ namespace SvdCli.DiskUtils
 {
     internal static class FsMaker
     {
-        public static void MakeNtfsRamDisk(ISvdStorage storage, bool createTempFolder)
+        public static void MakeNtfs(ISvdStorage storage, string label, Action<NtfsFileSystem> afterFormat = null)
         {
             storage.Init();
             using var sss = new SvdStorageStream(storage);
@@ -18,8 +18,8 @@ namespace SvdCli.DiskUtils
             BiosPartitionTable.Initialize(disk, WellKnownPartitionType.WindowsNtfs);
             var volMgr = new VolumeManager(disk);
 
-            using var destNtfs = NtfsFileSystem.Format(volMgr.GetLogicalVolumes()[0], "RamDIsk", new NtfsFormatOptions());
-            if (createTempFolder) destNtfs.CreateDirectory("Temp");
+            using var destNtfs = NtfsFileSystem.Format(volMgr.GetLogicalVolumes()[0], label, new NtfsFormatOptions());
+            afterFormat?.Invoke(destNtfs);
         }
 
         private class SvdStorageStream : Stream

SvdCli/Storage/ISvdStorage.cs

@@ -1,4 +1,6 @@
-namespace SvdCli.Storage
+using System;
+
+namespace SvdCli.Storage
 {
     public interface ISvdStorage
     {
@@ -27,5 +29,8 @@
         void Flush();
 
         void Trim(params TrimDescriptor[] descriptors);
+        
+        event EventHandler Mounted;
+        void FireMountedEvent();
     }
 }

SvdCli/Storage/Implements/RamStorage.cs

@@ -1,6 +1,5 @@
 using System;
 using System.Runtime.InteropServices;
-using System.Runtime.InteropServices.WindowsRuntime;
 
 namespace SvdCli.Storage.Implements
 {
@@ -16,6 +15,8 @@ namespace SvdCli.Storage.Implements
         public bool SupportTrim { get; } = false;
         public bool WriteProtect { get; set; }
 
+        public event EventHandler Mounted;
+
         public UnmanagedGigabyteRamDisk(int sizeInGigabyte)
         {
             _blocks = new IntPtr[sizeInGigabyte];
@@ -159,5 +160,10 @@ namespace SvdCli.Storage.Implements
         public void Trim(params TrimDescriptor[] descriptors)
         {
         }
+
+        public void FireMountedEvent()
+        {
+            Mounted?.Invoke(this, EventArgs.Empty);
+        }
     }
 }

SvdCli/Storage/Implements/RawImageStorage.cs

@@ -18,6 +18,8 @@ namespace SvdCli.Storage.Implements
         public ulong BlockCount { get; }
         public bool WriteProtect { get; }
 
+        public event EventHandler Mounted;
+
         public RawImageStorage(string imageFilePath, bool readOnly = false)
         {
             _imageFilePath = imageFilePath;
@@ -53,7 +55,6 @@ namespace SvdCli.Storage.Implements
                 _fileStream.Close();
                 _fileStream = null;
             }
-
         }
 
         public void ReadBlocks(byte[] buffer, ulong blockIndex, uint blockCount)
@@ -125,5 +126,10 @@ namespace SvdCli.Storage.Implements
                 }
             }
         }
+
+        public void FireMountedEvent()
+        {
+            Mounted?.Invoke(this, EventArgs.Empty);
+        }
     }
 }

SvdCli/Storage/Implements/StandaloneBddImageStorage.cs

@@ -1,13 +1,64 @@
 using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
 
 namespace SvdCli.Storage.Implements
 {
-    public class StandaloneBddImageStorage
+    public class StandaloneBddImageStorage : ISvdStorage
     {
         // TODO: StandaloneBddImageStorage
+        public string ProductId { get; } = "Svd BddImage";
+
+        public string ProductVersion { get; } = "1.0";
+        public uint BlockSize { get; } = 512;
+        public ulong BlockCount { get; }
+        public uint MaxTransferLength { get; }
+        public bool SupportTrim { get; }
+        public bool WriteProtect { get; }
+
+        public event EventHandler Mounted;
+
+        public void Init()
+        {
+            throw new NotImplementedException();
+        }
+
+        public void Exit()
+        {
+            throw new NotImplementedException();
+        }
+
+        public void ReadBlocks(byte[] buffer, ulong blockIndex, uint blockCount)
+        {
+            throw new NotImplementedException();
+        }
+
+        public void WriteBlocks(byte[] buffer, ulong blockIndex, uint blockCount)
+        {
+            throw new NotImplementedException();
+        }
+
+        public int Read(byte[] buffer, long offset, int index, int count)
+        {
+            throw new NotImplementedException();
+        }
+
+        public void Write(byte[] buffer, long offset, int index, int count)
+        {
+            throw new NotImplementedException();
+        }
+
+        public void Flush()
+        {
+            throw new NotImplementedException();
+        }
+
+        public void Trim(params TrimDescriptor[] descriptors)
+        {
+            throw new NotImplementedException();
+        }
+
+        public void FireMountedEvent()
+        {
+            Mounted?.Invoke(this, EventArgs.Empty);
+        }
     }
 }

SvdCli/SvdCli.csproj

@@ -135,6 +135,7 @@
     </Reference>
     <Reference Include="System" />
     <Reference Include="System.Core" />
+    <Reference Include="System.Management" />
     <Reference Include="System.ServiceProcess" />
     <Reference Include="System.Xml.Linq" />
     <Reference Include="System.Data.DataSetExtensions" />
@@ -148,7 +149,8 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="CliProgram.cs" />
-    <Compile Include="DiskUtils\FsMaker.cs" />
+    <Compile Include="ImageUtils\BlockDifferencingDiskImage\BddImage.cs" />
+    <Compile Include="ImageUtils\FsMaker.cs" />
     <Compile Include="SampleProgram.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
     <Compile Include="ServiceUtils\ServiceAccount.cs" />
@@ -175,6 +177,7 @@
     <None Include="packages.config" />
   </ItemGroup>
   <ItemGroup>
+    <Content Include="ImageUtils\BlockDifferencingDiskImage\Format.txt" />
     <Content Include="ServiceUtils\SOURCE.txt" />
   </ItemGroup>
   <ItemGroup />

SvdCli/TasteProgram.cs

@@ -5,6 +5,7 @@ using DiscUtils.Partitions;
 using DiscUtils.Streams;
 using DiscUtils.Vhd;
 using System.IO;
+using System.Management;
 
 namespace SvdCli
 {
@@ -12,6 +13,22 @@ namespace SvdCli
     {
         private static void Main(string[] args)
         {
+            var volumes = new ManagementObjectSearcher("root\\CIMV2", "SELECT * FROM Win32_Volume");
+            foreach (ManagementObject volume in volumes.Get())
+            {
+                volume.Get();
+                var letter = volume["DriveLetter"];
+                var label = volume["Label"];
+                if (label is string str && str == "RamDisk")
+                {
+                    volume["DriveLetter"] = "Z:";
+                    volume.Put();
+                }
+            }
+        }
+
+        private static void CreateNtfsImage()
+        {
             var diskSize = 5L * 1024 * 1024 * 1024;
 
             using (var fs = new FileStream(@"z:\test.img", FileMode.OpenOrCreate))
@@ -41,7 +58,6 @@ namespace SvdCli
                 using var sparseStream = volMgr.GetLogicalVolumes()[0].Open();
                 new NtfsFileSystemChecker(sparseStream).Check(Console.Out, ReportLevels.All);
             }
-
         }
     }
 }