tazusb annotate tazusb @ rev 0

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