[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)):
            os.mkdir(self.sandbox)

    def Exec(self):
        result = {}
        result['code'] = 500
        if (self.checkSign()):
            if "scan" in self.action:   #也就是说action要为“readscan”
                tmpfile = open("./%s/result.txt" % self.sandbox, 'w')
                resp = scan(self.param) #传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


@app.route("/geneSign", methods=['GET', 'POST'])
def geneSign():
    param = urllib.unquote(request.args.get("param", ""))   #接收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"))  #获取cookie参数action和sign，get参数param
    ip = request.remote_addr    #客户端的ip
    if (waf(param)):    #不能用用gopher和file协议
        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]    #设置param=flag.txt
    except:
        return "Connection Timeout"


def getSign(action, param):     #mde加密一个secret_key+我们传的param+scan
    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 Falsacs.get("param", ""))   #接收param参数
    action = "scan"
    return getSign(action, param)

getSign用MD5加密了一下

def getSign(action, param):     #mde加密一个secret_key+我们传的param+scan
    return hashlib.md5(secert_key + param + action).hexdigest()

把三个参数传入Task类里，在调用Task类里的Exec函数，

@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"))  #获取cookie参数action和sign，get参数param
    ip = request.remote_addr    #客户端的ip
    if (waf(param)):    #不能用用gopher和file协议
        return "No Hacker!!!!"a'c
    task = Task(action, param, sign, ip)
    return json.dumps(task.Exec())

在Exec可以看出，action判断要包含scan和read这两个字符

 def Exec(self):
        result = {}
        result['code'] = 500
        if (self.checkSign()):
            if "scan" in self.action:   #也就是说action要为“readscan”
                tmpfile = open("./%s/result.txt" % self.sandbox, 'w')
                resp = scan(self.param) #传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

self.checkSign()，虽然action要包含scan和read这两个字符，在getSign中如果设置param为flag.txtread，那么合起来就是flag.txtreadscan，也就是可以先访问/geneSign获取cookie，然后再通过/De1ta传值

def getSign(action, param):     #mde加密一个secret_key+我们传的param+scan
    return hashlib.md5(secert_key + param + action).hexdigest()

resp = scan(self.param)，

def scan(param):
    socket.setdefaulttimeout(1)
    try:
            return urllib.urlopen(param).read()[:50]    #设置param=flag.txt
    except:
        return "Connection Timeout"

先获取MD5值

再传参获取flag