[파이썬] PostgreSQL 데이터베이스 암호화

01 Sep 2023

python

암호화는 데이터베이스 보안에 중요한 역할을 합니다. 암호화를 통해 데이터의 기밀성을 유지하고 민감한 정보를 보호할 수 있습니다. PostgreSQL은 많은 기능을 제공하여 데이터베이스 암호화를 구현할 수 있으며, 이를 Python을 이용하여 쉽게 사용할 수 있습니다.

이번 블로그 포스트에서는 Python으로 PostgreSQL 데이터베이스를 암호화하는 방법에 대해 살펴보겠습니다.

psycopg2 사용하기

psycopg2는 Python에서 PostgreSQL을 사용하기 위한 완벽한 어댑터입니다. 이를 사용하여 Python 어플리케이션에서 PostgreSQL 데이터베이스에 접속하고 데이터를 암호화할 수 있습니다.

먼저 psycopg2 모듈을 설치해야합니다. pip를 사용하여 설치할 수 있습니다:

pip install psycopg2

이제 암호화된 연결을 위해 sslmode를 require로 설정해야합니다. 이는 SSL 암호화가 사용되어야 함을 의미합니다. 연결 설정은 데이터베이스에 대한 커넥션을 만들 때 설정할 수 있습니다.

import psycopg2

# 암호화된 연결을 위해 sslmode=require로 설정합니다.
conn = psycopg2.connect(
    host="localhost",
    dbname="mydatabase",
    user="myuser",
    password="mypassword",
    sslmode="require"
)

이제 연결이 암호화된 상태로 데이터베이스에 접속했습니다.

데이터 암호화하기

데이터베이스에 저장되는 민감한 정보를 암호화하기 위해 Python 내장 모듈인 cryptography를 사용할 수 있습니다. 이 모듈은 강력한 암호화 기능을 제공합니다.

먼저 cryptography 모듈을 설치해야합니다:

pip install cryptography

그런 다음 암호화 및 복호화를 위한 함수를 작성할 수 있습니다. 예를 들어, 문자열을 AES-256 알고리즘을 사용하여 암호화하고 복호화하는 함수는 다음과 같을 수 있습니다.

import base64
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.primitives import padding
from cryptography.hazmat.backends import default_backend

def encrypt_data(key, data):
    backend = default_backend()
    iv = os.urandom(16) # 초기화 벡터 생성
    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=backend)
    encryptor = cipher.encryptor()
    padder = padding.PKCS7(128).padder()    
    padded_data = padder.update(data) + padder.finalize()
    ciphertext = encryptor.update(padded_data) + encryptor.finalize()
    return base64.b64encode(iv + ciphertext)

def decrypt_data(key, encrypted_data):
    backend = default_backend()
    encrypted_data = base64.b64decode(encrypted_data)
    iv = encrypted_data[:16]
    ciphertext = encrypted_data[16:]
    
    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=backend)
    decryptor = cipher.decryptor()
    unpadder = padding.PKCS7(128).unpadder()
    decrypted_data = decryptor.update(ciphertext) + decryptor.finalize()
    return unpadder.update(decrypted_data) + unpadder.finalize()

이제 데이터베이스에 저장된 정보를 암호화하고 복호화하는 함수를 사용할 수 있습니다.

# 데이터 암호화 예제
plaintext_data = "This is my sensitive data"
key = b'mystrongencryptionkey'

encrypted_data = encrypt_data(key, plaintext_data)
print("Encrypted Data:", encrypted_data)

# 데이터 복호화 예제
decrypted_data = decrypt_data(key, encrypted_data)
print("Decrypted Data:", decrypted_data.decode())

위의 코드는 데이터를 암호화하고 복호화하기 위한 간단한 예제입니다. 실제 프로덕션 환경에서는 더 강력한 랜덤 키 생성 및 고급 암호화 알고리즘을 사용해야합니다.

결론

Python과 PostgreSQL을 함께 사용하여 데이터베이스에 암호화를 구현하는 방법을 알아보았습니다. psycopg2를 사용하여 PostgreSQL에 연결하고, cryptography 모듈을 사용하여 데이터를 암호화할 수 있습니다. 데이터베이스 암호화는 보안에 필수적이며 중요한 데이터를 보호하는데 큰 도움이 됩니다.

암호화는 데이터를 안전하게 보호하는 한 가지 방법일 뿐이므로, 데이터베이스 보안을 위해서는 암호화 외에도 다른 보안 조치를 적용해야합니다.

이러한 방법을 사용하여 PostgreSQL 데이터베이스에 암호화를 추가하고 암호화된 데이터를 안전하게 보호하세요!