193 lines
4.1 KiB
C++
193 lines
4.1 KiB
C++
#include "Controller.h"
|
||
|
||
#define MSG_DEBUG DEBUG
|
||
//#define MSG_DEBUG 0
|
||
#if MSG_DEBUG
|
||
#define msg_printf debug_printf
|
||
#else
|
||
__inline void msg_printf( const char *format, ... ) {}
|
||
#endif
|
||
|
||
// 构造控制器
|
||
Controller::Controller()
|
||
{
|
||
Port = NULL;
|
||
MinSize = 0;
|
||
Opened = false;
|
||
|
||
Received = NULL;
|
||
Param = NULL;
|
||
}
|
||
|
||
Controller::~Controller()
|
||
{
|
||
Close();
|
||
|
||
_Handlers.DeleteAll().Clear();
|
||
if(Port) delete Port;
|
||
|
||
debug_printf("TinyNet::UnInit\r\n");
|
||
}
|
||
|
||
void Controller::Open()
|
||
{
|
||
if(Opened) return;
|
||
|
||
assert_param2(Port, "还没有传输口呢");
|
||
|
||
// 注册收到数据事件
|
||
Port->Open();
|
||
Port->Register(Dispatch, this);
|
||
|
||
Opened = true;
|
||
}
|
||
|
||
void Controller::Close()
|
||
{
|
||
if(!Opened) return;
|
||
|
||
Opened = false;
|
||
}
|
||
|
||
Controller* _my = NULL;
|
||
uint Controller::Dispatch(ITransport* port, byte* buf, uint len, void* param)
|
||
{
|
||
assert_ptr(buf);
|
||
assert_ptr(param);
|
||
|
||
Controller* control = (Controller*)param;
|
||
if(!_my)
|
||
_my = control;
|
||
else
|
||
{
|
||
assert_param2(_my == control, "控制器指针已被改变");
|
||
}
|
||
|
||
// 这里使用数据流，可能多个消息粘包在一起
|
||
// 注意，此时指针位于0，而内容长度为缓冲区长度
|
||
Stream ms(buf, len);
|
||
while(ms.Remain() >= control->MinSize)
|
||
{
|
||
assert_param2(_my == control, "控制器指针已被改变");
|
||
// 如果不是有效数据包，则直接退出，避免产生死循环。当然，也可以逐字节移动测试，不过那样性能太差
|
||
if(!control->Dispatch(ms, NULL)) break;
|
||
assert_param2(_my == control, "控制器指针已被改变");
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
bool Controller::Dispatch(Stream& ms, Message* pmsg)
|
||
{
|
||
byte* buf = ms.Current();
|
||
|
||
#if MSG_DEBUG
|
||
/*msg_printf("TinyNet::Dispatch ");
|
||
// 输出整条信息
|
||
Sys.ShowHex(buf, ms.Length, '-');
|
||
msg_printf("\r\n");*/
|
||
#endif
|
||
|
||
Message& msg = *pmsg;
|
||
if(!msg.Read(ms)) return false;
|
||
|
||
// 校验
|
||
if(!msg.Valid()) return true;
|
||
|
||
if(!Valid(msg))
|
||
{
|
||
/*debug_printf("消息校验未通过\r\n");
|
||
msg.Show();
|
||
debug_printf("\r\n");*/
|
||
|
||
return true;
|
||
}
|
||
|
||
return OnReceive(msg);
|
||
}
|
||
|
||
bool Controller::Valid(Message& msg)
|
||
{
|
||
return true;
|
||
}
|
||
|
||
// 接收处理
|
||
bool Controller::OnReceive(Message& msg)
|
||
{
|
||
// 选择处理器来处理消息
|
||
// 倒序选择处理器来处理消息，让后注册处理器有更高优先权
|
||
for(int i = _Handlers.Count() - 1; i >= 0; i--)
|
||
{
|
||
HandlerLookup* lookup = _Handlers[i];
|
||
if(lookup && lookup->Code == msg.Code)
|
||
{
|
||
// 返回值决定是否成功处理，如果未成功处理，则技术遍历下一个处理器
|
||
bool rs = lookup->Handler(msg, lookup->Param);
|
||
if(rs) break;
|
||
}
|
||
}
|
||
|
||
// 外部公共消息事件
|
||
if(Received)
|
||
{
|
||
if(!Received(msg, Param)) return true;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
void Controller::Register(byte code, MessageHandler handler, void* param)
|
||
{
|
||
assert_param2(code, "注册指令码不能为空");
|
||
assert_param2(handler, "注册函数不能为空");
|
||
|
||
HandlerLookup* lookup;
|
||
|
||
#if DEBUG
|
||
// 检查是否已注册。一般指令码是固定的，所以只在DEBUG版本检查
|
||
for(int i=0; i<_Handlers.Count(); i++)
|
||
{
|
||
lookup = _Handlers[i];
|
||
if(lookup && lookup->Code == code)
|
||
{
|
||
debug_printf("Controller::Register Error! Code=%d was Registered to 0x%08x\r\n", code, lookup->Handler);
|
||
//return;
|
||
}
|
||
}
|
||
#endif
|
||
|
||
lookup = new HandlerLookup();
|
||
lookup->Code = code;
|
||
lookup->Handler = handler;
|
||
lookup->Param = param;
|
||
|
||
_Handlers.Add(lookup);
|
||
}
|
||
|
||
bool Controller::Send(Message& msg)
|
||
{
|
||
// 如果没有传输口处于打开状态，则发送失败
|
||
if(!Port->Open()) return false;
|
||
|
||
uint len = msg.Size();
|
||
|
||
// ms需要在外面这里声明，否则离开大括号作用域以后变量被销毁，导致缓冲区不可用
|
||
//Stream ms(len);
|
||
byte buf[512];
|
||
Stream ms(buf, ArrayLength(buf));
|
||
// 带有负载数据，需要合并成为一段连续的内存
|
||
msg.Write(ms);
|
||
assert_param2(len == ms.Position(), "消息标称大小和实际大小不符");
|
||
// 内存流扩容以后，指针会改变
|
||
byte* p = ms.GetBuffer();
|
||
|
||
return Port->Write(p, len);
|
||
}
|
||
|
||
bool Controller::Reply(Message& msg)
|
||
{
|
||
msg.Reply = 1;
|
||
|
||
return Send(msg);
|
||
}
|