DWORD avatars_server_connection::sendUploadAvatarRequest(MCONTACT hContact, WORD wRef, const BYTE *data, size_t datalen)
{
cookie_avatar *ack = (cookie_avatar*)SAFE_MALLOC(sizeof(cookie_avatar));
if (!ack)
return 0; // Failure: out of memory
ack->hContact = hContact;
DWORD dwCookie = ppro->AllocateCookie(CKT_AVATAR, ICQ_AVATAR_UPLOAD_REQUEST, 0, ack);
icq_packet packet;
serverPacketInit(&packet, 14 + datalen);
packFNACHeader(&packet, ICQ_AVATAR_FAMILY, ICQ_AVATAR_UPLOAD_REQUEST, 0, dwCookie);
packWord(&packet, wRef); // unknown, probably reference
packWord(&packet, (WORD)datalen);
packBuffer(&packet, data, datalen);
if (sendServerPacket(&packet)) {
ppro->debugLogA("Upload image packet sent.");
return dwCookie;
}
ppro->ReleaseCookie(dwCookie); // failed to send, free resources
return 0;
}
void NetManager::update()
{
mTick++;
ENetEvent event;
while(enet_host_service(mHost, &event, 0))
{
switch(event.type)
{
case ENET_EVENT_TYPE_CONNECT:
{
if(mIsServer)
{
spawnPlayer(event.peer->connectID);
mClients.push_back(event.peer);
}
break;
}
case ENET_EVENT_TYPE_RECEIVE:
{
if(mIsServer)
readClientPacket(event);
else
readServerPacket(event);
break;
}
default:
{
break;
}
}
}
if(mIsServer && mClients.size() > 0)
sendServerPacket();
else if(mServerPeer != NULL)
sendClientPacket();
enet_host_flush(mHost);
}
void avatars_server_connection::connectionThread()
{
// This is the "infinite" loop that receives the packets from the ICQ avatar server
NETLIBPACKETRECVER packetRecv = { 0 };
DWORD dwLastKeepAlive = time(0) + KEEPALIVE_INTERVAL;
hPacketRecver = (HANDLE)CallService(MS_NETLIB_CREATEPACKETRECVER, (WPARAM)hConnection, 65536);
packetRecv.cbSize = sizeof(packetRecv);
packetRecv.dwTimeout = 1000; // timeout - for stopThread to work
while (!stopThread) {
int recvResult = CallService(MS_NETLIB_GETMOREPACKETS, (WPARAM)hPacketRecver, (LPARAM)&packetRecv);
if (recvResult == 0) {
ppro->debugLogA("Clean closure of avatar socket");
break;
}
if (recvResult == SOCKET_ERROR) {
if (GetLastError() == ERROR_TIMEOUT) { // timeout, check if we should be still running
if (Miranda_Terminated())
break;
if (time(0) >= dwLastKeepAlive) { // limit frequency (HACK: on some systems select() does not work well)
if (!ppro->m_bGatewayMode && ppro->getByte("KeepAlive", DEFAULT_KEEPALIVE_ENABLED)) { // send keep-alive packet
icq_packet packet;
packet.wLen = 0;
write_flap(&packet, ICQ_PING_CHAN);
sendServerPacket(&packet);
}
dwLastKeepAlive = time(0) + KEEPALIVE_INTERVAL;
}
// check if we got something to request
checkRequestQueue();
continue;
}
if (!stopThread)
ppro->debugLogA("Avatar socket closed abortively, error: %d", GetLastError());
else
ppro->debugLogA("Avatar socket gracefully closed.");
break;
}
// Deal with the packet
packetRecv.bytesUsed = handleServerPackets(packetRecv.buffer, packetRecv.bytesAvailable);
if (isActive && (packetRecv.bytesAvailable == packetRecv.bytesUsed)) // no packets pending
checkRequestQueue(); // process request queue
}
{
// release connection
mir_cslock l(localSeqMutex);
NetLib_SafeCloseHandle(&hPacketRecver); // Close the packet receiver
NetLib_CloseConnection(&hConnection, FALSE); // Close the connection
}
{
// release rates
mir_cslock l(m_ratesMutex);
delete m_rates; m_rates = NULL;
}
SAFE_FREE((void**)&pCookie);
}
DWORD avatars_server_connection::sendGetAvatarRequest(MCONTACT hContact, DWORD dwUin, char *szUid, const BYTE *hash, size_t hashlen, const TCHAR *file)
{
int i;
DWORD dwNow = GetTickCount();
mir_cslockfull alck(ppro->m_avatarsMutex);
for (i = 0; i < runCount;) { // look for timeouted requests
if (runTime[i] < dwNow) { // found outdated, remove
runContact[i] = runContact[runCount - 1];
runTime[i] = runTime[runCount - 1];
runCount--;
}
else i++;
}
for (i = 0; i < runCount; i++) {
if (runContact[i] == hContact) {
ppro->debugLogA("Ignoring duplicate get %s image request.", strUID(dwUin, szUid));
return -1; // Success: request ignored
}
}
if (runCount < 4) { // 4 concurent requests at most
int bSendNow = TRUE;
{
// rate management
mir_cslock l(m_ratesMutex);
WORD wGroup = m_rates->getGroupFromSNAC(ICQ_AVATAR_FAMILY, ICQ_AVATAR_GET_REQUEST);
if (m_rates->getNextRateLevel(wGroup) < m_rates->getLimitLevel(wGroup, RML_ALERT)) { // we will be over quota if we send the request now, add to queue instead
bSendNow = FALSE;
ppro->debugLogA("Rates: Delay avatar request.");
}
}
if (bSendNow) {
runContact[runCount] = hContact;
runTime[runCount] = GetTickCount() + 30000; // 30sec to complete request
runCount++;
alck.unlock();
int nUinLen = getUIDLen(dwUin, szUid);
cookie_avatar *ack = (cookie_avatar*)SAFE_MALLOC(sizeof(cookie_avatar));
if (!ack)
return 0; // Failure: out of memory
ack->dwUin = 1; //dwUin; // I should be damned for this - only to identify get request
ack->hContact = hContact;
ack->hash = (BYTE*)SAFE_MALLOC(hashlen);
memcpy(ack->hash, hash, hashlen); // copy the data
ack->hashlen = hashlen;
ack->szFile = null_strdup(file); // duplicate the string
DWORD dwCookie = ppro->AllocateCookie(CKT_AVATAR, ICQ_AVATAR_GET_REQUEST, hContact, ack);
icq_packet packet;
serverPacketInit(&packet, 12 + nUinLen + hashlen);
packFNACHeader(&packet, ICQ_AVATAR_FAMILY, ICQ_AVATAR_GET_REQUEST, 0, dwCookie);
packUID(&packet, dwUin, szUid);
packByte(&packet, 1); // unknown, probably type of request: 1 = get icon :)
packBuffer(&packet, hash, hashlen);
if (sendServerPacket(&packet)) {
ppro->debugLogA("Request to get %s image sent.", strUID(dwUin, szUid));
return dwCookie;
}
ppro->FreeCookie(dwCookie); // sending failed, free resources
SAFE_FREE(&ack->szFile);
SAFE_FREE((void**)&ack->hash);
SAFE_FREE((void**)&ack);
}
}
return 0; // Failure
}
void avatars_server_connection::connectionThread()
{
// This is the "infinite" loop that receives the packets from the ICQ avatar server
NETLIBPACKETRECVER packetRecv = {0};
DWORD wLastKeepAlive = 0; // we send keep-alive at most one per 30secs
DWORD dwKeepAliveInterval = ppro->getDword("KeepAliveInterval", KEEPALIVE_INTERVAL);
hPacketRecver = (HANDLE)CallService(MS_NETLIB_CREATEPACKETRECVER, (WPARAM)hConnection, 65536);
packetRecv.cbSize = sizeof(packetRecv);
packetRecv.dwTimeout = dwKeepAliveInterval < KEEPALIVE_INTERVAL ? dwKeepAliveInterval: KEEPALIVE_INTERVAL; // timeout - for stopThread to work
while (!stopThread)
{
int recvResult = CallService(MS_NETLIB_GETMOREPACKETS, (WPARAM)hPacketRecver, (LPARAM)&packetRecv);
if (recvResult == 0)
{
ppro->debugLogA("Clean closure of server socket");
break;
}
if (recvResult == SOCKET_ERROR)
{
if (GetLastError() == ERROR_TIMEOUT)
{ // timeout, check if we should be still running
if (Miranda_Terminated())
{ // we must stop here, cause due to a hack in netlib, we always get timeout, even if the connection is already dead
stopThread = 1;
continue;
}
#ifdef _DEBUG
else ppro->debugLogA("Thread is Idle.");
#endif
if (GetTickCount() > wLastKeepAlive)
{ // limit frequency (HACK: on some systems select() does not work well)
if (!ppro->m_bGatewayMode && ppro->getByte("KeepAlive", DEFAULT_KEEPALIVE_ENABLED))
{ // send keep-alive packet
icq_packet packet;
packet.wLen = 0;
write_flap(&packet, ICQ_PING_CHAN);
sendServerPacket(&packet);
}
wLastKeepAlive = GetTickCount() + dwKeepAliveInterval;
}
else
{ // this is bad, the system does not handle select() properly
#ifdef _DEBUG
ppro->debugLogA("Thread is Forcing Idle.");
#endif
SleepEx(500, TRUE); // wait some time, can we do anything else ??
if (Miranda_Terminated())
{
stopThread = 1;
continue;
}
}
// check if we got something to request
checkRequestQueue();
continue;
}
if (!stopThread)
ppro->debugLogA("Abortive closure of server socket, error: %d", GetLastError());
else
ppro->debugLogA("Connection closed.");
break;
}
// Deal with the packet
packetRecv.bytesUsed = handleServerPackets(packetRecv.buffer, packetRecv.bytesAvailable);
if (isActive && (packetRecv.bytesAvailable == packetRecv.bytesUsed)) // no packets pending
{ // process request queue
checkRequestQueue();
}
}
{ // release connection
icq_lock l(localSeqMutex);
NetLib_SafeCloseHandle(&hPacketRecver); // Close the packet receiver
NetLib_CloseConnection(&hConnection, FALSE); // Close the connection
}
{ // release rates
icq_lock l(m_ratesMutex);
SAFE_DELETE((MZeroedObject**)&m_rates);
}
SAFE_FREE((void**)&pCookie);
}
void NetManager::update()
{
mTick++;
ENetEvent event;
// TODO: remove this block when we handle level switching properly
auto player = FAWorld::World::get()->getCurrentPlayer();
if(player->getLevel())
mLevelIndexTmp = player->getLevel()->getLevelIndex();
else
mLevelIndexTmp = -1;
while(enet_host_service(mHost, &event, 0))
{
switch(event.type)
{
case ENET_EVENT_TYPE_CONNECT:
{
if(mIsServer)
{
auto player = spawnPlayer(-1);
// send the client its player id
auto packet = enet_packet_create(NULL, sizeof(size_t), ENET_PACKET_FLAG_RELIABLE);
size_t position = 0;
writeToPacket(packet, position, player->getId());
enet_peer_send(event.peer, RELIABLE_CHANNEL_ID, packet);
sendLevel(0, event.peer);
sendLevel(1, event.peer);
mClients.push_back(event.peer);
mServerPlayerList[event.peer->connectID] = player;
}
break;
}
case ENET_EVENT_TYPE_RECEIVE:
{
if(mIsServer)
readClientPacket(event);
else
readServerPacket(event);
break;
}
default:
{
break;
}
}
}
if(mIsServer && mClients.size() > 0)
sendServerPacket();
else if(mServerPeer != NULL)
sendClientPacket();
enet_host_flush(mHost);
}
