#!/usr/bin/env python3
import os
import shutil

from flask import Flask, request, render_template, redirect

from flag import FLAG

APP = Flask(__name__)

UPLOAD_DIR = 'uploads'

@APP.route('/')
def index():
    files = os.listdir(UPLOAD_DIR)
    return render_template('index.html', files=files)

@APP.route('/upload', methods=['GET', 'POST'])
def upload_memo():
    if request.method == 'POST':
        filename = request.form.get('filename')
        content = request.form.get('content').encode('utf-8')

        if filename.find('..') != -1:
            return render_template('upload_result.html', data='bad characters,,')

        with open(f'{UPLOAD_DIR}/{filename}', 'wb') as f:
            f.write(content)

        return redirect('/')

    return render_template('upload.html')

@APP.route('/read')
def read_memo():
    error = False
    data = b''

    filename = request.args.get('name', '')

    try:
        with open(f'{UPLOAD_DIR}/{filename}', 'rb') as f:
            data = f.read()
    except (IsADirectoryError, FileNotFoundError):
        error = True

    return render_template('read.html',
                           filename=filename,
                           content=data.decode('utf-8'),
                           error=error)

if __name__ == '__main__':
    if os.path.exists(UPLOAD_DIR):
        shutil.rmtree(UPLOAD_DIR)

    os.mkdir(UPLOAD_DIR)

    APP.run(host='0.0.0.0', port=8000)

우선 코드를 확인해보겠습니다. 메인 페이지를 통해 upload 페이지로 이동하게 되면 파일명과 내용을 쓸 수 있습니다.

read 엔드포인트에서는 파일명을 이용해서 파일을 읽어들일 수 있습니다.

업로드 디렉토리는 UPLOAD_DIR = 'uploads' 로 정의되어 있습니다. 따라서 uploads 폴더 안에 생성된 파일들이 들어가게 됩니다.

read 엔드포인트에서는 name 파라미터를 통해 파일명을 받고 해당 파일명을 가진 파일을 찾아서 그 파일의 내용을 반환합니다.

문제 지문에서 flag.py를 읽어야 한다고 했습니다. 우선 flag.py 는 uploads 폴더에 들어있지 않습니다.

app.py와 같은 폴더에 있음을 짐작해서 요청 filename 을 ../flag.py 로 주면 될 것 같습니다.

http://host3.dreamhack.games:14729/read?name=../flag.py

DH{uploading_webshell_in_python_program_is_my_dream}