Home  >  Article  >  Backend Development  >  php-curl封装[避免下传文件二义性Bug]

php-curl封装[避免下传文件二义性Bug]

WBOY
WBOYOriginal
2016-06-13 13:13:16926browse

php-curl封装[避免上传文件二义性Bug]

由于php的curl在curl_setopt($curl, CURLOPT_POSTFIELDS, xxx)时, 当xxx为数组时, 如果值的第一个字符是@, 则认为是文件上传, 当同时需要上传文件, 也需要提交可能首字符为@的其他普通数据时, 存在冲突. 因此, 在api_common.php中的post数据的设置进行了封装


<?php /**
 * php-curl库封装
 * author: selfimpr
 * blog: http://blog.csdn.net/lgg201
 * mail: lgg860911@yahoo.com.cn
 */

define('API_CURL_UPLOAD_FILE',							'__file');

#支持的请求方法
define('REQUEST_METHOD_GET',							'GET');
define('REQUEST_METHOD_POST',							'POST');
define('REQUEST_METHOD_HEAD',							'HEAD');

#请求行为
define('REQUEST_BEHAVIOR_ALLOW_REDIRECT',				'allow_redirect');
define('REQUEST_BEHAVIOR_MAX_REDIRECT',					'max_redirect');
define('REQUEST_BEHAVIOR_USER_AGENT',					'user_agent');
define('REQUEST_BEHAVIOR_AUTOREFERER',					'autoreferer');
define('REQUEST_BEHAVIOR_UPLOAD',						'upload');
define('REQUEST_BEHAVIOR_CONNECTTIMEOUT',				'connecttimeout');
define('REQUEST_BEHAVIOR_DNS_CACHE_TIMEOUT',			'dns_cache_timeout');
define('REQUEST_BEHAVIOR_TIMEOUT',						'timeout');
define('REQUEST_BEHAVIOR_ENCODING',						'encoding');
define('REQUEST_BEHAVIOR_ERROR_HANDLER',				'error_handler');
define('REQUEST_BEHAVIORS',								'behaviors');
$GLOBALS[REQUEST_BEHAVIORS]	= array(
	REQUEST_BEHAVIOR_ALLOW_REDIRECT				=> TRUE, 
	REQUEST_BEHAVIOR_MAX_REDIRECT				=> 5, 
	REQUEST_BEHAVIOR_USER_AGENT					=> 'curl-lib', 
	REQUEST_BEHAVIOR_AUTOREFERER				=> TRUE, 
	REQUEST_BEHAVIOR_UPLOAD						=> FALSE, 
	REQUEST_BEHAVIOR_CONNECTTIMEOUT				=> 3, 
	REQUEST_BEHAVIOR_DNS_CACHE_TIMEOUT			=> 3600, 
	REQUEST_BEHAVIOR_TIMEOUT					=> 3, 
	REQUEST_BEHAVIOR_ENCODING					=> 'gzip', 
	REQUEST_BEHAVIOR_ERROR_HANDLER				=> '__default_curl_error_handler', 
);

define('MULTIPART_FORM_DATA_HEAD_FMT',				'Content-Type: multipart/form-data; boundary=----------------------------%s');
define('MULTIPART_FORM_DATA_BODY_STRING',			"------------------------------%s\r\nContent-Disposition: form-data; name=\"%s\"\r\n\r\n%s\r\n");
define('MULTIPART_FORM_DATA_BODY_FILE',				"------------------------------%s\r\nContent-Disposition: form-data; name=\"%s\"; filename=\"%s\"\r\nContent-Type: application/octet-stream\r\n\r\n%s\r\n");
define('MULTIPART_FORM_DATA_BODY_END',				"------------------------------%s--\r\n\r\n");

#响应键值
define('RESP_CODE',									'resp_code');
define('RESP_BODY',									'resp_body');
define('RESP_HEADER',								'resp_header');

