blob: 8100f224b4694a4f83d55973eae95db821908c37 [file] [log] [blame]
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -08001#! /usr/bin/env python
2# -*- Mode: python; py-indent-offset: 4; tab-width: 8; indent-tabs-mode: t; -*-
3
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -07004import copy
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -08005import pprint
6import os
7import sys
8import subprocess
9import json
10import argparse
11import io
12import time
13
14from flask import Flask, json, Response, render_template, make_response, request
15
16#
17# curl http://127.0.0.1:8080/wm/topology/route/00:00:00:00:00:00:0a:01/1/00:00:00:00:00:00:0a:04/1/json
18#
19
20## Global Var ##
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -070021ControllerIP = "127.0.0.1"
22ControllerPort = 8080
23MonitoringEnabled = False
Pavlin Radoslavovb549f082013-03-29 04:28:27 -070024MonitoringByOnos = False
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -070025ReadFromFile = ""
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080026
27DEBUG=0
28pp = pprint.PrettyPrinter(indent=4)
29
30app = Flask(__name__)
31
32## Worker Functions ##
33def log_error(txt):
34 print '%s' % (txt)
35
36def debug(txt):
37 if DEBUG:
38 print '%s' % (txt)
39
40# @app.route("/wm/topology/route/<srcdpid>/<srcport>/<destdpid>/<destport>/json")
41#
42# Sample output:
43# {'dstPort': {'port': {'value': 0}, 'dpid': {'value': '00:00:00:00:00:00:00:02'}}, 'srcPort': {'port': {'value': 0}, 'dpid': {'value': '00:00:00:00:00:00:00:01'}}, 'flowEntries': [{'outPort': {'value': 1}, 'flowEntryErrorState': None, 'flowEntryMatch': None, 'flowEntryActions': None, 'inPort': {'value': 0}, 'flowEntryId': None, 'flowEntryUserState': 'FE_USER_UNKNOWN', 'dpid': {'value': '00:00:00:00:00:00:00:01'}, 'flowEntrySwitchState': 'FE_SWITCH_UNKNOWN'}, {'outPort': {'value': 0}, 'flowEntryErrorState': None, 'flowEntryMatch': None, 'flowEntryActions': None, 'inPort': {'value': 9}, 'flowEntryId': None, 'flowEntryUserState': 'FE_USER_UNKNOWN', 'dpid': {'value': '00:00:00:00:00:00:00:02'}, 'flowEntrySwitchState': 'FE_SWITCH_UNKNOWN'}]}
44#
45def shortest_path(v1, p1, v2, p2):
46 try:
47 command = "curl -s http://%s:%s/wm/topology/route/%s/%s/%s/%s/json" % (ControllerIP, ControllerPort, v1, p1, v2, p2)
48 debug("shortest_path %s" % command)
Pavlin Radoslavoveb2d5a22013-03-14 19:58:14 -070049 parsedResult = []
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080050
51 result = os.popen(command).read()
52 debug("result %s" % result)
53 if len(result) == 0:
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -070054 log_error("No Path found from %s/%s to %s/%s" % (v1, p1, v2, p2))
55 else:
56 parsedResult = json.loads(result)
57 debug("parsed %s" % parsedResult)
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080058
59 except:
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -070060 log_error("Controller IF has issue: No Path found from %s/%s to %s/%s" % (v1, p1, v2, p2))
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080061
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -070062 return parsedResult
63
64def print_data_path(data_path):
65 if len(data_path) == 0:
66 return
67
68 srcSwitch = data_path['srcPort']['dpid']['value'];
69 srcPort = data_path['srcPort']['port']['value'];
70 dstSwitch = data_path['dstPort']['dpid']['value'];
71 dstPort = data_path['dstPort']['port']['value'];
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080072
73 print "DataPath: (src = %s/%s dst = %s/%s)" % (srcSwitch, srcPort, dstSwitch, dstPort);
74
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -070075 for f in data_path['flowEntries']:
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080076 inPort = f['inPort']['value'];
77 outPort = f['outPort']['value'];
78 dpid = f['dpid']['value']
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -070079 print " FlowEntry: (%s, %s, %s)" % (inPort, dpid, outPort)
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080080
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080081def add_flow_path(flow_path):
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -070082 flow_path_json = json.dumps(flow_path)
83
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080084 try:
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -070085 command = "curl -s -H 'Content-Type: application/json' -d '%s' http://%s:%s/wm/flow/add/json" % (flow_path_json, ControllerIP, ControllerPort)
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -080086 debug("add_flow_path %s" % command)
87 result = os.popen(command).read()
88 debug("result %s" % result)
89 # parsedResult = json.loads(result)
90 # debug("parsed %s" % parsedResult)
91 except:
92 log_error("Controller IF has issue")
93 exit(1)
94
Pavlin Radoslavovb549f082013-03-29 04:28:27 -070095def add_shortest_path_flow(flow_path):
96 flow_path_json = json.dumps(flow_path)
97
98 try:
99 command = "curl -s -H 'Content-Type: application/json' -d '%s' http://%s:%s/wm/flow/add-shortest-path/json" % (flow_path_json, ControllerIP, ControllerPort)
100 debug("add_shortest_path_flow %s" % command)
101 result = os.popen(command).read()
102 debug("result %s" % result)
103 # parsedResult = json.loads(result)
104 # debug("parsed %s" % parsedResult)
105 except:
106 log_error("Controller IF has issue")
107 exit(1)
108
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700109def delete_flow_path(flow_id):
110 command = "curl -s \"http://%s:%s/wm/flow/delete/%s/json\"" % (ControllerIP, ControllerPort, flow_id)
111 debug("delete_flow_path %s" % command)
112 result = os.popen(command).read()
113 debug("result %s" % result)
114 # parsedResult = json.loads(result)
115 # debug("parsed %s" % parsedResult)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700116
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700117def extract_flow_args(my_args):
118 # Check the arguments
119 if len(my_args) < 6:
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -0800120 log_error(usage_msg)
121 exit(1)
122
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700123 # Extract the mandatory arguments
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700124 my_flow_id = my_args[0]
125 my_installer_id = my_args[1]
126 my_src_dpid = my_args[2]
127 my_src_port = my_args[3]
128 my_dst_dpid = my_args[4]
129 my_dst_port = my_args[5]
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700130
131 #
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700132 # Extract the "flowPathFlags", "match" and "action" arguments
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700133 #
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700134 flowPathFlags = 0L
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700135 match = {}
136 matchInPortEnabled = True # NOTE: Enabled by default
137 actions = []
138 actionOutputEnabled = True # NOTE: Enabled by default
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700139 idx = 6
140 while idx < len(my_args):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700141 action = {}
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700142 arg1 = my_args[idx]
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700143 idx = idx + 1
144 # Extract the second argument
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700145 if idx >= len(my_args):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700146 error_arg = "ERROR: Missing or invalid '" + arg1 + "' argument"
147 log_error(error_arg)
148 log_error(usage_msg)
149 exit(1)
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700150 arg2 = my_args[idx]
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700151 idx = idx + 1
152
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700153 if arg1 == "flowPathFlags":
154 if "DISCARD_FIRST_HOP_ENTRY" in arg2:
155 flowPathFlags = flowPathFlags + 0x1
156 if "KEEP_ONLY_FIRST_HOP_ENTRY" in arg2:
157 flowPathFlags = flowPathFlags + 0x2
158 elif arg1 == "matchInPort":
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700159 # Just mark whether inPort matching is enabled
160 matchInPortEnabled = arg2 in ['True', 'true']
161 # inPort = {}
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700162 # inPort['value'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700163 # match['inPort'] = inPort
164 ## match['matchInPort'] = True
165 elif arg1 == "matchSrcMac":
166 srcMac = {}
167 srcMac['value'] = arg2
168 match['srcMac'] = srcMac
169 # match['matchSrcMac'] = True
170 elif arg1 == "matchDstMac":
171 dstMac = {}
172 dstMac['value'] = arg2
173 match['dstMac'] = dstMac
174 # match['matchDstMac'] = True
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700175 elif arg1 == "matchEthernetFrameType":
176 match['ethernetFrameType'] = int(arg2, 0)
177 # match['matchEthernetFrameType'] = True
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700178 elif arg1 == "matchVlanId":
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700179 match['vlanId'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700180 # match['matchVlanId'] = True
181 elif arg1 == "matchVlanPriority":
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700182 match['vlanPriority'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700183 # match['matchVlanPriority'] = True
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700184 elif arg1 == "matchSrcIPv4Net":
185 srcIPv4Net = {}
186 srcIPv4Net['value'] = arg2
187 match['srcIPv4Net'] = srcIPv4Net
188 # match['matchSrcIPv4Net'] = True
189 elif arg1 == "matchDstIPv4Net":
190 dstIPv4Net = {}
191 dstIPv4Net['value'] = arg2
192 match['dstIPv4Net'] = dstIPv4Net
193 # match['matchDstIPv4Net'] = True
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700194 elif arg1 == "matchIpProto":
195 match['ipProto'] = int(arg2, 0)
196 # match['matchIpProto'] = True
197 elif arg1 == "matchIpToS":
198 match['ipToS'] = int(arg2, 0)
199 # match['matchIpToS'] = True
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700200 elif arg1 == "matchSrcTcpUdpPort":
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700201 match['srcTcpUdpPort'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700202 # match['matchSrcTcpUdpPort'] = True
203 elif arg1 == "matchDstTcpUdpPort":
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700204 match['dstTcpUdpPort'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700205 # match['matchDstTcpUdpPort'] = True
206 elif arg1 == "actionOutput":
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700207 # Mark whether ACTION_OUTPUT action is enabled
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700208 actionOutputEnabled = arg2 in ['True', 'true']
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700209 # If ACTION_OUTPUT is explicitly enabled, add an entry with a fake
210 # port number. We need this entry to preserve the action ordering.
211 if actionOutputEnabled == True:
212 actionOutput = {}
213 outPort = {}
214 outPort['value'] = 0xffff
215 actionOutput['port'] = outPort
216 actionOutput['maxLen'] = 0
217 action['actionOutput'] = actionOutput
218 # action['actionType'] = 'ACTION_OUTPUT'
219 actions.append(action)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700220 #
221 elif arg1 == "actionSetVlanId":
222 vlanId = {}
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700223 vlanId['vlanId'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700224 action['actionSetVlanId'] = vlanId
225 # action['actionType'] = 'ACTION_SET_VLAN_VID'
226 actions.append(copy.deepcopy(action))
227 elif arg1 == "actionSetVlanPriority":
228 vlanPriority = {}
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700229 vlanPriority['vlanPriority'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700230 action['actionSetVlanPriority'] = vlanPriority
231 # action['actionType'] = 'ACTION_SET_VLAN_PCP'
232 actions.append(copy.deepcopy(action))
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700233 elif arg1 == "actionStripVlan":
234 stripVlan = {}
235 stripVlan['stripVlan'] = arg2 in ['True', 'true']
236 action['actionStripVlan'] = stripVlan
237 # action['actionType'] = 'ACTION_STRIP_VLAN'
238 actions.append(copy.deepcopy(action))
239 elif arg1 == "actionSetEthernetSrcAddr":
240 ethernetSrcAddr = {}
241 ethernetSrcAddr['value'] = arg2
242 setEthernetSrcAddr = {}
243 setEthernetSrcAddr['addr'] = ethernetSrcAddr
244 action['actionSetEthernetSrcAddr'] = setEthernetSrcAddr
245 # action['actionType'] = 'ACTION_SET_DL_SRC'
246 actions.append(copy.deepcopy(action))
247 elif arg1 == "actionSetEthernetDstAddr":
248 ethernetDstAddr = {}
249 ethernetDstAddr['value'] = arg2
250 setEthernetDstAddr = {}
251 setEthernetDstAddr['addr'] = ethernetDstAddr
252 action['actionSetEthernetDstAddr'] = setEthernetDstAddr
253 # action['actionType'] = 'ACTION_SET_DL_DST'
254 actions.append(copy.deepcopy(action))
255 elif arg1 == "actionSetIPv4SrcAddr":
256 IPv4SrcAddr = {}
257 IPv4SrcAddr['value'] = arg2
258 setIPv4SrcAddr = {}
259 setIPv4SrcAddr['addr'] = IPv4SrcAddr
260 action['actionSetIPv4SrcAddr'] = setIPv4SrcAddr
261 # action['actionType'] = 'ACTION_SET_NW_SRC'
262 actions.append(copy.deepcopy(action))
263 elif arg1 == "actionSetIPv4DstAddr":
264 IPv4DstAddr = {}
265 IPv4DstAddr['value'] = arg2
266 setIPv4DstAddr = {}
267 setIPv4DstAddr['addr'] = IPv4DstAddr
268 action['actionSetIPv4DstAddr'] = setIPv4DstAddr
269 # action['actionType'] = 'ACTION_SET_NW_DST'
270 actions.append(copy.deepcopy(action))
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700271 elif arg1 == "actionSetIpToS":
272 ipToS = {}
273 ipToS['ipToS'] = int(arg2, 0)
274 action['actionSetIpToS'] = ipToS
275 # action['actionType'] = 'ACTION_SET_NW_TOS'
276 actions.append(copy.deepcopy(action))
277 elif arg1 == "actionSetTcpUdpSrcPort":
278 tcpUdpSrcPort = {}
279 tcpUdpSrcPort['port'] = int(arg2, 0)
280 action['actionSetTcpUdpSrcPort'] = tcpUdpSrcPort
281 # action['actionType'] = 'ACTION_SET_TP_SRC'
282 actions.append(copy.deepcopy(action))
283 elif arg1 == "actionSetTcpUdpDstPort":
284 tcpUdpDstPort = {}
285 tcpUdpDstPort['port'] = int(arg2, 0)
286 action['actionSetTcpUdpDstPort'] = tcpUdpDstPort
287 # action['actionType'] = 'ACTION_SET_TP_DST'
288 actions.append(copy.deepcopy(action))
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700289 elif arg1 == "actionEnqueue":
290 # TODO: Implement ACTION_ENQUEUE
291 actionEnqueue = {}
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700292 # actionEnqueue['queueId'] = int(arg2, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700293 # enqueuePort = {}
Pavlin Radoslavov14748912013-03-12 15:44:56 -0700294 # enqueuePort['value'] = int(arg3, 0)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700295 # actionEnqueue['port'] = enqueuePort
296 # action['actionEnqueue'] = actionEnqueue
297 # # action['actionType'] = 'ACTION_ENQUEUE'
298 # actions.append(copy.deepcopy(action))
299 #
300 else:
301 log_error("ERROR: Unknown argument '%s'" % (arg1))
302 log_error(usage_msg)
303 exit(1)
304
Pavlin Radoslavov8a002d92013-03-13 20:20:43 -0700305 return {
306 'my_flow_id' : my_flow_id,
307 'my_installer_id' : my_installer_id,
308 'my_src_dpid' : my_src_dpid,
309 'my_src_port' : my_src_port,
310 'my_dst_dpid' : my_dst_dpid,
311 'my_dst_port' : my_dst_port,
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700312 'flowPathFlags' : flowPathFlags,
Pavlin Radoslavov8a002d92013-03-13 20:20:43 -0700313 'match' : match,
314 'matchInPortEnabled' : matchInPortEnabled,
315 'actions' : actions,
316 'actionOutputEnabled' : actionOutputEnabled
317 }
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700318
319def compute_data_path(parsed_args):
320
321 my_src_dpid = parsed_args['my_src_dpid']
322 my_src_port = parsed_args['my_src_port']
323 my_dst_dpid = parsed_args['my_dst_dpid']
324 my_dst_port = parsed_args['my_dst_port']
325
326 # Compute the shortest path
327 data_path = shortest_path(my_src_dpid, my_src_port, my_dst_dpid, my_dst_port)
328
329 debug("Data Path: %s" % data_path)
330 return data_path
331
332def compute_flow_path(parsed_args, data_path):
333
334 my_flow_id = parsed_args['my_flow_id']
335 my_installer_id = parsed_args['my_installer_id']
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700336 myFlowPathFlags = parsed_args['flowPathFlags']
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700337 match = parsed_args['match']
338 matchInPortEnabled = parsed_args['matchInPortEnabled']
339 actions = parsed_args['actions']
340 actionOutputEnabled = parsed_args['actionOutputEnabled']
341 my_data_path = copy.deepcopy(data_path)
342
343 flow_id = {}
344 flow_id['value'] = my_flow_id
345 installer_id = {}
346 installer_id['value'] = my_installer_id
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700347 flowPathFlags = {}
348 flowPathFlags['flags'] = myFlowPathFlags
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700349
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700350 flowEntryActions = {}
351
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700352 flow_path = {}
353 flow_path['flowId'] = flow_id
354 flow_path['installerId'] = installer_id
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700355 flow_path['flowPathFlags'] = flowPathFlags
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700356
Pavlin Radoslavov67b3ef32013-04-03 02:44:48 -0700357 if (len(match) > 0):
358 flow_path['flowEntryMatch'] = copy.deepcopy(match)
359
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700360 #
361 # Add the match conditions to each flow entry
362 #
363 if (len(match) > 0) or matchInPortEnabled:
364 idx = 0
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700365 while idx < len(my_data_path['flowEntries']):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700366 if matchInPortEnabled:
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700367 inPort = my_data_path['flowEntries'][idx]['inPort']
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700368 match['inPort'] = copy.deepcopy(inPort)
369 # match['matchInPort'] = True
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700370 my_data_path['flowEntries'][idx]['flowEntryMatch'] = copy.deepcopy(match)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700371 idx = idx + 1
372
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700373
374 if (len(actions) > 0):
375 flowEntryActions['actions'] = copy.deepcopy(actions)
376 flow_path['flowEntryActions'] = flowEntryActions
377
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700378 #
379 # Set the actions for each flow entry
380 # NOTE: The actions from the command line are aplied
381 # ONLY to the first flow entry.
382 #
383 # If ACTION_OUTPUT action is enabled, then apply it
384 # to each flow entry.
385 #
386 if (len(actions) > 0) or actionOutputEnabled:
387 idx = 0
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700388 while idx < len(my_data_path['flowEntries']):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700389 if idx > 0:
390 actions = [] # Reset the actions for all but first entry
391 action = {}
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700392 outPort = my_data_path['flowEntries'][idx]['outPort']
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700393 actionOutput = {}
394 actionOutput['port'] = copy.deepcopy(outPort)
395 # actionOutput['maxLen'] = 0 # TODO: not used for now
396 action['actionOutput'] = copy.deepcopy(actionOutput)
397 # action['actionType'] = 'ACTION_OUTPUT'
398 actions.append(copy.deepcopy(action))
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700399 flowEntryActions = {}
400 flowEntryActions['actions'] = copy.deepcopy(actions)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700401
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700402 my_data_path['flowEntries'][idx]['flowEntryActions'] = flowEntryActions
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700403 idx = idx + 1
404
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700405 flow_path['dataPath'] = my_data_path
406 debug("Flow Path: %s" % flow_path)
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700407 return flow_path
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -0800408
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700409def exec_monitoring_by_onos(parsed_args):
410 idx = 0
411 while idx < len(parsed_args):
412 data_path = {}
413 src_dpid = {}
414 src_port = {}
415 dst_dpid = {}
416 dst_port = {}
417 src_switch_port = {}
418 dst_switch_port = {}
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700419 flow_entries = []
420
421 src_dpid['value'] = parsed_args[idx]['my_src_dpid']
422 src_port['value'] = parsed_args[idx]['my_src_port']
423 dst_dpid['value'] = parsed_args[idx]['my_dst_dpid']
424 dst_port['value'] = parsed_args[idx]['my_dst_port']
425 src_switch_port['dpid'] = src_dpid
426 src_switch_port['port'] = src_port
427 dst_switch_port['dpid'] = dst_dpid
428 dst_switch_port['port'] = dst_port
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700429
430 data_path['srcPort'] = copy.deepcopy(src_switch_port)
431 data_path['dstPort'] = copy.deepcopy(dst_switch_port)
432 data_path['flowEntries'] = copy.deepcopy(flow_entries)
433
434 #
435 # XXX: Explicitly disable the InPort matching, and
436 # the Output action, because they get in the way
437 # during the compute_flow_path() processing.
438 #
439 parsed_args[idx]['matchInPortEnabled'] = False
440 parsed_args[idx]['actionOutputEnabled'] = False
441
442 flow_path = compute_flow_path(parsed_args[idx], data_path)
443 add_shortest_path_flow(flow_path)
444
445 idx = idx + 1
446
447
448def exec_processing_by_script(parsed_args):
449 #
450 # Initialization
451 #
452 last_data_paths = []
453 idx = 0
454 while idx < len(parsed_args):
455 last_data_path = []
456 last_data_paths.append(copy.deepcopy(last_data_path))
457 idx = idx + 1
458
459 #
460 # Do the work: install and/or periodically monitor each flow
461 #
462 while True:
463 idx = 0
464 while idx < len(parsed_args):
465 last_data_path = last_data_paths[idx]
466 my_flow_id = parsed_args[idx]['my_flow_id']
467 data_path = compute_data_path(parsed_args[idx])
468 if data_path != last_data_path:
469 print_data_path(data_path)
470 if len(last_data_path) > 0:
471 delete_flow_path(my_flow_id)
472 if len(data_path) > 0:
473 flow_path = compute_flow_path(parsed_args[idx], data_path)
474 add_flow_path(flow_path)
475 last_data_paths[idx] = copy.deepcopy(data_path)
476 idx = idx + 1
477
478 if MonitoringEnabled != True:
479 break
480 time.sleep(1)
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -0800481
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700482
483if __name__ == "__main__":
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700484 usage_msg = "Usage: %s [Flags] <flow-id> <installer-id> <src-dpid> <src-port> <dest-dpid> <dest-port> [Flow Path Flags] [Match Conditions] [Actions]\n" % (sys.argv[0])
Pavlin Radoslavov7be3bac2013-03-27 09:59:34 -0700485 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700486 usage_msg = usage_msg + " Flags:\n"
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700487 usage_msg = usage_msg + " -m [monitorname] Monitor and maintain the installed shortest path(s)\n"
488 usage_msg = usage_msg + " If 'monitorname' is specified and is set to 'ONOS'\n"
489 usage_msg = usage_msg + " ((case insensitive), then the flow generation and\n"
490 usage_msg = usage_msg + " maintanenance is done by ONOS itself.\n"
491 usage_msg = usage_msg + " Otherwise, it is done by this script.\n"
492 usage_msg = usage_msg + " -f <filename> Read the flow(s) to install from a file\n"
493 usage_msg = usage_msg + " File format: one line per flow starting with <flow-id>\n"
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700494 usage_msg = usage_msg + "\n"
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700495 usage_msg = usage_msg + " Flow Path Flags:\n"
496 usage_msg = usage_msg + " flowPathFlags <Flags> (flag names separated by ',')\n"
497 usage_msg = usage_msg + "\n"
498 usage_msg = usage_msg + " Known flags:\n"
499 usage_msg = usage_msg + " DISCARD_FIRST_HOP_ENTRY : Discard the first-hop flow entry\n"
500 usage_msg = usage_msg + " KEEP_ONLY_FIRST_HOP_ENTRY : Keep only the first-hop flow entry\n"
501 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700502 usage_msg = usage_msg + " Match Conditions:\n"
503 usage_msg = usage_msg + " matchInPort <True|False> (default to True)\n"
504 usage_msg = usage_msg + " matchSrcMac <source MAC address>\n"
505 usage_msg = usage_msg + " matchDstMac <destination MAC address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700506 usage_msg = usage_msg + " matchEthernetFrameType <Ethernet frame type>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700507 usage_msg = usage_msg + " matchVlanId <VLAN ID>\n"
508 usage_msg = usage_msg + " matchVlanPriority <VLAN priority>\n"
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700509 usage_msg = usage_msg + " matchSrcIPv4Net <source IPv4 network address>\n"
510 usage_msg = usage_msg + " matchDstIPv4Net <destination IPv4 network address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700511 usage_msg = usage_msg + " matchIpProto <IP protocol>\n"
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700512 usage_msg = usage_msg + " matchIpToS <IP ToS (DSCP field, 6 bits)>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700513 usage_msg = usage_msg + " matchSrcTcpUdpPort <source TCP/UDP port>\n"
514 usage_msg = usage_msg + " matchDstTcpUdpPort <destination TCP/UDP port>\n"
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700515 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700516 usage_msg = usage_msg + " Actions:\n"
517 usage_msg = usage_msg + " actionOutput <True|False> (default to True)\n"
Pavlin Radoslavovc1bafd12013-07-12 17:00:35 -0700518 usage_msg = usage_msg + " actionSetVlanId <VLAN ID>\n"
519 usage_msg = usage_msg + " actionSetVlanPriority <VLAN priority>\n"
520 usage_msg = usage_msg + " actionStripVlan <True|False>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700521 usage_msg = usage_msg + " actionSetEthernetSrcAddr <source MAC address>\n"
522 usage_msg = usage_msg + " actionSetEthernetDstAddr <destination MAC address>\n"
523 usage_msg = usage_msg + " actionSetIPv4SrcAddr <source IPv4 address>\n"
524 usage_msg = usage_msg + " actionSetIPv4DstAddr <destination IPv4 address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700525 usage_msg = usage_msg + " actionSetIpToS <IP ToS (DSCP field, 6 bits)>\n"
526 usage_msg = usage_msg + " actionSetTcpUdpSrcPort <source TCP/UDP port>\n"
527 usage_msg = usage_msg + " actionSetTcpUdpDstPort <destination TCP/UDP port>\n"
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700528 usage_msg = usage_msg + " Actions (not implemented yet):\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700529 usage_msg = usage_msg + " actionEnqueue <dummy argument>\n"
530
531 # app.debug = False;
532
533 # Usage info
534 if len(sys.argv) > 1 and (sys.argv[1] == "-h" or sys.argv[1] == "--help"):
535 print(usage_msg)
536 exit(0)
537
538 #
539 # Check the flags
540 #
541 start_argv_index = 1
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700542 idx = 1
543 while idx < len(sys.argv):
544 arg1 = sys.argv[idx]
545 idx = idx + 1
546 if arg1 == "-m":
547 MonitoringEnabled = True
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700548 if idx < len(sys.argv):
549 arg2 = sys.argv[idx]
550 if arg2.lower() == "onos":
551 MonitoringByOnos = True
552 idx = idx + 1
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700553 start_argv_index = idx
554 elif arg1 == "-f":
555 if idx >= len(sys.argv):
556 error_arg = "ERROR: Missing or invalid '" + arg1 + "' argument"
557 log_error(error_arg)
558 log_error(usage_msg)
559 exit(1)
560 ReadFromFile = sys.argv[idx]
561 idx = idx + 1
562 start_argv_index = idx
563 else:
564 break;
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700565
566 #
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700567 # Read the arguments from a file or from the remaining command line options
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700568 #
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700569 my_lines = []
570 if len(ReadFromFile) > 0:
571 f = open(ReadFromFile, "rt")
572 my_line = f.readline()
573 while my_line:
574 if len(my_line.rstrip()) > 0 and my_line[0] != "#":
575 my_token_line = my_line.rstrip().split()
576 my_lines.append(my_token_line)
577 my_line = f.readline()
578 else:
579 my_lines.append(copy.deepcopy(sys.argv[start_argv_index:]))
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700580
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700581 #
582 # Initialization
583 #
584 last_data_paths = []
585 parsed_args = []
586 idx = 0
587 while idx < len(my_lines):
588 last_data_path = []
589 last_data_paths.append(copy.deepcopy(last_data_path))
590 #
591 # Parse the flow arguments
592 #
593 my_args = my_lines[idx]
594 parsed_args.append(copy.deepcopy(extract_flow_args(my_args)))
595 # Cleanup leftover state
596 my_flow_id = parsed_args[idx]['my_flow_id']
597 delete_flow_path(my_flow_id)
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700598
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700599 idx = idx + 1
600
601 #
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700602 if MonitoringByOnos == True:
603 exec_monitoring_by_onos(parsed_args)
604 else:
605 exec_processing_by_script(parsed_args)
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700606