mirror of
https://codeberg.org/mclemens/tsmmonitor2.git
synced 2024-06-09 20:50:48 +00:00
321 lines
9.0 KiB
PHP
321 lines
9.0 KiB
PHP
<?php
|
|
|
|
/*
|
|
************************************************************************
|
|
This file is part of TSM Monitor.
|
|
|
|
TSM Monitor 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 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TSM Monitor 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 TSM Monitor. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
************************************************************************
|
|
*/
|
|
|
|
/**
|
|
*
|
|
* adodb.php, TSM Monitor
|
|
*
|
|
* DB Stuff for TSM Monitor
|
|
*
|
|
* @author Frank Fegert, Michael Clemens
|
|
* @package tsmmonitor
|
|
*/
|
|
|
|
|
|
/**
|
|
*
|
|
* Class ADOdb
|
|
*
|
|
*/
|
|
|
|
class ADOdb {
|
|
|
|
|
|
var $conn;
|
|
var $debug;
|
|
var $logfile;
|
|
|
|
/**
|
|
* constructor - establishes a DB connection via ADODB
|
|
*
|
|
* @param string $host the hostname of the DB server
|
|
* @param string $port the portnumber for the DB connection
|
|
* @param string $user the username for the DB connection
|
|
* @param string $pass the password for the DB connection
|
|
* @param string $db_name the name of the DB
|
|
* @param string $db_type the type of the DB (currently only 'mysql')
|
|
* @param string $retr the number attempts for the DB connection before a failure is reported
|
|
* @return 0
|
|
*/
|
|
function ADOdb($host, $port = "3306", $user, $pass, $db_name, $db_type, $retr = 20, $debug = FALSE, $logfile = FALSE) {
|
|
|
|
$this->debug = $debug;
|
|
if ($logfile != FALSE) $this->logfile = $logfile;
|
|
|
|
$try = 0;
|
|
$hostport = $host . ":" . $port;
|
|
$this->conn = NewADOConnection($db_type);
|
|
while ($try <= $retries) {
|
|
if ($this->conn->NConnect($hostport,$user,$pass,$db_name)) {
|
|
$this->conn = $this->conn;
|
|
return 0;
|
|
}
|
|
$try++;
|
|
usleep(50000);
|
|
}
|
|
|
|
die("FATAL: Cannot connect to database server on '$host':'$port'. Please make sure you have specified a valid database name in 'includes/config.php'\n");
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
* setDebug - enables or disabled debug mode
|
|
*
|
|
* @param string $debug On or Off
|
|
*/
|
|
function setDebug($debug) {
|
|
if ($debug == "On") {
|
|
$this->debug = TRUE;
|
|
} else {
|
|
$this->debug = FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* setLogfile - set logfile path and name
|
|
*
|
|
* @param string $logfile Logfile path and name
|
|
*/
|
|
function setLogfile($logfile) {
|
|
if ($logfile != "") {
|
|
$this->logfile = $logfile;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* writeMSG
|
|
*
|
|
* @param mixed $msg
|
|
* @access public
|
|
* @return void
|
|
*/
|
|
function writeMSG($msg) {
|
|
|
|
if ($this->logfile != "") {
|
|
if ($loghandle = fopen($this->logfile, 'a')) {
|
|
$starttime = microtime();
|
|
do {
|
|
$lock = flock($loghandle, LOCK_EX);
|
|
if (!$lock) usleep(round(rand(0, 100) * 1000));
|
|
} while (!$lock && ((microtime() - $starttime) < 1000));
|
|
|
|
if ($lock) {
|
|
fwrite($loghandle, $msg);
|
|
}
|
|
fclose($loghandle);
|
|
} else {
|
|
echo "ERROR: Cannot open logfile: '".$logfile."' for writing. Falling back to STDOUT.\n";
|
|
echo $msg;
|
|
}
|
|
} else {
|
|
echo $msg;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* closeDB - close an open DB connection
|
|
*
|
|
* @return string
|
|
*/
|
|
function closeDB() {
|
|
if ($this->conn) {
|
|
return $this->conn->Close();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* execDB - execute a SQL statement against the DB via ADODB
|
|
*
|
|
* @param string $sql SQL statement to execute
|
|
* @return ADORecordSet
|
|
*/
|
|
function execDB($sql) {
|
|
$this->conn->debug = $this->debug;
|
|
$sql = $this->sanitizeSQL($sql);
|
|
|
|
try {
|
|
$recordSet = &$this->conn->Execute($sql);
|
|
} catch (exception $e) {
|
|
$this->writeMSG(adodb_backtrace($e->gettrace()));
|
|
exit;
|
|
}
|
|
|
|
if (($recordSet) || ($this->conn->ErrorNo() == 0)) {
|
|
return($recordSet);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* fetchCellDB - execute a SQL query against the DB via ADODB and
|
|
* return only the first column of the fist row found
|
|
* or a specified column of the fist row found
|
|
*
|
|
* @param string $sql SQL statement to execute
|
|
* @param $column_name Column name to use instead of the first column
|
|
* @return string Content of the cell as a single variable
|
|
*/
|
|
function fetchCellDB($sql, $column_name) {
|
|
$this->conn->debug = $this->debug;
|
|
$sql = $this->sanitizeSQL($sql);
|
|
|
|
if ($column_name != '') {
|
|
$this->conn->SetFetchMode(ADODB_FETCH_ASSOC);
|
|
} else {
|
|
$this->conn->SetFetchMode(ADODB_FETCH_NUM);
|
|
}
|
|
try {
|
|
$recordSet = $this->conn->Execute($sql);
|
|
} catch (exception $e) {
|
|
$this->writeMSG(adodb_backtrace($e->gettrace()));
|
|
exit;
|
|
}
|
|
|
|
if (($recordSet) || ($this->conn->ErrorNo() == 0)) {
|
|
if (!$recordSet->EOF) {
|
|
if ($column_name != '') {
|
|
$column = $recordSet->fields[$column_name];
|
|
}else{
|
|
$column = $recordSet->fields[0];
|
|
}
|
|
$recordSet->close();
|
|
|
|
return($column);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* fetchRowDB - execute a SQL query against the DB via ADODB
|
|
* and return only the first row found
|
|
*
|
|
* @param string $sql SQL statement to execute
|
|
* @return array First row of results as an associative array
|
|
*/
|
|
function fetchRowDB($sql) {
|
|
$this->conn->debug = $this->debug;
|
|
$sql = $this->sanitizeSQL($sql);
|
|
|
|
$this->conn->SetFetchMode(ADODB_FETCH_ASSOC);
|
|
try {
|
|
$recordSet = $this->conn->Execute($sql);
|
|
} catch (exception $e) {
|
|
$this->writeMSG(adodb_backtrace($e->gettrace()));
|
|
exit;
|
|
}
|
|
|
|
if (($recordSet) || ($this->conn->ErrorNo() == 0)) {
|
|
if (!$recordSet->EOF) {
|
|
$recordFields = $recordSet->fields;
|
|
$recordSet->close();
|
|
|
|
return($recordFields);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* fetchArrayDB - execute a SQL query against the DB via ADODB
|
|
* and return results in an associative array.
|
|
*
|
|
* @param string $sql SQL statement to execute
|
|
* @return array All results in an associative array
|
|
*/
|
|
function fetchArrayDB($sql) {
|
|
$this->conn->debug = $this->debug;
|
|
$sql = $this->sanitizeSQL($sql);
|
|
|
|
$recordArray = array();
|
|
$this->conn->SetFetchMode(ADODB_FETCH_ASSOC);
|
|
try {
|
|
$recordSet = $this->conn->Execute($sql);
|
|
} catch (exception $e) {
|
|
$this->writeMSG(adodb_backtrace($e->gettrace()));
|
|
exit;
|
|
}
|
|
|
|
if (($recordSet) || ($this->conn->ErrorNo() == 0)) {
|
|
while ((!$recordSet->EOF) && ($recordSet)) {
|
|
$recordArray{sizeof($recordArray)} = $recordSet->fields;
|
|
$recordSet->MoveNext();
|
|
}
|
|
$recordSet->close();
|
|
return($recordArray);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* updateDB - execute a SQL update statement against the DB via ADODB
|
|
* to update a record. If the record is not found, an insert
|
|
* statement is generated and executed.
|
|
*
|
|
* @param string $table The name of the table containing the record to be updated
|
|
* @param array $cells An array of columnname/value pairs of the record to be updated
|
|
* @param string $keys Name of the primary key
|
|
* @param boolean $autoquote Use intelligent auto-quoting
|
|
* @param ADOConnection $this->conn DB connection ID to run the SQL against
|
|
* @return string Auto-increment ID if insert was performed
|
|
*/
|
|
function updateDB($table, $cells, $keys, $autoquote = TRUE) {
|
|
$this->conn->debug = $this->debug;
|
|
try {
|
|
$this->conn->Replace($table, $cells, $keys, $autoquote);
|
|
} catch (exception $e) {
|
|
$this->writeMSG(adodb_backtrace($e->gettrace()));
|
|
exit;
|
|
}
|
|
|
|
return $this->conn->Insert_ID();
|
|
}
|
|
|
|
|
|
/**
|
|
* sanitizeSQL - removes unwanted chars in values passed for use in
|
|
* SQL statements
|
|
*
|
|
* @param string $sql SQL expression to sanitize
|
|
* @return string
|
|
*/
|
|
function sanitizeSQL($sql) {
|
|
$sql = str_replace(";", "\;", $sql);
|
|
$sql = str_replace("\n", "", $sql);
|
|
$sql = str_replace("\r", "", $sql);
|
|
$sql = str_replace("\t", " ", $sql);
|
|
return $sql;
|
|
}
|
|
|
|
}
|
|
|
|
?>
|