修改ThinkPHP的驗證碼類

時間 2019-12-08
原文原文鏈接
　　今天用ThinkPHP從新開發一個系統，用到了ThinkPHP的驗證碼類，因爲我但願驗證碼別太複雜，但願驗證碼裏邊只有數字，卻發現該Verify類並未提供設置驗證碼中使用的字符的配置的方法，因而查看源碼，以爲這個類功能仍是不夠齊全，因此修改了一點點，讓用戶能夠配置驗證碼中使用的字符的範圍。php
  1 <?php
  2 // +----------------------------------------------------------------------
  3 // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
  4 // +----------------------------------------------------------------------
  5 // | Copyright (c) 2006-2013 http://thinkphp.cn All rights reserved.
  6 // +----------------------------------------------------------------------
  7 // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
  8 // +----------------------------------------------------------------------
  9 // | Author: Knight <S.Knight.Work@gmail.com> <http://www.getyourwant.com>
 10 // +----------------------------------------------------------------------
 11 // | Commet: 原做者：麥當苗兒 增長了驗證碼中使用的字符的配置，用戶能夠配置
 12 // | 驗證碼中使用的字符的範圍
 13 // +----------------------------------------------------------------------
 14 
 15 namespace Think;
 16 
 17 class Verify {
 18     protected $config =    array(
 19         'seKey'     => 'ThinkPHP.CN',   //驗證碼加密密鑰
 20         'expire'    => 1800,            // 驗證碼過時時間（s）
 21         'useZh'     => false,           // 使用中文驗證碼 
 22         'useImgBg'  => false,           // 使用背景圖片 
 23         'fontSize'  => 25,              // 驗證碼字體大小(px)
 24         'useCurve'  => true,            // 是否畫混淆曲線
 25         'useNoise'  => true,            // 是否添加雜點    
 26         'imageH'    => 0,               // 驗證碼圖片高度
 27         'imageW'    => 0,               // 驗證碼圖片寬度
 28         'length'    => 5,               // 驗證碼位數
 29         'fontttf'   => '',              // 驗證碼字體，不設置隨機獲取
 30         'bg'        => array(243, 251, 254),  // 背景顏色
 31         'codeSet'   => '2345678abcdefhijkmnpqrstuvwxyzABCDEFGHJKLMNPQRTUVWXY', /* 驗證碼中使用的字符，0--o,O 1--l,I, 9--g S--Z容易混淆，建議不用    */
 32         );
 33 
 34     private $_image   = NULL;     // 驗證碼圖片實例
 35     private $_color   = NULL;     // 驗證碼字體顏色
 36 
 37     // 架構方法 設置參數
 38     public function __construct($config=array()){
 39         $this->config   =   array_merge($this->config, $config);
 40     }
 41 
 42     /**
 43      * 使用 $this->name 獲取配置
 44      * @param  string $name 配置名稱
 45      * @return multitype    配置值
 46      */
 47     public function __get($name) {
 48         return $this->config[$name];
 49     }
 50 
 51     public function __set($name,$value){
 52         if(isset($this->config[$name])) {
 53             $this->config[$name]    =   $value;
 54         }
 55     }
 56 
 57     public function __isset($name){
 58         return isset($this->config[$name]);
 59     }
 60 
 61     /**
 62      * 驗證驗證碼是否正確
 63      *
 64      * @param string $code 用戶驗證碼
 65      * @return bool 用戶驗證碼是否正確
 66      */
 67     public function check($code, $id = '') {
 68         $key = $this->authcode($this->seKey);
 69         // 驗證碼不能爲空
 70         $session = session($key);
 71         if(empty($code) || empty($session)) {
 72             return false;
 73         }
 74 
 75         $secode = $id ? $session[$id] : $session;
 76         // session 過時
 77         if(NOW_TIME - $secode['verify_time'] > $this->expire) {
 78             session($key, null);
 79             return false;
 80         }
 81 
 82         if($this->authcode(strtoupper($code)) == $secode['verify_code']) {
 83             session($key, null);
 84             return true;
 85         }
 86 
 87         return false;
 88     }
 89 
 90     /**
 91      * 輸出驗證碼並把驗證碼的值保存的session中
 92      * 驗證碼保存到session的格式爲： array('code' => '驗證碼值', 'time' => '驗證碼建立時間');
 93      */
 94     public function entry($id = '') {
 95         // 圖片寬(px)
 96         $this->imageW || $this->imageW = $this->length*$this->fontSize*1.5 + $this->length*$this->fontSize/2; 
 97         // 圖片高(px)
 98         $this->imageH || $this->imageH = $this->fontSize * 2.5;
 99         // 創建一幅 $this->imageW x $this->imageH 的圖像
100         $this->_image = imagecreate($this->imageW, $this->imageH); 
101         // 設置背景      
102         imagecolorallocate($this->_image, $this->bg[0], $this->bg[1], $this->bg[2]); 
103 
104         // 驗證碼字體隨機顏色
105         $this->_color = imagecolorallocate($this->_image, mt_rand(1,150), mt_rand(1,150), mt_rand(1,150));
106         // 驗證碼使用隨機字體
107         $ttfPath = dirname(__FILE__) . '/Verify/' . ($this->useZh ? 'zhttfs' : 'ttfs') . '/';
108 
109         if(empty($this->fontttf)){
110             $dir = dir($ttfPath);
111             $ttfs = array();        
112             while (false !== ($file = $dir->read())) {
113                 if($file[0] != '.' && substr($file, -4) == '.ttf') {
114                     $ttfs[] = $file;
115                 }
116             }
117             $dir->close();
118             $this->fontttf = $ttfs[array_rand($ttfs)];
119         } 
120         $this->fontttf = $ttfPath . $this->fontttf;
121         
122         if($this->useImgBg) {
123             $this->_background();
124         }
125         
126         if ($this->useNoise) {
127             // 繪雜點
128             $this->_writeNoise();
129         } 
130         if ($this->useCurve) {
131             // 繪干擾線
132             $this->_writeCurve();
133         }
134         
135         // 繪驗證碼
136         $code = array(); // 驗證碼
137         $codeNX = 0; // 驗證碼第N個字符的左邊距
138         for ($i = 0; $i<$this->length; $i++) {
139             if($this->useZh) {
140                 $code[$i] = chr(mt_rand(0xB0,0xF7)).chr(mt_rand(0xA1,0xFE));
141             } else {
142                 $codeSetLen = strlen($this->config['codeSet']);
143                 $code[$i] = $this->config['codeSet'][mt_rand(0,  $codeSetLen - 1)]; //修改，將以前的$this->_codeSet改爲 $this->config['codeSet']，將[mt_rand(0, 51)]改爲[mt_rand(0,  $codeSetLen - 1)]
144                 $codeNX += mt_rand($this->fontSize*1.2, $this->fontSize*1.6);
145                 // 寫一個驗證碼字符
146                 $this->useZh || imagettftext($this->_image, $this->fontSize, mt_rand(-40, 40), $codeNX, $this->fontSize*1.6, $this->_color, $this->fontttf, $code[$i]);
147             }
148         }
149         
150         $this->useZh && imagettftext($this->_image, $this->fontSize, 0, ($this->imageW - $this->fontSize*$this->length*1.2)/3, $this->fontSize * 1.5, $this->_color, $this->fontttf, iconv("GB2312","UTF-8", join('', $code)));
151         
152         // 保存驗證碼
153         $key = $this->authcode($this->seKey);
154         $code = $this->authcode(strtoupper(implode('', $code)));
155         $session = array();
156         if($id) {
157             $session[$id]['verify_code'] = $code; // 把校驗碼保存到session
158             $session[$id]['verify_time'] = NOW_TIME;  // 驗證碼建立時間
159         } else {
160             $session['verify_code'] = $code; // 把校驗碼保存到session
161             $session['verify_time'] = NOW_TIME;  // 驗證碼建立時間
162         }
163         session($key, $session);
164 
165                 
166         header('Cache-Control: private, max-age=0, no-store, no-cache, must-revalidate');
167         header('Cache-Control: post-check=0, pre-check=0', false);        
168         header('Pragma: no-cache');
169         header("content-type: image/png");
170 
171         // 輸出圖像
172         imagepng($this->_image);
173         imagedestroy($this->_image);
174     }
175 
176     /** 
177      * 畫一條由兩條連在一塊兒構成的隨機正弦函數曲線做干擾線(你能夠改爲更帥的曲線函數) 
178      *      
179      *      高中的數學公式咋都忘了涅，寫出來
180      *        正弦型函數解析式：y=Asin(ωx+φ)+b
181      *      各常數值對函數圖像的影響：
182      *        A：決定峯值（即縱向拉伸壓縮的倍數）
183      *        b：表示波形在Y軸的位置關係或縱向移動距離（上加下減）
184      *        φ：決定波形與X軸位置關係或橫向移動距離（左加右減）
185      *        ω：決定週期（最小正週期T=2π/∣ω∣）
186      *
187      */
188     private function _writeCurve() {
189         $px = $py = 0;
190         
191         // 曲線前部分
192         $A = mt_rand(1, $this->imageH/2);                  // 振幅
193         $b = mt_rand(-$this->imageH/4, $this->imageH/4);   // Y軸方向偏移量
194         $f = mt_rand(-$this->imageH/4, $this->imageH/4);   // X軸方向偏移量
195         $T = mt_rand($this->imageH, $this->imageW*2);  // 週期
196         $w = (2* M_PI)/$T;
197                         
198         $px1 = 0;  // 曲線橫座標起始位置
199         $px2 = mt_rand($this->imageW/2, $this->imageW * 0.8);  // 曲線橫座標結束位置
200 
201         for ($px=$px1; $px<=$px2; $px = $px + 1) {
202             if ($w!=0) {
203                 $py = $A * sin($w*$px + $f)+ $b + $this->imageH/2;  // y = Asin(ωx+φ) + b
204                 $i = (int) ($this->fontSize/5);
205                 while ($i > 0) {    
206                     imagesetpixel($this->_image, $px + $i , $py + $i, $this->_color);  // 這裏(while)循環畫像素點比imagettftext和imagestring用字體大小一次畫出（不用這while循環）性能要好不少                
207                     $i--;
208                 }
209             }
210         }
211         
212         // 曲線後部分
213         $A = mt_rand(1, $this->imageH/2);                  // 振幅        
214         $f = mt_rand(-$this->imageH/4, $this->imageH/4);   // X軸方向偏移量
215         $T = mt_rand($this->imageH, $this->imageW*2);  // 週期
216         $w = (2* M_PI)/$T;        
217         $b = $py - $A * sin($w*$px + $f) - $this->imageH/2;
218         $px1 = $px2;
219         $px2 = $this->imageW;
220 
221         for ($px=$px1; $px<=$px2; $px=$px+ 1) {
222             if ($w!=0) {
223                 $py = $A * sin($w*$px + $f)+ $b + $this->imageH/2;  // y = Asin(ωx+φ) + b
224                 $i = (int) ($this->fontSize/5);
225                 while ($i > 0) {            
226                     imagesetpixel($this->_image, $px + $i, $py + $i, $this->_color);    
227                     $i--;
228                 }
229             }
230         }
231     }
232 
233     /**
234      * 畫雜點
235      * 往圖片上寫不一樣顏色的字母或數字
236      */
237     private function _writeNoise() {
238         for($i = 0; $i < 10; $i++){
239             //雜點顏色
240             $noiseColor = imagecolorallocate($this->_image, mt_rand(150,225), mt_rand(150,225), mt_rand(150,225));
241             for($j = 0; $j < 5; $j++) {
242                 // 繪雜點
243                 $tmpCodeSet = '2345678abcdefhijkmnpqrstuvwxyz';  //修改，新增了臨時變量，用於生成繪雜點
244                 imagestring($this->_image, 5, mt_rand(-10, $this->imageW),  mt_rand(-10, $this->imageH), $tmpCodeSet[mt_rand(0, strlen($tmpCodeSet)-1)], $noiseColor);
245             }
246         }
247     }
248 
249     /**
250      * 繪製背景圖片
251      * 注：若是驗證碼輸出圖片比較大，將佔用比較多的系統資源
252      */
253     private function _background() {
254         $path = dirname(__FILE__).'/Verify/bgs/';
255         $dir = dir($path);
256 
257         $bgs = array();        
258         while (false !== ($file = $dir->read())) {
259             if($file[0] != '.' && substr($file, -4) == '.jpg') {
260                 $bgs[] = $path . $file;
261             }
262         }
263         $dir->close();
264 
265         $gb = $bgs[array_rand($bgs)];
266 
267         list($width, $height) = @getimagesize($gb);
268         // Resample
269         $bgImage = @imagecreatefromjpeg($gb);
270         @imagecopyresampled($this->_image, $bgImage, 0, 0, 0, 0, $this->imageW, $this->imageH, $width, $height);
271         @imagedestroy($bgImage);
272     }
273 
274     /* 加密驗證碼 */
275     private function authcode($str){
276         $key = substr(md5($this->seKey), 5, 8);
277         $str = substr(md5($str), 8, 10);
278         return md5($key . $str);
279     }
280 
281 }
相關標籤/搜索
每日一句
每一个你不满意的现在，都有一个你没有努力的曾经。