|
|
@@ -0,0 +1,170 @@
|
|
|
+<?php
|
|
|
+/**
|
|
|
+ * Zend Framework
|
|
|
+ *
|
|
|
+ * LICENSE
|
|
|
+ *
|
|
|
+ * This source file is subject to the new BSD license that is bundled
|
|
|
+ * with this package in the file LICENSE.txt.
|
|
|
+ * It is also available through the world-wide-web at this URL:
|
|
|
+ * http://framework.zend.com/license/new-bsd
|
|
|
+ * If you did not receive a copy of the license and are unable to
|
|
|
+ * obtain it through the world-wide-web, please send an email
|
|
|
+ * to license@zend.com so we can send you a copy immediately.
|
|
|
+ *
|
|
|
+ * @category Zend
|
|
|
+ * @package Zend_Xml_Security
|
|
|
+ * @subpackage UnitTests
|
|
|
+ * @copyright Copyright (c) 2005-2014 Zend Technologies USA Inc. (http://www.zend.com)
|
|
|
+ * @license http://framework.zend.com/license/new-bsd New BSD License
|
|
|
+ * @version $Id$
|
|
|
+ */
|
|
|
+
|
|
|
+if (!defined('PHPUnit_MAIN_METHOD')) {
|
|
|
+ define('PHPUnit_MAIN_METHOD', 'Zend_Xml_SecurityTest::main');
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * @see Zend_Xml_Security
|
|
|
+ */
|
|
|
+require_once 'Zend/Xml/Security.php';
|
|
|
+
|
|
|
+require_once 'Zend/Xml/Exception.php';
|
|
|
+
|
|
|
+/**
|
|
|
+ * @category Zend
|
|
|
+ * @package Zend_Xml_Security
|
|
|
+ * @subpackage UnitTests
|
|
|
+ * @copyright Copyright (c) 2005-2014 Zend Technologies USA Inc. (http://www.zend.com)
|
|
|
+ * @license http://framework.zend.com/license/new-bsd New BSD License
|
|
|
+ * @group Zend_Xml
|
|
|
+ */
|
|
|
+class Zend_Xml_SecurityTest extends PHPUnit_Framework_TestCase
|
|
|
+{
|
|
|
+ public static function main()
|
|
|
+ {
|
|
|
+ $suite = new PHPUnit_Framework_TestSuite(__CLASS__);
|
|
|
+ $result = PHPUnit_TextUI_TestRunner::run($suite);
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanForXEE()
|
|
|
+ {
|
|
|
+ $xml = <<<XML
|
|
|
+<?xml version="1.0"?>
|
|
|
+<!DOCTYPE results [<!ENTITY harmless "completely harmless">]>
|
|
|
+<results>
|
|
|
+ <result>This result is &harmless;</result>
|
|
|
+</results>
|
|
|
+XML;
|
|
|
+
|
|
|
+ $this->setExpectedException('Zend_Xml_Exception');
|
|
|
+ $result = Zend_Xml_Security::scan($xml);
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanForXXE()
|
|
|
+ {
|
|
|
+ $file = tempnam(sys_get_temp_dir(), 'Zend_XML_Security');
|
|
|
+ file_put_contents($file, 'This is a remote content!');
|
|
|
+ $xml = <<<XML
|
|
|
+<?xml version="1.0"?>
|
|
|
+<!DOCTYPE root
|
|
|
+[
|
|
|
+<!ENTITY foo SYSTEM "file://$file">
|
|
|
+]>
|
|
|
+<results>
|
|
|
+ <result>&foo;</result>
|
|
|
+</results>
|
|
|
+XML;
|
|
|
+
|
|
|
+ try {
|
|
|
+ $result = Zend_Xml_Security::scan($xml);
|
|
|
+ } catch (Zend_Xml_Exception $e) {
|
|
|
+ unlink($file);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ $this->fail('An expected exception has not been raised.');
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanSimpleXmlResult()
|
|
|
+ {
|
|
|
+ $result = Zend_Xml_Security::scan($this->_getXml());
|
|
|
+ $this->assertTrue($result instanceof SimpleXMLElement);
|
|
|
+ $this->assertEquals($result->result, 'test');
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanDom()
|
|
|
+ {
|
|
|
+ $dom = new DOMDocument('1.0');
|
|
|
+ $result = Zend_Xml_Security::scan($this->_getXml(), $dom);
|
|
|
+ $this->assertTrue($result instanceof DOMDocument);
|
|
|
+ $node = $result->getElementsByTagName('result')->item(0);
|
|
|
+ $this->assertEquals($node->nodeValue, 'test');
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanInvalidXml()
|
|
|
+ {
|
|
|
+ $xml = <<<XML
|
|
|
+<foo>test</bar>
|
|
|
+XML;
|
|
|
+
|
|
|
+ $result = Zend_XML_Security::scan($xml);
|
|
|
+ $this->assertFalse($result);
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanInvalidXmlDom()
|
|
|
+ {
|
|
|
+ $xml = <<<XML
|
|
|
+<foo>test</bar>
|
|
|
+XML;
|
|
|
+
|
|
|
+ $dom = new DOMDocument('1.0');
|
|
|
+ $result = Zend_XML_Security::scan($xml, $dom);
|
|
|
+ $this->assertFalse($result);
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanFile()
|
|
|
+ {
|
|
|
+ $file = tempnam(sys_get_temp_dir(), 'Zend_XML_Security');
|
|
|
+ file_put_contents($file, $this->_getXml());
|
|
|
+
|
|
|
+ $result = Zend_Xml_Security::scanFile($file);
|
|
|
+ $this->assertTrue($result instanceof SimpleXMLElement);
|
|
|
+ $this->assertEquals($result->result, 'test');
|
|
|
+ unlink($file);
|
|
|
+ }
|
|
|
+
|
|
|
+ public function testScanXmlWithDTD()
|
|
|
+ {
|
|
|
+ $xml = <<<XML
|
|
|
+<?xml version="1.0"?>
|
|
|
+<!DOCTYPE results [
|
|
|
+<!ELEMENT results (result+)>
|
|
|
+<!ELEMENT result (#PCDATA)>
|
|
|
+]>
|
|
|
+<results>
|
|
|
+ <result>test</result>
|
|
|
+</results>
|
|
|
+XML;
|
|
|
+
|
|
|
+ $dom = new DOMDocument('1.0');
|
|
|
+ $result = Zend_Xml_Security::scan($xml, $dom);
|
|
|
+ $this->assertTrue($result instanceof DOMDocument);
|
|
|
+ $this->assertTrue($result->validate());
|
|
|
+ }
|
|
|
+
|
|
|
+ protected function _getXml()
|
|
|
+ {
|
|
|
+ return <<<XML
|
|
|
+<?xml version="1.0"?>
|
|
|
+<results>
|
|
|
+ <result>test</result>
|
|
|
+</results>
|
|
|
+XML;
|
|
|
+
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+if (PHPUnit_MAIN_METHOD == "Zend_Xml_SecurityTest::main") {
|
|
|
+ Zend_Xml_SecurityTest::main();
|
|
|
+}
|