#!/bin/bash
#
#-------------------------------------------------------------------------------
# FTPs files from the HMC efix ftp server
#
# Possible return codes are:
#
# 0 - no error
# 1 - usage error, invalid argument
# 2 - invalid userID/password combination
# 3 - source file not found on server
# 4 - 'ftp' connection error
# 5 - "other" 'ftp' error
# 6 - target file not written to HMC
#
#-------------------------------------------------------------------------------
#

function exit_cleanup {
#   rm -f $FTP_LOG_FILE
   chmod 777 $FTP_LOG_FILE
   exit $1
}

function ftpRemoteFiles {
   FTP_SERVER=$1
   REMOTE_FILE=$2
   REMOTE_DIR=`/usr/bin/dirname ${REMOTE_FILE}`
   FILE=`basename ${REMOTE_FILE}`
   USER=$3
   PASS=$4
   
   # remove any prior files in download directory first. Create the directory if it does not exist
   if [ -d $INSTALL_DIR/ ]; then
      rm -rf $INSTALL_DIR/*
   else
      mkdir -p $INSTALL_DIR > /dev/null
      chmod 777 $INSTALL_DIR
   fi
   cat << END_CLIENT_FTP | $FTP_CMD -vn $FTP_SERVER 2>&1 > $FTP_LOG_FILE
user $USER $PASS
bin
cd $REMOTE_DIR
get $FILE $INSTALL_DIR/$FILE
quit
END_CLIENT_FTP
} # end function getRemoteFiles

#-------------------------------------------------------------------------------
# "Main" program start   
#-------------------------------------------------------------------------------
FTP_LOG_FILE=/var/hsc/log/ftp.log
FTP_CMD=/usr/kerberos/bin/ftp
INSTALL_DIR=/dump/hsc_install.images

if [ "$1" = "" ]
then
   echo "Usage: getRemoteFiles <ftp server> <remote file> <user> <password>"
   exit_cleanup 1
fi

#-------------------------------------------------------------------------------
# 'ftp' the file from service server
#-------------------------------------------------------------------------------
ftpRemoteFiles $1 $2 $3 $4 

# good transfer, now see if the file actually got written to the HMC
if [ -f /dump/hsc_install.images/$FILE ]
then
   chmod 777 /dump/hsc_install.images/$FILE
   exit_cleanup 0
else
   # look for "Transfer complete", "Successful transfer" or equivalent msg code
   # careful on grep'ing for all these values - note the space
   grep -i "^226 " $FTP_LOG_FILE 2>&1 >/dev/null
   if [ $? -ne 0 ]
   then
     # ftp failed for some reason - find out why
     
     # file name not found on server
     grep -i "^550 " $FTP_LOG_FILE 2>&1 >/dev/null
     if [ $? -eq 0 ]
     then
        exit_cleanup 3
     fi

     # special case for i5 OS FTP
     grep -i "^250 " $FTP_LOG_FILE 2>&1 >/dev/null
     if [ $? -eq 0 ]
     then
	exit_cleanup 0
     fi

     ping -c 1 $FTP_SERVER >/dev/null 2>&1
     if [ $? -ne 0 ]
     then
	exit_cleanup 4
     fi

     grep -i "Connection timed out" $FTP_LOG_FILE 2>&1 >/dev/null
     if [ $? -eq 0 ]
     then
        exit_cleanup 4
     fi


     grep -i "^230 " $FTP_LOG_FILE 2>&1 >/dev/null
     if [ $? -ne 0 ]
     then
        exit_cleanup 2
     fi
   
     grep -i "^530 " $FTP_LOG_FILE 2>&1 >/dev/null
     if [ $? -eq 0 ]
     then
        exit_cleanup 2
     fi
   
     errRC=$(grep -i "Not connected" $FTP_LOG_FILE)
     if [ -n "$errRC" ]
     then
        exit_cleanup 4
     else
        # none of the above errors
        exit_cleanup 5
     fi
   else
      exit_cleanup 6
   fi
fi
exit_cleanup 0

