From 19ac561822125d943856b1527beee7f3631615e0 Mon Sep 17 00:00:00 2001
From: Scott Kitterman <scott@kitterman.com>
Date: Wed, 14 Feb 2018 01:01:26 -0500
Subject: [PATCH] Sign/Verify based on Mode config option

---
 dkimpy_milter/__init__.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/dkimpy_milter/__init__.py b/dkimpy_milter/__init__.py
index 100ff64..e283dd4 100644
--- a/dkimpy_milter/__init__.py
+++ b/dkimpy_milter/__init__.py
@@ -54,7 +54,7 @@ class dkimMilter(Milter.Base):
     self.mailfrom = None
     self.id = Milter.uniqueID()
     # we don't want config used to change during a connection
-    self.conf = config
+    self.conf = milterconfig
     self.fp = None
 
   @Milter.noreply
@@ -65,8 +65,8 @@ class dkimMilter(Milter.Base):
     self.receiver = self.getsymval('j').strip()
     if hostaddr and len(hostaddr) > 0:
       ipaddr = hostaddr[0]
-      if iniplist(ipaddr,self.conf.internal_connect):
-        self.internal_connection = True
+      """if iniplist(ipaddr,self.conf.internal_connect): FIXME
+        self.internal_connection = True"""
     else: ipaddr = ''
     self.connectip = ipaddr
     if self.internal_connection:
@@ -150,11 +150,11 @@ class dkimMilter(Milter.Base):
         self.log('REMOVE: ',val)
     # Check or sign DKIM
     self.fp.seek(0)
-    if self.internal_connection:
+    if self.internal_connection or conf.get('Mode') == 's' or conf.get('Mode') == 'sv':
       txt = self.fp.read()
       self.sign_dkim(txt)
       result = None
-    elif self.has_dkim:
+    if self.has_dkim and (conf.get('Mode') == 'v' or conf.get('Mode') == 'sv'):
       txt = self.fp.read()
       if self.check_dkim(txt):
         result = 'pass'