aboutsummaryrefslogtreecommitdiff
path: root/src/protocol.h
diff options
context:
space:
mode:
authorKamil Domanski <kdomanski@kdemail.net>2014-08-20 08:42:31 +0200
committerKamil Domanski <kdomanski@kdemail.net>2014-08-31 02:14:20 +0200
commit3d796f89962842e91e7d88e57c1d2d579f01052e (patch)
treed4be736c23fbc4d453a1e4e281eafb4f51cdffd8 /src/protocol.h
parent9f3d47677973cb894fdbb437b9b322e2062a1bf1 (diff)
downloadbitcoin-3d796f89962842e91e7d88e57c1d2d579f01052e.tar.xz
overhaul serialization code
The implementation of each class' serialization/deserialization is no longer passed within a macro. The implementation now lies within a template of form: template <typename T, typename Stream, typename Operation> inline static size_t SerializationOp(T thisPtr, Stream& s, Operation ser_action, int nType, int nVersion) { size_t nSerSize = 0; /* CODE */ return nSerSize; } In cases when codepath should depend on whether or not we are just deserializing (old fGetSize, fWrite, fRead flags) an additional clause can be used: bool fRead = boost::is_same<Operation, CSerActionUnserialize>(); The IMPLEMENT_SERIALIZE macro will now be a freestanding clause added within class' body (similiar to Qt's Q_OBJECT) to implement GetSerializeSize, Serialize and Unserialize. These are now wrappers around the "SerializationOp" template.
Diffstat (limited to 'src/protocol.h')
-rw-r--r--src/protocol.h61
1 files changed, 38 insertions, 23 deletions
diff --git a/src/protocol.h b/src/protocol.h
index d7565584af..851fca9d9e 100644
--- a/src/protocol.h
+++ b/src/protocol.h
@@ -36,12 +36,16 @@ class CMessageHeader
bool IsValid() const;
IMPLEMENT_SERIALIZE
- (
- READWRITE(FLATDATA(pchMessageStart));
- READWRITE(FLATDATA(pchCommand));
- READWRITE(nMessageSize);
- READWRITE(nChecksum);
- )
+
+ template <typename T, typename Stream, typename Operation>
+ inline static size_t SerializationOp(T thisPtr, Stream& s, Operation ser_action, int nType, int nVersion) {
+ size_t nSerSize = 0;
+ READWRITE(FLATDATA(thisPtr->pchMessageStart));
+ READWRITE(FLATDATA(thisPtr->pchCommand));
+ READWRITE(thisPtr->nMessageSize);
+ READWRITE(thisPtr->nChecksum);
+ return nSerSize;
+ }
// TODO: make private (improves encapsulation)
public:
@@ -84,19 +88,26 @@ class CAddress : public CService
void Init();
IMPLEMENT_SERIALIZE
- (
- CAddress* pthis = const_cast<CAddress*>(this);
- CService* pip = (CService*)pthis;
- if (fRead)
- pthis->Init();
- if (nType & SER_DISK)
- READWRITE(nVersion);
- if ((nType & SER_DISK) ||
- (nVersion >= CADDR_TIME_VERSION && !(nType & SER_GETHASH)))
- READWRITE(nTime);
- READWRITE(nServices);
- READWRITE(*pip);
- )
+
+ template <typename T, typename Stream, typename Operation>
+ inline static size_t SerializationOp(T thisPtr, Stream& s, Operation ser_action, int nType, int nVersion) {
+ size_t nSerSize = 0;
+ bool fRead = boost::is_same<Operation, CSerActionUnserialize>();
+
+ CAddress* pthis = const_cast<CAddress*>(thisPtr);
+ CService* pip = (CService*)pthis;
+ if (fRead)
+ pthis->Init();
+ if (nType & SER_DISK)
+ READWRITE(nVersion);
+ if ((nType & SER_DISK) ||
+ (nVersion >= CADDR_TIME_VERSION && !(nType & SER_GETHASH)))
+ READWRITE(thisPtr->nTime);
+ READWRITE(thisPtr->nServices);
+ READWRITE(*pip);
+
+ return nSerSize;
+ }
// TODO: make private (improves encapsulation)
public:
@@ -118,10 +129,14 @@ class CInv
CInv(const std::string& strType, const uint256& hashIn);
IMPLEMENT_SERIALIZE
- (
- READWRITE(type);
- READWRITE(hash);
- )
+
+ template <typename T, typename Stream, typename Operation>
+ inline static size_t SerializationOp(T thisPtr, Stream& s, Operation ser_action, int nType, int nVersion) {
+ size_t nSerSize = 0;
+ READWRITE(thisPtr->type);
+ READWRITE(thisPtr->hash);
+ return nSerSize;
+ }
friend bool operator<(const CInv& a, const CInv& b);