#HTTP 1xx状态验证
define('HTTP_1XX_RESP',								'/^HTTP\/1.[01] 1\d{2} \w+/');

#默认错误处理的错误消息
define('E_CURL_ERROR_FMT',								'curl "%s" error[%d]: %s');

#默认的curl错误处理
function __default_curl_error_handler($curl, $url, $errno, $errstr) {
	trigger_error(sprintf(E_CURL_ERROR_FMT, $url, $errno, $errstr), E_USER_ERROR);
}
#切换CURL请求方法
function __method_switch($curl, $method) {
	switch ( $method) {
		case REQUEST_METHOD_POST:
			__curl_setopt($curl, CURLOPT_POST, TRUE);
			break;
		case REQUEST_METHOD_HEAD:
			__curl_setopt($curl, CURLOPT_NOBODY, TRUE);
			break;
		case REQUEST_METHOD_GET:
			__curl_setopt($curl, CURLOPT_HTTPGET, TRUE);
			break;
		default:
			break;
	}
}
#设置默认头信息
function __default_header_set($curl) {
	__curl_setopt($curl, CURLOPT_RETURNTRANSFER,			TRUE);
	__curl_setopt($curl, CURLOPT_HEADER,					TRUE);
	__curl_setopt($curl, CURLOPT_FOLLOWLOCATION,			(bool)curl_behavior(REQUEST_BEHAVIOR_ALLOW_REDIRECT));
	__curl_setopt($curl, CURLOPT_MAXREDIRS,					(int)curl_behavior(REQUEST_BEHAVIOR_MAX_REDIRECT));
	__curl_setopt($curl, CURLOPT_USERAGENT,					(string)curl_behavior(REQUEST_BEHAVIOR_USER_AGENT));
	__curl_setopt($curl, CURLOPT_AUTOREFERER,				(bool)curl_behavior(REQUEST_BEHAVIOR_AUTOREFERER));
	__curl_setopt($curl, CURLOPT_UPLOAD,					(bool)curl_behavior(REQUEST_BEHAVIOR_UPLOAD));
	__curl_setopt($curl, CURLOPT_CONNECTTIMEOUT,			(int)curl_behavior(REQUEST_BEHAVIOR_CONNECTTIMEOUT));
	__curl_setopt($curl, CURLOPT_DNS_CACHE_TIMEOUT,			(int)curl_behavior(REQUEST_BEHAVIOR_DNS_CACHE_TIMEOUT));
	__curl_setopt($curl, CURLOPT_TIMEOUT,					(int)curl_behavior(REQUEST_BEHAVIOR_TIMEOUT));
	__curl_setopt($curl, CURLOPT_ENCODING,					(string)curl_behavior(REQUEST_BEHAVIOR_ENCODING));
}
#设置用户自定义头信息
function __custom_header_set($curl, $headers = NULL) {
	if ( empty($headers) ) return ;
	if ( is_string($headers) ) 
		$headers	= explode("\r\n", $headers);
	#类型修复
	foreach ( $headers as &$header ) 
		if ( is_array($header) ) 
			$header	= sprintf('%s: %s', $header[0], $header[1]);
	__curl_setopt($curl, CURLOPT_HTTPHEADER, $headers);
}
#设置请求body
function __datas_set($curl, $datas = NULL) {
	if ( empty($datas) ) return ;
	if ( is_array($datas) ) {
		$custom_body		= FALSE;
		$uniqid				= uniqid();
		$custom_body_str	= '';
		foreach ( $datas as $name => $data ) {
			if ( is_array($data) && array_key_exists(API_CURL_UPLOAD_FILE, $data) ) {
				$file	= $data[API_CURL_UPLOAD_FILE];
				if ( file_exists($file) ) {
					$custom_body		= TRUE;
					$custom_body_str	.= sprintf(MULTIPART_FORM_DATA_BODY_FILE, 
										$uniqid, $name, 
										$file, file_get_contents($file));
				}
			} else {
				$custom_body_str		.= sprintf(MULTIPART_FORM_DATA_BODY_STRING, 
										$uniqid, $name, $data);
			}
		}
		if ( $custom_body ) {
			curl_setopt($curl, CURLOPT_HTTPHEADER, array(sprintf(MULTIPART_FORM_DATA_HEAD_FMT, $uniqid)));
			$datas				= $custom_body_str . sprintf(MULTIPART_FORM_DATA_BODY_END, $uniqid);
		}
	}
	__curl_setopt($curl, CURLOPT_POSTFIELDS, $datas);
}
#对curl_setopt的封装
function __curl_setopt($curl, $optname, $optval) {
	curl_setopt($curl, $optname, $optval);
	__curl_error($curl);
}
#curl错误检查处理
function __curl_error($curl) {
	if ( curl_errno($curl) ) {
		$url	= curl_getinfo($curl, CURLINFO_EFFECTIVE_URL);
		$errno	= curl_errno($curl);
		$errstr	= curl_error($curl);
		$errh	= curl_behavior(REQUEST_BEHAVIOR_ERROR_HANDLER);
		if ( function_exists($errh) )
			$errh($curl, $url, $errno, $errstr);
	}
}

