145 lines
4.2 KiB
C
145 lines
4.2 KiB
C
#pragma once
|
|
/***
|
|
* A queue for requests to/from remote peers
|
|
* NOTE: This must handle multiple threads
|
|
*/
|
|
|
|
#include <pthread.h>
|
|
#include "libp2p/peer/peer.h"
|
|
#include "ipfs/exchange/bitswap/bitswap.h"
|
|
#include "ipfs/blocks/block.h"
|
|
|
|
struct CidEntry {
|
|
struct Cid* cid;
|
|
int cancel;
|
|
int cancel_has_been_sent;
|
|
int request_has_been_sent;
|
|
};
|
|
|
|
struct PeerRequest {
|
|
pthread_mutex_t request_mutex;
|
|
struct Libp2pPeer* peer;
|
|
// CidEntry collection of cids that they want
|
|
struct Libp2pVector* cids_they_want;
|
|
// CidEntry collection of cids that we want or are canceling
|
|
struct Libp2pVector* cids_we_want;
|
|
// blocks to send to them
|
|
struct Libp2pVector* blocks_we_want_to_send;
|
|
// blocks they sent us are processed immediately, so no queue necessary
|
|
// although the cid can go in cids_we_want again, with a cancel flag
|
|
};
|
|
|
|
struct PeerRequestEntry {
|
|
struct PeerRequestEntry* prior;
|
|
struct PeerRequest* current;
|
|
struct PeerRequestEntry* next;
|
|
};
|
|
|
|
struct PeerRequestQueue {
|
|
pthread_mutex_t queue_mutex;
|
|
struct PeerRequestEntry* first;
|
|
struct PeerRequestEntry* last;
|
|
};
|
|
|
|
/***
|
|
* Allocate memory for CidEntry
|
|
* @returns new CidEntry struct
|
|
*/
|
|
struct CidEntry* ipfs_bitswap_peer_request_cid_entry_new();
|
|
|
|
/**
|
|
* Allocate resources for a new PeerRequest
|
|
* @returns a new PeerRequest struct or NULL if there was a problem
|
|
*/
|
|
struct PeerRequest* ipfs_bitswap_peer_request_new();
|
|
|
|
/**
|
|
* Free resources from a PeerRequest
|
|
* @param request the request to free
|
|
* @returns true(1)
|
|
*/
|
|
int ipfs_bitswap_peer_request_free(struct PeerRequest* request);
|
|
|
|
/**
|
|
* Allocate resources for a new queue
|
|
* @returns a new PeerRequestQueue
|
|
*/
|
|
struct PeerRequestQueue* ipfs_bitswap_peer_request_queue_new();
|
|
|
|
/**
|
|
* Free all resources related to the queue
|
|
* @param queue the queue
|
|
* @returns true(1)
|
|
*/
|
|
int ipfs_bitswap_peer_request_queue_free(struct PeerRequestQueue* queue);
|
|
|
|
/**
|
|
* Adds a peer request to the end of the queue
|
|
* @param queue the queue
|
|
* @param request the request
|
|
* @returns true(1) on success, otherwise false
|
|
*/
|
|
int ipfs_bitswap_peer_request_queue_add(struct PeerRequestQueue* queue, struct PeerRequest* request);
|
|
|
|
/**
|
|
* Removes a peer request from the queue, no mather where it is
|
|
* @param queue the queue
|
|
* @param request the request
|
|
* @returns true(1) on success, otherwise false(0)
|
|
*/
|
|
int ipfs_bitswap_peer_request_quque_remove(struct PeerRequestQueue* queue, struct PeerRequest* request);
|
|
|
|
/**
|
|
* Pull a PeerRequest off the queue
|
|
* @param queue the queue
|
|
* @returns the PeerRequest that should be handled next.
|
|
*/
|
|
struct PeerRequest* ipfs_bitswap_peer_request_queue_pop(struct PeerRequestQueue* queue);
|
|
|
|
/**
|
|
* Finds a PeerRequestEntry that contains the specified PeerRequest
|
|
* @param queue the queue
|
|
* @param request what we're looking for
|
|
* @returns the PeerRequestEntry or NULL if not found
|
|
*/
|
|
struct PeerRequestEntry* ipfs_bitswap_peer_request_queue_find_entry(struct PeerRequestQueue* queue, struct Libp2pPeer* peer);
|
|
|
|
/***
|
|
* Add a block to the appropriate peer's queue
|
|
* @param queue the queue
|
|
* @param who the session context that identifies the peer
|
|
* @param block the block
|
|
* @returns true(1) on success, otherwise false(0)
|
|
*/
|
|
int ipfs_bitswap_peer_request_queue_fill(struct PeerRequestQueue* queue, struct Libp2pPeer* who, struct Block* block);
|
|
|
|
/***
|
|
* Allocate resources for a PeerRequestEntry struct
|
|
* @returns the allocated struct or NULL if there was a problem
|
|
*/
|
|
struct PeerRequestEntry* ipfs_bitswap_peer_request_entry_new();
|
|
|
|
/**
|
|
* Frees resources allocated
|
|
* @param entry the PeerRequestEntry to free
|
|
* @returns true(1)
|
|
*/
|
|
int ipfs_bitswap_peer_request_entry_free(struct PeerRequestEntry* entry);
|
|
|
|
/****
|
|
* Handle a PeerRequest
|
|
* @param context the BitswapContext
|
|
* @param request the request to process
|
|
* @returns true(1) on succes, otherwise false(0)
|
|
*/
|
|
int ipfs_bitswap_peer_request_process_entry(const struct BitswapContext* context, struct PeerRequest* request);
|
|
|
|
/***
|
|
* Find a PeerRequest related to a peer. If one is not found, it is created.
|
|
*
|
|
* @param peer_request_queue the queue to look through
|
|
* @param peer the peer to look for
|
|
* @returns a PeerRequestEntry or NULL on error
|
|
*/
|
|
struct PeerRequest* ipfs_peer_request_queue_find_peer(struct PeerRequestQueue* queue, struct Libp2pPeer* peer);
|
|
|