520 lines
11 KiB
C++
520 lines
11 KiB
C++
#include "Time.h"
|
||
|
||
#include "Net\Net.h"
|
||
#include "Net\DNS.h"
|
||
|
||
#include "TokenClient.h"
|
||
|
||
#include "TokenMessage.h"
|
||
#include "HelloMessage.h"
|
||
#include "LoginMessage.h"
|
||
#include "RegisterMessage.h"
|
||
|
||
#include "Security\MD5.h"
|
||
|
||
static bool OnTokenClientReceived(void* sender, Message& msg, void* param);
|
||
|
||
static void LoopTask(void* param);
|
||
|
||
TokenClient::TokenClient()
|
||
{
|
||
Token = 0;
|
||
|
||
Status = 0;
|
||
LoginTime = 0;
|
||
LastActive = 0;
|
||
Delay = 0;
|
||
|
||
Control = NULL;
|
||
|
||
Received = NULL;
|
||
Param = NULL;
|
||
|
||
Local = NULL;
|
||
}
|
||
|
||
void TokenClient::Open()
|
||
{
|
||
assert_param2(Control, "令牌客户端还没设置控制器呢");
|
||
|
||
Control->Received = OnTokenClientReceived;
|
||
Control->Param = this;
|
||
Control->Open();
|
||
|
||
auto ctrl = Control;
|
||
if(Local && Local != Control)
|
||
{
|
||
// 向服务端握手时，汇报内网本地端口，用户端将会通过该端口连接
|
||
ctrl = Local;
|
||
|
||
Local->Received = OnTokenClientReceived;
|
||
Local->Param = this;
|
||
Local->Open();
|
||
}
|
||
|
||
// 设置握手广播的本地地址和端口
|
||
auto sock = dynamic_cast<ISocket*>(ctrl->Port);
|
||
if(sock) Hello.EndPoint = sock->Local;
|
||
|
||
// 令牌客户端定时任务
|
||
_task = Sys.AddTask(LoopTask, this, 1000, 5000, "令牌客户端");
|
||
}
|
||
|
||
void TokenClient::Close()
|
||
{
|
||
Sys.RemoveTask(_task);
|
||
}
|
||
|
||
bool TokenClient::Send(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
// 未登录之前，只能 握手、登录、注册
|
||
if(Token == 0)
|
||
{
|
||
if(msg.Code != 0x01 && msg.Code != 0x02 && msg.Code != 0x07) return false;
|
||
}
|
||
|
||
if(!ctrl) ctrl = Control;
|
||
assert_param2(ctrl, "TokenClient::Send");
|
||
|
||
return ctrl->Send(msg);
|
||
}
|
||
|
||
bool TokenClient::Reply(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
// 未登录之前，只能 握手、登录、注册
|
||
if(Token == 0)
|
||
{
|
||
if(msg.Code != 0x01 && msg.Code != 0x02 && msg.Code != 0x07) return false;
|
||
}
|
||
|
||
if(!ctrl) ctrl = Control;
|
||
assert_param2(ctrl, "TokenClient::Reply");
|
||
|
||
return ctrl->Reply(msg);
|
||
}
|
||
|
||
bool TokenClient::OnReceive(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
LastActive = Sys.Ms();
|
||
|
||
switch(msg.Code)
|
||
{
|
||
case 0x01:
|
||
OnHello(msg, ctrl);
|
||
break;
|
||
case 0x02:
|
||
if(msg.Reply)
|
||
OnLogin(msg, ctrl);
|
||
else
|
||
Login(msg, ctrl);
|
||
break;
|
||
case 0x03:
|
||
OnPing(msg, ctrl);
|
||
break;
|
||
case 0x07:
|
||
OnRegister(msg, ctrl);
|
||
break;
|
||
}
|
||
|
||
// 消息转发
|
||
if(Received) Received(ctrl, msg, Param);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool OnTokenClientReceived(void* sender, Message& msg, void* param)
|
||
{
|
||
auto ctrl = (Controller*)sender;
|
||
assert_ptr(ctrl);
|
||
auto client = (TokenClient*)param;
|
||
assert_ptr(client);
|
||
|
||
return client->OnReceive((TokenMessage&)msg, ctrl);
|
||
}
|
||
|
||
// 常用系统级消息
|
||
|
||
// 定时任务
|
||
void LoopTask(void* param)
|
||
{
|
||
assert_ptr(param);
|
||
|
||
auto client = (TokenClient*)param;
|
||
//client->SayHello(false);
|
||
//if(client->Udp->BindPort != 3355)
|
||
// client->SayHello(true, 3355);
|
||
// 状态。0准备、1握手完成、2登录后
|
||
switch(client->Status)
|
||
{
|
||
case 0:
|
||
client->SayHello(false);
|
||
break;
|
||
case 1:
|
||
{
|
||
auto cfg = TokenConfig::Current;
|
||
|
||
if(cfg->User.Length() == 0)
|
||
client->Register();
|
||
else
|
||
client->Login();
|
||
|
||
break;
|
||
}
|
||
case 2:
|
||
client->Ping();
|
||
break;
|
||
}
|
||
}
|
||
|
||
// 发送发现消息，告诉大家我在这
|
||
// 请求：2版本 + S类型 + S名称 + 8本地时间 + 6本地IP端口 + S支持加密算法列表
|
||
// 响应：2版本 + S类型 + S名称 + 8本地时间 + 6对方IP端口 + 1加密算法 + N密钥
|
||
// 错误：0xFE + 1协议 + S服务器 + 2端口
|
||
void TokenClient::SayHello(bool broadcast, int port)
|
||
{
|
||
TokenMessage msg(0x01);
|
||
|
||
HelloMessage ext(Hello);
|
||
ext.Reply = false;
|
||
ext.LocalTime = Time.Now().TotalMicroseconds();
|
||
ext.WriteMessage(msg);
|
||
ext.Show(true);
|
||
|
||
// 广播消息直接用UDP发出
|
||
/*if(broadcast)
|
||
{
|
||
if(!Udp) return;
|
||
|
||
// 临时修改远程地址为广播地址
|
||
IPEndPoint ep = Udp->Remote;
|
||
Udp->Remote.Address = IPAddress::Broadcast;
|
||
Udp->Remote.Port = port;
|
||
|
||
debug_printf("握手广播 ");
|
||
Control->Send(msg);
|
||
|
||
// 还原回来原来的地址端口
|
||
Udp->Remote = ep;
|
||
|
||
return;
|
||
}*/
|
||
|
||
Send(msg);
|
||
}
|
||
|
||
// 握手响应
|
||
bool TokenClient::OnHello(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
// 解析数据
|
||
HelloMessage ext;
|
||
ext.Reply = msg.Reply;
|
||
|
||
ext.ReadMessage(msg);
|
||
ext.Show(true);
|
||
|
||
// 如果收到响应，并且来自来源服务器
|
||
if(msg.Reply)
|
||
{
|
||
if(msg.Error)
|
||
{
|
||
if(OnRedirect(ext)) return false;
|
||
|
||
Status = 0;
|
||
Token = 0;
|
||
debug_printf("握手失败，错误码=0x%02X ", ext.ErrCode);
|
||
|
||
ext.ErrMsg.Show(true);
|
||
}
|
||
else
|
||
{
|
||
// 通讯密码
|
||
if(ext.Key.Length() > 0)
|
||
{
|
||
auto ctrl2 = dynamic_cast<TokenController*>(ctrl);
|
||
if(ctrl2) ctrl2->Key = ext.Key;
|
||
|
||
debug_printf("握手得到通信密码：");
|
||
ext.Key.Show(true);
|
||
}
|
||
Status = 1;
|
||
|
||
// 握手完成后马上注册或登录
|
||
Sys.SetTask(_task, true, 0);
|
||
|
||
// 同步本地时间
|
||
if(ext.LocalTime > 0) ((TTime&)Time).SetTime(ext.LocalTime / 1000000UL);
|
||
}
|
||
}
|
||
else if(!msg.Reply)
|
||
{
|
||
TokenMessage rs;
|
||
rs.Code = msg.Code;
|
||
|
||
HelloMessage ext2(Hello);
|
||
ext2.Reply = true;
|
||
// 使用系统ID作为Name
|
||
ext2.Name = TokenConfig::Current->User;
|
||
// 使用系统ID作为Key
|
||
//ext2.Key.Copy(0, Sys.ID, 16);
|
||
ext2.Key = Sys.ID;
|
||
//auto ctrl3 = dynamic_cast<TokenController*>(ctrl);
|
||
//if(ctrl3) ctrl3->Key = ext2.Key;
|
||
|
||
ext2.Ciphers[0] = 0xFF;
|
||
//ext2.LocalTime = ext.LocalTime;
|
||
// 使用当前时间
|
||
ext2.LocalTime = Time.Now().TotalMicroseconds();
|
||
ext2.WriteMessage(rs);
|
||
|
||
// 源地址发回去
|
||
rs.State = msg.State;
|
||
|
||
Reply(rs);
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
bool TokenClient::OnRedirect(HelloMessage& msg)
|
||
{
|
||
if(!(msg.ErrCode == 0xFE || msg.ErrCode == 0xFD)) return false;
|
||
|
||
auto cfg = TokenConfig::Current;
|
||
cfg->Protocol = (ProtocolType)msg.Protocol;
|
||
|
||
cfg->Show();
|
||
|
||
/*uint len1 = ArrayLength(cfg->Server);
|
||
uint mslen1 = msg.Server.Length();
|
||
if(mslen1 > len1)
|
||
{
|
||
debug_printf("服务器地址超长 Max=%d Server=%s \r\n", len1, msg.Server.GetBuffer());
|
||
return false;
|
||
}*/
|
||
msg.Server = cfg->Server;
|
||
//cfg->Server[mslen1]= '\0';
|
||
|
||
cfg->ServerPort = msg.Port;
|
||
|
||
/*uint len2 = ArrayLength(cfg->VisitToken);
|
||
uint mslen2 = msg.VisitToken.Length();
|
||
if(mslen2> len2)
|
||
{
|
||
debug_printf("访问令牌超长 Max=%d VisitToken=%s \r\n", len2, msg.VisitToken.GetBuffer());
|
||
return false;
|
||
}*/
|
||
msg.VisitToken = cfg->VisitToken;
|
||
//cfg->VisitToken[mslen2] = '\0';
|
||
|
||
cfg->Show();
|
||
//msg.Server.CopyTo(cfg->Vendor, 0, 0);
|
||
// 0xFE永久改变厂商地址
|
||
if(msg.ErrCode == 0xFE)
|
||
{
|
||
cfg->Save();
|
||
//Sys.Reset();
|
||
|
||
//return true;
|
||
}
|
||
ChangeIPEndPoint(msg.Server, msg.Port);
|
||
Status = 0;
|
||
//auto flg = ChangeIPEndPoint(msg.Server,msg.Port);
|
||
//cfg->Save();
|
||
//Sys.Reset();
|
||
//if(!flg) Sys.Reset();
|
||
|
||
return true;
|
||
}
|
||
|
||
// 注册
|
||
void TokenClient::Register()
|
||
{
|
||
debug_printf("TokenClient::Register\r\n");
|
||
|
||
RegisterMessage re;
|
||
re.User = ByteArray((const void*)Sys.ID, 16).ToHex(0, 0);
|
||
re.Show(true);
|
||
|
||
TokenMessage msg(7);
|
||
re.WriteMessage(msg);
|
||
|
||
Send(msg);
|
||
}
|
||
|
||
void TokenClient::OnRegister(TokenMessage& msg ,Controller* ctrl)
|
||
{
|
||
if(!msg.Reply || msg.Error) return;
|
||
|
||
auto cfg = TokenConfig::Current;
|
||
|
||
RegisterMessage rm;
|
||
rm.ReadMessage(msg);
|
||
rm.User = cfg->User;
|
||
rm.Pass = cfg->Pass;
|
||
|
||
cfg->Show();
|
||
cfg->Save();
|
||
//cfg->Show();
|
||
|
||
//Sys.Reset();
|
||
Status = 0;
|
||
}
|
||
|
||
// 登录
|
||
void TokenClient::Login()
|
||
{
|
||
LoginMessage login;
|
||
|
||
auto cfg = TokenConfig::Current;
|
||
login.User = cfg->User;
|
||
//login.Key = cfg->Key;
|
||
//login.Pass = MD5::Hash(Array(cfg->Pass, ArrayLength(cfg->Pass))).ToHex(0, 0);
|
||
//login.Pass = MD5::Hash(String(cfg->Pass, ArrayLength(cfg->Pass)));
|
||
login.Pass = MD5::Hash(cfg->Pass);
|
||
/*// 临时代码，兼容旧云端
|
||
if(login.User.Length() < 4)
|
||
{
|
||
//Register();
|
||
login.User.Copy(Sys.ID, 16);
|
||
login.Pass.Copy(Sys.ID, 16);
|
||
}*/
|
||
TokenMessage msg(2);
|
||
login.WriteMessage(msg);
|
||
login.Show(true);
|
||
|
||
Send(msg);
|
||
}
|
||
|
||
void TokenClient::Login(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
if(msg.Error) return;
|
||
|
||
LoginMessage login;
|
||
// 这里需要随机密匙
|
||
//login.Key = Key.Copy(Sys.ID, 16);
|
||
// 随机令牌
|
||
login.Token = 123456;
|
||
login.Reply = true;
|
||
login.WriteMessage(msg);
|
||
|
||
Reply(msg);
|
||
|
||
auto ctrl2 = dynamic_cast<TokenController*>(ctrl);
|
||
ctrl2->Key = login.User;
|
||
ctrl2->Token = login.Token;
|
||
}
|
||
|
||
bool TokenClient::OnLogin(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
if(!msg.Reply) return false;
|
||
|
||
Stream ms(msg.Data, msg.Length);
|
||
|
||
if(msg.Error)
|
||
{
|
||
// 登录失败，清空令牌
|
||
Token = 0;
|
||
|
||
byte result = ms.ReadByte();
|
||
// 任何错误，重新握手
|
||
Status = 0;
|
||
|
||
debug_printf("登录失败，错误码 0x%02X！", result);
|
||
ms.ReadString().Show(true);
|
||
}
|
||
else
|
||
{
|
||
Status = 2;
|
||
debug_printf("登录成功！ ");
|
||
|
||
// 得到令牌
|
||
Token = ms.ReadUInt32();
|
||
debug_printf("令牌：0x%08X ", Token);
|
||
// 这里可能有通信密码
|
||
if(ms.Remain() > 0)
|
||
{
|
||
auto bs = ms.ReadArray();
|
||
if(bs.Length() > 0)
|
||
{
|
||
auto ctrl2 = dynamic_cast<TokenController*>(ctrl);
|
||
if(ctrl2) ctrl2->Key = bs;
|
||
|
||
debug_printf("通信密码：");
|
||
bs.Show();
|
||
}
|
||
}
|
||
debug_printf("\r\n");
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
// Ping指令用于保持与对方的活动状态
|
||
void TokenClient::Ping()
|
||
{
|
||
if(LastActive > 0 && LastActive + 30000 < Sys.Ms())
|
||
{
|
||
// 30秒无法联系，服务端可能已经掉线，重启Hello任务
|
||
debug_printf("30秒无法联系，服务端可能已经掉线，重新开始握手\r\n");
|
||
|
||
Status = 0;
|
||
|
||
return;
|
||
}
|
||
|
||
TokenMessage msg(3);
|
||
|
||
UInt64 time = Sys.Ms();
|
||
Stream ms = msg.ToStream();
|
||
ms.WriteArray(Array(&time, 8));
|
||
msg.Length = ms.Position();
|
||
|
||
Send(msg);
|
||
}
|
||
|
||
bool TokenClient::OnPing(TokenMessage& msg, Controller* ctrl)
|
||
{
|
||
// 忽略
|
||
if(!msg.Reply) return Reply(msg);
|
||
|
||
Stream ms = msg.ToStream();
|
||
|
||
UInt64 now = Sys.Ms();
|
||
UInt64 start = ms.ReadArray().ToUInt64();
|
||
int cost = (int)(now - start);
|
||
if(cost < 0) cost = -cost;
|
||
if(cost > 1000) ((TTime&)Time).SetTime(start / 1000);
|
||
|
||
if(Delay)
|
||
Delay = (Delay + cost) / 2;
|
||
else
|
||
Delay = cost;
|
||
|
||
debug_printf("心跳延迟 %dms / %dms \r\n", cost, Delay);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool TokenClient::ChangeIPEndPoint(const String& domain, ushort port)
|
||
{
|
||
debug_printf("ChangeIPEndPoint ");
|
||
|
||
domain.Show(true);
|
||
|
||
auto socket = dynamic_cast<ISocket*>(Control->Port);
|
||
if(socket == NULL) return false;
|
||
|
||
// 根据DNS获取云端IP地址
|
||
auto ip = DNS::Query(*(socket->Host), domain);
|
||
if(ip == IPAddress::Any()) return false;
|
||
|
||
debug_printf("服务器地址 %s %s:%d \r\n", domain.GetBuffer(), ip.ToString().GetBuffer(), port);
|
||
|
||
Control->Port->Close();
|
||
socket->Remote.Address = ip;
|
||
socket->Remote.Port = port;
|
||
Control->Port->Open();
|
||
|
||
return true;
|
||
}
|