A differential fuzz tester for the wikitext preprocessor. My changes don't pass it...

diff --git a/maintenance/preprocessorFuzzTest.php b/maintenance/preprocessorFuzzTest.php
new file mode 100644 (file)
index 0000000..5a029f6
--- /dev/null
+++ b/maintenance/preprocessorFuzzTest.php
@@ -0,0 +1,156 @@
+<?php
+
+require_once( dirname( __FILE__ ). '/../maintenance/commandLine.inc' );
+
+$wgHooks['BeforeParserFetchTemplateAndtitle'][] = 'PPFuzzTester::templateHook';
+
+class PPFuzzTester {
+       var $hairs = array(
+               '[[', ']]', '{{', '}}', '{{{', '}}}', 
+               '<', '>', '<nowiki', '<gallery', '</nowiki>', '</gallery>', '<nOwIkI>', '</NoWiKi>',
+               //'<!--' , '-->',
+               //'<ref>', '</ref>', '<references/>',
+               "\n==", "==\n",
+               '|', '=', "\n", ' ', "\t", "\x7f",
+               'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 
+               'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't',
+       );
+       var $minLength = 0;
+       var $maxLength = 20;
+       var $maxTemplates = 5;
+       var $outputTypes = array( 'OT_HTML', 'OT_WIKI', 'OT_MSG', 'OT_PREPROCESS' );
+       static $currentTest = false;
+
+       function execute() {
+               if ( !file_exists( 'results' ) ) {
+                       mkdir( 'results' );
+               }
+               if ( !is_dir( 'results' ) ) {
+                       echo "Unable to create 'results' directory\n";
+                       exit( 1 );
+               }
+               for ( $i = 0; true; $i++ ) {
+                       try {
+                               self::$currentTest = new PPFuzzTest( $this );
+                               self::$currentTest->execute();
+                       } catch ( MWException $e ) {
+                               $testReport = self::$currentTest->getReport();
+                               $exceptionReport = $e->getText();
+                               $hash = md5( $testReport );
+                               file_put_contents( "results/ppft-$hash.in", serialize( self::$currentTest ) );
+                               file_put_contents( "results/ppft-$hash.fail", 
+                                       "Input:\n$testReport\n\nException report:\n$exceptionReport\n" );
+                               print "Test $hash failed\n";
+                       }
+                       if ( $i % 1000 == 0 ) {
+                               print "$i tests done\n";
+                               /*
+                               $testReport = self::$currentTest->getReport();
+                               $filename = 'results/ppft-' . md5( $testReport ) . '.pass';
+                               file_put_contents( $filename, "Input:\n$testReport\n" );*/
+                       }
+               }
+       }
+
+       function makeInputText() {
+               $length = mt_rand( $this->minLength, $this->maxLength );
+               $s = '';
+               for ( $i = 0; $i < $length; $i++ ) {
+                       $hairIndex = mt_rand( 0, count( $this->hairs ) - 1 );
+                       $s .= $this->hairs[$hairIndex];
+               }
+               // Send through the UTF-8 normaliser
+               // This resolves a few differences between the old preprocessor and the 
+               // XML-based one, which doesn't like illegals and converts line endings.
+               // It's done by the MW UI, so it's a reasonably legitimate thing to do.
+               $s = UtfNormal::cleanUp( $s );
+               return $s;
+       }
+
+       function makeTitle() {
+               return Title::newFromText( mt_rand( 0, 1000000 ), mt_rand( 0, 10 ) );
+       }
+
+       function pickOutputType() {
+               $count = count( $this->outputTypes );
+               return $this->outputTypes[ mt_rand( 0, $count - 1 ) ];
+       }
+}
+
+class PPFuzzTest {
+       var $templates, $mainText, $title;
+
+       function __construct( $tester ) {
+               $this->parent = $tester;
+               $this->mainText = $tester->makeInputText();
+               $this->title = $tester->makeTitle();
+               $this->outputType = $tester->pickOutputType();
+               $this->templates = array();
+       }
+
+       function templateHook( $title ) {
+               $titleText = $title->getPrefixedDBkey();
+                               
+               if ( !isset( $this->templates[$titleText] ) ) {
+                       $finalTitle = $title;
+                       if ( count( $this->templates ) >= $this->parent->maxTemplates ) {
+                               // Too many templates
+                               $text = false;
+                       } else {
+                               if ( !mt_rand( 0, 1 ) ) {
+                                       // Redirect
+                                       $finalTitle = $this->parent->makeTitle();
+                               }
+                               if ( !mt_rand( 0, 5 ) ) {
+                                       // Doesn't exist
+                                       $text = false;
+                               } else {
+                                       $text = $this->parent->makeInputText();
+                               }
+                       }
+                       $this->templates[$titleText] = array(
+                               'text' => $text,
+                               'finalTitle' => $finalTitle );
+               }
+               return $this->templates[$titleText];
+       }
+
+       function execute() {
+               global $wgParser;
+               $options = new ParserOptions;
+               $options->setTemplateCallback( array( $this, 'templateHook' ) );
+               $wgParser->startExternalParse( $this->title, $options, constant( $this->outputType ) );
+               return $wgParser->srvus( $this->mainText );
+       }
+
+       function getReport() {
+               $s = "Title: " . $this->title->getPrefixedDBkey() . "\n" .
+                       "Output type: {$this->outputType}\n" . 
+                       "Main text: " . var_export( $this->mainText, true ) . "\n";
+               foreach ( $this->templates as $titleText => $template ) {
+                       $finalTitle = $template['finalTitle'];
+                       if ( $finalTitle != $titleText ) {
+                               $s .= "[[$titleText]] -> [[$finalTitle]]: " . var_export( $template['text'], true ) . "\n";
+                       } else {
+                               $s .= "[[$titleText]]: " . var_export( $template['text'], true ) . "\n";
+                       }
+               }
+               return $s;
+       }
+}
+
+ini_set( 'memory_limit', '50M' );
+if ( isset( $args[0] ) ) {
+       $testText = file_get_contents( $args[0] );
+       if ( !$testText ) {
+               print "File not found\n";
+               exit( 1 );
+       }
+       $test = unserialize( $testText );
+       print $test->getReport();
+       $result = $test->execute();
+       print "Test passed.\nResult: $result\n";
+} else {
+       $tester = new PPFuzzTester;
+       $tester->execute();
+}