summaryrefslogtreecommitdiff
path: root/kpov_judge
diff options
context:
space:
mode:
authorgasperfele@fri1.uni-lj.si <gasperfele@fri1.uni-lj.si@5cf9fbd1-b2bc-434c-b4b7-e852f4f63414>2015-03-05 17:06:08 +0000
committergasperfele@fri1.uni-lj.si <gasperfele@fri1.uni-lj.si@5cf9fbd1-b2bc-434c-b4b7-e852f4f63414>2015-03-05 17:06:08 +0000
commit060f54de39a45e4efbc1f82d7aef20732b5b1fc7 (patch)
treeb0dc9170c2aab9b5d2028742016f6ec52222a80d /kpov_judge
parentfc8be19f7ba116d451b03b2ad8f881350e4d00d6 (diff)
Test for nat_port_forward finished.
git-svn-id: https://svn.lusy.fri.uni-lj.si/kpov-public-svn/kpov-public@284 5cf9fbd1-b2bc-434c-b4b7-e852f4f63414
Diffstat (limited to 'kpov_judge')
-rw-r--r--kpov_judge/tasks/nat_port_forward/task.py126
1 files changed, 112 insertions, 14 deletions
diff --git a/kpov_judge/tasks/nat_port_forward/task.py b/kpov_judge/tasks/nat_port_forward/task.py
index d7dde03..a3fc740 100644
--- a/kpov_judge/tasks/nat_port_forward/task.py
+++ b/kpov_judge/tasks/nat_port_forward/task.py
@@ -63,37 +63,135 @@ networks = { 'net1': {'public': False}, 'test-net': {'public': True} }
params_meta = {
'IP_TEST_CLIENT': {'descriptions': {'si': 'Naslov TestClient'}, 'w': True, 'public':True, 'type': 'IP', 'generated': False},
+ 'IP_NAT': {'descriptions': {'si': 'Naslov NATServer, dostopen s TestClient'}, 'w': True, 'public':True, 'type': 'IP', 'generated': False},
'PORT_OUTER': {'descriptions': {'si': 'Zunanja vrata'}, 'w': False, 'public':True, 'type': 'port', 'generated': True},
'PORT_INNER': {'descriptions': {'si': 'Notranja vrata'}, 'w': False, 'public': True, 'type': 'port', 'generated': True},
'NET': {'descriptions': {'si': u'Področje naslovov med SimpleArbiter in TestClient'}, 'w': False, 'public': True, 'type': 'NET', 'generated': True},
}
-def task(IP_TEST_CLIENT, PORT_OUTER, PORT_INNER, NET):
+def task(IP_TEST_CLIENT, IP_NAT, PORT_OUTER, PORT_INNER, NET):
import pexpect
-
+ import pxssh
+ import re
+ import random
+ import time
+ # return None
+ peer_user = 'student'
+ peer_password = 'vaje'
+ results = dict()
+ tcpdump = pexpect.spawn('sudo /usr/sbin/tcpdump src host {} and dst port {}'.format(IP_TEST_CLIENT, PORT_INNER))
+ sshconn = pxssh.pxssh()
+ sshconn.login(IP_TEST_CLIENT, peer_user, peer_password)
+ r = random.Random()
+ k = r.randint(10, 15)
+ results['pre_nc'] = str(k)
+ results['nc_pre'] = ""
+ for i in xrange(k):
+ sshconn.sendline("nc {} {}".format(IP_NAT, PORT_OUTER))
+ sshconn.sendline()
+ sshconn.sendintr()
+ sshconn.prompt()
+ results['nc_pre'] += sshconn.before
+ nc = pexpect.spawn('nc -l -p {}'.format(PORT_INNER))
+ sshconn.sendline()
+ sshconn.prompt()
+ sshconn.sendline("nc {} {}".format(IP_NAT, PORT_OUTER))
+ results['post_nc'] = "".join([r.choice("abcd\n") for i in xrange(100)])
+ sshconn.sendline(results['post_nc'])
+ time.sleep(1)
+ sshconn.sendintr()
+ nc.expect(pexpect.EOF)
+ results['nc_ret'] = nc.before
+ results['route'] = pexpect.run('/sbin/ip route list 0/0')
+ results['traceroute'] = pexpect.run('traceroute {}'.format(IP_TEST_CLIENT))
+ # wait for traceroute
+ time.sleep(10)
+ tcpdump.sendintr()
+ tcpdump.expect(pexpect.EOF)
+ results['tcpdump'] = tcpdump.before
+ sshconn.prompt()
+ results['nc_post'] = sshconn.before
+ sshconn.close()
+ # nc.expect(pexpect.EOF)
return results
def gen_params(user_id, params_meta):
params = dict()
r = random.Random(user_id)
# IP_NM, DNS_NM, IP_static, DNS_static)
- params['PORT_INNER'] = r.randint(6000, 10000)
- params['PORT_OUTER'] = r.randint(10001, 15000)
- params['NET'] = kpov_random_helpers.IPv4_subnet_gen("10.36.0.0/14", 24)
+ params['PORT_INNER'] = str(r.randint(6000, 10000))
+ params['PORT_OUTER'] = str(r.randint(10001, 15000))
+ params['NET'] = kpov_random_helpers.IPv4_subnet_gen(r, "10.36.0.0/14", 24)
return params
def task_check(results, params):
import re
- score = -9
- if results['NM_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_NM'])) > -1:
- score += 3
- if results['static_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_static'])) > -1:
- score += 3
- if re.search(r'eth0 +802-.*connected', results['NM_nmcli']):
- score += 2
- if not re.search(r'eth0 +802-.*connected', results['static_nmcli']):
- score += 2
+ import cPickle
+ if results is None:
+ with open('bla.pickle') as f:
+ results = cPickle.load(f)
+ else:
+ with open('bla.pickle', 'w') as f:
+ cPickle.dump(results, f)
score = 0
+ local_net = params['NET'][:params['NET'].rfind('.')]
+ s = "default via ({}\\.[0-9]+)".format(re.escape(local_net))
+ res = re.search(s, results['route'])
+ if res:
+ ip_nat_internal = res.groups(1)[0]
+ score += 1
+ else:
+ ip_nat_internal = 'abrakadabra'
+ # print (s, results['route'],)
+ s = "traceroute to {ip_test} \\({ip_test}\\), 30 hops max, 60 byte packets\
+\r\n 1 {ip_nat} \\({ip_nat}\\) [0-9.]+ ms [0-9.]+ ms [0-9.]+ ms\
+.*{ip_test} \\({ip_test}\\)".format(
+ ip_nat = re.escape(ip_nat_internal),
+ ip_test = re.escape(params['IP_TEST_CLIENT'])
+ )
+ if re.search(s, results['traceroute'], re.DOTALL):
+ score += 1
+ else:
+ print (s, results['traceroute'],)
+ if len(results['post_nc']) == 100:
+ score += 1
+ else:
+ print (results['post_nc'], len(results['post_nc']))
+ if results['nc_ret'] == (results['post_nc'] + '\n').replace('\n', '\r\n'):
+ score += 1
+ else:
+ print (results['post_nc'], results['nc_ret'],)
+ s = "Connection refused"
+ res = re.findall(s, results['nc_pre'])
+ if len(res) >= 2:
+ score += 3
+ else:
+ print (s, results['nc_pre'],)
+ s = "\r\n"
+ if re.search(s, results['nc_post']):
+ score += 1
+ else:
+ print (s, results['nc_post'],)
+ rejected_count = int(results['pre_nc'])
+ accepted_count = results['nc_ret'].count('\r\n')
+ s = ".*verbose output suppressed.*listening on.*dropped by kernel.*"
+ if re.match(s, results['tcpdump'], re.DOTALL):
+ score += 1
+ else:
+ print (s, results['tcpdump'],)
+ res = re.findall("length .*\r\n", results['tcpdump'])
+ total_len = 0
+ n_empty = 0
+ for i in res:
+ k = int(i[len("length "):].strip())
+ total_len += k
+ if k == 0:
+ n_empty += 1
+ # print total_len, rejected_count, n_empty
+ if total_len == 101 and rejected_count <= n_empty:
+ score += 1
+ else:
+ print (s, results['tcpdump'],)
return score
def prepare_disks(templates, params):