#api默认行为切换
function curl_behavior($names, $values = NULL) {
	if ( !is_string($names) && !is_array($names) ) return ;
	if ( !is_null($values) ) {
		if ( is_string($names) ) 
			$GLOBALS[REQUEST_BEHAVIORS][$names]	= $values;
		else if ( is_array($names) && !is_array($values) )
			foreach ( $names as $name )
				$GLOBALS[REQUEST_BEHAVIORS][$name]	= $values;
		else if ( is_array($names) && is_array($values) )
			foreach ( $names as $k => $name ) 
				$GLOBALS[REQUEST_BEHAVIORS][$name]	= $values[$k];
	}
	if ( is_string($names) ) {
		$return	= $GLOBALS[REQUEST_BEHAVIORS][$names];
	} else if ( is_array($names) ) {
		$return	= array();
		foreach ( $names as $name ) 
			$return[$name]	= array_key_exists($name, $GLOBALS[REQUEST_BEHAVIORS]) 
							? $GLOBALS[REQUEST_BEHAVIORS][$name]
							: NULL;
	}
	return $return;
}
#请求入口
function curl_request($url, $method, $datas = NULL, $headers = NULL) {
	$curl	= curl_init($url);
	__method_switch($curl, $method);
	__default_header_set($curl);
	__custom_header_set($curl, $headers);
	__datas_set($curl, $datas);
	$response	= curl_exec($curl);
	__curl_error($curl);
	$status_code	= curl_getinfo($curl, CURLINFO_HTTP_CODE);
	$components		= explode("\r\n\r\n", $response);
	$i				= -1;
	while ( ++ $i  $status_code, 
		RESP_HEADER	=> $headers, 
		RESP_BODY	=> $body, 
	);
}
#GET请求
function curl_get($url, $headers = NULL) {
	return curl_request($url, REQUEST_METHOD_GET, NULL, $headers);
}
#POST请求
function curl_post($url, $datas = NULL, $headers = NULL) {
	return curl_request($url, REQUEST_METHOD_POST, $datas, $headers);
}
#HEAD请求
function curl_head($url, $headers = NULL) {
	return curl_request($url, REQUEST_METHOD_HEAD, NULL, $headers);
}
#构造上传文件字段
function curl_post_file($file) {
	return array(
		API_CURL_UPLOAD_FILE	=> $file, 
	);
}
#读取响应码
function curl_resp_code($resp) {
	return $resp[RESP_CODE];
}
#读取响应头
function curl_resp_header($resp) {
	return $resp[RESP_HEADER];
}
#读取响应体
function curl_resp_body($resp) {
	return $resp[RESP_BODY];
}


Statement:
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn