X-Git-Url: https://jasonwoof.com/gitweb/?p=wfpl.git;a=blobdiff_plain;f=fdb.php;h=91892f930ff6db25636643fc6670a926fa38af59;hp=4e289727f8e7cf3dd6a5e5e217ca4943fc6623fb;hb=HEAD;hpb=7b70792307cbb5ff5b4442c366e1aa0498e34e9c

diff --git a/fdb.php b/fdb.php
index 4e28972..91892f9 100644
--- a/fdb.php
+++ b/fdb.php
@@ -1,29 +1,17 @@
 <?php
 
-#  Copyright (C) 2007 Jason Woofenden
-#
-#  This file is part of wfpl.
-#
-#  wfpl is free software; you can redistribute it and/or modify it under the
-#  terms of the GNU Lesser General Public License as published by the Free
-#  Software Foundation; either version 2.1 of the License, or (at your option)
-#  any later version.
-#
-#  wfpl is distributed in the hope that it will be useful, but WITHOUT ANY
-#  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-#  FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for
-#  more details.
-#
-#  You should have received a copy of the GNU Lesser General Public License
-#  along with wfpl; if not, write to the Free Software Foundation, Inc., 51
-#  Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+# This program is in the public domain within the United States. Additionally,
+# we waive copyright and related rights in the work worldwide through the CC0
+# 1.0 Universal public domain dedication, which can be found at
+# http://creativecommons.org/publicdomain/zero/1.0/
 
 
 # This file contains code to use a web-writeable directory full of files as a
 # very simple database.
 
 # Keys are truncated to 32 bytes, made lowercase, and all characters that are
-# not alpha/numeric are replaced with underscores.
+# not alpha/numeric are replaced with underscores. Periods and hyphens are only
+# replaced if they are at the begining.
 
 # Data can be either a string or an array.
 
@@ -31,6 +19,9 @@
 # fdb_set_dir() passing the path to that directory.
 
 
+require_once(__DIR__.'/'.'file.php');
+require_once(__DIR__.'/'.'binary.php');
+
 # call this to set what directory is used to store the files
 function fdb_set_dir($dir) {
 	$GLOBALS['fdb_dir'] = $dir;
@@ -38,35 +29,19 @@ function fdb_set_dir($dir) {
 
 function fdb_get_dir() {
 	if(!isset($GLOBALS['fdb_dir'])) {
-		die('you must call fdb_set_dir() before calling other functions in code/wfpl/fdb.php');
+		die('you must call fdb_set_dir() before calling other functions in wfpl/fdb.php');
 	}
 	return $GLOBALS['fdb_dir'];
 }
 
-# return a 4 bytes that represent the passed integer as a big-endian binary number
-function to_raw_int($int) {
-	return chr($int >> 24) . chr(($int >> 16) & 0xff) . chr(($int >> 8) & 0xff) . chr($int & 0xff);
-}
-
-# return a php number from the string you pass in. The first 4 bytes of the
-# string are read in as a binary value in big-endian format.
-function from_raw_int($quad) {
-	return (ord(substr($quad, 0, 1)) << 24) + (ord(substr($quad, 1, 1)) << 16) + (ord(substr($quad, 2, 1)) << 8) + ord(substr($quad, 3, 1));
-}
-
-function int_at($string, $index) {
-	return from_raw_int(substr($string, $index * 4, 4));
-}
 
-# remove the first 4 bytes of the string, and return them as an int
-function pop_int($&string) {
-	$int = from_raw_int(substring($string, 0 4));
-	$string = substr($string, 4);
-	return $int;
+function fdb_fix_key($key) {
+	$key = preg_replace('|[^a-z0-9.-]|', '_', strtolower($key));
+	$key = preg_replace('|^[-.]|', '_', strtolower($key));
+	return substr($key, 0, 32);
 }
 
 
-
 function fdb_get_raw($key) {
 	$key = fdb_fix_key($key);
 	return read_whole_file_or_false(fdb_get_dir() . "/$key");
@@ -80,18 +55,11 @@ function fdb_set_raw($key, $data) {
 # like fdb_get() except it returns an array even when there's just one element
 function fdb_geta($key) {
 	$key = fdb_fix_key($key);
-	$data = fd_get_raw($key);
+	$data = fdb_get_raw($key);
 	if($data === false) {
 		return false;
 	}
-	$header_count = pop_int($data);
-	$out = array();
-	while($header_count--) {
-		$size = int_at($data);
-		$out[] = substr($data, 0, $size);
-		$data = substr($data, $size);
-	}
-	return $out;
+	return raw_to_array($data);
 }
 
 # returns:
@@ -104,6 +72,9 @@ function fdb_geta($key) {
 # array in this case too)
 function fdb_get($key) {
 	$ret = fdb_geta($key);
+	if($ret == false) {
+		return false;
+	}
 	if(count($ret) == 1) {
 		return $ret[0];
 	} else {
@@ -112,17 +83,19 @@ function fdb_get($key) {
 }
 
 # data can be a string or array
-function fdb_put($key, $data) {
+function fdb_set($key, $data) {
 	$key = fdb_fix_key($key);
 	if(!is_array($data)) {
 		$data = array($data);
 	}
-	$out = to_raw_int(count($data));
-	foreach($data as $dat) {
-		$out .= to_raw_int(strlen($dat));
-		$out .= $dat;
-	}
-	return $out;
+	fdb_set_raw($key, array_to_raw($data));
 }
 
-?>
+function fdb_delete($key) {
+	$key = fdb_fix_key($key);
+	$path = fdb_get_dir() . "/$key";
+	if(file_exists($path)) {
+		return unlink($path);
+	}
+	return false;
+}