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.
39 lines
882 B
39 lines
882 B
6 years ago
|
#!/usr/bin/env python3
|
||
|
|
"""Simple plugin to test the connected_hook.
|
||
|
|
|
||
|
|
It can mark some node_ids as rejects and it'll check for each
|
||
|
|
connection if it should be disconnected immediately or if it can
|
||
|
|
continue.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from lightning import Plugin
|
||
|
|
|
||
|
|
plugin = Plugin()
|
||
|
|
|
||
|
|
|
||
|
|
@plugin.hook('peer_connected')
|
||
|
|
def on_connected(peer, plugin):
|
||
|
|
if peer['id'] in plugin.reject_ids:
|
||
|
|
print("{} is in reject list, disconnecting".format(peer['id']))
|
||
|
|
return {'result': 'disconnect'}
|
||
|
|
|
||
|
|
print("{} is allowed".format(peer['id']))
|
||
|
|
return {'result': 'continue'}
|
||
|
|
|
||
|
|
|
||
|
|
@plugin.init()
|
||
|
|
def init(configuration, options, plugin):
|
||
|
|
plugin.reject_ids = []
|
||
|
|
|
||
|
|
|
||
|
|
@plugin.method('reject')
|
||
|
|
def reject(node_id, plugin):
|
||
|
|
"""Mark a given node_id as reject for future connections.
|
||
|
|
"""
|
||
|
|
print("Rejecting connections from {}".format(node_id))
|
||
|
|
plugin.reject_ids.append(node_id)
|
||
|
|
|
||
|
|
|
||
|
|
plugin.run()
|