forked from cawka/ndn.cxx
diff --git a/ndn-cpp/wire/ccnb.cc b/ndn-cpp/wire/ccnb.cc
new file mode 100644
index 0000000..86c3f5f
--- /dev/null
+++ b/ndn-cpp/wire/ccnb.cc
@@ -0,0 +1,259 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
+/*
+ * Copyright (c) 2013, Regents of the University of California
+ * Alexander Afanasyev
+ *
+ * BSD license, See the LICENSE file for more information
+ *
+ * Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
+ */
+
+#include "ccnb.h"
+#include "ndn-cpp/error.h"
+
+#include <boost/lexical_cast.hpp>
+
+namespace ndn {
+namespace wire {
+
+#define CCN_TT_BITS 3
+#define CCN_TT_MASK ((1 << CCN_TT_BITS) - 1)
+#define CCN_MAX_TINY ((1 << (7-CCN_TT_BITS)) - 1)
+#define CCN_TT_HBIT ((unsigned char)(1 << 7))
+
+void
+Ccnb::appendBlockHeader (std::ostream &os, size_t val, Ccnb::ccn_tt tt)
+{
+ unsigned char buf[1+8*((sizeof(val)+6)/7)];
+ unsigned char *p = &(buf[sizeof(buf)-1]);
+ size_t n = 1;
+ p[0] = (CCN_TT_HBIT & ~Ccnb::CCN_CLOSE_TAG) |
+ ((val & CCN_MAX_TINY) << CCN_TT_BITS) |
+ (CCN_TT_MASK & tt);
+ val >>= (7-CCN_TT_BITS);
+ while (val != 0) {
+ (--p)[0] = (((unsigned char)val) & ~CCN_TT_HBIT) | Ccnb::CCN_CLOSE_TAG;
+ n++;
+ val >>= 7;
+ }
+ os.write (reinterpret_cast<const char*> (p), n);
+ // return n;
+}
+
+void
+Ccnb::appendNumber (std::ostream &os, uint32_t number)
+{
+ std::string numberStr = boost::lexical_cast<std::string> (number);
+
+ appendBlockHeader (os, numberStr.size (), Ccnb::CCN_UDATA);
+ numberStr.size ();
+ os.write (numberStr.c_str (), numberStr.size ());
+}
+
+void
+Ccnb::appendName (std::ostream &os, const Name &name)
+{
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_Name, Ccnb::CCN_DTAG); // <Name>
+ for (Name::const_iterator component = name.begin (); component != name.end (); component ++)
+ {
+ appendTaggedBlob (os, Ccnb::CCN_DTAG_Component, component->buf (), component->size ());
+ }
+ Ccnb::appendCloser (os); // </Name>
+}
+
+void
+Ccnb::appendTimestampBlob (std::ostream &os, const TimeInterval &time)
+{
+ // CCNx method function implements some markers, which are not really defined anywhere else...
+
+ // Determine miminal number of bytes required to store the timestamp
+ int required_bytes = 2; // 12 bits for fractions of a second, 4 bits left for seconds. Sometimes it is enough
+ intmax_t ts = time.total_seconds () >> 4;
+ for (; required_bytes < 7 && ts != 0; ts >>= 8) // not more than 6 bytes?
+ required_bytes++;
+
+ appendBlockHeader(os, required_bytes, Ccnb::CCN_BLOB);
+
+ // write part with seconds
+ ts = time.total_seconds () >> 4;
+ for (int i = 0; i < required_bytes - 2; i++)
+ os.put ( ts >> (8 * (required_bytes - 3 - i)) );
+
+ /* arithmetic contortions are to avoid overflowing 31 bits */
+ ts = ((time.total_seconds () & 15) << 12) +
+ (((time.total_nanoseconds () % 1000000000) / 5 * 8 + 195312) / 390625);
+ for (int i = required_bytes - 2; i < required_bytes; i++)
+ os.put ( ts >> (8 * (required_bytes - 1 - i)) );
+
+ // return len + required_bytes;
+}
+
+void
+Ccnb::appendExclude (std::ostream &os, const Exclude &exclude)
+{
+ appendBlockHeader (os, Ccnb::CCN_DTAG_Exclude, Ccnb::CCN_DTAG); // <Exclude>
+
+ for (Exclude::const_reverse_iterator item = exclude.rbegin (); item != exclude.rend (); item ++)
+ {
+ if (!item->first.empty ())
+ appendTaggedBlob (os, Ccnb::CCN_DTAG_Component, item->first.buf (), item->first.size ());
+ if (item->second)
+ {
+ appendBlockHeader (os, Ccnb::CCN_DTAG_Any, Ccnb::CCN_DTAG); // <Any>
+ appendCloser (os); // </Any>
+ }
+ }
+ appendCloser (os); // </Exclude>
+}
+
+void
+Ccnb::appendInterest (std::ostream &os, const Interest &interest)
+{
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_Interest, Ccnb::CCN_DTAG); // <Interest>
+
+ // this is used for now as an interest template. Name should be empty
+ // Ccnb::appendName (os, interest.getName ());
+ Ccnb::appendName (os, Name ()); // <Component>...</Component>...
+
+ if (interest.getMinSuffixComponents () != Interest::ncomps)
+ {
+ appendTaggedNumber (os, Ccnb::CCN_DTAG_MinSuffixComponents, interest.getMinSuffixComponents ());
+ }
+ if (interest.getMaxSuffixComponents () != Interest::ncomps)
+ {
+ appendTaggedNumber (os, Ccnb::CCN_DTAG_MaxSuffixComponents, interest.getMaxSuffixComponents ());
+ }
+ if (interest.getExclude ().size () > 0)
+ {
+ appendExclude (os, interest.getExclude ());
+ }
+ if (interest.getChildSelector () != Interest::CHILD_DEFAULT)
+ {
+ appendTaggedNumber (os, Ccnb::CCN_DTAG_ChildSelector, interest.getChildSelector ());
+ }
+ if (interest.getAnswerOriginKind () != Interest::AOK_DEFAULT)
+ {
+ appendTaggedNumber (os, Ccnb::CCN_DTAG_AnswerOriginKind, interest.getAnswerOriginKind ());
+ }
+ if (interest.getScope () != Interest::NO_SCOPE)
+ {
+ appendTaggedNumber (os, Ccnb::CCN_DTAG_Scope, interest.getScope ());
+ }
+ if (!interest.getInterestLifetime ().is_negative ())
+ {
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_InterestLifetime, Ccnb::CCN_DTAG);
+ Ccnb::appendTimestampBlob (os, interest.getInterestLifetime ());
+ Ccnb::appendCloser (os);
+ }
+ // if (GetNonce()>0)
+ // {
+ // uint32_t nonce = interest.GetNonce();
+ // appendTaggedBlob (start, Ccnb::CCN_DTAG_Nonce, nonce);
+ // }
+
+ // if (GetNack ()>0)
+ // {
+ // appendBlockHeader (start, Ccnb::CCN_DTAG_Nack, Ccnb::CCN_DTAG);
+ // appendNumber (start, interest.GetNack ());
+ // appendCloser (start);
+ // }
+ Ccnb::appendCloser (os); // </Interest>
+}
+
+static void *SIGNATURE_Block = 0;
+static void *SINATURE_INFO_PublisherPublicKeyDigest = reinterpret_cast<void *> (1);
+static void *SINATURE_INFO_KeyLocator = reinterpret_cast<void *> (2);
+
+static const char TYPES [][3] = {
+ {0x0C, 0x04, 0xC0},
+ {0x10, 0xD0, 0x91},
+ {0x18, 0xE3, 0x44},
+ {0x28, 0x46, 0x3F},
+ {0x2C, 0x83, 0x4A},
+ {0x34, 0x00, 0x8A}
+};
+
+void
+Ccnb::appendSignature (std::ostream &os, const signature::Sha256WithRsa &signature, void *userData)
+{
+ if (userData == SIGNATURE_Block)
+ {
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_Signature, Ccnb::CCN_DTAG); // <Signature>
+ // if (signature.getDigestAlgorithm () != "2.16.840.1.101.3.4.2.1")
+ // {
+ // appendString (os, Ccnb::CCN_DTAG_DigestAlgorithm, signature.getDigestAlgorithm ());
+ // }
+ appendTaggedBlob (os, Ccnb::CCN_DTAG_SignatureBits, signature.getSignatureBits ());
+ Ccnb::appendCloser (os); // </Signature>
+ }
+ else if (userData == SINATURE_INFO_PublisherPublicKeyDigest)
+ {
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_PublisherPublicKeyDigest, signature.getPublisherKeyDigest ());
+ }
+ else if (userData == SINATURE_INFO_KeyLocator)
+ {
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_Signature, Ccnb::CCN_DTAG); // <Signature>
+ switch (signature.getKeyLocator ().getType ())
+ {
+ case KeyLocator::NOTSET:
+ break;
+ case KeyLocator::KEY:
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_Key, signature.getKeyLocator ().getKey ());
+ break;
+ case KeyLocator::CERTIFICATE:
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_Key, signature.getKeyLocator ().getCertificate ());
+ break;
+ case KeyLocator::KEYNAME:
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_KeyName, Ccnb::CCN_DTAG); // <KeyName>
+ Ccnb::appendName (os, signature.getKeyLocator ().getKeyName ());
+ Ccnb::appendCloser (os); // </KeyName>
+ break;
+ }
+ Ccnb::appendCloser (os); // </Signature>
+ }
+ // other cases should not be possible, but don't do anything
+}
+
+void
+Ccnb::appendData (std::ostream &os, const Data &data)
+{
+ if (!data.getSignature ())
+ BOOST_THROW_EXCEPTION (error::wire::Ccnb ()
+ << error::msg ("Signature is required, but not set"));
+
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_ContentObject, Ccnb::CCN_DTAG); // <ContentObject>
+
+ // necessary for now, because of the changed storage order
+ data.getSignature ()->doubleDispatch (os, *this, SIGNATURE_Block);
+
+ Ccnb::appendName (os, data.getName ());
+
+ Ccnb::appendBlockHeader (os, Ccnb::CCN_DTAG_SignedInfo, Ccnb::CCN_DTAG); // <SignedInfo>
+ data.getSignature ()->doubleDispatch (os, *this, SINATURE_INFO_PublisherPublicKeyDigest);
+
+ Ccnb::appendTimestampBlob (os, data.getContent ().getTimestamp ());
+
+ BOOST_ASSERT (sizeof (TYPES) == 3 * (static_cast<int> (Content::NACK)+1));
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_Type, TYPES [data.getContent ().getType ()], 3);
+
+ if (data.getContent ().getFreshness () != Content::noFreshness)
+ {
+ Ccnb::appendTaggedNumber (os, Ccnb::CCN_DTAG_FreshnessSeconds,
+ data.getContent ().getFreshness ().total_seconds ());
+ }
+
+ if (data.getContent ().getFinalBlockId () != Content::noFinalBlock)
+ {
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_FinalBlockID, data.getContent ().getFinalBlockId ());
+ }
+
+ data.getSignature ()->doubleDispatch (os, *this, SINATURE_INFO_KeyLocator);
+ Ccnb::appendCloser (os); // </SignedInfo>
+
+ Ccnb::appendTaggedBlob (os, Ccnb::CCN_DTAG_Content, data.content ());
+
+ Ccnb::appendCloser (os); // </ContentObject>
+}
+
+} // namespace wire
+} // namespace ndn