From 78b94c4e1b1b78b4c7402b4b984b0d5c6ecf10c0 Mon Sep 17 00:00:00 2001
From: Natalia Portillo <claunia@claunia.com>
Date: Sun, 23 Dec 2018 21:36:17 +0000
Subject: [PATCH] Add encoder for ATA IDENTIFY (PACKET) DEVICE.

---
 ATA/Identify.cs | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/ATA/Identify.cs b/ATA/Identify.cs
index d79bdc1..c76210b 100644
--- a/ATA/Identify.cs
+++ b/ATA/Identify.cs
@@ -1881,6 +1881,37 @@ namespace DiscImageChef.Decoders.ATA
             return ATAID;
         }
 
+        public static byte[] Encode(IdentifyDevice? identify)
+        {
+            if(identify is null) return null;
+
+            IdentifyDevice ataId = identify.Value;
+
+            ataId.WWN          = DescrambleWWN(ataId.WWN);
+            ataId.WWNExtension = DescrambleWWN(ataId.WWNExtension);
+
+            byte[] buf = new byte[512];
+            IntPtr ptr = Marshal.AllocHGlobal(512);
+            Marshal.StructureToPtr(ataId, ptr, false);
+            Marshal.Copy(ptr, buf, 0, 512);
+            Marshal.FreeHGlobal(ptr);
+
+            byte[] str = ScrambleATAString(ataId.SerialNumber, 20);
+            Array.Copy(str, 0, buf, 10 * 2, 20);
+            str = ScrambleATAString(ataId.FirmwareRevision, 8);
+            Array.Copy(str, 0, buf, 23 * 2, 8);
+            str = ScrambleATAString(ataId.Model, 40);
+            Array.Copy(str, 0, buf, 27 * 2, 40);
+            str = ScrambleATAString(ataId.AdditionalPID, 8);
+            Array.Copy(str, 0, buf, 170 * 2, 8);
+            str = ScrambleATAString(ataId.MediaSerial, 40);
+            Array.Copy(str, 0, buf, 176 * 2, 40);
+            str = ScrambleATAString(ataId.MediaManufacturer, 20);
+            Array.Copy(str, 0, buf, 196 * 2, 20);
+
+            return buf;
+        }
+
         public static string Prettify(byte[] IdentifyDeviceResponse)
         {
             if(IdentifyDeviceResponse.Length != 512) return null;
@@ -3540,5 +3571,13 @@ namespace DiscImageChef.Decoders.ATA
             string outStr = StringHandlers.CToString(outbuf);
             return outStr.Trim();
         }
+
+        static byte[] ScrambleATAString(string str, int length)
+        {
+            byte[] bytes = Encoding.ASCII.GetBytes(str);
+            byte[] buf   = new byte[length];
+            Array.Copy(bytes, 0, buf, 0, bytes.Length);
+            return buf;
+        }
     }
 }
\ No newline at end of file