본문 바로가기

카테고리 없음

드림핵 file-download-1

문제 서버를 생성하고 들어가보니 이와 같은 화면이 떴다.
 
 
 

Upload My Memo로 들어가니 무언가 작성할 수 있는 페이지가 떴다.
 

 

파일을 마음대로 만들어보니 파일 목록에 생성된 파일이 떴다.
 
 
 

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

 
문제에서 제공된 코드를 열어보았다. 
메모를 만들 때 메모 제목에 ‘..’가 들어간 것이 발견되면 bad characters 라는 페이지를 return한다.
제약조건에 ..를 메모 제목으로 사용할 수 없으므로 ../를 사용할 수 있다.
filename 변수의 매개변수로 name을 받는다. url에 있는 name을 이용할 수 있다.
 
 
 

url 끝 부분을 ../flag.py로 바꿔주니 이와 같이 플래그값이 떴다.