[De1CTF 2019]SSRF Me（flask框架）

题目首先提示了flag的位置，结合题目名字ssrf，可以得到一个大概的思路。

---

然后打开题目，直接给了源码：

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

这道题还是看了大佬的题解，了解了大概的步骤，然后自己分析一遍才做出来的。刚开始看不懂代码，去B站看了讲flask的教学视频，后来才慢慢的看懂。

一，首先看到这种web模板代码题，先看路由，然后从路由出发一步步分析

这里有三个路由：

• /geneSign
• /De1ta
• /

二，首先看跟路由/

作用就是把源码给我们。

三，然后分析路由/geneSign

首先接收了一个我们传进去的参数param，然后定义了一个参数action，它的值是scan。接下来把这两个参数传给了函数getSign(action, param)，它的作用是返回 md5(secert_key + param + action)的值。这里的secert_key我们是不知道的，param使我们能控制的，action=scan。

四，接着看路由/De1ta

定义了一个challenge()函数，通过cookie接受了两个参数action和sign，通过get方法得到param的值，这里的三个参数是可控的。

然后把param参数通过waf()函数，

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

可见，这里过滤了gopher和file这两个协议。

五，最后构造了一个Task类对象，并且执行它的Exec方法

    task = Task(action, param, sign, ip)
    return json.dumps(task.Exec())

---

我们一步一步分析这个对象，首先是__init__()

创建了一个名字是 md5(ip地址) 的目录。

---

然后是Exec()方法

    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

1，sign传给checkSign()函数

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

需要满足md5(secert_key + param + action)==sign

2，判断 参数action里有没有scan，有的话在上面创建的目录下创建一个result.txt ，然后通过scan()函数把名字为param的网址里的内容写到result.txt中，由于param是可控的，所以很容易想到这里把flag.txt传给param。

3，判断 参数action里有没有read，若有，读取result.txt的内容赋值给result。

六，总结思路

分析到这里梳理梳理思路，

路由/geneSign就是工具人，用以得到sign的值。

关键是路由/De1ta。

想要读取flag先要满足checkSign()的条件：if (getSign(self.action, self.param) == self.sign):这三个参数都是我们传进去的。

首先看左边：
要想读取到flag，self.param肯定等于flag.txt。还要满足后面两个if的条件self.action要包含read和scan
然后看右边：
后面的self.sign中的param是get得到的，action等于scan。

即：
md5(secert_key +flag.txt + action)=md5(secert_key +param + scan)
可见，要使上式成立，只要action=readscan，param=flag.txtread 即可。

七，利用payload得到flag

首先用路由/geneSign中获取if (getSign(self.action, self.param) == self.sign):中sign的值，传值param=flag.txtread

payload:
http://3d3e7fd7-9830-4021-9542-a41c112ec88c.node3.buuoj.cn/geneSign?param=flag.txtread
然后用路由/De1ta读取flag,txt，传值：

---

收获:这道题最开始完全看不懂代码，然后去B站上学flask框架，看得懂代码之后，看一遍题解，自己再慢慢一步一步分析，最后解出flag，做出来的一瞬间感觉特别有成就感，更重要的是学会了很多知识，hhhh。希望每天都有进步！