NetworkAuthenticationServerBase.cs
5.77 KB
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
namespace HslCommunication.Core.Net
{
/// <summary>
/// 带登录认证的服务器类
/// </summary>
public class NetworkAuthenticationServerBase : NetworkServerBase, IDisposable
{
/// <summary>
/// 当客户端的socket登录的时候额外检查的信息
/// </summary>
/// <param name="socket">套接字</param>
/// <param name="endPoint">终结点</param>
/// <returns>验证的结果</returns>
protected override OperateResult SocketAcceptExtraCheck( Socket socket, IPEndPoint endPoint )
{
if (IsUseAccountCertificate)
{
OperateResult<byte[], byte[]> receive = ReceiveAndCheckBytes( socket, 2000 );
if (!receive.IsSuccess) return new OperateResult( string.Format( "Client login failed[{0}]", endPoint ) );
if (BitConverter.ToInt32( receive.Content1, 0 ) != HslProtocol.ProtocolAccountLogin)
{
LogNet?.WriteError( ToString( ), StringResources.Language.NetClientAccountTimeout );
socket?.Close( );
return new OperateResult( string.Format( "Client login failed[{0}]", endPoint ) );
}
string[] infos = HslProtocol.UnPackStringArrayFromByte( receive.Content2 );
string ret = CheckAccountLegal( infos );
SendStringAndCheckReceive( socket, ret == "success" ? 1 : 0, new string[] { ret } );
if (ret != "success")
{
return new OperateResult( string.Format( "Client login failed[{0}]:{1}", endPoint, ret ) );
}
LogNet?.WriteDebug( ToString( ), string.Format( "Account Login:{0} Endpoint:[{1}]", infos[0], endPoint ) );
}
return OperateResult.CreateSuccessResult( );
}
#region Account Certification
/// <summary>
/// 获取或设置是否对客户端启动账号认证
/// </summary>
public bool IsUseAccountCertificate { get; set; }
private Dictionary<string, string> accounts = new Dictionary<string, string>( );
private SimpleHybirdLock lockLoginAccount = new SimpleHybirdLock( );
/// <summary>
/// 新增账户，如果想要启动账户登录，比如将<see cref="IsUseAccountCertificate"/>设置为<c>True</c>。
/// </summary>
/// <param name="userName">账户名称</param>
/// <param name="password">账户名称</param>
public void AddAccount( string userName, string password )
{
if (!string.IsNullOrEmpty( userName ))
{
lockLoginAccount.Enter( );
if (accounts.ContainsKey( userName ))
{
accounts[userName] = password;
}
else
{
accounts.Add( userName, password );
}
lockLoginAccount.Leave( );
}
}
/// <summary>
/// 删除一个账户的信息
/// </summary>
/// <param name="userName">账户名称</param>
public void DeleteAccount( string userName )
{
lockLoginAccount.Enter( );
if (accounts.ContainsKey( userName ))
{
accounts.Remove( userName );
}
lockLoginAccount.Leave( );
}
private string CheckAccountLegal( string[] infos )
{
if (infos?.Length < 2) return "User Name input wrong";
string ret = "";
lockLoginAccount.Enter( );
if (!accounts.ContainsKey( infos[0] ))
{
ret = "User Name input wrong";
}
else
{
if (accounts[infos[0]] != infos[1])
{
ret = "Password is not corrent";
}
else
{
ret = "success";
}
}
lockLoginAccount.Leave( );
return ret;
}
#endregion
#region IDisposable Support
private bool disposedValue = false; // 要检测冗余调用
/// <summary>
/// 释放当前的对象
/// </summary>
/// <param name="disposing">是否托管对象</param>
protected virtual void Dispose( bool disposing )
{
if (!disposedValue)
{
if (disposing)
{
ServerClose( );
lockLoginAccount?.Dispose( );
// TODO: 释放托管状态(托管对象)。
}
// TODO: 释放未托管的资源(未托管的对象)并在以下内容中替代终结器。
// TODO: 将大型字段设置为 null。
disposedValue = true;
}
}
// TODO: 仅当以上 Dispose(bool disposing) 拥有用于释放未托管资源的代码时才替代终结器。
// ~NetworkDataServerBase()
// {
// // 请勿更改此代码。将清理代码放入以上 Dispose(bool disposing) 中。
// Dispose(false);
// }
// 添加此代码以正确实现可处置模式。
/// <summary>
/// 释放当前的对象
/// </summary>
public void Dispose( )
{
// 请勿更改此代码。将清理代码放入以上 Dispose(bool disposing) 中。
Dispose( true );
// TODO: 如果在以上内容中替代了终结器，则取消注释以下行。
// GC.SuppressFinalize(this);
}
#endregion
}
}