Some more public registry value API

LTRData · Nov 12, 2023 · b626507 · b626507
1 parent bf4e5ed
commit b626507
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 6 deletions.
diff --git a/Library/DiscUtils.Registry/RegistryKey.cs b/Library/DiscUtils.Registry/RegistryKey.cs
@@ -738,7 +738,7 @@ public RegistryValue GetRegistryValue(string name)
         return null;
     }
 
-    private RegistryValue AddRegistryValue(string name)
+    public RegistryValue AddRegistryValue(string name)
     {
         var valueListMem = ArrayPool<byte>.Shared.Rent(_cell.NumValues * 4);
         try

diff --git a/Library/DiscUtils.Registry/RegistryValue.cs b/Library/DiscUtils.Registry/RegistryValue.cs
@@ -103,7 +103,26 @@ public object Value
             var buffer = ArrayPool<byte>.Shared.Rent(_cell.DataLength & int.MaxValue);
             try
             {
-                return ConvertToObject(GetData(buffer), DataType);
+                return ConvertToObject(GetRawData(buffer), DataType);
+            }
+            finally
+            {
+                ArrayPool<byte>.Shared.Return(buffer);
+            }
+        }
+    }
+
+    /// <summary>
+    /// Gets raw binary value without interpretation as any particular data type.
+    /// </summary>
+    public byte[] RawValue
+    {
+        get
+        {
+            var buffer = ArrayPool<byte>.Shared.Rent(_cell.DataLength & int.MaxValue);
+            try
+            {
+                return GetRawData(buffer).ToArray();
             }
             finally
             {
@@ -116,7 +135,7 @@ public object Value
     /// The raw value data as a byte array.
     /// </summary>
     /// <returns>The value as a raw byte array.</returns>
-    internal Span<byte> GetData(Span<byte> maxBytes)
+    internal Span<byte> GetRawData(Span<byte> maxBytes)
     {
         if (_cell.DataLength < 0)
         {
@@ -140,7 +159,7 @@ internal Span<byte> GetData(Span<byte> maxBytes)
     /// </summary>
     /// <param name="data">The data to store.</param>
     /// <param name="valueType">The type of the data.</param>
-    internal void SetData(ReadOnlySpan<byte> data, RegistryValueType valueType)
+    public void SetRawData(ReadOnlySpan<byte> data, RegistryValueType valueType)
     {
         // If we can place the data in the DataIndex field, do that to save space / allocation
         if ((valueType == RegistryValueType.Dword || valueType == RegistryValueType.DwordBigEndian) && data.Length <= 4)
@@ -184,7 +203,7 @@ internal void SetData(ReadOnlySpan<byte> data, RegistryValueType valueType)
     public void SetValue(object value, RegistryValueType valueType)
     {
         var data = ConvertToData(value, valueType);
-        SetData(data.Span, valueType);
+        SetRawData(data.Span, valueType);
     }
 
     /// <summary>
@@ -281,7 +300,7 @@ private string DataAsString()
         var buffer = ArrayPool<byte>.Shared.Rent(_cell.DataLength & int.MaxValue);
         try
         {
-            var data = GetData(buffer);
+            var data = GetRawData(buffer);
 
             switch (DataType)
             {