# FTP · FTPS 시험 서버 (pyftpdlib 2.2.0 · pyOpenSSL 26.4.0 - 시험용 파이썬 venv ..\tools\sftp_test_py 에 둠, 앱에는 들어가지 않음).
# 사용: python tool/ftp_test_server.py --root 폴더 [--user jj --password pw] [--mode plain|explicit|implicit] [--cert 파일]
#   127.0.0.1 의 빈 포트에 열고 "PORT n" · "FINGERPRINT SHA256:AA:BB:…" 를 알린다. 표준 입력이 닫히면 끝난다.
#   --cert 파일이 없으면 자체 서명 인증서를 만들어 그 파일에 둔다 (같은 파일이면 같은 지문).
import argparse
import hashlib
import os
import sys
import threading
import logging
if os.environ.get('JJ_FTP_DEBUG'):
    logging.basicConfig(level=logging.DEBUG)

from pyftpdlib.authorizers import DummyAuthorizer
from pyftpdlib.handlers import FTPHandler
from pyftpdlib.servers import FTPServer

ap = argparse.ArgumentParser()
ap.add_argument('--root', required=True)
ap.add_argument('--user', default='jj')
ap.add_argument('--password', default='pw')
ap.add_argument('--mode', default='explicit', choices=['plain', 'explicit', 'implicit'])
ap.add_argument('--cert', default='')
a = ap.parse_args()

auth = DummyAuthorizer()
auth.add_user(a.user, a.password, a.root, perm='elradfmwMT')
fingerprint = ''
if a.mode == 'plain':
    handler = FTPHandler
else:
    from OpenSSL import crypto
    from pyftpdlib.handlers import TLS_FTPHandler
    cert = a.cert or os.path.join(a.root, '..', 'ftp_cert.pem')
    if not os.path.exists(cert):
        k = crypto.PKey()
        k.generate_key(crypto.TYPE_RSA, 2048)
        c = crypto.X509()
        c.get_subject().CN = '127.0.0.1'
        c.set_serial_number(1)
        c.gmtime_adj_notBefore(0)
        c.gmtime_adj_notAfter(3600 * 24)
        c.set_issuer(c.get_subject())
        c.set_pubkey(k)
        c.sign(k, 'sha256')
        with open(cert, 'wb') as f:
            f.write(crypto.dump_privatekey(crypto.FILETYPE_PEM, k) + crypto.dump_certificate(crypto.FILETYPE_PEM, c))
    der = crypto.dump_certificate(crypto.FILETYPE_ASN1, crypto.load_certificate(crypto.FILETYPE_PEM, open(cert, 'rb').read()))
    fingerprint = 'SHA256:' + ':'.join('%02X' % b for b in hashlib.sha256(der).digest())

    from pyftpdlib.handlers import TLS_DTPHandler
    from pyftpdlib.handlers.ftps.ssl import SSLConnectionMixin

    # pyftpdlib 2.2.0 + pyOpenSSL 26: 자료 연결의 TLS 닫기에서 빈 SSL.Error 를 "다시 해 보기" 로 보고 끝없이 되풀이해
    # 올리기 (STOR) 가 끝나지 않았다 (226 이 오지 않음, 10/18). 시험 서버만: TLS 를 한 번 닫아 보고 바로 연결을 닫는다
    class D(TLS_DTPHandler):
        def _do_ssl_shutdown(self):
            try:
                self.socket.shutdown()
            except Exception:
                pass
            super(SSLConnectionMixin, self).close()

    class H(TLS_FTPHandler):
        certfile = cert
        dtp_handler = D
        tls_control_required = True
        tls_data_required = True

    if a.mode == 'implicit':
        # pyftpdlib 은 암시 FTPS 를 따로 두지 않는다 - 연결하자마자 TLS 로 (secure_connection)
        class HI(H):
            def handle(self):
                self.secure_connection(self.ssl_context)

            def handle_ssl_established(self):
                FTPHandler.handle(self)

            def ftp_AUTH(self, line):
                self.respond('503 Already using TLS.')

        handler = HI
    else:
        handler = H

handler.authorizer = auth
handler.passive_ports = None
handler.encoding = 'utf-8'
server = FTPServer(('127.0.0.1', 0), handler)
print('PORT', server.socket.getsockname()[1], flush=True)
print('FINGERPRINT', fingerprint, flush=True)


def watch():
    sys.stdin.read()
    server.close_all()
    os._exit(0)


threading.Thread(target=watch, daemon=True).start()
server.serve_forever(handle_exit=False)