src/AccessibilityInsights.SetupLibrary/FileHelpers.cs - main - NordPixel/AppInspectorWindows

src/AccessibilityInsights.SetupLibrary/FileHelpers.cs
101 строка4 KB
Dave Tryon
feat(upgrader-security): Consume signed manifests, add telemetry and docs (#1493)
30 ноя 2022, 23:44
30 ноя 2022, 23:44391a544
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
using Newtonsoft.Json;
using System;
using System.IO;
using System.Reflection.Metadata;
using System.Reflection.PortableExecutable;
using System.Text;

using static System.FormattableString;

namespace AccessibilityInsights.SetupLibrary
{
    /// <summary>
    /// Classes to help manage file serialization
    /// </summary>
    public static class FileHelpers
    {
        /// <summary>
        /// Serialize data to JSON format at the specified path
        /// </summary>
        /// <param name="data">The data being serialized</param>
        /// <param name="path">Config file location and name</param>
        public static void SerializeDataToJSON(object data, string path)
        {
            var json = JsonConvert.SerializeObject(data, Formatting.Indented);
            File.WriteAllText(path, json, Encoding.UTF8);
        }

        /// <summary>
        /// Load JSON serialized data from the specified path
        /// </summary>
        /// <typeparam name="T">Configuration type</typeparam>
        /// <param name="path">Config file location and name</param>
        /// <returns>The serialized data if successful, default(T) if not</returns>
        public static T LoadDataFromJSON<T>(string path)
        {
            T data = default(T);

            if (File.Exists(path))
            {
                var json = File.ReadAllText(path);
                data = JsonConvert.DeserializeObject<T>(json);
            }

            return data;
        }

        /// <summary>
        /// Rename the existing configuration to .bak file.
        /// </summary>
        /// <param name="path">The file to rename</param>
        /// <returns>true if the file was found and renamed</returns>
        public static bool RenameFileAsBackup(string path)
        {
            if (File.Exists(path))
            {
                File.Move(path, Invariant($"{path}.{DateTime.Now.Ticks}.bak"));
                return true;
            }

            return false;
        }

        /// <summary>
        /// Create folder if it doesn't exist.
        /// </summary>
        /// <param name="path"></param>
        public static void CreateFolder(string path)
        {
            if (Directory.Exists(path) == false)
            {
                Directory.CreateDirectory(path);
            }
        }

        internal static byte[] ExtractResourceFromStream(Stream stream, string resourceName)
        {
#pragma warning disable CA2000 // Dispose objects before losing scope
            // Don't dispose the StreamReader here, since it has the side effect of also
            // disposing the passed-in Stream, which we don't own.
            PEReader peReader = new PEReader(stream);
#pragma warning restore CA2000 // Dispose objects before losing scope
            MetadataReader metadataReader = peReader.GetMetadataReader();
            foreach (ManifestResourceHandle handle in metadataReader.ManifestResources)
            {
                ManifestResource manifestResource = metadataReader.GetManifestResource(handle);
                if (!metadataReader.StringComparer.Equals(manifestResource.Name, resourceName, false))
                    continue;

                PEMemoryBlock resourceDirectory = peReader.GetSectionData(peReader.PEHeaders.CorHeader.ResourcesDirectory.RelativeVirtualAddress);
                BlobReader blobReader = resourceDirectory.GetReader((int)manifestResource.Offset, resourceDirectory.Length - (int)manifestResource.Offset);
                uint size = blobReader.ReadUInt32();
                byte[] bytes = blobReader.ReadBytes((int)size);
                return bytes;
            }

            return null;
        }
    }
}