128 lines
3.8 KiB
C
128 lines
3.8 KiB
C
/**
|
|
* A basic storage building block of the IPFS system
|
|
*/
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "libp2p/crypto/sha256.h"
|
|
#include "mh/multihash.h"
|
|
#include "mh/hashes.h"
|
|
#include "ipfs/merkledag/merkledag.h"
|
|
#include "ipfs/unixfs/unixfs.h"
|
|
|
|
/***
|
|
* Convert a HashtableNode into a Block
|
|
* @param node the node to convert
|
|
* @param blockResult where to put the results
|
|
* @returns true(1) on success, false(0) otherwise
|
|
*/
|
|
int ipfs_merkledag_convert_node_to_block(struct HashtableNode* node, struct Block** blockResult) {
|
|
*blockResult = ipfs_block_new();
|
|
if (*blockResult == NULL)
|
|
return 0;
|
|
struct Block* block = *blockResult;
|
|
block->cid = ipfs_cid_new(1, node->hash, node->hash_size, CID_DAG_PROTOBUF);
|
|
if (block->cid == NULL) {
|
|
ipfs_block_free(block);
|
|
*blockResult = NULL;
|
|
return 0;
|
|
}
|
|
block->data_length = ipfs_hashtable_node_protobuf_encode_size(node);
|
|
block->data = malloc(block->data_length);
|
|
if (block->data == NULL) {
|
|
ipfs_block_free(block);
|
|
*blockResult = NULL;
|
|
return 0;
|
|
}
|
|
if (!ipfs_hashtable_node_protobuf_encode(node, block->data, block->data_length, &block->data_length)) {
|
|
ipfs_block_free(block);
|
|
*blockResult = NULL;
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/***
|
|
* Adds a node to the dagService and blockService
|
|
* @param node the node to add
|
|
* @param fs_repo the repo to add to
|
|
* @param bytes_written the number of bytes written
|
|
* @returns true(1) on success
|
|
*/
|
|
int ipfs_merkledag_add(struct HashtableNode* node, struct FSRepo* fs_repo, size_t* bytes_written) {
|
|
// taken from merkledag.go line 59
|
|
int retVal = 0;
|
|
|
|
// compute the hash if necessary
|
|
if (node->hash == NULL) {
|
|
size_t protobuf_size = ipfs_hashtable_node_protobuf_encode_size(node);
|
|
unsigned char protobuf[protobuf_size];
|
|
size_t bytes_encoded;
|
|
retVal = ipfs_hashtable_node_protobuf_encode(node, protobuf, protobuf_size, &bytes_encoded);
|
|
|
|
node->hash_size = 32;
|
|
node->hash = (unsigned char*)malloc(node->hash_size);
|
|
if (node->hash == NULL) {
|
|
return 0;
|
|
}
|
|
if (libp2p_crypto_hashing_sha256(protobuf, bytes_encoded, &node->hash[0]) == 0) {
|
|
free(node->hash);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
// write to block store & datastore
|
|
struct Block* block = NULL;
|
|
if (!ipfs_merkledag_convert_node_to_block(node, &block)) {
|
|
return 0;
|
|
}
|
|
retVal = ipfs_repo_fsrepo_block_write(block, fs_repo, bytes_written);
|
|
if (retVal == 0) {
|
|
ipfs_block_free(block);
|
|
return 0;
|
|
}
|
|
|
|
// TODO: call HasBlock (unsure why as yet)
|
|
return 1;
|
|
}
|
|
|
|
/***
|
|
* Retrieves a node from the datastore based on the hash
|
|
* @param hash the key to look for
|
|
* @param hash_size the length of the key
|
|
* @param node the node to be created
|
|
* @param fs_repo the repository
|
|
* @returns true(1) on success
|
|
*/
|
|
int ipfs_merkledag_get(const unsigned char* hash, size_t hash_size, struct HashtableNode** node, const struct FSRepo* fs_repo) {
|
|
int retVal = 1;
|
|
struct DatastoreRecord* datastore_record = NULL;
|
|
|
|
// look for the node in the datastore. If it is not there, it is not a node.
|
|
// If it exists, it is only a block.
|
|
retVal = fs_repo->config->datastore->datastore_get(hash, hash_size, &datastore_record, fs_repo->config->datastore);
|
|
if (retVal == 0)
|
|
return 0;
|
|
|
|
libp2p_datastore_record_free(datastore_record);
|
|
|
|
// we have the record from the db. Go get the node from the blockstore
|
|
if (!ipfs_repo_fsrepo_node_read(hash, hash_size, node, fs_repo))
|
|
return 0;
|
|
|
|
// set the hash
|
|
ipfs_hashtable_node_set_hash(*node, hash, hash_size);
|
|
|
|
return 1;
|
|
}
|
|
|
|
int ipfs_merkledag_get_by_multihash(const unsigned char* multihash, size_t multihash_length, struct HashtableNode** node, const struct FSRepo* fs_repo) {
|
|
// convert to hash
|
|
size_t hash_size = 0;
|
|
unsigned char* hash = NULL;
|
|
if (mh_multihash_digest(multihash, multihash_length, &hash, &hash_size) < 0) {
|
|
return 0;
|
|
}
|
|
return ipfs_merkledag_get(hash, hash_size, node, fs_repo);
|
|
}
|