mirror of
https://github.com/Telecominfraproject/OpenCellular.git
synced 2025-12-29 10:00:51 +00:00
f229fabd8b
This also moves flash related tests to use new 'hostcmd' console command. BUG=chrome-os-partner:10262 TEST=Test passed BRANCH=none Change-Id: I5616bfa93bcde0beb4cb2baf2d38e8b5d827c275 Signed-off-by: Vic Yang <victoryang@chromium.org> Reviewed-on: https://gerrit.chromium.org/gerrit/30665 Reviewed-by: Vincent Palatin <vpalatin@chromium.org>
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
#
|
|
# Utility functions for flash related test
|
|
#
|
|
|
|
import random
|
|
import re
|
|
import struct
|
|
|
|
# Fixed random seed.
|
|
random.seed(1234)
|
|
|
|
def hex_to_byte(text):
|
|
return ''.join(["%c" % chr(int(text[i:i+2], 16))
|
|
for i in range(0, len(text), 2)])
|
|
|
|
def byte_to_hex(byte_arr):
|
|
return ''.join(["%02x" % ord(c) for c in byte_arr])
|
|
|
|
def offset_size_pair(offset, size):
|
|
return byte_to_hex(struct.pack("II", offset, size))
|
|
|
|
def get_flash_info(helper):
|
|
helper.ec_command("hostcmd 0x10 0 00")
|
|
resp = helper.wait_output("Response: (?P<r>.{32,32})", use_re=True)["r"]
|
|
return struct.unpack("IIII", hex_to_byte(resp))
|
|
|
|
def get_flash_size(helper):
|
|
return get_flash_info(helper)[0]
|
|
|
|
def get_ro_size(helper):
|
|
helper.ec_command("rosize")
|
|
return int(helper.wait_output(
|
|
"RO image size = (?P<ro>0x[0-9a-f]+)", use_re=True)["ro"], 16)
|
|
|
|
def xor_sum(size, seed, mult, add):
|
|
ret = 0
|
|
for i in xrange(size):
|
|
ret ^= (seed & 0xff)
|
|
seed = seed * mult + add
|
|
return ret
|
|
|
|
def test_erase(helper, offset, size):
|
|
helper.ec_command("hostcmd 0x13 0 %s" % offset_size_pair(offset, size))
|
|
helper.wait_output("Flash erase at %x size %x" % (offset, size))
|
|
|
|
def _get_read_ref(helper, offset, size):
|
|
ret = []
|
|
retsub = []
|
|
assert size % 4 == 0
|
|
while size > 0:
|
|
helper.ec_command("rw %d" % offset)
|
|
h = helper.wait_output("read.*=\s+0x(?P<h>[0-9a-f]+)", use_re=True)["h"]
|
|
# Change endianess here
|
|
retsub.append(re.sub('(..)(..)(..)(..)', r'\4\3\2\1', h))
|
|
if len(retsub) == 8:
|
|
ret.append(''.join(retsub))
|
|
retsub = []
|
|
size = size - 4
|
|
offset = offset + 4
|
|
if retsub:
|
|
ret.append(''.join(retsub))
|
|
return ret
|
|
|
|
def test_read(helper, offset, size):
|
|
ref = _get_read_ref(helper, offset, size)
|
|
helper.ec_command("hostcmd 0x11 0 %s" % offset_size_pair(offset, size))
|
|
for line in ref:
|
|
helper.wait_output(line)
|
|
|
|
def _gen_data(size, seed, mult, add):
|
|
data = []
|
|
for i in xrange(size):
|
|
data.append("%02x" % (seed & 255))
|
|
seed = (seed * mult + add) & 4294967295;
|
|
return ''.join(data)
|
|
|
|
def test_write(helper, offset, size, expect_fail=False):
|
|
assert size <= 16
|
|
seed = random.randint(2, 10000)
|
|
mult = random.randint(2, 10000)
|
|
add = random.randint(2, 10000)
|
|
data = _gen_data(size, seed, mult, add)
|
|
payload = byte_to_hex(struct.pack("II", offset, size))
|
|
helper.ec_command("hostcmd 0x12 0 %s%s" %
|
|
(offset_size_pair(offset, size), data))
|
|
if expect_fail:
|
|
helper.wait_output("Command returned \d+", use_re=True)
|
|
else:
|
|
expected_sum = xor_sum(size, seed, mult, add)
|
|
helper.wait_output("Flash write at %x size %x XOR %x" %
|
|
(offset, size, expected_sum))
|