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
|
/*
Plugin of Miranda IM for communicating with users of the MSN Messenger protocol.
Copyright (c) 2012-2019 Miranda NG team
Copyright (c) 2007-2012 Boris Krasnovskiy.
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"
chunkedmsg::chunkedmsg(const char* tid, const size_t totsz, const bool tbychunk)
: size(totsz), recvsz(0), bychunk(tbychunk)
{
id = mir_strdup(tid);
msg = tbychunk ? nullptr : (char*)mir_alloc(totsz + 1);
}
chunkedmsg::~chunkedmsg()
{
mir_free(id);
mir_free(msg);
}
void chunkedmsg::add(const char* tmsg, size_t offset, size_t portion)
{
if (bychunk) {
size_t oldsz = recvsz;
recvsz += portion;
msg = (char*)mir_realloc(msg, recvsz + 1);
memcpy(msg + oldsz, tmsg, portion);
--size;
}
else {
size_t newsz = offset + portion;
if (newsz > size) {
portion = size - offset;
newsz = size;
}
memcpy(msg + offset, tmsg, portion);
if (newsz > recvsz) recvsz = newsz;
}
}
bool chunkedmsg::get(char*& tmsg, size_t& tsize)
{
bool alldata = bychunk ? size == 0 : recvsz == size;
if (alldata) {
msg[recvsz] = 0;
tmsg = msg;
tsize = recvsz;
msg = nullptr;
}
return alldata;
}
int CMsnProto::addCachedMsg(const char* id, const char* msg, const size_t offset,
const size_t portion, const size_t totsz, const bool bychunk)
{
int idx = msgCache.getIndex((chunkedmsg*)&id);
if (idx == -1) {
msgCache.insert(new chunkedmsg(id, totsz, bychunk));
idx = msgCache.getIndex((chunkedmsg*)&id);
}
msgCache[idx].add(msg, offset, portion);
return idx;
}
size_t CMsnProto::getCachedMsgSize(const char* id)
{
int idx = msgCache.getIndex((chunkedmsg*)&id);
return idx != -1 ? msgCache[idx].size : 0;
}
bool CMsnProto::getCachedMsg(int idx, char*& msg, size_t& size)
{
bool res = msgCache[idx].get(msg, size);
if (res)
msgCache.remove(idx);
return res;
}
bool CMsnProto::getCachedMsg(const char* id, char*& msg, size_t& size)
{
int idx = msgCache.getIndex((chunkedmsg*)&id);
return idx != -1 && getCachedMsg(idx, msg, size);
}
void CMsnProto::clearCachedMsg(int idx)
{
if (idx != -1)
msgCache.remove(idx);
else
msgCache.destroy();
}
void CMsnProto::CachedMsg_Uninit(void)
{
clearCachedMsg();
}
|