Shell 脚本模拟 milter 实现黑白名单及关键词过滤

程序执行流程:
1. 开始接受邮件。
2. 检查发件人是否在黑名单内,如果是拒绝接受;否则继续
3. 检查发件人是否在白名单内,如果是接收邮件;否则继续
4. 对邮件进行关键字过滤,如果邮件中包含被过滤的关键词信息,则拒绝接收
   该邮件;否则,接收该邮件。

Usage: sh mailfilter.sh  [-r reject-addr] [-a add-addr][-k keywords]
###########################################################################
#/bin/bash;                                                                 
#当有邮件到达时，先拷贝一份用户分析处理
If_Mail()
{
  User=`whoami`;
  if test -s /var/spool/mail/$User;
  then
    `cp /var/spool/mail/$User /var/spool/mail/tmp`;
    return 1;
  else
    # echo '暂无邮件要处理';
    return 0;
  fi
}

#定位制定email行号
Locate()
{
   if [ -f "/var/spool/mail/tmp" ];
   then
     Get_String=`grep -n $1  /var/spool/mail/tmp|head -1`;
     #${varible%%string*}从右向左截取最后一个string后的字符串
     Get_N=${Get_String%%:*};
     #Get_Sting=`sed '$n,touch@localhost.localdomain' /var/spool/mail/tmp|head -1`;
     #echo $Get_Sting;
   else
     Get_N=0;
   fi
  return $Get_N;
}

Deal_Black_List()
{
  #此邮件判断在黑名单中
  Mail_Addr=$1;
  Locate $Mail_Addr;
  #$?显示最后命令的退出状态
  if test $? -eq 1;
  then
    echo "$Mail_Addr 已被列入黑名单，拒收此Email!!!";
    #清空文件
    `echo > /var/spool/mail/$User`;
    # 删除空白行
    ` sed -i '/^$/d' /var/spool/mail/$User`;
    `rm /var/spool/mail/tmp`;
    return 1;
  else
    return 0;
  fi
}

Deal_White_List()
{
  Email_Addr=$1;
  Locate $Email_Addr;
  #$?显示最后命令的退出状态
  if test $? -eq 1;
  then
    return 1;
  else
    return 0;
  fi
}

Filter_Keywords()
{
    if  [ -f "/var/spool/mail/tmp" ] && [ 0 -lt `grep -c $1 /var/spool/mail/tmp` ];
    then
      echo "含有过滤关键词$1，拒收此Email!!!";
      #清空文件
      `echo > /var/spool/mail/$User`;
      # 删除空白行
      `sed -i '/^$/d' /var/spool/mail/$User`;
      `rm /var/spool/mail/tmp`;
      return 1;
    else
      echo "邮件接受";
      return 0;
    fi
}

while getopts :r:a:k:h opt
do  
    case $opt in
        r)
            Black_Addr=$OPTARG;
            ;;
        a)
            White_Addr=$OPTARG;
        ;;
        k)
            Keywords=$OPTARG;
            ;;
        h)
            echo "Usage: sh mail.sh  [-r reject-addr] [-a add-addr]";
            exit 0;
            ;;
        *)
            echo "-$opt 参数输入错误！";
            exit 0;
            ;;
    esac
done

echo '邮件过滤系统启用';
Old_Hash=`md5sum /var/spool/mail/$USER|cut -d' ' -f1`;
New_Hash=`md5sum /var/spool/mail/$USER|cut -d' ' -f1`;
while :
do
  sleep 3;
  n=0;
  If_Mail;
  if test $? -eq 0;
  then
     #邮件为空
     continue;
  else
     New_Hash=`md5sum /var/spool/mail/$USER|cut -d' ' -f1`;
  fi
  #判断是否有新文件
  if [ "$Old_Hash" = "$New_Hash" ];
  then
    continue;
  else
    echo '新邮件到达，系统即将处理！';
    Old_Hash=`md5sum /var/spool/mail/$USER|cut -d' ' -f1`;  
  fi
#黑名单过滤  
  Deal_Black_List $Black_Addr;
  if test $? -eq 1;
  then
    continue;
  fi
#白名单过滤  
  Deal_White_List $White_Addr;
  if test $? -eq 1;
  then
    echo "$White_Addr的邮件已接收"
    continue;
  fi
#关键词过滤
  Filter_Keywords $Keywords;
done
###########################################################################

这是我的一次作业，第一次写shell脚本，不乏纰漏。望指正！