blob: 0efa5504e13193cd04ec91ac8db559b90111948e [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 Radoslavovd28cf7c2013-10-26 11:27:43 -0700355 # NOTE: The 'flowPathType' might be rewritten later
356 flow_path['flowPathType'] = 'FP_TYPE_EXPLICIT_PATH'
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700357 flow_path['flowPathFlags'] = flowPathFlags
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700358
Pavlin Radoslavov67b3ef32013-04-03 02:44:48 -0700359 if (len(match) > 0):
360 flow_path['flowEntryMatch'] = copy.deepcopy(match)
361
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700362 #
363 # Add the match conditions to each flow entry
364 #
365 if (len(match) > 0) or matchInPortEnabled:
366 idx = 0
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700367 while idx < len(my_data_path['flowEntries']):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700368 if matchInPortEnabled:
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700369 inPort = my_data_path['flowEntries'][idx]['inPort']
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700370 match['inPort'] = copy.deepcopy(inPort)
371 # match['matchInPort'] = True
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700372 my_data_path['flowEntries'][idx]['flowEntryMatch'] = copy.deepcopy(match)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700373 idx = idx + 1
374
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700375
376 if (len(actions) > 0):
377 flowEntryActions['actions'] = copy.deepcopy(actions)
378 flow_path['flowEntryActions'] = flowEntryActions
379
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700380 #
381 # Set the actions for each flow entry
382 # NOTE: The actions from the command line are aplied
383 # ONLY to the first flow entry.
384 #
385 # If ACTION_OUTPUT action is enabled, then apply it
386 # to each flow entry.
387 #
388 if (len(actions) > 0) or actionOutputEnabled:
389 idx = 0
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700390 while idx < len(my_data_path['flowEntries']):
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700391 if idx > 0:
392 actions = [] # Reset the actions for all but first entry
393 action = {}
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700394 outPort = my_data_path['flowEntries'][idx]['outPort']
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700395 actionOutput = {}
396 actionOutput['port'] = copy.deepcopy(outPort)
397 # actionOutput['maxLen'] = 0 # TODO: not used for now
398 action['actionOutput'] = copy.deepcopy(actionOutput)
399 # action['actionType'] = 'ACTION_OUTPUT'
400 actions.append(copy.deepcopy(action))
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700401 flowEntryActions = {}
402 flowEntryActions['actions'] = copy.deepcopy(actions)
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700403
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700404 my_data_path['flowEntries'][idx]['flowEntryActions'] = flowEntryActions
Pavlin Radoslavovf13923a2013-03-11 19:42:17 -0700405 idx = idx + 1
406
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700407 flow_path['dataPath'] = my_data_path
408 debug("Flow Path: %s" % flow_path)
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700409 return flow_path
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -0800410
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700411def exec_monitoring_by_onos(parsed_args):
412 idx = 0
413 while idx < len(parsed_args):
414 data_path = {}
415 src_dpid = {}
416 src_port = {}
417 dst_dpid = {}
418 dst_port = {}
419 src_switch_port = {}
420 dst_switch_port = {}
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700421 flow_entries = []
422
423 src_dpid['value'] = parsed_args[idx]['my_src_dpid']
424 src_port['value'] = parsed_args[idx]['my_src_port']
425 dst_dpid['value'] = parsed_args[idx]['my_dst_dpid']
426 dst_port['value'] = parsed_args[idx]['my_dst_port']
427 src_switch_port['dpid'] = src_dpid
428 src_switch_port['port'] = src_port
429 dst_switch_port['dpid'] = dst_dpid
430 dst_switch_port['port'] = dst_port
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700431
432 data_path['srcPort'] = copy.deepcopy(src_switch_port)
433 data_path['dstPort'] = copy.deepcopy(dst_switch_port)
434 data_path['flowEntries'] = copy.deepcopy(flow_entries)
435
436 #
437 # XXX: Explicitly disable the InPort matching, and
438 # the Output action, because they get in the way
439 # during the compute_flow_path() processing.
440 #
441 parsed_args[idx]['matchInPortEnabled'] = False
442 parsed_args[idx]['actionOutputEnabled'] = False
443
444 flow_path = compute_flow_path(parsed_args[idx], data_path)
Pavlin Radoslavovd28cf7c2013-10-26 11:27:43 -0700445 flow_path['flowPathType'] = 'FP_TYPE_SHORTEST_PATH'
446
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700447 add_shortest_path_flow(flow_path)
448
449 idx = idx + 1
450
451
452def exec_processing_by_script(parsed_args):
453 #
454 # Initialization
455 #
456 last_data_paths = []
457 idx = 0
458 while idx < len(parsed_args):
459 last_data_path = []
460 last_data_paths.append(copy.deepcopy(last_data_path))
461 idx = idx + 1
462
463 #
464 # Do the work: install and/or periodically monitor each flow
465 #
466 while True:
467 idx = 0
468 while idx < len(parsed_args):
469 last_data_path = last_data_paths[idx]
470 my_flow_id = parsed_args[idx]['my_flow_id']
471 data_path = compute_data_path(parsed_args[idx])
472 if data_path != last_data_path:
473 print_data_path(data_path)
474 if len(last_data_path) > 0:
475 delete_flow_path(my_flow_id)
476 if len(data_path) > 0:
477 flow_path = compute_flow_path(parsed_args[idx], data_path)
478 add_flow_path(flow_path)
479 last_data_paths[idx] = copy.deepcopy(data_path)
480 idx = idx + 1
481
482 if MonitoringEnabled != True:
483 break
484 time.sleep(1)
Pavlin Radoslavovf4ad9892013-03-04 14:15:19 -0800485
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700486
487if __name__ == "__main__":
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700488 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 -0700489 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700490 usage_msg = usage_msg + " Flags:\n"
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700491 usage_msg = usage_msg + " -m [monitorname] Monitor and maintain the installed shortest path(s)\n"
492 usage_msg = usage_msg + " If 'monitorname' is specified and is set to 'ONOS'\n"
493 usage_msg = usage_msg + " ((case insensitive), then the flow generation and\n"
494 usage_msg = usage_msg + " maintanenance is done by ONOS itself.\n"
495 usage_msg = usage_msg + " Otherwise, it is done by this script.\n"
496 usage_msg = usage_msg + " -f <filename> Read the flow(s) to install from a file\n"
497 usage_msg = usage_msg + " File format: one line per flow starting with <flow-id>\n"
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700498 usage_msg = usage_msg + "\n"
Pavlin Radoslavov204b2862013-07-12 14:15:36 -0700499 usage_msg = usage_msg + " Flow Path Flags:\n"
500 usage_msg = usage_msg + " flowPathFlags <Flags> (flag names separated by ',')\n"
501 usage_msg = usage_msg + "\n"
502 usage_msg = usage_msg + " Known flags:\n"
503 usage_msg = usage_msg + " DISCARD_FIRST_HOP_ENTRY : Discard the first-hop flow entry\n"
504 usage_msg = usage_msg + " KEEP_ONLY_FIRST_HOP_ENTRY : Keep only the first-hop flow entry\n"
505 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700506 usage_msg = usage_msg + " Match Conditions:\n"
507 usage_msg = usage_msg + " matchInPort <True|False> (default to True)\n"
508 usage_msg = usage_msg + " matchSrcMac <source MAC address>\n"
509 usage_msg = usage_msg + " matchDstMac <destination MAC address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700510 usage_msg = usage_msg + " matchEthernetFrameType <Ethernet frame type>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700511 usage_msg = usage_msg + " matchVlanId <VLAN ID>\n"
512 usage_msg = usage_msg + " matchVlanPriority <VLAN priority>\n"
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700513 usage_msg = usage_msg + " matchSrcIPv4Net <source IPv4 network address>\n"
514 usage_msg = usage_msg + " matchDstIPv4Net <destination IPv4 network address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700515 usage_msg = usage_msg + " matchIpProto <IP protocol>\n"
Pavlin Radoslavovad3a1e62013-07-09 13:30:16 -0700516 usage_msg = usage_msg + " matchIpToS <IP ToS (DSCP field, 6 bits)>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700517 usage_msg = usage_msg + " matchSrcTcpUdpPort <source TCP/UDP port>\n"
518 usage_msg = usage_msg + " matchDstTcpUdpPort <destination TCP/UDP port>\n"
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700519 usage_msg = usage_msg + "\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700520 usage_msg = usage_msg + " Actions:\n"
521 usage_msg = usage_msg + " actionOutput <True|False> (default to True)\n"
Pavlin Radoslavovc1bafd12013-07-12 17:00:35 -0700522 usage_msg = usage_msg + " actionSetVlanId <VLAN ID>\n"
523 usage_msg = usage_msg + " actionSetVlanPriority <VLAN priority>\n"
524 usage_msg = usage_msg + " actionStripVlan <True|False>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700525 usage_msg = usage_msg + " actionSetEthernetSrcAddr <source MAC address>\n"
526 usage_msg = usage_msg + " actionSetEthernetDstAddr <destination MAC address>\n"
527 usage_msg = usage_msg + " actionSetIPv4SrcAddr <source IPv4 address>\n"
528 usage_msg = usage_msg + " actionSetIPv4DstAddr <destination IPv4 address>\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700529 usage_msg = usage_msg + " actionSetIpToS <IP ToS (DSCP field, 6 bits)>\n"
530 usage_msg = usage_msg + " actionSetTcpUdpSrcPort <source TCP/UDP port>\n"
531 usage_msg = usage_msg + " actionSetTcpUdpDstPort <destination TCP/UDP port>\n"
Pavlin Radoslavov1bc2c472013-07-17 18:11:37 -0700532 usage_msg = usage_msg + " Actions (not implemented yet):\n"
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700533 usage_msg = usage_msg + " actionEnqueue <dummy argument>\n"
534
535 # app.debug = False;
536
537 # Usage info
538 if len(sys.argv) > 1 and (sys.argv[1] == "-h" or sys.argv[1] == "--help"):
539 print(usage_msg)
540 exit(0)
541
542 #
543 # Check the flags
544 #
545 start_argv_index = 1
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700546 idx = 1
547 while idx < len(sys.argv):
548 arg1 = sys.argv[idx]
549 idx = idx + 1
550 if arg1 == "-m":
551 MonitoringEnabled = True
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700552 if idx < len(sys.argv):
553 arg2 = sys.argv[idx]
554 if arg2.lower() == "onos":
555 MonitoringByOnos = True
556 idx = idx + 1
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700557 start_argv_index = idx
558 elif arg1 == "-f":
559 if idx >= len(sys.argv):
560 error_arg = "ERROR: Missing or invalid '" + arg1 + "' argument"
561 log_error(error_arg)
562 log_error(usage_msg)
563 exit(1)
564 ReadFromFile = sys.argv[idx]
565 idx = idx + 1
566 start_argv_index = idx
567 else:
568 break;
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700569
570 #
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700571 # Read the arguments from a file or from the remaining command line options
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700572 #
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700573 my_lines = []
574 if len(ReadFromFile) > 0:
575 f = open(ReadFromFile, "rt")
576 my_line = f.readline()
577 while my_line:
578 if len(my_line.rstrip()) > 0 and my_line[0] != "#":
579 my_token_line = my_line.rstrip().split()
580 my_lines.append(my_token_line)
581 my_line = f.readline()
582 else:
583 my_lines.append(copy.deepcopy(sys.argv[start_argv_index:]))
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700584
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700585 #
586 # Initialization
587 #
588 last_data_paths = []
589 parsed_args = []
590 idx = 0
591 while idx < len(my_lines):
592 last_data_path = []
593 last_data_paths.append(copy.deepcopy(last_data_path))
594 #
595 # Parse the flow arguments
596 #
597 my_args = my_lines[idx]
598 parsed_args.append(copy.deepcopy(extract_flow_args(my_args)))
599 # Cleanup leftover state
600 my_flow_id = parsed_args[idx]['my_flow_id']
601 delete_flow_path(my_flow_id)
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700602
Pavlin Radoslavov2a0bd8b2013-03-15 18:45:51 -0700603 idx = idx + 1
604
605 #
Pavlin Radoslavovb549f082013-03-29 04:28:27 -0700606 if MonitoringByOnos == True:
607 exec_monitoring_by_onos(parsed_args)
608 else:
609 exec_processing_by_script(parsed_args)
Pavlin Radoslavov3f3778a2013-03-13 08:16:57 -0700610