-
Notifications
You must be signed in to change notification settings - Fork 0
/
PortForwarding.py
89 lines (71 loc) · 3.14 KB
/
PortForwarding.py
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/usr/bin/python
from pox.core import core
from pox.openflow import *
import string
import time
import threading
import pdb
from utils import *
from SimpleL2Learning import SimpleL2LearningSwitch
from pox.lib.packet.ethernet import ethernet
from pox.lib.packet.vlan import vlan
from pox.lib.packet.ipv4 import ipv4
from pox.lib.packet.arp import arp
from pox.lib.packet.tcp import tcp
log = core.getLogger() # Use central logging service
SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__))
FLOW_HARD_TIMEOUT = 30
FLOW_IDLE_TIMEOUT = 10
class PortForwardingSwitch(SimpleL2LearningSwitch):
def __init__(self, connection, config):
SimpleL2LearningSwitch.__init__(self, connection, False)
self._connection = connection;
self._serverip = config['server_ip']
self._origport = int(config['orig_port'])
self._forwport = int(config['forw_port'])
def _handle_PacketIn(self, event):
log.debug("Got a packet : " + str(event.parsed))
self.packet = event.parsed
self.event = event
self.macLearningHandle()
if packetIsTCP(self.packet, log) :
self._handle_PacketInTCP(event)
return
SimpleL2LearningSwitch._handle_PacketIn(self, event)
def _handle_PacketInTCP(self, event) :
inport = event.port
actions = []
out_port = self.get_out_port()
# XXX If packet is destined to serverip:original port
# make the appropriate rewrite
if packetDstIp(self.packet, self._serverip, log ):
if packetDstTCPPort(self.packet, self._origport, log) :
log.debug("Packet is TCP destined to %s:%d" % (self._serverip, self._origport))
newaction = createOFAction(of.OFPAT_SET_TP_DST, self._forwport, log)
actions.append(newaction)
# XXX If packet is sourced at serverip:forward port
# make the appropriate rewrite
if packetSrcIp(self.packet, self._serverip, log ):
if packetSrcTCPPort(self.packet, self._forwport, log) :
log.debug("Packet is TCP sourced at %s:%d" % (self._serverip, self._forwport))
newaction = createOFAction(of.OFPAT_SET_TP_SRC, self._origport, log)
actions.append(newaction)
# XXX Create the flow mod message in a variable
# called msg
newaction = createOFAction(of.OFPAT_OUTPUT, out_port, log)
actions.append(newaction)
match = getFullMatch(self.packet, inport)
msg = createFlowMod(match, actions, FLOW_HARD_TIMEOUT,
FLOW_IDLE_TIMEOUT, event.ofp.buffer_id)
event.connection.send(msg.pack())
class PortForwarding(object):
def __init__(self, config):
core.openflow.addListeners(self)
self._config=config
def _handle_ConnectionUp(self, event):
log.debug("Connection %s" % (event.connection,))
PortForwardingSwitch(event.connection, self._config)
def launch(config_file=os.path.join(SCRIPT_PATH, "port_forward.config")):
log.debug("PortForwarding " + config_file);
config = readConfigFile(config_file, log)
core.registerNew(PortForwarding, config["general"])