1 #!/usr/local/bin/python --
   2 
   3 import os, sys, string, time, traceback, re, socket
   4 
   5 Version  = '1.1'
   6 PyVersion  = '1.0'
   7 Logging  = 1
   8 Debug    = 0
   9 QmailD   = 82#这里需要和你/etc/password里面qmaild用户的一样
  10 LogFile  = None
  11 TestMode = None
  12 Filters  = []
  13 MailEnvelope = ''
  14 MailData     = ''
  15 ValidActions = { 'trap': '', 'drop' : '', 'block' :'' }
  16 
  17 #这里是你通过ucp协议将消息发送到哪个服务器的哪个端口
  18 def mail_sms(msg):
  19     host = "xxx.xxx.114.2"
  20     port = 9999
  21     buf = 500
  22     addr = (host,port)
  23 
  24     # Create socket
  25     UDPSock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
  26 
  27     def_msg = msg;
  28     UDPSock.sendto(def_msg,addr)
  29 
  30     # Close socket
  31     UDPSock.close()
  32 
  33 def openlog():
  34   global LogFile
  35 
  36   if not Logging:
  37      return
  38   try:
  39     LogFile = os.open('/var/log/qmfilt', os.O_APPEND|os.O_WRONLY|os.O_CREAT, 0744)
  40   except:
  41     if TestMode:
  42       print "Can't create /var/log/qmfilt"
  43     else:
  44       # Indicate a write error
  45       sys.exit(53)
  46 
  47 def log(message):
  48   message = time.asctime(time.localtime(time.time())) + " - " + message + "\n"
  49   if Logging:
  50     os.write(LogFile, message)
  51     # Indicate a write error
  52     #sys.exit(53)
  53   if TestMode:
  54     print message
  55 
  56 
  57 def showFilters():
  58 
  59   if len(Filters) == 0:
  60     print "No filters"
  61   for f in Filters:
  62     print "Filter -  %s - Action: %s  Regex: %s" % (f[0], f[1], f[2])
  63 
  64 def readFilters():
  65   global Filters
  66 
  67   try:
  68     file = open('/var/qmail/control/qmfilt', 'r')
  69     configLines = file.readlines()
  70     file.close()
  71   except:
  72     log("Can't read /var/qmail/control/qmfilt")
  73     if not TestMode:
  74       # Indicate a 'cannot read config file error'
  75       sys.exit(53)
  76 
  77   reg = re.compile('(.*)::(.+)::(.+)::')
  78   for line in configLines:
  79     if line[0] == '#':
  80       continue
  81     m = reg.match(line)
  82     if m != None:
  83       action = string.lower(m.group(2))
  84       if not ValidActions.has_key(action):
  85          log("Invalid action in config file [%s] - SKIPPED" %(action))
  86          continue
  87       Filters.append( [m.group(1), string.lower(m.group(2)), m.group(3)] )
  88 
  89 def readDescriptor(desc):
  90   result = ''
  91   while 1:
  92     data = os.read(desc, 16384)
  93     if data == '':
  94       break
  95     result = result + data
  96 
  97   return result
  98 
  99 def writeDescriptor(desc, data):
 100   while data:
 101     num  = os.write(desc, data)
 102     data = data[num:]
 103 
 104   os.close(desc)
 105 
 106 
 107 def filterHits():
 108 
 109   for regEx in Filters:
 110     reg = re.compile(regEx[2], re.M|re.I)
 111     match = reg.search(MailData)
 112     if match:
 113       log("Matched [%s]" % (regEx[0]))
 114       return regEx[1], regEx[0]
 115 
 116   return None,None
 117 
 118 
 119 def storeInTrap(hit):
 120   try:
 121     pid = os.getpid()
 122     mailFile = os.open('/var/qmail/qmfilt/qmfilt.%s' %(pid), os.O_WRONLY|os.O_CREAT|os.O_EXCL, 0744)
 123   except:
 124     log("Can't create /var/qmail/qmfilt/qmfilt.%s" %(pid))
 125     # Indicate a write error
 126     sys.exit(53)
 127 
 128   try:
 129     #如果不屏蔽会出现很多问题，至于为什么我还没弄明白，如果你找到问题所在，请告诉我。
 130     """(None, inode, None, None, None, None, size, None, None, None) = os.fstat(mailFile)"""
 131     os.rename('/var/qmail/qmfilt/qmfilt.%s' %(pid), '/var/qmail/qmfilt/%s.mail' %(inode))
 132   except:
 133     log("Can't rename /var/qmail/qmfilt/qmfilt.%s -> /var/qmail/qmfilt/%s.mail" %(pid, inode))
 134     # Indicate a write error
 135     sys.exit(53)
 136 
 137   try:
 138     envFile = os.open('/var/qmail/qmfilt/%s.envelope' %(inode), os.O_WRONLY|os.O_CREAT|os.O_EXCL, 0744)
 139     mesgFile = os.open('/var/qmail/qmfilt/%s.qmfilt' %(inode), os.O_WRONLY|os.O_CREAT|os.O_EXCL, 0744)
 140     writeDescriptor(mailFile, MailData)
 141     writeDescriptor(envFile,  MailEnvelope)
 142     writeDescriptor(mesgFile, "Matched filter [ %s] to file %s" %(hit, inode))
 143     log("Matched filter [ %s] to file %s" %(hit, inode))
 144   except:
 145     log("Can't create/write files into trap")
 146     # Indicate a write error
 147     sys.exit(53)
 148 
 149   return 0
 150 
 151 
 152 def sendToQmailQueue():
 153 
 154   # Open a pipe to qmail queue
 155   fd0 = os.pipe()
 156   fd1 = os.pipe()
 157   pid = os.fork()
 158   if pid < 0:
 159     log("Error couldn't fork")
 160     sys.exit(81)
 161   if pid == 0:
 162     # This is the child
 163     os.dup2(fd0[0], 0)
 164     os.dup2(fd1[0], 1)
 165     i = 2
 166     while (i < 64):
 167       try:
 168         os.close(i)
 169       except:
 170         pass
 171       i = i + 1
 172     os.chdir('/var/qmail')
 173     #time.sleep(10)
 174     os.execv("bin/qmail-queue", ('bin/qmail-queue',))
 175     log("Something went wrong")
 176     sys.exit(127)  # This is only reached on error
 177   else:
 178     # This is the parent
 179     # close the readable descriptors
 180     os.close(fd0[0])
 181     os.close(fd1[0])
 182 
 183 
 184   # Send the data
 185   recvHdr = "Received: (QMFILT: %s); " %(Version)
 186   recvHdr = recvHdr + time.strftime("%d %b %Y %H:%M:%S", time.gmtime(time.time()))
 187   recvHdr = recvHdr + " -0000\n"
 188   os.write(fd0[1], recvHdr)
 189   writeDescriptor(fd0[1], MailData)
 190   writeDescriptor(fd1[1], MailEnvelope)
 191 
 192   # Catch the exit code to return
 193   result = os.waitpid(pid, 0)[1]
 194   if PyVersion > '1.5.1':
 195     if os.WIFEXITED(result):
 196       return os.WEXITSTATUS(result)
 197     else:
 198       log("Didn't exit normally")
 199       sys.exit(81)
 200   else:
 201     return result
 202 
 203 def conver(msg):
 204   msg = msg.replace(chr(00),' ')
 205   msg = msg.replace('F','From:',1)
 206   msg = msg.replace(' T',' To:')
 207   return msg
 208 
 209 
 210 def main(argv, stdout, environ):
 211 
 212   global TestMode
 213   global MailData
 214   global MailEnvelope
 215   global PyVersion
 216 
 217   PyVersion = string.split(sys.version)[0]
 218 
 219   if len(argv) > 1 and argv[1] == '--test':
 220     TestMode = 1
 221 
 222   os.setuid(QmailD)
 223 
 224   # Insure Environment is OK
 225 
 226   # Try to open log
 227   openlog()
 228 
 229   if not os.path.exists('/var/qmail/qmfilt'):
 230     # Indicate a problem with the queue directory
 231     log("Directory /var/qmail/qmfilt doesn't exist")
 232     if not TestMode:
 233       sys.exit(61)
 234 
 235   if os.path.exists('/var/qmail/control/qmfilt'):
 236     readFilters()
 237   else:
 238     if TestMode:
 239       print "No filter file /var/qmail/control/qmfilt - no filters applied"
 240 
 241   if TestMode:
 242     showFilters()
 243 
 244   # Go no further if in test mode
 245   if TestMode:
 246     sys.exit(0)
 247 
 248 
 249 
 250   # Get the data
 251 
 252   # Read the data
 253   MailData     = readDescriptor(0)
 254 
 255   # Read the envelope
 256   MailEnvelope = readDescriptor(1)
 257   if Debug:
 258     log(MailData)
 259     log(MailEnvelope)
 260 
 261   action,hit = filterHits()
 262 
 263   if action == 'trap':
 264     storeInTrap(hit)
 265   if action == 'block':
 266     log("Matched filter [ %s] and email was BLOCKED/Refused delivery" %(hit))
 267     sys.exit(31)
 268   if action == 'drop':
 269     log("Matched filter [ %s] and email was DROPPED" %(hit))
 270   if action == None:
 271     sendToQmailQueue()
 272   #Log
 273   log(conver(MailEnvelope))
 274   #send sms
 275   mail_sms(conver(MailEnvelope))
 276 
 277   if Debug:
 278     log("qmailqueue returned [%d]" %(result))
 279   sys.exit(0)
 280 
 281 if __name__ == "__main__":
 282   try:
 283     main(sys.argv, sys.stdout, os.environ)
 284 
 285   # Catch the sys.exit() errors
 286   except SystemExit, val:
 287     sys.exit(val)
 288   except:
 289     # return a fatal error for the unknown error
 290     if TestMode:
 291       traceback.print_exc()
 292     sys.exit(81)