tazusb annotate tazusb @ rev 12

Interactive mode if no device are specified with gen-liveusb/format
author hacker@slitaz
date Thu Mar 13 00:05:10 2008 +0100 (2008-03-13)
parents bc7ca2e2616e
children a2778922137f
rev   line source
spode@0 1 #!/bin/sh
spode@0 2 # Tazusb - SliTaz LiveUSB
spode@0 3 #
spode@0 4 # Tazusb is an utility to generate, configure and manipulate SliTaz LiveUSB
spode@0 5 # bootable media and/or USB /home partition, such as flash keys, SD card or
spode@0 6 # USB harddisk.
spode@0 7 #
spode@0 8 # Authors : Christophe Lincoln (Pankso) <pankso@slitaz.org>
spode@0 9 # Andrew Miller (Spode) <spode@spodesabode.com>
spode@0 10 #
hacker@12 11 VERSION=20080312
spode@0 12
spode@0 13 COMMAND=$1
spode@0 14 TARGET_ROOT=/media/flash
spode@0 15 DRIVE_NAME=`cat /proc/sys/dev/cdrom/info | grep "drive name" | cut -f 3`
spode@0 16 CDROM=/dev/$DRIVE_NAME
spode@0 17
spode@0 18 #
spode@0 19 # Tazusb functions
spode@0 20 #
spode@0 21
spode@0 22 # Print the usage.
spode@0 23 usage ()
spode@0 24 {
spode@0 25 echo -e "\nSliTaz Live USB - Version: $VERSION\n
spode@0 26 \033[1mUsage: \033[0m `basename $0` [command] [compression|device]
spode@0 27 \033[1mCommands: \033[0m\n
spode@0 28 usage Print this short usage.
spode@0 29 writefs Write the current filesystem to rootfs.gz.
spode@0 30 tazSupported compression: lzma. gzip, none.
pankso@3 31 format Format and label device with ext3 filesystem
pankso@3 32 (for LiveUSB or /home).
spode@0 33 gen-liveusb Generate a bootable LiveUSB using files from the LiveCD.\n"
spode@0 34 }
spode@0 35
spode@0 36 # Status function.
spode@0 37 status()
spode@0 38 {
spode@0 39 local CHECK=$?
spode@0 40 echo -en "\\033[70G[ "
spode@0 41 if [ $CHECK = 0 ]; then
spode@0 42 echo -en "\\033[1;33mOK"
spode@0 43 else
spode@0 44 echo -en "\\033[1;31mFailed"
spode@0 45 fi
spode@0 46 echo -e "\\033[0;39m ]"
spode@0 47 }
spode@0 48
spode@0 49 # Exit if user is not root.
spode@0 50 check_root()
spode@0 51 {
spode@0 52 if test $(id -u) != 0 ; then
spode@0 53 echo -e "\nThis program requires being run as root.\n"
spode@0 54 exit 0
spode@0 55 fi
spode@0 56 }
spode@0 57
hacker@12 58 # Display a list of available partition.
hacker@12 59 fdisk_list()
hacker@12 60 {
hacker@12 61 echo ""
hacker@12 62 fdisk -l | grep ^/dev/sd*
hacker@12 63 echo ""
hacker@12 64 }
hacker@12 65
hacker@12 66 # We need a USB media to install.
hacker@12 67 ask_for_device()
hacker@12 68 {
hacker@12 69 echo -n "\
hacker@12 70 Please specify the target USB device to $COMMAND. You can type 'list' to
hacker@12 71 get a list of devices, type 'exit' or give an empty value to exit.
hacker@12 72
hacker@12 73 Device to use : "; read anser
hacker@12 74 while [ "$anser" == "list" ]; do
hacker@12 75 fdisk_list
hacker@12 76 echo -n "Device to use : "; read anser
hacker@12 77 done
hacker@12 78 if [ "$anser" = "" -o "$anser" = "exit" ]; then
hacker@12 79 echo -e "\nNo specified device or exit.\n"
hacker@12 80 exit 0
hacker@12 81 else
hacker@12 82 DEVICE=$anser
hacker@12 83 fi
hacker@12 84 }
hacker@12 85
spode@0 86 # Verify a device exists before format or install
spode@0 87 check_for_device()
spode@0 88 {
spode@9 89 IFDEV=`fdisk -l | grep $DEVICE`
spode@9 90 if [ -z "$IFDEV" ]; then
spode@0 91 echo -e "\nUnable to find device: $DEVICE\n"
spode@0 92 exit 0
spode@0 93 fi
spode@0 94 }
spode@0 95
spode@0 96 #gets the UUID and filesystem type
spode@0 97 get_part_info()
spode@0 98 {
spode@9 99 UUID=`blkid -s UUID -o value $DEVICE`
spode@9 100 FSTYPE=`blkid -s TYPE -o value $DEVICE`
spode@0 101 }
spode@0 102
spode@0 103 # Format target device and label partition.
spode@0 104 mkfs_ext3()
spode@0 105 {
spode@0 106 echo -n "Please specify a label for the partition (TazUSB): "
spode@0 107 read label
spode@0 108
spode@0 109 if [ -z $label ]; then
spode@0 110 label=TazUSB
spode@0 111 fi
spode@0 112
spode@0 113 echo "Label : $label"
spode@9 114 echo "Mkfs : mkfs.ext3 -L \"$label\" $DEVICE"
spode@0 115 echo "" && sleep 2
spode@9 116 mkfs.ext3 -L "$label" $DEVICE
spode@0 117
spode@0 118 }
spode@0 119
spode@0 120 # Mount an existing USB device.
spode@0 121 unmount_target_usb()
spode@0 122 {
spode@0 123 # If mount point is in use, unmount
spode@0 124 if mount | grep $TARGET_ROOT; then
spode@0 125 umount $TARGET_ROOT
spode@0 126 fi
spode@0 127
spode@0 128 # Device could be mounted elsewhere, so unmount
spode@9 129 if mount | grep $DEVICE; then
spode@0 130 echo "Unmounting USB target device..."
spode@9 131 umount $DEVICE
spode@0 132 fi
spode@0 133 }
spode@0 134
spode@0 135 # Mount an existing USB device.
spode@0 136 mount_target_usb()
spode@0 137 {
spode@0 138 echo "Mounting USB target device..."
spode@0 139 mkdir -p $TARGET_ROOT
spode@9 140 mount $DEVICE $TARGET_ROOT 2>/dev/null
spode@0 141 }
spode@0 142
spode@0 143 # Mount SliTaz LiveCD to get needed files.
spode@0 144 mount_cdrom()
spode@0 145 {
spode@0 146 echo "Mounting cdrom device..."
spode@0 147
spode@0 148 if mount | grep /media/cdrom; then
spode@0 149 umount /media/cdrom
spode@0 150 fi
spode@0 151
spode@0 152 mkdir -p /media/cdrom
hacker@12 153 mount -r -t iso9660 $CDROM /media/cdrom 2>/dev/null
spode@0 154
spode@0 155 if [ ! -f /media/cdrom/boot/rootfs.gz ]; then
hacker@12 156 echo -e "\nUnable to mount cdrom or to find a filesystem on it (rootfs.gz).\n"
spode@0 157 exit 0
spode@0 158 fi
spode@0 159 }
spode@0 160
spode@0 161 # All needed files are in the boot direcory of the cdrom.
spode@0 162 copy_cdrom_files()
spode@0 163 {
spode@0 164 echo -n "Copying needed files from cdrom..."
spode@0 165 mkdir -p $TARGET_ROOT/boot
spode@0 166 cp /media/cdrom/boot/bzImage $TARGET_ROOT/boot
spode@0 167 cp /media/cdrom/boot/rootfs.gz $TARGET_ROOT/boot
spode@0 168 status
spode@0 169 }
spode@0 170
spode@0 171 install_mbr()
spode@0 172 {
spode@0 173 # MBR
pankso@11 174 DISK=${DEVICE%[1-99]}
spode@0 175 if [ -f /usr/share/syslinux/mbr.bin ]; then
pankso@11 176 echo -n "Installing a new MBR to: $DISK"
pankso@11 177 cat /usr/share/syslinux/mbr.bin > $DISK
spode@0 178 status
spode@0 179 else
spode@0 180 # Skip MBR install (tazpkg get-install syslinux-extra ? and then cat)
hacker@12 181 echo "No new MBR installed to: $DISK"
spode@0 182 fi
spode@0 183 }
spode@0 184
spode@0 185 # ext/syslinux install
spode@0 186 install_boot()
spode@0 187 {
spode@0 188 #decide if we're installing syslinux or extlinux
spode@0 189 if [ "$FSTYPE" = "vfat" ]; then
spode@0 190 ST=syslinux
spode@9 191 STC="syslinux -d /boot/syslinux/ $DEVICE"
spode@0 192 STE=cfg
spode@0 193 else
spode@0 194 ST=extlinux
spode@0 195 STC="extlinux --install $TARGET_ROOT/boot/$ST"
spode@0 196 STE=conf
spode@0 197 fi
spode@0 198
spode@0 199 echo "Installing bootloader: $ST"
spode@0 200 mkdir -p $TARGET_ROOT/boot/$ST
spode@0 201 $STC
spode@0 202
spode@0 203 # extlinux.conf / syslinux.cfg
spode@0 204 cat > $TARGET_ROOT/boot/$ST/$ST.$STE << _EOT_
spode@0 205 display display.txt
spode@0 206 default slitaz
pankso@3 207 timeout 20
spode@0 208 label slitaz
spode@0 209 kernel /boot/bzImage
spode@0 210 append initrd=/boot/rootfs.gz rw root=/dev/null home=$UUID
spode@0 211
spode@0 212 label previous
spode@0 213 kernel /boot/bzImage
spode@0 214 append initrd=/boot/previous.gz rw root=/dev/null home=$UUID
spode@0 215
spode@0 216 _EOT_
spode@0 217
spode@0 218 # display.txt
spode@0 219 cat > $TARGET_ROOT/boot/$ST/display.txt << "EOT"
spode@0 220 _______. __ __ .___________. ___ ________
spode@0 221 / || | | | | | / \ | /
spode@0 222 | (----`| | | | `---| |---` / ^ \ `---/ /
spode@0 223 \ \ | | | | | | / /_\ \ / /
spode@0 224 .----) | | `----.| | | | / _____ \ / /----.
spode@0 225 |_______/ |_______||__| |__| /__/ \__\ /________|
spode@0 226
spode@0 227
spode@0 228 SliTaz GNU/Linux LiveUSB
spode@0 229 Simple Light Incredible Temporary Autonomus Zone
spode@0 230
spode@0 231
spode@0 232 EOT
spode@0 233 status
spode@0 234 }
spode@0 235
spode@0 236 # Let user exit or reboot.
spode@0 237 exit_or_reboot()
spode@0 238 {
spode@0 239 echo ""
spode@0 240 echo -n "Do you want to exit Tazusb or reboot system (Exit/reboot) ? "
spode@0 241 read anser
spode@0 242 if [ "$anser" == "reboot" ]; then
spode@0 243 umount $TARGET_ROOT
spode@0 244 umount /media/cdrom
spode@0 245 reboot || reboot -f
spode@0 246 else
spode@0 247 umount /media/cdrom
spode@0 248 echo "==============================================================================="
spode@0 249 echo ""
spode@0 250 exit 0
spode@0 251 fi
spode@0 252 }
spode@0 253
spode@0 254 set_bootable()
spode@0 255 {
spode@0 256 # As the boot flag is toggable, need to check it before hand
spode@0 257 ISSET=`fdisk -l $DEVICE | grep $DEVICE | grep "*"`
spode@0 258
spode@0 259 # If not set, set bootable
spode@0 260 if [ -z "$ISSET" ]; then
spode@0 261 umount $DEVICE
spode@9 262 echo "Setting $DEVICE as bootable..."
spode@0 263 fdisk $DEVICE >/dev/null << EOF
spode@0 264 a
spode@0 265 1
spode@0 266 w
spode@0 267 EOF
spode@0 268 status
spode@0 269 fi
spode@0 270 }
spode@0 271
spode@0 272 #
spode@0 273 # Tazusb sequence
spode@0 274 #
spode@0 275
spode@0 276 case $COMMAND in
spode@0 277 writefs)
spode@0 278 #writefs to rootfs.gz
spode@0 279 check_root
spode@0 280 if [ -z $2 ]; then
spode@0 281 COMPRESSION=none
spode@0 282 else
spode@0 283 COMPRESSION=$2
spode@0 284 fi
spode@0 285 #start info
spode@0 286 echo ""
spode@0 287 echo -e "\033[1mWrite filesystem\033[0m
spode@0 288 ===============================================================================
spode@0 289 The command writefs will write all the current filesystem into a suitable cpio
spode@0 290 archive (rootfs.gz) usable on a bootable LiveUSB media.
spode@0 291
spode@0 292 Archive compression: $COMPRESSION"
spode@0 293 echo ""
spode@0 294
spode@0 295 #clear out tazpkg cache
spode@0 296 rm /var/cache/tazpkg/* -r -f
spode@0 297
spode@0 298 #optionally remove sound card selection
spode@0 299 echo -n "Do you wish to remove the sound card selection (Yes/no/exit) ? "
pankso@5 300 read anser
pankso@5 301 case $anser in
pankso@5 302 e|E|"exit"|Exit)
pankso@5 303 exit 0
pankso@5 304 ;;
pankso@5 305 y|Y|yes|Yes)
pankso@5 306 echo -n "Removing current sound card selection..."
pankso@5 307 rm -f /var/lib/sound-card-driver
pankso@5 308 rm -f /etc/asound.state
pankso@5 309 ;;
pankso@5 310 *)
pankso@5 311 echo -n "Keeping current sound card selection..."
pankso@5 312 ;;
pankso@5 313 esac
spode@0 314 status
spode@0 315
spode@0 316 #create list of files
spode@0 317 find /bin /etc /init /sbin /var /dev /lib /mnt /root /usr >/tmp/list
spode@0 318
spode@0 319 for dir in /home /proc /sys /tmp /media /media/cdrom /media/flash /media/usbdisk
spode@0 320 do
spode@0 321 echo $dir >>/tmp/list
spode@0 322 done
spode@0 323
spode@0 324 #gen initramfs with specified compression
spode@0 325 if [ "$COMPRESSION" = "lzma" ]; then
spode@0 326 echo -n "Creating rootfs.gz with lzma compression... "
spode@0 327 cat /tmp/list | cpio -o -H newc | lzma e -si -so > /rootfs.gz
spode@0 328
spode@0 329 elif [ "$COMPRESSION" = "gzip" ]; then
spode@0 330 echo -n "Creating rootfs.gz with gzip compression... "
spode@0 331 cat /tmp/list | cpio -o -H newc | gzip -9 > /rootfs.gz
spode@0 332
spode@0 333 else
spode@0 334 echo -n "Creating rootfs.gz without compression... "
spode@0 335 cat /tmp/list | cpio -o -H newc > /rootfs.gz
spode@0 336 fi
spode@0 337
spode@0 338 #get initramfs size
spode@0 339 size=`du -sh /rootfs.gz | cut -f 1`
spode@0 340
spode@0 341 #if the bootable medium is where it should be, copy across
spode@0 342
spode@0 343 if (test -e /home/boot/bzImage); then
spode@0 344 echo "Moving rootfs.gz to media. Remember to unmount for delayed writes!"
spode@0 345
spode@0 346 #move the old filesystem with the unix timestamp for reference
spode@0 347 if (test -e /home/boot/previous.gz); then
spode@0 348 mv /home/boot/previous.gz /home/boot/rootfs.gz.$(date +%s)
spode@0 349 fi
spode@0 350
spode@0 351 mv /home/boot/rootfs.gz /home/boot/previous.gz
spode@0 352 mv /rootfs.gz /home/boot/.
spode@0 353 else
spode@0 354 echo "rootfs.gz is located in /"
spode@0 355 fi
spode@0 356
spode@0 357 echo "==============================================================================="
spode@0 358 echo "Root filesystem size: $size"
spode@0 359 echo ""
spode@0 360 ;;
spode@0 361 format)
spode@0 362 # Format a partitions in ext3.
spode@0 363 check_root
spode@0 364 echo ""
spode@0 365 echo -e "\033[1mFormat a device\033[0m"
spode@0 366 echo "==============================================================================="
spode@0 367 DEVICE=$2
spode@0 368 if [ -z $DEVICE ]; then
hacker@12 369 ask_for_device
hacker@12 370 check_for_device
hacker@12 371 else
hacker@12 372 echo "Device : $DEVICE"
spode@0 373 fi
spode@0 374 mkfs_ext3
spode@0 375 echo "==============================================================================="
spode@9 376 echo "Device $label ($DEVICE) is ready to use as LiveUSB and/or /home partition."
spode@0 377 echo ""
spode@0 378 ;;
spode@0 379 gen-liveusb)
spode@0 380 # Generate a LiveUSB media using files from a LiveCD.
spode@0 381 check_root
spode@0 382 echo ""
spode@0 383 echo -e "\033[1mGen a LiveUSB media\033[0m"
spode@0 384 echo "==============================================================================="
spode@0 385 DEVICE=$2
spode@0 386 if [ -z $DEVICE ]; then
hacker@12 387 ask_for_device
spode@0 388 fi
spode@0 389
spode@0 390 check_for_device
pankso@11 391 mount_cdrom
spode@0 392 get_part_info
spode@0 393 unmount_target_usb
spode@0 394 install_mbr
spode@0 395 set_bootable
spode@0 396 mount_target_usb
spode@0 397 copy_cdrom_files
spode@0 398 install_boot
spode@0 399 exit_or_reboot
spode@0 400 ;;
spode@0 401 usage|*)
spode@0 402 # Display usage by default.
spode@0 403 usage
spode@0 404 ;;
spode@0 405 esac
spode@0 406
spode@0 407 exit 0