👇👇关注后回复 “进群” ,拉你进程序员交流群👇👇
来源于网络,侵删
一个突发奇想,当你把usb插进去的时候,能不能自动执行usb上的程序。查了一下,发现只有windows上可以,具体的大家也可以搜索(搜索关键词usb autorun)到。但是,如果我想,比如,当一个usb插入时,在后台自动把usb里的重要文件神不知鬼不觉地拷贝到本地或者上传到某个服务器,就需要特殊的软件辅助。
于是我心想,能不能用python写一个程序,让它在后台运行。每当有u盘插入的时候,就自动拷贝其中重要文件。
如何判断U盘的插入与否?
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_jpg/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5phCZPeBLhhLZ04EtLM5AHmzG2VZR00zRUt0tOOvMmNr7cuWrbVibUJNA/640?wx_fmt=jpeg&wxfrom=5&wx_lazy=1&wx_co=1)
首先我们打开电脑终端,进入/Volumes目录,这时候插入U盘,可以发现它被挂载在了这个目录之下,也就是说,我们只要在固定时间扫描这个目录,当这个目录有新文件夹出现的时候,很可能有U盘被插入了。
我的设计是这样的,用time.sleep(3)函数,让程序保持运行状态,并且每隔三秒查看一下/Volumes/目录,如果多出来文件夹,就将其拷贝到另外的文件夹。
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pFaHJxu2OWq4nLAH3JBYP8DJ2SEL7ONsZynHCTG6SaxQL5RdU0CSnLA/0?wx_fmt=png)
就像标题所示,我们真的只用了10行(其实是11行,凑个整:)完成了这个“病毒”。我们可以发现usb中的目录,在插入半分钟后全部躺在了home目录下了。
如何选择性的复制文件?
刚刚我们写了一个很简易的脚本测试了一下这个想法的可行性,但是还是有问题。刚才之所以能把U盘中所有文件很快复制进去,是因为U盘中只有两三个文件,大小不超过15M。如果目标U盘中有很多电影,音乐,这些我们并不需要的文件,我们的程序就应该能跳过它们,仅仅选择一些重要的比如.docx比如.ppt文件,或者仅仅复制最近修改过的那些文件,或者排除所有大小大于5M的文件。我们可以用python做到吗?当然!
还是举个例子吧。
我在某目录下创建了testwalk文件夹,里面有file123.txt三个文件,folder123三个文件夹,其中folder1中有文件file4.txt以及folder4
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pTSP5HnalLMM5TCa68nC7dckxfK3OjZqqz2g9MW9yoEiaJYY0wc67UTw/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
现在我们来测试一下
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pZauOgaJIHiafRibEyh9A7NibsvXQCnRIqbLia5bWYeNWVeIibgbzpYkCicwg/0?wx_fmt=png)
root存放的是当前位置,它会把./testwalk/下所有的文件夹作为根目录,往下搜索
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pXQJIEHxysoaPMqhcOf8tMYwkHRTyvroks1Jze2pOIWdou7cFY8xaSg/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
单独查看 dirs
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5psScxMyL7vribuDicIxbQHdKsIKLL0dO6lGfBaJ8ysvrlaRh14mhcYWibw/0?wx_fmt=png)
单独查看 files
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pFA2UTEfK5uibVdxSJkO11kPFZY3RQIZw88kYSfB8nOzM5fPHiaxPuH0g/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
好了,我们现在需要递归usb文件夹,找到所有的file,查看大小,如果小于,比如3M,就拷贝进home,大于就舍去。
shutil模块
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pdBCR5YekeWDYia3leQviaPNORbFT4gyIbFCrU2armgQqepdVeMUWOpeQ/0?wx_fmt=png)
现在我们拿刚才的文件夹举例子,如果想把file1.txt拷贝到folder2:
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5p9uR4FPjcLbRoeS3s5qnbnkdfDvqbGgn1TCN287TBKvhELiaoaOJZxEw/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
还有许多使用工具在shutil里面这里就不详述了。
os.path.getsize()判断大小
os.path.getsize(文件名)返回的是一个单位为byte的数值,如果用来查看文件大小,我们则需要手动写一个函数,将其换算成容易阅读的形式。
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5p6goYvB8dgm14eTJQoCeYhIEK0M0O0icXPMmdCyaiaG0Jss7KW2kcl7SA/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
这里我们只要选择文件大小小于3M的即可,3M = 3 * 1024kB = 3 * 1024*1024byte
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pyDt4ibejGuRS3066qbYbutQ6NdXJQ6TVsVLrgJSjy2HGm8R5QtOfAwQ/0?wx_fmt=png)
结合shutil.copy2就可以把选定大小的文件复制进我们的目标文件夹了
如何指定文件类型
这里就需要正则表达式来帮助我们了。
正则表达式内容很多,《python核心编程》中用了整整一章来讲,所以我们也不深入了。
如下,我们让指定文件后缀以及指定文件大小可以复制进我们的目标文件:
别忘了导入 re
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5p4HTAlMgCqLTNZMJ5Vg5ZAouib9K7AWjnhia7XtRw4ZAlYEGXw1ib1icPJw/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
用更加复杂的正则表达式可以更好地指定文件类型
根据修改时间筛选文件
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5p7zqzgdgxc2WPfzKLa9wSSUdoWa4HhKM5Fbe5e6GzOp6K8rJrByL2Hg/640?wx_fmt=png&wxfrom=5&wx_lazy=1&wx_co=1)
这时候我在目录下创建了一个文件叫做newfile
![十行代码--用Python写一个USB病毒](http://mmbiz.qpic.cn/mmbiz_png/HZW0wwFxbQDVqO3hO5bFtdTaqXv4lx5pia6NT9NSYTtDJs2uUYhsFHbqxk6OZCzcSTvYxKu6aQjOQaI28pkh5mw/640?wx_fmt=jpeg&wxfrom=5&wx_lazy=1&wx_co=1)
总之,对每一个文件进行修改时间的筛选可以只复制那些近期,或者特定时期修改或者添加过的文件,这个功能在特定情况下很有用。
总结
其实,这就是一个小程序,也谈不上是什么高深的恶意代码。我更想通过这个例子,展示python对于文件处理的强大能力,引发大家的学习热情。以上实现都是基于macos,linux应该一样,windows稍加修改也可以成功。以上技术仅供学习,切勿用于其他行为。
-End-
最近有一些小伙伴,让我帮忙找一些 面试题 资料,于是我翻遍了收藏的 5T 资料后,汇总整理出来,可以说是程序员面试必备!所有资料都整理到网盘了,欢迎下载!
![](http://mmbiz.qpic.cn/mmbiz_png/2HyIlGuO02ZBfriaa54Tf9DXWD0KR1OaQpT2sThK9FrRMsHtrgOcYWG6OGKic7pTorJmKPo0iaIKHHg9EbFLcqdcA/640?wx_fmt=png)
在看点这里
好文分享给更多人↓↓