1 #! /usr/bin/sh 2 # 3 # CDDL HEADER START 4 # 5 # The contents of this file are subject to the terms of the 6 # Common Development and Distribution License (the "License"). 7 # You may not use this file except in compliance with the License. 8 # 9 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 # or http://www.opensolaris.org/os/licensing. 11 # See the License for the specific language governing permissions 12 # and limitations under the License. 13 # 14 # When distributing Covered Code, include this CDDL HEADER in each 15 # file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 # If applicable, add the following below this CDDL HEADER, with the 17 # fields enclosed by brackets "[]" replaced with your own identifying 18 # information: Portions Copyright [yyyy] [name of copyright owner] 19 # 20 # CDDL HEADER END 21 # 22 # Copyright 2010 Sun Microsystems, Inc. All rights reserved. 23 # Use is subject to license terms. 24 # 25 # 26 # This is a clean script for removable disks 27 # 28 # Following is the syntax for calling the script: 29 # scriptname [-s|-f|-i|-I] devicename [-A|-D] username zonename zonepath 30 # 31 # -s for standard cleanup by a user 32 # -f for forced cleanup by an administrator 33 # -i for boot-time initialization (when the system is booted with -r) 34 # -I to suppress error/warning messages; the script is run in the '-i' 35 # mode 36 # 37 # $1: devicename - device to be allocated/deallocated, e.g., sr0 38 # 39 # $2: -A if cleanup is for allocation, or -D if cleanup is for deallocation. 40 # 41 # $3: username - run the script as this user, rather than as the caller. 42 # 43 # $4: zonename - zone in which device to be allocated/deallocated 44 # 45 # $5: zonepath - root path of zonename 46 # 47 # A clean script for a removable media device should prompt the user to 48 # insert correctly labeled media at allocation time, and ensure that the 49 # media is ejected at deallocation time. 50 # 51 # Unless the clean script is being called for boot-time 52 # initialization, it may communicate with the user via stdin and 53 # stdout. To communicate with the user via CDE dialogs, create a 54 # script or link with the same name, but with ".windowing" appended. 55 # For example, if the clean script specified in device_allocate is 56 # /etc/security/xyz_clean, that script must use stdin/stdout. If a 57 # script named /etc/security/xyz_clean.windowing exists, it must use 58 # dialogs. To present dialogs to the user, the dtksh script 59 # /etc/security/lib/wdwmsg may be used. 60 # 61 # This particular script, disk_clean, will work using stdin/stdout, or 62 # using dialogs. A symbolic link disk_clean.windowing points to 63 # disk_clean. 64 # 65 66 # #################################################### 67 # ################ Local Functions ################# 68 # #################################################### 69 70 # 71 # Set up for windowing and non-windowing messages 72 # 73 msg_init() 74 { 75 if [ `basename $0` != `basename $0 .windowing` ]; then 76 WINDOWING="yes" 77 case $VOLUME_MEDIATYPE in 78 cdrom) TITLE="CD-ROM";; 79 rmdisk) TITLE="Removable Disk";; 80 floppy) TITLE="Floppy";; 81 *) TITLE="Disk";; 82 esac 83 84 if [ "$MODE" = "allocate" ]; then 85 TITLE="$TITLE Allocation" 86 else 87 TITLE="$TITLE Deallocation" 88 fi 89 else 90 WINDOWING="no" 91 fi 92 } 93 94 # 95 # Display a message for the user. For windowing, user must press OK button 96 # to continue. For non-windowing, no response is required. 97 # 98 msg() { 99 if [ "$WINDOWING" = "yes" ]; then 100 $WDWMSG "$*" "$TITLE" OK 101 elif [ "$silent" != "y" ]; then 102 echo "$*" > /dev/${MSGDEV} 103 fi 104 } 105 106 ok_msg() { 107 if [ "$WINDOWING" = "yes" ]; then 108 $WDWMSG "$*" "$TITLE" READY 109 else 110 form=`gettext "Media in %s is ready. Please store safely."` 111 printf "${form}\n" $PROG $DEVICE > /dev/{MSGDEV} 112 fi 113 } 114 115 error_msg() { 116 if [ "$WINDOWING" = "yes" ]; then 117 $WDWMSG "$*" "$TITLE" ERROR 118 else 119 form=`gettext "%s: Error cleaning up device %s."` 120 printf "${form}\n" $PROG $DEVICE > /dev/${MSGDEV} 121 fi 122 } 123 124 # 125 # Ask the user an OK/Cancel question. Return 0 for OK, 1 for Cancel. 126 # 127 okcancel() { 128 if [ "$WINDOWING" = "yes" ]; then 129 $WDWMSG "$*" "$TITLE" OK Cancel 130 elif [ "$silent" != "y" ]; then 131 get_reply "$* (y to continue, n to cancel) \c" y n 132 fi 133 } 134 135 # 136 # Ask the user an Yes/No question. Return 0 for Yes, 1 for No 137 # 138 yesno() { 139 if [ "$WINDOWING" = "yes" ]; then 140 $WDWMSG "$*" "$TITLE" Yes No 141 elif [ "$silent" != "y" ]; then 142 get_reply "$* (y/n) \c" y n 143 fi 144 } 145 146 # 147 # Display an error message, put the device in the error state, and exit. 148 # 149 error_exit() { 150 if [ "$silent" != "y" ]; then 151 msg "$2" "$3" \ 152 "\n\nDevice has been placed in allocation error state." \ 153 "\nPlease inform system administrator." 154 fi 155 exit 1 156 } 157 158 # 159 # get_reply prompt choice ... 160 # 161 get_reply() { 162 prompt=$1; shift 163 while true 164 do 165 echo $prompt > /dev/tty 166 read reply 167 i=0 168 for choice in $* 169 do 170 if [ "$choice" = "$reply" ] 171 then 172 return $i 173 else 174 i=`expr $i + 1` 175 fi 176 done 177 done 178 } 179 180 # 181 # Find the first disk slice containing a file system 182 # 183 find_fs() 184 { 185 # The list of files in device_maps(4) is in an unspecified order. 186 # To speed up the fstyp(1M) scanning below in most cases, perform 187 # the search for filesystems as follows: 188 # 1) Select only block device files of the form "/dev/dsk/*". 189 # 2) Sort the list of files in an order more likely to yield 190 # matches: first the fdisk(1M) partitions ("/dev/dsk/cNtNdNpN") 191 # then the format(1M) slices ("/dev/dsk/cNtNdNsN"), in ascending 192 # numeric order within each group. 193 DEVall="`echo $FILES | \ 194 /usr/bin/tr ' ' '\n' | \ 195 /usr/bin/sed '/^\/dev\/dsk\//!d; s/\([sp]\)\([0-9]*\)$/ \1 \2/;' | \ 196 /usr/bin/sort -t ' ' -k 2,2d -k 3,3n | \ 197 /usr/bin/tr -d ' '`" 198 for DEVn in $DEVall ; do 199 fstyp_output="`/usr/sbin/fstyp -a $DEVn 2>&1`" 200 if [ $? = 0 ]; then 201 FSPATH=$DEVn 202 gen_volume_label="`echo "$fstyp_output" | \ 203 sed -n '/^gen_volume_label: .\(.*\).$/s//\1/p'`" 204 if [ "$gen_volume_label" != "" ]; then 205 FSNAME="`echo $gen_volume_label | \ 206 /usr/xpg4/bin/tr '[:upper:] ' '[:lower:]_'`" 207 fi 208 # For consistency, hsfs filesystems detected at 209 # /dev/dsk/*p0 are mounted as /dev/dsk/*s2 210 FSTYPE=`echo "$fstyp_output" | /usr/bin/head -1` 211 if [ "$FSTYPE" = hsfs -a \ 212 `/usr/bin/expr $FSPATH : '.*p0'` -gt 0 ]; then 213 FSPATH=`echo $FSPATH | /usr/bin/sed 's/p0$/s2/'` 214 fi 215 return 216 fi 217 done 218 } 219 220 # 221 # Find all mountpoints in use for a set of device special files. 222 # Usage: findmounts devpath ... 223 # 224 225 findmounts() { 226 /usr/xpg4/bin/awk -f - -v vold_root="$VOLD_ROOT" -v devs="$*" /etc/mnttab <<\ 227 "ENDOFAWKPGM" 228 BEGIN { 229 split(devs, devlist, " "); 230 for (devN in devlist) { 231 dev = devlist[devN]; 232 realdevlist[dev] = 1; 233 sub(/.*\//, "", dev); 234 sub(/s[0-9]$/, "", dev); 235 if (vold_root != "") { 236 vold_dir[vold_root "/dev/dsk/" dev] = 1; 237 vold_dir[vold_root "/dev/rdsk/" dev] = 1; 238 } 239 } 240 } 241 242 { 243 for (dev in realdevlist) { 244 if ($1 == dev) { 245 mountpoint = $2; 246 print mountpoint; 247 } 248 } 249 for (dev in vold_dir) { 250 if (substr($1, 1, length(dev)) == dev) { 251 mountpoint = $2; 252 print mountpoint; 253 } 254 } 255 } 256 ENDOFAWKPGM 257 } 258 259 # 260 # Allocate a device. 261 # Ask the user to make sure the disk is properly labeled. 262 # Ask if the disk should be mounted. 263 # 264 do_allocate() 265 { 266 if [ $VOLUME_MEDIATYPE = floppy ]; then 267 # Determine if media is in drive 268 eject_msg="`eject -q $DEVFILE 2>&1`" 269 eject_status="$?" 270 case $eject_status in 271 1) # Media is not in drive 272 okcancel "Insert disk in $DEVICE." 273 if [ $? != 0 ]; then 274 exit 0 275 fi;; 276 3) # Error 277 error_exit $DEVICE \ 278 "Error checking for media in drive.";; 279 esac 280 else 281 okcancel "Insert disk in $DEVICE." 282 if [ $? != 0 ]; then 283 exit 0 284 fi 285 fi 286 287 yesno "Do you want $DEVICE mounted?" 288 if [ $? != 0 ]; then 289 exit 0 290 fi 291 292 if [ $VOLUME_MEDIATYPE = cdrom -o $VOLUME_MEDIATYPE = rmdisk ]; then 293 # Get the device path and volume name of a partition 294 find_fs 295 if [ "$FSPATH" != "" ]; then 296 VOLUME_PATH=$FSPATH 297 fi 298 if [ "$FSNAME" != "" ]; then 299 VOLUME_NAME=$FSNAME 300 fi 301 fi 302 VOLUME_ACTION=insert 303 304 # Give ourself write permission on device file so file system gets 305 # mounted read/write if possible. 306 # rmmount only cares about permissions not user... 307 chown $VOLUME_USER $VOLUME_PATH 308 chmod 700 $VOLUME_PATH 309 310 # Do the actual mount. VOLUME_* environment variables are inputs to 311 # rmmount. 312 rmmount_msg="`/usr/sbin/rmmount 2>&1`" 313 rmmount_status="$?" 314 if [ $rmmount_status -eq 0 ]; then 315 EXIT_STATUS=$CLEAN_MOUNT 316 elif [ $rmmount_status -gt 0 -a $VOLUME_MEDIATYPE != cdrom ]; then 317 # Try again in readonly mode. cdrom is always mounted ro, so 318 # no need to try again. 319 echo "Read-write mount of $DEVICE failed. Mounting read-only." 320 VOLUME_ACTION=remount; export VOLUME_ACTION 321 VOLUME_MOUNT_MODE=ro; export VOLUME_MOUNT_MODE 322 `/usr/sbin/rmmount` 323 if [ $? -eq 0 ]; then 324 EXIT_STATUS=$CLEAN_MOUNT 325 fi 326 fi 327 328 # Set permissions on directory used by vold, sdtvolcheck, etc. 329 if [ -d /tmp/.removable ]; then 330 chown root /tmp/.removable 331 chmod 777 /tmp/.removable 332 fi 333 } 334 335 336 do_deallocate() 337 { 338 if [ $VOLUME_MEDIATYPE = cdrom -o $VOLUME_MEDIATYPE = rmdisk ]; then 339 if [ -h /$VOLUME_MEDIATYPE/$DEVICE ]; then 340 # Get the device path and volume name of a partition 341 VOLUME_PATH=`ls -l /$VOLUME_MEDIATYPE/$DEVICE|\ 342 cut -d '>' -f2` 343 VOLUME_DEVICE=`mount -p|grep $VOLUME_PATH|\ 344 cut -d ' ' -f1` 345 fi 346 fi 347 348 if [ -d "$VOLUME_PATH" ]; then 349 VOLUME_ACTION=eject 350 # Do the actual unmount. 351 # VOLUME_* environment variables are inputs to rmmount. 352 rmmount_msg="`/usr/sbin/rmmount 2>&1`" 353 rmmount_status="$?" 354 355 # Remove symbolic links to mount point 356 for name in /$VOLUME_MEDIATYPE/*; do 357 if [ -h $name ]; then 358 target=`ls -l $name | awk '{ print $NF; }'` 359 target_dir=`dirname $target` 360 target_device=`echo $target_dir | \ 361 sed -e 's/^.*-\(.*\)$/\1/'` 362 if [ "$target_device" = "$DEVICE" ]; then 363 rm -f $name 364 fi 365 fi 366 done 367 else 368 rmmount_status=0 369 fi 370 371 case $rmmount_status in 372 1) # still mounted 373 error_exit $DEVICE "Error unmounting $DEVICE" "$rmmount_msg";; 374 0) # not mounted 375 # Eject the media 376 if [ "$FLAG" = "f" ] ; then 377 eject_msg="`eject -f $DEVICE 2>&1`" 378 else 379 eject_msg="`eject $DEVICE 2>&1`" 380 fi 381 eject_status="$?" 382 case $eject_status in 383 0|1|4) # Media has been ejected 384 case $VOLUME_MEDIATYPE in 385 floppy|cdrom|rmdisk) 386 msg "Please remove the disk from $DEVICE.";; 387 esac;; 388 3) # Media didn't eject 389 msg $DEVICE "Error ejecting disk from $DEVICE" \ 390 "$eject_msg";; 391 esac 392 esac 393 } 394 395 # 396 # Reclaim a device 397 # 398 do_init() 399 { 400 eject_msg="`eject -f $DEVICE 2>&1`" 401 eject_status="$?" 402 403 case $eject_status in 404 0) # Media has been ejected 405 if [ "$silent" != "y" ]; then 406 ok_msg 407 fi 408 exit 0;; 409 1) # Media not ejected 410 if [ "$silent" != "y" ]; then 411 error_msg 412 fi 413 exit 0;; 414 3) # Error 415 if [ "$silent" != "y" ]; then 416 error_msg 417 fi 418 msg $DEVICE "Error ejecting disk from $DEVICE" \ 419 "$eject_msg" 420 exit 2;; 421 esac 422 } 423 424 425 # #################################################### 426 # ################ Begin main program ################ 427 # #################################################### 428 429 trap "" INT TERM QUIT TSTP ABRT 430 431 PATH="/usr/bin:/usr/sbin" 432 MODE="allocate" 433 SILENT=n 434 WDWMSG="/etc/security/lib/wdwmsg" 435 VOLUME_ZONE_PATH="/" 436 USAGE="Usage: disk_clean [-s|-f|-i|-I] devicename -[A|D] [username] [zonename] [zonepath]" 437 EXIT_STATUS=0 438 CLEAN_MOUNT=4 439 MACH=`uname -p` 440 FLAG=i 441 # 442 # Parse the command line arguments 443 # 444 while getopts ifsI c 445 do 446 case $c in 447 i) 448 FLAG=$c;; 449 f) 450 FLAG=$c;; 451 s) 452 FLAG=$c;; 453 I) 454 FLAG=i 455 silent=y;; 456 \?) 457 echo $USAGE 458 exit 1;; 459 esac 460 done 461 462 shift `expr $OPTIND - 1` 463 464 DEVICE=$1 465 MODE="deallocate" 466 if [ "$2" = "-A" ]; then 467 MODE="allocate" 468 elif [ "$2" = "-D" ]; then 469 MODE="deallocate" 470 fi 471 472 #get the device_maps information 473 MAP=`/usr/sbin/list_devices -s -l $DEVICE` 474 FILES=`echo $MAP | cut -f4 -d:` # e.g., /dev/dsk/c0t6d0s0 /dev/dsk/c0t6d0s1 ... 475 DEVFILE=`echo $FILES | cut -f1 -d" "` # e.g., "/dev/dsk/c0t6d0s0" 476 477 # Set VOLUME_ variables that are inputs to rmmount 478 479 VOLUME_DEVICE=`echo $FILES | cut -f2 -d" "` # e.g., "/dev/dsk/c0t6d0s1" 480 MEDIATYPE=`echo $MAP | cut -f3 -d: | cut -f2 -d" "` 481 # e.g., "cdrom" or "floppy" 482 if [ "$MEDIATYPE" = "sr" ]; then 483 VOLUME_MEDIATYPE="cdrom" 484 elif [ "$MEDIATYPE" = "fd" ]; then 485 VOLUME_MEDIATYPE="floppy" 486 elif [ "$MEDIATYPE" = "rmdisk" ]; then 487 VOLUME_MEDIATYPE="rmdisk" 488 fi 489 490 VOLUME_PATH=$DEVFILE # e.g., "/dev/dsk/c0t6d0s0" 491 if [ "$MACH" = "i386" ] && [ "$MEDIATYPE" = "rmdisk" ]; then 492 VOLUME_PATH=`echo $DEVFILE | sed -e 's/s0/p0/'` 493 fi 494 495 SYMDEV=`echo $DEVICE | sed -e 's/_//'` # e.g., "cdrom" or "floppy" 496 SYMNUM=`echo $SYMDEV | sed -e 's/[a-z]*//g'` 497 SYMDEV=`echo $SYMDEV | sed -e 's/[0-9]*//g'` 498 if [ "$SYMDEV" = "sr" ]; then 499 VOLUME_SYMDEV="cdrom"$SYMNUM 500 elif [ "$SYMDEV" = "fd" ]; then 501 VOLUME_SYMDEV="floppy"$SYMNUM 502 elif [ "$SYMDEV" = "rmdisk" ]; then 503 VOLUME_SYMDEV="rmdisk"$SYMNUM 504 else 505 VOLUME_SYMDEV=$SYMDEV$SYMNUM 506 fi 507 508 VOLUME_ZONE_NAME=$4 509 510 VOLUME_ZONE_PATH=$5 511 512 if [ "$MODE" = "allocate" ]; then 513 if [ -n "$3" ]; then # e.g., "joeuser" 514 VOLUME_USER=$3 515 else 516 VOLUME_USER=`/usr/xpg4/bin/id -u -nr` 517 fi 518 else 519 # If there's a directory for the device under /<mediatype>, get the 520 # user name from there, to use in cleaning up that directory. Otherwise, 521 # the user name isn't actually used in deallocation. 522 if [ -d ${VOLUME_ZONE_PATH}/${VOLUME_MEDIATYPE}/*-${DEVICE} ]; then 523 VOLUME_USER=`ls -ld ${VOLUME_ZONE_PATH}/${VOLUME_MEDIATYPE}/*-${DEVICE} | awk '/^d/{print $3}'` 524 else 525 if [ -n "$3" ]; then 526 VOLUME_USER=$3 527 else 528 VOLUME_USER=`/usr/xpg4/bin/id -u -nr` 529 fi 530 fi 531 fi 532 533 VOLUME_NAME=unnamed_${VOLUME_MEDIATYPE} 534 # e.g., "joeuser-cdrom0/unnamed_cdrom" 535 536 if [ "$VOLUME_MEDIATYPE" = "rmdisk" ]; then 537 VOLUME_PCFS_ID=1 538 else 539 VOLUME_PCFS_ID= 540 fi 541 542 export VOLUME_ACTION VOLUME_DEVICE VOLUME_MEDIATYPE VOLUME_NAME VOLUME_PCFS_ID 543 export VOLUME_PATH VOLUME_SYMDEV VOLUME_USER VOLUME_ZONE_NAME VOLUME_ZONE_PATH 544 545 USERDIR=${VOLUME_USER}-${DEVICE} # e.g., "joeusr-cdrom0" 546 547 msg_init 548 549 if [ "$MODE" = "allocate" ]; then 550 MSGDEV=tty 551 do_allocate 552 else 553 if [ "$FLAG" = "i" ] ; then 554 MSGDEV=console 555 do_init 556 else 557 MSGDEV=tty 558 do_deallocate 559 fi 560 fi 561 562 exit $EXIT_STATUS