protocol.cpp

Go to the documentation of this file.

// Copyright (c) 2009-2010 Satoshi Nakamoto
// Copyright (c) 2009-2014 The Bitcoin developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#include "protocol.h"
 
#include "chainparams.h"
#include "util.h"
#include "utilstrencodings.h"
 
#ifndef WIN32
#include <arpa/inet.h>
#endif
 
namespace NetMsgType {
const char *VERSION="version";
const char *VERACK="verack";
const char *ADDR="addr";
const char *INV="inv";
const char *GETDATA="getdata";
const char *MERKLEBLOCK="merkleblock";
const char *GETBLOCKS="getblocks";
const char *GETHEADERS="getheaders";
const char *TX="tx";
const char *DSC="dsc";
const char *DSF="dsf";
const char *DSQ="dsq";
const char *DSR="dsr";
const char *DSTX="dstx";
const char *DSEE="dsee";
const char *DSEG="dseg";
const char *DSEEP="dseep";
const char *DSSU="dssu";
const char *HEADERS="headers";
const char *BLOCK="block";
const char *GETADDR="getaddr";
const char *MEMPOOL="mempool";
const char *PING="ping";
const char *PONG="pong";
const char *ALERT="alert";
const char *NOTFOUND="notfound";
const char *FILTERLOAD="filterload";
const char *FILTERADD="filteradd";
const char *FILTERCLEAR="filterclear";
const char *REJECT="reject";
const char *SENDHEADERS="sendheaders";
const char *IX="ix";
const char *IXLOCKVOTE="txlvote";
const char *MNBROADCAST="mnb";
const char *MNPING="mnp";
const char *MNWINNER="mnw";
const char *GETMNWINNERS="mnget";
const char *BUDGETPROPOSAL="mprop";
const char *BUDGETVOTE="mvote";
const char *BUDGETVOTESYNC="mnvs";
const char *FINALBUDGET="fbs";
const char *FINALBUDGETVOTE="fbvote";
const char *SYNCSTATUSCOUNT="ssc";
};
 
static const char* ppszTypeName[] =
{
    "ERROR", // Should never occur
    NetMsgType::TX,
    NetMsgType::DSC,
    NetMsgType::DSF,
    NetMsgType::DSQ,
    NetMsgType::DSR,
    NetMsgType::DSEE,
    NetMsgType::DSEG,
    NetMsgType::DSEEP,
    NetMsgType::DSSU,
    NetMsgType::HEADERS,
    NetMsgType::BLOCK,
    "filtered block", // Should never occur
    NetMsgType::IX,
    NetMsgType::IXLOCKVOTE,
    NetMsgType::MNBROADCAST,
    NetMsgType::MNPING,
    NetMsgType::MNWINNER,
    NetMsgType::GETMNWINNERS,
    NetMsgType::BUDGETPROPOSAL,
    NetMsgType::BUDGETVOTE,
    NetMsgType::FINALBUDGET,
    NetMsgType::FINALBUDGETVOTE
};
 
const static std::string allNetMessageTypes[] = {
    NetMsgType::VERSION,
    NetMsgType::VERACK,
    NetMsgType::ADDR,
    NetMsgType::INV,
    NetMsgType::GETDATA,
    NetMsgType::MERKLEBLOCK,
    NetMsgType::GETBLOCKS,
    NetMsgType::GETHEADERS,
    NetMsgType::TX,
    NetMsgType::DSC,
    NetMsgType::DSF,
    NetMsgType::DSQ,
    NetMsgType::DSR,
    NetMsgType::DSEE,
    NetMsgType::DSEG,
    NetMsgType::DSEEP,
    NetMsgType::DSSU,
    NetMsgType::HEADERS,
    NetMsgType::BLOCK,
    NetMsgType::GETADDR,
    NetMsgType::MEMPOOL,
    NetMsgType::PING,
    NetMsgType::PONG,
    NetMsgType::ALERT,
    NetMsgType::NOTFOUND,
    NetMsgType::FILTERLOAD,
    NetMsgType::FILTERADD,
    NetMsgType::FILTERCLEAR,
    NetMsgType::REJECT,
    NetMsgType::SENDHEADERS,
    NetMsgType::IX,
    NetMsgType::IXLOCKVOTE,
    NetMsgType::MNBROADCAST,
    NetMsgType::MNPING,
    NetMsgType::MNWINNER,
    NetMsgType::GETMNWINNERS,
    NetMsgType::BUDGETPROPOSAL,
    NetMsgType::BUDGETVOTE,
    NetMsgType::BUDGETVOTESYNC,
    NetMsgType::FINALBUDGET,
    NetMsgType::FINALBUDGETVOTE,
    NetMsgType::SYNCSTATUSCOUNT
};
const static std::vector<std::string> allNetMessageTypesVec(allNetMessageTypes, allNetMessageTypes+ARRAYLEN(allNetMessageTypes));
 
