이번에는 간단한 쿠키에 대한 문제이다.
쿠키로 인증 상태를 관린하는 간단한 로그인 서비스에 대해서, admin 계정으로 로그인에 성공하여 플래그를 획득해야한다.
문제 사이트의 코드는 아래와 같다.
#!/usr/bin/python3
from flask import Flask, request, render_template, make_response, redirect, url_for
app = Flask(__name__)
try:
FLAG = open('./flag.txt', 'r').read()
except:
FLAG = '[**FLAG**]'
users = {
'guest': 'guest',
'admin': FLAG
}
@app.route('/')
def index():
username = request.cookies.get('username', None)
if username:
return render_template('index.html', text=f'Hello {username}, {"flag is " + FLAG if username == "admin" else "you are not admin"}')
return render_template('index.html')
@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')) )
resp.set_cookie('username', username)
return resp
return '<script>alert("wrong password");history.go(-1);</script>'
app.run(host='0.0.0.0', port=8000)
make_response(redirect(url_for('index')) )
resp.set_cookie('username', username)
return resp
app.route함수의 아래쪽을 보면 이러한 코드가 보인다.
로그인시 쿠키를 'username':username값을 넣어서 만든다.
그리고 그 위에 app.route함수를 살펴보면
return render_template('index.html', text=f'Hello {username}, {"flag is " + FLAG if username == "admin" else "you are not admin"}')
이러한 코드가 나온다.
쿠키의 username값에 따라 결과를 다르게 출력하는 것이다.
즉 쿠키의 username값만 admin으로 바꿔서 변조해주면 Flag가 나오게 된다.