Extension:DatabaseFetchHook.php

<?php

1. Extension:DatabaseFetchObjectTemplate:Php

These are the MediaWiki extensions we're using and/or developing. Please refer to the information on the mediawiki.org wiki for installation and usage details. Extensions here which have no corresponding mediawiki article are either not ready for use or have been superseded. You can also browse our extension code in our local Subversion repository or our GitHub mirror.

1. - See http://www.mediawiki.org/wiki/Extension:DatabaseFetchObject for installation and usage details
2. - Started: 2007-03-05
3. - Licenced under LGPL (http://www.gnu.org/copyleft/lesser.html)

if (!defined('MEDIAWIKI')) die('Not an entry point.');

define('DFO_VERSION','1.0.2, 2007-05-19');

$wgExtensionCredits['parserhook'][] = array( 'name' => "DatabaseFetchObject", 'author' => 'User:Nad', 'description' => 'Dynamically adds a new hook to intercept database row requests', 'url' => 'http://www.mediawiki.org/wiki/Extension:DatabaseFetchObject', 'version' => DFO_VERSION );

1. Create a new secureLoadBalancer class by extending the existing one

$LoadBalancer = get_class($wgLoadBalancer); $secureLoadBalancer = "secure$LoadBalancer"; eval('class '.$secureLoadBalancer.' extends '.$LoadBalancer.' {

# Override the getConnection method to return a secureDatabase connection function &getConnection($i, $fail = true, $groups = array()) { $db =& parent::getConnection($i,$fail,$groups);

# Create a secure subclass of the database class with extended fetchObject method $dbClass = get_class($db); $dbSecure = "secure$dbClass"; if (!class_exists($dbSecure)) { eval(\'class \'.$dbSecure.\' extends \'.$dbClass.\' { function fetchObject($res) { $row = parent::fetchObject($res); wfRunHooks("DatabaseFetchObject", array(&$this,&$row)); return $row; } }\'); }

# Replace the DatabaseXXX connection with an identical secureDatabaseXXX connection $sdb = new $dbSecure; foreach(array_keys(get_class_vars($dbClass)) as $k) $sdb->$k = $db->$k; unset($db);

return $sdb; }

}');

1. Replace the global LoadBalancer object with an identical secureLoadBalancer

$oldLoadBalancer = $wgLoadBalancer; $wgLoadBalancer = new $secureLoadBalancer($oldLoadBalancer->mServers); foreach(array_keys(get_class_vars($LoadBalancer)) as $k) $wgLoadBalancer->$k = $oldLoadBalancer->$k;

1. Test the hook

$wgHooks['DatabaseFetchObject'][] = 'hookDatabaseFetchObject'; function hookDatabaseFetchObject(&$row) { if (is_object($row)) msg(join(',',array_keys(get_object_vars($row)))); return true; } ?>