JWT를 활용한 파이썬 웹 API의 접근 로그 저장 및 보안 감사 이벤트 분석 방법

22 Sep 2023

서론

파이썬을 이용하여 웹 API를 개발할 때, 사용자의 접근 로그를 저장하고 보안 감사 이벤트를 분석하는 것은 매우 중요합니다. 이를 위해 JWT(JSON Web Token)을 사용하여 사용자 인증 및 권한 부여를 구현하고, 로그 데이터를 수집하여 분석하는 방법을 알아보도록 하겠습니다.

JWT란?

JWT는 웹 애플리케이션 간에 정보를 안전하게 전송하기 위한 표준 방식입니다. JWT는 JSON 객체를 사용하여 사용자의 인증 및 클레임(권한, 속성) 정보를 암호화하여 토큰으로 전송합니다. 이 토큰은 서버에서 검증되며 사용자의 인증 및 권한 부여에 사용됩니다.

JWT를 사용한 웹 API 개발 단계

JWT 라이브러리 설치: 파이썬에서는 PyJWT 라이브러리를 사용하여 JWT를 구현할 수 있습니다. pip install pyjwt 명령으로 라이브러리를 설치합니다.
JWT 토큰 생성: 사용자가 로그인하면 서버에서 JWT 토큰을 생성합니다. 토큰은 사용자의 ID, 권한 등의 클레임 정보와 시크릿 키를 사용하여 암호화됩니다.

import jwt

# 토큰 생성 함수
def generate_token(user_id):
    payload = {
        'user_id': user_id,
        'role': 'user'
    }
    token = jwt.encode(payload, 'secret_key', algorithm='HS256')
    return token

API 보호: API 엔드포인트에 접근하기 위해서는 클라이언트가 JWT 토큰을 포함하여 요청해야 합니다. 서버에서는 토큰을 검증한 후 유효한 요청인지 확인합니다.

import jwt

# 인증 데코레이터
def require_token(func):
    def wrapper(request):
        token = request.headers.get('Authorization')
        if not token:
            return '인증 실패'
        try:
            payload = jwt.decode(token, 'secret_key', algorithms=['HS256'])
            # 토큰에 대한 유효성 검사 및 권한 확인 로직 추가
            return func(request)
        except jwt.InvalidTokenError:
            return '인증 실패'
    return wrapper

# 보호해야 할 API 엔드포인트
@require_token
def protected_endpoint(request):
    return '인증된 API 엔드포인트'

로그 저장 및 보안 감사 이벤트 분석: 사용자의 접근 로그를 저장하여 보안 감사 이벤트를 분석할 수 있습니다. 디비에 로그를 저장하는 예시 코드는 다음과 같습니다.

import pymongo

# MongoDB 연결
client = pymongo.MongoClient('mongodb://localhost:27017/')
db = client['로그저장DB']
collection = db['접근로그']

# 접근 로그 저장 함수
def save_log(user_id, event):
    log = {
        'user_id': user_id,
        'event': event
    }
    collection.insert_one(log)

마무리

JWT를 활용하여 파이썬 웹 API에 접근 로그 저장과 보안 감사 이벤트 분석 기능을 추가할 수 있습니다. 로그 저장을 위해 데이터베이스를 설정하고, 사용자의 인증과 요청에 대한 검증 로직을 구현하는 것이 중요합니다. 이를 통해 웹 API의 보안성을 강화하고, 사용자의 로그 데이터를 분석하여 보안 이슈를 신속하게 대응할 수 있습니다.

#보안 #JWT