[Dreamhack] xss-2

풀이

문제 페이지에 접속하면 위와 같다.

#!/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()
        # 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.route("/")
def index():
    return render_template("index.html")


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


@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_xss(param, {"name": "flag", "value": FLAG.strip()}):
            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", "")
    memo_text += text + "\n"
    return render_template("memo.html", memo=memo_text)


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

코드를 살펴보면 이번에는 xss-1 때의 문제와 다르게 param을 입력받으면  `render_template`함수를 사용해 처리해주고 있다.

`<script>location.href = "/memo?memo=" + document.cookie;</script>`

저번에 공격에 성공했던 코드를 입력해봐도 아무일이 일어나지 않는 걸 확인할 수 있다. 

 

그러면 이번에는 저 코드를 다른 방식으로 전달해서 쿠키를 탈취해야한다. 

<img src="one" onerror="location.href='/memo?memo='+document.cookie">

이번에는 <img> 태그를 이용해서 코드를 작성해봤다. "one"이라는 이미지 소스 로딩이 실패했을 때 onerror 이벤트를 처리하도록 하였다.

flag 페이지에서 입력하고 flag페이지를 확인하면, 이미지 로딩이 되지 않은 것을 확인할 수 있다. 

그리고 memo 페이지에 들어가면 쿠키 정보가 출력된 것을 확인할 수 있다.