From f674299b721d26ea97fc36e7e6818a84f3d311d3 Mon Sep 17 00:00:00 2001
From: Guilhem Moulin <guilhem.moulin@fripost.org>
Date: Tue, 22 Jan 2013 02:58:23 +0100
Subject: Better check for domain existence.

---
 lib/Fripost/Panel/Interface.pm | 4 ++--
 lib/Fripost/Schema/Domain.pm   | 9 ++++++---
 lib/Fripost/Schema/Local.pm    | 2 +-
 3 files changed, 9 insertions(+), 6 deletions(-)

(limited to 'lib')

diff --git a/lib/Fripost/Panel/Interface.pm b/lib/Fripost/Panel/Interface.pm
index cc7545a..aee0df4 100644
--- a/lib/Fripost/Panel/Interface.pm
+++ b/lib/Fripost/Panel/Interface.pm
@@ -225,7 +225,7 @@ sub EditLocal : Runmode {
 
     # Search for *the* matching user, alias or list.
     my ($d,$l) = ($self->split_path)[1,2];
-    $fp->domain->get ($d, -die => 404, -attrs => []);
+    $fp->domain->get ($d, -die => 404, -assert_exist => 1);
     my %local = $fp->local->get ($l.'@'.$d, -die => 404,
                                             -concat => "\x{0D}\x{0A}" );
     die "Unknown type" unless grep { $local{type} eq $_ }
@@ -488,7 +488,7 @@ sub AddLocal : Runmode {
 
         unless ($error) {
             my $fp = Fripost::Schema::->SASLauth( $self->authen->username, %CFG );
-            $fp->domain->get ($d, -die => 404, -attrs => []);
+            $fp->domain->get ($d, -die => 404, -assert_exist => 1);
             $error = $fp->$t->add( \%entry, -concat => "(\n|\x{0D}\x{0A})", %rest);
             $fp->done;
             return $self->redirect('./') unless $error;
diff --git a/lib/Fripost/Schema/Domain.pm b/lib/Fripost/Schema/Domain.pm
index 56e9737..8c3586f 100644
--- a/lib/Fripost/Schema/Domain.pm
+++ b/lib/Fripost/Schema/Domain.pm
@@ -17,7 +17,7 @@ use warnings;
 use utf8;
 
 use parent 'Fripost::Schema';
-use Net::LDAP qw/LDAP_SUCCESS LDAP_COMPARE_TRUE/;
+use Net::LDAP qw/LDAP_SUCCESS LDAP_COMPARE_TRUE LDAP_NO_SUCH_OBJECT/;
 use Fripost::Schema::Util qw/concat get_perms explode must_attrs
                              email_valid canonical_dn/;
 use Net::IDN::Encode qw/domain_to_ascii domain_to_unicode
@@ -113,6 +113,8 @@ sub get {
                          fripostOwner
                          fripostPostmaster/ ];
 
+    $attrs = [ '1.1' ] if $options{'-assert_exist'};
+
     my $domains = $self->ldap->search(
                       base => canonical_dn({fvd => $d}, @{$self->suffix}),
                       scope => 'base',
@@ -133,6 +135,7 @@ sub get {
         die $options{'-die'}."\n" if defined $options{'-die'};
         die "No such such domain: ‘$d’\n";
     }
+    return if $options{'-assert_exist'};
 
     return ( domain => domain_to_unicode($domain->get_value('fvd'))
            , isactive => $domain->get_value('fripostIsStatusActive') eq 'TRUE'
@@ -237,11 +240,11 @@ sub add {
                        scope => 'base',
                        deref => 'never',
                        filter => 'objectClass=FripostVirtualDomain',
-                       attrs => [] );
+                       attrs => [ '1.1' ] );
         if ($mesg->code == LDAP_SUCCESS) {
             die "Domain ‘".$domain."’ already exists.\n";
         }
-        elsif ($mesg->code != 32) {
+        elsif ($mesg->code != LDAP_NO_SUCH_OBJECT) {
             die $mesg->error."\n";
         }
 
diff --git a/lib/Fripost/Schema/Local.pm b/lib/Fripost/Schema/Local.pm
index 51cac34..7a1ae22 100644
--- a/lib/Fripost/Schema/Local.pm
+++ b/lib/Fripost/Schema/Local.pm
@@ -144,7 +144,7 @@ sub exists {
                                       , scope => 'base'
                                       , deref => 'never'
                                       , filter => 'objectClass=*'
-                                      , attrs => []
+                                      , attrs => [ '1.1' ]
         );
         return 1 unless $mesg->code; # 0  Success
         unless ($mesg->code == 32) { # 32 No such object
-- 
cgit v1.2.3