From c07bbe5137067aacfdb8ac5d5bbd6c9f0a074d53 Mon Sep 17 00:00:00 2001
From: Natalia Portillo <claunia@claunia.com>
Date: Fri, 15 Mar 2019 22:37:48 +0000
Subject: [PATCH] Use new marshaller in QEMU Copy-On-Write disk image.

---
 DiscImageChef.DiscImages/QCOW/Read.cs  | 13 ++++++-------
 DiscImageChef.DiscImages/QCOW/Write.cs |  9 ++++++---
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/DiscImageChef.DiscImages/QCOW/Read.cs b/DiscImageChef.DiscImages/QCOW/Read.cs
index 98584d30a..25683689a 100644
--- a/DiscImageChef.DiscImages/QCOW/Read.cs
+++ b/DiscImageChef.DiscImages/QCOW/Read.cs
@@ -33,13 +33,14 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Runtime.InteropServices;
 using DiscImageChef.CommonTypes;
 using DiscImageChef.CommonTypes.Enums;
 using DiscImageChef.CommonTypes.Interfaces;
 using DiscImageChef.Console;
-using DiscImageChef.Helpers;
 using SharpCompress.Compressors;
 using SharpCompress.Compressors.Deflate;
+using Marshal = DiscImageChef.Helpers.Marshal;
 
 namespace DiscImageChef.DiscImages
 {
@@ -108,10 +109,9 @@ namespace DiscImageChef.DiscImages
             byte[] l1TableB = new byte[l1Size * 8];
             stream.Seek((long)qHdr.l1_table_offset, SeekOrigin.Begin);
             stream.Read(l1TableB, 0, (int)l1Size * 8);
-            l1Table = new ulong[l1Size];
+            l1Table = MemoryMarshal.Cast<byte, ulong>(l1TableB).ToArray();
             DicConsole.DebugWriteLine("QCOW plugin", "Reading L1 table");
-            for(long i = 0; i < l1Table.LongLength; i++)
-                l1Table[i] = BigEndianBitConverter.ToUInt64(l1TableB, (int)(i * 8));
+            for(long i = 0; i < l1Table.LongLength; i++) l1Table[i] = Swapping.Swap(l1Table[i]);
 
             l1Mask = 0;
             int c = 0;
@@ -189,13 +189,12 @@ namespace DiscImageChef.DiscImages
 
             if(!l2TableCache.TryGetValue(l1Off, out ulong[] l2Table))
             {
-                l2Table = new ulong[l2Size];
                 imageStream.Seek((long)l1Table[l1Off], SeekOrigin.Begin);
                 byte[] l2TableB = new byte[l2Size * 8];
                 imageStream.Read(l2TableB, 0, l2Size * 8);
                 DicConsole.DebugWriteLine("QCOW plugin", "Reading L2 table #{0}", l1Off);
-                for(long i = 0; i < l2Table.LongLength; i++)
-                    l2Table[i] = BigEndianBitConverter.ToUInt64(l2TableB, (int)(i * 8));
+                l2Table = MemoryMarshal.Cast<byte, ulong>(l2TableB).ToArray();
+                for(long i = 0; i < l2Table.LongLength; i++) l2Table[i] = Swapping.Swap(l2Table[i]);
 
                 if(l2TableCache.Count >= maxL2TableCache) l2TableCache.Clear();
 
diff --git a/DiscImageChef.DiscImages/QCOW/Write.cs b/DiscImageChef.DiscImages/QCOW/Write.cs
index 0733c82e3..5caa50577 100644
--- a/DiscImageChef.DiscImages/QCOW/Write.cs
+++ b/DiscImageChef.DiscImages/QCOW/Write.cs
@@ -34,11 +34,12 @@ using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using System.Runtime.InteropServices;
 using DiscImageChef.CommonTypes;
 using DiscImageChef.CommonTypes.Enums;
 using DiscImageChef.CommonTypes.Structs;
-using DiscImageChef.Helpers;
 using Schemas;
+using Marshal = DiscImageChef.Helpers.Marshal;
 
 namespace DiscImageChef.DiscImages
 {
@@ -271,8 +272,10 @@ namespace DiscImageChef.DiscImages
             writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.l1_table_offset), 0, 8);
 
             writingStream.Seek((long)qHdr.l1_table_offset, SeekOrigin.Begin);
-            for(long i = 0; i < l1Table.LongLength; i++)
-                writingStream.Write(BigEndianBitConverter.GetBytes(l1Table[i]), 0, 8);
+            for(long i = 0; i < l1Table.LongLength; i++) l1Table[i] = Swapping.Swap(l1Table[i]);
+            byte[] l1TableB =
+                MemoryMarshal.Cast<ulong, byte>(l1Table).ToArray();
+            writingStream.Write(l1TableB, 0, l1TableB.Length);
 
             writingStream.Flush();
             writingStream.Close();