xref: /PHP-7.1/ext/dba/libflatfile/flatfile.c (revision ccd4716e)
1 /*
2    +----------------------------------------------------------------------+
3    | PHP Version 7                                                        |
4    +----------------------------------------------------------------------+
5    | Copyright (c) 1997-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: Marcus Boerger <helly@php.net>                              |
16    | based on ext/db/db.c by:                                             |
17    |          Rasmus Lerdorf <rasmus@php.net>                             |
18    |          Jim Winstead <jimw@php.net>                                 |
19    +----------------------------------------------------------------------+
20  */
21 
22 /* $Id: 464bb5af8f2e8462a17d20bf000e48ae24de5bd3 $ */
23 
24 #ifdef HAVE_CONFIG_H
25 #include "config.h"
26 #endif
27 
28 #include "php.h"
29 #include "php_globals.h"
30 
31 #include <stdlib.h>
32 #include <string.h>
33 #include <errno.h>
34 #if HAVE_UNISTD_H
35 #include <unistd.h>
36 #endif
37 
38 #include "flatfile.h"
39 
40 #define FLATFILE_BLOCK_SIZE 1024
41 
42 /*
43  * ret = -1 means that database was opened for read-only
44  * ret = 0  success
45  * ret = 1  key already exists - nothing done
46  */
47 
48 /* {{{ flatfile_store
49  */
flatfile_store(flatfile * dba,datum key_datum,datum value_datum,int mode)50 int flatfile_store(flatfile *dba, datum key_datum, datum value_datum, int mode) {
51 	if (mode == FLATFILE_INSERT) {
52 		if (flatfile_findkey(dba, key_datum)) {
53 			return 1;
54 		}
55 		php_stream_seek(dba->fp, 0L, SEEK_END);
56 		php_stream_printf(dba->fp, "%zu\n", key_datum.dsize);
57 		php_stream_flush(dba->fp);
58 		if (php_stream_write(dba->fp, key_datum.dptr, key_datum.dsize) < key_datum.dsize) {
59 			return -1;
60 		}
61 		php_stream_printf(dba->fp, "%zu\n", value_datum.dsize);
62 		php_stream_flush(dba->fp);
63 		if (php_stream_write(dba->fp, value_datum.dptr, value_datum.dsize) < value_datum.dsize) {
64 			return -1;
65 		}
66 	} else { /* FLATFILE_REPLACE */
67 		flatfile_delete(dba, key_datum);
68 		php_stream_printf(dba->fp, "%zu\n", key_datum.dsize);
69 		php_stream_flush(dba->fp);
70 		if (php_stream_write(dba->fp, key_datum.dptr, key_datum.dsize) < key_datum.dsize) {
71 			return -1;
72 		}
73 		php_stream_printf(dba->fp, "%zu\n", value_datum.dsize);
74 		if (php_stream_write(dba->fp, value_datum.dptr, value_datum.dsize) < value_datum.dsize) {
75 			return -1;
76 		}
77 	}
78 
79 	php_stream_flush(dba->fp);
80 	return 0;
81 }
82 /* }}} */
83 
84 /* {{{ flatfile_fetch
85  */
flatfile_fetch(flatfile * dba,datum key_datum)86 datum flatfile_fetch(flatfile *dba, datum key_datum) {
87 	datum value_datum = {NULL, 0};
88 	char buf[16];
89 
90 	if (flatfile_findkey(dba, key_datum)) {
91 		if (php_stream_gets(dba->fp, buf, sizeof(buf))) {
92 			value_datum.dsize = atoi(buf);
93 			value_datum.dptr = safe_emalloc(value_datum.dsize, 1, 1);
94 			value_datum.dsize = php_stream_read(dba->fp, value_datum.dptr, value_datum.dsize);
95 		} else {
96 			value_datum.dptr = NULL;
97 			value_datum.dsize = 0;
98 		}
99 	}
100 	return value_datum;
101 }
102 /* }}} */
103 
104 /* {{{ flatfile_delete
105  */
flatfile_delete(flatfile * dba,datum key_datum)106 int flatfile_delete(flatfile *dba, datum key_datum) {
107 	char *key = key_datum.dptr;
108 	size_t size = key_datum.dsize;
109 	size_t buf_size = FLATFILE_BLOCK_SIZE;
110 	char *buf = emalloc(buf_size);
111 	size_t num;
112 	size_t pos;
113 
114 	php_stream_rewind(dba->fp);
115 	while(!php_stream_eof(dba->fp)) {
116 		/* read in the length of the key name */
117 		if (!php_stream_gets(dba->fp, buf, 15)) {
118 			break;
119 		}
120 		num = atoi(buf);
121 		if (num >= buf_size) {
122 			buf_size = num + FLATFILE_BLOCK_SIZE;
123 			buf = erealloc(buf, buf_size);
124 		}
125 		pos = php_stream_tell(dba->fp);
126 
127 		/* read in the key name */
128 		num = php_stream_read(dba->fp, buf, num);
129 
130 		if (size == num && !memcmp(buf, key, size)) {
131 			php_stream_seek(dba->fp, pos, SEEK_SET);
132 			php_stream_putc(dba->fp, 0);
133 			php_stream_flush(dba->fp);
134 			php_stream_seek(dba->fp, 0L, SEEK_END);
135 			efree(buf);
136 			return SUCCESS;
137 		}
138 
139 		/* read in the length of the value */
140 		if (!php_stream_gets(dba->fp, buf, 15)) {
141 			break;
142 		}
143 		num = atoi(buf);
144 		if (num >= buf_size) {
145 			buf_size = num + FLATFILE_BLOCK_SIZE;
146 			buf = erealloc(buf, buf_size);
147 		}
148 		/* read in the value */
149 		num = php_stream_read(dba->fp, buf, num);
150 	}
151 	efree(buf);
152 	return FAILURE;
153 }
154 /* }}} */
155 
156 /* {{{ flatfile_findkey
157  */
flatfile_findkey(flatfile * dba,datum key_datum)158 int flatfile_findkey(flatfile *dba, datum key_datum) {
159 	size_t buf_size = FLATFILE_BLOCK_SIZE;
160 	char *buf = emalloc(buf_size);
161 	size_t num;
162 	int ret=0;
163 	void *key = key_datum.dptr;
164 	size_t size = key_datum.dsize;
165 
166 	php_stream_rewind(dba->fp);
167 	while (!php_stream_eof(dba->fp)) {
168 		if (!php_stream_gets(dba->fp, buf, 15)) {
169 			break;
170 		}
171 		num = atoi(buf);
172 		if (num >= buf_size) {
173 			buf_size = num + FLATFILE_BLOCK_SIZE;
174 			buf = erealloc(buf, buf_size);
175 		}
176 		num = php_stream_read(dba->fp, buf, num);
177 
178 		if (size == num) {
179 			if (!memcmp(buf, key, size)) {
180 				ret = 1;
181 				break;
182 			}
183 		}
184 		if (!php_stream_gets(dba->fp, buf, 15)) {
185 			break;
186 		}
187 		num = atoi(buf);
188 		if (num >= buf_size) {
189 			buf_size = num + FLATFILE_BLOCK_SIZE;
190 			buf = erealloc(buf, buf_size);
191 		}
192 		num = php_stream_read(dba->fp, buf, num);
193 	}
194 	efree(buf);
195 	return ret;
196 }
197 /* }}} */
198 
199 /* {{{ flatfile_firstkey
200  */
flatfile_firstkey(flatfile * dba)201 datum flatfile_firstkey(flatfile *dba) {
202 	datum res;
203 	size_t num;
204 	size_t buf_size = FLATFILE_BLOCK_SIZE;
205 	char *buf = emalloc(buf_size);
206 
207 	php_stream_rewind(dba->fp);
208 	while(!php_stream_eof(dba->fp)) {
209 		if (!php_stream_gets(dba->fp, buf, 15)) {
210 			break;
211 		}
212 		num = atoi(buf);
213 		if (num >= buf_size) {
214 			buf_size = num + FLATFILE_BLOCK_SIZE;
215 			buf = erealloc(buf, buf_size);
216 		}
217 		num = php_stream_read(dba->fp, buf, num);
218 
219 		if (*(buf) != 0) {
220 			dba->CurrentFlatFilePos = php_stream_tell(dba->fp);
221 			res.dptr = buf;
222 			res.dsize = num;
223 			return res;
224 		}
225 		if (!php_stream_gets(dba->fp, buf, 15)) {
226 			break;
227 		}
228 		num = atoi(buf);
229 		if (num >= buf_size) {
230 			buf_size = num + FLATFILE_BLOCK_SIZE;
231 			buf = erealloc(buf, buf_size);
232 		}
233 		num = php_stream_read(dba->fp, buf, num);
234 	}
235 	efree(buf);
236 	res.dptr = NULL;
237 	res.dsize = 0;
238 	return res;
239 }
240 /* }}} */
241 
242 /* {{{ flatfile_nextkey
243  */
flatfile_nextkey(flatfile * dba)244 datum flatfile_nextkey(flatfile *dba) {
245 	datum res;
246 	size_t num;
247 	size_t buf_size = FLATFILE_BLOCK_SIZE;
248 	char *buf = emalloc(buf_size);
249 
250 	php_stream_seek(dba->fp, dba->CurrentFlatFilePos, SEEK_SET);
251 	while(!php_stream_eof(dba->fp)) {
252 		if (!php_stream_gets(dba->fp, buf, 15)) {
253 			break;
254 		}
255 		num = atoi(buf);
256 		if (num >= buf_size) {
257 			buf_size = num + FLATFILE_BLOCK_SIZE;
258 			buf = erealloc(buf, buf_size);
259 		}
260 		num = php_stream_read(dba->fp, buf, num);
261 
262 		if (!php_stream_gets(dba->fp, buf, 15)) {
263 			break;
264 		}
265 		num = atoi(buf);
266 		if (num >= buf_size) {
267 			buf_size = num + FLATFILE_BLOCK_SIZE;
268 			buf = erealloc(buf, buf_size);
269 		}
270 		num = php_stream_read(dba->fp, buf, num);
271 
272 		if (*(buf)!=0) {
273 			dba->CurrentFlatFilePos = php_stream_tell(dba->fp);
274 			res.dptr = buf;
275 			res.dsize = num;
276 			return res;
277 		}
278 	}
279 	efree(buf);
280 	res.dptr = NULL;
281 	res.dsize = 0;
282 	return res;
283 }
284 /* }}} */
285 
286 /* {{{ flatfile_version */
flatfile_version()287 char *flatfile_version()
288 {
289 	return "1.0, $Id: 464bb5af8f2e8462a17d20bf000e48ae24de5bd3 $";
290 }
291 /* }}} */
292 
293 /*
294  * Local variables:
295  * tab-width: 4
296  * c-basic-offset: 4
297  * End:
298  * vim600: sw=4 ts=4 fdm=marker
299  * vim<600: sw=4 ts=4
300  */
301