aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/salticidae/network.h3
-rw-r--r--test/test_p2p_stress.cpp10
2 files changed, 11 insertions, 2 deletions
diff --git a/include/salticidae/network.h b/include/salticidae/network.h
index 1f21c92..d83e7f1 100644
--- a/include/salticidae/network.h
+++ b/include/salticidae/network.h
@@ -284,7 +284,7 @@ class PeerNetwork: public MsgNetwork<OpcodeType> {
NetAddr get_peer_addr() {
auto ret = *(static_cast<NetAddr *>(
get_net()->disp_tcall->call([this](ThreadCall::Handle &h) {
- h.set_result(peer ? peer->peer_addr : NetAddr());
+ h.set_result(peer ? NetAddr(peer->peer_addr) : NetAddr());
}).get()));
return ret;
}
@@ -650,6 +650,7 @@ void PeerNetwork<O, _, __>::on_teardown(const ConnPool::conn_t &_conn) {
known_peers[p->peer_addr] = std::make_pair(uint256_t(), TimerEvent());
{
pinfo_ulock_t __g(pid2peer_lock);
+ p->conn->peer = nullptr;
pid2peer.erase(p->peer_id);
}
this->user_tcall->async_call([this, conn](ThreadCall::Handle &) {
diff --git a/test/test_p2p_stress.cpp b/test/test_p2p_stress.cpp
index b8ef017..cb6595c 100644
--- a/test/test_p2p_stress.cpp
+++ b/test/test_p2p_stress.cpp
@@ -118,6 +118,9 @@ void install_proto(AppContext &app, const size_t &seg_buff_size) {
tc.hash = msg.hash;
net.send_msg(std::move(msg), conn);
};
+ net.reg_conn_handler([](const ConnPool::conn_t &conn, bool connected) {
+ return true;
+ });
net.reg_peer_handler([&, send_rand](const MyNet::conn_t &conn, bool connected) {
if (connected)
{
@@ -143,7 +146,12 @@ void install_proto(AppContext &app, const size_t &seg_buff_size) {
});
net.reg_handler([&, send_rand](MsgAck &&msg, const MyNet::conn_t &conn) {
auto addr = conn->get_peer_addr();
- assert(!addr.is_null());
+ if (addr.is_null()) return;
+ if (app.tc.find(addr) == app.tc.end())
+ {
+ SALTICIDAE_LOG_WARN("%s\n", std::string(addr).c_str());
+ throw std::runtime_error("violation");
+ }
auto &tc = app.tc[addr];
if (msg.view != tc.view)
{