[Dreamhack] csrf-1

풀이

문제 페이지에 들어가면 위와 같은 화면이 나온다. 

#!/usr/bin/python3
from flask import Flask, request, render_template
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import urllib
import os

app = Flask(__name__)
app.secret_key = os.urandom(32)

try:
    FLAG = open("./flag.txt", "r").read()
except:
    FLAG = "[**FLAG**]"


def read_url(url, cookie={"name": "name", "value": "value"}):
    cookie.update({"domain": "127.0.0.1"})
    try:
        service = Service(executable_path="/chromedriver")
        options = webdriver.ChromeOptions()
        for _ in [
            "headless",
            "window-size=1920x1080",
            "disable-gpu",
            "no-sandbox",
            "disable-dev-shm-usage",
        ]:
            options.add_argument(_)
        driver = webdriver.Chrome(service=service, options=options)
        driver.implicitly_wait(3)
        driver.set_page_load_timeout(3)
        driver.get("http://127.0.0.1:8000/")
        driver.add_cookie(cookie)
        driver.get(url)
    except Exception as e:
        driver.quit()
        print(str(e))
        # return str(e)
        return False
    driver.quit()
    return True


def check_csrf(param, cookie={"name": "name", "value": "value"}):
    url = f"http://127.0.0.1:8000/vuln?param={urllib.parse.quote(param)}"
    return read_url(url, cookie)


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/vuln")
def vuln():
    param = request.args.get("param", "").lower()
    xss_filter = ["frame", "script", "on"]
    for _ in xss_filter:
        param = param.replace(_, "*")
    return param


@app.route("/flag", methods=["GET", "POST"])
def flag():
    if request.method == "GET":
        return render_template("flag.html")
    elif request.method == "POST":
        param = request.form.get("param", "")
        if not check_csrf(param):
            return '<script>alert("wrong??");history.go(-1);</script>'

        return '<script>alert("good");history.go(-1);</script>'


memo_text = ""


@app.route("/memo")
def memo():
    global memo_text
    text = request.args.get("memo", None)
    if text:
        memo_text += text
    return render_template("memo.html", memo=memo_text)


@app.route("/admin/notice_flag")
def admin_notice_flag():
    global memo_text
    if request.remote_addr != "127.0.0.1":
        return "Access Denied"
    if request.args.get("userid", "") != "admin":
        return "Access Denied 2"
    memo_text += f"[Notice] flag is {FLAG}\n"
    return "Ok"


app.run(host="0.0.0.0", port=8000)

제공된 문제 코드를 살펴보면 xss 공격이 발생하지 않도록 키워드 치환하는 부분이 있고, 로컬호스트에서 `userid==admin`으로 접속했을 때 notice_flag 페이지에서 flag를 확인할 수 있단 것을 확인할 수 있다. 그래서 우리는 유저가 직접 해당 페이지에 접근해서 flag를 memo 페이지에 출력하도록 csrf 공격을 해야한다. 

@app.route("/vuln")
def vuln():
    param = request.args.get("param", "").lower()
    xss_filter = ["frame", "script", "on"]
    for _ in xss_filter:
        param = param.replace(_, "*")
    return param

코드의 이 부분을 살펴보면 frame, script, on이라는 키워드는 필터링하지만 그 외에 <,>, img 등은 필터링하지 않는 것을 볼 수 있다. 그러면 이를 활용해서 공격 코드를 작성할 수 있을 것 같다.

<img src="/admin/notice_flag?userid=admin" />

이렇게 공격 코드를 작성하고 flag 페이지에서 입력하면, memo 페이지에서 아래와 같이 flag를 확인할 수 있다.

 

 

++++

여기서는 문제에 대놓고 csrf 라고 적혀있었지만, 원래대로라면 csrf 취약점이 발생할 수 있는지를 확인해야한다.

드림핵 툴로 request bin을 생성하고 이 주소에 접속하는 공격 코드를 작성해서 flag 페이지에 입력해봤다.

<img src="https://hvazywy.request.dreamhack.games">

그리고 확인하면 접속 요청이 온 것을 확인할 수 있다!