using System;
using System.Buffers.Binary;
using System.Net;
using System.Runtime.InteropServices;

namespace linker.tun.test
{
    internal class Program
    {
        public static LinkerTunDeviceAdapter linkerTunDeviceAdapter;
        static void Main(string[] args)
        {

            linkerTunDeviceAdapter = new LinkerTunDeviceAdapter(new LinkerTunDeviceCallback());
            linkerTunDeviceAdapter.SetUp("linker111"
                , Guid.Parse("dc6d4efa-2b53-41bd-a403-f416c9bf7129")
                , IPAddress.Parse("192.168.55.2")
                , IPAddress.Parse("192.168.55.1"), 24);

            if (string.IsNullOrWhiteSpace(linkerTunDeviceAdapter.Error))
            {
                Console.WriteLine(linkerTunDeviceAdapter.Error);
            }

            Console.ReadLine();
        }
    }

    public unsafe sealed class LinkerTunDeviceCallback : ILinkerTunDeviceCallback
    {
        public void Callback(LinkerTunDevicPacket packet)
        {
            Memory<byte> writableMemory = MemoryMarshal.AsMemory(packet.Packet);
            fixed (byte* ptr = writableMemory.Span)
            {
                //icmp && request
                if (ptr[9] == 1 && ptr[20] == 8)
                {
                    Console.WriteLine($"ICMP to {new IPAddress(writableMemory.Span.Slice(16,4))}");

                    uint dist = BinaryPrimitives.ReadUInt32LittleEndian(writableMemory.Span.Slice(16, 4));

                    //目的地址变源地址，
                    * (uint*)(ptr + 16) = *(uint*)(ptr + 12);
                    //假装是网关回复的
                    *(uint*)(ptr + 12) = dist;

                    //计算一次IP头校验和
                    *(ushort*)(ptr + 10) = 0;
                    *(ushort*)(ptr + 10) = Program.linkerTunDeviceAdapter.Checksum((ushort*)ptr, 20);

                    //response
                    *(ushort*)(ptr + 20) = 0;

                    //计算ICMP校验和
                    *(ushort*)(ptr + 22) = 0;
                    *(ushort*)(ptr + 22) = Program.linkerTunDeviceAdapter.Checksum((ushort*)(ptr + 20), (uint)(writableMemory.Length - 20));

                    Program.linkerTunDeviceAdapter.Write(writableMemory);
                }
            }
        }
    }
}