php 필터 HTML 태그 속성 클래스

php 필터 HTML 태그 속성 클래스

HtmlAttributeFilter.class.php

<?php
/** HTML Attribute Filter
*   Date:   2013-09-22
*   Author: fdipzone
*   ver:    1.0
*
*   Func:
*   public  strip              필터 속성
*   public  setAllow           허용된 속성 설정
*   public  setException       특례를 설정하다
*   public  setIgnore          무시하기 위한 태그 설정
*   private findElements       처리할 요소 찾기
*   private findAttributes     속성 찾기
*   private removeAttributes   속성 제거
*   private isException        특례 여부를 판단하다
*   private createAttributes   속성 만들기
*   private protect            특수 문자 이스케이프
*/
 
class HtmlAttributeFilter{ // class start
 
    private $_str = '';            // 원본 문자열
    private $_allow = array();     // 보존할 수 있는 속성( 예)array('id','class','title')
    private $_exception = array(); // 특례 예를 들어:array('a'=>array('href','class'),'span'=>array('class'))
    private $_ignore = array();    // 약간 필터링된 태그 예를 들어:array('span','img')
 
 
    /** HTML 처리, 보존되지 않은 속성 필터링
    * @param  String $str 원본 문자열
    * @return String
    */
    public function strip($str){
        $this->_str = $str;
 
        if(is_string($this->_str) && strlen($this->_str)>0){ // 판단 문자열
 
            $this->_str = strtolower($this->_str); // 소문자로 변환
 
            $res = $this->findElements();
            if(is_string($res)){
                return $res;
            }
            $nodes = $this->findAttributes($res);
            $this->removeAttributes($nodes);
        }
 
        return $this->_str;
    }
 
 
    /** 허용된 속성 설정
    * @param Array $param
    */
    public function setAllow($param=array()){
        $this->_allow = $param;
    }
 
 
    /** 특례를 설정하다
    * @param Array $param
    */
    public function setException($param=array()){
        $this->_exception = $param;
    }
 
 
    /** 무시하기 위한 태그 설정
    * @param Array $param
    */
    public function setIgnore($param=array()){
        $this->_ignore = $param;
    }
 
 
    /** 처리할 요소 찾기 */
    private function findElements(){
        $nodes = array();
        preg_match_all("/<([^ !\/\>\n]+)([^>]*)>/i", $this->_str, $elements);
        foreach($elements[1] as $el_key => $element){
            if($elements[2][$el_key]){
                $literal = $elements[0][$el_key];
                $element_name = $elements[1][$el_key];
                $attributes = $elements[2][$el_key];
                if(is_array($this->_ignore) && !in_array($element_name, $this->_ignore)){
                    $nodes[] = array('literal'=>$literal, 'name'=>$element_name, 'attributes'=>$attributes);
                }
            }
        }
 
        if(!$nodes[0]){
            return $this->_str;
        }else{
            return $nodes;
        }
    }
 
 
    /** 속성 찾기
    *  @param Array $nodes 처리해야 할 요소
    */
    private function findAttributes($nodes){
        foreach($nodes as &$node){
            preg_match_all("/([^ =]+)\s*=\s*[\"|']{0,1}([^\"']*)[\"|']{0,1}/i", $node['attributes'], $attributes);
            if($attributes[1]){
                foreach($attributes[1] as $att_key=>$att){
                    $literal = $attributes[0][$att_key];
                    $attribute_name = $attributes[1][$att_key];
                    $value = $attributes[2][$att_key];
                    $atts[] = array('literal'=>$literal, 'name'=>$attribute_name, 'value'=>$value);
                }
            }else{
                $node['attributes'] = null;
            }
            $node['attributes'] = $atts;
            unset($atts);
        }
        return $nodes;
    }
 
 
    /** 속성 제거
    *  @param Array $nodes 처리해야 할 요소
    */
    private function removeAttributes($nodes){
        foreach($nodes as $node){
            $node_name = $node['name'];
            $new_attributes = '';
            if(is_array($node['attributes'])){
                foreach($node['attributes'] as $attribute){
                    if((is_array($this->_allow) && in_array($attribute['name'], $this->_allow)) || $this->isException($node_name, $attribute['name'], $this->_exception)){
                        $new_attributes = $this->createAttributes($new_attributes, $attribute['name'], $attribute['value']);
                    }
                }
            }
            $replacement = ($new_attributes) ? "<$node_name $new_attributes>" : "<$node_name>";
            $this->_str = preg_replace('/'.$this->protect($node['literal']).'/', $replacement, $this->_str);
        }
    }
 
 
    /** 특례 여부를 판단하다
    * @param String $element_name   원소명
    * @param String $attribute_name 속성 이름
    * @param Array  $exceptions     허용된 특례
    * @return boolean
    */
    private function isException($element_name, $attribute_name, $exceptions){
        if(array_key_exists($element_name, $this->_exception)){
            if(in_array($attribute_name, $this->_exception[$element_name])){
                return true;
            }
        }
        return false;
    }
 
 
    /** 속성 만들기
    * @param  String $new_attributes
    * @param  String $name
    * @param  String $value
    * @return String
    */
    private function createAttributes($new_attributes, $name, $value){
        if($new_attributes){
            $new_attributes .= " ";
        }
        $new_attributes .= "$name=\"$value\"";
        return $new_attributes;
    }
 
 
    /** 특수 문자 이스케이프
    * @param  String $str 원본 문자열
    * @return String
    */
    private function protect($str){
        $conversions = array(
            "^" => "\^", 
            "[" => "\[", 
            "." => "\.", 
            "$" => "\$", 
            "{" => "\{", 
            "*" => "\*", 
            "(" => "\(", 
            "\\" => "\\\\", 
            "/" => "\/", 
            "+" => "\+", 
            ")" => "\)", 
            "|" => "\|", 
            "?" => "\?", 
            "<" => "\<", 
            ">" => "\>" 
        );
        return strtr($str, $conversions);
    }
 
} // class end
 
?>

demo

<?php
require('HtmlAttributeFilter.class.php');
 
$str = '<div class="bd clearfix" id="index_hilite_ul"><ul class="list"><li><img src="http://su.bdimg.com/static/skin/img/logo_white.png" width="118" height="148"><div class="cover"><a class="text" href="http://www.csdn.net"><strong>yuna</strong><p>love</p></a><strong class="t g">want to know</strong><a href="/login.html" class="ppBtn"><strong class="text">YES</strong></a></div></li></ul></div>';
 
$obj = new HtmlAttributeFilter();
 
// ID 속성 허용하기
$obj->setAllow(array('id'));
 
$obj->setException(array(
                    'a' => array('href'),   // a 태그는 href 속성에 대한 예외를 허용합니다
                    'ul' => array('class')  // ul 태그에 class 속성 예외 허용
));
 
// img 탭 무시, 속성 필터링
$obj->setIgnore(array('img'));
 
echo 'source str:<br>';
echo htmlspecialchars($str).'<br><br>';
echo 'filter str:<br>';
echo htmlspecialchars($obj->strip($str));
?>