(SERVER) PARITY PROJECT TBA

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;

namespace TWRC.Utility.Hooks
{
    public class Connect : HookBase
    {

        [StructLayoutAttribute(LayoutKind.Sequential, CharSet = CharSet.Ansi)]
        public struct sockaddr_in
        {
            public short sin_family;
            public ushort sin_port;
            public in_addr sin_addr;
            [MarshalAsAttribute(UnmanagedType.ByValTStr, SizeConst = 8)]
            public string sin_zero;
        }

        [StructLayoutAttribute(LayoutKind.Sequential)]
        public struct in_addr
        {
            public Anonymous1 S_un;
        }

        [StructLayoutAttribute(LayoutKind.Explicit)]
        public struct Anonymous1
        {
            [FieldOffsetAttribute(0)]
            public Anonymous2 S_un_b;

            [FieldOffsetAttribute(0)]
            public Anonymous3 S_un_w;

            [FieldOffsetAttribute(0)]
            public uint S_addr;
        }

        [StructLayoutAttribute(LayoutKind.Sequential)]
        public struct Anonymous2
        {
            public byte s_b1;
            public byte s_b2;
            public byte s_b3;
            public byte s_b4;
        }

        [StructLayoutAttribute(LayoutKind.Sequential)]
        public struct Anonymous3
        {
            public ushort s_w1;
            public ushort s_w2;
        }

        [UnmanagedFunctionPointer(CallingConvention.StdCall,
            CharSet = CharSet.Ansi,
            SetLastError = true)]
        private delegate int Dconnect(IntPtr s, ref sockaddr_in addr, int addrsize);

        [DllImport("Ws2_32.dll")]
        private static extern int connect(IntPtr s, ref sockaddr_in addr, int addrsize);

        public Connect()
            : base((IntPtr)EasyHook.LocalHook.GetProcAddress("ws2_32.dll", "connect"))
        {
            this.SetDelegate(new Dconnect(_connect));
        }

        private int _connect(IntPtr s, ref sockaddr_in addr, int addrsize)
        {
            IPEndPoint ep = new IPEndPoint(new IPAddress(addr.sin_addr.S_un.S_addr), IPAddress.NetworkToHostOrder((short)addr.sin_port));
            if(ep.Port == 5330) // redirect to local one
            {
                    ep.Address = IPAddress.Parse(Utility.GetInstance().Authentication.ServerAddress);
                    addr.sin_addr.S_un.S_addr = (uint)ep.Address.Address;
                    addr.sin_port = (ushort)IPAddress.HostToNetworkOrder((short)ep.Port);
            }
            Console.WriteLine("Connect hook: {0} ({1})", ep, addr.sin_addr.S_un.S_addr);
            return connect(s, ref addr, addrsize);
        }

    }
}