preprocessorFuzzTest.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/commandLine.inc';

$wgHooks['BeforeParserFetchTemplateAndtitle'][] = 'PPFuzzTester::templateHook';

class PPFuzzTester {
    public $hairs = array(
        '[[', ']]', '{{', '{{', '}}', '}}', '{{{', '}}}',
        '<', '>', '<nowiki', '<gallery', '</nowiki>', '</gallery>', '<nOwIkI>', '</NoWiKi>',
        '<!--', '-->',
        "\n==", "==\n",
        '|', '=', "\n", ' ', "\t", "\x7f",
        '~~', '~~~', '~~~~', 'subst:',
        'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j',
        'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't',

        // extensions
        // '<ref>', '</ref>', '<references/>',
    );
    public $minLength = 0;
    public $maxLength = 20;
    public $maxTemplates = 5;
    // public $outputTypes = array( 'OT_HTML', 'OT_WIKI', 'OT_PREPROCESS' );
    public $entryPoints = array( 'testSrvus', 'testPst', 'testPreprocess' );
    public $verbose = false;

    private static $currentTest = false;

    function execute() {
        if ( !file_exists( 'results' ) ) {
            mkdir( 'results' );
        }
        if ( !is_dir( 'results' ) ) {
            echo "Unable to create 'results' directory\n";
            exit( 1 );
        }
        $overallStart = microtime( true );
        $reportInterval = 1000;
        for ( $i = 1; true; $i++ ) {
            $t = -microtime( true );
            try {
                self::$currentTest = new PPFuzzTest( $this );
                self::$currentTest->execute();
                $passed = 'passed';
            } 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";
                $passed = 'failed';
            }
            $t += microtime( true );

            if ( $this->verbose ) {
                printf( "Test $passed in %.3f seconds\n", $t );
                print self::$currentTest->getReport();
            }

            $reportMetric = ( microtime( true ) - $overallStart ) / $i * $reportInterval;
            if ( $reportMetric > 25 ) {
                if ( substr( $reportInterval, 0, 1 ) === '1' ) {
                    $reportInterval /= 2;
                } else {
                    $reportInterval /= 5;
                }
            } elseif ( $reportMetric < 4 ) {
                if ( substr( $reportInterval, 0, 1 ) === '1' ) {
                    $reportInterval *= 5;
                } else {
                    $reportInterval *= 2;
                }
            }
            if ( $i % $reportInterval == 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( $max = false ) {
        if ( $max === false ) {
            $max = $this->maxLength;
        }
        $length = mt_rand( $this->minLength, $max );
        $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.
        global $wgContLang;
        $s = $wgContLang->normalize( $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 ) ];
    }*/

    function pickEntryPoint() {
        $count = count( $this->entryPoints );

        return $this->entryPoints[mt_rand( 0, $count - 1 )];
    }
}

class PPFuzzTest {
    public $templates, $mainText, $title, $entryPoint, $output;

    function __construct( $tester ) {
        global $wgMaxSigChars;
        $this->parent = $tester;
        $this->mainText = $tester->makeInputText();
        $this->title = $tester->makeTitle();
        // $this->outputType = $tester->pickOutputType();
        $this->entryPoint = $tester->pickEntryPoint();
        $this->nickname = $tester->makeInputText( $wgMaxSigChars + 10 );
        $this->fancySig = (bool)mt_rand( 0, 1 );
        $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, $wgUser;

        $wgUser = new PPFuzzUser;
        $wgUser->mName = 'Fuzz';
        $wgUser->mFrom = 'name';
        $wgUser->ppfz_test = $this;

        $options = ParserOptions::newFromUser( $wgUser );
        $options->setTemplateCallback( array( $this, 'templateHook' ) );
        $options->setTimestamp( wfTimestampNow() );
        $this->output = call_user_func(
            array( $wgParser, $this->entryPoint ),
            $this->mainText,
            $this->title,
            $options
        );

        return $this->output;
    }

    function getReport() {
        $s = "Title: " . $this->title->getPrefixedDBkey() . "\n" .
//          "Output type: {$this->outputType}\n" .
            "Entry point: {$this->entryPoint}\n" .
            "User: " . ( $this->fancySig ? 'fancy' : 'no-fancy' ) .
            ' ' . var_export( $this->nickname, true ) . "\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";
            }
        }
        $s .= "Output: " . var_export( $this->output, true ) . "\n";

        return $s;
    }
}

class PPFuzzUser extends User {
    public $ppfz_test, $mDataLoaded;

    function load() {
        if ( $this->mDataLoaded ) {
            return;
        }
        $this->mDataLoaded = true;
        $this->loadDefaults( $this->mName );
    }

    function getOption( $oname, $defaultOverride = null, $ignoreHidden = false ) {
        if ( $oname === 'fancysig' ) {
            return $this->ppfz_test->fancySig;
        } elseif ( $oname === 'nickname' ) {
            return $this->ppfz_test->nickname;
        } else {
            return parent::getOption( $oname, $defaultOverride, $ignoreHidden );
        }
    }
}

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 );
    $result = $test->execute();
    print "Test passed.\n";
} else {
    $tester = new PPFuzzTester;
    $tester->verbose = isset( $options['verbose'] );
    $tester->execute();
}