mirror of
https://github.com/jopohl/urh.git
synced 2026-03-03 15:04:01 +01:00
cdc10d7d76
* Find correct parameters for crc and datarange * Adapt output to URH values * Added bit2int method and a length fix * Cython implementation of crc * cython implementation for crc (first shot) * Fixed cython crc algorithm and renamed old crc algorithm to reference_crc. Added tests that compare results of old and new algorithm. * Huge performance increase for crc bruteforce test * Performance test with cython code within python * Cleaned up * - Removed padding of data to %8 bits - Some bugfixes * performance improvement for guess_standard_parameters_and_datarange * minor updates * Delta Improvement * slight performance improvement of delta algorithm * Delta improvement implemented and used by default * Delta improvement implemented and used by default * Delta improvement implemented and used by default * Bugfix for homematic encoding/decoding
174 lines
8.4 KiB
Python
174 lines
8.4 KiB
Python
import unittest, time
|
|
|
|
from urh.signalprocessing.Encoding import Encoding
|
|
from urh.util import util
|
|
from urh.util.GenericCRC import GenericCRC
|
|
from urh.util.WSPChecksum import WSPChecksum
|
|
|
|
class TestCRC(unittest.TestCase):
|
|
def test_crc(self):
|
|
# http://depa.usst.edu.cn/chenjq/www2/software/crc/CRC_Javascript/CRCcalculation.htm
|
|
# CRC-16: polynomial="16_standard", start_value = False, final_xor = False, reverse_polynomial=False, reverse_all=False
|
|
# CRC-16-CCITT: polynomial="16_ccitt", start_value = False, final_xor = False, reverse_polynomial=False, reverse_all=False
|
|
|
|
# http://www.lammertbies.nl/comm/info/crc-calculation.html <- Fehler
|
|
# CRC-16: polynomial="16_standard", start_value = False, final_xor = False, reverse_polynomial=False, reverse_all=False
|
|
c = GenericCRC(polynomial=WSPChecksum.CRC_8_POLYNOMIAL)
|
|
e = Encoding()
|
|
|
|
bitstr = ["010101010110100111011010111011101110111011100110001011101010001011101110110110101101",
|
|
"010101010110101001101110111011101110111011100110001011101010001011101110110111100101",
|
|
"010101010110100111010010111011101110111011100110001011101010001011101110110110100101"]
|
|
|
|
expected = ["78", "c9", "f2"]
|
|
|
|
for value, expect in zip(bitstr, expected):
|
|
nv = ""
|
|
for i in range(0, len(value)):
|
|
if value[i] == "1":
|
|
nv += "0"
|
|
else:
|
|
nv += "1"
|
|
|
|
self.assertEqual(util.bit2hex(c.crc(e.str2bit(value[4:-8]))), expect)
|
|
|
|
def test_different_crcs(self):
|
|
c = GenericCRC(polynomial="16_standard", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
bitstring_set = [
|
|
"101001001010101010101011101111111000000000000111101010011101011",
|
|
"101001001010101101111010110111101010010110111010",
|
|
"00000000000000000000000000000000100000000000000000000000000000000001111111111111",
|
|
"1111111111111111111111111111111110111111111111111111110111111111111111110000000000"
|
|
"1"]
|
|
|
|
for j in c.DEFAULT_POLYNOMIALS:
|
|
c.polynomial = c.choose_polynomial(j)
|
|
for i in bitstring_set:
|
|
# Standard
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
|
|
# Special final xor
|
|
c.final_xor = c.str2bit("0000111100001111")
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.final_xor = [False] * 16
|
|
|
|
# Special start value
|
|
c.start_value = c.str2bit("1010101010101010")
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.start_value = [False] * 16
|
|
|
|
# reverse_polynomial
|
|
c.reverse_polynomial = True
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.reverse_polynomial = False
|
|
|
|
# lsb_first
|
|
c.lsb_first = True
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.lsb_first = False
|
|
|
|
# little_endian
|
|
c.little_endian = True
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.little_endian = False
|
|
|
|
# reverse all
|
|
c.reverse_all = True
|
|
crc_new = c.crc(c.str2bit(i))
|
|
crc_old = c.reference_crc(c.str2bit(i))
|
|
self.assertEqual(crc_new, crc_old)
|
|
c.reverse_all = False
|
|
|
|
def test_reverse_engineering(self):
|
|
c = GenericCRC(polynomial="16_standard", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
bitstring_set = [
|
|
"1110001111001011100010000101010100000010110111000101100010100100111110111101100110110111011001010010001011101010",
|
|
"1110010011001011100010000101010100000010110111000101100010100100111110111101100110110111011001010010001011101010",
|
|
"1110010111001011100010000101010100000010110111000101100010100100111110111101100110110111011001010010001011101010",
|
|
"1110011011001011100010000101010100000010110111000101100010100100111110111101100110110111011001010010001011101010"]
|
|
bitset = []
|
|
crcset = []
|
|
|
|
for i in bitstring_set:
|
|
tmp = c.str2bit(i)
|
|
bitset.append(tmp)
|
|
crcset.append(c.crc(tmp))
|
|
|
|
polynomial = c.reverse_engineer_polynomial(bitset, crcset)
|
|
if polynomial:
|
|
self.assertEqual(c.bit2str(polynomial), "1000000000000101")
|
|
self.assertEqual(util.bit2hex(polynomial), "8005")
|
|
|
|
def test_not_aligned_data_len(self):
|
|
c = GenericCRC(polynomial="16_standard", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
polynomials = ["8_standard", "16_standard", "16_ccitt", "16_dnp"]
|
|
crcs = {"8_standard": 0xd5, "16_standard": 0x8005, "16_ccitt": 0x1021, "16_dnp": 0x3d65}
|
|
for j in polynomials:
|
|
c.polynomial = c.choose_polynomial(j)
|
|
inpt = "1"
|
|
for i in range(0, 32):
|
|
val = c.bit2int(c.crc(c.str2bit(inpt)))
|
|
self.assertEqual(val, crcs[j])
|
|
inpt = "0" + inpt
|
|
|
|
def test_guess_standard_parameters_and_datarange(self):
|
|
c = GenericCRC(polynomial="16_ccitt", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
inpt = "101010101010101010000000111000000000000011100000001011010010110100000000111000000101001010000100000000000100111001111110010000000011011111111001001101100001100010100000000000111011110100010"
|
|
vrfy_crc = "0011101111010001"
|
|
|
|
result = c.guess_standard_parameters_and_datarange(c.str2arr(inpt), c.str2arr(vrfy_crc))
|
|
self.assertEqual(result, (2, 84, 172))
|
|
self.assertEqual(vrfy_crc, c.bit2str(c.crc(c.str2arr(inpt[result[1]:result[2]]))))
|
|
|
|
def test_guess_standard_parameters_and_datarange_improved(self):
|
|
c = GenericCRC(polynomial="16_ccitt", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
inpt = "101010101010101010000000111000000000000011100000001011010010110100000000111000000101001010000100000000000100111001111110010000000011011111111001001101100001100010100000000000111011110100010"
|
|
vrfy_crc = "0011101111010001"
|
|
|
|
t1 = 0
|
|
runs = 100
|
|
for i in range(0, runs):
|
|
t = time.time()
|
|
result = c.guess_standard_parameters_and_datarange(c.str2arr(inpt), c.str2arr(vrfy_crc))
|
|
t1 += time.time() - t
|
|
#print(result, c.bit2str(c.crc(c.str2arr(inpt[result[1]:result[2]]))))
|
|
self.assertEqual(result[0], 2) # Parameters = 2
|
|
self.assertEqual(result[1], len(inpt) - 1 - 16 - 88) # start of datarange
|
|
self.assertEqual(result[2], len(inpt) - 1 - 16) # end of datarange
|
|
inpt = "0"+inpt if i%2 == 0 else "1"+inpt
|
|
#print("Performance:", t1/runs)
|
|
self.assertLess(t1/runs, 0.1) # Should be faster than 100ms in average
|
|
|
|
def test_adaptive_crc_calculation(self):
|
|
c = GenericCRC(polynomial="16_ccitt", start_value=False, final_xor=False,
|
|
reverse_polynomial=False, reverse_all=False, lsb_first=False, little_endian=False)
|
|
inpt1 = "10101010101010"
|
|
inpt2 = "1010101010101001"
|
|
|
|
crc1 = c.crc(c.str2arr(inpt1))
|
|
crc2 = c.crc(c.str2arr(inpt2))
|
|
|
|
# Compute crc2 from crc1 in a faster way
|
|
# Note: In general only forward direction
|
|
delta = "01"
|
|
c.start_value = crc1
|
|
crcx = c.crc(c.str2arr(delta))
|
|
|
|
self.assertEqual(crcx, crc2) |