1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from Zorp.Core import *
from Zorp.Plug import *
from Zorp.Session import MasterSession
from Zorp.NAT import getNATPolicy, NAT_SNAT, NAT_DNAT
from Zorp.Zorp import quit
from traceback import *
config.options.kzorp_enabled = FALSE
def testcase(nat, session, addrs, type, expected_result):
res = nat.performTranslation(session, addrs, type)
if not res.equal(expected_result):
print 'invalid result res=%s, expected_result=%s' % (res.format(), expected_result.format())
raise 'test error'
def zorp():
try:
s = MasterSession()
s.setService(Service("s1", None))
NATPolicy('test', GeneralNAT(
[(InetDomain('10.0.0.0/8'), InetDomain('20.0.0.0/8')),
(InetDomain('11.0.0.0/8'), InetDomain('192.168.0.0/24')),
]))
nat = getNATPolicy('test')
testcase(nat, s, (None, SockAddrInet('10.0.0.1', 8888)), NAT_DNAT, SockAddrInet('20.0.0.1', 8888))
testcase(nat, s, (None, SockAddrInet('11.0.0.0', 8888)), NAT_DNAT, SockAddrInet('192.168.0.0', 8888))
testcase(nat, s, (None, SockAddrInet('11.0.1.1', 8888)), NAT_DNAT, SockAddrInet('192.168.0.1', 8888))
testcase(nat, s, (None, SockAddrInet('11.255.255.255', 8888)), NAT_DNAT, SockAddrInet('192.168.0.255', 8888))
except Exception, e:
print_exc()
quit(1)
return 1
quit(0)
return 1
|
