221 lines
		
	
	
	
		
			7.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			221 lines
		
	
	
	
		
			7.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| /**
 | |
|  * Smarty plugin
 | |
|  *
 | |
|  * @package    Smarty
 | |
|  * @subpackage PluginsFunction
 | |
|  */
 | |
| 
 | |
| /**
 | |
|  * Smarty {fetch} plugin
 | |
|  * Type:     function<br>
 | |
|  * Name:     fetch<br>
 | |
|  * Purpose:  fetch file, web or ftp data and display results
 | |
|  *
 | |
|  * @link   http://www.smarty.net/manual/en/language.function.fetch.php {fetch}
 | |
|  *         (Smarty online manual)
 | |
|  * @author Monte Ohrt <monte at ohrt dot com>
 | |
|  *
 | |
|  * @param array                    $params   parameters
 | |
|  * @param Smarty_Internal_Template $template template object
 | |
|  *
 | |
|  * @throws SmartyException
 | |
|  * @return string|null if the assign parameter is passed, Smarty assigns the result to a template variable
 | |
|  */
 | |
| function smarty_function_fetch($params, $template)
 | |
| {
 | |
|     if (empty($params['file'])) {
 | |
|         trigger_error("[plugin] fetch parameter 'file' cannot be empty", E_USER_NOTICE);
 | |
| 
 | |
|         return;
 | |
|     }
 | |
| 
 | |
|     // strip file protocol
 | |
|     if (stripos($params['file'], 'file://') === 0) {
 | |
|         $params['file'] = substr($params['file'], 7);
 | |
|     }
 | |
| 
 | |
|     $protocol = strpos($params['file'], '://');
 | |
|     if ($protocol !== false) {
 | |
|         $protocol = strtolower(substr($params['file'], 0, $protocol));
 | |
|     }
 | |
| 
 | |
|     if (isset($template->smarty->security_policy)) {
 | |
|         if ($protocol) {
 | |
|             // remote resource (or php stream, …)
 | |
|             if (!$template->smarty->security_policy->isTrustedUri($params['file'])) {
 | |
|                 return;
 | |
|             }
 | |
|         } else {
 | |
|             // local file
 | |
|             if (!$template->smarty->security_policy->isTrustedResourceDir($params['file'])) {
 | |
|                 return;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     $content = '';
 | |
|     if ($protocol == 'http') {
 | |
|         // http fetch
 | |
|         if ($uri_parts = parse_url($params['file'])) {
 | |
|             // set defaults
 | |
|             $host = $server_name = $uri_parts['host'];
 | |
|             $timeout = 30;
 | |
|             $accept = "image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, */*";
 | |
|             $agent = "Smarty Template Engine " . Smarty::SMARTY_VERSION;
 | |
|             $referer = "";
 | |
|             $uri = !empty($uri_parts['path']) ? $uri_parts['path'] : '/';
 | |
|             $uri .= !empty($uri_parts['query']) ? '?' . $uri_parts['query'] : '';
 | |
|             $_is_proxy = false;
 | |
|             if (empty($uri_parts['port'])) {
 | |
|                 $port = 80;
 | |
|             } else {
 | |
|                 $port = $uri_parts['port'];
 | |
|             }
 | |
|             if (!empty($uri_parts['user'])) {
 | |
|                 $user = $uri_parts['user'];
 | |
|             }
 | |
|             if (!empty($uri_parts['pass'])) {
 | |
|                 $pass = $uri_parts['pass'];
 | |
|             }
 | |
|             // loop through parameters, setup headers
 | |
|             foreach ($params as $param_key => $param_value) {
 | |
|                 switch ($param_key) {
 | |
|                     case "file":
 | |
|                     case "assign":
 | |
|                     case "assign_headers":
 | |
|                         break;
 | |
|                     case "user":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $user = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "pass":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $pass = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "accept":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $accept = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "header":
 | |
|                         if (!empty($param_value)) {
 | |
|                             if (!preg_match('![\w\d-]+: .+!', $param_value)) {
 | |
|                                 trigger_error("[plugin] invalid header format '" . $param_value . "'", E_USER_NOTICE);
 | |
| 
 | |
|                                 return;
 | |
|                             } else {
 | |
|                                 $extra_headers[] = $param_value;
 | |
|                             }
 | |
|                         }
 | |
|                         break;
 | |
|                     case "proxy_host":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $proxy_host = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "proxy_port":
 | |
|                         if (!preg_match('!\D!', $param_value)) {
 | |
|                             $proxy_port = (int) $param_value;
 | |
|                         } else {
 | |
|                             trigger_error("[plugin] invalid value for attribute '" . $param_key . "'", E_USER_NOTICE);
 | |
| 
 | |
|                             return;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "agent":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $agent = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "referer":
 | |
|                         if (!empty($param_value)) {
 | |
|                             $referer = $param_value;
 | |
|                         }
 | |
|                         break;
 | |
|                     case "timeout":
 | |
|                         if (!preg_match('!\D!', $param_value)) {
 | |
|                             $timeout = (int) $param_value;
 | |
|                         } else {
 | |
|                             trigger_error("[plugin] invalid value for attribute '" . $param_key . "'", E_USER_NOTICE);
 | |
| 
 | |
|                             return;
 | |
|                         }
 | |
|                         break;
 | |
|                     default:
 | |
|                         trigger_error("[plugin] unrecognized attribute '" . $param_key . "'", E_USER_NOTICE);
 | |
| 
 | |
|                         return;
 | |
|                 }
 | |
|             }
 | |
|             if (!empty($proxy_host) && !empty($proxy_port)) {
 | |
|                 $_is_proxy = true;
 | |
|                 $fp = fsockopen($proxy_host, $proxy_port, $errno, $errstr, $timeout);
 | |
|             } else {
 | |
|                 $fp = fsockopen($server_name, $port, $errno, $errstr, $timeout);
 | |
|             }
 | |
| 
 | |
|             if (!$fp) {
 | |
|                 trigger_error("[plugin] unable to fetch: $errstr ($errno)", E_USER_NOTICE);
 | |
| 
 | |
|                 return;
 | |
|             } else {
 | |
|                 if ($_is_proxy) {
 | |
|                     fputs($fp, 'GET ' . $params['file'] . " HTTP/1.0\r\n");
 | |
|                 } else {
 | |
|                     fputs($fp, "GET $uri HTTP/1.0\r\n");
 | |
|                 }
 | |
|                 if (!empty($host)) {
 | |
|                     fputs($fp, "Host: $host\r\n");
 | |
|                 }
 | |
|                 if (!empty($accept)) {
 | |
|                     fputs($fp, "Accept: $accept\r\n");
 | |
|                 }
 | |
|                 if (!empty($agent)) {
 | |
|                     fputs($fp, "User-Agent: $agent\r\n");
 | |
|                 }
 | |
|                 if (!empty($referer)) {
 | |
|                     fputs($fp, "Referer: $referer\r\n");
 | |
|                 }
 | |
|                 if (isset($extra_headers) && is_array($extra_headers)) {
 | |
|                     foreach ($extra_headers as $curr_header) {
 | |
|                         fputs($fp, $curr_header . "\r\n");
 | |
|                     }
 | |
|                 }
 | |
|                 if (!empty($user) && !empty($pass)) {
 | |
|                     fputs($fp, "Authorization: BASIC " . base64_encode("$user:$pass") . "\r\n");
 | |
|                 }
 | |
| 
 | |
|                 fputs($fp, "\r\n");
 | |
|                 while (!feof($fp)) {
 | |
|                     $content .= fgets($fp, 4096);
 | |
|                 }
 | |
|                 fclose($fp);
 | |
|                 $csplit = preg_split("!\r\n\r\n!", $content, 2);
 | |
| 
 | |
|                 $content = $csplit[1];
 | |
| 
 | |
|                 if (!empty($params['assign_headers'])) {
 | |
|                     $template->assign($params['assign_headers'], preg_split("!\r\n!", $csplit[0]));
 | |
|                 }
 | |
|             }
 | |
|         } else {
 | |
|             trigger_error("[plugin fetch] unable to parse URL, check syntax", E_USER_NOTICE);
 | |
| 
 | |
|             return;
 | |
|         }
 | |
|     } else {
 | |
|         $content = @file_get_contents($params['file']);
 | |
|         if ($content === false) {
 | |
|             throw new SmartyException("{fetch} cannot read resource '" . $params['file'] . "'");
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (!empty($params['assign'])) {
 | |
|         $template->assign($params['assign'], $content);
 | |
|     } else {
 | |
|         return $content;
 | |
|     }
 | |
| }
 |