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
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
#!/usr/bin/bash
export LC_ALL=C
CE="\033[31m" # Color red
CB="\033[1m" # Color bold
CR="\033[0m" # Color reset
CONFFILE=/etc/memtest86-efi.conf
PRGNAME=$(basename $0)
source $CONFFILE
[[ "$(uname -m)" == "i686" ]] && ARCH=ia32
[[ "$(uname -m)" == "x86_64" ]] && ARCH=x64
warn() {
echo -e "${CB}${CE}This script is unofficial, written by an AUR (Arch User Repository) user. Use it at YOUR OWN RISK.${CR}"
if [[ $EUID -ne 0 ]]; then
echo -e "${CE}You must be root to run $PRGNAME. Aborted.${CR}" > /dev/stderr
exit -1
fi
}
checkcommand() {
if ! $1 &> /dev/null; then
echo -e "${CE}Command $1 not found. Aborted.${CR}" > /dev/stderr
exit 3
fi
}
install() {
if [[ $install == 1 ]]; then
echo "MemTest86 is already installed in ESP. Nothing to do." > /dev/stderr
exit 1
fi
partition=$(fdisk -l | grep "EFI System" | awk '{print $1}' | tail -n1)
esp=$(mount | grep $partition | awk '{print $3}' | tail -n1)
[[ -n $esp ]] && mountpoint=" (mounted on ${CB}$esp${CR})"
echo -en "Press Enter if ${CB}$partition${CR}$mountpoint is your ESP partition, "
echo -en "else enter device path manually (like ${CB}/dev/sdXY${CR}): "
read choice
[[ -n $choice ]] && partition=$choice
esp=$(mount | grep $partition | awk '{print $3}' | tail -n1)
if [[ -z $esp ]]; then
echo -e "${CE}ESP $partition is not mounted, you need to mount it. Aborted.${CR}" > /dev/stderr
exit 2
fi
echo -e "The target is: ${CB}$partition${CR} (mounted on ${CB}$esp${CR}).\n"
echo "Select $PRGNAME action to perform:"
echo -e "${CB}1${CR}: Copy shellx64.efi file on ESP's root (bit safe)"
echo -e "${CB}2${CR}: Add a new EFI boot entry (more safe)"
echo -e "${CB}3${CR}: Add MemTest86 entry in GRUB2 menu"
echo -e "${CB}4${CR}: Cancel"
choice=0
while [[ $choice < 1 ]] || [[ $choice > 4 ]]; do
read choice
echo
done
if [[ $choice == 1 ]]; then
# Install MemTest86 in $esp
echo -e "MemTest86 will be installed in ${CB}$esp${CR}."
[[ -f "$esp/shell$ARCH.efi" ]] && mv -v "$esp/shell$ARCH.efi" "$esp/shell$ARCH.efi.bak" # Backup if exist
cp -v "$MEMTEST86_PATH/"* "$esp/" # Move files in ESP root
mv -v "$esp/boot$ARCH.efi" "$esp/shell$ARCH.efi" # Rename .efi file
elif [[ $choice == 2 ]]; then
# Install MemTest86 in $esp/EFI/memtest86 & Add a boot entry
checkcommand efibootmgr
echo -e "MemTest86 will be installed in ${CB}$esp/EFI/memtest86${CR}."
mkdir -pv "$esp/EFI/memtest86"
cp -v "$MEMTEST86_PATH/"* "$esp/EFI/memtest86/" # Move files in memtest ESP directory
mv -v "$esp/EFI/memtest86/boot$ARCH.efi" "$esp/EFI/memtest86/memtest$ARCH.efi" # Rename .efi file
echo -e "\nAdd a new EFI boot entry..."
efibootmgr -c -d ${partition:0:8} -p ${partition:8} -w -L "MemTest86" -l "\EFI\memtest86\memtest$ARCH.efi" # Manage efi entry
elif [[ $choice == 3 ]]; then
# Install MemTest86 in $esp/EFI/memtest86 & Add a file for GRUB2
checkcommand grub-mkconfig
echo -e "MemTest86 will be installed in ${CB}$esp/EFI/memtest86${CR}."
if [[ ! -d "/etc/grub.d/" ]]; then
echo -e "${CE}GRUB2 seems not installed on your system. Aborted.${CR}" > /dev/stderr
exit 2
fi
mkdir -pv "$esp/EFI/memtest86"
cp -v "$MEMTEST86_PATH/"* "$esp/EFI/memtest86/" # Move files in memtest ESP directory
mv -v "$esp/EFI/memtest86/boot$ARCH.efi" "$esp/EFI/memtest86/memtest$ARCH.efi" # Rename .efi file
echo -e "\nAdd a new configuration file for GRUB..."
echo -e "cat <<-EOF
if [ \"x\\\${grub_platform}\" == \"xefi\" ]; then
menuentry \"Memtest86\" {
search --set=root --no-floppy --fs-uuid $(blkid $partition -s UUID -o value)
chainloader /EFI/memtest86/memtest$ARCH.efi
}
fi
EOF" > "/etc/grub.d/86_memtest"
chmod +x "/etc/grub.d/86_memtest"
grub-mkconfig -o "/boot/grub/grub.cfg"
elif [[ $choice == 4 ]]; then
# Do nothing and quit
echo -e "Canceled. MemTest86 will not be installed."
exit 0
fi
echo "Writting configuration..."
sed -i "s|@PARTITION@|$partition|g" "$CONFFILE"
sed -i "s|@ESP@|$esp|g" "$CONFFILE"
sed -i "s|@CHOICE@|$choice|g" "$CONFFILE"
sed -i "s|install=0|install=1|g" "$CONFFILE"
echo -e "\nMemTest86 has been installed in ESP."
}
update() {
if [[ $install == 0 ]]; then
echo "MemTest86 is not installed in ESP: it cannot be updated." > /dev/stderr
exit 1
fi
if [[ $choice == 1 ]]; then
echo -e "MemTest86 will be updated in ${CB}$esp${CR}."
cp -fv "$MEMTEST86_PATH/"* "$esp/"
mv -fv "$esp/boot$ARCH.efi" "$esp/shell$ARCH.efi"
elif [[ $choice == 2 ]] || [[ $choice == 3 ]]; then
echo -e "MemTest86 will be updated in ${CB}$esp/EFI/memtest86${CR}."
cp -fv "$MEMTEST86_PATH/"* "$esp/EFI/memtest86/"
mv -fv "$esp/EFI/memtest86/boot$ARCH.efi" "$esp/EFI/memtest86/memtest$ARCH.efi"
fi
echo -e "\nMemTest86 has been updated in ESP."
}
remove() {
if [[ $install == 0 ]]; then
echo "MemTest86 is not installed in ESP: it cannot be removed." > /dev/stderr
exit 1
fi
if [[ $choice == 1 ]]; then
echo -e "MemTest86 will be removed from ${CB}$esp${CR}."
rm -v "$esp/mt86.png" "$esp/shell$ARCH.efi"
[[ -f "$esp/shell$ARCH.efi.bak" ]] && mv -v "$esp/shell$ARCH.efi.bak" "$esp/shell$ARCH.efi"
elif [[ $choice == 2 ]]; then
checkcommand efibootmgr
echo -e "MemTest86 will be removed from ${CB}$esp/EFI/memtest86${CR}."
rm -rfv "$esp/EFI/memtest86/"
echo -e "\nRemove MemTest86 EFI boot entry..."
efibootmgr -b $(efibootmgr | grep MemTest86 | cut -c 5-8) -B
elif [[ $choice == 3 ]]; then
checkcommand grub-mkconfig
echo -e "MemTest86 will be removed from ${CB}$esp/EFI/memtest86${CR}."
rm -rfv "$esp/EFI/memtest86/"
echo -e "\nRemove configuration file for GRUB..."
rm -v "/etc/grub.d/86_memtest"
grub-mkconfig -o "/boot/grub/grub.cfg"
fi
echo "Writting configuration..."
sed -i "s|$partition|@PARTITION@|g" "$CONFFILE"
sed -i "s|$esp|@ESP@|g" "$CONFFILE"
sed -i "s|$choice|@CHOICE@|g" "$CONFFILE"
sed -i "s|install=1|install=0|g" "$CONFFILE"
echo -e "\nMemTest86 has been removed from ESP."
}
help() {
echo -e "Usage: $PRGNAME ACTION\n"
echo -e "Available ACTION:"
echo -e "\t-i, --install\t Install MemTest86 in ESP"
echo -e "\t-u, --update\t Update an existing installation of MemTest86"
echo -e "\t-r, --remove\t Remove MemTest86 from ESP"
echo -e "\t-s, --status\t Return status (1 if installed, 0 else)"
echo -e "\t-h, --help\t Print this help and exit"
echo -e "\t-a, --about\t Print informations about $PRGNAME and exit"
}
about() {
echo -e "MemTest86 is a stand alone memory testing software, it cannot be run under an operating system."
echo -e "$PRGNAME is a script which helps you to easily use MemTest86 with your UEFI, as an EFI application."
}
case "$1" in
-i|--install) warn; install; exit 0;;
-u|--update) warn; update; exit 0;;
-r|--remove) warn; remove; exit 0;;
-s|--status) exit $install;;
-h|--help) help; exit 0;;
-a|--about) about; exit 0;;
*) help; exit 1;;
esac
|