글 읽기 - encrypt / decrypt 함수 예제

WeissBlume 2년 전 5

$ pip install pycrypto
# or
$ pip install pycryptodome

from Crypto.Cipher import AES

def decrypt(ctext: str, key: str) -> str:
	ctext, key = bytes.fromhex(ctext), bytes.fromhex(key)
	ptext = AES.new(key, AES.MODE_CBC, IV=b"\x00"*16).decrypt(ctext)
	return ptext.hex().upper()

def encrypt(ptext: str, key: str) -> str:
	ptext, key = bytes.fromhex(ptext), bytes.fromhex(key)
	ctext = AES.new(key, AES.MODE_CBC, IV=b"\x00"*16).encrypt(ptext)
	return ctext.hex().upper()

print(decrypt(
	decrypt("6323B4A5BC16C479ED6D94F5B58FF0C2", "70000000000000000000000000000000"),
	"A0000000000000000000000000000000"
)) # 00112233445566778899AABBCCDDEEFF

print(encrypt(
	encrypt("00112233445566778899AABBCCDDEEFF", "A0000000000000000000000000000000"),
	"70000000000000000000000000000000"
)) # 6323B4A5BC16C479ED6D94F5B58FF0C2

bubbler 2년 전 2

어젯밤에 슬랙 보고 아침에 찾아서 올려야지 했는데 다른 분이 먼저 하셨군요.

AES.new(key, AES.MODE_ECB)를 쓰면 조금 더 간단하게 됩니다.

설명을 조금 추가하자면 다음과 같습니다.

# 길이 32의 hex string을 받아 길이 16의 bytes object를 생성합니다.
# 대소문자는 상관 없습니다.
# bytes 대신 bytearray를 사용하면 바이트 값을 하나하나 수정할 수 있습니다. 문제 풀이에 도움이 될 수도..
key = bytes.fromhex('A0000000000000000000000000000000')

# 주어진 키를 이용해 AES object를 생성합니다.
# 키가 길이 16의 bytes/bytearray이면 AES128을 사용합니다.
cipher = AES.new(key, AES.MODE_ECB)

# cipher.encrypt와 decrypt 함수는 길이 16의 bytes를 받아 길이 16의 bytes를 생성합니다.
# (각각 암호화, 복호화)
# 같은 키로 암호화했다가 복호화하면 원래 데이터가 bytes 형태로 나옵니다.
data = bytes.fromhex('00112233445566778899AABBCCDDEEFF')
ciphertext = cipher.encrypt(data)
plaintext = cipher.decrypt(ciphertext)
assert data == plaintext

# bytes에서 hex로 만들려면 bytes.hex()를 사용합니다.
# 소문자를 주기 때문에 제출 형식에 맞추기 위해 .upper()가 필요합니다.
print(key.hex().upper())

jh05013 2년 전 1

WeissBlume님의 코드가 실행이 안 된다면 다음을 실행해 보시기 바랍니다.

from Crypto.Cipher import AES

def decrypt(ctext: str, key: str) -> str:
    ctext, key = bytes.fromhex(ctext), bytes.fromhex(key)
    ptext = AES.new(key, AES.MODE_ECB).decrypt(ctext)
    return ptext.hex().upper()

def encrypt(ptext: str, key: str) -> str:
    ptext, key = bytes.fromhex(ptext), bytes.fromhex(key)
    ctext = AES.new(key, AES.MODE_ECB).encrypt(ptext)
    return ctext.hex().upper()

print(decrypt(
    decrypt("6323B4A5BC16C479ED6D94F5B58FF0C2", "70000000000000000000000000000000"),
        "A0000000000000000000000000000000"
)) # 00112233445566778899AABBCCDDEEFF

print(encrypt(
    encrypt("00112233445566778899AABBCCDDEEFF", "A0000000000000000000000000000000"),
        "70000000000000000000000000000000"
)) # 6323B4A5BC16C479ED6D94F5B58FF0C2

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

encrypt / decrypt 함수 예제

WeissBlume 2년 전 5

bubbler 2년 전 2

jh05013 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말