X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=include%2Fdba.php;h=8d224b570f44299c431f087392cc5e735787ef27;hb=2347a4326c0a0877595ed1c634d04bcf765ce90f;hp=1421a703dc730ad1a7eb803e30c632ff0d17937c;hpb=ee714daf4d33f8258e1301438423fc10a3bdc916;p=friendica.git diff --git a/include/dba.php b/include/dba.php index 1421a703dc..8d224b570f 100644 --- a/include/dba.php +++ b/include/dba.php @@ -8,273 +8,213 @@ require_once('include/datetime.php'); * * For debugging, insert 'dbg(1);' anywhere in the program flow. * dbg(0); will turn it off. Logging is performed at LOGGER_DATA level. - * When logging, all binary info is converted to - * text and html entities are escaped so that - * the debugging stream is safe to view - * within both terminals and web pages. + * When logging, all binary info is converted to text and html entities are escaped so that + * the debugging stream is safe to view within both terminals and web pages. * */ + +if(! class_exists('dba')) { +class dba { + + private $debug = 0; + private $db; + public $mysqli = true; + public $connected = false; + public $error = false; + + function __construct($server,$user,$pass,$db,$install = false) { + + $server = trim($server); + $user = trim($user); + $pass = trim($pass); + $db = trim($db); + + if (!(strlen($server) && strlen($user))){ + $this->connected = false; + $this->db = null; + return; + } -if(! class_exists('dba')) { - - class dba { - - private $debug = 0; - private $db; - private $exceptions; - - public $mysqli = true; - public $connected = false; - public $error = false; - - function __construct($server,$user,$pass,$db,$install = false) { - - $server = trim($server); - $user = trim($user); - $pass = trim($pass); - $db = trim($db); - - //we need both, server and username, so fail if one is missing - if (!(strlen($server) && strlen($user))){ - $this->connected = false; - $this->db = null; - throw new InvalidArgumentException(t("Server name of user name are missing. ")); - } - - //when we are installing - if($install) { - if(strlen($server) && ($server !== 'localhost') && ($server !== '127.0.0.1')) { - if(! dns_get_record($server, DNS_A + DNS_CNAME + DNS_PTR)) { - $this->connected = false; - $this->db = null; - throw new InvalidArgumentException( t('Cannot locate DNS info for database server \'%s\''), $server); - } + if($install) { + if(strlen($server) && ($server !== 'localhost') && ($server !== '127.0.0.1')) { + if(! dns_get_record($server, DNS_A + DNS_CNAME + DNS_PTR)) { + $this->error = sprintf( t('Cannot locate DNS info for database server \'%s\''), $server); + $this->connected = false; + $this->db = null; + return; } } + } - if(class_exists('mysqli')) { - $this->db = new mysqli($server,$user,$pass,$db); - if(NULL === $this->db->connect_error) { - $this->connected = true; - } else { - throw new RuntimeException($this->db->connect_error); - } - } else { - $this->mysqli = false; - $this->db = mysql_connect($server,$user,$pass); - if($this->db && mysql_select_db($db,$this->db)) { - $this->connected = true; - } else { - throw new RuntimeException(mysql_error()); - } + if(class_exists('mysqli')) { + $this->db = @new mysqli($server,$user,$pass,$db); + if(! mysqli_connect_errno()) { + $this->connected = true; } } - - public function excep($excep) { - $this->exceptions=$excep; + else { + $this->mysqli = false; + $this->db = mysql_connect($server,$user,$pass); + if($this->db && mysql_select_db($db,$this->db)) { + $this->connected = true; + } } - - public function getdb() { - return $this->db; + if(! $this->connected) { + $this->db = null; + if(! $install) + system_unavailable(); } + } - public function q($sql) { - - if((! $this->db) || (! $this->connected)) { - $this->throwOrLog(new RuntimeException(t("There is no db connection. "))); - return; - } + public function getdb() { + return $this->db; + } - if($this->mysqli) { - $result = $this->db->query($sql); - } else { - $result = mysql_query($sql,$this->db); - } + public function q($sql) { + global $a; - //on debug mode or fail, the query is written to the log. - //this won't work if logger can not read it's logging level - //from the db. - if($this->debug || FALSE === $result) { - - $mesg = ''; - - if($result === false) { - $mesg = 'false '.$this->error(); - } elseif($result === true) { - $mesg = 'true'; - } else { - if($this->mysqli) { - $mesg = $result->num_rows . t(' results') . EOL; - } else { - $mesg = mysql_num_rows($result) . t(' results') . EOL; - } - } + if((! $this->db) || (! $this->connected)) + return false; - $str = 'SQL = ' . printable($sql) . EOL . t('SQL returned ') . $mesg . EOL; + $this->error = ''; + if(x($a->config,'system') && x($a->config['system'],'db_log')) + $stamp1 = microtime(true); - // If dbfail.out exists, we will write any failed calls directly to it, - // regardless of any logging that may or may nor be in effect. - // These usually indicate SQL syntax errors that need to be resolved. - if(file_exists('dbfail.out')) { - file_put_contents('dbfail.out', datetime_convert() . "\n" . $str . "\n", FILE_APPEND); - } - logger('dba: ' . $str ); - if(FALSE===$result) { - $this->throwOrLog(new RuntimeException('dba: ' . $str)); - return; - } + if($this->mysqli) + $result = @$this->db->query($sql); + else + $result = @mysql_query($sql,$this->db); + + if(x($a->config,'system') && x($a->config['system'],'db_log')) { + $stamp2 = microtime(true); + $duration = round($stamp2-$stamp1, 3); + if ($duration > $a->config["system"]["db_loglimit"]) { + $backtrace = debug_backtrace(DEBUG_BACKTRACE_IGNORE_ARGS); + @file_put_contents($a->config["system"]["db_log"], $duration."\t". + basename($backtrace[1]["file"])."\t". + $backtrace[1]["line"]."\t".$backtrace[2]["function"]."\t". + substr($sql, 0, 2000)."\n", FILE_APPEND); } - + } - if($result === true) { - return $result; - } + if($this->mysqli) { + if($this->db->errno) + $this->error = $this->db->error; + } + elseif(mysql_errno($this->db)) + $this->error = mysql_error($this->db); - $r = array(); - if($this->mysqli) { - if($result->num_rows) { - while($x = $result->fetch_array(MYSQLI_ASSOC)) { - $r[] = $x; - } - $result->free_result(); - } - } else { - if(mysql_num_rows($result)) { - while($x = mysql_fetch_array($result, MYSQL_ASSOC)) { - $r[] = $x; - } - mysql_free_result($result); - } - } + if(strlen($this->error)) { + logger('dba: ' . $this->error); + } + if($this->debug) { - if($this->debug) { - logger('dba: ' . printable(print_r($r, true))); - } - return($r); - } + $mesg = ''; - private function error() { - if($this->mysqli) { - return $this->db->error; - } else { - return mysql_error($this->db); + if($result === false) + $mesg = 'false'; + elseif($result === true) + $mesg = 'true'; + else { + if($this->mysqli) + $mesg = $result->num_rows . ' results' . EOL; + else + $mesg = mysql_num_rows($result) . ' results' . EOL; } - } - - private function throwOrLog(Exception $ex) { - if($this->exceptions) { - throw $ex; - } else { - logger('dba: '.$ex->getMessage()); - } - } - - /** - * starts a transaction. Transactions need to be finished with - * commit() or rollback(). Please mind that the db table engine may - * not support this. - */ - public function beginTransaction() { - if($this->mysqli) { - return $this->db->autocommit(false); - } else { - //no transaction support in mysql module... - mysql_query('SET AUTOCOMMIT = 0;', $db); - } - } - - /** - * rollback a transaction. So, rollback anything that was done since the last call - * to beginTransaction(). - */ - public function rollback() { - if($this->mysqli) { - return $this->db->rollback(); - } else { - //no transaction support in mysql module... - mysql_query('ROLLBACK;', $db); - } - $this->stopTransaction(); + + $str = 'SQL = ' . printable($sql) . EOL . 'SQL returned ' . $mesg + . (($this->error) ? ' error: ' . $this->error : '') + . EOL; + + logger('dba: ' . $str ); } /** - * commit a transaction. So, write any query to the database. + * If dbfail.out exists, we will write any failed calls directly to it, + * regardless of any logging that may or may nor be in effect. + * These usually indicate SQL syntax errors that need to be resolved. */ - public function commit() { - if($this->mysqli) { - return $this->db->commit(); - } else { - //no transaction support in mysql module... - mysql_query('COMMIT;', $db); - } - $this->stopTransaction(); - } - - private function stopTransaction() { - if($this->mysqli) { - return $this->db->autocommit(true); - } else { - //no transaction support in mysql module... - mysql_query('SET AUTOCOMMIT = 1;', $db); - } - } - - public function dbg($dbg) { - $this->debug = $dbg; + + if($result === false) { + logger('dba: ' . printable($sql) . ' returned false.' . "\n" . $this->error); + if(file_exists('dbfail.out')) + file_put_contents('dbfail.out', datetime_convert() . "\n" . printable($sql) . ' returned false' . "\n" . $this->error . "\n", FILE_APPEND); } - public function escape($str) { - if($this->db && $this->connected) { - if($this->mysqli) { - return $this->db->real_escape_string($str); - } else { - return mysql_real_escape_string($str,$this->db); - } + if(($result === true) || ($result === false)) + return $result; + + $r = array(); + if($this->mysqli) { + if($result->num_rows) { + while($x = $result->fetch_array(MYSQLI_ASSOC)) + $r[] = $x; + $result->free_result(); } } - - function __destruct() { - if ($this->db) { - if($this->mysqli) { - $this->db->close(); - } - } else { - mysql_close($this->db); + else { + if(mysql_num_rows($result)) { + while($x = mysql_fetch_array($result, MYSQL_ASSOC)) + $r[] = $x; + mysql_free_result($result); } } + + + if($this->debug) + logger('dba: ' . printable(print_r($r, true))); + return($r); } -} -if(! function_exists('printable')) { - function printable($s) { - $s = preg_replace("~([\x01-\x08\x0E-\x0F\x10-\x1F\x7F-\xFF])~",".", $s); - $s = str_replace("\x00",'.',$s); - if(x($_SERVER,'SERVER_NAME')) - $s = escape_tags($s); - return $s; + public function dbg($dbg) { + $this->debug = $dbg; } -} -// Procedural functions -if(! function_exists('dbg')) { - function dbg($state) { - global $db; - if($db) - $db->dbg($state); + public function escape($str) { + if($this->db && $this->connected) { + if($this->mysqli) + return @$this->db->real_escape_string($str); + else + return @mysql_real_escape_string($str,$this->db); + } } -} -if(! function_exists('dbesc')) { - function dbesc($str) { - global $db; - if($db && $db->connected) - return($db->escape($str)); - else - return(str_replace("'","\\'",$str)); + function __destruct() { + if ($this->db) + if($this->mysqli) + $this->db->close(); + else + mysql_close($this->db); } -} +}} + +if(! function_exists('printable')) { +function printable($s) { + $s = preg_replace("~([\x01-\x08\x0E-\x0F\x10-\x1F\x7F-\xFF])~",".", $s); + $s = str_replace("\x00",'.',$s); + if(x($_SERVER,'SERVER_NAME')) + $s = escape_tags($s); + return $s; +}} + +// Procedural functions +if(! function_exists('dbg')) { +function dbg($state) { + global $db; + if($db) + $db->dbg($state); +}} + +if(! function_exists('dbesc')) { +function dbesc($str) { + global $db; + if($db && $db->connected) + return($db->escape($str)); + else + return(str_replace("'","\\'",$str)); +}} @@ -283,31 +223,30 @@ if(! function_exists('dbesc')) { // Example: $r = q("SELECT * FROM `%s` WHERE `uid` = %d", // 'user', 1); -if(! function_exists('q')) { - function q($sql) { +if(! function_exists('q')) { +function q($sql) { - global $db; - $args = func_get_args(); - unset($args[0]); + global $db; + $args = func_get_args(); + unset($args[0]); - if($db && $db->connected) { - $stmt = vsprintf($sql,$args); - if($stmt === false) - logger('dba: vsprintf error: ' . print_r(debug_backtrace(),true)); - return $db->q($stmt); - } + if($db && $db->connected) { + $stmt = vsprintf($sql,$args); + if($stmt === false) + logger('dba: vsprintf error: ' . print_r(debug_backtrace(),true)); + return $db->q($stmt); + } - /** - * - * This will happen occasionally trying to store the - * session data after abnormal program termination - * - */ - logger('dba: no database: ' . print_r($args,true)); - return false; + /** + * + * This will happen occasionally trying to store the + * session data after abnormal program termination + * + */ + logger('dba: no database: ' . print_r($args,true)); + return false; - } -} +}} /** * @@ -315,39 +254,40 @@ if(! function_exists('q')) { * */ -if(! function_exists('dbq')) { - function dbq($sql) { +if(! function_exists('dbq')) { +function dbq($sql) { - global $db; - if($db && $db->connected) - $ret = $db->q($sql); - else - $ret = false; - return $ret; - } -} + global $db; + if($db && $db->connected) + $ret = $db->q($sql); + else + $ret = false; + return $ret; +}} -// Caller is responsible for ensuring that any integer arguments to +// Caller is responsible for ensuring that any integer arguments to // dbesc_array are actually integers and not malformed strings containing -// SQL injection vectors. All integer array elements should be specifically -// cast to int to avoid trouble. +// SQL injection vectors. All integer array elements should be specifically +// cast to int to avoid trouble. if(! function_exists('dbesc_array_cb')) { - function dbesc_array_cb(&$item, $key) { - if(is_string($item)) - $item = dbesc($item); - } -} +function dbesc_array_cb(&$item, $key) { + if(is_string($item)) + $item = dbesc($item); +}} if(! function_exists('dbesc_array')) { - function dbesc_array(&$arr) { - if(is_array($arr) && count($arr)) { - array_walk($arr,'dbesc_array_cb'); - } +function dbesc_array(&$arr) { + if(is_array($arr) && count($arr)) { + array_walk($arr,'dbesc_array_cb'); } -} +}} +function dba_timer() { + return microtime(true); +} +