dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

src/Npgsql/Netstandard20/UnixDomainSocketEndPoint.cs
89 строк3 KB
Brar Piening
Switch to UnixDomainSocketEndPoint where possible
10 ноя 2020, 19:51
10 ноя 2020, 19:518c513ae
#if NETSTANDARD2_0
using System.Net.Sockets;
using System.Text;

// ReSharper disable once CheckNamespace
namespace System.Net
{
    // Copied and adapted from https://github.com/mono/mono/blob/master/mcs/class/Mono.Posix/Mono.Unix/UnixEndPoint.cs
    class UnixDomainSocketEndPoint : EndPoint
    {
        string _filename;

        public UnixDomainSocketEndPoint (string filename)
        {
            if (filename == null)
                throw new ArgumentNullException(nameof(filename));
            if (filename == "")
                throw new ArgumentException ("Cannot be empty.", nameof(filename));
            _filename = filename;
        }

        public string Filename {
            get => _filename;
            set => _filename = value;
        }

        public override AddressFamily AddressFamily => AddressFamily.Unix;

        public override EndPoint Create(SocketAddress socketAddress)
        {
            /*
             * Should also check this
             *
            int addr = (int) AddressFamily.Unix;
            if (socketAddress [0] != (addr & 0xFF))
                throw new ArgumentException ("socketAddress is not a unix socket address.");
            if (socketAddress [1] != ((addr & 0xFF00) >> 8))
                throw new ArgumentException ("socketAddress is not a unix socket address.");
             */

            if (socketAddress.Size == 2) {
                // Empty filename.
                // Probably from RemoteEndPoint which on linux does not return the file name.
                return new UnixDomainSocketEndPoint("a") { _filename = "" };
            }
            var size = socketAddress.Size - 2;
            var bytes = new byte[size];
            for (var i = 0; i < bytes.Length; i++) {
                bytes[i] = socketAddress[i + 2];
                // There may be junk after the null terminator, so ignore it all.
                if (bytes[i] == 0) {
                    size = i;
                    break;
                }
            }

            var name = Encoding.UTF8.GetString(bytes, 0, size);
            return new UnixDomainSocketEndPoint(name);
        }

        public override SocketAddress Serialize()
        {
            var bytes = Encoding.UTF8.GetBytes(_filename);
            var sa = new SocketAddress(AddressFamily, 2 + bytes.Length + 1);
            // sa [0] -> family low byte, sa [1] -> family high byte
            for (var i = 0; i < bytes.Length; i++)
                sa[2 + i] = bytes[i];

            //NULL suffix for non-abstract path
            sa[2 + bytes.Length] = 0;

            return sa;
        }

        public override string ToString() => _filename;

        public override int GetHashCode () => _filename.GetHashCode();

        public override bool Equals(object? o)
        {
            var other = o as UnixDomainSocketEndPoint;
            if (other == null)
                return false;

            return (other._filename == _filename);
        }
    }
}
#endif