]> git.mxchange.org Git - friendica.git/blobdiff - include/dba.php
Merge pull request #276 from tomtom84/master
[friendica.git] / include / dba.php
index d3da8eae055471b656142dad8ee6b63fb0944740..c9f880241ba9003d38cc2a356c90e62daeb7fc59 100644 (file)
@@ -1,5 +1,7 @@
 <?php
 
+require_once('include/datetime.php');
+
 /**
  *
  * MySQL database class
@@ -16,16 +18,52 @@ class dba {
 
        private $debug = 0;
        private $db;
+       public  $mysqli = true;
        public  $connected = false;
+       public  $error = false;
 
        function __construct($server,$user,$pass,$db,$install = false) {
-               $this->db = @new mysqli($server,$user,$pass,$db);
-               if((mysqli_connect_errno()) && (! $install)) {
+
+               $server = trim($server);
+               $user = trim($user);
+               $pass = trim($pass);
+               $db = trim($db);
+
+               if (!(strlen($server) && strlen($user))){
+                       $this->connected = false;
                        $this->db = null;
-                       system_unavailable();
+                       return;                 
+               }
+               
+               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(! mysqli_connect_errno()) {
+                               $this->connected = true;
+                       }
+               }
+               else {
+                       $this->mysqli = false;
+                       $this->db = mysql_connect($server,$user,$pass);
+                       if($this->db && mysql_select_db($db,$this->db)) {
+                               $this->connected = true;
+                       }
+               }
+               if(! $this->connected) {
+                       $this->db = null;
+                       if(! $install)
+                               system_unavailable();
                }
-               else
-                       $this->connected = true;    
        }
 
        public function getdb() {
@@ -34,56 +72,83 @@ class dba {
 
        public function q($sql) {
                
-               if(! $this->db )
+               if((! $this->db) || (! $this->connected))
                        return false;
                
-               $result = @$this->db->query($sql);
+               $this->error = '';
+
+               if($this->mysqli)
+                       $result = @$this->db->query($sql);
+               else
+                       $result = @mysql_query($sql,$this->db);
+
+               if($this->mysqli) {
+                       if($this->db->errno)
+                               $this->error = $this->db->error;
+               }
+               elseif(mysql_errno($this->db))
+                               $this->error = mysql_error($this->db);  
+
+               if(strlen($this->error)) {
+                       logger('dba: ' . $this->error);
+               }
 
                if($this->debug) {
 
                        $mesg = '';
 
-                       if($this->db->errno)
-                               logger('dba: ' . $this->db->error);
-
                        if($result === false)
                                $mesg = 'false';
                        elseif($result === true)
                                $mesg = 'true';
-                       else
-                               $mesg = $result->num_rows . ' results' . EOL;
-        
-                       $str =  'SQL = ' . printable($sql) . EOL . 'SQL returned ' . $mesg . EOL;
+                       else {
+                               if($this->mysqli)
+                                       $mesg = $result->num_rows . ' results' . EOL;
+                       else
+                                       $mesg = mysql_num_rows($result) . ' results' . EOL;
+                       }
+    
+                       $str =  'SQL = ' . printable($sql) . EOL . 'SQL returned ' . $mesg 
+                               . (($this->error) ? ' error: ' . $this->error : '')
+                               . EOL;
 
                        logger('dba: ' . $str );
                }
-               else {
 
-                       /*
-                        * 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 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($result === false) {
-                               logger('dba: ' . printable($sql) . ' returned false.');
-                               if(file_exists('dbfail.out'))
-                                       file_put_contents('dbfail.out', printable($sql) . ' returned false' . "\n", FILE_APPEND);
-                       }
+               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);
                }
 
                if(($result === true) || ($result === false))
                        return $result;
 
                $r = array();
-               if($result->num_rows) {
-                       while($x = $result->fetch_array(MYSQL_ASSOC))
-                               $r[] = $x;
-                       $result->free_result();
+               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($this->debug)
-                       logger('dba: ' . printable(print_r($r, true)), LOGGER_DATA);
+                       logger('dba: ' . printable(print_r($r, true)));
                return($r);
        }
 
@@ -92,11 +157,20 @@ class dba {
        }
 
        public function escape($str) {
-               return @$this->db->real_escape_string($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);
+               }
        }
 
        function __destruct() {
-               @$this->db->close();
+               if ($this->db) 
+                       if($this->mysqli)
+                               $this->db->close();
+                       else
+                               mysql_close($this->db);
        }
 }}
 
@@ -113,19 +187,21 @@ function printable($s) {
 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->connected)
+       if($db && $db->connected)
                return($db->escape($str));
        else
                return(str_replace("'","\\'",$str));
 }}
 
 
+
 // Function: q($sql,$args);
 // Description: execute SQL query with printf style args.
 // Example: $r = q("SELECT * FROM `%s` WHERE `uid` = %d",
@@ -138,9 +214,11 @@ function q($sql) {
        $args = func_get_args();
        unset($args[0]);
 
-       if($db->connected) {
-               $ret = $db->q(vsprintf($sql,$args));
-               return $ret;
+       if($db && $db->connected) {
+               $stmt = vsprintf($sql,$args);
+               if($stmt === false)
+                       logger('dba: vsprintf error: ' . print_r(debug_backtrace(),true));
+               return $db->q($stmt);
        }
 
        /**
@@ -149,7 +227,6 @@ function q($sql) {
         * session data after abnormal program termination 
         *
         */
-
        logger('dba: no database: ' . print_r($args,true));
        return false; 
 
@@ -165,7 +242,7 @@ if(! function_exists('dbq')) {
 function dbq($sql) {
 
        global $db;
-       if($db->connected)
+       if($db && $db->connected)
                $ret = $db->q($sql);
        else
                $ret = false;
@@ -191,4 +268,4 @@ function dbesc_array(&$arr) {
        if(is_array($arr) && count($arr)) {
                array_walk($arr,'dbesc_array_cb');
        }
-}}             
\ No newline at end of file
+}}