#!/bin/sh
#Barry Kauler www.puppylinux.com
#LGPL 2007 Puppy Linux www.puppylinux.com
#17 june 2007
#21 June 2007 BK:
#some usb adaptors have slots for cards, plugging/unplugging these cards
#does not cause a hotplug event and the kernel does not update /proc.
#'disktype' does a direct hardware probe which forces /proc update.
#attempt here to run disktype when appropriate.
#v3.93 10 dec 2007 BK: updated for 2.6.24 kernel, no /dev/hd*.
#v3.97 31jan2008 BK: refinement for detecting kernel with /dev/hd support.
#v3.97 25feb2008 BK: guess_fstype does not work when testing 'makebootfat', use fdisk.
#v4.01 19may2008 BK: bugfix for 2.6.25.4, ram entries in /proc/partitions.
#v403 fix for SIZE of drives without partitions.
#v406 support for old kernel, /proc/ide, /dev/hd*
#v407 improved filesystem probe for optical discs.
#110126 remove code for SATADRIVES (not used in PUPSTATE anymore).
#120601 jemimah: fallback to use 'blkid' to find f.s. type. 120601 revert use of blkid, too slow.
#120602 kernel 3.2.18 major deviance from earlier kernels, /proc/partitions (and /sys/block) now shows /dev/sr0 when no disk.
#       new situation, getting duplicate /dev/sr0 so need 'sort -u'.
#120516 raspi, guess_fstype fails for ext4 and swap f.s., did detect fat. fallback blkid.
#120516 sort in old coreutils (as in wary/racy) does not have -V option
#130127 early kernels do not have sr0/sr1 in /proc/partitions, 3.2+ do, which can upset things. (tested 2.6.32.29 and 3.2.29+, so not sure exact version this difference came in)
#130128 improve detection of optical: distinguish iso9600, udf. no o/p if no media inserted.
#       previously, any optical media returned "iso9660", now have "udf" (also for video dvds), "audiocd" (audio cds), "none" (unknown f.s.).
#130526 Karl Godt: fix blkid parsing.

. /etc/rc.d/PUPSTATE

SUNITS="$1" #allowed params are '-k' or '-m'.
OUTPUT=""

if [ -f /root/.usb-drive-log-probepart ];then #force /proc upate mechanism
 for ONEUSBDRV in `cat /root/.usb-drive-log-probepart | tr '\n' ' '`
 do
  #disktype /dev/$ONEUSBDRV > /dev/null 2>&1
  dd if=/dev/$ONEUSBDRV of=/dev/null bs=512 count=1 >/dev/null 2>&1 #v3.97 faster.
 done
fi

#devices that have partitions... 
#([^k] is to eliminate mmcblk0 device -- allow mmcblk0p1 etc) v4.01 bugfix eliminate ram...
#130127 [^kr] screens out sr0-sr9. early kernels do not have these in /proc/partitions, 3.2+ do, which causes desktop icon to not appear when audio-cd inserted.
PARTITIONS="`grep '^ .*[^kr][0-9]$' /proc/partitions | tr -s ' ' | cut -f 4-5 -d ' ' | grep -vE ' loop| ram'`" #each line ex: 16076800 sda5
PARTNAMES="`echo "$PARTITIONS" | cut -f 2 -d ' '`" #120602

#all disk devices... 
#note, /proc/diskstats only updated when a disk accessed, so devs may be missing...
#NO, NO, NO, /sys is very flakey for hd devices...
ALLDRVS="`ls -1 /sys/block | grep -E '^scd|^sd|^mmc|^sr'`"
[ -e /proc/ide ] && ALLDRVS="${ALLDRVS}
`ls -1 /proc/ide | grep '^hd'`"

#all drives and partitions... 120602 sort...
#120516 sort in old coreutils (as in wary/racy) does not have -V option...
COREUTILSVER="$(sort --version | head -n 1 | rev | cut -f 1 -d ' ' | rev)"
if vercmp $COREUTILSVER le 6.12;then
 ALLDEVS="`echo "${PARTNAMES}
${ALLDRVS}" | tr '\n' ' '`"
else
 ALLDEVS="`echo "${PARTNAMES}
${ALLDRVS}" | sort -V -u | tr '\n' ' '`"
fi

