mirror of
https://github.com/mimblewimble/grin.git
synced 2025-01-21 11:31:08 +03:00
9cc7aed713
* Move Proof and Difficulty types to pow module. Difficulty scaling calculation.
* Diffculty scaling and size shift for 2nd PoW
* Backport e7eb26ee
* Test compilation fixes
* Scaling only need to go one way, as @tromp pointed out
96 lines
2.9 KiB
Rust
96 lines
2.9 KiB
Rust
// Copyright 2018 The Grin Developers
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
extern crate grin_core as core;
|
|
extern crate grin_p2p as p2p;
|
|
extern crate grin_pool as pool;
|
|
extern crate grin_store as store;
|
|
extern crate grin_util as util;
|
|
|
|
use std::net::{SocketAddr, TcpListener, TcpStream};
|
|
use std::sync::atomic::AtomicBool;
|
|
use std::sync::Arc;
|
|
use std::{thread, time};
|
|
|
|
use core::core::hash::Hash;
|
|
use core::pow::Difficulty;
|
|
use p2p::Peer;
|
|
|
|
fn open_port() -> u16 {
|
|
// use port 0 to allow the OS to assign an open port
|
|
// TcpListener's Drop impl will unbind the port as soon as
|
|
// listener goes out of scope
|
|
let listener = TcpListener::bind("127.0.0.1:0").unwrap();
|
|
listener.local_addr().unwrap().port()
|
|
}
|
|
|
|
// Starts a server and connects a client peer to it to check handshake,
|
|
// followed by a ping/pong exchange to make sure the connection is live.
|
|
#[test]
|
|
fn peer_handshake() {
|
|
util::init_test_logger();
|
|
|
|
let p2p_config = p2p::P2PConfig {
|
|
host: "0.0.0.0".parse().unwrap(),
|
|
port: open_port(),
|
|
peers_allow: None,
|
|
peers_deny: None,
|
|
..p2p::P2PConfig::default()
|
|
};
|
|
let net_adapter = Arc::new(p2p::DummyAdapter {});
|
|
let db_env = Arc::new(store::new_env(".grin".to_string()));
|
|
let server = Arc::new(
|
|
p2p::Server::new(
|
|
db_env,
|
|
p2p::Capabilities::UNKNOWN,
|
|
p2p_config.clone(),
|
|
net_adapter.clone(),
|
|
Hash::from_vec(&vec![]),
|
|
Arc::new(AtomicBool::new(false)),
|
|
false,
|
|
None,
|
|
).unwrap(),
|
|
);
|
|
|
|
let p2p_inner = server.clone();
|
|
let _ = thread::spawn(move || p2p_inner.listen());
|
|
|
|
thread::sleep(time::Duration::from_secs(1));
|
|
|
|
let addr = SocketAddr::new(p2p_config.host, p2p_config.port);
|
|
let mut socket = TcpStream::connect_timeout(&addr, time::Duration::from_secs(10)).unwrap();
|
|
|
|
let my_addr = "127.0.0.1:5000".parse().unwrap();
|
|
let mut peer = Peer::connect(
|
|
&mut socket,
|
|
p2p::Capabilities::UNKNOWN,
|
|
Difficulty::one(),
|
|
my_addr,
|
|
&p2p::handshake::Handshake::new(Hash::from_vec(&vec![]), p2p_config.clone()),
|
|
net_adapter,
|
|
).unwrap();
|
|
|
|
assert!(peer.info.user_agent.ends_with(env!("CARGO_PKG_VERSION")));
|
|
|
|
peer.start(socket);
|
|
thread::sleep(time::Duration::from_secs(1));
|
|
|
|
peer.send_ping(Difficulty::one(), 0).unwrap();
|
|
thread::sleep(time::Duration::from_secs(1));
|
|
|
|
let server_peer = server.peers.get_connected_peer(&my_addr).unwrap();
|
|
let server_peer = server_peer.read().unwrap();
|
|
assert_eq!(server_peer.info.total_difficulty, Difficulty::one());
|
|
assert!(server.peers.peer_count() > 0);
|
|
}
|