Python黑客编程2 入门demo--zip暴力破解

上一篇文章，我们在Kali Linux中搭建了基本的Python开发环境，本篇文章为了拉近Python和大家的距离，我们写一个暴力破解zip包密码的小程序。这个例子来自于《Voilent Python》一书，这也是一本python黑客编程的入门书，推荐大家看一看。

照顾没有接触过Python编程的同学，行文可能会有些啰嗦。

废话少说，我们进入正题。

2.1 准备基本材料

在/home/ziptest/目录下，我创建了两个文件，一个test.zip，是一个设置了密码的zip包，密码为456789。

dict.txt文件是一个字典文件，简单的配置了几个密码。

下面我们打开开发工具，开始编写测试代码。

2.2 zipfile

在python中操作zip文件，最简单的方式就是使用zipfile模块，使用该模块可以用来判断一个文件是否是压缩文件，创建、解压文件，获取zip文件的元数据信息。可以使用python的help方法查看该模块的使用方法。

这里我们首先关注下ZipFile类。

该类用来打开，读取，修改，解压zip文件。我们想要操作一个zip文件，第一步就是初始化ZipFile实例。下面我们打开我们准备好的text.zip文件。

import zipfile

zFile = zipfile.ZipFile("/home/test.zip");

我们只传了一个路径参数进去，从帮助文档我们可以看到，后面三个参数都有默认值，这里我们使用默认值就够了。

下面我们关注下ZipFile类的extractall方法。

extractall方法，是把压缩包里面的内容都解压出来，三个参数，path是解压的路径，members是需要解压出来的文件，pwd是密码。

现在我们可以测试下文件解压了。

import zipfile

zFile = zipfile.ZipFile("/home/ziptest/test.zip");

zFile.extractall("/home/",pwd="456789");

运行这个脚本。

密码正确的话，会正常解压文件。如果密码不正确会出现什么情况呢？我们在代码中输入一个错误的密码。

import zipfile

zFile = zipfile.ZipFile("/home/ziptest/test.zip");

zFile.extractall(path="/home/ziptest",pwd="4567890");

结果如下：

程序会抛出“bad password”的异常。

我们可以通过捕获异常，测试多个密码。

2.3 读取字典文件

在Python中打开文件，使用open方法，这是一个内置方法，查看open的帮助文档，可以看到该方法的参数说明。

open方法返回一个file对象，利用file对象，我们可以读取文件的具体内容。下面我们在代码中测试一下。

import zipfile

passFile = open('/home/ziptest/dict.txt');

for line in passFile.readlines():

password = line.strip('\n');

print(password);

运行结果如下：

下面我们利用读取到的密码来暴力测试zip文件。

import zipfile

zFile = zipfile.ZipFile("/home/ziptest/test.zip");

passFile = open('/home/ziptest/dict.txt');

for line in passFile.readlines():

password = line.strip('\n');

try:

zFile.extractall(path="/home/ziptest",pwd=password);

print("password is:"+password);

exit();

except:

pass;

在上面的代码中，我们使用try—except进行异常捕获，密码不正确的时候，程序跳过继续执行。密码正确的时候打印密码，终止程序。运行结果如下：

到目前为止，我们的脚本已经具备了暴力破解zip文件的功能，大家可以看到只有十几行的代码。下面我们为了提升程序的可用性，重构一下这个脚本，利用函数划分功能。

2.4 重构，用函数划分功能

首先定义一个extractFile函数，该函数接收三个参数，一个zipfile对象，解压的目标路径，密码，如果解密成功则返回密码。

def extractFile(toPath,zFile,password):

try:

zFile.extractall(path=toPath,pwd=password);

return password;

except Exception,e:

return;

下面我们再声明一个main方法。

def main():

zFile = zipfile.ZipFile("/home/ziptest/test.zip");

passFile = open('/home/zip/test/dict.txt');

for line in passFile.readlines():

password = line.strip('\n');

guess = extractFile("/home/",zFile,password);

if guess:

print('scucess'+password);

exit();

在main方法中，首先初始化了zipfile对象，然后打开字典文件，循环读取密码，传给extractFile方法调用。

分离了两个方法之后，我们需要在程序的入口处调用main方法，完整代码如下：

import zipfile

def extractFile(toPath,zFile,password):

try:

zFile.extractall(path=toPath,pwd=password);

print('scucess'+password);

return password;

except Exception,e:

return;

def main():

zFile = zipfile.ZipFile("/home/ziptest/test.zip");

passFile = open('/home/ziptest/dict.txt');

for line in passFile.readlines():

password = line.strip('\n');

guess = extractFile("/home/",zFile,password);

if guess:

print('scucess'+password);

exit();

if __name__=='__main__':

main();

这样一来，代码清晰了很多，但是我们更换zip文件和字典文件的时候，还是需要修改代码，很不方便，正常的程序都应该可以传递参数的，ok，下面我们引入optparse库。

2.5 optparse

Python 有两个内建的模块用于处理命令行参数：

一个是 getopt，《Deep in python》一书中也有提到，只能简单处理命令行参数；

另一个是 optparse，它功能强大，而且易于使用，可以方便地生成标准的、符合Unix/Posix 规范的命令行说明。

首先先引入optparse模块，然后强制添加两个参数，zip文件名和字典文件名。先看代码：

def main():

parser = optparse.OptionParser("usage%prog "+\

"-f <zipfile> -d <dictFile>");

parser.add_option('-f',dest='zname',type='string',help='specify zip file');

parser.add_option('-d',dest='dname',type='string',help='specify dict file');

(options,args)=parser.parse_args();

if(options.zname==None)|(options.dname==None):

print parse.usage;

exixt(0);

else:

zname=options.zname;

dname=options.dname;

zFile = zipfile.ZipFile(zname);

passFile = open(dname);

for line in passFile.readlines():

password = line.strip('\n');

guess = extractFile("/home/",zFile,password);

if guess:

print('scucess'+password);

exit(0);

首先初始化一个OptionParser对象，然后添加两个选项——“-f”和“-d”。之后在程序运行的时候通过parse_args方法获取输入的参数，如果参数为空，则打印使用方法，退出程序。

下面使用终端来测试这个程序。

无参数情况下：

输入参数：

2.6 小结

这个程序本身并没有什么难点，但是通过这个入门案例，我们可以体验Python编程的基本方法，如何查看帮助文档，如何引入模块，初始化对象，操作zip文件，打开本地文件，命令行参数设置。

原书中，作者有为程序增加多线程的例子，但是实在是太过于粗暴，容易误导大家都多线程的使用，所以我这篇文章直接略过了。

之后的文章，我们会在此基础上继续深入，敬请期待。

网络安全交流qq群：147098303

更多python黑客编程内容，关注我的微信订阅号，xuanhun521，为您持续推送：