Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| d7ef47d76b |
@@ -1,8 +0,0 @@
|
|||||||
*.pyc
|
|
||||||
build/
|
|
||||||
test/*.out
|
|
||||||
test/*.tstout
|
|
||||||
test/*.log
|
|
||||||
test.db
|
|
||||||
dist
|
|
||||||
MANIFEST
|
|
||||||
@@ -1,213 +1,3 @@
|
|||||||
# 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
|
|
||||||
#
|
|
||||||
# Revision 1.29 2011/06/09 15:45:27 customdesigned
|
|
||||||
# Print callback name for non-int return error.
|
|
||||||
#
|
|
||||||
# Revision 1.28 2011/06/08 23:13:48 customdesigned
|
|
||||||
# Generate special exception when callback return not int.
|
|
||||||
#
|
|
||||||
# Revision 1.27 2009/07/28 21:45:54 customdesigned
|
|
||||||
# Add getversion() to return runtime version.
|
|
||||||
#
|
|
||||||
# Revision 1.26 2009/07/28 21:08:20 customdesigned
|
|
||||||
# Increment del count.
|
|
||||||
#
|
|
||||||
# Revision 1.25 2009/07/28 20:58:55 customdesigned
|
|
||||||
# getdiag method
|
|
||||||
#
|
|
||||||
# Revision 1.24 2009/06/09 01:54:44 customdesigned
|
|
||||||
# Forgot to initialize optional parameter.
|
|
||||||
#
|
|
||||||
# Revision 1.23 2009/05/29 20:44:58 customdesigned
|
|
||||||
# Typo SMFIP_NO constants.
|
|
||||||
#
|
|
||||||
# Revision 1.22 2009/05/29 19:53:36 customdesigned
|
|
||||||
# Typo SMFIS_ALL_OPTS
|
|
||||||
#
|
|
||||||
# Revision 1.21 2009/05/29 19:49:40 customdesigned
|
|
||||||
# Typo calling helo instead of negotiate.
|
|
||||||
#
|
|
||||||
# Revision 1.20 2009/05/29 18:25:59 customdesigned
|
|
||||||
# Null terminate keyword list.
|
|
||||||
#
|
|
||||||
# Revision 1.19 2009/05/28 18:36:42 customdesigned
|
|
||||||
# Support new callbacks, including negotiate
|
|
||||||
#
|
|
||||||
# Revision 1.18 2009/05/21 21:53:05 customdesigned
|
|
||||||
# First cut at support unknown, data, negotiate callbacks.
|
|
||||||
#
|
|
||||||
# Revision 1.17 2009/02/06 04:28:08 customdesigned
|
|
||||||
# Oops! Missing options argument pointer for addrcpt.
|
|
||||||
#
|
|
||||||
# Revision 1.16 2008/12/16 04:21:05 customdesigned
|
|
||||||
# Fedora release
|
|
||||||
#
|
|
||||||
# Revision 1.15 2008/12/13 20:29:56 customdesigned
|
|
||||||
# Split off milter applications.
|
|
||||||
#
|
|
||||||
# Revision 1.14 2008/12/04 19:43:00 customdesigned
|
|
||||||
# Doc updates.
|
|
||||||
#
|
|
||||||
# Revision 1.13 2008/11/23 03:06:47 customdesigned
|
|
||||||
# Milter support for chgfrom.
|
|
||||||
#
|
|
||||||
# Revision 1.12 2008/11/21 20:42:52 customdesigned
|
|
||||||
# Support smfi_chgfrom and smfi_addrcpt_par.
|
|
||||||
#
|
|
||||||
# Revision 1.11 2007/09/25 02:26:29 customdesigned
|
|
||||||
# Update license.
|
|
||||||
#
|
|
||||||
# Revision 1.10 2006/02/12 02:00:42 customdesigned
|
|
||||||
# Resolve FIXME for wrap_close.
|
|
||||||
#
|
|
||||||
# Revision 1.9 2005/12/23 21:46:36 customdesigned
|
|
||||||
# Compile on sendmail-8.12 (ifdef SMFIR_INSHEADER)
|
|
||||||
#
|
|
||||||
# Revision 1.8 2005/10/20 23:23:36 customdesigned
|
|
||||||
# Include smfi_progress is SMFIR_PROGRESS defined
|
|
||||||
#
|
|
||||||
# Revision 1.7 2005/10/20 23:04:46 customdesigned
|
|
||||||
# Add optional idx for position of added header.
|
|
||||||
#
|
|
||||||
# Revision 1.6 2005/07/15 22:18:17 customdesigned
|
|
||||||
# Support callback exception policy
|
|
||||||
#
|
|
||||||
# Revision 1.5 2005/06/24 04:20:07 customdesigned
|
|
||||||
# Report context allocation error.
|
|
||||||
#
|
|
||||||
# Revision 1.4 2005/06/24 04:12:43 customdesigned
|
|
||||||
# Remove unused name argument to generic wrappers.
|
|
||||||
#
|
|
||||||
# Revision 1.3 2005/06/24 03:57:35 customdesigned
|
|
||||||
# Handle close called before connect.
|
|
||||||
#
|
|
||||||
# Revision 1.2 2005/06/02 04:18:55 customdesigned
|
|
||||||
# Update copyright notices after reading article on /.
|
|
||||||
#
|
|
||||||
# Revision 1.1.1.2 2005/05/31 18:09:06 customdesigned
|
|
||||||
# Release 0.7.1
|
|
||||||
#
|
|
||||||
# Revision 2.31 2004/08/23 02:24:36 stuart
|
|
||||||
# Support setbacklog
|
|
||||||
#
|
|
||||||
# Revision 2.30 2004/08/21 20:29:53 stuart
|
|
||||||
# Support option of 11 lines max for mlreply.
|
|
||||||
#
|
|
||||||
# Revision 2.29 2004/08/21 04:14:29 stuart
|
|
||||||
# mlreply support
|
|
||||||
#
|
|
||||||
# Revision 2.28 2004/08/21 02:45:21 stuart
|
|
||||||
# Don't leak int constants if module unloaded.
|
|
||||||
#
|
|
||||||
# Revision 2.27 2004/04/06 03:19:59 stuart
|
|
||||||
# Release 0.6.8
|
|
||||||
#
|
|
||||||
# Revision 2.26 2004/03/04 21:43:06 stuart
|
|
||||||
# Fix memory leak by removing unused dynamic template buffer,
|
|
||||||
# thanks again to Alexander Kourakos.
|
|
||||||
#
|
|
||||||
# Revision 2.25 2004/03/01 19:45:03 stuart
|
|
||||||
# Release 0.6.5
|
|
||||||
#
|
|
||||||
# Revision 2.24 2004/03/01 18:56:50 stuart
|
|
||||||
# Support progress reporting.
|
|
||||||
#
|
|
||||||
# Revision 2.23 2004/03/01 18:36:09 stuart
|
|
||||||
# Plug memory leak. Thanks to Alexander Kourakos.
|
|
||||||
#
|
|
||||||
# Revision 2.22 2003/11/02 03:01:46 stuart
|
|
||||||
# Adjust SMTP error codes after careful reading of standard.
|
|
||||||
#
|
|
||||||
# Revision 2.21 2003/06/24 19:57:04 stuart
|
|
||||||
# Allow removing a python milter callback by setting to None.
|
|
||||||
#
|
|
||||||
# Revision 2.20 2003/02/13 17:08:57 stuart
|
|
||||||
# IPV6 support
|
|
||||||
#
|
|
||||||
# Revision 2.19 2003/02/13 16:58:29 stuart
|
|
||||||
# Support passing None to setreply and chgheader.
|
|
||||||
#
|
|
||||||
# Revision 2.18 2002/12/11 16:44:06 stuart
|
|
||||||
# Support QUARANTINE if supported by libmilter.
|
|
||||||
#
|
|
||||||
# Revision 2.17 2002/04/18 20:20:35 stuart
|
|
||||||
# Fix for NULL hostaddr in connect callback from Jason Erickson.
|
|
||||||
#
|
|
||||||
# Revision 2.16 2001/09/26 13:29:09 stuart
|
|
||||||
# sa_len not supported by linux.
|
|
||||||
#
|
|
||||||
# Revision 2.15 2001/09/25 17:28:40 stuart
|
|
||||||
# Copyrights, documentation, release 0.3.1
|
|
||||||
#
|
|
||||||
# Revision 2.14 2001/09/25 00:36:57 stuart
|
|
||||||
# Pass hostaddr to python code in format used by standard socket module.
|
|
||||||
#
|
|
||||||
# Revision 2.13 2001/09/24 23:44:55 stuart
|
|
||||||
# Return old callback from setcallback functions.
|
|
||||||
#
|
|
||||||
# Revision 2.12 2001/09/24 20:02:30 stuart
|
|
||||||
# Remove redundant setpriv
|
|
||||||
#
|
|
||||||
# Revision 2.11 2001/09/23 22:26:35 stuart
|
|
||||||
# Update docs. Streamline Milter.py
|
|
||||||
# update testbms.py to reflect actual sendmail behaviour with multiple
|
|
||||||
# messages per connection.
|
|
||||||
#
|
|
||||||
# Revision 2.10 2001/09/22 15:33:42 stuart
|
|
||||||
# More doc comment updates.
|
|
||||||
#
|
|
||||||
# Revision 2.9 2001/09/22 14:52:27 stuart
|
|
||||||
# Actually return retval in _generic_return.
|
|
||||||
# Go over doc comments.
|
|
||||||
#
|
|
||||||
# Revision 2.8 2001/09/22 01:59:32 stuart
|
|
||||||
# Prevent reentrant call of milter_main, which libmilter doesn't support.
|
|
||||||
#
|
|
||||||
# Revision 2.7 2001/09/22 01:47:37 stuart
|
|
||||||
# Forgot to set milter interp.
|
|
||||||
#
|
|
||||||
# Revision 2.6 2001/09/22 01:23:53 stuart
|
|
||||||
# Added proper threading after research in python docs.
|
|
||||||
#
|
|
||||||
# Revision 2.5 2001/09/21 20:08:51 stuart
|
|
||||||
# Release 0.2.3
|
|
||||||
#
|
|
||||||
# Revision 2.4 2001/09/20 16:18:16 stuart
|
|
||||||
# libmilter checks in_eom state, so we don't have to.
|
|
||||||
#
|
|
||||||
# Revision 2.3 2001/09/19 06:02:33 stuart
|
|
||||||
# Make more stuff static.
|
|
||||||
#
|
|
||||||
# Revision 2.1 2001/09/19 04:24:13 stuart
|
|
||||||
# Use extension type to track context in python.
|
|
||||||
#
|
|
||||||
# Revision 1.4 2001/09/18 18:48:28 stuart
|
|
||||||
# clear private data reference in _clear_context
|
|
||||||
#
|
|
||||||
# Revision 1.3 2001/09/15 04:19:37 stuart
|
|
||||||
# nasty off by 1 mem overwrite bugs in wrap_env
|
|
||||||
# generic_set_callback
|
|
||||||
#
|
|
||||||
# Revision 1.2 2001/09/15 03:15:39 stuart
|
|
||||||
# several bugs fixed, works smoothly
|
|
||||||
#
|
|
||||||
# Revision 1.69 2006/11/04 22:09:39 customdesigned
|
# Revision 1.69 2006/11/04 22:09:39 customdesigned
|
||||||
# Another lame DSN heuristic. Block PTR cache poisoning attack.
|
# Another lame DSN heuristic. Block PTR cache poisoning attack.
|
||||||
#
|
#
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ PROJECT_NAME = pymilter
|
|||||||
# This could be handy for archiving the generated documentation or
|
# This could be handy for archiving the generated documentation or
|
||||||
# if some version control system is used.
|
# if some version control system is used.
|
||||||
|
|
||||||
PROJECT_NUMBER = 1.0.2
|
PROJECT_NUMBER = 0.9.8
|
||||||
|
|
||||||
# The OUTPUT_DIRECTORY tag is used to specify the (relative or absolute)
|
# The OUTPUT_DIRECTORY tag is used to specify the (relative or absolute)
|
||||||
# base path where the generated documentation will be put.
|
# base path where the generated documentation will be put.
|
||||||
|
|||||||
+27
-63
@@ -8,17 +8,12 @@
|
|||||||
# Copyright 2001,2009 Business Management Systems, Inc.
|
# Copyright 2001,2009 Business Management Systems, Inc.
|
||||||
# This code is under the GNU General Public License. See COPYING for details.
|
# This code is under the GNU General Public License. See COPYING for details.
|
||||||
|
|
||||||
from __future__ import print_function
|
__version__ = '0.9.8'
|
||||||
__version__ = '1.0.4'
|
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
import milter
|
import milter
|
||||||
try:
|
import thread
|
||||||
import thread
|
|
||||||
except:
|
|
||||||
# libmilter uses posix threads
|
|
||||||
import _thread as thread
|
|
||||||
|
|
||||||
from milter import *
|
from milter import *
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
@@ -48,12 +43,6 @@ OPTIONAL_CALLBACKS = {
|
|||||||
'header':(P_NR_HDR,P_NOHDRS)
|
'header':(P_NR_HDR,P_NOHDRS)
|
||||||
}
|
}
|
||||||
|
|
||||||
MACRO_CALLBACKS = {
|
|
||||||
'connect': M_CONNECT,
|
|
||||||
'hello': M_HELO, 'envfrom': M_ENVFROM, 'envrcpt': M_ENVRCPT,
|
|
||||||
'data': M_DATA, 'eom': M_EOM, 'eoh': M_EOH
|
|
||||||
}
|
|
||||||
|
|
||||||
## @private
|
## @private
|
||||||
R = re.compile(r'%+')
|
R = re.compile(r'%+')
|
||||||
|
|
||||||
@@ -112,7 +101,7 @@ def rejected_recipients(klass):
|
|||||||
return enable_protocols(klass,P_RCPT_REJ)
|
return enable_protocols(klass,P_RCPT_REJ)
|
||||||
|
|
||||||
## Milter leading space on headers. A class decorator that calls
|
## Milter leading space on headers. A class decorator that calls
|
||||||
# enable_protocols() with the P_HDR_LEADSPC flag. By default,
|
# enable_protocols() with the P_HEAD_LEADSPC flag. By default,
|
||||||
# header continuation lines are collected and joined before getting
|
# header continuation lines are collected and joined before getting
|
||||||
# sent to a milter. Headers modified or added by the milter are
|
# sent to a milter. Headers modified or added by the milter are
|
||||||
# folded by the MTA as necessary according to its own standards.
|
# folded by the MTA as necessary according to its own standards.
|
||||||
@@ -130,7 +119,7 @@ def rejected_recipients(klass):
|
|||||||
# @param klass the %milter application class to modify
|
# @param klass the %milter application class to modify
|
||||||
# @return the modified %milter class
|
# @return the modified %milter class
|
||||||
def header_leading_space(klass):
|
def header_leading_space(klass):
|
||||||
return enable_protocols(klass,P_HDR_LEADSPC)
|
return enable_protocols(klass,P_HEAD_LEADSPC)
|
||||||
|
|
||||||
## Function decorator to disable callback methods.
|
## Function decorator to disable callback methods.
|
||||||
# If the MTA supports it, tells the MTA not to invoke this callback,
|
# If the MTA supports it, tells the MTA not to invoke this callback,
|
||||||
@@ -147,7 +136,6 @@ def nocallback(func):
|
|||||||
except KeyError:
|
except KeyError:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
'@nocallback applied to non-optional method: '+func.__name__)
|
'@nocallback applied to non-optional method: '+func.__name__)
|
||||||
@wraps(func)
|
|
||||||
def wrapper(self,*args):
|
def wrapper(self,*args):
|
||||||
if func(self,*args) != CONTINUE:
|
if func(self,*args) != CONTINUE:
|
||||||
raise RuntimeError('%s return code must be CONTINUE with @nocallback'
|
raise RuntimeError('%s return code must be CONTINUE with @nocallback'
|
||||||
@@ -180,21 +168,6 @@ def noreply(func):
|
|||||||
wrapper.milter_protocol = nr_mask
|
wrapper.milter_protocol = nr_mask
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
## Function decorator to set macros used in a callback.
|
|
||||||
# By default, the MTA sends all macros defined for a callback.
|
|
||||||
# If some or all of these are unused, the bandwidth can be saved
|
|
||||||
# by listing the ones that are used.
|
|
||||||
# @since 1.0.2
|
|
||||||
def symlist(*syms):
|
|
||||||
if len(syms) > 5:
|
|
||||||
raise ValueError('@symlist limited to 5 macros by MTA: '+func.__name__)
|
|
||||||
def setsyms(func):
|
|
||||||
if func.__name__ not in MACRO_CALLBACKS:
|
|
||||||
raise ValueError('@symlist applied to non-symlist method: '+func.__name__)
|
|
||||||
func._symlist = syms
|
|
||||||
return func
|
|
||||||
return setsyms
|
|
||||||
|
|
||||||
## Disabled action exception.
|
## Disabled action exception.
|
||||||
# set_flags() can tell the MTA that this application will not use certain
|
# set_flags() can tell the MTA that this application will not use certain
|
||||||
# features (such as CHGFROM). This can also be negotiated for each
|
# features (such as CHGFROM). This can also be negotiated for each
|
||||||
@@ -284,7 +257,7 @@ class Base(object):
|
|||||||
## Defined by subclasses to write log messages.
|
## Defined by subclasses to write log messages.
|
||||||
def log(self,*msg): pass
|
def log(self,*msg): pass
|
||||||
## Called for each connection to the MTA. Called by the
|
## Called for each connection to the MTA. Called by the
|
||||||
# <a href="milter_api/xxfi_connect.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_connect">
|
||||||
# xxfi_connect</a> callback.
|
# xxfi_connect</a> callback.
|
||||||
# The <code>hostname</code> provided by the local MTA is either
|
# The <code>hostname</code> provided by the local MTA is either
|
||||||
# the PTR name or the IP in the form "[1.2.3.4]" if no PTR is available.
|
# the PTR name or the IP in the form "[1.2.3.4]" if no PTR is available.
|
||||||
@@ -321,7 +294,7 @@ class Base(object):
|
|||||||
@nocallback
|
@nocallback
|
||||||
def hello(self,hostname): return CONTINUE
|
def hello(self,hostname): return CONTINUE
|
||||||
## Called when the SMTP client says MAIL FROM. Called by the
|
## Called when the SMTP client says MAIL FROM. Called by the
|
||||||
# <a href="milter_api/xxfi_envfrom.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_envfrom">
|
||||||
# xxfi_envfrom</a> callback.
|
# xxfi_envfrom</a> callback.
|
||||||
# Returning REJECT rejects the message, but not the connection.
|
# Returning REJECT rejects the message, but not the connection.
|
||||||
# The sender is the "envelope" from as defined by
|
# The sender is the "envelope" from as defined by
|
||||||
@@ -332,7 +305,7 @@ class Base(object):
|
|||||||
@nocallback
|
@nocallback
|
||||||
def envfrom(self,f,*str): return CONTINUE
|
def envfrom(self,f,*str): return CONTINUE
|
||||||
## Called when the SMTP client says RCPT TO. Called by the
|
## Called when the SMTP client says RCPT TO. Called by the
|
||||||
# <a href="milter_api/xxfi_envrcpt.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_envrcpt">
|
||||||
# xxfi_envrcpt</a> callback.
|
# xxfi_envrcpt</a> callback.
|
||||||
# Returning REJECT rejects the current recipient, not the entire message.
|
# Returning REJECT rejects the current recipient, not the entire message.
|
||||||
# The recipient is the "envelope" recipient as defined by
|
# The recipient is the "envelope" recipient as defined by
|
||||||
@@ -392,13 +365,13 @@ class Base(object):
|
|||||||
for func,(nr,nc) in OPTIONAL_CALLBACKS.items():
|
for func,(nr,nc) in OPTIONAL_CALLBACKS.items():
|
||||||
func = getattr(klass,func)
|
func = getattr(klass,func)
|
||||||
ca = getattr(func,'milter_protocol',0)
|
ca = getattr(func,'milter_protocol',0)
|
||||||
#print(func,hex(nr),hex(nc),hex(ca))
|
#print func,hex(nr),hex(nc),hex(ca)
|
||||||
p |= (nr|nc) & ~ca
|
p |= (nr|nc) & ~ca
|
||||||
klass._protocol_mask = p
|
klass._protocol_mask = p
|
||||||
return p
|
return p
|
||||||
|
|
||||||
## Negotiate milter protocol options. Called by the
|
## Negotiate milter protocol options. Called by the
|
||||||
# <a href="milter_api/xxfi_negotiate.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_negotiate">
|
||||||
# xffi_negotiate</a> callback. This is an advanced callback,
|
# xffi_negotiate</a> callback. This is an advanced callback,
|
||||||
# do not override unless you know what you are doing. Most
|
# do not override unless you know what you are doing. Most
|
||||||
# negotiation can be done simply by using the supplied
|
# negotiation can be done simply by using the supplied
|
||||||
@@ -415,16 +388,11 @@ class Base(object):
|
|||||||
def negotiate(self,opts):
|
def negotiate(self,opts):
|
||||||
try:
|
try:
|
||||||
self._actions,p,f1,f2 = opts
|
self._actions,p,f1,f2 = opts
|
||||||
for func,stage in MACRO_CALLBACKS.items():
|
|
||||||
func = getattr(self,func)
|
|
||||||
syms = getattr(func,'_symlist',None)
|
|
||||||
if syms is not None:
|
|
||||||
self.setsymlist(stage,*syms)
|
|
||||||
opts[1] = self._protocol = p & ~self.protocol_mask()
|
opts[1] = self._protocol = p & ~self.protocol_mask()
|
||||||
opts[2] = 0
|
opts[2] = 0
|
||||||
opts[3] = 0
|
opts[3] = 0
|
||||||
#self.log("Negotiated:",opts)
|
#self.log("Negotiated:",opts)
|
||||||
except Exception as x:
|
except:
|
||||||
# don't change anything if something went wrong
|
# don't change anything if something went wrong
|
||||||
return ALL_OPTS
|
return ALL_OPTS
|
||||||
return CONTINUE
|
return CONTINUE
|
||||||
@@ -434,7 +402,7 @@ class Base(object):
|
|||||||
## Return the value of an MTA macro. Sendmail macro names
|
## Return the value of an MTA macro. Sendmail macro names
|
||||||
# are either single chars (e.g. "j") or multiple chars enclosed
|
# are either single chars (e.g. "j") or multiple chars enclosed
|
||||||
# in braces (e.g. "{auth_type}"). Macro names are MTA dependent.
|
# in braces (e.g. "{auth_type}"). Macro names are MTA dependent.
|
||||||
# See <a href="milter_api/smfi_getsymval.html">
|
# See <a href="https://www.milter.org/developers/api/smfi_getsymval">
|
||||||
# smfi_getsymval</a> for default sendmail macros.
|
# smfi_getsymval</a> for default sendmail macros.
|
||||||
# @param sym the macro name
|
# @param sym the macro name
|
||||||
def getsymval(self,sym):
|
def getsymval(self,sym):
|
||||||
@@ -448,7 +416,7 @@ class Base(object):
|
|||||||
# head scratching. What will <i>really</i> irritate you, however,
|
# head scratching. What will <i>really</i> irritate you, however,
|
||||||
# is that if you carefully double any '%%', your message will be
|
# is that if you carefully double any '%%', your message will be
|
||||||
# sent - but with the '%%' still doubled!
|
# sent - but with the '%%' still doubled!
|
||||||
# See <a href="milter_api/smfi_setreply.html">
|
# See <a href="https://www.milter.org/developers/api/smfi_setreply">
|
||||||
# smfi_setreply</a> for more information.
|
# smfi_setreply</a> for more information.
|
||||||
# @param rcode The three-digit (RFC 821/2821) SMTP reply code as a string.
|
# @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>.
|
# rcode cannot be None, and <b>must be a valid 4XX or 5XX reply code</b>.
|
||||||
@@ -470,32 +438,28 @@ class Base(object):
|
|||||||
# set. The protocol stages are M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT,
|
# set. The protocol stages are M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT,
|
||||||
# M_DATA, M_EOM, M_EOH.
|
# M_DATA, M_EOM, M_EOH.
|
||||||
#
|
#
|
||||||
# May only be called from negotiate callback. Hence, this is an advanced
|
# May only be called from negotiate callback.
|
||||||
# feature. Use the @@symlist function decorator to conviently set
|
|
||||||
# the macros used by a callback.
|
|
||||||
# @since 0.9.8, previous version was misspelled!
|
# @since 0.9.8, previous version was misspelled!
|
||||||
# @param stage the protocol stage to set to macro list for,
|
# @param stage the protocol stage to set to macro list for,
|
||||||
# one of the M_* constants defined in Milter
|
# one of the M_* constants defined in Milter
|
||||||
# @param macros space separated and/or lists of strings
|
# @param macros space separated and/or lists of strings
|
||||||
def setsymlist(self,stage,*macros):
|
def setsymlist(self,stage,*macros):
|
||||||
if not self._actions & SETSYMLIST: raise DisabledAction("SETSYMLIST")
|
if not self._actions & SETSYMLIST: raise DisabledAction("SETSYMLIST")
|
||||||
if len(macros) > 5:
|
|
||||||
raise ValueError('setsymlist limited to 5 macros by MTA')
|
|
||||||
a = []
|
a = []
|
||||||
for m in macros:
|
for m in macros:
|
||||||
try:
|
try:
|
||||||
m = m.encode('utf8')
|
m = m.encode('utf8')
|
||||||
except: pass
|
except: pass
|
||||||
try:
|
try:
|
||||||
m = m.split(b' ')
|
m = m.split(' ')
|
||||||
a += m
|
|
||||||
except: pass
|
except: pass
|
||||||
return self._ctx.setsymlist(stage,b' '.join(a))
|
a += m
|
||||||
|
return self._ctx.setsmlist(stage,' '.join(a))
|
||||||
|
|
||||||
# Milter methods which can only be called from eom callback.
|
# Milter methods which can only be called from eom callback.
|
||||||
|
|
||||||
## Add a mail header field.
|
## Add a mail header field.
|
||||||
# Calls <a href="milter_api/smfi_addheader.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_addheader">
|
||||||
# smfi_addheader</a>.
|
# smfi_addheader</a>.
|
||||||
# The <code>Milter.ADDHDRS</code> action flag must be set.
|
# The <code>Milter.ADDHDRS</code> action flag must be set.
|
||||||
#
|
#
|
||||||
@@ -509,7 +473,7 @@ class Base(object):
|
|||||||
return self._ctx.addheader(field,value,idx)
|
return self._ctx.addheader(field,value,idx)
|
||||||
|
|
||||||
## Change the value of a mail header field.
|
## Change the value of a mail header field.
|
||||||
# Calls <a href="milter_api/smfi_chgheader.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_chgheader">
|
||||||
# smfi_chgheader</a>.
|
# smfi_chgheader</a>.
|
||||||
# The <code>Milter.CHGHDRS</code> action flag must be set.
|
# The <code>Milter.CHGHDRS</code> action flag must be set.
|
||||||
#
|
#
|
||||||
@@ -523,7 +487,7 @@ class Base(object):
|
|||||||
return self._ctx.chgheader(field,idx,value)
|
return self._ctx.chgheader(field,idx,value)
|
||||||
|
|
||||||
## Add a recipient to the message.
|
## Add a recipient to the message.
|
||||||
# Calls <a href="milter_api/smfi_addrcpt.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_addrcpt">
|
||||||
# smfi_addrcpt</a>.
|
# smfi_addrcpt</a>.
|
||||||
# If no corresponding mail header is added, this is like a Bcc.
|
# If no corresponding mail header is added, this is like a Bcc.
|
||||||
# The syntax of the recipient is the same as used in the SMTP
|
# The syntax of the recipient is the same as used in the SMTP
|
||||||
@@ -543,7 +507,7 @@ class Base(object):
|
|||||||
raise DisabledAction("ADDRCPT_PAR")
|
raise DisabledAction("ADDRCPT_PAR")
|
||||||
return self._ctx.addrcpt(rcpt,params)
|
return self._ctx.addrcpt(rcpt,params)
|
||||||
## Delete a recipient from the message.
|
## Delete a recipient from the message.
|
||||||
# Calls <a href="milter_api/smfi_delrcpt.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_delrcpt">
|
||||||
# smfi_delrcpt</a>.
|
# smfi_delrcpt</a>.
|
||||||
# The recipient should match one passed to the envrcpt callback.
|
# The recipient should match one passed to the envrcpt callback.
|
||||||
# The <code>Milter.DELRCPT</code> action flag must be set.
|
# The <code>Milter.DELRCPT</code> action flag must be set.
|
||||||
@@ -556,7 +520,7 @@ class Base(object):
|
|||||||
return self._ctx.delrcpt(rcpt)
|
return self._ctx.delrcpt(rcpt)
|
||||||
|
|
||||||
## Replace the message body.
|
## Replace the message body.
|
||||||
# Calls <a href="milter_api/smfi_replacebody.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_replacebody">
|
||||||
# smfi_replacebody</a>.
|
# smfi_replacebody</a>.
|
||||||
# The entire message body must be replaced.
|
# The entire message body must be replaced.
|
||||||
# Call repeatedly with blocks of data until the entire body is transferred.
|
# Call repeatedly with blocks of data until the entire body is transferred.
|
||||||
@@ -570,7 +534,7 @@ class Base(object):
|
|||||||
return self._ctx.replacebody(body)
|
return self._ctx.replacebody(body)
|
||||||
|
|
||||||
## Change the SMTP envelope sender address.
|
## Change the SMTP envelope sender address.
|
||||||
# Calls <a href="milter_api/smfi_chgfrom.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_chgfrom">
|
||||||
# smfi_chgfrom</a>.
|
# smfi_chgfrom</a>.
|
||||||
# The syntax of the sender is that same as used in the SMTP
|
# The syntax of the sender is that same as used in the SMTP
|
||||||
# MAIL FROM command (and as delivered to the envfrom callback),
|
# MAIL FROM command (and as delivered to the envfrom callback),
|
||||||
@@ -587,7 +551,7 @@ class Base(object):
|
|||||||
return self._ctx.chgfrom(sender,params)
|
return self._ctx.chgfrom(sender,params)
|
||||||
|
|
||||||
## Quarantine the message.
|
## Quarantine the message.
|
||||||
# Calls <a href="milter_api/smfi_quarantine.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_quarantine">
|
||||||
# smfi_quarantine</a>.
|
# smfi_quarantine</a>.
|
||||||
# When quarantined, a message goes into the mailq as if to be delivered,
|
# When quarantined, a message goes into the mailq as if to be delivered,
|
||||||
# but delivery is deferred until the message is unquarantined.
|
# but delivery is deferred until the message is unquarantined.
|
||||||
@@ -601,7 +565,7 @@ class Base(object):
|
|||||||
return self._ctx.quarantine(reason)
|
return self._ctx.quarantine(reason)
|
||||||
|
|
||||||
## Tell the MTA to wait a bit longer.
|
## Tell the MTA to wait a bit longer.
|
||||||
# Calls <a href="milter_api/smfi_progress.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_progress">
|
||||||
# smfi_progress</a>.
|
# smfi_progress</a>.
|
||||||
# Resets timeouts in the MTA that detect a "hung" milter.
|
# Resets timeouts in the MTA that detect a "hung" milter.
|
||||||
def progress(self):
|
def progress(self):
|
||||||
@@ -615,9 +579,9 @@ class Milter(Base):
|
|||||||
|
|
||||||
## Provide simple logging to sys.stdout
|
## Provide simple logging to sys.stdout
|
||||||
def log(self,*msg):
|
def log(self,*msg):
|
||||||
print('Milter:',end=None)
|
print 'Milter:',
|
||||||
for i in msg: print(i,end=None)
|
for i in msg: print i,
|
||||||
print()
|
print
|
||||||
|
|
||||||
@noreply
|
@noreply
|
||||||
def connect(self,hostname,family,hostaddr):
|
def connect(self,hostname,family,hostaddr):
|
||||||
|
|||||||
+5
-8
@@ -46,9 +46,8 @@
|
|||||||
# Copyright 2001,2002,2003,2004,2005 Business Management Systems, Inc.
|
# Copyright 2001,2002,2003,2004,2005 Business Management Systems, Inc.
|
||||||
# This code is under the GNU General Public License. See COPYING for details.
|
# This code is under the GNU General Public License. See COPYING for details.
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import time
|
import time
|
||||||
from Milter.plock import PLock
|
from plock import PLock
|
||||||
|
|
||||||
class AddrCache(object):
|
class AddrCache(object):
|
||||||
time_format = '%Y%b%d %H:%M:%S %Z'
|
time_format = '%Y%b%d %H:%M:%S %Z'
|
||||||
@@ -132,8 +131,8 @@ class AddrCache(object):
|
|||||||
if not ts or ts > too_old:
|
if not ts or ts > too_old:
|
||||||
return res
|
return res
|
||||||
del self.cache[lsender]
|
del self.cache[lsender]
|
||||||
raise KeyError(sender)
|
raise KeyError, sender
|
||||||
except KeyError as x:
|
except KeyError,x:
|
||||||
try:
|
try:
|
||||||
user,host = sender.split('@',1)
|
user,host = sender.split('@',1)
|
||||||
return self.__getitem__(host)
|
return self.__getitem__(host)
|
||||||
@@ -148,8 +147,7 @@ class AddrCache(object):
|
|||||||
if not ts: return # already permanent
|
if not ts: return # already permanent
|
||||||
self.cache[lsender] = (None,res)
|
self.cache[lsender] = (None,res)
|
||||||
if not res:
|
if not res:
|
||||||
with open(self.fname,'a') as fp:
|
print >>open(self.fname,'a'),sender
|
||||||
print(sender,file=fp)
|
|
||||||
|
|
||||||
def __setitem__(self,sender,res):
|
def __setitem__(self,sender,res):
|
||||||
lsender = sender.lower()
|
lsender = sender.lower()
|
||||||
@@ -157,8 +155,7 @@ class AddrCache(object):
|
|||||||
self.cache[lsender] = (now,res)
|
self.cache[lsender] = (now,res)
|
||||||
if not res and self.fname:
|
if not res and self.fname:
|
||||||
s = time.strftime(AddrCache.time_format,time.localtime(now))
|
s = time.strftime(AddrCache.time_format,time.localtime(now))
|
||||||
with open(self.fname,'a') as fp:
|
print >>open(self.fname,'a'),sender,s # log refreshed senders
|
||||||
print(sender,s,file=fp) # log refreshed senders
|
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self.cache)
|
return len(self.cache)
|
||||||
|
|||||||
+8
-10
@@ -1,7 +1,6 @@
|
|||||||
## @package Milter.dns
|
## @package Milter.dns
|
||||||
# Provide a higher level interface to pydns.
|
# Provide a higher level interface to pydns.
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import DNS
|
import DNS
|
||||||
from DNS import DNSError
|
from DNS import DNSError
|
||||||
|
|
||||||
@@ -15,9 +14,9 @@ MAX_CNAME = 10
|
|||||||
# @return a list of ((name,type),data) tuples
|
# @return a list of ((name,type),data) tuples
|
||||||
def DNSLookup(name, qtype):
|
def DNSLookup(name, qtype):
|
||||||
try:
|
try:
|
||||||
# To be thread safe, we create a fresh DnsRequest with
|
# To be thread safe, we create a fresh DnsRequest with
|
||||||
# each call. It would be more efficient to reuse
|
# each call. It would be more efficient to reuse
|
||||||
# a req object stored in a Session.
|
# a req object stored in a Session.
|
||||||
req = DNS.DnsRequest(name, qtype=qtype)
|
req = DNS.DnsRequest(name, qtype=qtype)
|
||||||
resp = req.req()
|
resp = req.req()
|
||||||
#resp.show()
|
#resp.show()
|
||||||
@@ -26,8 +25,8 @@ def DNSLookup(name, qtype):
|
|||||||
# A RR as dotted quad. For consistency, this driver should
|
# A RR as dotted quad. For consistency, this driver should
|
||||||
# return both as binary string.
|
# return both as binary string.
|
||||||
return [((a['name'], a['typename']), a['data']) for a in resp.answers]
|
return [((a['name'], a['typename']), a['data']) for a in resp.answers]
|
||||||
except IOError as x:
|
except IOError, x:
|
||||||
raise DNSError(str(x))
|
raise DNSError, str(x)
|
||||||
|
|
||||||
class Session(object):
|
class Session(object):
|
||||||
"""A Session object has a simple cache with no TTL that is valid
|
"""A Session object has a simple cache with no TTL that is valid
|
||||||
@@ -74,7 +73,6 @@ class Session(object):
|
|||||||
if name.endswith('.'): name = name[:-1]
|
if name.endswith('.'): name = name[:-1]
|
||||||
if not reduce(lambda x,y:x and 0 < len(y) < 64, name.split('.'),True):
|
if not reduce(lambda x,y:x and 0 < len(y) < 64, name.split('.'),True):
|
||||||
return [] # invalid DNS name (too long or empty)
|
return [] # invalid DNS name (too long or empty)
|
||||||
name = name.lower()
|
|
||||||
result = self.cache.get( (name, qtype) )
|
result = self.cache.get( (name, qtype) )
|
||||||
cname = None
|
cname = None
|
||||||
if result: return result
|
if result: return result
|
||||||
@@ -98,7 +96,7 @@ class Session(object):
|
|||||||
#return result # if too many == NX_DOMAIN
|
#return result # if too many == NX_DOMAIN
|
||||||
raise DNSError('Length of CNAME chain exceeds %d' % MAX_CNAME)
|
raise DNSError('Length of CNAME chain exceeds %d' % MAX_CNAME)
|
||||||
cnames[name] = cname
|
cnames[name] = cname
|
||||||
if cname.lower().rstrip('.') in cnames:
|
if cname in cnames:
|
||||||
raise DNSError('CNAME loop')
|
raise DNSError('CNAME loop')
|
||||||
result = self.dns(cname, qtype, cnames=cnames)
|
result = self.dns(cname, qtype, cnames=cnames)
|
||||||
if result:
|
if result:
|
||||||
@@ -121,5 +119,5 @@ if __name__ == '__main__':
|
|||||||
import sys
|
import sys
|
||||||
s = Session()
|
s = Session()
|
||||||
for n,t in zip(*[iter(sys.argv[1:])]*2):
|
for n,t in zip(*[iter(sys.argv[1:])]*2):
|
||||||
print(n,t)
|
print n,t
|
||||||
print(s.dns(n,t))
|
print s.dns(n,t)
|
||||||
|
|||||||
+6
-7
@@ -69,7 +69,6 @@
|
|||||||
# a DSN or use a null MAIL FROM with an email address obtained from
|
# a DSN or use a null MAIL FROM with an email address obtained from
|
||||||
# anywhere else.
|
# anywhere else.
|
||||||
#
|
#
|
||||||
from __future__ import print_function
|
|
||||||
import smtplib
|
import smtplib
|
||||||
import socket
|
import socket
|
||||||
from email.Message import Message
|
from email.Message import Message
|
||||||
@@ -142,13 +141,13 @@ def send_dsn(mailfrom,receiver,msg=None,timeout=600,session=None,ourfrom=''):
|
|||||||
if badrcpts:
|
if badrcpts:
|
||||||
return badrcpts
|
return badrcpts
|
||||||
return None # success
|
return None # success
|
||||||
except smtplib.SMTPRecipientsRefused as x:
|
except smtplib.SMTPRecipientsRefused,x:
|
||||||
if len(x.recipients) == 1:
|
if len(x.recipients) == 1:
|
||||||
return x.recipients.values()[0] # permanent error
|
return x.recipients.values()[0] # permanent error
|
||||||
return x.recipients
|
return x.recipients
|
||||||
except smtplib.SMTPSenderRefused as x:
|
except smtplib.SMTPSenderRefused,x:
|
||||||
return x.args[:2] # does not accept DSN
|
return x.args[:2] # does not accept DSN
|
||||||
except smtplib.SMTPDataError as x:
|
except smtplib.SMTPDataError,x:
|
||||||
return x.args # permanent error
|
return x.args # permanent error
|
||||||
except smtplib.SMTPException:
|
except smtplib.SMTPException:
|
||||||
pass # any other error, try next MX
|
pass # any other error, try next MX
|
||||||
@@ -231,6 +230,6 @@ Subject: Test
|
|||||||
Test DSN template
|
Test DSN template
|
||||||
"""
|
"""
|
||||||
)
|
)
|
||||||
print(msg.as_string())
|
print msg.as_string()
|
||||||
# print(send_dsn(f,msg.as_string()))
|
# print send_dsn(f,msg.as_string())
|
||||||
# print(send_dsn(q.s,'mail.example.com',msg.as_string()))
|
# print send_dsn(q.s,'mail.example.com',msg.as_string())
|
||||||
|
|||||||
+4
-5
@@ -9,7 +9,6 @@
|
|||||||
# wiley-268-8196.roadrunner.nf.net at ('205.251.174.46', 4810)
|
# wiley-268-8196.roadrunner.nf.net at ('205.251.174.46', 4810)
|
||||||
# cbl-sd-02-79.aster.com.do at ('200.88.62.79', 4153)
|
# cbl-sd-02-79.aster.com.do at ('200.88.62.79', 4153)
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import re
|
import re
|
||||||
|
|
||||||
ip3 = re.compile('[0-9]{1,3}')
|
ip3 = re.compile('[0-9]{1,3}')
|
||||||
@@ -54,11 +53,11 @@ def is_dynip(host,addr):
|
|||||||
if host.find(addr) >= 0: return True
|
if host.find(addr) >= 0: return True
|
||||||
if addr.find(':') >= 0: return False # IP6
|
if addr.find(':') >= 0: return False # IP6
|
||||||
a = addr.split('.')
|
a = addr.split('.')
|
||||||
ia = list(map(int,a))
|
ia = map(int,a)
|
||||||
h = host
|
h = host
|
||||||
m = ip3.findall(host)
|
m = ip3.findall(host)
|
||||||
if m:
|
if m:
|
||||||
g = list(map(int,m))[:4]
|
g = map(int,m)[:4]
|
||||||
ia3 = (ia[1:],ia[:3])
|
ia3 = (ia[1:],ia[:3])
|
||||||
if g[-3:] in ia3: return True
|
if g[-3:] in ia3: return True
|
||||||
if g[0] == ia[3] and g[1:3] == ia[:2]: return True
|
if g[0] == ia[3] and g[1:3] == ia[:2]: return True
|
||||||
@@ -92,6 +91,6 @@ if __name__ == '__main__':
|
|||||||
if ip in seen: continue
|
if ip in seen: continue
|
||||||
seen.add(ip)
|
seen.add(ip)
|
||||||
if is_dynip(host,ip):
|
if is_dynip(host,ip):
|
||||||
print('%s\t%s DYN' % (ip,host))
|
print '%s\t%s DYN' % (ip,host)
|
||||||
else:
|
else:
|
||||||
print('%s\t%s' % (ip,host))
|
print '%s\t%s' % (ip,host)
|
||||||
|
|||||||
+1
-2
@@ -1,4 +1,3 @@
|
|||||||
from __future__ import print_function
|
|
||||||
import time
|
import time
|
||||||
import shelve
|
import shelve
|
||||||
import thread
|
import thread
|
||||||
@@ -59,7 +58,7 @@ class Greylist(object):
|
|||||||
cnt = 0
|
cnt = 0
|
||||||
dbp = self.dbp
|
dbp = self.dbp
|
||||||
for key, r in dbp.iteritems():
|
for key, r in dbp.iteritems():
|
||||||
#print(key,r,time.ctime(now))
|
#print key,r,time.ctime(now)
|
||||||
if now > r.lastseen + self.greylist_retain:
|
if now > r.lastseen + self.greylist_retain:
|
||||||
self.lock.acquire()
|
self.lock.acquire()
|
||||||
try:
|
try:
|
||||||
|
|||||||
+1
-4
@@ -2,10 +2,7 @@ import time
|
|||||||
import logging
|
import logging
|
||||||
import urllib
|
import urllib
|
||||||
import sqlite3
|
import sqlite3
|
||||||
try:
|
import thread
|
||||||
import thread
|
|
||||||
except:
|
|
||||||
import _thread as thread
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
log = logging.getLogger('milter.greylist')
|
log = logging.getLogger('milter.greylist')
|
||||||
|
|||||||
+3
-3
@@ -11,7 +11,7 @@ class PLock(object):
|
|||||||
self.basename = basename
|
self.basename = basename
|
||||||
self.fp = None
|
self.fp = None
|
||||||
|
|
||||||
def lock(self,lockname=None,mode=0o660,strict_perms=False):
|
def lock(self,lockname=None,mode=0660,strict_perms=False):
|
||||||
"Start an update transaction. Return FILE to write new version."
|
"Start an update transaction. Return FILE to write new version."
|
||||||
self.unlock()
|
self.unlock()
|
||||||
if not lockname:
|
if not lockname:
|
||||||
@@ -21,7 +21,7 @@ class PLock(object):
|
|||||||
st = os.stat(self.basename)
|
st = os.stat(self.basename)
|
||||||
mode |= st.st_mode
|
mode |= st.st_mode
|
||||||
except OSError: pass
|
except OSError: pass
|
||||||
u = os.umask(0o2)
|
u = os.umask(0002)
|
||||||
try:
|
try:
|
||||||
fd = os.open(lockname,os.O_WRONLY+os.O_CREAT+os.O_EXCL,mode)
|
fd = os.open(lockname,os.O_WRONLY+os.O_CREAT+os.O_EXCL,mode)
|
||||||
finally:
|
finally:
|
||||||
@@ -46,7 +46,7 @@ class PLock(object):
|
|||||||
def commit(self,backname=None):
|
def commit(self,backname=None):
|
||||||
"Commit update transaction with optional backup file."
|
"Commit update transaction with optional backup file."
|
||||||
if not self.fp:
|
if not self.fp:
|
||||||
raise IOError("File not locked")
|
raise IOError,"File not locked"
|
||||||
self.fp.close()
|
self.fp.close()
|
||||||
self.fp = None
|
self.fp = None
|
||||||
if backname:
|
if backname:
|
||||||
|
|||||||
+3
-4
@@ -6,7 +6,6 @@ This module is free software, and you may redistribute it and/or modify
|
|||||||
it under the same terms as Python itself, so long as this copyright message
|
it under the same terms as Python itself, so long as this copyright message
|
||||||
and disclaimer are retained in their original form.
|
and disclaimer are retained in their original form.
|
||||||
"""
|
"""
|
||||||
from __future__ import print_function
|
|
||||||
import struct
|
import struct
|
||||||
#from spf import RE_IP4
|
#from spf import RE_IP4
|
||||||
import re
|
import re
|
||||||
@@ -81,11 +80,11 @@ def inet_pton(p):
|
|||||||
(0, 0, 0, 0, 0, 65535, 258, 772)
|
(0, 0, 0, 0, 0, 65535, 258, 772)
|
||||||
|
|
||||||
>>> try: inet_pton('::1.2.3.4.5')
|
>>> try: inet_pton('::1.2.3.4.5')
|
||||||
... except ValueError as x: print(x)
|
... except ValueError,x: print x
|
||||||
::1.2.3.4.5
|
::1.2.3.4.5
|
||||||
"""
|
"""
|
||||||
if p == '::':
|
if p == '::':
|
||||||
return b'\0'*16
|
return '\0'*16
|
||||||
s = p
|
s = p
|
||||||
m = RE_IP4.search(s)
|
m = RE_IP4.search(s)
|
||||||
try:
|
try:
|
||||||
@@ -115,4 +114,4 @@ def inet_pton(p):
|
|||||||
return struct.pack('!HHHHHHHH',
|
return struct.pack('!HHHHHHHH',
|
||||||
*[int(s,16) for s in a[0].split(':')])
|
*[int(s,16) for s in a[0].split(':')])
|
||||||
except ValueError: pass
|
except ValueError: pass
|
||||||
raise ValueError(p)
|
raise ValueError,p
|
||||||
|
|||||||
@@ -1,553 +0,0 @@
|
|||||||
|
|
||||||
"""A parser for SGML, using the derived class as a static DTD."""
|
|
||||||
|
|
||||||
# XXX This only supports those SGML features used by HTML.
|
|
||||||
|
|
||||||
# XXX There should be a way to distinguish between PCDATA (parsed
|
|
||||||
# character data -- the normal case), RCDATA (replaceable character
|
|
||||||
# data -- only char and entity references and end tags are special)
|
|
||||||
# and CDATA (character data -- only end tags are special). RCDATA is
|
|
||||||
# not supported at all.
|
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
|
|
||||||
try:
|
|
||||||
import _markupbase
|
|
||||||
except:
|
|
||||||
import markupbase as _markupbase
|
|
||||||
import re
|
|
||||||
|
|
||||||
__all__ = ["SGMLParser", "SGMLParseError"]
|
|
||||||
|
|
||||||
# Regular expressions used for parsing
|
|
||||||
|
|
||||||
interesting = re.compile('[&<]')
|
|
||||||
incomplete = re.compile('&([a-zA-Z][a-zA-Z0-9]*|#[0-9]*)?|'
|
|
||||||
'<([a-zA-Z][^<>]*|'
|
|
||||||
'/([a-zA-Z][^<>]*)?|'
|
|
||||||
'![^<>]*)?')
|
|
||||||
|
|
||||||
entityref = re.compile('&([a-zA-Z][-.a-zA-Z0-9]*)[^a-zA-Z0-9]')
|
|
||||||
charref = re.compile('&#([0-9]+)[^0-9]')
|
|
||||||
|
|
||||||
starttagopen = re.compile('<[>a-zA-Z]')
|
|
||||||
shorttagopen = re.compile('<[a-zA-Z][-.a-zA-Z0-9]*/')
|
|
||||||
shorttag = re.compile('<([a-zA-Z][-.a-zA-Z0-9]*)/([^/]*)/')
|
|
||||||
piclose = re.compile('>')
|
|
||||||
endbracket = re.compile('[<>]')
|
|
||||||
tagfind = re.compile('[a-zA-Z][-_.a-zA-Z0-9]*')
|
|
||||||
attrfind = re.compile(
|
|
||||||
r'\s*([a-zA-Z_][-:.a-zA-Z_0-9]*)(\s*=\s*'
|
|
||||||
r'(\'[^\']*\'|"[^"]*"|[][\-a-zA-Z0-9./,:;+*%?!&$\(\)_#=~\'"@]*))?')
|
|
||||||
|
|
||||||
|
|
||||||
class SGMLParseError(RuntimeError):
|
|
||||||
"""Exception raised for all parse errors."""
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
# SGML parser base class -- find tags and call handler functions.
|
|
||||||
# Usage: p = SGMLParser(); p.feed(data); ...; p.close().
|
|
||||||
# The dtd is defined by deriving a class which defines methods
|
|
||||||
# with special names to handle tags: start_foo and end_foo to handle
|
|
||||||
# <foo> and </foo>, respectively, or do_foo to handle <foo> by itself.
|
|
||||||
# (Tags are converted to lower case for this purpose.) The data
|
|
||||||
# between tags is passed to the parser by calling self.handle_data()
|
|
||||||
# with some data as argument (the data may be split up in arbitrary
|
|
||||||
# chunks). Entity references are passed by calling
|
|
||||||
# self.handle_entityref() with the entity reference as argument.
|
|
||||||
|
|
||||||
class SGMLParser(_markupbase.ParserBase):
|
|
||||||
# Definition of entities -- derived classes may override
|
|
||||||
entity_or_charref = re.compile('&(?:'
|
|
||||||
'([a-zA-Z][-.a-zA-Z0-9]*)|#([0-9]+)'
|
|
||||||
')(;?)')
|
|
||||||
|
|
||||||
def __init__(self, verbose=0):
|
|
||||||
"""Initialize and reset this instance."""
|
|
||||||
self.verbose = verbose
|
|
||||||
self.reset()
|
|
||||||
|
|
||||||
def reset(self):
|
|
||||||
"""Reset this instance. Loses all unprocessed data."""
|
|
||||||
self.__starttag_text = None
|
|
||||||
self.rawdata = ''
|
|
||||||
self.stack = []
|
|
||||||
self.lasttag = '???'
|
|
||||||
self.nomoretags = 0
|
|
||||||
self.literal = 0
|
|
||||||
_markupbase.ParserBase.reset(self)
|
|
||||||
|
|
||||||
def setnomoretags(self):
|
|
||||||
"""Enter literal mode (CDATA) till EOF.
|
|
||||||
|
|
||||||
Intended for derived classes only.
|
|
||||||
"""
|
|
||||||
self.nomoretags = self.literal = 1
|
|
||||||
|
|
||||||
def setliteral(self, *args):
|
|
||||||
"""Enter literal mode (CDATA).
|
|
||||||
|
|
||||||
Intended for derived classes only.
|
|
||||||
"""
|
|
||||||
self.literal = 1
|
|
||||||
|
|
||||||
def feed(self, data):
|
|
||||||
"""Feed some data to the parser.
|
|
||||||
|
|
||||||
Call this as often as you want, with as little or as much text
|
|
||||||
as you want (may include '\n'). (This just saves the text,
|
|
||||||
all the processing is done by goahead().)
|
|
||||||
"""
|
|
||||||
|
|
||||||
self.rawdata = self.rawdata + data
|
|
||||||
self.goahead(0)
|
|
||||||
|
|
||||||
def close(self):
|
|
||||||
"""Handle the remaining data."""
|
|
||||||
self.goahead(1)
|
|
||||||
|
|
||||||
def error(self, message):
|
|
||||||
raise SGMLParseError(message)
|
|
||||||
|
|
||||||
# Internal -- handle data as far as reasonable. May leave state
|
|
||||||
# and data to be processed by a subsequent call. If 'end' is
|
|
||||||
# true, force handling all data as if followed by EOF marker.
|
|
||||||
def goahead(self, end):
|
|
||||||
rawdata = self.rawdata
|
|
||||||
i = 0
|
|
||||||
n = len(rawdata)
|
|
||||||
while i < n:
|
|
||||||
if self.nomoretags:
|
|
||||||
self.handle_data(rawdata[i:n])
|
|
||||||
i = n
|
|
||||||
break
|
|
||||||
match = interesting.search(rawdata, i)
|
|
||||||
if match: j = match.start()
|
|
||||||
else: j = n
|
|
||||||
if i < j:
|
|
||||||
self.handle_data(rawdata[i:j])
|
|
||||||
i = j
|
|
||||||
if i == n: break
|
|
||||||
if rawdata[i] == '<':
|
|
||||||
if starttagopen.match(rawdata, i):
|
|
||||||
if self.literal:
|
|
||||||
self.handle_data(rawdata[i])
|
|
||||||
i = i+1
|
|
||||||
continue
|
|
||||||
k = self.parse_starttag(i)
|
|
||||||
if k < 0: break
|
|
||||||
i = k
|
|
||||||
continue
|
|
||||||
if rawdata.startswith("</", i):
|
|
||||||
k = self.parse_endtag(i)
|
|
||||||
if k < 0: break
|
|
||||||
i = k
|
|
||||||
self.literal = 0
|
|
||||||
continue
|
|
||||||
if self.literal:
|
|
||||||
if n > (i + 1):
|
|
||||||
self.handle_data("<")
|
|
||||||
i = i+1
|
|
||||||
else:
|
|
||||||
# incomplete
|
|
||||||
break
|
|
||||||
continue
|
|
||||||
if rawdata.startswith("<!--", i):
|
|
||||||
# Strictly speaking, a comment is --.*--
|
|
||||||
# within a declaration tag <!...>.
|
|
||||||
# This should be removed,
|
|
||||||
# and comments handled only in parse_declaration.
|
|
||||||
k = self.parse_comment(i)
|
|
||||||
if k < 0: break
|
|
||||||
i = k
|
|
||||||
continue
|
|
||||||
if rawdata.startswith("<?", i):
|
|
||||||
k = self.parse_pi(i)
|
|
||||||
if k < 0: break
|
|
||||||
i = i+k
|
|
||||||
continue
|
|
||||||
if rawdata.startswith("<!", i):
|
|
||||||
# This is some sort of declaration; in "HTML as
|
|
||||||
# deployed," this should only be the document type
|
|
||||||
# declaration ("<!DOCTYPE html...>").
|
|
||||||
k = self.parse_declaration(i)
|
|
||||||
if k < 0: break
|
|
||||||
i = k
|
|
||||||
continue
|
|
||||||
elif rawdata[i] == '&':
|
|
||||||
if self.literal:
|
|
||||||
self.handle_data(rawdata[i])
|
|
||||||
i = i+1
|
|
||||||
continue
|
|
||||||
match = charref.match(rawdata, i)
|
|
||||||
if match:
|
|
||||||
name = match.group(1)
|
|
||||||
self.handle_charref(name)
|
|
||||||
i = match.end(0)
|
|
||||||
if rawdata[i-1] != ';': i = i-1
|
|
||||||
continue
|
|
||||||
match = entityref.match(rawdata, i)
|
|
||||||
if match:
|
|
||||||
name = match.group(1)
|
|
||||||
self.handle_entityref(name)
|
|
||||||
i = match.end(0)
|
|
||||||
if rawdata[i-1] != ';': i = i-1
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
self.error('neither < nor & ??')
|
|
||||||
# We get here only if incomplete matches but
|
|
||||||
# nothing else
|
|
||||||
match = incomplete.match(rawdata, i)
|
|
||||||
if not match:
|
|
||||||
self.handle_data(rawdata[i])
|
|
||||||
i = i+1
|
|
||||||
continue
|
|
||||||
j = match.end(0)
|
|
||||||
if j == n:
|
|
||||||
break # Really incomplete
|
|
||||||
self.handle_data(rawdata[i:j])
|
|
||||||
i = j
|
|
||||||
# end while
|
|
||||||
if end and i < n:
|
|
||||||
self.handle_data(rawdata[i:n])
|
|
||||||
i = n
|
|
||||||
self.rawdata = rawdata[i:]
|
|
||||||
# XXX if end: check for empty stack
|
|
||||||
|
|
||||||
# Extensions for the DOCTYPE scanner:
|
|
||||||
_decl_otherchars = '='
|
|
||||||
|
|
||||||
# Internal -- parse processing instr, return length or -1 if not terminated
|
|
||||||
def parse_pi(self, i):
|
|
||||||
rawdata = self.rawdata
|
|
||||||
if rawdata[i:i+2] != '<?':
|
|
||||||
self.error('unexpected call to parse_pi()')
|
|
||||||
match = piclose.search(rawdata, i+2)
|
|
||||||
if not match:
|
|
||||||
return -1
|
|
||||||
j = match.start(0)
|
|
||||||
self.handle_pi(rawdata[i+2: j])
|
|
||||||
j = match.end(0)
|
|
||||||
return j-i
|
|
||||||
|
|
||||||
def get_starttag_text(self):
|
|
||||||
return self.__starttag_text
|
|
||||||
|
|
||||||
# Internal -- handle starttag, return length or -1 if not terminated
|
|
||||||
def parse_starttag(self, i):
|
|
||||||
self.__starttag_text = None
|
|
||||||
start_pos = i
|
|
||||||
rawdata = self.rawdata
|
|
||||||
if shorttagopen.match(rawdata, i):
|
|
||||||
# SGML shorthand: <tag/data/ == <tag>data</tag>
|
|
||||||
# XXX Can data contain &... (entity or char refs)?
|
|
||||||
# XXX Can data contain < or > (tag characters)?
|
|
||||||
# XXX Can there be whitespace before the first /?
|
|
||||||
match = shorttag.match(rawdata, i)
|
|
||||||
if not match:
|
|
||||||
return -1
|
|
||||||
tag, data = match.group(1, 2)
|
|
||||||
self.__starttag_text = '<%s/' % tag
|
|
||||||
tag = tag.lower()
|
|
||||||
k = match.end(0)
|
|
||||||
self.finish_shorttag(tag, data)
|
|
||||||
self.__starttag_text = rawdata[start_pos:match.end(1) + 1]
|
|
||||||
return k
|
|
||||||
# XXX The following should skip matching quotes (' or ")
|
|
||||||
# As a shortcut way to exit, this isn't so bad, but shouldn't
|
|
||||||
# be used to locate the actual end of the start tag since the
|
|
||||||
# < or > characters may be embedded in an attribute value.
|
|
||||||
match = endbracket.search(rawdata, i+1)
|
|
||||||
if not match:
|
|
||||||
return -1
|
|
||||||
j = match.start(0)
|
|
||||||
# Now parse the data between i+1 and j into a tag and attrs
|
|
||||||
attrs = []
|
|
||||||
if rawdata[i:i+2] == '<>':
|
|
||||||
# SGML shorthand: <> == <last open tag seen>
|
|
||||||
k = j
|
|
||||||
tag = self.lasttag
|
|
||||||
else:
|
|
||||||
match = tagfind.match(rawdata, i+1)
|
|
||||||
if not match:
|
|
||||||
self.error('unexpected call to parse_starttag')
|
|
||||||
k = match.end(0)
|
|
||||||
tag = rawdata[i+1:k].lower()
|
|
||||||
self.lasttag = tag
|
|
||||||
while k < j:
|
|
||||||
match = attrfind.match(rawdata, k)
|
|
||||||
if not match: break
|
|
||||||
attrname, rest, attrvalue = match.group(1, 2, 3)
|
|
||||||
if not rest:
|
|
||||||
attrvalue = attrname
|
|
||||||
else:
|
|
||||||
if (attrvalue[:1] == "'" == attrvalue[-1:] or
|
|
||||||
attrvalue[:1] == '"' == attrvalue[-1:]):
|
|
||||||
# strip quotes
|
|
||||||
attrvalue = attrvalue[1:-1]
|
|
||||||
attrvalue = self.entity_or_charref.sub(
|
|
||||||
self._convert_ref, attrvalue)
|
|
||||||
attrs.append((attrname.lower(), attrvalue))
|
|
||||||
k = match.end(0)
|
|
||||||
if rawdata[j] == '>':
|
|
||||||
j = j+1
|
|
||||||
self.__starttag_text = rawdata[start_pos:j]
|
|
||||||
self.finish_starttag(tag, attrs)
|
|
||||||
return j
|
|
||||||
|
|
||||||
# Internal -- convert entity or character reference
|
|
||||||
def _convert_ref(self, match):
|
|
||||||
if match.group(2):
|
|
||||||
return self.convert_charref(match.group(2)) or \
|
|
||||||
'&#%s%s' % match.groups()[1:]
|
|
||||||
elif match.group(3):
|
|
||||||
return self.convert_entityref(match.group(1)) or \
|
|
||||||
'&%s;' % match.group(1)
|
|
||||||
else:
|
|
||||||
return '&%s' % match.group(1)
|
|
||||||
|
|
||||||
# Internal -- parse endtag
|
|
||||||
def parse_endtag(self, i):
|
|
||||||
rawdata = self.rawdata
|
|
||||||
match = endbracket.search(rawdata, i+1)
|
|
||||||
if not match:
|
|
||||||
return -1
|
|
||||||
j = match.start(0)
|
|
||||||
tag = rawdata[i+2:j].strip().lower()
|
|
||||||
if rawdata[j] == '>':
|
|
||||||
j = j+1
|
|
||||||
self.finish_endtag(tag)
|
|
||||||
return j
|
|
||||||
|
|
||||||
# Internal -- finish parsing of <tag/data/ (same as <tag>data</tag>)
|
|
||||||
def finish_shorttag(self, tag, data):
|
|
||||||
self.finish_starttag(tag, [])
|
|
||||||
self.handle_data(data)
|
|
||||||
self.finish_endtag(tag)
|
|
||||||
|
|
||||||
# Internal -- finish processing of start tag
|
|
||||||
# Return -1 for unknown tag, 0 for open-only tag, 1 for balanced tag
|
|
||||||
def finish_starttag(self, tag, attrs):
|
|
||||||
try:
|
|
||||||
method = getattr(self, 'start_' + tag)
|
|
||||||
except AttributeError:
|
|
||||||
try:
|
|
||||||
method = getattr(self, 'do_' + tag)
|
|
||||||
except AttributeError:
|
|
||||||
self.unknown_starttag(tag, attrs)
|
|
||||||
return -1
|
|
||||||
else:
|
|
||||||
self.handle_starttag(tag, method, attrs)
|
|
||||||
return 0
|
|
||||||
else:
|
|
||||||
self.stack.append(tag)
|
|
||||||
self.handle_starttag(tag, method, attrs)
|
|
||||||
return 1
|
|
||||||
|
|
||||||
# Internal -- finish processing of end tag
|
|
||||||
def finish_endtag(self, tag):
|
|
||||||
if not tag:
|
|
||||||
found = len(self.stack) - 1
|
|
||||||
if found < 0:
|
|
||||||
self.unknown_endtag(tag)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
if tag not in self.stack:
|
|
||||||
try:
|
|
||||||
method = getattr(self, 'end_' + tag)
|
|
||||||
except AttributeError:
|
|
||||||
self.unknown_endtag(tag)
|
|
||||||
else:
|
|
||||||
self.report_unbalanced(tag)
|
|
||||||
return
|
|
||||||
found = len(self.stack)
|
|
||||||
for i in range(found):
|
|
||||||
if self.stack[i] == tag: found = i
|
|
||||||
while len(self.stack) > found:
|
|
||||||
tag = self.stack[-1]
|
|
||||||
try:
|
|
||||||
method = getattr(self, 'end_' + tag)
|
|
||||||
except AttributeError:
|
|
||||||
method = None
|
|
||||||
if method:
|
|
||||||
self.handle_endtag(tag, method)
|
|
||||||
else:
|
|
||||||
self.unknown_endtag(tag)
|
|
||||||
del self.stack[-1]
|
|
||||||
|
|
||||||
# Overridable -- handle start tag
|
|
||||||
def handle_starttag(self, tag, method, attrs):
|
|
||||||
method(attrs)
|
|
||||||
|
|
||||||
# Overridable -- handle end tag
|
|
||||||
def handle_endtag(self, tag, method):
|
|
||||||
method()
|
|
||||||
|
|
||||||
# Example -- report an unbalanced </...> tag.
|
|
||||||
def report_unbalanced(self, tag):
|
|
||||||
if self.verbose:
|
|
||||||
print('*** Unbalanced </' + tag + '>')
|
|
||||||
print('*** Stack:', self.stack)
|
|
||||||
|
|
||||||
def convert_charref(self, name):
|
|
||||||
"""Convert character reference, may be overridden."""
|
|
||||||
try:
|
|
||||||
n = int(name)
|
|
||||||
except ValueError:
|
|
||||||
return
|
|
||||||
if not 0 <= n <= 127:
|
|
||||||
return
|
|
||||||
return self.convert_codepoint(n)
|
|
||||||
|
|
||||||
def convert_codepoint(self, codepoint):
|
|
||||||
return chr(codepoint)
|
|
||||||
|
|
||||||
def handle_charref(self, name):
|
|
||||||
"""Handle character reference, no need to override."""
|
|
||||||
replacement = self.convert_charref(name)
|
|
||||||
if replacement is None:
|
|
||||||
self.unknown_charref(name)
|
|
||||||
else:
|
|
||||||
self.handle_data(replacement)
|
|
||||||
|
|
||||||
# Definition of entities -- derived classes may override
|
|
||||||
entitydefs = \
|
|
||||||
{'lt': '<', 'gt': '>', 'amp': '&', 'quot': '"', 'apos': '\''}
|
|
||||||
|
|
||||||
def convert_entityref(self, name):
|
|
||||||
"""Convert entity references.
|
|
||||||
|
|
||||||
As an alternative to overriding this method; one can tailor the
|
|
||||||
results by setting up the self.entitydefs mapping appropriately.
|
|
||||||
"""
|
|
||||||
table = self.entitydefs
|
|
||||||
if name in table:
|
|
||||||
return table[name]
|
|
||||||
else:
|
|
||||||
return
|
|
||||||
|
|
||||||
def handle_entityref(self, name):
|
|
||||||
"""Handle entity references, no need to override."""
|
|
||||||
replacement = self.convert_entityref(name)
|
|
||||||
if replacement is None:
|
|
||||||
self.unknown_entityref(name)
|
|
||||||
else:
|
|
||||||
self.handle_data(replacement)
|
|
||||||
|
|
||||||
# Example -- handle data, should be overridden
|
|
||||||
def handle_data(self, data):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# Example -- handle comment, could be overridden
|
|
||||||
def handle_comment(self, data):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# Example -- handle declaration, could be overridden
|
|
||||||
def handle_decl(self, decl):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# Example -- handle processing instruction, could be overridden
|
|
||||||
def handle_pi(self, data):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# To be overridden -- handlers for unknown objects
|
|
||||||
def unknown_starttag(self, tag, attrs): pass
|
|
||||||
def unknown_endtag(self, tag): pass
|
|
||||||
def unknown_charref(self, ref): pass
|
|
||||||
def unknown_entityref(self, ref): pass
|
|
||||||
|
|
||||||
|
|
||||||
class TestSGMLParser(SGMLParser):
|
|
||||||
|
|
||||||
def __init__(self, verbose=0):
|
|
||||||
self.testdata = ""
|
|
||||||
SGMLParser.__init__(self, verbose)
|
|
||||||
|
|
||||||
def handle_data(self, data):
|
|
||||||
self.testdata = self.testdata + data
|
|
||||||
if len(repr(self.testdata)) >= 70:
|
|
||||||
self.flush()
|
|
||||||
|
|
||||||
def flush(self):
|
|
||||||
data = self.testdata
|
|
||||||
if data:
|
|
||||||
self.testdata = ""
|
|
||||||
print('data:', repr(data))
|
|
||||||
|
|
||||||
def handle_comment(self, data):
|
|
||||||
self.flush()
|
|
||||||
r = repr(data)
|
|
||||||
if len(r) > 68:
|
|
||||||
r = r[:32] + '...' + r[-32:]
|
|
||||||
print('comment:', r)
|
|
||||||
|
|
||||||
def unknown_starttag(self, tag, attrs):
|
|
||||||
self.flush()
|
|
||||||
if not attrs:
|
|
||||||
print('start tag: <' + tag + '>')
|
|
||||||
else:
|
|
||||||
print('start tag: <' + tag, end=' ')
|
|
||||||
for name, value in attrs:
|
|
||||||
print(name + '=' + '"' + value + '"', end=' ')
|
|
||||||
print('>')
|
|
||||||
|
|
||||||
def unknown_endtag(self, tag):
|
|
||||||
self.flush()
|
|
||||||
print('end tag: </' + tag + '>')
|
|
||||||
|
|
||||||
def unknown_entityref(self, ref):
|
|
||||||
self.flush()
|
|
||||||
print('*** unknown entity ref: &' + ref + ';')
|
|
||||||
|
|
||||||
def unknown_charref(self, ref):
|
|
||||||
self.flush()
|
|
||||||
print('*** unknown char ref: &#' + ref + ';')
|
|
||||||
|
|
||||||
def unknown_decl(self, data):
|
|
||||||
self.flush()
|
|
||||||
print('*** unknown decl: [' + data + ']')
|
|
||||||
|
|
||||||
def close(self):
|
|
||||||
SGMLParser.close(self)
|
|
||||||
self.flush()
|
|
||||||
|
|
||||||
|
|
||||||
def test(args = None):
|
|
||||||
import sys
|
|
||||||
|
|
||||||
if args is None:
|
|
||||||
args = sys.argv[1:]
|
|
||||||
|
|
||||||
if args and args[0] == '-s':
|
|
||||||
args = args[1:]
|
|
||||||
klass = SGMLParser
|
|
||||||
else:
|
|
||||||
klass = TestSGMLParser
|
|
||||||
|
|
||||||
if args:
|
|
||||||
file = args[0]
|
|
||||||
else:
|
|
||||||
file = 'test.html'
|
|
||||||
|
|
||||||
if file == '-':
|
|
||||||
f = sys.stdin
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
f = open(file, 'r')
|
|
||||||
except IOError as msg:
|
|
||||||
print(file, ":", msg)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
data = f.read()
|
|
||||||
if f is not sys.stdin:
|
|
||||||
f.close()
|
|
||||||
|
|
||||||
x = klass()
|
|
||||||
for c in data:
|
|
||||||
x.feed(c)
|
|
||||||
x.close()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
test()
|
|
||||||
+41
-89
@@ -1,12 +1,8 @@
|
|||||||
## @package Milter.test
|
## @package Milter.test
|
||||||
# A test framework for milters
|
# A test framework for milters
|
||||||
|
|
||||||
from __future__ import print_function
|
import rfc822
|
||||||
import mime
|
import StringIO
|
||||||
try:
|
|
||||||
from io import BytesIO
|
|
||||||
except:
|
|
||||||
from StringIO import StringIO as BytesIO
|
|
||||||
import Milter
|
import Milter
|
||||||
|
|
||||||
Milter.NOREPLY = Milter.CONTINUE
|
Milter.NOREPLY = Milter.CONTINUE
|
||||||
@@ -14,15 +10,12 @@ Milter.NOREPLY = Milter.CONTINUE
|
|||||||
## Test mixin for unit testing %milter applications.
|
## Test mixin for unit testing %milter applications.
|
||||||
# This mixin overrides many Milter.MilterBase methods
|
# This mixin overrides many Milter.MilterBase methods
|
||||||
# with stub versions that simply record what was done.
|
# with stub versions that simply record what was done.
|
||||||
# @deprecated Use Milter.test.TestCtx
|
|
||||||
# @since 0.9.8
|
# @since 0.9.8
|
||||||
class TestBase(object):
|
class TestBase(object):
|
||||||
|
|
||||||
def __init__(self,logfile='test/milter.log'):
|
def __init__(self,logfile='test/milter.log'):
|
||||||
self._protocol = 0
|
self._protocol = 0
|
||||||
self.logfp = open(logfile,"a")
|
self.logfp = open(logfile,"a")
|
||||||
## The MAIL FROM for the current email being fed to the %milter
|
|
||||||
self._sender = None
|
|
||||||
## List of recipients deleted
|
## List of recipients deleted
|
||||||
self._delrcpt = []
|
self._delrcpt = []
|
||||||
## List of recipients added
|
## List of recipients added
|
||||||
@@ -35,20 +28,15 @@ class TestBase(object):
|
|||||||
self._bodyreplaced = False
|
self._bodyreplaced = False
|
||||||
## True if the %milter changed any headers.
|
## True if the %milter changed any headers.
|
||||||
self._headerschanged = False
|
self._headerschanged = False
|
||||||
## True if the %milter changed the envelope from.
|
|
||||||
self._envfromchanged = False
|
|
||||||
## Reply codes and messages set by the %milter
|
## Reply codes and messages set by the %milter
|
||||||
self._reply = None
|
self._reply = None
|
||||||
## The rfc822 message object for the current email being fed to the %milter.
|
## The rfc822 message object for the current email being fed to the %milter.
|
||||||
self._msg = None
|
self._msg = None
|
||||||
## The protocol stage for macros returned
|
|
||||||
self._stage = None
|
|
||||||
## The macros returned by protocol stage
|
|
||||||
self._symlist = [ None, None, None, None, None, None, None ]
|
self._symlist = [ None, None, None, None, None, None, None ]
|
||||||
|
|
||||||
def log(self,*msg):
|
def log(self,*msg):
|
||||||
for i in msg: print(i,file=self.logfp,end=None)
|
for i in msg: print >>self.logfp, i,
|
||||||
print(file=self.logfp)
|
print >>self.logfp
|
||||||
|
|
||||||
## Set a macro value.
|
## Set a macro value.
|
||||||
# These are retrieved by the %milter with getsymval.
|
# These are retrieved by the %milter with getsymval.
|
||||||
@@ -58,40 +46,21 @@ class TestBase(object):
|
|||||||
self._macros[name] = val
|
self._macros[name] = val
|
||||||
|
|
||||||
def getsymval(self,name):
|
def getsymval(self,name):
|
||||||
stage = self._stage
|
# FIXME: track stage, and use _symlist
|
||||||
if stage >= 0:
|
return self._macros.get(name,'')
|
||||||
syms = self._symlist[stage]
|
|
||||||
if syms is not None and name not in syms:
|
|
||||||
return None
|
|
||||||
return self._macros.get(name,None)
|
|
||||||
|
|
||||||
def replacebody(self,chunk):
|
def replacebody(self,chunk):
|
||||||
if self._body:
|
if self._body:
|
||||||
self._body.write(chunk)
|
self._body.write(chunk)
|
||||||
self._bodyreplaced = True
|
self._bodyreplaced = True
|
||||||
else:
|
else:
|
||||||
raise IOError("replacebody not called from eom()")
|
raise IOError,"replacebody not called from eom()"
|
||||||
|
|
||||||
def chgfrom(self,sender,params=None):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("chgfrom not called from eom()")
|
|
||||||
self.log('chgfrom: sender=%s' % (sender))
|
|
||||||
self._envfromchanged = True
|
|
||||||
self._sender = sender
|
|
||||||
|
|
||||||
# TODO: write implement quarantine()
|
|
||||||
def quarantine(self,reason):
|
|
||||||
raise NotImplemented
|
|
||||||
|
|
||||||
# TODO: measure time between milter calls
|
|
||||||
def progress(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# FIXME: rfc822 indexing does not really reflect the way chg/add header
|
# FIXME: rfc822 indexing does not really reflect the way chg/add header
|
||||||
# work for a %milter
|
# work for a %milter
|
||||||
def chgheader(self,field,idx,value):
|
def chgheader(self,field,idx,value):
|
||||||
if not self._body:
|
if not self._body:
|
||||||
raise IOError("chgheader not called from eom()")
|
raise IOError,"chgheader not called from eom()"
|
||||||
self.log('chgheader: %s[%d]=%s' % (field,idx,value))
|
self.log('chgheader: %s[%d]=%s' % (field,idx,value))
|
||||||
if value == '':
|
if value == '':
|
||||||
del self._msg[field]
|
del self._msg[field]
|
||||||
@@ -101,19 +70,19 @@ class TestBase(object):
|
|||||||
|
|
||||||
def addheader(self,field,value,idx=-1):
|
def addheader(self,field,value,idx=-1):
|
||||||
if not self._body:
|
if not self._body:
|
||||||
raise IOError("addheader not called from eom()")
|
raise IOError,"addheader not called from eom()"
|
||||||
self.log('addheader: %s=%s' % (field,value))
|
self.log('addheader: %s=%s' % (field,value))
|
||||||
self._msg[field] = value
|
self._msg[field] = value
|
||||||
self._headerschanged = True
|
self._headerschanged = True
|
||||||
|
|
||||||
def delrcpt(self,rcpt):
|
def delrcpt(self,rcpt):
|
||||||
if not self._body:
|
if not self._body:
|
||||||
raise IOError("delrcpt not called from eom()")
|
raise IOError,"delrcpt not called from eom()"
|
||||||
self._delrcpt.append(rcpt)
|
self._delrcpt.append(rcpt)
|
||||||
|
|
||||||
def addrcpt(self,rcpt):
|
def addrcpt(self,rcpt):
|
||||||
if not self._body:
|
if not self._body:
|
||||||
raise IOError("addrcpt not called from eom()")
|
raise IOError,"addrcpt not called from eom()"
|
||||||
self._addrcpt.append(rcpt)
|
self._addrcpt.append(rcpt)
|
||||||
|
|
||||||
## Save the reply codes and messages in self._reply.
|
## Save the reply codes and messages in self._reply.
|
||||||
@@ -121,10 +90,7 @@ class TestBase(object):
|
|||||||
self._reply = (rcode,xcode) + msg
|
self._reply = (rcode,xcode) + msg
|
||||||
|
|
||||||
def setsymlist(self,stage,macros):
|
def setsymlist(self,stage,macros):
|
||||||
if not self._actions & Milter.SETSYMLIST:
|
if not self._actions & SETSYMLIST: raise DisabledAction("SETSYMLIST")
|
||||||
raise DisabledAction("SETSYMLIST")
|
|
||||||
if self._stage != -1:
|
|
||||||
raise RuntimeError("setsymlist may only be called from negotiate")
|
|
||||||
# not used yet, but just for grins we save the data
|
# not used yet, but just for grins we save the data
|
||||||
a = []
|
a = []
|
||||||
for m in macros:
|
for m in macros:
|
||||||
@@ -132,14 +98,9 @@ class TestBase(object):
|
|||||||
m = m.encode('utf8')
|
m = m.encode('utf8')
|
||||||
except: pass
|
except: pass
|
||||||
try:
|
try:
|
||||||
m = m.split(b' ')
|
m = m.split(' ')
|
||||||
except: pass
|
except: pass
|
||||||
a += m
|
a += m
|
||||||
if len(a) > 5:
|
|
||||||
raise ValueError('setsymlist limited to 5 macros by MTA')
|
|
||||||
if self._symlist[stage] is not None:
|
|
||||||
raise ValueError('setsymlist already called for stage:'+stage)
|
|
||||||
print('setsymlist',stage,a)
|
|
||||||
self._symlist[stage] = set(a)
|
self._symlist[stage] = set(a)
|
||||||
|
|
||||||
## Feed a file like object to the %milter. Calls envfrom, envrcpt for
|
## Feed a file like object to the %milter. Calls envfrom, envrcpt for
|
||||||
@@ -158,51 +119,48 @@ class TestBase(object):
|
|||||||
self._bodyreplaced = False
|
self._bodyreplaced = False
|
||||||
self._headerschanged = False
|
self._headerschanged = False
|
||||||
self._reply = None
|
self._reply = None
|
||||||
self._sender = '<%s>'%sender
|
msg = rfc822.Message(fp)
|
||||||
msg = mime.message_from_file(fp)
|
rc = self.envfrom('<%s>'%sender)
|
||||||
# envfrom
|
|
||||||
self._stage = Milter.M_ENVFROM
|
|
||||||
rc = self.envfrom(self._sender)
|
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
if rc != Milter.CONTINUE: return rc
|
||||||
# envrcpt
|
|
||||||
for rcpt in (rcpt,) + rcpts:
|
for rcpt in (rcpt,) + rcpts:
|
||||||
self._stage = Milter.M_ENVRCPT
|
|
||||||
rc = self.envrcpt('<%s>'%rcpt)
|
rc = self.envrcpt('<%s>'%rcpt)
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
if rc != Milter.CONTINUE: return rc
|
||||||
# data
|
line = None
|
||||||
self._stage = Milter.M_DATA
|
for h in msg.headers:
|
||||||
rc = self.data()
|
if h[:1].isspace():
|
||||||
self._stage = None
|
line = line + h
|
||||||
if rc != Milter.CONTINUE: return rc
|
continue
|
||||||
# header
|
if not line:
|
||||||
for h,val in msg.items():
|
line = h
|
||||||
rc = self.header(h,val)
|
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
|
if rc != Milter.CONTINUE: return rc
|
||||||
# eoh
|
|
||||||
self._stage = Milter.M_EOH
|
|
||||||
rc = self.eoh()
|
rc = self.eoh()
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
if rc != Milter.CONTINUE: return rc
|
||||||
# body
|
|
||||||
header,body = msg.as_bytes().split(b'\n\n',1)
|
|
||||||
bfp = BytesIO(body)
|
|
||||||
while 1:
|
while 1:
|
||||||
buf = bfp.read(8192)
|
buf = fp.read(8192)
|
||||||
if len(buf) == 0: break
|
if len(buf) == 0: break
|
||||||
rc = self.body(buf)
|
rc = self.body(buf)
|
||||||
if rc != Milter.CONTINUE: return rc
|
if rc != Milter.CONTINUE: return rc
|
||||||
self._msg = msg
|
self._msg = msg
|
||||||
self._body = BytesIO()
|
self._body = StringIO.StringIO()
|
||||||
self._stage = Milter.M_EOM
|
|
||||||
rc = self.eom()
|
rc = self.eom()
|
||||||
self._stage = None
|
|
||||||
if self._bodyreplaced:
|
if self._bodyreplaced:
|
||||||
body = self._body.getvalue()
|
body = self._body.getvalue()
|
||||||
self._body = BytesIO()
|
else:
|
||||||
self._body.write(header)
|
msg.rewindbody()
|
||||||
self._body.write(b'\n\n')
|
body = msg.fp.read()
|
||||||
|
self._body = StringIO.StringIO()
|
||||||
|
self._body.writelines(msg.headers)
|
||||||
|
self._body.write('\n')
|
||||||
self._body.write(body)
|
self._body.write(body)
|
||||||
return rc
|
return rc
|
||||||
|
|
||||||
@@ -211,7 +169,7 @@ class TestBase(object):
|
|||||||
# @param sender MAIL FROM
|
# @param sender MAIL FROM
|
||||||
# @param rcpts RCPT TO, multiple recipients may be supplied
|
# @param rcpts RCPT TO, multiple recipients may be supplied
|
||||||
def feedMsg(self,fname,sender="spam@adv.com",*rcpts):
|
def feedMsg(self,fname,sender="spam@adv.com",*rcpts):
|
||||||
with open('test/'+fname,'rb') as fp:
|
with open('test/'+fname,'r') as fp:
|
||||||
return self.feedFile(fp,sender,*rcpts)
|
return self.feedFile(fp,sender,*rcpts)
|
||||||
|
|
||||||
## Call the connect and helo callbacks.
|
## Call the connect and helo callbacks.
|
||||||
@@ -222,19 +180,13 @@ class TestBase(object):
|
|||||||
def connect(self,host='localhost',helo='spamrelay',ip='1.2.3.4'):
|
def connect(self,host='localhost',helo='spamrelay',ip='1.2.3.4'):
|
||||||
self._body = None
|
self._body = None
|
||||||
self._bodyreplaced = False
|
self._bodyreplaced = False
|
||||||
self._setctx(None)
|
|
||||||
opts = [ Milter.CURR_ACTS,~0,0,0 ]
|
opts = [ Milter.CURR_ACTS,~0,0,0 ]
|
||||||
self._stage = -1
|
|
||||||
rc = self.negotiate(opts)
|
rc = self.negotiate(opts)
|
||||||
self._stage = Milter.M_CONNECT
|
|
||||||
rc = super(TestBase,self).connect(host,1,(ip,1234))
|
rc = super(TestBase,self).connect(host,1,(ip,1234))
|
||||||
if rc != Milter.CONTINUE:
|
if rc != Milter.CONTINUE:
|
||||||
self._stage = None
|
|
||||||
self.close()
|
self.close()
|
||||||
return rc
|
return rc
|
||||||
self._stage = Milter.M_HELO
|
|
||||||
rc = self.hello(helo)
|
rc = self.hello(helo)
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE:
|
if rc != Milter.CONTINUE:
|
||||||
self.close()
|
self.close()
|
||||||
return rc
|
return rc
|
||||||
|
|||||||
@@ -1,297 +0,0 @@
|
|||||||
## @package Milter.testctx
|
|
||||||
# A test framework for milters that replaces milterContext rather
|
|
||||||
# than Milter.Base. Since miltermodule.c doesn't currently export
|
|
||||||
# a way to query callbacks set (and we might want to run without
|
|
||||||
# loading milter), we assume the callbacks set by Milter.runmilter().
|
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
from socket import AF_INET,AF_INET6
|
|
||||||
import time
|
|
||||||
import mime
|
|
||||||
try:
|
|
||||||
from io import BytesIO
|
|
||||||
except:
|
|
||||||
from StringIO import StringIO as BytesIO
|
|
||||||
import Milter
|
|
||||||
from Milter import utils
|
|
||||||
import mime
|
|
||||||
|
|
||||||
## Milter context for unit testing %milter applications.
|
|
||||||
# A substitute for milter.milterContext that can be passed to
|
|
||||||
# Milter.Base._setctx().
|
|
||||||
# @since 1.0.3
|
|
||||||
class TestCtx(object):
|
|
||||||
default_opts = [Milter.CURR_ACTS,0x1fffff,0,0]
|
|
||||||
def __init__(self,logfile='test/milter.log'):
|
|
||||||
## Usually the Milter application derived from Milter.Base
|
|
||||||
self._priv = None
|
|
||||||
## List of recipients deleted
|
|
||||||
self._delrcpt = []
|
|
||||||
## List of recipients added
|
|
||||||
self._addrcpt = []
|
|
||||||
## Macros defined
|
|
||||||
self._macros = { }
|
|
||||||
## Reply codes and messages set by the %milter
|
|
||||||
self._reply = None
|
|
||||||
## The macros returned by protocol stage
|
|
||||||
self._symlist = [ None, None, None, None, None, None, None ]
|
|
||||||
## 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
|
|
||||||
## The rfc822 message object for the current email being fed to the %milter.
|
|
||||||
self._msg = None
|
|
||||||
## The MAIL FROM for the current email being fed to the %milter
|
|
||||||
self._sender = None
|
|
||||||
## True if the %milter changed the envelope from.
|
|
||||||
self._envfromchanged = False
|
|
||||||
## List of recipients added
|
|
||||||
self._addrcpt = []
|
|
||||||
## Negotiated options
|
|
||||||
self._opts = TestCtx.default_opts
|
|
||||||
## Last activity
|
|
||||||
self._activity = time.time()
|
|
||||||
|
|
||||||
def getpriv(self):
|
|
||||||
return self._priv
|
|
||||||
|
|
||||||
def setpriv(self,priv):
|
|
||||||
self._priv = priv
|
|
||||||
|
|
||||||
def getsymval(self,name):
|
|
||||||
stage = self._stage
|
|
||||||
if stage >= 0:
|
|
||||||
try:
|
|
||||||
s = name.encode('utf8')
|
|
||||||
except: pass
|
|
||||||
syms = self._symlist[stage]
|
|
||||||
if syms is not None and s not in syms:
|
|
||||||
return None
|
|
||||||
return self._macros.get(name,None)
|
|
||||||
|
|
||||||
def _setsymval(self,name,val):
|
|
||||||
self._macros[name] = val
|
|
||||||
|
|
||||||
def setreply(self,rcode,xcode,*msg):
|
|
||||||
self._reply = (rcode,xcode) + msg
|
|
||||||
|
|
||||||
def setsymlist(self,stage,macros):
|
|
||||||
if self._stage != -1:
|
|
||||||
raise RuntimeError("setsymlist may only be called from negotiate")
|
|
||||||
# Records which macros are available to getsymval()
|
|
||||||
m = macros
|
|
||||||
try:
|
|
||||||
m = m.encode('utf8')
|
|
||||||
except: pass
|
|
||||||
try:
|
|
||||||
m = m.split(b' ')
|
|
||||||
except: pass
|
|
||||||
if len(m) > 5:
|
|
||||||
raise ValueError('setsymlist limited to 5 macros by MTA')
|
|
||||||
if self._symlist[stage] is not None:
|
|
||||||
raise ValueError('setsymlist already called for stage:'+stage)
|
|
||||||
if not m:
|
|
||||||
raise ValueError('setsymlist with empty list for stage:'+stage)
|
|
||||||
self._symlist[stage] = set(m)
|
|
||||||
|
|
||||||
def addheader(self,field,value,idx):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("addheader not called from eom()")
|
|
||||||
self._msg[field] = value
|
|
||||||
self._headerschanged = True
|
|
||||||
|
|
||||||
def chgheader(self,field,idx,value):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("chgheader not called from eom()")
|
|
||||||
if value == '':
|
|
||||||
del self._msg[field]
|
|
||||||
else:
|
|
||||||
self._msg[field] = value
|
|
||||||
self._headerschanged = True
|
|
||||||
|
|
||||||
def addrcpt(self,rcpt,params):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("addrcpt not called from eom()")
|
|
||||||
self._addrcpt.append((rcpt,params))
|
|
||||||
|
|
||||||
def delrcpt(self,rcpt):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("delrcpt not called from eom()")
|
|
||||||
self._delrcpt.append(rcpt)
|
|
||||||
|
|
||||||
def replacebody(self,chunk):
|
|
||||||
if self._body:
|
|
||||||
self._body.write(chunk)
|
|
||||||
self._bodyreplaced = True
|
|
||||||
else:
|
|
||||||
raise IOError("replacebody not called from eom()")
|
|
||||||
|
|
||||||
def chgfrom(self,sender,params=None):
|
|
||||||
if not self._body:
|
|
||||||
raise IOError("chgfrom not called from eom()")
|
|
||||||
self._envfromchanged = True
|
|
||||||
self._sender = sender
|
|
||||||
|
|
||||||
def quarantine(self,reason):
|
|
||||||
raise NotImplemented
|
|
||||||
|
|
||||||
## Reset activity timer.
|
|
||||||
def progress(self):
|
|
||||||
self._activity = time.time()
|
|
||||||
|
|
||||||
def _abort(self):
|
|
||||||
"What Milter sets for abort_callback"
|
|
||||||
self._priv.abort()
|
|
||||||
self._close()
|
|
||||||
|
|
||||||
def _close(self):
|
|
||||||
Milter.close_callback(self)
|
|
||||||
|
|
||||||
def _negotiate(self):
|
|
||||||
self._body = None
|
|
||||||
self._bodyreplaced = False
|
|
||||||
self._priv = None
|
|
||||||
self._opts = TestCtx.default_opts
|
|
||||||
self._stage = -1
|
|
||||||
rc = Milter.negotiate_callback(self,self._opts)
|
|
||||||
if rc == Milter.ALL_OPTS:
|
|
||||||
self._opts = TestCtx.default_opts
|
|
||||||
elif rc != Milter.CONTINUE:
|
|
||||||
self._abort()
|
|
||||||
self._close()
|
|
||||||
self._protocol = self._opts[1]
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _connect(self,host='localhost',helo='spamrelay',ip='1.2.3.4'):
|
|
||||||
rc = self._negotiate()
|
|
||||||
# FIXME: what if not CONTINUE or ALL_OPTS?
|
|
||||||
if self._protocol & Milter.P_NOCONNECT:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
if utils.ip4re.match(ip):
|
|
||||||
af = AF_INET
|
|
||||||
elif utils.ip6re.match(ip):
|
|
||||||
af = AF_INET6
|
|
||||||
else:
|
|
||||||
raise ValueError('TestCtx.connect: invalid ip address: '+ip)
|
|
||||||
self._stage = Milter.M_CONNECT
|
|
||||||
rc = Milter.connect_callback(self,host,af,ip)
|
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE:
|
|
||||||
self._close()
|
|
||||||
return rc
|
|
||||||
return self._helo(helo)
|
|
||||||
|
|
||||||
def _helo(self,helo):
|
|
||||||
if self._protocol & Milter.P_NOHELO:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
self._stage = Milter.M_HELO
|
|
||||||
rc = self._priv.hello(helo)
|
|
||||||
self._stage = None
|
|
||||||
if rc != Milter.CONTINUE:
|
|
||||||
self._close()
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _envfrom(self,*s):
|
|
||||||
self._sender = s[0]
|
|
||||||
if self._protocol & Milter.P_NOMAIL:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
self._stage = Milter.M_ENVFROM
|
|
||||||
rc = self._priv.envfrom(*s)
|
|
||||||
self._stage = None
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _envrcpt(self,s):
|
|
||||||
if self._protocol & Milter.P_NORCPT:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
self._stage = Milter.M_ENVRCPT
|
|
||||||
rc = self._priv.envrcpt(s)
|
|
||||||
self._stage = None
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _data(self):
|
|
||||||
if self._protocol & Milter.P_NODATA:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
self._stage = Milter.M_DATA
|
|
||||||
rc = self._priv.data()
|
|
||||||
self._stage = None
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _header(self,fld,val):
|
|
||||||
return self._priv.header(fld,val)
|
|
||||||
|
|
||||||
def _eoh(self):
|
|
||||||
if self._protocol & Milter.P_NOEOH:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
self._stage = Milter.M_EOH
|
|
||||||
rc = self._priv.eoh()
|
|
||||||
self._stage = None
|
|
||||||
return rc
|
|
||||||
|
|
||||||
def _feed_body(self,bfp):
|
|
||||||
if self._protocol & Milter.P_NOBODY:
|
|
||||||
return Milter.CONTINUE
|
|
||||||
while True:
|
|
||||||
buf = bfp.read(8192)
|
|
||||||
if len(buf) == 0: break
|
|
||||||
rc = self._priv.body(buf)
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
return Milter.CONTINUE
|
|
||||||
|
|
||||||
def _eom(self):
|
|
||||||
self._body = BytesIO()
|
|
||||||
self._stage = Milter.M_EOM
|
|
||||||
rc = self._priv.eom()
|
|
||||||
self._stage = None
|
|
||||||
return rc
|
|
||||||
|
|
||||||
## Feed a file like object to the ctx. Calls the callbacks in
|
|
||||||
# the same sequence as libmilter.
|
|
||||||
# @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 = mime.message_from_file(fp)
|
|
||||||
self._msg = msg
|
|
||||||
# envfrom
|
|
||||||
rc = self._envfrom('<%s>'%sender)
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
# envrcpt
|
|
||||||
for rcpt in (rcpt,) + rcpts:
|
|
||||||
rc = self._envrcpt('<%s>'%rcpt)
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
# data
|
|
||||||
rc = self._data()
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
# header
|
|
||||||
for h,val in msg.items():
|
|
||||||
rc = self._header(h,val)
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
# eoh
|
|
||||||
rc = self._eoh()
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
# body
|
|
||||||
header,body = msg.as_bytes().split(b'\n\n',1)
|
|
||||||
rc = self._feed_body(BytesIO(body))
|
|
||||||
if rc != Milter.CONTINUE: return rc
|
|
||||||
rc = self._eom()
|
|
||||||
if self._bodyreplaced:
|
|
||||||
body = self._body.getvalue()
|
|
||||||
self._body = BytesIO()
|
|
||||||
self._body.write(header)
|
|
||||||
self._body.write(b'\n\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,'rb') as fp:
|
|
||||||
return self._feedFile(fp,sender,*rcpts)
|
|
||||||
+20
-46
@@ -5,14 +5,12 @@
|
|||||||
import re
|
import re
|
||||||
import struct
|
import struct
|
||||||
import socket
|
import socket
|
||||||
import email.errors
|
import email.Errors
|
||||||
from email.header import decode_header
|
|
||||||
import email.base64mime
|
|
||||||
import email.utils
|
|
||||||
from fnmatch import fnmatchcase
|
from fnmatch import fnmatchcase
|
||||||
from binascii import a2b_base64
|
from email.Header import decode_header
|
||||||
|
#import email.Utils
|
||||||
|
import rfc822
|
||||||
|
|
||||||
dnsre = re.compile(r'^[a-z][-a-z\d.]+$', re.IGNORECASE)
|
|
||||||
PAT_IP4 = r'\.'.join([r'(?:\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])']*4)
|
PAT_IP4 = r'\.'.join([r'(?:\d|[1-9]\d|1\d\d|2[0-4]\d|25[0-5])']*4)
|
||||||
ip4re = re.compile(PAT_IP4+'$')
|
ip4re = re.compile(PAT_IP4+'$')
|
||||||
ip6re = re.compile( '(?:%(hex4)s:){6}%(ls32)s$'
|
ip6re = re.compile( '(?:%(hex4)s:){6}%(ls32)s$'
|
||||||
@@ -55,8 +53,8 @@ if hasattr(socket,'has_ipv6') and socket.has_ipv6:
|
|||||||
else:
|
else:
|
||||||
from pyip6 import inet_ntop, inet_pton
|
from pyip6 import inet_ntop, inet_pton
|
||||||
|
|
||||||
MASK = 0xFFFFFFFF
|
MASK = 0xFFFFFFFFL
|
||||||
MASK6 = 0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF
|
MASK6 = 0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFL
|
||||||
|
|
||||||
def cidr(i,n,mask=MASK):
|
def cidr(i,n,mask=MASK):
|
||||||
return ~(mask >> n) & mask & i
|
return ~(mask >> n) & mask & i
|
||||||
@@ -69,12 +67,6 @@ def iniplist(ipaddr,iplist):
|
|||||||
True
|
True
|
||||||
>>> iniplist('192.168.0.45',['192.168.0.*'])
|
>>> iniplist('192.168.0.45',['192.168.0.*'])
|
||||||
True
|
True
|
||||||
>>> iniplist('4.2.2.2',['b.resolvers.Level3.net'])
|
|
||||||
True
|
|
||||||
>>> iniplist('2606:2800:220:1::',['example.com/40'])
|
|
||||||
True
|
|
||||||
>>> iniplist('4.2.2.2',['nothing.example.com'])
|
|
||||||
False
|
|
||||||
>>> iniplist('2001:610:779:0:223:6cff:fe9a:9cf3',['127.0.0.1','172.20.1.0/24','2001:610:779::/48'])
|
>>> iniplist('2001:610:779:0:223:6cff:fe9a:9cf3',['127.0.0.1','172.20.1.0/24','2001:610:779::/48'])
|
||||||
True
|
True
|
||||||
>>> iniplist('2G01:610:779:0:223:6cff:fe9a:9cf3',['127.0.0.1','172.20.1.0/24','2001:610:779::/48'])
|
>>> iniplist('2G01:610:779:0:223:6cff:fe9a:9cf3',['127.0.0.1','172.20.1.0/24','2001:610:779::/48'])
|
||||||
@@ -83,10 +75,8 @@ def iniplist(ipaddr,iplist):
|
|||||||
ValueError: Invalid ip syntax:2G01:610:779:0:223:6cff:fe9a:9cf3
|
ValueError: Invalid ip syntax:2G01:610:779:0:223:6cff:fe9a:9cf3
|
||||||
"""
|
"""
|
||||||
if ip4re.match(ipaddr):
|
if ip4re.match(ipaddr):
|
||||||
fam = socket.AF_INET
|
|
||||||
ipnum = addr2bin(ipaddr)
|
ipnum = addr2bin(ipaddr)
|
||||||
elif ip6re.match(ipaddr):
|
elif ip6re.match(ipaddr):
|
||||||
fam = socket.AF_INET6
|
|
||||||
ipnum = bin2long6(inet_pton(ipaddr))
|
ipnum = bin2long6(inet_pton(ipaddr))
|
||||||
else:
|
else:
|
||||||
raise ValueError('Invalid ip syntax:'+ipaddr)
|
raise ValueError('Invalid ip syntax:'+ipaddr)
|
||||||
@@ -106,21 +96,13 @@ def iniplist(ipaddr,iplist):
|
|||||||
n = 128
|
n = 128
|
||||||
if cidr(bin2long6(inet_pton(p[0])),n,MASK6) == cidr(ipnum,n,MASK6):
|
if cidr(bin2long6(inet_pton(p[0])),n,MASK6) == cidr(ipnum,n,MASK6):
|
||||||
return True
|
return True
|
||||||
elif dnsre.match(p[0]):
|
|
||||||
try:
|
|
||||||
sfx = '/'.join(['']+p[1:])
|
|
||||||
addrlist = [r[4][0]+sfx for r in socket.getaddrinfo(p[0],25,fam)]
|
|
||||||
if iniplist(ipaddr,addrlist):
|
|
||||||
return True
|
|
||||||
except socket.gaierror: pass
|
|
||||||
elif fnmatchcase(ipaddr,pat):
|
elif fnmatchcase(ipaddr,pat):
|
||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
|
|
||||||
## Split email into Fullname and address.
|
## Split email into Fullname and address.
|
||||||
# This replaces <code>email.utils.parseaddr</code> but fixes
|
# This replaces <code>email.Utils.parseaddr</code> but fixes
|
||||||
# some <a href="http://bugs.python.org/issue1025395">tricky test cases</a>.
|
# some <a href="http://bugs.python.org/issue1025395">tricky test cases</a>.
|
||||||
# Additional tricky cases are still broken. Patches welcome.
|
|
||||||
#
|
#
|
||||||
def parseaddr(t):
|
def parseaddr(t):
|
||||||
"""Split email into Fullname and address.
|
"""Split email into Fullname and address.
|
||||||
@@ -134,10 +116,12 @@ def parseaddr(t):
|
|||||||
>>> parseaddr('God@heaven <@hop1.org,@hop2.net:jeff@spec.org>')
|
>>> parseaddr('God@heaven <@hop1.org,@hop2.net:jeff@spec.org>')
|
||||||
('God@heaven', 'jeff@spec.org')
|
('God@heaven', 'jeff@spec.org')
|
||||||
>>> parseaddr('Real Name ((comment)) <addr...@example.com>')
|
>>> parseaddr('Real Name ((comment)) <addr...@example.com>')
|
||||||
('Real Name (comment)', 'addr...@example.com')
|
('Real Name', 'addr...@example.com')
|
||||||
|
>>> parseaddr('a(WRONG)@b')
|
||||||
|
('WRONG', 'a@b')
|
||||||
"""
|
"""
|
||||||
#return email.utils.parseaddr(t)
|
#return email.Utils.parseaddr(t)
|
||||||
res = email.utils.parseaddr(t)
|
res = rfc822.parseaddr(t)
|
||||||
# dirty fix for some broken cases
|
# dirty fix for some broken cases
|
||||||
if not res[0]:
|
if not res[0]:
|
||||||
pos = t.find('<')
|
pos = t.find('<')
|
||||||
@@ -146,7 +130,7 @@ def parseaddr(t):
|
|||||||
pos1 = addrspec.rfind(':')
|
pos1 = addrspec.rfind(':')
|
||||||
if pos1 > 0:
|
if pos1 > 0:
|
||||||
addrspec = addrspec[pos1+1:]
|
addrspec = addrspec[pos1+1:]
|
||||||
return email.utils.parseaddr('"%s" <%s>' % (t[:pos].strip(),addrspec))
|
return rfc822.parseaddr('"%s" <%s>' % (t[:pos].strip(),addrspec))
|
||||||
if not res[1]:
|
if not res[1]:
|
||||||
pos = t.find('<')
|
pos = t.find('<')
|
||||||
if pos > 0 and t[-1] == '>':
|
if pos > 0 and t[-1] == '>':
|
||||||
@@ -154,19 +138,9 @@ def parseaddr(t):
|
|||||||
pos1 = addrspec.rfind(':')
|
pos1 = addrspec.rfind(':')
|
||||||
if pos1 > 0:
|
if pos1 > 0:
|
||||||
addrspec = addrspec[pos1+1:]
|
addrspec = addrspec[pos1+1:]
|
||||||
return email.utils.parseaddr('%s<%s>' % (t[:pos].strip(),addrspec))
|
return rfc822.parseaddr('%s<%s>' % (t[:pos].strip(),addrspec))
|
||||||
return res
|
return res
|
||||||
|
|
||||||
## Fix email.base64mime.decode to add any missing padding
|
|
||||||
def decode(s, convert_eols=None):
|
|
||||||
if not s: return s
|
|
||||||
while len(s) % 4: s += '=' # add missing padding
|
|
||||||
dec = a2b_base64(s)
|
|
||||||
if convert_eols:
|
|
||||||
return dec.replace(CRLF, convert_eols)
|
|
||||||
return dec
|
|
||||||
|
|
||||||
email.base64mime.decode = decode
|
|
||||||
|
|
||||||
def parse_addr(t):
|
def parse_addr(t):
|
||||||
"""Split email into user,domain.
|
"""Split email into user,domain.
|
||||||
@@ -211,18 +185,18 @@ def parse_header(val):
|
|||||||
for s,enc in h:
|
for s,enc in h:
|
||||||
if enc:
|
if enc:
|
||||||
try:
|
try:
|
||||||
u.append(s.decode(enc,'replace'))
|
u.append(unicode(s,enc,'replace'))
|
||||||
except LookupError:
|
except LookupError:
|
||||||
u.append(s.decode())
|
u.append(unicode(s))
|
||||||
else:
|
else:
|
||||||
u.append(s.decode())
|
u.append(unicode(s))
|
||||||
u = u''.join(u)
|
u = ''.join(u)
|
||||||
for enc in ('us-ascii','iso-8859-1','utf-8'):
|
for enc in ('us-ascii','iso-8859-1','utf8'):
|
||||||
try:
|
try:
|
||||||
return u.encode(enc)
|
return u.encode(enc)
|
||||||
except UnicodeError: continue
|
except UnicodeError: continue
|
||||||
except UnicodeDecodeError: pass
|
except UnicodeDecodeError: pass
|
||||||
except LookupError: pass
|
except LookupError: pass
|
||||||
except ValueError: pass
|
except ValueError: pass
|
||||||
except email.errors.HeaderParseError: pass
|
except email.Errors.HeaderParseError: pass
|
||||||
return val
|
return val
|
||||||
|
|||||||
@@ -11,24 +11,25 @@ any point, tell Sendmail to reject, discard, or accept the message.
|
|||||||
Requirements
|
Requirements
|
||||||
------------
|
------------
|
||||||
|
|
||||||
Python milter extension: http://https://pypi.python.org/pypi/pymilter/
|
This python milter extension: http://www.bmsi.com/python/milter.html
|
||||||
Python: http://www.python.org
|
Python: http://www.python.org
|
||||||
Sendmail: http://www.sendmail.org
|
Sendmail: http://www.sendmail.org
|
||||||
|
|
||||||
NB: From Sendmail's libmilter/README:
|
NB: From Sendmail's libmilter/README:
|
||||||
|
|
||||||
libmilter requires pthread support in the operating system. Moreover, it
|
libmilter requires pthread support in the operating system. Moreover, it
|
||||||
requires that the library functions it uses are thread safe; which is true
|
requires that the library functions it uses are thread safe; which is true
|
||||||
for the operating systems libmilter has been developed and tested on. On
|
for the operating systems libmilter has been developed and tested on. On
|
||||||
some operating systems this requires special compile time options (e.g.,
|
some operating systems this requires special compile time options (e.g.,
|
||||||
not just -pthread). libmilter is currently known to work on (modulo problems
|
not just -pthread). libmilter is currently known to work on (modulo
|
||||||
in the pthread support of some specific versions):
|
problems in the pthread support of some specific versions):
|
||||||
|
|
||||||
FreeBSD 3.x, 4.x
|
FreeBSD 3.x, 4.x
|
||||||
SunOS 5.x (x >= 5)
|
SunOS 5.x (x >= 5)
|
||||||
AIX 4.3.x
|
AIX 4.3.x
|
||||||
HP UX 11.x
|
HP UX 11.x
|
||||||
Linux (recent versions/distributions)
|
Linux (recent versions/distributions)
|
||||||
|
OpenBSD
|
||||||
|
AIX 4.1.5
|
||||||
|
|
||||||
libmilter is currently not supported on:
|
libmilter is currently not supported on:
|
||||||
|
|
||||||
@@ -109,6 +110,28 @@ _FFR_MILTER for the cf macros. For example,
|
|||||||
|
|
||||||
m4 -D_FFR_MILTER ../m4/cf.m4 myconfig.mc > myconfig.cf
|
m4 -D_FFR_MILTER ../m4/cf.m4 myconfig.mc > myconfig.cf
|
||||||
|
|
||||||
|
|
||||||
|
RedHat 6.2 Notes
|
||||||
|
----------------
|
||||||
|
|
||||||
|
The Redhat 6.2 sendmail RPM does not enable milter. You can obtain a
|
||||||
|
modified spec file at
|
||||||
|
|
||||||
|
http://www.bmsi.com/linux/rh62/sendmail-rhmilter.spec
|
||||||
|
|
||||||
|
use it to rebuild the Redhat 7.2 SRPM. The RH6.2 SRPM does not have
|
||||||
|
recent sendmail security patches.
|
||||||
|
|
||||||
|
RedHat 7.2 Notes
|
||||||
|
----------------
|
||||||
|
|
||||||
|
The Redhat 7.2 sendmail RPM enables milter in sendmail - but does not include
|
||||||
|
the headers needed for compiling a milter. You can obtain a modified spec
|
||||||
|
file with a sendmail-devel package that includes the needed static libraries
|
||||||
|
and headers at
|
||||||
|
|
||||||
|
http://www.bmsi.com/linux/sendmail-rh72.spec
|
||||||
|
|
||||||
IPv6 Notes
|
IPv6 Notes
|
||||||
----------
|
----------
|
||||||
|
|
||||||
|
|||||||
+3
-45
@@ -1,7 +1,7 @@
|
|||||||
## @mainpage Writing Milters in Python
|
## @mainpage Writing Milters in Python
|
||||||
#
|
#
|
||||||
# At the lowest level, the <code>milter</code> module provides a thin wrapper
|
# At the lowest level, the <code>milter</code> module provides a thin wrapper
|
||||||
# around the <a href="milter_api/index.html"> sendmail
|
# around the <a href="https://www.milter.org/developers/api/index"> sendmail
|
||||||
# libmilter API</a>. This API lets you register callbacks for a number of
|
# libmilter API</a>. This API lets you register callbacks for a number of
|
||||||
# events in the process of sendmail receiving a message via SMTP. These
|
# events in the process of sendmail receiving a message via SMTP. These
|
||||||
# events include the initial connection from a MTA, the envelope sender and
|
# events include the initial connection from a MTA, the envelope sender and
|
||||||
@@ -39,10 +39,10 @@
|
|||||||
# @section threading
|
# @section threading
|
||||||
#
|
#
|
||||||
# The libmilter library which pymilter wraps
|
# The libmilter library which pymilter wraps
|
||||||
# <a href="milter_overview#SignalHandling">handles
|
# <a href="https://www.milter.org/developers/overview#SignalHandling">handles
|
||||||
# all signals</a> itself, and expects to be called from a single main thread.
|
# all signals</a> itself, and expects to be called from a single main thread.
|
||||||
# It handles SIGTERM, SIGHUP, and SIGINT, mapping the first two to
|
# It handles SIGTERM, SIGHUP, and SIGINT, mapping the first two to
|
||||||
# <a href="milter_api/smfi_stop.html">smfi_stop</a>
|
# <a href="https://www.milter.org/developers/api/smfi_stop">smfi_stop</a>
|
||||||
# and the last to an internal ABORT.
|
# and the last to an internal ABORT.
|
||||||
#
|
#
|
||||||
# If you use python threads or threading modules, then signal handling gets
|
# If you use python threads or threading modules, then signal handling gets
|
||||||
@@ -54,45 +54,3 @@
|
|||||||
# multiprocessing</a> module useful. It can be a drop-in
|
# multiprocessing</a> module useful. It can be a drop-in
|
||||||
# replacement for threading as illustrated in
|
# replacement for threading as illustrated in
|
||||||
# <a href="milter-template_8py-example.html">milter-template.py</a>.
|
# <a href="milter-template_8py-example.html">milter-template.py</a>.
|
||||||
#
|
|
||||||
# @section Useful python packages for milters
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/pymilter">pymilter</a> - this package.
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/pyspf">pyspf</a> checks the
|
|
||||||
# SMTP envelope sender (MAIL FROM, passed to the Milter.Base.envfrom callback)
|
|
||||||
# against a Sender Policy published in DNS by the sending domain. This
|
|
||||||
# can prevent forgery of the MAIL FROM. SPF is Sender Policy Framework.
|
|
||||||
#
|
|
||||||
# <a href="https://launchpad.net/dkimpy">pydkim</a> checks a DKIM signature
|
|
||||||
# of the email body and headers against a public key published in DNS by
|
|
||||||
# the signing domain. DKIM is DomainKeys Identified Mail.
|
|
||||||
#
|
|
||||||
# The <a href="https://pypi.python.org/pypi/authres/">authres</a> module
|
|
||||||
# parses and formats the Authentication-Results email header, providing
|
|
||||||
# a standard place to summarize the results from DKIM, SPF, rDNS, SMTP AUTH,
|
|
||||||
# and other email authentication methods.
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/pydspam/">pydspam</a> wraps
|
|
||||||
# the libdspam API of the <a href="http://dspam.sourceforge.net/">DSPAM</a>
|
|
||||||
# project.
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/pysrs/">pysrs</a> rewrites
|
|
||||||
# MAIL FROM to include a timestamped signature so that "bounce spam"
|
|
||||||
# can be immediately rejected.
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/pygossip/">pygossip</a> is a
|
|
||||||
# system to track reputation by domain and authentication level and type,
|
|
||||||
# and a simple protocol to gossip about reputations with other mail servers.
|
|
||||||
#
|
|
||||||
# @section Milters written with pymilter
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/croessner/vrfydmn">Verify Domain</a> is a
|
|
||||||
# Postfix milter that rejects/fixes manipulated From: header
|
|
||||||
# on a mail host with multiple virtual domains.
|
|
||||||
#
|
|
||||||
# <a href="https://github.com/sdgathman/milter/">BMS Milter</a> has several
|
|
||||||
# milters, a big complicated spam filter that integrates multiple
|
|
||||||
# authentication protocols with pydspam, and two simple ones: spfmilter.py and
|
|
||||||
# dkim-milter.py.
|
|
||||||
#
|
|
||||||
|
|||||||
+20
-20
@@ -75,44 +75,44 @@ NOREPLY = 6
|
|||||||
# and converts function callbacks to instance method invocations.
|
# and converts function callbacks to instance method invocations.
|
||||||
#
|
#
|
||||||
class milterContext(object):
|
class milterContext(object):
|
||||||
## Calls <a href="milter_api/smfi_getsymval.html">smfi_getsymval</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_getsymval">smfi_getsymval</a>.
|
||||||
def getsymval(self,sym): pass
|
def getsymval(self,sym): pass
|
||||||
## Calls <a href="milter_api/smfi_setreply.html">
|
## Calls <a href="https://www.milter.org/developers/api/smfi_setreply">
|
||||||
# smfi_setreply</a> or
|
# smfi_setreply</a> or
|
||||||
# <a href="milter_api/smfi_setmlreply.html">
|
# <a href="https://www.milter.org/developers/api/smfi_setmlreply">
|
||||||
# smfi_setmlreply</a>.
|
# smfi_setmlreply</a>.
|
||||||
# @param rcode SMTP response code
|
# @param rcode SMTP response code
|
||||||
# @param xcode extended SMTP response code
|
# @param xcode extended SMTP response code
|
||||||
# @param msg one or more message lines. If the MTA does not support
|
# @param msg one or more message lines. If the MTA does not support
|
||||||
# multiline messages, only the first is used.
|
# multiline messages, only the first is used.
|
||||||
def setreply(self,rcode,xcode,*msg): pass
|
def setreply(self,rcode,xcode,*msg): pass
|
||||||
## Calls <a href="milter_api/smfi_addheader.html">smfi_addheader</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_addheader">smfi_addheader</a>.
|
||||||
def addheader(self,name,value,idx=-1): pass
|
def addheader(self,name,value,idx=-1): pass
|
||||||
## Calls <a href="milter_api/smfi_chgheader.html">smfi_chgheader</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_chgheader">smfi_chgheader</a>.
|
||||||
def chgheader(self,name,idx,value): pass
|
def chgheader(self,name,idx,value): pass
|
||||||
## Calls <a href="milter_api/smfi_addrcpt.html">smfi_addrcpt</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_addrcpt">smfi_addrcpt</a>.
|
||||||
def addrcpt(self,rcpt,params=None): pass
|
def addrcpt(self,rcpt,params=None): pass
|
||||||
## Calls <a href="milter_api/smfi_delrcpt.html">smfi_delrcpt</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_delrcpt">smfi_delrcpt</a>.
|
||||||
def delrcpt(self,rcpt): pass
|
def delrcpt(self,rcpt): pass
|
||||||
## Calls <a href="milter_api/smfi_replacebody.html">smfi_replacebody</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_replacebody">smfi_replacebody</a>.
|
||||||
def replacebody(self,data): pass
|
def replacebody(self,data): pass
|
||||||
## Attach a Python object to this connection context.
|
## Attach a Python object to this connection context.
|
||||||
# @return the old value or None
|
# @return the old value or None
|
||||||
def setpriv(self,priv): pass
|
def setpriv(self,priv): pass
|
||||||
## Return the Python object attached to this connection context.
|
## Return the Python object attached to this connection context.
|
||||||
def getpriv(self): pass
|
def getpriv(self): pass
|
||||||
## Calls <a href="milter_api/smfi_quarantine.html">smfi_quarantine</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_quarantine">smfi_quarantine</a>.
|
||||||
def quarantine(self,reason): pass
|
def quarantine(self,reason): pass
|
||||||
## Calls <a href="milter_api/smfi_progress.html">smfi_progress</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_progress">smfi_progress</a>.
|
||||||
def progress(self): pass
|
def progress(self): pass
|
||||||
## Calls <a href="milter_api/smfi_chgfrom.html">smfi_chgfrom</a>.
|
## Calls <a href="https://www.milter.org/developers/api/smfi_chgfrom">smfi_chgfrom</a>.
|
||||||
def chgfrom(self,sender,param=None): pass
|
def chgfrom(self,sender,param=None): pass
|
||||||
## Tell the MTA which macro values we are interested in for a given stage.
|
## 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.
|
# Of interest only when you need to squeeze a few more bytes of bandwidth.
|
||||||
# It may only be called from the negotiate callback.
|
# It may only be called from the negotiate callback.
|
||||||
# The protocol stages are
|
# The protocol stages are
|
||||||
# M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT, M_DATA, M_EOM, M_EOH.
|
# M_CONNECT, M_HELO, M_ENVFROM, M_ENVRCPT, M_DATA, M_EOM, M_EOH.
|
||||||
# Calls <a href="milter_api/smfi_setsymlist.html">smfi_setsymlist</a>.
|
# 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 stage protocol stage in which the macro list should be used
|
||||||
# @param macrolist a space separated list of macro names
|
# @param macrolist a space separated list of macro names
|
||||||
def setsymlist(self,stage,macrolist): pass
|
def setsymlist(self,stage,macrolist): pass
|
||||||
@@ -160,15 +160,15 @@ def set_exception_policy(code): pass
|
|||||||
# member functions are actually overridden by an application class.
|
# member functions are actually overridden by an application class.
|
||||||
# @param name the %milter name by which the MTA finds us
|
# @param name the %milter name by which the MTA finds us
|
||||||
# @param negotiate the
|
# @param negotiate the
|
||||||
# <a href="milter_api/xxfi_negotiate.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_negotiate">
|
||||||
# xxfi_negotiate</a> callback, called to negotiate supported
|
# xxfi_negotiate</a> callback, called to negotiate supported
|
||||||
# actions, callbacks, and protocol steps.
|
# actions, callbacks, and protocol steps.
|
||||||
# @param unknown the
|
# @param unknown the
|
||||||
# <a href="milter_api/xxfi_unknown.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_unknown">
|
||||||
# xxfi_unknown</a> callback, called when for SMTP commands
|
# xxfi_unknown</a> callback, called when for SMTP commands
|
||||||
# not recognized by the MTA. (Extend SMTP in your milter!)
|
# not recognized by the MTA. (Extend SMTP in your milter!)
|
||||||
# @param data the
|
# @param data the
|
||||||
# <a href="milter_api/xxfi_data.html">
|
# <a href="https://www.milter.org/developers/api/xxfi_data">
|
||||||
# xxfi_data</a> callback, called when the DATA
|
# xxfi_data</a> callback, called when the DATA
|
||||||
# SMTP command is received.
|
# SMTP command is received.
|
||||||
def register(name,negotiate=None,unknown=None,data=None): pass
|
def register(name,negotiate=None,unknown=None,data=None): pass
|
||||||
@@ -178,19 +178,19 @@ def register(name,negotiate=None,unknown=None,data=None): pass
|
|||||||
# call to milter.setconn() which will be the interface between MTAs and the
|
# 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
|
# %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.
|
# created. If this is not called, milter.main() will do so implicitly.
|
||||||
# Calls <a href="milter_api/smfi_opensocket.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_opensocket">
|
||||||
# smfi_opensocket</a>. While not documented for libmilter, my experiments
|
# smfi_opensocket</a>. While not documented for libmilter, my experiments
|
||||||
# indicate that you must call register() before calling opensocket().
|
# indicate that you must call register() before calling opensocket().
|
||||||
# @param rmsock Try to remove an existing unix domain socket if true.
|
# @param rmsock Try to remove an existing unix domain socket if true.
|
||||||
def opensocket(rmsock): pass
|
def opensocket(rmsock): pass
|
||||||
|
|
||||||
## Transfer control to libmilter.
|
## Transfer control to libmilter.
|
||||||
# Calls <a href="milter_api/smfi_main.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_main">
|
||||||
# smfi_main</a>.
|
# smfi_main</a>.
|
||||||
def main(): pass
|
def main(): pass
|
||||||
|
|
||||||
## Set the libmilter debugging level.
|
## Set the libmilter debugging level.
|
||||||
# <a href="milter_api/smfi_setdbg.html">smfi_setdbg</a>
|
# <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
|
# 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
|
# greater (more positive) the level the more detailed the debugging. Six is the
|
||||||
@@ -198,12 +198,12 @@ def main(): pass
|
|||||||
def setdbg(lev): pass
|
def setdbg(lev): pass
|
||||||
|
|
||||||
## Set timeout for MTA communication.
|
## Set timeout for MTA communication.
|
||||||
# Calls <a href="milter_api/smfi_settimeout.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_settimeout">
|
||||||
# smfi_settimeout</a>. Must be called before calling main().
|
# smfi_settimeout</a>. Must be called before calling main().
|
||||||
def settimeout(secs): pass
|
def settimeout(secs): pass
|
||||||
|
|
||||||
## Set socket backlog.
|
## Set socket backlog.
|
||||||
# Calls <a href="milter_api/smfi_setbacklog.html">
|
# Calls <a href="https://www.milter.org/developers/api/smfi_setbacklog">
|
||||||
# smfi_setbacklog</a>. Must be called before calling main().
|
# smfi_setbacklog</a>. Must be called before calling main().
|
||||||
def setbacklog(n): pass
|
def setbacklog(n): pass
|
||||||
|
|
||||||
|
|||||||
@@ -1,14 +1,16 @@
|
|||||||
web:
|
web:
|
||||||
doxygen
|
doxygen
|
||||||
test -L doc/html/milter_api || ln -sf /usr/share/doc/sendmail-milter-devel doc/html/milter_api
|
|
||||||
rsync -ravKk doc/html/ bmsi.com:/var/www/html/pymilter
|
|
||||||
cd doc/html; zip -r ../../doc .
|
cd doc/html; zip -r ../../doc .
|
||||||
|
rsync -ravK doc/html/ spidey2.bmsi.com:/Public/pymilter
|
||||||
|
|
||||||
VERSION=1.0.4
|
VERSION=1.0
|
||||||
|
CVSTAG=pymilter-1_0
|
||||||
PKG=pymilter-$(VERSION)
|
PKG=pymilter-$(VERSION)
|
||||||
SRCTAR=$(PKG).tar.gz
|
SRCTAR=$(PKG).tar.gz
|
||||||
|
|
||||||
$(SRCTAR):
|
$(SRCTAR):
|
||||||
git archive --format=tar.gz --prefix=$(PKG)/ -o $(SRCTAR) $(PKG)
|
cvs export -r$(CVSTAG) -d $(PKG) pymilter
|
||||||
|
tar cvfz $(PKG).tar.gz $(PKG)
|
||||||
|
rm -r $(PKG)
|
||||||
|
|
||||||
gittar: $(SRCTAR)
|
cvstar: $(SRCTAR)
|
||||||
|
|||||||
+4
-5
@@ -1,8 +1,7 @@
|
|||||||
## A very simple sample milter to prevent mixing of internal and external mail.
|
## A very simple milter to prevent mixing of internal and external mail.
|
||||||
# Internal is defined as using one of a list of internal top level domains.
|
# Internal is defined as using one of a list of internal top level domains.
|
||||||
# This code is open-source on the same terms as Python.
|
# This code is open-source on the same terms as Python.
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import Milter
|
import Milter
|
||||||
import time
|
import time
|
||||||
import sys
|
import sys
|
||||||
@@ -14,7 +13,7 @@ internal_tlds = ["corp", "personal"]
|
|||||||
# True if internal, False otherwise
|
# True if internal, False otherwise
|
||||||
def is_internal(hostname):
|
def is_internal(hostname):
|
||||||
components = hostname.split(".")
|
components = hostname.split(".")
|
||||||
return components.pop() in internal_tlds
|
return components.pop() in internal_tlds:
|
||||||
|
|
||||||
# Determine if internal and external hosts are mixed based on a list
|
# Determine if internal and external hosts are mixed based on a list
|
||||||
# of hostnames
|
# of hostnames
|
||||||
@@ -69,12 +68,12 @@ def main():
|
|||||||
timeout = 600
|
timeout = 600
|
||||||
# Register to have the Milter factory create instances of your class:
|
# Register to have the Milter factory create instances of your class:
|
||||||
Milter.factory = NoMixMilter
|
Milter.factory = NoMixMilter
|
||||||
print("%s milter startup" % time.strftime('%Y%b%d %H:%M:%S'))
|
print "%s milter startup" % time.strftime('%Y%b%d %H:%M:%S')
|
||||||
sys.stdout.flush()
|
sys.stdout.flush()
|
||||||
Milter.runmilter("nomixfilter",socketname,timeout)
|
Milter.runmilter("nomixfilter",socketname,timeout)
|
||||||
logq.put(None)
|
logq.put(None)
|
||||||
bt.join()
|
bt.join()
|
||||||
print("%s nomix milter shutdown" % time.strftime('%Y%b%d %H:%M:%S'))
|
print "%s nomix milter shutdown" % time.strftime('%Y%b%d %H:%M:%S')
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
+8
-14
@@ -1,18 +1,14 @@
|
|||||||
## To roll your own milter, create a class that extends Milter.
|
## To roll your own milter, create a class that extends Milter.
|
||||||
# See the pymilter project at http://bmsi.com/python/milter.html
|
# See the pymilter project at http://bmsi.com/python/milter.html
|
||||||
# based on Sendmail's milter API
|
# based on Sendmail's milter API http://www.milter.org/milter_api/api.html
|
||||||
# This code is open-source on the same terms as Python.
|
# This code is open-source on the same terms as Python.
|
||||||
|
|
||||||
## Milter calls methods of your class at milter events.
|
## Milter calls methods of your class at milter events.
|
||||||
## Return REJECT,TEMPFAIL,ACCEPT to short circuit processing for a message.
|
## Return REJECT,TEMPFAIL,ACCEPT to short circuit processing for a message.
|
||||||
## You can also add/del recipients, replacebody, add/del headers, etc.
|
## You can also add/del recipients, replacebody, add/del headers, etc.
|
||||||
|
|
||||||
from __future__ import print_function
|
|
||||||
import Milter
|
import Milter
|
||||||
try:
|
import StringIO
|
||||||
from StringIO import StringIO
|
|
||||||
except:
|
|
||||||
from io import StringIO
|
|
||||||
import time
|
import time
|
||||||
import email
|
import email
|
||||||
import sys
|
import sys
|
||||||
@@ -78,7 +74,7 @@ class myMilter(Milter.Base):
|
|||||||
# NOTE: self.fp is only an *internal* copy of message data. You
|
# NOTE: self.fp is only an *internal* copy of message data. You
|
||||||
# must use addheader, chgheader, replacebody to change the message
|
# must use addheader, chgheader, replacebody to change the message
|
||||||
# on the MTA.
|
# on the MTA.
|
||||||
self.fp = StringIO()
|
self.fp = StringIO.StringIO()
|
||||||
self.canon_from = '@'.join(parse_addr(mailfrom))
|
self.canon_from = '@'.join(parse_addr(mailfrom))
|
||||||
self.fp.write('From %s %s\n' % (self.canon_from,time.ctime()))
|
self.fp.write('From %s %s\n' % (self.canon_from,time.ctime()))
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
@@ -135,12 +131,10 @@ def background():
|
|||||||
t = logq.get()
|
t = logq.get()
|
||||||
if not t: break
|
if not t: break
|
||||||
msg,id,ts = t
|
msg,id,ts = t
|
||||||
print("%s [%d]" % (time.strftime('%Y%b%d %H:%M:%S',time.localtime(ts)),id),
|
print "%s [%d]" % (time.strftime('%Y%b%d %H:%M:%S',time.localtime(ts)),id),
|
||||||
end=None)
|
|
||||||
# 2005Oct13 02:34:11 [1] msg1 msg2 msg3 ...
|
# 2005Oct13 02:34:11 [1] msg1 msg2 msg3 ...
|
||||||
for i in msg: print(i,end=None)
|
for i in msg: print i,
|
||||||
print()
|
print
|
||||||
sys.stdout.flush()
|
|
||||||
|
|
||||||
## ===
|
## ===
|
||||||
|
|
||||||
@@ -155,12 +149,12 @@ def main():
|
|||||||
flags += Milter.ADDRCPT
|
flags += Milter.ADDRCPT
|
||||||
flags += Milter.DELRCPT
|
flags += Milter.DELRCPT
|
||||||
Milter.set_flags(flags) # tell Sendmail which features we use
|
Milter.set_flags(flags) # tell Sendmail which features we use
|
||||||
print("%s milter startup" % time.strftime('%Y%b%d %H:%M:%S'))
|
print "%s milter startup" % time.strftime('%Y%b%d %H:%M:%S')
|
||||||
sys.stdout.flush()
|
sys.stdout.flush()
|
||||||
Milter.runmilter("pythonfilter",socketname,timeout)
|
Milter.runmilter("pythonfilter",socketname,timeout)
|
||||||
logq.put(None)
|
logq.put(None)
|
||||||
bt.join()
|
bt.join()
|
||||||
print("%s bms milter shutdown" % time.strftime('%Y%b%d %H:%M:%S'))
|
print "%s bms milter shutdown" % time.strftime('%Y%b%d %H:%M:%S')
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
+232
-97
@@ -1,6 +1,6 @@
|
|||||||
/* Copyright (C) 2001 James Niemira (niemira@colltech.com, urmane@urmane.org)
|
/* Copyright (C) 2001 James Niemira (niemira@colltech.com, urmane@urmane.org)
|
||||||
* Portions Copyright (C) 2001,2002,2003,2004,2005,2006,2007
|
* Portions Copyright (C) 2001,2002,2003,2004,2005,2006,2007
|
||||||
* Stuart Gathman (stuart@gathman.org)
|
* Stuart Gathman (stuart@bmsi.com)
|
||||||
*
|
*
|
||||||
* This program is free software: you can redistribute it and/or modify it
|
* This program is free software: you can redistribute it and/or modify it
|
||||||
* under the terms of the GNU General Public License as published by the
|
* under the terms of the GNU General Public License as published by the
|
||||||
@@ -34,6 +34,217 @@ $ python setup.py help
|
|||||||
|
|
||||||
libraries=["milter","smutil","resolv"]
|
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
|
||||||
|
*
|
||||||
|
* Revision 1.29 2011/06/09 15:45:27 customdesigned
|
||||||
|
* Print callback name for non-int return error.
|
||||||
|
*
|
||||||
|
* Revision 1.28 2011/06/08 23:13:48 customdesigned
|
||||||
|
* Generate special exception when callback return not int.
|
||||||
|
*
|
||||||
|
* Revision 1.27 2009/07/28 21:45:54 customdesigned
|
||||||
|
* Add getversion() to return runtime version.
|
||||||
|
*
|
||||||
|
* Revision 1.26 2009/07/28 21:08:20 customdesigned
|
||||||
|
* Increment del count.
|
||||||
|
*
|
||||||
|
* Revision 1.25 2009/07/28 20:58:55 customdesigned
|
||||||
|
* getdiag method
|
||||||
|
*
|
||||||
|
* Revision 1.24 2009/06/09 01:54:44 customdesigned
|
||||||
|
* Forgot to initialize optional parameter.
|
||||||
|
*
|
||||||
|
* Revision 1.23 2009/05/29 20:44:58 customdesigned
|
||||||
|
* Typo SMFIP_NO constants.
|
||||||
|
*
|
||||||
|
* Revision 1.22 2009/05/29 19:53:36 customdesigned
|
||||||
|
* Typo SMFIS_ALL_OPTS
|
||||||
|
*
|
||||||
|
* Revision 1.21 2009/05/29 19:49:40 customdesigned
|
||||||
|
* Typo calling helo instead of negotiate.
|
||||||
|
*
|
||||||
|
* Revision 1.20 2009/05/29 18:25:59 customdesigned
|
||||||
|
* Null terminate keyword list.
|
||||||
|
*
|
||||||
|
* Revision 1.19 2009/05/28 18:36:42 customdesigned
|
||||||
|
* Support new callbacks, including negotiate
|
||||||
|
*
|
||||||
|
* Revision 1.18 2009/05/21 21:53:05 customdesigned
|
||||||
|
* First cut at support unknown, data, negotiate callbacks.
|
||||||
|
*
|
||||||
|
* Revision 1.17 2009/02/06 04:28:08 customdesigned
|
||||||
|
* Oops! Missing options argument pointer for addrcpt.
|
||||||
|
*
|
||||||
|
* Revision 1.16 2008/12/16 04:21:05 customdesigned
|
||||||
|
* Fedora release
|
||||||
|
*
|
||||||
|
* Revision 1.15 2008/12/13 20:29:56 customdesigned
|
||||||
|
* Split off milter applications.
|
||||||
|
*
|
||||||
|
* Revision 1.14 2008/12/04 19:43:00 customdesigned
|
||||||
|
* Doc updates.
|
||||||
|
*
|
||||||
|
* Revision 1.13 2008/11/23 03:06:47 customdesigned
|
||||||
|
* Milter support for chgfrom.
|
||||||
|
*
|
||||||
|
* Revision 1.12 2008/11/21 20:42:52 customdesigned
|
||||||
|
* Support smfi_chgfrom and smfi_addrcpt_par.
|
||||||
|
*
|
||||||
|
* Revision 1.11 2007/09/25 02:26:29 customdesigned
|
||||||
|
* Update license.
|
||||||
|
*
|
||||||
|
* Revision 1.10 2006/02/12 02:00:42 customdesigned
|
||||||
|
* Resolve FIXME for wrap_close.
|
||||||
|
*
|
||||||
|
* Revision 1.9 2005/12/23 21:46:36 customdesigned
|
||||||
|
* Compile on sendmail-8.12 (ifdef SMFIR_INSHEADER)
|
||||||
|
*
|
||||||
|
* Revision 1.8 2005/10/20 23:23:36 customdesigned
|
||||||
|
* Include smfi_progress is SMFIR_PROGRESS defined
|
||||||
|
*
|
||||||
|
* Revision 1.7 2005/10/20 23:04:46 customdesigned
|
||||||
|
* Add optional idx for position of added header.
|
||||||
|
*
|
||||||
|
* Revision 1.6 2005/07/15 22:18:17 customdesigned
|
||||||
|
* Support callback exception policy
|
||||||
|
*
|
||||||
|
* Revision 1.5 2005/06/24 04:20:07 customdesigned
|
||||||
|
* Report context allocation error.
|
||||||
|
*
|
||||||
|
* Revision 1.4 2005/06/24 04:12:43 customdesigned
|
||||||
|
* Remove unused name argument to generic wrappers.
|
||||||
|
*
|
||||||
|
* Revision 1.3 2005/06/24 03:57:35 customdesigned
|
||||||
|
* Handle close called before connect.
|
||||||
|
*
|
||||||
|
* Revision 1.2 2005/06/02 04:18:55 customdesigned
|
||||||
|
* Update copyright notices after reading article on /.
|
||||||
|
*
|
||||||
|
* Revision 1.1.1.2 2005/05/31 18:09:06 customdesigned
|
||||||
|
* Release 0.7.1
|
||||||
|
*
|
||||||
|
* Revision 2.31 2004/08/23 02:24:36 stuart
|
||||||
|
* Support setbacklog
|
||||||
|
*
|
||||||
|
* Revision 2.30 2004/08/21 20:29:53 stuart
|
||||||
|
* Support option of 11 lines max for mlreply.
|
||||||
|
*
|
||||||
|
* Revision 2.29 2004/08/21 04:14:29 stuart
|
||||||
|
* mlreply support
|
||||||
|
*
|
||||||
|
* Revision 2.28 2004/08/21 02:45:21 stuart
|
||||||
|
* Don't leak int constants if module unloaded.
|
||||||
|
*
|
||||||
|
* Revision 2.27 2004/04/06 03:19:59 stuart
|
||||||
|
* Release 0.6.8
|
||||||
|
*
|
||||||
|
* Revision 2.26 2004/03/04 21:43:06 stuart
|
||||||
|
* Fix memory leak by removing unused dynamic template buffer,
|
||||||
|
* thanks again to Alexander Kourakos.
|
||||||
|
*
|
||||||
|
* Revision 2.25 2004/03/01 19:45:03 stuart
|
||||||
|
* Release 0.6.5
|
||||||
|
*
|
||||||
|
* Revision 2.24 2004/03/01 18:56:50 stuart
|
||||||
|
* Support progress reporting.
|
||||||
|
*
|
||||||
|
* Revision 2.23 2004/03/01 18:36:09 stuart
|
||||||
|
* Plug memory leak. Thanks to Alexander Kourakos.
|
||||||
|
*
|
||||||
|
* Revision 2.22 2003/11/02 03:01:46 stuart
|
||||||
|
* Adjust SMTP error codes after careful reading of standard.
|
||||||
|
*
|
||||||
|
* Revision 2.21 2003/06/24 19:57:04 stuart
|
||||||
|
* Allow removing a python milter callback by setting to None.
|
||||||
|
*
|
||||||
|
* Revision 2.20 2003/02/13 17:08:57 stuart
|
||||||
|
* IPV6 support
|
||||||
|
*
|
||||||
|
* Revision 2.19 2003/02/13 16:58:29 stuart
|
||||||
|
* Support passing None to setreply and chgheader.
|
||||||
|
*
|
||||||
|
* Revision 2.18 2002/12/11 16:44:06 stuart
|
||||||
|
* Support QUARANTINE if supported by libmilter.
|
||||||
|
*
|
||||||
|
* Revision 2.17 2002/04/18 20:20:35 stuart
|
||||||
|
* Fix for NULL hostaddr in connect callback from Jason Erickson.
|
||||||
|
*
|
||||||
|
* Revision 2.16 2001/09/26 13:29:09 stuart
|
||||||
|
* sa_len not supported by linux.
|
||||||
|
*
|
||||||
|
* Revision 2.15 2001/09/25 17:28:40 stuart
|
||||||
|
* Copyrights, documentation, release 0.3.1
|
||||||
|
*
|
||||||
|
* Revision 2.14 2001/09/25 00:36:57 stuart
|
||||||
|
* Pass hostaddr to python code in format used by standard socket module.
|
||||||
|
*
|
||||||
|
* Revision 2.13 2001/09/24 23:44:55 stuart
|
||||||
|
* Return old callback from setcallback functions.
|
||||||
|
*
|
||||||
|
* Revision 2.12 2001/09/24 20:02:30 stuart
|
||||||
|
* Remove redundant setpriv
|
||||||
|
*
|
||||||
|
* Revision 2.11 2001/09/23 22:26:35 stuart
|
||||||
|
* Update docs. Streamline Milter.py
|
||||||
|
* update testbms.py to reflect actual sendmail behaviour with multiple
|
||||||
|
* messages per connection.
|
||||||
|
*
|
||||||
|
* Revision 2.10 2001/09/22 15:33:42 stuart
|
||||||
|
* More doc comment updates.
|
||||||
|
*
|
||||||
|
* Revision 2.9 2001/09/22 14:52:27 stuart
|
||||||
|
* Actually return retval in _generic_return.
|
||||||
|
* Go over doc comments.
|
||||||
|
*
|
||||||
|
* Revision 2.8 2001/09/22 01:59:32 stuart
|
||||||
|
* Prevent reentrant call of milter_main, which libmilter doesn't support.
|
||||||
|
*
|
||||||
|
* Revision 2.7 2001/09/22 01:47:37 stuart
|
||||||
|
* Forgot to set milter interp.
|
||||||
|
*
|
||||||
|
* Revision 2.6 2001/09/22 01:23:53 stuart
|
||||||
|
* Added proper threading after research in python docs.
|
||||||
|
*
|
||||||
|
* Revision 2.5 2001/09/21 20:08:51 stuart
|
||||||
|
* Release 0.2.3
|
||||||
|
*
|
||||||
|
* Revision 2.4 2001/09/20 16:18:16 stuart
|
||||||
|
* libmilter checks in_eom state, so we don't have to.
|
||||||
|
*
|
||||||
|
* Revision 2.3 2001/09/19 06:02:33 stuart
|
||||||
|
* Make more stuff static.
|
||||||
|
*
|
||||||
|
* Revision 2.1 2001/09/19 04:24:13 stuart
|
||||||
|
* Use extension type to track context in python.
|
||||||
|
*
|
||||||
|
* Revision 1.4 2001/09/18 18:48:28 stuart
|
||||||
|
* clear private data reference in _clear_context
|
||||||
|
*
|
||||||
|
* Revision 1.3 2001/09/15 04:19:37 stuart
|
||||||
|
* nasty off by 1 mem overwrite bugs in wrap_env
|
||||||
|
* generic_set_callback
|
||||||
|
*
|
||||||
|
* Revision 1.2 2001/09/15 03:15:39 stuart
|
||||||
|
* several bugs fixed, works smoothly
|
||||||
|
*
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef MAX_ML_REPLY
|
#ifndef MAX_ML_REPLY
|
||||||
@@ -71,7 +282,7 @@ $ python setup.py help
|
|||||||
* published. Unfortunately I know of no good way to do this
|
* published. Unfortunately I know of no good way to do this
|
||||||
* other than with OS-specific tests.
|
* other than with OS-specific tests.
|
||||||
*/
|
*/
|
||||||
#if defined(__FreeBSD__) || defined(__linux__) || defined(__sun__) || defined(__GLIBC__) || (defined(__APPLE__) && defined(__MACH__))
|
#if defined(__FreeBSD_kernel__) || defined(__linux__)
|
||||||
#define HAVE_IPV6_RFC2553
|
#define HAVE_IPV6_RFC2553
|
||||||
#include <arpa/inet.h>
|
#include <arpa/inet.h>
|
||||||
#endif
|
#endif
|
||||||
@@ -132,13 +343,7 @@ static struct MilterCallback {
|
|||||||
{ NULL , NULL }
|
{ NULL , NULL }
|
||||||
};
|
};
|
||||||
|
|
||||||
#if PY_MAJOR_VERSION >= 3
|
staticforward struct smfiDesc description; /* forward declaration */
|
||||||
static struct smfiDesc description; /* forward declaration */
|
|
||||||
static PyTypeObject milter_ContextType;
|
|
||||||
#else
|
|
||||||
staticforward struct smfiDesc description; /* forward declaration */
|
|
||||||
staticforward PyTypeObject milter_ContextType;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
static PyObject *MilterError;
|
static PyObject *MilterError;
|
||||||
/* The interpreter instance that called milter.main */
|
/* The interpreter instance that called milter.main */
|
||||||
@@ -150,6 +355,8 @@ typedef struct {
|
|||||||
|
|
||||||
static milter_Diag diag;
|
static milter_Diag diag;
|
||||||
|
|
||||||
|
staticforward PyTypeObject milter_ContextType;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
PyObject_HEAD
|
PyObject_HEAD
|
||||||
SMFICTX *ctx; /* libmilter thread state */
|
SMFICTX *ctx; /* libmilter thread state */
|
||||||
@@ -379,7 +586,7 @@ 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\
|
be multiple body chunks passed to the filter. End-of-lines are\n\
|
||||||
represented as received from SMTP (normally Carriage-Return/Line-Feed).\n\
|
represented as received from SMTP (normally Carriage-Return/Line-Feed).\n\
|
||||||
Function takes args (ctx, chunk) -> int\n\
|
Function takes args (ctx, chunk) -> int\n\
|
||||||
chunk -> bytes - body data";
|
chunk -> String - body data";
|
||||||
|
|
||||||
static PyObject *
|
static PyObject *
|
||||||
milter_set_body_callback(PyObject *self, PyObject *args) {
|
milter_set_body_callback(PyObject *self, PyObject *args) {
|
||||||
@@ -493,11 +700,7 @@ _generic_wrapper(milter_ContextObject *self, PyObject *cb, PyObject *arglist) {
|
|||||||
result = PyEval_CallObject(cb, arglist);
|
result = PyEval_CallObject(cb, arglist);
|
||||||
Py_DECREF(arglist);
|
Py_DECREF(arglist);
|
||||||
if (result == NULL) return _report_exception(self);
|
if (result == NULL) return _report_exception(self);
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
if (!PyLong_Check(result)) {
|
|
||||||
#else
|
|
||||||
if (!PyInt_Check(result)) {
|
if (!PyInt_Check(result)) {
|
||||||
#endif
|
|
||||||
const struct MilterCallback *p;
|
const struct MilterCallback *p;
|
||||||
const char *cbname = "milter";
|
const char *cbname = "milter";
|
||||||
char buf[40];
|
char buf[40];
|
||||||
@@ -512,11 +715,7 @@ _generic_wrapper(milter_ContextObject *self, PyObject *cb, PyObject *arglist) {
|
|||||||
PyErr_SetString(MilterError,buf);
|
PyErr_SetString(MilterError,buf);
|
||||||
return _report_exception(self);
|
return _report_exception(self);
|
||||||
}
|
}
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
retval = PyLong_AS_LONG(result);
|
|
||||||
#else
|
|
||||||
retval = PyInt_AS_LONG(result);
|
retval = PyInt_AS_LONG(result);
|
||||||
#endif
|
|
||||||
Py_DECREF(result);
|
Py_DECREF(result);
|
||||||
_release_thread(self->t);
|
_release_thread(self->t);
|
||||||
return retval;
|
return retval;
|
||||||
@@ -533,11 +732,7 @@ makeipaddr(struct sockaddr_in *addr) {
|
|||||||
sprintf(buf, "%d.%d.%d.%d",
|
sprintf(buf, "%d.%d.%d.%d",
|
||||||
(int) (x>>24) & 0xff, (int) (x>>16) & 0xff,
|
(int) (x>>24) & 0xff, (int) (x>>16) & 0xff,
|
||||||
(int) (x>> 8) & 0xff, (int) (x>> 0) & 0xff);
|
(int) (x>> 8) & 0xff, (int) (x>> 0) & 0xff);
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
return PyUnicode_FromString(buf);
|
|
||||||
#else
|
|
||||||
return PyString_FromString(buf);
|
return PyString_FromString(buf);
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef HAVE_IPV6_SUPPORT
|
#ifdef HAVE_IPV6_SUPPORT
|
||||||
@@ -545,13 +740,8 @@ static PyObject *
|
|||||||
makeip6addr(struct sockaddr_in6 *addr) {
|
makeip6addr(struct sockaddr_in6 *addr) {
|
||||||
char buf[100]; /* must be at least INET6_ADDRSTRLEN + 1 */
|
char buf[100]; /* must be at least INET6_ADDRSTRLEN + 1 */
|
||||||
const char *s = inet_ntop(AF_INET6, &addr->sin6_addr, buf, sizeof buf);
|
const char *s = inet_ntop(AF_INET6, &addr->sin6_addr, buf, sizeof buf);
|
||||||
#if PY_MAJOR_VERSION >= 3
|
if (s) return PyString_FromString(s);
|
||||||
if (s) return PyUnicode_FromString(s);
|
return PyString_FromString("inet6:unknown");
|
||||||
return PyUnicode_FromString("inet6:unknown");
|
|
||||||
#else
|
|
||||||
if (s) return PyString_FromString(s);
|
|
||||||
return PyString_FromString("inet6:unknown");
|
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@@ -642,11 +832,7 @@ generic_env_wrapper(SMFICTX *ctx, PyObject*cb, char **argv) {
|
|||||||
for (i=0;i<count;i++) {
|
for (i=0;i<count;i++) {
|
||||||
/* There's some error checking performed in do_mkvalue() for a string */
|
/* There's some error checking performed in do_mkvalue() for a string */
|
||||||
/* that's not currently done here - it probably should be */
|
/* that's not currently done here - it probably should be */
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
PyObject *o = PyUnicode_FromStringAndSize(argv[i], strlen(argv[i]));
|
|
||||||
#else
|
|
||||||
PyObject *o = PyString_FromStringAndSize(argv[i], strlen(argv[i]));
|
PyObject *o = PyString_FromStringAndSize(argv[i], strlen(argv[i]));
|
||||||
#endif
|
|
||||||
if (o == NULL) { /* out of memory */
|
if (o == NULL) { /* out of memory */
|
||||||
Py_DECREF(arglist);
|
Py_DECREF(arglist);
|
||||||
return _report_exception(self);
|
return _report_exception(self);
|
||||||
@@ -703,11 +889,7 @@ milter_wrap_body(SMFICTX *ctx, u_char *bodyp, size_t bodylen) {
|
|||||||
c = _get_context(ctx);
|
c = _get_context(ctx);
|
||||||
if (!c) return SMFIS_TEMPFAIL;
|
if (!c) return SMFIS_TEMPFAIL;
|
||||||
/* Unclear whether this should be s#, z#, or t# */
|
/* Unclear whether this should be s#, z#, or t# */
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
arglist = Py_BuildValue("(Oy#)", c, bodyp, bodylen);
|
|
||||||
#else
|
|
||||||
arglist = Py_BuildValue("(Os#)", c, bodyp, bodylen);
|
arglist = Py_BuildValue("(Os#)", c, bodyp, bodylen);
|
||||||
#endif
|
|
||||||
return _generic_wrapper(c, body_callback, arglist);
|
return _generic_wrapper(c, body_callback, arglist);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -768,17 +950,20 @@ milter_wrap_negotiate(SMFICTX *ctx,
|
|||||||
rc = _generic_wrapper(c, negotiate_callback, arglist);
|
rc = _generic_wrapper(c, negotiate_callback, arglist);
|
||||||
c->t = t;
|
c->t = t;
|
||||||
if (rc == SMFIS_CONTINUE) {
|
if (rc == SMFIS_CONTINUE) {
|
||||||
|
#if 0 // PyArgs_Parse deprecated and going away
|
||||||
|
if (!PyArgs_Parse(optlist,"[kkkk]",pf0,pf1,pf2,pf3)) {
|
||||||
|
PyErr_Print();
|
||||||
|
PyErr_Clear(); /* must clear since not returning to python */
|
||||||
|
rc = SMFIS_REJECT;
|
||||||
|
}
|
||||||
|
#else
|
||||||
unsigned long *pa[4] = { pf0,pf1,pf2,pf3 };
|
unsigned long *pa[4] = { pf0,pf1,pf2,pf3 };
|
||||||
unsigned long fa[4] = { f0,f1,f2,f3 };
|
unsigned long fa[4] = { f0,f1,f2,f3 };
|
||||||
int len = PyList_Size(optlist);
|
int len = PyList_Size(optlist);
|
||||||
int i;
|
int i;
|
||||||
for (i = 0; i < 4; ++i) {
|
for (i = 0; i < 4; ++i) {
|
||||||
*pa[i] = (i <= len)
|
*pa[i] = (i <= len)
|
||||||
#if PY_MAJOR_VERSION >= 3
|
? PyInt_AsUnsignedLongMask(PyList_GET_ITEM(optlist,i))
|
||||||
? PyLong_AsUnsignedLongMask(PyList_GET_ITEM(optlist,i))
|
|
||||||
#else
|
|
||||||
? PyInt_AsUnsignedLongMask(PyList_GET_ITEM(optlist,i))
|
|
||||||
#endif
|
|
||||||
: fa[i];
|
: fa[i];
|
||||||
}
|
}
|
||||||
if (PyErr_Occurred()) {
|
if (PyErr_Occurred()) {
|
||||||
@@ -786,6 +971,7 @@ milter_wrap_negotiate(SMFICTX *ctx,
|
|||||||
PyErr_Clear();
|
PyErr_Clear();
|
||||||
rc = SMFIS_REJECT;
|
rc = SMFIS_REJECT;
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
else if (rc != SMFIS_ALL_OPTS)
|
else if (rc != SMFIS_ALL_OPTS)
|
||||||
rc = SMFIS_REJECT;
|
rc = SMFIS_REJECT;
|
||||||
@@ -1145,7 +1331,7 @@ Change/delete a header in the message. \n\
|
|||||||
It is not checked for standards compliance; the mail filter\n\
|
It is not checked for standards compliance; the mail filter\n\
|
||||||
must ensure that no protocols are violated as a result of adding this header.\n\
|
must ensure that no protocols are violated as a result of adding this header.\n\
|
||||||
field - header field name\n\
|
field - header field name\n\
|
||||||
int - the Nth occurrence of this header\n\
|
int - the Nth occurence of this header\n\
|
||||||
value - header field value\n\
|
value - header field value\n\
|
||||||
field and value are strings.\n\
|
field and value are strings.\n\
|
||||||
This function can only be called from the EOM callback.";
|
This function can only be called from the EOM callback.";
|
||||||
@@ -1365,12 +1551,10 @@ static PyMethodDef context_methods[] = {
|
|||||||
{ NULL, NULL }
|
{ NULL, NULL }
|
||||||
};
|
};
|
||||||
|
|
||||||
#if PY_MAJOR_VERSION < 3
|
|
||||||
static PyObject *
|
static PyObject *
|
||||||
milter_Context_getattr(PyObject *self, char *name) {
|
milter_Context_getattr(PyObject *self, char *name) {
|
||||||
return Py_FindMethod(context_methods, self, name);
|
return Py_FindMethod(context_methods, self, name);
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
|
|
||||||
static struct smfiDesc description = { /* Set some reasonable defaults */
|
static struct smfiDesc description = { /* Set some reasonable defaults */
|
||||||
"pythonfilter",
|
"pythonfilter",
|
||||||
@@ -1420,23 +1604,14 @@ static PyMethodDef milter_methods[] = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
static PyTypeObject milter_ContextType = {
|
static PyTypeObject milter_ContextType = {
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
PyVarObject_HEAD_INIT(&PyType_Type,0)
|
|
||||||
"milter.Context",
|
|
||||||
#else
|
|
||||||
PyObject_HEAD_INIT(&PyType_Type)
|
PyObject_HEAD_INIT(&PyType_Type)
|
||||||
0,
|
0,
|
||||||
"milterContext",
|
"milterContext",
|
||||||
#endif
|
|
||||||
sizeof(milter_ContextObject),
|
sizeof(milter_ContextObject),
|
||||||
0,
|
0,
|
||||||
milter_Context_dealloc, /* tp_dealloc */
|
milter_Context_dealloc, /* tp_dealloc */
|
||||||
0, /* tp_print */
|
0, /* tp_print */
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
0, /* tp_getattr */
|
|
||||||
#else
|
|
||||||
milter_Context_getattr, /* tp_getattr */
|
milter_Context_getattr, /* tp_getattr */
|
||||||
#endif
|
|
||||||
0, /* tp_setattr */
|
0, /* tp_setattr */
|
||||||
0, /* tp_compare */
|
0, /* tp_compare */
|
||||||
0, /* tp_repr */
|
0, /* tp_repr */
|
||||||
@@ -1450,15 +1625,6 @@ static PyTypeObject milter_ContextType = {
|
|||||||
0, /* tp_setattro */
|
0, /* tp_setattro */
|
||||||
0, /* tp_as_buffer */
|
0, /* tp_as_buffer */
|
||||||
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
NULL, /* Documentation string */
|
|
||||||
0, /* call function for all accessible objects */
|
|
||||||
0, /* delete references to contained objects */
|
|
||||||
0, /* rich comparisons */
|
|
||||||
0, /* weak reference enabler */
|
|
||||||
0, 0, /* Iterators */
|
|
||||||
context_methods, /* Attribute descriptor and subclassing stuff */
|
|
||||||
#endif
|
|
||||||
};
|
};
|
||||||
|
|
||||||
static const char milter_documentation[] =
|
static const char milter_documentation[] =
|
||||||
@@ -1468,45 +1634,17 @@ Libmilter is currently marked FFR, and needs to be explicitly installed.\n\
|
|||||||
See <sendmailsource>/libmilter/README for details on setting it up.\n";
|
See <sendmailsource>/libmilter/README for details on setting it up.\n";
|
||||||
|
|
||||||
static void setitem(PyObject *d,const char *name,long val) {
|
static void setitem(PyObject *d,const char *name,long val) {
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
PyObject *v = PyLong_FromLong(val);
|
|
||||||
#else
|
|
||||||
PyObject *v = PyInt_FromLong(val);
|
PyObject *v = PyInt_FromLong(val);
|
||||||
#endif
|
|
||||||
PyDict_SetItemString(d,name,v);
|
PyDict_SetItemString(d,name,v);
|
||||||
Py_DECREF(v);
|
Py_DECREF(v);
|
||||||
}
|
}
|
||||||
|
|
||||||
#if PY_MAJOR_VERSION >= 3
|
void
|
||||||
|
initmilter(void) {
|
||||||
static struct PyModuleDef moduledef = {
|
|
||||||
PyModuleDef_HEAD_INIT,
|
|
||||||
"milter", /* m_name */
|
|
||||||
milter_documentation,/* m_doc */
|
|
||||||
-1, /* m_size */
|
|
||||||
milter_methods, /* m_methods */
|
|
||||||
NULL, /* m_reload */
|
|
||||||
NULL, /* m_traverse */
|
|
||||||
NULL, /* m_clear */
|
|
||||||
NULL, /* m_free */
|
|
||||||
};
|
|
||||||
|
|
||||||
PyMODINIT_FUNC PyInit_milter(void) {
|
|
||||||
PyObject *m, *d;
|
|
||||||
|
|
||||||
if (PyType_Ready(&milter_ContextType) < 0)
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
m = PyModule_Create(&moduledef);
|
|
||||||
if (m == NULL) return NULL;
|
|
||||||
#else
|
|
||||||
|
|
||||||
void initmilter(void) {
|
|
||||||
PyObject *m, *d;
|
PyObject *m, *d;
|
||||||
|
|
||||||
m = Py_InitModule4("milter", milter_methods, milter_documentation,
|
m = Py_InitModule4("milter", milter_methods, milter_documentation,
|
||||||
(PyObject*)NULL, PYTHON_API_VERSION);
|
(PyObject*)NULL, PYTHON_API_VERSION);
|
||||||
#endif
|
|
||||||
d = PyModule_GetDict(m);
|
d = PyModule_GetDict(m);
|
||||||
MilterError = PyErr_NewException("milter.error", NULL, NULL);
|
MilterError = PyErr_NewException("milter.error", NULL, NULL);
|
||||||
PyDict_SetItemString(d,"error", MilterError);
|
PyDict_SetItemString(d,"error", MilterError);
|
||||||
@@ -1572,7 +1710,4 @@ void initmilter(void) {
|
|||||||
setitem(d,"DISCARD", SMFIS_DISCARD);
|
setitem(d,"DISCARD", SMFIS_DISCARD);
|
||||||
setitem(d,"ACCEPT", SMFIS_ACCEPT);
|
setitem(d,"ACCEPT", SMFIS_ACCEPT);
|
||||||
setitem(d,"TEMPFAIL", SMFIS_TEMPFAIL);
|
setitem(d,"TEMPFAIL", SMFIS_TEMPFAIL);
|
||||||
#if PY_MAJOR_VERSION >= 3
|
|
||||||
return m;
|
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -93,34 +93,26 @@
|
|||||||
# Copyright 2001,2002,2003,2004,2005 Business Management Systems, Inc.
|
# Copyright 2001,2002,2003,2004,2005 Business Management Systems, Inc.
|
||||||
# This code is under the GNU General Public License. See COPYING for details.
|
# This code is under the GNU General Public License. See COPYING for details.
|
||||||
|
|
||||||
from __future__ import print_function
|
import StringIO
|
||||||
try:
|
|
||||||
from io import BytesIO, StringIO
|
|
||||||
except:
|
|
||||||
from StringIO import StringIO
|
|
||||||
BytesIO = StringIO
|
|
||||||
import socket
|
import socket
|
||||||
import Milter
|
import Milter
|
||||||
import zipfile
|
import zipfile
|
||||||
import sys
|
|
||||||
|
|
||||||
import email
|
import email
|
||||||
from email.message import Message
|
import email.Message
|
||||||
try:
|
from email.Message import Message
|
||||||
from email.generator import BytesGenerator
|
from email.Generator import Generator
|
||||||
from email import message_from_binary_file
|
from email.Utils import quote
|
||||||
except:
|
from email import Utils
|
||||||
from email.generator import Generator as BytesGenerator
|
from email.Parser import Parser
|
||||||
from email import message_from_file as message_from_binary_file
|
from email import Errors
|
||||||
from email.utils import quote
|
|
||||||
|
|
||||||
if not getattr(Message,'as_bytes',None):
|
from types import ListType,StringType
|
||||||
Message.as_bytes = Message.as_string
|
|
||||||
|
|
||||||
## Return a list of filenames in a zip file.
|
## Return a list of filenames in a zip file.
|
||||||
# Embedded zip files are recursively expanded.
|
# Embedded zip files are recursively expanded.
|
||||||
def zipnames(txt):
|
def zipnames(txt):
|
||||||
fp = BytesIO(txt)
|
fp = StringIO.StringIO(txt)
|
||||||
zipf = zipfile.ZipFile(fp,'r')
|
zipf = zipfile.ZipFile(fp,'r')
|
||||||
names = []
|
names = []
|
||||||
for nm in zipf.namelist():
|
for nm in zipf.namelist():
|
||||||
@@ -131,7 +123,7 @@ def zipnames(txt):
|
|||||||
|
|
||||||
## Fix multipart handling in email.Generator.
|
## Fix multipart handling in email.Generator.
|
||||||
#
|
#
|
||||||
class MimeGenerator(BytesGenerator):
|
class MimeGenerator(Generator):
|
||||||
def _dispatch(self, msg):
|
def _dispatch(self, msg):
|
||||||
# Get the Content-Type: for the message, then try to dispatch to
|
# Get the Content-Type: for the message, then try to dispatch to
|
||||||
# self._handle_<maintype>_<subtype>(). If there's no handler for the
|
# self._handle_<maintype>_<subtype>(). If there's no handler for the
|
||||||
@@ -141,7 +133,7 @@ class MimeGenerator(BytesGenerator):
|
|||||||
if msg.is_multipart() and main.lower() != 'multipart':
|
if msg.is_multipart() and main.lower() != 'multipart':
|
||||||
self._handle_multipart(msg)
|
self._handle_multipart(msg)
|
||||||
else:
|
else:
|
||||||
BytesGenerator._dispatch(self,msg)
|
Generator._dispatch(self,msg)
|
||||||
|
|
||||||
def unquote(s):
|
def unquote(s):
|
||||||
"""Remove quotes from a string."""
|
"""Remove quotes from a string."""
|
||||||
@@ -158,17 +150,19 @@ def unquote(s):
|
|||||||
return s[1:-1]
|
return s[1:-1]
|
||||||
return s
|
return s
|
||||||
|
|
||||||
|
from types import TupleType
|
||||||
|
|
||||||
def _unquotevalue(value):
|
def _unquotevalue(value):
|
||||||
if isinstance(value, tuple):
|
if isinstance(value, TupleType):
|
||||||
return value[0], value[1], unquote(value[2])
|
return value[0], value[1], unquote(value[2])
|
||||||
else:
|
else:
|
||||||
return unquote(value)
|
return unquote(value)
|
||||||
|
|
||||||
#email.Message._unquotevalue = _unquotevalue
|
#email.Message._unquotevalue = _unquotevalue
|
||||||
|
|
||||||
from email.message import _parseparam
|
from email.Message import _parseparam
|
||||||
|
|
||||||
## Enhance email.message.Message
|
## Enhance email.Message
|
||||||
#
|
#
|
||||||
# Tracks modifications to headers of body or any part independently.
|
# Tracks modifications to headers of body or any part independently.
|
||||||
|
|
||||||
@@ -209,7 +203,7 @@ class MimeMessage(Message):
|
|||||||
interpret as a name - and hence decide to execute this message."""
|
interpret as a name - and hence decide to execute this message."""
|
||||||
names = []
|
names = []
|
||||||
for attr,val in self._get_params_preserve([],'content-type'):
|
for attr,val in self._get_params_preserve([],'content-type'):
|
||||||
if isinstance(val, tuple):
|
if isinstance(val, TupleType):
|
||||||
# It's an RFC 2231 encoded parameter
|
# It's an RFC 2231 encoded parameter
|
||||||
newvalue = _unquotevalue(val)
|
newvalue = _unquotevalue(val)
|
||||||
if val[0]:
|
if val[0]:
|
||||||
@@ -244,9 +238,9 @@ class MimeMessage(Message):
|
|||||||
g = MimeGenerator(file)
|
g = MimeGenerator(file)
|
||||||
g.flatten(self,unixfrom=unixfrom)
|
g.flatten(self,unixfrom=unixfrom)
|
||||||
|
|
||||||
def as_bytes(self, unixfrom=False):
|
def as_string(self, unixfrom=False):
|
||||||
"Return the entire formatted message as a string."
|
"Return the entire formatted message as a string."
|
||||||
fp = BytesIO()
|
fp = StringIO.StringIO()
|
||||||
self.dump(fp,unixfrom=unixfrom)
|
self.dump(fp,unixfrom=unixfrom)
|
||||||
return fp.getvalue()
|
return fp.getvalue()
|
||||||
|
|
||||||
@@ -307,7 +301,7 @@ class MimeMessage(Message):
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
def message_from_file(fp):
|
def message_from_file(fp):
|
||||||
msg = message_from_binary_file(fp,MimeMessage)
|
msg = email.message_from_file(fp,MimeMessage)
|
||||||
for part in msg.walk():
|
for part in msg.walk():
|
||||||
part.modified = False
|
part.modified = False
|
||||||
assert not msg.ismodified()
|
assert not msg.ismodified()
|
||||||
@@ -318,7 +312,7 @@ ade,adp,asd,asx,asp,bas,bat,chm,cmd,com,cpl,crt,dll,exe,hlp,hta,inf,ins,isp,js,
|
|||||||
jse,lnk,mdb,mde,msc,msi,msp,mst,ocx,pcd,pif,reg,scr,sct,shs,url,vb,vbe,vbs,wsc,
|
jse,lnk,mdb,mde,msc,msi,msp,mst,ocx,pcd,pif,reg,scr,sct,shs,url,vb,vbe,vbs,wsc,
|
||||||
wsf,wsh
|
wsf,wsh
|
||||||
""".split())
|
""".split())
|
||||||
bad_extensions = ['.' + x for x in extlist.split(',')]
|
bad_extensions = map(lambda x:'.' + x,extlist.split(','))
|
||||||
|
|
||||||
def check_ext(name):
|
def check_ext(name):
|
||||||
"Check a name for dangerous Winblows extensions."
|
"Check a name for dangerous Winblows extensions."
|
||||||
@@ -357,6 +351,8 @@ def check_name(msg,savname=None,ckname=check_ext,scan_zip=False):
|
|||||||
msg["Content-Type"] = "text/plain; name="+name
|
msg["Content-Type"] = "text/plain; name="+name
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
|
import email.Iterators
|
||||||
|
|
||||||
def check_attachments(msg,check):
|
def check_attachments(msg,check):
|
||||||
"""Scan attachments.
|
"""Scan attachments.
|
||||||
msg MimeMessage
|
msg MimeMessage
|
||||||
@@ -402,21 +398,18 @@ class _defang:
|
|||||||
# emulate old defang function
|
# emulate old defang function
|
||||||
defang = _defang()
|
defang = _defang()
|
||||||
|
|
||||||
if sys.version < '3.0.0':
|
import sgmllib
|
||||||
from sgmllib import SGMLParser as HTMLParser
|
|
||||||
else:
|
|
||||||
from Milter.sgmllib import SGMLParser as HTMLParser
|
|
||||||
|
|
||||||
import re
|
import re
|
||||||
declname = re.compile(r'[a-zA-Z][-_.a-zA-Z0-9]*\s*')
|
declname = re.compile(r'[a-zA-Z][-_.a-zA-Z0-9]*\s*')
|
||||||
declstringlit = re.compile(r'(\'[^\']*\'|"[^"]*")\s*')
|
declstringlit = re.compile(r'(\'[^\']*\'|"[^"]*")\s*')
|
||||||
|
|
||||||
class SGMLFilter(HTMLParser):
|
class SGMLFilter(sgmllib.SGMLParser):
|
||||||
"""Parse HTML and pass through all constructs unchanged. It is intended for
|
"""Parse HTML and pass through all constructs unchanged. It is intended for
|
||||||
derived classes to implement exceptional processing for selected cases.
|
derived classes to implement exceptional processing for selected cases.
|
||||||
"""
|
"""
|
||||||
def __init__(self,out):
|
def __init__(self,out):
|
||||||
HTMLParser.__init__(self)
|
sgmllib.SGMLParser.__init__(self)
|
||||||
self.out = out
|
self.out = out
|
||||||
|
|
||||||
def handle_comment(self,comment):
|
def handle_comment(self,comment):
|
||||||
@@ -447,7 +440,7 @@ class SGMLFilter(HTMLParser):
|
|||||||
self.out.write("<!%s>" % data)
|
self.out.write("<!%s>" % data)
|
||||||
|
|
||||||
def write(self,buf):
|
def write(self,buf):
|
||||||
"Act like a writer. Why doesn't HTMLParser do this by default?"
|
"Act like a writer. Why doesn't SGMLParser do this by default?"
|
||||||
self.feed(buf)
|
self.feed(buf)
|
||||||
|
|
||||||
# Python-2.1 sgmllib rejects illegal declarations. Since various Microsoft
|
# Python-2.1 sgmllib rejects illegal declarations. Since various Microsoft
|
||||||
@@ -474,7 +467,7 @@ class SGMLFilter(HTMLParser):
|
|||||||
elif c in "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ":
|
elif c in "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ":
|
||||||
m = declname.match(rawdata, j)
|
m = declname.match(rawdata, j)
|
||||||
if not m:
|
if not m:
|
||||||
# incomplete or an error?
|
# incomplete or an error?
|
||||||
return -1
|
return -1
|
||||||
j = m.end()
|
j = m.end()
|
||||||
else:
|
else:
|
||||||
@@ -490,14 +483,11 @@ class HTMLScriptFilter(SGMLFilter):
|
|||||||
self.modified = False
|
self.modified = False
|
||||||
self.msg = "<!-- WARNING: embedded script removed -->"
|
self.msg = "<!-- WARNING: embedded script removed -->"
|
||||||
def start_script(self,unused):
|
def start_script(self,unused):
|
||||||
#print('beg script',unused)
|
|
||||||
self.ignoring += 1
|
self.ignoring += 1
|
||||||
self.modified = True
|
self.modified = True
|
||||||
|
self.out.write(self.msg)
|
||||||
def end_script(self):
|
def end_script(self):
|
||||||
#print('end script')
|
|
||||||
self.ignoring -= 1
|
self.ignoring -= 1
|
||||||
if not self.ignoring:
|
|
||||||
self.out.write(self.msg)
|
|
||||||
def handle_data(self,data):
|
def handle_data(self,data):
|
||||||
if not self.ignoring: SGMLFilter.handle_data(self,data)
|
if not self.ignoring: SGMLFilter.handle_data(self,data)
|
||||||
def handle_comment(self,comment):
|
def handle_comment(self,comment):
|
||||||
@@ -512,14 +502,14 @@ def check_html(msg,savname=None):
|
|||||||
if name and name.lower().endswith(".htm"):
|
if name and name.lower().endswith(".htm"):
|
||||||
msgtype = 'text/html'
|
msgtype = 'text/html'
|
||||||
if msgtype == 'text/html':
|
if msgtype == 'text/html':
|
||||||
out = StringIO()
|
out = StringIO.StringIO()
|
||||||
htmlfilter = HTMLScriptFilter(out)
|
htmlfilter = HTMLScriptFilter(out)
|
||||||
try:
|
try:
|
||||||
htmlfilter.write(msg.get_payload(decode=True).decode())
|
htmlfilter.write(msg.get_payload(decode=True))
|
||||||
htmlfilter.close()
|
htmlfilter.close()
|
||||||
#except sgmllib.SGMLParseError:
|
#except sgmllib.SGMLParseError:
|
||||||
except:
|
except:
|
||||||
mimetools.copyliteral(msg.get_payload(),open('debug.out','wb'))
|
#mimetools.copyliteral(msg.get_payload(),open('debug.out','w')
|
||||||
htmlfilter.close()
|
htmlfilter.close()
|
||||||
hostname = socket.gethostname()
|
hostname = socket.gethostname()
|
||||||
msg.set_payload(
|
msg.set_payload(
|
||||||
@@ -538,17 +528,18 @@ def check_html(msg,savname=None):
|
|||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
def _list_attach(msg):
|
def _list_attach(msg):
|
||||||
t = msg.get_content_type()
|
t = msg.get_content_type()
|
||||||
p = msg.get_payload(decode=True)
|
p = msg.get_payload(decode=True)
|
||||||
print(msg.get_filename(),msg.get_content_type(),type(p))
|
print msg.get_filename(),msg.get_content_type(),type(p)
|
||||||
msg = msg.get_submsg()
|
msg = msg.get_submsg()
|
||||||
if isinstance(msg,Message):
|
if isinstance(msg,Message):
|
||||||
return check_attachments(msg,_list_attach)
|
return check_attachments(msg,_list_attach)
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
for fname in sys.argv[1:]:
|
for fname in sys.argv[1:]:
|
||||||
fp = open(fname,'rb')
|
fp = open(fname)
|
||||||
msg = message_from_file(fp)
|
msg = message_from_file(fp)
|
||||||
email.iterators._structure(msg)
|
email.Iterators._structure(msg)
|
||||||
check_attachments(msg,_list_attach)
|
check_attachments(msg,_list_attach)
|
||||||
|
|||||||
@@ -1,6 +0,0 @@
|
|||||||
Check Description Justification
|
|
||||||
E111 req indent 4 Creates more continuation lines
|
|
||||||
E114 req indent 4 cmnt Same
|
|
||||||
E231 req space after , makes calls like print() harder to read
|
|
||||||
E266 no ## Required by Doxygen
|
|
||||||
W291 trailing spaces in cmnt Needed for space preserving para reformat
|
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
#!/bin/sh
|
|
||||||
ignore=`awk -F\\\\t '{ print $1 }' pep8.dat | tail -n +2`
|
|
||||||
a=(${ignore})
|
|
||||||
list=$(echo "${a[@]}"|tr '[ ]' '[,]')
|
|
||||||
echo python3 -m pep8 --ignore="$list" $@
|
|
||||||
+55
-182
@@ -1,223 +1,96 @@
|
|||||||
# we don't want to provide private python extension libs
|
%define __python python2.6
|
||||||
%global sum Python interface to sendmail milter API
|
%define pythonbase python
|
||||||
%global __provides_exclude_from ^(%{python2_sitearch})/.*\\.so$
|
|
||||||
%if 0%{?epel} == 7
|
|
||||||
%global python3 python34
|
|
||||||
%else
|
|
||||||
%global python3 python3
|
|
||||||
%endif
|
|
||||||
|
|
||||||
Summary: %{sum}
|
%define libdir %{_libdir}/pymilter
|
||||||
Name: python-pymilter
|
%{!?python_sitearch: %define python_sitearch %(%{__python} -c "from distutils.sysconfig import get_python_lib; print get_python_lib(1)")}
|
||||||
Version: 1.0.4
|
|
||||||
Release: 1%{?dist}
|
Summary: Python interface to sendmail milter API
|
||||||
Url: http://bmsi.com/pymilter
|
Name: %{pythonbase}-pymilter
|
||||||
Source: https://github.com/sdgathman/pymilter/archive/pymilter-%{version}.tar.gz
|
Version: 1.0
|
||||||
Source1: tmpfiles-python-pymilter.conf
|
Release: 1%{dist}
|
||||||
# remove unit tests that require network for check
|
Source: http://downloads.sourceforge.net/pymilter/pymilter-%{version}.tar.gz
|
||||||
Patch: pymilter-check.patch
|
Source1: pymilter.te
|
||||||
License: GPLv2+
|
License: GPLv2+
|
||||||
Group: Development/Libraries
|
Group: Development/Libraries
|
||||||
BuildRequires: python2-devel, %{python3}-devel, sendmail-devel >= 8.13
|
BuildRoot: %{_tmppath}/%{name}-%{version}-%{release}-root
|
||||||
|
Url: http://www.bmsi.com/python/milter.html
|
||||||
# python-2.6.4 gets RuntimeError: not holding the import lock
|
# python-2.6.4 gets RuntimeError: not holding the import lock
|
||||||
|
Requires: %{pythonbase} >= 2.6.5, sendmail >= 8.13
|
||||||
# Need python2.6 specific pydns, not the version for system python
|
# Need python2.6 specific pydns, not the version for system python
|
||||||
BuildRequires: gcc
|
Requires: %{pythonbase}-pydns
|
||||||
|
# Needed for callbacks, not a core function but highly useful for milters
|
||||||
|
BuildRequires: ed, %{pythonbase}-devel, sendmail-devel >= 8.13
|
||||||
|
|
||||||
%global _description\
|
%description
|
||||||
This is a python extension module to enable python scripts to\
|
This is a python extension module to enable python scripts to
|
||||||
attach to sendmail's libmilter functionality. Additional python\
|
attach to sendmail's libmilter functionality. Additional python
|
||||||
modules provide for navigating and modifying MIME parts, sending\
|
modules provide for navigating and modifying MIME parts, sending
|
||||||
DSNs, and doing CBV.
|
DSNs, and doing CBV.
|
||||||
|
|
||||||
%description %_description
|
|
||||||
|
|
||||||
%package -n python2-pymilter
|
|
||||||
Summary: %{sum}
|
|
||||||
%if 0%{?epel} >= 6
|
|
||||||
Requires: python-pydns
|
|
||||||
%else
|
|
||||||
Requires: python2-pydns
|
|
||||||
%endif
|
|
||||||
Requires: %{name}-common = %{version}-%{release}
|
|
||||||
%{?python_provide:%python_provide python2-pymilter}
|
|
||||||
|
|
||||||
%description -n python2-pymilter %_description
|
|
||||||
|
|
||||||
%package -n %{python3}-pymilter
|
|
||||||
Summary: %{sum}
|
|
||||||
%if 0%{?fedora} >= 26
|
|
||||||
Requires: %{python3}-py3dns
|
|
||||||
%endif
|
|
||||||
Requires: %{name}-common = %{version}-%{release}
|
|
||||||
%{?python_provide:%python_provide %{python3}-pymilter}
|
|
||||||
|
|
||||||
%description -n %{python3}-pymilter %_description
|
|
||||||
|
|
||||||
%package common
|
|
||||||
Summary: Common files and directories for python milters
|
|
||||||
BuildArch: noarch
|
|
||||||
|
|
||||||
%description common
|
|
||||||
Common files and directories used for python milters
|
|
||||||
|
|
||||||
%package selinux
|
%package selinux
|
||||||
Summary: SELinux policy module for pymilter
|
Summary: SELinux policy module for pymilter
|
||||||
Group: System Environment/Base
|
Group: System Environment/Base
|
||||||
Requires: policycoreutils, selinux-policy-targeted
|
Requires: policycoreutils, selinux-policy, %{name}
|
||||||
Requires: %{name} = %{version}-%{release}
|
BuildRequires: policycoreutils, checkpolicy
|
||||||
BuildArch: noarch
|
|
||||||
BuildRequires: policycoreutils, checkpolicy, selinux-policy-devel
|
|
||||||
%if 0%{?epel} >= 6
|
|
||||||
BuildRequires: policycoreutils-python
|
|
||||||
%else
|
|
||||||
BuildRequires: policycoreutils-python-utils
|
|
||||||
%endif
|
|
||||||
|
|
||||||
%description selinux
|
%description selinux
|
||||||
Give sendmail_t additional access to stream sockets used to communicate
|
SELinux policy module for using pymilter with sendmail with selinux enforcing
|
||||||
with milters.
|
|
||||||
|
|
||||||
%prep
|
%prep
|
||||||
%setup -q -n pymilter-pymilter-%{version}
|
%setup -q -n pymilter-%{version}
|
||||||
%patch -p1 -b .check
|
cp %{SOURCE1} pymilter.te
|
||||||
|
|
||||||
%build
|
%build
|
||||||
%py2_build
|
env CFLAGS="$RPM_OPT_FLAGS" %{__python} setup.py build
|
||||||
#patch -p1 -b -z .py3 <milter.patch # not needed since 1.0.3
|
|
||||||
%py3_build
|
|
||||||
checkmodule -m -M -o pymilter.mod pymilter.te
|
checkmodule -m -M -o pymilter.mod pymilter.te
|
||||||
semodule_package -o pymilter.pp -m pymilter.mod
|
semodule_package -o pymilter.pp -m pymilter.mod
|
||||||
|
|
||||||
%install
|
%install
|
||||||
%py2_install
|
rm -rf $RPM_BUILD_ROOT
|
||||||
%py3_install
|
%{__python} setup.py install --root=$RPM_BUILD_ROOT
|
||||||
|
mkdir -p $RPM_BUILD_ROOT%{_localstatedir}/run/milter
|
||||||
mkdir -p %{buildroot}/run/milter
|
mkdir -p $RPM_BUILD_ROOT%{_localstatedir}/log/milter
|
||||||
mkdir -p %{buildroot}%{_localstatedir}/log/milter
|
mkdir -p $RPM_BUILD_ROOT%{libdir}
|
||||||
mkdir -p %{buildroot}%{_libexecdir}/milter
|
|
||||||
mkdir -p %{buildroot}%{_prefix}/lib/tmpfiles.d
|
|
||||||
install -m 0644 %{SOURCE1} %{buildroot}%{_prefix}/lib/tmpfiles.d/%{name}.conf
|
|
||||||
|
|
||||||
# install selinux modules
|
# install selinux modules
|
||||||
mkdir -p %{buildroot}%{_datadir}/selinux/targeted
|
mkdir -p %{buildroot}%{_datadir}/selinux/targeted
|
||||||
cp -p pymilter.pp %{buildroot}%{_datadir}/selinux/targeted
|
cp -p pymilter.pp %{buildroot}%{_datadir}/selinux/targeted
|
||||||
|
|
||||||
%check
|
%files
|
||||||
py2path=$(ls -d build/lib.linux-*-2.*)
|
%defattr(-,root,root,-)
|
||||||
py3path=$(ls -d build/lib.linux-*-3.*)
|
|
||||||
PYTHONPATH=${py2path}:. python2 test.py &&
|
|
||||||
PYTHONPATH=${py3path}:. python3 test.py
|
|
||||||
|
|
||||||
%files -n python2-pymilter
|
|
||||||
%license COPYING
|
|
||||||
%doc README ChangeLog NEWS TODO CREDITS sample.py milter-template.py
|
%doc README ChangeLog NEWS TODO CREDITS sample.py milter-template.py
|
||||||
%{python2_sitearch}/*
|
%{python_sitearch}/*
|
||||||
|
%{libdir}
|
||||||
%files -n %{python3}-pymilter
|
%dir %attr(0755,mail,mail) %{_localstatedir}/run/milter
|
||||||
%license COPYING
|
|
||||||
%doc README ChangeLog NEWS TODO CREDITS sample.py milter-template.py
|
|
||||||
%{python3_sitearch}/*
|
|
||||||
|
|
||||||
%files common
|
|
||||||
%dir %{_libexecdir}/milter
|
|
||||||
%{_prefix}/lib/tmpfiles.d/%{name}.conf
|
|
||||||
%dir %attr(0755,mail,mail) %{_localstatedir}/log/milter
|
%dir %attr(0755,mail,mail) %{_localstatedir}/log/milter
|
||||||
%dir %attr(0755,mail,mail) /run/milter
|
|
||||||
|
|
||||||
%files selinux
|
%files selinux
|
||||||
%doc pymilter.te
|
%doc pymilter.te
|
||||||
%{_datadir}/selinux/targeted/*
|
%{_datadir}/selinux/targeted/*
|
||||||
|
|
||||||
|
%clean
|
||||||
|
rm -rf $RPM_BUILD_ROOT
|
||||||
|
|
||||||
%post selinux
|
%post selinux
|
||||||
%{_sbindir}/semodule -s targeted -i %{_datadir}/selinux/targeted/pymilter.pp \
|
/usr/sbin/semodule -s targeted -i %{_datadir}/selinux/targeted/pymilter.pp \
|
||||||
&>/dev/null || :
|
&>/dev/null || :
|
||||||
|
|
||||||
%postun selinux
|
%postun selinux
|
||||||
if [ $1 -eq 0 ] ; then
|
if [ $1 -eq 0 ] ; then
|
||||||
%{_sbindir}/semodule -s targeted -r pymilter &> /dev/null || :
|
/usr/sbin/semodule -s targeted -r pymilter &> /dev/null || :
|
||||||
fi
|
fi
|
||||||
|
|
||||||
%changelog
|
%changelog
|
||||||
* Wed Apr 17 2019 Stuart Gathman <stuart@gathman.org> - 1.0.4-1
|
* Sat Mar 1 2014 Stuart Gathman <stuart@gathman.org> 1.0-2
|
||||||
- New upstream release: cleanup unused files, additional platform support
|
- Remove start.sh to track EPEL repository, suggest daemonize as replacement
|
||||||
- Minor doc updates
|
- Selinux subpackage should not care about pymilter version
|
||||||
|
|
||||||
* Sun Dec 23 2018 Stuart Gathman <stuart@gathman.org> - 1.0.3-1
|
* Wed Jun 26 2013 Stuart Gathman <stuart@gathman.org> 1.0-1
|
||||||
- New upstream release
|
- Allow ACCEPT as untrapped exception policy
|
||||||
- patch step for python3 no longer required in build
|
- Optional dir for getaddrset and getaddrdict in Milter.config
|
||||||
|
- Show registered milter name in untrapped exception message.
|
||||||
* Sat Aug 4 2018 Stuart Gathman <stuart@gathman.org> - 1.0.2-4
|
- Include selinux subpackage
|
||||||
- Add unit tests to %%check
|
- Provide Milter.greylist export and Milter.greylist import to migrate data
|
||||||
|
|
||||||
* Sat Aug 4 2018 Stuart Gathman <stuart@gathman.org> - 1.0.2-3
|
|
||||||
- use libexec instead of libdir
|
|
||||||
|
|
||||||
* Sat Aug 4 2018 Stuart Gathman <stuart@gathman.org> - 1.0.2-2
|
|
||||||
- add python34 subpackage on el7
|
|
||||||
|
|
||||||
* Sat Aug 4 2018 Stuart Gathman <stuart@gathman.org> - 1.0.2-1
|
|
||||||
- build for both python2 and python3
|
|
||||||
- add selinux policy allowing sendmail_t access to milters
|
|
||||||
|
|
||||||
* Tue Jul 17 2018 Miro Hrončok <mhroncok@redhat.com> - 1.0-13
|
|
||||||
- Update Python macros to new packaging standards
|
|
||||||
(See https://fedoraproject.org/wiki/Changes/Move_usr_bin_python_into_separate_package)
|
|
||||||
|
|
||||||
* Sat Jul 14 2018 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-12
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_29_Mass_Rebuild
|
|
||||||
|
|
||||||
* Fri Feb 09 2018 Iryna Shcherbina <ishcherb@redhat.com> - 1.0-11
|
|
||||||
- Update Python 2 dependency declarations to new packaging standards
|
|
||||||
(See https://fedoraproject.org/wiki/FinalizingFedoraSwitchtoPython3)
|
|
||||||
|
|
||||||
* Fri Feb 09 2018 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-10
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_28_Mass_Rebuild
|
|
||||||
|
|
||||||
* Fri Feb 09 2018 Igor Gnatenko <ignatenkobrain@fedoraproject.org> - 1.0-9
|
|
||||||
- Escape macros in %%changelog
|
|
||||||
|
|
||||||
* Sat Aug 19 2017 Zbigniew Jędrzejewski-Szmek <zbyszek@in.waw.pl> - 1.0-8
|
|
||||||
- Python 2 binary package renamed to python2-pymilter
|
|
||||||
See https://fedoraproject.org/wiki/FinalizingFedoraSwitchtoPython3
|
|
||||||
|
|
||||||
* Thu Aug 03 2017 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-7
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_27_Binutils_Mass_Rebuild
|
|
||||||
|
|
||||||
* Thu Jul 27 2017 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-6
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_27_Mass_Rebuild
|
|
||||||
>>>>>>> 021796e51e5919812f1c300d1830ef9ed378db2d
|
|
||||||
|
|
||||||
* Sat Feb 11 2017 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-5
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_26_Mass_Rebuild
|
|
||||||
|
|
||||||
* Tue Jul 19 2016 Fedora Release Engineering <rel-eng@lists.fedoraproject.org> - 1.0-4
|
|
||||||
- https://fedoraproject.org/wiki/Changes/Automatic_Provides_for_Python_RPM_Packages
|
|
||||||
|
|
||||||
* Thu Feb 04 2016 Fedora Release Engineering <releng@fedoraproject.org> - 1.0-3
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_24_Mass_Rebuild
|
|
||||||
|
|
||||||
* Thu Jun 18 2015 Fedora Release Engineering <rel-eng@lists.fedoraproject.org> - 1.0-2
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_23_Mass_Rebuild
|
|
||||||
|
|
||||||
* Sat Sep 27 2014 Paul Wouters <pwouters@redhat.com> - 1.0-1
|
|
||||||
- Updated to 1.0
|
|
||||||
- Use tmpfiles and /run
|
|
||||||
|
|
||||||
* Sun Aug 17 2014 Fedora Release Engineering <rel-eng@lists.fedoraproject.org> - 0.9.8-6
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_21_22_Mass_Rebuild
|
|
||||||
|
|
||||||
* Sat Jun 07 2014 Fedora Release Engineering <rel-eng@lists.fedoraproject.org> - 0.9.8-5
|
|
||||||
- Rebuilt for https://fedoraproject.org/wiki/Fedora_21_Mass_Rebuild
|
|
||||||
|
|
||||||
* Fri Jan 10 2014 Paul Wouters <pwouters@redhat.com> - 0.9.8-4
|
|
||||||
- Add COPYING
|
|
||||||
- Fix buildroot macros and dist macro
|
|
||||||
|
|
||||||
* Fri Jan 10 2014 Paul Wouters <pwouters@redhat.com> - 0.9.8-3
|
|
||||||
- rebuilt with proper file permission
|
|
||||||
|
|
||||||
* Tue Jan 07 2014 Paul Wouters <pwouters@redhat.com> - 0.9.8-2
|
|
||||||
- Fixup for fedora release
|
|
||||||
|
|
||||||
* Sat Mar 9 2013 Stuart Gathman <stuart@bmsi.com> 0.9.8-1
|
* Sat Mar 9 2013 Stuart Gathman <stuart@bmsi.com> 0.9.8-1
|
||||||
- Add Milter.test module for unit testing milters.
|
- Add Milter.test module for unit testing milters.
|
||||||
@@ -225,13 +98,13 @@ fi
|
|||||||
- Change untrapped exception message to:
|
- Change untrapped exception message to:
|
||||||
- "pymilter: untrapped exception in milter app"
|
- "pymilter: untrapped exception in milter app"
|
||||||
|
|
||||||
* Sat Feb 25 2012 Stuart Gathman <stuart@bmsi.com> 0.9.7-1
|
* Thu Apr 12 2012 Stuart Gathman <stuart@bmsi.com> 0.9.7-1
|
||||||
- Raise RuntimeError when result != CONTINUE for @noreply and @nocallback
|
- Raise RuntimeError when result != CONTINUE for @noreply and @nocallback
|
||||||
- Remove redundant table in miltermodule
|
- Remove redundant table in miltermodule
|
||||||
- Fix CNAME chain duplicating TXT records in Milter.dns (from pyspf).
|
- Fix CNAME chain duplicating TXT records in Milter.dns (from pyspf).
|
||||||
|
|
||||||
* Sat Feb 25 2012 Stuart Gathman <stuart@bmsi.com> 0.9.6-1
|
* Sat Feb 25 2012 Stuart Gathman <stuart@bmsi.com> 0.9.6-1
|
||||||
- Raise ValueError on unescaped '%%' passed to setreply
|
- Raise ValueError on unescaped '%' passed to setreply
|
||||||
- Grace time at end of Greylist window
|
- Grace time at end of Greylist window
|
||||||
|
|
||||||
* Fri Aug 19 2011 Stuart Gathman <stuart@bmsi.com> 0.9.5-1
|
* Fri Aug 19 2011 Stuart Gathman <stuart@bmsi.com> 0.9.5-1
|
||||||
@@ -240,7 +113,7 @@ fi
|
|||||||
- Fix milter-template.py
|
- Fix milter-template.py
|
||||||
- Tweak Milter.utils.addr2bin and Milter.dynip to handle IP6
|
- Tweak Milter.utils.addr2bin and Milter.dynip to handle IP6
|
||||||
|
|
||||||
* Tue Mar 02 2010 Stuart Gathman <stuart@bmsi.com> 0.9.4-1
|
* Wed Mar 02 2010 Stuart Gathman <stuart@bmsi.com> 0.9.4-1
|
||||||
- Handle IP6 in Milter.utils.iniplist()
|
- Handle IP6 in Milter.utils.iniplist()
|
||||||
- python-2.6
|
- python-2.6
|
||||||
|
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
from __future__ import print_function
|
|
||||||
# A simple milter.
|
# A simple milter.
|
||||||
|
|
||||||
# Author: Stuart D. Gathman <stuart@bmsi.com>
|
# Author: Stuart D. Gathman <stuart@bmsi.com>
|
||||||
@@ -7,10 +7,8 @@ from __future__ import print_function
|
|||||||
|
|
||||||
import sys
|
import sys
|
||||||
import os
|
import os
|
||||||
try:
|
import StringIO
|
||||||
from io import BytesIO
|
import rfc822
|
||||||
except:
|
|
||||||
from StringIO import StringIO as BytesIO
|
|
||||||
import mime
|
import mime
|
||||||
import Milter
|
import Milter
|
||||||
import tempfile
|
import tempfile
|
||||||
@@ -23,9 +21,9 @@ class sampleMilter(Milter.Milter):
|
|||||||
"Milter to replace attachments poisonous to Windows with a WARNING message."
|
"Milter to replace attachments poisonous to Windows with a WARNING message."
|
||||||
|
|
||||||
def log(self,*msg):
|
def log(self,*msg):
|
||||||
print("%s [%d]" % (strftime('%Y%b%d %H:%M:%S'),self.id),end=None)
|
print "%s [%d]" % (strftime('%Y%b%d %H:%M:%S'),self.id),
|
||||||
for i in msg: print(i,end=None)
|
for i in msg: print i,
|
||||||
print()
|
print
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.tempname = None
|
self.tempname = None
|
||||||
@@ -33,25 +31,18 @@ class sampleMilter(Milter.Milter):
|
|||||||
self.fp = None
|
self.fp = None
|
||||||
self.bodysize = 0
|
self.bodysize = 0
|
||||||
self.id = Milter.uniqueID()
|
self.id = Milter.uniqueID()
|
||||||
self.user = None
|
|
||||||
|
|
||||||
# multiple messages can be received on a single connection
|
# multiple messages can be received on a single connection
|
||||||
# envfrom (MAIL FROM in the SMTP protocol) seems to mark the start
|
# envfrom (MAIL FROM in the SMTP protocol) seems to mark the start
|
||||||
# of each message.
|
# of each message.
|
||||||
@Milter.symlist('{auth_authen}')
|
|
||||||
@Milter.noreply
|
@Milter.noreply
|
||||||
def envfrom(self,f,*str):
|
def envfrom(self,f,*str):
|
||||||
"start of MAIL transaction"
|
"start of MAIL transaction"
|
||||||
self.fp = BytesIO()
|
self.log("mail from",f,str)
|
||||||
|
self.fp = StringIO.StringIO()
|
||||||
self.tempname = None
|
self.tempname = None
|
||||||
self.mailfrom = f
|
self.mailfrom = f
|
||||||
self.bodysize = 0
|
self.bodysize = 0
|
||||||
self.user = self.getsymval('{auth_authen}')
|
|
||||||
self.auth_type = self.getsymval('{auth_type}')
|
|
||||||
if self.user:
|
|
||||||
self.log("user",self.user,"sent mail from",f,str)
|
|
||||||
else:
|
|
||||||
self.log("mail from",f,str)
|
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
def envrcpt(self,to,*str):
|
def envrcpt(self,to,*str):
|
||||||
@@ -104,12 +95,12 @@ class sampleMilter(Milter.Milter):
|
|||||||
if lname in ('subject','x-mailer'):
|
if lname in ('subject','x-mailer'):
|
||||||
self.log('%s: %s' % (name,val))
|
self.log('%s: %s' % (name,val))
|
||||||
if self.fp:
|
if self.fp:
|
||||||
self.fp.write(("%s: %s\n" % (name,val)).encode()) # add header to buffer
|
self.fp.write("%s: %s\n" % (name,val)) # add header to buffer
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
def eoh(self):
|
def eoh(self):
|
||||||
if not self.fp: return Milter.TEMPFAIL # not seen by envfrom
|
if not self.fp: return Milter.TEMPFAIL # not seen by envfrom
|
||||||
self.fp.write(b'\n')
|
self.fp.write("\n")
|
||||||
self.fp.seek(0)
|
self.fp.seek(0)
|
||||||
# copy headers to a temp file for scanning the body
|
# copy headers to a temp file for scanning the body
|
||||||
headers = self.fp.getvalue()
|
headers = self.fp.getvalue()
|
||||||
@@ -147,16 +138,19 @@ class sampleMilter(Milter.Milter):
|
|||||||
self.log("Temp file:",self.tempname)
|
self.log("Temp file:",self.tempname)
|
||||||
self.tempname = None # prevent removal of original message copy
|
self.tempname = None # prevent removal of original message copy
|
||||||
# copy defanged message to a temp file
|
# copy defanged message to a temp file
|
||||||
with tempfile.TemporaryFile() as out:
|
out = tempfile.TemporaryFile()
|
||||||
|
try:
|
||||||
msg.dump(out)
|
msg.dump(out)
|
||||||
out.seek(0)
|
out.seek(0)
|
||||||
msg = mime.message_from_file(out)
|
msg = rfc822.Message(out)
|
||||||
fp = BytesIO(msg.as_bytes().split(b'\n\n',1)[1])
|
msg.rewindbody()
|
||||||
while 1:
|
while 1:
|
||||||
buf = fp.read(8192)
|
buf = out.read(8192)
|
||||||
if len(buf) == 0: break
|
if len(buf) == 0: break
|
||||||
self.replacebody(buf) # feed modified message to sendmail
|
self.replacebody(buf) # feed modified message to sendmail
|
||||||
return Milter.ACCEPT # ACCEPT modified message
|
return Milter.ACCEPT # ACCEPT modified message
|
||||||
|
finally:
|
||||||
|
out.close()
|
||||||
return Milter.TEMPFAIL
|
return Milter.TEMPFAIL
|
||||||
|
|
||||||
def close(self):
|
def close(self):
|
||||||
@@ -177,13 +171,13 @@ if __name__ == "__main__":
|
|||||||
socketname = os.getenv("HOME") + "/pythonsock"
|
socketname = os.getenv("HOME") + "/pythonsock"
|
||||||
Milter.factory = sampleMilter
|
Milter.factory = sampleMilter
|
||||||
Milter.set_flags(Milter.CHGBODY + Milter.CHGHDRS + Milter.ADDHDRS)
|
Milter.set_flags(Milter.CHGBODY + Milter.CHGHDRS + Milter.ADDHDRS)
|
||||||
print("""To use this with sendmail, add the following to sendmail.cf:
|
print """To use this with sendmail, add the following to sendmail.cf:
|
||||||
|
|
||||||
O InputMailFilters=pythonfilter
|
O InputMailFilters=pythonfilter
|
||||||
Xpythonfilter, S=local:%s
|
Xpythonfilter, S=local:%s
|
||||||
|
|
||||||
See the sendmail README for libmilter.
|
See the sendmail README for libmilter.
|
||||||
sample milter startup""" % socketname)
|
sample milter startup""" % socketname
|
||||||
sys.stdout.flush()
|
sys.stdout.flush()
|
||||||
Milter.runmilter("pythonfilter",socketname,240)
|
Milter.runmilter("pythonfilter",socketname,240)
|
||||||
print("sample milter shutdown")
|
print "sample milter shutdown"
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
[bdist_rpm]
|
[bdist_rpm]
|
||||||
python=python3
|
python=python2.6
|
||||||
doc_files=README NEWS TODO COPYING CREDITS
|
doc_files=README NEWS TODO COPYING CREDITS
|
||||||
packager=Stuart D. Gathman <stuart@gathman.org>
|
packager=Stuart D. Gathman <stuart@gathman.org>
|
||||||
release=1
|
release=1
|
||||||
|
|||||||
@@ -11,10 +11,9 @@ if sys.version < '2.6.5':
|
|||||||
#libs = ["milter", "smutil"]
|
#libs = ["milter", "smutil"]
|
||||||
libs = ["milter"]
|
libs = ["milter"]
|
||||||
libdirs = ["/usr/lib/libmilter"] # needed for Debian
|
libdirs = ["/usr/lib/libmilter"] # needed for Debian
|
||||||
modules = ["mime"]
|
|
||||||
|
|
||||||
# NOTE: importing Milter to obtain version fails when milter.so not built
|
# NOTE: importing Milter to obtain version fails when milter.so not built
|
||||||
setup(name = "pymilter", version = '1.0.4',
|
setup(name = "pymilter", version = '1.0',
|
||||||
description="Python interface to sendmail milter API",
|
description="Python interface to sendmail milter API",
|
||||||
long_description="""\
|
long_description="""\
|
||||||
This is a python extension module to enable python scripts to
|
This is a python extension module to enable python scripts to
|
||||||
@@ -25,19 +24,17 @@ sending DSNs or doing CBVs.
|
|||||||
author="Jim Niemira",
|
author="Jim Niemira",
|
||||||
author_email="urmane@urmane.org",
|
author_email="urmane@urmane.org",
|
||||||
maintainer="Stuart D. Gathman",
|
maintainer="Stuart D. Gathman",
|
||||||
maintainer_email="stuart@gathman.org",
|
maintainer_email="stuart@bmsi.com",
|
||||||
license="GPL",
|
license="GPL",
|
||||||
url="https://pythonhosted.org/milter/",
|
url="http://www.bmsi.com/python/milter.html",
|
||||||
py_modules=modules,
|
py_modules=["mime"],
|
||||||
packages = ['Milter'],
|
packages = ['Milter'],
|
||||||
ext_modules=[
|
ext_modules=[
|
||||||
Extension("milter", ["miltermodule.c"],
|
Extension("milter", ["miltermodule.c"],
|
||||||
library_dirs=libdirs,
|
library_dirs=libdirs,
|
||||||
libraries=libs,
|
libraries=libs,
|
||||||
# set MAX_ML_REPLY to 1 for sendmail < 8.13
|
# set MAX_ML_REPLY to 1 for sendmail < 8.13
|
||||||
define_macros = [ ('MAX_ML_REPLY',32) ],
|
define_macros = [ ('MAX_ML_REPLY',32) ]
|
||||||
# save lots of debugging time testing rfc2553 compliance
|
|
||||||
extra_compile_args = [ "-Werror=implicit-function-declaration" ]
|
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
keywords = ['sendmail','milter'],
|
keywords = ['sendmail','milter'],
|
||||||
|
|||||||
@@ -0,0 +1,19 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
appname="$1"
|
||||||
|
script="${2:-${appname}}"
|
||||||
|
datadir="/var/lib/milter"
|
||||||
|
logdir="/var/log/milter"
|
||||||
|
piddir="/var/run/milter"
|
||||||
|
libdir="/usr/lib/pymilter"
|
||||||
|
python="python2.4"
|
||||||
|
exec >>${logdir}/${appname}.log 2>&1
|
||||||
|
if test -s ${datadir}/${script}.py; then
|
||||||
|
cd ${datadir} # use version in data dir if it exists for debugging
|
||||||
|
elif test -s ${logdir}/${script}.py; then
|
||||||
|
cd ${logdir} # use version in log dir if it exists for debugging
|
||||||
|
else
|
||||||
|
cd ${libdir}
|
||||||
|
fi
|
||||||
|
|
||||||
|
${python} ${script}.py &
|
||||||
|
echo $! >${piddir}/${appname}.pid
|
||||||
+1
-2
@@ -8,8 +8,7 @@ class GreylistTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.fname = 'test.db'
|
self.fname = 'test.db'
|
||||||
if os.path.isfile(self.fname):
|
os.remove(self.fname)
|
||||||
os.remove(self.fname)
|
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
#os.remove(self.fname)
|
#os.remove(self.fname)
|
||||||
|
|||||||
+51
-79
@@ -26,21 +26,14 @@
|
|||||||
# Revision 1.20 2004/11/20 16:38:17 stuart
|
# Revision 1.20 2004/11/20 16:38:17 stuart
|
||||||
# Add rcs log
|
# Add rcs log
|
||||||
#
|
#
|
||||||
from __future__ import print_function
|
|
||||||
import unittest
|
import unittest
|
||||||
import mime
|
import mime
|
||||||
import socket
|
import socket
|
||||||
try:
|
import StringIO
|
||||||
from StringIO import StringIO
|
|
||||||
except:
|
|
||||||
from io import StringIO
|
|
||||||
import email
|
import email
|
||||||
import sys
|
import sys
|
||||||
import Milter
|
import Milter
|
||||||
try:
|
from email import Errors
|
||||||
from email import Errors as errors
|
|
||||||
except:
|
|
||||||
from email import errors
|
|
||||||
|
|
||||||
samp1_txt1 = """Dear Agent 1
|
samp1_txt1 = """Dear Agent 1
|
||||||
I hope you can read this. Whenever you write label it P.B.S kids.
|
I hope you can read this. Whenever you write label it P.B.S kids.
|
||||||
@@ -53,56 +46,48 @@ class MimeTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
# test mime parameter parsing
|
# test mime parameter parsing
|
||||||
def testParam(self):
|
def testParam(self):
|
||||||
plist = mime._parseparam('; boundary="----=_NextPart_000_4e56_490d_48e3"')
|
plist = mime._parseparam(
|
||||||
plist = [ x for x in plist if x ] # py2 doesn't include empty params
|
'; boundary="----=_NextPart_000_4e56_490d_48e3"')
|
||||||
self.assertEqual(1,len(plist))
|
self.failUnless(len(plist)==1)
|
||||||
self.assertTrue(plist[0] == 'boundary="----=_NextPart_000_4e56_490d_48e3"')
|
self.failUnless(plist[0] == 'boundary="----=_NextPart_000_4e56_490d_48e3"')
|
||||||
plist = mime._parseparam('; name="Jim&amp;Girlz.jpg"')
|
plist = mime._parseparam('; name="Jim&amp;Girlz.jpg"')
|
||||||
plist = [ x for x in plist if x ] # py2 doesn't include empty params
|
self.failUnless(len(plist)==1)
|
||||||
self.assertEqual(1,len(plist))
|
self.failUnless(plist[0] == 'name="Jim&amp;Girlz.jpg"')
|
||||||
self.assertTrue(plist[0] == 'name="Jim&amp;Girlz.jpg"')
|
|
||||||
|
|
||||||
def testParse(self,fname='samp1'):
|
def testParse(self,fname='samp1'):
|
||||||
with open('test/'+fname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+fname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
self.failUnless(msg.ismultipart())
|
||||||
self.assertTrue(msg.ismultipart())
|
|
||||||
parts = msg.get_payload()
|
parts = msg.get_payload()
|
||||||
self.assertTrue(len(parts) == 2)
|
self.failUnless(len(parts) == 2)
|
||||||
txt1 = parts[0].get_payload()
|
txt1 = parts[0].get_payload()
|
||||||
self.assertTrue(txt1.rstrip() == samp1_txt1,txt1)
|
self.failUnless(txt1.rstrip() == samp1_txt1,txt1)
|
||||||
with open('test/missingboundary',"rb") as fp:
|
msg = mime.message_from_file(open('test/missingboundary',"r"))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
# should get no exception as long as we don't try to parse
|
# should get no exception as long as we don't try to parse
|
||||||
# message attachments
|
# message attachments
|
||||||
mime.defang(msg,scan_rfc822=False)
|
mime.defang(msg,scan_rfc822=False)
|
||||||
with open('test/missingboundary.out','wb') as fp:
|
msg.dump(open('test/missingboundary.out','w'))
|
||||||
msg.dump(fp)
|
msg = mime.message_from_file(open('test/missingboundary',"r"))
|
||||||
with open('test/missingboundary',"rb") as fp:
|
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
try:
|
try:
|
||||||
mime.defang(msg)
|
mime.defang(msg)
|
||||||
# python 2.4 doesn't get exceptions on missing boundaries, and
|
# python 2.4 doesn't get exceptions on missing boundaries, and
|
||||||
# if message is modified, output is readable by mail clients
|
# if message is modified, output is readable by mail clients
|
||||||
if sys.hexversion < 0x02040000:
|
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:
|
except Errors.BoundaryError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def testDefang(self,vname='virus1',part=1,
|
def testDefang(self,vname='virus1',part=1,
|
||||||
fname='LOVE-LETTER-FOR-YOU.TXT.vbs'):
|
fname='LOVE-LETTER-FOR-YOU.TXT.vbs'):
|
||||||
with open('test/'+vname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+vname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg,scan_zip=True)
|
mime.defang(msg,scan_zip=True)
|
||||||
self.assertTrue(msg.ismodified(),"virus not removed")
|
self.failUnless(msg.ismodified(),"virus not removed")
|
||||||
oname = vname + '.out'
|
oname = vname + '.out'
|
||||||
with open('test/'+oname,"wb") as fp:
|
msg.dump(open('test/'+oname,"w"))
|
||||||
msg.dump(fp)
|
msg = mime.message_from_file(open('test/'+oname,"r"))
|
||||||
with open('test/'+oname,"rb") as fp:
|
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
txt2 = msg.get_payload()
|
txt2 = msg.get_payload()
|
||||||
if type(txt2) == list:
|
if type(txt2) == list:
|
||||||
txt2 = txt2[part].get_payload()
|
txt2 = txt2[part].get_payload()
|
||||||
self.assertTrue(
|
self.failUnless(
|
||||||
txt2.rstrip()+'\n' == mime.virus_msg % (fname,hostname,None),txt2)
|
txt2.rstrip()+'\n' == mime.virus_msg % (fname,hostname,None),txt2)
|
||||||
|
|
||||||
def testDefang3(self):
|
def testDefang3(self):
|
||||||
@@ -118,60 +103,51 @@ class MimeTestCase(unittest.TestCase):
|
|||||||
|
|
||||||
# virus6 has no parts - the virus is directly inline
|
# virus6 has no parts - the virus is directly inline
|
||||||
def testDefang6(self,vname="virus6",fname='FAX20.exe'):
|
def testDefang6(self,vname="virus6",fname='FAX20.exe'):
|
||||||
with open('test/'+vname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+vname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg)
|
mime.defang(msg)
|
||||||
oname = vname + '.out'
|
oname = vname + '.out'
|
||||||
with open('test/'+oname,"wb") as fp:
|
msg.dump(open('test/'+oname,"w"))
|
||||||
msg.dump(fp)
|
msg = mime.message_from_file(open('test/'+oname,"r"))
|
||||||
with open('test/'+oname,"rb") as fp:
|
self.failIf(msg.ismultipart())
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
self.assertFalse(msg.ismultipart())
|
|
||||||
txt2 = msg.get_payload()
|
txt2 = msg.get_payload()
|
||||||
self.assertTrue(txt2 == mime.virus_msg % \
|
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
|
# 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
|
# by email package in python-2.2.2, but not by mime.MimeMessage or 2.2.1
|
||||||
def testDefang7(self,vname="honey",fname='story[1].scr'):
|
def testDefang7(self,vname="honey",fname='story[1].scr'):
|
||||||
with open('test/'+vname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+vname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg)
|
mime.defang(msg)
|
||||||
oname = vname + '.out'
|
oname = vname + '.out'
|
||||||
with open('test/'+oname,"wb") as fp:
|
msg.dump(open('test/'+oname,"w"))
|
||||||
msg.dump(fp)
|
msg = mime.message_from_file(open('test/'+oname,"r"))
|
||||||
with open('test/'+oname,"rb") as fp:
|
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
parts = msg.get_payload()
|
parts = msg.get_payload()
|
||||||
txt2 = parts[1].get_payload()
|
txt2 = parts[1].get_payload()
|
||||||
txt3 = parts[2].get_payload()
|
txt3 = parts[2].get_payload()
|
||||||
self.assertTrue(txt2.rstrip()+'\n' == mime.virus_msg % \
|
self.failUnless(txt2.rstrip()+'\n' == mime.virus_msg % \
|
||||||
(fname,hostname,None),txt2)
|
(fname,hostname,None),txt2)
|
||||||
if txt3 != '':
|
if txt3 != '':
|
||||||
self.assertTrue(txt3.rstrip()+'\n' == mime.virus_msg % \
|
self.failUnless(txt3.rstrip()+'\n' == mime.virus_msg % \
|
||||||
('story[1].asp',hostname,None),txt3)
|
('story[1].asp',hostname,None),txt3)
|
||||||
|
|
||||||
def testParse2(self,fname="spam7"):
|
def testParse2(self,fname="spam7"):
|
||||||
with open('test/'+fname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+fname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
self.failUnless(msg.ismultipart())
|
||||||
self.assertTrue(msg.ismultipart())
|
|
||||||
parts = msg.get_payload()
|
parts = msg.get_payload()
|
||||||
self.assertTrue(len(parts) == 2)
|
self.failUnless(len(parts) == 2)
|
||||||
name = parts[1].getname()
|
name = parts[1].getname()
|
||||||
self.assertTrue(name == "Jim&amp;Girlz.jpg","name=%s"%name)
|
self.failUnless(name == "Jim&amp;Girlz.jpg","name=%s"%name)
|
||||||
|
|
||||||
def testZip(self,vname="zip1",fname='zip.zip'):
|
def testZip(self,vname="zip1",fname='zip.zip'):
|
||||||
self.testDefang(vname,1,'zip.zip')
|
self.testDefang(vname,1,'zip.zip')
|
||||||
# test scan_zip flag
|
# test scan_zip flag
|
||||||
with open('test/'+vname,"rb") as fp:
|
msg = mime.message_from_file(open('test/'+vname,"r"))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg,scan_zip=False)
|
mime.defang(msg,scan_zip=False)
|
||||||
self.assertFalse(msg.ismodified())
|
self.failIf(msg.ismodified())
|
||||||
# test ignoring empty zip (often found in DSNs)
|
# test ignoring empty zip (often found in DSNs)
|
||||||
with open('test/zip2','rb') as fp:
|
msg = mime.message_from_file(open('test/zip2','r'))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg,scan_zip=True)
|
mime.defang(msg,scan_zip=True)
|
||||||
self.assertFalse(msg.ismodified())
|
self.failIf(msg.ismodified())
|
||||||
# test corrupt zip (often an EXE named as a ZIP)
|
# test corrupt zip (often an EXE named as a ZIP)
|
||||||
self.testDefang('zip3',1,'zip.zip')
|
self.testDefang('zip3',1,'zip.zip')
|
||||||
# test zip within zip
|
# test zip within zip
|
||||||
@@ -188,24 +164,22 @@ class MimeTestCase(unittest.TestCase):
|
|||||||
mime.check_html(msg)
|
mime.check_html(msg)
|
||||||
# don't let a tricky virus slip one past us
|
# don't let a tricky virus slip one past us
|
||||||
msg = msg.get_submsg()
|
msg = msg.get_submsg()
|
||||||
if isinstance(msg,email.message.Message):
|
if isinstance(msg,email.Message.Message):
|
||||||
return mime.check_attachments(msg,self._chk_attach)
|
return mime.check_attachments(msg,self._chk_attach)
|
||||||
return Milter.CONTINUE
|
return Milter.CONTINUE
|
||||||
|
|
||||||
def testCheckAttach(self,fname="test1"):
|
def testCheckAttach(self,fname="test1"):
|
||||||
# test1 contains a very long filename
|
# test1 contains a very long filename
|
||||||
with open('test/'+fname,'rb') as fp:
|
msg = mime.message_from_file(open('test/'+fname,'r'))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
mime.defang(msg,scan_zip=True)
|
mime.defang(msg,scan_zip=True)
|
||||||
self.assertFalse(msg.ismodified())
|
self.failIf(msg.ismodified())
|
||||||
with open('test/test2','rb') as fp:
|
msg = mime.message_from_file(open('test/test2','r'))
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
rc = mime.check_attachments(msg,self._chk_attach)
|
rc = mime.check_attachments(msg,self._chk_attach)
|
||||||
self.assertEqual(self.filename,"7501'S FOR TWO GOLDEN SOURCES SHIPMENTS FOR TAX & DUTY PURPOSES ONLY.PDF")
|
self.assertEquals(self.filename,"7501'S FOR TWO GOLDEN SOURCES SHIPMENTS FOR TAX & DUTY PURPOSES ONLY.PDF")
|
||||||
self.assertEqual(rc,Milter.CONTINUE)
|
self.assertEquals(rc,Milter.CONTINUE)
|
||||||
|
|
||||||
def testHTML(self,fname=""):
|
def testHTML(self,fname=""):
|
||||||
result = StringIO()
|
result = StringIO.StringIO()
|
||||||
filter = mime.HTMLScriptFilter(result)
|
filter = mime.HTMLScriptFilter(result)
|
||||||
msg = """<! Illegal declaration used as comment>
|
msg = """<! Illegal declaration used as comment>
|
||||||
<![if conditional]> Optional SGML <![endif]>
|
<![if conditional]> Optional SGML <![endif]>
|
||||||
@@ -214,10 +188,8 @@ class MimeTestCase(unittest.TestCase):
|
|||||||
script = "<script lang=javascript> Dangerous script </script>"
|
script = "<script lang=javascript> Dangerous script </script>"
|
||||||
filter.feed(msg + script)
|
filter.feed(msg + script)
|
||||||
filter.close()
|
filter.close()
|
||||||
#print(result.getvalue())
|
#print result.getvalue()
|
||||||
#print('---')
|
self.failUnless(result.getvalue() == msg + filter.msg)
|
||||||
#print(msg + filter.msg)
|
|
||||||
self.assertTrue(result.getvalue() == msg + filter.msg)
|
|
||||||
|
|
||||||
def suite(): return unittest.makeSuite(MimeTestCase,'test')
|
def suite(): return unittest.makeSuite(MimeTestCase,'test')
|
||||||
|
|
||||||
@@ -226,7 +198,7 @@ if __name__ == '__main__':
|
|||||||
unittest.main()
|
unittest.main()
|
||||||
else:
|
else:
|
||||||
for fname in sys.argv[1:]:
|
for fname in sys.argv[1:]:
|
||||||
with open(fname,'rb') as fp:
|
fp = open(fname,'r')
|
||||||
msg = mime.message_from_file(fp)
|
msg = mime.message_from_file(fp)
|
||||||
mime.defang(msg,scan_zip=True)
|
mime.defang(msg,scan_zip=True)
|
||||||
print(msg.as_string())
|
print msg.as_string()
|
||||||
|
|||||||
+20
-50
@@ -2,8 +2,9 @@ import unittest
|
|||||||
import Milter
|
import Milter
|
||||||
import sample
|
import sample
|
||||||
import mime
|
import mime
|
||||||
|
import rfc822
|
||||||
|
import StringIO
|
||||||
from Milter.test import TestBase
|
from Milter.test import TestBase
|
||||||
from Milter.testctx import TestCtx
|
|
||||||
|
|
||||||
class TestMilter(TestBase,sample.sampleMilter):
|
class TestMilter(TestBase,sample.sampleMilter):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -12,47 +13,16 @@ class TestMilter(TestBase,sample.sampleMilter):
|
|||||||
|
|
||||||
class BMSMilterTestCase(unittest.TestCase):
|
class BMSMilterTestCase(unittest.TestCase):
|
||||||
|
|
||||||
def testCtx(self,fname='virus1'):
|
|
||||||
ctx = TestCtx()
|
|
||||||
Milter.factory = sample.sampleMilter
|
|
||||||
ctx._setsymval('{auth_authen}','batman')
|
|
||||||
ctx._setsymval('{auth_type}','batcomputer')
|
|
||||||
ctx._setsymval('j','mailhost')
|
|
||||||
rc = ctx._connect()
|
|
||||||
self.assertTrue(rc == Milter.CONTINUE)
|
|
||||||
rc = ctx._feedMsg(fname)
|
|
||||||
milter = ctx.getpriv()
|
|
||||||
# self.assertTrue(milter.user == 'batman',"getsymval failed: "+
|
|
||||||
# "%s != %s"%(milter.user,'batman'))
|
|
||||||
self.assertEquals(milter.user,'batman')
|
|
||||||
self.assertTrue(milter.auth_type != 'batcomputer',"setsymlist failed")
|
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
|
||||||
self.assertTrue(ctx._bodyreplaced,"Message body not replaced")
|
|
||||||
fp = ctx._body
|
|
||||||
open('test/'+fname+".tstout","wb").write(fp.getvalue())
|
|
||||||
#self.assertTrue(fp.getvalue() == open("test/virus1.out","r").read())
|
|
||||||
fp.seek(0)
|
|
||||||
msg = mime.message_from_file(fp)
|
|
||||||
s = msg.get_payload(1).get_payload()
|
|
||||||
milter.log(s)
|
|
||||||
ctx._close()
|
|
||||||
|
|
||||||
def testDefang(self,fname='virus1'):
|
def testDefang(self,fname='virus1'):
|
||||||
milter = TestMilter()
|
milter = TestMilter()
|
||||||
milter.setsymval('{auth_authen}','batman')
|
|
||||||
milter.setsymval('{auth_type}','batcomputer')
|
|
||||||
milter.setsymval('j','mailhost')
|
|
||||||
rc = milter.connect()
|
rc = milter.connect()
|
||||||
self.assertTrue(rc == Milter.CONTINUE)
|
self.failUnless(rc == Milter.CONTINUE)
|
||||||
rc = milter.feedMsg(fname)
|
rc = milter.feedMsg(fname)
|
||||||
self.assertTrue(milter.user == 'batman',"getsymval failed")
|
self.failUnless(rc == Milter.ACCEPT)
|
||||||
# setsymlist not working in TestBase
|
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||||
#self.assertTrue(milter.auth_type != 'batcomputer',"setsymlist failed")
|
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
|
||||||
self.assertTrue(milter._bodyreplaced,"Message body not replaced")
|
|
||||||
fp = milter._body
|
fp = milter._body
|
||||||
open('test/'+fname+".tstout","wb").write(fp.getvalue())
|
open('test/'+fname+".tstout","w").write(fp.getvalue())
|
||||||
#self.assertTrue(fp.getvalue() == open("test/virus1.out","r").read())
|
#self.failUnless(fp.getvalue() == open("test/virus1.out","r").read())
|
||||||
fp.seek(0)
|
fp.seek(0)
|
||||||
msg = mime.message_from_file(fp)
|
msg = mime.message_from_file(fp)
|
||||||
s = msg.get_payload(1).get_payload()
|
s = msg.get_payload(1).get_payload()
|
||||||
@@ -63,30 +33,30 @@ class BMSMilterTestCase(unittest.TestCase):
|
|||||||
milter = TestMilter()
|
milter = TestMilter()
|
||||||
milter.connect('somehost')
|
milter.connect('somehost')
|
||||||
rc = milter.feedMsg(fname)
|
rc = milter.feedMsg(fname)
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
self.failUnless(rc == Milter.ACCEPT)
|
||||||
self.assertFalse(milter._bodyreplaced,"Milter needlessly replaced body.")
|
self.failIf(milter._bodyreplaced,"Milter needlessly replaced body.")
|
||||||
fp = milter._body
|
fp = milter._body
|
||||||
open('test/'+fname+".tstout","wb").write(fp.getvalue())
|
open('test/'+fname+".tstout","w").write(fp.getvalue())
|
||||||
milter.close()
|
milter.close()
|
||||||
|
|
||||||
def testDefang2(self):
|
def testDefang2(self):
|
||||||
milter = TestMilter()
|
milter = TestMilter()
|
||||||
milter.connect('somehost')
|
milter.connect('somehost')
|
||||||
rc = milter.feedMsg('samp1')
|
rc = milter.feedMsg('samp1')
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
self.failUnless(rc == Milter.ACCEPT)
|
||||||
self.assertFalse(milter._bodyreplaced,"Milter needlessly replaced body.")
|
self.failIf(milter._bodyreplaced,"Milter needlessly replaced body.")
|
||||||
rc = milter.feedMsg("virus3")
|
rc = milter.feedMsg("virus3")
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
self.failUnless(rc == Milter.ACCEPT)
|
||||||
self.assertTrue(milter._bodyreplaced,"Message body not replaced")
|
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||||
fp = milter._body
|
fp = milter._body
|
||||||
open("test/virus3.tstout","wb").write(fp.getvalue())
|
open("test/virus3.tstout","w").write(fp.getvalue())
|
||||||
#self.assertTrue(fp.getvalue() == open("test/virus3.out","r").read())
|
#self.failUnless(fp.getvalue() == open("test/virus3.out","r").read())
|
||||||
rc = milter.feedMsg("virus6")
|
rc = milter.feedMsg("virus6")
|
||||||
self.assertTrue(rc == Milter.ACCEPT)
|
self.failUnless(rc == Milter.ACCEPT)
|
||||||
self.assertTrue(milter._bodyreplaced,"Message body not replaced")
|
self.failUnless(milter._bodyreplaced,"Message body not replaced")
|
||||||
self.assertTrue(milter._headerschanged,"Message headers not adjusted")
|
self.failUnless(milter._headerschanged,"Message headers not adjusted")
|
||||||
fp = milter._body
|
fp = milter._body
|
||||||
open("test/virus6.tstout","wb").write(fp.getvalue())
|
open("test/virus6.tstout","w").write(fp.getvalue())
|
||||||
milter.close()
|
milter.close()
|
||||||
|
|
||||||
def suite(): return unittest.makeSuite(BMSMilterTestCase,'test')
|
def suite(): return unittest.makeSuite(BMSMilterTestCase,'test')
|
||||||
|
|||||||
+11
-24
@@ -1,11 +1,9 @@
|
|||||||
from __future__ import print_function
|
|
||||||
import unittest
|
import unittest
|
||||||
import doctest
|
import doctest
|
||||||
import os
|
import os
|
||||||
import Milter.utils
|
import Milter.utils
|
||||||
from Milter.cache import AddrCache
|
from Milter.cache import AddrCache
|
||||||
from Milter.dynip import is_dynip
|
from Milter.dynip import is_dynip
|
||||||
from Milter.pyip6 import inet_ntop
|
|
||||||
|
|
||||||
class AddrCacheTestCase(unittest.TestCase):
|
class AddrCacheTestCase(unittest.TestCase):
|
||||||
|
|
||||||
@@ -13,48 +11,37 @@ class AddrCacheTestCase(unittest.TestCase):
|
|||||||
self.fname = 'test.dat'
|
self.fname = 'test.dat'
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
if os.path.exists(self.fname):
|
os.remove(self.fname)
|
||||||
os.remove(self.fname)
|
|
||||||
|
|
||||||
def testAdd(self):
|
def testAdd(self):
|
||||||
cache = AddrCache(fname=self.fname)
|
cache = AddrCache(fname=self.fname)
|
||||||
cache['foo@bar.com'] = None
|
cache['foo@bar.com'] = None
|
||||||
cache.addperm('baz@bar.com')
|
cache.addperm('baz@bar.com')
|
||||||
cache['temp@bar.com'] = 'testing'
|
cache['temp@bar.com'] = 'testing'
|
||||||
self.assertTrue(cache.has_key('foo@bar.com'))
|
self.failUnless(cache.has_key('foo@bar.com'))
|
||||||
self.assertTrue(not cache.has_key('hello@bar.com'))
|
self.failUnless(not cache.has_key('hello@bar.com'))
|
||||||
self.assertTrue('baz@bar.com' in cache)
|
self.failUnless('baz@bar.com' in cache)
|
||||||
self.assertEquals(cache['temp@bar.com'],'testing')
|
self.assertEquals(cache['temp@bar.com'],'testing')
|
||||||
s = open(self.fname).readlines()
|
s = open(self.fname).readlines()
|
||||||
self.assertTrue(len(s) == 2)
|
self.failUnless(len(s) == 2)
|
||||||
self.assertTrue(s[0].startswith('foo@bar.com '))
|
self.failUnless(s[0].startswith('foo@bar.com '))
|
||||||
self.assertEquals(s[1].strip(),'baz@bar.com')
|
self.assertEquals(s[1].strip(),'baz@bar.com')
|
||||||
# check that new result overrides old
|
# check that new result overrides old
|
||||||
cache['temp@bar.com'] = None
|
cache['temp@bar.com'] = None
|
||||||
self.assertTrue(not cache['temp@bar.com'])
|
self.failUnless(not cache['temp@bar.com'])
|
||||||
|
|
||||||
def testDomain(self):
|
def testDomain(self):
|
||||||
with open(self.fname,'w') as fp:
|
fp = open(self.fname,'w')
|
||||||
print('spammer.com',file=fp)
|
print >>fp,'spammer.com'
|
||||||
|
fp.close()
|
||||||
cache = AddrCache(fname=self.fname)
|
cache = AddrCache(fname=self.fname)
|
||||||
cache.load(self.fname,30)
|
cache.load(self.fname,30)
|
||||||
self.assertTrue('spammer.com' in cache)
|
self.failUnless('spammer.com' in cache)
|
||||||
|
|
||||||
def testParseHeader(self):
|
|
||||||
s='=?UTF-8?B?TGFzdCBGZXcgQ29sZHBsYXkgQWxidW0gQXJ0d29ya3MgQXZhaWxhYmxlAA?='
|
|
||||||
h = Milter.utils.parse_header(s)
|
|
||||||
self.assertEqual(h,b'Last Few Coldplay Album Artworks Available\x00')
|
|
||||||
|
|
||||||
@unittest.expectedFailure
|
|
||||||
def testParseAddress(self):
|
|
||||||
s = Milter.utils.parseaddr('a(WRONG)@b')
|
|
||||||
self.assertEqual(s,('WRONG', 'a@b'))
|
|
||||||
|
|
||||||
def suite():
|
def suite():
|
||||||
s = unittest.makeSuite(AddrCacheTestCase,'test')
|
s = unittest.makeSuite(AddrCacheTestCase,'test')
|
||||||
s.addTest(doctest.DocTestSuite(Milter.utils))
|
s.addTest(doctest.DocTestSuite(Milter.utils))
|
||||||
s.addTest(doctest.DocTestSuite(Milter.dynip))
|
s.addTest(doctest.DocTestSuite(Milter.dynip))
|
||||||
s.addTest(doctest.DocTestSuite(Milter.pyip6))
|
|
||||||
return s
|
return s
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|||||||
Reference in New Issue
Block a user