1 files changed, 182 insertions, 0 deletions

diff --git a/src/WixToolset.Core.WindowsInstaller/Msi/Record.cs b/src/WixToolset.Core.WindowsInstaller/Msi/Record.cs
new file mode 100644
index 00000000..438aa3b0
--- /dev/null
+++ b/src/WixToolset.Core.WindowsInstaller/Msi/Record.cs
@@ -0,0 +1,182 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Msi
+{
+    using System;
+    using System.ComponentModel;
+    using System.Text;
+    using WixToolset.Core.Native;
+
+    /// <summary>
+    /// Wrapper class around msi.dll interop for a record.
+    /// </summary>
+    public sealed class Record : MsiHandle
+    {
+        /// <summary>
+        /// Creates a record with the specified number of fields.
+        /// </summary>
+        /// <param name="fieldCount">Number of fields in record.</param>
+        public Record(int fieldCount)
+        {
+            this.Handle = MsiInterop.MsiCreateRecord(fieldCount);
+            if (0 == this.Handle)
+            {
+                throw new OutOfMemoryException();
+            }
+        }
+
+        /// <summary>
+        /// Creates a record from a handle.
+        /// </summary>
+        /// <param name="handle">Handle to create record from.</param>
+        internal Record(uint handle)
+        {
+            this.Handle = handle;
+        }
+
+        /// <summary>
+        /// Gets a string value at specified location.
+        /// </summary>
+        /// <param name="field">Index into record to get string.</param>
+        public string this[int field]
+        {
+            get { return this.GetString(field); }
+            set { this.SetString(field, (string)value); }
+        }
+
+        /// <summary>
+        /// Determines if the value is null at the specified location.
+        /// </summary>
+        /// <param name="field">Index into record of the field to query.</param>
+        /// <returns>true if the value is null, false otherwise.</returns>
+        public bool IsNull(int field)
+        {
+            int error = MsiInterop.MsiRecordIsNull(this.Handle, field);
+
+            switch (error)
+            {
+                case 0:
+                    return false;
+                case 1:
+                    return true;
+                default:
+                    throw new Win32Exception(error);
+            }
+        }
+
+        /// <summary>
+        /// Gets integer value at specified location.
+        /// </summary>
+        /// <param name="field">Index into record to get integer</param>
+        /// <returns>Integer value</returns>
+        public int GetInteger(int field)
+        {
+            return MsiInterop.MsiRecordGetInteger(this.Handle, field);
+        }
+
+        /// <summary>
+        /// Sets integer value at specified location.
+        /// </summary>
+        /// <param name="field">Index into record to set integer.</param>
+        /// <param name="value">Value to set into record.</param>
+        public void SetInteger(int field, int value)
+        {
+            int error = MsiInterop.MsiRecordSetInteger(this.Handle, field, value);
+            if (0 != error)
+            {
+                throw new Win32Exception(error);
+            }
+        }
+
+        /// <summary>
+        /// Gets string value at specified location.
+        /// </summary>
+        /// <param name="field">Index into record to get string.</param>
+        /// <returns>String value</returns>
+        public string GetString(int field)
+        {
+            int bufferSize = 255;
+            StringBuilder buffer = new StringBuilder(bufferSize);
+            int error = MsiInterop.MsiRecordGetString(this.Handle, field, buffer, ref bufferSize);
+            if (234 == error)
+            {
+                buffer.EnsureCapacity(++bufferSize);
+                error = MsiInterop.MsiRecordGetString(this.Handle, field, buffer, ref bufferSize);
+            }
+
+            if (0 != error)
+            {
+                throw new Win32Exception(error);
+            }
+
+            return (0 < buffer.Length ? buffer.ToString() : null);
+        }
+
+        /// <summary>
+        /// Set string value at specified location
+        /// </summary>
+        /// <param name="field">Index into record to set string.</param>
+        /// <param name="value">Value to set into record</param>
+        public void SetString(int field, string value)
+        {
+            int error = MsiInterop.MsiRecordSetString(this.Handle, field, value);
+            if (0 != error)
+            {
+                throw new Win32Exception(error);
+            }
+        }
+
+        /// <summary>
+        /// Get stream at specified location.
+        /// </summary>
+        /// <param name="field">Index into record to get stream.</param>
+        /// <param name="buffer">buffer to receive bytes from stream.</param>
+        /// <param name="requestedBufferSize">Buffer size to read.</param>
+        /// <returns>Stream read into string.</returns>
+        public int GetStream(int field, byte[] buffer, int requestedBufferSize)
+        {
+            int bufferSize = 255;
+            if (requestedBufferSize > 0)
+            {
+                bufferSize = requestedBufferSize;
+            }
+
+            int error = MsiInterop.MsiRecordReadStream(this.Handle, field, buffer, ref bufferSize);
+            if (0 != error)
+            {
+                throw new Win32Exception(error);
+            }
+
+            return bufferSize;
+        }
+
+        /// <summary>
+        /// Sets a stream at a specified location.
+        /// </summary>
+        /// <param name="field">Index into record to set stream.</param>
+        /// <param name="path">Path to file to read into stream.</param>
+        public void SetStream(int field, string path)
+        {
+            int error = MsiInterop.MsiRecordSetStream(this.Handle, field, path);
+            if (0 != error)
+            {
+                throw new Win32Exception(error);
+            }
+        }
+
+        /// <summary>
+        /// Gets the number of fields in record.
+        /// </summary>
+        /// <returns>Count of fields in record.</returns>
+        public int GetFieldCount()
+        {
+            int size = MsiInterop.MsiRecordGetFieldCount(this.Handle);
+            if (0 > size)
+            {
+                throw new Win32Exception();
+            }
+
+            return size;
+        }
+    }
+}