P2P-NAT穿透之NAT类型检测客户端代码片段

kkylove
已于 2024-06-05 16:44:06 修改
阅读量131
点赞数 3
分类专栏： P2P-NAT穿透 TCP/UDP穿透文章标签： c# p2p 网络协议
于 2024-06-05 15:54:21 首次发布
本文链接：https://blog.csdn.net/kkylove/article/details/139474554
版权
P2P-NAT穿透同时被 2 个专栏收录
4 篇文章 0 订阅
订阅专栏
TCP/UDP穿透
3 篇文章 0 订阅
订阅专栏
一言不合，直接上代码，有需要的小伙伴可以加作者QQ号：541655940；QQ群：700600021
 using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using P2P.Client.Enums;
using P2P.Client.Model;
using P2P.Client.Tasks;
using P2P.Network.ExtensionMethods;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace P2P.Client.Handler
{
    public class NatCheckHandler
    {
        private static readonly object _lock = new object();
        private static bool _step1Success = false;

        private bool Step1Success 
        {
            get 
            {
                lock (_lock) 
                {
                    return _step1Success;
                }
            }
            set 
            {
                lock (_lock) 
                {
                   _step1Success = value;
                }
            }
        }
        
        private static bool _step3Success = false;

        private bool Step3Success
        {
            get
            {
                lock (_lock)
                {
                    return _step3Success;
                }
            }
            set
            {
                lock (_lock)
                {
                    _step3Success = value;
                }
            }
        }

        private static bool _step5Success = false;

        private bool Step5Success
        {
            get
            {
                lock (_lock)
                {
                    return _step5Success;
                }
            }
            set
            {
                lock (_lock)
                {
                    _step5Success = value;
                }
            }
        }

        private ApiProtocol apiProtocol;

        public NatCheckHandler(ApiProtocol apiProtocol)
        {
            this.apiProtocol = apiProtocol;
        }

        public void NatCheckFinished() 
        {
            int natType;
            int.TryParse(apiProtocol.Params["NatType"].ToString(), out natType);

            SharedModel.Instance.NatType = (NatType)natType;

            Step1Success = false;
            Step3Success = false;
            Step5Success = false;
        }

        public void Step1() 
        {
            Step1Success = true;
        }

        public void Step2()
        {
            int count = 60;
            Task.Run(async () => 
            {
                while (true) 
                {
                    if (Step1Success)
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step2");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    if ((--count) == 0 && !Step1Success) 
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step3");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    await Task.Delay(100);
                }
            });
        }

        public void Step3() 
        {
            Step3Success = true;
        }

        public void Step4() 
        {
            int count = 60;
            Task.Run(async () =>
            {
                while (true)
                {
                    if (Step3Success)
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step4");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    if ((--count) == 0 && !Step3Success)
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step5");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, SharedModel.Instance.Hosts.UDPMainServerSecondPublicIP,
                            SharedModel.Instance.Hosts.UDPMainServerSecondPublicPort.Value);

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    await Task.Delay(100);
                }
            });
        }

        public void Step5() 
        {
            Step5Success = true;
        }

        public void Step6()
        {
            int count = 60;
            Task.Run(async () =>
            {
                while (true)
                {
                    if (Step5Success)
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step6");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    if ((--count) == 0 && !Step5Success)
                    {
                        JObject data = new JObject();
                        data.Add("Name", "NatCheck");
                        data.Add("Method", "Step7");
                        data.Add("Version", "1.0");

                        JObject param = new JObject();
                        param.Add("AccountId", SharedModel.Instance.AccountId);
                        data.Add("Params", param);

                        string ack = JsonConvert.SerializeObject(R<JObject>.Ok(data));

                        apiProtocol.Socket.SendTo(ack, apiProtocol.RemoteEndPoint);
                        return;
                    }

                    await Task.Delay(100);
                }
            });
        }
    }
}