def test_endpoint_by_name():
s = SDNConnect()
endpoint = s.endpoint_by_name('foo')
assert endpoint == None
endpoint = Endpoint('foo')
endpoint.endpoint_data = {
'tenant': 'foo', 'mac': '00:00:00:00:00:00', 'segment': 'foo', 'port': '1'}
s.endpoints.append(endpoint)
endpoint2 = s.endpoint_by_name('foo')
assert endpoint == endpoint2
def test_endpoint_by_name():
controller = Config().get_config()
s = SDNConnect(controller)
endpoint = s.endpoint_by_name('foo')
assert endpoint == None
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
s.endpoints[endpoint.name] = endpoint
endpoint2 = s.endpoint_by_name('foo')
assert endpoint == endpoint2
class Commands:
def __init__(self):
self.states = [
'active', 'inactive', 'known', 'unknown', 'mirroring', 'abnormal',
'shutdown', 'reinvestigating', 'queued'
]
self.sdnc = SDNConnect()
self.sdnc.get_stored_endpoints()
def _get_endpoints(self, args, idx):
''' get endpoints that match '''
eps = []
device = args.rsplit(' ', 1)[idx]
name_endpoint = self.sdnc.endpoint_by_name(device)
if name_endpoint:
eps.append(name_endpoint)
return eps
hash_endpoint = self.sdnc.endpoint_by_hash(device)
if hash_endpoint:
eps.append(hash_endpoint)
return eps
ip_endpoints = self.sdnc.endpoints_by_ip(device)
if len(ip_endpoints) > 0:
return ip_endpoints
mac_endpoints = self.sdnc.endpoints_by_mac(device)
if len(mac_endpoints) > 0:
return mac_endpoints
return eps
def what_is(self, args):
''' what is a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def history_of(self, args):
''' history of a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def where_is(self, args):
''' where topologically is a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def remove_inactives(self, args):
''' remove all inactive devices '''
endpoints = []
endpoint_names = []
for endpoint in self.sdnc.endpoints:
if endpoint.state == 'inactive':
endpoints.append(endpoint)
endpoint_names.append(endpoint.name)
self.sdnc.publish_action('poseidon.action.remove.inactives',
json.dumps(endpoint_names))
return endpoints
def remove_ignored(self, args):
''' remove all ignored devices '''
endpoints = []
endpoint_names = []
for endpoint in self.sdnc.endpoints:
if endpoint.ignore == True:
endpoints.append(endpoint)
endpoint_names.append(endpoint.name)
self.sdnc.publish_action('poseidon.action.remove.ignored',
json.dumps(endpoint_names))
return endpoints
def ignore(self, args):
''' ignore a specific thing '''
eps = []
device = args.rsplit(' ', 1)[0]
if device == 'inactive':
for endpoint in self.sdnc.endpoints:
if endpoint.state == 'inactive':
eps.append(endpoint)
else:
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
endpoints = []
endpoint_names = []
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
endpoint_names.append(endpoint.name)
self.sdnc.publish_action('poseidon.action.ignore',
json.dumps(endpoint_names))
return endpoints
def clear_ignored(self, args):
''' stop ignoring a specific thing '''
eps = []
device = args.rsplit(' ', 1)[0]
if device == 'ignored':
for endpoint in self.sdnc.endpoints:
if endpoint.ignore:
eps.append(endpoint)
else:
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
endpoints = []
endpoint_names = []
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
endpoint_names.append(endpoint.name)
self.sdnc.publish_action('poseidon.action.clear.ignored',
json.dumps(endpoint_names))
return endpoints
def remove(self, args):
''' remove and forget about a specific thing until it's seen again '''
endpoints = []
endpoint_names = []
eps = self._get_endpoints(args, 0)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
endpoint_names.append(endpoint.name)
self.sdnc.publish_action('poseidon.action.remove',
json.dumps(endpoint_names))
return endpoints
def show_devices(self, arg):
'''
show all devices that are of a specific filter. i.e. windows,
developer workstation, abnormal, mirroring, etc.
'''
return self.sdnc.show_endpoints(arg)
def change_devices(self, args):
''' change state of a specific thing '''
eps = []
endpoints = []
endpoint_names = []
device = args.split(' ', 1)[0]
state = args.rsplit(' ', 1)[-1]
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
endpoint_names.append((endpoint.name, state))
self.sdnc.publish_action('poseidon.action.change',
json.dumps(endpoint_names))
return endpoints
class Commands:
def __init__(self):
self.states = [
'active', 'inactive', 'known', 'unknown', 'mirroring', 'abnormal',
'shutdown', 'reinvestigating', 'queued'
]
self.sdnc = SDNConnect()
def _get_endpoints(self, args, idx):
''' get endpoints that match '''
eps = []
device = args.rsplit(' ', 1)[idx]
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
return eps
def what_is(self, args):
''' what is a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def history_of(self, args):
''' history of a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def where_is(self, args):
''' where topologically is a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
endpoints.append(endpoint)
return endpoints
def collect_on(self, args):
''' collect on a specific thing '''
endpoints = []
eps = self._get_endpoints(args, -1)
for endpoint in eps:
if endpoint:
self.sdnc.collect_on(endpoint)
endpoints.append(endpoint)
return endpoints
def remove_inactives(self, args):
''' remove all inactive devices '''
return self.sdnc.remove_inactive_endpoints()
def remove_ignored(self, args):
''' remove all ignored devices '''
return self.sdnc.remove_ignored_endpoints()
def ignore(self, args):
''' ignore a specific thing '''
eps = []
device = args.rsplit(' ', 1)[0]
if device == 'inactive':
for endpoint in self.sdnc.endpoints:
if endpoint.state == 'inactive':
eps.append(endpoint)
else:
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
endpoints = []
for endpoint in eps:
if endpoint:
self.sdnc.ignore_endpoint(endpoint)
endpoints.append(endpoint)
return endpoints
def clear_ignored(self, args):
''' stop ignoring a specific thing '''
eps = []
device = args.rsplit(' ', 1)[0]
if device == 'ignored':
for endpoint in self.sdnc.endpoints:
if endpoint.ignore:
eps.append(endpoint)
else:
eps.append(self.sdnc.endpoint_by_name(device))
eps.append(self.sdnc.endpoint_by_hash(device))
eps += self.sdnc.endpoints_by_ip(device)
eps += self.sdnc.endpoints_by_mac(device)
endpoints = []
for endpoint in eps:
if endpoint:
self.sdnc.clear_ignored_endpoint(endpoint)
endpoints.append(endpoint)
return endpoints
def remove(self, args):
''' remove and forget about a specific thing until it's seen again '''
endpoints = []
eps = self._get_endpoints(args, 0)
for endpoint in eps:
if endpoint:
self.sdnc.remove_endpoint(endpoint)
endpoints.append(endpoint)
return endpoints
def show_devices(self, args):
'''
show all devices that are of a specific filter. i.e. windows,
developer workstation, abnormal, mirroring, etc.
'''
state = None
type_filter = None
all_devices = False
query = args.rsplit(' ', 1)[0]
if query in self.states:
state = query
elif query == 'all':
all_devices = True
else:
type_filter = query
return self.sdnc.show_endpoints(state, type_filter, all_devices)
def test_endpoint_by_name():
s = SDNConnect()
endpoint = s.endpoint_by_name('foo')
assert endpoint == None
