lustre/tests/krb5_env.sh

   1 #!/bin/sh
   2
   3 #
   4 # KDC could be on remote hosts, but we suppose lgssd/lsvcgssd only
   5 # runs locally.
   6 #
   7
   8 export KDCHOST=${KDCHOST:-"localhost"}
   9 export KDCDIR=${KDCDIR:-"/usr/kerberos/sbin"}
  10 export KRB5DIR=${KRB5DIR:-"/usr/kerberos"}
  11 export LGSSD=${LGSSD:-"/usr/sbin/lgssd"}
  12 export SVCGSSD=${SVCGSSD:-"/usr/sbin/lsvcgssd"}
  13 export PDSH=${PDSH:-"ssh"}
  14
  15 export CHECK_KDC=${CHECKKDC:-"no"}
  16
  17 using_krb5_sec() {
  18     if [ "x$1" != "xkrb5i" -a "x$1" != "xkrb5p" ]; then
  19         echo "n"
  20     else
  21         echo "y"
  22     fi
  23 }
  24
  25 start_krb5_kdc() {
  26     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
  27         return 0
  28     fi
  29
  30     if [ "x$CHECK_KDC" == "xno" ]; then
  31         return 0
  32     fi
  33
  34     num=`$PDSH $KDCHOST "PATH=\$PATH:$KDCDIR; ps ax | grep krb5kdc | grep -v "grep" | wc -l"`
  35     if [ $num -eq 1 ]; then
  36         return 0
  37     fi
  38
  39     $PDSH $KDCHOST "PATH=\$PATH:$KDCDIR; krb5kdc"
  40     num=`$PDSH $KDCHOST "PATH=\$PATH:$KDCDIR; ps ax | grep krb5kdc | grep -v "grep" | wc -l"`
  41     if [ $num -ne 1 ]; then
  42         echo "fail to start krb5 KDC, check env KDCHOST and KDCDIR"
  43         return 1
  44     fi
  45     return 0
  46 }
  47
  48 prepare_krb5_cache() {
  49     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
  50         return 0
  51     fi
  52
  53     $KRB5DIR/bin/klist -5 -s
  54     invalid=$?
  55     if [ $invalid -eq 0 ]; then
  56         return 0
  57     fi
  58
  59     #
  60     # check installed service keytab for root
  61     #
  62     if [ $UID -eq 0 ]; then
  63         output=`$KRB5DIR/bin/klist -5 -k`
  64         if [ $? == 0 ]; then
  65             item=`echo $output | egrep "lustre_mds/.*@"`
  66             if [ "x$item" != "x" ]; then
  67                 echo "Using service keytab"
  68                 return 0
  69             fi
  70         fi
  71     fi
  72
  73     echo "***** refresh Kerberos V5 TGT for uid $UID *****"
  74     $KRB5DIR/bin/kinit
  75     ret=$?
  76     return $ret
  77 }
  78
  79 start_lsvcgssd() {
  80     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
  81         return 0
  82     fi
  83
  84     killall -q -9 lsvcgssd || true
  85
  86     `$SVCGSSD`
  87     num=`ps -o cmd -C "lsvcgssd" | grep lsvcgssd | wc -l`
  88     if [ $num -ne 1 ]; then
  89         echo "failed to start lsvcgssd"
  90         return 1
  91     fi
  92     return 0
  93 }
  94
  95 stop_lsvcgssd() {
  96     killall -q -9 lsvcgssd || true
  97     return 0
  98 }
  99
 100 start_lgssd() {
 101     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
 102         return 0
 103     fi
 104
 105     prepare_krb5_cache || exit 1
 106
 107     killall -q -9 lgssd || true
 108
 109     `$LGSSD`
 110     num=`ps -o cmd -C "lgssd" | grep lgssd | wc -l`
 111     if [ $num -ne 1 ]; then
 112         echo "failed to start lgssd $num"
 113         return 1
 114     fi
 115     return 0
 116 }
 117
 118 stop_lgssd() {
 119     killall -q -9 lgssd || true
 120     return 0
 121 }