src/protocol.cpp

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Copyright (c) 2009-2014 The Bitcoin Core developers
   3 // Distributed under the MIT software license, see the accompanying
   4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5
   6 #include "protocol.h"
   7
   8 #include "util.h"
   9 #include "utilstrencodings.h"
  10
  11 #ifndef WIN32
  12 # include <arpa/inet.h>
  13 #endif
  14
  15 static const char* ppszTypeName[] =
  16 {
  17     "ERROR",
  18     "tx",
  19     "block",
  20     "filtered block"
  21 };
  22
  23 CMessageHeader::CMessageHeader(const MessageStartChars& pchMessageStartIn)
  24 {
  25     memcpy(pchMessageStart, pchMessageStartIn, MESSAGE_START_SIZE);
  26     memset(pchCommand, 0, sizeof(pchCommand));
  27     nMessageSize = -1;
  28     nChecksum = 0;
  29 }
  30
  31 CMessageHeader::CMessageHeader(const MessageStartChars& pchMessageStartIn, const char* pszCommand, unsigned int nMessageSizeIn)
  32 {
  33     memcpy(pchMessageStart, pchMessageStartIn, MESSAGE_START_SIZE);
  34     memset(pchCommand, 0, sizeof(pchCommand));
  35     strncpy(pchCommand, pszCommand, COMMAND_SIZE);
  36     nMessageSize = nMessageSizeIn;
  37     nChecksum = 0;
  38 }
  39
  40 std::string CMessageHeader::GetCommand() const
  41 {
  42     return std::string(pchCommand, pchCommand + strnlen(pchCommand, COMMAND_SIZE));
  43 }
  44
  45 bool CMessageHeader::IsValid(const MessageStartChars& pchMessageStartIn) const
  46 {
  47     // Check start string
  48     if (memcmp(pchMessageStart, pchMessageStartIn, MESSAGE_START_SIZE) != 0)
  49         return false;
  50
  51     // Check the command string for errors
  52     for (const char* p1 = pchCommand; p1 < pchCommand + COMMAND_SIZE; p1++)
  53     {
  54         if (*p1 == 0)
  55         {
  56             // Must be all zeros after the first zero
  57             for (; p1 < pchCommand + COMMAND_SIZE; p1++)
  58                 if (*p1 != 0)
  59                     return false;
  60         }
  61         else if (*p1 < ' ' || *p1 > 0x7E)
  62             return false;
  63     }
  64
  65     // Message size
  66     if (nMessageSize > MAX_SIZE)
  67     {
  68         LogPrintf("CMessageHeader::IsValid(): (%s, %u bytes) nMessageSize > MAX_SIZE\n", GetCommand(), nMessageSize);
  69         return false;
  70     }
  71
  72     return true;
  73 }
  74
  75
  76
  77 CAddress::CAddress() : CService()
  78 {
  79     Init();
  80 }
  81
  82 CAddress::CAddress(CService ipIn, uint64_t nServicesIn) : CService(ipIn)
  83 {
  84     Init();
  85     nServices = nServicesIn;
  86 }
  87
  88 void CAddress::Init()
  89 {
  90     nServices = NODE_NETWORK;
  91     nTime = 100000000;
  92     nLastTry = 0;
  93 }
  94
  95 CInv::CInv()
  96 {
  97     type = 0;
  98     hash.SetNull();
  99 }
 100
 101 CInv::CInv(int typeIn, const uint256& hashIn)
 102 {
 103     type = typeIn;
 104     hash = hashIn;
 105 }
 106
 107 CInv::CInv(const std::string& strType, const uint256& hashIn)
 108 {
 109     unsigned int i;
 110     for (i = 1; i < ARRAYLEN(ppszTypeName); i++)
 111     {
 112         if (strType == ppszTypeName[i])
 113         {
 114             type = i;
 115             break;
 116         }
 117     }
 118     if (i == ARRAYLEN(ppszTypeName))
 119         throw std::out_of_range(strprintf("CInv::CInv(string, uint256): unknown type '%s'", strType));
 120     hash = hashIn;
 121 }
 122
 123 bool operator<(const CInv& a, const CInv& b)
 124 {
 125     return (a.type < b.type || (a.type == b.type && a.hash < b.hash));
 126 }
 127
 128 bool CInv::IsKnownType() const
 129 {
 130     return (type >= 1 && type < (int)ARRAYLEN(ppszTypeName));
 131 }
 132
 133 const char* CInv::GetCommand() const
 134 {
 135     if (!IsKnownType())
 136         throw std::out_of_range(strprintf("CInv::GetCommand(): type=%d unknown type", type));
 137     return ppszTypeName[type];
 138 }
 139
 140 std::string CInv::ToString() const
 141 {
 142     return strprintf("%s %s", GetCommand(), hash.ToString());
 143 }