[De1CTF 2019]SSRF Me

#! /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')

提示中给出了flag在./flag.txt里面。

首先Task类中有Exec函数，我们访问De1ta页面，传参，经过waf处理后，构造一个Task的对象。

我们可以看到在scan函数中用到了urlopen函数，这个可能是本题的关键，响应题目中的SSRF。

我们查看那里调用了scan函数，会发现旨在Task中的Exec函数里调用了一次，而且使用了我们可控的self.param参数。

这里是没有对self.param进行过滤的，直接就传给了urlopen函数。waf函数也只是检测我们是否使用gopher或file读取他的本地文件。但是，我们直接写入文件路径也是可以读取文件的。flag在./flag.txt中，这里是唯一可以读取文件的位置，这里应该就是破题的关键。

这个位置如果能够成功读取flag.txt，则将其中的内容写入result.txt中。

如果read在self.action中，则将result读取并保存到result['data']中，Exec函数的最后返回了result而在challenge函数中，返回了json.dumps(task.Exec()），也就是此处的result。

思路：先进入/De1ta中的challenge函数，在Exec中的scan部分中将flag.txt的内容存入result.txt，然后从read部分中将其存到result字典中读出，再以json形式返回到客户端，我们就能得到flag。

要想实现上面的思路，我们首先要通过self.checkSign()，

然后checkSign函数需要getSign(self.action,self.param) ==self.sign。

 

到现在，我们还面临一个问题是：我们还不知道secret_key，但是在/geneSign绑定的geneSign中，会返回getSign('scan', param)，通过这个返回值，直接在geneSign()中获取我们想要的字符串的哈希值。

在checkSign()函数的比较中，左边是getSign(self.action, self.param)，右边是getSign('scan', param1)，再进一步，左边是md5(key + self.param + self.action)，考虑到param会传入urlopen()，我们可以写成md5(key + 'flag.txt' + self.action)，其中，为了保证Exec()函数中scan部分和read部分都能被执行，self.action必须有'readscan'或'scanread'这样的子串。等号右边是md5(key + param1 + 'scan')，依此判断，self.action应包含'readscan'子串，可以将self.action定为'readscan'。

这样的话，等号左边为md5(key + 'flag.txt' + 'readscan')，等号右边的未定的参数也确定了，即md5(key + 'flag.txtread' + 'scan')。param1为'flag.txtread'依此传参，可得一个哈希值，这就是我们传给self.sign的值。

/geneSign?param=flag.txtread


#fcc344926d6f70e7f12f009f1ac84e0c

然后，传参给De1ta，修改cookie获取到最终flag。