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     echo "***** refresh Kerberos V5 TGT for uid $UID *****"
  60     $KRB5DIR/bin/kinit
  61     ret=$?
  62     return $ret
  63 }
  64
  65 start_lsvcgssd() {
  66     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
  67         return 0
  68     fi
  69
  70     killall -q -9 lsvcgssd || true
  71
  72     `$SVCGSSD`
  73     num=`ps -o cmd -C "lsvcgssd" | grep lsvcgssd | wc -l`
  74     if [ $num -ne 1 ]; then
  75         echo "failed to start lsvcgssd"
  76         return 1
  77     fi
  78     return 0
  79 }
  80
  81 stop_lsvcgssd() {
  82     killall -q -9 lsvcgssd || true
  83     return 0
  84 }
  85
  86 start_lgssd() {
  87     if [ `using_krb5_sec $SECURITY` == 'n' ] ; then
  88         return 0
  89     fi
  90
  91     prepare_krb5_cache || exit 1
  92
  93     killall -q -9 lgssd || true
  94
  95     `$LGSSD`
  96     num=`ps -o cmd -C "lgssd" | grep lgssd | wc -l`
  97     if [ $num -ne 1 ]; then
  98         echo "failed to start lgssd $num"
  99         return 1
 100     fi
 101     return 0
 102 }
 103
 104 stop_lgssd() {
 105     killall -q -9 lgssd || true
 106     return 0
 107 }