def main():
# Default subnet is required for the host tracker to work.
subnet_control = SubnetControl()
subnet_control.add_subnet("defaultSubnet", "10.0.0.254/8")
# Set up an affinity link
affinity_control = AffinityControl()
affinity_control.add_affinity_group("testAG1", ips=["10.0.0.1", "10.0.0.2"])
affinity_control.add_affinity_group("testAG2", ips=["10.0.0.3", "10.0.0.4"])
affinity_control.add_affinity_link("testAL", "testAG1", "testAG2")
raw_input("press enter ")
add_protocol_flows()
run_interactive_mode()
def run(self):
global sigint
high_rate_set = False
while not sigint:
is_high, is_low = self.stat.refresh()
if is_high and not high_rate_set:
ac = AffinityControl()
high_rate_set = True
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
ac.enable_affinity()
print "Adding isolate affinity to link: %s" % (link_name)
# time.sleep(3)
elif is_low and high_rate_set:
high_rate_set = False
ac.remove_isolate(link_name)
ac.enable_affinity()
print "Adding isolate affinity to link: %s" % (link_name)
time.sleep(10)
def run(self):
global sigint
high_rate_set = False
while not sigint:
is_high, is_low = self.stat.refresh()
if is_high and not high_rate_set:
ac = AffinityControl()
high_rate_set = True
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
ac.enable_affinity()
print "Adding isolate affinity to link: %s" % (link_name)
# time.sleep(3)
elif is_low and high_rate_set:
high_rate_set = False
ac.remove_isolate(link_name)
ac.enable_affinity()
print "Adding isolate affinity to link: %s" % (link_name)
time.sleep(10)
def main():
# Default subnet is required for the host tracker to work.
subnet_control = SubnetControl()
subnet_control.add_subnet("defaultSubnet", "10.0.0.254/8")
# Set up an affinity link
affinity_control = AffinityControl()
affinity_control.add_affinity_group("testAG1",
ips=["10.0.0.1", "10.0.0.2"])
affinity_control.add_affinity_group("testAG2",
ips=["10.0.0.3", "10.0.0.4"])
affinity_control.add_affinity_link("testAL", "testAG1", "testAG2")
raw_input("press enter ")
add_protocol_flows()
run_interactive_mode()
def run2(self):
global sigint
did_waypoint = False
while not sigint:
is_fast, is_big = self.stat.refresh()
if is_big and not did_waypoint:
print "Large flow detected (%d bytes, %d packets, %3.3f bit/s)" % (self.stat.get_bytes(), self.stat.get_packets(), self.stat.get_bit_rate())
print " ICMP: %d bytes, %d packets" % (self.stat.get_bytes(1), self.stat.get_packets(1))
print " UDP: %d bytes, %d packets" % (self.stat.get_bytes(17), self.stat.get_packets(17))
print " TCP: %d bytes, %d packets" % (self.stat.get_bytes(6), self.stat.get_packets(6))
print " other: %d bytes, %d packets" % (self.stat.get_bytes(-1), self.stat.get_packets(-1))
print("-------------------------")
ac = AffinityControl()
# First AG: Sources sending data into this subnet
src_ag_name = "sources"
src_ips = self.stat.get_large_incoming_hosts()
if (self.waypoint_address in src_ips):
src_ips.remove(self.waypoint_address)
ac.add_affinity_group(src_ag_name, ips=src_ips)
# Second AG: This entity
dst_ag_name = "client"
if (self.stat_type == Stats.TYPE_SUBNET):
ac.add_affinity_group(dst_ag_name, subnet=self.stat.subnet)
elif (self.stat_type == Stats.TYPE_HOST):
pass
else:
print "type", self.stat_type, "not supported for redirection"
# AL: Between them
link_name = "inflows"
ac.add_affinity_link(link_name, src_ag_name, dst_ag_name)
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
# ac.enable_waypoint(link_name)
ac.enable_affinity()
did_waypoint = True
raw_input("[Press Enter to disable affinity rules] ")
ac.disable_affinity()
# ac.disable_waypoint(link_name)
time.sleep(1)
def run(self):
global sigint
did_waypoint = False
while not sigint:
is_fast, is_big = self.stat.refresh()
if is_fast and not did_waypoint:
print "Fast flow detected (%3.3f bit/s)" % (self.stat.get_ewma_rate())
ac = AffinityControl()
# First AG: Sources sending data into this subnet
src_ag_name = "sources"
# Hosts that transmit > 10% of link capacity
src_ips = self.stat.get_large_incoming_hosts()
print "LFD src list = ", src_ips
print "Discard waypoint address"
if (self.waypoint_address in src_ips):
src_ips.remove(self.waypoint_address)
ac.add_affinity_group(src_ag_name, ips=src_ips)
# Second AG: This entity
dst_ag_name = "client"
if (self.stat_type == Stats.TYPE_SUBNET):
ac.add_affinity_group(dst_ag_name, subnet=self.stat.subnet)
elif (self.stat_type == Stats.TYPE_HOST):
ac.add_affinity_group(dst_ag_name, ips=self.dst)
else:
print "type", self.stat_type, "not supported for redirection"
# AL: Between them
link_name = "inflows"
ac.add_affinity_link(link_name, src_ag_name, dst_ag_name)
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
ac.enable_affinity()
did_waypoint = True
time.sleep(10)
# Flow now gone, disable affinity.
elif did_waypoint and not is_fast:
ac.disable_affinity()
did_waypoint = False
time.sleep(1)
def run2(self):
global sigint
did_waypoint = False
while not sigint:
is_fast, is_big = self.stat.refresh()
if is_big and not did_waypoint:
print "Large flow detected (%d bytes, %d packets, %3.3f bit/s)" % (
self.stat.get_bytes(), self.stat.get_packets(),
self.stat.get_bit_rate())
print " ICMP: %d bytes, %d packets" % (
self.stat.get_bytes(1), self.stat.get_packets(1))
print " UDP: %d bytes, %d packets" % (
self.stat.get_bytes(17), self.stat.get_packets(17))
print " TCP: %d bytes, %d packets" % (
self.stat.get_bytes(6), self.stat.get_packets(6))
print " other: %d bytes, %d packets" % (
self.stat.get_bytes(-1), self.stat.get_packets(-1))
print("-------------------------")
ac = AffinityControl()
# First AG: Sources sending data into this subnet
src_ag_name = "sources"
src_ips = self.stat.get_large_incoming_hosts()
if (self.waypoint_address in src_ips):
src_ips.remove(self.waypoint_address)
ac.add_affinity_group(src_ag_name, ips=src_ips)
# Second AG: This entity
dst_ag_name = "client"
if (self.stat_type == Stats.TYPE_SUBNET):
ac.add_affinity_group(dst_ag_name, subnet=self.stat.subnet)
elif (self.stat_type == Stats.TYPE_HOST):
pass
else:
print "type", self.stat_type, "not supported for redirection"
# AL: Between them
link_name = "inflows"
ac.add_affinity_link(link_name, src_ag_name, dst_ag_name)
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
# ac.enable_waypoint(link_name)
ac.enable_affinity()
did_waypoint = True
raw_input("[Press Enter to disable affinity rules] ")
ac.disable_affinity()
# ac.disable_waypoint(link_name)
time.sleep(1)
def run(self):
global sigint
did_waypoint = False
while not sigint:
is_fast, is_big = self.stat.refresh()
if is_fast and not did_waypoint:
print "Fast flow detected (%3.3f bit/s)" % (
self.stat.get_ewma_rate())
ac = AffinityControl()
# First AG: Sources sending data into this subnet
src_ag_name = "sources"
# Hosts that transmit > 10% of link capacity
src_ips = self.stat.get_large_incoming_hosts()
print "LFD src list = ", src_ips
print "Discard waypoint address"
if (self.waypoint_address in src_ips):
src_ips.remove(self.waypoint_address)
ac.add_affinity_group(src_ag_name, ips=src_ips)
# Second AG: This entity
dst_ag_name = "client"
if (self.stat_type == Stats.TYPE_SUBNET):
ac.add_affinity_group(dst_ag_name, subnet=self.stat.subnet)
elif (self.stat_type == Stats.TYPE_HOST):
ac.add_affinity_group(dst_ag_name, ips=self.dst)
else:
print "type", self.stat_type, "not supported for redirection"
# AL: Between them
link_name = "inflows"
ac.add_affinity_link(link_name, src_ag_name, dst_ag_name)
# ac.add_waypoint(link_name, self.waypoint_address)
ac.add_isolate(link_name)
ac.enable_affinity()
did_waypoint = True
time.sleep(10)
# Flow now gone, disable affinity.
elif did_waypoint and not is_fast:
ac.disable_affinity()
did_waypoint = False
time.sleep(1)
