dbinit.php

Network Administration Visualized 网络管理可视化源码

<?php
/*
 * $Id: dbinit.php 3259 2005-06-09 12:28:58Z andrs $
 *
 * Copyright 2002-2004 UNINETT AS
 * 
 * This file is part of Network Administration Visualized (NAV)
 *
 * NAV is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * NAV is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with NAV; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *
 * Authors: Andreas Aakre Solberg <[e-mail]>
 *
 */


/**
 * dbinit
 * 
 * The class dbinit handles database connectivitiy initialization.
 * Among the tasks of this class is to read database connect information from the
 * config file.
 */
class dbinit {

    var $dbh;
    var $dbh_dbcon;
    
    var $dbhk;
    var $dbhk_dbcon;
    
    // Constructor
    function dbinit() {
        //dl("pgsql.so");
        
        $this->dbh = null;
        $this->dbh_dbcon = null;
        
        $this->dbhk = null;
        $this->dbhk_dbcon = null;
    }

    function get_dbh() {
        if ($this->dbh == null) {
            $filename = PATH_DB . "/db.conf";
            
           // Exit if cannot find db configuration file.
            if (!file_exists($filename)) {
                print "<h1>" . gettext("File access error") . "</h1>";
                print "<p>" . gettext("Could not find the database configuration file.");
                exit(0);
            }
            
            // Get fileconfiglines            
            $conffile = file($filename);
            
            // Init variables, in case they dont exist in config file...
            $dhost = "localhost";
            $dport = "5432";
            $ddb = "navprofiles_";
            $duser = "navprofile_";
            $dpass = "";
            
            
            // Traverse all entries in db.conf file.
            foreach ($conffile as $confline) {
            	
				// Skip comments.
            		if (preg_match('/^\s*#/', $confline)) next;
            		
                $tvar = split('=', trim($confline));
                if (sizeof($tvar) > 1) {
					$prop = trim($tvar[0]); $value = trim($tvar[1]);
					
					//print "<p>Property [$prop] Value [$value]\n";
	
					switch ($prop) {
						case 'dbhost'		: $dhost = $value; break;
						case 'dbport'		: $dport = $value; break;
						case 'db_navprofile'	: $ddb   = $value; break;
						case 'script_navprofile' 	: $duser = $value; break;
						case 'userpw_' . $duser  	: $dpass = $value; break;
					}
                }
            }
            
            $cstr = "user=$duser password=$dpass dbname=$ddb";         
            
            	if (isset($dhost)) {
            		$cstr .= " host=$dhost";
            	}
            	if (isset($dport)) {
            		$cstr .= " port=$dport";
            	}
            //echo "<p>Connect string: " . $cstr;
            if (! $this->dbh_dbcon = pg_connect($cstr) ) {
                print "<h1>" . gettext("Database error") . "</h1>";
                print "<p>" . gettext("Could not connect to the navprofiles database. The database server could be down, or the logininfo could be corrupt in the database configuration file.");
                exit(0);
            } 

            $this->dbh = new DBH($this->dbh_dbcon);

        }
        return $this->dbh;
    }



    function get_dbhk() {
        if ($this->dbhk == null) {
            // Get fileconfiglines
            $filename = PATH_DB . "/db.conf";

            if (!file_exists($filename)) {
                print "<h1>" . gettext("File access error") . "</h1>";
                print "<p>" . gettext("Could not find the database configuration file.");
                exit(0);
            }

            
            $conffile = file($filename);
            
            // Init variables, in case they dont exist in config file...
            $dhost = "localhost";
            $dport = "5432";
            $ddb = "manage_";
            $duser = "navprofilemanage_";
            $dpass = "";
            

            // Traverse all entries in db.conf file.
            foreach ($conffile as $confline) {
            	
				// Skip comments.
            		if (preg_match('/^\s*#/', $confline)) next;
            		
                $tvar = split('=', trim($confline));
                if (sizeof($tvar) > 1) {
					$prop = trim($tvar[0]); $value = trim($tvar[1]);
					
					//print "<p>Property [$prop] Value [$value]\n";
	
					switch ($prop) {
						case 'dbhost'			: $dhost = $value; break;
						case 'dbport'			: $dport = $value; break;
						case 'db_navprofilemanage'		: $ddb   = $value; break;
						case 'script_navprofilemanage' 	: $duser = $value; break;
						case 'userpw_' . $duser 	: $dpass = $value; break;
					}
                }
            }
            
            $cstr = "user=$duser password=$dpass dbname=$ddb";         
            
            	if (isset($dhost)) {
            		$cstr .= " host=$dhost";
            	}
            	if (isset($dport)) {
            		$cstr .= " port=$dport";
            	}
   
            if (! $this->dbhk_dbcon = pg_connect($cstr) ) {
                print "<h1>" . gettext("Database error") . "</h1>";
                print "<p>" . gettext("Could not connect to the manage database. The database server could be down, or the logininfo could be corrupt in the database configuration file.");
                exit(0);
            } 

            $this->dbhk = new DBHK($this->dbhk_dbcon);

        }
        return $this->dbhk;
    }


    function closeall() {
        if (! is_null($this->dbh_con)) {
            pg_close($this->dbh_con);
        }
        
        if (! is_null($this->dbhk_con)) {
            pg_close($this->dbhk_con);
        }        
    }

}


$dbinit = new dbinit();

$dbh = $dbinit->get_dbh();



?>