/* */ #include "DHTRoutingTableSerializer.h" #include #include #include "DHTNode.h" #include "DlAbortEx.h" #include "DHTConstants.h" #include "bittorrent_helper.h" #include "Logger.h" #include "a2netcompat.h" #include "util.h" #include "TimeA2.h" namespace aria2 { DHTRoutingTableSerializer::DHTRoutingTableSerializer(int family): family_(family) {} DHTRoutingTableSerializer::~DHTRoutingTableSerializer() {} void DHTRoutingTableSerializer::setLocalNode (const SharedHandle& localNode) { localNode_ = localNode; } void DHTRoutingTableSerializer::setNodes (const std::vector >& nodes) { nodes_ = nodes; } void DHTRoutingTableSerializer::serialize(std::ostream& o) { char header[8]; memset(header, 0, sizeof(header)); // magic header[0] = 0xa1u; header[1] = 0xa2u; // format ID header[2] = 0x02u; // version header[6] = 0; header[7] = 0x03u; char zero[18]; memset(zero, 0, sizeof(zero)); o.write(header, 8); // write save date uint64_t ntime = hton64(Time().getTime()); o.write(reinterpret_cast(&ntime), sizeof(ntime)); // localnode // 8bytes reserved o.write(zero, 8); // 20bytes localnode ID o.write(reinterpret_cast(localNode_->getID()), DHT_ID_LENGTH); // 4bytes reserved o.write(zero, 4); // number of nodes uint32_t numNodes = htonl(nodes_.size()); o.write(reinterpret_cast(&numNodes), sizeof(uint32_t)); // 4bytes reserved o.write(zero, 4); const int clen = bittorrent::getCompactLength(family_); // nodes for(std::vector >::const_iterator i = nodes_.begin(), eoi = nodes_.end(); i != eoi; ++i) { const SharedHandle& node = *i; // Write IP address + port in Compact IP-address/port info form. unsigned char compactPeer[COMPACT_LEN_IPV6]; int compactlen = bittorrent::packcompact (compactPeer, node->getIPAddress(), node->getPort()); if(compactlen != clen) { memset(compactPeer, 0, clen); } // 1byte compact peer format length o << static_cast(clen); // 7bytes reserved o.write(zero, 7); // clen bytes compact peer o.write(reinterpret_cast(compactPeer), clen); // 24-clen bytes reserved o.write(zero, 24-clen); // 20bytes: node ID o.write(reinterpret_cast(node->getID()), DHT_ID_LENGTH); // 4bytes reserved o.write(zero, 4); } o.flush(); if(!o) { throw DL_ABORT_EX("Failed to save DHT routing table."); } } } // namespace aria2