6 #ifndef BITCOIN_STREAMS_H
7 #define BITCOIN_STREAMS_H
25 template<
typename Stream>
52 void write(
const char* pch,
size_t nSize)
57 void read(
char* pch,
size_t nSize)
96 template <
typename... Args>
97 CVectorWriter(
int nTypeIn,
int nVersionIn, std::vector<unsigned char>& vchDataIn,
size_t nPosIn, Args&&... args) :
CVectorWriter(nTypeIn, nVersionIn, vchDataIn, nPosIn)
101 void write(
const char* pch,
size_t nSize)
104 size_t nOverwrite = std::min(nSize,
vchData.size() -
nPos);
106 memcpy(
vchData.data() +
nPos,
reinterpret_cast<const unsigned char*
>(pch), nOverwrite);
108 if (nOverwrite < nSize) {
109 vchData.insert(
vchData.end(),
reinterpret_cast<const unsigned char*
>(pch) + nOverwrite,
reinterpret_cast<const unsigned char*
>(pch) + nSize);
169 Init(nTypeIn, nVersionIn);
174 Init(nTypeIn, nVersionIn);
177 CDataStream(
const char* pbegin,
const char* pend,
int nTypeIn,
int nVersionIn) :
vch(pbegin, pend)
179 Init(nTypeIn, nVersionIn);
184 Init(nTypeIn, nVersionIn);
189 Init(nTypeIn, nVersionIn);
194 Init(nTypeIn, nVersionIn);
197 template <
typename... Args>
200 Init(nTypeIn, nVersionIn);
204 void Init(
int nTypeIn,
int nVersionIn)
226 return (std::string(
begin(),
end()));
249 void insert(
iterator it, std::vector<char>::const_iterator first, std::vector<char>::const_iterator last)
251 if (last == first)
return;
252 assert(last - first > 0);
260 vch.insert(it, first, last);
265 if (last == first)
return;
266 assert(last - first > 0);
274 vch.insert(it, first, last);
286 return vch.erase(
vch.begin(),
vch.end());
291 return vch.erase(it);
299 if (last ==
vch.end())
302 return vch.erase(
vch.begin(),
vch.end());
311 return vch.erase(first, last);
342 void read(
char* pch,
size_t nSize)
344 if (nSize == 0)
return;
347 unsigned int nReadPosNext =
nReadPos + nSize;
348 if (nReadPosNext >=
vch.size())
350 if (nReadPosNext >
vch.size())
352 throw std::ios_base::failure(
"CDataStream::read(): end of data");
367 throw std::ios_base::failure(
"CDataStream::ignore(): nSize negative");
369 unsigned int nReadPosNext =
nReadPos + nSize;
370 if (nReadPosNext >=
vch.size())
372 if (nReadPosNext >
vch.size())
373 throw std::ios_base::failure(
"CDataStream::ignore(): end of data");
381 void write(
const char* pch,
size_t nSize)
384 vch.insert(
vch.end(), pch, pch + nSize);
387 template<
typename Stream>
392 s.write((
char*)&
vch[0],
vch.size() *
sizeof(
vch[0]));
421 void Xor(
const std::vector<unsigned char>& key)
423 if (key.size() == 0) {
508 void read(
char* pch,
size_t nSize)
511 throw std::ios_base::failure(
"CAutoFile::read: file handle is NULL");
512 if (fread(pch, 1, nSize,
file) != nSize)
513 throw std::ios_base::failure(feof(
file) ?
"CAutoFile::read: end of file" :
"CAutoFile::read: fread failed");
519 throw std::ios_base::failure(
"CAutoFile::ignore: file handle is NULL");
520 unsigned char data[4096];
522 size_t nNow = std::min<size_t>(nSize,
sizeof(data));
523 if (fread(data, 1, nNow,
file) != nNow)
524 throw std::ios_base::failure(feof(
file) ?
"CAutoFile::ignore: end of file" :
"CAutoFile::read: fread failed");
529 void write(
const char* pch,
size_t nSize)
532 throw std::ios_base::failure(
"CAutoFile::write: file handle is NULL");
533 if (fwrite(pch, 1, nSize,
file) != nSize)
534 throw std::ios_base::failure(
"CAutoFile::write: write failed");
542 throw std::ios_base::failure(
"CAutoFile::operator<<: file handle is NULL");
552 throw std::ios_base::failure(
"CAutoFile::operator>>: file handle is NULL");
585 unsigned int readNow =
vchBuf.size() - pos;
587 if (nAvail < readNow)
593 throw std::ios_base::failure(feof(
src) ?
"CBufferedFile::Fill: end of file" :
"CBufferedFile::Fill: fread failed");
601 CBufferedFile(FILE *fileIn, uint64_t nBufSize, uint64_t nRewindIn,
int nTypeIn,
int nVersionIn) :
629 void read(
char *pch,
size_t nSize) {
631 throw std::ios_base::failure(
"Read attempted past buffer limit");
633 throw std::ios_base::failure(
"Read larger than buffer size");
639 if (nNow + pos >
vchBuf.size())
640 nNow =
vchBuf.size() - pos;
670 long nLongPos = nPos;
671 if (nPos != (uint64_t)nLongPos)
673 if (fseek(
src, nLongPos, SEEK_SET))
675 nLongPos = ftell(
src);
Non-refcounted RAII wrapper for FILE*.
FILE * release()
Get wrapped FILE* with transfer of ownership.
void ignore(size_t nSize)
FILE * Get() const
Get wrapped FILE* without transfer of ownership.
CAutoFile(const CAutoFile &)
CAutoFile(FILE *filenew, int nTypeIn, int nVersionIn)
CAutoFile & operator=(const CAutoFile &)
bool IsNull() const
Return true if the wrapped FILE* is NULL, false otherwise.
void write(const char *pch, size_t nSize)
CAutoFile & operator<<(const T &obj)
CAutoFile & operator>>(T &obj)
void read(char *pch, size_t nSize)
Non-refcounted RAII wrapper around a FILE* that implements a ring buffer to deserialize from.
CBufferedFile(FILE *fileIn, uint64_t nBufSize, uint64_t nRewindIn, int nTypeIn, int nVersionIn)
std::vector< char > vchBuf
void read(char *pch, size_t nSize)
CBufferedFile & operator=(const CBufferedFile &)
CBufferedFile(const CBufferedFile &)
bool SetPos(uint64_t nPos)
CBufferedFile & operator>>(T &obj)
bool SetLimit(uint64_t nPos=(uint64_t)(-1))
Double ended buffer combining vector and stream-like interfaces.
CDataStream(const_iterator pbegin, const_iterator pend, int nTypeIn, int nVersionIn)
iterator insert(iterator it, const char &x=char())
const_iterator begin() const
vector_type::allocator_type allocator_type
vector_type::reference reference
CDataStream(const vector_type &vchIn, int nTypeIn, int nVersionIn)
CDataStream(int nTypeIn, int nVersionIn)
CDataStream(const std::vector< char > &vchIn, int nTypeIn, int nVersionIn)
void GetAndClear(CSerializeData &data)
CDataStream(int nTypeIn, int nVersionIn, Args &&... args)
void reserve(size_type n)
vector_type::value_type value_type
CDataStream & operator+=(const CDataStream &b)
void Xor(const std::vector< unsigned char > &key)
XOR the contents of this stream with a certain key.
CSerializeData vector_type
const_iterator end() const
iterator erase(iterator first, iterator last)
void read(char *pch, size_t nSize)
vector_type::size_type size_type
const_reference operator[](size_type pos) const
reference operator[](size_type pos)
vector_type::reverse_iterator reverse_iterator
void Init(int nTypeIn, int nVersionIn)
void insert(iterator it, const char *first, const char *last)
void Serialize(Stream &s) const
vector_type::difference_type difference_type
void insert(iterator it, std::vector< char >::const_iterator first, std::vector< char >::const_iterator last)
void resize(size_type n, value_type c=0)
CDataStream(const char *pbegin, const char *pend, int nTypeIn, int nVersionIn)
vector_type::const_iterator const_iterator
vector_type::iterator iterator
const value_type * data() const
CDataStream(const std::vector< unsigned char > &vchIn, int nTypeIn, int nVersionIn)
CDataStream & operator>>(T &obj)
void insert(iterator it, size_type n, const char &x)
iterator erase(iterator it)
vector_type::const_reference const_reference
friend CDataStream operator+(const CDataStream &a, const CDataStream &b)
void write(const char *pch, size_t nSize)
CDataStream & operator<<(const T &obj)
CVectorWriter & operator<<(const T &obj)
void write(const char *pch, size_t nSize)
CVectorWriter(int nTypeIn, int nVersionIn, std::vector< unsigned char > &vchDataIn, size_t nPosIn, Args &&... args)
CVectorWriter(int nTypeIn, int nVersionIn, std::vector< unsigned char > &vchDataIn, size_t nPosIn)
std::vector< unsigned char > & vchData
OverrideStream(Stream *stream_, int nType_, int nVersion_)
OverrideStream< Stream > & operator>>(T &obj)
void read(char *pch, size_t nSize)
OverrideStream< Stream > & operator<<(const T &obj)
void write(const char *pch, size_t nSize)
void * memcpy(void *a, const void *b, size_t c)
void SerializeMany(Stream &s)
void Serialize(Stream &s, char a)
void Unserialize(Stream &s, char &a)
OverrideStream< S > WithOrVersion(S *s, int nVersionFlag)
std::vector< char, zero_after_free_allocator< char > > CSerializeData