276 lines
7.9 KiB
Rust
276 lines
7.9 KiB
Rust
// Copyright 2015, 2016 Ethcore (UK) Ltd.
|
|
// This file is part of Parity.
|
|
|
|
// Parity 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.
|
|
|
|
// Parity 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 Parity. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
use std::collections::BTreeMap;
|
|
use std::collections::HashSet;
|
|
use std::cmp::PartialEq;
|
|
use std::str::FromStr;
|
|
use std::sync::Arc;
|
|
use util::RotatingLogger;
|
|
use ethcore::miner::{Miner, ExternalMiner};
|
|
use ethcore::client::Client;
|
|
use ethcore::account_provider::AccountProvider;
|
|
use ethcore::snapshot::SnapshotService;
|
|
use ethsync::{ManageNetwork, SyncProvider};
|
|
use ethcore_rpc::{Extendable, NetworkSettings};
|
|
pub use ethcore_rpc::SignerService;
|
|
|
|
|
|
#[derive(Debug, PartialEq, Clone, Eq, Hash)]
|
|
pub enum Api {
|
|
Web3,
|
|
Net,
|
|
Eth,
|
|
PersonalSafe,
|
|
PersonalAccounts,
|
|
Signer,
|
|
Ethcore,
|
|
EthcoreSet,
|
|
Traces,
|
|
Rpc,
|
|
}
|
|
|
|
impl FromStr for Api {
|
|
type Err = String;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
use self::Api::*;
|
|
|
|
match s {
|
|
"web3" => Ok(Web3),
|
|
"net" => Ok(Net),
|
|
"eth" => Ok(Eth),
|
|
"personal" => Ok(PersonalAccounts),
|
|
"personal_safe" => Ok(PersonalSafe),
|
|
"signer" => Ok(Signer),
|
|
"ethcore" => Ok(Ethcore),
|
|
"ethcore_set" => Ok(EthcoreSet),
|
|
"traces" => Ok(Traces),
|
|
"rpc" => Ok(Rpc),
|
|
api => Err(format!("Unknown api: {}", api))
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub enum ApiSet {
|
|
SafeContext,
|
|
UnsafeContext,
|
|
IpcContext,
|
|
List(HashSet<Api>),
|
|
}
|
|
|
|
impl Default for ApiSet {
|
|
fn default() -> Self {
|
|
ApiSet::UnsafeContext
|
|
}
|
|
}
|
|
|
|
impl PartialEq for ApiSet {
|
|
fn eq(&self, other: &Self) -> bool {
|
|
self.list_apis() == other.list_apis()
|
|
}
|
|
}
|
|
|
|
impl FromStr for ApiSet {
|
|
type Err = String;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
s.split(',')
|
|
.map(Api::from_str)
|
|
.collect::<Result<_, _>>()
|
|
.map(ApiSet::List)
|
|
}
|
|
}
|
|
|
|
pub struct Dependencies {
|
|
pub signer_service: Arc<SignerService>,
|
|
pub client: Arc<Client>,
|
|
pub snapshot: Arc<SnapshotService>,
|
|
pub sync: Arc<SyncProvider>,
|
|
pub net: Arc<ManageNetwork>,
|
|
pub secret_store: Arc<AccountProvider>,
|
|
pub miner: Arc<Miner>,
|
|
pub external_miner: Arc<ExternalMiner>,
|
|
pub logger: Arc<RotatingLogger>,
|
|
pub settings: Arc<NetworkSettings>,
|
|
pub net_service: Arc<ManageNetwork>,
|
|
pub geth_compatibility: bool,
|
|
pub dapps_port: Option<u16>,
|
|
}
|
|
|
|
fn to_modules(apis: &[Api]) -> BTreeMap<String, String> {
|
|
let mut modules = BTreeMap::new();
|
|
for api in apis {
|
|
let (name, version) = match *api {
|
|
Api::Web3 => ("web3", "1.0"),
|
|
Api::Net => ("net", "1.0"),
|
|
Api::Eth => ("eth", "1.0"),
|
|
Api::PersonalSafe => ("personal_safe", "1.0"),
|
|
Api::PersonalAccounts => ("personal", "1.0"),
|
|
Api::Signer => ("signer", "1.0"),
|
|
Api::Ethcore => ("ethcore", "1.0"),
|
|
Api::EthcoreSet => ("ethcore_set", "1.0"),
|
|
Api::Traces => ("traces", "1.0"),
|
|
Api::Rpc => ("rpc", "1.0"),
|
|
};
|
|
modules.insert(name.into(), version.into());
|
|
}
|
|
modules
|
|
}
|
|
|
|
impl ApiSet {
|
|
pub fn list_apis(&self) -> HashSet<Api> {
|
|
match *self {
|
|
ApiSet::List(ref apis) => apis.clone(),
|
|
ApiSet::UnsafeContext => {
|
|
vec![Api::Web3, Api::Net, Api::Eth, Api::Ethcore, Api::Traces, Api::Rpc, Api::PersonalSafe]
|
|
.into_iter().collect()
|
|
},
|
|
ApiSet::IpcContext => {
|
|
vec![Api::Web3, Api::Net, Api::Eth, Api::Ethcore, Api::Traces, Api::Rpc, Api::PersonalAccounts, Api::PersonalSafe]
|
|
.into_iter().collect()
|
|
},
|
|
ApiSet::SafeContext => {
|
|
vec![Api::Web3, Api::Net, Api::Eth, Api::PersonalAccounts, Api::PersonalSafe, Api::Signer, Api::Ethcore, Api::EthcoreSet, Api::Traces, Api::Rpc]
|
|
.into_iter().collect()
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn setup_rpc<T: Extendable>(server: T, deps: Arc<Dependencies>, apis: ApiSet) -> T {
|
|
use ethcore_rpc::v1::*;
|
|
|
|
// it's turned into vector, cause ont of the cases requires &[]
|
|
let apis = apis.list_apis().into_iter().collect::<Vec<_>>();
|
|
for api in &apis {
|
|
match *api {
|
|
Api::Web3 => {
|
|
server.add_delegate(Web3Client::new().to_delegate());
|
|
},
|
|
Api::Net => {
|
|
server.add_delegate(NetClient::new(&deps.sync).to_delegate());
|
|
},
|
|
Api::Eth => {
|
|
let client = EthClient::new(
|
|
&deps.client,
|
|
&deps.snapshot,
|
|
&deps.sync,
|
|
&deps.secret_store,
|
|
&deps.miner,
|
|
&deps.external_miner,
|
|
EthClientOptions {
|
|
allow_pending_receipt_query: !deps.geth_compatibility,
|
|
send_block_number_in_get_work: !deps.geth_compatibility,
|
|
}
|
|
);
|
|
server.add_delegate(client.to_delegate());
|
|
|
|
let filter_client = EthFilterClient::new(&deps.client, &deps.miner);
|
|
server.add_delegate(filter_client.to_delegate());
|
|
|
|
if deps.signer_service.is_enabled() {
|
|
server.add_delegate(EthSigningQueueClient::new(&deps.signer_service, &deps.client, &deps.miner, &deps.secret_store).to_delegate());
|
|
} else {
|
|
server.add_delegate(EthSigningUnsafeClient::new(&deps.client, &deps.secret_store, &deps.miner).to_delegate());
|
|
}
|
|
},
|
|
Api::PersonalAccounts => {
|
|
let enable_permanent_unlock = deps.geth_compatibility || deps.miner.seals_internally;
|
|
server.add_delegate(PersonalAccountsClient::new(&deps.secret_store, &deps.client, &deps.miner, enable_permanent_unlock).to_delegate());
|
|
},
|
|
Api::PersonalSafe => {
|
|
server.add_delegate(PersonalClient::new(&deps.secret_store, &deps.client).to_delegate());
|
|
},
|
|
Api::Signer => {
|
|
server.add_delegate(SignerClient::new(&deps.secret_store, &deps.client, &deps.miner, &deps.signer_service).to_delegate());
|
|
},
|
|
Api::Ethcore => {
|
|
let signer = match deps.signer_service.is_enabled() {
|
|
true => Some(deps.signer_service.clone()),
|
|
false => None,
|
|
};
|
|
server.add_delegate(EthcoreClient::new(
|
|
&deps.client,
|
|
&deps.miner,
|
|
&deps.sync,
|
|
&deps.net_service,
|
|
deps.logger.clone(),
|
|
deps.settings.clone(),
|
|
signer,
|
|
deps.dapps_port,
|
|
).to_delegate())
|
|
},
|
|
Api::EthcoreSet => {
|
|
server.add_delegate(EthcoreSetClient::new(&deps.client, &deps.miner, &deps.net_service).to_delegate())
|
|
},
|
|
Api::Traces => {
|
|
server.add_delegate(TracesClient::new(&deps.client, &deps.miner).to_delegate())
|
|
},
|
|
Api::Rpc => {
|
|
let modules = to_modules(&apis);
|
|
server.add_delegate(RpcClient::new(modules).to_delegate());
|
|
}
|
|
}
|
|
}
|
|
server
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::{Api, ApiSet};
|
|
|
|
#[test]
|
|
fn test_api_parsing() {
|
|
assert_eq!(Api::Web3, "web3".parse().unwrap());
|
|
assert_eq!(Api::Net, "net".parse().unwrap());
|
|
assert_eq!(Api::Eth, "eth".parse().unwrap());
|
|
assert_eq!(Api::PersonalAccounts, "personal".parse().unwrap());
|
|
assert_eq!(Api::PersonalSafe, "personal_safe".parse().unwrap());
|
|
assert_eq!(Api::Signer, "signer".parse().unwrap());
|
|
assert_eq!(Api::Ethcore, "ethcore".parse().unwrap());
|
|
assert_eq!(Api::EthcoreSet, "ethcore_set".parse().unwrap());
|
|
assert_eq!(Api::Traces, "traces".parse().unwrap());
|
|
assert_eq!(Api::Rpc, "rpc".parse().unwrap());
|
|
assert!("rp".parse::<Api>().is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_api_set_default() {
|
|
assert_eq!(ApiSet::UnsafeContext, ApiSet::default());
|
|
}
|
|
|
|
#[test]
|
|
fn test_api_set_parsing() {
|
|
assert_eq!(ApiSet::List(vec![Api::Web3, Api::Eth].into_iter().collect()), "web3,eth".parse().unwrap());
|
|
}
|
|
|
|
#[test]
|
|
fn test_api_set_unsafe_context() {
|
|
let expected = vec![Api::Web3, Api::Net, Api::Eth, Api::Ethcore, Api::Traces, Api::Rpc, Api::PersonalSafe]
|
|
.into_iter().collect();
|
|
assert_eq!(ApiSet::UnsafeContext.list_apis(), expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_api_set_safe_context() {
|
|
let expected = vec![Api::Web3, Api::Net, Api::Eth, Api::PersonalAccounts, Api::PersonalSafe, Api::Signer, Api::Ethcore, Api::EthcoreSet, Api::Traces, Api::Rpc]
|
|
.into_iter().collect();
|
|
assert_eq!(ApiSet::SafeContext.list_apis(), expected);
|
|
}
|
|
}
|