source: trunk/cgi-bin/IPDB.pm@ 66

# ipdb/cgi-bin/IPDB.pm
# Contains functions for IPDB - database access, subnet mangling, block allocation, etc
###
# SVN revision info
# $Date$
# SVN revision $Rev$
# Last update by $Author$
###
# Copyright (C) 2004 - Kris Deugau

package IPDB;

use strict;
use warnings;
use Exporter;
use Net::SMTP;
use vars qw($VERSION @ISA @EXPORT @EXPORT_OK %EXPORT_TAGS);

$VERSION        = 1.0;
@ISA            = qw(Exporter);
@EXPORT_OK      = qw(&connectDB &checkDBSanity &allocateBlock);

@EXPORT         = (); # Export nothing by default.
%EXPORT_TAGS    = ( ALL => [qw( &connectDB &checkDBSanity &allocateBlock )]
                  );


# Creates connection to IPDB.
# Default is a PostgreSQL db;  could be any DBMS with the
# right changes.  MySQL in comments.  Note that some DBMS's don't
# support transactions, this is a Bad Thing!
# Returns a handle to the db.
sub connectDB {
  my $dbh;
  my $DSN = "DBI:Pg:dbname=ipdb";
  my $user = 'ipdb';
  my $pw   = 'ipdbpwd';

# Note that we want to autocommit by default, and we will turn it off locally as necessary.
  $dbh = DBI->connect($DSN, $user, $pw, { AutoCommit => 1 } )
    or return undef if(!$dbh);

  return $dbh;
} # end connectDB

# Quick check to see if the db is responding.  A full integrity
# check will have to be a separate tool to walk the IP allocation trees.
sub checkDBSanity {
  my $dbh = connectDB();

  if (!$dbh) {
    print "Cannot connect to the database!";
  } else {
    # it connects, try a stmt.
    my $sth = $dbh->prepare('select cidr from masterblocks');
    my $err = $sth->execute();

    if ($sth->fetchrow()) {
      # all is well.
      return 1;
    } else {
      print "Connected to the database, but could not execute test statement.  ".$sth->errstr();
    }
  }
  # Clean up after ourselves.
  $dbh->disconnect;
} # end checkDBSanity


# allocateBlock()
# Does all of the magic of actually allocating a netblock
sub allocateBlock($) {
}


# mailNotify()
# Sends notification mail to recipients regarding an IPDB operation
sub mailNotify ($$$) {
  my ($recip,$subj,$message) = @_;
  my $mailer = new Net::SMTP "smtp.example.com";

  $mailer->mail('ipdb@example.com');
  $mailer->to($recip);
  $mailer->data("X-Mailer: IPDB Notify v$IPDB::Version\n",
                "Date: ".strftime("%a, %d %b %Y %H:%M:%S %z\n", localtime)."\n",
                "Organization: Example Corp.",
                "From: \"IP Database\" <ipdb\@example.com>\n",
                "Subject: $subj\n",
                "\n",
                "$message\n");
  $mailer->quit;
}

# Indicates module loaded OK.  Required by Perl.
1;