CMessageHeader::CMessageHeader()
{
    memcpy(pchMessageStart, Params().MessageStart(), MESSAGE_START_SIZE);
    memset(pchCommand, 0, sizeof(pchCommand));
    nMessageSize = -1;
    nChecksum = 0;
}
 
CMessageHeader::CMessageHeader(const char* pszCommand, unsigned int nMessageSizeIn)
{
    memcpy(pchMessageStart, Params().MessageStart(), MESSAGE_START_SIZE);
    memset(pchCommand, 0, sizeof(pchCommand));
    strncpy(pchCommand, pszCommand, COMMAND_SIZE);
    nMessageSize = nMessageSizeIn;
    nChecksum = 0;
}
 
std::string CMessageHeader::GetCommand() const
{
    return std::string(pchCommand, pchCommand + strnlen(pchCommand, COMMAND_SIZE));
}
 
bool CMessageHeader::IsValid() const
{
    // Check start string
    if (memcmp(pchMessageStart, Params().MessageStart(), MESSAGE_START_SIZE) != 0)
        return false;
 
    // Check the command string for errors
    for (const char* p1 = pchCommand; p1 < pchCommand + COMMAND_SIZE; p1++) {
        if (*p1 == 0) {
            // Must be all zeros after the first zero
            for (; p1 < pchCommand + COMMAND_SIZE; p1++)
                if (*p1 != 0)
                    return false;
        } else if (*p1 < ' ' || *p1 > 0x7E)
            return false;
    }
 
    // Message size
    if (nMessageSize > MAX_SIZE) {
        LogPrintf("CMessageHeader::IsValid() : (%s, %u bytes) nMessageSize > MAX_SIZE\n", GetCommand(), nMessageSize);
        return false;
    }
 
    return true;
}
 
 
CAddress::CAddress() : CService()
{
    Init();
}
 
CAddress::CAddress(CService ipIn, ServiceFlags nServicesIn) : CService(ipIn)
{
    Init();
    nServices = nServicesIn;
}
 
void CAddress::Init()
{
    nServices = NODE_NONE;
    nTime = 100000000;
}
 
CInv::CInv()
{
    type = 0;
    hash.SetNull();
}
 
CInv::CInv(int typeIn, const uint256& hashIn)
{
    type = typeIn;
    hash = hashIn;
}
 
CInv::CInv(const std::string& strType, const uint256& hashIn)
{
    unsigned int i;
    for (i = 1; i < ARRAYLEN(ppszTypeName); i++) {
        if (strType == ppszTypeName[i]) {
            type = i;
            break;
        }
    }
    if (i == ARRAYLEN(ppszTypeName))
        LogPrint(BCLog::NET, "CInv::CInv(string, uint256) : unknown type '%s'", strType);
    hash = hashIn;
}
 
bool operator<(const CInv& a, const CInv& b)
{
    return (a.type < b.type || (a.type == b.type && a.hash < b.hash));
}
 
bool CInv::IsKnownType() const
{
    return (type >= 1 && type < (int)ARRAYLEN(ppszTypeName));
}
 
bool CInv::IsMasterNodeType() const{
    return (type >= 6);
}
 
const char* CInv::GetCommand() const
{
    if (!IsKnownType()) {
        LogPrint(BCLog::NET, "CInv::GetCommand() : type=%d unknown type", type);
        return "UNKNOWN";
    }
 
    return ppszTypeName[type];
}
 
std::string CInv::ToString() const
{
    return strprintf("%s %s", GetCommand(), hash.ToString());
}
 
const std::vector<std::string> &getAllNetMessageTypes()
{
    return allNetMessageTypesVec;
}