blob: 5ba35ea783f982291a1ab6c6f383ff17465ce193 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
#!/bin/sh
. ../../lib/sh-test-lib
OUTPUT="$(pwd)/output"
RESULT_FILE="${OUTPUT}/result.txt"
ITERATION="5"
UNITS="MB/s"
usage() {
echo "Usage: $0 [-p <partition>] [-t <type>] [-i <iteration>] [-s <true>]" 1>&2
exit 1
}
while getopts "p:t:i:s:" o; do
case "$o" in
# The current working directory will be used by default.
# Use '-p' specify partition that used for dd test.
p) PARTITION="${OPTARG}" ;;
# CAUTION: if FS_TYPE not equal to the existing fs type of the partition
# specified with '-p', the partition will be formatted.
t) FS_TYPE="${OPTARG}" ;;
# You may need to run dd test 4-5 times for an accurate evaluation.
i) ITERATION="${OPTARG}" ;;
s) SKIP_INSTALL="${OPTARG}" ;;
*) usage ;;
esac
done
prepare_partition() {
if [ -n "${PARTITION}" ]; then
device_attribute="$(blkid | grep "${PARTITION}")"
[ -z "${device_attribute}" ] && error_msg "${PARTITION} NOT found"
fs_type=$(echo "${device_attribute}" \
| grep "TYPE=" \
| awk '{print $3}' \
| awk -F '"' '{print $2}')
# If PARTITION specified, its FS_TYPE needs to be specified explicitly.
[ -z "${FS_TYPE}" ] && error_msg "Please specify ${FS_TYPE} explicitly"
# Try to format the partition if it is unformatted or not the same as
# the filesystem type specified with parameter '-t'.
if [ -n "${FS_TYPE}" ]; then
if [ "${FS_TYPE}" != "${fs_type}" ]; then
umount "${PARTITION}" > /dev/null 2>&1
info_msg "Formatting ${PARTITION} to ${FS_TYPE}..."
if [ "${FS_TYPE}" = "fat32" ]; then
echo "y" | mkfs -t vfat -F 32 "${PARTITION}"
else
echo "y" | mkfs -t "${FS_TYPE}" "${PARTITION}"
fi
if [ $? -ne 0 ]; then
error_msg "unable to format ${PARTITION}"
else
info_msg "${PARTITION} formatted to ${FS_TYPE}"
fi
fi
fi
# Mount the partition and enter its mount point.
mount_point="$(df |grep "${PARTITION}" | awk '{print $NF}')"
if [ -z "${mount_point}" ]; then
mount_point="/mnt"
mount "${PARTITION}" "${mount_point}"
if [ $? -ne 0 ]; then
error_msg "Unable to mount ${PARTITIOIN}"
else
info_msg "${PARTITION} mounted to ${mount_point}"
fi
fi
cd "${mount_point}"
fi
}
dd_write() {
echo
echo "--- dd write speed test ---"
rm -f dd-write-output.txt
for i in $(seq "${ITERATION}"); do
echo "Running iteration ${i}..."
rm -f dd.img
echo 3 > /proc/sys/vm/drop_caches
dd if=/dev/zero of=dd.img bs=1048576 count=1024 conv=fsync 2>&1 \
| tee -a "${OUTPUT}"/dd-write-output.txt
done
}
dd_read() {
echo
echo "--- dd read speed test ---"
rm -f dd-read-output.txt
for i in $(seq "${ITERATION}"); do
echo "Running iteration ${i}..."
echo 3 > /proc/sys/vm/drop_caches
dd if=dd.img of=/dev/null bs=1048576 count=1024 2>&1 \
| tee -a "${OUTPUT}"/dd-read-output.txt
done
rm -f dd.img
}
parse_output() {
local test="$1"
local test_case_id="${test}"
if ! [ -f "${OUTPUT}/${test}-output.txt" ]; then
warn_msg "output file is missing"
return 1
fi
# Fixup test-case-id with filesystem type and partion name.
[ -n "${FS_TYPE}" ] && test_case_id="${FS_TYPE}-${test_case_id}"
if [ -n "${PARTITION}" ]; then
partition_no="$(echo "${PARTITION}" |awk -F '/' '{print $NF}')"
test_case_id="${partition_no}-${test_case_id}"
fi
itr=1
while read line; do
if echo "${line}" | egrep -q "(M|G)B/s"; then
measurement="$(echo "${line}" | awk '{print $(NF-1)}')"
units="$(echo "${line}" | awk '{print $NF}')"
if [ "${units}" = "GB/s" ]; then
measurement=$(( measurement * 1024 ))
elif [ "${units}" = "KB/s" ]; then
measurement=$(( measurement / 1024 ))
fi
add_metric "${test_case_id}-itr${itr}" "pass" "${measurement}" "${UNITS}"
itr=$(( itr + 1 ))
fi
done < "${OUTPUT}/${test}-output.txt"
# For mutiple times dd test, calculate the mean, min and max values.
# Save them to result.txt.
if [ "${ITERATION}" -gt 1 ]; then
eval "$(grep "${test}" "${OUTPUT}"/result.txt \
| awk '{
if(min=="") {min=max=$3};
if($3>max) {max=$3};
if($3< min) {min=$3};
total+=$3; count+=1;
}
END {
print "mean="total/count, "min="min, "max="max;
}')"
add_metric "${test_case_id}-mean" "pass" "${mean}" "${UNITS}"
add_metric "${test_case_id}-min" "pass" "${min}" "${UNITS}"
add_metric "${test_case_id}-max" "pass" "${max}" "${UNITS}"
fi
}
# Test run.
! check_root && error_msg "This script must be run as root"
[ -d "${OUTPUT}" ] && mv "${OUTPUT}" "${OUTPUT}_$(date +%Y%m%d%H%M%S)"
mkdir -p "${OUTPUT}"
info_msg "About to run dd test..."
info_msg "Output directory: ${OUTPUT}"
pkgs="e2fsprogs dosfstools"
install_deps "${pkgs}" "${SKIP_INSTALL}"
prepare_partition
info_msg "dd test directory: $(pwd)"
dd_write
parse_output "dd-write"
dd_read
parse_output "dd-read"
|