<?php

// WARNING: All the URI schemes are far to relaxed, we need to tighten
// the checks.
class HTMLPurifier_URISchemeTest extends HTMLPurifier_URIHarness {
	private $pngBase64;
	public function __construct() {
		$this->pngBase64 = 'iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAABGdBTUEAALGP' . 'C/xhBQAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB9YGARc5KB0XV+IA' . 'AAAddEVYdENvbW1lbnQAQ3JlYXRlZCB3aXRoIFRoZSBHSU1Q72QlbgAAAF1J' . 'REFUGNO9zL0NglAAxPEfdLTs4BZM4DIO4C7OwQg2JoQ9LE1exdlYvBBeZ7jq' . 'ch9//q1uH4TLzw4d6+ErXMMcXuHWxId3KOETnnXXV6MJpcq2MLaI97CER3N0' . 'vr4MkhoXe0rZigAAAABJRU5ErkJggg==';
	}
	protected function assertValidation($uri, $expect_uri = true) {
		$this->prepareURI ( $uri, $expect_uri );
		$this->config->set ( 'URI.AllowedSchemes', array (
				$uri->scheme 
		) );
		// convenience hack: the scheme should be explicitly specified
		$scheme = $uri->getSchemeObj ( $this->config, $this->context );
		$result = $scheme->validate ( $uri, $this->config, $this->context );
		$this->assertEitherFailOrIdentical ( $result, $uri, $expect_uri );
	}
	public function test_http_regular() {
		$this->assertValidation ( 'http://example.com/?s=q#fragment' );
	}
	public function test_http_uppercase() {
		$this->assertValidation ( 'http://example.com/FOO' );
	}
	public function test_http_removeDefaultPort() {
		$this->assertValidation ( 'http://example.com:80', 'http://example.com' );
	}
	public function test_http_removeUserInfo() {
		$this->assertValidation ( 'http://bob@example.com', 'http://example.com' );
	}
	public function test_http_preserveNonDefaultPort() {
		$this->assertValidation ( 'http://example.com:8080' );
	}
	public function test_https_regular() {
		$this->assertValidation ( 'https://user@example.com:443/?s=q#frag', 'https://example.com/?s=q#frag' );
	}
	public function test_ftp_regular() {
		$this->assertValidation ( 'ftp://user@example.com/path' );
	}
	public function test_ftp_removeDefaultPort() {
		$this->assertValidation ( 'ftp://example.com:21', 'ftp://example.com' );
	}
	public function test_ftp_removeQueryString() {
		$this->assertValidation ( 'ftp://example.com?s=q', 'ftp://example.com' );
	}
	public function test_ftp_preserveValidTypecode() {
		$this->assertValidation ( 'ftp://example.com/file.txt;type=a' );
	}
	public function test_ftp_removeInvalidTypecode() {
		$this->assertValidation ( 'ftp://example.com/file.txt;type=z', 'ftp://example.com/file.txt' );
	}
	public function test_ftp_encodeExtraSemicolons() {
		$this->assertValidation ( 'ftp://example.com/too;many;semicolons=1', 'ftp://example.com/too%3Bmany%3Bsemicolons=1' );
	}
	public function test_news_regular() {
		$this->assertValidation ( 'news:gmane.science.linguistics' );
	}
	public function test_news_explicit() {
		$this->assertValidation ( 'news:642@eagle.ATT.COM' );
	}
	public function test_news_removeNonPathComponents() {
		$this->assertValidation ( 'news://user@example.com:80/rec.music?path=foo#frag', 'news:/rec.music#frag' );
	}
	public function test_nntp_regular() {
		$this->assertValidation ( 'nntp://news.example.com/alt.misc/42#frag' );
	}
	public function test_nntp_removalOfRedundantOrUselessComponents() {
		$this->assertValidation ( 'nntp://user@news.example.com:119/alt.misc/42?s=q#frag', 'nntp://news.example.com/alt.misc/42#frag' );
	}
	public function test_mailto_regular() {
		$this->assertValidation ( 'mailto:bob@example.com' );
	}
	public function test_mailto_removalOfRedundantOrUselessComponents() {
		$this->assertValidation ( 'mailto://user@example.com:80/bob@example.com?subject=Foo#frag', 'mailto:/bob@example.com?subject=Foo#frag' );
	}
	public function test_data_png() {
		$this->assertValidation ( 'data:image/png;base64,' . $this->pngBase64 );
	}
	public function test_data_malformed() {
		$this->assertValidation ( 'data:image/png;base64,vr4MkhoXJRU5ErkJggg==', false );
	}
	public function test_data_implicit() {
		$this->assertValidation ( 'data:base64,' . $this->pngBase64, 'data:image/png;base64,' . $this->pngBase64 );
	}
	public function test_file_basic() {
		$this->assertValidation ( 'file://user@MYCOMPUTER:12/foo/bar?baz#frag', 'file://MYCOMPUTER/foo/bar#frag' );
	}
	public function test_file() {
		$this->assertValidation ( 'file:///foo/bar?baz#frag', 'file:///foo/bar#frag' );
	}
	public function test_ftp_empty_host() {
		$this->assertValidation ( 'ftp:///example.com', false );
	}
}

// vim: et sw=4 sts=4