file-integrity.sh

BASH Shell 编程 经典教程 《高级SHELL脚本编程》中文版

#!/bin/bash
# file-integrity.sh: 检查一个给定目录下的文件
#                    是否被改动了. 

E_DIR_NOMATCH=70
E_BAD_DBFILE=71

dbfile=File_record.md5
# 存储记录的文件名(数据库文件).


set_up_database ()
{
  echo ""$directory"" > "$dbfile"
  # 把目录名写到文件的第一行. 
  md5sum "$directory"/* >> "$dbfile"
  # 在文件中附上md5 checksum和filename. 
}

check_database ()
{
  local n=0
  local filename
  local checksum

  # ------------------------------------------- #
  #  这个文件检查其实是不必要的,
  #+ 但是能更安全一些.

  if [ ! -r "$dbfile" ]
  then
    echo "Unable to read checksum database file!"
    exit $E_BAD_DBFILE
  fi
  # ------------------------------------------- #

  while read record[n]
  do

    directory_checked="${record[0]}"
    if [ "$directory_checked" != "$directory" ]
    then
      echo "Directories do not match up!"
      # 换个目录试一下. 
      exit $E_DIR_NOMATCH
    fi

    if [ "$n" -gt 0 ]   # 不是目录名. 
    then
      filename[n]=$( echo ${record[$n]} | awk '{ print $2 }' )
      #  md5sum向后写记录, 
      #+ 先写checksum, 然后写filename. 
      checksum[n]=$( md5sum "${filename[n]}" )


      if [ "${record[n]}" = "${checksum[n]}" ]
      then
        echo "${filename[n]} unchanged."

      elif [ "`basename ${filename[n]}`" != "$dbfile" ]
             #  跳过checksum数据库文件, 
             #+ 因为在每次调用脚本它都会被修改. 
	     #  ---
	     #  这不幸的意味着当我们在$PWD中运行这个脚本时侯, 
	     #+ 篡改这个checksum数
	     #+ 据库文件将不会被检测出来. 
	     #  练习: 修正这个问题.
	then
          echo "${filename[n]} : CHECKSUM ERROR!"
        # 从上次的检查之后, 文件已经被修改. 
      fi

      fi



    let "n+=1"
  done <"$dbfile"       # 从checksum数据库文件中读.  

}  

# =================================================== #
# main ()

if [ -z  "$1" ]
then
  directory="$PWD"      #  如果没指定参数的话, 
else                    #+ 那么就使用当前的工作目录. 
  directory="$1"
fi  

clear                   # 清屏.
echo " Running file integrity check on $directory"
echo

# ------------------------------------------------------------------ #
  if [ ! -r "$dbfile" ] # 是否需要建立数据库文件? 
  then
    echo "Setting up database file, \""$directory"/"$dbfile"\"."; echo
    set_up_database
  fi  
# ------------------------------------------------------------------ #

check_database          # 调用主要处理函数. 

echo 

#  你可能想把这个脚本的输出重定向到文件中, 
#+ 尤其在这个目录中有很多文件的时候. 

exit 0

#  如果要对数量非常多的文件做完整性检查, 
#+ 可以考虑一下"Tripwire"包,
#+ http://sourceforge.net/projects/tripwire/.