Python2系で書かれたコードを3.7に変換したい

Question

前提・実現したいこと

Python2系で書かれたコードを3.7に変換したい

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "C:\Users(UserNane)\Desktop\新しいテキスト ドキュメント.txt", line 75, in <module>
    auth_key = encrypt(client_id, user_id, shared_key)
  File "C:\Users(UserNane)\Desktop\新しいテキスト ドキュメント.txt", line 39, in encrypt
    encrypted = cipher.encrypt(plain_text.encode("UTF-8"))
  File "C:\Users(UserNane)\AppData\Local\Programs\Python\Python37\lib\site-packages\Crypto\Cipher\_mode_cbc.py", line 183, in encrypt
    raise ValueError("Data must be padded to %d byte boundary in CBC mode" % self.block_size)
ValueError: Data must be padded to 8 byte boundary in CBC mode

該当のソースコード

#!/usr/bin/env python
# -*- coding: utf-8 -+-

"""
認証用文字列生成サンプル
"""

import base64
from builtins import Exception
import json
import random
import string
from struct import pack
import time
import urllib.request, urllib.error, urllib.parse

from Crypto.Cipher import Blowfish

BLOCK_SIZE = 8


def encrypt(client_id, user_id, shared_key):
    ''' 暗号文字列を生成します. '''
    #iv = ''.join(random.choice(string.letters) for i in range(BLOCK_SIZE))
    iv = ''.join(random.choices(string.ascii_letters, k = BLOCK_SIZE))
    # 暗号化前の平文文字列
    # 書式：[clientid]:[アクセス日時]:[ユーザーID]:[IV値]:
    plain_text = "{}:{}:{}:{}:".format(
        client_id, int(time.time()), user_id, iv)
    # 暗号文字列サイズが8の倍数になるように調整
    plen = BLOCK_SIZE - len(plain_text) % BLOCK_SIZE
    padding = [plen] * plen
    padding = pack('b' * plen, *padding)
    #plain_text += padding
    plain_text += str(padding)

    #cipher = Blowfish.new(shared_key, Blowfish.MODE_CBC, iv)
    cipher = Blowfish.new(shared_key.encode('utf-8'), Blowfish.MODE_CBC, iv.encode('utf-8'))
    encrypted = cipher.encrypt(plain_text.encode("UTF-8"))
    # IV値(8文字) + 暗号文字列をbase64エンコード
    return base64.urlsafe_b64encode(iv+encrypted)


def create_session_id(client_id, auth_key):
    
    #todo Python2系から変換する

if __name__ == '__main__':
    import sys

    # クライアントID
    #client_id = sys.argv[1]
    client_id = "kkcdev"
    # ユーザーＩＤ（任意）
    #user_id = sys.argv[2]
    user_id = "sss"
    # 共有キー
    #shared_key = sys.argv[3]
    shared_key = "PDSKKCDEV"

    auth_key = encrypt(client_id, user_id, shared_key)
    session_id = create_session_id(client_id, auth_key)
    print("Session ID: {}".format(session_id))

試したこと

encrypted = cipher.encrypt(plain_text)
だと

Traceback (most recent call last):
  File "C:\Users(UserNane)\Desktop\新しいテキスト ドキュメント.txt", line 75, in <module>
    auth_key = encrypt(client_id, user_id, shared_key)
  File "C:\Users(UserNane)\Desktop\新しいテキスト ドキュメント.txt", line 39, in encrypt
    encrypted = cipher.encrypt(plain_text)
  File "C:\Users(UserNane)\AppData\Local\Programs\Python\Python37\lib\site-packages\Crypto\Cipher\_mode_cbc.py", line 178, in encrypt
    c_uint8_ptr(plaintext),
  File "C:\Users(UserNane)\AppData\Local\Programs\Python\Python37\lib\site-packages\Crypto\Util\_raw_api.py", line 145, in c_uint8_ptr
    raise TypeError("Object type %s cannot be passed to C code" % type(data))
TypeError: Object type <class 'str'> cannot be passed to C code

となったので以下に修正
encrypted = cipher.encrypt(plain_text.encode("UTF-8"))

補足情報（FW/ツールのバージョンなど）

IDLE (Python 3.7 64-bit)
pycryptodome 3.9.0
Python 3.7

Answer 1

paddingはstr型のまま足しこむのがポイントかと思います。

def encrypt(client_id, user_id, shared_key):
    ''' 暗号文字列を生成します. '''
    iv = ''.join(random.choices(string.ascii_letters, k = BLOCK_SIZE))

    # 暗号化前の平文文字列
    # 書式：[clientid]:[アクセス日時]:[ユーザーID]:[IV値]:
    plain_text = "{}:{}:{}:{}:".format(
        client_id, int(time.time()), user_id, iv)

    # 暗号文字列サイズが8の倍数になるように調整
    plen = BLOCK_SIZE - len(plain_text) % BLOCK_SIZE

    # 文字列のまま足しこむ
    padding = str(plen) * plen
    plain_text += padding

    iv = iv.encode('utf-8') # byte型に

    cipher = Blowfish.new(shared_key.encode('utf-8'), Blowfish.MODE_CBC, iv)
    encrypted = cipher.encrypt(plain_text.encode("UTF-8"))
    # IV値(8文字) + 暗号文字列をbase64エンコード
    return base64.urlsafe_b64encode(iv+encrypted)

Comments

[NextToYou]

ありがとうございます。