[Dreamhack] csrf-2

csrf-1과 대부분의 인터페이스는 동일하지만 사이트에 들어가 보면 로그인을 해 달라는 문장이 크게 나온다. 코드를 살펴보자.

 

#!/usr/bin/python3
from flask import Flask, request, render_template, make_response, redirect, url_for
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**]"

users = {
    'guest': 'guest',
    'admin': FLAG
}

session_storage = {}

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():
    session_id = request.cookies.get('sessionid', None)
    try:
        username = session_storage[session_id]
    except KeyError:
        return render_template('index.html', text='please login')

    return render_template('index.html', text=f'Hello {username}, {"flag is " + FLAG if username == "admin" else "you are not an admin"}')


@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", "")
        session_id = os.urandom(16).hex()
        session_storage[session_id] = 'admin'
        if not check_csrf(param, {"name":"sessionid", "value": session_id}):
            return '<script>alert("wrong??");history.go(-1);</script>'

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


@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    elif request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        try:
            pw = users[username]
        except:
            return '<script>alert("not found user");history.go(-1);</script>'
        if pw == password:
            resp = make_response(redirect(url_for('index')) )
            session_id = os.urandom(8).hex()
            session_storage[session_id] = username
            resp.set_cookie('sessionid', session_id)
            return resp 
        return '<script>alert("wrong password");history.go(-1);</script>'


@app.route("/change_password")
def change_password():
    pw = request.args.get("pw", "")
    session_id = request.cookies.get('sessionid', None)
    try:
        username = session_storage[session_id]
    except KeyError:
        return render_template('index.html', text='please login')

    users[username] = pw
    return 'Done'

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

 

 

코드 상단에 정의된 users 딕셔너리에 guest와 admin의 이름으로 키와 밸류, 즉 아이디와 비밀번호가 저장된 것을 볼 수 있다. 코드에서 찾아볼 수 있는 5개의 엔드 포인트가 각각 어떤 기능을 수행하는지 정리해 보자.

 

/	인덱스 페이지, please login 또는 로그인 상태 또는 플래그 출력
/vuln	xss에 쓰이는 명령어를 필터링, 요청값을 그대로 return하므로 공격에 취약
/flag	GET : url 입력 폼 / POST : admin에 해당하는 session_id 저장
/login	GET : 로그인 폼 / POST : username, password 입력값으로 로그인
/change_password	session에 저장된 이용자의 패스워드(pw) 변경

 

@app.route("/")
def index():
    session_id = request.cookies.get('sessionid', None)
    try:
        username = session_storage[session_id]
    except KeyError:
        return render_template('index.html', text='please login')

    return render_template('index.html', text=f'Hello {username}, {"flag is " + FLAG if username == "admin" else "you are not an admin"}')

 

이 코드에서는 쿠키에서 sessionid를 가져와 사용자를 인증하고, admin일 경우 플래그를 출력하도록 한다. 그런데 우리는 지금 admin의 비밀번호를 알 수 없는 상황. /change_password 엔드포인트를 공략하여 admin의 비밀번호를 변경하고 그 계정으로 로그인에 성공하면, flag가 화면에 나올 것 같다.

 

csrf-1과 태그를 필터링하는 내용에 다름이 없으므로 해당 문제에서 사용했던 <img> 태그를 그대로 사용해 볼 수 있을 것 같다.

 

<img src="/change_password?pw=0000">

 

 

/change_password 페이지에서 pw, 즉 비밀번호를 0000으로 바꾸는 페이로드를 보낸다.

 

 

flag 페이지에 위 페이로드를 입력한 뒤 login에서 다시 admin에 비밀번호 0000을 입력하고 로그인하면 알아서 플래그를 화면에 띄워 준다.