140 lines
4.6 KiB
C++
140 lines
4.6 KiB
C++
#pragma once
|
|
|
|
// solanaceae port of sha1 fts for NGCFT1
|
|
|
|
#include <solanaceae/object_store/object_store.hpp>
|
|
#include <solanaceae/contact/contact_model3.hpp>
|
|
#include <solanaceae/message3/registry_message_model.hpp>
|
|
#include <solanaceae/tox_contacts/tox_contact_model2.hpp>
|
|
|
|
#include <solanaceae/ngc_ft1/ngcft1.hpp>
|
|
|
|
#include "./ft1_sha1_info.hpp"
|
|
#include "./sending_transfers.hpp"
|
|
#include "./receiving_transfers.hpp"
|
|
|
|
#include "./backends/sha1_mapped_filesystem.hpp"
|
|
|
|
#include <entt/container/dense_map.hpp>
|
|
|
|
#include <random>
|
|
#include <chrono>
|
|
|
|
class SHA1_NGCFT1 : public ToxEventI, public RegistryMessageModelEventI, public ObjectStoreEventI, public NGCFT1EventI, public NGCEXTEventI {
|
|
ObjectStore2& _os;
|
|
ObjectStore2::SubscriptionReference _os_sr;
|
|
// TODO: backend abstraction
|
|
Contact3Registry& _cr;
|
|
RegistryMessageModelI& _rmm;
|
|
RegistryMessageModelI::SubscriptionReference _rmm_sr;
|
|
NGCFT1& _nft;
|
|
NGCFT1::SubscriptionReference _nft_sr;
|
|
ToxContactModel2& _tcm;
|
|
ToxEventProviderI& _tep;
|
|
ToxEventProviderI::SubscriptionReference _tep_sr;
|
|
NGCEXTEventProvider& _neep;
|
|
NGCEXTEventProvider::SubscriptionReference _neep_sr;
|
|
|
|
Backends::SHA1MappedFilesystem _mfb;
|
|
|
|
bool _object_update_lock {false};
|
|
|
|
std::minstd_rand _rng {1337*11};
|
|
|
|
using clock = std::chrono::steady_clock;
|
|
clock::time_point _time_start_offset {clock::now()};
|
|
float getTimeNow(void) const {
|
|
return std::chrono::duration<float>{clock::now() - _time_start_offset}.count();
|
|
}
|
|
|
|
// limit this to each group?
|
|
entt::dense_map<SHA1Digest, ObjectHandle> _info_to_content;
|
|
|
|
// sha1 chunk index
|
|
// TODO: optimize lookup
|
|
// TODO: multiple contents. hashes might be unique, but data is not
|
|
entt::dense_map<SHA1Digest, ObjectHandle> _chunks;
|
|
|
|
// group_number, peer_number, content, chunk_hash, timer
|
|
std::deque<std::tuple<uint32_t, uint32_t, ObjectHandle, SHA1Digest, float>> _queue_requested_chunk;
|
|
//void queueUpRequestInfo(uint32_t group_number, uint32_t peer_number, const SHA1Digest& hash);
|
|
void queueUpRequestChunk(uint32_t group_number, uint32_t peer_number, ObjectHandle content, const SHA1Digest& hash);
|
|
|
|
SendingTransfers _sending_transfers;
|
|
ReceivingTransfers _receiving_transfers;
|
|
|
|
// makes request rotate around open content
|
|
std::deque<ObjectHandle> _queue_content_want_info;
|
|
|
|
struct QBitsetEntry {
|
|
Contact3Handle c;
|
|
ObjectHandle o;
|
|
};
|
|
std::deque<QBitsetEntry> _queue_send_bitset;
|
|
|
|
// FIXME: workaround missing contact events
|
|
// only used on peer exit (no, also used to quicken lookups)
|
|
entt::dense_map<uint64_t, Contact3Handle> _tox_peer_to_contact;
|
|
|
|
// reset every iterate; kept here as an allocation optimization
|
|
entt::dense_map<Contact3, size_t> _peer_open_requests;
|
|
|
|
void updateMessages(ObjectHandle ce);
|
|
|
|
std::optional<std::pair<uint32_t, uint32_t>> selectPeerForRequest(ObjectHandle ce);
|
|
|
|
void queueBitsetSendFull(Contact3Handle c, ObjectHandle o);
|
|
|
|
File2I* objGetFile2Write(ObjectHandle o);
|
|
File2I* objGetFile2Read(ObjectHandle o);
|
|
|
|
public: // TODO: config
|
|
bool _udp_only {false};
|
|
|
|
size_t _max_concurrent_in {4}; // info only
|
|
size_t _max_concurrent_out {4*10}; // HACK: allow "ideal" number for 10 peers
|
|
|
|
public:
|
|
SHA1_NGCFT1(
|
|
ObjectStore2& os,
|
|
Contact3Registry& cr,
|
|
RegistryMessageModelI& rmm,
|
|
NGCFT1& nft,
|
|
ToxContactModel2& tcm,
|
|
ToxEventProviderI& tep,
|
|
NGCEXTEventProvider& neep
|
|
);
|
|
|
|
float iterate(float delta);
|
|
|
|
void onSendFileHashFinished(ObjectHandle o, Message3Registry* reg_ptr, Contact3 c, uint64_t ts);
|
|
|
|
// construct the file part in a partially constructed message
|
|
ObjectHandle constructFileMessageInPlace(Message3Handle msg, NGCFT1_file_kind file_kind, ByteSpan file_id);
|
|
|
|
protected: // rmm events (actions)
|
|
bool sendFilePath(const Contact3 c, std::string_view file_name, std::string_view file_path) override;
|
|
|
|
protected: // os events (actions)
|
|
bool onEvent(const ObjectStore::Events::ObjectUpdate&) override;
|
|
|
|
protected: // events
|
|
bool onEvent(const Events::NGCFT1_recv_request&) override;
|
|
bool onEvent(const Events::NGCFT1_recv_init&) override;
|
|
bool onEvent(const Events::NGCFT1_recv_data&) override;
|
|
bool onEvent(const Events::NGCFT1_send_data&) override; // const?
|
|
bool onEvent(const Events::NGCFT1_recv_done&) override;
|
|
bool onEvent(const Events::NGCFT1_send_done&) override;
|
|
bool onEvent(const Events::NGCFT1_recv_message&) override;
|
|
|
|
bool onToxEvent(const Tox_Event_Group_Peer_Join* e) override;
|
|
bool onToxEvent(const Tox_Event_Group_Peer_Exit* e) override;
|
|
|
|
bool onEvent(const Events::NGCEXT_ft1_have&) override;
|
|
bool onEvent(const Events::NGCEXT_ft1_bitset&) override;
|
|
bool onEvent(const Events::NGCEXT_ft1_have_all&) override;
|
|
|
|
bool onEvent(const Events::NGCEXT_pc1_announce&) override;
|
|
};
|
|
|