130 lines
3.6 KiB
C
130 lines
3.6 KiB
C
#include <stdlib.h>
|
|
|
|
#include "../test_helper.h"
|
|
#include "ipfs/routing/routing.h"
|
|
#include "ipfs/repo/fsrepo/fs_repo.h"
|
|
|
|
void stop_kademlia(void);
|
|
|
|
int test_routing_supernode_start() {
|
|
int retVal = 0;
|
|
struct FSRepo* fs_repo = NULL;
|
|
struct IpfsNode* ipfs_node = NULL;
|
|
struct Stream* stream = NULL;
|
|
|
|
if (!drop_build_and_open_repo("/tmp/.ipfs", &fs_repo))
|
|
goto exit;
|
|
|
|
ipfs_node = (struct IpfsNode*)malloc(sizeof(struct IpfsNode));
|
|
ipfs_node->mode = MODE_ONLINE;
|
|
ipfs_node->identity = fs_repo->config->identity;
|
|
ipfs_node->repo = fs_repo;
|
|
ipfs_node->routing = ipfs_routing_new_kademlia(ipfs_node, &fs_repo->config->identity->private_key, stream);
|
|
|
|
if (ipfs_node->routing == NULL)
|
|
goto exit;
|
|
|
|
//TODO ping kademlia
|
|
|
|
retVal = 1;
|
|
exit:
|
|
if (ipfs_node->routing != NULL)
|
|
stop_kademlia();
|
|
return retVal;
|
|
}
|
|
|
|
int test_routing_supernode_get_value() {
|
|
int retVal = 0;
|
|
struct FSRepo* fs_repo = NULL;
|
|
struct IpfsNode* ipfs_node = NULL;
|
|
struct Stream* stream = NULL;
|
|
int file_size = 1000;
|
|
unsigned char bytes[file_size];
|
|
char* fileName = "temp_file.bin";
|
|
char* fullFileName = "/tmp/temp_file.bin";
|
|
struct Node* write_node = NULL;
|
|
size_t bytes_written = 0;
|
|
unsigned char base58Hash[100];
|
|
size_t results_size = 2048;
|
|
char results_buffer[results_size];
|
|
|
|
if (!drop_build_and_open_repo("/tmp/.ipfs", &fs_repo))
|
|
goto exit;
|
|
|
|
ipfs_node = (struct IpfsNode*)malloc(sizeof(struct IpfsNode));
|
|
ipfs_node->mode = MODE_ONLINE;
|
|
ipfs_node->identity = fs_repo->config->identity;
|
|
ipfs_node->repo = fs_repo;
|
|
ipfs_node->routing = ipfs_routing_new_kademlia(ipfs_node, &fs_repo->config->identity->private_key, stream);
|
|
|
|
if (ipfs_node->routing == NULL)
|
|
goto exit;
|
|
|
|
// create a file
|
|
create_bytes(&bytes[0], file_size);
|
|
create_file(fullFileName, bytes, file_size);
|
|
|
|
// write to ipfs
|
|
if (ipfs_import_file("/tmp", fileName, &write_node, fs_repo, &bytes_written, 1) == 0) {
|
|
goto exit;
|
|
}
|
|
|
|
if (!ipfs_node->routing->Provide(ipfs_node->routing, (char*)write_node->data, write_node->data_size))
|
|
goto exit;
|
|
// write_node->hash has the base32 key of the file. Convert this to a base58.
|
|
if (!ipfs_cid_hash_to_base58(write_node->hash, write_node->hash_size, base58Hash, 100))
|
|
goto exit;
|
|
|
|
// ask the network who can provide this
|
|
if (!ipfs_node->routing->FindProviders(ipfs_node->routing, (char*)base58Hash, 100, &results_buffer[0], &results_size))
|
|
goto exit;
|
|
|
|
// Q: What should FindProviders have in the results buffer? A: A struct of:
|
|
// 20 byte id
|
|
// 4 byte (or 16 byte for ip6) ip address
|
|
// 2 byte port number
|
|
// that means we have to attempt a connection and ask for peer ID
|
|
// TODO: Q: How do we determine ip4 vs ip6?
|
|
|
|
struct Libp2pLinkedList* multiaddress_head;
|
|
// get an IP4 ip and port
|
|
if (!ipfs_routing_supernode_parse_provider(results_buffer, results_size, &multiaddress_head))
|
|
goto exit;
|
|
|
|
struct Libp2pLinkedList* current_address = multiaddress_head;
|
|
struct MultiAddress* addr = NULL;
|
|
while (current_address != NULL) {
|
|
addr = (struct Multiaddress*)current_address->item;
|
|
if (multiaddress_is_ip4(addr))
|
|
break;
|
|
addr = NULL;
|
|
current_address = current_address->next;
|
|
}
|
|
|
|
if (addr == NULL)
|
|
goto exit;
|
|
|
|
// Connect to server
|
|
char* ip;
|
|
multiaddress_get_ip_address(addr, &ip);
|
|
struct Stream* file_stream = libp2p_net_multistream_connect(ip, multiaddress_get_ip_port(addr));
|
|
|
|
// Switch from multistream to NodeIO
|
|
if (!libp2p_nodeio_upgrade_stream(file_stream))
|
|
goto exit;
|
|
|
|
// Ask for file
|
|
struct Node* node = libp2p_nodeio_get(file_stream, base58Hash, 100);
|
|
if (node == NULL)
|
|
goto exit;
|
|
|
|
retVal = 1;
|
|
exit:
|
|
if (ipfs_node->routing != NULL)
|
|
stop_kademlia();
|
|
if (fs_repo != NULL)
|
|
ipfs_repo_fsrepo_free(fs_repo);
|
|
return retVal;
|
|
|
|
}
|