You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.8 KiB
69 lines
1.8 KiB
/*
|
|
This file is part of cpp-ethereum.
|
|
|
|
cpp-ethereum is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
cpp-ethereum is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
/** @file Capability.h
|
|
* @author Gav Wood <i@gavwood.com>
|
|
* @date 2014
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "Common.h"
|
|
#include "HostCapability.h"
|
|
|
|
namespace dev
|
|
{
|
|
namespace p2p
|
|
{
|
|
|
|
class ReputationManager;
|
|
|
|
class Capability: public std::enable_shared_from_this<Capability>
|
|
{
|
|
friend class Session;
|
|
|
|
public:
|
|
Capability(std::shared_ptr<Session> _s, HostCapabilityFace* _h, unsigned _idOffset);
|
|
virtual ~Capability() {}
|
|
|
|
// Implement these in the derived class.
|
|
/* static std::string name() { return ""; }
|
|
static u256 version() { return 0; }
|
|
static unsigned messageCount() { return 0; }
|
|
*/
|
|
std::shared_ptr<Session> session() const { return m_session.lock(); }
|
|
HostCapabilityFace* hostCapability() const { return m_hostCap; }
|
|
Host* host() const { return m_hostCap->host(); }
|
|
ReputationManager& repMan() const;
|
|
|
|
protected:
|
|
virtual bool interpret(unsigned _id, RLP const&) = 0;
|
|
|
|
void disable(std::string const& _problem);
|
|
|
|
RLPStream& prep(RLPStream& _s, unsigned _id, unsigned _args = 0);
|
|
void sealAndSend(RLPStream& _s);
|
|
void addRating(int _r);
|
|
|
|
private:
|
|
std::weak_ptr<Session> m_session;
|
|
HostCapabilityFace* m_hostCap;
|
|
bool m_enabled = true;
|
|
unsigned m_idOffset;
|
|
};
|
|
|
|
}
|
|
}
|
|
|