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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
#include "commonheaders.h"
list<pCNTX> CL; // CL.size() CL.empty()
HANDLE thread_timeout = 0;
unsigned __stdcall sttTimeoutThread(LPVOID);
// get context data on context id
pCNTX get_context_on_id(HANDLE context) {
if( !thread_timeout ) {
unsigned int tID;
thread_timeout = (HANDLE) _beginthreadex(NULL, 0, sttTimeoutThread, NULL, 0, &tID);
}
if( context ) {
pCNTX cntx = (pCNTX) context;
if( cntx->header == HEADER && cntx->footer == FOOTER )
return cntx;
#if defined(_DEBUG) || defined(NETLIB_LOG)
else
Sent_NetLog("get_context_on_id: corrupted context %08X", cntx);
#endif
}
return NULL;
}
/*
pCNTX get_context_on_id(HANDLE context) {
return get_context_on_id((int)context);
}
*/
// create context, return context id
HANDLE __cdecl cpp_create_context(int mode) {
list<pCNTX>::iterator i;
pCNTX cntx = NULL;
EnterCriticalSection(&localContextMutex);
if( !CL.empty() ) {
for(i=CL.begin(); i!=CL.end(); ++i) { // ищем пустой
if( (*i)->header==EMPTYH && (*i)->footer==EMPTYH ) {
cntx = (pCNTX) *i;
break;
}
}
}
if( !cntx ) { // не нашли - создаем новый
cntx = (pCNTX) malloc(sizeof(CNTX));
CL.push_back(cntx); // добавили в конец списка
}
memset(cntx,0,sizeof(CNTX)); // очищаем выделенный блок
cntx->header = HEADER;
cntx->footer = FOOTER;
cntx->mode = mode;
LeaveCriticalSection(&localContextMutex);
return (HANDLE)cntx;
}
// delete context
void __cdecl cpp_delete_context(HANDLE context) {
pCNTX tmp = get_context_on_id(context);
if(tmp) { // помечаем на удаление
tmp->deleted = gettime()+10; // будет удален через 10 секунд
}
}
// reset context
void __cdecl cpp_reset_context(HANDLE context) {
pCNTX tmp = get_context_on_id(context);
if(tmp) cpp_free_keys(tmp);
}
// allocate pdata
PBYTE cpp_alloc_pdata(pCNTX ptr) {
if( !ptr->pdata ) {
if( ptr->mode & MODE_PGP ) {
ptr->pdata = (PBYTE) malloc(sizeof(PGPDATA));
memset(ptr->pdata,0,sizeof(PGPDATA));
}
else
if( ptr->mode & MODE_GPG ) {
ptr->pdata = (PBYTE) malloc(sizeof(GPGDATA));
memset(ptr->pdata,0,sizeof(GPGDATA));
}
else
if( ptr->mode & MODE_RSA ) {
rsa_alloc(ptr);
}
else {
ptr->pdata = (PBYTE) malloc(sizeof(SIMDATA));
memset(ptr->pdata,0,sizeof(SIMDATA));
}
}
return ptr->pdata;
}
// free memory from keys
void cpp_free_keys(pCNTX ptr) {
SAFE_FREE(ptr->tmp);
cpp_alloc_pdata(ptr);
if( ptr->mode & MODE_PGP ) {
pPGPDATA p = (pPGPDATA) ptr->pdata;
SAFE_FREE(p->pgpKeyID);
SAFE_FREE(p->pgpKey);
SAFE_FREE(ptr->pdata);
}
else
if( ptr->mode & MODE_GPG ) {
pGPGDATA p = (pGPGDATA) ptr->pdata;
SAFE_FREE(p->gpgKeyID);
SAFE_FREE(ptr->pdata);
}
else
if( ptr->mode & MODE_RSA ) {
rsa_free(ptr);
SAFE_DELETE(ptr->pdata);
}
else {
pSIMDATA p = (pSIMDATA) ptr->pdata;
SAFE_FREE(p->PubA);
SAFE_FREE(p->KeyA);
SAFE_FREE(p->KeyB);
SAFE_FREE(p->KeyX);
SAFE_FREE(p->KeyP);
SAFE_DELETE(p->dh);
SAFE_FREE(ptr->pdata);
}
}
// search not established RSA/AES contexts && clear deleted contexts
unsigned __stdcall sttTimeoutThread( LPVOID ) {
list<pCNTX>::iterator i;
while(1) {
Sleep( 1000 ); // раз в секунду
if( CL.empty() ) continue;
u_int time = gettime();
// пробегаем все контексты
EnterCriticalSection(&localContextMutex);
for(i=CL.begin(); i!=CL.end(); ++i) {
pCNTX tmp = *i;
if( tmp->header!=HEADER || tmp->footer!=FOOTER ) continue;
// пропускаем приватные ключи
if( tmp->mode&MODE_PRIV_KEY ) continue;
else
if( tmp->deleted && tmp->deleted < time ) {
// удалить помеченный для удаления контекст
cpp_free_keys(tmp);
tmp->deleted = 0;
tmp->header = tmp->footer = EMPTYH;
}
else
if( tmp->mode&MODE_RSA && tmp->pdata ) {
// проверяем не протухло ли соединение
pRSADATA p = (pRSADATA) tmp->pdata;
if( p->time && p->time < time ) {
rsa_timeout((HANDLE)tmp,p);
}
}
} // for
LeaveCriticalSection(&localContextMutex);
} //while
}
// EOF
|
