dandri/i2pd
1
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2026-10-06 05:37:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

update piece's buffer after it gets loaded

Browse Source
This commit is contained in:
orignal
2026-09-15 08:52:27 -04:00
parent 42db1cf177
commit 79a4d7496b
2 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libi2pd_client/Torrents.cpp
+24 -2
View File
@@ -313,14 +313,20 @@ namespace torrents
bool Piece::Load (PieceFileFragment&& fragment)
{
if (fragment.fragmentOffset + fragment.fragmentSize > m_Size) return false;
if (!m_Data) NewDataBuffer ();
if (fragment.file->Load (fragment.fileOffset, m_Data + fragment.fragmentOffset, fragment.fragmentSize))
uint8_t * data = m_Data;
if (!data) data = new uint8_t[m_Size];
bool success = fragment.file->Load (fragment.fileOffset, data + fragment.fragmentOffset, fragment.fragmentSize);
UpdateDataBuffer (data);
if (success)
{
LogPrint (eLogDebug, "Torrents: Loaded bytes ", fragment.fileOffset, " - ", fragment.fileOffset + fragment.fragmentSize - 1, " from ", fragment.file->fullFilePath);
return true;
}
else
{
LogPrint (eLogError, "Torrents: Failed to load bytes ", fragment.fileOffset, " - ", fragment.fileOffset + fragment.fragmentSize - 1, " from ", fragment.file->fullFilePath);
return false;
}
}
void Piece::NewDataBuffer ()
@@ -345,6 +351,22 @@ namespace torrents
#endif
}
void Piece::UpdateDataBuffer (uint8_t * newData)
{
#if defined(__cpp_lib_atomic_ref)
std::atomic_ref<uint8_t *> data (m_Data);
auto old = data.exchange (newData);
if (old && old != newData) delete[] old;
#else
if (newData != m_Data)
{
auto old = m_Data;
m_Data = newData; delete[] old;
}
#endif
}
bool Piece::HasBlock (size_t offset) const
{
if (offset >= m_Size) return false;
libi2pd_client/Torrents.h
+1
View File
@@ -163,6 +163,7 @@ namespace torrents
size_t GetNumBlocks (size_t len) const;
void NewDataBuffer ();
void DeleteDataBuffer ();
void UpdateDataBuffer (uint8_t * newData);
private: