test/py: add generic CRC32 function

Add a generic function which can be used to compute the CRC32 value of
a region of RAM.

Signed-off-by: Liam Beguin <liambeguin@gmail.com>
Reviewed-by: Stephen Warren <swarren@nvidia.com>
This commit is contained in:
Liam Beguin 2018-03-14 19:15:15 -04:00 committed by Tom Rini
parent 64a2cebb6e
commit c3342cd58f

View file

@ -11,6 +11,7 @@ import os.path
import pytest import pytest
import sys import sys
import time import time
import re
def md5sum_data(data): def md5sum_data(data):
"""Calculate the MD5 hash of some data. """Calculate the MD5 hash of some data.
@ -310,3 +311,25 @@ def persistent_file_helper(u_boot_log, filename):
""" """
return PersistentFileHelperCtxMgr(u_boot_log, filename) return PersistentFileHelperCtxMgr(u_boot_log, filename)
def crc32(u_boot_console, address, count):
"""Helper function used to compute the CRC32 value of a section of RAM.
Args:
u_boot_console: A U-Boot console connection.
address: Address where data starts.
count: Amount of data to use for calculation.
Returns:
CRC32 value
"""
bcfg = u_boot_console.config.buildconfig
has_cmd_crc32 = bcfg.get('config_cmd_crc32', 'n') == 'y'
assert has_cmd_crc32, 'Cannot compute crc32 without CONFIG_CMD_CRC32.'
output = u_boot_console.run_command('crc32 %08x %x' % (address, count))
m = re.search('==> ([0-9a-fA-F]{8})$', output)
assert m, 'CRC32 operation failed.'
return m.group(1)