[Web hacking] csrf-2

#!/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)

 

소스코드 분석...................................................................

 

/flag:
POST로 요청하면 ->
랜덤 생성한 session_id에 대한 username을 "admin"으로 설정 
http://127.0.0.1:8000/에 "sessionid" 쿠키 추가
http://127.0.0.1:8000/vuln?param=form값 에 GET 요청

cookie_dict = 추가할 쿠키 정보를 담고 있는 딕셔너리
driver = webdriver.Chrome -> 새로운 브라우저 세션을 엶
driver.get('http://example.com/')
driver.add_cookie(cookie_dict) -> http://example.com 도메인에 쿠키(cookie_dict)를 추가

/change_password: 
request cookie의 "sessionid"에 해당하는 비밀번호를 request의 "pw"값으로 설정

/login:
POST로 요청하면 -> 
form의 "username"값에 해당하는 pw == form의 "password"값 이면 -> 
form의 "username"값에 랜덤 "sessionid" 부여
"sessionid"을 쿠키로 설정하여 /로 redirection

/: 
request cookie의 "sessionid"값에 매치되는 username이 "admin"이면 -> 

flag 포함된 index.html 렌더링

 

풀이.............................................................

 

>> Flag는 어디 있는가? 

/로 들어갔을 때, 그때의 "sessionid"값의 username이 "admin"이면, flag가 포함되어 렌더링됨

 

>> 어떻게 username을 "admin"으로 설정하는가?

/flag에 POST 요청을 보내면,

session_id를 랜덤 생성한 뒤, 이에 대한 username을 "admin"으로 설정하고, 이를 cookie에 추가한다.

 

>> 어떻게 sessionid를 cookie에 실은 채로 /에 GET 요청을 보내는가? 

/login에서 로그인에 성공하면 sessionid를 쿠키로 설정하고 /로 redirection된다.

 

>> 어떻게 admin으로서 로그인에 성공하는가? 

username에는 "admin",

password에는 "admin"의 pw

를 입력한다.

 

>> "admin"의 pw를 어떻게 알수 있는가? 

/change_password에서 직접 pw를 설정할 수 있다.

즉 현재 cookie의 sessionid에 대한 비밀번호를 request의 pw값으로 설정한다.

따라서, /change_password에 대한 request의 pw값을 단순히 "1"로 설정해주면

"admin"의 비밀번호가 1이 된다.

즉 /change_password?pw=1에 GET 요청을 보내야 한다.

 

>> 어떻게 /change_password?pw=1에 GET 요청을 보내는가?

/flag에의 POST 요청을 통해 http://127.0.0.1:8000/vuln?param=form값 에 GET 요청을 보낼 수 있다.

그런데 만약 이 form값에 img 태그가 들어가면, 이것의 src에도 따로 GET 요청을 보내게 된다.

따라서, src에 /change_password?pw=1을 넣어주면 된다.

즉, /flag에서 form에 <img src=/change_password?pw=1>을 입력한다.

 

>> 이제 진짜 로그인 

/login에서 username에는 admin, password에는 1을 입력해주면

/로 redirection되고 거기에는 flag가 있다.