tests/common

   1 #!/bin/bash
   2 #
   3 # Common routines for all tests
   4 #
   5
   6 _fail()
   7 {
   8         echo "$*" | tee -a $RESULTS
   9         exit 1
  10 }
  11
  12 _not_run()
  13 {
  14         echo "    [NOTRUN] $*"
  15         exit 0
  16 }
  17
  18 run_check()
  19 {
  20         echo "############### $@" >> $RESULTS 2>&1
  21         "$@" >> $RESULTS 2>&1 || _fail "failed: $@"
  22 }
  23
  24 check_prereq()
  25 {
  26         if ! [ -f $TOP/$1 ]; then
  27                 _fail "Failed prerequisities: $1";
  28         fi
  29 }
  30
  31 check_image()
  32 {
  33         image=$1
  34         echo "testing image $(basename $image)" >> $RESULTS
  35         $TOP/btrfs check $image >> $RESULTS 2>&1
  36         [ $? -eq 0 ] && _fail "btrfs check should have detected corruption"
  37
  38         run_check $TOP/btrfs check --repair $image
  39         run_check $TOP/btrfs check $image
  40 }
  41
  42 check_all_images()
  43 {
  44         dir=$1
  45         for i in $(find $dir -iname '*.img')
  46         do
  47                 echo "extracting image $(basename $i)" >> $RESULTS
  48                 $TOP/btrfs-image -r $i $i.restored || \
  49                         _fail "failed to extract image $i"
  50
  51                 check_image $i.restored
  52
  53                 rm $i.restored
  54         done
  55 }
  56
  57 # some tests need to mount the recovered image and do verifications call
  58 # 'setup_root_helper' and then check for have_root_helper == 1 if the test
  59 # needs to fail otherwise; using sudo by default for now
  60 SUDO_HELPER=
  61 NEED_SUDO_VALIDATE=unknown
  62 export SUDO_HELPER
  63 export NEED_SUDO_VALIDATE
  64 root_helper()
  65 {
  66         if [ $UID -eq 0 ]; then
  67                 "$@"
  68         else
  69                 if [ "$NEED_SUDO_VALIDATE" = 'yes' ]; then
  70                         sudo -v -n &>/dev/null || \
  71                                 _not_run "Need to validate sudo credentials"
  72                         sudo -n "$@"
  73                 elif [ "$NEED_SUDO_VALIDATE" = 'no' ]; then
  74                         sudo -n /bin/true &> /dev/null || \
  75                                 _not_run "Need to validate sudo user settings"
  76                         sudo -n "$@"
  77                 else
  78                         # should not happen
  79                         _not_run "Need to validate root privileges"
  80                 fi
  81         fi
  82 }
  83
  84 setup_root_helper()
  85 {
  86         if [ $UID -eq 0 ]; then
  87                 return
  88         fi
  89
  90         # Test for old sudo or special settings, which make sudo -v fail even
  91         # if user setting is NOPASSWD
  92         sudo -n /bin/true &>/dev/null && NEED_SUDO_VALIDATE=no
  93
  94         # Newer sudo or default sudo setting
  95         sudo -v -n &>/dev/null && NEED_SUDO_VALIDATE=yes
  96
  97         if [ "$NEED_SUDO_VALIDATE" = 'unknown' ]; then
  98                 _not_run "Need to validate root privileges"
  99         fi
 100         SUDO_HELPER=root_helper
 101 }