MD5 python实现

import math
import time

A = 0x67452301
B = 0xefcdab89
C = 0x98badcfe
D = 0x10325476

T = [math.floor(pow(2, 32) * abs(math.sin(i + 1))) for i in range(64)]
shifts = [[7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22],
          [5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20],
          [4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23],
          [6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21]]

def reverse_hex(hex_str):
    hex_str_list = []
    for i in range(0,len(hex_str),2):
        hex_str_list.append(hex_str[i:i+2])
    hex_str_list.reverse()
    hex_str_result = '0x' + ''.join(hex_str_list)
    return hex_str_result

def padding(m: str):
    # 附加填充比特
    bit_len = 0
    bit_m = ''
    tmp = m.encode('utf-8')
    for i in tmp:
        bit_len += 8
        bit_m += bin(i)[2:].zfill(8)

    x = bit_len % 512
    if x == 448:
        bit_need = 512
    elif x < 448:
        bit_need = 448 - x
    else:
        bit_need = 960 - x

    plug = '1'
    for i in range(bit_need - 1):
        plug += '0'

    # 附加报文长度值
    l = bin(bit_len)[2:].zfill(64)
    l_low_32 = reverse_hex(hex(int(l[32:],2))[2:].zfill(8))
    l_high_32 = reverse_hex(hex(int(l[:32],2))[2:].zfill(8))
    l = bin(int(l_low_32,16))[2:].zfill(32) + bin(int(l_high_32,16))[2:].zfill(32)

    # 把附加值都接上，以512比特分组列表输出
    res = []
    m = bit_m + plug + l
    for i in range(0, len(m), 512):
        res.append(m[i:512 + i])
    return res


def leftrotate(x, n):
    return ((x<<n)|(x>>(32-n)))&(0xffffffff)

def get_X(Y):
    X = []
    for i in range(0, len(Y), 32):
        X_tmp = hex(int(Y[i:i+32],2))[2:].zfill(8)
        X_reverse = reverse_hex(X_tmp)
        X.append(int(X_reverse,16))
    return X

def F(x, y, z):
    return (x & y) | ((~x) & z)

def G(x, y, z):
    return (x & z) | (y & (~z))

def H(x, y, z):
    return x ^ y ^ z

def I(x, y, z):
    return y ^ (x | (~z))


def process_block(X, a, b, c, d):
    CVq = [a, b ,c, d]
    # round 1
    for i in range(16):
        a = leftrotate((a + F(b, c, d) + X[i] + T[i]) & 0xFFFFFFFF, shifts[0][i])
        a = (a + b) & 0xFFFFFFFF
        a, b, c, d = d, a, b, c
    #     print(hex(a), hex(b), hex(c), hex(d))
    # print('--------------------------------')
     # round 2
    for i in range(16):
        a = leftrotate((a + G(b, c, d) + X[(1+5*i)%16] + T[16+i]) & 0xFFFFFFFF, shifts[1][i])
        a = (a + b) & 0xFFFFFFFF
        a, b, c, d = d, a, b, c
    #     print(hex(a), hex(b), hex(c), hex(d))
    # print('--------------------------------')
     # round 3
    for i in range(16):
        a = leftrotate((a + H(b, c, d) + X[(5+3*i)%16] + T[32+i]) & 0xFFFFFFFF, shifts[2][i])
        a = (a + b) & 0xFFFFFFFF
        a, b, c, d = d, a, b, c
    #     print(hex(a), hex(b), hex(c), hex(d))
    # print('--------------------------------')
     # round 4
    for i in range(16):
        a = leftrotate((a + I(b, c, d) + X[7*i%16] + T[48+i]) & 0xFFFFFFFF, shifts[3][i])
        a = (a + b) & 0xFFFFFFFF
        a, b, c, d = d, a, b, c
    #     print(hex(a), hex(b), hex(c), hex(d))
    # print('--------------------------------')

    a = (a + CVq[0]) & 0xFFFFFFFF
    b = (b + CVq[1]) & 0xFFFFFFFF
    c = (c + CVq[2]) & 0xFFFFFFFF
    d = (d + CVq[3]) & 0xFFFFFFFF

    return a, b, c, d


# 计算 MD5
def md5(m):
    a, b, c, d = A, B, C, D
    padded = padding(m)
    L = len(padded)
    for i in range(L):
        X = get_X(padded[i])
        a, b, c, d = process_block(X, a, b, c, d)
    res_a = reverse_hex(hex(a)[2:].zfill(8))[2:].zfill(8)
    res_b = reverse_hex(hex(b)[2:].zfill(8))[2:].zfill(8)
    res_c = reverse_hex(hex(c)[2:].zfill(8))[2:].zfill(8)
    res_d = reverse_hex(hex(d)[2:].zfill(8))[2:].zfill(8)
    hash = res_a + res_b + res_c + res_d
    return hash

m = input()
print(md5(m))