67 lines
2.3 KiB
C
67 lines
2.3 KiB
C
#include <pthread.h>
|
|
|
|
#include "ipfs/exchange/bitswap/want_manager.h"
|
|
#include "ipfs/exchange/bitswap/wantlist_queue.h"
|
|
|
|
/***
|
|
* Add a Cid to the wantlist
|
|
* @param context the context
|
|
* @param cid the Cid
|
|
* @returns the added entry
|
|
*/
|
|
struct WantListQueueEntry* ipfs_bitswap_want_manager_add(const struct BitswapContext* context, const struct Cid* cid, const struct WantListSession* session) {
|
|
// add if not there, and increment reference count
|
|
return ipfs_bitswap_wantlist_queue_add(context->localWantlist, cid, session);
|
|
}
|
|
|
|
/***
|
|
* Checks to see if the requested block has been received
|
|
* @param context the context
|
|
* @param cid the Cid
|
|
* @returns true(1) if it has been received, false(0) otherwise
|
|
*/
|
|
int ipfs_bitswap_want_manager_received(const struct BitswapContext* context, const struct Cid* cid) {
|
|
// find the entry
|
|
struct WantListQueueEntry* entry = ipfs_bitswap_wantlist_queue_find(context->localWantlist, cid);
|
|
// check the status
|
|
if (entry != NULL && entry->block != NULL) {
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***
|
|
* retrieve a block from the WantManager. NOTE: a call to want_manager_received should be done first
|
|
* @param context the context
|
|
* @param cid the Cid to get
|
|
* @param block a pointer to the block that will be allocated
|
|
* @returns true(1) on success, false(0) otherwise
|
|
*/
|
|
int ipfs_bitswap_want_manager_get_block(const struct BitswapContext* context, const struct Cid* cid, struct Block** block) {
|
|
struct WantListQueueEntry* entry = ipfs_bitswap_wantlist_queue_find(context->localWantlist, cid);
|
|
if (entry != NULL && entry->block != NULL) {
|
|
// return a copy of the block
|
|
*block = ipfs_block_copy(entry->block);
|
|
if ( (*block) != NULL) {
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/***
|
|
* We no longer are requesting this block, so remove it from the queue
|
|
* NOTE: This is reference counted, as another process may have asked for it.
|
|
* @param context the context
|
|
* @param cid the Cid
|
|
* @returns true(1) on success, false(0) otherwise.
|
|
*/
|
|
int ipfs_bitswap_want_manager_remove(const struct BitswapContext* context, const struct Cid* cid) {
|
|
// decrement the reference count
|
|
// if it is zero, remove the entry (lock first)
|
|
struct WantListSession session;
|
|
session.type = WANTLIST_SESSION_TYPE_LOCAL;
|
|
session.context = (void*) context->ipfsNode;
|
|
return ipfs_bitswap_wantlist_queue_remove(context->localWantlist, cid, &session);
|
|
}
|