c-ipfs/test/merkledag/test_merkledag.h

#include "ipfs/merkledag/merkledag.h"
#include "ipfs/node/node.h"
#include "../test_helper.h"

struct FSRepo* createAndOpenRepo(const char* dir) {
	int retVal = 1;
	// create a fresh repo
	retVal = drop_and_build_repository(dir);
	if (retVal == 0)
		return NULL;

	// open the fs repo
	struct RepoConfig* repo_config = NULL;
	struct FSRepo* fs_repo;

	// create the struct
	retVal = ipfs_repo_fsrepo_new(dir, repo_config, &fs_repo);
	if (retVal == 0)
		return NULL;

	// open the repository and read the config file
	retVal = ipfs_repo_fsrepo_open(fs_repo);
	if (retVal == 0) {
		ipfs_repo_fsrepo_free(fs_repo);
		return NULL;
	}
	return fs_repo;
}

int test_merkledag_get_data() {
	int retVal = 0;

	struct FSRepo* fs_repo = createAndOpenRepo("/tmp/.ipfs");

	// create data for node
	size_t binary_data_size = 256;
	unsigned char binary_data[binary_data_size];
	for(int i = 0; i < binary_data_size; i++) {
		binary_data[i] = i;
	}

	// create a node
	struct Node* node1;
	retVal = ipfs_node_new_from_data(binary_data, binary_data_size, &node1);

	retVal = ipfs_merkledag_add(node1, fs_repo);
	if (retVal == 0) {
		ipfs_node_free(node1);
		ipfs_repo_fsrepo_free(fs_repo);
		return 0;
	}

	// now retrieve it
	struct Node* results_node;
	retVal = ipfs_merkledag_get(node1->cached, &results_node, fs_repo);
	if (retVal == 0) {
		ipfs_node_free(node1);
		ipfs_node_free(results_node);
		ipfs_repo_fsrepo_free(fs_repo);
		return 0;
	}

	if (results_node->data_size != 256) {
		ipfs_node_free(node1);
		ipfs_node_free(results_node);
		ipfs_repo_fsrepo_free(fs_repo);
		return 0;
	}

	// the data should be the same
	for(int i = 0; i < results_node->data_size; i++) {
		if (results_node->data[i] != node1->data[i]) {
			ipfs_node_free(node1);
			ipfs_node_free(results_node);
			ipfs_repo_fsrepo_free(fs_repo);
			return 0;
		}
	}

	ipfs_node_free(node1);
	ipfs_node_free(results_node);
	ipfs_repo_fsrepo_free(fs_repo);

	return retVal;
}

int test_merkledag_add_data() {
	int retVal = 0;

	struct FSRepo* fs_repo = createAndOpenRepo("/tmp/.ipfs");
	if (fs_repo == NULL)
		return 0;

	// get the size of the database
	int start_file_size = os_utils_file_size("/tmp/.ipfs/datastore/data.mdb");

	// create data for node
	size_t binary_data_size = 256;
	unsigned char binary_data[binary_data_size];
	for(int i = 0; i < binary_data_size; i++) {
		binary_data[i] = i;
	}

	// create a node
	struct Node* node1;
	retVal = ipfs_node_new_from_data(binary_data, binary_data_size, &node1);

	retVal = ipfs_merkledag_add(node1, fs_repo);
	if (retVal == 0) {
		ipfs_node_free(node1);
		return 0;
	}

	// make sure everything is correct
	if (node1->cached == NULL)
		return 0;

	int first_add_size = os_utils_file_size("/tmp/.ipfs/datastore/data.mdb");
	if (first_add_size == start_file_size) { // uh oh, database should have increased in size
		ipfs_node_free(node1);
		return 0;
	}

	// adding the same binary again should do nothing (the hash should be the same)
	struct Node* node2;
	retVal = ipfs_node_new_from_data(binary_data, binary_data_size, &node2);
	retVal = ipfs_merkledag_add(node2, fs_repo);
	if (retVal == 0) {
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		return 0;
	}

	// make sure everything is correct
	if (node2->cached == NULL) {
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		return 0;
	}
	for(int i = 0; i < node1->cached->hash_length; i++) {
		if (node1->cached->hash[i] != node2->cached->hash[i]) {
			printf("hash of node1 does not match node2 at position %d\n", i);
			ipfs_node_free(node1);
			ipfs_node_free(node2);
			return 0;
		}
	}

	int second_add_size = os_utils_file_size("/tmp/.ipfs/datastore/data.mdb");
	if (first_add_size != second_add_size) { // uh oh, the database shouldn't have changed size
		printf("looks as if a new record was added when it shouldn't have. Old file size: %d, new file size: %d\n", first_add_size, second_add_size);
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		return 0;
	}

	// now change 1 byte, which should change the hash
	binary_data[10] = 0;
	// create a node
	struct Node* node3;
	retVal = ipfs_node_new_from_data(binary_data, binary_data_size, &node3);

	retVal = ipfs_merkledag_add(node3, fs_repo);
	if (retVal == 0) {
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		ipfs_node_free(node3);
		return 0;
	}

	// make sure everything is correct
	if (node3->cached == NULL) {
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		ipfs_node_free(node3);
		return 0;
	}

	ipfs_node_free(node1);
	ipfs_node_free(node2);
	ipfs_node_free(node3);
	int third_add_size = os_utils_file_size("/tmp/.ipfs/datastore/data.mdb");
	if (third_add_size == second_add_size || third_add_size < second_add_size) {// uh oh, it didn't add it
		printf("Node 3 should have been added, but the file size did not change from %d.\n", third_add_size);
		return 0;
	}

	ipfs_repo_fsrepo_free(fs_repo);

	return 1;
}

/**
 * Should save links
 */
int test_merkledag_add_node_with_links() {
	int retVal = 0;
	struct Link* link = NULL;
	struct Node* node1 = NULL;

	struct FSRepo* fs_repo = createAndOpenRepo("/tmp/.ipfs");
	if (fs_repo == NULL) {
		printf("Unable to create repo\n");
		return 0;
	}

	// make link
	retVal = ipfs_node_link_new("", "abc123", &link);
	retVal = ipfs_node_new_from_link(link, &node1);

	retVal = ipfs_merkledag_add(node1, fs_repo);
	if (retVal == 0) {
		ipfs_repo_fsrepo_free(fs_repo);
		ipfs_node_free(node1);
		printf("Unable to add node\n");
		return 0;
	}

	// now look for it
	struct Node* node2 = NULL;
	retVal = ipfs_merkledag_get(node1->cached, &node2, fs_repo);
	if (retVal == 0) {
		ipfs_repo_fsrepo_free(fs_repo);
		ipfs_node_free(node1);
		return 0;
	}

	if (node2->link_amount != node1->link_amount) {
		printf("Link amount %d does not match %d\n", node2->link_amount, node1->link_amount);
		ipfs_repo_fsrepo_free(fs_repo);
		ipfs_node_free(node1);
		ipfs_node_free(node2);
		return 0;
	}

	// make sure hashes match
	for(int i = 0; i < node1->links[0]->cid->hash_length; i++) {
		if(node1->links[0]->cid->hash[i] != node2->links[0]->cid->hash[i]) {
			printf("Hashes do not match\n");
			ipfs_repo_fsrepo_free(fs_repo);
			ipfs_node_free(node1);
			ipfs_node_free(node2);
			return 0;
		}
	}

	ipfs_node_free(node1);
	ipfs_node_free(node2);
	ipfs_repo_fsrepo_free(fs_repo);

	return 1;
}