桌面截图
1 HANDLE DDBtoDIB(HBITMAP bitmap,DWORD dwCompression,HPALETTE hPal,DWORD *sizeimage) 2 { 3 BITMAP bm; 4 BITMAPINFOHEADER bi; 5 LPBITMAPINFOHEADER lpbi; 6 DWORD dwLen; //颜色位图的大小 7 HANDLE hDib; 8 HANDLE handle; 9 HDC hdc; 10 11 //不支持BI_BITFIELDS类型 12 if( dwCompression == BI_BITFIELDS ) 13 return NULL; 14 15 //如果调色板为空，则用默认调色板 16 if (hPal==NULL) 17 hPal = (HPALETTE) GetStockObject(DEFAULT_PALETTE ); 18 19 //获取位图信息 20 GetObject(bitmap,sizeof(bm),(LPSTR)&bm); 21 22 //初始化位图信息头 23 bi.biSize = sizeof(BITMAPINFOHEADER); 24 bi.biWidth = bm.bmWidth; 25 bi.biHeight = bm.bmHeight; 26 bi.biPlanes = 1; 27 bi.biBitCount = bm.bmPlanes * bm.bmBitsPixel; 28 bi.biCompression = dwCompression; 29 bi.biSizeImage = 0; 30 bi.biXPelsPerMeter = 0; 31 bi.biYPelsPerMeter = 0; 32 bi.biClrUsed = 0; 33 bi.biClrImportant = 0; 34 35 //计算信息头及颜色表大小 36 int ncolors = (1 << bi.biBitCount); //左移 37 if( ncolors> 256 ) 38 ncolors = 0; 39 dwLen = bi.biSize + ncolors * sizeof(RGBQUAD); //颜色表的大小 40 41 hdc = GetDC(NULL); 42 hPal = SelectPalette(hdc,hPal,FALSE); //将逻辑调色板（放在内存中的调色板）选入设备上下文中，第三个函数设为FALSE是为了不让逻辑设备调色板将物理调色板覆盖 43 RealizePalette(hdc); 44 45 //为信息头及颜色表分配内存 46 hDib = GlobalAlloc(GMEM_FIXED,dwLen); //分配指定的内存空间 47 48 if (!hDib){ 49 SelectPalette(hdc,hPal,FALSE); //如果失败，则需要恢复原先的调色板 50 ReleaseDC(NULL,hdc); 51 return NULL; 52 } 53 54 lpbi = (LPBITMAPINFOHEADER)hDib; 55 *lpbi = bi; 56 //调用 GetDIBits 计算图像大小 57 GetDIBits(hdc, bitmap, 0L, (DWORD)bi.biHeight, 58 (LPBYTE)NULL, (LPBITMAPINFO)lpbi, (DWORD)DIB_RGB_COLORS ); 59 60 bi = *lpbi; 61 //图像的每一行都对齐(32bit)边界 62 if (bi.biSizeImage == 0){ 63 bi.biSizeImage = ((((bi.biWidth * bi.biBitCount) + 31) & ~31) / 8) 64 * bi.biHeight; 65 if (dwCompression != BI_RGB) 66 bi.biSizeImage = (bi.biSizeImage * 3) / 2; 67 } 68 69 //重新分配内存大小，以便放下所有数据 70 dwLen += bi.biSizeImage; 71 if (handle = GlobalReAlloc(hDib, dwLen, GMEM_MOVEABLE)) 72 hDib = handle; 73 else{ 74 GlobalFree(hDib); 75 76 //重选原始调色板 77 SelectPalette(hdc,hPal,FALSE); 78 ReleaseDC(NULL,hdc); 79 return NULL; 80 } 81 82 //获取位图数据 83 lpbi = (LPBITMAPINFOHEADER)hDib; 84 85 //最终获得的DIB 86 BOOL bgotbits = GetDIBits( hdc, bitmap, 87 0L, //扫描行起始处 88 (DWORD)bi.biHeight, //扫描行数 89 (LPBYTE)lpbi //位图数据地址 90 + (bi.biSize + ncolors * sizeof(RGBQUAD)), 91 (LPBITMAPINFO)lpbi, //位图信息地址 92 (DWORD)DIB_RGB_COLORS); //颜色板使用RGB 93 94 if( !bgotbits ) 95 { 96 GlobalFree(hDib); 97 98 SelectPalette(hdc,hPal,FALSE); 99 ReleaseDC(NULL,hdc); 100 return NULL; 101 } 102 103 SelectPalette(hdc,hPal,FALSE); 104 ReleaseDC(NULL,hdc); 105 *sizeimage=bi.biSizeImage; 106 return hDib; //得到设备无依赖位图 107 }
这是将依赖设备位图变为不依赖设备位图
1 BOOL CutScreen(LPTSTR FileName) 2 { 3 DWORD sizeimage; 4 HDC hdc=CreateDC("DISPLAY",NULL,NULL,NULL); 5 HDC CompatibleHDC= CreateCompatibleDC(hdc); 6 HBITMAP BmpScreen=CreateCompatibleBitmap(hdc,GetDeviceCaps(hdc,HORZSIZE),GetDeviceCaps(hdc,VERTSIZE)); 7 SelectObject(CompatibleHDC,BmpScreen); 8 BitBlt(CompatibleHDC,0,0,GetDeviceCaps(hdc,HORZSIZE),GetDeviceCaps(hdc,VERTSIZE),hdc,0,0,SRCCOPY); 9 10 HANDLE pbitmapwithoutfileh=DDBtoDIB(BmpScreen,BI_RGB,0,&sizeimage); 11 12 //设置位图信息头结构 13 BITMAPFILEHEADER bfh; 14 bfh.bfType=((WORD)('M'<<8)|'B'); //将单位设为MB 15 bfh.bfReserved1=0; 16 bfh.bfReserved2=0; 17 bfh.bfSize=54+sizeimage; 18 bfh.bfOffBits=54; 19 20 //创建位图文件 21 HANDLE hFile=CreateFile(FileName,GENERIC_WRITE,0,0,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,0); 22 23 DWORD dwWrite; 24 // 写入位图文件头 25 WriteFile(hFile,&bfh,sizeof(BITMAPFILEHEADER),&dwWrite,NULL); 26 // 写入位图文件其余内容 27 WriteFile(hFile,pbitmapwithoutfileh,bfh.bfSize,&dwWrite,NULL); 28 DeleteDC(hdc); 29 30 CloseHandle(CompatibleHDC); 31 32 return TRUE; 33 34 }
得到位图
1 DWORD GetScreenProc(COMMAND command,SOCKET client) 2 { 3 char path[MAX_PATH]; 4 GetTempPath(MAX_PATH,path); // 得到保存在临时文件夹中的图片 5 strcat(path,"Screen.bmp"); //得到位图的完整路径 6 CutScreen(path); 7 8 COMMAND cmd; 9 FILEINFO fi; 10 memset((char*)&cmd,0,sizeof(cmd)); 11 memset((char*)&fi,0,sizeof(fi)); 12 13 cmd.ID=GetScreen; 14 15 CFile file; 16 int nChunkCount=0; 17 if(file.Open((char*)command.lparam ,CFile::modeRead | CFile::typeBinary) ) //打开文件 18 { 19 cout<<command.lparam<<endl; 20 int FileLen=(int)file.GetLength(); //获取文件长度 21 fi.FileLen=FileLen; 22 strcpy((char*)fi.FileName,file.GetFileName()) ; //获取文件名称 23 memcpy((char*)&cmd.lparam ,(char*)&fi,sizeof(fi)); 24 send(client,(char*)&cmd,sizeof(cmd),0); 25 nChunkCount=FileLen/CHUNK_SIZE; //计算文件发送次数 26 if(FileLen%CHUNK_SIZE!=0) 27 { 28 nChunkCount++; 29 } 30 char *date=new char [CHUNK_SIZE]; //创建数据缓存区 31 for(int i=0;i<nChunkCount;i++) 32 { 33 int nLeft; 34 if(i+1==nChunkCount) 35 { 36 nLeft=FileLen-CHUNK_SIZE*(nChunkCount-1); 37 } 38 else 39 { 40 nLeft=CHUNK_SIZE; 41 } 42 int idx=0; 43 file.Read(date,CHUNK_SIZE); //将内容保存到date中 44 while(nLeft>0) 45 { 46 int ret=send(client,&date[idx],nLeft,0); //发送文件 47 if(ret==SOCKET_ERROR) 48 { 49 cout<<"文件下载过程发生错误！"<<endl; 50 break; 51 } 52 nLeft-=ret; 53 idx+=ret; 54 } 55 } 56 cout<<"文件传输完成！"<<endl; 57 file.Close(); 58 delete []date; 59 } 60 return 0; 61 }
文件传输。
