davejansen/lidgren-network-gen3
1
0
Fork 0
mirror of https://github.com/lidgren/lidgren-network-gen3.git synced 2026-05-16 07:06:30 +09:00
Code Issues Releases Activity

NAT intro work

Browse Source
This commit is contained in:
lidgren
2010-05-10 16:23:08 +00:00
parent 66802928b6
commit 51a18d5957
5 changed files with 103 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
Lidgren.Network/NetNatIntroduction.cs
View File

@@ -6,44 +6,94 @@ namespace Lidgren.Network
{
public partial class NetPeer
{
public void Introduce(IPEndPoint host, IPEndPoint client)
public void Introduce(
IPEndPoint hostInternal,
IPEndPoint hostExternal,
IPEndPoint clientInternal,
IPEndPoint clientExternal,
string token)
{
// send message to client
NetOutgoingMessage msg = CreateMessage(10);
NetOutgoingMessage msg = CreateMessage(10 + token.Length + 1);
msg.Write(false);
msg.WritePadBits();
msg.Write(host);
SendUnconnectedLibraryMessage(msg, NetMessageLibraryType.NatIntroduction, client);
msg.Write(hostInternal);
msg.Write(hostExternal);
msg.Write(token);
SendUnconnectedLibraryMessage(msg, NetMessageLibraryType.NatIntroduction, clientExternal);
// send message to host
msg = CreateMessage(10);
msg = CreateMessage(10 + token.Length + 1);
msg.Write(true);
msg.WritePadBits();
msg.Write(client);
SendUnconnectedLibraryMessage(msg, NetMessageLibraryType.NatIntroduction, host);
msg.Write(clientInternal);
msg.Write(clientExternal);
msg.Write(token);
SendUnconnectedLibraryMessage(msg, NetMessageLibraryType.NatIntroduction, hostExternal);
}
private void HandleNatIntroduction(int ptr, IPEndPoint senderEndpoint)
/// <summary>
/// Called when host/client receives a NatIntroduction message from a master server
/// </summary>
private void HandleNatIntroduction(int ptr)
{
VerifyNetworkThread();
// read intro
NetIncomingMessage tmp = new NetIncomingMessage(m_receiveBuffer, 1000); // never mind length
tmp.Position = (ptr * 8);
bool isHost = (tmp.ReadByte() == 0 ? false : true);
IPEndPoint ep = tmp.ReadIPEndpoint();
byte hostByte = tmp.ReadByte();
IPEndPoint remoteInternal = tmp.ReadIPEndpoint();
IPEndPoint remoteExternal = tmp.ReadIPEndpoint();
string token = tmp.ReadString();
bool isHost = (hostByte != 0);
// quickly; send nat punch
NetOutgoingMessage punch = CreateMessage(0);
SendUnconnectedLibraryMessage(punch, NetMessageLibraryType.NatPunchMessage, ep);
NetOutgoingMessage punch;
if (!isHost)
if (!isHost && m_configuration.IsMessageTypeEnabled(NetIncomingMessageType.NatIntroductionSuccess) == false)
return; // no need to punch - we're not listening for nat intros!
// send internal punch
punch = CreateMessage(1);
punch.Write(hostByte);
if (hostByte == 0)
{
NetIncomingMessage intro = CreateIncomingMessage(NetIncomingMessageType.NatIntroduction, 10);
intro.Write(ep);
intro.m_senderEndpoint = senderEndpoint;
ReleaseMessage(intro);
// only client needs to send token
punch.Write(token);
}
SendUnconnectedLibraryMessage(punch, NetMessageLibraryType.NatPunchMessage, remoteInternal);
// send external punch
punch = CreateMessage(1);
punch.Write(hostByte);
if (hostByte == 0)
{
// only client needs to send token
punch.Write(token);
}
SendUnconnectedLibraryMessage(punch, NetMessageLibraryType.NatPunchMessage, remoteExternal);
}
/// <summary>
/// Called when receiving a NatPunchMessage from a remote endpoint
/// </summary>
private void HandleNatPunch(int ptr, IPEndPoint senderEndpoint)
{
NetIncomingMessage tmp = new NetIncomingMessage(m_receiveBuffer, 1000); // never mind length
tmp.Position = (ptr * 8);
byte hostByte = tmp.ReadByte();
if (hostByte != 0)
return; // don't alert hosts about nat punch successes; only clients
string token = tmp.ReadString();
//
// Release punch success to client; enabling him to Connect() to msg.SenderIPEndPoint if token is ok
//
NetIncomingMessage punchSuccess = CreateIncomingMessage(NetIncomingMessageType.NatIntroductionSuccess, 10);
punchSuccess.m_senderEndpoint = senderEndpoint;
punchSuccess.Write(token);
ReleaseMessage(punchSuccess);
}
}
}