PXZ improvements

BRH-Media · Aug 19, 2020 · 007fe52 · 007fe52
1 parent e20cdf9
commit 007fe52
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/PlexDL.Common.Pxz/Structures/PxzRecordContent.cs b/PlexDL.Common.Pxz/Structures/PxzRecordContent.cs
@@ -1,4 +1,6 @@
-using PlexDL.Common.Pxz.Compressors;
+using PlexDL.Common.Enums;
+using PlexDL.Common.Pxz.Compressors;
+using PlexDL.Common.Security;
 using System.Drawing;
 using System.Text;
 using System.Xml;
@@ -8,6 +10,9 @@ namespace PlexDL.Common.Pxz.Structures
 {
     public class PxzRecordContent
     {
+        [XmlIgnore]
+        private bool _encrypted = false;
+
         public PxzRecordContent()
         {
             //blank initialiser
@@ -54,6 +59,11 @@ public XmlDocument ToXmlDocument()
 
         public Image ToImage() => Utilities.ByteToImage(Record);
 
+        /// <summary>
+        /// Determines whether the record's contents have been encrypted via DPAPI
+        /// </summary>
+        public bool Encrypted => _encrypted;
+
         /// <summary>
         /// Raw compressed base64-encoded (ASCII/UTF8) Gzipped bytes
         /// </summary>
@@ -75,7 +85,15 @@ public byte[] Record
         {
             get
             {
-                var r = !string.IsNullOrEmpty(RawRecord) ? GZipCompressor.DecompressBytes(RawRecord) : null;
+                var raw = RawRecord;
+
+                if (Encrypted)
+                {
+                    var provider = new ProtectedString(raw, StringProtectionMode.Decrypt);
+                    raw = provider.ProcessedValue;
+                }
+
+                var r = !string.IsNullOrEmpty(raw) ? GZipCompressor.DecompressBytes(raw) : null;
                 TmpRecord = r;
                 return r;
             }