1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*
Plugin for Miranda NG for communicating with users of the MSN Messenger protocol.
Copyright (c) 2012-18 Miranda NG team
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "stdafx.h"
#include "msn_proto.h"
void CMsnProto::AvatarQueue_Init()
{
hevAvatarQueue = ::CreateSemaphore(nullptr, 0, 255, nullptr);
ForkThread(&CMsnProto::MSN_AvatarsThread, nullptr);
}
void CMsnProto::AvatarQueue_Uninit()
{
::CloseHandle(hevAvatarQueue);
}
void CMsnProto::pushAvatarRequest(MCONTACT hContact, LPCSTR pszUrl)
{
ProtoBroadcastAck(hContact, ACKTYPE_AVATAR, ACKRESULT_STATUS, nullptr);
if (pszUrl != nullptr && *pszUrl != 0) {
mir_cslock lck(csAvatarQueue);
for (auto &it : lsAvatarQueue)
if (it->hContact == hContact)
return;
lsAvatarQueue.insert(new AvatarQueueEntry(hContact, pszUrl));
ReleaseSemaphore(hevAvatarQueue, 1, nullptr);
}
}
bool CMsnProto::loadHttpAvatar(AvatarQueueEntry *p)
{
NETLIBHTTPHEADER nlbhHeaders[1];
nlbhHeaders[0].szName = "User-Agent";
nlbhHeaders[0].szValue = (char*)MSN_USER_AGENT;
NETLIBHTTPREQUEST nlhr = { sizeof(nlhr) };
nlhr.requestType = REQUEST_GET;
nlhr.flags = NLHRF_HTTP11 | NLHRF_REDIRECT;
nlhr.szUrl = p->pszUrl;
nlhr.headers = (NETLIBHTTPHEADER*)&nlbhHeaders;
nlhr.headersCount = 1;
NETLIBHTTPREQUEST *nlhrReply = Netlib_HttpTransaction(m_hNetlibUser, &nlhr);
if (nlhrReply == nullptr)
return false;
if (nlhrReply->resultCode != 200 || nlhrReply->dataLength == 0) {
LBL_Error:
Netlib_FreeHttpRequest(nlhrReply);
return false;
}
const wchar_t *szExt;
int fmt = ProtoGetBufferFormat(nlhrReply->pData, &szExt);
if (fmt == PA_FORMAT_UNKNOWN)
goto LBL_Error;
PROTO_AVATAR_INFORMATION ai = { 0 };
ai.format = fmt;
ai.hContact = p->hContact;
MSN_GetAvatarFileName(ai.hContact, ai.filename, _countof(ai.filename), szExt);
_wremove(ai.filename);
int fileId = _wopen(ai.filename, _O_CREAT | _O_TRUNC | _O_WRONLY | O_BINARY, _S_IREAD | _S_IWRITE);
if (fileId == -1)
goto LBL_Error;
_write(fileId, nlhrReply->pData, (unsigned)nlhrReply->dataLength);
_close(fileId);
ProtoBroadcastAck(p->hContact, ACKTYPE_AVATAR, ACKRESULT_SUCCESS, &ai);
Netlib_FreeHttpRequest(nlhrReply);
return true;
}
void __cdecl CMsnProto::MSN_AvatarsThread(void*)
{
while (true) {
if (WaitForSingleObject(hevAvatarQueue, INFINITE) != WAIT_OBJECT_0)
break;
if (g_bTerminated)
break;
AvatarQueueEntry *p = nullptr;
{
mir_cslock lck(csAvatarQueue);
if (lsAvatarQueue.getCount() > 0) {
p = lsAvatarQueue[0];
lsAvatarQueue.remove(0);
}
}
if (p == nullptr)
continue;
if (!loadHttpAvatar(p))
ProtoBroadcastAck(p->hContact, ACKTYPE_AVATAR, ACKRESULT_FAILED, nullptr);
delete p;
}
mir_cslock lck(csAvatarQueue);
for (auto &it : lsAvatarQueue)
delete it;
lsAvatarQueue.destroy();
}
|
