Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 1efc262dc5 |
@@ -1,5 +1,5 @@
|
||||
Jim Niemira (urmane@urmane.org) wrote the original C module and some quick
|
||||
and dirty python to use it. Stuart D. Gathman (stuart@gathman.org) took that
|
||||
and dirty python to use it. Stuart D. Gathman (stuart@bmsi.com) took that
|
||||
kludge and added threading and context objects to it, wrote a proper OO
|
||||
wrapper (Milter.py) that handles attachments, did lots of testing, packaged
|
||||
it with distutils, and generally transformed it from a quick hack to a
|
||||
@@ -7,6 +7,7 @@ real, usable Python extension.
|
||||
|
||||
Other contributors (in random order):
|
||||
|
||||
|
||||
Daniel Troeder
|
||||
for pointing out a typo in @noreply
|
||||
arkanes@irc.freenode.net
|
||||
@@ -43,4 +44,4 @@ Business Management Systems - http://www.bmsi.com
|
||||
for hosting the website, and providing paying clients who need milter service
|
||||
so I can work on it as part of my day job.
|
||||
|
||||
If I have left anybody out, send me a reminder: stuart@gathman.org
|
||||
If I have left anybody out, send me a reminder: stuart@bmsi.com
|
||||
|
||||
+42
-44
@@ -8,7 +8,7 @@
|
||||
# Copyright 2001,2009 Business Management Systems, Inc.
|
||||
# This code is under the GNU General Public License. See COPYING for details.
|
||||
|
||||
__version__ = '0.9.8'
|
||||
__version__ = '0.9.7'
|
||||
|
||||
import os
|
||||
import re
|
||||
@@ -21,6 +21,12 @@ from functools import wraps
|
||||
_seq_lock = thread.allocate_lock()
|
||||
_seq = 0
|
||||
|
||||
## @fn set_flags(flags)
|
||||
# @brief Enable optional %milter actions.
|
||||
# Certain %milter actions need to be enabled before calling milter.runmilter()
|
||||
# or they throw an exception.
|
||||
# @param flags Bit ored mask of optional actions to enable
|
||||
|
||||
def uniqueID():
|
||||
"""Return a unique sequence number (incremented on each call).
|
||||
"""
|
||||
@@ -221,7 +227,7 @@ class Base(object):
|
||||
# Some optional actions may be disabled by calling milter.set_flags(), or
|
||||
# by overriding the negotiate callback. The bits include:
|
||||
# <code>ADDHDRS,CHGBODY,MODBODY,ADDRCPT,ADDRCPT_PAR,DELRCPT
|
||||
# CHGHDRS,QUARANTINE,CHGFROM,SETSYMLIST</code>.
|
||||
# CHGHDRS,QUARANTINE,CHGFROM,SETSMLIST</code>.
|
||||
# The <code>Milter.CURR_ACTS</code> bitmask is all actions
|
||||
# known when the milter module was compiled.
|
||||
# Application code can also inspect this field to determine
|
||||
@@ -410,21 +416,12 @@ class Base(object):
|
||||
|
||||
## Set the SMTP reply code and message.
|
||||
# If the MTA does not support setmlreply, then only the
|
||||
# first msg line is used. Any '%%' in a message line
|
||||
# first msg line is used. Any '%' in a message line
|
||||
# must be doubled, or libmilter will silently ignore the setreply.
|
||||
# Beginning with 0.9.6, we test for that case and throw ValueError to avoid
|
||||
# head scratching. What will <i>really</i> irritate you, however,
|
||||
# is that if you carefully double any '%%', your message will be
|
||||
# sent - but with the '%%' still doubled!
|
||||
# See <a href="https://www.milter.org/developers/api/smfi_setreply">
|
||||
# smfi_setreply</a> for more information.
|
||||
# @param rcode The three-digit (RFC 821/2821) SMTP reply code as a string.
|
||||
# rcode cannot be None, and <b>must be a valid 4XX or 5XX reply code</b>.
|
||||
# @param xcode The extended (RFC 1893/2034) reply code. If xcode is None,
|
||||
# no extended code is used. Otherwise, xcode must conform to RFC 1893/2034.
|
||||
# @param msg The text part of the SMTP reply. If msg is None,
|
||||
# an empty message is used.
|
||||
# @param ml Optional additional message lines.
|
||||
# is that if you carefully double any '%', your message will be
|
||||
# sent - but with the '%' still doubled!
|
||||
def setreply(self,rcode,xcode=None,msg=None,*ml):
|
||||
for m in (msg,)+ml:
|
||||
if 1 in [len(s)&1 for s in R.findall(m)]:
|
||||
@@ -432,29 +429,17 @@ class Base(object):
|
||||
return self._ctx.setreply(rcode,xcode,msg,*ml)
|
||||
|
||||
## Tell the MTA which macro names will be used.
|
||||
# This information can reduce the size of messages received from sendmail,
|
||||
# and hence could reduce bandwidth between sendmail and your milter where
|
||||
# that is a factor. The <code>Milter.SETSYMLIST</code> action flag must be
|
||||
# set. The protocol stages are M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT,
|
||||
# M_DATA, M_EOM, M_EOH.
|
||||
# The <code>Milter.SETSMLIST</code> action flag must be set.
|
||||
#
|
||||
# May only be called from negotiate callback.
|
||||
# @since 0.9.8, previous version was misspelled!
|
||||
# @param stage the protocol stage to set to macro list for,
|
||||
# one of the M_* constants defined in Milter
|
||||
# @param macros space separated and/or lists of strings
|
||||
def setsymlist(self,stage,*macros):
|
||||
if not self._actions & SETSYMLIST: raise DisabledAction("SETSYMLIST")
|
||||
a = []
|
||||
for m in macros:
|
||||
try:
|
||||
m = m.encode('utf8')
|
||||
except: pass
|
||||
try:
|
||||
m = m.split(' ')
|
||||
except: pass
|
||||
a += m
|
||||
return self._ctx.setsmlist(stage,' '.join(a))
|
||||
# @since 0.9.2
|
||||
# @param stage the protocol stage to set to macro list for
|
||||
# @param macros a string with a space delimited list of macros
|
||||
def setsmlist(self,stage,macros):
|
||||
if not self._actions & SETSMLIST: raise DisabledAction("SETSMLIST")
|
||||
if type(macros) in (list,tuple):
|
||||
macros = ' '.join(macros)
|
||||
return self._ctx.setsmlist(stage,macros)
|
||||
|
||||
# Milter methods which can only be called from eom callback.
|
||||
|
||||
@@ -718,7 +703,28 @@ def envcallback(c,args):
|
||||
# @param socketname the socket to be passed to milter.setconn()
|
||||
# @param timeout the time in secs the MTA should wait for a response before
|
||||
# considering this %milter dead
|
||||
def runmilter(name,socketname,timeout = 0,rmsock=True):
|
||||
def runmilter(name,socketname,timeout = 0):
|
||||
# This bit is here on the assumption that you will be starting this filter
|
||||
# before sendmail. If sendmail is not running and the socket already exists,
|
||||
# libmilter will throw a warning. If sendmail is running, this is still
|
||||
# safe if there are no messages currently being processed. It's safer to
|
||||
# shutdown sendmail, kill the filter process, restart the filter, and then
|
||||
# restart sendmail.
|
||||
pos = socketname.find(':')
|
||||
if pos > 1:
|
||||
s = socketname[:pos]
|
||||
fname = socketname[pos+1:]
|
||||
else:
|
||||
s = "unix"
|
||||
fname = socketname
|
||||
if s == "unix" or s == "local":
|
||||
print "Removing %s" % fname
|
||||
try:
|
||||
os.unlink(fname)
|
||||
except os.error, x:
|
||||
import errno
|
||||
if x.errno != errno.ENOENT:
|
||||
raise milter.error(x)
|
||||
|
||||
# The default flags set include everything
|
||||
# milter.set_flags(milter.ADDHDRS)
|
||||
@@ -749,14 +755,6 @@ def runmilter(name,socketname,timeout = 0,rmsock=True):
|
||||
unknown=lambda ctx,cmd: ctx.getpriv().unknown(cmd),
|
||||
negotiate=ncb
|
||||
)
|
||||
|
||||
# We remove the socket here by default on the assumption that you will be
|
||||
# starting this filter before sendmail. If sendmail is not running and the
|
||||
# socket already exists, libmilter will throw a warning. If sendmail is
|
||||
# running, this is still safe if there are no messages currently being
|
||||
# processed. It's safer to shutdown sendmail, kill the filter process,
|
||||
# restart the filter, and then restart sendmail.
|
||||
milter.opensocket(rmsock)
|
||||
start_seq = _seq
|
||||
try:
|
||||
milter.main()
|
||||
|
||||
+9
-12
@@ -10,9 +10,6 @@
|
||||
# CBV results.
|
||||
#
|
||||
# $Log$
|
||||
# Revision 1.9 2008/05/08 21:35:57 customdesigned
|
||||
# Allow explicitly whitelisted email from banned_users.
|
||||
#
|
||||
# Revision 1.8 2007/09/03 16:18:45 customdesigned
|
||||
# Delete unparseable timestamps when loading address cache. These have
|
||||
# arisen because of failure to parse MAIL FROM properly. Will have to
|
||||
@@ -75,8 +72,8 @@ class AddrCache(object):
|
||||
except OSError:
|
||||
fp = ()
|
||||
for ln in fp:
|
||||
try:
|
||||
rcpt,ts = ln.strip().split(None,1)
|
||||
try:
|
||||
rcpt,ts = ln.strip().split(None,1)
|
||||
try:
|
||||
l = time.strptime(ts,AddrCache.time_format)
|
||||
t = time.mktime(l)
|
||||
@@ -87,11 +84,11 @@ class AddrCache(object):
|
||||
except: # unparsable timestamp - likely garbage
|
||||
changed = True
|
||||
continue
|
||||
except: # manual entry (no timestamp)
|
||||
cache[ln.strip().lower()] = (now,None)
|
||||
wfp.write(ln)
|
||||
except: # manual entry (no timestamp)
|
||||
cache[ln.strip().lower()] = (now,None)
|
||||
wfp.write(ln)
|
||||
if changed:
|
||||
lock.commit(self.fname+'.old')
|
||||
lock.commit(self.fname+'.old')
|
||||
else:
|
||||
lock.unlock()
|
||||
except IOError:
|
||||
@@ -129,13 +126,13 @@ class AddrCache(object):
|
||||
ts,res = self.cache[lsender]
|
||||
too_old = time.time() - self.age*24*60*60 # max age in days
|
||||
if not ts or ts > too_old:
|
||||
return res
|
||||
return res
|
||||
del self.cache[lsender]
|
||||
raise KeyError, sender
|
||||
except KeyError,x:
|
||||
try:
|
||||
user,host = sender.split('@',1)
|
||||
return self.__getitem__(host)
|
||||
user,host = sender.split('@',1)
|
||||
return self.__getitem__(host)
|
||||
except ValueError:
|
||||
raise x
|
||||
|
||||
|
||||
+13
-15
@@ -1,5 +1,4 @@
|
||||
from ConfigParser import ConfigParser
|
||||
import os.path
|
||||
|
||||
class MilterConfigParser(ConfigParser):
|
||||
|
||||
@@ -21,7 +20,7 @@ class MilterConfigParser(ConfigParser):
|
||||
return [q.strip() for q in self.get(sect,opt).split(',')]
|
||||
return []
|
||||
|
||||
def getaddrset(self,sect,opt,dir=''):
|
||||
def getaddrset(self,sect,opt):
|
||||
if not self.has_option(sect,opt):
|
||||
return {}
|
||||
s = self.get(sect,opt)
|
||||
@@ -30,14 +29,13 @@ class MilterConfigParser(ConfigParser):
|
||||
q = q.strip()
|
||||
if q.startswith('file:'):
|
||||
domain = q[5:].lower()
|
||||
fname = os.path.join(dir,domain)
|
||||
d[domain] = d.setdefault(domain,[]) + open(fname,'r').read().split()
|
||||
d[domain] = d.setdefault(domain,[]) + open(domain,'r').read().split()
|
||||
else:
|
||||
user,domain = q.split('@')
|
||||
d.setdefault(domain.lower(),[]).append(user)
|
||||
user,domain = q.split('@')
|
||||
d.setdefault(domain.lower(),[]).append(user)
|
||||
return d
|
||||
|
||||
def getaddrdict(self,sect,opt,dir=''):
|
||||
def getaddrdict(self,sect,opt):
|
||||
if not self.has_option(sect,opt):
|
||||
return {}
|
||||
d = {}
|
||||
@@ -45,14 +43,14 @@ class MilterConfigParser(ConfigParser):
|
||||
q = q.strip()
|
||||
if self.has_option(sect,q):
|
||||
l = self.get(sect,q)
|
||||
for addr in l.split(','):
|
||||
addr = addr.strip()
|
||||
if addr.startswith('file:'):
|
||||
fname = os.path.join(dir,addr[5:])
|
||||
for a in open(fname,'r').read().split():
|
||||
d[a] = q
|
||||
else:
|
||||
d[addr] = q
|
||||
for addr in l.split(','):
|
||||
addr = addr.strip()
|
||||
if addr.startswith('file:'):
|
||||
fname = addr[5:]
|
||||
for a in open(fname,'r').read().split():
|
||||
d[a] = q
|
||||
else:
|
||||
d[addr] = q
|
||||
return d
|
||||
|
||||
def getdefault(self,sect,opt,default=None):
|
||||
|
||||
+8
-11
@@ -5,9 +5,6 @@
|
||||
# Send DSNs, do call back verification,
|
||||
# and generate DSN messages from a template
|
||||
# $Log$
|
||||
# Revision 1.22 2011/03/18 20:41:31 customdesigned
|
||||
# Python2.6 SMTP.close() fails when instance never connected.
|
||||
#
|
||||
# Revision 1.21 2011/03/03 05:11:58 customdesigned
|
||||
# Release 0.9.4
|
||||
#
|
||||
@@ -117,17 +114,17 @@ def send_dsn(mailfrom,receiver,msg=None,timeout=600,session=None,ourfrom=''):
|
||||
if a[0] == receiver:
|
||||
return (553,'Fraudulent MX for %s: %s' % (domain,host))
|
||||
if not (200 <= code <= 299):
|
||||
raise smtplib.SMTPHeloError(code, resp)
|
||||
raise smtplib.SMTPHeloError(code, resp)
|
||||
if msg:
|
||||
try:
|
||||
smtp.sendmail('<%s>'%ourfrom,mailfrom,msg)
|
||||
except smtplib.SMTPSenderRefused:
|
||||
smtp.sendmail('<%s>'%ourfrom,mailfrom,msg)
|
||||
except smtplib.SMTPSenderRefused:
|
||||
# does not accept DSN, try postmaster (at the risk of mail loops)
|
||||
smtp.sendmail('<postmaster@%s>'%receiver,mailfrom,msg)
|
||||
smtp.sendmail('<postmaster@%s>'%receiver,mailfrom,msg)
|
||||
else: # CBV
|
||||
code,resp = smtp.docmd('MAIL FROM: <%s>'%ourfrom)
|
||||
if code != 250:
|
||||
raise smtplib.SMTPSenderRefused(code, resp, '<%s>'%ourfrom)
|
||||
code,resp = smtp.docmd('MAIL FROM: <%s>'%ourfrom)
|
||||
if code != 250:
|
||||
raise smtplib.SMTPSenderRefused(code, resp, '<%s>'%ourfrom)
|
||||
if isinstance(mailfrom,basestring):
|
||||
mailfrom = [mailfrom]
|
||||
badrcpts = {}
|
||||
@@ -135,7 +132,7 @@ def send_dsn(mailfrom,receiver,msg=None,timeout=600,session=None,ourfrom=''):
|
||||
code,resp = smtp.rcpt(rcpt)
|
||||
if code not in (250,251):
|
||||
badrcpts[rcpt] = (code,resp)# permanent error
|
||||
smtp.quit()
|
||||
smtp.quit()
|
||||
if len(badrcpts) == 1:
|
||||
return badrcpts.values()[0] # permanent error
|
||||
if badrcpts:
|
||||
|
||||
+3
-3
@@ -68,8 +68,8 @@ def is_dynip(host,addr):
|
||||
if ia[2:] in (g[:2],g[-2:]): return True
|
||||
for m in ip3.finditer(host):
|
||||
if int(m.group()) == ia[3]:
|
||||
h = host[:m.start()] + '<3>' + host[m.end():]
|
||||
break
|
||||
h = host[:m.start()] + '<3>' + host[m.end():]
|
||||
break
|
||||
if rehmac.search(h): return True
|
||||
if host.find(''.join(a[:3])) >= 0: return True
|
||||
if host.find(''.join(a[1:])) >= 0: return True
|
||||
@@ -86,7 +86,7 @@ if __name__ == '__main__':
|
||||
if a[3:5] == ['connect','from']:
|
||||
host = a[5]
|
||||
if host.startswith('[') and host.endswith(']'):
|
||||
continue # no PTR
|
||||
continue # no PTR
|
||||
ip = a[7][2:-2]
|
||||
if ip in seen: continue
|
||||
seen.add(ip)
|
||||
|
||||
+8
-54
@@ -18,19 +18,13 @@ def quoteAddress(s):
|
||||
class Record(object):
|
||||
__slots__ = ( 'firstseen', 'lastseen', 'umis', 'cnt' )
|
||||
|
||||
def __init__(self,timeinc=0):
|
||||
now = time.time() + timeinc
|
||||
def __init__(self):
|
||||
now = time.time()
|
||||
self.firstseen = now
|
||||
self.lastseen = now
|
||||
self.cnt = 0
|
||||
self.umis = None
|
||||
|
||||
def __str__(self):
|
||||
return "Grey[%s:%s:%s:%d]" % (
|
||||
time.ctime(self.firstseen),time.ctime(self.lastseen),
|
||||
self.umis,self.cnt
|
||||
)
|
||||
|
||||
class Greylist(object):
|
||||
|
||||
def __init__(self,dbname,grey_time=10,grey_expire=4,grey_retain=36):
|
||||
@@ -40,38 +34,8 @@ class Greylist(object):
|
||||
self.greylist_retain = grey_retain * 24 * 3600 # days
|
||||
self.dbp = shelve.open(dbname,'c',protocol=2)
|
||||
self.lock = thread.allocate_lock()
|
||||
|
||||
def export_csv(self,fp,timeinc=0):
|
||||
"Export records to csv."
|
||||
import csv
|
||||
dbp = self.dbp
|
||||
w = csv.writer(fp)
|
||||
now = time.time() + timeinc
|
||||
for key, r in dbp.iteritems():
|
||||
if now > r.lastseen + self.greylist_retain: continue
|
||||
ip,sender,recipient = key.rsplit(':',2)
|
||||
w.writerow([ip,sender,recipient,r.firstseen,r.lastseen,r.cnt,r.umis])
|
||||
|
||||
def clean(self,timeinc=0):
|
||||
"Delete records past the retention limit."
|
||||
now = time.time() + timeinc
|
||||
cnt = 0
|
||||
dbp = self.dbp
|
||||
for key, r in dbp.iteritems():
|
||||
#print key,r,time.ctime(now)
|
||||
if now > r.lastseen + self.greylist_retain:
|
||||
self.lock.acquire()
|
||||
try:
|
||||
r = dbp[key]
|
||||
now = time.time() + timeinc
|
||||
if now > r.lastseen + self.greylist_retain:
|
||||
del dbp[key]
|
||||
cnt += 1
|
||||
finally:
|
||||
self.lock.release()
|
||||
return cnt
|
||||
|
||||
def check(self,ip,sender,recipient,timeinc=0):
|
||||
def check(self,ip,sender,recipient):
|
||||
"Return number of allowed messages for greylist triple."
|
||||
sender = quoteAddress(sender)
|
||||
recipient = quoteAddress(recipient)
|
||||
@@ -81,15 +45,15 @@ class Greylist(object):
|
||||
dbp = self.dbp
|
||||
try:
|
||||
r = dbp[key]
|
||||
now = time.time() + timeinc
|
||||
now = time.time()
|
||||
if now > r.lastseen + self.greylist_retain:
|
||||
# expired
|
||||
log.debug('Expired greylist: %s',key)
|
||||
r = Record(timeinc)
|
||||
r = Record()
|
||||
elif now < r.firstseen + self.greylist_time + 5:
|
||||
# still greylisted
|
||||
log.debug('Early greylist: %s',key)
|
||||
#r = Record(timeinc)
|
||||
#r = Record()
|
||||
r.lastseen = now
|
||||
elif r.cnt or now < r.firstseen + self.greylist_expire:
|
||||
# in greylist window or active
|
||||
@@ -99,22 +63,12 @@ class Greylist(object):
|
||||
else:
|
||||
# passed greylist window
|
||||
log.debug('Late greylist: %s',key)
|
||||
r = Record(timeinc)
|
||||
r = Record()
|
||||
dbp[key] = r
|
||||
except:
|
||||
r = Record(timeinc)
|
||||
r = Record()
|
||||
dbp[key] = r
|
||||
dbp.sync()
|
||||
finally:
|
||||
self.lock.release()
|
||||
return r.cnt
|
||||
|
||||
def close(self):
|
||||
self.dbp.close()
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
g = Greylist(sys.argv[1],5,24,36)
|
||||
try:
|
||||
g.export_csv(sys.stdout)
|
||||
finally: g.close()
|
||||
|
||||
@@ -1,106 +0,0 @@
|
||||
import time
|
||||
import logging
|
||||
import urllib
|
||||
import sqlite3
|
||||
import thread
|
||||
from datetime import datetime
|
||||
|
||||
log = logging.getLogger('milter.greylist')
|
||||
|
||||
_db_lock = thread.allocate_lock()
|
||||
|
||||
class Greylist(object):
|
||||
|
||||
def __init__(self,dbname,grey_time=10,grey_expire=4,grey_retain=36):
|
||||
self.ignoreLastByte = False
|
||||
self.greylist_time = grey_time * 60 # minutes
|
||||
self.greylist_expire = grey_expire * 3600 # hours
|
||||
self.greylist_retain = grey_retain * 24 * 3600 # days
|
||||
self.conn = sqlite3.connect(dbname)
|
||||
self.conn.row_factory = sqlite3.Row
|
||||
try:
|
||||
self.conn.execute('''create table greylist(
|
||||
ip text , sender text, recipient text,
|
||||
firstseen timestamp, lastseen timestamp, cnt integer, umis text,
|
||||
primary key (ip,sender,recipient))''')
|
||||
except: pass
|
||||
|
||||
def import_csv(self,fp):
|
||||
import csv
|
||||
rdr = csv.reader(fp)
|
||||
cur = self.conn.execute('begin immediate')
|
||||
try:
|
||||
for r in rdr:
|
||||
cur.execute('''insert into
|
||||
greylist(ip,sender,recipient,firstseen,lastseen,cnt,umis)
|
||||
values(?,?,?,?,?,?,?)''', r)
|
||||
self.conn.commit()
|
||||
finally:
|
||||
cur.close();
|
||||
|
||||
def clean(self,timeinc=0):
|
||||
"Delete records past the retention limit."
|
||||
now = time.time() + timeinc - self.greylist_retain
|
||||
cur = self.conn.cursor()
|
||||
try:
|
||||
cur.execute('delete from greylist where lastseen < ?',(now,))
|
||||
cnt = cur.rowcount
|
||||
self.conn.commit()
|
||||
finally: cur.close()
|
||||
return cnt
|
||||
|
||||
def check(self,ip,sender,recipient,timeinc=0):
|
||||
"Return number of allowed messages for greylist triple."
|
||||
_db_lock.acquire()
|
||||
cur = self.conn.execute('begin immediate')
|
||||
try:
|
||||
cur.execute('''select firstseen,lastseen,cnt,umis from greylist where
|
||||
ip=? and sender=? and recipient=?''',(ip,sender,recipient))
|
||||
r = cur.fetchone()
|
||||
now = time.time() + timeinc
|
||||
cnt = 0
|
||||
if not r:
|
||||
cur.execute('''insert into
|
||||
greylist(ip,sender,recipient,firstseen,lastseen,cnt,umis)
|
||||
values(?,?,?,?,?,?,?)''', (ip,sender,recipient,now,now,0,None))
|
||||
elif now > r['lastseen'] + self.greylist_retain:
|
||||
# expired
|
||||
log.debug('Expired greylist: %s:%s:%s',ip,sender,recipient)
|
||||
cur.execute('''update greylist set firstseen=?,lastseen=?,cnt=?,umis=?
|
||||
where ip=? and sender=? and recipient=?''',
|
||||
(now,now,0,None,ip,sender,recipient))
|
||||
elif now < r['firstseen'] + self.greylist_time + 5:
|
||||
# still greylisted
|
||||
log.debug('Early greylist: %s:%s:%s',ip,sender,recipient)
|
||||
#r = Record()
|
||||
cur.execute('''update greylist set lastseen=?
|
||||
where ip=? and sender=? and recipient=?''',
|
||||
(now,ip,sender,recipient))
|
||||
elif r['cnt'] or now < r['firstseen'] + self.greylist_expire:
|
||||
# in greylist window or active
|
||||
cnt = r['cnt'] + 1
|
||||
cur.execute('''update greylist set lastseen=?,cnt=?
|
||||
where ip=? and sender=? and recipient=?''',
|
||||
(now,cnt,ip,sender,recipient))
|
||||
log.debug('Active greylist(%d): %s:%s:%s',cnt,ip,sender,recipient)
|
||||
else:
|
||||
# passed greylist window
|
||||
log.debug('Late greylist: %s:%s:%s',ip,sender,recipient)
|
||||
cur.execute('''update greylist set firstseen=?,lastseen=?,cnt=?,umis=?
|
||||
where ip=? and sender=? and recipient=?''',
|
||||
(now,now,0,None,ip,sender,recipient))
|
||||
self.conn.commit()
|
||||
finally:
|
||||
cur.close()
|
||||
_db_lock.release()
|
||||
return cnt
|
||||
|
||||
def close(self):
|
||||
self.conn.close()
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
g = Greylist(sys.argv[1])
|
||||
try:
|
||||
g.import_csv(sys.stdin)
|
||||
finally: g.close()
|
||||
+3
-3
@@ -31,8 +31,8 @@ class PLock(object):
|
||||
os.chown(self.lockname,-1,st.st_gid)
|
||||
except:
|
||||
if strict_perms:
|
||||
self.unlock()
|
||||
raise
|
||||
self.unlock()
|
||||
raise
|
||||
return self.fp
|
||||
|
||||
def wlock(self,lockname=None):
|
||||
@@ -51,7 +51,7 @@ class PLock(object):
|
||||
self.fp = None
|
||||
if backname:
|
||||
try:
|
||||
os.remove(backname)
|
||||
os.remove(backname)
|
||||
except OSError: pass
|
||||
os.link(self.basename,backname)
|
||||
os.rename(self.lockname,self.basename)
|
||||
|
||||
+21
-21
@@ -48,11 +48,11 @@ def inet_ntop(s):
|
||||
e = n[:l]
|
||||
for i in range(9-l):
|
||||
if a[i:i+l] == e:
|
||||
if i == 0:
|
||||
return ':'+':%x'*(8-l) % a[l:]
|
||||
if i == 8 - l:
|
||||
return '%x:'*(8-l) % a[:-l] + ':'
|
||||
return '%x:'*i % a[:i] + ':%x'*(8-l-i) % a[i+l:]
|
||||
if i == 0:
|
||||
return ':'+':%x'*(8-l) % a[l:]
|
||||
if i == 8 - l:
|
||||
return '%x:'*(8-l) % a[:-l] + ':'
|
||||
return '%x:'*i % a[:i] + ':%x'*(8-l-i) % a[i+l:]
|
||||
return "%x:%x:%x:%x:%x:%x:%x:%x" % a
|
||||
|
||||
def inet_pton(p):
|
||||
@@ -89,29 +89,29 @@ def inet_pton(p):
|
||||
m = RE_IP4.search(s)
|
||||
try:
|
||||
if m:
|
||||
pos = m.start()
|
||||
ip4 = [int(i) for i in s[pos:].split('.')]
|
||||
if not pos:
|
||||
return struct.pack('!QLBBBB',0,65535,*ip4)
|
||||
s = s[:pos]+'%x%02x:%x%02x'%tuple(ip4)
|
||||
pos = m.start()
|
||||
ip4 = [int(i) for i in s[pos:].split('.')]
|
||||
if not pos:
|
||||
return struct.pack('!QLBBBB',0,65535,*ip4)
|
||||
s = s[:pos]+'%x%02x:%x%02x'%tuple(ip4)
|
||||
a = s.split('::')
|
||||
if len(a) == 2:
|
||||
l,r = a
|
||||
if not l:
|
||||
r = r.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
l,r = a
|
||||
if not l:
|
||||
r = r.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
*[0]*(8-len(r)) + [int(s,16) for s in r])
|
||||
if not r:
|
||||
l = l.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
if not r:
|
||||
l = l.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
*[int(s,16) for s in l] + [0]*(8-len(l)))
|
||||
l = l.split(':')
|
||||
r = r.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
l = l.split(':')
|
||||
r = r.split(':')
|
||||
return struct.pack('!HHHHHHHH',
|
||||
*[int(s,16) for s in l] + [0]*(8-len(l)-len(r))
|
||||
+ [int(s,16) for s in r])
|
||||
if len(a) == 1:
|
||||
return struct.pack('!HHHHHHHH',
|
||||
return struct.pack('!HHHHHHHH',
|
||||
*[int(s,16) for s in a[0].split(':')])
|
||||
except ValueError: pass
|
||||
raise ValueError,p
|
||||
|
||||
-192
@@ -1,192 +0,0 @@
|
||||
## @package Milter.test
|
||||
# A test framework for milters
|
||||
|
||||
import rfc822
|
||||
import StringIO
|
||||
import Milter
|
||||
|
||||
Milter.NOREPLY = Milter.CONTINUE
|
||||
|
||||
## Test mixin for unit testing %milter applications.
|
||||
# This mixin overrides many Milter.MilterBase methods
|
||||
# with stub versions that simply record what was done.
|
||||
# @since 0.9.8
|
||||
class TestBase(object):
|
||||
|
||||
def __init__(self,logfile='test/milter.log'):
|
||||
self._protocol = 0
|
||||
self.logfp = open(logfile,"a")
|
||||
## List of recipients deleted
|
||||
self._delrcpt = []
|
||||
## List of recipients added
|
||||
self._addrcpt = []
|
||||
## Macros defined
|
||||
self._macros = { }
|
||||
## The message body.
|
||||
self._body = None
|
||||
## True if the %milter replaced the message body.
|
||||
self._bodyreplaced = False
|
||||
## True if the %milter changed any headers.
|
||||
self._headerschanged = False
|
||||
## Reply codes and messages set by the %milter
|
||||
self._reply = None
|
||||
## The rfc822 message object for the current email being fed to the %milter.
|
||||
self._msg = None
|
||||
self._symlist = [ None, None, None, None, None, None, None ]
|
||||
|
||||
def log(self,*msg):
|
||||
for i in msg: print >>self.logfp, i,
|
||||
print >>self.logfp
|
||||
|
||||
## Set a macro value.
|
||||
# These are retrieved by the %milter with getsymval.
|
||||
# @param name the macro name, as passed to getsymval
|
||||
# @param val the macro value
|
||||
def setsymval(self,name,val):
|
||||
self._macros[name] = val
|
||||
|
||||
def getsymval(self,name):
|
||||
# FIXME: track stage, and use _symlist
|
||||
return self._macros.get(name,'')
|
||||
|
||||
def replacebody(self,chunk):
|
||||
if self._body:
|
||||
self._body.write(chunk)
|
||||
self._bodyreplaced = True
|
||||
else:
|
||||
raise IOError,"replacebody not called from eom()"
|
||||
|
||||
# FIXME: rfc822 indexing does not really reflect the way chg/add header
|
||||
# work for a %milter
|
||||
def chgheader(self,field,idx,value):
|
||||
if not self._body:
|
||||
raise IOError,"chgheader not called from eom()"
|
||||
self.log('chgheader: %s[%d]=%s' % (field,idx,value))
|
||||
if value == '':
|
||||
del self._msg[field]
|
||||
else:
|
||||
self._msg[field] = value
|
||||
self._headerschanged = True
|
||||
|
||||
def addheader(self,field,value,idx=-1):
|
||||
if not self._body:
|
||||
raise IOError,"addheader not called from eom()"
|
||||
self.log('addheader: %s=%s' % (field,value))
|
||||
self._msg[field] = value
|
||||
self._headerschanged = True
|
||||
|
||||
def delrcpt(self,rcpt):
|
||||
if not self._body:
|
||||
raise IOError,"delrcpt not called from eom()"
|
||||
self._delrcpt.append(rcpt)
|
||||
|
||||
def addrcpt(self,rcpt):
|
||||
if not self._body:
|
||||
raise IOError,"addrcpt not called from eom()"
|
||||
self._addrcpt.append(rcpt)
|
||||
|
||||
## Save the reply codes and messages in self._reply.
|
||||
def setreply(self,rcode,xcode,*msg):
|
||||
self._reply = (rcode,xcode) + msg
|
||||
|
||||
def setsymlist(self,stage,macros):
|
||||
if not self._actions & SETSYMLIST: raise DisabledAction("SETSYMLIST")
|
||||
# not used yet, but just for grins we save the data
|
||||
a = []
|
||||
for m in macros:
|
||||
try:
|
||||
m = m.encode('utf8')
|
||||
except: pass
|
||||
try:
|
||||
m = m.split(' ')
|
||||
except: pass
|
||||
a += m
|
||||
self._symlist[stage] = set(a)
|
||||
|
||||
## Feed a file like object to the %milter. Calls envfrom, envrcpt for
|
||||
# each recipient, header for each header field, body for each body
|
||||
# block, and finally eom. A return code from the %milter other than
|
||||
# CONTINUE returns immediately with that return code.
|
||||
#
|
||||
# This is a convenience method, a test could invoke the callbacks
|
||||
# in sequence on its own - and for some complex tests, this may
|
||||
# be necessary.
|
||||
# @param fp the file with rfc2822 message stream
|
||||
# @param sender the MAIL FROM
|
||||
# @param rcpt RCPT TO - additional recipients may follow
|
||||
def feedFile(self,fp,sender="spam@adv.com",rcpt="victim@lamb.com",*rcpts):
|
||||
self._body = None
|
||||
self._bodyreplaced = False
|
||||
self._headerschanged = False
|
||||
self._reply = None
|
||||
msg = rfc822.Message(fp)
|
||||
rc = self.envfrom('<%s>'%sender)
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
for rcpt in (rcpt,) + rcpts:
|
||||
rc = self.envrcpt('<%s>'%rcpt)
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
line = None
|
||||
for h in msg.headers:
|
||||
if h[:1].isspace():
|
||||
line = line + h
|
||||
continue
|
||||
if not line:
|
||||
line = h
|
||||
continue
|
||||
s = line.split(': ',1)
|
||||
if len(s) > 1: val = s[1].strip()
|
||||
else: val = ''
|
||||
rc = self.header(s[0],val)
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
line = h
|
||||
if line:
|
||||
s = line.split(': ',1)
|
||||
rc = self.header(s[0],s[1])
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
rc = self.eoh()
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
while 1:
|
||||
buf = fp.read(8192)
|
||||
if len(buf) == 0: break
|
||||
rc = self.body(buf)
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
self._msg = msg
|
||||
self._body = StringIO.StringIO()
|
||||
rc = self.eom()
|
||||
if self._bodyreplaced:
|
||||
body = self._body.getvalue()
|
||||
else:
|
||||
msg.rewindbody()
|
||||
body = msg.fp.read()
|
||||
self._body = StringIO.StringIO()
|
||||
self._body.writelines(msg.headers)
|
||||
self._body.write('\n')
|
||||
self._body.write(body)
|
||||
return rc
|
||||
|
||||
## Feed an email contained in a file to the %milter.
|
||||
# This is a convenience method that invokes @link #feedFile feedFile @endlink.
|
||||
# @param sender MAIL FROM
|
||||
# @param rcpts RCPT TO, multiple recipients may be supplied
|
||||
def feedMsg(self,fname,sender="spam@adv.com",*rcpts):
|
||||
with open('test/'+fname,'r') as fp:
|
||||
return self.feedFile(fp,sender,*rcpts)
|
||||
|
||||
## Call the connect and helo callbacks.
|
||||
# The helo callback is not called if connect does not return CONTINUE.
|
||||
# @param host the hostname passed to the connect callback
|
||||
# @param helo the hostname passed to the helo callback
|
||||
# @param ip the IP address passed to the connect callback
|
||||
def connect(self,host='localhost',helo='spamrelay',ip='1.2.3.4'):
|
||||
self._body = None
|
||||
self._bodyreplaced = False
|
||||
opts = [ Milter.CURR_ACTS,~0,0,0 ]
|
||||
rc = self.negotiate(opts)
|
||||
rc = super(TestBase,self).connect(host,1,(ip,1234))
|
||||
if rc != Milter.CONTINUE:
|
||||
self.close()
|
||||
return rc
|
||||
rc = self.hello(helo)
|
||||
if rc != Milter.CONTINUE:
|
||||
self.close()
|
||||
return rc
|
||||
+7
-7
@@ -84,14 +84,14 @@ def iniplist(ipaddr,iplist):
|
||||
p = pat.split('/',1)
|
||||
if ip4re.match(p[0]):
|
||||
if len(p) > 1:
|
||||
n = int(p[1])
|
||||
n = int(p[1])
|
||||
else:
|
||||
n = 32
|
||||
if cidr(addr2bin(p[0]),n) == cidr(ipnum,n):
|
||||
return True
|
||||
elif ip6re.match(p[0]):
|
||||
if len(p) > 1:
|
||||
n = int(p[1])
|
||||
n = int(p[1])
|
||||
else:
|
||||
n = 128
|
||||
if cidr(bin2long6(inet_pton(p[0])),n,MASK6) == cidr(ipnum,n,MASK6):
|
||||
@@ -185,15 +185,15 @@ def parse_header(val):
|
||||
for s,enc in h:
|
||||
if enc:
|
||||
try:
|
||||
u.append(unicode(s,enc,'replace'))
|
||||
except LookupError:
|
||||
u.append(unicode(s))
|
||||
u.append(unicode(s,enc,'replace'))
|
||||
except LookupError:
|
||||
u.append(unicode(s))
|
||||
else:
|
||||
u.append(unicode(s))
|
||||
u.append(unicode(s))
|
||||
u = ''.join(u)
|
||||
for enc in ('us-ascii','iso-8859-1','utf8'):
|
||||
try:
|
||||
return u.encode(enc)
|
||||
return u.encode(enc)
|
||||
except UnicodeError: continue
|
||||
except UnicodeDecodeError: pass
|
||||
except LookupError: pass
|
||||
|
||||
@@ -135,6 +135,8 @@ http://www.bmsi.com/linux/sendmail-rh72.spec
|
||||
IPv6 Notes
|
||||
----------
|
||||
|
||||
IPv6 is still experimental.
|
||||
|
||||
The IPv6 protocol is supported if your operation system supports it
|
||||
and if sendmail was compiled with IPv6 support. To determine if your
|
||||
sendmail supports IPv6, run "sendmail -d0" and check for the NETINET6
|
||||
@@ -192,7 +194,7 @@ Authors
|
||||
-------
|
||||
|
||||
Jim Niemira (urmane@urmane.org) wrote the original C module and some quick
|
||||
and dirty python to use it. Stuart D. Gathman (stuart@gathman.org) took that
|
||||
and dirty python to use it. Stuart D. Gathman (stuart@bmsi.com) took that
|
||||
kludge and added threading and context objects to it, wrote a proper OO
|
||||
wrapper (Milter.py) that handles attachments, did lots of testing, packaged
|
||||
it with distutils, and generally transformed it from a quick hack to a
|
||||
|
||||
@@ -1,2 +1,6 @@
|
||||
Support smfi_negotiate and auto negotiate only those callbacks for which
|
||||
Milter.Milter methods have been overridden. (Python should be able to
|
||||
do that.)
|
||||
|
||||
Lookup exact RFC syntax of real name / email and make
|
||||
Milter.utils.parse_addr() pass all unit tests.
|
||||
|
||||
+11
-14
@@ -7,32 +7,30 @@
|
||||
# events include the initial connection from a MTA, the envelope sender and
|
||||
# recipients, the top level mail headers, and the message body. There are
|
||||
# options to mangle all of these components of the message as it passes through
|
||||
# the %milter.
|
||||
# the milter.
|
||||
#
|
||||
# At the next level, the <code>Milter</code> module (note the case difference)
|
||||
# provides a Python friendly object oriented wrapper for the low level API. To
|
||||
# use the Milter module, an application registers a 'factory' to create an
|
||||
# object for each connection from a MTA to sendmail. These connection objects
|
||||
# must provide methods corresponding to the libmilter event callbacks.
|
||||
# must provide methods corresponding to the libmilter callback events.
|
||||
#
|
||||
# Each callback method returns a code to tell sendmail whether to proceed with
|
||||
# Each event method returns a code to tell sendmail whether to proceed with
|
||||
# processing the message. This is a big advantage of milters over other mail
|
||||
# filtering systems. Unwanted mail can be stopped in its tracks at the
|
||||
# earliest possible point. The callback return codes are
|
||||
# milter.CONTINUE, milter.REJECT, milter.DISCARD, milter.ACCEPT,
|
||||
# milter.TEMPFAIL, milter.SKIP, milter.NOREPLY.
|
||||
# earliest possible point.
|
||||
#
|
||||
# The Milter.Base class provides default implementations for
|
||||
# The <code>Milter.Base</code> class provides default implementations for
|
||||
# event methods that do nothing, and also provides wrappers for the libmilter
|
||||
# methods to mutate the message. It automatically negotiates with MTA
|
||||
# which protocol steps need to be processed by the %milter, based on
|
||||
# which protocol steps need to be processed by the milter, based on
|
||||
# which callback methods are overridden.
|
||||
#
|
||||
# The Milter.Milter class provides an alternate default
|
||||
# implementation that logs the main milter callbacks, but otherwise does
|
||||
# nothing. It is provided for compatibility.
|
||||
# The <code>Milter.Milter</code> class provides an alternate default
|
||||
# implementation that logs the main milter events, but otherwise does nothing.
|
||||
# It is provided for compatibility.
|
||||
#
|
||||
# The mime module provides a wrapper for the Python email package
|
||||
# The <code>mime</code> module provides a wrapper for the Python email package
|
||||
# that fixes some bugs, and simplifies modifying selected parts of a MIME
|
||||
# message.
|
||||
#
|
||||
@@ -52,5 +50,4 @@
|
||||
# You may find the
|
||||
# <a href="http://docs.python.org/release/2.6.6/library/multiprocessing.html">
|
||||
# multiprocessing</a> module useful. It can be a drop-in
|
||||
# replacement for threading as illustrated in
|
||||
# <a href="milter-template_8py-example.html">milter-template.py</a>.
|
||||
# replacement for threading as illustrated in @ref milter-template.py.
|
||||
|
||||
+19
-103
@@ -3,65 +3,10 @@
|
||||
|
||||
## @package milter
|
||||
#
|
||||
# A thin wrapper around libmilter. Most users will not import
|
||||
# milter directly, but will instead import Milter and subclass
|
||||
# Milter.Base. This module gives you ultimate low level control
|
||||
# from python.
|
||||
# A thin wrapper around libmilter.
|
||||
#
|
||||
|
||||
## Continue processing the current connection, message, or recipient.
|
||||
CONTINUE = 0
|
||||
## For a connection-oriented routine, reject this connection;
|
||||
# call Milter.Base.close(). For a message-oriented routine, except
|
||||
# Milter.Base.eom() or Milter.Base.abort(), reject this message. For a
|
||||
# recipient-oriented routine, reject the current recipient (but continue
|
||||
# processing the current message).
|
||||
REJECT = 1
|
||||
|
||||
## For a message- or recipient-oriented routine, accept this message, but
|
||||
# silently discard it. SMFIS_DISCARD should not be returned by a
|
||||
# connection-oriented routine.
|
||||
DISCARD = 2
|
||||
|
||||
## For a connection-oriented routine, accept this connection without further
|
||||
# filter processing; call Milter.Base.close(). For a message- or
|
||||
# recipient-oriented routine, accept this message without further filtering.
|
||||
ACCEPT = 3
|
||||
|
||||
## Return a temporary failure, i.e., the corresponding SMTP command will return
|
||||
# an appropriate 4xx status code. For a message-oriented routine, except
|
||||
# Milter.Base.envfrom(), fail for this message. For a connection-oriented
|
||||
# routine, fail for this connection; call Milter.Base.close(). For a recipient-oriented
|
||||
# routine, only
|
||||
# fail for the current recipient; continue message processing.
|
||||
TEMPFAIL = 4
|
||||
|
||||
## Skip further callbacks of the same type in this transaction.
|
||||
# Currently this return value is only allowed in Milter.Base.body(). It can be
|
||||
# used if a %milter has received sufficiently many body chunks to make a
|
||||
# decision, but still wants to invoke message modification functions that are
|
||||
# only allowed to be called from Milter.Base.eom(). Note: the %milter must
|
||||
# negotiate this behavior with the MTA, i.e., it must check whether the
|
||||
# protocol action SMFIP_SKIP is available and if so, the %milter must request
|
||||
# it.
|
||||
SKIP = 5
|
||||
|
||||
## Do not send a reply back to the MTA.
|
||||
# The %milter must negotiate this behavior with the MTA, i.e., it must check
|
||||
# whether the appropriate protocol action P_NR_* is available and if so,
|
||||
# the %milter must request it. If you set the P_NR_* protocol action for a
|
||||
# callback, that callback must always reply with NOREPLY. Using any other
|
||||
# reply code is a violation of the API. If in some cases your callback may
|
||||
# return another value (e.g., due to some resource shortages), then you must
|
||||
# not set P_NR_* and you must use CONTINUE as the default return
|
||||
# code. (Alternatively you can try to delay reporting the problem to a later
|
||||
# callback for which P_NR_* is not set.)
|
||||
#
|
||||
# This is negotiated and returned automatically by the Milter.noreply
|
||||
# function decorator.
|
||||
NOREPLY = 6
|
||||
|
||||
## Hold context for a %milter connection.
|
||||
## Hold context for a milter connection.
|
||||
# Each connection to sendmail creates a new <code>SMFICTX</code> struct within
|
||||
# libmilter. The milter module in turn creates a milterContext
|
||||
# tied to the <code>SMFICTX</code> struct via <code>smfi_setpriv</code>
|
||||
@@ -109,18 +54,12 @@ class milterContext(object):
|
||||
def chgfrom(self,sender,param=None): pass
|
||||
## Tell the MTA which macro values we are interested in for a given stage.
|
||||
# Of interest only when you need to squeeze a few more bytes of bandwidth.
|
||||
# It may only be called from the negotiate callback.
|
||||
# The protocol stages are
|
||||
# M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT, M_DATA, M_EOM, M_EOH.
|
||||
# Calls <a href="https://www.milter.org/developers/api/smfi_setsymlist">smfi_setsymlist</a>.
|
||||
# @param stage protocol stage in which the macro list should be used
|
||||
# @param macrolist a space separated list of macro names
|
||||
def setsymlist(self,stage,macrolist): pass
|
||||
def setsmlist(self,stage,macrolist): pass
|
||||
|
||||
class error(Exception): pass
|
||||
|
||||
## Enable optional %milter actions.
|
||||
# Certain %milter actions need to be enabled before calling main()
|
||||
## Enable optional milter actions.
|
||||
# Certain milter actions need to be enabled before calling main()
|
||||
# or they throw an exception. Pymilter enables them all by
|
||||
# default (since 0.9.2), but you may wish to disable unneeded
|
||||
# actions as an optimization.
|
||||
@@ -138,27 +77,19 @@ def set_abort_callback(cb): pass
|
||||
def set_close_callback(cb): pass
|
||||
|
||||
## Sets the return code for untrapped Python exceptions during a callback.
|
||||
# The default is TEMPFAIL. You should not depend on this handler. Your
|
||||
# application should have its own top level exception handler for each
|
||||
# callback. You can then choose your own reply message, log the stack track
|
||||
# were you please, and so on. However, if you miss one, this last ditch
|
||||
# handler will print a standard stack trace to sys.stderr, and return to
|
||||
# sendmail.
|
||||
# @param code one of #TEMPFAIL,#REJECT,#CONTINUE, or since 1.0, #ACCEPT
|
||||
# Must be one of TEMPFAIL,REJECT,CONTINUE
|
||||
def set_exception_policy(code): pass
|
||||
|
||||
## Register python %milter with libmilter.
|
||||
# The name we pass is used to identify the %milter in the MTA configuration.
|
||||
## Register python milter with libmilter.
|
||||
# The name we pass is used to identify the milter in the MTA configuration.
|
||||
# Callback functions must be set using the set_*_callback() functions before
|
||||
# registering the %milter.
|
||||
# registering the milter.
|
||||
# Three additional callbacks are specified as keyword parameters. These
|
||||
# were added by recent versions of libmilter. The keyword parameters is
|
||||
# a nicer way to do it, I think, since it makes clear that you have to do
|
||||
# it before registering. I may move all the callbacks in the future (perhaps
|
||||
# keeping the set functions for compatibility). Note that Milter.Base
|
||||
# automatically maps all callbacks to member functions, and negotiates which
|
||||
# member functions are actually overridden by an application class.
|
||||
# @param name the %milter name by which the MTA finds us
|
||||
# it before registering. I may move all the callbacks
|
||||
# in the future (perhaps keeping the set functions for compatibility).
|
||||
# @param name the milter name by which the MTA finds us
|
||||
# @param negotiate the
|
||||
# <a href="https://www.milter.org/developers/api/xxfi_negotiate">
|
||||
# xxfi_negotiate</a> callback, called to negotiate supported
|
||||
@@ -172,16 +103,6 @@ def set_exception_policy(code): pass
|
||||
# xxfi_data</a> callback, called when the DATA
|
||||
# SMTP command is received.
|
||||
def register(name,negotiate=None,unknown=None,data=None): pass
|
||||
|
||||
## Attempt to create the socket used to communicate with the MTA.
|
||||
# milter.opensocket() attempts to create the socket specified previously by a
|
||||
# call to milter.setconn() which will be the interface between MTAs and the
|
||||
# %milter. This allows the calling application to ensure that the socket can be
|
||||
# created. If this is not called, milter.main() will do so implicitly.
|
||||
# Calls <a href="https://www.milter.org/developers/api/smfi_opensocket">
|
||||
# smfi_opensocket</a>. While not documented for libmilter, my experiments
|
||||
# indicate that you must call register() before calling opensocket().
|
||||
# @param rmsock Try to remove an existing unix domain socket if true.
|
||||
def opensocket(rmsock): pass
|
||||
|
||||
## Transfer control to libmilter.
|
||||
@@ -191,7 +112,7 @@ def main(): pass
|
||||
|
||||
## Set the libmilter debugging level.
|
||||
# <a href="https://www.milter.org/developers/api/smfi_setdbg">smfi_setdbg</a>
|
||||
# sets the %milter library's internal debugging level to a new level
|
||||
# sets the milter library's internal debugging level to a new level
|
||||
# so that code details may be traced. A level of zero turns off debugging. The
|
||||
# greater (more positive) the level the more detailed the debugging. Six is the
|
||||
# current, highest, useful value. Must be called before calling main().
|
||||
@@ -212,21 +133,16 @@ def setbacklog(n): pass
|
||||
# unix, inet, or inet6 socket. By default, a unix domain socket
|
||||
# is used. It must not exist,
|
||||
# and sendmail will throw warnings if, eg, the file is under a
|
||||
# group or world writable directory. milter.setconn() will not fail with
|
||||
# an invalid socket - this will be detected only when calling milter.main()
|
||||
# or milter.opensocket().
|
||||
# @param s the socket address in proto:address format
|
||||
# group or world writable directory.
|
||||
# <pre>
|
||||
# milter.setconn('unix:/var/run/pythonfilter') # a named pipe
|
||||
# milter.setconn('local:/var/run/pythonfilter') # a named pipe
|
||||
# milter.setconn('inet:8800') # listen on ANY interface
|
||||
# milter.setconn('inet:7871@@publichost') # listen on a specific interface
|
||||
# milter.setconn('inet6:8020')
|
||||
# milter.setconn('inet6:8020@[2001:db8:1234::1]') # listen on specific IP
|
||||
# setconn('unix:/var/run/pythonfilter')
|
||||
# setconn('inet:8800') # listen on ANY interface
|
||||
# setconn('inet:7871@@publichost') # listen on a specific interface
|
||||
# setconn('inet6:8020')
|
||||
# </pre>
|
||||
def setconn(s): pass
|
||||
|
||||
## Stop the %milter gracefully.
|
||||
## Stop the milter gracefully.
|
||||
def stop(): pass
|
||||
|
||||
## Retrieve diagnostic info.
|
||||
|
||||
@@ -1,10 +1,9 @@
|
||||
web:
|
||||
doxygen
|
||||
cd doc/html; zip -r ../../doc .
|
||||
rsync -ravK doc/html/ spidey2.bmsi.com:/Public/pymilter
|
||||
|
||||
VERSION=1.0
|
||||
CVSTAG=pymilter-1_0
|
||||
VERSION=0.9.6
|
||||
CVSTAG=pymilter-0_9_6
|
||||
PKG=pymilter-$(VERSION)
|
||||
SRCTAR=$(PKG).tar.gz
|
||||
|
||||
|
||||
+1
-3
@@ -71,9 +71,6 @@ class myMilter(Milter.Base):
|
||||
self.fromparms = Milter.dictfromlist(str) # ESMTP parms
|
||||
self.user = self.getsymval('{auth_authen}') # authenticated user
|
||||
self.log("mail from:", mailfrom, *str)
|
||||
# NOTE: self.fp is only an *internal* copy of message data. You
|
||||
# must use addheader, chgheader, replacebody to change the message
|
||||
# on the MTA.
|
||||
self.fp = StringIO.StringIO()
|
||||
self.canon_from = '@'.join(parse_addr(mailfrom))
|
||||
self.fp.write('From %s %s\n' % (self.canon_from,time.ctime()))
|
||||
@@ -107,6 +104,7 @@ class myMilter(Milter.Base):
|
||||
def eom(self):
|
||||
self.fp.seek(0)
|
||||
msg = email.message_from_file(self.fp)
|
||||
self.setreply('250','2.5.1','Grokked by pymilter')
|
||||
# many milter functions can only be called from eom()
|
||||
# example of adding a Bcc:
|
||||
self.addrcpt('<%s>' % 'spy@example.com')
|
||||
|
||||
+49
-80
@@ -35,22 +35,6 @@ $ python setup.py help
|
||||
libraries=["milter","smutil","resolv"]
|
||||
|
||||
* $Log$
|
||||
* Revision 1.35 2013/03/14 22:11:25 customdesigned
|
||||
* Release 0.9.8
|
||||
*
|
||||
* Revision 1.34 2013/03/09 05:42:14 customdesigned
|
||||
* Make TestBase members private, fix getsymlist misspelling.
|
||||
*
|
||||
* Revision 1.33 2013/03/09 00:25:23 customdesigned
|
||||
* Better untrapped exception message. const char for doc comments.
|
||||
*
|
||||
* Revision 1.32 2013/01/13 01:46:16 customdesigned
|
||||
* Doc updates.
|
||||
*
|
||||
* Revision 1.31 2012/04/12 23:32:50 customdesigned
|
||||
* Replace redundant callback array with macros. If this doesn't break anything,
|
||||
* macros can be eliminated with code changes.
|
||||
*
|
||||
* Revision 1.30 2012/04/12 23:08:06 customdesigned
|
||||
* Support RFC2553 on BSD
|
||||
*
|
||||
@@ -452,7 +436,7 @@ _thread_return(PyThreadState *t,int val,char *errstr) {
|
||||
return _generic_return(val,errstr);
|
||||
}
|
||||
|
||||
static const char milter_set_flags__doc__[] =
|
||||
static char milter_set_flags__doc__[] =
|
||||
"set_flags(int) -> None\n\
|
||||
Set flags for filter capabilities; OR of one or more of:\n\
|
||||
ADDHDRS - filter may add headers\n\
|
||||
@@ -493,7 +477,7 @@ generic_set_callback(PyObject *args,char *t,PyObject **cb) {
|
||||
return Py_None;
|
||||
}
|
||||
|
||||
static const char milter_set_connect_callback__doc__[] =
|
||||
static char milter_set_connect_callback__doc__[] =
|
||||
"set_connect_callback(Function) -> None\n\
|
||||
Sets the Python function invoked when a connection is made to sendmail.\n\
|
||||
Function takes args (ctx, hostname, integer, hostaddr) -> int\n\
|
||||
@@ -520,7 +504,7 @@ milter_set_connect_callback(PyObject *self, PyObject *args) {
|
||||
"O:set_connect_callback", &connect_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_helo_callback__doc__[] =
|
||||
static char milter_set_helo_callback__doc__[] =
|
||||
"set_helo_callback(Function) -> None\n\
|
||||
Sets the Python function invoked upon SMTP HELO.\n\
|
||||
Function takes args (ctx, hostname) -> int\n\
|
||||
@@ -531,7 +515,7 @@ milter_set_helo_callback(PyObject *self, PyObject *args) {
|
||||
return generic_set_callback(args, "O:set_helo_callback", &helo_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_envfrom_callback__doc__[] =
|
||||
static char milter_set_envfrom_callback__doc__[] =
|
||||
"set_envfrom_callback(Function) -> None\n\
|
||||
Sets the Python function invoked on envelope from.\n\
|
||||
Function takes args (ctx, from, *str) -> int\n\
|
||||
@@ -544,7 +528,7 @@ milter_set_envfrom_callback(PyObject *self, PyObject *args) {
|
||||
&envfrom_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_envrcpt_callback__doc__[] =
|
||||
static char milter_set_envrcpt_callback__doc__[] =
|
||||
"set_envrcpt_callback(Function) -> None\n\
|
||||
Sets the Python function invoked on each envelope recipient.\n\
|
||||
Function takes args (ctx, rcpt, *str) -> int\n\
|
||||
@@ -557,7 +541,7 @@ milter_set_envrcpt_callback(PyObject *self, PyObject *args) {
|
||||
&envrcpt_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_header_callback__doc__[] =
|
||||
static char milter_set_header_callback__doc__[] =
|
||||
"set_header_callback(Function) -> None\n\
|
||||
Sets the Python function invoked on each message header.\n\
|
||||
Function takes args (ctx, field, value) ->int\n\
|
||||
@@ -570,7 +554,7 @@ milter_set_header_callback(PyObject *self, PyObject *args) {
|
||||
&header_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_eoh_callback__doc__[] =
|
||||
static char milter_set_eoh_callback__doc__[] =
|
||||
"set_eoh_callback(Function) -> None\n\
|
||||
Sets the Python function invoked at end of header.\n\
|
||||
Function takes args (ctx) -> int";
|
||||
@@ -580,7 +564,7 @@ milter_set_eoh_callback(PyObject *self, PyObject *args) {
|
||||
return generic_set_callback(args, "O:set_eoh_callback", &eoh_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_body_callback__doc__[] =
|
||||
static char milter_set_body_callback__doc__[] =
|
||||
"set_body_callback(Function) -> None\n\
|
||||
Sets the Python function invoked for each body chunk. There may\n\
|
||||
be multiple body chunks passed to the filter. End-of-lines are\n\
|
||||
@@ -593,7 +577,7 @@ milter_set_body_callback(PyObject *self, PyObject *args) {
|
||||
return generic_set_callback(args, "O:set_body_callback", &body_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_eom_callback__doc__[] =
|
||||
static char milter_set_eom_callback__doc__[] =
|
||||
"set_eom_callback(Function) -> None\n\
|
||||
Sets the Python function invoked at end of message.\n\
|
||||
This routine is the only place where special operations\n\
|
||||
@@ -606,7 +590,7 @@ milter_set_eom_callback(PyObject *self, PyObject *args) {
|
||||
return generic_set_callback(args, "O:set_eom_callback", &eom_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_abort_callback__doc__[] =
|
||||
static char milter_set_abort_callback__doc__[] =
|
||||
"set_abort_callback(Function) -> None\n\
|
||||
Sets the Python function invoked if message is aborted\n\
|
||||
outside of the control of the filter, for example,\n\
|
||||
@@ -620,7 +604,7 @@ milter_set_abort_callback(PyObject *self, PyObject *args) {
|
||||
return generic_set_callback(args, "O:set_abort_callback", &abort_callback);
|
||||
}
|
||||
|
||||
static const char milter_set_close_callback__doc__[] =
|
||||
static char milter_set_close_callback__doc__[] =
|
||||
"set_close_callback(Function) -> None\n\
|
||||
Sets the Python function invoked at end of the connection. This\n\
|
||||
is called on close even if the previous mail transaction was aborted.\n\
|
||||
@@ -633,7 +617,7 @@ milter_set_close_callback(PyObject *self, PyObject *args) {
|
||||
|
||||
static int exception_policy = SMFIS_TEMPFAIL;
|
||||
|
||||
static const char milter_set_exception_policy__doc__[] =
|
||||
static char milter_set_exception_policy__doc__[] =
|
||||
"set_exception_policy(i) -> None\n\
|
||||
Sets the policy for untrapped Python exceptions during a callback.\n\
|
||||
Must be one of TEMPFAIL,REJECT,CONTINUE";
|
||||
@@ -644,8 +628,7 @@ milter_set_exception_policy(PyObject *self, PyObject *args) {
|
||||
if (!PyArg_ParseTuple(args, "i:set_exception_policy", &i))
|
||||
return NULL;
|
||||
switch (i) {
|
||||
case SMFIS_REJECT: case SMFIS_TEMPFAIL:
|
||||
case SMFIS_CONTINUE: case SMFIS_ACCEPT:
|
||||
case SMFIS_REJECT: case SMFIS_TEMPFAIL: case SMFIS_CONTINUE:
|
||||
exception_policy = i;
|
||||
Py_INCREF(Py_None);
|
||||
return Py_None;
|
||||
@@ -660,30 +643,23 @@ _release_thread(PyThreadState *t) {
|
||||
PyEval_ReleaseThread(t);
|
||||
}
|
||||
|
||||
|
||||
/** Report and clear any python exception before returning to libmilter.
|
||||
The interpreter is locked when we are called, and we unlock it. */
|
||||
static int _report_exception(milter_ContextObject *self) {
|
||||
char untrapped_msg[80];
|
||||
if (PyErr_Occurred()) {
|
||||
sprintf(untrapped_msg,"pymilter: untrapped exception in %.40s",
|
||||
description.xxfi_name);
|
||||
PyErr_Print();
|
||||
PyErr_Clear(); /* must clear since not returning to python */
|
||||
_release_thread(self->t);
|
||||
switch (exception_policy) {
|
||||
case SMFIS_REJECT:
|
||||
smfi_setreply(self->ctx, "554", "5.3.0", untrapped_msg);
|
||||
smfi_setreply(self->ctx, "554", "5.3.0", "Filter failure");
|
||||
return SMFIS_REJECT;
|
||||
case SMFIS_TEMPFAIL:
|
||||
smfi_setreply(self->ctx, "451", "4.3.0", untrapped_msg);
|
||||
smfi_setreply(self->ctx, "451", "4.3.0", "Filter failure");
|
||||
return SMFIS_TEMPFAIL;
|
||||
}
|
||||
return exception_policy;
|
||||
return SMFIS_CONTINUE;
|
||||
}
|
||||
/* This should never happen, _report_exception is only called when
|
||||
* the caller has already detected a python exception. If it
|
||||
* does somehow happen, pretend nothing is wrong... */
|
||||
_release_thread(self->t);
|
||||
return SMFIS_CONTINUE;
|
||||
}
|
||||
@@ -1011,7 +987,7 @@ milter_wrap_close(SMFICTX *ctx) {
|
||||
return r;
|
||||
}
|
||||
|
||||
static const char milter_register__doc__[] =
|
||||
static char milter_register__doc__[] =
|
||||
"register(name,unknown=,data=,negotiate=) -> None\n\
|
||||
Registers the milter name with current callbacks, and flags.\n\
|
||||
Required before main() is called.";
|
||||
@@ -1056,7 +1032,7 @@ milter_register(PyObject *self, PyObject *args, PyObject *kwds) {
|
||||
return _generic_return(smfi_register(description), "cannot register");
|
||||
}
|
||||
|
||||
static const char milter_opensocket__doc__[] =
|
||||
static char milter_opensocket__doc__[] =
|
||||
"opensocket(rmsock) -> None\n\
|
||||
Attempts to create and open the socket provided with setconn.\n\
|
||||
Removes the socket first if rmsock is True.";
|
||||
@@ -1069,7 +1045,7 @@ milter_opensocket(PyObject *self, PyObject *args) {
|
||||
return _generic_return(smfi_opensocket(rmsock), "cannot opensocket");
|
||||
}
|
||||
|
||||
static const char milter_main__doc__[] =
|
||||
static char milter_main__doc__[] =
|
||||
"main() -> None\n\
|
||||
Main milter routine. Set any callbacks, and flags desired, then call\n\
|
||||
setconn(), then call register(name), and finally call main().";
|
||||
@@ -1093,7 +1069,7 @@ milter_main(PyObject *self, PyObject *args) {
|
||||
return o;
|
||||
}
|
||||
|
||||
static const char milter_setdbg__doc__[] =
|
||||
static char milter_setdbg__doc__[] =
|
||||
"setdbg(int) -> None\n\
|
||||
Sets debug level in sendmail/libmilter source. Dubious usefulness.";
|
||||
|
||||
@@ -1104,7 +1080,7 @@ milter_setdbg(PyObject *self, PyObject *args) {
|
||||
return _generic_return(smfi_setdbg(val), "cannot set debug value");
|
||||
}
|
||||
|
||||
static const char milter_setbacklog__doc__[] =
|
||||
static char milter_setbacklog__doc__[] =
|
||||
"setbacklog(int) -> None\n\
|
||||
Set the TCP connection queue size for the milter socket.";
|
||||
|
||||
@@ -1116,7 +1092,7 @@ milter_setbacklog(PyObject *self, PyObject *args) {
|
||||
return _generic_return(smfi_setbacklog(val), "cannot set backlog");
|
||||
}
|
||||
|
||||
static const char milter_settimeout__doc__[] =
|
||||
static char milter_settimeout__doc__[] =
|
||||
"settimeout(int) -> None\n\
|
||||
Set the time (in seconds) that sendmail will wait before\n\
|
||||
considering this filter dead.";
|
||||
@@ -1129,7 +1105,7 @@ milter_settimeout(PyObject *self, PyObject *args) {
|
||||
return _generic_return(smfi_settimeout(val), "cannot set timeout");
|
||||
}
|
||||
|
||||
static const char milter_setconn__doc__[] =
|
||||
static char milter_setconn__doc__[] =
|
||||
"setconn(filename) -> None\n\
|
||||
Sets the pathname to the unix, inet, or inet6 socket that\n\
|
||||
sendmail will use to communicate with this filter. By default,\n\
|
||||
@@ -1149,7 +1125,7 @@ milter_setconn(PyObject *self, PyObject *args) {
|
||||
return _generic_return(smfi_setconn(str), "cannot set connection");
|
||||
}
|
||||
|
||||
static const char milter_stop__doc__[] =
|
||||
static char milter_stop__doc__[] =
|
||||
"stop() -> None\n\
|
||||
This function appears to be a controlled method to tell sendmail to\n\
|
||||
stop using this filter. It will close the socket.";
|
||||
@@ -1162,7 +1138,7 @@ milter_stop(PyObject *self, PyObject *args) {
|
||||
return _thread_return(t,smfi_stop(), "cannot stop");
|
||||
}
|
||||
|
||||
static const char milter_getdiag__doc__[] =
|
||||
static char milter_getdiag__doc__[] =
|
||||
"getdiag() -> tuple\n\
|
||||
Return a tuple of diagnostic data. The first two items are context new\n\
|
||||
count and context del count. The rest are yet to be defined.";
|
||||
@@ -1172,7 +1148,7 @@ milter_getdiag(PyObject *self, PyObject *args) {
|
||||
return Py_BuildValue("(kk)", diag.contextNew,diag.contextDel);
|
||||
}
|
||||
|
||||
static const char milter_getversion__doc__[] =
|
||||
static char milter_getversion__doc__[] =
|
||||
"getversion() -> tuple\n\
|
||||
Return runtime libmilter version as a tuple of major,minor,patchlevel.";
|
||||
static PyObject *
|
||||
@@ -1186,7 +1162,7 @@ milter_getversion(PyObject *self, PyObject *args) {
|
||||
return Py_BuildValue("(kkk)", major,minor,patch);
|
||||
}
|
||||
|
||||
static const char milter_getsymval__doc__[] =
|
||||
static char milter_getsymval__doc__[] =
|
||||
"getsymval(String) -> String\n\
|
||||
Returns a symbol's value. Context-dependent, and unclear from the dox.";
|
||||
|
||||
@@ -1201,7 +1177,7 @@ milter_getsymval(PyObject *self, PyObject *args) {
|
||||
return Py_BuildValue("s", smfi_getsymval(ctx, str));
|
||||
}
|
||||
|
||||
static const char milter_setreply__doc__[] =
|
||||
static char milter_setreply__doc__[] =
|
||||
"setreply(rcode, xcode, message) -> None\n\
|
||||
Sets the specific reply code to be used in response\n\
|
||||
to the active command.\n\
|
||||
@@ -1265,7 +1241,7 @@ milter_setreply(PyObject *self, PyObject *args) {
|
||||
"cannot set reply");
|
||||
}
|
||||
|
||||
static const char milter_addheader__doc__[] =
|
||||
static char milter_addheader__doc__[] =
|
||||
"addheader(field, value, idx=-1) -> None\n\
|
||||
Add a header to the message. This header is not passed to other\n\
|
||||
filters. It is not checked for standards compliance;\n\
|
||||
@@ -1302,7 +1278,7 @@ milter_addheader(PyObject *self, PyObject *args) {
|
||||
}
|
||||
|
||||
#ifdef SMFIF_CHGFROM
|
||||
static const char milter_chgfrom__doc__[] =
|
||||
static char milter_chgfrom__doc__[] =
|
||||
"chgfrom(sender,params) -> None\n\
|
||||
Change the envelope sender (MAIL From) of the current message.\n\
|
||||
A filter which calls smfi_chgfrom must have set the CHGFROM flag\n\
|
||||
@@ -1325,7 +1301,7 @@ milter_chgfrom(PyObject *self, PyObject *args) {
|
||||
}
|
||||
#endif
|
||||
|
||||
static const char milter_chgheader__doc__[] =
|
||||
static char milter_chgheader__doc__[] =
|
||||
"chgheader(field, int, value) -> None\n\
|
||||
Change/delete a header in the message. \n\
|
||||
It is not checked for standards compliance; the mail filter\n\
|
||||
@@ -1353,7 +1329,7 @@ milter_chgheader(PyObject *self, PyObject *args) {
|
||||
"cannot change header");
|
||||
}
|
||||
|
||||
static const char milter_addrcpt__doc__[] =
|
||||
static char milter_addrcpt__doc__[] =
|
||||
"addrcpt(string,params=None) -> None\n\
|
||||
Add a recipient to the envelope. It must be in the same format\n\
|
||||
as is passed to the envrcpt callback in the first tuple element.\n\
|
||||
@@ -1383,7 +1359,7 @@ milter_addrcpt(PyObject *self, PyObject *args) {
|
||||
return _thread_return(t,rc, "cannot add recipient");
|
||||
}
|
||||
|
||||
static const char milter_delrcpt__doc__[] =
|
||||
static char milter_delrcpt__doc__[] =
|
||||
"delrcpt(string) -> None\n\
|
||||
Delete a recipient from the envelope.\n\
|
||||
This function can only be called from the EOM callback.";
|
||||
@@ -1401,7 +1377,7 @@ milter_delrcpt(PyObject *self, PyObject *args) {
|
||||
return _thread_return(t,smfi_delrcpt(ctx, rcpt), "cannot delete recipient");
|
||||
}
|
||||
|
||||
static const char milter_replacebody__doc__[] =
|
||||
static char milter_replacebody__doc__[] =
|
||||
"replacebody(string) -> None\n\
|
||||
Replace the body of the message. This routine may be called multiple\n\
|
||||
times if the body is longer than convenient to send in one call. End of\n\
|
||||
@@ -1423,7 +1399,7 @@ milter_replacebody(PyObject *self, PyObject *args) {
|
||||
(unsigned char *)bodyp, bodylen), "cannot replace message body");
|
||||
}
|
||||
|
||||
static const char milter_setpriv__doc__[] =
|
||||
static char milter_setpriv__doc__[] =
|
||||
"setpriv(object) -> object\n\
|
||||
Associates any Python object with this context, and returns\n\
|
||||
the old value or None. Use this to\n\
|
||||
@@ -1449,7 +1425,7 @@ milter_setpriv(PyObject *self, PyObject *args) {
|
||||
return old;
|
||||
}
|
||||
|
||||
static const char milter_getpriv__doc__[] =
|
||||
static char milter_getpriv__doc__[] =
|
||||
"getpriv() -> None\n\
|
||||
Returns the Python object associated with the current context (if any).\n\
|
||||
Use this in conjunction with setpriv to keep track of data in a thread-safe\n\
|
||||
@@ -1467,7 +1443,7 @@ milter_getpriv(PyObject *self, PyObject *args) {
|
||||
}
|
||||
|
||||
#ifdef SMFIF_QUARANTINE
|
||||
static const char milter_quarantine__doc__[] =
|
||||
static char milter_quarantine__doc__[] =
|
||||
"quarantine(string) -> None\n\
|
||||
Place the message in quarantine. A string with a description of the reason\n\
|
||||
is the only argument.";
|
||||
@@ -1488,7 +1464,7 @@ milter_quarantine(PyObject *self, PyObject *args) {
|
||||
#endif
|
||||
|
||||
#ifdef SMFIR_PROGRESS
|
||||
static const char milter_progress__doc__[] =
|
||||
static char milter_progress__doc__[] =
|
||||
"progress() -> None\n\
|
||||
Notify the MTA that we are working on a message so it will reset timeouts.";
|
||||
|
||||
@@ -1505,23 +1481,23 @@ milter_progress(PyObject *self, PyObject *args) {
|
||||
}
|
||||
#endif
|
||||
|
||||
#ifdef SMFIF_SETSYMLIST
|
||||
static const char milter_setsymlist__doc__[] =
|
||||
"setsymlist(stage,macrolist) -> None\n\
|
||||
#ifdef SMFIF_SETSMLIST
|
||||
static char milter_setsmlist__doc__[] =
|
||||
"setsmlist(stage,macrolist) -> None\n\
|
||||
Tell the MTA which macro values we are interested in for a given stage";
|
||||
|
||||
static PyObject *
|
||||
milter_setsymlist(PyObject *self, PyObject *args) {
|
||||
milter_setsmlist(PyObject *self, PyObject *args) {
|
||||
SMFICTX *ctx;
|
||||
PyThreadState *t;
|
||||
int stage = 0;
|
||||
char *smlist = 0;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "is:setsymlist",&stage, &smlist)) return NULL;
|
||||
if (!PyArg_ParseTuple(args, "is:setsmlist",&stage, &smlist)) return NULL;
|
||||
ctx = _find_context(self);
|
||||
if (ctx == NULL) return NULL;
|
||||
t = PyEval_SaveThread();
|
||||
return _thread_return(t,smfi_setsymlist(ctx,stage,smlist),
|
||||
return _thread_return(t,smfi_setsmlist(ctx,stage,smlist),
|
||||
"cannot set macro list");
|
||||
}
|
||||
#endif
|
||||
@@ -1545,8 +1521,8 @@ static PyMethodDef context_methods[] = {
|
||||
#ifdef SMFIF_CHGFROM
|
||||
{ "chgfrom", milter_chgfrom, METH_VARARGS, milter_chgfrom__doc__},
|
||||
#endif
|
||||
#ifdef SMFIF_SETSYMLIST
|
||||
{ "setsymlist", milter_setsymlist, METH_VARARGS, milter_setsymlist__doc__},
|
||||
#ifdef SMFIF_SETSMLIST
|
||||
{ "setsmlist", milter_setsmlist, METH_VARARGS, milter_setsmlist__doc__},
|
||||
#endif
|
||||
{ NULL, NULL }
|
||||
};
|
||||
@@ -1627,7 +1603,7 @@ static PyTypeObject milter_ContextType = {
|
||||
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
||||
};
|
||||
|
||||
static const char milter_documentation[] =
|
||||
static char milter_documentation[] =
|
||||
"This module interfaces with Sendmail's libmilter functionality,\n\
|
||||
allowing one to write email filters directly in Python.\n\
|
||||
Libmilter is currently marked FFR, and needs to be explicitly installed.\n\
|
||||
@@ -1669,15 +1645,8 @@ initmilter(void) {
|
||||
#ifdef SMFIF_CHGFROM
|
||||
setitem(d,"CHGFROM",SMFIF_CHGFROM);
|
||||
#endif
|
||||
#ifdef SMFIF_SETSYMLIST
|
||||
setitem(d,"SETSYMLIST",SMFIF_SETSYMLIST);
|
||||
setitem(d,"M_CONNECT",SMFIM_CONNECT);/* connect */
|
||||
setitem(d,"M_HELO",SMFIM_HELO); /* HELO/EHLO */
|
||||
setitem(d,"M_ENVFROM",SMFIM_ENVFROM);/* MAIL From */
|
||||
setitem(d,"M_ENVRCPT",SMFIM_ENVRCPT);/* RCPT To */
|
||||
setitem(d,"M_DATA",SMFIM_DATA); /* DATA */
|
||||
setitem(d,"M_EOM",SMFIM_EOM); /* end of message (final dot) */
|
||||
setitem(d,"M_EOH",SMFIM_EOH); /* end of header */
|
||||
#ifdef SMFIF_SETSMLIST
|
||||
setitem(d,"SETSMLIST",SMFIF_SETSMLIST);
|
||||
#endif
|
||||
#ifdef SMFIS_ALL_OPTS
|
||||
setitem(d,"P_RCPT_REJ",SMFIP_RCPT_REJ);
|
||||
|
||||
@@ -1,7 +1,4 @@
|
||||
# $Log$
|
||||
# Revision 1.8 2011/11/05 15:51:03 customdesigned
|
||||
# New example
|
||||
#
|
||||
# Revision 1.7 2009/06/13 21:15:12 customdesigned
|
||||
# Doxygen updates.
|
||||
#
|
||||
@@ -130,24 +127,24 @@ class MimeGenerator(Generator):
|
||||
# full MIME type, then dispatch to self._handle_<maintype>(). If
|
||||
# that's missing too, then dispatch to self._writeBody().
|
||||
main = msg.get_content_maintype()
|
||||
if msg.is_multipart() and main.lower() != 'multipart':
|
||||
self._handle_multipart(msg)
|
||||
else:
|
||||
Generator._dispatch(self,msg)
|
||||
if msg.is_multipart() and main.lower() != 'multipart':
|
||||
self._handle_multipart(msg)
|
||||
else:
|
||||
Generator._dispatch(self,msg)
|
||||
|
||||
def unquote(s):
|
||||
"""Remove quotes from a string."""
|
||||
if len(s) > 1:
|
||||
if s.startswith('"'):
|
||||
if s.endswith('"'):
|
||||
if s.endswith('"'):
|
||||
s = s[1:-1]
|
||||
else: # remove garbage after trailing quote
|
||||
try: s = s[1:s[1:].index('"')+1]
|
||||
except:
|
||||
return s
|
||||
return s.replace('\\\\', '\\').replace('\\"', '"')
|
||||
else: # remove garbage after trailing quote
|
||||
try: s = s[1:s[1:].index('"')+1]
|
||||
except:
|
||||
return s
|
||||
return s.replace('\\\\', '\\').replace('\\"', '"')
|
||||
if s.startswith('<') and s.endswith('>'):
|
||||
return s[1:-1]
|
||||
return s[1:-1]
|
||||
return s
|
||||
|
||||
from types import TupleType
|
||||
@@ -205,21 +202,21 @@ class MimeMessage(Message):
|
||||
for attr,val in self._get_params_preserve([],'content-type'):
|
||||
if isinstance(val, TupleType):
|
||||
# It's an RFC 2231 encoded parameter
|
||||
newvalue = _unquotevalue(val)
|
||||
if val[0]:
|
||||
val = unicode(newvalue[2], newvalue[0])
|
||||
else:
|
||||
val = unicode(newvalue[2])
|
||||
newvalue = _unquotevalue(val)
|
||||
if val[0]:
|
||||
val = unicode(newvalue[2], newvalue[0])
|
||||
else:
|
||||
val = unicode(newvalue[2])
|
||||
else:
|
||||
val = _unquotevalue(val.strip())
|
||||
val = _unquotevalue(val.strip())
|
||||
names.append((attr,val))
|
||||
names += [("filename",self.get_filename())]
|
||||
if scan_zip:
|
||||
for key,name in tuple(names): # copy by converting to tuple
|
||||
if name and name.lower().endswith('.zip'):
|
||||
txt = self.get_payload(decode=True)
|
||||
if txt.strip():
|
||||
names += zipnames(txt)
|
||||
if name and name.lower().endswith('.zip'):
|
||||
txt = self.get_payload(decode=True)
|
||||
if txt.strip():
|
||||
names += zipnames(txt)
|
||||
return names
|
||||
|
||||
def ismodified(self):
|
||||
@@ -290,13 +287,13 @@ class MimeMessage(Message):
|
||||
if t == 'message/rfc822' or t.startswith('multipart/'):
|
||||
if not self.submsg:
|
||||
txt = self.get_payload()
|
||||
if type(txt) == str:
|
||||
txt = self.get_payload(decode=True)
|
||||
self.submsg = email.message_from_string(txt,MimeMessage)
|
||||
for part in self.submsg.walk():
|
||||
part.modified = False
|
||||
else:
|
||||
self.submsg = txt[0]
|
||||
if type(txt) == str:
|
||||
txt = self.get_payload(decode=True)
|
||||
self.submsg = email.message_from_string(txt,MimeMessage)
|
||||
for part in self.submsg.walk():
|
||||
part.modified = False
|
||||
else:
|
||||
self.submsg = txt[0]
|
||||
return self.submsg
|
||||
return None
|
||||
|
||||
@@ -336,7 +333,7 @@ def check_name(msg,savname=None,ckname=check_ext,scan_zip=False):
|
||||
if badname:
|
||||
if key == 'zipname':
|
||||
badname = msg.get_filename()
|
||||
break
|
||||
break
|
||||
else:
|
||||
return Milter.CONTINUE
|
||||
except zipfile.BadZipfile:
|
||||
@@ -383,7 +380,7 @@ class _defang:
|
||||
return rc
|
||||
|
||||
def __call__(self,msg,savname=None,check=check_ext,scan_rfc822=True,
|
||||
scan_zip=False):
|
||||
scan_zip=False):
|
||||
"""Compatible entry point.
|
||||
Replace all attachments with dangerous names."""
|
||||
self._savname = savname
|
||||
@@ -453,25 +450,25 @@ class SGMLFilter(sgmllib.SGMLParser):
|
||||
n = len(rawdata)
|
||||
j = i + 2
|
||||
while j < n:
|
||||
c = rawdata[j]
|
||||
if c == ">":
|
||||
# end of declaration syntax
|
||||
self.handle_special(rawdata[i+2:j])
|
||||
return j + 1
|
||||
if c in "\"'":
|
||||
m = declstringlit.match(rawdata, j)
|
||||
if not m:
|
||||
c = rawdata[j]
|
||||
if c == ">":
|
||||
# end of declaration syntax
|
||||
self.handle_special(rawdata[i+2:j])
|
||||
return j + 1
|
||||
if c in "\"'":
|
||||
m = declstringlit.match(rawdata, j)
|
||||
if not m:
|
||||
# incomplete or an error?
|
||||
return -1
|
||||
j = m.end()
|
||||
elif c in "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ":
|
||||
m = declname.match(rawdata, j)
|
||||
if not m:
|
||||
return -1
|
||||
j = m.end()
|
||||
elif c in "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ":
|
||||
m = declname.match(rawdata, j)
|
||||
if not m:
|
||||
# incomplete or an error?
|
||||
return -1
|
||||
j = m.end()
|
||||
else:
|
||||
j += 1
|
||||
return -1
|
||||
j = m.end()
|
||||
else:
|
||||
j += 1
|
||||
# end of buffer between tokens
|
||||
return -1
|
||||
|
||||
@@ -500,7 +497,7 @@ def check_html(msg,savname=None):
|
||||
if msgtype == 'application/octet-stream':
|
||||
for (attr,name) in msg.getnames():
|
||||
if name and name.lower().endswith(".htm"):
|
||||
msgtype = 'text/html'
|
||||
msgtype = 'text/html'
|
||||
if msgtype == 'text/html':
|
||||
out = StringIO.StringIO()
|
||||
htmlfilter = HTMLScriptFilter(out)
|
||||
|
||||
+27
-50
@@ -1,15 +1,14 @@
|
||||
%define __python python2.6
|
||||
%define pythonbase python
|
||||
%define pythonbase python26
|
||||
|
||||
%define libdir %{_libdir}/pymilter
|
||||
%{!?python_sitearch: %define python_sitearch %(%{__python} -c "from distutils.sysconfig import get_python_lib; print get_python_lib(1)")}
|
||||
|
||||
Summary: Python interface to sendmail milter API
|
||||
Name: %{pythonbase}-pymilter
|
||||
Version: 1.0
|
||||
Version: 0.9.7
|
||||
Release: 1%{dist}
|
||||
Source: http://downloads.sourceforge.net/pymilter/pymilter-%{version}.tar.gz
|
||||
Source1: pymilter.te
|
||||
License: GPLv2+
|
||||
Group: Development/Libraries
|
||||
BuildRoot: %{_tmppath}/%{name}-%{version}-%{release}-root
|
||||
@@ -27,23 +26,11 @@ attach to sendmail's libmilter functionality. Additional python
|
||||
modules provide for navigating and modifying MIME parts, sending
|
||||
DSNs, and doing CBV.
|
||||
|
||||
%package selinux
|
||||
Summary: SELinux policy module for pymilter
|
||||
Group: System Environment/Base
|
||||
Requires: policycoreutils, selinux-policy, %{name}
|
||||
BuildRequires: policycoreutils, checkpolicy
|
||||
|
||||
%description selinux
|
||||
SELinux policy module for using pymilter with sendmail with selinux enforcing
|
||||
|
||||
%prep
|
||||
%setup -q -n pymilter-%{version}
|
||||
cp %{SOURCE1} pymilter.te
|
||||
|
||||
%build
|
||||
env CFLAGS="$RPM_OPT_FLAGS" %{__python} setup.py build
|
||||
checkmodule -m -M -o pymilter.mod pymilter.te
|
||||
semodule_package -o pymilter.pp -m pymilter.mod
|
||||
|
||||
%install
|
||||
rm -rf $RPM_BUILD_ROOT
|
||||
@@ -51,11 +38,31 @@ rm -rf $RPM_BUILD_ROOT
|
||||
mkdir -p $RPM_BUILD_ROOT%{_localstatedir}/run/milter
|
||||
mkdir -p $RPM_BUILD_ROOT%{_localstatedir}/log/milter
|
||||
mkdir -p $RPM_BUILD_ROOT%{libdir}
|
||||
cp start.sh $RPM_BUILD_ROOT%{libdir}
|
||||
ed $RPM_BUILD_ROOT%{libdir}/start.sh <<'EOF'
|
||||
/^datadir=/
|
||||
c
|
||||
datadir="%{_localstatedir}/log/milter"
|
||||
.
|
||||
/^piddir=/
|
||||
c
|
||||
piddir="%{_localstatedir}/run/milter"
|
||||
.
|
||||
/^libdir=/
|
||||
c
|
||||
libdir="%{libdir}"
|
||||
.
|
||||
/^python=/
|
||||
c
|
||||
python="%{__python}"
|
||||
.
|
||||
w
|
||||
q
|
||||
EOF
|
||||
chmod a+x $RPM_BUILD_ROOT%{libdir}/start.sh
|
||||
|
||||
# install selinux modules
|
||||
mkdir -p %{buildroot}%{_datadir}/selinux/targeted
|
||||
cp -p pymilter.pp %{buildroot}%{_datadir}/selinux/targeted
|
||||
|
||||
# start.sh is used by spfmilter, srsmilter, and milter, and could be used by
|
||||
# other milters using pymilter.
|
||||
%files
|
||||
%defattr(-,root,root,-)
|
||||
%doc README ChangeLog NEWS TODO CREDITS sample.py milter-template.py
|
||||
@@ -64,41 +71,11 @@ cp -p pymilter.pp %{buildroot}%{_datadir}/selinux/targeted
|
||||
%dir %attr(0755,mail,mail) %{_localstatedir}/run/milter
|
||||
%dir %attr(0755,mail,mail) %{_localstatedir}/log/milter
|
||||
|
||||
%files selinux
|
||||
%doc pymilter.te
|
||||
%{_datadir}/selinux/targeted/*
|
||||
|
||||
%clean
|
||||
rm -rf $RPM_BUILD_ROOT
|
||||
|
||||
%post selinux
|
||||
/usr/sbin/semodule -s targeted -i %{_datadir}/selinux/targeted/pymilter.pp \
|
||||
&>/dev/null || :
|
||||
|
||||
%postun selinux
|
||||
if [ $1 -eq 0 ] ; then
|
||||
/usr/sbin/semodule -s targeted -r pymilter &> /dev/null || :
|
||||
fi
|
||||
|
||||
%changelog
|
||||
* Sat Mar 1 2014 Stuart Gathman <stuart@gathman.org> 1.0-2
|
||||
- Remove start.sh to track EPEL repository, suggest daemonize as replacement
|
||||
- Selinux subpackage should not care about pymilter version
|
||||
|
||||
* Wed Jun 26 2013 Stuart Gathman <stuart@gathman.org> 1.0-1
|
||||
- Allow ACCEPT as untrapped exception policy
|
||||
- Optional dir for getaddrset and getaddrdict in Milter.config
|
||||
- Show registered milter name in untrapped exception message.
|
||||
- Include selinux subpackage
|
||||
- Provide Milter.greylist export and Milter.greylist import to migrate data
|
||||
|
||||
* Sat Mar 9 2013 Stuart Gathman <stuart@bmsi.com> 0.9.8-1
|
||||
- Add Milter.test module for unit testing milters.
|
||||
- Fix typo that prevented setsymlist from being active.
|
||||
- Change untrapped exception message to:
|
||||
- "pymilter: untrapped exception in milter app"
|
||||
|
||||
* Thu Apr 12 2012 Stuart Gathman <stuart@bmsi.com> 0.9.7-1
|
||||
* Sat Feb 25 2012 Stuart Gathman <stuart@bmsi.com> 0.9.7-1
|
||||
- Raise RuntimeError when result != CONTINUE for @noreply and @nocallback
|
||||
- Remove redundant table in miltermodule
|
||||
- Fix CNAME chain duplicating TXT records in Milter.dns (from pyspf).
|
||||
|
||||
-13
@@ -1,13 +0,0 @@
|
||||
module pymilter 1.0;
|
||||
|
||||
require {
|
||||
type sendmail_t;
|
||||
type var_run_t;
|
||||
type initrc_t;
|
||||
class sock_file { write getattr };
|
||||
class unix_stream_socket connectto;
|
||||
}
|
||||
|
||||
#============= sendmail_t ==============
|
||||
allow sendmail_t initrc_t:unix_stream_socket connectto;
|
||||
allow sendmail_t var_run_t:sock_file { write getattr };
|
||||
@@ -60,23 +60,23 @@ class sampleMilter(Milter.Milter):
|
||||
# even if we wanted the Taiwanese spam, we can't read Chinese
|
||||
# (delete if you read chinese mail)
|
||||
if val.startswith('=?big5') or val.startswith('=?ISO-2022-JP'):
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
#self.setreply('550','','Go away spammer')
|
||||
return Milter.REJECT
|
||||
return Milter.REJECT
|
||||
|
||||
# check for common spam keywords
|
||||
if val.find("$$$") >= 0 or val.find("XXX") >= 0 \
|
||||
or val.find("!!!") >= 0 or val.find("FREE") >= 0:
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
or val.find("!!!") >= 0 or val.find("FREE") >= 0:
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
#self.setreply('550','','Go away spammer')
|
||||
return Milter.REJECT
|
||||
return Milter.REJECT
|
||||
|
||||
# check for spam that pretends to be legal
|
||||
lval = val.lower()
|
||||
if lval.startswith("adv:") or lval.startswith("adv.") \
|
||||
or lval.find('viagra') >= 0:
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
return Milter.REJECT
|
||||
or lval.find('viagra') >= 0:
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
return Milter.REJECT
|
||||
|
||||
# check for invalid message id
|
||||
if lname == 'message-id' and len(val) < 4:
|
||||
@@ -86,7 +86,7 @@ class sampleMilter(Milter.Milter):
|
||||
|
||||
# check for common bulk mailers
|
||||
if lname == 'x-mailer' and \
|
||||
val.lower() in ('direct email','calypso','mail bomber'):
|
||||
val.lower() in ('direct email','calypso','mail bomber'):
|
||||
self.log('REJECT: %s: %s' % (name,val))
|
||||
#self.setreply('550','','Go away spammer')
|
||||
return Milter.REJECT
|
||||
@@ -123,7 +123,7 @@ class sampleMilter(Milter.Milter):
|
||||
h = msg.getheaders(name)
|
||||
cnt = len(h)
|
||||
for i in range(cnt,0,-1):
|
||||
self.chgheader(name,i-1,'')
|
||||
self.chgheader(name,i-1,'')
|
||||
|
||||
def eom(self):
|
||||
if not self.fp: return Milter.ACCEPT
|
||||
@@ -145,9 +145,9 @@ class sampleMilter(Milter.Milter):
|
||||
msg = rfc822.Message(out)
|
||||
msg.rewindbody()
|
||||
while 1:
|
||||
buf = out.read(8192)
|
||||
if len(buf) == 0: break
|
||||
self.replacebody(buf) # feed modified message to sendmail
|
||||
buf = out.read(8192)
|
||||
if len(buf) == 0: break
|
||||
self.replacebody(buf) # feed modified message to sendmail
|
||||
return Milter.ACCEPT # ACCEPT modified message
|
||||
finally:
|
||||
out.close()
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
[bdist_rpm]
|
||||
python=python2.6
|
||||
doc_files=README NEWS TODO COPYING CREDITS
|
||||
packager=Stuart D. Gathman <stuart@gathman.org>
|
||||
doc_files=README NEWS TODO
|
||||
packager=Stuart D. Gathman <stuart@bmsi.com>
|
||||
release=1
|
||||
|
||||
@@ -13,7 +13,7 @@ libs = ["milter"]
|
||||
libdirs = ["/usr/lib/libmilter"] # needed for Debian
|
||||
|
||||
# NOTE: importing Milter to obtain version fails when milter.so not built
|
||||
setup(name = "pymilter", version = '1.0',
|
||||
setup(name = "pymilter", version = '0.9.7',
|
||||
description="Python interface to sendmail milter API",
|
||||
long_description="""\
|
||||
This is a python extension module to enable python scripts to
|
||||
|
||||
@@ -2,7 +2,6 @@ import unittest
|
||||
import testmime
|
||||
import testsample
|
||||
import testutils
|
||||
import testgrey
|
||||
import os
|
||||
|
||||
def suite():
|
||||
@@ -10,7 +9,6 @@ def suite():
|
||||
s.addTest(testmime.suite())
|
||||
s.addTest(testsample.suite())
|
||||
s.addTest(testutils.suite())
|
||||
s.addTest(testgrey.suite())
|
||||
return s
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
-18587
File diff suppressed because it is too large
Load Diff
-55
@@ -1,55 +0,0 @@
|
||||
import unittest
|
||||
import doctest
|
||||
import os
|
||||
#from Milter.greylist import Greylist
|
||||
from Milter.greysql import Greylist
|
||||
|
||||
class GreylistTestCase(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.fname = 'test.db'
|
||||
os.remove(self.fname)
|
||||
|
||||
def tearDown(self):
|
||||
#os.remove(self.fname)
|
||||
pass
|
||||
|
||||
def testGrey(self):
|
||||
grey = Greylist(self.fname)
|
||||
# first time
|
||||
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com')
|
||||
self.assertEqual(rc,0)
|
||||
# not in window yet
|
||||
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=5*60)
|
||||
self.assertEqual(rc,0)
|
||||
# within window
|
||||
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=15*60)
|
||||
self.assertEqual(rc,1)
|
||||
# new triple
|
||||
rc = grey.check('1.2.3.5','foo@bar.com','baz@spat.com',timeinc=15*60)
|
||||
self.assertEqual(rc,0)
|
||||
# seen again
|
||||
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=5*3600)
|
||||
self.assertEqual(rc,2)
|
||||
# new one past expire
|
||||
rc = grey.check('1.2.3.5','foo@bar.com','baz@spat.com',timeinc=6*3600)
|
||||
self.assertEqual(rc,0)
|
||||
# original past retain
|
||||
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=37*24*3600)
|
||||
self.assertEqual(rc,0)
|
||||
# new one for testing expire
|
||||
rc = grey.check('1.2.3.5','flub@bar.com','baz@spat.com',timeinc=20*24*3600)
|
||||
self.assertEqual(rc,0)
|
||||
grey.close()
|
||||
# test cleanup
|
||||
grey = Greylist(self.fname)
|
||||
rc = grey.clean(timeinc=37*24*3600)
|
||||
self.assertEqual(rc,1)
|
||||
grey.close()
|
||||
|
||||
def suite():
|
||||
s = unittest.makeSuite(GreylistTestCase,'test')
|
||||
return s
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.TextTestRunner().run(suite())
|
||||
+5
-8
@@ -1,7 +1,4 @@
|
||||
# $Log$
|
||||
# Revision 1.5 2011/06/09 17:27:42 customdesigned
|
||||
# Documentation updates.
|
||||
#
|
||||
# Revision 1.4 2005/07/20 14:49:44 customdesigned
|
||||
# Handle corrupt and empty ZIP files.
|
||||
#
|
||||
@@ -72,12 +69,12 @@ class MimeTestCase(unittest.TestCase):
|
||||
# python 2.4 doesn't get exceptions on missing boundaries, and
|
||||
# if message is modified, output is readable by mail clients
|
||||
if sys.hexversion < 0x02040000:
|
||||
self.fail('should get boundary error parsing bad rfc822 attachment')
|
||||
self.fail('should get boundary error parsing bad rfc822 attachment')
|
||||
except Errors.BoundaryError:
|
||||
pass
|
||||
|
||||
def testDefang(self,vname='virus1',part=1,
|
||||
fname='LOVE-LETTER-FOR-YOU.TXT.vbs'):
|
||||
fname='LOVE-LETTER-FOR-YOU.TXT.vbs'):
|
||||
msg = mime.message_from_file(open('test/'+vname,"r"))
|
||||
mime.defang(msg,scan_zip=True)
|
||||
self.failUnless(msg.ismodified(),"virus not removed")
|
||||
@@ -111,7 +108,7 @@ class MimeTestCase(unittest.TestCase):
|
||||
self.failIf(msg.ismultipart())
|
||||
txt2 = msg.get_payload()
|
||||
self.failUnless(txt2 == mime.virus_msg % \
|
||||
(fname,hostname,None),txt2)
|
||||
(fname,hostname,None),txt2)
|
||||
|
||||
# honey virus has a sneaky ASP payload which is parsed correctly
|
||||
# by email package in python-2.2.2, but not by mime.MimeMessage or 2.2.1
|
||||
@@ -125,7 +122,7 @@ class MimeTestCase(unittest.TestCase):
|
||||
txt2 = parts[1].get_payload()
|
||||
txt3 = parts[2].get_payload()
|
||||
self.failUnless(txt2.rstrip()+'\n' == mime.virus_msg % \
|
||||
(fname,hostname,None),txt2)
|
||||
(fname,hostname,None),txt2)
|
||||
if txt3 != '':
|
||||
self.failUnless(txt3.rstrip()+'\n' == mime.virus_msg % \
|
||||
('story[1].asp',hostname,None),txt3)
|
||||
@@ -173,7 +170,7 @@ class MimeTestCase(unittest.TestCase):
|
||||
msg = mime.message_from_file(open('test/'+fname,'r'))
|
||||
mime.defang(msg,scan_zip=True)
|
||||
self.failIf(msg.ismodified())
|
||||
msg = mime.message_from_file(open('test/test2','r'))
|
||||
msg = mime.message_from_file(open('test/tmpytgcE5.fail','r'))
|
||||
rc = mime.check_attachments(msg,self._chk_attach)
|
||||
self.assertEquals(self.filename,"7501'S FOR TWO GOLDEN SOURCES SHIPMENTS FOR TAX & DUTY PURPOSES ONLY.PDF")
|
||||
self.assertEquals(rc,Milter.CONTINUE)
|
||||
|
||||
+95
-10
@@ -4,12 +4,97 @@ import sample
|
||||
import mime
|
||||
import rfc822
|
||||
import StringIO
|
||||
from Milter.test import TestBase
|
||||
|
||||
class TestMilter(TestBase,sample.sampleMilter):
|
||||
class TestMilter(sample.sampleMilter):
|
||||
|
||||
_protocol = 0
|
||||
def __init__(self):
|
||||
TestBase.__init__(self)
|
||||
sample.sampleMilter.__init__(self)
|
||||
self.logfp = open("test/milter.log","a")
|
||||
|
||||
def log(self,*msg):
|
||||
for i in msg: print >>self.logfp, i,
|
||||
print >>self.logfp
|
||||
|
||||
def replacebody(self,chunk):
|
||||
if self._body:
|
||||
self._body.write(chunk)
|
||||
self.bodyreplaced = True
|
||||
else:
|
||||
raise IOError,"replacebody not called from eom()"
|
||||
|
||||
# FIXME: rfc822 indexing does not really reflect the way chg/add header
|
||||
# work for a milter
|
||||
def chgheader(self,field,idx,value):
|
||||
self.log('chgheader: %s[%d]=%s' % (field,idx,value))
|
||||
if value == '':
|
||||
del self._msg[field]
|
||||
else:
|
||||
self._msg[field] = value
|
||||
self.headerschanged = True
|
||||
|
||||
def addheader(self,field,value):
|
||||
self.log('addheader: %s=%s' % (field,value))
|
||||
self._msg[field] = value
|
||||
self.headerschanged = True
|
||||
|
||||
def feedMsg(self,fname):
|
||||
self._body = None
|
||||
self.bodyreplaced = False
|
||||
self.headerschanged = 0
|
||||
fp = open('test/'+fname,'r')
|
||||
msg = rfc822.Message(fp)
|
||||
rc = self.envfrom('<spam@advertisements.com>')
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
rc = self.envrcpt('<victim@lamb.com>')
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
line = None
|
||||
for h in msg.headers:
|
||||
if h[:1].isspace():
|
||||
line = line + h
|
||||
continue
|
||||
if not line:
|
||||
line = h
|
||||
continue
|
||||
s = line.split(': ',1)
|
||||
rc = self.header(s[0],s[1].strip())
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
line = h
|
||||
if line:
|
||||
s = line.split(': ',1)
|
||||
rc = self.header(s[0],s[1])
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
rc = self.eoh()
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
while 1:
|
||||
buf = fp.read(8192)
|
||||
if len(buf) == 0: break
|
||||
rc = self.body(buf)
|
||||
if rc != Milter.CONTINUE: return rc
|
||||
self._msg = msg
|
||||
self._body = StringIO.StringIO()
|
||||
rc = self.eom()
|
||||
if self.bodyreplaced:
|
||||
body = self._body.getvalue()
|
||||
else:
|
||||
msg.rewindbody()
|
||||
body = msg.fp.read()
|
||||
self._body = StringIO.StringIO()
|
||||
self._body.writelines(msg.headers)
|
||||
self._body.write('\n')
|
||||
self._body.write(body)
|
||||
return rc
|
||||
|
||||
def connect(self,host='localhost'):
|
||||
self._body = None
|
||||
self.bodyreplaced = False
|
||||
rc = sample.sampleMilter.connect(self,host,1,0)
|
||||
if rc != Milter.CONTINUE and rc != Milter.ACCEPT:
|
||||
self.close()
|
||||
return rc
|
||||
rc = self.hello('spamrelay')
|
||||
if rc != Milter.CONTINUE:
|
||||
self.close()
|
||||
return rc
|
||||
|
||||
class BMSMilterTestCase(unittest.TestCase):
|
||||
|
||||
@@ -19,7 +104,7 @@ class BMSMilterTestCase(unittest.TestCase):
|
||||
self.failUnless(rc == Milter.CONTINUE)
|
||||
rc = milter.feedMsg(fname)
|
||||
self.failUnless(rc == Milter.ACCEPT)
|
||||
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||
self.failUnless(milter.bodyreplaced,"Message body not replaced")
|
||||
fp = milter._body
|
||||
open('test/'+fname+".tstout","w").write(fp.getvalue())
|
||||
#self.failUnless(fp.getvalue() == open("test/virus1.out","r").read())
|
||||
@@ -34,7 +119,7 @@ class BMSMilterTestCase(unittest.TestCase):
|
||||
milter.connect('somehost')
|
||||
rc = milter.feedMsg(fname)
|
||||
self.failUnless(rc == Milter.ACCEPT)
|
||||
self.failIf(milter._bodyreplaced,"Milter needlessly replaced body.")
|
||||
self.failIf(milter.bodyreplaced,"Milter needlessly replaced body.")
|
||||
fp = milter._body
|
||||
open('test/'+fname+".tstout","w").write(fp.getvalue())
|
||||
milter.close()
|
||||
@@ -44,17 +129,17 @@ class BMSMilterTestCase(unittest.TestCase):
|
||||
milter.connect('somehost')
|
||||
rc = milter.feedMsg('samp1')
|
||||
self.failUnless(rc == Milter.ACCEPT)
|
||||
self.failIf(milter._bodyreplaced,"Milter needlessly replaced body.")
|
||||
self.failIf(milter.bodyreplaced,"Milter needlessly replaced body.")
|
||||
rc = milter.feedMsg("virus3")
|
||||
self.failUnless(rc == Milter.ACCEPT)
|
||||
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||
self.failUnless(milter.bodyreplaced,"Message body not replaced")
|
||||
fp = milter._body
|
||||
open("test/virus3.tstout","w").write(fp.getvalue())
|
||||
#self.failUnless(fp.getvalue() == open("test/virus3.out","r").read())
|
||||
rc = milter.feedMsg("virus6")
|
||||
self.failUnless(rc == Milter.ACCEPT)
|
||||
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||
self.failUnless(milter._headerschanged,"Message headers not adjusted")
|
||||
self.failUnless(milter.bodyreplaced,"Message body not replaced")
|
||||
self.failUnless(milter.headerschanged,"Message headers not adjusted")
|
||||
fp = milter._body
|
||||
open("test/virus6.tstout","w").write(fp.getvalue())
|
||||
milter.close()
|
||||
|
||||
Reference in New Issue
Block a user