336 lines
6.7 KiB
C++
336 lines
6.7 KiB
C++
#include "Token.h"
|
||
|
||
#include "SerialPort.h"
|
||
#include "WatchDog.h"
|
||
#include "Config.h"
|
||
|
||
#include "Drivers\NRF24L01.h"
|
||
#include "Drivers\W5500.h"
|
||
#include "Drivers\ShunCom.h"
|
||
|
||
#include "Net\Dhcp.h"
|
||
#include "Net\DNS.h"
|
||
|
||
#include "TokenNet\Gateway.h"
|
||
#include "TokenNet\Token.h"
|
||
|
||
#include "Security\RC4.h"
|
||
|
||
#include "App\FlushPort.h"
|
||
|
||
#define ShunCom_Master 0
|
||
|
||
static void StartGateway(void* param);
|
||
|
||
static void OnDhcpStop5500(void* sender, void* param)
|
||
{
|
||
auto dhcp = (Dhcp*)sender;
|
||
if(!dhcp->Result)
|
||
{
|
||
// 失败后重新开始DHCP，等待网络连接
|
||
dhcp->Start();
|
||
|
||
return;
|
||
}
|
||
|
||
// 获取IP成功，重新设置参数
|
||
auto net = (W5500*)dhcp->Host;
|
||
net->Config();
|
||
net->ShowInfo();
|
||
net->SaveConfig();
|
||
|
||
if(dhcp->Times <= 1) Sys.AddTask(StartGateway, net, 0, -1, "启动网关");
|
||
}
|
||
|
||
ISocketHost* Token::CreateW5500(SPI_TypeDef* spi_, Pin irq, Pin rst, Pin power, IDataPort* led)
|
||
{
|
||
debug_printf("\r\nW5500::Create \r\n");
|
||
|
||
static Spi spi(spi_, 36000000);
|
||
|
||
debug_printf("\tPower: ");
|
||
static OutputPort pwr(power, true);
|
||
pwr = true;
|
||
|
||
static W5500 net;
|
||
net.LoadConfig();
|
||
net.Init(&spi, irq, rst);
|
||
net.Led = led;
|
||
|
||
// 打开DHCP
|
||
static UdpClient udp(&net);
|
||
static Dhcp dhcp(&udp);
|
||
dhcp.OnStop = OnDhcpStop5500;
|
||
dhcp.Start();
|
||
|
||
return &net;
|
||
}
|
||
|
||
ISocket* CreateW5500UDP(ISocketHost* host, TokenConfig* tk)
|
||
{
|
||
auto udp = new UdpClient((W5500*)host);
|
||
//udp->Local.Port = tk->Port;
|
||
udp->Remote.Port = tk->ServerPort;
|
||
udp->Remote.Address = IPAddress(tk->ServerIP);
|
||
|
||
return udp;
|
||
}
|
||
|
||
ISocket* CreateW5500TCP(ISocketHost* host, TokenConfig* tk)
|
||
{
|
||
auto tcp = new TcpClient((W5500*)host);
|
||
//tcp->Local.Port = tk->Port;
|
||
tcp->Remote.Port = tk->ServerPort;
|
||
tcp->Remote.Address = IPAddress(tk->ServerIP);
|
||
|
||
return tcp;
|
||
}
|
||
|
||
TokenClient* Token::CreateClient(ISocketHost* host)
|
||
{
|
||
debug_printf("\r\nCreateClient \r\n");
|
||
|
||
static TokenController token;
|
||
|
||
auto tk = TokenConfig::Current;
|
||
ISocket* socket = NULL;
|
||
if(tk->Protocol == 2)
|
||
socket = CreateW5500UDP(host, tk);
|
||
else if(tk->Protocol == 1)
|
||
socket = CreateW5500TCP(host, tk);
|
||
token.Port = dynamic_cast<ITransport*>(socket);
|
||
|
||
static TokenClient client;
|
||
client.Control = &token;
|
||
//client->Local = token;
|
||
|
||
// 如果是TCP，需要再建立一个本地UDP
|
||
//if(tk->Protocol == 1)
|
||
{
|
||
TokenConfig tc;
|
||
//tc.Port = tk->Port;
|
||
tc.ServerIP = IPAddress::Broadcast().Value;
|
||
tc.ServerPort = 3355; // 广播端口。其实用哪一个都不重要，因为不会主动广播
|
||
|
||
socket = CreateW5500UDP(host, &tc);
|
||
socket->Local.Port = tk->Port;
|
||
auto token2 = new TokenController();
|
||
token2->Port = dynamic_cast<ITransport*>(socket);
|
||
client.Local = token2;
|
||
}
|
||
|
||
return &client;
|
||
}
|
||
|
||
TinyServer* Token::CreateServer(ITransport* port)
|
||
{
|
||
debug_printf("CreateServer \r\n");
|
||
|
||
static TinyController ctrl;
|
||
ctrl.Port = port;
|
||
ctrl.QueueLength = 64;
|
||
ctrl.ApplyConfig();
|
||
|
||
// 新配置需要保存一下
|
||
auto tc = TinyConfig::Current;
|
||
if(tc && tc->New) tc->Save();
|
||
|
||
static TinyServer server(&ctrl);
|
||
server.Cfg = tc;
|
||
|
||
return &server;
|
||
}
|
||
|
||
uint OnSerial(ITransport* transport, Array& bs, void* param, void* param2)
|
||
{
|
||
debug_printf("OnSerial len=%d \t", bs.Length());
|
||
bs.Show(true);
|
||
|
||
//TokenClient* client = TokenClient::Current;
|
||
//if(client) client->Store.Write(1, bs);
|
||
|
||
return 0;
|
||
}
|
||
|
||
void Token::Setup(ushort code, const char* name, COM_Def message, int baudRate)
|
||
{
|
||
Sys.Code = code;
|
||
Sys.Name = (char*)name;
|
||
|
||
// 初始化系统
|
||
//Sys.Clock = 48000000;
|
||
Sys.Init();
|
||
#if DEBUG
|
||
Sys.MessagePort = message; // 指定printf输出的串口
|
||
Sys.ShowInfo();
|
||
#endif
|
||
|
||
#if DEBUG
|
||
// 设置一定量初始任务，减少堆分配
|
||
static Task ts[0x10];
|
||
Task::Scheduler()->Set(ts, ArrayLength(ts));
|
||
#endif
|
||
|
||
#if DEBUG
|
||
// 打开串口输入便于调试数据操作，但是会影响性能
|
||
if(baudRate > 0)
|
||
{
|
||
auto sp = SerialPort::GetMessagePort();
|
||
if(baudRate != 1024000)
|
||
{
|
||
sp->Close();
|
||
sp->SetBaudRate(baudRate);
|
||
}
|
||
sp->Register(OnSerial);
|
||
}
|
||
|
||
//WatchDog::Start(20000);
|
||
#else
|
||
WatchDog::Start();
|
||
#endif
|
||
|
||
// Flash最后一块作为配置区
|
||
Config::Current = &Config::CreateFlash();
|
||
}
|
||
|
||
void Fix2401(void* param)
|
||
{
|
||
auto& bs = *(Array*)param;
|
||
// 微网指令特殊处理长度
|
||
uint rs = bs.Length();
|
||
if(rs >= 8)
|
||
{
|
||
rs = bs[5] + 8;
|
||
if(rs < bs.Length()) bs.SetLength(rs);
|
||
}
|
||
}
|
||
|
||
ITransport* Token::Create2401(SPI_TypeDef* spi_, Pin ce, Pin irq, Pin power, bool powerInvert, IDataPort* led)
|
||
{
|
||
static Spi spi(spi_, 10000000, true);
|
||
static NRF24L01 nrf;
|
||
nrf.Init(&spi, ce, irq, power);
|
||
|
||
auto tc = TinyConfig::Create();
|
||
if(tc->Interval == 0)
|
||
{
|
||
tc->Channel = 120;
|
||
tc->Speed = 250;
|
||
|
||
tc->Interval= 40;
|
||
tc->Timeout = 1000;
|
||
}
|
||
|
||
nrf.AutoAnswer = false;
|
||
nrf.DynPayload = false;
|
||
nrf.Channel = tc->Channel;
|
||
//nrf.Channel = 120;
|
||
nrf.Speed = tc->Speed;
|
||
|
||
nrf.FixData = Fix2401;
|
||
nrf.Led = led;
|
||
|
||
nrf.Master = true;
|
||
|
||
return &nrf;
|
||
}
|
||
|
||
ITransport* Token::CreateShunCom(COM_Def index, int baudRate, Pin rst, Pin power, Pin slp, Pin cfg, IDataPort* led)
|
||
{
|
||
auto tc = TinyConfig::Create();
|
||
if(tc->Interval == 0)
|
||
{
|
||
tc->Channel = 0x0F;
|
||
tc->Speed = 250;
|
||
|
||
tc->Interval= 400;
|
||
tc->Timeout = 2000;
|
||
}
|
||
|
||
static SerialPort sp(index, baudRate);
|
||
static ShunCom zb;
|
||
zb.Power.Set(power);
|
||
zb.Sleep.Init(slp, true);
|
||
zb.Config.Init(cfg, true);
|
||
zb.Init(&sp, rst);
|
||
zb.Led = led;
|
||
|
||
#if ShunCom_Master
|
||
zb.AddrLength = 2;
|
||
|
||
if(tc->Channel != 0x0F)
|
||
{
|
||
if(zb.EnterConfig())
|
||
{
|
||
zb.ShowConfig();
|
||
zb.SetDevice(0x00);
|
||
//zb.SetPanID(0x4444);
|
||
//zb.EraConfig();
|
||
tc->Channel = 0x0F;
|
||
tc->Save();
|
||
zb.SetSend(0x01);
|
||
zb.PrintSrc(true);
|
||
zb.ExitConfig();
|
||
}
|
||
}
|
||
#endif
|
||
return &zb;
|
||
}
|
||
|
||
void StartGateway(void* param)
|
||
{
|
||
ISocket* socket = NULL;
|
||
auto gw = Gateway::Current;
|
||
if(gw) socket = dynamic_cast<ISocket*>(gw->Client->Control->Port);
|
||
|
||
auto tk = TokenConfig::Current;
|
||
|
||
if(tk && tk->Server[0])
|
||
{
|
||
auto net = (W5500*)param;
|
||
// 根据DNS获取云端IP地址
|
||
UdpClient udp(net);
|
||
DNS dns(&udp);
|
||
udp.Open();
|
||
|
||
for(int i=0; i<10; i++)
|
||
{
|
||
auto ip = dns.Query(tk->Server, 2000);
|
||
ip.Show(true);
|
||
|
||
if(ip != IPAddress::Any())
|
||
{
|
||
tk->ServerIP = ip.Value;
|
||
|
||
if(socket) socket->Remote.Address = ip;
|
||
|
||
break;
|
||
}
|
||
}
|
||
|
||
tk->Save();
|
||
}
|
||
|
||
// 此时启动网关服务
|
||
if(gw)
|
||
{
|
||
auto& ep = gw->Client->Hello.EndPoint;
|
||
if(socket) ep.Address = socket->Host->IP;
|
||
|
||
if(!gw->Running)
|
||
{
|
||
gw->Start();
|
||
|
||
debug_printf("\r\n");
|
||
// 启动时首先进入学习模式
|
||
//gw->SetMode(30);
|
||
}
|
||
}
|
||
}
|
||
|
||
void Token::SetPower(ITransport* port)
|
||
{
|
||
auto pwr = dynamic_cast<Power*>(port);
|
||
if(pwr) pwr->SetPower();
|
||
}
|