void CPushSessionHandler::OnRecvData(const char *szBuf, int32_t nBufSize)
{
m_Msg.Append(szBuf, nBufSize);
while (m_Msg.CheckMsgAvailable())
{
uint32_t cmd_id = m_Msg.GetCommandID();
switch (cmd_id) {
case IM::BaseDefine::CID_OTHER_HEARTBEAT:
_HandleHeartBeat(m_Msg.Data(), m_Msg.GetPduLength());
break;
case IM::BaseDefine::CID_OTHER_PUSH_TO_USER_REQ:
_HandlePushMsg(m_Msg.Data(), m_Msg.GetPduLength());
break;
default:
PUSH_SERVER_WARN("push session recv undefind msg, cmd id: %u.", cmd_id);
push_session_ptr pSession = CSessionManager::GetInstance()->GetPushSessionBySockID(_GetSockID());
if (pSession)
{
pSession->Stop();
}
break;
}
m_Msg.Remove(m_Msg.GetPduLength());
}
}
void CPushSessionHandler::OnException(uint32_t nsockid, int32_t nErrorCode)
{
PUSH_SERVER_WARN("push session has exception, sockid: %u, error code: %u.", nsockid, nErrorCode);
push_session_ptr pSession = CSessionManager::GetInstance()->GetPushSessionBySockID(nsockid);
if (pSession)
{
pSession->Stop();
}
}
BOOL CAPNSClient::_GetAPNSServerAddress()
{
BOOL bRet = FALSE;
char szIP[64] = {0};
uint32_t gateway_index = 0;
uint32_t feedback_index = 0;
if (GetSandBox())
{
gateway_index = 0;
feedback_index = 1;
}
else
{
gateway_index = 2;
feedback_index = 3;
}
//gateway host
if (S_GetHostByName(s_apn_servers[gateway_index].host, szIP) == 0)
{
m_strGatewayIP = szIP;
m_nGatewayPort = s_apn_servers[gateway_index].port;
}
else
{
PUSH_SERVER_WARN("parse gateway host failed, %s.", s_apn_servers[gateway_index].host);
return bRet;
}
//feedback host
if (S_GetHostByName(s_apn_servers[feedback_index].host, szIP) == 0)
{
m_strFeedbackIP = szIP;
m_nFeedbackPort = s_apn_servers[feedback_index].port;
}
else
{
PUSH_SERVER_WARN("parse feedback host failed, %s.", s_apn_servers[feedback_index].host);
return bRet;
}
bRet = TRUE;
return bRet;
}
void CPushSessionHandler::_HandlePushMsg(const char* szBuf, int32_t nBufSize)
{
CPduMsg pdu_msg;
if (!pdu_msg.ParseFromArray(szBuf, nBufSize))
{
PUSH_SERVER_ERROR("HandlePushMsg, msg parse failed.");
return;
}
push_session_ptr pSession = CSessionManager::GetInstance()->GetPushSessionBySockID(_GetSockID());
if (pSession == nullptr)
{
return;
}
pSession->SetHeartBeat(S_GetTickCount());
IM::Server::IMPushToUserReq msg;
if (!msg.ParseFromArray(pdu_msg.Body(), pdu_msg.GetBodyLength()))
{
PUSH_SERVER_ERROR("HandlePushMsg, msg parse failed.");
pSession->Stop();
return;
}
string strFlash = msg.flash();
string strUserData = msg.data();
apns_client_ptr pClient = CSessionManager::GetInstance()->GetAPNSClient();
if (pClient)
{
IM::Server::IMPushToUserRsp msg2;
for (uint32_t i = 0; i < msg.user_token_list_size(); i++)
{
IM::BaseDefine::PushResult* push_result = msg2.add_push_result_list();
IM::BaseDefine::UserTokenInfo user_token = msg.user_token_list(i);
if (user_token.user_type() != IM::BaseDefine::CLIENT_TYPE_IOS) {
push_result->set_user_token(user_token.token());
push_result->set_result_code(1);
continue;
}
m_NotificationID++;
PUSH_SERVER_INFO("HandlePushMsg, token: %s, push count: %d, push_type:%d, notification id: %u.", user_token.token().c_str(), user_token.push_count(),
user_token.push_type(), m_NotificationID);
CAPNSGateWayMsg msg;
msg.SetAlterBody(strFlash);
msg.SetCustomData(strUserData);
msg.SetDeviceToken(user_token.token());
time_t time_now = 0;
time(&time_now);
msg.SetExpirationDate(time_now + 3600);
if (user_token.push_type() == PUSH_TYPE_NORMAL)
{
msg.SetSound(TRUE);
}
else
{
msg.SetSound(FALSE);
}
msg.SetBadge(user_token.push_count());
msg.SetNotificationID(m_NotificationID);
if (msg.SerializeToArray())
{
pClient->SendPushMsgToGateway(msg.Data(), msg.GetDataLength());
push_result->set_result_code(0);
}
else
{
push_result->set_result_code(1);
PUSH_SERVER_WARN("HandlePushMsg, serialize CAPNSGateWayMsg failed.");
}
push_result->set_user_token(user_token.token());
}
CPduMsg pdu_msg2;
pdu_msg2.SetServiceID(IM::BaseDefine::SID_OTHER);
pdu_msg2.SetCommandID(IM::BaseDefine::CID_OTHER_PUSH_TO_USER_RSP);
if (pdu_msg2.SerializeToArray(&msg2))
{
pSession->SendMsg(pdu_msg2.Data(), pdu_msg2.GetDataLength());
}
else
{
PUSH_SERVER_WARN("HandlePushMsg, serialize IMPushToUserRsp failed.");
}
}
}
void CPushSessionHandler::OnClose(uint32_t nsockid)
{
PUSH_SERVER_WARN("push session closed, sockid: %u.", nsockid);
m_Msg.Clear();
CSessionManager::GetInstance()->RemovePushSessionBySockID(nsockid);
}
