[PATCH] unsubscribe.milter

[Eric Wong <e@80x24.org>]

---
 examples/unsubscribe.milter | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/examples/unsubscribe.milter b/examples/unsubscribe.milter
index 8bad8bd5..fc76cc54 100644
--- a/examples/unsubscribe.milter
+++ b/examples/unsubscribe.milter
@@ -33,8 +33,8 @@ if (open my $fh, '<', $allow_domains) {
 	local $/ = "\n";
 	chomp(my @l = <$fh>);
 	die "close: $!" unless eof($fh) && close($fh);
-	my %ad = map { lc($_) => 1 } @l;
-	$ALLOW_DOMAINS = \%ad;
+	my %l = map { lc($_) => 1 } @l;
+	$ALLOW_DOMAINS = \%l;
 } else {
 	warn <<EOM;
 W: open $allow_domains: $! (all domains allowed)
@@ -43,6 +43,10 @@ W: this is probably not what you want.
 EOM
 }
 
+# is a config file necessary?
+my @ALLOW_ADDR = (qr/\A::1\z/, qr/\A127\./);
+my $ALLOW_ADDR = join('|', @ALLOW_ADDR);
+
 my %cbs;
 $cbs{connect} = sub {
 	my ($ctx) = @_;
@@ -104,7 +108,12 @@ $cbs{eom} = sub {
 	eval {
 		my $priv = $ctx->getpriv;
 		$ctx->setpriv({ header => {}, envrcpt => {} });
+
 		my @rcpt = keys %{$priv->{envrcpt}};
+		if ($ALLOW_ADDR) {
+			my $x = $ctx->getsymval('{daemon_addr}');
+			return SMFIS_CONTINUE if $x && $x !~ /$ALLOW_ADDR/;
+		}
 
 		# one recipient, one unique HTTP(S) URL
 		return SMFIS_CONTINUE if @rcpt != 1;