[De1CTF 2019]SSRF Me 1（代码审计）

#! /usr/bin/env python
#encoding=utf-8
from flask import Flask
from flask import request
import socket
import hashlib
import urllib
import sys
import os
import json
reload(sys)
sys.setdefaultencoding('latin1')

app = Flask(__name__)

secert_key = os.urandom(16)


class Task:
    def __init__(self, action, param, sign, ip):
        self.action = action
        self.param = param
        self.sign = sign
        self.sandbox = md5(ip)
        if(not os.path.exists(self.sandbox)):          #SandBox For Remote_Addr
            os.mkdir(self.sandbox)

    def Exec(self):
        result = {}
        result['code'] = 500
        if (self.checkSign()):
            if "scan" in self.action:
                tmpfile = open("./%s/result.txt" % self.sandbox, 'w')
                resp = scan(self.param)
                if (resp == "Connection Timeout"):
                    result['data'] = resp
                else:
                    print resp
                    tmpfile.write(resp)
                    tmpfile.close()
                result['code'] = 200
            if "read" in self.action:
                f = open("./%s/result.txt" % self.sandbox, 'r')
                result['code'] = 200
                result['data'] = f.read()
            if result['code'] == 500:
                result['data'] = "Action Error"
        else:
            result['code'] = 500
            result['msg'] = "Sign Error"
        return result

    def checkSign(self):
        if (getSign(self.action, self.param) == self.sign):
            return True
        else:
            return False


#generate Sign For Action Scan.
@app.route("/geneSign", methods=['GET', 'POST'])
def geneSign():
    param = urllib.unquote(request.args.get("param", ""))
    action = "scan"
    return getSign(action, param)


@app.route('/De1ta',methods=['GET','POST'])
def challenge():
    action = urllib.unquote(request.cookies.get("action"))
    param = urllib.unquote(request.args.get("param", ""))
    sign = urllib.unquote(request.cookies.get("sign"))
    ip = request.remote_addr
    if(waf(param)):
        return "No Hacker!!!!"
    task = Task(action, param, sign, ip)
    return json.dumps(task.Exec())
@app.route('/')
def index():
    return open("code.txt","r").read()


def scan(param):
    socket.setdefaulttimeout(1)
    try:
        return urllib.urlopen(param).read()[:50]
    except:
        return "Connection Timeout"



def getSign(action, param):
    return hashlib.md5(secert_key + param + action).hexdigest()


def md5(content):
    return hashlib.md5(content).hexdigest()


def waf(param):
    check=param.strip().lower()
    if check.startswith("gopher") or check.startswith("file"):
        return True
    else:
        return False


if __name__ == '__main__':
    app.debug = False
    app.run(host='0.0.0.0')

题目提示：

先看代码：
@app.route(’/’)

scan()函数：（无关紧要，补充一下知识）
socket.setdefaulttimeout()
python程序根据url从互联网上批量下载图片时，设置HTTP或Socket超时，来防止爬虫爬取某个页面时间过长，导致程序卡置不前。
import socket
socket.setdefaulttimeout(t)
t：代表经过t秒后，如果还未下载成功，自动跳入下一次操作，此次下载失败。

getSign()函数
对secert_key + param + action进行md5加密
hash.digest()
返回摘要，作为二进制数据字符串值
hash.hexdigest()
返回摘要，作为十六进制数据字符串值

waf()函数
waf函数找到以gopher或者file开头的，所以在这里过滤了这两个协议，使我们不能通过协议读取文件

@app.route("/geneSign", methods=[‘GET’, ‘POST’])

geneSign()接收param，令action = “scan”，传入getSign()形成secert_key + param + action，进行md5加密

@app.route(’/De1ta’,methods=[‘GET’,‘POST’])

get方法传入param参数值，在cookie里面传递action和sign的值，将传递的param通过waf过滤，然后传入task的Exec()，利用json得到回显结果

exec()函数中先判断getSign()的结果是否与sign相同，然后若scan in cation，则将读取到的内容写入result.txt;若read in
action则将result.txt的内容显示出来

payload：
1.在geneSign里的getsign函数,他是用param和action做拼接的！而action是固定的scan ，也就是说，我们先使用geneSign，传参为flag.txtread，这样只会param+action拼接为 flag.txtreadscan，生成sign保存下来

2.再访问De1ta页面进行参数传递，拿到flag,因为我们能够通过字符串拼接绕过checkSign()方法,就可以拿到flag啦