python3 number.py", line 433, in bytes_to_long问题

python3 number.py", line 433, in bytes_to_long问题

在python2中使用Crypto.Util.number import getPrime,bytes_to_long不会出现问题,但在python3中使用却会报错:
File “D:\python\lib\site-packages\Crypto\Util\number.py”, line 434
acc = (acc << 32) + unpack(’>I’, s[i:i+4])[0]
TypeError: a bytes-like object is required, not 'str

问题分析

找到并打开D:\python\lib\site-packages\Crypto\Util\number.py文件,定位到434行,函数bytes_to_long的代码如下:

def bytes_to_long(s):
    """bytes_to_long(string) : long
    Convert a byte string to a long integer.

    This is (essentially) the inverse of long_to_bytes().
    """
    acc = 0
    unpack = struct.unpack
    length = len(s)
    if length % 4:
        extra = (4 - length % 4)
        s = b('\000') * extra + s
        length = length + extra
    for i in range(0, length, 4):
        acc = (acc << 32) + unpack('>I', s[i:i+4])[0]
    return acc

从代码中可以看到s应该时bytes型的不错啊,怎么会出现上面的“str"??
带着疑问在acc = (acc << 32) + unpack(’>I’, s[i:i+4])[0]上面一行添加了print(type(s[i:i+4])并运行了一次,果然,发现在最后一个循环输出的竟然是str!!!,猜测可能是数组越界了。

ps: python3对bytes和str做了更明确的区分,两者不能混用,

解决办法

找到了原因,那就好办了
直接将

for i in range(0, length, 4):
        acc = (acc << 32) + unpack('>I', s[i:i+4])[0]
    return acc

改为

    for i in range(0, length, 4):
        if isinstance(s[i:i+4],str):
        	acc = (acc << 32) + unpack('>I', s[i:i+4].encode())[0] 
        else:
        	acc = (acc << 32) + unpack('>I', s[i:i+4])[0]
    return acc

也就是加了一个判断,如果遇到str类型,则转变换为bytes型
再次运行,OK,问题解决。
PS:这里注意一下python很容易出现空格格式上下不一致问题,如果遇到了可以在上面的代码中复制一下前面的空格,然后粘贴到报错的那行替换掉之前的空格就行了

你可能感兴趣的:(python)