xref: /freebsd/usr.bin/mkimg/tests/mkimg_test.sh (revision 1f474190) 
1# $FreeBSD$
2
3mkimg_blksz_list="512 4096"
4mkimg_format_list="qcow qcow2 raw vhd vhdf vhdx vmdk"
5mkimg_geom_list="1x1 63x255"
6mkimg_scheme_list="apm bsd ebr gpt mbr vtoc8"
7
8bootcode()
9{
10    case $1 in
11      bsd)	echo 8192 ;;
12      gpt|mbr)	echo 512 ;;
13      *)	echo 0 ;;
14    esac
15    return 0
16}
17
18mkcontents()
19{
20    local byte count name
21
22    byte=$1
23    count=$2
24
25    name=_tmp-$byte-$count.bin
26    jot -b $byte $(($count/2)) > $name
27    echo $name
28    return 0
29}
30
31makeimage()
32{
33    local blksz bootarg bootsz format geom nhds nsecs partarg pfx scheme
34
35    format=$1
36    scheme=$2
37    blksz=$3
38    geom=$4
39    pfx=$5
40    shift 5
41
42    nsecs=${geom%x*}
43    nhds=${geom#*x}
44
45    bootsz=`bootcode $scheme`
46    if test $bootsz -gt 0; then
47	bootarg="-b `mkcontents B $bootsz`"
48    else
49	bootarg=""
50    fi
51
52    partarg=""
53    for P in $*; do
54	partarg="$partarg -p $P"
55    done
56    if test -z "$partarg"; then
57	local swap ufs
58	swap="-p freebsd-swap::128K"
59	ufs="-p freebsd-ufs:=$(atf_get_srcdir)/partition_data_4M.bin"
60	partarg="$ufs $swap"
61    fi
62
63    imagename=$pfx-$geom-$blksz-$scheme.$format
64
65    mkimg -y -f $format -o $imagename -s $scheme -P $blksz -H $nhds -T $nsecs \
66	    $bootarg $partarg
67    echo $imagename
68    return 0
69}
70
71mkimg_rebase()
72{
73    local baseline image result tmpfile update
74
75    image=$1
76    result=$2
77
78    baseline=$image.hex
79    update=yes
80
81    if test -f $baseline; then
82	tmpfile=_tmp-baseline
83	sed -e '/^#.*/D' < $baseline > $tmpfile
84	if diff -u $tmpfile $result; then
85	    update=no
86	fi
87    fi
88
89    if test $update = yes; then
90	# Prevent keyword expansion when writing the keyword.
91	(echo -n '# $'; echo -n FreeBSD; echo '$') > $baseline
92	cat $result >> $baseline
93    fi
94
95    rm $image $result _tmp-*
96    return 0
97}
98
99mkimg_test()
100{
101    local blksz format geom scheme
102
103    geom=$1
104    blksz=$2
105    scheme=$3
106    format=$4
107
108    case $scheme in
109      ebr|mbr)
110	bsd=`makeimage raw bsd $blksz $geom _tmp`
111	partinfo="freebsd:=$bsd"
112	;;
113      *)
114	partinfo=""
115	;;
116    esac
117    image=`makeimage $format $scheme $blksz $geom img $partinfo`
118    result=$image.out
119    hexdump -C $image > $result
120    if test "x$mkimg_update_baseline" = "xyes"; then
121	mkimg_rebase $image $result
122    else
123	baseline=`atf_get_srcdir`/$image
124	atf_check -s exit:0 diff -u $baseline $result
125    fi
126    return 0
127}
128
129atf_test_case rebase
130rebase_body()
131{
132    local nm
133
134    mkimg_update_baseline=yes
135    for nm in $mkimg_tests; do
136	${nm}_body
137    done
138    return 0
139}
140
141atf_init_test_cases()
142{
143    local B F G S nm
144
145    for G in $mkimg_geom_list; do
146	for B in $mkimg_blksz_list; do
147	    for S in $mkimg_scheme_list; do
148		for F in $mkimg_format_list; do
149		    nm="${S}_${G}_${B}_${F}"
150		    atf_test_case $nm
151		    eval "${nm}_body() { mkimg_test $G $B $S $F; }"
152		    mkimg_tests="${mkimg_tests} ${nm}"
153		    atf_add_test_case $nm
154		done
155	    done
156	done
157    done
158
159    # XXX hack to make updating the baseline easier
160    if test "${__RUNNING_INSIDE_ATF_RUN}" != "internal-yes-value"; then
161	atf_add_test_case rebase
162    fi
163}
164
165