[python] 웹 서버에 방화벽 기능 추가하기

05 Dec 2023

python

웹 서버에 방화벽 기능을 추가하여 보다 안전하게 서비스를 운영할 수 있습니다. 이번 포스트에서는 Python을 사용하여 웹 서버에 방화벽을 구현하는 방법에 대해 알아보겠습니다.

1. 방화벽 기능이란?

방화벽은 네트워크 트래픽을 모니터링하고, 허용되지 않은 액세스를 차단하여 네트워크의 보안을 강화하는 시스템입니다. 방화벽은 보안 정책을 기반으로 트래픽을 필터링하고, 허용된 트래픽만을 통과시킵니다.

2. Python으로 웹 서버 구현하기

Python의 http.server 모듈을 사용하여 간단한 웹 서버를 구현할 수 있습니다. 아래의 예제 코드를 통해 웹 서버를 실행할 수 있습니다.

import http.server
import socketserver

PORT = 8000

Handler = http.server.SimpleHTTPRequestHandler

with socketserver.TCPServer(("", PORT), Handler) as httpd:
    print("서버가 시작되었습니다.")
    httpd.serve_forever()

위의 예제 코드에서 PORT 변수는 서버가 실행될 포트번호를 지정하는 부분입니다.

3. 방화벽 기능 추가하기

Python의 socket 모듈을 사용하여 네트워크 연결을 처리할 수 있습니다. 아래의 예제 코드에서는 특정 IP 주소로의 접속을 차단하는 방화벽 기능을 추가한 것입니다.

import http.server
import socketserver

PORT = 8000

class FirewallHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
    blocked_ips = ["127.0.0.1", "192.168.0.1"]

    def handle_request(self):
        client_ip = self.client_address[0]
        if client_ip in self.blocked_ips:
            self.send_response(403)
            self.send_header("Content-type", "text/html")
            self.end_headers()
            self.wfile.write(bytes("접근이 차단되었습니다.", "utf-8"))
        else:
            super().handle_request()

with socketserver.TCPServer(("", PORT), FirewallHTTPRequestHandler) as httpd:
    print("서버가 시작되었습니다.")
    httpd.serve_forever()

위의 예제 코드에서 blocked_ips 변수는 차단할 IP 주소 목록을 지정하는 부분입니다. 클라이언트의 IP 주소를 확인하여 차단할 IP 목록에 있는 경우, 403 Forbidden 응답을 전송하도록 구현하였습니다.

위의 코드를 실행하면 웹 서버에 방화벽 기능이 추가된 상태로 실행됩니다.

마무리

이번 포스트에서는 Python을 사용하여 웹 서버에 방화벽을 추가하는 방법에 대해 알아보았습니다. 방화벽은 웹 서버의 보안을 강화하여 불필요한 접근을 차단함으로써 서비스의 안정성을 높일 수 있습니다.

더 자세한 내용은 다음 참고 자료를 참고하시기 바랍니다.

Python 공식 문서 (https://docs.python.org/3/library/http.server.html)
Socket 모듈 공식 문서 (https://docs.python.org/3/library/socket.html)