xref: /PHP-7.3/ext/mysqlnd/mysqlnd_loaddata.c (revision 96404eb8)
1 /*
2   +----------------------------------------------------------------------+
3   | PHP Version 7                                                        |
4   +----------------------------------------------------------------------+
5   | Copyright (c) 2006-2018 The PHP Group                                |
6   +----------------------------------------------------------------------+
7   | This source file is subject to version 3.01 of the PHP license,      |
8   | that is bundled with this package in the file LICENSE, and is        |
9   | available through the world-wide-web at the following url:           |
10   | http://www.php.net/license/3_01.txt                                  |
11   | If you did not receive a copy of the PHP license and are unable to   |
12   | obtain it through the world-wide-web, please send a note to          |
13   | license@php.net so we can mail you a copy immediately.               |
14   +----------------------------------------------------------------------+
15   | Authors: Andrey Hristov <andrey@php.net>                             |
16   |          Ulf Wendel <uw@php.net>                                     |
17   |          Georg Richter <georg@php.net>                               |
18   +----------------------------------------------------------------------+
19 */
20 #include "php.h"
21 #include "mysqlnd.h"
22 #include "mysqlnd_wireprotocol.h"
23 #include "mysqlnd_priv.h"
24 #include "mysqlnd_debug.h"
25 
26 /* {{{ mysqlnd_local_infile_init */
27 static
mysqlnd_local_infile_init(void ** ptr,const char * const filename)28 int mysqlnd_local_infile_init(void ** ptr, const char * const filename)
29 {
30 	MYSQLND_INFILE_INFO	*info;
31 	php_stream_context	*context = NULL;
32 
33 	DBG_ENTER("mysqlnd_local_infile_init");
34 
35 	info = ((MYSQLND_INFILE_INFO *)mnd_ecalloc(1, sizeof(MYSQLND_INFILE_INFO)));
36 	if (!info) {
37 		DBG_RETURN(1);
38 	}
39 
40 	*ptr = info;
41 
42 	/* check open_basedir */
43 	if (PG(open_basedir)) {
44 		if (php_check_open_basedir_ex(filename, 0) == -1) {
45 			strcpy(info->error_msg, "open_basedir restriction in effect. Unable to open file");
46 			info->error_no = CR_UNKNOWN_ERROR;
47 			DBG_RETURN(1);
48 		}
49 	}
50 
51 	info->filename = filename;
52 	info->fd = php_stream_open_wrapper_ex((char *)filename, "r", 0, NULL, context);
53 
54 	if (info->fd == NULL) {
55 		snprintf((char *)info->error_msg, sizeof(info->error_msg), "Can't find file '%-.64s'.", filename);
56 		info->error_no = MYSQLND_EE_FILENOTFOUND;
57 		DBG_RETURN(1);
58 	}
59 
60 	DBG_RETURN(0);
61 }
62 /* }}} */
63 
64 
65 /* {{{ mysqlnd_local_infile_read */
66 static
mysqlnd_local_infile_read(void * ptr,zend_uchar * buf,unsigned int buf_len)67 int mysqlnd_local_infile_read(void * ptr, zend_uchar * buf, unsigned int buf_len)
68 {
69 	MYSQLND_INFILE_INFO	*info = (MYSQLND_INFILE_INFO *)ptr;
70 	int count;
71 
72 	DBG_ENTER("mysqlnd_local_infile_read");
73 
74 	count = (int) php_stream_read(info->fd, (char *) buf, buf_len);
75 
76 	if (count < 0) {
77 		strcpy(info->error_msg, "Error reading file");
78 		info->error_no = CR_UNKNOWN_ERROR;
79 	}
80 
81 	DBG_RETURN(count);
82 }
83 /* }}} */
84 
85 
86 /* {{{ mysqlnd_local_infile_error */
87 static
mysqlnd_local_infile_error(void * ptr,char * error_buf,unsigned int error_buf_len)88 int	mysqlnd_local_infile_error(void * ptr, char *error_buf, unsigned int error_buf_len)
89 {
90 	MYSQLND_INFILE_INFO	*info = (MYSQLND_INFILE_INFO *)ptr;
91 
92 	DBG_ENTER("mysqlnd_local_infile_error");
93 
94 	if (info) {
95 		strlcpy(error_buf, info->error_msg, error_buf_len);
96 		DBG_INF_FMT("have info, %d", info->error_no);
97 		DBG_RETURN(info->error_no);
98 	}
99 
100 	strlcpy(error_buf, "Unknown error", error_buf_len);
101 	DBG_INF_FMT("no info, %d", CR_UNKNOWN_ERROR);
102 	DBG_RETURN(CR_UNKNOWN_ERROR);
103 }
104 /* }}} */
105 
106 
107 /* {{{ mysqlnd_local_infile_end */
108 static
mysqlnd_local_infile_end(void * ptr)109 void mysqlnd_local_infile_end(void * ptr)
110 {
111 	MYSQLND_INFILE_INFO	*info = (MYSQLND_INFILE_INFO *)ptr;
112 
113 	if (info) {
114 		/* php_stream_close segfaults on NULL */
115 		if (info->fd) {
116 			php_stream_close(info->fd);
117 			info->fd = NULL;
118 		}
119 		mnd_efree(info);
120 	}
121 }
122 /* }}} */
123 
124 
125 /* {{{ mysqlnd_local_infile_default */
126 PHPAPI void
mysqlnd_local_infile_default(MYSQLND_CONN_DATA * conn)127 mysqlnd_local_infile_default(MYSQLND_CONN_DATA * conn)
128 {
129 	conn->infile.local_infile_init = mysqlnd_local_infile_init;
130 	conn->infile.local_infile_read = mysqlnd_local_infile_read;
131 	conn->infile.local_infile_error = mysqlnd_local_infile_error;
132 	conn->infile.local_infile_end = mysqlnd_local_infile_end;
133 }
134 /* }}} */
135 
136 
137 static const char *lost_conn = "Lost connection to MySQL server during LOAD DATA of a local file";
138 
139 
140 /* {{{ mysqlnd_handle_local_infile */
141 enum_func_status
mysqlnd_handle_local_infile(MYSQLND_CONN_DATA * conn,const char * const filename,zend_bool * is_warning)142 mysqlnd_handle_local_infile(MYSQLND_CONN_DATA * conn, const char * const filename, zend_bool * is_warning)
143 {
144 	zend_uchar			*buf = NULL;
145 	zend_uchar			empty_packet[MYSQLND_HEADER_SIZE];
146 	enum_func_status	result = FAIL;
147 	unsigned int		buflen = 4096;
148 	void				*info = NULL;
149 	int					bufsize;
150 	size_t				ret;
151 	MYSQLND_INFILE		infile;
152 	MYSQLND_PFC			* net = conn->protocol_frame_codec;
153 	MYSQLND_VIO			* vio = conn->vio;
154 
155 	DBG_ENTER("mysqlnd_handle_local_infile");
156 
157 	if (!(conn->options->flags & CLIENT_LOCAL_FILES)) {
158 		php_error_docref(NULL, E_WARNING, "LOAD DATA LOCAL INFILE forbidden");
159 		SET_CLIENT_ERROR(conn->error_info, CR_UNKNOWN_ERROR, UNKNOWN_SQLSTATE,
160 						"LOAD DATA LOCAL INFILE is forbidden, check mysqli.allow_local_infile");
161 		/* write empty packet to server */
162 		ret = net->data->m.send(net, vio, empty_packet, 0, conn->stats, conn->error_info);
163 		*is_warning = TRUE;
164 		goto infile_error;
165 	}
166 
167 	infile = conn->infile;
168 	/* allocate buffer for reading data */
169 	buf = (zend_uchar *) mnd_ecalloc(1, buflen);
170 
171 	*is_warning = FALSE;
172 
173 	/* init handler: allocate read buffer and open file */
174 	if (infile.local_infile_init(&info, (char *)filename)) {
175 		char tmp_buf[sizeof(conn->error_info->error)];
176 		int tmp_error_no;
177 		*is_warning = TRUE;
178 		/* error occurred */
179 		tmp_error_no = infile.local_infile_error(info, tmp_buf, sizeof(tmp_buf));
180 		SET_CLIENT_ERROR(conn->error_info, tmp_error_no, UNKNOWN_SQLSTATE, tmp_buf);
181 		/* write empty packet to server */
182 		ret = net->data->m.send(net, vio, empty_packet, 0, conn->stats, conn->error_info);
183 		goto infile_error;
184 	}
185 
186 	/* read data */
187 	while ((bufsize = infile.local_infile_read (info, buf + MYSQLND_HEADER_SIZE, buflen - MYSQLND_HEADER_SIZE)) > 0) {
188 		if ((ret = net->data->m.send(net, vio, buf, bufsize, conn->stats, conn->error_info)) == 0) {
189 			DBG_ERR_FMT("Error during read : %d %s %s", CR_SERVER_LOST, UNKNOWN_SQLSTATE, lost_conn);
190 			SET_CLIENT_ERROR(conn->error_info, CR_SERVER_LOST, UNKNOWN_SQLSTATE, lost_conn);
191 			goto infile_error;
192 		}
193 	}
194 
195 	/* send empty packet for eof */
196 	if ((ret = net->data->m.send(net, vio, empty_packet, 0, conn->stats, conn->error_info)) == 0) {
197 		SET_CLIENT_ERROR(conn->error_info, CR_SERVER_LOST, UNKNOWN_SQLSTATE, lost_conn);
198 		goto infile_error;
199 	}
200 
201 	/* error during read occurred */
202 	if (bufsize < 0) {
203 		char tmp_buf[sizeof(conn->error_info->error)];
204 		int tmp_error_no;
205 		*is_warning = TRUE;
206 		DBG_ERR_FMT("Bufsize < 0, warning,  %d %s %s", CR_SERVER_LOST, UNKNOWN_SQLSTATE, lost_conn);
207 		tmp_error_no = infile.local_infile_error(info, tmp_buf, sizeof(tmp_buf));
208 		SET_CLIENT_ERROR(conn->error_info, tmp_error_no, UNKNOWN_SQLSTATE, tmp_buf);
209 		goto infile_error;
210 	}
211 
212 	result = PASS;
213 
214 infile_error:
215 	/* get response from server and update upsert values */
216 	if (FAIL == conn->payload_decoder_factory->m.send_command_handle_response(
217 											conn->payload_decoder_factory,
218 											PROT_OK_PACKET, FALSE, COM_QUERY, FALSE,
219 											conn->error_info,
220 											conn->upsert_status,
221 											&conn->last_message)) {
222 		result = FAIL;
223 	}
224 
225 	(*conn->infile.local_infile_end)(info);
226 	if (buf) {
227 		mnd_efree(buf);
228 	}
229 	DBG_INF_FMT("%s", result == PASS? "PASS":"FAIL");
230 	DBG_RETURN(result);
231 }
232 /* }}} */
233 
234 /*
235  * Local variables:
236  * tab-width: 4
237  * c-basic-offset: 4
238  * End:
239  * vim600: noet sw=4 ts=4 fdm=marker
240  * vim<600: noet sw=4 ts=4
241  */
242