-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathTcpSession.cs
79 lines (70 loc) · 2.11 KB
/
TcpSession.cs
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
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using SuperSocket.SocketBase;
using TLSHandler.Enums;
using TLSHandler.Handler;
using TLSHandler.Internal.TLS.Records;
namespace Https
{
public class TcpSession : AppSession<TcpSession, Request.TLSRequest>
{
public Context TLSContext { get; internal set; }
object _lock = new object();
public void Receive(TLSRecord record)
{
lock(_lock)
{
var response = this.TLSContext.Process_Record(record);
this.Send(response);
}
}
public void Send(Result resp)
{
if (resp != null)
{
if (resp is PacketResult hr)
{
this.Send(hr.Response);
}
else if (resp is AlertResult ar)
{
LogHelper.Error(this, ar);
if (ar.ShouldTerminate)
this.Close(CloseReason.ApplicationError);
return;
}
}
}
public void Send(IEnumerable<TLSRecord> tls)
{
var recordsBytes = new List<byte>();
foreach (var pkt in tls)
{
LogHelper.Debug(this, $"SEND TLS {pkt.Type}");
recordsBytes.AddRange(pkt.Data);
}
Send(recordsBytes.ToArray(), 0, recordsBytes.Count);
}
public void SendApplicationData(byte[] data)
{
var resp = this.TLSContext.GetEncryptedPacket(data);
if (resp != null)
{
if (resp is PacketResult hr)
{
this.Send(hr.Response);
}
else if (resp is AlertResult ar)
{
LogHelper.Error(this, ar);
if (ar.ShouldTerminate)
this.Close(CloseReason.ApplicationError);
return;
}
}
}
}
}