123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283 |
- /*
- +----------------------------------------------------------------------+
- | Copyright (c) The PHP Group |
- +----------------------------------------------------------------------+
- | This source file is subject to version 3.01 of the PHP license, |
- | that is bundled with this package in the file LICENSE, and is |
- | available through the world-wide-web at the following url: |
- | https://www.php.net/license/3_01.txt |
- | If you did not receive a copy of the PHP license and are unable to |
- | obtain it through the world-wide-web, please send a note to |
- | license@php.net so we can mail you a copy immediately. |
- +----------------------------------------------------------------------+
- | Authors: Marcus Boerger <helly@php.net> |
- | based on ext/db/db.c by: |
- | Rasmus Lerdorf <rasmus@php.net> |
- | Jim Winstead <jimw@php.net> |
- +----------------------------------------------------------------------+
- */
- /* $Id$ */
- #ifdef HAVE_CONFIG_H
- #include "config.h"
- #endif
- #include "php.h"
- #include "php_globals.h"
- #include <stdlib.h>
- #include <string.h>
- #include <errno.h>
- #if HAVE_UNISTD_H
- #include <unistd.h>
- #endif
- #include "flatfile.h"
- #define FLATFILE_BLOCK_SIZE 1024
- /*
- * ret = -1 means that database was opened for read-only
- * ret = 0 success
- * ret = 1 key already exists - nothing done
- */
- /* {{{ flatfile_store */
- int flatfile_store(flatfile *dba, datum key_datum, datum value_datum, int mode) {
- if (mode == FLATFILE_INSERT) {
- if (flatfile_findkey(dba, key_datum)) {
- return 1;
- }
- php_stream_seek(dba->fp, 0L, SEEK_END);
- php_stream_printf(dba->fp, "%zu\n", key_datum.dsize);
- php_stream_flush(dba->fp);
- if (php_stream_write(dba->fp, key_datum.dptr, key_datum.dsize) < key_datum.dsize) {
- return -1;
- }
- php_stream_printf(dba->fp, "%zu\n", value_datum.dsize);
- php_stream_flush(dba->fp);
- if (php_stream_write(dba->fp, value_datum.dptr, value_datum.dsize) < value_datum.dsize) {
- return -1;
- }
- } else { /* FLATFILE_REPLACE */
- flatfile_delete(dba, key_datum);
- php_stream_printf(dba->fp, "%zu\n", key_datum.dsize);
- php_stream_flush(dba->fp);
- if (php_stream_write(dba->fp, key_datum.dptr, key_datum.dsize) < key_datum.dsize) {
- return -1;
- }
- php_stream_printf(dba->fp, "%zu\n", value_datum.dsize);
- if (php_stream_write(dba->fp, value_datum.dptr, value_datum.dsize) < value_datum.dsize) {
- return -1;
- }
- }
- php_stream_flush(dba->fp);
- return 0;
- }
- /* }}} */
- /* {{{ flatfile_fetch */
- datum flatfile_fetch(flatfile *dba, datum key_datum) {
- datum value_datum = {NULL, 0};
- char buf[16];
- if (flatfile_findkey(dba, key_datum)) {
- if (php_stream_gets(dba->fp, buf, sizeof(buf))) {
- value_datum.dsize = atoi(buf);
- value_datum.dptr = safe_emalloc(value_datum.dsize, 1, 1);
- value_datum.dsize = php_stream_read(dba->fp, value_datum.dptr, value_datum.dsize);
- } else {
- value_datum.dptr = NULL;
- value_datum.dsize = 0;
- }
- }
- return value_datum;
- }
- /* }}} */
- /* {{{ flatfile_delete */
- int flatfile_delete(flatfile *dba, datum key_datum) {
- char *key = key_datum.dptr;
- size_t size = key_datum.dsize;
- size_t buf_size = FLATFILE_BLOCK_SIZE;
- char *buf = emalloc(buf_size);
- size_t num;
- size_t pos;
- php_stream_rewind(dba->fp);
- while(!php_stream_eof(dba->fp)) {
- /* read in the length of the key name */
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- pos = php_stream_tell(dba->fp);
- /* read in the key name */
- num = php_stream_read(dba->fp, buf, num);
- if (size == num && !memcmp(buf, key, size)) {
- php_stream_seek(dba->fp, pos, SEEK_SET);
- php_stream_putc(dba->fp, 0);
- php_stream_flush(dba->fp);
- php_stream_seek(dba->fp, 0L, SEEK_END);
- efree(buf);
- return SUCCESS;
- }
- /* read in the length of the value */
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- /* read in the value */
- num = php_stream_read(dba->fp, buf, num);
- }
- efree(buf);
- return FAILURE;
- }
- /* }}} */
- /* {{{ flatfile_findkey */
- int flatfile_findkey(flatfile *dba, datum key_datum) {
- size_t buf_size = FLATFILE_BLOCK_SIZE;
- char *buf = emalloc(buf_size);
- size_t num;
- int ret=0;
- void *key = key_datum.dptr;
- size_t size = key_datum.dsize;
- php_stream_rewind(dba->fp);
- while (!php_stream_eof(dba->fp)) {
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- if (size == num) {
- if (!memcmp(buf, key, size)) {
- ret = 1;
- break;
- }
- }
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- }
- efree(buf);
- return ret;
- }
- /* }}} */
- /* {{{ flatfile_firstkey */
- datum flatfile_firstkey(flatfile *dba) {
- datum res;
- size_t num;
- size_t buf_size = FLATFILE_BLOCK_SIZE;
- char *buf = emalloc(buf_size);
- php_stream_rewind(dba->fp);
- while(!php_stream_eof(dba->fp)) {
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- if (*(buf) != 0) {
- dba->CurrentFlatFilePos = php_stream_tell(dba->fp);
- res.dptr = buf;
- res.dsize = num;
- return res;
- }
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- }
- efree(buf);
- res.dptr = NULL;
- res.dsize = 0;
- return res;
- }
- /* }}} */
- /* {{{ flatfile_nextkey */
- datum flatfile_nextkey(flatfile *dba) {
- datum res;
- size_t num;
- size_t buf_size = FLATFILE_BLOCK_SIZE;
- char *buf = emalloc(buf_size);
- php_stream_seek(dba->fp, dba->CurrentFlatFilePos, SEEK_SET);
- while(!php_stream_eof(dba->fp)) {
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- if (!php_stream_gets(dba->fp, buf, 15)) {
- break;
- }
- num = atoi(buf);
- if (num >= buf_size) {
- buf_size = num + FLATFILE_BLOCK_SIZE;
- buf = erealloc(buf, buf_size);
- }
- num = php_stream_read(dba->fp, buf, num);
- if (*(buf)!=0) {
- dba->CurrentFlatFilePos = php_stream_tell(dba->fp);
- res.dptr = buf;
- res.dsize = num;
- return res;
- }
- }
- efree(buf);
- res.dptr = NULL;
- res.dsize = 0;
- return res;
- }
- /* }}} */
- /* {{{ flatfile_version */
- char *flatfile_version()
- {
- return "1.0, $Id$";
- }
- /* }}} */
|