e9e784a09e
Taint check checks for possible security issues by tracking html escaping and more by using phan. This slows done the phan-job a bit and requires more ram Keep the DoubleEscaped issues out to make reviewer easier Adds suppression for false positives Adds taint-annotation to help taint-check Removes suppression for code phan now understand better by the tracking of keys in taint-check Fix some small issues by adding int cast or htmlspecialchars calls Bug: T216348 Bug: T268920 Change-Id: I849ac4f120fd15b483e8939d4db45c98dc351259
111 lines
3.3 KiB
PHP
111 lines
3.3 KiB
PHP
<?php
|
|
/**
|
|
* Test revision text compression and decompression.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* @file
|
|
* @ingroup Maintenance ExternalStorage
|
|
*/
|
|
|
|
use MediaWiki\MediaWikiServices;
|
|
use MediaWiki\Revision\RevisionRecord;
|
|
use MediaWiki\Revision\SlotRecord;
|
|
|
|
$optionsWithArgs = [ 'start', 'limit', 'type' ];
|
|
require __DIR__ . '/../CommandLineInc.php';
|
|
|
|
if ( !isset( $args[0] ) ) {
|
|
echo "Usage: php testCompression.php [--type=<type>] [--start=<start-date>] " .
|
|
"[--limit=<num-revs>] <page-title>\n";
|
|
exit( 1 );
|
|
}
|
|
|
|
$lang = MediaWikiServices::getInstance()->getLanguageFactory()->getLanguage( 'en' );
|
|
$title = Title::newFromText( $args[0] );
|
|
if ( isset( $options['start'] ) ) {
|
|
$start = wfTimestamp( TS_MW, strtotime( $options['start'] ) );
|
|
echo "Starting from " . $lang->timeanddate( $start ) . "\n";
|
|
} else {
|
|
$start = '19700101000000';
|
|
}
|
|
if ( isset( $options['limit'] ) ) {
|
|
$limit = $options['limit'];
|
|
$untilHappy = false;
|
|
} else {
|
|
$limit = 1000;
|
|
$untilHappy = true;
|
|
}
|
|
$type = $options['type'] ?? ConcatenatedGzipHistoryBlob::class;
|
|
|
|
$dbr = wfGetDB( DB_REPLICA );
|
|
$revStore = MediaWikiServices::getInstance()->getRevisionStore();
|
|
$revQuery = $revStore->getQueryInfo( [ 'page' ] );
|
|
$res = $dbr->select(
|
|
$revQuery['tables'],
|
|
$revQuery['fields'],
|
|
[
|
|
'page_namespace' => $title->getNamespace(),
|
|
'page_title' => $title->getDBkey(),
|
|
'rev_timestamp > ' . $dbr->addQuotes( $dbr->timestamp( $start ) ),
|
|
],
|
|
__FILE__,
|
|
[ 'LIMIT' => $limit ],
|
|
$revQuery['joins']
|
|
);
|
|
|
|
$blob = new $type;
|
|
$hashes = [];
|
|
$keys = [];
|
|
$uncompressedSize = 0;
|
|
$t = -microtime( true );
|
|
foreach ( $res as $row ) {
|
|
$revRecord = $revStore->newRevisionFromRow( $row );
|
|
$text = $revRecord->getSlot( SlotRecord::MAIN, RevisionRecord::RAW )
|
|
->getContent()
|
|
->serialize();
|
|
$uncompressedSize += strlen( $text );
|
|
$hashes[$row->rev_id] = md5( $text );
|
|
$keys[$row->rev_id] = $blob->addItem( $text );
|
|
if ( $untilHappy && !$blob->isHappy() ) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
$serialized = serialize( $blob );
|
|
$t += microtime( true );
|
|
# print_r( $blob->mDiffMap );
|
|
|
|
printf( "%s\nCompression ratio for %d revisions: %5.2f, %s -> %d\n",
|
|
$type,
|
|
count( $hashes ),
|
|
$uncompressedSize / strlen( $serialized ),
|
|
$lang->formatSize( $uncompressedSize ),
|
|
strlen( $serialized )
|
|
);
|
|
printf( "Compression time: %5.2f ms\n", $t * 1000 );
|
|
|
|
$t = -microtime( true );
|
|
$blob = unserialize( $serialized );
|
|
foreach ( $keys as $id => $key ) {
|
|
$text = $blob->getItem( $key );
|
|
if ( md5( $text ) != $hashes[$id] ) {
|
|
echo "Content hash mismatch for rev_id $id\n";
|
|
# var_dump( $text );
|
|
}
|
|
}
|
|
$t += microtime( true );
|
|
printf( "Decompression time: %5.2f ms\n", $t * 1000 );
|