PHP Token(令牌)设计(2)

<?php 
/** 
* 原理：请求分配token的时候，想办法分配一个唯一的token, base64( time + rand + action) 
* 如果提交，将这个token记录，说明这个token以经使用，可以跟据它来避免重复提交。 
* 
*/ 
class GToken { 

    /** 
     * 得到当前所有的token 
     * 
     * @return array 
     */ 
    public static function getTokens(){ 
        $tokens = $_SESSION[GConfig::SESSION_KEY_TOKEN ]; 
        if (empty($tokens) && !is_array($tokens)) { 
            $tokens = array(); 
        } 
        return $tokens; 
    } 

    /** 
     * 产生一个新的Token 
     * 
     * @param string $formName 
     * @param 加密密钥 $key 
     * @return string 
     */ 

    public static function granteToken($formName,$key = GConfig::ENCRYPT_KEY ){ 
        $token = GEncrypt::encrypt($formName.":".session_id(),$key); 
        return $token; 
    } 

    /** 
     * 删除token,实际是向session 的一个数组里加入一个元素，说明这个token以经使用过，以避免数据重复提交。 
     * 
     * @param string $token 
     */ 
    public static function dropToken($token){ 
        $tokens = self::getTokens(); 
        $tokens[] = $token; 
        GSession::set(GConfig::SESSION_KEY_TOKEN ,$tokens); 
    } 

    /** 
     * 检查是否为指定的Token 
     * 
     * @param string $token    要检查的token值 
     * @param string $formName 
     * @param boolean $fromCheck 是否检查来路,如果为true,会判断token中附加的session_id是否和当前session_id一至. 
     * @param string $key 加密密钥 
     * @return boolean 
     */ 

    public static function isToken($token,$formName,$fromCheck = false,$key = GConfig::ENCRYPT_KEY){ 
        $tokens = self::getTokens(); 

        if (in_array($token,$tokens)) //如果存在，说明是以使用过的token 
            return false; 

        $source = split(":", GEncrypt::decrypt($token,$key)); 

        if($fromCheck) 
            return $source[1] == session_id() && $source[0] == $formName; 
        else 
            return $source[0] == $formName; 
    } 
} 
?> 

示例:

首先从$_POST里取出token,用isToken判断.

这一切看着似乎是没有问题了.
如果想判断是否是执行的匹配动作,可以把isToken里的formName改一下,运行,很好,没有匹配上.证明这个成功.

是否能避免重复提交,我没有验证,太简单的逻辑了.

余下的就是判断 来路检查 是否正常工作了.
把上面的示例产生的html copy到本地的一个网页内(以达到不同的域的目的),运行,检查来路不明,没有执行动作(需要把isToken的第三个参数设为true).
把isToken的第三个参数设置为false,提交,指定的动作执行了!

好了,到此为止,不知道哪个地方是否还存在BUG,这就要在长期运用中慢慢调试修改了!

您可能感兴趣的文章: