anduin zrewidował ten Gist 1 year ago. Przejdź do rewizji
1 file changed, 1 insertion
burn.sh
| @@ -38,6 +38,7 @@ function print_ok() { | |||
| 38 | 38 | ||
| 39 | 39 | function print_error() { | |
| 40 | 40 | echo -e "${ERROR} ${Red} $1 ${Font}" | |
| 41 | + | exit 1 | |
| 41 | 42 | } | |
| 42 | 43 | ||
| 43 | 44 | function print_warn() { | |
anduin zrewidował ten Gist 1 year ago. Przejdź do rewizji
1 file changed, 30 insertions, 19 deletions
burn.sh
| @@ -59,23 +59,23 @@ function judge() { | |||
| 59 | 59 | ||
| 60 | 60 | function burn() | |
| 61 | 61 | { | |
| 62 | - | # If the user want to burn /dev/sda, device should be 'sda' | |
| 63 | - | device=$1 | |
| 64 | - | iso=$2 | |
| 62 | + | # If the user want to burn /dev/sda, device should be '/dev/sda' | |
| 63 | + | iso=$1 | |
| 64 | + | device=$2 | |
| 65 | 65 | ||
| 66 | 66 | if [ -z "$device" ]; then | |
| 67 | - | print_error "Usage: $0 <device> <iso>" | |
| 67 | + | print_error "Usage: burn.sh /path/to/iso /dev/device" | |
| 68 | 68 | exit 1 | |
| 69 | 69 | fi | |
| 70 | 70 | ||
| 71 | - | # Ensure /dev/$device exists | |
| 72 | - | if [ ! -e "/dev/$device" ]; then | |
| 73 | - | print_error "Device /dev/$device does not exist." | |
| 71 | + | # Ensure $device exists | |
| 72 | + | if [ ! -e "$device" ]; then | |
| 73 | + | print_error "Device $device does not exist." | |
| 74 | 74 | exit 1 | |
| 75 | 75 | fi | |
| 76 | 76 | ||
| 77 | - | # Unmount all partitions on /dev/$device | |
| 78 | - | for mountpoint in $(lsblk -o MOUNTPOINT "/dev/$device" | tail -n +2); do | |
| 77 | + | # Unmount all partitions on /$device | |
| 78 | + | for mountpoint in $(lsblk -o MOUNTPOINT "$device" | tail -n +2); do | |
| 79 | 79 | print_ok "Unmounting $mountpoint" | |
| 80 | 80 | sudo umount "$mountpoint" | |
| 81 | 81 | judge "Unmount $mountpoint" | |
| @@ -101,33 +101,36 @@ function burn() | |||
| 101 | 101 | ||
| 102 | 102 | # Ensure the disk is larger than the iso | |
| 103 | 103 | iso_size=$(stat -c %s "$iso") | |
| 104 | - | device_size=$(sudo blockdev --getsize64 "/dev/$device") | |
| 104 | + | device_size=$(sudo blockdev --getsize64 "$device") | |
| 105 | 105 | if [ "$iso_size" -gt "$device_size" ]; then | |
| 106 | - | print_error "ISO $iso is larger than /dev/$device." | |
| 106 | + | print_error "ISO $iso is larger than $device." | |
| 107 | 107 | exit 1 | |
| 108 | 108 | else | |
| 109 | - | print_ok "ISO $iso is $iso_size bytes, /dev/$device is $device_size bytes. Device is large enough." | |
| 109 | + | print_ok "ISO $iso is $iso_size bytes, $device is $device_size bytes. Device is large enough." | |
| 110 | 110 | fi | |
| 111 | 111 | ||
| 112 | - | print_ok "Burning $iso to /dev/$device. This will take a while..." | |
| 113 | - | sudo dd if="$iso" of="/dev/$device" bs=4M status=progress oflag=sync | |
| 112 | + | print_ok "Burning $iso to $device. This will take a while..." | |
| 113 | + | sudo dd if="$iso" of="$device" bs=4M status=progress oflag=sync | |
| 114 | 114 | sync && sync && sync | |
| 115 | 115 | ||
| 116 | + | print_ok "Sleep 10 seconds to ensure the USB write buffer is flushed." | |
| 117 | + | sleep 10 | |
| 118 | + | ||
| 116 | 119 | # Verify the burn | |
| 117 | 120 | print_ok "Verifying the burn. This will take a while." | |
| 118 | 121 | ||
| 119 | 122 | # Foreach the mount point under $device, unmount it | |
| 120 | - | for mountpoint in $(lsblk -o MOUNTPOINT "/dev/$device" | tail -n +2); do | |
| 123 | + | for mountpoint in $(lsblk -o MOUNTPOINT "$device" | tail -n +2); do | |
| 121 | 124 | print_ok "Unmounting $mountpoint" | |
| 122 | 125 | sudo umount "$mountpoint" | |
| 123 | 126 | judge "Unmount $mountpoint" | |
| 124 | 127 | done | |
| 125 | 128 | ||
| 126 | 129 | # Mount the device to /tmp/burn/$device | |
| 127 | - | print_ok "Mounting /dev/$device to /tmp/burn/$device" | |
| 130 | + | print_ok "Mounting $device to /tmp/burn/$device" | |
| 128 | 131 | mkdir -p /tmp/burn/$device | |
| 129 | - | sudo mount "/dev/$device" /tmp/burn/$device | |
| 130 | - | judge "Mount /dev/$device to /tmp/burn/$device" | |
| 132 | + | sudo mount "$device" /tmp/burn/$device | |
| 133 | + | judge "Mount $device to /tmp/burn/$device" | |
| 131 | 134 | ||
| 132 | 135 | # Calculate the md5sum of the iso | |
| 133 | 136 | print_ok "Calculating the md5sum of the files in the iso" | |
| @@ -146,6 +149,7 @@ function burn() | |||
| 146 | 149 | ||
| 147 | 150 | # Unmount the device | |
| 148 | 151 | print_ok "Unmounting /tmp/burn/$device" | |
| 152 | + | sleep 1 | |
| 149 | 153 | sudo umount /tmp/burn/$device | |
| 150 | 154 | judge "Unmount /tmp/burn/$device" | |
| 151 | 155 | ||
| @@ -154,7 +158,14 @@ function burn() | |||
| 154 | 158 | rm -rf /tmp/burn/$device | |
| 155 | 159 | judge "Clean up" | |
| 156 | 160 | ||
| 157 | - | print_ok "Done. You can now remove the USB stick." | |
| 161 | + | # Ensure the USB stick safe to eject | |
| 162 | + | print_ok "Ejecting $device" | |
| 163 | + | sleep 5 | |
| 164 | + | udisksctl power-off -b "$device" | |
| 165 | + | judge "Eject $device" | |
| 166 | + | ||
| 167 | + | print_ok "Done. The usb drive is safe to remove." | |
| 158 | 168 | } | |
| 159 | 169 | ||
| 170 | + | clear | |
| 160 | 171 | burn "$@" | |
anduin zrewidował ten Gist 1 year ago. Przejdź do rewizji
1 file changed, 12 insertions, 2 deletions
burn.sh
| @@ -99,9 +99,19 @@ function burn() | |||
| 99 | 99 | exit 1 | |
| 100 | 100 | fi | |
| 101 | 101 | ||
| 102 | - | print_ok "Burning $iso to /dev/$device. This will take a while." | |
| 102 | + | # Ensure the disk is larger than the iso | |
| 103 | + | iso_size=$(stat -c %s "$iso") | |
| 104 | + | device_size=$(sudo blockdev --getsize64 "/dev/$device") | |
| 105 | + | if [ "$iso_size" -gt "$device_size" ]; then | |
| 106 | + | print_error "ISO $iso is larger than /dev/$device." | |
| 107 | + | exit 1 | |
| 108 | + | else | |
| 109 | + | print_ok "ISO $iso is $iso_size bytes, /dev/$device is $device_size bytes. Device is large enough." | |
| 110 | + | fi | |
| 111 | + | ||
| 112 | + | print_ok "Burning $iso to /dev/$device. This will take a while..." | |
| 103 | 113 | sudo dd if="$iso" of="/dev/$device" bs=4M status=progress oflag=sync | |
| 104 | - | sync | |
| 114 | + | sync && sync && sync | |
| 105 | 115 | ||
| 106 | 116 | # Verify the burn | |
| 107 | 117 | print_ok "Verifying the burn. This will take a while." | |
anduin zrewidował ten Gist 1 year ago. Przejdź do rewizji
1 file changed, 9 insertions, 2 deletions
burn.sh
| @@ -120,10 +120,17 @@ function burn() | |||
| 120 | 120 | judge "Mount /dev/$device to /tmp/burn/$device" | |
| 121 | 121 | ||
| 122 | 122 | # Calculate the md5sum of the iso | |
| 123 | + | print_ok "Calculating the md5sum of the files in the iso" | |
| 123 | 124 | ( | |
| 124 | 125 | cd /tmp/burn/$device && \ | |
| 125 | - | sudo md5sum -c md5sum.txt --quiet && \ | |
| 126 | - | print_ok "Burn successful." || \ | |
| 126 | + | sudo md5sum -c md5sum.txt && \ | |
| 127 | + | print_ok "First pass verification succeeded." || \ | |
| 128 | + | print_error "Burn failed." | |
| 129 | + | ) | |
| 130 | + | ( | |
| 131 | + | cd /tmp/burn/$device && \ | |
| 132 | + | sudo md5sum -c md5sum.txt && \ | |
| 133 | + | print_ok "Second pass verification succeeded." || \ | |
| 127 | 134 | print_error "Burn failed." | |
| 128 | 135 | ) | |
| 129 | 136 | ||
anduin zrewidował ten Gist 1 year ago. Przejdź do rewizji
1 file changed, 143 insertions
burn.sh(stworzono plik)
| @@ -0,0 +1,143 @@ | |||
| 1 | + | #!/bin/bash | |
| 2 | + | ||
| 3 | + | #========================== | |
| 4 | + | # Set up the environment | |
| 5 | + | #========================== | |
| 6 | + | set -e # exit on error | |
| 7 | + | set -o pipefail # exit on pipeline error | |
| 8 | + | set -u # treat unset variable as error | |
| 9 | + | ||
| 10 | + | #========================== | |
| 11 | + | # Basic Information | |
| 12 | + | #========================== | |
| 13 | + | export LC_ALL=C | |
| 14 | + | export LANG=en_US.UTF-8 | |
| 15 | + | export DEBIAN_FRONTEND=noninteractive | |
| 16 | + | export SCRIPT_DIR="$(dirname "$(readlink -f "$0")")" | |
| 17 | + | ||
| 18 | + | #========================== | |
| 19 | + | # Color | |
| 20 | + | #========================== | |
| 21 | + | Green="\033[32m" | |
| 22 | + | Red="\033[31m" | |
| 23 | + | Yellow="\033[33m" | |
| 24 | + | Blue="\033[36m" | |
| 25 | + | Font="\033[0m" | |
| 26 | + | GreenBG="\033[42;37m" | |
| 27 | + | RedBG="\033[41;37m" | |
| 28 | + | OK="${Green}[ OK ]${Font}" | |
| 29 | + | ERROR="${Red}[FAILED]${Font}" | |
| 30 | + | WARNING="${Yellow}[ WARN ]${Font}" | |
| 31 | + | ||
| 32 | + | #========================== | |
| 33 | + | # Print Colorful Text | |
| 34 | + | #========================== | |
| 35 | + | function print_ok() { | |
| 36 | + | echo -e "${OK} ${Blue} $1 ${Font}" | |
| 37 | + | } | |
| 38 | + | ||
| 39 | + | function print_error() { | |
| 40 | + | echo -e "${ERROR} ${Red} $1 ${Font}" | |
| 41 | + | } | |
| 42 | + | ||
| 43 | + | function print_warn() { | |
| 44 | + | echo -e "${WARNING} ${Yellow} $1 ${Font}" | |
| 45 | + | } | |
| 46 | + | ||
| 47 | + | #========================== | |
| 48 | + | # Judge function | |
| 49 | + | #========================== | |
| 50 | + | function judge() { | |
| 51 | + | if [[ 0 -eq $? ]]; then | |
| 52 | + | print_ok "$1 succeeded" | |
| 53 | + | sleep 0.2 | |
| 54 | + | else | |
| 55 | + | print_error "$1 failed" | |
| 56 | + | exit 1 | |
| 57 | + | fi | |
| 58 | + | } | |
| 59 | + | ||
| 60 | + | function burn() | |
| 61 | + | { | |
| 62 | + | # If the user want to burn /dev/sda, device should be 'sda' | |
| 63 | + | device=$1 | |
| 64 | + | iso=$2 | |
| 65 | + | ||
| 66 | + | if [ -z "$device" ]; then | |
| 67 | + | print_error "Usage: $0 <device> <iso>" | |
| 68 | + | exit 1 | |
| 69 | + | fi | |
| 70 | + | ||
| 71 | + | # Ensure /dev/$device exists | |
| 72 | + | if [ ! -e "/dev/$device" ]; then | |
| 73 | + | print_error "Device /dev/$device does not exist." | |
| 74 | + | exit 1 | |
| 75 | + | fi | |
| 76 | + | ||
| 77 | + | # Unmount all partitions on /dev/$device | |
| 78 | + | for mountpoint in $(lsblk -o MOUNTPOINT "/dev/$device" | tail -n +2); do | |
| 79 | + | print_ok "Unmounting $mountpoint" | |
| 80 | + | sudo umount "$mountpoint" | |
| 81 | + | judge "Unmount $mountpoint" | |
| 82 | + | done | |
| 83 | + | ||
| 84 | + | # Ensure the iso exists | |
| 85 | + | if [ ! -e "$iso" ]; then | |
| 86 | + | print_error "ISO $iso does not exist." | |
| 87 | + | exit 1 | |
| 88 | + | fi | |
| 89 | + | ||
| 90 | + | # Ensure the iso is a file | |
| 91 | + | if [ ! -f "$iso" ]; then | |
| 92 | + | print_error "ISO $iso is not a file." | |
| 93 | + | exit 1 | |
| 94 | + | fi | |
| 95 | + | ||
| 96 | + | # Ensure the iso is readable | |
| 97 | + | if [ ! -r "$iso" ]; then | |
| 98 | + | print_error "ISO $iso is not readable." | |
| 99 | + | exit 1 | |
| 100 | + | fi | |
| 101 | + | ||
| 102 | + | print_ok "Burning $iso to /dev/$device. This will take a while." | |
| 103 | + | sudo dd if="$iso" of="/dev/$device" bs=4M status=progress oflag=sync | |
| 104 | + | sync | |
| 105 | + | ||
| 106 | + | # Verify the burn | |
| 107 | + | print_ok "Verifying the burn. This will take a while." | |
| 108 | + | ||
| 109 | + | # Foreach the mount point under $device, unmount it | |
| 110 | + | for mountpoint in $(lsblk -o MOUNTPOINT "/dev/$device" | tail -n +2); do | |
| 111 | + | print_ok "Unmounting $mountpoint" | |
| 112 | + | sudo umount "$mountpoint" | |
| 113 | + | judge "Unmount $mountpoint" | |
| 114 | + | done | |
| 115 | + | ||
| 116 | + | # Mount the device to /tmp/burn/$device | |
| 117 | + | print_ok "Mounting /dev/$device to /tmp/burn/$device" | |
| 118 | + | mkdir -p /tmp/burn/$device | |
| 119 | + | sudo mount "/dev/$device" /tmp/burn/$device | |
| 120 | + | judge "Mount /dev/$device to /tmp/burn/$device" | |
| 121 | + | ||
| 122 | + | # Calculate the md5sum of the iso | |
| 123 | + | ( | |
| 124 | + | cd /tmp/burn/$device && \ | |
| 125 | + | sudo md5sum -c md5sum.txt --quiet && \ | |
| 126 | + | print_ok "Burn successful." || \ | |
| 127 | + | print_error "Burn failed." | |
| 128 | + | ) | |
| 129 | + | ||
| 130 | + | # Unmount the device | |
| 131 | + | print_ok "Unmounting /tmp/burn/$device" | |
| 132 | + | sudo umount /tmp/burn/$device | |
| 133 | + | judge "Unmount /tmp/burn/$device" | |
| 134 | + | ||
| 135 | + | # Clean up | |
| 136 | + | print_ok "Cleaning up" | |
| 137 | + | rm -rf /tmp/burn/$device | |
| 138 | + | judge "Clean up" | |
| 139 | + | ||
| 140 | + | print_ok "Done. You can now remove the USB stick." | |
| 141 | + | } | |
| 142 | + | ||
| 143 | + | burn "$@" | |