# guess_fstype really sucks for extended partitions - use fdisk but not in the main loop - too slow #140714
EXTD=''
printf "$ALLDRVS"|while read DISK
do
  FDISK="`fdisk -l /dev/"$DISK"`"
  EXTD="`echo "$FDISK"|grep -w 'Extended'|grep "$DISK"|awk '{print $1}'`" 
  EXTD=${EXTD##*/}
  [ "$EXTD" ] && echo -n "$EXTD " >> /tmp/extended_parts
done
[ -f /tmp/extended_parts ] && EXTD="`cat /tmp/extended_parts`"
rm /tmp/extended_parts 2>/dev/null

for ONEDEV in $ALLDEVS
do
 MEDIATYPE="" #130128
 FSTYPE="unknown"
 DPATTERN=' '${ONEDEV}'$'
 SIZE=`echo "$PARTITIONS" | grep "$DPATTERN" | cut -f 1 -d ' '`
 DEVICE="`echo "$PARTITIONS" | grep "$DPATTERN" | cut -f 2 -d ' '`"
 if [ ! $SIZE ];then
  [ "`echo "$PARTITIONS" | grep "$ONEDEV"`" != "" ] && continue
  #must be a device without partitions...
  SIZE=0
  DEVICE="$ONEDEV"
  case $DEVICE in
   hd*)
    [ "`cat /proc/ide/$DEVICE/media`" = "cdrom" ] && MEDIATYPE="optical" #130128
   ;;
   scd*|sr*) #usb,sata,scsi cd/dvd drive.
    MEDIATYPE="optical" #130128
   ;;
  esac
  if [ -e /sys/block/${DEVICE}/size ];then
   SIZE=`cat /sys/block/${DEVICE}/size`
   SIZE=$(($SIZE/2)) #get KB.
  fi
  #for hd* or sd* superfloppy, determine size (avoid probing optical drive)...
  if [ "$MEDIATYPE" = "" -a $SIZE -eq 0 ];then #130128
   BSIZE=`disktype /dev/$DEVICE 2>/dev/null | grep '^Block device' | cut -f 2 -d '(' | cut -f 1 -d ' '`
   [ $BSIZE ] && SIZE=$(($BSIZE/1024)) #KB
  fi
  if [ "$MEDIATYPE" = "optical" ];then #130128
   opticalPATTERN='^/dev/'"$DEVICE"
   if [ "`grep "$opticalPATTERN" /proc/mounts`" = "" ];then
    cddetect_quick -d/dev/${DEVICE} >/dev/null 2>&1 #very fast.
    #...returns 0 if disc inserted, else 255.
    [ $? -ne 0 ] && continue #130128 go no further, as no media inserted.
   fi
  fi
 fi
 FSTYPE="`guess_fstype /dev/$DEVICE 2>/dev/null`" #130128 note, audio-cd returns "unknown", as no f.s.
 echo "$EXTD"|grep -q "$DEVICE" && FSTYPE='unknown' #140714 'Extended' partition, gets changed to 'none'

 if [ "$FSTYPE" = "unknown" -a "$MEDIATYPE" = "optical" ];then #130128 probe optical a bit more.
  cddetect -q -d/dev/${DEVICE} > /dev/null 2>&1
  case $? in
   1) FSTYPE="audiocd" ;;
  esac
 fi
 
 #v3.97 guess_fstype fails sometimes...
 #(using makebootfat to setup a USB-FLOPPY/-HDD/-ZIP combined bootable FAT drive).
 xFSTYPE=''
 if [ "$FSTYPE" = "unknown" -a "$MEDIATYPE" = "" ];then #130128 not optical.
  fsPATTERN='^/dev/'"$DEVICE"' '
  xDEVICE="`echo -n "$DEVICE" | sed -e 's/[0-9]*$//' -e 's%p$%%'`" #"${DEVICE/[0-9]/}" #remove partition number. 120516 remove 'p' from mmcblk0p1
  xFSTYPE="`fdisk -l /dev/$xDEVICE 2>/dev/null | grep "$fsPATTERN" | head -n 1 | grep -o -E 'FAT|swap|Linux|NTFS'`" #120516 FAT12$|FAT16$|FAT32$
  case $xFSTYPE in #120516
   FAT) FSTYPE='vfat' ;;
   swap) FSTYPE='swap' ;;
   NTFS) FSTYPE='ntfs' ;;
   Linux)
    FSTYPE=$(blkid -c /dev/null /dev/${DEVICE} | grep -w "${DEVICE}" | grep -o ' TYPE=".*"' | cut -f2 -d'"')
    [ "$FSTYPE" = "" ] && FSTYPE="unknown"
   ;;
  esac
 fi
 
 [ "$FSTYPE" = "unknown" ] && FSTYPE="none"

 [ "$SUNITS" = "" ] && SIZE=$(($SIZE*2)) #want 512 byte blocks.
 [ "$SUNITS" = '-m' ] && SIZE=$(($SIZE/1024)) #want MB

 echo "/dev/$DEVICE|$FSTYPE|$SIZE"
 
 #keep record of usb sd*, for forced updating of /proc...
 case $DEVICE in
  sd*)
   #log if usb drive (not a ata drive)...
   DEVDRV="`echo -n "$DEVICE" | tr -d '[0-9]'`"
   if [ "`echo "$ATADRIVES" | grep "$DEVDRV"`" = "" ];then #110126
    echo "$DEVDRV" >> /root/.usb-drive-log-probepart
    sort -u /root/.usb-drive-log-probepart > /tmp/usb-drive-log-probepart-tmp
    mv -f /tmp/usb-drive-log-probepart-tmp /root/.usb-drive-log-probepart
   fi
   ;;
 esac
 
done

###END###
