mirror of
https://github.com/savoirfairelinux/opendht.git
synced 2026-11-08 12:28:42 +00:00
Adrien Béraud
5166b1add9
The network monitor allows to regularly check the status of an OpenDHT network, by performing listen and put operations at random places on the DHT. This first version exit with status code 1 if the test fails.
84 lines
2.8 KiB
Python
Executable File
84 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2015-2016 Savoir-faire Linux Inc.
|
|
# Author: Adrien Béraud <adrien.beraud@savoirfairelinux.com>
|
|
#
|
|
# This program 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.
|
|
#
|
|
# This program 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 this program; If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import time
|
|
import argparse
|
|
import time
|
|
import asyncio
|
|
from datetime import datetime
|
|
|
|
import opendht as dht
|
|
|
|
parser = argparse.ArgumentParser(description='Create a dht network of -n nodes')
|
|
parser.add_argument('-b', '--bootstrap', help='bootstrap address', default='bootstrap.ring.cx')
|
|
parser.add_argument('-n', '--num-ops', help='number of concurrent operations on the DHT', type=int, default=8)
|
|
parser.add_argument('-p', '--period', help='duration between each test (seconds)', type=int, default=60)
|
|
parser.add_argument('-t', '--timeout', help='timeout for a test to complete (seconds)', type=float, default=15)
|
|
args = parser.parse_args()
|
|
|
|
node1 = dht.DhtRunner()
|
|
node1.run()
|
|
|
|
node2 = dht.DhtRunner()
|
|
node2.run()
|
|
|
|
node1.bootstrap(args.bootstrap)
|
|
node2.bootstrap(args.bootstrap)
|
|
loop = asyncio.get_event_loop()
|
|
|
|
pending_tests = {}
|
|
keys = [dht.InfoHash.getRandom() for _ in range(args.num_ops)]
|
|
|
|
def listen_cb(key, val):
|
|
global pending_tests
|
|
kstr = str(key)
|
|
if kstr in pending_tests:
|
|
if pending_tests[kstr].id == val.id:
|
|
pending_tests.pop(kstr, None)
|
|
else:
|
|
print("Expected vid", val.id, "got", pending_tests[kstr].id)
|
|
return True
|
|
|
|
def listen(key):
|
|
node1.listen(key, lambda v: loop.call_soon_threadsafe(listen_cb, key, v))
|
|
|
|
for key in keys:
|
|
listen(key)
|
|
|
|
next_test = time.time()
|
|
while True:
|
|
start = time.time()
|
|
#print(datetime.fromtimestamp(start).strftime('%Y-%m-%d %H:%M:%S'), 'Test started')
|
|
for key in keys:
|
|
val = dht.Value(str(dht.InfoHash.getRandom()).encode())
|
|
pending_tests[str(key)] = {'v':val, 'c':0}
|
|
node2.put(key, val, lambda ok, nodes: ok)
|
|
while len(pending_tests):
|
|
loop.stop()
|
|
loop.run_forever()
|
|
time.sleep(1)
|
|
if time.time()-start > args.timeout:
|
|
print('Test timeout !')
|
|
exit(1)
|
|
|
|
end = time.time()
|
|
print(datetime.fromtimestamp(end).strftime('%Y-%m-%d %H:%M:%S'),
|
|
'Test completed successfully in', end-start)
|
|
next_test += args.period
|
|
if next_test > end:
|
|
time.sleep(next_test-end)
|