From f8bff2594897605c4fa6013c6050369b734c75e9 Mon Sep 17 00:00:00 2001
From: Al <albarrentine@gmail.com>
Date: Sat, 20 Jun 2015 12:29:11 -0500
Subject: [PATCH] [bloom] bloom filter I/O

---
 src/bloom.c | 129 +++++++++++++++++++++++++++++++++++++++++++++++++++-
 src/bloom.h |   9 ++++
 2 files changed, 136 insertions(+), 2 deletions(-)

diff --git a/src/bloom.c b/src/bloom.c
index d4a76da2..1c3930bb 100644
--- a/src/bloom.c
+++ b/src/bloom.c
@@ -3,6 +3,8 @@
 #include "bloom.h"
 #include "murmur/murmur.h"
 
+#define BLOOM_FILTER_SIGNATURE 0xBABABABA
+
 #define LOG2_SQUARED 0.4804530139182014
 #define LOG2 0.6931471805599453
 
@@ -56,8 +58,9 @@ int bloom_filter_add(bloom_filter_t *self, const char *key, size_t len) {
 bloom_filter_t *bloom_filter_new(uint64_t capacity, double error) {
     bloom_filter_t *bloom = malloc(sizeof(bloom_filter_t));
 
-    if (bloom == NULL)
+    if (bloom == NULL) {
         return NULL;
+    }
 
     bloom->ready = false;
 
@@ -90,9 +93,131 @@ exit_free_bloom:
     return NULL;
 }
 
+bool bloom_filter_write(bloom_filter_t *self, FILE *f) {
+    if (!file_write_uint64(f, BLOOM_FILTER_SIGNATURE)) {
+        return false;
+    }
 
+    if (!file_write_uint64(f, self->capacity)) {
+        return false;
+    }
+
+    if (!file_write_double(f, self->error)) {
+        return false;
+    }
+
+    if (!file_write_uint64(f, self->num_bits)) {
+        return false;
+    }
+
+    if (!file_write_uint64(f, self->num_bytes)) {
+        return false;
+    }
+
+    if (!file_write_uint32(f, self->num_hashes)) {
+        return false;
+    }
+
+    if (!file_write_double(f, self->bits_per_entry)) {
+        return false;
+    }
+
+    if (!file_write_chars(f, (char *)self->filter, self->num_bytes)) {
+        return false;
+    }
+
+    return true;
+}
+
+
+
+bool bloom_filter_save(bloom_filter_t *self, char *path) {
+    FILE *f;
+    if ((f = fopen(path, "wb")) == NULL) {
+        return false;
+    }
+    bool status = bloom_filter_write(self, f);
+    fclose(f);
+    return status;
+}
+
+bloom_filter_t *bloom_filter_read(FILE *f) {
+    bloom_filter_t *bloom = malloc(sizeof(bloom_filter_t));
+
+    if (bloom == NULL) {
+        return NULL;
+    }
+
+    bloom->ready = false;
+
+    uint64_t signature = 0;
+
+    if (!file_read_uint64(f, &signature) || signature != BLOOM_FILTER_SIGNATURE) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_uint64(f, &bloom->capacity)) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_double(f, &bloom->error)) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_uint64(f, &bloom->num_bits)) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_uint64(f, &bloom->num_bytes)) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_uint32(f, &bloom->num_hashes)) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_double(f, &bloom->bits_per_entry)) {
+        goto exit_bloom_filter_created;
+    }
+
+    bloom->filter = calloc(bloom->num_bytes, sizeof(char));
+    if (bloom->filter == NULL) {
+        goto exit_bloom_filter_created;
+    }
+
+    if (!file_read_chars(f, (char *)bloom->filter, bloom->num_bytes)) {
+        goto exit_bloom_filter_created;
+    }
+
+    bloom->ready = true;
+
+    return bloom;
+
+exit_bloom_filter_created:
+    bloom_filter_destroy(bloom);
+    bloom = NULL;
+    return bloom;
+}
+
+
+bloom_filter_t *bloom_filter_load(char *path) {
+    FILE *f;
+
+    if ((f = fopen(path, "rb")) == NULL) {
+        return NULL;
+    }
+
+    bloom_filter_t *bloom = bloom_filter_read(f);
+    fclose(f);
+    return bloom;
+}
 
 void bloom_filter_destroy(bloom_filter_t *self) {
-    free(self->filter);
+    if (self == NULL) return;
+
+    if (self->filter != NULL) {
+        free(self->filter);
+    }
+
     free(self);
 }
\ No newline at end of file
diff --git a/src/bloom.h b/src/bloom.h
index 8bc45450..e7eba8fc 100644
--- a/src/bloom.h
+++ b/src/bloom.h
@@ -4,6 +4,8 @@
 #include <stdlib.h>
 #include <stdbool.h>
 
+#include "file_utils.h"
+
 #define SALT_CONSTANT 0x66e8c41d
 
 typedef struct bloom_filter {
@@ -28,4 +30,11 @@ void bloom_filter_print(bloom_filter_t *self);
 
 void bloom_filter_destroy(bloom_filter_t *self);
 
+bool bloom_filter_write(bloom_filter_t *self, FILE *f);
+bool bloom_filter_save(bloom_filter_t *self, char *path);
+
+bloom_filter_t *bloom_filter_read(FILE *f);
+bloom_filter_t *bloom_filter_load(char *path);
+
+
 #endif
\ No newline at end of file