一个过滤超文本元素，并且允许指定的元素通过的Class

复制PHP内容到剪贴板

PHP代码:

<?php
/**
 * @copyright 2006 ajss
 * @author 徐智 <[[email protected]][email protected][/email]>
 * 
 * $Id: myfilterHtml.class.php 667 2007-07-03 04:48:47Z xuzhi $ 
 */

/**
 * 测试，使用
 */
$value = "<a href='a'>";
$obj =  new myfilterHtml(
    trim($value), 
    array(
        'center'    => array('center'   => 0),
        'br'        => array('br'   => 0),
        'font'      => array('font' => 0, 'size'  => 0, 'style' => 0),
        'a'         => array('a'    => 0, 'href'  => 0, 'title' => 0),
        'img'       => array('img'  => 0, 'src'   => array('javascript'), 'border' => 0, 'width' => 0, 'height' => 0, 'class' => 0, 'alt' => 0, 'title' => 0)
    )
);

$value = $obj->filterHtml();

/**
 * 过滤超文本元素，并且允许指定的元素通过
 *
 * myfilterHtml
 */
class myfilterHtml {
    const space = ' ';
    const quotes = '"';
    const quote = "'";
    const equ = "=";

    private $html = '';
    /**
     * array(
     *  'font' => array('style' => 0, 'size' => 0),
     *  )
     */
    private $allow = array();
    private $pos = 0;
    private $buf = '';
    private $ed = '';
    private $htmlStack = array();
    private $stackPos = 0;
    private $tagNum = 0;
    private $htmlTags = array();

    /**
     * 构造一个允许指定元素通过的Html过滤器
     *
     * @param   $html   string   需要过滤的文本
     * @param   $allow  array    允许通过的Html元素，key:允许的元素,Value:数组,允许的属性值,如果属性Key对应的值为一个数组,那么数组中可以指定,元素中不允许出现的关键字
     *          'img'       => array('img'  => 0, 'src'   => array('javascript'), 'border' => 0, 'width' => 0, 'height' => 0, 'class' => 0, 'alt' => 0, 'title' => 0)
     */
    public function __construct($html, $allow, $ed='utf-8') {
        $this->html = $html;
        $this->allow = $allow;
        $this->ed = $ed;
    }

    public function filterHtml() {
        for ($i = 0; $i < 10; $i++) {
            if ($this->nextHtmlTag() === false) break;
        }

        $htmlTags = $this->htmlTags;
        $html = &$this->html;
        $ed = &$this->ed;
        $p = 0;
        $buf = '';
        foreach ($htmlTags as $v) {
            if ($v[3] === true) {
                $buf .= htmlentities(mb_substr($html, $p, $v[1] - $p, $ed), ENT_QUOTES, $ed);
                $buf .= mb_substr($html, $v[1], $v[2] - $v[1] + 1, $ed);
                $p = $v[2] + 1;
            }
        }

        return $buf . htmlentities(mb_substr($html, $p, mb_strlen($html, $ed) - $p, $ed), ENT_QUOTES, $ed);
    }

    private function nextHtmlTag() {
        $html = &$this->html;
        $ed = &$this->ed;
        $p = &$this->pos;

        //找到html标签
        $spos = mb_strpos($html, '<', $p, $ed);
        if ($spos === false) return false;
        $p = $spos;

        $epos = mb_strpos($html, '>', $spos, $ed);
        if ($epos === false) return false;
        $p = $epos;

        if ($this->pushHtml($spos, $epos) === false) return false;
        
        return true;
    }

