[Summary view] [Print] [Text view]

   1  <?php
   2  /**
   3  *
   4  * This file is part of the phpBB Forum Software package.
   5  *
   6  * @copyright (c) phpBB Limited <https://www.phpbb.com>
   7  * @license GNU General Public License, version 2 (GPL-2.0)
   8  *
   9  * For full copyright and license information, please see
  10  * the docs/CREDITS.txt file.
  11  *
  12  */
  13  
  14  namespace phpbb\db\driver;
  15  
  16  /**
  17  * Unified ODBC functions
  18  * Unified ODBC functions support any database having ODBC driver, for example Adabas D, IBM DB2, iODBC, Solid, Sybase SQL Anywhere...
  19  * Here we only support MSSQL Server 2000+ because of the provided schema
  20  *
  21  * @note number of bytes returned for returning data depends on odbc.defaultlrl php.ini setting.
  22  * If it is limited to 4K for example only 4K of data is returned max, resulting in incomplete theme data for example.
  23  * @note odbc.defaultbinmode may affect UTF8 characters
  24  */
  25  class mssql_odbc extends \phpbb\db\driver\mssql_base
  26  {
  27      var $last_query_text = '';
  28      var $connect_error = '';
  29  
  30      /**
  31      * {@inheritDoc}
  32      */
  33  	function sql_connect($sqlserver, $sqluser, $sqlpassword, $database, $port = false, $persistency = false, $new_link = false)
  34      {
  35          $this->persistency = $persistency;
  36          $this->user = $sqluser;
  37          $this->dbname = $database;
  38  
  39          $port_delimiter = (defined('PHP_OS') && substr(PHP_OS, 0, 3) === 'WIN') ? ',' : ':';
  40          $this->server = $sqlserver . (($port) ? $port_delimiter . $port : '');
  41  
  42          $max_size = @ini_get('odbc.defaultlrl');
  43          if (!empty($max_size))
  44          {
  45              $unit = strtolower(substr($max_size, -1, 1));
  46              $max_size = (int) $max_size;
  47  
  48              if ($unit == 'k')
  49              {
  50                  $max_size = floor($max_size / 1024);
  51              }
  52              else if ($unit == 'g')
  53              {
  54                  $max_size *= 1024;
  55              }
  56              else if (is_numeric($unit))
  57              {
  58                  $max_size = floor((int) ($max_size . $unit) / 1048576);
  59              }
  60              $max_size = max(8, $max_size) . 'M';
  61  
  62              @ini_set('odbc.defaultlrl', $max_size);
  63          }
  64  
  65          if ($this->persistency)
  66          {
  67              if (!function_exists('odbc_pconnect'))
  68              {
  69                  $this->connect_error = 'odbc_pconnect function does not exist, is odbc extension installed?';
  70                  return $this->sql_error('');
  71              }
  72              $this->db_connect_id = @odbc_pconnect($this->server, $this->user, $sqlpassword);
  73          }
  74          else
  75          {
  76              if (!function_exists('odbc_connect'))
  77              {
  78                  $this->connect_error = 'odbc_connect function does not exist, is odbc extension installed?';
  79                  return $this->sql_error('');
  80              }
  81              $this->db_connect_id = @odbc_connect($this->server, $this->user, $sqlpassword);
  82          }
  83  
  84          return ($this->db_connect_id) ? $this->db_connect_id : $this->sql_error('');
  85      }
  86  
  87      /**
  88      * {@inheritDoc}
  89      */
  90  	function sql_server_info($raw = false, $use_cache = true)
  91      {
  92          global $cache;
  93  
  94          if (!$use_cache || empty($cache) || ($this->sql_server_version = $cache->get('mssqlodbc_version')) === false)
  95          {
  96              $result_id = @odbc_exec($this->db_connect_id, "SELECT SERVERPROPERTY('productversion'), SERVERPROPERTY('productlevel'), SERVERPROPERTY('edition')");
  97  
  98              $row = false;
  99              if ($result_id)
 100              {
 101                  $row = odbc_fetch_array($result_id);
 102                  odbc_free_result($result_id);
 103              }
 104  
 105              $this->sql_server_version = ($row) ? trim(implode(' ', $row)) : 0;
 106  
 107              if (!empty($cache) && $use_cache)
 108              {
 109                  $cache->put('mssqlodbc_version', $this->sql_server_version);
 110              }
 111          }
 112  
 113          if ($raw)
 114          {
 115              return $this->sql_server_version;
 116          }
 117  
 118          return ($this->sql_server_version) ? 'MSSQL (ODBC)<br />' . $this->sql_server_version : 'MSSQL (ODBC)';
 119      }
 120  
 121      /**
 122      * SQL Transaction
 123      * @access private
 124      */
 125  	function _sql_transaction($status = 'begin')
 126      {
 127          switch ($status)
 128          {
 129              case 'begin':
 130                  return @odbc_exec($this->db_connect_id, 'BEGIN TRANSACTION');
 131              break;
 132  
 133              case 'commit':
 134                  return @odbc_exec($this->db_connect_id, 'COMMIT TRANSACTION');
 135              break;
 136  
 137              case 'rollback':
 138                  return @odbc_exec($this->db_connect_id, 'ROLLBACK TRANSACTION');
 139              break;
 140          }
 141  
 142          return true;
 143      }
 144  
 145      /**
 146      * {@inheritDoc}
 147      */
 148  	function sql_query($query = '', $cache_ttl = 0)
 149      {
 150          if ($query != '')
 151          {
 152              global $cache;
 153  
 154              if ($this->debug_sql_explain)
 155              {
 156                  $this->sql_report('start', $query);
 157              }
 158              else if ($this->debug_load_time)
 159              {
 160                  $this->curtime = microtime(true);
 161              }
 162  
 163              $this->last_query_text = $query;
 164              $this->query_result = ($cache && $cache_ttl) ? $cache->sql_load($query) : false;
 165              $this->sql_add_num_queries($this->query_result);
 166  
 167              if ($this->query_result === false)
 168              {
 169                  if (($this->query_result = @odbc_exec($this->db_connect_id, $query)) === false)
 170                  {
 171                      $this->sql_error($query);
 172                  }
 173  
 174                  if ($this->debug_sql_explain)
 175                  {
 176                      $this->sql_report('stop', $query);
 177                  }
 178                  else if ($this->debug_load_time)
 179                  {
 180                      $this->sql_time += microtime(true) - $this->curtime;
 181                  }
 182  
 183                  if (!$this->query_result)
 184                  {
 185                      return false;
 186                  }
 187  
 188                  $safe_query_id = $this->clean_query_id($this->query_result);
 189  
 190                  if ($cache && $cache_ttl)
 191                  {
 192                      $this->open_queries[$safe_query_id] = $this->query_result;
 193                      $this->query_result = $cache->sql_save($this, $query, $this->query_result, $cache_ttl);
 194                  }
 195                  else if (strpos($query, 'SELECT') === 0)
 196                  {
 197                      $this->open_queries[$safe_query_id] = $this->query_result;
 198                  }
 199              }
 200              else if ($this->debug_sql_explain)
 201              {
 202                  $this->sql_report('fromcache', $query);
 203              }
 204          }
 205          else
 206          {
 207              return false;
 208          }
 209  
 210          return $this->query_result;
 211      }
 212  
 213      /**
 214      * Build LIMIT query
 215      */
 216  	function _sql_query_limit($query, $total, $offset = 0, $cache_ttl = 0)
 217      {
 218          $this->query_result = false;
 219  
 220          // Since TOP is only returning a set number of rows we won't need it if total is set to 0 (return all rows)
 221          if ($total)
 222          {
 223              // We need to grab the total number of rows + the offset number of rows to get the correct result
 224              if (strpos($query, 'SELECT DISTINCT') === 0)
 225              {
 226                  $query = 'SELECT DISTINCT TOP ' . ($total + $offset) . ' ' . substr($query, 15);
 227              }
 228              else
 229              {
 230                  $query = 'SELECT TOP ' . ($total + $offset) . ' ' . substr($query, 6);
 231              }
 232          }
 233  
 234          $result = $this->sql_query($query, $cache_ttl);
 235  
 236          // Seek by $offset rows
 237          if ($offset)
 238          {
 239              $this->sql_rowseek($offset, $result);
 240          }
 241  
 242          return $result;
 243      }
 244  
 245      /**
 246      * {@inheritDoc}
 247      */
 248  	function sql_affectedrows()
 249      {
 250          return ($this->db_connect_id) ? @odbc_num_rows($this->query_result) : false;
 251      }
 252  
 253      /**
 254      * {@inheritDoc}
 255      */
 256  	function sql_fetchrow($query_id = false)
 257      {
 258          global $cache;
 259  
 260          if ($query_id === false)
 261          {
 262              $query_id = $this->query_result;
 263          }
 264  
 265          $safe_query_id = $this->clean_query_id($query_id);
 266          if ($cache && $cache->sql_exists($safe_query_id))
 267          {
 268              return $cache->sql_fetchrow($safe_query_id);
 269          }
 270  
 271          return ($query_id) ? odbc_fetch_array($query_id) : false;
 272      }
 273  
 274      /**
 275       * {@inheritdoc}
 276       */
 277  	public function sql_last_inserted_id()
 278      {
 279          $result_id = @odbc_exec($this->db_connect_id, 'SELECT @@IDENTITY');
 280  
 281          if ($result_id)
 282          {
 283              if (odbc_fetch_array($result_id))
 284              {
 285                  $id = odbc_result($result_id, 1);
 286                  odbc_free_result($result_id);
 287                  return $id;
 288              }
 289              odbc_free_result($result_id);
 290          }
 291  
 292          return false;
 293      }
 294  
 295      /**
 296      * {@inheritDoc}
 297      */
 298  	function sql_freeresult($query_id = false)
 299      {
 300          global $cache;
 301  
 302          if ($query_id === false)
 303          {
 304              $query_id = $this->query_result;
 305          }
 306  
 307          $safe_query_id = $this->clean_query_id($query_id);
 308          if ($cache && $cache->sql_exists($safe_query_id))
 309          {
 310              return $cache->sql_freeresult($safe_query_id);
 311          }
 312  
 313          if (isset($this->open_queries[$safe_query_id]))
 314          {
 315              unset($this->open_queries[$safe_query_id]);
 316              return odbc_free_result($query_id);
 317          }
 318  
 319          return false;
 320      }
 321  
 322      /**
 323      * return sql error array
 324      * @access private
 325      */
 326  	function _sql_error()
 327      {
 328          if (function_exists('odbc_errormsg'))
 329          {
 330              $error = array(
 331                  'message'    => @odbc_errormsg(),
 332                  'code'        => @odbc_error(),
 333              );
 334          }
 335          else
 336          {
 337              $error = array(
 338                  'message'    => $this->connect_error,
 339                  'code'        => '',
 340              );
 341          }
 342  
 343          return $error;
 344      }
 345  
 346      /**
 347      * Close sql connection
 348      * @access private
 349      */
 350  	function _sql_close()
 351      {
 352          return @odbc_close($this->db_connect_id);
 353      }
 354  
 355      /**
 356      * Build db-specific report
 357      * @access private
 358      */
 359  	function _sql_report($mode, $query = '')
 360      {
 361          switch ($mode)
 362          {
 363              case 'start':
 364              break;
 365  
 366              case 'fromcache':
 367                  $endtime = explode(' ', microtime());
 368                  $endtime = $endtime[0] + $endtime[1];
 369  
 370                  $result = @odbc_exec($this->db_connect_id, $query);
 371                  if ($result)
 372                  {
 373                      while ($void = odbc_fetch_array($result))
 374                      {
 375                          // Take the time spent on parsing rows into account
 376                      }
 377                      odbc_free_result($result);
 378                  }
 379  
 380                  $splittime = explode(' ', microtime());
 381                  $splittime = $splittime[0] + $splittime[1];
 382  
 383                  $this->sql_report('record_fromcache', $query, $endtime, $splittime);
 384  
 385              break;
 386          }
 387      }
 388  }