NTFS: AttributeRecord: Added Clone method

Author: TalAloni

DiskAccessLibrary/FileSystems/NTFS/AttributeRecord/AttributeRecord.cs

@@ -71,6 +71,8 @@ public void WriteHeader(byte[] buffer, ushort nameOffset)
             }
         }
 
+        public abstract AttributeRecord Clone();
+
         public AttributeType AttributeType
         {
             get

DiskAccessLibrary/FileSystems/NTFS/AttributeRecord/FileNameAttributeRecord.cs

@@ -34,6 +34,13 @@ public override byte[] GetBytes()
             return base.GetBytes();
         }
 
+        public override AttributeRecord Clone()
+        {
+            FileNameAttributeRecord clone = (FileNameAttributeRecord)base.Clone();
+            clone.Record = this.Record.Clone();
+            return clone;
+        }
+
         public override ulong DataLength
         {
             get

DiskAccessLibrary/FileSystems/NTFS/AttributeRecord/IndexRootRecord.cs

@@ -61,6 +61,18 @@ public override byte[] GetBytes()
             return base.GetBytes();
         }
 
+        public override AttributeRecord Clone()
+        {
+            IndexRootRecord clone = (IndexRootRecord)base.Clone();
+            clone.m_indexHeader = m_indexHeader.Clone();
+            clone.IndexEntries = new List<IndexEntry>();
+            foreach (IndexEntry entry in IndexEntries)
+            {
+                clone.IndexEntries.Add(entry.Clone());
+            }
+            return clone;
+        }
+
         public override ulong DataLength
         {
             get

DiskAccessLibrary/FileSystems/NTFS/AttributeRecord/NonResidentAttributeRecord.cs

@@ -87,6 +87,13 @@ public KeyValuePairList<long, long> GetClustersInUse()
             return sequence;
         }
 
+        public override AttributeRecord Clone()
+        {
+            NonResidentAttributeRecord clone = (NonResidentAttributeRecord)this.MemberwiseClone();
+            clone.m_dataRunSequence = m_dataRunSequence.Clone();
+            return clone;
+        }
+
         /// <summary>
         /// Each attribute record must be aligned to 8-byte boundary, so RecordLength must be a multiple of 8.
         /// When reading attributes, they may contain additional padding,

DiskAccessLibrary/FileSystems/NTFS/AttributeRecord/ResidentAttributeRecord.cs

@@ -58,6 +58,13 @@ public override byte[] GetBytes()
             return buffer;
         }
 
+        public override AttributeRecord Clone()
+        {
+            ResidentAttributeRecord clone = (ResidentAttributeRecord)this.MemberwiseClone();
+            clone.Data = (byte[])this.Data.Clone();
+            return clone;
+        }
+
         public override ulong DataLength
         {
             get

DiskAccessLibrary/FileSystems/NTFS/Index/IndexEntry.cs

@@ -73,6 +73,13 @@ public void WriteBytes(byte[] buffer, int offset)
             }
         }
 
+        public IndexEntry Clone()
+        {
+            IndexEntry clone = (IndexEntry)this.MemberwiseClone();
+            clone.FileReference = this.FileReference.Clone();
+            return clone;
+        }
+
         public byte[] GetBytes()
         {
             byte[] buffer = new byte[Length];

DiskAccessLibrary/FileSystems/NTFS/Index/IndexHeader.cs

@@ -41,6 +41,11 @@ public void WriteBytes(byte[] buffer, int offset)
             ByteWriter.WriteByte(buffer, offset + 0x0C, (byte)IndexFlags);
         }
 
+        public IndexHeader Clone()
+        {
+            return (IndexHeader)this.MemberwiseClone();
+        }
+
         public bool IsParentNode
         {
             get

DiskAccessLibrary/FileSystems/NTFS/Structures/DataRun.cs

@@ -74,6 +74,11 @@ public byte[] GetBytes()
             return buffer;
         }
 
+        public DataRun Clone()
+        {
+            return (DataRun)this.MemberwiseClone();
+        }
+
         private static long ReadVarLong(ref byte[] buffer, int offset, int size)
         {
             ulong val = 0;

DiskAccessLibrary/FileSystems/NTFS/Structures/DataRunSequence.cs

@@ -21,6 +21,10 @@ public DataRunSequence() : base()
         {
         }
 
+        public DataRunSequence(int capacity) : base(capacity)
+        {
+        }
+
         public DataRunSequence(byte[] buffer, int offset, int length) : base()
         {
             int position = offset;
@@ -200,6 +204,16 @@ public long GetDataClusterLCN(long clusterVCN)
             throw new InvalidDataException("Invalid cluster VCN");
         }
 
+        public DataRunSequence Clone()
+        {
+            DataRunSequence clone = new DataRunSequence(this.Count);
+            foreach (DataRun run in this)
+            {
+                clone.Add(run.Clone());
+            }
+            return clone;
+        }
+
         public override string ToString()
         {
             StringBuilder builder = new StringBuilder();

DiskAccessLibrary/FileSystems/NTFS/Structures/FileNameRecord.cs

@@ -78,6 +78,13 @@ public byte[] GetBytes()
             return buffer;
         }
 
+        public FileNameRecord Clone()
+        {
+            FileNameRecord clone = (FileNameRecord)this.MemberwiseClone();
+            clone.ParentDirectory = this.ParentDirectory.Clone();
+            return clone;
+        }
+
         /// <param name="fileNameNamespace">POSIX or Win32 or DOS, multiple flags are not supported</param>
         public bool IsInNamespace(FileNameFlags fileNameNamespace)
         {