class BasicSetup(object):
def setup(self):
# run server / client
self.env = {}
self.ct = Conntrack()
self.nfct = NFCTSocket()
if self.ct.count() == 0:
self.ct.close()
self.nfct.close()
raise SkipTest('conntrack modules are not supported')
self.server = threading.Thread(target=server,
args=('127.0.0.1', 5591, self.env))
self.server.start()
e = None
for x in range(5):
try:
self.client = Client('127.0.0.1', 5591)
time.sleep(1)
break
except Exception as exc:
e = exc
else:
raise e
def teardown(self):
self.nfct.close()
self.ct.close()
self.client.stop()
self.server.join()
self.env = {}
class BasicSetup(object):
def setup(self):
# run server / client
self.env = {}
self.ct = Conntrack()
self.nfct = NFCTSocket()
if self.ct.count() == 0:
self.ct.close()
self.nfct.close()
raise SkipTest('conntrack modules are not supported')
self.server = threading.Thread(target=server,
args=('127.0.0.1', 5591, self.env))
self.server.start()
e = None
for x in range(5):
try:
self.client = Client('127.0.0.1', 5591)
time.sleep(1)
break
except Exception as exc:
e = exc
else:
raise e
def teardown(self):
self.nfct.close()
self.ct.close()
self.client.stop()
self.server.join()
self.env = {}
def setup(self):
require_user('root')
self.tuples = []
self.COUNT_CT = 20
self.ct = Conntrack()
for sport in range(20000, 20000 + self.COUNT_CT):
tuple_orig = self.add_tuple(
saddr='192.168.122.1', daddr='192.168.122.67',
proto=socket.IPPROTO_TCP, sport=sport, dport=5599)
self.ct.entry('add', timeout=60,
tuple_orig=tuple_orig,
tuple_reply=tuple_orig.reverse())
def setup(self):
require_user('root')
# run server / client
self.env = {}
self.ct = Conntrack()
self.nfct = NFCTSocket()
if self.ct.count() == 0:
self.ct.close()
self.nfct.close()
raise SkipTest('conntrack modules are not supported')
self.server = threading.Thread(target=server,
args=('127.0.0.1', 5591, self.env))
self.server.start()
e = None
for x in range(5):
try:
self.client = Client('127.0.0.1', 5591)
time.sleep(1)
break
except Exception as exc:
e = exc
else:
raise e
class TestDumpEntries(object):
def add_tuple(self, saddr, daddr, proto, sport, dport):
tuple_orig = NFCTAttrTuple(saddr=saddr,
daddr=daddr,
proto=proto,
sport=sport,
dport=dport)
self.tuples.append(tuple_orig)
return tuple_orig
def setup(self):
require_user('root')
self.tuples = []
self.COUNT_CT = 20
self.ct = Conntrack()
for sport in range(20000, 20000 + self.COUNT_CT):
tuple_orig = self.add_tuple(saddr='192.168.122.1',
daddr='192.168.122.67',
proto=socket.IPPROTO_TCP,
sport=sport,
dport=5599)
self.ct.entry('add',
timeout=60,
tuple_orig=tuple_orig,
tuple_reply=tuple_orig.reverse())
def test_dump(self):
tuple_match = NFCTAttrTuple(saddr='192.168.122.1',
daddr='192.168.122.67')
count_found = 0
tuple_filter = tuple_match
for entry in self.ct.dump_entries(tuple_orig=tuple_match):
count_found += 1
assert count_found == self.COUNT_CT
count_found = 0
tuple_filter = NFCTAttrTuple(proto=socket.IPPROTO_TCP)
for entry in self.ct.dump_entries(tuple_orig=tuple_filter):
if tuple_match == entry.tuple_orig:
count_found += 1
assert count_found == self.COUNT_CT
def teardown(self):
for tuple_orig in self.tuples:
self.ct.entry('del', tuple_orig=tuple_orig)