X-Git-Url: http://git.bytex64.net/?a=blobdiff_plain;f=database%2Fdatabase.c;h=585f508c06df5d9f8912e630671e1f7aad2bd041;hb=96ec261b36bdbb701e05f5ee1aab70dec44085f9;hp=1e33811c1d29f13f638ff603fa2481f32f3c5a29;hpb=1cf0f1ebfbd5b5a621af2a49ac0328fd0cec4bf4;p=blerg.git

diff --git a/database/database.c b/database/database.c
index 1e33811..585f508 100644
--- a/database/database.c
+++ b/database/database.c
@@ -10,6 +10,7 @@
 #include <sys/file.h>
 #include <fcntl.h>
 #include "database.h"
+#include "util.h"
 #include "config.h"
 
 uint64_t blerg_get_record_count(struct blerg *blerg) {
@@ -111,8 +112,8 @@ int blerg_exists(const char *name) {
 	int namelen = strlen(name);
 	char filename[512];
 
-	if (namelen > 32) {
-		perror("Name too long");
+	if (!valid_name(name)) {
+		fprintf(stderr, "Invalid name\n");
 		return 0;
 	}
 
@@ -129,8 +130,8 @@ struct blerg *blerg_open(const char *name) {
 	struct stat st;
 	uint64_t sequence;
 
-	if (namelen > 32) {
-		perror("Name too long");
+	if (!valid_name(name)) {
+		fprintf(stderr, "Invalid name\n");
 		return NULL;
 	}
 	struct blerg *blerg = malloc(sizeof(struct blerg));
@@ -203,7 +204,7 @@ int blerg_close(struct blerg *blerg) {
 
 int blerg_store(struct blerg *blerg, const char *data, int len) {
 	if (len > MAX_RECORD_SIZE) {
-		printf("len > 64K\n");
+		fprintf(stderr, "len > 64K\n");
 		return -1;
 	}
 
@@ -212,7 +213,7 @@ int blerg_store(struct blerg *blerg, const char *data, int len) {
 
 	uint64_t record = blerg_increment_record_count(blerg);
 	if (record == -1) {
-		printf("Could not find free record\n");
+		fprintf(stderr, "Could not find free record\n");
 		return -1;
 	}
 	int segment = record / RECORDS_PER_SEGMENT;
@@ -252,7 +253,7 @@ int blerg_store(struct blerg *blerg, const char *data, int len) {
 
 int blerg_fetch(struct blerg *blerg, int record, char **data, int *length) {
 	if (record < 0) {
-		printf("Invalid record\n");
+		fprintf(stderr, "Invalid record\n");
 		return 0;
 	}
 
@@ -262,7 +263,7 @@ int blerg_fetch(struct blerg *blerg, int record, char **data, int *length) {
 	int seg_rec = record % RECORDS_PER_SEGMENT;
 
 	if ((blerg->index[seg_rec].flags & 0x1) == 0) {
-		printf("Invalid record\n");
+		fprintf(stderr, "Invalid record\n");
 		return 0;
 	}
 
@@ -275,7 +276,7 @@ int blerg_fetch(struct blerg *blerg, int record, char **data, int *length) {
 		fstat(blerg->data_fd, &st);
 		blerg->data_size = st.st_size;
 		if (rec_offset > blerg->data_size) {
-			printf("Record offset outside of data!?");
+			fprintf(stderr, "Record offset outside of data!?");
 			return 0;
 		}
 
@@ -302,7 +303,7 @@ int blerg_fetch(struct blerg *blerg, int record, char **data, int *length) {
 
 time_t blerg_get_timestamp(struct blerg *blerg, int record) {
 	if (record < 0) {
-		printf("Invalid record\n");
+		fprintf(stderr, "Invalid record\n");
 		return 0;
 	}
 
@@ -312,7 +313,7 @@ time_t blerg_get_timestamp(struct blerg *blerg, int record) {
 	int seg_rec = record % RECORDS_PER_SEGMENT;
 
 	if ((blerg->index[seg_rec].flags & 0x1) == 0) {
-		printf("Invalid record\n");
+		fprintf(stderr, "Invalid record\n");
 		return 0;
 	}