src/protocol.cpp

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