bepin/BepInEx/ConsoleUtil/ConsoleEncoding/ConsoleEncoding.cs

// --------------------------------------------------
// UnityInjector - ConsoleEncoding.cs
// Copyright (c) Usagirei 2015 - 2015
// --------------------------------------------------

using System;
using System.Text;

namespace UnityInjector.ConsoleUtil
{
    // --------------------------------------------------
    // Code ported from
    // https://gist.github.com/asm256/9bfb88336a1433e2328a
    // Which in turn was seemingly ported from
    // http://jonskeet.uk/csharp/ebcdic/
    // using only safe (managed) code
    // --------------------------------------------------
    public partial class ConsoleEncoding : Encoding
    {
        private readonly uint _codePage;
        public override int CodePage => (int) _codePage;

        public static uint ConsoleCodePage
        {
            get { return GetConsoleOutputCP(); }
            set { SetConsoleOutputCP(value); }
        }

        public static uint GetActiveCodePage()
        {
            return GetACP();
        }

        private ConsoleEncoding(uint codePage)
        {
            _codePage = codePage;
        }

        public static ConsoleEncoding GetEncoding(uint codePage)
        {
            return new ConsoleEncoding(codePage);
        }

        public override int GetByteCount(char[] chars, int index, int count)
        {
            WriteCharBuffer(chars, index, count);
            int result = WideCharToMultiByte(_codePage, 0, _charBuffer, count, _zeroByte, 0, IntPtr.Zero, IntPtr.Zero);
            return result;
        }

        public override int GetBytes(char[] chars, int charIndex, int charCount, byte[] bytes, int byteIndex)
        {
            var byteCount = GetMaxByteCount(charCount);

            WriteCharBuffer(chars, charIndex, charCount);

            ExpandByteBuffer(byteCount);
            int result = WideCharToMultiByte(_codePage, 0, chars, charCount, _byteBuffer, byteCount, IntPtr.Zero, IntPtr.Zero);
            ReadByteBuffer(bytes, byteIndex, byteCount);

            return result;
        }

        public override int GetCharCount(byte[] bytes, int index, int count)
        {
            WriteByteBuffer(bytes, index, count);
            int result = MultiByteToWideChar(_codePage, 0, bytes, count, _zeroChar, 0);
            return result;
        }

        public override int GetChars(byte[] bytes, int byteIndex, int byteCount, char[] chars, int charIndex)
        {
            var charCount = GetMaxCharCount(byteCount);

            WriteByteBuffer(bytes, byteIndex, byteCount);

            ExpandCharBuffer(charCount);
            int result = MultiByteToWideChar(_codePage, 0, bytes, byteCount, _charBuffer, charCount);
            ReadCharBuffer(chars, charIndex, charCount);

            return result;
        }

        public override int GetMaxByteCount(int charCount) => charCount * 2;
        public override int GetMaxCharCount(int byteCount) => byteCount;
    }
}