1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
using Ryujinx.Common.Logging;
using Ryujinx.HLE.HOS.Services.Ldn.Types;
using System.Net;
using System.Net.Sockets;
namespace Ryujinx.HLE.HOS.Services.Ldn.UserServiceCreator.LdnMitm.Proxy
{
internal class LdnProxyTcpSession : NetCoreServer.TcpSession
{
private readonly LanProtocol _protocol;
internal int NodeId;
internal NodeInfo NodeInfo;
private byte[] _buffer;
private int _bufferEnd;
public LdnProxyTcpSession(LdnProxyTcpServer server, LanProtocol protocol) : base(server)
{
_protocol = protocol;
_protocol.Connect += OnConnect;
_buffer = new byte[LanProtocol.BufferSize];
OptionSendBufferSize = LanProtocol.TcpTxBufferSize;
OptionReceiveBufferSize = LanProtocol.TcpRxBufferSize;
OptionSendBufferLimit = LanProtocol.TxBufferSizeMax;
OptionReceiveBufferLimit = LanProtocol.RxBufferSizeMax;
}
public void OverrideInfo()
{
NodeInfo.NodeId = (byte)NodeId;
NodeInfo.IsConnected = (byte)(IsConnected ? 1 : 0);
}
protected override void OnConnected()
{
Logger.Info?.PrintMsg(LogClass.ServiceLdn, "LdnProxyTCPSession connected!");
}
protected override void OnDisconnected()
{
Logger.Info?.PrintMsg(LogClass.ServiceLdn, "LdnProxyTCPSession disconnected!");
_protocol.InvokeDisconnectStation(this);
}
protected override void OnReceived(byte[] buffer, long offset, long size)
{
_protocol.Read(ref _buffer, ref _bufferEnd, buffer, (int)offset, (int)size, this.Socket.RemoteEndPoint);
}
protected override void OnError(SocketError error)
{
Logger.Error?.PrintMsg(LogClass.ServiceLdn, $"LdnProxyTCPSession caught an error with code {error}");
Dispose();
}
protected override void Dispose(bool disposingManagedResources)
{
_protocol.Connect -= OnConnect;
base.Dispose(disposingManagedResources);
}
private void OnConnect(NodeInfo info, EndPoint endPoint)
{
try
{
if (endPoint.Equals(this.Socket.RemoteEndPoint))
{
NodeInfo = info;
_protocol.InvokeAccept(this);
}
}
catch (System.ObjectDisposedException)
{
Logger.Error?.PrintMsg(LogClass.ServiceLdn, $"LdnProxyTCPSession was disposed. [IP: {NodeInfo.Ipv4Address}]");
_protocol.InvokeDisconnectStation(this);
}
}
}
}
|
