首页
论坛
课程
招聘
[原创]Socket网络模式下获取客户端磁盘文件并上传下载
2021-5-26 15:53 1990

[原创]Socket网络模式下获取客户端磁盘文件并上传下载

2021-5-26 15:53
1990

服务端是用delphi做的界面和写的代码,客户端只是响应,用的是C++写客户端。

 

下载文件:

 

(由于我的代码是采用的是数据流和消息异步,贴码的话,太多,这里我就用口述+重要代码截图的方式);

 

1,建立连接后,先获取客户端的磁盘信息,这里我们给客户端发送一个命令,比如“PKT_DISK”等类似消息,在客户端接受消息,判断是接受到的相等命令,我们在客户端执行相关函数或操作;

 

这里调用函数【GetLogicalDriveStrings】获取电脑磁盘信息;然后将获取到的磁盘信息send给服务端;

 

 

 

2,在服务端recv到客户的电脑磁盘信息后,添加到界面,客户端的磁盘就已枚举好;

 

3,获取磁盘下的目录,同样也是send指定命令和想要下载的路径给客户端,客户端recv到相对应命令之后,响应相关代码,使用函数【FindFirstFile】【FindNextFile】找相关路径的文件,需要定义一个WIN32_FIND_DATA结构,返回给此结构,判断结构的dwFileAttributes属性不为“.”和“..”,循环查找此路径文件,然后一一将文件名,属性,时间,大小发送给服务端,最后调用【FindClose】来关闭打开的句柄;

 

 

4,服务端收到文件名等信息后,可枚举出此目录下的所有文件。

 

 

5,有了文件名之后,我们就可以执行下载命令了,给客户端发送下载命令,将想要下载的路径名发给客户端,客户端收到消息之后,使用函数【CreateFile】打开收到的文件路径,来判断是不是一个可以正常打开的文件,打开成功后,使用函数【GetFileTime】【GetFileAttributes】【GetFileSizeEx】来获取文件的时间,属性,大小等等,获取成功后,send给服务端。

 

 

6,当客户端确定可以打开后,服务端成功接受到消息,读取客户端发送的文件属性,大小,时间,使用函数【CreateFile】来创建文件,来预接受要下载的数据;创建成功后,给客户端发送消息,告诉客户端,你可以给我发送这个需要下载的文件的数据了;

 

7,客户端调用函数【ReadFile】读取刚才打开的文件,循环的发送给服务端,直到读取的字节数为0,则调用函数【CloseHandle】来关闭句柄,避免泄露句柄;

 

 

8,服务端则循环的recv的客户端的数据,利用函数【WriteFile】将收到的数据写入刚才创建的文件中,直到收到的消息为结束,则退出循环;接收完毕后,使用函数【SetFileAttributes】【setfiletime】设置文件的属性和时间,使用【CloseHandle】关闭句柄;

 

至此,下载一个文件成功 ;

 

 

上传文件:

 

1,服务端使用函数【CreatrFIle】打开想要上传的文件,利用函数【GetFileTime】【GetFileAttributes】【GetFileSizeEx】获取文件的时间, 属性,大小;将文件的信息发送给客户端;等待接收客户端是否可以创建文件的命令;

 

 

 

2,客户端接受到消息后,使用函数【CreateFile】创建文件,创建成功后,发送消息给服务端,创建文件成功,可以给我发送需要上传的数据了;

 

3,当服务端收到客户端创建文件成功后,使用函数【ReadFile】循环读取刚才打开的文件,一一发送数据给客户端,直到读取文件的字节数为0时,则发送一个完成的命令给客户端,并使用函数【CloseHandle】关闭句柄;

 

4,客户端开启循环模式,一直接收服务端发送的数据,然后利用函数【WriteFile】给刚才创建成功的文件写入数据,直到收到的命令是完成,则给服务端再send一条命令,告诉服务端,接收上传的数据成功,使用函数【SetFileTime】设置文件时间,最后一步使用函数【CloseHandle】【SetFileAttributes】来关闭句柄和设置文件属性。

5,服务端这边再执行一次recv,收到成功后,且没有错误的情况下,至此,上传文件完成;

 

 

服务端和客户端的单个文件的上传下载就分享到这里;

 

如果想要下载/上传多个文件,则在两端增加一个变量,用于记录需要下载的个数,循环下载/上传文件;

 

如果想要下载一个文件夹,我采用的方式是比较简单的递归算法,枚举文件夹内的所有文件,如果遇到文件夹再次递归,最多再调用创建文件夹的函数就可以上传/下载文件夹了;

 

欢迎大家多多探讨,多多指导,多多交流;


[注意] 欢迎加入看雪团队!base上海,招聘安全工程师、逆向工程师多个坑位等你投递!

最后于 2021-5-26 22:09 被PE_Hacker编辑 ,原因:
收藏
点赞0
打赏
分享
最新回复 (1)
雪    币: 245
能力值: ( LV1,RANK:0 )
在线值:
发帖
回帖
粉丝
abel346 活跃值 2021-5-27 05:17
2
0
代码附件上传就行了嘛
游客
登录 | 注册 方可回帖
返回