crc.py

import itertools

from ntype import uint32, BigStream


def calculate_crc(data: BigStream) -> bytearray:
    t1 = t2 = t3 = t4 = t5 = t6 = 0xDF26F436

    u32 = 0xFFFFFFFF

    m1 = data.read_bytes(0x1000, 0x100000)
    words = map(uint32.value, zip(m1[0::4], m1[1::4], m1[2::4], m1[3::4]))

    m2 = data.read_bytes(0x750, 0x100)
    words2 = map(uint32.value, zip(m2[0::4], m2[1::4], m2[2::4], m2[3::4]))

    for d, d2 in zip(words, itertools.cycle(words2)):
        # keep t2 and t6 in u32 for comparisons; others can wait to be truncated
        if ((t6 + d) & u32) < t6:
            t4 += 1

        t6 = (t6+d) & u32
        t3 ^= d
        shift = d & 0x1F
        r = ((d << shift) | (d >> (32 - shift)))
        t5 += r

        if t2 > d:
            t2 ^= r & u32
        else:
            t2 ^= t6 ^ d

        t1 += d2 ^ d

    crc0 = (t6 ^ t4 ^ t3) & u32
    crc1 = (t5 ^ t2 ^ t1) & u32

    return uint32.bytes(crc0) + uint32.bytes(crc1)