mirror of
https://github.com/mjg59/python-broadlink.git
synced 2024-11-22 23:17:47 +01:00
27 lines
689 B
Python
27 lines
689 B
Python
"""Helper functions."""
|
|
from ctypes import c_ushort
|
|
|
|
|
|
def calculate_crc16(input_data: bytes) -> int:
|
|
"""Calculate the CRC-16 of a byte string."""
|
|
crc16_tab = []
|
|
crc16_constant = 0xA001
|
|
|
|
for i in range(0, 256):
|
|
crc = c_ushort(i).value
|
|
for j in range(0, 8):
|
|
if crc & 0x0001:
|
|
crc = c_ushort(crc >> 1).value ^ crc16_constant
|
|
else:
|
|
crc = c_ushort(crc >> 1).value
|
|
crc16_tab.append(hex(crc))
|
|
|
|
crcValue = 0xFFFF
|
|
|
|
for c in input_data:
|
|
tmp = crcValue ^ c
|
|
rotated = c_ushort(crcValue >> 8).value
|
|
crcValue = rotated ^ int(crc16_tab[(tmp & 0x00FF)], 0)
|
|
|
|
return crcValue
|