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
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
| class vncClient;
typedef SHORT vncClientId;
#if (!defined(_WINVNC_VNCCLIENT))
#define _WINVNC_VNCCLIENT
#include <list>
typedef std::list<vncClientId> vncClientList;
// Includes
#include "stdhdrs.h"
#include "VSocket.h"
#include <omnithread.h>
// Custom
#include "vncDesktop.h"
#include "rfbRegion.h"
#include "rfbUpdateTracker.h"
#include "vncBuffer.h"
#include "vncEncodeMgr.h"
#include "TextChat.h" // sf@2002 - TextChat
#include "ZipUnZip32/zipUnZip32.h"
//#include "timer.h"
// The vncClient class itself
typedef UINT (WINAPI *pSendinput)(UINT,LPINPUT,INT);
#define SPI_GETMOUSESPEED 0x0070
#define SPI_SETMOUSESPEED 0x0071
#define MOUSEEVENTF_VIRTUALDESK 0x4000
extern int CheckUserGroupPasswordUni(char * userin,char *password,const char *machine);
using namespace rfb;
class vncClient
{
public:
// Constructor/destructor
vncClient();
~vncClient();
// Allow the client thread to see inside the client object
friend class vncClientThread;
friend class vncClientUpdateThread;
// Init
virtual BOOL Init(vncServer *server,
VSocket *socket,
BOOL auth,
BOOL shared,
vncClientId newid);
// Kill
// The server uses this to close the client socket, causing the
// client thread to fail, which in turn deletes the client object
virtual void Kill();
// Client manipulation functions for use by the server
virtual void SetBuffer(vncBuffer *buffer);
// Update handling functions
// These all lock the UpdateLock themselves
virtual void UpdateMouse();
virtual void UpdateClipText(const char* text);
virtual void UpdatePalette();
virtual void UpdateLocalFormat();
// Is the client waiting on an update?
// YES IFF there is an incremental update region,
// AND no changed or copied updates intersect it
virtual BOOL UpdateWanted() {
omni_mutex_lock l(GetUpdateLock());
return !m_incr_rgn.is_empty() &&
m_incr_rgn.intersect(m_update_tracker.get_changed_region()).is_empty() &&
m_incr_rgn.intersect(m_update_tracker.get_cached_region()).is_empty() &&
m_incr_rgn.intersect(m_update_tracker.get_copied_region()).is_empty();
};
// Has the client sent an input event?
virtual BOOL RemoteEventReceived() {
BOOL result = m_remoteevent;
m_remoteevent = FALSE;
return result;
};
// The UpdateLock
// This must be held for a number of routines to be successfully invoked...
virtual omni_mutex& GetUpdateLock() {return m_encodemgr.GetUpdateLock();};
// Functions for setting & getting the client settings
virtual void EnableKeyboard(BOOL enable) {m_keyboardenabled = enable;};
virtual void EnablePointer(BOOL enable) {m_pointerenabled = enable;};
virtual void SetCapability(int capability) {m_capability = capability;};
virtual int GetCapability() {return m_capability;};
virtual const char *GetClientName();
virtual vncClientId GetClientId() {return m_id;};
// Disable/enable protocol messages to the client
virtual void DisableProtocol();
virtual void EnableProtocol();
// resize desktop
virtual BOOL SetNewSWSize(long w,long h,BOOL desktop);
virtual void SetSWOffset(int x,int y);
virtual void SetScreenOffset(int x,int y,int type);
virtual TextChat* GetTextChatPointer() { return m_pTextChat; }; // sf@2002
virtual void SetUltraViewer(bool fTrue) { m_fUltraViewer = fTrue;};
virtual bool IsUltraViewer() { return m_fUltraViewer;};
virtual void EnableCache(BOOL enabled);
// sf@2002
virtual void SetConnectTime(long lTime) {m_lConnectTime = lTime;};
virtual long GetConnectTime() {return m_lConnectTime;};
virtual bool IsSlowEncoding() {return m_encodemgr.IsSlowEncoding();};
virtual bool IsUltraEncoding() {return m_encodemgr.IsUltraEncoding();};
void SetProtocolVersion(rfbProtocolVersionMsg *protocolMsg);
void Clear_Update_Tracker();
void UpdateCursorShape();
// sf@2004 - Asynchronous FileTransfer - Delta Transfer
int GenerateFileChecksums(HANDLE hFile, char* lpCSBuffer, int nCSBufferSize);
bool ReceiveDestinationFileChecksums(int nSize, int nLen);
void ReceiveFileChunk(int nLen, int nSize);
void FinishFileReception();
void SendFileChunk();
void FinishFileSending();
bool GetSpecialFolderPath(int nId, char* szPath);
int ZipPossibleDirectory(LPSTR szSrcFileName);
int CheckAndZipDirectoryForChecksuming(LPSTR szSrcFileName);
int UnzipPossibleDirectory(LPSTR szFileName);
bool MyGetFileSize(char* szFilePath, ULARGE_INTEGER* n2FileSize);
bool DoFTUserImpersonation();
void UndoFTUserImpersonation();
// sf@2002
// Update routines
protected:
BOOL SendUpdate(rfb::SimpleUpdateTracker &update);
BOOL SendRFBMsg(CARD8 type, BYTE *buffer, int buflen);
BOOL SendRectangles(const rfb::RectVector &rects);
BOOL SendRectangle(const rfb::Rect &rect);
BOOL SendCopyRect(const rfb::Rect &dest, const rfb::Point &source);
BOOL SendPalette();
// CACHE
BOOL SendCacheRectangles(const rfb::RectVector &rects);
BOOL SendCacheRect(const rfb::Rect &dest);
BOOL SendCacheZip(const rfb::RectVector &rects); // sf@2002
// Tight - CURSOR HANDLING
BOOL SendCursorShapeUpdate();
// nyama/marscha - PointerPos
BOOL SendCursorPosUpdate();
BOOL SendLastRect(); // Tight
void TriggerUpdateThread();
void PollWindow(HWND hwnd);
// Specialised client-side UpdateTracker
protected:
// This update tracker stores updates it receives and
// kicks the client update thread every time one is received
class ClientUpdateTracker : public rfb::SimpleUpdateTracker {
public:
ClientUpdateTracker() : m_client(0) {};
virtual ~ClientUpdateTracker() {};
void init(vncClient *client) {m_client=client;}
virtual void add_changed(const rfb::Region2D ®ion) {
{
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::add_changed(region);
m_client->TriggerUpdateThread();
}
}
virtual void add_cached(const rfb::Region2D ®ion) {
{
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::add_cached(region);
m_client->TriggerUpdateThread();
}
}
virtual void add_copied(const rfb::Region2D &dest, const rfb::Point &delta) {
{
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::add_copied(dest, delta);
m_client->TriggerUpdateThread();
}
}
virtual void clear() {
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::clear();
}
virtual void flush_update(rfb::UpdateInfo &info, const rfb::Region2D &cliprgn) {;
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::flush_update(info, cliprgn);
}
virtual void flush_update(rfb::UpdateTracker &to, const rfb::Region2D &cliprgn) {;
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::flush_update(to, cliprgn);
}
virtual void get_update(rfb::UpdateInfo &info) const {;
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::get_update(info);
}
virtual void get_update(rfb::UpdateTracker &to) const {
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
SimpleUpdateTracker::get_update(to);
}
virtual bool is_empty() const{
// RealVNC 336 change - omni_mutex_lock l(m_client->GetUpdateLock());
return SimpleUpdateTracker::is_empty();
}
protected:
vncClient *m_client;
};
friend class ClientUpdateTracker;
// Make the update tracker available externally
public:
rfb::UpdateTracker &GetUpdateTracker() {return m_update_tracker;};
int m_SWOffsetx;
int m_SWOffsety;
int m_ScreenOffsetx;
int m_ScreenOffsety;
int m_display_type;
BOOL m_NewSWUpdateWaiting;
rfbProtocolVersionMsg ProtocolVersionMsg;
//Timer Sendtimer;
//int roundrobin_counter;
//int timearray[rfbEncodingZRLE+1][31];
//int sizearray[rfbEncodingZRLE+1][31];
//int Totalsend;
BOOL client_settings_passed;
// Internal stuffs
protected:
// Per-client settings
BOOL m_IsLoopback;
BOOL m_keyboardenabled;
BOOL m_pointerenabled;
int m_capability;
vncClientId m_id;
long m_lConnectTime;
// Pixel translation & encoding handler
vncEncodeMgr m_encodemgr;
// The server
vncServer *m_server;
// The socket
VSocket *m_socket;
char *m_client_name;
// The client thread
omni_thread *m_thread;
// Count to indicate whether updates, clipboards, etc can be sent
// to the client. If 0 then OK, otherwise not.
ULONG m_disable_protocol;
// User input information
rfb::Rect m_oldmousepos;
BOOL m_mousemoved;
rfbPointerEventMsg m_ptrevent;
// vncKeymap m_keymap;
// Update tracking structures
ClientUpdateTracker m_update_tracker;
// Client update transmission thread
vncClientUpdateThread *m_updatethread;
// Requested update region & requested flag
rfb::Region2D m_incr_rgn;
// Full screen rectangle
// rfb::Rect m_fullscreen;
// When the local display is palettized, it sometimes changes...
BOOL m_palettechanged;
// Information used in polling mode!
BOOL m_remoteevent;
// Clipboard data
char* m_clipboard_text;
//SINGLE WINDOW
BOOL m_use_NewSWSize;
BOOL m_NewSWDesktop;
int NewsizeW;
int NewsizeH;
// CURSOR HANDLING
BOOL m_cursor_update_pending;
BOOL m_cursor_update_sent;
// nyama/marscha - PointerPos
BOOL m_cursor_pos_changed;
BOOL m_use_PointerPos;
POINT m_cursor_pos;
// Modif sf@2002 - FileTransfer
BOOL m_fFileTransferRunning;
CZipUnZip32 *m_pZipUnZip;
char m_szFullDestName[MAX_PATH + 64];
char m_szFileTime[18];
char* m_pBuff;
char* m_pCompBuff;
HANDLE m_hDestFile;
DWORD m_dwFileSize;
DWORD m_dwNbPackets;
DWORD m_dwNbReceivedPackets;
DWORD m_dwNbBytesWritten;
DWORD m_dwTotalNbBytesWritten;
bool m_fFileDownloadError;
bool m_fFileDownloadRunning;
char m_szSrcFileName[MAX_PATH + 64]; // Path + timestring
HANDLE m_hSrcFile;
bool m_fEof;
DWORD m_dwNbBytesRead;
DWORD m_dwTotalNbBytesRead;
int m_nPacketCount;
bool m_fCompressionEnabled;
bool m_fFileUploadError;
bool m_fFileUploadRunning;
// sf@2004 - Delta Transfer
char* m_lpCSBuffer;
int m_nCSOffset;
int m_nCSBufferSize;
// Modif sf@2002 - Scaling
rfb::Rect m_ScaledScreen;
UINT m_nScale;
bool fNewScale;
bool m_fPalmVNCScaling;
bool fFTRequest;
// sf@2002 - DSM Plugin Rects alignement buffer
BYTE* m_pNetRectBuf;
bool m_fReadFromNetRectBuf; //
bool m_fWriteToNetRectBuf;
int m_nNetRectBufOffset;
int m_nReadSize;
int m_nNetRectBufSize;
// sf@2002
BYTE* m_pCacheZipBuf;
int m_nCacheZipBufSize;
BYTE* m_pRawCacheZipBuf;
int m_nRawCacheZipBufSize;
friend class TextChat;
TextChat *m_pTextChat; // Modif sf@2002 - Text Chat
bool m_fUltraViewer; // sf@2002
// sf@2005 - FTUserImpersonation
bool m_fFTUserImpersonatedOk;
DWORD m_lLastFTUserImpersonationTime;
//stats
int totalraw;
pSendinput Sendinput;
// Modif cs@2005
#ifdef DSHOW
HANDLE m_hmtxEncodeAccess;
#endif
};
#endif |