    private function pushHtml($s, $e) {
        $html = &$this->html;
        $ed = &$this->ed;
        $spos = $s + 1;

        //检测是否是html结束标记
        if (mb_substr($html, $spos, 1, $ed) == '/') {
            //结束标记
            $htmlTag = array(
                0 => mb_substr($html, $spos + 1, $e - $spos - 1, $ed),
                1 => $s,
                2 => $e,
                3 => false,
            );
            if ($this->stackPos > 0) {
                $stack = &$this->htmlStack[--$this->stackPos];
                if ($stack[0] == $htmlTag[0]) {
                    $this->htmlTags[] = &$htmlTag;
                    $htmlTag[3] = true;
                    $stack[3] = true;
                } else {
                    $this->stackPos++;
                }
            }
        } else {
            //开始标记
            $pos = mb_strpos($html, ' ', $spos, $ed);
            $endTag = mb_substr($html, $e - 1, 1, $ed);
            if ($pos > $e || $pos === false) {
                $pos = $e;
                if ($endTag == '/') $pos--;
            }
            $pos--;

            $htmlTag = array(
                0 => mb_substr($html, $spos, $pos - $spos + 1, $ed),
                1 => $s,
                2 => $e,
                3 => false,
            );

            //过滤非法的html标记
            if (true && $this->checkHtml($htmlTag)) {
                $this->htmlStack[$this->stackPos++] = &$htmlTag;
                $this->htmlTags[] = &$htmlTag;

                //关闭标记
                if ($endTag == '/') {
                    $htmlTag[3] = true; 
                    $this->stackPos--;
                }
            }
        }

        return true;
    }

    private function checkHtml(&$htmlTag) {
        //TODO: 考虑使用更优的解决方案
        $ed = &$this->ed;
        $allow = &$this->allow;
        if (!isset($allow[$htmlTag[0]])) return false;
        $allow = &$allow[$htmlTag[0]];
        $buf = mb_substr($this->html, $htmlTag[1] + 1, $htmlTag[2] - $htmlTag[1] - 1, $ed);

        $len = strlen($buf);
        $pos = -1;
        $tag = '';
        $att = '';
        for ($i = 0; $i < $len; $i++) {
            switch ($buf{$i}) {
                case self::space: 
                    if ($tag == self::quote || $tag == self::quotes || $tag == self::equ || $tag == self::space) break; 
                    if ($pos >= $i - 1) break;
                    $pos = $i;
                    $tag = self::space;
                    break;
                case self::quote: 
                    break;
                    if ($att == '') return false;
                    if ($tag == self::quotes) break;
                    if ($tag == self::quote) {
                        if ($pos >= $i - 1) break;
                        //过滤非法属性值
                        if ($allow[$att] !== 0) {
                            $attValue = trim(substr($buf, $pos + 1, $i - $pos - 1));
                            foreach ($allow[$att] as $v) {
                                if (stripos($attValue, $v) !== false) return false;
                            }
                        }
                        $att = '';
                        $pos = $i - 1;
                        $tag = '';
                    } else {
                        $tag = self::quote;
                    }
                    break; 
                case self::quotes:
                    if ($att == '') return false;
                    if ($tag == self::quote) break;
                    if ($tag == self::quotes) {
                        if ($pos >= $i - 1) break;
                        //过滤非法属性值
                        if ($allow[$att] !== 0) {
                            $attValue = trim(substr($buf, $pos + 1, $i - $pos - 1));
                            foreach ($allow[$att] as $v) {
                                if (stripos($attValue, $v) !== false) return false;
                            }
                        }
                        $att = '';
                        $pos = $i - 1;
                        $tag = '';
                    } else {
                        $tag = self::quotes;
                    }
                    break;
                case self::equ:
                    if ($pos >= $i - 1) return false;
                    if ($tag == self::equ) return false;
                    if ($tag == self::space) {
                        $att = trim(substr($buf, $pos + 1, $i - $pos - 1));
                        if (!isset($allow[$att])) return false;
                    }
                    $tag = self::equ;
                    $pos = $i;
                    break;
            default:
                ;
            }
        }

        return true;
    }
  

}
?>

引用:

原帖由 luzhou 于 2008-1-19 08:58 发表
应鼓励原创精品。
另，不要离开PHP！