Changeset 149 for branches/sql-cleanup/cgi-bin/IPDB.pm

Timestamp:

02/03/05 14:23:34 (19 years ago)

Author:

Kris Deugau

Message:

/branches/sql-cleanup

Fixed a bunch of general problems with pool allocations, and
rearranged some code, alloctypes, and the poolips table (ugh)
to more cleanly support different types of IP pool.

File:

• branches/sql-cleanup/cgi-bin/IPDB.pm (modified) (5 diffs)

branches/sql-cleanup/cgi-bin/IPDB.pm

@@ -251 +251 @@
 
           # And initialize the pool, if necessary
+          # PPPoE pools (currently dialup, DSL, and WiFi) get all IPs made available
+          # "DHCP" or "real-subnet" pools have the net, gw, and bcast IPs removed.
           if ($type =~ /^.p$/) {
-            $msg = "Could not initialize IPs in new $disp_alloctypes{$type} pool $cidr";
-            initPool($dbh,$cidr,$type,$city,($type eq 'dp' ? "all" : "normal"));
+            $msg = "Could not initialize IPs in new $disp_alloctypes{$type} $cidr";
+            my ($code,$rmsg) = initPool($dbh,$cidr,$type,$city,"all");
+            die $rmsg if $code eq 'FAIL';
+          } elsif ($type =~ /^.d$/) {
+            $msg = "Could not initialize IPs in new $disp_alloctypes{$type} $cidr";
+            my ($code,$rmsg) = initPool($dbh,$cidr,$type,$city,"normal");
+            die $rmsg if $code eq 'FAIL';
           }
 
@@ -261 +268 @@
       }; # end of eval
       if ($@) {
-        $msg = $@;
+        $msg .= ": ".$@;
         eval { $dbh->rollback; };
-        return ('FAIL',$@);
+        return ('FAIL',$msg);
       } else {
         return ('OK',"OK");
@@ -367 +374 @@
   my $pool = new NetAddr::IP $_[1];
 
-  my ($pooltype) = ($type =~ /^(.)p$/);
+##fixme Need to just replace 2nd char of type with i rather than capturing 1st char of type
+  $type =~ s/[pd]$/i/;
   my $sth;
-
-  # have to insert all pool IPs into poolips table as "unallocated".
-  $sth = $dbh->prepare("insert into poolips (pool,ip,custid,city,ptype)".
-        " values ('$pool', ?, '6750400', '$city', '$pooltype')");
-  my @poolip_list = $pool->hostenum;
-  if ($class eq 'all') { # (DSL-ish block - *all* IPs available
-    $sth->execute($pool->addr);
-    for (my $i=0; $i<=$#poolip_list; $i++) {
-      $sth->execute($poolip_list[$i]->addr);
-    }
-    $pool--;
-    $sth->execute($pool->addr);
-  } else { # (real netblock)
-    for (my $i=1; $i<=$#poolip_list; $i++) {
-      $sth->execute($poolip_list[$i]->addr);
-    }
+  my $msg;
+
+  # Trap errors so we can pass them back to the caller.  Even if the
+  # caller is only ever supposed to be local, and therefore already
+  # trapping errors.  >:(
+  local $dbh->{AutoCommit} = 0; # These need to be local so we don't
+  local $dbh->{RaiseError} = 1; # step on our toes by accident.
+
+  eval {
+    # have to insert all pool IPs into poolips table as "unallocated".
+    $sth = $dbh->prepare("insert into poolips (pool,ip,custid,city,type)".
+        " values ('$pool', ?, '6750400', '$city', '$type')");
+    my @poolip_list = $pool->hostenum;
+    if ($class eq 'all') { # (DSL-ish block - *all* IPs available
+      $sth->execute($pool->addr);
+      for (my $i=0; $i<=$#poolip_list; $i++) {
+        $sth->execute($poolip_list[$i]->addr);
+      }
+      $pool--;
+      $sth->execute($pool->addr);
+    } else { # (real netblock)
+      for (my $i=1; $i<=$#poolip_list; $i++) {
+        $sth->execute($poolip_list[$i]->addr);
+      }
+    }
+  };
+  if ($@) {
+    $msg = "'".$sth->errstr."'";
+    eval { $dbh->rollback; };
+    return ('FAIL',$msg);
+  } else {
+    return ('OK',"OK");
   }
 } # end initPool()
@@ -412 +436 @@
 
     eval {
-      $msg = "Unable to deallocate $type $cidr";
+      $msg = "Unable to deallocate $disp_alloctypes{$type} $cidr";
       $sth = $dbh->prepare("update poolips set custid='6750400',available='y',".
         "city=(select city from allocations where cidr >>= '$cidr'),".
@@ -469 +493 @@
         $sth->execute;
         # Special case - delete pool IPs
-        if ($type =~ /^.p$/) {
+        if ($type =~ /^.[pd]$/) {
           # We have to delete the IPs from the pool listing.
           $sth = $dbh->prepare("delete from poolips where pool='$cidr'");