From 860d54bd7366e198f0c84bc8560df93d3b453f93 Mon Sep 17 00:00:00 2001 From: Claire Robinson Date: Wed, 27 Aug 2014 15:41:48 +0100 Subject: Initial commit to github --- dorsync | 446 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 446 insertions(+) create mode 100644 dorsync (limited to 'dorsync') diff --git a/dorsync b/dorsync new file mode 100644 index 0000000..67def1a --- /dev/null +++ b/dorsync @@ -0,0 +1,446 @@ +#!/bin/bash +############################################ +# If dcfldd is installed you will get a # +# progress indicator when dumping to USB # +############################################ +# Enter a few variables. # +# location is where you stored the ISOs # +# release is the directory on the server # +# user is the rsync login username # +# password is the rsync password # +############################################ + +location="" +release="" +user="" +password="" + +############################################ +# To exclude directories from sync, create # +# a file, dorsync.skip in the start folder # # +# containing the directory names only, one # +# per line. # +############################################ + +####### Don't touch below this line ######## +# # +############################################ +# dorsync v2.8 # +############################################ +# Claire Robinson (MrsB) # +# for Mageia QA # +############################################ +# Thanks also go to.. # +# Alfred Kretschmer (alfred) # +# Barry Jackson (barjac) # +############################################ + +red='\e[0;31m' +green='\e[0;32m' +bold='\e[01m' +endcolour='\e[0m' +rsyncargs=() +rsyncargs+=" -avHP" +if [[ -e dorsync.skip ]]; then + while read line; do + rsyncargs+=" --exclude=$line" + done < dorsync.skip +fi +rsyncargs+=" rsync://$user@bcd.mageia.org/isos/$release/ ." +cd "$location" +# Gap so I remember to add command line arguments + +# Function to do the actual rsync and run checks afterwards +dorsync () { +echo "Starting rsync.." +RSYNC_PASSWORD="$password" rsync ${rsyncargs[@]} +OUT=$? +echo "" +if [ $OUT -eq 0 ];then + echo "Completed rsync" +else + echo "Rsync failed. Exiting." + exit +fi + +echo "" +du -h | sort -k 2 | sed "s|\./||" | sed -e "s|\(\s\)\.|\1Total\n|" +echo "" +echo "Starting checks, please ensure the dates are correct" +echo "" +# Loop through folders in $location and read info into arrays +# Hopefully able to use this later to dump to USB stick +# Perform checks too and highlight green OK red Failed +i=0 +OIFS="$IFS" +IFS=$'\n' +for line in $(find . -type f -name "*.iso" -printf '%P\n') +do + directory[$i]=`dirname "$line"` + iso[$i]=`basename "$line"` + echo "PATH: ${directory[$i]}" + echo "ISO: ${iso[$i]}" + cd "${directory[$i]}" + # Check for DATE.txt and display it or warn if missing + if [ -e DATE.txt ] + then + echo "DATE: `cat DATE.txt`" + missingdate[$i]=false + else + echo -e "DATE: ${red}Missing${endcolour}" + missingdate[$i]=true + fi + # Check for md5 file and then md5sum it or warn if missing + if [ -e "${iso[$i]}.md5" ] + then + md5sum -c --quiet --status *.md5 2>/dev/null + OUT=$? + if [ $OUT -eq 0 ];then + echo -e "MD5: ${green}OK${endcolour} \c" + md5ok[$i]=true + missingmd5[$i]=false + else + echo -e "MD5: ${red}Failed${endcolour} \c" + md5ok[$i]=false + missingmd5[$i]=false + fi + else + echo -e "MD5: ${red}Missing${endcolour} \c" + missingmd5[$i]=true + fi + + # Check for sha1 file and then sha1sum it or warn if missing + if [ -e ${iso[$i]}.sha1 ] + then + sha1sum -c --quiet --status *.sha1 2>/dev/null + OUT=$? + if [ $OUT -eq 0 ];then + echo -e "SHA1: ${green}OK${endcolour}" + sha1ok[$i]=true + missingsha1[$i]=false + else + echo -e "SHA1: ${red}Failed${endcolour}" + sha1ok[$i]=false + missingsha1[$i]=false + fi + else + echo -e "SHA1: ${red}Missing${endcolour}" + sha1ok[$i]=false + missingsha1[$i]=true + fi + echo "" + cd "$location" + (( i++ )) +done +IFS="$OIFS" +# Check if anything failed and provide a summary +total=${#directory[@]} +warnings=0 +startwarning="--------------------------------------------------\n${bold}Warning:${endcolour}\n" +for (( i=0; i<=$(( $total -1 )); i++ )) +do + # Warn of missing DATE.txt files + if [ ${missingdate[$i]} = true ] + then + if [ $warnings = 0 ] + then + echo -e $startwarning + fi + echo "${directory[$i]}/DATE.txt missing" + (( warnings++ )) + fi + # Warn of missing md5 files + if [ ${missingmd5[$i]} = true ] + then + if [ $warnings = 0 ] + then + echo -e $startwarning + fi + echo "${directory[$i]}/${iso[$i]}.md5 missing" + (( warnings++ )) + else + # Warn of bad md5sum if present + if [ ${md5ok[$i]} = false ] + then + if [ $warnings = 0 ] + then + echo -e $startwarning + fi + echo "${iso[$i]} md5sum failed" + (( warnings++ )) + fi + fi + # Warn of missing sha1 files + if [ ${missingsha1[$i]} = true ] + then + if [ $warnings = 0 ] + then + echo -e $startwarning + fi + echo "${directory[$i]}/${iso[$i]}.sha1 missing" + (( warnings++ )) + else + # Warn of bad sha1sum if present + if [ ${sha1ok[$i]} = false ] + then + if [ $warnings = 0 ] + then + echo -e $startwarning + fi + echo "${iso[$i]} sha1sum failed" + (( warnings++ )) + fi + fi +done +echo "" +# If anything failed, provide a count +if [ $warnings != 0 ] +then + if [ $warnings -gt 1 ] + then + message="There are $warnings potential problems." + else + message="There is 1 potential problem." + fi + echo -e "${bold}$message${endcolour}" + echo "--------------------------------------------------" +else + echo "------------------------------------" + echo "" + echo -e "${bold}All tests passed OK. Yippee!${endcolour}" + echo "Please check the dates are correct." + echo "" + echo "------------------------------------" +fi + +} + + +# Function to list USB sticks +liststicks () { + usbs="" + for device in /sys/block/*/device + do + if echo $(readlink -f $device)|egrep -q "usb" + then + adisk=`echo $device | cut -f4 -d/` + usbs="$usbs /dev/$adisk" + fi + done + + # Find info for each from udisks + i=0 + for usb in $usbs + do + showinfo=`udisks --show-info $usb` + isremovable=`echo "$showinfo" | grep " removable:"| cut -f2 -d: | sed 's/^[ ]*//'` + + # Check udisks shows it as removable + if [ $isremovable = 1 ] + then + usbdev[$i]="$usb" + found[$i]=`echo "$showinfo" | grep " detected at:"| cut -f2- -d: | sed 's/^[ ]*//'` + vendor[$i]=`echo "$showinfo" | grep "vendor:" | cut -f2 -d: | sed 's/^[ ]*//'` + model[$i]=`echo "$showinfo" | grep "model:" | cut -f2 -d: | sed 's/^[ ]*//'` + sizebytes[$i]=`echo "$showinfo" | grep " size:" | cut -f2 -d: | sed 's/^[ ]*//'` + sizegb[$i]=`echo "scale=1;${sizebytes[$i]}/1024/1024/1024" | bc` + (( i++ )) + fi + done +} + + +# Minimal function to find isos if rsync is skipped +findisos () { + i=0 + OIFS="$IFS" + IFS=$'\n' + for line in $(find . -type f -name "*.iso" -printf '%P\n') + do + directory[$i]=`dirname "$line"` + iso[$i]=`basename "$line"` + (( i++ )) + done + IFS="$OIFS" +} + + +# Function to wait for usb stick to be inserted +dodoh () { + while true + do + echo "No USB drives found. Please insert one now." + read -n1 -e -p "Press Q to quit or any other key to continue: " num + case $num in + [Qq]* ) exit;; + * ) liststicks + echo "" + if ! [ ${#found[@]} = 0 ] + then + doh="Yay" + break + fi + ;; + esac + done +} + + +# Function to actually do the dump onto usb +dothedeed () { + echo "" + echo "Please be patient this can take a while" + echo "" + echo "This operation requires root privileges" + echo "Please enter your root password.." + echo "" + export rootwhere="${usbdev[$(($usbchoice-1))]}" + export rootiso="$fullisopath" + if [ -e /usr/bin/dcfldd ] + then + su -c 'dcfldd sizeprobe=if statusinterval=5 if="$rootiso" of="$rootwhere" bs=1M && sync' + else + su -c 'dd if="$rootiso" of="$rootwhere" bs=1M && sync' + fi + OUT=$? + echo "" + if [ $OUT -eq 0 ];then + echo "Completed OK." + else + echo "Failed. Exiting." + exit + fi +} + +# Function to choose usb stick & iso and check it'll fit +dodump () { + echo "" + didrsync=$1 + if [ $didrsync == 1 ] + then + # Ask to dump to USB or exit + echo "" + while true + do + read -n1 -e -p "Do you want to dump one onto USB? [y/n]" yn + case $yn in + [Yy] ) liststicks + break;; + [Nn] ) exit;; + * ) echo "Please choose y or n";; + esac + done + echo "" + else + liststicks + fi + + # If no USBs found D'oh! loop until there is one.. + total=${#found[@]} + if [ $total = 0 ] + then + doh="doh" + while [ "$doh" = "doh" ] + do + dodoh + done + total=${#found[@]} + fi + + # Now there is a USB print the info + for (( i=0; i<=$(( $total -1 )); i++ )) + do + if [ $i = 0 ] + then + echo "" + echo "USB drives found:" + echo "" + echo -e " Device\tSize\tWhen Found\t\t\tModel" + fi + echo -e "$((i+1))) ${usbdev[$i]}\t${sizegb[$i]}Gb\t${found[$i]}\t${vendor[$i]} ${model[$i]}" + done + + # Choose which USB to use +echo "" + while true + do + read -e -p "Please choose which USB to use or Q to quit and press : " usbchoice + if [[ $usbchoice = [Qq] ]]; then + exit + elif [[ "$usbchoice" =~ ^[0-9]+$ ]] && (( $usbchoice <= $total )) && (( $usbchoice > 0 )); then + break + else echo "Invalid selection" + fi + done + + # List the ISOs + total=${#iso[@]} + for (( i=0; i<=$(( $total -1 )); i++ )) + do + if [ $i = 0 ] + then + echo "" + echo "List of ISOs:" + fi + echo "$(($i+1))) ${iso[$i]}" + done + echo "" + + # Choose which ISO to use + while true + do + read -e -p "Please choose which ISO to use or Q to quit and press : " isochoice + if [[ $isochoice = [Qq] ]]; then + exit + elif [[ "$isochoice" =~ ^[0-9]+$ ]] && (( $isochoice <= $total )) && (( $isochoice > 0 )); then + break + else echo "Invalid selection" + fi + done + + # Will it fit? + fullisopath=`readlink -f "${directory[$(($isochoice-1))]}/${iso[$(($isochoice-1))]}"` + isosize=`du -b "$fullisopath" | cut -f1` + usbsize=${sizebytes[$(($usbchoice-1))]} + if [ $usbsize -lt $isosize ] + then + echo "" + echo "Sorry. It won't fit! Exiting." + exit + fi + + # Warn + usbinfo="${usbdev[$(($usbchoice-1))]} (${sizegb[$(($usbchoice-1))]}Gb ${vendor[$(($usbchoice-1))]} ${model[$(($usbchoice-1))]} found at ${found[$(($usbchoice-1))]})" + echo "" + echo -e "${bold}About to dump $fullisopath${endcolour}" + echo -e "${bold}onto $usbinfo${endcolour}" + echo "" + echo "This will destroy any data already on the USB device." + + # Confirm + while true + do + read -n1 -e -p "Press Y to confirm or Q to quit: " yq + case $yq in + [Qq] ) exit;; + [Yy] ) dothedeed + break;; + * ) echo -e "Invalid selection\n";; + esac + done +} + + +# Offer to skip rsync and just dump to usb +echo "" +while true +do + read -n1 -e -p "Do you want to Rsync (R), skip Rsync and just dump to usb (D) or quit (Q): " rdq +case $rdq in + [Rr] ) dorsync; dodump 1; exit;; + [Dd] ) findisos; dodump 0; exit;; + [Qq] ) exit;; + * ) echo " $rdq Invalid selection"; echo "";; + esac +done + -- cgit v1.2.1