Enhance the UMS test to optionally mount a partition and read/write a file to it, validating that the content written and read back are identical. This enhancement is backwards-compatible; old boardenv contents that don't define the new configuration data will cause the test code to perform as before. test/ums/ is deleted since the Python test now performs the same testing that it did. The code is also re-written to make use of the recently added utility module, and split it up into nested functions so the overall logic of the test process can be followed more easily without the details cluttering the code. Signed-off-by: Stephen Warren <swarren@nvidia.com> Acked-by: Lukasz Majewski <l.majewski@samsung.com> Acked-by: Simon Glass <sjg@chromium.org>master
Stephen Warren
9 years ago
committed by
Simon Glass
parent
76b4693928
commit
d054f4c2cb
@ -1,30 +0,0 @@ |
||||
UMS test script. |
||||
|
||||
ums_gadget_test.sh |
||||
================== |
||||
|
||||
Example usage: |
||||
1. On the target: |
||||
create UMS exportable partitions (with e.g. gpt write), or specify a |
||||
partition number (PART_NUM) as "-" to use the entire device |
||||
ums 0 mmc 0 |
||||
2. On the host: |
||||
sudo test/ums/ums_gadget_test.sh VID PID PART_NUM [-f FILE_SYSTEM] [test_file] |
||||
e.g. sudo test/ums/ums_gadget_test.sh 0525 a4a5 6 -f vfat ./dat_14M.img |
||||
|
||||
... where: |
||||
VID - UMS device USB Vendor ID |
||||
PID - UMS device USB Product ID |
||||
PART_NUM - is the partition number on which UMS operates or "-" to use the |
||||
whole device |
||||
|
||||
Information about available partitions on the target one can read with using |
||||
the 'mmc part' or 'part list' commands. |
||||
|
||||
The partition num (PART_NUM) can be specified as '-' for using the whole device. |
||||
|
||||
The [-f FILE_SYSTEM] optional switch allows for formatting target partition to |
||||
FILE_SYSTEM. |
||||
|
||||
The last, optional [test_file] parameter is for specifying the exact test file |
||||
to use. |
||||
@ -1,183 +0,0 @@ |
||||
#! /bin/bash |
||||
|
||||
# Copyright (C) 2014 Samsung Electronics |
||||
# Lukasz Majewski <l.majewski@samsung.com> |
||||
# |
||||
# UMS operation test script |
||||
# |
||||
# SPDX-License-Identifier: GPL-2.0+ |
||||
|
||||
clear |
||||
|
||||
COLOUR_RED="\33[31m" |
||||
COLOUR_GREEN="\33[32m" |
||||
COLOUR_ORANGE="\33[33m" |
||||
COLOUR_DEFAULT="\33[0m" |
||||
|
||||
DIR=./ |
||||
SUFFIX=img |
||||
RCV_DIR=rcv/ |
||||
LOG_FILE=./log/log-`date +%d-%m-%Y_%H-%M-%S` |
||||
|
||||
cd `dirname $0` |
||||
../dfu/dfu_gadget_test_init.sh 33M 97M |
||||
|
||||
cleanup () { |
||||
rm -rf $RCV_DIR $MNT_DIR |
||||
} |
||||
|
||||
control_c() |
||||
# run if user hits control-c |
||||
{ |
||||
echo -en "\n*** CTRL+C ***\n" |
||||
umount $MNT_DIR |
||||
cleanup |
||||
exit 0 |
||||
} |
||||
|
||||
# trap keyboard interrupt (control-c) |
||||
trap control_c SIGINT |
||||
|
||||
die () { |
||||
printf " $COLOUR_RED FAILED $COLOUR_DEFAULT \n" |
||||
cleanup |
||||
exit 1 |
||||
} |
||||
|
||||
calculate_md5sum () { |
||||
MD5SUM=`md5sum $1` |
||||
MD5SUM=`echo $MD5SUM | cut -d ' ' -f1` |
||||
echo "md5sum:"$MD5SUM |
||||
} |
||||
|
||||
ums_test_file () { |
||||
printf "$COLOUR_GREEN========================================================================================= $COLOUR_DEFAULT\n" |
||||
printf "File:$COLOUR_GREEN %s $COLOUR_DEFAULT\n" $1 |
||||
|
||||
mount /dev/$MEM_DEV $MNT_DIR |
||||
if [ -f $MNT_DIR/dat_* ]; then |
||||
rm $MNT_DIR/dat_* |
||||
fi |
||||
|
||||
cp ./$1 $MNT_DIR |
||||
|
||||
while true; do |
||||
umount $MNT_DIR > /dev/null 2>&1 |
||||
if [ $? -eq 0 ]; then |
||||
break |
||||
fi |
||||
printf "$COLOUR_ORANGE\tSleeping to wait for umount...$COLOUR_DEFAULT\n" |
||||
sleep 1 |
||||
done |
||||
|
||||
echo -n "TX: " |
||||
calculate_md5sum $1 |
||||
|
||||
MD5_TX=$MD5SUM |
||||
sleep 1 |
||||
N_FILE=$DIR$RCV_DIR${1:2}"_rcv" |
||||
|
||||
mount /dev/$MEM_DEV $MNT_DIR |
||||
cp $MNT_DIR/$1 $N_FILE || die $? |
||||
rm $MNT_DIR/$1 |
||||
umount $MNT_DIR |
||||
|
||||
echo -n "RX: " |
||||
calculate_md5sum $N_FILE |
||||
MD5_RX=$MD5SUM |
||||
|
||||
if [ "$MD5_TX" == "$MD5_RX" ]; then |
||||
printf " $COLOUR_GREEN -------> OK $COLOUR_DEFAULT \n" |
||||
else |
||||
printf " $COLOUR_RED -------> FAILED $COLOUR_DEFAULT \n" |
||||
cleanup |
||||
exit 1 |
||||
fi |
||||
} |
||||
|
||||
printf "$COLOUR_GREEN========================================================================================= $COLOUR_DEFAULT\n" |
||||
echo "U-boot UMS test program" |
||||
|
||||
if [ $EUID -ne 0 ]; then |
||||
echo "You must be root to do this." 1>&2 |
||||
exit 100 |
||||
fi |
||||
|
||||
if [ $# -lt 3 ]; then |
||||
echo "Wrong number of arguments" |
||||
echo "Example:" |
||||
echo "sudo ./ums_gadget_test.sh VID PID PART_NUM [-f ext4] [test_file]" |
||||
die |
||||
fi |
||||
|
||||
MNT_DIR="/mnt/tmp-ums-test" |
||||
|
||||
VID=$1; shift |
||||
PID=$1; shift |
||||
PART_NUM=$1; shift |
||||
|
||||
if [ "$1" == "-f" ]; then |
||||
shift |
||||
FS_TO_FORMAT=$1; shift |
||||
fi |
||||
|
||||
TEST_FILE=$1 |
||||
|
||||
for f in `find /sys -type f -name idProduct`; do |
||||
d=`dirname ${f}` |
||||
if [ `cat ${d}/idVendor` != "${VID}" ]; then |
||||
continue |
||||
fi |
||||
if [ `cat ${d}/idProduct` != "${PID}" ]; then |
||||
continue |
||||
fi |
||||
USB_DEV=${d} |
||||
break |
||||
done |
||||
|
||||
if [ -z "${USB_DEV}" ]; then |
||||
echo "Connect target" |
||||
echo "e.g. ums 0 mmc 0" |
||||
exit 1 |
||||
fi |
||||
|
||||
MEM_DEV=`find $USB_DEV -type d -name "sd[a-z]" | awk -F/ '{print $(NF)}' -` |
||||
|
||||
mkdir -p $RCV_DIR |
||||
if [ ! -d $MNT_DIR ]; then |
||||
mkdir -p $MNT_DIR |
||||
fi |
||||
|
||||
if [ "$PART_NUM" == "-" ]; then |
||||
PART_NUM="" |
||||
fi |
||||
MEM_DEV=$MEM_DEV$PART_NUM |
||||
|
||||
if [ -n "$FS_TO_FORMAT" ]; then |
||||
echo -n "Formatting partition /dev/$MEM_DEV to $FS_TO_FORMAT" |
||||
mkfs -t $FS_TO_FORMAT /dev/$MEM_DEV > /dev/null 2>&1 |
||||
if [ $? -eq 0 ]; then |
||||
printf " $COLOUR_GREEN DONE $COLOUR_DEFAULT \n" |
||||
else |
||||
die |
||||
fi |
||||
fi |
||||
|
||||
printf "Mount: /dev/$MEM_DEV \n" |
||||
|
||||
if [ -n "$TEST_FILE" ]; then |
||||
if [ ! -e $TEST_FILE ]; then |
||||
echo "No file: $TEST_FILE" |
||||
die |
||||
fi |
||||
ums_test_file $TEST_FILE |
||||
else |
||||
for file in $DIR*.$SUFFIX |
||||
do |
||||
ums_test_file $file |
||||
done |
||||
fi |
||||
|
||||
cleanup |
||||
|
||||
exit 0 |
||||
Loading…
Reference in new issue