includes/DatabaseFunctions.php

   1 <?
   2 include_once( "FulltextStoplist.php" );
   3 include_once( "CacheManager.php" );
   4
   5 define( "DB_READ", -1 );
   6 define( "DB_WRITE", -2 );
   7 define( "DB_LAST", -3 );
   8
   9 $wgLastDatabaseQuery = "";
  10
  11 function wfGetDB( $altuser = "", $altpassword = "", $altserver = "", $altdb = "" )
  12 {
  13         global $wgDBserver, $wgDBuser, $wgDBpassword;
  14         global $wgDBname, $wgDBconnection, $wgEmergencyContact;
  15
  16         $noconn = wfMsgNoDB( "noconnect", $wgDBserver );
  17         $nodb = wfMsgNoDB( "nodb", $wgDBname );
  18
  19         $helpme = "\n<p>If this error persists after reloading and clearing " .
  20           "your browser cache, please notify the <a href=\"mailto:" .
  21           $wgEmergencyContact . "\">Wikipedia developers</a>.</p>";
  22
  23         if ( $altuser != "" ) {
  24                 $serve = ($altserver ? $altserver : $wgDBserver );
  25                 $db = ($altdb ? $altdb : $wgDBname );
  26                 $wgDBconnection = mysql_connect( $serve, $altuser, $altpassword )
  27                         or die( "bad sql user" );
  28                 mysql_select_db( $db, $wgDBconnection ) or die(
  29                   htmlspecialchars(mysql_error()) );
  30         }
  31
  32         if ( ! $wgDBconnection ) {
  33                 @$wgDBconnection = mysql_pconnect( $wgDBserver, $wgDBuser, $wgDBpassword )
  34                         or wfEmergencyAbort();
  35
  36                 if( !mysql_select_db( $wgDBname, $wgDBconnection ) ) {
  37                         /* Persistent connections may become stuck in an unusable state */
  38                         wfDebug( "Persistent connection is broken?\n", true );
  39
  40                         @$wgDBconnection = mysql_connect( $wgDBserver, $wgDBuser, $wgDBpassword )
  41                                 or wfEmergencyAbort();
  42
  43                         @mysql_select_db( $wgDBname, $wgDBconnection )
  44                                 or wfEmergencyAbort();
  45                 }
  46         }
  47         # mysql_ping( $wgDBconnection );
  48         return $wgDBconnection;
  49 }
  50
  51 /* Call this function if we couldn't contact the database...
  52    We'll try to use the cache to display something in the meantime */
  53 function wfEmergencyAbort( $msg = "" ) {
  54         global $wgTitle, $wgUseFileCache, $title, $wgOutputEncoding;
  55
  56         header( "Content-type: text/html; charset=$wgOutputEncoding" );
  57         if($msg == "") $msg = wfMsgNoDB( "noconnect" );
  58         $text = $msg;
  59
  60         if($wgUseFileCache) {
  61                 if($wgTitle) {
  62                         $t =& $wgTitle;
  63                 } else {
  64                         if($title) {
  65                                 $t = Title::newFromURL( $title );
  66                         } else {
  67                                 $t = Title::newFromText( wfMsgNoDB( "mainpage" ) );
  68                         }
  69                 }
  70
  71                 $cache = new CacheManager( $t );
  72                 if( $cache->isFileCached() ) {
  73                         $msg = "<p style='color: red'><b>$msg<br>\n" .
  74                                 wfMsgNoDB( "cachederror" ) . "</b></p>\n";
  75
  76                         $tag = "<div id='article'>";
  77                         $text = str_replace(
  78                                 $tag,
  79                                 $tag . $msg,
  80                                 $cache->fetchPageText() );
  81                 }
  82         }
  83
  84         /* Don't cache error pages!  They cause no end of trouble... */
  85         header( "Cache-control: none" );
  86         header( "Pragma: nocache" );
  87         echo $text;
  88         exit;
  89 }
  90
  91 # $db: DB_READ  = -1    read from slave (or only server)
  92 #      DB_WRITE = -2    write to master (or only server)
  93 #      0,1,2,...        query a database with a specific index
  94 # Replication is not actually implemented just yet
  95 function wfQuery( $sql, $db, $fname = "" )
  96 {
  97         global $wgLastDatabaseQuery, $wgOut, $wgDebugDumpSql;
  98
  99         # wfGeneralizeSQL will probably cut down the query to reasonable
 100         # logging size most of the time. The substr is really just a sanity check.
 101         $profName = "wfQuery: " . substr( wfGeneralizeSQL( $sql ), 0, 255 );
 102
 103         wfProfileIn( $profName );
 104
 105         if ( !is_numeric( $db ) ) {
 106                 # Someone has tried to call this the old way
 107                 $wgOut->fatalError( wfMsgNoDB( "wrong_wfQuery_params", $db, $sql ) );
 108         }
 109
 110         $wgLastDatabaseQuery = $sql;
 111
 112         if( $wgDebugDumpSql ) {
 113                 $sqlx = substr( $sql, 0, 500 );
 114                 $sqlx = wordwrap(strtr($sqlx,"\t\n","  "));
 115                 wfDebug( "SQL: $sqlx\n" );
 116         }
 117
 118         $conn = wfGetDB();
 119         $ret = mysql_query( $sql, $conn );
 120
 121         if ( false === $ret ) {
 122                 $wgOut->databaseError( $fname );
 123                 exit;
 124         }
 125         wfProfileOut( $profName );
 126         return $ret;
 127 }
 128
 129 function wfFreeResult( $res ) { mysql_free_result( $res ); }
 130 function wfFetchObject( $res ) { return mysql_fetch_object( $res ); }
 131 function wfNumRows( $res ) { return mysql_num_rows( $res ); }
 132 function wfNumFields( $res ) { return mysql_num_fields( $res ); }
 133 function wfFieldName( $res, $n ) { return mysql_field_name( $res, $n ); }
 134 function wfInsertId() { return mysql_insert_id( wfGetDB() ); }
 135 function wfDataSeek( $res, $row ) { return mysql_data_seek( $res, $row ); }
 136 function wfLastErrno() { return mysql_errno(); }
 137 function wfLastError() { return mysql_error(); }
 138 function wfAffectedRows() { return mysql_affected_rows( wfGetDB() ); }
 139
 140 function wfLastDBquery()
 141 {
 142         global $wgLastDatabaseQuery;
 143         return $wgLastDatabaseQuery;
 144 }
 145
 146 function wfSetSQL( $table, $var, $value, $cond )
 147 {
 148         $sql = "UPDATE $table SET $var = '" .
 149           wfStrencode( $value ) . "' WHERE ($cond)";
 150         wfQuery( $sql, DB_WRITE, "wfSetSQL" );
 151 }
 152
 153 function wfGetSQL( $table, $var, $cond )
 154 {
 155         $sql = "SELECT $var FROM $table WHERE ($cond)";
 156         $result = wfQuery( $sql, DB_READ, "wfGetSQL" );
 157
 158         $ret = "";
 159         if ( mysql_num_rows( $result ) > 0 ) {
 160                 $s = mysql_fetch_object( $result );
 161                 $ret = $s->$var;
 162                 mysql_free_result( $result );
 163         }
 164         return $ret;
 165 }
 166
 167 function wfStrencode( $s )
 168 {
 169         return addslashes( $s );
 170 }
 171
 172 # Ideally we'd be using actual time fields in the db
 173 function wfTimestamp2Unix( $ts ) {
 174         return gmmktime( ( (int)substr( $ts, 8, 2) ),
 175                   (int)substr( $ts, 10, 2 ), (int)substr( $ts, 12, 2 ),
 176                   (int)substr( $ts, 4, 2 ), (int)substr( $ts, 6, 2 ),
 177                   (int)substr( $ts, 0, 4 ) );
 178 }
 179
 180 function wfUnix2Timestamp( $unixtime ) {
 181         return gmdate( "YmdHis", $unixtime );
 182 }
 183
 184 function wfTimestampNow() {
 185         # return NOW
 186         return gmdate( "YmdHis" );
 187 }
 188
 189 # Sorting hack for MySQL 3, which doesn't use index sorts for DESC
 190 function wfInvertTimestamp( $ts ) {
 191         return strtr(
 192                 $ts,
 193                 "0123456789",
 194                 "9876543210"
 195         );
 196 }
 197
 198 # Removes most variables from an SQL query and replaces them with X or N for numbers.
 199 # It's only slightly flawed. Don't use for anything important.
 200 function wfGeneralizeSQL( $sql )
 201 {
 202         # This could be done faster with some arrays and a single preg_replace,
 203         # but this show more clearly what's going on.
 204         $sql = preg_replace ( "/-?\d+/" , "N", $sql);
 205         $sql = preg_replace ( "/([\'\"]).+?[^\\\\]\\1/", "\\1X\\1", $sql);
 206         $sql = preg_replace ( "/\s+/", " ", $sql);
 207         return $sql;
 208 }
 209
 210 function wfFieldExists( $table, $field )
 211 {
 212         $fname = "wfFieldExists";
 213         $res = wfQuery( "DESCRIBE $table", DB_READ, $fname );
 214         $found = false;
 215
 216         while ( $row = wfFetchObject( $res ) ) {
 217                 if ( $row->Field == $field ) {
 218                         $found = true;
 219                         break;
 220                 }
 221         }
 222         return $found;
 223 }
 224
 225 function wfIndexExists( $table, $index )
 226 {
 227         global $wgDBname;
 228         $fname = "wfIndexExists";
 229         $sql = "SHOW INDEXES FROM $table";
 230         $res = wfQuery( $sql, DB_READ, $fname );
 231         $found = false;
 232         while ( $row = wfFetchObject( $res ) ) {
 233                 if ( $row->Key_name == $index ) {
 234                         $found = true;
 235                         break;
 236                 }
 237         }
 238         return $found;
 239 }
 240 ?>